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(32),password varchar(32) ); //创建一个叫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,0,NULL,CLIENT_FOUND_ROWS))
    {
        printf("Connect Success!
");
        res=mysql_query(&conn,sql);
        if(res)
        {
            printf("Insert Error!
");
        }
        else
        {
            printf("Insert Success!
");
        }
    }
    else
    {
        printf("Connect Failed!
");
        exit(-1);
    }
    return 0;
}

　　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,0,NULL,CLIENT_FOUND_ROWS))
    {
        printf("Connect Success!
");
        res=mysql_query(&conn,sql);
        if(res)
        {
            printf("Delete Error!
");
        }
        else
        {
            printf("Delete Success!
");
        }
    }
    else
    {
        printf("Connect Failed!
");
        exit(-1);
    }
    return 0;
}

　　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,0,NULL,CLIENT_FOUND_ROWS))
    {
        printf("Connect Success!
");
        res=mysql_query(&conn,sql);
        if(res)
        {
            printf("Update Error!
");
        }
        else
        {
            printf("Update Success!
");
        }
    }
    else
    {
        printf("Connect Failed!
");
        exit(-1);
    }
    return 0;
}

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

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

　　对应的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
",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[256]={0};
    strcpy(sql,"select * from clients where username="");
    strcat(sql,su.name);
    strcat(sql,"" and password="");
    strcat(sql,su.pwd);
    strcat(sql,"";");
    printf("查询的sql:%s
",sql);

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

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

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

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

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

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

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

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