src/xdgmimecache.c

   1 /* -*- mode: C; c-file-style: "gnu" -*- */
   2 /* xdgmimealias.c: Private file.  mmappable caches for mime data
   3  *
   4  * More info can be found at http://www.freedesktop.org/standards/
   5  *
   6  * Copyright (C) 2005  Matthias Clasen <mclasen@redhat.com>
   7  *
   8  * Licensed under the Academic Free License version 2.0
   9  * Or under the following terms:
  10  *
  11  * This library is free software; you can redistribute it and/or
  12  * modify it under the terms of the GNU Lesser General Public
  13  * License as published by the Free Software Foundation; either
  14  * version 2 of the License, or (at your option) any later version.
  15  *
  16  * This library is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19  * Lesser General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU Lesser General Public
  22  * License along with this library; if not, write to the
  23  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  24  * Boston, MA 02111-1307, USA.
  25  */
  26
  27 #ifdef HAVE_CONFIG_H
  28 #include <config.h>
  29 #endif
  30
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34
  35 #include <fcntl.h>
  36 #include <unistd.h>
  37 #include <fnmatch.h>
  38 #include <assert.h>
  39
  40 #include <netinet/in.h> /* for ntohl/ntohs */
  41
  42 #ifdef HAVE_MMAP
  43 #include <sys/mman.h>
  44 #else
  45 #warning Building xdgmime without MMAP support. Binary "mime.info" cache files will not be used.
  46 #endif
  47
  48 #include <sys/stat.h>
  49 #include <sys/types.h>
  50
  51 #include "xdgmimecache.h"
  52 #include "xdgmimeint.h"
  53
  54 #ifndef MAX
  55 #define MAX(a,b) ((a) > (b) ? (a) : (b))
  56 #endif
  57
  58 #ifndef FALSE
  59 #define FALSE   (0)
  60 #endif
  61
  62 #ifndef TRUE
  63 #define TRUE    (!FALSE)
  64 #endif
  65
  66 #ifndef _O_BINARY
  67 #define _O_BINARY 0
  68 #endif
  69
  70 #ifndef MAP_FAILED
  71 #define MAP_FAILED ((void *) -1)
  72 #endif
  73
  74 #define MAJOR_VERSION 1
  75 #define MINOR_VERSION_MIN 1
  76 #define MINOR_VERSION_MAX 2
  77
  78 struct _XdgMimeCache
  79 {
  80   int ref_count;
  81   int minor;
  82
  83   size_t  size;
  84   char   *buffer;
  85 };
  86
  87 #define GET_UINT16(cache,offset) (ntohs(*(xdg_uint16_t*)((cache) + (offset))))
  88 #define GET_UINT32(cache,offset) (ntohl(*(xdg_uint32_t*)((cache) + (offset))))
  89
  90 XdgMimeCache *
  91 _xdg_mime_cache_ref (XdgMimeCache *cache)
  92 {
  93   cache->ref_count++;
  94   return cache;
  95 }
  96
  97 void
  98 _xdg_mime_cache_unref (XdgMimeCache *cache)
  99 {
 100   cache->ref_count--;
 101
 102   if (cache->ref_count == 0)
 103     {
 104 #ifdef HAVE_MMAP
 105       munmap (cache->buffer, cache->size);
 106 #endif
 107       free (cache);
 108     }
 109 }
 110
 111 XdgMimeCache *
 112 _xdg_mime_cache_new_from_file (const char *file_name)
 113 {
 114   XdgMimeCache *cache = NULL;
 115
 116 #ifdef HAVE_MMAP
 117   int fd = -1;
 118   struct stat st;
 119   char *buffer = NULL;
 120   int minor;
 121
 122   /* Open the file and map it into memory */
 123   fd = open (file_name, O_RDONLY|_O_BINARY, 0);
 124
 125   if (fd < 0)
 126     return NULL;
 127
 128   if (fstat (fd, &st) < 0 || st.st_size < 4)
 129     goto done;
 130
 131   buffer = (char *) mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
 132
 133   if (buffer == MAP_FAILED)
 134     goto done;
 135
 136   minor = GET_UINT16 (buffer, 2);
 137   /* Verify version */
 138   if (GET_UINT16 (buffer, 0) != MAJOR_VERSION ||
 139       (minor < MINOR_VERSION_MIN ||
 140        minor > MINOR_VERSION_MAX))
 141     {
 142       munmap (buffer, st.st_size);
 143
 144       goto done;
 145     }
 146
 147   cache = (XdgMimeCache *) malloc (sizeof (XdgMimeCache));
 148   cache->minor = minor;
 149   cache->ref_count = 1;
 150   cache->buffer = buffer;
 151   cache->size = st.st_size;
 152
 153  done:
 154   if (fd != -1)
 155     close (fd);
 156
 157 #endif  /* HAVE_MMAP */
 158
 159   return cache;
 160 }
 161
 162 static int
 163 cache_magic_matchlet_compare_to_data (XdgMimeCache *cache,
 164                                       xdg_uint32_t  offset,
 165                                       const void   *data,
 166                                       size_t        len)
 167 {
 168   xdg_uint32_t range_start = GET_UINT32 (cache->buffer, offset);
 169   xdg_uint32_t range_length = GET_UINT32 (cache->buffer, offset + 4);
 170   xdg_uint32_t data_length = GET_UINT32 (cache->buffer, offset + 12);
 171   xdg_uint32_t data_offset = GET_UINT32 (cache->buffer, offset + 16);
 172   xdg_uint32_t mask_offset = GET_UINT32 (cache->buffer, offset + 20);
 173
 174   int i, j;
 175
 176   for (i = range_start; i <= range_start + range_length; i++)
 177     {
 178       int valid_matchlet = TRUE;
 179
 180       if (i + data_length > len)
 181         return FALSE;
 182
 183       if (mask_offset)
 184         {
 185           for (j = 0; j < data_length; j++)
 186             {
 187               if ((((unsigned char *)cache->buffer)[data_offset + j] & ((unsigned char *)cache->buffer)[mask_offset + j]) !=
 188                   ((((unsigned char *) data)[j + i]) & ((unsigned char *)cache->buffer)[mask_offset + j]))
 189                 {
 190                   valid_matchlet = FALSE;
 191                   break;
 192                 }
 193             }
 194         }
 195       else
 196         {
 197           for (j = 0; j < data_length; j++)
 198             {
 199               if (((unsigned char *)cache->buffer)[data_offset + j] != ((unsigned char *) data)[j + i])
 200                 {
 201                   valid_matchlet = FALSE;
 202                   break;
 203                 }
 204             }
 205         }
 206
 207       if (valid_matchlet)
 208         return TRUE;
 209     }
 210
 211   return FALSE;
 212 }
 213
 214 static int
 215 cache_magic_matchlet_compare (XdgMimeCache *cache,
 216                               xdg_uint32_t  offset,
 217                               const void   *data,
 218                               size_t        len)
 219 {
 220   xdg_uint32_t n_children = GET_UINT32 (cache->buffer, offset + 24);
 221   xdg_uint32_t child_offset = GET_UINT32 (cache->buffer, offset + 28);
 222
 223   int i;
 224
 225   if (cache_magic_matchlet_compare_to_data (cache, offset, data, len))
 226     {
 227       if (n_children == 0)
 228         return TRUE;
 229
 230       for (i = 0; i < n_children; i++)
 231         {
 232           if (cache_magic_matchlet_compare (cache, child_offset + 32 * i,
 233                                             data, len))
 234             return TRUE;
 235         }
 236     }
 237
 238   return FALSE;
 239 }
 240
 241 static const char *
 242 cache_magic_compare_to_data (XdgMimeCache *cache,
 243                              xdg_uint32_t  offset,
 244                              const void   *data,
 245                              size_t        len,
 246                              int          *prio)
 247 {
 248   xdg_uint32_t priority = GET_UINT32 (cache->buffer, offset);
 249   xdg_uint32_t mimetype_offset = GET_UINT32 (cache->buffer, offset + 4);
 250   xdg_uint32_t n_matchlets = GET_UINT32 (cache->buffer, offset + 8);
 251   xdg_uint32_t matchlet_offset = GET_UINT32 (cache->buffer, offset + 12);
 252
 253   int i;
 254
 255   for (i = 0; i < n_matchlets; i++)
 256     {
 257       if (cache_magic_matchlet_compare (cache, matchlet_offset + i * 32,
 258                                         data, len))
 259         {
 260           *prio = priority;
 261
 262           return cache->buffer + mimetype_offset;
 263         }
 264     }
 265
 266   return NULL;
 267 }
 268
 269 static const char *
 270 cache_magic_lookup_data (XdgMimeCache *cache,
 271                          const void   *data,
 272                          size_t        len,
 273                          int          *prio,
 274                          const char   *mime_types[],
 275                          int           n_mime_types)
 276 {
 277   xdg_uint32_t list_offset;
 278   xdg_uint32_t n_entries;
 279   xdg_uint32_t offset;
 280
 281   int j, n;
 282
 283   *prio = 0;
 284
 285   list_offset = GET_UINT32 (cache->buffer, 24);
 286   n_entries = GET_UINT32 (cache->buffer, list_offset);
 287   offset = GET_UINT32 (cache->buffer, list_offset + 8);
 288
 289   for (j = 0; j < n_entries; j++)
 290     {
 291       const char *match;
 292
 293       match = cache_magic_compare_to_data (cache, offset + 16 * j,
 294                                            data, len, prio);
 295       if (match)
 296         return match;
 297       else
 298         {
 299           xdg_uint32_t mimetype_offset;
 300           const char *non_match;
 301
 302           mimetype_offset = GET_UINT32 (cache->buffer, offset + 16 * j + 4);
 303           non_match = cache->buffer + mimetype_offset;
 304
 305           for (n = 0; n < n_mime_types; n++)
 306             {
 307               if (mime_types[n] &&
 308                   _xdg_mime_mime_type_equal (mime_types[n], non_match))
 309                 mime_types[n] = NULL;
 310             }
 311         }
 312     }
 313
 314   return NULL;
 315 }
 316
 317 static const char *
 318 cache_alias_lookup (const char *alias)
 319 {
 320   const char *ptr;
 321   int i, min, max, mid, cmp;
 322
 323   for (i = 0; _caches[i]; i++)
 324     {
 325       XdgMimeCache *cache = _caches[i];
 326       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 4);
 327       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 328       xdg_uint32_t offset;
 329
 330       min = 0;
 331       max = n_entries - 1;
 332       while (max >= min)
 333         {
 334           mid = (min + max) / 2;
 335
 336           offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid);
 337           ptr = cache->buffer + offset;
 338           cmp = strcmp (ptr, alias);
 339
 340           if (cmp < 0)
 341             min = mid + 1;
 342           else if (cmp > 0)
 343             max = mid - 1;
 344           else
 345             {
 346               offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid + 4);
 347               return cache->buffer + offset;
 348             }
 349         }
 350     }
 351
 352   return NULL;
 353 }
 354
 355 typedef struct {
 356   const char *mime;
 357   int weight;
 358 } MimeWeight;
 359
 360 static int
 361 cache_glob_lookup_literal (const char *file_name,
 362                            const char *mime_types[],
 363                            int         n_mime_types)
 364 {
 365   const char *ptr;
 366   int i, min, max, mid, cmp;
 367
 368   for (i = 0; _caches[i]; i++)
 369     {
 370       XdgMimeCache *cache = _caches[i];
 371       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 12);
 372       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 373       xdg_uint32_t offset;
 374
 375       min = 0;
 376       max = n_entries - 1;
 377       while (max >= min)
 378         {
 379           mid = (min + max) / 2;
 380
 381           offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * mid);
 382           ptr = cache->buffer + offset;
 383           cmp = strcmp (ptr, file_name);
 384
 385           if (cmp < 0)
 386             min = mid + 1;
 387           else if (cmp > 0)
 388             max = mid - 1;
 389           else
 390             {
 391               offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * mid + 4);
 392               mime_types[0] = (const char *)(cache->buffer + offset);
 393
 394               return 1;
 395             }
 396         }
 397     }
 398
 399   return 0;
 400 }
 401
 402 static int
 403 cache_glob_lookup_fnmatch (const char *file_name,
 404                            MimeWeight  mime_types[],
 405                            int         n_mime_types)
 406 {
 407   const char *mime_type;
 408   const char *ptr;
 409
 410   int i, j, n;
 411
 412   n = 0;
 413   for (i = 0; _caches[i]; i++)
 414     {
 415       XdgMimeCache *cache = _caches[i];
 416
 417       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 20);
 418       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 419
 420       for (j = 0; j < n_entries && n < n_mime_types; j++)
 421         {
 422           xdg_uint32_t offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j);
 423           xdg_uint32_t mimetype_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j + 4);
 424           int weight = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j + 8);
 425           ptr = cache->buffer + offset;
 426           mime_type = cache->buffer + mimetype_offset;
 427
 428           /* FIXME: Not UTF-8 safe */
 429           if (fnmatch (ptr, file_name, 0) == 0)
 430             {
 431               mime_types[n].mime = mime_type;
 432               mime_types[n].weight = weight;
 433               n++;
 434             }
 435         }
 436
 437       if (n > 0)
 438         return n;
 439     }
 440
 441   return 0;
 442 }
 443
 444 static int
 445 cache_glob_node_lookup_suffix (XdgMimeCache  *cache,
 446                                xdg_uint32_t   n_entries,
 447                                xdg_uint32_t   offset,
 448                                xdg_unichar_t *file_name,
 449                                int            len,
 450                                int            ignore_case,
 451                                MimeWeight     mime_types[],
 452                                int            n_mime_types)
 453 {
 454   xdg_unichar_t character;
 455   xdg_unichar_t match_char;
 456   xdg_uint32_t mimetype_offset;
 457   xdg_uint32_t n_children;
 458   xdg_uint32_t child_offset;
 459   int weight;
 460
 461   int min, max, mid, n, i;
 462
 463   character = file_name[len - 1];
 464   if (ignore_case)
 465     character = _xdg_ucs4_to_lower (character);
 466
 467   assert (character != 0);
 468
 469   min = 0;
 470   max = n_entries - 1;
 471   while (max >= min)
 472     {
 473       mid = (min + max) /  2;
 474       match_char = GET_UINT32 (cache->buffer, offset + 12 * mid);
 475       if (match_char < character)
 476         min = mid + 1;
 477       else if (match_char > character)
 478         max = mid - 1;
 479       else
 480         {
 481           len--;
 482           n = 0;
 483           n_children = GET_UINT32 (cache->buffer, offset + 12 * mid + 4);
 484           child_offset = GET_UINT32 (cache->buffer, offset + 12 * mid + 8);
 485
 486           if (len > 0)
 487             {
 488               n = cache_glob_node_lookup_suffix (cache,
 489                                                  n_children, child_offset,
 490                                                  file_name, len,
 491                                                  ignore_case,
 492                                                  mime_types,
 493                                                  n_mime_types);
 494             }
 495           if (n == 0)
 496             {
 497               i = 0;
 498               while (n < n_mime_types && i < n_children)
 499                 {
 500                   match_char = GET_UINT32 (cache->buffer, child_offset + 12 * i);
 501                   if (match_char != 0)
 502                     break;
 503
 504                   mimetype_offset = GET_UINT32 (cache->buffer, child_offset + 12 * i + 4);
 505                   weight = GET_UINT32 (cache->buffer, child_offset + 12 * i + 8);
 506
 507                   mime_types[n].mime = cache->buffer + mimetype_offset;
 508                   mime_types[n].weight = weight;
 509                   n++;
 510                   i++;
 511                 }
 512             }
 513           return n;
 514         }
 515     }
 516   return 0;
 517 }
 518
 519 static int
 520 cache_glob_lookup_suffix (xdg_unichar_t *file_name,
 521                           int            len,
 522                           int            ignore_case,
 523                           MimeWeight     mime_types[],
 524                           int            n_mime_types)
 525 {
 526   int i, n;
 527
 528   for (i = 0; _caches[i]; i++)
 529     {
 530       XdgMimeCache *cache = _caches[i];
 531
 532       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 16);
 533       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 534       xdg_uint32_t offset = GET_UINT32 (cache->buffer, list_offset + 4);
 535
 536       n = cache_glob_node_lookup_suffix (cache,
 537                                          n_entries, offset,
 538                                          file_name, len,
 539                                          ignore_case,
 540                                          mime_types,
 541                                          n_mime_types);
 542       if (n > 0)
 543         return n;
 544     }
 545
 546   return 0;
 547 }
 548
 549 static int compare_mime_weight (const void *a, const void *b)
 550 {
 551   const MimeWeight *aa = (const MimeWeight *)a;
 552   const MimeWeight *bb = (const MimeWeight *)b;
 553
 554   return aa->weight - bb->weight;
 555 }
 556
 557 static int
 558 cache_glob_lookup_file_name (const char *file_name,
 559                              const char *mime_types[],
 560                              int         n_mime_types)
 561 {
 562   int n;
 563   MimeWeight mimes[10];
 564   int n_mimes = 10;
 565   int i;
 566   xdg_unichar_t *ucs4;
 567   int len;
 568
 569   assert (file_name != NULL && n_mime_types > 0);
 570
 571   /* First, check the literals */
 572   n = cache_glob_lookup_literal (file_name, mime_types, n_mime_types);
 573   if (n > 0)
 574     return n;
 575
 576   ucs4 = _xdg_convert_to_ucs4 (file_name, &len);
 577   n = cache_glob_lookup_suffix (ucs4, len, FALSE, mimes, n_mimes);
 578
 579   if (n == 0)
 580     n = cache_glob_lookup_suffix (ucs4, len, TRUE, mimes, n_mimes);
 581   free(ucs4);
 582
 583   /* Last, try fnmatch */
 584   if (n == 0)
 585     n = cache_glob_lookup_fnmatch (file_name, mimes, n_mimes);
 586
 587   qsort (mimes, n, sizeof (MimeWeight), compare_mime_weight);
 588
 589   if (n_mime_types < n)
 590     n = n_mime_types;
 591
 592   for (i = 0; i < n; i++)
 593     mime_types[i] = mimes[i].mime;
 594
 595   return n;
 596 }
 597
 598 int
 599 _xdg_mime_cache_get_max_buffer_extents (void)
 600 {
 601   xdg_uint32_t offset;
 602   xdg_uint32_t max_extent;
 603   int i;
 604
 605   max_extent = 0;
 606   for (i = 0; _caches[i]; i++)
 607     {
 608       XdgMimeCache *cache = _caches[i];
 609
 610       offset = GET_UINT32 (cache->buffer, 24);
 611       max_extent = MAX (max_extent, GET_UINT32 (cache->buffer, offset + 4));
 612     }
 613
 614   return max_extent;
 615 }
 616
 617 static const char *
 618 cache_get_mime_type_for_data (const void *data,
 619                               size_t      len,
 620                               int        *result_prio,
 621                               const char *mime_types[],
 622                               int         n_mime_types)
 623 {
 624   const char *mime_type;
 625   int i, n, priority;
 626
 627   priority = 0;
 628   mime_type = NULL;
 629   for (i = 0; _caches[i]; i++)
 630     {
 631       XdgMimeCache *cache = _caches[i];
 632
 633       int prio;
 634       const char *match;
 635
 636       match = cache_magic_lookup_data (cache, data, len, &prio,
 637                                        mime_types, n_mime_types);
 638       if (prio > priority)
 639         {
 640           priority = prio;
 641           mime_type = match;
 642         }
 643     }
 644
 645   if (result_prio)
 646     *result_prio = priority;
 647
 648   if (priority > 0)
 649     return mime_type;
 650
 651   for (n = 0; n < n_mime_types; n++)
 652     {
 653
 654       if (mime_types[n])
 655         return mime_types[n];
 656     }
 657
 658   return XDG_MIME_TYPE_UNKNOWN;
 659 }
 660
 661 const char *
 662 _xdg_mime_cache_get_mime_type_for_data (const void *data,
 663                                         size_t      len,
 664                                         int        *result_prio)
 665 {
 666   return cache_get_mime_type_for_data (data, len, result_prio, NULL, 0);
 667 }
 668
 669 const char *
 670 _xdg_mime_cache_get_mime_type_for_file (const char  *file_name,
 671                                         struct stat *statbuf)
 672 {
 673   const char *mime_type;
 674   const char *mime_types[10];
 675   FILE *file;
 676   unsigned char *data;
 677   int max_extent;
 678   int bytes_read;
 679   struct stat buf;
 680   const char *base_name;
 681   int n;
 682
 683   if (file_name == NULL)
 684     return NULL;
 685
 686   if (! _xdg_utf8_validate (file_name))
 687     return NULL;
 688
 689   base_name = _xdg_get_base_name (file_name);
 690   n = cache_glob_lookup_file_name (base_name, mime_types, 10);
 691
 692   if (n == 1)
 693     return mime_types[0];
 694
 695   if (!statbuf)
 696     {
 697       if (stat (file_name, &buf) != 0)
 698         return XDG_MIME_TYPE_UNKNOWN;
 699
 700       statbuf = &buf;
 701     }
 702
 703   if (!S_ISREG (statbuf->st_mode))
 704     return XDG_MIME_TYPE_UNKNOWN;
 705
 706   /* FIXME: Need to make sure that max_extent isn't totally broken.  This could
 707    * be large and need getting from a stream instead of just reading it all
 708    * in. */
 709   max_extent = _xdg_mime_cache_get_max_buffer_extents ();
 710   data = malloc (max_extent);
 711   if (data == NULL)
 712     return XDG_MIME_TYPE_UNKNOWN;
 713
 714   file = fopen (file_name, "r");
 715   if (file == NULL)
 716     {
 717       free (data);
 718       return XDG_MIME_TYPE_UNKNOWN;
 719     }
 720
 721   bytes_read = fread (data, 1, max_extent, file);
 722   if (ferror (file))
 723     {
 724       free (data);
 725       fclose (file);
 726       return XDG_MIME_TYPE_UNKNOWN;
 727     }
 728
 729   mime_type = cache_get_mime_type_for_data (data, bytes_read, NULL,
 730                                             mime_types, n);
 731
 732   free (data);
 733   fclose (file);
 734
 735   return mime_type;
 736 }
 737
 738 const char *
 739 _xdg_mime_cache_get_mime_type_from_file_name (const char *file_name)
 740 {
 741   const char *mime_type;
 742
 743   if (cache_glob_lookup_file_name (file_name, &mime_type, 1))
 744     return mime_type;
 745   else
 746     return XDG_MIME_TYPE_UNKNOWN;
 747 }
 748
 749 int
 750 _xdg_mime_cache_get_mime_types_from_file_name (const char *file_name,
 751                                                const char  *mime_types[],
 752                                                int          n_mime_types)
 753 {
 754   return cache_glob_lookup_file_name (file_name, mime_types, n_mime_types);
 755 }
 756
 757 #if 1
 758 static int
 759 is_super_type (const char *mime)
 760 {
 761   int length;
 762   const char *type;
 763
 764   length = strlen (mime);
 765   type = &(mime[length - 2]);
 766
 767   if (strcmp (type, "/*") == 0)
 768     return 1;
 769
 770   return 0;
 771 }
 772 #endif
 773
 774 int
 775 _xdg_mime_cache_mime_type_subclass (const char *mime,
 776                                     const char *base)
 777 {
 778   const char *umime, *ubase;
 779
 780   int i, j, min, max, med, cmp;
 781
 782   umime = _xdg_mime_cache_unalias_mime_type (mime);
 783   ubase = _xdg_mime_cache_unalias_mime_type (base);
 784
 785   if (strcmp (umime, ubase) == 0)
 786     return 1;
 787
 788   /* We really want to handle text/ * in GtkFileFilter, so we just
 789    * turn on the supertype matching
 790    */
 791 #if 1
 792   /* Handle supertypes */
 793   if (is_super_type (ubase) &&
 794       xdg_mime_media_type_equal (umime, ubase))
 795     return 1;
 796 #endif
 797
 798   /*  Handle special cases text/plain and application/octet-stream */
 799   if (strcmp (ubase, "text/plain") == 0 &&
 800       strncmp (umime, "text/", 5) == 0)
 801     return 1;
 802
 803   if (strcmp (ubase, "application/octet-stream") == 0)
 804     return 1;
 805
 806   for (i = 0; _caches[i]; i++)
 807     {
 808       XdgMimeCache *cache = _caches[i];
 809
 810       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 8);
 811       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 812       xdg_uint32_t offset, n_parents, parent_offset;
 813
 814       min = 0;
 815       max = n_entries - 1;
 816       while (max >= min)
 817         {
 818           med = (min + max)/2;
 819
 820           offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * med);
 821           cmp = strcmp (cache->buffer + offset, umime);
 822           if (cmp < 0)
 823             min = med + 1;
 824           else if (cmp > 0)
 825             max = med - 1;
 826           else
 827             {
 828               offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * med + 4);
 829               n_parents = GET_UINT32 (cache->buffer, offset);
 830
 831               for (j = 0; j < n_parents; j++)
 832                 {
 833                   parent_offset = GET_UINT32 (cache->buffer, offset + 4 + 4 * j);
 834                   if (_xdg_mime_cache_mime_type_subclass (cache->buffer + parent_offset, ubase))
 835                     return 1;
 836                 }
 837
 838               break;
 839             }
 840         }
 841     }
 842
 843   return 0;
 844 }
 845
 846 const char *
 847 _xdg_mime_cache_unalias_mime_type (const char *mime)
 848 {
 849   const char *lookup;
 850
 851   lookup = cache_alias_lookup (mime);
 852
 853   if (lookup)
 854     return lookup;
 855
 856   return mime;
 857 }
 858
 859 char **
 860 _xdg_mime_cache_list_mime_parents (const char *mime)
 861 {
 862   int i, j, k, l, p;
 863   char *all_parents[128]; /* we'll stop at 128 */
 864   char **result;
 865
 866   mime = xdg_mime_unalias_mime_type (mime);
 867
 868   p = 0;
 869   for (i = 0; _caches[i]; i++)
 870     {
 871       XdgMimeCache *cache = _caches[i];
 872
 873       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 8);
 874       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 875
 876       for (j = 0; j < n_entries; j++)
 877         {
 878           xdg_uint32_t mimetype_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * j);
 879           xdg_uint32_t parents_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * j + 4);
 880
 881           if (strcmp (cache->buffer + mimetype_offset, mime) == 0)
 882             {
 883               xdg_uint32_t parent_mime_offset;
 884               xdg_uint32_t n_parents = GET_UINT32 (cache->buffer, parents_offset);
 885
 886               for (k = 0; k < n_parents && p < 127; k++)
 887                 {
 888                   parent_mime_offset = GET_UINT32 (cache->buffer, parents_offset + 4 + 4 * k);
 889
 890                   /* Don't add same parent multiple times.
 891                    * This can happen for instance if the same type is listed in multiple directories
 892                    */
 893                   for (l = 0; l < p; l++)
 894                     {
 895                       if (strcmp (all_parents[l], cache->buffer + parent_mime_offset) == 0)
 896                         break;
 897                     }
 898
 899                   if (l == p)
 900                     all_parents[p++] = cache->buffer + parent_mime_offset;
 901                 }
 902
 903               break;
 904             }
 905         }
 906     }
 907   all_parents[p++] = NULL;
 908
 909   result = (char **) malloc (p * sizeof (char *));
 910   memcpy (result, all_parents, p * sizeof (char *));
 911
 912   return result;
 913 }
 914
 915 static const char *
 916 cache_lookup_icon (const char *mime, int header)
 917 {
 918   const char *ptr;
 919   int i, min, max, mid, cmp;
 920
 921   for (i = 0; _caches[i]; i++)
 922     {
 923       XdgMimeCache *cache = _caches[i];
 924       xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, header);
 925       xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
 926       xdg_uint32_t offset;
 927
 928       min = 0;
 929       max = n_entries - 1;
 930       while (max >= min)
 931         {
 932           mid = (min + max) / 2;
 933
 934           offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid);
 935           ptr = cache->buffer + offset;
 936           cmp = strcmp (ptr, mime);
 937
 938           if (cmp < 0)
 939             min = mid + 1;
 940           else if (cmp > 0)
 941             max = mid - 1;
 942           else
 943             {
 944               offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid + 4);
 945               return cache->buffer + offset;
 946             }
 947         }
 948     }
 949
 950   return NULL;
 951 }
 952
 953 const char *
 954 _xdg_mime_cache_get_generic_icon (const char *mime)
 955 {
 956   return cache_lookup_icon (mime, 36);
 957 }
 958
 959 const char *
 960 _xdg_mime_cache_get_icon (const char *mime)
 961 {
 962   const char *icon;
 963
 964   icon = cache_lookup_icon (mime, 32);
 965
 966   if (icon == NULL)
 967     icon = _xdg_mime_cache_get_generic_icon (mime);
 968
 969   return icon;
 970 }
 971
 972 static void
 973 dump_glob_node (XdgMimeCache *cache,
 974                 xdg_uint32_t  offset,
 975                 int           depth)
 976 {
 977   xdg_unichar_t character;
 978   xdg_uint32_t mime_offset;
 979   xdg_uint32_t n_children;
 980   xdg_uint32_t child_offset;
 981   int i;
 982
 983   character = GET_UINT32 (cache->buffer, offset);
 984   mime_offset = GET_UINT32 (cache->buffer, offset + 4);
 985   n_children = GET_UINT32 (cache->buffer, offset + 8);
 986   child_offset = GET_UINT32 (cache->buffer, offset + 12);
 987   for (i = 0; i < depth; i++)
 988     printf (" ");
 989   printf ("%c", character);
 990   if (mime_offset)
 991     printf (" - %s", cache->buffer + mime_offset);
 992   printf ("\n");
 993   if (child_offset)
 994   {
 995     for (i = 0; i < n_children; i++)
 996       dump_glob_node (cache, child_offset + 20 * i, depth + 1);
 997   }
 998 }
 999
1000 void
1001 _xdg_mime_cache_glob_dump (void)
1002 {
1003   int i, j;
1004   for (i = 0; _caches[i]; i++)
1005   {
1006     XdgMimeCache *cache = _caches[i];
1007     xdg_uint32_t list_offset;
1008     xdg_uint32_t n_entries;
1009     xdg_uint32_t offset;
1010     list_offset = GET_UINT32 (cache->buffer, 16);
1011     n_entries = GET_UINT32 (cache->buffer, list_offset);
1012     offset = GET_UINT32 (cache->buffer, list_offset + 4);
1013     for (j = 0; j < n_entries; j++)
1014             dump_glob_node (cache, offset + 20 * j, 0);
1015   }
1016 }
1017
1018