auto_mount_usb_cdrom.c

自动挂载 USB CDROM的程序代码,很好的东西哦

#include <stdio.h>

#include <stdlib.h>

#include <error.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <string.h>

#include <sys/mount.h>

#include <unistd.h>

#include <pthread.h>

 

//setMsg

extern void set_ALERT_Msg(char * buf);

 

 

#define FILE_DEV_CHECK "/proc/scsi/scsi" //用来检测设备数和类型

#define FILE_MOUNT_CHECK "/proc/mounts"  //用来检测设备是否被mount

#define FILE_DISC_PARTS_CHECK "/proc/partitions" //用来检测设备的分区情况

 

#define FILE_DEV_STATU_TEMPL "/proc/scsi/usb-storage-%d/%d"   //用来检测某一设备是否在插入状态

#define FILE_DEV_PART_TEMPL "/dev/scsi/host%d/bus0/target0/lun0/" //具体的设备

 

#define USB_CDROM_MP "/tmp/cdrom"

#define USB_DISK_MP "/tmp/usbdisk"

 

#define DBG_PRINT(x) printf("[AutoMounter]");printf x

#define DBG_PRINT2(x) printf("[AutoMounter]");printf x

 

#define MAX_NAME_LEN 64

#define MAX_PART_NUM 6   //最多允许6个分区

typedef struct s_scsi_usb_dev

{

       int type; /*1 cdrom  2 disk */

       int index; /*like host0  host1*/

       char file_statu[MAX_NAME_LEN]; /*like "/proc/scsi/usb-storage-%d/%d"*/

       char devfile_parts[MAX_PART_NUM][MAX_NAME_LEN]; /*存储每个分区的设备文件*/

       char mount_path[MAX_PART_NUM][MAX_NAME_LEN]; /*与上面对应的mount点*/

       int part_num; //分区数

       struct s_scsi_usb_dev * next_dev;  //指向下一个备

} SCSI_USB_DEV;

static int is_manual_umount = 1;

static SCSI_USB_DEV * f_first_dev = NULL;

 

 

static int CHECK_PARTS(SCSI_USB_DEV * dev);

static int  ADD_DEV(SCSI_USB_DEV * dev);

static int INIT_DEV(SCSI_USB_DEV * dev, int index, char * type);

static void CLEAR_DEV(void);

 

static int find_device(void);

static int check_attach(SCSI_USB_DEV * dev);

static int check_mount(SCSI_USB_DEV * dev);

static int do_mount(SCSI_USB_DEV * dev);

static int do_umount(SCSI_USB_DEV * dev);

static int process_dev(SCSI_USB_DEV * dev);

 

static void * main_thread(void);

 

//device opt fuction

static int  ADD_DEV(SCSI_USB_DEV * dev)

{

       if(f_first_dev)

       {

              dev->next_dev = f_first_dev;

              f_first_dev = dev;

       }

       else

       {

              f_first_dev = dev;

              dev->next_dev = NULL;

       }

       return 1;

}

static int INIT_DEV(SCSI_USB_DEV * dev, int index, char * type)

{

       dev->index = index;

       sprintf(dev->file_statu, FILE_DEV_STATU_TEMPL, index, index);

       

       if (!strncmp(type, "CD-ROM", 6))

       {//usb cd-rom

              dev->type = 1;

              dev->part_num = 1;

              sprintf(dev->devfile_parts[0], FILE_DEV_PART_TEMPL"cd", index);

              strcpy(dev->mount_path[0],USB_CDROM_MP);

              DBG_PRINT(("---%d---,%s,%s\n", dev->index, dev->devfile_parts[0],dev->mount_path[0]));

       }

       else

       {//usb disk

              dev->type = 2;

              dev->part_num = CHECK_PARTS(dev);

       }

       

       return 1;

}

static int CHECK_PARTS(SCSI_USB_DEV * dev)

