linux下的TCP小程序

迷南。 2024-04-18 21:24 29阅读 0赞

服务端是用多线程写的并发服务器，比较简单。

功能：

1．  服务端运行后，即监听指定的端口。允许多个连接并发运行

2．  服务端接收到客户端请求后，根据客户端传过来的指令完成特定任务：

a)         向客户端传送服务端所在机器的当前时间

b)        向客户端传送服务端所在机器的名称

c)         向客户端传送当前连接的所有客户端信息

d)        其它（向客户端传送键盘输入的内容）

3．  客户端运行后，可以执行以下任务：

a)         连接到指定地址和端口的服务端

b)        断开与服务端的连接

c)         请求服务端给出当前时间

d)        请求服务端给出其机器的名称

e)         请求服务端给出当前连接的所有客户端信息（编号、IP地址、端口等）

f)         给其他客户端发送键盘输入的内容

g)        退出客户端程序

h)        其它（向服务器端传送键盘输入的内容）

客户端：

#include <stdio.h>
    #include <sys/types.h>
    #include <sys/socket.h>
    #include <sys/un.h>
    #include <netdb.h>
    #include <unistd.h>
    #include <time.h>
    
    int main(int argc,char *argv[])
    {
        int connect_fd;
    	int ret;
    	char snd_buf[1024];
    	int i;
    	int port;
    	int len,lens;
    
    	static struct sockaddr_in srv_addr;
    
    	//判断客户端运行需要给出具体的连接地址和端口 
    	if(argc!=3)
    	{
    	    printf("Usage: %s server_ip_address port\n",argv[0]);
    		return 1;
    	}
    
    	//获得输入的端口
    	port=atoi(argv[2]);
    
    	//创建套节字用于客户端的连接
    	connect_fd=socket(PF_INET,SOCK_STREAM,0);
    	if(connect_fd<0)
    	{
    	    perror("cannot create communication socket");
    		return 1;
    	}
    
    	//填充关于服务器的套节字信息
    	memset(&srv_addr,0,sizeof(srv_addr));
    	srv_addr.sin_family=AF_INET;
    	srv_addr.sin_addr.s_addr=inet_addr(argv[1]);
    	srv_addr.sin_port=htons(port);
    
    	//连接服务器，如果连接失败 ret==-1，结束程序 
    	ret=connect(connect_fd,(struct sockaddr *)&srv_addr,sizeof(srv_addr));
    	if(ret==-1)
    	{
    	    perror("cannot connect to the server");
    		close(connect_fd);
    		return 1;
    	}
    
    	memset(snd_buf,0,1024);
    	//成功连接服务器 
    	while(1)
    	{
    	/*	time_t timeval;
    		(void)time(&timeval);*/ 
    		lens=strlen(ctime(&timeval));
    
    	    	printf("Please input char:"); 
    		fgets(snd_buf,1024,stdin); //键盘输入内容 
    		len = strlen(snd_buf);
    		if(len>0)
    		{
    			write(connect_fd,snd_buf,len);//发送信息到服务器 
    		}
    		len=read(connect_fd,snd_buf,1024);//如果服务器发来信息，则读取，否则等待。。 
    		if(len>0)
    			printf("Message form server(%d): %s\n",len,snd_buf);//打印从服务器收到的信息 
    		if(snd_buf[0]=='@')//@为退出字符。 跳出循环 
    			break;
    	}
    	close(connect_fd);//断开与服务器的连接。 
    	return 0;
    }

服务端：

