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

　　在这一小节中实现了文件的下载，具体的思路是根据用户的uid和用户提供的文件名filename联合两张表，取得md5唯一标识符，然后操作这个标识符对应的文件发送给客户端。

　　实现下载的小小网盘程序

　　client.cpp增加下面这个函数以实现文件的下载。

 1 int file_pull(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     FILE * fp=NULL;

 9 

10     host=gethostbyname(addr.host);

11     servAddr.sin_family=AF_INET;

12     servAddr.sin_addr=*((struct in_addr *)host->h_addr);

13     servAddr.sin_port=htons(addr.port);

14 

15     if(host==NULL)

16     {

17         perror("获取IP地址失败");

18         exit(-1);

19     }

20     if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1)

21     {

22         perror("socket创建失败");

23         exit(-1);

24     }

25     if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1)

26     {

27         perror("connect 失败");

28         exit(-1);

29     }

30 

31     //传输控制信号

32     control.control=FILE_PULL;

33     control.uid=user.uid;

34     if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0)

35     {

36         perror("控制信号发送失败");

37         exit(-1);

38     }

39     strcpy(file.filename,filenames);

40     file.uid=user.uid;

41     if(send(sockfd,(char *)&file,sizeof(struct File),0)<0)

42     {

43         perror("文件指纹发送失败");

44         exit(-1);

45     }

46     if((fp=fopen("data","wb"))==NULL)

47     {

48         perror("文件打开失败");

49         exit(-1);

50     }

51     int size=0;

52     int data_len=0;

53     char buffer[BUFFER_SIZE];

54     memset(buffer,0,sizeof(buffer));

55     recv(sockfd,buffer,64,0);

56     if(buffer[0]=='n')

57     {

58         printf("服务器中没有该文件，请确认后再输入，如不知道是否有文件，可以使用file list查看\n");

59         return 0;

60     }

61     memset(buffer,0,sizeof(buffer));

62     while(data_len=recv(sockfd,buffer,BUFFER_SIZE,0))

63     {

64         if(data_len<0)

65         {

66             perror("接收数据有误");

67         }

68         size++;

69         if(size==1)

70         {

71             printf("正在接收来自服务器的文件");

72         }

73         else

74         {

75             printf(".");

76         }

77         int write_len=fwrite(buffer,sizeof(char),data_len,fp);

78         if(write_len>data_len)

79         {

80             perror("写入数据有误");

81         }

82         bzero(buffer,BUFFER_SIZE);

83     }

84     printf("\n文件接收完毕\n");

85     fclose(fp);

86     rename("data",filenames);

87     close(sockfd);

88     return 0;

89 }

