1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* GTK - The GIMP Toolkit
3 * Copyright (C) David Zeuthen <davidz@redhat.com>
4 * Copyright (C) 2001 Havoc Pennington
5 * Copyright (C) 2005-2007 Vincent Untz
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the
19 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
20 * Boston, MA 02111-1307, USA.
24 * Modified by the GTK+ Team and others 1997-2000. See the AUTHORS
25 * file for a list of people on the GTK+ Team. See the ChangeLog
26 * files for a list of changes. These files are distributed with
27 * GTK+ at ftp://ftp.gtk.org/pub/gtk/.
36 #include <X11/Xatom.h>
37 #include <gtk/gtkicontheme.h>
40 /* for the kill(2) system call and errno - POSIX.1-2001 and later */
41 #include <sys/types.h>
45 #include "gtkmountoperationprivate.h"
48 /* ---------------------------------------------------------------------------------------------------- */
49 /* these functions are based on code from libwnck (LGPLv2) */
51 static gboolean get_window_list (Display *xdisplay,
57 static char* get_utf8_property (Display *xdisplay,
61 static gboolean get_cardinal (Display *xdisplay,
66 static gboolean read_rgb_icon (Display *xdisplay,
76 get_cardinal (Display *xdisplay,
90 gdk_error_trap_push ();
92 result = XGetWindowProperty (xdisplay,
96 False, XA_CARDINAL, &type, &format, &nitems,
97 &bytes_after, (void*)&num);
98 XSync (xdisplay, False);
99 err = gdk_error_trap_pop ();
101 if (err != Success ||
105 if (type != XA_CARDINAL)
119 get_utf8_property (Display *xdisplay,
132 utf8_string = gdk_x11_get_xatom_by_name ("UTF8_STRING");
134 gdk_error_trap_push ();
137 result = XGetWindowProperty (xdisplay,
142 &type, &format, &nitems,
143 &bytes_after, (guchar **)&val);
144 XSync (xdisplay, False);
145 err = gdk_error_trap_pop ();
147 if (err != Success ||
151 if (type != utf8_string ||
160 if (!g_utf8_validate (val, nitems, NULL))
162 g_warning ("Property %s contained invalid UTF-8\n",
163 gdk_x11_get_xatom_name (atom));
168 retval = g_strndup (val, nitems);
176 find_largest_sizes (gulong *data,
192 return FALSE; /* no space for w, h */
197 if (nitems < ((w * h) + 2))
198 return FALSE; /* not enough data */
200 *width = MAX (w, *width);
201 *height = MAX (h, *height);
204 nitems -= (w * h) + 2;
211 find_best_size (gulong *data,
222 int max_width, max_height;
228 if (!find_largest_sizes (data, nitems, &max_width, &max_height))
232 ideal_width = max_width;
233 if (ideal_height < 0)
234 ideal_height = max_height;
248 return FALSE; /* no space for w, h */
253 if (nitems < ((w * h) + 2))
254 break; /* not enough data */
256 if (best_start == NULL)
262 /* work with averages */
263 const int ideal_size = (ideal_width + ideal_height) / 2;
264 int best_size = (best_w + best_h) / 2;
265 int this_size = (w + h) / 2;
267 /* larger than desired is always better than smaller */
268 if (best_size < ideal_size &&
269 this_size >= ideal_size)
271 /* if we have too small, pick anything bigger */
272 else if (best_size < ideal_size &&
273 this_size > best_size)
275 /* if we have too large, pick anything smaller
276 * but still >= the ideal
278 else if (best_size > ideal_size &&
279 this_size >= ideal_size &&
280 this_size < best_size)
286 best_start = data + 2;
292 nitems -= (w * h) + 2;
307 argbdata_to_pixdata (gulong *argb_data,
314 *pixdata = g_new (guchar, len * 4);
317 /* One could speed this up a lot. */
325 rgba = (argb << 8) | (argb >> 24);
329 *p = (rgba >> 16) & 0xff;
331 *p = (rgba >> 8) & 0xff;
341 read_rgb_icon (Display *xdisplay,
358 gdk_error_trap_push ();
361 result = XGetWindowProperty (xdisplay,
363 gdk_x11_get_xatom_by_name ("_NET_WM_ICON"),
365 False, XA_CARDINAL, &type, &format, &nitems,
366 &bytes_after, (void*)&data);
368 XSync (xdisplay, False);
369 err = gdk_error_trap_pop ();
371 if (err != Success ||
375 if (type != XA_CARDINAL)
381 if (!find_best_size (data, nitems,
382 ideal_width, ideal_height,
392 argbdata_to_pixdata (best, w * h, pixdata);
400 free_pixels (guchar *pixels, gpointer data)
406 scaled_from_pixdata (guchar *pixdata,
415 src = gdk_pixbuf_new_from_data (pixdata,
433 tmp = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8, size, size);
437 gdk_pixbuf_fill (tmp, 0);
438 gdk_pixbuf_copy_area (src, 0, 0, w, h,
440 (size - w) / 2, (size - h) / 2);
442 g_object_unref (src);
447 if (w != new_w || h != new_h)
449 dest = gdk_pixbuf_scale_simple (src, new_w, new_h, GDK_INTERP_BILINEAR);
451 g_object_unref (G_OBJECT (src));
462 get_window_list (Display *xdisplay,
478 gdk_error_trap_push ();
480 result = XGetWindowProperty (xdisplay,
484 False, XA_WINDOW, &type, &format, &nitems,
485 &bytes_after, (void*)&data);
486 XSync (xdisplay, False);
487 err = gdk_error_trap_pop ();
489 if (err != Success ||
493 if (type != XA_WINDOW)
499 *windows = g_new (Window, nitems);
500 memcpy (*windows, data, sizeof (Window) * nitems);
509 /* ---------------------------------------------------------------------------------------------------- */
511 struct _GtkMountOperationLookupContext
513 /* Hash from pid (gint) -> XID (gint)
515 * Note that XIDs are at most 27 bits - however, also note that sizeof(XID) == 8 on
516 * x86_64 - that's just xlib brokenness. So it's safe to stuff the XID into a pointer.
518 GHashTable *pid_to_window;
522 GtkMountOperationLookupContext *
523 _gtk_mount_operation_lookup_context_get (GdkDisplay *display)
525 GtkMountOperationLookupContext *context;
530 context = g_new0 (GtkMountOperationLookupContext, 1);
532 context->pid_to_window = g_hash_table_new (g_direct_hash, g_direct_equal);
533 context->display = display;
537 get_window_list (GDK_DISPLAY_XDISPLAY (context->display),
539 gdk_x11_get_xatom_by_name_for_display (context->display,
543 for (n = 0; n < mapping_length; n++)
547 if (!get_cardinal (GDK_DISPLAY_XDISPLAY (context->display),
549 gdk_x11_get_xatom_by_name_for_display (context->display,
554 g_hash_table_insert (context->pid_to_window,
555 GINT_TO_POINTER (pid),
556 GINT_TO_POINTER ((gint) mapping[n]));
564 _gtk_mount_operation_lookup_context_free (GtkMountOperationLookupContext *context)
566 g_hash_table_unref (context->pid_to_window);
570 /* ---------------------------------------------------------------------------------------------------- */
575 pid_get_parent (GPid pid)
579 gchar *stat_filename;
580 gchar *stat_contents;
585 stat_contents = NULL;
586 stat_filename = NULL;
588 /* fail if trying to get the parent of the init process (no such thing) */
592 stat_filename = g_strdup_printf ("/proc/%d/status", pid);
593 if (g_file_get_contents (stat_filename,
600 tokens = g_strsplit (stat_contents, "\n", 0);
602 for (n = 0; tokens[n] != NULL; n++)
604 if (g_str_has_prefix (tokens[n], "PPid:"))
609 ppid = strtoll (tokens[n] + sizeof "PPid:" - 1, &endp, 10);
610 if (endp == NULL || *endp != '\0')
612 g_warning ("Error parsing contents of `%s'. Parent pid is malformed.",
625 g_free (stat_contents);
626 g_free (stat_filename);
632 pid_get_env (GPid pid,
644 key_len = strlen (key);
646 env_filename = g_strdup_printf ("/proc/%d/environ", pid);
647 if (g_file_get_contents (env_filename,
654 /* /proc/<pid>/environ in Linux is split at '\0' points, g_strsplit() can't handle that... */
658 if (env[n] == '\0' || n >= env_len)
661 if (g_str_has_prefix (env + n, key) && (*(env + n + key_len) == '='))
663 ret = g_strdup (env + n + key_len + 1);
665 /* skip invalid UTF-8 */
666 if (!g_utf8_validate (ret, -1, (const gchar *) &end))
671 for (; env[n] != '\0' && n < env_len; n++)
677 g_free (env_filename);
683 pid_get_command_line (GPid pid)
685 gchar *cmdline_filename;
686 gchar *cmdline_contents;
691 cmdline_contents = NULL;
693 cmdline_filename = g_strdup_printf ("/proc/%d/cmdline", pid);
694 if (!g_file_get_contents (cmdline_filename,
700 /* /proc/<pid>/cmdline separates args by NUL-bytes - replace with spaces */
701 for (n = 0; n < cmdline_len - 1; n++)
703 if (cmdline_contents[n] == '\0')
704 cmdline_contents[n] = ' ';
707 /* skip invalid UTF-8 */
708 if (!g_utf8_validate (cmdline_contents, -1, (const gchar *) &end))
712 g_free (cmdline_filename);
714 return cmdline_contents;
717 /* ---------------------------------------------------------------------------------------------------- */
720 /* TODO: please implement for your OS - must return valid UTF-8 */
723 pid_get_parent (GPid pid)
729 pid_get_env (GPid pid,
736 pid_get_command_line (GPid pid)
743 /* ---------------------------------------------------------------------------------------------------- */
746 get_name_for_window_with_pid (GtkMountOperationLookupContext *context,
750 Window windowid_window;
755 window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));
758 gchar *windowid_value;
760 /* check for $WINDOWID (set by terminals) and see if we can get the title that way */
761 windowid_value = pid_get_env (pid, "WINDOWID");
762 if (windowid_value != NULL)
767 windowid_window = (Window) g_ascii_strtoll (windowid_value, &endp, 10);
768 if (endp != NULL || *endp == '\0')
770 window = windowid_window;
772 g_free (windowid_value);
775 /* otherwise, check for parents */
780 pid = pid_get_parent (pid);
784 window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));
794 ret = get_utf8_property (GDK_DISPLAY_XDISPLAY (context->display),
796 gdk_x11_get_xatom_by_name_for_display (context->display,
799 ret = get_utf8_property (GDK_DISPLAY_XDISPLAY (context->display),
800 window, gdk_x11_get_xatom_by_name_for_display (context->display,
801 "_NET_WM_ICON_NAME"));
807 /* ---------------------------------------------------------------------------------------------------- */
810 get_pixbuf_for_window_with_pid (GtkMountOperationLookupContext *context,
819 window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));
822 /* otherwise, check for parents */
825 pid = pid_get_parent (pid);
829 window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));
842 if (read_rgb_icon (GDK_DISPLAY_XDISPLAY (context->display),
844 size_pixels, size_pixels,
849 ret = scaled_from_pixdata (pixdata,
851 size_pixels, size_pixels);
858 /* ---------------------------------------------------------------------------------------------------- */
860 static const gchar *well_known_commands[] =
862 "less", N_("Terminal Pager"),
863 "top", N_("Top Command"),
864 "bash", N_("Bourne Again Shell"),
865 "sh", N_("Bourne Shell"),
866 "zsh", N_("Z Shell"),
871 _gtk_mount_operation_lookup_info (GtkMountOperationLookupContext *context,
875 gchar **out_command_line,
876 GdkPixbuf **out_pixbuf)
878 g_return_val_if_fail (out_name != NULL && *out_name == NULL, FALSE);
879 g_return_val_if_fail (out_command_line != NULL && *out_command_line == NULL, FALSE);
880 g_return_val_if_fail (out_pixbuf != NULL && *out_pixbuf == NULL, FALSE);
882 /* We perform two different lookups for name and icon size.. this is
883 * because we want the name from the window with WINDOWID and this
884 * normally does not give you an icon
886 * (the canonical example is a tab in gnome-terminal - the shell/command running
887 * in the shell will have WINDOWID set - but this window won't have an icon - so
888 * we want to continue up until the gnome-terminal window so we can get that icon)
891 *out_command_line = pid_get_command_line (pid);
893 *out_name = get_name_for_window_with_pid (context, pid);
895 *out_pixbuf = get_pixbuf_for_window_with_pid (context, pid, size_pixels);
897 /* if we didn't manage to find the name via X, fall back to the basename
898 * of the first element of the command line and, for maximum geek-comfort,
899 * map a few well-known commands to proper translated names
901 if (*out_name == NULL && *out_command_line != NULL &&
902 strlen (*out_command_line) > 0 && (*out_command_line)[0] != ' ')
908 /* find the first character after the first argument */
909 s = strchr (*out_command_line, ' ');
911 s = *out_command_line + strlen (*out_command_line);
913 for (p = s; p > *out_command_line; p--)
922 *out_name = g_strndup (p, s - p);
924 for (n = 0; well_known_commands[n] != NULL; n += 2)
926 /* sometimes the command is prefixed with a -, e.g. '-bash' instead
927 * of 'bash' - handle that as well
929 if ((strcmp (well_known_commands[n], *out_name) == 0) ||
930 ((*out_name)[0] == '-' && (strcmp (well_known_commands[n], (*out_name) + 1) == 0)))
933 *out_name = g_strdup (_(well_known_commands[n+1]));
943 _gtk_mount_operation_kill_process (GPid pid,
950 if (kill ((pid_t) pid, SIGTERM) != 0)
952 /* TODO: On EPERM, we could use a setuid helper using polkit (very easy to implement
953 * via pkexec(1)) to allow the user to e.g. authenticate to gain the authorization
954 * to kill the process. But that's not how things currently work.
960 g_io_error_from_errno (errno),
961 _("Cannot end process with pid %d: %s"),