gio/glib-compile-resources.c

   1 /*
   2  * Copyright © 2011 Red Hat, Inc
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the licence, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  *
  19  * Author: Alexander Larsson <alexl@redhat.com>
  20  */
  21
  22 #include "config.h"
  23
  24 #include <glib.h>
  25 #include <gstdio.h>
  26 #include <gi18n.h>
  27 #include <gioenums.h>
  28
  29 #include <string.h>
  30 #include <stdio.h>
  31 #include <locale.h>
  32 #include <errno.h>
  33 #ifdef G_OS_WIN32
  34 #include <io.h>
  35 #endif
  36 #ifdef HAVE_SYS_WAIT_H
  37 #include <sys/wait.h>
  38 #endif
  39
  40 #include <gio/gmemoryoutputstream.h>
  41 #include <gio/gzlibcompressor.h>
  42 #include <gio/gconverteroutputstream.h>
  43
  44 #ifdef HAVE_UNISTD_H
  45 #include <unistd.h>
  46 #endif
  47
  48 #include <glib.h>
  49 #include "gvdb/gvdb-builder.h"
  50
  51 #include "gconstructor_as_data.h"
  52
  53 typedef struct
  54 {
  55   char *filename;
  56   char *content;
  57   gsize content_size;
  58   gsize size;
  59   guint32 flags;
  60 } FileData;
  61
  62 typedef struct
  63 {
  64   GHashTable *table; /* resource path -> FileData */
  65
  66   gboolean collect_data;
  67
  68   /* per gresource */
  69   char *prefix;
  70
  71   /* per file */
  72   char *alias;
  73   gboolean compressed;
  74   char *preproc_options;
  75
  76   GString *string;  /* non-NULL when accepting text */
  77 } ParseState;
  78
  79 static gchar **sourcedirs = NULL;
  80 static gchar *xmllint = NULL;
  81 static gchar *gdk_pixbuf_pixdata = NULL;
  82
  83 static void
  84 file_data_free (FileData *data)
  85 {
  86   g_free (data->filename);
  87   g_free (data->content);
  88   g_free (data);
  89 }
  90
  91 static void
  92 start_element (GMarkupParseContext  *context,
  93                const gchar          *element_name,
  94                const gchar         **attribute_names,
  95                const gchar         **attribute_values,
  96                gpointer              user_data,
  97                GError              **error)
  98 {
  99   ParseState *state = user_data;
 100   const GSList *element_stack;
 101   const gchar *container;
 102
 103   element_stack = g_markup_parse_context_get_element_stack (context);
 104   container = element_stack->next ? element_stack->next->data : NULL;
 105
 106 #define COLLECT(first, ...) \
 107   g_markup_collect_attributes (element_name,                                 \
 108                                attribute_names, attribute_values, error,     \
 109                                first, __VA_ARGS__, G_MARKUP_COLLECT_INVALID)
 110 #define OPTIONAL   G_MARKUP_COLLECT_OPTIONAL
 111 #define STRDUP     G_MARKUP_COLLECT_STRDUP
 112 #define STRING     G_MARKUP_COLLECT_STRING
 113 #define BOOL       G_MARKUP_COLLECT_BOOLEAN
 114 #define NO_ATTRS()  COLLECT (G_MARKUP_COLLECT_INVALID, NULL)
 115
 116   if (container == NULL)
 117     {
 118       if (strcmp (element_name, "gresources") == 0)
 119         return;
 120     }
 121   else if (strcmp (container, "gresources") == 0)
 122     {
 123       if (strcmp (element_name, "gresource") == 0)
 124         {
 125           COLLECT (OPTIONAL | STRDUP,
 126                    "prefix", &state->prefix);
 127           return;
 128         }
 129     }
 130   else if (strcmp (container, "gresource") == 0)
 131     {
 132       if (strcmp (element_name, "file") == 0)
 133         {
 134           COLLECT (OPTIONAL | STRDUP, "alias", &state->alias,
 135                    OPTIONAL | BOOL, "compressed", &state->compressed,
 136                    OPTIONAL | STRDUP, "preprocess", &state->preproc_options);
 137           state->string = g_string_new ("");
 138           return;
 139         }
 140     }
 141
 142   if (container)
 143     g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_UNKNOWN_ELEMENT,
 144                  _("Element <%s> not allowed inside <%s>"),
 145                  element_name, container);
 146   else
 147     g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_UNKNOWN_ELEMENT,
 148                  _("Element <%s> not allowed at toplevel"), element_name);
 149
 150 }
 151
 152 static GvdbItem *
 153 get_parent (GHashTable *table,
 154             gchar      *key,
 155             gint        length)
 156 {
 157   GvdbItem *grandparent, *parent;
 158
 159   if (length == 1)
 160     return NULL;
 161
 162   while (key[--length - 1] != '/');
 163   key[length] = '\0';
 164
 165   parent = g_hash_table_lookup (table, key);
 166
 167   if (parent == NULL)
 168     {
 169       parent = gvdb_hash_table_insert (table, key);
 170
 171       grandparent = get_parent (table, key, length);
 172
 173       if (grandparent != NULL)
 174         gvdb_item_set_parent (parent, grandparent);
 175     }
 176
 177   return parent;
 178 }
 179
 180 static gchar *
 181 find_file (const gchar *filename)
 182 {
 183   guint i;
 184   gchar *real_file;
 185   gboolean exists;
 186
 187   if (g_path_is_absolute (filename))
 188     return g_strdup (filename);
 189
 190   /* search all the sourcedirs for the correct files in order */
 191   for (i = 0; sourcedirs[i] != NULL; i++)
 192     {
 193         real_file = g_build_filename (sourcedirs[i], filename, NULL);
 194         exists = g_file_test (real_file, G_FILE_TEST_EXISTS);
 195         if (exists)
 196           return real_file;
 197         g_free (real_file);
 198     }
 199     return NULL;
 200 }
 201
 202 static void
 203 end_element (GMarkupParseContext  *context,
 204              const gchar          *element_name,
 205              gpointer              user_data,
 206              GError              **error)
 207 {
 208   ParseState *state = user_data;
 209   GError *my_error = NULL;
 210
 211   if (strcmp (element_name, "gresource") == 0)
 212     {
 213       g_free (state->prefix);
 214       state->prefix = NULL;
 215     }
 216
 217   else if (strcmp (element_name, "file") == 0)
 218     {
 219       gchar *file, *real_file;
 220       gchar *key;
 221       FileData *data;
 222       char *tmp_file = NULL;
 223       char *tmp_file2 = NULL;
 224
 225       file = state->string->str;
 226       key = file;
 227       if (state->alias)
 228         key = state->alias;
 229
 230       if (state->prefix)
 231         key = g_build_path ("/", "/", state->prefix, key, NULL);
 232       else
 233         key = g_build_path ("/", "/", key, NULL);
 234
 235       if (g_hash_table_lookup (state->table, key) != NULL)
 236         {
 237           g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
 238                        _("File %s appears multiple times in the resource"),
 239                        key);
 240           return;
 241         }
 242
 243       data = g_new0 (FileData, 1);
 244
 245       if (sourcedirs != NULL)
 246         {
 247           real_file = find_file (file);
 248           if (real_file == NULL)
 249             {
 250                 g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 251                              _("Failed to locate '%s' in any source directory"), file);
 252                 return;
 253             }
 254         }
 255       else
 256         {
 257           gboolean exists;
 258           exists = g_file_test (file, G_FILE_TEST_EXISTS);
 259           if (!exists)
 260             {
 261               g_set_error (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 262                            _("Failed to locate '%s' in current directory"), file);
 263               return;
 264             }
 265           real_file = g_strdup (file);
 266         }
 267
 268       data->filename = g_strdup (real_file);
 269       if (!state->collect_data)
 270         goto done;
 271
 272       if (state->preproc_options)
 273         {
 274           gchar **options;
 275           guint i;
 276           gboolean xml_stripblanks = FALSE;
 277           gboolean to_pixdata = FALSE;
 278
 279           options = g_strsplit (state->preproc_options, ",", -1);
 280
 281           for (i = 0; options[i]; i++)
 282             {
 283               if (!strcmp (options[i], "xml-stripblanks"))
 284                 xml_stripblanks = TRUE;
 285               else if (!strcmp (options[i], "to-pixdata"))
 286                 to_pixdata = TRUE;
 287               else
 288                 {
 289                   g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
 290                                _("Unknown processing option \"%s\""), options[i]);
 291                   g_strfreev (options);
 292                   goto cleanup;
 293                 }
 294             }
 295           g_strfreev (options);
 296
 297           if (xml_stripblanks && xmllint != NULL)
 298             {
 299               gchar *argv[8];
 300               int status, fd, argc;
 301
 302               tmp_file = g_strdup ("resource-XXXXXXXX");
 303               if ((fd = g_mkstemp (tmp_file)) == -1)
 304                 {
 305                   int errsv = errno;
 306
 307                   g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv),
 308                                _("Failed to create temp file: %s"),
 309                               g_strerror (errsv));
 310                   g_free (tmp_file);
 311                   tmp_file = NULL;
 312                   goto cleanup;
 313                 }
 314               close (fd);
 315
 316               argc = 0;
 317               argv[argc++] = (gchar *) xmllint;
 318               argv[argc++] = "--nonet";
 319               argv[argc++] = "--noblanks";
 320               argv[argc++] = "--output";
 321               argv[argc++] = tmp_file;
 322               argv[argc++] = real_file;
 323               argv[argc++] = NULL;
 324               g_assert (argc <= G_N_ELEMENTS (argv));
 325
 326               if (!g_spawn_sync (NULL /* cwd */, argv, NULL /* envv */,
 327                                  G_SPAWN_STDOUT_TO_DEV_NULL |
 328                                  G_SPAWN_STDERR_TO_DEV_NULL,
 329                                  NULL, NULL, NULL, NULL, &status, &my_error))
 330                 {
 331                   g_propagate_error (error, my_error);
 332                   goto cleanup;
 333                 }
 334 #ifdef HAVE_SYS_WAIT_H
 335               if (!WIFEXITED (status) || WEXITSTATUS (status) != 0)
 336                 {
 337                   g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 338                                       _("Error processing input file with xmllint"));
 339                   goto cleanup;
 340                 }
 341 #endif
 342
 343               g_free (real_file);
 344               real_file = g_strdup (tmp_file);
 345             }
 346
 347           if (to_pixdata)
 348             {
 349               gchar *argv[4];
 350               int status, fd, argc;
 351
 352               if (gdk_pixbuf_pixdata == NULL)
 353                 {
 354                   g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 355                                        "to-pixbuf preprocessing requested but GDK_PIXBUF_PIXDATA "
 356                                        "not set and gdk-pixbuf-pixdata not found in path");
 357                   goto cleanup;
 358                 }
 359
 360               tmp_file2 = g_strdup ("resource-XXXXXXXX");
 361               if ((fd = g_mkstemp (tmp_file2)) == -1)
 362                 {
 363                   int errsv = errno;
 364
 365                   g_set_error (error, G_IO_ERROR, g_io_error_from_errno (errsv),
 366                                _("Failed to create temp file: %s"),
 367                                g_strerror (errsv));
 368                   g_free (tmp_file2);
 369                   tmp_file2 = NULL;
 370                   goto cleanup;
 371                 }
 372               close (fd);
 373
 374               argc = 0;
 375               argv[argc++] = (gchar *) gdk_pixbuf_pixdata;
 376               argv[argc++] = real_file;
 377               argv[argc++] = tmp_file2;
 378               argv[argc++] = NULL;
 379               g_assert (argc <= G_N_ELEMENTS (argv));
 380
 381               if (!g_spawn_sync (NULL /* cwd */, argv, NULL /* envv */,
 382                                  G_SPAWN_STDOUT_TO_DEV_NULL |
 383                                  G_SPAWN_STDERR_TO_DEV_NULL,
 384                                  NULL, NULL, NULL, NULL, &status, &my_error))
 385                 {
 386                   g_propagate_error (error, my_error);
 387                   goto cleanup;
 388                 }
 389 #ifdef HAVE_SYS_WAIT_H
 390               if (!WIFEXITED (status) || WEXITSTATUS (status) != 0)
 391                 {
 392                   g_set_error_literal (error, G_IO_ERROR, G_IO_ERROR_FAILED,
 393                                        _("Error processing input file with to-pixdata"));
 394                   goto cleanup;
 395                 }
 396 #endif
 397
 398               g_free (real_file);
 399               real_file = g_strdup (tmp_file2);
 400             }
 401         }
 402
 403       if (!g_file_get_contents (real_file, &data->content, &data->size, &my_error))
 404         {
 405           g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
 406                        _("Error reading file %s: %s"),
 407                        real_file, my_error->message);
 408           g_clear_error (&my_error);
 409           goto cleanup;
 410         }
 411       /* Include zero termination in content_size for uncompressed files (but not in size) */
 412       data->content_size = data->size + 1;
 413
 414       if (state->compressed)
 415         {
 416           GOutputStream *out = g_memory_output_stream_new (NULL, 0, g_realloc, g_free);
 417           GZlibCompressor *compressor =
 418             g_zlib_compressor_new (G_ZLIB_COMPRESSOR_FORMAT_ZLIB, 9);
 419           GOutputStream *out2 = g_converter_output_stream_new (out, G_CONVERTER (compressor));
 420
 421           if (!g_output_stream_write_all (out2, data->content, data->size,
 422                                           NULL, NULL, NULL) ||
 423               !g_output_stream_close (out2, NULL, NULL))
 424             {
 425               g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
 426                            _("Error compressing file %s"),
 427                            real_file);
 428               goto cleanup;
 429             }
 430
 431           g_free (data->content);
 432           data->content_size = g_memory_output_stream_get_size (G_MEMORY_OUTPUT_STREAM (out));
 433           data->content = g_memory_output_stream_steal_data (G_MEMORY_OUTPUT_STREAM (out));
 434
 435           g_object_unref (compressor);
 436           g_object_unref (out);
 437           g_object_unref (out2);
 438
 439           data->flags |= G_RESOURCE_FLAGS_COMPRESSED;
 440         }
 441
 442     done:
 443
 444       g_hash_table_insert (state->table, key, data);
 445
 446     cleanup:
 447       /* Cleanup */
 448
 449       g_free (state->alias);
 450       state->alias = NULL;
 451       g_string_free (state->string, TRUE);
 452       state->string = NULL;
 453       g_free (state->preproc_options);
 454       state->preproc_options = NULL;
 455
 456       g_free (real_file);
 457
 458       if (tmp_file)
 459         {
 460           unlink (tmp_file);
 461           g_free (tmp_file);
 462         }
 463
 464       if (tmp_file2)
 465         {
 466           unlink (tmp_file2);
 467           g_free (tmp_file2);
 468         }
 469     }
 470 }
 471
 472 static void
 473 text (GMarkupParseContext  *context,
 474       const gchar          *text,
 475       gsize                 text_len,
 476       gpointer              user_data,
 477       GError              **error)
 478 {
 479   ParseState *state = user_data;
 480   gsize i;
 481
 482   for (i = 0; i < text_len; i++)
 483     if (!g_ascii_isspace (text[i]))
 484       {
 485         if (state->string)
 486           g_string_append_len (state->string, text, text_len);
 487
 488         else
 489           g_set_error (error, G_MARKUP_ERROR, G_MARKUP_ERROR_INVALID_CONTENT,
 490                        _("text may not appear inside <%s>"),
 491                        g_markup_parse_context_get_element (context));
 492
 493         break;
 494       }
 495 }
 496
 497 static GHashTable *
 498 parse_resource_file (const gchar *filename,
 499                      gboolean collect_data)
 500 {
 501   GMarkupParser parser = { start_element, end_element, text };
 502   ParseState state = { 0, };
 503   GMarkupParseContext *context;
 504   GError *error = NULL;
 505   gchar *contents;
 506   GHashTable *table = NULL;
 507   gsize size;
 508
 509   if (!g_file_get_contents (filename, &contents, &size, &error))
 510     {
 511       g_printerr ("%s\n", error->message);
 512       g_clear_error (&error);
 513       return NULL;
 514     }
 515
 516   state.table = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, (GDestroyNotify)file_data_free);
 517   state.collect_data = collect_data;
 518
 519   context = g_markup_parse_context_new (&parser,
 520                                         G_MARKUP_TREAT_CDATA_AS_TEXT |
 521                                         G_MARKUP_PREFIX_ERROR_POSITION,
 522                                         &state, NULL);
 523
 524   if (!g_markup_parse_context_parse (context, contents, size, &error) ||
 525       !g_markup_parse_context_end_parse (context, &error))
 526     {
 527       g_printerr ("%s: %s.\n", filename, error->message);
 528       g_clear_error (&error);
 529     }
 530   else if (collect_data)
 531     {
 532       GHashTableIter iter;
 533       const char *key;
 534       char *mykey;
 535       gsize key_len;
 536       FileData *data;
 537       GVariant *v_data;
 538       GVariantBuilder builder;
 539       GvdbItem *item;
 540
 541       table = gvdb_hash_table_new (NULL, NULL);
 542
 543       g_hash_table_iter_init (&iter, state.table);
 544       while (g_hash_table_iter_next (&iter, (gpointer *)&key, (gpointer *)&data))
 545         {
 546           key_len = strlen (key);
 547           mykey = g_strdup (key);
 548
 549           item = gvdb_hash_table_insert (table, key);
 550           gvdb_item_set_parent (item,
 551                                 get_parent (table, mykey, key_len));
 552
 553           g_free (mykey);
 554
 555           g_variant_builder_init (&builder, G_VARIANT_TYPE ("(uuay)"));
 556
 557           g_variant_builder_add (&builder, "u", data->size); /* Size */
 558           g_variant_builder_add (&builder, "u", data->flags); /* Flags */
 559
 560           v_data = g_variant_new_from_data (G_VARIANT_TYPE("ay"),
 561                                             data->content, data->content_size, TRUE,
 562                                             g_free, data->content);
 563           g_variant_builder_add_value (&builder, v_data);
 564           data->content = NULL; /* Take ownership */
 565
 566           gvdb_item_set_value (item,
 567                                g_variant_builder_end (&builder));
 568         }
 569     }
 570   else
 571     {
 572       table = g_hash_table_ref (state.table);
 573     }
 574
 575   g_hash_table_unref (state.table);
 576   g_markup_parse_context_free (context);
 577   g_free (contents);
 578
 579   return table;
 580 }
 581
 582 static gboolean
 583 write_to_file (GHashTable   *table,
 584                const gchar  *filename,
 585                GError      **error)
 586 {
 587   gboolean success;
 588
 589   success = gvdb_table_write_contents (table, filename,
 590                                        G_BYTE_ORDER != G_LITTLE_ENDIAN,
 591                                        error);
 592
 593   return success;
 594 }
 595
 596 int
 597 main (int argc, char **argv)
 598 {
 599   GError *error;
 600   GHashTable *table;
 601   gchar *srcfile;
 602   gchar *target = NULL;
 603   gchar *binary_target = NULL;
 604   gboolean generate_automatic = FALSE;
 605   gboolean generate_source = FALSE;
 606   gboolean generate_header = FALSE;
 607   gboolean manual_register = FALSE;
 608   gboolean generate_dependencies = FALSE;
 609   char *c_name = NULL;
 610   char *c_name_no_underscores;
 611   GOptionContext *context;
 612   GOptionEntry entries[] = {
 613     { "target", 0, 0, G_OPTION_ARG_FILENAME, &target, N_("name of the output file"), N_("FILE") },
 614     { "sourcedir", 0, 0, G_OPTION_ARG_FILENAME_ARRAY, &sourcedirs, N_("The directories where files are to be read from (default to current directory)"), N_("DIRECTORY") },
 615     { "generate", 0, 0, G_OPTION_ARG_NONE, &generate_automatic, N_("Generate output in the format selected for by the target filename extension"), NULL },
 616     { "generate-header", 0, 0, G_OPTION_ARG_NONE, &generate_header, N_("Generate source header"), NULL },
 617     { "generate-source", 0, 0, G_OPTION_ARG_NONE, &generate_source, N_("Generate sourcecode used to link in the resource file into your code"), NULL },
 618     { "generate-dependencies", 0, 0, G_OPTION_ARG_NONE, &generate_dependencies, N_("Generate dependency list"), NULL },
 619     { "manual-register", 0, 0, G_OPTION_ARG_NONE, &manual_register, N_("Don't automatically create and register resource"), NULL },
 620     { "c-name", 0, 0, G_OPTION_ARG_STRING, &c_name, N_("C identifier name used for the generated source code"), NULL },
 621     { NULL }
 622   };
 623
 624 #ifdef G_OS_WIN32
 625   extern gchar *_glib_get_locale_dir (void);
 626   gchar *tmp;
 627 #endif
 628
 629   setlocale (LC_ALL, "");
 630   textdomain (GETTEXT_PACKAGE);
 631
 632 #ifdef G_OS_WIN32
 633   tmp = _glib_get_locale_dir ();
 634   bindtextdomain (GETTEXT_PACKAGE, tmp);
 635   g_free (tmp);
 636 #else
 637   bindtextdomain (GETTEXT_PACKAGE, GLIB_LOCALE_DIR);
 638 #endif
 639
 640 #ifdef HAVE_BIND_TEXTDOMAIN_CODESET
 641   bind_textdomain_codeset (GETTEXT_PACKAGE, "UTF-8");
 642 #endif
 643
 644   g_type_init ();
 645
 646   context = g_option_context_new (N_("FILE"));
 647   g_option_context_set_translation_domain (context, GETTEXT_PACKAGE);
 648   g_option_context_set_summary (context,
 649     N_("Compile a resource specification into a resource file.\n"
 650        "Resource specification files have the extension .gresource.xml,\n"
 651        "and the resource file have the extension called .gresource."));
 652   g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
 653
 654   error = NULL;
 655   if (!g_option_context_parse (context, &argc, &argv, &error))
 656     {
 657       g_printerr ("%s\n", error->message);
 658       return 1;
 659     }
 660
 661   g_option_context_free (context);
 662
 663   if (argc != 2)
 664     {
 665       g_printerr (_("You should give exactly one file name\n"));
 666       return 1;
 667     }
 668
 669   srcfile = argv[1];
 670
 671   xmllint = g_strdup (g_getenv ("XMLLINT"));
 672   if (xmllint == NULL)
 673     xmllint = g_find_program_in_path ("xmllint");
 674   if (xmllint == NULL)
 675     g_printerr ("XMLLINT not set and xmllint not found in path; skipping xml preprocessing.\n");
 676
 677   gdk_pixbuf_pixdata = g_strdup (g_getenv ("GDK_PIXBUF_PIXDATA"));
 678   if (gdk_pixbuf_pixdata == NULL)
 679     gdk_pixbuf_pixdata = g_find_program_in_path ("gdk-pixbuf-pixdata");
 680
 681   if (target == NULL)
 682     {
 683       char *dirname = g_path_get_dirname (srcfile);
 684       char *base = g_path_get_basename (srcfile);
 685       char *target_basename;
 686       if (g_str_has_suffix (base, ".xml"))
 687         base[strlen(base) - strlen (".xml")] = 0;
 688
 689       if (generate_source)
 690         {
 691           if (g_str_has_suffix (base, ".gresource"))
 692             base[strlen(base) - strlen (".gresource")] = 0;
 693           target_basename = g_strconcat (base, ".c", NULL);
 694         }
 695       else
 696         {
 697           if (g_str_has_suffix (base, ".gresource"))
 698             target_basename = g_strdup (base);
 699           else
 700             target_basename = g_strconcat (base, ".gresource", NULL);
 701         }
 702
 703       target = g_build_filename (dirname, target_basename, NULL);
 704       g_free (target_basename);
 705       g_free (dirname);
 706       g_free (base);
 707     }
 708   else if (generate_automatic)
 709     {
 710       if (g_str_has_suffix (target, ".c"))
 711         generate_source = TRUE;
 712       else if (g_str_has_suffix (target, ".h"))
 713         generate_header = TRUE;
 714       else if (g_str_has_suffix (target, ".gresource"))
 715         ;
 716     }
 717
 718   if ((table = parse_resource_file (srcfile, !generate_dependencies)) == NULL)
 719     {
 720       g_free (target);
 721       return 1;
 722     }
 723
 724   if (generate_dependencies)
 725     {
 726       GHashTableIter iter;
 727       gpointer key, data;
 728       FileData *file_data;
 729
 730       g_hash_table_iter_init (&iter, table);
 731       while (g_hash_table_iter_next (&iter, &key, &data))
 732         {
 733           file_data = data;
 734           g_print ("%s\n",file_data->filename);
 735         }
 736     }
 737   else if (generate_source || generate_header)
 738     {
 739       if (generate_source)
 740         {
 741           int fd = g_file_open_tmp (NULL, &binary_target, NULL);
 742           if (fd == -1)
 743             {
 744               g_printerr ("Can't open temp file\n");
 745               return 1;
 746             }
 747           close (fd);
 748         }
 749
 750       if (c_name == NULL)
 751         {
 752           char *base = g_path_get_basename (srcfile);
 753           GString *s;
 754           char *dot;
 755           int i;
 756
 757           /* Remove extensions */
 758           dot = strchr (base, '.');
 759           if (dot)
 760             *dot = 0;
 761
 762           s = g_string_new ("");
 763
 764           for (i = 0; base[i] != 0; i++)
 765             {
 766               const char *first = G_CSET_A_2_Z G_CSET_a_2_z "_";
 767               const char *rest = G_CSET_A_2_Z G_CSET_a_2_z G_CSET_DIGITS "_";
 768               if (strchr ((i == 0) ? first : rest, base[i]) != NULL)
 769                 g_string_append_c (s, base[i]);
 770               else if (base[i] == '-')
 771                 g_string_append_c (s, '_');
 772
 773             }
 774
 775           c_name = g_string_free (s, FALSE);
 776         }
 777     }
 778   else
 779     binary_target = g_strdup (target);
 780
 781   c_name_no_underscores = c_name;
 782   while (c_name_no_underscores && *c_name_no_underscores == '_')
 783     c_name_no_underscores++;
 784
 785   if (binary_target != NULL &&
 786       !write_to_file (table, binary_target, &error))
 787     {
 788       g_printerr ("%s\n", error->message);
 789       g_free (target);
 790       return 1;
 791     }
 792
 793   if (generate_header)
 794     {
 795       FILE *file;
 796
 797       file = fopen (target, "w");
 798       if (file == NULL)
 799         {
 800           g_printerr ("can't write to file %s", target);
 801           return 1;
 802         }
 803
 804       fprintf (file,
 805                "#ifndef __RESOURCE_%s_H__\n"
 806                "#define __RESOURCE_%s_H__\n"
 807                "\n"
 808                "#include <gio/gio.h>\n"
 809                "\n"
 810                "extern GResource *%s_get_resource (void);\n",
 811                c_name, c_name, c_name);
 812
 813       if (manual_register)
 814         fprintf (file,
 815                  "\n"
 816                  "extern void %s_register_resource (void);\n"
 817                  "extern void %s_unregister_resource (void);\n"
 818                  "\n",
 819                  c_name, c_name);
 820
 821       fprintf (file,
 822                "#endif\n");
 823
 824       fclose (file);
 825     }
 826   else if (generate_source)
 827     {
 828       FILE *file;
 829       guint8 *data;
 830       gsize data_size;
 831       gsize i;
 832
 833       if (!g_file_get_contents (binary_target, (char **)&data,
 834                                 &data_size, NULL))
 835         {
 836           g_printerr ("can't read back temporary file");
 837           return 1;
 838         }
 839       g_unlink (binary_target);
 840
 841       file = fopen (target, "w");
 842       if (file == NULL)
 843         {
 844           g_printerr ("can't write to file %s", target);
 845           return 1;
 846         }
 847
 848       fprintf (file,
 849                "#include <gio/gio.h>\n"
 850                "\n"
 851                "#if defined (__ELF__) && ( __GNUC__ > 2 || (__GNUC__ == 2 && __GNUC_MINOR__ >= 6))\n"
 852                "# define SECTION __attribute__ ((section (\".gresource.%s\"), aligned (8)))\n"
 853                "#else\n"
 854                "# define SECTION\n"
 855                "#endif\n"
 856                "\n"
 857                "static const SECTION union { const guint8 data[%"G_GSIZE_FORMAT"]; const double alignment; void * const ptr;}  %s_resource_data = { {\n",
 858                c_name_no_underscores, data_size, c_name);
 859
 860       for (i = 0; i < data_size; i++) {
 861         if (i % 8 == 0)
 862           fprintf (file, "  ");
 863         fprintf (file, "0x%2.2x", (int)data[i]);
 864         if (i != data_size - 1)
 865           fprintf (file, ", ");
 866         if ((i % 8 == 7) || (i == data_size - 1))
 867           fprintf (file, "\n");
 868       }
 869
 870       fprintf (file, "} };\n");
 871
 872       fprintf (file,
 873                "\n"
 874                "static GStaticResource static_resource = { %s_resource_data.data, sizeof (%s_resource_data.data) };\n"
 875                "extern GResource *%s_get_resource (void);\n"
 876                "GResource *%s_get_resource (void)\n"
 877                "{\n"
 878                "  return g_static_resource_get_resource (&static_resource);\n"
 879                "}\n",
 880                c_name, c_name, c_name, c_name);
 881
 882
 883       if (manual_register)
 884         {
 885           fprintf (file,
 886                    "\n"
 887                    "extern void %s_unregister_resource (void);\n"
 888                    "void %s_unregister_resource (void)\n"
 889                    "{\n"
 890                    "  g_static_resource_fini (&static_resource);\n"
 891                    "}\n"
 892                    "\n"
 893                    "extern void %s_register_resource (void);\n"
 894                    "void %s_register_resource (void)\n"
 895                    "{\n"
 896                    "  g_static_resource_init (&static_resource);\n"
 897                    "}\n",
 898                    c_name, c_name, c_name, c_name);
 899         }
 900       else
 901         {
 902           fprintf (file, "%s", gconstructor_code);
 903           fprintf (file,
 904                    "\n"
 905                    "#ifdef G_HAS_CONSTRUCTORS\n"
 906                    "\n"
 907                    "#ifdef G_DEFINE_CONSTRUCTOR_NEEDS_PRAGMA\n"
 908                    "#pragma G_DEFINE_CONSTRUCTOR_PRAGMA_ARGS(resource_constructor)\n"
 909                    "#endif\n"
 910                    "G_DEFINE_CONSTRUCTOR(resource_constructor)\n"
 911                    "#ifdef G_DEFINE_DESTRUCTOR_NEEDS_PRAGMA\n"
 912                    "#pragma G_DEFINE_DESTRUCTOR_PRAGMA_ARGS(resource_destructor)\n"
 913                    "#endif\n"
 914                    "G_DEFINE_DESTRUCTOR(resource_destructor)\n"
 915                    "\n"
 916                    "#else\n"
 917                    "#warning \"Constructor not supported on this compiler, linking in resources will not work\"\n"
 918                    "#endif\n"
 919                    "\n"
 920                    "static void resource_constructor (void)\n"
 921                    "{\n"
 922                    "  g_static_resource_init (&static_resource);\n"
 923                    "}\n"
 924                    "\n"
 925                    "static void resource_destructor (void)\n"
 926                    "{\n"
 927                    "  g_static_resource_fini (&static_resource);\n"
 928                    "}\n");
 929         }
 930
 931       fclose (file);
 932
 933       g_free (data);
 934     }
 935
 936   g_free (binary_target);
 937   g_free (target);
 938   g_hash_table_destroy (table);
 939   g_free (xmllint);
 940
 941   return 0;
 942 }