Socket网络编程--聊天程序(7)
接上一小节,本来是计划这一节用来讲数据库的增删改查,但是在实现的过程中,出现了一点小问题,也不是技术的问题,就是在字符界面上比较不好操作。比如要注册一个帐号,就需要弄个字符界面提示,然后输入数字表示选择,在依次输入信息。(这一点,用C写过什么管理系统就知道,很是麻烦。)考虑到本程序讲的是网络编程,就不弄增删改查了,就判断一个用户名密码是否正确就行了。
首先是安装mysql和创建数据库,所用到的命令如下:
1 su root //使用root用户 2 yum install mysql-devel* 3 yum install mysql-server* 4 service mysqld start
创建一个users数据库和一个client表
1 $mysql -u root 2 mysql> create database users; 3 mysql> use users; //切换到users数据库中 4 mysql> create table clients(id int,username varchar(32),password varchar(32) ); //创建一个叫test的表 5 mysql> show create table clients; //显示刚才创建的表信息 6 mysql> select * from clients; //查询client表中数据 7 mysql> quit
1.插入数据
1 #include <stdio.h> 2 #include <mysql.h> 3 #include <stdlib.h> 4 5 #define HOST "localhost" 6 #define USERNAME "root" 7 #define PASSWORD "" 8 #define DATABASE "users" 9 10 int main(int argc,char *argv[]) 11 { 12 MYSQL conn; 13 int res; 14 char *sql="insert into clients values('1','user1','123456');"; 15 mysql_init(&conn); 16 17 if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,0,NULL,CLIENT_FOUND_ROWS)) 18 { 19 printf("Connect Success!\n"); 20 res=mysql_query(&conn,sql); 21 if(res) 22 { 23 printf("Insert Error!\n"); 24 } 25 else 26 { 27 printf("Insert Success!\n"); 28 } 29 } 30 else 31 { 32 printf("Connect Failed!\n"); 33 exit(-1); 34 } 35 return 0; 36 }
2.删除数据
1 #include <stdio.h> 2 #include <mysql.h> 3 #include <stdlib.h> 4 5 #define HOST "localhost" 6 #define USERNAME "root" 7 #define PASSWORD "" 8 #define DATABASE "users" 9 10 int main(int argc,char *argv[]) 11 { 12 MYSQL conn; 13 int res; 14 char *sql="delete from clients where username=\"user1\";"; 15 16 mysql_init(&conn); 17 18 if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,0,NULL,CLIENT_FOUND_ROWS)) 19 { 20 printf("Connect Success!\n"); 21 res=mysql_query(&conn,sql); 22 if(res) 23 { 24 printf("Delete Error!\n"); 25 } 26 else 27 { 28 printf("Delete Success!\n"); 29 } 30 } 31 else 32 { 33 printf("Connect Failed!\n"); 34 exit(-1); 35 } 36 return 0; 37 }
3.修改数据
1 #include <stdio.h> 2 #include <mysql.h> 3 #include <stdlib.h> 4 5 #define HOST "localhost" 6 #define USERNAME "root" 7 #define PASSWORD "" 8 #define DATABASE "users" 9 10 int main(int argc,char *argv[]) 11 { 12 MYSQL conn; 13 int res; 14 char *sql="update clients set username=\"user2\" where username=\"user1\";"; 15 mysql_init(&conn); 16 17 if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,0,NULL,CLIENT_FOUND_ROWS)) 18 { 19 printf("Connect Success!\n"); 20 res=mysql_query(&conn,sql); 21 if(res) 22 { 23 printf("Update Error!\n"); 24 } 25 else 26 { 27 printf("Update Success!\n"); 28 } 29 } 30 else 31 { 32 printf("Connect Failed!\n"); 33 exit(-1); 34 } 35 return 0; 36 }
4.查询数据
1 #include <stdio.h> 2 #include <mysql.h> 3 #include <stdlib.h> 4 5 #define HOST "localhost" 6 #define USERNAME "root" 7 #define PASSWORD "" 8 #define DATABASE "users" 9 10 int main(int argc,char *argv[]) 11 { 12 MYSQL conn; 13 MYSQL_RES *res_ptr;//指向查询结果的指针 14 MYSQL_FIELD *field;//字段结构体指针 15 MYSQL_ROW result_row;//按行返回的查询信息 16 17 int res; 18 int i,j; 19 int row,column;//查询返回的行数和列数 20 char *sql="select * from clients;"; 21 mysql_init(&conn); 22 23 24 if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,0,NULL,CLIENT_FOUND_ROWS)) 25 { 26 printf("Connect Success!\n"); 27 res=mysql_query(&conn,sql); 28 //res=0表示查询成功 29 if(res) 30 { 31 printf("Select Error!\n"); 32 } 33 else 34 { 35 printf("Select Success!\n"); 36 //将查询的结果给res_ptr 37 res_ptr=mysql_store_result(&conn); 38 //如果结果不为空,就可以输出了 39 if(res_ptr)//可以用这个的真假来判断用户名密码是否正确 40 { 41 //有结果了 42 column=mysql_num_fields(res_ptr);//获取列数 43 row=mysql_num_rows(res_ptr)+1;//加1是因为第一行是字段名 44 printf("查询到的数据有 %d 行\n",row); 45 46 //输出结果的字段名 47 for(i=0;field=mysql_fetch_field(res_ptr);i++) 48 { 49 printf("%s\t",field->name); 50 } 51 printf("\n"); 52 //按行输出结果 53 for(i=1;i<row;i++) 54 { 55 result_row=mysql_fetch_row(res_ptr); 56 for(j=0;j<column;j++) 57 { 58 printf("%s\t",result_row[j]); 59 } 60 printf("\n"); 61 } 62 } 63 else 64 { 65 //没有结果 66 printf("没有查询到匹配的数据。"); 67 } 68 } 69 } 70 else 71 { 72 printf("Connect Failed!\n"); 73 exit(-1); 74 } 75 mysql_close(&conn); 76 return 0; 77 }
对应的makefile
1 main: 2 gcc insert.c `mysql_config --cflags --libs` -o insert 3 gcc delete.c `mysql_config --cflags --libs` -o delete 4 gcc update.c `mysql_config --cflags --libs` -o update 5 gcc select.c `mysql_config --cflags --libs` -o select
由于本次的重点不是数据库的操作,所以就简单的给出常用的基本方法,在我们的聊天程序中只加入查询用户密码的功能,其他的如注册用户什么的就不写了,应该没有什么难度,就是有点烦。
加入数据库连接后的聊天程序
client.c 不变
server.c 修改如下
... 17 #include <mysql.h>//用于mysql连接 18 ...
37 38 void print_time(char * ch,time_t *now) 39 { 40 struct tm*stm; 41 stm=localtime(now);//立即获取当前时间 42 sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec); 43 } 44 45 int mysql_check_login(struct user su) 46 { 47 MYSQL conn; 48 MYSQL_RES *res_ptr; 49 int res; 50 int row; 51 char sql[256]={0}; 52 strcpy(sql,"select * from clients where username=\""); 53 strcat(sql,su.name); 54 strcat(sql,"\" and password=\""); 55 strcat(sql,su.pwd); 56 strcat(sql,"\";"); 57 printf("查询的sql:%s\n",sql); 58 59 mysql_init(&conn); 60 if(mysql_real_connect(&conn,"localhost","root","","users",0,NULL,CLIENT_FOUND_ROWS)) 61 { 62 res=mysql_query(&conn,sql); 63 if(res) 64 { 65 perror("Select SQL Error!\n"); 66 exit(-1); 67 } 68 else 69 { 70 res_ptr=mysql_store_result(&conn); 71 if(res_ptr) 72 { 73 row=mysql_num_rows(res_ptr); 74 if(row==0) 75 return 0; 76 return 1;//1表示验证成功 77 } 78 else 79 { 80 return 0;//0表示验证失败 81 } 82 } 83 } 84 else 85 { 86 perror("Database Connect Failed!"); 87 exit(-1); 88 } 89 mysql_close(&conn); 90 return 0; 91 } 92 93 int main(int argc,char *argv[]) 94 { ...
110 struct timeval timeout; 111 struct user use; 112 time_t now; struct tm *timenow; 113 ... 168 while(1) 169 { ... 174 switch(select(max_servfd+1,&servfd,NULL,NULL,&timeout)) 175 { 176 case -1: 177 perror("select error"); 178 break; 179 case 0: 180 break; 181 default: 182 //printf("has datas to offer accept\n"); 183 if(FD_ISSET(sockfd,&servfd))//sockfd 有数据表示可以进行accept 184 { ...
200 printf("客户端发来的用户名是:%s,密码:%s\n",use.name,use.pwd); 201 memset(recvBuf,0,sizeof(recvBuf)); 202 if(mysql_check_login(use)) 203 { 204 printf("验证成功!\n"); 205 strcpy(sendBuf,"yes"); 206 } 207 else 208 { 209 printf("验证失败!\n"); 210 strcpy(sendBuf,"no"); 211 } 212 if((sendSize=send(clientfd,sendBuf,MAX_DATA_SIZE,0))==-1) 213 { 214 perror("fail to receive datas"); 215 } ...
221 } 222 break; 223 } 224 //FD_COPY(recvfd,servfd); 225 for(i=0;i<MAX_CON_NO;i++)//最大队列进行判断,优化的话,可以使用链表 226 { ...
231 } 232 233 switch(select(max_recvfd+1,&recvfd,NULL,NULL,&timeout)) 234 { ...
288 } 289 } 290 return 0; 291 }
本次就增加了一个mysql_check_login这个函数用于用户名和密码的判断。也没有加入太多的功能。只要有个大概思路就可以了,剩下的注册和修改基本相似,就不加入到程序中了。
接下来还是老规矩,上一张运行时的截图
数据库中的用户名是user1,密码是123456。看,完成了验证了。顿时感觉好极了。
作者:无脑仔的小明 出处:http://www.cnblogs.com/wunaozai/ 本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。 如果文中有什么错误,欢迎指出。以免更多的人被误导。有需要沟通的,可以站内私信,文章留言,或者关注“无脑仔的小明”公众号私信我。一定尽力回答。 |