http://www.cnblogs.com/wunaozai/p/3887728.html

　這一節將不會介紹太多的技術的問題，這節主要是搭建一個小小的框架，為了方便接下來的繼續編寫擴展程序。本次會在上一小節的基礎上加上一個身份驗證的功能。

　　因為網盤程序不像聊天程序，網盤是屬于主動向服務器拉取信息，而聊天程序有可能要被動獲取信息，所以為了減輕服務器壓力，每次要向服務器獲取服務就建立一個短連接，而不像聊天程序一樣的長連接，微信的公眾平臺，輸入指令獲取服務，就是這個樣子了。具體看一下代碼就知道了。

　　還有為了方便處理，我增加了一個控制信號，這個控制信號在以前的聊天程序中講到過，但是當時為了簡單就沒有實現，現在就簡單實現一些，方便以后可以參考。

　　還有就是這次的網盤程序我將盡量實現下面的這些功能:

　　(1) file push filename ? 　　　　　　　　//用戶上傳文件到服務器上

　　(2) file pull filename ? ?　　　　　　　　//用戶下載文件到本地上

　　(3) file list　　　　　　　　　　　　　　//列出用戶在服務器中的所有文件

　　(4) file sendto ?filename　username　//共享文件給其他用戶

　　(5) file delete filename　　　　　　　　//刪除服務器中的文件

　　(6) 用戶登錄，自動完成

　　本次數據庫名為filetranslate,目前有表user

1 create table user 2 ( 3 uid int; 4 username varchar(64); 5 password varchar(64); 6 );

?

　　修改后的client.cpp