#include <pthread.h>
    #include <stdio.h>
    #include <sys/types.h>
    #include <sys/socket.h>
    #include <sys/utsname.h>
    #include <sys/un.h>
    #include <errno.h>
    #include <unistd.h>
    #include <signal.h>
    #include <sys/wait.h>
    #include <netdb.h>
    #include <time.h>
    #include <stdlib.h>
    
    
    struct sockaddr_in clt_addr;
    struct sockaddr_in srv_addr;
    
    //线程执行函数负责读写
    void *thr_fn(void *arg)
    {
    	int size,j,lens;
    	int *parg=(int *)arg;
    	int new_fd=*parg;
    	char recv_buf[1024];
    	char hostdate[1024];
    	char snd_buf[1024];
    
    	fprintf(stderr,"Server get connection from %s\n",(char* )inet_ntoa(clt_addr.sin_addr));//显示连接的客户端ip，并打印出来 
    	printf("new_fd=%d\n",new_fd);//客户端连接编号 
    	memset(snd_buf,0,1024);
    
    	time_t timeval;
    	(void)time(&timeval);
    
    	while((size=read(new_fd,recv_buf,1024))>0)//判断是否有客户端发送信息过来 
    	   {
    			if(recv_buf[0]== '@')//exit
    				break;
    			if(recv_buf[0]== '!')//time
    			{
    				printf("Message from client(%d): %s\n",new_fd,recv_buf);
    				lens=strlen(ctime(&timeval));
    				write(new_fd,ctime(&timeval),lens);	
    			}
    			else if(recv_buf[0]== '#')//host message
    			{
    				char computer[256];
    				struct utsname uts;
    				if(gethostname(computer,255)!=0||uname(&uts)< 0)
    				{
    				fprintf(stderr, "Coule not get host information\n");
    				exit(1);
    				}
    				lens=strlen(uts.nodename);
    				char* hostdate=uts.nodename;
    				printf("Message from client(%d): %s\n",new_fd,recv_buf);
    				write(new_fd,hostdate,lens);			
    			}
    			else if(recv_buf[0]== '$')  
    			{	
    				
    			}
    			else
    			{
    			printf("Message from client(%d): %s\n",new_fd,recv_buf);
    			}
    			//服务器向客户端发信息 
    		printf("Please input char:");
    		fgets(snd_buf,1024,stdin); 
    		write(new_fd,snd_buf,strlen(snd_buf)); 
    		printf("\n");
    	}
    	close(new_fd);
    	return 0;
    }
    
    
    int main(int argc,char *argv[])
    {
        socklen_t clt_addr_len;
    	int listen_fd;
    	int com_fd;
    	int ret;
    	int i;
    	static char recv_buf[1024];
    	static char rev_buf[1024];
    	int len;
    	int port;
    	pthread_t tid;
    
    	//判断客户端是否填写端口 
    	if(argc!=2)   
    	{
    	    printf("Usage:%s port\n",argv[0]);
    		return 1;
    	}
    
    	//端口的字符转换 
    	port=atoi(argv[1]);
    
    	//创建套接字用于服务器的监听
    	listen_fd=socket(PF_INET,SOCK_STREAM,0);
    	if(listen_fd<0)
    	{
    	    perror("cannot create listening socket");
    		return 1;
    	}
    
    
    	memset(&srv_addr,0,sizeof(srv_addr));
    	//填充服务器信息 
    	srv_addr.sin_family=AF_INET;
    	srv_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    	srv_addr.sin_port=htons(port);
    
    
    	//将服务器和套节字绑定
    	ret=bind(listen_fd,(struct sockaddr *)&srv_addr,sizeof(srv_addr));
    	if(ret==-1)
    	{
    	    perror("cannot bind server socket");
    		close(listen_fd);
    		return 1;
    	}
    
    	//设置连接5个客户端
    	ret=listen(listen_fd,5);
    	if(ret==-1)
    	{
    	    perror("cannot listen the client connect request");
    		close(listen_fd);
    		return 1;
    	}
    
    	//对每个连接来的客户端创建一个线程，单独与其进行通信 ，首先调用read函数读取客户端发送来的信息 
    	while(1)
    	{
    	    len=sizeof(clt_addr);
    		com_fd=accept(listen_fd,(struct sockaddr *)&clt_addr,&len);
    		if(com_fd<0)
    		{
    		    if(errno==EINTR)
    			{
    			    continue;
    			}
    			else
    			{
    			    perror("cannot accept client connect request");
    				close(listen_fd);
    				return 1;
    			}
    		}
    		if((pthread_create(&tid,NULL,thr_fn,&com_fd))==-1)//多线程调用 ，跳到thr_fn函数 
    		{
    		    perror("pthread_create error");
    			close(listen_fd);
    			close(com_fd);
    			return 1;
    		}
    	}
    	return 0;
    }