Linux Socket 网络编程 基于GTK+ 的多线程实现的局域网通信软件

本程序的通信是基于UDP协议的传输实现

服务器端：

server.h

/*********************************** * server.h * * the header files and functions * ***********************************/ #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <stdio.h> #include <errno.h> #include <stdlib.h> #include <string.h> #include <semaphore.h> #include <unistd.h> #include <pthread.h> #include <arpa/inet.h> #define bool _Bool #define true 1 #define false 0 #define PORT 8888 #define MAXSIZE 1024 void *recv_func(void *arg); int build_socket(); void send_text(void); void send_func(const char *); void show_remote_text(char rcvd_mess[]); void show_err(char *err); void clean_send_text(void); void startup(void);

se_sock.c

/************************************************ * se_sock.c * * function: * * create a socket * * create a new thread,for receive function * * send message * ************************************************/ #include "server.h" socklen_t len; bool from_host;//表示客户端是否已经向本服务器发送来信息 int sockfd; struct sockaddr_in saddr,caddr;//saddr表示本地服务器地址信息，caddr用于保存向本服务器发送信息的主机地址信息 int build_socket() { int res; pthread_t recv_thread; pthread_attr_t thread_attr; len=sizeof(caddr); /* Set status of thread */ res=pthread_attr_init(&thread_attr); if(res!=0) { show_err("Setting detached attribute failed"); exit(EXIT_FAILURE); } /* Create a socket */ if((sockfd=socket(AF_INET,SOCK_DGRAM,0))<0) { show_err("Socket Error"); exit(1); } bzero(&saddr,sizeof(struct sockaddr)); saddr.sin_family=AF_INET; saddr.sin_addr.s_addr=htonl(INADDR_ANY); saddr.sin_port=htons(PORT); if(bind(sockfd,(struct sockaddr *)&saddr,sizeof(struct sockaddr_in))==-1) { show_err("Bind Error"); exit(1); } /* Set the status of thread,don't wait for return of the subthread */ res=pthread_attr_setdetachstate(&thread_attr,PTHREAD_CREATE_DETACHED); if(res!=0) { show_err("Setting detached attribute failed"); exit(EXIT_FAILURE); } /* Create a sub thread,call recv_func() */ res=pthread_create(&recv_thread,&thread_attr,recv_func,NULL); if(res!=0) { show_err("Thread create error"); exit(EXIT_FAILURE); } /* Callback the attribute of thread */ (void)pthread_attr_destroy(&thread_attr); } /* Receive function */ void *recv_func(void *arg) { char recv_text[MAXSIZE]; while(1) { /* To Receive message from client and get the address infomation */ if(recvfrom(sockfd,recv_text,sizeof(recv_text),0,(struct sockaddr*)&caddr,&len)<0) { show_err("server recv error"); exit(1); } if( from_host==false ) from_host=true; show_remote_text(recv_text); } } /* Send function,send message to client */ void send_func(const char * send_text) { /* If there is no text,continue */ if(strlen(send_text)==1) return; if( !from_host ) show_err("Unknown the client's address infomation/nPlease wait the client to connect you first !/n"); /* Send message */ if(from_host && sendto(sockfd,send_text,MAXSIZE,0,(struct sockaddr*)&caddr,len)<0) { show_err("S send error"); exit(1); } }

server.c

