socket epoll网络编程实例

其他 2019-05-14 10:30:55 阅读次数: 0
版权声明：墨痕诉清风 https://blog.csdn.net/u012206617/article/details/89301624

总结了各个博主的经验，写出了简单的demo，实例为ET模式，转载请写明出处，如有宝贵意见请留言。

第一版：

服务端：

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


#define MAXBUF 1024
#define MAXEPOLLSIZE 10000


int main(int argc, char *argv[])
{
	//设置端口
	if(argc != 2)
	{  
		printf("请设置端口号！\n");
	}
	int port = atoi(argv[1]);  
	
	int listener, conn_sock, kdpfd, nfds, n, ret, curfds;
	socklen_t len;
	struct sockaddr_in server_addr, client_addr;
	struct epoll_event ev;
	struct epoll_event pevent[MAXEPOLLSIZE];
	struct rlimit rt;
	rt.rlim_max = rt.rlim_cur = MAXEPOLLSIZE;
	
	//设置系统资源，打开最大文件数
	if (setrlimit(RLIMIT_NOFILE, &rt) == -1)
	{
		perror("setrlimit");
		exit(EXIT_FAILURE);
	}
	else
	{
		printf("设置系统资源参数成功！\n");
	}
	
	//创建socket
	if( (listener = socket(AF_INET, SOCK_STREAM, 0)) == -1)
	{
		perror("socket");
		exit(EXIT_FAILURE);
	}
	else
	{
		printf("socket 创建成功！\n");
	}
	
	//设置非堵塞
	if (fcntl(listener, F_SETFL, fcntl(listener, F_GETFL, 0) | O_NONBLOCK) == -1)
	{
		perror("fcntl");
		exit(EXIT_FAILURE);
	}
	
	bzero(&server_addr, sizeof(server_addr));
	server_addr.sin_family = AF_INET;
	server_addr.sin_port = htons(port);
	server_addr.sin_addr.s_addr = INADDR_ANY;  //0.0.0.0所有地址
	
	//绑定
	if ( bind( listener, (struct sockaddr*)&server_addr, sizeof(struct sockaddr)) == -1 )
	{
		perror("bind");
		exit(EXIT_FAILURE);
	}
	else
	{
		printf("IP 地址和端口绑定成功\n");
	}
	
	if (listen(listener, 10) == -1)
	{
		perror("listen");
		exit(EXIT_FAILURE);
	}
	else
	{
		printf("开启服务成功！\n");
	}

	//创建epoll为ET模式
	kdpfd = epoll_create(MAXEPOLLSIZE);
	len = sizeof(struct sockaddr_in);
	ev.events = EPOLLIN | EPOLLET;
	ev.data.fd = listener;
	
	//socket加入epoll
	if( epoll_ctl(kdpfd, EPOLL_CTL_ADD, listener, &ev) < 0 )
	{
		fprintf( stderr, "epoll set insertion error: fd=%d\n", listener );
		exit(EXIT_FAILURE);
	}
	else
	{
		printf("监听 socket 加入 epoll 成功！\n");
	}
	
	//设置延迟和事件个数,事件由累加完成
	curfds = 1;
	//int timeout = 10*1000;
	while(1)
	{
		//等待有事件发生
		//nfds = epoll_wait(kdpfd, pevent, curfds, timeout);
		nfds = epoll_wait(kdpfd, pevent, curfds, -1);
		if( nfds == -1 )
		{
			perror("epoll_wait");
			break;
		}
		else if (nfds == 0)
		{
			printf("waiting for connecting...\n");
			continue;
		}
		
		for (n = 0; n < nfds; ++n)
		{
			if ((pevent[n].events & EPOLLERR) || (pevent[n].events & EPOLLHUP) || (!(pevent[n].events & EPOLLIN)))
			{
				//此FD上发生错误，或者套接字未准备好读取（那么为什么通知我们？）
				fprintf (stderr, "epoll error\n");
				close(pevent[n].data.fd);
				continue;
			}
			else if (pevent[n].data.fd == listener)
			{
				//我们在监听套接字上有一个通知,这意味着一个或多个传入连接
				while (1)
				{
					conn_sock = accept(listener, (struct sockaddr*)&client_addr, &len);
					if( conn_sock == -1 )
					{
						if ((errno == EAGAIN) || (errno == EWOULDBLOCK))
						{
							//我们已经处理了所有传入的连接
							break;
						}
						else
						{
							perror ("accept error");
							break;
						}
					}
					//else	
					//	printf("有连接来自于： %s:%d， 分配的 socket 为:%d\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port), conn_sock);
					
					char hbuf[1024], sbuf[1024];
					if ( 0 == getnameinfo((struct sockaddr*)&client_addr, len, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV))
						printf("Accepted connection on descriptor %d (host=%s, port=%s)\n", conn_sock, hbuf, sbuf);
					
					if (fcntl(conn_sock, F_SETFL, fcntl(conn_sock, F_GETFL, 0) | O_NONBLOCK) == -1)
					{
						perror("fcntl");
						break;
					}

					ev.events = EPOLLIN | EPOLLET;
					ev.data.fd = conn_sock;
					
					if( -1 == epoll_ctl( kdpfd, EPOLL_CTL_ADD, conn_sock, &ev))
					{
						fprintf(stderr, "把 socket '%d' 加入 epoll 失败！%s\n", conn_sock, strerror(errno));
						exit(EXIT_FAILURE);
					}
					
					curfds ++;					
				}
				continue;
			}
			else
			{
				if (do_use_fd(pevent[n].data.fd) < 0)
				{
					printf ("关闭 %d\n", pevent[n].data.fd);					
					epoll_ctl(kdpfd, EPOLL_CTL_DEL, pevent[n].data.fd,&ev);
                    close(pevent[n].data.fd);
					curfds--;
				}
			}
		}
	}
	
	close(listener);
	close(kdpfd);
	return 0;
}

