Linux之—网络通信（进程和线程实现并发通信）

进程：

 tcpserver_fork.c

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define MY_PORT 3333

int main(int argc ,char **argv)
{
    int listen_fd,accept_fd;
    struct sockaddr_in     client_addr;
    int n;
    int nbytes;
 
     if((listen_fd=socket(AF_INET,SOCK_STREAM,0))<0)
      {
            printf("Socket Error:%s\n\a",strerror(errno));
            exit(1);
      }
 
     bzero(&client_addr,sizeof(struct sockaddr_in));
     client_addr.sin_family=AF_INET;
     client_addr.sin_port=htons(MY_PORT);
     client_addr.sin_addr.s_addr=htonl(INADDR_ANY);
     n=1;
 
     /* 如果服务器终止后,服务器可以第二次快速启动而不用等待一段时间  */
     setsockopt(listen_fd,SOL_SOCKET,SO_REUSEADDR,&n,sizeof(int));
     if(bind(listen_fd,(struct sockaddr *)&client_addr,sizeof(client_addr))<0)
      {
            printf("Bind Error:%s\n\a",strerror(errno));
            exit(1);
      }
 
      listen(listen_fd,5);
      
      while(1)
      {
           accept_fd=accept(listen_fd,NULL,NULL);
           if((accept_fd<0)&&(errno==EINTR))
                  continue;
           else if(accept_fd<0)
            {
                printf("Accept Error:%s\n\a",strerror(errno));
                continue;
            }
          if((n=fork())==0)
           {
                /* 子进程处理客户端的连接 */
                char buffer[1024];

                if((nbytes=read(accept_fd,buffer,1024))==-1)
            {
                fprintf(stderr,"Read Error:%s\n",strerror(errno));
                exit(1);
            }         
            buffer[nbytes]='\0';
            printf("Server received %s\n",buffer);
                
                close(listen_fd);
                close(accept_fd);
                exit(0);
           }
           else
               close(accept_fd);
    }
}

线程：tcpserver_pthread.c

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define portnumber 3333

void * read_msg(void *arg)
{
    int fd = *((int *)arg);
	int nread = 0;
	char buffer[1024];

	while((nread = read(fd,buffer,sizeof(buffer))) > 0)
	{
	    buffer[nread] = '\0';
		printf("get client message: %s\n",buffer);
		memset(buffer,0,sizeof(buffer));
	}
}

void * write_msg(void * arg)
{
   
    int fd = *((int *)arg);
    
	while(1)
	{
	    write(fd,"hello",5);
	    sleep(2);
	}
}

int main(int argc, char *argv[])
{
	int sockfd,new_fd;
	struct sockaddr_in server_addr;
	struct sockaddr_in client_addr;
	int sin_size;
	int nbytes;
	char buffer[1024];

	pthread_t id;


	/* 服务器端开始建立sockfd描述符 */
	if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:IPV4;SOCK_STREAM:TCP
	{
		fprintf(stderr,"Socket error:%s\n\a",strerror(errno));
		exit(1);
	}

	/* 服务器端填充 sockaddr结构 */
	bzero(&server_addr,sizeof(struct sockaddr_in)); // 初始化,置0
	server_addr.sin_family=AF_INET;                 // Internet
	server_addr.sin_addr.s_addr=htonl(INADDR_ANY);  // (将本机器上的long数据转化为网络上的long数据)和任何主机通信  //INADDR_ANY 表示可以接收任意IP地址的数据，即绑定到所有的IP
	//server_addr.sin_addr.s_addr=inet_addr("192.168.1.1");  //用于绑定到一个固定IP,inet_addr用于把数字加格式的ip转化为整形ip
	server_addr.sin_port=htons(portnumber);         // (将本机器上的short数据转化为网络上的short数据)端口号

	/* 捆绑sockfd描述符到IP地址 */
	if(bind(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
	{
		fprintf(stderr,"Bind error:%s\n\a",strerror(errno));
		exit(1);
	}

	/* 设置允许连接的最大客户端数 */
	if(listen(sockfd,5)==-1)
	{
		fprintf(stderr,"Listen error:%s\n\a",strerror(errno));
		exit(1);
	}

	while(1)
	{
		/* 服务器阻塞,直到客户程序建立连接 */
		sin_size=sizeof(struct sockaddr_in);
		printf("accepting!\n");
		if((new_fd=accept(sockfd,(struct sockaddr *)(&client_addr),&sin_size))==-1)
		{
			fprintf(stderr,"Accept error:%s\n\a",strerror(errno));
			exit(1);
		}
		fprintf(stderr,"Server get connection from %s\n",inet_ntoa(client_addr.sin_addr)); // 将网络地址转换成.字符串
        
        pthread_create(&id,NULL,(void *)read_msg,(void *)&new_fd);
        pthread_create(&id,NULL,(void *)write_msg,(void *)&new_fd);

	}

	/* 结束通讯 */
	close(sockfd);
	exit(0);
}

tcpclient.c（进程和线程的客户端基本一样）

#include  
#include  
#include  
#include  
#include  
#include  
#include  
#include  

#define portnumber 3333

void * read_msg(void *arg)
{
    int fd = *((int *)arg);
	int nread = 0;
	char buffer[1024];

	while((nread = read(fd,buffer,sizeof(buffer))) > 0)
	{
	    buffer[nread] = '\0';
		printf("get server message: %s\n",buffer);
		memset(buffer,0,sizeof(buffer));
		sleep(2);
	}
}

int main(int argc, char *argv[]) 
{ 
	int sockfd; 
	char buffer[1024]; 
	struct sockaddr_in server_addr; 
	struct hostent *host; 

        /* 使用hostname查询host 名字 */
	if(argc!=2) 
	{ 
		fprintf(stderr,"Usage:%s hostname \a\n",argv[0]); 
		exit(1); 
	} 

	if((host=gethostbyname(argv[1]))==NULL) 
	{ 
		fprintf(stderr,"Gethostname error\n"); 
		exit(1); 
	} 

	/* 客户程序开始建立 sockfd描述符 */ 
	if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:Internet;SOCK_STREAM:TCP
	{ 
		fprintf(stderr,"Socket Error:%s\a\n",strerror(errno)); 
		exit(1); 
	} 

	/* 客户程序填充服务端的资料 */ 
	bzero(&server_addr,sizeof(server_addr)); // 初始化,置0
	server_addr.sin_family=AF_INET;          // IPV4
	server_addr.sin_port=htons(portnumber);  // (将本机器上的short数据转化为网络上的short数据)端口号
	server_addr.sin_addr=*((struct in_addr *)host->h_addr); // IP地址
	
	/* 客户程序发起连接请求 */ 
	if(connect(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1) 
	{ 
		fprintf(stderr,"Connect Error:%s\a\n",strerror(errno)); 
		exit(1); 
	} 
    
	pthread_t id;

	pthread_create(&id,NULL,(void *)read_msg,(int *)&sockfd);
	/* 连接成功了 */ 
	while(1)
	{
	    printf("Please input char:\n");
	
	    /* 发送数据 */
	    scanf("%s",buffer); 
	    write(sockfd,buffer,strlen(buffer)); 
	}

	/* 结束通讯 */ 
	close(sockfd); 
	exit(0); 
}