/* Chrysalide - Outil d'analyse de fichiers binaires
 * raw-ui.c - opérandes représentant des instructions de données brutes sous forme graphique
 *
 * Copyright (C) 2025 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 "raw-ui.h"


#include <assert.h>
#include <ctype.h>
#include <malloc.h>


#include "raw.h"
#include "../operand-ui.h"
#include "../operands/immediate.h"
#include "../../glibext/objhole.h"
#include "../../glibext/options/asm.h"



/* Etablit dans une ligne de rendu le contenu représenté. */
static void g_raw_instruction_ui_populate_line(const GTokenGenerator *, size_t, size_t, GBufferLine *, void *);



/******************************************************************************
*                                                                             *
*  Paramètres  : iface = interface GLib à initialiser.                        *
*                                                                             *
*  Description : Procède à l'initialisation de l'interface de génération.     *
*                                                                             *
*  Retour      : -                                                            *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

void g_raw_instruction_ui_token_generator_iface_init(GTokenGeneratorInterface *iface)
{
    iface->populate = g_raw_instruction_ui_populate_line;

}


/******************************************************************************
*                                                                             *
*  Paramètres  : generator = générateur à utiliser pour l'impression.         *
*                index     = indice de cette même ligne dans le tampon global.*
*                repeat    = indice d'utilisations successives du générateur. *
*                line      = ligne de rendu à compléter.                      *
*                data      = éventuelle donnée complémentaire fournie.        *
*                                                                             *
*  Description : Etablit dans une ligne de rendu le contenu représenté.       *
*                                                                             *
*  Retour      : -                                                            *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

static void g_raw_instruction_ui_populate_line(const GTokenGenerator *generator, size_t index, size_t repeat, GBufferLine *line, void *data)
{
    GArchInstruction *instr;                /* Version spécialisée #1      */
    GRawInstruction *raw;                   /* Version spécialisée #2      */
    GBinContent *content;                   /* Contenu brut d'origine      */
    mrange_t range;                         /* Emplacement couvert         */
    phys_t max_displayed_len;               /* Quantité de code affichée   */
    char *key;                              /* Mot clef principal          */
    char *string;                           /* Chaîne reconstituée         */
    size_t iter;                            /* Tête d'écriture             */
    bool first;                             /* Mémorise une énumération    */
    size_t count;                           /* Nombre d'opérandes en place */
    size_t i;                               /* Boucle de parcours          */
    GArchOperand *op;                       /* Opérande à manipuler        */
    GImmediateOperand *imm;                 /* Version opérande de valeur  */
    char byte;                              /* Octet à afficher (ou pas)   */
#ifndef NDEBUG
    bool status;                            /* Bilan d'une récupération    */
#endif

    instr = G_ARCH_INSTRUCTION(generator);
    raw = G_RAW_INSTRUCTION(instr);
    content = G_BIN_CONTENT(data);

    g_thick_object_lock(G_THICK_OBJECT(instr));

    /* Prologue */

    if (g_arch_instruction_get_range(instr, &range))
    {
        /* Localisation */

        g_buffer_line_fill_physical(line, ACO_PHYSICAL, MDS_32_BITS_UNSIGNED, get_mrange_addr(&range));

        g_buffer_line_fill_virtual(line, ACO_VIRTUAL, MDS_32_BITS_UNSIGNED, get_mrange_addr(&range));

        /* Contenu */

        if (g_raw_instruction_is_padding(raw))
            max_displayed_len = 0;

        else if (g_raw_instruction_is_string(raw))
            max_displayed_len = 1;

        else
        {
            max_displayed_len = get_mrange_length(&range);
            max_displayed_len /= g_arch_instruction_count_operands(instr);
        }

        g_buffer_line_fill_content(line, ACO_BINARY, content, &range, max_displayed_len);

    }

    /* Instruction proprement dite */

    key = g_arch_instruction_get_keyword(instr);

    g_buffer_line_append_text(line, ACO_ASSEMBLY_HEAD, TRT_INSTRUCTION, SL(key), NULL, G_OBJECT(instr));

    free(key);

    /* Contenu sous forme d'opérandes */

    if (g_raw_instruction_is_padding(raw))
        g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_CHR_PRINTABLE, STCSL("..."), NULL, NULL);

    else
    {
        string = NULL;
        iter = 0;

        first = true;

        count = g_arch_instruction_count_operands(instr);

        for (i = 0; i < count; i++)
        {
            op = g_arch_instruction_get_operand(instr, i);

            if (!G_IS_IMMEDIATE_OPERAND(op))
                goto fallback;

            imm = G_IMMEDIATE_OPERAND(op);

            if (g_immediate_operand_get_size(imm) != MDS_8_BITS)
                goto fallback;

            if (!g_raw_instruction_is_string(raw) && g_immediate_operand_get_display(imm) != IOD_CHAR)
                goto fallback;

#ifndef NDEBUG
            status = g_immediate_operand_get_value(imm, MDS_8_BITS, &byte);
            assert(status);
#else
            g_immediate_operand_get_value(imm, MDS_8_BITS, &byte);
#endif

            /* Si le caractère doit apparaître en hexadécimal... */

            if (!isprint(byte))
                goto fallback;

            /* Impression de l'octet */

            if (string == NULL)
            {
                string = calloc(count + 3, sizeof(char));

                strcpy(string, "\"");
                iter = 1;

            }

            string[iter++] = byte;

            unref_object(op);

            continue;

 fallback:

            /* Si une chaîne précède */

            if (string != NULL && iter > 1)
            {
                if (!first)
                {
                    g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_PUNCT, STCSL(","), NULL, NULL);
                    g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_CHR_PRINTABLE, STCSL(" "), NULL, NULL);
                }
                else
                    first = false;

                string[iter++] = '"';

                g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_STRING, string, iter, NULL, NULL);

                iter = 1;

            }

            /* Intégration en tant qu'opérande classique */

            if (!first)
            {
                g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_PUNCT, STCSL(","), NULL, NULL);
                g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_CHR_PRINTABLE, STCSL(" "), NULL, NULL);
            }
            else
                first = false;

            g_arch_operand_ui_print(G_ARCH_OPERAND_UI(op), line);

            unref_object(op);

        }

        /* Si au final une chaîne traine encore */

        if (string != NULL && iter > 1)
        {
            if (!first)
            {
                g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_PUNCT, STCSL(","), NULL, NULL);
                g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_CHR_PRINTABLE, STCSL(" "), NULL, NULL);
            }

            string[iter++] = '"';

            g_buffer_line_append_text(line, ACO_ASSEMBLY, TRT_STRING, string, iter, NULL, NULL);

        }

        if (string != NULL)
            free(string);

    }

    g_thick_object_unlock(G_THICK_OBJECT(instr));

}