int do_use_fd(int connfd)
{
	int done = 0;
	
	while(1)
	{
		char buf[MAXBUF + 1];
		bzero(buf, MAXBUF + 1);
		int nread;
		
		//读取客户端socket流
		nread = recv(connfd, buf, MAXBUF, 0);
		if (nread == -1)
		{
			if (errno != EAGAIN)
			{
				perror ("recv");
				done = -1;
			}
			break;
		}
		else if (nread == 0)
		{
			done = -1;
			break;
		}
		
		printf("%d接收消息成功:'%s'，共%d个字节的数据\n", connfd, buf, nread);
		
		//响应客户端 
		if ( -1 == send(connfd, buf, strlen(buf), 0))
			perror ("write");
	}

    return done;
}

第二版：

.h

#ifndef _INITNETWORK_H_
#define _INITNETWORK_H_

#include <stdio.h>
#include <sys/epoll.h>
#include <stdlib.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/resource.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <pthread.h>


#define MAXBUF 1024
#define MAXEPOLLSIZE 10000


typedef struct SOCKET_EPOLL_PAR{
	int                listener;              //socket句柄
	int                port;                  //socket端口
	int                kdpfd;                 //epoll句柄
	int                curfds;                //socket连接个数
	socklen_t          len;                   //len = sizeof(struct sockaddr_in);
	struct epoll_event ev;                    //epoll 模式
	int                fd;                    //当前事件
	struct epoll_event pevent[MAXEPOLLSIZE];  //事件数集合
}SEPAR;

static void *thread_conn(void *arg);
static void *thread_do_use_fd(void *arg);
int init_system_res();
int init_socket(int *listener, int *port);
int init_epoll(SEPAR *se_par);
void run_network(int port);
int close_network(int listener, int kdpfd);


#endif

.c

#include "initnetwork.h"


void *thread_conn(void *arg)
{
	SEPAR *pse_par = (SEPAR*)arg;
	
	struct sockaddr_in client_addr;
	
	while (1)
	{	
		int conn_sock = accept(pse_par->listener, (struct sockaddr*)&client_addr, &(pse_par->len));
		if( conn_sock == -1 )
		{
			if ((errno == EAGAIN) || (errno == EWOULDBLOCK))
			{
				//我们已经处理了所有传入的连接
				break;
			}
			else
			{
				perror ("accept error");
				break;
			}
		}
		
		char hbuf[1024], sbuf[1024];
		if ( 0 == getnameinfo((struct sockaddr*)&client_addr, pse_par->len, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV))
			printf("Accepted connection on descriptor %d (host=%s, port=%s)\n", conn_sock, hbuf, sbuf);
		
		if (fcntl(conn_sock, F_SETFL, fcntl(conn_sock, F_GETFL, 0) | O_NONBLOCK) == -1)
		{
			perror("fcntl");
			break;
		}

		pse_par->ev.events = EPOLLIN | EPOLLET;
		pse_par->ev.data.fd = conn_sock;
		
		if( -1 == epoll_ctl( pse_par->kdpfd, EPOLL_CTL_ADD, conn_sock, &(pse_par->ev)))
		{
			fprintf(stderr, "把 socket '%d' 加入 epoll 失败！%s\n", conn_sock, strerror(errno));
			exit(EXIT_FAILURE);
		}
		
		pse_par->curfds++;
		printf ("conn curfds: %d\n", pse_par->curfds);
	}
}


