首页 > 代码库 > 【UNIX网络编程(四)】TCP套接字编程详细分析

【UNIX网络编程(四)】TCP套接字编程详细分析

引言:

套接字编程其实跟进程间通信有一定的相似性,可能也正因为此,stevens这位大神才会将套接字编程与进程间的通信都归为“网络编程”,并分别写成了两本书《UNP1》《UNP2》。TCP套接字编程是套接字编程中非常重要的一种,仔细分析,其实它的原理并不复杂。现在就以一个例子来详细分析TCP套接字编程。


一、示例要求:

本节中试着编写一个完成的TCP客户/服务器程序示例,并对它进行深入的探讨。该示例会用到绝大多数的基本函数,未用到但比较重要的函数会在后面的补充上。示例要完成的内容是实现一个客户/服务器程序:

1、客户从标准输入读入一行文本,并写给服务器;

2、服务器从网络输入读入这行文本,并回射给客户。

3、客户从网络输入读入这行回射文本,并显示在标准输出上。

用图表示给该客户/服务器示例就是:


大致分析该客户/服务器程序:首先,客户程序通过fgets函数从标准输入得到文本,然后客户程序调用write函数发给服务器。服务器通过read函数读取客户程序发过来的文本,然后服务器调用write函数将“从客户端读取的文本”重新发给客户端。客户端调用read函数读取服务器重新发给客户端的文本,之后客户程序调用fputs函数将读取的文本重新发送到标准输出。

从上面的分析可以看出,TCP客户程序与TCP服务器程序是两个独立的程序,它们可以组合使用从而形成一个“TCP回射客户/服务器程序”。下面我们首先分别写好客户端程序、服务器端程序,并用Socket调试工具,进行调试。调试结束后,组合这两个程序,形成一个客户/服务器程序。


二、示例分析:

在写客户/服务器程序之前,我们先用图来展示一下各个阶段是如何工作的,以及客户、服务器是如何建立连接并传递数据的。


从上到下按照顺序分析客户端应用程序以及服务器段应用程序如下:

1、服务器端调用socket()函数创建一个套接字,由socket()函数创建的套接字都是主动套接字,也就是说,即使我们的初衷是在服务器端创建的套接字,但此时内核并不知道我们的意图,此时的套接字仅仅是一个要调用connect()函数发起连接的客户套接字,那么该如何使它变成一个被动套接字呢?也就是说如何告诉内核应该接受指向该套接字的连接请求呢?必须要经过后面的bind()函数和listen()函数。

2、调用完socket()函数后,TCP此时的状态是CLOSED状态,接下来我们调用bind()函数,该函数完成对ip地址和端口号的绑定。注意bind函数一般是由服务器调用的,客户端一般不会调用该函数绑定地址,因为对于端口号,内核会自动的帮助客户端确定它的端口号为一个临时端口号;对于IP地址,内核将根据所用外出网络接口来选择源IP地址。

3、服务器端此时调用listen()函数,它的任务就是将上面步骤1中socket()函数创建的主动套接字转换为一个被动套接字,指示内核应该接受指定该套接字的连接请求,同时规定了内核应该为相应套接字排队的最大连接数。listen()函数调用完成之后,TCP状态由CLOSED状态转换到LISTEN状态。

4、服务器端调用accept()函数,它用于从已完成连接队列对头返回下一个已完成连接。如果已完成连接队列为空,那么进程休眠。注意,accept()函数的第二三个参数分别用来返回已连接的对端进程(客户)的协议地址和套接字地址结构的长度。如果成功,它的返回值是由内核自动生成的一个全兴描述符,代表与所返回客户的TCP连接。accept()函数的第一个参数就是socket函数创建的套接字,我们称之为监听套接字返回值为已连接套接字描述符

5、因为accept()函数会使服务器进程阻塞,等待客户端的连接。客户端用socket()函数创建一个套接字后,接着客户进程调用connect()函数建立与TCP服务器的连接(注意,客户端没用bind、listen函数)。对于TCP套接字,connect函数会激发TCP的三路握手过程,即,客户端发SYN给服务器,服务器接到后回发SYN和ACK给客户端,客户端接受到服务器的SYN后,回发ACK给服务器端。三路握手完成后,连接建立。此时的客户端与服务器端都处于ESTABLISHED状态。连接建立后,我们就可以通过read和write函数在客户和服务器端进行文本的传送了。

6、当文本发送完毕后,客户给服务器端发送一个FIN结束标志,服务器端接到FIN后给客户端回发一个FIN结束标志,然后客户接到服务器的FIN,在给服务器发送一个确认标志ACK。之后,连接断开。两端的TCP状态都处于CLOSED状态。


三、示例代码及讲解

首先给出客户端的程序并用TCP/UDP Socket调试工具进行调试,并给出结果:

代码:

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

#define MAXLINE 2048

int
main(int argc, char **argv)
{
	int sockfd;
	char sendline[MAXLINE], recvline[MAXLINE];
	struct sockaddr_in	cliaddr;

	sockfd = socket(AF_INET, SOCK_STREAM, 0);/*创建套接字*/
	if(sockfd < 0){
		printf("socket function fail.\n");	
		return -1;
	}

	bzero(&cliaddr, sizeof(cliaddr));
	cliaddr.sin_family = AF_INET;
	cliaddr.sin_port = htons(9877);
	inet_pton(AF_INET, argv[1], &cliaddr.sin_addr);/*对套接字地址结构进行操作*/

	connect(sockfd, (struct sockaddr *)&cliaddr, sizeof(cliaddr));/*建立连接*/

	while(fgets(sendline, MAXLINE, stdin) != NULL){
		write(sockfd, sendline, strlen(sendline));	/*利用套接字进行数据发送*/
	}
}
编译该程序:gcc client.c -o client

