long readn(SOCKET skt, char* buffer, long length)

{

    long cnt, rc;

 

    cnt = length;

 

    while(cnt > 0)

    {

        rc = recv(skt, buffer, cnt, 0);

 

        if(rc < 0)

            return -1;  

 

        if(rc == 0)

            return length - cnt;

        

        buffer += rc;

        cnt -= rc;

    }

 

    return length;

}

bool InitClient()

{

///////////////////////////////// WSAStartup ///////////////////////////////////////////

    WORD wVersionRequested;

    WSADATA wsaData;

    int err;

 

    wVersionRequested = MAKEWORD( 2, 2 );

    err = WSAStartup( wVersionRequested, &wsaData );

 

    if ( err != 0 )

    {      

        child->AddToErrorLog("could not find a usable WinSock DLL");        

        return false;

    }

 

    // Confirm that the WinSock DLL supports 2.2

    if ( LOBYTE( wsaData.wVersion ) != 2 ||HIBYTE( wsaData.wVersion ) != 2 )

    {

        child->AddToErrorLog("could not find a usable WinSock DLL");                                

        WSACleanup();      

        return false;

    }

 

/////////////////////////////////// connect to server //////////////////////////////////////////        

    struct sockaddr_in peer;

    SOCKET skt=0;

    int res;

 

    memset(&peer, 0, sizeof(peer));

    peer.sin_family = AF_INET;              

    peer.sin_port = htons(4000);

    peer.sin_addr.s_addr = inet_addr(m_ActiveServer_ip);    

    

    if( (skt = socket(AF_INET, SOCK_STREAM, 0)) < 0)

    {

        child->ShowErrorMessage(WSAGetLastError());

        return false;

    }

 

    TRACE("\nconnecting to server. port=4000 ip=%s\n", m_ActiveServer_ip);  

 

    if( (res = connect(skt, (struct sockaddr*)&peer, sizeof(peer)) ) != 0 )

    {

        child->ShowErrorMessage(WSAGetLastError());

        closesocket(skt);                  

        return false;

    }

 

 

///////////////////////////// get server connection //////////////////////////////////

    SOCKET listen_skt, read_skt;

    struct sockaddr_in local;

    local.sin_family = PF_INET;        

    local.sin_port = htons(4001);

    local.sin_addr.s_addr = 0;

 

    // create socket

    if( (listen_skt = socket( PF_INET, SOCK_STREAM, 0) ) < 0)

    {

            child->ShowErrorMessage(WSAGetLastError());

    }

 

    // bind address and port to listening socket

    if( (res = bind(listen_skt, (struct sockaddr*)&local, sizeof(local)) ) < 0)

    {

        child->ShowErrorMessage(WSAGetLastError());

    }

    

    // mark socket as listening

    if( (res = listen(listen_skt, 5)) != 0)

    {  

        child->ShowErrorMessage(WSAGetLastError());

    }

        

    if( (read_skt = accept(listen_skt, NULL, NULL)) < 0)

    {      

        child->ShowErrorMessage(WSAGetLastError());

    }

 

    m_sendsocket = skt;

    m_readsocket = read_skt;

 

      AfxBeginThread(SendFunc, NULL);

    AfxBeginThread(GetFunc, NULL);

 

      return true;

}

 

//bool InitServer() - аналогично InitClient()

 

 

UINT SendFunc(LPVOID pParam)

{

    int skt = m_sendsocket;

    long counter=0;

    long size = 0;              

    tag_message1 message;

    

    for(long i=0; i<50000; i++)

        message.param[i] = 'W';

    message.param[i] = '{text}';

 

    while(1)

    {

        size = sizeof(tag_message1);

                

        // send size

        if( send(skt, (char*)&size, sizeof(long), 0) < 0)

        {

            DWORD s = WSAGetLastError();

            break;

        }

            

        char* p = reinterpret_cast<char*>(&message);    

        

        // send structure

        if( send(skt, p, size, 0) < 0)

        {

            DWORD s = WSAGetLastError();

            break;      

        }

    }

 

    return 0;

}

 

 

 

 

UINT GetFunc(LPVOID pParam)

{

    int skt = m_readsocket;

    long counter=0;

    long ret;

    long size = 0;      

    tag_message1 message;

 

    for(long i=0; i<50000; i++)

        message.param[i] = 'W';

 

    message.param[i] = '{text}';

 

    while(1)

    {

        // get parameter length

        size = sizeof(tag_message1);//strlen(message.param);        

 

        if( recv(skt, (char*)&size, sizeof(long), 0) <=0)

        {

            DWORD s = WSAGetLastError();

            break;

        }

        

        char* buff = new char [size];

        //tag_message1 pmsg = new tag_message1;

            

        long nBytes = recv( skt, buff, size, 0);

        tag_message1 *pmsg = reinterpret_cast<tag_message1*>(buff);

            

        TRACE("\ngot\t%d", counter++);

        delete [] buff;    

    }

 

    return 0;

}

SOCKET WSASocket(

    int af,

    int type,

    int protocol,

    LPWSAPROTOCOL_INFO lpProtocolInfo,

    GROUP g,

    DWORD dwFlags);