#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <netdb.h>
#include <iostream>

#define MAX_BUF 1024

int main(int argc, char* argv[])
{
  int sockd;
  int lbajt;
  struct sockaddr_in my_name, serv_name;
  struct hostent *he;  //added to support get host by name
  char buf[MAX_BUF];
  int status;
  int addrlen;

  if (argc < 3)
  {
    fprintf(stderr, "Usage: %s ip_address port_number\n", argv[0]);
    exit(1);
  }
  /* create a socket */
  sockd = socket(AF_INET, SOCK_DGRAM, 0);
  printf("socket: %d\n", sockd);
  if (sockd == -1)
  {
    perror("Socket creation error");
    exit(1);
  }

  /* client address */
  my_name.sin_family = AF_INET;
  my_name.sin_addr.s_addr = INADDR_ANY;
  my_name.sin_port = 0;

  status = bind(sockd, (struct sockaddr*)&my_name, sizeof(my_name));
  printf("bind: %d\n", status);

    //added to translate server address to/from tcip
    if ((he=gethostbyname(argv[1]))==NULL) 
                perror("gethostbyname");

  /* server address */ 
  serv_name.sin_family = AF_INET;
  //serv_name.sin_addr.s_addr = inet_addr(argv[1]); // uses ip number to resolve hostname
  serv_name.sin_addr = *((struct in_addr *) he->h_addr); 
  serv_name.sin_port = htons(atoi(argv[2]));

  strcpy(buf, "Hello world\n");
  status = sendto(sockd, buf, strlen(buf)+1, 0, 
                 (struct sockaddr*)&serv_name, sizeof(serv_name));
  printf("sendto: %d\n", status);

  addrlen = sizeof(serv_name);
  status = recvfrom(sockd, buf, MAX_BUF, 0, 
                   (struct sockaddr*)&serv_name, &addrlen);
  printf("recvfrom: %d\n", status);
  buf[status] = 0;

  printf("\nReceived:\n%s", buf);

  close(sockd);
  return 0;
}