Did away with lprintf all together now its called CtdlLogPrintf()
[citadel.git] / citadel / modules / rssclient / serv_rssclient.c
index fa0fa76c2de02daff9a71ed178ebb5114842db56..605b0772d70524977cae7784f783954ecf8784ea 100644 (file)
 #include <errno.h>
 #include <sys/types.h>
 #include <sys/stat.h>
+#include <expat.h>
+#include <libcitadel.h>
 #include "citadel.h"
 #include "server.h"
 #include "citserver.h"
 #include "support.h"
 #include "config.h"
-#include "tools.h"
+#include "threads.h"
 #include "room_ops.h"
 #include "ctdl_module.h"
 #include "clientsocket.h"
 #include "msgbase.h"
+#include "parsedate.h"
 #include "database.h"
 #include "citadel_dirs.h"
 #include "md5.h"
 
-#ifdef HAVE_EXPAT
-#include <expat.h>
-
 
 struct rssnetcfg {
        struct rssnetcfg *next;
@@ -115,7 +115,7 @@ void rss_save_item(struct rss_item *ri) {
        cdbut = cdb_fetch(CDB_USETABLE, utmsgid, strlen(utmsgid));
        if (cdbut != NULL) {
                /* Item has already been seen */
-               lprintf(CTDL_DEBUG, "%s has already been seen\n", utmsgid);
+               CtdlLogPrintf(CTDL_DEBUG, "%s has already been seen\n", utmsgid);
                cdb_free(cdbut);
 
                /* rewrite the record anyway, to update the timestamp */
@@ -126,6 +126,7 @@ void rss_save_item(struct rss_item *ri) {
        else {
                /* Item has not been seen, so save it. */
 
+               if (ri->description == NULL) ri->description = strdup("");
                for (i=strlen(ri->description); i>=0; --i) {
                        if (isspace(ri->description[i])) {
                                ri->description[i] = ' ';
@@ -161,13 +162,13 @@ void rss_save_item(struct rss_item *ri) {
 
                CtdlSubmitMsg(msg, recp, NULL);
                CtdlFreeMessage(msg);
-               free_recipients(recp);
 
                /* write the uidl to the use table so we don't store this item again */
                strcpy(ut.ut_msgid, utmsgid);
                ut.ut_timestamp = time(NULL);
                cdb_store(CDB_USETABLE, utmsgid, strlen(utmsgid), &ut, sizeof(struct UseTable) );
        }
+       free_recipients(recp);
 }
 
 
@@ -178,6 +179,7 @@ void rss_save_item(struct rss_item *ri) {
 time_t rdf_parsedate(char *p)
 {
        struct tm tm;
+       time_t t = 0;
 
        if (!p) return 0L;
        if (strlen(p) < 10) return 0L;
@@ -194,13 +196,16 @@ time_t rdf_parsedate(char *p)
                        tm.tm_hour = atoi(&p[11]);
                        tm.tm_min = atoi(&p[14]);
                }
+               return mktime(&tm);
        }
 
-       else {
-               /* FIXME try an imap timestamp conversion */
-       }
+       /* hmm... try RFC822 date stamp format */
+
+       t = parsedate(p);
+       if (t > 0) return(t);
 
-       return mktime(&tm);
+       /* yeesh.  ok, just return the current date and time. */
+       return(time(NULL));
 }
 
 
@@ -291,7 +296,7 @@ void rss_xml_end(void *data, const char *supplied_el) {
        }
 
        if ( (!strcasecmp(el, "rss")) || (!strcasecmp(el, "rdf")) ) {
-               lprintf(CTDL_DEBUG, "End of feed detected.  Closing parser.\n");
+               CtdlLogPrintf(CTDL_DEBUG, "End of feed detected.  Closing parser.\n");
                ri->done_parsing = 1;
        }
 
@@ -397,12 +402,15 @@ void rss_do_fetching(char *url, char *rooms) {
 
        /* Parse the URL */
        if (parse_url(url, rsshost, &rssport, rssurl) != 0) {
-               lprintf(CTDL_ALERT, "Invalid URL: %s\n", url);
+               CtdlLogPrintf(CTDL_ALERT, "Invalid URL: %s\n", url);
        }
+       
+       if (CtdlThreadCheckStop())
+               return;
 
        xp = XML_ParserCreateNS("UTF-8", ':');
        if (!xp) {
-               lprintf(CTDL_ALERT, "Cannot create XML parser!\n");
+               CtdlLogPrintf(CTDL_ALERT, "Cannot create XML parser!\n");
                return;
        }
 
@@ -412,44 +420,72 @@ void rss_do_fetching(char *url, char *rooms) {
        XML_SetCharacterDataHandler(xp, rss_xml_chardata);
        XML_SetUserData(xp, &ri);
 
-retry: lprintf(CTDL_NOTICE, "Connecting to <%s>\n", rsshost);
+       if (CtdlThreadCheckStop())
+       {
+               XML_ParserFree(xp);
+               return;
+       }
+       
+retry: CtdlLogPrintf(CTDL_NOTICE, "Connecting to <%s>\n", rsshost);
        sprintf(buf, "%d", rssport);
        sock = sock_connect(rsshost, buf, "tcp");
        if (sock >= 0) {
-               lprintf(CTDL_DEBUG, "Connected!\n");
+               CtdlLogPrintf(CTDL_DEBUG, "Connected!\n");
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                snprintf(buf, sizeof buf, "GET %s HTTP/1.0", rssurl);
-               lprintf(CTDL_DEBUG, "<%s\n", buf);
+               CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
                sock_puts(sock, buf);
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                snprintf(buf, sizeof buf, "Host: %s", rsshost);
-               lprintf(CTDL_DEBUG, "<%s\n", buf);
+               CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
                sock_puts(sock, buf);
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                snprintf(buf, sizeof buf, "User-Agent: %s", CITADEL);
-               lprintf(CTDL_DEBUG, "<%s\n", buf);
+               CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
                sock_puts(sock, buf);
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                snprintf(buf, sizeof buf, "Accept: */*");
-               lprintf(CTDL_DEBUG, "<%s\n", buf);
+               CtdlLogPrintf(CTDL_DEBUG, "<%s\n", buf);
                sock_puts(sock, buf);
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                sock_puts(sock, "");
 
+               if (CtdlThreadCheckStop())
+                       goto shutdown ;
+                       
                if (sock_getln(sock, buf, sizeof buf) >= 0) {
-               lprintf(CTDL_DEBUG, ">%s\n", buf);
+                       CtdlLogPrintf(CTDL_DEBUG, ">%s\n", buf);
                        remove_token(buf, 0, ' ');
 
                        /* 200 OK */
                        if (buf[0] == '2') {
 
-                               while (got_bytes = sock_getln(sock, buf, sizeof buf),
+                               while (got_bytes = sock_getln(sock, buf, sizeof buf),
                                      (got_bytes >= 0 && (strcmp(buf, "")) && (strcmp(buf, "\r"))) ) {
+                                       if (CtdlThreadCheckStop())
+                                               goto shutdown ;
                                        /* discard headers */
                                }
 
                                while (got_bytes = sock_read(sock, buf, sizeof buf, 0),
                                      ((got_bytes>=0) && (ri.done_parsing == 0)) ) {
+                                       if (CtdlThreadCheckStop())
+                                               goto shutdown ;
                                        XML_Parse(xp, buf, got_bytes, 0);
                                }
                                if (ri.done_parsing == 0) XML_Parse(xp, "", 0, 1);
@@ -459,25 +495,29 @@ retry:    lprintf(CTDL_NOTICE, "Connecting to <%s>\n", rsshost);
                        else if ( (!strncmp(buf, "30", 2)) && (redirect_count < 16) ) {
                                while (got_bytes = sock_getln(sock, buf, sizeof buf),
                                      (got_bytes >= 0 && (strcmp(buf, "")) && (strcmp(buf, "\r"))) ) {
+                                       if (CtdlThreadCheckStop())
+                                               goto shutdown ;
                                        if (!strncasecmp(buf, "Location:", 9)) {
                                                ++redirect_count;
                                                strcpy(buf, &buf[9]);
                                                striplt(buf);
                                                if (parse_url(buf, rsshost, &rssport, rssurl) == 0) {
+                                                       sock_close(sock);
                                                        goto retry;
                                                }
                                                else {
-                                                       lprintf(CTDL_ALERT, "Invalid URL: %s\n", buf);
+                                                       CtdlLogPrintf(CTDL_ALERT, "Invalid URL: %s\n", buf);
                                                }
                                        }
                                }
                        }
 
                }
+shutdown:
                sock_close(sock);
        }
        else {
-               lprintf(CTDL_ERR, "Could not connect: %s\n", strerror(errno));
+               CtdlLogPrintf(CTDL_ERR, "Could not connect: %s\n", strerror(errno));
        }
 
        XML_ParserFree(xp);
@@ -516,13 +556,16 @@ void rssclient_scan_room(struct ctdlroom *qrbuf, void *data)
 
        assoc_file_name(filename, sizeof filename, qrbuf, ctdl_netcfg_dir);
 
+       if (CtdlThreadCheckStop())
+               return;
+               
        /* Only do net processing for rooms that have netconfigs */
        fp = fopen(filename, "r");
        if (fp == NULL) {
                return;
        }
 
-       while (fgets(buf, sizeof buf, fp) != NULL) {
+       while (fgets(buf, sizeof buf, fp) != NULL && !CtdlThreadCheckStop()) {
                buf[strlen(buf)-1] = 0;
 
                extract_token(instr, buf, 0, '|', sizeof instr);
@@ -579,17 +622,19 @@ void rssclient_scan_room(struct ctdlroom *qrbuf, void *data)
 /*
  * Scan for rooms that have RSS client requests configured
  */
-void rssclient_scan(void) {
+void *rssclient_scan(void *args) {
        static time_t last_run = 0L;
        static int doing_rssclient = 0;
        struct rssnetcfg *rptr = NULL;
+       struct CitContext rssclientCC;
 
-       /*
-        * Run RSS client no more frequently than once every n seconds
-        */
-       if ( (time(NULL) - last_run) < config.c_net_freq ) {
-               return;
-       }
+       /* Give this thread its own private CitContext */
+       memset(&rssclientCC, 0, sizeof(struct CitContext));
+       rssclientCC.internal_pgm = 1;
+       rssclientCC.cs_pid = 0;
+       pthread_setspecific(MyConKey, (void *)&rssclientCC );
+
+       CtdlThreadAllocTSD();
 
        /*
         * This is a simple concurrency check to make sure only one rssclient run
@@ -597,13 +642,13 @@ void rssclient_scan(void) {
         * don't really require extremely fine granularity here, we'll do it
         * with a static variable instead.
         */
-       if (doing_rssclient) return;
+       if (doing_rssclient) return NULL;
        doing_rssclient = 1;
 
-       lprintf(CTDL_DEBUG, "rssclient started\n");
+       CtdlLogPrintf(CTDL_DEBUG, "rssclient started\n");
        ForEachRoom(rssclient_scan_room, NULL);
 
-       while (rnclist != NULL) {
+       while (rnclist != NULL && !CtdlThreadCheckStop()) {
                rss_do_fetching(rnclist->url, rnclist->rooms);
                rptr = rnclist;
                rnclist = rnclist->next;
@@ -611,21 +656,23 @@ void rssclient_scan(void) {
                free(rptr);
        }
 
-       lprintf(CTDL_DEBUG, "rssclient ended\n");
+       CtdlLogPrintf(CTDL_DEBUG, "rssclient ended\n");
        last_run = time(NULL);
        doing_rssclient = 0;
+       if (!CtdlThreadCheckStop())
+               CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, last_run + config.c_net_freq);
+       else
+               CtdlLogPrintf(CTDL_DEBUG, "rssclient: Task STOPPED.\n");
+       return NULL;
 }
 
 
-#endif /* HAVE_EXPAT */
-
 CTDL_MODULE_INIT(rssclient)
 {
-#ifdef HAVE_EXPAT
-       CtdlRegisterSessionHook(rssclient_scan, EVT_TIMER);
-#else
-        lprintf(CTDL_INFO, "This server is missing the Expat XML parser.  RSS client will be disabled.\n");
-#endif
+       if (threading)
+       {
+               CtdlThreadSchedule ("RSS Client", CTDLTHREAD_BIGSTACK, rssclient_scan, NULL, 0);
+       }
        /* return our Subversion id for the Log */
         return "$Id: serv_rssclient.c 5652 2007-10-29 20:14:48Z ajc $";
 }