void *thread_do_use_fd(void *arg)
{
	SEPAR *pse_par = (SEPAR*)arg;
	int done = 0;
	
	int conn_fd = pse_par->pevent[pse_par->fd].data.fd;
	
	while(1)
	{
		char buf[MAXBUF + 1];
		bzero(buf, MAXBUF + 1);
		int nread;
		
		//读取客户端socket流
		nread = recv(conn_fd, buf, MAXBUF, 0);
		if (nread == -1)
		{
			if (errno != EAGAIN)
			{
				perror ("recv");
				done = -1;
			}
			break;
		}
		else if (nread == 0)
		{
			done = -1;
			break;
		}
		
		printf("%d接收消息成功:'%s'，共%d个字节的数据\n", conn_fd, buf, nread);
		
		//响应客户端 
		if ( -1 == send(conn_fd, buf, strlen(buf), 0))
			perror ("write");
	}

	if (done < 0)
	{
		printf ("关闭 %d\n", conn_fd);
		epoll_ctl(pse_par->kdpfd, EPOLL_CTL_DEL, conn_fd,&(pse_par->ev));
		close(conn_fd);
		pse_par->curfds--;
		printf ("close curfds: %d\n", pse_par->curfds);
	}
}


int init_system_res()
{
	struct rlimit rt;
	rt.rlim_max = rt.rlim_cur = MAXEPOLLSIZE;
	if (setrlimit(RLIMIT_NOFILE, &rt) == -1)
	{
		perror("setrlimit");
		return -1;
	}
	else
		printf("设置系统资源参数成功！\n");
	
	return 0;
}


int init_socket(int *listener, int *port)
{
	//创建socket
	if( (*listener = socket(AF_INET, SOCK_STREAM, 0)) == -1)
	{
		perror("socket");
		return -1;
	}
	else
		printf("socket 创建成功！\n");
	
	//设置非堵塞
	if (fcntl(*listener, F_SETFL, fcntl(*listener, F_GETFL, 0) | O_NONBLOCK) == -1)
	{
		perror("fcntl");
		return -1;
	}
	
	struct sockaddr_in server_addr;
	bzero(&server_addr, sizeof(server_addr));
	server_addr.sin_family = AF_INET;
	server_addr.sin_port = htons(*port);
	server_addr.sin_addr.s_addr = INADDR_ANY;  //0.0.0.0所有地址
	
	//绑定
	if (bind( *listener, (struct sockaddr*)&server_addr, sizeof(struct sockaddr)) == -1 )
	{
		perror("bind");
		return -1;
	}
	else
		printf("IP 地址和端口绑定成功\n");
	
	//监听
	if (listen( *listener, MAXEPOLLSIZE) == -1)
	{
		perror("listen");
		return -1;
	}
	else
		printf("开启服务成功！\n");
	
	return 0;
}


int init_epoll(SEPAR *se_par)
{
	//创建epoll为ET模式
	se_par->kdpfd = epoll_create(MAXEPOLLSIZE);
	se_par->len = sizeof(struct sockaddr_in);
	se_par->ev.events = EPOLLIN | EPOLLET;
	se_par->ev.data.fd = se_par->listener;
	
	//socket加入epoll
	if( epoll_ctl(se_par->kdpfd, EPOLL_CTL_ADD, se_par->listener, &(se_par->ev)) < 0 )
	{
		fprintf( stderr, "epoll set insertion error: fd=%d\n", se_par->listener );
		return -1;
	}
	else
		printf("监听 socket 加入 epoll 成功！\n");
	
	return 0;
}


