Keskustelu: Koodit: C++: TCP serveri ja clientti

// server.cpp

#include <winsock2.h>
#include <iostream>

using std::cout;
using std::endl;

int VastOtaViesti( SOCKET, void*, size_t );
int Vastaa( SOCKET, const void*, size_t );

int main( )
{
    WSADATA ws;
    SOCKET TCP;
    BOOL opt = TRUE;

    // Otetaan käyttöön winsocket 2.2
    if( WSAStartup( MAKEWORD(2,2), &ws ) != 0 ) {
        cout << "Ei tukea winsocket 2.2:lle";
        WSACleanup();
        return 2;
    }

    // luodaan socket
    if( (TCP = socket( PF_INET, SOCK_STREAM, IPPROTO_TCP )) == INVALID_SOCKET ) {
        cout << "Ei voi luoda sockettia! Error: " << WSAGetLastError();
        WSACleanup();
        return 3;
    }

    servent* pent;
    if( (pent = getservbyname( "echo", 0 )) == 0 ) {
        cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 4;
    }

    struct sockaddr_in sa;
    sa.sin_family = AF_INET;
    sa.sin_port = pent-> s_port;
    sa.sin_addr.s_addr = htonl(INADDR_ANY); // sallitaan yhteys keneltä tahansa

    if( bind( TCP, (sockaddr*) &sa, sizeof(sockaddr_in) ) == SOCKET_ERROR ) {
        cout  << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 5;
    }

    if( setsockopt( TCP, SOL_SOCKET, SO_REUSEADDR, (char*) &opt, sizeof(BOOL) ) == SOCKET_ERROR )
    {
        cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 6;
    }
    if( listen( TCP, 0 ) == SOCKET_ERROR )
    {
        cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 7;
    }

    SOCKET TCPC;
    sockaddr_in srcaddr;

    int srcaddrlen = sizeof(sockaddr_in);
    cout << "Linkku's %sber TCP server. ", "\x81\x81";
    cout << "Odotetaan yhteytt\x84...\n";

    // Sallitaan sisään tuleva yhteys accept:lla
    if( (TCPC = accept( TCP, (sockaddr*) &srcaddr, &srcaddrlen )) == INVALID_SOCKET )
    {
        cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 8;
    }
    cout << "Yhteys avattu!\n" << "<" << inet_ntoa( srcaddr.sin_addr ) << ">" << endl;


    int iNum = 1, iPit;
    char cBuf[1024];

    while(1)
    {
        // vastaanotetaan viestit
        if( (iPit = VastOtaViesti( TCPC, cBuf, sizeof(cBuf) )) == SOCKET_ERROR ) {
            cout << endl << iNum-1 << " viesti\x84 vastaanotettu!";
            getchar();
            return 9;
        }

        cBuf[iPit] = '\0';
        cout << iNum++ << ": " << iPit << " - " << cBuf << endl;

        if( Vastaa( TCPC, cBuf, iPit ) == SOCKET_ERROR ) {
            cout << "Ei voi vastata viestiin! Error: " << WSAGetLastError();
            WSACleanup();
            getchar();
            return -1;
        }

    }

    getchar();
    // vapautetaan resurssit
    WSACleanup();
    return 0;
}

int VastOtaViesti( SOCKET S, void* cBuf, size_t Palautus )
{
    long expect;
    Palautus = recv( S, (char*) &expect, sizeof(long), 0 );
    expect = ntohl( expect );

    Palautus = recv( S, (char*) cBuf, (size_t) expect, 0 );
    return Palautus;
}

int Vastaa( SOCKET S, const void* cBuffer, size_t iSize )
{
    long len = htonl( (long) iSize );
    send( S, (const char*) &len, sizeof(long), 0 );
    send( S, (const char*) cBuffer, iSize, 0 );
    return 0;
}

// client.cpp


#include <winsock2.h>
#include <iostream>

using std::cout;
using std::endl;
using std::cin;

int Vastaus( SOCKET, void*, size_t );
int LahetaViesti( SOCKET, const void*, size_t );

