You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.1 KiB
65 lines
1.1 KiB
8 years ago
|
// socket.h
|
||
|
|
|
||
|
|
#ifndef SOCKET_H
|
||
|
|
#define SOCKET_H
|
||
|
|
|
||
|
|
#pragma comment(lib, "wsock32.lib")
|
||
|
|
|
||
|
|
#include <winsock.h>
|
||
|
|
|
||
|
|
|
||
|
|
class WinsockInit
|
||
|
|
{
|
||
|
|
public :
|
||
|
|
WSADATA m_wsd;
|
||
|
|
int m_iStatus;
|
||
|
|
|
||
|
|
WinsockInit(WORD wVersionRequested = 0x0101);
|
||
|
|
~WinsockInit();
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class InetAddr : public sockaddr_in
|
||
|
|
{
|
||
|
|
public :
|
||
|
|
InetAddr(WORD wPort = 0);
|
||
|
|
InetAddr(LPCSTR lpszAddress, WORD wPort = 0);
|
||
|
|
InetAddr& operator = (LPCSTR lpszAddress);
|
||
|
|
|
||
|
|
protected :
|
||
|
|
void Resolve(LPCSTR lpszAddress, WORD wPort = 0);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
class Socket
|
||
|
|
{
|
||
|
|
public :
|
||
|
|
Socket();
|
||
|
|
Socket(SOCKET s);
|
||
|
|
Socket(const Socket& s);
|
||
|
|
virtual ~Socket();
|
||
|
|
|
||
|
|
bool Create();
|
||
|
|
void Close();
|
||
|
|
bool Bind(const InetAddr& addr);
|
||
|
|
bool Connect(const InetAddr& addr);
|
||
|
|
bool Listen();
|
||
|
|
Socket Accept(InetAddr& addr);
|
||
|
|
int Send(const unsigned char* buf, int cbBuf);
|
||
|
|
int Send(const char* fmt, ...);
|
||
|
|
int Receive(unsigned char* buf, int cbBuf);
|
||
|
|
bool SetOpt(int opt, const char* pBuf, int cbBuf);
|
||
|
|
bool GetOpt(int opt, char* pBuf, int& cbBuf);
|
||
|
|
operator SOCKET& () const { return (SOCKET&)m_sock; }
|
||
|
|
operator bool() const { return m_sock != INVALID_SOCKET; }
|
||
|
|
|
||
|
|
protected :
|
||
|
|
SOCKET m_sock;
|
||
|
|
|
||
|
|
private :
|
||
|
|
bool m_bOwnSocket;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif
|