sh_tcp.cpp

来自「rsa算法打的一个包」· C++ 代码 · 共 86 行

// SH_Tcp.cpp: implementation of the SH_Tcp class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "SH_Tcp.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

SH_Tcp::SH_Tcp()
{

}

SH_Tcp::~SH_Tcp()
{

}
BOOL SH_Tcp::CreateReceiver(LPCTSTR cszIP,const UINT uPort)
{
    LONG opt = 1;
    if(Open(AF_INET,SOCK_DGRAM,0))
    {
        if(Bind(cszIP,uPort))
        {
            if(SetSockOpt(SOL_SOCKET,SO_REUSEADDR,(char*)&opt,sizeof(opt)))
            {
                if(Listen() && SelectEx(FD_ACCEPT))
                    return TRUE;
            }
        }
    }
    Close();
    return FALSE;
}

BOOL SH_Tcp::CreateReceiver(const DWORD dwIP,const UINT uPort)
{
    LONG opt = 1;
    if(Open(AF_INET,SOCK_DGRAM,0))
    {
        if(Bind(dwIP,uPort))
        {
            if(SetSockOpt(SOL_SOCKET,SO_REUSEADDR,(char*)&opt,sizeof(opt)))
            {
                if(Listen() && SelectEx(FD_ACCEPT))
                    return TRUE;
            }
        }
    }
    Close();
    return FALSE;
}

BOOL SH_Tcp::CreateReceiver(const struct sockaddr_in * addr)
{
    LONG opt = 1;

    if(Open(AF_INET,SOCK_DGRAM,0))
    {
        if(Bind(addr))
        {
            if(SetSockOpt(SOL_SOCKET,SO_REUSEADDR,(char*)&opt,sizeof(opt)))
            {
                if(Listen() && SelectEx(FD_ACCEPT))
                    return TRUE;
            }
        }
    }
    Close();
    return FALSE;
}

BOOL SH_Tcp::CreateSender()
{
    return Open(AF_INET,SOCK_DGRAM,0);
}