* rss_save_item(): encode sender & subject, so webcit will correctly display umlauts...
[citadel.git] / citadel / modules / rssclient / serv_rssclient.c
1 /*
2  * $Id$
3  *
4  * Bring external RSS feeds into rooms.
5  *
6  * Copyright (c) 2007-2009 by the citadel.org team
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 3 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  */
22
23 #include <stdlib.h>
24 #include <unistd.h>
25 #include <stdio.h>
26
27 #if TIME_WITH_SYS_TIME
28 # include <sys/time.h>
29 # include <time.h>
30 #else
31 # if HAVE_SYS_TIME_H
32 #  include <sys/time.h>
33 # else
34 #  include <time.h>
35 # endif
36 #endif
37
38 #include <ctype.h>
39 #include <string.h>
40 #include <errno.h>
41 #include <sys/types.h>
42 #include <sys/stat.h>
43 #include <expat.h>
44 #include <curl/curl.h>
45 #include <libcitadel.h>
46 #include "citadel.h"
47 #include "server.h"
48 #include "citserver.h"
49 #include "support.h"
50 #include "config.h"
51 #include "threads.h"
52 #include "ctdl_module.h"
53 #include "clientsocket.h"
54 #include "msgbase.h"
55 #include "parsedate.h"
56 #include "database.h"
57 #include "citadel_dirs.h"
58 #include "md5.h"
59 #include "context.h"
60
61 typedef struct rssnetcfg rssnetcfg;
62
63 struct rssnetcfg {
64         rssnetcfg *next;
65         char url[256];
66         char *rooms;
67         time_t last_error_when;
68         int ItemType;
69 };
70
71 #define RSS_UNSET 0
72 #define RSS_RSS 1
73 #define RSS_ATOM 2
74
75 typedef struct _rss_item {
76         char *chardata;
77         int chardata_len;
78         char *roomlist;
79         int done_parsing;
80         char *guid;
81         char *title;
82         char *link;
83         char *description;
84         time_t pubdate;
85         char channel_title[256];
86         int item_tag_nesting;
87         char *author_or_creator;
88         char *author_url;
89 }rss_item;
90
91
92 typedef struct _rsscollection {
93         rss_item *Item;
94         rssnetcfg *Cfg;
95         
96
97 } rsscollection;
98
99 struct rssnetcfg *rnclist = NULL;
100
101
102 #if 0
103 #ifdef HAVE_ICONV
104 #include <iconv.h>
105
106
107 /* 
108  * dug this out of the trashcan of the midgard project, lets see if it works for us.
109  * original code by Alexander Bokovoy <bokovoy@avilink.ne> distributed under GPL V2 or later
110  */
111
112 /* Returns: 
113  >= 0 - successfull, 0 means conversion doesn't use multibyte sequences 
114    -1 - error during iconv_open call 
115    -2 - error during iconv_close call 
116    ---------------------------------- 
117    This function expects that multibyte encoding in 'charset' wouldn't have 
118    characters with more than 3 bytes. It is not intended to convert UTF-8 because 
119    we'll never receive UTF-8 in our handler (it is handled by Exat itself). 
120 */ 
121 static int 
122 fill_encoding_info (const char *charset, XML_Encoding * info) 
123
124   iconv_t cd = (iconv_t)(-1); 
125   int flag; 
126         CtdlLogPrintf(0, "RSS: fill encoding info ...\n");
127  
128 #if G_BYTE_ORDER == G_LITTLE_ENDIAN 
129   cd = iconv_open ("UCS-2LE", charset); 
130 #else 
131   cd = iconv_open ("UCS-2BE", charset); 
132 #endif 
133  
134   if (cd == (iconv_t) (-1)) 
135     { 
136       return -1; 
137     } 
138  
139   { 
140     unsigned short out; 
141     unsigned char buf[4]; 
142     unsigned int i0, i1, i2; 
143     int result; 
144     flag = 0; 
145     for (i0 = 0; i0 < 0x100; i0++) 
146       { 
147         buf[0] = i0; 
148         info->map[i0] = 0; 
149         //result = try (cd, buf, 1, &out); 
150         if (result < 0) 
151           { 
152           } 
153         else if (result > 0) 
154           { 
155             info->map[i0] = out; 
156           } 
157         else 
158           { 
159             for (i1 = 0; i1 < 0x100; i1++) 
160               { 
161                 buf[1] = i1; 
162                 ///result = try (cd, buf, 2, &out); 
163                 if (result < 0) 
164                   { 
165                   } 
166                 else if (result > 0) 
167                   { 
168                     flag++; 
169                     info->map[i0] = -2; 
170                   } 
171                 else 
172                   { 
173                     for (i2 = 0; i2 < 0x100; i2++) 
174                       { 
175                         buf[2] = i2; 
176                         ////result = try (cd, buf, 3, &out); 
177                         if (result < 0) 
178                           { 
179                           } 
180                         else if (result > 0) 
181                           { 
182                             flag++; 
183                             info->map[i0] = -3; 
184                           } 
185                       } 
186                   } 
187               } 
188           } 
189       } 
190   } 
191  
192   if (iconv_close (cd) < 0) 
193     { 
194       return -2; 
195     } 
196   return flag; 
197
198
199 static int 
200 iconv_convertor (void *data, const char *s) 
201
202   XML_Encoding *info = data; 
203   int res; 
204         CtdlLogPrintf(0, "RSS: Converting ...\n");
205
206   if (s == NULL) 
207     return -1; 
208 /*
209   GByteArray *result; 
210   result = g_byte_array_new (); 
211   if (process_block (info->data, (char *) s, strlen (s), result) == 0) 
212     { 
213       res = *(result->data); 
214       g_byte_array_free (result, TRUE); 
215       return res; 
216     } 
217   g_byte_array_free (result, TRUE); 
218 */
219   return -1; 
220
221
222 static void 
223 my_release (void *data) 
224
225   iconv_t cd = (iconv_t) data; 
226   if (iconv_close (cd) != 0) 
227     { 
228 /// TODO: uh no.      exit (1); 
229     } 
230
231 int 
232 handle_unknown_xml_encoding (void *encodingHandleData, 
233                              const XML_Char * name, 
234                              XML_Encoding * info) 
235
236   int result; 
237   CtdlLogPrintf(0, "RSS: unknown encoding ...\n");
238   result = fill_encoding_info (name, info); 
239   if (result >= 0) 
240     { 
241       /*  
242         Special case: client asked for reverse conversion, we'll provide him with 
243         iconv descriptor which handles it. Client should release it by himself. 
244       */ 
245       if(encodingHandleData != NULL) 
246             *((iconv_t *)encodingHandleData) = iconv_open(name, "UTF-8"); 
247       /*  
248          Optimization: we do not need conversion function if encoding is one-to-one,  
249          info->map table will be enough  
250        */ 
251       if (result == 0) 
252         { 
253           info->data = NULL; 
254           info->convert = NULL; 
255           info->release = NULL; 
256           return 1; 
257         } 
258       /*  
259          We do need conversion function because this encoding uses multibyte sequences 
260        */ 
261       info->data = (void *) iconv_open ("UTF-8", name); 
262       if ((int)info->data == -1) 
263         return -1; 
264       info->convert = iconv_convertor; 
265       info->release = my_release; 
266       return 1; 
267     } 
268   if(encodingHandleData != NULL)  
269     *(iconv_t *)encodingHandleData = NULL; 
270   return 0; 
271
272
273 #endif
274 #endif
275
276 /*
277  * Commit a fetched and parsed RSS item to disk
278  */
279 void rss_save_item(rsscollection *rssc) {
280
281         struct MD5Context md5context;
282         u_char rawdigest[MD5_DIGEST_LEN];
283         int i;
284         char utmsgid[SIZ];
285         struct cdbdata *cdbut;
286         struct UseTable ut;
287         struct CtdlMessage *msg;
288         struct recptypes *recp = NULL;
289         int msglen = 0;
290         rss_item *ri = rssc->Item;
291
292         recp = (struct recptypes *) malloc(sizeof(struct recptypes));
293         if (recp == NULL) return;
294         memset(recp, 0, sizeof(struct recptypes));
295         memset(&ut, 0, sizeof(struct UseTable));
296         recp->recp_room = strdup(ri->roomlist);
297         recp->num_room = num_tokens(ri->roomlist, '|');
298         recp->recptypes_magic = RECPTYPES_MAGIC;
299    
300         /* Construct a GUID to use in the S_USETABLE table.
301          * If one is not present in the item itself, make one up.
302          */
303         if (ri->guid != NULL) {
304                 snprintf(utmsgid, sizeof utmsgid, "rss/%s", ri->guid);
305         }
306         else {
307                 MD5Init(&md5context);
308                 if (ri->title != NULL) {
309                         MD5Update(&md5context, (unsigned char*)ri->title, strlen(ri->title));
310                 }
311                 if (ri->link != NULL) {
312                         MD5Update(&md5context, (unsigned char*)ri->link, strlen(ri->link));
313                 }
314                 MD5Final(rawdigest, &md5context);
315                 for (i=0; i<MD5_DIGEST_LEN; i++) {
316                         sprintf(&utmsgid[i*2], "%02X", (unsigned char) (rawdigest[i] & 0xff));
317                         utmsgid[i*2] = tolower(utmsgid[i*2]);
318                         utmsgid[(i*2)+1] = tolower(utmsgid[(i*2)+1]);
319                 }
320                 strcat(utmsgid, "_rss2ctdl");
321         }
322
323         /* Find out if we've already seen this item */
324         cdbut = cdb_fetch(CDB_USETABLE, utmsgid, strlen(utmsgid));
325         if (cdbut != NULL) {
326                 /* Item has already been seen */
327                 CtdlLogPrintf(CTDL_DEBUG, "%s has already been seen\n", utmsgid);
328                 cdb_free(cdbut);
329
330                 /* rewrite the record anyway, to update the timestamp */
331                 strcpy(ut.ut_msgid, utmsgid);
332                 ut.ut_timestamp = time(NULL);
333                 cdb_store(CDB_USETABLE, utmsgid, strlen(utmsgid), &ut, sizeof(struct UseTable) );
334         }
335         else {
336                 /* Item has not been seen, so save it. */
337                 CtdlLogPrintf(CTDL_DEBUG, "RSS: saving item...\n");
338                 if (ri->description == NULL) ri->description = strdup("");
339                 for (i=strlen(ri->description); i>=0; --i) {
340                         if (isspace(ri->description[i])) {
341                                 ri->description[i] = ' ';
342                         }
343                 }
344
345                 msg = malloc(sizeof(struct CtdlMessage));
346                 memset(msg, 0, sizeof(struct CtdlMessage));
347                 msg->cm_magic = CTDLMESSAGE_MAGIC;
348                 msg->cm_anon_type = MES_NORMAL;
349                 msg->cm_format_type = FMT_RFC822;
350
351                 if (ri->guid != NULL) {
352                         msg->cm_fields['E'] = strdup(ri->guid);
353                 }
354
355                 if (ri->author_or_creator != NULL) {
356                         char *From;
357                         StrBuf *Encoded, *QPEncoded;
358                         StrBuf *UserName;
359                         StrBuf *EmailAddress;
360                         StrBuf *EncBuf;
361                         
362                         UserName = NewStrBuf();
363                         EmailAddress = NewStrBuf();
364                         EncBuf = NewStrBuf();
365
366                         From = html_to_ascii(ri->author_or_creator,
367                                              strlen(ri->author_or_creator), 
368                                              512, 0);
369
370                         Encoded = NewStrBufPlain(From, -1);
371                         free(From);
372                         QPEncoded = StrBufSanitizeEmailRecipientVector(Encoded, UserName, EmailAddress, EncBuf);
373                         msg->cm_fields['A'] = SmashStrBuf(&QPEncoded);
374
375                         FreeStrBuf(&Encoded);
376                         FreeStrBuf(&UserName);
377                         FreeStrBuf(&EmailAddress);
378                         FreeStrBuf(&EncBuf);
379
380                 }
381                 else {
382                         msg->cm_fields['A'] = strdup("rss");
383                 }
384
385                 msg->cm_fields['N'] = strdup(NODENAME);
386                 if (ri->title != NULL) {
387                         long len;
388                         char *Sbj;
389                         StrBuf *Encoded, *QPEncoded;
390
391                         QPEncoded = NULL;
392                         len = strlen(ri->title);
393                         Sbj = html_to_ascii(ri->title, len, 512, 0);
394                         Encoded = NewStrBufPlain(Sbj, -1);
395                         free(Sbj);
396
397                         StrBufTrim(Encoded);
398                         StrBufRFC2047encode(&QPEncoded, Encoded);
399
400                         msg->cm_fields['U'] = SmashStrBuf(&QPEncoded);
401                         FreeStrBuf(&Encoded);
402                 }
403                 msg->cm_fields['T'] = malloc(64);
404                 snprintf(msg->cm_fields['T'], 64, "%ld", ri->pubdate);
405                 if (ri->channel_title != NULL) {
406                         if (!IsEmptyStr(ri->channel_title)) {
407                                 msg->cm_fields['O'] = strdup(ri->channel_title);
408                         }
409                 }
410                 if (ri->link == NULL) 
411                         ri->link = strdup("");
412                 msglen += 1024 + strlen(ri->link) + strlen(ri->description) ;
413                 msg->cm_fields['M'] = malloc(msglen);
414                 snprintf(msg->cm_fields['M'], msglen,
415                         "Content-type: text/html; charset=\"UTF-8\"\r\n\r\n"
416                         "<html><body>\n"
417                         "%s<br><br>\n"
418                         "<a href=\"%s\">%s</a>\n"
419                         "</body></html>\n"
420                         ,
421                         ri->description,
422                         ri->link, ri->link
423                 );
424
425                 CtdlSubmitMsg(msg, recp, NULL, 0);
426                 CtdlFreeMessage(msg);
427
428                 /* write the uidl to the use table so we don't store this item again */
429                 strcpy(ut.ut_msgid, utmsgid);
430                 ut.ut_timestamp = time(NULL);
431                 cdb_store(CDB_USETABLE, utmsgid, strlen(utmsgid), &ut, sizeof(struct UseTable) );
432         }
433         free_recipients(recp);
434 }
435
436
437
438 /*
439  * Convert an RDF/RSS datestamp into a time_t
440  */
441 time_t rdf_parsedate(char *p)
442 {
443         struct tm tm;
444         time_t t = 0;
445
446         if (!p) return 0L;
447         if (strlen(p) < 10) return 0L;
448
449         memset(&tm, 0, sizeof tm);
450
451         /*
452          * If the timestamp appears to be in W3C datetime format, try to
453          * parse it.  See also: http://www.w3.org/TR/NOTE-datetime
454          *
455          * This code, along with parsedate.c, is a potential candidate for
456          * moving into libcitadel.
457          */
458         if ( (p[4] == '-') && (p[7] == '-') ) {
459                 tm.tm_year = atoi(&p[0]) - 1900;
460                 tm.tm_mon = atoi(&p[5]) - 1;
461                 tm.tm_mday = atoi(&p[8]);
462                 if ( (p[10] == 'T') && (p[13] == ':') ) {
463                         tm.tm_hour = atoi(&p[11]);
464                         tm.tm_min = atoi(&p[14]);
465                 }
466                 return mktime(&tm);
467         }
468
469         /* hmm... try RFC822 date stamp format */
470
471         t = parsedate(p);
472         if (t > 0) return(t);
473
474         /* yeesh.  ok, just return the current date and time. */
475         return(time(NULL));
476 }
477
478 #define RSS_UNSET 0
479 #define RSS_RSS 1
480 #define RSS_ATOM 2
481
482 void flush_rss_ite(rss_item *ri)
483 {
484         /* Initialize the feed item data structure */
485         if (ri->guid != NULL) free(ri->guid);
486         ri->guid = NULL;
487         if (ri->title != NULL) free(ri->title);
488         ri->title = NULL;
489         if (ri->link != NULL) free(ri->link);
490         ri->link = NULL;
491         if (ri->author_or_creator != NULL) free(ri->author_or_creator);
492         ri->author_or_creator = NULL;
493         if (ri->author_url != NULL) free(ri->author_url);
494         ri->author_url = NULL;
495         if (ri->description != NULL) free(ri->description);
496         ri->description = NULL;
497         /* Throw away any existing character data */
498         if (ri->chardata_len > 0) {
499                 free(ri->chardata);
500                 ri->chardata = 0;
501                 ri->chardata_len = 0;
502         }
503 }
504
505 void rss_xml_start(void *data, const char *supplied_el, const char **attr) {
506         rsscollection *rssc = (rsscollection*) data;
507         rss_item *ri = rssc->Item;
508         char el[256];
509         char *sep = NULL;
510
511         /* Axe the namespace, we don't care about it */
512 ///     CtdlLogPrintf(0, "RSS: supplied el %d: %s...\n", rssc->Cfg->ItemType, supplied_el);
513         safestrncpy(el, supplied_el, sizeof el);
514         while (sep = strchr(el, ':'), sep) {
515                 strcpy(el, ++sep);
516         }
517
518         if ((rssc->Cfg->ItemType == RSS_UNSET) && !strcasecmp(el, "rss")) 
519         {
520                 CtdlLogPrintf(9, "RSS: This is an RSS feed.\n");
521                 rssc->Cfg->ItemType = RSS_RSS;
522         }
523         if ((rssc->Cfg->ItemType == RSS_UNSET) && !strcasecmp(el, "rdf")) 
524         {
525                 CtdlLogPrintf(9, "RSS: This is an RDF feed.\n");
526                 rssc->Cfg->ItemType = RSS_RSS;
527         }
528         else if ((rssc->Cfg->ItemType == RSS_UNSET) && !strcasecmp(el, "feed")) 
529         {
530                 CtdlLogPrintf(9, "RSS: This is an ATOM feed.\n");
531                 rssc->Cfg->ItemType = RSS_ATOM;
532         }
533         else if ((rssc->Cfg->ItemType == RSS_RSS) &&
534             !strcasecmp(el, "item")) 
535         {
536                 ri->item_tag_nesting ++ ;
537                 flush_rss_ite(ri);
538         }
539         else if ( (rssc->Cfg->ItemType == RSS_ATOM) &&
540                  !strcasecmp(el, "entry")) 
541         { /* Atom feed... */
542                 ++ri->item_tag_nesting;
543                 flush_rss_ite(ri);
544         }
545         else if ((rssc->Cfg->ItemType == RSS_ATOM) &&
546                  !strcasecmp(el, "link"))
547         {
548                 int found ;
549                 int i;
550
551                 for (found = 0, i = 0;!found && attr[i] != NULL; i+=2)
552                 {
553                         if (!strcmp(attr[i], "href"))
554                         {
555                                 found = 1;
556                                 if (ri->link != NULL)
557                                         free(ri->link);
558                                 ri->link = strdup(attr[i+1]);
559                                 striplt(ri->link);
560                         }
561                 }
562
563         }
564
565 }
566
567 void rss_xml_end(void *data, const char *supplied_el) {
568         rsscollection *rssc = (rsscollection*) data;
569         rss_item *ri = rssc->Item;
570         char el[256];
571         char *sep = NULL;
572
573         /* Axe the namespace, we don't care about it */
574         safestrncpy(el, supplied_el, sizeof el);
575         while (sep = strchr(el, ':'), sep) {
576                 strcpy(el, ++sep);
577         }
578 //      CtdlLogPrintf(0, "RSS: END %s...\n", el);
579
580         if ( (!strcasecmp(el, "title")) && (ri->item_tag_nesting == 0) && (ri->chardata != NULL) ) {
581                 safestrncpy(ri->channel_title, ri->chardata, sizeof ri->channel_title);
582                 striplt(ri->channel_title);
583         }
584
585         if ( (rssc->Cfg->ItemType == RSS_RSS) && 
586              (!strcasecmp(el, "guid")) && (ri->chardata != NULL) ) {
587                 if (ri->guid != NULL) free(ri->guid);
588                 striplt(ri->chardata);
589                 ri->guid = strdup(ri->chardata);
590         }
591         else if ( (rssc->Cfg->ItemType == RSS_ATOM) && 
592                   (!strcasecmp(el, "id")) && (ri->chardata != NULL) ) {
593                 if (ri->guid != NULL) free(ri->guid);
594                 striplt(ri->chardata);
595                 ri->guid = strdup(ri->chardata);
596         }
597
598         else if ( (rssc->Cfg->ItemType == RSS_RSS) && (!strcasecmp(el, "link")) && (ri->chardata != NULL) ) {
599                 if (ri->link != NULL) free(ri->link);
600                 striplt(ri->chardata);
601                 ri->link = strdup(ri->chardata);
602         }
603
604         else if ( (!strcasecmp(el, "title")) && (ri->chardata != NULL) ) {
605                 if (ri->title != NULL) free(ri->title);
606                 striplt(ri->chardata);
607                 ri->title = strdup(ri->chardata);
608         }
609
610         else if ((rssc->Cfg->ItemType == RSS_ATOM) && 
611                  (!strcasecmp(el, "content")) && 
612                  (ri->chardata != NULL) ) {
613                 if (ri->description != NULL) free(ri->description);
614                 ri->description = strdup(ri->chardata);
615         }
616         else if ( (rssc->Cfg->ItemType == RSS_RSS) && 
617                   (!strcasecmp(el, "description")) &&
618                   (ri->chardata != NULL) ) {
619                 if (ri->description != NULL) free(ri->description);
620                 ri->description = strdup(ri->chardata);
621         }
622                   
623         else if ((rssc->Cfg->ItemType == RSS_ATOM) && 
624                  ((!strcasecmp(el, "published")) ||
625                   (!strcasecmp(el, "updated"))) &&
626                  (ri->chardata != NULL) ) {
627                 striplt(ri->chardata);
628                 ri->pubdate = rdf_parsedate(ri->chardata);
629         }
630
631
632         else if ((rssc->Cfg->ItemType == RSS_RSS) && 
633                  ((!strcasecmp(el, "pubdate")) || 
634                   (!strcasecmp(el, "date"))) && 
635                  (ri->chardata != NULL) ) {
636                 striplt(ri->chardata);
637                 ri->pubdate = rdf_parsedate(ri->chardata);
638         }
639
640         else if ((rssc->Cfg->ItemType == RSS_RSS) && 
641                  ((!strcasecmp(el, "author")) || 
642                   (!strcasecmp(el, "creator"))) && 
643                  (ri->chardata != NULL) ) {
644                 if (ri->author_or_creator != NULL) free(ri->author_or_creator);
645                 striplt(ri->chardata);
646                 ri->author_or_creator = strdup(ri->chardata);
647         }
648
649         else if ((rssc->Cfg->ItemType == RSS_ATOM) && 
650                  (!strcasecmp(el, "name")) && 
651                  (ri->chardata != NULL) ) {
652                 if (ri->author_or_creator != NULL) free(ri->author_or_creator);
653                 striplt(ri->chardata);
654                 ri->author_or_creator = strdup(ri->chardata);
655         }
656         else if ((rssc->Cfg->ItemType == RSS_ATOM) && 
657                  (!strcasecmp(el, "uri")) && 
658                  (ri->chardata != NULL) ) {
659                 if (ri->author_url != NULL) free(ri->author_url);
660                 striplt(ri->chardata);
661                 ri->author_url = strdup(ri->chardata);
662         }
663
664         else if ((rssc->Cfg->ItemType == RSS_RSS) && 
665                  !strcasecmp(el, "item")) {
666                 --ri->item_tag_nesting;
667                 rss_save_item(rssc);
668         }
669         else if ((rssc->Cfg->ItemType == RSS_ATOM) && 
670                  !strcasecmp(el, "entry")) {
671                 --ri->item_tag_nesting;
672                 rss_save_item(rssc);
673         }
674
675         else if ( (!strcasecmp(el, "rss")) || 
676                   (!strcasecmp(el, "rdf")) ) {
677 //              CtdlLogPrintf(CTDL_DEBUG, "End of feed detected.  Closing parser.\n");
678                 ri->done_parsing = 1;
679         }
680
681         if (ri->chardata_len > 0) {
682                 free(ri->chardata);
683                 ri->chardata = 0;
684                 ri->chardata_len = 0;
685         }
686
687 }
688
689
690 /*
691  * This callback stores up the data which appears in between tags.
692  */
693 void rss_xml_chardata(void *data, const XML_Char *s, int len) {
694         rsscollection *rssc = (rsscollection*) data;
695         rss_item *ri = rssc->Item;
696         int old_len;
697         int new_len;
698         char *new_buffer;
699
700         old_len = ri->chardata_len;
701         new_len = old_len + len;
702         new_buffer = realloc(ri->chardata, new_len + 1);
703         if (new_buffer != NULL) {
704                 memcpy(&new_buffer[old_len], s, len);
705                 new_buffer[new_len] = 0;
706                 ri->chardata = new_buffer;
707                 ri->chardata_len = new_len;
708         }
709 }
710
711
712
713 /*
714  * Callback function for passing libcurl's output to expat for parsing
715  */
716 size_t rss_libcurl_callback(void *ptr, size_t size, size_t nmemb, void *stream)
717 {
718         XML_Parse((XML_Parser)stream, ptr, (size * nmemb), 0);
719         return (size*nmemb);
720 }
721
722
723
724 /*
725  * Begin a feed parse
726  */
727 void rss_do_fetching(rssnetcfg *Cfg) {
728         rsscollection rssc;
729         rss_item ri;
730         XML_Parser xp;
731
732         CURL *curl;
733         CURLcode res;
734         char errmsg[1024] = "";
735
736         rssc.Item = &ri;
737         rssc.Cfg = Cfg;
738
739         CtdlLogPrintf(CTDL_DEBUG, "Fetching RSS feed <%s>\n", Cfg->url);
740
741         curl = curl_easy_init();
742         if (!curl) {
743                 CtdlLogPrintf(CTDL_ALERT, "Unable to initialize libcurl.\n");
744                 return;
745         }
746
747         xp = XML_ParserCreateNS("UTF-8", ':');
748         if (!xp) {
749                 CtdlLogPrintf(CTDL_ALERT, "Cannot create XML parser!\n");
750                 curl_easy_cleanup(curl);
751                 return;
752         }
753
754         curl_easy_setopt(curl, CURLOPT_URL, Cfg->url);
755         curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0);
756         curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0);
757         curl_easy_setopt(curl, CURLOPT_WRITEDATA, xp);
758         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, rss_libcurl_callback);
759         curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, errmsg);
760         curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1);
761 #ifdef CURLOPT_HTTP_CONTENT_DECODING
762         curl_easy_setopt(curl, CURLOPT_HTTP_CONTENT_DECODING, 1);
763         curl_easy_setopt(curl, CURLOPT_ENCODING, "");
764 #endif
765         curl_easy_setopt(curl, CURLOPT_USERAGENT, CITADEL);
766         curl_easy_setopt(curl, CURLOPT_TIMEOUT, 180);           /* die after 180 seconds */
767         if (!IsEmptyStr(config.c_ip_addr)) {
768                 curl_easy_setopt(curl, CURLOPT_INTERFACE, config.c_ip_addr);
769         }
770
771         memset(&ri, 0, sizeof(rss_item));
772         ri.roomlist = Cfg->rooms;
773 #ifdef HAVE_ICONV
774 #if 0
775         XML_SetUnknownEncodingHandler(xp,
776                                       handle_unknown_xml_encoding,
777                                       NULL);
778 #endif
779 #endif
780         XML_SetElementHandler(xp, rss_xml_start, rss_xml_end);
781         XML_SetCharacterDataHandler(xp, rss_xml_chardata);
782         XML_SetUserData(xp, &rssc);
783         if (CtdlThreadCheckStop())
784         {
785                 XML_ParserFree(xp);
786                 curl_easy_cleanup(curl);
787                 return;
788         }
789         
790         if (CtdlThreadCheckStop())
791                 goto shutdown ;
792
793         res = curl_easy_perform(curl);
794         if (res) {
795                 CtdlLogPrintf(CTDL_ALERT, "libcurl error %d: %s\n", res, errmsg);
796         }
797
798         if (CtdlThreadCheckStop())
799                 goto shutdown ;
800
801         if (ri.done_parsing == 0)
802                 XML_Parse(xp, "", 0, 1);
803
804
805         CtdlLogPrintf(CTDL_ALERT, "RSS: XML Status [%s] \n", 
806                       XML_ErrorString(
807                               XML_GetErrorCode(xp)));
808
809 shutdown:
810         curl_easy_cleanup(curl);
811         XML_ParserFree(xp);
812
813         flush_rss_ite(&ri);
814 }
815
816
817 /*
818  * Scan a room's netconfig to determine whether it is requesting any RSS feeds
819  */
820 void rssclient_scan_room(struct ctdlroom *qrbuf, void *data)
821 {
822         char filename[PATH_MAX];
823         char buf[1024];
824         char instr[32];
825         FILE *fp;
826         char feedurl[256];
827         rssnetcfg *rncptr = NULL;
828         rssnetcfg *use_this_rncptr = NULL;
829         int len = 0;
830         char *ptr = NULL;
831
832         assoc_file_name(filename, sizeof filename, qrbuf, ctdl_netcfg_dir);
833
834         if (CtdlThreadCheckStop())
835                 return;
836                 
837         /* Only do net processing for rooms that have netconfigs */
838         fp = fopen(filename, "r");
839         if (fp == NULL) {
840                 return;
841         }
842
843         while (fgets(buf, sizeof buf, fp) != NULL && !CtdlThreadCheckStop()) {
844                 buf[strlen(buf)-1] = 0;
845
846                 extract_token(instr, buf, 0, '|', sizeof instr);
847                 if (!strcasecmp(instr, "rssclient")) {
848
849                         use_this_rncptr = NULL;
850
851                         extract_token(feedurl, buf, 1, '|', sizeof feedurl);
852
853                         /* If any other rooms have requested the same feed, then we will just add this
854                          * room to the target list for that client request.
855                          */
856                         for (rncptr=rnclist; rncptr!=NULL; rncptr=rncptr->next) {
857                                 if (!strcmp(rncptr->url, feedurl)) {
858                                         use_this_rncptr = rncptr;
859                                 }
860                         }
861
862                         /* Otherwise create a new client request */
863                         if (use_this_rncptr == NULL) {
864                                 rncptr = (rssnetcfg *) malloc(sizeof(rssnetcfg));
865                                 rncptr->ItemType = RSS_UNSET;
866                                 if (rncptr != NULL) {
867                                         rncptr->next = rnclist;
868                                         safestrncpy(rncptr->url, feedurl, sizeof rncptr->url);
869                                         rncptr->rooms = NULL;
870                                         rnclist = rncptr;
871                                         use_this_rncptr = rncptr;
872                                 }
873                         }
874
875                         /* Add the room name to the request */
876                         if (use_this_rncptr != NULL) {
877                                 if (use_this_rncptr->rooms == NULL) {
878                                         rncptr->rooms = strdup(qrbuf->QRname);
879                                 }
880                                 else {
881                                         len = strlen(use_this_rncptr->rooms) + strlen(qrbuf->QRname) + 5;
882                                         ptr = realloc(use_this_rncptr->rooms, len);
883                                         if (ptr != NULL) {
884                                                 strcat(ptr, "|");
885                                                 strcat(ptr, qrbuf->QRname);
886                                                 use_this_rncptr->rooms = ptr;
887                                         }
888                                 }
889                         }
890                 }
891
892         }
893
894         fclose(fp);
895
896 }
897
898 /*
899  * Scan for rooms that have RSS client requests configured
900  */
901 void *rssclient_scan(void *args) {
902         static time_t last_run = 0L;
903         static int doing_rssclient = 0;
904         rssnetcfg *rptr = NULL;
905         CitContext rssclientCC;
906
907         /* Give this thread its own private CitContext */
908         CtdlFillSystemContext(&rssclientCC, "rssclient");
909         citthread_setspecific(MyConKey, (void *)&rssclientCC );
910
911         /*
912          * This is a simple concurrency check to make sure only one rssclient run
913          * is done at a time.  We could do this with a mutex, but since we
914          * don't really require extremely fine granularity here, we'll do it
915          * with a static variable instead.
916          */
917         if (doing_rssclient) return NULL;
918         doing_rssclient = 1;
919
920         CtdlLogPrintf(CTDL_DEBUG, "rssclient started\n");
921         CtdlForEachRoom(rssclient_scan_room, NULL);
922
923         while (rnclist != NULL && !CtdlThreadCheckStop()) {
924                 rss_do_fetching(rnclist);
925                 rptr = rnclist;
926                 rnclist = rnclist->next;
927                 if (rptr->rooms != NULL) free(rptr->rooms);
928                 free(rptr);
929         }
930
931         CtdlLogPrintf(CTDL_DEBUG, "rssclient ended\n");
932         last_run = time(NULL);
933         doing_rssclient = 0;
934         if (!CtdlThreadCheckStop())
935                 CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, last_run + config.c_net_freq);
936         else
937                 CtdlLogPrintf(CTDL_DEBUG, "rssclient: Task STOPPED.\n");
938         CtdlClearSystemContext();
939         return NULL;
940 }
941
942
943 CTDL_MODULE_INIT(rssclient)
944 {
945         if (threading)
946         {
947                 CtdlLogPrintf(CTDL_INFO, "%s\n", curl_version());
948                 CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, 0);
949         }
950         /* return our Subversion id for the Log */
951         return "$Id$";
952 }