void run_network(int port)
{
	SEPAR se_par;
	int nfds, n;
	
	se_par.port = port;
	
	//设置系统资源，打开最大文件数
	if(init_system_res() == -1)
		abort();
	
	//初始化网络
	if (init_socket(&se_par.listener, &se_par.port) == -1)
		abort();

	//创建epoll并添加socket
	if (init_epoll(&se_par) == -1)
		abort();
	
	//设置延迟和事件个数,事件由累加完成
	se_par.curfds = 1;
	//int timeout = 10*1000;
	while(1)
	{
		//等待有事件发生
		//nfds = epoll_wait(kdpfd, pevent, curfds, timeout);
		nfds = epoll_wait(se_par.kdpfd, se_par.pevent, se_par.curfds, -1);
		if( nfds == -1 )
		{
			perror("epoll_wait");
			break;
		}
		else if (nfds == 0)
		{
			printf("waiting for connecting...\n");
			continue;
		}
		
		for (n = 0; n < nfds; ++n)
		{
			if ((se_par.pevent[n].events & EPOLLERR) || (se_par.pevent[n].events & EPOLLHUP) || (!(se_par.pevent[n].events & EPOLLIN)))
			{
				//此FD上发生错误，或者套接字未准备好读取（那么为什么通知我们？）
				fprintf (stderr, "epoll error\n");
				close(se_par.pevent[n].data.fd);
				continue;
			}
			else if (se_par.pevent[n].data.fd == se_par.listener)
			{
				//我们在监听套接字上有一个通知,这意味着一个或多个传入连接				
				pthread_t tid;
				int ret = pthread_create(&tid, NULL, thread_conn, (void*)&se_par);
				if (ret != 0)
					printf ("创建epoll连接线程出错！");
				
				continue;
			}
			else
			{
				//处理事件
				pthread_t tid;
				se_par.fd = n;
				int ret = pthread_create(&tid, NULL, thread_do_use_fd, (void*)&se_par);
				if (ret != 0)
					printf ("创建epoll事件线程出错！");
			}
		}
	}
	
	if (close_network(se_par.listener, se_par.kdpfd) == -1)
		abort();
}


int close_network(int listener, int kdpfd)
{
	if (close(listener) == -1)
	{
		perror ("close listener");
		return -1;
	}
	
	if (close(kdpfd) == -1)
	{
		perror ("close kdpfd");
		return -1;
	}
	
	return 0;
}

测试客户端：

/*
 * *   socket简单编程 客户端
 *  */
 
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
 
#define BUFFSIZE 1024
 
int main(int argc, char *argv[])
{
    int client_sockfd = 0;
    int len = 0;
    struct sockaddr_in server_addr;
    char buf[BUFFSIZE] = {0};
    bzero(&server_addr, sizeof(struct sockaddr_in));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(8080);
 
    if((client_sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    {
        perror("socket error!\n");
        return -1;
    }
 
    if(connect(client_sockfd, (struct sockaddr*)&server_addr, sizeof(struct sockaddr)) < 0)
    {
        perror("client error!\n");
        return -1;
    }
 
    printf("connect success!\n");
 
//    len = recv(client_sockfd, buf, BUFFSIZE, 0);
//    buf[len] = '\0';
//    printf("client_buf = %s\n",buf);
 
   while(1)
    {
        printf("Enter string to send:");
		bzero(buf, sizeof(buf));
        scanf("%s",buf);
        if(!strcmp(buf,"quit"))
        {
            break;
        }
 
        len = send(client_sockfd, buf, strlen(buf), 0);
		bzero(buf, sizeof(buf));
        len = recv(client_sockfd, buf, BUFFSIZE, 0);
        buf[len] = '\0';
        printf("received: %s\n",buf);
    }
 
    close(client_sockfd);
 
    return 0;
}

猜你喜欢

转载自blog.csdn.net/u012206617/article/details/89301624
今日推荐
《美国对全球网络空间安全与发展的威胁和破坏》报告发布
火速冲上 GitHub 热榜 —— 开源编程语言、框架哪有这么可爱？
北京人形机器人创新中心发布全球首个纯电驱拟人奔跑的全尺寸人形机器人“天工”
LFOSSA 源来如此公开课 | 掌握云原生未来：CNCF 认证全面攻略与备考秘籍
周排行
循环神经网络（rnn）讲解
Tigao教程四：单独的关节运动
金蝶K3WISE15.0-注册套打教程
如何在Mac上配置Kubernetes
Android应用结束自身进程的方法
SpringMVC学习 十三 拦截器栈
中国驻洛杉矶总领馆举行新春招待会
HttpClient get post 发送
11 - three.js 笔记 - 绘制三维字体模型
Mysql递归获取某个父节点下面的所有子节点和子节点上的所有父节点
每日归档
更多
2024-05-01(4)
2024-04-30(1)
2024-04-29(40)
2024-04-28(0)
2024-04-27(56)
2024-04-26(39)
2024-04-25(22)
2024-04-24(36)
2024-04-23(26)
2024-04-22(39)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022