/* 
* Copyright(c)2008 
* 
* 文件名称:  mbuf 
* 文件标识: 
* 摘    要: 仿照tcp/ip的mbuf机制实现自己的mbuf 
			pWriter只仅仅表示我们当前的写指针位置，
			而我们的iUnused只仅仅的表示我们的写指
			针位置距离我们的buff+MBUF_BUFFER_SIZE
			还有多少个可用字节数目。
			pReader只仅仅表示我们当前的读指针位置，
			而我们的iUsed也只仅仅表示我们的读指针
			位置距离我们的pWriter指针还有多少字节
			这个再次注意，我们的iUnused+iUsed不一定
			就等于buff的总体长度MBUF_BUFFER_SIZE，
			这两者之间没有任何联系
* 
* 当前版本: cobra_for_window 1.00 
* 作    者: 关中刀客 
* E-Mail  : guanzhongdaoke@gmail.com 
* Blog    : http://guan-zhong-dao-ke.blog.163.com/ 
* 完成时间: 2008年04月22日 
*/ 


#include <Windows.h>


struct  mbuf
{
public:
	enum
	{
		MBUF_BUFFER_SIZE = 2048
	};

public:
	struct  mbuf*  pNext;
	struct  mbuf*  pRef;
	char*          pReader;
	char*          pWriter;
	size_t         iUsed;
	size_t         iUnused;
	char           buff[MBUF_BUFFER_SIZE];

public:
	inline  void  init_mbuf()
	{
		pNext    =  NULL;
		pRef     =  this;
		pReader  =  buff;
		pWriter  =  buff;
		iUsed    =  0;
		iUnused  =  MBUF_BUFFER_SIZE;
		
		memset(buff, 0, sizeof(buff));
	}
};


struct  mbuf;
class   mbufpool
{
private:
	struct  mbuf*  m_pBegin;
	struct  mbuf*  m_pEnd;
	struct  mbuf*  m_pFinish;
	size_t         m_iBytes;

public:
	mbufpool();
	~mbufpool();

public:
	/*
		得到数据开始的节点
	*/
	mbuf*    begin();
	/*
		得到数据结尾的节点
	*/
	mbuf*    end();
	/*
		得到当前mbufpool中所存字节的总数
	*/
	size_t   get_bytes_number();
	/*
		设计当前mbufpool中所存字节的总数
	*/
	void     set_bytes_number(size_t iBytes);
	/*
		增加制定数目的mbuf节点
	*/
	void     resize(size_t iNum);
	/*
		m_pBegin向后移动
	*/
	void     next();
	/*
		压入一些节点
	*/
	void     push_node(mbuf* pNode);
	/*
		压入一定长度的数据
	*/
	void     push_data(const char* pData, size_t iNum);
	/*
		得到一定数目的字节
	*/
	bool     pop_data(char* pData, size_t iNum);
	/*
		销毁mbufpool
	*/
	void     destroy();

private:
	/*
		从系统中分配节点
	*/
	void     malloc_mbuf_from_system();
};



class cobra_socket
{
private:
	SOCKET    m_sock;
	mbufpool  m_recvpool;
	mbufpool  m_sendpool;

public:
	cobra_socket();
	~cobra_socket();

public:
	void    init_cobra_socket(size_t iRecvpollSize, size_t iSendPoolSize);
	bool    create_socket();
	bool    connect_to_server(const char* pIP, u_short port);
	bool    bind_to_window(HWND hWnd, int wMsg, int iMsg);
	bool    logic_recv(char* pBuff, size_t iNum);
	void    logic_send(const char* pBuff, size_t iNum);
	bool    start_recv();
	bool    start_send();
	void    close_socket();
	void    destroy_cobra_socket();
};

• JavaEye问答大赛开始了！ 从6月23日 至 7月6日，奖品丰厚 ！
• Oracle专区上线，有Oracle最新文章，重要下载及知识库等精彩内容，欢迎访问。
• 快来参加7月17日在成都举行的SOA中国技术论坛
• 搜狐网站诚聘Java、PHP和C++工程师
• 北京: 千橡集团暨校内网诚聘软件研发工程师

• JavaEye问答大赛开始了！ 从6月23日 至 7月6日，奖品丰厚 ！
• Oracle专区上线，有Oracle最新文章，重要下载及知识库等精彩内容，欢迎访问。
• 搜狐网站诚聘Java、PHP和C++工程师
• 快来参加7月17日在成都举行的SOA中国技术论坛
• 北京: 千橡集团暨校内网诚聘软件研发工程师

