#include <afxwin.h> // MFC core and standard components
#include <afxext.h> // MFC extensions
#include <afxsock.h> // MFC socket extensions

#include <winsock.h>
//#include <windows.h>

//CSocket gensoc;

void server( int port)
{
printf("\n Server \n\n");

CSocket gensoc;
gensoc.Create ( port );

SOCKADDR_IN addr;

memset( &addr, 0, sizeof( SOCKADDR_IN ) );
addr.sin_family = AF_INET;
addr.sin_port = htons (port);

addr.sin_addr.s_addr = INADDR_ANY;


if (bind( gensoc, (SOCKADDR *)&addr , sizeof(addr)))
{
printf ("\n Bind success");
}
else
{
printf ("\n Bind fail");
}

//gensoc.Listen();


if (listen( gensoc,0) )
{
printf ("\n listening" );

}
else
{
printf ("\n socket error" );
}


CSocket newsoc;
int nlen = sizeof(addr);

//gensoc.Accept(newsoc);

if( accept (gensoc,(SOCKADDR *)&addr , &nlen))

{
printf ("\n Accept() function success" );

}
else
{
printf ("\n Accept() function failed" );
}

int nBytesRead;
char cbBuffer[2000];

nBytesRead = gensoc.Receive ( cbBuffer, sizeof ( cbBuffer ));
gensoc.close();
newsoc.close();

}

void client( char* remote_mac, int port)
{
printf("\n Client \n\n");

CSocket gensoc;
gensoc.Create ( port );

SOCKADDR_IN addr;

memset( &addr, 0, sizeof( SOCKADDR_IN ) );
addr.sin_family = AF_INET;
addr.sin_port = htons (port);

addr.sin_addr.s_addr = INADDR_ANY;


/****************************************/

addr.sin_addr.s_addr = inet_addr( remote_mac );

/*
if (addr.sin_addr.s_addr == INADDR_NONE)
{
HOSTENT *lpHostEnt;
lpHostEnt = gethostbyname( remote_mac );
if (!lpHostEnt)
{
char* lpAddr = ( char* ) inet_ntoa(*(LPIN_ADDR)*(lpHostEnt ->h_addr_list) );
addr.sin_addr.s_addr = **(int**)(lpHostEnt->h_addr_list);
}
}*/

//if (gensoc.Connect( (SOCKADDR*) &addr, sizeof( SOCKADDR_IN ) ) )
if (connect(gensoc, (SOCKADDR*) &addr, sizeof( SOCKADDR_IN ) ) )
{
printf( "\n connect was successful" );
}
else
{
printf( "\nconnect was not successful" );
}


int nBytesWritten;
static char msg[] = "Hello socket";

nBytesWritten = gensoc.Send ( msg, sizeof ( msg ));
gensoc.close();

}

void main( int argc, char* argv[] )
{

switch( argc )
{
case 2:
server( atoi( argv[1] ) );
break;
case 3:
client( argv[1], atoi( argv[2]) );
break;
default:
printf("Usage: main[RemoteMachien] port#\n");
printf("If remote machine is not provided, application will run in server mode listening to specified port number \n");
printf("If remote machine is provided, application will run in client trying to reach specified server \n");
}



}