/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

/* GTK - The GIMP Toolkit

 * Copyright (C) David Zeuthen <davidz@redhat.com>

 * Copyright (C) 2001 Havoc Pennington

 * Copyright (C) 2005-2007 Vincent Untz

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2 of the License, or (at your option) any later version.

 *

 * This library 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with this library; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

/*

 * Modified by the GTK+ Team and others 1997-2000.  See the AUTHORS

 * file for a list of people on the GTK+ Team.  See the ChangeLog

 * files for a list of changes.  These files are distributed with

 * GTK+ at ftp://ftp.gtk.org/pub/gtk/.

 */

#include "config.h"

#include <string.h>

#include <stdlib.h>

#include <gio/gio.h>

#include "x11/gdkx.h"

#include <X11/Xatom.h>

#include <gtk/gtkicontheme.h>

#include "gtkintl.h"

/* for the kill(2) system call and errno - POSIX.1-2001 and later */

#include <sys/types.h>

#include <signal.h>

#include <errno.h>

#if defined(__OpenBSD__)

#include <sys/param.h>

#include <kvm.h>

#include <fcntl.h>

#include <sys/sysctl.h>

#endif

#include "gtkmountoperationprivate.h"

#include "gtkalias.h"

/* ---------------------------------------------------------------------------------------------------- */

/* these functions are based on code from libwnck (LGPLv2) */

static gboolean get_window_list   (Display   *xdisplay,

                                   Window     xwindow,

                                   Atom       atom,

                                   Window   **windows,

                                   int       *len);

static char*    get_utf8_property (Display   *xdisplay,

                                   Window     xwindow,

                                   Atom       atom);

static gboolean get_cardinal      (Display   *xdisplay,

                                   Window     xwindow,

                                   Atom       atom,

                                   int       *val);

static gboolean read_rgb_icon     (Display   *xdisplay,

                                   Window     xwindow,

                                   int        ideal_width,

                                   int        ideal_height,

                                   int       *width,

                                   int       *height,

                                   guchar   **pixdata);

static gboolean

get_cardinal (Display *xdisplay,

              Window   xwindow,

              Atom     atom,

              int     *val)

{

  Atom type;

  int format;

  gulong nitems;

  gulong bytes_after;

  gulong *num;

  int err, result;

  *val = 0;

  gdk_error_trap_push ();

  type = None;

  result = XGetWindowProperty (xdisplay,

                               xwindow,

                               atom,

                               0, G_MAXLONG,

                               False, XA_CARDINAL, &type, &format, &nitems,

                               &bytes_after, (void*)&num);

  XSync (xdisplay, False);

  err = gdk_error_trap_pop ();

  if (err != Success ||

      result != Success)

    return FALSE;

  if (type != XA_CARDINAL)

    {

      XFree (num);

      return FALSE;

    }

  *val = *num;

  XFree (num);

  return TRUE;

}

static char*

get_utf8_property (Display *xdisplay,

                   Window   xwindow,

                   Atom     atom)

