首页 > 代码库 > Socket网络编程--小小网盘程序(4)
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 else74 {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 else121 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 else138 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 else233 {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 else244 {245 result_row=mysql_fetch_row(res_ptr);246 if(result_row[0]==NULL)247 {248 rt=-1;249 strcpy(file->md5,"");250 }251 else252 strcpy(file->md5,result_row[0]);253 }254 }255 else256 {257 rt=-1;258 printf("没有数据\n");259 }260 }261 }262 else263 {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,";");
好了，至此我们已经实现了文件的上传和下载了，并且还能进行用户权限的控制。话说FTP是不是就差不多这个样子啊。
本文地址: http://www.cnblogs.com/wunaozai/p/3892729.html
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。