/* librairie standard ... */
#include <stdlib.h>
/* pour getopt */
#include <unistd.h>
/* déclaration des types de base */
#include <sys/types.h>
/* constantes relatives aux domaines, types et protocoles */
#include <sys/socket.h>
/* constantes et structures propres au domaine UNIX */
#include <sys/un.h>
/* constantes et structures propres au domaine INTERNET */
#include <netinet/in.h>
/* structures retournées par les fonctions de gestion de la base de
données du réseau */
#include <netdb.h>
/* pour les entrées/sorties */
#include <stdio.h>
/* pour la gestion des erreurs */
#include <errno.h>
#define BASE_SIZE 10

void construire_message(char *message, char motif, int lg, int num_message) {
  int i;
  printf("\nSOURCE : Envoi n°%d (%d) [",num_message,lg);
  char* indice_message = malloc(sizeof(char)* 5);
  _itoa(num_message,indice_message,10);

  int nb_zeros=1;

  while (num_message > 0){
	  num_message/=10;
	  nb_zeros++;
  }
  
  for (int j=0; j< 5-nb_zeros; j++){
	  
	  message[j]='-';
  }
  
  int l=0;
  for (int k=5-nb_zeros;k<5;k++){
	message[k]=indice_message[l];
	l++;
  }

  for (i=0;i<lg;i++) message[i] = motif;
  strcpy(message,indice_message);
  printf("]\n");
}

void afficher_message_source(char *message, int lg, int num_message) {
  int i;
  printf("\nPUIT : Reception n°%d (%d) [",num_message+1,lg);
  for (i=0;i<lg;i++){
	  printf("%c", message[i]);
  }
  printf("]\n");
}


int main (int argc, char **argv)
{
	int c;
	extern char *optarg;
	extern int optind;
	int nb_message = -1; /* Nb de messages à envoyer ou à recevoir, par défaut : 10 en émission, infini en réception */
	int source = -1 ; /* 0=puits, 1=source */
	int udp=0; /* 0=TCP, 1=UDP */
	
	while ((c = getopt(argc, argv, "pn:su")) != -1) {
		switch (c) {
		case 'p':
			if (source == 1) {
				printf("usage: cmd [-p|-s][-n ##]\n");
				exit(1);
			}
			source = 0;
			break;

		case 's':
			if (source == 0) {
				printf("usage: cmd [-p|-s][-n ##]\n");
				exit(1) ;
			}
			source = 1;
			break;

		case 'n':
			nb_message = atoi(optarg);
			break;

		case 'u':
		  udp=1;
		  break;

		default:
			printf("usage: cmd [-p|-s][-n ##]\n");
			break;
		}
	}
	
	char* nom_machine_distante;

	// Recuperation du port
	int port=atoi(argv[argc-1]);
	port = htons(port);

	char* message = malloc(BASE_SIZE * sizeof(char));

	if (source == -1) {
		printf("usage: cmd [-p|-s][-n ##]\n");
		exit(1) ;
	}

	if (source == 1) {
		printf("on est dans le source\n");

		//Recuperation du nom logique
		nom_machine_distante=argv[argc-2];
	}
	
	else
		printf("on est dans le puits\n");

	if (nb_message != -1) {
		if (source == 1)
			printf("nb de tampons à envoyer : %d\n", nb_message);
		else
			printf("nb de tampons à recevoir : %d\n", nb_message);
	} else {
		if (source == 1) {
			nb_message = 10 ;
			printf("nb de tampons à envoyer = 10 par défaut\n");
		} else
		printf("nb de tampons à envoyer = infini\n");

	}

	if (udp==1){
	    // Creation du socket local
	    int sock= socket(AF_INET,SOCK_DGRAM,0);

	  if (source==1) {



	    // Creation de l'adresse du socket distant
	    struct hostent *hp ;
	    struct sockaddr_in adr_dest;
	    int longueur_adr_dest = sizeof(adr_dest);

	    int longueur_message = BASE_SIZE;
	    
	    memset((char *)& adr_dest, 0, sizeof(adr_dest)) ;
	    
	    adr_dest.sin_family=AF_INET;
	    adr_dest.sin_port=port;


	    if ((hp = gethostbyname(nom_machine_distante)) == NULL) {
	      printf("erreur gethostbyname\n") ;
	      exit(1) ;
	    }

	    memcpy( (char*)&(adr_dest.sin_addr.s_addr),
		    hp->h_addr,
		    hp->h_length ) ;
	    
	    for (int i = 0; i < nb_message; i++) {
		    // Construction du message
		    construire_message(message, 'a' + (i % 26), BASE_SIZE,i);

		    // Envoi du message

		    sendto(sock,message,longueur_message,0,(struct sockaddr*)&adr_dest,longueur_adr_dest);
	    }

	    exit(0);
	    
	  }

	  else {


	    // Creation de l'adresse du socket distant
	    struct sockaddr_in adr_locale;
	    int longueur_adr_locale = sizeof(adr_locale);

	    int longueur_message = BASE_SIZE;
	    
	    memset((char *)& adr_locale, 0, sizeof(adr_locale)) ;
	    
	    adr_locale.sin_family=AF_INET;
	    adr_locale.sin_port=port;
	    adr_locale.sin_addr.s_addr = INADDR_ANY;

	    // Bind the local socket to any local address (ie any available interface)
	    if (bind(sock, (struct sockaddr *) &adr_locale, longueur_adr_locale) == -1) {
		    printf("failed to bind\n");
		    exit(1);
	    }

	    int i=0;
	    while (1) {
		    // Receive a single message because we are lazy
		    recvfrom(sock, message, longueur_message, 0, NULL, NULL);

		    // Afficher notre seule et unique triste message
		    afficher_message_source(message, longueur_message,i);
		    i++;
	    }

	    exit(0);


	    
	  }
        }
}