r1885: tighten the cache consistency with the ntprinters.tdb entry an the in memory...
[Samba/ekacnet.git] / source / printing / nt_printing.c
blobb54e1f738d91faadff3681a5943420235e04541b
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"
214 static const struct table_node archi_table[]= {
216 {"Windows 4.0", SPL_ARCH_WIN40, 0 },
217 {"Windows NT x86", SPL_ARCH_W32X86, 2 },
218 {"Windows NT R4000", SPL_ARCH_W32MIPS, 2 },
219 {"Windows NT Alpha_AXP", SPL_ARCH_W32ALPHA, 2 },
220 {"Windows NT PowerPC", SPL_ARCH_W32PPC, 2 },
221 {"Windows IA64", SPL_ARCH_IA64, 3 },
222 {NULL, "", -1 }
225 static BOOL upgrade_to_version_3(void)
227 TDB_DATA kbuf, newkey, dbuf;
229 DEBUG(0,("upgrade_to_version_3: upgrading print tdb's to version 3\n"));
231 for (kbuf = tdb_firstkey(tdb_drivers); kbuf.dptr;
232 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
234 dbuf = tdb_fetch(tdb_drivers, kbuf);
236 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) == 0) {
237 DEBUG(0,("upgrade_to_version_3:moving form\n"));
238 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) {
239 SAFE_FREE(dbuf.dptr);
240 DEBUG(0,("upgrade_to_version_3: failed to move form. Error (%s).\n", tdb_errorstr(tdb_forms)));
241 return False;
243 if (tdb_delete(tdb_drivers, kbuf) != 0) {
244 SAFE_FREE(dbuf.dptr);
245 DEBUG(0,("upgrade_to_version_3: failed to delete form. Error (%s)\n", tdb_errorstr(tdb_drivers)));
246 return False;
250 if (strncmp(kbuf.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX)) == 0) {
251 DEBUG(0,("upgrade_to_version_3:moving printer\n"));
252 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
253 SAFE_FREE(dbuf.dptr);
254 DEBUG(0,("upgrade_to_version_3: failed to move printer. Error (%s)\n", tdb_errorstr(tdb_printers)));
255 return False;
257 if (tdb_delete(tdb_drivers, kbuf) != 0) {
258 SAFE_FREE(dbuf.dptr);
259 DEBUG(0,("upgrade_to_version_3: failed to delete printer. Error (%s)\n", tdb_errorstr(tdb_drivers)));
260 return False;
264 if (strncmp(kbuf.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX)) == 0) {
265 DEBUG(0,("upgrade_to_version_3:moving secdesc\n"));
266 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
267 SAFE_FREE(dbuf.dptr);
268 DEBUG(0,("upgrade_to_version_3: failed to move secdesc. Error (%s)\n", tdb_errorstr(tdb_printers)));
269 return False;
271 if (tdb_delete(tdb_drivers, kbuf) != 0) {
272 SAFE_FREE(dbuf.dptr);
273 DEBUG(0,("upgrade_to_version_3: failed to delete secdesc. Error (%s)\n", tdb_errorstr(tdb_drivers)));
274 return False;
278 SAFE_FREE(dbuf.dptr);
281 return True;
284 /****************************************************************************
285 Open the NT printing tdbs. Done once before fork().
286 ****************************************************************************/
288 BOOL nt_printing_init(void)
290 static pid_t local_pid;
291 const char *vstring = "INFO/version";
293 if (tdb_drivers && tdb_printers && tdb_forms && local_pid == sys_getpid())
294 return True;
296 if (tdb_drivers)
297 tdb_close(tdb_drivers);
298 tdb_drivers = tdb_open_log(lock_path("ntdrivers.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
299 if (!tdb_drivers) {
300 DEBUG(0,("nt_printing_init: Failed to open nt drivers database %s (%s)\n",
301 lock_path("ntdrivers.tdb"), strerror(errno) ));
302 return False;
305 if (tdb_printers)
306 tdb_close(tdb_printers);
307 tdb_printers = tdb_open_log(lock_path("ntprinters.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
308 if (!tdb_printers) {
309 DEBUG(0,("nt_printing_init: Failed to open nt printers database %s (%s)\n",
310 lock_path("ntprinters.tdb"), strerror(errno) ));
311 return False;
314 if (tdb_forms)
315 tdb_close(tdb_forms);
316 tdb_forms = tdb_open_log(lock_path("ntforms.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
317 if (!tdb_forms) {
318 DEBUG(0,("nt_printing_init: Failed to open nt forms database %s (%s)\n",
319 lock_path("ntforms.tdb"), strerror(errno) ));
320 return False;
323 local_pid = sys_getpid();
325 /* handle a Samba upgrade */
326 tdb_lock_bystring(tdb_drivers, vstring, 0);
328 int32 vers_id;
330 /* Cope with byte-reversed older versions of the db. */
331 vers_id = tdb_fetch_int32(tdb_drivers, vstring);
332 if ((vers_id == NTDRIVERS_DATABASE_VERSION_2) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_2)) {
333 /* Written on a bigendian machine with old fetch_int code. Save as le. */
334 /* The only upgrade between V2 and V3 is to save the version in little-endian. */
335 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION);
336 vers_id = NTDRIVERS_DATABASE_VERSION;
339 if (vers_id != NTDRIVERS_DATABASE_VERSION) {
341 if ((vers_id == NTDRIVERS_DATABASE_VERSION_1) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_1)) {
342 if (!upgrade_to_version_3())
343 return False;
344 } else
345 tdb_traverse(tdb_drivers, tdb_traverse_delete_fn, NULL);
347 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION);
350 tdb_unlock_bystring(tdb_drivers, vstring);
352 update_c_setprinter(True);
355 * register callback to handle updating printers as new
356 * drivers are installed
359 message_register( MSG_PRINTER_DRVUPGRADE, do_drv_upgrade_printer );
362 * register callback to handle updating printer data
363 * when a driver is initialized
366 message_register( MSG_PRINTERDATA_INIT_RESET, reset_all_printerdata );
369 * register callback to handle invalidating the printer cache
370 * between smbd processes.
373 message_register( MSG_PRINTER_MOD, receive_printer_mod_msg);
375 /* of course, none of the message callbacks matter if you don't
376 tell messages.c that you interested in receiving PRINT_GENERAL
377 msgs. This is done in claim_connection() */
380 return True;
383 /*******************************************************************
384 Function to allow filename parsing "the old way".
385 ********************************************************************/
387 static BOOL driver_unix_convert(char *name,connection_struct *conn,
388 char *saved_last_component, BOOL *bad_path, SMB_STRUCT_STAT *pst)
390 unix_format(name);
391 unix_clean_name(name);
392 trim_string(name,"/","/");
393 return unix_convert(name, conn, saved_last_component, bad_path, pst);
396 /*******************************************************************
397 tdb traversal function for counting printers.
398 ********************************************************************/
400 static int traverse_counting_printers(TDB_CONTEXT *t, TDB_DATA key,
401 TDB_DATA data, void *context)
403 int *printer_count = (int*)context;
405 if (memcmp(PRINTERS_PREFIX, key.dptr, sizeof(PRINTERS_PREFIX)-1) == 0) {
406 (*printer_count)++;
407 DEBUG(10,("traverse_counting_printers: printer = [%s] printer_count = %d\n", key.dptr, *printer_count));
410 return 0;
413 /*******************************************************************
414 Update the spooler global c_setprinter. This variable is initialized
415 when the parent smbd starts with the number of existing printers. It
416 is monotonically increased by the current number of printers *after*
417 each add or delete printer RPC. Only Microsoft knows why... JRR020119
418 ********************************************************************/
420 uint32 update_c_setprinter(BOOL initialize)
422 int32 c_setprinter;
423 int32 printer_count = 0;
425 tdb_lock_bystring(tdb_printers, GLOBAL_C_SETPRINTER, 0);
427 /* Traverse the tdb, counting the printers */
428 tdb_traverse(tdb_printers, traverse_counting_printers, (void *)&printer_count);
430 /* If initializing, set c_setprinter to current printers count
431 * otherwise, bump it by the current printer count
433 if (!initialize)
434 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER) + printer_count;
435 else
436 c_setprinter = printer_count;
438 DEBUG(10,("update_c_setprinter: c_setprinter = %u\n", (unsigned int)c_setprinter));
439 tdb_store_int32(tdb_printers, GLOBAL_C_SETPRINTER, c_setprinter);
441 tdb_unlock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
443 return (uint32)c_setprinter;
446 /*******************************************************************
447 Get the spooler global c_setprinter, accounting for initialization.
448 ********************************************************************/
450 uint32 get_c_setprinter(void)
452 int32 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER);
454 if (c_setprinter == (int32)-1)
455 c_setprinter = update_c_setprinter(True);
457 DEBUG(10,("get_c_setprinter: c_setprinter = %d\n", c_setprinter));
459 return (uint32)c_setprinter;
462 /****************************************************************************
463 Get builtin form struct list.
464 ****************************************************************************/
466 int get_builtin_ntforms(nt_forms_struct **list)
468 *list = (nt_forms_struct *)memdup(&default_forms[0], sizeof(default_forms));
469 return sizeof(default_forms) / sizeof(default_forms[0]);
472 /****************************************************************************
473 get a builtin form struct
474 ****************************************************************************/
476 BOOL get_a_builtin_ntform(UNISTR2 *uni_formname,nt_forms_struct *form)
478 int i,count;
479 fstring form_name;
480 unistr2_to_ascii(form_name, uni_formname, sizeof(form_name)-1);
481 DEBUGADD(6,("Looking for builtin form %s \n", form_name));
482 count = sizeof(default_forms) / sizeof(default_forms[0]);
483 for (i=0;i<count;i++) {
484 if (strequal(form_name,default_forms[i].name)) {
485 DEBUGADD(6,("Found builtin form %s \n", form_name));
486 memcpy(form,&default_forms[i],sizeof(*form));
487 break;
491 return (i !=count);
494 /****************************************************************************
495 get a form struct list
496 ****************************************************************************/
497 int get_ntforms(nt_forms_struct **list)
499 TDB_DATA kbuf, newkey, dbuf;
500 nt_forms_struct *tl;
501 nt_forms_struct form;
502 int ret;
503 int i;
504 int n = 0;
506 for (kbuf = tdb_firstkey(tdb_forms);
507 kbuf.dptr;
508 newkey = tdb_nextkey(tdb_forms, kbuf), safe_free(kbuf.dptr), kbuf=newkey)
510 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) != 0)
511 continue;
513 dbuf = tdb_fetch(tdb_forms, kbuf);
514 if (!dbuf.dptr)
515 continue;
517 fstrcpy(form.name, kbuf.dptr+strlen(FORMS_PREFIX));
518 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddddddd",
519 &i, &form.flag, &form.width, &form.length, &form.left,
520 &form.top, &form.right, &form.bottom);
521 SAFE_FREE(dbuf.dptr);
522 if (ret != dbuf.dsize)
523 continue;
525 tl = Realloc(*list, sizeof(nt_forms_struct)*(n+1));
526 if (!tl) {
527 DEBUG(0,("get_ntforms: Realloc fail.\n"));
528 return 0;
530 *list = tl;
531 (*list)[n] = form;
532 n++;
536 return n;
539 /****************************************************************************
540 write a form struct list
541 ****************************************************************************/
542 int write_ntforms(nt_forms_struct **list, int number)
544 pstring buf, key;
545 int len;
546 TDB_DATA kbuf,dbuf;
547 int i;
549 for (i=0;i<number;i++) {
550 /* save index, so list is rebuilt in correct order */
551 len = tdb_pack(buf, sizeof(buf), "dddddddd",
552 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
553 (*list)[i].left, (*list)[i].top, (*list)[i].right,
554 (*list)[i].bottom);
555 if (len > sizeof(buf)) break;
556 slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[i].name);
557 kbuf.dsize = strlen(key)+1;
558 kbuf.dptr = key;
559 dbuf.dsize = len;
560 dbuf.dptr = buf;
561 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) break;
564 return i;
567 /****************************************************************************
568 add a form struct at the end of the list
569 ****************************************************************************/
570 BOOL add_a_form(nt_forms_struct **list, const FORM *form, int *count)
572 int n=0;
573 BOOL update;
574 fstring form_name;
575 nt_forms_struct *tl;
578 * NT tries to add forms even when
579 * they are already in the base
580 * only update the values if already present
583 update=False;
585 unistr2_to_ascii(form_name, &form->name, sizeof(form_name)-1);
586 for (n=0; n<*count; n++) {
587 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
588 DEBUG(103, ("NT workaround, [%s] already exists\n", form_name));
589 update=True;
590 break;
594 if (update==False) {
595 if((tl=Realloc(*list, (n+1)*sizeof(nt_forms_struct))) == NULL) {
596 DEBUG(0,("add_a_form: failed to enlarge forms list!\n"));
597 return False;
599 *list = tl;
600 unistr2_to_ascii((*list)[n].name, &form->name, sizeof((*list)[n].name)-1);
601 (*count)++;
604 (*list)[n].flag=form->flags;
605 (*list)[n].width=form->size_x;
606 (*list)[n].length=form->size_y;
607 (*list)[n].left=form->left;
608 (*list)[n].top=form->top;
609 (*list)[n].right=form->right;
610 (*list)[n].bottom=form->bottom;
612 return True;
615 /****************************************************************************
616 Delete a named form struct.
617 ****************************************************************************/
619 BOOL delete_a_form(nt_forms_struct **list, UNISTR2 *del_name, int *count, WERROR *ret)
621 pstring key;
622 TDB_DATA kbuf;
623 int n=0;
624 fstring form_name;
626 *ret = WERR_OK;
628 unistr2_to_ascii(form_name, del_name, sizeof(form_name)-1);
630 for (n=0; n<*count; n++) {
631 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
632 DEBUG(103, ("delete_a_form, [%s] in list\n", form_name));
633 break;
637 if (n == *count) {
638 DEBUG(10,("delete_a_form, [%s] not found\n", form_name));
639 *ret = WERR_INVALID_PARAM;
640 return False;
643 slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[n].name);
644 kbuf.dsize = strlen(key)+1;
645 kbuf.dptr = key;
646 if (tdb_delete(tdb_forms, kbuf) != 0) {
647 *ret = WERR_NOMEM;
648 return False;
651 return True;
654 /****************************************************************************
655 Update a form struct.
656 ****************************************************************************/
658 void update_a_form(nt_forms_struct **list, const FORM *form, int count)
660 int n=0;
661 fstring form_name;
662 unistr2_to_ascii(form_name, &(form->name), sizeof(form_name)-1);
664 DEBUG(106, ("[%s]\n", form_name));
665 for (n=0; n<count; n++) {
666 DEBUGADD(106, ("n [%d]:[%s]\n", n, (*list)[n].name));
667 if (!strncmp((*list)[n].name, form_name, strlen(form_name)))
668 break;
671 if (n==count) return;
673 (*list)[n].flag=form->flags;
674 (*list)[n].width=form->size_x;
675 (*list)[n].length=form->size_y;
676 (*list)[n].left=form->left;
677 (*list)[n].top=form->top;
678 (*list)[n].right=form->right;
679 (*list)[n].bottom=form->bottom;
682 /****************************************************************************
683 Get the nt drivers list.
684 Traverse the database and look-up the matching names.
685 ****************************************************************************/
686 int get_ntdrivers(fstring **list, const char *architecture, uint32 version)
688 int total=0;
689 const char *short_archi;
690 fstring *fl;
691 pstring key;
692 TDB_DATA kbuf, newkey;
694 short_archi = get_short_archi(architecture);
695 slprintf(key, sizeof(key)-1, "%s%s/%d/", DRIVERS_PREFIX, short_archi, version);
697 for (kbuf = tdb_firstkey(tdb_drivers);
698 kbuf.dptr;
699 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
701 if (strncmp(kbuf.dptr, key, strlen(key)) != 0)
702 continue;
704 if((fl = Realloc(*list, sizeof(fstring)*(total+1))) == NULL) {
705 DEBUG(0,("get_ntdrivers: failed to enlarge list!\n"));
706 return -1;
708 else *list = fl;
710 fstrcpy((*list)[total], kbuf.dptr+strlen(key));
711 total++;
714 return(total);
717 /****************************************************************************
718 function to do the mapping between the long architecture name and
719 the short one.
720 ****************************************************************************/
721 const char *get_short_archi(const char *long_archi)
723 int i=-1;
725 DEBUG(107,("Getting architecture dependant directory\n"));
726 do {
727 i++;
728 } while ( (archi_table[i].long_archi!=NULL ) &&
729 StrCaseCmp(long_archi, archi_table[i].long_archi) );
731 if (archi_table[i].long_archi==NULL) {
732 DEBUGADD(10,("Unknown architecture [%s] !\n", long_archi));
733 return NULL;
736 /* this might be client code - but shouldn't this be an fstrcpy etc? */
739 DEBUGADD(108,("index: [%d]\n", i));
740 DEBUGADD(108,("long architecture: [%s]\n", archi_table[i].long_archi));
741 DEBUGADD(108,("short architecture: [%s]\n", archi_table[i].short_archi));
743 return archi_table[i].short_archi;
746 /****************************************************************************
747 Version information in Microsoft files is held in a VS_VERSION_INFO structure.
748 There are two case to be covered here: PE (Portable Executable) and NE (New
749 Executable) files. Both files support the same INFO structure, but PE files
750 store the signature in unicode, and NE files store it as !unicode.
751 returns -1 on error, 1 on version info found, and 0 on no version info found.
752 ****************************************************************************/
754 static int get_file_version(files_struct *fsp, char *fname,uint32 *major, uint32 *minor)
756 int i;
757 char *buf = NULL;
758 ssize_t byte_count;
760 if ((buf=malloc(PE_HEADER_SIZE)) == NULL) {
761 DEBUG(0,("get_file_version: PE file [%s] PE Header malloc failed bytes = %d\n",
762 fname, PE_HEADER_SIZE));
763 goto error_exit;
766 /* Note: DOS_HEADER_SIZE < malloc'ed PE_HEADER_SIZE */
767 if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
768 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %lu\n",
769 fname, (unsigned long)byte_count));
770 goto no_version_info;
773 /* Is this really a DOS header? */
774 if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
775 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
776 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
777 goto no_version_info;
780 /* Skip OEM header (if any) and the DOS stub to start of Windows header */
781 if (SMB_VFS_LSEEK(fsp, fsp->fd, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (SMB_OFF_T)-1) {
782 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
783 fname, errno));
784 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
785 goto no_version_info;
788 if ((byte_count = vfs_read_data(fsp, buf, PE_HEADER_SIZE)) < PE_HEADER_SIZE) {
789 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %lu\n",
790 fname, (unsigned long)byte_count));
791 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
792 goto no_version_info;
795 /* The header may be a PE (Portable Executable) or an NE (New Executable) */
796 if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
797 unsigned int num_sections;
798 unsigned int section_table_bytes;
800 if (SVAL(buf,PE_HEADER_MACHINE_OFFSET) != PE_HEADER_MACHINE_I386) {
801 DEBUG(3,("get_file_version: PE file [%s] wrong machine = 0x%x\n",
802 fname, SVAL(buf,PE_HEADER_MACHINE_OFFSET)));
803 /* At this point, we assume the file is in error. It still could be somthing
804 * else besides a PE file, but it unlikely at this point.
806 goto error_exit;
809 /* get the section table */
810 num_sections = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
811 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
812 if (section_table_bytes == 0)
813 goto error_exit;
815 SAFE_FREE(buf);
816 if ((buf=malloc(section_table_bytes)) == NULL) {
817 DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
818 fname, section_table_bytes));
819 goto error_exit;
822 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
823 DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %lu\n",
824 fname, (unsigned long)byte_count));
825 goto error_exit;
828 /* Iterate the section table looking for the resource section ".rsrc" */
829 for (i = 0; i < num_sections; i++) {
830 int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
832 if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
833 unsigned int section_pos = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
834 unsigned int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
836 if (section_bytes == 0)
837 goto error_exit;
839 SAFE_FREE(buf);
840 if ((buf=malloc(section_bytes)) == NULL) {
841 DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
842 fname, section_bytes));
843 goto error_exit;
846 /* Seek to the start of the .rsrc section info */
847 if (SMB_VFS_LSEEK(fsp, fsp->fd, section_pos, SEEK_SET) == (SMB_OFF_T)-1) {
848 DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
849 fname, errno));
850 goto error_exit;
853 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
854 DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %lu\n",
855 fname, (unsigned long)byte_count));
856 goto error_exit;
859 if (section_bytes < VS_VERSION_INFO_UNICODE_SIZE)
860 goto error_exit;
862 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
863 /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
864 if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
865 /* Align to next long address */
866 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
868 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
869 *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
870 *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
872 DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
873 fname, *major, *minor,
874 (*major>>16)&0xffff, *major&0xffff,
875 (*minor>>16)&0xffff, *minor&0xffff));
876 SAFE_FREE(buf);
877 return 1;
884 /* Version info not found, fall back to origin date/time */
885 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
886 SAFE_FREE(buf);
887 return 0;
889 } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
890 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
891 DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
892 fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
893 /* At this point, we assume the file is in error. It still could be somthing
894 * else besides a NE file, but it unlikely at this point. */
895 goto error_exit;
898 /* Allocate a bit more space to speed up things */
899 SAFE_FREE(buf);
900 if ((buf=malloc(VS_NE_BUF_SIZE)) == NULL) {
901 DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes = %d\n",
902 fname, PE_HEADER_SIZE));
903 goto error_exit;
906 /* This is a HACK! I got tired of trying to sort through the messy
907 * 'NE' file format. If anyone wants to clean this up please have at
908 * it, but this works. 'NE' files will eventually fade away. JRR */
909 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
910 /* Cover case that should not occur in a well formed 'NE' .dll file */
911 if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
913 for(i=0; i<byte_count; i++) {
914 /* Fast skip past data that can't possibly match */
915 if (buf[i] != 'V') continue;
917 /* Potential match data crosses buf boundry, move it to beginning
918 * of buf, and fill the buf with as much as it will hold. */
919 if (i>byte_count-VS_VERSION_INFO_SIZE) {
920 int bc;
922 memcpy(buf, &buf[i], byte_count-i);
923 if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
924 (byte_count-i))) < 0) {
926 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
927 fname, errno));
928 goto error_exit;
931 byte_count = bc + (byte_count - i);
932 if (byte_count<VS_VERSION_INFO_SIZE) break;
934 i = 0;
937 /* Check that the full signature string and the magic number that
938 * follows exist (not a perfect solution, but the chances that this
939 * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
940 * twice, as it is simpler to read the code. */
941 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
942 /* Compute skip alignment to next long address */
943 int skip = -(SMB_VFS_LSEEK(fsp, fsp->fd, 0, SEEK_CUR) - (byte_count - i) +
944 sizeof(VS_SIGNATURE)) & 3;
945 if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
947 *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
948 *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
949 DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
950 fname, *major, *minor,
951 (*major>>16)&0xffff, *major&0xffff,
952 (*minor>>16)&0xffff, *minor&0xffff));
953 SAFE_FREE(buf);
954 return 1;
959 /* Version info not found, fall back to origin date/time */
960 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
961 SAFE_FREE(buf);
962 return 0;
964 } else
965 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
966 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
967 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
969 no_version_info:
970 SAFE_FREE(buf);
971 return 0;
973 error_exit:
974 SAFE_FREE(buf);
975 return -1;
978 /****************************************************************************
979 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
980 share one or more files. During the MS installation process files are checked
981 to insure that only a newer version of a shared file is installed over an
982 older version. There are several possibilities for this comparison. If there
983 is no previous version, the new one is newer (obviously). If either file is
984 missing the version info structure, compare the creation date (on Unix use
985 the modification date). Otherwise chose the numerically larger version number.
986 ****************************************************************************/
988 static int file_version_is_newer(connection_struct *conn, fstring new_file, fstring old_file)
990 BOOL use_version = True;
991 pstring filepath;
993 uint32 new_major;
994 uint32 new_minor;
995 time_t new_create_time;
997 uint32 old_major;
998 uint32 old_minor;
999 time_t old_create_time;
1001 int access_mode;
1002 int action;
1003 files_struct *fsp = NULL;
1004 SMB_STRUCT_STAT st;
1005 SMB_STRUCT_STAT stat_buf;
1006 BOOL bad_path;
1008 ZERO_STRUCT(st);
1009 ZERO_STRUCT(stat_buf);
1010 new_create_time = (time_t)0;
1011 old_create_time = (time_t)0;
1013 /* Get file version info (if available) for previous file (if it exists) */
1014 pstrcpy(filepath, old_file);
1016 driver_unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
1018 fsp = open_file_shared(conn, filepath, &stat_buf,
1019 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1020 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1021 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1022 if (!fsp) {
1023 /* Old file not found, so by definition new file is in fact newer */
1024 DEBUG(10,("file_version_is_newer: Can't open old file [%s], errno = %d\n",
1025 filepath, errno));
1026 return True;
1028 } else {
1029 int ret = get_file_version(fsp, old_file, &old_major, &old_minor);
1030 if (ret == -1) goto error_exit;
1032 if (!ret) {
1033 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1034 old_file));
1035 use_version = False;
1036 if (SMB_VFS_FSTAT(fsp, fsp->fd, &st) == -1) goto error_exit;
1037 old_create_time = st.st_mtime;
1038 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", old_create_time));
1041 close_file(fsp, True);
1043 /* Get file version info (if available) for new file */
1044 pstrcpy(filepath, new_file);
1045 driver_unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
1047 fsp = open_file_shared(conn, filepath, &stat_buf,
1048 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1049 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1050 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1051 if (!fsp) {
1052 /* New file not found, this shouldn't occur if the caller did its job */
1053 DEBUG(3,("file_version_is_newer: Can't open new file [%s], errno = %d\n",
1054 filepath, errno));
1055 goto error_exit;
1057 } else {
1058 int ret = get_file_version(fsp, new_file, &new_major, &new_minor);
1059 if (ret == -1) goto error_exit;
1061 if (!ret) {
1062 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1063 new_file));
1064 use_version = False;
1065 if (SMB_VFS_FSTAT(fsp, fsp->fd, &st) == -1) goto error_exit;
1066 new_create_time = st.st_mtime;
1067 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", new_create_time));
1070 close_file(fsp, True);
1072 if (use_version && (new_major != old_major || new_minor != old_minor)) {
1073 /* Compare versions and choose the larger version number */
1074 if (new_major > old_major ||
1075 (new_major == old_major && new_minor > old_minor)) {
1077 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1078 return True;
1080 else {
1081 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1082 return False;
1085 } else {
1086 /* Compare modification time/dates and choose the newest time/date */
1087 if (new_create_time > old_create_time) {
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;
1097 error_exit:
1098 if(fsp)
1099 close_file(fsp, True);
1100 return -1;
1103 /****************************************************************************
1104 Determine the correct cVersion associated with an architecture and driver
1105 ****************************************************************************/
1106 static uint32 get_correct_cversion(const char *architecture, fstring driverpath_in,
1107 struct current_user *user, WERROR *perr)
1109 int cversion;
1110 int access_mode;
1111 int action;
1112 NTSTATUS nt_status;
1113 pstring driverpath;
1114 DATA_BLOB null_pw;
1115 fstring res_type;
1116 files_struct *fsp = NULL;
1117 BOOL bad_path;
1118 SMB_STRUCT_STAT st;
1119 connection_struct *conn;
1121 ZERO_STRUCT(st);
1123 *perr = WERR_INVALID_PARAM;
1125 /* If architecture is Windows 95/98/ME, the version is always 0. */
1126 if (strcmp(architecture, "WIN40") == 0) {
1127 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
1128 *perr = WERR_OK;
1129 return 0;
1133 * Connect to the print$ share under the same account as the user connected
1134 * to the rpc pipe. Note we must still be root to do this.
1137 /* Null password is ok - we are already an authenticated user... */
1138 null_pw = data_blob(NULL, 0);
1139 fstrcpy(res_type, "A:");
1140 become_root();
1141 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1142 unbecome_root();
1144 if (conn == NULL) {
1145 DEBUG(0,("get_correct_cversion: Unable to connect\n"));
1146 *perr = ntstatus_to_werror(nt_status);
1147 return -1;
1150 /* We are temporarily becoming the connection user. */
1151 if (!become_user(conn, user->vuid)) {
1152 DEBUG(0,("get_correct_cversion: Can't become user!\n"));
1153 *perr = WERR_ACCESS_DENIED;
1154 return -1;
1157 /* Open the driver file (Portable Executable format) and determine the
1158 * deriver the cversion. */
1159 slprintf(driverpath, sizeof(driverpath)-1, "%s/%s", architecture, driverpath_in);
1161 driver_unix_convert(driverpath,conn,NULL,&bad_path,&st);
1163 fsp = open_file_shared(conn, driverpath, &st,
1164 SET_DENY_MODE(DENY_NONE)|SET_OPEN_MODE(DOS_OPEN_RDONLY),
1165 (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
1166 FILE_ATTRIBUTE_NORMAL, INTERNAL_OPEN_ONLY, &access_mode, &action);
1167 if (!fsp) {
1168 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = %d\n",
1169 driverpath, errno));
1170 *perr = WERR_ACCESS_DENIED;
1171 goto error_exit;
1173 else {
1174 uint32 major;
1175 uint32 minor;
1176 int ret = get_file_version(fsp, driverpath, &major, &minor);
1177 if (ret == -1) goto error_exit;
1179 if (!ret) {
1180 DEBUG(6,("get_correct_cversion: Version info not found [%s]\n", driverpath));
1181 goto error_exit;
1185 * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
1186 * for more details. Version in this case is not just the version of the
1187 * file, but the version in the sense of kernal mode (2) vs. user mode
1188 * (3) drivers. Other bits of the version fields are the version info.
1189 * JRR 010716
1191 cversion = major & 0x0000ffff;
1192 switch (cversion) {
1193 case 2: /* WinNT drivers */
1194 case 3: /* Win2K drivers */
1195 break;
1197 default:
1198 DEBUG(6,("get_correct_cversion: cversion invalid [%s] cversion = %d\n",
1199 driverpath, cversion));
1200 goto error_exit;
1203 DEBUG(10,("get_correct_cversion: Version info found [%s] major = 0x%x minor = 0x%x\n",
1204 driverpath, major, minor));
1207 DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
1208 driverpath, cversion));
1210 close_file(fsp, True);
1211 close_cnum(conn, user->vuid);
1212 unbecome_user();
1213 *perr = WERR_OK;
1214 return cversion;
1217 error_exit:
1219 if(fsp)
1220 close_file(fsp, True);
1222 close_cnum(conn, user->vuid);
1223 unbecome_user();
1224 return -1;
1227 /****************************************************************************
1228 ****************************************************************************/
1229 static WERROR clean_up_driver_struct_level_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver,
1230 struct current_user *user)
1232 const char *architecture;
1233 fstring new_name;
1234 char *p;
1235 int i;
1236 WERROR err;
1238 /* clean up the driver name.
1239 * we can get .\driver.dll
1240 * or worse c:\windows\system\driver.dll !
1242 /* using an intermediate string to not have overlaping memcpy()'s */
1243 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1244 fstrcpy(new_name, p+1);
1245 fstrcpy(driver->driverpath, new_name);
1248 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1249 fstrcpy(new_name, p+1);
1250 fstrcpy(driver->datafile, new_name);
1253 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1254 fstrcpy(new_name, p+1);
1255 fstrcpy(driver->configfile, new_name);
1258 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1259 fstrcpy(new_name, p+1);
1260 fstrcpy(driver->helpfile, new_name);
1263 if (driver->dependentfiles) {
1264 for (i=0; *driver->dependentfiles[i]; i++) {
1265 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1266 fstrcpy(new_name, p+1);
1267 fstrcpy(driver->dependentfiles[i], new_name);
1272 architecture = get_short_archi(driver->environment);
1274 /* jfm:7/16/2000 the client always sends the cversion=0.
1275 * The server should check which version the driver is by reading
1276 * the PE header of driver->driverpath.
1278 * For Windows 95/98 the version is 0 (so the value sent is correct)
1279 * For Windows NT (the architecture doesn't matter)
1280 * NT 3.1: cversion=0
1281 * NT 3.5/3.51: cversion=1
1282 * NT 4: cversion=2
1283 * NT2K: cversion=3
1285 if ((driver->cversion = get_correct_cversion( architecture,
1286 driver->driverpath, user, &err)) == -1)
1287 return err;
1289 return WERR_OK;
1292 /****************************************************************************
1293 ****************************************************************************/
1294 static WERROR clean_up_driver_struct_level_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver, struct current_user *user)
1296 const char *architecture;
1297 fstring new_name;
1298 char *p;
1299 int i;
1300 WERROR err;
1302 /* clean up the driver name.
1303 * we can get .\driver.dll
1304 * or worse c:\windows\system\driver.dll !
1306 /* using an intermediate string to not have overlaping memcpy()'s */
1307 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1308 fstrcpy(new_name, p+1);
1309 fstrcpy(driver->driverpath, new_name);
1312 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1313 fstrcpy(new_name, p+1);
1314 fstrcpy(driver->datafile, new_name);
1317 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1318 fstrcpy(new_name, p+1);
1319 fstrcpy(driver->configfile, new_name);
1322 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1323 fstrcpy(new_name, p+1);
1324 fstrcpy(driver->helpfile, new_name);
1327 if (driver->dependentfiles) {
1328 for (i=0; *driver->dependentfiles[i]; i++) {
1329 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1330 fstrcpy(new_name, p+1);
1331 fstrcpy(driver->dependentfiles[i], new_name);
1336 architecture = get_short_archi(driver->environment);
1338 /* jfm:7/16/2000 the client always sends the cversion=0.
1339 * The server should check which version the driver is by reading
1340 * the PE header of driver->driverpath.
1342 * For Windows 95/98 the version is 0 (so the value sent is correct)
1343 * For Windows NT (the architecture doesn't matter)
1344 * NT 3.1: cversion=0
1345 * NT 3.5/3.51: cversion=1
1346 * NT 4: cversion=2
1347 * NT2K: cversion=3
1349 if ((driver->version = get_correct_cversion(architecture, driver->driverpath, user, &err)) == -1)
1350 return err;
1352 return WERR_OK;
1355 /****************************************************************************
1356 ****************************************************************************/
1357 WERROR clean_up_driver_struct(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract,
1358 uint32 level, struct current_user *user)
1360 switch (level) {
1361 case 3:
1363 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1364 driver=driver_abstract.info_3;
1365 return clean_up_driver_struct_level_3(driver, user);
1367 case 6:
1369 NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver;
1370 driver=driver_abstract.info_6;
1371 return clean_up_driver_struct_level_6(driver, user);
1373 default:
1374 return WERR_INVALID_PARAM;
1378 /****************************************************************************
1379 This function sucks and should be replaced. JRA.
1380 ****************************************************************************/
1382 static void convert_level_6_to_level3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *dst, NT_PRINTER_DRIVER_INFO_LEVEL_6 *src)
1384 dst->cversion = src->version;
1386 fstrcpy( dst->name, src->name);
1387 fstrcpy( dst->environment, src->environment);
1388 fstrcpy( dst->driverpath, src->driverpath);
1389 fstrcpy( dst->datafile, src->datafile);
1390 fstrcpy( dst->configfile, src->configfile);
1391 fstrcpy( dst->helpfile, src->helpfile);
1392 fstrcpy( dst->monitorname, src->monitorname);
1393 fstrcpy( dst->defaultdatatype, src->defaultdatatype);
1394 dst->dependentfiles = src->dependentfiles;
1397 #if 0 /* Debugging function */
1399 static char* ffmt(unsigned char *c){
1400 int i;
1401 static char ffmt_str[17];
1403 for (i=0; i<16; i++) {
1404 if ((c[i] < ' ') || (c[i] > '~'))
1405 ffmt_str[i]='.';
1406 else
1407 ffmt_str[i]=c[i];
1409 ffmt_str[16]='\0';
1410 return ffmt_str;
1413 #endif
1415 /****************************************************************************
1416 ****************************************************************************/
1417 BOOL move_driver_to_download_area(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract, uint32 level,
1418 struct current_user *user, WERROR *perr)
1420 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1421 NT_PRINTER_DRIVER_INFO_LEVEL_3 converted_driver;
1422 const char *architecture;
1423 pstring new_dir;
1424 pstring old_name;
1425 pstring new_name;
1426 DATA_BLOB null_pw;
1427 connection_struct *conn;
1428 NTSTATUS nt_status;
1429 pstring inbuf;
1430 pstring outbuf;
1431 fstring res_type;
1432 BOOL bad_path;
1433 SMB_STRUCT_STAT st;
1434 int ver = 0;
1435 int i;
1437 memset(inbuf, '\0', sizeof(inbuf));
1438 memset(outbuf, '\0', sizeof(outbuf));
1439 *perr = WERR_OK;
1441 if (level==3)
1442 driver=driver_abstract.info_3;
1443 else if (level==6) {
1444 convert_level_6_to_level3(&converted_driver, driver_abstract.info_6);
1445 driver = &converted_driver;
1446 } else {
1447 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)level ));
1448 return False;
1451 architecture = get_short_archi(driver->environment);
1454 * Connect to the print$ share under the same account as the user connected to the rpc pipe.
1455 * Note we must be root to do this.
1458 null_pw = data_blob(NULL, 0);
1459 fstrcpy(res_type, "A:");
1460 become_root();
1461 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1462 unbecome_root();
1464 if (conn == NULL) {
1465 DEBUG(0,("move_driver_to_download_area: Unable to connect\n"));
1466 *perr = ntstatus_to_werror(nt_status);
1467 return False;
1471 * Save who we are - we are temporarily becoming the connection user.
1474 if (!become_user(conn, conn->vuid)) {
1475 DEBUG(0,("move_driver_to_download_area: Can't become user!\n"));
1476 return False;
1480 * make the directories version and version\driver_name
1481 * under the architecture directory.
1483 DEBUG(5,("Creating first directory\n"));
1484 slprintf(new_dir, sizeof(new_dir)-1, "%s/%d", architecture, driver->cversion);
1485 driver_unix_convert(new_dir, conn, NULL, &bad_path, &st);
1486 mkdir_internal(conn, new_dir);
1488 /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1489 * listed for this driver which has already been moved, skip it (note:
1490 * drivers may list the same file name several times. Then check if the
1491 * file already exists in archi\cversion\, if so, check that the version
1492 * info (or time stamps if version info is unavailable) is newer (or the
1493 * date is later). If it is, move it to archi\cversion\filexxx.yyy.
1494 * Otherwise, delete the file.
1496 * If a file is not moved to archi\cversion\ because of an error, all the
1497 * rest of the 'unmoved' driver files are removed from archi\. If one or
1498 * more of the driver's files was already moved to archi\cversion\, it
1499 * potentially leaves the driver in a partially updated state. Version
1500 * trauma will most likely occur if an client attempts to use any printer
1501 * bound to the driver. Perhaps a rewrite to make sure the moves can be
1502 * done is appropriate... later JRR
1505 DEBUG(5,("Moving files now !\n"));
1507 if (driver->driverpath && strlen(driver->driverpath)) {
1508 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->driverpath);
1509 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->driverpath);
1510 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1511 NTSTATUS status;
1512 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1513 status = rename_internals(conn, new_name, old_name, 0, True);
1514 if (!NT_STATUS_IS_OK(status)) {
1515 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1516 new_name, old_name));
1517 *perr = ntstatus_to_werror(status);
1518 unlink_internals(conn, 0, new_name);
1519 ver = -1;
1521 } else {
1522 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1523 unlink_internals(conn, 0, new_name);
1527 if (driver->datafile && strlen(driver->datafile)) {
1528 if (!strequal(driver->datafile, driver->driverpath)) {
1529 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->datafile);
1530 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->datafile);
1531 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1532 NTSTATUS status;
1533 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1534 status = rename_internals(conn, new_name, old_name, 0, True);
1535 if (!NT_STATUS_IS_OK(status)) {
1536 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1537 new_name, old_name));
1538 *perr = ntstatus_to_werror(status);
1539 unlink_internals(conn, 0, new_name);
1540 ver = -1;
1542 } else {
1543 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1544 unlink_internals(conn, 0, new_name);
1549 if (driver->configfile && strlen(driver->configfile)) {
1550 if (!strequal(driver->configfile, driver->driverpath) &&
1551 !strequal(driver->configfile, driver->datafile)) {
1552 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->configfile);
1553 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->configfile);
1554 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1555 NTSTATUS status;
1556 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1557 status = rename_internals(conn, new_name, old_name, 0, True);
1558 if (!NT_STATUS_IS_OK(status)) {
1559 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1560 new_name, old_name));
1561 *perr = ntstatus_to_werror(status);
1562 unlink_internals(conn, 0, new_name);
1563 ver = -1;
1565 } else {
1566 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1567 unlink_internals(conn, 0, new_name);
1572 if (driver->helpfile && strlen(driver->helpfile)) {
1573 if (!strequal(driver->helpfile, driver->driverpath) &&
1574 !strequal(driver->helpfile, driver->datafile) &&
1575 !strequal(driver->helpfile, driver->configfile)) {
1576 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->helpfile);
1577 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->helpfile);
1578 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1579 NTSTATUS status;
1580 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1581 status = rename_internals(conn, new_name, old_name, 0, True);
1582 if (!NT_STATUS_IS_OK(status)) {
1583 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1584 new_name, old_name));
1585 *perr = ntstatus_to_werror(status);
1586 unlink_internals(conn, 0, new_name);
1587 ver = -1;
1589 } else {
1590 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1591 unlink_internals(conn, 0, new_name);
1596 if (driver->dependentfiles) {
1597 for (i=0; *driver->dependentfiles[i]; i++) {
1598 if (!strequal(driver->dependentfiles[i], driver->driverpath) &&
1599 !strequal(driver->dependentfiles[i], driver->datafile) &&
1600 !strequal(driver->dependentfiles[i], driver->configfile) &&
1601 !strequal(driver->dependentfiles[i], driver->helpfile)) {
1602 int j;
1603 for (j=0; j < i; j++) {
1604 if (strequal(driver->dependentfiles[i], driver->dependentfiles[j])) {
1605 goto NextDriver;
1609 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->dependentfiles[i]);
1610 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->dependentfiles[i]);
1611 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1612 NTSTATUS status;
1613 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1614 status = rename_internals(conn, new_name, old_name, 0, True);
1615 if (!NT_STATUS_IS_OK(status)) {
1616 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1617 new_name, old_name));
1618 *perr = ntstatus_to_werror(status);
1619 unlink_internals(conn, 0, new_name);
1620 ver = -1;
1622 } else {
1623 driver_unix_convert(new_name, conn, NULL, &bad_path, &st);
1624 unlink_internals(conn, 0, new_name);
1627 NextDriver: ;
1631 close_cnum(conn, user->vuid);
1632 unbecome_user();
1634 return ver == -1 ? False : True;
1637 /****************************************************************************
1638 ****************************************************************************/
1639 static uint32 add_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver)
1641 int len, buflen;
1642 const char *architecture;
1643 pstring directory;
1644 fstring temp_name;
1645 pstring key;
1646 char *buf;
1647 int i, ret;
1648 TDB_DATA kbuf, dbuf;
1650 architecture = get_short_archi(driver->environment);
1652 /* The names are relative. We store them in the form: \print$\arch\version\driver.xxx
1653 * \\server is added in the rpc server layer.
1654 * It does make sense to NOT store the server's name in the printer TDB.
1657 slprintf(directory, sizeof(directory)-1, "\\print$\\%s\\%d\\", architecture, driver->cversion);
1659 /* .inf files do not always list a file for each of the four standard files.
1660 * Don't prepend a path to a null filename, or client claims:
1661 * "The server on which the printer resides does not have a suitable
1662 * <printer driver name> printer driver installed. Click OK if you
1663 * wish to install the driver on your local machine."
1665 if (strlen(driver->driverpath)) {
1666 fstrcpy(temp_name, driver->driverpath);
1667 slprintf(driver->driverpath, sizeof(driver->driverpath)-1, "%s%s", directory, temp_name);
1670 if (strlen(driver->datafile)) {
1671 fstrcpy(temp_name, driver->datafile);
1672 slprintf(driver->datafile, sizeof(driver->datafile)-1, "%s%s", directory, temp_name);
1675 if (strlen(driver->configfile)) {
1676 fstrcpy(temp_name, driver->configfile);
1677 slprintf(driver->configfile, sizeof(driver->configfile)-1, "%s%s", directory, temp_name);
1680 if (strlen(driver->helpfile)) {
1681 fstrcpy(temp_name, driver->helpfile);
1682 slprintf(driver->helpfile, sizeof(driver->helpfile)-1, "%s%s", directory, temp_name);
1685 if (driver->dependentfiles) {
1686 for (i=0; *driver->dependentfiles[i]; i++) {
1687 fstrcpy(temp_name, driver->dependentfiles[i]);
1688 slprintf(driver->dependentfiles[i], sizeof(driver->dependentfiles[i])-1, "%s%s", directory, temp_name);
1692 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, driver->cversion, driver->name);
1694 DEBUG(5,("add_a_printer_driver_3: Adding driver with key %s\n", key ));
1696 buf = NULL;
1697 len = buflen = 0;
1699 again:
1700 len = 0;
1701 len += tdb_pack(buf+len, buflen-len, "dffffffff",
1702 driver->cversion,
1703 driver->name,
1704 driver->environment,
1705 driver->driverpath,
1706 driver->datafile,
1707 driver->configfile,
1708 driver->helpfile,
1709 driver->monitorname,
1710 driver->defaultdatatype);
1712 if (driver->dependentfiles) {
1713 for (i=0; *driver->dependentfiles[i]; i++) {
1714 len += tdb_pack(buf+len, buflen-len, "f",
1715 driver->dependentfiles[i]);
1719 if (len != buflen) {
1720 char *tb;
1722 tb = (char *)Realloc(buf, len);
1723 if (!tb) {
1724 DEBUG(0,("add_a_printer_driver_3: failed to enlarge buffer\n!"));
1725 ret = -1;
1726 goto done;
1728 else buf = tb;
1729 buflen = len;
1730 goto again;
1734 kbuf.dptr = key;
1735 kbuf.dsize = strlen(key)+1;
1736 dbuf.dptr = buf;
1737 dbuf.dsize = len;
1739 ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
1741 done:
1742 if (ret)
1743 DEBUG(0,("add_a_printer_driver_3: Adding driver with key %s failed.\n", key ));
1745 SAFE_FREE(buf);
1746 return ret;
1749 /****************************************************************************
1750 ****************************************************************************/
1751 static uint32 add_a_printer_driver_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver)
1753 NT_PRINTER_DRIVER_INFO_LEVEL_3 info3;
1755 ZERO_STRUCT(info3);
1756 info3.cversion = driver->version;
1757 fstrcpy(info3.name,driver->name);
1758 fstrcpy(info3.environment,driver->environment);
1759 fstrcpy(info3.driverpath,driver->driverpath);
1760 fstrcpy(info3.datafile,driver->datafile);
1761 fstrcpy(info3.configfile,driver->configfile);
1762 fstrcpy(info3.helpfile,driver->helpfile);
1763 fstrcpy(info3.monitorname,driver->monitorname);
1764 fstrcpy(info3.defaultdatatype,driver->defaultdatatype);
1765 info3.dependentfiles = driver->dependentfiles;
1767 return add_a_printer_driver_3(&info3);
1771 /****************************************************************************
1772 ****************************************************************************/
1773 static WERROR get_a_printer_driver_3_default(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, const char *driver, const char *arch)
1775 NT_PRINTER_DRIVER_INFO_LEVEL_3 info;
1777 ZERO_STRUCT(info);
1779 fstrcpy(info.name, driver);
1780 fstrcpy(info.defaultdatatype, "RAW");
1782 fstrcpy(info.driverpath, "");
1783 fstrcpy(info.datafile, "");
1784 fstrcpy(info.configfile, "");
1785 fstrcpy(info.helpfile, "");
1787 if ((info.dependentfiles=(fstring *)malloc(2*sizeof(fstring))) == NULL)
1788 return WERR_NOMEM;
1790 memset(info.dependentfiles, '\0', 2*sizeof(fstring));
1791 fstrcpy(info.dependentfiles[0], "");
1793 *info_ptr = memdup(&info, sizeof(info));
1795 return WERR_OK;
1798 /****************************************************************************
1799 ****************************************************************************/
1800 static WERROR get_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring drivername, const char *arch, uint32 version)
1802 NT_PRINTER_DRIVER_INFO_LEVEL_3 driver;
1803 TDB_DATA kbuf, dbuf;
1804 const char *architecture;
1805 int len = 0;
1806 int i;
1807 pstring key;
1809 ZERO_STRUCT(driver);
1811 architecture = get_short_archi(arch);
1813 /* Windows 4.0 (i.e. win9x) should always use a version of 0 */
1815 if ( strcmp( architecture, SPL_ARCH_WIN40 ) == 0 )
1816 version = 0;
1818 DEBUG(8,("get_a_printer_driver_3: [%s%s/%d/%s]\n", DRIVERS_PREFIX, architecture, version, drivername));
1820 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, version, drivername);
1822 kbuf.dptr = key;
1823 kbuf.dsize = strlen(key)+1;
1825 dbuf = tdb_fetch(tdb_drivers, kbuf);
1826 if (!dbuf.dptr)
1827 return WERR_UNKNOWN_PRINTER_DRIVER;
1829 len += tdb_unpack(dbuf.dptr, dbuf.dsize, "dffffffff",
1830 &driver.cversion,
1831 driver.name,
1832 driver.environment,
1833 driver.driverpath,
1834 driver.datafile,
1835 driver.configfile,
1836 driver.helpfile,
1837 driver.monitorname,
1838 driver.defaultdatatype);
1840 i=0;
1841 while (len < dbuf.dsize) {
1842 fstring *tddfs;
1844 tddfs = (fstring *)Realloc(driver.dependentfiles,
1845 sizeof(fstring)*(i+2));
1846 if (tddfs == NULL) {
1847 DEBUG(0,("get_a_printer_driver_3: failed to enlarge buffer!\n"));
1848 break;
1850 else driver.dependentfiles = tddfs;
1852 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f",
1853 &driver.dependentfiles[i]);
1854 i++;
1857 if (driver.dependentfiles != NULL)
1858 fstrcpy(driver.dependentfiles[i], "");
1860 SAFE_FREE(dbuf.dptr);
1862 if (len != dbuf.dsize) {
1863 SAFE_FREE(driver.dependentfiles);
1865 return get_a_printer_driver_3_default(info_ptr, drivername, arch);
1868 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&driver, sizeof(driver));
1870 return WERR_OK;
1873 /****************************************************************************
1874 Debugging function, dump at level 6 the struct in the logs.
1875 ****************************************************************************/
1877 static uint32 dump_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
1879 uint32 result;
1880 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
1881 int i;
1883 DEBUG(20,("Dumping printer driver at level [%d]\n", level));
1885 switch (level)
1887 case 3:
1889 if (driver.info_3 == NULL)
1890 result=5;
1891 else {
1892 info3=driver.info_3;
1894 DEBUGADD(20,("version:[%d]\n", info3->cversion));
1895 DEBUGADD(20,("name:[%s]\n", info3->name));
1896 DEBUGADD(20,("environment:[%s]\n", info3->environment));
1897 DEBUGADD(20,("driverpath:[%s]\n", info3->driverpath));
1898 DEBUGADD(20,("datafile:[%s]\n", info3->datafile));
1899 DEBUGADD(20,("configfile:[%s]\n", info3->configfile));
1900 DEBUGADD(20,("helpfile:[%s]\n", info3->helpfile));
1901 DEBUGADD(20,("monitorname:[%s]\n", info3->monitorname));
1902 DEBUGADD(20,("defaultdatatype:[%s]\n", info3->defaultdatatype));
1904 for (i=0; info3->dependentfiles &&
1905 *info3->dependentfiles[i]; i++) {
1906 DEBUGADD(20,("dependentfile:[%s]\n",
1907 info3->dependentfiles[i]));
1909 result=0;
1911 break;
1913 default:
1914 DEBUGADD(20,("dump_a_printer_driver: Level %u not implemented\n", (unsigned int)level));
1915 result=1;
1916 break;
1919 return result;
1922 /****************************************************************************
1923 ****************************************************************************/
1924 int pack_devicemode(NT_DEVICEMODE *nt_devmode, char *buf, int buflen)
1926 int len = 0;
1928 len += tdb_pack(buf+len, buflen-len, "p", nt_devmode);
1930 if (!nt_devmode)
1931 return len;
1933 len += tdb_pack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
1934 nt_devmode->devicename,
1935 nt_devmode->formname,
1937 nt_devmode->specversion,
1938 nt_devmode->driverversion,
1939 nt_devmode->size,
1940 nt_devmode->driverextra,
1941 nt_devmode->orientation,
1942 nt_devmode->papersize,
1943 nt_devmode->paperlength,
1944 nt_devmode->paperwidth,
1945 nt_devmode->scale,
1946 nt_devmode->copies,
1947 nt_devmode->defaultsource,
1948 nt_devmode->printquality,
1949 nt_devmode->color,
1950 nt_devmode->duplex,
1951 nt_devmode->yresolution,
1952 nt_devmode->ttoption,
1953 nt_devmode->collate,
1954 nt_devmode->logpixels,
1956 nt_devmode->fields,
1957 nt_devmode->bitsperpel,
1958 nt_devmode->pelswidth,
1959 nt_devmode->pelsheight,
1960 nt_devmode->displayflags,
1961 nt_devmode->displayfrequency,
1962 nt_devmode->icmmethod,
1963 nt_devmode->icmintent,
1964 nt_devmode->mediatype,
1965 nt_devmode->dithertype,
1966 nt_devmode->reserved1,
1967 nt_devmode->reserved2,
1968 nt_devmode->panningwidth,
1969 nt_devmode->panningheight,
1970 nt_devmode->private);
1973 if (nt_devmode->private) {
1974 len += tdb_pack(buf+len, buflen-len, "B",
1975 nt_devmode->driverextra,
1976 nt_devmode->private);
1979 DEBUG(8,("Packed devicemode [%s]\n", nt_devmode->formname));
1981 return len;
1984 /****************************************************************************
1985 Pack all values in all printer keys
1986 ***************************************************************************/
1988 static int pack_values(NT_PRINTER_DATA *data, char *buf, int buflen)
1990 int len = 0;
1991 int i, j;
1992 REGISTRY_VALUE *val;
1993 REGVAL_CTR *val_ctr;
1994 pstring path;
1995 int num_values;
1997 if ( !data )
1998 return 0;
2000 /* loop over all keys */
2002 for ( i=0; i<data->num_keys; i++ ) {
2003 val_ctr = &data->keys[i].values;
2004 num_values = regval_ctr_numvals( val_ctr );
2006 /* loop over all values */
2008 for ( j=0; j<num_values; j++ ) {
2009 /* pathname should be stored as <key>\<value> */
2011 val = regval_ctr_specific_value( val_ctr, j );
2012 pstrcpy( path, data->keys[i].name );
2013 pstrcat( path, "\\" );
2014 pstrcat( path, regval_name(val) );
2016 len += tdb_pack(buf+len, buflen-len, "pPdB",
2017 val,
2018 path,
2019 regval_type(val),
2020 regval_size(val),
2021 regval_data_p(val) );
2026 /* terminator */
2028 len += tdb_pack(buf+len, buflen-len, "p", NULL);
2030 return len;
2034 /****************************************************************************
2035 Delete a printer - this just deletes the printer info file, any open
2036 handles are not affected.
2037 ****************************************************************************/
2039 uint32 del_a_printer(char *sharename)
2041 pstring key;
2042 TDB_DATA kbuf;
2044 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
2046 kbuf.dptr=key;
2047 kbuf.dsize=strlen(key)+1;
2049 tdb_delete(tdb_printers, kbuf);
2050 return 0;
2053 /****************************************************************************
2054 ****************************************************************************/
2055 static WERROR update_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info)
2057 pstring key;
2058 char *buf;
2059 int buflen, len;
2060 WERROR ret;
2061 TDB_DATA kbuf, dbuf;
2064 * in addprinter: no servername and the printer is the name
2065 * in setprinter: servername is \\server
2066 * and printer is \\server\\printer
2068 * Samba manages only local printers.
2069 * we currently don't support things like i
2070 * path=\\other_server\printer
2072 * We only store the printername, not \\server\printername
2075 if ( info->servername[0] != '\0' ) {
2076 trim_string(info->printername, info->servername, NULL);
2077 trim_char(info->printername, '\\', '\0');
2078 info->servername[0]='\0';
2082 * JFM: one day I'll forget.
2083 * below that's info->portname because that's the SAMBA sharename
2084 * and I made NT 'thinks' it's the portname
2085 * the info->sharename is the thing you can name when you add a printer
2086 * that's the short-name when you create shared printer for 95/98
2087 * So I've made a limitation in SAMBA: you can only have 1 printer model
2088 * behind a SAMBA share.
2091 buf = NULL;
2092 buflen = 0;
2094 again:
2095 len = 0;
2096 len += tdb_pack(buf+len, buflen-len, "dddddddddddfffffPfffff",
2097 info->attributes,
2098 info->priority,
2099 info->default_priority,
2100 info->starttime,
2101 info->untiltime,
2102 info->status,
2103 info->cjobs,
2104 info->averageppm,
2105 info->changeid,
2106 info->c_setprinter,
2107 info->setuptime,
2108 info->servername,
2109 info->printername,
2110 info->sharename,
2111 info->portname,
2112 info->drivername,
2113 info->comment,
2114 info->location,
2115 info->sepfile,
2116 info->printprocessor,
2117 info->datatype,
2118 info->parameters);
2120 len += pack_devicemode(info->devmode, buf+len, buflen-len);
2122 len += pack_values( &info->data, buf+len, buflen-len );
2124 if (buflen != len) {
2125 char *tb;
2127 tb = (char *)Realloc(buf, len);
2128 if (!tb) {
2129 DEBUG(0,("update_a_printer_2: failed to enlarge buffer!\n"));
2130 ret = WERR_NOMEM;
2131 goto done;
2133 else buf = tb;
2134 buflen = len;
2135 goto again;
2139 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, info->sharename);
2141 kbuf.dptr = key;
2142 kbuf.dsize = strlen(key)+1;
2143 dbuf.dptr = buf;
2144 dbuf.dsize = len;
2146 ret = (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) == 0? WERR_OK : WERR_NOMEM);
2148 done:
2149 if (!W_ERROR_IS_OK(ret))
2150 DEBUG(8, ("error updating printer to tdb on disk\n"));
2152 SAFE_FREE(buf);
2154 DEBUG(8,("packed printer [%s] with driver [%s] portname=[%s] len=%d\n",
2155 info->sharename, info->drivername, info->portname, len));
2157 return ret;
2161 /****************************************************************************
2162 Malloc and return an NT devicemode.
2163 ****************************************************************************/
2165 NT_DEVICEMODE *construct_nt_devicemode(const fstring default_devicename)
2168 char adevice[MAXDEVICENAME];
2169 NT_DEVICEMODE *nt_devmode = (NT_DEVICEMODE *)malloc(sizeof(NT_DEVICEMODE));
2171 if (nt_devmode == NULL) {
2172 DEBUG(0,("construct_nt_devicemode: malloc fail.\n"));
2173 return NULL;
2176 ZERO_STRUCTP(nt_devmode);
2178 safe_strcpy(adevice, default_devicename, sizeof(adevice)-1);
2179 fstrcpy(nt_devmode->devicename, adevice);
2181 fstrcpy(nt_devmode->formname, "Letter");
2183 nt_devmode->specversion = 0x0401;
2184 nt_devmode->driverversion = 0x0400;
2185 nt_devmode->size = 0x00DC;
2186 nt_devmode->driverextra = 0x0000;
2187 nt_devmode->fields = FORMNAME | TTOPTION | PRINTQUALITY |
2188 DEFAULTSOURCE | COPIES | SCALE |
2189 PAPERSIZE | ORIENTATION;
2190 nt_devmode->orientation = 1;
2191 nt_devmode->papersize = PAPER_LETTER;
2192 nt_devmode->paperlength = 0;
2193 nt_devmode->paperwidth = 0;
2194 nt_devmode->scale = 0x64;
2195 nt_devmode->copies = 1;
2196 nt_devmode->defaultsource = BIN_FORMSOURCE;
2197 nt_devmode->printquality = RES_HIGH; /* 0x0258 */
2198 nt_devmode->color = COLOR_MONOCHROME;
2199 nt_devmode->duplex = DUP_SIMPLEX;
2200 nt_devmode->yresolution = 0;
2201 nt_devmode->ttoption = TT_SUBDEV;
2202 nt_devmode->collate = COLLATE_FALSE;
2203 nt_devmode->icmmethod = 0;
2204 nt_devmode->icmintent = 0;
2205 nt_devmode->mediatype = 0;
2206 nt_devmode->dithertype = 0;
2208 /* non utilisés par un driver d'imprimante */
2209 nt_devmode->logpixels = 0;
2210 nt_devmode->bitsperpel = 0;
2211 nt_devmode->pelswidth = 0;
2212 nt_devmode->pelsheight = 0;
2213 nt_devmode->displayflags = 0;
2214 nt_devmode->displayfrequency = 0;
2215 nt_devmode->reserved1 = 0;
2216 nt_devmode->reserved2 = 0;
2217 nt_devmode->panningwidth = 0;
2218 nt_devmode->panningheight = 0;
2220 nt_devmode->private = NULL;
2221 return nt_devmode;
2224 /****************************************************************************
2225 Deepcopy an NT devicemode.
2226 ****************************************************************************/
2228 NT_DEVICEMODE *dup_nt_devicemode(NT_DEVICEMODE *nt_devicemode)
2230 NT_DEVICEMODE *new_nt_devicemode = NULL;
2232 if ( !nt_devicemode )
2233 return NULL;
2235 if ((new_nt_devicemode = (NT_DEVICEMODE *)memdup(nt_devicemode, sizeof(NT_DEVICEMODE))) == NULL) {
2236 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2237 return NULL;
2240 new_nt_devicemode->private = NULL;
2241 if (nt_devicemode->private != NULL) {
2242 if ((new_nt_devicemode->private = memdup(nt_devicemode->private, nt_devicemode->driverextra)) == NULL) {
2243 SAFE_FREE(new_nt_devicemode);
2244 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2245 return NULL;
2249 return new_nt_devicemode;
2252 /****************************************************************************
2253 Clean up and deallocate a (maybe partially) allocated NT_DEVICEMODE.
2254 ****************************************************************************/
2256 void free_nt_devicemode(NT_DEVICEMODE **devmode_ptr)
2258 NT_DEVICEMODE *nt_devmode = *devmode_ptr;
2260 if(nt_devmode == NULL)
2261 return;
2263 DEBUG(106,("free_nt_devicemode: deleting DEVMODE\n"));
2265 SAFE_FREE(nt_devmode->private);
2266 SAFE_FREE(*devmode_ptr);
2269 /****************************************************************************
2270 Clean up and deallocate a (maybe partially) allocated NT_PRINTER_INFO_LEVEL_2.
2271 ****************************************************************************/
2272 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr)
2274 NT_PRINTER_INFO_LEVEL_2 *info = *info_ptr;
2275 NT_PRINTER_DATA *data;
2276 int i;
2278 if ( !info )
2279 return;
2281 DEBUG(106,("free_nt_printer_info_level_2: deleting info\n"));
2283 free_nt_devicemode(&info->devmode);
2285 /* clean up all registry keys */
2287 data = &info->data;
2288 for ( i=0; i<data->num_keys; i++ ) {
2289 SAFE_FREE( data->keys[i].name );
2290 regval_ctr_destroy( &data->keys[i].values );
2292 SAFE_FREE( data->keys );
2294 /* finally the top level structure */
2296 SAFE_FREE( *info_ptr );
2300 /****************************************************************************
2301 ****************************************************************************/
2302 int unpack_devicemode(NT_DEVICEMODE **nt_devmode, char *buf, int buflen)
2304 int len = 0;
2305 int extra_len = 0;
2306 NT_DEVICEMODE devmode;
2308 ZERO_STRUCT(devmode);
2310 len += tdb_unpack(buf+len, buflen-len, "p", nt_devmode);
2312 if (!*nt_devmode) return len;
2314 len += tdb_unpack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2315 devmode.devicename,
2316 devmode.formname,
2318 &devmode.specversion,
2319 &devmode.driverversion,
2320 &devmode.size,
2321 &devmode.driverextra,
2322 &devmode.orientation,
2323 &devmode.papersize,
2324 &devmode.paperlength,
2325 &devmode.paperwidth,
2326 &devmode.scale,
2327 &devmode.copies,
2328 &devmode.defaultsource,
2329 &devmode.printquality,
2330 &devmode.color,
2331 &devmode.duplex,
2332 &devmode.yresolution,
2333 &devmode.ttoption,
2334 &devmode.collate,
2335 &devmode.logpixels,
2337 &devmode.fields,
2338 &devmode.bitsperpel,
2339 &devmode.pelswidth,
2340 &devmode.pelsheight,
2341 &devmode.displayflags,
2342 &devmode.displayfrequency,
2343 &devmode.icmmethod,
2344 &devmode.icmintent,
2345 &devmode.mediatype,
2346 &devmode.dithertype,
2347 &devmode.reserved1,
2348 &devmode.reserved2,
2349 &devmode.panningwidth,
2350 &devmode.panningheight,
2351 &devmode.private);
2353 if (devmode.private) {
2354 /* the len in tdb_unpack is an int value and
2355 * devmode.driverextra is only a short
2357 len += tdb_unpack(buf+len, buflen-len, "B", &extra_len, &devmode.private);
2358 devmode.driverextra=(uint16)extra_len;
2360 /* check to catch an invalid TDB entry so we don't segfault */
2361 if (devmode.driverextra == 0) {
2362 devmode.private = NULL;
2366 *nt_devmode = (NT_DEVICEMODE *)memdup(&devmode, sizeof(devmode));
2368 DEBUG(8,("Unpacked devicemode [%s](%s)\n", devmode.devicename, devmode.formname));
2369 if (devmode.private)
2370 DEBUG(8,("with a private section of %d bytes\n", devmode.driverextra));
2372 return len;
2375 /****************************************************************************
2376 Allocate and initialize a new slot.
2377 ***************************************************************************/
2379 static int add_new_printer_key( NT_PRINTER_DATA *data, const char *name )
2381 NT_PRINTER_KEY *d;
2382 int key_index;
2384 if ( !data || !name )
2385 return -1;
2387 /* allocate another slot in the NT_PRINTER_KEY array */
2389 d = Realloc( data->keys, sizeof(NT_PRINTER_KEY)*(data->num_keys+1) );
2390 if ( d )
2391 data->keys = d;
2393 key_index = data->num_keys;
2395 /* initialze new key */
2397 data->num_keys++;
2398 data->keys[key_index].name = strdup( name );
2400 ZERO_STRUCTP( &data->keys[key_index].values );
2402 regval_ctr_init( &data->keys[key_index].values );
2404 DEBUG(10,("add_new_printer_key: Inserted new data key [%s]\n", name ));
2406 return key_index;
2409 /****************************************************************************
2410 search for a registry key name in the existing printer data
2411 ***************************************************************************/
2413 int lookup_printerkey( NT_PRINTER_DATA *data, const char *name )
2415 int key_index = -1;
2416 int i;
2418 if ( !data || !name )
2419 return -1;
2421 DEBUG(12,("lookup_printerkey: Looking for [%s]\n", name));
2423 /* loop over all existing keys */
2425 for ( i=0; i<data->num_keys; i++ ) {
2426 if ( strequal(data->keys[i].name, name) ) {
2427 DEBUG(12,("lookup_printerkey: Found [%s]!\n", name));
2428 key_index = i;
2429 break;
2434 return key_index;
2437 /****************************************************************************
2438 ***************************************************************************/
2440 uint32 get_printer_subkeys( NT_PRINTER_DATA *data, const char* key, fstring **subkeys )
2442 int i, j;
2443 int key_len;
2444 int num_subkeys = 0;
2445 char *p;
2446 fstring *ptr, *subkeys_ptr = NULL;
2447 fstring subkeyname;
2449 if ( !data )
2450 return 0;
2452 for ( i=0; i<data->num_keys; i++ ) {
2453 if ( StrnCaseCmp(data->keys[i].name, key, strlen(key)) == 0 ) {
2454 /* match sure it is a subkey and not the key itself */
2456 key_len = strlen( key );
2457 if ( strlen(data->keys[i].name) == key_len )
2458 continue;
2460 /* get subkey path */
2462 p = data->keys[i].name + key_len;
2463 if ( *p == '\\' )
2464 p++;
2465 fstrcpy( subkeyname, p );
2466 if ( (p = strchr( subkeyname, '\\' )) )
2467 *p = '\0';
2469 /* don't add a key more than once */
2471 for ( j=0; j<num_subkeys; j++ ) {
2472 if ( strequal( subkeys_ptr[j], subkeyname ) )
2473 break;
2476 if ( j != num_subkeys )
2477 continue;
2479 /* found a match, so allocate space and copy the name */
2481 if ( !(ptr = Realloc( subkeys_ptr, (num_subkeys+2)*sizeof(fstring))) ) {
2482 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
2483 num_subkeys+1));
2484 SAFE_FREE( subkeys );
2485 return 0;
2488 subkeys_ptr = ptr;
2489 fstrcpy( subkeys_ptr[num_subkeys], subkeyname );
2490 num_subkeys++;
2495 /* tag of the end */
2497 if (num_subkeys)
2498 fstrcpy(subkeys_ptr[num_subkeys], "" );
2500 *subkeys = subkeys_ptr;
2502 return num_subkeys;
2505 #ifdef HAVE_ADS
2506 static void map_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2507 const char *sz)
2509 smb_ucs2_t conv_str[1024];
2510 size_t str_size;
2512 regval_ctr_delvalue(ctr, val_name);
2513 str_size = push_ucs2(NULL, conv_str, sz, sizeof(conv_str),
2514 STR_TERMINATE | STR_NOALIGN);
2515 regval_ctr_addvalue(ctr, val_name, REG_SZ,
2516 (char *) conv_str, str_size);
2519 static void map_dword_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2520 uint32 dword)
2522 regval_ctr_delvalue(ctr, val_name);
2523 regval_ctr_addvalue(ctr, val_name, REG_DWORD,
2524 (char *) &dword, sizeof(dword));
2527 static void map_bool_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2528 BOOL b)
2530 uint8 bin_bool = (b ? 1 : 0);
2531 regval_ctr_delvalue(ctr, val_name);
2532 regval_ctr_addvalue(ctr, val_name, REG_BINARY,
2533 (char *) &bin_bool, sizeof(bin_bool));
2536 static void map_single_multi_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
2537 const char *multi_sz)
2539 smb_ucs2_t *conv_strs = NULL;
2540 size_t str_size;
2542 /* a multi-sz has to have a null string terminator, i.e., the last
2543 string must be followed by two nulls */
2544 str_size = (strlen(multi_sz) + 2) * sizeof(smb_ucs2_t);
2545 conv_strs = calloc(str_size, 1);
2547 push_ucs2(NULL, conv_strs, multi_sz, str_size,
2548 STR_TERMINATE | STR_NOALIGN);
2550 regval_ctr_delvalue(ctr, val_name);
2551 regval_ctr_addvalue(ctr, val_name, REG_MULTI_SZ,
2552 (char *) conv_strs, str_size);
2553 safe_free(conv_strs);
2557 /****************************************************************************
2558 * Map the NT_PRINTER_INFO_LEVEL_2 data into DsSpooler keys for publishing.
2560 * @param info2 NT_PRINTER_INFO_LEVEL_2 describing printer - gets modified
2561 * @return BOOL indicating success or failure
2562 ***************************************************************************/
2564 static BOOL map_nt_printer_info2_to_dsspooler(NT_PRINTER_INFO_LEVEL_2 *info2)
2566 REGVAL_CTR *ctr = NULL;
2567 fstring longname;
2568 char *allocated_string = NULL;
2569 const char *ascii_str;
2570 int i;
2572 if ((i = lookup_printerkey(&info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
2573 i = add_new_printer_key(&info2->data, SPOOL_DSSPOOLER_KEY);
2574 ctr = &info2->data.keys[i].values;
2576 map_sz_into_ctr(ctr, SPOOL_REG_PRINTERNAME, info2->sharename);
2577 map_sz_into_ctr(ctr, SPOOL_REG_SHORTSERVERNAME, global_myname());
2579 get_mydnsfullname(longname);
2580 map_sz_into_ctr(ctr, SPOOL_REG_SERVERNAME, longname);
2582 asprintf(&allocated_string, "\\\\%s\\%s", longname, info2->sharename);
2583 map_sz_into_ctr(ctr, SPOOL_REG_UNCNAME, allocated_string);
2584 SAFE_FREE(allocated_string);
2586 map_dword_into_ctr(ctr, SPOOL_REG_VERSIONNUMBER, 4);
2587 map_sz_into_ctr(ctr, SPOOL_REG_DRIVERNAME, info2->drivername);
2588 map_sz_into_ctr(ctr, SPOOL_REG_LOCATION, info2->location);
2589 map_sz_into_ctr(ctr, SPOOL_REG_DESCRIPTION, info2->comment);
2590 map_single_multi_sz_into_ctr(ctr, SPOOL_REG_PORTNAME, info2->portname);
2591 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSEPARATORFILE, info2->sepfile);
2592 map_dword_into_ctr(ctr, SPOOL_REG_PRINTSTARTTIME, info2->starttime);
2593 map_dword_into_ctr(ctr, SPOOL_REG_PRINTENDTIME, info2->untiltime);
2594 map_dword_into_ctr(ctr, SPOOL_REG_PRIORITY, info2->priority);
2596 map_bool_into_ctr(ctr, SPOOL_REG_PRINTKEEPPRINTEDJOBS,
2597 (info2->attributes &
2598 PRINTER_ATTRIBUTE_KEEPPRINTEDJOBS));
2600 switch (info2->attributes & 0x3) {
2601 case 0:
2602 ascii_str = SPOOL_REGVAL_PRINTWHILESPOOLING;
2603 break;
2604 case 1:
2605 ascii_str = SPOOL_REGVAL_PRINTAFTERSPOOLED;
2606 break;
2607 case 2:
2608 ascii_str = SPOOL_REGVAL_PRINTDIRECT;
2609 break;
2610 default:
2611 ascii_str = "unknown";
2613 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSPOOLING, ascii_str);
2615 return True;
2618 static void store_printer_guid(NT_PRINTER_INFO_LEVEL_2 *info2,
2619 struct uuid guid)
2621 int i;
2622 REGVAL_CTR *ctr=NULL;
2624 /* find the DsSpooler key */
2625 if ((i = lookup_printerkey(&info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
2626 i = add_new_printer_key(&info2->data, SPOOL_DSSPOOLER_KEY);
2627 ctr = &info2->data.keys[i].values;
2629 regval_ctr_delvalue(ctr, "objectGUID");
2630 regval_ctr_addvalue(ctr, "objectGUID", REG_BINARY,
2631 (char *) &guid, sizeof(struct uuid));
2634 static WERROR publish_it(NT_PRINTER_INFO_LEVEL *printer)
2636 ADS_STATUS ads_rc;
2637 TALLOC_CTX *ctx = talloc_init("publish_it");
2638 ADS_MODLIST mods = ads_init_mods(ctx);
2639 char *prt_dn = NULL, *srv_dn, *srv_cn_0;
2640 char *srv_dn_utf8, **srv_cn_utf8;
2641 void *res = NULL;
2642 ADS_STRUCT *ads;
2643 const char *attrs[] = {"objectGUID", NULL};
2644 struct uuid guid;
2645 WERROR win_rc = WERR_OK;
2647 ZERO_STRUCT(guid);
2648 /* set the DsSpooler info and attributes */
2649 if (!(map_nt_printer_info2_to_dsspooler(printer->info_2)))
2650 return WERR_NOMEM;
2651 printer->info_2->attributes |= PRINTER_ATTRIBUTE_PUBLISHED;
2652 win_rc = mod_a_printer(*printer, 2);
2653 if (!W_ERROR_IS_OK(win_rc)) {
2654 DEBUG(3, ("err %d saving data\n",
2655 W_ERROR_V(win_rc)));
2656 return win_rc;
2659 /* Build the ads mods */
2660 get_local_printer_publishing_data(ctx, &mods,
2661 &printer->info_2->data);
2662 ads_mod_str(ctx, &mods, SPOOL_REG_PRINTERNAME,
2663 printer->info_2->sharename);
2665 /* initial ads structure */
2667 ads = ads_init(NULL, NULL, NULL);
2668 if (!ads) {
2669 DEBUG(3, ("ads_init() failed\n"));
2670 return WERR_SERVER_UNAVAILABLE;
2672 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
2673 SAFE_FREE(ads->auth.password);
2674 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
2675 NULL, NULL);
2677 /* ads_connect() will find the DC for us */
2678 ads_rc = ads_connect(ads);
2679 if (!ADS_ERR_OK(ads_rc)) {
2680 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
2681 ads_destroy(&ads);
2682 return WERR_ACCESS_DENIED;
2685 /* figure out where to publish */
2686 ads_find_machine_acct(ads, &res, global_myname());
2688 /* We use ldap_get_dn here as we need the answer
2689 * in utf8 to call ldap_explode_dn(). JRA. */
2691 srv_dn_utf8 = ldap_get_dn(ads->ld, res);
2692 if (!srv_dn_utf8) {
2693 ads_destroy(&ads);
2694 return WERR_SERVER_UNAVAILABLE;
2696 ads_msgfree(ads, res);
2697 srv_cn_utf8 = ldap_explode_dn(srv_dn_utf8, 1);
2698 if (!srv_cn_utf8) {
2699 ldap_memfree(srv_dn_utf8);
2700 ads_destroy(&ads);
2701 return WERR_SERVER_UNAVAILABLE;
2703 /* Now convert to CH_UNIX. */
2704 if (pull_utf8_allocate(&srv_dn, srv_dn_utf8) == (size_t)-1) {
2705 ldap_memfree(srv_dn_utf8);
2706 ldap_memfree(srv_cn_utf8);
2707 ads_destroy(&ads);
2708 return WERR_SERVER_UNAVAILABLE;
2710 if (pull_utf8_allocate(&srv_cn_0, srv_cn_utf8[0]) == (size_t)-1) {
2711 ldap_memfree(srv_dn_utf8);
2712 ldap_memfree(srv_cn_utf8);
2713 ads_destroy(&ads);
2714 SAFE_FREE(srv_dn);
2715 return WERR_SERVER_UNAVAILABLE;
2718 ldap_memfree(srv_dn_utf8);
2719 ldap_memfree(srv_cn_utf8);
2721 asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn_0,
2722 printer->info_2->sharename, srv_dn);
2724 SAFE_FREE(srv_dn);
2725 SAFE_FREE(srv_cn_0);
2727 /* publish it */
2728 ads_rc = ads_add_printer_entry(ads, prt_dn, ctx, &mods);
2729 if (LDAP_ALREADY_EXISTS == ads_rc.err.rc)
2730 ads_rc = ads_mod_printer_entry(ads, prt_dn, ctx,&mods);
2732 /* retreive the guid and store it locally */
2733 if (ADS_ERR_OK(ads_search_dn(ads, &res, prt_dn, attrs))) {
2734 ads_pull_guid(ads, res, &guid);
2735 ads_msgfree(ads, res);
2736 store_printer_guid(printer->info_2, guid);
2737 win_rc = mod_a_printer(*printer, 2);
2740 SAFE_FREE(prt_dn);
2741 ads_destroy(&ads);
2743 return WERR_OK;
2746 WERROR unpublish_it(NT_PRINTER_INFO_LEVEL *printer)
2748 ADS_STATUS ads_rc;
2749 ADS_STRUCT *ads;
2750 void *res;
2751 char *prt_dn = NULL;
2752 WERROR win_rc;
2754 printer->info_2->attributes ^= PRINTER_ATTRIBUTE_PUBLISHED;
2755 win_rc = mod_a_printer(*printer, 2);
2756 if (!W_ERROR_IS_OK(win_rc)) {
2757 DEBUG(3, ("err %d saving data\n",
2758 W_ERROR_V(win_rc)));
2759 return win_rc;
2762 ads = ads_init(NULL, NULL, NULL);
2763 if (!ads) {
2764 DEBUG(3, ("ads_init() failed\n"));
2765 return WERR_SERVER_UNAVAILABLE;
2767 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
2768 SAFE_FREE(ads->auth.password);
2769 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
2770 NULL, NULL);
2772 /* ads_connect() will find the DC for us */
2773 ads_rc = ads_connect(ads);
2774 if (!ADS_ERR_OK(ads_rc)) {
2775 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
2776 ads_destroy(&ads);
2777 return WERR_ACCESS_DENIED;
2780 /* remove the printer from the directory */
2781 ads_rc = ads_find_printer_on_server(ads, &res,
2782 printer->info_2->sharename, global_myname());
2783 if (ADS_ERR_OK(ads_rc) && ads_count_replies(ads, res)) {
2784 prt_dn = ads_get_dn(ads, res);
2785 ads_msgfree(ads, res);
2786 ads_rc = ads_del_dn(ads, prt_dn);
2787 ads_memfree(ads, prt_dn);
2790 ads_destroy(&ads);
2791 return WERR_OK;
2794 /****************************************************************************
2795 * Publish a printer in the directory
2797 * @param snum describing printer service
2798 * @return WERROR indicating status of publishing
2799 ***************************************************************************/
2801 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
2803 NT_PRINTER_INFO_LEVEL *printer = NULL;
2804 WERROR win_rc;
2806 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
2807 if (!W_ERROR_IS_OK(win_rc))
2808 return win_rc;
2810 switch(action) {
2811 case SPOOL_DS_PUBLISH:
2812 case SPOOL_DS_UPDATE:
2813 win_rc = publish_it(printer);
2814 break;
2815 case SPOOL_DS_UNPUBLISH:
2816 win_rc = unpublish_it(printer);
2817 break;
2818 default:
2819 win_rc = WERR_NOT_SUPPORTED;
2823 free_a_printer(&printer, 2);
2824 return win_rc;
2827 BOOL is_printer_published(Printer_entry *print_hnd, int snum,
2828 struct uuid *guid)
2830 NT_PRINTER_INFO_LEVEL *printer = NULL;
2831 REGVAL_CTR *ctr;
2832 REGISTRY_VALUE *guid_val;
2833 WERROR win_rc;
2834 int i;
2837 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
2838 if (!W_ERROR_IS_OK(win_rc))
2839 return False;
2841 if (!(printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED))
2842 return False;
2844 if ((i = lookup_printerkey(&printer->info_2->data,
2845 SPOOL_DSSPOOLER_KEY)) < 0)
2846 return False;
2848 if (!(ctr = &printer->info_2->data.keys[i].values)) {
2849 return False;
2852 if (!(guid_val = regval_ctr_getvalue(ctr, "objectGUID"))) {
2853 return False;
2856 if (regval_size(guid_val) == sizeof(struct uuid))
2857 memcpy(guid, regval_data_p(guid_val), sizeof(struct uuid));
2859 return True;
2862 #else
2863 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
2865 return WERR_OK;
2867 BOOL is_printer_published(Printer_entry *print_hnd, int snum,
2868 struct uuid *guid)
2870 return False;
2872 #endif
2873 /****************************************************************************
2874 ***************************************************************************/
2876 WERROR delete_all_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key )
2878 NT_PRINTER_DATA *data;
2879 int i;
2880 int removed_keys = 0;
2881 int empty_slot;
2883 data = &p2->data;
2884 empty_slot = data->num_keys;
2886 if ( !key )
2887 return WERR_INVALID_PARAM;
2889 /* remove all keys */
2891 if ( !strlen(key) ) {
2892 for ( i=0; i<data->num_keys; i++ ) {
2893 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
2894 data->keys[i].name));
2896 SAFE_FREE( data->keys[i].name );
2897 regval_ctr_destroy( &data->keys[i].values );
2900 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from printer [%s]\n",
2901 p2->printername ));
2903 SAFE_FREE( data->keys );
2904 ZERO_STRUCTP( data );
2906 return WERR_OK;
2909 /* remove a specific key (and all subkeys) */
2911 for ( i=0; i<data->num_keys; i++ ) {
2912 if ( StrnCaseCmp( data->keys[i].name, key, strlen(key)) == 0 ) {
2913 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
2914 data->keys[i].name));
2916 SAFE_FREE( data->keys[i].name );
2917 regval_ctr_destroy( &data->keys[i].values );
2919 /* mark the slot as empty */
2921 ZERO_STRUCTP( &data->keys[i] );
2925 /* find the first empty slot */
2927 for ( i=0; i<data->num_keys; i++ ) {
2928 if ( !data->keys[i].name ) {
2929 empty_slot = i;
2930 removed_keys++;
2931 break;
2935 if ( i == data->num_keys )
2936 /* nothing was removed */
2937 return WERR_INVALID_PARAM;
2939 /* move everything down */
2941 for ( i=empty_slot+1; i<data->num_keys; i++ ) {
2942 if ( data->keys[i].name ) {
2943 memcpy( &data->keys[empty_slot], &data->keys[i], sizeof(NT_PRINTER_KEY) );
2944 ZERO_STRUCTP( &data->keys[i] );
2945 empty_slot++;
2946 removed_keys++;
2950 /* update count */
2952 data->num_keys -= removed_keys;
2954 /* sanity check to see if anything is left */
2956 if ( !data->num_keys ) {
2957 DEBUG(8,("delete_all_printer_data: No keys left for printer [%s]\n", p2->printername ));
2959 SAFE_FREE( data->keys );
2960 ZERO_STRUCTP( data );
2963 return WERR_OK;
2966 /****************************************************************************
2967 ***************************************************************************/
2969 WERROR delete_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
2971 WERROR result = WERR_OK;
2972 int key_index;
2974 /* we must have names on non-zero length */
2976 if ( !key || !*key|| !value || !*value )
2977 return WERR_INVALID_NAME;
2979 /* find the printer key first */
2981 key_index = lookup_printerkey( &p2->data, key );
2982 if ( key_index == -1 )
2983 return WERR_OK;
2985 /* make sure the value exists so we can return the correct error code */
2987 if ( !regval_ctr_getvalue( &p2->data.keys[key_index].values, value ) )
2988 return WERR_BADFILE;
2990 regval_ctr_delvalue( &p2->data.keys[key_index].values, value );
2992 DEBUG(8,("delete_printer_data: Removed key => [%s], value => [%s]\n",
2993 key, value ));
2995 return result;
2998 /****************************************************************************
2999 ***************************************************************************/
3001 WERROR add_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value,
3002 uint32 type, uint8 *data, int real_len )
3004 WERROR result = WERR_OK;
3005 int key_index;
3007 /* we must have names on non-zero length */
3009 if ( !key || !*key|| !value || !*value )
3010 return WERR_INVALID_NAME;
3012 /* find the printer key first */
3014 key_index = lookup_printerkey( &p2->data, key );
3015 if ( key_index == -1 )
3016 key_index = add_new_printer_key( &p2->data, key );
3018 if ( key_index == -1 )
3019 return WERR_NOMEM;
3021 regval_ctr_addvalue( &p2->data.keys[key_index].values, value,
3022 type, (const char *)data, real_len );
3024 DEBUG(8,("add_printer_data: Added key => [%s], value => [%s], type=> [%d], size => [%d]\n",
3025 key, value, type, real_len ));
3027 return result;
3030 /****************************************************************************
3031 ***************************************************************************/
3033 REGISTRY_VALUE* get_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3035 int key_index;
3037 if ( (key_index = lookup_printerkey( &p2->data, key )) == -1 )
3038 return NULL;
3040 DEBUG(8,("get_printer_data: Attempting to lookup key => [%s], value => [%s]\n",
3041 key, value ));
3043 return regval_ctr_getvalue( &p2->data.keys[key_index].values, value );
3046 /****************************************************************************
3047 Unpack a list of registry values frem the TDB
3048 ***************************************************************************/
3050 static int unpack_values(NT_PRINTER_DATA *printer_data, char *buf, int buflen)
3052 int len = 0;
3053 uint32 type;
3054 pstring string, valuename, keyname;
3055 char *str;
3056 int size;
3057 uint8 *data_p;
3058 REGISTRY_VALUE *regval_p;
3059 int key_index;
3061 /* add the "PrinterDriverData" key first for performance reasons */
3063 add_new_printer_key( printer_data, SPOOL_PRINTERDATA_KEY );
3065 /* loop and unpack the rest of the registry values */
3067 while ( True ) {
3069 /* check to see if there are any more registry values */
3071 regval_p = NULL;
3072 len += tdb_unpack(buf+len, buflen-len, "p", &regval_p);
3073 if ( !regval_p )
3074 break;
3076 /* unpack the next regval */
3078 len += tdb_unpack(buf+len, buflen-len, "fdB",
3079 string,
3080 &type,
3081 &size,
3082 &data_p);
3085 * break of the keyname from the value name.
3086 * Valuenames can have embedded '\'s so be careful.
3087 * only support one level of keys. See the
3088 * "Konica Fiery S300 50C-K v1.1. enu" 2k driver.
3089 * -- jerry
3092 str = strchr_m( string, '\\');
3094 /* Put in "PrinterDriverData" is no key specified */
3096 if ( !str ) {
3097 pstrcpy( keyname, SPOOL_PRINTERDATA_KEY );
3098 pstrcpy( valuename, string );
3100 else {
3101 *str = '\0';
3102 pstrcpy( keyname, string );
3103 pstrcpy( valuename, str+1 );
3106 /* see if we need a new key */
3108 if ( (key_index=lookup_printerkey( printer_data, keyname )) == -1 )
3109 key_index = add_new_printer_key( printer_data, keyname );
3111 if ( key_index == -1 ) {
3112 DEBUG(0,("unpack_values: Failed to allocate a new key [%s]!\n",
3113 keyname));
3114 break;
3117 /* add the new value */
3119 regval_ctr_addvalue( &printer_data->keys[key_index].values, valuename, type, (const char *)data_p, size );
3121 SAFE_FREE(data_p); /* 'B' option to tdbpack does a malloc() */
3123 DEBUG(8,("specific: [%s:%s], len: %d\n", keyname, valuename, size));
3126 return len;
3129 /****************************************************************************
3130 ***************************************************************************/
3132 static void map_to_os2_driver(fstring drivername)
3134 static BOOL initialised=False;
3135 static fstring last_from,last_to;
3136 char *mapfile = lp_os2_driver_map();
3137 char **lines = NULL;
3138 int numlines = 0;
3139 int i;
3141 if (!strlen(drivername))
3142 return;
3144 if (!*mapfile)
3145 return;
3147 if (!initialised) {
3148 *last_from = *last_to = 0;
3149 initialised = True;
3152 if (strequal(drivername,last_from)) {
3153 DEBUG(3,("Mapped Windows driver %s to OS/2 driver %s\n",drivername,last_to));
3154 fstrcpy(drivername,last_to);
3155 return;
3158 lines = file_lines_load(mapfile, &numlines);
3159 if (numlines == 0) {
3160 DEBUG(0,("No entries in OS/2 driver map %s\n",mapfile));
3161 return;
3164 DEBUG(4,("Scanning OS/2 driver map %s\n",mapfile));
3166 for( i = 0; i < numlines; i++) {
3167 char *nt_name = lines[i];
3168 char *os2_name = strchr(nt_name,'=');
3170 if (!os2_name)
3171 continue;
3173 *os2_name++ = 0;
3175 while (isspace(*nt_name))
3176 nt_name++;
3178 if (!*nt_name || strchr("#;",*nt_name))
3179 continue;
3182 int l = strlen(nt_name);
3183 while (l && isspace(nt_name[l-1])) {
3184 nt_name[l-1] = 0;
3185 l--;
3189 while (isspace(*os2_name))
3190 os2_name++;
3193 int l = strlen(os2_name);
3194 while (l && isspace(os2_name[l-1])) {
3195 os2_name[l-1] = 0;
3196 l--;
3200 if (strequal(nt_name,drivername)) {
3201 DEBUG(3,("Mapped windows driver %s to os2 driver%s\n",drivername,os2_name));
3202 fstrcpy(last_from,drivername);
3203 fstrcpy(last_to,os2_name);
3204 fstrcpy(drivername,os2_name);
3205 file_lines_free(lines);
3206 return;
3210 file_lines_free(lines);
3213 /****************************************************************************
3214 Get a default printer info 2 struct.
3215 ****************************************************************************/
3216 static WERROR get_a_printer_2_default(NT_PRINTER_INFO_LEVEL_2 **info_ptr, const char *sharename)
3218 int snum;
3219 NT_PRINTER_INFO_LEVEL_2 info;
3221 ZERO_STRUCT(info);
3223 snum = lp_servicenumber(sharename);
3225 slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", get_called_name());
3226 slprintf(info.printername, sizeof(info.printername)-1, "\\\\%s\\%s",
3227 get_called_name(), sharename);
3228 fstrcpy(info.sharename, sharename);
3229 fstrcpy(info.portname, SAMBA_PRINTER_PORT_NAME);
3231 /* by setting the driver name to an empty string, a local NT admin
3232 can now run the **local** APW to install a local printer driver
3233 for a Samba shared printer in 2.2. Without this, drivers **must** be
3234 installed on the Samba server for NT clients --jerry */
3235 #if 0 /* JERRY --do not uncomment-- */
3236 if (!*info.drivername)
3237 fstrcpy(info.drivername, "NO DRIVER AVAILABLE FOR THIS PRINTER");
3238 #endif
3241 DEBUG(10,("get_a_printer_2_default: driver name set to [%s]\n", info.drivername));
3243 pstrcpy(info.comment, "");
3244 fstrcpy(info.printprocessor, "winprint");
3245 fstrcpy(info.datatype, "RAW");
3247 info.attributes = PRINTER_ATTRIBUTE_SAMBA;
3249 info.starttime = 0; /* Minutes since 12:00am GMT */
3250 info.untiltime = 0; /* Minutes since 12:00am GMT */
3251 info.priority = 1;
3252 info.default_priority = 1;
3253 info.setuptime = (uint32)time(NULL);
3256 * I changed this as I think it is better to have a generic
3257 * DEVMODE than to crash Win2k explorer.exe --jerry
3258 * See the HP Deskjet 990c Win2k drivers for an example.
3260 * However the default devmode appears to cause problems
3261 * with the HP CLJ 8500 PCL driver. Hence the addition of
3262 * the "default devmode" parameter --jerry 22/01/2002
3265 if (lp_default_devmode(snum)) {
3266 if ((info.devmode = construct_nt_devicemode(info.printername)) == NULL)
3267 goto fail;
3269 else {
3270 info.devmode = NULL;
3273 /* This will get the current RPC talloc context, but we should be
3274 passing this as a parameter... fixme... JRA ! */
3276 if (!nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf))
3277 goto fail;
3279 *info_ptr = (NT_PRINTER_INFO_LEVEL_2 *)memdup(&info, sizeof(info));
3280 if (! *info_ptr) {
3281 DEBUG(0,("get_a_printer_2_default: malloc fail.\n"));
3282 goto fail;
3285 return WERR_OK;
3287 fail:
3288 if (info.devmode)
3289 free_nt_devicemode(&info.devmode);
3290 return WERR_ACCESS_DENIED;
3293 /****************************************************************************
3294 ****************************************************************************/
3295 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr, const char *sharename)
3297 pstring key;
3298 NT_PRINTER_INFO_LEVEL_2 info;
3299 int len = 0;
3300 TDB_DATA kbuf, dbuf;
3301 fstring printername;
3302 char adevice[MAXDEVICENAME];
3304 ZERO_STRUCT(info);
3306 slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
3308 kbuf.dptr = key;
3309 kbuf.dsize = strlen(key)+1;
3311 dbuf = tdb_fetch(tdb_printers, kbuf);
3312 if (!dbuf.dptr)
3313 return get_a_printer_2_default(info_ptr, sharename);
3315 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "dddddddddddfffffPfffff",
3316 &info.attributes,
3317 &info.priority,
3318 &info.default_priority,
3319 &info.starttime,
3320 &info.untiltime,
3321 &info.status,
3322 &info.cjobs,
3323 &info.averageppm,
3324 &info.changeid,
3325 &info.c_setprinter,
3326 &info.setuptime,
3327 info.servername,
3328 info.printername,
3329 info.sharename,
3330 info.portname,
3331 info.drivername,
3332 info.comment,
3333 info.location,
3334 info.sepfile,
3335 info.printprocessor,
3336 info.datatype,
3337 info.parameters);
3339 /* Samba has to have shared raw drivers. */
3340 info.attributes = PRINTER_ATTRIBUTE_SAMBA;
3342 /* Restore the stripped strings. */
3343 slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", get_called_name());
3344 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", get_called_name(),
3345 info.printername);
3346 fstrcpy(info.printername, printername);
3348 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
3351 * Some client drivers freak out if there is a NULL devmode
3352 * (probably the driver is not checking before accessing
3353 * the devmode pointer) --jerry
3355 * See comments in get_a_printer_2_default()
3358 if (lp_default_devmode(lp_servicenumber(sharename)) && !info.devmode) {
3359 DEBUG(8,("get_a_printer_2: Constructing a default device mode for [%s]\n",
3360 printername));
3361 info.devmode = construct_nt_devicemode(printername);
3364 safe_strcpy(adevice, info.printername, sizeof(adevice)-1);
3365 if (info.devmode) {
3366 fstrcpy(info.devmode->devicename, adevice);
3369 len += unpack_values( &info.data, dbuf.dptr+len, dbuf.dsize-len );
3371 /* This will get the current RPC talloc context, but we should be
3372 passing this as a parameter... fixme... JRA ! */
3374 nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf);
3376 /* Fix for OS/2 drivers. */
3378 if (get_remote_arch() == RA_OS2)
3379 map_to_os2_driver(info.drivername);
3381 SAFE_FREE(dbuf.dptr);
3382 *info_ptr=memdup(&info, sizeof(info));
3384 DEBUG(9,("Unpacked printer [%s] name [%s] running driver [%s]\n",
3385 sharename, info.printername, info.drivername));
3387 return WERR_OK;
3390 /****************************************************************************
3391 Debugging function, dump at level 6 the struct in the logs.
3392 ****************************************************************************/
3393 static uint32 dump_a_printer(NT_PRINTER_INFO_LEVEL printer, uint32 level)
3395 uint32 result;
3396 NT_PRINTER_INFO_LEVEL_2 *info2;
3398 DEBUG(106,("Dumping printer at level [%d]\n", level));
3400 switch (level) {
3401 case 2:
3403 if (printer.info_2 == NULL)
3404 result=5;
3405 else
3407 info2=printer.info_2;
3409 DEBUGADD(106,("attributes:[%d]\n", info2->attributes));
3410 DEBUGADD(106,("priority:[%d]\n", info2->priority));
3411 DEBUGADD(106,("default_priority:[%d]\n", info2->default_priority));
3412 DEBUGADD(106,("starttime:[%d]\n", info2->starttime));
3413 DEBUGADD(106,("untiltime:[%d]\n", info2->untiltime));
3414 DEBUGADD(106,("status:[%d]\n", info2->status));
3415 DEBUGADD(106,("cjobs:[%d]\n", info2->cjobs));
3416 DEBUGADD(106,("averageppm:[%d]\n", info2->averageppm));
3417 DEBUGADD(106,("changeid:[%d]\n", info2->changeid));
3418 DEBUGADD(106,("c_setprinter:[%d]\n", info2->c_setprinter));
3419 DEBUGADD(106,("setuptime:[%d]\n", info2->setuptime));
3421 DEBUGADD(106,("servername:[%s]\n", info2->servername));
3422 DEBUGADD(106,("printername:[%s]\n", info2->printername));
3423 DEBUGADD(106,("sharename:[%s]\n", info2->sharename));
3424 DEBUGADD(106,("portname:[%s]\n", info2->portname));
3425 DEBUGADD(106,("drivername:[%s]\n", info2->drivername));
3426 DEBUGADD(106,("comment:[%s]\n", info2->comment));
3427 DEBUGADD(106,("location:[%s]\n", info2->location));
3428 DEBUGADD(106,("sepfile:[%s]\n", info2->sepfile));
3429 DEBUGADD(106,("printprocessor:[%s]\n", info2->printprocessor));
3430 DEBUGADD(106,("datatype:[%s]\n", info2->datatype));
3431 DEBUGADD(106,("parameters:[%s]\n", info2->parameters));
3432 result=0;
3434 break;
3436 default:
3437 DEBUGADD(106,("dump_a_printer: Level %u not implemented\n", (unsigned int)level ));
3438 result=1;
3439 break;
3442 return result;
3445 /****************************************************************************
3446 Update the changeid time.
3447 This is SO NASTY as some drivers need this to change, others need it
3448 static. This value will change every second, and I must hope that this
3449 is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
3450 UTAH ! JRA.
3451 ****************************************************************************/
3453 static uint32 rev_changeid(void)
3455 struct timeval tv;
3457 get_process_uptime(&tv);
3459 #if 1 /* JERRY */
3460 /* Return changeid as msec since spooler restart */
3461 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
3462 #else
3464 * This setting seems to work well but is too untested
3465 * to replace the above calculation. Left in for experiementation
3466 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
3468 return tv.tv_sec * 10 + tv.tv_usec / 100000;
3469 #endif
3472 /********************************************************************
3473 Send a message to all smbds about the printer that just changed
3474 ********************************************************************/
3476 static BOOL send_printer_mod_msg( char* printername )
3478 int len = strlen(printername);
3480 if (!len)
3481 return False;
3483 DEBUG(10,("send_printer_mod_msg: Sending message about printer change [%s]\n",
3484 printername));
3486 /* spam everyone that we just changed this printer */
3488 message_send_all( conn_tdb_ctx(), MSG_PRINTER_MOD, printername, len+1, False, NULL );
3490 return True;
3494 * The function below are the high level ones.
3495 * only those ones must be called from the spoolss code.
3496 * JFM.
3499 /****************************************************************************
3500 Modify a printer. This is called from SETPRINTERDATA/DELETEPRINTERDATA.
3501 ****************************************************************************/
3503 WERROR mod_a_printer(NT_PRINTER_INFO_LEVEL printer, uint32 level)
3505 WERROR result;
3507 dump_a_printer(printer, level);
3510 * invalidate cache for all open handles to this printer.
3511 * cache for a given handle will be updated on the next
3512 * get_a_printer()
3515 invalidate_printer_hnd_cache( printer.info_2->sharename );
3516 send_printer_mod_msg( printer.info_2->sharename );
3518 switch (level) {
3519 case 2:
3522 * Update the changestamp. Emperical tests show that the
3523 * ChangeID is always updated,but c_setprinter is
3524 * global spooler variable (not per printer).
3527 /* ChangeID **must** be increasing over the lifetime
3528 of client's spoolss service in order for the
3529 client's cache to show updates */
3531 printer.info_2->changeid = rev_changeid();
3534 * Because one day someone will ask:
3535 * NT->NT An admin connection to a remote
3536 * printer show changes imeediately in
3537 * the properities dialog
3539 * A non-admin connection will only show the
3540 * changes after viewing the properites page
3541 * 2 times. Seems to be related to a
3542 * race condition in the client between the spooler
3543 * updating the local cache and the Explorer.exe GUI
3544 * actually displaying the properties.
3546 * This is fixed in Win2k. admin/non-admin
3547 * connections both display changes immediately.
3549 * 14/12/01 --jerry
3552 result=update_a_printer_2(printer.info_2);
3554 break;
3556 default:
3557 result=WERR_UNKNOWN_LEVEL;
3558 break;
3561 return result;
3564 /****************************************************************************
3565 Initialize printer devmode & data with previously saved driver init values.
3566 ****************************************************************************/
3568 static BOOL set_driver_init_2( NT_PRINTER_INFO_LEVEL_2 *info_ptr )
3570 int len = 0;
3571 pstring key;
3572 TDB_DATA kbuf, dbuf;
3573 NT_PRINTER_INFO_LEVEL_2 info;
3576 ZERO_STRUCT(info);
3579 * Delete any printer data 'values' already set. When called for driver
3580 * replace, there will generally be some, but during an add printer, there
3581 * should not be any (if there are delete them).
3584 delete_all_printer_data( info_ptr, "" );
3586 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info_ptr->drivername);
3588 kbuf.dptr = key;
3589 kbuf.dsize = strlen(key)+1;
3591 dbuf = tdb_fetch(tdb_drivers, kbuf);
3592 if (!dbuf.dptr) {
3594 * When changing to a driver that has no init info in the tdb, remove
3595 * the previous drivers init info and leave the new on blank.
3597 free_nt_devicemode(&info_ptr->devmode);
3598 return False;
3602 * Get the saved DEVMODE..
3605 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
3608 * The saved DEVMODE contains the devicename from the printer used during
3609 * the initialization save. Change it to reflect the new printer.
3612 if ( info.devmode ) {
3613 ZERO_STRUCT(info.devmode->devicename);
3614 fstrcpy(info.devmode->devicename, info_ptr->printername);
3618 * NT/2k does not change out the entire DeviceMode of a printer
3619 * when changing the driver. Only the driverextra, private, &
3620 * driverversion fields. --jerry (Thu Mar 14 08:58:43 CST 2002)
3622 * Later examination revealed that Windows NT/2k does reset the
3623 * the printer's device mode, bit **only** when you change a
3624 * property of the device mode such as the page orientation.
3625 * --jerry
3629 /* Bind the saved DEVMODE to the new the printer */
3631 free_nt_devicemode(&info_ptr->devmode);
3632 info_ptr->devmode = info.devmode;
3634 DEBUG(10,("set_driver_init_2: Set printer [%s] init %s DEVMODE for driver [%s]\n",
3635 info_ptr->printername, info_ptr->devmode?"VALID":"NULL", info_ptr->drivername));
3637 /* Add the printer data 'values' to the new printer */
3639 len += unpack_values( &info_ptr->data, dbuf.dptr+len, dbuf.dsize-len );
3642 SAFE_FREE(dbuf.dptr);
3644 return True;
3647 /****************************************************************************
3648 Initialize printer devmode & data with previously saved driver init values.
3649 When a printer is created using AddPrinter, the drivername bound to the
3650 printer is used to lookup previously saved driver initialization info, which
3651 is bound to the new printer.
3652 ****************************************************************************/
3654 BOOL set_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
3656 BOOL result = False;
3658 switch (level) {
3659 case 2:
3660 result = set_driver_init_2(printer->info_2);
3661 break;
3663 default:
3664 DEBUG(0,("set_driver_init: Programmer's error! Unknown driver_init level [%d]\n",
3665 level));
3666 break;
3669 return result;
3672 /****************************************************************************
3673 Delete driver init data stored for a specified driver
3674 ****************************************************************************/
3676 BOOL del_driver_init(char *drivername)
3678 pstring key;
3679 TDB_DATA kbuf;
3681 if (!drivername || !*drivername) {
3682 DEBUG(3,("del_driver_init: No drivername specified!\n"));
3683 return False;
3686 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, drivername);
3688 kbuf.dptr = key;
3689 kbuf.dsize = strlen(key)+1;
3691 DEBUG(6,("del_driver_init: Removing driver init data for [%s]\n", drivername));
3693 return (tdb_delete(tdb_drivers, kbuf) == 0);
3696 /****************************************************************************
3697 Pack up the DEVMODE and values for a printer into a 'driver init' entry
3698 in the tdb. Note: this is different from the driver entry and the printer
3699 entry. There should be a single driver init entry for each driver regardless
3700 of whether it was installed from NT or 2K. Technically, they should be
3701 different, but they work out to the same struct.
3702 ****************************************************************************/
3704 static uint32 update_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info)
3706 pstring key;
3707 char *buf;
3708 int buflen, len, ret;
3709 TDB_DATA kbuf, dbuf;
3711 buf = NULL;
3712 buflen = 0;
3714 again:
3715 len = 0;
3716 len += pack_devicemode(info->devmode, buf+len, buflen-len);
3718 len += pack_values( &info->data, buf+len, buflen-len );
3720 if (buflen != len) {
3721 char *tb;
3723 tb = (char *)Realloc(buf, len);
3724 if (!tb) {
3725 DEBUG(0, ("update_driver_init_2: failed to enlarge buffer!\n"));
3726 ret = -1;
3727 goto done;
3729 else
3730 buf = tb;
3731 buflen = len;
3732 goto again;
3735 slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info->drivername);
3737 kbuf.dptr = key;
3738 kbuf.dsize = strlen(key)+1;
3739 dbuf.dptr = buf;
3740 dbuf.dsize = len;
3742 ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
3744 done:
3745 if (ret == -1)
3746 DEBUG(8, ("update_driver_init_2: error updating printer init to tdb on disk\n"));
3748 SAFE_FREE(buf);
3750 DEBUG(10,("update_driver_init_2: Saved printer [%s] init DEVMODE & values for driver [%s]\n",
3751 info->sharename, info->drivername));
3753 return ret;
3756 /****************************************************************************
3757 Update (i.e. save) the driver init info (DEVMODE and values) for a printer
3758 ****************************************************************************/
3760 uint32 update_driver_init(NT_PRINTER_INFO_LEVEL printer, uint32 level)
3762 uint32 result;
3764 dump_a_printer(printer, level);
3766 switch (level) {
3767 case 2:
3768 result = update_driver_init_2(printer.info_2);
3769 break;
3770 default:
3771 result = 1;
3772 break;
3775 return result;
3778 /****************************************************************************
3779 Convert the printer data value, a REG_BINARY array, into an initialization
3780 DEVMODE. Note: the array must be parsed as if it was a DEVMODE in an rpc...
3781 got to keep the endians happy :).
3782 ****************************************************************************/
3784 static BOOL convert_driver_init( TALLOC_CTX *ctx, NT_DEVICEMODE *nt_devmode, uint8 *data, uint32 data_len )
3786 BOOL result = False;
3787 prs_struct ps;
3788 DEVICEMODE devmode;
3790 ZERO_STRUCT(devmode);
3792 prs_init(&ps, 0, ctx, UNMARSHALL);
3793 ps.data_p = (char *)data;
3794 ps.buffer_size = data_len;
3796 if (spoolss_io_devmode("phantom DEVMODE", &ps, 0, &devmode))
3797 result = convert_devicemode("", &devmode, &nt_devmode);
3798 else
3799 DEBUG(10,("convert_driver_init: error parsing DEVMODE\n"));
3801 return result;
3804 /****************************************************************************
3805 Set the DRIVER_INIT info in the tdb. Requires Win32 client code that:
3807 1. Use the driver's config DLL to this UNC printername and:
3808 a. Call DrvPrintEvent with PRINTER_EVENT_INITIALIZE
3809 b. Call DrvConvertDevMode with CDM_DRIVER_DEFAULT to get default DEVMODE
3810 2. Call SetPrinterData with the 'magic' key and the DEVMODE as data.
3812 The last step triggers saving the "driver initialization" information for
3813 this printer into the tdb. Later, new printers that use this driver will
3814 have this initialization information bound to them. This simulates the
3815 driver initialization, as if it had run on the Samba server (as it would
3816 have done on NT).
3818 The Win32 client side code requirement sucks! But until we can run arbitrary
3819 Win32 printer driver code on any Unix that Samba runs on, we are stuck with it.
3821 It would have been easier to use SetPrinter because all the UNMARSHALLING of
3822 the DEVMODE is done there, but 2K/XP clients do not set the DEVMODE... think
3823 about it and you will realize why. JRR 010720
3824 ****************************************************************************/
3826 static WERROR save_driver_init_2(NT_PRINTER_INFO_LEVEL *printer, uint8 *data, uint32 data_len )
3828 WERROR status = WERR_OK;
3829 TALLOC_CTX *ctx = NULL;
3830 NT_DEVICEMODE *nt_devmode = NULL;
3831 NT_DEVICEMODE *tmp_devmode = printer->info_2->devmode;
3834 * When the DEVMODE is already set on the printer, don't try to unpack it.
3836 DEBUG(8,("save_driver_init_2: Enter...\n"));
3838 if ( !printer->info_2->devmode && data_len ) {
3840 * Set devmode on printer info, so entire printer initialization can be
3841 * saved to tdb.
3844 if ((ctx = talloc_init("save_driver_init_2")) == NULL)
3845 return WERR_NOMEM;
3847 if ((nt_devmode = (NT_DEVICEMODE*)malloc(sizeof(NT_DEVICEMODE))) == NULL) {
3848 status = WERR_NOMEM;
3849 goto done;
3852 ZERO_STRUCTP(nt_devmode);
3855 * The DEVMODE is held in the 'data' component of the param in raw binary.
3856 * Convert it to to a devmode structure
3858 if ( !convert_driver_init( ctx, nt_devmode, data, data_len )) {
3859 DEBUG(10,("save_driver_init_2: error converting DEVMODE\n"));
3860 status = WERR_INVALID_PARAM;
3861 goto done;
3864 printer->info_2->devmode = nt_devmode;
3868 * Pack up and add (or update) the DEVMODE and any current printer data to
3869 * a 'driver init' element in the tdb
3873 if ( update_driver_init(*printer, 2) != 0 ) {
3874 DEBUG(10,("save_driver_init_2: error updating DEVMODE\n"));
3875 status = WERR_NOMEM;
3876 goto done;
3880 * If driver initialization info was successfully saved, set the current
3881 * printer to match it. This allows initialization of the current printer
3882 * as well as the driver.
3884 status = mod_a_printer(*printer, 2);
3885 if (!W_ERROR_IS_OK(status)) {
3886 DEBUG(10,("save_driver_init_2: error setting DEVMODE on printer [%s]\n",
3887 printer->info_2->printername));
3890 done:
3891 talloc_destroy(ctx);
3892 free_nt_devicemode( &nt_devmode );
3894 printer->info_2->devmode = tmp_devmode;
3896 return status;
3899 /****************************************************************************
3900 Update the driver init info (DEVMODE and specifics) for a printer
3901 ****************************************************************************/
3903 WERROR save_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level, uint8 *data, uint32 data_len)
3905 WERROR status = WERR_OK;
3907 switch (level) {
3908 case 2:
3909 status = save_driver_init_2( printer, data, data_len );
3910 break;
3911 default:
3912 status = WERR_UNKNOWN_LEVEL;
3913 break;
3916 return status;
3919 /****************************************************************************
3920 Deep copy a NT_PRINTER_DATA
3921 ****************************************************************************/
3923 static NTSTATUS copy_printer_data( NT_PRINTER_DATA *dst, NT_PRINTER_DATA *src )
3925 int i, j, num_vals, new_key_index;
3926 REGVAL_CTR *src_key, *dst_key;
3928 if ( !dst || !src )
3929 return NT_STATUS_NO_MEMORY;
3931 for ( i=0; i<src->num_keys; i++ ) {
3933 /* create a new instance of the printerkey in the destination
3934 printer_data object */
3936 new_key_index = add_new_printer_key( dst, src->keys[i].name );
3937 dst_key = &dst->keys[new_key_index].values;
3939 src_key = &src->keys[i].values;
3940 num_vals = regval_ctr_numvals( src_key );
3942 /* dup the printer entire printer key */
3944 for ( j=0; j<num_vals; j++ ) {
3945 regval_ctr_copyvalue( dst_key, regval_ctr_specific_value(src_key, j) );
3949 return NT_STATUS_OK;
3952 /****************************************************************************
3953 Deep copy a NT_PRINTER_INFO_LEVEL_2 structure using malloc()'d memeory
3954 Caller must free.
3955 ****************************************************************************/
3957 NT_PRINTER_INFO_LEVEL_2* dup_printer_2( TALLOC_CTX *ctx, NT_PRINTER_INFO_LEVEL_2 *printer )
3959 NT_PRINTER_INFO_LEVEL_2 *copy;
3961 if ( !printer )
3962 return NULL;
3964 if ( !(copy = (NT_PRINTER_INFO_LEVEL_2 *)malloc(sizeof(NT_PRINTER_INFO_LEVEL_2))) )
3965 return NULL;
3967 memcpy( copy, printer, sizeof(NT_PRINTER_INFO_LEVEL_2) );
3969 /* malloc()'d members copied here */
3971 copy->devmode = dup_nt_devicemode( printer->devmode );
3973 ZERO_STRUCT( copy->data );
3974 copy_printer_data( &copy->data, &printer->data );
3976 /* this is talloc()'d; very ugly that we have a structure that
3977 is half malloc()'d and half talloc()'d but that is the way
3978 that the PRINTER_INFO stuff is written right now. --jerry */
3980 copy->secdesc_buf = dup_sec_desc_buf( ctx, printer->secdesc_buf );
3982 return copy;
3985 /****************************************************************************
3986 Get a NT_PRINTER_INFO_LEVEL struct. It returns malloced memory.
3988 Previously the code had a memory allocation problem because it always
3989 used the TALLOC_CTX from the Printer_entry*. This context lasts
3990 as a long as the original handle is open. So if the client made a lot
3991 of getprinter[data]() calls, the memory usage would climb. Now we use
3992 a short lived TALLOC_CTX for printer_info_2 objects returned. We
3993 still use the Printer_entry->ctx for maintaining the cache copy though
3994 since that object must live as long as the handle by definition.
3995 --jerry
3997 ****************************************************************************/
3999 WERROR get_a_printer( Printer_entry *print_hnd, NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level,
4000 const char *sharename)
4002 WERROR result;
4003 NT_PRINTER_INFO_LEVEL *printer = NULL;
4005 *pp_printer = NULL;
4007 DEBUG(10,("get_a_printer: [%s] level %u\n", sharename, (unsigned int)level));
4009 switch (level) {
4010 case 2:
4011 if ((printer = (NT_PRINTER_INFO_LEVEL *)malloc(sizeof(NT_PRINTER_INFO_LEVEL))) == NULL) {
4012 DEBUG(0,("get_a_printer: malloc fail.\n"));
4013 return WERR_NOMEM;
4015 ZERO_STRUCTP(printer);
4018 * check for cache first. A Printer handle cannot changed
4019 * to another printer object so we only check that the printer
4020 * is actually for a printer and that the printer_info pointer
4021 * is valid
4023 if ( print_hnd
4024 && (print_hnd->printer_type==PRINTER_HANDLE_IS_PRINTER)
4025 && print_hnd->printer_info )
4027 /* get_talloc_ctx() works here because we need a short
4028 lived talloc context */
4030 if ( !(printer->info_2 = dup_printer_2(get_talloc_ctx(), print_hnd->printer_info->info_2)) )
4032 DEBUG(0,("get_a_printer: unable to copy cached printer info!\n"));
4034 SAFE_FREE(printer);
4035 return WERR_NOMEM;
4038 DEBUG(10,("get_a_printer: using cached copy of printer_info_2\n"));
4040 *pp_printer = printer;
4041 result = WERR_OK;
4043 break;
4046 /* no cache for this handle; see if we can match one from another handle.
4047 Make sure to use a short lived talloc ctx */
4049 if ( print_hnd )
4050 result = find_printer_in_print_hnd_cache(get_talloc_ctx(), &printer->info_2, sharename);
4052 /* fail to disk if we don't have it with any open handle */
4054 if ( !print_hnd || !W_ERROR_IS_OK(result) )
4055 result = get_a_printer_2(&printer->info_2, sharename);
4057 /* we have a new printer now. Save it with this handle */
4059 if ( W_ERROR_IS_OK(result) ) {
4060 dump_a_printer(*printer, level);
4062 /* save a copy in cache */
4063 if ( print_hnd && (print_hnd->printer_type==PRINTER_HANDLE_IS_PRINTER)) {
4064 if ( !print_hnd->printer_info )
4065 print_hnd->printer_info = (NT_PRINTER_INFO_LEVEL *)malloc(sizeof(NT_PRINTER_INFO_LEVEL));
4067 if ( print_hnd->printer_info ) {
4068 /* make sure to use the handle's talloc ctx here since
4069 the printer_2 object must last until the handle is closed */
4071 print_hnd->printer_info->info_2 = dup_printer_2(print_hnd->ctx, printer->info_2);
4073 /* don't fail the lookup just because the cache update failed */
4074 if ( !print_hnd->printer_info->info_2 )
4075 DEBUG(0,("get_a_printer: unable to copy new printer info!\n"));
4078 *pp_printer = printer;
4080 else
4081 SAFE_FREE(printer);
4083 break;
4085 default:
4086 result=WERR_UNKNOWN_LEVEL;
4087 break;
4090 DEBUG(10,("get_a_printer: [%s] level %u returning %s\n", sharename, (unsigned int)level, dos_errstr(result)));
4092 return result;
4095 /****************************************************************************
4096 Deletes a NT_PRINTER_INFO_LEVEL struct.
4097 ****************************************************************************/
4099 uint32 free_a_printer(NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level)
4101 uint32 result;
4102 NT_PRINTER_INFO_LEVEL *printer = *pp_printer;
4104 DEBUG(104,("freeing a printer at level [%d]\n", level));
4106 if (printer == NULL)
4107 return 0;
4109 switch (level) {
4110 case 2:
4111 if (printer->info_2 != NULL) {
4112 free_nt_printer_info_level_2(&printer->info_2);
4113 result=0;
4114 } else
4115 result=4;
4116 break;
4118 default:
4119 result=1;
4120 break;
4123 SAFE_FREE(*pp_printer);
4124 return result;
4127 /****************************************************************************
4128 ****************************************************************************/
4129 uint32 add_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4131 uint32 result;
4132 DEBUG(104,("adding a printer at level [%d]\n", level));
4133 dump_a_printer_driver(driver, level);
4135 switch (level) {
4136 case 3:
4137 result=add_a_printer_driver_3(driver.info_3);
4138 break;
4140 case 6:
4141 result=add_a_printer_driver_6(driver.info_6);
4142 break;
4144 default:
4145 result=1;
4146 break;
4149 return result;
4151 /****************************************************************************
4152 ****************************************************************************/
4154 WERROR get_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL *driver, uint32 level,
4155 fstring drivername, const char *architecture, uint32 version)
4157 WERROR result;
4159 switch (level) {
4160 case 3:
4161 /* Sometime we just want any version of the driver */
4163 if ( version == DRIVER_ANY_VERSION ) {
4164 /* look for Win2k first and then for NT4 */
4165 result = get_a_printer_driver_3(&driver->info_3, drivername,
4166 architecture, 3);
4168 if ( !W_ERROR_IS_OK(result) ) {
4169 result = get_a_printer_driver_3( &driver->info_3,
4170 drivername, architecture, 2 );
4172 } else {
4173 result = get_a_printer_driver_3(&driver->info_3, drivername,
4174 architecture, version);
4176 break;
4178 default:
4179 result=W_ERROR(1);
4180 break;
4183 if (W_ERROR_IS_OK(result))
4184 dump_a_printer_driver(*driver, level);
4186 return result;
4189 /****************************************************************************
4190 ****************************************************************************/
4191 uint32 free_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4193 uint32 result;
4195 switch (level) {
4196 case 3:
4198 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
4199 if (driver.info_3 != NULL)
4201 info3=driver.info_3;
4202 SAFE_FREE(info3->dependentfiles);
4203 ZERO_STRUCTP(info3);
4204 SAFE_FREE(info3);
4205 result=0;
4206 } else {
4207 result=4;
4209 break;
4211 case 6:
4213 NT_PRINTER_DRIVER_INFO_LEVEL_6 *info6;
4214 if (driver.info_6 != NULL) {
4215 info6=driver.info_6;
4216 SAFE_FREE(info6->dependentfiles);
4217 SAFE_FREE(info6->previousnames);
4218 ZERO_STRUCTP(info6);
4219 SAFE_FREE(info6);
4220 result=0;
4221 } else {
4222 result=4;
4224 break;
4226 default:
4227 result=1;
4228 break;
4230 return result;
4234 /****************************************************************************
4235 Determine whether or not a particular driver is currently assigned
4236 to a printer
4237 ****************************************************************************/
4239 BOOL printer_driver_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3 )
4241 int snum;
4242 int n_services = lp_numservices();
4243 NT_PRINTER_INFO_LEVEL *printer = NULL;
4245 if ( !info_3 )
4246 return False;
4248 DEBUG(5,("printer_driver_in_use: Beginning search through ntprinters.tdb...\n"));
4250 /* loop through the printers.tdb and check for the drivername */
4252 for (snum=0; snum<n_services; snum++) {
4253 if ( !(lp_snum_ok(snum) && lp_print_ok(snum) ) )
4254 continue;
4256 if ( !W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))) )
4257 continue;
4259 if ( !StrCaseCmp(info_3->name, printer->info_2->drivername) ) {
4260 free_a_printer( &printer, 2 );
4261 return True;
4264 free_a_printer( &printer, 2 );
4267 DEBUG(5,("printer_driver_in_use: Completed search through ntprinters.tdb...\n"));
4269 /* report that the driver is not in use by default */
4271 return False;
4275 /**********************************************************************
4276 Check to see if a ogiven file is in use by *info
4277 *********************************************************************/
4279 static BOOL drv_file_in_use( char* file, NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
4281 int i = 0;
4283 if ( !info )
4284 return False;
4286 if ( strequal(file, info->driverpath) )
4287 return True;
4289 if ( strequal(file, info->datafile) )
4290 return True;
4292 if ( strequal(file, info->configfile) )
4293 return True;
4295 if ( strequal(file, info->helpfile) )
4296 return True;
4298 /* see of there are any dependent files to examine */
4300 if ( !info->dependentfiles )
4301 return False;
4303 while ( *info->dependentfiles[i] ) {
4304 if ( strequal(file, info->dependentfiles[i]) )
4305 return True;
4306 i++;
4309 return False;
4313 /**********************************************************************
4314 Utility function to remove the dependent file pointed to by the
4315 input parameter from the list
4316 *********************************************************************/
4318 static void trim_dependent_file( fstring files[], int idx )
4321 /* bump everything down a slot */
4323 while( *files[idx+1] ) {
4324 fstrcpy( files[idx], files[idx+1] );
4325 idx++;
4328 *files[idx] = '\0';
4330 return;
4333 /**********************************************************************
4334 Check if any of the files used by src are also used by drv
4335 *********************************************************************/
4337 static BOOL trim_overlap_drv_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *src,
4338 NT_PRINTER_DRIVER_INFO_LEVEL_3 *drv )
4340 BOOL in_use = False;
4341 int i = 0;
4343 if ( !src || !drv )
4344 return False;
4346 /* check each file. Remove it from the src structure if it overlaps */
4348 if ( drv_file_in_use(src->driverpath, drv) ) {
4349 in_use = True;
4350 DEBUG(10,("Removing driverfile [%s] from list\n", src->driverpath));
4351 fstrcpy( src->driverpath, "" );
4354 if ( drv_file_in_use(src->datafile, drv) ) {
4355 in_use = True;
4356 DEBUG(10,("Removing datafile [%s] from list\n", src->datafile));
4357 fstrcpy( src->datafile, "" );
4360 if ( drv_file_in_use(src->configfile, drv) ) {
4361 in_use = True;
4362 DEBUG(10,("Removing configfile [%s] from list\n", src->configfile));
4363 fstrcpy( src->configfile, "" );
4366 if ( drv_file_in_use(src->helpfile, drv) ) {
4367 in_use = True;
4368 DEBUG(10,("Removing helpfile [%s] from list\n", src->helpfile));
4369 fstrcpy( src->helpfile, "" );
4372 /* are there any dependentfiles to examine? */
4374 if ( !src->dependentfiles )
4375 return in_use;
4377 while ( *src->dependentfiles[i] ) {
4378 if ( drv_file_in_use(src->dependentfiles[i], drv) ) {
4379 in_use = True;
4380 DEBUG(10,("Removing [%s] from dependent file list\n", src->dependentfiles[i]));
4381 trim_dependent_file( src->dependentfiles, i );
4382 } else
4383 i++;
4386 return in_use;
4389 /****************************************************************************
4390 Determine whether or not a particular driver files are currently being
4391 used by any other driver.
4393 Return value is True if any files were in use by other drivers
4394 and False otherwise.
4396 Upon return, *info has been modified to only contain the driver files
4397 which are not in use
4398 ****************************************************************************/
4400 BOOL printer_driver_files_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
4402 int i;
4403 int ndrivers;
4404 uint32 version;
4405 fstring *list = NULL;
4406 NT_PRINTER_DRIVER_INFO_LEVEL driver;
4408 if ( !info )
4409 return False;
4411 version = info->cversion;
4413 /* loop over all driver versions */
4415 DEBUG(5,("printer_driver_files_in_use: Beginning search through ntdrivers.tdb...\n"));
4417 /* get the list of drivers */
4419 list = NULL;
4420 ndrivers = get_ntdrivers(&list, info->environment, version);
4422 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n",
4423 ndrivers, info->environment, version));
4425 /* check each driver for overlap in files */
4427 for (i=0; i<ndrivers; i++) {
4428 DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
4430 ZERO_STRUCT(driver);
4432 if ( !W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, list[i], info->environment, version)) ) {
4433 SAFE_FREE(list);
4434 return True;
4437 /* check if d2 uses any files from d1 */
4438 /* only if this is a different driver than the one being deleted */
4440 if ( !strequal(info->name, driver.info_3->name) ) {
4441 if ( trim_overlap_drv_files(info, driver.info_3) ) {
4442 free_a_printer_driver(driver, 3);
4443 SAFE_FREE( list );
4444 return True;
4448 free_a_printer_driver(driver, 3);
4451 SAFE_FREE(list);
4453 DEBUG(5,("printer_driver_files_in_use: Completed search through ntdrivers.tdb...\n"));
4455 driver.info_3 = info;
4457 if ( DEBUGLEVEL >= 20 )
4458 dump_a_printer_driver( driver, 3 );
4460 return False;
4463 /****************************************************************************
4464 Actually delete the driver files. Make sure that
4465 printer_driver_files_in_use() return False before calling
4466 this.
4467 ****************************************************************************/
4469 static BOOL delete_driver_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user )
4471 int i = 0;
4472 char *s;
4473 connection_struct *conn;
4474 DATA_BLOB null_pw;
4475 NTSTATUS nt_status;
4476 fstring res_type;
4477 BOOL bad_path;
4478 SMB_STRUCT_STAT st;
4480 if ( !info_3 )
4481 return False;
4483 DEBUG(6,("delete_driver_files: deleting driver [%s] - version [%d]\n", info_3->name, info_3->cversion));
4486 * Connect to the print$ share under the same account as the
4487 * user connected to the rpc pipe. Note we must be root to
4488 * do this.
4491 null_pw = data_blob( NULL, 0 );
4492 fstrcpy(res_type, "A:");
4493 become_root();
4494 conn = make_connection_with_chdir( "print$", null_pw, res_type, user->vuid, &nt_status );
4495 unbecome_root();
4497 if ( !conn ) {
4498 DEBUG(0,("delete_driver_files: Unable to connect\n"));
4499 return False;
4502 /* Save who we are - we are temporarily becoming the connection user. */
4504 if ( !become_user(conn, conn->vuid) ) {
4505 DEBUG(0,("delete_driver_files: Can't become user!\n"));
4506 return False;
4509 /* now delete the files; must strip the '\print$' string from
4510 fron of path */
4512 if ( *info_3->driverpath ) {
4513 if ( (s = strchr( &info_3->driverpath[1], '\\' )) != NULL ) {
4514 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4515 DEBUG(10,("deleting driverfile [%s]\n", s));
4516 unlink_internals(conn, 0, s);
4520 if ( *info_3->configfile ) {
4521 if ( (s = strchr( &info_3->configfile[1], '\\' )) != NULL ) {
4522 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4523 DEBUG(10,("deleting configfile [%s]\n", s));
4524 unlink_internals(conn, 0, s);
4528 if ( *info_3->datafile ) {
4529 if ( (s = strchr( &info_3->datafile[1], '\\' )) != NULL ) {
4530 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4531 DEBUG(10,("deleting datafile [%s]\n", s));
4532 unlink_internals(conn, 0, s);
4536 if ( *info_3->helpfile ) {
4537 if ( (s = strchr( &info_3->helpfile[1], '\\' )) != NULL ) {
4538 driver_unix_convert(s, conn, NULL, &bad_path, &st);
4539 DEBUG(10,("deleting helpfile [%s]\n", s));
4540 unlink_internals(conn, 0, s);
4544 /* check if we are done removing files */
4546 if ( info_3->dependentfiles ) {
4547 while ( *info_3->dependentfiles[i] ) {
4548 char *file;
4550 /* bypass the "\print$" portion of the path */
4552 if ( (file = strchr( info_3->dependentfiles[i]+1, '\\' )) != NULL ) {
4553 driver_unix_convert(file, conn, NULL, &bad_path, &st);
4554 DEBUG(10,("deleting dependent file [%s]\n", file));
4555 unlink_internals(conn, 0, file );
4558 i++;
4562 unbecome_user();
4564 return True;
4567 /****************************************************************************
4568 Remove a printer driver from the TDB. This assumes that the the driver was
4569 previously looked up.
4570 ***************************************************************************/
4572 WERROR delete_printer_driver( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user,
4573 uint32 version, BOOL delete_files )
4575 pstring key;
4576 const char *arch;
4577 TDB_DATA kbuf, dbuf;
4578 NT_PRINTER_DRIVER_INFO_LEVEL ctr;
4580 /* delete the tdb data first */
4582 arch = get_short_archi(info_3->environment);
4583 slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX,
4584 arch, version, info_3->name);
4586 DEBUG(5,("delete_printer_driver: key = [%s] delete_files = %s\n",
4587 key, delete_files ? "TRUE" : "FALSE" ));
4589 ctr.info_3 = info_3;
4590 dump_a_printer_driver( ctr, 3 );
4592 kbuf.dptr=key;
4593 kbuf.dsize=strlen(key)+1;
4595 /* check if the driver actually exists for this environment */
4597 dbuf = tdb_fetch( tdb_drivers, kbuf );
4598 if ( !dbuf.dptr ) {
4599 DEBUG(8,("delete_printer_driver: Driver unknown [%s]\n", key));
4600 return WERR_UNKNOWN_PRINTER_DRIVER;
4603 SAFE_FREE( dbuf.dptr );
4605 /* ok... the driver exists so the delete should return success */
4607 if (tdb_delete(tdb_drivers, kbuf) == -1) {
4608 DEBUG (0,("delete_printer_driver: fail to delete %s!\n", key));
4609 return WERR_ACCESS_DENIED;
4613 * now delete any associated files if delete_files == True
4614 * even if this part failes, we return succes because the
4615 * driver doesn not exist any more
4618 if ( delete_files )
4619 delete_driver_files( info_3, user );
4622 DEBUG(5,("delete_printer_driver: driver delete successful [%s]\n", key));
4624 return WERR_OK;
4627 /****************************************************************************
4628 Store a security desc for a printer.
4629 ****************************************************************************/
4631 WERROR nt_printing_setsec(const char *printername, SEC_DESC_BUF *secdesc_ctr)
4633 SEC_DESC_BUF *new_secdesc_ctr = NULL;
4634 SEC_DESC_BUF *old_secdesc_ctr = NULL;
4635 prs_struct ps;
4636 TALLOC_CTX *mem_ctx = NULL;
4637 fstring key;
4638 WERROR status;
4640 mem_ctx = talloc_init("nt_printing_setsec");
4641 if (mem_ctx == NULL)
4642 return WERR_NOMEM;
4644 /* The old owner and group sids of the security descriptor are not
4645 present when new ACEs are added or removed by changing printer
4646 permissions through NT. If they are NULL in the new security
4647 descriptor then copy them over from the old one. */
4649 if (!secdesc_ctr->sec->owner_sid || !secdesc_ctr->sec->grp_sid) {
4650 DOM_SID *owner_sid, *group_sid;
4651 SEC_ACL *dacl, *sacl;
4652 SEC_DESC *psd = NULL;
4653 size_t size;
4655 nt_printing_getsec(mem_ctx, printername, &old_secdesc_ctr);
4657 /* Pick out correct owner and group sids */
4659 owner_sid = secdesc_ctr->sec->owner_sid ?
4660 secdesc_ctr->sec->owner_sid :
4661 old_secdesc_ctr->sec->owner_sid;
4663 group_sid = secdesc_ctr->sec->grp_sid ?
4664 secdesc_ctr->sec->grp_sid :
4665 old_secdesc_ctr->sec->grp_sid;
4667 dacl = secdesc_ctr->sec->dacl ?
4668 secdesc_ctr->sec->dacl :
4669 old_secdesc_ctr->sec->dacl;
4671 sacl = secdesc_ctr->sec->sacl ?
4672 secdesc_ctr->sec->sacl :
4673 old_secdesc_ctr->sec->sacl;
4675 /* Make a deep copy of the security descriptor */
4677 psd = make_sec_desc(mem_ctx, secdesc_ctr->sec->revision, secdesc_ctr->sec->type,
4678 owner_sid, group_sid,
4679 sacl,
4680 dacl,
4681 &size);
4683 new_secdesc_ctr = make_sec_desc_buf(mem_ctx, size, psd);
4686 if (!new_secdesc_ctr) {
4687 new_secdesc_ctr = secdesc_ctr;
4690 /* Store the security descriptor in a tdb */
4692 prs_init(&ps, (uint32)sec_desc_size(new_secdesc_ctr->sec) +
4693 sizeof(SEC_DESC_BUF), mem_ctx, MARSHALL);
4695 if (!sec_io_desc_buf("nt_printing_setsec", &new_secdesc_ctr,
4696 &ps, 1)) {
4697 status = WERR_BADFUNC;
4698 goto out;
4701 slprintf(key, sizeof(key)-1, "SECDESC/%s", printername);
4703 if (tdb_prs_store(tdb_printers, key, &ps)==0) {
4704 status = WERR_OK;
4705 } else {
4706 DEBUG(1,("Failed to store secdesc for %s\n", printername));
4707 status = WERR_BADFUNC;
4710 /* Free malloc'ed memory */
4712 out:
4714 prs_mem_free(&ps);
4715 if (mem_ctx)
4716 talloc_destroy(mem_ctx);
4717 return status;
4720 /****************************************************************************
4721 Construct a default security descriptor buffer for a printer.
4722 ****************************************************************************/
4724 static SEC_DESC_BUF *construct_default_printer_sdb(TALLOC_CTX *ctx)
4726 SEC_ACE ace[3];
4727 SEC_ACCESS sa;
4728 SEC_ACL *psa = NULL;
4729 SEC_DESC_BUF *sdb = NULL;
4730 SEC_DESC *psd = NULL;
4731 DOM_SID owner_sid;
4732 size_t sd_size;
4734 /* Create an ACE where Everyone is allowed to print */
4736 init_sec_access(&sa, PRINTER_ACE_PRINT);
4737 init_sec_ace(&ace[0], &global_sid_World, SEC_ACE_TYPE_ACCESS_ALLOWED,
4738 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
4740 /* Make the security descriptor owned by the Administrators group
4741 on the PDC of the domain. */
4743 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
4744 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4745 } else {
4747 /* Backup plan - make printer owned by admins.
4748 This should emulate a lanman printer as security
4749 settings can't be changed. */
4751 sid_copy(&owner_sid, get_global_sam_sid());
4752 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4755 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
4756 init_sec_ace(&ace[1], &owner_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
4757 sa, SEC_ACE_FLAG_OBJECT_INHERIT |
4758 SEC_ACE_FLAG_INHERIT_ONLY);
4760 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
4761 init_sec_ace(&ace[2], &owner_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
4762 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
4764 /* The ACL revision number in rpc_secdesc.h differs from the one
4765 created by NT when setting ACE entries in printer
4766 descriptors. NT4 complains about the property being edited by a
4767 NT5 machine. */
4769 if ((psa = make_sec_acl(ctx, NT4_ACL_REVISION, 3, ace)) != NULL) {
4770 psd = make_sec_desc(ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
4771 &owner_sid, NULL,
4772 NULL, psa, &sd_size);
4775 if (!psd) {
4776 DEBUG(0,("construct_default_printer_sd: Failed to make SEC_DESC.\n"));
4777 return NULL;
4780 sdb = make_sec_desc_buf(ctx, sd_size, psd);
4782 DEBUG(4,("construct_default_printer_sdb: size = %u.\n",
4783 (unsigned int)sd_size));
4785 return sdb;
4788 /****************************************************************************
4789 Get a security desc for a printer.
4790 ****************************************************************************/
4792 BOOL nt_printing_getsec(TALLOC_CTX *ctx, const char *printername, SEC_DESC_BUF **secdesc_ctr)
4794 prs_struct ps;
4795 fstring key;
4796 char *temp;
4798 if (strlen(printername) > 2 && (temp = strchr(printername + 2, '\\'))) {
4799 printername = temp + 1;
4802 /* Fetch security descriptor from tdb */
4804 slprintf(key, sizeof(key)-1, "SECDESC/%s", printername);
4806 if (tdb_prs_fetch(tdb_printers, key, &ps, ctx)!=0 ||
4807 !sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
4809 DEBUG(4,("using default secdesc for %s\n", printername));
4811 if (!(*secdesc_ctr = construct_default_printer_sdb(ctx))) {
4812 return False;
4815 /* Save default security descriptor for later */
4817 prs_init(&ps, (uint32)sec_desc_size((*secdesc_ctr)->sec) +
4818 sizeof(SEC_DESC_BUF), ctx, MARSHALL);
4820 if (sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1))
4821 tdb_prs_store(tdb_printers, key, &ps);
4823 prs_mem_free(&ps);
4825 return True;
4828 /* If security descriptor is owned by S-1-1-0 and winbindd is up,
4829 this security descriptor has been created when winbindd was
4830 down. Take ownership of security descriptor. */
4832 if (sid_equal((*secdesc_ctr)->sec->owner_sid, &global_sid_World)) {
4833 DOM_SID owner_sid;
4835 /* Change sd owner to workgroup administrator */
4837 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
4838 SEC_DESC_BUF *new_secdesc_ctr = NULL;
4839 SEC_DESC *psd = NULL;
4840 size_t size;
4842 /* Create new sd */
4844 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
4846 psd = make_sec_desc(ctx, (*secdesc_ctr)->sec->revision, (*secdesc_ctr)->sec->type,
4847 &owner_sid,
4848 (*secdesc_ctr)->sec->grp_sid,
4849 (*secdesc_ctr)->sec->sacl,
4850 (*secdesc_ctr)->sec->dacl,
4851 &size);
4853 new_secdesc_ctr = make_sec_desc_buf(ctx, size, psd);
4855 /* Swap with other one */
4857 *secdesc_ctr = new_secdesc_ctr;
4859 /* Set it */
4861 nt_printing_setsec(printername, *secdesc_ctr);
4865 if (DEBUGLEVEL >= 10) {
4866 SEC_ACL *the_acl = (*secdesc_ctr)->sec->dacl;
4867 int i;
4869 DEBUG(10, ("secdesc_ctr for %s has %d aces:\n",
4870 printername, the_acl->num_aces));
4872 for (i = 0; i < the_acl->num_aces; i++) {
4873 fstring sid_str;
4875 sid_to_string(sid_str, &the_acl->ace[i].trustee);
4877 DEBUG(10, ("%s %d %d 0x%08x\n", sid_str,
4878 the_acl->ace[i].type, the_acl->ace[i].flags,
4879 the_acl->ace[i].info.mask));
4883 prs_mem_free(&ps);
4884 return True;
4887 /* error code:
4888 0: everything OK
4889 1: level not implemented
4890 2: file doesn't exist
4891 3: can't allocate memory
4892 4: can't free memory
4893 5: non existant struct
4897 A printer and a printer driver are 2 different things.
4898 NT manages them separatelly, Samba does the same.
4899 Why ? Simply because it's easier and it makes sense !
4901 Now explanation: You have 3 printers behind your samba server,
4902 2 of them are the same make and model (laser A and B). But laser B
4903 has an 3000 sheet feeder and laser A doesn't such an option.
4904 Your third printer is an old dot-matrix model for the accounting :-).
4906 If the /usr/local/samba/lib directory (default dir), you will have
4907 5 files to describe all of this.
4909 3 files for the printers (1 by printer):
4910 NTprinter_laser A
4911 NTprinter_laser B
4912 NTprinter_accounting
4913 2 files for the drivers (1 for the laser and 1 for the dot matrix)
4914 NTdriver_printer model X
4915 NTdriver_printer model Y
4917 jfm: I should use this comment for the text file to explain
4918 same thing for the forms BTW.
4919 Je devrais mettre mes commentaires en francais, ca serait mieux :-)
4923 /* Convert generic access rights to printer object specific access rights.
4924 It turns out that NT4 security descriptors use generic access rights and
4925 NT5 the object specific ones. */
4927 void map_printer_permissions(SEC_DESC *sd)
4929 int i;
4931 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
4932 se_map_generic(&sd->dacl->ace[i].info.mask,
4933 &printer_generic_mapping);
4937 /****************************************************************************
4938 Check a user has permissions to perform the given operation. We use the
4939 permission constants defined in include/rpc_spoolss.h to check the various
4940 actions we perform when checking printer access.
4942 PRINTER_ACCESS_ADMINISTER:
4943 print_queue_pause, print_queue_resume, update_printer_sec,
4944 update_printer, spoolss_addprinterex_level_2,
4945 _spoolss_setprinterdata
4947 PRINTER_ACCESS_USE:
4948 print_job_start
4950 JOB_ACCESS_ADMINISTER:
4951 print_job_delete, print_job_pause, print_job_resume,
4952 print_queue_purge
4954 ****************************************************************************/
4955 BOOL print_access_check(struct current_user *user, int snum, int access_type)
4957 SEC_DESC_BUF *secdesc = NULL;
4958 uint32 access_granted;
4959 NTSTATUS status;
4960 BOOL result;
4961 const char *pname;
4962 TALLOC_CTX *mem_ctx = NULL;
4963 extern struct current_user current_user;
4965 /* If user is NULL then use the current_user structure */
4967 if (!user)
4968 user = &current_user;
4970 /* Always allow root or printer admins to do anything */
4972 if (user->uid == 0 ||
4973 user_in_list(uidtoname(user->uid), lp_printer_admin(snum), user->groups, user->ngroups)) {
4974 return True;
4977 /* Get printer name */
4979 pname = PRINTERNAME(snum);
4981 if (!pname || !*pname) {
4982 errno = EACCES;
4983 return False;
4986 /* Get printer security descriptor */
4988 if(!(mem_ctx = talloc_init("print_access_check"))) {
4989 errno = ENOMEM;
4990 return False;
4993 nt_printing_getsec(mem_ctx, pname, &secdesc);
4995 if (access_type == JOB_ACCESS_ADMINISTER) {
4996 SEC_DESC_BUF *parent_secdesc = secdesc;
4998 /* Create a child security descriptor to check permissions
4999 against. This is because print jobs are child objects
5000 objects of a printer. */
5002 secdesc = se_create_child_secdesc(mem_ctx, parent_secdesc->sec, False);
5004 /* Now this is the bit that really confuses me. The access
5005 type needs to be changed from JOB_ACCESS_ADMINISTER to
5006 PRINTER_ACCESS_ADMINISTER for this to work. Something
5007 to do with the child (job) object becoming like a
5008 printer?? -tpot */
5010 access_type = PRINTER_ACCESS_ADMINISTER;
5013 /* Check access */
5015 map_printer_permissions(secdesc->sec);
5017 result = se_access_check(secdesc->sec, user->nt_user_token, access_type,
5018 &access_granted, &status);
5020 DEBUG(4, ("access check was %s\n", result ? "SUCCESS" : "FAILURE"));
5022 talloc_destroy(mem_ctx);
5024 if (!result)
5025 errno = EACCES;
5027 return result;
5030 /****************************************************************************
5031 Check the time parameters allow a print operation.
5032 *****************************************************************************/
5034 BOOL print_time_access_check(int snum)
5036 NT_PRINTER_INFO_LEVEL *printer = NULL;
5037 BOOL ok = False;
5038 time_t now = time(NULL);
5039 struct tm *t;
5040 uint32 mins;
5042 if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))))
5043 return False;
5045 if (printer->info_2->starttime == 0 && printer->info_2->untiltime == 0)
5046 ok = True;
5048 t = gmtime(&now);
5049 mins = (uint32)t->tm_hour*60 + (uint32)t->tm_min;
5051 if (mins >= printer->info_2->starttime && mins <= printer->info_2->untiltime)
5052 ok = True;
5054 free_a_printer(&printer, 2);
5056 if (!ok)
5057 errno = EACCES;
5059 return ok;