From 70438138096a47b2505ac55634cd94947ce378b6 Mon Sep 17 00:00:00 2001 From: Perberos Date: Wed, 9 Nov 2011 22:53:33 -0300 Subject: initial --- src/fr-command-7z.c | 686 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 686 insertions(+) create mode 100644 src/fr-command-7z.c (limited to 'src/fr-command-7z.c') diff --git a/src/fr-command-7z.c b/src/fr-command-7z.c new file mode 100644 index 0000000..76fe48d --- /dev/null +++ b/src/fr-command-7z.c @@ -0,0 +1,686 @@ +/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */ + +/* + * File-Roller + * + * Copyright (C) 2004, 2008 Free Software Foundation, Inc. + * + * This program 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 2 of the License, or + * (at your option) any later version. + * + * This program 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 this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA. + */ + +#include +#include +#include +#include +#include + +#include +#include + +#include "file-data.h" +#include "file-utils.h" +#include "glib-utils.h" +#include "fr-command.h" +#include "fr-command-7z.h" + +static void fr_command_7z_class_init (FrCommand7zClass *class); +static void fr_command_7z_init (FrCommand *afile); +static void fr_command_7z_finalize (GObject *object); + +/* Parent Class */ + +static FrCommandClass *parent_class = NULL; + + +/* -- list -- */ + + +static time_t +mktime_from_string (char *date_s, + char *time_s) +{ + struct tm tm = {0, }; + char **fields; + + tm.tm_isdst = -1; + + /* date */ + + fields = g_strsplit (date_s, "-", 3); + if (fields[0] != NULL) { + tm.tm_year = atoi (fields[0]) - 1900; + tm.tm_mon = atoi (fields[1]) - 1; + tm.tm_mday = atoi (fields[2]); + } + g_strfreev (fields); + + /* time */ + + fields = g_strsplit (time_s, ":", 3); + if (fields[0] != NULL) { + tm.tm_hour = atoi (fields[0]); + if (fields[1] != NULL) { + tm.tm_min = atoi (fields[1]); + if (fields[2] != NULL) + tm.tm_sec = atoi (fields[2]); + } + } + g_strfreev (fields); + + return mktime (&tm); +} + + +static void +list__process_line (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + FrCommand7z *p7z_comm = FR_COMMAND_7Z (comm); + char **fields; + FileData *fdata; + + g_return_if_fail (line != NULL); + + if (! p7z_comm->list_started) { + if (strncmp (line, "p7zip Version ", 14) == 0) { + const char *ver_start; + int ver_len; + char version[256]; + + ver_start = eat_spaces (line + 14); + ver_len = strchr (ver_start, ' ') - ver_start; + strncpy (version, ver_start, ver_len); + version[ver_len] = 0; + + if (strcmp (version, "4.55") < 0) + p7z_comm->old_style = TRUE; + else + p7z_comm->old_style = FALSE; + } + else if (p7z_comm->old_style && (strncmp (line, "Listing archive: ", 17) == 0)) + p7z_comm->list_started = TRUE; + else if (! p7z_comm->old_style && (strcmp (line, "----------") == 0)) + p7z_comm->list_started = TRUE; + else if (strncmp (line, "Multivolume = ", 14) == 0) { + fields = g_strsplit (line, " = ", 2); + comm->multi_volume = (strcmp (fields[1], "+") == 0); + g_strfreev (fields); + } + return; + } + + if (strcmp (line, "") == 0) { + if (p7z_comm->fdata != NULL) { + if (p7z_comm->fdata->original_path == NULL) { + file_data_free (p7z_comm->fdata); + p7z_comm->fdata = NULL; + } + else { + fdata = p7z_comm->fdata; + if (fdata->dir) + fdata->name = dir_name_from_path (fdata->full_path); + else + fdata->name = g_strdup (file_name_from_path (fdata->full_path)); + fdata->path = remove_level_from_path (fdata->full_path); + fr_command_add_file (comm, fdata); + p7z_comm->fdata = NULL; + } + } + return; + } + + if (p7z_comm->fdata == NULL) + p7z_comm->fdata = file_data_new (); + + fields = g_strsplit (line, " = ", 2); + + if (g_strv_length (fields) < 2) { + g_strfreev (fields); + return; + } + + fdata = p7z_comm->fdata; + + if (strcmp (fields[0], "Path") == 0) { + fdata->free_original_path = TRUE; + fdata->original_path = g_strdup (fields[1]); + fdata->full_path = g_strconcat ((fdata->original_path[0] != '/') ? "/" : "", + fdata->original_path, + (fdata->dir && (fdata->original_path[strlen (fdata->original_path - 1)] != '/')) ? "/" : "", + NULL); + } + else if (strcmp (fields[0], "Folder") == 0) { + fdata->dir = (strcmp (fields[1], "+") == 0); + } + else if (strcmp (fields[0], "Size") == 0) { + fdata->size = g_ascii_strtoull (fields[1], NULL, 10); + } + else if (strcmp (fields[0], "Modified") == 0) { + char **modified_fields; + + modified_fields = g_strsplit (fields[1], " ", 2); + if (modified_fields[0] != NULL) + fdata->modified = mktime_from_string (modified_fields[0], modified_fields[1]); + g_strfreev (modified_fields); + } + else if (strcmp (fields[0], "Encrypted") == 0) { + if (strcmp (fields[1], "+") == 0) + fdata->encrypted = TRUE; + } + else if (strcmp (fields[0], "Method") == 0) { + if (strstr (fields[1], "AES") != NULL) + fdata->encrypted = TRUE; + } + else if (strcmp (fields[0], "Attributes") == 0) { + if (fields[1][0] == 'D') + fdata->dir = TRUE; + } + g_strfreev (fields); +} + + +static void +fr_command_7z_begin_command (FrCommand *comm) +{ + if (is_program_in_path ("7z")) + fr_process_begin_command (comm->process, "7z"); + else if (is_program_in_path ("7za")) + fr_process_begin_command (comm->process, "7za"); + else if (is_program_in_path ("7zr")) + fr_process_begin_command (comm->process, "7zr"); +} + + +static void +add_password_arg (FrCommand *comm, + const char *password, + gboolean always_specify) +{ + if (always_specify || ((password != NULL) && (*password != 0))) { + char *arg; + + arg = g_strconcat ("-p", password, NULL); + fr_process_add_arg (comm->process, arg); + g_free (arg); + } +} + + +static void +list__begin (gpointer data) +{ + FrCommand7z *p7z_comm = data; + + if (p7z_comm->fdata != NULL) { + file_data_free (p7z_comm->fdata); + p7z_comm->fdata = NULL; + } + p7z_comm->list_started = FALSE; +} + + +static void +fr_command_7z_list (FrCommand *comm) +{ + fr_process_set_out_line_func (comm->process, list__process_line, comm); + + fr_command_7z_begin_command (comm); + fr_process_set_begin_func (comm->process, list__begin, comm); + fr_process_add_arg (comm->process, "l"); + fr_process_add_arg (comm->process, "-slt"); + fr_process_add_arg (comm->process, "-bd"); + fr_process_add_arg (comm->process, "-y"); + add_password_arg (comm, comm->password, FALSE); + fr_process_add_arg (comm->process, "--"); + fr_process_add_arg (comm->process, comm->filename); + fr_process_end_command (comm->process); + + fr_process_start (comm->process); +} + + +static char Progress_Message[4196]; +static char Progress_Filename[4096]; + + +static void +parse_progress_line (FrCommand *comm, + const char *prefix, + const char *message_prefix, + const char *line) +{ + int prefix_len; + + prefix_len = strlen (prefix); + if (strncmp (line, prefix, prefix_len) == 0) { + double fraction; + + strcpy (Progress_Filename, line + prefix_len); + sprintf (Progress_Message, "%s%s", message_prefix, file_name_from_path (Progress_Filename)); + fr_command_message (comm, Progress_Message); + + fraction = (double) ++comm->n_file / (comm->n_files + 1); + fr_command_progress (comm, fraction); + } +} + + +static void +process_line__add (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + + if ((comm->volume_size > 0) && (strncmp (line, "Creating archive ", 17) == 0)) { + char *volume_filename; + + volume_filename = g_strconcat (comm->filename, ".001", NULL); + fr_command_set_multi_volume (comm, volume_filename); + g_free (volume_filename); + } + + if (comm->n_files != 0) + parse_progress_line (comm, "Compressing ", _("Adding file: "), line); +} + + +static void +fr_command_7z_add (FrCommand *comm, + const char *from_file, + GList *file_list, + const char *base_dir, + gboolean update, + gboolean recursive) +{ + GList *scan; + + fr_process_use_standard_locale (comm->process, TRUE); + fr_process_set_out_line_func (comm->process, + process_line__add, + comm); + + fr_command_7z_begin_command (comm); + + if (update) + fr_process_add_arg (comm->process, "u"); + else + fr_process_add_arg (comm->process, "a"); + + if (base_dir != NULL) { + fr_process_set_working_dir (comm->process, base_dir); + fr_process_add_arg_concat (comm->process, "-w", base_dir, NULL); + } + + if (is_mime_type (comm->mime_type, "application/zip") + || is_mime_type (comm->mime_type, "application/x-cbz")) + { + fr_process_add_arg (comm->process, "-tzip"); + fr_process_add_arg (comm->process, "-mem=AES128"); + } + + fr_process_add_arg (comm->process, "-bd"); + fr_process_add_arg (comm->process, "-y"); + fr_process_add_arg (comm->process, "-l"); + add_password_arg (comm, comm->password, FALSE); + if ((comm->password != NULL) + && (*comm->password != 0) + && comm->encrypt_header + && fr_command_is_capable_of (comm, FR_COMMAND_CAN_ENCRYPT_HEADER)) + { + fr_process_add_arg (comm->process, "-mhe=on"); + } + + /* fr_process_add_arg (comm->process, "-ms=off"); FIXME: solid mode off? */ + + switch (comm->compression) { + case FR_COMPRESSION_VERY_FAST: + fr_process_add_arg (comm->process, "-mx=1"); break; + case FR_COMPRESSION_FAST: + fr_process_add_arg (comm->process, "-mx=5"); break; + case FR_COMPRESSION_NORMAL: + fr_process_add_arg (comm->process, "-mx=5"); break; + case FR_COMPRESSION_MAXIMUM: + fr_process_add_arg (comm->process, "-mx=7"); break; + } + + if (is_mime_type (comm->mime_type, "application/x-ms-dos-executable")) + fr_process_add_arg (comm->process, "-sfx"); + + if (comm->volume_size > 0) + fr_process_add_arg_printf (comm->process, "-v%ub", comm->volume_size); + + if (from_file != NULL) + fr_process_add_arg_concat (comm->process, "-i@", from_file, NULL); + + fr_process_add_arg (comm->process, "--"); + fr_process_add_arg (comm->process, comm->filename); + if (from_file == NULL) + for (scan = file_list; scan; scan = scan->next) + fr_process_add_arg (comm->process, scan->data); + + fr_process_end_command (comm->process); +} + + +static void +fr_command_7z_delete (FrCommand *comm, + const char *from_file, + GList *file_list) +{ + GList *scan; + + fr_command_7z_begin_command (comm); + fr_process_add_arg (comm->process, "d"); + fr_process_add_arg (comm->process, "-bd"); + fr_process_add_arg (comm->process, "-y"); + if (is_mime_type (comm->mime_type, "application/x-ms-dos-executable")) + fr_process_add_arg (comm->process, "-sfx"); + + if (from_file != NULL) + fr_process_add_arg_concat (comm->process, "-i@", from_file, NULL); + + fr_process_add_arg (comm->process, "--"); + fr_process_add_arg (comm->process, comm->filename); + if (from_file == NULL) + for (scan = file_list; scan; scan = scan->next) + fr_process_add_arg (comm->process, scan->data); + + fr_process_end_command (comm->process); +} + + +static void +process_line__extract (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + + if (comm->n_files != 0) + parse_progress_line (comm, "Extracting ", _("Extracting file: "), line); +} + + +static void +fr_command_7z_extract (FrCommand *comm, + const char *from_file, + GList *file_list, + const char *dest_dir, + gboolean overwrite, + gboolean skip_older, + gboolean junk_paths) +{ + GList *scan; + + fr_process_use_standard_locale (comm->process, TRUE); + fr_process_set_out_line_func (comm->process, + process_line__extract, + comm); + fr_command_7z_begin_command (comm); + + if (junk_paths) + fr_process_add_arg (comm->process, "e"); + else + fr_process_add_arg (comm->process, "x"); + + fr_process_add_arg (comm->process, "-bd"); + fr_process_add_arg (comm->process, "-y"); + add_password_arg (comm, comm->password, FALSE); + + if (dest_dir != NULL) + fr_process_add_arg_concat (comm->process, "-o", dest_dir, NULL); + + if (from_file != NULL) + fr_process_add_arg_concat (comm->process, "-i@", from_file, NULL); + + fr_process_add_arg (comm->process, "--"); + fr_process_add_arg (comm->process, comm->filename); + if (from_file == NULL) + for (scan = file_list; scan; scan = scan->next) + fr_process_add_arg (comm->process, scan->data); + + fr_process_end_command (comm->process); +} + + +static void +fr_command_7z_test (FrCommand *comm) +{ + fr_command_7z_begin_command (comm); + fr_process_add_arg (comm->process, "t"); + fr_process_add_arg (comm->process, "-bd"); + fr_process_add_arg (comm->process, "-y"); + add_password_arg (comm, comm->password, FALSE); + fr_process_add_arg (comm->process, "--"); + fr_process_add_arg (comm->process, comm->filename); + fr_process_end_command (comm->process); +} + + +static void +fr_command_7z_handle_error (FrCommand *comm, + FrProcError *error) +{ + if (error->type == FR_PROC_ERROR_NONE) { + FileData *first; + char *basename; + char *testname; + + /* This is a way to fix bug #582712. */ + + if (comm->files->len != 1) + return; + + if (! g_str_has_suffix (comm->filename, ".001")) + return; + + first = g_ptr_array_index (comm->files, 0); + basename = g_path_get_basename (comm->filename); + testname = g_strconcat (first->original_path, ".001", NULL); + + if (strcmp (basename, testname) == 0) + error->type = FR_PROC_ERROR_ASK_PASSWORD; + + g_free (testname); + g_free (basename); + + return; + } + + if (error->status <= 1) { + error->type = FR_PROC_ERROR_NONE; + } + else { + GList *scan; + + for (scan = g_list_last (comm->process->out.raw); scan; scan = scan->prev) { + char *line = scan->data; + + if ((strstr (line, "Wrong password?") != NULL) + || (strstr (line, "Enter password") != NULL)) + { + error->type = FR_PROC_ERROR_ASK_PASSWORD; + break; + } + } + } +} + + +const char *sevenz_mime_types[] = { "application/x-7z-compressed", + "application/x-arj", + "application/vnd.ms-cab-compressed", + "application/x-cd-image", + /*"application/x-cbr",*/ + "application/x-cbz", + "application/x-ms-dos-executable", + "application/x-rar", + "application/zip", + NULL }; + + +const char ** +fr_command_7z_get_mime_types (FrCommand *comm) +{ + return sevenz_mime_types; +} + + +FrCommandCap +fr_command_7z_get_capabilities (FrCommand *comm, + const char *mime_type, + gboolean check_command) +{ + FrCommandCap capabilities; + + capabilities = FR_COMMAND_CAN_ARCHIVE_MANY_FILES; + if (! is_program_available ("7za", check_command) && ! is_program_available ("7zr", check_command) && ! is_program_available ("7z", check_command)) + return capabilities; + + if (is_mime_type (mime_type, "application/x-7z-compressed")) { + capabilities |= FR_COMMAND_CAN_READ_WRITE | FR_COMMAND_CAN_CREATE_VOLUMES; + if (is_program_available ("7z", check_command)) + capabilities |= FR_COMMAND_CAN_ENCRYPT | FR_COMMAND_CAN_ENCRYPT_HEADER; + } + else if (is_mime_type (mime_type, "application/x-7z-compressed-tar")) { + capabilities |= FR_COMMAND_CAN_READ_WRITE; + if (is_program_available ("7z", check_command)) + capabilities |= FR_COMMAND_CAN_ENCRYPT | FR_COMMAND_CAN_ENCRYPT_HEADER; + } + else if (is_program_available ("7z", check_command)) { + if (is_mime_type (mime_type, "application/x-rar") + || is_mime_type (mime_type, "application/x-cbr")) + { + if (! check_command || g_file_test ("/usr/lib/p7zip/Codecs/Rar29.so", G_FILE_TEST_EXISTS)) + capabilities |= FR_COMMAND_CAN_READ; + } + else + capabilities |= FR_COMMAND_CAN_READ; + + if (is_mime_type (mime_type, "application/x-cbz") + || is_mime_type (mime_type, "application/x-ms-dos-executable") + || is_mime_type (mime_type, "application/zip")) + { + capabilities |= FR_COMMAND_CAN_WRITE | FR_COMMAND_CAN_ENCRYPT; + } + } + else if (is_program_available ("7za", check_command)) { + if (is_mime_type (mime_type, "application/vnd.ms-cab-compressed") + || is_mime_type (mime_type, "application/zip")) + { + capabilities |= FR_COMMAND_CAN_READ; + } + + if (is_mime_type (mime_type, "application/zip")) + capabilities |= FR_COMMAND_CAN_WRITE; + } + + /* multi-volumes are read-only */ + if ((comm->files->len > 0) && comm->multi_volume && (capabilities & FR_COMMAND_CAN_WRITE)) + capabilities ^= FR_COMMAND_CAN_WRITE; + + return capabilities; +} + + +static const char * +fr_command_7z_get_packages (FrCommand *comm, + const char *mime_type) +{ + if (is_mime_type (mime_type, "application/x-rar")) + return PACKAGES ("p7zip,p7zip-rar"); + else if (is_mime_type (mime_type, "application/zip") || is_mime_type (mime_type, "application/vnd.ms-cab-compressed")) + return PACKAGES ("p7zip,p7zip-full"); + else + return PACKAGES ("p7zip"); +} + + +static void +fr_command_7z_class_init (FrCommand7zClass *class) +{ + GObjectClass *gobject_class = G_OBJECT_CLASS (class); + FrCommandClass *afc; + + parent_class = g_type_class_peek_parent (class); + afc = (FrCommandClass*) class; + + gobject_class->finalize = fr_command_7z_finalize; + + afc->list = fr_command_7z_list; + afc->add = fr_command_7z_add; + afc->delete = fr_command_7z_delete; + afc->extract = fr_command_7z_extract; + afc->test = fr_command_7z_test; + afc->handle_error = fr_command_7z_handle_error; + afc->get_mime_types = fr_command_7z_get_mime_types; + afc->get_capabilities = fr_command_7z_get_capabilities; + afc->get_packages = fr_command_7z_get_packages; +} + + +static void +fr_command_7z_init (FrCommand *comm) +{ + comm->propAddCanUpdate = TRUE; + comm->propAddCanReplace = TRUE; + comm->propAddCanStoreFolders = TRUE; + comm->propExtractCanAvoidOverwrite = FALSE; + comm->propExtractCanSkipOlder = FALSE; + comm->propExtractCanJunkPaths = TRUE; + comm->propPassword = TRUE; + comm->propTest = TRUE; + comm->propListFromFile = TRUE; +} + + +static void +fr_command_7z_finalize (GObject *object) +{ + g_return_if_fail (object != NULL); + g_return_if_fail (FR_IS_COMMAND_7Z (object)); + + /* Chain up */ + if (G_OBJECT_CLASS (parent_class)->finalize) + G_OBJECT_CLASS (parent_class)->finalize (object); +} + + +GType +fr_command_7z_get_type () +{ + static GType type = 0; + + if (! type) { + GTypeInfo type_info = { + sizeof (FrCommand7zClass), + NULL, + NULL, + (GClassInitFunc) fr_command_7z_class_init, + NULL, + NULL, + sizeof (FrCommand7z), + 0, + (GInstanceInitFunc) fr_command_7z_init + }; + + type = g_type_register_static (FR_TYPE_COMMAND, + "FRCommand7z", + &type_info, + 0); + } + + return type; +} -- cgit v1.2.1