Line data    Source code

       1             : /*
       2             :  *  Unix SMB/CIFS implementation.
       3             :  *  RPC Pipe client / server routines
       4             :  *
       5             :  *  Copyright (c) Andreas Schneider            2010.
       6             :  *  Copyright (C) Bjoern Baumbach <bb@sernet.de> 2011
       7             :  *
       8             :  *  This program is free software; you can redistribute it and/or modify
       9             :  *  it under the terms of the GNU General Public License as published by
      10             :  *  the Free Software Foundation; either version 3 of the License, or
      11             :  *  (at your option) any later version.
      12             :  *
      13             :  *  This program is distributed in the hope that it will be useful,
      14             :  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
      15             :  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      16             :  *  GNU General Public License for more details.
      17             :  *
      18             :  *  You should have received a copy of the GNU General Public License
      19             :  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
      20             :  */
      21             : 
      22             : #include "includes.h"
      23             : #include "printing/nt_printing_migrate.h"
      24             : 
      25             : #include "rpc_client/rpc_client.h"
      26             : #include "librpc/gen_ndr/ndr_ntprinting.h"
      27             : #include "librpc/gen_ndr/ndr_spoolss_c.h"
      28             : #include "librpc/gen_ndr/ndr_security.h"
      29             : #include "rpc_client/cli_winreg_spoolss.h"
      30             : 
      31           0 : static const char *driver_file_basename(const char *file)
      32             : {
      33           0 :         const char *basefile;
      34             : 
      35           0 :         basefile = strrchr(file, '\\');
      36           0 :         if (basefile == NULL) {
      37           0 :                 basefile = file;
      38             :         } else {
      39           0 :                 basefile++;
      40             :         }
      41             : 
      42           0 :         return basefile;
      43             : }
      44             : 
      45           0 : NTSTATUS printing_tdb_migrate_form(TALLOC_CTX *mem_ctx,
      46             :                                    struct rpc_pipe_client *winreg_pipe,
      47             :                                    const char *key_name,
      48             :                                    unsigned char *data,
      49             :                                    size_t length)
      50             : {
      51           0 :         struct dcerpc_binding_handle *b = winreg_pipe->binding_handle;
      52           0 :         enum ndr_err_code ndr_err;
      53           0 :         struct ntprinting_form r;
      54           0 :         struct spoolss_AddFormInfo1 f1;
      55           0 :         DATA_BLOB blob;
      56           0 :         WERROR result;
      57             : 
      58           0 :         blob = data_blob_const(data, length);
      59             : 
      60           0 :         ZERO_STRUCT(r);
      61             : 
      62           0 :         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
      63             :                    (ndr_pull_flags_fn_t)ndr_pull_ntprinting_form);
      64           0 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
      65           0 :                 DEBUG(2, ("Form pull failed: %s\n",
      66             :                           ndr_errstr(ndr_err)));
      67           0 :                 return NT_STATUS_NO_MEMORY;
      68             :         }
      69             : 
      70             :         /* Don't migrate builtin forms */
      71           0 :         if (r.flag == SPOOLSS_FORM_BUILTIN) {
      72           0 :                 return NT_STATUS_OK;
      73             :         }
      74             : 
      75           0 :         DEBUG(2, ("Migrating Form: %s\n", key_name));
      76             : 
      77           0 :         f1.form_name = key_name;
      78           0 :         f1.flags = r.flag;
      79             : 
      80           0 :         f1.size.width = r.width;
      81           0 :         f1.size.height = r.length;
      82             : 
      83           0 :         f1.area.top = r.top;
      84           0 :         f1.area.right = r.right;
      85           0 :         f1.area.bottom = r.bottom;
      86           0 :         f1.area.left = r.left;
      87             : 
      88           0 :         result = winreg_printer_addform1(mem_ctx,
      89             :                                          b,
      90             :                                          &f1);
      91           0 :         if (W_ERROR_EQUAL(result, WERR_FILE_EXISTS)) {
      92             :                 /* Don't migrate form if it already exists. */
      93           0 :                 result = WERR_OK;
      94             :         }
      95           0 :         if (!W_ERROR_IS_OK(result)) {
      96           0 :                 return werror_to_ntstatus(result);
      97             :         }
      98             : 
      99           0 :         return NT_STATUS_OK;
     100             : }
     101             : 
     102           0 : NTSTATUS printing_tdb_migrate_driver(TALLOC_CTX *mem_ctx,
     103             :                                      struct rpc_pipe_client *winreg_pipe,
     104             :                                      const char *key_name,
     105             :                                      unsigned char *data,
     106             :                                      size_t length,
     107             :                                      bool do_string_conversion)
     108             : {
     109           0 :         struct dcerpc_binding_handle *b = winreg_pipe->binding_handle;
     110           0 :         enum ndr_err_code ndr_err;
     111           0 :         struct ntprinting_driver r;
     112           0 :         struct spoolss_AddDriverInfoCtr d;
     113           0 :         struct spoolss_AddDriverInfo3 d3;
     114           0 :         struct spoolss_StringArray a;
     115           0 :         DATA_BLOB blob;
     116           0 :         WERROR result;
     117           0 :         const char *driver_name;
     118           0 :         uint32_t driver_version;
     119           0 :         int i;
     120             : 
     121           0 :         blob = data_blob_const(data, length);
     122             : 
     123           0 :         ZERO_STRUCT(r);
     124             : 
     125           0 :         if (do_string_conversion) {
     126           0 :                 r.string_flags = LIBNDR_FLAG_STR_ASCII;
     127             :         }
     128             : 
     129           0 :         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
     130             :                    (ndr_pull_flags_fn_t)ndr_pull_ntprinting_driver);
     131           0 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
     132           0 :                 DEBUG(2, ("Driver pull failed: %s\n",
     133             :                           ndr_errstr(ndr_err)));
     134           0 :                 return NT_STATUS_NO_MEMORY;
     135             :         }
     136             : 
     137           0 :         DEBUG(2, ("Migrating Printer Driver: %s\n", key_name));
     138             : 
     139           0 :         ZERO_STRUCT(d3);
     140           0 :         ZERO_STRUCT(a);
     141             : 
     142             :         /* remove paths from file names */
     143           0 :         if (r.dependent_files != NULL) {
     144           0 :                 for (i = 0 ; r.dependent_files[i] != NULL; i++) {
     145           0 :                         r.dependent_files[i] = driver_file_basename(r.dependent_files[i]);
     146             :                 }
     147             :         }
     148           0 :         a.string = r.dependent_files;
     149             : 
     150           0 :         r.driverpath = driver_file_basename(r.driverpath);
     151           0 :         r.configfile = driver_file_basename(r.configfile);
     152           0 :         r.datafile = driver_file_basename(r.datafile);
     153           0 :         r.helpfile = driver_file_basename(r.helpfile);
     154             : 
     155           0 :         d3.architecture = r.environment;
     156           0 :         d3.config_file = r.configfile;
     157           0 :         d3.data_file = r.datafile;
     158           0 :         d3.default_datatype = r.defaultdatatype;
     159           0 :         d3.dependent_files = &a;
     160           0 :         d3.driver_path = r.driverpath;
     161           0 :         d3.help_file = r.helpfile;
     162           0 :         d3.monitor_name = r.monitorname;
     163           0 :         d3.driver_name = r.name;
     164           0 :         d3.version = r.version;
     165             : 
     166           0 :         d.level = 3;
     167           0 :         d.info.info3 = &d3;
     168             : 
     169           0 :         result = winreg_add_driver(mem_ctx,
     170             :                                    b,
     171             :                                    &d,
     172             :                                    &driver_name,
     173             :                                    &driver_version);
     174           0 :         if (!W_ERROR_IS_OK(result)) {
     175           0 :                 return werror_to_ntstatus(result);
     176             :         }
     177             : 
     178           0 :         return NT_STATUS_OK;
     179             : }
     180             : 
     181           0 : NTSTATUS printing_tdb_migrate_printer(TALLOC_CTX *mem_ctx,
     182             :                                       struct rpc_pipe_client *winreg_pipe,
     183             :                                       const char *key_name,
     184             :                                       unsigned char *data,
     185             :                                       size_t length,
     186             :                                       bool do_string_conversion)
     187             : {
     188           0 :         struct dcerpc_binding_handle *b = winreg_pipe->binding_handle;
     189           0 :         enum ndr_err_code ndr_err;
     190           0 :         struct ntprinting_printer r;
     191           0 :         struct spoolss_SetPrinterInfo2 info2;
     192           0 :         struct spoolss_DeviceMode dm;
     193           0 :         struct spoolss_DevmodeContainer devmode_ctr;
     194           0 :         DATA_BLOB blob;
     195           0 :         NTSTATUS status;
     196           0 :         WERROR result;
     197           0 :         int j;
     198           0 :         uint32_t info2_mask = (SPOOLSS_PRINTER_INFO_ALL)
     199             :                                 & ~SPOOLSS_PRINTER_INFO_SECDESC;
     200             : 
     201           0 :         if (strequal(key_name, "printers")) {
     202           0 :                 return NT_STATUS_OK;
     203             :         }
     204             : 
     205           0 :         blob = data_blob_const(data, length);
     206             : 
     207           0 :         ZERO_STRUCT(r);
     208             : 
     209           0 :         if (do_string_conversion) {
     210           0 :                 r.info.string_flags = LIBNDR_FLAG_STR_ASCII;
     211             :         }
     212             : 
     213           0 :         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &r,
     214             :                    (ndr_pull_flags_fn_t) ndr_pull_ntprinting_printer);
     215           0 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
     216           0 :                 DEBUG(2, ("printer pull failed: %s\n",
     217             :                           ndr_errstr(ndr_err)));
     218           0 :                 return NT_STATUS_NO_MEMORY;
     219             :         }
     220             : 
     221           0 :         DEBUG(2, ("Migrating Printer: %s\n", key_name));
     222             : 
     223           0 :         ZERO_STRUCT(devmode_ctr);
     224             : 
     225             :         /* Create printer info level 2 */
     226           0 :         ZERO_STRUCT(info2);
     227             : 
     228           0 :         info2.attributes = r.info.attributes;
     229           0 :         info2.averageppm = r.info.averageppm;
     230           0 :         info2.cjobs = r.info.cjobs;
     231           0 :         info2.comment = r.info.comment;
     232           0 :         info2.datatype = r.info.datatype;
     233           0 :         info2.defaultpriority = r.info.default_priority;
     234           0 :         info2.drivername = r.info.drivername;
     235           0 :         info2.location = r.info.location;
     236           0 :         info2.parameters = r.info.parameters;
     237           0 :         info2.portname = r.info.portname;
     238           0 :         info2.printername = r.info.printername;
     239           0 :         info2.printprocessor = r.info.printprocessor;
     240           0 :         info2.priority = r.info.priority;
     241           0 :         info2.sepfile = r.info.sepfile;
     242           0 :         info2.sharename = r.info.sharename;
     243           0 :         info2.starttime = r.info.starttime;
     244           0 :         info2.status = r.info.status;
     245           0 :         info2.untiltime = r.info.untiltime;
     246             : 
     247             :         /* Create Device Mode */
     248           0 :         if (r.devmode == NULL) {
     249           0 :                 info2_mask &= ~SPOOLSS_PRINTER_INFO_DEVMODE;
     250             :         } else {
     251           0 :                 ZERO_STRUCT(dm);
     252             : 
     253           0 :                 dm.bitsperpel              = r.devmode->bitsperpel;
     254           0 :                 dm.collate                 = r.devmode->collate;
     255           0 :                 dm.color                   = r.devmode->color;
     256           0 :                 dm.copies                  = r.devmode->copies;
     257           0 :                 dm.defaultsource           = r.devmode->defaultsource;
     258           0 :                 dm.devicename              = r.devmode->devicename;
     259           0 :                 dm.displayflags            = r.devmode->displayflags;
     260           0 :                 dm.displayfrequency        = r.devmode->displayfrequency;
     261           0 :                 dm.dithertype              = r.devmode->dithertype;
     262           0 :                 dm.driverversion           = r.devmode->driverversion;
     263           0 :                 dm.duplex                  = r.devmode->duplex;
     264           0 :                 dm.fields                  = r.devmode->fields;
     265           0 :                 dm.formname                = r.devmode->formname;
     266           0 :                 dm.icmintent               = r.devmode->icmintent;
     267           0 :                 dm.icmmethod               = r.devmode->icmmethod;
     268           0 :                 dm.logpixels               = r.devmode->logpixels;
     269           0 :                 dm.mediatype               = r.devmode->mediatype;
     270           0 :                 dm.orientation             = r.devmode->orientation;
     271           0 :                 dm.panningheight           = r.devmode->pelsheight;
     272           0 :                 dm.panningwidth            = r.devmode->panningwidth;
     273           0 :                 dm.paperlength             = r.devmode->paperlength;
     274           0 :                 dm.papersize               = r.devmode->papersize;
     275           0 :                 dm.paperwidth              = r.devmode->paperwidth;
     276           0 :                 dm.pelsheight              = r.devmode->pelsheight;
     277           0 :                 dm.pelswidth               = r.devmode->pelswidth;
     278           0 :                 dm.printquality            = r.devmode->printquality;
     279           0 :                 dm.size                    = r.devmode->size;
     280           0 :                 dm.scale                   = r.devmode->scale;
     281           0 :                 dm.specversion             = r.devmode->specversion;
     282           0 :                 dm.ttoption                = r.devmode->ttoption;
     283           0 :                 dm.yresolution             = r.devmode->yresolution;
     284             : 
     285           0 :                 if (r.devmode->nt_dev_private != NULL) {
     286           0 :                         dm.driverextra_data.data   = r.devmode->nt_dev_private->data;
     287           0 :                         dm.driverextra_data.length = r.devmode->nt_dev_private->length;
     288           0 :                         dm.__driverextra_length    = r.devmode->nt_dev_private->length;
     289             :                 }
     290             : 
     291           0 :                 devmode_ctr.devmode = &dm;
     292             :         }
     293             : 
     294           0 :         result = winreg_update_printer(mem_ctx, b,
     295             :                                        key_name,
     296             :                                        info2_mask,
     297             :                                        &info2,
     298             :                                        &dm,
     299             :                                        NULL);
     300           0 :         if (!W_ERROR_IS_OK(result)) {
     301           0 :                 DEBUG(2, ("SetPrinter(%s) level 2 refused -- %s.\n",
     302             :                           key_name, win_errstr(result)));
     303           0 :                 status = werror_to_ntstatus(result);
     304           0 :                 goto done;
     305             :         }
     306             : 
     307             :         /* migrate printerdata */
     308           0 :         for (j = 0; j < r.count; j++) {
     309           0 :                 char *valuename;
     310           0 :                 const char *keyname;
     311             : 
     312           0 :                 if (r.printer_data[j].type == REG_NONE) {
     313           0 :                         continue;
     314             :                 }
     315             : 
     316           0 :                 keyname = r.printer_data[j].name;
     317           0 :                 valuename = strchr(keyname, '\\');
     318           0 :                 if (valuename == NULL) {
     319           0 :                         continue;
     320             :                 } else {
     321           0 :                         valuename[0] = '\0';
     322           0 :                         valuename++;
     323             :                 }
     324             : 
     325           0 :                 result = winreg_set_printer_dataex(mem_ctx, b,
     326             :                                                    key_name,
     327             :                                                    keyname,
     328             :                                                    valuename,
     329           0 :                                                    r.printer_data[j].type,
     330           0 :                                                    r.printer_data[j].data.data,
     331           0 :                                                    r.printer_data[j].data.length);
     332           0 :                 if (!W_ERROR_IS_OK(result)) {
     333           0 :                         DEBUG(2, ("SetPrinterDataEx: printer [%s], keyname [%s], "
     334             :                                   "valuename [%s] refused -- %s.\n",
     335             :                                   key_name, keyname, valuename,
     336             :                                   win_errstr(result)));
     337           0 :                         status = werror_to_ntstatus(result);
     338           0 :                         break;
     339             :                 }
     340             :         }
     341             : 
     342           0 :         status = NT_STATUS_OK;
     343           0 :  done:
     344             : 
     345           0 :         return status;
     346             : }
     347             : 
     348           0 : NTSTATUS printing_tdb_migrate_secdesc(TALLOC_CTX *mem_ctx,
     349             :                                       struct rpc_pipe_client *winreg_pipe,
     350             :                                       const char *key_name,
     351             :                                       unsigned char *data,
     352             :                                       size_t length)
     353             : {
     354           0 :         struct dcerpc_binding_handle *b = winreg_pipe->binding_handle;
     355           0 :         enum ndr_err_code ndr_err;
     356           0 :         struct sec_desc_buf secdesc_ctr;
     357           0 :         DATA_BLOB blob;
     358           0 :         WERROR result;
     359             : 
     360           0 :         if (strequal(key_name, "printers")) {
     361           0 :                 return NT_STATUS_OK;
     362             :         }
     363             : 
     364           0 :         blob = data_blob_const(data, length);
     365             : 
     366           0 :         ZERO_STRUCT(secdesc_ctr);
     367             : 
     368           0 :         ndr_err = ndr_pull_struct_blob(&blob, mem_ctx, &secdesc_ctr,
     369             :                    (ndr_pull_flags_fn_t)ndr_pull_sec_desc_buf);
     370           0 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
     371           0 :                 DEBUG(2, ("security descriptor pull failed: %s\n",
     372             :                           ndr_errstr(ndr_err)));
     373           0 :                 return NT_STATUS_NO_MEMORY;
     374             :         }
     375             : 
     376           0 :         DEBUG(2, ("Migrating Security Descriptor: %s\n", key_name));
     377             : 
     378           0 :         result = winreg_set_printer_secdesc(mem_ctx, b,
     379             :                                             key_name,
     380           0 :                                             secdesc_ctr.sd);
     381           0 :         if (!W_ERROR_IS_OK(result)) {
     382           0 :                 return werror_to_ntstatus(result);
     383             :         }
     384             : 
     385           0 :         return NT_STATUS_OK;
     386             : }