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