citadel/modules/rssclient/serv_rssclient.c

   1 /*
   2  * $Id: serv_rssclient.c 5652 2007-10-29 20:14:48Z ajc $
   3  *
   4  * Bring external RSS feeds into rooms.
   5  *
   6  */
   7
   8 #include <stdlib.h>
   9 #include <unistd.h>
  10 #include <stdio.h>
  11
  12 #if TIME_WITH_SYS_TIME
  13 # include <sys/time.h>
  14 # include <time.h>
  15 #else
  16 # if HAVE_SYS_TIME_H
  17 #  include <sys/time.h>
  18 # else
  19 #  include <time.h>
  20 # endif
  21 #endif
  22
  23 #include <ctype.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28 #include <expat.h>
  29 #include <curl/curl.h>
  30 #include <libcitadel.h>
  31 #include "citadel.h"
  32 #include "server.h"
  33 #include "citserver.h"
  34 #include "support.h"
  35 #include "config.h"
  36 #include "threads.h"
  37 #include "room_ops.h"
  38 #include "ctdl_module.h"
  39 #include "clientsocket.h"
  40 #include "msgbase.h"
  41 #include "parsedate.h"
  42 #include "database.h"
  43 #include "citadel_dirs.h"
  44 #include "md5.h"
  45
  46
  47 struct rssnetcfg {
  48         struct rssnetcfg *next;
  49         char url[256];
  50         char *rooms;
  51 };
  52
  53 struct rss_item {
  54         char *chardata;
  55         int chardata_len;
  56         char *roomlist;
  57         int done_parsing;
  58         char *guid;
  59         char *title;
  60         char *link;
  61         char *description;
  62         time_t pubdate;
  63         char channel_title[256];
  64         int item_tag_nesting;
  65 };
  66
  67 struct rssnetcfg *rnclist = NULL;
  68
  69
  70 /*
  71  * Commit a fetched and parsed RSS item to disk
  72  */
  73 void rss_save_item(struct rss_item *ri) {
  74
  75         struct MD5Context md5context;
  76         u_char rawdigest[MD5_DIGEST_LEN];
  77         int i;
  78         char utmsgid[SIZ];
  79         struct cdbdata *cdbut;
  80         struct UseTable ut;
  81         struct CtdlMessage *msg;
  82         struct recptypes *recp = NULL;
  83         int msglen = 0;
  84
  85         recp = (struct recptypes *) malloc(sizeof(struct recptypes));
  86         if (recp == NULL) return;
  87         memset(recp, 0, sizeof(struct recptypes));
  88         recp->recp_room = strdup(ri->roomlist);
  89         recp->num_room = num_tokens(ri->roomlist, '|');
  90         recp->recptypes_magic = RECPTYPES_MAGIC;
  91
  92         /* Construct a GUID to use in the S_USETABLE table.
  93          * If one is not present in the item itself, make one up.
  94          */
  95         if (ri->guid != NULL) {
  96                 snprintf(utmsgid, sizeof utmsgid, "rss/%s", ri->guid);
  97         }
  98         else {
  99                 MD5Init(&md5context);
 100                 if (ri->title != NULL) {
 101                         MD5Update(&md5context, (unsigned char*)ri->title, strlen(ri->title));
 102                 }
 103                 if (ri->link != NULL) {
 104                         MD5Update(&md5context, (unsigned char*)ri->link, strlen(ri->link));
 105                 }
 106                 MD5Final(rawdigest, &md5context);
 107                 for (i=0; i<MD5_DIGEST_LEN; i++) {
 108                         sprintf(&utmsgid[i*2], "%02X", (unsigned char) (rawdigest[i] & 0xff));
 109                         utmsgid[i*2] = tolower(utmsgid[i*2]);
 110                         utmsgid[(i*2)+1] = tolower(utmsgid[(i*2)+1]);
 111                 }
 112                 strcat(utmsgid, "_rss2ctdl");
 113         }
 114
 115         /* Find out if we've already seen this item */
 116         cdbut = cdb_fetch(CDB_USETABLE, utmsgid, strlen(utmsgid));
 117         if (cdbut != NULL) {
 118                 /* Item has already been seen */
 119                 CtdlLogPrintf(CTDL_DEBUG, "%s has already been seen\n", utmsgid);
 120                 cdb_free(cdbut);
 121
 122                 /* rewrite the record anyway, to update the timestamp */
 123                 strcpy(ut.ut_msgid, utmsgid);
 124                 ut.ut_timestamp = time(NULL);
 125                 cdb_store(CDB_USETABLE, utmsgid, strlen(utmsgid), &ut, sizeof(struct UseTable) );
 126         }
 127         else {
 128                 /* Item has not been seen, so save it. */
 129
 130                 if (ri->description == NULL) ri->description = strdup("");
 131                 for (i=strlen(ri->description); i>=0; --i) {
 132                         if (isspace(ri->description[i])) {
 133                                 ri->description[i] = ' ';
 134                         }
 135                 }
 136
 137                 msg = malloc(sizeof(struct CtdlMessage));
 138                 memset(msg, 0, sizeof(struct CtdlMessage));
 139                 msg->cm_magic = CTDLMESSAGE_MAGIC;
 140                 msg->cm_anon_type = MES_NORMAL;
 141                 msg->cm_format_type = FMT_RFC822;
 142                 msg->cm_fields['A'] = strdup("rss");
 143                 msg->cm_fields['N'] = strdup(NODENAME);
 144                 msg->cm_fields['U'] = strdup(ri->title);
 145                 msg->cm_fields['T'] = malloc(64);
 146                 snprintf(msg->cm_fields['T'], 64, "%ld", ri->pubdate);
 147                 if (!IsEmptyStr(ri->channel_title)) {
 148                         msg->cm_fields['O'] = strdup(ri->channel_title);
 149                 }
 150
 151                 msglen = 1024 + strlen(ri->link) + strlen(ri->description) ;
 152                 msg->cm_fields['M'] = malloc(msglen);
 153                 snprintf(msg->cm_fields['M'], msglen,
 154                         "Content-type: text/html\r\n\r\n"
 155                         "<html><body>\n"
 156                         "%s<br><br>\n"
 157                         "<a href=\"%s\">%s</a>\n"
 158                         "</body></html>\n"
 159                         ,
 160                         ri->description,
 161                         ri->link, ri->link
 162                 );
 163
 164                 CtdlSubmitMsg(msg, recp, NULL, 0);
 165                 CtdlFreeMessage(msg);
 166
 167                 /* write the uidl to the use table so we don't store this item again */
 168                 strcpy(ut.ut_msgid, utmsgid);
 169                 ut.ut_timestamp = time(NULL);
 170                 cdb_store(CDB_USETABLE, utmsgid, strlen(utmsgid), &ut, sizeof(struct UseTable) );
 171         }
 172         free_recipients(recp);
 173 }
 174
 175
 176
 177 /*
 178  * Convert an RDF/RSS datestamp into a time_t
 179  */
 180 time_t rdf_parsedate(char *p)
 181 {
 182         struct tm tm;
 183         time_t t = 0;
 184
 185         if (!p) return 0L;
 186         if (strlen(p) < 10) return 0L;
 187
 188         memset(&tm, 0, sizeof tm);
 189
 190         /* YYYY-MM-DDTHH:MM format...
 191          */
 192         if ( (p[4] == '-') && (p[7] == '-') ) {
 193                 tm.tm_year = atoi(&p[0]) - 1900;
 194                 tm.tm_mon = atoi(&p[5]) - 1;
 195                 tm.tm_mday = atoi(&p[8]);
 196                 if ( (p[10] == 'T') && (p[13] == ':') ) {
 197                         tm.tm_hour = atoi(&p[11]);
 198                         tm.tm_min = atoi(&p[14]);
 199                 }
 200                 return mktime(&tm);
 201         }
 202
 203         /* hmm... try RFC822 date stamp format */
 204
 205         t = parsedate(p);
 206         if (t > 0) return(t);
 207
 208         /* yeesh.  ok, just return the current date and time. */
 209         return(time(NULL));
 210 }
 211
 212
 213
 214 void rss_xml_start(void *data, const char *supplied_el, const char **attr) {
 215         struct rss_item *ri = (struct rss_item *) data;
 216         char el[256];
 217         char *sep = NULL;
 218
 219         /* Axe the namespace, we don't care about it */
 220         safestrncpy(el, supplied_el, sizeof el);
 221         while (sep = strchr(el, ':'), sep) {
 222                 strcpy(el, ++sep);
 223         }
 224
 225         if (!strcasecmp(el, "item")) {
 226                 ++ri->item_tag_nesting;
 227
 228                 /* Initialize the feed item data structure */
 229                 if (ri->guid != NULL) free(ri->guid);
 230                 ri->guid = NULL;
 231                 if (ri->title != NULL) free(ri->title);
 232                 ri->title = NULL;
 233                 if (ri->link != NULL) free(ri->link);
 234                 ri->link = NULL;
 235                 if (ri->description != NULL) free(ri->description);
 236                 ri->description = NULL;
 237
 238                 /* Throw away any existing character data */
 239                 if (ri->chardata_len > 0) {
 240                         free(ri->chardata);
 241                         ri->chardata = 0;
 242                         ri->chardata_len = 0;
 243                 }
 244         }
 245
 246
 247
 248 }
 249
 250 void rss_xml_end(void *data, const char *supplied_el) {
 251         struct rss_item *ri = (struct rss_item *) data;
 252         char el[256];
 253         char *sep = NULL;
 254
 255         /* Axe the namespace, we don't care about it */
 256         safestrncpy(el, supplied_el, sizeof el);
 257         while (sep = strchr(el, ':'), sep) {
 258                 strcpy(el, ++sep);
 259         }
 260
 261         if ( (!strcasecmp(el, "title")) && (ri->item_tag_nesting == 0) && (ri->chardata != NULL) ) {
 262                 safestrncpy(ri->channel_title, ri->chardata, sizeof ri->channel_title);
 263                 striplt(ri->channel_title);
 264         }
 265
 266         if ( (!strcasecmp(el, "guid")) && (ri->chardata != NULL) ) {
 267                 if (ri->guid != NULL) free(ri->guid);
 268                 striplt(ri->chardata);
 269                 ri->guid = strdup(ri->chardata);
 270         }
 271
 272         if ( (!strcasecmp(el, "title")) && (ri->chardata != NULL) ) {
 273                 if (ri->title != NULL) free(ri->title);
 274                 striplt(ri->chardata);
 275                 ri->title = strdup(ri->chardata);
 276         }
 277
 278         if ( (!strcasecmp(el, "link")) && (ri->chardata != NULL) ) {
 279                 if (ri->link != NULL) free(ri->link);
 280                 striplt(ri->chardata);
 281                 ri->link = strdup(ri->chardata);
 282         }
 283
 284         if ( (!strcasecmp(el, "description")) && (ri->chardata != NULL) ) {
 285                 if (ri->description != NULL) free(ri->description);
 286                 ri->description = strdup(ri->chardata);
 287         }
 288
 289         if ( ((!strcasecmp(el, "pubdate")) || (!strcasecmp(el, "date"))) && (ri->chardata != NULL) ) {
 290                 striplt(ri->chardata);
 291                 ri->pubdate = rdf_parsedate(ri->chardata);
 292         }
 293
 294         if (!strcasecmp(el, "item")) {
 295                 --ri->item_tag_nesting;
 296                 rss_save_item(ri);
 297         }
 298
 299         if ( (!strcasecmp(el, "rss")) || (!strcasecmp(el, "rdf")) ) {
 300                 CtdlLogPrintf(CTDL_DEBUG, "End of feed detected.  Closing parser.\n");
 301                 ri->done_parsing = 1;
 302         }
 303
 304         if (ri->chardata_len > 0) {
 305                 free(ri->chardata);
 306                 ri->chardata = 0;
 307                 ri->chardata_len = 0;
 308         }
 309
 310 }
 311
 312
 313 /*
 314  * This callback stores up the data which appears in between tags.
 315  */
 316 void rss_xml_chardata(void *data, const XML_Char *s, int len) {
 317         struct rss_item *ri = (struct rss_item *) data;
 318         int old_len;
 319         int new_len;
 320         char *new_buffer;
 321
 322         old_len = ri->chardata_len;
 323         new_len = old_len + len;
 324         new_buffer = realloc(ri->chardata, new_len + 1);
 325         if (new_buffer != NULL) {
 326                 memcpy(&new_buffer[old_len], s, len);
 327                 new_buffer[new_len] = 0;
 328                 ri->chardata = new_buffer;
 329                 ri->chardata_len = new_len;
 330         }
 331 }
 332
 333
 334
 335 /*
 336  * Callback function for passing libcurl's output to expat for parsing
 337  */
 338 size_t rss_libcurl_callback(void *ptr, size_t size, size_t nmemb, void *stream)
 339 {
 340         XML_Parse((XML_Parser)stream, ptr, (size * nmemb), 0);
 341         return (size*nmemb);
 342 }
 343
 344
 345
 346 /*
 347  * Begin a feed parse
 348  */
 349 void rss_do_fetching(char *url, char *rooms) {
 350         struct rss_item ri;
 351         XML_Parser xp;
 352
 353         CURL *curl;
 354         CURLcode res;
 355         char errmsg[1024] = "";
 356
 357         CtdlLogPrintf(CTDL_DEBUG, "Fetching RSS feed <%s>\n", url);
 358
 359         curl = curl_easy_init();
 360         if (!curl) {
 361                 CtdlLogPrintf(CTDL_ALERT, "Unable to initialize libcurl.\n");
 362                 return;
 363         }
 364
 365         xp = XML_ParserCreateNS("UTF-8", ':');
 366         if (!xp) {
 367                 CtdlLogPrintf(CTDL_ALERT, "Cannot create XML parser!\n");
 368                 curl_easy_cleanup(curl);
 369                 return;
 370         }
 371
 372         curl_easy_setopt(curl, CURLOPT_URL, url);
 373         curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0);
 374         curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0);
 375         curl_easy_setopt(curl, CURLOPT_WRITEDATA, xp);
 376         curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, rss_libcurl_callback);
 377         curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, errmsg);
 378         curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1);
 379
 380         memset(&ri, 0, sizeof(struct rss_item));
 381         ri.roomlist = rooms;
 382         XML_SetElementHandler(xp, rss_xml_start, rss_xml_end);
 383         XML_SetCharacterDataHandler(xp, rss_xml_chardata);
 384         XML_SetUserData(xp, &ri);
 385
 386         if (CtdlThreadCheckStop())
 387         {
 388                 XML_ParserFree(xp);
 389                 curl_easy_cleanup(curl);
 390                 return;
 391         }
 392
 393         if (CtdlThreadCheckStop())
 394                 goto shutdown ;
 395
 396         res = curl_easy_perform(curl);
 397         if (res) {
 398                 CtdlLogPrintf(CTDL_ALERT, "libcurl error %d: %s\n", res, errmsg);
 399         }
 400
 401         if (CtdlThreadCheckStop())
 402                 goto shutdown ;
 403
 404         if (ri.done_parsing == 0) XML_Parse(xp, "", 0, 1);
 405
 406
 407 shutdown:
 408         curl_easy_cleanup(curl);
 409         XML_ParserFree(xp);
 410
 411         /* Free the feed item data structure */
 412         if (ri.guid != NULL) free(ri.guid);
 413         ri.guid = NULL;
 414         if (ri.title != NULL) free(ri.title);
 415         ri.title = NULL;
 416         if (ri.link != NULL) free(ri.link);
 417         ri.link = NULL;
 418         if (ri.description != NULL) free(ri.description);
 419         ri.description = NULL;
 420         if (ri.chardata_len > 0) {
 421                 free(ri.chardata);
 422                 ri.chardata = 0;
 423                 ri.chardata_len = 0;
 424         }
 425 }
 426
 427
 428 /*
 429  * Scan a room's netconfig to determine whether it is requesting any RSS feeds
 430  */
 431 void rssclient_scan_room(struct ctdlroom *qrbuf, void *data)
 432 {
 433         char filename[PATH_MAX];
 434         char buf[1024];
 435         char instr[32];
 436         FILE *fp;
 437         char feedurl[256];
 438         struct rssnetcfg *rncptr = NULL;
 439         struct rssnetcfg *use_this_rncptr = NULL;
 440         int len = 0;
 441         char *ptr = NULL;
 442
 443         assoc_file_name(filename, sizeof filename, qrbuf, ctdl_netcfg_dir);
 444
 445         if (CtdlThreadCheckStop())
 446                 return;
 447
 448         /* Only do net processing for rooms that have netconfigs */
 449         fp = fopen(filename, "r");
 450         if (fp == NULL) {
 451                 return;
 452         }
 453
 454         while (fgets(buf, sizeof buf, fp) != NULL && !CtdlThreadCheckStop()) {
 455                 buf[strlen(buf)-1] = 0;
 456
 457                 extract_token(instr, buf, 0, '|', sizeof instr);
 458                 if (!strcasecmp(instr, "rssclient")) {
 459
 460                         use_this_rncptr = NULL;
 461
 462                         extract_token(feedurl, buf, 1, '|', sizeof feedurl);
 463
 464                         /* If any other rooms have requested the same feed, then we will just add this
 465                          * room to the target list for that client request.
 466                          */
 467                         for (rncptr=rnclist; rncptr!=NULL; rncptr=rncptr->next) {
 468                                 if (!strcmp(rncptr->url, feedurl)) {
 469                                         use_this_rncptr = rncptr;
 470                                 }
 471                         }
 472
 473                         /* Otherwise create a new client request */
 474                         if (use_this_rncptr == NULL) {
 475                                 rncptr = (struct rssnetcfg *) malloc(sizeof(struct rssnetcfg));
 476                                 if (rncptr != NULL) {
 477                                         rncptr->next = rnclist;
 478                                         safestrncpy(rncptr->url, feedurl, sizeof rncptr->url);
 479                                         rncptr->rooms = NULL;
 480                                         rnclist = rncptr;
 481                                         use_this_rncptr = rncptr;
 482                                 }
 483                         }
 484
 485                         /* Add the room name to the request */
 486                         if (use_this_rncptr != NULL) {
 487                                 if (use_this_rncptr->rooms == NULL) {
 488                                         rncptr->rooms = strdup(qrbuf->QRname);
 489                                 }
 490                                 else {
 491                                         len = strlen(use_this_rncptr->rooms) + strlen(qrbuf->QRname) + 5;
 492                                         ptr = realloc(use_this_rncptr->rooms, len);
 493                                         if (ptr != NULL) {
 494                                                 strcat(ptr, "|");
 495                                                 strcat(ptr, qrbuf->QRname);
 496                                                 use_this_rncptr->rooms = ptr;
 497                                         }
 498                                 }
 499                         }
 500                 }
 501
 502         }
 503
 504         fclose(fp);
 505
 506 }
 507
 508 /*
 509  * Scan for rooms that have RSS client requests configured
 510  */
 511 void *rssclient_scan(void *args) {
 512         static time_t last_run = 0L;
 513         static int doing_rssclient = 0;
 514         struct rssnetcfg *rptr = NULL;
 515         struct CitContext rssclientCC;
 516
 517         /* Give this thread its own private CitContext */
 518         memset(&rssclientCC, 0, sizeof(struct CitContext));
 519         rssclientCC.internal_pgm = 1;
 520         rssclientCC.cs_pid = 0;
 521         pthread_setspecific(MyConKey, (void *)&rssclientCC );
 522
 523         CtdlThreadAllocTSD();
 524
 525         /*
 526          * This is a simple concurrency check to make sure only one rssclient run
 527          * is done at a time.  We could do this with a mutex, but since we
 528          * don't really require extremely fine granularity here, we'll do it
 529          * with a static variable instead.
 530          */
 531         if (doing_rssclient) return NULL;
 532         doing_rssclient = 1;
 533
 534         CtdlLogPrintf(CTDL_DEBUG, "rssclient started\n");
 535         ForEachRoom(rssclient_scan_room, NULL);
 536
 537         while (rnclist != NULL && !CtdlThreadCheckStop()) {
 538                 rss_do_fetching(rnclist->url, rnclist->rooms);
 539                 rptr = rnclist;
 540                 rnclist = rnclist->next;
 541                 if (rptr->rooms != NULL) free(rptr->rooms);
 542                 free(rptr);
 543         }
 544
 545         CtdlLogPrintf(CTDL_DEBUG, "rssclient ended\n");
 546         last_run = time(NULL);
 547         doing_rssclient = 0;
 548         if (!CtdlThreadCheckStop())
 549                 CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, last_run + config.c_net_freq);
 550         else
 551                 CtdlLogPrintf(CTDL_DEBUG, "rssclient: Task STOPPED.\n");
 552         return NULL;
 553 }
 554
 555
 556 CTDL_MODULE_INIT(rssclient)
 557 {
 558         if (threading)
 559         {
 560                 CtdlLogPrintf(CTDL_INFO, "%s\n", curl_version());
 561                 CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, 0);
 562         }
 563         /* return our Subversion id for the Log */
 564         return "$Id: serv_rssclient.c 5652 2007-10-29 20:14:48Z ajc $";
 565 }