?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s3-utils/net_rpc_printer.c: return on read error in net_copy_file()
[Samba.git] / source3 / utils / net_rpc_printer.c
bloba8221d093159646f196de1c28da857cdc8e760f2
1 /*
2 Samba Unix/Linux SMB client library
3 Distributed SMB/CIFS Server Management Utility
4 Copyright (C) 2004,2009 Guenther Deschner (gd@samba.org)
5
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
10
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
18 */
19 #include "includes.h"
20 #include "system/filesys.h"
21 #include "utils/net.h"
22 #include "rpc_client/rpc_client.h"
23 #include "../librpc/gen_ndr/ndr_spoolss_c.h"
24 #include "rpc_client/cli_spoolss.h"
25 #include "rpc_client/init_spoolss.h"
26 #include "nt_printing.h"
27 #include "registry/reg_objects.h"
28 #include "../libcli/security/security.h"
29 #include "../libcli/registry/util_reg.h"
30 #include "libsmb/libsmb.h"
31
32 /* support itanium as well */
33 static const struct print_architecture_table_node archi_table[]= {
34
35 {"Windows 4.0", "WIN40", 0 },
36 {"Windows NT x86", "W32X86", 2 },
37 {"Windows NT x86", "W32X86", 3 },
38 {"Windows NT R4000", "W32MIPS", 2 },
39 {"Windows NT Alpha_AXP", "W32ALPHA", 2 },
40 {"Windows NT PowerPC", "W32PPC", 2 },
41 {"Windows IA64", "IA64", 3 },
42 {"Windows x64", "x64", 3 },
43 {NULL, "", -1 }
44 };
45
46
47 /**
48 * This display-printdriver-functions was borrowed from rpcclient/cmd_spoolss.c.
49 * It is here for debugging purpose and should be removed later on.
50 **/
51
52 /****************************************************************************
53 Printer info level 3 display function.
54 ****************************************************************************/
55
56 static void display_print_driver3(struct spoolss_DriverInfo3 *r)
57 {
58 int i;
59
60 if (!r) {
61 return;
62 }
63
64 printf(_("Printer Driver Info 3:\n"));
65 printf(_("\tVersion: [%x]\n"), r->version);
66 printf(_("\tDriver Name: [%s]\n"), r->driver_name);
67 printf(_("\tArchitecture: [%s]\n"), r->architecture);
68 printf(_("\tDriver Path: [%s]\n"), r->driver_path);
69 printf(_("\tDatafile: [%s]\n"), r->data_file);
70 printf(_("\tConfigfile: [%s]\n\n"), r->config_file);
71 printf(_("\tHelpfile: [%s]\n\n"), r->help_file);
72
73 for (i=0; r->dependent_files && r->dependent_files[i] != NULL; i++) {
74 printf(_("\tDependentfiles: [%s]\n"), r->dependent_files[i]);
75 }
76
77 printf("\n");
78
79 printf(_("\tMonitorname: [%s]\n"), r->monitor_name);
80 printf(_("\tDefaultdatatype: [%s]\n\n"), r->default_datatype);
81 }
82
83 static void display_reg_value(const char *subkey, struct regval_blob *value)
84 {
85 const char *text;
86 DATA_BLOB blob;
87
88 switch(regval_type(value)) {
89 case REG_DWORD:
90 d_printf(_("\t[%s:%s]: REG_DWORD: 0x%08x\n"), subkey,
91 regval_name(value), *((uint32_t *) regval_data_p(value)));
92 break;
93
94 case REG_SZ:
95 blob = data_blob_const(regval_data_p(value), regval_size(value));
96 pull_reg_sz(talloc_tos(), &blob, &text);
97 if (!text) {
98 break;
99 }
100 d_printf(_("\t[%s:%s]: REG_SZ: %s\n"), subkey, regval_name(value),
101 text);
102 break;
103
104 case REG_BINARY:
105 d_printf(_("\t[%s:%s]: REG_BINARY: unknown length value not "
106 "displayed\n"),
107 subkey, regval_name(value));
108 break;
109
110 case REG_MULTI_SZ: {
111 uint32_t i;
112 const char **values;
113 blob = data_blob_const(regval_data_p(value), regval_size(value));
114
115 if (!pull_reg_multi_sz(NULL, &blob, &values)) {
116 d_printf("pull_reg_multi_sz failed\n");
117 break;
118 }
119
120 printf("%s: REG_MULTI_SZ: \n", regval_name(value));
121 for (i=0; values[i] != NULL; i++) {
122 d_printf("%s\n", values[i]);
123 }
124 TALLOC_FREE(values);
125 break;
126 }
127
128 default:
129 d_printf(_("\t%s: unknown type %d\n"), regval_name(value),
130 regval_type(value));
131 }
132
133 }
134
135 /**
136 * Copies ACLs, DOS-attributes and timestamps from one
137 * file or directory from one connected share to another connected share
138 *
139 * @param c A net_context structure
140 * @param mem_ctx A talloc-context
141 * @param cli_share_src A connected cli_state
142 * @param cli_share_dst A connected cli_state
143 * @param src_file The source file-name
144 * @param dst_file The destination file-name
145 * @param copy_acls Whether to copy acls
146 * @param copy_attrs Whether to copy DOS attributes
147 * @param copy_timestamps Whether to preserve timestamps
148 * @param is_file Whether this file is a file or a dir
149 *
150 * @return Normal NTSTATUS return.
151 **/
152
153 NTSTATUS net_copy_fileattr(struct net_context *c,
154 TALLOC_CTX *mem_ctx,
155 struct cli_state *cli_share_src,
156 struct cli_state *cli_share_dst,
157 const char *src_name, const char *dst_name,
158 bool copy_acls, bool copy_attrs,
159 bool copy_timestamps, bool is_file)
160 {
161 NTSTATUS nt_status;
162 uint16_t fnum_src = 0;
163 uint16_t fnum_dst = 0;
164 struct security_descriptor *sd = NULL;
165 uint16_t attr;
166 time_t f_atime, f_ctime, f_mtime;
167
168 if (!copy_timestamps && !copy_acls && !copy_attrs)
169 return NT_STATUS_OK;
170
171 /* open file/dir on the originating server */
172
173 DEBUGADD(3,("opening %s %s on originating server\n",
174 is_file?"file":"dir", src_name));
175
176 nt_status = cli_ntcreate(cli_share_src, src_name, 0,
177 READ_CONTROL_ACCESS, 0,
178 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN,
179 0x0, 0x0, &fnum_src);
180 if (!NT_STATUS_IS_OK(nt_status)) {
181 DEBUGADD(0,("cannot open %s %s on originating server %s\n",
182 is_file?"file":"dir", src_name, nt_errstr(nt_status)));
183 goto out;
184 }
185
186 if (copy_acls) {
187 /* get the security descriptor */
188 sd = cli_query_secdesc(cli_share_src, fnum_src, mem_ctx);
189 if (!sd) {
190 DEBUG(0,("failed to get security descriptor: %s\n",
191 cli_errstr(cli_share_src)));
192 nt_status = cli_nt_error(cli_share_src);
193 goto out;
194 }
195
196 if (c->opt_verbose && DEBUGLEVEL >= 3)
197 display_sec_desc(sd);
198 }
199
200 if (copy_attrs || copy_timestamps) {
201
202 /* get file attributes */
203 nt_status = cli_getattrE(cli_share_src, fnum_src, &attr, NULL,
204 &f_ctime, &f_atime, &f_mtime);
205 if (!NT_STATUS_IS_OK(nt_status)) {
206 DEBUG(0,("failed to get file-attrs: %s\n",
207 nt_errstr(nt_status)));
208 goto out;
209 }
210 }
211
212 /* open the file/dir on the destination server */
213 nt_status = cli_ntcreate(cli_share_dst, dst_name, 0,
214 WRITE_DAC_ACCESS | WRITE_OWNER_ACCESS, 0,
215 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN,
216 0x0, 0x0, &fnum_dst);
217 if (!NT_STATUS_IS_OK(nt_status)) {
218 DEBUG(0,("failed to open %s on the destination server: %s: %s\n",
219 is_file?"file":"dir", dst_name, nt_errstr(nt_status)));
220 goto out;
221 }
222
223 if (copy_timestamps) {
224 /* set timestamps */
225 nt_status = cli_setattrE(cli_share_dst, fnum_dst, f_ctime, f_atime, f_mtime);
226 if (!NT_STATUS_IS_OK(nt_status)) {
227 DEBUG(0,("failed to set file-attrs (timestamps): %s\n",
228 nt_errstr(nt_status)));
229 goto out;
230 }
231 }
232
233 if (copy_acls) {
234 /* set acls */
235 nt_status = cli_set_secdesc(cli_share_dst, fnum_dst, sd);
236 if (!NT_STATUS_IS_OK(nt_status)) {
237 DEBUG(0, ("could not set secdesc on %s: %s\n",
238 dst_name, nt_errstr(nt_status)));
239 goto out;
240 }
241 }
242
243 if (copy_attrs) {
244 /* set attrs */
245 nt_status = cli_setatr(cli_share_dst, dst_name, attr, 0);
246 if (!NT_STATUS_IS_OK(nt_status)) {
247 DEBUG(0,("failed to set file-attrs: %s\n",
248 nt_errstr(nt_status)));
249 goto out;
250 }
251 }
252
253
254 /* closing files */
255 nt_status = cli_close(cli_share_src, fnum_src);
256 if (!NT_STATUS_IS_OK(nt_status)) {
257 d_fprintf(stderr,
258 _("could not close %s on originating server: %s\n"),
259 is_file?"file":"dir", nt_errstr(nt_status));
260 goto out;
261 }
262
263 nt_status = cli_close(cli_share_dst, fnum_dst);
264 if (!NT_STATUS_IS_OK(nt_status)) {
265 d_fprintf(stderr,
266 _("could not close %s on destination server: %s\n"),
267 is_file?"file":"dir", nt_errstr(nt_status));
268 goto out;
269 }
270
271
272 nt_status = NT_STATUS_OK;
273
274 out:
275
276 /* cleaning up */
277 if (fnum_src)
278 cli_close(cli_share_src, fnum_src);
279
280 if (fnum_dst)
281 cli_close(cli_share_dst, fnum_dst);
282
283 return nt_status;
284 }
285
286 /**
287 * Copy a file or directory from a connected share to another connected share
288 *
289 * @param c A net_context structure
290 * @param mem_ctx A talloc-context
291 * @param cli_share_src A connected cli_state
292 * @param cli_share_dst A connected cli_state
293 * @param src_file The source file-name
294 * @param dst_file The destination file-name
295 * @param copy_acls Whether to copy acls
296 * @param copy_attrs Whether to copy DOS attributes
297 * @param copy_timestamps Whether to preserve timestamps
298 * @param is_file Whether this file is a file or a dir
299 *
300 * @return Normal NTSTATUS return.
301 **/
302
303 NTSTATUS net_copy_file(struct net_context *c,
304 TALLOC_CTX *mem_ctx,
305 struct cli_state *cli_share_src,
306 struct cli_state *cli_share_dst,
307 const char *src_name, const char *dst_name,
308 bool copy_acls, bool copy_attrs,
309 bool copy_timestamps, bool is_file)
310 {
311 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
312 uint16_t fnum_src = 0;
313 uint16_t fnum_dst = 0;
314 static int io_bufsize = 64512;
315 int read_size = io_bufsize;
316 char *data = NULL;
317 off_t nread = 0;
318
319
320 if (!src_name || !dst_name)
321 goto out;
322
323 if (cli_share_src == NULL || cli_share_dst == NULL)
324 goto out;
325
326 /* open on the originating server */
327 DEBUGADD(3,("opening %s %s on originating server\n",
328 is_file ? "file":"dir", src_name));
329 if (is_file)
330 nt_status = cli_open(cli_share_src, src_name, O_RDONLY, DENY_NONE, &fnum_src);
331 else
332 nt_status = cli_ntcreate(cli_share_src, src_name, 0, READ_CONTROL_ACCESS, 0,
333 FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, 0x0, 0x0, &fnum_src);
334
335 if (!NT_STATUS_IS_OK(nt_status)) {
336 DEBUGADD(0,("cannot open %s %s on originating server %s\n",
337 is_file ? "file":"dir",
338 src_name, nt_errstr(nt_status)));
339 goto out;
340 }
341
342
343 if (is_file) {
344
345 /* open file on the destination server */
346 DEBUGADD(3,("opening file %s on destination server\n", dst_name));
347 nt_status = cli_open(cli_share_dst, dst_name,
348 O_RDWR|O_CREAT|O_TRUNC, DENY_NONE, &fnum_dst);
349
350 if (!NT_STATUS_IS_OK(nt_status)) {
351 DEBUGADD(1,("cannot create file %s on destination server: %s\n",
352 dst_name, nt_errstr(nt_status)));
353 goto out;
354 }
355
356 /* allocate memory */
357 if (!(data = (char *)SMB_MALLOC(read_size))) {
358 d_fprintf(stderr, _("malloc fail for size %d\n"),
359 read_size);
360 nt_status = NT_STATUS_NO_MEMORY;
361 goto out;
362 }
363
364 }
365
366
367 if (c->opt_verbose) {
368
369 d_printf(_("copying [\\\\%s\\%s%s] => [\\\\%s\\%s%s] "
370 "%s ACLs and %s DOS Attributes %s\n"),
371 cli_state_remote_name(cli_share_src),
372 cli_share_src->share, src_name,
373 cli_state_remote_name(cli_share_dst),
374 cli_share_dst->share, dst_name,
375 copy_acls ? _("with") : _("without"),
376 copy_attrs ? _("with") : _("without"),
377 copy_timestamps ? _("(preserving timestamps)") : "" );
378 }
379
380
381 while (is_file) {
382
383 /* copying file */
384 size_t n;
385
386 nt_status = cli_read(cli_share_src, fnum_src, data, nread,
387 read_size, &n);
388 if (!NT_STATUS_IS_OK(nt_status)) {
389 d_fprintf(stderr,
390 _("Error reading file [\\\\%s\%s%s]: %s\n"),
391 cli_state_remote_name(cli_share_src),
392 cli_share_src->share,
393 src_name, nt_errstr(nt_status));
394 goto out;
395 }
396
397 if (n == 0)
398 break;
399
400 nt_status = cli_writeall(cli_share_dst, fnum_dst, 0,
401 (uint8_t *)data, nread, n, NULL);
402
403 if (!NT_STATUS_IS_OK(nt_status)) {
404 d_fprintf(stderr, _("Error writing file: %s\n"),
405 nt_errstr(nt_status));
406 goto out;
407 }
408
409 nread += n;
410 }
411
412
413 if (!is_file && !NT_STATUS_IS_OK(cli_chkpath(cli_share_dst, dst_name))) {
414
415 /* creating dir */
416 DEBUGADD(3,("creating dir %s on the destination server\n",
417 dst_name));
418
419 nt_status = cli_mkdir(cli_share_dst, dst_name);
420 if (!NT_STATUS_IS_OK(nt_status)) {
421 DEBUG(0,("cannot create directory %s: %s\n",
422 dst_name, nt_errstr(nt_status)));
423 nt_status = NT_STATUS_NO_SUCH_FILE;
424 }
425
426
427 nt_status = cli_chkpath(cli_share_dst, dst_name);
428 if (!NT_STATUS_IS_OK(nt_status)) {
429 d_fprintf(stderr,
430 _("cannot check for directory %s: %s\n"),
431 dst_name, nt_errstr(nt_status));
432 goto out;
433 }
434 }
435
436
437 /* closing files */
438 nt_status = cli_close(cli_share_src, fnum_src);
439 if (!NT_STATUS_IS_OK(nt_status)) {
440 d_fprintf(stderr,
441 _("could not close file on originating server: %s\n"),
442 nt_errstr(nt_status));
443 goto out;
444 }
445
446 if (is_file) {
447 nt_status = cli_close(cli_share_dst, fnum_dst);
448 if (!NT_STATUS_IS_OK(nt_status)) {
449 d_fprintf(stderr,
450 _("could not close file on destination server: %s\n"),
451 nt_errstr(nt_status));
452 goto out;
453 }
454 }
455
456 /* possibly we have to copy some file-attributes / acls / sd */
457 nt_status = net_copy_fileattr(c, mem_ctx, cli_share_src, cli_share_dst,
458 src_name, dst_name, copy_acls,
459 copy_attrs, copy_timestamps, is_file);
460 if (!NT_STATUS_IS_OK(nt_status))
461 goto out;
462
463
464 nt_status = NT_STATUS_OK;
465
466 out:
467
468 /* cleaning up */
469 if (fnum_src)
470 cli_close(cli_share_src, fnum_src);
471
472 if (fnum_dst)
473 cli_close(cli_share_dst, fnum_dst);
474
475 SAFE_FREE(data);
476
477 return nt_status;
478 }
479
480 /**
481 * Copy a driverfile from on connected share to another connected share
482 * This silently assumes that a driver-file is picked up from
483 *
484 * \\src_server\print$\{arch}\{version}\file
485 *
486 * and copied to
487 *
488 * \\dst_server\print$\{arch}\file
489 *
490 * to be added via setdriver-calls later.
491 * @param c A net_context structure
492 * @param mem_ctx A talloc-context
493 * @param cli_share_src A cli_state connected to source print$-share
494 * @param cli_share_dst A cli_state connected to destination print$-share
495 * @param file The file-name to be copied
496 * @param short_archi The name of the driver-architecture (short form)
497 *
498 * @return Normal NTSTATUS return.
499 **/
500
501 static NTSTATUS net_copy_driverfile(struct net_context *c,
502 TALLOC_CTX *mem_ctx,
503 struct cli_state *cli_share_src,
504 struct cli_state *cli_share_dst,
505 const char *file, const char *short_archi) {
506
507 const char *p;
508 char *src_name;
509 char *dst_name;
510 char *version = NULL;
511 char *filename = NULL;
512 char *tok;
513
514 if (!file) {
515 return NT_STATUS_OK;
516 }
517
518 /* scroll through the file until we have the part
519 beyond archi_table.short_archi */
520 p = file;
521 while (next_token_talloc(mem_ctx, &p, &tok, "\\")) {
522 if (strequal(tok, short_archi)) {
523 next_token_talloc(mem_ctx, &p, &version, "\\");
524 next_token_talloc(mem_ctx, &p, &filename, "\\");
525 }
526 }
527
528 if (version == NULL || filename == NULL) {
529 return NT_STATUS_UNSUCCESSFUL;
530 }
531
532 /* build source file name */
533 src_name = talloc_asprintf(mem_ctx, "\\%s\\%s\\%s",
534 short_archi, version, filename);
535 if (src_name == NULL) {
536 return NT_STATUS_NO_MEMORY;
537 }
538
539 /* create destination file name */
540 dst_name = talloc_asprintf(mem_ctx, "\\%s\\%s", short_archi, filename);
541 if (dst_name == NULL) {
542 return NT_STATUS_NO_MEMORY;
543 }
544
545
546 /* finally copy the file */
547 return net_copy_file(c, mem_ctx, cli_share_src, cli_share_dst,
548 src_name, dst_name, false, false, false, true);
549 }
550
551 /**
552 * Check for existing Architecture directory on a given server
553 *
554 * @param cli_share A cli_state connected to a print$-share
555 * @param short_archi The Architecture for the print-driver
556 *
557 * @return Normal NTSTATUS return.
558 **/
559
560 static NTSTATUS check_arch_dir(struct cli_state *cli_share, const char *short_archi)
561 {
562
563 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
564 char *dir;
565
566 if (asprintf(&dir, "\\%s", short_archi) < 0) {
567 return NT_STATUS_NO_MEMORY;
568 }
569
570 DEBUG(10,("creating print-driver dir for architecture: %s\n",
571 short_archi));
572
573 nt_status = cli_mkdir(cli_share, dir);
574 if (!NT_STATUS_IS_OK(nt_status)) {
575 DEBUG(1,("cannot create directory %s: %s\n",
576 dir, nt_errstr(nt_status)));
577 }
578
579 nt_status = cli_chkpath(cli_share, dir);
580 if (!NT_STATUS_IS_OK(nt_status)) {
581 d_fprintf(stderr, _("cannot check %s: %s\n"),
582 dir, nt_errstr(nt_status));
583 goto out;
584 }
585
586 nt_status = NT_STATUS_OK;
587
588 out:
589 SAFE_FREE(dir);
590 return nt_status;
591 }
592
593 /**
594 * Copy a print-driver (level 3) from one connected print$-share to another
595 * connected print$-share
596 *
597 * @param c A net_context structure
598 * @param mem_ctx A talloc-context
599 * @param cli_share_src A cli_state connected to a print$-share
600 * @param cli_share_dst A cli_state connected to a print$-share
601 * @param short_archi The Architecture for the print-driver
602 * @param i1 The DRIVER_INFO_3-struct
603 *
604 * @return Normal NTSTATUS return.
605 **/
606
607 static NTSTATUS copy_print_driver_3(struct net_context *c,
608 TALLOC_CTX *mem_ctx,
609 struct cli_state *cli_share_src,
610 struct cli_state *cli_share_dst,
611 const char *short_archi,
612 struct spoolss_DriverInfo3 *r)
613 {
614 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
615 int i;
616
617 if (r == NULL) {
618 return nt_status;
619 }
620
621 if (c->opt_verbose)
622 d_printf(_("copying driver: [%s], for architecture: [%s], "
623 "version: [%d]\n"),
624 r->driver_name, short_archi, r->version);
625
626 nt_status = net_copy_driverfile(c, mem_ctx, cli_share_src, cli_share_dst,
627 r->driver_path, short_archi);
628 if (!NT_STATUS_IS_OK(nt_status))
629 return nt_status;
630
631 nt_status = net_copy_driverfile(c, mem_ctx, cli_share_src, cli_share_dst,
632 r->data_file, short_archi);
633 if (!NT_STATUS_IS_OK(nt_status))
634 return nt_status;
635
636 nt_status = net_copy_driverfile(c, mem_ctx, cli_share_src, cli_share_dst,
637 r->config_file, short_archi);
638 if (!NT_STATUS_IS_OK(nt_status))
639 return nt_status;
640
641 nt_status = net_copy_driverfile(c, mem_ctx, cli_share_src, cli_share_dst,
642 r->help_file, short_archi);
643 if (!NT_STATUS_IS_OK(nt_status))
644 return nt_status;
645
646 for (i=0; r->dependent_files[i] != NULL; i++) {
647
648 nt_status = net_copy_driverfile(c, mem_ctx,
649 cli_share_src, cli_share_dst,
650 r->dependent_files[i], short_archi);
651 if (!NT_STATUS_IS_OK(nt_status)) {
652 return nt_status;
653 }
654 }
655
656 return NT_STATUS_OK;
657 }
658
659 /**
660 * net_spoolss-functions
661 * =====================
662 *
663 * the net_spoolss-functions aim to simplify spoolss-client-functions
664 * required during the migration-process wrt buffer-sizes, returned
665 * error-codes, etc.
666 *
667 * this greatly reduces the complexitiy of the migrate-functions.
668 *
669 **/
670
671 static bool net_spoolss_enum_printers(struct rpc_pipe_client *pipe_hnd,
672 TALLOC_CTX *mem_ctx,
673 char *name,
674 uint32_t flags,
675 uint32_t level,
676 uint32_t *num_printers,
677 union spoolss_PrinterInfo **info)
678 {
679 WERROR result;
680
681 /* enum printers */
682
683 result = rpccli_spoolss_enumprinters(pipe_hnd, mem_ctx,
684 flags,
685 name,
686 level,
687 0,
688 num_printers,
689 info);
690 if (!W_ERROR_IS_OK(result)) {
691 printf(_("cannot enum printers: %s\n"), win_errstr(result));
692 return false;
693 }
694
695 return true;
696 }
697
698 static bool net_spoolss_open_printer_ex(struct rpc_pipe_client *pipe_hnd,
699 TALLOC_CTX *mem_ctx,
700 const char *printername,
701 uint32_t access_required,
702 const char *username,
703 struct policy_handle *hnd)
704 {
705 WERROR result;
706 fstring printername2;
707
708 fstrcpy(printername2, pipe_hnd->srv_name_slash);
709 fstrcat(printername2, "\\");
710 fstrcat(printername2, printername);
711
712 DEBUG(10,("connecting to: %s as %s for %s and access: %x\n",
713 pipe_hnd->srv_name_slash, username, printername2, access_required));
714
715 /* open printer */
716 result = rpccli_spoolss_openprinter_ex(pipe_hnd, mem_ctx,
717 printername2,
718 access_required,
719 hnd);
720
721 /* be more verbose */
722 if (W_ERROR_V(result) == W_ERROR_V(WERR_ACCESS_DENIED)) {
723 d_fprintf(stderr,
724 _("no access to printer [%s] on [%s] for user [%s] "
725 "granted\n"),
726 printername2, pipe_hnd->srv_name_slash, username);
727 return false;
728 }
729
730 if (!W_ERROR_IS_OK(result)) {
731 d_fprintf(stderr,_("cannot open printer %s on server %s: %s\n"),
732 printername2, pipe_hnd->srv_name_slash, win_errstr(result));
733 return false;
734 }
735
736 DEBUG(2,("got printer handle for printer: %s, server: %s\n",
737 printername2, pipe_hnd->srv_name_slash));
738
739 return true;
740 }
741
742 static bool net_spoolss_getprinter(struct rpc_pipe_client *pipe_hnd,
743 TALLOC_CTX *mem_ctx,
744 struct policy_handle *hnd,
745 uint32_t level,
746 union spoolss_PrinterInfo *info)
747 {
748 WERROR result;
749
750 /* getprinter call */
751 result = rpccli_spoolss_getprinter(pipe_hnd, mem_ctx,
752 hnd,
753 level,
754 0, /* offered */
755 info);
756 if (!W_ERROR_IS_OK(result)) {
757 printf(_("cannot get printer-info: %s\n"), win_errstr(result));
758 return false;
759 }
760
761 return true;
762 }
763
764 static bool net_spoolss_setprinter(struct rpc_pipe_client *pipe_hnd,
765 TALLOC_CTX *mem_ctx,
766 struct policy_handle *hnd,
767 uint32_t level,
768 union spoolss_PrinterInfo *info)
769 {
770 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
771 WERROR result;
772 NTSTATUS status;
773 struct spoolss_SetPrinterInfoCtr info_ctr;
774 struct spoolss_SetPrinterInfo2 info2;
775 struct spoolss_DevmodeContainer devmode_ctr;
776 struct sec_desc_buf secdesc_ctr;
777
778 ZERO_STRUCT(devmode_ctr);
779 ZERO_STRUCT(secdesc_ctr);
780
781 /* setprinter call */
782
783 info_ctr.level = level;
784 switch (level) {
785 case 0:
786 info_ctr.info.info0 = (struct spoolss_SetPrinterInfo0 *)
787 (void *)&info->info0;
788 break;
789 case 1:
790 info_ctr.info.info1 = (struct spoolss_SetPrinterInfo1 *)
791 (void *)&info->info1;
792 break;
793 case 2:
794 spoolss_printerinfo2_to_setprinterinfo2(&info->info2, &info2);
795 info_ctr.info.info2 = &info2;
796 break;
797 case 3:
798 info_ctr.info.info3 = (struct spoolss_SetPrinterInfo3 *)
799 (void *)&info->info3;
800 break;
801 case 4:
802 info_ctr.info.info4 = (struct spoolss_SetPrinterInfo4 *)
803 (void *)&info->info4;
804 break;
805 case 5:
806 info_ctr.info.info5 = (struct spoolss_SetPrinterInfo5 *)
807 (void *)&info->info5;
808 break;
809 case 6:
810 info_ctr.info.info6 = (struct spoolss_SetPrinterInfo6 *)
811 (void *)&info->info6;
812 break;
813 case 7:
814 info_ctr.info.info7 = (struct spoolss_SetPrinterInfo7 *)
815 (void *)&info->info7;
816 break;
817 #if 0 /* FIXME GD */
818 case 8:
819 info_ctr.info.info8 = (struct spoolss_SetPrinterInfo8 *)
820 (void *)&info->info8;
821 break;
822 case 9:
823 info_ctr.info.info9 = (struct spoolss_SetPrinterInfo9 *)
824 (void *)&info->info9;
825 break;
826 #endif
827 default:
828 break; /* FIXME */
829 }
830
831 status = dcerpc_spoolss_SetPrinter(b, mem_ctx,
832 hnd,
833 &info_ctr,
834 &devmode_ctr,
835 &secdesc_ctr,
836 0, /* command */
837 &result);
838 if (!NT_STATUS_IS_OK(status)) {
839 printf(_("cannot set printer-info: %s\n"), nt_errstr(status));
840 return false;
841 }
842 if (!W_ERROR_IS_OK(result)) {
843 printf(_("cannot set printer-info: %s\n"), win_errstr(result));
844 return false;
845 }
846
847 return true;
848 }
849
850
851 static bool net_spoolss_setprinterdata(struct rpc_pipe_client *pipe_hnd,
852 TALLOC_CTX *mem_ctx,
853 struct policy_handle *hnd,
854 const char *value_name,
855 enum winreg_Type type,
856 uint8_t *data,
857 uint32_t offered)
858 {
859 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
860 WERROR result;
861 NTSTATUS status;
862
863 /* setprinterdata call */
864 status = dcerpc_spoolss_SetPrinterData(b, mem_ctx,
865 hnd,
866 value_name,
867 type,
868 data,
869 offered,
870 &result);
871 if (!NT_STATUS_IS_OK(status)) {
872 printf (_("unable to set printerdata: %s\n"),
873 nt_errstr(status));
874 return false;
875 }
876 if (!W_ERROR_IS_OK(result)) {
877 printf (_("unable to set printerdata: %s\n"),
878 win_errstr(result));
879 return false;
880 }
881
882 return true;
883 }
884
885
886 static bool net_spoolss_enumprinterkey(struct rpc_pipe_client *pipe_hnd,
887 TALLOC_CTX *mem_ctx,
888 struct policy_handle *hnd,
889 const char *keyname,
890 const char ***keylist)
891 {
892 WERROR result;
893
894 /* enumprinterkey call */
895 result = rpccli_spoolss_enumprinterkey(pipe_hnd, mem_ctx, hnd, keyname, keylist, 0);
896
897 if (!W_ERROR_IS_OK(result)) {
898 printf(_("enumprinterkey failed: %s\n"), win_errstr(result));
899 return false;
900 }
901
902 return true;
903 }
904
905 static bool net_spoolss_enumprinterdataex(struct rpc_pipe_client *pipe_hnd,
906 TALLOC_CTX *mem_ctx,
907 uint32_t offered,
908 struct policy_handle *hnd,
909 const char *keyname,
910 uint32_t *count,
911 struct spoolss_PrinterEnumValues **info)
912 {
913 WERROR result;
914
915 /* enumprinterdataex call */
916 result = rpccli_spoolss_enumprinterdataex(pipe_hnd, mem_ctx,
917 hnd,
918 keyname,
919 0, /* offered */
920 count,
921 info);
922
923 if (!W_ERROR_IS_OK(result)) {
924 printf(_("enumprinterdataex failed: %s\n"), win_errstr(result));
925 return false;
926 }
927
928 return true;
929 }
930
931
932 static bool net_spoolss_setprinterdataex(struct rpc_pipe_client *pipe_hnd,
933 TALLOC_CTX *mem_ctx,
934 struct policy_handle *hnd,
935 const char *keyname,
936 struct regval_blob *value)
937 {
938 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
939 WERROR result;
940 NTSTATUS status;
941
942 /* setprinterdataex call */
943 status = dcerpc_spoolss_SetPrinterDataEx(b, mem_ctx,
944 hnd,
945 keyname,
946 regval_name(value),
947 regval_type(value),
948 regval_data_p(value),
949 regval_size(value),
950 &result);
951 if (!NT_STATUS_IS_OK(status)) {
952 printf(_("could not set printerdataex: %s\n"),
953 nt_errstr(status));
954 return false;
955 }
956 if (!W_ERROR_IS_OK(result)) {
957 printf(_("could not set printerdataex: %s\n"),
958 win_errstr(result));
959 return false;
960 }
961
962 return true;
963 }
964
965 static bool net_spoolss_enumforms(struct rpc_pipe_client *pipe_hnd,
966 TALLOC_CTX *mem_ctx,
967 struct policy_handle *hnd,
968 int level,
969 uint32_t *num_forms,
970 union spoolss_FormInfo **forms)
971 {
972 WERROR result;
973
974 /* enumforms call */
975 result = rpccli_spoolss_enumforms(pipe_hnd, mem_ctx,
976 hnd,
977 level,
978 0,
979 num_forms,
980 forms);
981 if (!W_ERROR_IS_OK(result)) {
982 printf(_("could not enum forms: %s\n"), win_errstr(result));
983 return false;
984 }
985
986 return true;
987 }
988
989 static bool net_spoolss_enumprinterdrivers (struct rpc_pipe_client *pipe_hnd,
990 TALLOC_CTX *mem_ctx,
991 uint32_t level, const char *env,
992 uint32_t *count,
993 union spoolss_DriverInfo **info)
994 {
995 WERROR result;
996
997 /* enumprinterdrivers call */
998 result = rpccli_spoolss_enumprinterdrivers(pipe_hnd, mem_ctx,
999 pipe_hnd->srv_name_slash,
1000 env,
1001 level,
1002 0,
1003 count,
1004 info);
1005 if (!W_ERROR_IS_OK(result)) {
1006 if (W_ERROR_V(result) != W_ERROR_V(WERR_INVALID_ENVIRONMENT)) {
1007 printf(_("cannot enum drivers for environment %s: %s\n"), env,
1008 win_errstr(result));
1009 return false;
1010 } else {
1011 printf(_("Server does not support environment [%s]\n"),
1012 env);
1013 }
1014 }
1015
1016 return true;
1017 }
1018
1019 static bool net_spoolss_getprinterdriver(struct rpc_pipe_client *pipe_hnd,
1020 TALLOC_CTX *mem_ctx,
1021 struct policy_handle *hnd, uint32_t level,
1022 const char *env, int version,
1023 union spoolss_DriverInfo *info)
1024 {
1025 WERROR result;
1026 uint32_t server_major_version;
1027 uint32_t server_minor_version;
1028
1029 /* getprinterdriver call */
1030 result = rpccli_spoolss_getprinterdriver2(pipe_hnd, mem_ctx,
1031 hnd,
1032 env,
1033 level,
1034 0,
1035 version,
1036 2,
1037 info,
1038 &server_major_version,
1039 &server_minor_version);
1040 if (!W_ERROR_IS_OK(result)) {
1041 DEBUG(1,("cannot get driver (for architecture: %s): %s\n",
1042 env, win_errstr(result)));
1043 if (W_ERROR_V(result) != W_ERROR_V(WERR_UNKNOWN_PRINTER_DRIVER) &&
1044 W_ERROR_V(result) != W_ERROR_V(WERR_INVALID_ENVIRONMENT)) {
1045 printf(_("cannot get driver: %s\n"),
1046 win_errstr(result));
1047 }
1048 return false;
1049 }
1050
1051 return true;
1052 }
1053
1054
1055 static bool net_spoolss_addprinterdriver(struct rpc_pipe_client *pipe_hnd,
1056 TALLOC_CTX *mem_ctx, uint32_t level,
1057 union spoolss_DriverInfo *info)
1058 {
1059 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
1060 WERROR result;
1061 NTSTATUS status;
1062 struct spoolss_AddDriverInfoCtr info_ctr;
1063
1064 info_ctr.level = level;
1065
1066 switch (level) {
1067 case 2:
1068 info_ctr.info.info2 = (struct spoolss_AddDriverInfo2 *)
1069 (void *)&info->info2;
1070 break;
1071 case 3:
1072 info_ctr.info.info3 = (struct spoolss_AddDriverInfo3 *)
1073 (void *)&info->info3;
1074 break;
1075 default:
1076 printf(_("unsupported info level: %d\n"), level);
1077 return false;
1078 }
1079
1080 /* addprinterdriver call */
1081 status = dcerpc_spoolss_AddPrinterDriver(b, mem_ctx,
1082 pipe_hnd->srv_name_slash,
1083 &info_ctr,
1084 &result);
1085 if (!NT_STATUS_IS_OK(status)) {
1086 printf(_("cannot add driver: %s\n"), nt_errstr(status));
1087 return false;
1088 }
1089 /* be more verbose */
1090 if (W_ERROR_V(result) == W_ERROR_V(WERR_ACCESS_DENIED)) {
1091 printf(_("You are not allowed to add drivers\n"));
1092 return false;
1093 }
1094 if (!W_ERROR_IS_OK(result)) {
1095 printf(_("cannot add driver: %s\n"), win_errstr(result));
1096 return false;
1097 }
1098
1099 return true;
1100 }
1101
1102 /**
1103 * abstraction function to get uint32_t num_printers and PRINTER_INFO_CTR ctr
1104 * for a single printer or for all printers depending on argc/argv
1105 **/
1106
1107 static bool get_printer_info(struct rpc_pipe_client *pipe_hnd,
1108 TALLOC_CTX *mem_ctx,
1109 int level,
1110 int argc,
1111 const char **argv,
1112 uint32_t *num_printers,
1113 union spoolss_PrinterInfo **info_p)
1114 {
1115 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
1116 struct policy_handle hnd;
1117 WERROR werr;
1118
1119 /* no arguments given, enumerate all printers */
1120 if (argc == 0) {
1121
1122 if (!net_spoolss_enum_printers(pipe_hnd, mem_ctx, NULL,
1123 PRINTER_ENUM_LOCAL|PRINTER_ENUM_SHARED,
1124 level, num_printers, info_p))
1125 return false;
1126
1127 goto out;
1128 }
1129
1130 /* argument given, get a single printer by name */
1131 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, argv[0],
1132 MAXIMUM_ALLOWED_ACCESS,
1133 pipe_hnd->auth->user_name,
1134 &hnd))
1135 return false;
1136
1137 *info_p = talloc_zero(mem_ctx, union spoolss_PrinterInfo);
1138 if (*info_p == NULL) {
1139 return false;
1140 }
1141
1142 if (!net_spoolss_getprinter(pipe_hnd, mem_ctx, &hnd, level, *info_p)) {
1143 dcerpc_spoolss_ClosePrinter(b, mem_ctx, &hnd, &werr);
1144 return false;
1145 }
1146
1147 dcerpc_spoolss_ClosePrinter(b, mem_ctx, &hnd, &werr);
1148
1149 *num_printers = 1;
1150
1151 out:
1152 DEBUG(3,("got %d printers\n", *num_printers));
1153
1154 return true;
1155
1156 }
1157
1158 /**
1159 * List print-queues (including local printers that are not shared)
1160 *
1161 * All parameters are provided by the run_rpc_command function, except for
1162 * argc, argv which are passed through.
1163 *
1164 * @param c A net_context structure
1165 * @param domain_sid The domain sid aquired from the remote server
1166 * @param cli A cli_state connected to the server.
1167 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1168 * @param argc Standard main() style argc
1169 * @param argv Standard main() style argv. Initial components are already
1170 * stripped
1171 *
1172 * @return Normal NTSTATUS return.
1173 **/
1174
1175 NTSTATUS rpc_printer_list_internals(struct net_context *c,
1176 const struct dom_sid *domain_sid,
1177 const char *domain_name,
1178 struct cli_state *cli,
1179 struct rpc_pipe_client *pipe_hnd,
1180 TALLOC_CTX *mem_ctx,
1181 int argc,
1182 const char **argv)
1183 {
1184 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1185 uint32_t i, num_printers;
1186 uint32_t level = 2;
1187 const char *printername, *sharename;
1188 union spoolss_PrinterInfo *info;
1189
1190 printf("listing printers\n");
1191
1192 if (!get_printer_info(pipe_hnd, mem_ctx, level, argc, argv, &num_printers, &info))
1193 return nt_status;
1194
1195 for (i = 0; i < num_printers; i++) {
1196
1197 /* do some initialization */
1198 printername = info[i].info2.printername;
1199 sharename = info[i].info2.sharename;
1200
1201 if (printername && sharename) {
1202 d_printf(_("printer %d: %s, shared as: %s\n"),
1203 i+1, printername, sharename);
1204 }
1205 }
1206
1207 return NT_STATUS_OK;
1208 }
1209
1210 /**
1211 * List printer-drivers from a server
1212 *
1213 * All parameters are provided by the run_rpc_command function, except for
1214 * argc, argv which are passed through.
1215 *
1216 * @param c A net_context structure
1217 * @param domain_sid The domain sid aquired from the remote server
1218 * @param cli A cli_state connected to the server.
1219 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1220 * @param argc Standard main() style argc
1221 * @param argv Standard main() style argv. Initial components are already
1222 * stripped
1223 *
1224 * @return Normal NTSTATUS return.
1225 **/
1226
1227 NTSTATUS rpc_printer_driver_list_internals(struct net_context *c,
1228 const struct dom_sid *domain_sid,
1229 const char *domain_name,
1230 struct cli_state *cli,
1231 struct rpc_pipe_client *pipe_hnd,
1232 TALLOC_CTX *mem_ctx,
1233 int argc,
1234 const char **argv)
1235 {
1236 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1237 uint32_t i;
1238 uint32_t level = 3;
1239 union spoolss_DriverInfo *info;
1240 int d;
1241
1242 printf(_("listing printer-drivers\n"));
1243
1244 for (i=0; archi_table[i].long_archi!=NULL; i++) {
1245
1246 uint32_t num_drivers;
1247
1248 /* enum remote drivers */
1249 if (!net_spoolss_enumprinterdrivers(pipe_hnd, mem_ctx, level,
1250 archi_table[i].long_archi,
1251 &num_drivers, &info)) {
1252 nt_status = NT_STATUS_UNSUCCESSFUL;
1253 goto done;
1254 }
1255
1256 if (num_drivers == 0) {
1257 d_printf(_("no drivers found on server for "
1258 "architecture: [%s].\n"),
1259 archi_table[i].long_archi);
1260 continue;
1261 }
1262
1263 d_printf(_("got %d printer-drivers for architecture: [%s]\n"),
1264 num_drivers, archi_table[i].long_archi);
1265
1266
1267 /* do something for all drivers for architecture */
1268 for (d = 0; d < num_drivers; d++) {
1269 display_print_driver3(&info[d].info3);
1270 }
1271 }
1272
1273 nt_status = NT_STATUS_OK;
1274
1275 done:
1276 return nt_status;
1277
1278 }
1279
1280 /**
1281 * Publish print-queues with args-wrapper
1282 *
1283 * @param cli A cli_state connected to the server.
1284 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1285 * @param argc Standard main() style argc
1286 * @param argv Standard main() style argv. Initial components are already
1287 * stripped
1288 * @param action
1289 *
1290 * @return Normal NTSTATUS return.
1291 **/
1292
1293 static NTSTATUS rpc_printer_publish_internals_args(struct rpc_pipe_client *pipe_hnd,
1294 TALLOC_CTX *mem_ctx,
1295 int argc,
1296 const char **argv,
1297 uint32_t action)
1298 {
1299 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
1300 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1301 uint32_t i, num_printers;
1302 uint32_t level = 7;
1303 const char *printername, *sharename;
1304 union spoolss_PrinterInfo *info_enum;
1305 union spoolss_PrinterInfo info;
1306 struct spoolss_SetPrinterInfoCtr info_ctr;
1307 struct spoolss_DevmodeContainer devmode_ctr;
1308 struct sec_desc_buf secdesc_ctr;
1309 struct policy_handle hnd;
1310 WERROR result;
1311 const char *action_str;
1312
1313 if (!get_printer_info(pipe_hnd, mem_ctx, 2, argc, argv, &num_printers, &info_enum))
1314 return nt_status;
1315
1316 for (i = 0; i < num_printers; i++) {
1317
1318 /* do some initialization */
1319 printername = info_enum[i].info2.printername;
1320 sharename = info_enum[i].info2.sharename;
1321 if (!printername || !sharename) {
1322 goto done;
1323 }
1324
1325 /* open printer handle */
1326 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
1327 PRINTER_ALL_ACCESS, pipe_hnd->auth->user_name, &hnd))
1328 goto done;
1329
1330 /* check for existing dst printer */
1331 if (!net_spoolss_getprinter(pipe_hnd, mem_ctx, &hnd, level, &info))
1332 goto done;
1333
1334 /* check action and set string */
1335 switch (action) {
1336 case DSPRINT_PUBLISH:
1337 action_str = N_("published");
1338 break;
1339 case DSPRINT_UPDATE:
1340 action_str = N_("updated");
1341 break;
1342 case DSPRINT_UNPUBLISH:
1343 action_str = N_("unpublished");
1344 break;
1345 default:
1346 action_str = N_("unknown action");
1347 printf(_("unknown action: %d\n"), action);
1348 break;
1349 }
1350
1351 info.info7.action = action;
1352 info_ctr.level = 7;
1353 info_ctr.info.info7 = (struct spoolss_SetPrinterInfo7 *)
1354 (void *)&info.info7;
1355
1356 ZERO_STRUCT(devmode_ctr);
1357 ZERO_STRUCT(secdesc_ctr);
1358
1359 nt_status = dcerpc_spoolss_SetPrinter(b, mem_ctx,
1360 &hnd,
1361 &info_ctr,
1362 &devmode_ctr,
1363 &secdesc_ctr,
1364 0, /* command */
1365 &result);
1366 if (!NT_STATUS_IS_OK(nt_status)) {
1367 printf(_("cannot set printer-info: %s\n"),
1368 nt_errstr(nt_status));
1369 goto done;
1370 }
1371 if (!W_ERROR_IS_OK(result) && !W_ERROR_EQUAL(result, WERR_IO_PENDING)) {
1372 if ((action == DSPRINT_UPDATE) && W_ERROR_EQUAL(result, W_ERROR(0x80070002))) {
1373 printf(_("printer not published yet\n"));
1374 } else {
1375 printf(_("cannot set printer-info: %s\n"),
1376 win_errstr(result));
1377 }
1378 nt_status = werror_to_ntstatus(result);
1379 goto done;
1380 }
1381
1382 printf(_("successfully %s printer %s in Active Directory\n"),
1383 action_str, sharename);
1384 }
1385
1386 nt_status = NT_STATUS_OK;
1387
1388 done:
1389 if (is_valid_policy_hnd(&hnd)) {
1390 dcerpc_spoolss_ClosePrinter(b, mem_ctx, &hnd, &result);
1391 }
1392
1393 return nt_status;
1394 }
1395
1396 NTSTATUS rpc_printer_publish_publish_internals(struct net_context *c,
1397 const struct dom_sid *domain_sid,
1398 const char *domain_name,
1399 struct cli_state *cli,
1400 struct rpc_pipe_client *pipe_hnd,
1401 TALLOC_CTX *mem_ctx,
1402 int argc,
1403 const char **argv)
1404 {
1405 return rpc_printer_publish_internals_args(pipe_hnd, mem_ctx, argc, argv, DSPRINT_PUBLISH);
1406 }
1407
1408 NTSTATUS rpc_printer_publish_unpublish_internals(struct net_context *c,
1409 const struct dom_sid *domain_sid,
1410 const char *domain_name,
1411 struct cli_state *cli,
1412 struct rpc_pipe_client *pipe_hnd,
1413 TALLOC_CTX *mem_ctx,
1414 int argc,
1415 const char **argv)
1416 {
1417 return rpc_printer_publish_internals_args(pipe_hnd, mem_ctx, argc, argv, DSPRINT_UNPUBLISH);
1418 }
1419
1420 NTSTATUS rpc_printer_publish_update_internals(struct net_context *c,
1421 const struct dom_sid *domain_sid,
1422 const char *domain_name,
1423 struct cli_state *cli,
1424 struct rpc_pipe_client *pipe_hnd,
1425 TALLOC_CTX *mem_ctx,
1426 int argc,
1427 const char **argv)
1428 {
1429 return rpc_printer_publish_internals_args(pipe_hnd, mem_ctx, argc, argv, DSPRINT_UPDATE);
1430 }
1431
1432 /**
1433 * List print-queues w.r.t. their publishing state
1434 *
1435 * All parameters are provided by the run_rpc_command function, except for
1436 * argc, argv which are passed through.
1437 *
1438 * @param c A net_context structure
1439 * @param domain_sid The domain sid aquired from the remote server
1440 * @param cli A cli_state connected to the server.
1441 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1442 * @param argc Standard main() style argc
1443 * @param argv Standard main() style argv. Initial components are already
1444 * stripped
1445 *
1446 * @return Normal NTSTATUS return.
1447 **/
1448
1449 NTSTATUS rpc_printer_publish_list_internals(struct net_context *c,
1450 const struct dom_sid *domain_sid,
1451 const char *domain_name,
1452 struct cli_state *cli,
1453 struct rpc_pipe_client *pipe_hnd,
1454 TALLOC_CTX *mem_ctx,
1455 int argc,
1456 const char **argv)
1457 {
1458 struct dcerpc_binding_handle *b = pipe_hnd->binding_handle;
1459 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1460 uint32_t i, num_printers;
1461 uint32_t level = 7;
1462 const char *printername, *sharename;
1463 union spoolss_PrinterInfo *info_enum;
1464 union spoolss_PrinterInfo info;
1465 struct policy_handle hnd;
1466 int state;
1467 WERROR werr;
1468
1469 if (!get_printer_info(pipe_hnd, mem_ctx, 2, argc, argv, &num_printers, &info_enum))
1470 return nt_status;
1471
1472 for (i = 0; i < num_printers; i++) {
1473
1474 /* do some initialization */
1475 printername = info_enum[i].info2.printername;
1476 sharename = info_enum[i].info2.sharename;
1477
1478 if (!printername || !sharename) {
1479 goto done;
1480 }
1481
1482 /* open printer handle */
1483 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
1484 PRINTER_ALL_ACCESS, cli->user_name, &hnd))
1485 goto done;
1486
1487 /* check for existing dst printer */
1488 if (!net_spoolss_getprinter(pipe_hnd, mem_ctx, &hnd, level, &info))
1489 goto done;
1490
1491 if (!info.info7.guid) {
1492 goto done;
1493 }
1494 state = info.info7.action;
1495 switch (state) {
1496 case DSPRINT_PUBLISH:
1497 printf(_("printer [%s] is published"),
1498 sharename);
1499 if (c->opt_verbose)
1500 printf(_(", guid: %s"),info.info7.guid);
1501 printf("\n");
1502 break;
1503 case DSPRINT_UNPUBLISH:
1504 printf(_("printer [%s] is unpublished\n"),
1505 sharename);
1506 break;
1507 case DSPRINT_UPDATE:
1508 printf(_("printer [%s] is currently updating\n"),
1509 sharename);
1510 break;
1511 default:
1512 printf(_("unknown state: %d\n"), state);
1513 break;
1514 }
1515 }
1516
1517 nt_status = NT_STATUS_OK;
1518
1519 done:
1520 if (is_valid_policy_hnd(&hnd)) {
1521 dcerpc_spoolss_ClosePrinter(b, mem_ctx, &hnd, &werr);
1522 }
1523
1524 return nt_status;
1525 }
1526
1527 /**
1528 * Migrate Printer-ACLs from a source server to the destination server
1529 *
1530 * All parameters are provided by the run_rpc_command function, except for
1531 * argc, argv which are passed through.
1532 *
1533 * @param c A net_context structure
1534 * @param domain_sid The domain sid aquired from the remote server
1535 * @param cli A cli_state connected to the server.
1536 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1537 * @param argc Standard main() style argc
1538 * @param argv Standard main() style argv. Initial components are already
1539 * stripped
1540 *
1541 * @return Normal NTSTATUS return.
1542 **/
1543
1544 NTSTATUS rpc_printer_migrate_security_internals(struct net_context *c,
1545 const struct dom_sid *domain_sid,
1546 const char *domain_name,
1547 struct cli_state *cli,
1548 struct rpc_pipe_client *pipe_hnd,
1549 TALLOC_CTX *mem_ctx,
1550 int argc,
1551 const char **argv)
1552 {
1553 struct dcerpc_binding_handle *b_src = pipe_hnd->binding_handle;
1554 /* TODO: what now, info2 or info3 ?
1555 convince jerry that we should add clientside setacls level 3 at least
1556 */
1557 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1558 uint32_t i = 0;
1559 uint32_t num_printers;
1560 uint32_t level = 2;
1561 const char *printername, *sharename;
1562 struct rpc_pipe_client *pipe_hnd_dst = NULL;
1563 struct dcerpc_binding_handle *b_dst = NULL;
1564 struct policy_handle hnd_src, hnd_dst;
1565 union spoolss_PrinterInfo *info_enum;
1566 struct cli_state *cli_dst = NULL;
1567 union spoolss_PrinterInfo info_src, info_dst;
1568 WERROR werr;
1569
1570 DEBUG(3,("copying printer ACLs\n"));
1571
1572 /* connect destination PI_SPOOLSS */
1573 nt_status = connect_dst_pipe(c, &cli_dst, &pipe_hnd_dst,
1574 &ndr_table_spoolss.syntax_id);
1575 if (!NT_STATUS_IS_OK(nt_status)) {
1576 return nt_status;
1577 }
1578 b_dst = pipe_hnd_dst->binding_handle;
1579
1580 /* enum source printers */
1581 if (!get_printer_info(pipe_hnd, mem_ctx, level, argc, argv, &num_printers, &info_enum)) {
1582 nt_status = NT_STATUS_UNSUCCESSFUL;
1583 goto done;
1584 }
1585
1586 if (!num_printers) {
1587 printf (_("no printers found on server.\n"));
1588 nt_status = NT_STATUS_OK;
1589 goto done;
1590 }
1591
1592 /* do something for all printers */
1593 for (i = 0; i < num_printers; i++) {
1594
1595 /* do some initialization */
1596 printername = info_enum[i].info2.printername;
1597 sharename = info_enum[i].info2.sharename;
1598
1599 if (!printername || !sharename) {
1600 nt_status = NT_STATUS_UNSUCCESSFUL;
1601 goto done;
1602 }
1603
1604 /* we can reset NT_STATUS here because we do not
1605 get any real NT_STATUS-codes anymore from now on */
1606 nt_status = NT_STATUS_UNSUCCESSFUL;
1607
1608 d_printf(_("migrating printer ACLs for: [%s] / [%s]\n"),
1609 printername, sharename);
1610
1611 /* according to msdn you have specify these access-rights
1612 to see the security descriptor
1613 - READ_CONTROL (DACL)
1614 - ACCESS_SYSTEM_SECURITY (SACL)
1615 */
1616
1617 /* open src printer handle */
1618 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
1619 MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
1620 goto done;
1621
1622 /* open dst printer handle */
1623 if (!net_spoolss_open_printer_ex(pipe_hnd_dst, mem_ctx, sharename,
1624 PRINTER_ALL_ACCESS, cli_dst->user_name, &hnd_dst))
1625 goto done;
1626
1627 /* check for existing dst printer */
1628 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, level, &info_dst))
1629 goto done;
1630
1631 /* check for existing src printer */
1632 if (!net_spoolss_getprinter(pipe_hnd, mem_ctx, &hnd_src, 3, &info_src))
1633 goto done;
1634
1635 /* Copy Security Descriptor */
1636
1637 /* copy secdesc (info level 2) */
1638 info_dst.info2.devmode = NULL;
1639 info_dst.info2.secdesc = dup_sec_desc(mem_ctx, info_src.info3.secdesc);
1640
1641 if (c->opt_verbose)
1642 display_sec_desc(info_dst.info2.secdesc);
1643
1644 if (!net_spoolss_setprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, 2, &info_dst))
1645 goto done;
1646
1647 DEBUGADD(1,("\tSetPrinter of SECDESC succeeded\n"));
1648
1649
1650 /* close printer handles here */
1651 if (is_valid_policy_hnd(&hnd_src)) {
1652 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &werr);
1653 }
1654
1655 if (is_valid_policy_hnd(&hnd_dst)) {
1656 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &werr);
1657 }
1658
1659 }
1660
1661 nt_status = NT_STATUS_OK;
1662
1663 done:
1664
1665 if (is_valid_policy_hnd(&hnd_src)) {
1666 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &werr);
1667 }
1668
1669 if (is_valid_policy_hnd(&hnd_dst)) {
1670 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &werr);
1671 }
1672
1673 if (cli_dst) {
1674 cli_shutdown(cli_dst);
1675 }
1676 return nt_status;
1677 }
1678
1679 /**
1680 * Migrate printer-forms from a src server to the dst server
1681 *
1682 * All parameters are provided by the run_rpc_command function, except for
1683 * argc, argv which are passed through.
1684 *
1685 * @param c A net_context structure
1686 * @param domain_sid The domain sid aquired from the remote server
1687 * @param cli A cli_state connected to the server.
1688 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1689 * @param argc Standard main() style argc
1690 * @param argv Standard main() style argv. Initial components are already
1691 * stripped
1692 *
1693 * @return Normal NTSTATUS return.
1694 **/
1695
1696 NTSTATUS rpc_printer_migrate_forms_internals(struct net_context *c,
1697 const struct dom_sid *domain_sid,
1698 const char *domain_name,
1699 struct cli_state *cli,
1700 struct rpc_pipe_client *pipe_hnd,
1701 TALLOC_CTX *mem_ctx,
1702 int argc,
1703 const char **argv)
1704 {
1705 struct dcerpc_binding_handle *b_src = pipe_hnd->binding_handle;
1706 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1707 WERROR result;
1708 uint32_t i, f;
1709 uint32_t num_printers;
1710 uint32_t level = 1;
1711 const char *printername, *sharename;
1712 struct rpc_pipe_client *pipe_hnd_dst = NULL;
1713 struct dcerpc_binding_handle *b_dst = NULL;
1714 struct policy_handle hnd_src, hnd_dst;
1715 union spoolss_PrinterInfo *info_enum;
1716 union spoolss_PrinterInfo info_dst;
1717 uint32_t num_forms;
1718 union spoolss_FormInfo *forms;
1719 struct cli_state *cli_dst = NULL;
1720
1721 DEBUG(3,("copying forms\n"));
1722
1723 /* connect destination PI_SPOOLSS */
1724 nt_status = connect_dst_pipe(c, &cli_dst, &pipe_hnd_dst,
1725 &ndr_table_spoolss.syntax_id);
1726 if (!NT_STATUS_IS_OK(nt_status)) {
1727 return nt_status;
1728 }
1729 b_dst = pipe_hnd_dst->binding_handle;
1730
1731 /* enum src printers */
1732 if (!get_printer_info(pipe_hnd, mem_ctx, 2, argc, argv, &num_printers, &info_enum)) {
1733 nt_status = NT_STATUS_UNSUCCESSFUL;
1734 goto done;
1735 }
1736
1737 if (!num_printers) {
1738 printf (_("no printers found on server.\n"));
1739 nt_status = NT_STATUS_OK;
1740 goto done;
1741 }
1742
1743 /* do something for all printers */
1744 for (i = 0; i < num_printers; i++) {
1745
1746 /* do some initialization */
1747 printername = info_enum[i].info2.printername;
1748 sharename = info_enum[i].info2.sharename;
1749
1750 if (!printername || !sharename) {
1751 nt_status = NT_STATUS_UNSUCCESSFUL;
1752 goto done;
1753 }
1754 /* we can reset NT_STATUS here because we do not
1755 get any real NT_STATUS-codes anymore from now on */
1756 nt_status = NT_STATUS_UNSUCCESSFUL;
1757
1758 d_printf(_("migrating printer forms for: [%s] / [%s]\n"),
1759 printername, sharename);
1760
1761
1762 /* open src printer handle */
1763 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
1764 MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
1765 goto done;
1766
1767 /* open dst printer handle */
1768 if (!net_spoolss_open_printer_ex(pipe_hnd_dst, mem_ctx, sharename,
1769 PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst))
1770 goto done;
1771
1772 /* check for existing dst printer */
1773 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, level, &info_dst))
1774 goto done;
1775
1776 /* finally migrate forms */
1777 if (!net_spoolss_enumforms(pipe_hnd, mem_ctx, &hnd_src, level, &num_forms, &forms))
1778 goto done;
1779
1780 DEBUG(1,("got %d forms for printer\n", num_forms));
1781
1782
1783 for (f = 0; f < num_forms; f++) {
1784
1785 union spoolss_AddFormInfo info;
1786 NTSTATUS status;
1787
1788 /* only migrate FORM_PRINTER types, according to jerry
1789 FORM_BUILTIN-types are hard-coded in samba */
1790 if (forms[f].info1.flags != SPOOLSS_FORM_PRINTER)
1791 continue;
1792
1793 if (c->opt_verbose)
1794 d_printf(_("\tmigrating form # %d [%s] of type "
1795 "[%d]\n"),
1796 f, forms[f].info1.form_name,
1797 forms[f].info1.flags);
1798
1799 info.info1 = (struct spoolss_AddFormInfo1 *)
1800 (void *)&forms[f].info1;
1801
1802 /* FIXME: there might be something wrong with samba's
1803 builtin-forms */
1804 status = dcerpc_spoolss_AddForm(b_dst, mem_ctx,
1805 &hnd_dst,
1806 1,
1807 info,
1808 &result);
1809 if (!NT_STATUS_IS_OK(status)) {
1810 d_printf(_("\tdcerpc_spoolss_AddForm form %d: [%s] - %s\n"),
1811 f, forms[f].info1.form_name, nt_errstr(status));
1812 continue;
1813 }
1814 if (!W_ERROR_IS_OK(result)) {
1815 d_printf(_("\tAddForm form %d: [%s] refused.\n"),
1816 f, forms[f].info1.form_name);
1817 continue;
1818 }
1819
1820 DEBUGADD(1,("\tAddForm of [%s] succeeded\n",
1821 forms[f].info1.form_name));
1822 }
1823
1824
1825 /* close printer handles here */
1826 if (is_valid_policy_hnd(&hnd_src)) {
1827 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
1828 }
1829
1830 if (is_valid_policy_hnd(&hnd_dst)) {
1831 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
1832 }
1833 }
1834
1835 nt_status = NT_STATUS_OK;
1836
1837 done:
1838
1839 if (is_valid_policy_hnd(&hnd_src)) {
1840 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
1841 }
1842
1843 if (is_valid_policy_hnd(&hnd_dst)) {
1844 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
1845 }
1846
1847 if (cli_dst) {
1848 cli_shutdown(cli_dst);
1849 }
1850 return nt_status;
1851 }
1852
1853 /**
1854 * Migrate printer-drivers from a src server to the dst server
1855 *
1856 * All parameters are provided by the run_rpc_command function, except for
1857 * argc, argv which are passed through.
1858 *
1859 * @param c A net_context structure
1860 * @param domain_sid The domain sid aquired from the remote server
1861 * @param cli A cli_state connected to the server.
1862 * @param mem_ctx Talloc context, destoyed on compleation of the function.
1863 * @param argc Standard main() style argc
1864 * @param argv Standard main() style argv. Initial components are already
1865 * stripped
1866 *
1867 * @return Normal NTSTATUS return.
1868 **/
1869
1870 NTSTATUS rpc_printer_migrate_drivers_internals(struct net_context *c,
1871 const struct dom_sid *domain_sid,
1872 const char *domain_name,
1873 struct cli_state *cli,
1874 struct rpc_pipe_client *pipe_hnd,
1875 TALLOC_CTX *mem_ctx,
1876 int argc,
1877 const char **argv)
1878 {
1879 struct dcerpc_binding_handle *b_src = pipe_hnd->binding_handle;
1880 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
1881 uint32_t i, p;
1882 uint32_t num_printers;
1883 uint32_t level = 3;
1884 const char *printername, *sharename;
1885 bool got_src_driver_share = false;
1886 bool got_dst_driver_share = false;
1887 struct rpc_pipe_client *pipe_hnd_dst = NULL;
1888 struct dcerpc_binding_handle *b_dst = NULL;
1889 struct policy_handle hnd_src, hnd_dst;
1890 union spoolss_DriverInfo drv_info_src;
1891 union spoolss_PrinterInfo *info_enum;
1892 union spoolss_PrinterInfo info_dst;
1893 struct cli_state *cli_dst = NULL;
1894 struct cli_state *cli_share_src = NULL;
1895 struct cli_state *cli_share_dst = NULL;
1896 const char *drivername = NULL;
1897 WERROR werr;
1898
1899 DEBUG(3,("copying printer-drivers\n"));
1900
1901 nt_status = connect_dst_pipe(c, &cli_dst, &pipe_hnd_dst,
1902 &ndr_table_spoolss.syntax_id);
1903 if (!NT_STATUS_IS_OK(nt_status)) {
1904 return nt_status;
1905 }
1906 b_dst = pipe_hnd_dst->binding_handle;
1907
1908 /* open print$-share on the src server */
1909 nt_status = connect_to_service(c, &cli_share_src,
1910 cli_state_remote_sockaddr(cli),
1911 cli_state_remote_name(cli),
1912 "print$", "A:");
1913 if (!NT_STATUS_IS_OK(nt_status))
1914 goto done;
1915
1916 got_src_driver_share = true;
1917
1918
1919 /* open print$-share on the dst server */
1920 nt_status = connect_to_service(c, &cli_share_dst,
1921 cli_state_remote_sockaddr(cli_dst),
1922 cli_state_remote_name(cli_dst),
1923 "print$", "A:");
1924 if (!NT_STATUS_IS_OK(nt_status))
1925 return nt_status;
1926
1927 got_dst_driver_share = true;
1928
1929
1930 /* enum src printers */
1931 if (!get_printer_info(pipe_hnd, mem_ctx, 2, argc, argv, &num_printers, &info_enum)) {
1932 nt_status = NT_STATUS_UNSUCCESSFUL;
1933 goto done;
1934 }
1935
1936 if (num_printers == 0) {
1937 printf (_("no printers found on server.\n"));
1938 nt_status = NT_STATUS_OK;
1939 goto done;
1940 }
1941
1942
1943 /* do something for all printers */
1944 for (p = 0; p < num_printers; p++) {
1945
1946 /* do some initialization */
1947 printername = info_enum[p].info2.printername;
1948 sharename = info_enum[p].info2.sharename;
1949
1950 if (!printername || !sharename) {
1951 nt_status = NT_STATUS_UNSUCCESSFUL;
1952 goto done;
1953 }
1954
1955 /* we can reset NT_STATUS here because we do not
1956 get any real NT_STATUS-codes anymore from now on */
1957 nt_status = NT_STATUS_UNSUCCESSFUL;
1958
1959 d_printf(_("migrating printer driver for: [%s] / [%s]\n"),
1960 printername, sharename);
1961
1962 /* open dst printer handle */
1963 if (!net_spoolss_open_printer_ex(pipe_hnd_dst, mem_ctx, sharename,
1964 PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst))
1965 goto done;
1966
1967 /* check for existing dst printer */
1968 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, 2, &info_dst))
1969 goto done;
1970
1971
1972 /* open src printer handle */
1973 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
1974 MAXIMUM_ALLOWED_ACCESS,
1975 pipe_hnd->auth->user_name,
1976 &hnd_src))
1977 goto done;
1978
1979 /* in a first step call getdriver for each shared printer (per arch)
1980 to get a list of all files that have to be copied */
1981
1982 for (i=0; archi_table[i].long_archi!=NULL; i++) {
1983
1984 /* getdriver src */
1985 if (!net_spoolss_getprinterdriver(pipe_hnd, mem_ctx, &hnd_src,
1986 level, archi_table[i].long_archi,
1987 archi_table[i].version, &drv_info_src))
1988 continue;
1989
1990 drivername = drv_info_src.info3.driver_name;
1991
1992 if (c->opt_verbose)
1993 display_print_driver3(&drv_info_src.info3);
1994
1995 /* check arch dir */
1996 nt_status = check_arch_dir(cli_share_dst, archi_table[i].short_archi);
1997 if (!NT_STATUS_IS_OK(nt_status))
1998 goto done;
1999
2000
2001 /* copy driver-files */
2002 nt_status = copy_print_driver_3(c, mem_ctx, cli_share_src, cli_share_dst,
2003 archi_table[i].short_archi,
2004 &drv_info_src.info3);
2005 if (!NT_STATUS_IS_OK(nt_status))
2006 goto done;
2007
2008
2009 /* adddriver dst */
2010 if (!net_spoolss_addprinterdriver(pipe_hnd_dst, mem_ctx, level, &drv_info_src)) {
2011 nt_status = NT_STATUS_UNSUCCESSFUL;
2012 goto done;
2013 }
2014
2015 DEBUGADD(1,("Sucessfully added driver [%s] for printer [%s]\n",
2016 drivername, printername));
2017
2018 }
2019
2020 if (!drivername || strlen(drivername) == 0) {
2021 DEBUGADD(1,("Did not get driver for printer %s\n",
2022 printername));
2023 goto done;
2024 }
2025
2026 /* setdriver dst */
2027 info_dst.info2.drivername = drivername;
2028
2029 if (!net_spoolss_setprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, 2, &info_dst)) {
2030 nt_status = NT_STATUS_UNSUCCESSFUL;
2031 goto done;
2032 }
2033
2034 DEBUGADD(1,("Sucessfully set driver %s for printer %s\n",
2035 drivername, printername));
2036
2037 /* close dst */
2038 if (is_valid_policy_hnd(&hnd_dst)) {
2039 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &werr);
2040 }
2041
2042 /* close src */
2043 if (is_valid_policy_hnd(&hnd_src)) {
2044 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &werr);
2045 }
2046 }
2047
2048 nt_status = NT_STATUS_OK;
2049
2050 done:
2051
2052 if (is_valid_policy_hnd(&hnd_dst)) {
2053 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &werr);
2054 }
2055
2056 /* close src */
2057 if (is_valid_policy_hnd(&hnd_src)) {
2058 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &werr);
2059 }
2060
2061 if (cli_dst) {
2062 cli_shutdown(cli_dst);
2063 }
2064
2065 if (got_src_driver_share)
2066 cli_shutdown(cli_share_src);
2067
2068 if (got_dst_driver_share)
2069 cli_shutdown(cli_share_dst);
2070
2071 return nt_status;
2072
2073 }
2074
2075 /**
2076 * Migrate printer-queues from a src to the dst server
2077 * (requires a working "addprinter command" to be installed for the local smbd)
2078 *
2079 * All parameters are provided by the run_rpc_command function, except for
2080 * argc, argv which are passed through.
2081 *
2082 * @param c A net_context structure
2083 * @param domain_sid The domain sid aquired from the remote server
2084 * @param cli A cli_state connected to the server.
2085 * @param mem_ctx Talloc context, destoyed on compleation of the function.
2086 * @param argc Standard main() style argc
2087 * @param argv Standard main() style argv. Initial components are already
2088 * stripped
2089 *
2090 * @return Normal NTSTATUS return.
2091 **/
2092
2093 NTSTATUS rpc_printer_migrate_printers_internals(struct net_context *c,
2094 const struct dom_sid *domain_sid,
2095 const char *domain_name,
2096 struct cli_state *cli,
2097 struct rpc_pipe_client *pipe_hnd,
2098 TALLOC_CTX *mem_ctx,
2099 int argc,
2100 const char **argv)
2101 {
2102 struct dcerpc_binding_handle *b_src = pipe_hnd->binding_handle;
2103 WERROR result;
2104 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
2105 uint32_t i = 0, num_printers;
2106 uint32_t level = 2;
2107 union spoolss_PrinterInfo info_dst, info_src;
2108 union spoolss_PrinterInfo *info_enum;
2109 struct cli_state *cli_dst = NULL;
2110 struct policy_handle hnd_dst, hnd_src;
2111 const char *printername, *sharename;
2112 struct rpc_pipe_client *pipe_hnd_dst = NULL;
2113 struct dcerpc_binding_handle *b_dst = NULL;
2114 struct spoolss_SetPrinterInfoCtr info_ctr;
2115
2116 DEBUG(3,("copying printers\n"));
2117
2118 /* connect destination PI_SPOOLSS */
2119 nt_status = connect_dst_pipe(c, &cli_dst, &pipe_hnd_dst,
2120 &ndr_table_spoolss.syntax_id);
2121 if (!NT_STATUS_IS_OK(nt_status)) {
2122 return nt_status;
2123 }
2124 b_dst = pipe_hnd_dst->binding_handle;
2125
2126 /* enum printers */
2127 if (!get_printer_info(pipe_hnd, mem_ctx, level, argc, argv, &num_printers, &info_enum)) {
2128 nt_status = NT_STATUS_UNSUCCESSFUL;
2129 goto done;
2130 }
2131
2132 if (!num_printers) {
2133 printf (_("no printers found on server.\n"));
2134 nt_status = NT_STATUS_OK;
2135 goto done;
2136 }
2137
2138 /* do something for all printers */
2139 for (i = 0; i < num_printers; i++) {
2140
2141 struct spoolss_SetPrinterInfo2 info2;
2142
2143 /* do some initialization */
2144 printername = info_enum[i].info2.printername;
2145 sharename = info_enum[i].info2.sharename;
2146
2147 if (!printername || !sharename) {
2148 nt_status = NT_STATUS_UNSUCCESSFUL;
2149 goto done;
2150 }
2151 /* we can reset NT_STATUS here because we do not
2152 get any real NT_STATUS-codes anymore from now on */
2153 nt_status = NT_STATUS_UNSUCCESSFUL;
2154
2155 d_printf(_("migrating printer queue for: [%s] / [%s]\n"),
2156 printername, sharename);
2157
2158 /* open dst printer handle */
2159 if (!net_spoolss_open_printer_ex(pipe_hnd_dst, mem_ctx, sharename,
2160 PRINTER_ALL_ACCESS, cli->user_name, &hnd_dst)) {
2161
2162 DEBUG(1,("could not open printer: %s\n", sharename));
2163 }
2164
2165 /* check for existing dst printer */
2166 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, level, &info_dst)) {
2167 printf (_("could not get printer, creating printer.\n"));
2168 } else {
2169 DEBUG(1,("printer already exists: %s\n", sharename));
2170 /* close printer handle here - dst only, not got src yet. */
2171 if (is_valid_policy_hnd(&hnd_dst)) {
2172 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
2173 }
2174 continue;
2175 }
2176
2177 /* now get again src printer ctr via getprinter,
2178 we first need a handle for that */
2179
2180 /* open src printer handle */
2181 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
2182 MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
2183 goto done;
2184
2185 /* getprinter on the src server */
2186 if (!net_spoolss_getprinter(pipe_hnd, mem_ctx, &hnd_src, level, &info_src))
2187 goto done;
2188
2189 /* copy each src printer to a dst printer 1:1,
2190 maybe some values have to be changed though */
2191 d_printf(_("creating printer: %s\n"), printername);
2192
2193 info_ctr.level = level;
2194 spoolss_printerinfo2_to_setprinterinfo2(&info_src.info2, &info2);
2195 info_ctr.info.info2 = &info2;
2196
2197 result = rpccli_spoolss_addprinterex(pipe_hnd_dst,
2198 mem_ctx,
2199 &info_ctr);
2200
2201 if (W_ERROR_IS_OK(result))
2202 d_printf (_("printer [%s] successfully added.\n"),
2203 printername);
2204 else if (W_ERROR_V(result) == W_ERROR_V(WERR_PRINTER_ALREADY_EXISTS))
2205 d_fprintf (stderr, _("printer [%s] already exists.\n"),
2206 printername);
2207 else {
2208 d_fprintf (stderr, _("could not create printer [%s]\n"),
2209 printername);
2210 goto done;
2211 }
2212
2213 /* close printer handles here */
2214 if (is_valid_policy_hnd(&hnd_src)) {
2215 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
2216 }
2217
2218 if (is_valid_policy_hnd(&hnd_dst)) {
2219 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
2220 }
2221 }
2222
2223 nt_status = NT_STATUS_OK;
2224
2225 done:
2226 if (is_valid_policy_hnd(&hnd_src)) {
2227 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
2228 }
2229
2230 if (is_valid_policy_hnd(&hnd_dst)) {
2231 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
2232 }
2233
2234 if (cli_dst) {
2235 cli_shutdown(cli_dst);
2236 }
2237 return nt_status;
2238 }
2239
2240 /**
2241 * Migrate Printer-Settings from a src server to the dst server
2242 * (for this to work, printers and drivers already have to be migrated earlier)
2243 *
2244 * All parameters are provided by the run_rpc_command function, except for
2245 * argc, argv which are passed through.
2246 *
2247 * @param c A net_context structure
2248 * @param domain_sid The domain sid aquired from the remote server
2249 * @param cli A cli_state connected to the server.
2250 * @param mem_ctx Talloc context, destoyed on compleation of the function.
2251 * @param argc Standard main() style argc
2252 * @param argv Standard main() style argv. Initial components are already
2253 * stripped
2254 *
2255 * @return Normal NTSTATUS return.
2256 **/
2257
2258 NTSTATUS rpc_printer_migrate_settings_internals(struct net_context *c,
2259 const struct dom_sid *domain_sid,
2260 const char *domain_name,
2261 struct cli_state *cli,
2262 struct rpc_pipe_client *pipe_hnd,
2263 TALLOC_CTX *mem_ctx,
2264 int argc,
2265 const char **argv)
2266 {
2267 struct dcerpc_binding_handle *b_src = pipe_hnd->binding_handle;
2268
2269 /* FIXME: Here the nightmare begins */
2270
2271 WERROR result;
2272 NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL;
2273 uint32_t i = 0, j = 0;
2274 uint32_t num_printers;
2275 uint32_t level = 2;
2276 const char *printername, *sharename;
2277 struct rpc_pipe_client *pipe_hnd_dst = NULL;
2278 struct dcerpc_binding_handle *b_dst = NULL;
2279 struct policy_handle hnd_src, hnd_dst;
2280 union spoolss_PrinterInfo *info_enum;
2281 union spoolss_PrinterInfo info_dst_publish;
2282 union spoolss_PrinterInfo info_dst;
2283 struct cli_state *cli_dst = NULL;
2284 char *devicename = NULL, *unc_name = NULL, *url = NULL;
2285 const char *longname;
2286 const char **keylist = NULL;
2287
2288 /* FIXME GD */
2289 ZERO_STRUCT(info_dst_publish);
2290
2291 DEBUG(3,("copying printer settings\n"));
2292
2293 /* connect destination PI_SPOOLSS */
2294 nt_status = connect_dst_pipe(c, &cli_dst, &pipe_hnd_dst,
2295 &ndr_table_spoolss.syntax_id);
2296 if (!NT_STATUS_IS_OK(nt_status)) {
2297 return nt_status;
2298 }
2299 b_dst = pipe_hnd_dst->binding_handle;
2300
2301 /* enum src printers */
2302 if (!get_printer_info(pipe_hnd, mem_ctx, level, argc, argv, &num_printers, &info_enum)) {
2303 nt_status = NT_STATUS_UNSUCCESSFUL;
2304 goto done;
2305 }
2306
2307 if (!num_printers) {
2308 printf (_("no printers found on server.\n"));
2309 nt_status = NT_STATUS_OK;
2310 goto done;
2311 }
2312
2313
2314 /* needed for dns-strings in regkeys */
2315 longname = get_mydnsfullname();
2316 if (!longname) {
2317 nt_status = NT_STATUS_UNSUCCESSFUL;
2318 goto done;
2319 }
2320
2321 /* do something for all printers */
2322 for (i = 0; i < num_printers; i++) {
2323
2324 uint32_t value_needed;
2325 uint32_t data_needed;
2326 enum winreg_Type type;
2327 struct spoolss_EnumPrinterData r;
2328
2329 /* do some initialization */
2330 printername = info_enum[i].info2.printername;
2331 sharename = info_enum[i].info2.sharename;
2332
2333 if (!printername || !sharename) {
2334 nt_status = NT_STATUS_UNSUCCESSFUL;
2335 goto done;
2336 }
2337 /* we can reset NT_STATUS here because we do not
2338 get any real NT_STATUS-codes anymore from now on */
2339 nt_status = NT_STATUS_UNSUCCESSFUL;
2340
2341 d_printf(_("migrating printer settings for: [%s] / [%s]\n"),
2342 printername, sharename);
2343
2344
2345 /* open src printer handle */
2346 if (!net_spoolss_open_printer_ex(pipe_hnd, mem_ctx, sharename,
2347 MAXIMUM_ALLOWED_ACCESS, cli->user_name, &hnd_src))
2348 goto done;
2349
2350 /* open dst printer handle */
2351 if (!net_spoolss_open_printer_ex(pipe_hnd_dst, mem_ctx, sharename,
2352 PRINTER_ALL_ACCESS, cli_dst->user_name, &hnd_dst))
2353 goto done;
2354
2355 /* check for existing dst printer */
2356 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst,
2357 level, &info_dst))
2358 goto done;
2359
2360
2361 /* STEP 1: COPY DEVICE-MODE and other
2362 PRINTER_INFO_2-attributes
2363 */
2364
2365 info_dst.info2 = info_enum[i].info2;
2366
2367 /* why is the port always disconnected when the printer
2368 is correctly installed (incl. driver ???) */
2369 info_dst.info2.portname = SAMBA_PRINTER_PORT_NAME;
2370
2371 /* check if printer is published */
2372 if (info_enum[i].info2.attributes & PRINTER_ATTRIBUTE_PUBLISHED) {
2373
2374 /* check for existing dst printer */
2375 if (!net_spoolss_getprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, 7, &info_dst_publish))
2376 goto done;
2377
2378 info_dst_publish.info7.action = DSPRINT_PUBLISH;
2379
2380 /* ignore false from setprinter due to WERR_IO_PENDING */
2381 net_spoolss_setprinter(pipe_hnd_dst, mem_ctx, &hnd_dst, 7, &info_dst_publish);
2382
2383 DEBUG(3,("republished printer\n"));
2384 }
2385
2386 if (info_enum[i].info2.devmode != NULL) {
2387
2388 /* copy devmode (info level 2) */
2389 info_dst.info2.devmode = info_enum[i].info2.devmode;
2390
2391 /* do not copy security descriptor (we have another
2392 * command for that) */
2393 info_dst.info2.secdesc = NULL;
2394
2395 #if 0
2396 info_dst.info2.devmode.devicename =
2397 talloc_asprintf(mem_ctx, "\\\\%s\\%s",
2398 longname, printername);
2399 if (!info_dst.info2.devmode.devicename) {
2400 nt_status = NT_STATUS_NO_MEMORY;
2401 goto done;
2402 }
2403 #endif
2404 if (!net_spoolss_setprinter(pipe_hnd_dst, mem_ctx, &hnd_dst,
2405 level, &info_dst))
2406 goto done;
2407
2408 DEBUGADD(1,("\tSetPrinter of DEVICEMODE succeeded\n"));
2409 }
2410
2411 /* STEP 2: COPY REGISTRY VALUES */
2412
2413 /* please keep in mind that samba parse_spools gives horribly
2414 crippled results when used to rpccli_spoolss_enumprinterdataex
2415 a win2k3-server. (Bugzilla #1851)
2416 FIXME: IIRC I've seen it too on a win2k-server
2417 */
2418
2419 r.in.handle = &hnd_src;
2420 r.in.enum_index = 0;
2421 r.in.value_offered = 0;
2422 r.in.data_offered = 0;
2423 r.out.value_name = NULL;
2424 r.out.value_needed = &value_needed;
2425 r.out.type = &type;
2426 r.out.data = NULL;
2427 r.out.data_needed = &data_needed;
2428
2429 /* enumerate data on src handle */
2430 nt_status = dcerpc_spoolss_EnumPrinterData_r(b_src, mem_ctx, &r);
2431
2432 r.in.data_offered = *r.out.data_needed;
2433 r.in.value_offered = *r.out.value_needed;
2434 r.out.data = talloc_zero_array(mem_ctx, uint8_t, r.in.data_offered);
2435 r.out.value_name = talloc_zero_array(mem_ctx, char, r.in.value_offered);
2436
2437 /* loop for all printerdata of "PrinterDriverData" */
2438 while (NT_STATUS_IS_OK(nt_status) && W_ERROR_IS_OK(r.out.result)) {
2439
2440 r.in.enum_index++;
2441
2442 nt_status = dcerpc_spoolss_EnumPrinterData_r(b_src, mem_ctx, &r);
2443
2444 /* loop for all reg_keys */
2445 if (NT_STATUS_IS_OK(nt_status) && W_ERROR_IS_OK(r.out.result)) {
2446
2447 /* display_value */
2448 if (c->opt_verbose) {
2449 struct regval_blob *v;
2450
2451 v = regval_compose(talloc_tos(),
2452 r.out.value_name,
2453 *r.out.type,
2454 r.out.data,
2455 r.in.data_offered);
2456 if (v == NULL) {
2457 nt_status = NT_STATUS_NO_MEMORY;
2458 goto done;
2459 }
2460
2461 display_reg_value(SPOOL_PRINTERDATA_KEY, v);
2462 talloc_free(v);
2463 }
2464
2465 /* set_value */
2466 if (!net_spoolss_setprinterdata(pipe_hnd_dst, mem_ctx,
2467 &hnd_dst, r.out.value_name,
2468 *r.out.type, r.out.data, r.in.data_offered))
2469 goto done;
2470
2471 DEBUGADD(1,("\tSetPrinterData of [%s] succeeded\n",
2472 r.out.value_name));
2473 }
2474 }
2475
2476 /* STEP 3: COPY SUBKEY VALUES */
2477
2478 /* here we need to enum all printer_keys and then work
2479 on the result with enum_printer_key_ex. nt4 does not
2480 respond to enumprinterkey, win2k does, so continue
2481 in case of an error */
2482
2483 if (!net_spoolss_enumprinterkey(pipe_hnd, mem_ctx, &hnd_src, "", &keylist)) {
2484 printf(_("got no key-data\n"));
2485 continue;
2486 }
2487
2488
2489 /* work on a list of printer keys
2490 each key has to be enumerated to get all required
2491 information. information is then set via setprinterdataex-calls */
2492
2493 if (keylist == NULL)
2494 continue;
2495
2496 for (i=0; keylist && keylist[i] != NULL; i++) {
2497
2498 const char *subkey = keylist[i];
2499 uint32_t count;
2500 struct spoolss_PrinterEnumValues *info;
2501
2502 /* enumerate all src subkeys */
2503 if (!net_spoolss_enumprinterdataex(pipe_hnd, mem_ctx, 0,
2504 &hnd_src, subkey,
2505 &count, &info)) {
2506 goto done;
2507 }
2508
2509 for (j=0; j < count; j++) {
2510
2511 struct regval_blob *value;
2512 DATA_BLOB blob;
2513
2514 ZERO_STRUCT(blob);
2515
2516 /* although samba replies with sane data in most cases we
2517 should try to avoid writing wrong registry data */
2518
2519 if (strequal(info[j].value_name, SPOOL_REG_PORTNAME) ||
2520 strequal(info[j].value_name, SPOOL_REG_UNCNAME) ||
2521 strequal(info[j].value_name, SPOOL_REG_URL) ||
2522 strequal(info[j].value_name, SPOOL_REG_SHORTSERVERNAME) ||
2523 strequal(info[j].value_name, SPOOL_REG_SERVERNAME)) {
2524
2525 if (strequal(info[j].value_name, SPOOL_REG_PORTNAME)) {
2526
2527 /* although windows uses a multi-sz, we use a sz */
2528 push_reg_sz(mem_ctx, &blob, SAMBA_PRINTER_PORT_NAME);
2529 }
2530
2531 if (strequal(info[j].value_name, SPOOL_REG_UNCNAME)) {
2532
2533 if (asprintf(&unc_name, "\\\\%s\\%s", longname, sharename) < 0) {
2534 nt_status = NT_STATUS_NO_MEMORY;
2535 goto done;
2536 }
2537 push_reg_sz(mem_ctx, &blob, unc_name);
2538 }
2539
2540 if (strequal(info[j].value_name, SPOOL_REG_URL)) {
2541
2542 continue;
2543
2544 #if 0
2545 /* FIXME: should we really do that ??? */
2546 if (asprintf(&url, "http://%s:631/printers/%s", longname, sharename) < 0) {
2547 nt_status = NT_STATUS_NO_MEMORY;
2548 goto done;
2549 }
2550 push_reg_sz(mem_ctx, NULL, &blob, url);
2551 fstrcpy(value.valuename, SPOOL_REG_URL);
2552 #endif
2553 }
2554
2555 if (strequal(info[j].value_name, SPOOL_REG_SERVERNAME)) {
2556
2557 push_reg_sz(mem_ctx, &blob, longname);
2558 }
2559
2560 if (strequal(info[j].value_name, SPOOL_REG_SHORTSERVERNAME)) {
2561
2562 push_reg_sz(mem_ctx, &blob, lp_netbios_name());
2563 }
2564
2565 value = regval_compose(talloc_tos(),
2566 info[j].value_name,
2567 REG_SZ,
2568 blob.length == 0 ? NULL : blob.data,
2569 blob.length);
2570 if (value == NULL) {
2571 nt_status = NT_STATUS_NO_MEMORY;
2572 goto done;
2573 }
2574
2575 if (c->opt_verbose)
2576 display_reg_value(subkey, value);
2577
2578 /* here we have to set all subkeys on the dst server */
2579 if (!net_spoolss_setprinterdataex(pipe_hnd_dst, mem_ctx, &hnd_dst,
2580 subkey, value))
2581 {
2582 talloc_free(value);
2583 goto done;
2584 }
2585
2586 talloc_free(value);
2587 } else {
2588
2589 struct regval_blob *v;
2590
2591 v = regval_compose(talloc_tos(),
2592 info[j].value_name,
2593 info[j].type,
2594 info[j].data->data,
2595 info[j].data->length);
2596 if (v == NULL) {
2597 nt_status = NT_STATUS_NO_MEMORY;
2598 goto done;
2599 }
2600
2601 if (c->opt_verbose) {
2602 display_reg_value(subkey, v);
2603 }
2604
2605 /* here we have to set all subkeys on the dst server */
2606 if (!net_spoolss_setprinterdataex(pipe_hnd_dst, mem_ctx, &hnd_dst,
2607 subkey, v)) {
2608 goto done;
2609 }
2610
2611 talloc_free(v);
2612 }
2613
2614 DEBUGADD(1,("\tSetPrinterDataEx of key [%s\\%s] succeeded\n",
2615 subkey, info[j].value_name));
2616
2617 }
2618 }
2619
2620 TALLOC_FREE(keylist);
2621
2622 /* close printer handles here */
2623 if (is_valid_policy_hnd(&hnd_src)) {
2624 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
2625 }
2626
2627 if (is_valid_policy_hnd(&hnd_dst)) {
2628 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
2629 }
2630 }
2631
2632 nt_status = NT_STATUS_OK;
2633
2634 done:
2635 SAFE_FREE(devicename);
2636 SAFE_FREE(url);
2637 SAFE_FREE(unc_name);
2638
2639 if (is_valid_policy_hnd(&hnd_src)) {
2640 dcerpc_spoolss_ClosePrinter(b_src, mem_ctx, &hnd_src, &result);
2641 }
2642
2643 if (is_valid_policy_hnd(&hnd_dst)) {
2644 dcerpc_spoolss_ClosePrinter(b_dst, mem_ctx, &hnd_dst, &result);
2645 }
2646
2647 if (cli_dst) {
2648 cli_shutdown(cli_dst);
2649 }
2650 return nt_status;
2651 }
Samba GIT mirror