　　server.cpp 同样的实现一个相同的功能

  1 int main(int argc,char *argv[])

  2 {

  3     struct sockaddr_in server_addr;

  4     struct sockaddr_in client_addr;

  5     struct User user;

  6     struct Control control;

  7     char ch[64];

  8     int clientfd;

  9     pid_t pid;

 10     socklen_t length;

 11     bzero(&server_addr,sizeof(server_addr));

 12     server_addr.sin_family=AF_INET;

 13     server_addr.sin_addr.s_addr=htons(INADDR_ANY);

 14     server_addr.sin_port=htons(SERVER_PORT);

 15 

 16     //创建套接字

 17     int sockfd=socket(AF_INET,SOCK_STREAM,0);

 18     if(sockfd<0)

 19     {

 20         perror("创建套接字失败");

 21         exit(-1);

 22     }

 23 

 24     if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr))==-1)

 25     {

 26         perror("bind 失败");

 27         exit(-1);

 28     }

 29 

 30     if(listen(sockfd,LISTEN_QUEUE))

 31     {

 32         perror("listen 失败");

 33         exit(-1);

 34     }

 35 

 36     length=sizeof(struct sockaddr);

 37 

 38     while(1)

 39     {

 40         clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&length);

 41         if(clientfd==-1)

 42         {

 43             perror("accept 失败");

 44             continue;

 45         }

 46         printf(">>>>>%s:%d 连接成功，当前所在的ID(fd)号: %d \n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),clientfd);

 47         print_time(ch);

 48         printf("加入的时间是:%s\n",ch);

 49 

 50         //来一个连接就创建一个进程进行处理

 51         pid=fork();

 52         if(pid<0)

 53         {

 54             perror("fork error");

 55         }

 56         else if(pid==0)

 57         {

 58             recv(clientfd,(char *)&control,sizeof(struct Control),0);

 59             printf("用户 %d 使用命令 %d\n",control.uid,control.control);

 60             switch(control.control)

 61             {

 62                 case USER_CHECK_LOGIN:

 63                     {

 64                         //身份验证处理

 65                         recv(clientfd,(char *)&user,sizeof(struct User),0);

 66                         printf("客户端发送过来的用户名是:%s,密码:%s\n",user.username,user.password);

 67                         if((user.uid=mysql_check_login(user))>0)

 68                         {

 69                             printf("验证成功\n");

 70                         }

 71                         else

 72                         {

 73                             printf("验证失败\n");

 74                         }

 75                         send(clientfd,(char *)&user,sizeof(struct User),0);

 76                         break;

 77                     }

 78                 case FILE_PUSH:

 79                     {

 80                         char buffer[BUFFER_SIZE];

 81                         int data_len;

 82                         FILE * fp=NULL;

 83                         struct File file;

 84                         //获取文件指纹

 85                         recv(clientfd,(char *)&file,sizeof(struct File),0);

 86                         printf("获取到的用户名ID: %d 文件名:%s  MD5:%s\n",file.uid,file.filename,file.md5);

 87                         //对文件进行验证，如果文件已经存在就不用进行接收了

 88                         int t=mysql_check_md5(file);

 89                         char ch[64]={0};

 90                         printf("t=%d\n",t);

 91                         if(t!=0)

 92                         {

 93                             printf("该文件存在,使用秒传功能\n");

 94                             strcpy(ch,"yes");

 95                             send(clientfd,ch,64,0);

 96                             mysql_file_in(file.uid,t);

 97                             //continue;

 98                         }

 99                         strcpy(ch,"no");

100                         send(clientfd,ch,64,0);

101                         printf("md5验证后得到的fid:%d\n",t);

102                         bzero(buffer,BUFFER_SIZE);

103                         if((fp=fopen("data","wb"))==NULL)

104                         {

105                             perror("文件打开失败");

106                             exit(-1);

107                         }

108                         //循环接收数据

109                         int size=0;//表示有多少个块

110                         while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0))

111                         {

112                             if(data_len<0)

113                             {

114                                 perror("接收数据错误");

115                                 exit(-1);

116                             }

117                             size++;

118                             if(size==1)

119                                 printf("正在接收来自%s:%d的文件\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));

120                             else

121                                 printf(".");

122                             //向文件中写入

123                             int write_len=fwrite(buffer,sizeof(char),data_len,fp);

124                             if(write_len>data_len)

125                             {

126                                 perror("写入数据错误");

127                                 exit(-1);

128                             }

129                             bzero(buffer,BUFFER_SIZE);

130                         }

131                         if(size>0)

132                         {

133                             printf("\n%s:%d的文件传送完毕\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));

134                             //如果文件传输成功那么就可以写入数据库了

135                             mysql_file_in(file);

136                         }

137                         else

138                             printf("\n%s:%d的文件传送失败\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));

139                         fclose(fp);

140                         rename("data",file.md5);//这里可以修改文件的名字

141                         exit(0);

142                         break;

143                     }

144                 case FILE_PULL:

145                     {

146                         struct File file;

147                         char ch[64];

148                         memset(ch,0,sizeof(ch));

149                         //获取接下来要发送的文件

150                         recv(clientfd,(char *)&file,sizeof(struct File),0);

151                         //根据uid和filename获取服务器中的唯一文件，然后发送

152                         int t=mysql_get_md5_from_file(&file);

153                         printf("获取到的MD5:%s\n",file.md5);

154                         if(t==-1||file.md5[0]==0)//服务器没有对应的文件

155                         {

156                             printf("没有对应的文件\n");;

157                             strcpy(ch,"no");

158                             send(clientfd,ch,64,0);

159                             continue;

160                         }

161                         strcpy(ch,"yes");

162                         send(clientfd,ch,64,0);

163 

164                         FILE * fp = NULL;

165                         if((fp=fopen(file.md5,"rb"))==NULL)

166                         {

167                             perror("文件打开失败");

168                         }

169                         char buffer[BUFFER_SIZE];

170                         bzero(buffer,BUFFER_SIZE);

171                         printf("正在传输文件");

172                         int len=0;

173                         while((len=fread(buffer,sizeof(char),BUFFER_SIZE,fp))>0)

174                         {

175                             if(send(clientfd,buffer,BUFFER_SIZE,0)<0)

176                             {

177                                 perror("发送数据失败");

178                             }

179                             bzero(buffer,BUFFER_SIZE);

180                             printf(".");

181                         }

182                         printf("传输完成\n");

183                         fclose(fp);

184                         break;

185                     }

186                 case FILE_LIST:

187                     {

188                         break;

189                     }

190                 case FILE_DELECT:

191                     {

192                         break;

193                     }

194                 default:

195                     {

196                         break;

197                     }

198             }

199             shutdown(clientfd,2);//这里要注意加一个shutdown否则客户端接收不到结束符而一直等待接收数据。普通的close是不会发送结束符的

200             close(clientfd);//短连接结束

201             exit(0);//退出子进程

202         }

203     }

204 

205     return 0;

206 }

