linux Tcp 套接字

1、服务器代码

#include<stdio.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<string.h>
int startup(const char *ip,int port)
{
	int sock = socket(AF_INET,SOCK_STREAM,0);
	if(sock < 0)
	{
		perror("socket");
		exit(2);
	}
	//bind
	struct sockaddr_in local;
	local.sin_family = AF_INET;
	local.sin_port = htons(port);
	local.sin_addr.s_addr = inet_addr(ip);

	if(bind(sock,(struct sockaddr*)&local,sizeof(local)) < 0)
	{
		perror("bind");
		exit(3);
	}

	if(listen(sock,10) < 0)
	{
		perror("listen");
		exit(4);
	}
	return sock;
}
static void usage(const char *proc)
{
	printf("%s[local ip] [local port]\n",proc);
}

int main(int argc,char *argv[])
{
	if(argc != 3){
		usage(argv[0]);
		return 1;
	}
	//jianting
	
	int listen_sock = startup(argv[1],atoi(argv[2]));
	struct sockaddr_in client;
	socklen_t len = sizeof(client);
	char buf[1024];
	while(1){
		int sock = accept(listen_sock,(struct sockaddr*)&client,&len);
		if(sock < 0){
			perror("accept");
			continue;
		}
		printf("client ip: %s,port: %d\n",inet_ntoa(client.sin_addr),ntohs(client.sin_port));
		while(1){
			ssize_t s = read(sock,buf,sizeof((buf)-1));
			if(s > 0){
				buf[s] = 0;
				printf("client say# %s\n",buf);
				write(sock,buf,strlen(buf));
			}else if(s == 0){
				printf("client quit!\n");
				break;
			}
		}
	}
	return 0;
}		

2、客户端代码

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

static void usage(const char *proc)
{
	printf("%s[server ip] [server port]\n",proc);
}

int main(int argc,char* argv[])
{

	if(argc != 3)
	{
		usage(argv[0]);
		return 1;
	}
	int sock = socket(AF_INET,SOCK_STREAM,0);
	if(sock < 0){
		perror("socket");
		return 2;
	}

	struct sockaddr_in peer;
	peer.sin_family = AF_INET;
	peer.sin_port = htons(atoi(argv[2]));
	peer.sin_addr.s_addr = inet_addr(argv[1]);

	if(connect(sock,(struct sockaddr*)&peer,sizeof(peer)) < 0){
		perror("connect");
		return 3;
	}

	char buf[1024];
	while(1)
	{
		printf("please send#");
		fflush(stdout);
		ssize_t s = read(0,buf,sizeof(buf)-1);
		if(s > 0){
			buf[s-1] = 0;
			write(sock,buf,strlen(buf));
			ssize_t c = read(sock,buf,sizeof(buf)-1);
			if(c > 0){
				buf[c] = 0;
				printf("server echo# %s\n",buf);
			}
		}
	}
	close(sock);
	return 0;
}

3、测试结果