linux下u盘检测程序

    
  获得U盘的插入或者拔取得信息的传统方法是在内核级运行hotplug程序，相关参数通过环境变量传递过来，再由hotplug通知其他关注hotplug的应用程序,但是效率比较低.

     网上查找知道:

   用户空间的程序与设备通信的方法，主要有以下几种方式，

　　3. 用write/read/mmap去操作/dev目录下对应的设备，这也是设备驱动程序提供的接口。像framebuffer等都是这样做的。

      
上面的方法在大多数情况下，都可以正常工作，但是对于热插拨(hotplug)的设备，比如像U盘，就有点困难了，因为不知道：什么时候设备插上了，什么时候设备拔掉了。这就是所谓的hotplug问题了。

   新的方法是采用NETLINK实现的，这是一种特殊类型的socket，专门用于内核空间与用户空间的异步通信。

   先说明几个总要的结构体:

  sockaddr_nl结构：

}nl;

 
int setsockopt(

贴出代码:

#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <stdlib.h>
#include <linux/netlink.h>
#include <dirent.h>
#include <sys/statfs.h>

int init_socket()
{
	struct sockaddr_nl snl;
	const int BufferSize= 1024;
	int retval;

	memset(&snl,0,sizeof(struct sockaddr_nl));
	snl.nl_family = AF_NETLINK;
	snl.nl_pid = getpid();
	snl.nl_groups = 1;

	int Sock_id = socket(PF_NETLINK,SOCK_DGRAM,NETLINK_KOBJECT_UEVENT);

	if(Sock_id == -1)
	printf("sock err:%m\n"),exit(-1);

	// set reveive buffer
	setsockopt(Sock_id,SOL_SOCKET,SO_RCVBUFFORCE,&BufferSize,sizeof(BufferSize));
	retval = bind(Sock_id,(struct sockaddr*)&snl,sizeof(struct sockaddr_nl));
	if(retval==-1)
	printf("bind err:%m\n"),close(Sock_id),exit(-1);
	return Sock_id;
}

// 该函数主要作用时检测u盘的 总空间,剩余空间,剩余空间百分比
double GetDiskFreeSpacePercent(const char *pDisk,double* freespace,double*  totalspace)
{
	struct statfs disk_statfs;
	double freeSpacePercent =0;
	if(statfs(pDisk,&disk_statfs) == 0)
	{

		*freespace = (disk_statfs.f_bsize * disk_statfs.f_bfree) / (1024*1024*1024.0);
		*totalspace = (disk_statfs.f_bsize * disk_statfs.f_blocks) / (1024*1024*1024.0);
	}
	return freeSpacePercent = (*freespace)/(*totalspace)*100;

}

#define BUFFER_SIZE 2048

int main()
{
	DIR *dp;
	double f=0;
	double t=0;
	double percent=0;
	const char* path="/media/cjl/disk";
	int sd= init_socket();
	while(1)
	{
		char buf[BUFFER_SIZE] = {0};
		recv(sd,&buf,sizeof(buf),0);
		//printf("%s\n",buf);
			if(!memcmp(buf,"[email protected]",4) /*&& !memcmp(&buf[strlen(buf) - 4],"/sdb",4)*/)
		{
			printf("Found U Disk\n");
			break;
		}
	}
	printf("是否打开u盘Y/N\n");
	char c;
	scanf("%c",&c);
	if(c=='Y' || c=='y')
	{
		if((dp = opendir(path)) ==NULL)
		{
			printf("打开失败!\n");
		}
		else
		{
			system("ls -l /media/cjl/disk");
		}
	}
	else if(c=='N' || c=='n')

	percent = GetDiskFreeSpacePercent(path,&f,&t);

	printf("u盘剩余空间: %.2f\n",f);
	printf("u盘总空间:   %.2f\n",t);
	printf("u盘剩余空间百分比: %0.2f%%\n",percent);
	return 0;
}