Created helpers for dealing with archives.

3 files changed, 228 insertions, 0 deletions

diff --git a/src/common/Makefile.am b/src/common/Makefile.am
index b5b008c..7a13aa8 100644
--- a/src/common/Makefile.am
+++ b/src/common/Makefile.am
@@ -6,6 +6,7 @@ libcommon_la_SOURCES =					\
 	asm.h asm.c							\
 	bconst.h							\
 	bits.h bits.c						\
+	compression.h compression.c			\
 	cpp.h								\
 	dllist.h dllist.c					\
 	endianness.h endianness.c			\
diff --git a/src/common/compression.c b/src/common/compression.c
new file mode 100644
index 0000000..d443924
--- /dev/null
+++ b/src/common/compression.c
@@ -0,0 +1,175 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * compression.c - facilités de manipulation des archives
+ *
+ * Copyright (C) 2018 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 "compression.h"
+
+
+#include <fcntl.h>
+#include <stdio.h>
+
+
+#include "io.h"
+
+
+
+/* Fixe le tampon pour la lecture des fichiers à inclure */
+#define ARCHIVE_RBUF_SIZE 2048
+
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : output   = archive dont le contenu est à composer.           *
+*                filename = chemin d'accès au fichier d'entrée.               *
+*                path     = chemin d'accès dans l'archive.                    *
+*                                                                             *
+*  Description : Ajoute un élement à une archive.                             *
+*                                                                             *
+*  Retour      : Code de retour pour l'opération.                             *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+CPError add_file_into_archive(struct archive *output, const char *filename, const char *path)
+{
+    CPError result;                     /* Code de retour à renvoyer   */
+    struct stat info;                   /* Informations d'origine      */
+    int ret;                                /* Bilan d'un appel            */
+    struct archive_entry *entry;        /* Elément de l'archive        */
+    int fd;                             /* Flux ouvert en lecture      */
+    char buffer[ARCHIVE_RBUF_SIZE];     /* Tampon pour les transferts  */
+    ssize_t len;                        /* Quantité de données lues    */
+
+    result = CPE_ARCHIVE_ERROR;
+
+    ret = stat(filename, &info);
+    if (ret != 0)
+    {
+        perror("stat");
+        result = CPE_SYSTEM_ERROR;
+        goto afia_exit;
+    }
+
+    entry = archive_entry_new();
+
+    archive_entry_copy_stat(entry, &info);
+    archive_entry_set_pathname(entry, path);
+
+    ret = archive_write_header(output, entry);
+    if (ret != 0) goto afia_exit;
+
+    fd = open(filename, O_RDONLY);
+    if (fd == -1)
+    {
+        perror("open");
+        result = CPE_SYSTEM_ERROR;
+        goto afia_exit;
+    }
+
+    for (len = safe_read_partial(fd, buffer, ARCHIVE_RBUF_SIZE);
+         len > 0;
+         len = safe_read_partial(fd, buffer, ARCHIVE_RBUF_SIZE))
+    {
+        if (archive_write_data(output, buffer, len) != len)
+            goto afia_exit;
+    }
+
+    close(fd);
+
+    archive_entry_free(entry);
+
+    return CPE_NO_ERROR;
+
+ afia_exit:
+
+    archive_entry_free(entry);
+
+    return result;
+
+}
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : intput   = archive dont le contenu est à extraire.           *
+*                entry    = entrée de l'archive à extraire.                   *
+*                filename = chemin d'accès au fichier de sortie.              *
+*                                                                             *
+*  Description : Extrait un élement d'une archive.                            *
+*                                                                             *
+*  Retour      : Bilan de l'opération.                                        *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+bool dump_archive_entry_into_file(struct archive *input, struct archive_entry *entry, const char *filename)
+{
+    bool result;                            /* Conclusion à retourner      */
+    int flags;                              /* Propriétés à extraire       */
+    struct archive *output;                 /* Extracteur générique        */
+    int ret;                                /* Bilan d'un appel            */
+    const void *buff;                       /* Tampon de copie             */
+    size_t size;                            /* Quantité copiée             */
+    __LA_INT64_T offset;                    /* Position de lecture         */
+
+    result = false;
+
+    archive_entry_set_pathname(entry, filename);
+
+    /* Propriétés à restaurer */
+    flags = ARCHIVE_EXTRACT_TIME;
+    flags |= ARCHIVE_EXTRACT_PERM;
+    flags |= ARCHIVE_EXTRACT_ACL;
+    flags |= ARCHIVE_EXTRACT_FFLAGS;
+
+    output = archive_write_disk_new();
+    archive_write_disk_set_options(output, flags);
+    archive_write_disk_set_standard_lookup(output);
+
+    ret = archive_write_header(output, entry);
+    if (ret != ARCHIVE_OK) goto daeif_exit;
+
+    for (ret = archive_read_data_block(input, &buff, &size, &offset);
+         ret == ARCHIVE_OK;
+         ret = archive_read_data_block(input, &buff, &size, &offset))
+    {
+        ret = archive_write_data_block(output, buff, size, offset);
+    }
+
+    if (ret != ARCHIVE_EOF)
+        goto daeif_exit;
+
+    ret = archive_write_finish_entry(output);
+
+    result = (ret == ARCHIVE_OK);
+
+ daeif_exit:
+
+    archive_write_close(output);
+    archive_write_free(output);
+
+    return result;
+
+}
diff --git a/src/common/compression.h b/src/common/compression.h
new file mode 100644
index 0000000..632147a
--- /dev/null
+++ b/src/common/compression.h
@@ -0,0 +1,52 @@
+
+/* Chrysalide - Outil d'analyse de fichiers binaires
+ * compression.h - prototypes pour les facilités de manipulation des archives
+ *
+ * Copyright (C) 2018 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 _COMMON_COMPRESSION_H
+#define _COMMON_COMPRESSION_H
+
+
+#include <archive.h>
+#include <archive_entry.h>
+#include <stdbool.h>
+
+
+
+/* Codes de retour pour la compression */
+typedef enum _CPError
+{
+    CPE_NO_ERROR,                           /* Aucun souci particulier     */
+    CPE_SYSTEM_ERROR,                       /* Le soucis vient de l'archive*/
+    CPE_ARCHIVE_ERROR                       /* Le soucis vient du système  */
+
+} CPError;
+
+
+/* Ajoute un élement à une archive. */
+CPError add_file_into_archive(struct archive *, const char *, const char *);
+
+/* Extrait un élement d'une archive. */
+bool dump_archive_entry_into_file(struct archive *, struct archive_entry *, const char *);
+
+
+
+#endif  /* _COMMON_COMPRESSION_H */