Linux 下读取目录下文件，并将文件信息写入mysql数据库

chunshui

1. #include <stdio.h>
   
2. #include <string.h>
   
3. #include <unistd.h>
   
4. #include <sys/stat.h>
   
5. #include <sys/types.h>
   
6. #include <stdlib.h>
   
7. #include <errno.h>
   
8. #include <fcntl.h>
   
9. #include <dirent.h>
   
10. #include <pwd.h>
    
11. #include <time.h>
    
12. #include <grp.h>
    
13. #include <mysql/mysql.h>
    
14. #define N 7
    
15. MYSQL my_sql;
    
16. DIR * my_dir = NULL;
    
17. char *args[N] = {NULL,NULL,NULL,NULL,NULL,NULL,NULL};
    
18. void read_dir(char *);
    
19. void connect_mysql();
    
20. void set_work();
    
21. void get_file_infor(const char *,char *);
    
22. void write_mysql();
    
23. int main(int argc,char *argv[])
    
24. {
    
25.     if(argc != 2)
    
26.     {
    
27.         fprintf(stderr,"arguments is not two!\n");
    
28.         return -1;
    
29.     }else
    
30.     {
    
31.         connect_mysql();
    
32.         set_work();
    
33.         read_dir(argv[1]);
    
34.     }
    
35.     mysql_close(&my_sql);
    
36.     return EXIT_SUCCESS;
    
37. }
    
38. 
    
39. void read_dir(char *dirname)
    
40. {
    
41.     struct dirent * dir;
    
42.     int n = 0;
    
43.     my_dir = opendir(dirname);
    
44.     if(!my_dir)
    
45.     {
    
46.         fprintf(stderr,"open dir fail %s\n",strerror(errno));
    
47.     }else
    
48.     {
    
49.          
    
50.         while((dir = readdir(my_dir)))
    
51.         {
    
52.             if(strcmp(".",dir->d_name)&&strcmp("..",dir->d_name))
    
53.             {
    
54.                 for(;n < N;n++)
    
55.                 {
    
56.                     args[n] = (char *)malloc(256);
    
57.                     if(!args[n])
    
58.                     {
    
59.                         fprintf(stderr,"malloc fail\n");
    
60.                         closedir(my_dir);
    
61.                         exit(-1);
    
62.                     }
    
63.                 }
    
64.                 args[0] = dir->d_name;
    
65.                 get_file_infor(dir->d_name,dirname);
    
66.                 printf("\n");
    
67.             }
    
68.          
    
69.         }
    
70. 
    
71.      
    
72.     }
    
73. }
    
74. 
    
75. void connect_mysql()
    
76. {
    
77.     mysql_init(&my_sql);
    
78.     if(mysql_real_connect(&my_sql,"localhost","ma","123456ma",
    
79.                             "student",0,0,0))
    
80.     {
    
81.         printf("connect mysql success\n");
    
82.     }else{
    
83.         fprintf(stderr,"connect mysql fail %s",mysql_error(&my_sql));
    
84.         exit(-1);
    
85.     }
    
86. }
    
87. 
    
88. void set_work()
    
89. {
    
90.     int res1,res2;
    
91.     res1 =mysql_query(&my_sql,"set names utf8");
    
92.     res2 = mysql_query(&my_sql,"use student");
    
93.     if(res1)
    
94.     {
    
95.         fprintf(stderr,"set character fail! %s\n",mysql_error(&my_sql));
    
96.     }else
    
97.            printf("set character success !\n");
    
98.     if(res2)
    
99.     {
    
100.         fprintf(stderr,"use student database fail ! %s\n",mysql_error(&my_sql));
     
101.     }else
     
102.            printf("use database success !\n");
     
103. 
     
104. }
     
105. 
     
106. void write_mysql()
     
107. {
     
108.     printf("\t#######start write into mysql#######\n");
     
109.     int res ;
     
110.     char * infor = (char *)malloc(512);
     
111.     memset(infor,0,512);
     
112.     sprintf(infor,"INSERT INTO File(filename,filemode,filelink,fileown,filegroup,filetime,filesize) VALUES ('%s','%s',%s,'%s','%s','%s',%s)",args[0],args[1],args[2],args[3],args[4],args[5],args[6]);
     
113.     infor[strlen(infor)] = '\0';
     
114. //  printf("%s\n",infor);
     
115.     res = mysql_query(&my_sql,infor);
     
116.     if(!res)
     
117.            printf("Insert %lu rows\n",(unsigned long)mysql_affected_rows(&my_sql));
     
118.     else
     
119.            fprintf(stderr,"Insert data fail %s\n",mysql_error(&my_sql));
     
120. 
     
121. }
     
122. 
     
123. void get_file_infor(const char *path1,char *arg)
     
124. {
     
125.     struct stat  filestat;
     
126.     int n = 0;
     
127.     char *path = (char *)malloc(256);
     
128.     memset(path,0,256);
     
129.     sprintf(path,"%s/%s",arg,path1);
     
130.     if(lstat(path,&filestat))
     
131.     {
     
132.         fprintf(stderr,"get inode information fail %s\n",strerror(errno));
     
133.         return;
     
134.     }else
     
135.     {
     
136.          switch(filestat.st_mode & S_IFMT)
     
137.         {
     
138.             case S_IFBLK: args[1]  = "block";break;
     
139.             case S_IFCHR: args[1] = "character";break;
     
140.             case S_IFDIR: args[1] = "directory";break;
     
141.             case S_IFIFO: args[1] = "FIFO/pipe";break;
     
142.             case S_IFLNK: args[1] = "symlink";break;
     
143.             case S_IFREG: args[1] = "regularfile";break;
     
144.             case S_IFSOCK: args[1] = "socket";break;
     
145.             default: args[1] = "unknown";
     
146.          
     
147.         }
     
148.         sprintf(args[2],"%ld ",(unsigned long)filestat.st_nlink);
     
149.         struct passwd * passwd = getpwuid(filestat.st_uid);
     
150.         sprintf(args[3],"%s ",passwd->pw_name);
     
151.         struct group * group = getgrgid(filestat.st_gid);
     
152.         sprintf(args[4],"%s ",group->gr_name);
     
153.         struct tm *filetime = gmtime(&(filestat.st_mtime));
     
154.         sprintf(args[5],"%d/%d/%d",filetime->tm_year+1900,filetime->tm_mon,filetime->tm_mday);
     
155.         sprintf(args[6],"%lld",filestat.st_size);
     
156.         write_mysql();
     
157.     }
     
158. 
     
159. }

复制代码