/*
 * tools
 * Creation et manipulation de sockets
 * Thomas Nemeth -- le 15 juin 2001
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>
#include <strings.h>
#include <arpa/inet.h>
#include "defines.h"
#include "xmem.h"
#include "tools.h"

/*
 * Ouverture d'une socket de type flot de donnees
 *
 * creation d'une socket serveur : port != 0
 *
 */
int ouvre_socket (int port, char *adr_ip, struct sockaddr_in *ptr_adresse) {
    int                desc_soc;
    int                lg_adresse = sizeof (struct sockaddr_in);
    int                lg_linger  = sizeof (struct linger);
    struct sockaddr_in adresse;
    struct linger      ling;

    /*
     * Creation de la socket
     *
     */
    if ( (desc_soc = socket (AF_INET, SOCK_STREAM, 0) ) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Ne peut pas ouvrir la socket de connexion.\n");
#endif
        perror (__FILE__ " socket");
        return -1;
    }

    /*
     * Affectation des options
     *
     */
    if (port != 0) {
        int opt = 1;
        if (setsockopt (desc_soc, SOL_SOCKET,
                        SO_REUSEADDR, &opt, sizeof (int) ) == -1) {
#ifdef DEBUG
            fprintf (stderr, "Impossible de positionner REUSEADDR.\n");
#endif
            perror (__FILE__ " setsockopt -- REUSEADDR");
            return -1;
        }
    }
    ling.l_onoff  = 0;
    ling.l_linger = 0;
    if (setsockopt (desc_soc, SOL_SOCKET, SO_LINGER, &ling, lg_linger) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Impossible de positionner LINGER.\n");
#endif
        perror (__FILE__ " setsockopt -- LINGER");
        return -1;
    }

    /*
     * Preparation de l'adresse d'attachement
     *
     */
    memset (&adresse, 0, sizeof (adresse) );
    adresse.sin_family = AF_INET;
    if (adr_ip) {
        struct in_addr ia;

        if (inet_aton(adr_ip, &ia)) {
            adresse.sin_addr = ia;
        } else {
            struct hostent *h;

            h = gethostbyname(adr_ip);
            if (!h) {
                fprintf (stderr, "Adresse %s inconnue.\n", adr_ip);
                return -1;
            }
            memcpy (&adresse.sin_addr.s_addr, h->h_addr, h->h_length);
        }
    } else
        adresse.sin_addr.s_addr = htonl (INADDR_ANY);
    adresse.sin_port        = htons (port);

    /*
     * Demande d'attachement de la socket
     *
     */
    if (bind (desc_soc, (struct sockaddr*) &adresse, lg_adresse) == -1) {
#ifdef DEBUG
        fprintf (stderr,
                 "Ne peut donner de nom a la socket de connexion.\n"
                 "Attendez quelques minutes avant de relancer.\n");
#endif
        perror (__FILE__ " bind");
        close (desc_soc);
        return -1;
    }

    /*
     * Recuperation de l'adresse effective d'attachement
     *
     */
    if (ptr_adresse != NULL)
        getsockname (desc_soc, (struct sockaddr *) ptr_adresse, &lg_adresse);

  return desc_soc;
}


int connect_server (const char *machine, int port) {
    struct hostent     *serveur;
    struct sockaddr_in  adresse_serveur, adresse_client;
    int                 socket_client;

    /* Recherche de la machine ou se trouve le serveur */
    if ( (serveur = gethostbyname (machine) ) == NULL) {
#ifdef DEBUG
        fprintf (stderr, "Machine %s inconnue...\n", machine);
#endif
        return -1;
    }

    /* Creation et attachement de la socket du client --
     * port quelconque => 0
     */
    if ( (socket_client = ouvre_socket
          (0, NULL, (struct sockaddr_in *) &adresse_client)) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Creation de la socket du client impossible.\n");
#endif
        return -1;
    }

    /* Preparation de l'adresse du serveur */
    adresse_serveur.sin_family = AF_INET;
    adresse_serveur.sin_port   = htons (port);
    memcpy (&adresse_serveur.sin_addr.s_addr,
            serveur->h_addr, serveur->h_length);

    /* Demande de connexion au serveur */
    if (connect (socket_client, (struct sockaddr *) &adresse_serveur,
                 sizeof (adresse_serveur) ) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Impossible de se connecter au serveur.\n");
#endif
        perror (__FILE__ " connect");
        return -1;
    }

    return socket_client;
}


