Socket网络编程--聊天程序(7)

1 su root  //使用root用户
2 yum install mysql-devel*
3 yum install mysql-server*
4 service mysqld start

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 #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 }

　　... 
 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 }