Remove two statics
[Samba.git] / source / printing / nt_printing.c
blob55dd63be78c487cfc8a8271f1108a3b4c9c130d0
1 /*
2 * Unix SMB/CIFS implementation.
3 * RPC Pipe client / server routines
4 * Copyright (C) Andrew Tridgell 1992-2000,
5 * Copyright (C) Jean François Micouleau 1998-2000.
6 * Copyright (C) Gerald Carter 2002-2005.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
24 extern struct current_user current_user;
26 static TDB_CONTEXT *tdb_forms; /* used for forms files */
27 static TDB_CONTEXT *tdb_drivers; /* used for driver files */
28 static TDB_CONTEXT *tdb_printers; /* used for printers files */
30 #define FORMS_PREFIX "FORMS/"
31 #define DRIVERS_PREFIX "DRIVERS/"
32 #define DRIVER_INIT_PREFIX "DRIVER_INIT/"
33 #define PRINTERS_PREFIX "PRINTERS/"
34 #define SECDESC_PREFIX "SECDESC/"
35 #define GLOBAL_C_SETPRINTER "GLOBALS/c_setprinter"
37 #define NTDRIVERS_DATABASE_VERSION_1 1
38 #define NTDRIVERS_DATABASE_VERSION_2 2
39 #define NTDRIVERS_DATABASE_VERSION_3 3 /* little endian version of v2 */
40 #define NTDRIVERS_DATABASE_VERSION_4 4 /* fix generic bits in security descriptors */
41 #define NTDRIVERS_DATABASE_VERSION_5 5 /* normalize keys in ntprinters.tdb */
43 /* Map generic permissions to printer object specific permissions */
45 const struct generic_mapping printer_generic_mapping = {
46 PRINTER_READ,
47 PRINTER_WRITE,
48 PRINTER_EXECUTE,
49 PRINTER_ALL_ACCESS
52 const struct standard_mapping printer_std_mapping = {
53 PRINTER_READ,
54 PRINTER_WRITE,
55 PRINTER_EXECUTE,
56 PRINTER_ALL_ACCESS
59 /* Map generic permissions to print server object specific permissions */
61 const struct generic_mapping printserver_generic_mapping = {
62 SERVER_READ,
63 SERVER_WRITE,
64 SERVER_EXECUTE,
65 SERVER_ALL_ACCESS
68 const struct generic_mapping printserver_std_mapping = {
69 SERVER_READ,
70 SERVER_WRITE,
71 SERVER_EXECUTE,
72 SERVER_ALL_ACCESS
75 /* We need one default form to support our default printer. Msoft adds the
76 forms it wants and in the ORDER it wants them (note: DEVMODE papersize is an
77 array index). Letter is always first, so (for the current code) additions
78 always put things in the correct order. */
79 static const nt_forms_struct default_forms[] = {
80 {"Letter",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
81 {"Letter Small",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
82 {"Tabloid",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
83 {"Ledger",0x1,0x696b8,0x44368,0x0,0x0,0x696b8,0x44368},
84 {"Legal",0x1,0x34b5c,0x56d10,0x0,0x0,0x34b5c,0x56d10},
85 {"Statement",0x1,0x221b4,0x34b5c,0x0,0x0,0x221b4,0x34b5c},
86 {"Executive",0x1,0x2cf56,0x411cc,0x0,0x0,0x2cf56,0x411cc},
87 {"A3",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
88 {"A4",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
89 {"A4 Small",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
90 {"A5",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
91 {"B4 (JIS)",0x1,0x3ebe8,0x58de0,0x0,0x0,0x3ebe8,0x58de0},
92 {"B5 (JIS)",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
93 {"Folio",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
94 {"Quarto",0x1,0x347d8,0x43238,0x0,0x0,0x347d8,0x43238},
95 {"10x14",0x1,0x3e030,0x56d10,0x0,0x0,0x3e030,0x56d10},
96 {"11x17",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
97 {"Note",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
98 {"Envelope #9",0x1,0x18079,0x37091,0x0,0x0,0x18079,0x37091},
99 {"Envelope #10",0x1,0x19947,0x3ae94,0x0,0x0,0x19947,0x3ae94},
100 {"Envelope #11",0x1,0x1be7c,0x40565,0x0,0x0,0x1be7c,0x40565},
101 {"Envelope #12",0x1,0x1d74a,0x44368,0x0,0x0,0x1d74a,0x44368},
102 {"Envelope #14",0x1,0x1f018,0x47504,0x0,0x0,0x1f018,0x47504},
103 {"C size sheet",0x1,0x696b8,0x886d0,0x0,0x0,0x696b8,0x886d0},
104 {"D size sheet",0x1,0x886d0,0xd2d70,0x0,0x0,0x886d0,0xd2d70},
105 {"E size sheet",0x1,0xd2d70,0x110da0,0x0,0x0,0xd2d70,0x110da0},
106 {"Envelope DL",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
107 {"Envelope C5",0x1,0x278d0,0x37e88,0x0,0x0,0x278d0,0x37e88},
108 {"Envelope C3",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
109 {"Envelope C4",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
110 {"Envelope C6",0x1,0x1bd50,0x278d0,0x0,0x0,0x1bd50,0x278d0},
111 {"Envelope C65",0x1,0x1bd50,0x37e88,0x0,0x0,0x1bd50,0x37e88},
112 {"Envelope B4",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
113 {"Envelope B5",0x1,0x2af80,0x3d090,0x0,0x0,0x2af80,0x3d090},
114 {"Envelope B6",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
115 {"Envelope",0x1,0x1adb0,0x38270,0x0,0x0,0x1adb0,0x38270},
116 {"Envelope Monarch",0x1,0x18079,0x2e824,0x0,0x0,0x18079,0x2e824},
117 {"6 3/4 Envelope",0x1,0x167ab,0x284ec,0x0,0x0,0x167ab,0x284ec},
118 {"US Std Fanfold",0x1,0x5c3e1,0x44368,0x0,0x0,0x5c3e1,0x44368},
119 {"German Std Fanfold",0x1,0x34b5c,0x4a6a0,0x0,0x0,0x34b5c,0x4a6a0},
120 {"German Legal Fanfold",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
121 {"B4 (ISO)",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
122 {"Japanese Postcard",0x1,0x186a0,0x24220,0x0,0x0,0x186a0,0x24220},
123 {"9x11",0x1,0x37cf8,0x44368,0x0,0x0,0x37cf8,0x44368},
124 {"10x11",0x1,0x3e030,0x44368,0x0,0x0,0x3e030,0x44368},
125 {"15x11",0x1,0x5d048,0x44368,0x0,0x0,0x5d048,0x44368},
126 {"Envelope Invite",0x1,0x35b60,0x35b60,0x0,0x0,0x35b60,0x35b60},
127 {"Reserved48",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
128 {"Reserved49",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
129 {"Letter Extra",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
130 {"Legal Extra",0x1,0x3ae94,0x5d048,0x0,0x0,0x3ae94,0x5d048},
131 {"Tabloid Extra",0x1,0x4a6a0,0x6f9f0,0x0,0x0,0x4a6a0,0x6f9f0},
132 {"A4 Extra",0x1,0x397c2,0x4eb16,0x0,0x0,0x397c2,0x4eb16},
133 {"Letter Transverse",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
134 {"A4 Transverse",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
135 {"Letter Extra Transverse",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
136 {"Super A",0x1,0x376b8,0x56ea0,0x0,0x0,0x376b8,0x56ea0},
137 {"Super B",0x1,0x4a768,0x76e58,0x0,0x0,0x4a768,0x76e58},
138 {"Letter Plus",0x1,0x34b5c,0x4eb16,0x0,0x0,0x34b5c,0x4eb16},
139 {"A4 Plus",0x1,0x33450,0x50910,0x0,0x0,0x33450,0x50910},
140 {"A5 Transverse",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
141 {"B5 (JIS) Transverse",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
142 {"A3 Extra",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
143 {"A5 Extra",0x1,0x2a7b0,0x395f8,0x0,0x0,0x2a7b0,0x395f8},
144 {"B5 (ISO) Extra",0x1,0x31128,0x43620,0x0,0x0,0x31128,0x43620},
145 {"A2",0x1,0x668a0,0x91050,0x0,0x0,0x668a0,0x91050},
146 {"A3 Transverse",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
147 {"A3 Extra Transverse",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
148 {"Japanese Double Postcard",0x1,0x30d40,0x24220,0x0,0x0,0x30d40,0x24220},
149 {"A6",0x1,0x19a28,0x24220,0x0,0x0,0x19a28,0x24220},
150 {"Japanese Envelope Kaku #2",0x1,0x3a980,0x510e0,0x0,0x0,0x3a980,0x510e0},
151 {"Japanese Envelope Kaku #3",0x1,0x34bc0,0x43a08,0x0,0x0,0x34bc0,0x43a08},
152 {"Japanese Envelope Chou #3",0x1,0x1d4c0,0x395f8,0x0,0x0,0x1d4c0,0x395f8},
153 {"Japanese Envelope Chou #4",0x1,0x15f90,0x320c8,0x0,0x0,0x15f90,0x320c8},
154 {"Letter Rotated",0x1,0x44368,0x34b5c,0x0,0x0,0x44368,0x34b5c},
155 {"A3 Rotated",0x1,0x668a0,0x48828,0x0,0x0,0x668a0,0x48828},
156 {"A4 Rotated",0x1,0x48828,0x33450,0x0,0x0,0x48828,0x33450},
157 {"A5 Rotated",0x1,0x33450,0x24220,0x0,0x0,0x33450,0x24220},
158 {"B4 (JIS) Rotated",0x1,0x58de0,0x3ebe8,0x0,0x0,0x58de0,0x3ebe8},
159 {"B5 (JIS) Rotated",0x1,0x3ebe8,0x2c6f0,0x0,0x0,0x3ebe8,0x2c6f0},
160 {"Japanese Postcard Rotated",0x1,0x24220,0x186a0,0x0,0x0,0x24220,0x186a0},
161 {"Double Japan Postcard Rotated",0x1,0x24220,0x30d40,0x0,0x0,0x24220,0x30d40},
162 {"A6 Rotated",0x1,0x24220,0x19a28,0x0,0x0,0x24220,0x19a28},
163 {"Japan Envelope Kaku #2 Rotated",0x1,0x510e0,0x3a980,0x0,0x0,0x510e0,0x3a980},
164 {"Japan Envelope Kaku #3 Rotated",0x1,0x43a08,0x34bc0,0x0,0x0,0x43a08, 0x34bc0},
165 {"Japan Envelope Chou #3 Rotated",0x1,0x395f8,0x1d4c0,0x0,0x0,0x395f8,0x1d4c0},
166 {"Japan Envelope Chou #4 Rotated",0x1,0x320c8,0x15f90,0x0,0x0,0x320c8,0x15f90},
167 {"B6 (JIS)",0x1,0x1f400,0x2c6f0,0x0,0x0,0x1f400,0x2c6f0},
168 {"B6 (JIS) Rotated",0x1,0x2c6f0,0x1f400,0x0,0x0,0x2c6f0,0x1f400},
169 {"12x11",0x1,0x4a724,0x443e1,0x0,0x0,0x4a724,0x443e1},
170 {"Japan Envelope You #4",0x1,0x19a28,0x395f8,0x0,0x0,0x19a28,0x395f8},
171 {"Japan Envelope You #4 Rotated",0x1,0x395f8,0x19a28,0x0,0x0,0x395f8,0x19a28},
172 {"PRC 16K",0x1,0x2de60,0x3f7a0,0x0,0x0,0x2de60,0x3f7a0},
173 {"PRC 32K",0x1,0x1fbd0,0x2cec0,0x0,0x0,0x1fbd0,0x2cec0},
174 {"PRC 32K(Big)",0x1,0x222e0,0x318f8,0x0,0x0,0x222e0,0x318f8},
175 {"PRC Envelope #1",0x1,0x18e70,0x28488,0x0,0x0,0x18e70,0x28488},
176 {"PRC Envelope #2",0x1,0x18e70,0x2af80,0x0,0x0,0x18e70,0x2af80},
177 {"PRC Envelope #3",0x1,0x1e848,0x2af80,0x0,0x0,0x1e848,0x2af80},
178 {"PRC Envelope #4",0x1,0x1adb0,0x32c80,0x0,0x0,0x1adb0,0x32c80},
179 {"PRC Envelope #5",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
180 {"PRC Envelope #6",0x1,0x1d4c0,0x38270,0x0,0x0,0x1d4c0,0x38270},
181 {"PRC Envelope #7",0x1,0x27100,0x38270,0x0,0x0,0x27100,0x38270},
182 {"PRC Envelope #8",0x1,0x1d4c0,0x4b708,0x0,0x0,0x1d4c0,0x4b708},
183 {"PRC Envelope #9",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
184 {"PRC Envelope #10",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
185 {"PRC 16K Rotated",0x1,0x3f7a0,0x2de60,0x0,0x0,0x3f7a0,0x2de60},
186 {"PRC 32K Rotated",0x1,0x2cec0,0x1fbd0,0x0,0x0,0x2cec0,0x1fbd0},
187 {"PRC 32K(Big) Rotated",0x1,0x318f8,0x222e0,0x0,0x0,0x318f8,0x222e0},
188 {"PRC Envelope #1 Rotated",0x1,0x28488,0x18e70,0x0,0x0,0x28488,0x18e70},
189 {"PRC Envelope #2 Rotated",0x1,0x2af80,0x18e70,0x0,0x0,0x2af80,0x18e70},
190 {"PRC Envelope #3 Rotated",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
191 {"PRC Envelope #4 Rotated",0x1,0x32c80,0x1adb0,0x0,0x0,0x32c80,0x1adb0},
192 {"PRC Envelope #5 Rotated",0x1,0x35b60,0x1adb0,0x0,0x0,0x35b60,0x1adb0},
193 {"PRC Envelope #6 Rotated",0x1,0x38270,0x1d4c0,0x0,0x0,0x38270,0x1d4c0},
194 {"PRC Envelope #7 Rotated",0x1,0x38270,0x27100,0x0,0x0,0x38270,0x27100},
195 {"PRC Envelope #8 Rotated",0x1,0x4b708,0x1d4c0,0x0,0x0,0x4b708,0x1d4c0},
196 {"PRC Envelope #9 Rotated",0x1,0x4f1a0,0x37e88,0x0,0x0,0x4f1a0,0x37e88},
197 {"PRC Envelope #10 Rotated",0x1,0x6fd10,0x4f1a0,0x0,0x0,0x6fd10,0x4f1a0}
200 struct table_node {
201 const char *long_archi;
202 const char *short_archi;
203 int version;
206 #define SPL_ARCH_WIN40 "WIN40"
207 #define SPL_ARCH_W32X86 "W32X86"
208 #define SPL_ARCH_W32MIPS "W32MIPS"
209 #define SPL_ARCH_W32ALPHA "W32ALPHA"
210 #define SPL_ARCH_W32PPC "W32PPC"
211 #define SPL_ARCH_IA64 "IA64"
212 #define SPL_ARCH_X64 "x64"
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 {"Windows x64", SPL_ARCH_X64, 3 },
223 {NULL, "", -1 }
227 /****************************************************************************
228 generate a new TDB_DATA key for storing a printer
229 ****************************************************************************/
231 static TDB_DATA make_printer_tdbkey(TALLOC_CTX *ctx, const char *sharename )
233 fstring share;
234 char *keystr = NULL;
235 TDB_DATA key;
237 fstrcpy(share, sharename);
238 strlower_m(share);
240 keystr = talloc_asprintf(ctx, "%s%s", PRINTERS_PREFIX, share);
241 key = string_term_tdb_data(keystr ? keystr : "");
243 return key;
246 /****************************************************************************
247 generate a new TDB_DATA key for storing a printer security descriptor
248 ****************************************************************************/
250 static TDB_DATA make_printers_secdesc_tdbkey(TALLOC_CTX *ctx,
251 const char* sharename )
253 fstring share;
254 char *keystr = NULL;
255 TDB_DATA key;
257 fstrcpy(share, sharename );
258 strlower_m(share);
260 keystr = talloc_asprintf(ctx, "%s%s", SECDESC_PREFIX, share);
261 key = string_term_tdb_data(keystr ? keystr : "");
263 return key;
266 /****************************************************************************
267 ****************************************************************************/
269 static bool upgrade_to_version_3(void)
271 TDB_DATA kbuf, newkey, dbuf;
273 DEBUG(0,("upgrade_to_version_3: upgrading print tdb's to version 3\n"));
275 for (kbuf = tdb_firstkey(tdb_drivers); kbuf.dptr;
276 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
278 dbuf = tdb_fetch(tdb_drivers, kbuf);
280 if (strncmp((const char *)kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) == 0) {
281 DEBUG(0,("upgrade_to_version_3:moving form\n"));
282 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) {
283 SAFE_FREE(dbuf.dptr);
284 DEBUG(0,("upgrade_to_version_3: failed to move form. Error (%s).\n", tdb_errorstr(tdb_forms)));
285 return False;
287 if (tdb_delete(tdb_drivers, kbuf) != 0) {
288 SAFE_FREE(dbuf.dptr);
289 DEBUG(0,("upgrade_to_version_3: failed to delete form. Error (%s)\n", tdb_errorstr(tdb_drivers)));
290 return False;
294 if (strncmp((const char *)kbuf.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX)) == 0) {
295 DEBUG(0,("upgrade_to_version_3:moving printer\n"));
296 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
297 SAFE_FREE(dbuf.dptr);
298 DEBUG(0,("upgrade_to_version_3: failed to move printer. Error (%s)\n", tdb_errorstr(tdb_printers)));
299 return False;
301 if (tdb_delete(tdb_drivers, kbuf) != 0) {
302 SAFE_FREE(dbuf.dptr);
303 DEBUG(0,("upgrade_to_version_3: failed to delete printer. Error (%s)\n", tdb_errorstr(tdb_drivers)));
304 return False;
308 if (strncmp((const char *)kbuf.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX)) == 0) {
309 DEBUG(0,("upgrade_to_version_3:moving secdesc\n"));
310 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
311 SAFE_FREE(dbuf.dptr);
312 DEBUG(0,("upgrade_to_version_3: failed to move secdesc. Error (%s)\n", tdb_errorstr(tdb_printers)));
313 return False;
315 if (tdb_delete(tdb_drivers, kbuf) != 0) {
316 SAFE_FREE(dbuf.dptr);
317 DEBUG(0,("upgrade_to_version_3: failed to delete secdesc. Error (%s)\n", tdb_errorstr(tdb_drivers)));
318 return False;
322 SAFE_FREE(dbuf.dptr);
325 return True;
328 /*******************************************************************
329 Fix an issue with security descriptors. Printer sec_desc must
330 use more than the generic bits that were previously used
331 in <= 3.0.14a. They must also have a owner and group SID assigned.
332 Otherwise, any printers than have been migrated to a Windows
333 host using printmig.exe will not be accessible.
334 *******************************************************************/
336 static int sec_desc_upg_fn( TDB_CONTEXT *the_tdb, TDB_DATA key,
337 TDB_DATA data, void *state )
339 prs_struct ps;
340 SEC_DESC_BUF *sd_orig = NULL;
341 SEC_DESC_BUF *sd_new, *sd_store;
342 SEC_DESC *sec, *new_sec;
343 TALLOC_CTX *ctx = state;
344 int result, i;
345 uint32 sd_size;
346 size_t size_new_sec;
347 DOM_SID sid;
349 if (!data.dptr || data.dsize == 0) {
350 return 0;
353 if ( strncmp((const char *) key.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX) ) != 0 ) {
354 return 0;
357 /* upgrade the security descriptor */
359 ZERO_STRUCT( ps );
361 prs_init( &ps, 0, ctx, UNMARSHALL );
362 prs_give_memory( &ps, (char *)data.dptr, data.dsize, False );
364 if ( !sec_io_desc_buf( "sec_desc_upg_fn", &sd_orig, &ps, 1 ) ) {
365 /* delete bad entries */
366 DEBUG(0,("sec_desc_upg_fn: Failed to parse original sec_desc for %si. Deleting....\n",
367 (const char *)key.dptr ));
368 tdb_delete( tdb_printers, key );
369 prs_mem_free( &ps );
370 return 0;
373 if (!sd_orig) {
374 prs_mem_free( &ps );
375 return 0;
377 sec = sd_orig->sd;
379 /* is this even valid? */
381 if ( !sec->dacl ) {
382 prs_mem_free( &ps );
383 return 0;
386 /* update access masks */
388 for ( i=0; i<sec->dacl->num_aces; i++ ) {
389 switch ( sec->dacl->aces[i].access_mask ) {
390 case (GENERIC_READ_ACCESS | GENERIC_WRITE_ACCESS | GENERIC_EXECUTE_ACCESS):
391 sec->dacl->aces[i].access_mask = PRINTER_ACE_PRINT;
392 break;
394 case GENERIC_ALL_ACCESS:
395 sec->dacl->aces[i].access_mask = PRINTER_ACE_FULL_CONTROL;
396 break;
398 case READ_CONTROL_ACCESS:
399 sec->dacl->aces[i].access_mask = PRINTER_ACE_MANAGE_DOCUMENTS;
401 default: /* no change */
402 break;
406 /* create a new SEC_DESC with the appropriate owner and group SIDs */
408 string_to_sid(&sid, "S-1-5-32-544" );
409 new_sec = make_sec_desc( ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
410 &sid, &sid,
411 NULL, NULL, &size_new_sec );
412 if (!new_sec) {
413 prs_mem_free( &ps );
414 return 0;
416 sd_new = make_sec_desc_buf( ctx, size_new_sec, new_sec );
417 if (!sd_new) {
418 prs_mem_free( &ps );
419 return 0;
422 if ( !(sd_store = sec_desc_merge( ctx, sd_new, sd_orig )) ) {
423 DEBUG(0,("sec_desc_upg_fn: Failed to update sec_desc for %s\n", key.dptr ));
424 prs_mem_free( &ps );
425 return 0;
428 prs_mem_free( &ps );
430 /* store it back */
432 sd_size = sec_desc_size(sd_store->sd) + sizeof(SEC_DESC_BUF);
433 prs_init(&ps, sd_size, ctx, MARSHALL);
435 if ( !sec_io_desc_buf( "sec_desc_upg_fn", &sd_store, &ps, 1 ) ) {
436 DEBUG(0,("sec_desc_upg_fn: Failed to parse new sec_desc for %s\n", key.dptr ));
437 prs_mem_free( &ps );
438 return 0;
441 data.dptr = (uint8 *)prs_data_p( &ps );
442 data.dsize = sd_size;
444 result = tdb_store( tdb_printers, key, data, TDB_REPLACE );
446 prs_mem_free( &ps );
448 /* 0 to continue and non-zero to stop traversal */
450 return (result == -1);
453 /*******************************************************************
454 *******************************************************************/
456 static bool upgrade_to_version_4(void)
458 TALLOC_CTX *ctx;
459 int result;
461 DEBUG(0,("upgrade_to_version_4: upgrading printer security descriptors\n"));
463 if ( !(ctx = talloc_init( "upgrade_to_version_4" )) )
464 return False;
466 result = tdb_traverse( tdb_printers, sec_desc_upg_fn, ctx );
468 talloc_destroy( ctx );
470 return ( result != -1 );
473 /*******************************************************************
474 Fix an issue with security descriptors. Printer sec_desc must
475 use more than the generic bits that were previously used
476 in <= 3.0.14a. They must also have a owner and group SID assigned.
477 Otherwise, any printers than have been migrated to a Windows
478 host using printmig.exe will not be accessible.
479 *******************************************************************/
481 static int normalize_printers_fn( TDB_CONTEXT *the_tdb, TDB_DATA key,
482 TDB_DATA data, void *state )
484 TALLOC_CTX *ctx = talloc_tos();
485 TDB_DATA new_key;
487 if (!data.dptr || data.dsize == 0)
488 return 0;
490 /* upgrade printer records and security descriptors */
492 if ( strncmp((const char *) key.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX) ) == 0 ) {
493 new_key = make_printer_tdbkey(ctx, (const char *)key.dptr+strlen(PRINTERS_PREFIX) );
495 else if ( strncmp((const char *) key.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX) ) == 0 ) {
496 new_key = make_printers_secdesc_tdbkey(ctx, (const char *)key.dptr+strlen(SECDESC_PREFIX) );
498 else {
499 /* ignore this record */
500 return 0;
503 /* delete the original record and store under the normalized key */
505 if ( tdb_delete( the_tdb, key ) != 0 ) {
506 DEBUG(0,("normalize_printers_fn: tdb_delete for [%s] failed!\n",
507 key.dptr));
508 return 1;
511 if ( tdb_store( the_tdb, new_key, data, TDB_REPLACE) != 0 ) {
512 DEBUG(0,("normalize_printers_fn: failed to store new record for [%s]!\n",
513 key.dptr));
514 return 1;
517 return 0;
520 /*******************************************************************
521 *******************************************************************/
523 static bool upgrade_to_version_5(void)
525 TALLOC_CTX *ctx;
526 int result;
528 DEBUG(0,("upgrade_to_version_5: normalizing printer keys\n"));
530 if ( !(ctx = talloc_init( "upgrade_to_version_5" )) )
531 return False;
533 result = tdb_traverse( tdb_printers, normalize_printers_fn, NULL );
535 talloc_destroy( ctx );
537 return ( result != -1 );
540 /****************************************************************************
541 Open the NT printing tdbs. Done once before fork().
542 ****************************************************************************/
544 bool nt_printing_init(struct messaging_context *msg_ctx)
546 const char *vstring = "INFO/version";
547 WERROR win_rc;
548 int32 vers_id;
550 if ( tdb_drivers && tdb_printers && tdb_forms )
551 return True;
553 if (tdb_drivers)
554 tdb_close(tdb_drivers);
555 tdb_drivers = tdb_open_log(state_path("ntdrivers.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
556 if (!tdb_drivers) {
557 DEBUG(0,("nt_printing_init: Failed to open nt drivers database %s (%s)\n",
558 state_path("ntdrivers.tdb"), strerror(errno) ));
559 return False;
562 if (tdb_printers)
563 tdb_close(tdb_printers);
564 tdb_printers = tdb_open_log(state_path("ntprinters.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
565 if (!tdb_printers) {
566 DEBUG(0,("nt_printing_init: Failed to open nt printers database %s (%s)\n",
567 state_path("ntprinters.tdb"), strerror(errno) ));
568 return False;
571 if (tdb_forms)
572 tdb_close(tdb_forms);
573 tdb_forms = tdb_open_log(state_path("ntforms.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
574 if (!tdb_forms) {
575 DEBUG(0,("nt_printing_init: Failed to open nt forms database %s (%s)\n",
576 state_path("ntforms.tdb"), strerror(errno) ));
577 return False;
580 /* handle a Samba upgrade */
582 vers_id = tdb_fetch_int32(tdb_drivers, vstring);
583 if (vers_id == -1) {
584 DEBUG(10, ("Fresh database\n"));
585 tdb_store_int32( tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_5 );
586 vers_id = NTDRIVERS_DATABASE_VERSION_5;
589 if ( vers_id != NTDRIVERS_DATABASE_VERSION_5 ) {
591 if ((vers_id == NTDRIVERS_DATABASE_VERSION_1) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_1)) {
592 if (!upgrade_to_version_3())
593 return False;
594 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_3);
595 vers_id = NTDRIVERS_DATABASE_VERSION_3;
598 if ((vers_id == NTDRIVERS_DATABASE_VERSION_2) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_2)) {
599 /* Written on a bigendian machine with old fetch_int code. Save as le. */
600 /* The only upgrade between V2 and V3 is to save the version in little-endian. */
601 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_3);
602 vers_id = NTDRIVERS_DATABASE_VERSION_3;
605 if (vers_id == NTDRIVERS_DATABASE_VERSION_3 ) {
606 if ( !upgrade_to_version_4() )
607 return False;
608 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_4);
609 vers_id = NTDRIVERS_DATABASE_VERSION_4;
612 if (vers_id == NTDRIVERS_DATABASE_VERSION_4 ) {
613 if ( !upgrade_to_version_5() )
614 return False;
615 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_5);
616 vers_id = NTDRIVERS_DATABASE_VERSION_5;
620 if ( vers_id != NTDRIVERS_DATABASE_VERSION_5 ) {
621 DEBUG(0,("nt_printing_init: Unknown printer database version [%d]\n", vers_id));
622 return False;
626 update_c_setprinter(True);
629 * register callback to handle updating printers as new
630 * drivers are installed
633 messaging_register(msg_ctx, NULL, MSG_PRINTER_DRVUPGRADE,
634 do_drv_upgrade_printer);
637 * register callback to handle updating printer data
638 * when a driver is initialized
641 messaging_register(msg_ctx, NULL, MSG_PRINTERDATA_INIT_RESET,
642 reset_all_printerdata);
644 /* of course, none of the message callbacks matter if you don't
645 tell messages.c that you interested in receiving PRINT_GENERAL
646 msgs. This is done in claim_connection() */
649 if ( lp_security() == SEC_ADS ) {
650 win_rc = check_published_printers();
651 if (!W_ERROR_IS_OK(win_rc))
652 DEBUG(0, ("nt_printing_init: error checking published printers: %s\n", dos_errstr(win_rc)));
655 return True;
658 /*******************************************************************
659 Function to allow filename parsing "the old way".
660 ********************************************************************/
662 static char *driver_unix_convert(connection_struct *conn,
663 const char *old_name,
664 SMB_STRUCT_STAT *pst)
666 TALLOC_CTX *ctx = talloc_tos();
667 char *name = talloc_strdup(ctx, old_name);
668 char *new_name = NULL;
670 if (!name) {
671 return NULL;
673 unix_format(name);
674 name = unix_clean_name(ctx, name);
675 if (!name) {
676 return NULL;
678 trim_string(name,"/","/");
679 unix_convert(ctx,conn, name, false, &new_name, NULL, pst);
680 return new_name;
683 /*******************************************************************
684 tdb traversal function for counting printers.
685 ********************************************************************/
687 static int traverse_counting_printers(TDB_CONTEXT *t, TDB_DATA key,
688 TDB_DATA data, void *context)
690 int *printer_count = (int*)context;
692 if (memcmp(PRINTERS_PREFIX, key.dptr, sizeof(PRINTERS_PREFIX)-1) == 0) {
693 (*printer_count)++;
694 DEBUG(10,("traverse_counting_printers: printer = [%s] printer_count = %d\n", key.dptr, *printer_count));
697 return 0;
700 /*******************************************************************
701 Update the spooler global c_setprinter. This variable is initialized
702 when the parent smbd starts with the number of existing printers. It
703 is monotonically increased by the current number of printers *after*
704 each add or delete printer RPC. Only Microsoft knows why... JRR020119
705 ********************************************************************/
707 uint32 update_c_setprinter(bool initialize)
709 int32 c_setprinter;
710 int32 printer_count = 0;
712 tdb_lock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
714 /* Traverse the tdb, counting the printers */
715 tdb_traverse(tdb_printers, traverse_counting_printers, (void *)&printer_count);
717 /* If initializing, set c_setprinter to current printers count
718 * otherwise, bump it by the current printer count
720 if (!initialize)
721 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER) + printer_count;
722 else
723 c_setprinter = printer_count;
725 DEBUG(10,("update_c_setprinter: c_setprinter = %u\n", (unsigned int)c_setprinter));
726 tdb_store_int32(tdb_printers, GLOBAL_C_SETPRINTER, c_setprinter);
728 tdb_unlock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
730 return (uint32)c_setprinter;
733 /*******************************************************************
734 Get the spooler global c_setprinter, accounting for initialization.
735 ********************************************************************/
737 uint32 get_c_setprinter(void)
739 int32 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER);
741 if (c_setprinter == (int32)-1)
742 c_setprinter = update_c_setprinter(True);
744 DEBUG(10,("get_c_setprinter: c_setprinter = %d\n", c_setprinter));
746 return (uint32)c_setprinter;
749 /****************************************************************************
750 Get builtin form struct list.
751 ****************************************************************************/
753 int get_builtin_ntforms(nt_forms_struct **list)
755 *list = (nt_forms_struct *)memdup(&default_forms[0], sizeof(default_forms));
756 if (!*list) {
757 return 0;
759 return sizeof(default_forms) / sizeof(default_forms[0]);
762 /****************************************************************************
763 get a builtin form struct
764 ****************************************************************************/
766 bool get_a_builtin_ntform(UNISTR2 *uni_formname,nt_forms_struct *form)
768 int i,count;
769 fstring form_name;
770 unistr2_to_ascii(form_name, uni_formname, sizeof(form_name));
771 DEBUGADD(6,("Looking for builtin form %s \n", form_name));
772 count = sizeof(default_forms) / sizeof(default_forms[0]);
773 for (i=0;i<count;i++) {
774 if (strequal(form_name,default_forms[i].name)) {
775 DEBUGADD(6,("Found builtin form %s \n", form_name));
776 memcpy(form,&default_forms[i],sizeof(*form));
777 break;
781 return (i !=count);
784 /****************************************************************************
785 get a form struct list.
786 ****************************************************************************/
788 int get_ntforms(nt_forms_struct **list)
790 TDB_DATA kbuf, newkey, dbuf;
791 nt_forms_struct form;
792 int ret;
793 int i;
794 int n = 0;
796 *list = NULL;
798 for (kbuf = tdb_firstkey(tdb_forms);
799 kbuf.dptr;
800 newkey = tdb_nextkey(tdb_forms, kbuf), safe_free(kbuf.dptr), kbuf=newkey)
802 if (strncmp((const char *)kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) != 0)
803 continue;
805 dbuf = tdb_fetch(tdb_forms, kbuf);
806 if (!dbuf.dptr)
807 continue;
809 fstrcpy(form.name, (const char *)kbuf.dptr+strlen(FORMS_PREFIX));
810 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddddddd",
811 &i, &form.flag, &form.width, &form.length, &form.left,
812 &form.top, &form.right, &form.bottom);
813 SAFE_FREE(dbuf.dptr);
814 if (ret != dbuf.dsize)
815 continue;
817 *list = SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1);
818 if (!*list) {
819 DEBUG(0,("get_ntforms: Realloc fail.\n"));
820 return 0;
822 (*list)[n] = form;
823 n++;
827 return n;
830 /****************************************************************************
831 write a form struct list
832 ****************************************************************************/
834 int write_ntforms(nt_forms_struct **list, int number)
836 TALLOC_CTX *ctx = talloc_tos();
837 char *buf = NULL;
838 char *key = NULL;
839 int len;
840 TDB_DATA dbuf;
841 int i;
843 for (i=0;i<number;i++) {
844 /* save index, so list is rebuilt in correct order */
845 len = tdb_pack(NULL, 0, "dddddddd",
846 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
847 (*list)[i].left, (*list)[i].top, (*list)[i].right,
848 (*list)[i].bottom);
849 if (!len) {
850 continue;
852 buf = TALLOC_ARRAY(ctx, char, len);
853 if (!buf) {
854 return 0;
856 len = tdb_pack((uint8 *)buf, len, "dddddddd",
857 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
858 (*list)[i].left, (*list)[i].top, (*list)[i].right,
859 (*list)[i].bottom);
860 key = talloc_asprintf(ctx, "%s%s", FORMS_PREFIX, (*list)[i].name);
861 if (!key) {
862 return 0;
864 dbuf.dsize = len;
865 dbuf.dptr = (uint8 *)buf;
866 if (tdb_store_bystring(tdb_forms, key, dbuf, TDB_REPLACE) != 0) {
867 TALLOC_FREE(key);
868 TALLOC_FREE(buf);
869 break;
871 TALLOC_FREE(key);
872 TALLOC_FREE(buf);
875 return i;
878 /****************************************************************************
879 add a form struct at the end of the list
880 ****************************************************************************/
881 bool add_a_form(nt_forms_struct **list, const FORM *form, int *count)
883 int n=0;
884 bool update;
885 fstring form_name;
888 * NT tries to add forms even when
889 * they are already in the base
890 * only update the values if already present
893 update=False;
895 unistr2_to_ascii(form_name, &form->name, sizeof(form_name));
896 for (n=0; n<*count; n++) {
897 if ( strequal((*list)[n].name, form_name) ) {
898 update=True;
899 break;
903 if (update==False) {
904 if((*list=SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1)) == NULL) {
905 DEBUG(0,("add_a_form: failed to enlarge forms list!\n"));
906 return False;
908 unistr2_to_ascii((*list)[n].name, &form->name, sizeof((*list)[n].name));
909 (*count)++;
912 (*list)[n].flag=form->flags;
913 (*list)[n].width=form->size_x;
914 (*list)[n].length=form->size_y;
915 (*list)[n].left=form->left;
916 (*list)[n].top=form->top;
917 (*list)[n].right=form->right;
918 (*list)[n].bottom=form->bottom;
920 DEBUG(6,("add_a_form: Successfully %s form [%s]\n",
921 update ? "updated" : "added", form_name));
923 return True;
926 /****************************************************************************
927 Delete a named form struct.
928 ****************************************************************************/
930 bool delete_a_form(nt_forms_struct **list, UNISTR2 *del_name, int *count, WERROR *ret)
932 char *key = NULL;
933 int n=0;
934 fstring form_name;
936 *ret = WERR_OK;
938 unistr2_to_ascii(form_name, del_name, sizeof(form_name));
940 for (n=0; n<*count; n++) {
941 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
942 DEBUG(103, ("delete_a_form, [%s] in list\n", form_name));
943 break;
947 if (n == *count) {
948 DEBUG(10,("delete_a_form, [%s] not found\n", form_name));
949 *ret = WERR_INVALID_PARAM;
950 return False;
953 if (asprintf(&key, "%s%s", FORMS_PREFIX, (*list)[n].name) < 0) {
954 *ret = WERR_NOMEM;
955 return false;
957 if (tdb_delete_bystring(tdb_forms, key) != 0) {
958 SAFE_FREE(key);
959 *ret = WERR_NOMEM;
960 return False;
962 SAFE_FREE(key);
963 return true;
966 /****************************************************************************
967 Update a form struct.
968 ****************************************************************************/
970 void update_a_form(nt_forms_struct **list, const FORM *form, int count)
972 int n=0;
973 fstring form_name;
974 unistr2_to_ascii(form_name, &(form->name), sizeof(form_name));
976 DEBUG(106, ("[%s]\n", form_name));
977 for (n=0; n<count; n++) {
978 DEBUGADD(106, ("n [%d]:[%s]\n", n, (*list)[n].name));
979 if (!strncmp((*list)[n].name, form_name, strlen(form_name)))
980 break;
983 if (n==count) return;
985 (*list)[n].flag=form->flags;
986 (*list)[n].width=form->size_x;
987 (*list)[n].length=form->size_y;
988 (*list)[n].left=form->left;
989 (*list)[n].top=form->top;
990 (*list)[n].right=form->right;
991 (*list)[n].bottom=form->bottom;
994 /****************************************************************************
995 Get the nt drivers list.
996 Traverse the database and look-up the matching names.
997 ****************************************************************************/
998 int get_ntdrivers(fstring **list, const char *architecture, uint32 version)
1000 int total=0;
1001 const char *short_archi;
1002 char *key = NULL;
1003 TDB_DATA kbuf, newkey;
1005 short_archi = get_short_archi(architecture);
1006 if (!short_archi) {
1007 return 0;
1010 if (asprintf(&key, "%s%s/%d/", DRIVERS_PREFIX,
1011 short_archi, version) < 0) {
1012 return 0;
1015 for (kbuf = tdb_firstkey(tdb_drivers);
1016 kbuf.dptr;
1017 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
1019 if (strncmp((const char *)kbuf.dptr, key, strlen(key)) != 0)
1020 continue;
1022 if((*list = SMB_REALLOC_ARRAY(*list, fstring, total+1)) == NULL) {
1023 DEBUG(0,("get_ntdrivers: failed to enlarge list!\n"));
1024 SAFE_FREE(key);
1025 return -1;
1028 fstrcpy((*list)[total], (const char *)kbuf.dptr+strlen(key));
1029 total++;
1032 SAFE_FREE(key);
1033 return(total);
1036 /****************************************************************************
1037 Function to do the mapping between the long architecture name and
1038 the short one.
1039 ****************************************************************************/
1041 const char *get_short_archi(const char *long_archi)
1043 int i=-1;
1045 DEBUG(107,("Getting architecture dependant directory\n"));
1046 do {
1047 i++;
1048 } while ( (archi_table[i].long_archi!=NULL ) &&
1049 StrCaseCmp(long_archi, archi_table[i].long_archi) );
1051 if (archi_table[i].long_archi==NULL) {
1052 DEBUGADD(10,("Unknown architecture [%s] !\n", long_archi));
1053 return NULL;
1056 /* this might be client code - but shouldn't this be an fstrcpy etc? */
1058 DEBUGADD(108,("index: [%d]\n", i));
1059 DEBUGADD(108,("long architecture: [%s]\n", archi_table[i].long_archi));
1060 DEBUGADD(108,("short architecture: [%s]\n", archi_table[i].short_archi));
1062 return archi_table[i].short_archi;
1065 /****************************************************************************
1066 Version information in Microsoft files is held in a VS_VERSION_INFO structure.
1067 There are two case to be covered here: PE (Portable Executable) and NE (New
1068 Executable) files. Both files support the same INFO structure, but PE files
1069 store the signature in unicode, and NE files store it as !unicode.
1070 returns -1 on error, 1 on version info found, and 0 on no version info found.
1071 ****************************************************************************/
1073 static int get_file_version(files_struct *fsp, char *fname,uint32 *major, uint32 *minor)
1075 int i;
1076 char *buf = NULL;
1077 ssize_t byte_count;
1079 if ((buf=(char *)SMB_MALLOC(DOS_HEADER_SIZE)) == NULL) {
1080 DEBUG(0,("get_file_version: PE file [%s] DOS Header malloc failed bytes = %d\n",
1081 fname, DOS_HEADER_SIZE));
1082 goto error_exit;
1085 if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
1086 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %lu\n",
1087 fname, (unsigned long)byte_count));
1088 goto no_version_info;
1091 /* Is this really a DOS header? */
1092 if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
1093 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
1094 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
1095 goto no_version_info;
1098 /* Skip OEM header (if any) and the DOS stub to start of Windows header */
1099 if (SMB_VFS_LSEEK(fsp, fsp->fh->fd, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (SMB_OFF_T)-1) {
1100 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
1101 fname, errno));
1102 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1103 goto no_version_info;
1106 /* Note: DOS_HEADER_SIZE and NE_HEADER_SIZE are incidentally same */
1107 if ((byte_count = vfs_read_data(fsp, buf, NE_HEADER_SIZE)) < NE_HEADER_SIZE) {
1108 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %lu\n",
1109 fname, (unsigned long)byte_count));
1110 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1111 goto no_version_info;
1114 /* The header may be a PE (Portable Executable) or an NE (New Executable) */
1115 if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
1116 unsigned int num_sections;
1117 unsigned int section_table_bytes;
1119 /* Just skip over optional header to get to section table */
1120 if (SMB_VFS_LSEEK(fsp, fsp->fh->fd,
1121 SVAL(buf,PE_HEADER_OPTIONAL_HEADER_SIZE)-(NE_HEADER_SIZE-PE_HEADER_SIZE),
1122 SEEK_CUR) == (SMB_OFF_T)-1) {
1123 DEBUG(3,("get_file_version: File [%s] Windows optional header too short, errno = %d\n",
1124 fname, errno));
1125 goto error_exit;
1128 /* get the section table */
1129 num_sections = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
1130 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
1131 if (section_table_bytes == 0)
1132 goto error_exit;
1134 SAFE_FREE(buf);
1135 if ((buf=(char *)SMB_MALLOC(section_table_bytes)) == NULL) {
1136 DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
1137 fname, section_table_bytes));
1138 goto error_exit;
1141 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
1142 DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %lu\n",
1143 fname, (unsigned long)byte_count));
1144 goto error_exit;
1147 /* Iterate the section table looking for the resource section ".rsrc" */
1148 for (i = 0; i < num_sections; i++) {
1149 int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
1151 if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
1152 unsigned int section_pos = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
1153 unsigned int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
1155 if (section_bytes == 0)
1156 goto error_exit;
1158 SAFE_FREE(buf);
1159 if ((buf=(char *)SMB_MALLOC(section_bytes)) == NULL) {
1160 DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
1161 fname, section_bytes));
1162 goto error_exit;
1165 /* Seek to the start of the .rsrc section info */
1166 if (SMB_VFS_LSEEK(fsp, fsp->fh->fd, section_pos, SEEK_SET) == (SMB_OFF_T)-1) {
1167 DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
1168 fname, errno));
1169 goto error_exit;
1172 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
1173 DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %lu\n",
1174 fname, (unsigned long)byte_count));
1175 goto error_exit;
1178 if (section_bytes < VS_VERSION_INFO_UNICODE_SIZE)
1179 goto error_exit;
1181 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
1182 /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
1183 if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
1184 /* Align to next long address */
1185 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
1187 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
1188 *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
1189 *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
1191 DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
1192 fname, *major, *minor,
1193 (*major>>16)&0xffff, *major&0xffff,
1194 (*minor>>16)&0xffff, *minor&0xffff));
1195 SAFE_FREE(buf);
1196 return 1;
1203 /* Version info not found, fall back to origin date/time */
1204 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
1205 SAFE_FREE(buf);
1206 return 0;
1208 } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
1209 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
1210 DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
1211 fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
1212 /* At this point, we assume the file is in error. It still could be somthing
1213 * else besides a NE file, but it unlikely at this point. */
1214 goto error_exit;
1217 /* Allocate a bit more space to speed up things */
1218 SAFE_FREE(buf);
1219 if ((buf=(char *)SMB_MALLOC(VS_NE_BUF_SIZE)) == NULL) {
1220 DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes = %d\n",
1221 fname, PE_HEADER_SIZE));
1222 goto error_exit;
1225 /* This is a HACK! I got tired of trying to sort through the messy
1226 * 'NE' file format. If anyone wants to clean this up please have at
1227 * it, but this works. 'NE' files will eventually fade away. JRR */
1228 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
1229 /* Cover case that should not occur in a well formed 'NE' .dll file */
1230 if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
1232 for(i=0; i<byte_count; i++) {
1233 /* Fast skip past data that can't possibly match */
1234 if (buf[i] != 'V') continue;
1236 /* Potential match data crosses buf boundry, move it to beginning
1237 * of buf, and fill the buf with as much as it will hold. */
1238 if (i>byte_count-VS_VERSION_INFO_SIZE) {
1239 int bc;
1241 memcpy(buf, &buf[i], byte_count-i);
1242 if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
1243 (byte_count-i))) < 0) {
1245 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
1246 fname, errno));
1247 goto error_exit;
1250 byte_count = bc + (byte_count - i);
1251 if (byte_count<VS_VERSION_INFO_SIZE) break;
1253 i = 0;
1256 /* Check that the full signature string and the magic number that
1257 * follows exist (not a perfect solution, but the chances that this
1258 * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
1259 * twice, as it is simpler to read the code. */
1260 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
1261 /* Compute skip alignment to next long address */
1262 int skip = -(SMB_VFS_LSEEK(fsp, fsp->fh->fd, 0, SEEK_CUR) - (byte_count - i) +
1263 sizeof(VS_SIGNATURE)) & 3;
1264 if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
1266 *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
1267 *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
1268 DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
1269 fname, *major, *minor,
1270 (*major>>16)&0xffff, *major&0xffff,
1271 (*minor>>16)&0xffff, *minor&0xffff));
1272 SAFE_FREE(buf);
1273 return 1;
1278 /* Version info not found, fall back to origin date/time */
1279 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
1280 SAFE_FREE(buf);
1281 return 0;
1283 } else
1284 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1285 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
1286 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
1288 no_version_info:
1289 SAFE_FREE(buf);
1290 return 0;
1292 error_exit:
1293 SAFE_FREE(buf);
1294 return -1;
1297 /****************************************************************************
1298 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
1299 share one or more files. During the MS installation process files are checked
1300 to insure that only a newer version of a shared file is installed over an
1301 older version. There are several possibilities for this comparison. If there
1302 is no previous version, the new one is newer (obviously). If either file is
1303 missing the version info structure, compare the creation date (on Unix use
1304 the modification date). Otherwise chose the numerically larger version number.
1305 ****************************************************************************/
1307 static int file_version_is_newer(connection_struct *conn, fstring new_file, fstring old_file)
1309 bool use_version = true;
1310 char *filepath = NULL;
1312 uint32 new_major;
1313 uint32 new_minor;
1314 time_t new_create_time;
1316 uint32 old_major;
1317 uint32 old_minor;
1318 time_t old_create_time;
1320 files_struct *fsp = NULL;
1321 SMB_STRUCT_STAT st;
1322 SMB_STRUCT_STAT stat_buf;
1324 NTSTATUS status;
1326 SET_STAT_INVALID(st);
1327 SET_STAT_INVALID(stat_buf);
1328 new_create_time = (time_t)0;
1329 old_create_time = (time_t)0;
1331 /* Get file version info (if available) for previous file (if it exists) */
1332 filepath = driver_unix_convert(conn,old_file,&stat_buf);
1333 if (!filepath) {
1334 goto error_exit;
1337 status = open_file_ntcreate(conn, NULL, filepath, &stat_buf,
1338 FILE_GENERIC_READ,
1339 FILE_SHARE_READ|FILE_SHARE_WRITE,
1340 FILE_OPEN,
1342 FILE_ATTRIBUTE_NORMAL,
1343 INTERNAL_OPEN_ONLY,
1344 NULL, &fsp);
1346 if (!NT_STATUS_IS_OK(status)) {
1347 /* Old file not found, so by definition new file is in fact newer */
1348 DEBUG(10,("file_version_is_newer: Can't open old file [%s], errno = %d\n",
1349 filepath, errno));
1350 return 1;
1352 } else {
1353 int ret = get_file_version(fsp, old_file, &old_major, &old_minor);
1354 if (ret == -1) {
1355 goto error_exit;
1358 if (!ret) {
1359 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1360 old_file));
1361 use_version = false;
1362 if (SMB_VFS_FSTAT(fsp, fsp->fh->fd, &st) == -1) {
1363 goto error_exit;
1365 old_create_time = st.st_mtime;
1366 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", old_create_time));
1369 close_file(fsp, NORMAL_CLOSE);
1371 /* Get file version info (if available) for new file */
1372 filepath = driver_unix_convert(conn,new_file,&stat_buf);
1373 if (!filepath) {
1374 goto error_exit;
1377 status = open_file_ntcreate(conn, NULL, filepath, &stat_buf,
1378 FILE_GENERIC_READ,
1379 FILE_SHARE_READ|FILE_SHARE_WRITE,
1380 FILE_OPEN,
1382 FILE_ATTRIBUTE_NORMAL,
1383 INTERNAL_OPEN_ONLY,
1384 NULL, &fsp);
1386 if (!NT_STATUS_IS_OK(status)) {
1387 /* New file not found, this shouldn't occur if the caller did its job */
1388 DEBUG(3,("file_version_is_newer: Can't open new file [%s], errno = %d\n",
1389 filepath, errno));
1390 goto error_exit;
1392 } else {
1393 int ret = get_file_version(fsp, new_file, &new_major, &new_minor);
1394 if (ret == -1) {
1395 goto error_exit;
1398 if (!ret) {
1399 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1400 new_file));
1401 use_version = false;
1402 if (SMB_VFS_FSTAT(fsp, fsp->fh->fd, &st) == -1) {
1403 goto error_exit;
1405 new_create_time = st.st_mtime;
1406 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", new_create_time));
1409 close_file(fsp, NORMAL_CLOSE);
1411 if (use_version && (new_major != old_major || new_minor != old_minor)) {
1412 /* Compare versions and choose the larger version number */
1413 if (new_major > old_major ||
1414 (new_major == old_major && new_minor > old_minor)) {
1416 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1417 return 1;
1419 else {
1420 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1421 return 0;
1424 } else {
1425 /* Compare modification time/dates and choose the newest time/date */
1426 if (new_create_time > old_create_time) {
1427 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1428 return 1;
1430 else {
1431 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1432 return 0;
1436 error_exit:
1437 if(fsp)
1438 close_file(fsp, NORMAL_CLOSE);
1439 return -1;
1442 /****************************************************************************
1443 Determine the correct cVersion associated with an architecture and driver
1444 ****************************************************************************/
1445 static uint32 get_correct_cversion(const char *architecture, fstring driverpath_in,
1446 struct current_user *user, WERROR *perr)
1448 int cversion;
1449 NTSTATUS nt_status;
1450 char *driverpath = NULL;
1451 DATA_BLOB null_pw;
1452 fstring res_type;
1453 files_struct *fsp = NULL;
1454 SMB_STRUCT_STAT st;
1455 connection_struct *conn;
1456 NTSTATUS status;
1458 SET_STAT_INVALID(st);
1460 *perr = WERR_INVALID_PARAM;
1462 /* If architecture is Windows 95/98/ME, the version is always 0. */
1463 if (strcmp(architecture, SPL_ARCH_WIN40) == 0) {
1464 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
1465 *perr = WERR_OK;
1466 return 0;
1469 /* If architecture is Windows x64, the version is always 3. */
1470 if (strcmp(architecture, SPL_ARCH_X64) == 0) {
1471 DEBUG(10,("get_correct_cversion: Driver is x64, cversion = 3\n"));
1472 *perr = WERR_OK;
1473 return 3;
1477 * Connect to the print$ share under the same account as the user connected
1478 * to the rpc pipe. Note we must still be root to do this.
1481 /* Null password is ok - we are already an authenticated user... */
1482 null_pw = data_blob_null;
1483 fstrcpy(res_type, "A:");
1484 become_root();
1485 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1486 unbecome_root();
1488 if (conn == NULL) {
1489 DEBUG(0,("get_correct_cversion: Unable to connect\n"));
1490 *perr = ntstatus_to_werror(nt_status);
1491 return -1;
1494 /* We are temporarily becoming the connection user. */
1495 if (!become_user(conn, user->vuid)) {
1496 DEBUG(0,("get_correct_cversion: Can't become user!\n"));
1497 *perr = WERR_ACCESS_DENIED;
1498 return -1;
1501 /* Open the driver file (Portable Executable format) and determine the
1502 * deriver the cversion. */
1503 driverpath = talloc_asprintf(talloc_tos(),
1504 "%s/%s",
1505 architecture,
1506 driverpath_in);
1507 if (!driverpath) {
1508 *perr = WERR_NOMEM;
1509 goto error_exit;
1512 driverpath = driver_unix_convert(conn,driverpath,&st);
1513 if (!driverpath) {
1514 *perr = WERR_NOMEM;
1515 goto error_exit;
1518 if (!vfs_file_exist(conn, driverpath, &st)) {
1519 *perr = WERR_BADFILE;
1520 goto error_exit;
1523 status = open_file_ntcreate(conn, NULL, driverpath, &st,
1524 FILE_GENERIC_READ,
1525 FILE_SHARE_READ|FILE_SHARE_WRITE,
1526 FILE_OPEN,
1528 FILE_ATTRIBUTE_NORMAL,
1529 INTERNAL_OPEN_ONLY,
1530 NULL, &fsp);
1532 if (!NT_STATUS_IS_OK(status)) {
1533 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = %d\n",
1534 driverpath, errno));
1535 *perr = WERR_ACCESS_DENIED;
1536 goto error_exit;
1537 } else {
1538 uint32 major;
1539 uint32 minor;
1540 int ret = get_file_version(fsp, driverpath, &major, &minor);
1541 if (ret == -1) goto error_exit;
1543 if (!ret) {
1544 DEBUG(6,("get_correct_cversion: Version info not found [%s]\n", driverpath));
1545 goto error_exit;
1549 * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
1550 * for more details. Version in this case is not just the version of the
1551 * file, but the version in the sense of kernal mode (2) vs. user mode
1552 * (3) drivers. Other bits of the version fields are the version info.
1553 * JRR 010716
1555 cversion = major & 0x0000ffff;
1556 switch (cversion) {
1557 case 2: /* WinNT drivers */
1558 case 3: /* Win2K drivers */
1559 break;
1561 default:
1562 DEBUG(6,("get_correct_cversion: cversion invalid [%s] cversion = %d\n",
1563 driverpath, cversion));
1564 goto error_exit;
1567 DEBUG(10,("get_correct_cversion: Version info found [%s] major = 0x%x minor = 0x%x\n",
1568 driverpath, major, minor));
1571 DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
1572 driverpath, cversion));
1574 close_file(fsp, NORMAL_CLOSE);
1575 close_cnum(conn, user->vuid);
1576 unbecome_user();
1577 *perr = WERR_OK;
1578 return cversion;
1581 error_exit:
1583 if(fsp)
1584 close_file(fsp, NORMAL_CLOSE);
1586 close_cnum(conn, user->vuid);
1587 unbecome_user();
1588 return -1;
1591 /****************************************************************************
1592 ****************************************************************************/
1593 static WERROR clean_up_driver_struct_level_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver,
1594 struct current_user *user)
1596 const char *architecture;
1597 fstring new_name;
1598 char *p;
1599 int i;
1600 WERROR err;
1602 /* clean up the driver name.
1603 * we can get .\driver.dll
1604 * or worse c:\windows\system\driver.dll !
1606 /* using an intermediate string to not have overlaping memcpy()'s */
1607 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1608 fstrcpy(new_name, p+1);
1609 fstrcpy(driver->driverpath, new_name);
1612 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1613 fstrcpy(new_name, p+1);
1614 fstrcpy(driver->datafile, new_name);
1617 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1618 fstrcpy(new_name, p+1);
1619 fstrcpy(driver->configfile, new_name);
1622 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1623 fstrcpy(new_name, p+1);
1624 fstrcpy(driver->helpfile, new_name);
1627 if (driver->dependentfiles) {
1628 for (i=0; *driver->dependentfiles[i]; i++) {
1629 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1630 fstrcpy(new_name, p+1);
1631 fstrcpy(driver->dependentfiles[i], new_name);
1636 architecture = get_short_archi(driver->environment);
1637 if (!architecture) {
1638 return WERR_UNKNOWN_PRINTER_DRIVER;
1641 /* jfm:7/16/2000 the client always sends the cversion=0.
1642 * The server should check which version the driver is by reading
1643 * the PE header of driver->driverpath.
1645 * For Windows 95/98 the version is 0 (so the value sent is correct)
1646 * For Windows NT (the architecture doesn't matter)
1647 * NT 3.1: cversion=0
1648 * NT 3.5/3.51: cversion=1
1649 * NT 4: cversion=2
1650 * NT2K: cversion=3
1652 if ((driver->cversion = get_correct_cversion( architecture, driver->driverpath, user, &err)) == -1)
1653 return err;
1655 return WERR_OK;
1658 /****************************************************************************
1659 ****************************************************************************/
1660 static WERROR clean_up_driver_struct_level_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver, struct current_user *user)
1662 const char *architecture;
1663 fstring new_name;
1664 char *p;
1665 int i;
1666 WERROR err;
1668 /* clean up the driver name.
1669 * we can get .\driver.dll
1670 * or worse c:\windows\system\driver.dll !
1672 /* using an intermediate string to not have overlaping memcpy()'s */
1673 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1674 fstrcpy(new_name, p+1);
1675 fstrcpy(driver->driverpath, new_name);
1678 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1679 fstrcpy(new_name, p+1);
1680 fstrcpy(driver->datafile, new_name);
1683 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1684 fstrcpy(new_name, p+1);
1685 fstrcpy(driver->configfile, new_name);
1688 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1689 fstrcpy(new_name, p+1);
1690 fstrcpy(driver->helpfile, new_name);
1693 if (driver->dependentfiles) {
1694 for (i=0; *driver->dependentfiles[i]; i++) {
1695 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1696 fstrcpy(new_name, p+1);
1697 fstrcpy(driver->dependentfiles[i], new_name);
1702 architecture = get_short_archi(driver->environment);
1703 if (!architecture) {
1704 return WERR_UNKNOWN_PRINTER_DRIVER;
1707 /* jfm:7/16/2000 the client always sends the cversion=0.
1708 * The server should check which version the driver is by reading
1709 * the PE header of driver->driverpath.
1711 * For Windows 95/98 the version is 0 (so the value sent is correct)
1712 * For Windows NT (the architecture doesn't matter)
1713 * NT 3.1: cversion=0
1714 * NT 3.5/3.51: cversion=1
1715 * NT 4: cversion=2
1716 * NT2K: cversion=3
1719 if ((driver->version = get_correct_cversion(architecture, driver->driverpath, user, &err)) == -1)
1720 return err;
1722 return WERR_OK;
1725 /****************************************************************************
1726 ****************************************************************************/
1727 WERROR clean_up_driver_struct(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract,
1728 uint32 level, struct current_user *user)
1730 switch (level) {
1731 case 3:
1733 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1734 driver=driver_abstract.info_3;
1735 return clean_up_driver_struct_level_3(driver, user);
1737 case 6:
1739 NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver;
1740 driver=driver_abstract.info_6;
1741 return clean_up_driver_struct_level_6(driver, user);
1743 default:
1744 return WERR_INVALID_PARAM;
1748 /****************************************************************************
1749 This function sucks and should be replaced. JRA.
1750 ****************************************************************************/
1752 static void convert_level_6_to_level3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *dst, NT_PRINTER_DRIVER_INFO_LEVEL_6 *src)
1754 dst->cversion = src->version;
1756 fstrcpy( dst->name, src->name);
1757 fstrcpy( dst->environment, src->environment);
1758 fstrcpy( dst->driverpath, src->driverpath);
1759 fstrcpy( dst->datafile, src->datafile);
1760 fstrcpy( dst->configfile, src->configfile);
1761 fstrcpy( dst->helpfile, src->helpfile);
1762 fstrcpy( dst->monitorname, src->monitorname);
1763 fstrcpy( dst->defaultdatatype, src->defaultdatatype);
1764 dst->dependentfiles = src->dependentfiles;
1767 #if 0 /* Debugging function */
1769 static char* ffmt(unsigned char *c){
1770 int i;
1771 static char ffmt_str[17];
1773 for (i=0; i<16; i++) {
1774 if ((c[i] < ' ') || (c[i] > '~'))
1775 ffmt_str[i]='.';
1776 else
1777 ffmt_str[i]=c[i];
1779 ffmt_str[16]='\0';
1780 return ffmt_str;
1783 #endif
1785 /****************************************************************************
1786 ****************************************************************************/
1787 WERROR move_driver_to_download_area(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract, uint32 level,
1788 struct current_user *user, WERROR *perr)
1790 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1791 NT_PRINTER_DRIVER_INFO_LEVEL_3 converted_driver;
1792 const char *architecture;
1793 char *new_dir = NULL;
1794 char *old_name = NULL;
1795 char *new_name = NULL;
1796 DATA_BLOB null_pw;
1797 connection_struct *conn;
1798 NTSTATUS nt_status;
1799 fstring res_type;
1800 SMB_STRUCT_STAT st;
1801 int i;
1802 TALLOC_CTX *ctx = talloc_tos();
1803 int ver = 0;
1805 *perr = WERR_OK;
1807 if (level==3)
1808 driver=driver_abstract.info_3;
1809 else if (level==6) {
1810 convert_level_6_to_level3(&converted_driver, driver_abstract.info_6);
1811 driver = &converted_driver;
1812 } else {
1813 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)level ));
1814 return WERR_UNKNOWN_LEVEL;
1817 architecture = get_short_archi(driver->environment);
1818 if (!architecture) {
1819 return WERR_UNKNOWN_PRINTER_DRIVER;
1823 * Connect to the print$ share under the same account as the user connected to the rpc pipe.
1824 * Note we must be root to do this.
1827 null_pw = data_blob_null;
1828 fstrcpy(res_type, "A:");
1829 become_root();
1830 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1831 unbecome_root();
1833 if (conn == NULL) {
1834 DEBUG(0,("move_driver_to_download_area: Unable to connect\n"));
1835 *perr = ntstatus_to_werror(nt_status);
1836 return WERR_NO_SUCH_SHARE;
1840 * Save who we are - we are temporarily becoming the connection user.
1843 if (!become_user(conn, conn->vuid)) {
1844 DEBUG(0,("move_driver_to_download_area: Can't become user!\n"));
1845 return WERR_ACCESS_DENIED;
1848 /* WE ARE NOW RUNNING AS USER conn->vuid !!!!! */
1851 * make the directories version and version\driver_name
1852 * under the architecture directory.
1854 DEBUG(5,("Creating first directory\n"));
1855 new_dir = talloc_asprintf(ctx,
1856 "%s/%d",
1857 architecture,
1858 driver->cversion);
1859 if (!new_dir) {
1860 *perr = WERR_NOMEM;
1861 goto err_exit;
1863 new_dir = driver_unix_convert(conn,new_dir,&st);
1864 if (!new_dir) {
1865 *perr = WERR_NOMEM;
1866 goto err_exit;
1869 create_directory(conn, new_dir);
1871 /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1872 * listed for this driver which has already been moved, skip it (note:
1873 * drivers may list the same file name several times. Then check if the
1874 * file already exists in archi\cversion\, if so, check that the version
1875 * info (or time stamps if version info is unavailable) is newer (or the
1876 * date is later). If it is, move it to archi\cversion\filexxx.yyy.
1877 * Otherwise, delete the file.
1879 * If a file is not moved to archi\cversion\ because of an error, all the
1880 * rest of the 'unmoved' driver files are removed from archi\. If one or
1881 * more of the driver's files was already moved to archi\cversion\, it
1882 * potentially leaves the driver in a partially updated state. Version
1883 * trauma will most likely occur if an client attempts to use any printer
1884 * bound to the driver. Perhaps a rewrite to make sure the moves can be
1885 * done is appropriate... later JRR
1888 DEBUG(5,("Moving files now !\n"));
1890 if (driver->driverpath && strlen(driver->driverpath)) {
1891 new_name = talloc_asprintf(ctx,
1892 "%s/%s",
1893 architecture,
1894 driver->driverpath);
1895 if (!new_name) {
1896 *perr = WERR_NOMEM;
1897 goto err_exit;
1899 old_name = talloc_asprintf(ctx,
1900 "%s/%s",
1901 new_dir,
1902 driver->driverpath);
1903 if (!old_name) {
1904 *perr = WERR_NOMEM;
1905 goto err_exit;
1908 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1909 new_name = driver_unix_convert(conn,new_name,&st);
1910 if (!new_name) {
1911 *perr = WERR_NOMEM;
1912 goto err_exit;
1914 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1915 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1916 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1917 new_name, old_name));
1918 *perr = WERR_ACCESS_DENIED;
1919 ver = -1;
1924 if (driver->datafile && strlen(driver->datafile)) {
1925 if (!strequal(driver->datafile, driver->driverpath)) {
1926 new_name = talloc_asprintf(ctx,
1927 "%s/%s",
1928 architecture,
1929 driver->datafile);
1930 if (!new_name) {
1931 *perr = WERR_NOMEM;
1932 goto err_exit;
1934 old_name = talloc_asprintf(ctx,
1935 "%s/%s",
1936 new_dir,
1937 driver->datafile);
1938 if (!old_name) {
1939 *perr = WERR_NOMEM;
1940 goto err_exit;
1942 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1943 new_name = driver_unix_convert(conn,new_name,&st);
1944 if (!new_name) {
1945 *perr = WERR_NOMEM;
1946 goto err_exit;
1948 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1949 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1950 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1951 new_name, old_name));
1952 *perr = WERR_ACCESS_DENIED;
1953 ver = -1;
1959 if (driver->configfile && strlen(driver->configfile)) {
1960 if (!strequal(driver->configfile, driver->driverpath) &&
1961 !strequal(driver->configfile, driver->datafile)) {
1962 new_name = talloc_asprintf(ctx,
1963 "%s/%s",
1964 architecture,
1965 driver->configfile);
1966 if (!new_name) {
1967 *perr = WERR_NOMEM;
1968 goto err_exit;
1970 old_name = talloc_asprintf(ctx,
1971 "%s/%s",
1972 new_dir,
1973 driver->configfile);
1974 if (!old_name) {
1975 *perr = WERR_NOMEM;
1976 goto err_exit;
1978 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1979 new_name = driver_unix_convert(conn,new_name,&st);
1980 if (!new_name) {
1981 *perr = WERR_NOMEM;
1982 goto err_exit;
1984 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1985 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1986 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1987 new_name, old_name));
1988 *perr = WERR_ACCESS_DENIED;
1989 ver = -1;
1995 if (driver->helpfile && strlen(driver->helpfile)) {
1996 if (!strequal(driver->helpfile, driver->driverpath) &&
1997 !strequal(driver->helpfile, driver->datafile) &&
1998 !strequal(driver->helpfile, driver->configfile)) {
1999 new_name = talloc_asprintf(ctx,
2000 "%s/%s",
2001 architecture,
2002 driver->helpfile);
2003 if (!new_name) {
2004 *perr = WERR_NOMEM;
2005 goto err_exit;
2007 old_name = talloc_asprintf(ctx,
2008 "%s/%s",
2009 new_dir,
2010 driver->helpfile);
2011 if (!old_name) {
2012 *perr = WERR_NOMEM;
2013 goto err_exit;
2015 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
2016 new_name = driver_unix_convert(conn,new_name,&st);
2017 if (!new_name) {
2018 *perr = WERR_NOMEM;
2019 goto err_exit;
2021 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
2022 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
2023 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
2024 new_name, old_name));
2025 *perr = WERR_ACCESS_DENIED;
2026 ver = -1;
2032 if (driver->dependentfiles) {
2033 for (i=0; *driver->dependentfiles[i]; i++) {
2034 if (!strequal(driver->dependentfiles[i], driver->driverpath) &&
2035 !strequal(driver->dependentfiles[i], driver->datafile) &&
2036 !strequal(driver->dependentfiles[i], driver->configfile) &&
2037 !strequal(driver->dependentfiles[i], driver->helpfile)) {
2038 int j;
2039 for (j=0; j < i; j++) {
2040 if (strequal(driver->dependentfiles[i], driver->dependentfiles[j])) {
2041 goto NextDriver;
2045 new_name = talloc_asprintf(ctx,
2046 "%s/%s",
2047 architecture,
2048 driver->dependentfiles[i]);
2049 if (!new_name) {
2050 *perr = WERR_NOMEM;
2051 goto err_exit;
2053 old_name = talloc_asprintf(ctx,
2054 "%s/%s",
2055 new_dir,
2056 driver->dependentfiles[i]);
2057 if (!old_name) {
2058 *perr = WERR_NOMEM;
2059 goto err_exit;
2061 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
2062 new_name = driver_unix_convert(conn,new_name,&st);
2063 if (!new_name) {
2064 *perr = WERR_NOMEM;
2065 goto err_exit;
2067 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name,
2068 OPENX_FILE_EXISTS_TRUNCATE|
2069 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
2070 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
2071 new_name, old_name));
2072 *perr = WERR_ACCESS_DENIED;
2073 ver = -1;
2077 NextDriver: ;
2081 err_exit:
2083 close_cnum(conn, user->vuid);
2084 unbecome_user();
2086 if (W_ERROR_EQUAL(*perr, WERR_OK)) {
2087 return WERR_OK;
2089 if (ver == -1) {
2090 return WERR_UNKNOWN_PRINTER_DRIVER;
2092 return (*perr);
2095 /****************************************************************************
2096 ****************************************************************************/
2098 static uint32 add_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver)
2100 TALLOC_CTX *ctx = talloc_tos();
2101 int len, buflen;
2102 const char *architecture;
2103 char *directory = NULL;
2104 fstring temp_name;
2105 char *key = NULL;
2106 uint8 *buf;
2107 int i, ret;
2108 TDB_DATA dbuf;
2110 architecture = get_short_archi(driver->environment);
2111 if (!architecture) {
2112 return (uint32)-1;
2115 /* The names are relative. We store them in the form: \print$\arch\version\driver.xxx
2116 * \\server is added in the rpc server layer.
2117 * It does make sense to NOT store the server's name in the printer TDB.
2120 directory = talloc_asprintf(ctx, "\\print$\\%s\\%d\\",
2121 architecture, driver->cversion);
2122 if (!directory) {
2123 return (uint32)-1;
2126 /* .inf files do not always list a file for each of the four standard files.
2127 * Don't prepend a path to a null filename, or client claims:
2128 * "The server on which the printer resides does not have a suitable
2129 * <printer driver name> printer driver installed. Click OK if you
2130 * wish to install the driver on your local machine."
2132 if (strlen(driver->driverpath)) {
2133 fstrcpy(temp_name, driver->driverpath);
2134 slprintf(driver->driverpath, sizeof(driver->driverpath)-1, "%s%s", directory, temp_name);
2137 if (strlen(driver->datafile)) {
2138 fstrcpy(temp_name, driver->datafile);
2139 slprintf(driver->datafile, sizeof(driver->datafile)-1, "%s%s", directory, temp_name);
2142 if (strlen(driver->configfile)) {
2143 fstrcpy(temp_name, driver->configfile);
2144 slprintf(driver->configfile, sizeof(driver->configfile)-1, "%s%s", directory, temp_name);
2147 if (strlen(driver->helpfile)) {
2148 fstrcpy(temp_name, driver->helpfile);
2149 slprintf(driver->helpfile, sizeof(driver->helpfile)-1, "%s%s", directory, temp_name);
2152 if (driver->dependentfiles) {
2153 for (i=0; *driver->dependentfiles[i]; i++) {
2154 fstrcpy(temp_name, driver->dependentfiles[i]);
2155 slprintf(driver->dependentfiles[i], sizeof(driver->dependentfiles[i])-1, "%s%s", directory, temp_name);
2159 key = talloc_asprintf(ctx, "%s%s/%d/%s", DRIVERS_PREFIX,
2160 architecture, driver->cversion, driver->name);
2161 if (!key) {
2162 return (uint32)-1;
2165 DEBUG(5,("add_a_printer_driver_3: Adding driver with key %s\n", key ));
2167 buf = NULL;
2168 len = buflen = 0;
2170 again:
2171 len = 0;
2172 len += tdb_pack(buf+len, buflen-len, "dffffffff",
2173 driver->cversion,
2174 driver->name,
2175 driver->environment,
2176 driver->driverpath,
2177 driver->datafile,
2178 driver->configfile,
2179 driver->helpfile,
2180 driver->monitorname,
2181 driver->defaultdatatype);
2183 if (driver->dependentfiles) {
2184 for (i=0; *driver->dependentfiles[i]; i++) {
2185 len += tdb_pack(buf+len, buflen-len, "f",
2186 driver->dependentfiles[i]);
2190 if (len != buflen) {
2191 buf = (uint8 *)SMB_REALLOC(buf, len);
2192 if (!buf) {
2193 DEBUG(0,("add_a_printer_driver_3: failed to enlarge buffer\n!"));
2194 ret = -1;
2195 goto done;
2197 buflen = len;
2198 goto again;
2201 dbuf.dptr = buf;
2202 dbuf.dsize = len;
2204 ret = tdb_store_bystring(tdb_drivers, key, dbuf, TDB_REPLACE);
2206 done:
2207 if (ret)
2208 DEBUG(0,("add_a_printer_driver_3: Adding driver with key %s failed.\n", key ));
2210 SAFE_FREE(buf);
2211 return ret;
2214 /****************************************************************************
2215 ****************************************************************************/
2216 static uint32 add_a_printer_driver_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver)
2218 NT_PRINTER_DRIVER_INFO_LEVEL_3 info3;
2220 ZERO_STRUCT(info3);
2221 info3.cversion = driver->version;
2222 fstrcpy(info3.name,driver->name);
2223 fstrcpy(info3.environment,driver->environment);
2224 fstrcpy(info3.driverpath,driver->driverpath);
2225 fstrcpy(info3.datafile,driver->datafile);
2226 fstrcpy(info3.configfile,driver->configfile);
2227 fstrcpy(info3.helpfile,driver->helpfile);
2228 fstrcpy(info3.monitorname,driver->monitorname);
2229 fstrcpy(info3.defaultdatatype,driver->defaultdatatype);
2230 info3.dependentfiles = driver->dependentfiles;
2232 return add_a_printer_driver_3(&info3);
2236 /****************************************************************************
2237 ****************************************************************************/
2238 static WERROR get_a_printer_driver_3_default(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, const char *driver, const char *arch)
2240 NT_PRINTER_DRIVER_INFO_LEVEL_3 info;
2242 ZERO_STRUCT(info);
2244 fstrcpy(info.name, driver);
2245 fstrcpy(info.defaultdatatype, "RAW");
2247 fstrcpy(info.driverpath, "");
2248 fstrcpy(info.datafile, "");
2249 fstrcpy(info.configfile, "");
2250 fstrcpy(info.helpfile, "");
2252 if ((info.dependentfiles= SMB_MALLOC_ARRAY(fstring, 2)) == NULL)
2253 return WERR_NOMEM;
2255 memset(info.dependentfiles, '\0', 2*sizeof(fstring));
2256 fstrcpy(info.dependentfiles[0], "");
2258 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&info, sizeof(info));
2259 if (!*info_ptr) {
2260 SAFE_FREE(info.dependentfiles);
2261 return WERR_NOMEM;
2264 return WERR_OK;
2267 /****************************************************************************
2268 ****************************************************************************/
2269 static WERROR get_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring drivername, const char *arch, uint32 version)
2271 NT_PRINTER_DRIVER_INFO_LEVEL_3 driver;
2272 TDB_DATA dbuf;
2273 const char *architecture;
2274 int len = 0;
2275 int i;
2276 char *key = NULL;
2278 ZERO_STRUCT(driver);
2280 architecture = get_short_archi(arch);
2281 if ( !architecture ) {
2282 return WERR_UNKNOWN_PRINTER_DRIVER;
2285 /* Windows 4.0 (i.e. win9x) should always use a version of 0 */
2287 if ( strcmp( architecture, SPL_ARCH_WIN40 ) == 0 )
2288 version = 0;
2290 DEBUG(8,("get_a_printer_driver_3: [%s%s/%d/%s]\n", DRIVERS_PREFIX, architecture, version, drivername));
2292 if (asprintf(&key, "%s%s/%d/%s", DRIVERS_PREFIX,
2293 architecture, version, drivername) < 0) {
2294 return WERR_NOMEM;
2297 dbuf = tdb_fetch_bystring(tdb_drivers, key);
2298 if (!dbuf.dptr) {
2299 SAFE_FREE(key);
2300 return WERR_UNKNOWN_PRINTER_DRIVER;
2303 len += tdb_unpack(dbuf.dptr, dbuf.dsize, "dffffffff",
2304 &driver.cversion,
2305 driver.name,
2306 driver.environment,
2307 driver.driverpath,
2308 driver.datafile,
2309 driver.configfile,
2310 driver.helpfile,
2311 driver.monitorname,
2312 driver.defaultdatatype);
2314 i=0;
2315 while (len < dbuf.dsize) {
2316 driver.dependentfiles = SMB_REALLOC_ARRAY(driver.dependentfiles, fstring, i+2);
2317 if ( !driver.dependentfiles ) {
2318 DEBUG(0,("get_a_printer_driver_3: failed to enlarge buffer!\n"));
2319 break;
2322 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f",
2323 &driver.dependentfiles[i]);
2324 i++;
2327 if ( driver.dependentfiles )
2328 fstrcpy( driver.dependentfiles[i], "" );
2330 SAFE_FREE(dbuf.dptr);
2331 SAFE_FREE(key);
2333 if (len != dbuf.dsize) {
2334 SAFE_FREE(driver.dependentfiles);
2336 return get_a_printer_driver_3_default(info_ptr, drivername, arch);
2339 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&driver, sizeof(driver));
2340 if (!*info_ptr) {
2341 SAFE_FREE(driver.dependentfiles);
2342 return WERR_NOMEM;
2345 return WERR_OK;
2348 /****************************************************************************
2349 Debugging function, dump at level 6 the struct in the logs.
2350 ****************************************************************************/
2352 static uint32 dump_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
2354 uint32 result;
2355 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
2356 int i;
2358 DEBUG(20,("Dumping printer driver at level [%d]\n", level));
2360 switch (level)
2362 case 3:
2364 if (driver.info_3 == NULL)
2365 result=5;
2366 else {
2367 info3=driver.info_3;
2369 DEBUGADD(20,("version:[%d]\n", info3->cversion));
2370 DEBUGADD(20,("name:[%s]\n", info3->name));
2371 DEBUGADD(20,("environment:[%s]\n", info3->environment));
2372 DEBUGADD(20,("driverpath:[%s]\n", info3->driverpath));
2373 DEBUGADD(20,("datafile:[%s]\n", info3->datafile));
2374 DEBUGADD(20,("configfile:[%s]\n", info3->configfile));
2375 DEBUGADD(20,("helpfile:[%s]\n", info3->helpfile));
2376 DEBUGADD(20,("monitorname:[%s]\n", info3->monitorname));
2377 DEBUGADD(20,("defaultdatatype:[%s]\n", info3->defaultdatatype));
2379 for (i=0; info3->dependentfiles &&
2380 *info3->dependentfiles[i]; i++) {
2381 DEBUGADD(20,("dependentfile:[%s]\n",
2382 info3->dependentfiles[i]));
2384 result=0;
2386 break;
2388 default:
2389 DEBUGADD(20,("dump_a_printer_driver: Level %u not implemented\n", (unsigned int)level));
2390 result=1;
2391 break;
2394 return result;
2397 /****************************************************************************
2398 ****************************************************************************/
2399 int pack_devicemode(NT_DEVICEMODE *nt_devmode, uint8 *buf, int buflen)
2401 int len = 0;
2403 len += tdb_pack(buf+len, buflen-len, "p", nt_devmode);
2405 if (!nt_devmode)
2406 return len;
2408 len += tdb_pack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2409 nt_devmode->devicename,
2410 nt_devmode->formname,
2412 nt_devmode->specversion,
2413 nt_devmode->driverversion,
2414 nt_devmode->size,
2415 nt_devmode->driverextra,
2416 nt_devmode->orientation,
2417 nt_devmode->papersize,
2418 nt_devmode->paperlength,
2419 nt_devmode->paperwidth,
2420 nt_devmode->scale,
2421 nt_devmode->copies,
2422 nt_devmode->defaultsource,
2423 nt_devmode->printquality,
2424 nt_devmode->color,
2425 nt_devmode->duplex,
2426 nt_devmode->yresolution,
2427 nt_devmode->ttoption,
2428 nt_devmode->collate,
2429 nt_devmode->logpixels,
2431 nt_devmode->fields,
2432 nt_devmode->bitsperpel,
2433 nt_devmode->pelswidth,
2434 nt_devmode->pelsheight,
2435 nt_devmode->displayflags,
2436 nt_devmode->displayfrequency,
2437 nt_devmode->icmmethod,
2438 nt_devmode->icmintent,
2439 nt_devmode->mediatype,
2440 nt_devmode->dithertype,
2441 nt_devmode->reserved1,
2442 nt_devmode->reserved2,
2443 nt_devmode->panningwidth,
2444 nt_devmode->panningheight,
2445 nt_devmode->nt_dev_private);
2447 if (nt_devmode->nt_dev_private) {
2448 len += tdb_pack(buf+len, buflen-len, "B",
2449 nt_devmode->driverextra,
2450 nt_devmode->nt_dev_private);
2453 DEBUG(8,("Packed devicemode [%s]\n", nt_devmode->formname));
2455 return len;
2458 /****************************************************************************
2459 Pack all values in all printer keys
2460 ***************************************************************************/
2462 static int pack_values(NT_PRINTER_DATA *data, uint8 *buf, int buflen)
2464 int len = 0;
2465 int i, j;
2466 REGISTRY_VALUE *val;
2467 REGVAL_CTR *val_ctr;
2468 char *path = NULL;
2469 int num_values;
2471 if ( !data )
2472 return 0;
2474 /* loop over all keys */
2476 for ( i=0; i<data->num_keys; i++ ) {
2477 val_ctr = data->keys[i].values;
2478 num_values = regval_ctr_numvals( val_ctr );
2480 /* pack the keyname followed by a empty value */
2482 len += tdb_pack(buf+len, buflen-len, "pPdB",
2483 &data->keys[i].name,
2484 data->keys[i].name,
2485 REG_NONE,
2487 NULL);
2489 /* now loop over all values */
2491 for ( j=0; j<num_values; j++ ) {
2492 /* pathname should be stored as <key>\<value> */
2494 val = regval_ctr_specific_value( val_ctr, j );
2495 if (asprintf(&path, "%s\\%s",
2496 data->keys[i].name,
2497 regval_name(val)) < 0) {
2498 return -1;
2501 len += tdb_pack(buf+len, buflen-len, "pPdB",
2502 val,
2503 path,
2504 regval_type(val),
2505 regval_size(val),
2506 regval_data_p(val) );
2508 DEBUG(8,("specific: [%s], len: %d\n", regval_name(val), regval_size(val)));
2509 SAFE_FREE(path);
2514 /* terminator */
2516 len += tdb_pack(buf+len, buflen-len, "p", NULL);
2518 return len;
2522 /****************************************************************************
2523 Delete a printer - this just deletes the printer info file, any open
2524 handles are not affected.
2525 ****************************************************************************/
2527 uint32 del_a_printer(const char *sharename)
2529 TDB_DATA kbuf;
2530 char *printdb_path = NULL;
2531 TALLOC_CTX *ctx = talloc_tos();
2533 kbuf = make_printer_tdbkey(ctx, sharename);
2534 tdb_delete(tdb_printers, kbuf);
2536 kbuf= make_printers_secdesc_tdbkey(ctx, sharename);
2537 tdb_delete(tdb_printers, kbuf);
2539 close_all_print_db();
2541 if (geteuid() == 0) {
2542 if (asprintf(&printdb_path, "%s%s.tdb",
2543 lock_path("printing/"),
2544 sharename) < 0) {
2545 return (uint32)-1;
2547 unlink(printdb_path);
2548 SAFE_FREE(printdb_path);
2551 return 0;
2554 /****************************************************************************
2555 ****************************************************************************/
2556 static WERROR update_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info)
2558 uint8 *buf;
2559 int buflen, len;
2560 int retlen;
2561 WERROR ret;
2562 TDB_DATA kbuf, dbuf;
2565 * in addprinter: no servername and the printer is the name
2566 * in setprinter: servername is \\server
2567 * and printer is \\server\\printer
2569 * Samba manages only local printers.
2570 * we currently don't support things like i
2571 * path=\\other_server\printer
2573 * We only store the printername, not \\server\printername
2576 if ( info->servername[0] != '\0' ) {
2577 trim_string(info->printername, info->servername, NULL);
2578 trim_char(info->printername, '\\', '\0');
2579 info->servername[0]='\0';
2583 * JFM: one day I'll forget.
2584 * below that's info->portname because that's the SAMBA sharename
2585 * and I made NT 'thinks' it's the portname
2586 * the info->sharename is the thing you can name when you add a printer
2587 * that's the short-name when you create shared printer for 95/98
2588 * So I've made a limitation in SAMBA: you can only have 1 printer model
2589 * behind a SAMBA share.
2592 buf = NULL;
2593 buflen = 0;
2595 again:
2596 len = 0;
2597 len += tdb_pack(buf+len, buflen-len, "dddddddddddfffffPfffff",
2598 info->attributes,
2599 info->priority,
2600 info->default_priority,
2601 info->starttime,
2602 info->untiltime,
2603 info->status,
2604 info->cjobs,
2605 info->averageppm,
2606 info->changeid,
2607 info->c_setprinter,
2608 info->setuptime,
2609 info->servername,
2610 info->printername,
2611 info->sharename,
2612 info->portname,
2613 info->drivername,
2614 info->comment,
2615 info->location,
2616 info->sepfile,
2617 info->printprocessor,
2618 info->datatype,
2619 info->parameters);
2621 len += pack_devicemode(info->devmode, buf+len, buflen-len);
2622 retlen = pack_values( info->data, buf+len, buflen-len );
2623 if (retlen == -1) {
2624 ret = WERR_NOMEM;
2625 goto done;
2627 len += retlen;
2629 if (buflen != len) {
2630 buf = (uint8 *)SMB_REALLOC(buf, len);
2631 if (!buf) {
2632 DEBUG(0,("update_a_printer_2: failed to enlarge buffer!\n"));
2633 ret = WERR_NOMEM;
2634 goto done;
2636 buflen = len;
2637 goto again;
2640 kbuf = make_printer_tdbkey(talloc_tos(), info->sharename );
2642 dbuf.dptr = buf;
2643 dbuf.dsize = len;
2645 ret = (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) == 0? WERR_OK : WERR_NOMEM);
2647 done:
2648 if (!W_ERROR_IS_OK(ret))
2649 DEBUG(8, ("error updating printer to tdb on disk\n"));
2651 SAFE_FREE(buf);
2653 DEBUG(8,("packed printer [%s] with driver [%s] portname=[%s] len=%d\n",
2654 info->sharename, info->drivername, info->portname, len));
2656 return ret;
2660 /****************************************************************************
2661 Malloc and return an NT devicemode.
2662 ****************************************************************************/
2664 NT_DEVICEMODE *construct_nt_devicemode(const fstring default_devicename)
2667 char adevice[MAXDEVICENAME];
2668 NT_DEVICEMODE *nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE);
2670 if (nt_devmode == NULL) {
2671 DEBUG(0,("construct_nt_devicemode: malloc fail.\n"));
2672 return NULL;
2675 ZERO_STRUCTP(nt_devmode);
2677 slprintf(adevice, sizeof(adevice), "%s", default_devicename);
2678 fstrcpy(nt_devmode->devicename, adevice);
2680 fstrcpy(nt_devmode->formname, "Letter");
2682 nt_devmode->specversion = 0x0401;
2683 nt_devmode->driverversion = 0x0400;
2684 nt_devmode->size = 0x00DC;
2685 nt_devmode->driverextra = 0x0000;
2686 nt_devmode->fields = FORMNAME | TTOPTION | PRINTQUALITY |
2687 DEFAULTSOURCE | COPIES | SCALE |
2688 PAPERSIZE | ORIENTATION;
2689 nt_devmode->orientation = 1;
2690 nt_devmode->papersize = PAPER_LETTER;
2691 nt_devmode->paperlength = 0;
2692 nt_devmode->paperwidth = 0;
2693 nt_devmode->scale = 0x64;
2694 nt_devmode->copies = 1;
2695 nt_devmode->defaultsource = BIN_FORMSOURCE;
2696 nt_devmode->printquality = RES_HIGH; /* 0x0258 */
2697 nt_devmode->color = COLOR_MONOCHROME;
2698 nt_devmode->duplex = DUP_SIMPLEX;
2699 nt_devmode->yresolution = 0;
2700 nt_devmode->ttoption = TT_SUBDEV;
2701 nt_devmode->collate = COLLATE_FALSE;
2702 nt_devmode->icmmethod = 0;
2703 nt_devmode->icmintent = 0;
2704 nt_devmode->mediatype = 0;
2705 nt_devmode->dithertype = 0;
2707 /* non utilisés par un driver d'imprimante */
2708 nt_devmode->logpixels = 0;
2709 nt_devmode->bitsperpel = 0;
2710 nt_devmode->pelswidth = 0;
2711 nt_devmode->pelsheight = 0;
2712 nt_devmode->displayflags = 0;
2713 nt_devmode->displayfrequency = 0;
2714 nt_devmode->reserved1 = 0;
2715 nt_devmode->reserved2 = 0;
2716 nt_devmode->panningwidth = 0;
2717 nt_devmode->panningheight = 0;
2719 nt_devmode->nt_dev_private = NULL;
2720 return nt_devmode;
2723 /****************************************************************************
2724 Deepcopy an NT devicemode.
2725 ****************************************************************************/
2727 NT_DEVICEMODE *dup_nt_devicemode(NT_DEVICEMODE *nt_devicemode)
2729 NT_DEVICEMODE *new_nt_devicemode = NULL;
2731 if ( !nt_devicemode )
2732 return NULL;
2734 if ((new_nt_devicemode = (NT_DEVICEMODE *)memdup(nt_devicemode, sizeof(NT_DEVICEMODE))) == NULL) {
2735 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2736 return NULL;
2739 new_nt_devicemode->nt_dev_private = NULL;
2740 if (nt_devicemode->nt_dev_private != NULL) {
2741 if ((new_nt_devicemode->nt_dev_private = (uint8 *)memdup(nt_devicemode->nt_dev_private, nt_devicemode->driverextra)) == NULL) {
2742 SAFE_FREE(new_nt_devicemode);
2743 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2744 return NULL;
2748 return new_nt_devicemode;
2751 /****************************************************************************
2752 Clean up and deallocate a (maybe partially) allocated NT_DEVICEMODE.
2753 ****************************************************************************/
2755 void free_nt_devicemode(NT_DEVICEMODE **devmode_ptr)
2757 NT_DEVICEMODE *nt_devmode = *devmode_ptr;
2759 if(nt_devmode == NULL)
2760 return;
2762 DEBUG(106,("free_nt_devicemode: deleting DEVMODE\n"));
2764 SAFE_FREE(nt_devmode->nt_dev_private);
2765 SAFE_FREE(*devmode_ptr);
2768 /****************************************************************************
2769 Clean up and deallocate a (maybe partially) allocated NT_PRINTER_INFO_LEVEL_2.
2770 ****************************************************************************/
2772 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr)
2774 NT_PRINTER_INFO_LEVEL_2 *info = *info_ptr;
2776 if ( !info )
2777 return;
2779 free_nt_devicemode(&info->devmode);
2781 TALLOC_FREE( *info_ptr );
2785 /****************************************************************************
2786 ****************************************************************************/
2787 int unpack_devicemode(NT_DEVICEMODE **nt_devmode, const uint8 *buf, int buflen)
2789 int len = 0;
2790 int extra_len = 0;
2791 NT_DEVICEMODE devmode;
2793 ZERO_STRUCT(devmode);
2795 len += tdb_unpack(buf+len, buflen-len, "p", nt_devmode);
2797 if (!*nt_devmode) return len;
2799 len += tdb_unpack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2800 devmode.devicename,
2801 devmode.formname,
2803 &devmode.specversion,
2804 &devmode.driverversion,
2805 &devmode.size,
2806 &devmode.driverextra,
2807 &devmode.orientation,
2808 &devmode.papersize,
2809 &devmode.paperlength,
2810 &devmode.paperwidth,
2811 &devmode.scale,
2812 &devmode.copies,
2813 &devmode.defaultsource,
2814 &devmode.printquality,
2815 &devmode.color,
2816 &devmode.duplex,
2817 &devmode.yresolution,
2818 &devmode.ttoption,
2819 &devmode.collate,
2820 &devmode.logpixels,
2822 &devmode.fields,
2823 &devmode.bitsperpel,
2824 &devmode.pelswidth,
2825 &devmode.pelsheight,
2826 &devmode.displayflags,
2827 &devmode.displayfrequency,
2828 &devmode.icmmethod,
2829 &devmode.icmintent,
2830 &devmode.mediatype,
2831 &devmode.dithertype,
2832 &devmode.reserved1,
2833 &devmode.reserved2,
2834 &devmode.panningwidth,
2835 &devmode.panningheight,
2836 &devmode.nt_dev_private);
2838 if (devmode.nt_dev_private) {
2839 /* the len in tdb_unpack is an int value and
2840 * devmode.driverextra is only a short
2842 len += tdb_unpack(buf+len, buflen-len, "B", &extra_len, &devmode.nt_dev_private);
2843 devmode.driverextra=(uint16)extra_len;
2845 /* check to catch an invalid TDB entry so we don't segfault */
2846 if (devmode.driverextra == 0) {
2847 devmode.nt_dev_private = NULL;
2851 *nt_devmode = (NT_DEVICEMODE *)memdup(&devmode, sizeof(devmode));
2852 if (!*nt_devmode) {
2853 SAFE_FREE(devmode.nt_dev_private);
2854 return -1;
2857 DEBUG(8,("Unpacked devicemode [%s](%s)\n", devmode.devicename, devmode.formname));
2858 if (devmode.nt_dev_private)
2859 DEBUG(8,("with a private section of %d bytes\n", devmode.driverextra));
2861 return len;
2864 /****************************************************************************
2865 Allocate and initialize a new slot.
2866 ***************************************************************************/
2868 int add_new_printer_key( NT_PRINTER_DATA *data, const char *name )
2870 NT_PRINTER_KEY *d;
2871 int key_index;
2873 if ( !name || !data )
2874 return -1;
2876 /* allocate another slot in the NT_PRINTER_KEY array */
2878 if ( !(d = TALLOC_REALLOC_ARRAY( data, data->keys, NT_PRINTER_KEY, data->num_keys+1)) ) {
2879 DEBUG(0,("add_new_printer_key: Realloc() failed!\n"));
2880 return -1;
2883 data->keys = d;
2885 key_index = data->num_keys;
2887 /* initialze new key */
2889 data->keys[key_index].name = talloc_strdup( data, name );
2891 if ( !(data->keys[key_index].values = TALLOC_ZERO_P( data, REGVAL_CTR )) )
2892 return -1;
2894 data->num_keys++;
2896 DEBUG(10,("add_new_printer_key: Inserted new data key [%s]\n", name ));
2898 return key_index;
2901 /****************************************************************************
2902 search for a registry key name in the existing printer data
2903 ***************************************************************************/
2905 int delete_printer_key( NT_PRINTER_DATA *data, const char *name )
2907 int i;
2909 for ( i=0; i<data->num_keys; i++ ) {
2910 if ( strequal( data->keys[i].name, name ) ) {
2912 /* cleanup memory */
2914 TALLOC_FREE( data->keys[i].name );
2915 TALLOC_FREE( data->keys[i].values );
2917 /* if not the end of the array, move remaining elements down one slot */
2919 data->num_keys--;
2920 if ( data->num_keys && (i < data->num_keys) )
2921 memmove( &data->keys[i], &data->keys[i+1], sizeof(NT_PRINTER_KEY)*(data->num_keys-i) );
2923 break;
2928 return data->num_keys;
2931 /****************************************************************************
2932 search for a registry key name in the existing printer data
2933 ***************************************************************************/
2935 int lookup_printerkey( NT_PRINTER_DATA *data, const char *name )
2937 int key_index = -1;
2938 int i;
2940 if ( !data || !name )
2941 return -1;
2943 DEBUG(12,("lookup_printerkey: Looking for [%s]\n", name));
2945 /* loop over all existing keys */
2947 for ( i=0; i<data->num_keys; i++ ) {
2948 if ( strequal(data->keys[i].name, name) ) {
2949 DEBUG(12,("lookup_printerkey: Found [%s]!\n", name));
2950 key_index = i;
2951 break;
2956 return key_index;
2959 /****************************************************************************
2960 ***************************************************************************/
2962 int get_printer_subkeys( NT_PRINTER_DATA *data, const char* key, fstring **subkeys )
2964 int i, j;
2965 int key_len;
2966 int num_subkeys = 0;
2967 char *p;
2968 fstring *subkeys_ptr = NULL;
2969 fstring subkeyname;
2971 *subkeys = NULL;
2973 if ( !data )
2974 return 0;
2976 if ( !key )
2977 return -1;
2979 /* special case of asking for the top level printer data registry key names */
2981 if ( strlen(key) == 0 ) {
2982 for ( i=0; i<data->num_keys; i++ ) {
2984 /* found a match, so allocate space and copy the name */
2986 if ( !(subkeys_ptr = SMB_REALLOC_ARRAY( subkeys_ptr, fstring, num_subkeys+2)) ) {
2987 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
2988 num_subkeys+1));
2989 return -1;
2992 fstrcpy( subkeys_ptr[num_subkeys], data->keys[i].name );
2993 num_subkeys++;
2996 goto done;
2999 /* asking for the subkeys of some key */
3000 /* subkey paths are stored in the key name using '\' as the delimiter */
3002 for ( i=0; i<data->num_keys; i++ ) {
3003 if ( StrnCaseCmp(data->keys[i].name, key, strlen(key)) == 0 ) {
3005 /* if we found the exact key, then break */
3006 key_len = strlen( key );
3007 if ( strlen(data->keys[i].name) == key_len )
3008 break;
3010 /* get subkey path */
3012 p = data->keys[i].name + key_len;
3013 if ( *p == '\\' )
3014 p++;
3015 fstrcpy( subkeyname, p );
3016 if ( (p = strchr( subkeyname, '\\' )) )
3017 *p = '\0';
3019 /* don't add a key more than once */
3021 for ( j=0; j<num_subkeys; j++ ) {
3022 if ( strequal( subkeys_ptr[j], subkeyname ) )
3023 break;
3026 if ( j != num_subkeys )
3027 continue;
3029 /* found a match, so allocate space and copy the name */
3031 if ( !(subkeys_ptr = SMB_REALLOC_ARRAY( subkeys_ptr, fstring, num_subkeys+2)) ) {
3032 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
3033 num_subkeys+1));
3034 return 0;
3037 fstrcpy( subkeys_ptr[num_subkeys], subkeyname );
3038 num_subkeys++;
3043 /* return error if the key was not found */
3045 if ( i == data->num_keys ) {
3046 SAFE_FREE(subkeys_ptr);
3047 return -1;
3050 done:
3051 /* tag off the end */
3053 if (num_subkeys)
3054 fstrcpy(subkeys_ptr[num_subkeys], "" );
3056 *subkeys = subkeys_ptr;
3058 return num_subkeys;
3061 #ifdef HAVE_ADS
3062 static void map_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3063 const char *sz)
3065 smb_ucs2_t conv_str[1024];
3066 size_t str_size;
3068 regval_ctr_delvalue(ctr, val_name);
3069 str_size = push_ucs2(NULL, conv_str, sz, sizeof(conv_str),
3070 STR_TERMINATE | STR_NOALIGN);
3071 regval_ctr_addvalue(ctr, val_name, REG_SZ,
3072 (char *) conv_str, str_size);
3075 static void map_dword_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3076 uint32 dword)
3078 regval_ctr_delvalue(ctr, val_name);
3079 regval_ctr_addvalue(ctr, val_name, REG_DWORD,
3080 (char *) &dword, sizeof(dword));
3083 static void map_bool_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3084 bool b)
3086 uint8 bin_bool = (b ? 1 : 0);
3087 regval_ctr_delvalue(ctr, val_name);
3088 regval_ctr_addvalue(ctr, val_name, REG_BINARY,
3089 (char *) &bin_bool, sizeof(bin_bool));
3092 static void map_single_multi_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3093 const char *multi_sz)
3095 smb_ucs2_t *conv_strs = NULL;
3096 size_t str_size;
3098 /* a multi-sz has to have a null string terminator, i.e., the last
3099 string must be followed by two nulls */
3100 str_size = strlen(multi_sz) + 2;
3101 conv_strs = SMB_CALLOC_ARRAY(smb_ucs2_t, str_size);
3102 if (!conv_strs) {
3103 return;
3106 /* Change to byte units. */
3107 str_size *= sizeof(smb_ucs2_t);
3108 push_ucs2(NULL, conv_strs, multi_sz, str_size,
3109 STR_TERMINATE | STR_NOALIGN);
3111 regval_ctr_delvalue(ctr, val_name);
3112 regval_ctr_addvalue(ctr, val_name, REG_MULTI_SZ,
3113 (char *) conv_strs, str_size);
3114 safe_free(conv_strs);
3118 /****************************************************************************
3119 * Map the NT_PRINTER_INFO_LEVEL_2 data into DsSpooler keys for publishing.
3121 * @param info2 NT_PRINTER_INFO_LEVEL_2 describing printer - gets modified
3122 * @return bool indicating success or failure
3123 ***************************************************************************/
3125 static bool map_nt_printer_info2_to_dsspooler(NT_PRINTER_INFO_LEVEL_2 *info2)
3127 REGVAL_CTR *ctr = NULL;
3128 fstring longname;
3129 const char *dnssuffix;
3130 char *allocated_string = NULL;
3131 const char *ascii_str;
3132 int i;
3134 if ((i = lookup_printerkey(info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
3135 i = add_new_printer_key(info2->data, SPOOL_DSSPOOLER_KEY);
3136 ctr = info2->data->keys[i].values;
3138 map_sz_into_ctr(ctr, SPOOL_REG_PRINTERNAME, info2->sharename);
3139 map_sz_into_ctr(ctr, SPOOL_REG_SHORTSERVERNAME, global_myname());
3141 /* we make the assumption that the netbios name is the same
3142 as the DNS name sinc ethe former will be what we used to
3143 join the domain */
3145 dnssuffix = get_mydnsdomname(talloc_tos());
3146 if (dnssuffix && *dnssuffix) {
3147 fstr_sprintf( longname, "%s.%s", global_myname(), dnssuffix );
3148 } else {
3149 fstrcpy( longname, global_myname() );
3152 map_sz_into_ctr(ctr, SPOOL_REG_SERVERNAME, longname);
3154 asprintf(&allocated_string, "\\\\%s\\%s", longname, info2->sharename);
3155 map_sz_into_ctr(ctr, SPOOL_REG_UNCNAME, allocated_string);
3156 SAFE_FREE(allocated_string);
3158 map_dword_into_ctr(ctr, SPOOL_REG_VERSIONNUMBER, 4);
3159 map_sz_into_ctr(ctr, SPOOL_REG_DRIVERNAME, info2->drivername);
3160 map_sz_into_ctr(ctr, SPOOL_REG_LOCATION, info2->location);
3161 map_sz_into_ctr(ctr, SPOOL_REG_DESCRIPTION, info2->comment);
3162 map_single_multi_sz_into_ctr(ctr, SPOOL_REG_PORTNAME, info2->portname);
3163 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSEPARATORFILE, info2->sepfile);
3164 map_dword_into_ctr(ctr, SPOOL_REG_PRINTSTARTTIME, info2->starttime);
3165 map_dword_into_ctr(ctr, SPOOL_REG_PRINTENDTIME, info2->untiltime);
3166 map_dword_into_ctr(ctr, SPOOL_REG_PRIORITY, info2->priority);
3168 map_bool_into_ctr(ctr, SPOOL_REG_PRINTKEEPPRINTEDJOBS,
3169 (info2->attributes &
3170 PRINTER_ATTRIBUTE_KEEPPRINTEDJOBS));
3172 switch (info2->attributes & 0x3) {
3173 case 0:
3174 ascii_str = SPOOL_REGVAL_PRINTWHILESPOOLING;
3175 break;
3176 case 1:
3177 ascii_str = SPOOL_REGVAL_PRINTAFTERSPOOLED;
3178 break;
3179 case 2:
3180 ascii_str = SPOOL_REGVAL_PRINTDIRECT;
3181 break;
3182 default:
3183 ascii_str = "unknown";
3185 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSPOOLING, ascii_str);
3187 return True;
3190 /*****************************************************************
3191 ****************************************************************/
3193 static void store_printer_guid(NT_PRINTER_INFO_LEVEL_2 *info2,
3194 struct GUID guid)
3196 int i;
3197 REGVAL_CTR *ctr=NULL;
3198 UNISTR2 unistr_guid;
3200 /* find the DsSpooler key */
3201 if ((i = lookup_printerkey(info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
3202 i = add_new_printer_key(info2->data, SPOOL_DSSPOOLER_KEY);
3203 ctr = info2->data->keys[i].values;
3205 regval_ctr_delvalue(ctr, "objectGUID");
3207 /* We used to store this as a REG_BINARY but that causes
3208 Vista to whine */
3210 ZERO_STRUCT( unistr_guid );
3212 init_unistr2( &unistr_guid, smb_uuid_string(talloc_tos(), guid),
3213 UNI_STR_TERMINATE );
3215 regval_ctr_addvalue(ctr, "objectGUID", REG_SZ,
3216 (char *)unistr_guid.buffer,
3217 unistr_guid.uni_max_len*2);
3221 static WERROR nt_printer_publish_ads(ADS_STRUCT *ads,
3222 NT_PRINTER_INFO_LEVEL *printer)
3224 ADS_STATUS ads_rc;
3225 LDAPMessage *res;
3226 char *prt_dn = NULL, *srv_dn, *srv_cn_0, *srv_cn_escaped, *sharename_escaped;
3227 char *srv_dn_utf8, **srv_cn_utf8;
3228 TALLOC_CTX *ctx;
3229 ADS_MODLIST mods;
3230 const char *attrs[] = {"objectGUID", NULL};
3231 struct GUID guid;
3232 WERROR win_rc = WERR_OK;
3234 DEBUG(5, ("publishing printer %s\n", printer->info_2->printername));
3236 /* figure out where to publish */
3237 ads_find_machine_acct(ads, &res, global_myname());
3239 /* We use ldap_get_dn here as we need the answer
3240 * in utf8 to call ldap_explode_dn(). JRA. */
3242 srv_dn_utf8 = ldap_get_dn((LDAP *)ads->ldap.ld, (LDAPMessage *)res);
3243 if (!srv_dn_utf8) {
3244 ads_destroy(&ads);
3245 return WERR_SERVER_UNAVAILABLE;
3247 ads_msgfree(ads, res);
3248 srv_cn_utf8 = ldap_explode_dn(srv_dn_utf8, 1);
3249 if (!srv_cn_utf8) {
3250 ldap_memfree(srv_dn_utf8);
3251 ads_destroy(&ads);
3252 return WERR_SERVER_UNAVAILABLE;
3254 /* Now convert to CH_UNIX. */
3255 if (pull_utf8_allocate(&srv_dn, srv_dn_utf8) == (size_t)-1) {
3256 ldap_memfree(srv_dn_utf8);
3257 ldap_memfree(srv_cn_utf8);
3258 ads_destroy(&ads);
3259 return WERR_SERVER_UNAVAILABLE;
3261 if (pull_utf8_allocate(&srv_cn_0, srv_cn_utf8[0]) == (size_t)-1) {
3262 ldap_memfree(srv_dn_utf8);
3263 ldap_memfree(srv_cn_utf8);
3264 ads_destroy(&ads);
3265 SAFE_FREE(srv_dn);
3266 return WERR_SERVER_UNAVAILABLE;
3269 ldap_memfree(srv_dn_utf8);
3270 ldap_memfree(srv_cn_utf8);
3272 srv_cn_escaped = escape_rdn_val_string_alloc(srv_cn_0);
3273 if (!srv_cn_escaped) {
3274 SAFE_FREE(srv_cn_0);
3275 ldap_memfree(srv_dn_utf8);
3276 ads_destroy(&ads);
3277 return WERR_SERVER_UNAVAILABLE;
3279 sharename_escaped = escape_rdn_val_string_alloc(printer->info_2->sharename);
3280 if (!sharename_escaped) {
3281 SAFE_FREE(srv_cn_escaped);
3282 SAFE_FREE(srv_cn_0);
3283 ldap_memfree(srv_dn_utf8);
3284 ads_destroy(&ads);
3285 return WERR_SERVER_UNAVAILABLE;
3289 asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn_escaped, sharename_escaped, srv_dn);
3291 SAFE_FREE(srv_dn);
3292 SAFE_FREE(srv_cn_0);
3293 SAFE_FREE(srv_cn_escaped);
3294 SAFE_FREE(sharename_escaped);
3296 /* build the ads mods */
3297 ctx = talloc_init("nt_printer_publish_ads");
3298 if (ctx == NULL) {
3299 SAFE_FREE(prt_dn);
3300 return WERR_NOMEM;
3303 mods = ads_init_mods(ctx);
3305 if (mods == NULL) {
3306 SAFE_FREE(prt_dn);
3307 talloc_destroy(ctx);
3308 return WERR_NOMEM;
3311 get_local_printer_publishing_data(ctx, &mods, printer->info_2->data);
3312 ads_mod_str(ctx, &mods, SPOOL_REG_PRINTERNAME,
3313 printer->info_2->sharename);
3315 /* publish it */
3316 ads_rc = ads_mod_printer_entry(ads, prt_dn, ctx, &mods);
3317 if (ads_rc.err.rc == LDAP_NO_SUCH_OBJECT)
3318 ads_rc = ads_add_printer_entry(ads, prt_dn, ctx, &mods);
3320 if (!ADS_ERR_OK(ads_rc))
3321 DEBUG(3, ("error publishing %s: %s\n", printer->info_2->sharename, ads_errstr(ads_rc)));
3323 talloc_destroy(ctx);
3325 /* retreive the guid and store it locally */
3326 if (ADS_ERR_OK(ads_search_dn(ads, &res, prt_dn, attrs))) {
3327 ZERO_STRUCT(guid);
3328 ads_pull_guid(ads, res, &guid);
3329 ads_msgfree(ads, res);
3330 store_printer_guid(printer->info_2, guid);
3331 win_rc = mod_a_printer(printer, 2);
3334 SAFE_FREE(prt_dn);
3335 return win_rc;
3338 static WERROR nt_printer_unpublish_ads(ADS_STRUCT *ads,
3339 NT_PRINTER_INFO_LEVEL *printer)
3341 ADS_STATUS ads_rc;
3342 LDAPMessage *res;
3343 char *prt_dn = NULL;
3345 DEBUG(5, ("unpublishing printer %s\n", printer->info_2->printername));
3347 /* remove the printer from the directory */
3348 ads_rc = ads_find_printer_on_server(ads, &res,
3349 printer->info_2->sharename, global_myname());
3351 if (ADS_ERR_OK(ads_rc) && ads_count_replies(ads, res)) {
3352 prt_dn = ads_get_dn(ads, res);
3353 if (!prt_dn) {
3354 ads_msgfree(ads, res);
3355 return WERR_NOMEM;
3357 ads_rc = ads_del_dn(ads, prt_dn);
3358 ads_memfree(ads, prt_dn);
3361 ads_msgfree(ads, res);
3362 return WERR_OK;
3365 /****************************************************************************
3366 * Publish a printer in the directory
3368 * @param snum describing printer service
3369 * @return WERROR indicating status of publishing
3370 ***************************************************************************/
3372 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
3374 ADS_STATUS ads_rc;
3375 ADS_STRUCT *ads = NULL;
3376 NT_PRINTER_INFO_LEVEL *printer = NULL;
3377 WERROR win_rc;
3379 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
3380 if (!W_ERROR_IS_OK(win_rc))
3381 goto done;
3383 switch (action) {
3384 case SPOOL_DS_PUBLISH:
3385 case SPOOL_DS_UPDATE:
3386 /* set the DsSpooler info and attributes */
3387 if (!(map_nt_printer_info2_to_dsspooler(printer->info_2))) {
3388 win_rc = WERR_NOMEM;
3389 goto done;
3392 printer->info_2->attributes |= PRINTER_ATTRIBUTE_PUBLISHED;
3393 break;
3394 case SPOOL_DS_UNPUBLISH:
3395 printer->info_2->attributes ^= PRINTER_ATTRIBUTE_PUBLISHED;
3396 break;
3397 default:
3398 win_rc = WERR_NOT_SUPPORTED;
3399 goto done;
3402 win_rc = mod_a_printer(printer, 2);
3403 if (!W_ERROR_IS_OK(win_rc)) {
3404 DEBUG(3, ("err %d saving data\n", W_ERROR_V(win_rc)));
3405 goto done;
3408 ads = ads_init(lp_realm(), lp_workgroup(), NULL);
3409 if (!ads) {
3410 DEBUG(3, ("ads_init() failed\n"));
3411 win_rc = WERR_SERVER_UNAVAILABLE;
3412 goto done;
3414 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
3415 SAFE_FREE(ads->auth.password);
3416 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
3417 NULL, NULL);
3419 /* ads_connect() will find the DC for us */
3420 ads_rc = ads_connect(ads);
3421 if (!ADS_ERR_OK(ads_rc)) {
3422 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
3423 win_rc = WERR_ACCESS_DENIED;
3424 goto done;
3427 switch (action) {
3428 case SPOOL_DS_PUBLISH:
3429 case SPOOL_DS_UPDATE:
3430 win_rc = nt_printer_publish_ads(ads, printer);
3431 break;
3432 case SPOOL_DS_UNPUBLISH:
3433 win_rc = nt_printer_unpublish_ads(ads, printer);
3434 break;
3437 done:
3438 free_a_printer(&printer, 2);
3439 ads_destroy(&ads);
3440 return win_rc;
3443 WERROR check_published_printers(void)
3445 ADS_STATUS ads_rc;
3446 ADS_STRUCT *ads = NULL;
3447 int snum;
3448 int n_services = lp_numservices();
3449 NT_PRINTER_INFO_LEVEL *printer = NULL;
3451 ads = ads_init(lp_realm(), lp_workgroup(), NULL);
3452 if (!ads) {
3453 DEBUG(3, ("ads_init() failed\n"));
3454 return WERR_SERVER_UNAVAILABLE;
3456 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
3457 SAFE_FREE(ads->auth.password);
3458 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
3459 NULL, NULL);
3461 /* ads_connect() will find the DC for us */
3462 ads_rc = ads_connect(ads);
3463 if (!ADS_ERR_OK(ads_rc)) {
3464 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
3465 ads_destroy(&ads);
3466 ads_kdestroy("MEMORY:prtpub_cache");
3467 return WERR_ACCESS_DENIED;
3470 for (snum = 0; snum < n_services; snum++) {
3471 if (!(lp_snum_ok(snum) && lp_print_ok(snum)))
3472 continue;
3474 if (W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2,
3475 lp_servicename(snum))) &&
3476 (printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED))
3477 nt_printer_publish_ads(ads, printer);
3479 free_a_printer(&printer, 2);
3482 ads_destroy(&ads);
3483 ads_kdestroy("MEMORY:prtpub_cache");
3484 return WERR_OK;
3487 bool is_printer_published(Printer_entry *print_hnd, int snum,
3488 struct GUID *guid)
3490 NT_PRINTER_INFO_LEVEL *printer = NULL;
3491 REGVAL_CTR *ctr;
3492 REGISTRY_VALUE *guid_val;
3493 WERROR win_rc;
3494 int i;
3495 bool ret = False;
3497 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
3499 if (!W_ERROR_IS_OK(win_rc) ||
3500 !(printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED) ||
3501 ((i = lookup_printerkey(printer->info_2->data, SPOOL_DSSPOOLER_KEY)) < 0) ||
3502 !(ctr = printer->info_2->data->keys[i].values) ||
3503 !(guid_val = regval_ctr_getvalue(ctr, "objectGUID")))
3505 free_a_printer(&printer, 2);
3506 return False;
3509 /* fetching printer guids really ought to be a separate function. */
3511 if ( guid ) {
3512 fstring guid_str;
3514 /* We used to store the guid as REG_BINARY, then swapped
3515 to REG_SZ for Vista compatibility so check for both */
3517 switch ( regval_type(guid_val) ){
3518 case REG_SZ:
3519 rpcstr_pull( guid_str, regval_data_p(guid_val),
3520 sizeof(guid_str)-1, -1, STR_TERMINATE );
3521 ret = smb_string_to_uuid( guid_str, guid );
3522 break;
3523 case REG_BINARY:
3524 if ( regval_size(guid_val) != sizeof(struct GUID) ) {
3525 ret = False;
3526 break;
3528 memcpy(guid, regval_data_p(guid_val), sizeof(struct GUID));
3529 break;
3530 default:
3531 DEBUG(0,("is_printer_published: GUID value stored as "
3532 "invaluid type (%d)\n", regval_type(guid_val) ));
3533 break;
3537 free_a_printer(&printer, 2);
3538 return ret;
3540 #else
3541 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
3543 return WERR_OK;
3546 WERROR check_published_printers(void)
3548 return WERR_OK;
3551 bool is_printer_published(Printer_entry *print_hnd, int snum,
3552 struct GUID *guid)
3554 return False;
3556 #endif /* HAVE_ADS */
3558 /****************************************************************************
3559 ***************************************************************************/
3561 WERROR delete_all_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key )
3563 NT_PRINTER_DATA *data;
3564 int i;
3565 int removed_keys = 0;
3566 int empty_slot;
3568 data = p2->data;
3569 empty_slot = data->num_keys;
3571 if ( !key )
3572 return WERR_INVALID_PARAM;
3574 /* remove all keys */
3576 if ( !strlen(key) ) {
3578 TALLOC_FREE( data );
3580 p2->data = NULL;
3582 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from printer [%s]\n",
3583 p2->printername ));
3585 return WERR_OK;
3588 /* remove a specific key (and all subkeys) */
3590 for ( i=0; i<data->num_keys; i++ ) {
3591 if ( StrnCaseCmp( data->keys[i].name, key, strlen(key)) == 0 ) {
3592 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
3593 data->keys[i].name));
3595 TALLOC_FREE( data->keys[i].name );
3596 TALLOC_FREE( data->keys[i].values );
3598 /* mark the slot as empty */
3600 ZERO_STRUCTP( &data->keys[i] );
3604 /* find the first empty slot */
3606 for ( i=0; i<data->num_keys; i++ ) {
3607 if ( !data->keys[i].name ) {
3608 empty_slot = i;
3609 removed_keys++;
3610 break;
3614 if ( i == data->num_keys )
3615 /* nothing was removed */
3616 return WERR_INVALID_PARAM;
3618 /* move everything down */
3620 for ( i=empty_slot+1; i<data->num_keys; i++ ) {
3621 if ( data->keys[i].name ) {
3622 memcpy( &data->keys[empty_slot], &data->keys[i], sizeof(NT_PRINTER_KEY) );
3623 ZERO_STRUCTP( &data->keys[i] );
3624 empty_slot++;
3625 removed_keys++;
3629 /* update count */
3631 data->num_keys -= removed_keys;
3633 /* sanity check to see if anything is left */
3635 if ( !data->num_keys ) {
3636 DEBUG(8,("delete_all_printer_data: No keys left for printer [%s]\n", p2->printername ));
3638 SAFE_FREE( data->keys );
3639 ZERO_STRUCTP( data );
3642 return WERR_OK;
3645 /****************************************************************************
3646 ***************************************************************************/
3648 WERROR delete_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3650 WERROR result = WERR_OK;
3651 int key_index;
3653 /* we must have names on non-zero length */
3655 if ( !key || !*key|| !value || !*value )
3656 return WERR_INVALID_NAME;
3658 /* find the printer key first */
3660 key_index = lookup_printerkey( p2->data, key );
3661 if ( key_index == -1 )
3662 return WERR_OK;
3664 /* make sure the value exists so we can return the correct error code */
3666 if ( !regval_ctr_getvalue( p2->data->keys[key_index].values, value ) )
3667 return WERR_BADFILE;
3669 regval_ctr_delvalue( p2->data->keys[key_index].values, value );
3671 DEBUG(8,("delete_printer_data: Removed key => [%s], value => [%s]\n",
3672 key, value ));
3674 return result;
3677 /****************************************************************************
3678 ***************************************************************************/
3680 WERROR add_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value,
3681 uint32 type, uint8 *data, int real_len )
3683 WERROR result = WERR_OK;
3684 int key_index;
3686 /* we must have names on non-zero length */
3688 if ( !key || !*key|| !value || !*value )
3689 return WERR_INVALID_NAME;
3691 /* find the printer key first */
3693 key_index = lookup_printerkey( p2->data, key );
3694 if ( key_index == -1 )
3695 key_index = add_new_printer_key( p2->data, key );
3697 if ( key_index == -1 )
3698 return WERR_NOMEM;
3700 regval_ctr_addvalue( p2->data->keys[key_index].values, value,
3701 type, (const char *)data, real_len );
3703 DEBUG(8,("add_printer_data: Added key => [%s], value => [%s], type=> [%d], size => [%d]\n",
3704 key, value, type, real_len ));
3706 return result;
3709 /****************************************************************************
3710 ***************************************************************************/
3712 REGISTRY_VALUE* get_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3714 int key_index;
3716 if ( (key_index = lookup_printerkey( p2->data, key )) == -1 )
3717 return NULL;
3719 DEBUG(8,("get_printer_data: Attempting to lookup key => [%s], value => [%s]\n",
3720 key, value ));
3722 return regval_ctr_getvalue( p2->data->keys[key_index].values, value );
3725 /****************************************************************************
3726 Unpack a list of registry values frem the TDB
3727 ***************************************************************************/
3729 static int unpack_values(NT_PRINTER_DATA *printer_data, const uint8 *buf, int buflen)
3731 int len = 0;
3732 uint32 type;
3733 fstring string;
3734 const char *valuename = NULL;
3735 const char *keyname = NULL;
3736 char *str;
3737 int size;
3738 uint8 *data_p;
3739 REGISTRY_VALUE *regval_p;
3740 int key_index;
3742 /* add the "PrinterDriverData" key first for performance reasons */
3744 add_new_printer_key( printer_data, SPOOL_PRINTERDATA_KEY );
3746 /* loop and unpack the rest of the registry values */
3748 while ( True ) {
3750 /* check to see if there are any more registry values */
3752 regval_p = NULL;
3753 len += tdb_unpack(buf+len, buflen-len, "p", &regval_p);
3754 if ( !regval_p )
3755 break;
3757 /* unpack the next regval */
3759 len += tdb_unpack(buf+len, buflen-len, "fdB",
3760 string,
3761 &type,
3762 &size,
3763 &data_p);
3765 /* lookup for subkey names which have a type of REG_NONE */
3766 /* there's no data with this entry */
3768 if ( type == REG_NONE ) {
3769 if ( (key_index=lookup_printerkey( printer_data, string)) == -1 )
3770 add_new_printer_key( printer_data, string );
3771 continue;
3775 * break of the keyname from the value name.
3776 * Valuenames can have embedded '\'s so be careful.
3777 * only support one level of keys. See the
3778 * "Konica Fiery S300 50C-K v1.1. enu" 2k driver.
3779 * -- jerry
3782 str = strchr_m( string, '\\');
3784 /* Put in "PrinterDriverData" is no key specified */
3786 if ( !str ) {
3787 keyname = SPOOL_PRINTERDATA_KEY;
3788 valuename = string;
3790 else {
3791 *str = '\0';
3792 keyname = string;
3793 valuename = str+1;
3796 /* see if we need a new key */
3798 if ( (key_index=lookup_printerkey( printer_data, keyname )) == -1 )
3799 key_index = add_new_printer_key( printer_data, keyname );
3801 if ( key_index == -1 ) {
3802 DEBUG(0,("unpack_values: Failed to allocate a new key [%s]!\n",
3803 keyname));
3804 break;
3807 DEBUG(8,("specific: [%s:%s], len: %d\n", keyname, valuename, size));
3809 /* Vista doesn't like unknown REG_BINARY values in DsSpooler.
3810 Thanks to Martin Zielinski for the hint. */
3812 if ( type == REG_BINARY &&
3813 strequal( keyname, SPOOL_DSSPOOLER_KEY ) &&
3814 strequal( valuename, "objectGUID" ) )
3816 struct GUID guid;
3817 UNISTR2 unistr_guid;
3819 ZERO_STRUCT( unistr_guid );
3821 /* convert the GUID to a UNICODE string */
3823 memcpy( &guid, data_p, sizeof(struct GUID) );
3825 init_unistr2( &unistr_guid,
3826 smb_uuid_string(talloc_tos(), guid),
3827 UNI_STR_TERMINATE );
3829 regval_ctr_addvalue( printer_data->keys[key_index].values,
3830 valuename, REG_SZ,
3831 (const char *)unistr_guid.buffer,
3832 unistr_guid.uni_str_len*2 );
3834 } else {
3835 /* add the value */
3837 regval_ctr_addvalue( printer_data->keys[key_index].values,
3838 valuename, type, (const char *)data_p,
3839 size );
3842 SAFE_FREE(data_p); /* 'B' option to tdbpack does a malloc() */
3846 return len;
3849 /****************************************************************************
3850 ***************************************************************************/
3852 static void map_to_os2_driver(fstring drivername)
3854 static bool initialised=False;
3855 static fstring last_from,last_to;
3856 char *mapfile = lp_os2_driver_map();
3857 char **lines = NULL;
3858 int numlines = 0;
3859 int i;
3861 if (!strlen(drivername))
3862 return;
3864 if (!*mapfile)
3865 return;
3867 if (!initialised) {
3868 *last_from = *last_to = 0;
3869 initialised = True;
3872 if (strequal(drivername,last_from)) {
3873 DEBUG(3,("Mapped Windows driver %s to OS/2 driver %s\n",drivername,last_to));
3874 fstrcpy(drivername,last_to);
3875 return;
3878 lines = file_lines_load(mapfile, &numlines,0);
3879 if (numlines == 0 || lines == NULL) {
3880 DEBUG(0,("No entries in OS/2 driver map %s\n",mapfile));
3881 SAFE_FREE(lines);
3882 return;
3885 DEBUG(4,("Scanning OS/2 driver map %s\n",mapfile));
3887 for( i = 0; i < numlines; i++) {
3888 char *nt_name = lines[i];
3889 char *os2_name = strchr(nt_name,'=');
3891 if (!os2_name)
3892 continue;
3894 *os2_name++ = 0;
3896 while (isspace(*nt_name))
3897 nt_name++;
3899 if (!*nt_name || strchr("#;",*nt_name))
3900 continue;
3903 int l = strlen(nt_name);
3904 while (l && isspace(nt_name[l-1])) {
3905 nt_name[l-1] = 0;
3906 l--;
3910 while (isspace(*os2_name))
3911 os2_name++;
3914 int l = strlen(os2_name);
3915 while (l && isspace(os2_name[l-1])) {
3916 os2_name[l-1] = 0;
3917 l--;
3921 if (strequal(nt_name,drivername)) {
3922 DEBUG(3,("Mapped windows driver %s to os2 driver%s\n",drivername,os2_name));
3923 fstrcpy(last_from,drivername);
3924 fstrcpy(last_to,os2_name);
3925 fstrcpy(drivername,os2_name);
3926 file_lines_free(lines);
3927 return;
3931 file_lines_free(lines);
3934 /****************************************************************************
3935 Get a default printer info 2 struct.
3936 ****************************************************************************/
3937 static WERROR get_a_printer_2_default(NT_PRINTER_INFO_LEVEL_2 *info, const char *servername, const char* sharename)
3939 int snum = lp_servicenumber(sharename);
3941 slprintf(info->servername, sizeof(info->servername)-1, "\\\\%s", servername);
3942 slprintf(info->printername, sizeof(info->printername)-1, "\\\\%s\\%s",
3943 servername, sharename);
3944 fstrcpy(info->sharename, sharename);
3945 fstrcpy(info->portname, SAMBA_PRINTER_PORT_NAME);
3947 /* by setting the driver name to an empty string, a local NT admin
3948 can now run the **local** APW to install a local printer driver
3949 for a Samba shared printer in 2.2. Without this, drivers **must** be
3950 installed on the Samba server for NT clients --jerry */
3951 #if 0 /* JERRY --do not uncomment-- */
3952 if (!*info->drivername)
3953 fstrcpy(info->drivername, "NO DRIVER AVAILABLE FOR THIS PRINTER");
3954 #endif
3957 DEBUG(10,("get_a_printer_2_default: driver name set to [%s]\n", info->drivername));
3959 strlcpy(info->comment, "", sizeof(info->comment));
3960 fstrcpy(info->printprocessor, "winprint");
3961 fstrcpy(info->datatype, "RAW");
3963 #ifdef HAVE_CUPS
3964 if ( (enum printing_types)lp_printing(snum) == PRINT_CUPS ) {
3965 /* Pull the location and comment strings from cups if we don't
3966 already have one */
3967 if ( !strlen(info->location) || !strlen(info->comment) )
3968 cups_pull_comment_location( info );
3970 #endif
3972 info->attributes = PRINTER_ATTRIBUTE_SAMBA;
3974 info->starttime = 0; /* Minutes since 12:00am GMT */
3975 info->untiltime = 0; /* Minutes since 12:00am GMT */
3976 info->priority = 1;
3977 info->default_priority = 1;
3978 info->setuptime = (uint32)time(NULL);
3981 * I changed this as I think it is better to have a generic
3982 * DEVMODE than to crash Win2k explorer.exe --jerry
3983 * See the HP Deskjet 990c Win2k drivers for an example.
3985 * However the default devmode appears to cause problems
3986 * with the HP CLJ 8500 PCL driver. Hence the addition of
3987 * the "default devmode" parameter --jerry 22/01/2002
3990 if (lp_default_devmode(snum)) {
3991 if ((info->devmode = construct_nt_devicemode(info->printername)) == NULL) {
3992 goto fail;
3994 } else {
3995 info->devmode = NULL;
3998 if (!nt_printing_getsec(info, sharename, &info->secdesc_buf)) {
3999 goto fail;
4002 return WERR_OK;
4004 fail:
4005 if (info->devmode)
4006 free_nt_devicemode(&info->devmode);
4008 return WERR_ACCESS_DENIED;
4011 /****************************************************************************
4012 ****************************************************************************/
4013 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info, const char *servername, const char *sharename)
4015 int len = 0;
4016 int snum = lp_servicenumber(sharename);
4017 TDB_DATA kbuf, dbuf;
4018 fstring printername;
4019 char adevice[MAXDEVICENAME];
4020 char *comment = NULL;
4022 kbuf = make_printer_tdbkey(talloc_tos(), sharename);
4024 dbuf = tdb_fetch(tdb_printers, kbuf);
4025 if (!dbuf.dptr) {
4026 return get_a_printer_2_default(info, servername, sharename);
4029 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "dddddddddddfffffPfffff",
4030 &info->attributes,
4031 &info->priority,
4032 &info->default_priority,
4033 &info->starttime,
4034 &info->untiltime,
4035 &info->status,
4036 &info->cjobs,
4037 &info->averageppm,
4038 &info->changeid,
4039 &info->c_setprinter,
4040 &info->setuptime,
4041 info->servername,
4042 info->printername,
4043 info->sharename,
4044 info->portname,
4045 info->drivername,
4046 &comment,
4047 info->location,
4048 info->sepfile,
4049 info->printprocessor,
4050 info->datatype,
4051 info->parameters);
4053 if (comment) {
4054 strlcpy(info->comment, comment, sizeof(info->comment));
4055 SAFE_FREE(comment);
4058 /* Samba has to have shared raw drivers. */
4059 info->attributes |= PRINTER_ATTRIBUTE_SAMBA;
4060 info->attributes &= ~PRINTER_ATTRIBUTE_NOT_SAMBA;
4062 /* Restore the stripped strings. */
4063 slprintf(info->servername, sizeof(info->servername)-1, "\\\\%s", servername);
4065 if ( lp_force_printername(snum) ) {
4066 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, sharename );
4067 } else {
4068 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, info->printername);
4071 fstrcpy(info->printername, printername);
4073 #ifdef HAVE_CUPS
4074 if ( (enum printing_types)lp_printing(snum) == PRINT_CUPS ) {
4075 /* Pull the location and comment strings from cups if we don't
4076 already have one */
4077 if ( !strlen(info->location) || !strlen(info->comment) )
4078 cups_pull_comment_location( info );
4080 #endif
4082 len += unpack_devicemode(&info->devmode,dbuf.dptr+len, dbuf.dsize-len);
4085 * Some client drivers freak out if there is a NULL devmode
4086 * (probably the driver is not checking before accessing
4087 * the devmode pointer) --jerry
4089 * See comments in get_a_printer_2_default()
4092 if (lp_default_devmode(snum) && !info->devmode) {
4093 DEBUG(8,("get_a_printer_2: Constructing a default device mode for [%s]\n",
4094 printername));
4095 info->devmode = construct_nt_devicemode(printername);
4098 slprintf( adevice, sizeof(adevice), "%s", info->printername );
4099 if (info->devmode) {
4100 fstrcpy(info->devmode->devicename, adevice);
4103 if ( !(info->data = TALLOC_ZERO_P( info, NT_PRINTER_DATA )) ) {
4104 DEBUG(0,("unpack_values: talloc() failed!\n"));
4105 SAFE_FREE(dbuf.dptr);
4106 return WERR_NOMEM;
4108 len += unpack_values( info->data, dbuf.dptr+len, dbuf.dsize-len );
4110 /* This will get the current RPC talloc context, but we should be
4111 passing this as a parameter... fixme... JRA ! */
4113 if (!nt_printing_getsec(info, sharename, &info->secdesc_buf)) {
4114 SAFE_FREE(dbuf.dptr);
4115 return WERR_NOMEM;
4118 /* Fix for OS/2 drivers. */
4120 if (get_remote_arch() == RA_OS2) {
4121 map_to_os2_driver(info->drivername);
4124 SAFE_FREE(dbuf.dptr);
4126 DEBUG(9,("Unpacked printer [%s] name [%s] running driver [%s]\n",
4127 sharename, info->printername, info->drivername));
4129 return WERR_OK;
4132 /****************************************************************************
4133 Debugging function, dump at level 6 the struct in the logs.
4134 ****************************************************************************/
4135 static uint32 dump_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4137 uint32 result;
4138 NT_PRINTER_INFO_LEVEL_2 *info2;
4140 DEBUG(106,("Dumping printer at level [%d]\n", level));
4142 switch (level) {
4143 case 2:
4145 if (printer->info_2 == NULL)
4146 result=5;
4147 else
4149 info2=printer->info_2;
4151 DEBUGADD(106,("attributes:[%d]\n", info2->attributes));
4152 DEBUGADD(106,("priority:[%d]\n", info2->priority));
4153 DEBUGADD(106,("default_priority:[%d]\n", info2->default_priority));
4154 DEBUGADD(106,("starttime:[%d]\n", info2->starttime));
4155 DEBUGADD(106,("untiltime:[%d]\n", info2->untiltime));
4156 DEBUGADD(106,("status:[%d]\n", info2->status));
4157 DEBUGADD(106,("cjobs:[%d]\n", info2->cjobs));
4158 DEBUGADD(106,("averageppm:[%d]\n", info2->averageppm));
4159 DEBUGADD(106,("changeid:[%d]\n", info2->changeid));
4160 DEBUGADD(106,("c_setprinter:[%d]\n", info2->c_setprinter));
4161 DEBUGADD(106,("setuptime:[%d]\n", info2->setuptime));
4163 DEBUGADD(106,("servername:[%s]\n", info2->servername));
4164 DEBUGADD(106,("printername:[%s]\n", info2->printername));
4165 DEBUGADD(106,("sharename:[%s]\n", info2->sharename));
4166 DEBUGADD(106,("portname:[%s]\n", info2->portname));
4167 DEBUGADD(106,("drivername:[%s]\n", info2->drivername));
4168 DEBUGADD(106,("comment:[%s]\n", info2->comment));
4169 DEBUGADD(106,("location:[%s]\n", info2->location));
4170 DEBUGADD(106,("sepfile:[%s]\n", info2->sepfile));
4171 DEBUGADD(106,("printprocessor:[%s]\n", info2->printprocessor));
4172 DEBUGADD(106,("datatype:[%s]\n", info2->datatype));
4173 DEBUGADD(106,("parameters:[%s]\n", info2->parameters));
4174 result=0;
4176 break;
4178 default:
4179 DEBUGADD(106,("dump_a_printer: Level %u not implemented\n", (unsigned int)level ));
4180 result=1;
4181 break;
4184 return result;
4187 /****************************************************************************
4188 Update the changeid time.
4189 This is SO NASTY as some drivers need this to change, others need it
4190 static. This value will change every second, and I must hope that this
4191 is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
4192 UTAH ! JRA.
4193 ****************************************************************************/
4195 static uint32 rev_changeid(void)
4197 struct timeval tv;
4199 get_process_uptime(&tv);
4201 #if 1 /* JERRY */
4202 /* Return changeid as msec since spooler restart */
4203 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
4204 #else
4206 * This setting seems to work well but is too untested
4207 * to replace the above calculation. Left in for experiementation
4208 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
4210 return tv.tv_sec * 10 + tv.tv_usec / 100000;
4211 #endif
4216 * The function below are the high level ones.
4217 * only those ones must be called from the spoolss code.
4218 * JFM.
4221 /****************************************************************************
4222 Modify a printer. This is called from SETPRINTERDATA/DELETEPRINTERDATA.
4223 ****************************************************************************/
4225 WERROR mod_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4227 WERROR result;
4229 dump_a_printer(printer, level);
4231 switch (level) {
4232 case 2:
4235 * Update the changestamp. Emperical tests show that the
4236 * ChangeID is always updated,but c_setprinter is
4237 * global spooler variable (not per printer).
4240 /* ChangeID **must** be increasing over the lifetime
4241 of client's spoolss service in order for the
4242 client's cache to show updates */
4244 printer->info_2->changeid = rev_changeid();
4247 * Because one day someone will ask:
4248 * NT->NT An admin connection to a remote
4249 * printer show changes imeediately in
4250 * the properities dialog
4252 * A non-admin connection will only show the
4253 * changes after viewing the properites page
4254 * 2 times. Seems to be related to a
4255 * race condition in the client between the spooler
4256 * updating the local cache and the Explorer.exe GUI
4257 * actually displaying the properties.
4259 * This is fixed in Win2k. admin/non-admin
4260 * connections both display changes immediately.
4262 * 14/12/01 --jerry
4265 result=update_a_printer_2(printer->info_2);
4266 break;
4268 default:
4269 result=WERR_UNKNOWN_LEVEL;
4270 break;
4273 return result;
4276 /****************************************************************************
4277 Initialize printer devmode & data with previously saved driver init values.
4278 ****************************************************************************/
4280 static bool set_driver_init_2( NT_PRINTER_INFO_LEVEL_2 *info_ptr )
4282 int len = 0;
4283 char *key = NULL;
4284 TDB_DATA dbuf;
4285 NT_PRINTER_INFO_LEVEL_2 info;
4288 ZERO_STRUCT(info);
4291 * Delete any printer data 'values' already set. When called for driver
4292 * replace, there will generally be some, but during an add printer, there
4293 * should not be any (if there are delete them).
4296 if ( info_ptr->data )
4297 delete_all_printer_data( info_ptr, "" );
4299 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX,
4300 info_ptr->drivername) < 0) {
4301 return false;
4304 dbuf = tdb_fetch_bystring(tdb_drivers, key);
4305 if (!dbuf.dptr) {
4307 * When changing to a driver that has no init info in the tdb, remove
4308 * the previous drivers init info and leave the new on blank.
4310 free_nt_devicemode(&info_ptr->devmode);
4311 SAFE_FREE(key);
4312 return false;
4315 SAFE_FREE(key);
4317 * Get the saved DEVMODE..
4320 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
4323 * The saved DEVMODE contains the devicename from the printer used during
4324 * the initialization save. Change it to reflect the new printer.
4327 if ( info.devmode ) {
4328 ZERO_STRUCT(info.devmode->devicename);
4329 fstrcpy(info.devmode->devicename, info_ptr->printername);
4333 * NT/2k does not change out the entire DeviceMode of a printer
4334 * when changing the driver. Only the driverextra, private, &
4335 * driverversion fields. --jerry (Thu Mar 14 08:58:43 CST 2002)
4337 * Later examination revealed that Windows NT/2k does reset the
4338 * the printer's device mode, bit **only** when you change a
4339 * property of the device mode such as the page orientation.
4340 * --jerry
4344 /* Bind the saved DEVMODE to the new the printer */
4346 free_nt_devicemode(&info_ptr->devmode);
4347 info_ptr->devmode = info.devmode;
4349 DEBUG(10,("set_driver_init_2: Set printer [%s] init %s DEVMODE for driver [%s]\n",
4350 info_ptr->printername, info_ptr->devmode?"VALID":"NULL", info_ptr->drivername));
4352 /* Add the printer data 'values' to the new printer */
4354 if ( !(info_ptr->data = TALLOC_ZERO_P( info_ptr, NT_PRINTER_DATA )) ) {
4355 DEBUG(0,("set_driver_init_2: talloc() failed!\n"));
4356 return False;
4359 len += unpack_values( info_ptr->data, dbuf.dptr+len, dbuf.dsize-len );
4361 SAFE_FREE(dbuf.dptr);
4363 return true;
4366 /****************************************************************************
4367 Initialize printer devmode & data with previously saved driver init values.
4368 When a printer is created using AddPrinter, the drivername bound to the
4369 printer is used to lookup previously saved driver initialization info, which
4370 is bound to the new printer.
4371 ****************************************************************************/
4373 bool set_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4375 bool result = False;
4377 switch (level) {
4378 case 2:
4379 result = set_driver_init_2(printer->info_2);
4380 break;
4382 default:
4383 DEBUG(0,("set_driver_init: Programmer's error! Unknown driver_init level [%d]\n",
4384 level));
4385 break;
4388 return result;
4391 /****************************************************************************
4392 Delete driver init data stored for a specified driver
4393 ****************************************************************************/
4395 bool del_driver_init(char *drivername)
4397 char *key;
4398 bool ret;
4400 if (!drivername || !*drivername) {
4401 DEBUG(3,("del_driver_init: No drivername specified!\n"));
4402 return false;
4405 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX, drivername) < 0) {
4406 return false;
4409 DEBUG(6,("del_driver_init: Removing driver init data for [%s]\n",
4410 drivername));
4412 ret = (tdb_delete_bystring(tdb_drivers, key) == 0);
4413 SAFE_FREE(key);
4414 return ret;
4417 /****************************************************************************
4418 Pack up the DEVMODE and values for a printer into a 'driver init' entry
4419 in the tdb. Note: this is different from the driver entry and the printer
4420 entry. There should be a single driver init entry for each driver regardless
4421 of whether it was installed from NT or 2K. Technically, they should be
4422 different, but they work out to the same struct.
4423 ****************************************************************************/
4425 static uint32 update_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info)
4427 char *key = NULL;
4428 uint8 *buf;
4429 int buflen, len, ret;
4430 int retlen;
4431 TDB_DATA dbuf;
4433 buf = NULL;
4434 buflen = 0;
4436 again:
4437 len = 0;
4438 len += pack_devicemode(info->devmode, buf+len, buflen-len);
4440 retlen = pack_values( info->data, buf+len, buflen-len );
4441 if (retlen == -1) {
4442 ret = -1;
4443 goto done;
4445 len += retlen;
4447 if (buflen < len) {
4448 buf = (uint8 *)SMB_REALLOC(buf, len);
4449 if (!buf) {
4450 DEBUG(0, ("update_driver_init_2: failed to enlarge buffer!\n"));
4451 ret = -1;
4452 goto done;
4454 buflen = len;
4455 goto again;
4458 SAFE_FREE(key);
4459 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX, info->drivername) < 0) {
4460 ret = (uint32)-1;
4461 goto done;
4464 dbuf.dptr = buf;
4465 dbuf.dsize = len;
4467 ret = tdb_store_bystring(tdb_drivers, key, dbuf, TDB_REPLACE);
4469 done:
4470 if (ret == -1)
4471 DEBUG(8, ("update_driver_init_2: error updating printer init to tdb on disk\n"));
4473 SAFE_FREE(buf);
4475 DEBUG(10,("update_driver_init_2: Saved printer [%s] init DEVMODE & values for driver [%s]\n",
4476 info->sharename, info->drivername));
4478 return ret;
4481 /****************************************************************************
4482 Update (i.e. save) the driver init info (DEVMODE and values) for a printer
4483 ****************************************************************************/
4485 static uint32 update_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4487 uint32 result;
4489 dump_a_printer(printer, level);
4491 switch (level) {
4492 case 2:
4493 result = update_driver_init_2(printer->info_2);
4494 break;
4495 default:
4496 result = 1;
4497 break;
4500 return result;
4503 /****************************************************************************
4504 Convert the printer data value, a REG_BINARY array, into an initialization
4505 DEVMODE. Note: the array must be parsed as if it was a DEVMODE in an rpc...
4506 got to keep the endians happy :).
4507 ****************************************************************************/
4509 static bool convert_driver_init( TALLOC_CTX *ctx, NT_DEVICEMODE *nt_devmode, uint8 *data, uint32 data_len )
4511 bool result = False;
4512 prs_struct ps;
4513 DEVICEMODE devmode;
4515 ZERO_STRUCT(devmode);
4517 prs_init(&ps, 0, ctx, UNMARSHALL);
4518 ps.data_p = (char *)data;
4519 ps.buffer_size = data_len;
4521 if (spoolss_io_devmode("phantom DEVMODE", &ps, 0, &devmode))
4522 result = convert_devicemode("", &devmode, &nt_devmode);
4523 else
4524 DEBUG(10,("convert_driver_init: error parsing DEVMODE\n"));
4526 return result;
4529 /****************************************************************************
4530 Set the DRIVER_INIT info in the tdb. Requires Win32 client code that:
4532 1. Use the driver's config DLL to this UNC printername and:
4533 a. Call DrvPrintEvent with PRINTER_EVENT_INITIALIZE
4534 b. Call DrvConvertDevMode with CDM_DRIVER_DEFAULT to get default DEVMODE
4535 2. Call SetPrinterData with the 'magic' key and the DEVMODE as data.
4537 The last step triggers saving the "driver initialization" information for
4538 this printer into the tdb. Later, new printers that use this driver will
4539 have this initialization information bound to them. This simulates the
4540 driver initialization, as if it had run on the Samba server (as it would
4541 have done on NT).
4543 The Win32 client side code requirement sucks! But until we can run arbitrary
4544 Win32 printer driver code on any Unix that Samba runs on, we are stuck with it.
4546 It would have been easier to use SetPrinter because all the UNMARSHALLING of
4547 the DEVMODE is done there, but 2K/XP clients do not set the DEVMODE... think
4548 about it and you will realize why. JRR 010720
4549 ****************************************************************************/
4551 static WERROR save_driver_init_2(NT_PRINTER_INFO_LEVEL *printer, uint8 *data, uint32 data_len )
4553 WERROR status = WERR_OK;
4554 TALLOC_CTX *ctx = NULL;
4555 NT_DEVICEMODE *nt_devmode = NULL;
4556 NT_DEVICEMODE *tmp_devmode = printer->info_2->devmode;
4559 * When the DEVMODE is already set on the printer, don't try to unpack it.
4561 DEBUG(8,("save_driver_init_2: Enter...\n"));
4563 if ( !printer->info_2->devmode && data_len ) {
4565 * Set devmode on printer info, so entire printer initialization can be
4566 * saved to tdb.
4569 if ((ctx = talloc_init("save_driver_init_2")) == NULL)
4570 return WERR_NOMEM;
4572 if ((nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE)) == NULL) {
4573 status = WERR_NOMEM;
4574 goto done;
4577 ZERO_STRUCTP(nt_devmode);
4580 * The DEVMODE is held in the 'data' component of the param in raw binary.
4581 * Convert it to to a devmode structure
4583 if ( !convert_driver_init( ctx, nt_devmode, data, data_len )) {
4584 DEBUG(10,("save_driver_init_2: error converting DEVMODE\n"));
4585 status = WERR_INVALID_PARAM;
4586 goto done;
4589 printer->info_2->devmode = nt_devmode;
4593 * Pack up and add (or update) the DEVMODE and any current printer data to
4594 * a 'driver init' element in the tdb
4598 if ( update_driver_init(printer, 2) != 0 ) {
4599 DEBUG(10,("save_driver_init_2: error updating DEVMODE\n"));
4600 status = WERR_NOMEM;
4601 goto done;
4605 * If driver initialization info was successfully saved, set the current
4606 * printer to match it. This allows initialization of the current printer
4607 * as well as the driver.
4609 status = mod_a_printer(printer, 2);
4610 if (!W_ERROR_IS_OK(status)) {
4611 DEBUG(10,("save_driver_init_2: error setting DEVMODE on printer [%s]\n",
4612 printer->info_2->printername));
4615 done:
4616 talloc_destroy(ctx);
4617 free_nt_devicemode( &nt_devmode );
4619 printer->info_2->devmode = tmp_devmode;
4621 return status;
4624 /****************************************************************************
4625 Update the driver init info (DEVMODE and specifics) for a printer
4626 ****************************************************************************/
4628 WERROR save_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level, uint8 *data, uint32 data_len)
4630 WERROR status = WERR_OK;
4632 switch (level) {
4633 case 2:
4634 status = save_driver_init_2( printer, data, data_len );
4635 break;
4636 default:
4637 status = WERR_UNKNOWN_LEVEL;
4638 break;
4641 return status;
4644 /****************************************************************************
4645 Get a NT_PRINTER_INFO_LEVEL struct. It returns malloced memory.
4647 Previously the code had a memory allocation problem because it always
4648 used the TALLOC_CTX from the Printer_entry*. This context lasts
4649 as a long as the original handle is open. So if the client made a lot
4650 of getprinter[data]() calls, the memory usage would climb. Now we use
4651 a short lived TALLOC_CTX for printer_info_2 objects returned. We
4652 still use the Printer_entry->ctx for maintaining the cache copy though
4653 since that object must live as long as the handle by definition.
4654 --jerry
4656 ****************************************************************************/
4658 WERROR get_a_printer( Printer_entry *print_hnd, NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level,
4659 const char *sharename)
4661 WERROR result;
4662 fstring servername;
4664 DEBUG(10,("get_a_printer: [%s] level %u\n", sharename, (unsigned int)level));
4666 if ( !(*pp_printer = TALLOC_ZERO_P(NULL, NT_PRINTER_INFO_LEVEL)) ) {
4667 DEBUG(0,("get_a_printer: talloc() fail.\n"));
4668 return WERR_NOMEM;
4671 switch (level) {
4672 case 2:
4673 if ( !((*pp_printer)->info_2 = TALLOC_ZERO_P(*pp_printer, NT_PRINTER_INFO_LEVEL_2)) ) {
4674 DEBUG(0,("get_a_printer: talloc() fail.\n"));
4675 TALLOC_FREE( *pp_printer );
4676 return WERR_NOMEM;
4679 if ( print_hnd )
4680 fstrcpy( servername, print_hnd->servername );
4681 else {
4682 fstrcpy( servername, "%L" );
4683 standard_sub_basic( "", "", servername,
4684 sizeof(servername)-1 );
4687 result = get_a_printer_2( (*pp_printer)->info_2, servername, sharename );
4690 /* we have a new printer now. Save it with this handle */
4692 if ( !W_ERROR_IS_OK(result) ) {
4693 TALLOC_FREE( *pp_printer );
4694 DEBUG(10,("get_a_printer: [%s] level %u returning %s\n",
4695 sharename, (unsigned int)level, dos_errstr(result)));
4696 return result;
4699 dump_a_printer( *pp_printer, level);
4701 break;
4703 default:
4704 TALLOC_FREE( *pp_printer );
4705 return WERR_UNKNOWN_LEVEL;
4708 return WERR_OK;
4711 /****************************************************************************
4712 Deletes a NT_PRINTER_INFO_LEVEL struct.
4713 ****************************************************************************/
4715 uint32 free_a_printer(NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level)
4717 NT_PRINTER_INFO_LEVEL *printer = *pp_printer;
4719 if ( !printer )
4720 return 0;
4722 switch (level) {
4723 case 2:
4724 if ( printer->info_2 )
4725 free_nt_printer_info_level_2(&printer->info_2);
4726 break;
4728 default:
4729 DEBUG(0,("free_a_printer: unknown level! [%d]\n", level ));
4730 return 1;
4733 TALLOC_FREE(*pp_printer);
4735 return 0;
4738 /****************************************************************************
4739 ****************************************************************************/
4740 uint32 add_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4742 uint32 result;
4743 DEBUG(104,("adding a printer at level [%d]\n", level));
4744 dump_a_printer_driver(driver, level);
4746 switch (level) {
4747 case 3:
4748 result=add_a_printer_driver_3(driver.info_3);
4749 break;
4751 case 6:
4752 result=add_a_printer_driver_6(driver.info_6);
4753 break;
4755 default:
4756 result=1;
4757 break;
4760 return result;
4762 /****************************************************************************
4763 ****************************************************************************/
4765 WERROR get_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL *driver, uint32 level,
4766 fstring drivername, const char *architecture, uint32 version)
4768 WERROR result;
4770 switch (level) {
4771 case 3:
4772 /* Sometime we just want any version of the driver */
4774 if ( version == DRIVER_ANY_VERSION ) {
4775 /* look for Win2k first and then for NT4 */
4776 result = get_a_printer_driver_3(&driver->info_3, drivername,
4777 architecture, 3);
4779 if ( !W_ERROR_IS_OK(result) ) {
4780 result = get_a_printer_driver_3( &driver->info_3,
4781 drivername, architecture, 2 );
4783 } else {
4784 result = get_a_printer_driver_3(&driver->info_3, drivername,
4785 architecture, version);
4787 break;
4789 default:
4790 result=W_ERROR(1);
4791 break;
4794 if (W_ERROR_IS_OK(result))
4795 dump_a_printer_driver(*driver, level);
4797 return result;
4800 /****************************************************************************
4801 ****************************************************************************/
4802 uint32 free_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4804 uint32 result;
4806 switch (level) {
4807 case 3:
4809 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
4810 if (driver.info_3 != NULL)
4812 info3=driver.info_3;
4813 SAFE_FREE(info3->dependentfiles);
4814 ZERO_STRUCTP(info3);
4815 SAFE_FREE(info3);
4816 result=0;
4817 } else {
4818 result=4;
4820 break;
4822 case 6:
4824 NT_PRINTER_DRIVER_INFO_LEVEL_6 *info6;
4825 if (driver.info_6 != NULL) {
4826 info6=driver.info_6;
4827 SAFE_FREE(info6->dependentfiles);
4828 SAFE_FREE(info6->previousnames);
4829 ZERO_STRUCTP(info6);
4830 SAFE_FREE(info6);
4831 result=0;
4832 } else {
4833 result=4;
4835 break;
4837 default:
4838 result=1;
4839 break;
4841 return result;
4845 /****************************************************************************
4846 Determine whether or not a particular driver is currently assigned
4847 to a printer
4848 ****************************************************************************/
4850 bool printer_driver_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3 )
4852 int snum;
4853 int n_services = lp_numservices();
4854 NT_PRINTER_INFO_LEVEL *printer = NULL;
4855 bool in_use = False;
4857 if ( !info_3 )
4858 return False;
4860 DEBUG(10,("printer_driver_in_use: Beginning search through ntprinters.tdb...\n"));
4862 /* loop through the printers.tdb and check for the drivername */
4864 for (snum=0; snum<n_services && !in_use; snum++) {
4865 if ( !(lp_snum_ok(snum) && lp_print_ok(snum) ) )
4866 continue;
4868 if ( !W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))) )
4869 continue;
4871 if ( strequal(info_3->name, printer->info_2->drivername) )
4872 in_use = True;
4874 free_a_printer( &printer, 2 );
4877 DEBUG(10,("printer_driver_in_use: Completed search through ntprinters.tdb...\n"));
4879 if ( in_use ) {
4880 NT_PRINTER_DRIVER_INFO_LEVEL d;
4881 WERROR werr;
4883 DEBUG(5,("printer_driver_in_use: driver \"%s\" is currently in use\n", info_3->name));
4885 /* we can still remove the driver if there is one of
4886 "Windows NT x86" version 2 or 3 left */
4888 if ( !strequal( "Windows NT x86", info_3->environment ) ) {
4889 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", DRIVER_ANY_VERSION );
4891 else {
4892 switch ( info_3->cversion ) {
4893 case 2:
4894 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", 3 );
4895 break;
4896 case 3:
4897 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", 2 );
4898 break;
4899 default:
4900 DEBUG(0,("printer_driver_in_use: ERROR! unknown driver version (%d)\n",
4901 info_3->cversion));
4902 werr = WERR_UNKNOWN_PRINTER_DRIVER;
4903 break;
4907 /* now check the error code */
4909 if ( W_ERROR_IS_OK(werr) ) {
4910 /* it's ok to remove the driver, we have other architctures left */
4911 in_use = False;
4912 free_a_printer_driver( d, 3 );
4916 /* report that the driver is not in use by default */
4918 return in_use;
4922 /**********************************************************************
4923 Check to see if a ogiven file is in use by *info
4924 *********************************************************************/
4926 static bool drv_file_in_use( char* file, NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
4928 int i = 0;
4930 if ( !info )
4931 return False;
4933 if ( strequal(file, info->driverpath) )
4934 return True;
4936 if ( strequal(file, info->datafile) )
4937 return True;
4939 if ( strequal(file, info->configfile) )
4940 return True;
4942 if ( strequal(file, info->helpfile) )
4943 return True;
4945 /* see of there are any dependent files to examine */
4947 if ( !info->dependentfiles )
4948 return False;
4950 while ( *info->dependentfiles[i] ) {
4951 if ( strequal(file, info->dependentfiles[i]) )
4952 return True;
4953 i++;
4956 return False;
4960 /**********************************************************************
4961 Utility function to remove the dependent file pointed to by the
4962 input parameter from the list
4963 *********************************************************************/
4965 static void trim_dependent_file( fstring files[], int idx )
4968 /* bump everything down a slot */
4970 while( *files[idx+1] ) {
4971 fstrcpy( files[idx], files[idx+1] );
4972 idx++;
4975 *files[idx] = '\0';
4977 return;
4980 /**********************************************************************
4981 Check if any of the files used by src are also used by drv
4982 *********************************************************************/
4984 static bool trim_overlap_drv_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *src,
4985 NT_PRINTER_DRIVER_INFO_LEVEL_3 *drv )
4987 bool in_use = False;
4988 int i = 0;
4990 if ( !src || !drv )
4991 return False;
4993 /* check each file. Remove it from the src structure if it overlaps */
4995 if ( drv_file_in_use(src->driverpath, drv) ) {
4996 in_use = True;
4997 DEBUG(10,("Removing driverfile [%s] from list\n", src->driverpath));
4998 fstrcpy( src->driverpath, "" );
5001 if ( drv_file_in_use(src->datafile, drv) ) {
5002 in_use = True;
5003 DEBUG(10,("Removing datafile [%s] from list\n", src->datafile));
5004 fstrcpy( src->datafile, "" );
5007 if ( drv_file_in_use(src->configfile, drv) ) {
5008 in_use = True;
5009 DEBUG(10,("Removing configfile [%s] from list\n", src->configfile));
5010 fstrcpy( src->configfile, "" );
5013 if ( drv_file_in_use(src->helpfile, drv) ) {
5014 in_use = True;
5015 DEBUG(10,("Removing helpfile [%s] from list\n", src->helpfile));
5016 fstrcpy( src->helpfile, "" );
5019 /* are there any dependentfiles to examine? */
5021 if ( !src->dependentfiles )
5022 return in_use;
5024 while ( *src->dependentfiles[i] ) {
5025 if ( drv_file_in_use(src->dependentfiles[i], drv) ) {
5026 in_use = True;
5027 DEBUG(10,("Removing [%s] from dependent file list\n", src->dependentfiles[i]));
5028 trim_dependent_file( src->dependentfiles, i );
5029 } else
5030 i++;
5033 return in_use;
5036 /****************************************************************************
5037 Determine whether or not a particular driver files are currently being
5038 used by any other driver.
5040 Return value is True if any files were in use by other drivers
5041 and False otherwise.
5043 Upon return, *info has been modified to only contain the driver files
5044 which are not in use
5045 ****************************************************************************/
5047 bool printer_driver_files_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
5049 int i;
5050 int ndrivers;
5051 uint32 version;
5052 fstring *list = NULL;
5053 NT_PRINTER_DRIVER_INFO_LEVEL driver;
5055 if ( !info )
5056 return False;
5058 version = info->cversion;
5060 /* loop over all driver versions */
5062 DEBUG(5,("printer_driver_files_in_use: Beginning search through ntdrivers.tdb...\n"));
5064 /* get the list of drivers */
5066 list = NULL;
5067 ndrivers = get_ntdrivers(&list, info->environment, version);
5069 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n",
5070 ndrivers, info->environment, version));
5072 /* check each driver for overlap in files */
5074 for (i=0; i<ndrivers; i++) {
5075 DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
5077 ZERO_STRUCT(driver);
5079 if ( !W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, list[i], info->environment, version)) ) {
5080 SAFE_FREE(list);
5081 return True;
5084 /* check if d2 uses any files from d1 */
5085 /* only if this is a different driver than the one being deleted */
5087 if ( !strequal(info->name, driver.info_3->name) ) {
5088 if ( trim_overlap_drv_files(info, driver.info_3) ) {
5089 free_a_printer_driver(driver, 3);
5090 SAFE_FREE( list );
5091 return True;
5095 free_a_printer_driver(driver, 3);
5098 SAFE_FREE(list);
5100 DEBUG(5,("printer_driver_files_in_use: Completed search through ntdrivers.tdb...\n"));
5102 driver.info_3 = info;
5104 if ( DEBUGLEVEL >= 20 )
5105 dump_a_printer_driver( driver, 3 );
5107 return False;
5110 /****************************************************************************
5111 Actually delete the driver files. Make sure that
5112 printer_driver_files_in_use() return False before calling
5113 this.
5114 ****************************************************************************/
5116 static bool delete_driver_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user )
5118 int i = 0;
5119 char *s;
5120 const char *file;
5121 connection_struct *conn;
5122 DATA_BLOB null_pw;
5123 NTSTATUS nt_status;
5124 fstring res_type;
5125 SMB_STRUCT_STAT st;
5127 if ( !info_3 )
5128 return False;
5130 DEBUG(6,("delete_driver_files: deleting driver [%s] - version [%d]\n", info_3->name, info_3->cversion));
5133 * Connect to the print$ share under the same account as the
5134 * user connected to the rpc pipe. Note we must be root to
5135 * do this.
5138 null_pw = data_blob_null;
5139 fstrcpy(res_type, "A:");
5140 become_root();
5141 conn = make_connection_with_chdir( "print$", null_pw, res_type, user->vuid, &nt_status );
5142 unbecome_root();
5144 if ( !conn ) {
5145 DEBUG(0,("delete_driver_files: Unable to connect\n"));
5146 return False;
5149 if ( !CAN_WRITE(conn) ) {
5150 DEBUG(3,("delete_driver_files: Cannot delete print driver when [print$] is read-only\n"));
5151 return False;
5154 /* Save who we are - we are temporarily becoming the connection user. */
5156 if ( !become_user(conn, conn->vuid) ) {
5157 DEBUG(0,("delete_driver_files: Can't become user!\n"));
5158 return False;
5161 /* now delete the files; must strip the '\print$' string from
5162 fron of path */
5164 if ( *info_3->driverpath ) {
5165 if ( (s = strchr( &info_3->driverpath[1], '\\' )) != NULL ) {
5166 file = s;
5167 driver_unix_convert(conn,file,&st);
5168 DEBUG(10,("deleting driverfile [%s]\n", s));
5169 unlink_internals(conn, NULL, 0, file, False);
5173 if ( *info_3->configfile ) {
5174 if ( (s = strchr( &info_3->configfile[1], '\\' )) != NULL ) {
5175 file = s;
5176 driver_unix_convert(conn,file,&st);
5177 DEBUG(10,("deleting configfile [%s]\n", s));
5178 unlink_internals(conn, NULL, 0, file, False);
5182 if ( *info_3->datafile ) {
5183 if ( (s = strchr( &info_3->datafile[1], '\\' )) != NULL ) {
5184 file = s;
5185 driver_unix_convert(conn,file,&st);
5186 DEBUG(10,("deleting datafile [%s]\n", s));
5187 unlink_internals(conn, NULL, 0, file, False);
5191 if ( *info_3->helpfile ) {
5192 if ( (s = strchr( &info_3->helpfile[1], '\\' )) != NULL ) {
5193 file = s;
5194 driver_unix_convert(conn,file,&st);
5195 DEBUG(10,("deleting helpfile [%s]\n", s));
5196 unlink_internals(conn, NULL, 0, file, False);
5200 /* check if we are done removing files */
5202 if ( info_3->dependentfiles ) {
5203 while ( info_3->dependentfiles[i][0] ) {
5204 char *p;
5206 /* bypass the "\print$" portion of the path */
5208 if ( (p = strchr( info_3->dependentfiles[i]+1, '\\' )) != NULL ) {
5209 file = p;
5210 driver_unix_convert(conn,file,&st);
5211 DEBUG(10,("deleting dependent file [%s]\n", file));
5212 unlink_internals(conn, NULL, 0, file, False);
5215 i++;
5219 unbecome_user();
5221 return true;
5224 /****************************************************************************
5225 Remove a printer driver from the TDB. This assumes that the the driver was
5226 previously looked up.
5227 ***************************************************************************/
5229 WERROR delete_printer_driver( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user,
5230 uint32 version, bool delete_files )
5232 char *key = NULL;
5233 const char *arch;
5234 TDB_DATA dbuf;
5235 NT_PRINTER_DRIVER_INFO_LEVEL ctr;
5237 /* delete the tdb data first */
5239 arch = get_short_archi(info_3->environment);
5240 if (!arch) {
5241 return WERR_UNKNOWN_PRINTER_DRIVER;
5243 if (asprintf(&key, "%s%s/%d/%s", DRIVERS_PREFIX,
5244 arch, version, info_3->name) < 0) {
5245 return WERR_NOMEM;
5248 DEBUG(5,("delete_printer_driver: key = [%s] delete_files = %s\n",
5249 key, delete_files ? "TRUE" : "FALSE" ));
5251 ctr.info_3 = info_3;
5252 dump_a_printer_driver( ctr, 3 );
5254 /* check if the driver actually exists for this environment */
5256 dbuf = tdb_fetch_bystring( tdb_drivers, key );
5257 if ( !dbuf.dptr ) {
5258 DEBUG(8,("delete_printer_driver: Driver unknown [%s]\n", key));
5259 SAFE_FREE(key);
5260 return WERR_UNKNOWN_PRINTER_DRIVER;
5263 SAFE_FREE( dbuf.dptr );
5265 /* ok... the driver exists so the delete should return success */
5267 if (tdb_delete_bystring(tdb_drivers, key) == -1) {
5268 DEBUG (0,("delete_printer_driver: fail to delete %s!\n", key));
5269 SAFE_FREE(key);
5270 return WERR_ACCESS_DENIED;
5274 * now delete any associated files if delete_files == True
5275 * even if this part failes, we return succes because the
5276 * driver doesn not exist any more
5279 if ( delete_files )
5280 delete_driver_files( info_3, user );
5282 DEBUG(5,("delete_printer_driver: driver delete successful [%s]\n", key));
5283 SAFE_FREE(key);
5285 return WERR_OK;
5288 /****************************************************************************
5289 Store a security desc for a printer.
5290 ****************************************************************************/
5292 WERROR nt_printing_setsec(const char *sharename, SEC_DESC_BUF *secdesc_ctr)
5294 SEC_DESC_BUF *new_secdesc_ctr = NULL;
5295 SEC_DESC_BUF *old_secdesc_ctr = NULL;
5296 prs_struct ps;
5297 TALLOC_CTX *mem_ctx = NULL;
5298 TDB_DATA kbuf;
5299 WERROR status;
5301 mem_ctx = talloc_init("nt_printing_setsec");
5302 if (mem_ctx == NULL)
5303 return WERR_NOMEM;
5305 /* The old owner and group sids of the security descriptor are not
5306 present when new ACEs are added or removed by changing printer
5307 permissions through NT. If they are NULL in the new security
5308 descriptor then copy them over from the old one. */
5310 if (!secdesc_ctr->sd->owner_sid || !secdesc_ctr->sd->group_sid) {
5311 DOM_SID *owner_sid, *group_sid;
5312 SEC_ACL *dacl, *sacl;
5313 SEC_DESC *psd = NULL;
5314 size_t size;
5316 if (!nt_printing_getsec(mem_ctx, sharename, &old_secdesc_ctr)) {
5317 status = WERR_NOMEM;
5318 goto out;
5321 /* Pick out correct owner and group sids */
5323 owner_sid = secdesc_ctr->sd->owner_sid ?
5324 secdesc_ctr->sd->owner_sid :
5325 old_secdesc_ctr->sd->owner_sid;
5327 group_sid = secdesc_ctr->sd->group_sid ?
5328 secdesc_ctr->sd->group_sid :
5329 old_secdesc_ctr->sd->group_sid;
5331 dacl = secdesc_ctr->sd->dacl ?
5332 secdesc_ctr->sd->dacl :
5333 old_secdesc_ctr->sd->dacl;
5335 sacl = secdesc_ctr->sd->sacl ?
5336 secdesc_ctr->sd->sacl :
5337 old_secdesc_ctr->sd->sacl;
5339 /* Make a deep copy of the security descriptor */
5341 psd = make_sec_desc(mem_ctx, secdesc_ctr->sd->revision, secdesc_ctr->sd->type,
5342 owner_sid, group_sid,
5343 sacl,
5344 dacl,
5345 &size);
5347 if (!psd) {
5348 status = WERR_NOMEM;
5349 goto out;
5352 new_secdesc_ctr = make_sec_desc_buf(mem_ctx, size, psd);
5355 if (!new_secdesc_ctr) {
5356 new_secdesc_ctr = secdesc_ctr;
5359 /* Store the security descriptor in a tdb */
5361 prs_init(&ps, (uint32)sec_desc_size(new_secdesc_ctr->sd) +
5362 sizeof(SEC_DESC_BUF), mem_ctx, MARSHALL);
5364 if (!sec_io_desc_buf("nt_printing_setsec", &new_secdesc_ctr,
5365 &ps, 1)) {
5366 status = WERR_BADFUNC;
5367 goto out;
5370 kbuf = make_printers_secdesc_tdbkey(mem_ctx, sharename );
5372 if (tdb_prs_store(tdb_printers, kbuf, &ps)==0) {
5373 status = WERR_OK;
5374 } else {
5375 DEBUG(1,("Failed to store secdesc for %s\n", sharename));
5376 status = WERR_BADFUNC;
5379 /* Free malloc'ed memory */
5381 out:
5383 prs_mem_free(&ps);
5384 if (mem_ctx)
5385 talloc_destroy(mem_ctx);
5386 return status;
5389 /****************************************************************************
5390 Construct a default security descriptor buffer for a printer.
5391 ****************************************************************************/
5393 static SEC_DESC_BUF *construct_default_printer_sdb(TALLOC_CTX *ctx)
5395 SEC_ACE ace[5]; /* max number of ace entries */
5396 int i = 0;
5397 SEC_ACCESS sa;
5398 SEC_ACL *psa = NULL;
5399 SEC_DESC_BUF *sdb = NULL;
5400 SEC_DESC *psd = NULL;
5401 DOM_SID adm_sid;
5402 size_t sd_size;
5404 /* Create an ACE where Everyone is allowed to print */
5406 init_sec_access(&sa, PRINTER_ACE_PRINT);
5407 init_sec_ace(&ace[i++], &global_sid_World, SEC_ACE_TYPE_ACCESS_ALLOWED,
5408 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5410 /* Add the domain admins group if we are a DC */
5412 if ( IS_DC ) {
5413 DOM_SID domadmins_sid;
5415 sid_copy(&domadmins_sid, get_global_sam_sid());
5416 sid_append_rid(&domadmins_sid, DOMAIN_GROUP_RID_ADMINS);
5418 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5419 init_sec_ace(&ace[i++], &domadmins_sid,
5420 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5421 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5422 init_sec_ace(&ace[i++], &domadmins_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
5423 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5425 else if (secrets_fetch_domain_sid(lp_workgroup(), &adm_sid)) {
5426 sid_append_rid(&adm_sid, DOMAIN_USER_RID_ADMIN);
5428 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5429 init_sec_ace(&ace[i++], &adm_sid,
5430 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5431 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5432 init_sec_ace(&ace[i++], &adm_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
5433 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5436 /* add BUILTIN\Administrators as FULL CONTROL */
5438 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5439 init_sec_ace(&ace[i++], &global_sid_Builtin_Administrators,
5440 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5441 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5442 init_sec_ace(&ace[i++], &global_sid_Builtin_Administrators,
5443 SEC_ACE_TYPE_ACCESS_ALLOWED,
5444 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5446 /* Make the security descriptor owned by the BUILTIN\Administrators */
5448 /* The ACL revision number in rpc_secdesc.h differs from the one
5449 created by NT when setting ACE entries in printer
5450 descriptors. NT4 complains about the property being edited by a
5451 NT5 machine. */
5453 if ((psa = make_sec_acl(ctx, NT4_ACL_REVISION, i, ace)) != NULL) {
5454 psd = make_sec_desc(ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
5455 &global_sid_Builtin_Administrators,
5456 &global_sid_Builtin_Administrators,
5457 NULL, psa, &sd_size);
5460 if (!psd) {
5461 DEBUG(0,("construct_default_printer_sd: Failed to make SEC_DESC.\n"));
5462 return NULL;
5465 sdb = make_sec_desc_buf(ctx, sd_size, psd);
5467 DEBUG(4,("construct_default_printer_sdb: size = %u.\n",
5468 (unsigned int)sd_size));
5470 return sdb;
5473 /****************************************************************************
5474 Get a security desc for a printer.
5475 ****************************************************************************/
5477 bool nt_printing_getsec(TALLOC_CTX *ctx, const char *sharename, SEC_DESC_BUF **secdesc_ctr)
5479 prs_struct ps;
5480 TDB_DATA kbuf;
5481 char *temp;
5483 if (strlen(sharename) > 2 && (temp = strchr(sharename + 2, '\\'))) {
5484 sharename = temp + 1;
5487 ZERO_STRUCT(ps);
5489 /* Fetch security descriptor from tdb */
5491 kbuf = make_printers_secdesc_tdbkey(ctx, sharename );
5493 if (tdb_prs_fetch(tdb_printers, kbuf, &ps, ctx)!=0 ||
5494 !sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
5496 prs_mem_free(&ps);
5498 DEBUG(4,("using default secdesc for %s\n", sharename));
5500 if (!(*secdesc_ctr = construct_default_printer_sdb(ctx))) {
5501 return False;
5504 /* Save default security descriptor for later */
5506 prs_init(&ps, (uint32)sec_desc_size((*secdesc_ctr)->sd) +
5507 sizeof(SEC_DESC_BUF), ctx, MARSHALL);
5509 if (sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
5510 tdb_prs_store(tdb_printers, kbuf, &ps);
5513 prs_mem_free(&ps);
5515 return True;
5518 prs_mem_free(&ps);
5520 /* If security descriptor is owned by S-1-1-0 and winbindd is up,
5521 this security descriptor has been created when winbindd was
5522 down. Take ownership of security descriptor. */
5524 if (sid_equal((*secdesc_ctr)->sd->owner_sid, &global_sid_World)) {
5525 DOM_SID owner_sid;
5527 /* Change sd owner to workgroup administrator */
5529 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
5530 SEC_DESC_BUF *new_secdesc_ctr = NULL;
5531 SEC_DESC *psd = NULL;
5532 size_t size;
5534 /* Create new sd */
5536 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
5538 psd = make_sec_desc(ctx, (*secdesc_ctr)->sd->revision, (*secdesc_ctr)->sd->type,
5539 &owner_sid,
5540 (*secdesc_ctr)->sd->group_sid,
5541 (*secdesc_ctr)->sd->sacl,
5542 (*secdesc_ctr)->sd->dacl,
5543 &size);
5545 if (!psd) {
5546 return False;
5549 new_secdesc_ctr = make_sec_desc_buf(ctx, size, psd);
5550 if (!new_secdesc_ctr) {
5551 return False;
5554 /* Swap with other one */
5556 *secdesc_ctr = new_secdesc_ctr;
5558 /* Set it */
5560 nt_printing_setsec(sharename, *secdesc_ctr);
5564 if (DEBUGLEVEL >= 10) {
5565 SEC_ACL *the_acl = (*secdesc_ctr)->sd->dacl;
5566 int i;
5568 DEBUG(10, ("secdesc_ctr for %s has %d aces:\n",
5569 sharename, the_acl->num_aces));
5571 for (i = 0; i < the_acl->num_aces; i++) {
5572 fstring sid_str;
5574 sid_to_string(sid_str, &the_acl->aces[i].trustee);
5576 DEBUG(10, ("%s %d %d 0x%08x\n", sid_str,
5577 the_acl->aces[i].type, the_acl->aces[i].flags,
5578 the_acl->aces[i].access_mask));
5582 return True;
5585 /* error code:
5586 0: everything OK
5587 1: level not implemented
5588 2: file doesn't exist
5589 3: can't allocate memory
5590 4: can't free memory
5591 5: non existant struct
5595 A printer and a printer driver are 2 different things.
5596 NT manages them separatelly, Samba does the same.
5597 Why ? Simply because it's easier and it makes sense !
5599 Now explanation: You have 3 printers behind your samba server,
5600 2 of them are the same make and model (laser A and B). But laser B
5601 has an 3000 sheet feeder and laser A doesn't such an option.
5602 Your third printer is an old dot-matrix model for the accounting :-).
5604 If the /usr/local/samba/lib directory (default dir), you will have
5605 5 files to describe all of this.
5607 3 files for the printers (1 by printer):
5608 NTprinter_laser A
5609 NTprinter_laser B
5610 NTprinter_accounting
5611 2 files for the drivers (1 for the laser and 1 for the dot matrix)
5612 NTdriver_printer model X
5613 NTdriver_printer model Y
5615 jfm: I should use this comment for the text file to explain
5616 same thing for the forms BTW.
5617 Je devrais mettre mes commentaires en francais, ca serait mieux :-)
5621 /* Convert generic access rights to printer object specific access rights.
5622 It turns out that NT4 security descriptors use generic access rights and
5623 NT5 the object specific ones. */
5625 void map_printer_permissions(SEC_DESC *sd)
5627 int i;
5629 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
5630 se_map_generic(&sd->dacl->aces[i].access_mask,
5631 &printer_generic_mapping);
5635 /****************************************************************************
5636 Check a user has permissions to perform the given operation. We use the
5637 permission constants defined in include/rpc_spoolss.h to check the various
5638 actions we perform when checking printer access.
5640 PRINTER_ACCESS_ADMINISTER:
5641 print_queue_pause, print_queue_resume, update_printer_sec,
5642 update_printer, spoolss_addprinterex_level_2,
5643 _spoolss_setprinterdata
5645 PRINTER_ACCESS_USE:
5646 print_job_start
5648 JOB_ACCESS_ADMINISTER:
5649 print_job_delete, print_job_pause, print_job_resume,
5650 print_queue_purge
5652 Try access control in the following order (for performance reasons):
5653 1) root ans SE_PRINT_OPERATOR can do anything (easy check)
5654 2) check security descriptor (bit comparisons in memory)
5655 3) "printer admins" (may result in numerous calls to winbind)
5657 ****************************************************************************/
5658 bool print_access_check(struct current_user *user, int snum, int access_type)
5660 SEC_DESC_BUF *secdesc = NULL;
5661 uint32 access_granted;
5662 NTSTATUS status;
5663 bool result;
5664 const char *pname;
5665 TALLOC_CTX *mem_ctx = NULL;
5666 SE_PRIV se_printop = SE_PRINT_OPERATOR;
5668 /* If user is NULL then use the current_user structure */
5670 if (!user)
5671 user = &current_user;
5673 /* Always allow root or SE_PRINT_OPERATROR to do anything */
5675 if ( user->ut.uid == 0 || user_has_privileges(user->nt_user_token, &se_printop ) ) {
5676 return True;
5679 /* Get printer name */
5681 pname = PRINTERNAME(snum);
5683 if (!pname || !*pname) {
5684 errno = EACCES;
5685 return False;
5688 /* Get printer security descriptor */
5690 if(!(mem_ctx = talloc_init("print_access_check"))) {
5691 errno = ENOMEM;
5692 return False;
5695 if (!nt_printing_getsec(mem_ctx, pname, &secdesc)) {
5696 talloc_destroy(mem_ctx);
5697 errno = ENOMEM;
5698 return False;
5701 if (access_type == JOB_ACCESS_ADMINISTER) {
5702 SEC_DESC_BUF *parent_secdesc = secdesc;
5704 /* Create a child security descriptor to check permissions
5705 against. This is because print jobs are child objects
5706 objects of a printer. */
5708 secdesc = se_create_child_secdesc(mem_ctx, parent_secdesc->sd, False);
5710 if (!secdesc) {
5711 talloc_destroy(mem_ctx);
5712 errno = ENOMEM;
5713 return False;
5716 /* Now this is the bit that really confuses me. The access
5717 type needs to be changed from JOB_ACCESS_ADMINISTER to
5718 PRINTER_ACCESS_ADMINISTER for this to work. Something
5719 to do with the child (job) object becoming like a
5720 printer?? -tpot */
5722 access_type = PRINTER_ACCESS_ADMINISTER;
5725 /* Check access */
5727 map_printer_permissions(secdesc->sd);
5729 result = se_access_check(secdesc->sd, user->nt_user_token, access_type,
5730 &access_granted, &status);
5732 DEBUG(4, ("access check was %s\n", result ? "SUCCESS" : "FAILURE"));
5734 /* see if we need to try the printer admin list */
5736 if ((access_granted == 0) &&
5737 (token_contains_name_in_list(uidtoname(user->ut.uid), NULL,
5738 user->nt_user_token,
5739 lp_printer_admin(snum)))) {
5740 talloc_destroy(mem_ctx);
5741 return True;
5744 talloc_destroy(mem_ctx);
5746 if (!result) {
5747 errno = EACCES;
5750 return result;
5753 /****************************************************************************
5754 Check the time parameters allow a print operation.
5755 *****************************************************************************/
5757 bool print_time_access_check(const char *servicename)
5759 NT_PRINTER_INFO_LEVEL *printer = NULL;
5760 bool ok = False;
5761 time_t now = time(NULL);
5762 struct tm *t;
5763 uint32 mins;
5765 if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, servicename)))
5766 return False;
5768 if (printer->info_2->starttime == 0 && printer->info_2->untiltime == 0)
5769 ok = True;
5771 t = gmtime(&now);
5772 mins = (uint32)t->tm_hour*60 + (uint32)t->tm_min;
5774 if (mins >= printer->info_2->starttime && mins <= printer->info_2->untiltime)
5775 ok = True;
5777 free_a_printer(&printer, 2);
5779 if (!ok)
5780 errno = EACCES;
5782 return ok;
5785 /****************************************************************************
5786 Fill in the servername sent in the _spoolss_open_printer_ex() call
5787 ****************************************************************************/
5789 char* get_server_name( Printer_entry *printer )
5791 return printer->servername;