linux中简单聊天程序实现原理,基于Linux C 的简单聊天程序

服务器端：

// tcp-server.c

//tcp server

#include

#include

#include

#include

#include

#include

#include

#define SERVER_PORT 8080

#define NET_IP "127.0.0.1"

#define MAX_LINK 10

#define BUF_SIZE 1024

#define STATU_CONNECT 1

#define STATU_DISCONN 0

int main(int argc, char** argv)

{

char buf[BUF_SIZE],tmp_buf[BUF_SIZE];

unsigned char conn_fd[FD_SETSIZE];

struct sockaddr_in server_sin,client_sin; //ip

int server_sock_fd,client_sock_fd; //socket 文件描述符

int fd,tmp_fd;//temp fd

int ctrl_flag=1; //端口复用标志

int recvLen;

socklen_t client_len; //client_sin 长度，发送数据时用到

fd_set inset,tmp_inset;

server_sock_fd=socket(AF_INET,SOCK_STREAM,0);

if(server_sock_fd<0)

{

perror("socket");

exit(1);

}

printf("socket fd is %d\n",server_sock_fd);

bzero(&server_sin,sizeof(server_sin));

server_sin.sin_family=AF_INET;

server_sin.sin_port=htons(SERVER_PORT);

server_sin.sin_addr.s_addr=htonl(INADDR_ANY);//用这句绑定的是默认ip

if(argc==2)

{

if(inet_aton(argv[1],&server_sin.sin_addr)<0)//这里绑定的是指定地址

{

perror("inet_aton");

close(server_sock_fd);

exit(1);

}

}

setsockopt(server_sock_fd,SOL_SOCKET,SO_REUSEADDR,&ctrl_flag,sizeof(ctrl_flag));//设置端口重用

if(bind(server_sock_fd,(struct sockaddr *)&server_sin,sizeof(server_sin))<0)

{

perror("bind");

close(server_sock_fd);

exit(1);

}

if(listen(server_sock_fd,MAX_LINK)<0)

{

perror("listen");

close(server_sock_fd);

exit(1);

}

printf("listening...\n");

FD_ZERO(&inset);

FD_SET(server_sock_fd,&inset);

FD_SET(0,&inset);

conn_fd[server_sock_fd]=STATU_CONNECT;

while(1)

{

FD_ZERO(&tmp_inset);

tmp_inset=inset;

client_len=sizeof(client_sin);

memset(buf,0,sizeof(buf));

memset(tmp_buf,0,sizeof(tmp_buf));

if(!select(FD_SETSIZE,&tmp_inset,NULL,NULL,NULL)>0)//监控输入文件表述符

{

perror("select");

}

for(fd=0;fd

{

if(FD_ISSET(fd,&tmp_inset)>0)

{

if(fd==0)

{

if(read(0,buf,sizeof(buf))>0)

{

buf[strlen(buf)]='\0';

if(strcmp(buf,"exit\n")==0)

{

exit(0);

}

sprintf(tmp_buf,"system message: %s",buf);

for(tmp_fd=0;tmp_fd

{

if(conn_fd[tmp_fd] && tmp_fd!=server_sock_fd && tmp_fd!=fd)

{

send(tmp_fd,tmp_buf,strlen(tmp_buf),0);

}

}

}

}

else if(fd==server_sock_fd)

{

if((client_sock_fd=accept(server_sock_fd,(struct sockaddr *)&client_sin,&client_len))==-1)

{

perror("accept");

close(server_sock_fd);

exit(1);

}

FD_SET(client_sock_fd,&inset);

conn_fd[client_sock_fd]=STATU_CONNECT;

sprintf(buf,"new connection : fd=%d\t ip=%s\n",client_sock_fd,inet_ntoa(client_sin.sin_addr));

printf("%s",buf);

for(tmp_fd=0;tmp_fd

{

if(conn_fd[tmp_fd] && tmp_fd!=server_sock_fd && tmp_fd!=fd)

{

send(tmp_fd,buf,strlen(buf),0);

}

}

client_sock_fd=-1;

}

else

{

recvLen=recv(fd,buf,sizeof(buf),0);

if(recvLen<=0)

{

sprintf(buf,"client %d(socket) has disconnect.\n",fd);

printf("client %d(socket) has disconnect.\n",fd);

FD_CLR(fd,&inset);

conn_fd[fd]=STATU_DISCONN;

close(fd);

for(tmp_fd=0;tmp_fd

{

if(conn_fd[tmp_fd] && tmp_fd!=server_sock_fd && tmp_fd!=fd)

{

send(tmp_fd,buf,strlen(buf),0);

}

}

}

else

{

buf[strlen(buf)]='\0';

sprintf(tmp_buf,"fd %d say: %s",fd,buf);

printf("recieve from client %d: %s",fd,buf);

for(tmp_fd=0;tmp_fd

{

if(conn_fd[tmp_fd] && tmp_fd!=server_sock_fd && tmp_fd!=fd)

{

send(tmp_fd,tmp_buf,strlen(tmp_buf),0);

}

}

}//if(recvLen<=0)

}//if(fd==server_sock_fd)

}//if(FD_ISSET(fd,&tmp_inset)>0)

}//for(fd=0;fd

}//while(1)

for(tmp_fd=0;tmp_fd

{

if(conn_fd[tmp_fd] && tmp_fd!=server_sock_fd)

{

close(tmp_fd);

}

}

close(server_sock_fd);

return 0;

}

编译：gcc -o talk_tcp_server talk_tcp_server.c

客户端：

//talk_tcp_client.c

//talk client

#include

#include

#include

#include

#include

#include

#include

#include //gethostbyname()要用到

#include

#include

#define BUF_SIZE 1024

#define NET_IP "172.0.0.1"

#define NET_PORT 8080

void release_data(void *data)

{

if(data!=NULL)

{

free(data);

data=NULL;

}

}

void *recv_msg(void *arg)

{

char buf[BUF_SIZE];

int recvLen;

int fd;

fd=*((int *)arg);

while(1)

{

memset(buf,0,sizeof(buf));

recvLen=recv(fd,buf,BUF_SIZE,0);

if(recvLen<=0)

{

printf("connect has discard.\n");

close(fd);

exit(1);

}

else

{

buf[recvLen]='\0';

printf("%s",buf);

}

}

}

int main(int argc, char** argv)

{

char buf[BUF_SIZE];

struct sockaddr_in server_sin;

pthread_t tid;

pthread_attr_t attr;

struct hostent *host;

int sock_fd;

int recvLen;

memset(&server_sin,0,sizeof(server_sin));

server_sin.sin_family=AF_INET;

server_sin.sin_port=htons(NET_PORT);

if(argc > 2)

{

printf("Usage: tcp_client [host name/host ip]\n");

return -1;

}

if(argc==2)

{

if((host=gethostbyname(argv[1]))==NULL)

{

perror("gethostbyname");

return -1;

}

//server_sin.sin_addr=*((struct in_addr *)host->h_aliases);

}

else

{

if((host=gethostbyname(NET_IP))==NULL)

{

perror("gethostbyname");

return -1;

}

}

server_sin.sin_addr=*((struct in_addr *)host->h_aliases);

if((sock_fd=socket(AF_INET,SOCK_STREAM,0))<0)

{

perror("socket");

return -1;

}

if(connect(sock_fd,(struct sockaddr *)&server_sin,sizeof(server_sin))<0)

{

perror("connect");

close(sock_fd);

return -1;

}

pthread_attr_init(&attr);

pthread_attr_setdetachstate(&attr,PTHREAD_CREATE_DETACHED);

pthread_create(&tid,&attr,recv_msg,(void *)&sock_fd);

sleep(1);

do

{

memset(buf,0,sizeof(buf));

fgets(buf,sizeof(buf),stdin);

if(strcmp(buf,"exit\n")==0)

{

break;

}

send(sock_fd,buf,strlen(buf),0);

}while(1);

pthread_cancel(tid);

//pthread_join(tid);

pthread_attr_destroy(&attr);

close(sock_fd);

return 0;

}

编译：gcc -o talk_tcp_client talk_tcp_client.c -lpthread

##################################

运行：

在一个终端里运行talk_tcp_server:

./talk_tcp_server

在另一个终端运行talk_tcp_client: ./talk_tcp_client 127.0.0.1       //这里的ip根据情况自行设置，本机的就用127.0.0.1或者localhost或者192.168.0.102(这是我的局域网内的ip)