/************************************************* * server.c * * create the server window * * send message to client * *************************************************/ #include "server.h" #include<gtk/gtk.h> bool socket_start; GtkTextBuffer *show_buffer,*input_buffer; void quit_win(GtkWidget *,gpointer); int main(int argc,char **argv) { GtkWidget *window; GtkWidget *show_text,*input_text; GtkWidget *send_button,*quit_button,*cls_button,*start_button; GtkWidget *hbox,*vbox; GtkWidget *scrolled1,*scrolled2; GtkWidget *space_label; gtk_init(&argc,&argv); window=gtk_window_new(GTK_WINDOW_TOPLEVEL); gtk_window_set_title(GTK_WINDOW(window),"Server"); gtk_window_set_position(GTK_WINDOW(window),GTK_WIN_POS_CENTER); gtk_window_set_default_size(GTK_WINDOW(window),430,320); /* "quit" button */ g_signal_connect(GTK_OBJECT(window),"destroy",GTK_SIGNAL_FUNC(quit_win),NULL); space_label=gtk_label_new(" "); /* set button */ send_button=gtk_button_new_with_label("Send"); quit_button=gtk_button_new_with_label("Close"); cls_button=gtk_button_new_with_label("Clear"); start_button=gtk_button_new_with_label("Startup"); /* set textbox */ show_text=gtk_text_view_new(); input_text=gtk_text_view_new(); /* get the buffer of textbox */ show_buffer=gtk_text_view_get_buffer(GTK_TEXT_VIEW(show_text)); input_buffer=gtk_text_view_get_buffer(GTK_TEXT_VIEW(input_text)); /* set textbox to diseditable */ gtk_text_view_set_editable(GTK_TEXT_VIEW(show_text),FALSE); /* scroll window */ scrolled1=gtk_scrolled_window_new(NULL,NULL); scrolled2=gtk_scrolled_window_new(NULL,NULL); /* create a textbox */ gtk_scrolled_window_add_with_viewport(GTK_SCROLLED_WINDOW(scrolled1),show_text); gtk_scrolled_window_add_with_viewport(GTK_SCROLLED_WINDOW(scrolled2),input_text); /* setting of window */ gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolled1),GTK_POLICY_AUTOMATIC,GTK_POLICY_AUTOMATIC); gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolled2),GTK_POLICY_AUTOMATIC,GTK_POLICY_AUTOMATIC); hbox=gtk_hbox_new(FALSE,2); vbox=gtk_vbox_new(FALSE,2); /* click quit to call quit_win*/ g_signal_connect(GTK_OBJECT(quit_button),"clicked",GTK_SIGNAL_FUNC(quit_win),NULL); /* click to clear input screen*/ g_signal_connect(GTK_OBJECT(cls_button),"clicked",GTK_SIGNAL_FUNC(clean_send_text),NULL); /* click Start to build socket*/ g_signal_connect(GTK_OBJECT(start_button),"clicked",GTK_SIGNAL_FUNC(startup),NULL); /* create window */ gtk_box_pack_start(GTK_BOX(hbox),start_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),space_label,TRUE,TRUE,2); gtk_box_pack_start(GTK_BOX(hbox),send_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),cls_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),quit_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(vbox),scrolled1,TRUE,TRUE,2); gtk_box_pack_start(GTK_BOX(vbox),scrolled2,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(vbox),hbox,FALSE,FALSE,2); gtk_container_add(GTK_CONTAINER(window),vbox); /* click send button ,then call send_text*/ gtk_signal_connect(GTK_OBJECT(send_button),"clicked",GTK_SIGNAL_FUNC(send_text),NULL); gtk_widget_show_all(window); gtk_main(); return 0; } /* show the input text */ void show_local_text(const gchar* text) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"Me:/n",4);/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,text,strlen(text));/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"/n",1);/*插入文本到缓冲区*/ } /* clean the input text */ void clean_send_text() { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(input_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_delete(GTK_TEXT_BUFFER(input_buffer),&start,&end);/*插入到缓冲区*/ } /* get the input text,and send it */ void send_text() { GtkTextIter start,end; gchar *text; /* Socket creating has succeed ,so send message */ text=(gchar *)malloc(MAXSIZE); if(text==NULL) { printf("Malloc error!/n"); exit(1); } /* get text */ gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(input_buffer),&start,&end); text=gtk_text_buffer_get_text(GTK_TEXT_BUFFER(input_buffer),&start,&end,FALSE); /* If there is no input,do nothing but return */ if(strcmp(text,"")!=0) { send_func(text); clean_send_text(); show_local_text(text); } else show_err("The message can not be empty .../n"); free(text); } void startup(void) { if( !socket_start ) { build_socket(); socket_start=true; return; } show_err("The server has been started !/n"); } /* show errors such as "no input","haven't create sockt" etc. */ void show_err(char *err) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,err,strlen(err)); } /* show the received message */ void show_remote_text(char rcvd_mess[]) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"Client:/n",8);/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,rcvd_mess,strlen(rcvd_mess));/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"/n",1);/*插入换行到缓冲区*/ } /* quit */ void quit_win(GtkWidget *window,gpointer data) { gtk_main_quit(); }

客户端

client.h

/*********************************** * client.h * * the header files and functions * ***********************************/ #include <stdio.h> #include <string.h> #include <stdlib.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <arpa/inet.h> #include <errno.h> #include <unistd.h> #include <pthread.h> #define PORT 8888 #define MAXSIZE 1024 int build_socket(const char *); void send_text(void); void send_func(const char *); void *recv_func(void *); void show_remote_text(char rcvd_mess[]); void clean_send_text(void); void show_err(char *err);

cl_sock.c

