• 10.5作业

    磕磕绊绊还是差不多完成了,tcp多客户端在线词典

    代码：

    数据库导入：有点粗糙，不知道怎么搞成两列，一个单词中间还是空格卧槽难搞

    1. #include 
    2. #include 
    3. #include 
    4. #include 
    5. #define ERR_MSG(msg) do{\
    6. 	fprintf(stderr,"__%d__",__LINE__);\
    7. 	perror(msg);\
    8. }while(0)
    9. int main(int argc, const char *argv[])
    10. {
    11. 	//词典数据库初始化
    12. 	sqlite3 *ppDb;
    13. 	if(sqlite3_open("../ele_directory.db",&ppDb)!=0){
    14. 		printf("__%d__:sqlite3_open error\n",__LINE__);
    15. 		return -1;
    16. 	}
    17. 	
    18. 	char sql[128]="create table if not exists directory (word char);";
    19. 	char *errmsg="";
    20.  
    21. 	int sql_res=sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg);
    22. 	if(sql_res!=0){
    23. 		printf("__%d__:sqlite3_exec :%s\n",__LINE__,errmsg);
    24. 		return -1;
    25. 	}
    26. 	//导入数据
    27. 	FILE* fp=fopen("../dict.txt","r");
    28. 	if(NULL==fp){
    29. 		ERR_MSG("fopen");
    30. 		return -1;
    31. 	}
    32. 	while(1){
    33. 		char line[64]="";
    34. 		bzero(line,sizeof(line));
    35. 		char* fs_res=fgets(line,sizeof(line),fp);
    36. 		if(NULL==fs_res){
    37. 			puts("文件读取完毕");
    38. 			break;
    39. 		}
    40. 		bzero(sql,sizeof(sql));
    41.  
    42. 		sprintf(sql,"insert into directory values(\"%s\");",line);
    43. 		sql_res=sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg);
    44. 		printf("%s\n",sql);
    45. 		if(sql_res!=0){
    46. 			printf("__%d__:sqlite3_exec:%s\n",__LINE__,errmsg);
    47. 			return -1;
    48. 		}
    49. 	}
    50. 	puts("数据导入完成");
    51. 	
    52.  
    53. 	if(sqlite3_close(ppDb)!=0){
    54. 		printf("__%d__:sqlite3_close error\n",__LINE__);
    55. 		return -1;
    56. 	}
    57.  
    58. 	return 0;
    59. }

    服务器：

    server.c

    1. /*
    2.  * function:    在线词典服务器TCP多线程
    3.  * @param [ in] 
    4.  * @param [out] 
    5.  * @return      
    6.  */
    7. #include "head.h"
    8.  
    9.  
    10. int main(int argc, const char *argv[])
    11. {
    12.  
    13.  
    14. 	//TCP服务器创建
    15. 	int sfd=socket(AF_INET,SOCK_STREAM,0);
    16. 	if(sfd<0){
    17. 		ERR_MSG("socket");
    18. 		return -1;
    19. 	}
    20. 	//允许端口快速复用
    21. 	int reuse = 1;
    22. 	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    23. 	{
    24. 		ERR_MSG("setsockopt");
    25. 		return -1;
    26. 	}
    27. 	printf("允许端口快速复用成功\n");
    28.  
    29. 	struct sockaddr_in addr;
    30. 	addr.sin_family=AF_INET;
    31. 	addr.sin_port=htons(PORT);
    32. 	addr.sin_addr.s_addr=inet_addr(IP);
    33. 	if(bind(sfd,(struct sockaddr*)&addr,sizeof(addr))<0){
    34. 		ERR_MSG("bind");
    35. 		return -1;
    36. 	}
    37. 	printf("绑定成功\n");
    38. 	if(listen(sfd,128)<0){
    39. 		ERR_MSG("listen");
    40. 		return -1;
    41. 	}
    42. 	printf("监听状态\n");
    43. 	struct sockaddr_in cli_addr;
    44. 	socklen_t addrlen=sizeof(cli_addr);
    45. 	pthread_t pth;
    46.  
    47. 	
    48. 	while(1){
    49. 		printf("客户端：%s:%d:连接前\n",\
    50. 				inet_ntoa(cli_addr.sin_addr),\
    51. 				ntohs(cli_addr.sin_addr.s_addr));
    52. 		int cfd=accept(sfd,(struct sockaddr*)&cli_addr,&addrlen);
    53. 		if(cfd<0){
    54. 			ERR_MSG("accept");
    55. 			return -1;
    56. 		}
    57. 		printf("客户端：%s:%d:连接成功\n",\
    58. 				inet_ntoa(cli_addr.sin_addr),\
    59. 				ntohs(cli_addr.sin_port));
    60. 		//调用线程
    61. 		cli_info client;	
    62. 		client.cfd=cfd;
    63. 		client.cli_addr=cli_addr;
    64. 		if(pthread_create(&pth,NULL,fun,(void*)&client)<0){
    65. 			ERR_MSG("pthread_create");
    66. 			return -1;
    67. 		}
    68. 		printf("[%d]线程创建成功\n",cfd);
    69. 		pthread_detach(pth);//分离线程交由系统回收
    70. 		
    71. 	}
    72. 	close(sfd);
    73. 	return 0;
    74. }

    fun.c

    1. #include "head.h"
    2.  
    3.  
    4. void * fun(void*arg){
    5. 	int cfd=((cli_info*)arg)->cfd;
    6. 	struct sockaddr_in cli_addr=((cli_info*)arg)->cli_addr;
    7. 	char buf[128]="";
    8. 	char user[16]="";
    9. 	//打开数据库
    10. 	sqlite3 *ppDb;
    11. 	if(sqlite3_open("../ele_directory.db",&ppDb)!=0){
    12. 		printf("__%d__:sqlite3_open error\n",__LINE__);
    13. 		return NULL;
    14. 	}
    15. 	//第一层
    16. 	//登录判断，（注册判断）1代表登录，2代表注册
    17. 	socklen_t addrlen=sizeof(cli_addr);	
    18.  
    19. 	int recv_res=recv(cfd,&buf,sizeof(buf),0);
    20. 	if(recv_res<0){
    21. 		ERR_MSG("recv");
    22. 		return NULL;
    23. 	}
    24. 	if(recv_res==0){
    25. 		printf("客户端：%s:%d:断开连接\n",\
    26. 					inet_ntoa(cli_addr.sin_addr),\
    27. 					ntohs(cli_addr.sin_port));
    28. 		return NULL;
    29. 	}
    30. 	printf("11111=%s\n",buf);
    31. 	if(buf[0]=='1'){
    32. 		//登录判断
    33. 		printf("login...\n");
    34. 		int login_res;
    35. 		while(login_res=isLogin(ppDb,cfd,&user)){
    36. 			printf("login_res=%d\n",login_res);
    37. 			char*l_res=NULL;
    38. 			if(1==login_res)
    39. 				l_res="1";
    40. 			else if(2==login_res)
    41. 				l_res="2";
    42. 			else if(3==login_res) 
    43. 				l_res="3";
    44. 			else
    45. 				return NULL;
    46. 			printf("l_res=%s\n",l_res);
    47. 			int send_res=send(cfd,l_res,sizeof(recv_res),0);
    48. 			if(send<0){
    49. 				ERR_MSG("send");
    50. 				return NULL;
    51. 			}
    52. 		}
    53. 		int send_res=send(cfd,"0",1,0);
    54. 			if(send<0){
    55. 				ERR_MSG("send");
    56. 				return NULL;
    57. 			}
    58. 		printf("%s登陆成功\n",user);
    59. 	}else if(buf[0]=='2'){
    60. 		//注册判断
    61. 		printf("register...\n");
    62. 		int register_res;
    63. 		while(register_res=isRegister(ppDb,cfd,&user)){
    64. 			printf("register_res=%d",register_res);
    65. 			if(register_res!=1)
    66. 				return NULL;
    67. 		}
    68. 		printf("%s注册成功\n",user);
    69. 		return NULL;
    70. 	}else{
    71. 		printf("__%d__unknown error\n",__LINE__);
    72. 		return NULL;
    73. 	}
    74. 	//第二层
    75. 	//单词查询或者记录查询或退出
    76. 	while(1){
    77. 		char sql[128]="";
    78. 		char* errmsg=NULL;
    79. 		bzero(buf,sizeof(buf));
    80. 		recv_res=recv(cfd,&buf,sizeof(buf),0);
    81. 		if(recv_res<0){
    82. 			ERR_MSG("recv");
    83. 			return NULL;
    84. 		}
    85. 		if(recv_res==0){
    86. 			printf("客户端：%s:%d:断开连接\n",\
    87. 					inet_ntoa(cli_addr.sin_addr),\
    88. 					ntohs(cli_addr.sin_port));
    89. 		//账号状态设为未登录
    90. 		bzero(sql,sizeof(sql));
    91. 		sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);
    92. 		if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    93. 			fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);
    94. 			return NULL;
    95. 		}
    96. 		}
    97. 		if(buf[0]=='3'){
    98. 			//单词查询
    99. 			printf("word finding...\n");
    100. 			findWord(ppDb,cfd,user);	
    101. 		}else if(buf[0]=='4'){
    102. 			//记录查询
    103. 			printf("record finding...\n");
    104. 			findRecord(ppDb,cfd,user);
    105. 		}else if(buf[0]=='5'){
    106. 			printf("客户端：%s:退出登录\n",user);
    107. 		//账号状态设为未登录
    108. 		bzero(sql,sizeof(sql));
    109. 		sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);
    110. 		if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    111. 			fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);
    112. 			return NULL;
    113. 		}
    114.  
    115. 		return NULL;
    116.  
    117. 		}else{
    118. 			printf("__%d__unknown error\n",__LINE__);
    119. 			return NULL;
    120. 		}
    121.  
    122. 	}
    123.  
    124. 	//关闭数据库
    125. 	if(sqlite3_close(ppDb)!=0){
    126. 		printf("__%d__:sqlite3_close error\n",__LINE__);
    127. 		return NULL;
    128. 	}
    129.  
    130. }
    131. int isLogin(sqlite3 *ppDb,int cfd,char (*user)[]){
    132. 	char account[10]="";
    133. 	char password[10]="";
    134. 	char buf[128]="";
    135. 	//数据库查询，账号是否存在，若存在密码是否正确
    136. 	int recv_res=recv(cfd,&buf,sizeof(buf),0);
    137. 	if(recv_res<0){
    138. 		ERR_MSG("recv");
    139. 		return -1;
    140. 	}else if(recv_res==0){
    141. 		printf("客户端：%s:断开连接\n",*user);
    142. 		return -1;
    143.  
    144. 	}
    145. 	printf("buf=%s\n",buf);
    146. 	sscanf(buf,"%s %s",account,password);
    147. 	
    148. 	char sql[128]="";
    149. 	char** pres = NULL;
    150.     int row, column;
    151.     char* errmsg = NULL;
    152. 	
    153. 	sprintf(sql,"select * from account where account=\"%s\";",account);
    154. 	if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){
    155. 		fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    156.         return -1;
    157. 	}
    158. 	printf("sql=%s\n",sql);
    159. 	int exist_res=0;//是否存在
    160.  
    161. 	for (int i=0; i<(row+1)*column; i++)
    162. 	{
    163. 		printf("%s\t", pres[i]);
    164.         if((i+1)%column == 0)
    165.             putchar('\n');
    166. 		exist_res=1;//存在为1
    167. 	}
    168. 	printf("exist_res=%d\n",exist_res);
    169. 	if(0==exist_res){
    170. 		printf("账号不存在\n");	
    171. 		return 1;//账号不存在返回1
    172. 	}else{
    173. 		printf("账号存在,判断密码\n");
    174. 		bzero(sql,sizeof(sql));
    175. 		sprintf(sql,"select status from account where account=\"%s\" and password=\"%s\";",account,password);
    176.  
    177. 		if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){
    178. 			fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    179. 			return -1;
    180. 		}
    181. 		printf("sql=%s\n",sql);
    182. 		int login_res=0;//是否存在账号密码匹配
    183.  
    184. 		for (int i=column; i<(row+1)*column; i++)
    185. 		{
    186. 			printf("%s\t", pres[i]);
    187. 			if((i+1)%column == 0)
    188. 				putchar('\n');
    189. 			login_res=1;//存在为1
    190. 		}
    191. 		if(0==login_res){
    192. 			printf("密码错误\n");
    193. 			return 2;//账号密码错误返回2
    194. 		}else if(0==strcmp(pres[column],"0")){
    195. 			printf("账号重复登录\n");
    196. 			return 3;
    197. 		}
    198. 		//账号状态设为已登录
    199. 		bzero(sql,sizeof(sql));
    200. 		sprintf(sql,"update account set status=\"0\" where account=\"%s\";",account);
    201. 		if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    202. 			fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);
    203. 			return -1;
    204. 		}
    205.  
    206. 		printf("密码正确\n");
    207. 		strcpy(*user,account);
    208. 		return 0;//登陆成功返回0
    209. 	}
    210. }
    211. int isRegister(sqlite3 *ppDb,int cfd,char(*user)[]){
    212. 	char account[10]="";
    213. 	char password[10]="";
    214. 	char buf[128]="";
    215. 	//数据库查询，账号是否存在
    216. 	bzero(buf,sizeof(buf));
    217. 	int recv_res=recv(cfd,&buf,sizeof(buf),0);
    218. 	if(recv_res<0){
    219. 		ERR_MSG("recv");
    220. 		return -1;
    221. 	}else if(recv_res==0){
    222. 		printf("客户端：%s:断开连接\n",*user);
    223. 			return -1;
    224.  
    225. 	}
    226.  
    227.  
    228. 	printf("buf=%s\n",buf);
    229. 	sscanf(buf,"%s %s",account,password);
    230. 	
    231. 	char sql[128]="";
    232. 	char** pres = NULL;
    233.     int row, column;
    234.     char* errmsg = NULL;
    235.  
    236. 	sprintf(sql,"select * from account where account=\"%s\";",account);
    237. 	if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){
    238. 		fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    239.         return -1;
    240. 	}
    241. 	printf("sql=%s\n",sql);
    242. 	int reg_res=0;//是否存在
    243.  
    244. 	for (int i=0; i<(row+1)*column; i++)
    245. 	{
    246. 		printf("%s\t", pres[i]);
    247.         if((i+1)%column == 0)
    248.             putchar('\n');
    249. 		reg_res=1;//存在为1
    250. 	}
    251. 	printf("reg_res=%d\n",reg_res);
    252. 	if(0==reg_res){
    253. 		//不存在
    254. 		printf("账号不存在，开始注册：\n");	
    255. 		bzero(sql,sizeof(sql));
    256. 		sprintf(sql,"insert into account values (\"%s\",\"%s\",\"1\");",account,password);
    257. 		printf("sql=%s\n",sql);
    258. 		if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    259. 			fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    260.         	return -1;
    261. 		}
    262. 		printf("注册完成\n");
    263. 		strcpy(*user,account);
    264. 		char *r_res="0";//注册成功
    265. 		printf("r_res=%s\n",r_res);	
    266. 		int send_res=send(cfd,r_res,sizeof(recv_res),0);
    267. 		if(send<0){
    268. 			ERR_MSG("send");
    269. 			return -1;
    270. 		}
    271.  
    272. 		return 0;
    273. 	}else{
    274. 		printf("账号已存在\n");
    275. 		char *r_res="1";
    276. 		printf("r_res=%s\n",r_res);	
    277. 		int send_res=send(cfd,r_res,sizeof(recv_res),0);
    278. 		if(send<0){
    279. 			ERR_MSG("send");
    280. 			return -1;
    281. 		}
    282. 		return 1;
    283. 	}
    284.  
    285. }
    286. int findWord(sqlite3 *ppDb,int cfd,char*user){
    287. 	printf("user=%s\n",user);
    288. 	char buf[128]="";
    289. 	char word[32]="";
    290. 	char sql[128]="";
    291. 	char*errmsg=NULL;
    292. 	int recv_res=recv(cfd,&word,sizeof(buf),0);
    293. 	if(recv_res<0){
    294. 		ERR_MSG("recv");
    295. 		return -1;
    296. 	}
    297. 	if(recv_res==0){
    298. 			printf("客户端：%s:断开连接\n",user);
    299. 		//账号状态设为未登录
    300. 		sprintf(sql,"update account set status=\"1\" where account=\"%s\";",user);
    301. 		if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    302. 			fprintf(stderr, "sqlite3_exec: %s __%d__\n", errmsg, __LINE__);
    303. 			return -1;
    304. 		}
    305. 	}
    306. 	printf("word=%s\n",word);
    307. 	char** pres = NULL;
    308.     int row, column;
    309. 	bzero(sql,sizeof(sql));
    310. 	sprintf(sql,"select * from directory where word like \"%s%\";",word);
    311. 	if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){
    312. 		fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    313.         return -1;
    314. 	}
    315. 	printf("sql=%s\n",sql);
    316. 	for (int i=column; i<(row+1)*column; i++)
    317. 	{
    318. 		bzero(buf,sizeof(buf));
    319.  
    320. 		printf("%s\t", pres[i]);
    321. 		strcpy(buf,pres[i]);
    322. 		strcat(buf," ");
    323.         if((i+1)%column == 0)
    324.             strcat(buf,"\n");
    325. 		printf("buf=%s\n",buf);
    326. 		int send_res=send(cfd,buf,sizeof(buf),0);
    327. 		if(send<0){
    328. 		ERR_MSG("send");
    329. 		return -1;
    330. 		}
    331. 	
    332.         if((i+1)%column == 0)
    333.             putchar('\n');
    334. 	}
    335. 	int send_res=send(cfd,"find over",strlen("find over"),0);
    336. 		if(send<0){
    337. 		ERR_MSG("send");
    338. 		return -1;
    339. 		}
    340. 	printf("查找完毕\n");
    341. 	//记录(word,time,user)
    342. 	bzero(sql,sizeof(sql));
    343. 	time_t t=time(NULL);
    344. 	struct tm *timeinfo=localtime(&t);
    345. 	char r_time[64]="";
    346. 	sprintf(r_time,"%d-%d-%d:%d:%d:%d",\
    347. 			timeinfo->tm_year+1900,timeinfo->tm_mon+1,timeinfo->tm_mday,\
    348. 			timeinfo->tm_hour,timeinfo->tm_min,timeinfo->tm_sec);
    349. 	printf("word:%s;time:%s;user:%s\n",word,r_time,user);
    350. 	printf("sql=%s\n",sql);
    351. 	sprintf(sql,"insert into record values(\"%s\",\"%s\",\"%s\");",word,r_time,user);
    352. 	printf("sql=%s\n",sql);
    353. 	
    354. 	if(sqlite3_exec(ppDb,sql,NULL,NULL,&errmsg)!=0){
    355. 		fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    356. 		return -1;
    357. 	}
    358. 	
    359. 	printf("记录完成\n");
    360. 	return 0;
    361. }
    362. int findRecord(sqlite3 *ppDb,int cfd,char*user){
    363. 	printf("user=%s\n",user);
    364. 	char sql[128]="";
    365. 	char** pres = NULL;
    366.     int row, column;
    367.     char* errmsg = NULL;
    368. 	char buf[128]="";
    369.  
    370. 	sprintf(sql,"select * from record where user=\"%s\";",user);
    371. 	if(sqlite3_get_table(ppDb,sql,&pres,&row,&column,&errmsg)!=0){
    372. 		fprintf(stderr, "sqlite3_get_table: %s __%d__\n", errmsg, __LINE__);
    373.         return -1;
    374. 	}
    375. 	printf("sql=%s\n",sql);
    376. 	for (int i=column; i<(row+1)*column; i++)
    377. 	{
    378. 		bzero(buf,sizeof(buf));
    379. 		printf("%s;%ld\t", pres[i],strlen(pres[i]));
    380. 		strcpy(buf,pres[i]);
    381. 		strcat(buf," ");
    382. 		if((i+1)%column == 0)
    383.             strcat(buf,"\n");
    384.  
    385. 		int send_res=send(cfd,buf,sizeof(buf),0);
    386. 		if(send<0){
    387. 		ERR_MSG("send");
    388. 		return -1;
    389. 		}
    390. 	
    391.         if((i+1)%column == 0)
    392.             putchar('\n');
    393. 	}
    394. 	printf("查找完毕\n");
    395. 	int send_res=send(cfd,"find record over",strlen("find record over"),0);
    396. 		if(send<0){
    397. 		ERR_MSG("send");
    398. 		return -1;
    399. 		}
    400. 	printf("记录查找完毕\n");
    401. 	return 0;
    402. }

    head.h

    1. #ifndef __HEAD_H__
    2. #define __HEAD_H__
    3.  
    4. #define ERR_MSG(msg) do{\
    5. 	fprintf(stderr,"__%d__",__LINE__);\
    6. 	perror(msg);\
    7. }while(0)
    8. #define PORT 8888
    9. #define IP "192.168.1.6"
    10. #include 
    11. #include 
    12. #include 
    13. #include   
    14. #include 
    15. #include 
    16. #include 
    17. #include 
    18. #include 
    19. #include 
    20.  
    21. typedef struct clientInfo{
    22. 	int cfd;
    23. 	struct sockaddr_in cli_addr;
    24. }cli_info;
    25.  
    26. void * fun(void*arg);
    27. int isLogin(sqlite3 *ppDb,int cfd,char(*user)[]);
    28. int isRegister(sqlite3 *ppDb,int cfd,char(*user)[]);
    29. int findWord(sqlite3 *ppDb,int cfd,char*user);
    30. int findRecord(sqlite3 *ppDb,int cfd,char*user);
    31. #endif

    客户端：

    client.c

    1. #include 
    2. #include 
    3. #include 
    4. #include 
    5. #include 
    6. #include 
    7. #include 
    8.  
    9. //自定义报错提示
    10. #define ERR_MSG(msg) do{\
    11. 	fprintf(stderr,"__%d__",__LINE__);\
    12. 	perror(msg);\
    13. }while(0)
    14. #define SER_PORT 8888
    15. #define SER_IP "192.168.1.6"
    16. /*
    17.  * function:    TCP客户端
    18.  * @param [ in] 
    19.  * @param [out] 
    20.  * @return      
    21.  */
    22. int login(int cfd);
    23. int registe(int cfd);
    24. int findWord(int cfd);
    25. int findRecord(int cfd);
    26.  
    27. int main(int argc, const char *argv[])
    28. {
    29. 	//1.创建socket套接字，
    30. 	int cfd=socket(AF_INET,SOCK_STREAM,0);
    31. 	if(cfd<0){
    32. 		ERR_MSG("socket");
    33. 		return -1;
    34. 	}
    35. 	puts("socket create");
    36.  
    37. 	//允许端口快速复用
    38. 	int reuse = 1;
    39. 	if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    40. 	{
    41. 		ERR_MSG("setsockopt");
    42. 		return -1;
    43. 	}
    44. 	printf("允许端口快速复用成功\n");
    45.  
    46. 	//2.连接服务端connect
    47. 	struct sockaddr_in addr;
    48. 	addr.sin_family=AF_INET;
    49. 	addr.sin_port=htons(SER_PORT);
    50. 	addr.sin_addr.s_addr=inet_addr(SER_IP);
    51. 	if(connect(cfd,(struct sockaddr*)&addr,sizeof(addr))<0){
    52. 		ERR_MSG("connect");
    53. 		return -1;
    54. 	}
    55. 	puts("connect success");
    56. 	printf("-------------------------------------------------\n");
    57. 	printf("-------Wecome to eletronic directory system------\n");
    58. 	printf("-------------------------------------------------\n");
    59. 	printf("--------------1.登录-----------------------------\n");
    60. 	printf("--------------2.注册-----------------------------\n");
    61. 	printf("--------------3.退出-----------------------------\n");
    62. 	printf("-------------------------------------------------\n");
    63. 	printf("-------------------------------------------------\n");
    64. 	printf("-------------------------------------------------\n");
    65. 	char choice=0;
    66. 	while(1){
    67.  
    68. 		printf("请选择1.登录；2注册；3.退出>>>");
    69. 		choice=getchar();
    70. 		while(getchar()!='\n');
    71. 		switch(choice){
    72. 		case '1':
    73. 			//登录
    74. 			if(5==login(cfd))
    75. 				return 0;
    76. 			break;
    77. 		case '2':
    78. 			//注册
    79. 			if(0==registe(cfd))
    80.  
    81. 				return 0;
    82. 		case '3':
    83. 			//退出
    84. 			return 0;
    85. 		default :
    86. 			printf("错误输入,请重输\n");
    87. 			break;	
    88. 		}
    89.  
    90. 	}
    91. 	close(cfd);
    92. 	return 0;
    93. }
    94. int login(int cfd){
    95. 	char buf[32]="";
    96. 	char status=0;
    97. 	int send_res=send(cfd,"1",1,0);
    98. 	if(send_res<0){
    99. 		ERR_MSG("send");
    100. 		return -1;
    101. 	}
    102. 	while(1){
    103. 		
    104. 		printf("请输入账号 密码 (空格分开)>>>");
    105. 		fgets(buf,sizeof(buf),stdin);
    106.  
    107. 		int send_res=send(cfd,buf,sizeof(buf),0);
    108. 		if(send_res<0){
    109. 			ERR_MSG("send");
    110. 			return -1;
    111. 		}
    112. 		int recv_res=recv(cfd,&status,sizeof(buf),0);
    113. 		if(recv_res<0){
    114. 			ERR_MSG("recv");
    115. 			return -1;
    116. 		}else if(recv_res==0){
    117. 			printf("socket peer has shutdown\n");
    118. 			return -2;
    119. 		}
    120. 		printf("status=%c\n",status);
    121. 		if('1'==status){
    122. 			printf("账号不存在\n");
    123. 		}else if('2'==status){
    124. 			printf("密码错误\n");
    125. 		}else if('3'==status){
    126. 			printf("账号已登录\n");
    127. 		}else{
    128. 		printf("登陆成功\n");
    129. 		break;}
    130. 	}
    131. 	printf("-----------------------------------\n");
    132. 	printf("-------请输入你的操作--------------\n");
    133. 	printf("---------3.查找单词----------------\n");
    134. 	printf("---------4.记录查询----------------\n");
    135. 	printf("---------5.退出--------------------\n");
    136. 	printf("-----------------------------------\n");
    137. 	char choice=0;
    138. 	while(1){
    139.  
    140. 		printf("请选择>>>3.查找单词4.记录查询5.退出");
    141. 		choice=getchar();
    142. 		while(getchar()!='\n');
    143. 		switch(choice){
    144. 		case '3':
    145. 			//查找单词
    146. 			findWord(cfd);
    147. 			break;
    148. 		case '4':
    149. 			//查询记录
    150. 			findRecord(cfd);
    151. 			break;
    152. 		case '5':
    153. 			//退出
    154. 			send_res=send(cfd,"5",1,0);
    155. 			if(send_res<0){
    156. 				ERR_MSG("send");
    157. 				return -1;
    158. 			}
    159. 			return 5;
    160. 		default :
    161. 			printf("错误输入,请重输\n");
    162. 			break;	
    163. 		}
    164.  
    165. 	}
    166.  
    167. 	return 0;
    168. }
    169. int registe(int cfd){
    170. 	char buf[32]="";
    171. 	char status=0;
    172. 	int send_res=send(cfd,"2",1,0);
    173. 	if(send_res<0){
    174. 		ERR_MSG("send");
    175. 		return -1;
    176. 	}
    177. 	while(1){
    178. 		bzero(buf,sizeof(buf));
    179. 		printf("请输入账号 密码 (空格分开)>>>");
    180.  
    181. 		fgets(buf,sizeof(buf),stdin);
    182. 		printf("%s",buf);
    183. 		int send_res=send(cfd,buf,sizeof(buf),0);
    184. 		if(send_res<0){
    185. 			ERR_MSG("send");
    186. 			return -1;
    187. 		}
    188. 		int recv_res=recv(cfd,&status,sizeof(buf),0);
    189.  
    190. 		if(recv_res<0){
    191. 			ERR_MSG("recv");
    192. 			return -1;
    193. 		}else if(recv_res==0){
    194. 			printf("socket peer has shutdown\n");
    195. 			return -2;
    196. 		}
    197. 		printf("status=%c\n",status);
    198. 		if(status=='0'){
    199. 			printf("注册成功,请重新进入系统\n");
    200. 			return 0;
    201. 		}else if(status=='1'){
    202. 			printf("账号已存在\n");
    203. 		}
    204. 	}
    205. }
    206. int findWord(int cfd){
    207. 	char buf[128]="";
    208. 	int send_res=send(cfd,"3",1,0);
    209. 	if(send_res<0){
    210. 		ERR_MSG("send");
    211. 		return -1;
    212. 	}
    213. 	printf("请输入你要查询的单词(支持模糊查询)>>>");
    214. 	fgets(buf,sizeof(buf),stdin);
    215. 	printf("%s",buf);
    216. 	buf[strlen(buf)-1] = '\0';
    217. 	send_res=send(cfd,buf,sizeof(buf),0);
    218. 	if(send_res<0){
    219. 		ERR_MSG("send");
    220. 		return -1;
    221. 	}
    222. 	while(1){
    223.  
    224. 		bzero(buf,sizeof(buf));
    225. 		int recv_res=recv(cfd,buf,sizeof(buf),0);
    226.  
    227. 		if(recv_res<0){
    228. 			ERR_MSG("recv");
    229. 			return -1;
    230. 		}else if(recv_res==0){
    231. 			printf("socket peer has shutdown\n");
    232. 	return -2;
    233. 		}
    234. 		if(0==strcmp(buf,"find over"))
    235. 			break;
    236.  
    237. 		printf("%s\n",buf);
    238. 	}
    239. 	putchar(10);
    240. 	printf("查找单词完毕\n");
    241. 	return 0;
    242. }
    243. int findRecord(int cfd){
    244. 	char buf[128]="";
    245. 	int send_res=send(cfd,"4",1,0);
    246. 	if(send_res<0){
    247. 		ERR_MSG("send");
    248. 		return -1;
    249. 	}
    250. 	while(1){
    251. 		bzero(buf,sizeof(buf));
    252. 		int recv_res=recv(cfd,buf,sizeof(buf),0);
    253.  
    254. 		if(recv_res<0){
    255. 			ERR_MSG("recv");
    256. 			return -1;
    257. 		}else if(recv_res==0){
    258. 			printf("socket peer has shutdown\n");
    259. 			return -2;
    260. 		}
    261. 		if(0==strcmp(buf,"find record over"))
    262. 			break;
    263. 	
    264. 		printf("%s",buf);
    265. 	}
    266. 	putchar(10);
    267. 	printf("查找记录完毕\n");
    268. 	return 0;
    269. }

    又是依托屎山代码，好乱，不想去整理了

  • 相关阅读:
    【老生谈算法】matlab实现控制系统稳定性——控制系统
    Hudi（三）集成Flink
    ELF 和 二进制文件的区别
    DlhSoft Gantt Chart Hyper Library for HTML5 Standard Edition
    自动化工具：PyAutoGUI的鼠标与键盘控制，解放双手的利器
    网页设计作业
    ora-00439 未启用 bit-mapped indexes
    计算机毕业设计Java靓车汽车销售网站(源码+mysql数据库+系统+lw文档）
    数字孪生扫除智慧城市“盲点”，赋能社会数字发展
    Webpack与Vite热更新差异对比
  • 原文地址：https://blog.csdn.net/weixin_53762703/article/details/133563372