r4089: fix logic error in add_a_form() that only compared N characters instead of...
[Samba.git] / source / printing / nt_printing.c
blob3ffedcf1858cebf7f1909aac43cee9c4e0f461e2
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-2003.
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 2 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, write to the Free Software
20 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 #include "includes.h"
25 extern DOM_SID global_sid_World;
27 static TDB_CONTEXT *tdb_forms; /* used for forms files */
28 static TDB_CONTEXT *tdb_drivers; /* used for driver files */
29 static TDB_CONTEXT *tdb_printers; /* used for printers files */
31 #define FORMS_PREFIX "FORMS/"
32 #define DRIVERS_PREFIX "DRIVERS/"
33 #define DRIVER_INIT_PREFIX "DRIVER_INIT/"
34 #define PRINTERS_PREFIX "PRINTERS/"
35 #define SECDESC_PREFIX "SECDESC/"
36 #define GLOBAL_C_SETPRINTER "GLOBALS/c_setprinter"
38 #define NTDRIVERS_DATABASE_VERSION_1 1
39 #define NTDRIVERS_DATABASE_VERSION_2 2
40 #define NTDRIVERS_DATABASE_VERSION_3 3 /* little endian version of v2 */
42 #define NTDRIVERS_DATABASE_VERSION NTDRIVERS_DATABASE_VERSION_3
44 /* Map generic permissions to printer object specific permissions */
46 GENERIC_MAPPING printer_generic_mapping = {
47 PRINTER_READ,
48 PRINTER_WRITE,
49 PRINTER_EXECUTE,
50 PRINTER_ALL_ACCESS
53 STANDARD_MAPPING printer_std_mapping = {
54 PRINTER_READ,
55 PRINTER_WRITE,
56 PRINTER_EXECUTE,
57 PRINTER_ALL_ACCESS
60 /* Map generic permissions to print server object specific permissions */
62 GENERIC_MAPPING printserver_generic_mapping = {
63 SERVER_READ,
64 SERVER_WRITE,
65 SERVER_EXECUTE,
66 SERVER_ALL_ACCESS
69 STANDARD_MAPPING printserver_std_mapping = {
70 SERVER_READ,
71 SERVER_WRITE,
72 SERVER_EXECUTE,
73 SERVER_ALL_ACCESS
76 /* We need one default form to support our default printer. Msoft adds the
77 forms it wants and in the ORDER it wants them (note: DEVMODE papersize is an
78 array index). Letter is always first, so (for the current code) additions
79 always put things in the correct order. */
80 static const nt_forms_struct default_forms[] = {
81 {"Letter",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
82 {"Letter Small",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
83 {"Tabloid",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
84 {"Ledger",0x1,0x696b8,0x44368,0x0,0x0,0x696b8,0x44368},
85 {"Legal",0x1,0x34b5c,0x56d10,0x0,0x0,0x34b5c,0x56d10},
86 {"Statement",0x1,0x221b4,0x34b5c,0x0,0x0,0x221b4,0x34b5c},
87 {"Executive",0x1,0x2cf56,0x411cc,0x0,0x0,0x2cf56,0x411cc},
88 {"A3",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
89 {"A4",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
90 {"A4 Small",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
91 {"A5",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
92 {"B4 (JIS)",0x1,0x3ebe8,0x58de0,0x0,0x0,0x3ebe8,0x58de0},
93 {"B5 (JIS)",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
94 {"Folio",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
95 {"Quarto",0x1,0x347d8,0x43238,0x0,0x0,0x347d8,0x43238},
96 {"10x14",0x1,0x3e030,0x56d10,0x0,0x0,0x3e030,0x56d10},
97 {"11x17",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
98 {"Note",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
99 {"Envelope #9",0x1,0x18079,0x37091,0x0,0x0,0x18079,0x37091},
100 {"Envelope #10",0x1,0x19947,0x3ae94,0x0,0x0,0x19947,0x3ae94},
101 {"Envelope #11",0x1,0x1be7c,0x40565,0x0,0x0,0x1be7c,0x40565},
102 {"Envelope #12",0x1,0x1d74a,0x44368,0x0,0x0,0x1d74a,0x44368},
103 {"Envelope #14",0x1,0x1f018,0x47504,0x0,0x0,0x1f018,0x47504},
104 {"C size sheet",0x1,0x696b8,0x886d0,0x0,0x0,0x696b8,0x886d0},
105 {"D size sheet",0x1,0x886d0,0xd2d70,0x0,0x0,0x886d0,0xd2d70},
106 {"E size sheet",0x1,0xd2d70,0x110da0,0x0,0x0,0xd2d70,0x110da0},
107 {"Envelope DL",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
108 {"Envelope C5",0x1,0x278d0,0x37e88,0x0,0x0,0x278d0,0x37e88},
109 {"Envelope C3",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
110 {"Envelope C4",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
111 {"Envelope C6",0x1,0x1bd50,0x278d0,0x0,0x0,0x1bd50,0x278d0},
112 {"Envelope C65",0x1,0x1bd50,0x37e88,0x0,0x0,0x1bd50,0x37e88},
113 {"Envelope B4",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
114 {"Envelope B5",0x1,0x2af80,0x3d090,0x0,0x0,0x2af80,0x3d090},
115 {"Envelope B6",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
116 {"Envelope",0x1,0x1adb0,0x38270,0x0,0x0,0x1adb0,0x38270},
117 {"Envelope Monarch",0x1,0x18079,0x2e824,0x0,0x0,0x18079,0x2e824},
118 {"6 3/4 Envelope",0x1,0x167ab,0x284ec,0x0,0x0,0x167ab,0x284ec},
119 {"US Std Fanfold",0x1,0x5c3e1,0x44368,0x0,0x0,0x5c3e1,0x44368},
120 {"German Std Fanfold",0x1,0x34b5c,0x4a6a0,0x0,0x0,0x34b5c,0x4a6a0},
121 {"German Legal Fanfold",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
122 {"B4 (ISO)",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
123 {"Japanese Postcard",0x1,0x186a0,0x24220,0x0,0x0,0x186a0,0x24220},
124 {"9x11",0x1,0x37cf8,0x44368,0x0,0x0,0x37cf8,0x44368},
125 {"10x11",0x1,0x3e030,0x44368,0x0,0x0,0x3e030,0x44368},
126 {"15x11",0x1,0x5d048,0x44368,0x0,0x0,0x5d048,0x44368},
127 {"Envelope Invite",0x1,0x35b60,0x35b60,0x0,0x0,0x35b60,0x35b60},
128 {"Reserved48",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
129 {"Reserved49",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
130 {"Letter Extra",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
131 {"Legal Extra",0x1,0x3ae94,0x5d048,0x0,0x0,0x3ae94,0x5d048},
132 {"Tabloid Extra",0x1,0x4a6a0,0x6f9f0,0x0,0x0,0x4a6a0,0x6f9f0},
133 {"A4 Extra",0x1,0x397c2,0x4eb16,0x0,0x0,0x397c2,0x4eb16},
134 {"Letter Transverse",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
135 {"A4 Transverse",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
136 {"Letter Extra Transverse",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
137 {"Super A",0x1,0x376b8,0x56ea0,0x0,0x0,0x376b8,0x56ea0},
138 {"Super B",0x1,0x4a768,0x76e58,0x0,0x0,0x4a768,0x76e58},
139 {"Letter Plus",0x1,0x34b5c,0x4eb16,0x0,0x0,0x34b5c,0x4eb16},
140 {"A4 Plus",0x1,0x33450,0x50910,0x0,0x0,0x33450,0x50910},
141 {"A5 Transverse",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
142 {"B5 (JIS) Transverse",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
143 {"A3 Extra",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
144 {"A5 Extra",0x1,0x2a7b0,0x395f8,0x0,0x0,0x2a7b0,0x395f8},
145 {"B5 (ISO) Extra",0x1,0x31128,0x43620,0x0,0x0,0x31128,0x43620},
146 {"A2",0x1,0x668a0,0x91050,0x0,0x0,0x668a0,0x91050},
147 {"A3 Transverse",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
148 {"A3 Extra Transverse",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
149 {"Japanese Double Postcard",0x1,0x30d40,0x24220,0x0,0x0,0x30d40,0x24220},
150 {"A6",0x1,0x19a28,0x24220,0x0,0x0,0x19a28,0x24220},
151 {"Japanese Envelope Kaku #2",0x1,0x3a980,0x510e0,0x0,0x0,0x3a980,0x510e0},
152 {"Japanese Envelope Kaku #3",0x1,0x34bc0,0x43a08,0x0,0x0,0x34bc0,0x43a08},
153 {"Japanese Envelope Chou #3",0x1,0x1d4c0,0x395f8,0x0,0x0,0x1d4c0,0x395f8},
154 {"Japanese Envelope Chou #4",0x1,0x15f90,0x320c8,0x0,0x0,0x15f90,0x320c8},
155 {"Letter Rotated",0x1,0x44368,0x34b5c,0x0,0x0,0x44368,0x34b5c},
156 {"A3 Rotated",0x1,0x668a0,0x48828,0x0,0x0,0x668a0,0x48828},
157 {"A4 Rotated",0x1,0x48828,0x33450,0x0,0x0,0x48828,0x33450},
158 {"A5 Rotated",0x1,0x33450,0x24220,0x0,0x0,0x33450,0x24220},
159 {"B4 (JIS) Rotated",0x1,0x58de0,0x3ebe8,0x0,0x0,0x58de0,0x3ebe8},
160 {"B5 (JIS) Rotated",0x1,0x3ebe8,0x2c6f0,0x0,0x0,0x3ebe8,0x2c6f0},
161 {"Japanese Postcard Rotated",0x1,0x24220,0x186a0,0x0,0x0,0x24220,0x186a0},
162 {"Double Japan Postcard Rotated",0x1,0x24220,0x30d40,0x0,0x0,0x24220,0x30d40},
163 {"A6 Rotated",0x1,0x24220,0x19a28,0x0,0x0,0x24220,0x19a28},
164 {"Japan Envelope Kaku #2 Rotated",0x1,0x510e0,0x3a980,0x0,0x0,0x510e0,0x3a980},
165 {"Japan Envelope Kaku #3 Rotated",0x1,0x43a08,0x34bc0,0x0,0x0,0x43a08, 0x34bc0},
166 {"Japan Envelope Chou #3 Rotated",0x1,0x395f8,0x1d4c0,0x0,0x0,0x395f8,0x1d4c0},
167 {"Japan Envelope Chou #4 Rotated",0x1,0x320c8,0x15f90,0x0,0x0,0x320c8,0x15f90},
168 {"B6 (JIS)",0x1,0x1f400,0x2c6f0,0x0,0x0,0x1f400,0x2c6f0},
169 {"B6 (JIS) Rotated",0x1,0x2c6f0,0x1f400,0x0,0x0,0x2c6f0,0x1f400},
170 {"12x11",0x1,0x4a724,0x443e1,0x0,0x0,0x4a724,0x443e1},
171 {"Japan Envelope You #4",0x1,0x19a28,0x395f8,0x0,0x0,0x19a28,0x395f8},
172 {"Japan Envelope You #4 Rotated",0x1,0x395f8,0x19a28,0x0,0x0,0x395f8,0x19a28},
173 {"PRC 16K",0x1,0x2de60,0x3f7a0,0x0,0x0,0x2de60,0x3f7a0},
174 {"PRC 32K",0x1,0x1fbd0,0x2cec0,0x0,0x0,0x1fbd0,0x2cec0},
175 {"PRC 32K(Big)",0x1,0x222e0,0x318f8,0x0,0x0,0x222e0,0x318f8},
176 {"PRC Envelope #1",0x1,0x18e70,0x28488,0x0,0x0,0x18e70,0x28488},
177 {"PRC Envelope #2",0x1,0x18e70,0x2af80,0x0,0x0,0x18e70,0x2af80},
178 {"PRC Envelope #3",0x1,0x1e848,0x2af80,0x0,0x0,0x1e848,0x2af80},
179 {"PRC Envelope #4",0x1,0x1adb0,0x32c80,0x0,0x0,0x1adb0,0x32c80},
180 {"PRC Envelope #5",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
181 {"PRC Envelope #6",0x1,0x1d4c0,0x38270,0x0,0x0,0x1d4c0,0x38270},
182 {"PRC Envelope #7",0x1,0x27100,0x38270,0x0,0x0,0x27100,0x38270},
183 {"PRC Envelope #8",0x1,0x1d4c0,0x4b708,0x0,0x0,0x1d4c0,0x4b708},
184 {"PRC Envelope #9",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
185 {"PRC Envelope #10",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
186 {"PRC 16K Rotated",0x1,0x3f7a0,0x2de60,0x0,0x0,0x3f7a0,0x2de60},
187 {"PRC 32K Rotated",0x1,0x2cec0,0x1fbd0,0x0,0x0,0x2cec0,0x1fbd0},
188 {"PRC 32K(Big) Rotated",0x1,0x318f8,0x222e0,0x0,0x0,0x318f8,0x222e0},
189 {"PRC Envelope #1 Rotated",0x1,0x28488,0x18e70,0x0,0x0,0x28488,0x18e70},
190 {"PRC Envelope #2 Rotated",0x1,0x2af80,0x18e70,0x0,0x0,0x2af80,0x18e70},
191 {"PRC Envelope #3 Rotated",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
192 {"PRC Envelope #4 Rotated",0x1,0x32c80,0x1adb0,0x0,0x0,0x32c80,0x1adb0},
193 {"PRC Envelope #5 Rotated",0x1,0x35b60,0x1adb0,0x0,0x0,0x35b60,0x1adb0},
194 {"PRC Envelope #6 Rotated",0x1,0x38270,0x1d4c0,0x0,0x0,0x38270,0x1d4c0},
195 {"PRC Envelope #7 Rotated",0x1,0x38270,0x27100,0x0,0x0,0x38270,0x27100},
196 {"PRC Envelope #8 Rotated",0x1,0x4b708,0x1d4c0,0x0,0x0,0x4b708,0x1d4c0},
197 {"PRC Envelope #9 Rotated",0x1,0x4f1a0,0x37e88,0x0,0x0,0x4f1a0,0x37e88},
198 {"PRC Envelope #10 Rotated",0x1,0x6fd10,0x4f1a0,0x0,0x0,0x6fd10,0x4f1a0}
201 struct table_node {
202 const char *long_archi;
203 const char *short_archi;
204 int version;
207 #define SPL_ARCH_WIN40 "WIN40"
208 #define SPL_ARCH_W32X86 "W32X86"
209 #define SPL_ARCH_W32MIPS "W32MIPS"
210 #define SPL_ARCH_W32ALPHA "W32ALPHA"
211 #define SPL_ARCH_W32PPC "W32PPC"
212 #define SPL_ARCH_IA64 "IA64"
213 #define SPL_ARCH_X64 "x64"
215 static const struct table_node archi_table[]= {
217 {"Windows 4.0", SPL_ARCH_WIN40, 0 },
218 {"Windows NT x86", SPL_ARCH_W32X86, 2 },
219 {"Windows NT R4000", SPL_ARCH_W32MIPS, 2 },
220 {"Windows NT Alpha_AXP", SPL_ARCH_W32ALPHA, 2 },
221 {"Windows NT PowerPC", SPL_ARCH_W32PPC, 2 },
222 {"Windows IA64", SPL_ARCH_IA64, 3 },
223 {"Windows x64", SPL_ARCH_X64, 3 },
224 {NULL, "", -1 }
227 static BOOL upgrade_to_version_3(void)
229 TDB_DATA kbuf, newkey, dbuf;
231 DEBUG(0,("upgrade_to_version_3: upgrading print tdb's to version 3\n"));
233 for (kbuf = tdb_firstkey(tdb_drivers); kbuf.dptr;
234 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
236 dbuf = tdb_fetch(tdb_drivers, kbuf);
238 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) == 0) {
239 DEBUG(0,("upgrade_to_version_3:moving form\n"));
240 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) {
241 SAFE_FREE(dbuf.dptr);
242 DEBUG(0,("upgrade_to_version_3: failed to move form. Error (%s).\n", tdb_errorstr(tdb_forms)));
243 return False;
245 if (tdb_delete(tdb_drivers, kbuf) != 0) {
246 SAFE_FREE(dbuf.dptr);
247 DEBUG(0,("upgrade_to_version_3: failed to delete form. Error (%s)\n", tdb_errorstr(tdb_drivers)));
248 return False;
252 if (strncmp(kbuf.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX)) == 0) {
253 DEBUG(0,("upgrade_to_version_3:moving printer\n"));
254 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
255 SAFE_FREE(dbuf.dptr);
256 DEBUG(0,("upgrade_to_version_3: failed to move printer. Error (%s)\n", tdb_errorstr(tdb_printers)));
257 return False;
259 if (tdb_delete(tdb_drivers, kbuf) != 0) {
260 SAFE_FREE(dbuf.dptr);
261 DEBUG(0,("upgrade_to_version_3: failed to delete printer. Error (%s)\n", tdb_errorstr(tdb_drivers)));
262 return False;
266 if (strncmp(kbuf.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX)) == 0) {
267 DEBUG(0,("upgrade_to_version_3:moving secdesc\n"));
268 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
269 SAFE_FREE(dbuf.dptr);
270 DEBUG(0,("upgrade_to_version_3: failed to move secdesc. Error (%s)\n", tdb_errorstr(tdb_printers)));
271 return False;
273 if (tdb_delete(tdb_drivers, kbuf) != 0) {
274 SAFE_FREE(dbuf.dptr);
275 DEBUG(0,("upgrade_to_version_3: failed to delete secdesc. Error (%s)\n", tdb_errorstr(tdb_drivers)));
276 return False;
280 SAFE_FREE(dbuf.dptr);
283 return True;
286 /****************************************************************************
287 Open the NT printing tdbs. Done once before fork().
288 ****************************************************************************/
290 BOOL nt_printing_init(void)
292 static pid_t local_pid;
293 const char *vstring = "INFO/version";
294 WERROR win_rc;
296 if (tdb_drivers && tdb_printers && tdb_forms && local_pid == sys_getpid())
297 return True;
299 if (tdb_drivers)
300 tdb_close(tdb_drivers);
301 tdb_drivers = tdb_open_log(lock_path("ntdrivers.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
302 if (!tdb_drivers) {
303 DEBUG(0,("nt_printing_init: Failed to open nt drivers database %s (%s)\n",
304 lock_path("ntdrivers.tdb"), strerror(errno) ));
305 return False;
308 if (tdb_printers)
309 tdb_close(tdb_printers);
310 tdb_printers = tdb_open_log(lock_path("ntprinters.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
311 if (!tdb_printers) {
312 DEBUG(0,("nt_printing_init: Failed to open nt printers database %s (%s)\n",
313 lock_path("ntprinters.tdb"), strerror(errno) ));
314 return False;
317 if (tdb_forms)
318 tdb_close(tdb_forms);
319 tdb_forms = tdb_open_log(lock_path("ntforms.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
320 if (!tdb_forms) {
321 DEBUG(0,("nt_printing_init: Failed to open nt forms database %s (%s)\n",
322 lock_path("ntforms.tdb"), strerror(errno) ));
323 return False;
326 local_pid = sys_getpid();
328 /* handle a Samba upgrade */
329 tdb_lock_bystring(tdb_drivers, vstring, 0);
331 int32 vers_id;
333 /* Cope with byte-reversed older versions of the db. */
334 vers_id = tdb_fetch_int32(tdb_drivers, vstring);
335 if ((vers_id == NTDRIVERS_DATABASE_VERSION_2) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_2)) {
336 /* Written on a bigendian machine with old fetch_int code. Save as le. */
337 /* The only upgrade between V2 and V3 is to save the version in little-endian. */
338 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION);
339 vers_id = NTDRIVERS_DATABASE_VERSION;
342 if (vers_id != NTDRIVERS_DATABASE_VERSION) {
344 if ((vers_id == NTDRIVERS_DATABASE_VERSION_1) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_1)) {
345 if (!upgrade_to_version_3())
346 return False;
347 } else
348 tdb_traverse(tdb_drivers, tdb_traverse_delete_fn, NULL);
350 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION);
353 tdb_unlock_bystring(tdb_drivers, vstring);
355 update_c_setprinter(True);
358 * register callback to handle updating printers as new
359 * drivers are installed
362 message_register( MSG_PRINTER_DRVUPGRADE, do_drv_upgrade_printer );
365 * register callback to handle updating printer data
366 * when a driver is initialized
369 message_register( MSG_PRINTERDATA_INIT_RESET, reset_all_printerdata );
372 * register callback to handle invalidating the printer cache
373 * between smbd processes.
376 message_register( MSG_PRINTER_MOD, receive_printer_mod_msg);
378 /* of course, none of the message callbacks matter if you don't
379 tell messages.c that you interested in receiving PRINT_GENERAL
380 msgs. This is done in claim_connection() */
383 if ( lp_security() == SEC_ADS ) {
384 win_rc = check_published_printers();
385 if (!W_ERROR_IS_OK(win_rc))
386 DEBUG(0, ("nt_printing_init: error checking published printers: %s\n", dos_errstr(win_rc)));
389 return True;
392 /*******************************************************************
393 Function to allow filename parsing "the old way".
394 ********************************************************************/
396 static BOOL driver_unix_convert(char *name,connection_struct *conn,
397 char *saved_last_component, BOOL *bad_path, SMB_STRUCT_STAT *pst)
399 unix_format(name);
400 unix_clean_name(name);
401 trim_string(name,"/","/");
402 return unix_convert(name, conn, saved_last_component, bad_path, pst);
405 /*******************************************************************
406 tdb traversal function for counting printers.
407 ********************************************************************/
409 static int traverse_counting_printers(TDB_CONTEXT *t, TDB_DATA key,
410 TDB_DATA data, void *context)
412 int *printer_count = (int*)context;
414 if (memcmp(PRINTERS_PREFIX, key.dptr, sizeof(PRINTERS_PREFIX)-1) == 0) {
415 (*printer_count)++;
416 DEBUG(10,("traverse_counting_printers: printer = [%s] printer_count = %d\n", key.dptr, *printer_count));
419 return 0;
422 /*******************************************************************
423 Update the spooler global c_setprinter. This variable is initialized
424 when the parent smbd starts with the number of existing printers. It
425 is monotonically increased by the current number of printers *after*
426 each add or delete printer RPC. Only Microsoft knows why... JRR020119
427 ********************************************************************/
429 uint32 update_c_setprinter(BOOL initialize)
431 int32 c_setprinter;
432 int32 printer_count = 0;
434 tdb_lock_bystring(tdb_printers, GLOBAL_C_SETPRINTER, 0);
436 /* Traverse the tdb, counting the printers */
437 tdb_traverse(tdb_printers, traverse_counting_printers, (void *)&printer_count);
439 /* If initializing, set c_setprinter to current printers count
440 * otherwise, bump it by the current printer count
442 if (!initialize)
443 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER) + printer_count;
444 else
445 c_setprinter = printer_count;
447 DEBUG(10,("update_c_setprinter: c_setprinter = %u\n", (unsigned int)c_setprinter));
448 tdb_store_int32(tdb_printers, GLOBAL_C_SETPRINTER, c_setprinter);
450 tdb_unlock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
452 return (uint32)c_setprinter;
455 /*******************************************************************
456 Get the spooler global c_setprinter, accounting for initialization.
457 ********************************************************************/
459 uint32 get_c_setprinter(void)
461 int32 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER);
463 if (c_setprinter == (int32)-1)
464 c_setprinter = update_c_setprinter(True);
466 DEBUG(10,("get_c_setprinter: c_setprinter = %d\n", c_setprinter));
468 return (uint32)c_setprinter;
471 /****************************************************************************
472 Get builtin form struct list.
473 ****************************************************************************/
475 int get_builtin_ntforms(nt_forms_struct **list)
477 *list = (nt_forms_struct *)memdup(&default_forms[0], sizeof(default_forms));
478 return sizeof(default_forms) / sizeof(default_forms[0]);
481 /****************************************************************************
482 get a builtin form struct
483 ****************************************************************************/
485 BOOL get_a_builtin_ntform(UNISTR2 *uni_formname,nt_forms_struct *form)
487 int i,count;
488 fstring form_name;
489 unistr2_to_ascii(form_name, uni_formname, sizeof(form_name)-1);
490 DEBUGADD(6,("Looking for builtin form %s \n", form_name));
491 count = sizeof(default_forms) / sizeof(default_forms[0]);
492 for (i=0;i<count;i++) {
493 if (strequal(form_name,default_forms[i].name)) {
494 DEBUGADD(6,("Found builtin form %s \n", form_name));
495 memcpy(form,&default_forms[i],sizeof(*form));
496 break;
500 return (i !=count);
503 /****************************************************************************
504 get a form struct list
505 ****************************************************************************/
506 int get_ntforms(nt_forms_struct **list)
508 TDB_DATA kbuf, newkey, dbuf;
509 nt_forms_struct *tl;
510 nt_forms_struct form;
511 int ret;
512 int i;
513 int n = 0;
515 for (kbuf = tdb_firstkey(tdb_forms);
516 kbuf.dptr;
517 newkey = tdb_nextkey(tdb_forms, kbuf), safe_free(kbuf.dptr), kbuf=newkey)
519 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) != 0)
520 continue;
522 dbuf = tdb_fetch(tdb_forms, kbuf);
523 if (!dbuf.dptr)
524 continue;
526 fstrcpy(form.name, kbuf.dptr+strlen(FORMS_PREFIX));
527 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddddddd",
528 &i, &form.flag, &form.width, &form.length, &form.left,
529 &form.top, &form.right, &form.bottom);
530 SAFE_FREE(dbuf.dptr);
531 if (ret != dbuf.dsize)
532 continue;
534 tl = SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1);
535 if (!tl) {
536 DEBUG(0,("get_ntforms: Realloc fail.\n"));
537 return 0;
539 *list = tl;
540 (*list)[n] = form;
541 n++;
545 return n;
548 /****************************************************************************
549 write a form struct list
550 ****************************************************************************/
551 int write_ntforms(nt_forms_struct **list, int number)
553 pstring buf, key;
554 int len;
555 TDB_DATA kbuf,dbuf;
556 int i;
558 for (i=0;i<number;i++) {
559 /* save index, so list is rebuilt in correct order */
560 len = tdb_pack(buf, sizeof(buf), "dddddddd",
561 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
562 (*list)[i].left, (*list)[i].top, (*list)[i].right,
563 (*list)[i].bottom);
564 if (len > sizeof(buf)) break;
565 slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[i].name);
566 kbuf.dsize = strlen(key)+1;
567 kbuf.dptr = key;
568 dbuf.dsize = len;
569 dbuf.dptr = buf;
570 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) break;
573 return i;
576 /****************************************************************************
577 add a form struct at the end of the list
578 ****************************************************************************/
579 BOOL add_a_form(nt_forms_struct **list, const FORM *form, int *count)
581 int n=0;
582 BOOL update;
583 fstring form_name;
584 nt_forms_struct *tl;
587 * NT tries to add forms even when
588 * they are already in the base
589 * only update the values if already present
592 update=False;
594 unistr2_to_ascii(form_name, &form->name, sizeof(form_name)-1);
595 for (n=0; n<*count; n++) {
596 if ( strequal((*list)[n].name, form_name) ) {
597 update=True;
598 break;
602 if (update==False) {
603 if((tl=SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1)) == NULL) {
604 DEBUG(0,("add_a_form: failed to enlarge forms list!\n"));
605 return False;
607 *list = tl;
608 unistr2_to_ascii((*list)[n].name, &form->name, sizeof((*list)[n].name)-1);
609 (*count)++;
612 (*list)[n].flag=form->flags;
613 (*list)[n].width=form->size_x;
614 (*list)[n].length=form->size_y;
615 (*list)[n].left=form->left;
616 (*list)[n].top=form->top;
617 (*list)[n].right=form->right;
618 (*list)[n].bottom=form->bottom;
620 DEBUG(6,("add_a_form: Successfully %s form [%s]\n",
621 update ? "updated" : "added", form_name));
623 return True;
626 /****************************************************************************
627 Delete a named form struct.
628 ****************************************************************************/
630 BOOL delete_a_form(nt_forms_struct **list, UNISTR2 *del_name, int *count, WERROR *ret)
632 pstring key;
633 TDB_DATA kbuf;
634 int n=0;
635 fstring form_name;
637 *ret = WERR_OK;
639 unistr2_to_ascii(form_name, del_name, sizeof(form_name)-1);
641 for (n=0; n<*count; n++) {
642 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
643 DEBUG(103, ("delete_a_form, [%s] in list\n", form_name));
644 break;
648 if (n == *count) {
649 DEBUG(10,("delete_a_form, [%s] not found\n", form_name));
650 *ret = WERR_INVALID_PARAM;
651 return False;
654 slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[n].name);
655 kbuf.dsize = strlen(key)+1;
656 kbuf.dptr = key;
657 if (tdb_delete(tdb_forms, kbuf) != 0) {
658 *ret = WERR_NOMEM;
659 return False;
662 return True;
665 /****************************************************************************
666 Update a form struct.
667 ****************************************************************************/
669 void update_a_form(nt_forms_struct **list, const FORM *form, int count)
671 int n=0;
672 fstring form_name;
673 unistr2_to_ascii(form_name, &(form->name), sizeof(form_name)-1);
675 DEBUG(106, ("[%s]\n", form_name));
676 for (n=0; n<count; n++) {
677 DEBUGADD(106, ("n [%d]:[%s]\n", n, (*list)[n].name));
678 if (!strncmp((*list)[n].name, form_name, strlen(form_name)))
679 break;
682 if (n==count) return;
684 (*list)[n].flag=form->flags;
685 (*list)[n].width=form->size_x;
686 (*list)[n].length=form->size_y;
687 (*list)[n].left=form->left;
688 (*list)[n].top=form->top;
689 (*list)[n].right=form->right;
690 (*list)[n].bottom=form->bottom;
693 /****************************************************************************
694 Get the nt drivers list.
695 Traverse the database and look-up the matching names.
696 ****************************************************************************/
697 int get_ntdrivers(fstring **list, const char *architecture, uint32 version)
699 int total=0;
700 const char *short_archi;
701 fstring *fl;
702 pstring key;
703 TDB_DATA kbuf, newkey;
705 short_archi = get_short_archi(architecture);
706 slprintf(key, sizeof(key)-1, "%s%s/%d/", DRIVERS_PREFIX, short_archi, version);
708 for (kbuf = tdb_firstkey(tdb_drivers);
709 kbuf.dptr;
710 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
712 if (strncmp(kbuf.dptr, key, strlen(key)) != 0)
713 continue;
715 if((fl = SMB_REALLOC_ARRAY(*list, fstring, total+1)) == NULL) {
716 DEBUG(0,("get_ntdrivers: failed to enlarge list!\n"));
717 return -1;
719 else *list = fl;
721 fstrcpy((*list)[total], kbuf.dptr+strlen(key));
722 total++;
725 return(total);
728 /****************************************************************************
729 function to do the mapping between the long architecture name and
730 the short one.
731 ****************************************************************************/
732 const char *get_short_archi(const char *long_archi)
734 int i=-1;
736 DEBUG(107,("Getting architecture dependant directory\n"));
737 do {
738 i++;
739 } while ( (archi_table[i].long_archi!=NULL ) &&
740 StrCaseCmp(long_archi, archi_table[i].long_archi) );
742 if (archi_table[i].long_archi==NULL) {
743 DEBUGADD(10,("Unknown architecture [%s] !\n", long_archi));
744 return NULL;
747 /* this might be client code - but shouldn't this be an fstrcpy etc? */
750 DEBUGADD(108,("index: [%d]\n", i));
751 DEBUGADD(108,("long architecture: [%s]\n", archi_table[i].long_archi));
752 DEBUGADD(108,("short architecture: [%s]\n", archi_table[i].short_archi));
754 return archi_table[i].short_archi;
757 /****************************************************************************
758 Version information in Microsoft files is held in a VS_VERSION_INFO structure.
759 There are two case to be covered here: PE (Portable Executable) and NE (New
760 Executable) files. Both files support the same INFO structure, but PE files
761 store the signature in unicode, and NE files store it as !unicode.
762 returns -1 on error, 1 on version info found, and 0 on no version info found.
763 ****************************************************************************/
765 static int get_file_version(files_struct *fsp, char *fname,uint32 *major, uint32 *minor)
767 int i;
768 char *buf = NULL;
769 ssize_t byte_count;
771 if ((buf=SMB_MALLOC(PE_HEADER_SIZE)) == NULL) {
772 DEBUG(0,("get_file_version: PE file [%s] PE Header malloc failed bytes = %d\n",
773 fname, PE_HEADER_SIZE));
774 goto error_exit;
777 /* Note: DOS_HEADER_SIZE < malloc'ed PE_HEADER_SIZE */
778 if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
779 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %lu\n",
780 fname, (unsigned long)byte_count));
781 goto no_version_info;
784 /* Is this really a DOS header? */
785 if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
786 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
787 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
788 goto no_version_info;
791 /* Skip OEM header (if any) and the DOS stub to start of Windows header */
792 if (SMB_VFS_LSEEK(fsp, fsp->fd, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (SMB_OFF_T)-1) {
793 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
794 fname, errno));
795 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
796 goto no_version_info;
799 if ((byte_count = vfs_read_data(fsp, buf, PE_HEADER_SIZE)) < PE_HEADER_SIZE) {
800 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %lu\n",
801 fname, (unsigned long)byte_count));
802 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
803 goto no_version_info;
806 /* The header may be a PE (Portable Executable) or an NE (New Executable) */
807 if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
808 unsigned int num_sections;
809 unsigned int section_table_bytes;
811 if (SVAL(buf,PE_HEADER_MACHINE_OFFSET) != PE_HEADER_MACHINE_I386) {
812 DEBUG(3,("get_file_version: PE file [%s] wrong machine = 0x%x\n",
813 fname, SVAL(buf,PE_HEADER_MACHINE_OFFSET)));
814 /* At this point, we assume the file is in error. It still could be somthing
815 * else besides a PE file, but it unlikely at this point.
817 goto error_exit;
820 /* get the section table */
821 num_sections = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
822 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
823 if (section_table_bytes == 0)
824 goto error_exit;
826 SAFE_FREE(buf);
827 if ((buf=SMB_MALLOC(section_table_bytes)) == NULL) {
828 DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
829 fname, section_table_bytes));
830 goto error_exit;
833 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
834 DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %lu\n",
835 fname, (unsigned long)byte_count));
836 goto error_exit;
839 /* Iterate the section table looking for the resource section ".rsrc" */
840 for (i = 0; i < num_sections; i++) {
841 int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
843 if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
844 unsigned int section_pos = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
845 unsigned int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
847 if (section_bytes == 0)
848 goto error_exit;
850 SAFE_FREE(buf);
851 if ((buf=SMB_MALLOC(section_bytes)) == NULL) {
852 DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
853 fname, section_bytes));
854 goto error_exit;
857 /* Seek to the start of the .rsrc section info */
858 if (SMB_VFS_LSEEK(fsp, fsp->fd, section_pos, SEEK_SET) == (SMB_OFF_T)-1) {
859 DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
860 fname, errno));
861 goto error_exit;
864 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
865 DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %lu\n",
866 fname, (unsigned long)byte_count));
867 goto error_exit;
870 if (section_bytes < VS_VERSION_INFO_UNICODE_SIZE)
871 goto error_exit;
873 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
874 /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
875 if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
876 /* Align to next long address */
877 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
879 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
880 *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
881 *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
883 DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
884 fname, *major, *minor,
885 (*major>>16)&0xffff, *major&0xffff,
886 (*minor>>16)&0xffff, *minor&0xffff));
887 SAFE_FREE(buf);
888 return 1;
895 /* Version info not found, fall back to origin date/time */
896 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
897 SAFE_FREE(buf);
898 return 0;
900 } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
901 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
902 DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
903 fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
904 /* At this point, we assume the file is in error. It still could be somthing
905 * else besides a NE file, but it unlikely at this point. */
906 goto error_exit;
909 /* Allocate a bit more space to speed up things */
910 SAFE_FREE(buf);
911 if ((buf=SMB_MALLOC(VS_NE_BUF_SIZE)) == NULL) {
912 DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes = %d\n",
913 fname, PE_HEADER_SIZE));
914 goto error_exit;
917 /* This is a HACK! I got tired of trying to sort through the messy
918 * 'NE' file format. If anyone wants to clean this up please have at
919 * it, but this works. 'NE' files will eventually fade away. JRR */
920 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
921 /* Cover case that should not occur in a well formed 'NE' .dll file */
922 if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
924 for(i=0; i<byte_count; i++) {
925 /* Fast skip past data that can't possibly match */
926 if (buf[i] != 'V') continue;
928 /* Potential match data crosses buf boundry, move it to beginning
929 * of buf, and fill the buf with as much as it will hold. */
930 if (i>byte_count-VS_VERSION_INFO_SIZE) {
931 int bc;
933 memcpy(buf, &buf[i], byte_count-i);
934 if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
935 (byte_count-i))) < 0) {
937 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
938 fname, errno));
939 goto error_exit;
942 byte_count = bc + (byte_count - i);
943 if (byte_count<VS_VERSION_INFO_SIZE) break;
945 i = 0;
948 /* Check that the full signature string and the magic number that
949 * follows exist (not a perfect solution, but the chances that this
950 * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
951 * twice, as it is simpler to read the code. */
952 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
953 /* Compute skip alignment to next long address */
954 int skip = -(SMB_VFS_LSEEK(fsp, fsp->fd, 0, SEEK_CUR) - (byte_count - i) +
955 sizeof(VS_SIGNATURE)) & 3;
956 if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
958 *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
959 *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
960 DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
961 fname, *major, *minor,
962 (*major>>16)&0xffff, *major&0xffff,
963 (*minor>>16)&0xffff, *minor&0xffff));
964 SAFE_FREE(buf);
965 return 1;
970 /* Version info not found, fall back to origin date/time */
971 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
972 SAFE_FREE(buf);
973 return 0;
975 } else
976 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
977 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
978 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
980 no_version_info:
981 SAFE_FREE(buf);
982 return 0;
984 error_exit:
985 SAFE_FREE(buf);
986 return -1;
989 /****************************************************************************
990 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
991 share one or more files. During the MS installation process files are checked
992 to insure that only a newer version of a shared file is installed over an
993 older version. There are several possibilities for this comparison. If there
994 is no previous version, the new one is newer (obviously). If either file is
995 missing the version info structure, compare the creation date (on Unix use
996 the modification date). Otherwise chose the numerically larger version number.
997 ****************************************************************************/
999 static int file_version_is_newer(connection_struct *conn, fstring new_file, fstring old_file)
1001 BOOL use_version = True;
1002 pstring filepath;
1004 uint32 new_major;
1005 uint32 new_minor;
1006 time_t new_create_time;
1008 uint32 old_major;
1009 uint32 old_minor;
1010 time_t old_create_time;
1012 int access_mode;
1013 int action;
1014 files_struct *fsp = NULL;
1015 SMB_STRUCT_STAT st;
1016 SMB_STRUCT_STAT stat_buf;
1017 BOOL bad_path;
1019 ZERO_STRUCT(st);
1020 ZERO_STRUCT(stat_buf);
1021 new_create_time = (time_t)0;
1022 old_create_time = (time_t)0;
1024 /* Get file version info (if available) for previous file (if it exists) */
1025 pstrcpy(filepath, old_file);
1027 driver_unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
1029 fsp = open_file_shared(conn, filepath, &stat_buf,
1030 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1031 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1032 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1033 if (!fsp) {
1034 /* Old file not found, so by definition new file is in fact newer */
1035 DEBUG(10,("file_version_is_newer: Can't open old file [%s], errno = %d\n",
1036 filepath, errno));
1037 return True;
1039 } else {
1040 int ret = get_file_version(fsp, old_file, &old_major, &old_minor);
1041 if (ret == -1) goto error_exit;
1043 if (!ret) {
1044 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1045 old_file));
1046 use_version = False;
1047 if (SMB_VFS_FSTAT(fsp, fsp->fd, &st) == -1) goto error_exit;
1048 old_create_time = st.st_mtime;
1049 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", old_create_time));
1052 close_file(fsp, True);
1054 /* Get file version info (if available) for new file */
1055 pstrcpy(filepath, new_file);
1056 driver_unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
1058 fsp = open_file_shared(conn, filepath, &stat_buf,
1059 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1060 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1061 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1062 if (!fsp) {
1063 /* New file not found, this shouldn't occur if the caller did its job */
1064 DEBUG(3,("file_version_is_newer: Can't open new file [%s], errno = %d\n",
1065 filepath, errno));
1066 goto error_exit;
1068 } else {
1069 int ret = get_file_version(fsp, new_file, &new_major, &new_minor);
1070 if (ret == -1) goto error_exit;
1072 if (!ret) {
1073 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1074 new_file));
1075 use_version = False;
1076 if (SMB_VFS_FSTAT(fsp, fsp->fd, &st) == -1) goto error_exit;
1077 new_create_time = st.st_mtime;
1078 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", new_create_time));
1081 close_file(fsp, True);
1083 if (use_version && (new_major != old_major || new_minor != old_minor)) {
1084 /* Compare versions and choose the larger version number */
1085 if (new_major > old_major ||
1086 (new_major == old_major && new_minor > old_minor)) {
1088 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1089 return True;
1091 else {
1092 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1093 return False;
1096 } else {
1097 /* Compare modification time/dates and choose the newest time/date */
1098 if (new_create_time > old_create_time) {
1099 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1100 return True;
1102 else {
1103 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1104 return False;
1108 error_exit:
1109 if(fsp)
1110 close_file(fsp, True);
1111 return -1;
1114 /****************************************************************************
1115 Determine the correct cVersion associated with an architecture and driver
1116 ****************************************************************************/
1117 static uint32 get_correct_cversion(const char *architecture, fstring driverpath_in,
1118 struct current_user *user, WERROR *perr)
1120 int cversion;
1121 int access_mode;
1122 int action;
1123 NTSTATUS nt_status;
1124 pstring driverpath;
1125 DATA_BLOB null_pw;
1126 fstring res_type;
1127 files_struct *fsp = NULL;
1128 BOOL bad_path;
1129 SMB_STRUCT_STAT st;
1130 connection_struct *conn;
1132 ZERO_STRUCT(st);
1134 *perr = WERR_INVALID_PARAM;
1136 /* If architecture is Windows 95/98/ME, the version is always 0. */
1137 if (strcmp(architecture, "WIN40") == 0) {
1138 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
1139 *perr = WERR_OK;
1140 return 0;
1144 * Connect to the print$ share under the same account as the user connected
1145 * to the rpc pipe. Note we must still be root to do this.
1148 /* Null password is ok - we are already an authenticated user... */
1149 null_pw = data_blob(NULL, 0);
1150 fstrcpy(res_type, "A:");
1151 become_root();
1152 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1153 unbecome_root();
1155 if (conn == NULL) {
1156 DEBUG(0,("get_correct_cversion: Unable to connect\n"));
1157 *perr = ntstatus_to_werror(nt_status);
1158 return -1;
1161 /* We are temporarily becoming the connection user. */
1162 if (!become_user(conn, user->vuid)) {
1163 DEBUG(0,("get_correct_cversion: Can't become user!\n"));
1164 *perr = WERR_ACCESS_DENIED;
1165 return -1;
1168 /* Open the driver file (Portable Executable format) and determine the
1169 * deriver the cversion. */
1170 slprintf(driverpath, sizeof(driverpath)-1, "%s/%s", architecture, driverpath_in);
1172 driver_unix_convert(driverpath,conn,NULL,&bad_path,&st);
1174 fsp = open_file_shared(conn, driverpath, &st,
1175 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1176 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1177 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1178 if (!fsp) {
1179 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = %d\n",
1180 driverpath, errno));
1181 *perr = WERR_ACCESS_DENIED;
1182 goto error_exit;
1184 else {
1185 uint32 major;
1186 uint32 minor;
1187 int ret = get_file_version(fsp, driverpath, &major, &minor);
1188 if (ret == -1) goto error_exit;
1190 if (!ret) {
1191 DEBUG(6,("get_correct_cversion: Version info not found [%s]\n", driverpath));
1192 goto error_exit;
1196 * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
1197 * for more details. Version in this case is not just the version of the
1198 * file, but the version in the sense of kernal mode (2) vs. user mode
1199 * (3) drivers. Other bits of the version fields are the version info.
1200 * JRR 010716
1202 cversion = major & 0x0000ffff;
1203 switch (cversion) {
1204 case 2: /* WinNT drivers */
1205 case 3: /* Win2K drivers */
1206 break;
1208 default:
1209 DEBUG(6,("get_correct_cversion: cversion invalid [%s] cversion = %d\n",
1210 driverpath, cversion));
1211 goto error_exit;
1214 DEBUG(10,("get_correct_cversion: Version info found [%s] major = 0x%x minor = 0x%x\n",
1215 driverpath, major, minor));
1218 DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
1219 driverpath, cversion));
1221 close_file(fsp, True);
1222 close_cnum(conn, user->vuid);
1223 unbecome_user();
1224 *perr = WERR_OK;
1225 return cversion;
1228 error_exit:
1230 if(fsp)
1231 close_file(fsp, True);
1233 close_cnum(conn, user->vuid);
1234 unbecome_user();
1235 return -1;
1238 /****************************************************************************
1239 ****************************************************************************/
1240 static WERROR clean_up_driver_struct_level_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver,
1241 struct current_user *user)
1243 const char *architecture;
1244 fstring new_name;
1245 char *p;
1246 int i;
1247 WERROR err;
1249 /* clean up the driver name.
1250 * we can get .\driver.dll
1251 * or worse c:\windows\system\driver.dll !
1253 /* using an intermediate string to not have overlaping memcpy()'s */
1254 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1255 fstrcpy(new_name, p+1);
1256 fstrcpy(driver->driverpath, new_name);
1259 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1260 fstrcpy(new_name, p+1);
1261 fstrcpy(driver->datafile, new_name);
1264 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1265 fstrcpy(new_name, p+1);
1266 fstrcpy(driver->configfile, new_name);
1269 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1270 fstrcpy(new_name, p+1);
1271 fstrcpy(driver->helpfile, new_name);
1274 if (driver->dependentfiles) {
1275 for (i=0; *driver->dependentfiles[i]; i++) {
1276 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1277 fstrcpy(new_name, p+1);
1278 fstrcpy(driver->dependentfiles[i], new_name);
1283 architecture = get_short_archi(driver->environment);
1285 /* jfm:7/16/2000 the client always sends the cversion=0.
1286 * The server should check which version the driver is by reading
1287 * the PE header of driver->driverpath.
1289 * For Windows 95/98 the version is 0 (so the value sent is correct)
1290 * For Windows NT (the architecture doesn't matter)
1291 * NT 3.1: cversion=0
1292 * NT 3.5/3.51: cversion=1
1293 * NT 4: cversion=2
1294 * NT2K: cversion=3
1296 if ((driver->cversion = get_correct_cversion( architecture,
1297 driver->driverpath, user, &err)) == -1)
1298 return err;
1300 return WERR_OK;
1303 /****************************************************************************
1304 ****************************************************************************/
1305 static WERROR clean_up_driver_struct_level_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver, struct current_user *user)
1307 const char *architecture;
1308 fstring new_name;
1309 char *p;
1310 int i;
1311 WERROR err;
1313 /* clean up the driver name.
1314 * we can get .\driver.dll
1315 * or worse c:\windows\system\driver.dll !
1317 /* using an intermediate string to not have overlaping memcpy()'s */
1318 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1319 fstrcpy(new_name, p+1);
1320 fstrcpy(driver->driverpath, new_name);
1323 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1324 fstrcpy(new_name, p+1);
1325 fstrcpy(driver->datafile, new_name);
1328 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1329 fstrcpy(new_name, p+1);
1330 fstrcpy(driver->configfile, new_name);
1333 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1334 fstrcpy(new_name, p+1);
1335 fstrcpy(driver->helpfile, new_name);
1338 if (driver->dependentfiles) {
1339 for (i=0; *driver->dependentfiles[i]; i++) {
1340 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1341 fstrcpy(new_name, p+1);
1342 fstrcpy(driver->dependentfiles[i], new_name);
1347 architecture = get_short_archi(driver->environment);
1349 /* jfm:7/16/2000 the client always sends the cversion=0.
1350 * The server should check which version the driver is by reading
1351 * the PE header of driver->driverpath.
1353 * For Windows 95/98 the version is 0 (so the value sent is correct)
1354 * For Windows NT (the architecture doesn't matter)
1355 * NT 3.1: cversion=0
1356 * NT 3.5/3.51: cversion=1
1357 * NT 4: cversion=2
1358 * NT2K: cversion=3
1360 if ((driver->version = get_correct_cversion(architecture, driver->driverpath, user, &err)) == -1)
1361 return err;
1363 return WERR_OK;
1366 /****************************************************************************
1367 ****************************************************************************/
1368 WERROR clean_up_driver_struct(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract,
1369 uint32 level, struct current_user *user)
1371 switch (level) {
1372 case 3:
1374 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1375 driver=driver_abstract.info_3;
1376 return clean_up_driver_struct_level_3(driver, user);
1378 case 6:
1380 NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver;
1381 driver=driver_abstract.info_6;
1382 return clean_up_driver_struct_level_6(driver, user);
1384 default:
1385 return WERR_INVALID_PARAM;
1389 /****************************************************************************
1390 This function sucks and should be replaced. JRA.
1391 ****************************************************************************/
1393 static void convert_level_6_to_level3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *dst, NT_PRINTER_DRIVER_INFO_LEVEL_6 *src)
1395 dst->cversion = src->version;
1397 fstrcpy( dst->name, src->name);
1398 fstrcpy( dst->environment, src->environment);
1399 fstrcpy( dst->driverpath, src->driverpath);
1400 fstrcpy( dst->datafile, src->datafile);
1401 fstrcpy( dst->configfile, src->configfile);
1402 fstrcpy( dst->helpfile, src->helpfile);
1403 fstrcpy( dst->monitorname, src->monitorname);
1404 fstrcpy( dst->defaultdatatype, src->defaultdatatype);
1405 dst->dependentfiles = src->dependentfiles;
1408 #if 0 /* Debugging function */
1410 static char* ffmt(unsigned char *c){
1411 int i;
1412 static char ffmt_str[17];
1414 for (i=0; i<16; i++) {
1415 if ((c[i] < ' ') || (c[i] > '~'))
1416 ffmt_str[i]='.';
1417 else
1418 ffmt_str[i]=c[i];
1420 ffmt_str[16]='\0';
1421 return ffmt_str;
1424 #endif
1426 /****************************************************************************
1427 ****************************************************************************/
1428 BOOL move_driver_to_download_area(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract, uint32 level,
1429 struct current_user *user, WERROR *perr)
1431 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1432 NT_PRINTER_DRIVER_INFO_LEVEL_3 converted_driver;
1433 const char *architecture;
1434 pstring new_dir;
1435 pstring old_name;
1436 pstring new_name;
1437 DATA_BLOB null_pw;
1438 connection_struct *conn;
1439 NTSTATUS nt_status;
1440 pstring inbuf;
1441 pstring outbuf;
1442 fstring res_type;
1443 BOOL bad_path;
1444 SMB_STRUCT_STAT st;
1445 int ver = 0;
1446 int i;
1448 memset(inbuf, '\0', sizeof(inbuf));
1449 memset(outbuf, '\0', sizeof(outbuf));
1450 *perr = WERR_OK;
1452 if (level==3)
1453 driver=driver_abstract.info_3;
1454 else if (level==6) {
1455 convert_level_6_to_level3(&converted_driver, driver_abstract.info_6);
1456 driver = &converted_driver;
1457 } else {
1458 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)level ));
1459 return False;
1462 architecture = get_short_archi(driver->environment);
1465 * Connect to the print$ share under the same account as the user connected to the rpc pipe.
1466 * Note we must be root to do this.
1469 null_pw = data_blob(NULL, 0);
1470 fstrcpy(res_type, "A:");
1471 become_root();
1472 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1473 unbecome_root();
1475 if (conn == NULL) {
1476 DEBUG(0,("move_driver_to_download_area: Unable to connect\n"));
1477 *perr = ntstatus_to_werror(nt_status);
1478 return False;
1482 * Save who we are - we are temporarily becoming the connection user.
1485 if (!become_user(conn, conn->vuid)) {
1486 DEBUG(0,("move_driver_to_download_area: Can't become user!\n"));
1487 return False;
1491 * make the directories version and version\driver_name
1492 * under the architecture directory.
1494 DEBUG(5,("Creating first directory\n"));
1495 slprintf(new_dir, sizeof(new_dir)-1, "%s/%d", architecture, driver->cversion);
1496 driver_unix_convert(new_dir, conn, NULL, &bad_path, &st);
1497 mkdir_internal(conn, new_dir);
1499 /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1500 * listed for this driver which has already been moved, skip it (note:
1501 * drivers may list the same file name several times. Then check if the
1502 * file already exists in archi\cversion\, if so, check that the version
1503 * info (or time stamps if version info is unavailable) is newer (or the
1504 * date is later). If it is, move it to archi\cversion\filexxx.yyy.
1505 * Otherwise, delete the file.
1507 * If a file is not moved to archi\cversion\ because of an error, all the
1508 * rest of the 'unmoved' driver files are removed from archi\. If one or
1509 * more of the driver's files was already moved to archi\cversion\, it
1510 * potentially leaves the driver in a partially updated state. Version
1511 * trauma will most likely occur if an client attempts to use any printer
1512 * bound to the driver. Perhaps a rewrite to make sure the moves can be
1513 * done is appropriate... later JRR
1516 DEBUG(5,("Moving files now !\n"));
1518 if (driver->driverpath && strlen(driver->driverpath)) {
1519 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->driverpath);
1520 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->driverpath);
1521 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1522 NTSTATUS status;
1523 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1524 status = rename_internals(conn, new_name, old_name, 0, True);
1525 if (!NT_STATUS_IS_OK(status)) {
1526 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1527 new_name, old_name));
1528 *perr = ntstatus_to_werror(status);
1529 unlink_internals(conn, 0, new_name);
1530 ver = -1;
1532 } else {
1533 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1534 unlink_internals(conn, 0, new_name);
1538 if (driver->datafile && strlen(driver->datafile)) {
1539 if (!strequal(driver->datafile, driver->driverpath)) {
1540 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->datafile);
1541 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->datafile);
1542 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1543 NTSTATUS status;
1544 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1545 status = rename_internals(conn, new_name, old_name, 0, True);
1546 if (!NT_STATUS_IS_OK(status)) {
1547 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1548 new_name, old_name));
1549 *perr = ntstatus_to_werror(status);
1550 unlink_internals(conn, 0, new_name);
1551 ver = -1;
1553 } else {
1554 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1555 unlink_internals(conn, 0, new_name);
1560 if (driver->configfile && strlen(driver->configfile)) {
1561 if (!strequal(driver->configfile, driver->driverpath) &&
1562 !strequal(driver->configfile, driver->datafile)) {
1563 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->configfile);
1564 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->configfile);
1565 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1566 NTSTATUS status;
1567 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1568 status = rename_internals(conn, new_name, old_name, 0, True);
1569 if (!NT_STATUS_IS_OK(status)) {
1570 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1571 new_name, old_name));
1572 *perr = ntstatus_to_werror(status);
1573 unlink_internals(conn, 0, new_name);
1574 ver = -1;
1576 } else {
1577 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1578 unlink_internals(conn, 0, new_name);
1583 if (driver->helpfile && strlen(driver->helpfile)) {
1584 if (!strequal(driver->helpfile, driver->driverpath) &&
1585 !strequal(driver->helpfile, driver->datafile) &&
1586 !strequal(driver->helpfile, driver->configfile)) {
1587 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->helpfile);
1588 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->helpfile);
1589 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1590 NTSTATUS status;
1591 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1592 status = rename_internals(conn, new_name, old_name, 0, True);
1593 if (!NT_STATUS_IS_OK(status)) {
1594 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1595 new_name, old_name));
1596 *perr = ntstatus_to_werror(status);
1597 unlink_internals(conn, 0, new_name);
1598 ver = -1;
1600 } else {
1601 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1602 unlink_internals(conn, 0, new_name);
1607 if (driver->dependentfiles) {
1608 for (i=0; *driver->dependentfiles[i]; i++) {
1609 if (!strequal(driver->dependentfiles[i], driver->driverpath) &&
1610 !strequal(driver->dependentfiles[i], driver->datafile) &&
1611 !strequal(driver->dependentfiles[i], driver->configfile) &&
1612 !strequal(driver->dependentfiles[i], driver->helpfile)) {
1613 int j;
1614 for (j=0; j < i; j++) {
1615 if (strequal(driver->dependentfiles[i], driver->dependentfiles[j])) {
1616 goto NextDriver;
1620 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->dependentfiles[i]);
1621 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->dependentfiles[i]);
1622 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1623 NTSTATUS status;
1624 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1625 status = rename_internals(conn, new_name, old_name, 0, True);
1626 if (!NT_STATUS_IS_OK(status)) {
1627 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1628 new_name, old_name));
1629 *perr = ntstatus_to_werror(status);
1630 unlink_internals(conn, 0, new_name);
1631 ver = -1;
1633 } else {
1634 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1635 unlink_internals(conn, 0, new_name);
1638 NextDriver: ;
1642 close_cnum(conn, user->vuid);
1643 unbecome_user();
1645 return ver == -1 ? False : True;
1648 /****************************************************************************
1649 ****************************************************************************/
1650 static uint32 add_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver)
1652 int len, buflen;
1653 const char *architecture;
1654 pstring directory;
1655 fstring temp_name;
1656 pstring key;
1657 char *buf;
1658 int i, ret;
1659 TDB_DATA kbuf, dbuf;
1661 architecture = get_short_archi(driver->environment);
1663 /* The names are relative. We store them in the form: \print$\arch\version\driver.xxx
1664 * \\server is added in the rpc server layer.
1665 * It does make sense to NOT store the server's name in the printer TDB.
1668 slprintf(directory, sizeof(directory)-1, "\\print$\\%s\\%d\\", architecture, driver->cversion);
1670 /* .inf files do not always list a file for each of the four standard files.
1671 * Don't prepend a path to a null filename, or client claims:
1672 * "The server on which the printer resides does not have a suitable
1673 * <printer driver name> printer driver installed. Click OK if you
1674 * wish to install the driver on your local machine."
1676 if (strlen(driver->driverpath)) {
1677 fstrcpy(temp_name, driver->driverpath);
1678 slprintf(driver->driverpath, sizeof(driver->driverpath)-1, "%s%s", directory, temp_name);
1681 if (strlen(driver->datafile)) {
1682 fstrcpy(temp_name, driver->datafile);
1683 slprintf(driver->datafile, sizeof(driver->datafile)-1, "%s%s", directory, temp_name);
1686 if (strlen(driver->configfile)) {
1687 fstrcpy(temp_name, driver->configfile);
1688 slprintf(driver->configfile, sizeof(driver->configfile)-1, "%s%s", directory, temp_name);
1691 if (strlen(driver->helpfile)) {
1692 fstrcpy(temp_name, driver->helpfile);
1693 slprintf(driver->helpfile, sizeof(driver->helpfile)-1, "%s%s", directory, temp_name);
1696 if (driver->dependentfiles) {
1697 for (i=0; *driver->dependentfiles[i]; i++) {
1698 fstrcpy(temp_name, driver->dependentfiles[i]);
1699 slprintf(driver->dependentfiles[i], sizeof(driver->dependentfiles[i])-1, "%s%s", directory, temp_name);
1703 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, driver->cversion, driver->name);
1705 DEBUG(5,("add_a_printer_driver_3: Adding driver with key %s\n", key ));
1707 buf = NULL;
1708 len = buflen = 0;
1710 again:
1711 len = 0;
1712 len += tdb_pack(buf+len, buflen-len, "dffffffff",
1713 driver->cversion,
1714 driver->name,
1715 driver->environment,
1716 driver->driverpath,
1717 driver->datafile,
1718 driver->configfile,
1719 driver->helpfile,
1720 driver->monitorname,
1721 driver->defaultdatatype);
1723 if (driver->dependentfiles) {
1724 for (i=0; *driver->dependentfiles[i]; i++) {
1725 len += tdb_pack(buf+len, buflen-len, "f",
1726 driver->dependentfiles[i]);
1730 if (len != buflen) {
1731 char *tb;
1733 tb = (char *)SMB_REALLOC(buf, len);
1734 if (!tb) {
1735 DEBUG(0,("add_a_printer_driver_3: failed to enlarge buffer\n!"));
1736 ret = -1;
1737 goto done;
1739 else buf = tb;
1740 buflen = len;
1741 goto again;
1745 kbuf.dptr = key;
1746 kbuf.dsize = strlen(key)+1;
1747 dbuf.dptr = buf;
1748 dbuf.dsize = len;
1750 ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
1752 done:
1753 if (ret)
1754 DEBUG(0,("add_a_printer_driver_3: Adding driver with key %s failed.\n", key ));
1756 SAFE_FREE(buf);
1757 return ret;
1760 /****************************************************************************
1761 ****************************************************************************/
1762 static uint32 add_a_printer_driver_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver)
1764 NT_PRINTER_DRIVER_INFO_LEVEL_3 info3;
1766 ZERO_STRUCT(info3);
1767 info3.cversion = driver->version;
1768 fstrcpy(info3.name,driver->name);
1769 fstrcpy(info3.environment,driver->environment);
1770 fstrcpy(info3.driverpath,driver->driverpath);
1771 fstrcpy(info3.datafile,driver->datafile);
1772 fstrcpy(info3.configfile,driver->configfile);
1773 fstrcpy(info3.helpfile,driver->helpfile);
1774 fstrcpy(info3.monitorname,driver->monitorname);
1775 fstrcpy(info3.defaultdatatype,driver->defaultdatatype);
1776 info3.dependentfiles = driver->dependentfiles;
1778 return add_a_printer_driver_3(&info3);
1782 /****************************************************************************
1783 ****************************************************************************/
1784 static WERROR get_a_printer_driver_3_default(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, const char *driver, const char *arch)
1786 NT_PRINTER_DRIVER_INFO_LEVEL_3 info;
1788 ZERO_STRUCT(info);
1790 fstrcpy(info.name, driver);
1791 fstrcpy(info.defaultdatatype, "RAW");
1793 fstrcpy(info.driverpath, "");
1794 fstrcpy(info.datafile, "");
1795 fstrcpy(info.configfile, "");
1796 fstrcpy(info.helpfile, "");
1798 if ((info.dependentfiles= SMB_MALLOC_ARRAY(fstring, 2)) == NULL)
1799 return WERR_NOMEM;
1801 memset(info.dependentfiles, '\0', 2*sizeof(fstring));
1802 fstrcpy(info.dependentfiles[0], "");
1804 *info_ptr = memdup(&info, sizeof(info));
1806 return WERR_OK;
1809 /****************************************************************************
1810 ****************************************************************************/
1811 static WERROR get_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring drivername, const char *arch, uint32 version)
1813 NT_PRINTER_DRIVER_INFO_LEVEL_3 driver;
1814 TDB_DATA kbuf, dbuf;
1815 const char *architecture;
1816 int len = 0;
1817 int i;
1818 pstring key;
1820 ZERO_STRUCT(driver);
1822 architecture = get_short_archi(arch);
1824 /* Windows 4.0 (i.e. win9x) should always use a version of 0 */
1826 if ( strcmp( architecture, SPL_ARCH_WIN40 ) == 0 )
1827 version = 0;
1829 DEBUG(8,("get_a_printer_driver_3: [%s%s/%d/%s]\n", DRIVERS_PREFIX, architecture, version, drivername));
1831 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, version, drivername);
1833 kbuf.dptr = key;
1834 kbuf.dsize = strlen(key)+1;
1836 dbuf = tdb_fetch(tdb_drivers, kbuf);
1837 if (!dbuf.dptr)
1838 return WERR_UNKNOWN_PRINTER_DRIVER;
1840 len += tdb_unpack(dbuf.dptr, dbuf.dsize, "dffffffff",
1841 &driver.cversion,
1842 driver.name,
1843 driver.environment,
1844 driver.driverpath,
1845 driver.datafile,
1846 driver.configfile,
1847 driver.helpfile,
1848 driver.monitorname,
1849 driver.defaultdatatype);
1851 i=0;
1852 while (len < dbuf.dsize) {
1853 fstring *tddfs;
1855 tddfs = SMB_REALLOC_ARRAY(driver.dependentfiles, fstring, i+2);
1856 if (tddfs == NULL) {
1857 DEBUG(0,("get_a_printer_driver_3: failed to enlarge buffer!\n"));
1858 break;
1860 else driver.dependentfiles = tddfs;
1862 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f",
1863 &driver.dependentfiles[i]);
1864 i++;
1867 if (driver.dependentfiles != NULL)
1868 fstrcpy(driver.dependentfiles[i], "");
1870 SAFE_FREE(dbuf.dptr);
1872 if (len != dbuf.dsize) {
1873 SAFE_FREE(driver.dependentfiles);
1875 return get_a_printer_driver_3_default(info_ptr, drivername, arch);
1878 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&driver, sizeof(driver));
1880 return WERR_OK;
1883 /****************************************************************************
1884 Debugging function, dump at level 6 the struct in the logs.
1885 ****************************************************************************/
1887 static uint32 dump_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
1889 uint32 result;
1890 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
1891 int i;
1893 DEBUG(20,("Dumping printer driver at level [%d]\n", level));
1895 switch (level)
1897 case 3:
1899 if (driver.info_3 == NULL)
1900 result=5;
1901 else {
1902 info3=driver.info_3;
1904 DEBUGADD(20,("version:[%d]\n", info3->cversion));
1905 DEBUGADD(20,("name:[%s]\n", info3->name));
1906 DEBUGADD(20,("environment:[%s]\n", info3->environment));
1907 DEBUGADD(20,("driverpath:[%s]\n", info3->driverpath));
1908 DEBUGADD(20,("datafile:[%s]\n", info3->datafile));
1909 DEBUGADD(20,("configfile:[%s]\n", info3->configfile));
1910 DEBUGADD(20,("helpfile:[%s]\n", info3->helpfile));
1911 DEBUGADD(20,("monitorname:[%s]\n", info3->monitorname));
1912 DEBUGADD(20,("defaultdatatype:[%s]\n", info3->defaultdatatype));
1914 for (i=0; info3->dependentfiles &&
1915 *info3->dependentfiles[i]; i++) {
1916 DEBUGADD(20,("dependentfile:[%s]\n",
1917 info3->dependentfiles[i]));
1919 result=0;
1921 break;
1923 default:
1924 DEBUGADD(20,("dump_a_printer_driver: Level %u not implemented\n", (unsigned int)level));
1925 result=1;
1926 break;
1929 return result;
1932 /****************************************************************************
1933 ****************************************************************************/
1934 int pack_devicemode(NT_DEVICEMODE *nt_devmode, char *buf, int buflen)
1936 int len = 0;
1938 len += tdb_pack(buf+len, buflen-len, "p", nt_devmode);
1940 if (!nt_devmode)
1941 return len;
1943 len += tdb_pack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
1944 nt_devmode->devicename,
1945 nt_devmode->formname,
1947 nt_devmode->specversion,
1948 nt_devmode->driverversion,
1949 nt_devmode->size,
1950 nt_devmode->driverextra,
1951 nt_devmode->orientation,
1952 nt_devmode->papersize,
1953 nt_devmode->paperlength,
1954 nt_devmode->paperwidth,
1955 nt_devmode->scale,
1956 nt_devmode->copies,
1957 nt_devmode->defaultsource,
1958 nt_devmode->printquality,
1959 nt_devmode->color,
1960 nt_devmode->duplex,
1961 nt_devmode->yresolution,
1962 nt_devmode->ttoption,
1963 nt_devmode->collate,
1964 nt_devmode->logpixels,
1966 nt_devmode->fields,
1967 nt_devmode->bitsperpel,
1968 nt_devmode->pelswidth,
1969 nt_devmode->pelsheight,
1970 nt_devmode->displayflags,
1971 nt_devmode->displayfrequency,
1972 nt_devmode->icmmethod,
1973 nt_devmode->icmintent,
1974 nt_devmode->mediatype,
1975 nt_devmode->dithertype,
1976 nt_devmode->reserved1,
1977 nt_devmode->reserved2,
1978 nt_devmode->panningwidth,
1979 nt_devmode->panningheight,
1980 nt_devmode->private);
1983 if (nt_devmode->private) {
1984 len += tdb_pack(buf+len, buflen-len, "B",
1985 nt_devmode->driverextra,
1986 nt_devmode->private);
1989 DEBUG(8,("Packed devicemode [%s]\n", nt_devmode->formname));
1991 return len;
1994 /****************************************************************************
1995 Pack all values in all printer keys
1996 ***************************************************************************/
1998 static int pack_values(NT_PRINTER_DATA *data, char *buf, int buflen)
2000 int len = 0;
2001 int i, j;
2002 REGISTRY_VALUE *val;
2003 REGVAL_CTR *val_ctr;
2004 pstring path;
2005 int num_values;
2007 if ( !data )
2008 return 0;
2010 /* loop over all keys */
2012 for ( i=0; i<data->num_keys; i++ ) {
2013 val_ctr = &data->keys[i].values;
2014 num_values = regval_ctr_numvals( val_ctr );
2016 /* loop over all values */
2018 for ( j=0; j<num_values; j++ ) {
2019 /* pathname should be stored as <key>\<value> */
2021 val = regval_ctr_specific_value( val_ctr, j );
2022 pstrcpy( path, data->keys[i].name );
2023 pstrcat( path, "\\" );
2024 pstrcat( path, regval_name(val) );
2026 len += tdb_pack(buf+len, buflen-len, "pPdB",
2027 val,
2028 path,
2029 regval_type(val),
2030 regval_size(val),
2031 regval_data_p(val) );
2036 /* terminator */
2038 len += tdb_pack(buf+len, buflen-len, "p", NULL);
2040 return len;
2044 /****************************************************************************
2045 Delete a printer - this just deletes the printer info file, any open
2046 handles are not affected.
2047 ****************************************************************************/
2049 uint32 del_a_printer(char *sharename)
2051 pstring key;
2052 TDB_DATA kbuf;
2054 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
2056 kbuf.dptr=key;
2057 kbuf.dsize=strlen(key)+1;
2059 tdb_delete(tdb_printers, kbuf);
2060 return 0;
2063 /****************************************************************************
2064 ****************************************************************************/
2065 static WERROR update_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info)
2067 pstring key;
2068 char *buf;
2069 int buflen, len;
2070 WERROR ret;
2071 TDB_DATA kbuf, dbuf;
2074 * in addprinter: no servername and the printer is the name
2075 * in setprinter: servername is \\server
2076 * and printer is \\server\\printer
2078 * Samba manages only local printers.
2079 * we currently don't support things like i
2080 * path=\\other_server\printer
2082 * We only store the printername, not \\server\printername
2085 if ( info->servername[0] != '\0' ) {
2086 trim_string(info->printername, info->servername, NULL);
2087 trim_char(info->printername, '\\', '\0');
2088 info->servername[0]='\0';
2092 * JFM: one day I'll forget.
2093 * below that's info->portname because that's the SAMBA sharename
2094 * and I made NT 'thinks' it's the portname
2095 * the info->sharename is the thing you can name when you add a printer
2096 * that's the short-name when you create shared printer for 95/98
2097 * So I've made a limitation in SAMBA: you can only have 1 printer model
2098 * behind a SAMBA share.
2101 buf = NULL;
2102 buflen = 0;
2104 again:
2105 len = 0;
2106 len += tdb_pack(buf+len, buflen-len, "dddddddddddfffffPfffff",
2107 info->attributes,
2108 info->priority,
2109 info->default_priority,
2110 info->starttime,
2111 info->untiltime,
2112 info->status,
2113 info->cjobs,
2114 info->averageppm,
2115 info->changeid,
2116 info->c_setprinter,
2117 info->setuptime,
2118 info->servername,
2119 info->printername,
2120 info->sharename,
2121 info->portname,
2122 info->drivername,
2123 info->comment,
2124 info->location,
2125 info->sepfile,
2126 info->printprocessor,
2127 info->datatype,
2128 info->parameters);
2130 len += pack_devicemode(info->devmode, buf+len, buflen-len);
2132 len += pack_values( &info->data, buf+len, buflen-len );
2134 if (buflen != len) {
2135 char *tb;
2137 tb = (char *)SMB_REALLOC(buf, len);
2138 if (!tb) {
2139 DEBUG(0,("update_a_printer_2: failed to enlarge buffer!\n"));
2140 ret = WERR_NOMEM;
2141 goto done;
2143 else buf = tb;
2144 buflen = len;
2145 goto again;
2149 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, info->sharename);
2151 kbuf.dptr = key;
2152 kbuf.dsize = strlen(key)+1;
2153 dbuf.dptr = buf;
2154 dbuf.dsize = len;
2156 ret = (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) == 0? WERR_OK : WERR_NOMEM);
2158 done:
2159 if (!W_ERROR_IS_OK(ret))
2160 DEBUG(8, ("error updating printer to tdb on disk\n"));
2162 SAFE_FREE(buf);
2164 DEBUG(8,("packed printer [%s] with driver [%s] portname=[%s] len=%d\n",
2165 info->sharename, info->drivername, info->portname, len));
2167 return ret;
2171 /****************************************************************************
2172 Malloc and return an NT devicemode.
2173 ****************************************************************************/
2175 NT_DEVICEMODE *construct_nt_devicemode(const fstring default_devicename)
2178 char adevice[MAXDEVICENAME];
2179 NT_DEVICEMODE *nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE);
2181 if (nt_devmode == NULL) {
2182 DEBUG(0,("construct_nt_devicemode: malloc fail.\n"));
2183 return NULL;
2186 ZERO_STRUCTP(nt_devmode);
2188 slprintf(adevice, sizeof(adevice), "%s", default_devicename);
2189 fstrcpy(nt_devmode->devicename, adevice);
2191 fstrcpy(nt_devmode->formname, "Letter");
2193 nt_devmode->specversion = 0x0401;
2194 nt_devmode->driverversion = 0x0400;
2195 nt_devmode->size = 0x00DC;
2196 nt_devmode->driverextra = 0x0000;
2197 nt_devmode->fields = FORMNAME | TTOPTION | PRINTQUALITY |
2198 DEFAULTSOURCE | COPIES | SCALE |
2199 PAPERSIZE | ORIENTATION;
2200 nt_devmode->orientation = 1;
2201 nt_devmode->papersize = PAPER_LETTER;
2202 nt_devmode->paperlength = 0;
2203 nt_devmode->paperwidth = 0;
2204 nt_devmode->scale = 0x64;
2205 nt_devmode->copies = 1;
2206 nt_devmode->defaultsource = BIN_FORMSOURCE;
2207 nt_devmode->printquality = RES_HIGH; /* 0x0258 */
2208 nt_devmode->color = COLOR_MONOCHROME;
2209 nt_devmode->duplex = DUP_SIMPLEX;
2210 nt_devmode->yresolution = 0;
2211 nt_devmode->ttoption = TT_SUBDEV;
2212 nt_devmode->collate = COLLATE_FALSE;
2213 nt_devmode->icmmethod = 0;
2214 nt_devmode->icmintent = 0;
2215 nt_devmode->mediatype = 0;
2216 nt_devmode->dithertype = 0;
2218 /* non utilisés par un driver d'imprimante */
2219 nt_devmode->logpixels = 0;
2220 nt_devmode->bitsperpel = 0;
2221 nt_devmode->pelswidth = 0;
2222 nt_devmode->pelsheight = 0;
2223 nt_devmode->displayflags = 0;
2224 nt_devmode->displayfrequency = 0;
2225 nt_devmode->reserved1 = 0;
2226 nt_devmode->reserved2 = 0;
2227 nt_devmode->panningwidth = 0;
2228 nt_devmode->panningheight = 0;
2230 nt_devmode->private = NULL;
2231 return nt_devmode;
2234 /****************************************************************************
2235 Deepcopy an NT devicemode.
2236 ****************************************************************************/
2238 NT_DEVICEMODE *dup_nt_devicemode(NT_DEVICEMODE *nt_devicemode)
2240 NT_DEVICEMODE *new_nt_devicemode = NULL;
2242 if ( !nt_devicemode )
2243 return NULL;
2245 if ((new_nt_devicemode = (NT_DEVICEMODE *)memdup(nt_devicemode, sizeof(NT_DEVICEMODE))) == NULL) {
2246 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2247 return NULL;
2250 new_nt_devicemode->private = NULL;
2251 if (nt_devicemode->private != NULL) {
2252 if ((new_nt_devicemode->private = memdup(nt_devicemode->private, nt_devicemode->driverextra)) == NULL) {
2253 SAFE_FREE(new_nt_devicemode);
2254 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2255 return NULL;
2259 return new_nt_devicemode;
2262 /****************************************************************************
2263 Clean up and deallocate a (maybe partially) allocated NT_DEVICEMODE.
2264 ****************************************************************************/
2266 void free_nt_devicemode(NT_DEVICEMODE **devmode_ptr)
2268 NT_DEVICEMODE *nt_devmode = *devmode_ptr;
2270 if(nt_devmode == NULL)
2271 return;
2273 DEBUG(106,("free_nt_devicemode: deleting DEVMODE\n"));
2275 SAFE_FREE(nt_devmode->private);
2276 SAFE_FREE(*devmode_ptr);
2279 /****************************************************************************
2280 Clean up and deallocate a (maybe partially) allocated NT_PRINTER_INFO_LEVEL_2.
2281 ****************************************************************************/
2282 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr)
2284 NT_PRINTER_INFO_LEVEL_2 *info = *info_ptr;
2285 NT_PRINTER_DATA *data;
2286 int i;
2288 if ( !info )
2289 return;
2291 DEBUG(106,("free_nt_printer_info_level_2: deleting info\n"));
2293 free_nt_devicemode(&info->devmode);
2295 /* clean up all registry keys */
2297 data = &info->data;
2298 for ( i=0; i<data->num_keys; i++ ) {
2299 SAFE_FREE( data->keys[i].name );
2300 regval_ctr_destroy( &data->keys[i].values );
2302 SAFE_FREE( data->keys );
2304 /* finally the top level structure */
2306 SAFE_FREE( *info_ptr );
2310 /****************************************************************************
2311 ****************************************************************************/
2312 int unpack_devicemode(NT_DEVICEMODE **nt_devmode, char *buf, int buflen)
2314 int len = 0;
2315 int extra_len = 0;
2316 NT_DEVICEMODE devmode;
2318 ZERO_STRUCT(devmode);
2320 len += tdb_unpack(buf+len, buflen-len, "p", nt_devmode);
2322 if (!*nt_devmode) return len;
2324 len += tdb_unpack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2325 devmode.devicename,
2326 devmode.formname,
2328 &devmode.specversion,
2329 &devmode.driverversion,
2330 &devmode.size,
2331 &devmode.driverextra,
2332 &devmode.orientation,
2333 &devmode.papersize,
2334 &devmode.paperlength,
2335 &devmode.paperwidth,
2336 &devmode.scale,
2337 &devmode.copies,
2338 &devmode.defaultsource,
2339 &devmode.printquality,
2340 &devmode.color,
2341 &devmode.duplex,
2342 &devmode.yresolution,
2343 &devmode.ttoption,
2344 &devmode.collate,
2345 &devmode.logpixels,
2347 &devmode.fields,
2348 &devmode.bitsperpel,
2349 &devmode.pelswidth,
2350 &devmode.pelsheight,
2351 &devmode.displayflags,
2352 &devmode.displayfrequency,
2353 &devmode.icmmethod,
2354 &devmode.icmintent,
2355 &devmode.mediatype,
2356 &devmode.dithertype,
2357 &devmode.reserved1,
2358 &devmode.reserved2,
2359 &devmode.panningwidth,
2360 &devmode.panningheight,
2361 &devmode.private);
2363 if (devmode.private) {
2364 /* the len in tdb_unpack is an int value and
2365 * devmode.driverextra is only a short
2367 len += tdb_unpack(buf+len, buflen-len, "B", &extra_len, &devmode.private);
2368 devmode.driverextra=(uint16)extra_len;
2370 /* check to catch an invalid TDB entry so we don't segfault */
2371 if (devmode.driverextra == 0) {
2372 devmode.private = NULL;
2376 *nt_devmode = (NT_DEVICEMODE *)memdup(&devmode, sizeof(devmode));
2378 DEBUG(8,("Unpacked devicemode [%s](%s)\n", devmode.devicename, devmode.formname));
2379 if (devmode.private)
2380 DEBUG(8,("with a private section of %d bytes\n", devmode.driverextra));
2382 return len;
2385 /****************************************************************************
2386 Allocate and initialize a new slot.
2387 ***************************************************************************/
2389 static int add_new_printer_key( NT_PRINTER_DATA *data, const char *name )
2391 NT_PRINTER_KEY *d;
2392 int key_index;
2394 if ( !data || !name )
2395 return -1;
2397 /* allocate another slot in the NT_PRINTER_KEY array */
2399 d = SMB_REALLOC_ARRAY( data->keys, NT_PRINTER_KEY, data->num_keys+1);
2400 if ( d )
2401 data->keys = d;
2403 key_index = data->num_keys;
2405 /* initialze new key */
2407 data->num_keys++;
2408 data->keys[key_index].name = SMB_STRDUP( name );
2410 ZERO_STRUCTP( &data->keys[key_index].values );
2412 regval_ctr_init( &data->keys[key_index].values );
2414 DEBUG(10,("add_new_printer_key: Inserted new data key [%s]\n", name ));
2416 return key_index;
2419 /****************************************************************************
2420 search for a registry key name in the existing printer data
2421 ***************************************************************************/
2423 int lookup_printerkey( NT_PRINTER_DATA *data, const char *name )
2425 int key_index = -1;
2426 int i;
2428 if ( !data || !name )
2429 return -1;
2431 DEBUG(12,("lookup_printerkey: Looking for [%s]\n", name));
2433 /* loop over all existing keys */
2435 for ( i=0; i<data->num_keys; i++ ) {
2436 if ( strequal(data->keys[i].name, name) ) {
2437 DEBUG(12,("lookup_printerkey: Found [%s]!\n", name));
2438 key_index = i;
2439 break;
2444 return key_index;
2447 /****************************************************************************
2448 ***************************************************************************/
2450 uint32 get_printer_subkeys( NT_PRINTER_DATA *data, const char* key, fstring **subkeys )
2452 int i, j;
2453 int key_len;
2454 int num_subkeys = 0;
2455 char *p;
2456 fstring *ptr, *subkeys_ptr = NULL;
2457 fstring subkeyname;
2459 if ( !data )
2460 return 0;
2462 for ( i=0; i<data->num_keys; i++ ) {
2463 if ( StrnCaseCmp(data->keys[i].name, key, strlen(key)) == 0 ) {
2464 /* match sure it is a subkey and not the key itself */
2466 key_len = strlen( key );
2467 if ( strlen(data->keys[i].name) == key_len )
2468 continue;
2470 /* get subkey path */
2472 p = data->keys[i].name + key_len;
2473 if ( *p == '\\' )
2474 p++;
2475 fstrcpy( subkeyname, p );
2476 if ( (p = strchr( subkeyname, '\\' )) )
2477 *p = '\0';
2479 /* don't add a key more than once */
2481 for ( j=0; j<num_subkeys; j++ ) {
2482 if ( strequal( subkeys_ptr[j], subkeyname ) )
2483 break;
2486 if ( j != num_subkeys )
2487 continue;
2489 /* found a match, so allocate space and copy the name */
2491 if ( !(ptr = SMB_REALLOC_ARRAY( subkeys_ptr, fstring, num_subkeys+2)) ) {
2492 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
2493 num_subkeys+1));
2494 SAFE_FREE( subkeys );
2495 return 0;
2498 subkeys_ptr = ptr;
2499 fstrcpy( subkeys_ptr[num_subkeys], subkeyname );
2500 num_subkeys++;
2505 /* tag of the end */
2507 if (num_subkeys)
2508 fstrcpy(subkeys_ptr[num_subkeys], "" );
2510 *subkeys = subkeys_ptr;
2512 return num_subkeys;
2515 #ifdef HAVE_ADS
2516 static void map_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2517 const char *sz)
2519 smb_ucs2_t conv_str[1024];
2520 size_t str_size;
2522 regval_ctr_delvalue(ctr, val_name);
2523 str_size = push_ucs2(NULL, conv_str, sz, sizeof(conv_str),
2524 STR_TERMINATE | STR_NOALIGN);
2525 regval_ctr_addvalue(ctr, val_name, REG_SZ,
2526 (char *) conv_str, str_size);
2529 static void map_dword_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2530 uint32 dword)
2532 regval_ctr_delvalue(ctr, val_name);
2533 regval_ctr_addvalue(ctr, val_name, REG_DWORD,
2534 (char *) &dword, sizeof(dword));
2537 static void map_bool_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2538 BOOL b)
2540 uint8 bin_bool = (b ? 1 : 0);
2541 regval_ctr_delvalue(ctr, val_name);
2542 regval_ctr_addvalue(ctr, val_name, REG_BINARY,
2543 (char *) &bin_bool, sizeof(bin_bool));
2546 static void map_single_multi_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2547 const char *multi_sz)
2549 smb_ucs2_t *conv_strs = NULL;
2550 size_t str_size;
2552 /* a multi-sz has to have a null string terminator, i.e., the last
2553 string must be followed by two nulls */
2554 str_size = strlen(multi_sz) + 2;
2555 conv_strs = SMB_CALLOC_ARRAY(smb_ucs2_t, str_size);
2556 if (!conv_strs) {
2557 return;
2560 /* Change to byte units. */
2561 str_size *= sizeof(smb_ucs2_t);
2562 push_ucs2(NULL, conv_strs, multi_sz, str_size,
2563 STR_TERMINATE | STR_NOALIGN);
2565 regval_ctr_delvalue(ctr, val_name);
2566 regval_ctr_addvalue(ctr, val_name, REG_MULTI_SZ,
2567 (char *) conv_strs, str_size);
2568 safe_free(conv_strs);
2572 /****************************************************************************
2573 * Map the NT_PRINTER_INFO_LEVEL_2 data into DsSpooler keys for publishing.
2575 * @param info2 NT_PRINTER_INFO_LEVEL_2 describing printer - gets modified
2576 * @return BOOL indicating success or failure
2577 ***************************************************************************/
2579 static BOOL map_nt_printer_info2_to_dsspooler(NT_PRINTER_INFO_LEVEL_2 *info2)
2581 REGVAL_CTR *ctr = NULL;
2582 fstring longname;
2583 char *allocated_string = NULL;
2584 const char *ascii_str;
2585 int i;
2587 if ((i = lookup_printerkey(&info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
2588 i = add_new_printer_key(&info2->data, SPOOL_DSSPOOLER_KEY);
2589 ctr = &info2->data.keys[i].values;
2591 map_sz_into_ctr(ctr, SPOOL_REG_PRINTERNAME, info2->sharename);
2592 map_sz_into_ctr(ctr, SPOOL_REG_SHORTSERVERNAME, global_myname());
2594 get_mydnsfullname(longname);
2595 map_sz_into_ctr(ctr, SPOOL_REG_SERVERNAME, longname);
2597 asprintf(&allocated_string, "\\\\%s\\%s", longname, info2->sharename);
2598 map_sz_into_ctr(ctr, SPOOL_REG_UNCNAME, allocated_string);
2599 SAFE_FREE(allocated_string);
2601 map_dword_into_ctr(ctr, SPOOL_REG_VERSIONNUMBER, 4);
2602 map_sz_into_ctr(ctr, SPOOL_REG_DRIVERNAME, info2->drivername);
2603 map_sz_into_ctr(ctr, SPOOL_REG_LOCATION, info2->location);
2604 map_sz_into_ctr(ctr, SPOOL_REG_DESCRIPTION, info2->comment);
2605 map_single_multi_sz_into_ctr(ctr, SPOOL_REG_PORTNAME, info2->portname);
2606 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSEPARATORFILE, info2->sepfile);
2607 map_dword_into_ctr(ctr, SPOOL_REG_PRINTSTARTTIME, info2->starttime);
2608 map_dword_into_ctr(ctr, SPOOL_REG_PRINTENDTIME, info2->untiltime);
2609 map_dword_into_ctr(ctr, SPOOL_REG_PRIORITY, info2->priority);
2611 map_bool_into_ctr(ctr, SPOOL_REG_PRINTKEEPPRINTEDJOBS,
2612 (info2->attributes &
2613 PRINTER_ATTRIBUTE_KEEPPRINTEDJOBS));
2615 switch (info2->attributes & 0x3) {
2616 case 0:
2617 ascii_str = SPOOL_REGVAL_PRINTWHILESPOOLING;
2618 break;
2619 case 1:
2620 ascii_str = SPOOL_REGVAL_PRINTAFTERSPOOLED;
2621 break;
2622 case 2:
2623 ascii_str = SPOOL_REGVAL_PRINTDIRECT;
2624 break;
2625 default:
2626 ascii_str = "unknown";
2628 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSPOOLING, ascii_str);
2630 return True;
2633 static void store_printer_guid(NT_PRINTER_INFO_LEVEL_2 *info2,
2634 struct uuid guid)
2636 int i;
2637 REGVAL_CTR *ctr=NULL;
2639 /* find the DsSpooler key */
2640 if ((i = lookup_printerkey(&info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
2641 i = add_new_printer_key(&info2->data, SPOOL_DSSPOOLER_KEY);
2642 ctr = &info2->data.keys[i].values;
2644 regval_ctr_delvalue(ctr, "objectGUID");
2645 regval_ctr_addvalue(ctr, "objectGUID", REG_BINARY,
2646 (char *) &guid, sizeof(struct uuid));
2649 static WERROR nt_printer_publish_ads(ADS_STRUCT *ads,
2650 NT_PRINTER_INFO_LEVEL *printer)
2652 ADS_STATUS ads_rc;
2653 void *res;
2654 char *prt_dn = NULL, *srv_dn, *srv_cn_0;
2655 char *srv_dn_utf8, **srv_cn_utf8;
2656 TALLOC_CTX *ctx;
2657 ADS_MODLIST mods;
2658 const char *attrs[] = {"objectGUID", NULL};
2659 struct uuid guid;
2660 WERROR win_rc = WERR_OK;
2662 DEBUG(5, ("publishing printer %s\n", printer->info_2->printername));
2664 /* figure out where to publish */
2665 ads_find_machine_acct(ads, &res, global_myname());
2667 /* We use ldap_get_dn here as we need the answer
2668 * in utf8 to call ldap_explode_dn(). JRA. */
2670 srv_dn_utf8 = ldap_get_dn(ads->ld, res);
2671 if (!srv_dn_utf8) {
2672 ads_destroy(&ads);
2673 return WERR_SERVER_UNAVAILABLE;
2675 ads_msgfree(ads, res);
2676 srv_cn_utf8 = ldap_explode_dn(srv_dn_utf8, 1);
2677 if (!srv_cn_utf8) {
2678 ldap_memfree(srv_dn_utf8);
2679 ads_destroy(&ads);
2680 return WERR_SERVER_UNAVAILABLE;
2682 /* Now convert to CH_UNIX. */
2683 if (pull_utf8_allocate(&srv_dn, srv_dn_utf8) == (size_t)-1) {
2684 ldap_memfree(srv_dn_utf8);
2685 ldap_memfree(srv_cn_utf8);
2686 ads_destroy(&ads);
2687 return WERR_SERVER_UNAVAILABLE;
2689 if (pull_utf8_allocate(&srv_cn_0, srv_cn_utf8[0]) == (size_t)-1) {
2690 ldap_memfree(srv_dn_utf8);
2691 ldap_memfree(srv_cn_utf8);
2692 ads_destroy(&ads);
2693 SAFE_FREE(srv_dn);
2694 return WERR_SERVER_UNAVAILABLE;
2697 ldap_memfree(srv_dn_utf8);
2698 ldap_memfree(srv_cn_utf8);
2700 asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn_0,
2701 printer->info_2->sharename, srv_dn);
2703 SAFE_FREE(srv_dn);
2704 SAFE_FREE(srv_cn_0);
2706 /* build the ads mods */
2707 ctx = talloc_init("nt_printer_publish_ads");
2708 mods = ads_init_mods(ctx);
2710 get_local_printer_publishing_data(ctx, &mods,
2711 &printer->info_2->data);
2712 ads_mod_str(ctx, &mods, SPOOL_REG_PRINTERNAME,
2713 printer->info_2->sharename);
2715 /* publish it */
2716 ads_rc = ads_mod_printer_entry(ads, prt_dn, ctx, &mods);
2717 if (ads_rc.err.rc == LDAP_NO_SUCH_OBJECT)
2718 ads_rc = ads_add_printer_entry(ads, prt_dn, ctx, &mods);
2720 if (!ADS_ERR_OK(ads_rc))
2721 DEBUG(3, ("error publishing %s: %s\n", printer->info_2->sharename, ads_errstr(ads_rc)));
2723 talloc_destroy(ctx);
2725 /* retreive the guid and store it locally */
2726 if (ADS_ERR_OK(ads_search_dn(ads, &res, prt_dn, attrs))) {
2727 ZERO_STRUCT(guid);
2728 ads_pull_guid(ads, res, &guid);
2729 ads_msgfree(ads, res);
2730 store_printer_guid(printer->info_2, guid);
2731 win_rc = mod_a_printer(printer, 2);
2734 SAFE_FREE(prt_dn);
2735 return win_rc;
2738 static WERROR nt_printer_unpublish_ads(ADS_STRUCT *ads,
2739 NT_PRINTER_INFO_LEVEL *printer)
2741 ADS_STATUS ads_rc;
2742 void *res;
2743 char *prt_dn = NULL;
2745 DEBUG(5, ("unpublishing printer %s\n", printer->info_2->printername));
2747 /* remove the printer from the directory */
2748 ads_rc = ads_find_printer_on_server(ads, &res,
2749 printer->info_2->sharename, global_myname());
2751 if (ADS_ERR_OK(ads_rc) && ads_count_replies(ads, res)) {
2752 prt_dn = ads_get_dn(ads, res);
2753 ads_rc = ads_del_dn(ads, prt_dn);
2754 ads_memfree(ads, prt_dn);
2757 ads_msgfree(ads, res);
2758 return WERR_OK;
2761 /****************************************************************************
2762 * Publish a printer in the directory
2764 * @param snum describing printer service
2765 * @return WERROR indicating status of publishing
2766 ***************************************************************************/
2768 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
2770 ADS_STATUS ads_rc;
2771 ADS_STRUCT *ads = NULL;
2772 NT_PRINTER_INFO_LEVEL *printer = NULL;
2773 WERROR win_rc;
2775 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
2776 if (!W_ERROR_IS_OK(win_rc))
2777 goto done;
2779 switch (action) {
2780 case SPOOL_DS_PUBLISH:
2781 case SPOOL_DS_UPDATE:
2782 /* set the DsSpooler info and attributes */
2783 if (!(map_nt_printer_info2_to_dsspooler(printer->info_2))) {
2784 win_rc = WERR_NOMEM;
2785 goto done;
2788 printer->info_2->attributes |= PRINTER_ATTRIBUTE_PUBLISHED;
2789 break;
2790 case SPOOL_DS_UNPUBLISH:
2791 printer->info_2->attributes ^= PRINTER_ATTRIBUTE_PUBLISHED;
2792 break;
2793 default:
2794 win_rc = WERR_NOT_SUPPORTED;
2795 goto done;
2798 win_rc = mod_a_printer(printer, 2);
2799 if (!W_ERROR_IS_OK(win_rc)) {
2800 DEBUG(3, ("err %d saving data\n", W_ERROR_V(win_rc)));
2801 goto done;
2804 ads = ads_init(NULL, NULL, NULL);
2805 if (!ads) {
2806 DEBUG(3, ("ads_init() failed\n"));
2807 win_rc = WERR_SERVER_UNAVAILABLE;
2808 goto done;
2810 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
2811 SAFE_FREE(ads->auth.password);
2812 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
2813 NULL, NULL);
2815 /* ads_connect() will find the DC for us */
2816 ads_rc = ads_connect(ads);
2817 if (!ADS_ERR_OK(ads_rc)) {
2818 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
2819 win_rc = WERR_ACCESS_DENIED;
2820 goto done;
2823 switch (action) {
2824 case SPOOL_DS_PUBLISH:
2825 case SPOOL_DS_UPDATE:
2826 win_rc = nt_printer_publish_ads(ads, printer);
2827 break;
2828 case SPOOL_DS_UNPUBLISH:
2829 win_rc = nt_printer_unpublish_ads(ads, printer);
2830 break;
2833 done:
2834 free_a_printer(&printer, 2);
2835 ads_destroy(&ads);
2836 return win_rc;
2839 WERROR check_published_printers(void)
2841 ADS_STATUS ads_rc;
2842 ADS_STRUCT *ads = NULL;
2843 int snum;
2844 int n_services = lp_numservices();
2845 NT_PRINTER_INFO_LEVEL *printer = NULL;
2847 ads = ads_init(NULL, NULL, NULL);
2848 if (!ads) {
2849 DEBUG(3, ("ads_init() failed\n"));
2850 return WERR_SERVER_UNAVAILABLE;
2852 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
2853 SAFE_FREE(ads->auth.password);
2854 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
2855 NULL, NULL);
2857 /* ads_connect() will find the DC for us */
2858 ads_rc = ads_connect(ads);
2859 if (!ADS_ERR_OK(ads_rc)) {
2860 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
2861 ads_destroy(&ads);
2862 return WERR_ACCESS_DENIED;
2865 for (snum = 0; snum < n_services; snum++) {
2866 if (!(lp_snum_ok(snum) && lp_print_ok(snum)))
2867 continue;
2869 if (W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2,
2870 lp_servicename(snum))) &&
2871 (printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED))
2872 nt_printer_publish_ads(ads, printer);
2874 free_a_printer(&printer, 2);
2877 ads_destroy(&ads);
2878 return WERR_OK;
2881 BOOL is_printer_published(Printer_entry *print_hnd, int snum,
2882 struct uuid *guid)
2884 NT_PRINTER_INFO_LEVEL *printer = NULL;
2885 REGVAL_CTR *ctr;
2886 REGISTRY_VALUE *guid_val;
2887 WERROR win_rc;
2888 int i;
2890 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
2892 if (!W_ERROR_IS_OK(win_rc) ||
2893 !(printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED) ||
2894 ((i = lookup_printerkey(&printer->info_2->data,
2895 SPOOL_DSSPOOLER_KEY)) < 0) ||
2896 !(ctr = &printer->info_2->data.keys[i].values) ||
2897 !(guid_val = regval_ctr_getvalue(ctr, "objectGUID"))) {
2898 free_a_printer(&printer, 2);
2899 return False;
2902 if (regval_size(guid_val) == sizeof(struct uuid))
2903 memcpy(guid, regval_data_p(guid_val), sizeof(struct uuid));
2905 free_a_printer(&printer, 2);
2906 return True;
2908 #else
2909 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
2911 return WERR_OK;
2914 WERROR check_published_printers(void)
2916 return WERR_OK;
2919 BOOL is_printer_published(Printer_entry *print_hnd, int snum,
2920 struct uuid *guid)
2922 return False;
2924 #endif /* HAVE_ADS */
2926 /****************************************************************************
2927 ***************************************************************************/
2929 WERROR delete_all_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key )
2931 NT_PRINTER_DATA *data;
2932 int i;
2933 int removed_keys = 0;
2934 int empty_slot;
2936 data = &p2->data;
2937 empty_slot = data->num_keys;
2939 if ( !key )
2940 return WERR_INVALID_PARAM;
2942 /* remove all keys */
2944 if ( !strlen(key) ) {
2945 for ( i=0; i<data->num_keys; i++ ) {
2946 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
2947 data->keys[i].name));
2949 SAFE_FREE( data->keys[i].name );
2950 regval_ctr_destroy( &data->keys[i].values );
2953 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from printer [%s]\n",
2954 p2->printername ));
2956 SAFE_FREE( data->keys );
2957 ZERO_STRUCTP( data );
2959 return WERR_OK;
2962 /* remove a specific key (and all subkeys) */
2964 for ( i=0; i<data->num_keys; i++ ) {
2965 if ( StrnCaseCmp( data->keys[i].name, key, strlen(key)) == 0 ) {
2966 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
2967 data->keys[i].name));
2969 SAFE_FREE( data->keys[i].name );
2970 regval_ctr_destroy( &data->keys[i].values );
2972 /* mark the slot as empty */
2974 ZERO_STRUCTP( &data->keys[i] );
2978 /* find the first empty slot */
2980 for ( i=0; i<data->num_keys; i++ ) {
2981 if ( !data->keys[i].name ) {
2982 empty_slot = i;
2983 removed_keys++;
2984 break;
2988 if ( i == data->num_keys )
2989 /* nothing was removed */
2990 return WERR_INVALID_PARAM;
2992 /* move everything down */
2994 for ( i=empty_slot+1; i<data->num_keys; i++ ) {
2995 if ( data->keys[i].name ) {
2996 memcpy( &data->keys[empty_slot], &data->keys[i], sizeof(NT_PRINTER_KEY) );
2997 ZERO_STRUCTP( &data->keys[i] );
2998 empty_slot++;
2999 removed_keys++;
3003 /* update count */
3005 data->num_keys -= removed_keys;
3007 /* sanity check to see if anything is left */
3009 if ( !data->num_keys ) {
3010 DEBUG(8,("delete_all_printer_data: No keys left for printer [%s]\n", p2->printername ));
3012 SAFE_FREE( data->keys );
3013 ZERO_STRUCTP( data );
3016 return WERR_OK;
3019 /****************************************************************************
3020 ***************************************************************************/
3022 WERROR delete_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3024 WERROR result = WERR_OK;
3025 int key_index;
3027 /* we must have names on non-zero length */
3029 if ( !key || !*key|| !value || !*value )
3030 return WERR_INVALID_NAME;
3032 /* find the printer key first */
3034 key_index = lookup_printerkey( &p2->data, key );
3035 if ( key_index == -1 )
3036 return WERR_OK;
3038 /* make sure the value exists so we can return the correct error code */
3040 if ( !regval_ctr_getvalue( &p2->data.keys[key_index].values, value ) )
3041 return WERR_BADFILE;
3043 regval_ctr_delvalue( &p2->data.keys[key_index].values, value );
3045 DEBUG(8,("delete_printer_data: Removed key => [%s], value => [%s]\n",
3046 key, value ));
3048 return result;
3051 /****************************************************************************
3052 ***************************************************************************/
3054 WERROR add_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value,
3055 uint32 type, uint8 *data, int real_len )
3057 WERROR result = WERR_OK;
3058 int key_index;
3060 /* we must have names on non-zero length */
3062 if ( !key || !*key|| !value || !*value )
3063 return WERR_INVALID_NAME;
3065 /* find the printer key first */
3067 key_index = lookup_printerkey( &p2->data, key );
3068 if ( key_index == -1 )
3069 key_index = add_new_printer_key( &p2->data, key );
3071 if ( key_index == -1 )
3072 return WERR_NOMEM;
3074 regval_ctr_addvalue( &p2->data.keys[key_index].values, value,
3075 type, (const char *)data, real_len );
3077 DEBUG(8,("add_printer_data: Added key => [%s], value => [%s], type=> [%d], size => [%d]\n",
3078 key, value, type, real_len ));
3080 return result;
3083 /****************************************************************************
3084 ***************************************************************************/
3086 REGISTRY_VALUE* get_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3088 int key_index;
3090 if ( (key_index = lookup_printerkey( &p2->data, key )) == -1 )
3091 return NULL;
3093 DEBUG(8,("get_printer_data: Attempting to lookup key => [%s], value => [%s]\n",
3094 key, value ));
3096 return regval_ctr_getvalue( &p2->data.keys[key_index].values, value );
3099 /****************************************************************************
3100 Unpack a list of registry values frem the TDB
3101 ***************************************************************************/
3103 static int unpack_values(NT_PRINTER_DATA *printer_data, char *buf, int buflen)
3105 int len = 0;
3106 uint32 type;
3107 pstring string, valuename, keyname;
3108 char *str;
3109 int size;
3110 uint8 *data_p;
3111 REGISTRY_VALUE *regval_p;
3112 int key_index;
3114 /* add the "PrinterDriverData" key first for performance reasons */
3116 add_new_printer_key( printer_data, SPOOL_PRINTERDATA_KEY );
3118 /* loop and unpack the rest of the registry values */
3120 while ( True ) {
3122 /* check to see if there are any more registry values */
3124 regval_p = NULL;
3125 len += tdb_unpack(buf+len, buflen-len, "p", &regval_p);
3126 if ( !regval_p )
3127 break;
3129 /* unpack the next regval */
3131 len += tdb_unpack(buf+len, buflen-len, "fdB",
3132 string,
3133 &type,
3134 &size,
3135 &data_p);
3138 * break of the keyname from the value name.
3139 * Valuenames can have embedded '\'s so be careful.
3140 * only support one level of keys. See the
3141 * "Konica Fiery S300 50C-K v1.1. enu" 2k driver.
3142 * -- jerry
3145 str = strchr_m( string, '\\');
3147 /* Put in "PrinterDriverData" is no key specified */
3149 if ( !str ) {
3150 pstrcpy( keyname, SPOOL_PRINTERDATA_KEY );
3151 pstrcpy( valuename, string );
3153 else {
3154 *str = '\0';
3155 pstrcpy( keyname, string );
3156 pstrcpy( valuename, str+1 );
3159 /* see if we need a new key */
3161 if ( (key_index=lookup_printerkey( printer_data, keyname )) == -1 )
3162 key_index = add_new_printer_key( printer_data, keyname );
3164 if ( key_index == -1 ) {
3165 DEBUG(0,("unpack_values: Failed to allocate a new key [%s]!\n",
3166 keyname));
3167 break;
3170 /* add the new value */
3172 regval_ctr_addvalue( &printer_data->keys[key_index].values, valuename, type, (const char *)data_p, size );
3174 SAFE_FREE(data_p); /* 'B' option to tdbpack does a malloc() */
3176 DEBUG(8,("specific: [%s:%s], len: %d\n", keyname, valuename, size));
3179 return len;
3182 /****************************************************************************
3183 ***************************************************************************/
3185 static void map_to_os2_driver(fstring drivername)
3187 static BOOL initialised=False;
3188 static fstring last_from,last_to;
3189 char *mapfile = lp_os2_driver_map();
3190 char **lines = NULL;
3191 int numlines = 0;
3192 int i;
3194 if (!strlen(drivername))
3195 return;
3197 if (!*mapfile)
3198 return;
3200 if (!initialised) {
3201 *last_from = *last_to = 0;
3202 initialised = True;
3205 if (strequal(drivername,last_from)) {
3206 DEBUG(3,("Mapped Windows driver %s to OS/2 driver %s\n",drivername,last_to));
3207 fstrcpy(drivername,last_to);
3208 return;
3211 lines = file_lines_load(mapfile, &numlines);
3212 if (numlines == 0) {
3213 DEBUG(0,("No entries in OS/2 driver map %s\n",mapfile));
3214 return;
3217 DEBUG(4,("Scanning OS/2 driver map %s\n",mapfile));
3219 for( i = 0; i < numlines; i++) {
3220 char *nt_name = lines[i];
3221 char *os2_name = strchr(nt_name,'=');
3223 if (!os2_name)
3224 continue;
3226 *os2_name++ = 0;
3228 while (isspace(*nt_name))
3229 nt_name++;
3231 if (!*nt_name || strchr("#;",*nt_name))
3232 continue;
3235 int l = strlen(nt_name);
3236 while (l && isspace(nt_name[l-1])) {
3237 nt_name[l-1] = 0;
3238 l--;
3242 while (isspace(*os2_name))
3243 os2_name++;
3246 int l = strlen(os2_name);
3247 while (l && isspace(os2_name[l-1])) {
3248 os2_name[l-1] = 0;
3249 l--;
3253 if (strequal(nt_name,drivername)) {
3254 DEBUG(3,("Mapped windows driver %s to os2 driver%s\n",drivername,os2_name));
3255 fstrcpy(last_from,drivername);
3256 fstrcpy(last_to,os2_name);
3257 fstrcpy(drivername,os2_name);
3258 file_lines_free(lines);
3259 return;
3263 file_lines_free(lines);
3266 /****************************************************************************
3267 Get a default printer info 2 struct.
3268 ****************************************************************************/
3269 static WERROR get_a_printer_2_default(NT_PRINTER_INFO_LEVEL_2 **info_ptr, const char *servername, const char* sharename)
3271 int snum;
3272 NT_PRINTER_INFO_LEVEL_2 info;
3274 ZERO_STRUCT(info);
3276 snum = lp_servicenumber(sharename);
3278 slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", servername);
3279 slprintf(info.printername, sizeof(info.printername)-1, "\\\\%s\\%s",
3280 servername, sharename);
3281 fstrcpy(info.sharename, sharename);
3282 fstrcpy(info.portname, SAMBA_PRINTER_PORT_NAME);
3284 /* by setting the driver name to an empty string, a local NT admin
3285 can now run the **local** APW to install a local printer driver
3286 for a Samba shared printer in 2.2. Without this, drivers **must** be
3287 installed on the Samba server for NT clients --jerry */
3288 #if 0 /* JERRY --do not uncomment-- */
3289 if (!*info.drivername)
3290 fstrcpy(info.drivername, "NO DRIVER AVAILABLE FOR THIS PRINTER");
3291 #endif
3294 DEBUG(10,("get_a_printer_2_default: driver name set to [%s]\n", info.drivername));
3296 pstrcpy(info.comment, "");
3297 fstrcpy(info.printprocessor, "winprint");
3298 fstrcpy(info.datatype, "RAW");
3300 info.attributes = PRINTER_ATTRIBUTE_SAMBA;
3302 info.starttime = 0; /* Minutes since 12:00am GMT */
3303 info.untiltime = 0; /* Minutes since 12:00am GMT */
3304 info.priority = 1;
3305 info.default_priority = 1;
3306 info.setuptime = (uint32)time(NULL);
3309 * I changed this as I think it is better to have a generic
3310 * DEVMODE than to crash Win2k explorer.exe --jerry
3311 * See the HP Deskjet 990c Win2k drivers for an example.
3313 * However the default devmode appears to cause problems
3314 * with the HP CLJ 8500 PCL driver. Hence the addition of
3315 * the "default devmode" parameter --jerry 22/01/2002
3318 if (lp_default_devmode(snum)) {
3319 if ((info.devmode = construct_nt_devicemode(info.printername)) == NULL)
3320 goto fail;
3322 else {
3323 info.devmode = NULL;
3326 /* This will get the current RPC talloc context, but we should be
3327 passing this as a parameter... fixme... JRA ! */
3329 if (!nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf))
3330 goto fail;
3332 *info_ptr = (NT_PRINTER_INFO_LEVEL_2 *)memdup(&info, sizeof(info));
3333 if (! *info_ptr) {
3334 DEBUG(0,("get_a_printer_2_default: malloc fail.\n"));
3335 goto fail;
3338 return WERR_OK;
3340 fail:
3341 if (info.devmode)
3342 free_nt_devicemode(&info.devmode);
3343 return WERR_ACCESS_DENIED;
3346 /****************************************************************************
3347 ****************************************************************************/
3348 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr, const char *servername, const char *sharename)
3350 pstring key;
3351 NT_PRINTER_INFO_LEVEL_2 info;
3352 int len = 0;
3353 int snum = lp_servicenumber(sharename);
3354 TDB_DATA kbuf, dbuf;
3355 fstring printername;
3356 char adevice[MAXDEVICENAME];
3358 ZERO_STRUCT(info);
3360 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
3362 kbuf.dptr = key;
3363 kbuf.dsize = strlen(key)+1;
3365 dbuf = tdb_fetch(tdb_printers, kbuf);
3366 if (!dbuf.dptr)
3367 return get_a_printer_2_default(info_ptr, servername, sharename);
3369 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "dddddddddddfffffPfffff",
3370 &info.attributes,
3371 &info.priority,
3372 &info.default_priority,
3373 &info.starttime,
3374 &info.untiltime,
3375 &info.status,
3376 &info.cjobs,
3377 &info.averageppm,
3378 &info.changeid,
3379 &info.c_setprinter,
3380 &info.setuptime,
3381 info.servername,
3382 info.printername,
3383 info.sharename,
3384 info.portname,
3385 info.drivername,
3386 info.comment,
3387 info.location,
3388 info.sepfile,
3389 info.printprocessor,
3390 info.datatype,
3391 info.parameters);
3393 /* Samba has to have shared raw drivers. */
3394 info.attributes |= PRINTER_ATTRIBUTE_SAMBA;
3395 info.attributes &= ~PRINTER_ATTRIBUTE_NOT_SAMBA;
3397 /* Restore the stripped strings. */
3398 slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", servername);
3400 if ( lp_force_printername(snum) )
3401 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, sharename );
3402 else
3403 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, info.printername);
3405 fstrcpy(info.printername, printername);
3407 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
3410 * Some client drivers freak out if there is a NULL devmode
3411 * (probably the driver is not checking before accessing
3412 * the devmode pointer) --jerry
3414 * See comments in get_a_printer_2_default()
3417 if (lp_default_devmode(snum) && !info.devmode) {
3418 DEBUG(8,("get_a_printer_2: Constructing a default device mode for [%s]\n",
3419 printername));
3420 info.devmode = construct_nt_devicemode(printername);
3423 slprintf( adevice, sizeof(adevice), "%s", info.printername );
3424 if (info.devmode) {
3425 fstrcpy(info.devmode->devicename, adevice);
3428 len += unpack_values( &info.data, dbuf.dptr+len, dbuf.dsize-len );
3430 /* This will get the current RPC talloc context, but we should be
3431 passing this as a parameter... fixme... JRA ! */
3433 nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf);
3435 /* Fix for OS/2 drivers. */
3437 if (get_remote_arch() == RA_OS2)
3438 map_to_os2_driver(info.drivername);
3440 SAFE_FREE(dbuf.dptr);
3441 *info_ptr=memdup(&info, sizeof(info));
3443 DEBUG(9,("Unpacked printer [%s] name [%s] running driver [%s]\n",
3444 sharename, info.printername, info.drivername));
3446 return WERR_OK;
3449 /****************************************************************************
3450 Debugging function, dump at level 6 the struct in the logs.
3451 ****************************************************************************/
3452 static uint32 dump_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
3454 uint32 result;
3455 NT_PRINTER_INFO_LEVEL_2 *info2;
3457 DEBUG(106,("Dumping printer at level [%d]\n", level));
3459 switch (level) {
3460 case 2:
3462 if (printer->info_2 == NULL)
3463 result=5;
3464 else
3466 info2=printer->info_2;
3468 DEBUGADD(106,("attributes:[%d]\n", info2->attributes));
3469 DEBUGADD(106,("priority:[%d]\n", info2->priority));
3470 DEBUGADD(106,("default_priority:[%d]\n", info2->default_priority));
3471 DEBUGADD(106,("starttime:[%d]\n", info2->starttime));
3472 DEBUGADD(106,("untiltime:[%d]\n", info2->untiltime));
3473 DEBUGADD(106,("status:[%d]\n", info2->status));
3474 DEBUGADD(106,("cjobs:[%d]\n", info2->cjobs));
3475 DEBUGADD(106,("averageppm:[%d]\n", info2->averageppm));
3476 DEBUGADD(106,("changeid:[%d]\n", info2->changeid));
3477 DEBUGADD(106,("c_setprinter:[%d]\n", info2->c_setprinter));
3478 DEBUGADD(106,("setuptime:[%d]\n", info2->setuptime));
3480 DEBUGADD(106,("servername:[%s]\n", info2->servername));
3481 DEBUGADD(106,("printername:[%s]\n", info2->printername));
3482 DEBUGADD(106,("sharename:[%s]\n", info2->sharename));
3483 DEBUGADD(106,("portname:[%s]\n", info2->portname));
3484 DEBUGADD(106,("drivername:[%s]\n", info2->drivername));
3485 DEBUGADD(106,("comment:[%s]\n", info2->comment));
3486 DEBUGADD(106,("location:[%s]\n", info2->location));
3487 DEBUGADD(106,("sepfile:[%s]\n", info2->sepfile));
3488 DEBUGADD(106,("printprocessor:[%s]\n", info2->printprocessor));
3489 DEBUGADD(106,("datatype:[%s]\n", info2->datatype));
3490 DEBUGADD(106,("parameters:[%s]\n", info2->parameters));
3491 result=0;
3493 break;
3495 default:
3496 DEBUGADD(106,("dump_a_printer: Level %u not implemented\n", (unsigned int)level ));
3497 result=1;
3498 break;
3501 return result;
3504 /****************************************************************************
3505 Update the changeid time.
3506 This is SO NASTY as some drivers need this to change, others need it
3507 static. This value will change every second, and I must hope that this
3508 is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
3509 UTAH ! JRA.
3510 ****************************************************************************/
3512 static uint32 rev_changeid(void)
3514 struct timeval tv;
3516 get_process_uptime(&tv);
3518 #if 1 /* JERRY */
3519 /* Return changeid as msec since spooler restart */
3520 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
3521 #else
3523 * This setting seems to work well but is too untested
3524 * to replace the above calculation. Left in for experiementation
3525 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
3527 return tv.tv_sec * 10 + tv.tv_usec / 100000;
3528 #endif
3531 /********************************************************************
3532 Send a message to all smbds about the printer that just changed
3533 ********************************************************************/
3535 static BOOL send_printer_mod_msg( char* printername )
3537 int len = strlen(printername);
3539 if (!len)
3540 return False;
3542 DEBUG(10,("send_printer_mod_msg: Sending message about printer change [%s]\n",
3543 printername));
3545 /* spam everyone that we just changed this printer */
3547 message_send_all( conn_tdb_ctx(), MSG_PRINTER_MOD, printername, len+1, False, NULL );
3549 return True;
3553 * The function below are the high level ones.
3554 * only those ones must be called from the spoolss code.
3555 * JFM.
3558 /****************************************************************************
3559 Modify a printer. This is called from SETPRINTERDATA/DELETEPRINTERDATA.
3560 ****************************************************************************/
3562 WERROR mod_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
3564 WERROR result;
3566 dump_a_printer(printer, level);
3569 * invalidate cache for all open handles to this printer.
3570 * cache for a given handle will be updated on the next
3571 * get_a_printer()
3574 invalidate_printer_hnd_cache( printer->info_2->sharename );
3575 send_printer_mod_msg( printer->info_2->sharename );
3577 switch (level) {
3578 case 2:
3581 * Update the changestamp. Emperical tests show that the
3582 * ChangeID is always updated,but c_setprinter is
3583 * global spooler variable (not per printer).
3586 /* ChangeID **must** be increasing over the lifetime
3587 of client's spoolss service in order for the
3588 client's cache to show updates */
3590 printer->info_2->changeid = rev_changeid();
3593 * Because one day someone will ask:
3594 * NT->NT An admin connection to a remote
3595 * printer show changes imeediately in
3596 * the properities dialog
3598 * A non-admin connection will only show the
3599 * changes after viewing the properites page
3600 * 2 times. Seems to be related to a
3601 * race condition in the client between the spooler
3602 * updating the local cache and the Explorer.exe GUI
3603 * actually displaying the properties.
3605 * This is fixed in Win2k. admin/non-admin
3606 * connections both display changes immediately.
3608 * 14/12/01 --jerry
3611 result=update_a_printer_2(printer->info_2);
3613 break;
3615 default:
3616 result=WERR_UNKNOWN_LEVEL;
3617 break;
3620 return result;
3623 /****************************************************************************
3624 Initialize printer devmode & data with previously saved driver init values.
3625 ****************************************************************************/
3627 static BOOL set_driver_init_2( NT_PRINTER_INFO_LEVEL_2 *info_ptr )
3629 int len = 0;
3630 pstring key;
3631 TDB_DATA kbuf, dbuf;
3632 NT_PRINTER_INFO_LEVEL_2 info;
3635 ZERO_STRUCT(info);
3638 * Delete any printer data 'values' already set. When called for driver
3639 * replace, there will generally be some, but during an add printer, there
3640 * should not be any (if there are delete them).
3643 delete_all_printer_data( info_ptr, "" );
3645 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info_ptr->drivername);
3647 kbuf.dptr = key;
3648 kbuf.dsize = strlen(key)+1;
3650 dbuf = tdb_fetch(tdb_drivers, kbuf);
3651 if (!dbuf.dptr) {
3653 * When changing to a driver that has no init info in the tdb, remove
3654 * the previous drivers init info and leave the new on blank.
3656 free_nt_devicemode(&info_ptr->devmode);
3657 return False;
3661 * Get the saved DEVMODE..
3664 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
3667 * The saved DEVMODE contains the devicename from the printer used during
3668 * the initialization save. Change it to reflect the new printer.
3671 if ( info.devmode ) {
3672 ZERO_STRUCT(info.devmode->devicename);
3673 fstrcpy(info.devmode->devicename, info_ptr->printername);
3677 * NT/2k does not change out the entire DeviceMode of a printer
3678 * when changing the driver. Only the driverextra, private, &
3679 * driverversion fields. --jerry (Thu Mar 14 08:58:43 CST 2002)
3681 * Later examination revealed that Windows NT/2k does reset the
3682 * the printer's device mode, bit **only** when you change a
3683 * property of the device mode such as the page orientation.
3684 * --jerry
3688 /* Bind the saved DEVMODE to the new the printer */
3690 free_nt_devicemode(&info_ptr->devmode);
3691 info_ptr->devmode = info.devmode;
3693 DEBUG(10,("set_driver_init_2: Set printer [%s] init %s DEVMODE for driver [%s]\n",
3694 info_ptr->printername, info_ptr->devmode?"VALID":"NULL", info_ptr->drivername));
3696 /* Add the printer data 'values' to the new printer */
3698 len += unpack_values( &info_ptr->data, dbuf.dptr+len, dbuf.dsize-len );
3701 SAFE_FREE(dbuf.dptr);
3703 return True;
3706 /****************************************************************************
3707 Initialize printer devmode & data with previously saved driver init values.
3708 When a printer is created using AddPrinter, the drivername bound to the
3709 printer is used to lookup previously saved driver initialization info, which
3710 is bound to the new printer.
3711 ****************************************************************************/
3713 BOOL set_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
3715 BOOL result = False;
3717 switch (level) {
3718 case 2:
3719 result = set_driver_init_2(printer->info_2);
3720 break;
3722 default:
3723 DEBUG(0,("set_driver_init: Programmer's error! Unknown driver_init level [%d]\n",
3724 level));
3725 break;
3728 return result;
3731 /****************************************************************************
3732 Delete driver init data stored for a specified driver
3733 ****************************************************************************/
3735 BOOL del_driver_init(char *drivername)
3737 pstring key;
3738 TDB_DATA kbuf;
3740 if (!drivername || !*drivername) {
3741 DEBUG(3,("del_driver_init: No drivername specified!\n"));
3742 return False;
3745 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, drivername);
3747 kbuf.dptr = key;
3748 kbuf.dsize = strlen(key)+1;
3750 DEBUG(6,("del_driver_init: Removing driver init data for [%s]\n", drivername));
3752 return (tdb_delete(tdb_drivers, kbuf) == 0);
3755 /****************************************************************************
3756 Pack up the DEVMODE and values for a printer into a 'driver init' entry
3757 in the tdb. Note: this is different from the driver entry and the printer
3758 entry. There should be a single driver init entry for each driver regardless
3759 of whether it was installed from NT or 2K. Technically, they should be
3760 different, but they work out to the same struct.
3761 ****************************************************************************/
3763 static uint32 update_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info)
3765 pstring key;
3766 char *buf;
3767 int buflen, len, ret;
3768 TDB_DATA kbuf, dbuf;
3770 buf = NULL;
3771 buflen = 0;
3773 again:
3774 len = 0;
3775 len += pack_devicemode(info->devmode, buf+len, buflen-len);
3777 len += pack_values( &info->data, buf+len, buflen-len );
3779 if (buflen < len) {
3780 char *tb;
3782 tb = (char *)SMB_REALLOC(buf, len);
3783 if (!tb) {
3784 DEBUG(0, ("update_driver_init_2: failed to enlarge buffer!\n"));
3785 ret = -1;
3786 goto done;
3788 else
3789 buf = tb;
3790 buflen = len;
3791 goto again;
3794 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info->drivername);
3796 kbuf.dptr = key;
3797 kbuf.dsize = strlen(key)+1;
3798 dbuf.dptr = buf;
3799 dbuf.dsize = len;
3801 ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
3803 done:
3804 if (ret == -1)
3805 DEBUG(8, ("update_driver_init_2: error updating printer init to tdb on disk\n"));
3807 SAFE_FREE(buf);
3809 DEBUG(10,("update_driver_init_2: Saved printer [%s] init DEVMODE & values for driver [%s]\n",
3810 info->sharename, info->drivername));
3812 return ret;
3815 /****************************************************************************
3816 Update (i.e. save) the driver init info (DEVMODE and values) for a printer
3817 ****************************************************************************/
3819 static uint32 update_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
3821 uint32 result;
3823 dump_a_printer(printer, level);
3825 switch (level) {
3826 case 2:
3827 result = update_driver_init_2(printer->info_2);
3828 break;
3829 default:
3830 result = 1;
3831 break;
3834 return result;
3837 /****************************************************************************
3838 Convert the printer data value, a REG_BINARY array, into an initialization
3839 DEVMODE. Note: the array must be parsed as if it was a DEVMODE in an rpc...
3840 got to keep the endians happy :).
3841 ****************************************************************************/
3843 static BOOL convert_driver_init( TALLOC_CTX *ctx, NT_DEVICEMODE *nt_devmode, uint8 *data, uint32 data_len )
3845 BOOL result = False;
3846 prs_struct ps;
3847 DEVICEMODE devmode;
3849 ZERO_STRUCT(devmode);
3851 prs_init(&ps, 0, ctx, UNMARSHALL);
3852 ps.data_p = (char *)data;
3853 ps.buffer_size = data_len;
3855 if (spoolss_io_devmode("phantom DEVMODE", &ps, 0, &devmode))
3856 result = convert_devicemode("", &devmode, &nt_devmode);
3857 else
3858 DEBUG(10,("convert_driver_init: error parsing DEVMODE\n"));
3860 return result;
3863 /****************************************************************************
3864 Set the DRIVER_INIT info in the tdb. Requires Win32 client code that:
3866 1. Use the driver's config DLL to this UNC printername and:
3867 a. Call DrvPrintEvent with PRINTER_EVENT_INITIALIZE
3868 b. Call DrvConvertDevMode with CDM_DRIVER_DEFAULT to get default DEVMODE
3869 2. Call SetPrinterData with the 'magic' key and the DEVMODE as data.
3871 The last step triggers saving the "driver initialization" information for
3872 this printer into the tdb. Later, new printers that use this driver will
3873 have this initialization information bound to them. This simulates the
3874 driver initialization, as if it had run on the Samba server (as it would
3875 have done on NT).
3877 The Win32 client side code requirement sucks! But until we can run arbitrary
3878 Win32 printer driver code on any Unix that Samba runs on, we are stuck with it.
3880 It would have been easier to use SetPrinter because all the UNMARSHALLING of
3881 the DEVMODE is done there, but 2K/XP clients do not set the DEVMODE... think
3882 about it and you will realize why. JRR 010720
3883 ****************************************************************************/
3885 static WERROR save_driver_init_2(NT_PRINTER_INFO_LEVEL *printer, uint8 *data, uint32 data_len )
3887 WERROR status = WERR_OK;
3888 TALLOC_CTX *ctx = NULL;
3889 NT_DEVICEMODE *nt_devmode = NULL;
3890 NT_DEVICEMODE *tmp_devmode = printer->info_2->devmode;
3893 * When the DEVMODE is already set on the printer, don't try to unpack it.
3895 DEBUG(8,("save_driver_init_2: Enter...\n"));
3897 if ( !printer->info_2->devmode && data_len ) {
3899 * Set devmode on printer info, so entire printer initialization can be
3900 * saved to tdb.
3903 if ((ctx = talloc_init("save_driver_init_2")) == NULL)
3904 return WERR_NOMEM;
3906 if ((nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE)) == NULL) {
3907 status = WERR_NOMEM;
3908 goto done;
3911 ZERO_STRUCTP(nt_devmode);
3914 * The DEVMODE is held in the 'data' component of the param in raw binary.
3915 * Convert it to to a devmode structure
3917 if ( !convert_driver_init( ctx, nt_devmode, data, data_len )) {
3918 DEBUG(10,("save_driver_init_2: error converting DEVMODE\n"));
3919 status = WERR_INVALID_PARAM;
3920 goto done;
3923 printer->info_2->devmode = nt_devmode;
3927 * Pack up and add (or update) the DEVMODE and any current printer data to
3928 * a 'driver init' element in the tdb
3932 if ( update_driver_init(printer, 2) != 0 ) {
3933 DEBUG(10,("save_driver_init_2: error updating DEVMODE\n"));
3934 status = WERR_NOMEM;
3935 goto done;
3939 * If driver initialization info was successfully saved, set the current
3940 * printer to match it. This allows initialization of the current printer
3941 * as well as the driver.
3943 status = mod_a_printer(printer, 2);
3944 if (!W_ERROR_IS_OK(status)) {
3945 DEBUG(10,("save_driver_init_2: error setting DEVMODE on printer [%s]\n",
3946 printer->info_2->printername));
3949 done:
3950 talloc_destroy(ctx);
3951 free_nt_devicemode( &nt_devmode );
3953 printer->info_2->devmode = tmp_devmode;
3955 return status;
3958 /****************************************************************************
3959 Update the driver init info (DEVMODE and specifics) for a printer
3960 ****************************************************************************/
3962 WERROR save_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level, uint8 *data, uint32 data_len)
3964 WERROR status = WERR_OK;
3966 switch (level) {
3967 case 2:
3968 status = save_driver_init_2( printer, data, data_len );
3969 break;
3970 default:
3971 status = WERR_UNKNOWN_LEVEL;
3972 break;
3975 return status;
3978 /****************************************************************************
3979 Deep copy a NT_PRINTER_DATA
3980 ****************************************************************************/
3982 static NTSTATUS copy_printer_data( NT_PRINTER_DATA *dst, NT_PRINTER_DATA *src )
3984 int i, j, num_vals, new_key_index;
3985 REGVAL_CTR *src_key, *dst_key;
3987 if ( !dst || !src )
3988 return NT_STATUS_NO_MEMORY;
3990 for ( i=0; i<src->num_keys; i++ ) {
3992 /* create a new instance of the printerkey in the destination
3993 printer_data object */
3995 new_key_index = add_new_printer_key( dst, src->keys[i].name );
3996 dst_key = &dst->keys[new_key_index].values;
3998 src_key = &src->keys[i].values;
3999 num_vals = regval_ctr_numvals( src_key );
4001 /* dup the printer entire printer key */
4003 for ( j=0; j<num_vals; j++ ) {
4004 regval_ctr_copyvalue( dst_key, regval_ctr_specific_value(src_key, j) );
4008 return NT_STATUS_OK;
4011 /****************************************************************************
4012 Deep copy a NT_PRINTER_INFO_LEVEL_2 structure using malloc()'d memeory
4013 Caller must free.
4014 ****************************************************************************/
4016 NT_PRINTER_INFO_LEVEL_2* dup_printer_2( TALLOC_CTX *ctx, NT_PRINTER_INFO_LEVEL_2 *printer )
4018 NT_PRINTER_INFO_LEVEL_2 *copy;
4020 if ( !printer )
4021 return NULL;
4023 if ( !(copy = SMB_MALLOC_P(NT_PRINTER_INFO_LEVEL_2)) )
4024 return NULL;
4026 memcpy( copy, printer, sizeof(NT_PRINTER_INFO_LEVEL_2) );
4028 /* malloc()'d members copied here */
4030 copy->devmode = dup_nt_devicemode( printer->devmode );
4032 ZERO_STRUCT( copy->data );
4033 copy_printer_data( &copy->data, &printer->data );
4035 /* this is talloc()'d; very ugly that we have a structure that
4036 is half malloc()'d and half talloc()'d but that is the way
4037 that the PRINTER_INFO stuff is written right now. --jerry */
4039 copy->secdesc_buf = dup_sec_desc_buf( ctx, printer->secdesc_buf );
4041 return copy;
4044 /****************************************************************************
4045 Get a NT_PRINTER_INFO_LEVEL struct. It returns malloced memory.
4047 Previously the code had a memory allocation problem because it always
4048 used the TALLOC_CTX from the Printer_entry*. This context lasts
4049 as a long as the original handle is open. So if the client made a lot
4050 of getprinter[data]() calls, the memory usage would climb. Now we use
4051 a short lived TALLOC_CTX for printer_info_2 objects returned. We
4052 still use the Printer_entry->ctx for maintaining the cache copy though
4053 since that object must live as long as the handle by definition.
4054 --jerry
4056 ****************************************************************************/
4058 WERROR get_a_printer( Printer_entry *print_hnd, NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level,
4059 const char *sharename)
4061 WERROR result;
4062 NT_PRINTER_INFO_LEVEL *printer = NULL;
4063 fstring servername;
4065 *pp_printer = NULL;
4067 DEBUG(10,("get_a_printer: [%s] level %u\n", sharename, (unsigned int)level));
4069 switch (level) {
4070 case 2:
4071 if ((printer = SMB_MALLOC_P(NT_PRINTER_INFO_LEVEL)) == NULL) {
4072 DEBUG(0,("get_a_printer: malloc fail.\n"));
4073 return WERR_NOMEM;
4075 ZERO_STRUCTP(printer);
4077 if ( print_hnd )
4078 fstrcpy( servername, print_hnd->servername );
4079 else {
4080 fstrcpy( servername, "%L" );
4081 standard_sub_basic( "", servername, sizeof(servername)-1 );
4085 * check for cache first. A Printer handle cannot changed
4086 * to another printer object so we only check that the printer
4087 * is actually for a printer and that the printer_info pointer
4088 * is valid
4090 if ( print_hnd
4091 && (print_hnd->printer_type==PRINTER_HANDLE_IS_PRINTER)
4092 && print_hnd->printer_info )
4094 /* get_talloc_ctx() works here because we need a short
4095 lived talloc context */
4097 if ( !(printer->info_2 = dup_printer_2(get_talloc_ctx(), print_hnd->printer_info->info_2)) )
4099 DEBUG(0,("get_a_printer: unable to copy cached printer info!\n"));
4101 SAFE_FREE(printer);
4102 return WERR_NOMEM;
4105 DEBUG(10,("get_a_printer: using cached copy of printer_info_2\n"));
4107 *pp_printer = printer;
4108 result = WERR_OK;
4110 break;
4113 /* no cache for this handle; see if we can match one from another handle.
4114 Make sure to use a short lived talloc ctx */
4116 if ( print_hnd )
4117 result = find_printer_in_print_hnd_cache(get_talloc_ctx(), &printer->info_2, servername, sharename);
4119 /* fail to disk if we don't have it with any open handle */
4121 if ( !print_hnd || !W_ERROR_IS_OK(result) )
4122 result = get_a_printer_2(&printer->info_2, servername, sharename );
4124 /* we have a new printer now. Save it with this handle */
4126 if ( W_ERROR_IS_OK(result) ) {
4127 dump_a_printer(printer, level);
4129 /* save a copy in cache */
4130 if ( print_hnd && (print_hnd->printer_type==PRINTER_HANDLE_IS_PRINTER)) {
4131 if ( !print_hnd->printer_info )
4132 print_hnd->printer_info = SMB_MALLOC_P(NT_PRINTER_INFO_LEVEL);
4134 if ( print_hnd->printer_info ) {
4135 /* make sure to use the handle's talloc ctx here since
4136 the printer_2 object must last until the handle is closed */
4138 print_hnd->printer_info->info_2 = dup_printer_2(print_hnd->ctx, printer->info_2);
4140 /* don't fail the lookup just because the cache update failed */
4141 if ( !print_hnd->printer_info->info_2 )
4142 DEBUG(0,("get_a_printer: unable to copy new printer info!\n"));
4145 *pp_printer = printer;
4147 else
4148 SAFE_FREE(printer);
4150 break;
4152 default:
4153 result=WERR_UNKNOWN_LEVEL;
4154 break;
4157 DEBUG(10,("get_a_printer: [%s] level %u returning %s\n", sharename, (unsigned int)level, dos_errstr(result)));
4159 return result;
4162 /****************************************************************************
4163 Deletes a NT_PRINTER_INFO_LEVEL struct.
4164 ****************************************************************************/
4166 uint32 free_a_printer(NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level)
4168 uint32 result;
4169 NT_PRINTER_INFO_LEVEL *printer = *pp_printer;
4171 DEBUG(104,("freeing a printer at level [%d]\n", level));
4173 if (printer == NULL)
4174 return 0;
4176 switch (level) {
4177 case 2:
4178 if (printer->info_2 != NULL) {
4179 free_nt_printer_info_level_2(&printer->info_2);
4180 result=0;
4181 } else
4182 result=4;
4183 break;
4185 default:
4186 result=1;
4187 break;
4190 SAFE_FREE(*pp_printer);
4191 return result;
4194 /****************************************************************************
4195 ****************************************************************************/
4196 uint32 add_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4198 uint32 result;
4199 DEBUG(104,("adding a printer at level [%d]\n", level));
4200 dump_a_printer_driver(driver, level);
4202 switch (level) {
4203 case 3:
4204 result=add_a_printer_driver_3(driver.info_3);
4205 break;
4207 case 6:
4208 result=add_a_printer_driver_6(driver.info_6);
4209 break;
4211 default:
4212 result=1;
4213 break;
4216 return result;
4218 /****************************************************************************
4219 ****************************************************************************/
4221 WERROR get_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL *driver, uint32 level,
4222 fstring drivername, const char *architecture, uint32 version)
4224 WERROR result;
4226 switch (level) {
4227 case 3:
4228 /* Sometime we just want any version of the driver */
4230 if ( version == DRIVER_ANY_VERSION ) {
4231 /* look for Win2k first and then for NT4 */
4232 result = get_a_printer_driver_3(&driver->info_3, drivername,
4233 architecture, 3);
4235 if ( !W_ERROR_IS_OK(result) ) {
4236 result = get_a_printer_driver_3( &driver->info_3,
4237 drivername, architecture, 2 );
4239 } else {
4240 result = get_a_printer_driver_3(&driver->info_3, drivername,
4241 architecture, version);
4243 break;
4245 default:
4246 result=W_ERROR(1);
4247 break;
4250 if (W_ERROR_IS_OK(result))
4251 dump_a_printer_driver(*driver, level);
4253 return result;
4256 /****************************************************************************
4257 ****************************************************************************/
4258 uint32 free_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4260 uint32 result;
4262 switch (level) {
4263 case 3:
4265 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
4266 if (driver.info_3 != NULL)
4268 info3=driver.info_3;
4269 SAFE_FREE(info3->dependentfiles);
4270 ZERO_STRUCTP(info3);
4271 SAFE_FREE(info3);
4272 result=0;
4273 } else {
4274 result=4;
4276 break;
4278 case 6:
4280 NT_PRINTER_DRIVER_INFO_LEVEL_6 *info6;
4281 if (driver.info_6 != NULL) {
4282 info6=driver.info_6;
4283 SAFE_FREE(info6->dependentfiles);
4284 SAFE_FREE(info6->previousnames);
4285 ZERO_STRUCTP(info6);
4286 SAFE_FREE(info6);
4287 result=0;
4288 } else {
4289 result=4;
4291 break;
4293 default:
4294 result=1;
4295 break;
4297 return result;
4301 /****************************************************************************
4302 Determine whether or not a particular driver is currently assigned
4303 to a printer
4304 ****************************************************************************/
4306 BOOL printer_driver_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3 )
4308 int snum;
4309 int n_services = lp_numservices();
4310 NT_PRINTER_INFO_LEVEL *printer = NULL;
4312 if ( !info_3 )
4313 return False;
4315 DEBUG(5,("printer_driver_in_use: Beginning search through ntprinters.tdb...\n"));
4317 /* loop through the printers.tdb and check for the drivername */
4319 for (snum=0; snum<n_services; snum++) {
4320 if ( !(lp_snum_ok(snum) && lp_print_ok(snum) ) )
4321 continue;
4323 if ( !W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))) )
4324 continue;
4326 if ( !StrCaseCmp(info_3->name, printer->info_2->drivername) ) {
4327 free_a_printer( &printer, 2 );
4328 return True;
4331 free_a_printer( &printer, 2 );
4334 DEBUG(5,("printer_driver_in_use: Completed search through ntprinters.tdb...\n"));
4336 /* report that the driver is not in use by default */
4338 return False;
4342 /**********************************************************************
4343 Check to see if a ogiven file is in use by *info
4344 *********************************************************************/
4346 static BOOL drv_file_in_use( char* file, NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
4348 int i = 0;
4350 if ( !info )
4351 return False;
4353 if ( strequal(file, info->driverpath) )
4354 return True;
4356 if ( strequal(file, info->datafile) )
4357 return True;
4359 if ( strequal(file, info->configfile) )
4360 return True;
4362 if ( strequal(file, info->helpfile) )
4363 return True;
4365 /* see of there are any dependent files to examine */
4367 if ( !info->dependentfiles )
4368 return False;
4370 while ( *info->dependentfiles[i] ) {
4371 if ( strequal(file, info->dependentfiles[i]) )
4372 return True;
4373 i++;
4376 return False;
4380 /**********************************************************************
4381 Utility function to remove the dependent file pointed to by the
4382 input parameter from the list
4383 *********************************************************************/
4385 static void trim_dependent_file( fstring files[], int idx )
4388 /* bump everything down a slot */
4390 while( *files[idx+1] ) {
4391 fstrcpy( files[idx], files[idx+1] );
4392 idx++;
4395 *files[idx] = '\0';
4397 return;
4400 /**********************************************************************
4401 Check if any of the files used by src are also used by drv
4402 *********************************************************************/
4404 static BOOL trim_overlap_drv_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *src,
4405 NT_PRINTER_DRIVER_INFO_LEVEL_3 *drv )
4407 BOOL in_use = False;
4408 int i = 0;
4410 if ( !src || !drv )
4411 return False;
4413 /* check each file. Remove it from the src structure if it overlaps */
4415 if ( drv_file_in_use(src->driverpath, drv) ) {
4416 in_use = True;
4417 DEBUG(10,("Removing driverfile [%s] from list\n", src->driverpath));
4418 fstrcpy( src->driverpath, "" );
4421 if ( drv_file_in_use(src->datafile, drv) ) {
4422 in_use = True;
4423 DEBUG(10,("Removing datafile [%s] from list\n", src->datafile));
4424 fstrcpy( src->datafile, "" );
4427 if ( drv_file_in_use(src->configfile, drv) ) {
4428 in_use = True;
4429 DEBUG(10,("Removing configfile [%s] from list\n", src->configfile));
4430 fstrcpy( src->configfile, "" );
4433 if ( drv_file_in_use(src->helpfile, drv) ) {
4434 in_use = True;
4435 DEBUG(10,("Removing helpfile [%s] from list\n", src->helpfile));
4436 fstrcpy( src->helpfile, "" );
4439 /* are there any dependentfiles to examine? */
4441 if ( !src->dependentfiles )
4442 return in_use;
4444 while ( *src->dependentfiles[i] ) {
4445 if ( drv_file_in_use(src->dependentfiles[i], drv) ) {
4446 in_use = True;
4447 DEBUG(10,("Removing [%s] from dependent file list\n", src->dependentfiles[i]));
4448 trim_dependent_file( src->dependentfiles, i );
4449 } else
4450 i++;
4453 return in_use;
4456 /****************************************************************************
4457 Determine whether or not a particular driver files are currently being
4458 used by any other driver.
4460 Return value is True if any files were in use by other drivers
4461 and False otherwise.
4463 Upon return, *info has been modified to only contain the driver files
4464 which are not in use
4465 ****************************************************************************/
4467 BOOL printer_driver_files_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
4469 int i;
4470 int ndrivers;
4471 uint32 version;
4472 fstring *list = NULL;
4473 NT_PRINTER_DRIVER_INFO_LEVEL driver;
4475 if ( !info )
4476 return False;
4478 version = info->cversion;
4480 /* loop over all driver versions */
4482 DEBUG(5,("printer_driver_files_in_use: Beginning search through ntdrivers.tdb...\n"));
4484 /* get the list of drivers */
4486 list = NULL;
4487 ndrivers = get_ntdrivers(&list, info->environment, version);
4489 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n",
4490 ndrivers, info->environment, version));
4492 /* check each driver for overlap in files */
4494 for (i=0; i<ndrivers; i++) {
4495 DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
4497 ZERO_STRUCT(driver);
4499 if ( !W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, list[i], info->environment, version)) ) {
4500 SAFE_FREE(list);
4501 return True;
4504 /* check if d2 uses any files from d1 */
4505 /* only if this is a different driver than the one being deleted */
4507 if ( !strequal(info->name, driver.info_3->name) ) {
4508 if ( trim_overlap_drv_files(info, driver.info_3) ) {
4509 free_a_printer_driver(driver, 3);
4510 SAFE_FREE( list );
4511 return True;
4515 free_a_printer_driver(driver, 3);
4518 SAFE_FREE(list);
4520 DEBUG(5,("printer_driver_files_in_use: Completed search through ntdrivers.tdb...\n"));
4522 driver.info_3 = info;
4524 if ( DEBUGLEVEL >= 20 )
4525 dump_a_printer_driver( driver, 3 );
4527 return False;
4530 /****************************************************************************
4531 Actually delete the driver files. Make sure that
4532 printer_driver_files_in_use() return False before calling
4533 this.
4534 ****************************************************************************/
4536 static BOOL delete_driver_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user )
4538 int i = 0;
4539 char *s;
4540 connection_struct *conn;
4541 DATA_BLOB null_pw;
4542 NTSTATUS nt_status;
4543 fstring res_type;
4544 BOOL bad_path;
4545 SMB_STRUCT_STAT st;
4547 if ( !info_3 )
4548 return False;
4550 DEBUG(6,("delete_driver_files: deleting driver [%s] - version [%d]\n", info_3->name, info_3->cversion));
4553 * Connect to the print$ share under the same account as the
4554 * user connected to the rpc pipe. Note we must be root to
4555 * do this.
4558 null_pw = data_blob( NULL, 0 );
4559 fstrcpy(res_type, "A:");
4560 become_root();
4561 conn = make_connection_with_chdir( "print$", null_pw, res_type, user->vuid, &nt_status );
4562 unbecome_root();
4564 if ( !conn ) {
4565 DEBUG(0,("delete_driver_files: Unable to connect\n"));
4566 return False;
4569 /* Save who we are - we are temporarily becoming the connection user. */
4571 if ( !become_user(conn, conn->vuid) ) {
4572 DEBUG(0,("delete_driver_files: Can't become user!\n"));
4573 return False;
4576 /* now delete the files; must strip the '\print$' string from
4577 fron of path */
4579 if ( *info_3->driverpath ) {
4580 if ( (s = strchr( &info_3->driverpath[1], '\\' )) != NULL ) {
4581 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4582 DEBUG(10,("deleting driverfile [%s]\n", s));
4583 unlink_internals(conn, 0, s);
4587 if ( *info_3->configfile ) {
4588 if ( (s = strchr( &info_3->configfile[1], '\\' )) != NULL ) {
4589 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4590 DEBUG(10,("deleting configfile [%s]\n", s));
4591 unlink_internals(conn, 0, s);
4595 if ( *info_3->datafile ) {
4596 if ( (s = strchr( &info_3->datafile[1], '\\' )) != NULL ) {
4597 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4598 DEBUG(10,("deleting datafile [%s]\n", s));
4599 unlink_internals(conn, 0, s);
4603 if ( *info_3->helpfile ) {
4604 if ( (s = strchr( &info_3->helpfile[1], '\\' )) != NULL ) {
4605 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4606 DEBUG(10,("deleting helpfile [%s]\n", s));
4607 unlink_internals(conn, 0, s);
4611 /* check if we are done removing files */
4613 if ( info_3->dependentfiles ) {
4614 while ( *info_3->dependentfiles[i] ) {
4615 char *file;
4617 /* bypass the "\print$" portion of the path */
4619 if ( (file = strchr( info_3->dependentfiles[i]+1, '\\' )) != NULL ) {
4620 driver_unix_convert(file, conn, NULL, &bad_path, &st);
4621 DEBUG(10,("deleting dependent file [%s]\n", file));
4622 unlink_internals(conn, 0, file );
4625 i++;
4629 unbecome_user();
4631 return True;
4634 /****************************************************************************
4635 Remove a printer driver from the TDB. This assumes that the the driver was
4636 previously looked up.
4637 ***************************************************************************/
4639 WERROR delete_printer_driver( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user,
4640 uint32 version, BOOL delete_files )
4642 pstring key;
4643 const char *arch;
4644 TDB_DATA kbuf, dbuf;
4645 NT_PRINTER_DRIVER_INFO_LEVEL ctr;
4647 /* delete the tdb data first */
4649 arch = get_short_archi(info_3->environment);
4650 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX,
4651 arch, version, info_3->name);
4653 DEBUG(5,("delete_printer_driver: key = [%s] delete_files = %s\n",
4654 key, delete_files ? "TRUE" : "FALSE" ));
4656 ctr.info_3 = info_3;
4657 dump_a_printer_driver( ctr, 3 );
4659 kbuf.dptr=key;
4660 kbuf.dsize=strlen(key)+1;
4662 /* check if the driver actually exists for this environment */
4664 dbuf = tdb_fetch( tdb_drivers, kbuf );
4665 if ( !dbuf.dptr ) {
4666 DEBUG(8,("delete_printer_driver: Driver unknown [%s]\n", key));
4667 return WERR_UNKNOWN_PRINTER_DRIVER;
4670 SAFE_FREE( dbuf.dptr );
4672 /* ok... the driver exists so the delete should return success */
4674 if (tdb_delete(tdb_drivers, kbuf) == -1) {
4675 DEBUG (0,("delete_printer_driver: fail to delete %s!\n", key));
4676 return WERR_ACCESS_DENIED;
4680 * now delete any associated files if delete_files == True
4681 * even if this part failes, we return succes because the
4682 * driver doesn not exist any more
4685 if ( delete_files )
4686 delete_driver_files( info_3, user );
4689 DEBUG(5,("delete_printer_driver: driver delete successful [%s]\n", key));
4691 return WERR_OK;
4694 /****************************************************************************
4695 Store a security desc for a printer.
4696 ****************************************************************************/
4698 WERROR nt_printing_setsec(const char *printername, SEC_DESC_BUF *secdesc_ctr)
4700 SEC_DESC_BUF *new_secdesc_ctr = NULL;
4701 SEC_DESC_BUF *old_secdesc_ctr = NULL;
4702 prs_struct ps;
4703 TALLOC_CTX *mem_ctx = NULL;
4704 fstring key;
4705 WERROR status;
4707 mem_ctx = talloc_init("nt_printing_setsec");
4708 if (mem_ctx == NULL)
4709 return WERR_NOMEM;
4711 /* The old owner and group sids of the security descriptor are not
4712 present when new ACEs are added or removed by changing printer
4713 permissions through NT. If they are NULL in the new security
4714 descriptor then copy them over from the old one. */
4716 if (!secdesc_ctr->sec->owner_sid || !secdesc_ctr->sec->grp_sid) {
4717 DOM_SID *owner_sid, *group_sid;
4718 SEC_ACL *dacl, *sacl;
4719 SEC_DESC *psd = NULL;
4720 size_t size;
4722 nt_printing_getsec(mem_ctx, printername, &old_secdesc_ctr);
4724 /* Pick out correct owner and group sids */
4726 owner_sid = secdesc_ctr->sec->owner_sid ?
4727 secdesc_ctr->sec->owner_sid :
4728 old_secdesc_ctr->sec->owner_sid;
4730 group_sid = secdesc_ctr->sec->grp_sid ?
4731 secdesc_ctr->sec->grp_sid :
4732 old_secdesc_ctr->sec->grp_sid;
4734 dacl = secdesc_ctr->sec->dacl ?
4735 secdesc_ctr->sec->dacl :
4736 old_secdesc_ctr->sec->dacl;
4738 sacl = secdesc_ctr->sec->sacl ?
4739 secdesc_ctr->sec->sacl :
4740 old_secdesc_ctr->sec->sacl;
4742 /* Make a deep copy of the security descriptor */
4744 psd = make_sec_desc(mem_ctx, secdesc_ctr->sec->revision, secdesc_ctr->sec->type,
4745 owner_sid, group_sid,
4746 sacl,
4747 dacl,
4748 &size);
4750 new_secdesc_ctr = make_sec_desc_buf(mem_ctx, size, psd);
4753 if (!new_secdesc_ctr) {
4754 new_secdesc_ctr = secdesc_ctr;
4757 /* Store the security descriptor in a tdb */
4759 prs_init(&ps, (uint32)sec_desc_size(new_secdesc_ctr->sec) +
4760 sizeof(SEC_DESC_BUF), mem_ctx, MARSHALL);
4762 if (!sec_io_desc_buf("nt_printing_setsec", &new_secdesc_ctr,
4763 &ps, 1)) {
4764 status = WERR_BADFUNC;
4765 goto out;
4768 slprintf(key, sizeof(key)-1, "SECDESC/%s", printername);
4770 if (tdb_prs_store(tdb_printers, key, &ps)==0) {
4771 status = WERR_OK;
4772 } else {
4773 DEBUG(1,("Failed to store secdesc for %s\n", printername));
4774 status = WERR_BADFUNC;
4777 /* Free malloc'ed memory */
4779 out:
4781 prs_mem_free(&ps);
4782 if (mem_ctx)
4783 talloc_destroy(mem_ctx);
4784 return status;
4787 /****************************************************************************
4788 Construct a default security descriptor buffer for a printer.
4789 ****************************************************************************/
4791 static SEC_DESC_BUF *construct_default_printer_sdb(TALLOC_CTX *ctx)
4793 SEC_ACE ace[3];
4794 SEC_ACCESS sa;
4795 SEC_ACL *psa = NULL;
4796 SEC_DESC_BUF *sdb = NULL;
4797 SEC_DESC *psd = NULL;
4798 DOM_SID owner_sid;
4799 size_t sd_size;
4801 /* Create an ACE where Everyone is allowed to print */
4803 init_sec_access(&sa, PRINTER_ACE_PRINT);
4804 init_sec_ace(&ace[0], &global_sid_World, SEC_ACE_TYPE_ACCESS_ALLOWED,
4805 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
4807 /* Make the security descriptor owned by the Administrators group
4808 on the PDC of the domain. */
4810 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
4811 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4812 } else {
4814 /* Backup plan - make printer owned by admins.
4815 This should emulate a lanman printer as security
4816 settings can't be changed. */
4818 sid_copy(&owner_sid, get_global_sam_sid());
4819 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4822 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
4823 init_sec_ace(&ace[1], &owner_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
4824 sa, SEC_ACE_FLAG_OBJECT_INHERIT |
4825 SEC_ACE_FLAG_INHERIT_ONLY);
4827 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
4828 init_sec_ace(&ace[2], &owner_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
4829 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
4831 /* The ACL revision number in rpc_secdesc.h differs from the one
4832 created by NT when setting ACE entries in printer
4833 descriptors. NT4 complains about the property being edited by a
4834 NT5 machine. */
4836 if ((psa = make_sec_acl(ctx, NT4_ACL_REVISION, 3, ace)) != NULL) {
4837 psd = make_sec_desc(ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
4838 &owner_sid, NULL,
4839 NULL, psa, &sd_size);
4842 if (!psd) {
4843 DEBUG(0,("construct_default_printer_sd: Failed to make SEC_DESC.\n"));
4844 return NULL;
4847 sdb = make_sec_desc_buf(ctx, sd_size, psd);
4849 DEBUG(4,("construct_default_printer_sdb: size = %u.\n",
4850 (unsigned int)sd_size));
4852 return sdb;
4855 /****************************************************************************
4856 Get a security desc for a printer.
4857 ****************************************************************************/
4859 BOOL nt_printing_getsec(TALLOC_CTX *ctx, const char *printername, SEC_DESC_BUF **secdesc_ctr)
4861 prs_struct ps;
4862 fstring key;
4863 char *temp;
4865 if (strlen(printername) > 2 && (temp = strchr(printername + 2, '\\'))) {
4866 printername = temp + 1;
4869 /* Fetch security descriptor from tdb */
4871 slprintf(key, sizeof(key)-1, "SECDESC/%s", printername);
4873 if (tdb_prs_fetch(tdb_printers, key, &ps, ctx)!=0 ||
4874 !sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
4876 DEBUG(4,("using default secdesc for %s\n", printername));
4878 if (!(*secdesc_ctr = construct_default_printer_sdb(ctx))) {
4879 return False;
4882 /* Save default security descriptor for later */
4884 prs_init(&ps, (uint32)sec_desc_size((*secdesc_ctr)->sec) +
4885 sizeof(SEC_DESC_BUF), ctx, MARSHALL);
4887 if (sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1))
4888 tdb_prs_store(tdb_printers, key, &ps);
4890 prs_mem_free(&ps);
4892 return True;
4895 /* If security descriptor is owned by S-1-1-0 and winbindd is up,
4896 this security descriptor has been created when winbindd was
4897 down. Take ownership of security descriptor. */
4899 if (sid_equal((*secdesc_ctr)->sec->owner_sid, &global_sid_World)) {
4900 DOM_SID owner_sid;
4902 /* Change sd owner to workgroup administrator */
4904 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
4905 SEC_DESC_BUF *new_secdesc_ctr = NULL;
4906 SEC_DESC *psd = NULL;
4907 size_t size;
4909 /* Create new sd */
4911 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4913 psd = make_sec_desc(ctx, (*secdesc_ctr)->sec->revision, (*secdesc_ctr)->sec->type,
4914 &owner_sid,
4915 (*secdesc_ctr)->sec->grp_sid,
4916 (*secdesc_ctr)->sec->sacl,
4917 (*secdesc_ctr)->sec->dacl,
4918 &size);
4920 new_secdesc_ctr = make_sec_desc_buf(ctx, size, psd);
4922 /* Swap with other one */
4924 *secdesc_ctr = new_secdesc_ctr;
4926 /* Set it */
4928 nt_printing_setsec(printername, *secdesc_ctr);
4932 if (DEBUGLEVEL >= 10) {
4933 SEC_ACL *the_acl = (*secdesc_ctr)->sec->dacl;
4934 int i;
4936 DEBUG(10, ("secdesc_ctr for %s has %d aces:\n",
4937 printername, the_acl->num_aces));
4939 for (i = 0; i < the_acl->num_aces; i++) {
4940 fstring sid_str;
4942 sid_to_string(sid_str, &the_acl->ace[i].trustee);
4944 DEBUG(10, ("%s %d %d 0x%08x\n", sid_str,
4945 the_acl->ace[i].type, the_acl->ace[i].flags,
4946 the_acl->ace[i].info.mask));
4950 prs_mem_free(&ps);
4951 return True;
4954 /* error code:
4955 0: everything OK
4956 1: level not implemented
4957 2: file doesn't exist
4958 3: can't allocate memory
4959 4: can't free memory
4960 5: non existant struct
4964 A printer and a printer driver are 2 different things.
4965 NT manages them separatelly, Samba does the same.
4966 Why ? Simply because it's easier and it makes sense !
4968 Now explanation: You have 3 printers behind your samba server,
4969 2 of them are the same make and model (laser A and B). But laser B
4970 has an 3000 sheet feeder and laser A doesn't such an option.
4971 Your third printer is an old dot-matrix model for the accounting :-).
4973 If the /usr/local/samba/lib directory (default dir), you will have
4974 5 files to describe all of this.
4976 3 files for the printers (1 by printer):
4977 NTprinter_laser A
4978 NTprinter_laser B
4979 NTprinter_accounting
4980 2 files for the drivers (1 for the laser and 1 for the dot matrix)
4981 NTdriver_printer model X
4982 NTdriver_printer model Y
4984 jfm: I should use this comment for the text file to explain
4985 same thing for the forms BTW.
4986 Je devrais mettre mes commentaires en francais, ca serait mieux :-)
4990 /* Convert generic access rights to printer object specific access rights.
4991 It turns out that NT4 security descriptors use generic access rights and
4992 NT5 the object specific ones. */
4994 void map_printer_permissions(SEC_DESC *sd)
4996 int i;
4998 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
4999 se_map_generic(&sd->dacl->ace[i].info.mask,
5000 &printer_generic_mapping);
5004 /****************************************************************************
5005 Check a user has permissions to perform the given operation. We use the
5006 permission constants defined in include/rpc_spoolss.h to check the various
5007 actions we perform when checking printer access.
5009 PRINTER_ACCESS_ADMINISTER:
5010 print_queue_pause, print_queue_resume, update_printer_sec,
5011 update_printer, spoolss_addprinterex_level_2,
5012 _spoolss_setprinterdata
5014 PRINTER_ACCESS_USE:
5015 print_job_start
5017 JOB_ACCESS_ADMINISTER:
5018 print_job_delete, print_job_pause, print_job_resume,
5019 print_queue_purge
5021 ****************************************************************************/
5022 BOOL print_access_check(struct current_user *user, int snum, int access_type)
5024 SEC_DESC_BUF *secdesc = NULL;
5025 uint32 access_granted;
5026 NTSTATUS status;
5027 BOOL result;
5028 const char *pname;
5029 TALLOC_CTX *mem_ctx = NULL;
5030 extern struct current_user current_user;
5032 /* If user is NULL then use the current_user structure */
5034 if (!user)
5035 user = &current_user;
5037 /* Always allow root or printer admins to do anything */
5039 if (user->uid == 0 ||
5040 user_in_list(uidtoname(user->uid), lp_printer_admin(snum), user->groups, user->ngroups)) {
5041 return True;
5044 /* Get printer name */
5046 pname = PRINTERNAME(snum);
5048 if (!pname || !*pname) {
5049 errno = EACCES;
5050 return False;
5053 /* Get printer security descriptor */
5055 if(!(mem_ctx = talloc_init("print_access_check"))) {
5056 errno = ENOMEM;
5057 return False;
5060 nt_printing_getsec(mem_ctx, pname, &secdesc);
5062 if (access_type == JOB_ACCESS_ADMINISTER) {
5063 SEC_DESC_BUF *parent_secdesc = secdesc;
5065 /* Create a child security descriptor to check permissions
5066 against. This is because print jobs are child objects
5067 objects of a printer. */
5069 secdesc = se_create_child_secdesc(mem_ctx, parent_secdesc->sec, False);
5071 /* Now this is the bit that really confuses me. The access
5072 type needs to be changed from JOB_ACCESS_ADMINISTER to
5073 PRINTER_ACCESS_ADMINISTER for this to work. Something
5074 to do with the child (job) object becoming like a
5075 printer?? -tpot */
5077 access_type = PRINTER_ACCESS_ADMINISTER;
5080 /* Check access */
5082 map_printer_permissions(secdesc->sec);
5084 result = se_access_check(secdesc->sec, user->nt_user_token, access_type,
5085 &access_granted, &status);
5087 DEBUG(4, ("access check was %s\n", result ? "SUCCESS" : "FAILURE"));
5089 talloc_destroy(mem_ctx);
5091 if (!result)
5092 errno = EACCES;
5094 return result;
5097 /****************************************************************************
5098 Check the time parameters allow a print operation.
5099 *****************************************************************************/
5101 BOOL print_time_access_check(int snum)
5103 NT_PRINTER_INFO_LEVEL *printer = NULL;
5104 BOOL ok = False;
5105 time_t now = time(NULL);
5106 struct tm *t;
5107 uint32 mins;
5109 if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))))
5110 return False;
5112 if (printer->info_2->starttime == 0 && printer->info_2->untiltime == 0)
5113 ok = True;
5115 t = gmtime(&now);
5116 mins = (uint32)t->tm_hour*60 + (uint32)t->tm_min;
5118 if (mins >= printer->info_2->starttime && mins <= printer->info_2->untiltime)
5119 ok = True;
5121 free_a_printer(&printer, 2);
5123 if (!ok)
5124 errno = EACCES;
5126 return ok;
5129 /****************************************************************************
5130 Fill in the servername sent in the _spoolss_open_printer_ex() call
5131 ****************************************************************************/
5132 char* get_server_name( Printer_entry *printer )
5134 return printer->servername;