diff options
Diffstat (limited to 'src/analysis/storage')
| -rw-r--r-- | src/analysis/storage/Makefile.am | 30 | ||||
| -rw-r--r-- | src/analysis/storage/cache-int.h | 62 | ||||
| -rw-r--r-- | src/analysis/storage/cache.c | 334 | ||||
| -rw-r--r-- | src/analysis/storage/cache.h | 62 | ||||
| -rw-r--r-- | src/analysis/storage/container-int.h | 62 | ||||
| -rw-r--r-- | src/analysis/storage/container.c | 140 | ||||
| -rw-r--r-- | src/analysis/storage/container.h | 66 | ||||
| -rw-r--r-- | src/analysis/storage/serialize-int.h | 55 | ||||
| -rw-r--r-- | src/analysis/storage/serialize.c | 119 | ||||
| -rw-r--r-- | src/analysis/storage/serialize.h | 65 | ||||
| -rw-r--r-- | src/analysis/storage/storage-int.h | 66 | ||||
| -rw-r--r-- | src/analysis/storage/storage.c | 436 | ||||
| -rw-r--r-- | src/analysis/storage/storage.h | 70 | ||||
| -rw-r--r-- | src/analysis/storage/tpmem.c | 454 | ||||
| -rw-r--r-- | src/analysis/storage/tpmem.h | 70 |
15 files changed, 2091 insertions, 0 deletions
diff --git a/src/analysis/storage/Makefile.am b/src/analysis/storage/Makefile.am new file mode 100644 index 0000000..94f5041 --- /dev/null +++ b/src/analysis/storage/Makefile.am @@ -0,0 +1,30 @@ + +noinst_LTLIBRARIES = libanalysisstorage.la + + +libanalysisstorage_la_SOURCES = \ + cache-int.h \ + cache.h cache.c \ + container-int.h \ + container.h container.c \ + serialize-int.h \ + serialize.h serialize.c \ + storage-int.h \ + storage.h storage.c \ + tpmem.h tpmem.c + +libanalysisstorage_la_LIBADD = + +libanalysisstorage_la_LDFLAGS = $(LIBSSL_LIBS) + + +devdir = $(includedir)/chrysalide/$(subdir:src/%=%) + +dev_HEADERS = $(libanalysisstorage_la_SOURCES:%c=) + + +AM_CPPFLAGS = $(LIBGTK_CFLAGS) $(LIBXML_CFLAGS) $(LIBARCHIVE_CFLAGS) $(LIBSQLITE_CFLAGS) $(LIBSSL_CFLAGS) + +AM_CFLAGS = $(DEBUG_CFLAGS) $(WARNING_FLAGS) $(COMPLIANCE_FLAGS) + +SUBDIRS = diff --git a/src/analysis/storage/cache-int.h b/src/analysis/storage/cache-int.h new file mode 100644 index 0000000..17be85d --- /dev/null +++ b/src/analysis/storage/cache-int.h @@ -0,0 +1,62 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * cache.h - prototypes internes pour la conservation hors mémoire d'objets choisis + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#ifndef _ANALYSIS_STORAGE_CACHE_INT_H +#define _ANALYSIS_STORAGE_CACHE_INT_H + + +#include "cache.h" + + + +/* Définition d'un cache d'objets entreposables (instance) */ +struct _GObjectCache +{ + GObject parent; /* A laisser en premier */ + + GLoadedContent *loaded; /* Contenu principal */ + + char *filename; /* Fichier local utilisé */ + int fd; /* Descripteur du flux associé */ + + GCacheContainer **containers; /* Objets en sursis */ + size_t count; /* Quantité de ces objets */ + size_t free_ptr; /* Point d'enregistrement */ + GMutex mutex; /* Contrôle d'accès à la liste */ + +}; + +/* Définition d'un cache d'objets entreposables (classe) */ +struct _GObjectCacheClass +{ + GObjectClass parent; /* A laisser en premier */ + +}; + + +/* Associe un contenu à un cache d'objets. */ +bool g_object_cache_open_for(GObjectCache *, GLoadedContent *); + + + +#endif /* _ANALYSIS_STORAGE_CACHE_INT_H */ diff --git a/src/analysis/storage/cache.c b/src/analysis/storage/cache.c new file mode 100644 index 0000000..3d0966b --- /dev/null +++ b/src/analysis/storage/cache.c @@ -0,0 +1,334 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * cache.c - conservation hors mémoire d'objets choisis + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#include "cache.h" + + +#include <assert.h> +#include <fcntl.h> +#include <malloc.h> +#include <unistd.h> + + +#include "cache-int.h" +#include "../../core/logs.h" + + + +/* Initialise la classe des caches d'objets entreposables. */ +static void g_object_cache_class_init(GObjectCacheClass *); + +/* Initialise une instance de cache d'objets entreposables. */ +static void g_object_cache_init(GObjectCache *); + +/* Supprime toutes les références externes. */ +static void g_object_cache_dispose(GObjectCache *); + +/* Procède à la libération totale de la mémoire. */ +static void g_object_cache_finalize(GObjectCache *); + + + +/* Indique le type défini pour un cache d'objets entreposables. */ +G_DEFINE_TYPE(GObjectCache, g_object_cache, G_TYPE_OBJECT); + + +/****************************************************************************** +* * +* Paramètres : klass = classe à initialiser. * +* * +* Description : Initialise la classe des caches d'objets entreposables. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_cache_class_init(GObjectCacheClass *klass) +{ + GObjectClass *object; /* Autre version de la classe */ + + object = G_OBJECT_CLASS(klass); + + object->dispose = (GObjectFinalizeFunc/* ! */)g_object_cache_dispose; + object->finalize = (GObjectFinalizeFunc)g_object_cache_finalize; + +} + + +/****************************************************************************** +* * +* Paramètres : cache = instance à initialiser. * +* * +* Description : Initialise une instance de cache d'objets entreposables. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_cache_init(GObjectCache *cache) +{ + cache->loaded = NULL; + + cache->filename = NULL; + cache->fd = -1; + + cache->containers = NULL; + cache->count = 0; + cache->free_ptr = 0; + g_mutex_init(&cache->mutex); + +} + + +/****************************************************************************** +* * +* Paramètres : cache = instance d'objet GLib à traiter. * +* * +* Description : Supprime toutes les références externes. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_cache_dispose(GObjectCache *cache) +{ + size_t i; /* Boucle de parcours */ + + g_clear_object(&cache->loaded); + + g_mutex_lock(&cache->mutex); + + for (i = 0; i < cache->count; i++) + g_clear_object(&cache->containers[i]); + + g_mutex_unlock(&cache->mutex); + + g_mutex_clear(&cache->mutex); + + G_OBJECT_CLASS(g_object_cache_parent_class)->dispose(G_OBJECT(cache)); + +} + + +/****************************************************************************** +* * +* Paramètres : cache = instance d'objet GLib à traiter. * +* * +* Description : Procède à la libération totale de la mémoire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_cache_finalize(GObjectCache *cache) +{ + int ret; /* Bilan d'un appel */ + + ret = access(cache->filename, W_OK); + if (ret == 0) + { + ret = unlink(cache->filename); + if (ret != 0) LOG_ERROR_N("unlink"); + } + + free(cache->filename); + + if (cache->containers != NULL) + free(cache->containers); + + G_OBJECT_CLASS(g_object_cache_parent_class)->finalize(G_OBJECT(cache)); + +} + + +/****************************************************************************** +* * +* Paramètres : loaded = contenu binaire à associer. * +* * +* Description : Crée le support d'un cache d'objets entreposables. * +* * +* Retour : Mécanismes mis en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GObjectCache *g_object_cache_new(GLoadedContent *loaded) +{ + GObjectCache *result; /* Structure à retourner */ + + result = g_object_new(G_TYPE_OBJECT_CACHE, NULL); + + if (!g_object_cache_open_for(result, loaded)) + { + g_object_unref(G_OBJECT(result)); + result = NULL; + } + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : cache = cache d'objets à manipuler. * +* loaded = contenu binaire à associer. * +* * +* Description : Associe un contenu à un cache d'objets. * +* * +* Retour : Bilan de l'opéation. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_object_cache_open_for(GObjectCache *cache, GLoadedContent *loaded) +{ + bool result; /* Bilan à retourner */ + GBinContent *content; /* Contenu binaire traité */ + const gchar *checksum; /* Empreinte de ce contenu */ + + result = (cache->loaded == NULL); + assert(result); + + if (result) goto done; + + cache->loaded = loaded; + g_object_ref(G_OBJECT(loaded)); + + /* Constitution du fichier de cache */ + + content = g_loaded_content_get_content(loaded); + + checksum = g_binary_content_get_checksum(content); + + asprintf(&cache->filename, "/dev/shm/%s.cache", checksum); + + g_object_unref(G_OBJECT(content)); + + /* Ouverture dudit fichier */ + + cache->fd = open(cache->filename, O_CREAT | O_TRUNC | O_LARGEFILE, 0600); + if (cache->fd == -1) + { + LOG_ERROR_N("open"); + result = false; + goto done; + } + + /* Préparation du cache */ + + cache->count = 1000; + + cache->containers = calloc(cache->count, sizeof(GCacheContainer *)); + + result = true; + + done: + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : cache = cache d'objets à manipuler. * +* container = objet à placer dans le cache. * +* * +* Description : Introduit un contenu dans un cache d'objets. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +void g_object_cache_add(GObjectCache *cache, GCacheContainer *container) +{ + bool loop; /* Détection de trop-plein */ + size_t i; /* Boucle de parcours */ + + /* Recherche d'un emplacement libre */ + + loop = false; + + i = cache->free_ptr; + + do + { + if (cache->containers[i] != NULL) + { + g_cache_container_lock_unlock(cache->containers[i], true); + + if (g_cache_container_can_store(cache->containers[i])) + { + if (true) // TODO + g_clear_object(&cache->containers[i]); + } + + g_cache_container_lock_unlock(cache->containers[i], false); + + } + + if (cache->containers[i] == NULL) + break; + + i++; + + if (i == cache->count) + i = 0; + + loop = (i == cache->free_ptr); + + } while (!loop); + + if (loop) + { + log_simple_message(LMT_WARNING, _("Instruction cache is full!")); + goto exit; + } + + /* Inscription à la liste des sursis */ + + cache->containers[i] = container; + + cache->free_ptr = (i + 1); + + if (cache->free_ptr == cache->count) + cache->free_ptr = 0; + + exit: + + ; + +} diff --git a/src/analysis/storage/cache.h b/src/analysis/storage/cache.h new file mode 100644 index 0000000..7f65c09 --- /dev/null +++ b/src/analysis/storage/cache.h @@ -0,0 +1,62 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * cache.h - prototypes pour la conservation hors mémoire d'objets choisis + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#ifndef _ANALYSIS_STORAGE_CACHE_H +#define _ANALYSIS_STORAGE_CACHE_H + + +#include <glib-object.h> + + +#include "container.h" +#include "../loaded.h" + + + +#define G_TYPE_OBJECT_CACHE g_object_cache_get_type() +#define G_OBJECT_CACHE(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), G_TYPE_OBJECT_CACHE, GObjectCache)) +#define G_IS_OBJECT_CACHE(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), G_TYPE_OBJECT_CACHE)) +#define G_OBJECT_CACHE_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), G_TYPE_OBJECT_CACHE, GObjectCacheClass)) +#define G_IS_OBJECT_CACHE_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), G_TYPE_OBJECT_CACHE)) +#define G_OBJECT_CACHE_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), G_TYPE_OBJECT_CACHE, GObjectCacheClass)) + + +/* Définition d'un cache d'objets entreposables (instance) */ +typedef struct _GObjectCache GObjectCache; + +/* Définition d'un cache d'objets entreposables (classe) */ +typedef struct _GObjectCacheClass GObjectCacheClass; + + +/* Indique le type défini pour un cache d'objets entreposables. */ +GType g_object_cache_get_type(void); + +/* Crée le support d'un cache d'objets entreposables. */ +GObjectCache *g_object_cache_new(GLoadedContent *); + +/* Introduit un contenu dans un cache d'objets. */ +void g_object_cache_add(GObjectCache *, GCacheContainer *); + + + +#endif /* _ANALYSIS_STORAGE_CACHE_H */ diff --git a/src/analysis/storage/container-int.h b/src/analysis/storage/container-int.h new file mode 100644 index 0000000..19e77df --- /dev/null +++ b/src/analysis/storage/container-int.h @@ -0,0 +1,62 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * container-int.h - définitions internes propres aux conteneurs d'objets entreposables dans un cache + * + * Copyright (C) 2020 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/>. + */ + + +#ifndef _ANALYSIS_STORAGE_CONTAINER_INT_H +#define _ANALYSIS_STORAGE_CONTAINER_INT_H + + +#include "container.h" + + + +/* Contrôle l'accès au contenu d'un conteneur. */ +typedef void (* lock_unlock_container_cb) (GCacheContainer *, bool); + +/* Détermine si le conteneur a ses accès verrouillés. */ +#ifndef NDEBUG +typedef bool (* is_locked_container_cb) (GCacheContainer *); +#endif + +/* Indique si le contenu d'un conteneur peut être mis en cache. */ +typedef bool (* can_store_container_cb) (GCacheContainer *); + +/* Intermédiaire pour un conteneur d'objets entreposables (interface) */ +struct _GCacheContainerIface +{ + GTypeInterface base_iface; /* A laisser en premier */ + + lock_unlock_container_cb lock_unlock; /* Contrôle d'accès au contenu */ +#ifndef NDEBUG + is_locked_container_cb is_locked; /* Validation des verrous */ +#endif + can_store_container_cb can_store; /* Mise en cache possible ? */ + +}; + + +/* Redéfinition */ +typedef GCacheContainerIface GCacheContainerInterface; + + + +#endif /* _ANALYSIS_STORAGE_CONTAINER_INT_H */ diff --git a/src/analysis/storage/container.c b/src/analysis/storage/container.c new file mode 100644 index 0000000..60d6d74 --- /dev/null +++ b/src/analysis/storage/container.c @@ -0,0 +1,140 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * container.c - conteneurs d'objets entreposables dans un cache + * + * Copyright (C) 2020 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 "container.h" + + +#include <assert.h> + + +#include "container-int.h" + + + +/* Procède à l'initialisation de l'interface de conteneur. */ +static void g_cache_container_default_init(GCacheContainerInterface *); + + + +/* Détermine le type d'une interface pour un conteneur d'objets entreposables. */ +G_DEFINE_INTERFACE(GCacheContainer, g_cache_container, G_TYPE_OBJECT) + + +/****************************************************************************** +* * +* Paramètres : iface = interface GLib à initialiser. * +* * +* Description : Procède à l'initialisation de l'interface de conteneur. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_cache_container_default_init(GCacheContainerInterface *iface) +{ + +} + + +/****************************************************************************** +* * +* Paramètres : container = conteneur à manipuler. * +* lock = indique une demande de verrou. * +* * +* Description : Contrôle l'accès au contenu d'un conteneur. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +void g_cache_container_lock_unlock(GCacheContainer *container, bool lock) +{ + GCacheContainerIface *iface; /* Interface utilisée */ + + assert(g_cache_container_is_locked(container)); + + iface = G_CACHE_CONTAINER_GET_IFACE(container); + + iface->lock_unlock(container, lock); + +} + + +/****************************************************************************** +* * +* Paramètres : container = conteneur à consulter. * +* * +* Description : Détermine si le conteneur a ses accès verrouillés. * +* * +* Retour : Bilan de la consultation. * +* * +* Remarques : - * +* * +******************************************************************************/ + +#ifndef NDEBUG +bool g_cache_container_is_locked(GCacheContainer *container) +{ + bool result; /* Bilan à retourner */ + GCacheContainerIface *iface; /* Interface utilisée */ + + iface = G_CACHE_CONTAINER_GET_IFACE(container); + + result = iface->is_locked(container); + + return result; + +} +#endif + + +/****************************************************************************** +* * +* Paramètres : container = conteneur à consulter. * +* * +* Description : Indique si le contenu d'un conteneur peut être mis en cache. * +* * +* Retour : Bilan de la consultation. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_cache_container_can_store(GCacheContainer *container) +{ + bool result; /* Bilan à retourner */ + GCacheContainerIface *iface; /* Interface utilisée */ + + assert(g_cache_container_is_locked(container)); + + iface = G_CACHE_CONTAINER_GET_IFACE(container); + + result = iface->can_store(container); + + return result; + +} diff --git a/src/analysis/storage/container.h b/src/analysis/storage/container.h new file mode 100644 index 0000000..b85793d --- /dev/null +++ b/src/analysis/storage/container.h @@ -0,0 +1,66 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * container.h - prototypes pour les conteneurs d'objets entreposables dans un cache + * + * Copyright (C) 2020 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/>. + */ + + +#ifndef _ANALYSIS_STORAGE_CONTAINER_H +#define _ANALYSIS_STORAGE_CONTAINER_H + + +#include <glib-object.h> + + +#include "../../common/packed.h" + + + +#define G_TYPE_CACHE_CONTAINER g_cache_container_get_type() +#define G_CACHE_CONTAINER(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), G_TYPE_CACHE_CONTAINER, GCacheContainer)) +#define G_CACHE_CONTAINER_CLASS(vtable) (G_TYPE_CHECK_CLASS_CAST((vtable), G_TYPE_CACHE_CONTAINER, GCacheContainerIface)) +#define G_IS_CACHE_CONTAINER(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), G_TYPE_CACHE_CONTAINER)) +#define G_IS_CACHE_CONTAINER_CLASS(vtable) (G_TYPE_CHECK_CLASS_TYPE((vtable), G_TYPE_CACHE_CONTAINER)) +#define G_CACHE_CONTAINER_GET_IFACE(inst) (G_TYPE_INSTANCE_GET_INTERFACE((inst), G_TYPE_CACHE_CONTAINER, GCacheContainerIface)) + + +/* Intermédiaire pour un conteneur d'objets entreposables (coquille vide) */ +typedef struct _GCacheContainer GCacheContainer; + +/* Intermédiaire pour un conteneur d'objets entreposables (interface) */ +typedef struct _GCacheContainerIface GCacheContainerIface; + + +/* Détermine le type d'une interface pour un conteneur d'objets entreposables. */ +GType g_cache_container_get_type(void) G_GNUC_CONST; + +/* Contrôle l'accès au contenu d'un conteneur. */ +void g_cache_container_lock_unlock(GCacheContainer *, bool); + +/* Détermine si le conteneur a ses accès verrouillés. */ +#ifndef NDEBUG +bool g_cache_container_is_locked(GCacheContainer *); +#endif + +/* Indique si le contenu d'un conteneur peut être mis en cache. */ +bool g_cache_container_can_store(GCacheContainer *); + + + +#endif /* _ANALYSIS_STORAGE_CONTAINER_H */ diff --git a/src/analysis/storage/serialize-int.h b/src/analysis/storage/serialize-int.h new file mode 100644 index 0000000..3a99b52 --- /dev/null +++ b/src/analysis/storage/serialize-int.h @@ -0,0 +1,55 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * serialize-int.h - définitions internes propres aux objets entreposables dans un cache + * + * Copyright (C) 2020 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/>. + */ + + +#ifndef _ANALYSIS_STORAGE_SERIALIZE_INT_H +#define _ANALYSIS_STORAGE_SERIALIZE_INT_H + + +#include "serialize.h" + + + +/* Charge un objet depuis une mémoire tampon. */ +typedef bool (* load_serializable_object_cb) (GSerializableObject *, GObjectStorage *, GTypeMemory *, packed_buffer *); + +/* Sauvegarde un objet dans une mémoire tampon. */ +typedef bool (* store_serializable_object_cb) (const GSerializableObject *, GObjectStorage *, GTypeMemory *, packed_buffer *); + + +/* Intermédiaire pour la mise en cache d'objet (interface) */ +struct _GSerializableObjectIface +{ + GTypeInterface base_iface; /* A laisser en premier */ + + load_serializable_object_cb load; /* Chargement */ + store_serializable_object_cb store; /* Enregistrement */ + +}; + + +/* Redéfinition */ +typedef GSerializableObjectIface GSerializableObjectInterface; + + + +#endif /* _ANALYSIS_STORAGE_SERIALIZE_INT_H */ diff --git a/src/analysis/storage/serialize.c b/src/analysis/storage/serialize.c new file mode 100644 index 0000000..6ed1eab --- /dev/null +++ b/src/analysis/storage/serialize.c @@ -0,0 +1,119 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * serialize.h - objets entreposables dans un cache + * + * Copyright (C) 2020 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 "serialize.h" + + +#include "serialize-int.h" + + + +/* Procède à l'initialisation de l'interface de mise en cache. */ +static void g_serializable_object_default_init(GSerializableObjectInterface *); + + + +/* Détermine le type d'une interface pour la mise en cache d'objet. */ +G_DEFINE_INTERFACE(GSerializableObject, g_serializable_object, G_TYPE_OBJECT) + + +/****************************************************************************** +* * +* Paramètres : iface = interface GLib à initialiser. * +* * +* Description : Procède à l'initialisation de l'interface de mise en cache. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_serializable_object_default_init(GSerializableObjectInterface *iface) +{ + +} + + +/****************************************************************************** +* * +* Paramètres : object = instruction d'assemblage à consulter. * +* storage = conservateur de données à manipuler ou NULL. * +* tpmem = mémoire des types d'objets à compléter. * +* pbuf = zone tampon à remplir. * +* * +* Description : Charge un objet depuis une mémoire tampon. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_serializable_object_load(GSerializableObject *object, GObjectStorage *storage, GTypeMemory *tpmem, packed_buffer *pbuf) +{ + bool result; /* Bilan à retourner */ + GSerializableObjectIface *iface; /* Interface utilisée */ + + iface = G_SERIALIZABLE_OBJECT_GET_IFACE(object); + + result = iface->load(object, storage, tpmem, pbuf); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : object = instruction d'assemblage à consulter. * +* storage = conservateur de données à manipuler ou NULL. * +* tpmem = mémoire des types d'objets à compléter. * +* pbuf = zone tampon à remplir. * +* * +* Description : Sauvegarde un objet dans une mémoire tampon. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_serializable_object_store(const GSerializableObject *object, GObjectStorage *storage, GTypeMemory *tpmem, packed_buffer *pbuf) +{ + bool result; /* Bilan à retourner */ + GSerializableObjectIface *iface; /* Interface utilisée */ + + result = g_type_memory_store_object_gtype(tpmem, G_OBJECT(object), pbuf); + + if (result) + { + iface = G_SERIALIZABLE_OBJECT_GET_IFACE(object); + + result = iface->store(object, storage, tpmem, pbuf); + + } + + return result; + +} diff --git a/src/analysis/storage/serialize.h b/src/analysis/storage/serialize.h new file mode 100644 index 0000000..d248b65 --- /dev/null +++ b/src/analysis/storage/serialize.h @@ -0,0 +1,65 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * serialize.h - prototypes pour les objets entreposables dans un cache + * + * Copyright (C) 2020 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/>. + */ + + +#ifndef _ANALYSIS_STORAGE_SERIALIZE_H +#define _ANALYSIS_STORAGE_SERIALIZE_H + + +#include <glib-object.h> + + +#include "tpmem.h" +#include "../../common/packed.h" + + + +#define G_TYPE_SERIALIZABLE_OBJECT g_serializable_object_get_type() +#define G_SERIALIZABLE_OBJECT(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), G_TYPE_SERIALIZABLE_OBJECT, GSerializableObject)) +#define G_SERIALIZABLE_OBJECT_CLASS(vtable) (G_TYPE_CHECK_CLASS_CAST((vtable), G_TYPE_SERIALIZABLE_OBJECT, GSerializableObjectIface)) +#define G_IS_SERIALIZABLE_OBJECT(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), G_TYPE_SERIALIZABLE_OBJECT)) +#define G_IS_SERIALIZABLE_OBJECT_CLASS(vtable) (G_TYPE_CHECK_CLASS_TYPE((vtable), G_TYPE_SERIALIZABLE_OBJECT)) +#define G_SERIALIZABLE_OBJECT_GET_IFACE(inst) (G_TYPE_INSTANCE_GET_INTERFACE((inst), G_TYPE_SERIALIZABLE_OBJECT, GSerializableObjectIface)) + + +/* Intermédiaire pour la mise en cache d'objet (coquille vide) */ +typedef struct _GSerializableObject GSerializableObject; + +/* Intermédiaire pour la mise en cache d'objet (interface) */ +typedef struct _GSerializableObjectIface GSerializableObjectIface; + + +/* Détermine le type d'une interface pour la mise en cache d'objet. */ +GType g_serializable_object_get_type(void) G_GNUC_CONST; + +/* storage.h : définition d'une conservation d'objets construits */ +typedef struct _GObjectStorage GObjectStorage; + +/* Charge un objet depuis une mémoire tampon. */ +bool g_serializable_object_load(GSerializableObject *, GObjectStorage *, GTypeMemory *, packed_buffer *); + +/* Sauvegarde un objet dans une mémoire tampon. */ +bool g_serializable_object_store(const GSerializableObject *, GObjectStorage *, GTypeMemory *, packed_buffer *); + + + +#endif /* _ANALYSIS_STORAGE_SERIALIZE_H */ diff --git a/src/analysis/storage/storage-int.h b/src/analysis/storage/storage-int.h new file mode 100644 index 0000000..7e01cbc --- /dev/null +++ b/src/analysis/storage/storage-int.h @@ -0,0 +1,66 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * storage.h - prototypes internes pour la conservation sur disque d'objets construits + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#ifndef _ANALYSIS_STORAGE_STORAGE_INT_H +#define _ANALYSIS_STORAGE_STORAGE_INT_H + + +#include "storage.h" + + + +/* Gestion d'enregistrements spécifiques */ +typedef struct _storage_backend_t +{ + char *name; /* Désignation du groupe */ + + char *filename; /* Nom du fichier associé */ + int fd; /* Flux d'accès correspondant */ + +} storage_backend_t; + +/* Définition d'une conservation d'objets construits (instance) */ +struct _GObjectStorage +{ + GObject parent; /* A laisser en premier */ + + GTypeMemory *tpmem; /* Mémorisation de types */ + + GLoadedContent *loaded; /* Contenu principal */ + + storage_backend_t *backends; /* Gestionnaires existants */ + size_t count; /* Quantité de gestionnaires */ + GMutex mutex; /* Contrôle d'accès à la liste */ + +}; + +/* Définition d'une conservation d'objets construits (classe) */ +struct _GObjectStorageClass +{ + GObjectClass parent; /* A laisser en premier */ + +}; + + + +#endif /* _ANALYSIS_STORAGE_STORAGE_INT_H */ diff --git a/src/analysis/storage/storage.c b/src/analysis/storage/storage.c new file mode 100644 index 0000000..d64a08e --- /dev/null +++ b/src/analysis/storage/storage.c @@ -0,0 +1,436 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * storage.c - conservation hors mémoire d'objets choisis + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#include "storage.h" + + +#include <assert.h> +#include <malloc.h> +#include <string.h> +#include <unistd.h> + + +#include "storage-int.h" +#include "../../core/logs.h" + + + +/* Initialise la classe des conservations d'objets en place. */ +static void g_object_storage_class_init(GObjectStorageClass *); + +/* Initialise une instance de conservation d'objets en place. */ +static void g_object_storage_init(GObjectStorage *); + +/* Supprime toutes les références externes. */ +static void g_object_storage_dispose(GObjectStorage *); + +/* Procède à la libération totale de la mémoire. */ +static void g_object_storage_finalize(GObjectStorage *); + +/* Retrouve l'encadrement pour un nouveau groupe d'objets. */ +static storage_backend_t *g_object_storage_find_backend(GObjectStorage *, const char *); + + + +/* Indique le type défini pour une conservation d'objets construits. */ +G_DEFINE_TYPE(GObjectStorage, g_object_storage, G_TYPE_OBJECT); + + +/****************************************************************************** +* * +* Paramètres : klass = classe à initialiser. * +* * +* Description : Initialise la classe des conservations d'objets en place. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_storage_class_init(GObjectStorageClass *klass) +{ + GObjectClass *object; /* Autre version de la classe */ + + object = G_OBJECT_CLASS(klass); + + object->dispose = (GObjectFinalizeFunc/* ! */)g_object_storage_dispose; + object->finalize = (GObjectFinalizeFunc)g_object_storage_finalize; + +} + + +/****************************************************************************** +* * +* Paramètres : storage = instance à initialiser. * +* * +* Description : Initialise une instance de conservation d'objets en place. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_storage_init(GObjectStorage *storage) +{ + storage->tpmem = g_type_memory_new(); + + storage->loaded = NULL; + + storage->backends = NULL; + storage->count = 0; + g_mutex_init(&storage->mutex); + +} + + +/****************************************************************************** +* * +* Paramètres : storage = instance d'objet GLib à traiter. * +* * +* Description : Supprime toutes les références externes. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_storage_dispose(GObjectStorage *storage) +{ + g_clear_object(&storage->tpmem); + + g_clear_object(&storage->loaded); + + G_OBJECT_CLASS(g_object_storage_parent_class)->dispose(G_OBJECT(storage)); + +} + + +/****************************************************************************** +* * +* Paramètres : storage = instance d'objet GLib à traiter. * +* * +* Description : Procède à la libération totale de la mémoire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_object_storage_finalize(GObjectStorage *storage) +{ + size_t i; /* Boucle de parcours */ + storage_backend_t *backend; /* Gestionnaire à manipuler */ + int ret; /* Bilan d'un appel */ + + g_mutex_lock(&storage->mutex); + + for (i = 0; i < storage->count; i++) + { + backend = &storage->backends[i]; + + if (backend->fd != -1) + close(backend->fd); + else + assert(false); + + ret = access(backend->filename, W_OK); + if (ret == 0) + { + ret = unlink(backend->filename); + if (ret != 0) LOG_ERROR_N("unlink"); + } + + free(backend->name); + + free(backend->filename); + + } + + if (storage->backends != NULL) + free(storage->backends); + + g_mutex_unlock(&storage->mutex); + + g_mutex_clear(&storage->mutex); + + G_OBJECT_CLASS(g_object_storage_parent_class)->finalize(G_OBJECT(storage)); + +} + + +/****************************************************************************** +* * +* Paramètres : loaded = contenu binaire à associer. * +* * +* Description : Crée le support d'une conservation d'objets en place. * +* * +* Retour : Mécanismes mis en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GObjectStorage *g_object_storage_new(GLoadedContent *loaded) +{ + GObjectStorage *result; /* Structure à retourner */ + + result = g_object_new(G_TYPE_OBJECT_STORAGE, NULL); + + result->loaded = loaded; + g_object_ref(G_OBJECT(loaded)); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : storage = gestionnaire de conservations à compléter. * +* name = désignation d'un nouveau groupe d'objets. * +* * +* Description : Retrouve l'encadrement pour un nouveau groupe d'objets. * +* * +* Retour : Informations liées à un groupe ou NULL en cas d'échec. * +* * +* Remarques : - * +* * +******************************************************************************/ + +static storage_backend_t *g_object_storage_find_backend(GObjectStorage *storage, const char *name) +{ + storage_backend_t *result; /* Encadrement à retourner */ + size_t i; /* Boucle de parcours */ + + assert(!g_mutex_trylock(&storage->mutex)); + + for (i = 0; i < storage->count; i++) + if (strcmp(storage->backends[i].name, name) == 0) + break; + + if (i == storage->count) + result = NULL; + else + result = &storage->backends[i]; + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : storage = gestionnaire de conservations à compléter. * +* name = désignation d'un nouveau groupe d'objets. * +* filename = éventuel nom de fichier à utiliser ou NULL. * +* * +* Description : Ajoute le support d'un nouveau groupe d'objets construits. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_object_storage_add_backend(GObjectStorage *storage, const char *name, const char *filename) +{ + bool result; /* Bilan à retourner */ + GBinContent *content; /* Contenu binaire traité */ + const gchar *checksum; /* Empreinte de ce contenu */ + char *prefix; /* Début de nom de fichier */ + storage_backend_t backend; /* Informations à intégrer */ + + result = false; + + g_mutex_lock(&storage->mutex); + + if (g_object_storage_find_backend(storage, name) != NULL) + goto exit; + + /* Préparatifs */ + + content = g_loaded_content_get_content(storage->loaded); + + checksum = g_binary_content_get_checksum(content); + + asprintf(&prefix, "%s-%s", checksum, name); + + g_object_unref(G_OBJECT(content)); + + backend.fd = make_tmp_file(prefix, "cache", &backend.filename); + + free(prefix); + + if (backend.fd == -1) + goto exit; + + /* Inscription en bonne et due forme */ + + backend.name = strdup(name); + + storage->backends = realloc(storage->backends, ++storage->count * sizeof(storage_backend_t)); + + storage->backends[storage->count - 1] = backend; + + exit: + + g_mutex_unlock(&storage->mutex); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : storage = gestionnaire à manipuler. * +* name = désignation d'un nouveau groupe d'objets. * +* pos = tête de lecture avant écriture. * +* * +* Description : Charge un objet à partir de données rassemblées. * +* * +* Retour : Objet restauré en mémoire ou NULL en cas d'échec. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GSerializableObject *g_object_storage_load_object(GObjectStorage *storage, const char *name, off64_t pos) +{ + GSerializableObject *result; /* Instance à retourner */ + bool status; /* Bilan d'une opération */ + storage_backend_t *backend; /* Informations à consulter */ + packed_buffer pbuf; /* Tampon des données à lire */ + off64_t new; /* Nouvelle position de lecture*/ + + result = NULL; + + /* Chargement */ + + status = false; + + g_mutex_lock(&storage->mutex); + + backend = g_object_storage_find_backend(storage, name); + + if (backend != NULL) + { + new = lseek64(backend->fd, pos, SEEK_SET); + + if (new == pos) + { + reset_packed_buffer(&pbuf); + status = read_packed_buffer(&pbuf, backend->fd); + } + + } + + g_mutex_unlock(&storage->mutex); + + if (!status) + goto exit; + + /* Phase de conversion */ + + result = G_SERIALIZABLE_OBJECT(g_type_memory_create_object(storage->tpmem, &pbuf)); + + if (result) + { + status = g_serializable_object_load(result, storage, storage->tpmem, &pbuf); + + if (!status) + g_clear_object(&result); + + } + + exit_packed_buffer(&pbuf); + + exit: + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : storage = gestionnaire à manipuler. * +* name = désignation d'un nouveau groupe d'objets. * +* pbuf = zone tampon à lire. * +* pos = tête de lecture avant écriture, ou NULL. [OUT] * +* * +* Description : Sauvegarde un object sous forme de données rassemblées. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_object_storage_store_object(GObjectStorage *storage, const char *name, const GSerializableObject *object, off64_t *pos) +{ + bool result; /* Bilan à retourner */ + packed_buffer pbuf; /* Tampon des données à écrire */ + storage_backend_t *backend; /* Informations à consulter */ + off64_t tmp; /* Conservation éphémère */ + + /* Phase de conversion */ + + init_packed_buffer(&pbuf); + + result = g_serializable_object_store(object, storage, storage->tpmem, &pbuf); + if (!result) goto exit; + + /* Enregistrement */ + + g_mutex_lock(&storage->mutex); + + backend = g_object_storage_find_backend(storage, name); + + if (backend != NULL) + { + if (pos == NULL) + pos = &tmp; + + *pos = lseek64(backend->fd, 0, SEEK_CUR); + + if (*pos != (off64_t)-1) + result = write_packed_buffer(&pbuf, backend->fd); + + } + + g_mutex_unlock(&storage->mutex); + + /* Sortie propre */ + + exit: + + exit_packed_buffer(&pbuf); + + return result; + +} diff --git a/src/analysis/storage/storage.h b/src/analysis/storage/storage.h new file mode 100644 index 0000000..611e49e --- /dev/null +++ b/src/analysis/storage/storage.h @@ -0,0 +1,70 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * storage.h - prototypes pour la conservation sur disque d'objets construits + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#ifndef _ANALYSIS_STORAGE_STORAGE_H +#define _ANALYSIS_STORAGE_STORAGE_H + + +#include <glib-object.h> +#include <stdbool.h> + + +#include "serialize.h" +#include "tpmem.h" +#include "../loaded.h" + + + +#define G_TYPE_OBJECT_STORAGE g_object_storage_get_type() +#define G_OBJECT_STORAGE(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), G_TYPE_OBJECT_STORAGE, GObjectStorage)) +#define G_IS_OBJECT_STORAGE(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), G_TYPE_OBJECT_STORAGE)) +#define G_OBJECT_STORAGE_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), G_TYPE_OBJECT_STORAGE, GObjectStorageClass)) +#define G_IS_OBJECT_STORAGE_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), G_TYPE_OBJECT_STORAGE)) +#define G_OBJECT_STORAGE_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), G_TYPE_OBJECT_STORAGE, GObjectStorageClass)) + + +/* Définition d'une conservation d'objets construits (instance) */ +typedef struct _GObjectStorage GObjectStorage; + +/* Définition d'une conservation d'objets construits (classe) */ +typedef struct _GObjectStorageClass GObjectStorageClass; + + +/* Indique le type défini pour une conservation d'objets construits. */ +GType g_object_storage_get_type(void); + +/* Crée le support d'une conservation d'objets en place. */ +GObjectStorage *g_object_storage_new(GLoadedContent *); + +/* Ajoute le support d'un nouveau groupe d'objets construits. */ +bool g_object_storage_add_backend(GObjectStorage *, const char *, const char *); + +/* Charge un objet à partir de données rassemblées. */ +GSerializableObject *g_object_storage_load_object(GObjectStorage *, const char *, off64_t); + +/* Sauvegarde un object sous forme de données rassemblées. */ +bool g_object_storage_store_object(GObjectStorage *, const char *, const GSerializableObject *, off64_t *); + + + +#endif /* _ANALYSIS_STORAGE_STORAGE_H */ diff --git a/src/analysis/storage/tpmem.c b/src/analysis/storage/tpmem.c new file mode 100644 index 0000000..64f54d2 --- /dev/null +++ b/src/analysis/storage/tpmem.c @@ -0,0 +1,454 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * tpmem.c - mémorisation des types d'objets mis en cache + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#include "tpmem.h" + + +#include <assert.h> +#include <stdint.h> + + +#include "../../arch/operands/target.h" +#include "../../core/logs.h" + + + +/* Conservation d'une référence sur un type */ +typedef struct _gtype_ref_info_t +{ + GType gtype; /* Type pour la GLib */ + gpointer gclass; /* Lien vers sa classe */ + + /** + * La GLib n'est pas très claire sur la taille de GType : + * + * #if GLIB_SIZEOF_SIZE_T != GLIB_SIZEOF_LONG || !defined __cplusplus + * typedef gsize GType; + * #else // for historic reasons, C++ links against gulong GTypes + * typedef gulong GType; + * #endif + * + * Et : + * + * typedef unsigned $glib_size_type_define gsize; + * + * On prend donc le parti de conserver ces types sous forme de valeurs 64 bits + * lors des enregistrements. + */ + +} gtype_ref_info_t; + +/* Définition d'une mémoire de types d'objets (instance) */ +struct _GTypeMemory +{ + GObject parent; /* A laisser en premier */ + + gtype_ref_info_t *gtypes; /* Types des objets reconnus */ + uint64_t count; /* Quantité de ces objets */ + GMutex mutex; /* Contrôle d'accès à la liste */ + +}; + +/* Définition d'une mémoire de types d'objets (classe) */ +struct _GTypeMemoryClass +{ + GObjectClass parent; /* A laisser en premier */ + +}; + + +/* Initialise la classe des mémoires de types d'objets. */ +static void g_type_memory_class_init(GTypeMemoryClass *); + +/* Initialise une instance de mémoire de types d'objets. */ +static void g_type_memory_init(GTypeMemory *); + +/* Supprime toutes les références externes. */ +static void g_type_memory_dispose(GTypeMemory *); + +/* Procède à la libération totale de la mémoire. */ +static void g_type_memory_finalize(GTypeMemory *); + + + +/* Indique le type défini pour une mémoire de types d'objets. */ +G_DEFINE_TYPE(GTypeMemory, g_type_memory, G_TYPE_OBJECT); + + +/****************************************************************************** +* * +* Paramètres : klass = classe à initialiser. * +* * +* Description : Initialise la classe des mémoires de types d'objets. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_type_memory_class_init(GTypeMemoryClass *klass) +{ + GObjectClass *object; /* Autre version de la classe */ + + object = G_OBJECT_CLASS(klass); + + object->dispose = (GObjectFinalizeFunc/* ! */)g_type_memory_dispose; + object->finalize = (GObjectFinalizeFunc)g_type_memory_finalize; + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = instance à initialiser. * +* * +* Description : Initialise une instance de mémoire de types d'objets. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_type_memory_init(GTypeMemory *tpmem) +{ + tpmem->gtypes = NULL; + tpmem->count = 0; + g_mutex_init(&tpmem->mutex); + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = instance d'objet GLib à traiter. * +* * +* Description : Supprime toutes les références externes. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_type_memory_dispose(GTypeMemory *tpmem) +{ + uint64_t i; /* Boucle de parcours */ + + g_mutex_lock(&tpmem->mutex); + + for (i = 0; i < tpmem->count; i++) + if (tpmem->gtypes[i].gclass != NULL) + g_type_class_unref(tpmem->gtypes[i].gclass); + + g_mutex_unlock(&tpmem->mutex); + + g_mutex_clear(&tpmem->mutex); + + G_OBJECT_CLASS(g_type_memory_parent_class)->dispose(G_OBJECT(tpmem)); + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = instance d'objet GLib à traiter. * +* * +* Description : Procède à la libération totale de la mémoire. * +* * +* Retour : - * +* * +* Remarques : - * +* * +******************************************************************************/ + +static void g_type_memory_finalize(GTypeMemory *tpmem) +{ + if (tpmem->gtypes != NULL) + free(tpmem->gtypes); + + G_OBJECT_CLASS(g_type_memory_parent_class)->finalize(G_OBJECT(tpmem)); + +} + + +/****************************************************************************** +* * +* Paramètres : - * +* * +* Description : Crée une mémoire pour types d'objets. * +* * +* Retour : Instance mise en place. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GTypeMemory *g_type_memory_new(void) +{ + GTypeMemory *result; /* Structure à retourner */ + + result = g_object_new(G_TYPE_TYPE_MEMORY, NULL); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = mémoire à compléter. * +* fd = flux ouvert en lecture. * +* * +* Description : Apprend tous les types mémorisés dans un flux. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_type_memory_read_types(GTypeMemory *tpmem, int fd) +{ + bool result; /* Bilan à enregistrer */ + packed_buffer pbuf; /* Tampon des données à écrire */ + uint64_t i; /* Boucle de parcours */ + unsigned char len; /* Taille d'un nom de type */ + char *name; /* Désignation d'un type */ + + init_packed_buffer(&pbuf); + + result = read_packed_buffer(&pbuf, fd); + + if (result) + { + g_mutex_lock(&tpmem->mutex); + + result = extract_packed_buffer(&pbuf, &tpmem->count, sizeof(uint64_t), true); + + if (result) + { + assert(tpmem->gtypes == NULL); + tpmem->gtypes = calloc(tpmem->count, sizeof(gtype_ref_info_t)); + } + + for (i = 0; i < tpmem->count && result; i++) + { + result = extract_packed_buffer(&pbuf, &len, sizeof(unsigned char), false); + + if (result) + { + name = malloc(len); + + result = extract_packed_buffer(&pbuf, name, len, false); + + if (result) + { + tpmem->gtypes[i].gtype = g_type_from_name(name); + result = (tpmem->gtypes[i].gtype != 0); + + if (!result) + log_variadic_message(LMT_ERROR, "Unknown type: '%s'", name); + + } + + if (result) + tpmem->gtypes[i].gclass = g_type_class_ref(tpmem->gtypes[i].gtype); + + free(name); + + } + + } + + g_mutex_unlock(&tpmem->mutex); + + } + + exit_packed_buffer(&pbuf); + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = mémoire à manipuler. * +* pbuf = zone tampon à venir lire. * +* * +* Description : Crée une nouvelle instance d'objet à partir de son type. * +* * +* Retour : Instance issue de l'opération ou NULL. * +* * +* Remarques : - * +* * +******************************************************************************/ + +GObject *g_type_memory_create_object(GTypeMemory *tpmem, packed_buffer *pbuf) +{ + GObject *result; /* Nouvelle instance à renvoyer*/ + uint64_t index; /* Indice du point d'insertion */ + bool status; /* Bilan d'une récupération */ + + result = NULL; + + status = extract_packed_buffer(pbuf, &index, sizeof(uint64_t), true); + + if (status) + { + g_mutex_lock(&tpmem->mutex); + + if (index < tpmem->count) + result = g_object_new(tpmem->gtypes[index].gtype, NULL); + + g_mutex_unlock(&tpmem->mutex); + + } + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = mémoire à manipuler. * +* obj = instance dont le type est à mémoriser. * +* pbuf = zone tampon à remplir. * +* * +* Description : Sauvegarde le type d'un objet instancié. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_type_memory_store_object_gtype(GTypeMemory *tpmem, GObject *obj, packed_buffer *pbuf) +{ + bool result; /* Bilan à retourner */ + GType gtype; /* Type à enregistrer */ + uint64_t index; /* Indice du point d'insertion */ + + gtype = G_TYPE_FROM_INSTANCE(obj); + + /** + * Pour quelques explications sur l'esquive suivante, se rapporter aux + * commentaires de g_target_operand_unserialize(). + * + * Dans la situation présente, on ne doit pas enregistrer le type dans le tampon, + * car l'opérande va relancer l'opération entière (avec un opérande temporaire), + * ce qui conduirait à l'enregistrement de deux types successifs dans les données. + */ + + if (gtype == G_TYPE_TARGET_OPERAND) + result = true; + + else + { + g_mutex_lock(&tpmem->mutex); + + for (index = 0; index < tpmem->count; index++) + if (tpmem->gtypes[index].gtype == gtype) + break; + + if (index == tpmem->count) + { + tpmem->gtypes = realloc(tpmem->gtypes, ++tpmem->count * sizeof(gtype_ref_info_t)); + + assert(tpmem->count > 0); + + tpmem->gtypes[index].gtype = gtype; + tpmem->gtypes[index].gclass = g_type_class_ref(gtype); + + } + + g_mutex_unlock(&tpmem->mutex); + + result = extend_packed_buffer(pbuf, &index, sizeof(uint64_t), true); + + } + + return result; + +} + + +/****************************************************************************** +* * +* Paramètres : tpmem = mémoire à consulter. * +* fd = flux ouvert en écriture. * +* * +* Description : Enregistre tous les types mémorisés dans un flux. * +* * +* Retour : Bilan de l'opération. * +* * +* Remarques : - * +* * +******************************************************************************/ + +bool g_type_memory_write_types(GTypeMemory *tpmem, int fd) +{ + bool result; /* Bilan à enregistrer */ + packed_buffer pbuf; /* Tampon des données à écrire */ + uint64_t i; /* Boucle de parcours */ + const gchar *name; /* Désignation d'un type */ + size_t len; /* Taille de ce nom */ + + init_packed_buffer(&pbuf); + + g_mutex_lock(&tpmem->mutex); + + result = extend_packed_buffer(&pbuf, &tpmem->count, sizeof(uint64_t), true); + + for (i = 0; i < tpmem->count && result; i++) + { + name = g_type_name(tpmem->gtypes[i].gtype); + len = strlen(name) + 1; + + if (len > (2 << (sizeof(unsigned char) * 8 - 1))) + { + log_variadic_message(LMT_ERROR, "Type name too long: '%s' (%zu bytes)", name, len); + result = false; + break; + } + + result = extend_packed_buffer(&pbuf, (unsigned char []) { len }, sizeof(unsigned char), false); + + if (result) + result = extend_packed_buffer(&pbuf, name, len, false); + + } + + if (result) + result = write_packed_buffer(&pbuf, fd); + + g_mutex_unlock(&tpmem->mutex); + + exit_packed_buffer(&pbuf); + + return result; + +} diff --git a/src/analysis/storage/tpmem.h b/src/analysis/storage/tpmem.h new file mode 100644 index 0000000..4c82a1f --- /dev/null +++ b/src/analysis/storage/tpmem.h @@ -0,0 +1,70 @@ + +/* Chrysalide - Outil d'analyse de fichiers binaires + * tpmem.h - prototypes pour la mémorisation des types d'objets mis en cache + * + * Copyright (C) 2020 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 Foobar. If not, see <http://www.gnu.org/licenses/>. + */ + + +#ifndef _ANALYSIS_STORAGE_TPMEM_H +#define _ANALYSIS_STORAGE_TPMEM_H + + +#include <glib-object.h> + + +#include "../../common/packed.h" + + + +#define G_TYPE_TYPE_MEMORY g_type_memory_get_type() +#define G_TYPE_MEMORY(obj) (G_TYPE_CHECK_INSTANCE_CAST((obj), G_TYPE_TYPE_MEMORY, GTypeMemory)) +#define G_IS_TYPE_MEMORY(obj) (G_TYPE_CHECK_INSTANCE_TYPE((obj), G_TYPE_TYPE_MEMORY)) +#define G_TYPE_MEMORY_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST((klass), G_TYPE_TYPE_MEMORY, GTypeMemoryClass)) +#define G_IS_TYPE_MEMORY_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE((klass), G_TYPE_TYPE_MEMORY)) +#define G_TYPE_MEMORY_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS((obj), G_TYPE_TYPE_MEMORY, GTypeMemoryClass)) + + +/* Définition d'une mémoire de types d'objets (instance) */ +typedef struct _GTypeMemory GTypeMemory; + +/* Définition d'une mémoire de types d'objets (classe) */ +typedef struct _GTypeMemoryClass GTypeMemoryClass; + + +/* Indique le type défini pour une mémoire de types d'objets. */ +GType g_type_memory_get_type(void); + +/* Crée une mémoire pour types d'objets. */ +GTypeMemory *g_type_memory_new(void); + +/* Apprend tous les types mémorisés dans un flux. */ +bool g_type_memory_read_types(GTypeMemory *, int); + +/* Crée une nouvelle instance d'objet à partir de son type. */ +GObject *g_type_memory_create_object(GTypeMemory *, packed_buffer *); + +/* Sauvegarde le type d'un objet instancié. */ +bool g_type_memory_store_object_gtype(GTypeMemory *, GObject *, packed_buffer *); + +/* Enregistre tous les types mémorisés dans un flux. */ +bool g_type_memory_write_types(GTypeMemory *, int); + + + +#endif /* _ANALYSIS_STORAGE_TPMEM_H */ |