113 lines
5.3 KiB
C++
113 lines
5.3 KiB
C++
#ifndef _CUSTOM_SOCKET_H_
|
||
#define _CUSTOM_SOCKET_H_
|
||
|
||
/******************************************************************
|
||
*
|
||
* wyLib库 2008 - 2010
|
||
*
|
||
* $ 通用套接字处理类 $
|
||
*
|
||
* - 主要功能 -
|
||
*
|
||
* 基本的套接字封装类，支持阻塞和非阻塞模式。
|
||
*
|
||
*****************************************************************/
|
||
#include "_osdef.h"
|
||
#include "SocketConfig.h"
|
||
|
||
namespace wylib
|
||
{
|
||
namespace inet
|
||
{
|
||
namespace socket
|
||
{
|
||
/* 基本TCP套接字类
|
||
|
||
*/
|
||
class CCustomSocket
|
||
{
|
||
protected:
|
||
SOCKET m_nSocket; //套接字
|
||
SOCKADDR_IN m_LocalAddr; //绑定的本地地址
|
||
SOCKADDR_IN m_RemoteAddr; //远程地址
|
||
bool m_boBlock; //是否非阻塞模式
|
||
bool m_boConnected; //是否已建立连接
|
||
bool m_boConnecting; //是否正在连接
|
||
protected:
|
||
//提供向最外层逻辑通知连接建立的事件
|
||
virtual VOID OnConnected();
|
||
//提供向最外层逻辑通知连接断开的事件
|
||
virtual VOID OnDisconnected();
|
||
//提供向最外层逻辑通知连接发生错误的事件
|
||
virtual VOID OnError(int errorCode);
|
||
protected:
|
||
//继承类通过重载此函数处理建立连接需要进行的操作
|
||
virtual VOID Connected();
|
||
//继承类通过重载此函数处理断开连接需要进行的操作
|
||
virtual VOID Disconnected();
|
||
//继承类通过重载此函数处理连接发生了错误的所需要进行的操作
|
||
virtual VOID SocketError(int nErrorCode);
|
||
public:
|
||
CCustomSocket();
|
||
virtual ~CCustomSocket();
|
||
|
||
//获得SOCKET
|
||
inline SOCKET getSocket(){ return m_nSocket; }
|
||
//设置SOCKET
|
||
VOID setSocket(const SOCKET socket);
|
||
//获取是阻塞模式。true表示阻塞，false表示非阻塞
|
||
inline bool getBlockMode(){ return m_boBlock; }
|
||
//设置阻塞模式。true表示阻塞，false表示非阻塞，返回值为socket错误号，0表示成功
|
||
INT setBlockMode(const bool block);
|
||
//判断SOCKET是否处于连接状态
|
||
inline bool connected(){ return m_boConnected; }
|
||
//判断SOCKET是否处于正在连接状态
|
||
inline bool connecting(){ return m_boConnecting; }
|
||
//断开连接并关闭套接字
|
||
VOID close();
|
||
//断开连接,sd表示断开的方式,分别有SD_RECEIVE、SD_SEND和SD_BOTH
|
||
INT shutDown(const INT sd);
|
||
//绑定指定32位的IP地址和字符串绑定到制定的地址和端口，返回值为socket错误号，0表示成功
|
||
INT bind(const ULONG addr, const INT port);
|
||
//通过指定IP地址字符串和端口绑定到指定的地址和端口，返回值为socket错误号，0表示成功
|
||
INT bind(const char * addr, const INT port);
|
||
//获取绑定的本地地址和端口的sockaddr_in结构
|
||
inline VOID getLoaclAddr(PSOCKADDR_IN addr_in){ *addr_in = m_LocalAddr; }
|
||
//获取远程的地址和端口的sockaddr_in结构
|
||
inline VOID getRemoteAddr(PSOCKADDR_IN addr_in){ *addr_in = m_RemoteAddr; }
|
||
//设置远程地址信息
|
||
inline VOID setRemoteAddr(const PSOCKADDR_IN addr_in){ m_RemoteAddr = *addr_in; }
|
||
//获取接收缓冲区大小，返回值为socket错误号，0表示成功。
|
||
INT getRecvBufSize(ULONG *size);
|
||
//设置接收缓冲区大小，返回值为socket错误号，0表示成功。
|
||
INT setRecvBufSize(ULONG size);
|
||
//获取发送缓冲区大小，返回值为socket错误号，0表示成功。
|
||
INT getSendBufSize(ULONG *size);
|
||
//设置发送缓冲区大小，返回值为socket错误号，0表示成功。
|
||
INT setSendBufSize(ULONG size);
|
||
//开始监听，backlog表示每次可接受的连接数量，返回值为socket错误号，0表示成功
|
||
INT listen(const INT backlog = 5);
|
||
//通过32位IP地址连接到制定的端口，返回值为socket错误号，0表示成功。对于非阻塞模式返回0并不表示已经建立连接
|
||
INT connect(const ULONG addr, const INT port);
|
||
//通过IP地址字符串连接到指定的端口，返回值为socket错误号，0表示成功。对于非阻塞模式返回0并不表示已经建立连接
|
||
INT connect(const char * addr, const INT port);
|
||
|
||
INT connect(const char * addr, const INT port, int timeout);
|
||
INT connect(const ULONG addr, const INT port, int timeout);
|
||
|
||
//接收新的连接，wait_msec表示最大等待毫秒数，函数返回0表示成功且socket被接收的套接字填充，返回SOCKET_ERROR-1表示等待超时，否则函数返回socket错误号
|
||
INT accept(SOCKET * socket, unsigned long wait_msec, PSOCKADDR_IN addr);
|
||
//创建套接字，返回值为socket错误号，0表示成功且socket的内容被填充为新建立的socket
|
||
INT createSocket(OUT SOCKET *socket, const INT af = AF_INET, const int type = SOCK_STREAM, const int protocol = IPPROTO_TCP);
|
||
//读取套接字内容。返回0表示连接已断开，返回SOCKET_ERROR表示发生错误，返回SOCKET_ERROR-1表示没有数据可读(仅对于非阻塞模式套接字)，否则表示实际接收的字节数
|
||
virtual INT recv(LPVOID buf, INT len, const INT flags = 0);
|
||
//向套接字写入内容，返回0表示连接已断开，返回SOCKET_ERROR表示发生错误，返回SOCKET_ERROR-1表示send操作会发生阻塞且没有数据被发送(仅对于非阻塞模式套接字)，否则返回发送的字节数。
|
||
virtual INT send(LPVOID buf, INT len, const INT flags = 0);
|
||
};
|
||
};
|
||
};
|
||
};
|
||
|
||
#endif
|
||
|