Updated the MIME Parser API to include Content-ID in
[citadel.git] / libcitadel / lib / mime_parser.c
index 0da90aeb047e01ed9f759fdee2da521d91b6d59e..68f09a277514719acd1dfc9e3cc14ffbbc1eb6fa 100644 (file)
@@ -16,6 +16,8 @@
 #include <ctype.h>
 #include <string.h>
 #include <sys/stat.h>
+#include <sys/types.h>
+#include <dirent.h>
 #include <errno.h>
 
 #include "xdgmime/xdgmime.h"
@@ -114,6 +116,7 @@ void mime_decode(char *partnum,
                 char *part_start, size_t length,
                 char *content_type, char *charset, char *encoding,
                 char *disposition,
+                char *id,
                 char *name, char *filename,
                 void (*CallBack)
                  (char *cbname,
@@ -125,6 +128,7 @@ void mime_decode(char *partnum,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
                 void (*PreMultiPartCallBack)
                  (char *cbname,
@@ -136,6 +140,7 @@ void mime_decode(char *partnum,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
                 void (*PostMultiPartCallBack)
                  (char *cbname,
@@ -147,6 +152,7 @@ void mime_decode(char *partnum,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
                  void *userdata,
                  int dont_decode
@@ -169,7 +175,7 @@ void mime_decode(char *partnum,
                if (CallBack != NULL) {
                        CallBack(name, filename, fixed_partnum(partnum),
                                disposition, part_start,
-                               content_type, charset, length, encoding, userdata);
+                               content_type, charset, length, encoding, id, userdata);
                        }
                return;
        }
@@ -201,7 +207,7 @@ void mime_decode(char *partnum,
        if (bytes_decoded > 0) if (CallBack != NULL) {
                CallBack(name, filename, fixed_partnum(partnum),
                        disposition, decoded,
-                       content_type, charset, bytes_decoded, "binary", userdata);
+                       content_type, charset, bytes_decoded, "binary", id, userdata);
        }
 
        free(decoded);
@@ -225,6 +231,7 @@ void the_mime_parser(char *partnum,
                       char *cbcharset,
                       size_t cblength,
                       char *cbencoding,
+                      char *cbid,
                       void *cbuserdata),
                     void (*PreMultiPartCallBack)
                      (char *cbname,
@@ -236,6 +243,7 @@ void the_mime_parser(char *partnum,
                       char *cbcharset,
                       size_t cblength,
                       char *cbencoding,
+                      char *cbid,
                       void *cbuserdata),
                     void (*PostMultiPartCallBack)
                      (char *cbname,
@@ -247,6 +255,7 @@ void the_mime_parser(char *partnum,
                       char *cbcharset,
                       size_t cblength,
                       char *cbencoding,
+                      char *cbid,
                       void *cbuserdata),
                      void *userdata,
                      int dont_decode
@@ -268,6 +277,7 @@ void the_mime_parser(char *partnum,
        size_t content_length;
        char *encoding;
        char *disposition;
+       char *id;
        char *name = NULL;
        char *content_type_name;
        char *content_disposition_name;
@@ -318,6 +328,9 @@ void the_mime_parser(char *partnum,
        disposition = malloc(SIZ);
        memset(disposition, 0, SIZ);
 
+       id = malloc(SIZ);
+       memset(id, 0, SIZ);
+
        /* If the caller didn't supply an endpointer, generate one by measure */
        if (content_end == NULL) {
                content_end = &content_start[strlen(content_start)];
@@ -357,6 +370,11 @@ void the_mime_parser(char *partnum,
                                extract_key(content_disposition_name, disposition, "name");
                                extract_key(filename, disposition, "filename");
                        }
+                       if (!strncasecmp(header, "Content-ID:", 11)) {
+                               strcpy(id, &header[11]);
+                               striplt(id);
+                               stripallbut(id, '<', '>');
+                       }
                        if (!strncasecmp(header, "Content-length: ", 15)) {
                                char clbuf[10];
                                safestrncpy(clbuf, &header[15], sizeof clbuf);
@@ -398,7 +416,7 @@ void the_mime_parser(char *partnum,
                if (PreMultiPartCallBack != NULL) {
                        PreMultiPartCallBack("", "", partnum, "",
                                NULL, content_type, charset,
-                               0, encoding, userdata);
+                               0, encoding, id, userdata);
                }
 
                /* Figure out where the boundaries are */
@@ -476,7 +494,7 @@ void the_mime_parser(char *partnum,
 
                if (PostMultiPartCallBack != NULL) {
                        PostMultiPartCallBack("", "", partnum, "", NULL,
-                               content_type, charset, 0, encoding, userdata);
+                               content_type, charset, 0, encoding, id, userdata);
                }
                goto end_parser;
        }
@@ -491,18 +509,15 @@ void the_mime_parser(char *partnum,
                }
                part_end = content_end;
 
-               /******
-                * I thought there was an off-by-one error here, but there isn't.
-                * This probably means that there's an off-by-one error somewhere
-                * else ... or maybe only in certain messages?
-               --part_end;
-               --length;
-               ******/
-               
-               /* Truncate if the header told us to */
-               if ( (content_length > 0) && (length > content_length) ) {
-                       length = content_length;
-               }
+
+               /* The following code will truncate the MIME part to the size
+                * specified by the Content-length: header.   We have commented it
+                * out because these headers have a tendency to be wrong.
+                *
+                *      if ( (content_length > 0) && (length > content_length) ) {
+                *              length = content_length;
+                *      }
+                 */
 
                /* Sometimes the "name" field is tacked on to Content-type,
                 * and sometimes it's tacked on to Content-disposition.  Use
@@ -519,7 +534,7 @@ void the_mime_parser(char *partnum,
                 */
                mime_decode(partnum,
                        part_start, length,
-                       content_type, charset, encoding, disposition,
+                       content_type, charset, encoding, disposition, id,
                        name, filename,
                        CallBack, NULL, NULL,
                        userdata, dont_decode
@@ -533,7 +548,7 @@ void the_mime_parser(char *partnum,
                        if (PreMultiPartCallBack != NULL) {
                                PreMultiPartCallBack("", "", partnum, "",
                                        NULL, content_type, charset,
-                                       0, encoding, userdata);
+                                       0, encoding, id, userdata);
                        }
                        if (CallBack != NULL) {
                                if (strlen(partnum) > 0) {
@@ -558,7 +573,7 @@ void the_mime_parser(char *partnum,
                        }
                        if (PostMultiPartCallBack != NULL) {
                                PostMultiPartCallBack("", "", partnum, "", NULL,
-                                       content_type, charset, 0, encoding, userdata);
+                                       content_type, charset, 0, encoding, id, userdata);
                        }
 
 
@@ -578,6 +593,7 @@ end_parser: /* free the buffers!  end the oppression!! */
        free(content_disposition_name);
        free(filename);
        free(disposition);
+       free(id);
 }
 
 
@@ -601,6 +617,7 @@ void mime_parser(char *content_start,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
 
                 void (*PreMultiPartCallBack)
@@ -613,6 +630,7 @@ void mime_parser(char *content_start,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
 
                 void (*PostMultiPartCallBack)
@@ -625,6 +643,7 @@ void mime_parser(char *content_start,
                   char *cbcharset,
                   size_t cblength,
                   char *cbencoding,
+                  char *cbid,
                   void *cbuserdata),
 
                  void *userdata,
@@ -679,10 +698,8 @@ MimeGuess MyMimes [] = {
 };
 
 
-const char *GuessMimeType(char *data, size_t dlen)
+const char *GuessMimeType(const char *data, size_t dlen)
 {
-       return (xdg_mime_get_mime_type_for_data(data, dlen));
-/*
        int MimeIndex = 0;
 
        while (MyMimes[MimeIndex].PatternLen != 0)
@@ -693,19 +710,21 @@ const char *GuessMimeType(char *data, size_t dlen)
                            &data[MyMimes[MimeIndex].PatternOffset], 
                            MyMimes[MimeIndex].PatternLen) == 0)
                {
-                       break;
+                       return MyMimes[MimeIndex].MimeString;
                }
                MimeIndex ++;
        }
-       return MyMimes[MimeIndex].MimeString;
-*/
+       /* 
+        * 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)
 {
-       return xdg_mime_get_mime_type_from_file_name(what);
-/*
+       /* 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))
@@ -735,6 +754,142 @@ const char* GuessMimeByFilename(const char *what, size_t len)
        else if (!strncasecmp(&what[len - 5], ".wbmp", 5))
                return "image/vnd.wap.wbmp";
        else
-               return "application/octet-stream";
+               /* 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);
+       free(Icon);
+}
+
+/*
+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(1, NULL);
+       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);
+       }
+       closedir(filedir);
+       return 1;
+}
+
+const char *GetIconFilename(char *MimeType, size_t len)
+{
+       void *vIcon;
+       IconName *Icon;
+       
+       if(IconHash == NULL)
+               return NULL;
+
+       GetHash(IconHash, MimeType, len, &vIcon), Icon = (IconName*) vIcon;
+       /* 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, &vIcon),
+                               Icon = (IconName*) vIcon;
+               }
+       }
+       if (Icon == NULL) {
+               return NULL;
+       }
+
+       /*printf("Getting: [%s] == [%s] -> [%s]\n", MimeType, Icon->FlatName, Icon->FileName);*/
+       return Icon->FileName;
+}
+
+void ShutDownLibCitadel(void)
+{
+       DeleteHash(&IconHash);
 }