{

       char buf[1024];

       char hoststr[16];

       int fd,len;

       int part_num = 0;

 

       char * line;

       char * delim="\n";

       char * strtok_tmp_ptr;

       char * seek;

       char * part_blocks;

       

       fd = open(FILE_DISC_PARTS_CHECK, O_RDONLY);

       if( fd > 0 )

       {

              len = read(fd, buf, sizeof(buf));

              close(fd);

              if (len > 0)

              {

                     sprintf(hoststr,"scsi/host%d", dev->index);

                     line = strtok_r(buf, delim, &strtok_tmp_ptr);

                     while(line)

                     {

                            seek = strstr(line, hoststr);

                            if (seek)

                            {

                                   part_blocks = seek - 3;

                                   if (strncmp(part_blocks, " 1 ", 3))

                                   {//not extend part

                                          sprintf(dev->devfile_parts[part_num],"/dev/%s", seek);

                                          sprintf(dev->mount_path[part_num], USB_DISK_MP"/disk%d/part%d",dev->index,part_num);

                                          DBG_PRINT(("---%d---,%s,%s\n", dev->index, dev->devfile_parts[part_num],dev->mount_path[part_num]));

                                          part_num ++;

                                          if (part_num == MAX_PART_NUM)

                                          {

                                                 break;//too many parts ignore 

                                          }

                                   }

                            }

                            line = strtok_r(NULL, delim, &strtok_tmp_ptr);

                     }

              }

       }

       else

       {

              perror(FILE_DISC_PARTS_CHECK);

       }

       return part_num;

}

static void CLEAR_DEV(void)

{

       SCSI_USB_DEV * cur_dev = f_first_dev;

       SCSI_USB_DEV * tmp_dev;

       

       while (cur_dev)

       {

              tmp_dev = cur_dev;

              cur_dev = cur_dev->next_dev;

              free(tmp_dev);

       }

       f_first_dev = NULL;

}

 

//

static int find_device()

{//搜索设备加入列表

       char buf[1024];

       int fd;

       int len;

       int dev_num = 0;

       char * seek = NULL;

       SCSI_USB_DEV * new_dev;

       //clear exist device

       CLEAR_DEV();

 

       //add new device

       fd = open(FILE_DEV_CHECK, O_RDONLY);

       if( fd > 0 )

       {

              len = read(fd, buf, sizeof(buf));

              close(fd);

              if ( len > 0 )

              {

                     //printf("%d\n", len);

                     seek = buf;

                     while(seek)

                     {

                            seek = strstr(seek, "Host: scsi");

                            if (seek)

                            {

                                   seek += strlen( "Host: scsi");

                                   seek = strstr(seek, "Type:");

                                   if (seek)

                                   {

                                          seek += strlen("Type:");

                                          while(*seek == ' ') seek++;

                                          new_dev = malloc(sizeof(SCSI_USB_DEV));

                                          INIT_DEV(new_dev, dev_num, seek);

                                          ADD_DEV(new_dev);       

                                          dev_num ++;

                                   }

                            }

                     }

                     DBG_PRINT(("dev_num = %d\n", dev_num));

              }

       }

       else

       {

              perror(FILE_DEV_CHECK);

       }

       return dev_num;

}

 

 

static int check_attach(SCSI_USB_DEV * dev)

{//检测设备是否连接

//linux中只要设备被插入过就会记住该设备

//只能通过Attached来判断是否连接

 

       char buf[512];

       int fd;

       int len;

       char * seek;

       

       fd = open(dev->file_statu, O_RDONLY);

       if( fd > 0 )

       {

              len = read(fd, buf, sizeof(buf));

              close(fd);

              if ( len > 0 )

              {

                     seek = strstr(buf, "Attached:");

                     if (seek)

                     {

                            seek += strlen( "Attached:");

                            while(*seek == ' ') seek++;

                            return *seek=='Y';

                     }

              }

       }

       else

       {

              perror(dev->file_statu);

       }

       return 0;

}

static int check_mount(SCSI_USB_DEV * dev)

{

       char buf[1024];

       int fd;

       int len;

       char * seek;

       int i = 0;

       

       fd = open(FILE_MOUNT_CHECK, O_RDONLY);

       if( fd > 0 )

       {

              len = read(fd, buf, sizeof(buf));

              close(fd);

              if ( len > 0 )

              {

                     buf[len] = '\0';

                     if(dev->type == 2 && dev->part_num > 1)  i ++; /*if disk ignore first part disc*/

                     for(; i<dev->part_num; i++)

                     {

                            //printf("index=%d type=%d %s\n",dev->index,dev->type,dev->devfile_parts[i]);

                            seek = strstr(buf, dev->devfile_parts[i]);

                            if  (seek!=NULL) return 1;/*have one part mounted return 1*/