回声客户端的实现

客户端向服务器发送任何字符，服务器接收到后原样发回去：

SERVER:

 

Server：
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>

#define BUF_SIZE 1024

void error_handing(char *message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}


int main(int argc,char *argv[])
{
	int sock_c,sock_s;
	char message[BUF_SIZE];
	int str_len,i;

	struct sockaddr_in serv_adr,clnt_adr;
	socklen_t clnt_adr_sz;

	sock_s = socket(PF_INET,SOCK_STREAM,0);
	if(sock_s == -1)
		error_handing("socket() error!\n");

	memset(&serv_adr,0,sizeof(serv_adr));
	serv_adr.sin_family = AF_INET;
	serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
	serv_adr.sin_port = htons(atoi("8888"));

	if(bind(sock_s,(struct sockaddr*)&serv_adr,sizeof(serv_adr))==-1)
		error_handing("bind() error!\n");

	if(listen(sock_s,5) == -1)
		error_handing("listen() error!\n");

	clnt_adr_sz = sizeof(clnt_adr);

	for(i=0;i<5;i++)
	{
		sock_c = accept(sock_s,(struct sockaddr*)&clnt_adr,&clnt_adr_sz);
		if(sock_c == -1)
			error_handing("accept() error!\n");
		else
			printf("connect client: %d!\n",i+1);

		while(str_len = read(sock_c,message,BUF_SIZE))//此时如果客户端没有数据发来，则read阻塞
		{
			printf("connect: %d str_len: %d \n",i+1,str_len);
			write(sock_c,message,str_len);
		}
		close(sock_c);
		printf("disconnect:%d \n",i+1);
	}
	close(sock_s);
	return 0;
}

Client：

 

 

Client：
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 1024

void error_handing(char *message)
{
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}

int main(int argc,char *argv[])
{
	int sock;
	struct sockaddr_in serv_addr;
	char message[BUF_SIZE];
	int str_len;
	
	sock = socket(PF_INET,SOCK_STREAM,0);
	if(sock == -1)
		error_handing("socket() error!\n");

	memset(&serv_addr,0,sizeof(serv_addr));
		serv_addr.sin_family = AF_INET;
	serv_addr.sin_addr.s_addr = inet_addr("192.168.1.172");
	serv_addr.sin_port = htons(atoi("8888"));
	
	if(connect(sock,(struct sockaddr*)&serv_addr,sizeof(serv_addr)) == -1)
		error_handing("connect() error!\n");
	else
		puts("connect.......\n");
	
	while(1)
	{
		fputs("Input message(Q to quit):",stdout);
		fgets(message,BUF_SIZE,stdin);

		if(!strcmp(message,"q\n") || !strcmp(message,"Q\n"))
			break;
		write(sock,message,strlen(message));
		str_len = read(sock,message,BUF_SIZE-1);
		message[str_len] = 0;
		printf("Message from server:%s ",message);
	}
	close(sock);
	return 0;
}

 

 

由于TCP协议不存在数据边界，因此字符串可能被分多次传输，例如字符串太长的情况下。因此每个回合一次性收发可能会发生错误。
因此改进一下，即当接受到的字节数小于发送时则循环接受：
将while中的内容改成：

 

 

 

 

 

 

	while(1)
	{
		fputs("Input message(Q to quit):",stdout);
		fgets(message,BUF_SIZE,stdin);

		if(!strcmp(message,"q\n") || !strcmp(message,"Q\n"))
			break;
		
		str_len = write(sock,message,strlen(message));
		recv_len = 0;
		while(recv_len < str_len)
		{
			recv_cnt = read(sock,&message[recv_len],BUF_SIZE-1); //message[]从上次的通知的地方继续接受
			if(recv_cnt == -1)
				error_handing("read() error!");
			recv_len += recv_cnt;
		}
		message[str_len] = 0;
		printf("Message from server:%s ",message);
	}