notifier/rss.c

   1 /*
   2  * "$Id: rss.c 7824 2008-08-01 21:11:55Z mike $"
   3  *
   4  *   RSS notifier for the Common UNIX Printing System (CUPS).
   5  *
   6  *   Copyright 2007 by Apple Inc.
   7  *   Copyright 2007 by Easy Software Products.
   8  *
   9  *   These coded instructions, statements, and computer programs are the
  10  *   property of Apple Inc. and are protected by Federal copyright
  11  *   law.  Distribution and use rights are outlined in the file "LICENSE.txt"
  12  *   which should have been included with this file.  If this file is
  13  *   file is missing or damaged, see the license at "http://www.cups.org/".
  14  *
  15  * Contents:
  16  *
  17  *   main()           - Main entry for the test notifier.
  18  *   compare_rss()    - Compare two messages.
  19  *   delete_message() - Free all memory used by a message.
  20  *   load_rss()       - Load an existing RSS feed file.
  21  *   new_message()    - Create a new RSS message.
  22  *   password_cb()    - Return the cached password.
  23  *   save_rss()       - Save messages to a RSS file.
  24  *   xml_escape()     - Copy a string, escaping &, <, and > as needed.
  25  */
  26
  27 /*
  28  * Include necessary headers...
  29  */
  30
  31 #include <cups/cups.h>
  32 #include <cups/language.h>
  33 #include <cups/string.h>
  34 #include <cups/array.h>
  35 #include <errno.h>
  36 #include <sys/select.h>
  37
  38
  39 /*
  40  * Structures...
  41  */
  42
  43 typedef struct _cups_rss_s              /**** RSS message data ****/
  44 {
  45   int           sequence_number;        /* notify-sequence-number */
  46   char          *subject,               /* Message subject/summary */
  47                 *text,                  /* Message text */
  48                 *link_url;              /* Link to printer */
  49   time_t        event_time;             /* When the event occurred */
  50 } _cups_rss_t;
  51
  52
  53 /*
  54  * Local globals...
  55  */
  56
  57 static char             *rss_password;  /* Password for remote RSS */
  58
  59
  60 /*
  61  * Local functions...
  62  */
  63
  64 static int              compare_rss(_cups_rss_t *a, _cups_rss_t *b);
  65 static void             delete_message(_cups_rss_t *rss);
  66 static void             load_rss(cups_array_t *rss, const char *filename);
  67 static _cups_rss_t      *new_message(int sequence_number, char *subject,
  68                                      char *text, char *link_url,
  69                                      time_t event_time);
  70 static const char       *password_cb(const char *prompt);
  71 static int              save_rss(cups_array_t *rss, const char *filename,
  72                                  const char *baseurl);
  73 static char             *xml_escape(const char *s);
  74
  75
  76 /*
  77  * 'main()' - Main entry for the test notifier.
  78  */
  79
  80 int                                     /* O - Exit status */
  81 main(int  argc,                         /* I - Number of command-line arguments */
  82      char *argv[])                      /* I - Command-line arguments */
  83 {
  84   int           i;                      /* Looping var */
  85   ipp_t         *event;                 /* Event from scheduler */
  86   ipp_state_t   state;                  /* IPP event state */
  87   char          scheme[32],             /* URI scheme ("rss") */
  88                 username[256],          /* Username for remote RSS */
  89                 host[1024],             /* Hostname for remote RSS */
  90                 resource[1024],         /* RSS file */
  91                 *options;               /* Options */
  92   int           port,                   /* Port number for remote RSS */
  93                 max_events;             /* Maximum number of events */
  94   http_t        *http;                  /* Connection to remote server */
  95   http_status_t status;                 /* HTTP GET/PUT status code */
  96   char          filename[1024],         /* Local filename */
  97                 newname[1024];          /* filename.N */
  98   cups_lang_t   *language;              /* Language information */
  99   ipp_attribute_t *printer_up_time,     /* Timestamp on event */
 100                 *notify_sequence_number,/* Sequence number */
 101                 *notify_printer_uri;    /* Printer URI */
 102   char          *subject,               /* Subject for notification message */
 103                 *text,                  /* Text for notification message */
 104                 link_url[1024],         /* Link to printer */
 105                 link_scheme[32],        /* Scheme for link */
 106                 link_username[256],     /* Username for link */
 107                 link_host[1024],        /* Host for link */
 108                 link_resource[1024];    /* Resource for link */
 109   int           link_port;              /* Link port */
 110   cups_array_t  *rss;                   /* RSS message array */
 111   _cups_rss_t   *msg;                   /* RSS message */
 112   char          baseurl[1024];          /* Base URL */
 113   fd_set        input;                  /* Input set for select() */
 114   struct timeval timeout;               /* Timeout for select() */
 115   int           changed;                /* Has the RSS data changed? */
 116   int           exit_status;            /* Exit status */
 117
 118
 119   fprintf(stderr, "DEBUG: argc=%d\n", argc);
 120   for (i = 0; i < argc; i ++)
 121     fprintf(stderr, "DEBUG: argv[%d]=\"%s\"\n", i, argv[i]);
 122
 123  /*
 124   * See whether we are publishing this RSS feed locally or remotely...
 125   */
 126
 127   if (httpSeparateURI(HTTP_URI_CODING_ALL, argv[1], scheme, sizeof(scheme),
 128                       username, sizeof(username), host, sizeof(host), &port,
 129                       resource, sizeof(resource)) < HTTP_URI_OK)
 130   {
 131     fprintf(stderr, "ERROR: Bad RSS URI \"%s\"!\n", argv[1]);
 132     return (1);
 133   }
 134
 135   max_events = 20;
 136
 137   if ((options = strchr(resource, '?')) != NULL)
 138   {
 139     *options++ = '\0';
 140
 141     if (!strncmp(options, "max_events=", 11))
 142     {
 143       max_events = atoi(options + 11);
 144
 145       if (max_events <= 0)
 146         max_events = 20;
 147     }
 148   }
 149
 150   rss = cupsArrayNew((cups_array_func_t)compare_rss, NULL);
 151
 152   if (host[0])
 153   {
 154    /*
 155     * Remote feed, see if we can get the current file...
 156     */
 157
 158     int fd;                             /* Temporary file */
 159
 160
 161     if ((rss_password = strchr(username, ':')) != NULL)
 162       *rss_password++ = '\0';
 163
 164     cupsSetPasswordCB(password_cb);
 165     cupsSetUser(username);
 166
 167     if ((fd = cupsTempFd(filename, sizeof(filename))) < 0)
 168     {
 169       fprintf(stderr, "ERROR: Unable to create temporary file: %s\n",
 170               strerror(errno));
 171
 172       return (1);
 173     }
 174
 175     if ((http = httpConnect(host, port)) == NULL)
 176     {
 177       fprintf(stderr, "ERROR: Unable to connect to %s on port %d: %s\n",
 178               host, port, strerror(errno));
 179
 180       close(fd);
 181       unlink(filename);
 182
 183       return (1);
 184     }
 185
 186     status = cupsGetFd(http, resource, fd);
 187
 188     close(fd);
 189
 190     if (status != HTTP_OK && status != HTTP_NOT_FOUND)
 191     {
 192       fprintf(stderr, "ERROR: Unable to GET %s from %s on port %d: %d %s\n",
 193               resource, host, port, status, httpStatus(status));
 194
 195       httpClose(http);
 196       unlink(filename);
 197
 198       return (1);
 199     }
 200
 201     strlcpy(newname, filename, sizeof(newname));
 202
 203     httpAssembleURI(HTTP_URI_CODING_ALL, baseurl, sizeof(baseurl), "http",
 204                     NULL, host, port, resource);
 205   }
 206   else
 207   {
 208     const char  *cachedir,              /* CUPS_CACHEDIR */
 209                 *server_name,           /* SERVER_NAME */
 210                 *server_port;           /* SERVER_PORT */
 211
 212
 213     http = NULL;
 214
 215     if ((cachedir = getenv("CUPS_CACHEDIR")) == NULL)
 216       cachedir = CUPS_CACHEDIR;
 217
 218     if ((server_name = getenv("SERVER_NAME")) == NULL)
 219       server_name = "localhost";
 220
 221     if ((server_port = getenv("SERVER_PORT")) == NULL)
 222       server_port = "631";
 223
 224     snprintf(filename, sizeof(filename), "%s/rss%s", cachedir, resource);
 225     snprintf(newname, sizeof(newname), "%s.N", filename);
 226
 227     httpAssembleURIf(HTTP_URI_CODING_ALL, baseurl, sizeof(baseurl), "http",
 228                      NULL, server_name, atoi(server_port), "/rss%s", resource);
 229   }
 230
 231  /*
 232   * Load the previous RSS file, if any...
 233   */
 234
 235   load_rss(rss, filename);
 236
 237   changed = cupsArrayCount(rss) == 0;
 238
 239  /*
 240   * Localize for the user's chosen language...
 241   */
 242
 243   language = cupsLangDefault();
 244
 245  /*
 246   * Read events and update the RSS file until we are out of events.
 247   */
 248
 249   for (exit_status = 0, event = NULL;;)
 250   {
 251     if (changed)
 252     {
 253      /*
 254       * Save the messages to the file again, uploading as needed...
 255       */
 256
 257       if (save_rss(rss, newname, baseurl))
 258       {
 259         if (http)
 260         {
 261          /*
 262           * Upload the RSS file...
 263           */
 264
 265           if ((status = cupsPutFile(http, resource, filename)) != HTTP_CREATED)
 266             fprintf(stderr, "ERROR: Unable to PUT %s from %s on port %d: %d %s\n",
 267                     resource, host, port, status, httpStatus(status));
 268         }
 269         else
 270         {
 271          /*
 272           * Move the new RSS file over top the old one...
 273           */
 274
 275           if (rename(newname, filename))
 276             fprintf(stderr, "ERROR: Unable to rename %s to %s: %s\n",
 277                     newname, filename, strerror(errno));
 278         }
 279
 280         changed = 0;
 281       }
 282     }
 283
 284    /*
 285     * Wait up to 30 seconds for an event...
 286     */
 287
 288     timeout.tv_sec  = 30;
 289     timeout.tv_usec = 0;
 290
 291     FD_ZERO(&input);
 292     FD_SET(0, &input);
 293
 294     if (select(1, &input, NULL, NULL, &timeout) < 0)
 295       continue;
 296     else if (!FD_ISSET(0, &input))
 297     {
 298       fprintf(stderr, "DEBUG: %s is bored, exiting...\n", argv[1]);
 299       break;
 300     }
 301
 302    /*
 303     * Read the next event...
 304     */
 305
 306     event = ippNew();
 307     while ((state = ippReadFile(0, event)) != IPP_DATA)
 308     {
 309       if (state <= IPP_IDLE)
 310         break;
 311     }
 312
 313     if (state == IPP_ERROR)
 314       fputs("DEBUG: ippReadFile() returned IPP_ERROR!\n", stderr);
 315
 316     if (state <= IPP_IDLE)
 317       break;
 318
 319    /*
 320     * Collect the info from the event...
 321     */
 322
 323     printer_up_time        = ippFindAttribute(event, "printer-up-time",
 324                                               IPP_TAG_INTEGER);
 325     notify_sequence_number = ippFindAttribute(event, "notify-sequence-number",
 326                                               IPP_TAG_INTEGER);
 327     notify_printer_uri     = ippFindAttribute(event, "notify-printer-uri",
 328                                               IPP_TAG_URI);
 329     subject                = cupsNotifySubject(language, event);
 330     text                   = cupsNotifyText(language, event);
 331
 332     if (printer_up_time && notify_sequence_number && subject && text)
 333     {
 334      /*
 335       * Create a new RSS message...
 336       */
 337
 338       if (notify_printer_uri)
 339       {
 340         httpSeparateURI(HTTP_URI_CODING_ALL,
 341                         notify_printer_uri->values[0].string.text,
 342                         link_scheme, sizeof(link_scheme),
 343                         link_username, sizeof(link_username),
 344                         link_host, sizeof(link_host), &link_port,
 345                         link_resource, sizeof(link_resource));
 346         httpAssembleURI(HTTP_URI_CODING_ALL, link_url, sizeof(link_url),
 347                         "http", link_username, link_host, link_port,
 348                         link_resource);
 349       }
 350
 351       msg = new_message(notify_sequence_number->values[0].integer,
 352                         xml_escape(subject), xml_escape(text),
 353                         notify_printer_uri ? xml_escape(link_url) : NULL,
 354                         printer_up_time->values[0].integer);
 355
 356       if (!msg)
 357       {
 358         fprintf(stderr, "ERROR: Unable to create message: %s\n",
 359                 strerror(errno));
 360         exit_status = 1;
 361         break;
 362       }
 363
 364      /*
 365       * Add it to the array...
 366       */
 367
 368       cupsArrayAdd(rss, msg);
 369
 370       changed = 1;
 371
 372      /*
 373       * Trim the array as needed...
 374       */
 375
 376       while (cupsArrayCount(rss) > max_events)
 377       {
 378         msg = cupsArrayFirst(rss);
 379
 380         cupsArrayRemove(rss, msg);
 381
 382         delete_message(msg);
 383       }
 384     }
 385
 386     if (subject)
 387       free(subject);
 388
 389     if (text)
 390       free(text);
 391
 392     ippDelete(event);
 393     event = NULL;
 394   }
 395
 396  /*
 397   * We only get here when idle or error...
 398   */
 399
 400   ippDelete(event);
 401
 402   if (http)
 403   {
 404     unlink(filename);
 405     httpClose(http);
 406   }
 407
 408   return (exit_status);
 409 }
 410
 411
 412 /*
 413  * 'compare_rss()' - Compare two messages.
 414  */
 415
 416 static int                              /* O - Result of comparison */
 417 compare_rss(_cups_rss_t *a,             /* I - First message */
 418             _cups_rss_t *b)             /* I - Second message */
 419 {
 420   return (a->sequence_number - b->sequence_number);
 421 }
 422
 423
 424 /*
 425  * 'delete_message()' - Free all memory used by a message.
 426  */
 427
 428 static void
 429 delete_message(_cups_rss_t *msg)        /* I - RSS message */
 430 {
 431   if (msg->subject)
 432     free(msg->subject);
 433
 434   if (msg->text)
 435     free(msg->text);
 436
 437   if (msg->link_url)
 438     free(msg->link_url);
 439
 440   free(msg);
 441 }
 442
 443
 444 /*
 445  * 'load_rss()' - Load an existing RSS feed file.
 446  */
 447
 448 static void
 449 load_rss(cups_array_t *rss,             /* I - RSS messages */
 450          const char   *filename)        /* I - File to load */
 451 {
 452   FILE          *fp;                    /* File pointer */
 453   char          line[4096],             /* Line from file */
 454                 *subject,               /* Subject */
 455                 *text,                  /* Text */
 456                 *link_url,              /* Link URL */
 457                 *start,                 /* Start of element */
 458                 *end;                   /* End of element */
 459   time_t        event_time;             /* Event time */
 460   int           sequence_number;        /* Sequence number */
 461   int           in_item;                /* In an item */
 462   _cups_rss_t   *msg;                   /* New message */
 463
 464
 465   if ((fp = fopen(filename, "r")) == NULL)
 466   {
 467     if (errno != ENOENT)
 468       fprintf(stderr, "ERROR: Unable to open %s: %s\n", filename,
 469               strerror(errno));
 470
 471     return;
 472   }
 473
 474   subject         = NULL;
 475   text            = NULL;
 476   link_url        = NULL;
 477   event_time      = 0;
 478   sequence_number = 0;
 479   in_item         = 0;
 480
 481   while (fgets(line, sizeof(line), fp))
 482   {
 483     if (strstr(line, "<item>"))
 484       in_item = 1;
 485     else if (strstr(line, "</item>") && in_item)
 486     {
 487       if (subject && text)
 488       {
 489         msg = new_message(sequence_number, subject, text, link_url,
 490                           event_time);
 491
 492         if (msg)
 493           cupsArrayAdd(rss, msg);
 494
 495       }
 496       else
 497       {
 498         if (subject)
 499           free(subject);
 500
 501         if (text)
 502           free(text);
 503
 504         if (link_url)
 505           free(link_url);
 506       }
 507
 508       subject         = NULL;
 509       text            = NULL;
 510       link_url        = NULL;
 511       event_time      = 0;
 512       sequence_number = 0;
 513       in_item         = 0;
 514     }
 515     else if (!in_item)
 516       continue;
 517     else if ((start = strstr(line, "<title>")) != NULL)
 518     {
 519       start += 7;
 520       if ((end = strstr(start, "</title>")) != NULL)
 521       {
 522         *end    = '\0';
 523         subject = strdup(start);
 524       }
 525     }
 526     else if ((start = strstr(line, "<description>")) != NULL)
 527     {
 528       start += 13;
 529       if ((end = strstr(start, "</description>")) != NULL)
 530       {
 531         *end = '\0';
 532         text = strdup(start);
 533       }
 534     }
 535     else if ((start = strstr(line, "<link>")) != NULL)
 536     {
 537       start += 6;
 538       if ((end = strstr(start, "</link>")) != NULL)
 539       {
 540         *end     = '\0';
 541         link_url = strdup(start);
 542       }
 543     }
 544     else if ((start = strstr(line, "<pubDate>")) != NULL)
 545     {
 546       start += 9;
 547       if ((end = strstr(start, "</pubDate>")) != NULL)
 548       {
 549         *end       = '\0';
 550         event_time = httpGetDateTime(start);
 551       }
 552     }
 553     else if ((start = strstr(line, "<guid>")) != NULL)
 554       sequence_number = atoi(start + 6);
 555   }
 556
 557   fclose(fp);
 558 }
 559
 560
 561 /*
 562  * 'new_message()' - Create a new RSS message.
 563  */
 564
 565 static _cups_rss_t *                    /* O - New message */
 566 new_message(int    sequence_number,     /* I - notify-sequence-number */
 567             char   *subject,            /* I - Subject/summary */
 568             char   *text,               /* I - Text */
 569             char   *link_url,           /* I - Link to printer */
 570             time_t event_time)          /* I - Date/time of event */
 571 {
 572   _cups_rss_t   *msg;                   /* New message */
 573
 574
 575   if ((msg = calloc(1, sizeof(_cups_rss_t))) == NULL)
 576     return (NULL);
 577
 578   msg->sequence_number = sequence_number;
 579   msg->subject         = subject;
 580   msg->text            = text;
 581   msg->link_url        = link_url;
 582   msg->event_time      = event_time;
 583
 584   return (msg);
 585 }
 586
 587
 588 /*
 589  * 'password_cb()' - Return the cached password.
 590  */
 591
 592 static const char *                     /* O - Cached password */
 593 password_cb(const char *prompt)         /* I - Prompt string, unused */
 594 {
 595   (void)prompt;
 596
 597   return (rss_password);
 598 }
 599
 600
 601 /*
 602  * 'save_rss()' - Save messages to a RSS file.
 603  */
 604
 605 static int                              /* O - 1 on success, 0 on failure */
 606 save_rss(cups_array_t *rss,             /* I - RSS messages */
 607          const char   *filename,        /* I - File to save to */
 608          const char   *baseurl)         /* I - Base URL */
 609 {
 610   FILE          *fp;                    /* File pointer */
 611   _cups_rss_t   *msg;                   /* Current message */
 612   char          date[1024];             /* Current date */
 613   char          *href;                  /* Escaped base URL */
 614
 615
 616   if ((fp = fopen(filename, "w")) == NULL)
 617   {
 618     fprintf(stderr, "ERROR: Unable to create %s: %s\n", filename,
 619             strerror(errno));
 620     return (0);
 621   }
 622
 623   fputs("<?xml version=\"1.0\"?>\n", fp);
 624   fputs("<rss version=\"2.0\">\n", fp);
 625   fputs("  <channel>\n", fp);
 626   fputs("    <title>CUPS RSS Feed</title>\n", fp);
 627
 628   href = xml_escape(baseurl);
 629   fprintf(fp, "    <link>%s</link>\n", href);
 630   free(href);
 631
 632   fputs("    <description>CUPS RSS Feed</description>\n", fp);
 633   fputs("    <generator>" CUPS_SVERSION "</generator>\n", fp);
 634   fputs("    <ttl>1</ttl>\n", fp);
 635
 636   fprintf(fp, "    <pubDate>%s</pubDate>\n",
 637           httpGetDateString2(time(NULL), date, sizeof(date)));
 638
 639   for (msg = (_cups_rss_t *)cupsArrayLast(rss);
 640        msg;
 641        msg = (_cups_rss_t *)cupsArrayPrev(rss))
 642   {
 643     fputs("    <item>\n", fp);
 644     fprintf(fp, "      <title>%s</title>\n", msg->subject);
 645     fprintf(fp, "      <description>%s</description>\n", msg->text);
 646     if (msg->link_url)
 647       fprintf(fp, "      <link>%s</link>\n", msg->link_url);
 648     fprintf(fp, "      <pubDate>%s</pubDate>\n",
 649             httpGetDateString2(msg->event_time, date, sizeof(date)));
 650     fprintf(fp, "      <guid>%d</guid>\n", msg->sequence_number);
 651     fputs("    </item>\n", fp);
 652   }
 653
 654   fputs(" </channel>\n", fp);
 655   fputs("</rss>\n", fp);
 656
 657   return (!fclose(fp));
 658 }
 659
 660
 661 /*
 662  * 'xml_escape()' - Copy a string, escaping &, <, and > as needed.
 663  */
 664
 665 static char *                           /* O - Escaped string */
 666 xml_escape(const char *s)               /* I - String to escape */
 667 {
 668   char          *e,                     /* Escaped string */
 669                 *eptr;                  /* Pointer into escaped string */
 670   const char    *sptr;                  /* Pointer into string */
 671   size_t        bytes;                  /* Bytes needed for string */
 672
 673
 674  /*
 675   * First figure out how many extra bytes we need...
 676   */
 677
 678   for (bytes = 0, sptr = s; *sptr; sptr ++)
 679     if (*sptr == '&')
 680       bytes += 4;                       /* &amp; */
 681     else if (*sptr == '<' || *sptr == '>')
 682       bytes += 3;                       /* &lt; and &gt; */
 683
 684  /*
 685   * If there is nothing to escape, just strdup() it...
 686   */
 687
 688   if (bytes == 0)
 689     return (strdup(s));
 690
 691  /*
 692   * Otherwise allocate memory and copy...
 693   */
 694
 695   if ((e = malloc(bytes + 1 + strlen(s))) == NULL)
 696     return (NULL);
 697
 698   for (eptr = e, sptr = s; *sptr; sptr ++)
 699     if (*sptr == '&')
 700     {
 701       *eptr++ = '&';
 702       *eptr++ = 'a';
 703       *eptr++ = 'm';
 704       *eptr++ = 'p';
 705       *eptr++ = ';';
 706     }
 707     else if (*sptr == '<')
 708     {
 709       *eptr++ = '&';
 710       *eptr++ = 'l';
 711       *eptr++ = 't';
 712       *eptr++ = ';';
 713     }
 714     else if (*sptr == '>')
 715     {
 716       *eptr++ = '&';
 717       *eptr++ = 'g';
 718       *eptr++ = 't';
 719       *eptr++ = ';';
 720     }
 721     else
 722       *eptr++ = *sptr;
 723
 724   *eptr = '\0';
 725
 726   return (e);
 727 }
 728
 729
 730 /*
 731  * End of "$Id: rss.c 7824 2008-08-01 21:11:55Z mike $".
 732  */