/* 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;

}