{

  Atom type;

  int format;

  gulong nitems;

  gulong bytes_after;

  gchar *val;

  int err, result;

  char *retval;

  Atom utf8_string;

  utf8_string = gdk_x11_get_xatom_by_name ("UTF8_STRING");

  gdk_error_trap_push ();

  type = None;

  val = NULL;

  result = XGetWindowProperty (xdisplay,

                               xwindow,

                               atom,

                               0, G_MAXLONG,

                               False, utf8_string,

                               &type, &format, &nitems,

                               &bytes_after, (guchar **)&val;;

  XSync (xdisplay, False);

  err = gdk_error_trap_pop ();

  if (err != Success ||

      result != Success)

    return NULL;

  if (type != utf8_string ||

      format != 8 ||

      nitems == 0)

    {

      if (val)

        XFree (val);

      return NULL;

    }

  if (!g_utf8_validate (val, nitems, NULL))

    {

      g_warning ("Property %s contained invalid UTF-8\n",

                 gdk_x11_get_xatom_name (atom));

      XFree (val);

      return NULL;

    }

  retval = g_strndup (val, nitems);

  XFree (val);

  return retval;

}

static gboolean

find_largest_sizes (gulong *data,

                    gulong  nitems,

                    int    *width,

                    int    *height)

{

  *width = 0;

  *height = 0;

  while (nitems > 0)

    {

      int w, h;

      gboolean replace;

      replace = FALSE;

      if (nitems < 3)

        return FALSE; /* no space for w, h */

      w = data[0];

      h = data[1];

      if (nitems < ((w * h) + 2))

        return FALSE; /* not enough data */

      *width = MAX (w, *width);

      *height = MAX (h, *height);

      data += (w * h) + 2;

      nitems -= (w * h) + 2;

    }

  return TRUE;

}

static gboolean

find_best_size (gulong  *data,

                gulong   nitems,

                int      ideal_width,

                int      ideal_height,

                int     *width,

                int     *height,

                gulong **start)

{

  int best_w;

  int best_h;

  gulong *best_start;

  int max_width, max_height;

  *width = 0;

  *height = 0;

  *start = NULL;

  if (!find_largest_sizes (data, nitems, &max_width, &max_height))

    return FALSE;

  if (ideal_width < 0)

    ideal_width = max_width;

  if (ideal_height < 0)

    ideal_height = max_height;

  best_w = 0;

  best_h = 0;

  best_start = NULL;

  while (nitems > 0)

    {

      int w, h;

      gboolean replace;

      replace = FALSE;

      if (nitems < 3)

        return FALSE; /* no space for w, h */

      w = data[0];

      h = data[1];

      if (nitems < ((w * h) + 2))

        break; /* not enough data */

      if (best_start == NULL)

        {

          replace = TRUE;

        }

      else

        {

          /* work with averages */

          const int ideal_size = (ideal_width + ideal_height) / 2;

          int best_size = (best_w + best_h) / 2;

          int this_size = (w + h) / 2;

          /* larger than desired is always better than smaller */

          if (best_size < ideal_size &&

              this_size >= ideal_size)

            replace = TRUE;

          /* if we have too small, pick anything bigger */

          else if (best_size < ideal_size &&

                   this_size > best_size)

            replace = TRUE;

          /* if we have too large, pick anything smaller

           * but still >= the ideal

           */

          else if (best_size > ideal_size &&

                   this_size >= ideal_size &&

                   this_size < best_size)

            replace = TRUE;

        }

      if (replace)

        {

          best_start = data + 2;

          best_w = w;

          best_h = h;

        }

      data += (w * h) + 2;

      nitems -= (w * h) + 2;

    }

  if (best_start)

    {

      *start = best_start;

      *width = best_w;

      *height = best_h;

      return TRUE;

    }

  else

    return FALSE;

}

static void

argbdata_to_pixdata (gulong  *argb_data,

                     int      len,

                     guchar **pixdata)

{

  guchar *p;

  int i;

  *pixdata = g_new (guchar, len * 4);

  p = *pixdata;

  /* One could speed this up a lot. */

  i = 0;

  while (i < len)

    {

      guint argb;

      guint rgba;

      argb = argb_data[i];

      rgba = (argb << 8) | (argb >> 24);

      *p = rgba >> 24;

      ++p;

      *p = (rgba >> 16) & 0xff;

      ++p;

      *p = (rgba >> 8) & 0xff;

      ++p;

      *p = rgba & 0xff;

      ++p;

      ++i;

    }

}

static gboolean

read_rgb_icon (Display   *xdisplay,

               Window     xwindow,

               int        ideal_width,

               int        ideal_height,

               int       *width,

               int       *height,

               guchar   **pixdata)

{

  Atom type;

  int format;

  gulong nitems;

  gulong bytes_after;

  int result, err;

  gulong *data;

  gulong *best;

  int w, h;

  gdk_error_trap_push ();

  type = None;

  data = NULL;

  result = XGetWindowProperty (xdisplay,

                               xwindow,

                               gdk_x11_get_xatom_by_name ("_NET_WM_ICON"),

                               0, G_MAXLONG,

                               False, XA_CARDINAL, &type, &format, &nitems,

                               &bytes_after, (void*)&data);

  XSync (xdisplay, False);

  err = gdk_error_trap_pop ();

  if (err != Success ||

      result != Success)

    return FALSE;

  if (type != XA_CARDINAL)

    {

      XFree (data);

      return FALSE;

    }

  if (!find_best_size (data, nitems,

                       ideal_width, ideal_height,

                       &w, &h, &best))

    {

      XFree (data);

      return FALSE;

    }

  *width = w;

  *height = h;

  argbdata_to_pixdata (best, w * h, pixdata);

  XFree (data);

  return TRUE;

}

static void

free_pixels (guchar *pixels, gpointer data)

{

  g_free (pixels);

}

static GdkPixbuf*

scaled_from_pixdata (guchar *pixdata,

                     int     w,

                     int     h,

                     int     new_w,

                     int     new_h)

{

  GdkPixbuf *src;

  GdkPixbuf *dest;

  src = gdk_pixbuf_new_from_data (pixdata,

                                  GDK_COLORSPACE_RGB,

                                  TRUE,

                                  8,

                                  w, h, w * 4,

                                  free_pixels,

                                  NULL);

  if (src == NULL)

    return NULL;

  if (w != h)

    {

      GdkPixbuf *tmp;

      int size;

      size = MAX (w, h);

      tmp = gdk_pixbuf_new (GDK_COLORSPACE_RGB, TRUE, 8, size, size);

      if (tmp != NULL)

        {

          gdk_pixbuf_fill (tmp, 0);

          gdk_pixbuf_copy_area (src, 0, 0, w, h,

                                tmp,

                                (size - w) / 2, (size - h) / 2);

          g_object_unref (src);

          src = tmp;

        }

    }

  if (w != new_w || h != new_h)

    {

      dest = gdk_pixbuf_scale_simple (src, new_w, new_h, GDK_INTERP_BILINEAR);

      g_object_unref (G_OBJECT (src));

    }

  else

    {

      dest = src;

    }

  return dest;

}

static gboolean

get_window_list (Display  *xdisplay,

                 Window    xwindow,

                 Atom      atom,

                 Window  **windows,

                 int      *len)

{

  Atom type;

  int format;

  gulong nitems;

  gulong bytes_after;

  Window *data;

  int err, result;

  *windows = NULL;

  *len = 0;

  gdk_error_trap_push ();

  type = None;

  result = XGetWindowProperty (xdisplay,

                               xwindow,

                               atom,

                               0, G_MAXLONG,

                               False, XA_WINDOW, &type, &format, &nitems,

                               &bytes_after, (void*)&data);

  XSync (xdisplay, False);

  err = gdk_error_trap_pop ();

  if (err != Success ||

      result != Success)

    return FALSE;

  if (type != XA_WINDOW)

    {

      XFree (data);

      return FALSE;

    }

  *windows = g_new (Window, nitems);

  memcpy (*windows, data, sizeof (Window) * nitems);

  *len = nitems;

  XFree (data);

  return TRUE;

}

/* ---------------------------------------------------------------------------------------------------- */

struct _GtkMountOperationLookupContext

{

  /* Hash from pid (gint) -> XID (gint)

   *

   * Note that XIDs are at most 27 bits - however, also note that sizeof(XID) == 8 on

   * x86_64 - that's just xlib brokenness. So it's safe to stuff the XID into a pointer.

   */

  GHashTable *pid_to_window;

  GdkDisplay *display;

};

GtkMountOperationLookupContext *

_gtk_mount_operation_lookup_context_get (GdkDisplay *display)

{

  GtkMountOperationLookupContext *context;

  Window *mapping;

  gint mapping_length;

  gint n;

  context = g_new0 (GtkMountOperationLookupContext, 1);

  context->pid_to_window = g_hash_table_new (g_direct_hash, g_direct_equal);

  context->display = display;

  mapping = NULL;

  mapping_length = 0;

  get_window_list (GDK_DISPLAY_XDISPLAY (context->display),

                   GDK_ROOT_WINDOW(),

                   gdk_x11_get_xatom_by_name_for_display (context->display,

                                                          "_NET_CLIENT_LIST"),

                   &mapping,

                   &mapping_length);

  for (n = 0; n < mapping_length; n++)

    {

      gint pid;

      if (!get_cardinal (GDK_DISPLAY_XDISPLAY (context->display),

                         mapping[n],

                         gdk_x11_get_xatom_by_name_for_display (context->display,

                                                                "_NET_WM_PID"),

                         &pid))

        continue;

      g_hash_table_insert (context->pid_to_window,

                           GINT_TO_POINTER (pid),

                           GINT_TO_POINTER ((gint) mapping[n]));

    }

  g_free (mapping);

  return context;

}

void

_gtk_mount_operation_lookup_context_free (GtkMountOperationLookupContext *context)

{

  g_hash_table_unref (context->pid_to_window);

  g_free (context);

}

/* ---------------------------------------------------------------------------------------------------- */

#ifdef __linux__

static GPid

pid_get_parent (GPid pid)

{

  GPid ppid;

  gchar **tokens;

  gchar *stat_filename;

  gchar *stat_contents;

  gsize stat_len;

  ppid = 0;

  tokens = NULL;

  stat_contents = NULL;

  stat_filename = NULL;

  /* fail if trying to get the parent of the init process (no such thing) */

  if (pid == 1)

      goto out;

  stat_filename = g_strdup_printf ("/proc/%d/status", pid);

  if (g_file_get_contents (stat_filename,

                           &stat_contents,

                           &stat_len,

                           NULL))

    {

      guint n;

      tokens = g_strsplit (stat_contents, "\n", 0);

      for (n = 0; tokens[n] != NULL; n++)

        {

          if (g_str_has_prefix (tokens[n], "PPid:"))

            {

              gchar *endp;

              endp = NULL;

              ppid = strtoll (tokens[n] + sizeof "PPid:" - 1, &endp, 10);

              if (endp == NULL || *endp != '\0')

                {

                  g_warning ("Error parsing contents of `%s'. Parent pid is malformed.",

                             stat_filename);

                  ppid = 0;

                  goto out;

                }

              break;

            }

        }

    }

 out:

  g_strfreev (tokens);

  g_free (stat_contents);

  g_free (stat_filename);

  return ppid;

}

static gchar *

pid_get_env (GPid         pid,

             const gchar *key)

{

  gchar *ret;

  gchar *env_filename;

  gchar *env;

  gsize env_len;

  gsize key_len;

  gchar *end;

  ret = NULL;

  key_len = strlen (key);

  env_filename = g_strdup_printf ("/proc/%d/environ", pid);

  if (g_file_get_contents (env_filename,

                           &env,

                           &env_len,

                           NULL))

    {

      guint n;

      /* /proc/<pid>/environ in Linux is split at '\0' points, g_strsplit() can't handle that... */

      n = 0;

      while (TRUE)

        {

          if (env[n] == '\0' || n >= env_len)

            break;

          if (g_str_has_prefix (env + n, key) && (*(env + n + key_len) == '='))

            {

              ret = g_strdup (env + n + key_len + 1);

              /* skip invalid UTF-8 */

              if (!g_utf8_validate (ret, -1, (const gchar **) &end))

                *end = '\0';

              break;

            }

          for (; env[n] != '\0' && n < env_len; n++)

            ;

          n++;

        }

      g_free (env);

    }

  g_free (env_filename);

  return ret;

}

static gchar *

pid_get_command_line (GPid pid)

{

  gchar *cmdline_filename;

  gchar *cmdline_contents;

  gsize cmdline_len;

  guint n;

  gchar *end;

  cmdline_contents = NULL;

  cmdline_filename = g_strdup_printf ("/proc/%d/cmdline", pid);

  if (!g_file_get_contents (cmdline_filename,

                            &cmdline_contents,

                            &cmdline_len,

                            NULL))

    goto out;

  /* /proc/<pid>/cmdline separates args by NUL-bytes - replace with spaces */

  for (n = 0; n < cmdline_len - 1; n++)

    {

      if (cmdline_contents[n] == '\0')

        cmdline_contents[n] = ' ';

    }

  /* skip invalid UTF-8 */

  if (!g_utf8_validate (cmdline_contents, -1, (const gchar **) &end))

      *end = '\0';

 out:

  g_free (cmdline_filename);

  return cmdline_contents;

}

/* ---------------------------------------------------------------------------------------------------- */

#else

/* TODO: please implement for your OS - must return valid UTF-8 */

static GPid

pid_get_parent (GPid pid)

{

  return 0;

}

static gchar *

pid_get_env (GPid         pid,

             const gchar *key)

{

  return NULL;

}

static gchar *

pid_get_command_line (GPid pid)

{

  return NULL;

}

#endif

/* ---------------------------------------------------------------------------------------------------- */

static gchar *

get_name_for_window_with_pid (GtkMountOperationLookupContext *context,

                              GPid                            pid)

{

  Window window;

  Window windowid_window;

  gchar *ret;

  ret = NULL;

  window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));

  if (window == None)

    {

      gchar *windowid_value;

      /* check for $WINDOWID (set by terminals) and see if we can get the title that way */

      windowid_value = pid_get_env (pid, "WINDOWID");

      if (windowid_value != NULL)

        {

          gchar *endp;

          endp = NULL;

          windowid_window = (Window) g_ascii_strtoll (windowid_value, &endp, 10);

          if (endp != NULL || *endp == '\0')

            {

              window = windowid_window;

            }

          g_free (windowid_value);

        }

      /* otherwise, check for parents */

      if (window == None)

        {

          do

            {

              pid = pid_get_parent (pid);

              if (pid == 0)

                break;

              window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));

              if (window != None)

                break;

            }

          while (TRUE);

        }

    }

  if (window != None)

    {

      ret = get_utf8_property (GDK_DISPLAY_XDISPLAY (context->display),

                               window,

                               gdk_x11_get_xatom_by_name_for_display (context->display,

                                                                      "_NET_WM_NAME"));

      if (ret == NULL)

        ret = get_utf8_property (GDK_DISPLAY_XDISPLAY (context->display),

                                 window, gdk_x11_get_xatom_by_name_for_display (context->display,

                                                                                "_NET_WM_ICON_NAME"));

    }

  return ret;

}

