From a4f2f3ec4b4cf7b894d6976c884fbc446396cd00 Mon Sep 17 00:00:00 2001 From: Cyrille Bagard Date: Mon, 11 Apr 2016 23:46:03 +0200 Subject: Distinguished the internal server from the remote one using Unix sockets. --- ChangeLog | 18 ++++ src/analysis/binary.c | 4 +- src/analysis/db/cdb.c | 30 +++---- src/analysis/db/client.c | 164 +++++++++++++++++++++++++++++----- src/analysis/db/client.h | 7 +- src/analysis/db/server.c | 226 +++++++++++++++++++++++++++++++++-------------- src/analysis/db/server.h | 7 +- src/main.c | 2 +- 8 files changed, 342 insertions(+), 116 deletions(-) diff --git a/ChangeLog b/ChangeLog index 5822754..8f0fc98 100644 --- a/ChangeLog +++ b/ChangeLog @@ -1,3 +1,21 @@ +16-04-11 Cyrille Bagard + + * src/analysis/binary.c: + * src/analysis/db/cdb.c: + Update code. + + * src/analysis/db/client.c: + * src/analysis/db/client.h: + Distinguish the internal server from the remote one using Unix sockets. + + * src/analysis/db/server.c: + * src/analysis/db/server.h: + Distinguish the internal server from the remote one using Unix sockets. + Delete the useless cdb_client structure. + + * src/main.c: + Update code. + 16-04-10 Cyrille Bagard * src/analysis/binary.c: diff --git a/src/analysis/binary.c b/src/analysis/binary.c index c71bc84..c422e29 100644 --- a/src/analysis/binary.c +++ b/src/analysis/binary.c @@ -1047,7 +1047,7 @@ static bool g_loaded_binary_connect_internal(GLoadedBinary *binary) /* ... */; - result = g_db_client_start(binary->local, host, port); + result = g_db_client_start_internal(binary->local); printf("DB status :: %d\n", result); @@ -1123,7 +1123,7 @@ static bool g_loaded_binary_connect_remote(GLoadedBinary *binary) /* ... */; - result = g_db_client_start(binary->local, host, port); + result = g_db_client_start_remote(binary->local, host, port); printf("DB status :: %d\n", result); diff --git a/src/analysis/db/cdb.c b/src/analysis/db/cdb.c index a42f838..e6a7005 100644 --- a/src/analysis/db/cdb.c +++ b/src/analysis/db/cdb.c @@ -49,7 +49,6 @@ #include "../../common/extstr.h" #include "../../common/io.h" #include "../../common/pathname.h" -#include "../../common/xdg.h" #include "../../common/xml.h" #include "../../core/collections.h" @@ -256,6 +255,7 @@ static void g_cdb_archive_finalize(GCdbArchive *archive) { //void close_xml_file(xmlDocPtr, xmlXPathContextPtr); + free(archive->filename); G_OBJECT_CLASS(g_cdb_archive_parent_class)->finalize(G_OBJECT(archive)); @@ -265,10 +265,10 @@ static void g_cdb_archive_finalize(GCdbArchive *archive) /****************************************************************************** * * -* Paramètres : owner = description humaine du serveur d'accueil. * -* hash = empreinte du binaire à représenter. * -* user = désignation d'un éventuel nouveau créateur. * -* error = indication éventuelle en cas d'échec. [OUT] * +* Paramètres : basedir = répertoire de stockage des enregistrements. * +* hash = empreinte du binaire à représenter. * +* user = désignation d'un éventuel nouveau créateur. * +* error = indication éventuelle en cas d'échec. [OUT] * * * * Description : Définit ou ouvre une archive d'éléments utilisateur. * * * @@ -279,10 +279,9 @@ static void g_cdb_archive_finalize(GCdbArchive *archive) * * ******************************************************************************/ -GCdbArchive *g_cdb_archive_new(const char *owner, const rle_string *hash, const rle_string *user, DBError *error) +GCdbArchive *g_cdb_archive_new(const char *basedir, const rle_string *hash, const rle_string *user, DBError *error) { GCdbArchive *result; /* Adresse à retourner */ - char *suffix; /* Fin du nom de fichier */ struct stat finfo; /* Information sur l'archive */ int ret; /* Retour d'un appel */ @@ -292,17 +291,10 @@ GCdbArchive *g_cdb_archive_new(const char *owner, const rle_string *hash, const /* Chemin de l'archive */ - suffix = strdup("chrysalide" G_DIR_SEPARATOR_S); - suffix = stradd(suffix, owner); - suffix = stradd(suffix, G_DIR_SEPARATOR_S); - suffix = stradd(suffix, hash->data); - suffix = stradd(suffix, ".tar.xz"); - - result->filename = get_xdg_config_dir(suffix); - - printf("dealing with '%s'...\n", result->filename); - - free(suffix); + result->filename = strdup(basedir); + result->filename = stradd(result->filename, G_DIR_SEPARATOR_S); + result->filename = stradd(result->filename, hash->data); + result->filename = stradd(result->filename, ".tar.xz"); if (!mkpath(result->filename)) goto gcan_error; @@ -894,8 +886,6 @@ static void *g_cdb_archive_process(GCdbArchive *archive) /* Lancement d'une phase de surveillance */ - printf("(%p) POLL %d\n", archive, nfds); - ret = poll(fds, nfds, -1); if (ret == -1) { diff --git a/src/analysis/db/client.c b/src/analysis/db/client.c index bb900ed..0598e32 100644 --- a/src/analysis/db/client.c +++ b/src/analysis/db/client.c @@ -30,6 +30,7 @@ #include #include #include +#include #include @@ -39,6 +40,7 @@ #include "protocol.h" #include "misc/rlestr.h" #include "../../common/io.h" +#include "../../common/xdg.h" #include "../../gui/panels/log.h" @@ -80,6 +82,9 @@ static void g_db_client_init(GDbClient *); /* Procède à la libération totale de la mémoire. */ static void g_db_client_finalize(GDbClient *); +/* Démarre réellement la connexion à la base de données. */ +static bool g_db_client_start_common(GDbClient *, const char *); + /* Assure l'accueil des nouvelles mises à jour. */ static void *g_db_client_update(GDbClient *); @@ -193,10 +198,83 @@ GDbClient *g_db_client_new(char *author, char *kfile, const char *name, const ch /****************************************************************************** * * * Paramètres : client = client pour les accès distants à manipuler. * +* * +* Description : Démarre la connexion à la base de données interne. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_db_client_start_internal(GDbClient *client) +{ + char *suffix; /* Suffixe pour un fichier */ + char *sock_path; /* Chemin vers le canal UNIX */ + struct sockaddr_un addr; /* Adresse de transmission */ + int ret; /* Bilan d'un appel */ + char *desc; /* Description du serveur ciblé*/ + bool status; /* Bilan de la connexion */ + + /* Identification du serveur à contacter */ + + asprintf(&suffix, "chrysalide" G_DIR_SEPARATOR_S ".internal_server.%d", getpid()); + sock_path = get_xdg_config_dir(suffix); + free(suffix); + + memset(&addr, 0, sizeof(struct sockaddr_un)); + +#define UNIX_PATH_MAX 108 + + addr.sun_family = AF_UNIX; + strncpy(addr.sun_path, sock_path, UNIX_PATH_MAX - 1); + + free(sock_path); + + /* Création d'un canal de communication */ + + client->fd = socket(AF_UNIX, SOCK_STREAM, 0); + if (client->fd == -1) + { + perror("socket"); + return false; + } + + ret = connect(client->fd, (struct sockaddr *)&addr, sizeof(struct sockaddr_un)); + if (ret == -1) + { + perror("connect"); + goto gdcsi_no_listening; + } + + asprintf(&desc, "unix://.internal_server.%d", getpid()); + + status = g_db_client_start_common(client, desc); + + free(desc); + + if (!status) + goto gdcsi_no_listening; + + return true; + + gdcsi_no_listening: + + close(client->fd); + client->fd = -1; + + return false; + +} + + +/****************************************************************************** +* * +* Paramètres : client = client pour les accès distants à manipuler. * * host = hôte à représenter pour le service. * * port = port de connexion pour les clients. * * * -* Description : Démarre la connexion à la base de données. * +* Description : Démarre la connexion à la base de données distante. * * * * Retour : Bilan de l'opération. * * * @@ -204,18 +282,13 @@ GDbClient *g_db_client_new(char *author, char *kfile, const char *name, const ch * * ******************************************************************************/ -bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) +bool g_db_client_start_remote(GDbClient *client, const char *host, unsigned short port) { struct hostent *hp; /* Informations sur l'hôte */ struct sockaddr_in addr; /* Adresse de transmission */ int ret; /* Bilan d'un appel */ - rle_string user; /* Nom d'utilisateur associé */ - GChecksum *checksum; /* Empreinte MD5 à signer */ - unsigned char md5_digest[16]; /* Empreinte MD5 calculée */ - RSA *key; /* Clef pour la signature */ - unsigned char sig[RSA_USED_SIZE]; /* Signature effectuée */ - uint32_t data; /* Mot de données lues */ - DBError error; /* Validation de la connexion */ + char *desc; /* Description du serveur ciblé*/ + bool status; /* Bilan de la connexion */ /* Identification du serveur à contacter */ @@ -240,9 +313,54 @@ bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) if (ret == -1) { perror("connect"); - goto gdcs_no_listening; + goto gdcsr_no_listening; } + asprintf(&desc, "%s:%hu", host, port); + + status = g_db_client_start_common(client, desc); + + free(desc); + + if (!status) + goto gdcsr_no_listening; + + return true; + + gdcsr_no_listening: + + close(client->fd); + client->fd = -1; + + return false; + +} + + +/****************************************************************************** +* * +* Paramètres : client = client pour les accès distants à manipuler. * +* host = hôte à représenter pour le service. * +* port = port de connexion pour les clients. * +* * +* Description : Démarre réellement la connexion à la base de données. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static bool g_db_client_start_common(GDbClient *client, const char *desc) +{ + rle_string user; /* Nom d'utilisateur associé */ + GChecksum *checksum; /* Empreinte MD5 à signer */ + unsigned char md5_digest[16]; /* Empreinte MD5 calculée */ + RSA *key; /* Clef pour la signature */ + unsigned char sig[RSA_USED_SIZE]; /* Signature effectuée */ + uint32_t data; /* Mot de données lues */ + DBError error; /* Validation de la connexion */ + /** * On réalise l'envoi initial ; le premier paquet doit contenir : * - la commande 'DBC_HELO'. @@ -296,7 +414,7 @@ bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) if (be32toh(data) != DBC_WELCOME) { - log_variadic_message(LMT_ERROR, _("The server '%s:%hu' did not welcome us!"), host, port); + log_variadic_message(LMT_ERROR, _("The server '%s' did not welcome us!"), desc); goto gdcs_error; } @@ -308,30 +426,30 @@ bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) switch (error) { case DBE_NONE: - log_variadic_message(LMT_INFO, _("Connected to the server '%s:%hu'!"), host, port); + log_variadic_message(LMT_INFO, _("Connected to the server '%s'!"), desc); break; case DBE_WRONG_VERSION: - log_variadic_message(LMT_ERROR, _("The server '%s:%hu' does not use our protocol version (0x%08x)..."), - host, port, CDB_PROTOCOL_VERSION); + log_variadic_message(LMT_ERROR, _("The server '%s' does not use our protocol version (0x%08x)..."), + desc, CDB_PROTOCOL_VERSION); goto gdcs_error; break; case DBE_XML_VERSION_ERROR: - log_variadic_message(LMT_ERROR, _("The archive from the server '%s:%hu' does not use our protocol version (0x%08x)..."), - host, port, CDB_PROTOCOL_VERSION); + log_variadic_message(LMT_ERROR, _("The archive from the server '%s' does not use our protocol version (0x%08x)..."), + desc, CDB_PROTOCOL_VERSION); goto gdcs_error; break; case DBE_DB_LOADING_ERROR: - log_variadic_message(LMT_ERROR, _("The server '%s:%hu' got into troubles while loading the database...."), - host, port); + log_variadic_message(LMT_ERROR, _("The server '%s' got into troubles while loading the database...."), + desc); goto gdcs_error; break; default: - log_variadic_message(LMT_ERROR, _("The server '%s:%hu' has run into an error (%u)..."), - host, port, error); + log_variadic_message(LMT_ERROR, _("The server '%s' has run into an error (%u)..."), + desc, error); goto gdcs_error; break; @@ -340,8 +458,8 @@ bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) client->update = g_thread_try_new("cdb_client", (GThreadFunc)g_db_client_update, client, NULL); if (client->update == NULL) { - log_variadic_message(LMT_ERROR, _("Failed to start a listening thread for the server '%s:%hu'!"), - host, port); + log_variadic_message(LMT_ERROR, _("Failed to start a listening thread for the server '%s'!"), + desc); goto gdcs_error; } @@ -351,8 +469,6 @@ bool g_db_client_start(GDbClient *client, const char *host, unsigned short port) unset_rle_string(&user); - gdcs_no_listening: - close(client->fd); client->fd = -1; diff --git a/src/analysis/db/client.h b/src/analysis/db/client.h index 749b8aa..e17c347 100644 --- a/src/analysis/db/client.h +++ b/src/analysis/db/client.h @@ -53,8 +53,11 @@ GType g_db_client_get_type(void); /* Prépare un client pour une connexion à une BD. */ GDbClient *g_db_client_new(char *, char *, const char *, const char *, GList *); -/* Démarre la connexion à la base de données. */ -bool g_db_client_start(GDbClient *, const char *, unsigned short); +/* Démarre la connexion à la base de données interne. */ +bool g_db_client_start_internal(GDbClient *); + +/* Démarre la connexion à la base de données distante. */ +bool g_db_client_start_remote(GDbClient *, const char *, unsigned short); /* Arrête la connexion à la base de données. */ void g_db_client_stop(GDbClient *); diff --git a/src/analysis/db/server.c b/src/analysis/db/server.c index 360b266..48126e8 100644 --- a/src/analysis/db/server.c +++ b/src/analysis/db/server.c @@ -32,6 +32,7 @@ #include #include #include +#include #include "cdb.h" @@ -39,6 +40,7 @@ #include "protocol.h" #include "misc/rlestr.h" #include "../../common/io.h" +#include "../../common/xdg.h" #include "../../core/params.h" #include "../../gui/panels/log.h" @@ -52,17 +54,17 @@ typedef struct _registered_user } registered_user; -/* Informations relatives à un client */ -typedef struct _cdb_client +/* Format générique des adresses de connexion */ +typedef union _gen_sockaddr_t { - GDbServer *server; /* Accès facile en mémoire */ + struct sockaddr_in inet_addr; /* Adresse d'écoute IPv4 */ + struct sockaddr_un unix_addr; /* Adresse d'écoute Unix */ - int fd; /* Canal de communication */ - struct sockaddr_in peer; /* Adresse distante */ - - GThread *thread; /* Procédure de traitement */ +} gen_sockaddr_t; -} cdb_client; +#ifndef UNIX_PATH_MAX +# define UNIX_PATH_MAX 108 +#endif /* Description de serveur à l'écoute (instance) */ struct _GDbServer @@ -73,15 +75,16 @@ struct _GDbServer size_t users_count; /* Nombre d'enregistrés */ int fd; /* Canal de communication */ - char *hostname; /* Désignation humaine */ + int domain; /* Domaine du canal */ + gen_sockaddr_t addr; /* Adresse d'écoute générique */ + socklen_t sock_len; /* Taille de cette adresse */ char *desc; /* Désignation du serveur */ - struct sockaddr_in addr; /* Adresse d'écoute */ + + char *basedir; /* Répertoire de stockage */ GThread *listener; /* Procédure de traitement */ GList *archives; /* Liste des binaires ouverts */ - cdb_client **clients; /* Connexions en place */ - size_t count; /* Quantité de clients */ GMutex mutex; /* Verrou pour l'accès */ }; @@ -181,7 +184,7 @@ static void g_db_server_finalize(GDbServer *server) { g_db_server_unregister_all_user(server); - free(server->hostname); + free(server->desc); G_OBJECT_CLASS(g_db_server_parent_class)->finalize(G_OBJECT(server)); @@ -192,10 +195,8 @@ static void g_db_server_finalize(GDbServer *server) * * * Paramètres : author = utilisateur à représenter via le client. * * kfile = clef menant à sa clef publique. * -* host = hôte à représenter pour le service. * -* port = port de connexion pour les clients. * * * -* Description : Prépare un serveur de BD pour les clients. * +* Description : Prépare un serveur de BD pour les clients internes. * * * * Retour : Structure mise en place ou NULL en cas d'échec. * * * @@ -203,18 +204,97 @@ static void g_db_server_finalize(GDbServer *server) * * ******************************************************************************/ -GDbServer *g_db_server_new(const char *author, char *kfile, const char *host, short port) +GDbServer *g_db_server_new_internal(const char *author, char *kfile) { GDbServer *result; /* Adresse à retourner */ + bool ret; /* Bilan d'un appel */ + char *suffix; /* Suffixe pour un fichier */ + char *sock_path; /* Chemin vers le canal UNIX */ + + result = g_object_new(G_TYPE_DB_SERVER, NULL); + + /* Chargement du profil */ + + ret = g_db_server_register_user(result, author, kfile); + if (!ret) goto gdsni_error; + + /* Détermination du point d'écoute */ + + result->domain = AF_UNIX; + + asprintf(&suffix, "chrysalide" G_DIR_SEPARATOR_S ".internal_server.%d", getpid()); + sock_path = get_xdg_config_dir(suffix); + free(suffix); + + memset(&result->addr, 0, sizeof(struct sockaddr_un)); + + result->addr.unix_addr.sun_family = AF_UNIX; + strncpy(result->addr.unix_addr.sun_path, sock_path, UNIX_PATH_MAX - 1); + + free(sock_path); + + result->sock_len = sizeof(struct sockaddr_un); + + /* Désignation humaine */ + + asprintf(&result->desc, "unix://.internal_server.%d", getpid()); + + /* Répertoire de stockage */ + + result->basedir = get_xdg_config_dir("chrysalide" G_DIR_SEPARATOR_S "cdbs"); + + return result; + + gdsni_error: + + g_object_unref(G_OBJECT(result)); + + return NULL; + +} + + +/****************************************************************************** +* * +* Paramètres : conf = fichier de configuration à interpréter. * +* * +* Description : Prépare un serveur de BD pour les clients distants. * +* * +* Retour : Structure mise en place ou NULL en cas d'échec. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GDbServer *g_db_server_new_remote(const char *conf) +{ + GDbServer *result; /* Adresse à retourner */ + + char *host; + short port; + + struct hostent *hp; /* Informations sur l'hôte */ size_t desclen; /* Taille de désignation */ const char *ip; /* Adresse IPv4 ou IPv6 */ result = g_object_new(G_TYPE_DB_SERVER, NULL); - /* ... =*/g_db_server_register_user(result, author, kfile); + /* Chargement des profils */ - result->hostname = strdup(host); + + /* + ret = g_db_server_register_user(result, author, kfile); + if (!ret) goto gdsni_error; + */ + + result->domain = AF_INET; + + host = "localhost"; + port = 9999; + + + /* Détermination du point d'écoute */ hp = gethostbyname(host); if (hp == NULL) @@ -223,23 +303,33 @@ GDbServer *g_db_server_new(const char *author, char *kfile, const char *host, sh goto gdsn_error; } - result->addr.sin_family = hp->h_addrtype; - memcpy(&result->addr.sin_addr, hp->h_addr_list[0], sizeof(struct in_addr)); + result->addr.inet_addr.sin_family = hp->h_addrtype; + memcpy(&result->addr.inet_addr.sin_addr, hp->h_addr_list[0], sizeof(struct in_addr)); + + result->addr.inet_addr.sin_port = htons(port); - result->addr.sin_port = htons(port); + result->sock_len = sizeof(struct sockaddr_in); desclen = INET6_ADDRSTRLEN + 1 + 5 + 1; result->desc = (char *)calloc(desclen, sizeof(char)); - ip = inet_ntop(AF_INET, &result->addr.sin_addr, result->desc, INET6_ADDRSTRLEN); + ip = inet_ntop(AF_INET, &result->addr.inet_addr.sin_addr, result->desc, INET6_ADDRSTRLEN); if (ip == NULL) { perror("inet_ntop"); goto gdsn_error; } + /* Désignation humaine */ + snprintf(result->desc + strlen(ip), 1 + 5, ":%hu", port); + /* Répertoire de stockage */ + + + result->basedir = strdup("/tmp/"); /* TODO */ + + return result; gdsn_error: @@ -382,10 +472,10 @@ static void *g_db_server_listener(GDbServer *server) { struct pollfd fds; /* Surveillance des flux */ int ret; /* Bilan d'un appel */ - struct sockaddr_in peer; /* Adresse cliente */ + gen_sockaddr_t peer; /* Adresse cliente */ int fd; /* Canal établi vers un client */ - char source[INET6_ADDRSTRLEN]; /* Adresse du client (IPv4/6) */ const char *ip; /* Statut de la conversion */ + char *peer_name; /* Désignation du correspondant*/ DBError error; /* Validation de la connexion */ GCdbArchive *archive; /* Destinataire final du client*/ uint32_t cmd; /* Commande initiale lue */ @@ -409,18 +499,31 @@ static void *g_db_server_listener(GDbServer *server) if (fds.revents & (POLLIN | POLLPRI)) { - fd = accept(server->fd, &peer, (socklen_t []) { sizeof(struct sockaddr_in) }); + fd = accept(server->fd, (struct sockaddr *)&peer, (socklen_t []) { sizeof(gen_sockaddr_t) }); if (fd == -1) { perror("accept"); continue; } - ip = inet_ntop(AF_INET, &peer.sin_addr, source, sizeof(source)); - if (ip == NULL) + /* Construction d'une représentation */ + + if (*((sa_family_t *)&peer) == AF_UNIX) + peer_name = strdup(server->desc); + + else if (*((sa_family_t *)&peer) == AF_INET) { - perror("inet_ntop"); - goto gdsl_error; + peer_name = (char *)calloc(INET6_ADDRSTRLEN + 1 + 5 + 1, sizeof(char)); + + ip = inet_ntop(AF_INET, &peer.inet_addr.sin_addr, peer_name, INET6_ADDRSTRLEN); + if (ip == NULL) + { + perror("inet_ntop"); + goto gdsl_error; + } + + snprintf(peer_name + strlen(ip), 1 + 5, ":%hu", ntohs(peer.inet_addr.sin_port)); + } error = DBE_NONE; @@ -439,72 +542,72 @@ static void *g_db_server_listener(GDbServer *server) if (!safe_recv(fd, &cmd, sizeof(uint32_t), 0)) { - log_variadic_message(LMT_ERROR, _("Error while getting the initial command from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("Error while getting the initial command from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (!safe_recv(fd, &version, sizeof(uint32_t), 0)) { - log_variadic_message(LMT_ERROR, _("Error while getting the protocol version from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("Error while getting the protocol version from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (!recv_rle_string(&hash, fd, 0)) { - log_variadic_message(LMT_ERROR, _("Error while getting the binary hash from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("Error while getting the binary hash from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (!recv_rle_string(&user, fd, 0)) { - log_variadic_message(LMT_ERROR, _("Error while getting the user name from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("Error while getting the user name from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (!safe_recv(fd, sig, RSA_USED_SIZE, 0)) { - log_variadic_message(LMT_ERROR, _("Error while getting the signature from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("Error while getting the signature from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (be32toh(cmd) != DBC_HELO) { - log_variadic_message(LMT_ERROR, _("The client from '%s:%hu' did not introduce itself!"), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("The client from '%s' did not introduce itself!"), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (be32toh(version) != CDB_PROTOCOL_VERSION) { - log_variadic_message(LMT_ERROR, _("The client from '%s:%hu' does not use the same protocol: 0x%08x vs 0x%08x..."), - source, ntohs(peer.sin_port), be32toh(version), CDB_PROTOCOL_VERSION); + log_variadic_message(LMT_ERROR, _("The client from '%s' does not use the same protocol: 0x%08x vs 0x%08x..."), + peer_name, be32toh(version), CDB_PROTOCOL_VERSION); error = DBE_WRONG_VERSION; goto gdsl_error_sending; } if (is_rle_string_empty(&hash)) { - log_variadic_message(LMT_ERROR, _("The submitted binary hash from '%s:%hu' is empty!"), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("The submitted binary hash from '%s' is empty!"), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } if (is_rle_string_empty(&user)) { - log_variadic_message(LMT_ERROR, _("No user is associated with the client from '%s:%hu'..."), - source, ntohs(peer.sin_port)); + log_variadic_message(LMT_ERROR, _("No user is associated with the client from '%s'..."), + peer_name); error = DBE_BAD_EXCHANGE; goto gdsl_error_sending; } @@ -533,7 +636,7 @@ static void *g_db_server_listener(GDbServer *server) } if (iter == NULL) - archive = g_cdb_archive_new(server->desc, &hash, &user, &error); + archive = g_cdb_archive_new(server->basedir, &hash, &user, &error); /** * Le serveur doit répondre pour un message type : @@ -563,6 +666,8 @@ static void *g_db_server_listener(GDbServer *server) server->archives = g_list_append(server->archives, archive); error = g_cdb_archive_add_client(archive, fd, &user); + free(peer_name); + exit_rle_string(&hash); exit_rle_string(&user); @@ -574,6 +679,8 @@ static void *g_db_server_listener(GDbServer *server) gdsl_error: + free(peer_name); + exit_rle_string(&hash); exit_rle_string(&user); @@ -605,7 +712,7 @@ bool g_db_server_start(GDbServer *server) int ret; /* Bilan d'un appel */ int backlog; /* Nombre de connexions maximal*/ - server->fd = socket(AF_INET, SOCK_STREAM, 0); + server->fd = socket(server->domain, SOCK_STREAM, 0); if (server->fd == -1) { perror("socket"); @@ -616,10 +723,11 @@ bool g_db_server_start(GDbServer *server) if (ret == -1) { perror("setsockopt"); + exit(0); goto gdss_error; } - ret = bind(server->fd, (struct sockaddr *)&server->addr, sizeof(struct sockaddr_in)); + ret = bind(server->fd, (struct sockaddr *)&server->addr, server->sock_len); if (ret == -1) { perror("bind"); @@ -638,8 +746,7 @@ bool g_db_server_start(GDbServer *server) server->listener = g_thread_new("cdb_listener", (GThreadFunc)g_db_server_listener, server); - log_variadic_message(LMT_PROCESS, _("Server started and listening at %s:%hu"), - server->hostname, ntohs(server->addr.sin_port)); + log_variadic_message(LMT_PROCESS, _("Server started and listening at %s"), server->desc); return true; @@ -667,9 +774,6 @@ bool g_db_server_start(GDbServer *server) void g_db_server_stop(GDbServer *server) { - size_t i; /* Boucle de parcours */ - GThread *thread; /* Procédure de traitement */ - if (server->fd != -1) return; @@ -678,14 +782,6 @@ void g_db_server_stop(GDbServer *server) g_thread_join(server->listener); - for (i = 0; i < server->count; i++) - { - /* Sauvegarde de la référene, qui peut disparaître */ - thread = server->clients[i]->thread; - - close(server->clients[i]->fd); - g_thread_join(thread); - - } + /* TODO : s'occuper des archives ouvertes */ } diff --git a/src/analysis/db/server.h b/src/analysis/db/server.h index a06e89f..8d3c9b6 100644 --- a/src/analysis/db/server.h +++ b/src/analysis/db/server.h @@ -47,8 +47,11 @@ typedef struct _GDbServerClass GDbServerClass; /* Indique le type défini pour une description de serveur à l'écoute. */ GType g_db_server_get_type(void); -/* Prépare un serveur de BD pour les clients. */ -GDbServer *g_db_server_new(const char *, char *, const char *, short); +/* Prépare un serveur de BD pour les clients internes. */ +GDbServer *g_db_server_new_internal(const char *, char *); + +/* Prépare un serveur de BD pour les clients distants. */ +GDbServer *g_db_server_new_remote(const char *); /* Démarre le serveur de base de données. */ bool g_db_server_start(GDbServer *); diff --git a/src/main.c b/src/main.c index 09fcf42..ec379bc 100644 --- a/src/main.c +++ b/src/main.c @@ -193,7 +193,7 @@ int main(int argc, char **argv) pub = get_xdg_config_dir("chrysalide" G_DIR_SEPARATOR_S "id_rsa.pub"); - server = g_db_server_new(author, pub, "localhost", 1337); + server = g_db_server_new_internal(author, pub); g_db_server_start(server); /* Charge le dernier projet */ -- cgit v0.11.2-87-g4458