/************************************************ * cl_sock.c * * function: * * create a socket * * create a new thread,for receive function * * send message * ************************************************/ #include "client.h" int sockfd; struct sockaddr_in saddr; int build_socket(const char *serv_ip) { int res; pthread_t recv_thread; pthread_attr_t thread_attr; /* set status of thread */ res=pthread_attr_init(&thread_attr); if(res!=0) { perror("Setting detached attribute failed"); exit(EXIT_FAILURE); } sockfd=socket(AF_INET,SOCK_DGRAM,0); /* create a socket */ if(sockfd==-1) { perror("Socket Error"); exit(1); } bzero(&saddr,sizeof(saddr)); saddr.sin_family=AF_INET; saddr.sin_port=htons(PORT); res=inet_pton(AF_INET,serv_ip,&saddr.sin_addr); if(res==0){ /* the serv_ip is invalid */ return 1; } else if(res==-1){ return -1; } /* set the stats of thread:means do not wait for the return value of subthread */ res=pthread_attr_setdetachstate(&thread_attr,PTHREAD_CREATE_DETACHED); if(res!=0) { perror("Setting detached attribute failed"); exit(EXIT_FAILURE); } /* Create a thread,to process the receive function. */ res=pthread_create(&recv_thread,&thread_attr,&recv_func,NULL); if(res!=0) { perror("Thread create error"); exit(EXIT_FAILURE); } /* callback the attribute */ (void)pthread_attr_destroy(&thread_attr); return 0; } /* send function */ void send_func(const char *text) { int n; socklen_t len=sizeof(saddr); n=sendto(sockfd,text,MAXSIZE,0,(const struct sockaddr*)&saddr,len); if(n<0) { perror("S send error"); exit(1); } } /* a new thread,to receive message */ void *recv_func(void *arg) { char rcvd_mess[MAXSIZE]; while(1) { bzero(rcvd_mess,MAXSIZE); if(recvfrom(sockfd,rcvd_mess,MAXSIZE,0,NULL,NULL)<0) /*阻塞直到收到客户端发的消息*/ { perror("server recv error"); exit(1); } show_remote_text(rcvd_mess); } }

client.c