/* ---------------------------------------------------------------------------------------------------- */

static GdkPixbuf *

get_pixbuf_for_window_with_pid (GtkMountOperationLookupContext *context,

                                GPid                            pid,

                                gint                            size_pixels)

{

  Window window;

  GdkPixbuf *ret;

  ret = NULL;

  window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));

  if (window == None)

    {

      /* otherwise, check for parents */

      do

        {

          pid = pid_get_parent (pid);

          if (pid == 0)

            break;

          window = GPOINTER_TO_INT (g_hash_table_lookup (context->pid_to_window, GINT_TO_POINTER (pid)));

          if (window != None)

            break;

        }

      while (TRUE);

    }

  if (window != None)

    {

      gint    width;

      gint    height;

      guchar *pixdata;

      if (read_rgb_icon (GDK_DISPLAY_XDISPLAY (context->display),

                         window,

                         size_pixels, size_pixels,

                         &width, &height,

                         &pixdata))

        {

          /* steals pixdata */

          ret = scaled_from_pixdata (pixdata,

                                     width, height,

                                     size_pixels, size_pixels);

        }

    }

  return ret;

}

/* ---------------------------------------------------------------------------------------------------- */

static const gchar *well_known_commands[] =

{

  /* translators: this string is a name for the 'less' command */

  "less", N_("Terminal Pager"),

  "top", N_("Top Command"),

  "bash", N_("Bourne Again Shell"),

  "sh", N_("Bourne Shell"),

  "zsh", N_("Z Shell"),

  NULL,

};

gboolean