// Yuri Zarechny for News2News (c) 2006 
 
#include "stdafx.h"
#include < stdlib.h >
#include < winsock2.h > // dependencies: ws2_32.lib
#include "conio.h"
 
const int iPort=1234;
 
void Fatal(LPSTR sError)
{
    printf("Fatal error: %s",sError);
    exit(EXIT_FAILURE);
}
 
SOCKET ListenSocket, DataSocket;
WSAData WSData;
sockaddr_in sAddr;
int addrsize, datasize, i;
char databuf[1024];
 
int _tmain(int argc, _TCHAR* argv[])
{
    if (WSAStartup(MAKEWORD(2,2), &WSData)>0)
    {
        sprintf(databuf,"Sockets interface not available. "
            "Socket error: #%d\n", WSAGetLastError);
        Fatal(databuf);
    }
 
    printf(databuf,"Interface found: %s",WSData.szDescription);
    ListenSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
 
    if (ListenSocket==INVALID_SOCKET)
    {
        sprintf(databuf,"Can not open listening socket. "
            "Socket error: #%d\n", WSAGetLastError);
        Fatal(databuf);
    }
 
    printf("Waiting for incoming connection...\n");
 
    ZeroMemory(&sAddr,sizeof(sAddr)); 
    sAddr.sin_family=AF_INET;
    sAddr.sin_port=htons(iPort);
 
    if (bind(ListenSocket,(SOCKADDR *) &sAddr, 
        sizeof(sAddr))==SOCKET_ERROR) 
    {
        sprintf(databuf,"Can not bind() listening socket. "
            "Socket error: #%d\n", WSAGetLastError);
 
        closesocket(ListenSocket);
        Fatal(databuf);
    } 
    else if (listen(ListenSocket,1)==SOCKET_ERROR) 
    {
        sprintf(databuf,"Can not listen() on socket. Socket error: #%d\n",
            WSAGetLastError);
 
        closesocket(ListenSocket);
        Fatal(databuf);
    }
    else
    {
        ZeroMemory(&sAddr, sizeof(sAddr));
        addrsize=sizeof(sAddr);
        DataSocket=accept(ListenSocket,(SOCKADDR *) &sAddr, &addrsize);
 
        if (DataSocket==INVALID_SOCKET)
        {
            sprintf(databuf,"Can nott accept() connection. "
                "Socket error: #%d\n", WSAGetLastError);
 
            closesocket(ListenSocket);
            Fatal(databuf);
        }
        else
        {
            getpeername(DataSocket,(SOCKADDR *) &sAddr,&addrsize);
            strcpy(databuf,inet_ntoa(sAddr.sin_addr));
            i=ntohs(sAddr.sin_port);
 
            printf("Connection established. (IP %s; Port %d)\n",databuf,i);
 
            do {
                datasize=recv(DataSocket,(char *) &databuf, 
                    sizeof(databuf),0);
                databuf[datasize]=0;
 
                if (datasize>1 || (datasize>0 && databuf[0]!=(char)"*")) 
                    printf("= %s\n",databuf);
            } while (datasize>0 && databuf[0]!=(char)"*");
 
            closesocket(DataSocket);
            printf("Connection dropped.");
        }
    }
 
    closesocket(ListenSocket);
 
    printf("\nPress Enter to close this window.\n"); 
    getch();
    return 0;
}