linux下socket编程

1.实现一般socket编程 

        socket编程的基本函数有socket()、bind()、listen()、accept()、send()、sendto()、recv()、recvfrom(),其中根据是客户端还是服务器端，或者根据使用TCP协议还是UDP协议，这些函数的调用流程都有所区别。这里先对每个函数进行说明。

socket():

该函数用于建立一个套接字，一条通信路线的端点。在建立了socket之后，可对sockaddr或sockaddr_in结构进行初始化，已保存所建立的socket地址信息。

bind():

该函数用于将sockaddr结构的地址信息与套接字进程绑定。它主要用于TCP的连接，而在UDP的连接中则没有必要。

listen():

在服务器端程序成功建立套接字和与地址进行绑定之后，还需要准备在该套接字上接收新的连接请求。此时调用listen()函数来创建一个等待队列，在其中存放未处理的客户端连接请求。

accept():

服务器端程序调用listen()函数创建等待队列之后，调用accept()函数等待并接收客户端的连接请求。它通常从由listen()所创建的等待队列中取出第一个未处理的连接请求。

connet():

客户端通过一个未命名套接字和服务器监听套接字之间建立的方法来连接到服务器。该工作客户端通过使用connect()函数来实现。

send()和recv():

这两个函数分别用于发送和接收数据，可以用在TCP中，也可以冻灾UDP中。当用在UDP中时，可以在connect()函数建立连接之后再使用。

sendto()和recvfrom():

这两个函数的作用与send()和recv()函数类似，也可以用在TCP和UDP中。当用在TCP中时，后边的几个与地址有关的参数不起作用，函数作用等同于send()和recv()；当用在UDP中时，可以用再之前没有使用connect()的情况下，这两个函数可以自动寻找指定地址并进行连接。

服务器端与客户端使用TCP协议的流程如下：

服务器与客户端使用UDP协议的流程入下： 

 

TCP协议代码如下，运行时需要先启动服务器，在启动客户端。

服务器端

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

#define PORT				4321
#define BUFFER_SIZE			1024
#define MAX_QUE_CONN_NM		5

int main()
{
	struct sockaddr_in server_sockaddr,client_sockaddr;
	int sin_size,recvbytes;
	int sockfd,client_fd;
	char buf[BUFFER_SIZE];
	
	//建立socket连接
	if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1)
	{
		perror("socket");
		exit(1);
	}
	printf("Socket id = %d\n",sockfd);
	
	//设置sockaddr_in 结构体中相关参数
	server_sockaddr.sin_family 			= AF_INET;
	server_sockaddr.sin_port			= htons(PORT);
	server_sockaddr.sin_addr.s_addr		= INADDR_ANY;
	bzero((&server_sockaddr.sin_zero),8);
	
	int i = 1;//允许重复使用本地址与套接字进行绑定
	setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR, &i, sizeof(i));
	
	//绑定函数bind()
	if(bind(sockfd,(struct sockaddr *)&server_sockaddr,sizeof(struct sockaddr)) == -1)
	{
		perror("bind");
		exit(1);
	}
	
	printf("Bind success!\n");
	
	//调用listen()函数，创建未处理请求的队列
	if(listen(sockfd,MAX_QUE_CONN_NM) == -1)
	{
		perror("listen");
		exit(1);
	}
	printf("Listening...\n");
	
	//调用accept()函数，等待客户端的连接
	sin_size = sizeof(client_sockaddr);
	if((client_fd = accept(sockfd,(struct sockaddr *)&client_sockaddr, &sin_size)) == -1)
	{
		perror("accept");
		exit(1);
	}
	
	//调用recv()函数，接受客户端的请求
	memset(buf, 0, sizeof(buf));
	if((recvbytes = recv(client_fd,buf, BUFFER_SIZE, 0)) == -1)
	{
		perror("recv");
		exit(1);
	}
	
	printf("Receivd a message: %s\n", buf);
	close(sockfd);
	exit(0);
	
}

执行结果

客户端

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

#define PORT				4321
#define BUFFER_SIZE			1024

