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-rar.c | 813 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 813 insertions(+) create mode 100644 src/fr-command-rar.c (limited to 'src/fr-command-rar.c') diff --git a/src/fr-command-rar.c b/src/fr-command-rar.c new file mode 100644 index 0000000..2c42ba4 --- /dev/null +++ b/src/fr-command-rar.c @@ -0,0 +1,813 @@ +/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */ + +/* + * File-Roller + * + * Copyright (C) 2001 The 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 "file-data.h" +#include "file-utils.h" +#include "gio-utils.h" +#include "glib-utils.h" +#include "fr-command.h" +#include "fr-command-rar.h" +#include "fr-error.h" + +static void fr_command_rar_class_init (FrCommandRarClass *class); +static void fr_command_rar_init (FrCommand *afile); +static void fr_command_rar_finalize (GObject *object); + +/* Parent Class */ + +static FrCommandClass *parent_class = NULL; + + +static gboolean +have_rar (void) +{ + return is_program_in_path ("rar"); +} + + +/* -- 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_mday = atoi (fields[0]); + if (fields[1] != NULL) { + tm.tm_mon = atoi (fields[1]) - 1; + if (fields[2] != NULL) + tm.tm_year = 100 + atoi (fields[2]); + } + } + g_strfreev (fields); + + /* time */ + + fields = g_strsplit (time_s, ":", 2); + if (fields[0] != NULL) { + tm.tm_hour = atoi (fields[0]); + if (fields[1] != NULL) + tm.tm_min = atoi (fields[1]); + } + g_strfreev (fields); + + return mktime (&tm); +} + + +static void +process_line (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + FrCommandRar *rar_comm = FR_COMMAND_RAR (comm); + char **fields; + const char *name_field; + + g_return_if_fail (line != NULL); + + if (! rar_comm->list_started) { + if (strncmp (line, "--------", 8) == 0) { + rar_comm->list_started = TRUE; + rar_comm->odd_line = TRUE; + } + else if (strncmp (line, "Volume ", 7) == 0) + comm->multi_volume = TRUE; + return; + } + + if (strncmp (line, "--------", 8) == 0) { + rar_comm->list_started = FALSE; + return; + } + + if (! rar_comm->odd_line) { + FileData *fdata; + + fdata = rar_comm->fdata; + + /* read file info. */ + + fields = split_line (line, 6); + if (g_strv_length (fields) < 6) { + /* wrong line format, treat this line as a filename line */ + g_strfreev (fields); + file_data_free (rar_comm->fdata); + rar_comm->fdata = NULL; + rar_comm->odd_line = TRUE; + } + else { + if ((strcmp (fields[2], "<->") == 0) + || (strcmp (fields[2], "<--") == 0)) + { + /* ignore files that span more volumes */ + + file_data_free (rar_comm->fdata); + rar_comm->fdata = NULL; + } + else { + fdata->size = g_ascii_strtoull (fields[0], NULL, 10); + fdata->modified = mktime_from_string (fields[3], fields[4]); + + if ((fields[5][1] == 'D') || (fields[5][0] == 'd')) { + char *tmp; + + tmp = fdata->full_path; + fdata->full_path = g_strconcat (fdata->full_path, "/", NULL); + + fdata->original_path = g_strdup (fdata->original_path); + fdata->free_original_path = TRUE; + + g_free (tmp); + + fdata->name = dir_name_from_path (fdata->full_path); + fdata->dir = TRUE; + } + else + fdata->name = g_strdup (file_name_from_path (fdata->full_path)); + + fr_command_add_file (comm, fdata); + rar_comm->fdata = NULL; + } + + g_strfreev (fields); + } + } + + if (rar_comm->odd_line) { + FileData *fdata; + + rar_comm->fdata = fdata = file_data_new (); + + /* read file name. */ + + fdata->encrypted = (line[0] == '*') ? TRUE : FALSE; + + name_field = line + 1; + + if (*name_field == '/') { + fdata->full_path = g_strdup (name_field); + fdata->original_path = fdata->full_path; + } + else { + fdata->full_path = g_strconcat ("/", name_field, NULL); + fdata->original_path = fdata->full_path + 1; + } + + fdata->link = NULL; + fdata->path = remove_level_from_path (fdata->full_path); + } + else { + + } + + rar_comm->odd_line = ! rar_comm->odd_line; +} + + +static void +add_password_arg (FrCommand *comm, + const char *password, + gboolean disable_query) +{ + if ((password != NULL) && (password[0] != '\0')) { + if (comm->encrypt_header) + fr_process_add_arg_concat (comm->process, "-hp", password, NULL); + else + fr_process_add_arg_concat (comm->process, "-p", password, NULL); + } + else if (disable_query) + fr_process_add_arg (comm->process, "-p-"); +} + + +typedef enum { + FIRST_VOLUME_IS_000, + FIRST_VOLUME_IS_001, + FIRST_VOLUME_IS_RAR +} FirstVolumeExtension; + + +static char * +get_first_volume_name (const char *name, + const char *pattern, + FirstVolumeExtension extension_type) +{ + char *volume_name = NULL; + GRegex *re; + + re = g_regex_new (pattern, G_REGEX_CASELESS, 0, NULL); + if (g_regex_match (re, name, 0, NULL)) { + char **parts; + int l, i; + + parts = g_regex_split (re, name, 0); + l = strlen (parts[2]); + switch (extension_type) { + case FIRST_VOLUME_IS_000: + for (i = 0; i < l; i++) + parts[2][i] = '0'; + break; + + case FIRST_VOLUME_IS_001: + for (i = 0; i < l; i++) + parts[2][i] = (i < l - 1) ? '0' : '1'; + break; + + case FIRST_VOLUME_IS_RAR: + if (g_str_has_suffix (parts[1], "r")) { + parts[2][0] = 'a'; + parts[2][1] = 'r'; + } + else { + parts[2][0] = 'A'; + parts[2][1] = 'R'; + } + break; + } + + volume_name = g_strjoinv ("", parts); + + g_strfreev (parts); + } + g_regex_unref (re); + + if (volume_name != NULL) { + char *tmp; + + tmp = volume_name; + volume_name = g_filename_from_utf8 (tmp, -1, NULL, NULL, NULL); + g_free (tmp); + } + + return volume_name; +} + + +static void +check_multi_vomule (FrCommand *comm) +{ + GFile *file; + char buffer[11]; + + file = g_file_new_for_path (comm->filename); + if (! g_load_file_in_buffer (file, buffer, 11, NULL)) { + g_object_unref (file); + return; + } + + if ((buffer[10] & 0x01) == 0x01) { + char *volume_name = NULL; + char *name; + + name = g_filename_to_utf8 (file_name_from_path (comm->filename), -1, NULL, NULL, NULL); + + volume_name = get_first_volume_name (name, "^(.*\\.part)([0-9]+)(\\.rar)$", FIRST_VOLUME_IS_001); + if (volume_name == NULL) + volume_name = get_first_volume_name (name, "^(.*\\.r)([0-9]+)$", FIRST_VOLUME_IS_RAR); + if (volume_name == NULL) + volume_name = get_first_volume_name (name, "^(.*\\.)([0-9]+)$", FIRST_VOLUME_IS_001); + + if (volume_name != NULL) { + GFile *parent; + GFile *child; + char *volume_filename; + + parent = g_file_get_parent (file); + child = g_file_get_child (parent, volume_name); + volume_filename = g_file_get_path (child); + fr_command_set_multi_volume (comm, volume_filename); + + g_free (volume_filename); + g_object_unref (child); + g_object_unref (parent); + } + + g_free (name); + } + + g_object_unref (file); +} + + +static void +list__begin (gpointer data) +{ + FrCommandRar *comm = data; + + comm->list_started = FALSE; +} + + +static void +fr_command_rar_list (FrCommand *comm) +{ + check_multi_vomule (comm); + + fr_process_set_out_line_func (comm->process, process_line, comm); + + if (have_rar ()) + fr_process_begin_command (comm->process, "rar"); + else + fr_process_begin_command (comm->process, "unrar"); + fr_process_set_begin_func (comm->process, list__begin, comm); + fr_process_add_arg (comm->process, "v"); + fr_process_add_arg (comm->process, "-c-"); + fr_process_add_arg (comm->process, "-v"); + + add_password_arg (comm, comm->password, TRUE); + + /* stop switches scanning */ + 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; + int len; + char *b_idx; + + strcpy (Progress_Filename, line + prefix_len); + + /* when a new volume is created a sequence of backspaces is + * issued, remove the backspaces from the filename */ + b_idx = strchr (Progress_Filename, '\x08'); + if (b_idx != NULL) + *b_idx = 0; + + /* remove the OK at the end of the filename */ + len = strlen (Progress_Filename); + if ((len > 5) && (strncmp (Progress_Filename + len - 5, " OK ", 5) == 0)) + Progress_Filename[len - 5] = 0; + + 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 (strncmp (line, "Creating archive ", 17) == 0) { + const char *archive_filename = line + 17; + char *uri; + + uri = g_filename_to_uri (archive_filename, NULL, NULL); + if ((comm->volume_size > 0) + && g_regex_match_simple ("^.*\\.part(0)*2\\.rar$", uri, G_REGEX_CASELESS, 0)) + { + char *volume_filename; + + volume_filename = g_strdup (archive_filename); + volume_filename[strlen (volume_filename) - 5] = '1'; + fr_command_set_multi_volume (comm, volume_filename); + g_free (volume_filename); + } + fr_command_working_archive (comm, uri); + + g_free (uri); + return; + } + + if (comm->n_files != 0) + parse_progress_line (comm, "Adding ", _("Adding file: "), line); +} + + +static void +fr_command_rar_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_process_begin_command (comm->process, "rar"); + + if (base_dir != NULL) + fr_process_set_working_dir (comm->process, base_dir); + + if (update) + fr_process_add_arg (comm->process, "u"); + else + fr_process_add_arg (comm->process, "a"); + + switch (comm->compression) { + case FR_COMPRESSION_VERY_FAST: + fr_process_add_arg (comm->process, "-m1"); break; + case FR_COMPRESSION_FAST: + fr_process_add_arg (comm->process, "-m2"); break; + case FR_COMPRESSION_NORMAL: + fr_process_add_arg (comm->process, "-m3"); break; + case FR_COMPRESSION_MAXIMUM: + fr_process_add_arg (comm->process, "-m5"); break; + } + + add_password_arg (comm, comm->password, FALSE); + + if (comm->volume_size > 0) + fr_process_add_arg_printf (comm->process, "-v%ub", comm->volume_size); + + /* disable percentage indicator */ + fr_process_add_arg (comm->process, "-Idp"); + + 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); + else + fr_process_add_arg_concat (comm->process, "@", from_file, NULL); + + fr_process_end_command (comm->process); +} + + +static void +process_line__delete (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + + if (strncmp (line, "Deleting from ", 14) == 0) { + char *uri; + + uri = g_filename_to_uri (line + 14, NULL, NULL); + fr_command_working_archive (comm, uri); + g_free (uri); + + return; + } + + if (comm->n_files != 0) + parse_progress_line (comm, "Deleting ", _("Removing file: "), line); +} + + +static void +fr_command_rar_delete (FrCommand *comm, + const char *from_file, + GList *file_list) +{ + GList *scan; + + fr_process_use_standard_locale (comm->process, TRUE); + fr_process_set_out_line_func (comm->process, + process_line__delete, + comm); + + fr_process_begin_command (comm->process, "rar"); + fr_process_add_arg (comm->process, "d"); + + 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); + else + fr_process_add_arg_concat (comm->process, "@", from_file, NULL); + + fr_process_end_command (comm->process); +} + + +static void +process_line__extract (char *line, + gpointer data) +{ + FrCommand *comm = FR_COMMAND (data); + + if (strncmp (line, "Extracting from ", 16) == 0) { + char *uri; + + uri = g_filename_to_uri (line + 16, NULL, NULL); + fr_command_working_archive (comm, uri); + g_free (uri); + + return; + } + + if (comm->n_files != 0) + parse_progress_line (comm, "Extracting ", _("Extracting file: "), line); +} + + +static void +fr_command_rar_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); + + if (have_rar ()) + fr_process_begin_command (comm->process, "rar"); + else + fr_process_begin_command (comm->process, "unrar"); + + fr_process_add_arg (comm->process, "x"); + + /* keep broken extracted files */ + fr_process_add_arg (comm->process, "-kb"); + + if (overwrite) + fr_process_add_arg (comm->process, "-o+"); + else + fr_process_add_arg (comm->process, "-o-"); + + if (skip_older) + fr_process_add_arg (comm->process, "-u"); + + if (junk_paths) + fr_process_add_arg (comm->process, "-ep"); + + add_password_arg (comm, comm->password, TRUE); + + /* disable percentage indicator */ + fr_process_add_arg (comm->process, "-Idp"); + + 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); + else + fr_process_add_arg_concat (comm->process, "@", from_file, NULL); + + if (dest_dir != NULL) + fr_process_add_arg (comm->process, dest_dir); + + fr_process_end_command (comm->process); +} + + +static void +fr_command_rar_test (FrCommand *comm) +{ + if (have_rar ()) + fr_process_begin_command (comm->process, "rar"); + else + fr_process_begin_command (comm->process, "unrar"); + + fr_process_add_arg (comm->process, "t"); + + add_password_arg (comm, comm->password, TRUE); + + /* disable percentage indicator */ + fr_process_add_arg (comm->process, "-Idp"); + + /* stop switches scanning */ + fr_process_add_arg (comm->process, "--"); + + fr_process_add_arg (comm->process, comm->filename); + fr_process_end_command (comm->process); +} + + +static void +fr_command_rar_handle_error (FrCommand *comm, + FrProcError *error) +{ + GList *scan; + +#if 0 + { + GList *scan; + + for (scan = g_list_last (comm->process->err.raw); scan; scan = scan->prev) + g_print ("%s\n", (char*)scan->data); + } +#endif + + if (error->type == FR_PROC_ERROR_NONE) + return; + + /*if (error->status == 3) + error->type = FR_PROC_ERROR_ASK_PASSWORD; + else */ + if (error->status <= 1) + error->type = FR_PROC_ERROR_NONE; + + for (scan = g_list_last (comm->process->err.raw); scan; scan = scan->prev) { + char *line = scan->data; + + if (strstr (line, "password incorrect") != NULL) { + error->type = FR_PROC_ERROR_ASK_PASSWORD; + break; + } + + if (strncmp (line, "Unexpected end of archive", 25) == 0) { + /* FIXME: handle this type of errors at a higher level when the freeze is over. */ + } + + if (strncmp (line, "Cannot find volume", 18) == 0) { + char *volume_filename; + + g_clear_error (&error->gerror); + + error->type = FR_PROC_ERROR_MISSING_VOLUME; + volume_filename = g_path_get_basename (line + strlen ("Cannot find volume ")); + error->gerror = g_error_new (FR_ERROR, error->status, _("Could not find the volume: %s"), volume_filename); + g_free (volume_filename); + break; + } + } +} + + +const char *rar_mime_type[] = { "application/x-cbr", + "application/x-rar", + NULL }; + + +const char ** +fr_command_rar_get_mime_types (FrCommand *comm) +{ + return rar_mime_type; +} + + +FrCommandCap +fr_command_rar_get_capabilities (FrCommand *comm, + const char *mime_type, + gboolean check_command) +{ + FrCommandCap capabilities; + + capabilities = FR_COMMAND_CAN_ARCHIVE_MANY_FILES | FR_COMMAND_CAN_ENCRYPT | FR_COMMAND_CAN_ENCRYPT_HEADER; + if (is_program_available ("rar", check_command)) + capabilities |= FR_COMMAND_CAN_READ_WRITE | FR_COMMAND_CAN_CREATE_VOLUMES; + else if (is_program_available ("unrar", check_command)) + capabilities |= FR_COMMAND_CAN_READ; + + /* 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_rar_get_packages (FrCommand *comm, + const char *mime_type) +{ + return PACKAGES ("rar,unrar"); +} + + +static void +fr_command_rar_class_init (FrCommandRarClass *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_rar_finalize; + + afc->list = fr_command_rar_list; + afc->add = fr_command_rar_add; + afc->delete = fr_command_rar_delete; + afc->extract = fr_command_rar_extract; + afc->test = fr_command_rar_test; + afc->handle_error = fr_command_rar_handle_error; + afc->get_mime_types = fr_command_rar_get_mime_types; + afc->get_capabilities = fr_command_rar_get_capabilities; + afc->get_packages = fr_command_rar_get_packages; +} + + +static void +fr_command_rar_init (FrCommand *comm) +{ + comm->propAddCanUpdate = TRUE; + comm->propAddCanReplace = TRUE; + comm->propAddCanStoreFolders = TRUE; + comm->propExtractCanAvoidOverwrite = TRUE; + comm->propExtractCanSkipOlder = TRUE; + comm->propExtractCanJunkPaths = TRUE; + comm->propPassword = TRUE; + comm->propTest = TRUE; + comm->propListFromFile = TRUE; +} + + +static void +fr_command_rar_finalize (GObject *object) +{ + g_return_if_fail (object != NULL); + g_return_if_fail (FR_IS_COMMAND_RAR (object)); + + /* Chain up */ + if (G_OBJECT_CLASS (parent_class)->finalize) + G_OBJECT_CLASS (parent_class)->finalize (object); +} + + +GType +fr_command_rar_get_type () +{ + static GType type = 0; + + if (! type) { + GTypeInfo type_info = { + sizeof (FrCommandRarClass), + NULL, + NULL, + (GClassInitFunc) fr_command_rar_class_init, + NULL, + NULL, + sizeof (FrCommandRar), + 0, + (GInstanceInitFunc) fr_command_rar_init + }; + + type = g_type_register_static (FR_TYPE_COMMAND, + "FRCommandRar", + &type_info, + 0); + } + + return type; +} -- cgit v1.2.1