用调试工具进行调试,用该工具建立TCP服务器,端口号设为9877,然后执行上面生成的可执行程序./client 59.73.166.231(该IP地址,是我电脑的IP地址,Linux的IP地址是59.73.166.32,它会出现在调试工具的客户端地址部分)。并在客户端发送数据,abc,然后就会在调试工具里显示该文本abc,接着发送def,同样会在调试工具里出现def文本。效果如下图所示:


由上面可以看出客户端的端口号是53633,它是由内核自动分配的。

下面给出服务器端的程序如下:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <netinet/in.h>
#include <sys/socket.h>

char c[2048];

int
main(int argc, char **argv)
{
	int 	n;
	int		sockfd, coundfd;
	struct sockaddr_in serv, client;
	socklen_t socklen;

	/*creat sockfd using the socket function*/
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sockfd == -1){
		printf("socket function fail.\n");	
		return -1;
	}

	/*bind the ip addr and port using the bind function*/
	bzero(&serv, sizeof(serv));
	serv.sin_family 	  = AF_INET;
	serv.sin_addr.s_addr  = htonl(INADDR_ANY);
	serv.sin_port		  = 9877;
	if(bind(sockfd, (struct sockaddr *)&serv, sizeof(struct sockaddr_in))){
		printf("bind function fail.\n");	
		return -1;
	}

	/*listen function*/
	if(listen(sockfd, 5)){
		printf("listen function fail.\n");	
		return -1;
	}

	/*accept function*/
	socklen = sizeof(struct sockaddr_in);
	printf("accept wait \n");
	coundfd = accept(sockfd, (struct sockaddr *)&client, &socklen);
	printf("accept wait \n");

	while(1){
		if((n = read(coundfd, c, 2048) > 0)){
			write(coundfd, c, n);	
		}
	}

}
跟上面的客户端程序一样,也可以进行同样的服务器端程序的测试,此处略去了。


四、完成的TCP回射客户/服务器程序

由于上面的程序是进行的测试,客户端并没有回射部分,下面的代码给出回射部分。

客户程序:

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

#define MAXLINE 2048
#define SERV_PORT 9877

void str_cli(FILE *, int);

int
main(int argc, char **argv)
{
	int 	sockfd;
	struct  sockaddr_in cliaddr;

	if(argc != 2){
		printf("usage:tcpcli <IPaddress>");	
		return -1;
	}

	sockfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(sockfd == -1){
		printf("socket fail.\n");	
		return -1;
	}

	bzero(&cliaddr, sizeof(cliaddr));
	cliaddr.sin_family = AF_INET;
	cliaddr.sin_port   = htons(SERV_PORT);
	if(!(inet_pton(AF_INET, argv[1], &cliaddr.sin_addr))){
		printf("inet pton fail.\n");	
		return -1;
	}

	if(connect(sockfd, (struct sockaddr *)&cliaddr, sizeof(cliaddr))){
		printf("connect fail.\n");	
		return -1;
	}

	str_cli(stdin, sockfd);

	exit(0);

}

void
str_cli(FILE *fp, int sockfd)
{
	char 	sendline[MAXLINE], recvline[MAXLINE];

	while(fgets(sendline, MAXLINE, fp) != NULL){
		write(sockfd, sendline, strlen(sendline));
		
		if(read(sockfd, recvline, MAXLINE) == 0)	
			printf("str_cli:server terminated prematurely");

		fputs(recvline, stdout);
	}
}
服务器程序:

#include <stdio.h>
#include <errno.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

#define SERV_PORT	9876
#define MAXLINE		2048

void
str_echo(int sockfd)
{
	ssize_t	n;
	char	buf[MAXLINE];

again:
	while((n = read(sockfd, buf, MAXLINE)) > 0)
		write(sockfd, buf, n);

	if(n < 0 && errno == EINTR)
		goto again;
	else if(n < 0)
		printf("str_echo : read error");
}

int
main(int argc, char **argv)
{
	int	listenfd, connfd;
	pid_t	childpid;
	socklen_t	clilen;
	struct	sockaddr_in	cliaddr, servaddr;

	listenfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if(listenfd == -1){
		printf("socket fail.\n");	
		return -1;
	}

	bzero(&servaddr, sizeof(struct sockaddr_in));
	servaddr.sin_family 	 = AF_INET;
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	servaddr.sin_port 		 = htons(SERV_PORT);

	if(bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr))){
		printf("bind fail.\n");	
		return -1;
	}
	if(listen(listenfd, 5)){
		printf("listen fail.\n");	
		return -1;
	}

	printf("listen stat.\n");

	for(;;){
		clilen = sizeof(cliaddr);				
		connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);
		if(connfd == -1){
			printf("accept fail.\n");	
			return -1;
		}
		printf("accept stat.\n");
		if((childpid = fork()) == 0){
			close(listenfd);	
			printf("servers.\n");
			str_echo(connfd);
			exit(0);
		}
		close(connfd);
	}

	return 0;
}