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