X-Git-Url: https://www.fi.muni.cz/~kas/git//home/kas/public_html/git/?a=blobdiff_plain;f=backend%2Fcomics%2Fcomics-document.c;h=c871860dd4a6e704121a11844ebcca521404c65c;hb=ae68dd7c9fd270f04d7c1392b21662a8a1c1e7db;hp=29a1581d143fede0359cccd8efd7fa17ee7af604;hpb=6d6a66984b5f4467b02fe6b8f6d4370750f291cb;p=evince.git diff --git a/backend/comics/comics-document.c b/backend/comics/comics-document.c index 29a1581d..c871860d 100644 --- a/backend/comics/comics-document.c +++ b/backend/comics/comics-document.c @@ -17,29 +17,25 @@ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ -#include "config.h" - #include + #include #include -#include -#include -#include #include #include #include +#include +#include +#include +#include + #include "comics-document.h" #include "ev-document-misc.h" #include "ev-document-thumbnails.h" #include "ev-file-helpers.h" -struct _ComicsDocumentClass -{ - GObjectClass parent_class; -}; - -typedef enum +typedef enum { RARLABS, GNAUNRAR, @@ -47,33 +43,66 @@ typedef enum P7ZIP } ComicBookDecompressType; +typedef struct _ComicsDocumentClass ComicsDocumentClass; + +struct _ComicsDocumentClass +{ + EvDocumentClass parent_class; +}; + struct _ComicsDocument { - GObject parent_instance; + EvDocument parent_instance; + gchar *archive, *dir; - GSList *page_names; - gint n_pages; + GPtrArray *page_names; gchar *selected_command; gchar *extract_command, *list_command, *decompress_tmp; gboolean regex_arg; + gint offset; ComicBookDecompressType command_usage; }; -struct -{ - char *extract, *list, *decompress_tmp; - gboolean regex_arg; -} command_usage_def[] = { - {"%s p -c- -ierr", "%s vb -c- -- %s", NULL , FALSE}, - {NULL , "%s t %s" , "%s -xf %s %s", TRUE }, - {"%s -p -C" , "%s -Z -1 -- %s" , NULL , TRUE }, - {"%s x -so" , "%s l -- %s" , NULL , TRUE } -}; +#define OFFSET_7Z 53 +#define NO_OFFSET 0 +/* For perfomance reasons of 7z* we've choosen to decompress on the temporary + * directory instead of decompressing on the stdout */ -typedef struct _ComicsDocumentClass ComicsDocumentClass; +/** + * @extract: command line arguments to pass to extract a file from the archive + * to stdout. The archive file and the file to extract will be appended after + * a "--". + * @list: command line arguments to list the archive contents + * @decompress_tmp: command line arguments to pass to extract the archive + * into a directory. The archive file and the directory to extract to will be + * appended after a "--". + * @regex_arg: whether the command expects one filename or accepts a regex (glob?) + * @offset: the byte offset of the filename on each line in the output of + * running the @list command + */ +typedef struct { + char *extract; + char *list; + char *decompress_tmp; + gboolean regex_arg; + gint offset; +} ComicBookDecompressCommand; + +static const ComicBookDecompressCommand command_usage_def[] = { + /* RARLABS unrar */ + {"%s p -c- -ierr", "%s vb -c- -- %s", NULL , FALSE, NO_OFFSET}, + + /* GNA! unrar */ + {NULL , "%s t %s" , "%s -xf %s %s" , TRUE , NO_OFFSET}, + + /* unzip */ + {"%s -p -C" , "%s -Z -1 -- %s" , NULL , TRUE , NO_OFFSET}, + + /* 7zip */ + {NULL , "%s l -- %s" , "%s x -y %s -o%s", FALSE, OFFSET_7Z} +}; -static void comics_document_document_iface_init (EvDocumentIface *iface); static void comics_document_document_thumbnails_iface_init (EvDocumentThumbnailsIface *iface); static GSList* get_supported_image_extensions (void); @@ -183,7 +212,7 @@ static gboolean comics_generate_command_lines (ComicsDocument *comics_document, GError **error) { - gchar *quoted_file, *checksum; + gchar *quoted_file; ComicBookDecompressType type; type = comics_document->command_usage; @@ -197,39 +226,22 @@ comics_generate_command_lines (ComicsDocument *comics_document, comics_document->selected_command, quoted_file); comics_document->regex_arg = command_usage_def[type].regex_arg; + comics_document->offset = command_usage_def[type].offset; if (command_usage_def[type].decompress_tmp) { - checksum = - g_compute_checksum_for_string (G_CHECKSUM_MD5, - comics_document->archive, - -1); - comics_document->dir = g_build_filename (ev_tmp_dir (), - checksum, NULL); - comics_document->decompress_tmp = + comics_document->dir = ev_mkdtemp ("evince-comics-XXXXXX", error); + if (comics_document->dir == NULL) + return FALSE; + + /* unrar-free can't create directories, but ev_mkdtemp already created the dir */ + + comics_document->decompress_tmp = g_strdup_printf (command_usage_def[type].decompress_tmp, comics_document->selected_command, quoted_file, comics_document->dir); g_free (quoted_file); - g_free (checksum); - /* unrar-free can't create directories so we do it on its - * behalf */ - if (type == GNAUNRAR) { - if (g_mkdir_with_parents (comics_document->dir, 0700) != - 0) { - int errsv = errno; - g_set_error (error, - EV_DOCUMENT_ERROR, - EV_DOCUMENT_ERROR_INVALID, - _("Failed to create a temporary " - "directory.")); - g_warning ("Failed to create directory %s: %s", - comics_document->dir, - g_strerror (errsv)); - - return FALSE; - } - } - if (!comics_decompress_temp_dir (comics_document->decompress_tmp, + + if (!comics_decompress_temp_dir (comics_document->decompress_tmp, comics_document->selected_command, error)) return FALSE; else @@ -318,14 +330,26 @@ comics_check_decompress_command (gchar *mime_type, } else if (!strcmp (mime_type, "application/x-cb7") || !strcmp (mime_type, "application/x-7z-compressed")) { - /* 7zr is a light stand-alone executable that supports only - * 7z/LZMA/BCJ */ + /* 7zr, 7za and 7z are the commands from the p7zip project able + * to decompress .7z files */ comics_document->selected_command = g_find_program_in_path ("7zr"); if (comics_document->selected_command) { comics_document->command_usage = P7ZIP; return TRUE; } + comics_document->selected_command = + g_find_program_in_path ("7za"); + if (comics_document->selected_command) { + comics_document->command_usage = P7ZIP; + return TRUE; + } + comics_document->selected_command = + g_find_program_in_path ("7z"); + if (comics_document->selected_command) { + comics_document->command_usage = P7ZIP; + return TRUE; + } } else { g_set_error (error, EV_DOCUMENT_ERROR, @@ -337,11 +361,18 @@ comics_check_decompress_command (gchar *mime_type, g_set_error_literal (error, EV_DOCUMENT_ERROR, EV_DOCUMENT_ERROR_INVALID, - _("Can't find an appropiate command to " + _("Can't find an appropriate command to " "decompress this type of comic book")); return FALSE; } +static int +sort_page_names (gconstpointer a, + gconstpointer b) +{ + return strcmp (* (const char **) a, * (const char **) b); +} + static gboolean comics_document_load (EvDocument *document, const char *uri, @@ -351,7 +382,7 @@ comics_document_load (EvDocument *document, GSList *supported_extensions; gchar *std_out; gchar *mime_type; - gchar **cbr_files; + gchar **cb_files, *cb_file; gboolean success; int i, retval; GError *err = NULL; @@ -400,10 +431,10 @@ comics_document_load (EvDocument *document, } /* FIXME: is this safe against filenames containing \n in the archive ? */ - cbr_files = g_strsplit (std_out, "\n", 0); + cb_files = g_strsplit (std_out, "\n", 0); g_free (std_out); - if (!cbr_files) { + if (!cb_files) { g_set_error_literal (error, EV_DOCUMENT_ERROR, EV_DOCUMENT_ERROR_INVALID, @@ -411,31 +442,38 @@ comics_document_load (EvDocument *document, return FALSE; } + comics_document->page_names = g_ptr_array_sized_new (64); + supported_extensions = get_supported_image_extensions (); - for (i = 0; cbr_files[i] != NULL; i++) { - gchar *suffix = g_strrstr (cbr_files[i], "."); + for (i = 0; cb_files[i] != NULL; i++) { + if (comics_document->offset != NO_OFFSET) { + if (g_utf8_strlen (cb_files[i],-1) > + comics_document->offset) { + cb_file = + g_utf8_offset_to_pointer (cb_files[i], + comics_document->offset); + } else { + continue; + } + } else { + cb_file = cb_files[i]; + } + gchar *suffix = g_strrstr (cb_file, "."); if (!suffix) continue; suffix = g_ascii_strdown (suffix + 1, -1); - if (g_slist_find_custom (supported_extensions, suffix, (GCompareFunc) strcmp) != NULL) { - comics_document->page_names = - g_slist_insert_sorted ( - comics_document->page_names, - g_strdup (g_strstrip (cbr_files[i])), - (GCompareFunc) strcmp); - comics_document->n_pages++; + g_ptr_array_add (comics_document->page_names, + g_strstrip (g_strdup (cb_file))); } - g_free (suffix); } - - g_strfreev (cbr_files); + g_strfreev (cb_files); g_slist_foreach (supported_extensions, (GFunc) g_free, NULL); g_slist_free (supported_extensions); - if (comics_document->n_pages == 0) { + if (comics_document->page_names->len == 0) { g_set_error (error, EV_DOCUMENT_ERROR, EV_DOCUMENT_ERROR_INVALID, @@ -444,6 +482,9 @@ comics_document_load (EvDocument *document, return FALSE; } + /* Now sort the pages */ + g_ptr_array_sort (comics_document->page_names, sort_page_names); + return TRUE; } @@ -461,7 +502,12 @@ comics_document_save (EvDocument *document, static int comics_document_get_n_pages (EvDocument *document) { - return COMICS_DOCUMENT (document)->n_pages; + ComicsDocument *comics_document = COMICS_DOCUMENT (document); + + if (comics_document->page_names == NULL) + return 0; + + return comics_document->page_names->len; } static void @@ -520,10 +566,8 @@ comics_document_get_page_size (EvDocument *document, g_spawn_close_pid (child_pid); g_object_unref (loader); } else { - filename = g_build_filename (comics_document->dir, - (char*) g_slist_nth_data ( - comics_document->page_names, - page->index), + filename = g_build_filename (comics_document->dir, + (char *) comics_document->page_names->pdata[page->index], NULL); pixbuf = gdk_pixbuf_new_from_file (filename, NULL); g_free (filename); @@ -595,9 +639,7 @@ comics_document_render_pixbuf (EvDocument *document, } else { filename = g_build_filename (comics_document->dir, - (char*) g_slist_nth_data ( - comics_document->page_names, - rc->page->index), + (char *) comics_document->page_names->pdata[rc->page->index], NULL); gdk_pixbuf_get_file_info (filename, &width, &height); @@ -682,13 +724,11 @@ comics_document_finalize (GObject *object) g_warning (_("There was an error deleting “%s”."), comics_document->dir); g_free (comics_document->dir); - g_remove (ev_tmp_dir ()); } if (comics_document->page_names) { - g_slist_foreach (comics_document->page_names, - (GFunc) g_free, NULL); - g_slist_free (comics_document->page_names); + g_ptr_array_foreach (comics_document->page_names, (GFunc) g_free, NULL); + g_ptr_array_free (comics_document->page_names, TRUE); } g_free (comics_document->archive); @@ -702,27 +742,16 @@ comics_document_finalize (GObject *object) static void comics_document_class_init (ComicsDocumentClass *klass) { - GObjectClass *gobject_class = G_OBJECT_CLASS (klass); - gobject_class->finalize = comics_document_finalize; -} + GObjectClass *gobject_class = G_OBJECT_CLASS (klass); + EvDocumentClass *ev_document_class = EV_DOCUMENT_CLASS (klass); -static EvDocumentInfo * -comics_document_get_info (EvDocument *document) -{ - EvDocumentInfo *info; - info = g_new0 (EvDocumentInfo, 1); - return info; -} + gobject_class->finalize = comics_document_finalize; -static void -comics_document_document_iface_init (EvDocumentIface *iface) -{ - iface->load = comics_document_load; - iface->save = comics_document_save; - iface->get_n_pages = comics_document_get_n_pages; - iface->get_page_size = comics_document_get_page_size; - iface->render = comics_document_render; - iface->get_info = comics_document_get_info; + ev_document_class->load = comics_document_load; + ev_document_class->save = comics_document_save; + ev_document_class->get_n_pages = comics_document_get_n_pages; + ev_document_class->get_page_size = comics_document_get_page_size; + ev_document_class->render = comics_document_render; } static void @@ -731,7 +760,6 @@ comics_document_init (ComicsDocument *comics_document) comics_document->archive = NULL; comics_document->page_names = NULL; comics_document->extract_command = NULL; - comics_document->n_pages = 0; } /* Returns a list of file extensions supported by gdk-pixbuf */ @@ -812,13 +840,14 @@ extract_argv (EvDocument *document, gint page) char *command_line, *quoted_archive, *quoted_filename; GError *err = NULL; + if (page >= comics_document->page_names->len) + return NULL; + quoted_archive = g_shell_quote (comics_document->archive); if (comics_document->regex_arg) { - quoted_filename = comics_regex_quote ( - g_slist_nth_data (comics_document->page_names, page)); + quoted_filename = comics_regex_quote (comics_document->page_names->pdata[page]); } else { - quoted_filename = g_shell_quote ( - g_slist_nth_data (comics_document->page_names, page)); + quoted_filename = g_shell_quote (comics_document->page_names->pdata[page]); } command_line = g_strdup_printf ("%s -- %s %s", @@ -827,13 +856,13 @@ extract_argv (EvDocument *document, gint page) quoted_filename); g_shell_parse_argv (command_line, NULL, &argv, &err); - + if (err) { g_warning (_("Error %s"), err->message); g_error_free (err); return NULL; } - + g_free (command_line); g_free (quoted_archive); g_free (quoted_filename);