多路复用

1、tcp服务器的多路复用

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <sys/wait.h>
#include <sys/select.h>


#define SIZE 1024
#define PORT 9999

// sfd 用于和客户端通信
void handl_client(int sfd, fd_set *allset)
{
	char buf[SIZE] = {0};
	int ret = read (sfd, buf, SIZE-1);
	if (ret == -1)
	{
		perror ("read");
		return;
	}
	if (ret == 0)
	{
		printf ("客户端退出\n");
		FD_CLR(sfd, allset);
		return;
	}
	buf[ret] = '\0';
	printf ("read %d bytes : %s\n", ret, buf);
	
	int i; 
	for (i = 0; i < ret - 1; i++)
	{
		buf[i] = buf[i] + 'A' - 'a';
	}
	
	ret = write (sfd, buf, ret);
}

// 监听套接字
int init()
{
	int listen_socket = socket(AF_INET, SOCK_STREAM, 0);
	if(-1 == listen_socket)
	{
		perror("创建套接字失败");
		return -1;
	}
	
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(struct sockaddr_in));
	addr.sin_family      = AF_INET;             /* Internet地址族 */
	addr.sin_port        = htons(PORT);         /* 端口号 */
	addr.sin_addr.s_addr = htonl(INADDR_ANY);   /* IP地址, 绑定本地的所有ip地址*/

	int ret = bind(listen_socket, (const struct sockaddr *)&addr, sizeof(addr));
	if(-1 == ret)
	{
		perror("绑定失败");
		return -1;
	}
	
	// 3、监听套接字
	ret = listen(listen_socket, 5);
	if(-1 == ret)
	{
		perror("监听失败");
		return -1;
	}
	
	return listen_socket;
}

// 通信套接字
int myAccept(int listen_socket)
{
	struct sockaddr_in client_addr;
	socklen_t  len = sizeof(client_addr);
	int client_socket = accept(listen_socket, (struct sockaddr *)&client_addr, &len);
	if (-1 == client_socket)
	{
		perror("accept  失败");
		return -1;
	}
	
	printf ("客户端的 ip = %s, 端口 = %d\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
	
	
	return client_socket;
}


int main()
{
	// 初始化套接字
	int listen_fd = init();
	if (listen_fd == -1)
		return -1;
	

	int maxfd = listen_fd;
	fd_set allset;
	fd_set rset;
	
	FD_ZERO(&allset);              // 将文件描述符集清空 
	FD_SET(listen_fd, &allset);    // 将监听文件描述符加入到监听序列
	
	int client_fd[FD_SETSIZE];     // 代表select能够检测客户端数量
	
	// 对数组进行初始化，全部置为-1，凡是值为-1的位置，都是空位置
	int i;
	for (i = 0; i < FD_SETSIZE; i++)
	{
		client_fd[i] = -1;
	}
	
	while (1)
	{
		rset = allset;
		int count = select(maxfd + 1, &rset, NULL, NULL, NULL);
		if (count == -1)
		{
			perror ("select");
			break;
		}
		
		if (FD_ISSET(listen_fd, &rset))  // 查看是否有客户端要连接
		{
			int conn_fd = myAccept(listen_fd);
			if (conn_fd == -1)
			{
				continue;
			}
			
			// 找个空位置存与客户端通信的文件描述符
			for (i = 0; i < FD_SETSIZE; i++)
			{
				if (client_fd[i] == -1)     // 找到一个空位置
				{
					client_fd[i] = conn_fd;
					break;
				}
			}
			
			// 判断文件描述符集是否已达上限
			if (i == FD_SETSIZE)
			{
				close (conn_fd);
			}
			else
			{
				FD_SET(conn_fd, &allset);
				if (maxfd < conn_fd)
					maxfd = conn_fd;
			}
			
			if (--count <= 0)
				continue;
		}
		
		for (i = 0; i < FD_SETSIZE; i++)
		{
			if (client_fd[i] == -1)
				continue;
			
			if (FD_ISSET(client_fd[i], &rset))
			{
				handl_client(client_fd[i], &allset);
				if (--count <= 0)
					break;
			}
		}
	}
	
	close (listen_fd);
	return 0;
}

2、并发

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

#define  PORT   9998

int conm_socket; 

typedef struct _msg
{
	char buf[256];
	struct sockaddr_in client_addr;
}Msg;

Msg task[100];   // 任务
int front = 0;
int rear  = 0;


pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;  // 互斥锁
pthread_cond_t cond1;    // 条件变量


void handl_client(Msg *data)
{
	int i;
	for (i = 0; i < strlen(data->buf)-1; i++)
	{
		data->buf[i] += 'A' - 'a';
	}
	
	sendto(conm_socket, data->buf, strlen(data->buf), 0, (struct sockaddr *)&(data->client_addr),  sizeof(data->client_addr));// 回
}


// 监听套接字
int init()
{
	int conm_socket = socket(AF_INET, SOCK_DGRAM, 0);
	if(-1 == conm_socket)
	{
		perror("创建套接字失败");
		return -1;
	}
	
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(struct sockaddr_in));
	addr.sin_family      = AF_INET;             /* Internet地址族 */
	addr.sin_port        = htons(PORT);         /* 端口号 */
	addr.sin_addr.s_addr = htonl(INADDR_ANY);   /* IP地址, 绑定本地的所有ip地址*/

	int ret = bind(conm_socket, (const struct sockaddr *)&addr, sizeof(addr));
	if(-1 == ret)
	{
		perror("绑定失败");
		return -1;
	}
	
	return conm_socket;
}

void *consume(void *v)
{
	while (1)
	{
		Msg data;
		pthread_mutex_lock(&mutex);
		while (rear == front)
		{
			pthread_cond_wait(&cond1, &mutex);
		}
		
		front = (front+1)%20;
		data = task[front];
		
		pthread_mutex_unlock(&mutex);
		
		handl_client(&data);
	}
}

void  produce()
{
	char buf[1024];
	while (1)
	{
		struct sockaddr_in client_addr;
		socklen_t  len = sizeof(client_addr);
		ssize_t ret = recvfrom(conm_socket, buf, 1023, 0, (struct sockaddr *)&client_addr, &len);// 收
		buf[ret] = '\0';
		
		printf ("[%s:%d] %s", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port), buf);
		
		pthread_mutex_lock(&mutex);
		
		if ((rear+1)%20 == front)
			continue;
		
		rear = (rear+1)%20;
		strcpy(task[rear].buf, buf);
		task[rear].client_addr = client_addr;
		
		pthread_cond_broadcast(&cond1);     // 广播
		pthread_mutex_unlock(&mutex);	
	}
}

int main(int argc, char **argv)
{
	long i;
	pthread_t thread;
	for (i = 0; i < 8; i++)
	{
		pthread_create(&thread, NULL, consume, (void*)(i+1));
		pthread_detach(thread);
	}
	
	pthread_cond_init(&cond1, NULL);
	
	conm_socket = init();
	if (-1 == conm_socket)
		return;
	
	produce();
	
	return 0;
}