c++ 封装socket和epoll编程

最新推荐文章于 2024-06-24 18:34:08 发布

jlaij

最新推荐文章于 2024-06-24 18:34:08 发布

阅读量2.9k

点赞数

分类专栏： c/c++ linux学习文章标签： c++ epoll socket

本文链接：https://blog.csdn.net/chinabhlt/article/details/50014735

版权

c/c++ 同时被 2 个专栏收录

223 篇文章 0 订阅

订阅专栏

linux学习

75 篇文章 0 订阅

订阅专栏

本文介绍了如何在C++中封装socket和epoll进行编程，通过实例展示了具体实现过程，包括CSocket和CSocketEvent的使用，并提供了编译命令。

摘要由CSDN通过智能技术生成

编程实例：

CSocket.h文件请参考前一篇文章

CSocketEvent.h

#include <sys/epoll.h>  
#include <iostream>
#include <cstdio>
#include "CSocket.h"

const int MaxEvents = 256;
const int EpollSize = 500;

class CSocketEvent
{
public:
    CSocketEvent();
    virtual ~CSocketEvent();
    bool eventSet(CSocket *socket, bool readable, bool writeable);
	bool eventAdd(CSocket *socket, bool readable, bool writeable);
	bool eventDel(CSocket *socket);
	int getEvents(int timeout);
	struct epoll_event events[MaxEvents];
private:
	int epfd;
	CSocketEvent(const CSocketEvent&);
	CSocketEvent& operator=(const CSocketEvent&);
};

CSocketEvent::CSocketEvent()
{
	if((epfd = epoll_create(EpollSize)) < 0)
	{
		std::cout << "create epoll handle err\n";
		return;
	}
	memset(&events, 0, sizeof(struct epoll_event)*MaxEvents);
}

CSocketEvent::~CSocketEvent()
{
	if(epfd > 0)
	{
		close(epfd);
		epfd = -1;
	}
}

bool CSocketEvent::eventSet(CSocket *socket, bool readable, bool writeable)
{
	int	sockfd = socket->getSockfd();
	if(sockfd < 0)
	{
		std::cout << "eventSet the socket fd err\n";
		return false;
	}
	struct epoll_event ev;
	memset(&ev, 0, sizeof(ev));
	ev.events |= EPOLLET; //ET modle as default	
	if(readable)
		ev.events |= EPOLLIN;
	if(writeable)
		ev.events |= EPOLLOUT;
	//ev.data.ptr = (void*)sockfd;
	return ( 0 == epoll_ctl(epfd, EPOLL_CTL_MOD, sockfd, &ev));
}

bool CSocketEvent::eventAdd(CSocket *socket, bool readable, bool writeable)
{
	int sockfd = -1;
	if(socket->setNonBlock(true))
		sockfd = socket->getSockfd();
	if(sockfd < 0)
	{
		std::cout << "eventAdd the socket fd err\n";
		return false;
	}
	struct epoll_event ev;
	memset(&ev, 0, sizeof(ev));
	ev.events |= EPOLLET; //ET modle as default
	if(readable)
		ev.events |= EPOLLIN;
	if(writeable)
		ev.events |= EPOLLOUT;
	//ev.data.ptr = sockfd;
	return ( 0 == epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev));
}

bool CSocketEvent::eventDel(CSocket *socket)
{
	int sockfd = socket->getSockfd();
	if(sockfd < 0)
	{
		std::cout << "eventDel the socket fd err\n";
		return false;
	}
	struct epoll_event ev;
	memset(&ev, 0, sizeof(ev));
	ev.events = 0;
	return (0==epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, &ev));
}

int CSocketEvent::getEvents(int timeout)
{
	int ret = epoll_wait(epfd, events, MaxEvents, timeout);
	return ret;
}

TestSocketEvent.cpp

#include "CSocketEvent.h"

int main(int argc, char *argv[])
{
	CServSocket servSocket;
	CSocketEvent socketEvent;
	char buf[256];
	memset(buf,0, sizeof(buf));
	if(!servSocket.Listen(9999))
	{
		std::cout << "listen err\n";		
	}
	for(;;)
	{
	    int ret = socketEvent.getEvents(1000);
		for(int i=0; i< ret; ++i)
		{
			if(socketEvent.events[i].data.fd == servSocket.getSockfd())
			{
				CSocket *socket = NULL;
				if((socket = servSocket.Accept()) == NULL)
				{
					std::cout << "accpt error\n";
					break;
				}
				socketEvent.eventAdd(socket, true, true);
				delete socket;
			}
			else if(socketEvent.events[i].events & EPOLLIN)
			{
				std::cout << "event trigger ...\n";
				int sockfd = socketEvent.events[i].data.fd;
				CSocket *socket = new CSocket();
				socket->setUp(sockfd, NULL);
				socket->Recv(buf, 10, 0);
				std::cout << buf << "\n";
				socketEvent.eventSet(socket, true, true);
				delete socket;
			}
		}
	}
    return 0;
}

编译：

g++ -o TestSocketEvent TestSocketEvent.cpp CSocketEvent.h CSocket.h