#ifndef  _LC_COBRA_EPOLLER_H
#define  _LC_COBRA_EPOLLER_H



/*
* Copyright(c)2008
*
* 文件名称: Epoller
* 文件标识:
* 摘    要: Epoller是我自己封装的epoll，仿照
            Windows平台的IOCP(完成端口)来实现类似
			的一套高性能的网络底层io策略。
			Epoller实现的原理是：
			但个线程来循环我们的epoll来等待底层的网络事件
			当有事件的时候，我们就把这些事件对应的数据放入
			到我们的上层Epoller缓冲队列中，并且发出信号，通知
			阻塞在这个上面的线程可以取数据去处理。
*
* 当前版本: cobra 1.05
* 作    者: 关中刀客
* E-Mail  : guanzhongdaoke@gmail.com
* Blog    : http://guan-zhong-dao-ke.blog.163.com/
* 完成时间: 2008年04月28日
*/


#include "../common/Header.h"
#include <sys/epoll.h>
#include <sys/socket.h>
#include <pthread.h>
#include <semaphore.h>


class   EpollerEvent;
class   Epoller
{
public:
	/*
		需要监听的套接字Epoller消息
	*/
	enum EPOLLERMESSAGE
	{
		EPOLLERREAD,
		EPOLLERWRITE
	};

private:
	int                  m_epoller;   // epoller句柄
	pthread_mutex_t      m_lock;      // 中间队列的锁
	pthread_cond_t       m_cond;      // 条件变量锁
	sem_t                m_sem;       // 主轮循等待线程处理的信号量
	Queue<EpollerEvent, EmptyLock>  m_queue;  // 中间的数据结构缓冲队列 

public:
	Epoller();
	~Epoller();

public:
	/*
		初始化epoller
		queuesize : 初始化的时候我们的缓冲队列默认的大小
		number    : 处理的数目
	*/
	bool   init_epoller(size_t queuesize, size_t number);
	/*
		向epoller注册套接字需要监听事件
		sock   :   指定的套接字
		iMsg   :   注册到Epoller上的消息
		pEvent :   指定的事件数据体
	*/
	bool   register_epoller_socket_event(SOCKET sock, int iMsg, EpollerEvent* pEvent);
	/*
		改变Epoller中套接字监听的事件
		sock   :   指定的套接字
		iMsg   :   注册到Epoller上的消息
		pEvent :   指定的事件数据体
	*/
	bool   change_epoller_socket_event(SOCKET sock, int iMsg, EpollerEvent* pEvent);
	/*
		删除Epoller中套接字的所有事件
	*/
	void   delete_epoller_socket_event(SOCKET sock, int iMsg);
	/*
		向Epoller发送指定的事件
		pEvent :   指定的事件结构体
		通常情况下，我们使用这个函数控制各Epoller线程的退出
		类似于windows下的PostQueuedCompletionStatus
	*/
	void   post_event_to_epoller(EpollerEvent* pEvent);
	/*
		得到Epoller中的事件结构体数据
		pEvent :  表示得到的数据体
		一般情况下，这个函数由各Epoller线程函数调用,然后回得到指定的事件，
		然后根据事件的信息去处理
		类似于windows下的GetQueuedCompletionStatus
	*/
	bool   get_event_from_epoller(EpollerEvent* pEvent, size_t timeout);
	/*
		单独线程来循环Epoller主循环，检测套接字的事件
		timeout :  超时的时间
	*/
	int    listen_epoller_event(int timeout);
	/*
		销毁epoller
	*/
	void   destroy_epoller();
};
extern  Epoller  g_Epoller;


#endif

• 搜狐网站诚聘Java、PHP和C++工程师
• 北京: 千橡集团暨校内网诚聘软件研发工程师
• JavaEye问答大赛开始了！ 从6月23日 至 7月6日，奖品丰厚 ！
• 快来参加7月17日在成都举行的SOA中国技术论坛
• Oracle专区上线，有Oracle最新文章，重要下载及知识库等精彩内容，欢迎访问。

• 快来参加7月17日在成都举行的SOA中国技术论坛
• 北京: 千橡集团暨校内网诚聘软件研发工程师
• Oracle专区上线，有Oracle最新文章，重要下载及知识库等精彩内容，欢迎访问。
• JavaEye问答大赛开始了！ 从6月23日 至 7月6日，奖品丰厚 ！
• 搜狐网站诚聘Java、PHP和C++工程师