s3-spoolss: use architecture in spoolss_MonitorInfo calls consistently
[Samba.git] / source3 / printing / nt_printing.c
blob394a3e5e7e1a4d701343881ae91c9a3dde576fe5
1 /*
2 * Unix SMB/CIFS implementation.
3 * RPC Pipe client / server routines
4 * Copyright (C) Andrew Tridgell 1992-2000,
5 * Copyright (C) Jean François Micouleau 1998-2000.
6 * Copyright (C) Gerald Carter 2002-2005.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "printing/nt_printing_tdb.h"
24 #include "../librpc/gen_ndr/ndr_spoolss.h"
25 #include "rpc_server/spoolss/srv_spoolss_util.h"
26 #include "nt_printing.h"
27 #include "secrets.h"
28 #include "../librpc/gen_ndr/netlogon.h"
29 #include "../libcli/security/security.h"
30 #include "passdb/machine_sid.h"
31 #include "smbd/smbd.h"
32 #include "smbd/globals.h"
33 #include "auth.h"
34 #include "messages.h"
35 #include "rpc_server/spoolss/srv_spoolss_nt.h"
36 #include "rpc_client/cli_winreg_spoolss.h"
38 /* Map generic permissions to printer object specific permissions */
40 const struct generic_mapping printer_generic_mapping = {
41 PRINTER_READ,
42 PRINTER_WRITE,
43 PRINTER_EXECUTE,
44 PRINTER_ALL_ACCESS
47 /* Map generic permissions to print server object specific permissions */
49 const struct generic_mapping printserver_generic_mapping = {
50 SERVER_READ,
51 SERVER_WRITE,
52 SERVER_EXECUTE,
53 SERVER_ALL_ACCESS
56 /* Map generic permissions to job object specific permissions */
58 const struct generic_mapping job_generic_mapping = {
59 JOB_READ,
60 JOB_WRITE,
61 JOB_EXECUTE,
62 JOB_ALL_ACCESS
65 static const struct print_architecture_table_node archi_table[]= {
67 {"Windows 4.0", SPL_ARCH_WIN40, 0 },
68 {"Windows NT x86", SPL_ARCH_W32X86, 2 },
69 {"Windows NT R4000", SPL_ARCH_W32MIPS, 2 },
70 {"Windows NT Alpha_AXP", SPL_ARCH_W32ALPHA, 2 },
71 {"Windows NT PowerPC", SPL_ARCH_W32PPC, 2 },
72 {"Windows IA64", SPL_ARCH_IA64, 3 },
73 {"Windows x64", SPL_ARCH_X64, 3 },
74 {NULL, "", -1 }
77 static bool print_driver_directories_init(void)
79 int service, i;
80 char *driver_path;
81 bool ok;
82 TALLOC_CTX *mem_ctx = talloc_stackframe();
84 service = lp_servicenumber("print$");
85 if (service < 0) {
86 /* We don't have a print$ share */
87 DEBUG(5, ("No print$ share has been configured.\n"));
88 talloc_free(mem_ctx);
89 return true;
92 driver_path = lp_path(mem_ctx, service);
93 if (driver_path == NULL) {
94 talloc_free(mem_ctx);
95 return false;
98 ok = directory_create_or_exist(driver_path, 0755);
99 if (!ok) {
100 DEBUG(1, ("Failed to create printer driver directory %s\n",
101 driver_path));
102 talloc_free(mem_ctx);
103 return false;
106 for (i = 0; archi_table[i].long_archi != NULL; i++) {
107 const char *arch_path;
109 arch_path = talloc_asprintf(mem_ctx,
110 "%s/%s",
111 driver_path,
112 archi_table[i].short_archi);
113 if (arch_path == NULL) {
114 talloc_free(mem_ctx);
115 return false;
118 ok = directory_create_or_exist(arch_path, 0755);
119 if (!ok) {
120 DEBUG(1, ("Failed to create printer driver "
121 "architecture directory %s\n",
122 arch_path));
123 talloc_free(mem_ctx);
124 return false;
128 talloc_free(mem_ctx);
129 return true;
132 /****************************************************************************
133 Forward a MSG_PRINTER_DRVUPGRADE message from another smbd to the
134 background lpq updater.
135 ****************************************************************************/
137 static void forward_drv_upgrade_printer_msg(struct messaging_context *msg,
138 void *private_data,
139 uint32_t msg_type,
140 struct server_id server_id,
141 DATA_BLOB *data)
143 extern pid_t background_lpq_updater_pid;
145 if (background_lpq_updater_pid == -1) {
146 DEBUG(3,("no background lpq queue updater\n"));
147 return;
150 messaging_send_buf(msg,
151 pid_to_procid(background_lpq_updater_pid),
152 MSG_PRINTER_DRVUPGRADE,
153 data->data,
154 data->length);
157 /****************************************************************************
158 Open the NT printing tdbs. Done once before fork().
159 ****************************************************************************/
161 bool nt_printing_init(struct messaging_context *msg_ctx)
163 WERROR win_rc;
165 if (!print_driver_directories_init()) {
166 return false;
169 if (!nt_printing_tdb_upgrade()) {
170 return false;
174 * register callback to handle updating printers as new
175 * drivers are installed. Forwards to background lpq updater.
177 messaging_register(msg_ctx, NULL, MSG_PRINTER_DRVUPGRADE,
178 forward_drv_upgrade_printer_msg);
180 /* of course, none of the message callbacks matter if you don't
181 tell messages.c that you interested in receiving PRINT_GENERAL
182 msgs. This is done in serverid_register() */
184 if ( lp_security() == SEC_ADS ) {
185 win_rc = check_published_printers(msg_ctx);
186 if (!W_ERROR_IS_OK(win_rc))
187 DEBUG(0, ("nt_printing_init: error checking published printers: %s\n", win_errstr(win_rc)));
190 return true;
193 /*******************************************************************
194 Function to allow filename parsing "the old way".
195 ********************************************************************/
197 static NTSTATUS driver_unix_convert(connection_struct *conn,
198 const char *old_name,
199 struct smb_filename **smb_fname)
201 NTSTATUS status;
202 TALLOC_CTX *ctx = talloc_tos();
203 char *name = talloc_strdup(ctx, old_name);
205 if (!name) {
206 return NT_STATUS_NO_MEMORY;
208 unix_format(name);
209 name = unix_clean_name(ctx, name);
210 if (!name) {
211 return NT_STATUS_NO_MEMORY;
213 trim_string(name,"/","/");
215 status = unix_convert(ctx, conn, name, smb_fname, 0);
216 if (!NT_STATUS_IS_OK(status)) {
217 return NT_STATUS_NO_MEMORY;
220 return NT_STATUS_OK;
223 /****************************************************************************
224 Function to do the mapping between the long architecture name and
225 the short one.
226 ****************************************************************************/
228 const char *get_short_archi(const char *long_archi)
230 int i=-1;
232 DEBUG(107,("Getting architecture dependent directory\n"));
233 do {
234 i++;
235 } while ( (archi_table[i].long_archi!=NULL ) &&
236 strcasecmp_m(long_archi, archi_table[i].long_archi) );
238 if (archi_table[i].long_archi==NULL) {
239 DEBUGADD(10,("Unknown architecture [%s] !\n", long_archi));
240 return NULL;
243 /* this might be client code - but shouldn't this be an fstrcpy etc? */
245 DEBUGADD(108,("index: [%d]\n", i));
246 DEBUGADD(108,("long architecture: [%s]\n", archi_table[i].long_archi));
247 DEBUGADD(108,("short architecture: [%s]\n", archi_table[i].short_archi));
249 return archi_table[i].short_archi;
252 /****************************************************************************
253 Version information in Microsoft files is held in a VS_VERSION_INFO structure.
254 There are two case to be covered here: PE (Portable Executable) and NE (New
255 Executable) files. Both files support the same INFO structure, but PE files
256 store the signature in unicode, and NE files store it as !unicode.
257 returns -1 on error, 1 on version info found, and 0 on no version info found.
258 ****************************************************************************/
260 static int get_file_version(files_struct *fsp, char *fname,uint32_t *major, uint32_t *minor)
262 int i;
263 char *buf = NULL;
264 ssize_t byte_count;
266 if ((buf=(char *)SMB_MALLOC(DOS_HEADER_SIZE)) == NULL) {
267 DEBUG(0,("get_file_version: PE file [%s] DOS Header malloc failed bytes = %d\n",
268 fname, DOS_HEADER_SIZE));
269 goto error_exit;
272 if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
273 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %lu\n",
274 fname, (unsigned long)byte_count));
275 goto no_version_info;
278 /* Is this really a DOS header? */
279 if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
280 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
281 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
282 goto no_version_info;
285 /* Skip OEM header (if any) and the DOS stub to start of Windows header */
286 if (SMB_VFS_LSEEK(fsp, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (off_t)-1) {
287 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
288 fname, errno));
289 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
290 goto no_version_info;
293 /* Note: DOS_HEADER_SIZE and NE_HEADER_SIZE are incidentally same */
294 if ((byte_count = vfs_read_data(fsp, buf, NE_HEADER_SIZE)) < NE_HEADER_SIZE) {
295 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %lu\n",
296 fname, (unsigned long)byte_count));
297 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
298 goto no_version_info;
301 /* The header may be a PE (Portable Executable) or an NE (New Executable) */
302 if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
303 unsigned int num_sections;
304 unsigned int section_table_bytes;
306 /* Just skip over optional header to get to section table */
307 if (SMB_VFS_LSEEK(fsp,
308 SVAL(buf,PE_HEADER_OPTIONAL_HEADER_SIZE)-(NE_HEADER_SIZE-PE_HEADER_SIZE),
309 SEEK_CUR) == (off_t)-1) {
310 DEBUG(3,("get_file_version: File [%s] Windows optional header too short, errno = %d\n",
311 fname, errno));
312 goto error_exit;
315 /* get the section table */
316 num_sections = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
317 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
318 if (section_table_bytes == 0)
319 goto error_exit;
321 SAFE_FREE(buf);
322 if ((buf=(char *)SMB_MALLOC(section_table_bytes)) == NULL) {
323 DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
324 fname, section_table_bytes));
325 goto error_exit;
328 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
329 DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %lu\n",
330 fname, (unsigned long)byte_count));
331 goto error_exit;
334 /* Iterate the section table looking for the resource section ".rsrc" */
335 for (i = 0; i < num_sections; i++) {
336 int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
338 if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
339 unsigned int section_pos = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
340 unsigned int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
342 if (section_bytes == 0)
343 goto error_exit;
345 SAFE_FREE(buf);
346 if ((buf=(char *)SMB_MALLOC(section_bytes)) == NULL) {
347 DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
348 fname, section_bytes));
349 goto error_exit;
352 /* Seek to the start of the .rsrc section info */
353 if (SMB_VFS_LSEEK(fsp, section_pos, SEEK_SET) == (off_t)-1) {
354 DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
355 fname, errno));
356 goto error_exit;
359 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
360 DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %lu\n",
361 fname, (unsigned long)byte_count));
362 goto error_exit;
365 if (section_bytes < VS_VERSION_INFO_UNICODE_SIZE)
366 goto error_exit;
368 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
369 /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
370 if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
371 /* Align to next long address */
372 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
374 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
375 *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
376 *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
378 DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
379 fname, *major, *minor,
380 (*major>>16)&0xffff, *major&0xffff,
381 (*minor>>16)&0xffff, *minor&0xffff));
382 SAFE_FREE(buf);
383 return 1;
390 /* Version info not found, fall back to origin date/time */
391 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
392 SAFE_FREE(buf);
393 return 0;
395 } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
396 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
397 DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
398 fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
399 /* At this point, we assume the file is in error. It still could be somthing
400 * else besides a NE file, but it unlikely at this point. */
401 goto error_exit;
404 /* Allocate a bit more space to speed up things */
405 SAFE_FREE(buf);
406 if ((buf=(char *)SMB_MALLOC(VS_NE_BUF_SIZE)) == NULL) {
407 DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes = %d\n",
408 fname, PE_HEADER_SIZE));
409 goto error_exit;
412 /* This is a HACK! I got tired of trying to sort through the messy
413 * 'NE' file format. If anyone wants to clean this up please have at
414 * it, but this works. 'NE' files will eventually fade away. JRR */
415 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
416 /* Cover case that should not occur in a well formed 'NE' .dll file */
417 if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
419 for(i=0; i<byte_count; i++) {
420 /* Fast skip past data that can't possibly match */
421 if (buf[i] != 'V') continue;
423 /* Potential match data crosses buf boundry, move it to beginning
424 * of buf, and fill the buf with as much as it will hold. */
425 if (i>byte_count-VS_VERSION_INFO_SIZE) {
426 int bc;
428 memcpy(buf, &buf[i], byte_count-i);
429 if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
430 (byte_count-i))) < 0) {
432 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
433 fname, errno));
434 goto error_exit;
437 byte_count = bc + (byte_count - i);
438 if (byte_count<VS_VERSION_INFO_SIZE) break;
440 i = 0;
443 /* Check that the full signature string and the magic number that
444 * follows exist (not a perfect solution, but the chances that this
445 * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
446 * twice, as it is simpler to read the code. */
447 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
448 /* Compute skip alignment to next long address */
449 int skip = -(SMB_VFS_LSEEK(fsp, 0, SEEK_CUR) - (byte_count - i) +
450 sizeof(VS_SIGNATURE)) & 3;
451 if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
453 *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
454 *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
455 DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
456 fname, *major, *minor,
457 (*major>>16)&0xffff, *major&0xffff,
458 (*minor>>16)&0xffff, *minor&0xffff));
459 SAFE_FREE(buf);
460 return 1;
465 /* Version info not found, fall back to origin date/time */
466 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
467 SAFE_FREE(buf);
468 return 0;
470 } else
471 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
472 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
473 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
475 no_version_info:
476 SAFE_FREE(buf);
477 return 0;
479 error_exit:
480 SAFE_FREE(buf);
481 return -1;
484 /****************************************************************************
485 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
486 share one or more files. During the MS installation process files are checked
487 to insure that only a newer version of a shared file is installed over an
488 older version. There are several possibilities for this comparison. If there
489 is no previous version, the new one is newer (obviously). If either file is
490 missing the version info structure, compare the creation date (on Unix use
491 the modification date). Otherwise chose the numerically larger version number.
492 ****************************************************************************/
494 static int file_version_is_newer(connection_struct *conn, fstring new_file, fstring old_file)
496 bool use_version = true;
498 uint32_t new_major;
499 uint32_t new_minor;
500 time_t new_create_time;
502 uint32_t old_major;
503 uint32_t old_minor;
504 time_t old_create_time;
506 struct smb_filename *smb_fname = NULL;
507 files_struct *fsp = NULL;
508 SMB_STRUCT_STAT st;
510 NTSTATUS status;
511 int ret;
513 SET_STAT_INVALID(st);
514 new_create_time = (time_t)0;
515 old_create_time = (time_t)0;
517 /* Get file version info (if available) for previous file (if it exists) */
518 status = driver_unix_convert(conn, old_file, &smb_fname);
519 if (!NT_STATUS_IS_OK(status)) {
520 goto error_exit;
523 status = SMB_VFS_CREATE_FILE(
524 conn, /* conn */
525 NULL, /* req */
526 0, /* root_dir_fid */
527 smb_fname, /* fname */
528 FILE_GENERIC_READ, /* access_mask */
529 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
530 FILE_OPEN, /* create_disposition*/
531 0, /* create_options */
532 FILE_ATTRIBUTE_NORMAL, /* file_attributes */
533 INTERNAL_OPEN_ONLY, /* oplock_request */
534 NULL, /* lease */
535 0, /* allocation_size */
536 0, /* private_flags */
537 NULL, /* sd */
538 NULL, /* ea_list */
539 &fsp, /* result */
540 NULL, /* pinfo */
541 NULL, NULL); /* create context */
543 if (!NT_STATUS_IS_OK(status)) {
544 /* Old file not found, so by definition new file is in fact newer */
545 DEBUG(10,("file_version_is_newer: Can't open old file [%s], "
546 "errno = %d\n", smb_fname_str_dbg(smb_fname),
547 errno));
548 ret = 1;
549 goto done;
551 } else {
552 ret = get_file_version(fsp, old_file, &old_major, &old_minor);
553 if (ret == -1) {
554 goto error_exit;
557 if (!ret) {
558 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
559 old_file));
560 use_version = false;
561 if (SMB_VFS_FSTAT(fsp, &st) == -1) {
562 goto error_exit;
564 old_create_time = convert_timespec_to_time_t(st.st_ex_mtime);
565 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n",
566 (long)old_create_time));
569 close_file(NULL, fsp, NORMAL_CLOSE);
570 fsp = NULL;
572 /* Get file version info (if available) for new file */
573 status = driver_unix_convert(conn, new_file, &smb_fname);
574 if (!NT_STATUS_IS_OK(status)) {
575 goto error_exit;
578 status = SMB_VFS_CREATE_FILE(
579 conn, /* conn */
580 NULL, /* req */
581 0, /* root_dir_fid */
582 smb_fname, /* fname */
583 FILE_GENERIC_READ, /* access_mask */
584 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
585 FILE_OPEN, /* create_disposition*/
586 0, /* create_options */
587 FILE_ATTRIBUTE_NORMAL, /* file_attributes */
588 INTERNAL_OPEN_ONLY, /* oplock_request */
589 NULL, /* lease */
590 0, /* allocation_size */
591 0, /* private_flags */
592 NULL, /* sd */
593 NULL, /* ea_list */
594 &fsp, /* result */
595 NULL, /* pinfo */
596 NULL, NULL); /* create context */
598 if (!NT_STATUS_IS_OK(status)) {
599 /* New file not found, this shouldn't occur if the caller did its job */
600 DEBUG(3,("file_version_is_newer: Can't open new file [%s], "
601 "errno = %d\n", smb_fname_str_dbg(smb_fname), errno));
602 goto error_exit;
604 } else {
605 ret = get_file_version(fsp, new_file, &new_major, &new_minor);
606 if (ret == -1) {
607 goto error_exit;
610 if (!ret) {
611 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
612 new_file));
613 use_version = false;
614 if (SMB_VFS_FSTAT(fsp, &st) == -1) {
615 goto error_exit;
617 new_create_time = convert_timespec_to_time_t(st.st_ex_mtime);
618 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n",
619 (long)new_create_time));
622 close_file(NULL, fsp, NORMAL_CLOSE);
623 fsp = NULL;
625 if (use_version && (new_major != old_major || new_minor != old_minor)) {
626 /* Compare versions and choose the larger version number */
627 if (new_major > old_major ||
628 (new_major == old_major && new_minor > old_minor)) {
630 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
631 ret = 1;
632 goto done;
634 else {
635 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
636 ret = 0;
637 goto done;
640 } else {
641 /* Compare modification time/dates and choose the newest time/date */
642 if (new_create_time > old_create_time) {
643 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
644 ret = 1;
645 goto done;
647 else {
648 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
649 ret = 0;
650 goto done;
654 error_exit:
655 if(fsp)
656 close_file(NULL, fsp, NORMAL_CLOSE);
657 ret = -1;
658 done:
659 TALLOC_FREE(smb_fname);
660 return ret;
663 /****************************************************************************
664 Determine the correct cVersion associated with an architecture and driver
665 ****************************************************************************/
666 static uint32_t get_correct_cversion(struct auth_session_info *session_info,
667 const char *architecture,
668 const char *driverpath_in,
669 WERROR *perr)
671 int cversion = -1;
672 NTSTATUS nt_status;
673 struct smb_filename *smb_fname = NULL;
674 char *driverpath = NULL;
675 files_struct *fsp = NULL;
676 connection_struct *conn = NULL;
677 char *oldcwd;
678 char *printdollar = NULL;
679 int printdollar_snum;
681 *perr = WERR_INVALID_PARAMETER;
683 /* If architecture is Windows 95/98/ME, the version is always 0. */
684 if (strcmp(architecture, SPL_ARCH_WIN40) == 0) {
685 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
686 *perr = WERR_OK;
687 return 0;
690 /* If architecture is Windows x64, the version is always 3. */
691 if (strcmp(architecture, SPL_ARCH_X64) == 0) {
692 DEBUG(10,("get_correct_cversion: Driver is x64, cversion = 3\n"));
693 *perr = WERR_OK;
694 return 3;
697 printdollar_snum = find_service(talloc_tos(), "print$", &printdollar);
698 if (!printdollar) {
699 *perr = WERR_NOT_ENOUGH_MEMORY;
700 return -1;
702 if (printdollar_snum == -1) {
703 *perr = WERR_BAD_NET_NAME;
704 return -1;
707 nt_status = create_conn_struct_cwd(talloc_tos(),
708 server_event_context(),
709 server_messaging_context(),
710 &conn,
711 printdollar_snum,
712 lp_path(talloc_tos(), printdollar_snum),
713 session_info, &oldcwd);
714 if (!NT_STATUS_IS_OK(nt_status)) {
715 DEBUG(0,("get_correct_cversion: create_conn_struct "
716 "returned %s\n", nt_errstr(nt_status)));
717 *perr = ntstatus_to_werror(nt_status);
718 return -1;
721 nt_status = set_conn_force_user_group(conn, printdollar_snum);
722 if (!NT_STATUS_IS_OK(nt_status)) {
723 DEBUG(0, ("failed set force user / group\n"));
724 *perr = ntstatus_to_werror(nt_status);
725 goto error_free_conn;
728 if (!become_user_by_session(conn, session_info)) {
729 DEBUG(0, ("failed to become user\n"));
730 *perr = WERR_ACCESS_DENIED;
731 goto error_free_conn;
734 /* Open the driver file (Portable Executable format) and determine the
735 * deriver the cversion. */
736 driverpath = talloc_asprintf(talloc_tos(),
737 "%s/%s",
738 architecture,
739 driverpath_in);
740 if (!driverpath) {
741 *perr = WERR_NOT_ENOUGH_MEMORY;
742 goto error_exit;
745 nt_status = driver_unix_convert(conn, driverpath, &smb_fname);
746 if (!NT_STATUS_IS_OK(nt_status)) {
747 *perr = ntstatus_to_werror(nt_status);
748 goto error_exit;
751 nt_status = vfs_file_exist(conn, smb_fname);
752 if (!NT_STATUS_IS_OK(nt_status)) {
753 DEBUG(3,("get_correct_cversion: vfs_file_exist failed\n"));
754 *perr = WERR_FILE_NOT_FOUND;
755 goto error_exit;
758 nt_status = SMB_VFS_CREATE_FILE(
759 conn, /* conn */
760 NULL, /* req */
761 0, /* root_dir_fid */
762 smb_fname, /* fname */
763 FILE_GENERIC_READ, /* access_mask */
764 FILE_SHARE_READ | FILE_SHARE_WRITE, /* share_access */
765 FILE_OPEN, /* create_disposition*/
766 0, /* create_options */
767 FILE_ATTRIBUTE_NORMAL, /* file_attributes */
768 INTERNAL_OPEN_ONLY, /* oplock_request */
769 NULL, /* lease */
770 0, /* private_flags */
771 0, /* allocation_size */
772 NULL, /* sd */
773 NULL, /* ea_list */
774 &fsp, /* result */
775 NULL, /* pinfo */
776 NULL, NULL); /* create context */
778 if (!NT_STATUS_IS_OK(nt_status)) {
779 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = "
780 "%d\n", smb_fname_str_dbg(smb_fname), errno));
781 *perr = WERR_ACCESS_DENIED;
782 goto error_exit;
783 } else {
784 uint32_t major;
785 uint32_t minor;
786 int ret;
788 ret = get_file_version(fsp, smb_fname->base_name, &major, &minor);
789 if (ret == -1) {
790 *perr = WERR_INVALID_PARAMETER;
791 goto error_exit;
792 } else if (!ret) {
793 DEBUG(6,("get_correct_cversion: Version info not "
794 "found [%s]\n",
795 smb_fname_str_dbg(smb_fname)));
796 *perr = WERR_INVALID_PARAMETER;
797 goto error_exit;
801 * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
802 * for more details. Version in this case is not just the version of the
803 * file, but the version in the sense of kernal mode (2) vs. user mode
804 * (3) drivers. Other bits of the version fields are the version info.
805 * JRR 010716
807 cversion = major & 0x0000ffff;
808 switch (cversion) {
809 case 2: /* WinNT drivers */
810 case 3: /* Win2K drivers */
811 break;
813 default:
814 DEBUG(6,("get_correct_cversion: cversion "
815 "invalid [%s] cversion = %d\n",
816 smb_fname_str_dbg(smb_fname),
817 cversion));
818 goto error_exit;
821 DEBUG(10,("get_correct_cversion: Version info found [%s] major"
822 " = 0x%x minor = 0x%x\n",
823 smb_fname_str_dbg(smb_fname), major, minor));
826 DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
827 smb_fname_str_dbg(smb_fname), cversion));
828 *perr = WERR_OK;
830 error_exit:
831 unbecome_user();
832 error_free_conn:
833 TALLOC_FREE(smb_fname);
834 if (fsp != NULL) {
835 close_file(NULL, fsp, NORMAL_CLOSE);
837 if (conn != NULL) {
838 vfs_ChDir(conn, oldcwd);
839 SMB_VFS_DISCONNECT(conn);
840 conn_free(conn);
842 if (!W_ERROR_IS_OK(*perr)) {
843 cversion = -1;
846 return cversion;
849 /****************************************************************************
850 ****************************************************************************/
852 #define strip_driver_path(_mem_ctx, _element) do { \
853 if (_element && ((_p = strrchr((_element), '\\')) != NULL)) { \
854 (_element) = talloc_asprintf((_mem_ctx), "%s", _p+1); \
855 W_ERROR_HAVE_NO_MEMORY((_element)); \
857 } while (0);
859 static WERROR clean_up_driver_struct_level(TALLOC_CTX *mem_ctx,
860 struct auth_session_info *session_info,
861 const char *architecture,
862 const char **driver_path,
863 const char **data_file,
864 const char **config_file,
865 const char **help_file,
866 struct spoolss_StringArray *dependent_files,
867 enum spoolss_DriverOSVersion *version,
868 uint32_t flags,
869 const char **driver_directory)
871 const char *short_architecture;
872 int i;
873 WERROR err;
874 char *_p;
876 if (!*driver_path || !*data_file) {
877 return WERR_INVALID_PARAMETER;
880 if (!strequal(architecture, SPOOLSS_ARCHITECTURE_4_0) && !*config_file) {
881 return WERR_INVALID_PARAMETER;
884 if (flags & APD_COPY_FROM_DIRECTORY) {
885 char *path;
886 char *q;
889 * driver_path is set to:
891 * \\PRINTSRV\print$\x64\{279245b0-a8bd-4431-bf6f-baee92ac15c0}\pscript5.dll
893 path = talloc_strdup(mem_ctx, *driver_path);
894 if (path == NULL) {
895 return WERR_NOT_ENOUGH_MEMORY;
898 /* Remove pscript5.dll */
899 q = strrchr_m(path, '\\');
900 if (q == NULL) {
901 return WERR_INVALID_PARAMETER;
903 *q = '\0';
905 /* Get \{279245b0-a8bd-4431-bf6f-baee92ac15c0} */
906 q = strrchr_m(path, '\\');
907 if (q == NULL) {
908 return WERR_INVALID_PARAMETER;
912 * Set driver_directory to:
914 * {279245b0-a8bd-4431-bf6f-baee92ac15c0}
916 * This is the directory where all the files have been uploaded
918 *driver_directory = q + 1;
921 /* clean up the driver name.
922 * we can get .\driver.dll
923 * or worse c:\windows\system\driver.dll !
925 /* using an intermediate string to not have overlaping memcpy()'s */
927 strip_driver_path(mem_ctx, *driver_path);
928 strip_driver_path(mem_ctx, *data_file);
929 if (*config_file) {
930 strip_driver_path(mem_ctx, *config_file);
932 if (help_file) {
933 strip_driver_path(mem_ctx, *help_file);
936 if (dependent_files && dependent_files->string) {
937 for (i=0; dependent_files->string[i]; i++) {
938 strip_driver_path(mem_ctx, dependent_files->string[i]);
942 short_architecture = get_short_archi(architecture);
943 if (!short_architecture) {
944 return WERR_UNKNOWN_PRINTER_DRIVER;
947 /* jfm:7/16/2000 the client always sends the cversion=0.
948 * The server should check which version the driver is by reading
949 * the PE header of driver->driverpath.
951 * For Windows 95/98 the version is 0 (so the value sent is correct)
952 * For Windows NT (the architecture doesn't matter)
953 * NT 3.1: cversion=0
954 * NT 3.5/3.51: cversion=1
955 * NT 4: cversion=2
956 * NT2K: cversion=3
959 *version = get_correct_cversion(session_info, short_architecture,
960 *driver_path, &err);
961 if (*version == -1) {
962 return err;
965 return WERR_OK;
968 /****************************************************************************
969 ****************************************************************************/
971 WERROR clean_up_driver_struct(TALLOC_CTX *mem_ctx,
972 struct auth_session_info *session_info,
973 struct spoolss_AddDriverInfoCtr *r,
974 uint32_t flags,
975 const char **driver_directory)
977 switch (r->level) {
978 case 3:
979 return clean_up_driver_struct_level(mem_ctx, session_info,
980 r->info.info3->architecture,
981 &r->info.info3->driver_path,
982 &r->info.info3->data_file,
983 &r->info.info3->config_file,
984 &r->info.info3->help_file,
985 r->info.info3->dependent_files,
986 &r->info.info3->version,
987 flags,
988 driver_directory);
989 case 6:
990 return clean_up_driver_struct_level(mem_ctx, session_info,
991 r->info.info6->architecture,
992 &r->info.info6->driver_path,
993 &r->info.info6->data_file,
994 &r->info.info6->config_file,
995 &r->info.info6->help_file,
996 r->info.info6->dependent_files,
997 &r->info.info6->version,
998 flags,
999 driver_directory);
1000 case 8:
1001 return clean_up_driver_struct_level(mem_ctx, session_info,
1002 r->info.info8->architecture,
1003 &r->info.info8->driver_path,
1004 &r->info.info8->data_file,
1005 &r->info.info8->config_file,
1006 &r->info.info8->help_file,
1007 r->info.info8->dependent_files,
1008 &r->info.info8->version,
1009 flags,
1010 driver_directory);
1011 default:
1012 return WERR_NOT_SUPPORTED;
1016 /****************************************************************************
1017 This function sucks and should be replaced. JRA.
1018 ****************************************************************************/
1020 static void convert_level_6_to_level3(struct spoolss_AddDriverInfo3 *dst,
1021 const struct spoolss_AddDriverInfo6 *src)
1023 dst->version = src->version;
1025 dst->driver_name = src->driver_name;
1026 dst->architecture = src->architecture;
1027 dst->driver_path = src->driver_path;
1028 dst->data_file = src->data_file;
1029 dst->config_file = src->config_file;
1030 dst->help_file = src->help_file;
1031 dst->monitor_name = src->monitor_name;
1032 dst->default_datatype = src->default_datatype;
1033 dst->_ndr_size_dependent_files = src->_ndr_size_dependent_files;
1034 dst->dependent_files = src->dependent_files;
1037 static void convert_level_8_to_level3(struct spoolss_AddDriverInfo3 *dst,
1038 const struct spoolss_AddDriverInfo8 *src)
1040 dst->version = src->version;
1042 dst->driver_name = src->driver_name;
1043 dst->architecture = src->architecture;
1044 dst->driver_path = src->driver_path;
1045 dst->data_file = src->data_file;
1046 dst->config_file = src->config_file;
1047 dst->help_file = src->help_file;
1048 dst->monitor_name = src->monitor_name;
1049 dst->default_datatype = src->default_datatype;
1050 dst->_ndr_size_dependent_files = src->_ndr_size_dependent_files;
1051 dst->dependent_files = src->dependent_files;
1054 /****************************************************************************
1055 ****************************************************************************/
1057 static WERROR move_driver_file_to_download_area(TALLOC_CTX *mem_ctx,
1058 connection_struct *conn,
1059 const char *driver_file,
1060 const char *short_architecture,
1061 uint32_t driver_version,
1062 uint32_t version,
1063 const char *driver_directory)
1065 struct smb_filename *smb_fname_old = NULL;
1066 struct smb_filename *smb_fname_new = NULL;
1067 char *old_name = NULL;
1068 char *new_name = NULL;
1069 NTSTATUS status;
1070 WERROR ret;
1072 if (driver_directory != NULL) {
1073 old_name = talloc_asprintf(mem_ctx,
1074 "%s/%s/%s",
1075 short_architecture,
1076 driver_directory,
1077 driver_file);
1078 } else {
1079 old_name = talloc_asprintf(mem_ctx,
1080 "%s/%s",
1081 short_architecture,
1082 driver_file);
1084 if (old_name == NULL) {
1085 return WERR_NOT_ENOUGH_MEMORY;
1088 new_name = talloc_asprintf(mem_ctx, "%s/%d/%s",
1089 short_architecture, driver_version, driver_file);
1090 if (new_name == NULL) {
1091 TALLOC_FREE(old_name);
1092 return WERR_NOT_ENOUGH_MEMORY;
1095 if (version != -1 && (version = file_version_is_newer(conn, old_name, new_name)) > 0) {
1097 status = driver_unix_convert(conn, old_name, &smb_fname_old);
1098 if (!NT_STATUS_IS_OK(status)) {
1099 ret = WERR_NOT_ENOUGH_MEMORY;
1100 goto out;
1103 /* Setup a synthetic smb_filename struct */
1104 smb_fname_new = talloc_zero(mem_ctx, struct smb_filename);
1105 if (!smb_fname_new) {
1106 ret = WERR_NOT_ENOUGH_MEMORY;
1107 goto out;
1110 smb_fname_new->base_name = new_name;
1112 DEBUG(10,("move_driver_file_to_download_area: copying '%s' to "
1113 "'%s'\n", smb_fname_old->base_name,
1114 smb_fname_new->base_name));
1116 status = copy_file(mem_ctx, conn, smb_fname_old, smb_fname_new,
1117 OPENX_FILE_EXISTS_TRUNCATE |
1118 OPENX_FILE_CREATE_IF_NOT_EXIST,
1119 0, false);
1121 if (!NT_STATUS_IS_OK(status)) {
1122 DEBUG(0,("move_driver_file_to_download_area: Unable "
1123 "to rename [%s] to [%s]: %s\n",
1124 smb_fname_old->base_name, new_name,
1125 nt_errstr(status)));
1126 ret = WERR_ACCESS_DENIED;
1127 goto out;
1131 ret = WERR_OK;
1132 out:
1133 TALLOC_FREE(smb_fname_old);
1134 TALLOC_FREE(smb_fname_new);
1135 return ret;
1138 WERROR move_driver_to_download_area(struct auth_session_info *session_info,
1139 struct spoolss_AddDriverInfoCtr *r,
1140 const char *driver_directory)
1142 struct spoolss_AddDriverInfo3 *driver;
1143 struct spoolss_AddDriverInfo3 converted_driver;
1144 const char *short_architecture;
1145 struct smb_filename *smb_dname = NULL;
1146 char *new_dir = NULL;
1147 connection_struct *conn = NULL;
1148 NTSTATUS nt_status;
1149 int i;
1150 TALLOC_CTX *ctx = talloc_tos();
1151 int ver = 0;
1152 char *oldcwd;
1153 char *printdollar = NULL;
1154 int printdollar_snum;
1155 WERROR err = WERR_OK;
1157 switch (r->level) {
1158 case 3:
1159 driver = r->info.info3;
1160 break;
1161 case 6:
1162 convert_level_6_to_level3(&converted_driver, r->info.info6);
1163 driver = &converted_driver;
1164 break;
1165 case 8:
1166 convert_level_8_to_level3(&converted_driver, r->info.info8);
1167 driver = &converted_driver;
1168 break;
1169 default:
1170 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)r->level));
1171 return WERR_INVALID_LEVEL;
1174 short_architecture = get_short_archi(driver->architecture);
1175 if (!short_architecture) {
1176 return WERR_UNKNOWN_PRINTER_DRIVER;
1179 printdollar_snum = find_service(ctx, "print$", &printdollar);
1180 if (!printdollar) {
1181 return WERR_NOT_ENOUGH_MEMORY;
1183 if (printdollar_snum == -1) {
1184 return WERR_BAD_NET_NAME;
1187 nt_status = create_conn_struct_cwd(talloc_tos(),
1188 server_event_context(),
1189 server_messaging_context(),
1190 &conn,
1191 printdollar_snum,
1192 lp_path(talloc_tos(), printdollar_snum),
1193 session_info, &oldcwd);
1194 if (!NT_STATUS_IS_OK(nt_status)) {
1195 DEBUG(0,("move_driver_to_download_area: create_conn_struct "
1196 "returned %s\n", nt_errstr(nt_status)));
1197 err = ntstatus_to_werror(nt_status);
1198 return err;
1201 nt_status = set_conn_force_user_group(conn, printdollar_snum);
1202 if (!NT_STATUS_IS_OK(nt_status)) {
1203 DEBUG(0, ("failed set force user / group\n"));
1204 err = ntstatus_to_werror(nt_status);
1205 goto err_free_conn;
1208 if (!become_user_by_session(conn, session_info)) {
1209 DEBUG(0, ("failed to become user\n"));
1210 err = WERR_ACCESS_DENIED;
1211 goto err_free_conn;
1214 new_dir = talloc_asprintf(ctx,
1215 "%s/%d",
1216 short_architecture,
1217 driver->version);
1218 if (!new_dir) {
1219 err = WERR_NOT_ENOUGH_MEMORY;
1220 goto err_exit;
1222 nt_status = driver_unix_convert(conn, new_dir, &smb_dname);
1223 if (!NT_STATUS_IS_OK(nt_status)) {
1224 err = WERR_NOT_ENOUGH_MEMORY;
1225 goto err_exit;
1228 DEBUG(5,("Creating first directory: %s\n", smb_dname->base_name));
1230 nt_status = create_directory(conn, NULL, smb_dname);
1231 if (!NT_STATUS_IS_OK(nt_status)
1232 && !NT_STATUS_EQUAL(nt_status, NT_STATUS_OBJECT_NAME_COLLISION)) {
1233 DEBUG(0, ("failed to create driver destination directory: %s\n",
1234 nt_errstr(nt_status)));
1235 err = ntstatus_to_werror(nt_status);
1236 goto err_exit;
1239 /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1240 * listed for this driver which has already been moved, skip it (note:
1241 * drivers may list the same file name several times. Then check if the
1242 * file already exists in archi\version\, if so, check that the version
1243 * info (or time stamps if version info is unavailable) is newer (or the
1244 * date is later). If it is, move it to archi\version\filexxx.yyy.
1245 * Otherwise, delete the file.
1247 * If a file is not moved to archi\version\ because of an error, all the
1248 * rest of the 'unmoved' driver files are removed from archi\. If one or
1249 * more of the driver's files was already moved to archi\version\, it
1250 * potentially leaves the driver in a partially updated state. Version
1251 * trauma will most likely occur if an client attempts to use any printer
1252 * bound to the driver. Perhaps a rewrite to make sure the moves can be
1253 * done is appropriate... later JRR
1256 DEBUG(5,("Moving files now !\n"));
1258 if (driver->driver_path && strlen(driver->driver_path)) {
1260 err = move_driver_file_to_download_area(ctx,
1261 conn,
1262 driver->driver_path,
1263 short_architecture,
1264 driver->version,
1265 ver,
1266 driver_directory);
1267 if (!W_ERROR_IS_OK(err)) {
1268 goto err_exit;
1272 if (driver->data_file && strlen(driver->data_file)) {
1273 if (!strequal(driver->data_file, driver->driver_path)) {
1275 err = move_driver_file_to_download_area(ctx,
1276 conn,
1277 driver->data_file,
1278 short_architecture,
1279 driver->version,
1280 ver,
1281 driver_directory);
1282 if (!W_ERROR_IS_OK(err)) {
1283 goto err_exit;
1288 if (driver->config_file && strlen(driver->config_file)) {
1289 if (!strequal(driver->config_file, driver->driver_path) &&
1290 !strequal(driver->config_file, driver->data_file)) {
1292 err = move_driver_file_to_download_area(ctx,
1293 conn,
1294 driver->config_file,
1295 short_architecture,
1296 driver->version,
1297 ver,
1298 driver_directory);
1299 if (!W_ERROR_IS_OK(err)) {
1300 goto err_exit;
1305 if (driver->help_file && strlen(driver->help_file)) {
1306 if (!strequal(driver->help_file, driver->driver_path) &&
1307 !strequal(driver->help_file, driver->data_file) &&
1308 !strequal(driver->help_file, driver->config_file)) {
1310 err = move_driver_file_to_download_area(ctx,
1311 conn,
1312 driver->help_file,
1313 short_architecture,
1314 driver->version,
1315 ver,
1316 driver_directory);
1317 if (!W_ERROR_IS_OK(err)) {
1318 goto err_exit;
1323 if (driver->dependent_files && driver->dependent_files->string) {
1324 for (i=0; driver->dependent_files->string[i]; i++) {
1325 if (!strequal(driver->dependent_files->string[i], driver->driver_path) &&
1326 !strequal(driver->dependent_files->string[i], driver->data_file) &&
1327 !strequal(driver->dependent_files->string[i], driver->config_file) &&
1328 !strequal(driver->dependent_files->string[i], driver->help_file)) {
1329 int j;
1330 for (j=0; j < i; j++) {
1331 if (strequal(driver->dependent_files->string[i], driver->dependent_files->string[j])) {
1332 goto NextDriver;
1336 err = move_driver_file_to_download_area(ctx,
1337 conn,
1338 driver->dependent_files->string[i],
1339 short_architecture,
1340 driver->version,
1341 ver,
1342 driver_directory);
1343 if (!W_ERROR_IS_OK(err)) {
1344 goto err_exit;
1347 NextDriver: ;
1351 err = WERR_OK;
1352 err_exit:
1353 unbecome_user();
1354 err_free_conn:
1355 TALLOC_FREE(smb_dname);
1357 if (conn != NULL) {
1358 vfs_ChDir(conn, oldcwd);
1359 SMB_VFS_DISCONNECT(conn);
1360 conn_free(conn);
1363 return err;
1366 /****************************************************************************
1367 Determine whether or not a particular driver is currently assigned
1368 to a printer
1369 ****************************************************************************/
1371 bool printer_driver_in_use(TALLOC_CTX *mem_ctx,
1372 struct dcerpc_binding_handle *b,
1373 const struct spoolss_DriverInfo8 *r)
1375 int snum;
1376 int n_services = lp_numservices();
1377 bool in_use = False;
1378 struct spoolss_PrinterInfo2 *pinfo2 = NULL;
1379 WERROR result;
1381 if (!r) {
1382 return false;
1385 DEBUG(10,("printer_driver_in_use: Beginning search through ntprinters.tdb...\n"));
1387 /* loop through the printers.tdb and check for the drivername */
1389 for (snum=0; snum<n_services && !in_use; snum++) {
1390 if (!lp_snum_ok(snum) || !lp_printable(snum)) {
1391 continue;
1394 result = winreg_get_printer(mem_ctx, b,
1395 lp_servicename(talloc_tos(), snum),
1396 &pinfo2);
1397 if (!W_ERROR_IS_OK(result)) {
1398 continue; /* skip */
1401 if (strequal(r->driver_name, pinfo2->drivername)) {
1402 in_use = True;
1405 TALLOC_FREE(pinfo2);
1408 DEBUG(10,("printer_driver_in_use: Completed search through ntprinters.tdb...\n"));
1410 if ( in_use ) {
1411 struct spoolss_DriverInfo8 *driver = NULL;
1412 WERROR werr;
1414 DEBUG(5,("printer_driver_in_use: driver \"%s\" is currently in use\n", r->driver_name));
1416 /* we can still remove the driver if there is one of
1417 "Windows NT x86" version 2 or 3 left */
1419 if (!strequal("Windows NT x86", r->architecture)) {
1420 werr = winreg_get_driver(mem_ctx, b,
1421 "Windows NT x86",
1422 r->driver_name,
1423 DRIVER_ANY_VERSION,
1424 &driver);
1425 } else if (r->version == 2) {
1426 werr = winreg_get_driver(mem_ctx, b,
1427 "Windows NT x86",
1428 r->driver_name,
1429 3, &driver);
1430 } else if (r->version == 3) {
1431 werr = winreg_get_driver(mem_ctx, b,
1432 "Windows NT x86",
1433 r->driver_name,
1434 2, &driver);
1435 } else {
1436 DEBUG(0, ("printer_driver_in_use: ERROR!"
1437 " unknown driver version (%d)\n",
1438 r->version));
1439 werr = WERR_UNKNOWN_PRINTER_DRIVER;
1442 /* now check the error code */
1444 if ( W_ERROR_IS_OK(werr) ) {
1445 /* it's ok to remove the driver, we have other architctures left */
1446 in_use = False;
1447 talloc_free(driver);
1451 /* report that the driver is not in use by default */
1453 return in_use;
1457 /**********************************************************************
1458 Check to see if a ogiven file is in use by *info
1459 *********************************************************************/
1461 static bool drv_file_in_use(const char *file, const struct spoolss_DriverInfo8 *info)
1463 int i = 0;
1465 if ( !info )
1466 return False;
1468 /* mz: skip files that are in the list but already deleted */
1469 if (!file || !file[0]) {
1470 return false;
1473 if (strequal(file, info->driver_path))
1474 return True;
1476 if (strequal(file, info->data_file))
1477 return True;
1479 if (strequal(file, info->config_file))
1480 return True;
1482 if (strequal(file, info->help_file))
1483 return True;
1485 /* see of there are any dependent files to examine */
1487 if (!info->dependent_files)
1488 return False;
1490 while (info->dependent_files[i] && *info->dependent_files[i]) {
1491 if (strequal(file, info->dependent_files[i]))
1492 return True;
1493 i++;
1496 return False;
1500 /**********************************************************************
1501 Utility function to remove the dependent file pointed to by the
1502 input parameter from the list
1503 *********************************************************************/
1505 static void trim_dependent_file(TALLOC_CTX *mem_ctx, const char **files, int idx)
1508 /* bump everything down a slot */
1510 while (files && files[idx+1]) {
1511 files[idx] = talloc_strdup(mem_ctx, files[idx+1]);
1512 idx++;
1515 files[idx] = NULL;
1517 return;
1520 /**********************************************************************
1521 Check if any of the files used by src are also used by drv
1522 *********************************************************************/
1524 static bool trim_overlap_drv_files(TALLOC_CTX *mem_ctx,
1525 struct spoolss_DriverInfo8 *src,
1526 const struct spoolss_DriverInfo8 *drv)
1528 bool in_use = False;
1529 int i = 0;
1531 if ( !src || !drv )
1532 return False;
1534 /* check each file. Remove it from the src structure if it overlaps */
1536 if (drv_file_in_use(src->driver_path, drv)) {
1537 in_use = True;
1538 DEBUG(10,("Removing driverfile [%s] from list\n", src->driver_path));
1539 src->driver_path = talloc_strdup(mem_ctx, "");
1540 if (!src->driver_path) { return false; }
1543 if (drv_file_in_use(src->data_file, drv)) {
1544 in_use = True;
1545 DEBUG(10,("Removing datafile [%s] from list\n", src->data_file));
1546 src->data_file = talloc_strdup(mem_ctx, "");
1547 if (!src->data_file) { return false; }
1550 if (drv_file_in_use(src->config_file, drv)) {
1551 in_use = True;
1552 DEBUG(10,("Removing configfile [%s] from list\n", src->config_file));
1553 src->config_file = talloc_strdup(mem_ctx, "");
1554 if (!src->config_file) { return false; }
1557 if (drv_file_in_use(src->help_file, drv)) {
1558 in_use = True;
1559 DEBUG(10,("Removing helpfile [%s] from list\n", src->help_file));
1560 src->help_file = talloc_strdup(mem_ctx, "");
1561 if (!src->help_file) { return false; }
1564 /* are there any dependentfiles to examine? */
1566 if (!src->dependent_files)
1567 return in_use;
1569 while (src->dependent_files[i] && *src->dependent_files[i]) {
1570 if (drv_file_in_use(src->dependent_files[i], drv)) {
1571 in_use = True;
1572 DEBUG(10,("Removing [%s] from dependent file list\n", src->dependent_files[i]));
1573 trim_dependent_file(mem_ctx, src->dependent_files, i);
1574 } else
1575 i++;
1578 return in_use;
1581 /****************************************************************************
1582 Determine whether or not a particular driver files are currently being
1583 used by any other driver.
1585 Return value is True if any files were in use by other drivers
1586 and False otherwise.
1588 Upon return, *info has been modified to only contain the driver files
1589 which are not in use
1591 Fix from mz:
1593 This needs to check all drivers to ensure that all files in use
1594 have been removed from *info, not just the ones in the first
1595 match.
1596 ****************************************************************************/
1598 bool printer_driver_files_in_use(TALLOC_CTX *mem_ctx,
1599 struct dcerpc_binding_handle *b,
1600 struct spoolss_DriverInfo8 *info)
1602 int i;
1603 uint32_t version;
1604 struct spoolss_DriverInfo8 *driver;
1605 bool in_use = false;
1606 uint32_t num_drivers;
1607 const char **drivers;
1608 WERROR result;
1610 if ( !info )
1611 return False;
1613 version = info->version;
1615 /* loop over all driver versions */
1617 DEBUG(5,("printer_driver_files_in_use: Beginning search of drivers...\n"));
1619 /* get the list of drivers */
1621 result = winreg_get_driver_list(mem_ctx, b,
1622 info->architecture, version,
1623 &num_drivers, &drivers);
1624 if (!W_ERROR_IS_OK(result)) {
1625 return true;
1628 DEBUGADD(4, ("we have:[%d] drivers in environment [%s] and version [%d]\n",
1629 num_drivers, info->architecture, version));
1631 /* check each driver for overlap in files */
1633 for (i = 0; i < num_drivers; i++) {
1634 DEBUGADD(5,("\tdriver: [%s]\n", drivers[i]));
1636 driver = NULL;
1638 result = winreg_get_driver(mem_ctx, b,
1639 info->architecture, drivers[i],
1640 version, &driver);
1641 if (!W_ERROR_IS_OK(result)) {
1642 talloc_free(drivers);
1643 return True;
1646 /* check if d2 uses any files from d1 */
1647 /* only if this is a different driver than the one being deleted */
1649 if (!strequal(info->driver_name, driver->driver_name)) {
1650 if (trim_overlap_drv_files(mem_ctx, info, driver)) {
1651 /* mz: Do not instantly return -
1652 * we need to ensure this file isn't
1653 * also in use by other drivers. */
1654 in_use = true;
1658 talloc_free(driver);
1661 talloc_free(drivers);
1663 DEBUG(5,("printer_driver_files_in_use: Completed search of drivers...\n"));
1665 return in_use;
1668 static NTSTATUS driver_unlink_internals(connection_struct *conn,
1669 const char *short_arch,
1670 int vers,
1671 const char *fname)
1673 TALLOC_CTX *tmp_ctx = talloc_new(conn);
1674 struct smb_filename *smb_fname = NULL;
1675 char *print_dlr_path;
1676 NTSTATUS status = NT_STATUS_NO_MEMORY;
1678 print_dlr_path = talloc_asprintf(tmp_ctx, "%s/%d/%s",
1679 short_arch, vers, fname);
1680 if (print_dlr_path == NULL) {
1681 goto err_out;
1684 smb_fname = synthetic_smb_fname(tmp_ctx, print_dlr_path, NULL, NULL, 0);
1685 if (smb_fname == NULL) {
1686 goto err_out;
1689 status = unlink_internals(conn, NULL, 0, smb_fname, false);
1690 err_out:
1691 talloc_free(tmp_ctx);
1692 return status;
1695 /****************************************************************************
1696 Actually delete the driver files. Make sure that
1697 printer_driver_files_in_use() return False before calling
1698 this.
1699 ****************************************************************************/
1701 bool delete_driver_files(const struct auth_session_info *session_info,
1702 const struct spoolss_DriverInfo8 *r)
1704 const char *short_arch;
1705 connection_struct *conn;
1706 NTSTATUS nt_status;
1707 char *oldcwd;
1708 char *printdollar = NULL;
1709 int printdollar_snum;
1710 bool ret = false;
1712 if (!r) {
1713 return false;
1716 DEBUG(6,("delete_driver_files: deleting driver [%s] - version [%d]\n",
1717 r->driver_name, r->version));
1719 printdollar_snum = find_service(talloc_tos(), "print$", &printdollar);
1720 if (!printdollar) {
1721 return false;
1723 if (printdollar_snum == -1) {
1724 return false;
1727 nt_status = create_conn_struct_cwd(talloc_tos(),
1728 server_event_context(),
1729 server_messaging_context(),
1730 &conn,
1731 printdollar_snum,
1732 lp_path(talloc_tos(), printdollar_snum),
1733 session_info, &oldcwd);
1734 if (!NT_STATUS_IS_OK(nt_status)) {
1735 DEBUG(0,("delete_driver_files: create_conn_struct "
1736 "returned %s\n", nt_errstr(nt_status)));
1737 return false;
1740 nt_status = set_conn_force_user_group(conn, printdollar_snum);
1741 if (!NT_STATUS_IS_OK(nt_status)) {
1742 DEBUG(0, ("failed set force user / group\n"));
1743 ret = false;
1744 goto err_free_conn;
1747 if (!become_user_by_session(conn, session_info)) {
1748 DEBUG(0, ("failed to become user\n"));
1749 ret = false;
1750 goto err_free_conn;
1753 if ( !CAN_WRITE(conn) ) {
1754 DEBUG(3,("delete_driver_files: Cannot delete print driver when [print$] is read-only\n"));
1755 ret = false;
1756 goto err_out;
1759 short_arch = get_short_archi(r->architecture);
1760 if (short_arch == NULL) {
1761 DEBUG(0, ("bad architecture %s\n", r->architecture));
1762 ret = false;
1763 goto err_out;
1766 /* now delete the files */
1768 if (r->driver_path && r->driver_path[0]) {
1769 DEBUG(10,("deleting driverfile [%s]\n", r->driver_path));
1770 driver_unlink_internals(conn, short_arch, r->version, r->driver_path);
1773 if (r->config_file && r->config_file[0]) {
1774 DEBUG(10,("deleting configfile [%s]\n", r->config_file));
1775 driver_unlink_internals(conn, short_arch, r->version, r->config_file);
1778 if (r->data_file && r->data_file[0]) {
1779 DEBUG(10,("deleting datafile [%s]\n", r->data_file));
1780 driver_unlink_internals(conn, short_arch, r->version, r->data_file);
1783 if (r->help_file && r->help_file[0]) {
1784 DEBUG(10,("deleting helpfile [%s]\n", r->help_file));
1785 driver_unlink_internals(conn, short_arch, r->version, r->help_file);
1788 if (r->dependent_files) {
1789 int i = 0;
1790 while (r->dependent_files[i] && r->dependent_files[i][0]) {
1791 DEBUG(10,("deleting dependent file [%s]\n", r->dependent_files[i]));
1792 driver_unlink_internals(conn, short_arch, r->version, r->dependent_files[i]);
1793 i++;
1797 ret = true;
1798 err_out:
1799 unbecome_user();
1800 err_free_conn:
1801 if (conn != NULL) {
1802 vfs_ChDir(conn, oldcwd);
1803 SMB_VFS_DISCONNECT(conn);
1804 conn_free(conn);
1806 return ret;
1809 /* error code:
1810 0: everything OK
1811 1: level not implemented
1812 2: file doesn't exist
1813 3: can't allocate memory
1814 4: can't free memory
1815 5: non existent struct
1819 A printer and a printer driver are 2 different things.
1820 NT manages them separatelly, Samba does the same.
1821 Why ? Simply because it's easier and it makes sense !
1823 Now explanation: You have 3 printers behind your samba server,
1824 2 of them are the same make and model (laser A and B). But laser B
1825 has an 3000 sheet feeder and laser A doesn't such an option.
1826 Your third printer is an old dot-matrix model for the accounting :-).
1828 If the /usr/local/samba/lib directory (default dir), you will have
1829 5 files to describe all of this.
1831 3 files for the printers (1 by printer):
1832 NTprinter_laser A
1833 NTprinter_laser B
1834 NTprinter_accounting
1835 2 files for the drivers (1 for the laser and 1 for the dot matrix)
1836 NTdriver_printer model X
1837 NTdriver_printer model Y
1839 jfm: I should use this comment for the text file to explain
1840 same thing for the forms BTW.
1841 Je devrais mettre mes commentaires en francais, ca serait mieux :-)
1845 /* Convert generic access rights to printer object specific access rights.
1846 It turns out that NT4 security descriptors use generic access rights and
1847 NT5 the object specific ones. */
1849 void map_printer_permissions(struct security_descriptor *sd)
1851 int i;
1853 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
1854 se_map_generic(&sd->dacl->aces[i].access_mask,
1855 &printer_generic_mapping);
1859 void map_job_permissions(struct security_descriptor *sd)
1861 int i;
1863 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
1864 se_map_generic(&sd->dacl->aces[i].access_mask,
1865 &job_generic_mapping);
1870 /****************************************************************************
1871 Check a user has permissions to perform the given operation. We use the
1872 permission constants defined in include/rpc_spoolss.h to check the various
1873 actions we perform when checking printer access.
1875 PRINTER_ACCESS_ADMINISTER:
1876 print_queue_pause, print_queue_resume, update_printer_sec,
1877 update_printer, spoolss_addprinterex_level_2,
1878 _spoolss_setprinterdata
1880 PRINTER_ACCESS_USE:
1881 print_job_start
1883 JOB_ACCESS_ADMINISTER:
1884 print_job_delete, print_job_pause, print_job_resume,
1885 print_queue_purge
1887 Try access control in the following order (for performance reasons):
1888 1) root and SE_PRINT_OPERATOR can do anything (easy check)
1889 2) check security descriptor (bit comparisons in memory)
1890 3) "printer admins" (may result in numerous calls to winbind)
1892 ****************************************************************************/
1893 WERROR print_access_check(const struct auth_session_info *session_info,
1894 struct messaging_context *msg_ctx, int snum,
1895 int access_type)
1897 struct spoolss_security_descriptor *secdesc = NULL;
1898 uint32_t access_granted;
1899 size_t sd_size;
1900 NTSTATUS status;
1901 WERROR result;
1902 const char *pname;
1903 TALLOC_CTX *mem_ctx = NULL;
1905 /* If user is NULL then use the current_user structure */
1907 /* Always allow root or SE_PRINT_OPERATROR to do anything */
1909 if ((session_info->unix_token->uid == sec_initial_uid())
1910 || security_token_has_privilege(session_info->security_token,
1911 SEC_PRIV_PRINT_OPERATOR)) {
1912 return WERR_OK;
1915 /* Get printer name */
1917 pname = lp_printername(talloc_tos(), snum);
1919 if (!pname || !*pname) {
1920 return WERR_ACCESS_DENIED;
1923 /* Get printer security descriptor */
1925 if(!(mem_ctx = talloc_init("print_access_check"))) {
1926 return WERR_NOT_ENOUGH_MEMORY;
1929 result = winreg_get_printer_secdesc_internal(mem_ctx,
1930 get_session_info_system(),
1931 msg_ctx,
1932 pname,
1933 &secdesc);
1934 if (!W_ERROR_IS_OK(result)) {
1935 talloc_destroy(mem_ctx);
1936 return WERR_NOT_ENOUGH_MEMORY;
1939 if (access_type == JOB_ACCESS_ADMINISTER) {
1940 struct spoolss_security_descriptor *parent_secdesc = secdesc;
1942 /* Create a child security descriptor to check permissions
1943 against. This is because print jobs are child objects
1944 objects of a printer. */
1945 status = se_create_child_secdesc(mem_ctx,
1946 &secdesc,
1947 &sd_size,
1948 parent_secdesc,
1949 parent_secdesc->owner_sid,
1950 parent_secdesc->group_sid,
1951 false);
1952 if (!NT_STATUS_IS_OK(status)) {
1953 talloc_destroy(mem_ctx);
1954 return ntstatus_to_werror(status);
1957 map_job_permissions(secdesc);
1958 } else {
1959 map_printer_permissions(secdesc);
1962 /* Check access */
1963 status = se_access_check(secdesc, session_info->security_token, access_type,
1964 &access_granted);
1966 DEBUG(4, ("access check was %s\n", NT_STATUS_IS_OK(status) ? "SUCCESS" : "FAILURE"));
1968 talloc_destroy(mem_ctx);
1970 return ntstatus_to_werror(status);
1973 /****************************************************************************
1974 Check the time parameters allow a print operation.
1975 *****************************************************************************/
1977 bool print_time_access_check(const struct auth_session_info *session_info,
1978 struct messaging_context *msg_ctx,
1979 const char *servicename)
1981 struct spoolss_PrinterInfo2 *pinfo2 = NULL;
1982 WERROR result;
1983 bool ok = False;
1984 time_t now = time(NULL);
1985 struct tm *t;
1986 uint32_t mins;
1988 result = winreg_get_printer_internal(NULL, session_info, msg_ctx,
1989 servicename, &pinfo2);
1990 if (!W_ERROR_IS_OK(result)) {
1991 return False;
1994 if (pinfo2->starttime == 0 && pinfo2->untiltime == 0) {
1995 ok = True;
1998 t = gmtime(&now);
1999 mins = (uint32_t)t->tm_hour*60 + (uint32_t)t->tm_min;
2001 if (mins >= pinfo2->starttime && mins <= pinfo2->untiltime) {
2002 ok = True;
2005 TALLOC_FREE(pinfo2);
2007 if (!ok) {
2008 errno = EACCES;
2011 return ok;
2014 void nt_printer_remove(TALLOC_CTX *mem_ctx,
2015 const struct auth_session_info *session_info,
2016 struct messaging_context *msg_ctx,
2017 const char *printer)
2019 WERROR result;
2021 result = winreg_delete_printer_key_internal(mem_ctx, session_info, msg_ctx,
2022 printer, "");
2023 if (!W_ERROR_IS_OK(result)) {
2024 DEBUG(0, ("nt_printer_remove: failed to remove printer %s: "
2025 "%s\n", printer, win_errstr(result)));
2029 void nt_printer_add(TALLOC_CTX *mem_ctx,
2030 const struct auth_session_info *session_info,
2031 struct messaging_context *msg_ctx,
2032 const char *printer)
2034 WERROR result;
2036 result = winreg_create_printer_internal(mem_ctx, session_info, msg_ctx,
2037 printer);
2038 if (!W_ERROR_IS_OK(result)) {
2039 DEBUG(0, ("nt_printer_add: failed to add printer %s: %s\n",
2040 printer, win_errstr(result)));