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