cocos2dx Socket (一)

#ifndef __CLIENT_SOCKET__H__
#define __CLIENT_SOCKET__H__

#ifdef WIN32
#include <winsock.h>
#else
#include <sys/socket.h>
#include <sys/errno.h>
#include <fcntl.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <netdb.h>
#endif

enum{
EVENT_CONNECT_ERROR = -10,/*buf = NULL len = 0*/
EVENT_READ_ERROR,
EVENT_ERROR = -1,/*buf = NULL len = 0*/
EVENT_PROGRESS = 0,/*socket阻塞状态*/
EVENT_CONNECTED,/*buf = NULL len = 0*/
EVENT_READ,/*buf = 收到数据类型 len = 收到数据长度*/
};

//event: 消息类型
//buf: 数据
//len: 长度
typedef void (*MY_SOCKET_CALLBACK)(int event,void* buf,int len);

/*
当我们以非阻塞的方式来进行连接的时候，
返回的结果如果是 -1,这并不代表这次连
接发生了错误，如果它的返回结果是 EINPROGRESS，
那么就代表连接还在进行中。 后面可以通过poll或
者select来判断socket是否可写，如果可以写，说
明连接完成了。
*/
class CMySocket
{
public:
CMySocket( char* ip, int port );
~CMySocket();

public:
int MyCreateSocket(void);//创建socket并设置为非阻塞
int MyConnect(void);
int MyRecv( void* buf, int len );
int MySend( void* buf, int len );
int MyClose(void);
void MySetSocketCallBack(MY_SOCKET_CALLBACK cb);
int MyGetSockFd(void){return m_sock_fd;}
MY_SOCKET_CALLBACK MyGetSocketCallBack(void){

return m_callback;
}
static char* InetNtoa(unsigned int  ip);
static char*  DnsToIp(char *dns);
#ifdef WIN32
static void WinSocketStart(void);
static void WinSocketClean(void);
static bool isWinSocketStart;
static int socketNum;
#endif
private:
//socket句柄
int m_sock_fd;
char m_ip[24];
int m_port;
MY_SOCKET_CALLBACK m_callback;
};

#endif

#include "ClientSocket.h"
#include  <iostream>
using namespace std;
#ifdef WIN32
#pragma comment(lib, "wsock32")
#endif
static char socketIP[30];
#ifdef WIN32
bool CMySocket::isWinSocketStart = false;
int CMySocket::socketNum = 0;
void CMySocket::WinSocketStart(void)
{
socketNum ++;
if(isWinSocketStart == false)
{
WSADATA wsaData;
//#define MAKEWORD(a,b) ((WORD) (((BYTE) (a)) | ((WORD) ((BYTE) (b))) << 8))
WORD version = MAKEWORD(2, 0);
int ret = WSAStartup(version, &wsaData);//win sock start up
isWinSocketStart = true;
}
}

void CMySocket::WinSocketClean(void)
{

if(isWinSocketStart == true)
{
if(socketNum == 1)
{
WSACleanup();
isWinSocketStart = false;
}
}
socketNum -- ;
}
#endif
char* CMySocket::InetNtoa(unsigned int  ip)
{
struct in_addr addr1;
memcpy(&addr1, &ip, 4 );
char *tem = inet_ntoa(addr1);
strcpy_s(socketIP, tem);
return socketIP;
}

char*  CMySocket::DnsToIp(char *dns)
{
#ifdef WIN32
WinSocketStart();
#endif
struct hostent *host;
host=gethostbyname(dns);
char *tem = inet_ntoa(*(struct in_addr*)(host->h_addr));
strcpy_s(socketIP, tem);
return socketIP;
}

CMySocket::CMySocket( char* ip, int port )
{
#ifdef WIN32
WinSocketStart();
#endif

m_port = port;
strcpy_s( m_ip, ip );
m_sock_fd = -1;
m_callback = NULL;
}

CMySocket::~CMySocket()
{
this->MyClose();
#ifdef WIN32
WinSocketClean();
#endif
}

int CMySocket::MyCreateSocket(void)
{
int flags, s;
int sock_fd = socket( AF_INET, SOCK_STREAM, 0 );//在进行网络连接前，需要用socket函数向系统申请一个通信端口
if( sock_fd < 0 )
{
return -1;
}
#ifdef WIN32

#else
//Set non-block
flags = fcntl (sock_fd, F_GETFL, 0);//get sock fd flags
if (flags == -1)
{
close(sock_fd);
return -1;
}
flags |= O_NONBLOCK;//set fd's flag to non-block
s = fcntl (sock_fd, F_SETFL, flags);
if (s == -1)
{
close(sock_fd);
return -1;
}
#endif
m_sock_fd = sock_fd;
return m_sock_fd;
}

int CMySocket::MyConnect(void)
{
int ret;
struct sockaddr_in clientAddr;
memset(&clientAddr, 0, sizeof(clientAddr));//置字节字符串s的前n个字节为零且包括‘\0’。
clientAddr.sin_family = AF_INET;//一般来说 AF_INET（地址族）PF_INET（协议族 ）
clientAddr.sin_port = htons(m_port);//必须要采用网络数据格式,普通数字可以用htons()函数转换成网络数据格式的数字
clientAddr.sin_addr.s_addr = inet_addr(m_ip);//ip地址
ret = connect(m_sock_fd, (sockaddr*)&clientAddr, sizeof(sockaddr_in));//建立与一个端的连接 若无错误发生 连接成功返回0
#ifdef WIN32
#else
if( ret == -1 )
{
if( errno == EINPROGRESS )
{
return EVENT_PROGRESS;
}
}
#endif
return ret;
}
int CMySocket::MyRecv( void* buf, int len )
{
return recv( m_sock_fd, (char *)buf, len, 0 );
}

int CMySocket::MySend( void* buf, int len )
{
return send( m_sock_fd, (char *)buf, len, 0 );
}

int CMySocket::MyClose(void)
{
if( m_sock_fd >= 0 )
{
#ifdef WIN32
closesocket(m_sock_fd);
#else
close( m_sock_fd );
#endif
m_sock_fd = -1;
}
return 0;
}

void CMySocket::MySetSocketCallBack(MY_SOCKET_CALLBACK cb)
{
m_callback = cb;
}