Socket網絡編程--聊天程序(7)

  接上一小節,原本是計劃這一節用來說數據庫的增刪改查,可是在實現的過程當中,出現了一點小問題,也不是技術的問題,就是在字符界面上比較很差操做。好比要註冊一個賬號,就須要弄個字符界面提示,而後輸入數字表示選擇,在依次輸入信息。(這一點,用C寫過什麼管理系統就知道,非常麻煩。)考慮到本程序講的是網絡編程,就不弄增刪改查了,就判斷一個用戶名密碼是否正確就好了。html

  首先是安裝mysql和建立數據庫,所用到的命令以下:mysql

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

  建立一個users數據庫和一個client表sql

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 }
View Code

  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 }
View Code

  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 }
View Code

  4.查詢數據ide

 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
View Code

  因爲本次的重點不是數據庫的操做,因此就簡單的給出經常使用的基本方法,在咱們的聊天程序中只加入查詢用戶密碼的功能,其餘的如註冊用戶什麼的就不寫了,應該沒有什麼難度,就是有點煩。fetch

  加入數據庫鏈接後的聊天程序優化

  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/p/3876134.html 

相關文章
相關標籤/搜索