linux——网络编程优化

1、服务器、客户端运行能自己输入配置ip地址和端口号。当有客户端接入，服务器不退出

服务器代码：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
//#include <linux/in.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc,char **argv)
{
	int s_fd;
	char readbuf[128];
	int nread;

	char *msg="i get your message\n";
	struct sockaddr_in s_addr;
	struct sockaddr_in c_addr;

	
	memset(&s_addr,0,sizeof(struct sockaddr_in));
	memset(&c_addr,0,sizeof(struct sockaddr_in));

	//1.socket
	s_fd=socket(AF_INET,SOCK_STREAM,0);
	if(s_fd==-1){
		perror("socket");
		exit(-1);
	}
	//2.bind
	s_addr.sin_family=AF_INET;
	s_addr.sin_port=htons(atoi(argv[2]));
	inet_aton(argv[1],&s_addr.sin_addr);

	bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr_in));
	//3.listen
	listen(s_fd,10);
	//4.accept
	int clen=sizeof(struct sockaddr_in);
	while(1){
		int c_fd=accept(s_fd,(struct sockaddr*)&c_addr,&clen);
		if(c_fd==-1){
			perror("accept");
		}
		printf("connect\n");
		printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));
		if(fork()==0){
	//5.read
			nread=read(c_fd,readbuf,128);
			if(nread==-1){
				perror("read");
			}else{
				printf("get message:%d,%s\n",nread,readbuf);
			}
	//6.write
			write(c_fd,msg,strlen(msg));
			break;
		}	
	}
	return 0;
}

客户端代码：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
//#include <linux/in.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc, char **argv)
{
	int c_fd;
	char readbuf[128];
	int nread;

	char *msg="msg from client\n";
	struct sockaddr_in c_addr;

	
	memset(&c_addr,0,sizeof(struct sockaddr_in));

	//1.socket
	c_fd=socket(AF_INET,SOCK_STREAM,0);
	if(c_fd==-1){
		perror("socket");
		exit(-1);
	}
	//2.connect
	c_addr.sin_family=AF_INET;
	c_addr.sin_port=htons(atoi(argv[2]));
	inet_aton(argv[1],&c_addr.sin_addr);

	if(connect(c_fd,(struct sockaddr*)&c_addr,sizeof(struct sockaddr))==-1){
		perror("connect");
		exit(-1);
	}
	//3.write
	write(c_fd,msg,strlen(msg));
	
	//4.read
	nread=read(c_fd,readbuf,128);
	if(nread==-1){
		perror("read");
	}else{
		printf("get message from sever:%d,%s\n",nread,readbuf);
	}

	return 0;
}

编译：

        gcc sever.c -o sever

        gcc client.c -o client

运行：./sever ip地址 端口号

        ./client ip地址 端口号，windows充当客户端需要telnet ip地址 客户端实现通信

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.5
get message:16,msg from client

connect
get connect:192.168.1.6
get message:1,s
connect
get connect:192.168.1.6
get message:0,

2、代码优化二：修改代码，实现输入字符多机通信

服务器代码；

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
//#include <linux/in.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc,char **argv)
{
	int s_fd;
	char readbuf[128];
	int nread;

	char msg[128]={0};
	struct sockaddr_in s_addr;
	struct sockaddr_in c_addr;
	if(argc!=3){
		printf("param is not good\n");
		exit(-1);
	}
	
	memset(&s_addr,0,sizeof(struct sockaddr_in));
	memset(&c_addr,0,sizeof(struct sockaddr_in));

	//1.socket
	s_fd=socket(AF_INET,SOCK_STREAM,0);
	if(s_fd==-1){
		perror("socket");
		exit(-1);
	}
	//2.bind
	s_addr.sin_family=AF_INET;
	s_addr.sin_port=htons(atoi(argv[2]));
	inet_aton(argv[1],&s_addr.sin_addr);

	bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr_in));
	//3.listen
	listen(s_fd,10);
	//4.accept
	int clen=sizeof(struct sockaddr_in);
	while(1){
		int c_fd=accept(s_fd,(struct sockaddr*)&c_addr,&clen);
		if(c_fd==-1){
			perror("accept");
		}
		printf("connect\n");
		printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));
		if(fork()==0){
			if(fork()==0){
				while(1){
					memset(msg,0,sizeof(msg));
					printf("input: ");
					gets(msg);
					//6.write
					write(c_fd,msg,strlen(msg));
				}
			}
			while(1){
				//5.read
				memset(readbuf,0,sizeof(readbuf));
				nread=read(c_fd,readbuf,128);
				if(nread==-1){
					perror("read");
				}else{
					printf("get message:%d,%s\n",nread,readbuf);
				}
			}
			break;
		}	
	}
	return 0;
}

客户端代码；

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
//#include <linux/in.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdlib.h>
#include <string.h>

int main(int argc, char **argv)
{
	int c_fd;
	char readbuf[128];
	int nread;

	char msg[128]={0};
	struct sockaddr_in c_addr;
	if(argc!=3){
		printf("param is not good\n");
		exit(-1);
	}
	
	memset(&c_addr,0,sizeof(struct sockaddr_in));

	//1.socket
	c_fd=socket(AF_INET,SOCK_STREAM,0);
	if(c_fd==-1){
		perror("socket");
		exit(-1);
	}
	//2.connect
	c_addr.sin_family=AF_INET;
	c_addr.sin_port=htons(atoi(argv[2]));
	inet_aton(argv[1],&c_addr.sin_addr);

	if(connect(c_fd,(struct sockaddr*)&c_addr,sizeof(struct sockaddr))==-1){
		perror("connect");
		exit(-1);
	}
	while(1){
		if(fork()==0){
			while(1){
				memset(msg,0,sizeof(msg));
				printf("input: ");
				gets(msg);
				//3.write
				write(c_fd,msg,strlen(msg));
			}
		}
		while(1){
			//4.read
			memset(readbuf,0,sizeof(readbuf));
			nread=read(c_fd,readbuf,128);
			if(nread==-1){
				perror("read");
			}else{
				printf("get message from sever:%d,%s\n",nread,readbuf);
			}
		}
	}

	return 0;
}

编译运行：可以实现服务器与客户端之间的收发消息

有一个bug，如果服务器连接多个客户端，发消息会被这几个客户端其中一个收到（随机）