source/client/smbspool.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    Version 2.0.
   4    SMB backend for the Common UNIX Printing System ("CUPS")
   5    Copyright 1999 by Easy Software Products
   6    Copyright Andrew Tridgell 1994-1998
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  21 */
  22
  23 #define NO_SYSLOG
  24
  25 #include "includes.h"
  26
  27 /*
  28  * Globals...
  29  */
  30
  31 extern BOOL             in_client;      /* Boolean for client library */
  32 extern struct in_addr   ipzero;         /* Any address */
  33
  34
  35 /*
  36  * Local functions...
  37  */
  38
  39 static void             list_devices(void);
  40 static struct cli_state *smb_connect(char *, char *, char *, char *, char *);
  41 static int              smb_print(struct cli_state *, char *, FILE *);
  42
  43
  44 /*
  45  * 'main()' - Main entry for SMB backend.
  46  */
  47
  48  int                            /* O - Exit status */
  49  main(int  argc,                        /* I - Number of command-line arguments */
  50      char *argv[])              /* I - Command-line arguments */
  51 {
  52   int           i;              /* Looping var */
  53   int           copies;         /* Number of copies */
  54   char          uri[1024],      /* URI */
  55                 *sep,           /* Pointer to separator */
  56                 *username,      /* Username */
  57                 *password,      /* Password */
  58                 *workgroup,     /* Workgroup */
  59                 *server,        /* Server name */
  60                 *printer;       /* Printer name */
  61   FILE          *fp;            /* File to print */
  62   int           status=0;               /* Status of LPD job */
  63   struct cli_state *cli;        /* SMB interface */
  64
  65   /* we expect the URI in argv[0]. Detect the case where it is in argv[1] and cope */
  66   if (argc > 2 && strncmp(argv[0],"smb://", 6) && !strncmp(argv[1],"smb://", 6)) {
  67           argv++;
  68           argc--;
  69   }
  70
  71   if (argc == 1)
  72   {
  73    /*
  74     * NEW!  In CUPS 1.1 the backends are run with no arguments to list the
  75     *       available devices.  These can be devices served by this backend
  76     *       or any other backends (i.e. you can have an SNMP backend that
  77     *       is only used to enumerate the available network printers... :)
  78     */
  79
  80     list_devices();
  81     return (0);
  82   }
  83
  84   if (argc < 6 || argc > 7)
  85   {
  86     fprintf(stderr, "Usage: %s [DEVICE_URI] job-id user title copies options [file]\n",
  87             argv[0]);
  88     fputs("       The DEVICE_URI environment variable can also contain the\n", stderr);
  89     fputs("       destination printer:\n", stderr);
  90     fputs("\n", stderr);
  91     fputs("           smb://[username:password@][workgroup/]server/printer\n", stderr);
  92     return (1);
  93   }
  94
  95  /*
  96   * If we have 7 arguments, print the file named on the command-line.
  97   * Otherwise, print data from stdin...
  98   */
  99
 100   if (argc == 6)
 101   {
 102    /*
 103     * Print from Copy stdin to a temporary file...
 104     */
 105
 106     fp     = stdin;
 107     copies = 1;
 108   }
 109   else if ((fp = fopen(argv[6], "rb")) == NULL)
 110   {
 111     perror("ERROR: Unable to open print file");
 112     return (1);
 113   }
 114   else
 115     copies = atoi(argv[4]);
 116
 117  /*
 118   * Find the URI...
 119   */
 120
 121   if (strncmp(argv[0], "smb://", 6) == 0)
 122     strncpy(uri, argv[0], sizeof(uri) - 1);
 123   else if (getenv("DEVICE_URI") != NULL)
 124     strncpy(uri, getenv("DEVICE_URI"), sizeof(uri) - 1);
 125   else
 126   {
 127     fputs("ERROR: No device URI found in argv[0] or DEVICE_URI environment variable!\n", stderr);
 128     return (1);
 129   }
 130
 131   uri[sizeof(uri) - 1] = '\0';
 132
 133  /*
 134   * Extract the destination from the URI...
 135   */
 136
 137   if ((sep = strrchr(uri, '@')) != NULL)
 138   {
 139     username = uri + 6;
 140     *sep++ = '\0';
 141
 142     server = sep;
 143
 144    /*
 145     * Extract password as needed...
 146     */
 147
 148     if ((password = strchr(username, ':')) != NULL)
 149       *password++ = '\0';
 150     else
 151       password = "";
 152   }
 153   else
 154   {
 155     username = "";
 156     password = "";
 157     server   = uri + 6;
 158   }
 159
 160   if ((sep = strchr(server, '/')) == NULL)
 161   {
 162     fputs("ERROR: Bad URI - need printer name!\n", stderr);
 163     return (1);
 164   }
 165
 166   *sep++ = '\0';
 167   printer = sep;
 168
 169   if ((sep = strchr(printer, '/')) != NULL)
 170   {
 171    /*
 172     * Convert to smb://[username:password@]workgroup/server/printer...
 173     */
 174
 175     *sep++ = '\0';
 176
 177     workgroup = server;
 178     server    = printer;
 179     printer   = sep;
 180   }
 181   else
 182     workgroup = NULL;
 183
 184  /*
 185   * Setup the SAMBA server state...
 186   */
 187
 188   setup_logging("smbspool", True);
 189
 190   TimeInit();
 191   charset_initialise();
 192
 193   in_client = True;   /* Make sure that we tell lp_load we are */
 194
 195   if (!lp_load(CONFIGFILE, True, False, False))
 196   {
 197     fprintf(stderr, "ERROR: Can't load %s - run testparm to debug it\n", CONFIGFILE);
 198     return (1);
 199   }
 200
 201   if (workgroup == NULL)
 202     workgroup = lp_workgroup();
 203
 204   codepage_initialise(lp_client_code_page());
 205
 206   load_interfaces();
 207
 208   do
 209   {
 210     if ((cli = smb_connect(workgroup, server, printer, username, password)) == NULL)
 211     {
 212       if (getenv("CLASS") == NULL)
 213       {
 214         perror("ERROR: Unable to connect to SAMBA host, will retry in 60 seconds...");
 215         sleep (60);
 216       }
 217       else
 218       {
 219         perror("ERROR: Unable to connect to SAMBA host, trying next printer...");
 220         return (1);
 221       }
 222     }
 223   }
 224   while (cli == NULL);
 225
 226  /*
 227   * Now that we are connected to the server, ignore SIGTERM so that we
 228   * can finish out any page data the driver sends (e.g. to eject the
 229   * current page...  Only ignore SIGTERM if we are printing data from
 230   * stdin (otherwise you can't cancel raw jobs...)
 231   */
 232
 233   if (argc < 7)
 234     CatchSignal(SIGTERM, SIG_IGN);
 235
 236  /*
 237   * Queue the job...
 238   */
 239
 240   for (i = 0; i < copies; i ++)
 241     if ((status = smb_print(cli, argv[3] /* title */, fp)) != 0)
 242       break;
 243
 244   cli_shutdown(cli);
 245
 246  /*
 247   * Return the queue status...
 248   */
 249
 250   return (status);
 251 }
 252
 253
 254 /*
 255  * 'list_devices()' - List the available printers seen on the network...
 256  */
 257
 258 static void
 259 list_devices(void)
 260 {
 261  /*
 262   * Eventually, search the local workgroup for available hosts and printers.
 263   */
 264
 265   puts("network smb \"Unknown\" \"Windows Printer via SAMBA\"");
 266 }
 267
 268
 269 /*
 270  * 'smb_connect()' - Return a connection to a server.
 271  */
 272
 273 static struct cli_state *               /* O - SMB connection */
 274 smb_connect(char *workgroup,            /* I - Workgroup */
 275             char *server,               /* I - Server */
 276             char *share,                /* I - Printer */
 277             char *username,             /* I - Username */
 278             char *password)             /* I - Password */
 279 {
 280   struct cli_state      *c;             /* New connection */
 281   struct nmb_name       called,         /* NMB name of server */
 282                         calling;        /* NMB name of client */
 283   struct in_addr        ip;             /* IP address of server */
 284   pstring               myname;         /* Client name */
 285
 286
 287  /*
 288   * Get the names and addresses of the client and server...
 289   */
 290
 291   get_myname(myname);
 292
 293   ip = ipzero;
 294
 295   make_nmb_name(&calling, myname, 0x0);
 296   make_nmb_name(&called, server, 0x20);
 297
 298  /*
 299   * Open a new connection to the SMB server...
 300   */
 301
 302   if ((c = cli_initialise(NULL)) == NULL)
 303   {
 304     fputs("ERROR: cli_initialise() failed...\n", stderr);
 305     return (NULL);
 306   }
 307
 308   if (!cli_set_port(c, SMB_PORT))
 309   {
 310     fputs("ERROR: cli_set_port() failed...\n", stderr);
 311     return (NULL);
 312   }
 313
 314   if (!cli_connect(c, server, &ip))
 315   {
 316     fputs("ERROR: cli_connect() failed...\n", stderr);
 317     return (NULL);
 318   }
 319
 320   if (!cli_session_request(c, &calling, &called))
 321   {
 322     fputs("ERROR: cli_session_request() failed...\n", stderr);
 323     return (NULL);
 324   }
 325
 326   if (!cli_negprot(c))
 327   {
 328     fputs("ERROR: SMB protocol negotiation failed\n", stderr);
 329     cli_shutdown(c);
 330     return (NULL);
 331   }
 332
 333  /*
 334   * Do password stuff...
 335   */
 336
 337   if (!cli_session_setup(c, username,
 338                          password, strlen(password),
 339                          password, strlen(password),
 340                          workgroup))
 341   {
 342     fprintf(stderr, "ERROR: SMB session setup failed: %s\n", cli_errstr(c));
 343     return (NULL);
 344   }
 345
 346   if (!cli_send_tconX(c, share, "?????",
 347                       password, strlen(password)+1))
 348   {
 349     fprintf(stderr, "ERROR: SMB tree connect failed: %s\n", cli_errstr(c));
 350     cli_shutdown(c);
 351     return (NULL);
 352   }
 353
 354  /*
 355   * Return the new connection...
 356   */
 357
 358   return (c);
 359 }
 360
 361
 362 /*
 363  * 'smb_print()' - Queue a job for printing using the SMB protocol.
 364  */
 365
 366 static int                              /* O - 0 = success, non-0 = failure */
 367 smb_print(struct cli_state *cli,        /* I - SMB connection */
 368           char             *title,      /* I - Title/job name */
 369           FILE             *fp)         /* I - File to print */
 370 {
 371   int   fnum;           /* File number */
 372   int   nbytes,         /* Number of bytes read */
 373         tbytes;         /* Total bytes read */
 374   char  buffer[8192],   /* Buffer for copy */
 375         *ptr;           /* Pointer into tile */
 376
 377
 378  /*
 379   * Sanitize the title...
 380   */
 381
 382   for (ptr = title; *ptr; ptr ++)
 383     if (!isalnum((int)*ptr) && !isspace((int)*ptr))
 384       *ptr = '_';
 385
 386  /*
 387   * Open the printer device...
 388   */
 389
 390   if ((fnum = cli_open(cli, title, O_RDWR | O_CREAT | O_TRUNC, DENY_NONE)) == -1)
 391   {
 392     fprintf(stderr, "ERROR: %s opening remote file %s\n",
 393             cli_errstr(cli), title);
 394     return (1);
 395   }
 396
 397  /*
 398   * Copy the file to the printer...
 399   */
 400
 401   if (fp != stdin)
 402     rewind(fp);
 403
 404   tbytes = 0;
 405
 406   while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0)
 407   {
 408     if (cli_write(cli, fnum, 0, buffer, tbytes, nbytes) != nbytes)
 409     {
 410       fprintf(stderr, "ERROR: Error writing file: %s\n", cli_errstr(cli));
 411       break;
 412     }
 413
 414     tbytes += nbytes;
 415   }
 416
 417   if (!cli_close(cli, fnum))
 418   {
 419     fprintf(stderr, "ERROR: %s closing remote file %s\n",
 420             cli_errstr(cli), title);
 421     return (1);
 422   }
 423   else
 424     return (0);
 425 }