一个封装的socket类

 

#include "stdafx.h"
#include "libyfun.h"
#include "mytcp.h"
//#include "ty_server.h"

//构造函数
myCTcp::myCTcp ()
{
    m_timeout = 3000;
    m_nSock = -1;
    WSADATA sockData;
    if (WSAStartup (MAKEWORD (2, 2), &sockData))
    {
        printf("TCP加载失败\n");
        exit (0);
    }
}

//析构函数
myCTcp::~myCTcp ()
{
    Close ();
    WSACleanup ();
}

/*赋值
  入参：nSockfd - socket句柄
  出参：赋值后的socket句柄
*/
int myCTcp::operator = (int nSockfd)
{
    assert ((m_nSock == -1) && (nSockfd > -1));
    m_nSock = nSockfd;

    return m_nSock;
}

/*取出socket句柄
  入参：无
  出参：取出的socket句柄
*/
int myCTcp::GetHandle () const
{
    return m_nSock;
}

/*创建socket
  入参：无
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::Open ()
{
    assert (m_nSock == -1);

    if(m_nSock != -1)
        Close ();

    m_nSock = socket (AF_INET, SOCK_STREAM, 0);

    setsockopt (m_nSock, SOL_SOCKET, SO_SNDTIMEO, (char *)&m_timeout, sizeof(m_timeout));

    setsockopt (m_nSock, SOL_SOCKET, SO_RCVTIMEO, (char *)&m_timeout, sizeof(m_timeout));


    return (m_nSock != -1);
}

/*关闭socket
  入参：无
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::Close ()
{
    if (m_nSock != -1)
    {
        if(closesocket (m_nSock) != 0)
        {
            int err = WSAGetLastError();
//            DEBUG (gbl.is_debug, gbl.debug_level, gbl.log_file, "WSAGetLastError %d", err);
        }

        m_nSock = -1;    
    }
    return 1;
}

/*连接
  入参：pHost - IP地址或主机名
      nPort - 端口
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::Connect (const char *pHost, int nPort) const
{
    assert ((m_nSock != -1) && pHost && (nPort > 0));
    struct sockaddr_in addr;
    struct hostent *phe = NULL;

    memset (&addr, 0, sizeof (addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons (nPort);

    if ((addr.sin_addr.s_addr = inet_addr (pHost)) == (unsigned long int) -1)
    {
        if ((phe = gethostbyname (pHost)) == NULL)
            return 0;

        memcpy ((char *)&addr.sin_addr, phe->h_addr, phe->h_length);
    }

    return (connect (m_nSock, (struct sockaddr *)&addr, sizeof (addr)) == 0);
}

/*绑定
  入参：pIP - IP地址
      nPort - 端口
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::Bind (const char *pIP, int nPort) const
{
    assert ((m_nSock != -1) && (nPort > 0));
    struct sockaddr_in addr;
    struct hostent *phe = NULL;
    int opt=1;

    if (setsockopt (m_nSock, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt, sizeof(int)) == -1)
    {    
        return 0;
    }

    memset (&addr, 0, sizeof (addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons (nPort);

    if (!pIP)
    {
        addr.sin_addr.s_addr = htonl (INADDR_ANY);
    }
    else
    {
        if ((addr.sin_addr.s_addr = inet_addr (pIP)) == (unsigned long int) -1)
        {
            if ((phe = gethostbyname (pIP)) == NULL)
                return 0;

            memcpy ((char *)&addr.sin_addr, phe->h_addr, phe->h_length);
        }
    }

    return (bind (m_nSock, (struct sockaddr *)&addr, sizeof (addr)) == 0);
}

/*监听
  入参：nNum - 监听数目
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::Listen (int nNum) const
{
    assert ((m_nSock != -1) && (nNum > 0));

    return (listen (m_nSock, nNum) == 0);
}

/*接受连接
  入参：无
  出参：其他: 连接套接字句柄 ; -1: 失败
*/
int myCTcp::Accept () const
{
    assert (m_nSock != -1);

    return (accept (m_nSock, (struct sockaddr *)NULL, NULL));
}

/*发送指定数目数据
  入参：pBuf - 发送缓存
      nCount - 需发送字节数
      nMicsec - socket超时值，单位：微妙，缺省：500000微妙
  出参：实际发送字节数 ，如果发送失败，返回负数
*/
int myCTcp::Sendn (const void *pBuf, int nCount, int nMicsec) const
{
    assert ((m_nSock != -1) && pBuf && (nCount > 0));
    int nLeft = nCount, nWritten = 0;
    const char *ptr = (const char *)pBuf;
    
    while (nLeft > 0)
    {
        if ((nWritten = send (m_nSock, ptr, nLeft, 0)) <= 0)
        {
            return MY_SOCK_ERROR;
        }

        nLeft -= nWritten;
        ptr += nWritten;
    }

    return nCount;
}

/*接收指定数目数据
  入参：pBuf - 接收缓存
      nCount - 需接收字节数
      nMicsec - socket超时值，单位：微妙，缺省：500000微妙
  出参：实际接收字节数 ，如果接收失败，返回负数，如果对方关闭，返回0
*/
int myCTcp::Recvn (void *pBuf, int nCount, int nMicsec) const
{
     assert ((m_nSock != -1) && pBuf && (nCount > 0));
    int nLeft = nCount, nRead = 0;
    char *ptr = (char *)pBuf;

    while (nLeft > 0)
    {
        if ((nRead = recv (m_nSock, ptr, nLeft, 0)) < 0)
        {
            return MY_SOCK_ERROR;
        }
        else if (nRead == 0)
        {
            return 0;
        }

        nLeft -= nRead;
        ptr += nRead;
    }

    return nCount;
}

/*设置地址可重用
  入参：无
  出参：1: 成功 ; 0: 失败
*/
int myCTcp::SetReuseAddr ()
{
    assert (m_nSock != -1);
    int opt = 1;
    
    if (setsockopt (m_nSock, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt, sizeof(int)) == -1)
    {
        return 0;
    }
    
    return 1;
}