diff options
Diffstat (limited to 'src/analysis')
| -rw-r--r-- | src/analysis/db/client.c | 41 |
1 files changed, 26 insertions, 15 deletions
diff --git a/src/analysis/db/client.c b/src/analysis/db/client.c index 2256a21..f97335c 100644 --- a/src/analysis/db/client.c +++ b/src/analysis/db/client.c @@ -65,6 +65,7 @@ struct _GDbClient int fd; /* Canal de communication */ SSL *tls_fd; /* Même canal, mais sécurisé */ + char *desc; /* Description du lien */ GMutex sending_lock; /* Concurrence des envois */ GThread *update; /* Procédure de traitement */ @@ -89,7 +90,7 @@ static void g_db_client_init(GDbClient *); 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 *); +static bool g_db_client_start_common(GDbClient *, char *); /* Assure l'accueil des nouvelles mises à jour. */ static void *g_db_client_update(GDbClient *); @@ -143,6 +144,7 @@ static void g_db_client_init(GDbClient *client) client->fd = -1; client->tls_fd = NULL; + client->desc = NULL; } @@ -176,6 +178,9 @@ static void g_db_client_finalize(GDbClient *client) assert(client->tls_fd == NULL); + if (client->desc != NULL) + free(client->desc); + G_OBJECT_CLASS(g_db_client_parent_class)->finalize(G_OBJECT(client)); } @@ -260,8 +265,6 @@ bool g_db_client_start_internal(GDbClient *client) status = g_db_client_start_common(client, desc); - free(desc); - if (!status) goto gdcsi_no_listening; @@ -332,8 +335,6 @@ bool g_db_client_start_remote(GDbClient *client, const char *host, unsigned shor status = g_db_client_start_common(client, desc); - free(desc); - if (!status) goto gdcsr_no_listening; @@ -363,7 +364,7 @@ bool g_db_client_start_remote(GDbClient *client, const char *host, unsigned shor * * ******************************************************************************/ -static bool g_db_client_start_common(GDbClient *client, const char *desc) +static bool g_db_client_start_common(GDbClient *client, char *desc) { const SSL_METHOD *method; /* Mode du canal sécurisé */ int ret; /* Bilan d'un appel */ @@ -378,6 +379,8 @@ static bool g_db_client_start_common(GDbClient *client, const char *desc) uint32_t data; /* Mot de données lues */ DBError error; /* Validation de la connexion */ + client->desc = desc; + /** * Mise en place d'un environnement sécurisé. */ @@ -432,7 +435,7 @@ static bool g_db_client_start_common(GDbClient *client, const char *desc) status = pack_rle_string(&client->hash, &out_pbuf); if (!status) goto setup_error; - dup_into_rle_string(&user, client->author); /* FIXME : src ? */ + dup_into_rle_string(&user, client->author); status = pack_rle_string(&user, &out_pbuf); if (!status) goto setup_error; @@ -577,6 +580,7 @@ static void *g_db_client_update(GDbClient *client) uint32_t command; /* Commande de la requête */ DBError error; /* Bilan d'une commande passée */ GDbCollection *collec; /* Collection visée au final */ + char *msg; /* Message d'erreur à imprimer */ fds.fd = client->fd; fds.events = POLLIN | POLLPRI; @@ -639,19 +643,19 @@ static void *g_db_client_update(GDbClient *client) gdcu_bad_exchange: - printf("Bad reception...\n"); - - /* TODO : close conn */ + asprintf(&msg, _("Bad reception from %s"), client->desc); - ; + LOG_ERROR(LMT_ERROR, msg); + free(msg); + break; } } - //g_db_client_stop(client); + g_db_client_stop(client); exit_packed_buffer(&in_pbuf); @@ -680,17 +684,24 @@ void g_db_client_stop(GDbClient *client) if (client->fd != -1) { - assert(client->tls_ctx == NULL); + /** + * Si la fermture est forcée, le thread de traitement va terminer en erreur. + * Donc cette fonction sera appelée deux fois. Seule la première va affecter + * le contexte, donc on le peut pas s'assurer de la condition suivante dans + * tous les cas. + */ + + /*assert(client->tls_ctx == NULL);*/ return; } ret = close(client->fd); if (ret == -1) perror("close"); - g_thread_join(client->update); - client->fd = -1; + g_thread_join(client->update); + /* Environnement TLS */ SSL_CTX_free(client->tls_ctx); |