1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* xdgmimealias.c: Private file. mmappable caches for mime data
4 * More info can be found at http://www.freedesktop.org/standards/
6 * Copyright (C) 2005 Matthias Clasen <mclasen@redhat.com>
8 * Licensed under the Academic Free License version 2.0
9 * Or under the following terms:
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.
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.
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.
40 #include <netinet/in.h> /* for ntohl/ntohs */
45 #warning Building xdgmime without MMAP support. Binary "mime.info" cache files will not be used.
49 #include <sys/types.h>
51 #include "xdgmimecache.h"
52 #include "xdgmimeint.h"
55 #define MAX(a,b) ((a) > (b) ? (a) : (b))
71 #define MAP_FAILED ((void *) -1)
74 #define MAJOR_VERSION 1
75 #define MINOR_VERSION_MIN 1
76 #define MINOR_VERSION_MAX 2
87 #define GET_UINT16(cache,offset) (ntohs(*(xdg_uint16_t*)((cache) + (offset))))
88 #define GET_UINT32(cache,offset) (ntohl(*(xdg_uint32_t*)((cache) + (offset))))
91 _xdg_mime_cache_ref (XdgMimeCache *cache)
98 _xdg_mime_cache_unref (XdgMimeCache *cache)
102 if (cache->ref_count == 0)
105 munmap (cache->buffer, cache->size);
112 _xdg_mime_cache_new_from_file (const char *file_name)
114 XdgMimeCache *cache = NULL;
122 /* Open the file and map it into memory */
123 fd = open (file_name, O_RDONLY|_O_BINARY, 0);
128 if (fstat (fd, &st) < 0 || st.st_size < 4)
131 buffer = (char *) mmap (NULL, st.st_size, PROT_READ, MAP_SHARED, fd, 0);
133 if (buffer == MAP_FAILED)
136 minor = GET_UINT16 (buffer, 2);
138 if (GET_UINT16 (buffer, 0) != MAJOR_VERSION ||
139 (minor < MINOR_VERSION_MIN ||
140 minor > MINOR_VERSION_MAX))
142 munmap (buffer, st.st_size);
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;
157 #endif /* HAVE_MMAP */
163 cache_magic_matchlet_compare_to_data (XdgMimeCache *cache,
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);
176 for (i = range_start; i < range_start + range_length; i++)
178 int valid_matchlet = TRUE;
180 if (i + data_length > len)
185 for (j = 0; j < data_length; j++)
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]))
190 valid_matchlet = FALSE;
197 for (j = 0; j < data_length; j++)
199 if (((unsigned char *)cache->buffer)[data_offset + j] != ((unsigned char *) data)[j + i])
201 valid_matchlet = FALSE;
215 cache_magic_matchlet_compare (XdgMimeCache *cache,
220 xdg_uint32_t n_children = GET_UINT32 (cache->buffer, offset + 24);
221 xdg_uint32_t child_offset = GET_UINT32 (cache->buffer, offset + 28);
225 if (cache_magic_matchlet_compare_to_data (cache, offset, data, len))
230 for (i = 0; i < n_children; i++)
232 if (cache_magic_matchlet_compare (cache, child_offset + 32 * i,
242 cache_magic_compare_to_data (XdgMimeCache *cache,
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);
255 for (i = 0; i < n_matchlets; i++)
257 if (cache_magic_matchlet_compare (cache, matchlet_offset + i * 32,
262 return cache->buffer + mimetype_offset;
270 cache_magic_lookup_data (XdgMimeCache *cache,
274 const char *mime_types[],
277 xdg_uint32_t list_offset;
278 xdg_uint32_t n_entries;
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);
289 for (j = 0; j < n_entries; j++)
293 match = cache_magic_compare_to_data (cache, offset + 16 * j,
299 xdg_uint32_t mimetype_offset;
300 const char *non_match;
302 mimetype_offset = GET_UINT32 (cache->buffer, offset + 16 * j + 4);
303 non_match = cache->buffer + mimetype_offset;
305 for (n = 0; n < n_mime_types; n++)
308 _xdg_mime_mime_type_equal (mime_types[n], non_match))
309 mime_types[n] = NULL;
318 cache_alias_lookup (const char *alias)
321 int i, min, max, mid, cmp;
323 for (i = 0; _caches[i]; i++)
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);
334 mid = (min + max) / 2;
336 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid);
337 ptr = cache->buffer + offset;
338 cmp = strcmp (ptr, alias);
346 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid + 4);
347 return cache->buffer + offset;
361 cache_glob_lookup_literal (const char *file_name,
362 const char *mime_types[],
364 int case_sensitive_check)
367 int i, min, max, mid, cmp;
369 for (i = 0; _caches[i]; i++)
371 XdgMimeCache *cache = _caches[i];
372 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 12);
373 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
380 mid = (min + max) / 2;
382 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * mid);
383 ptr = cache->buffer + offset;
384 cmp = strcmp (ptr, file_name);
392 int weight = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * mid + 8);
393 int case_sensitive = weight & 0x100;
394 weight = weight & 0xff;
396 if (case_sensitive_check || !case_sensitive)
398 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * mid + 4);
399 mime_types[0] = (const char *)(cache->buffer + offset);
412 cache_glob_lookup_fnmatch (const char *file_name,
413 MimeWeight mime_types[],
416 const char *mime_type;
422 for (i = 0; _caches[i]; i++)
424 XdgMimeCache *cache = _caches[i];
426 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 20);
427 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
429 for (j = 0; j < n_entries && n < n_mime_types; j++)
431 xdg_uint32_t offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j);
432 xdg_uint32_t mimetype_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j + 4);
433 int weight = GET_UINT32 (cache->buffer, list_offset + 4 + 12 * j + 8);
434 weight = weight & 0xff;
435 ptr = cache->buffer + offset;
436 mime_type = cache->buffer + mimetype_offset;
438 /* FIXME: Not UTF-8 safe */
439 if (fnmatch (ptr, file_name, 0) == 0)
441 mime_types[n].mime = mime_type;
442 mime_types[n].weight = weight;
455 cache_glob_node_lookup_suffix (XdgMimeCache *cache,
456 xdg_uint32_t n_entries,
458 const char *file_name,
460 int case_sensitive_check,
461 MimeWeight mime_types[],
464 xdg_unichar_t character;
465 xdg_unichar_t match_char;
466 xdg_uint32_t mimetype_offset;
467 xdg_uint32_t n_children;
468 xdg_uint32_t child_offset;
472 int min, max, mid, n, i;
474 character = file_name[len - 1];
476 assert (character != 0);
482 mid = (min + max) / 2;
483 match_char = GET_UINT32 (cache->buffer, offset + 12 * mid);
484 if (match_char < character)
486 else if (match_char > character)
492 n_children = GET_UINT32 (cache->buffer, offset + 12 * mid + 4);
493 child_offset = GET_UINT32 (cache->buffer, offset + 12 * mid + 8);
497 n = cache_glob_node_lookup_suffix (cache,
498 n_children, child_offset,
500 case_sensitive_check,
507 while (n < n_mime_types && i < n_children)
509 match_char = GET_UINT32 (cache->buffer, child_offset + 12 * i);
513 mimetype_offset = GET_UINT32 (cache->buffer, child_offset + 12 * i + 4);
514 weight = GET_UINT32 (cache->buffer, child_offset + 12 * i + 8);
515 case_sensitive = weight & 0x100;
516 weight = weight & 0xff;
518 if (case_sensitive_check || !case_sensitive)
520 mime_types[n].mime = cache->buffer + mimetype_offset;
521 mime_types[n].weight = weight;
534 cache_glob_lookup_suffix (const char *file_name,
537 MimeWeight mime_types[],
542 for (i = 0; _caches[i]; i++)
544 XdgMimeCache *cache = _caches[i];
546 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 16);
547 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
548 xdg_uint32_t offset = GET_UINT32 (cache->buffer, list_offset + 4);
550 n = cache_glob_node_lookup_suffix (cache,
563 static int compare_mime_weight (const void *a, const void *b)
565 const MimeWeight *aa = (const MimeWeight *)a;
566 const MimeWeight *bb = (const MimeWeight *)b;
568 return bb->weight - aa->weight;
571 #define ISUPPER(c) ((c) >= 'A' && (c) <= 'Z')
573 ascii_tolower (const char *str)
577 lower = strdup (str);
582 *p++ = ISUPPER (c) ? c - 'A' + 'a' : c;
588 cache_glob_lookup_file_name (const char *file_name,
589 const char *mime_types[],
593 MimeWeight mimes[10];
599 assert (file_name != NULL && n_mime_types > 0);
601 /* First, check the literals */
603 lower_case = ascii_tolower (file_name);
605 n = cache_glob_lookup_literal (lower_case, mime_types, n_mime_types, FALSE);
612 n = cache_glob_lookup_literal (file_name, mime_types, n_mime_types, TRUE);
619 len = strlen (file_name);
620 n = cache_glob_lookup_suffix (lower_case, len, FALSE, mimes, n_mimes);
622 n = cache_glob_lookup_suffix (file_name, len, TRUE, mimes, n_mimes);
626 /* Last, try fnmatch */
628 n = cache_glob_lookup_fnmatch (file_name, mimes, n_mimes);
630 qsort (mimes, n, sizeof (MimeWeight), compare_mime_weight);
632 if (n_mime_types < n)
635 for (i = 0; i < n; i++)
636 mime_types[i] = mimes[i].mime;
642 _xdg_mime_cache_get_max_buffer_extents (void)
645 xdg_uint32_t max_extent;
649 for (i = 0; _caches[i]; i++)
651 XdgMimeCache *cache = _caches[i];
653 offset = GET_UINT32 (cache->buffer, 24);
654 max_extent = MAX (max_extent, GET_UINT32 (cache->buffer, offset + 4));
661 cache_get_mime_type_for_data (const void *data,
664 const char *mime_types[],
667 const char *mime_type;
672 for (i = 0; _caches[i]; i++)
674 XdgMimeCache *cache = _caches[i];
679 match = cache_magic_lookup_data (cache, data, len, &prio,
680 mime_types, n_mime_types);
689 *result_prio = priority;
694 for (n = 0; n < n_mime_types; n++)
698 return mime_types[n];
701 return XDG_MIME_TYPE_UNKNOWN;
705 _xdg_mime_cache_get_mime_type_for_data (const void *data,
709 return cache_get_mime_type_for_data (data, len, result_prio, NULL, 0);
713 _xdg_mime_cache_get_mime_type_for_file (const char *file_name,
714 struct stat *statbuf)
716 const char *mime_type;
717 const char *mime_types[10];
723 const char *base_name;
726 if (file_name == NULL)
729 if (! _xdg_utf8_validate (file_name))
732 base_name = _xdg_get_base_name (file_name);
733 n = cache_glob_lookup_file_name (base_name, mime_types, 10);
736 return mime_types[0];
740 if (stat (file_name, &buf) != 0)
741 return XDG_MIME_TYPE_UNKNOWN;
746 if (!S_ISREG (statbuf->st_mode))
747 return XDG_MIME_TYPE_UNKNOWN;
749 /* FIXME: Need to make sure that max_extent isn't totally broken. This could
750 * be large and need getting from a stream instead of just reading it all
752 max_extent = _xdg_mime_cache_get_max_buffer_extents ();
753 data = malloc (max_extent);
755 return XDG_MIME_TYPE_UNKNOWN;
757 file = fopen (file_name, "r");
761 return XDG_MIME_TYPE_UNKNOWN;
764 bytes_read = fread (data, 1, max_extent, file);
769 return XDG_MIME_TYPE_UNKNOWN;
772 mime_type = cache_get_mime_type_for_data (data, bytes_read, NULL,
782 _xdg_mime_cache_get_mime_type_from_file_name (const char *file_name)
784 const char *mime_type;
786 if (cache_glob_lookup_file_name (file_name, &mime_type, 1))
789 return XDG_MIME_TYPE_UNKNOWN;
793 _xdg_mime_cache_get_mime_types_from_file_name (const char *file_name,
794 const char *mime_types[],
797 return cache_glob_lookup_file_name (file_name, mime_types, n_mime_types);
802 is_super_type (const char *mime)
807 length = strlen (mime);
808 type = &(mime[length - 2]);
810 if (strcmp (type, "/*") == 0)
818 _xdg_mime_cache_mime_type_subclass (const char *mime,
821 const char *umime, *ubase;
823 int i, j, min, max, med, cmp;
825 umime = _xdg_mime_cache_unalias_mime_type (mime);
826 ubase = _xdg_mime_cache_unalias_mime_type (base);
828 if (strcmp (umime, ubase) == 0)
831 /* We really want to handle text/ * in GtkFileFilter, so we just
832 * turn on the supertype matching
835 /* Handle supertypes */
836 if (is_super_type (ubase) &&
837 xdg_mime_media_type_equal (umime, ubase))
841 /* Handle special cases text/plain and application/octet-stream */
842 if (strcmp (ubase, "text/plain") == 0 &&
843 strncmp (umime, "text/", 5) == 0)
846 if (strcmp (ubase, "application/octet-stream") == 0)
849 for (i = 0; _caches[i]; i++)
851 XdgMimeCache *cache = _caches[i];
853 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 8);
854 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
855 xdg_uint32_t offset, n_parents, parent_offset;
863 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * med);
864 cmp = strcmp (cache->buffer + offset, umime);
871 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * med + 4);
872 n_parents = GET_UINT32 (cache->buffer, offset);
874 for (j = 0; j < n_parents; j++)
876 parent_offset = GET_UINT32 (cache->buffer, offset + 4 + 4 * j);
877 if (_xdg_mime_cache_mime_type_subclass (cache->buffer + parent_offset, ubase))
890 _xdg_mime_cache_unalias_mime_type (const char *mime)
894 lookup = cache_alias_lookup (mime);
903 _xdg_mime_cache_list_mime_parents (const char *mime)
906 char *all_parents[128]; /* we'll stop at 128 */
909 mime = xdg_mime_unalias_mime_type (mime);
912 for (i = 0; _caches[i]; i++)
914 XdgMimeCache *cache = _caches[i];
916 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, 8);
917 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
919 for (j = 0; j < n_entries; j++)
921 xdg_uint32_t mimetype_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * j);
922 xdg_uint32_t parents_offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * j + 4);
924 if (strcmp (cache->buffer + mimetype_offset, mime) == 0)
926 xdg_uint32_t parent_mime_offset;
927 xdg_uint32_t n_parents = GET_UINT32 (cache->buffer, parents_offset);
929 for (k = 0; k < n_parents && p < 127; k++)
931 parent_mime_offset = GET_UINT32 (cache->buffer, parents_offset + 4 + 4 * k);
933 /* Don't add same parent multiple times.
934 * This can happen for instance if the same type is listed in multiple directories
936 for (l = 0; l < p; l++)
938 if (strcmp (all_parents[l], cache->buffer + parent_mime_offset) == 0)
943 all_parents[p++] = cache->buffer + parent_mime_offset;
950 all_parents[p++] = NULL;
952 result = (char **) malloc (p * sizeof (char *));
953 memcpy (result, all_parents, p * sizeof (char *));
959 cache_lookup_icon (const char *mime, int header)
962 int i, min, max, mid, cmp;
964 for (i = 0; _caches[i]; i++)
966 XdgMimeCache *cache = _caches[i];
967 xdg_uint32_t list_offset = GET_UINT32 (cache->buffer, header);
968 xdg_uint32_t n_entries = GET_UINT32 (cache->buffer, list_offset);
975 mid = (min + max) / 2;
977 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid);
978 ptr = cache->buffer + offset;
979 cmp = strcmp (ptr, mime);
987 offset = GET_UINT32 (cache->buffer, list_offset + 4 + 8 * mid + 4);
988 return cache->buffer + offset;
997 _xdg_mime_cache_get_generic_icon (const char *mime)
999 return cache_lookup_icon (mime, 36);
1003 _xdg_mime_cache_get_icon (const char *mime)
1005 return cache_lookup_icon (mime, 32);
1009 dump_glob_node (XdgMimeCache *cache,
1010 xdg_uint32_t offset,
1013 xdg_unichar_t character;
1014 xdg_uint32_t mime_offset;
1015 xdg_uint32_t n_children;
1016 xdg_uint32_t child_offset;
1019 character = GET_UINT32 (cache->buffer, offset);
1020 mime_offset = GET_UINT32 (cache->buffer, offset + 4);
1021 n_children = GET_UINT32 (cache->buffer, offset + 8);
1022 child_offset = GET_UINT32 (cache->buffer, offset + 12);
1023 for (i = 0; i < depth; i++)
1025 printf ("%c", character);
1027 printf (" - %s", cache->buffer + mime_offset);
1031 for (i = 0; i < n_children; i++)
1032 dump_glob_node (cache, child_offset + 20 * i, depth + 1);
1037 _xdg_mime_cache_glob_dump (void)
1040 for (i = 0; _caches[i]; i++)
1042 XdgMimeCache *cache = _caches[i];
1043 xdg_uint32_t list_offset;
1044 xdg_uint32_t n_entries;
1045 xdg_uint32_t offset;
1046 list_offset = GET_UINT32 (cache->buffer, 16);
1047 n_entries = GET_UINT32 (cache->buffer, list_offset);
1048 offset = GET_UINT32 (cache->buffer, list_offset + 4);
1049 for (j = 0; j < n_entries; j++)
1050 dump_glob_node (cache, offset + 20 * j, 0);