132 lines
4.2 KiB
C++
132 lines
4.2 KiB
C++
#ifndef _BASE_SOCKET_H_
|
||
#define _BASE_SOCKET_H_
|
||
|
||
/******************************************************************
|
||
*
|
||
* $ 通用套接字处理类 $
|
||
*
|
||
* - 主要功能 -
|
||
*
|
||
* 基本的套接字封装类，支持阻塞和非阻塞模式。
|
||
*
|
||
*****************************************************************/
|
||
#ifdef _MSC_VER
|
||
#define socklen_t int
|
||
#include <winsock.h>
|
||
#else
|
||
#define INVALID_SOCKET -1
|
||
#define SOCKET_ERROR -1
|
||
#define SOCKET int
|
||
#define WSAGetLastError() errno
|
||
#define WSAEWOULDBLOCK EWOULDBLOCK
|
||
#define closesocket close
|
||
#define OUT
|
||
|
||
#include <errno.h>
|
||
#include <sys/socket.h>
|
||
#include <fcntl.h>
|
||
#include <netinet/in.h>
|
||
#include <netdb.h>
|
||
#endif
|
||
|
||
namespace net
|
||
{
|
||
/* 基本TCP套接字类
|
||
*/
|
||
class BaseSocket
|
||
{
|
||
protected:
|
||
SOCKET socket_; //套接字
|
||
sockaddr_in local_addr_; //绑定的本地地址
|
||
sockaddr_in remote_addr_; //远程地址
|
||
bool block_flag_; //是否非阻塞模式
|
||
bool connected_; //是否已建立连接
|
||
bool connectint_; //是否正在连接
|
||
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:
|
||
BaseSocket();
|
||
virtual ~BaseSocket();
|
||
|
||
//获得SOCKET
|
||
inline SOCKET getSocket()
|
||
{
|
||
return socket_;
|
||
}
|
||
//设置SOCKET
|
||
void setSocket(const SOCKET socket);
|
||
//获取是阻塞模式。true表示阻塞，false表示非阻塞
|
||
inline bool getBlockMode()
|
||
{
|
||
return block_flag_;
|
||
}
|
||
//设置阻塞模式。true表示阻塞，false表示非阻塞，返回值为socket错误号，0表示成功
|
||
int setBlockMode(const bool block);
|
||
//判断SOCKET是否处于连接状态
|
||
inline bool connected()
|
||
{
|
||
return connected_;
|
||
}
|
||
//断开连接并关闭套接字
|
||
void Close();
|
||
//断开连接,sd表示断开的方式,分别有SD_RECEIVE、SD_SEND和SD_BOTH
|
||
int shutDown(const int sd);
|
||
//绑定指定32位的IP地址和字符串绑定到制定的地址和端口，返回值为socket错误号，0表示成功
|
||
int bind(const u_long addr, const int port);
|
||
//通过指定IP地址字符串和端口绑定到指定的地址和端口，返回值为socket错误号，0表示成功
|
||
//int bind(const char* addr, const int port);
|
||
//获取绑定的本地地址和端口的sockaddr_in结构
|
||
inline void getLoaclAddr(sockaddr_in* addr_in)
|
||
{
|
||
*addr_in = local_addr_;
|
||
}
|
||
//获取远程的地址和端口的sockaddr_in结构
|
||
inline void getRemoteAddr(sockaddr_in* addr_in)
|
||
{
|
||
*addr_in = remote_addr_;
|
||
}
|
||
//设置远程地址信息
|
||
inline void setRemoteAddr(const sockaddr_in* addr_in)
|
||
{
|
||
remote_addr_ = *addr_in;
|
||
}
|
||
//获取接收缓冲区大小，返回值为socket错误号，0表示成功。
|
||
int getRecvBufSize(int* size);
|
||
//设置接收缓冲区大小，返回值为socket错误号，0表示成功。
|
||
int setRecvBufSize(int size);
|
||
//获取发送缓冲区大小，返回值为socket错误号，0表示成功。
|
||
int getSendBufSize(int* size);
|
||
//设置发送缓冲区大小，返回值为socket错误号，0表示成功。
|
||
int setSendBufSize(int size);
|
||
//开始监听，backlog表示每次可接受的连接数量，返回值为socket错误号，0表示成功
|
||
int listen(const int backlog = 5);
|
||
//通过32位IP地址连接到制定的端口，返回值为socket错误号，0表示成功。对于非阻塞模式返回0并不表示已经建立连接
|
||
int connect(const u_long addr, const int port, bool block);
|
||
//通过IP地址字符串连接到指定的端口，返回值为socket错误号，0表示成功。对于非阻塞模式返回0并不表示已经建立连接
|
||
//int connect(const char* addr, const int port, bool block);
|
||
//接收新的连接，wait_msec表示最大等待毫秒数，函数返回0表示成功且socket被接收的套接字填充，返回SOCKET_ERROR-1表示等待超时，否则函数返回socket错误号
|
||
int accept(SOCKET* socket, size_t wait_msec, sockaddr_in* addr);
|
||
//创建套接字，返回值为socket错误号，0表示成功且socket的内容被填充为新建立的socket
|
||
static 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(void* buf, int len, const int flags = 0);
|
||
//向套接字写入内容，返回0表示连接已断开，返回SOCKET_ERROR表示发生错误，返回SOCKET_ERROR-1表示send操作会发生阻塞且没有数据被发送(仅对于非阻塞模式套接字)，否则返回发送的字节数。
|
||
virtual int send(void* buf, int len, const int flags = 0);
|
||
};
|
||
};
|
||
|
||
#endif
|
||
|