]> git.ipfire.org Git - thirdparty/cups.git/blobdiff - backend/socket.c
Fix Linux build errors.
[thirdparty/cups.git] / backend / socket.c
index 3c3c086b17e0178e1b934f884c883e6962266268..579e1907742a61eebab6e7ebc015f16d6c9d1b5f 100644 (file)
@@ -1,47 +1,34 @@
 /*
- * "$Id: socket.c,v 1.39 2003/10/09 19:13:28 mike Exp $"
+ * "$Id$"
  *
  *   AppSocket backend for the Common UNIX Printing System (CUPS).
  *
- *   Copyright 1997-2003 by Easy Software Products, all rights reserved.
+ *   Copyright 2007-2010 by Apple Inc.
+ *   Copyright 1997-2007 by Easy Software Products, all rights reserved.
  *
  *   These coded instructions, statements, and computer programs are the
- *   property of Easy Software Products and are protected by Federal
- *   copyright law.  Distribution and use rights are outlined in the file
+ *   property of Apple Inc. and are protected by Federal copyright
+ *   law.  Distribution and use rights are outlined in the file "LICENSE.txt"
  *   "LICENSE" which should have been included with this file.  If this
- *   file is missing or damaged please contact Easy Software Products
- *   at:
- *
- *       Attn: CUPS Licensing Information
- *       Easy Software Products
- *       44141 Airport View Drive, Suite 204
- *       Hollywood, Maryland 20636-3111 USA
- *
- *       Voice: (301) 373-9603
- *       EMail: cups-info@cups.org
- *         WWW: http://www.cups.org
+ *   file is missing or damaged, see the license at "http://www.cups.org/".
  *
  *   This file is subject to the Apple OS-Developed Software exception.
  *
  * Contents:
  *
- *   main() - Send a file to the printer or server.
+ *   main()    - Send a file to the printer or server.
+ *   wait_bc() - Wait for back-channel data...
  */
 
 /*
  * Include necessary headers.
  */
 
-#include <cups/cups.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <stdarg.h>
 #include <cups/http-private.h>
-#include <cups/string.h>
-#include <errno.h>
+#include "backend-private.h"
+#include <stdarg.h>
 #include <sys/types.h>
 #include <sys/stat.h>
-#include <signal.h>
 
 #ifdef WIN32
 #  include <winsock.h>
@@ -59,7 +46,7 @@
  * Local functions...
  */
 
-void   print_backchannel(const unsigned char *buffer, int nbytes);
+static int     wait_bc(int device_fd, int secs);
 
 
 /*
@@ -70,31 +57,43 @@ void        print_backchannel(const unsigned char *buffer, int nbytes);
  *    printer-uri job-id user title copies options [file]
  */
 
