gio/gio-querymodules.c

   1 /* GIO - GLib Input, Output and Streaming Library
   2  *
   3  * Copyright (C) 2006-2007 Red Hat, Inc.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.
  17  *
  18  * Author: Alexander Larsson <alexl@redhat.com>
  19  */
  20
  21 #include "config.h"
  22 #include "giomodule.h"
  23 #include "giomodule-priv.h"
  24
  25 #include <gstdio.h>
  26 #include <errno.h>
  27 #include <locale.h>
  28
  29 #include "glib/glib-private.h"
  30
  31 static gboolean
  32 is_valid_module_name (const gchar *basename)
  33 {
  34 #if !defined(G_OS_WIN32) && !defined(G_WITH_CYGWIN)
  35   return
  36     g_str_has_prefix (basename, "lib") &&
  37     g_str_has_suffix (basename, ".so");
  38 #else
  39   return g_str_has_suffix (basename, ".dll");
  40 #endif
  41 }
  42
  43 static void
  44 query_dir (const char *dirname)
  45 {
  46   GString *data;
  47   GDir *dir;
  48   GList *list = NULL, *iterator = NULL;
  49   const char *name;
  50   char *cachename;
  51   char **(* query)  (void);
  52   GError *error;
  53   int i;
  54
  55   if (!g_module_supported ())
  56     return;
  57
  58   error = NULL;
  59   dir = g_dir_open (dirname, 0, &error);
  60   if (!dir)
  61     {
  62       g_printerr ("Unable to open directory %s: %s\n", dirname, error->message);
  63       g_error_free (error);
  64       return;
  65     }
  66
  67   data = g_string_new ("");
  68
  69   while ((name = g_dir_read_name (dir)))
  70     list = g_list_prepend (list, g_strdup (name));
  71
  72   list = g_list_sort (list, (GCompareFunc) g_strcmp0);
  73   for (iterator = list; iterator; iterator = iterator->next)
  74     {
  75       GModule *module;
  76       gchar     *path;
  77       char **extension_points;
  78
  79       name = iterator->data;
  80       if (!is_valid_module_name (name))
  81         continue;
  82
  83       path = g_build_filename (dirname, name, NULL);
  84       module = g_module_open (path, G_MODULE_BIND_LAZY | G_MODULE_BIND_LOCAL);
  85       g_free (path);
  86
  87       if (module)
  88         {
  89           gchar *modulename;
  90           gchar *symname;
  91
  92           modulename = _g_io_module_extract_name (name);
  93           symname = g_strconcat ("g_io_", modulename, "_query", NULL);
  94           g_module_symbol (module, symname, (gpointer) &query);
  95           g_free (symname);
  96           g_free (modulename);
  97
  98           if (!query)
  99             {
 100               /* Fallback to old name */
 101               g_module_symbol (module, "g_io_module_query", (gpointer) &query);
 102             }
 103
 104           if (query)
 105             {
 106               extension_points = query ();
 107
 108               if (extension_points)
 109                 {
 110                   g_string_append_printf (data, "%s: ", name);
 111
 112                   for (i = 0; extension_points[i] != NULL; i++)
 113                     g_string_append_printf (data, "%s%s", i == 0 ? "" : ",", extension_points[i]);
 114
 115                   g_string_append (data, "\n");
 116                   g_strfreev (extension_points);
 117                 }
 118             }
 119
 120           g_module_close (module);
 121         }
 122     }
 123
 124   g_dir_close (dir);
 125   g_list_free_full (list, g_free);
 126
 127   cachename = g_build_filename (dirname, "giomodule.cache", NULL);
 128
 129   if (data->len > 0)
 130     {
 131       error = NULL;
 132
 133       if (!g_file_set_contents (cachename, data->str, data->len, &error))
 134         {
 135           g_printerr ("Unable to create %s: %s\n", cachename, error->message);
 136           g_error_free (error);
 137         }
 138     }
 139   else
 140     {
 141       if (g_unlink (cachename) != 0 && errno != ENOENT)
 142         {
 143           int errsv = errno;
 144           g_printerr ("Unable to unlink %s: %s\n", cachename, g_strerror (errsv));
 145         }
 146     }
 147
 148   g_free (cachename);
 149   g_string_free (data, TRUE);
 150 }
 151
 152 int
 153 main (gint   argc,
 154       gchar *argv[])
 155 {
 156   int i;
 157
 158   if (argc == 1)
 159     {
 160       g_print ("Usage: gio-querymodules <directory1> [<directory2> ...]\n");
 161       g_print ("Will update giomodule.cache in the listed directories\n");
 162       return 1;
 163     }
 164
 165   setlocale (LC_ALL, GLIB_DEFAULT_LOCALE);
 166
 167   /* Be defensive and ensure we're linked to GObject */
 168   g_type_ensure (G_TYPE_OBJECT);
 169
 170   for (i = 1; i < argc; i++)
 171     query_dir (argv[i]);
 172
 173   return 0;
 174 }