source3/rpc_server/srv_spoolss_nt.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *  RPC Pipe client / server routines
   4  *  Copyright (C) Andrew Tridgell              1992-2000,
   5  *  Copyright (C) Luke Kenneth Casson Leighton 1996-2000,
   6  *  Copyright (C) Jean François Micouleau      1998-2000,
   7  *  Copyright (C) Jeremy Allison               2001-2002,
   8  *  Copyright (C) Gerald Carter                2000-2004,
   9  *  Copyright (C) Tim Potter                   2001-2002.
  10  *
  11  *  This program is free software; you can redistribute it and/or modify
  12  *  it under the terms of the GNU General Public License as published by
  13  *  the Free Software Foundation; either version 2 of the License, or
  14  *  (at your option) any later version.
  15  *
  16  *  This program is distributed in the hope that it will be useful,
  17  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  *  GNU General Public License for more details.
  20  *
  21  *  You should have received a copy of the GNU General Public License
  22  *  along with this program; if not, write to the Free Software
  23  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  24  */
  25
  26 /* Since the SPOOLSS rpc routines are basically DOS 16-bit calls wrapped
  27    up, all the errors returned are DOS errors, not NT status codes. */
  28
  29 #include "includes.h"
  30
  31 extern userdom_struct current_user_info;
  32
  33 #undef DBGC_CLASS
  34 #define DBGC_CLASS DBGC_RPC_SRV
  35
  36 #ifndef MAX_OPEN_PRINTER_EXS
  37 #define MAX_OPEN_PRINTER_EXS 50
  38 #endif
  39
  40 #define MAGIC_DISPLAY_FREQUENCY 0xfade2bad
  41 #define PHANTOM_DEVMODE_KEY "_p_f_a_n_t_0_m_"
  42
  43
  44 /* Table to map the driver version */
  45 /* to OS */
  46 static const char * drv_ver_to_os[] = {
  47         "WIN9X",   /* driver version/cversion 0 */
  48         "",        /* unused ? */
  49         "WINNT",   /* driver version/cversion 2 */
  50         "WIN2K",   /* driver version/cversion 3 */
  51 };
  52
  53 static const char *get_drv_ver_to_os(int ver)
  54 {
  55         if (ver < 0 || ver > 3)
  56                 return "";
  57         return drv_ver_to_os[ver];
  58 }
  59
  60 struct table_node {
  61         const char    *long_archi;
  62         const char    *short_archi;
  63         int     version;
  64 };
  65
  66 static Printer_entry *printers_list;
  67
  68 typedef struct _counter_printer_0 {
  69         struct _counter_printer_0 *next;
  70         struct _counter_printer_0 *prev;
  71
  72         int snum;
  73         uint32 counter;
  74 } counter_printer_0;
  75
  76 static counter_printer_0 *counter_list;
  77
  78 static struct rpc_pipe_client *notify_cli_pipe; /* print notify back-channel pipe handle*/
  79 static uint32 smb_connections=0;
  80
  81
  82 /* in printing/nt_printing.c */
  83
  84 extern STANDARD_MAPPING printer_std_mapping, printserver_std_mapping;
  85
  86 #define OUR_HANDLE(hnd) (((hnd)==NULL)?"NULL":(IVAL((hnd)->data5,4)==(uint32)sys_getpid()?"OURS":"OTHER")), \
  87 ((unsigned int)IVAL((hnd)->data5,4)),((unsigned int)sys_getpid())
  88
  89
  90 /* API table for Xcv Monitor functions */
  91
  92 struct xcv_api_table {
  93         const char *name;
  94         WERROR(*fn) (NT_USER_TOKEN *token, RPC_BUFFER *in, RPC_BUFFER *out, uint32 *needed);
  95 };
  96
  97
  98 /* translate between internal status numbers and NT status numbers */
  99 static int nt_printj_status(int v)
 100 {
 101         switch (v) {
 102         case LPQ_QUEUED:
 103                 return 0;
 104         case LPQ_PAUSED:
 105                 return JOB_STATUS_PAUSED;
 106         case LPQ_SPOOLING:
 107                 return JOB_STATUS_SPOOLING;
 108         case LPQ_PRINTING:
 109                 return JOB_STATUS_PRINTING;
 110         case LPQ_ERROR:
 111                 return JOB_STATUS_ERROR;
 112         case LPQ_DELETING:
 113                 return JOB_STATUS_DELETING;
 114         case LPQ_OFFLINE:
 115                 return JOB_STATUS_OFFLINE;
 116         case LPQ_PAPEROUT:
 117                 return JOB_STATUS_PAPEROUT;
 118         case LPQ_PRINTED:
 119                 return JOB_STATUS_PRINTED;
 120         case LPQ_DELETED:
 121                 return JOB_STATUS_DELETED;
 122         case LPQ_BLOCKED:
 123                 return JOB_STATUS_BLOCKED;
 124         case LPQ_USER_INTERVENTION:
 125                 return JOB_STATUS_USER_INTERVENTION;
 126         }
 127         return 0;
 128 }
 129
 130 static int nt_printq_status(int v)
 131 {
 132         switch (v) {
 133         case LPQ_PAUSED:
 134                 return PRINTER_STATUS_PAUSED;
 135         case LPQ_QUEUED:
 136         case LPQ_SPOOLING:
 137         case LPQ_PRINTING:
 138                 return 0;
 139         }
 140         return 0;
 141 }
 142
 143 /****************************************************************************
 144  Functions to handle SPOOL_NOTIFY_OPTION struct stored in Printer_entry.
 145 ****************************************************************************/
 146
 147 static void free_spool_notify_option(SPOOL_NOTIFY_OPTION **pp)
 148 {
 149         if (*pp == NULL)
 150                 return;
 151
 152         SAFE_FREE((*pp)->ctr.type);
 153         SAFE_FREE(*pp);
 154 }
 155
 156 /***************************************************************************
 157  Disconnect from the client
 158 ****************************************************************************/
 159
 160 static void srv_spoolss_replycloseprinter(int snum, POLICY_HND *handle)
 161 {
 162         WERROR result;
 163
 164         /*
 165          * Tell the specific printing tdb we no longer want messages for this printer
 166          * by deregistering our PID.
 167          */
 168
 169         if (!print_notify_deregister_pid(snum))
 170                 DEBUG(0,("print_notify_register_pid: Failed to register our pid for printer %s\n", lp_const_servicename(snum) ));
 171
 172         /* weird if the test succeds !!! */
 173         if (smb_connections==0) {
 174                 DEBUG(0,("srv_spoolss_replycloseprinter:Trying to close non-existant notify backchannel !\n"));
 175                 return;
 176         }
 177
 178         result = rpccli_spoolss_reply_close_printer(notify_cli_pipe, notify_cli_pipe->cli->mem_ctx, handle);
 179
 180         if (!W_ERROR_IS_OK(result))
 181                 DEBUG(0,("srv_spoolss_replycloseprinter: reply_close_printer failed [%s].\n",
 182                         dos_errstr(result)));
 183
 184         /* if it's the last connection, deconnect the IPC$ share */
 185         if (smb_connections==1) {
 186
 187                 cli_shutdown( notify_cli_pipe->cli );
 188                 notify_cli_pipe = NULL; /* The above call shuts downn the pipe also. */
 189
 190                 message_deregister(MSG_PRINTER_NOTIFY2);
 191
 192                 /* Tell the connections db we're no longer interested in
 193                  * printer notify messages. */
 194
 195                 register_message_flags( False, FLAG_MSG_PRINT_NOTIFY );
 196         }
 197
 198         smb_connections--;
 199 }
 200
 201 /****************************************************************************
 202  Functions to free a printer entry datastruct.
 203 ****************************************************************************/
 204
 205 static void free_printer_entry(void *ptr)
 206 {
 207         Printer_entry *Printer = (Printer_entry *)ptr;
 208
 209         if (Printer->notify.client_connected==True) {
 210                 int snum = -1;
 211
 212                 if ( Printer->printer_type == SPLHND_SERVER) {
 213                         snum = -1;
 214                         srv_spoolss_replycloseprinter(snum, &Printer->notify.client_hnd);
 215                 } else if (Printer->printer_type == SPLHND_PRINTER) {
 216                         snum = print_queue_snum(Printer->sharename);
 217                         if (snum != -1)
 218                                 srv_spoolss_replycloseprinter(snum,
 219                                                 &Printer->notify.client_hnd);
 220                 }
 221         }
 222
 223         Printer->notify.flags=0;
 224         Printer->notify.options=0;
 225         Printer->notify.localmachine[0]='\0';
 226         Printer->notify.printerlocal=0;
 227         free_spool_notify_option(&Printer->notify.option);
 228         Printer->notify.option=NULL;
 229         Printer->notify.client_connected=False;
 230
 231         free_nt_devicemode( &Printer->nt_devmode );
 232         free_a_printer( &Printer->printer_info, 2 );
 233
 234         talloc_destroy( Printer->ctx );
 235
 236         /* Remove from the internal list. */
 237         DLIST_REMOVE(printers_list, Printer);
 238
 239         SAFE_FREE(Printer);
 240 }
 241
 242 /****************************************************************************
 243  Functions to duplicate a SPOOL_NOTIFY_OPTION struct stored in Printer_entry.
 244 ****************************************************************************/
 245
 246 static SPOOL_NOTIFY_OPTION *dup_spool_notify_option(SPOOL_NOTIFY_OPTION *sp)
 247 {
 248         SPOOL_NOTIFY_OPTION *new_sp = NULL;
 249
 250         if (!sp)
 251                 return NULL;
 252
 253         new_sp = SMB_MALLOC_P(SPOOL_NOTIFY_OPTION);
 254         if (!new_sp)
 255                 return NULL;
 256
 257         *new_sp = *sp;
 258
 259         if (sp->ctr.count) {
 260                 new_sp->ctr.type = (SPOOL_NOTIFY_OPTION_TYPE *)memdup(sp->ctr.type, sizeof(SPOOL_NOTIFY_OPTION_TYPE) * sp->ctr.count);
 261
 262                 if (!new_sp->ctr.type) {
 263                         SAFE_FREE(new_sp);
 264                         return NULL;
 265                 }
 266         }
 267
 268         return new_sp;
 269 }
 270
 271 /****************************************************************************
 272   find printer index by handle
 273 ****************************************************************************/
 274
 275 static Printer_entry *find_printer_index_by_hnd(pipes_struct *p, POLICY_HND *hnd)
 276 {
 277         Printer_entry *find_printer = NULL;
 278
 279         if(!find_policy_by_hnd(p,hnd,(void **)(void *)&find_printer)) {
 280                 DEBUG(2,("find_printer_index_by_hnd: Printer handle not found: "));
 281                 return NULL;
 282         }
 283
 284         return find_printer;
 285 }
 286
 287 /****************************************************************************
 288  Close printer index by handle.
 289 ****************************************************************************/
 290
 291 static BOOL close_printer_handle(pipes_struct *p, POLICY_HND *hnd)
 292 {
 293         Printer_entry *Printer = find_printer_index_by_hnd(p, hnd);
 294
 295         if (!Printer) {
 296                 DEBUG(2,("close_printer_handle: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(hnd)));
 297                 return False;
 298         }
 299
 300         close_policy_hnd(p, hnd);
 301
 302         return True;
 303 }
 304
 305 /****************************************************************************
 306  Delete a printer given a handle.
 307 ****************************************************************************/
 308 WERROR delete_printer_hook( NT_USER_TOKEN *token, const char *sharename )
 309 {
 310         char *cmd = lp_deleteprinter_cmd();
 311         pstring command;
 312         int ret;
 313         SE_PRIV se_printop = SE_PRINT_OPERATOR;
 314         BOOL is_print_op = False;
 315
 316         /* can't fail if we don't try */
 317
 318         if ( !*cmd )
 319                 return WERR_OK;
 320
 321         pstr_sprintf(command, "%s \"%s\"", cmd, sharename);
 322
 323         if ( token )
 324                 is_print_op = user_has_privileges( token, &se_printop );
 325
 326         DEBUG(10,("Running [%s]\n", command));
 327
 328         /********** BEGIN SePrintOperatorPrivlege BLOCK **********/
 329
 330         if ( is_print_op )
 331                 become_root();
 332
 333         if ( (ret = smbrun(command, NULL)) == 0 ) {
 334                 /* Tell everyone we updated smb.conf. */
 335                 message_send_all(conn_tdb_ctx(), MSG_SMB_CONF_UPDATED, NULL, 0, False, NULL);
 336         }
 337
 338         if ( is_print_op )
 339                 unbecome_root();
 340
 341         /********** END SePrintOperatorPrivlege BLOCK **********/
 342
 343         DEBUGADD(10,("returned [%d]\n", ret));
 344
 345         if (ret != 0)
 346                 return WERR_BADFID; /* What to return here? */
 347
 348         /* go ahead and re-read the services immediately */
 349         reload_services( False );
 350
 351         if ( lp_servicenumber( sharename )  < 0 )
 352                 return WERR_ACCESS_DENIED;
 353
 354         return WERR_OK;
 355 }
 356
 357 /****************************************************************************
 358  Delete a printer given a handle.
 359 ****************************************************************************/
 360
 361 static WERROR delete_printer_handle(pipes_struct *p, POLICY_HND *hnd)
 362 {
 363         Printer_entry *Printer = find_printer_index_by_hnd(p, hnd);
 364
 365         if (!Printer) {
 366                 DEBUG(2,("delete_printer_handle: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(hnd)));
 367                 return WERR_BADFID;
 368         }
 369
 370         /*
 371          * It turns out that Windows allows delete printer on a handle
 372          * opened by an admin user, then used on a pipe handle created
 373          * by an anonymous user..... but they're working on security.... riiight !
 374          * JRA.
 375          */
 376
 377         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER) {
 378                 DEBUG(3, ("delete_printer_handle: denied by handle\n"));
 379                 return WERR_ACCESS_DENIED;
 380         }
 381
 382         /* this does not need a become root since the access check has been
 383            done on the handle already */
 384
 385         if (del_a_printer( Printer->sharename ) != 0) {
 386                 DEBUG(3,("Error deleting printer %s\n", Printer->sharename));
 387                 return WERR_BADFID;
 388         }
 389
 390         return delete_printer_hook( p->pipe_user.nt_user_token, Printer->sharename );
 391 }
 392
 393 /****************************************************************************
 394  Return the snum of a printer corresponding to an handle.
 395 ****************************************************************************/
 396
 397 static BOOL get_printer_snum(pipes_struct *p, POLICY_HND *hnd, int *number)
 398 {
 399         Printer_entry *Printer = find_printer_index_by_hnd(p, hnd);
 400
 401         if (!Printer) {
 402                 DEBUG(2,("get_printer_snum: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(hnd)));
 403                 return False;
 404         }
 405
 406         switch (Printer->printer_type) {
 407                 case SPLHND_PRINTER:
 408                         DEBUG(4,("short name:%s\n", Printer->sharename));
 409                         *number = print_queue_snum(Printer->sharename);
 410                         return (*number != -1);
 411                 case SPLHND_SERVER:
 412                         return False;
 413                 default:
 414                         return False;
 415         }
 416 }
 417
 418 /****************************************************************************
 419  Set printer handle type.
 420  Check if it's \\server or \\server\printer
 421 ****************************************************************************/
 422
 423 static BOOL set_printer_hnd_printertype(Printer_entry *Printer, char *handlename)
 424 {
 425         DEBUG(3,("Setting printer type=%s\n", handlename));
 426
 427         if ( strlen(handlename) < 3 ) {
 428                 DEBUGADD(4,("A print server must have at least 1 char ! %s\n", handlename));
 429                 return False;
 430         }
 431
 432         /* it's a print server */
 433         if (*handlename=='\\' && *(handlename+1)=='\\' && !strchr_m(handlename+2, '\\')) {
 434                 DEBUGADD(4,("Printer is a print server\n"));
 435                 Printer->printer_type = SPLHND_SERVER;
 436         }
 437         /* it's a printer (set_printer_hnd_name() will handle port monitors */
 438         else {
 439                 DEBUGADD(4,("Printer is a printer\n"));
 440                 Printer->printer_type = SPLHND_PRINTER;
 441         }
 442
 443         return True;
 444 }
 445
 446 /****************************************************************************
 447  Set printer handle name..  Accept names like \\server, \\server\printer,
 448  \\server\SHARE, & "\\server\,XcvMonitor Standard TCP/IP Port"    See
 449  the MSDN docs regarding OpenPrinter() for details on the XcvData() and
 450  XcvDataPort() interface.
 451 ****************************************************************************/
 452
 453 static BOOL set_printer_hnd_name(Printer_entry *Printer, char *handlename)
 454 {
 455         int snum;
 456         int n_services=lp_numservices();
 457         char *aprinter, *printername;
 458         const char *servername;
 459         fstring sname;
 460         BOOL found=False;
 461         NT_PRINTER_INFO_LEVEL *printer;
 462         WERROR result;
 463
 464         DEBUG(4,("Setting printer name=%s (len=%lu)\n", handlename, (unsigned long)strlen(handlename)));
 465
 466         aprinter = handlename;
 467         if ( *handlename == '\\' ) {
 468                 servername = handlename + 2;
 469                 if ( (aprinter = strchr_m( handlename+2, '\\' )) != NULL ) {
 470                         *aprinter = '\0';
 471                         aprinter++;
 472                 }
 473         }
 474         else {
 475                 servername = "";
 476         }
 477
 478         /* save the servername to fill in replies on this handle */
 479
 480         if ( !is_myname_or_ipaddr( servername ) )
 481                 return False;
 482
 483         fstrcpy( Printer->servername, servername );
 484
 485         if ( Printer->printer_type == SPLHND_SERVER )
 486                 return True;
 487
 488         if ( Printer->printer_type != SPLHND_PRINTER )
 489                 return False;
 490
 491         DEBUGADD(5, ("searching for [%s]\n", aprinter ));
 492
 493         /* check for the Port Monitor Interface */
 494
 495         if ( strequal( aprinter, SPL_XCV_MONITOR_TCPMON ) ) {
 496                 Printer->printer_type = SPLHND_PORTMON_TCP;
 497                 fstrcpy(sname, SPL_XCV_MONITOR_TCPMON);
 498                 found = True;
 499         }
 500         else if ( strequal( aprinter, SPL_XCV_MONITOR_LOCALMON ) ) {
 501                 Printer->printer_type = SPLHND_PORTMON_LOCAL;
 502                 fstrcpy(sname, SPL_XCV_MONITOR_LOCALMON);
 503                 found = True;
 504         }
 505
 506         /* Search all sharenames first as this is easier than pulling
 507            the printer_info_2 off of disk. Don't use find_service() since
 508            that calls out to map_username() */
 509
 510         /* do another loop to look for printernames */
 511
 512         for (snum=0; !found && snum<n_services; snum++) {
 513
 514                 /* no point going on if this is not a printer */
 515
 516                 if ( !(lp_snum_ok(snum) && lp_print_ok(snum)) )
 517                         continue;
 518
 519                 fstrcpy(sname, lp_servicename(snum));
 520                 if ( strequal( aprinter, sname ) ) {
 521                         found = True;
 522                         break;
 523                 }
 524
 525                 /* no point looking up the printer object if
 526                    we aren't allowing printername != sharename */
 527
 528                 if ( lp_force_printername(snum) )
 529                         continue;
 530
 531                 fstrcpy(sname, lp_servicename(snum));
 532
 533                 printer = NULL;
 534                 result = get_a_printer( NULL, &printer, 2, sname );
 535                 if ( !W_ERROR_IS_OK(result) ) {
 536                         DEBUG(0,("set_printer_hnd_name: failed to lookup printer [%s] -- result [%s]\n",
 537                                 sname, dos_errstr(result)));
 538                         continue;
 539                 }
 540
 541                 /* printername is always returned as \\server\printername */
 542                 if ( !(printername = strchr_m(&printer->info_2->printername[2], '\\')) ) {
 543                         DEBUG(0,("set_printer_hnd_name: info2->printername in wrong format! [%s]\n",
 544                                 printer->info_2->printername));
 545                         free_a_printer( &printer, 2);
 546                         continue;
 547                 }
 548
 549                 printername++;
 550
 551                 if ( strequal(printername, aprinter) ) {
 552                         found = True;
 553                         break;
 554                 }
 555
 556                 DEBUGADD(10, ("printername: %s\n", printername));
 557
 558                 free_a_printer( &printer, 2);
 559         }
 560
 561         if ( !found ) {
 562                 DEBUGADD(4,("Printer not found\n"));
 563                 return False;
 564         }
 565
 566         DEBUGADD(4,("set_printer_hnd_name: Printer found: %s -> %s\n", aprinter, sname));
 567
 568         fstrcpy(Printer->sharename, sname);
 569
 570         return True;
 571 }
 572
 573 /****************************************************************************
 574  Find first available printer slot. creates a printer handle for you.
 575  ****************************************************************************/
 576
 577 static BOOL open_printer_hnd(pipes_struct *p, POLICY_HND *hnd, char *name, uint32 access_granted)
 578 {
 579         Printer_entry *new_printer;
 580
 581         DEBUG(10,("open_printer_hnd: name [%s]\n", name));
 582
 583         if((new_printer=SMB_MALLOC_P(Printer_entry)) == NULL)
 584                 return False;
 585
 586         ZERO_STRUCTP(new_printer);
 587
 588         if (!create_policy_hnd(p, hnd, free_printer_entry, new_printer)) {
 589                 SAFE_FREE(new_printer);
 590                 return False;
 591         }
 592
 593         /* Add to the internal list. */
 594         DLIST_ADD(printers_list, new_printer);
 595
 596         new_printer->notify.option=NULL;
 597
 598         if ( !(new_printer->ctx = talloc_init("Printer Entry [%p]", hnd)) ) {
 599                 DEBUG(0,("open_printer_hnd: talloc_init() failed!\n"));
 600                 close_printer_handle(p, hnd);
 601                 return False;
 602         }
 603
 604         if (!set_printer_hnd_printertype(new_printer, name)) {
 605                 close_printer_handle(p, hnd);
 606                 return False;
 607         }
 608
 609         if (!set_printer_hnd_name(new_printer, name)) {
 610                 close_printer_handle(p, hnd);
 611                 return False;
 612         }
 613
 614         new_printer->access_granted = access_granted;
 615
 616         DEBUG(5, ("%d printer handles active\n", (int)p->pipe_handles->count ));
 617
 618         return True;
 619 }
 620
 621 /***************************************************************************
 622  check to see if the client motify handle is monitoring the notification
 623  given by (notify_type, notify_field).
 624  **************************************************************************/
 625
 626 static BOOL is_monitoring_event_flags(uint32 flags, uint16 notify_type,
 627                                       uint16 notify_field)
 628 {
 629         return True;
 630 }
 631
 632 static BOOL is_monitoring_event(Printer_entry *p, uint16 notify_type,
 633                                 uint16 notify_field)
 634 {
 635         SPOOL_NOTIFY_OPTION *option = p->notify.option;
 636         uint32 i, j;
 637
 638         /*
 639          * Flags should always be zero when the change notify
 640          * is registered by the client's spooler.  A user Win32 app
 641          * might use the flags though instead of the NOTIFY_OPTION_INFO
 642          * --jerry
 643          */
 644
 645         if (!option) {
 646                 return False;
 647         }
 648
 649         if (p->notify.flags)
 650                 return is_monitoring_event_flags(
 651                         p->notify.flags, notify_type, notify_field);
 652
 653         for (i = 0; i < option->count; i++) {
 654
 655                 /* Check match for notify_type */
 656
 657                 if (option->ctr.type[i].type != notify_type)
 658                         continue;
 659
 660                 /* Check match for field */
 661
 662                 for (j = 0; j < option->ctr.type[i].count; j++) {
 663                         if (option->ctr.type[i].fields[j] == notify_field) {
 664                                 return True;
 665                         }
 666                 }
 667         }
 668
 669         DEBUG(10, ("Open handle for \\\\%s\\%s is not monitoring 0x%02x/0x%02x\n",
 670                    p->servername, p->sharename, notify_type, notify_field));
 671
 672         return False;
 673 }
 674
 675 /* Convert a notification message to a SPOOL_NOTIFY_INFO_DATA struct */
 676
 677 static void notify_one_value(struct spoolss_notify_msg *msg,
 678                              SPOOL_NOTIFY_INFO_DATA *data,
 679                              TALLOC_CTX *mem_ctx)
 680 {
 681         data->notify_data.value[0] = msg->notify.value[0];
 682         data->notify_data.value[1] = 0;
 683 }
 684
 685 static void notify_string(struct spoolss_notify_msg *msg,
 686                           SPOOL_NOTIFY_INFO_DATA *data,
 687                           TALLOC_CTX *mem_ctx)
 688 {
 689         UNISTR2 unistr;
 690
 691         /* The length of the message includes the trailing \0 */
 692
 693         init_unistr2(&unistr, msg->notify.data, UNI_STR_TERMINATE);
 694
 695         data->notify_data.data.length = msg->len * 2;
 696         data->notify_data.data.string = TALLOC_ARRAY(mem_ctx, uint16, msg->len);
 697
 698         if (!data->notify_data.data.string) {
 699                 data->notify_data.data.length = 0;
 700                 return;
 701         }
 702
 703         memcpy(data->notify_data.data.string, unistr.buffer, msg->len * 2);
 704 }
 705
 706 static void notify_system_time(struct spoolss_notify_msg *msg,
 707                                SPOOL_NOTIFY_INFO_DATA *data,
 708                                TALLOC_CTX *mem_ctx)
 709 {
 710         SYSTEMTIME systime;
 711         prs_struct ps;
 712
 713         if (msg->len != sizeof(time_t)) {
 714                 DEBUG(5, ("notify_system_time: received wrong sized message (%d)\n",
 715                           msg->len));
 716                 return;
 717         }
 718
 719         if (!prs_init(&ps, RPC_MAX_PDU_FRAG_LEN, mem_ctx, MARSHALL)) {
 720                 DEBUG(5, ("notify_system_time: prs_init() failed\n"));
 721                 return;
 722         }
 723
 724         if (!make_systemtime(&systime, gmtime((time_t *)msg->notify.data))) {
 725                 DEBUG(5, ("notify_system_time: unable to make systemtime\n"));
 726                 return;
 727         }
 728
 729         if (!spoolss_io_system_time("", &ps, 0, &systime))
 730                 return;
 731
 732         data->notify_data.data.length = prs_offset(&ps);
 733         data->notify_data.data.string = TALLOC(mem_ctx, prs_offset(&ps));
 734
 735         prs_copy_all_data_out((char *)data->notify_data.data.string, &ps);
 736
 737         prs_mem_free(&ps);
 738 }
 739
 740 struct notify2_message_table {
 741         const char *name;
 742         void (*fn)(struct spoolss_notify_msg *msg,
 743                    SPOOL_NOTIFY_INFO_DATA *data, TALLOC_CTX *mem_ctx);
 744 };
 745
 746 static struct notify2_message_table printer_notify_table[] = {
 747         /* 0x00 */ { "PRINTER_NOTIFY_SERVER_NAME", notify_string },
 748         /* 0x01 */ { "PRINTER_NOTIFY_PRINTER_NAME", notify_string },
 749         /* 0x02 */ { "PRINTER_NOTIFY_SHARE_NAME", notify_string },
 750         /* 0x03 */ { "PRINTER_NOTIFY_PORT_NAME", notify_string },
 751         /* 0x04 */ { "PRINTER_NOTIFY_DRIVER_NAME", notify_string },
 752         /* 0x05 */ { "PRINTER_NOTIFY_COMMENT", notify_string },
 753         /* 0x06 */ { "PRINTER_NOTIFY_LOCATION", notify_string },
 754         /* 0x07 */ { "PRINTER_NOTIFY_DEVMODE", NULL },
 755         /* 0x08 */ { "PRINTER_NOTIFY_SEPFILE", notify_string },
 756         /* 0x09 */ { "PRINTER_NOTIFY_PRINT_PROCESSOR", notify_string },
 757         /* 0x0a */ { "PRINTER_NOTIFY_PARAMETERS", NULL },
 758         /* 0x0b */ { "PRINTER_NOTIFY_DATATYPE", notify_string },
 759         /* 0x0c */ { "PRINTER_NOTIFY_SECURITY_DESCRIPTOR", NULL },
 760         /* 0x0d */ { "PRINTER_NOTIFY_ATTRIBUTES", notify_one_value },
 761         /* 0x0e */ { "PRINTER_NOTIFY_PRIORITY", notify_one_value },
 762         /* 0x0f */ { "PRINTER_NOTIFY_DEFAULT_PRIORITY", NULL },
 763         /* 0x10 */ { "PRINTER_NOTIFY_START_TIME", NULL },
 764         /* 0x11 */ { "PRINTER_NOTIFY_UNTIL_TIME", NULL },
 765         /* 0x12 */ { "PRINTER_NOTIFY_STATUS", notify_one_value },
 766 };
 767
 768 static struct notify2_message_table job_notify_table[] = {
 769         /* 0x00 */ { "JOB_NOTIFY_PRINTER_NAME", NULL },
 770         /* 0x01 */ { "JOB_NOTIFY_MACHINE_NAME", NULL },
 771         /* 0x02 */ { "JOB_NOTIFY_PORT_NAME", NULL },
 772         /* 0x03 */ { "JOB_NOTIFY_USER_NAME", notify_string },
 773         /* 0x04 */ { "JOB_NOTIFY_NOTIFY_NAME", NULL },
 774         /* 0x05 */ { "JOB_NOTIFY_DATATYPE", NULL },
 775         /* 0x06 */ { "JOB_NOTIFY_PRINT_PROCESSOR", NULL },
 776         /* 0x07 */ { "JOB_NOTIFY_PARAMETERS", NULL },
 777         /* 0x08 */ { "JOB_NOTIFY_DRIVER_NAME", NULL },
 778         /* 0x09 */ { "JOB_NOTIFY_DEVMODE", NULL },
 779         /* 0x0a */ { "JOB_NOTIFY_STATUS", notify_one_value },
 780         /* 0x0b */ { "JOB_NOTIFY_STATUS_STRING", NULL },
 781         /* 0x0c */ { "JOB_NOTIFY_SECURITY_DESCRIPTOR", NULL },
 782         /* 0x0d */ { "JOB_NOTIFY_DOCUMENT", notify_string },
 783         /* 0x0e */ { "JOB_NOTIFY_PRIORITY", NULL },
 784         /* 0x0f */ { "JOB_NOTIFY_POSITION", NULL },
 785         /* 0x10 */ { "JOB_NOTIFY_SUBMITTED", notify_system_time },
 786         /* 0x11 */ { "JOB_NOTIFY_START_TIME", NULL },
 787         /* 0x12 */ { "JOB_NOTIFY_UNTIL_TIME", NULL },
 788         /* 0x13 */ { "JOB_NOTIFY_TIME", NULL },
 789         /* 0x14 */ { "JOB_NOTIFY_TOTAL_PAGES", notify_one_value },
 790         /* 0x15 */ { "JOB_NOTIFY_PAGES_PRINTED", NULL },
 791         /* 0x16 */ { "JOB_NOTIFY_TOTAL_BYTES", notify_one_value },
 792         /* 0x17 */ { "JOB_NOTIFY_BYTES_PRINTED", NULL },
 793 };
 794
 795
 796 /***********************************************************************
 797  Allocate talloc context for container object
 798  **********************************************************************/
 799
 800 static void notify_msg_ctr_init( SPOOLSS_NOTIFY_MSG_CTR *ctr )
 801 {
 802         if ( !ctr )
 803                 return;
 804
 805         ctr->ctx = talloc_init("notify_msg_ctr_init %p", ctr);
 806
 807         return;
 808 }
 809
 810 /***********************************************************************
 811  release all allocated memory and zero out structure
 812  **********************************************************************/
 813
 814 static void notify_msg_ctr_destroy( SPOOLSS_NOTIFY_MSG_CTR *ctr )
 815 {
 816         if ( !ctr )
 817                 return;
 818
 819         if ( ctr->ctx )
 820                 talloc_destroy(ctr->ctx);
 821
 822         ZERO_STRUCTP(ctr);
 823
 824         return;
 825 }
 826
 827 /***********************************************************************
 828  **********************************************************************/
 829
 830 static TALLOC_CTX* notify_ctr_getctx( SPOOLSS_NOTIFY_MSG_CTR *ctr )
 831 {
 832         if ( !ctr )
 833                 return NULL;
 834
 835         return ctr->ctx;
 836 }
 837
 838 /***********************************************************************
 839  **********************************************************************/
 840
 841 static SPOOLSS_NOTIFY_MSG_GROUP* notify_ctr_getgroup( SPOOLSS_NOTIFY_MSG_CTR *ctr, uint32 idx )
 842 {
 843         if ( !ctr || !ctr->msg_groups )
 844                 return NULL;
 845
 846         if ( idx >= ctr->num_groups )
 847                 return NULL;
 848
 849         return &ctr->msg_groups[idx];
 850
 851 }
 852
 853 /***********************************************************************
 854  How many groups of change messages do we have ?
 855  **********************************************************************/
 856
 857 static int notify_msg_ctr_numgroups( SPOOLSS_NOTIFY_MSG_CTR *ctr )
 858 {
 859         if ( !ctr )
 860                 return 0;
 861
 862         return ctr->num_groups;
 863 }
 864
 865 /***********************************************************************
 866  Add a SPOOLSS_NOTIFY_MSG_CTR to the correct group
 867  **********************************************************************/
 868
 869 static int notify_msg_ctr_addmsg( SPOOLSS_NOTIFY_MSG_CTR *ctr, SPOOLSS_NOTIFY_MSG *msg )
 870 {
 871         SPOOLSS_NOTIFY_MSG_GROUP        *groups = NULL;
 872         SPOOLSS_NOTIFY_MSG_GROUP        *msg_grp = NULL;
 873         SPOOLSS_NOTIFY_MSG              *msg_list = NULL;
 874         int                             i, new_slot;
 875
 876         if ( !ctr || !msg )
 877                 return 0;
 878
 879         /* loop over all groups looking for a matching printer name */
 880
 881         for ( i=0; i<ctr->num_groups; i++ ) {
 882                 if ( strcmp(ctr->msg_groups[i].printername, msg->printer) == 0 )
 883                         break;
 884         }
 885
 886         /* add a new group? */
 887
 888         if ( i == ctr->num_groups ) {
 889                 ctr->num_groups++;
 890
 891                 if ( !(groups = TALLOC_REALLOC_ARRAY( ctr->ctx, ctr->msg_groups, SPOOLSS_NOTIFY_MSG_GROUP, ctr->num_groups)) ) {
 892                         DEBUG(0,("notify_msg_ctr_addmsg: talloc_realloc() failed!\n"));
 893                         return 0;
 894                 }
 895                 ctr->msg_groups = groups;
 896
 897                 /* clear the new entry and set the printer name */
 898
 899                 ZERO_STRUCT( ctr->msg_groups[ctr->num_groups-1] );
 900                 fstrcpy( ctr->msg_groups[ctr->num_groups-1].printername, msg->printer );
 901         }
 902
 903         /* add the change messages; 'i' is the correct index now regardless */
 904
 905         msg_grp = &ctr->msg_groups[i];
 906
 907         msg_grp->num_msgs++;
 908
 909         if ( !(msg_list = TALLOC_REALLOC_ARRAY( ctr->ctx, msg_grp->msgs, SPOOLSS_NOTIFY_MSG, msg_grp->num_msgs )) ) {
 910                 DEBUG(0,("notify_msg_ctr_addmsg: talloc_realloc() failed for new message [%d]!\n", msg_grp->num_msgs));
 911                 return 0;
 912         }
 913         msg_grp->msgs = msg_list;
 914
 915         new_slot = msg_grp->num_msgs-1;
 916         memcpy( &msg_grp->msgs[new_slot], msg, sizeof(SPOOLSS_NOTIFY_MSG) );
 917
 918         /* need to allocate own copy of data */
 919
 920         if ( msg->len != 0 )
 921                 msg_grp->msgs[new_slot].notify.data = TALLOC_MEMDUP( ctr->ctx, msg->notify.data, msg->len );
 922
 923         return ctr->num_groups;
 924 }
 925
 926 /***********************************************************************
 927  Send a change notication message on all handles which have a call
 928  back registered
 929  **********************************************************************/
 930
 931 static void send_notify2_changes( SPOOLSS_NOTIFY_MSG_CTR *ctr, uint32 idx )
 932 {
 933         Printer_entry            *p;
 934         TALLOC_CTX               *mem_ctx = notify_ctr_getctx( ctr );
 935         SPOOLSS_NOTIFY_MSG_GROUP *msg_group = notify_ctr_getgroup( ctr, idx );
 936         SPOOLSS_NOTIFY_MSG       *messages;
 937         int                      sending_msg_count;
 938
 939         if ( !msg_group ) {
 940                 DEBUG(5,("send_notify2_changes() called with no msg group!\n"));
 941                 return;
 942         }
 943
 944         messages = msg_group->msgs;
 945
 946         if ( !messages ) {
 947                 DEBUG(5,("send_notify2_changes() called with no messages!\n"));
 948                 return;
 949         }
 950
 951         DEBUG(8,("send_notify2_changes: Enter...[%s]\n", msg_group->printername));
 952
 953         /* loop over all printers */
 954
 955         for (p = printers_list; p; p = p->next) {
 956                 SPOOL_NOTIFY_INFO_DATA *data;
 957                 uint32  data_len = 0;
 958                 uint32  id;
 959                 int     i;
 960
 961                 /* Is there notification on this handle? */
 962
 963                 if ( !p->notify.client_connected )
 964                         continue;
 965
 966                 DEBUG(10,("Client connected! [\\\\%s\\%s]\n", p->servername, p->sharename));
 967
 968                 /* For this printer?  Print servers always receive
 969                    notifications. */
 970
 971                 if ( ( p->printer_type == SPLHND_PRINTER )  &&
 972                     ( !strequal(msg_group->printername, p->sharename) ) )
 973                         continue;
 974
 975                 DEBUG(10,("Our printer\n"));
 976
 977                 /* allocate the max entries possible */
 978
 979                 data = TALLOC_ARRAY( mem_ctx, SPOOL_NOTIFY_INFO_DATA, msg_group->num_msgs);
 980                 ZERO_STRUCTP(data);
 981
 982                 /* build the array of change notifications */
 983
 984                 sending_msg_count = 0;
 985
 986                 for ( i=0; i<msg_group->num_msgs; i++ ) {
 987                         SPOOLSS_NOTIFY_MSG      *msg = &messages[i];
 988
 989                         /* Are we monitoring this event? */
 990
 991                         if (!is_monitoring_event(p, msg->type, msg->field))
 992                                 continue;
 993
 994                         sending_msg_count++;
 995
 996
 997                         DEBUG(10,("process_notify2_message: Sending message type [0x%x] field [0x%2x] for printer [%s]\n",
 998                                 msg->type, msg->field, p->sharename));
 999
1000                         /*
1001                          * if the is a printer notification handle and not a job notification
1002                          * type, then set the id to 0.  Other wise just use what was specified
1003                          * in the message.
1004                          *
1005                          * When registering change notification on a print server handle
1006                          * we always need to send back the id (snum) matching the printer
1007                          * for which the change took place.  For change notify registered
1008                          * on a printer handle, this does not matter and the id should be 0.
1009                          *
1010                          * --jerry
1011                          */
1012
1013                         if ( ( p->printer_type == SPLHND_PRINTER ) && ( msg->type == PRINTER_NOTIFY_TYPE ) )
1014                                 id = 0;
1015                         else
1016                                 id = msg->id;
1017
1018
1019                         /* Convert unix jobid to smb jobid */
1020
1021                         if (msg->flags & SPOOLSS_NOTIFY_MSG_UNIX_JOBID) {
1022                                 id = sysjob_to_jobid(msg->id);
1023
1024                                 if (id == -1) {
1025                                         DEBUG(3, ("no such unix jobid %d\n", msg->id));
1026                                         goto done;
1027                                 }
1028                         }
1029
1030                         construct_info_data( &data[data_len], msg->type, msg->field, id );
1031
1032                         switch(msg->type) {
1033                         case PRINTER_NOTIFY_TYPE:
1034                                 if ( printer_notify_table[msg->field].fn )
1035                                         printer_notify_table[msg->field].fn(msg, &data[data_len], mem_ctx);
1036                                 break;
1037
1038                         case JOB_NOTIFY_TYPE:
1039                                 if ( job_notify_table[msg->field].fn )
1040                                         job_notify_table[msg->field].fn(msg, &data[data_len], mem_ctx);
1041                                 break;
1042
1043                         default:
1044                                 DEBUG(5, ("Unknown notification type %d\n", msg->type));
1045                                 goto done;
1046                         }
1047
1048                         data_len++;
1049                 }
1050
1051                 if ( sending_msg_count ) {
1052                         rpccli_spoolss_rrpcn( notify_cli_pipe, mem_ctx, &p->notify.client_hnd,
1053                                         data_len, data, p->notify.change, 0 );
1054                 }
1055         }
1056
1057 done:
1058         DEBUG(8,("send_notify2_changes: Exit...\n"));
1059         return;
1060 }
1061
1062 /***********************************************************************
1063  **********************************************************************/
1064
1065 static BOOL notify2_unpack_msg( SPOOLSS_NOTIFY_MSG *msg, struct timeval *tv, void *buf, size_t len )
1066 {
1067
1068         uint32 tv_sec, tv_usec;
1069         size_t offset = 0;
1070
1071         /* Unpack message */
1072
1073         offset += tdb_unpack((char *)buf + offset, len - offset, "f",
1074                              msg->printer);
1075
1076         offset += tdb_unpack((char *)buf + offset, len - offset, "ddddddd",
1077                                 &tv_sec, &tv_usec,
1078                                 &msg->type, &msg->field, &msg->id, &msg->len, &msg->flags);
1079
1080         if (msg->len == 0)
1081                 tdb_unpack((char *)buf + offset, len - offset, "dd",
1082                            &msg->notify.value[0], &msg->notify.value[1]);
1083         else
1084                 tdb_unpack((char *)buf + offset, len - offset, "B",
1085                            &msg->len, &msg->notify.data);
1086
1087         DEBUG(3, ("notify2_unpack_msg: got NOTIFY2 message for printer %s, jobid %u type %d, field 0x%02x, flags 0x%04x\n",
1088                   msg->printer, (unsigned int)msg->id, msg->type, msg->field, msg->flags));
1089
1090         tv->tv_sec = tv_sec;
1091         tv->tv_usec = tv_usec;
1092
1093         if (msg->len == 0)
1094                 DEBUG(3, ("notify2_unpack_msg: value1 = %d, value2 = %d\n", msg->notify.value[0],
1095                           msg->notify.value[1]));
1096         else
1097                 dump_data(3, msg->notify.data, msg->len);
1098
1099         return True;
1100 }
1101
1102 /********************************************************************
1103  Receive a notify2 message list
1104  ********************************************************************/
1105
1106 static void receive_notify2_message_list(int msg_type, struct process_id src,
1107                                          void *msg, size_t len)
1108 {
1109         size_t                  msg_count, i;
1110         char                    *buf = (char *)msg;
1111         char                    *msg_ptr;
1112         size_t                  msg_len;
1113         SPOOLSS_NOTIFY_MSG      notify;
1114         SPOOLSS_NOTIFY_MSG_CTR  messages;
1115         int                     num_groups;
1116
1117         if (len < 4) {
1118                 DEBUG(0,("receive_notify2_message_list: bad message format (len < 4)!\n"));
1119                 return;
1120         }
1121
1122         msg_count = IVAL(buf, 0);
1123         msg_ptr = buf + 4;
1124
1125         DEBUG(5, ("receive_notify2_message_list: got %lu messages in list\n", (unsigned long)msg_count));
1126
1127         if (msg_count == 0) {
1128                 DEBUG(0,("receive_notify2_message_list: bad message format (msg_count == 0) !\n"));
1129                 return;
1130         }
1131
1132         /* initialize the container */
1133
1134         ZERO_STRUCT( messages );
1135         notify_msg_ctr_init( &messages );
1136
1137         /*
1138          * build message groups for each printer identified
1139          * in a change_notify msg.  Remember that a PCN message
1140          * includes the handle returned for the srv_spoolss_replyopenprinter()
1141          * call.  Therefore messages are grouped according to printer handle.
1142          */
1143
1144         for ( i=0; i<msg_count; i++ ) {
1145                 struct timeval msg_tv;
1146
1147                 if (msg_ptr + 4 - buf > len) {
1148                         DEBUG(0,("receive_notify2_message_list: bad message format (len > buf_size) !\n"));
1149                         return;
1150                 }
1151
1152                 msg_len = IVAL(msg_ptr,0);
1153                 msg_ptr += 4;
1154
1155                 if (msg_ptr + msg_len - buf > len) {
1156                         DEBUG(0,("receive_notify2_message_list: bad message format (bad len) !\n"));
1157                         return;
1158                 }
1159
1160                 /* unpack messages */
1161
1162                 ZERO_STRUCT( notify );
1163                 notify2_unpack_msg( &notify, &msg_tv, msg_ptr, msg_len );
1164                 msg_ptr += msg_len;
1165
1166                 /* add to correct list in container */
1167
1168                 notify_msg_ctr_addmsg( &messages, &notify );
1169
1170                 /* free memory that might have been allocated by notify2_unpack_msg() */
1171
1172                 if ( notify.len != 0 )
1173                         SAFE_FREE( notify.notify.data );
1174         }
1175
1176         /* process each group of messages */
1177
1178         num_groups = notify_msg_ctr_numgroups( &messages );
1179         for ( i=0; i<num_groups; i++ )
1180                 send_notify2_changes( &messages, i );
1181
1182
1183         /* cleanup */
1184
1185         DEBUG(10,("receive_notify2_message_list: processed %u messages\n", (uint32)msg_count ));
1186
1187         notify_msg_ctr_destroy( &messages );
1188
1189         return;
1190 }
1191
1192 /********************************************************************
1193  Send a message to ourself about new driver being installed
1194  so we can upgrade the information for each printer bound to this
1195  driver
1196  ********************************************************************/
1197
1198 static BOOL srv_spoolss_drv_upgrade_printer(char* drivername)
1199 {
1200         int len = strlen(drivername);
1201
1202         if (!len)
1203                 return False;
1204
1205         DEBUG(10,("srv_spoolss_drv_upgrade_printer: Sending message about driver upgrade [%s]\n",
1206                 drivername));
1207
1208         message_send_pid(pid_to_procid(sys_getpid()),
1209                          MSG_PRINTER_DRVUPGRADE, drivername, len+1, False);
1210
1211         return True;
1212 }
1213
1214 /**********************************************************************
1215  callback to receive a MSG_PRINTER_DRVUPGRADE message and interate
1216  over all printers, upgrading ones as necessary
1217  **********************************************************************/
1218
1219 void do_drv_upgrade_printer(int msg_type, struct process_id src, void *buf, size_t len)
1220 {
1221         fstring drivername;
1222         int snum;
1223         int n_services = lp_numservices();
1224
1225         len = MIN(len,sizeof(drivername)-1);
1226         strncpy(drivername, buf, len);
1227
1228         DEBUG(10,("do_drv_upgrade_printer: Got message for new driver [%s]\n", drivername ));
1229
1230         /* Iterate the printer list */
1231
1232         for (snum=0; snum<n_services; snum++)
1233         {
1234                 if (lp_snum_ok(snum) && lp_print_ok(snum) )
1235                 {
1236                         WERROR result;
1237                         NT_PRINTER_INFO_LEVEL *printer = NULL;
1238
1239                         result = get_a_printer(NULL, &printer, 2, lp_const_servicename(snum));
1240                         if (!W_ERROR_IS_OK(result))
1241                                 continue;
1242
1243                         if (printer && printer->info_2 && !strcmp(drivername, printer->info_2->drivername))
1244                         {
1245                                 DEBUG(6,("Updating printer [%s]\n", printer->info_2->printername));
1246
1247                                 /* all we care about currently is the change_id */
1248
1249                                 result = mod_a_printer(printer, 2);
1250                                 if (!W_ERROR_IS_OK(result)) {
1251                                         DEBUG(3,("do_drv_upgrade_printer: mod_a_printer() failed with status [%s]\n",
1252                                                 dos_errstr(result)));
1253                                 }
1254                         }
1255
1256                         free_a_printer(&printer, 2);
1257                 }
1258         }
1259
1260         /* all done */
1261 }
1262
1263 /********************************************************************
1264  Update the cache for all printq's with a registered client
1265  connection
1266  ********************************************************************/
1267
1268 void update_monitored_printq_cache( void )
1269 {
1270         Printer_entry *printer = printers_list;
1271         int snum;
1272
1273         /* loop through all printers and update the cache where
1274            client_connected == True */
1275         while ( printer )
1276         {
1277                 if ( (printer->printer_type == SPLHND_PRINTER)
1278                         && printer->notify.client_connected )
1279                 {
1280                         snum = print_queue_snum(printer->sharename);
1281                         print_queue_status( snum, NULL, NULL );
1282                 }
1283
1284                 printer = printer->next;
1285         }
1286
1287         return;
1288 }
1289 /********************************************************************
1290  Send a message to ourself about new driver being installed
1291  so we can upgrade the information for each printer bound to this
1292  driver
1293  ********************************************************************/
1294
1295 static BOOL srv_spoolss_reset_printerdata(char* drivername)
1296 {
1297         int len = strlen(drivername);
1298
1299         if (!len)
1300                 return False;
1301
1302         DEBUG(10,("srv_spoolss_reset_printerdata: Sending message about resetting printerdata [%s]\n",
1303                 drivername));
1304
1305         message_send_pid(pid_to_procid(sys_getpid()),
1306                          MSG_PRINTERDATA_INIT_RESET, drivername, len+1, False);
1307
1308         return True;
1309 }
1310
1311 /**********************************************************************
1312  callback to receive a MSG_PRINTERDATA_INIT_RESET message and interate
1313  over all printers, resetting printer data as neessary
1314  **********************************************************************/
1315
1316 void reset_all_printerdata(int msg_type, struct process_id src,
1317                            void *buf, size_t len)
1318 {
1319         fstring drivername;
1320         int snum;
1321         int n_services = lp_numservices();
1322
1323         len = MIN( len, sizeof(drivername)-1 );
1324         strncpy( drivername, buf, len );
1325
1326         DEBUG(10,("reset_all_printerdata: Got message for new driver [%s]\n", drivername ));
1327
1328         /* Iterate the printer list */
1329
1330         for ( snum=0; snum<n_services; snum++ )
1331         {
1332                 if ( lp_snum_ok(snum) && lp_print_ok(snum) )
1333                 {
1334                         WERROR result;
1335                         NT_PRINTER_INFO_LEVEL *printer = NULL;
1336
1337                         result = get_a_printer( NULL, &printer, 2, lp_const_servicename(snum) );
1338                         if ( !W_ERROR_IS_OK(result) )
1339                                 continue;
1340
1341                         /*
1342                          * if the printer is bound to the driver,
1343                          * then reset to the new driver initdata
1344                          */
1345
1346                         if ( printer && printer->info_2 && !strcmp(drivername, printer->info_2->drivername) )
1347                         {
1348                                 DEBUG(6,("reset_all_printerdata: Updating printer [%s]\n", printer->info_2->printername));
1349
1350                                 if ( !set_driver_init(printer, 2) ) {
1351                                         DEBUG(5,("reset_all_printerdata: Error resetting printer data for printer [%s], driver [%s]!\n",
1352                                                 printer->info_2->printername, printer->info_2->drivername));
1353                                 }
1354
1355                                 result = mod_a_printer( printer, 2 );
1356                                 if ( !W_ERROR_IS_OK(result) ) {
1357                                         DEBUG(3,("reset_all_printerdata: mod_a_printer() failed!  (%s)\n",
1358                                                 get_dos_error_msg(result)));
1359                                 }
1360                         }
1361
1362                         free_a_printer( &printer, 2 );
1363                 }
1364         }
1365
1366         /* all done */
1367
1368         return;
1369 }
1370
1371 /********************************************************************
1372  Copy routines used by convert_to_openprinterex()
1373  *******************************************************************/
1374
1375 static DEVICEMODE* dup_devicemode(TALLOC_CTX *ctx, DEVICEMODE *devmode)
1376 {
1377         DEVICEMODE *d;
1378         int len;
1379
1380         if (!devmode)
1381                 return NULL;
1382
1383         DEBUG (8,("dup_devmode\n"));
1384
1385         /* bulk copy first */
1386
1387         d = TALLOC_MEMDUP(ctx, devmode, sizeof(DEVICEMODE));
1388         if (!d)
1389                 return NULL;
1390
1391         /* dup the pointer members separately */
1392
1393         len = unistrlen(devmode->devicename.buffer);
1394         if (len != -1) {
1395                 d->devicename.buffer = TALLOC_ARRAY(ctx, uint16, len);
1396                 if (unistrcpy(d->devicename.buffer, devmode->devicename.buffer) != len)
1397                         return NULL;
1398         }
1399
1400
1401         len = unistrlen(devmode->formname.buffer);
1402         if (len != -1) {
1403                 d->devicename.buffer = TALLOC_ARRAY(ctx, uint16, len);
1404                 if (unistrcpy(d->formname.buffer, devmode->formname.buffer) != len)
1405                         return NULL;
1406         }
1407
1408         d->dev_private = TALLOC_MEMDUP(ctx, devmode->dev_private, devmode->driverextra);
1409
1410         return d;
1411 }
1412
1413 static void copy_devmode_ctr(TALLOC_CTX *ctx, DEVMODE_CTR *new_ctr, DEVMODE_CTR *ctr)
1414 {
1415         if (!new_ctr || !ctr)
1416                 return;
1417
1418         DEBUG(8,("copy_devmode_ctr\n"));
1419
1420         new_ctr->size = ctr->size;
1421         new_ctr->devmode_ptr = ctr->devmode_ptr;
1422
1423         if(ctr->devmode_ptr)
1424                 new_ctr->devmode = dup_devicemode(ctx, ctr->devmode);
1425 }
1426
1427 static void copy_printer_default(TALLOC_CTX *ctx, PRINTER_DEFAULT *new_def, PRINTER_DEFAULT *def)
1428 {
1429         if (!new_def || !def)
1430                 return;
1431
1432         DEBUG(8,("copy_printer_defaults\n"));
1433
1434         new_def->datatype_ptr = def->datatype_ptr;
1435
1436         if (def->datatype_ptr)
1437                 copy_unistr2(&new_def->datatype, &def->datatype);
1438
1439         copy_devmode_ctr(ctx, &new_def->devmode_cont, &def->devmode_cont);
1440
1441         new_def->access_required = def->access_required;
1442 }
1443
1444 /********************************************************************
1445  * Convert a SPOOL_Q_OPEN_PRINTER structure to a
1446  * SPOOL_Q_OPEN_PRINTER_EX structure
1447  ********************************************************************/
1448
1449 static WERROR convert_to_openprinterex(TALLOC_CTX *ctx, SPOOL_Q_OPEN_PRINTER_EX *q_u_ex, SPOOL_Q_OPEN_PRINTER *q_u)
1450 {
1451         if (!q_u_ex || !q_u)
1452                 return WERR_OK;
1453
1454         DEBUG(8,("convert_to_openprinterex\n"));
1455
1456         if ( q_u->printername ) {
1457                 q_u_ex->printername = TALLOC_ZERO_P( ctx, UNISTR2 );
1458                 if (q_u_ex->printername == NULL)
1459                         return WERR_NOMEM;
1460                 copy_unistr2(q_u_ex->printername, q_u->printername);
1461         }
1462
1463         copy_printer_default(ctx, &q_u_ex->printer_default, &q_u->printer_default);
1464
1465         return WERR_OK;
1466 }
1467
1468 /********************************************************************
1469  * spoolss_open_printer
1470  *
1471  * called from the spoolss dispatcher
1472  ********************************************************************/
1473
1474 WERROR _spoolss_open_printer(pipes_struct *p, SPOOL_Q_OPEN_PRINTER *q_u, SPOOL_R_OPEN_PRINTER *r_u)
1475 {
1476         SPOOL_Q_OPEN_PRINTER_EX q_u_ex;
1477         SPOOL_R_OPEN_PRINTER_EX r_u_ex;
1478
1479         if (!q_u || !r_u)
1480                 return WERR_NOMEM;
1481
1482         ZERO_STRUCT(q_u_ex);
1483         ZERO_STRUCT(r_u_ex);
1484
1485         /* convert the OpenPrinter() call to OpenPrinterEx() */
1486
1487         r_u_ex.status = convert_to_openprinterex(p->mem_ctx, &q_u_ex, q_u);
1488         if (!W_ERROR_IS_OK(r_u_ex.status))
1489                 return r_u_ex.status;
1490
1491         r_u_ex.status = _spoolss_open_printer_ex(p, &q_u_ex, &r_u_ex);
1492
1493         /* convert back to OpenPrinter() */
1494
1495         memcpy(r_u, &r_u_ex, sizeof(*r_u));
1496
1497         return r_u->status;
1498 }
1499
1500 /********************************************************************
1501  ********************************************************************/
1502
1503 WERROR _spoolss_open_printer_ex( pipes_struct *p, SPOOL_Q_OPEN_PRINTER_EX *q_u, SPOOL_R_OPEN_PRINTER_EX *r_u)
1504 {
1505         PRINTER_DEFAULT         *printer_default = &q_u->printer_default;
1506         POLICY_HND              *handle = &r_u->handle;
1507
1508         fstring name;
1509         int snum;
1510         struct current_user user;
1511         Printer_entry *Printer=NULL;
1512
1513         if ( !q_u->printername )
1514                 return WERR_INVALID_PRINTER_NAME;
1515
1516         /* some sanity check because you can open a printer or a print server */
1517         /* aka: \\server\printer or \\server */
1518
1519         unistr2_to_ascii(name, q_u->printername, sizeof(name)-1);
1520
1521         DEBUGADD(3,("checking name: %s\n",name));
1522
1523         if (!open_printer_hnd(p, handle, name, 0))
1524                 return WERR_INVALID_PRINTER_NAME;
1525
1526         Printer=find_printer_index_by_hnd(p, handle);
1527         if ( !Printer ) {
1528                 DEBUG(0,(" _spoolss_open_printer_ex: logic error.  Can't find printer "
1529                         "handle we created for printer %s\n", name ));
1530                 close_printer_handle(p,handle);
1531                 return WERR_INVALID_PRINTER_NAME;
1532         }
1533
1534         get_current_user(&user, p);
1535
1536         /*
1537          * First case: the user is opening the print server:
1538          *
1539          * Disallow MS AddPrinterWizard if parameter disables it. A Win2k
1540          * client 1st tries an OpenPrinterEx with access==0, MUST be allowed.
1541          *
1542          * Then both Win2k and WinNT clients try an OpenPrinterEx with
1543          * SERVER_ALL_ACCESS, which we allow only if the user is root (uid=0)
1544          * or if the user is listed in the smb.conf printer admin parameter.
1545          *
1546          * Then they try OpenPrinterEx with SERVER_READ which we allow. This lets the
1547          * client view printer folder, but does not show the MSAPW.
1548          *
1549          * Note: this test needs code to check access rights here too. Jeremy
1550          * could you look at this?
1551          *
1552          * Second case: the user is opening a printer:
1553          * NT doesn't let us connect to a printer if the connecting user
1554          * doesn't have print permission.
1555          *
1556          * Third case: user is opening a Port Monitor
1557          * access checks same as opening a handle to the print server.
1558          */
1559
1560         switch (Printer->printer_type )
1561         {
1562         case SPLHND_SERVER:
1563         case SPLHND_PORTMON_TCP:
1564         case SPLHND_PORTMON_LOCAL:
1565                 /* Printserver handles use global struct... */
1566
1567                 snum = -1;
1568
1569                 /* Map standard access rights to object specific access rights */
1570
1571                 se_map_standard(&printer_default->access_required,
1572                                 &printserver_std_mapping);
1573
1574                 /* Deny any object specific bits that don't apply to print
1575                    servers (i.e printer and job specific bits) */
1576
1577                 printer_default->access_required &= SPECIFIC_RIGHTS_MASK;
1578
1579                 if (printer_default->access_required &
1580                     ~(SERVER_ACCESS_ADMINISTER | SERVER_ACCESS_ENUMERATE)) {
1581                         DEBUG(3, ("access DENIED for non-printserver bits\n"));
1582                         close_printer_handle(p, handle);
1583                         return WERR_ACCESS_DENIED;
1584                 }
1585
1586                 /* Allow admin access */
1587
1588                 if ( printer_default->access_required & SERVER_ACCESS_ADMINISTER )
1589                 {
1590                         SE_PRIV se_printop = SE_PRINT_OPERATOR;
1591
1592                         if (!lp_ms_add_printer_wizard()) {
1593                                 close_printer_handle(p, handle);
1594                                 return WERR_ACCESS_DENIED;
1595                         }
1596
1597                         /* if the user is not root, doesn't have SE_PRINT_OPERATOR privilege,
1598                            and not a printer admin, then fail */
1599
1600                         if ((user.ut.uid != 0) &&
1601                             !user_has_privileges(user.nt_user_token,
1602                                                  &se_printop ) &&
1603                             !token_contains_name_in_list(
1604                                     uidtoname(user.ut.uid), NULL,
1605                                     user.nt_user_token,
1606                                     lp_printer_admin(snum))) {
1607                                 close_printer_handle(p, handle);
1608                                 return WERR_ACCESS_DENIED;
1609                         }
1610
1611                         printer_default->access_required = SERVER_ACCESS_ADMINISTER;
1612                 }
1613                 else
1614                 {
1615                         printer_default->access_required = SERVER_ACCESS_ENUMERATE;
1616                 }
1617
1618                 DEBUG(4,("Setting print server access = %s\n", (printer_default->access_required == SERVER_ACCESS_ADMINISTER)
1619                         ? "SERVER_ACCESS_ADMINISTER" : "SERVER_ACCESS_ENUMERATE" ));
1620
1621                 /* We fall through to return WERR_OK */
1622                 break;
1623
1624         case SPLHND_PRINTER:
1625                 /* NT doesn't let us connect to a printer if the connecting user
1626                    doesn't have print permission.  */
1627
1628                 if (!get_printer_snum(p, handle, &snum)) {
1629                         close_printer_handle(p, handle);
1630                         return WERR_BADFID;
1631                 }
1632
1633                 se_map_standard(&printer_default->access_required, &printer_std_mapping);
1634
1635                 /* map an empty access mask to the minimum access mask */
1636                 if (printer_default->access_required == 0x0)
1637                         printer_default->access_required = PRINTER_ACCESS_USE;
1638
1639                 /*
1640                  * If we are not serving the printer driver for this printer,
1641                  * map PRINTER_ACCESS_ADMINISTER to PRINTER_ACCESS_USE.  This
1642                  * will keep NT clients happy  --jerry
1643                  */
1644
1645                 if (lp_use_client_driver(snum)
1646                         && (printer_default->access_required & PRINTER_ACCESS_ADMINISTER))
1647                 {
1648                         printer_default->access_required = PRINTER_ACCESS_USE;
1649                 }
1650
1651                 /* check smb.conf parameters and the the sec_desc */
1652
1653                 if ( !check_access(smbd_server_fd(), lp_hostsallow(snum), lp_hostsdeny(snum)) ) {
1654                         DEBUG(3, ("access DENIED (hosts allow/deny) for printer open\n"));
1655                         return WERR_ACCESS_DENIED;
1656                 }
1657
1658                 if (!user_ok_token(uidtoname(user.ut.uid), user.nt_user_token,
1659                                    snum) ||
1660                     !print_access_check(&user, snum,
1661                                         printer_default->access_required)) {
1662                         DEBUG(3, ("access DENIED for printer open\n"));
1663                         close_printer_handle(p, handle);
1664                         return WERR_ACCESS_DENIED;
1665                 }
1666
1667                 if ((printer_default->access_required & SPECIFIC_RIGHTS_MASK)& ~(PRINTER_ACCESS_ADMINISTER|PRINTER_ACCESS_USE)) {
1668                         DEBUG(3, ("access DENIED for printer open - unknown bits\n"));
1669                         close_printer_handle(p, handle);
1670                         return WERR_ACCESS_DENIED;
1671                 }
1672
1673                 if (printer_default->access_required & PRINTER_ACCESS_ADMINISTER)
1674                         printer_default->access_required = PRINTER_ACCESS_ADMINISTER;
1675                 else
1676                         printer_default->access_required = PRINTER_ACCESS_USE;
1677
1678                 DEBUG(4,("Setting printer access = %s\n", (printer_default->access_required == PRINTER_ACCESS_ADMINISTER)
1679                         ? "PRINTER_ACCESS_ADMINISTER" : "PRINTER_ACCESS_USE" ));
1680
1681                 break;
1682
1683         default:
1684                 /* sanity check to prevent programmer error */
1685                 return WERR_BADFID;
1686         }
1687
1688         Printer->access_granted = printer_default->access_required;
1689
1690         /*
1691          * If the client sent a devmode in the OpenPrinter() call, then
1692          * save it here in case we get a job submission on this handle
1693          */
1694
1695          if ( (Printer->printer_type != SPLHND_SERVER)
1696                 && q_u->printer_default.devmode_cont.devmode_ptr )
1697          {
1698                 convert_devicemode( Printer->sharename, q_u->printer_default.devmode_cont.devmode,
1699                         &Printer->nt_devmode );
1700          }
1701
1702 #if 0   /* JERRY -- I'm doubtful this is really effective */
1703         /* HACK ALERT!!! Sleep for 1/3 of a second to try trigger a LAN/WAN
1704            optimization in Windows 2000 clients  --jerry */
1705
1706         if ( (printer_default->access_required == PRINTER_ACCESS_ADMINISTER)
1707                 && (RA_WIN2K == get_remote_arch()) )
1708         {
1709                 DEBUG(10,("_spoolss_open_printer_ex: Enabling LAN/WAN hack for Win2k clients.\n"));
1710                 sys_usleep( 500000 );
1711         }
1712 #endif
1713
1714         return WERR_OK;
1715 }
1716
1717 /****************************************************************************
1718 ****************************************************************************/
1719
1720 static BOOL convert_printer_info(const SPOOL_PRINTER_INFO_LEVEL *uni,
1721                                 NT_PRINTER_INFO_LEVEL *printer, uint32 level)
1722 {
1723         BOOL ret;
1724
1725         switch (level) {
1726                 case 2:
1727                         /* allocate memory if needed.  Messy because
1728                            convert_printer_info is used to update an existing
1729                            printer or build a new one */
1730
1731                         if ( !printer->info_2 ) {
1732                                 printer->info_2 = TALLOC_ZERO_P( printer, NT_PRINTER_INFO_LEVEL_2 );
1733                                 if ( !printer->info_2 ) {
1734                                         DEBUG(0,("convert_printer_info: talloc() failed!\n"));
1735                                         return False;
1736                                 }
1737                         }
1738
1739                         ret = uni_2_asc_printer_info_2(uni->info_2, printer->info_2);
1740                         printer->info_2->setuptime = time(NULL);
1741
1742                         return ret;
1743         }
1744
1745         return False;
1746 }
1747
1748 static BOOL convert_printer_driver_info(const SPOOL_PRINTER_DRIVER_INFO_LEVEL *uni,
1749                                         NT_PRINTER_DRIVER_INFO_LEVEL *printer, uint32 level)
1750 {
1751         BOOL result = True;
1752
1753         switch (level) {
1754                 case 3:
1755                         printer->info_3=NULL;
1756                         if (!uni_2_asc_printer_driver_3(uni->info_3, &printer->info_3))
1757                                 result = False;
1758                         break;
1759                 case 6:
1760                         printer->info_6=NULL;
1761                         if (!uni_2_asc_printer_driver_6(uni->info_6, &printer->info_6))
1762                                 result = False;
1763                         break;
1764                 default:
1765                         break;
1766         }
1767
1768         return result;
1769 }
1770
1771 BOOL convert_devicemode(const char *printername, const DEVICEMODE *devmode,
1772                                 NT_DEVICEMODE **pp_nt_devmode)
1773 {
1774         NT_DEVICEMODE *nt_devmode = *pp_nt_devmode;
1775
1776         /*
1777          * Ensure nt_devmode is a valid pointer
1778          * as we will be overwriting it.
1779          */
1780
1781         if (nt_devmode == NULL) {
1782                 DEBUG(5, ("convert_devicemode: allocating a generic devmode\n"));
1783                 if ((nt_devmode = construct_nt_devicemode(printername)) == NULL)
1784                         return False;
1785         }
1786
1787         rpcstr_pull(nt_devmode->devicename,devmode->devicename.buffer, 31, -1, 0);
1788         rpcstr_pull(nt_devmode->formname,devmode->formname.buffer, 31, -1, 0);
1789
1790         nt_devmode->specversion=devmode->specversion;
1791         nt_devmode->driverversion=devmode->driverversion;
1792         nt_devmode->size=devmode->size;
1793         nt_devmode->fields=devmode->fields;
1794         nt_devmode->orientation=devmode->orientation;
1795         nt_devmode->papersize=devmode->papersize;
1796         nt_devmode->paperlength=devmode->paperlength;
1797         nt_devmode->paperwidth=devmode->paperwidth;
1798         nt_devmode->scale=devmode->scale;
1799         nt_devmode->copies=devmode->copies;
1800         nt_devmode->defaultsource=devmode->defaultsource;
1801         nt_devmode->printquality=devmode->printquality;
1802         nt_devmode->color=devmode->color;
1803         nt_devmode->duplex=devmode->duplex;
1804         nt_devmode->yresolution=devmode->yresolution;
1805         nt_devmode->ttoption=devmode->ttoption;
1806         nt_devmode->collate=devmode->collate;
1807
1808         nt_devmode->logpixels=devmode->logpixels;
1809         nt_devmode->bitsperpel=devmode->bitsperpel;
1810         nt_devmode->pelswidth=devmode->pelswidth;
1811         nt_devmode->pelsheight=devmode->pelsheight;
1812         nt_devmode->displayflags=devmode->displayflags;
1813         nt_devmode->displayfrequency=devmode->displayfrequency;
1814         nt_devmode->icmmethod=devmode->icmmethod;
1815         nt_devmode->icmintent=devmode->icmintent;
1816         nt_devmode->mediatype=devmode->mediatype;
1817         nt_devmode->dithertype=devmode->dithertype;
1818         nt_devmode->reserved1=devmode->reserved1;
1819         nt_devmode->reserved2=devmode->reserved2;
1820         nt_devmode->panningwidth=devmode->panningwidth;
1821         nt_devmode->panningheight=devmode->panningheight;
1822
1823         /*
1824          * Only change private and driverextra if the incoming devmode
1825          * has a new one. JRA.
1826          */
1827
1828         if ((devmode->driverextra != 0) && (devmode->dev_private != NULL)) {
1829                 SAFE_FREE(nt_devmode->nt_dev_private);
1830                 nt_devmode->driverextra=devmode->driverextra;
1831                 if((nt_devmode->nt_dev_private=SMB_MALLOC_ARRAY(uint8, nt_devmode->driverextra)) == NULL)
1832                         return False;
1833                 memcpy(nt_devmode->nt_dev_private, devmode->dev_private, nt_devmode->driverextra);
1834         }
1835
1836         *pp_nt_devmode = nt_devmode;
1837
1838         return True;
1839 }
1840
1841 /********************************************************************
1842  * _spoolss_enddocprinter_internal.
1843  ********************************************************************/
1844
1845 static WERROR _spoolss_enddocprinter_internal(pipes_struct *p, POLICY_HND *handle)
1846 {
1847         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
1848         int snum;
1849
1850         if (!Printer) {
1851                 DEBUG(2,("_spoolss_enddocprinter_internal: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(handle)));
1852                 return WERR_BADFID;
1853         }
1854
1855         if (!get_printer_snum(p, handle, &snum))
1856                 return WERR_BADFID;
1857
1858         Printer->document_started=False;
1859         print_job_end(snum, Printer->jobid,NORMAL_CLOSE);
1860         /* error codes unhandled so far ... */
1861
1862         return WERR_OK;
1863 }
1864
1865 /********************************************************************
1866  * api_spoolss_closeprinter
1867  ********************************************************************/
1868
1869 WERROR _spoolss_closeprinter(pipes_struct *p, SPOOL_Q_CLOSEPRINTER *q_u, SPOOL_R_CLOSEPRINTER *r_u)
1870 {
1871         POLICY_HND *handle = &q_u->handle;
1872
1873         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
1874
1875         if (Printer && Printer->document_started)
1876                 _spoolss_enddocprinter_internal(p, handle);          /* print job was not closed */
1877
1878         if (!close_printer_handle(p, handle))
1879                 return WERR_BADFID;
1880
1881         /* clear the returned printer handle.  Observed behavior
1882            from Win2k server.  Don't think this really matters.
1883            Previous code just copied the value of the closed
1884            handle.    --jerry */
1885
1886         memset(&r_u->handle, '\0', sizeof(r_u->handle));
1887
1888         return WERR_OK;
1889 }
1890
1891 /********************************************************************
1892  * api_spoolss_deleteprinter
1893
1894  ********************************************************************/
1895
1896 WERROR _spoolss_deleteprinter(pipes_struct *p, SPOOL_Q_DELETEPRINTER *q_u, SPOOL_R_DELETEPRINTER *r_u)
1897 {
1898         POLICY_HND *handle = &q_u->handle;
1899         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
1900         WERROR result;
1901
1902         if (Printer && Printer->document_started)
1903                 _spoolss_enddocprinter_internal(p, handle);  /* print job was not closed */
1904
1905         memcpy(&r_u->handle, &q_u->handle, sizeof(r_u->handle));
1906
1907         result = delete_printer_handle(p, handle);
1908
1909         update_c_setprinter(False);
1910
1911         return result;
1912 }
1913
1914 /*******************************************************************
1915  * static function to lookup the version id corresponding to an
1916  * long architecture string
1917  ******************************************************************/
1918
1919 static int get_version_id (char * arch)
1920 {
1921         int i;
1922         struct table_node archi_table[]= {
1923
1924                 {"Windows 4.0",          "WIN40",       0 },
1925                 {"Windows NT x86",       "W32X86",      2 },
1926                 {"Windows NT R4000",     "W32MIPS",     2 },
1927                 {"Windows NT Alpha_AXP", "W32ALPHA",    2 },
1928                 {"Windows NT PowerPC",   "W32PPC",      2 },
1929                 {"Windows IA64",         "IA64",        3 },
1930                 {"Windows x64",          "x64",         3 },
1931                 {NULL,                   "",            -1 }
1932         };
1933
1934         for (i=0; archi_table[i].long_archi != NULL; i++)
1935         {
1936                 if (strcmp(arch, archi_table[i].long_archi) == 0)
1937                         return (archi_table[i].version);
1938         }
1939
1940         return -1;
1941 }
1942
1943 /********************************************************************
1944  * _spoolss_deleteprinterdriver
1945  ********************************************************************/
1946
1947 WERROR _spoolss_deleteprinterdriver(pipes_struct *p, SPOOL_Q_DELETEPRINTERDRIVER *q_u, SPOOL_R_DELETEPRINTERDRIVER *r_u)
1948 {
1949         fstring                         driver;
1950         fstring                         arch;
1951         NT_PRINTER_DRIVER_INFO_LEVEL    info;
1952         NT_PRINTER_DRIVER_INFO_LEVEL    info_win2k;
1953         int                             version;
1954         struct current_user             user;
1955         WERROR                          status;
1956         WERROR                          status_win2k = WERR_ACCESS_DENIED;
1957         SE_PRIV                         se_printop = SE_PRINT_OPERATOR;
1958
1959         get_current_user(&user, p);
1960
1961         /* if the user is not root, doesn't have SE_PRINT_OPERATOR privilege,
1962            and not a printer admin, then fail */
1963
1964         if ( (user.ut.uid != 0)
1965                 && !user_has_privileges(user.nt_user_token, &se_printop )
1966                 && !token_contains_name_in_list( uidtoname(user.ut.uid),
1967                     NULL, user.nt_user_token, lp_printer_admin(-1)) )
1968         {
1969                 return WERR_ACCESS_DENIED;
1970         }
1971
1972         unistr2_to_ascii(driver, &q_u->driver, sizeof(driver)-1 );
1973         unistr2_to_ascii(arch,   &q_u->arch,   sizeof(arch)-1   );
1974
1975         /* check that we have a valid driver name first */
1976
1977         if ((version=get_version_id(arch)) == -1)
1978                 return WERR_INVALID_ENVIRONMENT;
1979
1980         ZERO_STRUCT(info);
1981         ZERO_STRUCT(info_win2k);
1982
1983         if (!W_ERROR_IS_OK(get_a_printer_driver(&info, 3, driver, arch, version)))
1984         {
1985                 /* try for Win2k driver if "Windows NT x86" */
1986
1987                 if ( version == 2 ) {
1988                         version = 3;
1989                         if (!W_ERROR_IS_OK(get_a_printer_driver(&info, 3, driver, arch, version))) {
1990                                 status = WERR_UNKNOWN_PRINTER_DRIVER;
1991                                 goto done;
1992                         }
1993                 }
1994                 /* otherwise it was a failure */
1995                 else {
1996                         status = WERR_UNKNOWN_PRINTER_DRIVER;
1997                         goto done;
1998                 }
1999
2000         }
2001
2002         if (printer_driver_in_use(info.info_3)) {
2003                 status = WERR_PRINTER_DRIVER_IN_USE;
2004                 goto done;
2005         }
2006
2007         if ( version == 2 )
2008         {
2009                 if (W_ERROR_IS_OK(get_a_printer_driver(&info_win2k, 3, driver, arch, 3)))
2010                 {
2011                         /* if we get to here, we now have 2 driver info structures to remove */
2012                         /* remove the Win2k driver first*/
2013
2014                         status_win2k = delete_printer_driver(info_win2k.info_3, &user, 3, False );
2015                         free_a_printer_driver( info_win2k, 3 );
2016
2017                         /* this should not have failed---if it did, report to client */
2018                         if ( !W_ERROR_IS_OK(status_win2k) )
2019                         {
2020                                 status = status_win2k;
2021                                 goto done;
2022                         }
2023                 }
2024         }
2025
2026         status = delete_printer_driver(info.info_3, &user, version, False);
2027
2028         /* if at least one of the deletes succeeded return OK */
2029
2030         if ( W_ERROR_IS_OK(status) || W_ERROR_IS_OK(status_win2k) )
2031                 status = WERR_OK;
2032
2033 done:
2034         free_a_printer_driver( info, 3 );
2035
2036         return status;
2037 }
2038
2039 /********************************************************************
2040  * spoolss_deleteprinterdriverex
2041  ********************************************************************/
2042
2043 WERROR _spoolss_deleteprinterdriverex(pipes_struct *p, SPOOL_Q_DELETEPRINTERDRIVEREX *q_u, SPOOL_R_DELETEPRINTERDRIVEREX *r_u)
2044 {
2045         fstring                         driver;
2046         fstring                         arch;
2047         NT_PRINTER_DRIVER_INFO_LEVEL    info;
2048         NT_PRINTER_DRIVER_INFO_LEVEL    info_win2k;
2049         int                             version;
2050         uint32                          flags = q_u->delete_flags;
2051         BOOL                            delete_files;
2052         struct current_user             user;
2053         WERROR                          status;
2054         WERROR                          status_win2k = WERR_ACCESS_DENIED;
2055         SE_PRIV                         se_printop = SE_PRINT_OPERATOR;
2056
2057         get_current_user(&user, p);
2058
2059         /* if the user is not root, doesn't have SE_PRINT_OPERATOR privilege,
2060            and not a printer admin, then fail */
2061
2062         if ( (user.ut.uid != 0)
2063                 && !user_has_privileges(user.nt_user_token, &se_printop )
2064                 && !token_contains_name_in_list( uidtoname(user.ut.uid),
2065                     NULL, user.nt_user_token, lp_printer_admin(-1)) )
2066         {
2067                 return WERR_ACCESS_DENIED;
2068         }
2069
2070         unistr2_to_ascii(driver, &q_u->driver, sizeof(driver)-1 );
2071         unistr2_to_ascii(arch,   &q_u->arch,   sizeof(arch)-1   );
2072
2073         /* check that we have a valid driver name first */
2074         if ((version=get_version_id(arch)) == -1) {
2075                 /* this is what NT returns */
2076                 return WERR_INVALID_ENVIRONMENT;
2077         }
2078
2079         if ( flags & DPD_DELETE_SPECIFIC_VERSION )
2080                 version = q_u->version;
2081
2082         ZERO_STRUCT(info);
2083         ZERO_STRUCT(info_win2k);
2084
2085         status = get_a_printer_driver(&info, 3, driver, arch, version);
2086
2087         if ( !W_ERROR_IS_OK(status) )
2088         {
2089                 /*
2090                  * if the client asked for a specific version,
2091                  * or this is something other than Windows NT x86,
2092                  * then we've failed
2093                  */
2094
2095                 if ( (flags&DPD_DELETE_SPECIFIC_VERSION) || (version !=2) )
2096                         goto done;
2097
2098                 /* try for Win2k driver if "Windows NT x86" */
2099
2100                 version = 3;
2101                 if (!W_ERROR_IS_OK(get_a_printer_driver(&info, 3, driver, arch, version))) {
2102                         status = WERR_UNKNOWN_PRINTER_DRIVER;
2103                         goto done;
2104                 }
2105         }
2106
2107         if ( printer_driver_in_use(info.info_3) ) {
2108                 status = WERR_PRINTER_DRIVER_IN_USE;
2109                 goto done;
2110         }
2111
2112         /*
2113          * we have a couple of cases to consider.
2114          * (1) Are any files in use?  If so and DPD_DELTE_ALL_FILE is set,
2115          *     then the delete should fail if **any** files overlap with
2116          *     other drivers
2117          * (2) If DPD_DELTE_UNUSED_FILES is sert, then delete all
2118          *     non-overlapping files
2119          * (3) If neither DPD_DELTE_ALL_FILE nor DPD_DELTE_ALL_FILES
2120          *     is set, the do not delete any files
2121          * Refer to MSDN docs on DeletePrinterDriverEx() for details.
2122          */
2123
2124         delete_files = flags & (DPD_DELETE_ALL_FILES|DPD_DELETE_UNUSED_FILES);
2125
2126         /* fail if any files are in use and DPD_DELETE_ALL_FILES is set */
2127
2128         if ( delete_files && printer_driver_files_in_use(info.info_3) & (flags&DPD_DELETE_ALL_FILES) ) {
2129                 /* no idea of the correct error here */
2130                 status = WERR_ACCESS_DENIED;
2131                 goto done;
2132         }
2133
2134
2135         /* also check for W32X86/3 if necessary; maybe we already have? */
2136
2137         if ( (version == 2) && ((flags&DPD_DELETE_SPECIFIC_VERSION) != DPD_DELETE_SPECIFIC_VERSION)  ) {
2138                 if (W_ERROR_IS_OK(get_a_printer_driver(&info_win2k, 3, driver, arch, 3)))
2139                 {
2140
2141                         if ( delete_files && printer_driver_files_in_use(info_win2k.info_3) & (flags&DPD_DELETE_ALL_FILES) ) {
2142                                 /* no idea of the correct error here */
2143                                 free_a_printer_driver( info_win2k, 3 );
2144                                 status = WERR_ACCESS_DENIED;
2145                                 goto done;
2146                         }
2147
2148                         /* if we get to here, we now have 2 driver info structures to remove */
2149                         /* remove the Win2k driver first*/
2150
2151                         status_win2k = delete_printer_driver(info_win2k.info_3, &user, 3, delete_files);
2152                         free_a_printer_driver( info_win2k, 3 );
2153
2154                         /* this should not have failed---if it did, report to client */
2155
2156                         if ( !W_ERROR_IS_OK(status_win2k) )
2157                                 goto done;
2158                 }
2159         }
2160
2161         status = delete_printer_driver(info.info_3, &user, version, delete_files);
2162
2163         if ( W_ERROR_IS_OK(status) || W_ERROR_IS_OK(status_win2k) )
2164                 status = WERR_OK;
2165 done:
2166         free_a_printer_driver( info, 3 );
2167
2168         return status;
2169 }
2170
2171
2172 /****************************************************************************
2173  Internal routine for retreiving printerdata
2174  ***************************************************************************/
2175
2176 static WERROR get_printer_dataex( TALLOC_CTX *ctx, NT_PRINTER_INFO_LEVEL *printer,
2177                                   const char *key, const char *value, uint32 *type, uint8 **data,
2178                                   uint32 *needed, uint32 in_size  )
2179 {
2180         REGISTRY_VALUE          *val;
2181         uint32                  size;
2182         int                     data_len;
2183
2184         if ( !(val = get_printer_data( printer->info_2, key, value)) )
2185                 return WERR_BADFILE;
2186
2187         *type = regval_type( val );
2188
2189         DEBUG(5,("get_printer_dataex: allocating %d\n", in_size));
2190
2191         size = regval_size( val );
2192
2193         /* copy the min(in_size, len) */
2194
2195         if ( in_size ) {
2196                 data_len = (size > in_size) ? in_size : size*sizeof(uint8);
2197
2198                 /* special case for 0 length values */
2199                 if ( data_len ) {
2200                         if ( (*data  = (uint8 *)TALLOC_MEMDUP(ctx, regval_data_p(val), data_len)) == NULL )
2201                                 return WERR_NOMEM;
2202                 }
2203                 else {
2204                         if ( (*data  = (uint8 *)TALLOC_ZERO(ctx, in_size)) == NULL )
2205                                 return WERR_NOMEM;
2206                 }
2207         }
2208         else
2209                 *data = NULL;
2210
2211         *needed = size;
2212
2213         DEBUG(5,("get_printer_dataex: copy done\n"));
2214
2215         return WERR_OK;
2216 }
2217
2218 /****************************************************************************
2219  Internal routine for removing printerdata
2220  ***************************************************************************/
2221
2222 static WERROR delete_printer_dataex( NT_PRINTER_INFO_LEVEL *printer, const char *key, const char *value )
2223 {
2224         return delete_printer_data( printer->info_2, key, value );
2225 }
2226
2227 /****************************************************************************
2228  Internal routine for storing printerdata
2229  ***************************************************************************/
2230
2231 WERROR set_printer_dataex( NT_PRINTER_INFO_LEVEL *printer, const char *key, const char *value,
2232                                   uint32 type, uint8 *data, int real_len  )
2233 {
2234         /* the registry objects enforce uniqueness based on value name */
2235
2236         return add_printer_data( printer->info_2, key, value, type, data, real_len );
2237 }
2238
2239 /********************************************************************
2240  GetPrinterData on a printer server Handle.
2241 ********************************************************************/
2242
2243 static WERROR getprinterdata_printer_server(TALLOC_CTX *ctx, fstring value, uint32 *type, uint8 **data, uint32 *needed, uint32 in_size)
2244 {
2245         int i;
2246
2247         DEBUG(8,("getprinterdata_printer_server:%s\n", value));
2248
2249         if (!StrCaseCmp(value, "W3SvcInstalled")) {
2250                 *type = REG_DWORD;
2251                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2252                         return WERR_NOMEM;
2253                 *needed = 0x4;
2254                 return WERR_OK;
2255         }
2256
2257         if (!StrCaseCmp(value, "BeepEnabled")) {
2258                 *type = REG_DWORD;
2259                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2260                         return WERR_NOMEM;
2261                 SIVAL(*data, 0, 0x00);
2262                 *needed = 0x4;
2263                 return WERR_OK;
2264         }
2265
2266         if (!StrCaseCmp(value, "EventLog")) {
2267                 *type = REG_DWORD;
2268                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2269                         return WERR_NOMEM;
2270                 /* formally was 0x1b */
2271                 SIVAL(*data, 0, 0x0);
2272                 *needed = 0x4;
2273                 return WERR_OK;
2274         }
2275
2276         if (!StrCaseCmp(value, "NetPopup")) {
2277                 *type = REG_DWORD;
2278                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2279                         return WERR_NOMEM;
2280                 SIVAL(*data, 0, 0x00);
2281                 *needed = 0x4;
2282                 return WERR_OK;
2283         }
2284
2285         if (!StrCaseCmp(value, "MajorVersion")) {
2286                 *type = REG_DWORD;
2287                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2288                         return WERR_NOMEM;
2289
2290                 /* Windows NT 4.0 seems to not allow uploading of drivers
2291                    to a server that reports 0x3 as the MajorVersion.
2292                    need to investigate more how Win2k gets around this .
2293                    -- jerry */
2294
2295                 if ( RA_WINNT == get_remote_arch() )
2296                         SIVAL(*data, 0, 2);
2297                 else
2298                         SIVAL(*data, 0, 3);
2299
2300                 *needed = 0x4;
2301                 return WERR_OK;
2302         }
2303
2304         if (!StrCaseCmp(value, "MinorVersion")) {
2305                 *type = REG_DWORD;
2306                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2307                         return WERR_NOMEM;
2308                 SIVAL(*data, 0, 0);
2309                 *needed = 0x4;
2310                 return WERR_OK;
2311         }
2312
2313         /* REG_BINARY
2314          *  uint32 size          = 0x114
2315          *  uint32 major         = 5
2316          *  uint32 minor         = [0|1]
2317          *  uint32 build         = [2195|2600]
2318          *  extra unicode string = e.g. "Service Pack 3"
2319          */
2320         if (!StrCaseCmp(value, "OSVersion")) {
2321                 *type = REG_BINARY;
2322                 *needed = 0x114;
2323
2324                 if ( !(*data = TALLOC_ZERO_ARRAY(ctx, uint8, (*needed > in_size) ? *needed:in_size )) )
2325                         return WERR_NOMEM;
2326
2327                 SIVAL(*data, 0, *needed);       /* size */
2328                 SIVAL(*data, 4, 5);             /* Windows 2000 == 5.0 */
2329                 SIVAL(*data, 8, 0);
2330                 SIVAL(*data, 12, 2195);         /* build */
2331
2332                 /* leave extra string empty */
2333
2334                 return WERR_OK;
2335         }
2336
2337
2338         if (!StrCaseCmp(value, "DefaultSpoolDirectory")) {
2339                 const char *string="C:\\PRINTERS";
2340                 *type = REG_SZ;
2341                 *needed = 2*(strlen(string)+1);
2342                 if((*data  = (uint8 *)TALLOC(ctx, (*needed > in_size) ? *needed:in_size )) == NULL)
2343                         return WERR_NOMEM;
2344                 memset(*data, 0, (*needed > in_size) ? *needed:in_size);
2345
2346                 /* it's done by hand ready to go on the wire */
2347                 for (i=0; i<strlen(string); i++) {
2348                         (*data)[2*i]=string[i];
2349                         (*data)[2*i+1]='\0';
2350                 }
2351                 return WERR_OK;
2352         }
2353
2354         if (!StrCaseCmp(value, "Architecture")) {
2355                 const char *string="Windows NT x86";
2356                 *type = REG_SZ;
2357                 *needed = 2*(strlen(string)+1);
2358                 if((*data  = (uint8 *)TALLOC(ctx, (*needed > in_size) ? *needed:in_size )) == NULL)
2359                         return WERR_NOMEM;
2360                 memset(*data, 0, (*needed > in_size) ? *needed:in_size);
2361                 for (i=0; i<strlen(string); i++) {
2362                         (*data)[2*i]=string[i];
2363                         (*data)[2*i+1]='\0';
2364                 }
2365                 return WERR_OK;
2366         }
2367
2368         if (!StrCaseCmp(value, "DsPresent")) {
2369                 *type = REG_DWORD;
2370                 if ( !(*data = TALLOC_ARRAY(ctx, uint8, sizeof(uint32) )) )
2371                         return WERR_NOMEM;
2372
2373                 /* only show the publish check box if we are a
2374                    memeber of a AD domain */
2375
2376                 if ( lp_security() == SEC_ADS )
2377                         SIVAL(*data, 0, 0x01);
2378                 else
2379                         SIVAL(*data, 0, 0x00);
2380
2381                 *needed = 0x4;
2382                 return WERR_OK;
2383         }
2384
2385         if (!StrCaseCmp(value, "DNSMachineName")) {
2386                 pstring hostname;
2387
2388                 if (!get_mydnsfullname(hostname))
2389                         return WERR_BADFILE;
2390                 *type = REG_SZ;
2391                 *needed = 2*(strlen(hostname)+1);
2392                 if((*data  = (uint8 *)TALLOC(ctx, (*needed > in_size) ? *needed:in_size )) == NULL)
2393                         return WERR_NOMEM;
2394                 memset(*data, 0, (*needed > in_size) ? *needed:in_size);
2395                 for (i=0; i<strlen(hostname); i++) {
2396                         (*data)[2*i]=hostname[i];
2397                         (*data)[2*i+1]='\0';
2398                 }
2399                 return WERR_OK;
2400         }
2401
2402
2403         return WERR_BADFILE;
2404 }
2405
2406 /********************************************************************
2407  * spoolss_getprinterdata
2408  ********************************************************************/
2409
2410 WERROR _spoolss_getprinterdata(pipes_struct *p, SPOOL_Q_GETPRINTERDATA *q_u, SPOOL_R_GETPRINTERDATA *r_u)
2411 {
2412         POLICY_HND      *handle = &q_u->handle;
2413         UNISTR2         *valuename = &q_u->valuename;
2414         uint32          in_size = q_u->size;
2415         uint32          *type = &r_u->type;
2416         uint32          *out_size = &r_u->size;
2417         uint8           **data = &r_u->data;
2418         uint32          *needed = &r_u->needed;
2419         WERROR          status;
2420         fstring         value;
2421         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
2422         NT_PRINTER_INFO_LEVEL   *printer = NULL;
2423         int             snum = 0;
2424
2425         /*
2426          * Reminder: when it's a string, the length is in BYTES
2427          * even if UNICODE is negociated.
2428          *
2429          * JFM, 4/19/1999
2430          */
2431
2432         *out_size = in_size;
2433
2434         /* in case of problem, return some default values */
2435
2436         *needed = 0;
2437         *type   = 0;
2438
2439         DEBUG(4,("_spoolss_getprinterdata\n"));
2440
2441         if ( !Printer ) {
2442                 DEBUG(2,("_spoolss_getprinterdata: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
2443                 status = WERR_BADFID;
2444                 goto done;
2445         }
2446
2447         unistr2_to_ascii(value, valuename, sizeof(value)-1);
2448
2449         if ( Printer->printer_type == SPLHND_SERVER )
2450                 status = getprinterdata_printer_server( p->mem_ctx, value, type, data, needed, *out_size );
2451         else
2452         {
2453                 if ( !get_printer_snum(p,handle, &snum) ) {
2454                         status = WERR_BADFID;
2455                         goto done;
2456                 }
2457
2458                 status = get_a_printer(Printer, &printer, 2, lp_servicename(snum));
2459                 if ( !W_ERROR_IS_OK(status) )
2460                         goto done;
2461
2462                 /* XP sends this and wants to change id value from the PRINTER_INFO_0 */
2463
2464                 if ( strequal(value, "ChangeId") ) {
2465                         *type = REG_DWORD;
2466                         *needed = sizeof(uint32);
2467                         if ( (*data = (uint8*)TALLOC(p->mem_ctx, sizeof(uint32))) == NULL) {
2468                                 status = WERR_NOMEM;
2469                                 goto done;
2470                         }
2471                         SIVAL( *data, 0, printer->info_2->changeid );
2472                         status = WERR_OK;
2473                 }
2474                 else
2475                         status = get_printer_dataex( p->mem_ctx, printer, SPOOL_PRINTERDATA_KEY, value, type, data, needed, *out_size );
2476         }
2477
2478         if (*needed > *out_size)
2479                 status = WERR_MORE_DATA;
2480
2481 done:
2482         if ( !W_ERROR_IS_OK(status) )
2483         {
2484                 DEBUG(5, ("error %d: allocating %d\n", W_ERROR_V(status),*out_size));
2485
2486                 /* reply this param doesn't exist */
2487
2488                 if ( *out_size ) {
2489                         if((*data=(uint8 *)TALLOC_ZERO_ARRAY(p->mem_ctx, uint8, *out_size)) == NULL) {
2490                                 if ( printer )
2491                                         free_a_printer( &printer, 2 );
2492                                 return WERR_NOMEM;
2493                 }
2494                 }
2495                 else {
2496                         *data = NULL;
2497                 }
2498         }
2499
2500         /* cleanup & exit */
2501
2502         if ( printer )
2503                 free_a_printer( &printer, 2 );
2504
2505         return status;
2506 }
2507
2508 /*********************************************************
2509  Connect to the client machine.
2510 **********************************************************/
2511
2512 static BOOL spoolss_connect_to_client(struct rpc_pipe_client **pp_pipe,
2513                         struct in_addr *client_ip, const char *remote_machine)
2514 {
2515         NTSTATUS ret;
2516         struct cli_state *the_cli;
2517         struct in_addr rm_addr;
2518
2519         if ( is_zero_ip(*client_ip) ) {
2520                 if ( !resolve_name( remote_machine, &rm_addr, 0x20) ) {
2521                         DEBUG(2,("spoolss_connect_to_client: Can't resolve address for %s\n", remote_machine));
2522                         return False;
2523                 }
2524
2525                 if ( ismyip( rm_addr )) {
2526                         DEBUG(0,("spoolss_connect_to_client: Machine %s is one of our addresses. Cannot add to ourselves.\n", remote_machine));
2527                         return False;
2528                 }
2529         } else {
2530                 rm_addr.s_addr = client_ip->s_addr;
2531                 DEBUG(5,("spoolss_connect_to_client: Using address %s (no name resolution necessary)\n",
2532                         inet_ntoa(*client_ip) ));
2533         }
2534
2535         /* setup the connection */
2536
2537         ret = cli_full_connection( &the_cli, global_myname(), remote_machine,
2538                 &rm_addr, 0, "IPC$", "IPC",
2539                 "", /* username */
2540                 "", /* domain */
2541                 "", /* password */
2542                 0, lp_client_signing(), NULL );
2543
2544         if ( !NT_STATUS_IS_OK( ret ) ) {
2545                 DEBUG(2,("spoolss_connect_to_client: connection to [%s] failed!\n",
2546                         remote_machine ));
2547                 return False;
2548         }
2549
2550         if ( the_cli->protocol != PROTOCOL_NT1 ) {
2551                 DEBUG(0,("spoolss_connect_to_client: machine %s didn't negotiate NT protocol.\n", remote_machine));
2552                 cli_shutdown(the_cli);
2553                 return False;
2554         }
2555
2556         /*
2557          * Ok - we have an anonymous connection to the IPC$ share.
2558          * Now start the NT Domain stuff :-).
2559          */
2560
2561         if ( !(*pp_pipe = cli_rpc_pipe_open_noauth(the_cli, PI_SPOOLSS, &ret)) ) {
2562                 DEBUG(2,("spoolss_connect_to_client: unable to open the spoolss pipe on machine %s. Error was : %s.\n",
2563                         remote_machine, nt_errstr(ret)));
2564                 cli_shutdown(the_cli);
2565                 return False;
2566         }
2567
2568         /* make sure to save the cli_state pointer.  Keep its own talloc_ctx */
2569
2570         (*pp_pipe)->cli = the_cli;
2571
2572         return True;
2573 }
2574
2575 /***************************************************************************
2576  Connect to the client.
2577 ****************************************************************************/
2578
2579 static BOOL srv_spoolss_replyopenprinter(int snum, const char *printer,
2580                                         uint32 localprinter, uint32 type,
2581                                         POLICY_HND *handle, struct in_addr *client_ip)
2582 {
2583         WERROR result;
2584
2585         /*
2586          * If it's the first connection, contact the client
2587          * and connect to the IPC$ share anonymously
2588          */
2589         if (smb_connections==0) {
2590                 fstring unix_printer;
2591
2592                 fstrcpy(unix_printer, printer+2); /* the +2 is to strip the leading 2 backslashs */
2593
2594                 if ( !spoolss_connect_to_client( &notify_cli_pipe, client_ip, unix_printer ))
2595                         return False;
2596
2597                 message_register(MSG_PRINTER_NOTIFY2, receive_notify2_message_list);
2598                 /* Tell the connections db we're now interested in printer
2599                  * notify messages. */
2600                 register_message_flags( True, FLAG_MSG_PRINT_NOTIFY );
2601         }
2602
2603         /*
2604          * Tell the specific printing tdb we want messages for this printer
2605          * by registering our PID.
2606          */
2607
2608         if (!print_notify_register_pid(snum))
2609                 DEBUG(0,("print_notify_register_pid: Failed to register our pid for printer %s\n", printer ));
2610
2611         smb_connections++;
2612
2613         result = rpccli_spoolss_reply_open_printer(notify_cli_pipe, notify_cli_pipe->cli->mem_ctx, printer, localprinter,
2614                         type, handle);
2615
2616         if (!W_ERROR_IS_OK(result))
2617                 DEBUG(5,("srv_spoolss_reply_open_printer: Client RPC returned [%s]\n",
2618                         dos_errstr(result)));
2619
2620         return (W_ERROR_IS_OK(result));
2621 }
2622
2623 /********************************************************************
2624  * _spoolss_rffpcnex
2625  * ReplyFindFirstPrinterChangeNotifyEx
2626  *
2627  * before replying OK: status=0 a rpc call is made to the workstation
2628  * asking ReplyOpenPrinter
2629  *
2630  * in fact ReplyOpenPrinter is the changenotify equivalent on the spoolss pipe
2631  * called from api_spoolss_rffpcnex
2632  ********************************************************************/
2633
2634 WERROR _spoolss_rffpcnex(pipes_struct *p, SPOOL_Q_RFFPCNEX *q_u, SPOOL_R_RFFPCNEX *r_u)
2635 {
2636         POLICY_HND *handle = &q_u->handle;
2637         uint32 flags = q_u->flags;
2638         uint32 options = q_u->options;
2639         UNISTR2 *localmachine = &q_u->localmachine;
2640         uint32 printerlocal = q_u->printerlocal;
2641         int snum = -1;
2642         SPOOL_NOTIFY_OPTION *option = q_u->option;
2643         struct in_addr client_ip;
2644
2645         /* store the notify value in the printer struct */
2646
2647         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
2648
2649         if (!Printer) {
2650                 DEBUG(2,("_spoolss_rffpcnex: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
2651                 return WERR_BADFID;
2652         }
2653
2654         Printer->notify.flags=flags;
2655         Printer->notify.options=options;
2656         Printer->notify.printerlocal=printerlocal;
2657
2658         if (Printer->notify.option)
2659                 free_spool_notify_option(&Printer->notify.option);
2660
2661         Printer->notify.option=dup_spool_notify_option(option);
2662
2663         unistr2_to_ascii(Printer->notify.localmachine, localmachine,
2664                        sizeof(Printer->notify.localmachine)-1);
2665
2666         /* Connect to the client machine and send a ReplyOpenPrinter */
2667
2668         if ( Printer->printer_type == SPLHND_SERVER)
2669                 snum = -1;
2670         else if ( (Printer->printer_type == SPLHND_PRINTER) &&
2671                         !get_printer_snum(p, handle, &snum) )
2672                 return WERR_BADFID;
2673
2674         client_ip.s_addr = inet_addr(p->conn->client_address);
2675
2676         if(!srv_spoolss_replyopenprinter(snum, Printer->notify.localmachine,
2677                                         Printer->notify.printerlocal, 1,
2678                                         &Printer->notify.client_hnd, &client_ip))
2679                 return WERR_SERVER_UNAVAILABLE;
2680
2681         Printer->notify.client_connected=True;
2682
2683         return WERR_OK;
2684 }
2685
2686 /*******************************************************************
2687  * fill a notify_info_data with the servername
2688  ********************************************************************/
2689
2690 void spoolss_notify_server_name(int snum,
2691                                        SPOOL_NOTIFY_INFO_DATA *data,
2692                                        print_queue_struct *queue,
2693                                        NT_PRINTER_INFO_LEVEL *printer,
2694                                        TALLOC_CTX *mem_ctx)
2695 {
2696         pstring temp;
2697         uint32 len;
2698
2699         len = rpcstr_push(temp, printer->info_2->servername, sizeof(temp)-2, STR_TERMINATE);
2700
2701         data->notify_data.data.length = len;
2702         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2703
2704         if (!data->notify_data.data.string) {
2705                 data->notify_data.data.length = 0;
2706                 return;
2707         }
2708
2709         memcpy(data->notify_data.data.string, temp, len);
2710 }
2711
2712 /*******************************************************************
2713  * fill a notify_info_data with the printername (not including the servername).
2714  ********************************************************************/
2715
2716 void spoolss_notify_printer_name(int snum,
2717                                         SPOOL_NOTIFY_INFO_DATA *data,
2718                                         print_queue_struct *queue,
2719                                         NT_PRINTER_INFO_LEVEL *printer,
2720                                         TALLOC_CTX *mem_ctx)
2721 {
2722         pstring temp;
2723         uint32 len;
2724
2725         /* the notify name should not contain the \\server\ part */
2726         char *p = strrchr(printer->info_2->printername, '\\');
2727
2728         if (!p) {
2729                 p = printer->info_2->printername;
2730         } else {
2731                 p++;
2732         }
2733
2734         len = rpcstr_push(temp, p, sizeof(temp)-2, STR_TERMINATE);
2735
2736         data->notify_data.data.length = len;
2737         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2738
2739         if (!data->notify_data.data.string) {
2740                 data->notify_data.data.length = 0;
2741                 return;
2742         }
2743
2744         memcpy(data->notify_data.data.string, temp, len);
2745 }
2746
2747 /*******************************************************************
2748  * fill a notify_info_data with the servicename
2749  ********************************************************************/
2750
2751 void spoolss_notify_share_name(int snum,
2752                                       SPOOL_NOTIFY_INFO_DATA *data,
2753                                       print_queue_struct *queue,
2754                                       NT_PRINTER_INFO_LEVEL *printer,
2755                                       TALLOC_CTX *mem_ctx)
2756 {
2757         pstring temp;
2758         uint32 len;
2759
2760         len = rpcstr_push(temp, lp_servicename(snum), sizeof(temp)-2, STR_TERMINATE);
2761
2762         data->notify_data.data.length = len;
2763         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2764
2765         if (!data->notify_data.data.string) {
2766                 data->notify_data.data.length = 0;
2767                 return;
2768         }
2769
2770         memcpy(data->notify_data.data.string, temp, len);
2771 }
2772
2773 /*******************************************************************
2774  * fill a notify_info_data with the port name
2775  ********************************************************************/
2776
2777 void spoolss_notify_port_name(int snum,
2778                                      SPOOL_NOTIFY_INFO_DATA *data,
2779                                      print_queue_struct *queue,
2780                                      NT_PRINTER_INFO_LEVEL *printer,
2781                                      TALLOC_CTX *mem_ctx)
2782 {
2783         pstring temp;
2784         uint32 len;
2785
2786         /* even if it's strange, that's consistant in all the code */
2787
2788         len = rpcstr_push(temp, printer->info_2->portname, sizeof(temp)-2, STR_TERMINATE);
2789
2790         data->notify_data.data.length = len;
2791         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2792
2793         if (!data->notify_data.data.string) {
2794                 data->notify_data.data.length = 0;
2795                 return;
2796         }
2797
2798         memcpy(data->notify_data.data.string, temp, len);
2799 }
2800
2801 /*******************************************************************
2802  * fill a notify_info_data with the printername
2803  * but it doesn't exist, have to see what to do
2804  ********************************************************************/
2805
2806 void spoolss_notify_driver_name(int snum,
2807                                        SPOOL_NOTIFY_INFO_DATA *data,
2808                                        print_queue_struct *queue,
2809                                        NT_PRINTER_INFO_LEVEL *printer,
2810                                        TALLOC_CTX *mem_ctx)
2811 {
2812         pstring temp;
2813         uint32 len;
2814
2815         len = rpcstr_push(temp, printer->info_2->drivername, sizeof(temp)-2, STR_TERMINATE);
2816
2817         data->notify_data.data.length = len;
2818         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2819
2820         if (!data->notify_data.data.string) {
2821                 data->notify_data.data.length = 0;
2822                 return;
2823         }
2824
2825         memcpy(data->notify_data.data.string, temp, len);
2826 }
2827
2828 /*******************************************************************
2829  * fill a notify_info_data with the comment
2830  ********************************************************************/
2831
2832 void spoolss_notify_comment(int snum,
2833                                    SPOOL_NOTIFY_INFO_DATA *data,
2834                                    print_queue_struct *queue,
2835                                    NT_PRINTER_INFO_LEVEL *printer,
2836                                    TALLOC_CTX *mem_ctx)
2837 {
2838         pstring temp;
2839         uint32 len;
2840
2841         if (*printer->info_2->comment == '\0')
2842                 len = rpcstr_push(temp, lp_comment(snum), sizeof(temp)-2, STR_TERMINATE);
2843         else
2844                 len = rpcstr_push(temp, printer->info_2->comment, sizeof(temp)-2, STR_TERMINATE);
2845
2846         data->notify_data.data.length = len;
2847         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2848
2849         if (!data->notify_data.data.string) {
2850                 data->notify_data.data.length = 0;
2851                 return;
2852         }
2853
2854         memcpy(data->notify_data.data.string, temp, len);
2855 }
2856
2857 /*******************************************************************
2858  * fill a notify_info_data with the comment
2859  * location = "Room 1, floor 2, building 3"
2860  ********************************************************************/
2861
2862 void spoolss_notify_location(int snum,
2863                                     SPOOL_NOTIFY_INFO_DATA *data,
2864                                     print_queue_struct *queue,
2865                                     NT_PRINTER_INFO_LEVEL *printer,
2866                                     TALLOC_CTX *mem_ctx)
2867 {
2868         pstring temp;
2869         uint32 len;
2870
2871         len = rpcstr_push(temp, printer->info_2->location,sizeof(temp)-2, STR_TERMINATE);
2872
2873         data->notify_data.data.length = len;
2874         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2875
2876         if (!data->notify_data.data.string) {
2877                 data->notify_data.data.length = 0;
2878                 return;
2879         }
2880
2881         memcpy(data->notify_data.data.string, temp, len);
2882 }
2883
2884 /*******************************************************************
2885  * fill a notify_info_data with the device mode
2886  * jfm:xxxx don't to it for know but that's a real problem !!!
2887  ********************************************************************/
2888
2889 static void spoolss_notify_devmode(int snum,
2890                                    SPOOL_NOTIFY_INFO_DATA *data,
2891                                    print_queue_struct *queue,
2892                                    NT_PRINTER_INFO_LEVEL *printer,
2893                                    TALLOC_CTX *mem_ctx)
2894 {
2895 }
2896
2897 /*******************************************************************
2898  * fill a notify_info_data with the separator file name
2899  ********************************************************************/
2900
2901 void spoolss_notify_sepfile(int snum,
2902                                    SPOOL_NOTIFY_INFO_DATA *data,
2903                                    print_queue_struct *queue,
2904                                    NT_PRINTER_INFO_LEVEL *printer,
2905                                    TALLOC_CTX *mem_ctx)
2906 {
2907         pstring temp;
2908         uint32 len;
2909
2910         len = rpcstr_push(temp, printer->info_2->sepfile, sizeof(temp)-2, STR_TERMINATE);
2911
2912         data->notify_data.data.length = len;
2913         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2914
2915         if (!data->notify_data.data.string) {
2916                 data->notify_data.data.length = 0;
2917                 return;
2918         }
2919
2920         memcpy(data->notify_data.data.string, temp, len);
2921 }
2922
2923 /*******************************************************************
2924  * fill a notify_info_data with the print processor
2925  * jfm:xxxx return always winprint to indicate we don't do anything to it
2926  ********************************************************************/
2927
2928 void spoolss_notify_print_processor(int snum,
2929                                            SPOOL_NOTIFY_INFO_DATA *data,
2930                                            print_queue_struct *queue,
2931                                            NT_PRINTER_INFO_LEVEL *printer,
2932                                            TALLOC_CTX *mem_ctx)
2933 {
2934         pstring temp;
2935         uint32 len;
2936
2937         len = rpcstr_push(temp,  printer->info_2->printprocessor, sizeof(temp)-2, STR_TERMINATE);
2938
2939         data->notify_data.data.length = len;
2940         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2941
2942         if (!data->notify_data.data.string) {
2943                 data->notify_data.data.length = 0;
2944                 return;
2945         }
2946
2947         memcpy(data->notify_data.data.string, temp, len);
2948 }
2949
2950 /*******************************************************************
2951  * fill a notify_info_data with the print processor options
2952  * jfm:xxxx send an empty string
2953  ********************************************************************/
2954
2955 void spoolss_notify_parameters(int snum,
2956                                       SPOOL_NOTIFY_INFO_DATA *data,
2957                                       print_queue_struct *queue,
2958                                       NT_PRINTER_INFO_LEVEL *printer,
2959                                       TALLOC_CTX *mem_ctx)
2960 {
2961         pstring temp;
2962         uint32 len;
2963
2964         len = rpcstr_push(temp,  printer->info_2->parameters, sizeof(temp)-2, STR_TERMINATE);
2965
2966         data->notify_data.data.length = len;
2967         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2968
2969         if (!data->notify_data.data.string) {
2970                 data->notify_data.data.length = 0;
2971                 return;
2972         }
2973
2974         memcpy(data->notify_data.data.string, temp, len);
2975 }
2976
2977 /*******************************************************************
2978  * fill a notify_info_data with the data type
2979  * jfm:xxxx always send RAW as data type
2980  ********************************************************************/
2981
2982 void spoolss_notify_datatype(int snum,
2983                                     SPOOL_NOTIFY_INFO_DATA *data,
2984                                     print_queue_struct *queue,
2985                                     NT_PRINTER_INFO_LEVEL *printer,
2986                                     TALLOC_CTX *mem_ctx)
2987 {
2988         pstring temp;
2989         uint32 len;
2990
2991         len = rpcstr_push(temp, printer->info_2->datatype, sizeof(pstring)-2, STR_TERMINATE);
2992
2993         data->notify_data.data.length = len;
2994         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
2995
2996         if (!data->notify_data.data.string) {
2997                 data->notify_data.data.length = 0;
2998                 return;
2999         }
3000
3001         memcpy(data->notify_data.data.string, temp, len);
3002 }
3003
3004 /*******************************************************************
3005  * fill a notify_info_data with the security descriptor
3006  * jfm:xxxx send an null pointer to say no security desc
3007  * have to implement security before !
3008  ********************************************************************/
3009
3010 static void spoolss_notify_security_desc(int snum,
3011                                          SPOOL_NOTIFY_INFO_DATA *data,
3012                                          print_queue_struct *queue,
3013                                          NT_PRINTER_INFO_LEVEL *printer,
3014                                          TALLOC_CTX *mem_ctx)
3015 {
3016         data->notify_data.sd.size = printer->info_2->secdesc_buf->len;
3017         data->notify_data.sd.desc = dup_sec_desc( mem_ctx, printer->info_2->secdesc_buf->sec ) ;
3018 }
3019
3020 /*******************************************************************
3021  * fill a notify_info_data with the attributes
3022  * jfm:xxxx a samba printer is always shared
3023  ********************************************************************/
3024
3025 void spoolss_notify_attributes(int snum,
3026                                       SPOOL_NOTIFY_INFO_DATA *data,
3027                                       print_queue_struct *queue,
3028                                       NT_PRINTER_INFO_LEVEL *printer,
3029                                       TALLOC_CTX *mem_ctx)
3030 {
3031         data->notify_data.value[0] = printer->info_2->attributes;
3032         data->notify_data.value[1] = 0;
3033 }
3034
3035 /*******************************************************************
3036  * fill a notify_info_data with the priority
3037  ********************************************************************/
3038
3039 static void spoolss_notify_priority(int snum,
3040                                     SPOOL_NOTIFY_INFO_DATA *data,
3041                                     print_queue_struct *queue,
3042                                     NT_PRINTER_INFO_LEVEL *printer,
3043                                     TALLOC_CTX *mem_ctx)
3044 {
3045         data->notify_data.value[0] = printer->info_2->priority;
3046         data->notify_data.value[1] = 0;
3047 }
3048
3049 /*******************************************************************
3050  * fill a notify_info_data with the default priority
3051  ********************************************************************/
3052
3053 static void spoolss_notify_default_priority(int snum,
3054                                             SPOOL_NOTIFY_INFO_DATA *data,
3055                                             print_queue_struct *queue,
3056                                             NT_PRINTER_INFO_LEVEL *printer,
3057                                             TALLOC_CTX *mem_ctx)
3058 {
3059         data->notify_data.value[0] = printer->info_2->default_priority;
3060         data->notify_data.value[1] = 0;
3061 }
3062
3063 /*******************************************************************
3064  * fill a notify_info_data with the start time
3065  ********************************************************************/
3066
3067 static void spoolss_notify_start_time(int snum,
3068                                       SPOOL_NOTIFY_INFO_DATA *data,
3069                                       print_queue_struct *queue,
3070                                       NT_PRINTER_INFO_LEVEL *printer,
3071                                       TALLOC_CTX *mem_ctx)
3072 {
3073         data->notify_data.value[0] = printer->info_2->starttime;
3074         data->notify_data.value[1] = 0;
3075 }
3076
3077 /*******************************************************************
3078  * fill a notify_info_data with the until time
3079  ********************************************************************/
3080
3081 static void spoolss_notify_until_time(int snum,
3082                                       SPOOL_NOTIFY_INFO_DATA *data,
3083                                       print_queue_struct *queue,
3084                                       NT_PRINTER_INFO_LEVEL *printer,
3085                                       TALLOC_CTX *mem_ctx)
3086 {
3087         data->notify_data.value[0] = printer->info_2->untiltime;
3088         data->notify_data.value[1] = 0;
3089 }
3090
3091 /*******************************************************************
3092  * fill a notify_info_data with the status
3093  ********************************************************************/
3094
3095 static void spoolss_notify_status(int snum,
3096                                   SPOOL_NOTIFY_INFO_DATA *data,
3097                                   print_queue_struct *queue,
3098                                   NT_PRINTER_INFO_LEVEL *printer,
3099                                   TALLOC_CTX *mem_ctx)
3100 {
3101         print_status_struct status;
3102
3103         print_queue_length(snum, &status);
3104         data->notify_data.value[0]=(uint32) status.status;
3105         data->notify_data.value[1] = 0;
3106 }
3107
3108 /*******************************************************************
3109  * fill a notify_info_data with the number of jobs queued
3110  ********************************************************************/
3111
3112 void spoolss_notify_cjobs(int snum,
3113                                  SPOOL_NOTIFY_INFO_DATA *data,
3114                                  print_queue_struct *queue,
3115                                  NT_PRINTER_INFO_LEVEL *printer,
3116                                  TALLOC_CTX *mem_ctx)
3117 {
3118         data->notify_data.value[0] = print_queue_length(snum, NULL);
3119         data->notify_data.value[1] = 0;
3120 }
3121
3122 /*******************************************************************
3123  * fill a notify_info_data with the average ppm
3124  ********************************************************************/
3125
3126 static void spoolss_notify_average_ppm(int snum,
3127                                        SPOOL_NOTIFY_INFO_DATA *data,
3128                                        print_queue_struct *queue,
3129                                        NT_PRINTER_INFO_LEVEL *printer,
3130                                        TALLOC_CTX *mem_ctx)
3131 {
3132         /* always respond 8 pages per minutes */
3133         /* a little hard ! */
3134         data->notify_data.value[0] = printer->info_2->averageppm;
3135         data->notify_data.value[1] = 0;
3136 }
3137
3138 /*******************************************************************
3139  * fill a notify_info_data with username
3140  ********************************************************************/
3141
3142 static void spoolss_notify_username(int snum,
3143                                     SPOOL_NOTIFY_INFO_DATA *data,
3144                                     print_queue_struct *queue,
3145                                     NT_PRINTER_INFO_LEVEL *printer,
3146                                     TALLOC_CTX *mem_ctx)
3147 {
3148         pstring temp;
3149         uint32 len;
3150
3151         len = rpcstr_push(temp, queue->fs_user, sizeof(temp)-2, STR_TERMINATE);
3152
3153         data->notify_data.data.length = len;
3154         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
3155
3156         if (!data->notify_data.data.string) {
3157                 data->notify_data.data.length = 0;
3158                 return;
3159         }
3160
3161         memcpy(data->notify_data.data.string, temp, len);
3162 }
3163
3164 /*******************************************************************
3165  * fill a notify_info_data with job status
3166  ********************************************************************/
3167
3168 static void spoolss_notify_job_status(int snum,
3169                                       SPOOL_NOTIFY_INFO_DATA *data,
3170                                       print_queue_struct *queue,
3171                                       NT_PRINTER_INFO_LEVEL *printer,
3172                                       TALLOC_CTX *mem_ctx)
3173 {
3174         data->notify_data.value[0]=nt_printj_status(queue->status);
3175         data->notify_data.value[1] = 0;
3176 }
3177
3178 /*******************************************************************
3179  * fill a notify_info_data with job name
3180  ********************************************************************/
3181
3182 static void spoolss_notify_job_name(int snum,
3183                                     SPOOL_NOTIFY_INFO_DATA *data,
3184                                     print_queue_struct *queue,
3185                                     NT_PRINTER_INFO_LEVEL *printer,
3186                                     TALLOC_CTX *mem_ctx)
3187 {
3188         pstring temp;
3189         uint32 len;
3190
3191         len = rpcstr_push(temp, queue->fs_file, sizeof(temp)-2, STR_TERMINATE);
3192
3193         data->notify_data.data.length = len;
3194         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
3195
3196         if (!data->notify_data.data.string) {
3197                 data->notify_data.data.length = 0;
3198                 return;
3199         }
3200
3201         memcpy(data->notify_data.data.string, temp, len);
3202 }
3203
3204 /*******************************************************************
3205  * fill a notify_info_data with job status
3206  ********************************************************************/
3207
3208 static void spoolss_notify_job_status_string(int snum,
3209                                              SPOOL_NOTIFY_INFO_DATA *data,
3210                                              print_queue_struct *queue,
3211                                              NT_PRINTER_INFO_LEVEL *printer,
3212                                              TALLOC_CTX *mem_ctx)
3213 {
3214         /*
3215          * Now we're returning job status codes we just return a "" here. JRA.
3216          */
3217
3218         const char *p = "";
3219         pstring temp;
3220         uint32 len;
3221
3222 #if 0 /* NO LONGER NEEDED - JRA. 02/22/2001 */
3223         p = "unknown";
3224
3225         switch (queue->status) {
3226         case LPQ_QUEUED:
3227                 p = "Queued";
3228                 break;
3229         case LPQ_PAUSED:
3230                 p = "";    /* NT provides the paused string */
3231                 break;
3232         case LPQ_SPOOLING:
3233                 p = "Spooling";
3234                 break;
3235         case LPQ_PRINTING:
3236                 p = "Printing";
3237                 break;
3238         }
3239 #endif /* NO LONGER NEEDED. */
3240
3241         len = rpcstr_push(temp, p, sizeof(temp) - 2, STR_TERMINATE);
3242
3243         data->notify_data.data.length = len;
3244         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
3245
3246         if (!data->notify_data.data.string) {
3247                 data->notify_data.data.length = 0;
3248                 return;
3249         }
3250
3251         memcpy(data->notify_data.data.string, temp, len);
3252 }
3253
3254 /*******************************************************************
3255  * fill a notify_info_data with job time
3256  ********************************************************************/
3257
3258 static void spoolss_notify_job_time(int snum,
3259                                     SPOOL_NOTIFY_INFO_DATA *data,
3260                                     print_queue_struct *queue,
3261                                     NT_PRINTER_INFO_LEVEL *printer,
3262                                     TALLOC_CTX *mem_ctx)
3263 {
3264         data->notify_data.value[0]=0x0;
3265         data->notify_data.value[1]=0;
3266 }
3267
3268 /*******************************************************************
3269  * fill a notify_info_data with job size
3270  ********************************************************************/
3271
3272 static void spoolss_notify_job_size(int snum,
3273                                     SPOOL_NOTIFY_INFO_DATA *data,
3274                                     print_queue_struct *queue,
3275                                     NT_PRINTER_INFO_LEVEL *printer,
3276                                     TALLOC_CTX *mem_ctx)
3277 {
3278         data->notify_data.value[0]=queue->size;
3279         data->notify_data.value[1]=0;
3280 }
3281
3282 /*******************************************************************
3283  * fill a notify_info_data with page info
3284  ********************************************************************/
3285 static void spoolss_notify_total_pages(int snum,
3286                                 SPOOL_NOTIFY_INFO_DATA *data,
3287                                 print_queue_struct *queue,
3288                                 NT_PRINTER_INFO_LEVEL *printer,
3289                                 TALLOC_CTX *mem_ctx)
3290 {
3291         data->notify_data.value[0]=queue->page_count;
3292         data->notify_data.value[1]=0;
3293 }
3294
3295 /*******************************************************************
3296  * fill a notify_info_data with pages printed info.
3297  ********************************************************************/
3298 static void spoolss_notify_pages_printed(int snum,
3299                                 SPOOL_NOTIFY_INFO_DATA *data,
3300                                 print_queue_struct *queue,
3301                                 NT_PRINTER_INFO_LEVEL *printer,
3302                                 TALLOC_CTX *mem_ctx)
3303 {
3304         data->notify_data.value[0]=0;  /* Add code when back-end tracks this */
3305         data->notify_data.value[1]=0;
3306 }
3307
3308 /*******************************************************************
3309  Fill a notify_info_data with job position.
3310  ********************************************************************/
3311
3312 static void spoolss_notify_job_position(int snum,
3313                                         SPOOL_NOTIFY_INFO_DATA *data,
3314                                         print_queue_struct *queue,
3315                                         NT_PRINTER_INFO_LEVEL *printer,
3316                                         TALLOC_CTX *mem_ctx)
3317 {
3318         data->notify_data.value[0]=queue->job;
3319         data->notify_data.value[1]=0;
3320 }
3321
3322 /*******************************************************************
3323  Fill a notify_info_data with submitted time.
3324  ********************************************************************/
3325
3326 static void spoolss_notify_submitted_time(int snum,
3327                                           SPOOL_NOTIFY_INFO_DATA *data,
3328                                           print_queue_struct *queue,
3329                                           NT_PRINTER_INFO_LEVEL *printer,
3330                                           TALLOC_CTX *mem_ctx)
3331 {
3332         struct tm *t;
3333         uint32 len;
3334         SYSTEMTIME st;
3335         char *p;
3336
3337         t=gmtime(&queue->time);
3338
3339         len = sizeof(SYSTEMTIME);
3340
3341         data->notify_data.data.length = len;
3342         data->notify_data.data.string = (uint16 *)TALLOC(mem_ctx, len);
3343
3344         if (!data->notify_data.data.string) {
3345                 data->notify_data.data.length = 0;
3346                 return;
3347         }
3348
3349         make_systemtime(&st, t);
3350
3351         /*
3352          * Systemtime must be linearized as a set of UINT16's.
3353          * Fix from Benjamin (Bj) Kuit bj@it.uts.edu.au
3354          */
3355
3356         p = (char *)data->notify_data.data.string;
3357         SSVAL(p, 0, st.year);
3358         SSVAL(p, 2, st.month);
3359         SSVAL(p, 4, st.dayofweek);
3360         SSVAL(p, 6, st.day);
3361         SSVAL(p, 8, st.hour);
3362         SSVAL(p, 10, st.minute);
3363         SSVAL(p, 12, st.second);
3364         SSVAL(p, 14, st.milliseconds);
3365 }
3366
3367 struct s_notify_info_data_table
3368 {
3369         uint16 type;
3370         uint16 field;
3371         const char *name;
3372         uint32 size;
3373         void (*fn) (int snum, SPOOL_NOTIFY_INFO_DATA *data,
3374                     print_queue_struct *queue,
3375                     NT_PRINTER_INFO_LEVEL *printer, TALLOC_CTX *mem_ctx);
3376 };
3377
3378 /* A table describing the various print notification constants and
3379    whether the notification data is a pointer to a variable sized
3380    buffer, a one value uint32 or a two value uint32. */
3381
3382 static const struct s_notify_info_data_table notify_info_data_table[] =
3383 {
3384 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_SERVER_NAME,         "PRINTER_NOTIFY_SERVER_NAME",         NOTIFY_STRING,   spoolss_notify_server_name },
3385 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PRINTER_NAME,        "PRINTER_NOTIFY_PRINTER_NAME",        NOTIFY_STRING,   spoolss_notify_printer_name },
3386 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_SHARE_NAME,          "PRINTER_NOTIFY_SHARE_NAME",          NOTIFY_STRING,   spoolss_notify_share_name },
3387 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PORT_NAME,           "PRINTER_NOTIFY_PORT_NAME",           NOTIFY_STRING,   spoolss_notify_port_name },
3388 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_DRIVER_NAME,         "PRINTER_NOTIFY_DRIVER_NAME",         NOTIFY_STRING,   spoolss_notify_driver_name },
3389 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_COMMENT,             "PRINTER_NOTIFY_COMMENT",             NOTIFY_STRING,   spoolss_notify_comment },
3390 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_LOCATION,            "PRINTER_NOTIFY_LOCATION",            NOTIFY_STRING,   spoolss_notify_location },
3391 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_DEVMODE,             "PRINTER_NOTIFY_DEVMODE",             NOTIFY_POINTER,   spoolss_notify_devmode },
3392 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_SEPFILE,             "PRINTER_NOTIFY_SEPFILE",             NOTIFY_STRING,   spoolss_notify_sepfile },
3393 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PRINT_PROCESSOR,     "PRINTER_NOTIFY_PRINT_PROCESSOR",     NOTIFY_STRING,   spoolss_notify_print_processor },
3394 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PARAMETERS,          "PRINTER_NOTIFY_PARAMETERS",          NOTIFY_STRING,   spoolss_notify_parameters },
3395 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_DATATYPE,            "PRINTER_NOTIFY_DATATYPE",            NOTIFY_STRING,   spoolss_notify_datatype },
3396 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_SECURITY_DESCRIPTOR, "PRINTER_NOTIFY_SECURITY_DESCRIPTOR", NOTIFY_SECDESC,   spoolss_notify_security_desc },
3397 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_ATTRIBUTES,          "PRINTER_NOTIFY_ATTRIBUTES",          NOTIFY_ONE_VALUE, spoolss_notify_attributes },
3398 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PRIORITY,            "PRINTER_NOTIFY_PRIORITY",            NOTIFY_ONE_VALUE, spoolss_notify_priority },
3399 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_DEFAULT_PRIORITY,    "PRINTER_NOTIFY_DEFAULT_PRIORITY",    NOTIFY_ONE_VALUE, spoolss_notify_default_priority },
3400 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_START_TIME,          "PRINTER_NOTIFY_START_TIME",          NOTIFY_ONE_VALUE, spoolss_notify_start_time },
3401 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_UNTIL_TIME,          "PRINTER_NOTIFY_UNTIL_TIME",          NOTIFY_ONE_VALUE, spoolss_notify_until_time },
3402 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_STATUS,              "PRINTER_NOTIFY_STATUS",              NOTIFY_ONE_VALUE, spoolss_notify_status },
3403 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_STATUS_STRING,       "PRINTER_NOTIFY_STATUS_STRING",       NOTIFY_POINTER,   NULL },
3404 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_CJOBS,               "PRINTER_NOTIFY_CJOBS",               NOTIFY_ONE_VALUE, spoolss_notify_cjobs },
3405 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_AVERAGE_PPM,         "PRINTER_NOTIFY_AVERAGE_PPM",         NOTIFY_ONE_VALUE, spoolss_notify_average_ppm },
3406 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_TOTAL_PAGES,         "PRINTER_NOTIFY_TOTAL_PAGES",         NOTIFY_POINTER,   NULL },
3407 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_PAGES_PRINTED,       "PRINTER_NOTIFY_PAGES_PRINTED",       NOTIFY_POINTER,   NULL },
3408 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_TOTAL_BYTES,         "PRINTER_NOTIFY_TOTAL_BYTES",         NOTIFY_POINTER,   NULL },
3409 { PRINTER_NOTIFY_TYPE, PRINTER_NOTIFY_BYTES_PRINTED,       "PRINTER_NOTIFY_BYTES_PRINTED",       NOTIFY_POINTER,   NULL },
3410 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PRINTER_NAME,            "JOB_NOTIFY_PRINTER_NAME",            NOTIFY_STRING,   spoolss_notify_printer_name },
3411 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_MACHINE_NAME,            "JOB_NOTIFY_MACHINE_NAME",            NOTIFY_STRING,   spoolss_notify_server_name },
3412 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PORT_NAME,               "JOB_NOTIFY_PORT_NAME",               NOTIFY_STRING,   spoolss_notify_port_name },
3413 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_USER_NAME,               "JOB_NOTIFY_USER_NAME",               NOTIFY_STRING,   spoolss_notify_username },
3414 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_NOTIFY_NAME,             "JOB_NOTIFY_NOTIFY_NAME",             NOTIFY_STRING,   spoolss_notify_username },
3415 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_DATATYPE,                "JOB_NOTIFY_DATATYPE",                NOTIFY_STRING,   spoolss_notify_datatype },
3416 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PRINT_PROCESSOR,         "JOB_NOTIFY_PRINT_PROCESSOR",         NOTIFY_STRING,   spoolss_notify_print_processor },
3417 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PARAMETERS,              "JOB_NOTIFY_PARAMETERS",              NOTIFY_STRING,   spoolss_notify_parameters },
3418 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_DRIVER_NAME,             "JOB_NOTIFY_DRIVER_NAME",             NOTIFY_STRING,   spoolss_notify_driver_name },
3419 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_DEVMODE,                 "JOB_NOTIFY_DEVMODE",                 NOTIFY_POINTER,   spoolss_notify_devmode },
3420 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_STATUS,                  "JOB_NOTIFY_STATUS",                  NOTIFY_ONE_VALUE, spoolss_notify_job_status },
3421 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_STATUS_STRING,           "JOB_NOTIFY_STATUS_STRING",           NOTIFY_STRING,   spoolss_notify_job_status_string },
3422 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_SECURITY_DESCRIPTOR,     "JOB_NOTIFY_SECURITY_DESCRIPTOR",     NOTIFY_POINTER,   NULL },
3423 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_DOCUMENT,                "JOB_NOTIFY_DOCUMENT",                NOTIFY_STRING,   spoolss_notify_job_name },
3424 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PRIORITY,                "JOB_NOTIFY_PRIORITY",                NOTIFY_ONE_VALUE, spoolss_notify_priority },
3425 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_POSITION,                "JOB_NOTIFY_POSITION",                NOTIFY_ONE_VALUE, spoolss_notify_job_position },
3426 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_SUBMITTED,               "JOB_NOTIFY_SUBMITTED",               NOTIFY_POINTER,   spoolss_notify_submitted_time },
3427 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_START_TIME,              "JOB_NOTIFY_START_TIME",              NOTIFY_ONE_VALUE, spoolss_notify_start_time },
3428 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_UNTIL_TIME,              "JOB_NOTIFY_UNTIL_TIME",              NOTIFY_ONE_VALUE, spoolss_notify_until_time },
3429 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_TIME,                    "JOB_NOTIFY_TIME",                    NOTIFY_ONE_VALUE, spoolss_notify_job_time },
3430 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_TOTAL_PAGES,             "JOB_NOTIFY_TOTAL_PAGES",             NOTIFY_ONE_VALUE, spoolss_notify_total_pages },
3431 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_PAGES_PRINTED,           "JOB_NOTIFY_PAGES_PRINTED",           NOTIFY_ONE_VALUE, spoolss_notify_pages_printed },
3432 { JOB_NOTIFY_TYPE,     JOB_NOTIFY_TOTAL_BYTES,             "JOB_NOTIFY_TOTAL_BYTES",             NOTIFY_ONE_VALUE, spoolss_notify_job_size },
3433 { PRINT_TABLE_END, 0x0, NULL, 0x0, NULL },
3434 };
3435
3436 /*******************************************************************
3437  Return the size of info_data structure.
3438 ********************************************************************/
3439
3440 static uint32 size_of_notify_info_data(uint16 type, uint16 field)
3441 {
3442         int i=0;
3443
3444         for (i = 0; i < sizeof(notify_info_data_table); i++)
3445         {
3446                 if ( (notify_info_data_table[i].type == type)
3447                         && (notify_info_data_table[i].field == field) )
3448                 {
3449                         switch(notify_info_data_table[i].size)
3450                         {
3451                         case NOTIFY_ONE_VALUE:
3452                         case NOTIFY_TWO_VALUE:
3453                                 return 1;
3454                         case NOTIFY_STRING:
3455                                 return 2;
3456
3457                         /* The only pointer notify data I have seen on
3458                            the wire is the submitted time and this has
3459                            the notify size set to 4. -tpot */
3460
3461                         case NOTIFY_POINTER:
3462                                 return 4;
3463
3464                                 case NOTIFY_SECDESC:
3465                                         return 5;
3466                         }
3467                 }
3468         }
3469
3470         DEBUG(5, ("invalid notify data type %d/%d\n", type, field));
3471
3472         return 0;
3473 }
3474
3475 /*******************************************************************
3476  Return the type of notify_info_data.
3477 ********************************************************************/
3478
3479 static int type_of_notify_info_data(uint16 type, uint16 field)
3480 {
3481         int i=0;
3482
3483         for (i = 0; i < sizeof(notify_info_data_table); i++) {
3484                 if (notify_info_data_table[i].type == type &&
3485                     notify_info_data_table[i].field == field)
3486                         return notify_info_data_table[i].size;
3487         }
3488
3489         return False;
3490 }
3491
3492 /****************************************************************************
3493 ****************************************************************************/
3494
3495 static int search_notify(uint16 type, uint16 field, int *value)
3496 {
3497         int i;
3498
3499         for (i = 0; notify_info_data_table[i].type != PRINT_TABLE_END; i++) {
3500                 if (notify_info_data_table[i].type == type &&
3501                     notify_info_data_table[i].field == field &&
3502                     notify_info_data_table[i].fn != NULL) {
3503                         *value = i;
3504                         return True;
3505                 }
3506         }
3507
3508         return False;
3509 }
3510
3511 /****************************************************************************
3512 ****************************************************************************/
3513
3514 void construct_info_data(SPOOL_NOTIFY_INFO_DATA *info_data, uint16 type, uint16 field, int id)
3515 {
3516         info_data->type     = type;
3517         info_data->field    = field;
3518         info_data->reserved = 0;
3519
3520         info_data->size     = size_of_notify_info_data(type, field);
3521         info_data->enc_type = type_of_notify_info_data(type, field);
3522
3523         info_data->id = id;
3524
3525 }
3526
3527
3528 /*******************************************************************
3529  *
3530  * fill a notify_info struct with info asked
3531  *
3532  ********************************************************************/
3533
3534 static BOOL construct_notify_printer_info(Printer_entry *print_hnd, SPOOL_NOTIFY_INFO *info, int
3535                                           snum, SPOOL_NOTIFY_OPTION_TYPE
3536                                           *option_type, uint32 id,
3537                                           TALLOC_CTX *mem_ctx)
3538 {
3539         int field_num,j;
3540         uint16 type;
3541         uint16 field;
3542
3543         SPOOL_NOTIFY_INFO_DATA *current_data;
3544         NT_PRINTER_INFO_LEVEL *printer = NULL;
3545         print_queue_struct *queue=NULL;
3546
3547         type=option_type->type;
3548
3549         DEBUG(4,("construct_notify_printer_info: Notify type: [%s], number of notify info: [%d] on printer: [%s]\n",
3550                 (option_type->type==PRINTER_NOTIFY_TYPE?"PRINTER_NOTIFY_TYPE":"JOB_NOTIFY_TYPE"),
3551                 option_type->count, lp_servicename(snum)));
3552
3553         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &printer, 2, lp_const_servicename(snum))))
3554                 return False;
3555
3556         for(field_num=0; field_num<option_type->count; field_num++) {
3557                 field = option_type->fields[field_num];
3558
3559                 DEBUG(4,("construct_notify_printer_info: notify [%d]: type [%x], field [%x]\n", field_num, type, field));
3560
3561                 if (!search_notify(type, field, &j) )
3562                         continue;
3563
3564                 if((info->data=SMB_REALLOC_ARRAY(info->data, SPOOL_NOTIFY_INFO_DATA, info->count+1)) == NULL) {
3565                         DEBUG(2,("construct_notify_printer_info: failed to enlarge buffer info->data!\n"));
3566                         return False;
3567                 }
3568
3569                 current_data = &info->data[info->count];
3570
3571                 construct_info_data(current_data, type, field, id);
3572
3573                 DEBUG(10,("construct_notify_printer_info: calling [%s]  snum=%d  printername=[%s])\n",
3574                                 notify_info_data_table[j].name, snum, printer->info_2->printername ));
3575
3576                 notify_info_data_table[j].fn(snum, current_data, queue,
3577                                              printer, mem_ctx);
3578
3579                 info->count++;
3580         }
3581
3582         free_a_printer(&printer, 2);
3583         return True;
3584 }
3585
3586 /*******************************************************************
3587  *
3588  * fill a notify_info struct with info asked
3589  *
3590  ********************************************************************/
3591
3592 static BOOL construct_notify_jobs_info(print_queue_struct *queue,
3593                                        SPOOL_NOTIFY_INFO *info,
3594                                        NT_PRINTER_INFO_LEVEL *printer,
3595                                        int snum, SPOOL_NOTIFY_OPTION_TYPE
3596                                        *option_type, uint32 id,
3597                                        TALLOC_CTX *mem_ctx)
3598 {
3599         int field_num,j;
3600         uint16 type;
3601         uint16 field;
3602
3603         SPOOL_NOTIFY_INFO_DATA *current_data;
3604
3605         DEBUG(4,("construct_notify_jobs_info\n"));
3606
3607         type = option_type->type;
3608
3609         DEBUGADD(4,("Notify type: [%s], number of notify info: [%d]\n",
3610                 (option_type->type==PRINTER_NOTIFY_TYPE?"PRINTER_NOTIFY_TYPE":"JOB_NOTIFY_TYPE"),
3611                 option_type->count));
3612
3613         for(field_num=0; field_num<option_type->count; field_num++) {
3614                 field = option_type->fields[field_num];
3615
3616                 if (!search_notify(type, field, &j) )
3617                         continue;
3618
3619                 if((info->data=SMB_REALLOC_ARRAY(info->data, SPOOL_NOTIFY_INFO_DATA, info->count+1)) == NULL) {
3620                         DEBUG(2,("construct_notify_jobs_info: failed to enlarg buffer info->data!\n"));
3621                         return False;
3622                 }
3623
3624                 current_data=&(info->data[info->count]);
3625
3626                 construct_info_data(current_data, type, field, id);
3627                 notify_info_data_table[j].fn(snum, current_data, queue,
3628                                              printer, mem_ctx);
3629                 info->count++;
3630         }
3631
3632         return True;
3633 }
3634
3635 /*
3636  * JFM: The enumeration is not that simple, it's even non obvious.
3637  *
3638  * let's take an example: I want to monitor the PRINTER SERVER for
3639  * the printer's name and the number of jobs currently queued.
3640  * So in the NOTIFY_OPTION, I have one NOTIFY_OPTION_TYPE structure.
3641  * Its type is PRINTER_NOTIFY_TYPE and it has 2 fields NAME and CJOBS.
3642  *
3643  * I have 3 printers on the back of my server.
3644  *
3645  * Now the response is a NOTIFY_INFO structure, with 6 NOTIFY_INFO_DATA
3646  * structures.
3647  *   Number     Data                    Id
3648  *      1       printer 1 name          1
3649  *      2       printer 1 cjob          1
3650  *      3       printer 2 name          2
3651  *      4       printer 2 cjob          2
3652  *      5       printer 3 name          3
3653  *      6       printer 3 name          3
3654  *
3655  * that's the print server case, the printer case is even worse.
3656  */
3657
3658 /*******************************************************************
3659  *
3660  * enumerate all printers on the printserver
3661  * fill a notify_info struct with info asked
3662  *
3663  ********************************************************************/
3664
3665 static WERROR printserver_notify_info(pipes_struct *p, POLICY_HND *hnd,
3666                                       SPOOL_NOTIFY_INFO *info,
3667                                       TALLOC_CTX *mem_ctx)
3668 {
3669         int snum;
3670         Printer_entry *Printer=find_printer_index_by_hnd(p, hnd);
3671         int n_services=lp_numservices();
3672         int i;
3673         SPOOL_NOTIFY_OPTION *option;
3674         SPOOL_NOTIFY_OPTION_TYPE *option_type;
3675
3676         DEBUG(4,("printserver_notify_info\n"));
3677
3678         if (!Printer)
3679                 return WERR_BADFID;
3680
3681         option=Printer->notify.option;
3682         info->version=2;
3683         info->data=NULL;
3684         info->count=0;
3685
3686         /* a bug in xp sp2 rc2 causes it to send a fnpcn request without
3687            sending a ffpcn() request first */
3688
3689         if ( !option )
3690                 return WERR_BADFID;
3691
3692         for (i=0; i<option->count; i++) {
3693                 option_type=&(option->ctr.type[i]);
3694
3695                 if (option_type->type!=PRINTER_NOTIFY_TYPE)
3696                         continue;
3697
3698                 for (snum=0; snum<n_services; snum++)
3699                 {
3700                         if ( lp_browseable(snum) && lp_snum_ok(snum) && lp_print_ok(snum) )
3701                                 construct_notify_printer_info ( Printer, info, snum, option_type, snum, mem_ctx );
3702                 }
3703         }
3704
3705 #if 0
3706         /*
3707          * Debugging information, don't delete.
3708          */
3709
3710         DEBUG(1,("dumping the NOTIFY_INFO\n"));
3711         DEBUGADD(1,("info->version:[%d], info->flags:[%d], info->count:[%d]\n", info->version, info->flags, info->count));
3712         DEBUGADD(1,("num\ttype\tfield\tres\tid\tsize\tenc_type\n"));
3713
3714         for (i=0; i<info->count; i++) {
3715                 DEBUGADD(1,("[%d]\t[%d]\t[%d]\t[%d]\t[%d]\t[%d]\t[%d]\n",
3716                 i, info->data[i].type, info->data[i].field, info->data[i].reserved,
3717                 info->data[i].id, info->data[i].size, info->data[i].enc_type));
3718         }
3719 #endif
3720
3721         return WERR_OK;
3722 }
3723
3724 /*******************************************************************
3725  *
3726  * fill a notify_info struct with info asked
3727  *
3728  ********************************************************************/
3729
3730 static WERROR printer_notify_info(pipes_struct *p, POLICY_HND *hnd, SPOOL_NOTIFY_INFO *info,
3731                                   TALLOC_CTX *mem_ctx)
3732 {
3733         int snum;
3734         Printer_entry *Printer=find_printer_index_by_hnd(p, hnd);
3735         int i;
3736         uint32 id;
3737         SPOOL_NOTIFY_OPTION *option;
3738         SPOOL_NOTIFY_OPTION_TYPE *option_type;
3739         int count,j;
3740         print_queue_struct *queue=NULL;
3741         print_status_struct status;
3742
3743         DEBUG(4,("printer_notify_info\n"));
3744
3745         if (!Printer)
3746                 return WERR_BADFID;
3747
3748         option=Printer->notify.option;
3749         id = 0x0;
3750         info->version=2;
3751         info->data=NULL;
3752         info->count=0;
3753
3754         /* a bug in xp sp2 rc2 causes it to send a fnpcn request without
3755            sending a ffpcn() request first */
3756
3757         if ( !option )
3758                 return WERR_BADFID;
3759
3760         get_printer_snum(p, hnd, &snum);
3761
3762         for (i=0; i<option->count; i++) {
3763                 option_type=&option->ctr.type[i];
3764
3765                 switch ( option_type->type ) {
3766                 case PRINTER_NOTIFY_TYPE:
3767                         if(construct_notify_printer_info(Printer, info, snum,
3768                                                          option_type, id,
3769                                                          mem_ctx))
3770                                 id--;
3771                         break;
3772
3773                 case JOB_NOTIFY_TYPE: {
3774                         NT_PRINTER_INFO_LEVEL *printer = NULL;
3775
3776                         count = print_queue_status(snum, &queue, &status);
3777
3778                         if (!W_ERROR_IS_OK(get_a_printer(Printer, &printer, 2, lp_const_servicename(snum))))
3779                                 goto done;
3780
3781                         for (j=0; j<count; j++) {
3782                                 construct_notify_jobs_info(&queue[j], info,
3783                                                            printer, snum,
3784                                                            option_type,
3785                                                            queue[j].job,
3786                                                            mem_ctx);
3787                         }
3788
3789                         free_a_printer(&printer, 2);
3790
3791                 done:
3792                         SAFE_FREE(queue);
3793                         break;
3794                 }
3795                 }
3796         }
3797
3798         /*
3799          * Debugging information, don't delete.
3800          */
3801         /*
3802         DEBUG(1,("dumping the NOTIFY_INFO\n"));
3803         DEBUGADD(1,("info->version:[%d], info->flags:[%d], info->count:[%d]\n", info->version, info->flags, info->count));
3804         DEBUGADD(1,("num\ttype\tfield\tres\tid\tsize\tenc_type\n"));
3805
3806         for (i=0; i<info->count; i++) {
3807                 DEBUGADD(1,("[%d]\t[%d]\t[%d]\t[%d]\t[%d]\t[%d]\t[%d]\n",
3808                 i, info->data[i].type, info->data[i].field, info->data[i].reserved,
3809                 info->data[i].id, info->data[i].size, info->data[i].enc_type));
3810         }
3811         */
3812         return WERR_OK;
3813 }
3814
3815 /********************************************************************
3816  * spoolss_rfnpcnex
3817  ********************************************************************/
3818
3819 WERROR _spoolss_rfnpcnex( pipes_struct *p, SPOOL_Q_RFNPCNEX *q_u, SPOOL_R_RFNPCNEX *r_u)
3820 {
3821         POLICY_HND *handle = &q_u->handle;
3822         SPOOL_NOTIFY_INFO *info = &r_u->info;
3823
3824         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
3825         WERROR result = WERR_BADFID;
3826
3827         /* we always have a NOTIFY_INFO struct */
3828         r_u->info_ptr=0x1;
3829
3830         if (!Printer) {
3831                 DEBUG(2,("_spoolss_rfnpcnex: Invalid handle (%s:%u:%u).\n",
3832                          OUR_HANDLE(handle)));
3833                 goto done;
3834         }
3835
3836         DEBUG(4,("Printer type %x\n",Printer->printer_type));
3837
3838         /*
3839          *      We are now using the change value, and
3840          *      I should check for PRINTER_NOTIFY_OPTIONS_REFRESH but as
3841          *      I don't have a global notification system, I'm sending back all the
3842          *      informations even when _NOTHING_ has changed.
3843          */
3844
3845         /* We need to keep track of the change value to send back in
3846            RRPCN replies otherwise our updates are ignored. */
3847
3848         Printer->notify.fnpcn = True;
3849
3850         if (Printer->notify.client_connected) {
3851                 DEBUG(10,("_spoolss_rfnpcnex: Saving change value in request [%x]\n", q_u->change));
3852                 Printer->notify.change = q_u->change;
3853         }
3854
3855         /* just ignore the SPOOL_NOTIFY_OPTION */
3856
3857         switch (Printer->printer_type) {
3858                 case SPLHND_SERVER:
3859                         result = printserver_notify_info(p, handle, info, p->mem_ctx);
3860                         break;
3861
3862                 case SPLHND_PRINTER:
3863                         result = printer_notify_info(p, handle, info, p->mem_ctx);
3864                         break;
3865         }
3866
3867         Printer->notify.fnpcn = False;
3868
3869 done:
3870         return result;
3871 }
3872
3873 /********************************************************************
3874  * construct_printer_info_0
3875  * fill a printer_info_0 struct
3876  ********************************************************************/
3877
3878 static BOOL construct_printer_info_0(Printer_entry *print_hnd, PRINTER_INFO_0 *printer, int snum)
3879 {
3880         pstring chaine;
3881         int count;
3882         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
3883         counter_printer_0 *session_counter;
3884         uint32 global_counter;
3885         struct tm *t;
3886         time_t setuptime;
3887         print_status_struct status;
3888
3889         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
3890                 return False;
3891
3892         count = print_queue_length(snum, &status);
3893
3894         /* check if we already have a counter for this printer */
3895         for(session_counter = counter_list; session_counter; session_counter = session_counter->next) {
3896                 if (session_counter->snum == snum)
3897                         break;
3898         }
3899
3900         /* it's the first time, add it to the list */
3901         if (session_counter==NULL) {
3902                 if((session_counter=SMB_MALLOC_P(counter_printer_0)) == NULL) {
3903                         free_a_printer(&ntprinter, 2);
3904                         return False;
3905                 }
3906                 ZERO_STRUCTP(session_counter);
3907                 session_counter->snum=snum;
3908                 session_counter->counter=0;
3909                 DLIST_ADD(counter_list, session_counter);
3910         }
3911
3912         /* increment it */
3913         session_counter->counter++;
3914
3915         /* JFM:
3916          * the global_counter should be stored in a TDB as it's common to all the clients
3917          * and should be zeroed on samba startup
3918          */
3919         global_counter=session_counter->counter;
3920
3921         pstrcpy(chaine,ntprinter->info_2->printername);
3922
3923         init_unistr(&printer->printername, chaine);
3924
3925         slprintf(chaine,sizeof(chaine)-1,"\\\\%s", get_server_name(print_hnd));
3926         init_unistr(&printer->servername, chaine);
3927
3928         printer->cjobs = count;
3929         printer->total_jobs = 0;
3930         printer->total_bytes = 0;
3931
3932         setuptime = (time_t)ntprinter->info_2->setuptime;
3933         t=gmtime(&setuptime);
3934
3935         printer->year = t->tm_year+1900;
3936         printer->month = t->tm_mon+1;
3937         printer->dayofweek = t->tm_wday;
3938         printer->day = t->tm_mday;
3939         printer->hour = t->tm_hour;
3940         printer->minute = t->tm_min;
3941         printer->second = t->tm_sec;
3942         printer->milliseconds = 0;
3943
3944         printer->global_counter = global_counter;
3945         printer->total_pages = 0;
3946
3947         /* in 2.2 we reported ourselves as 0x0004 and 0x0565 */
3948         printer->major_version = 0x0005;        /* NT 5 */
3949         printer->build_version = 0x0893;        /* build 2195 */
3950
3951         printer->unknown7 = 0x1;
3952         printer->unknown8 = 0x0;
3953         printer->unknown9 = 0x0;
3954         printer->session_counter = session_counter->counter;
3955         printer->unknown11 = 0x0;
3956         printer->printer_errors = 0x0;          /* number of print failure */
3957         printer->unknown13 = 0x0;
3958         printer->unknown14 = 0x1;
3959         printer->unknown15 = 0x024a;            /* 586 Pentium ? */
3960         printer->unknown16 =  0x0;
3961         printer->change_id = ntprinter->info_2->changeid; /* ChangeID in milliseconds*/
3962         printer->unknown18 =  0x0;
3963         printer->status = nt_printq_status(status.status);
3964         printer->unknown20 =  0x0;
3965         printer->c_setprinter = get_c_setprinter(); /* monotonically increasing sum of delta printer counts */
3966         printer->unknown22 = 0x0;
3967         printer->unknown23 = 0x6;               /* 6  ???*/
3968         printer->unknown24 = 0;                 /* unknown 24 to 26 are always 0 */
3969         printer->unknown25 = 0;
3970         printer->unknown26 = 0;
3971         printer->unknown27 = 0;
3972         printer->unknown28 = 0;
3973         printer->unknown29 = 0;
3974
3975         free_a_printer(&ntprinter,2);
3976         return (True);
3977 }
3978
3979 /********************************************************************
3980  * construct_printer_info_1
3981  * fill a printer_info_1 struct
3982  ********************************************************************/
3983 static BOOL construct_printer_info_1(Printer_entry *print_hnd, uint32 flags, PRINTER_INFO_1 *printer, int snum)
3984 {
3985         pstring chaine;
3986         pstring chaine2;
3987         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
3988
3989         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
3990                 return False;
3991
3992         printer->flags=flags;
3993
3994         if (*ntprinter->info_2->comment == '\0') {
3995                 init_unistr(&printer->comment, lp_comment(snum));
3996                 slprintf(chaine,sizeof(chaine)-1,"%s,%s,%s", ntprinter->info_2->printername,
3997                         ntprinter->info_2->drivername, lp_comment(snum));
3998         }
3999         else {
4000                 init_unistr(&printer->comment, ntprinter->info_2->comment); /* saved comment. */
4001                 slprintf(chaine,sizeof(chaine)-1,"%s,%s,%s", ntprinter->info_2->printername,
4002                         ntprinter->info_2->drivername, ntprinter->info_2->comment);
4003         }
4004
4005         slprintf(chaine2,sizeof(chaine)-1,"%s", ntprinter->info_2->printername);
4006
4007         init_unistr(&printer->description, chaine);
4008         init_unistr(&printer->name, chaine2);
4009
4010         free_a_printer(&ntprinter,2);
4011
4012         return True;
4013 }
4014
4015 /****************************************************************************
4016  Free a DEVMODE struct.
4017 ****************************************************************************/
4018
4019 static void free_dev_mode(DEVICEMODE *dev)
4020 {
4021         if (dev == NULL)
4022                 return;
4023
4024         SAFE_FREE(dev->dev_private);
4025         SAFE_FREE(dev);
4026 }
4027
4028
4029 /****************************************************************************
4030  Convert an NT_DEVICEMODE to a DEVICEMODE structure.  Both pointers
4031  should be valid upon entry
4032 ****************************************************************************/
4033
4034 static BOOL convert_nt_devicemode( DEVICEMODE *devmode, NT_DEVICEMODE *ntdevmode )
4035 {
4036         if ( !devmode || !ntdevmode )
4037                 return False;
4038
4039         init_unistr(&devmode->devicename, ntdevmode->devicename);
4040
4041         init_unistr(&devmode->formname, ntdevmode->formname);
4042
4043         devmode->specversion      = ntdevmode->specversion;
4044         devmode->driverversion    = ntdevmode->driverversion;
4045         devmode->size             = ntdevmode->size;
4046         devmode->driverextra      = ntdevmode->driverextra;
4047         devmode->fields           = ntdevmode->fields;
4048
4049         devmode->orientation      = ntdevmode->orientation;
4050         devmode->papersize        = ntdevmode->papersize;
4051         devmode->paperlength      = ntdevmode->paperlength;
4052         devmode->paperwidth       = ntdevmode->paperwidth;
4053         devmode->scale            = ntdevmode->scale;
4054         devmode->copies           = ntdevmode->copies;
4055         devmode->defaultsource    = ntdevmode->defaultsource;
4056         devmode->printquality     = ntdevmode->printquality;
4057         devmode->color            = ntdevmode->color;
4058         devmode->duplex           = ntdevmode->duplex;
4059         devmode->yresolution      = ntdevmode->yresolution;
4060         devmode->ttoption         = ntdevmode->ttoption;
4061         devmode->collate          = ntdevmode->collate;
4062         devmode->icmmethod        = ntdevmode->icmmethod;
4063         devmode->icmintent        = ntdevmode->icmintent;
4064         devmode->mediatype        = ntdevmode->mediatype;
4065         devmode->dithertype       = ntdevmode->dithertype;
4066
4067         if (ntdevmode->nt_dev_private != NULL) {
4068                 if ((devmode->dev_private=(uint8 *)memdup(ntdevmode->nt_dev_private, ntdevmode->driverextra)) == NULL)
4069                         return False;
4070         }
4071
4072         return True;
4073 }
4074
4075 /****************************************************************************
4076  Create a DEVMODE struct. Returns malloced memory.
4077 ****************************************************************************/
4078
4079 DEVICEMODE *construct_dev_mode(int snum)
4080 {
4081         NT_PRINTER_INFO_LEVEL   *printer = NULL;
4082         DEVICEMODE              *devmode = NULL;
4083
4084         DEBUG(7,("construct_dev_mode\n"));
4085
4086         DEBUGADD(8,("getting printer characteristics\n"));
4087
4088         if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_const_servicename(snum))))
4089                 return NULL;
4090
4091         if ( !printer->info_2->devmode ) {
4092                 DEBUG(5, ("BONG! There was no device mode!\n"));
4093                 goto done;
4094         }
4095
4096         if ((devmode = SMB_MALLOC_P(DEVICEMODE)) == NULL) {
4097                 DEBUG(2,("construct_dev_mode: malloc fail.\n"));
4098                 goto done;
4099         }
4100
4101         ZERO_STRUCTP(devmode);
4102
4103         DEBUGADD(8,("loading DEVICEMODE\n"));
4104
4105         if ( !convert_nt_devicemode( devmode, printer->info_2->devmode ) ) {
4106                 free_dev_mode( devmode );
4107                 devmode = NULL;
4108         }
4109
4110 done:
4111         free_a_printer(&printer,2);
4112
4113         return devmode;
4114 }
4115
4116 /********************************************************************
4117  * construct_printer_info_2
4118  * fill a printer_info_2 struct
4119  ********************************************************************/
4120
4121 static BOOL construct_printer_info_2(Printer_entry *print_hnd, PRINTER_INFO_2 *printer, int snum)
4122 {
4123         int count;
4124         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
4125
4126         print_status_struct status;
4127
4128         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
4129                 return False;
4130
4131         count = print_queue_length(snum, &status);
4132
4133         init_unistr(&printer->servername, ntprinter->info_2->servername); /* servername*/
4134         init_unistr(&printer->printername, ntprinter->info_2->printername);                             /* printername*/
4135         init_unistr(&printer->sharename, lp_servicename(snum));                 /* sharename */
4136         init_unistr(&printer->portname, ntprinter->info_2->portname);                   /* port */
4137         init_unistr(&printer->drivername, ntprinter->info_2->drivername);       /* drivername */
4138
4139         if (*ntprinter->info_2->comment == '\0')
4140                 init_unistr(&printer->comment, lp_comment(snum));                       /* comment */
4141         else
4142                 init_unistr(&printer->comment, ntprinter->info_2->comment); /* saved comment. */
4143
4144         init_unistr(&printer->location, ntprinter->info_2->location);           /* location */
4145         init_unistr(&printer->sepfile, ntprinter->info_2->sepfile);             /* separator file */
4146         init_unistr(&printer->printprocessor, ntprinter->info_2->printprocessor);/* print processor */
4147         init_unistr(&printer->datatype, ntprinter->info_2->datatype);           /* datatype */
4148         init_unistr(&printer->parameters, ntprinter->info_2->parameters);       /* parameters (of print processor) */
4149
4150         printer->attributes = ntprinter->info_2->attributes;
4151
4152         printer->priority = ntprinter->info_2->priority;                                /* priority */
4153         printer->defaultpriority = ntprinter->info_2->default_priority;         /* default priority */
4154         printer->starttime = ntprinter->info_2->starttime;                      /* starttime */
4155         printer->untiltime = ntprinter->info_2->untiltime;                      /* untiltime */
4156         printer->status = nt_printq_status(status.status);                      /* status */
4157         printer->cjobs = count;                                                 /* jobs */
4158         printer->averageppm = ntprinter->info_2->averageppm;                    /* average pages per minute */
4159
4160         if ( !(printer->devmode = construct_dev_mode(snum)) )
4161                 DEBUG(8, ("Returning NULL Devicemode!\n"));
4162
4163         printer->secdesc = NULL;
4164
4165         if ( ntprinter->info_2->secdesc_buf
4166                 && ntprinter->info_2->secdesc_buf->len != 0 )
4167         {
4168                 /* don't use talloc_steal() here unless you do a deep steal of all
4169                    the SEC_DESC members */
4170
4171                 printer->secdesc = dup_sec_desc( get_talloc_ctx(),
4172                         ntprinter->info_2->secdesc_buf->sec );
4173         }
4174
4175         free_a_printer(&ntprinter, 2);
4176
4177         return True;
4178 }
4179
4180 /********************************************************************
4181  * construct_printer_info_3
4182  * fill a printer_info_3 struct
4183  ********************************************************************/
4184
4185 static BOOL construct_printer_info_3(Printer_entry *print_hnd, PRINTER_INFO_3 **pp_printer, int snum)
4186 {
4187         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
4188         PRINTER_INFO_3 *printer = NULL;
4189
4190         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
4191                 return False;
4192
4193         *pp_printer = NULL;
4194         if ((printer = SMB_MALLOC_P(PRINTER_INFO_3)) == NULL) {
4195                 DEBUG(2,("construct_printer_info_3: malloc fail.\n"));
4196                 return False;
4197         }
4198
4199         ZERO_STRUCTP(printer);
4200
4201         /* These are the components of the SD we are returning. */
4202
4203         printer->flags = 0x4;
4204
4205         if (ntprinter->info_2->secdesc_buf && ntprinter->info_2->secdesc_buf->len != 0) {
4206                 /* don't use talloc_steal() here unless you do a deep steal of all
4207                    the SEC_DESC members */
4208
4209                 printer->secdesc = dup_sec_desc( get_talloc_ctx(),
4210                         ntprinter->info_2->secdesc_buf->sec );
4211         }
4212
4213         free_a_printer(&ntprinter, 2);
4214
4215         *pp_printer = printer;
4216         return True;
4217 }
4218
4219 /********************************************************************
4220  * construct_printer_info_4
4221  * fill a printer_info_4 struct
4222  ********************************************************************/
4223
4224 static BOOL construct_printer_info_4(Printer_entry *print_hnd, PRINTER_INFO_4 *printer, int snum)
4225 {
4226         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
4227
4228         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
4229                 return False;
4230
4231         init_unistr(&printer->printername, ntprinter->info_2->printername);                             /* printername*/
4232         init_unistr(&printer->servername, ntprinter->info_2->servername); /* servername*/
4233         printer->attributes = ntprinter->info_2->attributes;
4234
4235         free_a_printer(&ntprinter, 2);
4236         return True;
4237 }
4238
4239 /********************************************************************
4240  * construct_printer_info_5
4241  * fill a printer_info_5 struct
4242  ********************************************************************/
4243
4244 static BOOL construct_printer_info_5(Printer_entry *print_hnd, PRINTER_INFO_5 *printer, int snum)
4245 {
4246         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
4247
4248         if (!W_ERROR_IS_OK(get_a_printer(print_hnd, &ntprinter, 2, lp_const_servicename(snum))))
4249                 return False;
4250
4251         init_unistr(&printer->printername, ntprinter->info_2->printername);
4252         init_unistr(&printer->portname, ntprinter->info_2->portname);
4253         printer->attributes = ntprinter->info_2->attributes;
4254
4255         /* these two are not used by NT+ according to MSDN */
4256
4257         printer->device_not_selected_timeout = 0x0;  /* have seen 0x3a98 */
4258         printer->transmission_retry_timeout  = 0x0;  /* have seen 0xafc8 */
4259
4260         free_a_printer(&ntprinter, 2);
4261
4262         return True;
4263 }
4264
4265 /********************************************************************
4266  * construct_printer_info_7
4267  * fill a printer_info_7 struct
4268  ********************************************************************/
4269
4270 static BOOL construct_printer_info_7(Printer_entry *print_hnd, PRINTER_INFO_7 *printer, int snum)
4271 {
4272         char *guid_str = NULL;
4273         struct uuid guid;
4274
4275         if (is_printer_published(print_hnd, snum, &guid)) {
4276                 asprintf(&guid_str, "{%s}", smb_uuid_string_static(guid));
4277                 strupper_m(guid_str);
4278                 init_unistr(&printer->guid, guid_str);
4279                 printer->action = SPOOL_DS_PUBLISH;
4280         } else {
4281                 init_unistr(&printer->guid, "");
4282                 printer->action = SPOOL_DS_UNPUBLISH;
4283         }
4284
4285         return True;
4286 }
4287
4288 /********************************************************************
4289  Spoolss_enumprinters.
4290 ********************************************************************/
4291
4292 static WERROR enum_all_printers_info_1(uint32 flags, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4293 {
4294         int snum;
4295         int i;
4296         int n_services=lp_numservices();
4297         PRINTER_INFO_1 *printers=NULL;
4298         PRINTER_INFO_1 current_prt;
4299         WERROR result = WERR_OK;
4300
4301         DEBUG(4,("enum_all_printers_info_1\n"));
4302
4303         for (snum=0; snum<n_services; snum++) {
4304                 if (lp_browseable(snum) && lp_snum_ok(snum) && lp_print_ok(snum) ) {
4305                         DEBUG(4,("Found a printer in smb.conf: %s[%x]\n", lp_servicename(snum), snum));
4306
4307                         if (construct_printer_info_1(NULL, flags, &current_prt, snum)) {
4308                                 if((printers=SMB_REALLOC_ARRAY(printers, PRINTER_INFO_1, *returned +1)) == NULL) {
4309                                         DEBUG(2,("enum_all_printers_info_1: failed to enlarge printers buffer!\n"));
4310                                         *returned=0;
4311                                         return WERR_NOMEM;
4312                                 }
4313                                 DEBUG(4,("ReAlloced memory for [%d] PRINTER_INFO_1\n", *returned));
4314
4315                                 memcpy(&printers[*returned], &current_prt, sizeof(PRINTER_INFO_1));
4316                                 (*returned)++;
4317                         }
4318                 }
4319         }
4320
4321         /* check the required size. */
4322         for (i=0; i<*returned; i++)
4323                 (*needed) += spoolss_size_printer_info_1(&printers[i]);
4324
4325         if (*needed > offered) {
4326                 result = WERR_INSUFFICIENT_BUFFER;
4327                 goto out;
4328         }
4329
4330         if (!rpcbuf_alloc_size(buffer, *needed)) {
4331                 result = WERR_NOMEM;
4332                 goto out;
4333         }
4334
4335         /* fill the buffer with the structures */
4336         for (i=0; i<*returned; i++)
4337                 smb_io_printer_info_1("", buffer, &printers[i], 0);
4338
4339 out:
4340         /* clear memory */
4341
4342         SAFE_FREE(printers);
4343
4344         if ( !W_ERROR_IS_OK(result) )
4345                 *returned = 0;
4346
4347         return result;
4348 }
4349
4350 /********************************************************************
4351  enum_all_printers_info_1_local.
4352 *********************************************************************/
4353
4354 static WERROR enum_all_printers_info_1_local(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4355 {
4356         DEBUG(4,("enum_all_printers_info_1_local\n"));
4357
4358         return enum_all_printers_info_1(PRINTER_ENUM_ICON8, buffer, offered, needed, returned);
4359 }
4360
4361 /********************************************************************
4362  enum_all_printers_info_1_name.
4363 *********************************************************************/
4364
4365 static WERROR enum_all_printers_info_1_name(fstring name, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4366 {
4367         char *s = name;
4368
4369         DEBUG(4,("enum_all_printers_info_1_name\n"));
4370
4371         if ((name[0] == '\\') && (name[1] == '\\'))
4372                 s = name + 2;
4373
4374         if (is_myname_or_ipaddr(s)) {
4375                 return enum_all_printers_info_1(PRINTER_ENUM_ICON8, buffer, offered, needed, returned);
4376         }
4377         else
4378                 return WERR_INVALID_NAME;
4379 }
4380
4381 #if 0   /* JERRY -- disabled for now.  Don't think this is used, tested, or correct */
4382 /********************************************************************
4383  enum_all_printers_info_1_remote.
4384 *********************************************************************/
4385
4386 static WERROR enum_all_printers_info_1_remote(fstring name, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4387 {
4388         PRINTER_INFO_1 *printer;
4389         fstring printername;
4390         fstring desc;
4391         fstring comment;
4392         DEBUG(4,("enum_all_printers_info_1_remote\n"));
4393         WERROR result = WERR_OK;
4394
4395         /* JFM: currently it's more a place holder than anything else.
4396          * In the spooler world there is a notion of server registration.
4397          * the print servers are registered on the PDC (in the same domain)
4398          *
4399          * We should have a TDB here. The registration is done thru an
4400          * undocumented RPC call.
4401          */
4402
4403         if((printer=SMB_MALLOC_P(PRINTER_INFO_1)) == NULL)
4404                 return WERR_NOMEM;
4405
4406         *returned=1;
4407
4408         slprintf(printername, sizeof(printername)-1,"Windows NT Remote Printers!!\\\\%s", name);
4409         slprintf(desc, sizeof(desc)-1,"%s", name);
4410         slprintf(comment, sizeof(comment)-1, "Logged on Domain");
4411
4412         init_unistr(&printer->description, desc);
4413         init_unistr(&printer->name, printername);
4414         init_unistr(&printer->comment, comment);
4415         printer->flags=PRINTER_ENUM_ICON3|PRINTER_ENUM_CONTAINER;
4416
4417         /* check the required size. */
4418         *needed += spoolss_size_printer_info_1(printer);
4419
4420         if (*needed > offered) {
4421                 result = WERR_INSUFFICIENT_BUFFER;
4422                 goto out;
4423         }
4424
4425         if (!rpcbuf_alloc_size(buffer, *needed)) {
4426                 result = WERR_NOMEM;
4427                 goto out;
4428         }
4429
4430         /* fill the buffer with the structures */
4431         smb_io_printer_info_1("", buffer, printer, 0);
4432
4433 out:
4434         /* clear memory */
4435         SAFE_FREE(printer);
4436
4437         if ( !W_ERROR_IS_OK(result) )
4438                 *returned = 0;
4439
4440         return result;
4441 }
4442
4443 #endif
4444
4445 /********************************************************************
4446  enum_all_printers_info_1_network.
4447 *********************************************************************/
4448
4449 static WERROR enum_all_printers_info_1_network(fstring name, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4450 {
4451         char *s = name;
4452
4453         DEBUG(4,("enum_all_printers_info_1_network\n"));
4454
4455         /* If we respond to a enum_printers level 1 on our name with flags
4456            set to PRINTER_ENUM_REMOTE with a list of printers then these
4457            printers incorrectly appear in the APW browse list.
4458            Specifically the printers for the server appear at the workgroup
4459            level where all the other servers in the domain are
4460            listed. Windows responds to this call with a
4461            WERR_CAN_NOT_COMPLETE so we should do the same. */
4462
4463         if (name[0] == '\\' && name[1] == '\\')
4464                  s = name + 2;
4465
4466         if (is_myname_or_ipaddr(s))
4467                  return WERR_CAN_NOT_COMPLETE;
4468
4469         return enum_all_printers_info_1(PRINTER_ENUM_UNKNOWN_8, buffer, offered, needed, returned);
4470 }
4471
4472 /********************************************************************
4473  * api_spoolss_enumprinters
4474  *
4475  * called from api_spoolss_enumprinters (see this to understand)
4476  ********************************************************************/
4477
4478 static WERROR enum_all_printers_info_2(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
4479 {
4480         int snum;
4481         int i;
4482         int n_services=lp_numservices();
4483         PRINTER_INFO_2 *printers=NULL;
4484         PRINTER_INFO_2 current_prt;
4485         WERROR result = WERR_OK;
4486
4487         *returned = 0;
4488
4489         for (snum=0; snum<n_services; snum++) {
4490                 if (lp_browseable(snum) && lp_snum_ok(snum) && lp_print_ok(snum) ) {
4491                         DEBUG(4,("Found a printer in smb.conf: %s[%x]\n", lp_servicename(snum), snum));
4492
4493                         if (construct_printer_info_2(NULL, &current_prt, snum)) {
4494                                 if ( !(printers=SMB_REALLOC_ARRAY(printers, PRINTER_INFO_2, *returned +1)) ) {
4495                                         DEBUG(2,("enum_all_printers_info_2: failed to enlarge printers buffer!\n"));
4496                                         *returned = 0;
4497                                         return WERR_NOMEM;
4498                                 }
4499
4500                                 DEBUG(4,("ReAlloced memory for [%d] PRINTER_INFO_2\n", *returned + 1));
4501
4502                                 memcpy(&printers[*returned], &current_prt, sizeof(PRINTER_INFO_2));
4503
4504                                 (*returned)++;
4505                         }
4506                 }
4507         }
4508
4509         /* check the required size. */
4510         for (i=0; i<*returned; i++)
4511                 (*needed) += spoolss_size_printer_info_2(&printers[i]);
4512
4513         if (*needed > offered) {
4514                 result = WERR_INSUFFICIENT_BUFFER;
4515                 goto out;
4516         }
4517
4518         if (!rpcbuf_alloc_size(buffer, *needed)) {
4519                 result = WERR_NOMEM;
4520                 goto out;
4521         }
4522
4523         /* fill the buffer with the structures */
4524         for (i=0; i<*returned; i++)
4525                 smb_io_printer_info_2("", buffer, &(printers[i]), 0);
4526
4527 out:
4528         /* clear memory */
4529
4530         for (i=0; i<*returned; i++)
4531                 free_devmode(printers[i].devmode);
4532
4533         SAFE_FREE(printers);
4534
4535         if ( !W_ERROR_IS_OK(result) )
4536                 *returned = 0;
4537
4538         return result;
4539 }
4540
4541 /********************************************************************
4542  * handle enumeration of printers at level 1
4543  ********************************************************************/
4544
4545 static WERROR enumprinters_level1( uint32 flags, fstring name,
4546                                  RPC_BUFFER *buffer, uint32 offered,
4547                                  uint32 *needed, uint32 *returned)
4548 {
4549         /* Not all the flags are equals */
4550
4551         if (flags & PRINTER_ENUM_LOCAL)
4552                 return enum_all_printers_info_1_local(buffer, offered, needed, returned);
4553
4554         if (flags & PRINTER_ENUM_NAME)
4555                 return enum_all_printers_info_1_name(name, buffer, offered, needed, returned);
4556
4557 #if 0   /* JERRY - disabled for now */
4558         if (flags & PRINTER_ENUM_REMOTE)
4559                 return enum_all_printers_info_1_remote(name, buffer, offered, needed, returned);
4560 #endif
4561
4562         if (flags & PRINTER_ENUM_NETWORK)
4563                 return enum_all_printers_info_1_network(name, buffer, offered, needed, returned);
4564
4565         return WERR_OK; /* NT4sp5 does that */
4566 }
4567
4568 /********************************************************************
4569  * handle enumeration of printers at level 2
4570  ********************************************************************/
4571
4572 static WERROR enumprinters_level2( uint32 flags, fstring servername,
4573                                  RPC_BUFFER *buffer, uint32 offered,
4574                                  uint32 *needed, uint32 *returned)
4575 {
4576         char *s = servername;
4577
4578         if (flags & PRINTER_ENUM_LOCAL) {
4579                         return enum_all_printers_info_2(buffer, offered, needed, returned);
4580         }
4581
4582         if (flags & PRINTER_ENUM_NAME) {
4583                 if ((servername[0] == '\\') && (servername[1] == '\\'))
4584                         s = servername + 2;
4585                 if (is_myname_or_ipaddr(s))
4586                         return enum_all_printers_info_2(buffer, offered, needed, returned);
4587                 else
4588                         return WERR_INVALID_NAME;
4589         }
4590
4591         if (flags & PRINTER_ENUM_REMOTE)
4592                 return WERR_UNKNOWN_LEVEL;
4593
4594         return WERR_OK;
4595 }
4596
4597 /********************************************************************
4598  * handle enumeration of printers at level 5
4599  ********************************************************************/
4600
4601 static WERROR enumprinters_level5( uint32 flags, fstring servername,
4602                                  RPC_BUFFER *buffer, uint32 offered,
4603                                  uint32 *needed, uint32 *returned)
4604 {
4605 /*      return enum_all_printers_info_5(buffer, offered, needed, returned);*/
4606         return WERR_OK;
4607 }
4608
4609 /********************************************************************
4610  * api_spoolss_enumprinters
4611  *
4612  * called from api_spoolss_enumprinters (see this to understand)
4613  ********************************************************************/
4614
4615 WERROR _spoolss_enumprinters( pipes_struct *p, SPOOL_Q_ENUMPRINTERS *q_u, SPOOL_R_ENUMPRINTERS *r_u)
4616 {
4617         uint32 flags = q_u->flags;
4618         UNISTR2 *servername = &q_u->servername;
4619         uint32 level = q_u->level;
4620         RPC_BUFFER *buffer = NULL;
4621         uint32 offered = q_u->offered;
4622         uint32 *needed = &r_u->needed;
4623         uint32 *returned = &r_u->returned;
4624
4625         fstring name;
4626
4627         /* that's an [in out] buffer */
4628
4629         if (!q_u->buffer) {
4630                 return WERR_INVALID_PARAM;
4631         }
4632
4633         rpcbuf_move(q_u->buffer, &r_u->buffer);
4634         buffer = r_u->buffer;
4635
4636         DEBUG(4,("_spoolss_enumprinters\n"));
4637
4638         *needed=0;
4639         *returned=0;
4640
4641         /*
4642          * Level 1:
4643          *          flags==PRINTER_ENUM_NAME
4644          *           if name=="" then enumerates all printers
4645          *           if name!="" then enumerate the printer
4646          *          flags==PRINTER_ENUM_REMOTE
4647          *          name is NULL, enumerate printers
4648          * Level 2: name!="" enumerates printers, name can't be NULL
4649          * Level 3: doesn't exist
4650          * Level 4: does a local registry lookup
4651          * Level 5: same as Level 2
4652          */
4653
4654         unistr2_to_ascii(name, servername, sizeof(name)-1);
4655         strupper_m(name);
4656
4657         switch (level) {
4658         case 1:
4659                 return enumprinters_level1(flags, name, buffer, offered, needed, returned);
4660         case 2:
4661                 return enumprinters_level2(flags, name, buffer, offered, needed, returned);
4662         case 5:
4663                 return enumprinters_level5(flags, name, buffer, offered, needed, returned);
4664         case 3:
4665         case 4:
4666                 break;
4667         }
4668         return WERR_UNKNOWN_LEVEL;
4669 }
4670
4671 /****************************************************************************
4672 ****************************************************************************/
4673
4674 static WERROR getprinter_level_0(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4675 {
4676         PRINTER_INFO_0 *printer=NULL;
4677         WERROR result = WERR_OK;
4678
4679         if((printer=SMB_MALLOC_P(PRINTER_INFO_0)) == NULL)
4680                 return WERR_NOMEM;
4681
4682         construct_printer_info_0(print_hnd, printer, snum);
4683
4684         /* check the required size. */
4685         *needed += spoolss_size_printer_info_0(printer);
4686
4687         if (*needed > offered) {
4688                 result = WERR_INSUFFICIENT_BUFFER;
4689                 goto out;
4690         }
4691
4692         if (!rpcbuf_alloc_size(buffer, *needed)) {
4693                 result = WERR_NOMEM;
4694                 goto out;
4695         }
4696
4697         /* fill the buffer with the structures */
4698         smb_io_printer_info_0("", buffer, printer, 0);
4699
4700 out:
4701         /* clear memory */
4702
4703         SAFE_FREE(printer);
4704
4705         return result;
4706 }
4707
4708 /****************************************************************************
4709 ****************************************************************************/
4710
4711 static WERROR getprinter_level_1(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4712 {
4713         PRINTER_INFO_1 *printer=NULL;
4714         WERROR result = WERR_OK;
4715
4716         if((printer=SMB_MALLOC_P(PRINTER_INFO_1)) == NULL)
4717                 return WERR_NOMEM;
4718
4719         construct_printer_info_1(print_hnd, PRINTER_ENUM_ICON8, printer, snum);
4720
4721         /* check the required size. */
4722         *needed += spoolss_size_printer_info_1(printer);
4723
4724         if (*needed > offered) {
4725                 result = WERR_INSUFFICIENT_BUFFER;
4726                 goto out;
4727         }
4728
4729         if (!rpcbuf_alloc_size(buffer, *needed)) {
4730                 result = WERR_NOMEM;
4731                 goto out;
4732         }
4733
4734         /* fill the buffer with the structures */
4735         smb_io_printer_info_1("", buffer, printer, 0);
4736
4737 out:
4738         /* clear memory */
4739         SAFE_FREE(printer);
4740
4741         return result;
4742 }
4743
4744 /****************************************************************************
4745 ****************************************************************************/
4746
4747 static WERROR getprinter_level_2(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4748 {
4749         PRINTER_INFO_2 *printer=NULL;
4750         WERROR result = WERR_OK;
4751
4752         if((printer=SMB_MALLOC_P(PRINTER_INFO_2))==NULL)
4753                 return WERR_NOMEM;
4754
4755         construct_printer_info_2(print_hnd, printer, snum);
4756
4757         /* check the required size. */
4758         *needed += spoolss_size_printer_info_2(printer);
4759
4760         if (*needed > offered) {
4761                 result = WERR_INSUFFICIENT_BUFFER;
4762                 goto out;
4763         }
4764
4765         if (!rpcbuf_alloc_size(buffer, *needed)) {
4766                 result = WERR_NOMEM;
4767                 goto out;
4768         }
4769
4770         /* fill the buffer with the structures */
4771         if (!smb_io_printer_info_2("", buffer, printer, 0))
4772                 result = WERR_NOMEM;
4773
4774 out:
4775         /* clear memory */
4776         free_printer_info_2(printer);
4777
4778         return result;
4779 }
4780
4781 /****************************************************************************
4782 ****************************************************************************/
4783
4784 static WERROR getprinter_level_3(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4785 {
4786         PRINTER_INFO_3 *printer=NULL;
4787         WERROR result = WERR_OK;
4788
4789         if (!construct_printer_info_3(print_hnd, &printer, snum))
4790                 return WERR_NOMEM;
4791
4792         /* check the required size. */
4793         *needed += spoolss_size_printer_info_3(printer);
4794
4795         if (*needed > offered) {
4796                 result = WERR_INSUFFICIENT_BUFFER;
4797                 goto out;
4798         }
4799
4800         if (!rpcbuf_alloc_size(buffer, *needed)) {
4801                 result = WERR_NOMEM;
4802                 goto out;
4803         }
4804
4805         /* fill the buffer with the structures */
4806         smb_io_printer_info_3("", buffer, printer, 0);
4807
4808 out:
4809         /* clear memory */
4810         free_printer_info_3(printer);
4811
4812         return result;
4813 }
4814
4815 /****************************************************************************
4816 ****************************************************************************/
4817
4818 static WERROR getprinter_level_4(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4819 {
4820         PRINTER_INFO_4 *printer=NULL;
4821         WERROR result = WERR_OK;
4822
4823         if((printer=SMB_MALLOC_P(PRINTER_INFO_4))==NULL)
4824                 return WERR_NOMEM;
4825
4826         if (!construct_printer_info_4(print_hnd, printer, snum)) {
4827                 SAFE_FREE(printer);
4828                 return WERR_NOMEM;
4829         }
4830
4831         /* check the required size. */
4832         *needed += spoolss_size_printer_info_4(printer);
4833
4834         if (*needed > offered) {
4835                 result = WERR_INSUFFICIENT_BUFFER;
4836                 goto out;
4837         }
4838
4839         if (!rpcbuf_alloc_size(buffer, *needed)) {
4840                 result = WERR_NOMEM;
4841                 goto out;
4842         }
4843
4844         /* fill the buffer with the structures */
4845         smb_io_printer_info_4("", buffer, printer, 0);
4846
4847 out:
4848         /* clear memory */
4849         free_printer_info_4(printer);
4850
4851         return result;
4852 }
4853
4854 /****************************************************************************
4855 ****************************************************************************/
4856
4857 static WERROR getprinter_level_5(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4858 {
4859         PRINTER_INFO_5 *printer=NULL;
4860         WERROR result = WERR_OK;
4861
4862         if((printer=SMB_MALLOC_P(PRINTER_INFO_5))==NULL)
4863                 return WERR_NOMEM;
4864
4865         if (!construct_printer_info_5(print_hnd, printer, snum))
4866                 return WERR_NOMEM;
4867
4868         /* check the required size. */
4869         *needed += spoolss_size_printer_info_5(printer);
4870
4871         if (*needed > offered) {
4872                 result = WERR_INSUFFICIENT_BUFFER;
4873                 goto out;
4874         }
4875
4876         if (!rpcbuf_alloc_size(buffer, *needed)) {
4877                 result = WERR_NOMEM;
4878                 goto out;
4879         }
4880
4881         /* fill the buffer with the structures */
4882         smb_io_printer_info_5("", buffer, printer, 0);
4883
4884 out:
4885         /* clear memory */
4886         free_printer_info_5(printer);
4887
4888         return result;
4889 }
4890
4891 static WERROR getprinter_level_7(Printer_entry *print_hnd, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
4892 {
4893         PRINTER_INFO_7 *printer=NULL;
4894         WERROR result = WERR_OK;
4895
4896         if((printer=SMB_MALLOC_P(PRINTER_INFO_7))==NULL)
4897                 return WERR_NOMEM;
4898
4899         if (!construct_printer_info_7(print_hnd, printer, snum))
4900                 return WERR_NOMEM;
4901
4902         /* check the required size. */
4903         *needed += spoolss_size_printer_info_7(printer);
4904
4905         if (*needed > offered) {
4906                 result = WERR_INSUFFICIENT_BUFFER;
4907                 goto out;
4908         }
4909
4910         if (!rpcbuf_alloc_size(buffer, *needed)) {
4911                 result = WERR_NOMEM;
4912                 goto out;
4913
4914         }
4915
4916         /* fill the buffer with the structures */
4917         smb_io_printer_info_7("", buffer, printer, 0);
4918
4919 out:
4920         /* clear memory */
4921         free_printer_info_7(printer);
4922
4923         return result;
4924 }
4925
4926 /****************************************************************************
4927 ****************************************************************************/
4928
4929 WERROR _spoolss_getprinter(pipes_struct *p, SPOOL_Q_GETPRINTER *q_u, SPOOL_R_GETPRINTER *r_u)
4930 {
4931         POLICY_HND *handle = &q_u->handle;
4932         uint32 level = q_u->level;
4933         RPC_BUFFER *buffer = NULL;
4934         uint32 offered = q_u->offered;
4935         uint32 *needed = &r_u->needed;
4936         Printer_entry *Printer=find_printer_index_by_hnd(p, handle);
4937
4938         int snum;
4939
4940         /* that's an [in out] buffer */
4941
4942         if (!q_u->buffer) {
4943                 return WERR_INVALID_PARAM;
4944         }
4945
4946         rpcbuf_move(q_u->buffer, &r_u->buffer);
4947         buffer = r_u->buffer;
4948
4949         *needed=0;
4950
4951         if (!get_printer_snum(p, handle, &snum))
4952                 return WERR_BADFID;
4953
4954         switch (level) {
4955         case 0:
4956                 return getprinter_level_0(Printer, snum, buffer, offered, needed);
4957         case 1:
4958                 return getprinter_level_1(Printer, snum, buffer, offered, needed);
4959         case 2:
4960                 return getprinter_level_2(Printer, snum, buffer, offered, needed);
4961         case 3:
4962                 return getprinter_level_3(Printer, snum, buffer, offered, needed);
4963         case 4:
4964                 return getprinter_level_4(Printer, snum, buffer, offered, needed);
4965         case 5:
4966                 return getprinter_level_5(Printer, snum, buffer, offered, needed);
4967         case 7:
4968                 return getprinter_level_7(Printer, snum, buffer, offered, needed);
4969         }
4970         return WERR_UNKNOWN_LEVEL;
4971 }
4972
4973 /********************************************************************
4974  * fill a DRIVER_INFO_1 struct
4975  ********************************************************************/
4976
4977 static void fill_printer_driver_info_1(DRIVER_INFO_1 *info, NT_PRINTER_DRIVER_INFO_LEVEL driver, fstring servername, fstring architecture)
4978 {
4979         init_unistr( &info->name, driver.info_3->name);
4980 }
4981
4982 /********************************************************************
4983  * construct_printer_driver_info_1
4984  ********************************************************************/
4985
4986 static WERROR construct_printer_driver_info_1(DRIVER_INFO_1 *info, int snum, fstring servername, fstring architecture, uint32 version)
4987 {
4988         NT_PRINTER_INFO_LEVEL *printer = NULL;
4989         NT_PRINTER_DRIVER_INFO_LEVEL driver;
4990
4991         ZERO_STRUCT(driver);
4992
4993         if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_const_servicename(snum))))
4994                 return WERR_INVALID_PRINTER_NAME;
4995
4996         if (!W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version)))
4997                 return WERR_UNKNOWN_PRINTER_DRIVER;
4998
4999         fill_printer_driver_info_1(info, driver, servername, architecture);
5000
5001         free_a_printer(&printer,2);
5002
5003         return WERR_OK;
5004 }
5005
5006 /********************************************************************
5007  * construct_printer_driver_info_2
5008  * fill a printer_info_2 struct
5009  ********************************************************************/
5010
5011 static void fill_printer_driver_info_2(DRIVER_INFO_2 *info, NT_PRINTER_DRIVER_INFO_LEVEL driver, fstring servername)
5012 {
5013         pstring temp;
5014
5015         info->version=driver.info_3->cversion;
5016
5017         init_unistr( &info->name, driver.info_3->name );
5018         init_unistr( &info->architecture, driver.info_3->environment );
5019
5020
5021     if (strlen(driver.info_3->driverpath)) {
5022                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->driverpath);
5023                 init_unistr( &info->driverpath, temp );
5024     } else
5025         init_unistr( &info->driverpath, "" );
5026
5027         if (strlen(driver.info_3->datafile)) {
5028                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->datafile);
5029                 init_unistr( &info->datafile, temp );
5030         } else
5031                 init_unistr( &info->datafile, "" );
5032
5033         if (strlen(driver.info_3->configfile)) {
5034                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->configfile);
5035                 init_unistr( &info->configfile, temp );
5036         } else
5037                 init_unistr( &info->configfile, "" );
5038 }
5039
5040 /********************************************************************
5041  * construct_printer_driver_info_2
5042  * fill a printer_info_2 struct
5043  ********************************************************************/
5044
5045 static WERROR construct_printer_driver_info_2(DRIVER_INFO_2 *info, int snum, fstring servername, fstring architecture, uint32 version)
5046 {
5047         NT_PRINTER_INFO_LEVEL *printer = NULL;
5048         NT_PRINTER_DRIVER_INFO_LEVEL driver;
5049
5050         ZERO_STRUCT(printer);
5051         ZERO_STRUCT(driver);
5052
5053         if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_const_servicename(snum))))
5054                 return WERR_INVALID_PRINTER_NAME;
5055
5056         if (!W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version)))
5057                 return WERR_UNKNOWN_PRINTER_DRIVER;
5058
5059         fill_printer_driver_info_2(info, driver, servername);
5060
5061         free_a_printer(&printer,2);
5062
5063         return WERR_OK;
5064 }
5065
5066 /********************************************************************
5067  * copy a strings array and convert to UNICODE
5068  *
5069  * convert an array of ascii string to a UNICODE string
5070  ********************************************************************/
5071
5072 static uint32 init_unistr_array(uint16 **uni_array, fstring *char_array, const char *servername)
5073 {
5074         int i=0;
5075         int j=0;
5076         const char *v;
5077         pstring line;
5078
5079         DEBUG(6,("init_unistr_array\n"));
5080         *uni_array=NULL;
5081
5082         while (True)
5083         {
5084                 if ( !char_array )
5085                         v = "";
5086                 else
5087                 {
5088                         v = char_array[i];
5089                         if (!v)
5090                                 v = ""; /* hack to handle null lists */
5091                 }
5092
5093                 /* hack to allow this to be used in places other than when generating
5094                    the list of dependent files */
5095
5096                 if ( servername )
5097                         slprintf( line, sizeof(line)-1, "\\\\%s%s", servername, v );
5098                 else
5099                         pstrcpy( line, v );
5100
5101                 DEBUGADD(6,("%d:%s:%lu\n", i, line, (unsigned long)strlen(line)));
5102
5103                 /* add one extra unit16 for the second terminating NULL */
5104
5105                 if ( (*uni_array=SMB_REALLOC_ARRAY(*uni_array, uint16, j+1+strlen(line)+2)) == NULL ) {
5106                         DEBUG(2,("init_unistr_array: Realloc error\n" ));
5107                         return 0;
5108                 }
5109
5110                 if ( !strlen(v) )
5111                         break;
5112
5113                 j += (rpcstr_push((*uni_array+j), line, sizeof(uint16)*strlen(line)+2, STR_TERMINATE) / sizeof(uint16));
5114                 i++;
5115         }
5116
5117         if (*uni_array) {
5118                 /* special case for ""; we need to add both NULL's here */
5119                 if (!j)
5120                         (*uni_array)[j++]=0x0000;
5121                 (*uni_array)[j]=0x0000;
5122         }
5123
5124         DEBUGADD(6,("last one:done\n"));
5125
5126         /* return size of array in uint16's */
5127
5128         return j+1;
5129 }
5130
5131 /********************************************************************
5132  * construct_printer_info_3
5133  * fill a printer_info_3 struct
5134  ********************************************************************/
5135
5136 static void fill_printer_driver_info_3(DRIVER_INFO_3 *info, NT_PRINTER_DRIVER_INFO_LEVEL driver, fstring servername)
5137 {
5138         pstring temp;
5139
5140         ZERO_STRUCTP(info);
5141
5142         info->version=driver.info_3->cversion;
5143
5144         init_unistr( &info->name, driver.info_3->name );
5145         init_unistr( &info->architecture, driver.info_3->environment );
5146
5147         if (strlen(driver.info_3->driverpath)) {
5148                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->driverpath);
5149                 init_unistr( &info->driverpath, temp );
5150         } else
5151                 init_unistr( &info->driverpath, "" );
5152
5153         if (strlen(driver.info_3->datafile)) {
5154                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->datafile);
5155                 init_unistr( &info->datafile, temp );
5156         } else
5157                 init_unistr( &info->datafile, "" );
5158
5159         if (strlen(driver.info_3->configfile)) {
5160                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->configfile);
5161                 init_unistr( &info->configfile, temp );
5162         } else
5163                 init_unistr( &info->configfile, "" );
5164
5165         if (strlen(driver.info_3->helpfile)) {
5166                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->helpfile);
5167                 init_unistr( &info->helpfile, temp );
5168         } else
5169                 init_unistr( &info->helpfile, "" );
5170
5171         init_unistr( &info->monitorname, driver.info_3->monitorname );
5172         init_unistr( &info->defaultdatatype, driver.info_3->defaultdatatype );
5173
5174         info->dependentfiles=NULL;
5175         init_unistr_array(&info->dependentfiles, driver.info_3->dependentfiles, servername);
5176 }
5177
5178 /********************************************************************
5179  * construct_printer_info_3
5180  * fill a printer_info_3 struct
5181  ********************************************************************/
5182
5183 static WERROR construct_printer_driver_info_3(DRIVER_INFO_3 *info, int snum, fstring servername, fstring architecture, uint32 version)
5184 {
5185         NT_PRINTER_INFO_LEVEL *printer = NULL;
5186         NT_PRINTER_DRIVER_INFO_LEVEL driver;
5187         WERROR status;
5188         ZERO_STRUCT(driver);
5189
5190         status=get_a_printer(NULL, &printer, 2, lp_const_servicename(snum) );
5191         DEBUG(8,("construct_printer_driver_info_3: status: %s\n", dos_errstr(status)));
5192         if (!W_ERROR_IS_OK(status))
5193                 return WERR_INVALID_PRINTER_NAME;
5194
5195         status=get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version);
5196         DEBUG(8,("construct_printer_driver_info_3: status: %s\n", dos_errstr(status)));
5197
5198 #if 0   /* JERRY */
5199
5200         /*
5201          * I put this code in during testing.  Helpful when commenting out the
5202          * support for DRIVER_INFO_6 in regards to win2k.  Not needed in general
5203          * as win2k always queries the driver using an infor level of 6.
5204          * I've left it in (but ifdef'd out) because I'll probably
5205          * use it in experimentation again in the future.   --jerry 22/01/2002
5206          */
5207
5208         if (!W_ERROR_IS_OK(status)) {
5209                 /*
5210                  * Is this a W2k client ?
5211                  */
5212                 if (version == 3) {
5213                         /* Yes - try again with a WinNT driver. */
5214                         version = 2;
5215                         status=get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version);
5216                         DEBUG(8,("construct_printer_driver_info_3: status: %s\n", dos_errstr(status)));
5217                 }
5218 #endif
5219
5220                 if (!W_ERROR_IS_OK(status)) {
5221                         free_a_printer(&printer,2);
5222                         return WERR_UNKNOWN_PRINTER_DRIVER;
5223                 }
5224
5225 #if 0   /* JERRY */
5226         }
5227 #endif
5228
5229
5230         fill_printer_driver_info_3(info, driver, servername);
5231
5232         free_a_printer(&printer,2);
5233
5234         return WERR_OK;
5235 }
5236
5237 /********************************************************************
5238  * construct_printer_info_6
5239  * fill a printer_info_6 struct - we know that driver is really level 3. This sucks. JRA.
5240  ********************************************************************/
5241
5242 static void fill_printer_driver_info_6(DRIVER_INFO_6 *info, NT_PRINTER_DRIVER_INFO_LEVEL driver, fstring servername)
5243 {
5244         pstring temp;
5245         fstring nullstr;
5246
5247         ZERO_STRUCTP(info);
5248         memset(&nullstr, '\0', sizeof(fstring));
5249
5250         info->version=driver.info_3->cversion;
5251
5252         init_unistr( &info->name, driver.info_3->name );
5253         init_unistr( &info->architecture, driver.info_3->environment );
5254
5255         if (strlen(driver.info_3->driverpath)) {
5256                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->driverpath);
5257                 init_unistr( &info->driverpath, temp );
5258         } else
5259                 init_unistr( &info->driverpath, "" );
5260
5261         if (strlen(driver.info_3->datafile)) {
5262                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->datafile);
5263                 init_unistr( &info->datafile, temp );
5264         } else
5265                 init_unistr( &info->datafile, "" );
5266
5267         if (strlen(driver.info_3->configfile)) {
5268                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->configfile);
5269                 init_unistr( &info->configfile, temp );
5270         } else
5271                 init_unistr( &info->configfile, "" );
5272
5273         if (strlen(driver.info_3->helpfile)) {
5274                 slprintf(temp, sizeof(temp)-1, "\\\\%s%s", servername, driver.info_3->helpfile);
5275                 init_unistr( &info->helpfile, temp );
5276         } else
5277                 init_unistr( &info->helpfile, "" );
5278
5279         init_unistr( &info->monitorname, driver.info_3->monitorname );
5280         init_unistr( &info->defaultdatatype, driver.info_3->defaultdatatype );
5281
5282         info->dependentfiles = NULL;
5283         init_unistr_array( &info->dependentfiles, driver.info_3->dependentfiles, servername );
5284
5285         info->previousdrivernames=NULL;
5286         init_unistr_array(&info->previousdrivernames, &nullstr, servername);
5287
5288         info->driver_date.low=0;
5289         info->driver_date.high=0;
5290
5291         info->padding=0;
5292         info->driver_version_low=0;
5293         info->driver_version_high=0;
5294
5295         init_unistr( &info->mfgname, "");
5296         init_unistr( &info->oem_url, "");
5297         init_unistr( &info->hardware_id, "");
5298         init_unistr( &info->provider, "");
5299 }
5300
5301 /********************************************************************
5302  * construct_printer_info_6
5303  * fill a printer_info_6 struct
5304  ********************************************************************/
5305
5306 static WERROR construct_printer_driver_info_6(DRIVER_INFO_6 *info, int snum,
5307               fstring servername, fstring architecture, uint32 version)
5308 {
5309         NT_PRINTER_INFO_LEVEL           *printer = NULL;
5310         NT_PRINTER_DRIVER_INFO_LEVEL    driver;
5311         WERROR                          status;
5312
5313         ZERO_STRUCT(driver);
5314
5315         status=get_a_printer(NULL, &printer, 2, lp_const_servicename(snum) );
5316
5317         DEBUG(8,("construct_printer_driver_info_6: status: %s\n", dos_errstr(status)));
5318
5319         if (!W_ERROR_IS_OK(status))
5320                 return WERR_INVALID_PRINTER_NAME;
5321
5322         status = get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version);
5323
5324         DEBUG(8,("construct_printer_driver_info_6: status: %s\n", dos_errstr(status)));
5325
5326         if (!W_ERROR_IS_OK(status))
5327         {
5328                 /*
5329                  * Is this a W2k client ?
5330                  */
5331
5332                 if (version < 3) {
5333                         free_a_printer(&printer,2);
5334                         return WERR_UNKNOWN_PRINTER_DRIVER;
5335                 }
5336
5337                 /* Yes - try again with a WinNT driver. */
5338                 version = 2;
5339                 status=get_a_printer_driver(&driver, 3, printer->info_2->drivername, architecture, version);
5340                 DEBUG(8,("construct_printer_driver_info_6: status: %s\n", dos_errstr(status)));
5341                 if (!W_ERROR_IS_OK(status)) {
5342                         free_a_printer(&printer,2);
5343                         return WERR_UNKNOWN_PRINTER_DRIVER;
5344                 }
5345         }
5346
5347         fill_printer_driver_info_6(info, driver, servername);
5348
5349         free_a_printer(&printer,2);
5350         free_a_printer_driver(driver, 3);
5351
5352         return WERR_OK;
5353 }
5354
5355 /****************************************************************************
5356 ****************************************************************************/
5357
5358 static void free_printer_driver_info_3(DRIVER_INFO_3 *info)
5359 {
5360         SAFE_FREE(info->dependentfiles);
5361 }
5362
5363 /****************************************************************************
5364 ****************************************************************************/
5365
5366 static void free_printer_driver_info_6(DRIVER_INFO_6 *info)
5367 {
5368         SAFE_FREE(info->dependentfiles);
5369 }
5370
5371 /****************************************************************************
5372 ****************************************************************************/
5373
5374 static WERROR getprinterdriver2_level1(fstring servername, fstring architecture, uint32 version, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
5375 {
5376         DRIVER_INFO_1 *info=NULL;
5377         WERROR result;
5378
5379         if((info=SMB_MALLOC_P(DRIVER_INFO_1)) == NULL)
5380                 return WERR_NOMEM;
5381
5382         result = construct_printer_driver_info_1(info, snum, servername, architecture, version);
5383         if (!W_ERROR_IS_OK(result))
5384                 goto out;
5385
5386         /* check the required size. */
5387         *needed += spoolss_size_printer_driver_info_1(info);
5388
5389         if (*needed > offered) {
5390                 result = WERR_INSUFFICIENT_BUFFER;
5391                 goto out;
5392         }
5393
5394         if (!rpcbuf_alloc_size(buffer, *needed)) {
5395                 result = WERR_NOMEM;
5396                 goto out;
5397         }
5398
5399         /* fill the buffer with the structures */
5400         smb_io_printer_driver_info_1("", buffer, info, 0);
5401
5402 out:
5403         /* clear memory */
5404         SAFE_FREE(info);
5405
5406         return result;
5407 }
5408
5409 /****************************************************************************
5410 ****************************************************************************/
5411
5412 static WERROR getprinterdriver2_level2(fstring servername, fstring architecture, uint32 version, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
5413 {
5414         DRIVER_INFO_2 *info=NULL;
5415         WERROR result;
5416
5417         if((info=SMB_MALLOC_P(DRIVER_INFO_2)) == NULL)
5418                 return WERR_NOMEM;
5419
5420         result = construct_printer_driver_info_2(info, snum, servername, architecture, version);
5421         if (!W_ERROR_IS_OK(result))
5422                 goto out;
5423
5424         /* check the required size. */
5425         *needed += spoolss_size_printer_driver_info_2(info);
5426
5427         if (*needed > offered) {
5428                 result = WERR_INSUFFICIENT_BUFFER;
5429                 goto out;
5430         }
5431
5432         if (!rpcbuf_alloc_size(buffer, *needed)) {
5433                 result = WERR_NOMEM;
5434                 goto out;
5435         }
5436
5437         /* fill the buffer with the structures */
5438         smb_io_printer_driver_info_2("", buffer, info, 0);
5439
5440 out:
5441         /* clear memory */
5442         SAFE_FREE(info);
5443
5444         return result;
5445 }
5446
5447 /****************************************************************************
5448 ****************************************************************************/
5449
5450 static WERROR getprinterdriver2_level3(fstring servername, fstring architecture, uint32 version, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
5451 {
5452         DRIVER_INFO_3 info;
5453         WERROR result;
5454
5455         ZERO_STRUCT(info);
5456
5457         result = construct_printer_driver_info_3(&info, snum, servername, architecture, version);
5458         if (!W_ERROR_IS_OK(result))
5459                 goto out;
5460
5461         /* check the required size. */
5462         *needed += spoolss_size_printer_driver_info_3(&info);
5463
5464         if (*needed > offered) {
5465                 result = WERR_INSUFFICIENT_BUFFER;
5466                 goto out;
5467         }
5468
5469         if (!rpcbuf_alloc_size(buffer, *needed)) {
5470                 result = WERR_NOMEM;
5471                 goto out;
5472         }
5473
5474         /* fill the buffer with the structures */
5475         smb_io_printer_driver_info_3("", buffer, &info, 0);
5476
5477 out:
5478         free_printer_driver_info_3(&info);
5479
5480         return result;
5481 }
5482
5483 /****************************************************************************
5484 ****************************************************************************/
5485
5486 static WERROR getprinterdriver2_level6(fstring servername, fstring architecture, uint32 version, int snum, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
5487 {
5488         DRIVER_INFO_6 info;
5489         WERROR result;
5490
5491         ZERO_STRUCT(info);
5492
5493         result = construct_printer_driver_info_6(&info, snum, servername, architecture, version);
5494         if (!W_ERROR_IS_OK(result))
5495                 goto out;
5496
5497         /* check the required size. */
5498         *needed += spoolss_size_printer_driver_info_6(&info);
5499
5500         if (*needed > offered) {
5501                 result = WERR_INSUFFICIENT_BUFFER;
5502                 goto out;
5503         }
5504
5505         if (!rpcbuf_alloc_size(buffer, *needed)) {
5506                 result = WERR_NOMEM;
5507                 goto out;
5508         }
5509
5510         /* fill the buffer with the structures */
5511         smb_io_printer_driver_info_6("", buffer, &info, 0);
5512
5513 out:
5514         free_printer_driver_info_6(&info);
5515
5516         return result;
5517 }
5518
5519 /****************************************************************************
5520 ****************************************************************************/
5521
5522 WERROR _spoolss_getprinterdriver2(pipes_struct *p, SPOOL_Q_GETPRINTERDRIVER2 *q_u, SPOOL_R_GETPRINTERDRIVER2 *r_u)
5523 {
5524         POLICY_HND *handle = &q_u->handle;
5525         UNISTR2 *uni_arch = &q_u->architecture;
5526         uint32 level = q_u->level;
5527         uint32 clientmajorversion = q_u->clientmajorversion;
5528         RPC_BUFFER *buffer = NULL;
5529         uint32 offered = q_u->offered;
5530         uint32 *needed = &r_u->needed;
5531         uint32 *servermajorversion = &r_u->servermajorversion;
5532         uint32 *serverminorversion = &r_u->serverminorversion;
5533         Printer_entry *printer;
5534
5535         fstring servername;
5536         fstring architecture;
5537         int snum;
5538
5539         /* that's an [in out] buffer */
5540
5541         if (!q_u->buffer) {
5542                 return WERR_INVALID_PARAM;
5543         }
5544
5545         rpcbuf_move(q_u->buffer, &r_u->buffer);
5546         buffer = r_u->buffer;
5547
5548         DEBUG(4,("_spoolss_getprinterdriver2\n"));
5549
5550         if ( !(printer = find_printer_index_by_hnd( p, handle )) ) {
5551                 DEBUG(0,("_spoolss_getprinterdriver2: invalid printer handle!\n"));
5552                 return WERR_INVALID_PRINTER_NAME;
5553         }
5554
5555         *needed = 0;
5556         *servermajorversion = 0;
5557         *serverminorversion = 0;
5558
5559         fstrcpy(servername, get_server_name( printer ));
5560         unistr2_to_ascii(architecture, uni_arch, sizeof(architecture)-1);
5561
5562         if (!get_printer_snum(p, handle, &snum))
5563                 return WERR_BADFID;
5564
5565         switch (level) {
5566         case 1:
5567                 return getprinterdriver2_level1(servername, architecture, clientmajorversion, snum, buffer, offered, needed);
5568         case 2:
5569                 return getprinterdriver2_level2(servername, architecture, clientmajorversion, snum, buffer, offered, needed);
5570         case 3:
5571                 return getprinterdriver2_level3(servername, architecture, clientmajorversion, snum, buffer, offered, needed);
5572         case 6:
5573                 return getprinterdriver2_level6(servername, architecture, clientmajorversion, snum, buffer, offered, needed);
5574 #if 0   /* JERRY */
5575         case 101:
5576                 /* apparently this call is the equivalent of
5577                    EnumPrinterDataEx() for the DsDriver key */
5578                 break;
5579 #endif
5580         }
5581
5582         return WERR_UNKNOWN_LEVEL;
5583 }
5584
5585 /****************************************************************************
5586 ****************************************************************************/
5587
5588 WERROR _spoolss_startpageprinter(pipes_struct *p, SPOOL_Q_STARTPAGEPRINTER *q_u, SPOOL_R_STARTPAGEPRINTER *r_u)
5589 {
5590         POLICY_HND *handle = &q_u->handle;
5591
5592         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5593
5594         if (!Printer) {
5595                 DEBUG(3,("Error in startpageprinter printer handle\n"));
5596                 return WERR_BADFID;
5597         }
5598
5599         Printer->page_started=True;
5600         return WERR_OK;
5601 }
5602
5603 /****************************************************************************
5604 ****************************************************************************/
5605
5606 WERROR _spoolss_endpageprinter(pipes_struct *p, SPOOL_Q_ENDPAGEPRINTER *q_u, SPOOL_R_ENDPAGEPRINTER *r_u)
5607 {
5608         POLICY_HND *handle = &q_u->handle;
5609         int snum;
5610
5611         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5612
5613         if (!Printer) {
5614                 DEBUG(2,("_spoolss_endpageprinter: Invalid handle (%s:%u:%u).\n",OUR_HANDLE(handle)));
5615                 return WERR_BADFID;
5616         }
5617
5618         if (!get_printer_snum(p, handle, &snum))
5619                 return WERR_BADFID;
5620
5621         Printer->page_started=False;
5622         print_job_endpage(snum, Printer->jobid);
5623
5624         return WERR_OK;
5625 }
5626
5627 /********************************************************************
5628  * api_spoolss_getprinter
5629  * called from the spoolss dispatcher
5630  *
5631  ********************************************************************/
5632
5633 WERROR _spoolss_startdocprinter(pipes_struct *p, SPOOL_Q_STARTDOCPRINTER *q_u, SPOOL_R_STARTDOCPRINTER *r_u)
5634 {
5635         POLICY_HND *handle = &q_u->handle;
5636         DOC_INFO *docinfo = &q_u->doc_info_container.docinfo;
5637         uint32 *jobid = &r_u->jobid;
5638
5639         DOC_INFO_1 *info_1 = &docinfo->doc_info_1;
5640         int snum;
5641         pstring jobname;
5642         fstring datatype;
5643         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5644         struct current_user user;
5645
5646         if (!Printer) {
5647                 DEBUG(2,("_spoolss_startdocprinter: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(handle)));
5648                 return WERR_BADFID;
5649         }
5650
5651         get_current_user(&user, p);
5652
5653         /*
5654          * a nice thing with NT is it doesn't listen to what you tell it.
5655          * when asked to send _only_ RAW datas, it tries to send datas
5656          * in EMF format.
5657          *
5658          * So I add checks like in NT Server ...
5659          */
5660
5661         if (info_1->p_datatype != 0) {
5662                 unistr2_to_ascii(datatype, &info_1->datatype, sizeof(datatype));
5663                 if (strcmp(datatype, "RAW") != 0) {
5664                         (*jobid)=0;
5665                         return WERR_INVALID_DATATYPE;
5666                 }
5667         }
5668
5669         /* get the share number of the printer */
5670         if (!get_printer_snum(p, handle, &snum)) {
5671                 return WERR_BADFID;
5672         }
5673
5674         unistr2_to_ascii(jobname, &info_1->docname, sizeof(jobname));
5675
5676         Printer->jobid = print_job_start(&user, snum, jobname, Printer->nt_devmode);
5677
5678         /* An error occured in print_job_start() so return an appropriate
5679            NT error code. */
5680
5681         if (Printer->jobid == -1) {
5682                 return map_werror_from_unix(errno);
5683         }
5684
5685         Printer->document_started=True;
5686         (*jobid) = Printer->jobid;
5687
5688         return WERR_OK;
5689 }
5690
5691 /********************************************************************
5692  * api_spoolss_getprinter
5693  * called from the spoolss dispatcher
5694  *
5695  ********************************************************************/
5696
5697 WERROR _spoolss_enddocprinter(pipes_struct *p, SPOOL_Q_ENDDOCPRINTER *q_u, SPOOL_R_ENDDOCPRINTER *r_u)
5698 {
5699         POLICY_HND *handle = &q_u->handle;
5700
5701         return _spoolss_enddocprinter_internal(p, handle);
5702 }
5703
5704 /****************************************************************************
5705 ****************************************************************************/
5706
5707 WERROR _spoolss_writeprinter(pipes_struct *p, SPOOL_Q_WRITEPRINTER *q_u, SPOOL_R_WRITEPRINTER *r_u)
5708 {
5709         POLICY_HND *handle = &q_u->handle;
5710         uint32 buffer_size = q_u->buffer_size;
5711         uint8 *buffer = q_u->buffer;
5712         uint32 *buffer_written = &q_u->buffer_size2;
5713         int snum;
5714         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5715
5716         if (!Printer) {
5717                 DEBUG(2,("_spoolss_writeprinter: Invalid handle (%s:%u:%u)\n",OUR_HANDLE(handle)));
5718                 r_u->buffer_written = q_u->buffer_size2;
5719                 return WERR_BADFID;
5720         }
5721
5722         if (!get_printer_snum(p, handle, &snum))
5723                 return WERR_BADFID;
5724
5725         (*buffer_written) = (uint32)print_job_write(snum, Printer->jobid, (const char *)buffer,
5726                                         (SMB_OFF_T)-1, (size_t)buffer_size);
5727         if (*buffer_written == (uint32)-1) {
5728                 r_u->buffer_written = 0;
5729                 if (errno == ENOSPC)
5730                         return WERR_NO_SPOOL_SPACE;
5731                 else
5732                         return WERR_ACCESS_DENIED;
5733         }
5734
5735         r_u->buffer_written = q_u->buffer_size2;
5736
5737         return WERR_OK;
5738 }
5739
5740 /********************************************************************
5741  * api_spoolss_getprinter
5742  * called from the spoolss dispatcher
5743  *
5744  ********************************************************************/
5745
5746 static WERROR control_printer(POLICY_HND *handle, uint32 command,
5747                               pipes_struct *p)
5748 {
5749         struct current_user user;
5750         int snum;
5751         WERROR errcode = WERR_BADFUNC;
5752         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5753
5754         get_current_user(&user, p);
5755
5756         if (!Printer) {
5757                 DEBUG(2,("control_printer: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(handle)));
5758                 return WERR_BADFID;
5759         }
5760
5761         if (!get_printer_snum(p, handle, &snum))
5762                 return WERR_BADFID;
5763
5764         switch (command) {
5765         case PRINTER_CONTROL_PAUSE:
5766                 if (print_queue_pause(&user, snum, &errcode)) {
5767                         errcode = WERR_OK;
5768                 }
5769                 break;
5770         case PRINTER_CONTROL_RESUME:
5771         case PRINTER_CONTROL_UNPAUSE:
5772                 if (print_queue_resume(&user, snum, &errcode)) {
5773                         errcode = WERR_OK;
5774                 }
5775                 break;
5776         case PRINTER_CONTROL_PURGE:
5777                 if (print_queue_purge(&user, snum, &errcode)) {
5778                         errcode = WERR_OK;
5779                 }
5780                 break;
5781         default:
5782                 return WERR_UNKNOWN_LEVEL;
5783         }
5784
5785         return errcode;
5786 }
5787
5788 /********************************************************************
5789  * api_spoolss_abortprinter
5790  * From MSDN: "Deletes printer's spool file if printer is configured
5791  * for spooling"
5792  ********************************************************************/
5793
5794 WERROR _spoolss_abortprinter(pipes_struct *p, SPOOL_Q_ABORTPRINTER *q_u, SPOOL_R_ABORTPRINTER *r_u)
5795 {
5796         POLICY_HND      *handle = &q_u->handle;
5797         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
5798         int             snum;
5799         struct          current_user user;
5800         WERROR          errcode = WERR_OK;
5801
5802         if (!Printer) {
5803                 DEBUG(2,("_spoolss_abortprinter: Invalid handle (%s:%u:%u)\n",OUR_HANDLE(handle)));
5804                 return WERR_BADFID;
5805         }
5806
5807         if (!get_printer_snum(p, handle, &snum))
5808                 return WERR_BADFID;
5809
5810         get_current_user( &user, p );
5811
5812         print_job_delete( &user, snum, Printer->jobid, &errcode );
5813
5814         return errcode;
5815 }
5816
5817 /********************************************************************
5818  * called by spoolss_api_setprinter
5819  * when updating a printer description
5820  ********************************************************************/
5821
5822 static WERROR update_printer_sec(POLICY_HND *handle, uint32 level,
5823                                  const SPOOL_PRINTER_INFO_LEVEL *info,
5824                                  pipes_struct *p, SEC_DESC_BUF *secdesc_ctr)
5825 {
5826         SEC_DESC_BUF *new_secdesc_ctr = NULL, *old_secdesc_ctr = NULL;
5827         WERROR result;
5828         int snum;
5829
5830         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
5831
5832         if (!Printer || !get_printer_snum(p, handle, &snum)) {
5833                 DEBUG(2,("update_printer_sec: Invalid handle (%s:%u:%u)\n",
5834                          OUR_HANDLE(handle)));
5835
5836                 result = WERR_BADFID;
5837                 goto done;
5838         }
5839
5840         /* Check the user has permissions to change the security
5841            descriptor.  By experimentation with two NT machines, the user
5842            requires Full Access to the printer to change security
5843            information. */
5844
5845         if ( Printer->access_granted != PRINTER_ACCESS_ADMINISTER ) {
5846                 DEBUG(4,("update_printer_sec: updated denied by printer permissions\n"));
5847                 result = WERR_ACCESS_DENIED;
5848                 goto done;
5849         }
5850
5851         /* NT seems to like setting the security descriptor even though
5852            nothing may have actually changed. */
5853
5854         nt_printing_getsec(p->mem_ctx, Printer->sharename, &old_secdesc_ctr);
5855
5856         if (DEBUGLEVEL >= 10) {
5857                 SEC_ACL *the_acl;
5858                 int i;
5859
5860                 the_acl = old_secdesc_ctr->sec->dacl;
5861                 DEBUG(10, ("old_secdesc_ctr for %s has %d aces:\n",
5862                            PRINTERNAME(snum), the_acl->num_aces));
5863
5864                 for (i = 0; i < the_acl->num_aces; i++) {
5865                         fstring sid_str;
5866
5867                         sid_to_string(sid_str, &the_acl->ace[i].trustee);
5868
5869                         DEBUG(10, ("%s 0x%08x\n", sid_str,
5870                                   the_acl->ace[i].info.mask));
5871                 }
5872
5873                 the_acl = secdesc_ctr->sec->dacl;
5874
5875                 if (the_acl) {
5876                         DEBUG(10, ("secdesc_ctr for %s has %d aces:\n",
5877                                    PRINTERNAME(snum), the_acl->num_aces));
5878
5879                         for (i = 0; i < the_acl->num_aces; i++) {
5880                                 fstring sid_str;
5881
5882                                 sid_to_string(sid_str, &the_acl->ace[i].trustee);
5883
5884                                 DEBUG(10, ("%s 0x%08x\n", sid_str,
5885                                            the_acl->ace[i].info.mask));
5886                         }
5887                 } else {
5888                         DEBUG(10, ("dacl for secdesc_ctr is NULL\n"));
5889                 }
5890         }
5891
5892         new_secdesc_ctr = sec_desc_merge(p->mem_ctx, secdesc_ctr, old_secdesc_ctr);
5893
5894         if (sec_desc_equal(new_secdesc_ctr->sec, old_secdesc_ctr->sec)) {
5895                 result = WERR_OK;
5896                 goto done;
5897         }
5898
5899         result = nt_printing_setsec(Printer->sharename, new_secdesc_ctr);
5900
5901  done:
5902
5903         return result;
5904 }
5905
5906 /********************************************************************
5907  Canonicalize printer info from a client
5908
5909  ATTN: It does not matter what we set the servername to hear
5910  since we do the necessary work in get_a_printer() to set it to
5911  the correct value based on what the client sent in the
5912  _spoolss_open_printer_ex().
5913  ********************************************************************/
5914
5915 static BOOL check_printer_ok(NT_PRINTER_INFO_LEVEL_2 *info, int snum)
5916 {
5917         fstring printername;
5918         const char *p;
5919
5920         DEBUG(5,("check_printer_ok: servername=%s printername=%s sharename=%s "
5921                 "portname=%s drivername=%s comment=%s location=%s\n",
5922                 info->servername, info->printername, info->sharename,
5923                 info->portname, info->drivername, info->comment, info->location));
5924
5925         /* we force some elements to "correct" values */
5926         slprintf(info->servername, sizeof(info->servername)-1, "\\\\%s", global_myname());
5927         fstrcpy(info->sharename, lp_servicename(snum));
5928
5929         /* check to see if we allow printername != sharename */
5930
5931         if ( lp_force_printername(snum) ) {
5932                 slprintf(info->printername, sizeof(info->printername)-1, "\\\\%s\\%s",
5933                         global_myname(), info->sharename );
5934         } else {
5935
5936                 /* make sure printername is in \\server\printername format */
5937
5938                 fstrcpy( printername, info->printername );
5939                 p = printername;
5940                 if ( printername[0] == '\\' && printername[1] == '\\' ) {
5941                         if ( (p = strchr_m( &printername[2], '\\' )) != NULL )
5942                                 p++;
5943                 }
5944
5945                 slprintf(info->printername, sizeof(info->printername)-1, "\\\\%s\\%s",
5946                          global_myname(), p );
5947         }
5948
5949         info->attributes |= PRINTER_ATTRIBUTE_SAMBA;
5950         info->attributes &= ~PRINTER_ATTRIBUTE_NOT_SAMBA;
5951
5952
5953
5954         return True;
5955 }
5956
5957 /****************************************************************************
5958 ****************************************************************************/
5959
5960 WERROR add_port_hook(NT_USER_TOKEN *token, const char *portname, const char *uri )
5961 {
5962         char *cmd = lp_addport_cmd();
5963         pstring command;
5964         int ret;
5965         int fd;
5966         SE_PRIV se_printop = SE_PRINT_OPERATOR;
5967         BOOL is_print_op = False;
5968
5969         if ( !*cmd ) {
5970                 return WERR_ACCESS_DENIED;
5971         }
5972
5973         slprintf(command, sizeof(command)-1, "%s \"%s\" \"%s\"", cmd, portname, uri );
5974
5975         if ( token )
5976                 is_print_op = user_has_privileges( token, &se_printop );
5977
5978         DEBUG(10,("Running [%s]\n", command));
5979
5980         /********* BEGIN SePrintOperatorPrivilege **********/
5981
5982         if ( is_print_op )
5983                 become_root();
5984
5985         ret = smbrun(command, &fd);
5986
5987         if ( is_print_op )
5988                 unbecome_root();
5989
5990         /********* END SePrintOperatorPrivilege **********/
5991
5992         DEBUGADD(10,("returned [%d]\n", ret));
5993
5994         if ( ret != 0 ) {
5995                 if (fd != -1)
5996                         close(fd);
5997                 return WERR_ACCESS_DENIED;
5998         }
5999
6000         return WERR_OK;
6001 }
6002
6003 /****************************************************************************
6004 ****************************************************************************/
6005
6006 BOOL add_printer_hook(NT_USER_TOKEN *token, NT_PRINTER_INFO_LEVEL *printer)
6007 {
6008         char *cmd = lp_addprinter_cmd();
6009         char **qlines;
6010         pstring command;
6011         int numlines;
6012         int ret;
6013         int fd;
6014         fstring remote_machine = "%m";
6015         SE_PRIV se_printop = SE_PRINT_OPERATOR;
6016         BOOL is_print_op = False;
6017
6018         standard_sub_basic(current_user_info.smb_name, remote_machine,sizeof(remote_machine));
6019
6020         slprintf(command, sizeof(command)-1, "%s \"%s\" \"%s\" \"%s\" \"%s\" \"%s\" \"%s\" \"%s\"",
6021                         cmd, printer->info_2->printername, printer->info_2->sharename,
6022                         printer->info_2->portname, printer->info_2->drivername,
6023                         printer->info_2->location, printer->info_2->comment, remote_machine);
6024
6025         if ( token )
6026                 is_print_op = user_has_privileges( token, &se_printop );
6027
6028         DEBUG(10,("Running [%s]\n", command));
6029
6030         /********* BEGIN SePrintOperatorPrivilege **********/
6031
6032         if ( is_print_op )
6033                 become_root();
6034
6035         if ( (ret = smbrun(command, &fd)) == 0 ) {
6036                 /* Tell everyone we updated smb.conf. */
6037                 message_send_all(conn_tdb_ctx(), MSG_SMB_CONF_UPDATED, NULL, 0, False, NULL);
6038         }
6039
6040         if ( is_print_op )
6041                 unbecome_root();
6042
6043         /********* END SePrintOperatorPrivilege **********/
6044
6045         DEBUGADD(10,("returned [%d]\n", ret));
6046
6047         if ( ret != 0 ) {
6048                 if (fd != -1)
6049                         close(fd);
6050                 return False;
6051         }
6052
6053         /* reload our services immediately */
6054         reload_services( False );
6055
6056         numlines = 0;
6057         /* Get lines and convert them back to dos-codepage */
6058         qlines = fd_lines_load(fd, &numlines, 0);
6059         DEBUGADD(10,("Lines returned = [%d]\n", numlines));
6060         close(fd);
6061
6062         /* Set the portname to what the script says the portname should be. */
6063         /* but don't require anything to be return from the script exit a good error code */
6064
6065         if (numlines) {
6066                 /* Set the portname to what the script says the portname should be. */
6067                 strncpy(printer->info_2->portname, qlines[0], sizeof(printer->info_2->portname));
6068                 DEBUGADD(6,("Line[0] = [%s]\n", qlines[0]));
6069         }
6070
6071         file_lines_free(qlines);
6072         return True;
6073 }
6074
6075
6076 /********************************************************************
6077  * Called by spoolss_api_setprinter
6078  * when updating a printer description.
6079  ********************************************************************/
6080
6081 static WERROR update_printer(pipes_struct *p, POLICY_HND *handle, uint32 level,
6082                            const SPOOL_PRINTER_INFO_LEVEL *info,
6083                            DEVICEMODE *devmode)
6084 {
6085         int snum;
6086         NT_PRINTER_INFO_LEVEL *printer = NULL, *old_printer = NULL;
6087         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
6088         WERROR result;
6089         UNISTR2 buffer;
6090         fstring asc_buffer;
6091
6092         DEBUG(8,("update_printer\n"));
6093
6094         result = WERR_OK;
6095
6096         if (!Printer) {
6097                 result = WERR_BADFID;
6098                 goto done;
6099         }
6100
6101         if (!get_printer_snum(p, handle, &snum)) {
6102                 result = WERR_BADFID;
6103                 goto done;
6104         }
6105
6106         if (!W_ERROR_IS_OK(get_a_printer(Printer, &printer, 2, lp_const_servicename(snum))) ||
6107             (!W_ERROR_IS_OK(get_a_printer(Printer, &old_printer, 2, lp_const_servicename(snum))))) {
6108                 result = WERR_BADFID;
6109                 goto done;
6110         }
6111
6112         DEBUGADD(8,("Converting info_2 struct\n"));
6113
6114         /*
6115          * convert_printer_info converts the incoming
6116          * info from the client and overwrites the info
6117          * just read from the tdb in the pointer 'printer'.
6118          */
6119
6120         if (!convert_printer_info(info, printer, level)) {
6121                 result =  WERR_NOMEM;
6122                 goto done;
6123         }
6124
6125         if (devmode) {
6126                 /* we have a valid devmode
6127                    convert it and link it*/
6128
6129                 DEBUGADD(8,("update_printer: Converting the devicemode struct\n"));
6130                 if (!convert_devicemode(printer->info_2->printername, devmode,
6131                                 &printer->info_2->devmode)) {
6132                         result =  WERR_NOMEM;
6133                         goto done;
6134                 }
6135         }
6136
6137         /* Do sanity check on the requested changes for Samba */
6138
6139         if (!check_printer_ok(printer->info_2, snum)) {
6140                 result = WERR_INVALID_PARAM;
6141                 goto done;
6142         }
6143
6144         /* FIXME!!! If the driver has changed we really should verify that
6145            it is installed before doing much else   --jerry */
6146
6147         /* Check calling user has permission to update printer description */
6148
6149         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER) {
6150                 DEBUG(3, ("update_printer: printer property change denied by handle\n"));
6151                 result = WERR_ACCESS_DENIED;
6152                 goto done;
6153         }
6154
6155         /* Call addprinter hook */
6156         /* Check changes to see if this is really needed */
6157
6158         if ( *lp_addprinter_cmd()
6159                 && (!strequal(printer->info_2->drivername, old_printer->info_2->drivername)
6160                         || !strequal(printer->info_2->comment, old_printer->info_2->comment)
6161                         || !strequal(printer->info_2->portname, old_printer->info_2->portname)
6162                         || !strequal(printer->info_2->location, old_printer->info_2->location)) )
6163         {
6164                 /* add_printer_hook() will call reload_services() */
6165
6166                 if ( !add_printer_hook(p->pipe_user.nt_user_token, printer) ) {
6167                         result = WERR_ACCESS_DENIED;
6168                         goto done;
6169                 }
6170         }
6171
6172         /*
6173          * When a *new* driver is bound to a printer, the drivername is used to
6174          * lookup previously saved driver initialization info, which is then
6175          * bound to the printer, simulating what happens in the Windows arch.
6176          */
6177         if (!strequal(printer->info_2->drivername, old_printer->info_2->drivername))
6178         {
6179                 if (!set_driver_init(printer, 2))
6180                 {
6181                         DEBUG(5,("update_printer: Error restoring driver initialization data for driver [%s]!\n",
6182                                 printer->info_2->drivername));
6183                 }
6184
6185                 DEBUG(10,("update_printer: changing driver [%s]!  Sending event!\n",
6186                         printer->info_2->drivername));
6187
6188                 notify_printer_driver(snum, printer->info_2->drivername);
6189         }
6190
6191         /*
6192          * flag which changes actually occured.  This is a small subset of
6193          * all the possible changes.  We also have to update things in the
6194          * DsSpooler key.
6195          */
6196
6197         if (!strequal(printer->info_2->comment, old_printer->info_2->comment)) {
6198                 init_unistr2( &buffer, printer->info_2->comment, UNI_STR_TERMINATE);
6199                 set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "description",
6200                         REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6201
6202                 notify_printer_comment(snum, printer->info_2->comment);
6203         }
6204
6205         if (!strequal(printer->info_2->sharename, old_printer->info_2->sharename)) {
6206                 init_unistr2( &buffer, printer->info_2->sharename, UNI_STR_TERMINATE);
6207                 set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "shareName",
6208                         REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6209
6210                 notify_printer_sharename(snum, printer->info_2->sharename);
6211         }
6212
6213         if (!strequal(printer->info_2->printername, old_printer->info_2->printername)) {
6214                 char *pname;
6215
6216                 if ( (pname = strchr_m( printer->info_2->printername+2, '\\' )) != NULL )
6217                         pname++;
6218                 else
6219                         pname = printer->info_2->printername;
6220
6221
6222                 init_unistr2( &buffer, pname, UNI_STR_TERMINATE);
6223                 set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "printerName",
6224                         REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6225
6226                 notify_printer_printername( snum, pname );
6227         }
6228
6229         if (!strequal(printer->info_2->portname, old_printer->info_2->portname)) {
6230                 init_unistr2( &buffer, printer->info_2->portname, UNI_STR_TERMINATE);
6231                 set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "portName",
6232                         REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6233
6234                 notify_printer_port(snum, printer->info_2->portname);
6235         }
6236
6237         if (!strequal(printer->info_2->location, old_printer->info_2->location)) {
6238                 init_unistr2( &buffer, printer->info_2->location, UNI_STR_TERMINATE);
6239                 set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "location",
6240                         REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6241
6242                 notify_printer_location(snum, printer->info_2->location);
6243         }
6244
6245         /* here we need to update some more DsSpooler keys */
6246         /* uNCName, serverName, shortServerName */
6247
6248         init_unistr2( &buffer, global_myname(), UNI_STR_TERMINATE);
6249         set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "serverName",
6250                 REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6251         set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "shortServerName",
6252                 REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6253
6254         slprintf( asc_buffer, sizeof(asc_buffer)-1, "\\\\%s\\%s",
6255                  global_myname(), printer->info_2->sharename );
6256         init_unistr2( &buffer, asc_buffer, UNI_STR_TERMINATE);
6257         set_printer_dataex( printer, SPOOL_DSSPOOLER_KEY, "uNCName",
6258                 REG_SZ, (uint8*)buffer.buffer, buffer.uni_str_len*2 );
6259
6260         /* Update printer info */
6261         result = mod_a_printer(printer, 2);
6262
6263 done:
6264         free_a_printer(&printer, 2);
6265         free_a_printer(&old_printer, 2);
6266
6267
6268         return result;
6269 }
6270
6271 /****************************************************************************
6272 ****************************************************************************/
6273 static WERROR publish_or_unpublish_printer(pipes_struct *p, POLICY_HND *handle,
6274                                    const SPOOL_PRINTER_INFO_LEVEL *info)
6275 {
6276 #ifdef HAVE_ADS
6277         SPOOL_PRINTER_INFO_LEVEL_7 *info7 = info->info_7;
6278         int snum;
6279         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
6280
6281         DEBUG(5,("publish_or_unpublish_printer, action = %d\n",info7->action));
6282
6283         if (!Printer)
6284                 return WERR_BADFID;
6285
6286         if (!get_printer_snum(p, handle, &snum))
6287                 return WERR_BADFID;
6288
6289         nt_printer_publish(Printer, snum, info7->action);
6290
6291         return WERR_OK;
6292 #else
6293         return WERR_UNKNOWN_LEVEL;
6294 #endif
6295 }
6296 /****************************************************************************
6297 ****************************************************************************/
6298
6299 WERROR _spoolss_setprinter(pipes_struct *p, SPOOL_Q_SETPRINTER *q_u, SPOOL_R_SETPRINTER *r_u)
6300 {
6301         POLICY_HND *handle = &q_u->handle;
6302         uint32 level = q_u->level;
6303         SPOOL_PRINTER_INFO_LEVEL *info = &q_u->info;
6304         DEVMODE_CTR devmode_ctr = q_u->devmode_ctr;
6305         SEC_DESC_BUF *secdesc_ctr = q_u->secdesc_ctr;
6306         uint32 command = q_u->command;
6307         WERROR result;
6308
6309         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
6310
6311         if (!Printer) {
6312                 DEBUG(2,("_spoolss_setprinter: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(handle)));
6313                 return WERR_BADFID;
6314         }
6315
6316         /* check the level */
6317         switch (level) {
6318                 case 0:
6319                         return control_printer(handle, command, p);
6320                 case 2:
6321                         result = update_printer(p, handle, level, info, devmode_ctr.devmode);
6322                         if (!W_ERROR_IS_OK(result))
6323                                 return result;
6324                         if (secdesc_ctr)
6325                                 result = update_printer_sec(handle, level, info, p, secdesc_ctr);
6326                         return result;
6327                 case 3:
6328                         return update_printer_sec(handle, level, info, p,
6329                                                   secdesc_ctr);
6330                 case 7:
6331                         return publish_or_unpublish_printer(p, handle, info);
6332                 default:
6333                         return WERR_UNKNOWN_LEVEL;
6334         }
6335 }
6336
6337 /****************************************************************************
6338 ****************************************************************************/
6339
6340 WERROR _spoolss_fcpn(pipes_struct *p, SPOOL_Q_FCPN *q_u, SPOOL_R_FCPN *r_u)
6341 {
6342         POLICY_HND *handle = &q_u->handle;
6343         Printer_entry *Printer= find_printer_index_by_hnd(p, handle);
6344
6345         if (!Printer) {
6346                 DEBUG(2,("_spoolss_fcpn: Invalid handle (%s:%u:%u)\n", OUR_HANDLE(handle)));
6347                 return WERR_BADFID;
6348         }
6349
6350         if (Printer->notify.client_connected==True) {
6351                 int snum = -1;
6352
6353                 if ( Printer->printer_type == SPLHND_SERVER)
6354                         snum = -1;
6355                 else if ( (Printer->printer_type == SPLHND_PRINTER) &&
6356                                 !get_printer_snum(p, handle, &snum) )
6357                         return WERR_BADFID;
6358
6359                 srv_spoolss_replycloseprinter(snum, &Printer->notify.client_hnd);
6360         }
6361
6362         Printer->notify.flags=0;
6363         Printer->notify.options=0;
6364         Printer->notify.localmachine[0]='\0';
6365         Printer->notify.printerlocal=0;
6366         if (Printer->notify.option)
6367                 free_spool_notify_option(&Printer->notify.option);
6368         Printer->notify.client_connected=False;
6369
6370         return WERR_OK;
6371 }
6372
6373 /****************************************************************************
6374 ****************************************************************************/
6375
6376 WERROR _spoolss_addjob(pipes_struct *p, SPOOL_Q_ADDJOB *q_u, SPOOL_R_ADDJOB *r_u)
6377 {
6378         /* that's an [in out] buffer */
6379
6380         if (!q_u->buffer) {
6381                 return WERR_INVALID_PARAM;
6382         }
6383
6384         rpcbuf_move(q_u->buffer, &r_u->buffer);
6385
6386         r_u->needed = 0;
6387         return WERR_INVALID_PARAM; /* this is what a NT server
6388                                            returns for AddJob. AddJob
6389                                            must fail on non-local
6390                                            printers */
6391 }
6392
6393 /****************************************************************************
6394 ****************************************************************************/
6395
6396 static void fill_job_info_1(JOB_INFO_1 *job_info, print_queue_struct *queue,
6397                             int position, int snum,
6398                             NT_PRINTER_INFO_LEVEL *ntprinter)
6399 {
6400         struct tm *t;
6401
6402         t=gmtime(&queue->time);
6403
6404         job_info->jobid=queue->job;
6405         init_unistr(&job_info->printername, lp_servicename(snum));
6406         init_unistr(&job_info->machinename, ntprinter->info_2->servername);
6407         init_unistr(&job_info->username, queue->fs_user);
6408         init_unistr(&job_info->document, queue->fs_file);
6409         init_unistr(&job_info->datatype, "RAW");
6410         init_unistr(&job_info->text_status, "");
6411         job_info->status=nt_printj_status(queue->status);
6412         job_info->priority=queue->priority;
6413         job_info->position=position;
6414         job_info->totalpages=queue->page_count;
6415         job_info->pagesprinted=0;
6416
6417         make_systemtime(&job_info->submitted, t);
6418 }
6419
6420 /****************************************************************************
6421 ****************************************************************************/
6422
6423 static BOOL fill_job_info_2(JOB_INFO_2 *job_info, print_queue_struct *queue,
6424                             int position, int snum,
6425                             NT_PRINTER_INFO_LEVEL *ntprinter,
6426                             DEVICEMODE *devmode)
6427 {
6428         struct tm *t;
6429
6430         t=gmtime(&queue->time);
6431
6432         job_info->jobid=queue->job;
6433
6434         init_unistr(&job_info->printername, ntprinter->info_2->printername);
6435
6436         init_unistr(&job_info->machinename, ntprinter->info_2->servername);
6437         init_unistr(&job_info->username, queue->fs_user);
6438         init_unistr(&job_info->document, queue->fs_file);
6439         init_unistr(&job_info->notifyname, queue->fs_user);
6440         init_unistr(&job_info->datatype, "RAW");
6441         init_unistr(&job_info->printprocessor, "winprint");
6442         init_unistr(&job_info->parameters, "");
6443         init_unistr(&job_info->drivername, ntprinter->info_2->drivername);
6444         init_unistr(&job_info->text_status, "");
6445
6446 /* and here the security descriptor */
6447
6448         job_info->status=nt_printj_status(queue->status);
6449         job_info->priority=queue->priority;
6450         job_info->position=position;
6451         job_info->starttime=0;
6452         job_info->untiltime=0;
6453         job_info->totalpages=queue->page_count;
6454         job_info->size=queue->size;
6455         make_systemtime(&(job_info->submitted), t);
6456         job_info->timeelapsed=0;
6457         job_info->pagesprinted=0;
6458
6459         job_info->devmode = devmode;
6460
6461         return (True);
6462 }
6463
6464 /****************************************************************************
6465  Enumjobs at level 1.
6466 ****************************************************************************/
6467
6468 static WERROR enumjobs_level1(print_queue_struct *queue, int snum,
6469                               NT_PRINTER_INFO_LEVEL *ntprinter,
6470                               RPC_BUFFER *buffer, uint32 offered,
6471                               uint32 *needed, uint32 *returned)
6472 {
6473         JOB_INFO_1 *info;
6474         int i;
6475         WERROR result = WERR_OK;
6476
6477         info=SMB_MALLOC_ARRAY(JOB_INFO_1,*returned);
6478         if (info==NULL) {
6479                 SAFE_FREE(queue);
6480                 *returned=0;
6481                 return WERR_NOMEM;
6482         }
6483
6484         for (i=0; i<*returned; i++)
6485                 fill_job_info_1( &info[i], &queue[i], i, snum, ntprinter );
6486
6487         SAFE_FREE(queue);
6488
6489         /* check the required size. */
6490         for (i=0; i<*returned; i++)
6491                 (*needed) += spoolss_size_job_info_1(&info[i]);
6492
6493         if (*needed > offered) {
6494                 result = WERR_INSUFFICIENT_BUFFER;
6495                 goto out;
6496         }
6497
6498         if (!rpcbuf_alloc_size(buffer, *needed)) {
6499                 result = WERR_NOMEM;
6500                 goto out;
6501         }
6502
6503         /* fill the buffer with the structures */
6504         for (i=0; i<*returned; i++)
6505                 smb_io_job_info_1("", buffer, &info[i], 0);
6506
6507 out:
6508         /* clear memory */
6509         SAFE_FREE(info);
6510
6511         if ( !W_ERROR_IS_OK(result) )
6512                 *returned = 0;
6513
6514         return result;
6515 }
6516
6517 /****************************************************************************
6518  Enumjobs at level 2.
6519 ****************************************************************************/
6520
6521 static WERROR enumjobs_level2(print_queue_struct *queue, int snum,
6522                               NT_PRINTER_INFO_LEVEL *ntprinter,
6523                               RPC_BUFFER *buffer, uint32 offered,
6524                               uint32 *needed, uint32 *returned)
6525 {
6526         JOB_INFO_2 *info = NULL;
6527         int i;
6528         WERROR result = WERR_OK;
6529         DEVICEMODE *devmode = NULL;
6530
6531         if ( !(info = SMB_MALLOC_ARRAY(JOB_INFO_2,*returned)) ) {
6532                 *returned=0;
6533                 return WERR_NOMEM;
6534         }
6535
6536         /* this should not be a failure condition if the devmode is NULL */
6537
6538         devmode = construct_dev_mode(snum);
6539
6540         for (i=0; i<*returned; i++)
6541                 fill_job_info_2(&(info[i]), &queue[i], i, snum, ntprinter, devmode);
6542
6543         free_a_printer(&ntprinter, 2);
6544         SAFE_FREE(queue);
6545
6546         /* check the required size. */
6547         for (i=0; i<*returned; i++)
6548                 (*needed) += spoolss_size_job_info_2(&info[i]);
6549
6550         if (*needed > offered) {
6551                 result = WERR_INSUFFICIENT_BUFFER;
6552                 goto out;
6553         }
6554
6555         if (!rpcbuf_alloc_size(buffer, *needed)) {
6556                 result = WERR_NOMEM;
6557                 goto out;
6558         }
6559
6560         /* fill the buffer with the structures */
6561         for (i=0; i<*returned; i++)
6562                 smb_io_job_info_2("", buffer, &info[i], 0);
6563
6564 out:
6565         free_devmode(devmode);
6566         SAFE_FREE(info);
6567
6568         if ( !W_ERROR_IS_OK(result) )
6569                 *returned = 0;
6570
6571         return result;
6572
6573 }
6574
6575 /****************************************************************************
6576  Enumjobs.
6577 ****************************************************************************/
6578
6579 WERROR _spoolss_enumjobs( pipes_struct *p, SPOOL_Q_ENUMJOBS *q_u, SPOOL_R_ENUMJOBS *r_u)
6580 {
6581         POLICY_HND *handle = &q_u->handle;
6582         uint32 level = q_u->level;
6583         RPC_BUFFER *buffer = NULL;
6584         uint32 offered = q_u->offered;
6585         uint32 *needed = &r_u->needed;
6586         uint32 *returned = &r_u->returned;
6587         WERROR wret;
6588         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
6589         int snum;
6590         print_status_struct prt_status;
6591         print_queue_struct *queue=NULL;
6592
6593         /* that's an [in out] buffer */
6594
6595         if (!q_u->buffer) {
6596                 return WERR_INVALID_PARAM;
6597         }
6598
6599         rpcbuf_move(q_u->buffer, &r_u->buffer);
6600         buffer = r_u->buffer;
6601
6602         DEBUG(4,("_spoolss_enumjobs\n"));
6603
6604         *needed=0;
6605         *returned=0;
6606
6607         /* lookup the printer snum and tdb entry */
6608
6609         if (!get_printer_snum(p, handle, &snum))
6610                 return WERR_BADFID;
6611
6612         wret = get_a_printer(NULL, &ntprinter, 2, lp_servicename(snum));
6613         if ( !W_ERROR_IS_OK(wret) )
6614                 return wret;
6615
6616         *returned = print_queue_status(snum, &queue, &prt_status);
6617         DEBUGADD(4,("count:[%d], status:[%d], [%s]\n", *returned, prt_status.status, prt_status.message));
6618
6619         if (*returned == 0) {
6620                 SAFE_FREE(queue);
6621                 return WERR_OK;
6622         }
6623
6624         switch (level) {
6625         case 1:
6626                 wret = enumjobs_level1(queue, snum, ntprinter, buffer, offered, needed, returned);
6627                 return wret;
6628         case 2:
6629                 wret = enumjobs_level2(queue, snum, ntprinter, buffer, offered, needed, returned);
6630                 return wret;
6631         default:
6632                 SAFE_FREE(queue);
6633                 *returned=0;
6634                 wret = WERR_UNKNOWN_LEVEL;
6635         }
6636
6637         free_a_printer( &ntprinter, 2 );
6638         return wret;
6639 }
6640
6641 /****************************************************************************
6642 ****************************************************************************/
6643
6644 WERROR _spoolss_schedulejob( pipes_struct *p, SPOOL_Q_SCHEDULEJOB *q_u, SPOOL_R_SCHEDULEJOB *r_u)
6645 {
6646         return WERR_OK;
6647 }
6648
6649 /****************************************************************************
6650 ****************************************************************************/
6651
6652 WERROR _spoolss_setjob(pipes_struct *p, SPOOL_Q_SETJOB *q_u, SPOOL_R_SETJOB *r_u)
6653 {
6654         POLICY_HND *handle = &q_u->handle;
6655         uint32 jobid = q_u->jobid;
6656         uint32 command = q_u->command;
6657
6658         struct current_user user;
6659         int snum;
6660         WERROR errcode = WERR_BADFUNC;
6661
6662         if (!get_printer_snum(p, handle, &snum)) {
6663                 return WERR_BADFID;
6664         }
6665
6666         if (!print_job_exists(lp_const_servicename(snum), jobid)) {
6667                 return WERR_INVALID_PRINTER_NAME;
6668         }
6669
6670         get_current_user(&user, p);
6671
6672         switch (command) {
6673         case JOB_CONTROL_CANCEL:
6674         case JOB_CONTROL_DELETE:
6675                 if (print_job_delete(&user, snum, jobid, &errcode)) {
6676                         errcode = WERR_OK;
6677                 }
6678                 break;
6679         case JOB_CONTROL_PAUSE:
6680                 if (print_job_pause(&user, snum, jobid, &errcode)) {
6681                         errcode = WERR_OK;
6682                 }
6683                 break;
6684         case JOB_CONTROL_RESTART:
6685         case JOB_CONTROL_RESUME:
6686                 if (print_job_resume(&user, snum, jobid, &errcode)) {
6687                         errcode = WERR_OK;
6688                 }
6689                 break;
6690         default:
6691                 return WERR_UNKNOWN_LEVEL;
6692         }
6693
6694         return errcode;
6695 }
6696
6697 /****************************************************************************
6698  Enumerates all printer drivers at level 1.
6699 ****************************************************************************/
6700
6701 static WERROR enumprinterdrivers_level1(fstring servername, fstring architecture, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
6702 {
6703         int i;
6704         int ndrivers;
6705         uint32 version;
6706         fstring *list = NULL;
6707         NT_PRINTER_DRIVER_INFO_LEVEL driver;
6708         DRIVER_INFO_1 *driver_info_1=NULL;
6709         WERROR result = WERR_OK;
6710
6711         *returned=0;
6712
6713         for (version=0; version<DRIVER_MAX_VERSION; version++) {
6714                 list=NULL;
6715                 ndrivers=get_ntdrivers(&list, architecture, version);
6716                 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n", ndrivers, architecture, version));
6717
6718                 if(ndrivers == -1)
6719                         return WERR_NOMEM;
6720
6721                 if(ndrivers != 0) {
6722                         if((driver_info_1=SMB_REALLOC_ARRAY(driver_info_1, DRIVER_INFO_1, *returned+ndrivers )) == NULL) {
6723                                 DEBUG(0,("enumprinterdrivers_level1: failed to enlarge driver info buffer!\n"));
6724                                 SAFE_FREE(list);
6725                                 return WERR_NOMEM;
6726                         }
6727                 }
6728
6729                 for (i=0; i<ndrivers; i++) {
6730                         WERROR status;
6731                         DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
6732                         ZERO_STRUCT(driver);
6733                         status = get_a_printer_driver(&driver, 3, list[i],
6734                                                       architecture, version);
6735                         if (!W_ERROR_IS_OK(status)) {
6736                                 SAFE_FREE(list);
6737                                 return status;
6738                         }
6739                         fill_printer_driver_info_1(&driver_info_1[*returned+i], driver, servername, architecture );
6740                         free_a_printer_driver(driver, 3);
6741                 }
6742
6743                 *returned+=ndrivers;
6744                 SAFE_FREE(list);
6745         }
6746
6747         /* check the required size. */
6748         for (i=0; i<*returned; i++) {
6749                 DEBUGADD(6,("adding driver [%d]'s size\n",i));
6750                 *needed += spoolss_size_printer_driver_info_1(&driver_info_1[i]);
6751         }
6752
6753         if (*needed > offered) {
6754                 result = WERR_INSUFFICIENT_BUFFER;
6755                 goto out;
6756         }
6757
6758         if (!rpcbuf_alloc_size(buffer, *needed)) {
6759                 result = WERR_NOMEM;
6760                 goto out;
6761         }
6762
6763         /* fill the buffer with the driver structures */
6764         for (i=0; i<*returned; i++) {
6765                 DEBUGADD(6,("adding driver [%d] to buffer\n",i));
6766                 smb_io_printer_driver_info_1("", buffer, &driver_info_1[i], 0);
6767         }
6768
6769 out:
6770         SAFE_FREE(driver_info_1);
6771
6772         if ( !W_ERROR_IS_OK(result) )
6773                 *returned = 0;
6774
6775         return result;
6776 }
6777
6778 /****************************************************************************
6779  Enumerates all printer drivers at level 2.
6780 ****************************************************************************/
6781
6782 static WERROR enumprinterdrivers_level2(fstring servername, fstring architecture, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
6783 {
6784         int i;
6785         int ndrivers;
6786         uint32 version;
6787         fstring *list = NULL;
6788         NT_PRINTER_DRIVER_INFO_LEVEL driver;
6789         DRIVER_INFO_2 *driver_info_2=NULL;
6790         WERROR result = WERR_OK;
6791
6792         *returned=0;
6793
6794         for (version=0; version<DRIVER_MAX_VERSION; version++) {
6795                 list=NULL;
6796                 ndrivers=get_ntdrivers(&list, architecture, version);
6797                 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n", ndrivers, architecture, version));
6798
6799                 if(ndrivers == -1)
6800                         return WERR_NOMEM;
6801
6802                 if(ndrivers != 0) {
6803                         if((driver_info_2=SMB_REALLOC_ARRAY(driver_info_2, DRIVER_INFO_2, *returned+ndrivers )) == NULL) {
6804                                 DEBUG(0,("enumprinterdrivers_level2: failed to enlarge driver info buffer!\n"));
6805                                 SAFE_FREE(list);
6806                                 return WERR_NOMEM;
6807                         }
6808                 }
6809
6810                 for (i=0; i<ndrivers; i++) {
6811                         WERROR status;
6812
6813                         DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
6814                         ZERO_STRUCT(driver);
6815                         status = get_a_printer_driver(&driver, 3, list[i],
6816                                                       architecture, version);
6817                         if (!W_ERROR_IS_OK(status)) {
6818                                 SAFE_FREE(list);
6819                                 return status;
6820                         }
6821                         fill_printer_driver_info_2(&driver_info_2[*returned+i], driver, servername);
6822                         free_a_printer_driver(driver, 3);
6823                 }
6824
6825                 *returned+=ndrivers;
6826                 SAFE_FREE(list);
6827         }
6828
6829         /* check the required size. */
6830         for (i=0; i<*returned; i++) {
6831                 DEBUGADD(6,("adding driver [%d]'s size\n",i));
6832                 *needed += spoolss_size_printer_driver_info_2(&(driver_info_2[i]));
6833         }
6834
6835         if (*needed > offered) {
6836                 result = WERR_INSUFFICIENT_BUFFER;
6837                 goto out;
6838         }
6839
6840         if (!rpcbuf_alloc_size(buffer, *needed)) {
6841                 result = WERR_NOMEM;
6842                 goto out;
6843         }
6844
6845         /* fill the buffer with the form structures */
6846         for (i=0; i<*returned; i++) {
6847                 DEBUGADD(6,("adding driver [%d] to buffer\n",i));
6848                 smb_io_printer_driver_info_2("", buffer, &(driver_info_2[i]), 0);
6849         }
6850
6851 out:
6852         SAFE_FREE(driver_info_2);
6853
6854         if ( !W_ERROR_IS_OK(result) )
6855                 *returned = 0;
6856
6857         return result;
6858 }
6859
6860 /****************************************************************************
6861  Enumerates all printer drivers at level 3.
6862 ****************************************************************************/
6863
6864 static WERROR enumprinterdrivers_level3(fstring servername, fstring architecture, RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
6865 {
6866         int i;
6867         int ndrivers;
6868         uint32 version;
6869         fstring *list = NULL;
6870         NT_PRINTER_DRIVER_INFO_LEVEL driver;
6871         DRIVER_INFO_3 *driver_info_3=NULL;
6872         WERROR result = WERR_OK;
6873
6874         *returned=0;
6875
6876         for (version=0; version<DRIVER_MAX_VERSION; version++) {
6877                 list=NULL;
6878                 ndrivers=get_ntdrivers(&list, architecture, version);
6879                 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n", ndrivers, architecture, version));
6880
6881                 if(ndrivers == -1)
6882                         return WERR_NOMEM;
6883
6884                 if(ndrivers != 0) {
6885                         if((driver_info_3=SMB_REALLOC_ARRAY(driver_info_3, DRIVER_INFO_3, *returned+ndrivers )) == NULL) {
6886                                 DEBUG(0,("enumprinterdrivers_level3: failed to enlarge driver info buffer!\n"));
6887                                 SAFE_FREE(list);
6888                                 return WERR_NOMEM;
6889                         }
6890                 }
6891
6892                 for (i=0; i<ndrivers; i++) {
6893                         WERROR status;
6894
6895                         DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
6896                         ZERO_STRUCT(driver);
6897                         status = get_a_printer_driver(&driver, 3, list[i],
6898                                                       architecture, version);
6899                         if (!W_ERROR_IS_OK(status)) {
6900                                 SAFE_FREE(list);
6901                                 return status;
6902                         }
6903                         fill_printer_driver_info_3(&driver_info_3[*returned+i], driver, servername);
6904                         free_a_printer_driver(driver, 3);
6905                 }
6906
6907                 *returned+=ndrivers;
6908                 SAFE_FREE(list);
6909         }
6910
6911         /* check the required size. */
6912         for (i=0; i<*returned; i++) {
6913                 DEBUGADD(6,("adding driver [%d]'s size\n",i));
6914                 *needed += spoolss_size_printer_driver_info_3(&driver_info_3[i]);
6915         }
6916
6917         if (*needed > offered) {
6918                 result = WERR_INSUFFICIENT_BUFFER;
6919                 goto out;
6920         }
6921
6922         if (!rpcbuf_alloc_size(buffer, *needed)) {
6923                 result = WERR_NOMEM;
6924                 goto out;
6925         }
6926
6927         /* fill the buffer with the driver structures */
6928         for (i=0; i<*returned; i++) {
6929                 DEBUGADD(6,("adding driver [%d] to buffer\n",i));
6930                 smb_io_printer_driver_info_3("", buffer, &driver_info_3[i], 0);
6931         }
6932
6933 out:
6934         for (i=0; i<*returned; i++)
6935                 SAFE_FREE(driver_info_3[i].dependentfiles);
6936
6937         SAFE_FREE(driver_info_3);
6938
6939         if ( !W_ERROR_IS_OK(result) )
6940                 *returned = 0;
6941
6942         return result;
6943 }
6944
6945 /****************************************************************************
6946  Enumerates all printer drivers.
6947 ****************************************************************************/
6948
6949 WERROR _spoolss_enumprinterdrivers( pipes_struct *p, SPOOL_Q_ENUMPRINTERDRIVERS *q_u, SPOOL_R_ENUMPRINTERDRIVERS *r_u)
6950 {
6951         uint32 level = q_u->level;
6952         RPC_BUFFER *buffer = NULL;
6953         uint32 offered = q_u->offered;
6954         uint32 *needed = &r_u->needed;
6955         uint32 *returned = &r_u->returned;
6956
6957         fstring servername;
6958         fstring architecture;
6959
6960         /* that's an [in out] buffer */
6961
6962         if (!q_u->buffer) {
6963                 return WERR_INVALID_PARAM;
6964         }
6965
6966         rpcbuf_move(q_u->buffer, &r_u->buffer);
6967         buffer = r_u->buffer;
6968
6969         DEBUG(4,("_spoolss_enumprinterdrivers\n"));
6970
6971         *needed   = 0;
6972         *returned = 0;
6973
6974         unistr2_to_ascii(architecture, &q_u->environment, sizeof(architecture)-1);
6975         unistr2_to_ascii(servername, &q_u->name, sizeof(servername)-1);
6976
6977         if ( !is_myname_or_ipaddr( servername ) )
6978                 return WERR_UNKNOWN_PRINTER_DRIVER;
6979
6980         switch (level) {
6981         case 1:
6982                 return enumprinterdrivers_level1(servername, architecture, buffer, offered, needed, returned);
6983         case 2:
6984                 return enumprinterdrivers_level2(servername, architecture, buffer, offered, needed, returned);
6985         case 3:
6986                 return enumprinterdrivers_level3(servername, architecture, buffer, offered, needed, returned);
6987         default:
6988                 return WERR_UNKNOWN_LEVEL;
6989         }
6990 }
6991
6992 /****************************************************************************
6993 ****************************************************************************/
6994
6995 static void fill_form_1(FORM_1 *form, nt_forms_struct *list)
6996 {
6997         form->flag=list->flag;
6998         init_unistr(&form->name, list->name);
6999         form->width=list->width;
7000         form->length=list->length;
7001         form->left=list->left;
7002         form->top=list->top;
7003         form->right=list->right;
7004         form->bottom=list->bottom;
7005 }
7006
7007 /****************************************************************************
7008 ****************************************************************************/
7009
7010 WERROR _spoolss_enumforms(pipes_struct *p, SPOOL_Q_ENUMFORMS *q_u, SPOOL_R_ENUMFORMS *r_u)
7011 {
7012         uint32 level = q_u->level;
7013         RPC_BUFFER *buffer = NULL;
7014         uint32 offered = q_u->offered;
7015         uint32 *needed = &r_u->needed;
7016         uint32 *numofforms = &r_u->numofforms;
7017         uint32 numbuiltinforms;
7018
7019         nt_forms_struct *list=NULL;
7020         nt_forms_struct *builtinlist=NULL;
7021         FORM_1 *forms_1;
7022         int buffer_size=0;
7023         int i;
7024
7025         /* that's an [in out] buffer */
7026
7027         if (!q_u->buffer) {
7028                 return WERR_INVALID_PARAM;
7029         }
7030
7031         rpcbuf_move(q_u->buffer, &r_u->buffer);
7032         buffer = r_u->buffer;
7033
7034         DEBUG(4,("_spoolss_enumforms\n"));
7035         DEBUGADD(5,("Offered buffer size [%d]\n", offered));
7036         DEBUGADD(5,("Info level [%d]\n",          level));
7037
7038         numbuiltinforms = get_builtin_ntforms(&builtinlist);
7039         DEBUGADD(5,("Number of builtin forms [%d]\n",     numbuiltinforms));
7040         *numofforms = get_ntforms(&list);
7041         DEBUGADD(5,("Number of user forms [%d]\n",     *numofforms));
7042         *numofforms += numbuiltinforms;
7043
7044         if (*numofforms == 0)
7045                 return WERR_NO_MORE_ITEMS;
7046
7047         switch (level) {
7048         case 1:
7049                 if ((forms_1=SMB_MALLOC_ARRAY(FORM_1, *numofforms)) == NULL) {
7050                         *numofforms=0;
7051                         return WERR_NOMEM;
7052                 }
7053
7054                 /* construct the list of form structures */
7055                 for (i=0; i<numbuiltinforms; i++) {
7056                         DEBUGADD(6,("Filling form number [%d]\n",i));
7057                         fill_form_1(&forms_1[i], &builtinlist[i]);
7058                 }
7059
7060                 SAFE_FREE(builtinlist);
7061
7062                 for (; i<*numofforms; i++) {
7063                         DEBUGADD(6,("Filling form number [%d]\n",i));
7064                         fill_form_1(&forms_1[i], &list[i-numbuiltinforms]);
7065                 }
7066
7067                 SAFE_FREE(list);
7068
7069                 /* check the required size. */
7070                 for (i=0; i<numbuiltinforms; i++) {
7071                         DEBUGADD(6,("adding form [%d]'s size\n",i));
7072                         buffer_size += spoolss_size_form_1(&forms_1[i]);
7073                 }
7074                 for (; i<*numofforms; i++) {
7075                         DEBUGADD(6,("adding form [%d]'s size\n",i));
7076                         buffer_size += spoolss_size_form_1(&forms_1[i]);
7077                 }
7078
7079                 *needed=buffer_size;
7080
7081                 if (*needed > offered) {
7082                         SAFE_FREE(forms_1);
7083                         *numofforms=0;
7084                         return WERR_INSUFFICIENT_BUFFER;
7085                 }
7086
7087                 if (!rpcbuf_alloc_size(buffer, buffer_size)){
7088                         SAFE_FREE(forms_1);
7089                         *numofforms=0;
7090                         return WERR_NOMEM;
7091                 }
7092
7093                 /* fill the buffer with the form structures */
7094                 for (i=0; i<numbuiltinforms; i++) {
7095                         DEBUGADD(6,("adding form [%d] to buffer\n",i));
7096                         smb_io_form_1("", buffer, &forms_1[i], 0);
7097                 }
7098                 for (; i<*numofforms; i++) {
7099                         DEBUGADD(6,("adding form [%d] to buffer\n",i));
7100                         smb_io_form_1("", buffer, &forms_1[i], 0);
7101                 }
7102
7103                 SAFE_FREE(forms_1);
7104
7105                 return WERR_OK;
7106
7107         default:
7108                 SAFE_FREE(list);
7109                 SAFE_FREE(builtinlist);
7110                 return WERR_UNKNOWN_LEVEL;
7111         }
7112
7113 }
7114
7115 /****************************************************************************
7116 ****************************************************************************/
7117
7118 WERROR _spoolss_getform(pipes_struct *p, SPOOL_Q_GETFORM *q_u, SPOOL_R_GETFORM *r_u)
7119 {
7120         uint32 level = q_u->level;
7121         UNISTR2 *uni_formname = &q_u->formname;
7122         RPC_BUFFER *buffer = NULL;
7123         uint32 offered = q_u->offered;
7124         uint32 *needed = &r_u->needed;
7125
7126         nt_forms_struct *list=NULL;
7127         nt_forms_struct builtin_form;
7128         BOOL foundBuiltin;
7129         FORM_1 form_1;
7130         fstring form_name;
7131         int buffer_size=0;
7132         int numofforms=0, i=0;
7133
7134         /* that's an [in out] buffer */
7135
7136         if (!q_u->buffer) {
7137                 return WERR_INVALID_PARAM;
7138         }
7139
7140         rpcbuf_move(q_u->buffer, &r_u->buffer);
7141         buffer = r_u->buffer;
7142
7143         unistr2_to_ascii(form_name, uni_formname, sizeof(form_name)-1);
7144
7145         DEBUG(4,("_spoolss_getform\n"));
7146         DEBUGADD(5,("Offered buffer size [%d]\n", offered));
7147         DEBUGADD(5,("Info level [%d]\n",          level));
7148
7149         foundBuiltin = get_a_builtin_ntform(uni_formname,&builtin_form);
7150         if (!foundBuiltin) {
7151                 numofforms = get_ntforms(&list);
7152                 DEBUGADD(5,("Number of forms [%d]\n",     numofforms));
7153
7154                 if (numofforms == 0)
7155                         return WERR_BADFID;
7156         }
7157
7158         switch (level) {
7159         case 1:
7160                 if (foundBuiltin) {
7161                         fill_form_1(&form_1, &builtin_form);
7162                 } else {
7163
7164                         /* Check if the requested name is in the list of form structures */
7165                         for (i=0; i<numofforms; i++) {
7166
7167                                 DEBUG(4,("_spoolss_getform: checking form %s (want %s)\n", list[i].name, form_name));
7168
7169                                 if (strequal(form_name, list[i].name)) {
7170                                         DEBUGADD(6,("Found form %s number [%d]\n", form_name, i));
7171                                         fill_form_1(&form_1, &list[i]);
7172                                         break;
7173                                 }
7174                         }
7175
7176                         SAFE_FREE(list);
7177                         if (i == numofforms) {
7178                                 return WERR_BADFID;
7179                         }
7180                 }
7181                 /* check the required size. */
7182
7183                 *needed=spoolss_size_form_1(&form_1);
7184
7185                 if (*needed > offered)
7186                         return WERR_INSUFFICIENT_BUFFER;
7187
7188                 if (!rpcbuf_alloc_size(buffer, buffer_size))
7189                         return WERR_NOMEM;
7190
7191                 /* fill the buffer with the form structures */
7192                 DEBUGADD(6,("adding form %s [%d] to buffer\n", form_name, i));
7193                 smb_io_form_1("", buffer, &form_1, 0);
7194
7195                 return WERR_OK;
7196
7197         default:
7198                 SAFE_FREE(list);
7199                 return WERR_UNKNOWN_LEVEL;
7200         }
7201 }
7202
7203 /****************************************************************************
7204 ****************************************************************************/
7205
7206 static void fill_port_1(PORT_INFO_1 *port, const char *name)
7207 {
7208         init_unistr(&port->port_name, name);
7209 }
7210
7211 /****************************************************************************
7212  TODO: This probably needs distinguish between TCP/IP and Local ports
7213  somehow.
7214 ****************************************************************************/
7215
7216 static void fill_port_2(PORT_INFO_2 *port, const char *name)
7217 {
7218         init_unistr(&port->port_name, name);
7219         init_unistr(&port->monitor_name, "Local Monitor");
7220         init_unistr(&port->description, SPL_LOCAL_PORT );
7221         port->port_type=PORT_TYPE_WRITE;
7222         port->reserved=0x0;
7223 }
7224
7225
7226 /****************************************************************************
7227  wrapper around the enumer ports command
7228 ****************************************************************************/
7229
7230 WERROR enumports_hook( int *count, char ***lines )
7231 {
7232         char *cmd = lp_enumports_cmd();
7233         char **qlines;
7234         pstring command;
7235         int numlines;
7236         int ret;
7237         int fd;
7238
7239
7240         /* if no hook then just fill in the default port */
7241
7242         if ( !*cmd ) {
7243                 qlines = SMB_MALLOC_ARRAY( char*, 2 );
7244                 qlines[0] = SMB_STRDUP( SAMBA_PRINTER_PORT_NAME );
7245                 qlines[1] = NULL;
7246                 numlines = 1;
7247         }
7248         else {
7249                 /* we have a valid enumport command */
7250
7251                 slprintf(command, sizeof(command)-1, "%s \"%d\"", cmd, 1);
7252
7253                 DEBUG(10,("Running [%s]\n", command));
7254                 ret = smbrun(command, &fd);
7255                 DEBUG(10,("Returned [%d]\n", ret));
7256                 if (ret != 0) {
7257                         if (fd != -1)
7258                                 close(fd);
7259
7260                         return WERR_ACCESS_DENIED;
7261                 }
7262
7263                 numlines = 0;
7264                 qlines = fd_lines_load(fd, &numlines, 0);
7265                 DEBUGADD(10,("Lines returned = [%d]\n", numlines));
7266                 close(fd);
7267         }
7268
7269         *count = numlines;
7270         *lines = qlines;
7271
7272         return WERR_OK;
7273 }
7274
7275 /****************************************************************************
7276  enumports level 1.
7277 ****************************************************************************/
7278
7279 static WERROR enumports_level_1(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
7280 {
7281         PORT_INFO_1 *ports=NULL;
7282         int i=0;
7283         WERROR result = WERR_OK;
7284         char **qlines;
7285         int numlines;
7286
7287         if ( !W_ERROR_IS_OK(result = enumports_hook( &numlines, &qlines )) )
7288                 return result;
7289
7290         if(numlines) {
7291                 if((ports=SMB_MALLOC_ARRAY( PORT_INFO_1, numlines )) == NULL) {
7292                         DEBUG(10,("Returning WERR_NOMEM [%s]\n",
7293                                   dos_errstr(WERR_NOMEM)));
7294                         file_lines_free(qlines);
7295                         return WERR_NOMEM;
7296                 }
7297
7298                 for (i=0; i<numlines; i++) {
7299                         DEBUG(6,("Filling port number [%d] with port [%s]\n", i, qlines[i]));
7300                         fill_port_1(&ports[i], qlines[i]);
7301                 }
7302
7303                 file_lines_free(qlines);
7304         }
7305
7306         *returned = numlines;
7307
7308         /* check the required size. */
7309         for (i=0; i<*returned; i++) {
7310                 DEBUGADD(6,("adding port [%d]'s size\n", i));
7311                 *needed += spoolss_size_port_info_1(&ports[i]);
7312         }
7313
7314         if (*needed > offered) {
7315                 result = WERR_INSUFFICIENT_BUFFER;
7316                 goto out;
7317         }
7318
7319         if (!rpcbuf_alloc_size(buffer, *needed)) {
7320                 result = WERR_NOMEM;
7321                 goto out;
7322         }
7323
7324         /* fill the buffer with the ports structures */
7325         for (i=0; i<*returned; i++) {
7326                 DEBUGADD(6,("adding port [%d] to buffer\n", i));
7327                 smb_io_port_1("", buffer, &ports[i], 0);
7328         }
7329
7330 out:
7331         SAFE_FREE(ports);
7332
7333         if ( !W_ERROR_IS_OK(result) )
7334                 *returned = 0;
7335
7336         return result;
7337 }
7338
7339 /****************************************************************************
7340  enumports level 2.
7341 ****************************************************************************/
7342
7343 static WERROR enumports_level_2(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
7344 {
7345         PORT_INFO_2 *ports=NULL;
7346         int i=0;
7347         WERROR result = WERR_OK;
7348         char **qlines;
7349         int numlines;
7350
7351         if ( !W_ERROR_IS_OK(result = enumports_hook( &numlines, &qlines )) )
7352                 return result;
7353
7354
7355         if(numlines) {
7356                 if((ports=SMB_MALLOC_ARRAY( PORT_INFO_2, numlines)) == NULL) {
7357                         file_lines_free(qlines);
7358                         return WERR_NOMEM;
7359                 }
7360
7361                 for (i=0; i<numlines; i++) {
7362                         DEBUG(6,("Filling port number [%d] with port [%s]\n", i, qlines[i]));
7363                         fill_port_2(&(ports[i]), qlines[i]);
7364                 }
7365
7366                 file_lines_free(qlines);
7367         }
7368
7369         *returned = numlines;
7370
7371         /* check the required size. */
7372         for (i=0; i<*returned; i++) {
7373                 DEBUGADD(6,("adding port [%d]'s size\n", i));
7374                 *needed += spoolss_size_port_info_2(&ports[i]);
7375         }
7376
7377         if (*needed > offered) {
7378                 result = WERR_INSUFFICIENT_BUFFER;
7379                 goto out;
7380         }
7381
7382         if (!rpcbuf_alloc_size(buffer, *needed)) {
7383                 result = WERR_NOMEM;
7384                 goto out;
7385         }
7386
7387         /* fill the buffer with the ports structures */
7388         for (i=0; i<*returned; i++) {
7389                 DEBUGADD(6,("adding port [%d] to buffer\n", i));
7390                 smb_io_port_2("", buffer, &ports[i], 0);
7391         }
7392
7393 out:
7394         SAFE_FREE(ports);
7395
7396         if ( !W_ERROR_IS_OK(result) )
7397                 *returned = 0;
7398
7399         return result;
7400 }
7401
7402 /****************************************************************************
7403  enumports.
7404 ****************************************************************************/
7405
7406 WERROR _spoolss_enumports( pipes_struct *p, SPOOL_Q_ENUMPORTS *q_u, SPOOL_R_ENUMPORTS *r_u)
7407 {
7408         uint32 level = q_u->level;
7409         RPC_BUFFER *buffer = NULL;
7410         uint32 offered = q_u->offered;
7411         uint32 *needed = &r_u->needed;
7412         uint32 *returned = &r_u->returned;
7413
7414         /* that's an [in out] buffer */
7415
7416         if (!q_u->buffer) {
7417                 return WERR_INVALID_PARAM;
7418         }
7419
7420         rpcbuf_move(q_u->buffer, &r_u->buffer);
7421         buffer = r_u->buffer;
7422
7423         DEBUG(4,("_spoolss_enumports\n"));
7424
7425         *returned=0;
7426         *needed=0;
7427
7428         switch (level) {
7429         case 1:
7430                 return enumports_level_1(buffer, offered, needed, returned);
7431         case 2:
7432                 return enumports_level_2(buffer, offered, needed, returned);
7433         default:
7434                 return WERR_UNKNOWN_LEVEL;
7435         }
7436 }
7437
7438 /****************************************************************************
7439 ****************************************************************************/
7440
7441 static WERROR spoolss_addprinterex_level_2( pipes_struct *p, const UNISTR2 *uni_srv_name,
7442                                 const SPOOL_PRINTER_INFO_LEVEL *info,
7443                                 DEVICEMODE *devmode, SEC_DESC_BUF *sec_desc_buf,
7444                                 uint32 user_switch, const SPOOL_USER_CTR *user,
7445                                 POLICY_HND *handle)
7446 {
7447         NT_PRINTER_INFO_LEVEL *printer = NULL;
7448         fstring name;
7449         int     snum;
7450         WERROR err = WERR_OK;
7451
7452         if ( !(printer = TALLOC_ZERO_P(NULL, NT_PRINTER_INFO_LEVEL)) ) {
7453                 DEBUG(0,("spoolss_addprinterex_level_2: malloc fail.\n"));
7454                 return WERR_NOMEM;
7455         }
7456
7457         /* convert from UNICODE to ASCII - this allocates the info_2 struct inside *printer.*/
7458         if (!convert_printer_info(info, printer, 2)) {
7459                 free_a_printer(&printer, 2);
7460                 return WERR_NOMEM;
7461         }
7462
7463         /* check to see if the printer already exists */
7464
7465         if ((snum = print_queue_snum(printer->info_2->sharename)) != -1) {
7466                 DEBUG(5, ("spoolss_addprinterex_level_2: Attempted to add a printer named [%s] when one already existed!\n",
7467                         printer->info_2->sharename));
7468                 free_a_printer(&printer, 2);
7469                 return WERR_PRINTER_ALREADY_EXISTS;
7470         }
7471
7472         /* FIXME!!!  smbd should check to see if the driver is installed before
7473            trying to add a printer like this  --jerry */
7474
7475         if (*lp_addprinter_cmd() ) {
7476                 if ( !add_printer_hook(p->pipe_user.nt_user_token, printer) ) {
7477                         free_a_printer(&printer,2);
7478                         return WERR_ACCESS_DENIED;
7479                 }
7480         } else {
7481                 DEBUG(0,("spoolss_addprinterex_level_2: add printer for printer %s called and no"
7482                         "smb.conf parameter \"addprinter command\" is defined. This"
7483                         "parameter must exist for this call to succeed\n",
7484                         printer->info_2->sharename ));
7485         }
7486
7487         /* use our primary netbios name since get_a_printer() will convert
7488            it to what the client expects on a case by case basis */
7489
7490         slprintf(name, sizeof(name)-1, "\\\\%s\\%s", global_myname(),
7491              printer->info_2->sharename);
7492
7493
7494         if ((snum = print_queue_snum(printer->info_2->sharename)) == -1) {
7495                 free_a_printer(&printer,2);
7496                 return WERR_ACCESS_DENIED;
7497         }
7498
7499         /* you must be a printer admin to add a new printer */
7500         if (!print_access_check(NULL, snum, PRINTER_ACCESS_ADMINISTER)) {
7501                 free_a_printer(&printer,2);
7502                 return WERR_ACCESS_DENIED;
7503         }
7504
7505         /*
7506          * Do sanity check on the requested changes for Samba.
7507          */
7508
7509         if (!check_printer_ok(printer->info_2, snum)) {
7510                 free_a_printer(&printer,2);
7511                 return WERR_INVALID_PARAM;
7512         }
7513
7514         /*
7515          * When a printer is created, the drivername bound to the printer is used
7516          * to lookup previously saved driver initialization info, which is then
7517          * bound to the new printer, simulating what happens in the Windows arch.
7518          */
7519
7520         if (!devmode)
7521         {
7522                 set_driver_init(printer, 2);
7523         }
7524         else
7525         {
7526                 /* A valid devmode was included, convert and link it
7527                 */
7528                 DEBUGADD(10, ("spoolss_addprinterex_level_2: devmode included, converting\n"));
7529
7530                 if (!convert_devicemode(printer->info_2->printername, devmode,
7531                                 &printer->info_2->devmode))
7532                         return  WERR_NOMEM;
7533         }
7534
7535         /* write the ASCII on disk */
7536         err = mod_a_printer(printer, 2);
7537         if (!W_ERROR_IS_OK(err)) {
7538                 free_a_printer(&printer,2);
7539                 return err;
7540         }
7541
7542         if (!open_printer_hnd(p, handle, name, PRINTER_ACCESS_ADMINISTER)) {
7543                 /* Handle open failed - remove addition. */
7544                 del_a_printer(printer->info_2->sharename);
7545                 free_a_printer(&printer,2);
7546                 return WERR_ACCESS_DENIED;
7547         }
7548
7549         update_c_setprinter(False);
7550         free_a_printer(&printer,2);
7551
7552         return WERR_OK;
7553 }
7554
7555 /****************************************************************************
7556 ****************************************************************************/
7557
7558 WERROR _spoolss_addprinterex( pipes_struct *p, SPOOL_Q_ADDPRINTEREX *q_u, SPOOL_R_ADDPRINTEREX *r_u)
7559 {
7560         UNISTR2 *uni_srv_name = q_u->server_name;
7561         uint32 level = q_u->level;
7562         SPOOL_PRINTER_INFO_LEVEL *info = &q_u->info;
7563         DEVICEMODE *devmode = q_u->devmode_ctr.devmode;
7564         SEC_DESC_BUF *sdb = q_u->secdesc_ctr;
7565         uint32 user_switch = q_u->user_switch;
7566         SPOOL_USER_CTR *user = &q_u->user_ctr;
7567         POLICY_HND *handle = &r_u->handle;
7568
7569         switch (level) {
7570                 case 1:
7571                         /* we don't handle yet */
7572                         /* but I know what to do ... */
7573                         return WERR_UNKNOWN_LEVEL;
7574                 case 2:
7575                         return spoolss_addprinterex_level_2(p, uni_srv_name, info,
7576                                                             devmode, sdb,
7577                                                             user_switch, user, handle);
7578                 default:
7579                         return WERR_UNKNOWN_LEVEL;
7580         }
7581 }
7582
7583 /****************************************************************************
7584 ****************************************************************************/
7585
7586 WERROR _spoolss_addprinterdriver(pipes_struct *p, SPOOL_Q_ADDPRINTERDRIVER *q_u, SPOOL_R_ADDPRINTERDRIVER *r_u)
7587 {
7588         uint32 level = q_u->level;
7589         SPOOL_PRINTER_DRIVER_INFO_LEVEL *info = &q_u->info;
7590         WERROR err = WERR_OK;
7591         NT_PRINTER_DRIVER_INFO_LEVEL driver;
7592         struct current_user user;
7593         fstring driver_name;
7594         uint32 version;
7595
7596         ZERO_STRUCT(driver);
7597
7598         get_current_user(&user, p);
7599
7600         if (!convert_printer_driver_info(info, &driver, level)) {
7601                 err = WERR_NOMEM;
7602                 goto done;
7603         }
7604
7605         DEBUG(5,("Cleaning driver's information\n"));
7606         err = clean_up_driver_struct(driver, level, &user);
7607         if (!W_ERROR_IS_OK(err))
7608                 goto done;
7609
7610         DEBUG(5,("Moving driver to final destination\n"));
7611         if( !W_ERROR_IS_OK(err = move_driver_to_download_area(driver, level, &user, &err)) ) {
7612                 goto done;
7613         }
7614
7615         if (add_a_printer_driver(driver, level)!=0) {
7616                 err = WERR_ACCESS_DENIED;
7617                 goto done;
7618         }
7619
7620         /* BEGIN_ADMIN_LOG */
7621         switch(level) {
7622             case 3:
7623                 fstrcpy(driver_name, driver.info_3->name ? driver.info_3->name : "");
7624                 sys_adminlog(LOG_INFO,"Added printer driver. Print driver name: %s. Print driver OS: %s. Administrator name: %s.",
7625                         driver_name, get_drv_ver_to_os(driver.info_3->cversion),uidtoname(user.ut.uid));
7626                 break;
7627             case 6:
7628                 fstrcpy(driver_name, driver.info_6->name ?  driver.info_6->name : "");
7629                 sys_adminlog(LOG_INFO,"Added printer driver. Print driver name: %s. Print driver OS: %s. Administrator name: %s.",
7630                         driver_name, get_drv_ver_to_os(driver.info_6->version),uidtoname(user.ut.uid));
7631                 break;
7632         }
7633         /* END_ADMIN_LOG */
7634
7635         /*
7636          * I think this is where he DrvUpgradePrinter() hook would be
7637          * be called in a driver's interface DLL on a Windows NT 4.0/2k
7638          * server.  Right now, we just need to send ourselves a message
7639          * to update each printer bound to this driver.   --jerry
7640          */
7641
7642         if (!srv_spoolss_drv_upgrade_printer(driver_name)) {
7643                 DEBUG(0,("_spoolss_addprinterdriver: Failed to send message about upgrading driver [%s]!\n",
7644                         driver_name));
7645         }
7646
7647         /*
7648          * Based on the version (e.g. driver destination dir: 0=9x,2=Nt/2k,3=2k/Xp),
7649          * decide if the driver init data should be deleted. The rules are:
7650          *  1) never delete init data if it is a 9x driver, they don't use it anyway
7651          *  2) delete init data only if there is no 2k/Xp driver
7652          *  3) always delete init data
7653          * The generalized rule is always use init data from the highest order driver.
7654          * It is necessary to follow the driver install by an initialization step to
7655          * finish off this process.
7656         */
7657         if (level == 3)
7658                 version = driver.info_3->cversion;
7659         else if (level == 6)
7660                 version = driver.info_6->version;
7661         else
7662                 version = -1;
7663         switch (version) {
7664                 /*
7665                  * 9x printer driver - never delete init data
7666                 */
7667                 case 0:
7668                         DEBUG(10,("_spoolss_addprinterdriver: init data not deleted for 9x driver [%s]\n",
7669                                         driver_name));
7670                         break;
7671
7672                 /*
7673                  * Nt or 2k (compatiblity mode) printer driver - only delete init data if
7674                  * there is no 2k/Xp driver init data for this driver name.
7675                 */
7676                 case 2:
7677                 {
7678                         NT_PRINTER_DRIVER_INFO_LEVEL driver1;
7679
7680                         if (!W_ERROR_IS_OK(get_a_printer_driver(&driver1, 3, driver_name, "Windows NT x86", 3))) {
7681                                 /*
7682                                  * No 2k/Xp driver found, delete init data (if any) for the new Nt driver.
7683                                 */
7684                                 if (!del_driver_init(driver_name))
7685                                         DEBUG(6,("_spoolss_addprinterdriver: del_driver_init(%s) Nt failed!\n", driver_name));
7686                         } else {
7687                                 /*
7688                                  * a 2k/Xp driver was found, don't delete init data because Nt driver will use it.
7689                                 */
7690                                 free_a_printer_driver(driver1,3);
7691                                 DEBUG(10,("_spoolss_addprinterdriver: init data not deleted for Nt driver [%s]\n",
7692                                                 driver_name));
7693                         }
7694                 }
7695                 break;
7696
7697                 /*
7698                  * 2k or Xp printer driver - always delete init data
7699                 */
7700                 case 3:
7701                         if (!del_driver_init(driver_name))
7702                                 DEBUG(6,("_spoolss_addprinterdriver: del_driver_init(%s) 2k/Xp failed!\n", driver_name));
7703                         break;
7704
7705                 default:
7706                         DEBUG(0,("_spoolss_addprinterdriver: invalid level=%d\n", level));
7707                         break;
7708         }
7709
7710
7711 done:
7712         free_a_printer_driver(driver, level);
7713         return err;
7714 }
7715
7716 /********************************************************************
7717  * spoolss_addprinterdriverex
7718  ********************************************************************/
7719
7720 WERROR _spoolss_addprinterdriverex(pipes_struct *p, SPOOL_Q_ADDPRINTERDRIVEREX *q_u, SPOOL_R_ADDPRINTERDRIVEREX *r_u)
7721 {
7722         SPOOL_Q_ADDPRINTERDRIVER q_u_local;
7723         SPOOL_R_ADDPRINTERDRIVER r_u_local;
7724
7725         /*
7726          * we only support the semantics of AddPrinterDriver()
7727          * i.e. only copy files that are newer than existing ones
7728          */
7729
7730         if ( q_u->copy_flags != APD_COPY_NEW_FILES )
7731                 return WERR_ACCESS_DENIED;
7732
7733         ZERO_STRUCT(q_u_local);
7734         ZERO_STRUCT(r_u_local);
7735
7736         /* just pass the information off to _spoolss_addprinterdriver() */
7737         q_u_local.server_name_ptr = q_u->server_name_ptr;
7738         copy_unistr2(&q_u_local.server_name, &q_u->server_name);
7739         q_u_local.level = q_u->level;
7740         memcpy( &q_u_local.info, &q_u->info, sizeof(SPOOL_PRINTER_DRIVER_INFO_LEVEL) );
7741
7742         return _spoolss_addprinterdriver( p, &q_u_local, &r_u_local );
7743 }
7744
7745 /****************************************************************************
7746 ****************************************************************************/
7747
7748 static void fill_driverdir_1(DRIVER_DIRECTORY_1 *info, char *name)
7749 {
7750         init_unistr(&info->name, name);
7751 }
7752
7753 /****************************************************************************
7754 ****************************************************************************/
7755
7756 static WERROR getprinterdriverdir_level_1(UNISTR2 *name, UNISTR2 *uni_environment, RPC_BUFFER *buffer, uint32 offered, uint32 *needed)
7757 {
7758         pstring path;
7759         pstring long_archi;
7760         fstring servername;
7761         char *pservername;
7762         const char *short_archi;
7763         DRIVER_DIRECTORY_1 *info=NULL;
7764         WERROR result = WERR_OK;
7765
7766         unistr2_to_ascii(servername, name, sizeof(servername)-1);
7767         unistr2_to_ascii(long_archi, uni_environment, sizeof(long_archi)-1);
7768
7769         /* check for beginning double '\'s and that the server
7770            long enough */
7771
7772         pservername = servername;
7773         if ( *pservername == '\\' && strlen(servername)>2 ) {
7774                 pservername += 2;
7775         }
7776
7777         if ( !is_myname_or_ipaddr( pservername ) )
7778                 return WERR_INVALID_PARAM;
7779
7780         if (!(short_archi = get_short_archi(long_archi)))
7781                 return WERR_INVALID_ENVIRONMENT;
7782
7783         if((info=SMB_MALLOC_P(DRIVER_DIRECTORY_1)) == NULL)
7784                 return WERR_NOMEM;
7785
7786         slprintf(path, sizeof(path)-1, "\\\\%s\\print$\\%s", pservername, short_archi);
7787
7788         DEBUG(4,("printer driver directory: [%s]\n", path));
7789
7790         fill_driverdir_1(info, path);
7791
7792         *needed += spoolss_size_driverdir_info_1(info);
7793
7794         if (*needed > offered) {
7795                 result = WERR_INSUFFICIENT_BUFFER;
7796                 goto out;
7797         }
7798
7799         if (!rpcbuf_alloc_size(buffer, *needed)) {
7800                 result = WERR_NOMEM;
7801                 goto out;
7802         }
7803
7804         smb_io_driverdir_1("", buffer, info, 0);
7805
7806 out:
7807         SAFE_FREE(info);
7808
7809         return result;
7810 }
7811
7812 /****************************************************************************
7813 ****************************************************************************/
7814
7815 WERROR _spoolss_getprinterdriverdirectory(pipes_struct *p, SPOOL_Q_GETPRINTERDRIVERDIR *q_u, SPOOL_R_GETPRINTERDRIVERDIR *r_u)
7816 {
7817         UNISTR2 *name = &q_u->name;
7818         UNISTR2 *uni_environment = &q_u->environment;
7819         uint32 level = q_u->level;
7820         RPC_BUFFER *buffer = NULL;
7821         uint32 offered = q_u->offered;
7822         uint32 *needed = &r_u->needed;
7823
7824         /* that's an [in out] buffer */
7825
7826         if (!q_u->buffer ) {
7827                 return WERR_INVALID_PARAM;
7828         }
7829
7830         rpcbuf_move(q_u->buffer, &r_u->buffer);
7831         buffer = r_u->buffer;
7832
7833         DEBUG(4,("_spoolss_getprinterdriverdirectory\n"));
7834
7835         *needed=0;
7836
7837         switch(level) {
7838         case 1:
7839                 return getprinterdriverdir_level_1(name, uni_environment, buffer, offered, needed);
7840         default:
7841                 return WERR_UNKNOWN_LEVEL;
7842         }
7843 }
7844
7845 /****************************************************************************
7846 ****************************************************************************/
7847
7848 WERROR _spoolss_enumprinterdata(pipes_struct *p, SPOOL_Q_ENUMPRINTERDATA *q_u, SPOOL_R_ENUMPRINTERDATA *r_u)
7849 {
7850         POLICY_HND *handle = &q_u->handle;
7851         uint32 idx               = q_u->index;
7852         uint32 in_value_len      = q_u->valuesize;
7853         uint32 in_data_len       = q_u->datasize;
7854         uint32 *out_max_value_len = &r_u->valuesize;
7855         uint16 **out_value       = &r_u->value;
7856         uint32 *out_value_len    = &r_u->realvaluesize;
7857         uint32 *out_type         = &r_u->type;
7858         uint32 *out_max_data_len = &r_u->datasize;
7859         uint8  **data_out        = &r_u->data;
7860         uint32 *out_data_len     = &r_u->realdatasize;
7861
7862         NT_PRINTER_INFO_LEVEL *printer = NULL;
7863
7864         uint32          biggest_valuesize;
7865         uint32          biggest_datasize;
7866         uint32          data_len;
7867         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
7868         int             snum;
7869         WERROR          result;
7870         REGISTRY_VALUE  *val = NULL;
7871         NT_PRINTER_DATA *p_data;
7872         int             i, key_index, num_values;
7873         int             name_length;
7874
7875         *out_type = 0;
7876
7877         *out_max_data_len = 0;
7878         *data_out         = NULL;
7879         *out_data_len     = 0;
7880
7881         DEBUG(5,("spoolss_enumprinterdata\n"));
7882
7883         if (!Printer) {
7884                 DEBUG(2,("_spoolss_enumprinterdata: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
7885                 return WERR_BADFID;
7886         }
7887
7888         if (!get_printer_snum(p,handle, &snum))
7889                 return WERR_BADFID;
7890
7891         result = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
7892         if (!W_ERROR_IS_OK(result))
7893                 return result;
7894
7895         p_data = printer->info_2->data;
7896         key_index = lookup_printerkey( p_data, SPOOL_PRINTERDATA_KEY );
7897
7898         result = WERR_OK;
7899
7900         /*
7901          * The NT machine wants to know the biggest size of value and data
7902          *
7903          * cf: MSDN EnumPrinterData remark section
7904          */
7905
7906         if ( !in_value_len && !in_data_len && (key_index != -1) )
7907         {
7908                 DEBUGADD(6,("Activating NT mega-hack to find sizes\n"));
7909
7910                 biggest_valuesize = 0;
7911                 biggest_datasize  = 0;
7912
7913                 num_values = regval_ctr_numvals( p_data->keys[key_index].values );
7914
7915                 for ( i=0; i<num_values; i++ )
7916                 {
7917                         val = regval_ctr_specific_value( p_data->keys[key_index].values, i );
7918
7919                         name_length = strlen(val->valuename);
7920                         if ( strlen(val->valuename) > biggest_valuesize )
7921                                 biggest_valuesize = name_length;
7922
7923                         if ( val->size > biggest_datasize )
7924                                 biggest_datasize = val->size;
7925
7926                         DEBUG(6,("current values: [%d], [%d]\n", biggest_valuesize,
7927                                 biggest_datasize));
7928                 }
7929
7930                 /* the value is an UNICODE string but real_value_size is the length
7931                    in bytes including the trailing 0 */
7932
7933                 *out_value_len = 2 * (1+biggest_valuesize);
7934                 *out_data_len  = biggest_datasize;
7935
7936                 DEBUG(6,("final values: [%d], [%d]\n", *out_value_len, *out_data_len));
7937
7938                 goto done;
7939         }
7940
7941         /*
7942          * the value len is wrong in NT sp3
7943          * that's the number of bytes not the number of unicode chars
7944          */
7945
7946         if ( key_index != -1 )
7947                 val = regval_ctr_specific_value( p_data->keys[key_index].values, idx );
7948
7949         if ( !val )
7950         {
7951
7952                 /* out_value should default to "" or else NT4 has
7953                    problems unmarshalling the response */
7954
7955                 *out_max_value_len=(in_value_len/sizeof(uint16));
7956
7957                 if((*out_value=(uint16 *)TALLOC_ZERO(p->mem_ctx, in_value_len*sizeof(uint8))) == NULL)
7958                 {
7959                         result = WERR_NOMEM;
7960                         goto done;
7961                 }
7962
7963                 *out_value_len = (uint32)rpcstr_push((char *)*out_value, "", in_value_len, 0);
7964
7965                 /* the data is counted in bytes */
7966
7967                 *out_max_data_len = in_data_len;
7968                 *out_data_len     = in_data_len;
7969
7970                 /* only allocate when given a non-zero data_len */
7971
7972                 if ( in_data_len && ((*data_out=(uint8 *)TALLOC_ZERO(p->mem_ctx, in_data_len*sizeof(uint8))) == NULL) )
7973                 {
7974                         result = WERR_NOMEM;
7975                         goto done;
7976                 }
7977
7978                 result = WERR_NO_MORE_ITEMS;
7979         }
7980         else
7981         {
7982                 /*
7983                  * the value is:
7984                  * - counted in bytes in the request
7985                  * - counted in UNICODE chars in the max reply
7986                  * - counted in bytes in the real size
7987                  *
7988                  * take a pause *before* coding not *during* coding
7989                  */
7990
7991                 /* name */
7992                 *out_max_value_len=(in_value_len/sizeof(uint16));
7993                 if ( (*out_value = (uint16 *)TALLOC_ZERO(p->mem_ctx, in_value_len*sizeof(uint8))) == NULL )
7994                 {
7995                         result = WERR_NOMEM;
7996                         goto done;
7997                 }
7998
7999                 *out_value_len = (uint32)rpcstr_push((char *)*out_value, regval_name(val), in_value_len, 0);
8000
8001                 /* type */
8002
8003                 *out_type = regval_type( val );
8004
8005                 /* data - counted in bytes */
8006
8007                 *out_max_data_len = in_data_len;
8008                 if ( in_data_len && (*data_out = (uint8 *)TALLOC_ZERO(p->mem_ctx, in_data_len*sizeof(uint8))) == NULL)
8009                 {
8010                         result = WERR_NOMEM;
8011                         goto done;
8012                 }
8013                 data_len = regval_size(val);
8014                 if ( *data_out )
8015                         memcpy( *data_out, regval_data_p(val), data_len );
8016                 *out_data_len = data_len;
8017         }
8018
8019 done:
8020         free_a_printer(&printer, 2);
8021         return result;
8022 }
8023
8024 /****************************************************************************
8025 ****************************************************************************/
8026
8027 WERROR _spoolss_setprinterdata( pipes_struct *p, SPOOL_Q_SETPRINTERDATA *q_u, SPOOL_R_SETPRINTERDATA *r_u)
8028 {
8029         POLICY_HND              *handle = &q_u->handle;
8030         UNISTR2                 *value = &q_u->value;
8031         uint32                  type = q_u->type;
8032         uint8                   *data = q_u->data;
8033         uint32                  real_len = q_u->real_len;
8034
8035         NT_PRINTER_INFO_LEVEL   *printer = NULL;
8036         int                     snum=0;
8037         WERROR                  status = WERR_OK;
8038         Printer_entry           *Printer=find_printer_index_by_hnd(p, handle);
8039         fstring                 valuename;
8040
8041         DEBUG(5,("spoolss_setprinterdata\n"));
8042
8043         if (!Printer) {
8044                 DEBUG(2,("_spoolss_setprinterdata: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8045                 return WERR_BADFID;
8046         }
8047
8048         if ( Printer->printer_type == SPLHND_SERVER ) {
8049                 DEBUG(10,("_spoolss_setprinterdata: Not implemented for server handles yet\n"));
8050                 return WERR_INVALID_PARAM;
8051         }
8052
8053         if (!get_printer_snum(p,handle, &snum))
8054                 return WERR_BADFID;
8055
8056         /*
8057          * Access check : NT returns "access denied" if you make a
8058          * SetPrinterData call without the necessary privildge.
8059          * we were originally returning OK if nothing changed
8060          * which made Win2k issue **a lot** of SetPrinterData
8061          * when connecting to a printer  --jerry
8062          */
8063
8064         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER)
8065         {
8066                 DEBUG(3, ("_spoolss_setprinterdata: change denied by handle access permissions\n"));
8067                 status = WERR_ACCESS_DENIED;
8068                 goto done;
8069         }
8070
8071         status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
8072         if (!W_ERROR_IS_OK(status))
8073                 return status;
8074
8075         unistr2_to_ascii( valuename, value, sizeof(valuename)-1 );
8076
8077         /*
8078          * When client side code sets a magic printer data key, detect it and save
8079          * the current printer data and the magic key's data (its the DEVMODE) for
8080          * future printer/driver initializations.
8081          */
8082         if ( (type == REG_BINARY) && strequal( valuename, PHANTOM_DEVMODE_KEY))
8083         {
8084                 /* Set devmode and printer initialization info */
8085                 status = save_driver_init( printer, 2, data, real_len );
8086
8087                 srv_spoolss_reset_printerdata( printer->info_2->drivername );
8088         }
8089         else
8090         {
8091         status = set_printer_dataex( printer, SPOOL_PRINTERDATA_KEY, valuename,
8092                                         type, data, real_len );
8093                 if ( W_ERROR_IS_OK(status) )
8094                         status = mod_a_printer(printer, 2);
8095         }
8096
8097 done:
8098         free_a_printer(&printer, 2);
8099
8100         return status;
8101 }
8102
8103 /****************************************************************************
8104 ****************************************************************************/
8105
8106 WERROR _spoolss_resetprinter(pipes_struct *p, SPOOL_Q_RESETPRINTER *q_u, SPOOL_R_RESETPRINTER *r_u)
8107 {
8108         POLICY_HND      *handle = &q_u->handle;
8109         Printer_entry   *Printer=find_printer_index_by_hnd(p, handle);
8110         int             snum;
8111
8112         DEBUG(5,("_spoolss_resetprinter\n"));
8113
8114         /*
8115          * All we do is to check to see if the handle and queue is valid.
8116          * This call really doesn't mean anything to us because we only
8117          * support RAW printing.   --jerry
8118          */
8119
8120         if (!Printer) {
8121                 DEBUG(2,("_spoolss_resetprinter: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8122                 return WERR_BADFID;
8123         }
8124
8125         if (!get_printer_snum(p,handle, &snum))
8126                 return WERR_BADFID;
8127
8128
8129         /* blindly return success */
8130         return WERR_OK;
8131 }
8132
8133
8134 /****************************************************************************
8135 ****************************************************************************/
8136
8137 WERROR _spoolss_deleteprinterdata(pipes_struct *p, SPOOL_Q_DELETEPRINTERDATA *q_u, SPOOL_R_DELETEPRINTERDATA *r_u)
8138 {
8139         POLICY_HND      *handle = &q_u->handle;
8140         UNISTR2         *value = &q_u->valuename;
8141
8142         NT_PRINTER_INFO_LEVEL   *printer = NULL;
8143         int             snum=0;
8144         WERROR          status = WERR_OK;
8145         Printer_entry   *Printer=find_printer_index_by_hnd(p, handle);
8146         pstring         valuename;
8147
8148         DEBUG(5,("spoolss_deleteprinterdata\n"));
8149
8150         if (!Printer) {
8151                 DEBUG(2,("_spoolss_deleteprinterdata: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8152                 return WERR_BADFID;
8153         }
8154
8155         if (!get_printer_snum(p, handle, &snum))
8156                 return WERR_BADFID;
8157
8158         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER) {
8159                 DEBUG(3, ("_spoolss_deleteprinterdata: printer properties change denied by handle\n"));
8160                 return WERR_ACCESS_DENIED;
8161         }
8162
8163         status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
8164         if (!W_ERROR_IS_OK(status))
8165                 return status;
8166
8167         unistr2_to_ascii( valuename, value, sizeof(valuename)-1 );
8168
8169         status = delete_printer_dataex( printer, SPOOL_PRINTERDATA_KEY, valuename );
8170
8171         if ( W_ERROR_IS_OK(status) )
8172                 mod_a_printer( printer, 2 );
8173
8174         free_a_printer(&printer, 2);
8175
8176         return status;
8177 }
8178
8179 /****************************************************************************
8180 ****************************************************************************/
8181
8182 WERROR _spoolss_addform( pipes_struct *p, SPOOL_Q_ADDFORM *q_u, SPOOL_R_ADDFORM *r_u)
8183 {
8184         POLICY_HND *handle = &q_u->handle;
8185         FORM *form = &q_u->form;
8186         nt_forms_struct tmpForm;
8187         int snum;
8188         WERROR status = WERR_OK;
8189         NT_PRINTER_INFO_LEVEL *printer = NULL;
8190
8191         int count=0;
8192         nt_forms_struct *list=NULL;
8193         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
8194
8195         DEBUG(5,("spoolss_addform\n"));
8196
8197         if (!Printer) {
8198                 DEBUG(2,("_spoolss_addform: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8199                 return WERR_BADFID;
8200         }
8201
8202
8203         /* forms can be added on printer of on the print server handle */
8204
8205         if ( Printer->printer_type == SPLHND_PRINTER )
8206         {
8207                 if (!get_printer_snum(p,handle, &snum))
8208                         return WERR_BADFID;
8209
8210                 status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
8211                 if (!W_ERROR_IS_OK(status))
8212                         goto done;
8213         }
8214
8215         if ( !(Printer->access_granted & (PRINTER_ACCESS_ADMINISTER|SERVER_ACCESS_ADMINISTER)) ) {
8216                 DEBUG(2,("_spoolss_addform: denied by handle permissions.\n"));
8217                 status = WERR_ACCESS_DENIED;
8218                 goto done;
8219         }
8220
8221         /* can't add if builtin */
8222
8223         if (get_a_builtin_ntform(&form->name,&tmpForm)) {
8224                 status = WERR_ALREADY_EXISTS;
8225                 goto done;
8226         }
8227
8228         count = get_ntforms(&list);
8229
8230         if(!add_a_form(&list, form, &count)) {
8231                 status =  WERR_NOMEM;
8232                 goto done;
8233         }
8234
8235         write_ntforms(&list, count);
8236
8237         /*
8238          * ChangeID must always be set if this is a printer
8239          */
8240
8241         if ( Printer->printer_type == SPLHND_PRINTER )
8242                 status = mod_a_printer(printer, 2);
8243
8244 done:
8245         if ( printer )
8246                 free_a_printer(&printer, 2);
8247         SAFE_FREE(list);
8248
8249         return status;
8250 }
8251
8252 /****************************************************************************
8253 ****************************************************************************/
8254
8255 WERROR _spoolss_deleteform( pipes_struct *p, SPOOL_Q_DELETEFORM *q_u, SPOOL_R_DELETEFORM *r_u)
8256 {
8257         POLICY_HND *handle = &q_u->handle;
8258         UNISTR2 *form_name = &q_u->name;
8259         nt_forms_struct tmpForm;
8260         int count=0;
8261         nt_forms_struct *list=NULL;
8262         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
8263         int snum;
8264         WERROR status = WERR_OK;
8265         NT_PRINTER_INFO_LEVEL *printer = NULL;
8266
8267         DEBUG(5,("spoolss_deleteform\n"));
8268
8269         if (!Printer) {
8270                 DEBUG(2,("_spoolss_deleteform: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8271                 return WERR_BADFID;
8272         }
8273
8274         /* forms can be deleted on printer of on the print server handle */
8275
8276         if ( Printer->printer_type == SPLHND_PRINTER )
8277         {
8278                 if (!get_printer_snum(p,handle, &snum))
8279                         return WERR_BADFID;
8280
8281                 status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
8282                 if (!W_ERROR_IS_OK(status))
8283                         goto done;
8284         }
8285
8286         if ( !(Printer->access_granted & (PRINTER_ACCESS_ADMINISTER|SERVER_ACCESS_ADMINISTER)) ) {
8287                 DEBUG(2,("_spoolss_deleteform: denied by handle permissions.\n"));
8288                 status = WERR_ACCESS_DENIED;
8289                 goto done;
8290         }
8291
8292         /* can't delete if builtin */
8293
8294         if (get_a_builtin_ntform(form_name,&tmpForm)) {
8295                 status = WERR_INVALID_PARAM;
8296                 goto done;
8297         }
8298
8299         count = get_ntforms(&list);
8300
8301         if ( !delete_a_form(&list, form_name, &count, &status ))
8302                 goto done;
8303
8304         /*
8305          * ChangeID must always be set if this is a printer
8306          */
8307
8308         if ( Printer->printer_type == SPLHND_PRINTER )
8309                 status = mod_a_printer(printer, 2);
8310
8311 done:
8312         if ( printer )
8313                 free_a_printer(&printer, 2);
8314         SAFE_FREE(list);
8315
8316         return status;
8317 }
8318
8319 /****************************************************************************
8320 ****************************************************************************/
8321
8322 WERROR _spoolss_setform(pipes_struct *p, SPOOL_Q_SETFORM *q_u, SPOOL_R_SETFORM *r_u)
8323 {
8324         POLICY_HND *handle = &q_u->handle;
8325         FORM *form = &q_u->form;
8326         nt_forms_struct tmpForm;
8327         int snum;
8328         WERROR status = WERR_OK;
8329         NT_PRINTER_INFO_LEVEL *printer = NULL;
8330
8331         int count=0;
8332         nt_forms_struct *list=NULL;
8333         Printer_entry *Printer = find_printer_index_by_hnd(p, handle);
8334
8335         DEBUG(5,("spoolss_setform\n"));
8336
8337         if (!Printer) {
8338                 DEBUG(2,("_spoolss_setform: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8339                 return WERR_BADFID;
8340         }
8341
8342         /* forms can be modified on printer of on the print server handle */
8343
8344         if ( Printer->printer_type == SPLHND_PRINTER )
8345         {
8346                 if (!get_printer_snum(p,handle, &snum))
8347                         return WERR_BADFID;
8348
8349                 status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
8350                 if (!W_ERROR_IS_OK(status))
8351                         goto done;
8352         }
8353
8354         if ( !(Printer->access_granted & (PRINTER_ACCESS_ADMINISTER|SERVER_ACCESS_ADMINISTER)) ) {
8355                 DEBUG(2,("_spoolss_setform: denied by handle permissions\n"));
8356                 status = WERR_ACCESS_DENIED;
8357                 goto done;
8358         }
8359
8360         /* can't set if builtin */
8361         if (get_a_builtin_ntform(&form->name,&tmpForm)) {
8362                 status = WERR_INVALID_PARAM;
8363                 goto done;
8364         }
8365
8366         count = get_ntforms(&list);
8367         update_a_form(&list, form, count);
8368         write_ntforms(&list, count);
8369
8370         /*
8371          * ChangeID must always be set if this is a printer
8372          */
8373
8374         if ( Printer->printer_type == SPLHND_PRINTER )
8375                 status = mod_a_printer(printer, 2);
8376
8377
8378 done:
8379         if ( printer )
8380                 free_a_printer(&printer, 2);
8381         SAFE_FREE(list);
8382
8383         return status;
8384 }
8385
8386 /****************************************************************************
8387  enumprintprocessors level 1.
8388 ****************************************************************************/
8389
8390 static WERROR enumprintprocessors_level_1(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
8391 {
8392         PRINTPROCESSOR_1 *info_1=NULL;
8393         WERROR result = WERR_OK;
8394
8395         if((info_1 = SMB_MALLOC_P(PRINTPROCESSOR_1)) == NULL)
8396                 return WERR_NOMEM;
8397
8398         (*returned) = 0x1;
8399
8400         init_unistr(&info_1->name, "winprint");
8401
8402         *needed += spoolss_size_printprocessor_info_1(info_1);
8403
8404         if (*needed > offered) {
8405                 result = WERR_INSUFFICIENT_BUFFER;
8406                 goto out;
8407         }
8408
8409         if (!rpcbuf_alloc_size(buffer, *needed)) {
8410                 result = WERR_NOMEM;
8411                 goto out;
8412         }
8413
8414         smb_io_printprocessor_info_1("", buffer, info_1, 0);
8415
8416 out:
8417         SAFE_FREE(info_1);
8418
8419         if ( !W_ERROR_IS_OK(result) )
8420                 *returned = 0;
8421
8422         return result;
8423 }
8424
8425 /****************************************************************************
8426 ****************************************************************************/
8427
8428 WERROR _spoolss_enumprintprocessors(pipes_struct *p, SPOOL_Q_ENUMPRINTPROCESSORS *q_u, SPOOL_R_ENUMPRINTPROCESSORS *r_u)
8429 {
8430         uint32 level = q_u->level;
8431         RPC_BUFFER *buffer = NULL;
8432         uint32 offered = q_u->offered;
8433         uint32 *needed = &r_u->needed;
8434         uint32 *returned = &r_u->returned;
8435
8436         /* that's an [in out] buffer */
8437
8438         if (!q_u->buffer) {
8439                 return WERR_INVALID_PARAM;
8440         }
8441
8442         rpcbuf_move(q_u->buffer, &r_u->buffer);
8443         buffer = r_u->buffer;
8444
8445         DEBUG(5,("spoolss_enumprintprocessors\n"));
8446
8447         /*
8448          * Enumerate the print processors ...
8449          *
8450          * Just reply with "winprint", to keep NT happy
8451          * and I can use my nice printer checker.
8452          */
8453
8454         *returned=0;
8455         *needed=0;
8456
8457         switch (level) {
8458         case 1:
8459                 return enumprintprocessors_level_1(buffer, offered, needed, returned);
8460         default:
8461                 return WERR_UNKNOWN_LEVEL;
8462         }
8463 }
8464
8465 /****************************************************************************
8466  enumprintprocdatatypes level 1.
8467 ****************************************************************************/
8468
8469 static WERROR enumprintprocdatatypes_level_1(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
8470 {
8471         PRINTPROCDATATYPE_1 *info_1=NULL;
8472         WERROR result = WERR_NOMEM;
8473
8474         if((info_1 = SMB_MALLOC_P(PRINTPROCDATATYPE_1)) == NULL)
8475                 return WERR_NOMEM;
8476
8477         (*returned) = 0x1;
8478
8479         init_unistr(&info_1->name, "RAW");
8480
8481         *needed += spoolss_size_printprocdatatype_info_1(info_1);
8482
8483         if (*needed > offered) {
8484                 result = WERR_INSUFFICIENT_BUFFER;
8485                 goto out;
8486         }
8487
8488         if (!rpcbuf_alloc_size(buffer, *needed)) {
8489                 result = WERR_NOMEM;
8490                 goto out;
8491         }
8492
8493         smb_io_printprocdatatype_info_1("", buffer, info_1, 0);
8494
8495 out:
8496         SAFE_FREE(info_1);
8497
8498         if ( !W_ERROR_IS_OK(result) )
8499                 *returned = 0;
8500
8501         return result;
8502 }
8503
8504 /****************************************************************************
8505 ****************************************************************************/
8506
8507 WERROR _spoolss_enumprintprocdatatypes(pipes_struct *p, SPOOL_Q_ENUMPRINTPROCDATATYPES *q_u, SPOOL_R_ENUMPRINTPROCDATATYPES *r_u)
8508 {
8509         uint32 level = q_u->level;
8510         RPC_BUFFER *buffer = NULL;
8511         uint32 offered = q_u->offered;
8512         uint32 *needed = &r_u->needed;
8513         uint32 *returned = &r_u->returned;
8514
8515         /* that's an [in out] buffer */
8516
8517         if (!q_u->buffer) {
8518                 return WERR_INVALID_PARAM;
8519         }
8520
8521         rpcbuf_move(q_u->buffer, &r_u->buffer);
8522         buffer = r_u->buffer;
8523
8524         DEBUG(5,("_spoolss_enumprintprocdatatypes\n"));
8525
8526         *returned=0;
8527         *needed=0;
8528
8529         switch (level) {
8530         case 1:
8531                 return enumprintprocdatatypes_level_1(buffer, offered, needed, returned);
8532         default:
8533                 return WERR_UNKNOWN_LEVEL;
8534         }
8535 }
8536
8537 /****************************************************************************
8538  enumprintmonitors level 1.
8539 ****************************************************************************/
8540
8541 static WERROR enumprintmonitors_level_1(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
8542 {
8543         PRINTMONITOR_1 *info_1;
8544         WERROR result = WERR_OK;
8545         int i;
8546
8547         if((info_1 = SMB_MALLOC_ARRAY(PRINTMONITOR_1, 2)) == NULL)
8548                 return WERR_NOMEM;
8549
8550         *returned = 2;
8551
8552         init_unistr(&(info_1[0].name), SPL_LOCAL_PORT );
8553         init_unistr(&(info_1[1].name), SPL_TCPIP_PORT );
8554
8555         for ( i=0; i<*returned; i++ ) {
8556                 *needed += spoolss_size_printmonitor_info_1(&info_1[i]);
8557         }
8558
8559         if (*needed > offered) {
8560                 result = WERR_INSUFFICIENT_BUFFER;
8561                 goto out;
8562         }
8563
8564         if (!rpcbuf_alloc_size(buffer, *needed)) {
8565                 result = WERR_NOMEM;
8566                 goto out;
8567         }
8568
8569         for ( i=0; i<*returned; i++ ) {
8570                 smb_io_printmonitor_info_1("", buffer, &info_1[i], 0);
8571         }
8572
8573 out:
8574         SAFE_FREE(info_1);
8575
8576         if ( !W_ERROR_IS_OK(result) )
8577                 *returned = 0;
8578
8579         return result;
8580 }
8581
8582 /****************************************************************************
8583  enumprintmonitors level 2.
8584 ****************************************************************************/
8585
8586 static WERROR enumprintmonitors_level_2(RPC_BUFFER *buffer, uint32 offered, uint32 *needed, uint32 *returned)
8587 {
8588         PRINTMONITOR_2 *info_2;
8589         WERROR result = WERR_OK;
8590         int i;
8591
8592         if((info_2 = SMB_MALLOC_ARRAY(PRINTMONITOR_2, 2)) == NULL)
8593                 return WERR_NOMEM;
8594
8595         *returned = 2;
8596
8597         init_unistr( &(info_2[0].name), SPL_LOCAL_PORT );
8598         init_unistr( &(info_2[0].environment), "Windows NT X86" );
8599         init_unistr( &(info_2[0].dll_name), "localmon.dll" );
8600
8601         init_unistr( &(info_2[1].name), SPL_TCPIP_PORT );
8602         init_unistr( &(info_2[1].environment), "Windows NT X86" );
8603         init_unistr( &(info_2[1].dll_name), "tcpmon.dll" );
8604
8605         for ( i=0; i<*returned; i++ ) {
8606                 *needed += spoolss_size_printmonitor_info_2(&info_2[i]);
8607         }
8608
8609         if (*needed > offered) {
8610                 result = WERR_INSUFFICIENT_BUFFER;
8611                 goto out;
8612         }
8613
8614         if (!rpcbuf_alloc_size(buffer, *needed)) {
8615                 result = WERR_NOMEM;
8616                 goto out;
8617         }
8618
8619         for ( i=0; i<*returned; i++ ) {
8620                 smb_io_printmonitor_info_2("", buffer, &info_2[i], 0);
8621         }
8622
8623 out:
8624         SAFE_FREE(info_2);
8625
8626         if ( !W_ERROR_IS_OK(result) )
8627                 *returned = 0;
8628
8629         return result;
8630 }
8631
8632 /****************************************************************************
8633 ****************************************************************************/
8634
8635 WERROR _spoolss_enumprintmonitors(pipes_struct *p, SPOOL_Q_ENUMPRINTMONITORS *q_u, SPOOL_R_ENUMPRINTMONITORS *r_u)
8636 {
8637         uint32 level = q_u->level;
8638         RPC_BUFFER *buffer = NULL;
8639         uint32 offered = q_u->offered;
8640         uint32 *needed = &r_u->needed;
8641         uint32 *returned = &r_u->returned;
8642
8643         /* that's an [in out] buffer */
8644
8645         if (!q_u->buffer) {
8646                 return WERR_INVALID_PARAM;
8647         }
8648
8649         rpcbuf_move(q_u->buffer, &r_u->buffer);
8650         buffer = r_u->buffer;
8651
8652         DEBUG(5,("spoolss_enumprintmonitors\n"));
8653
8654         /*
8655          * Enumerate the print monitors ...
8656          *
8657          * Just reply with "Local Port", to keep NT happy
8658          * and I can use my nice printer checker.
8659          */
8660
8661         *returned=0;
8662         *needed=0;
8663
8664         switch (level) {
8665         case 1:
8666                 return enumprintmonitors_level_1(buffer, offered, needed, returned);
8667         case 2:
8668                 return enumprintmonitors_level_2(buffer, offered, needed, returned);
8669         default:
8670                 return WERR_UNKNOWN_LEVEL;
8671         }
8672 }
8673
8674 /****************************************************************************
8675 ****************************************************************************/
8676
8677 static WERROR getjob_level_1(print_queue_struct **queue, int count, int snum,
8678                              NT_PRINTER_INFO_LEVEL *ntprinter,
8679                              uint32 jobid, RPC_BUFFER *buffer, uint32 offered,
8680                              uint32 *needed)
8681 {
8682         int i=0;
8683         BOOL found=False;
8684         JOB_INFO_1 *info_1=NULL;
8685         WERROR result = WERR_OK;
8686
8687         info_1=SMB_MALLOC_P(JOB_INFO_1);
8688
8689         if (info_1 == NULL) {
8690                 return WERR_NOMEM;
8691         }
8692
8693         for (i=0; i<count && found==False; i++) {
8694                 if ((*queue)[i].job==(int)jobid)
8695                         found=True;
8696         }
8697
8698         if (found==False) {
8699                 SAFE_FREE(info_1);
8700                 /* NT treats not found as bad param... yet another bad choice */
8701                 return WERR_INVALID_PARAM;
8702         }
8703
8704         fill_job_info_1( info_1, &((*queue)[i-1]), i, snum, ntprinter );
8705
8706         *needed += spoolss_size_job_info_1(info_1);
8707
8708         if (*needed > offered) {
8709                 result = WERR_INSUFFICIENT_BUFFER;
8710                 goto out;
8711         }
8712
8713         if (!rpcbuf_alloc_size(buffer, *needed)) {
8714                 result = WERR_NOMEM;
8715                 goto out;
8716         }
8717
8718         smb_io_job_info_1("", buffer, info_1, 0);
8719
8720 out:
8721         SAFE_FREE(info_1);
8722
8723         return result;
8724 }
8725
8726 /****************************************************************************
8727 ****************************************************************************/
8728
8729 static WERROR getjob_level_2(print_queue_struct **queue, int count, int snum,
8730                              NT_PRINTER_INFO_LEVEL *ntprinter,
8731                              uint32 jobid, RPC_BUFFER *buffer, uint32 offered,
8732                              uint32 *needed)
8733 {
8734         int             i = 0;
8735         BOOL            found = False;
8736         JOB_INFO_2      *info_2;
8737         WERROR          result;
8738         DEVICEMODE      *devmode = NULL;
8739         NT_DEVICEMODE   *nt_devmode = NULL;
8740
8741         if ( !(info_2=SMB_MALLOC_P(JOB_INFO_2)) )
8742                 return WERR_NOMEM;
8743
8744         ZERO_STRUCTP(info_2);
8745
8746         for ( i=0; i<count && found==False; i++ )
8747         {
8748                 if ((*queue)[i].job == (int)jobid)
8749                         found = True;
8750         }
8751
8752         if ( !found ) {
8753                 /* NT treats not found as bad param... yet another bad
8754                    choice */
8755                 result = WERR_INVALID_PARAM;
8756                 goto done;
8757         }
8758
8759         /*
8760          * if the print job does not have a DEVMODE associated with it,
8761          * just use the one for the printer. A NULL devicemode is not
8762          *  a failure condition
8763          */
8764
8765         if ( !(nt_devmode=print_job_devmode( lp_const_servicename(snum), jobid )) )
8766                 devmode = construct_dev_mode(snum);
8767         else {
8768                 if ((devmode = SMB_MALLOC_P(DEVICEMODE)) != NULL) {
8769                         ZERO_STRUCTP( devmode );
8770                         convert_nt_devicemode( devmode, nt_devmode );
8771                 }
8772         }
8773
8774         fill_job_info_2(info_2, &((*queue)[i-1]), i, snum, ntprinter, devmode);
8775
8776         *needed += spoolss_size_job_info_2(info_2);
8777
8778         if (*needed > offered) {
8779                 result = WERR_INSUFFICIENT_BUFFER;
8780                 goto done;
8781         }
8782
8783         if (!rpcbuf_alloc_size(buffer, *needed)) {
8784                 result = WERR_NOMEM;
8785                 goto done;
8786         }
8787
8788         smb_io_job_info_2("", buffer, info_2, 0);
8789
8790         result = WERR_OK;
8791
8792  done:
8793         /* Cleanup allocated memory */
8794
8795         free_job_info_2(info_2);        /* Also frees devmode */
8796         SAFE_FREE(info_2);
8797
8798         return result;
8799 }
8800
8801 /****************************************************************************
8802 ****************************************************************************/
8803
8804 WERROR _spoolss_getjob( pipes_struct *p, SPOOL_Q_GETJOB *q_u, SPOOL_R_GETJOB *r_u)
8805 {
8806         POLICY_HND *handle = &q_u->handle;
8807         uint32 jobid = q_u->jobid;
8808         uint32 level = q_u->level;
8809         RPC_BUFFER *buffer = NULL;
8810         uint32 offered = q_u->offered;
8811         uint32 *needed = &r_u->needed;
8812         WERROR          wstatus = WERR_OK;
8813         NT_PRINTER_INFO_LEVEL *ntprinter = NULL;
8814         int snum;
8815         int count;
8816         print_queue_struct      *queue = NULL;
8817         print_status_struct prt_status;
8818
8819         /* that's an [in out] buffer */
8820
8821         if (!q_u->buffer) {
8822                 return WERR_INVALID_PARAM;
8823         }
8824
8825         rpcbuf_move(q_u->buffer, &r_u->buffer);
8826         buffer = r_u->buffer;
8827
8828         DEBUG(5,("spoolss_getjob\n"));
8829
8830         *needed = 0;
8831
8832         if (!get_printer_snum(p, handle, &snum))
8833                 return WERR_BADFID;
8834
8835         wstatus = get_a_printer(NULL, &ntprinter, 2, lp_servicename(snum));
8836         if ( !W_ERROR_IS_OK(wstatus) )
8837                 return wstatus;
8838
8839         count = print_queue_status(snum, &queue, &prt_status);
8840
8841         DEBUGADD(4,("count:[%d], prt_status:[%d], [%s]\n",
8842                      count, prt_status.status, prt_status.message));
8843
8844         switch ( level ) {
8845         case 1:
8846                         wstatus = getjob_level_1(&queue, count, snum, ntprinter, jobid,
8847                                 buffer, offered, needed);
8848                         break;
8849         case 2:
8850                         wstatus = getjob_level_2(&queue, count, snum, ntprinter, jobid,
8851                                 buffer, offered, needed);
8852                         break;
8853         default:
8854                         wstatus = WERR_UNKNOWN_LEVEL;
8855                         break;
8856         }
8857
8858         SAFE_FREE(queue);
8859         free_a_printer( &ntprinter, 2 );
8860
8861         return wstatus;
8862 }
8863
8864 /********************************************************************
8865  spoolss_getprinterdataex
8866
8867  From MSDN documentation of GetPrinterDataEx: pass request
8868  to GetPrinterData if key is "PrinterDriverData".
8869  ********************************************************************/
8870
8871 WERROR _spoolss_getprinterdataex(pipes_struct *p, SPOOL_Q_GETPRINTERDATAEX *q_u, SPOOL_R_GETPRINTERDATAEX *r_u)
8872 {
8873         POLICY_HND      *handle = &q_u->handle;
8874         uint32          in_size = q_u->size;
8875         uint32          *type = &r_u->type;
8876         uint32          *out_size = &r_u->size;
8877         uint8           **data = &r_u->data;
8878         uint32          *needed = &r_u->needed;
8879         fstring         keyname, valuename;
8880
8881         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
8882
8883         NT_PRINTER_INFO_LEVEL   *printer = NULL;
8884         int                     snum = 0;
8885         WERROR                  status = WERR_OK;
8886
8887         DEBUG(4,("_spoolss_getprinterdataex\n"));
8888
8889         unistr2_to_ascii(keyname, &q_u->keyname, sizeof(keyname) - 1);
8890         unistr2_to_ascii(valuename, &q_u->valuename, sizeof(valuename) - 1);
8891
8892         DEBUG(10, ("_spoolss_getprinterdataex: key => [%s], value => [%s]\n",
8893                 keyname, valuename));
8894
8895         /* in case of problem, return some default values */
8896
8897         *needed   = 0;
8898         *type     = 0;
8899         *out_size = in_size;
8900
8901         if (!Printer) {
8902                 DEBUG(2,("_spoolss_getprinterdataex: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8903                 status = WERR_BADFID;
8904                 goto done;
8905         }
8906
8907         /* Is the handle to a printer or to the server? */
8908
8909         if (Printer->printer_type == SPLHND_SERVER) {
8910                 DEBUG(10,("_spoolss_getprinterdataex: Not implemented for server handles yet\n"));
8911                 status = WERR_INVALID_PARAM;
8912                 goto done;
8913         }
8914
8915         if ( !get_printer_snum(p,handle, &snum) )
8916                 return WERR_BADFID;
8917
8918         status = get_a_printer(Printer, &printer, 2, lp_servicename(snum));
8919         if ( !W_ERROR_IS_OK(status) )
8920                 goto done;
8921
8922         /* check to see if the keyname is valid */
8923         if ( !strlen(keyname) ) {
8924                 status = WERR_INVALID_PARAM;
8925                 goto done;
8926         }
8927
8928         if ( lookup_printerkey( printer->info_2->data, keyname ) == -1 ) {
8929                 DEBUG(4,("_spoolss_getprinterdataex: Invalid keyname [%s]\n", keyname ));
8930                 free_a_printer( &printer, 2 );
8931                 status = WERR_BADFILE;
8932                 goto done;
8933         }
8934
8935         /* When given a new keyname, we should just create it */
8936
8937         status = get_printer_dataex( p->mem_ctx, printer, keyname, valuename, type, data, needed, in_size );
8938
8939         if (*needed > *out_size)
8940                 status = WERR_MORE_DATA;
8941
8942 done:
8943         if ( !W_ERROR_IS_OK(status) )
8944         {
8945                 DEBUG(5, ("error: allocating %d\n", *out_size));
8946
8947                 /* reply this param doesn't exist */
8948
8949                 if ( *out_size )
8950                 {
8951                         if( (*data=(uint8 *)TALLOC_ZERO(p->mem_ctx, *out_size*sizeof(uint8))) == NULL ) {
8952                                 status = WERR_NOMEM;
8953                                 goto done;
8954                         }
8955                 }
8956                 else {
8957                         *data = NULL;
8958         }
8959         }
8960
8961         if ( printer )
8962         free_a_printer( &printer, 2 );
8963
8964         return status;
8965 }
8966
8967 /********************************************************************
8968  * spoolss_setprinterdataex
8969  ********************************************************************/
8970
8971 WERROR _spoolss_setprinterdataex(pipes_struct *p, SPOOL_Q_SETPRINTERDATAEX *q_u, SPOOL_R_SETPRINTERDATAEX *r_u)
8972 {
8973         POLICY_HND              *handle = &q_u->handle;
8974         uint32                  type = q_u->type;
8975         uint8                   *data = q_u->data;
8976         uint32                  real_len = q_u->real_len;
8977
8978         NT_PRINTER_INFO_LEVEL   *printer = NULL;
8979         int                     snum = 0;
8980         WERROR                  status = WERR_OK;
8981         Printer_entry           *Printer = find_printer_index_by_hnd(p, handle);
8982         fstring                 valuename;
8983         fstring                 keyname;
8984         char                    *oid_string;
8985
8986         DEBUG(4,("_spoolss_setprinterdataex\n"));
8987
8988         /* From MSDN documentation of SetPrinterDataEx: pass request to
8989            SetPrinterData if key is "PrinterDriverData" */
8990
8991         if (!Printer) {
8992                 DEBUG(2,("_spoolss_setprinterdataex: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
8993                 return WERR_BADFID;
8994         }
8995
8996         if ( Printer->printer_type == SPLHND_SERVER ) {
8997                 DEBUG(10,("_spoolss_setprinterdataex: Not implemented for server handles yet\n"));
8998                 return WERR_INVALID_PARAM;
8999         }
9000
9001         if ( !get_printer_snum(p,handle, &snum) )
9002                 return WERR_BADFID;
9003
9004         /*
9005          * Access check : NT returns "access denied" if you make a
9006          * SetPrinterData call without the necessary privildge.
9007          * we were originally returning OK if nothing changed
9008          * which made Win2k issue **a lot** of SetPrinterData
9009          * when connecting to a printer  --jerry
9010          */
9011
9012         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER)
9013         {
9014                 DEBUG(3, ("_spoolss_setprinterdataex: change denied by handle access permissions\n"));
9015                 return WERR_ACCESS_DENIED;
9016         }
9017
9018         status = get_a_printer(Printer, &printer, 2, lp_servicename(snum));
9019         if (!W_ERROR_IS_OK(status))
9020                 return status;
9021
9022         unistr2_to_ascii( valuename, &q_u->value, sizeof(valuename) - 1);
9023         unistr2_to_ascii( keyname, &q_u->key, sizeof(keyname) - 1);
9024
9025         /* check for OID in valuename */
9026
9027         if ( (oid_string = strchr( valuename, ',' )) != NULL )
9028         {
9029                 *oid_string = '\0';
9030                 oid_string++;
9031         }
9032
9033         /* save the registry data */
9034
9035         status = set_printer_dataex( printer, keyname, valuename, type, data, real_len );
9036
9037         if ( W_ERROR_IS_OK(status) )
9038         {
9039                 /* save the OID if one was specified */
9040                 if ( oid_string ) {
9041                         fstrcat( keyname, "\\" );
9042                         fstrcat( keyname, SPOOL_OID_KEY );
9043
9044                         /*
9045                          * I'm not checking the status here on purpose.  Don't know
9046                          * if this is right, but I'm returning the status from the
9047                          * previous set_printer_dataex() call.  I have no idea if
9048                          * this is right.    --jerry
9049                          */
9050
9051                         set_printer_dataex( printer, keyname, valuename,
9052                                             REG_SZ, (void*)oid_string, strlen(oid_string)+1 );
9053                 }
9054
9055                 status = mod_a_printer(printer, 2);
9056         }
9057
9058         free_a_printer(&printer, 2);
9059
9060         return status;
9061 }
9062
9063
9064 /********************************************************************
9065  * spoolss_deleteprinterdataex
9066  ********************************************************************/
9067
9068 WERROR _spoolss_deleteprinterdataex(pipes_struct *p, SPOOL_Q_DELETEPRINTERDATAEX *q_u, SPOOL_R_DELETEPRINTERDATAEX *r_u)
9069 {
9070         POLICY_HND      *handle = &q_u->handle;
9071         UNISTR2         *value = &q_u->valuename;
9072         UNISTR2         *key = &q_u->keyname;
9073
9074         NT_PRINTER_INFO_LEVEL   *printer = NULL;
9075         int             snum=0;
9076         WERROR          status = WERR_OK;
9077         Printer_entry   *Printer=find_printer_index_by_hnd(p, handle);
9078         pstring         valuename, keyname;
9079
9080         DEBUG(5,("spoolss_deleteprinterdataex\n"));
9081
9082         if (!Printer) {
9083                 DEBUG(2,("_spoolss_deleteprinterdata: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
9084                 return WERR_BADFID;
9085         }
9086
9087         if (!get_printer_snum(p, handle, &snum))
9088                 return WERR_BADFID;
9089
9090         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER) {
9091                 DEBUG(3, ("_spoolss_deleteprinterdataex: printer properties change denied by handle\n"));
9092                 return WERR_ACCESS_DENIED;
9093         }
9094
9095         status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
9096         if (!W_ERROR_IS_OK(status))
9097                 return status;
9098
9099         unistr2_to_ascii( valuename, value, sizeof(valuename)-1 );
9100         unistr2_to_ascii( keyname, key, sizeof(keyname)-1 );
9101
9102         status = delete_printer_dataex( printer, keyname, valuename );
9103
9104         if ( W_ERROR_IS_OK(status) )
9105                 mod_a_printer( printer, 2 );
9106
9107         free_a_printer(&printer, 2);
9108
9109         return status;
9110 }
9111
9112 /********************************************************************
9113  * spoolss_enumprinterkey
9114  ********************************************************************/
9115
9116
9117 WERROR _spoolss_enumprinterkey(pipes_struct *p, SPOOL_Q_ENUMPRINTERKEY *q_u, SPOOL_R_ENUMPRINTERKEY *r_u)
9118 {
9119         fstring         key;
9120         fstring         *keynames = NULL;
9121         uint16          *enumkeys = NULL;
9122         int             num_keys;
9123         int             printerkey_len;
9124         POLICY_HND      *handle = &q_u->handle;
9125         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
9126         NT_PRINTER_DATA *data;
9127         NT_PRINTER_INFO_LEVEL   *printer = NULL;
9128         int             snum = 0;
9129         WERROR          status = WERR_BADFILE;
9130
9131
9132         DEBUG(4,("_spoolss_enumprinterkey\n"));
9133
9134         if (!Printer) {
9135                 DEBUG(2,("_spoolss_enumprinterkey: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
9136                 return WERR_BADFID;
9137         }
9138
9139         if ( !get_printer_snum(p,handle, &snum) )
9140                 return WERR_BADFID;
9141
9142         status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
9143         if (!W_ERROR_IS_OK(status))
9144                 return status;
9145
9146         /* get the list of subkey names */
9147
9148         unistr2_to_ascii( key, &q_u->key, sizeof(key)-1 );
9149         data = printer->info_2->data;
9150
9151         num_keys = get_printer_subkeys( data, key, &keynames );
9152
9153         if ( num_keys == -1 ) {
9154                 status = WERR_BADFILE;
9155                 goto done;
9156         }
9157
9158         printerkey_len = init_unistr_array( &enumkeys,  keynames, NULL );
9159
9160         r_u->needed = printerkey_len*2;
9161
9162         if ( q_u->size < r_u->needed ) {
9163                 status = WERR_MORE_DATA;
9164                 goto done;
9165         }
9166
9167         if (!make_spoolss_buffer5(p->mem_ctx, &r_u->keys, printerkey_len, enumkeys)) {
9168                 status = WERR_NOMEM;
9169                 goto done;
9170         }
9171
9172         status = WERR_OK;
9173
9174         if ( q_u->size < r_u->needed )
9175                 status = WERR_MORE_DATA;
9176
9177 done:
9178         free_a_printer( &printer, 2 );
9179         SAFE_FREE( keynames );
9180
9181         return status;
9182 }
9183
9184 /********************************************************************
9185  * spoolss_deleteprinterkey
9186  ********************************************************************/
9187
9188 WERROR _spoolss_deleteprinterkey(pipes_struct *p, SPOOL_Q_DELETEPRINTERKEY *q_u, SPOOL_R_DELETEPRINTERKEY *r_u)
9189 {
9190         POLICY_HND              *handle = &q_u->handle;
9191         Printer_entry           *Printer = find_printer_index_by_hnd(p, &q_u->handle);
9192         fstring                 key;
9193         NT_PRINTER_INFO_LEVEL   *printer = NULL;
9194         int                     snum=0;
9195         WERROR                  status;
9196
9197         DEBUG(5,("spoolss_deleteprinterkey\n"));
9198
9199         if (!Printer) {
9200                 DEBUG(2,("_spoolss_deleteprinterkey: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(handle)));
9201                 return WERR_BADFID;
9202         }
9203
9204         /* if keyname == NULL, return error */
9205
9206         if ( !q_u->keyname.buffer )
9207                 return WERR_INVALID_PARAM;
9208
9209         if (!get_printer_snum(p, handle, &snum))
9210                 return WERR_BADFID;
9211
9212         if (Printer->access_granted != PRINTER_ACCESS_ADMINISTER) {
9213                 DEBUG(3, ("_spoolss_deleteprinterkey: printer properties change denied by handle\n"));
9214                 return WERR_ACCESS_DENIED;
9215         }
9216
9217         status = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
9218         if (!W_ERROR_IS_OK(status))
9219                 return status;
9220
9221         /* delete the key and all subneys */
9222
9223         unistr2_to_ascii(key, &q_u->keyname, sizeof(key) - 1);
9224
9225         status = delete_all_printer_data( printer->info_2, key );
9226
9227         if ( W_ERROR_IS_OK(status) )
9228                 status = mod_a_printer(printer, 2);
9229
9230         free_a_printer( &printer, 2 );
9231
9232         return status;
9233 }
9234
9235
9236 /********************************************************************
9237  * spoolss_enumprinterdataex
9238  ********************************************************************/
9239
9240 WERROR _spoolss_enumprinterdataex(pipes_struct *p, SPOOL_Q_ENUMPRINTERDATAEX *q_u, SPOOL_R_ENUMPRINTERDATAEX *r_u)
9241 {
9242         POLICY_HND      *handle = &q_u->handle;
9243         uint32          in_size = q_u->size;
9244         uint32          num_entries,
9245                         needed;
9246         NT_PRINTER_INFO_LEVEL   *printer = NULL;
9247         PRINTER_ENUM_VALUES     *enum_values = NULL;
9248         NT_PRINTER_DATA         *p_data;
9249         fstring         key;
9250         Printer_entry   *Printer = find_printer_index_by_hnd(p, handle);
9251         int             snum;
9252         WERROR          result;
9253         int             key_index;
9254         int             i;
9255         REGISTRY_VALUE  *val;
9256         char            *value_name;
9257         uint32          data_len;
9258
9259
9260         DEBUG(4,("_spoolss_enumprinterdataex\n"));
9261
9262         if (!Printer) {
9263                 DEBUG(2,("_spoolss_enumprinterdataex: Invalid handle (%s:%u:%u1<).\n", OUR_HANDLE(handle)));
9264                 return WERR_BADFID;
9265         }
9266
9267         /*
9268          * first check for a keyname of NULL or "".  Win2k seems to send
9269          * this a lot and we should send back WERR_INVALID_PARAM
9270          * no need to spend time looking up the printer in this case.
9271          * --jerry
9272          */
9273
9274         unistr2_to_ascii(key, &q_u->key, sizeof(key) - 1);
9275         if ( !strlen(key) ) {
9276                 result = WERR_INVALID_PARAM;
9277                 goto done;
9278         }
9279
9280         /* get the printer off of disk */
9281
9282         if (!get_printer_snum(p,handle, &snum))
9283                 return WERR_BADFID;
9284
9285         ZERO_STRUCT(printer);
9286         result = get_a_printer(Printer, &printer, 2, lp_const_servicename(snum));
9287         if (!W_ERROR_IS_OK(result))
9288                 return result;
9289
9290         /* now look for a match on the key name */
9291
9292         p_data = printer->info_2->data;
9293
9294         unistr2_to_ascii(key, &q_u->key, sizeof(key) - 1);
9295         if ( (key_index = lookup_printerkey( p_data, key)) == -1  )
9296         {
9297                 DEBUG(10,("_spoolss_enumprinterdataex: Unknown keyname [%s]\n", key));
9298                 result = WERR_INVALID_PARAM;
9299                 goto done;
9300         }
9301
9302         result = WERR_OK;
9303         needed = 0;
9304
9305         /* allocate the memory for the array of pointers -- if necessary */
9306
9307         num_entries = regval_ctr_numvals( p_data->keys[key_index].values );
9308         if ( num_entries )
9309         {
9310                 if ( (enum_values=TALLOC_ARRAY(p->mem_ctx, PRINTER_ENUM_VALUES, num_entries)) == NULL )
9311                 {
9312                         DEBUG(0,("_spoolss_enumprinterdataex: talloc() failed to allocate memory for [%lu] bytes!\n",
9313                                 (unsigned long)num_entries*sizeof(PRINTER_ENUM_VALUES)));
9314                         result = WERR_NOMEM;
9315                         goto done;
9316                 }
9317
9318                 memset( enum_values, 0x0, num_entries*sizeof(PRINTER_ENUM_VALUES) );
9319         }
9320
9321         /*
9322          * loop through all params and build the array to pass
9323          * back to the  client
9324          */
9325
9326         for ( i=0; i<num_entries; i++ )
9327         {
9328                 /* lookup the registry value */
9329
9330                 val = regval_ctr_specific_value( p_data->keys[key_index].values, i );
9331                 DEBUG(10,("retrieved value number [%d] [%s]\n", i, regval_name(val) ));
9332
9333                 /* copy the data */
9334
9335                 value_name = regval_name( val );
9336                 init_unistr( &enum_values[i].valuename, value_name );
9337                 enum_values[i].value_len = (strlen(value_name)+1) * 2;
9338                 enum_values[i].type      = regval_type( val );
9339
9340                 data_len = regval_size( val );
9341                 if ( data_len ) {
9342                         if ( !(enum_values[i].data = TALLOC_MEMDUP(p->mem_ctx, regval_data_p(val), data_len)) )
9343                         {
9344                                 DEBUG(0,("talloc_memdup failed to allocate memory [data_len=%d] for data!\n",
9345                                         data_len ));
9346                                 result = WERR_NOMEM;
9347                                 goto done;
9348                         }
9349                 }
9350                 enum_values[i].data_len = data_len;
9351
9352                 /* keep track of the size of the array in bytes */
9353
9354                 needed += spoolss_size_printer_enum_values(&enum_values[i]);
9355         }
9356
9357         /* housekeeping information in the reply */
9358
9359         r_u->needed     = needed;
9360         r_u->returned   = num_entries;
9361
9362         if (needed > in_size) {
9363                 result = WERR_MORE_DATA;
9364                 goto done;
9365         }
9366
9367         /* copy data into the reply */
9368
9369         r_u->ctr.size           = r_u->needed;
9370         r_u->ctr.size_of_array  = r_u->returned;
9371         r_u->ctr.values         = enum_values;
9372
9373
9374
9375 done:
9376         if ( printer )
9377         free_a_printer(&printer, 2);
9378
9379         return result;
9380 }
9381
9382 /****************************************************************************
9383 ****************************************************************************/
9384
9385 static void fill_printprocessordirectory_1(PRINTPROCESSOR_DIRECTORY_1 *info, char *name)
9386 {
9387         init_unistr(&info->name, name);
9388 }
9389
9390 static WERROR getprintprocessordirectory_level_1(UNISTR2 *name,
9391                                                  UNISTR2 *environment,
9392                                                  RPC_BUFFER *buffer,
9393                                                  uint32 offered,
9394                                                  uint32 *needed)
9395 {
9396         pstring path;
9397         pstring long_archi;
9398         PRINTPROCESSOR_DIRECTORY_1 *info=NULL;
9399         WERROR result = WERR_OK;
9400
9401         unistr2_to_ascii(long_archi, environment, sizeof(long_archi)-1);
9402
9403         if (!get_short_archi(long_archi))
9404                 return WERR_INVALID_ENVIRONMENT;
9405
9406         if((info=SMB_MALLOC_P(PRINTPROCESSOR_DIRECTORY_1)) == NULL)
9407                 return WERR_NOMEM;
9408
9409         pstrcpy(path, "C:\\WINNT\\System32\\spool\\PRTPROCS\\W32X86");
9410
9411         fill_printprocessordirectory_1(info, path);
9412
9413         *needed += spoolss_size_printprocessordirectory_info_1(info);
9414
9415         if (*needed > offered) {
9416                 result = WERR_INSUFFICIENT_BUFFER;
9417                 goto out;
9418         }
9419
9420         if (!rpcbuf_alloc_size(buffer, *needed)) {
9421                 result = WERR_INSUFFICIENT_BUFFER;
9422                 goto out;
9423         }
9424
9425         smb_io_printprocessordirectory_1("", buffer, info, 0);
9426
9427 out:
9428         SAFE_FREE(info);
9429
9430         return result;
9431 }
9432
9433 WERROR _spoolss_getprintprocessordirectory(pipes_struct *p, SPOOL_Q_GETPRINTPROCESSORDIRECTORY *q_u, SPOOL_R_GETPRINTPROCESSORDIRECTORY *r_u)
9434 {
9435         uint32 level = q_u->level;
9436         RPC_BUFFER *buffer = NULL;
9437         uint32 offered = q_u->offered;
9438         uint32 *needed = &r_u->needed;
9439         WERROR result;
9440
9441         /* that's an [in out] buffer */
9442
9443         if (!q_u->buffer) {
9444                 return WERR_INVALID_PARAM;
9445         }
9446
9447         rpcbuf_move(q_u->buffer, &r_u->buffer);
9448         buffer = r_u->buffer;
9449
9450         DEBUG(5,("_spoolss_getprintprocessordirectory\n"));
9451
9452         *needed=0;
9453
9454         switch(level) {
9455         case 1:
9456                 result = getprintprocessordirectory_level_1
9457                   (&q_u->name, &q_u->environment, buffer, offered, needed);
9458                 break;
9459         default:
9460                 result = WERR_UNKNOWN_LEVEL;
9461         }
9462
9463         return result;
9464 }
9465
9466 /*******************************************************************
9467  Streams the monitor UI DLL name in UNICODE
9468 *******************************************************************/
9469
9470 static WERROR xcvtcp_monitorui( NT_USER_TOKEN *token, RPC_BUFFER *in,
9471                                 RPC_BUFFER *out, uint32 *needed )
9472 {
9473         const char *dllname = "tcpmonui.dll";
9474
9475         *needed = (strlen(dllname)+1) * 2;
9476
9477         if ( rpcbuf_get_size(out) < *needed ) {
9478                 return WERR_INSUFFICIENT_BUFFER;
9479         }
9480
9481         if ( !make_monitorui_buf( out, dllname ) ) {
9482                 return WERR_NOMEM;
9483         }
9484
9485         return WERR_OK;
9486 }
9487
9488 /*******************************************************************
9489  Create a new TCP/IP port
9490 *******************************************************************/
9491
9492 static WERROR xcvtcp_addport( NT_USER_TOKEN *token, RPC_BUFFER *in,
9493                               RPC_BUFFER *out, uint32 *needed )
9494 {
9495         NT_PORT_DATA_1 port1;
9496         pstring device_uri;
9497
9498         ZERO_STRUCT( port1 );
9499
9500         /* convert to our internal port data structure */
9501
9502         if ( !convert_port_data_1( &port1, in ) ) {
9503                 return WERR_NOMEM;
9504         }
9505
9506         /* create the device URI and call the add_port_hook() */
9507
9508         switch ( port1.protocol ) {
9509         case PORT_PROTOCOL_DIRECT:
9510                 pstr_sprintf( device_uri, "socket://%s:%d/", port1.hostaddr, port1.port );
9511                 break;
9512
9513         case PORT_PROTOCOL_LPR:
9514                 pstr_sprintf( device_uri, "lpr://%s/%s", port1.hostaddr, port1.queue );
9515                 break;
9516
9517         default:
9518                 return WERR_UNKNOWN_PORT;
9519         }
9520
9521         return add_port_hook( token, port1.name, device_uri );
9522 }
9523
9524 /*******************************************************************
9525 *******************************************************************/
9526
9527 struct xcv_api_table xcvtcp_cmds[] = {
9528         { "MonitorUI",  xcvtcp_monitorui },
9529         { "AddPort",    xcvtcp_addport},
9530         { NULL,         NULL }
9531 };
9532
9533 static WERROR process_xcvtcp_command( NT_USER_TOKEN *token, const char *command,
9534                                       RPC_BUFFER *inbuf, RPC_BUFFER *outbuf,
9535                                       uint32 *needed )
9536 {
9537         int i;
9538
9539         DEBUG(10,("process_xcvtcp_command: Received command \"%s\"\n", command));
9540
9541         for ( i=0; xcvtcp_cmds[i].name; i++ ) {
9542                 if ( strcmp( command, xcvtcp_cmds[i].name ) == 0 )
9543                         return xcvtcp_cmds[i].fn( token, inbuf, outbuf, needed );
9544         }
9545
9546         return WERR_BADFUNC;
9547 }
9548
9549 /*******************************************************************
9550 *******************************************************************/
9551 #if 0   /* don't support management using the "Local Port" monitor */
9552
9553 static WERROR xcvlocal_monitorui( NT_USER_TOKEN *token, RPC_BUFFER *in,
9554                                   RPC_BUFFER *out, uint32 *needed )
9555 {
9556         const char *dllname = "localui.dll";
9557
9558         *needed = (strlen(dllname)+1) * 2;
9559
9560         if ( rpcbuf_get_size(out) < *needed ) {
9561                 return WERR_INSUFFICIENT_BUFFER;
9562         }
9563
9564         if ( !make_monitorui_buf( out, dllname )) {
9565                 return WERR_NOMEM;
9566         }
9567
9568         return WERR_OK;
9569 }
9570
9571 /*******************************************************************
9572 *******************************************************************/
9573
9574 struct xcv_api_table xcvlocal_cmds[] = {
9575         { "MonitorUI",  xcvlocal_monitorui },
9576         { NULL,         NULL }
9577 };
9578 #else
9579 struct xcv_api_table xcvlocal_cmds[] = {
9580         { NULL,         NULL }
9581 };
9582 #endif
9583
9584
9585
9586 /*******************************************************************
9587 *******************************************************************/
9588
9589 static WERROR process_xcvlocal_command( NT_USER_TOKEN *token, const char *command,
9590                                         RPC_BUFFER *inbuf, RPC_BUFFER *outbuf,
9591                                         uint32 *needed )
9592 {
9593         int i;
9594
9595         DEBUG(10,("process_xcvlocal_command: Received command \"%s\"\n", command));
9596
9597         for ( i=0; xcvlocal_cmds[i].name; i++ ) {
9598                 if ( strcmp( command, xcvlocal_cmds[i].name ) == 0 )
9599                         return xcvlocal_cmds[i].fn( token, inbuf, outbuf , needed );
9600         }
9601         return WERR_BADFUNC;
9602 }
9603
9604 /*******************************************************************
9605 *******************************************************************/
9606
9607 WERROR _spoolss_xcvdataport(pipes_struct *p, SPOOL_Q_XCVDATAPORT *q_u, SPOOL_R_XCVDATAPORT *r_u)
9608 {
9609         Printer_entry *Printer = find_printer_index_by_hnd(p, &q_u->handle);
9610         fstring command;
9611
9612         if (!Printer) {
9613                 DEBUG(2,("_spoolss_xcvdataport: Invalid handle (%s:%u:%u).\n", OUR_HANDLE(&q_u->handle)));
9614                 return WERR_BADFID;
9615         }
9616
9617         /* Has to be a handle to the TCP/IP port monitor */
9618
9619         if ( !(Printer->printer_type & (SPLHND_PORTMON_LOCAL|SPLHND_PORTMON_TCP)) ) {
9620                 DEBUG(2,("_spoolss_xcvdataport: Call only valid for Port Monitors\n"));
9621                 return WERR_BADFID;
9622         }
9623
9624         /* requires administrative access to the server */
9625
9626         if ( !(Printer->access_granted & SERVER_ACCESS_ADMINISTER) ) {
9627                 DEBUG(2,("_spoolss_xcvdataport: denied by handle permissions.\n"));
9628                 return WERR_ACCESS_DENIED;
9629         }
9630
9631         /* Get the command name.  There's numerous commands supported by the
9632            TCPMON interface. */
9633
9634         rpcstr_pull(command, q_u->dataname.buffer, sizeof(command),
9635                 q_u->dataname.uni_str_len*2, 0);
9636
9637         /* Allocate the outgoing buffer */
9638
9639         rpcbuf_init( &r_u->outdata, q_u->offered, p->mem_ctx );
9640
9641         switch ( Printer->printer_type ) {
9642         case SPLHND_PORTMON_TCP:
9643                 return process_xcvtcp_command( p->pipe_user.nt_user_token, command,
9644                         &q_u->indata, &r_u->outdata, &r_u->needed );
9645         case SPLHND_PORTMON_LOCAL:
9646                 return process_xcvlocal_command( p->pipe_user.nt_user_token, command,
9647                         &q_u->indata, &r_u->outdata, &r_u->needed );
9648         }
9649
9650         return WERR_INVALID_PRINT_MONITOR;
9651 }
9652
9653