* fix freeing.. well, most of it
[citadel.git] / webcit / html2html.c
1 /*
2  * $Id$
3  */
4 /**
5  * \defgroup HTML2HTML Output an HTML message, modifying it slightly to make sure it plays nice
6  * with the rest of our web framework.
7  * \ingroup WebcitHttpServer
8  */
9 /*@{*/
10 #include "webcit.h"
11 #include "webserver.h"
12
13
14 /**
15  * \brief       Strip surrounding single or double quotes from a string.
16  *
17  * \param s     String to be stripped.
18  */
19 void stripquotes(char *s)
20 {
21         int len;
22
23         if (!s) return;
24
25         len = strlen(s);
26         if (len < 2) return;
27
28         if ( ( (s[0] == '\"') && (s[len-1] == '\"') ) || ( (s[0] == '\'') && (s[len-1] == '\'') ) ) {
29                 s[len-1] = 0;
30                 strcpy(s, &s[1]);
31         }
32 }
33
34
35 /**
36  * \brief Check to see if a META tag has overridden the declared MIME character set.
37  *
38  * \param charset               Character set name (left unchanged if we don't do anything)
39  * \param meta_http_equiv       Content of the "http-equiv" portion of the META tag
40  * \param meta_content          Content of the "content" portion of the META tag
41  */
42 void extract_charset_from_meta(char *charset, char *meta_http_equiv, char *meta_content)
43 {
44         char *ptr;
45         char buf[64];
46
47         if (!charset) return;
48         if (!meta_http_equiv) return;
49         if (!meta_content) return;
50
51
52         if (strcasecmp(meta_http_equiv, "Content-type")) return;
53
54         ptr = strchr(meta_content, ';');
55         if (!ptr) return;
56
57         safestrncpy(buf, ++ptr, sizeof buf);
58         striplt(buf);
59         if (!strncasecmp(buf, "charset=", 8)) {
60                 strcpy(charset, &buf[8]);
61
62                 /*
63                  * The brain-damaged webmail program in Microsoft Exchange declares
64                  * a charset of "unicode" when they really mean "UTF-8".  GNU iconv
65                  * treats "unicode" as an alias for "UTF-16" so we have to manually
66                  * fix this here, otherwise messages generated in Exchange webmail
67                  * show up as a big pile of weird characters.
68                  */
69                 if (!strcasecmp(charset, "unicode")) {
70                         strcpy(charset, "UTF-8");
71                 }
72
73         }
74 }
75
76
77
78 /**
79  * \brief Sanitize and enhance an HTML message for display.
80  *        Also convert weird character sets to UTF-8 if necessary.
81  *        Also fixup img src="cid:..." type inline images to fetch the image
82  *
83  * \param supplied_charset the input charset as declared in the MIME headers
84  */
85 void output_html(const char *supplied_charset, int treat_as_wiki, int msgnum, StrBuf *Source, StrBuf *Target) {
86         char buf[SIZ];
87         char *msg;
88         char *ptr;
89         char *msgstart;
90         char *msgend;
91         StrBuf *converted_msg;
92         int buffer_length = 1;
93         int line_length = 0;
94         int content_length = 0;
95         char new_window[SIZ];
96         int brak = 0;
97         int alevel = 0;
98         int scriptlevel = 0;
99         int script_start_pos = (-1);
100         int i;
101         int linklen;
102         char charset[128];
103 #ifdef HAVE_ICONV
104         iconv_t ic = (iconv_t)(-1) ;
105         char *ibuf;                   /**< Buffer of characters to be converted */
106         char *obuf;                   /**< Buffer for converted characters      */
107         size_t ibuflen;               /**< Length of input buffer               */
108         size_t obuflen;               /**< Length of output buffer              */
109         char *osav;                   /**< Saved pointer to output buffer       */
110 #endif
111         if (Target == NULL)
112                 Target = WC->WBuf;
113
114         safestrncpy(charset, supplied_charset, sizeof charset);
115         msg = strdup("");
116         sprintf(new_window, "<a target=\"%s\" href=", TARGET);
117
118         if (Source == NULL) while (serv_getln(buf, sizeof buf), strcmp(buf, "000")) {
119                 line_length = strlen(buf);
120                 buffer_length = content_length + line_length + 2;
121                 ptr = realloc(msg, buffer_length);
122                 if (ptr == NULL) {
123                         StrBufAppendPrintf(Target, "<b>");
124                         StrBufAppendPrintf(Target, _("realloc() error! couldn't get %d bytes: %s"),
125                                 buffer_length + 1,
126                                 strerror(errno));
127                         StrBufAppendPrintf(Target, "</b><br /><br />\n");
128                         while (serv_getln(buf, sizeof buf), strcmp(buf, "000")) {
129                                 /** flush */
130                         }
131                         free(msg);
132                         return;
133                 }
134                 msg = ptr;
135                 strcpy(&msg[content_length], buf);
136                 content_length += line_length;
137                 strcpy(&msg[content_length], "\n");
138                 content_length += 1;
139         }
140         else {
141                 content_length = StrLength(Source);
142                 free(msg);
143                 msg = (char*) ChrPtr(Source);/* TODO: remove cast */
144                 buffer_length = content_length;
145         }
146
147         /** Do a first pass to isolate the message body */
148         ptr = msg + 1;
149         msgstart = msg;
150         msgend = &msg[content_length];
151
152         while (ptr < msgend) {
153
154                 /** Advance to next tag */
155                 ptr = strchr(ptr, '<');
156                 if ((ptr == NULL) || (ptr >= msgend)) break;
157                 ++ptr;
158                 if ((ptr == NULL) || (ptr >= msgend)) break;
159
160                 /**
161                  *  Look for META tags.  Some messages (particularly in
162                  *  Asian locales) illegally declare a message's character
163                  *  set in the HTML instead of in the MIME headers.  This
164                  *  is wrong but we have to work around it anyway.
165                  */
166                 if (!strncasecmp(ptr, "META", 4)) {
167
168                         char *meta_start;
169                         char *meta_end;
170                         int meta_length;
171                         char *meta;
172                         char *meta_http_equiv;
173                         char *meta_content;
174                         char *spaceptr;
175
176                         meta_start = &ptr[4];
177                         meta_end = strchr(ptr, '>');
178                         if ((meta_end != NULL) && (meta_end <= msgend)) {
179                                 meta_length = meta_end - meta_start + 1;
180                                 meta = malloc(meta_length + 1);
181                                 safestrncpy(meta, meta_start, meta_length);
182                                 meta[meta_length] = 0;
183                                 striplt(meta);
184                                 if (!strncasecmp(meta, "HTTP-EQUIV=", 11)) {
185                                         meta_http_equiv = strdup(&meta[11]);
186                                         spaceptr = strchr(meta_http_equiv, ' ');
187                                         if (spaceptr != NULL) {
188                                                 *spaceptr = 0;
189                                                 meta_content = strdup(++spaceptr);
190                                                 if (!strncasecmp(meta_content, "content=", 8)) {
191                                                         strcpy(meta_content, &meta_content[8]);
192                                                         stripquotes(meta_http_equiv);
193                                                         stripquotes(meta_content);
194                                                         extract_charset_from_meta(charset,
195                                                                 meta_http_equiv, meta_content);
196                                                 }
197                                                 free(meta_content);
198                                         }
199                                         free(meta_http_equiv);
200                                 }
201                                 free(meta);
202                         }
203                 }
204
205                 /**
206                  * Any of these tags cause everything up to and including
207                  * the tag to be removed.
208                  */     
209                 if ( (!strncasecmp(ptr, "HTML", 4))
210                    ||(!strncasecmp(ptr, "HEAD", 4))
211                    ||(!strncasecmp(ptr, "/HEAD", 5))
212                    ||(!strncasecmp(ptr, "BODY", 4)) ) {
213                         ptr = strchr(ptr, '>');
214                         if ((ptr == NULL) || (ptr >= msgend)) break;
215                         ++ptr;
216                         if ((ptr == NULL) || (ptr >= msgend)) break;
217                         msgstart = ptr;
218                 }
219
220                 /**
221                  * Any of these tags cause everything including and following
222                  * the tag to be removed.
223                  */
224                 if ( (!strncasecmp(ptr, "/HTML", 5))
225                    ||(!strncasecmp(ptr, "/BODY", 5)) ) {
226                         --ptr;
227                         msgend = ptr;
228                         strcpy(ptr, "");
229                         
230                 }
231
232                 ++ptr;
233         }
234         if (msgstart > msg) {
235                 strcpy(msg, msgstart);
236         }
237
238         /** Now go through the message, parsing tags as necessary. */
239         converted_msg = NewStrBufPlain(NULL, content_length + 8192);
240
241
242         /** Convert foreign character sets to UTF-8 if necessary. */
243 #ifdef HAVE_ICONV
244         if ( (strcasecmp(charset, "us-ascii"))
245            && (strcasecmp(charset, "UTF-8"))
246            && (strcasecmp(charset, ""))
247         ) {
248                 lprintf(9, "Converting %s to UTF-8\n", charset);
249                 ctdl_iconv_open("UTF-8", charset, &ic);
250                 if (ic == (iconv_t)(-1) ) {
251                         lprintf(5, "%s:%d iconv_open() failed: %s\n",
252                                 __FILE__, __LINE__, strerror(errno));
253                 }
254         }
255         if  (Source == NULL) {
256                 if (ic != (iconv_t)(-1) ) {
257                         ibuf = msg;
258                         ibuflen = content_length;
259                         obuflen = content_length + (content_length / 2) ;
260                         obuf = (char *) malloc(obuflen);
261                         osav = obuf;
262                         iconv(ic, &ibuf, &ibuflen, &obuf, &obuflen);
263                         content_length = content_length + (content_length / 2) - obuflen;
264                         osav[content_length] = 0;
265                         free(msg);
266                         msg = osav;
267                         iconv_close(ic);
268                 }
269         }
270         else {
271                 if (ic != (iconv_t)(-1) ) {
272                         StrBuf *Buf = NewStrBufPlain(NULL, StrLength(Source) + 8096);;
273                         StrBufConvert(Source, Buf, &ic);
274                         FreeStrBuf(&Buf);
275                         iconv_close(ic);
276                 }
277         }
278                 
279 #endif
280
281         /**
282          *      At this point, the message has been stripped down to
283          *      only the content inside the <BODY></BODY> tags, and has
284          *      been converted to UTF-8 if it was originally in a foreign
285          *      character set.  The text is also guaranteed to be null
286          *      terminated now.
287          */
288
289         if (converted_msg == NULL) {
290                 StrBufAppendPrintf(Target, "Error %d: %s<br />%s:%d", errno, strerror(errno), __FILE__, __LINE__);
291                 goto BAIL;
292         }
293
294         ptr = msg;
295         msgend = strchr(msg, 0);
296         while (ptr < msgend) {
297
298                 /** Try to sanitize the html of any rogue scripts */
299                 if (!strncasecmp(ptr, "<script", 7)) {
300                         if (scriptlevel == 0) {
301                                 script_start_pos = StrLength(converted_msg);
302                         }
303                         ++scriptlevel;
304                 }
305                 if (!strncasecmp(ptr, "</script", 8)) {
306                         --scriptlevel;
307                 }
308
309                 /**
310                  * Change mailto: links to WebCit mail, by replacing the
311                  * link with one that points back to our mail room.  Due to
312                  * the way we parse URL's, it'll even handle mailto: links
313                  * that have "?subject=" in them.
314                  */
315                 if (!strncasecmp(ptr, "<a href=\"mailto:", 16)) {
316                         content_length += 64;
317                         StrBufAppendPrintf(converted_msg,
318                                 "<a href=\"display_enter?force_room=_MAIL_&recp=");
319                         ptr = &ptr[16];
320                         ++alevel;
321                         ++brak;
322                 }
323                 /** Make external links open in a separate window */
324                 else if (!strncasecmp(ptr, "<a href=\"", 9)) {
325                         ++alevel;
326                         ++brak;
327                         if ( ((strchr(ptr, ':') < strchr(ptr, '/')))
328                              &&  ((strchr(ptr, '/') < strchr(ptr, '>'))) 
329                              ) {
330                                 /* open external links to new window */
331                                 StrBufAppendPrintf(converted_msg, new_window);
332                                 ptr = &ptr[8];
333                         }
334                         else if ( (treat_as_wiki) && (strncasecmp(ptr, "<a href=\"wiki?", 14)) ) {
335                                 content_length += 64;
336                                 StrBufAppendPrintf(converted_msg, "<a href=\"wiki?page=");
337                                 ptr = &ptr[9];
338                         }
339                         else {
340                                 StrBufAppendPrintf(converted_msg, "<a href=\"");
341                                 ptr = &ptr[9];
342                         }
343                 }
344                 /** Fixup <img src="cid:... ...> to fetch the mime part */
345                 else if (!strncasecmp(ptr, "<img ", 5)) {
346                         char* tag_end=strchr(ptr,'>');
347                         char* src=strstr(ptr, " src=\"cid:");
348                         char *cid_start, *cid_end;
349                         ++brak;
350
351                         if (src && 
352                                 (cid_start=strchr(src,':')) && 
353                                 (cid_end=strchr(cid_start,'"')) &&
354                                 (cid_end < tag_end)) {
355
356                                 /* copy tag and attributes up to src="cid: */
357                                 StrBufAppendBufPlain(converted_msg, ptr, src - ptr, 0);
358                                 cid_start++;
359
360                                 /* add in /webcit/mimepart/<msgno>/CID/ 
361                                    trailing / stops dumb URL filters getting excited */
362                                 StrBufAppendPrintf(converted_msg,
363                                         "src=\"/webcit/mimepart/%d/",msgnum);
364                                 StrBufAppendBufPlain(converted_msg, cid_start, cid_end - cid_start, 0);
365                                 StrBufAppendBufPlain(converted_msg, "/\"", -1, 0);
366                                 
367                                 ptr = cid_end+1;
368                         }
369                         StrBufAppendBufPlain(converted_msg, ptr, tag_end - ptr, 0);
370                         ptr = tag_end;
371                 }
372
373                 /**
374                  * Turn anything that looks like a URL into a real link, as long
375                  * as it's not inside a tag already
376                  */
377                 else if ( (brak == 0) && (alevel == 0)
378                      && (!strncasecmp(ptr, "http://", 7))) {
379                                 /** Find the end of the link */
380                                 int strlenptr;
381                                 linklen = 0;
382                                 
383                                 strlenptr = strlen(ptr);
384                                 for (i=0; i<=strlenptr; ++i) {
385                                         if ((ptr[i]==0)
386                                            ||(isspace(ptr[i]))
387                                            ||(ptr[i]==10)
388                                            ||(ptr[i]==13)
389                                            ||(ptr[i]=='(')
390                                            ||(ptr[i]==')')
391                                            ||(ptr[i]=='<')
392                                            ||(ptr[i]=='>')
393                                            ||(ptr[i]=='[')
394                                            ||(ptr[i]==']')
395                                            ||(ptr[i]=='"')
396                                            ||(ptr[i]=='\'')
397                                         ) linklen = i;
398                                         /* did s.b. send us an entity? */
399                                         if (ptr[i] == '&') {
400                                                 if ((ptr[i+2] ==';') ||
401                                                     (ptr[i+3] ==';') ||
402                                                     (ptr[i+5] ==';') ||
403                                                     (ptr[i+6] ==';') ||
404                                                     (ptr[i+7] ==';'))
405                                                         linklen = i;
406                                         }
407                                         if (linklen > 0) break;
408                                 }
409                                 if (linklen > 0) {
410                                         char *ltreviewptr;
411                                         char *nbspreviewptr;
412                                         char linkedchar;
413                                         int len = linklen;
414                                         
415                                         len = linklen;
416                                         linkedchar = ptr[len];
417                                         ptr[len] = '\0';
418                                         /* spot for some subject strings tinymce tends to give us. */
419                                         ltreviewptr = strchr(ptr, '<');
420                                         if (ltreviewptr != NULL) {
421                                                 *ltreviewptr = '\0';
422                                                 linklen = ltreviewptr - ptr;
423                                         }
424
425                                         nbspreviewptr = strstr(ptr, "&nbsp;");
426                                         if (nbspreviewptr != NULL) {
427                                                 ///*nbspreviewptr = '\0';
428                                                 linklen = nbspreviewptr - ptr;
429                                         }
430                                         if (ltreviewptr != 0)
431                                                 *ltreviewptr = '<';
432
433                                         ptr[len] = linkedchar;
434
435                                         content_length += (32 + linklen);
436                                         StrBufAppendPrintf(converted_msg, "%s\"", new_window);
437                                         StrBufAppendBufPlain(converted_msg, ptr, linklen, 0);
438                                         StrBufAppendPrintf(converted_msg, "\">");
439                                         StrBufAppendBufPlain(converted_msg, ptr, linklen, 0);
440                                         ptr += linklen;
441                                         StrBufAppendPrintf(converted_msg, "</A>");
442                                 }
443                 }
444                 else {
445                         StrBufAppendBufPlain(converted_msg, ptr, 1, 0);
446                         ptr++;
447                 }
448
449                 /**
450                  * We need to know when we're inside a tag,
451                  * so we don't turn things that look like URL's into
452                  * links, when they're already links - or image sources.
453                  */
454                 if (*(ptr-1) == '<') {
455                         ++brak;
456                 }
457                 if (*(ptr-1) == '>') {
458                         --brak;
459                         if ((scriptlevel == 0) && (script_start_pos >= 0)) {
460                                 StrBufCutRight(converted_msg, StrLength(converted_msg) - script_start_pos);
461                                 script_start_pos = (-1);
462                         }
463                 }
464                 if (!strncasecmp(ptr, "</A>", 3)) --alevel;
465         }
466
467         /**     uncomment these two lines to override conversion        */
468         /**     memcpy(converted_msg, msg, content_length);             */
469         /**     output_length = content_length;                         */
470
471         /** Output our big pile of markup */
472         StrBufAppendBuf(Target, converted_msg, 0);
473
474 BAIL:   /** A little trailing vertical whitespace... */
475         StrBufAppendPrintf(Target, "<br /><br />\n");
476
477         /** Now give back the memory */
478         FreeStrBuf(&converted_msg);
479         if ((msg != NULL) && (Source == NULL)) free(msg);
480 }
481
482 /*@}*/