#include <sys/types.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <netdb.h>

#define BUF_SIZE 500
#define MAX_CON 5                            /* nombre maximum de connexions */

struct addrinfo *result;                     /* tableau des adresses réseaux des clients */

void cree_socket(struct addrinfo hints, char* port){
  int s = getaddrinfo(NULL, port, &hints, &result);
  if (s != 0) {
    fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(s));
    exit(EXIT_FAILURE);
  }
}

void con_client(int *sfd){
  struct addrinfo *rp;                       /* structure de l'adresse du client */
  /* getaddrinfo () retourne une liste de structures d'adresses.
     Essayez chaque adresse jusqu'à ce que nous ayons réussi à bind(2).
     Si socket() (ou bind()) échoue, nous (fermons le socket et)
     essayons l'adresse suivante */

  for (rp = result; rp != NULL; rp = rp->ai_next) {
    /* socket crée un point de communication et renvoie un descripteur */
    *sfd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
    if (*sfd == -1){
      continue;
    }
    if (bind(*sfd, rp->ai_addr, rp->ai_addrlen) == 0){
      break;                                  /* Succès */
    }
    close(*sfd);
  }
  if (rp == NULL) {                           /* Aucune adresse n'a pu marcher */
    fprintf(stderr, "Could not connect\n");
    exit(EXIT_FAILURE);
  }
  if (listen(*sfd, MAX_CON) == -1){
    fprintf(stderr, "erreur du listen");
    exit(EXIT_FAILURE);
  }
  freeaddrinfo(result);                       /* libère la structure d'adresse client */
}

void traite_connexion(int *sfd){
  ssize_t nread;
  char buf[BUF_SIZE];
  struct sockaddr_storage peer_addr;
  socklen_t peer_addr_len;
  int s, new_sfd;
  char host[NI_MAXHOST], service[NI_MAXSERV];

  /* Lit les messages et les renvoi au client */
  for (;;) {
    peer_addr_len = sizeof(struct sockaddr_storage);
    new_sfd = accept(*sfd, (struct sockaddr*) &peer_addr, &peer_addr_len);   /* accepte la connexion */
    nread = read(new_sfd, buf, BUF_SIZE);                                    /* lit dans le buffer de la connexion, renvoie la taille du message */
    if (nread == -1){
      continue;                                                              /* ignore la mauvaise requete */
    }
    s = getnameinfo((struct sockaddr *) &peer_addr, peer_addr_len, host, NI_MAXHOST, service, NI_MAXSERV, NI_NUMERICSERV);
    if (s == 0){
      printf("Recu %ld octets de %s:%s\n", (long) nread, host, service);
      if (write(new_sfd, buf, nread) != nread){
        fprintf(stderr, "Erreur d'envoi de la reponse\n");
      }
      close(new_sfd);
    }
    else{
      fprintf(stderr, "getnameinfo: %s\n", gai_strerror(s));
    }
  }
}

int main(int argc, char *argv[]){
  struct addrinfo hints;
  int s;                                       /* s code de retour de getaddrinfo */
  int sfd;                                     /* code de retour de l'ouverture du socket */

  if (argc != 2) {                             /* test du nombre d'arguments entrés */
    fprintf(stderr, "Usage: %s port\n", argv[0]);
    exit(EXIT_FAILURE);
  }

  char* port = argv[1];                        /* récupération du port passée en paramètre du programme */

  /* remplissage de la structure hints contenants la configuration réseau du serveur */
  memset(&hints, 0, sizeof(struct addrinfo));
  hints.ai_family = AF_UNSPEC;                 /* utilisation du serveur sur IPv4/v6 (AF_INET/AF_INET6) */
  hints.ai_socktype = SOCK_STREAM;             /* socket en mode datagramme */
  hints.ai_protocol = IPPROTO_TCP;             /* protocol de connexion */

  cree_socket(hints, port);                    /* création du socket */

  con_client(&sfd);                            /* connexion au client */

  traite_connexion(&sfd);
  printf("traite_connexion ok\n");
}