#include <winsock2.h>
#include <iostream.h>
#include <string.h>
#include "addr.h"


#define DEFHOST "localhost"
#define DEFPORT "5555"

int main(int argc, char* argv[])
{
   SOCKET      s;
   SOCKADDR_IN server, peer;
   int         peerlen;
   char        buf[256];

   init();

   if( argc < 2 ){
      cout<<"Default host "<<DEFHOST<<" is used"<<endl;
      cout<<"Default port "<<DEFPORT<<" is used"<<endl;
      set_address(DEFHOST, DEFPORT, "udp", &server);
   }
   else if( argc < 3 ){
      cout<<"Host "<<argv[1]<<" is used\n";
      cout<<"Default port "<<DEFPORT<<" is used"<<endl;
      set_address(argv[1], DEFPORT, "udp", &server);
   }
   else{
      cout<<"Port "<<argv[2]<<" is used"<<endl;
      set_address(argv[1], argv[2], "udp", &server);
   }
   print_address(server);

   if( (s=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == INVALID_SOCKET ){
      cerr<<"failed to create socket\n";
      return 2;
   }

   int i = 3;
   while(i){
      cout<<i<<"# ";
      cin>>buf;
      peerlen = sizeof(server);
      if( sendto(s, buf, sizeof(buf), 0, (SOCKADDR*)&server, peerlen) < 0 ){
         cerr<<"Error writing into socket"<<endl;
         return 5;
      }
      cout<<"S: ";
      if( recvfrom(s, buf, sizeof(buf), 0, (SOCKADDR*)&peer, &peerlen) < 0 ){
         cerr<<"Error reading from socket"<<endl;
         return 4;
      }
      cout<<"Received: "<<buf<<endl;
      i--;
   }

   return 0;
}
//---------------------------------------------------------------------------