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

　　接上一小节，本来是计划这一节用来讲数据库的增删改查，但是在实现的过程中，出现了一点小问题，也不是技术的问题，就是在字符界面上比较不好操作。比如要注册一个帐号，就需要弄个字符界面提示，然后输入数字表示选择，在依次输入信息。(这一点，用C写过什么管理系统就知道，很是麻烦。)考虑到本程序讲的是网络编程，就不弄增删改查了，就判断一个用户名密码是否正确就行了。

　　首先是安装mysql和创建数据库，所用到的命令如下：

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

　　创建一个users数据库和一个client表

 $mysql -u root
 mysql> create database users;
 mysql> use users;    //切换到users数据库中
 mysql> create table clients(id int,username varchar(),password varchar() ); //创建一个叫test的表
 mysql> show create table clients;  //显示刚才创建的表信息
 mysql> select * from clients; 　　//查询client表中数据
 mysql> quit

　　1.插入数据

 #include <stdio.h>
 #include <mysql.h>
 #include <stdlib.h>

 #define HOST "localhost"
 #define USERNAME "root"
 #define PASSWORD ""
 #define DATABASE "users"

 int main(int argc,char *argv[])
 {
     MYSQL conn;
     int res;
     char *sql="insert into clients values('1','user1','123456');";
     mysql_init(&conn);

     if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,,NULL,CLIENT_FOUND_ROWS))
     {
         printf("Connect Success!\n");
         res=mysql_query(&conn,sql);
         if(res)
         {
             printf("Insert Error!\n");
         }
         else
         {
             printf("Insert Success!\n");
         }
     }
     else
     {
         printf("Connect Failed!\n");
         exit(-);
     }
     return ;
 }

　　2.删除数据

 #include <stdio.h>
 #include <mysql.h>
 #include <stdlib.h>

 #define HOST "localhost"
 #define USERNAME "root"
 #define PASSWORD ""
 #define DATABASE "users"

 int main(int argc,char *argv[])
 {
     MYSQL conn;
     int res;
     char *sql="delete from clients where username=\"user1\";";

     mysql_init(&conn);

     if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,,NULL,CLIENT_FOUND_ROWS))
     {
         printf("Connect Success!\n");
         res=mysql_query(&conn,sql);
         if(res)
         {
             printf("Delete Error!\n");
         }
         else
         {
             printf("Delete Success!\n");
         }
     }
     else
     {
         printf("Connect Failed!\n");
         exit(-);
     }
     return ;
 }

　　3.修改数据

 #include <stdio.h>
 #include <mysql.h>
 #include <stdlib.h>

 #define HOST "localhost"
 #define USERNAME "root"
 #define PASSWORD ""
 #define DATABASE "users"

 int main(int argc,char *argv[])
 {
     MYSQL conn;
     int res;
     char *sql="update clients set username=\"user2\" where username=\"user1\";";
     mysql_init(&conn);

     if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,,NULL,CLIENT_FOUND_ROWS))
     {
         printf("Connect Success!\n");
         res=mysql_query(&conn,sql);
         if(res)
         {
             printf("Update Error!\n");
         }
         else
         {
             printf("Update Success!\n");
         }
     }
     else
     {
         printf("Connect Failed!\n");
         exit(-);
     }
     return ;
 }

　　4.查询数据

 #include <stdio.h>
 #include <mysql.h>
 #include <stdlib.h>

 #define HOST "localhost"
 #define USERNAME "root"
 #define PASSWORD ""
 #define DATABASE "users"

 int main(int argc,char *argv[])
 {
     MYSQL conn;
     MYSQL_RES *res_ptr;//指向查询结果的指针
     MYSQL_FIELD *field;//字段结构体指针
     MYSQL_ROW result_row;//按行返回的查询信息

     int res;
     int i,j;
     int row,column;//查询返回的行数和列数
     char *sql="select * from clients;";
     mysql_init(&conn);

     if(mysql_real_connect(&conn,HOST,USERNAME,PASSWORD,DATABASE,,NULL,CLIENT_FOUND_ROWS))
     {
         printf("Connect Success!\n");
         res=mysql_query(&conn,sql);
         //res=0表示查询成功
         if(res)
         {
             printf("Select Error!\n");
         }
         else
         {
             printf("Select Success!\n");
             //将查询的结果给res_ptr
             res_ptr=mysql_store_result(&conn);
             //如果结果不为空，就可以输出了
             if(res_ptr)//可以用这个的真假来判断用户名密码是否正确
             {
                 //有结果了
                 column=mysql_num_fields(res_ptr);//获取列数
                 row=mysql_num_rows(res_ptr)+;//加1是因为第一行是字段名
                 printf("查询到的数据有 %d 行\n",row);

                 //输出结果的字段名
                 for(i=;field=mysql_fetch_field(res_ptr);i++)
                 {
                     printf("%s\t",field->name);
                 }
                 printf("\n");
                 //按行输出结果
                 for(i=;i<row;i++)
                 {
                     result_row=mysql_fetch_row(res_ptr);
                     for(j=;j<column;j++)
                     {
                         printf("%s\t",result_row[j]);
                     }
                     printf("\n");
                 }
             }
             else
             {
                 //没有结果
                 printf("没有查询到匹配的数据。");
             }
         }
     }
     else
     {
         printf("Connect Failed!\n");
         exit(-);
     }
     mysql_close(&conn);
     return ;
 }

