]> code.citadel.org Git - citadel.git/blobdiff - libcitadel/lib/mime_parser.c
* fixed several crash / invalid data cases in the hash implementation
[citadel.git] / libcitadel / lib / mime_parser.c
index 33fc854f74d55084d23b59464eb69fa3e5ee05b2..14f292a6def95a637e5fe9275f2804e03a21f88a 100644 (file)
@@ -3,7 +3,7 @@
  *
  * This is the MIME parser for Citadel.
  *
- * Copyright (c) 1998-2007 by Art Cancro
+ * Copyright (c) 1998-2007 by the citadel.org development team.
  * This code is distributed under the GNU General Public License v3.
  *
  */
 #include <ctype.h>
 #include <string.h>
 #include <sys/stat.h>
+#include <sys/types.h>
+#include <dirent.h>
 #include <errno.h>
 
+#include "xdgmime/xdgmime.h"
 #include "libcitadel.h"
 
-
 void extract_key(char *target, char *source, char *key)
 {
        char *ptr;
@@ -638,3 +640,232 @@ void mime_parser(char *content_start,
                        PostMultiPartCallBack,
                        userdata, dont_decode);
 }
+
+
+
+
+
+
+typedef struct _MimeGuess {
+       const char *Pattern;
+       size_t PatternLen;
+       long PatternOffset;
+       const char *MimeString;
+} MimeGuess;
+
+MimeGuess MyMimes [] = {
+       {
+               "GIF",
+               3,
+               0,
+               "image/gif"
+       },
+       {
+               "\xff\xd8",
+               2,
+               0,
+               "image/jpeg"
+       },
+       {
+               "\x89PNG",
+               4,
+               0,
+               "image/png"
+       },
+       { // last...
+               "",
+               0,
+               0,
+               ""
+       }
+};
+
+
+const char *GuessMimeType(char *data, size_t dlen)
+{
+       int MimeIndex = 0;
+
+       while (MyMimes[MimeIndex].PatternLen != 0)
+       {
+               if ((MyMimes[MimeIndex].PatternLen + 
+                    MyMimes[MimeIndex].PatternOffset < dlen) &&
+                   strncmp(MyMimes[MimeIndex].Pattern, 
+                           &data[MyMimes[MimeIndex].PatternOffset], 
+                           MyMimes[MimeIndex].PatternLen) == 0)
+               {
+                       return MyMimes[MimeIndex].MimeString;
+               }
+               MimeIndex ++;
+       }
+       /* 
+        * ok, our simple minded algorythm didn't find anything, 
+        * let the big chegger try it, he wil default to application/octet-stream
+        */
+       return (xdg_mime_get_mime_type_for_data(data, dlen));
+}
+
+
+const char* GuessMimeByFilename(const char *what, size_t len)
+{
+       /* we know some hardcoded on our own, try them... */
+       if (!strncasecmp(&what[len - 4], ".gif", 4))
+               return "image/gif";
+       else if (!strncasecmp(&what[len - 3], ".js", 3))
+               return  "text/javascript";
+       else if (!strncasecmp(&what[len - 4], ".txt", 4))
+               return "text/plain";
+       else if (!strncasecmp(&what[len - 4], ".css", 4))
+               return "text/css";
+       else if (!strncasecmp(&what[len - 4], ".jpg", 4))
+               return "image/jpeg";
+       else if (!strncasecmp(&what[len - 4], ".png", 4))
+               return "image/png";
+       else if (!strncasecmp(&what[len - 4], ".ico", 4))
+               return "image/x-icon";
+       else if (!strncasecmp(&what[len - 5], ".html", 5))
+               return "text/html";
+       else if (!strncasecmp(&what[len - 4], ".htm", 4))
+               return "text/html";
+       else if (!strncasecmp(&what[len - 4], ".wml", 4))
+               return "text/vnd.wap.wml";
+       else if (!strncasecmp(&what[len - 5], ".wmls", 5))
+               return "text/vnd.wap.wmlscript";
+       else if (!strncasecmp(&what[len - 5], ".wmlc", 5))
+               return "application/vnd.wap.wmlc";
+       else if (!strncasecmp(&what[len - 6], ".wmlsc", 6))
+               return "application/vnd.wap.wmlscriptc";
+       else if (!strncasecmp(&what[len - 5], ".wbmp", 5))
+               return "image/vnd.wap.wbmp";
+       else
+               /* and let xdgmime do the fallback. */
+               return xdg_mime_get_mime_type_from_file_name(what);
+}
+
+static HashList *IconHash = NULL;
+
+typedef struct IconName IconName;
+
+struct IconName {
+       char *FlatName;
+       char *FileName;
+};
+
+static void DeleteIcon(void *IconNamePtr)
+{
+       IconName *Icon = (IconName*) IconNamePtr;
+       free(Icon->FlatName);
+       free(Icon->FileName);
+}
+
+static const char *PrintFlat(void *IconNamePtr)
+{
+       IconName *Icon = (IconName*) IconNamePtr;
+       return Icon->FlatName;
+}
+static const char *PrintFile(void *IconNamePtr)
+{
+       IconName *Icon = (IconName*) IconNamePtr;
+       return Icon->FileName;
+}
+#define GENSTR "x-generic"
+#define IGNORE_PREFIX_1 "gnome-mime"
+int LoadIconDir(const char *DirName)
+{
+       DIR *filedir = NULL;
+       struct dirent *filedir_entry;
+       int d_namelen;
+       int d_without_ext;
+       IconName *Icon;
+
+       filedir = opendir (DirName);
+       IconHash = NewHash();
+       if (filedir == NULL) {
+               return 0;
+       }
+
+       while ((filedir_entry = readdir(filedir)))
+       {
+               char *MinorPtr;
+               char *PStart;
+#ifdef _DIRENT_HAVE_D_NAMELEN
+               d_namelen = filedir_entry->d_namelen;
+#else
+               d_namelen = strlen(filedir_entry->d_name);
+#endif
+               d_without_ext = d_namelen;
+               while ((d_without_ext > 0) && (filedir_entry->d_name[d_without_ext] != '.'))
+                       d_without_ext --;
+               if ((d_without_ext == 0) || (d_namelen < 3))
+                       continue;
+
+               if ((sizeof(IGNORE_PREFIX_1) < d_namelen) &&
+                   (strncmp(IGNORE_PREFIX_1, 
+                            filedir_entry->d_name, 
+                            sizeof(IGNORE_PREFIX_1) - 1) == 0)) {
+                       PStart = filedir_entry->d_name + sizeof(IGNORE_PREFIX_1);
+                       d_without_ext -= sizeof(IGNORE_PREFIX_1);
+               }
+               else {
+                       PStart = filedir_entry->d_name;
+               }
+               Icon = malloc(sizeof(IconName));
+
+               Icon->FileName = malloc(d_namelen + 1);
+               memcpy(Icon->FileName, filedir_entry->d_name, d_namelen + 1);
+
+               Icon->FlatName = malloc(d_without_ext + 1);
+               memcpy(Icon->FlatName, PStart, d_without_ext);
+               Icon->FlatName[d_without_ext] = '\0';
+               /* Try to find Minor type in image-jpeg */
+               MinorPtr = strchr(Icon->FlatName, '-');
+               if (MinorPtr != NULL) {
+                       size_t MinorLen;
+                       MinorLen = 1 + d_without_ext - (MinorPtr - Icon->FlatName + 1);
+                       if ((MinorLen == sizeof(GENSTR)) && 
+                           (strncmp(MinorPtr + 1, GENSTR, sizeof(GENSTR)) == 0)) {
+                               /* ok, we found a generic filename. cut the generic. */
+                               *MinorPtr = '\0';
+                               d_without_ext = d_without_ext - (MinorPtr - Icon->FlatName);
+                       }
+                       else { /* Map the major / minor separator to / */
+                               *MinorPtr = '/';
+                       }
+               }
+
+//             PrintHash(IconHash, PrintFlat, PrintFile);
+//             printf("%s - %s\n", Icon->FlatName, Icon->FileName);
+               Put(IconHash, Icon->FlatName, d_without_ext, Icon, DeleteIcon);
+//             PrintHash(IconHash, PrintFlat, PrintFile);
+       }
+       return 1;
+}
+
+const char *GetIconFilename(char *MimeType, size_t len)
+{
+       IconName *Icon;
+       
+       if(IconHash == NULL)
+               return NULL;
+
+       GetHash(IconHash, MimeType, len, (void**)&Icon);
+       /* didn't find the exact mimetype? try major only. */
+       if (Icon == NULL) {
+               char * pMinor;
+               pMinor = strchr(MimeType, '/');
+               if (pMinor != NULL) {
+                       *pMinor = '\0';
+                       GetHash(IconHash, MimeType, pMinor - MimeType, (void**)&Icon);
+               }
+       }
+       if (Icon == NULL) {
+               return NULL;
+       }
+
+       /*printf("Getting: [%s] == [%s] -> [%s]\n", MimeType, Icon->FlatName, Icon->FileName);*/
+       return Icon->FileName;
+}
+
+void ShutDownLibCitadel(void)
+{
+       DeleteHash(&IconHash);
+}