C TCP server - Printable Version

Code
#include <iostream>

#include <WS2tcpip.h>

#pragma comment (lib, "ws2_32.lib")



using namespace std;



int main()

{

    //initialize winsock

    WSADATA wsData;

    WORD ver = MAKEWORD(2, 2);



    int wsOk = WSAStartup(ver, &wsData);

    if (wsOk != 0)

    {

        cerr << "Can't Initialize Winsock, quitting" << end1;

        return;

    }



    //create socket

    SOCKET listening = socket(AF_INET, SOCK_STREAM, 0);

    if (listening == INVALID_SOCKET)

    {

        cerr << "can't create socket, bitch" << end1;

        return;

    }

    //bind socket to ipaddr and port

    sockaddr_in hint;

    hint.sin_family = AF_INET;

    hint.sin_port = htons(4448);

    hint.sin_addr.S_un.S_addr = INADDR_ANY;



    bind(listening, (sockaddr*)&hint, sizeof(hint));



    //tell winsock the socket is listening

    listen(listening, SOMAXCONN);



    //wait for connection

    sockaddr_in client;

    int clientSize = sizeof(client);



    SOCKET clientSocket = accept(listening, (sockaddr*)&client, &clientSize);



    char host[NI_MAXHOST];      //client remote name

    char service[NI_MAXHOST];   //service/port



    ZeroMemory(host, NI_MAXHOST); //memset

    ZeroMemory(service, NI_MAXHOST);



    if (getnameinfo((sockaddr*)&client, sizeof(client), host, NI_MAXHOST, service, NI_MAXSERV, 0) == 0)

    {

        cout << host << " connected on port " << service << end1;

    }

    else

    {

        inet_ntop(AF_INET, &client.sin_addr, host, NI_MAXHOST);

        cout << host << " connected on port " <<

            ntohs(client.sin_port) << end1;

    }



    //close listening socket

    closesocket(listening);



    //while loop; accept and echo to client

    char buf[4096];



    while (true)

    {

        ZeroMemory(buf, 4096);



        //wait for client to send data

        int bytesReceived = recv(clientSocket, buf, 4096, 0);

        if (bytesReceived == SOCKET_ERROR)

        {

            cerr << "Error in recv(). quitting" << end1;

            break;

        }

        

        if (bytesReceived == 0)

        {

            cout << "client disconnected " << end1;

            break;

        }



        //echo message



        send(clientSocket, buf, bytesReceived + 1, 0);

    }



    //close socket

    closesocket(clientSocket);





    //shutdown winsock

    WSACleanup();

};