Socket网络编程--小小网盘程序(5)

  各位好呀!这一小节应该就是这个小小网盘程序的最后一小节了,这一节将实现最后的三个功能,即列出用户在服务器中的文件列表,还有删除用户在服务器中的文件,最后的可以共享文件给好友。

  列出用户在服务器中的文件列表

  增加一个结构体

1 struct FileList
2 {
3     int cnt;
4     char list[16][128];
5 };

  为了方便我就假设服务器最多可以存16个单个用户的文件。如果想要支持更多的文件,这里可以增加一个int pages;用于分页作用,我们在服务器中获取文件时,可以根据分页进行发送。这样既方便又能支持多文件。

  client.cpp这个客户端文件增加一个函数

 1 int file_list(struct Addr addr,struct User user)
 2 {
 3     struct sockaddr_in servAddr;
 4     struct hostent *host;
 5     struct Control control;
 6     struct FileList filelist;
 7     int sockfd;
 8 
 9     host=gethostbyname(addr.host);
10     servAddr.sin_family=AF_INET;
11     servAddr.sin_addr=*((struct in_addr *)host->h_addr);
12     servAddr.sin_port=htons(addr.port);
13     if(host==NULL)
14     {
15         perror("获取IP地址失败");
16         exit(-1);
17     }
18     if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
19     {
20         perror("socket创建失败");
21         exit(-1);
22     }
23     if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
24     {
25         perror("connect 失败");
26         exit(-1);
27     }
28 
29     //控制信号
30     control.control=FILE_LIST;
31     control.uid=user.uid;
32     if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)
33     {
34         perror("文件指纹发送失败");
35         exit(-1);
36     }
37     if(recv(sockfd,(char *)&filelist,sizeof(struct FileList),0)<0)
38     {
39         perror("获取文件列表失败");
40         exit(-1);
41     }
42     for(int i=0;i<filelist.cnt;i++)
43     {
44         printf("--> %s\n",filelist.list[i]);
45     }
46 
47     close(sockfd);
48     return 0;
49 }

  然后在主函数中调用即可。

  server.cpp实现,在主函数的case FILE_LIST:处修改如下

