在這一小節(jié)中實(shí)現(xiàn)了文件的下載，具體的思路是根據(jù)用戶的uid和用戶提供的文件名filename聯(lián)合兩張表，取得md5唯一標(biāo)識(shí)符，然后操作這個(gè)標(biāo)識(shí)符對應(yīng)的文件發(fā)送給客戶端。

　　實(shí)現(xiàn)下載的小小網(wǎng)盤程序

　　client.cpp增加下面這個(gè)函數(shù)以實(shí)現(xiàn)文件的下載。

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創(chuàng)建失敗"); 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 //傳輸控制信號(hào) 32 control.control=FILE_PULL; 33 control.uid=user.uid; 34 if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0) 35 { 36 perror("控制信號(hào)發(fā)送失敗"); 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("文件指紋發(fā)送失敗"); 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("服務(wù)器中沒有該文件，請確認(rèn)后再輸入，如不知道是否有文件，可以使用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("接收數(shù)據(jù)有誤"); 67 } 68 size++; 69 if(size==1) 70 { 71 printf("正在接收來自服務(wù)器的文件"); 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("寫入數(shù)據(jù)有誤"); 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 同樣的實(shí)現(xiàn)一個(gè)相同的功能

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 //創(chuàng)建套接字 17 int sockfd=socket(AF_INET,SOCK_STREAM,0); 18 if(sockfd<0) 19 { 20 perror("創(chuàng)建套接字失敗"); 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 連接成功，當(dāng)前所在的ID(fd)號(hào): %d \n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),clientfd); 47 print_time(ch); 48 printf("加入的時(shí)間是:%s\n",ch); 49 50 //來一個(gè)連接就創(chuàng)建一個(gè)進(jìn)程進(jìn)行處理 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 //身份驗(yàn)證處理 65 recv(clientfd,(char *)&user,sizeof(struct User),0); 66 printf("客戶端發(fā)送過來的用戶名是:%s,密碼:%s\n",user.username,user.password); 67 if((user.uid=mysql_check_login(user))>0) 68 { 69 printf("驗(yàn)證成功\n"); 70 } 71 else 72 { 73 printf("驗(yàn)證失敗\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 //對文件進(jìn)行驗(yàn)證，如果文件已經(jīng)存在就不用進(jìn)行接收了 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驗(yàn)證后得到的fid:%d\n",t); 102 bzero(buffer,BUFFER_SIZE); 103 if((fp=fopen("data","wb"))==NULL) 104 { 105 perror("文件打開失敗"); 106 exit(-1); 107 } 108 //循環(huán)接收數(shù)據(jù) 109 int size=0;//表示有多少個(gè)塊 110 while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0)) 111 { 112 if(data_len<0) 113 { 114 perror("接收數(shù)據(jù)錯(cuò)誤"); 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("寫入數(shù)據(jù)錯(cuò)誤"); 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 //如果文件傳輸成功那么就可以寫入數(shù)據(jù)庫了 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 //獲取接下來要發(fā)送的文件 150 recv(clientfd,(char *)&file,sizeof(struct File),0); 151 //根據(jù)uid和filename獲取服務(wù)器中的唯一文件，然后發(fā)送 152 int t=mysql_get_md5_from_file(&file); 153 printf("獲取到的MD5:%s\n",file.md5); 154 if(t==-1||file.md5[0]==0)//服務(wù)器沒有對應(yīng)的文件 155 { 156 printf("沒有對應(yīng)的文件\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("發(fā)送數(shù)據(jù)失敗"); 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);//這里要注意加一個(gè)shutdown否則客戶端接收不到結(jié)束符而一直等待接收數(shù)據(jù)。普通的close是不會(huì)發(fā)送結(jié)束符的 200 close(clientfd);//短連接結(jié)束 201 exit(0);//退出子進(jìn)程 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("沒有數(shù)據(jù)\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這個(gè)函數(shù)是利用用戶的uid和文件名進(jìn)行查詢的。因?yàn)楸４嬖诜?wù)器中的文件是以MD5的值作為文件名的(目的是不同的用戶可以有相同的文件名，但是所對應(yīng)的文件確實(shí)不同的)，查詢后返回一個(gè)唯一文件標(biāo)識(shí)MD5值，根據(jù)這個(gè)值取得文件然后發(fā)送給客戶端。

　　還有一個(gè)要注意的是在server.cpp的199行處，使用了shutdown來結(jié)束服務(wù)器的發(fā)送，一開始我沒有使用該函數(shù)，造成的結(jié)果是服務(wù)器發(fā)送數(shù)據(jù)后，調(diào)用close，調(diào)用exit退出子進(jìn)程，結(jié)束socket連接。雖然是關(guān)閉了socket，但是在客戶端卻在recv處一直處于阻塞，好像是還有數(shù)據(jù)沒有接收完。經(jīng)過調(diào)試還有查資料才知道，這里要調(diào)用shutdown，否則客戶端接收不到結(jié)束符而一直等待接收數(shù)據(jù)。普通的close是不會(huì)發(fā)送結(jié)束符的。

　　#include <sys/socket.h>

　　int shutdown(int sockfd, int how); ?//返回值: 如果成功則返回0，否則出錯(cuò)返回-1

　　套接字通信是雙向的?？梢圆捎煤瘮?shù)shutdown來禁止套接字上的輸入/輸出。如果how是SHUT_RD(關(guān)閉讀端 0 )，那么就無法從套接字讀取數(shù)據(jù)；如果how是SHUT_WR(關(guān)閉寫端 1 )，那么無法使用套接字發(fā)送數(shù)據(jù)；使用SHUT_RDWR則將同時(shí)無法讀取和發(fā)送數(shù)據(jù)(2).

