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 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
  24 #include <gstdio.h>
  25 #include <errno.h>
  26 #include <locale.h>
  27
  28 static gboolean
  29 is_valid_module_name (const gchar *basename)
  30 {
  31 #if !defined(G_OS_WIN32) && !defined(G_WITH_CYGWIN)
  32   return
  33     g_str_has_prefix (basename, "lib") &&
  34     g_str_has_suffix (basename, ".so");
  35 #else
  36   return g_str_has_suffix (basename, ".dll");
  37 #endif
  38 }
  39
  40 static void
  41 query_dir (const char *dirname)
  42 {
  43   GString *data;
  44   GDir *dir;
  45   const char *name;
  46   char *cachename;
  47   char **(* query)  (void);
  48   GError *error;
  49   int i;
  50
  51   if (!g_module_supported ())
  52     return;
  53
  54   error = NULL;
  55   dir = g_dir_open (dirname, 0, &error);
  56   if (!dir)
  57     {
  58       g_printerr ("Unable to open directory %s: %s\n", dirname, error->message);
  59       g_error_free (error);
  60       return;
  61     }
  62
  63   data = g_string_new ("");
  64
  65   while ((name = g_dir_read_name (dir)))
  66     {
  67       GModule *module;
  68       gchar     *path;
  69       char **extension_points;
  70
  71       if (!is_valid_module_name (name))
  72         continue;
  73
  74       path = g_build_filename (dirname, name, NULL);
  75       module = g_module_open (path, G_MODULE_BIND_LAZY | G_MODULE_BIND_LOCAL);
  76       g_free (path);
  77
  78       if (module)
  79         {
  80           g_module_symbol (module, "g_io_module_query", (gpointer) &query);
  81
  82           if (query)
  83             {
  84               extension_points = query ();
  85
  86               if (extension_points)
  87                 {
  88                   g_string_append_printf (data, "%s: ", name);
  89
  90                   for (i = 0; extension_points[i] != NULL; i++)
  91                     g_string_append_printf (data, "%s%s", i == 0 ? "" : ",", extension_points[i]);
  92
  93                   g_string_append (data, "\n");
  94                   g_strfreev (extension_points);
  95                 }
  96             }
  97
  98           g_module_close (module);
  99         }
 100     }
 101
 102   g_dir_close (dir);
 103
 104   cachename = g_build_filename (dirname, "giomodule.cache", NULL);
 105
 106   if (data->len > 0)
 107     {
 108       error = NULL;
 109
 110       if (!g_file_set_contents (cachename, data->str, data->len, &error))
 111         {
 112           g_printerr ("Unable to create %s: %s\n", cachename, error->message);
 113           g_error_free (error);
 114         }
 115     }
 116   else
 117     {
 118       if (g_unlink (cachename) != 0 && errno != ENOENT)
 119         g_printerr ("Unable to unlink %s: %s\n", cachename, g_strerror (errno));
 120     }
 121
 122   g_string_free (data, TRUE);
 123 }
 124
 125 int
 126 main (gint   argc,
 127       gchar *argv[])
 128 {
 129   int i;
 130
 131   if (argc == 1)
 132     {
 133       g_print ("Usage: gio-querymodules <directory1> [<directory2> ...]\n");
 134       g_print ("Will update giomodule.cache in the listed directories\n");
 135       return 1;
 136     }
 137
 138   setlocale (LC_ALL, "");
 139
 140   /* Be defensive and ensure we're linked to GObject */
 141   g_type_ensure (G_TYPE_OBJECT);
 142
 143   for (i = 1; i < argc; i++)
 144     query_dir (argv[i]);
 145
 146   return 0;
 147 }