linux的tcp非阻塞客户端与服务端demo源码

客户端

#include <stdio.h>  
#include <string.h>  
#include <errno.h>  
#include <sys/socket.h>  
#include <resolv.h>  
#include <stdlib.h>  
#include <netinet/in.h>  
#include <arpa/inet.h>  
#include <unistd.h>  
#include <sys/time.h>  
#include <sys/types.h>  
  
#define MAXBUF 1024  
#define PORT 8888
int main(int argc, char **argv)  
{  
        int sockfd, len;  
        struct sockaddr_in dest;  
        char buf[MAXBUF + 1];  
        fd_set rfds,wfds;  
        struct timeval tv;  
		int retval, maxfd = -1;  
		char sendbuf[128]="helloworld";
		if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {  
            perror("Socket");  
            exit(errno);  
        }  
		char *seraddr="192.168.0.224";
		bzero(&dest, sizeof(dest));  
        dest.sin_family = AF_INET;  
        dest.sin_port = htons(PORT);  
        if (inet_aton(seraddr, (struct in_addr *) &dest.sin_addr.s_addr) == 0) {  
            perror(seraddr);  
            exit(errno);  
        }  
  
		if(connect(sockfd, (struct sockaddr *) &dest, sizeof(dest)) != 0) {  
            perror("Connect ");  
            exit(errno);  
        }  
  
		printf("connect to server...\n");         
		while (1)   
		{  
  
                FD_ZERO(&rfds);             
                FD_SET(sockfd, &rfds);  
				FD_ZERO(&wfds);             
                FD_SET(sockfd, &wfds); 
                maxfd = 0;  
              
                //FD_SET(sockfd, &rfds);  
                if (sockfd > maxfd)  
                    maxfd = sockfd;  
              
                tv.tv_sec = 1;  
                tv.tv_usec = 0;  
            
                    retval = select(maxfd + 1, &rfds, &wfds, NULL, &tv);  
  
				if (retval == -1)   
				{  
                        printf("select error! %s", strerror(errno));                
						break;  
                } 
				else if (retval == 0) 
				{   
                        continue;  
                } 
				else 
				{  
					if (FD_ISSET(sockfd, &wfds))   
					{                  
                            //bzero(buf, MAXBUF + 1);  
                            len = send(sockfd, sendbuf, strlen(sendbuf), 0);  
                            if (len > 0)  
                                printf("msg:%s send successful，totalbytes: %d！\n", sendbuf, len);  
                            else {  
                                printf("msg:'%s  failed!\n", sendbuf);  
                                break;  
                            }  
							
                    }  
                    if(FD_ISSET(sockfd, &rfds))   
					{   
                                bzero(buf, MAXBUF + 1);  
                                len = recv(sockfd, buf, MAXBUF, 0);  
                                if (len > 0)  
                                    printf("recv:'%s, total: %d \n", buf, len);  
                                else    
								{  
									if (len < 0)   
                                            printf("recv failed！errno:%d，error msg: '%s'\n", errno, strerror(errno));  
                                    else  
                                            printf("other exit，terminal chat\n");  
                                    break;  
								}  
                    }                 
            } 
			sleep(1);
    }  
  
    close(sockfd);  
    return 0;  
}  



服务端
#include <stdio.h>  
#include <stdlib.h>  
#include <errno.h>  
#include <string.h>  
#include <unistd.h>  
#include <arpa/inet.h>  
#include <netinet/in.h>  
#include <sys/types.h>  
#include <sys/socket.h>  
#include <sys/wait.h>  
#include <sys/time.h>  
#include <sys/types.h>  
  
#define MAXBUF 1024  
  
typedef struct CLIENT {  
    int fd;  
    struct sockaddr_in addr;      
}CLIENT;  

/*************************** 
**server for multi-client  
**PF_SETSIZE=1024 
****************************/  
int main(int argc, char** argv)  
{  
    int i,n,maxi = -1;  
	int j=0;
    int nready;  
    int slisten,sockfd,maxfd=-1,connectfd;
	int maxifd=-1;
    int sendn=-1;
	char sendbuf[100]="helloworld";
    unsigned int myport,lisnum;   
    CLIENT client[FD_SETSIZE];  
    struct sockaddr_in  my_addr,addr;  
    struct timeval tv;  
      
    socklen_t len;  
    fd_set rset,allset,wset;   
      
    char buf[MAXBUF + 1];  
     
  
    myport = 8883;  
  
    lisnum = 5;  
  
    if((slisten = socket(AF_INET,SOCK_STREAM,0)) == -1)  
    {  
        perror("socket");  
        exit(1);  
    }  
  
    bzero(&my_addr,sizeof(my_addr));  
    my_addr.sin_family = AF_INET;  
    my_addr.sin_port = htons(myport);  
    my_addr.sin_addr.s_addr = INADDR_ANY;  
	char snd_buf[128]="helloworld";
    if(bind(slisten, (struct sockaddr *)&my_addr, sizeof(my_addr)) == -1) {  
            perror("bind");  
            exit(1);  
        }  
  
    if (listen(slisten, lisnum) == -1) {  
            perror("listen");  
            exit(1);  
        }  
  
    for(i=0;i<5;i++)  
    {  
        client[i].fd = -1;  
    }  
 
	printf("Waiting for connections and data...\n");
    while (1)  
    {         
        tv.tv_sec = 1;  
        tv.tv_usec = 0;
		FD_ZERO(&allset);             
		FD_SET(slisten, &allset);    
		maxfd = slisten;  		
		if(select(maxfd+1,&allset,NULL,NULL,&tv)>0)
		{
			len = sizeof(struct sockaddr);  
	
			if((connectfd = accept(slisten,(struct sockaddr*)&addr,&len)) == -1)  
			{  
				perror("accept() error\n");  
			}
			for(i=0;i<5;i++)
			{
				if(client[i].fd<0)
				{
					client[i].fd=connectfd;
					break;
				}
			}
			if(i==5)
			{
				printf("too many connect\n");
			}
		}
		for(j=0;j<5;j++)
		{
			if(client[j].fd<0)
				continue;
			FD_ZERO(&rset);
			FD_SET(client[j].fd,&rset);
			FD_ZERO(&wset);
			FD_SET(client[j].fd,&wset);	
			maxifd = client[j].fd;				
			if(select(maxifd + 1, &rset,&wset, NULL, &tv)>0)
			{
				if(FD_ISSET(client[j].fd,&rset))  
				{                         
					bzero(buf,MAXBUF + 1);  
					if((n = recv(client[j].fd,buf,MAXBUF,0)) > 0)  
					{  
						printf("received data:%s \n",buf);    
					}                 
				}
				if(FD_ISSET(client[j].fd,&wset))
				{
					n = send(client[j].fd,snd_buf,128,0) > 0 ;
				}
			}
		}			
		sleep(1);
    }  
    close(slisten);  
}