　　既然能夠close關(guān)閉套接字，那么為什么還要用shutdown呢？理由如下：首先，close只有在最后一個(gè)活動(dòng)引用被關(guān)閉時(shí)才釋放網(wǎng)絡(luò)端點(diǎn)【這就是為什么我以前的章節(jié)可以用close來結(jié)束上傳，那是因?yàn)榭蛻舳说陌l(fā)送數(shù)據(jù)是在一個(gè)函數(shù)里面的一個(gè)連接，函數(shù)結(jié)束，連接也就斷了。而這次出現(xiàn)不能下載，就是因?yàn)槲业姆?wù)器是使用多進(jìn)程的，而進(jìn)程clientfd實(shí)在fork的前面定義的，所以當(dāng)發(fā)送完畢后調(diào)用close是因?yàn)閏lientfd還被引用到，而不是最后一個(gè)活動(dòng)應(yīng)用】。這意味著如果復(fù)制一個(gè)套接字(如采用dup)，套接字直到關(guān)閉了最后一個(gè)引用它的文件描述符之后才會(huì)被釋放。而shutdown允許使一個(gè)套接字處于不活動(dòng)狀態(tài)，無論引用它的文件描述符數(shù)目多少。其次，有時(shí)只有關(guān)閉套接字雙向傳輸中的一個(gè)方向會(huì)很方便。例如，如果想讓所通信的進(jìn)程能夠確定數(shù)據(jù)發(fā)送何時(shí)結(jié)束，可以關(guān)閉該套接字的寫端，然而通過該套接字讀端仍然可以繼續(xù)接收數(shù)據(jù)。

?　　下面給出運(yùn)行時(shí)的截圖

　　運(yùn)行的順序具體看一下命令就知道了，最后出現(xiàn)了一個(gè)問題，就是使用不同的用戶居然可以下載src這個(gè)文件，想想應(yīng)該是數(shù)據(jù)庫sql沒有寫好。我們修改如下：

　　在上面server.cpp代碼的第222行處加上下面代碼即可，實(shí)現(xiàn)對用戶的權(quán)限控制。

1 strcat(sql," and relations.uid="); 2 sprintf(ch,"%d",file->uid); 3 strcat(sql,ch); 4 strcat(sql,";"); View Code

　　好了，至此我們已經(jīng)實(shí)現(xiàn)了文件的上傳和下載了，并且還能進(jìn)行用戶權(quán)限的控制。話說FTP是不是就差不多這個(gè)樣子啊。

?

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

轉(zhuǎn)載于:https://www.cnblogs.com/wunaozai/p/3892729.html

總結(jié)

以上是生活随笔為你收集整理的Socket网络编程--小小网盘程序(4)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。