int install_server (int port, char *adr_ip, struct sockaddr_in *ecoute) {
    int socket_ecoute;

    /* Ouverture de la socket */
    if ( (socket_ecoute = ouvre_socket (port, adr_ip, ecoute)) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Creation de la socket du serveur impossible.\n");
#endif
        return -1;
    }

    /* Declaration d'ouverture du service */
    if (listen (socket_ecoute, 10) == -1) {
#ifdef DEBUG
        fprintf (stderr, "Impossible d'enregistrer le service.\n");
#endif
        perror (__FILE__ " listen");
        return -1;
    }

    return socket_ecoute;
}


/*
 * Fonction de reception de donnees
 *
 */
int read_infos (int socket_data, char *infos) {
    int  nb;

    /* SSIZE_MAX = 2147483647 */
    memset (infos, 0, MAXSTRLEN);
    nb = read (socket_data, infos, MAXSTRLEN - 1);

    if ( (nb == -1) || (nb == 0) ) {
#ifdef DEBUG
        fprintf (stderr, "Erreur de connexion reception !\n");
#endif
        if ( (errno == EPIPE) || (nb == 0) ) return FALSE; /* cont = FALSE; */
    }

#ifdef DEBUG
    printf ("Recu : [%s] lg = %d/%d\n", infos, strlen (infos), nb);
#endif

    return TRUE;
}


/*
 * Fonction d'emisson de donnees
 *
 */
int send_infos (int socket_data, const char *infos) {

#ifdef DEBUG
    printf ("Emission de : %s [%d]\n", infos, strlen (infos));
#endif

    if (write (socket_data, (void *) infos, strlen (infos) + 1) == -1) {
        /* Erreur : plus de socket ! */
#ifdef DEBUG
        fprintf (stderr, "Erreur de connexion emission !\n");
#endif
        if (errno == EPIPE) return FALSE; /*cont = FALSE;*/
    }
#ifdef DEBUG
        fprintf (stderr, "Emission ok !\n");
#endif
    return TRUE;
}


void get_sock_infos (int sock, char **nom, char **adr, int *prl, int *prd) {
    struct sockaddr_in  name;
#ifdef OpenBSD
    int                 namelen;
#else
    socklen_t           namelen;
#endif
    struct hostent     *machine;

    /* Initialisations en cas d'echecs */
    *nom = NULL;
    *adr = NULL;
    *prd = 0;

    /* Infos locales (port) */
    namelen = sizeof (name);
    if (getsockname (sock, (struct sockaddr *) &name, &namelen) == 0)
        *prl = ntohs (name.sin_port);
    else {
        perror ("getsockname() ");
        *prl = 0;
    }

    /* Port et adresse IP distants */
    namelen = sizeof (name);
    if (getpeername (sock, (struct sockaddr *) &name, &namelen) != 0) {
#ifdef DEBUG
        perror ("Erreur de recuperation d'information sur pair !\n"
                "getpeername() ");
#else
        perror ("Impossible d'obtenir les infos distantes ");
#endif
        return;
    }

    *adr = xstrdup (inet_ntoa (name.sin_addr));
    *prd = ntohs (name.sin_port);

    /* Nom de machine distante */
    machine = gethostbyaddr ((char *)&name.sin_addr,
                             sizeof name.sin_addr, name.sin_family);
    if (machine == NULL) {
#ifdef DEBUG
        perror ("Erreur de recuperation de nom de pair.\n"
                "gethostbyaddr() ");
        fprintf (stderr, "Adresse : %s (%s)\tFamille : %d (2 == AF_INET)\n",
                 *adr, (char *)&name.sin_addr, name.sin_family);
#else
        perror ("Impossible d'obtenir le nom d'hote distant ");
#endif
        return;
    }

    *nom = xstrdup (machine->h_name);
}