1  case FILE_LIST:
2  {
3       struct File file;
4       struct FileList filelist;
5       file.uid=control.uid;
6       mysql_get_file_list(file,&filelist);
7       send(clientfd,(char *)&filelist,sizeof(struct FileList),0);
8       break;
9   }

  然后再增加一个对应的mysql_get_file_list函数

 1 int mysql_get_file_list(struct File file,struct FileList *filelist)
 2 {
 3     MYSQL conn;
 4     MYSQL_RES * res_ptr;
 5     MYSQL_ROW result_row;
 6     int res;int row;int column;
 7     int i,j;
 8     char sql[256]={0};
 9     char ch[64];
10     //select filename from files,relations where relations.uid=[file].uid and relations.fid=files.fid;
11     strcpy(sql,"select filename from files,relations where relations.uid=");
12     sprintf(ch,"%d",file.uid);
13     strcat(sql,ch);
14     strcat(sql," and relations.fid=files.fid ;");
15     //printf("==>%s\n",sql);
16 
17     mysql_init(&conn);
18     if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
19     {
20         res=mysql_query(&conn,sql);
21         if(res)
22         {
23             perror("select sql error1");
24         }
25         else
26         {
27             res_ptr=mysql_store_result(&conn);
28             if(res_ptr)
29             {
30                 column=mysql_num_fields(res_ptr);
31                 row=mysql_num_rows(res_ptr)+1;
32                 //按行输出结果
33                 filelist->cnt=row-1;
34                 for(i=1;i<row;i++)
35                 {
36                     result_row=mysql_fetch_row(res_ptr);
37                     strcpy(filelist->list[i-1],result_row[0]);
38                     //printf("%s",result_row[0]);
39                 }
40             }
41             else
42             {
43                 printf("没有数据\n");
44             }
45         }
46     }
47     else
48     {
49         perror("Connect Failed1\n");
50         exit(-1);
51     }
52     mysql_close(&conn);
53     return 0;
54 }

  运行时的截图

 

  删除服务器中的用户文件

  在client.cpp中增加一个file_delete函数

 1 int file_delect(struct Addr addr,struct User user,char *filenames)
 2 {
 3     struct sockaddr_in servAddr;
 4     struct hostent *host;
 5     struct Control control;
 6     struct File file;
 7     int sockfd;
 8 
 9     host=gethostbyname(addr.host);
10     servAddr.sin_family=AF_INET;
11     servAddr.sin_addr=*((struct in_addr *)host->h_addr);
12     servAddr.sin_port=htons(addr.port);
13     if(host==NULL)
14     {
15         perror("获取IP地址失败");
16         exit(-1);
17     }
18     if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
19     {
20         perror("socket创建失败");
21         exit(-1);
22     }
23     if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
24     {
25         perror("connect 失败");
26         exit(-1);
27     }
28 
29     //控制信号
30     control.control=FILE_DELECT;
31     control.uid=user.uid;
32     if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)
33     {
34         perror("文件指纹发送失败");
35         exit(-1);
36     }
37     file.uid=user.uid;
38     strcpy(file.filename,filenames);
39     if(send(sockfd,(char *)&file,sizeof(struct File),0)<0)
40     {
41         perror("删除文件失败");
42         exit(-1);
43     }
44     char ch[32];
45     memset(ch,0,sizeof(ch));
46     if(recv(sockfd,ch,sizeof(ch),0)<0)
47     {
48         perror("删除文件失败");
49         exit(-1);
50     }
51     if(ch[0]=='y') //删除成功
52     {
53         printf("删除成功\n");;
54     }
55     else if(ch[0]=='n') //删除失败
56     {
57         printf("删除失败,确认是否有该文件\n");;
58     }
59     close(sockfd);
60     return 0;
61 }

  在server.cpp的main函数中增加

 1 case FILE_DELECT:
 2 {
 3     struct File file;
 4     char ch[64];
 5     memset(ch,0,sizeof(ch));
 6     recv(clientfd,(char *)&file,sizeof(struct File),0);
 7     int t=mysql_delete_file(file);
 8     if(t==-1)
 9     {
10          printf("没有对应的文件\n");;
11          strcpy(ch,"no");
12          send(clientfd,ch,64,0);
13          break;
14      }
15       strcpy(ch,"yes");
16       send(clientfd,ch,64,0);
17       printf("删除成功\n");
18       break;
19 }

  然后在server.cpp中再增加一个mysql_delete_file函数

 1 int mysql_delete_file(struct File file)
 2 {
 3     MYSQL conn;
 4     MYSQL_RES * res_ptr;
 5     MYSQL_ROW result_row;
 6     int res;int row;int column;
 7     int i,j;
 8     char sql[256];
 9     char ch[64];
10     int fid;int rt=0;
11 
12     mysql_init(&conn);
13     if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
14     {
15         //select files.fid from files,relations where relations.fid=files.fid and filename= [file].filename
16         strcpy(sql,"select files.fid from files,relations where relations.fid=files.fid and filename=\"");
17         strcat(sql,file.filename);
18         strcat(sql,"\";");
19         res=mysql_query(&conn,sql);
20         fid=0;
21         if(res)
22         {
23             perror("Select Sql Error!");
24         }
25         else
26         {
27             res_ptr=mysql_store_result(&conn);
28             if(res_ptr)
29             {
30                 column=mysql_num_fields(res_ptr);
31                 row=mysql_num_rows(res_ptr)+1;
32                 if(row<=1)
33                     ;
34                 else
35                 {
36                     result_row=mysql_fetch_row(res_ptr);
37                     if(result_row[0]==NULL)
38                     {
39                         fid=0;
40                     }
41                     else
42                     {
43                         fid=atoi(result_row[0]);
44                     }
45                 }
46             }
47             else
48             {
49                 fid=0;
50             }
51         }
52         if(fid==0)
53         {
54             mysql_close(&conn);
55             return -1;
56         }
57         //根据获取到的fid然后删除relations对应fid和uid
58         //delect relations where uid='uid' and fid='fid'
59         strcpy(sql,"delete from relations where uid=");
60         sprintf(ch,"%d",file.uid);
61         strcat(sql,ch);
62         strcat(sql," and fid=");
63         sprintf(ch,"%d",fid);
64         strcat(sql,ch);
65         res=mysql_query(&conn,sql);
66         if(res)
67         {
68             printf("Delete Error\n");
69         }
70         else
71         {
72             ;;
73         }
74     }
75     else
76     {
77         perror("Connect Failed!");
78         exit(-1);
79     }
80 
81     mysql_close(&conn);
82     return rt;
83 }

  从上面的sql语句可以知道我们只是删除了relations表中的链接而已。而没有真正的删除已经上传上去的文件。这一点可以参考以前给过的资料。

  运行的截图如下

 

  文件共享给好友

  client.cpp文件加入一个函数

 1 int file_sendto(struct Addr addr,struct User user,char *filenames,struct User to)
 2 {
 3     struct sockaddr_in servAddr;
 4     struct hostent *host;
 5     struct Control control;
 6     struct File file;
 7     int sockfd;
 8 
 9     host=gethostbyname(addr.host);
10     servAddr.sin_family=AF_INET;
11     servAddr.sin_addr=*((struct in_addr *)host->h_addr);
12     servAddr.sin_port=htons(addr.port);
13     if(host==NULL)
14     {
15         perror("获取IP地址失败");
16         exit(-1);
17     }
18     if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)
19     {
20         perror("socket创建失败");
21         exit(-1);
22     }
23     if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)
24     {
25         perror("connect 失败");
26         exit(-1);
27     }
28 
29     //控制信号
30     control.control=FILE_SENDTO;
31     control.uid=user.uid;
32     if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)
33     {
34         perror("文件指纹发送失败");
35         exit(-1);
36     }
37     file.uid=user.uid;
38     strcpy(file.filename,filenames);
39     if(send(sockfd,(char *)&file,sizeof(struct File),0)<0)
40     {
41         perror("共享文件失败");
42         exit(-1);
43     }
44     if(send(sockfd,(char *)&to,sizeof(struct User),0)<0)
45     {
46         perror("共享用户发送失败");
47         exit(-1);
48     }
49     char ch[32];
50     memset(ch,0,sizeof(ch));
51     if(recv(sockfd,ch,sizeof(ch),0)<0)
52     {
53         perror("共享文件失败");
54         exit(-1);
55     }
56     if(ch[0]=='y') //删除成功
57     {
58         printf("共享成功\n");;
59     }
60     else if(ch[0]=='n') //删除失败
61     {
62         if(ch[2]=='1')
63             printf("共享失败,确认是否有该文件\n");
64         else if(ch[2]=='2')
65             printf("共享失败,确认是否有该用户\n");
66     }
67     close(sockfd);
68     return 0;
69 }

  而server.cpp在主函数main中switch中增加如下

 1 case FILE_SENDTO:
 2 {
 3      struct File file;
 4      struct User to;
 5      recv(clientfd,(char *)&file,sizeof(struct File),0);
 6      recv(clientfd,(char *)&to,sizeof(struct User),0);
 7      int t=mysql_sendto(file,to);
 8      char ch[64];
 9      memset(ch,0,sizeof(ch));
10      if(t==-1)
11      {
12           printf("没有对应的文件\n");;
13           strcpy(ch,"no1");
14           send(clientfd,ch,64,0);
15           break;
16      }
17      else if(t==-2)
18      {
19           printf("没有对应的用户\n");;
20           strcpy(ch,"no2");
21           send(clientfd,ch,64,0);
22           break;
23      }
24           strcpy(ch,"yes");
25           send(clientfd,ch,64,0);
26           printf("共享成功\n");
27           break;
28 }

  然后对应的增加下面一个函数

  1 int mysql_sendto(struct File file,struct User to)
  2 {
  3     //insert into relations values(uid,fid);
  4     MYSQL conn;
  5     MYSQL_RES * res_ptr;
  6     MYSQL_ROW result_row;
  7     int res;int row;int column;
  8     int i,j;int fid;int uid;int rt=0;
  9     char sql[256];
 10     char ch[64];
 11 
 12     mysql_init(&conn);
 13     if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))
 14     {
 15         //select files.fid from files,relations where relations.fid=files.fid and filename=files.filename;
 16         //得到fid后
 17         strcpy(sql,"select files.fid from files,relations where relations.fid=files.fid and filename=\"");
 18         strcat(sql,file.filename);
 19         strcat(sql,"\";");
 20         res=mysql_query(&conn,sql);
 21         fid=0;
 22         if(res)
 23         {
 24             perror("Select Sql Error!");
 25         }
 26         else
 27         {
 28             res_ptr=mysql_store_result(&conn);
 29             if(res_ptr)
 30             {
 31                 column=mysql_num_fields(res_ptr);
 32                 row=mysql_num_rows(res_ptr)+1;
 33                 if(row<=1)
 34                     ;
 35                 else
 36                 {
 37                     result_row=mysql_fetch_row(res_ptr);
 38                     if(result_row[0]==NULL)
 39                     {
 40                         fid=0;
 41                     }
 42                     else
 43                     {
 44                         fid=atoi(result_row[0]);
 45                     }
 46                 }
 47             }
 48             else
 49             {
 50                 fid=0;
 51             }
 52         }
 53         if(fid==0)
 54         {
 55             mysql_close(&conn);
 56             return -1;//表示没有该文件
 57         }
 58 
 59         //select uid from users where username=[to].username;
 60         //得到uid后
 61         strcpy(sql,"select uid from users where username=\"");
 62         strcat(sql,to.username);
 63         strcat(sql,"\"");
 64         res=mysql_query(&conn,sql);
 65         uid=0;
 66         if(res)
 67         {
 68             perror("Select Sql Error!");
 69         }
 70         else
 71         {
 72             res_ptr=mysql_store_result(&conn);
 73             if(res_ptr)
 74             {
 75                 column=mysql_num_fields(res_ptr);
 76                 row=mysql_num_rows(res_ptr)+1;
 77                 if(row<=1)
 78                     ;
 79                 else
 80                 {
 81                     result_row=mysql_fetch_row(res_ptr);
 82                     if(result_row[0]==NULL)
 83                     {
 84                         uid=0;
 85                     }
 86                     else
 87                     {
 88                         uid=atoi(result_row[0]);
 89                     }
 90                 }
 91             }
 92             else
 93             {
 94                 uid=0;
 95             }
 96         }
 97         if(uid==0)
 98         {
 99             mysql_close(&conn);
100             return -2;//表示没有该用户
101         }
102 
103         //将获取到的uid fid插入到数据库relations中
104         //insert into relations values(uid,fid);
105         strcpy(sql,"insert into relations values( ");
106         sprintf(ch,"%d",uid);
107         strcat(sql,ch);
108         strcat(sql,", ");
109         sprintf(ch,"%d",fid);
110         strcat(sql,ch);
111         strcat(sql,");");
112         res=mysql_query(&conn,sql);
113         printf("==========> uid=%d  fid=%d\n",uid,fid);
114         if(res)
115         {
116             rt=-1;
117             printf("Insert Error\n");
118         }
119         else
120         {
121             printf("Insert Success\n");
122         }
123     }
124     else
125     {
126         perror("Connect Failed!");
127         exit(-1);
128     }
129     mysql_close(&conn);
130     return rt;
131 }

  下面这个是运行时的截图

  在本次程序的最后,送上程序代码结构,及本人的开发环境。

  从程序中可以看出很多代码是有冗余的,如果进行重构的话,估计代码可以节省50%。可怕的新手啊(◑﹏◐)。从代码量上看,这次的代码量也不少了,相比与上次聊天程序,代码量有过之而不及,不过可喜的是这次都把具体的功能封装成一个一个的函数,即使有冗余代码。不过还是有点小进步了。

 

  Socket网络编程--小小网盘程序各个小节的传送门

  Socket网络编程--小小网盘程序(1) http://www.cnblogs.com/wunaozai/p/3886588.html
  Socket网络编程--小小网盘程序(2) http://www.cnblogs.com/wunaozai/p/3887728.html
  Socket网络编程--小小网盘程序(3) http://www.cnblogs.com/wunaozai/p/3891062.html
  Socket网络编程--小小网盘程序(4) http://www.cnblogs.com/wunaozai/p/3892729.html
  Socket网络编程--小小网盘程序(5) http://www.cnblogs.com/wunaozai/p/3893469.html

  本文地址: http://www.cnblogs.com/wunaozai/p/3893469.html 

  代码下载: https://files.cnblogs.com/wunaozai/xiaoxiaowangpan.zip

posted @ 2014-08-06 09:17  无脑仔的小明  阅读(2129)  评论(4编辑  收藏  举报