debian/midori
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Handle unique_filename and null-check data source

Browse source
This commit is contained in:
Alexander Butenko 2011-06-09 00:37:25 +02:00 committed by Christian Dywan
parent 3a51aafd7f
commit 38da85ee43
1 changed files with 26 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
midori/midori-browser.c
View file

@ -1050,6 +1050,7 @@ midori_browser_save_uri (MidoriBrowser* browser,
frame = webkit_web_view_get_main_frame (WEBKIT_WEB_VIEW (web_view)); frame = webkit_web_view_get_main_frame (WEBKIT_WEB_VIEW (web_view));
data_source = webkit_web_frame_get_data_source (frame); data_source = webkit_web_frame_get_data_source (frame);
data = webkit_web_data_source_get_data (data_source); data = webkit_web_data_source_get_data (data_source);
if (data)
midori_browser_save_source (uri, data->str, data->len, filename); midori_browser_save_source (uri, data->str, data->len, filename);
g_free (last_dir); g_free (last_dir);
@ -3300,8 +3301,6 @@ midori_browser_save_source (const gchar* uri,
const size_t len, const size_t len,
const gchar* outfile) const gchar* outfile)
{ {
gchar* filename;
gchar* extension;
gchar* unique_filename; gchar* unique_filename;
gint fd; gint fd;
FILE* fp; FILE* fp;
@ -3312,16 +3311,20 @@ midori_browser_save_source (const gchar* uri,
if (!outfile) if (!outfile)
{ {
gchar* filename;
gchar* extension;
extension = midori_browser_get_uri_extension (uri); extension = midori_browser_get_uri_extension (uri);
filename = g_strdup_printf ("%uXXXXXX%s", filename = g_strdup_printf ("%uXXXXXX%s",
g_str_hash (uri), extension); g_str_hash (uri), extension);
g_free (extension); g_free (extension);
fd = g_file_open_tmp (filename, &unique_filename, NULL); fd = g_file_open_tmp (filename, &unique_filename, NULL);
g_free (filename);
} }
else else
{ {
filename = g_strdup (outfile); unique_filename = g_strdup (outfile);
fd = g_open (filename, O_WRONLY|O_CREAT, 0644); fd = g_open (unique_filename, O_WRONLY|O_CREAT, 0644);
} }
if (fd != -1) if (fd != -1)
@ -3334,12 +3337,11 @@ midori_browser_save_source (const gchar* uri,
{ {
g_warning ("Error writing to file %s " g_warning ("Error writing to file %s "
"in midori_browser_source_transfer_cb()", unique_filename); "in midori_browser_source_transfer_cb()", unique_filename);
g_free (unique_filename); katze_assign (unique_filename, NULL);
} }
} }
close (fd); close (fd);
} }
g_free (filename);
return unique_filename; return unique_filename;
} }
@ -3347,6 +3349,9 @@ static void
_action_source_view_activate (GtkAction* action, _action_source_view_activate (GtkAction* action,
MidoriBrowser* browser) MidoriBrowser* browser)
{ {
WebKitWebDataSource *data_source;
WebKitWebFrame *frame;
const GString *data;
GtkWidget* view; GtkWidget* view;
GtkWidget* web_view; GtkWidget* web_view;
gchar* text_editor; gchar* text_editor;
@ -3358,40 +3363,34 @@ _action_source_view_activate (GtkAction* action,
g_object_get (browser->settings, "text-editor", &text_editor, NULL); g_object_get (browser->settings, "text-editor", &text_editor, NULL);
uri = midori_view_get_display_uri (MIDORI_VIEW (view)); uri = midori_view_get_display_uri (MIDORI_VIEW (view));
web_view = midori_view_get_web_view (MIDORI_VIEW (view));
frame = webkit_web_view_get_main_frame (WEBKIT_WEB_VIEW (web_view));
data_source = webkit_web_frame_get_data_source (frame);
data = webkit_web_data_source_get_data (data_source);
if (data)
filename = midori_browser_save_source (uri, data->str, data->len, NULL);
if (!filename)
return;
if (!(text_editor && *text_editor)) if (!(text_editor && *text_editor))
{ {
GtkWidget* source; GtkWidget* source;
GtkWidget* web_view; GtkWidget* web_view;
gchar* source_uri;
source_uri = g_filename_to_uri (filename, NULL, NULL);
g_free (filename);
source = midori_view_new (NULL); source = midori_view_new (NULL);
midori_view_set_settings (MIDORI_VIEW (source), browser->settings); midori_view_set_settings (MIDORI_VIEW (source), browser->settings);
web_view = midori_view_get_web_view (MIDORI_VIEW (source)); web_view = midori_view_get_web_view (MIDORI_VIEW (source));
webkit_web_view_set_view_source_mode (WEBKIT_WEB_VIEW (web_view), TRUE); webkit_web_view_set_view_source_mode (WEBKIT_WEB_VIEW (web_view), TRUE);
webkit_web_view_load_uri (WEBKIT_WEB_VIEW (web_view), uri); webkit_web_view_load_uri (WEBKIT_WEB_VIEW (web_view), source_uri);
gtk_widget_show (source); gtk_widget_show (source);
midori_browser_add_tab (browser, source); midori_browser_add_tab (browser, source);
g_free (text_editor);
return;
} }
if (g_str_has_prefix (uri, "file://"))
filename = g_filename_from_uri (uri, NULL, NULL);
else else
{
WebKitWebDataSource *data_source;
WebKitWebFrame *frame;
const GString *data;
web_view = midori_view_get_web_view (MIDORI_VIEW (view));
frame = webkit_web_view_get_main_frame (WEBKIT_WEB_VIEW (web_view));
data_source = webkit_web_frame_get_data_source (frame);
data = webkit_web_data_source_get_data (data_source);
filename = midori_browser_save_source (uri, data->str, data->len, NULL);
}
if (filename)
{ {
sokoke_spawn_program (text_editor, filename); sokoke_spawn_program (text_editor, filename);
g_free (filename); g_free (filename);