/* Chrysalide - Outil d'analyse de fichiers binaires
* export.c - assistant d'exportation de contenu binaire
*
* Copyright (C) 2010-2014 Cyrille Bagard
*
* This file is part of Chrysalide.
*
* OpenIDA 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.
*
* OpenIDA 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 .
*/
#include "export.h"
#include
#include
#include
#include
#include
#include "../gtkext/easygtk.h"
/* ------------------------ PARTIE PRINCIPALE DE L'ASSISTANT ------------------------ */
/* Conservation des informations utiles */
struct _export_data
{
int fd; /* Flux ouvert en écriture */
BufferExportType type; /* Type d'exportation */
bool addr; /* Affichage des adresses */
bool code; /* Affichage du code binaire */
bool content; /* Affichage du contenu humain */
};
/* Ferme l'assistant sans dérouler la procédure. */
static void export_assistant_cancel(GtkAssistant *, gpointer);
/* Ferme l'assistant et déroule la procédure. */
static void export_assistant_close(GtkAssistant *, GObject *);
/* Assure l'exportation en différé. */
static bool export_buffer_line(GCodeBuffer *, GBufferLine *, struct _export_data *);
/* -------------------- DEFINITION DES REGLAGES DE L'EXPORTATION -------------------- */
/* Ajoute le panneau de choix du type de sortie. */
static void register_output_panel(GtkAssistant *);
/* Réagit un changement du nom de fichier pour l'exportation. */
static void on_export_filename_changed(GtkEntry *, GtkAssistant *);
/* Sélectionne ou non un nouveau fichier de sortie. */
static void on_filename_browsing_clicked(GtkButton *, GObject *);
/* ------------------------- SELECTION DU CONTENU A TRAITER ------------------------- */
/* Ajoute le panneau de sélection du contenu à exporter. */
static void register_content_panel(GtkAssistant *);
/* ---------------------------------------------------------------------------------- */
/* PARTIE PRINCIPALE DE L'ASSISTANT */
/* ---------------------------------------------------------------------------------- */
/******************************************************************************
* *
* Paramètres : binary = binaire chargé en mémoire à traiter. *
* parent = fenêtre principale de l'éditeur. *
* *
* Description : Crée et affiche un assistant d'aide à l'exportation. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
void run_export_assistant(GLoadedBinary *binary, GtkWindow *parent)
{
GtkWidget *assistant; /* Fenêtre à afficher */
GObject *ref; /* Espace de référencement */
assistant = gtk_assistant_new();
gtk_widget_set_size_request(assistant, 450, 300);
gtk_window_set_position(GTK_WINDOW(assistant), GTK_WIN_POS_CENTER);
gtk_window_set_title(GTK_WINDOW(assistant), _("Export assistant"));
gtk_window_set_modal(GTK_WINDOW(assistant), TRUE);
gtk_window_set_transient_for(GTK_WINDOW(assistant), parent);
ref = G_OBJECT(assistant);
g_object_set_data(ref, "binary", binary);
register_output_panel(GTK_ASSISTANT(assistant));
register_content_panel(GTK_ASSISTANT(assistant));
g_signal_connect(G_OBJECT(assistant), "cancel", G_CALLBACK(export_assistant_cancel), NULL);
g_signal_connect(G_OBJECT(assistant), "close", G_CALLBACK(export_assistant_close), ref);
gtk_widget_show_all(assistant);
}
/******************************************************************************
* *
* Paramètres : assistant = fenêtre à compléter et référencement global. *
* data = adresse non utilisée ici. *
* *
* Description : Ferme l'assistant sans dérouler la procédure. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void export_assistant_cancel(GtkAssistant *assistant, gpointer data)
{
gtk_widget_destroy(GTK_WIDGET(assistant));
}
/******************************************************************************
* *
* Paramètres : assistant = fenêtre à compléter et référencement global. *
* ref = adresse de l'espace de référencement global. *
* *
* Description : Ferme l'assistant et déroule la procédure. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void export_assistant_close(GtkAssistant *assistant, GObject *ref)
{
GtkEntry *entry; /* Zone de saisie */
const gchar *filename; /* Chemin d'accès du fichier */
int fd; /* Descripteur de la sortie */
struct _export_data *export; /* Informations à faire suivre */
GtkToggleButton *checkbutton; /* Coche à retrouver */
GLoadedBinary *binary; /* Binaire chargé à parcourir */
GCodeBuffer *buffer; /* Tampon de code à traiter */
/* Fichier de sortie */
entry = GTK_ENTRY(g_object_get_data(ref, "filename"));
filename = gtk_entry_get_text(entry);
fd = open(filename, O_CREAT | O_WRONLY, S_IRUSR | S_IWUSR);
if (fd == -1)
{
perror("open");
return;
}
export = (struct _export_data *)calloc(1, sizeof(struct _export_data));
export->fd = fd;
/* Type d'exportation */
export->type = BET_TEXT;
/* Eléments à afficher */
checkbutton = GTK_TOGGLE_BUTTON(g_object_get_data(ref, "virtual_addr"));
export->addr = gtk_toggle_button_get_active(checkbutton);
checkbutton = GTK_TOGGLE_BUTTON(g_object_get_data(ref, "binary_code"));
export->code = gtk_toggle_button_get_active(checkbutton);
checkbutton = GTK_TOGGLE_BUTTON(g_object_get_data(ref, "assembly_code"));
export->content = gtk_toggle_button_get_active(checkbutton);
/* Programmation de la tâche */
binary = G_LOADED_BINARY(g_object_get_data(ref, "binary"));
buffer = g_loaded_binary_get_disassembled_buffer(binary);
g_buffer_code_scan(buffer, 0, VMPA_MAX, _("Exporting binary lines..."),
(process_line_fc)export_buffer_line, export);
gtk_widget_destroy(GTK_WIDGET(assistant));
}
/******************************************************************************
* *
* Paramètres : buffer = tampon de code contenant toutes les lignes. *
* line = ligne dont le contenu est à exporter. *
* export = informations utiles à la manoeuvre. *
* *
* Description : Assure l'exportation en différé. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static bool export_buffer_line(GCodeBuffer *buffer, GBufferLine *line, struct _export_data *export)
{
/* Si les traitements sont terminés... */
if (line == NULL)
{
close(export->fd);
free(export);
}
else
g_buffer_line_export(line, export->fd, export->type,
export->addr, export->code, export->content);
return true;
}
/* ---------------------------------------------------------------------------------- */
/* DEFINITION DES REGLAGES DE L'EXPORTATION */
/* ---------------------------------------------------------------------------------- */
/******************************************************************************
* *
* Paramètres : assistant = fenêtre à compléter et référencement global. *
* *
* Description : Ajoute le panneau de choix du type de sortie. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void register_output_panel(GtkAssistant *assistant)
{
GtkWidget *alignment; /* Disposition sur le support */
GtkWidget *vbox; /* Support principal #1 */
GtkWidget *hbox; /* Support principal #2 */
GtkWidget *label; /* Etiquette d'indication */
GtkWidget *combobox; /* Sélection du format */
GtkWidget *entry; /* Zone de saisie de texte */
GtkWidget *button; /* Sélection de fichier */
GLoadedBinary *binary; /* Binaire chargé à parcourir */
const char *filename; /* Chemin d'accès par défaut */
alignment = qck_create_padded_alignment(8, 8, 8, 8);
vbox = gtk_vbox_new(TRUE, 0);
gtk_widget_show(vbox);
gtk_container_add(GTK_CONTAINER(alignment), vbox);
/* Format de sortie */
hbox = gtk_hbox_new(FALSE, 0);
gtk_widget_show(hbox);
gtk_box_pack_start(GTK_BOX(vbox), hbox, FALSE, FALSE, 0);
label = qck_create_label(NULL, NULL, _("Format: "));
gtk_box_pack_start(GTK_BOX(hbox), label, FALSE, FALSE, 0);
combobox = qck_create_combobox(NULL, NULL, G_CALLBACK(NULL), NULL);
gtk_box_pack_start(GTK_BOX(hbox), combobox, TRUE, TRUE, 0);
gtk_combo_box_text_append_text(GTK_COMBO_BOX_TEXT(combobox), _("Simple text"));
gtk_combo_box_set_active(GTK_COMBO_BOX(combobox), 0);
/* Fichier de sortie */
hbox = gtk_hbox_new(FALSE, 0);
gtk_widget_show(hbox);
gtk_box_pack_start(GTK_BOX(vbox), hbox, FALSE, FALSE, 0);
label = qck_create_label(NULL, NULL, _("File: "));
gtk_box_pack_start(GTK_BOX(hbox), label, FALSE, FALSE, 0);
entry = qck_create_entry(G_OBJECT(assistant), "filename", NULL);
gtk_box_pack_start(GTK_BOX(hbox), entry, TRUE, TRUE, 0);
button = qck_create_button(NULL, NULL, "...", G_CALLBACK(on_filename_browsing_clicked), assistant);
gtk_box_pack_start(GTK_BOX(hbox), button, FALSE, FALSE, 0);
/* Intégration */
gtk_assistant_append_page(assistant, alignment);
gtk_assistant_set_page_title(assistant, alignment, _("Output"));
gtk_assistant_set_page_type(assistant, alignment, GTK_ASSISTANT_PAGE_INTRO);
gtk_assistant_set_page_complete(assistant, alignment, TRUE);
/* Choix par défaut */
binary = G_LOADED_BINARY(g_object_get_data(G_OBJECT(assistant), "binary"));
filename = g_loaded_binary_get_name(binary, true);
gtk_entry_set_text(GTK_ENTRY(entry), filename);
gtk_editable_insert_text(GTK_EDITABLE(entry), ".txt", -1, (gint []) { strlen(filename) });
g_signal_connect(G_OBJECT(entry), "changed", G_CALLBACK(on_export_filename_changed), assistant);
}
/******************************************************************************
* *
* Paramètres : entry = zone de texte dont le contenu vient de changer. *
* assistant = fenêtre affichée et référencement global. *
* *
* Description : Réagit un changement du nom de fichier pour l'exportation. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void on_export_filename_changed(GtkEntry *entry, GtkAssistant *assistant)
{
const gchar *text; /* Texte saisi dans la zone */
gint num; /* Etape courante */
GtkWidget *page; /* Support de cette étape */
text = gtk_entry_get_text(entry);
num = gtk_assistant_get_current_page(assistant);
page = gtk_assistant_get_nth_page(assistant, num);
gtk_assistant_set_page_complete(assistant, page, (strlen(text) > 0));
}
/******************************************************************************
* *
* Paramètres : button = bouton d'édition de la sélection. *
* ref = espace de référencement principal. *
* *
* Description : Sélectionne ou non un nouveau fichier de sortie. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void on_filename_browsing_clicked(GtkButton *button, GObject *ref)
{
GtkWidget *dialog; /* Boîte à afficher */
gchar *filename; /* Nom du fichier à intégrer */
GtkEntry *entry; /* Zone de saisie à maj. */
dialog = gtk_file_chooser_dialog_new(_("Choose an output filename"), GTK_WINDOW(ref),
GTK_FILE_CHOOSER_ACTION_SAVE,
GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL,
GTK_STOCK_OPEN, GTK_RESPONSE_ACCEPT,
NULL);
entry = GTK_ENTRY(g_object_get_data(ref, "filename"));
gtk_file_chooser_set_filename(GTK_FILE_CHOOSER(dialog), gtk_entry_get_text(entry));
if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT)
{
filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog));
gtk_entry_set_text(GTK_ENTRY(entry), filename);
g_free(filename);
}
gtk_widget_destroy(dialog);
}
/* ---------------------------------------------------------------------------------- */
/* SELECTION DU CONTENU A TRAITER */
/* ---------------------------------------------------------------------------------- */
/******************************************************************************
* *
* Paramètres : assistant = fenêtre à compléter et référencement global. *
* *
* Description : Ajoute le panneau de sélection du contenu à exporter. *
* *
* Retour : - *
* *
* Remarques : - *
* *
******************************************************************************/
static void register_content_panel(GtkAssistant *assistant)
{
GtkWidget *alignment; /* Disposition sur le support */
GtkWidget *vbox; /* Support principal */
GtkWidget *frame; /* Support avec encadrement */
GtkWidget *subalign; /* Disposition des options */
GtkWidget *sub_vbox; /* Division verticale */
GtkWidget *checkbutton; /* Coche pour une option */
alignment = qck_create_padded_alignment(8, 8, 8, 8);
vbox = gtk_vbox_new(TRUE, 0);
gtk_widget_show(vbox);
gtk_container_add(GTK_CONTAINER(alignment), vbox);
/* Eléments à afficher */
frame = qck_create_frame(_("Items to display"), &subalign, 0, 0, 12, 0);
gtk_box_pack_start(GTK_BOX(vbox), frame, FALSE, FALSE, 0);
sub_vbox = gtk_vbox_new(FALSE, 0);
gtk_widget_show(sub_vbox);
gtk_container_add(GTK_CONTAINER(subalign), sub_vbox);
checkbutton = qck_create_check_button(G_OBJECT(assistant), "virtual_addr", _("Virtual address"), NULL, NULL);
gtk_box_pack_start(GTK_BOX(sub_vbox), checkbutton, FALSE, FALSE, 0);
gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(checkbutton), TRUE);
checkbutton = qck_create_check_button(G_OBJECT(assistant), "binary_code", _("Binary code"), NULL, NULL);
gtk_box_pack_start(GTK_BOX(sub_vbox), checkbutton, FALSE, FALSE, 0);
gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(checkbutton), TRUE);
checkbutton = qck_create_check_button(G_OBJECT(assistant), "assembly_code", _("Assembly code"), NULL, NULL);
gtk_box_pack_start(GTK_BOX(sub_vbox), checkbutton, FALSE, FALSE, 0);
gtk_toggle_button_set_active(GTK_TOGGLE_BUTTON(checkbutton), TRUE);
/* Intégration */
gtk_assistant_append_page(assistant, alignment);
gtk_assistant_set_page_title(assistant, alignment, _("Exported content"));
gtk_assistant_set_page_type(assistant, alignment, GTK_ASSISTANT_PAGE_CONFIRM);
gtk_assistant_set_page_complete(assistant, alignment, TRUE);
}