仿mbuf机制实现自己的不定长内存池
最近在考虑一个客户端底层通讯模块的实现问题，我需要做的就是把我的套接字绑定到窗口上，当套接字有事件产生，比如可读，关闭等消息的时候，直接通过窗口消息通知我，我在做一定的处理，客户端需要实现一个可伸缩的内存池来管理通讯的数据，上层在一桢中可压入许多待发送的数据，底层在一桢可接受许多服务器端的数据。所以我打算使用变长的内存池来实现，以前看过tcp/ip底层的mbuf实现机制，感觉很不错，所以这次做了一些改进，来实现我的不定长内存池策略。/* * 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>structmbuf{public:enum{MBUF_BUFFER_SIZE = 2048};public:structmbuf*pNext;structmbuf*pRef;char* pReader;char* pWriter;size_t iUsed;size_t iUnused;char buff;public:inlinevoidinit_mbuf(){pNext =NULL;pRef =this;pReader=buff;pWriter=buff;iUsed =0;iUnused=MBUF_BUFFER_SIZE;memset(buff, 0, sizeof(buff));}};structmbuf;class mbufpool{private:structmbuf*m_pBegin;structmbuf*m_pEnd;structmbuf*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;mbufpoolm_recvpool;mbufpoolm_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();};感觉这个方法挺不错的，当然，如果各位朋友有更好的建议或者意见的话，欢迎和我联系^__^
页:
[1]