source3/utils/net_rpc_printer.c

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