基于socket编程的多人聊天室

效果图：

　　　　    

    server.c

/* 服务器端 server.c */ 
#include <glib.h>
#include <stdio.h>
#include <fcntl.h>
#include <signal.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/time.h>
#include <unistd.h> 
#include <netdb.h> 
#include <netinet/in.h>
#include <arpa/inet.h>  
#define MAX_USERS 800   //最大访问人数

//定义用户数据结构
struct _client {
　　 gint sd; 
　　 gboolean in_use; 
 　　gchar name[64]; 
 　　gchar buf[1024]; 
       　　　　 };
typedef struct _client client;

//定义用户数据区
client user[MAX_USERS];
//定义服务线程
void do_service (gpointer id){ 
 　　  gint j; 
  　　 char tobuf[1024]; 
　　　　while(read(user[GPOINTER_TO_INT(id)].sd,user[GPOINTER_TO_INT(id)].buf,1024)!=-1)
　　　　{
     　　 sprintf(tobuf,"%s：%s
",user[GPOINTER_TO_INT(id)].name,
     　　 user[GPOINTER_TO_INT(id)].buf);
         for(j=0; j<MAX_USERS; j++)
　　　　　　　　{ 
        　　if(user[j].in_use)
       　　 { 
            write(user[j].sd,tobuf,1024);
            //g_print("%s",tobuf);
        　　}
         }
　　　　}
//
　　　  user[GPOINTER_TO_INT(id)].in_use = FALSE;
　　   close(user[GPOINTER_TO_INT(id)].sd);
　　　　exit(0);

} 
int main(int argc, char* argv[]) 
{
 　　gint sd, newsd;
　　 struct sockaddr_in *sin; 
　　 gint slen;
　　 gint count = 0; 
　　 gint flags; 
　　 gchar buf[1024]; 
　　 gchar tobuf[1024]; 
　　 gint length,i,j; 
　　 if(!g_thread_supported())
   　　　　 g_thread_init(NULL);
    else 
   　　　　 g_print("thread not supported
"); 

 63     sd = socket(AF_INET,SOCK_STREAM,0);   //1.......创建套接子
　　if(sd == -1)
　　{ 
　　    g_print("create socket error!
"); 
       return -1;
　　}     
   　　  sin = g_new(struct sockaddr_in,1);
   　 　 sin->sin_family = AF_INET;
         sin->sin_port = htons(1234);
         slen = sizeof(struct sockaddr_in); 
        sin->sin_addr.s_addr=inet_addr("127.0.0.1");

　　if(bind(sd,(struct sockaddr*)sin,slen)<0) //2......绑定
　　{ 
   　　 g_print("bind error!
");
   　　  return -1; 
　　} 
　　if(listen(sd,8)<0)        //3......开启监听
　　{
       g_print("listen error!
"); return -1; 
　　} 
　　for(i=0; i<MAX_USERS; i++)    //判断访问人数
  　　  user[i].in_use = FALSE; 
   　　 flags = fcntl(sd,F_GETFL);
   　　 fcntl(sd,F_SETFL,flags&~O_NDELAY); 
 88　　 for(;;)
　　{ 
   　　  newsd = accept(sd,sin,&slen); //4....接收
   　　　if(newsd == -1){ 
       　　 g_print("accept error!
"); 
       　　 break; 
　　　　　　　　　　　　　 }
　　　　else 
 96　　　　 { 
 97　　　　　　 if(count >= MAX_USERS) 
　　　　　　{
  　　　　　　 sprintf(buf,"用户数量过多服务器不能连接。
");
  　　　　　　 write(newsd,buf,1024);
 　　　　　　  close(newsd);
　　　　　　}
　　　　　　else 
　　　　　　{
   　　　　　　 flags = fcntl(user[i].sd,F_GETFL); 
   　　　　　　 fcntl(user[i].sd,F_SETFL,O_NONBLOCK);  
    　　　　　　user[count].sd = newsd;  
   　　　　　　 user[count].in_use = TRUE; 
   　　　　　　 read(newsd,user[count].name,64); 
　　　　　　　　// 创建为用户服务的线程 
   　　　　　　g_thread_create((GThreadFunc)do_service, 
 
  　　　　　　(gpointer)count,TRUE,NULL); 
 　　　　　　 count++; 
    　　　　}
      }
　}//
  　　for(;;) 
 　　 close(sd);
 　　 g_free(sin);
} 

client.c

/* 客户端 client.c */
#include <gtk/gtk.h>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h> 
#include</usr/include/mysql/mysql.h>
#define OURPORT 8088 //定义端口号
#define MAXLINE 800

static GtkWidget *text1;
static GtkWidget *text2;
static GtkWidget *login;
static GtkWidget *entry_id;
static GtkWidget *entry_pwd;
static GtkWidget *regwindow;

static GtkWidget *reg_entry1;
static GtkWidget *reg_entry2;

static GtkWidget *window1;//聊天室主窗口
static GtkWidget *win;//昵称登录窗口
static GtkWidget *text_view;

static GtkTextBuffer *buffer; //显示对话内容的文本显示缓冲区
static GtkTextBuffer *buffer1;
static GtkTextBuffer *buffer2;
static GtkWidget *textview1; //显示输入消息的单行录入控件
static GtkWidget *name_entry;//输入用户名的单行录入控件
static GtkWidget *login_button; //登录按钮
void btn_cal_clicked();
void on_send (GtkButton* button, gpointer data) ;
void update_widget_bg(GtkWidget * widget, gchar * img_file);

gint sd; //套接字句柄
struct sockaddr_in s_in; //套接字数据结构
gchar username[64]; //用户名
gchar buf[1024]; //写缓冲区
gchar get_buf[1048]; //读缓冲区
gboolean isconnected = FALSE; // 定义逻辑值表示是否连接

MYSQL_RES *rs;
MYSQL_ROW row;
MYSQL_RES *exe_sql(char sql[MAXLINE])
{
    const char *host = "localhost";
    const char *user = "root";
    const char *pass = "";
    const char *db = "talk";
    
    /* 定义mysql变量 */
    MYSQL_RES *result;
    MYSQL * mysql;
    if(!(mysql=mysql_init(NULL)))
    {
        printf("mysql_init wrong!");
        mysql_close(mysql);
        exit(0);
    }
    /* 连接数据库 */
    if (!mysql_real_connect(mysql, host, user, pass, db, 3306, NULL, 0)) 
    {
        printf("connect wrong!");
        mysql_close(mysql);
        exit(0);
    }
    printf("%s
",sql);
    if(mysql_query(mysql,sql))
    {
        printf("mysql_query_wrong!");
        mysql_close(mysql);
        exit(0);
    }
    result = mysql_store_result(mysql);    /* 获取查询结果 */
    return result;
}
//socket通信
void get_message(void) 
{  
    GtkTextIter iter; 
        gchar get_buf[1024]; 
        gchar buf[1024]; 
    while(read(sd,buf,1024) != -1) // 只要读取数据成功就循环执行
     { 
        sprintf(get_buf,"%s",buf);
        printf("%s
",buf);
        gdk_threads_enter(); //进入
        gtk_text_buffer_get_end_iter(buffer,&iter);
        gtk_text_buffer_insert(buffer,&iter,get_buf,-1);//显示读取的数据
        gdk_threads_leave(); //离开
    }
}
// 连接多人聊天服务器
gboolean do_connect(void) 
{  
    GtkTextIter iter; 
    gint slen; 
    sd = socket(AF_INET,SOCK_STREAM,0);//创建
    printf("%d
",sd);
    if(sd < 0)
    { 
        gtk_text_buffer_get_end_iter(buffer,&iter);
        gtk_text_buffer_insert(buffer,&iter,"打开套接字时出错！
",-1);
    return FALSE; 
    }       
    s_in.sin_family = AF_INET;
    s_in.sin_port = htons(1234);
    slen = sizeof(s_in);
    if(connect(sd,&s_in,slen) < 0) //
    { 
        gtk_text_buffer_get_end_iter(buffer,&iter);
        gtk_text_buffer_insert(buffer,&iter,"连接服务器时出错！
",-1);
        return FALSE; 
    }
    else 
    { 

        gtk_text_buffer_get_end_iter(buffer,&iter);
        gtk_text_buffer_insert(buffer,&iter,username,-1);
        gtk_text_buffer_get_end_iter(buffer,&iter);
        gtk_text_buffer_insert(buffer,&iter,"
成功与服务器连接....
",-1);
        //
        write(sd,username,64);//向服务器发送用户名//
        isconnected = TRUE;
        return TRUE; 
     }
} 
void on_send (GtkButton* button, gpointer data) 
{ 
    //向服务器发送数据
    const char* message; 
    GtkTextIter start;
    GtkTextIter end;
    char sql[MAXLINE];
     if(isconnected==FALSE)
        return; 
    buffer1=gtk_text_view_get_buffer(GTK_TEXT_VIEW(textview1));
    gtk_text_buffer_get_start_iter(buffer1,&start);
    gtk_text_buffer_get_end_iter(buffer1,&end);
    message=gtk_text_buffer_get_text(buffer1,&start,&end,TRUE);
    sprintf(buf,"%s
",message);
    write(sd,buf,1024);//发送 
    sprintf(sql,"insert into history (user,message,date) values ('%s','%s',now())",username,message);
    
    rs = exe_sql(sql);
    gtk_text_buffer_set_text(buffer1,"",-1);////清除文字

}     
void on_login(GtkWidget *button, gpointer data)
{ 
    //点击登录按钮时执行 
    create_win();
} 
void cls()//清屏
{
    gtk_text_buffer_set_text(buffer,"",-1);
}
void history()
{
    int rows_num;
    int count;
    GtkBuilder *gb;
    //查看历史窗口
    GtkWidget *history_win;//查看聊天历史界面
    GtkWidget *history_view;
    char *sql[MAXLINE];
    gb=gtk_builder_new();
    gtk_builder_add_from_file(gb,"history.glade",NULL);
    history_win=GTK_WIDGET(gtk_builder_get_object(gb,"history_win"));
    history_view=GTK_WIDGET(gtk_builder_get_object(gb,"history_view"));
    update_widget_bg(history_win, "ico/bg.jpg");
    gtk_widget_show(GTK_WIDGET(history_win));
    //显示数据
    strcpy(sql,"select * from history where date <=now() and date >=now()-interval 5 minute");
    rs=exe_sql(sql);
    printf("%s
",sql);
    rows_num=mysql_num_rows(rs);

    for(count=0;count<rows_num;count++)
    {
        GtkTextIter end;
        row=mysql_fetch_row(rs);
        buffer1 = gtk_text_view_get_buffer(GTK_TEXT_VIEW(history_view));
        gtk_text_buffer_get_end_iter(buffer1,&end);
        gtk_text_buffer_insert(buffer1, &end, row[1], -1);  
        gtk_text_buffer_insert(buffer1, &end, " : ", -1);
        gtk_text_buffer_insert(buffer1, &end, row[2], -1); 
        gtk_text_buffer_insert(buffer1, &end, "  ", -1);
        gtk_text_buffer_insert(buffer1, &end, row[3], -1); 
        gtk_text_buffer_insert(buffer1, &end, "
", -1);
    
    }
}
//当点击登录窗口的登录按钮时执行
void on_button_clicked(GtkButton *button, gpointer data)
{
      buffer = gtk_text_view_get_buffer(GTK_TEXT_VIEW(text_view));  
    gtk_text_buffer_set_text(buffer, "Hello    ", -1); 
    gchar *name;    
    name = gtk_entry_get_text(GTK_ENTRY(name_entry));
    
    sprintf(username,"%s",name);

    if(do_connect())
    { 
        gtk_widget_set_sensitive(login_button,FALSE);
        g_thread_create((GThreadFunc)get_message,NULL,FALSE,NULL);
    }
        gtk_widget_destroy(data);
    } 

//当前关闭登录窗口时执行 
void on_destroy(GtkWidget *widget, GdkEvent *event, gpointer data)
{
    sprintf(username,"guest");
    if(do_connect()==TRUE)
    { 
        gtk_widget_set_sensitive(login_button,FALSE);
        g_thread_create((GThreadFunc)get_message,NULL,FALSE,NULL);
    }
    gtk_widget_destroy(widget);
} 
//背景图片的显示功能
void update_widget_bg(GtkWidget * widget, gchar * img_file)
{
    GtkStyle *style;
    GdkPixbuf *pixbuf;
    GdkPixmap *pixmap;
    gint width, height;

    pixbuf = gdk_pixbuf_new_from_file(img_file, NULL);
    width = gdk_pixbuf_get_width(pixbuf);
    height = gdk_pixbuf_get_height(pixbuf);
    pixmap = gdk_pixmap_new(NULL, width, height, 24);
    gdk_pixbuf_render_pixmap_and_mask(pixbuf, &pixmap, NULL, 0);
    style = gtk_style_copy(GTK_WIDGET(widget)->style);
    if (style->bg_pixmap[GTK_STATE_NORMAL])
        g_object_unref(style->bg_pixmap[GTK_STATE_NORMAL]);
    style->bg_pixmap[GTK_STATE_NORMAL] = g_object_ref(pixmap);
    style->bg_pixmap[GTK_STATE_ACTIVE] = g_object_ref(pixmap);
    style->bg_pixmap[GTK_STATE_PRELIGHT] = g_object_ref(pixmap);
    style->bg_pixmap[GTK_STATE_SELECTED] = g_object_ref(pixmap);
    style->bg_pixmap[GTK_STATE_INSENSITIVE] = g_object_ref(pixmap);
    gtk_widget_set_style(GTK_WIDGET(widget), style);
    g_object_unref(style);
}
void btn_cal(GtkWidget *dialog,gpointer data)
{
    gtk_widget_destroy(win);
    return FALSE;
}
void create_win(void)
{     //昵称窗口
    GtkBuilder *gb;
    GtkWidget *btn_ok;
    gb=gtk_builder_new();
    gtk_builder_add_from_file(gb,"log.glade",NULL);
        win=GTK_WIDGET(gtk_builder_get_object(gb,"win"));
    name_entry=GTK_WIDGET(gtk_builder_get_object(gb,"name_entry"));
    btn_ok=GTK_WIDGET(gtk_builder_get_object(gb,"btn_ok"));    
    update_widget_bg(win, "ico/bg.jpg");
    gtk_widget_show(GTK_WIDGET(win));
    g_signal_connect(GTK_OBJECT(btn_ok),"clicked",G_CALLBACK(on_button_clicked),win);
} 
//聊天室关闭按钮
void button1_clicked()
{
    close(sd);//关闭 
    gtk_main_quit();
}
/* 弹出对话框代码开始*/
void show_info(char *msgtip)
{
    GtkWidget *dialog;
    dialog = gtk_message_dialog_new(NULL, GTK_DIALOG_DESTROY_WITH_PARENT, GTK_MESSAGE_INFO, GTK_BUTTONS_OK, msgtip);
    gtk_window_set_title(GTK_WINDOW(dialog), "Information");
    gtk_dialog_run(GTK_DIALOG(dialog));
    gtk_widget_destroy(dialog);
}
/*glade主界面*/
#define w_(builder,type,name) name=GTK_##type(gtk_builder_get_object(builder,#name))
GtkBuilder* gtk_load_glade(gchar* filename)

{

  GtkBuilder *gb;

  gb=gtk_builder_new();

  if(!gtk_builder_add_from_file(gb,filename,NULL)) 
      return NULL;

  gtk_builder_connect_signals(gb,NULL);

  return gb;

}
void cal_widget_init()

{

  gtk_widget_show(GTK_WIDGET(window1));

}

void cal_get_widgets(GtkBuilder* gb)

{

  w_(gb,WINDOW,login);

  w_(gb,ENTRY,entry_id);
  w_(gb,ENTRY,entry_pwd);
 
  w_(gb,WINDOW,window1);

  w_(gb,TEXT_VIEW,text_view);
  w_(gb,WIDGET,textview1);
  w_(gb,ENTRY,name_entry);

}
//登录按钮的响应函数
void mysql_connect()
{
    char sql[MAXLINE];
    const gchar *entry1;
    const gchar *entry2;
    entry1=gtk_entry_get_text(GTK_ENTRY(entry_id));
    entry2=gtk_entry_get_text(GTK_ENTRY(entry_pwd));

    if (strcmp(entry1, "") == 0) {
        show_info("帐号不能为空！");
        return;
    }
    if (strcmp(entry2, "") == 0) {
        show_info("密码不能为空！");
        return;
    }

    sprintf(sql,"select count(*) from users where user='%s' and passwd='%s'",entry1,entry2);
    
    rs=exe_sql(sql);
    
        row=mysql_fetch_row(rs);
    
    const gchar *a=row[0];
    
    if(strcmp(a,"1")==0)
        {
            show_info("成功登录！");
             if(!g_thread_supported())
            g_thread_init(NULL); 
            //初始线程 

              GtkBuilder *gb;

              gb=gtk_load_glade("try1.glade");
            gtk_widget_destroy(login);
            if(gb==NULL)  
            return -1;

              cal_get_widgets(gb);

              cal_widget_init();
            return 0; 
        }
    else
        
        show_info("用户名或密码失败
");
        return;
}
//创建button按钮
GtkWidget *create_button()
{
    GtkWidget *button = gtk_button_new();
    return button;
}
//退出按钮
void btn_cal_clicked(GtkWidget *dialog,gpointer data)
{
    gtk_widget_destroy(regwindow);
    return FALSE;
}
//注册的响应函数
void reg_btn2_clicked()
{
    int rows;
    MYSQL_ROW *mysql_row; 
    const char *name=gtk_entry_get_text(GTK_ENTRY(reg_entry1));
    const char *passwd=gtk_entry_get_text(GTK_ENTRY(reg_entry2));
    if (strcmp(name, "") == 0) 
    {
        show_info("帐号不能为空！");
        return;
    }
    if (strcmp(passwd, "") == 0)
    {
        show_info("密码不能为空！");
        return;
    }
    char sql[MAXLINE];
    char query[MAXLINE];
    sprintf(query,"select * from users where user='%s'",name);
    rs = exe_sql(query);
    rows=mysql_num_rows(rs);
    //mysql_row=mysql_fetch_row(rs);
    //const gchar *b=row[0];
    
    if(rows>=1)
    {    
        show_info("用户名已被注册
");
        return;
    }
    else
    sprintf(sql, "insert into users(user,passwd) values('%s','%s')",name,passwd);
    rs = exe_sql(sql);
    show_info("注册成功");
}
//注册响应函数+注册界面
void btn_reg_clicked()
{
    GtkBuilder *gb;
    GtkWidget *reg_btn1;
    GtkWidget *reg_btn2;
    gb=gtk_builder_new();
    gtk_builder_add_from_file(gb,"reg.glade",NULL);
        regwindow=GTK_WIDGET(gtk_builder_get_object(gb,"regwindow"));
    reg_entry1=GTK_WIDGET(gtk_builder_get_object(gb,"reg_entry1"));
    reg_entry2=GTK_WIDGET(gtk_builder_get_object(gb,"reg_entry2"));
    reg_btn1=GTK_WIDGET(gtk_builder_get_object(gb,"reg_btn1"));
    reg_btn2=GTK_WIDGET(gtk_builder_get_object(gb,"reg_btn2"));
    update_widget_bg(regwindow, "ico/bg.jpg");    
    gtk_widget_show(GTK_WIDGET(regwindow));
    g_signal_connect(GTK_OBJECT(reg_btn1),"clicked",G_CALLBACK(btn_cal_clicked),regwindow);
    g_signal_connect(GTK_OBJECT(reg_btn2),"clicked",G_CALLBACK(reg_btn2_clicked),regwindow);
}
GtkWidget *create_window() //创建窗体和控件!!
{
    gtk_init(NULL, NULL); //初始化
      GtkBuilder *gb;

      gb=gtk_load_glade("client.glade");

      cal_get_widgets(gb);
      update_widget_bg(login, "ico/bg.jpg");

      gtk_widget_show_all(login); //显示窗体和控件
      return 0; 
}
int main(int argc,char *argv[])
{
    
    create_window(); //创建窗体和控件!!
    gtk_main(); //事件循环处理
    return 0;
}

由于编写时调用了glade界面和mysql数据库，界面和表就不再上传了。