3.2 Linux随手简记之网络编程篇（TCP代码详解）

TCP服务端代码小例子，注释详解

#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>
#include <signal.h>


/* socket
 * bind
 * listen
 * accept
 * send/recv
 */

#define SERVER_PORT 8888  //定义服务器端口
#define BACKLOG     10  //定义最大监听10路连接

int main(int argc, char **argv)
{
	int iSocketServer;
	int iSocketClient; //客户端描述符
	struct sockaddr_in tSocketServerAddr;   //sockaddr_in结构体，其中包含两字节端口，四字节IP， 详见博客
	struct sockaddr_in tSocketClientAddr;
	int iRet;
	int iAddrLen; //sockaddr_in结构体大小

	int iRecvLen;
	unsigned char ucRecvBuf[1000];

	int iClientNum = -1;

	signal(SIGCHLD,SIG_IGN);//解决client退出之后，导致的僵死进程
	
	iSocketServer = socket(AF_INET, SOCK_STREAM, 0);  //创建套接字，参数：ipv4，tcp，0
	if (-1 == iSocketServer)
	{
		printf("socket error!\n");
		return -1;
	}

	tSocketServerAddr.sin_family      = AF_INET;  //
	
	//监测服务器端口，将端口转化为网络字节序
	tSocketServerAddr.sin_port        = htons(SERVER_PORT);  /* host to net, short 把主机的字节序转为网络字节序*/ 
 	tSocketServerAddr.sin_addr.s_addr = INADDR_ANY;  //一个电脑上可能有多个IP，表示本机上所有的IP
	memset(tSocketServerAddr.sin_zero, 0, 8);  //设置为0  
	
	//绑定套接字到地址（ip,端口），第二个参数是结构体，第三个是结构体的长度
	iRet = bind(iSocketServer, (const struct sockaddr *)&tSocketServerAddr, sizeof(struct sockaddr));  
	if (-1 == iRet)
	{
		printf("bind error!\n");
		return -1;
	}
	//第一个参数是开始监听的套接字，第二个是定义最大长度队列，即它可以同时监听多少路连接
	iRet = listen(iSocketServer, BACKLOG);
	if (-1 == iRet)
	{
		printf("listen error!\n");
		return -1;
	}

	while (1)
	{
		iAddrLen = sizeof(struct sockaddr);//结构体大小
		//来连接的客户端的IP和端口等信息保存在tSocketClientAddr结构体上   地址长度
		iSocketClient = accept(iSocketServer, (struct sockaddr *)&tSocketClientAddr, &iAddrLen);  //等待接受一条连接
		if (-1 != iSocketClient)
		{
			iClientNum++;    //第几个客户端
			//打印出客户端与客户端的ip地址，使用inet_ntoa，将ip地址转化为ASCII码显示出来，如192.168.1.220  
			printf("Get connect from client %d : %s\n",  iClientNum, inet_ntoa(tSocketClientAddr.sin_addr));
			//为了接收更多的客户端，开多进程
			
			/*
			fork调用的一个奇妙之处就是它仅仅被调用一次，却能够返回两次，它可能有三种不同的返回值：
			在父进程中，fork返回新创建子进程的进程ID；
			在子进程中，fork返回0；
			如果出现错误，fork返回一个负值；
			*/
			
			if (!fork())  //复制主进程的代码，fork出一个子进程。fork() == 0,为子进程，子进程执行以下代码，父进程不执行
			{
				/* 子进程的源码 */
				while (1)
				{
					/* 接收客户端发来的数据并显示出来 */
					//得到接收到的数据长度。第一个参数为客户端描述符，第二个是：接收到的数据放入缓冲区buf中，第三：最多时候接收数据长度999.源，目的。长度
					iRecvLen = recv(iSocketClient, ucRecvBuf, 999, 0);
					if (iRecvLen <= 0)
					{
						close(iSocketClient); //关闭客户端描述符
						return -1;
					}
					else
					{
						ucRecvBuf[iRecvLen] = '\0';//加个结束符
						//打印消息
						printf("Get Msg From Client %d: %s\n", iClientNum, ucRecvBuf);
					}
				}				
			}
		}
	}
	
	close(iSocketServer);
	return 0;
}



```

TCP客户端小例子代码，注释详解

#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>

/* socket
 * connect
 * send/recv
 */

#define SERVER_PORT 8888  //服务器端口

int main(int argc, char **argv)
{
	int iSocketClient;
	struct sockaddr_in tSocketServerAddr;
	
	int iRet;
	unsigned char ucSendBuf[1000];
	int iSendLen;

	if (argc != 2)
	{
		printf("Usage:\n");
		printf("%s <server_ip>\n", argv[0]);
		return -1;
	}

	iSocketClient = socket(AF_INET, SOCK_STREAM, 0);

	tSocketServerAddr.sin_family      = AF_INET;   //协议族
	tSocketServerAddr.sin_port        = htons(SERVER_PORT);  /* host to net, short */
 	//tSocketServerAddr.sin_addr.s_addr = INADDR_ANY;
 	if (0 == inet_aton(argv[1], &tSocketServerAddr.sin_addr))  //把字符串IP转化为网络字节序存到&tSocketServerAddr.sin_addr
 	{
		printf("invalid server_ip\n");
		return -1;
	}
	memset(tSocketServerAddr.sin_zero, 0, 8);


	iRet = connect(iSocketClient, (const struct sockaddr *)&tSocketServerAddr, sizeof(struct sockaddr));	
	if (-1 == iRet)
	{
		printf("connect error!\n");
		return -1;
	}

	while (1)
	{
		if (fgets(ucSendBuf, 999, stdin))   //从标准输入文件中获取数据，即键盘输入到标准输入文件，再从标准输入文件中获取数据。将获取到的数据放进ucSendBuf
		{ //iSocketClient已经与服务端连接成功，返回值是发送长度
			iSendLen = send(iSocketClient, ucSendBuf, strlen(ucSendBuf), 0);
			if (iSendLen <= 0)
			{
				close(iSocketClient);
				return -1;
			}
		}
	}
	
	return 0;
}