int main(int argc, char *argv[])
{
	int sockfd,sendbytes;
	char buf[BUFFER_SIZE];
	struct hostent *host;
	struct sockaddr_in serv_addr;
	
	if(argc < 3)
	{
		fprintf(stderr,"USAGE: ./client Hostname(or ip address) Text\n");
		exit(1);
	}
	//地址解析函数
	if((host = gethostbyname(argv[1])) == NULL)
	{
		perror("gethostbyname");
		exit(1);
	}
	
	memset(buf, 0, sizeof(buf));
	sprintf(buf,"%s",argv[2]);
	//创建socket
	if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
	{
		perror("socket");
		exit(1);
	}
	
	//设置sockaddr_in 结构体中相关参数
	serv_addr.sin_family 		= AF_INET;
	serv_addr.sin_port			= htons(PORT);
	serv_addr.sin_addr			= *((struct in_addr *)host->h_addr);
	bzero(&(serv_addr.sin_zero), 8);
	
	//调用connect()函数主动发起对服务器端的连接
	if(connect(sockfd,(struct sockaddr *)&serv_addr, sizeof(struct sockaddr)) == -1)
	{
		perror("connect");
		exit(1);
	}
	//发送消息给服务器
	if((sendbytes = send(sockfd, buf, strlen(buf), 0)) == -1)
	{
		perror("send");
		exit(1);
	}
	close(sockfd);
	exit(0);
	
}

执行结果

2.实现高级socket()编程

在实际应用中，人们往往遇到多个客户端连接服务器的情况。上边使用了阻塞函数，如果资源没有准备好，则调用该函数的进程将进入睡眠状态，就无法处理其他请求。

优化措施：

1.非阻塞和异步I/O       ----> 使用fcntl()函数

2.多路复用处理            ---->使用select()或者poll()函数 （最优方法）

3.多进程或多线程        

这里主要讲多路复用处理。

使用fcntl()函数虽然可以实现非阻塞，但在实际使用时往往会对资源是否准备完毕进行循环测试，这样就大大增加了不必要的CPU资源的占用，多进程或多线程也是有同样的问题。使用select()函数来解决这个问题

关于select()我的另一篇博文有详细讲解 linux下select()函数_pingxiaozhao的博客-CSDN博客

主要优化服务器，代码如下：

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

#define PORT				4321
#define BUFFER_SIZE			1024
#define MAX_QUE_CONN_NM		5
#define MAX_SOCK_FD			FD_SETSIZE

int main()
{
	struct sockaddr_in server_sockaddr,client_sockaddr;
	int sin_size,count;
	fd_set inset, tmp_inset;
	int sockfd,client_fd, fd;
	char buf[BUFFER_SIZE];
	
	//建立socket连接
	if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1)
	{
		perror("socket");
		exit(1);
	}
	printf("Socket id = %d\n",sockfd);
	
	//设置sockaddr_in 结构体中相关参数
	server_sockaddr.sin_family 			= AF_INET;
	server_sockaddr.sin_port			= htons(PORT);
	server_sockaddr.sin_addr.s_addr		= INADDR_ANY;
	bzero((&server_sockaddr.sin_zero),8);
	
	int i = 1;//允许重复使用本地址与套接字进行绑定
	setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &i, sizeof(i));
	
	//绑定函数bind()
	if(bind(sockfd,(struct sockaddr *)&server_sockaddr,sizeof(struct sockaddr)) == -1)
	{
		perror("bind");
		exit(1);
	}
	
	printf("Bind success!\n");
	
	//调用listen()函数，创建未处理请求的队列
	if(listen(sockfd,MAX_QUE_CONN_NM) == -1)
	{
		perror("listen");
		exit(1);
	}
	printf("Listening...\n");
	//将调用socket()函数的描述符作为文件描述符
	FD_ZERO(&inset);
	FD_SET(sockfd, &inset);
	while(1)
	{
		//文件描述符集合的备份，这样可以避免每次都进行初始化
		tmp_inset = inset;
		sin_size = sizeof(struct sockaddr_in);
		memset(buf, 0, sizeof(buf));
		
		//调用select()函数
		if(!(select(MAX_SOCK_FD, &tmp_inset, NULL, NULL, NULL) > 0))
		{
			perror("select");
		}
		for(fd = 0;fd < MAX_SOCK_FD; fd++)
		{
			if(FD_ISSET(fd, &tmp_inset) > 0)
			{
				if(fd == sockfd)
				{
					//服务器端接收客户端的连接器,调用accept()函数，等待客户端的连接
					if((client_fd = accept(sockfd,(struct sockaddr *)&client_sockaddr, &sin_size)) == -1)
					{
						perror("accept");
						exit(1);
					}
					//将新连接的客户端套接字加入观察列表中
					FD_SET(client_fd, &inset);
					printf("New connection from %d(socket)\n",client_fd);
				}
				else //处理从客户端发来的消息
				{
					if((count = recv(client_fd, buf, BUFFER_SIZE, 0)) > 0)
					{
						printf("Receivd a message from %d: %s\n", client_fd, buf);
					}
					else
					{
						close(fd);
						FD_CLR(fd, &inset);
						printf("Client %d(socket) has left\n",fd);
					}
				}
			}
		}
	}
	close(sockfd);
	exit(0);
	
}

服务器执行结果 

客户端代码和上边客户端代码一样 ，执行结果如下