diff options
Diffstat (limited to 'src/analysis/db/admin.c')
-rw-r--r-- | src/analysis/db/admin.c | 292 |
1 files changed, 292 insertions, 0 deletions
diff --git a/src/analysis/db/admin.c b/src/analysis/db/admin.c new file mode 100644 index 0000000..355f8c0 --- /dev/null +++ b/src/analysis/db/admin.c @@ -0,0 +1,292 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * admin.c - connexion en administrateur à un serveur Chrysalide + * + * Copyright (C) 2014-2019 Cyrille Bagard + * + * This file is part of Chrysalide. + * + * Chrysalide is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * Chrysalide is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with Chrysalide. If not, see <http://www.gnu.org/licenses/>. + */ + + +#include "admin.h" + + +#include <assert.h> +#include <poll.h> + + +#include "client-int.h" +#include "../../core/logs.h" + + + +/* Description de client à l'écoute (instance) */ +struct _GAdminClient +{ + GHubClient parent; /* A laisser en premier */ + +}; + +/* Description de client à l'écoute (classe) */ +struct _GAdminClientClass +{ + GHubClientClass parent; /* A laisser en premier */ + +}; + + +/* Initialise la classe des descriptions de fichier binaire. */ +static void g_admin_client_class_init(GAdminClientClass *); + +/* Initialise une description de fichier binaire. */ +static void g_admin_client_init(GAdminClient *); + +/* Supprime toutes les références externes. */ +static void g_admin_client_dispose(GAdminClient *); + +/* Procède à la libération totale de la mémoire. */ +static void g_admin_client_finalize(GAdminClient *); + +/* Assure l'accueil des nouvelles mises à jour. */ +static void *g_admin_client_update(GAdminClient *); + + + +/* Indique le type défini pour une description de client à l'écoute. */ +G_DEFINE_TYPE(GAdminClient, g_admin_client, G_TYPE_HUB_CLIENT); + + +/****************************************************************************** +* * +* Paramètres : klass = classe à initialiser. * +* * +* Description : Initialise la classe des descriptions de fichier binaire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_admin_client_class_init(GAdminClientClass *klass) +{ + GObjectClass *object; /* Autre version de la classe */ + GHubClientClass *client; /* Classe parente */ + + object = G_OBJECT_CLASS(klass); + + object->dispose = (GObjectFinalizeFunc/* ! */)g_admin_client_dispose; + object->finalize = (GObjectFinalizeFunc)g_admin_client_finalize; + + client = G_HUB_CLIENT_CLASS(klass); + + client->role = CRL_ADMIN; + client->recv_func = (GThreadFunc)g_admin_client_update; + +} + + +/****************************************************************************** +* * +* Paramètres : client = instance à initialiser. * +* * +* Description : Initialise une description de fichier binaire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_admin_client_init(GAdminClient *client) +{ + +} + + +/****************************************************************************** +* * +* Paramètres : archive = instance d'objet GLib à traiter. * +* * +* Description : Supprime toutes les références externes. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_admin_client_dispose(GAdminClient *client) +{ + g_hub_client_stop(G_HUB_CLIENT(client)); + + G_OBJECT_CLASS(g_admin_client_parent_class)->dispose(G_OBJECT(client)); + +} + + +/****************************************************************************** +* * +* Paramètres : client = instance d'objet GLib à traiter. * +* * +* Description : Procède à la libération totale de la mémoire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_admin_client_finalize(GAdminClient *client) +{ + G_OBJECT_CLASS(g_admin_client_parent_class)->finalize(G_OBJECT(client)); + +} + + +/****************************************************************************** +* * +* Paramètres : - * +* * +* Description : Prépare un client pour une connexion à une BD. * +* * +* Retour : Structure mise en place ou NULL en cas d'échec. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GAdminClient *g_admin_client_new(void) +{ + GAdminClient *result; /* Adresse à retourner */ + + result = g_object_new(G_TYPE_ADMIN_CLIENT, NULL); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : client = client pour les accès distants à manipuler. * +* * +* Description : Assure l'accueil des nouvelles mises à jour. * +* * +* Retour : NULL. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void *g_admin_client_update(GAdminClient *client) +{ + GHubClient *base; /* Base de l'instance */ + struct pollfd fds[2]; /* Surveillance des flux */ + packed_buffer in_pbuf; /* Tampon de réception */ + int ret; /* Bilan d'un appel */ + bool status; /* Bilan d'une opération */ + uint32_t command; /* Commande de la requête */ + //packed_buffer out_pbuf; /* Tampon d'émission */ + char *msg; /* Message d'erreur à imprimer */ + + base = G_HUB_CLIENT(client); + + /** + * Phase d'écoute continue... + */ + + fds[0].fd = base->stop_ctrl[0]; + fds[0].events = POLLIN | POLLPRI; + + fds[1].fd = SSL_get_fd(base->tls_fd); + fds[1].events = POLLIN | POLLPRI; + + init_packed_buffer(&in_pbuf); + + while (true) + { + ret = poll(fds, 2, -1); + if (ret == -1) + { + LOG_ERROR_N("poll"); + break; + } + + /* Demande expresse d'arrêt des procédures */ + if (fds[0].revents) + break; + + /* Le canal est fermé, une sortie doit être demandée... */ + if (fds[1].revents & POLLNVAL) + break; + + /** + * Même chose, cf. "TCP: When is EPOLLHUP generated?" + * https://stackoverflow.com/questions/52976152/tcp-when-is-epollhup-generated/52976327#52976327 + */ + + if (fds[1].revents & (POLLHUP | POLLRDHUP)) + break; + + if (fds[1].revents & (POLLIN | POLLPRI)) + { + reset_packed_buffer(&in_pbuf); + + status = ssl_recv_packed_buffer(&in_pbuf, base->tls_fd); + if (!status) goto bad_exchange; + + next_command: + + status = extract_packed_buffer(&in_pbuf, &command, sizeof(uint32_t), true); + if (!status) goto bad_exchange; + + switch (command) + { + default: + log_variadic_message(LMT_INFO, + _("This command is not available on this side: 0x%08x"), command); + goto bad_exchange; + break; + + } + + if (has_more_data_in_packed_buffer(&in_pbuf)) + goto next_command; + + continue; + + bad_exchange: + + asprintf(&msg, _("Bad reception from %s"), base->desc); + + LOG_ERROR(LMT_ERROR, msg); + + free(msg); + + break; + + } + + } + + g_hub_client_stop(G_HUB_CLIENT(client)); + + exit_packed_buffer(&in_pbuf); + + return NULL; + +} |