int main( int ac, char* av[] )
{
    unsigned long N = 0;
    BOOL opt = TRUE;
    HOSTENT* Hostent;
    WSADATA ws;
    SOCKET TCP;

    // Otetaan käyttöön winsocket 2.2
    if(WSAStartup(MAKEWORD(2,2), &ws )!= 0 ) {
        std::cout << "Ei tukea winsocket 2.2:lle!";
        WSACleanup();
        getchar();
        return 2;
    }

    // luodaan TCP socket
    if( (TCP = socket( PF_INET, SOCK_STREAM, IPPROTO_TCP )) == INVALID_SOCKET ) {
        std::cout << "Ei voi luoda sockettia! Error: " << WSAGetLastError();
        WSACleanup();
        return 3;
    }

    struct sockaddr_in sa;
    sa.sin_family = AF_INET;
    sa.sin_port = 0; // otetaan nollasta seuraava vapaa portti
    sa.sin_addr.s_addr = htonl(INADDR_ANY);
    if( bind( TCP, (sockaddr*) &sa, sizeof(sockaddr_in) ) == SOCKET_ERROR ) {
        std::cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 4;
    }
    // Säädetään asetukset
    if( setsockopt( TCP, SOL_SOCKET, SO_REUSEADDR, (char*) &opt, sizeof(BOOL) ) == SOCKET_ERROR )
    {
        std::cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 5;
    }

    servent* dd;
    if( (dd = getservbyname( "echo", 0 )) == 0 ) {
        std::cout << "Socket error: " << WSAGetLastError();
        WSACleanup();
        return 6;
    }

    sockaddr_in srvaddr;
    srvaddr.sin_family = AF_INET;
    srvaddr.sin_port = dd-> s_port;

    if( (Hostent = gethostbyname("localhost")) != 0 )
        srvaddr.sin_addr.s_addr = *(long*) Hostent-> h_addr;
    else if( (srvaddr.sin_addr.s_addr = inet_addr( av[1] )) == INADDR_NONE ) {
        std::cout << "Socket error: " << WSAGetLastError() ;
        WSACleanup();
        return 7;
    }

    cout << "Linkku's %sber TCP client. ", "\x81\x81";
    cout << "Yhdistet\x84\x84n...\n";

    // Yritetään etsiä serveri
    if( connect( TCP, (sockaddr*) &srvaddr, sizeof(srvaddr) ) == SOCKET_ERROR ) {
        std::cout << "Ei voi yhdist\x84\x84 palvelimeen [palvelin ei ehk\x84 k\x84ynniss\x84]\nError: ";
        std::cout << WSAGetLastError();
        WSACleanup();
        getchar();
        return 8;
    }
    std::cout << "Yhdistetty!\n";
    int iKor = 3;
    while(1)
    {
        char cBuf[1024];
        cout << "\n> " ;
        cin.getline( cBuf, sizeof(cBuf));

        size_t iPit = strlen(cBuf);

        if( iPit==0 ) {
          break;
        }
        N++;

        // siirretään viesti
        if( LahetaViesti( TCP, cBuf, iPit ) == SOCKET_ERROR ) {
            cout << "Ei voi l\x84hett\x84\x84 viesti\x84! Error: " << WSAGetLastError() ;
            WSACleanup();
            getchar();
            return 9;
        }

        char cVBuf[sizeof(cBuf)] = " ";
        size_t rlen = SOCKET_ERROR;
        size_t tlen = strlen(cBuf);

        // odotetaan vastausta
        if( (rlen = Vastaus( TCP, cVBuf, sizeof(cVBuf) )) == SOCKET_ERROR )
        {
            std::cout << "Ei saatu vastausta palvelimelta! Error: " << WSAGetLastError();
            getchar();
            WSACleanup();
            return -1;
        }
        cout << cVBuf << endl;

        // tarkistetaan serverin lähettämä data
        // ei kuitenkaan mikään übervarma tarkistus
        if( rlen != tlen || memcmp( cVBuf, cBuf, rlen ) ) {
            cout << "\nViesti korruptoitunut!\n" ;
        }

    }


    std::cout << "\n\n" << N << " viesti\x84 l\x84hetetty";

    // vapautetaan resurssit
    getchar();
    WSACleanup();
    return 0;
}


int LahetaViesti( SOCKET S, const void* cBuffer, size_t iSize )
{
    long len = htonl( (long) iSize );
    send( S, (const char*) &len, sizeof(long), 0 );
    send( S, (const char*) cBuffer, iSize, 0 );
    return 0;
}

int Vastaus( SOCKET S, void* cBuffer, size_t Palautus )
{
    long expect;
    Palautus = recv( S, (char*) &expect, sizeof(long), 0 );
    expect = ntohl( expect );

    Palautus = recv( S, (char*) cBuffer, (size_t) expect, 0 );
    return Palautus;
}