source: branches/samba-3.3.x/source/utils/net_rpc_printer.c

Last change on this file was 206, checked in by Herwig Bauernfeind, 16 years ago

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