/********************************* * Client.c * * Function: * * create the chat window * *********************************/ #include "client.h" #include <gtk/gtk.h> int issucceed=-1; GtkTextBuffer *show_buffer,*input_buffer; void get_ip(GtkWidget *,gpointer); void quit_win(GtkWidget *,gpointer); int main(int argc,char **argv) { GtkWidget *window; GtkWidget *show_text,*input_text,*ip_text; GtkWidget *ip_label,*space_label; GtkWidget *link_button,*send_button,*quit_button,*cls_button; GtkWidget *hbox,*vbox; GtkWidget *scrolled1,*scrolled2; gtk_init(&argc,&argv); window=gtk_window_new(GTK_WINDOW_TOPLEVEL); gtk_window_set_title(GTK_WINDOW(window),"Client"); gtk_window_set_position(GTK_WINDOW(window),GTK_WIN_POS_CENTER); gtk_window_set_default_size(GTK_WINDOW(window),430,320); /* "quit" button */ g_signal_connect(GTK_OBJECT(window),"destroy",GTK_SIGNAL_FUNC(quit_win),NULL); ip_label=gtk_label_new("IP:"); space_label=gtk_label_new(" "); /* set button */ link_button=gtk_button_new_with_label("Connect"); send_button=gtk_button_new_with_label("Send"); quit_button=gtk_button_new_with_label("Close"); cls_button=gtk_button_new_with_label("Clear"); /* set textbox */ ip_text=gtk_entry_new(); show_text=gtk_text_view_new(); input_text=gtk_text_view_new(); /* set length of IP box */ gtk_entry_set_max_length(GTK_ENTRY(ip_text),15); /* get the buffer of textbox */ show_buffer=gtk_text_view_get_buffer(GTK_TEXT_VIEW(show_text)); input_buffer=gtk_text_view_get_buffer(GTK_TEXT_VIEW(input_text)); /* set textbox to diseditable */ gtk_text_view_set_editable(GTK_TEXT_VIEW(show_text),FALSE); /* scroll window */ scrolled1=gtk_scrolled_window_new(NULL,NULL); scrolled2=gtk_scrolled_window_new(NULL,NULL); /* create a textbox */ gtk_scrolled_window_add_with_viewport(GTK_SCROLLED_WINDOW(scrolled1),show_text); gtk_scrolled_window_add_with_viewport(GTK_SCROLLED_WINDOW(scrolled2),input_text); /* setting of window */ gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolled1),GTK_POLICY_AUTOMATIC,GTK_POLICY_AUTOMATIC); gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(scrolled2),GTK_POLICY_AUTOMATIC,GTK_POLICY_AUTOMATIC); hbox=gtk_hbox_new(FALSE,2); vbox=gtk_vbox_new(FALSE,2); /* click quit to call quit_win*/ g_signal_connect(GTK_OBJECT(quit_button),"clicked",GTK_SIGNAL_FUNC(quit_win),NULL); /* click "connect" to call get_ip */ g_signal_connect(GTK_OBJECT(link_button),"clicked",GTK_SIGNAL_FUNC(get_ip),ip_text); /* click to clear input screen*/ g_signal_connect(GTK_OBJECT(cls_button),"clicked",GTK_SIGNAL_FUNC(clean_send_text),NULL); /* create window */ gtk_box_pack_start(GTK_BOX(hbox),ip_label,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),ip_text,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),link_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),space_label,TRUE,TRUE,2); gtk_box_pack_start(GTK_BOX(hbox),send_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),cls_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(hbox),quit_button,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(vbox),scrolled1,TRUE,TRUE,2); gtk_box_pack_start(GTK_BOX(vbox),scrolled2,FALSE,FALSE,2); gtk_box_pack_start(GTK_BOX(vbox),hbox,FALSE,FALSE,2); gtk_container_add(GTK_CONTAINER(window),vbox); /* click send button ,then call send_text*/ gtk_signal_connect(GTK_OBJECT(send_button),"clicked",GTK_SIGNAL_FUNC(send_text),NULL); gtk_widget_show_all(window); gtk_main(); return 0; } /* show the input text */ void show_local_text(const gchar* text) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"Me:/n",4);/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,text,strlen(text));/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"/n",1);/*插入文本到缓冲区*/ } /* clean the input text */ void clean_send_text() { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(input_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_delete(GTK_TEXT_BUFFER(input_buffer),&start,&end);/*插入到缓冲区*/ } /* get the input text,and send it */ void send_text() { GtkTextIter start,end; gchar *text; if(issucceed==-1){ /* Haven't create a socket */ show_err("Not connected.../n"); } else { /* Socket creating has succeed ,so send message */ text=(gchar *)malloc(MAXSIZE); if(text==NULL) { printf("Malloc error!/n"); exit(1); } /* get text */ gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(input_buffer),&start,&end); text=gtk_text_buffer_get_text(GTK_TEXT_BUFFER(input_buffer),&start,&end,FALSE); /* If there is no input,do nothing but return */ if(strcmp(text,"")!=0) { send_func(text); clean_send_text(); show_local_text(text); } else show_err("The message can not be empty.../n"); free(text); } } /* show errors such as "no input","haven't create sockt" etc. */ void show_err(char *err) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,err,strlen(err)); } /* get IP address and create socket */ void get_ip(GtkWidget *button,gpointer ip_text) { gchar *serv_ip; int res; serv_ip=(gchar *)gtk_entry_get_text(GTK_ENTRY((GtkWidget *)ip_text)); res=build_socket(serv_ip); if(res==1) show_err("IP Address is Invalid.../n"); else if(res==-1) show_err("Connect Failure... /n");/*插入文本到缓冲区*/ else{ show_err("Connect Successful... /n"); issucceed=0; } } /* show the received message */ void show_remote_text(char rcvd_mess[]) { GtkTextIter start,end; gtk_text_buffer_get_bounds(GTK_TEXT_BUFFER(show_buffer),&start,&end);/*获得缓冲区开始和结束位置的Iter*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"Server:/n",8);/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,rcvd_mess,strlen(rcvd_mess));/*插入文本到缓冲区*/ gtk_text_buffer_insert(GTK_TEXT_BUFFER(show_buffer),&end,"/n",1);/*插入换行到缓冲区*/ } /* quit */ void quit_win(GtkWidget *window,gpointer data) { gtk_main_quit(); }

makefile代码

OBJ=client.o cl_sock.o

OBJ1=server.o se_sock.o

server:server.o se_sock.o server.h

gcc -Wall $(OBJ1) -o server `pkg-config gtk+-2.0 --libs` -l pthread

server.o:server.c server.h

gcc -Wall -c $< `pkg-config gtk+-2.0 --cflags`

server_sock.o:se_sock.c server.h

gcc -Wall -c $<

client:client.o cl_sock.o client.h

gcc -Wall $(OBJ) -o client `pkg-config gtk+-2.0 --libs` -l pthread

client.o:client.c client.h

gcc -Wall -c $< `pkg-config gtk+-2.0 --cflags`

client_sock.o:cl_sock.c client.h

gcc -Wall -c $<

clean:

rm *.o