1 #include <netinet/in.h> // sockaddr_in 2 #include <sys/types.h> //socket 3 #include <sys/socket.h> //socket 4 #include <netdb.h> //gethostbyname 5 #include <unistd.h> //close 6 #include <stdio.h> 7 #include <stdlib.h> 8 #include <string.h> 9 #include <time.h> 10 #include <arpa/inet.h> //inet_addr 11 12 #define SERVVER_PORT 12138 13 #define LISTEN_QUEUE 20 14 #define BUFFER_SIZE 1024 15 16 //傳輸控制信號宏定義 17 //struct Control中control的取值 18 #define USER_CHECK_LOGIN 1 19 #define FILE_PUSH 2 20 #define FILE_PULL 3 21 #define FILE_LIST 4 22 #define FILE_SENDTO 5 23 #define FILE_DELECT 6 24 25 struct Addr 26 { 27 char host[64]; 28 int port; 29 }; 30 struct User 31 { 32 int uid; 33 char username[64]; 34 char password[64]; 35 }; 36 struct Control 37 { 38 int uid; 39 int control; 40 }; 41 42 void print_time(char *ch);//打印時間 43 int file_push(struct Addr addr,struct User user,char *filenames); 44 int check_login(struct Addr addr,struct User * user); 45 46 47 int main(int argc,char *argv[]) 48 { 49 char orderbuf[BUFFER_SIZE]; 50 struct Addr addr; 51 char arg1[32],arg2[32],arg3[32],arg4[32]; 52 struct User user; 53 54 if(argc!=5) 55 { 56 perror("usage: ./client [serverhost] [serverport] [username] [password]"); 57 exit(-1); 58 } 59 strcpy(addr.host,argv[1]); 60 addr.port=atoi(argv[2]); 61 strcpy(user.username,argv[3]); 62 strcpy(user.password,argv[4]); 63 int uid=check_login(addr,&user); 64 if(uid<=0) 65 { 66 perror("用戶驗證失敗"); 67 exit(-1); 68 } 69 printf("驗證登陸成功\n"); 70 while(1) 71 { 72 printf("\n請輸入操作指令:"); 73 fgets(orderbuf,BUFFER_SIZE,stdin); 74 memset(arg1,0,sizeof(arg1)); 75 memset(arg2,0,sizeof(arg2)); 76 memset(arg3,0,sizeof(arg3)); 77 memset(arg4,0,sizeof(arg4)); 78 sscanf(orderbuf,"%s%s%s%s",arg1,arg2,arg3,arg4); 79 if(strcmp(arg1,"file")==0) 80 { 81 if(strcmp(arg2,"push")==0) 82 { 83 strcpy(orderbuf,arg3); 84 file_push(addr,user,orderbuf); 85 } 86 else if(strcmp(arg2,"pull")==0) 87 { 88 ; 89 } 90 else if(strcmp(arg2,"list")==0) 91 { 92 ; 93 } 94 else if(strcmp(arg2,"sendto")==0) 95 { 96 ; 97 } 98 else if(strcmp(arg2,"delect")==0) 99 { 100 ; 101 } 102 else 103 { 104 printf("該命令不支持\n"); 105 } 106 } 107 else 108 { 109 printf("該命令不支持\n"); 110 } 111 } 112 113 114 return 0; 115 } 116 117 //驗證成功時返回大于0的uid號碼，錯誤返回-1 118 int check_login(struct Addr addr,struct User * user) 119 { 120 struct sockaddr_in servAddr; 121 struct hostent * host; 122 struct Control control; 123 int sockfd; 124 125 host=gethostbyname(addr.host); 126 servAddr.sin_family=AF_INET; 127 servAddr.sin_addr=*((struct in_addr *)host->h_addr); 128 //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1"); 129 servAddr.sin_port=htons(addr.port); 130 if(host==NULL) 131 { 132 perror("獲取IP地址失敗"); 133 exit(-1); 134 } 135 if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) 136 { 137 perror("socket創建失敗"); 138 exit(-1); 139 } 140 if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1) 141 { 142 perror("connect 失敗"); 143 exit(-1); 144 } 145 146 control.control=USER_CHECK_LOGIN; 147 control.uid=0; 148 if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0) 149 { 150 perror("發送數據失敗"); 151 exit(-1); 152 } 153 154 if(send(sockfd,(char *)user,sizeof(struct User),0)<0) 155 { 156 perror("發送數據失敗"); 157 exit(-1); 158 } 159 bzero(user,sizeof(struct User)); 160 if(recv(sockfd,(char *)user,sizeof(struct User),0)<0) 161 { 162 perror("接收數據失敗"); 163 exit(-1); 164 } 165 printf("獲取后用戶名:%s 密碼:%s ID號:%d\n",user->username,user->password,user->uid); 166 167 close(sockfd);//關閉socket連接 168 return user->uid; 169 } 170 171 int file_push(struct Addr addr,struct User user,char *filenames) 172 { 173 struct sockaddr_in servAddr; 174 struct hostent * host; 175 struct Control control; 176 int sockfd; 177 FILE *fp; 178 179 host=gethostbyname(addr.host); 180 servAddr.sin_family=AF_INET; 181 servAddr.sin_addr=*((struct in_addr *)host->h_addr); 182 //servAddr.sin_addr.s_addr=inet_addr("127.0.0.1"); 183 servAddr.sin_port=htons(addr.port); 184 if(host==NULL) 185 { 186 perror("獲取IP地址失敗"); 187 exit(-1); 188 } 189 if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) 190 { 191 perror("socket創建失敗"); 192 exit(-1); 193 } 194 195 if(connect(sockfd,(struct sockaddr *)&servAddr,sizeof(struct sockaddr_in))==-1) 196 { 197 perror("connect 失敗"); 198 exit(-1); 199 } 200 201 //打開文件 202 if((fp=fopen(filenames,"rb"))==NULL) 203 { 204 perror("文件打開失敗"); 205 exit(-1); 206 } 207 //這里傳輸控制信號 208 control.control=FILE_PUSH; 209 control.uid=user.uid; 210 if(send(sockfd,(char *)&control,sizeof(struct Control),0)<0) 211 { 212 perror("控制信號發送失敗"); 213 exit(-1); 214 } 215 char buffer[BUFFER_SIZE]; 216 bzero(buffer,BUFFER_SIZE); 217 printf("正在傳輸文件"); 218 int len=0; 219 //不斷的讀取文件直到文件結束 220 while((len=fread(buffer,1,BUFFER_SIZE,fp))>0) 221 { 222 if(send(sockfd,buffer,len,0)<0) 223 { 224 perror("發送數據失敗"); 225 exit(-1); 226 } 227 bzero(buffer,BUFFER_SIZE); 228 printf(".");//1K打印一個點//如果要實現百分比，就要計算文件大小，然后再處理即可 229 } 230 231 printf("傳輸完畢\n"); 232 fclose(fp);//關閉文件流 233 close(sockfd);//關閉socket連接 234 235 return 0; 236 } 237 238 void print_time(char *ch) 239 { 240 time_t now; 241 struct tm * stm; 242 time(&now); 243 stm=localtime(&now); 244 sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec); 245 return ; 246 }

