source4/client/smbspool.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    SMB backend for the Common UNIX Printing System ("CUPS")
   4    Copyright 1999 by Easy Software Products
   5    Copyright Andrew Tridgell 1994-1998
   6    Copyright Andrew Bartlett 2002
   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 3 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, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23
  24 /*
  25  * Local functions...
  26  */
  27
  28 static void             list_devices(void);
  29 static struct smbcli_state      *smb_connect(const char *, const char *, const char **, const char *, const char *, const char *);
  30 static int              smb_print(struct smbcli_state *, char *, FILE *);
  31
  32
  33 /*
  34  * 'main()' - Main entry for SMB backend.
  35  */
  36
  37  int                            /* O - Exit status */
  38  main(int  argc,                        /* I - Number of command-line arguments */
  39      char *argv[])              /* I - Command-line arguments */
  40 {
  41   int           i;              /* Looping var */
  42   int           copies;         /* Number of copies */
  43   char          uri[1024],      /* URI */
  44                 *sep,           /* Pointer to separator */
  45                 *password;      /* Password */
  46   const char    *username,      /* Username */
  47                 *server,        /* Server name */
  48                 *printer;       /* Printer name */
  49   const char    *workgroup;     /* Workgroup */
  50   FILE          *fp;            /* File to print */
  51   int           status=0;               /* Status of LPD job */
  52   struct smbcli_state *cli;     /* SMB interface */
  53   struct loadparm_context *lp_ctx;
  54
  55   /* we expect the URI in argv[0]. Detect the case where it is in argv[1] and cope */
  56   if (argc > 2 && strncmp(argv[0],"smb://", 6) && !strncmp(argv[1],"smb://", 6)) {
  57           argv++;
  58           argc--;
  59   }
  60
  61   if (argc == 1)
  62   {
  63    /*
  64     * NEW!  In CUPS 1.1 the backends are run with no arguments to list the
  65     *       available devices.  These can be devices served by this backend
  66     *       or any other backends (i.e. you can have an SNMP backend that
  67     *       is only used to enumerate the available network printers... :)
  68     */
  69
  70     list_devices();
  71     return (0);
  72   }
  73
  74   if (argc < 6 || argc > 7)
  75   {
  76     fprintf(stderr, "Usage: %s [DEVICE_URI] job-id user title copies options [file]\n",
  77             argv[0]);
  78     fputs("       The DEVICE_URI environment variable can also contain the\n", stderr);
  79     fputs("       destination printer:\n", stderr);
  80     fputs("\n", stderr);
  81     fputs("           smb://[username:password@][workgroup/]server/printer\n", stderr);
  82     return (1);
  83   }
  84
  85  /*
  86   * If we have 7 arguments, print the file named on the command-line.
  87   * Otherwise, print data from stdin...
  88   */
  89
  90   if (argc == 6)
  91   {
  92    /*
  93     * Print from Copy stdin to a temporary file...
  94     */
  95
  96     fp     = stdin;
  97     copies = 1;
  98   }
  99   else if ((fp = fopen(argv[6], "rb")) == NULL)
 100   {
 101     perror("ERROR: Unable to open print file");
 102     return (1);
 103   }
 104   else
 105     copies = atoi(argv[4]);
 106
 107  /*
 108   * Find the URI...
 109   */
 110
 111   if (strncmp(argv[0], "smb://", 6) == 0)
 112     strncpy(uri, argv[0], sizeof(uri) - 1);
 113   else if (getenv("DEVICE_URI") != NULL)
 114     strncpy(uri, getenv("DEVICE_URI"), sizeof(uri) - 1);
 115   else
 116   {
 117     fputs("ERROR: No device URI found in argv[0] or DEVICE_URI environment variable!\n", stderr);
 118     return (1);
 119   }
 120
 121   uri[sizeof(uri) - 1] = '\0';
 122
 123  /*
 124   * Extract the destination from the URI...
 125   */
 126
 127   if ((sep = strrchr_m(uri, '@')) != NULL)
 128   {
 129     username = uri + 6;
 130     *sep++ = '\0';
 131
 132     server = sep;
 133
 134    /*
 135     * Extract password as needed...
 136     */
 137
 138     if ((password = strchr_m(username, ':')) != NULL)
 139       *password++ = '\0';
 140     else
 141       password = "";
 142   }
 143   else
 144   {
 145     username = "";
 146     password = "";
 147     server   = uri + 6;
 148   }
 149
 150   if ((sep = strchr_m(server, '/')) == NULL)
 151   {
 152     fputs("ERROR: Bad URI - need printer name!\n", stderr);
 153     return (1);
 154   }
 155
 156   *sep++ = '\0';
 157   printer = sep;
 158
 159   if ((sep = strchr_m(printer, '/')) != NULL)
 160   {
 161    /*
 162     * Convert to smb://[username:password@]workgroup/server/printer...
 163     */
 164
 165     *sep++ = '\0';
 166
 167     workgroup = server;
 168     server    = printer;
 169     printer   = sep;
 170   }
 171   else
 172     workgroup = NULL;
 173
 174  /*
 175   * Setup the SAMBA server state...
 176   */
 177
 178   setup_logging(argv[0], DEBUG_STDOUT);
 179
 180   lp_ctx = loadparm_init(talloc_autofree_context());
 181
 182   if (!lp_load(lp_ctx, dyn_CONFIGFILE)) {
 183           fprintf(stderr, "ERROR: Can't load %s - run testparm to debug it\n", lp_config_file());
 184           return (1);
 185   }
 186
 187   if (workgroup == NULL)
 188     workgroup = lp_workgroup(lp_ctx);
 189
 190   do
 191   {
 192     if ((cli = smb_connect(workgroup, server, lp_smb_ports(lp_ctx), printer, username, password)) == NULL)
 193     {
 194       if (getenv("CLASS") == NULL)
 195       {
 196         fprintf(stderr, "ERROR: Unable to connect to SAMBA host, will retry in 60 seconds...");
 197         sleep (60);
 198       }
 199       else
 200       {
 201         fprintf(stderr, "ERROR: Unable to connect to SAMBA host, trying next printer...");
 202         return (1);
 203       }
 204     }
 205   }
 206   while (cli == NULL);
 207
 208  /*
 209   * Now that we are connected to the server, ignore SIGTERM so that we
 210   * can finish out any page data the driver sends (e.g. to eject the
 211   * current page...  Only ignore SIGTERM if we are printing data from
 212   * stdin (otherwise you can't cancel raw jobs...)
 213   */
 214
 215   if (argc < 7)
 216     CatchSignal(SIGTERM, SIG_IGN);
 217
 218  /*
 219   * Queue the job...
 220   */
 221
 222   for (i = 0; i < copies; i ++)
 223     if ((status = smb_print(cli, argv[3] /* title */, fp)) != 0)
 224       break;
 225
 226   talloc_free(cli);
 227
 228  /*
 229   * Return the queue status...
 230   */
 231
 232   return (status);
 233 }
 234
 235
 236 /*
 237  * 'list_devices()' - List the available printers seen on the network...
 238  */
 239
 240 static void
 241 list_devices(void)
 242 {
 243  /*
 244   * Eventually, search the local workgroup for available hosts and printers.
 245   */
 246
 247   puts("network smb \"Unknown\" \"Windows Printer via SAMBA\"");
 248 }
 249
 250
 251 /*
 252  * 'smb_connect()' - Return a connection to a server.
 253  */
 254
 255 static struct smbcli_state *            /* O - SMB connection */
 256 smb_connect(const char *workgroup,              /* I - Workgroup */
 257             const char *server,         /* I - Server */
 258             const char **ports,           /* I - Ports */
 259             const char *share,          /* I - Printer */
 260             const char *username,               /* I - Username */
 261             const char *password)               /* I - Password */
 262 {
 263   struct smbcli_state   *c;             /* New connection */
 264   char *myname;         /* Client name */
 265   NTSTATUS nt_status;
 266
 267  /*
 268   * Get the names and addresses of the client and server...
 269   */
 270
 271   myname = get_myname();
 272
 273   nt_status = smbcli_full_connection(NULL, &c, myname, server, ports, share,
 274                                      NULL, username, workgroup, password, NULL);
 275
 276   free(myname);
 277   if (!NT_STATUS_IS_OK(nt_status)) {
 278           fprintf(stderr, "ERROR:  Connection failed with error %s\n", nt_errstr(nt_status));
 279           return NULL;
 280   }
 281
 282   /*
 283    * Return the new connection...
 284    */
 285
 286   return (c);
 287 }
 288
 289
 290 /*
 291  * 'smb_print()' - Queue a job for printing using the SMB protocol.
 292  */
 293
 294 static int                              /* O - 0 = success, non-0 = failure */
 295 smb_print(struct smbcli_state *cli,     /* I - SMB connection */
 296           char             *title,      /* I - Title/job name */
 297           FILE             *fp)         /* I - File to print */
 298 {
 299   int   fnum;           /* File number */
 300   int   nbytes,         /* Number of bytes read */
 301         tbytes;         /* Total bytes read */
 302   char  buffer[8192],   /* Buffer for copy */
 303         *ptr;           /* Pointer into tile */
 304
 305
 306  /*
 307   * Sanitize the title...
 308   */
 309
 310   for (ptr = title; *ptr; ptr ++)
 311     if (!isalnum((int)*ptr) && !isspace((int)*ptr))
 312       *ptr = '_';
 313
 314  /*
 315   * Open the printer device...
 316   */
 317
 318   if ((fnum = smbcli_open(cli, title, O_RDWR | O_CREAT | O_TRUNC, DENY_NONE)) == -1)
 319   {
 320     fprintf(stderr, "ERROR: %s opening remote file %s\n",
 321             smbcli_errstr(cli), title);
 322     return (1);
 323   }
 324
 325  /*
 326   * Copy the file to the printer...
 327   */
 328
 329   if (fp != stdin)
 330     rewind(fp);
 331
 332   tbytes = 0;
 333
 334   while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0)
 335   {
 336     if (smbcli_write(cli, fnum, 0, buffer, tbytes, nbytes) != nbytes)
 337     {
 338       fprintf(stderr, "ERROR: Error writing file: %s\n", smbcli_errstr(cli));
 339       break;
 340     }
 341
 342     tbytes += nbytes;
 343   }
 344
 345   if (!smbcli_close(cli, fnum))
 346   {
 347     fprintf(stderr, "ERROR: %s closing remote file %s\n",
 348             smbcli_errstr(cli), title);
 349     return (1);
 350   }
 351   else
 352     return (0);
 353 }