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