backend/socket.c

   1 /*
   2  * "$Id: socket.c,v 1.24 2001/10/30 20:37:13 mike Exp $"
   3  *
   4  *   AppSocket backend for the Common UNIX Printing System (CUPS).
   5  *
   6  *   Copyright 1997-2001 by Easy Software Products, all rights reserved.
   7  *
   8  *   These coded instructions, statements, and computer programs are the
   9  *   property of Easy Software Products and are protected by Federal
  10  *   copyright law.  Distribution and use rights are outlined in the file
  11  *   "LICENSE" which should have been included with this file.  If this
  12  *   file is missing or damaged please contact Easy Software Products
  13  *   at:
  14  *
  15  *       Attn: CUPS Licensing Information
  16  *       Easy Software Products
  17  *       44141 Airport View Drive, Suite 204
  18  *       Hollywood, Maryland 20636-3111 USA
  19  *
  20  *       Voice: (301) 373-9603
  21  *       EMail: cups-info@cups.org
  22  *         WWW: http://www.cups.org
  23  *
  24  * Contents:
  25  *
  26  *   main() - Send a file to the printer or server.
  27  */
  28
  29 /*
  30  * Include necessary headers.
  31  */
  32
  33 #include <cups/cups.h>
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <stdarg.h>
  37 #include <cups/string.h>
  38 #include <errno.h>
  39 #include <sys/types.h>
  40 #include <sys/stat.h>
  41 #include <signal.h>
  42
  43 #if defined(WIN32) || defined(__EMX__)
  44 #  include <winsock.h>
  45 #else
  46 #  include <sys/socket.h>
  47 #  include <netinet/in.h>
  48 #  include <arpa/inet.h>
  49 #  include <netdb.h>
  50 #endif /* WIN32 || __EMX__ */
  51
  52
  53 /*
  54  * 'main()' - Send a file to the printer or server.
  55  *
  56  * Usage:
  57  *
  58  *    printer-uri job-id user title copies options [file]
  59  */
  60
  61 int                     /* O - Exit status */
  62 main(int  argc,         /* I - Number of command-line arguments (6 or 7) */
  63      char *argv[])      /* I - Command-line arguments */
  64 {
  65   char          method[255],    /* Method in URI */
  66                 hostname[1024], /* Hostname */
  67                 username[255],  /* Username info (not used) */
  68                 resource[1024]; /* Resource info (not used) */
  69   FILE          *fp;            /* Print file */
  70   int           copies;         /* Number of copies to print */
  71   int           port;           /* Port number */
  72   int           delay;          /* Delay for retries... */
  73   int           fd;             /* AppSocket */
  74   int           error;          /* Error code (if any) */
  75   struct sockaddr_in addr;      /* Socket address */
  76   struct hostent *hostaddr;     /* Host address */
  77   int           wbytes;         /* Number of bytes written */
  78   size_t        nbytes,         /* Number of bytes read */
  79                 tbytes;         /* Total number of bytes written */
  80   char          buffer[8192],   /* Output buffer */
  81                 *bufptr;        /* Pointer into buffer */
  82   struct timeval timeout;       /* Timeout for select() */
  83   fd_set        input;          /* Input set for select() */
  84 #if defined(HAVE_SIGACTION) && !defined(HAVE_SIGSET)
  85   struct sigaction action;      /* Actions for POSIX signals */
  86 #endif /* HAVE_SIGACTION && !HAVE_SIGSET */
  87
  88
  89  /*
  90   * Make sure status messages are not buffered...
  91   */
  92
  93   setbuf(stderr, NULL);
  94
  95  /*
  96   * Check command-line...
  97   */
  98
  99   if (argc == 1)
 100   {
 101     puts("network socket \"Unknown\" \"AppSocket/HP JetDirect\"");
 102     return (0);
 103   }
 104   else if (argc < 6 || argc > 7)
 105   {
 106     fprintf(stderr, "Usage: %s job-id user title copies options [file]\n",
 107             argv[0]);
 108     return (1);
 109   }
 110
 111  /*
 112   * If we have 7 arguments, print the file named on the command-line.
 113   * Otherwise, send stdin instead...
 114   */
 115
 116   if (argc == 6)
 117   {
 118     fp     = stdin;
 119     copies = 1;
 120   }
 121   else
 122   {
 123    /*
 124     * Try to open the print file...
 125     */
 126
 127     if ((fp = fopen(argv[6], "rb")) == NULL)
 128     {
 129       perror("ERROR: unable to open print file");
 130       return (1);
 131     }
 132
 133     copies = atoi(argv[4]);
 134   }
 135
 136  /*
 137   * Extract the hostname and port number from the URI...
 138   */
 139
 140   httpSeparate(argv[0], method, username, hostname, &port, resource);
 141
 142   if (port == 0)
 143     port = 9100;        /* Default to HP JetDirect/Tektronix PhaserShare */
 144
 145  /*
 146   * Then try to connect to the remote host...
 147   */
 148
 149   if ((hostaddr = httpGetHostByName(hostname)) == NULL)
 150   {
 151     fprintf(stderr, "ERROR: Unable to locate printer \'%s\' - %s\n",
 152             hostname, strerror(errno));
 153     return (1);
 154   }
 155
 156   fprintf(stderr, "INFO: Attempting to connect to host %s on port %d\n",
 157           hostname, port);
 158
 159   memset(&addr, 0, sizeof(addr));
 160   memcpy(&(addr.sin_addr), hostaddr->h_addr, hostaddr->h_length);
 161   addr.sin_family = hostaddr->h_addrtype;
 162   addr.sin_port   = htons(port);
 163
 164   while (copies > 0)
 165   {
 166     for (delay = 5;;)
 167     {
 168       if ((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
 169       {
 170         perror("ERROR: Unable to create socket");
 171         return (1);
 172       }
 173
 174       if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) < 0)
 175       {
 176         error = errno;
 177         close(fd);
 178         fd = -1;
 179
 180         if (error == ECONNREFUSED)
 181         {
 182           fprintf(stderr, "INFO: Network host \'%s\' is busy; will retry in %d seconds...\n",
 183                   hostname, delay);
 184           sleep(delay);
 185
 186           if (delay < 30)
 187             delay += 5;
 188         }
 189         else
 190         {
 191           perror("ERROR: Unable to connect to printer (retrying in 30 seconds)");
 192           sleep(30);
 193         }
 194       }
 195       else
 196         break;
 197     }
 198
 199    /*
 200     * Now that we are "connected" to the port, ignore SIGTERM so that we
 201     * can finish out any page data the driver sends (e.g. to eject the
 202     * current page...  Only ignore SIGTERM if we are printing data from
 203     * stdin (otherwise you can't cancel raw jobs...)
 204     */
 205
 206     if (argc < 7)
 207     {
 208 #ifdef HAVE_SIGSET /* Use System V signals over POSIX to avoid bugs */
 209       sigset(SIGTERM, SIG_IGN);
 210 #elif defined(HAVE_SIGACTION)
 211       memset(&action, 0, sizeof(action));
 212
 213       sigemptyset(&action.sa_mask);
 214       action.sa_handler = SIG_IGN;
 215       sigaction(SIGTERM, &action, NULL);
 216 #else
 217       signal(SIGTERM, SIG_IGN);
 218 #endif /* HAVE_SIGSET */
 219     }
 220
 221    /*
 222     * Finally, send the print file...
 223     */
 224
 225     copies --;
 226
 227     if (fp != stdin)
 228     {
 229       fputs("PAGE: 1 1\n", stderr);
 230       rewind(fp);
 231     }
 232
 233     fputs("INFO: Connected to host, sending print job...\n", stderr);
 234
 235     tbytes = 0;
 236     while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0)
 237     {
 238      /*
 239       * Write the print data to the printer...
 240       */
 241
 242       tbytes += nbytes;
 243       bufptr = buffer;
 244
 245       while (nbytes > 0)
 246       {
 247         if ((wbytes = send(fd, bufptr, nbytes, 0)) < 0)
 248         {
 249           perror("ERROR: Unable to send print file to printer");
 250           break;
 251         }
 252
 253         nbytes -= wbytes;
 254         bufptr += wbytes;
 255       }
 256
 257      /*
 258       * Check for possible data coming back from the printer...
 259       */
 260
 261       timeout.tv_sec  = 0;
 262       timeout.tv_usec = 0;
 263
 264       FD_ZERO(&input);
 265       FD_SET(fd, &input);
 266 #ifdef __hpux
 267       if (select(fd + 1, (int *)&input, NULL, NULL, &timeout) > 0)
 268 #else
 269       if (select(fd + 1, &input, NULL, NULL, &timeout) > 0)
 270 #endif /* __hpux */
 271       {
 272        /*
 273         * Grab the data coming back and spit it out to stderr...
 274         */
 275
 276         if ((nbytes = recv(fd, buffer, sizeof(buffer), 0)) > 0)
 277           fprintf(stderr, "INFO: Received %u bytes of back-channel data!\n",
 278                   nbytes);
 279       }
 280       else if (argc > 6)
 281         fprintf(stderr, "INFO: Sending print file, %u bytes...\n", tbytes);
 282     }
 283
 284    /*
 285     * Shutdown the socket and wait for the other end to finish...
 286     */
 287
 288     fputs("INFO: Print file sent, waiting for printer to finish...\n", stderr);
 289
 290     shutdown(fd, 1);
 291
 292     for (;;)
 293     {
 294      /*
 295       * Wait a maximum of 90 seconds for backchannel data or a closed
 296       * connection...
 297       */
 298
 299       timeout.tv_sec  = 90;
 300       timeout.tv_usec = 0;
 301
 302       FD_ZERO(&input);
 303       FD_SET(fd, &input);
 304
 305 #ifdef __hpux
 306       if (select(fd + 1, (int *)&input, NULL, NULL, &timeout) > 0)
 307 #else
 308       if (select(fd + 1, &input, NULL, NULL, &timeout) > 0)
 309 #endif /* __hpux */
 310       {
 311        /*
 312         * Grab the data coming back and spit it out to stderr...
 313         */
 314
 315         if ((nbytes = recv(fd, buffer, sizeof(buffer), 0)) > 0)
 316           fprintf(stderr, "INFO: Received %u bytes of back-channel data!\n",
 317                   nbytes);
 318         else
 319           break;
 320       }
 321       else
 322         break;
 323     }
 324
 325    /*
 326     * Close the socket connection...
 327     */
 328
 329     close(fd);
 330   }
 331
 332  /*
 333   * Close the input file and return...
 334   */
 335
 336   if (fp != stdin)
 337     fclose(fp);
 338
 339   fputs("INFO: Ready to print.\n", stderr);
 340
 341   return (0);
 342 }
 343
 344
 345 /*
 346  * End of "$Id: socket.c,v 1.24 2001/10/30 20:37:13 mike Exp $".
 347  */