207 

208 ///////////////////

209 

210 int mysql_get_md5_from_file(struct File * file)

211 {

212     //select md5 from files,relations where files.fid=relations.fid and file.md5=file.md5;

213     MYSQL conn;

214     MYSQL_RES * res_ptr;

215     MYSQL_ROW result_row;

216     int res;int row;int column;

217     int rt;

218     char sql[256]={0};

219     rt=0;

220     strcpy(sql,"select md5 from files,relations where files.fid=relations.fid and files.filename=\"");

221     strcat(sql,file->filename);

222     strcat(sql,"\"");

223     printf("查询的sql:%s\n",sql);

224     mysql_init(&conn);

225     if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS))

226     {

227         res=mysql_query(&conn,sql);

228         if(res)

229         {

230             perror("select sql error!");

231         }

232         else

233         {

234             res_ptr=mysql_store_result(&conn);

235             if(res_ptr)

236             {

237                 column=mysql_num_fields(res_ptr);

238                 row=mysql_num_rows(res_ptr)+1;

239                 if(row<=1)

240                 {

241                     ;

242                 }

243                 else

244                 {

245                     result_row=mysql_fetch_row(res_ptr);

246                     if(result_row[0]==NULL)

247                     {

248                         rt=-1;

249                         strcpy(file->md5,"");

250                     }

251                     else

252                         strcpy(file->md5,result_row[0]);

253                 }

254             }

255             else

256             {

257                 rt=-1;

258                 printf("没有数据\n");

259             }

260         }

261     }

262     else

263     {

264         perror("Connect Failed1");

265         exit(-1);

266     }

267     mysql_close(&conn);

268     return rt;

269 }

　　mysql_get_md5_from_file这个函数是利用用户的uid和文件名进行查询的。因为保存在服务器中的文件是以MD5的值作为文件名的(目的是不同的用户可以有相同的文件名，但是所对应的文件确实不同的)，查询后返回一个唯一文件标识MD5值，根据这个值取得文件然后发送给客户端。

　　还有一个要注意的是在server.cpp的199行处，使用了shutdown来结束服务器的发送，一开始我没有使用该函数，造成的结果是服务器发送数据后，调用close，调用exit退出子进程，结束socket连接。虽然是关闭了socket，但是在客户端却在recv处一直处于阻塞，好像是还有数据没有接收完。经过调试还有查资料才知道，这里要调用shutdown，否则客户端接收不到结束符而一直等待接收数据。普通的close是不会发送结束符的。

　　#include <sys/socket.h>

　　int shutdown(int sockfd, int how);  //返回值: 如果成功则返回0，否则出错返回-1

　　套接字通信是双向的。可以采用函数shutdown来禁止套接字上的输入/输出。如果how是SHUT_RD(关闭读端 0 )，那么就无法从套接字读取数据；如果how是SHUT_WR(关闭写端 1 )，那么无法使用套接字发送数据；使用SHUT_RDWR则将同时无法读取和发送数据(2).

　　既然能够close关闭套接字，那么为什么还要用shutdown呢？理由如下：首先，close只有在最后一个活动引用被关闭时才释放网络端点【这就是为什么我以前的章节可以用close来结束上传，那是因为客户端的发送数据是在一个函数里面的一个连接，函数结束，连接也就断了。而这次出现不能下载，就是因为我的服务器是使用多进程的，而进程clientfd实在fork的前面定义的，所以当发送完毕后调用close是因为clientfd还被引用到，而不是最后一个活动应用】。这意味着如果复制一个套接字(如采用dup)，套接字直到关闭了最后一个引用它的文件描述符之后才会被释放。而shutdown允许使一个套接字处于不活动状态，无论引用它的文件描述符数目多少。其次，有时只有关闭套接字双向传输中的一个方向会很方便。例如，如果想让所通信的进程能够确定数据发送何时结束，可以关闭该套接字的写端，然而通过该套接字读端仍然可以继续接收数据。

 　　下面给出运行时的截图

　　运行的顺序具体看一下命令就知道了，最后出现了一个问题，就是使用不同的用户居然可以下载src这个文件，想想应该是数据库sql没有写好。我们修改如下：

　　在上面server.cpp代码的第222行处加上下面代码即可，实现对用户的权限控制。

1     strcat(sql," and relations.uid=");

2     sprintf(ch,"%d",file->uid);

3     strcat(sql,ch);

4     strcat(sql,";");

View Code

　　好了，至此我们已经实现了文件的上传和下载了，并且还能进行用户权限的控制。话说FTP是不是就差不多这个样子啊。

 

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