　　对应的makefile

 main:
         gcc insert.c `mysql_config --cflags --libs` -o insert
         gcc delete.c `mysql_config --cflags --libs` -o delete
         gcc update.c `mysql_config --cflags --libs` -o update
         gcc select.c `mysql_config --cflags --libs` -o select

　　由于本次的重点不是数据库的操作，所以就简单的给出常用的基本方法，在我们的聊天程序中只加入查询用户密码的功能，其他的如注册用户什么的就不写了，应该没有什么难度，就是有点烦。

　　加入数据库连接后的聊天程序

　　client.c 不变

　　server.c 修改如下

　　...
 #include <mysql.h>//用于mysql连接

　　... 
 37
 void print_time(char * ch,time_t *now)
 {
     struct tm*stm;
     stm=localtime(now);//立即获取当前时间
     sprintf(ch,"%02d:%02d:%02d\n",stm->tm_hour,stm->tm_min,stm->tm_sec);
 }

 int mysql_check_login(struct user su)
 {
     MYSQL conn;
     MYSQL_RES *res_ptr;
     int res;
     int row;
     char sql[]={};
     strcpy(sql,"select * from clients where username=\"");
     strcat(sql,su.name);
     strcat(sql,"\" and password=\"");
     strcat(sql,su.pwd);
     strcat(sql,"\";");
     printf("查询的sql:%s\n",sql);

     mysql_init(&conn);
     if(mysql_real_connect(&conn,"localhost","root","","users",,NULL,CLIENT_FOUND_ROWS))
     {
         res=mysql_query(&conn,sql);
         if(res)
         {
             perror("Select SQL Error!\n");
             exit(-);
         }
         else
         {
             res_ptr=mysql_store_result(&conn);
             if(res_ptr)
             {
                 row=mysql_num_rows(res_ptr);
                 if(row==)
                     return ;
                 return ;//1表示验证成功
             }
             else
             {
                 return ;//0表示验证失败
             }
         }
     }
     else
     {
         perror("Database Connect Failed!");
         exit(-);
     }
     mysql_close(&conn);
     return ;
 }

 int main(int argc,char *argv[])
 {
　　　　...
     struct timeval timeout;
     struct user use;
     time_t now; struct tm *timenow;

　　　　...
     while()
     {
　　　　　　　...
         switch(select(max_servfd+,&servfd,NULL,NULL,&timeout))
         {
             case -:
                 perror("select error");
                 break;
             case :
                 break;
             default:
                 //printf("has datas to offer accept\n");
                 if(FD_ISSET(sockfd,&servfd))//sockfd 有数据表示可以进行accept
                 {
                     　　...
                     printf("客户端发来的用户名是:%s,密码:%s\n",use.name,use.pwd);
                     memset(recvBuf,,sizeof(recvBuf));
                     if(mysql_check_login(use))
                     {
                         printf("验证成功！\n");
                         strcpy(sendBuf,"yes");
                     }
                     else
                     {
                         printf("验证失败！\n");
                         strcpy(sendBuf,"no");
                     }
                     if((sendSize=send(clientfd,sendBuf,MAX_DATA_SIZE,))==-)
                     {
                         perror("fail to receive datas");
                     }
　　　　　　　　　　　　　　...
                 }
                 break;
         }
         //FD_COPY(recvfd,servfd);
         for(i=;i<MAX_CON_NO;i++)//最大队列进行判断，优化的话，可以使用链表
         {
　　　　　　　　...
         }

         switch(select(max_recvfd+,&recvfd,NULL,NULL,&timeout))
         {
　　　　　　　　...
         }
     }
     return ;
 }

　　本次就增加了一个mysql_check_login这个函数用于用户名和密码的判断。也没有加入太多的功能。只要有个大概思路就可以了，剩下的注册和修改基本相似，就不加入到程序中了。

　　接下来还是老规矩，上一张运行时的截图

　　数据库中的用户名是user1，密码是123456。看，完成了验证了。顿时感觉好极了。

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