-int                            /* O - Exit status */
-main(int  argc,                        /* I - Number of command-line arguments (6 or 7) */
-     char *argv[])             /* I - Command-line arguments */
+int                                    /* O - Exit status */
+main(int  argc,                                /* I - Number of command-line arguments (6 or 7) */
+     char *argv[])                     /* I - Command-line arguments */
 {
-  char         method[255],    /* Method in URI */
-               hostname[1024], /* Hostname */
-               username[255],  /* Username info (not used) */
-               resource[1024]; /* Resource info (not used) */
-  int          fp;             /* Print file */
-  int          copies;         /* Number of copies to print */
-  int          port;           /* Port number */
-  int          delay;          /* Delay for retries... */
-  int          fd;             /* AppSocket */
-  int          error;          /* Error code (if any) */
-  struct sockaddr_in addr;     /* Socket address */
-  struct hostent *hostaddr;    /* Host address */
-  int          wbytes;         /* Number of bytes written */
-  int          nbytes;         /* Number of bytes read */
-  size_t       tbytes;         /* Total number of bytes written */
-  char         buffer[8192],   /* Output buffer */
-               *bufptr;        /* Pointer into buffer */
-  struct timeval timeout;      /* Timeout for select() */
-  fd_set       input;          /* Input set for select() */
+  const char   *device_uri;            /* Device URI */
+  char         scheme[255],            /* Scheme in URI */
+               hostname[1024],         /* Hostname */
+               username[255],          /* Username info (not used) */
+               resource[1024],         /* Resource info (not used) */
+               *options,               /* Pointer to options */
+               *name,                  /* Name of option */
+               *value,                 /* Value of option */
+               sep;                    /* Option separator */
+  int          print_fd;               /* Print file */
+  int          copies;                 /* Number of copies to print */
+  time_t       start_time;             /* Time of first connect */
+#ifdef __APPLE__
+  time_t       current_time,           /* Current time */
+               wait_time;              /* Time to wait before shutting down socket */
+#endif /* __APPLE__ */
+  int          contimeout;             /* Connection timeout */
+  int          waiteof;                /* Wait for end-of-file? */
+  int          port;                   /* Port number */
+  char         portname[255];          /* Port name */
+  int          delay;                  /* Delay for retries... */
+  int          device_fd;              /* AppSocket */
+  int          error;                  /* Error code (if any) */
+  http_addrlist_t *addrlist,           /* Address list */
+               *addr;                  /* Connected address */
+  char         addrname[256];          /* Address name */
+  int          snmp_fd,                /* SNMP socket */
+               start_count,            /* Page count via SNMP at start */
+               page_count,             /* Page count via SNMP */
+               have_supplies;          /* Printer supports supply levels? */
+  ssize_t      tbytes;                 /* Total number of bytes written */
 #if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
-  struct sigaction action;     /* Actions for POSIX signals */
+  struct sigaction action;             /* Actions for POSIX signals */
 #endif /* HAVE_SIGACTION && !HAVE_SIGSET */
 
 
@@ -124,14 +123,16 @@ main(int  argc,                   /* I - Number of command-line arguments (6 or 7) */
 
   if (argc == 1)
   {
-    puts("network socket \"Unknown\" \"AppSocket/HP JetDirect\"");
-    return (0);
+    printf("network socket \"Unknown\" \"%s\"\n",
+           _cupsLangString(cupsLangDefault(), _("AppSocket/HP JetDirect")));
+    return (CUPS_BACKEND_OK);
   }
   else if (argc < 6 || argc > 7)
   {
-    fprintf(stderr, "Usage: %s job-id user title copies options [file]\n",
-            argv[0]);
-    return (1);
+    _cupsLangPrintf(stderr,
+                    _("Usage: %s job-id user title copies options [file]\n"),
+                    argv[0]);
+    return (CUPS_BACKEND_FAILED);
   }
 
  /*
@@ -141,8 +142,8 @@ main(int  argc,                     /* I - Number of command-line arguments (6 or 7) */
 
   if (argc == 6)
   {
-    fp     = 0;
-    copies = 1;
+    print_fd = 0;
+    copies   = 1;
   }
   else
   {
@@ -150,10 +151,12 @@ main(int  argc,                   /* I - Number of command-line arguments (6 or 7) */
     * Try to open the print file...
     */
 
-    if ((fp = open(argv[6], O_RDONLY)) < 0)
+    if ((print_fd = open(argv[6], O_RDONLY)) < 0)
     {
-      perror("ERROR: unable to open print file");
-      return (1);
+      _cupsLangPrintf(stderr,
+                      _("ERROR: Unable to open print file \"%s\": %s\n"),
+                      argv[6], strerror(errno));
+      return (CUPS_BACKEND_FAILED);
     }
 
     copies = atoi(argv[4]);
@@ -163,256 +166,330 @@ main(int  argc,                 /* I - Number of command-line arguments (6 or 7) */
   * Extract the hostname and port number from the URI...
   */
 
-  httpSeparate(argv[0], method, username, hostname, &port, resource);
+  if ((device_uri = cupsBackendDeviceURI(argv)) == NULL)
+    return (CUPS_BACKEND_FAILED);
+
+  httpSeparateURI(HTTP_URI_CODING_ALL, device_uri, scheme, sizeof(scheme),
+                  username, sizeof(username), hostname, sizeof(hostname), &port,
+                 resource, sizeof(resource));
 
   if (port == 0)
     port = 9100;       /* Default to HP JetDirect/Tektronix PhaserShare */
 
  /*
-  * Then try to connect to the remote host...
+  * Get options, if any...
   */
 
-  if ((hostaddr = httpGetHostByName(hostname)) == NULL)
-  {
-    fprintf(stderr, "ERROR: Unable to locate printer \'%s\' - %s\n",
-            hostname, hstrerror(h_errno));
-    return (1);
-  }
-
-  fprintf(stderr, "INFO: Attempting to connect to host %s on port %d\n",
-          hostname, port);
+  waiteof    = 1;
+  contimeout = 7 * 24 * 60 * 60;
 
-  memset(&addr, 0, sizeof(addr));
-  memcpy(&(addr.sin_addr), hostaddr->h_addr, hostaddr->h_length);
-  addr.sin_family = hostaddr->h_addrtype;
-  addr.sin_port   = htons(port);
-
-  while (copies > 0)
+  if ((options = strchr(resource, '?')) != NULL)
   {
-    for (delay = 5;;)
-    {
-      if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
-      {
-       perror("ERROR: Unable to create socket");
-       return (1);
-      }
-
-      if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
-      {
-       error = errno;
-       close(fd);
-       fd = -1;
-
-       if (error == ECONNREFUSED || error == EHOSTDOWN ||
-            error == EHOSTUNREACH)
-       {
-         fprintf(stderr, "INFO: Network host \'%s\' is busy; will retry in %d seconds...\n",
-                  hostname, delay);
-         sleep(delay);
-
-         if (delay < 30)
-           delay += 5;
-       }
-       else
-       {
-         perror("ERROR: Unable to connect to printer (retrying in 30 seconds)");
-         sleep(30);
-       }
-      }
-      else
-       break;
-    }
-
    /*
-    * Now that we are "connected" to the port, ignore SIGTERM so that we
-    * can finish out any page data the driver sends (e.g. to eject the
-    * current page...  Only ignore SIGTERM if we are printing data from
-    * stdin (otherwise you can't cancel raw jobs...)
+    * Yup, terminate the device name string and move to the first
+    * character of the options...
     */
 
-    if (argc < 7)
-    {
-#ifdef HAVE_SIGSET /* Use System V signals over POSIX to avoid bugs */
-      sigset(SIGTERM, SIG_IGN);
-#elif defined(HAVE_SIGACTION)
-      memset(&action, 0, sizeof(action));
-
-      sigemptyset(&action.sa_mask);
-      action.sa_handler = SIG_IGN;
-      sigaction(SIGTERM, &action, NULL);
-#else
-      signal(SIGTERM, SIG_IGN);
-#endif /* HAVE_SIGSET */
-    }
+    *options++ = '\0';
 
    /*
-    * Finally, send the print file...
+    * Parse options...
     */
 
-    copies --;
-
-    if (fp != 0)
-    {
-      fputs("PAGE: 1 1\n", stderr);
-      lseek(fp, 0, SEEK_SET);
-    }
-
-    fputs("INFO: Connected to host, sending print job...\n", stderr);
-
-    tbytes = 0;
-    while ((nbytes = read(fp, buffer, sizeof(buffer))) > 0)
+    while (*options)
     {
      /*
-      * Write the print data to the printer...
+      * Get the name...
       */
 
-      tbytes += nbytes;
-      bufptr = buffer;
+      name = options;
+
+      while (*options && *options != '=' && *options != '+' && *options != '&')
+        options ++;
+
+      if ((sep = *options) != '\0')
+        *options++ = '\0';
 
-      while (nbytes > 0)
+      if (sep == '=')
       {
-       if ((wbytes = send(fd, bufptr, nbytes, 0)) < 0)
-       {
-         perror("ERROR: Unable to send print file to printer");
-         break;
-       }
+       /*
+        * Get the value...
+       */
 
-       nbytes -= wbytes;
-       bufptr += wbytes;
-      }
+        value = options;
 
-      if (wbytes < 0)
-        break;
+       while (*options && *options != '+' && *options != '&')
+         options ++;
+
+        if (*options)
+         *options++ = '\0';
+      }
+      else
+        value = (char *)"";
 
      /*
-      * Check for possible data coming back from the printer...
+      * Process the option...
       */
 
-      timeout.tv_sec  = 0;
-      timeout.tv_usec = 0;
+      if (!strcasecmp(name, "waiteof"))
+      {
+       /*
+        * Set the wait-for-eof value...
+       */
 
-      FD_ZERO(&input);
-      FD_SET(fd, &input);
-      if (select(fd + 1, &input, NULL, NULL, &timeout) > 0)
+        waiteof = !value[0] || !strcasecmp(value, "on") ||
+                 !strcasecmp(value, "yes") || !strcasecmp(value, "true");
+      }
+      else if (!strcasecmp(name, "contimeout"))
       {
        /*
-       * Grab the data coming back and spit it out to stderr...
+        * Set the connection timeout...
        */
 
-       if ((nbytes = recv(fd, buffer, sizeof(buffer), 0)) > 0)
-       {
-         fprintf(stderr, "INFO: Received %d bytes of back-channel data!\n",
-                 nbytes);
-          print_backchannel((unsigned char *)buffer, nbytes);
-        }
+       if (atoi(value) > 0)
+         contimeout = atoi(value);
       }
-      else if (argc > 6)
-       fprintf(stderr, "INFO: Sending print file, %lu bytes...\n",
-               (unsigned long)tbytes);
     }
+  }
 
  /*
-    * Shutdown the socket and wait for the other end to finish...
-    */
+ /*
+  * Then try to connect to the remote host...
+  */
 
-    fputs("INFO: Print file sent, waiting for printer to finish...\n", stderr);
+  start_time = time(NULL);
 
-    shutdown(fd, 1);
+  sprintf(portname, "%d", port);
 
-    for (;;)
-    {
-     /*
-      * Wait a maximum of 90 seconds for backchannel data or a closed
-      * connection...
-      */
+  fputs("STATE: +connecting-to-device\n", stderr);
+  fprintf(stderr, "DEBUG: Looking up \"%s\"...\n", hostname);
 
-      timeout.tv_sec  = 90;
-      timeout.tv_usec = 0;
+  if ((addrlist = httpAddrGetList(hostname, AF_UNSPEC, portname)) == NULL)
+  {
+    _cupsLangPrintf(stderr, _("ERROR: Unable to locate printer \'%s\'\n"),
+                    hostname);
+    return (CUPS_BACKEND_STOP);
+  }
 
-      FD_ZERO(&input);
-      FD_SET(fd, &input);
+  fprintf(stderr, "DEBUG: Connecting to %s:%d\n", hostname, port);
+  _cupsLangPuts(stderr, _("INFO: Connecting to printer...\n"));
 
-#ifdef __hpux
-      if (select(fd + 1, (int *)&input, NULL, NULL, &timeout) > 0)
-#else
-      if (select(fd + 1, &input, NULL, NULL, &timeout) > 0)
-#endif /* __hpux */
+  for (delay = 5;;)
+  {
+    if ((addr = httpAddrConnect(addrlist, &device_fd)) == NULL)
+    {
+      error     = errno;
+      device_fd = -1;
+
+      if (getenv("CLASS") != NULL)
       {
        /*
-       * Grab the data coming back and spit it out to stderr...
+        * If the CLASS environment variable is set, the job was submitted
+       * to a class and not to a specific queue.  In this case, we want
+       * to abort immediately so that the job can be requeued on the next
+       * available printer in the class.
+       */
+
+        _cupsLangPuts(stderr,
+                     _("INFO: Unable to contact printer, queuing on next "
+                       "printer in class...\n"));
+
+       /*
+        * Sleep 5 seconds to keep the job from requeuing too rapidly...
        */
 
-       if ((nbytes = recv(fd, buffer, sizeof(buffer), 0)) > 0)
+       sleep(5);
+
+        return (CUPS_BACKEND_FAILED);
+      }
+
+      if (error == ECONNREFUSED || error == EHOSTDOWN ||
+          error == EHOSTUNREACH)
+      {
+        if (contimeout && (time(NULL) - start_time) > contimeout)
        {
-         fprintf(stderr, "INFO: Received %d bytes of back-channel data!\n",
-                 nbytes);
-          print_backchannel((unsigned char *)buffer, nbytes);
-        }
-       else
-         break;
+         _cupsLangPuts(stderr, _("ERROR: Printer not responding\n"));
+         return (CUPS_BACKEND_FAILED);
+       }
+
+       _cupsLangPrintf(stderr,
+                       _("WARNING: Network host \'%s\' is busy; will retry in "
+                         "%d seconds...\n"), hostname, delay);
+
+       sleep(delay);
+
+       if (delay < 30)
+         delay += 5;
       }
       else
-        break;
+      {
+        _cupsLangPrintf(stderr, "DEBUG: Connection error: %s\n",
+                       strerror(errno));
+       _cupsLangPuts(stderr,
+                     _("ERROR: Unable to connect to printer; will retry in 30 "
+                       "seconds...\n"));
+       sleep(30);
+      }
+    }
+    else
+      break;
+  }
+
+  fputs("STATE: -connecting-to-device\n", stderr);
+  _cupsLangPuts(stderr, _("INFO: Connected to printer...\n"));
+
+#ifdef AF_INET6
+  if (addr->addr.addr.sa_family == AF_INET6)
+    fprintf(stderr, "DEBUG: Connected to [%s]:%d (IPv6)...\n", 
+           httpAddrString(&addr->addr, addrname, sizeof(addrname)),
+           ntohs(addr->addr.ipv6.sin6_port));
+  else
+#endif /* AF_INET6 */
+    if (addr->addr.addr.sa_family == AF_INET)
+      fprintf(stderr, "DEBUG: Connected to %s:%d (IPv4)...\n",
+             httpAddrString(&addr->addr, addrname, sizeof(addrname)),
+             ntohs(addr->addr.ipv4.sin_port));
+
+ /*
+  * See if the printer supports SNMP...
+  */
+
+  if ((snmp_fd = _cupsSNMPOpen(addr->addr.addr.sa_family)) >= 0)
+  {
+    have_supplies = !backendSNMPSupplies(snmp_fd, &(addr->addr), &start_count,
+                                         NULL);
+  }
+  else
+    have_supplies = start_count = 0;
+
+ /*
+  * Print everything...
+  */
+
+  tbytes = 0;
+
+  while (copies > 0 && tbytes >= 0)
+  {
+    copies --;
+
+    if (print_fd != 0)
+    {
+      fputs("PAGE: 1 1\n", stderr);
+      lseek(print_fd, 0, SEEK_SET);
     }
 
+    tbytes = backendRunLoop(print_fd, device_fd, snmp_fd, &(addr->addr), 1, 0, 
+                            backendNetworkSideCB);
+
+    if (print_fd != 0 && tbytes >= 0)
+      _cupsLangPrintf(stderr,
+#ifdef HAVE_LONG_LONG
+                     _("INFO: Sent print file, %lld bytes...\n"),
+#else
+                     _("INFO: Sent print file, %ld bytes...\n"),
+#endif /* HAVE_LONG_LONG */
+                     CUPS_LLCAST tbytes);
+  }
+
+#ifdef __APPLE__
+ /*
+  * Wait up to 5 seconds to get any pending back-channel data...
+  */
+
+  wait_time = time(NULL) + 5;
+  while (wait_time >= time(&current_time))
+    if (wait_bc(device_fd, wait_time - current_time) <= 0)
+      break;
+#endif /* __APPLE__ */
+
+  if (waiteof)
+  {
    /*
-    * Close the socket connection...
+    * Shutdown the socket and wait for the other end to finish...
     */
 
-    close(fd);
+    _cupsLangPuts(stderr,
+                  _("INFO: Print file sent, waiting for printer to finish...\n"));
+
+    shutdown(device_fd, 1);
+
+    while (wait_bc(device_fd, 90) > 0);
   }
 
+ /*
+  * Collect the final page count as needed...
+  */
+
+  if (have_supplies && 
+      !backendSNMPSupplies(snmp_fd, &(addr->addr), &page_count, NULL) &&
+      page_count > start_count)
+    fprintf(stderr, "PAGE: total %d\n", page_count - start_count);
+
+ /*
+  * Close the socket connection...
+  */
+
+  close(device_fd);
+
+  httpAddrFreeList(addrlist);
+
  /*
   * Close the input file and return...
   */
 
-  if (fp != 0)
-    close(fp);
+  if (print_fd != 0)
+    close(print_fd);
 
-  fputs("INFO: Ready to print.\n", stderr);
+  if (tbytes >= 0)
+    _cupsLangPuts(stderr, _("INFO: Ready to print.\n"));
 
-  return (0);
+  return (tbytes < 0 ? CUPS_BACKEND_FAILED : CUPS_BACKEND_OK);
 }
 
 
 /*
- * 'print_backchannel()' - Print the contents of a back-channel buffer.
+ * 'wait_bc()' - Wait for back-channel data...
  */
 
-void
-print_backchannel(const unsigned char *buffer, /* I - Data buffer */
-                  int                 nbytes)  /* I - Number of bytes */
+static int                             /* O - # bytes read or -1 on error */
+wait_bc(int device_fd,                 /* I - Socket */
+        int secs)                      /* I - Seconds to wait */
 {
-  char line[255],                              /* Formatted line */
-       *lineptr;                               /* Pointer into line */
+  struct timeval timeout;              /* Timeout for select() */
+  fd_set       input;                  /* Input set for select() */
+  ssize_t      bytes;                  /* Number of back-channel bytes read */
+  char         buffer[1024];           /* Back-channel buffer */
+
+
+ /*
+  * Wait up to "secs" seconds for backchannel data...
+  */
+
+  timeout.tv_sec  = secs;
+  timeout.tv_usec = 0;
 
+  FD_ZERO(&input);
+  FD_SET(device_fd, &input);
 
-  for (lineptr = line; nbytes > 0; buffer ++, nbytes --)
+  if (select(device_fd + 1, &input, NULL, NULL, &timeout) > 0)
   {
-    if (*buffer < 0x20 || *buffer >= 0x7f)
-    {
-      snprintf(lineptr, sizeof(line) - (lineptr - line), "<%02X>", *buffer);
-      lineptr += strlen(lineptr);
-    }
-    else
-      *lineptr++ = *buffer;
+   /*
+    * Grab the data coming back and spit it out to stderr...
+    */
 
-    if ((lineptr - line) > 72)
+    if ((bytes = read(device_fd, buffer, sizeof(buffer))) > 0)
     {
-      *lineptr = '\0';
-      fprintf(stderr, "DEBUG: DATA: %s\n", line);
-      lineptr = line;
+      fprintf(stderr, "DEBUG: Received %d bytes of back-channel data\n",
+             (int)bytes);
+      cupsBackChannelWrite(buffer, bytes, 1.0);
     }
-  }
 
-  if (lineptr > line)
-  {
-    *lineptr = '\0';
-    fprintf(stderr, "DEBUG: DATA: %s\n", line);
+    return (bytes);
   }
+  else
+    return (-1);
 }
 
 
 /*
- * End of "$Id: socket.c,v 1.39 2003/10/09 19:13:28 mike Exp $".
+ * End of "$Id$".
  */