　　下面這個是server.cpp

1 #include <netinet/in.h> 2 #include <sys/types.h> 3 #include <sys/socket.h> 4 #include <sys/select.h> 5 #include <netdb.h> 6 #include <stdlib.h> 7 #include <time.h> 8 #include <string.h> 9 #include <unistd.h> 10 #include <stdio.h> 11 #include <arpa/inet.h> //inet_ntoa 12 #include <mysql.h> 13 14 #define SERVER_PORT 12138 15 #define LISTEN_QUEUE 20 16 #define BACKLOG 200 17 #define BUFFER_SIZE 1024 18 19 //傳輸控制信號宏定義 20 //struct Control中control的取值 21 #define USER_CHECK_LOGIN 1 22 #define FILE_PUSH 2 23 #define FILE_PULL 3 24 #define FILE_LIST 4 25 #define FILE_SENDTO 5 26 #define FILE_DELECT 6 27 28 struct User 29 { 30 int uid; 31 char username[64]; 32 char password[64]; 33 }; 34 35 struct Control 36 { 37 int uid; 38 int control; 39 }; 40 41 void print_time(char *ch);//打印時間 42 int MAX(int a,int b); 43 int mysql_check_login(struct User user); 44 45 46 int main(int argc,char *argv[]) 47 { 48 struct sockaddr_in server_addr; 49 struct sockaddr_in client_addr; 50 struct User user; 51 struct Control control; 52 char ch[64]; 53 int clientfd; 54 pid_t pid; 55 socklen_t length; 56 bzero(&server_addr,sizeof(server_addr)); 57 server_addr.sin_family=AF_INET; 58 server_addr.sin_addr.s_addr=htons(INADDR_ANY); 59 server_addr.sin_port=htons(SERVER_PORT); 60 61 //創建套接字 62 int sockfd=socket(AF_INET,SOCK_STREAM,0); 63 if(sockfd<0) 64 { 65 perror("創建套接字失敗"); 66 exit(-1); 67 } 68 69 if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr))==-1) 70 { 71 perror("bind 失敗"); 72 exit(-1); 73 } 74 75 if(listen(sockfd,LISTEN_QUEUE)) 76 { 77 perror("listen 失敗"); 78 exit(-1); 79 } 80 81 length=sizeof(struct sockaddr); 82 83 while(1) 84 { 85 clientfd=accept(sockfd,(struct sockaddr *)&client_addr,&length); 86 if(clientfd==-1) 87 { 88 perror("accept 失敗"); 89 continue; 90 } 91 printf(">>>>>%s:%d 連接成功，當前所在的ID(fd)號: %d \n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port),clientfd); 92 print_time(ch); 93 printf("加入的時間是:%s\n",ch); 94 95 //來一個連接就創建一個進程進行處理 96 pid=fork(); 97 if(pid<0) 98 { 99 perror("fork error"); 100 } 101 else if(pid==0) 102 { 103 recv(clientfd,(char *)&control,sizeof(struct Control),0); 104 printf("用戶 %d 使用命令 %d\n",control.uid,control.control); 105 switch(control.control) 106 { 107 case USER_CHECK_LOGIN: 108 { 109 //身份驗證處理 110 recv(clientfd,(char *)&user,sizeof(struct User),0); 111 printf("客戶端發送過來的用戶名是:%s,密碼:%s\n",user.username,user.password); 112 if((user.uid=mysql_check_login(user))>0) 113 { 114 printf("驗證成功\n"); 115 } 116 else 117 { 118 printf("驗證失敗\n"); 119 } 120 send(clientfd,(char *)&user,sizeof(struct User),0); 121 break; 122 } 123 case FILE_PUSH: 124 { 125 char buffer[BUFFER_SIZE]; 126 int data_len; 127 FILE * fp=NULL; 128 bzero(buffer,BUFFER_SIZE); 129 if((fp=fopen("data","wb"))==NULL) 130 { 131 perror("文件打開失敗"); 132 exit(-1); 133 } 134 //循環接收數據 135 int size=0;//表示有多少個塊 136 while(data_len=recv(clientfd,buffer,BUFFER_SIZE,0)) 137 { 138 if(data_len<0) 139 { 140 perror("接收數據錯誤"); 141 exit(-1); 142 } 143 size++; 144 if(size==1) 145 printf("正在接收來自%s:%d的文件\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 146 else 147 printf("."); 148 //向文件中寫入 149 int write_len=fwrite(buffer,sizeof(char),data_len,fp); 150 if(write_len>data_len) 151 { 152 perror("寫入數據錯誤"); 153 exit(-1); 154 } 155 bzero(buffer,BUFFER_SIZE); 156 } 157 if(size>0) 158 printf("\n%s:%d的文件傳送完畢\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 159 else 160 printf("\n%s:%d的文件傳送失敗\n",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 161 fclose(fp); 162 //rename("data","asdf");//這里可以修改文件的名字 163 exit(0); 164 break; 165 } 166 case FILE_PULL: 167 { 168 break; 169 } 170 case FILE_LIST: 171 { 172 break; 173 } 174 case FILE_DELECT: 175 { 176 break; 177 } 178 default: 179 { 180 break; 181 } 182 } 183 close(clientfd);//短連接結束 184 exit(0);//退出子進程 185 } 186 } 187 188 return 0; 189 } 190 191 192 //函數定義 193 int mysql_check_login(struct User user) 194 { 195 MYSQL conn; 196 MYSQL_RES *res_ptr; 197 MYSQL_ROW result_row; 198 int res; 199 int row; 200 int column; 201 int uid; 202 char sql[256]={0}; 203 strcpy(sql,"select uid from users where username=\""); 204 strcat(sql,user.username); 205 strcat(sql,"\" and password=\""); 206 strcat(sql,user.password); 207 strcat(sql,"\";"); 208 printf("查詢的sql:%s\n",sql); 209 uid=-1; 210 mysql_init(&conn); 211 if(mysql_real_connect(&conn,"localhost","root","","filetranslate",0,NULL,CLIENT_FOUND_ROWS)) 212 { 213 res=mysql_query(&conn,sql); 214 if(res) 215 { 216 perror("Select SQL ERROR!"); 217 exit(-1); 218 } 219 else 220 { 221 res_ptr=mysql_store_result(&conn); 222 if(res_ptr) 223 { 224 column=mysql_num_fields(res_ptr);//獲取列數 225 row=mysql_num_rows(res_ptr)+1;//獲取行數，加1表示還有第一行字段名 226 if(row<=1) 227 { 228 ;//驗證失敗 229 } 230 else 231 { 232 //這里先假定username是唯一 233 result_row=mysql_fetch_row(res_ptr); 234 printf("獲取到的uid是:%s\n",result_row[0]); 235 uid=atoi(result_row[0]); 236 } 237 } 238 else 239 { 240 printf("沒有查詢到匹配的數據\n"); 241 } 242 } 243 } 244 else 245 { 246 perror("Connect Failed!\n"); 247 exit(-1); 248 } 249 mysql_close(&conn); 250 return uid; 251 } 252 253 void print_time(char *ch) 254 { 255 time_t now; 256 struct tm * stm; 257 time(&now); 258 stm=localtime(&now); 259 sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec); 260 return ; 261 } 262 263 int MAX(int a,int b) 264 { 265 if(a>b) 266 return a; 267 return b; 268 }

　　下面給張運行時的截圖

?

　　關于c語言調用mysql:?http://www.cnblogs.com/wunaozai/p/3876134.html

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。