Revert "Split lookup_name() and create a new functiong called"
[Samba.git] / source / printing / nt_printing.c
blobded985c260d5af5bc28a044d0d4c600e740bfbf4
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 static TDB_CONTEXT *tdb_forms; /* used for forms files */
25 static TDB_CONTEXT *tdb_drivers; /* used for driver files */
26 static TDB_CONTEXT *tdb_printers; /* used for printers files */
28 #define FORMS_PREFIX "FORMS/"
29 #define DRIVERS_PREFIX "DRIVERS/"
30 #define DRIVER_INIT_PREFIX "DRIVER_INIT/"
31 #define PRINTERS_PREFIX "PRINTERS/"
32 #define SECDESC_PREFIX "SECDESC/"
33 #define GLOBAL_C_SETPRINTER "GLOBALS/c_setprinter"
35 #define NTDRIVERS_DATABASE_VERSION_1 1
36 #define NTDRIVERS_DATABASE_VERSION_2 2
37 #define NTDRIVERS_DATABASE_VERSION_3 3 /* little endian version of v2 */
38 #define NTDRIVERS_DATABASE_VERSION_4 4 /* fix generic bits in security descriptors */
39 #define NTDRIVERS_DATABASE_VERSION_5 5 /* normalize keys in ntprinters.tdb */
41 /* Map generic permissions to printer object specific permissions */
43 const struct generic_mapping printer_generic_mapping = {
44 PRINTER_READ,
45 PRINTER_WRITE,
46 PRINTER_EXECUTE,
47 PRINTER_ALL_ACCESS
50 const struct standard_mapping printer_std_mapping = {
51 PRINTER_READ,
52 PRINTER_WRITE,
53 PRINTER_EXECUTE,
54 PRINTER_ALL_ACCESS
57 /* Map generic permissions to print server object specific permissions */
59 const struct generic_mapping printserver_generic_mapping = {
60 SERVER_READ,
61 SERVER_WRITE,
62 SERVER_EXECUTE,
63 SERVER_ALL_ACCESS
66 const struct generic_mapping printserver_std_mapping = {
67 SERVER_READ,
68 SERVER_WRITE,
69 SERVER_EXECUTE,
70 SERVER_ALL_ACCESS
73 /* Map generic permissions to job object specific permissions */
75 const struct generic_mapping job_generic_mapping = {
76 JOB_READ,
77 JOB_WRITE,
78 JOB_EXECUTE,
79 JOB_ALL_ACCESS
82 /* We need one default form to support our default printer. Msoft adds the
83 forms it wants and in the ORDER it wants them (note: DEVMODE papersize is an
84 array index). Letter is always first, so (for the current code) additions
85 always put things in the correct order. */
86 static const nt_forms_struct default_forms[] = {
87 {"Letter",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
88 {"Letter Small",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
89 {"Tabloid",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
90 {"Ledger",0x1,0x696b8,0x44368,0x0,0x0,0x696b8,0x44368},
91 {"Legal",0x1,0x34b5c,0x56d10,0x0,0x0,0x34b5c,0x56d10},
92 {"Statement",0x1,0x221b4,0x34b5c,0x0,0x0,0x221b4,0x34b5c},
93 {"Executive",0x1,0x2cf56,0x411cc,0x0,0x0,0x2cf56,0x411cc},
94 {"A3",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
95 {"A4",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
96 {"A4 Small",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
97 {"A5",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
98 {"B4 (JIS)",0x1,0x3ebe8,0x58de0,0x0,0x0,0x3ebe8,0x58de0},
99 {"B5 (JIS)",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
100 {"Folio",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
101 {"Quarto",0x1,0x347d8,0x43238,0x0,0x0,0x347d8,0x43238},
102 {"10x14",0x1,0x3e030,0x56d10,0x0,0x0,0x3e030,0x56d10},
103 {"11x17",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
104 {"Note",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
105 {"Envelope #9",0x1,0x18079,0x37091,0x0,0x0,0x18079,0x37091},
106 {"Envelope #10",0x1,0x19947,0x3ae94,0x0,0x0,0x19947,0x3ae94},
107 {"Envelope #11",0x1,0x1be7c,0x40565,0x0,0x0,0x1be7c,0x40565},
108 {"Envelope #12",0x1,0x1d74a,0x44368,0x0,0x0,0x1d74a,0x44368},
109 {"Envelope #14",0x1,0x1f018,0x47504,0x0,0x0,0x1f018,0x47504},
110 {"C size sheet",0x1,0x696b8,0x886d0,0x0,0x0,0x696b8,0x886d0},
111 {"D size sheet",0x1,0x886d0,0xd2d70,0x0,0x0,0x886d0,0xd2d70},
112 {"E size sheet",0x1,0xd2d70,0x110da0,0x0,0x0,0xd2d70,0x110da0},
113 {"Envelope DL",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
114 {"Envelope C5",0x1,0x278d0,0x37e88,0x0,0x0,0x278d0,0x37e88},
115 {"Envelope C3",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
116 {"Envelope C4",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
117 {"Envelope C6",0x1,0x1bd50,0x278d0,0x0,0x0,0x1bd50,0x278d0},
118 {"Envelope C65",0x1,0x1bd50,0x37e88,0x0,0x0,0x1bd50,0x37e88},
119 {"Envelope B4",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
120 {"Envelope B5",0x1,0x2af80,0x3d090,0x0,0x0,0x2af80,0x3d090},
121 {"Envelope B6",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
122 {"Envelope",0x1,0x1adb0,0x38270,0x0,0x0,0x1adb0,0x38270},
123 {"Envelope Monarch",0x1,0x18079,0x2e824,0x0,0x0,0x18079,0x2e824},
124 {"6 3/4 Envelope",0x1,0x167ab,0x284ec,0x0,0x0,0x167ab,0x284ec},
125 {"US Std Fanfold",0x1,0x5c3e1,0x44368,0x0,0x0,0x5c3e1,0x44368},
126 {"German Std Fanfold",0x1,0x34b5c,0x4a6a0,0x0,0x0,0x34b5c,0x4a6a0},
127 {"German Legal Fanfold",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
128 {"B4 (ISO)",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
129 {"Japanese Postcard",0x1,0x186a0,0x24220,0x0,0x0,0x186a0,0x24220},
130 {"9x11",0x1,0x37cf8,0x44368,0x0,0x0,0x37cf8,0x44368},
131 {"10x11",0x1,0x3e030,0x44368,0x0,0x0,0x3e030,0x44368},
132 {"15x11",0x1,0x5d048,0x44368,0x0,0x0,0x5d048,0x44368},
133 {"Envelope Invite",0x1,0x35b60,0x35b60,0x0,0x0,0x35b60,0x35b60},
134 {"Reserved48",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
135 {"Reserved49",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
136 {"Letter Extra",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
137 {"Legal Extra",0x1,0x3ae94,0x5d048,0x0,0x0,0x3ae94,0x5d048},
138 {"Tabloid Extra",0x1,0x4a6a0,0x6f9f0,0x0,0x0,0x4a6a0,0x6f9f0},
139 {"A4 Extra",0x1,0x397c2,0x4eb16,0x0,0x0,0x397c2,0x4eb16},
140 {"Letter Transverse",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
141 {"A4 Transverse",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
142 {"Letter Extra Transverse",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
143 {"Super A",0x1,0x376b8,0x56ea0,0x0,0x0,0x376b8,0x56ea0},
144 {"Super B",0x1,0x4a768,0x76e58,0x0,0x0,0x4a768,0x76e58},
145 {"Letter Plus",0x1,0x34b5c,0x4eb16,0x0,0x0,0x34b5c,0x4eb16},
146 {"A4 Plus",0x1,0x33450,0x50910,0x0,0x0,0x33450,0x50910},
147 {"A5 Transverse",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
148 {"B5 (JIS) Transverse",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
149 {"A3 Extra",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
150 {"A5 Extra",0x1,0x2a7b0,0x395f8,0x0,0x0,0x2a7b0,0x395f8},
151 {"B5 (ISO) Extra",0x1,0x31128,0x43620,0x0,0x0,0x31128,0x43620},
152 {"A2",0x1,0x668a0,0x91050,0x0,0x0,0x668a0,0x91050},
153 {"A3 Transverse",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
154 {"A3 Extra Transverse",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
155 {"Japanese Double Postcard",0x1,0x30d40,0x24220,0x0,0x0,0x30d40,0x24220},
156 {"A6",0x1,0x19a28,0x24220,0x0,0x0,0x19a28,0x24220},
157 {"Japanese Envelope Kaku #2",0x1,0x3a980,0x510e0,0x0,0x0,0x3a980,0x510e0},
158 {"Japanese Envelope Kaku #3",0x1,0x34bc0,0x43a08,0x0,0x0,0x34bc0,0x43a08},
159 {"Japanese Envelope Chou #3",0x1,0x1d4c0,0x395f8,0x0,0x0,0x1d4c0,0x395f8},
160 {"Japanese Envelope Chou #4",0x1,0x15f90,0x320c8,0x0,0x0,0x15f90,0x320c8},
161 {"Letter Rotated",0x1,0x44368,0x34b5c,0x0,0x0,0x44368,0x34b5c},
162 {"A3 Rotated",0x1,0x668a0,0x48828,0x0,0x0,0x668a0,0x48828},
163 {"A4 Rotated",0x1,0x48828,0x33450,0x0,0x0,0x48828,0x33450},
164 {"A5 Rotated",0x1,0x33450,0x24220,0x0,0x0,0x33450,0x24220},
165 {"B4 (JIS) Rotated",0x1,0x58de0,0x3ebe8,0x0,0x0,0x58de0,0x3ebe8},
166 {"B5 (JIS) Rotated",0x1,0x3ebe8,0x2c6f0,0x0,0x0,0x3ebe8,0x2c6f0},
167 {"Japanese Postcard Rotated",0x1,0x24220,0x186a0,0x0,0x0,0x24220,0x186a0},
168 {"Double Japan Postcard Rotated",0x1,0x24220,0x30d40,0x0,0x0,0x24220,0x30d40},
169 {"A6 Rotated",0x1,0x24220,0x19a28,0x0,0x0,0x24220,0x19a28},
170 {"Japan Envelope Kaku #2 Rotated",0x1,0x510e0,0x3a980,0x0,0x0,0x510e0,0x3a980},
171 {"Japan Envelope Kaku #3 Rotated",0x1,0x43a08,0x34bc0,0x0,0x0,0x43a08, 0x34bc0},
172 {"Japan Envelope Chou #3 Rotated",0x1,0x395f8,0x1d4c0,0x0,0x0,0x395f8,0x1d4c0},
173 {"Japan Envelope Chou #4 Rotated",0x1,0x320c8,0x15f90,0x0,0x0,0x320c8,0x15f90},
174 {"B6 (JIS)",0x1,0x1f400,0x2c6f0,0x0,0x0,0x1f400,0x2c6f0},
175 {"B6 (JIS) Rotated",0x1,0x2c6f0,0x1f400,0x0,0x0,0x2c6f0,0x1f400},
176 {"12x11",0x1,0x4a724,0x443e1,0x0,0x0,0x4a724,0x443e1},
177 {"Japan Envelope You #4",0x1,0x19a28,0x395f8,0x0,0x0,0x19a28,0x395f8},
178 {"Japan Envelope You #4 Rotated",0x1,0x395f8,0x19a28,0x0,0x0,0x395f8,0x19a28},
179 {"PRC 16K",0x1,0x2de60,0x3f7a0,0x0,0x0,0x2de60,0x3f7a0},
180 {"PRC 32K",0x1,0x1fbd0,0x2cec0,0x0,0x0,0x1fbd0,0x2cec0},
181 {"PRC 32K(Big)",0x1,0x222e0,0x318f8,0x0,0x0,0x222e0,0x318f8},
182 {"PRC Envelope #1",0x1,0x18e70,0x28488,0x0,0x0,0x18e70,0x28488},
183 {"PRC Envelope #2",0x1,0x18e70,0x2af80,0x0,0x0,0x18e70,0x2af80},
184 {"PRC Envelope #3",0x1,0x1e848,0x2af80,0x0,0x0,0x1e848,0x2af80},
185 {"PRC Envelope #4",0x1,0x1adb0,0x32c80,0x0,0x0,0x1adb0,0x32c80},
186 {"PRC Envelope #5",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
187 {"PRC Envelope #6",0x1,0x1d4c0,0x38270,0x0,0x0,0x1d4c0,0x38270},
188 {"PRC Envelope #7",0x1,0x27100,0x38270,0x0,0x0,0x27100,0x38270},
189 {"PRC Envelope #8",0x1,0x1d4c0,0x4b708,0x0,0x0,0x1d4c0,0x4b708},
190 {"PRC Envelope #9",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
191 {"PRC Envelope #10",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
192 {"PRC 16K Rotated",0x1,0x3f7a0,0x2de60,0x0,0x0,0x3f7a0,0x2de60},
193 {"PRC 32K Rotated",0x1,0x2cec0,0x1fbd0,0x0,0x0,0x2cec0,0x1fbd0},
194 {"PRC 32K(Big) Rotated",0x1,0x318f8,0x222e0,0x0,0x0,0x318f8,0x222e0},
195 {"PRC Envelope #1 Rotated",0x1,0x28488,0x18e70,0x0,0x0,0x28488,0x18e70},
196 {"PRC Envelope #2 Rotated",0x1,0x2af80,0x18e70,0x0,0x0,0x2af80,0x18e70},
197 {"PRC Envelope #3 Rotated",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
198 {"PRC Envelope #4 Rotated",0x1,0x32c80,0x1adb0,0x0,0x0,0x32c80,0x1adb0},
199 {"PRC Envelope #5 Rotated",0x1,0x35b60,0x1adb0,0x0,0x0,0x35b60,0x1adb0},
200 {"PRC Envelope #6 Rotated",0x1,0x38270,0x1d4c0,0x0,0x0,0x38270,0x1d4c0},
201 {"PRC Envelope #7 Rotated",0x1,0x38270,0x27100,0x0,0x0,0x38270,0x27100},
202 {"PRC Envelope #8 Rotated",0x1,0x4b708,0x1d4c0,0x0,0x0,0x4b708,0x1d4c0},
203 {"PRC Envelope #9 Rotated",0x1,0x4f1a0,0x37e88,0x0,0x0,0x4f1a0,0x37e88},
204 {"PRC Envelope #10 Rotated",0x1,0x6fd10,0x4f1a0,0x0,0x0,0x6fd10,0x4f1a0}
207 struct table_node {
208 const char *long_archi;
209 const char *short_archi;
210 int version;
213 #define SPL_ARCH_WIN40 "WIN40"
214 #define SPL_ARCH_W32X86 "W32X86"
215 #define SPL_ARCH_W32MIPS "W32MIPS"
216 #define SPL_ARCH_W32ALPHA "W32ALPHA"
217 #define SPL_ARCH_W32PPC "W32PPC"
218 #define SPL_ARCH_IA64 "IA64"
219 #define SPL_ARCH_X64 "x64"
221 static const struct table_node archi_table[]= {
223 {"Windows 4.0", SPL_ARCH_WIN40, 0 },
224 {"Windows NT x86", SPL_ARCH_W32X86, 2 },
225 {"Windows NT R4000", SPL_ARCH_W32MIPS, 2 },
226 {"Windows NT Alpha_AXP", SPL_ARCH_W32ALPHA, 2 },
227 {"Windows NT PowerPC", SPL_ARCH_W32PPC, 2 },
228 {"Windows IA64", SPL_ARCH_IA64, 3 },
229 {"Windows x64", SPL_ARCH_X64, 3 },
230 {NULL, "", -1 }
234 /****************************************************************************
235 generate a new TDB_DATA key for storing a printer
236 ****************************************************************************/
238 static TDB_DATA make_printer_tdbkey(TALLOC_CTX *ctx, const char *sharename )
240 fstring share;
241 char *keystr = NULL;
242 TDB_DATA key;
244 fstrcpy(share, sharename);
245 strlower_m(share);
247 keystr = talloc_asprintf(ctx, "%s%s", PRINTERS_PREFIX, share);
248 key = string_term_tdb_data(keystr ? keystr : "");
250 return key;
253 /****************************************************************************
254 generate a new TDB_DATA key for storing a printer security descriptor
255 ****************************************************************************/
257 static TDB_DATA make_printers_secdesc_tdbkey(TALLOC_CTX *ctx,
258 const char* sharename )
260 fstring share;
261 char *keystr = NULL;
262 TDB_DATA key;
264 fstrcpy(share, sharename );
265 strlower_m(share);
267 keystr = talloc_asprintf(ctx, "%s%s", SECDESC_PREFIX, share);
268 key = string_term_tdb_data(keystr ? keystr : "");
270 return key;
273 /****************************************************************************
274 ****************************************************************************/
276 static bool upgrade_to_version_3(void)
278 TDB_DATA kbuf, newkey, dbuf;
280 DEBUG(0,("upgrade_to_version_3: upgrading print tdb's to version 3\n"));
282 for (kbuf = tdb_firstkey(tdb_drivers); kbuf.dptr;
283 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
285 dbuf = tdb_fetch(tdb_drivers, kbuf);
287 if (strncmp((const char *)kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) == 0) {
288 DEBUG(0,("upgrade_to_version_3:moving form\n"));
289 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) {
290 SAFE_FREE(dbuf.dptr);
291 DEBUG(0,("upgrade_to_version_3: failed to move form. Error (%s).\n", tdb_errorstr(tdb_forms)));
292 return False;
294 if (tdb_delete(tdb_drivers, kbuf) != 0) {
295 SAFE_FREE(dbuf.dptr);
296 DEBUG(0,("upgrade_to_version_3: failed to delete form. Error (%s)\n", tdb_errorstr(tdb_drivers)));
297 return False;
301 if (strncmp((const char *)kbuf.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX)) == 0) {
302 DEBUG(0,("upgrade_to_version_3:moving printer\n"));
303 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
304 SAFE_FREE(dbuf.dptr);
305 DEBUG(0,("upgrade_to_version_3: failed to move printer. Error (%s)\n", tdb_errorstr(tdb_printers)));
306 return False;
308 if (tdb_delete(tdb_drivers, kbuf) != 0) {
309 SAFE_FREE(dbuf.dptr);
310 DEBUG(0,("upgrade_to_version_3: failed to delete printer. Error (%s)\n", tdb_errorstr(tdb_drivers)));
311 return False;
315 if (strncmp((const char *)kbuf.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX)) == 0) {
316 DEBUG(0,("upgrade_to_version_3:moving secdesc\n"));
317 if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
318 SAFE_FREE(dbuf.dptr);
319 DEBUG(0,("upgrade_to_version_3: failed to move secdesc. Error (%s)\n", tdb_errorstr(tdb_printers)));
320 return False;
322 if (tdb_delete(tdb_drivers, kbuf) != 0) {
323 SAFE_FREE(dbuf.dptr);
324 DEBUG(0,("upgrade_to_version_3: failed to delete secdesc. Error (%s)\n", tdb_errorstr(tdb_drivers)));
325 return False;
329 SAFE_FREE(dbuf.dptr);
332 return True;
335 /*******************************************************************
336 Fix an issue with security descriptors. Printer sec_desc must
337 use more than the generic bits that were previously used
338 in <= 3.0.14a. They must also have a owner and group SID assigned.
339 Otherwise, any printers than have been migrated to a Windows
340 host using printmig.exe will not be accessible.
341 *******************************************************************/
343 static int sec_desc_upg_fn( TDB_CONTEXT *the_tdb, TDB_DATA key,
344 TDB_DATA data, void *state )
346 prs_struct ps;
347 SEC_DESC_BUF *sd_orig = NULL;
348 SEC_DESC_BUF *sd_new, *sd_store;
349 SEC_DESC *sec, *new_sec;
350 TALLOC_CTX *ctx = state;
351 int result, i;
352 uint32 sd_size;
353 size_t size_new_sec;
355 if (!data.dptr || data.dsize == 0) {
356 return 0;
359 if ( strncmp((const char *) key.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX) ) != 0 ) {
360 return 0;
363 /* upgrade the security descriptor */
365 ZERO_STRUCT( ps );
367 prs_init_empty( &ps, ctx, UNMARSHALL );
368 prs_give_memory( &ps, (char *)data.dptr, data.dsize, False );
370 if ( !sec_io_desc_buf( "sec_desc_upg_fn", &sd_orig, &ps, 1 ) ) {
371 /* delete bad entries */
372 DEBUG(0,("sec_desc_upg_fn: Failed to parse original sec_desc for %si. Deleting....\n",
373 (const char *)key.dptr ));
374 tdb_delete( tdb_printers, key );
375 prs_mem_free( &ps );
376 return 0;
379 if (!sd_orig) {
380 prs_mem_free( &ps );
381 return 0;
383 sec = sd_orig->sd;
385 /* is this even valid? */
387 if ( !sec->dacl ) {
388 prs_mem_free( &ps );
389 return 0;
392 /* update access masks */
394 for ( i=0; i<sec->dacl->num_aces; i++ ) {
395 switch ( sec->dacl->aces[i].access_mask ) {
396 case (GENERIC_READ_ACCESS | GENERIC_WRITE_ACCESS | GENERIC_EXECUTE_ACCESS):
397 sec->dacl->aces[i].access_mask = PRINTER_ACE_PRINT;
398 break;
400 case GENERIC_ALL_ACCESS:
401 sec->dacl->aces[i].access_mask = PRINTER_ACE_FULL_CONTROL;
402 break;
404 case READ_CONTROL_ACCESS:
405 sec->dacl->aces[i].access_mask = PRINTER_ACE_MANAGE_DOCUMENTS;
407 default: /* no change */
408 break;
412 /* create a new SEC_DESC with the appropriate owner and group SIDs */
414 new_sec = make_sec_desc( ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
415 &global_sid_Builtin_Administrators,
416 &global_sid_Builtin_Administrators,
417 NULL, NULL, &size_new_sec );
418 if (!new_sec) {
419 prs_mem_free( &ps );
420 return 0;
422 sd_new = make_sec_desc_buf( ctx, size_new_sec, new_sec );
423 if (!sd_new) {
424 prs_mem_free( &ps );
425 return 0;
428 if ( !(sd_store = sec_desc_merge( ctx, sd_new, sd_orig )) ) {
429 DEBUG(0,("sec_desc_upg_fn: Failed to update sec_desc for %s\n", key.dptr ));
430 prs_mem_free( &ps );
431 return 0;
434 prs_mem_free( &ps );
436 /* store it back */
438 sd_size = ndr_size_security_descriptor(sd_store->sd, 0)
439 + sizeof(SEC_DESC_BUF);
440 if ( !prs_init(&ps, sd_size, ctx, MARSHALL) ) {
441 DEBUG(0,("sec_desc_upg_fn: Failed to allocate prs memory for %s\n", key.dptr ));
442 return 0;
445 if ( !sec_io_desc_buf( "sec_desc_upg_fn", &sd_store, &ps, 1 ) ) {
446 DEBUG(0,("sec_desc_upg_fn: Failed to parse new sec_desc for %s\n", key.dptr ));
447 prs_mem_free( &ps );
448 return 0;
451 data.dptr = (uint8 *)prs_data_p( &ps );
452 data.dsize = sd_size;
454 result = tdb_store( tdb_printers, key, data, TDB_REPLACE );
456 prs_mem_free( &ps );
458 /* 0 to continue and non-zero to stop traversal */
460 return (result == -1);
463 /*******************************************************************
464 *******************************************************************/
466 static bool upgrade_to_version_4(void)
468 TALLOC_CTX *ctx;
469 int result;
471 DEBUG(0,("upgrade_to_version_4: upgrading printer security descriptors\n"));
473 if ( !(ctx = talloc_init( "upgrade_to_version_4" )) )
474 return False;
476 result = tdb_traverse( tdb_printers, sec_desc_upg_fn, ctx );
478 talloc_destroy( ctx );
480 return ( result != -1 );
483 /*******************************************************************
484 Fix an issue with security descriptors. Printer sec_desc must
485 use more than the generic bits that were previously used
486 in <= 3.0.14a. They must also have a owner and group SID assigned.
487 Otherwise, any printers than have been migrated to a Windows
488 host using printmig.exe will not be accessible.
489 *******************************************************************/
491 static int normalize_printers_fn( TDB_CONTEXT *the_tdb, TDB_DATA key,
492 TDB_DATA data, void *state )
494 TALLOC_CTX *ctx = talloc_tos();
495 TDB_DATA new_key;
497 if (!data.dptr || data.dsize == 0)
498 return 0;
500 /* upgrade printer records and security descriptors */
502 if ( strncmp((const char *) key.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX) ) == 0 ) {
503 new_key = make_printer_tdbkey(ctx, (const char *)key.dptr+strlen(PRINTERS_PREFIX) );
505 else if ( strncmp((const char *) key.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX) ) == 0 ) {
506 new_key = make_printers_secdesc_tdbkey(ctx, (const char *)key.dptr+strlen(SECDESC_PREFIX) );
508 else {
509 /* ignore this record */
510 return 0;
513 /* delete the original record and store under the normalized key */
515 if ( tdb_delete( the_tdb, key ) != 0 ) {
516 DEBUG(0,("normalize_printers_fn: tdb_delete for [%s] failed!\n",
517 key.dptr));
518 return 1;
521 if ( tdb_store( the_tdb, new_key, data, TDB_REPLACE) != 0 ) {
522 DEBUG(0,("normalize_printers_fn: failed to store new record for [%s]!\n",
523 key.dptr));
524 return 1;
527 return 0;
530 /*******************************************************************
531 *******************************************************************/
533 static bool upgrade_to_version_5(void)
535 TALLOC_CTX *ctx;
536 int result;
538 DEBUG(0,("upgrade_to_version_5: normalizing printer keys\n"));
540 if ( !(ctx = talloc_init( "upgrade_to_version_5" )) )
541 return False;
543 result = tdb_traverse( tdb_printers, normalize_printers_fn, NULL );
545 talloc_destroy( ctx );
547 return ( result != -1 );
550 /****************************************************************************
551 Open the NT printing tdbs. Done once before fork().
552 ****************************************************************************/
554 bool nt_printing_init(struct messaging_context *msg_ctx)
556 const char *vstring = "INFO/version";
557 WERROR win_rc;
558 int32 vers_id;
560 if ( tdb_drivers && tdb_printers && tdb_forms )
561 return True;
563 if (tdb_drivers)
564 tdb_close(tdb_drivers);
565 tdb_drivers = tdb_open_log(state_path("ntdrivers.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
566 if (!tdb_drivers) {
567 DEBUG(0,("nt_printing_init: Failed to open nt drivers database %s (%s)\n",
568 state_path("ntdrivers.tdb"), strerror(errno) ));
569 return False;
572 if (tdb_printers)
573 tdb_close(tdb_printers);
574 tdb_printers = tdb_open_log(state_path("ntprinters.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
575 if (!tdb_printers) {
576 DEBUG(0,("nt_printing_init: Failed to open nt printers database %s (%s)\n",
577 state_path("ntprinters.tdb"), strerror(errno) ));
578 return False;
581 if (tdb_forms)
582 tdb_close(tdb_forms);
583 tdb_forms = tdb_open_log(state_path("ntforms.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
584 if (!tdb_forms) {
585 DEBUG(0,("nt_printing_init: Failed to open nt forms database %s (%s)\n",
586 state_path("ntforms.tdb"), strerror(errno) ));
587 return False;
590 /* handle a Samba upgrade */
592 vers_id = tdb_fetch_int32(tdb_drivers, vstring);
593 if (vers_id == -1) {
594 DEBUG(10, ("Fresh database\n"));
595 tdb_store_int32( tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_5 );
596 vers_id = NTDRIVERS_DATABASE_VERSION_5;
599 if ( vers_id != NTDRIVERS_DATABASE_VERSION_5 ) {
601 if ((vers_id == NTDRIVERS_DATABASE_VERSION_1) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_1)) {
602 if (!upgrade_to_version_3())
603 return False;
604 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_3);
605 vers_id = NTDRIVERS_DATABASE_VERSION_3;
608 if ((vers_id == NTDRIVERS_DATABASE_VERSION_2) || (IREV(vers_id) == NTDRIVERS_DATABASE_VERSION_2)) {
609 /* Written on a bigendian machine with old fetch_int code. Save as le. */
610 /* The only upgrade between V2 and V3 is to save the version in little-endian. */
611 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_3);
612 vers_id = NTDRIVERS_DATABASE_VERSION_3;
615 if (vers_id == NTDRIVERS_DATABASE_VERSION_3 ) {
616 if ( !upgrade_to_version_4() )
617 return False;
618 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_4);
619 vers_id = NTDRIVERS_DATABASE_VERSION_4;
622 if (vers_id == NTDRIVERS_DATABASE_VERSION_4 ) {
623 if ( !upgrade_to_version_5() )
624 return False;
625 tdb_store_int32(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION_5);
626 vers_id = NTDRIVERS_DATABASE_VERSION_5;
630 if ( vers_id != NTDRIVERS_DATABASE_VERSION_5 ) {
631 DEBUG(0,("nt_printing_init: Unknown printer database version [%d]\n", vers_id));
632 return False;
636 update_c_setprinter(True);
639 * register callback to handle updating printers as new
640 * drivers are installed
643 messaging_register(msg_ctx, NULL, MSG_PRINTER_DRVUPGRADE,
644 do_drv_upgrade_printer);
647 * register callback to handle updating printer data
648 * when a driver is initialized
651 messaging_register(msg_ctx, NULL, MSG_PRINTERDATA_INIT_RESET,
652 reset_all_printerdata);
654 /* of course, none of the message callbacks matter if you don't
655 tell messages.c that you interested in receiving PRINT_GENERAL
656 msgs. This is done in claim_connection() */
659 if ( lp_security() == SEC_ADS ) {
660 win_rc = check_published_printers();
661 if (!W_ERROR_IS_OK(win_rc))
662 DEBUG(0, ("nt_printing_init: error checking published printers: %s\n", dos_errstr(win_rc)));
665 return True;
668 /*******************************************************************
669 Function to allow filename parsing "the old way".
670 ********************************************************************/
672 static char *driver_unix_convert(connection_struct *conn,
673 const char *old_name,
674 SMB_STRUCT_STAT *pst)
676 TALLOC_CTX *ctx = talloc_tos();
677 char *name = talloc_strdup(ctx, old_name);
678 char *new_name = NULL;
680 if (!name) {
681 return NULL;
683 unix_format(name);
684 name = unix_clean_name(ctx, name);
685 if (!name) {
686 return NULL;
688 trim_string(name,"/","/");
689 unix_convert(ctx,conn, name, false, &new_name, NULL, pst);
690 return new_name;
693 /*******************************************************************
694 tdb traversal function for counting printers.
695 ********************************************************************/
697 static int traverse_counting_printers(TDB_CONTEXT *t, TDB_DATA key,
698 TDB_DATA data, void *context)
700 int *printer_count = (int*)context;
702 if (memcmp(PRINTERS_PREFIX, key.dptr, sizeof(PRINTERS_PREFIX)-1) == 0) {
703 (*printer_count)++;
704 DEBUG(10,("traverse_counting_printers: printer = [%s] printer_count = %d\n", key.dptr, *printer_count));
707 return 0;
710 /*******************************************************************
711 Update the spooler global c_setprinter. This variable is initialized
712 when the parent smbd starts with the number of existing printers. It
713 is monotonically increased by the current number of printers *after*
714 each add or delete printer RPC. Only Microsoft knows why... JRR020119
715 ********************************************************************/
717 uint32 update_c_setprinter(bool initialize)
719 int32 c_setprinter;
720 int32 printer_count = 0;
722 tdb_lock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
724 /* Traverse the tdb, counting the printers */
725 tdb_traverse(tdb_printers, traverse_counting_printers, (void *)&printer_count);
727 /* If initializing, set c_setprinter to current printers count
728 * otherwise, bump it by the current printer count
730 if (!initialize)
731 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER) + printer_count;
732 else
733 c_setprinter = printer_count;
735 DEBUG(10,("update_c_setprinter: c_setprinter = %u\n", (unsigned int)c_setprinter));
736 tdb_store_int32(tdb_printers, GLOBAL_C_SETPRINTER, c_setprinter);
738 tdb_unlock_bystring(tdb_printers, GLOBAL_C_SETPRINTER);
740 return (uint32)c_setprinter;
743 /*******************************************************************
744 Get the spooler global c_setprinter, accounting for initialization.
745 ********************************************************************/
747 uint32 get_c_setprinter(void)
749 int32 c_setprinter = tdb_fetch_int32(tdb_printers, GLOBAL_C_SETPRINTER);
751 if (c_setprinter == (int32)-1)
752 c_setprinter = update_c_setprinter(True);
754 DEBUG(10,("get_c_setprinter: c_setprinter = %d\n", c_setprinter));
756 return (uint32)c_setprinter;
759 /****************************************************************************
760 Get builtin form struct list.
761 ****************************************************************************/
763 int get_builtin_ntforms(nt_forms_struct **list)
765 *list = (nt_forms_struct *)memdup(&default_forms[0], sizeof(default_forms));
766 if (!*list) {
767 return 0;
769 return sizeof(default_forms) / sizeof(default_forms[0]);
772 /****************************************************************************
773 get a builtin form struct
774 ****************************************************************************/
776 bool get_a_builtin_ntform(UNISTR2 *uni_formname,nt_forms_struct *form)
778 int i,count;
779 fstring form_name;
780 unistr2_to_ascii(form_name, uni_formname, sizeof(form_name));
781 DEBUGADD(6,("Looking for builtin form %s \n", form_name));
782 count = sizeof(default_forms) / sizeof(default_forms[0]);
783 for (i=0;i<count;i++) {
784 if (strequal(form_name,default_forms[i].name)) {
785 DEBUGADD(6,("Found builtin form %s \n", form_name));
786 memcpy(form,&default_forms[i],sizeof(*form));
787 break;
791 return (i !=count);
794 /****************************************************************************
795 get a form struct list.
796 ****************************************************************************/
798 int get_ntforms(nt_forms_struct **list)
800 TDB_DATA kbuf, newkey, dbuf;
801 nt_forms_struct form;
802 int ret;
803 int i;
804 int n = 0;
806 *list = NULL;
808 for (kbuf = tdb_firstkey(tdb_forms);
809 kbuf.dptr;
810 newkey = tdb_nextkey(tdb_forms, kbuf), safe_free(kbuf.dptr), kbuf=newkey)
812 if (strncmp((const char *)kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) != 0)
813 continue;
815 dbuf = tdb_fetch(tdb_forms, kbuf);
816 if (!dbuf.dptr)
817 continue;
819 fstrcpy(form.name, (const char *)kbuf.dptr+strlen(FORMS_PREFIX));
820 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddddddd",
821 &i, &form.flag, &form.width, &form.length, &form.left,
822 &form.top, &form.right, &form.bottom);
823 SAFE_FREE(dbuf.dptr);
824 if (ret != dbuf.dsize)
825 continue;
827 *list = SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1);
828 if (!*list) {
829 DEBUG(0,("get_ntforms: Realloc fail.\n"));
830 return 0;
832 (*list)[n] = form;
833 n++;
837 return n;
840 /****************************************************************************
841 write a form struct list
842 ****************************************************************************/
844 int write_ntforms(nt_forms_struct **list, int number)
846 TALLOC_CTX *ctx = talloc_tos();
847 char *buf = NULL;
848 char *key = NULL;
849 int len;
850 TDB_DATA dbuf;
851 int i;
853 for (i=0;i<number;i++) {
854 /* save index, so list is rebuilt in correct order */
855 len = tdb_pack(NULL, 0, "dddddddd",
856 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
857 (*list)[i].left, (*list)[i].top, (*list)[i].right,
858 (*list)[i].bottom);
859 if (!len) {
860 continue;
862 buf = TALLOC_ARRAY(ctx, char, len);
863 if (!buf) {
864 return 0;
866 len = tdb_pack((uint8 *)buf, len, "dddddddd",
867 i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
868 (*list)[i].left, (*list)[i].top, (*list)[i].right,
869 (*list)[i].bottom);
870 key = talloc_asprintf(ctx, "%s%s", FORMS_PREFIX, (*list)[i].name);
871 if (!key) {
872 return 0;
874 dbuf.dsize = len;
875 dbuf.dptr = (uint8 *)buf;
876 if (tdb_store_bystring(tdb_forms, key, dbuf, TDB_REPLACE) != 0) {
877 TALLOC_FREE(key);
878 TALLOC_FREE(buf);
879 break;
881 TALLOC_FREE(key);
882 TALLOC_FREE(buf);
885 return i;
888 /****************************************************************************
889 add a form struct at the end of the list
890 ****************************************************************************/
891 bool add_a_form(nt_forms_struct **list, const FORM *form, int *count)
893 int n=0;
894 bool update;
895 fstring form_name;
898 * NT tries to add forms even when
899 * they are already in the base
900 * only update the values if already present
903 update=False;
905 unistr2_to_ascii(form_name, &form->name, sizeof(form_name));
906 for (n=0; n<*count; n++) {
907 if ( strequal((*list)[n].name, form_name) ) {
908 update=True;
909 break;
913 if (update==False) {
914 if((*list=SMB_REALLOC_ARRAY(*list, nt_forms_struct, n+1)) == NULL) {
915 DEBUG(0,("add_a_form: failed to enlarge forms list!\n"));
916 return False;
918 unistr2_to_ascii((*list)[n].name, &form->name, sizeof((*list)[n].name));
919 (*count)++;
922 (*list)[n].flag=form->flags;
923 (*list)[n].width=form->size_x;
924 (*list)[n].length=form->size_y;
925 (*list)[n].left=form->left;
926 (*list)[n].top=form->top;
927 (*list)[n].right=form->right;
928 (*list)[n].bottom=form->bottom;
930 DEBUG(6,("add_a_form: Successfully %s form [%s]\n",
931 update ? "updated" : "added", form_name));
933 return True;
936 /****************************************************************************
937 Delete a named form struct.
938 ****************************************************************************/
940 bool delete_a_form(nt_forms_struct **list, UNISTR2 *del_name, int *count, WERROR *ret)
942 char *key = NULL;
943 int n=0;
944 fstring form_name;
946 *ret = WERR_OK;
948 unistr2_to_ascii(form_name, del_name, sizeof(form_name));
950 for (n=0; n<*count; n++) {
951 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
952 DEBUG(103, ("delete_a_form, [%s] in list\n", form_name));
953 break;
957 if (n == *count) {
958 DEBUG(10,("delete_a_form, [%s] not found\n", form_name));
959 *ret = WERR_INVALID_PARAM;
960 return False;
963 if (asprintf(&key, "%s%s", FORMS_PREFIX, (*list)[n].name) < 0) {
964 *ret = WERR_NOMEM;
965 return false;
967 if (tdb_delete_bystring(tdb_forms, key) != 0) {
968 SAFE_FREE(key);
969 *ret = WERR_NOMEM;
970 return False;
972 SAFE_FREE(key);
973 return true;
976 /****************************************************************************
977 Update a form struct.
978 ****************************************************************************/
980 void update_a_form(nt_forms_struct **list, const FORM *form, int count)
982 int n=0;
983 fstring form_name;
984 unistr2_to_ascii(form_name, &(form->name), sizeof(form_name));
986 DEBUG(106, ("[%s]\n", form_name));
987 for (n=0; n<count; n++) {
988 DEBUGADD(106, ("n [%d]:[%s]\n", n, (*list)[n].name));
989 if (!strncmp((*list)[n].name, form_name, strlen(form_name)))
990 break;
993 if (n==count) return;
995 (*list)[n].flag=form->flags;
996 (*list)[n].width=form->size_x;
997 (*list)[n].length=form->size_y;
998 (*list)[n].left=form->left;
999 (*list)[n].top=form->top;
1000 (*list)[n].right=form->right;
1001 (*list)[n].bottom=form->bottom;
1004 /****************************************************************************
1005 Get the nt drivers list.
1006 Traverse the database and look-up the matching names.
1007 ****************************************************************************/
1008 int get_ntdrivers(fstring **list, const char *architecture, uint32 version)
1010 int total=0;
1011 const char *short_archi;
1012 char *key = NULL;
1013 TDB_DATA kbuf, newkey;
1015 short_archi = get_short_archi(architecture);
1016 if (!short_archi) {
1017 return 0;
1020 if (asprintf(&key, "%s%s/%d/", DRIVERS_PREFIX,
1021 short_archi, version) < 0) {
1022 return 0;
1025 for (kbuf = tdb_firstkey(tdb_drivers);
1026 kbuf.dptr;
1027 newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
1029 if (strncmp((const char *)kbuf.dptr, key, strlen(key)) != 0)
1030 continue;
1032 if((*list = SMB_REALLOC_ARRAY(*list, fstring, total+1)) == NULL) {
1033 DEBUG(0,("get_ntdrivers: failed to enlarge list!\n"));
1034 SAFE_FREE(key);
1035 return -1;
1038 fstrcpy((*list)[total], (const char *)kbuf.dptr+strlen(key));
1039 total++;
1042 SAFE_FREE(key);
1043 return(total);
1046 /****************************************************************************
1047 Function to do the mapping between the long architecture name and
1048 the short one.
1049 ****************************************************************************/
1051 const char *get_short_archi(const char *long_archi)
1053 int i=-1;
1055 DEBUG(107,("Getting architecture dependant directory\n"));
1056 do {
1057 i++;
1058 } while ( (archi_table[i].long_archi!=NULL ) &&
1059 StrCaseCmp(long_archi, archi_table[i].long_archi) );
1061 if (archi_table[i].long_archi==NULL) {
1062 DEBUGADD(10,("Unknown architecture [%s] !\n", long_archi));
1063 return NULL;
1066 /* this might be client code - but shouldn't this be an fstrcpy etc? */
1068 DEBUGADD(108,("index: [%d]\n", i));
1069 DEBUGADD(108,("long architecture: [%s]\n", archi_table[i].long_archi));
1070 DEBUGADD(108,("short architecture: [%s]\n", archi_table[i].short_archi));
1072 return archi_table[i].short_archi;
1075 /****************************************************************************
1076 Version information in Microsoft files is held in a VS_VERSION_INFO structure.
1077 There are two case to be covered here: PE (Portable Executable) and NE (New
1078 Executable) files. Both files support the same INFO structure, but PE files
1079 store the signature in unicode, and NE files store it as !unicode.
1080 returns -1 on error, 1 on version info found, and 0 on no version info found.
1081 ****************************************************************************/
1083 static int get_file_version(files_struct *fsp, char *fname,uint32 *major, uint32 *minor)
1085 int i;
1086 char *buf = NULL;
1087 ssize_t byte_count;
1089 if ((buf=(char *)SMB_MALLOC(DOS_HEADER_SIZE)) == NULL) {
1090 DEBUG(0,("get_file_version: PE file [%s] DOS Header malloc failed bytes = %d\n",
1091 fname, DOS_HEADER_SIZE));
1092 goto error_exit;
1095 if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
1096 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %lu\n",
1097 fname, (unsigned long)byte_count));
1098 goto no_version_info;
1101 /* Is this really a DOS header? */
1102 if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
1103 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
1104 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
1105 goto no_version_info;
1108 /* Skip OEM header (if any) and the DOS stub to start of Windows header */
1109 if (SMB_VFS_LSEEK(fsp, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (SMB_OFF_T)-1) {
1110 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
1111 fname, errno));
1112 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1113 goto no_version_info;
1116 /* Note: DOS_HEADER_SIZE and NE_HEADER_SIZE are incidentally same */
1117 if ((byte_count = vfs_read_data(fsp, buf, NE_HEADER_SIZE)) < NE_HEADER_SIZE) {
1118 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %lu\n",
1119 fname, (unsigned long)byte_count));
1120 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1121 goto no_version_info;
1124 /* The header may be a PE (Portable Executable) or an NE (New Executable) */
1125 if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
1126 unsigned int num_sections;
1127 unsigned int section_table_bytes;
1129 /* Just skip over optional header to get to section table */
1130 if (SMB_VFS_LSEEK(fsp,
1131 SVAL(buf,PE_HEADER_OPTIONAL_HEADER_SIZE)-(NE_HEADER_SIZE-PE_HEADER_SIZE),
1132 SEEK_CUR) == (SMB_OFF_T)-1) {
1133 DEBUG(3,("get_file_version: File [%s] Windows optional header too short, errno = %d\n",
1134 fname, errno));
1135 goto error_exit;
1138 /* get the section table */
1139 num_sections = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
1140 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
1141 if (section_table_bytes == 0)
1142 goto error_exit;
1144 SAFE_FREE(buf);
1145 if ((buf=(char *)SMB_MALLOC(section_table_bytes)) == NULL) {
1146 DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
1147 fname, section_table_bytes));
1148 goto error_exit;
1151 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
1152 DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %lu\n",
1153 fname, (unsigned long)byte_count));
1154 goto error_exit;
1157 /* Iterate the section table looking for the resource section ".rsrc" */
1158 for (i = 0; i < num_sections; i++) {
1159 int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
1161 if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
1162 unsigned int section_pos = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
1163 unsigned int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
1165 if (section_bytes == 0)
1166 goto error_exit;
1168 SAFE_FREE(buf);
1169 if ((buf=(char *)SMB_MALLOC(section_bytes)) == NULL) {
1170 DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
1171 fname, section_bytes));
1172 goto error_exit;
1175 /* Seek to the start of the .rsrc section info */
1176 if (SMB_VFS_LSEEK(fsp, section_pos, SEEK_SET) == (SMB_OFF_T)-1) {
1177 DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
1178 fname, errno));
1179 goto error_exit;
1182 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
1183 DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %lu\n",
1184 fname, (unsigned long)byte_count));
1185 goto error_exit;
1188 if (section_bytes < VS_VERSION_INFO_UNICODE_SIZE)
1189 goto error_exit;
1191 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
1192 /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
1193 if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
1194 /* Align to next long address */
1195 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
1197 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
1198 *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
1199 *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
1201 DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
1202 fname, *major, *minor,
1203 (*major>>16)&0xffff, *major&0xffff,
1204 (*minor>>16)&0xffff, *minor&0xffff));
1205 SAFE_FREE(buf);
1206 return 1;
1213 /* Version info not found, fall back to origin date/time */
1214 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
1215 SAFE_FREE(buf);
1216 return 0;
1218 } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
1219 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
1220 DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
1221 fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
1222 /* At this point, we assume the file is in error. It still could be somthing
1223 * else besides a NE file, but it unlikely at this point. */
1224 goto error_exit;
1227 /* Allocate a bit more space to speed up things */
1228 SAFE_FREE(buf);
1229 if ((buf=(char *)SMB_MALLOC(VS_NE_BUF_SIZE)) == NULL) {
1230 DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes = %d\n",
1231 fname, PE_HEADER_SIZE));
1232 goto error_exit;
1235 /* This is a HACK! I got tired of trying to sort through the messy
1236 * 'NE' file format. If anyone wants to clean this up please have at
1237 * it, but this works. 'NE' files will eventually fade away. JRR */
1238 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
1239 /* Cover case that should not occur in a well formed 'NE' .dll file */
1240 if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
1242 for(i=0; i<byte_count; i++) {
1243 /* Fast skip past data that can't possibly match */
1244 if (buf[i] != 'V') continue;
1246 /* Potential match data crosses buf boundry, move it to beginning
1247 * of buf, and fill the buf with as much as it will hold. */
1248 if (i>byte_count-VS_VERSION_INFO_SIZE) {
1249 int bc;
1251 memcpy(buf, &buf[i], byte_count-i);
1252 if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
1253 (byte_count-i))) < 0) {
1255 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
1256 fname, errno));
1257 goto error_exit;
1260 byte_count = bc + (byte_count - i);
1261 if (byte_count<VS_VERSION_INFO_SIZE) break;
1263 i = 0;
1266 /* Check that the full signature string and the magic number that
1267 * follows exist (not a perfect solution, but the chances that this
1268 * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
1269 * twice, as it is simpler to read the code. */
1270 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
1271 /* Compute skip alignment to next long address */
1272 int skip = -(SMB_VFS_LSEEK(fsp, 0, SEEK_CUR) - (byte_count - i) +
1273 sizeof(VS_SIGNATURE)) & 3;
1274 if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
1276 *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
1277 *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
1278 DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
1279 fname, *major, *minor,
1280 (*major>>16)&0xffff, *major&0xffff,
1281 (*minor>>16)&0xffff, *minor&0xffff));
1282 SAFE_FREE(buf);
1283 return 1;
1288 /* Version info not found, fall back to origin date/time */
1289 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
1290 SAFE_FREE(buf);
1291 return 0;
1293 } else
1294 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
1295 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
1296 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
1298 no_version_info:
1299 SAFE_FREE(buf);
1300 return 0;
1302 error_exit:
1303 SAFE_FREE(buf);
1304 return -1;
1307 /****************************************************************************
1308 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
1309 share one or more files. During the MS installation process files are checked
1310 to insure that only a newer version of a shared file is installed over an
1311 older version. There are several possibilities for this comparison. If there
1312 is no previous version, the new one is newer (obviously). If either file is
1313 missing the version info structure, compare the creation date (on Unix use
1314 the modification date). Otherwise chose the numerically larger version number.
1315 ****************************************************************************/
1317 static int file_version_is_newer(connection_struct *conn, fstring new_file, fstring old_file)
1319 bool use_version = true;
1320 char *filepath = NULL;
1322 uint32 new_major;
1323 uint32 new_minor;
1324 time_t new_create_time;
1326 uint32 old_major;
1327 uint32 old_minor;
1328 time_t old_create_time;
1330 files_struct *fsp = NULL;
1331 SMB_STRUCT_STAT st;
1332 SMB_STRUCT_STAT stat_buf;
1334 NTSTATUS status;
1336 SET_STAT_INVALID(st);
1337 SET_STAT_INVALID(stat_buf);
1338 new_create_time = (time_t)0;
1339 old_create_time = (time_t)0;
1341 /* Get file version info (if available) for previous file (if it exists) */
1342 filepath = driver_unix_convert(conn,old_file,&stat_buf);
1343 if (!filepath) {
1344 goto error_exit;
1347 status = open_file_ntcreate(conn, NULL, filepath, &stat_buf,
1348 FILE_GENERIC_READ,
1349 FILE_SHARE_READ|FILE_SHARE_WRITE,
1350 FILE_OPEN,
1352 FILE_ATTRIBUTE_NORMAL,
1353 INTERNAL_OPEN_ONLY,
1354 NULL, &fsp);
1356 if (!NT_STATUS_IS_OK(status)) {
1357 /* Old file not found, so by definition new file is in fact newer */
1358 DEBUG(10,("file_version_is_newer: Can't open old file [%s], errno = %d\n",
1359 filepath, errno));
1360 return 1;
1362 } else {
1363 int ret = get_file_version(fsp, old_file, &old_major, &old_minor);
1364 if (ret == -1) {
1365 goto error_exit;
1368 if (!ret) {
1369 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1370 old_file));
1371 use_version = false;
1372 if (SMB_VFS_FSTAT(fsp, &st) == -1) {
1373 goto error_exit;
1375 old_create_time = st.st_mtime;
1376 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", old_create_time));
1379 close_file(fsp, NORMAL_CLOSE);
1381 /* Get file version info (if available) for new file */
1382 filepath = driver_unix_convert(conn,new_file,&stat_buf);
1383 if (!filepath) {
1384 goto error_exit;
1387 status = open_file_ntcreate(conn, NULL, filepath, &stat_buf,
1388 FILE_GENERIC_READ,
1389 FILE_SHARE_READ|FILE_SHARE_WRITE,
1390 FILE_OPEN,
1392 FILE_ATTRIBUTE_NORMAL,
1393 INTERNAL_OPEN_ONLY,
1394 NULL, &fsp);
1396 if (!NT_STATUS_IS_OK(status)) {
1397 /* New file not found, this shouldn't occur if the caller did its job */
1398 DEBUG(3,("file_version_is_newer: Can't open new file [%s], errno = %d\n",
1399 filepath, errno));
1400 goto error_exit;
1402 } else {
1403 int ret = get_file_version(fsp, new_file, &new_major, &new_minor);
1404 if (ret == -1) {
1405 goto error_exit;
1408 if (!ret) {
1409 DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
1410 new_file));
1411 use_version = false;
1412 if (SMB_VFS_FSTAT(fsp, &st) == -1) {
1413 goto error_exit;
1415 new_create_time = st.st_mtime;
1416 DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", new_create_time));
1419 close_file(fsp, NORMAL_CLOSE);
1421 if (use_version && (new_major != old_major || new_minor != old_minor)) {
1422 /* Compare versions and choose the larger version number */
1423 if (new_major > old_major ||
1424 (new_major == old_major && new_minor > old_minor)) {
1426 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1427 return 1;
1429 else {
1430 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1431 return 0;
1434 } else {
1435 /* Compare modification time/dates and choose the newest time/date */
1436 if (new_create_time > old_create_time) {
1437 DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
1438 return 1;
1440 else {
1441 DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
1442 return 0;
1446 error_exit:
1447 if(fsp)
1448 close_file(fsp, NORMAL_CLOSE);
1449 return -1;
1452 /****************************************************************************
1453 Determine the correct cVersion associated with an architecture and driver
1454 ****************************************************************************/
1455 static uint32 get_correct_cversion(const char *architecture, fstring driverpath_in,
1456 struct current_user *user, WERROR *perr)
1458 int cversion;
1459 NTSTATUS nt_status;
1460 char *driverpath = NULL;
1461 DATA_BLOB null_pw;
1462 fstring res_type;
1463 files_struct *fsp = NULL;
1464 SMB_STRUCT_STAT st;
1465 connection_struct *conn;
1466 NTSTATUS status;
1468 SET_STAT_INVALID(st);
1470 *perr = WERR_INVALID_PARAM;
1472 /* If architecture is Windows 95/98/ME, the version is always 0. */
1473 if (strcmp(architecture, SPL_ARCH_WIN40) == 0) {
1474 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
1475 *perr = WERR_OK;
1476 return 0;
1479 /* If architecture is Windows x64, the version is always 3. */
1480 if (strcmp(architecture, SPL_ARCH_X64) == 0) {
1481 DEBUG(10,("get_correct_cversion: Driver is x64, cversion = 3\n"));
1482 *perr = WERR_OK;
1483 return 3;
1487 * Connect to the print$ share under the same account as the user connected
1488 * to the rpc pipe. Note we must still be root to do this.
1491 /* Null password is ok - we are already an authenticated user... */
1492 null_pw = data_blob_null;
1493 fstrcpy(res_type, "A:");
1494 become_root();
1495 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1496 unbecome_root();
1498 if (conn == NULL) {
1499 DEBUG(0,("get_correct_cversion: Unable to connect\n"));
1500 *perr = ntstatus_to_werror(nt_status);
1501 return -1;
1504 /* We are temporarily becoming the connection user. */
1505 if (!become_user(conn, user->vuid)) {
1506 DEBUG(0,("get_correct_cversion: Can't become user!\n"));
1507 *perr = WERR_ACCESS_DENIED;
1508 return -1;
1511 /* Open the driver file (Portable Executable format) and determine the
1512 * deriver the cversion. */
1513 driverpath = talloc_asprintf(talloc_tos(),
1514 "%s/%s",
1515 architecture,
1516 driverpath_in);
1517 if (!driverpath) {
1518 *perr = WERR_NOMEM;
1519 goto error_exit;
1522 driverpath = driver_unix_convert(conn,driverpath,&st);
1523 if (!driverpath) {
1524 *perr = WERR_NOMEM;
1525 goto error_exit;
1528 if (!vfs_file_exist(conn, driverpath, &st)) {
1529 *perr = WERR_BADFILE;
1530 goto error_exit;
1533 status = open_file_ntcreate(conn, NULL, driverpath, &st,
1534 FILE_GENERIC_READ,
1535 FILE_SHARE_READ|FILE_SHARE_WRITE,
1536 FILE_OPEN,
1538 FILE_ATTRIBUTE_NORMAL,
1539 INTERNAL_OPEN_ONLY,
1540 NULL, &fsp);
1542 if (!NT_STATUS_IS_OK(status)) {
1543 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = %d\n",
1544 driverpath, errno));
1545 *perr = WERR_ACCESS_DENIED;
1546 goto error_exit;
1547 } else {
1548 uint32 major;
1549 uint32 minor;
1550 int ret = get_file_version(fsp, driverpath, &major, &minor);
1551 if (ret == -1) goto error_exit;
1553 if (!ret) {
1554 DEBUG(6,("get_correct_cversion: Version info not found [%s]\n", driverpath));
1555 goto error_exit;
1559 * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
1560 * for more details. Version in this case is not just the version of the
1561 * file, but the version in the sense of kernal mode (2) vs. user mode
1562 * (3) drivers. Other bits of the version fields are the version info.
1563 * JRR 010716
1565 cversion = major & 0x0000ffff;
1566 switch (cversion) {
1567 case 2: /* WinNT drivers */
1568 case 3: /* Win2K drivers */
1569 break;
1571 default:
1572 DEBUG(6,("get_correct_cversion: cversion invalid [%s] cversion = %d\n",
1573 driverpath, cversion));
1574 goto error_exit;
1577 DEBUG(10,("get_correct_cversion: Version info found [%s] major = 0x%x minor = 0x%x\n",
1578 driverpath, major, minor));
1581 DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
1582 driverpath, cversion));
1584 close_file(fsp, NORMAL_CLOSE);
1585 close_cnum(conn, user->vuid);
1586 unbecome_user();
1587 *perr = WERR_OK;
1588 return cversion;
1591 error_exit:
1593 if(fsp)
1594 close_file(fsp, NORMAL_CLOSE);
1596 close_cnum(conn, user->vuid);
1597 unbecome_user();
1598 return -1;
1601 /****************************************************************************
1602 ****************************************************************************/
1603 static WERROR clean_up_driver_struct_level_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver,
1604 struct current_user *user)
1606 const char *architecture;
1607 fstring new_name;
1608 char *p;
1609 int i;
1610 WERROR err;
1612 /* clean up the driver name.
1613 * we can get .\driver.dll
1614 * or worse c:\windows\system\driver.dll !
1616 /* using an intermediate string to not have overlaping memcpy()'s */
1617 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1618 fstrcpy(new_name, p+1);
1619 fstrcpy(driver->driverpath, new_name);
1622 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1623 fstrcpy(new_name, p+1);
1624 fstrcpy(driver->datafile, new_name);
1627 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1628 fstrcpy(new_name, p+1);
1629 fstrcpy(driver->configfile, new_name);
1632 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1633 fstrcpy(new_name, p+1);
1634 fstrcpy(driver->helpfile, new_name);
1637 if (driver->dependentfiles) {
1638 for (i=0; *driver->dependentfiles[i]; i++) {
1639 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1640 fstrcpy(new_name, p+1);
1641 fstrcpy(driver->dependentfiles[i], new_name);
1646 architecture = get_short_archi(driver->environment);
1647 if (!architecture) {
1648 return WERR_UNKNOWN_PRINTER_DRIVER;
1651 /* jfm:7/16/2000 the client always sends the cversion=0.
1652 * The server should check which version the driver is by reading
1653 * the PE header of driver->driverpath.
1655 * For Windows 95/98 the version is 0 (so the value sent is correct)
1656 * For Windows NT (the architecture doesn't matter)
1657 * NT 3.1: cversion=0
1658 * NT 3.5/3.51: cversion=1
1659 * NT 4: cversion=2
1660 * NT2K: cversion=3
1662 if ((driver->cversion = get_correct_cversion( architecture, driver->driverpath, user, &err)) == -1)
1663 return err;
1665 return WERR_OK;
1668 /****************************************************************************
1669 ****************************************************************************/
1670 static WERROR clean_up_driver_struct_level_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver, struct current_user *user)
1672 const char *architecture;
1673 fstring new_name;
1674 char *p;
1675 int i;
1676 WERROR err;
1678 /* clean up the driver name.
1679 * we can get .\driver.dll
1680 * or worse c:\windows\system\driver.dll !
1682 /* using an intermediate string to not have overlaping memcpy()'s */
1683 if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1684 fstrcpy(new_name, p+1);
1685 fstrcpy(driver->driverpath, new_name);
1688 if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1689 fstrcpy(new_name, p+1);
1690 fstrcpy(driver->datafile, new_name);
1693 if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1694 fstrcpy(new_name, p+1);
1695 fstrcpy(driver->configfile, new_name);
1698 if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1699 fstrcpy(new_name, p+1);
1700 fstrcpy(driver->helpfile, new_name);
1703 if (driver->dependentfiles) {
1704 for (i=0; *driver->dependentfiles[i]; i++) {
1705 if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1706 fstrcpy(new_name, p+1);
1707 fstrcpy(driver->dependentfiles[i], new_name);
1712 architecture = get_short_archi(driver->environment);
1713 if (!architecture) {
1714 return WERR_UNKNOWN_PRINTER_DRIVER;
1717 /* jfm:7/16/2000 the client always sends the cversion=0.
1718 * The server should check which version the driver is by reading
1719 * the PE header of driver->driverpath.
1721 * For Windows 95/98 the version is 0 (so the value sent is correct)
1722 * For Windows NT (the architecture doesn't matter)
1723 * NT 3.1: cversion=0
1724 * NT 3.5/3.51: cversion=1
1725 * NT 4: cversion=2
1726 * NT2K: cversion=3
1729 if ((driver->version = get_correct_cversion(architecture, driver->driverpath, user, &err)) == -1)
1730 return err;
1732 return WERR_OK;
1735 /****************************************************************************
1736 ****************************************************************************/
1737 WERROR clean_up_driver_struct(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract,
1738 uint32 level, struct current_user *user)
1740 switch (level) {
1741 case 3:
1743 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1744 driver=driver_abstract.info_3;
1745 return clean_up_driver_struct_level_3(driver, user);
1747 case 6:
1749 NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver;
1750 driver=driver_abstract.info_6;
1751 return clean_up_driver_struct_level_6(driver, user);
1753 default:
1754 return WERR_INVALID_PARAM;
1758 /****************************************************************************
1759 This function sucks and should be replaced. JRA.
1760 ****************************************************************************/
1762 static void convert_level_6_to_level3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *dst, NT_PRINTER_DRIVER_INFO_LEVEL_6 *src)
1764 dst->cversion = src->version;
1766 fstrcpy( dst->name, src->name);
1767 fstrcpy( dst->environment, src->environment);
1768 fstrcpy( dst->driverpath, src->driverpath);
1769 fstrcpy( dst->datafile, src->datafile);
1770 fstrcpy( dst->configfile, src->configfile);
1771 fstrcpy( dst->helpfile, src->helpfile);
1772 fstrcpy( dst->monitorname, src->monitorname);
1773 fstrcpy( dst->defaultdatatype, src->defaultdatatype);
1774 dst->dependentfiles = src->dependentfiles;
1777 #if 0 /* Debugging function */
1779 static char* ffmt(unsigned char *c){
1780 int i;
1781 static char ffmt_str[17];
1783 for (i=0; i<16; i++) {
1784 if ((c[i] < ' ') || (c[i] > '~'))
1785 ffmt_str[i]='.';
1786 else
1787 ffmt_str[i]=c[i];
1789 ffmt_str[16]='\0';
1790 return ffmt_str;
1793 #endif
1795 /****************************************************************************
1796 ****************************************************************************/
1797 WERROR move_driver_to_download_area(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract, uint32 level,
1798 struct current_user *user, WERROR *perr)
1800 NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1801 NT_PRINTER_DRIVER_INFO_LEVEL_3 converted_driver;
1802 const char *architecture;
1803 char *new_dir = NULL;
1804 char *old_name = NULL;
1805 char *new_name = NULL;
1806 DATA_BLOB null_pw;
1807 connection_struct *conn;
1808 NTSTATUS nt_status;
1809 fstring res_type;
1810 SMB_STRUCT_STAT st;
1811 int i;
1812 TALLOC_CTX *ctx = talloc_tos();
1813 int ver = 0;
1815 *perr = WERR_OK;
1817 if (level==3)
1818 driver=driver_abstract.info_3;
1819 else if (level==6) {
1820 convert_level_6_to_level3(&converted_driver, driver_abstract.info_6);
1821 driver = &converted_driver;
1822 } else {
1823 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)level ));
1824 return WERR_UNKNOWN_LEVEL;
1827 architecture = get_short_archi(driver->environment);
1828 if (!architecture) {
1829 return WERR_UNKNOWN_PRINTER_DRIVER;
1833 * Connect to the print$ share under the same account as the user connected to the rpc pipe.
1834 * Note we must be root to do this.
1837 null_pw = data_blob_null;
1838 fstrcpy(res_type, "A:");
1839 become_root();
1840 conn = make_connection_with_chdir("print$", null_pw, res_type, user->vuid, &nt_status);
1841 unbecome_root();
1843 if (conn == NULL) {
1844 DEBUG(0,("move_driver_to_download_area: Unable to connect\n"));
1845 *perr = ntstatus_to_werror(nt_status);
1846 return WERR_NO_SUCH_SHARE;
1850 * Save who we are - we are temporarily becoming the connection user.
1853 if (!become_user(conn, conn->vuid)) {
1854 DEBUG(0,("move_driver_to_download_area: Can't become user!\n"));
1855 return WERR_ACCESS_DENIED;
1858 /* WE ARE NOW RUNNING AS USER conn->vuid !!!!! */
1861 * make the directories version and version\driver_name
1862 * under the architecture directory.
1864 DEBUG(5,("Creating first directory\n"));
1865 new_dir = talloc_asprintf(ctx,
1866 "%s/%d",
1867 architecture,
1868 driver->cversion);
1869 if (!new_dir) {
1870 *perr = WERR_NOMEM;
1871 goto err_exit;
1873 new_dir = driver_unix_convert(conn,new_dir,&st);
1874 if (!new_dir) {
1875 *perr = WERR_NOMEM;
1876 goto err_exit;
1879 create_directory(conn, NULL, new_dir);
1881 /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1882 * listed for this driver which has already been moved, skip it (note:
1883 * drivers may list the same file name several times. Then check if the
1884 * file already exists in archi\cversion\, if so, check that the version
1885 * info (or time stamps if version info is unavailable) is newer (or the
1886 * date is later). If it is, move it to archi\cversion\filexxx.yyy.
1887 * Otherwise, delete the file.
1889 * If a file is not moved to archi\cversion\ because of an error, all the
1890 * rest of the 'unmoved' driver files are removed from archi\. If one or
1891 * more of the driver's files was already moved to archi\cversion\, it
1892 * potentially leaves the driver in a partially updated state. Version
1893 * trauma will most likely occur if an client attempts to use any printer
1894 * bound to the driver. Perhaps a rewrite to make sure the moves can be
1895 * done is appropriate... later JRR
1898 DEBUG(5,("Moving files now !\n"));
1900 if (driver->driverpath && strlen(driver->driverpath)) {
1901 new_name = talloc_asprintf(ctx,
1902 "%s/%s",
1903 architecture,
1904 driver->driverpath);
1905 if (!new_name) {
1906 *perr = WERR_NOMEM;
1907 goto err_exit;
1909 old_name = talloc_asprintf(ctx,
1910 "%s/%s",
1911 new_dir,
1912 driver->driverpath);
1913 if (!old_name) {
1914 *perr = WERR_NOMEM;
1915 goto err_exit;
1918 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1919 new_name = driver_unix_convert(conn,new_name,&st);
1920 if (!new_name) {
1921 *perr = WERR_NOMEM;
1922 goto err_exit;
1924 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1925 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1926 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1927 new_name, old_name));
1928 *perr = WERR_ACCESS_DENIED;
1929 ver = -1;
1934 if (driver->datafile && strlen(driver->datafile)) {
1935 if (!strequal(driver->datafile, driver->driverpath)) {
1936 new_name = talloc_asprintf(ctx,
1937 "%s/%s",
1938 architecture,
1939 driver->datafile);
1940 if (!new_name) {
1941 *perr = WERR_NOMEM;
1942 goto err_exit;
1944 old_name = talloc_asprintf(ctx,
1945 "%s/%s",
1946 new_dir,
1947 driver->datafile);
1948 if (!old_name) {
1949 *perr = WERR_NOMEM;
1950 goto err_exit;
1952 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1953 new_name = driver_unix_convert(conn,new_name,&st);
1954 if (!new_name) {
1955 *perr = WERR_NOMEM;
1956 goto err_exit;
1958 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1959 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1960 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1961 new_name, old_name));
1962 *perr = WERR_ACCESS_DENIED;
1963 ver = -1;
1969 if (driver->configfile && strlen(driver->configfile)) {
1970 if (!strequal(driver->configfile, driver->driverpath) &&
1971 !strequal(driver->configfile, driver->datafile)) {
1972 new_name = talloc_asprintf(ctx,
1973 "%s/%s",
1974 architecture,
1975 driver->configfile);
1976 if (!new_name) {
1977 *perr = WERR_NOMEM;
1978 goto err_exit;
1980 old_name = talloc_asprintf(ctx,
1981 "%s/%s",
1982 new_dir,
1983 driver->configfile);
1984 if (!old_name) {
1985 *perr = WERR_NOMEM;
1986 goto err_exit;
1988 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1989 new_name = driver_unix_convert(conn,new_name,&st);
1990 if (!new_name) {
1991 *perr = WERR_NOMEM;
1992 goto err_exit;
1994 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
1995 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
1996 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1997 new_name, old_name));
1998 *perr = WERR_ACCESS_DENIED;
1999 ver = -1;
2005 if (driver->helpfile && strlen(driver->helpfile)) {
2006 if (!strequal(driver->helpfile, driver->driverpath) &&
2007 !strequal(driver->helpfile, driver->datafile) &&
2008 !strequal(driver->helpfile, driver->configfile)) {
2009 new_name = talloc_asprintf(ctx,
2010 "%s/%s",
2011 architecture,
2012 driver->helpfile);
2013 if (!new_name) {
2014 *perr = WERR_NOMEM;
2015 goto err_exit;
2017 old_name = talloc_asprintf(ctx,
2018 "%s/%s",
2019 new_dir,
2020 driver->helpfile);
2021 if (!old_name) {
2022 *perr = WERR_NOMEM;
2023 goto err_exit;
2025 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
2026 new_name = driver_unix_convert(conn,new_name,&st);
2027 if (!new_name) {
2028 *perr = WERR_NOMEM;
2029 goto err_exit;
2031 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name, OPENX_FILE_EXISTS_TRUNCATE|
2032 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
2033 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
2034 new_name, old_name));
2035 *perr = WERR_ACCESS_DENIED;
2036 ver = -1;
2042 if (driver->dependentfiles) {
2043 for (i=0; *driver->dependentfiles[i]; i++) {
2044 if (!strequal(driver->dependentfiles[i], driver->driverpath) &&
2045 !strequal(driver->dependentfiles[i], driver->datafile) &&
2046 !strequal(driver->dependentfiles[i], driver->configfile) &&
2047 !strequal(driver->dependentfiles[i], driver->helpfile)) {
2048 int j;
2049 for (j=0; j < i; j++) {
2050 if (strequal(driver->dependentfiles[i], driver->dependentfiles[j])) {
2051 goto NextDriver;
2055 new_name = talloc_asprintf(ctx,
2056 "%s/%s",
2057 architecture,
2058 driver->dependentfiles[i]);
2059 if (!new_name) {
2060 *perr = WERR_NOMEM;
2061 goto err_exit;
2063 old_name = talloc_asprintf(ctx,
2064 "%s/%s",
2065 new_dir,
2066 driver->dependentfiles[i]);
2067 if (!old_name) {
2068 *perr = WERR_NOMEM;
2069 goto err_exit;
2071 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
2072 new_name = driver_unix_convert(conn,new_name,&st);
2073 if (!new_name) {
2074 *perr = WERR_NOMEM;
2075 goto err_exit;
2077 if ( !NT_STATUS_IS_OK(copy_file(ctx,conn, new_name, old_name,
2078 OPENX_FILE_EXISTS_TRUNCATE|
2079 OPENX_FILE_CREATE_IF_NOT_EXIST, 0, False))) {
2080 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
2081 new_name, old_name));
2082 *perr = WERR_ACCESS_DENIED;
2083 ver = -1;
2087 NextDriver: ;
2091 err_exit:
2093 close_cnum(conn, user->vuid);
2094 unbecome_user();
2096 if (W_ERROR_EQUAL(*perr, WERR_OK)) {
2097 return WERR_OK;
2099 if (ver == -1) {
2100 return WERR_UNKNOWN_PRINTER_DRIVER;
2102 return (*perr);
2105 /****************************************************************************
2106 ****************************************************************************/
2108 static uint32 add_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver)
2110 TALLOC_CTX *ctx = talloc_tos();
2111 int len, buflen;
2112 const char *architecture;
2113 char *directory = NULL;
2114 fstring temp_name;
2115 char *key = NULL;
2116 uint8 *buf;
2117 int i, ret;
2118 TDB_DATA dbuf;
2120 architecture = get_short_archi(driver->environment);
2121 if (!architecture) {
2122 return (uint32)-1;
2125 /* The names are relative. We store them in the form: \print$\arch\version\driver.xxx
2126 * \\server is added in the rpc server layer.
2127 * It does make sense to NOT store the server's name in the printer TDB.
2130 directory = talloc_asprintf(ctx, "\\print$\\%s\\%d\\",
2131 architecture, driver->cversion);
2132 if (!directory) {
2133 return (uint32)-1;
2136 /* .inf files do not always list a file for each of the four standard files.
2137 * Don't prepend a path to a null filename, or client claims:
2138 * "The server on which the printer resides does not have a suitable
2139 * <printer driver name> printer driver installed. Click OK if you
2140 * wish to install the driver on your local machine."
2142 if (strlen(driver->driverpath)) {
2143 fstrcpy(temp_name, driver->driverpath);
2144 slprintf(driver->driverpath, sizeof(driver->driverpath)-1, "%s%s", directory, temp_name);
2147 if (strlen(driver->datafile)) {
2148 fstrcpy(temp_name, driver->datafile);
2149 slprintf(driver->datafile, sizeof(driver->datafile)-1, "%s%s", directory, temp_name);
2152 if (strlen(driver->configfile)) {
2153 fstrcpy(temp_name, driver->configfile);
2154 slprintf(driver->configfile, sizeof(driver->configfile)-1, "%s%s", directory, temp_name);
2157 if (strlen(driver->helpfile)) {
2158 fstrcpy(temp_name, driver->helpfile);
2159 slprintf(driver->helpfile, sizeof(driver->helpfile)-1, "%s%s", directory, temp_name);
2162 if (driver->dependentfiles) {
2163 for (i=0; *driver->dependentfiles[i]; i++) {
2164 fstrcpy(temp_name, driver->dependentfiles[i]);
2165 slprintf(driver->dependentfiles[i], sizeof(driver->dependentfiles[i])-1, "%s%s", directory, temp_name);
2169 key = talloc_asprintf(ctx, "%s%s/%d/%s", DRIVERS_PREFIX,
2170 architecture, driver->cversion, driver->name);
2171 if (!key) {
2172 return (uint32)-1;
2175 DEBUG(5,("add_a_printer_driver_3: Adding driver with key %s\n", key ));
2177 buf = NULL;
2178 len = buflen = 0;
2180 again:
2181 len = 0;
2182 len += tdb_pack(buf+len, buflen-len, "dffffffff",
2183 driver->cversion,
2184 driver->name,
2185 driver->environment,
2186 driver->driverpath,
2187 driver->datafile,
2188 driver->configfile,
2189 driver->helpfile,
2190 driver->monitorname,
2191 driver->defaultdatatype);
2193 if (driver->dependentfiles) {
2194 for (i=0; *driver->dependentfiles[i]; i++) {
2195 len += tdb_pack(buf+len, buflen-len, "f",
2196 driver->dependentfiles[i]);
2200 if (len != buflen) {
2201 buf = (uint8 *)SMB_REALLOC(buf, len);
2202 if (!buf) {
2203 DEBUG(0,("add_a_printer_driver_3: failed to enlarge buffer\n!"));
2204 ret = -1;
2205 goto done;
2207 buflen = len;
2208 goto again;
2211 dbuf.dptr = buf;
2212 dbuf.dsize = len;
2214 ret = tdb_store_bystring(tdb_drivers, key, dbuf, TDB_REPLACE);
2216 done:
2217 if (ret)
2218 DEBUG(0,("add_a_printer_driver_3: Adding driver with key %s failed.\n", key ));
2220 SAFE_FREE(buf);
2221 return ret;
2224 /****************************************************************************
2225 ****************************************************************************/
2226 static uint32 add_a_printer_driver_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver)
2228 NT_PRINTER_DRIVER_INFO_LEVEL_3 info3;
2230 ZERO_STRUCT(info3);
2231 info3.cversion = driver->version;
2232 fstrcpy(info3.name,driver->name);
2233 fstrcpy(info3.environment,driver->environment);
2234 fstrcpy(info3.driverpath,driver->driverpath);
2235 fstrcpy(info3.datafile,driver->datafile);
2236 fstrcpy(info3.configfile,driver->configfile);
2237 fstrcpy(info3.helpfile,driver->helpfile);
2238 fstrcpy(info3.monitorname,driver->monitorname);
2239 fstrcpy(info3.defaultdatatype,driver->defaultdatatype);
2240 info3.dependentfiles = driver->dependentfiles;
2242 return add_a_printer_driver_3(&info3);
2246 /****************************************************************************
2247 ****************************************************************************/
2248 static WERROR get_a_printer_driver_3_default(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, const char *driver, const char *arch)
2250 NT_PRINTER_DRIVER_INFO_LEVEL_3 info;
2252 ZERO_STRUCT(info);
2254 fstrcpy(info.name, driver);
2255 fstrcpy(info.defaultdatatype, "RAW");
2257 fstrcpy(info.driverpath, "");
2258 fstrcpy(info.datafile, "");
2259 fstrcpy(info.configfile, "");
2260 fstrcpy(info.helpfile, "");
2262 if ((info.dependentfiles= SMB_MALLOC_ARRAY(fstring, 2)) == NULL)
2263 return WERR_NOMEM;
2265 memset(info.dependentfiles, '\0', 2*sizeof(fstring));
2266 fstrcpy(info.dependentfiles[0], "");
2268 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&info, sizeof(info));
2269 if (!*info_ptr) {
2270 SAFE_FREE(info.dependentfiles);
2271 return WERR_NOMEM;
2274 return WERR_OK;
2277 /****************************************************************************
2278 ****************************************************************************/
2279 static WERROR get_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring drivername, const char *arch, uint32 version)
2281 NT_PRINTER_DRIVER_INFO_LEVEL_3 driver;
2282 TDB_DATA dbuf;
2283 const char *architecture;
2284 int len = 0;
2285 int i;
2286 char *key = NULL;
2288 ZERO_STRUCT(driver);
2290 architecture = get_short_archi(arch);
2291 if ( !architecture ) {
2292 return WERR_UNKNOWN_PRINTER_DRIVER;
2295 /* Windows 4.0 (i.e. win9x) should always use a version of 0 */
2297 if ( strcmp( architecture, SPL_ARCH_WIN40 ) == 0 )
2298 version = 0;
2300 DEBUG(8,("get_a_printer_driver_3: [%s%s/%d/%s]\n", DRIVERS_PREFIX, architecture, version, drivername));
2302 if (asprintf(&key, "%s%s/%d/%s", DRIVERS_PREFIX,
2303 architecture, version, drivername) < 0) {
2304 return WERR_NOMEM;
2307 dbuf = tdb_fetch_bystring(tdb_drivers, key);
2308 if (!dbuf.dptr) {
2309 SAFE_FREE(key);
2310 return WERR_UNKNOWN_PRINTER_DRIVER;
2313 len += tdb_unpack(dbuf.dptr, dbuf.dsize, "dffffffff",
2314 &driver.cversion,
2315 driver.name,
2316 driver.environment,
2317 driver.driverpath,
2318 driver.datafile,
2319 driver.configfile,
2320 driver.helpfile,
2321 driver.monitorname,
2322 driver.defaultdatatype);
2324 i=0;
2325 while (len < dbuf.dsize) {
2326 driver.dependentfiles = SMB_REALLOC_ARRAY(driver.dependentfiles, fstring, i+2);
2327 if ( !driver.dependentfiles ) {
2328 DEBUG(0,("get_a_printer_driver_3: failed to enlarge buffer!\n"));
2329 break;
2332 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f",
2333 &driver.dependentfiles[i]);
2334 i++;
2337 if ( driver.dependentfiles )
2338 fstrcpy( driver.dependentfiles[i], "" );
2340 SAFE_FREE(dbuf.dptr);
2341 SAFE_FREE(key);
2343 if (len != dbuf.dsize) {
2344 SAFE_FREE(driver.dependentfiles);
2346 return get_a_printer_driver_3_default(info_ptr, drivername, arch);
2349 *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&driver, sizeof(driver));
2350 if (!*info_ptr) {
2351 SAFE_FREE(driver.dependentfiles);
2352 return WERR_NOMEM;
2355 return WERR_OK;
2358 /****************************************************************************
2359 Debugging function, dump at level 6 the struct in the logs.
2360 ****************************************************************************/
2362 static uint32 dump_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
2364 uint32 result;
2365 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
2366 int i;
2368 DEBUG(20,("Dumping printer driver at level [%d]\n", level));
2370 switch (level)
2372 case 3:
2374 if (driver.info_3 == NULL)
2375 result=5;
2376 else {
2377 info3=driver.info_3;
2379 DEBUGADD(20,("version:[%d]\n", info3->cversion));
2380 DEBUGADD(20,("name:[%s]\n", info3->name));
2381 DEBUGADD(20,("environment:[%s]\n", info3->environment));
2382 DEBUGADD(20,("driverpath:[%s]\n", info3->driverpath));
2383 DEBUGADD(20,("datafile:[%s]\n", info3->datafile));
2384 DEBUGADD(20,("configfile:[%s]\n", info3->configfile));
2385 DEBUGADD(20,("helpfile:[%s]\n", info3->helpfile));
2386 DEBUGADD(20,("monitorname:[%s]\n", info3->monitorname));
2387 DEBUGADD(20,("defaultdatatype:[%s]\n", info3->defaultdatatype));
2389 for (i=0; info3->dependentfiles &&
2390 *info3->dependentfiles[i]; i++) {
2391 DEBUGADD(20,("dependentfile:[%s]\n",
2392 info3->dependentfiles[i]));
2394 result=0;
2396 break;
2398 default:
2399 DEBUGADD(20,("dump_a_printer_driver: Level %u not implemented\n", (unsigned int)level));
2400 result=1;
2401 break;
2404 return result;
2407 /****************************************************************************
2408 ****************************************************************************/
2409 int pack_devicemode(NT_DEVICEMODE *nt_devmode, uint8 *buf, int buflen)
2411 int len = 0;
2413 len += tdb_pack(buf+len, buflen-len, "p", nt_devmode);
2415 if (!nt_devmode)
2416 return len;
2418 len += tdb_pack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2419 nt_devmode->devicename,
2420 nt_devmode->formname,
2422 nt_devmode->specversion,
2423 nt_devmode->driverversion,
2424 nt_devmode->size,
2425 nt_devmode->driverextra,
2426 nt_devmode->orientation,
2427 nt_devmode->papersize,
2428 nt_devmode->paperlength,
2429 nt_devmode->paperwidth,
2430 nt_devmode->scale,
2431 nt_devmode->copies,
2432 nt_devmode->defaultsource,
2433 nt_devmode->printquality,
2434 nt_devmode->color,
2435 nt_devmode->duplex,
2436 nt_devmode->yresolution,
2437 nt_devmode->ttoption,
2438 nt_devmode->collate,
2439 nt_devmode->logpixels,
2441 nt_devmode->fields,
2442 nt_devmode->bitsperpel,
2443 nt_devmode->pelswidth,
2444 nt_devmode->pelsheight,
2445 nt_devmode->displayflags,
2446 nt_devmode->displayfrequency,
2447 nt_devmode->icmmethod,
2448 nt_devmode->icmintent,
2449 nt_devmode->mediatype,
2450 nt_devmode->dithertype,
2451 nt_devmode->reserved1,
2452 nt_devmode->reserved2,
2453 nt_devmode->panningwidth,
2454 nt_devmode->panningheight,
2455 nt_devmode->nt_dev_private);
2457 if (nt_devmode->nt_dev_private) {
2458 len += tdb_pack(buf+len, buflen-len, "B",
2459 nt_devmode->driverextra,
2460 nt_devmode->nt_dev_private);
2463 DEBUG(8,("Packed devicemode [%s]\n", nt_devmode->formname));
2465 return len;
2468 /****************************************************************************
2469 Pack all values in all printer keys
2470 ***************************************************************************/
2472 static int pack_values(NT_PRINTER_DATA *data, uint8 *buf, int buflen)
2474 int len = 0;
2475 int i, j;
2476 REGISTRY_VALUE *val;
2477 REGVAL_CTR *val_ctr;
2478 char *path = NULL;
2479 int num_values;
2481 if ( !data )
2482 return 0;
2484 /* loop over all keys */
2486 for ( i=0; i<data->num_keys; i++ ) {
2487 val_ctr = data->keys[i].values;
2488 num_values = regval_ctr_numvals( val_ctr );
2490 /* pack the keyname followed by a empty value */
2492 len += tdb_pack(buf+len, buflen-len, "pPdB",
2493 &data->keys[i].name,
2494 data->keys[i].name,
2495 REG_NONE,
2497 NULL);
2499 /* now loop over all values */
2501 for ( j=0; j<num_values; j++ ) {
2502 /* pathname should be stored as <key>\<value> */
2504 val = regval_ctr_specific_value( val_ctr, j );
2505 if (asprintf(&path, "%s\\%s",
2506 data->keys[i].name,
2507 regval_name(val)) < 0) {
2508 return -1;
2511 len += tdb_pack(buf+len, buflen-len, "pPdB",
2512 val,
2513 path,
2514 regval_type(val),
2515 regval_size(val),
2516 regval_data_p(val) );
2518 DEBUG(8,("specific: [%s], len: %d\n", regval_name(val), regval_size(val)));
2519 SAFE_FREE(path);
2524 /* terminator */
2526 len += tdb_pack(buf+len, buflen-len, "p", NULL);
2528 return len;
2532 /****************************************************************************
2533 Delete a printer - this just deletes the printer info file, any open
2534 handles are not affected.
2535 ****************************************************************************/
2537 uint32 del_a_printer(const char *sharename)
2539 TDB_DATA kbuf;
2540 char *printdb_path = NULL;
2541 TALLOC_CTX *ctx = talloc_tos();
2543 kbuf = make_printer_tdbkey(ctx, sharename);
2544 tdb_delete(tdb_printers, kbuf);
2546 kbuf= make_printers_secdesc_tdbkey(ctx, sharename);
2547 tdb_delete(tdb_printers, kbuf);
2549 close_all_print_db();
2551 if (geteuid() == 0) {
2552 if (asprintf(&printdb_path, "%s%s.tdb",
2553 lock_path("printing/"),
2554 sharename) < 0) {
2555 return (uint32)-1;
2557 unlink(printdb_path);
2558 SAFE_FREE(printdb_path);
2561 return 0;
2564 /****************************************************************************
2565 ****************************************************************************/
2566 static WERROR update_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info)
2568 uint8 *buf;
2569 int buflen, len;
2570 int retlen;
2571 WERROR ret;
2572 TDB_DATA kbuf, dbuf;
2575 * in addprinter: no servername and the printer is the name
2576 * in setprinter: servername is \\server
2577 * and printer is \\server\\printer
2579 * Samba manages only local printers.
2580 * we currently don't support things like i
2581 * path=\\other_server\printer
2583 * We only store the printername, not \\server\printername
2586 if ( info->servername[0] != '\0' ) {
2587 trim_string(info->printername, info->servername, NULL);
2588 trim_char(info->printername, '\\', '\0');
2589 info->servername[0]='\0';
2593 * JFM: one day I'll forget.
2594 * below that's info->portname because that's the SAMBA sharename
2595 * and I made NT 'thinks' it's the portname
2596 * the info->sharename is the thing you can name when you add a printer
2597 * that's the short-name when you create shared printer for 95/98
2598 * So I've made a limitation in SAMBA: you can only have 1 printer model
2599 * behind a SAMBA share.
2602 buf = NULL;
2603 buflen = 0;
2605 again:
2606 len = 0;
2607 len += tdb_pack(buf+len, buflen-len, "dddddddddddfffffPfffff",
2608 info->attributes,
2609 info->priority,
2610 info->default_priority,
2611 info->starttime,
2612 info->untiltime,
2613 info->status,
2614 info->cjobs,
2615 info->averageppm,
2616 info->changeid,
2617 info->c_setprinter,
2618 info->setuptime,
2619 info->servername,
2620 info->printername,
2621 info->sharename,
2622 info->portname,
2623 info->drivername,
2624 info->comment,
2625 info->location,
2626 info->sepfile,
2627 info->printprocessor,
2628 info->datatype,
2629 info->parameters);
2631 len += pack_devicemode(info->devmode, buf+len, buflen-len);
2632 retlen = pack_values( info->data, buf+len, buflen-len );
2633 if (retlen == -1) {
2634 ret = WERR_NOMEM;
2635 goto done;
2637 len += retlen;
2639 if (buflen != len) {
2640 buf = (uint8 *)SMB_REALLOC(buf, len);
2641 if (!buf) {
2642 DEBUG(0,("update_a_printer_2: failed to enlarge buffer!\n"));
2643 ret = WERR_NOMEM;
2644 goto done;
2646 buflen = len;
2647 goto again;
2650 kbuf = make_printer_tdbkey(talloc_tos(), info->sharename );
2652 dbuf.dptr = buf;
2653 dbuf.dsize = len;
2655 ret = (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) == 0? WERR_OK : WERR_NOMEM);
2657 done:
2658 if (!W_ERROR_IS_OK(ret))
2659 DEBUG(8, ("error updating printer to tdb on disk\n"));
2661 SAFE_FREE(buf);
2663 DEBUG(8,("packed printer [%s] with driver [%s] portname=[%s] len=%d\n",
2664 info->sharename, info->drivername, info->portname, len));
2666 return ret;
2670 /****************************************************************************
2671 Malloc and return an NT devicemode.
2672 ****************************************************************************/
2674 NT_DEVICEMODE *construct_nt_devicemode(const fstring default_devicename)
2677 char adevice[MAXDEVICENAME];
2678 NT_DEVICEMODE *nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE);
2680 if (nt_devmode == NULL) {
2681 DEBUG(0,("construct_nt_devicemode: malloc fail.\n"));
2682 return NULL;
2685 ZERO_STRUCTP(nt_devmode);
2687 slprintf(adevice, sizeof(adevice), "%s", default_devicename);
2688 fstrcpy(nt_devmode->devicename, adevice);
2690 fstrcpy(nt_devmode->formname, "Letter");
2692 nt_devmode->specversion = 0x0401;
2693 nt_devmode->driverversion = 0x0400;
2694 nt_devmode->size = 0x00DC;
2695 nt_devmode->driverextra = 0x0000;
2696 nt_devmode->fields = FORMNAME | TTOPTION | PRINTQUALITY |
2697 DEFAULTSOURCE | COPIES | SCALE |
2698 PAPERSIZE | ORIENTATION;
2699 nt_devmode->orientation = 1;
2700 nt_devmode->papersize = PAPER_LETTER;
2701 nt_devmode->paperlength = 0;
2702 nt_devmode->paperwidth = 0;
2703 nt_devmode->scale = 0x64;
2704 nt_devmode->copies = 1;
2705 nt_devmode->defaultsource = BIN_FORMSOURCE;
2706 nt_devmode->printquality = RES_HIGH; /* 0x0258 */
2707 nt_devmode->color = COLOR_MONOCHROME;
2708 nt_devmode->duplex = DUP_SIMPLEX;
2709 nt_devmode->yresolution = 0;
2710 nt_devmode->ttoption = TT_SUBDEV;
2711 nt_devmode->collate = COLLATE_FALSE;
2712 nt_devmode->icmmethod = 0;
2713 nt_devmode->icmintent = 0;
2714 nt_devmode->mediatype = 0;
2715 nt_devmode->dithertype = 0;
2717 /* non utilisés par un driver d'imprimante */
2718 nt_devmode->logpixels = 0;
2719 nt_devmode->bitsperpel = 0;
2720 nt_devmode->pelswidth = 0;
2721 nt_devmode->pelsheight = 0;
2722 nt_devmode->displayflags = 0;
2723 nt_devmode->displayfrequency = 0;
2724 nt_devmode->reserved1 = 0;
2725 nt_devmode->reserved2 = 0;
2726 nt_devmode->panningwidth = 0;
2727 nt_devmode->panningheight = 0;
2729 nt_devmode->nt_dev_private = NULL;
2730 return nt_devmode;
2733 /****************************************************************************
2734 Deepcopy an NT devicemode.
2735 ****************************************************************************/
2737 NT_DEVICEMODE *dup_nt_devicemode(NT_DEVICEMODE *nt_devicemode)
2739 NT_DEVICEMODE *new_nt_devicemode = NULL;
2741 if ( !nt_devicemode )
2742 return NULL;
2744 if ((new_nt_devicemode = (NT_DEVICEMODE *)memdup(nt_devicemode, sizeof(NT_DEVICEMODE))) == NULL) {
2745 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2746 return NULL;
2749 new_nt_devicemode->nt_dev_private = NULL;
2750 if (nt_devicemode->nt_dev_private != NULL) {
2751 if ((new_nt_devicemode->nt_dev_private = (uint8 *)memdup(nt_devicemode->nt_dev_private, nt_devicemode->driverextra)) == NULL) {
2752 SAFE_FREE(new_nt_devicemode);
2753 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2754 return NULL;
2758 return new_nt_devicemode;
2761 /****************************************************************************
2762 Clean up and deallocate a (maybe partially) allocated NT_DEVICEMODE.
2763 ****************************************************************************/
2765 void free_nt_devicemode(NT_DEVICEMODE **devmode_ptr)
2767 NT_DEVICEMODE *nt_devmode = *devmode_ptr;
2769 if(nt_devmode == NULL)
2770 return;
2772 DEBUG(106,("free_nt_devicemode: deleting DEVMODE\n"));
2774 SAFE_FREE(nt_devmode->nt_dev_private);
2775 SAFE_FREE(*devmode_ptr);
2778 /****************************************************************************
2779 Clean up and deallocate a (maybe partially) allocated NT_PRINTER_INFO_LEVEL_2.
2780 ****************************************************************************/
2782 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr)
2784 NT_PRINTER_INFO_LEVEL_2 *info = *info_ptr;
2786 if ( !info )
2787 return;
2789 free_nt_devicemode(&info->devmode);
2791 TALLOC_FREE( *info_ptr );
2795 /****************************************************************************
2796 ****************************************************************************/
2797 int unpack_devicemode(NT_DEVICEMODE **nt_devmode, const uint8 *buf, int buflen)
2799 int len = 0;
2800 int extra_len = 0;
2801 NT_DEVICEMODE devmode;
2803 ZERO_STRUCT(devmode);
2805 len += tdb_unpack(buf+len, buflen-len, "p", nt_devmode);
2807 if (!*nt_devmode) return len;
2809 len += tdb_unpack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2810 devmode.devicename,
2811 devmode.formname,
2813 &devmode.specversion,
2814 &devmode.driverversion,
2815 &devmode.size,
2816 &devmode.driverextra,
2817 &devmode.orientation,
2818 &devmode.papersize,
2819 &devmode.paperlength,
2820 &devmode.paperwidth,
2821 &devmode.scale,
2822 &devmode.copies,
2823 &devmode.defaultsource,
2824 &devmode.printquality,
2825 &devmode.color,
2826 &devmode.duplex,
2827 &devmode.yresolution,
2828 &devmode.ttoption,
2829 &devmode.collate,
2830 &devmode.logpixels,
2832 &devmode.fields,
2833 &devmode.bitsperpel,
2834 &devmode.pelswidth,
2835 &devmode.pelsheight,
2836 &devmode.displayflags,
2837 &devmode.displayfrequency,
2838 &devmode.icmmethod,
2839 &devmode.icmintent,
2840 &devmode.mediatype,
2841 &devmode.dithertype,
2842 &devmode.reserved1,
2843 &devmode.reserved2,
2844 &devmode.panningwidth,
2845 &devmode.panningheight,
2846 &devmode.nt_dev_private);
2848 if (devmode.nt_dev_private) {
2849 /* the len in tdb_unpack is an int value and
2850 * devmode.driverextra is only a short
2852 len += tdb_unpack(buf+len, buflen-len, "B", &extra_len, &devmode.nt_dev_private);
2853 devmode.driverextra=(uint16)extra_len;
2855 /* check to catch an invalid TDB entry so we don't segfault */
2856 if (devmode.driverextra == 0) {
2857 devmode.nt_dev_private = NULL;
2861 *nt_devmode = (NT_DEVICEMODE *)memdup(&devmode, sizeof(devmode));
2862 if (!*nt_devmode) {
2863 SAFE_FREE(devmode.nt_dev_private);
2864 return -1;
2867 DEBUG(8,("Unpacked devicemode [%s](%s)\n", devmode.devicename, devmode.formname));
2868 if (devmode.nt_dev_private)
2869 DEBUG(8,("with a private section of %d bytes\n", devmode.driverextra));
2871 return len;
2874 /****************************************************************************
2875 Allocate and initialize a new slot.
2876 ***************************************************************************/
2878 int add_new_printer_key( NT_PRINTER_DATA *data, const char *name )
2880 NT_PRINTER_KEY *d;
2881 int key_index;
2883 if ( !name || !data )
2884 return -1;
2886 /* allocate another slot in the NT_PRINTER_KEY array */
2888 if ( !(d = TALLOC_REALLOC_ARRAY( data, data->keys, NT_PRINTER_KEY, data->num_keys+1)) ) {
2889 DEBUG(0,("add_new_printer_key: Realloc() failed!\n"));
2890 return -1;
2893 data->keys = d;
2895 key_index = data->num_keys;
2897 /* initialze new key */
2899 data->keys[key_index].name = talloc_strdup( data, name );
2901 if ( !(data->keys[key_index].values = TALLOC_ZERO_P( data, REGVAL_CTR )) )
2902 return -1;
2904 data->num_keys++;
2906 DEBUG(10,("add_new_printer_key: Inserted new data key [%s]\n", name ));
2908 return key_index;
2911 /****************************************************************************
2912 search for a registry key name in the existing printer data
2913 ***************************************************************************/
2915 int delete_printer_key( NT_PRINTER_DATA *data, const char *name )
2917 int i;
2919 for ( i=0; i<data->num_keys; i++ ) {
2920 if ( strequal( data->keys[i].name, name ) ) {
2922 /* cleanup memory */
2924 TALLOC_FREE( data->keys[i].name );
2925 TALLOC_FREE( data->keys[i].values );
2927 /* if not the end of the array, move remaining elements down one slot */
2929 data->num_keys--;
2930 if ( data->num_keys && (i < data->num_keys) )
2931 memmove( &data->keys[i], &data->keys[i+1], sizeof(NT_PRINTER_KEY)*(data->num_keys-i) );
2933 break;
2938 return data->num_keys;
2941 /****************************************************************************
2942 search for a registry key name in the existing printer data
2943 ***************************************************************************/
2945 int lookup_printerkey( NT_PRINTER_DATA *data, const char *name )
2947 int key_index = -1;
2948 int i;
2950 if ( !data || !name )
2951 return -1;
2953 DEBUG(12,("lookup_printerkey: Looking for [%s]\n", name));
2955 /* loop over all existing keys */
2957 for ( i=0; i<data->num_keys; i++ ) {
2958 if ( strequal(data->keys[i].name, name) ) {
2959 DEBUG(12,("lookup_printerkey: Found [%s]!\n", name));
2960 key_index = i;
2961 break;
2966 return key_index;
2969 /****************************************************************************
2970 ***************************************************************************/
2972 int get_printer_subkeys( NT_PRINTER_DATA *data, const char* key, fstring **subkeys )
2974 int i, j;
2975 int key_len;
2976 int num_subkeys = 0;
2977 char *p;
2978 fstring *subkeys_ptr = NULL;
2979 fstring subkeyname;
2981 *subkeys = NULL;
2983 if ( !data )
2984 return 0;
2986 if ( !key )
2987 return -1;
2989 /* special case of asking for the top level printer data registry key names */
2991 if ( strlen(key) == 0 ) {
2992 for ( i=0; i<data->num_keys; i++ ) {
2994 /* found a match, so allocate space and copy the name */
2996 if ( !(subkeys_ptr = SMB_REALLOC_ARRAY( subkeys_ptr, fstring, num_subkeys+2)) ) {
2997 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
2998 num_subkeys+1));
2999 return -1;
3002 fstrcpy( subkeys_ptr[num_subkeys], data->keys[i].name );
3003 num_subkeys++;
3006 goto done;
3009 /* asking for the subkeys of some key */
3010 /* subkey paths are stored in the key name using '\' as the delimiter */
3012 for ( i=0; i<data->num_keys; i++ ) {
3013 if ( StrnCaseCmp(data->keys[i].name, key, strlen(key)) == 0 ) {
3015 /* if we found the exact key, then break */
3016 key_len = strlen( key );
3017 if ( strlen(data->keys[i].name) == key_len )
3018 break;
3020 /* get subkey path */
3022 p = data->keys[i].name + key_len;
3023 if ( *p == '\\' )
3024 p++;
3025 fstrcpy( subkeyname, p );
3026 if ( (p = strchr( subkeyname, '\\' )) )
3027 *p = '\0';
3029 /* don't add a key more than once */
3031 for ( j=0; j<num_subkeys; j++ ) {
3032 if ( strequal( subkeys_ptr[j], subkeyname ) )
3033 break;
3036 if ( j != num_subkeys )
3037 continue;
3039 /* found a match, so allocate space and copy the name */
3041 if ( !(subkeys_ptr = SMB_REALLOC_ARRAY( subkeys_ptr, fstring, num_subkeys+2)) ) {
3042 DEBUG(0,("get_printer_subkeys: Realloc failed for [%d] entries!\n",
3043 num_subkeys+1));
3044 return 0;
3047 fstrcpy( subkeys_ptr[num_subkeys], subkeyname );
3048 num_subkeys++;
3053 /* return error if the key was not found */
3055 if ( i == data->num_keys ) {
3056 SAFE_FREE(subkeys_ptr);
3057 return -1;
3060 done:
3061 /* tag off the end */
3063 if (num_subkeys)
3064 fstrcpy(subkeys_ptr[num_subkeys], "" );
3066 *subkeys = subkeys_ptr;
3068 return num_subkeys;
3071 #ifdef HAVE_ADS
3072 static void map_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3073 const char *sz)
3075 smb_ucs2_t conv_str[1024];
3076 size_t str_size;
3078 regval_ctr_delvalue(ctr, val_name);
3079 str_size = push_ucs2(NULL, conv_str, sz, sizeof(conv_str),
3080 STR_TERMINATE | STR_NOALIGN);
3081 regval_ctr_addvalue(ctr, val_name, REG_SZ,
3082 (char *) conv_str, str_size);
3085 static void map_dword_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3086 uint32 dword)
3088 regval_ctr_delvalue(ctr, val_name);
3089 regval_ctr_addvalue(ctr, val_name, REG_DWORD,
3090 (char *) &dword, sizeof(dword));
3093 static void map_bool_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3094 bool b)
3096 uint8 bin_bool = (b ? 1 : 0);
3097 regval_ctr_delvalue(ctr, val_name);
3098 regval_ctr_addvalue(ctr, val_name, REG_BINARY,
3099 (char *) &bin_bool, sizeof(bin_bool));
3102 static void map_single_multi_sz_into_ctr(REGVAL_CTR *ctr, const char *val_name,
3103 const char *multi_sz)
3105 smb_ucs2_t *conv_strs = NULL;
3106 size_t str_size;
3108 /* a multi-sz has to have a null string terminator, i.e., the last
3109 string must be followed by two nulls */
3110 str_size = strlen(multi_sz) + 2;
3111 conv_strs = SMB_CALLOC_ARRAY(smb_ucs2_t, str_size);
3112 if (!conv_strs) {
3113 return;
3116 /* Change to byte units. */
3117 str_size *= sizeof(smb_ucs2_t);
3118 push_ucs2(NULL, conv_strs, multi_sz, str_size,
3119 STR_TERMINATE | STR_NOALIGN);
3121 regval_ctr_delvalue(ctr, val_name);
3122 regval_ctr_addvalue(ctr, val_name, REG_MULTI_SZ,
3123 (char *) conv_strs, str_size);
3124 safe_free(conv_strs);
3128 /****************************************************************************
3129 * Map the NT_PRINTER_INFO_LEVEL_2 data into DsSpooler keys for publishing.
3131 * @param info2 NT_PRINTER_INFO_LEVEL_2 describing printer - gets modified
3132 * @return bool indicating success or failure
3133 ***************************************************************************/
3135 static bool map_nt_printer_info2_to_dsspooler(NT_PRINTER_INFO_LEVEL_2 *info2)
3137 REGVAL_CTR *ctr = NULL;
3138 fstring longname;
3139 const char *dnssuffix;
3140 char *allocated_string = NULL;
3141 const char *ascii_str;
3142 int i;
3144 if ((i = lookup_printerkey(info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
3145 i = add_new_printer_key(info2->data, SPOOL_DSSPOOLER_KEY);
3146 ctr = info2->data->keys[i].values;
3148 map_sz_into_ctr(ctr, SPOOL_REG_PRINTERNAME, info2->sharename);
3149 map_sz_into_ctr(ctr, SPOOL_REG_SHORTSERVERNAME, global_myname());
3151 /* we make the assumption that the netbios name is the same
3152 as the DNS name sinc ethe former will be what we used to
3153 join the domain */
3155 dnssuffix = get_mydnsdomname(talloc_tos());
3156 if (dnssuffix && *dnssuffix) {
3157 fstr_sprintf( longname, "%s.%s", global_myname(), dnssuffix );
3158 } else {
3159 fstrcpy( longname, global_myname() );
3162 map_sz_into_ctr(ctr, SPOOL_REG_SERVERNAME, longname);
3164 asprintf(&allocated_string, "\\\\%s\\%s", longname, info2->sharename);
3165 map_sz_into_ctr(ctr, SPOOL_REG_UNCNAME, allocated_string);
3166 SAFE_FREE(allocated_string);
3168 map_dword_into_ctr(ctr, SPOOL_REG_VERSIONNUMBER, 4);
3169 map_sz_into_ctr(ctr, SPOOL_REG_DRIVERNAME, info2->drivername);
3170 map_sz_into_ctr(ctr, SPOOL_REG_LOCATION, info2->location);
3171 map_sz_into_ctr(ctr, SPOOL_REG_DESCRIPTION, info2->comment);
3172 map_single_multi_sz_into_ctr(ctr, SPOOL_REG_PORTNAME, info2->portname);
3173 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSEPARATORFILE, info2->sepfile);
3174 map_dword_into_ctr(ctr, SPOOL_REG_PRINTSTARTTIME, info2->starttime);
3175 map_dword_into_ctr(ctr, SPOOL_REG_PRINTENDTIME, info2->untiltime);
3176 map_dword_into_ctr(ctr, SPOOL_REG_PRIORITY, info2->priority);
3178 map_bool_into_ctr(ctr, SPOOL_REG_PRINTKEEPPRINTEDJOBS,
3179 (info2->attributes &
3180 PRINTER_ATTRIBUTE_KEEPPRINTEDJOBS));
3182 switch (info2->attributes & 0x3) {
3183 case 0:
3184 ascii_str = SPOOL_REGVAL_PRINTWHILESPOOLING;
3185 break;
3186 case 1:
3187 ascii_str = SPOOL_REGVAL_PRINTAFTERSPOOLED;
3188 break;
3189 case 2:
3190 ascii_str = SPOOL_REGVAL_PRINTDIRECT;
3191 break;
3192 default:
3193 ascii_str = "unknown";
3195 map_sz_into_ctr(ctr, SPOOL_REG_PRINTSPOOLING, ascii_str);
3197 return True;
3200 /*****************************************************************
3201 ****************************************************************/
3203 static void store_printer_guid(NT_PRINTER_INFO_LEVEL_2 *info2,
3204 struct GUID guid)
3206 int i;
3207 REGVAL_CTR *ctr=NULL;
3208 UNISTR2 unistr_guid;
3210 /* find the DsSpooler key */
3211 if ((i = lookup_printerkey(info2->data, SPOOL_DSSPOOLER_KEY)) < 0)
3212 i = add_new_printer_key(info2->data, SPOOL_DSSPOOLER_KEY);
3213 ctr = info2->data->keys[i].values;
3215 regval_ctr_delvalue(ctr, "objectGUID");
3217 /* We used to store this as a REG_BINARY but that causes
3218 Vista to whine */
3220 ZERO_STRUCT( unistr_guid );
3222 init_unistr2( &unistr_guid, smb_uuid_string(talloc_tos(), guid),
3223 UNI_STR_TERMINATE );
3225 regval_ctr_addvalue(ctr, "objectGUID", REG_SZ,
3226 (char *)unistr_guid.buffer,
3227 unistr_guid.uni_max_len*2);
3231 static WERROR nt_printer_publish_ads(ADS_STRUCT *ads,
3232 NT_PRINTER_INFO_LEVEL *printer)
3234 ADS_STATUS ads_rc;
3235 LDAPMessage *res;
3236 char *prt_dn = NULL, *srv_dn, *srv_cn_0, *srv_cn_escaped, *sharename_escaped;
3237 char *srv_dn_utf8, **srv_cn_utf8;
3238 TALLOC_CTX *ctx;
3239 ADS_MODLIST mods;
3240 const char *attrs[] = {"objectGUID", NULL};
3241 struct GUID guid;
3242 WERROR win_rc = WERR_OK;
3243 size_t converted_size;
3245 DEBUG(5, ("publishing printer %s\n", printer->info_2->printername));
3247 /* figure out where to publish */
3248 ads_find_machine_acct(ads, &res, global_myname());
3250 /* We use ldap_get_dn here as we need the answer
3251 * in utf8 to call ldap_explode_dn(). JRA. */
3253 srv_dn_utf8 = ldap_get_dn((LDAP *)ads->ldap.ld, (LDAPMessage *)res);
3254 if (!srv_dn_utf8) {
3255 ads_destroy(&ads);
3256 return WERR_SERVER_UNAVAILABLE;
3258 ads_msgfree(ads, res);
3259 srv_cn_utf8 = ldap_explode_dn(srv_dn_utf8, 1);
3260 if (!srv_cn_utf8) {
3261 ldap_memfree(srv_dn_utf8);
3262 ads_destroy(&ads);
3263 return WERR_SERVER_UNAVAILABLE;
3265 /* Now convert to CH_UNIX. */
3266 if (!pull_utf8_allocate(&srv_dn, srv_dn_utf8, &converted_size)) {
3267 ldap_memfree(srv_dn_utf8);
3268 ldap_memfree(srv_cn_utf8);
3269 ads_destroy(&ads);
3270 return WERR_SERVER_UNAVAILABLE;
3272 if (!pull_utf8_allocate(&srv_cn_0, srv_cn_utf8[0], &converted_size)) {
3273 ldap_memfree(srv_dn_utf8);
3274 ldap_memfree(srv_cn_utf8);
3275 ads_destroy(&ads);
3276 SAFE_FREE(srv_dn);
3277 return WERR_SERVER_UNAVAILABLE;
3280 ldap_memfree(srv_dn_utf8);
3281 ldap_memfree(srv_cn_utf8);
3283 srv_cn_escaped = escape_rdn_val_string_alloc(srv_cn_0);
3284 if (!srv_cn_escaped) {
3285 SAFE_FREE(srv_cn_0);
3286 ldap_memfree(srv_dn_utf8);
3287 ads_destroy(&ads);
3288 return WERR_SERVER_UNAVAILABLE;
3290 sharename_escaped = escape_rdn_val_string_alloc(printer->info_2->sharename);
3291 if (!sharename_escaped) {
3292 SAFE_FREE(srv_cn_escaped);
3293 SAFE_FREE(srv_cn_0);
3294 ldap_memfree(srv_dn_utf8);
3295 ads_destroy(&ads);
3296 return WERR_SERVER_UNAVAILABLE;
3300 asprintf(&prt_dn, "cn=%s-%s,%s", srv_cn_escaped, sharename_escaped, srv_dn);
3302 SAFE_FREE(srv_dn);
3303 SAFE_FREE(srv_cn_0);
3304 SAFE_FREE(srv_cn_escaped);
3305 SAFE_FREE(sharename_escaped);
3307 /* build the ads mods */
3308 ctx = talloc_init("nt_printer_publish_ads");
3309 if (ctx == NULL) {
3310 SAFE_FREE(prt_dn);
3311 return WERR_NOMEM;
3314 mods = ads_init_mods(ctx);
3316 if (mods == NULL) {
3317 SAFE_FREE(prt_dn);
3318 talloc_destroy(ctx);
3319 return WERR_NOMEM;
3322 get_local_printer_publishing_data(ctx, &mods, printer->info_2->data);
3323 ads_mod_str(ctx, &mods, SPOOL_REG_PRINTERNAME,
3324 printer->info_2->sharename);
3326 /* publish it */
3327 ads_rc = ads_mod_printer_entry(ads, prt_dn, ctx, &mods);
3328 if (ads_rc.err.rc == LDAP_NO_SUCH_OBJECT) {
3329 int i;
3330 for (i=0; mods[i] != 0; i++)
3332 mods[i] = (LDAPMod *)-1;
3333 ads_rc = ads_add_printer_entry(ads, prt_dn, ctx, &mods);
3336 if (!ADS_ERR_OK(ads_rc))
3337 DEBUG(3, ("error publishing %s: %s\n", printer->info_2->sharename, ads_errstr(ads_rc)));
3339 talloc_destroy(ctx);
3341 /* retreive the guid and store it locally */
3342 if (ADS_ERR_OK(ads_search_dn(ads, &res, prt_dn, attrs))) {
3343 ZERO_STRUCT(guid);
3344 ads_pull_guid(ads, res, &guid);
3345 ads_msgfree(ads, res);
3346 store_printer_guid(printer->info_2, guid);
3347 win_rc = mod_a_printer(printer, 2);
3350 SAFE_FREE(prt_dn);
3351 return win_rc;
3354 static WERROR nt_printer_unpublish_ads(ADS_STRUCT *ads,
3355 NT_PRINTER_INFO_LEVEL *printer)
3357 ADS_STATUS ads_rc;
3358 LDAPMessage *res;
3359 char *prt_dn = NULL;
3361 DEBUG(5, ("unpublishing printer %s\n", printer->info_2->printername));
3363 /* remove the printer from the directory */
3364 ads_rc = ads_find_printer_on_server(ads, &res,
3365 printer->info_2->sharename, global_myname());
3367 if (ADS_ERR_OK(ads_rc) && ads_count_replies(ads, res)) {
3368 prt_dn = ads_get_dn(ads, res);
3369 if (!prt_dn) {
3370 ads_msgfree(ads, res);
3371 return WERR_NOMEM;
3373 ads_rc = ads_del_dn(ads, prt_dn);
3374 ads_memfree(ads, prt_dn);
3377 ads_msgfree(ads, res);
3378 return WERR_OK;
3381 /****************************************************************************
3382 * Publish a printer in the directory
3384 * @param snum describing printer service
3385 * @return WERROR indicating status of publishing
3386 ***************************************************************************/
3388 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
3390 ADS_STATUS ads_rc;
3391 ADS_STRUCT *ads = NULL;
3392 NT_PRINTER_INFO_LEVEL *printer = NULL;
3393 WERROR win_rc;
3395 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
3396 if (!W_ERROR_IS_OK(win_rc))
3397 goto done;
3399 switch (action) {
3400 case SPOOL_DS_PUBLISH:
3401 case SPOOL_DS_UPDATE:
3402 /* set the DsSpooler info and attributes */
3403 if (!(map_nt_printer_info2_to_dsspooler(printer->info_2))) {
3404 win_rc = WERR_NOMEM;
3405 goto done;
3408 printer->info_2->attributes |= PRINTER_ATTRIBUTE_PUBLISHED;
3409 break;
3410 case SPOOL_DS_UNPUBLISH:
3411 printer->info_2->attributes ^= PRINTER_ATTRIBUTE_PUBLISHED;
3412 break;
3413 default:
3414 win_rc = WERR_NOT_SUPPORTED;
3415 goto done;
3418 win_rc = mod_a_printer(printer, 2);
3419 if (!W_ERROR_IS_OK(win_rc)) {
3420 DEBUG(3, ("err %d saving data\n", W_ERROR_V(win_rc)));
3421 goto done;
3424 ads = ads_init(lp_realm(), lp_workgroup(), NULL);
3425 if (!ads) {
3426 DEBUG(3, ("ads_init() failed\n"));
3427 win_rc = WERR_SERVER_UNAVAILABLE;
3428 goto done;
3430 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
3431 SAFE_FREE(ads->auth.password);
3432 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
3433 NULL, NULL);
3435 /* ads_connect() will find the DC for us */
3436 ads_rc = ads_connect(ads);
3437 if (!ADS_ERR_OK(ads_rc)) {
3438 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
3439 win_rc = WERR_ACCESS_DENIED;
3440 goto done;
3443 switch (action) {
3444 case SPOOL_DS_PUBLISH:
3445 case SPOOL_DS_UPDATE:
3446 win_rc = nt_printer_publish_ads(ads, printer);
3447 break;
3448 case SPOOL_DS_UNPUBLISH:
3449 win_rc = nt_printer_unpublish_ads(ads, printer);
3450 break;
3453 done:
3454 free_a_printer(&printer, 2);
3455 ads_destroy(&ads);
3456 return win_rc;
3459 WERROR check_published_printers(void)
3461 ADS_STATUS ads_rc;
3462 ADS_STRUCT *ads = NULL;
3463 int snum;
3464 int n_services = lp_numservices();
3465 NT_PRINTER_INFO_LEVEL *printer = NULL;
3467 ads = ads_init(lp_realm(), lp_workgroup(), NULL);
3468 if (!ads) {
3469 DEBUG(3, ("ads_init() failed\n"));
3470 return WERR_SERVER_UNAVAILABLE;
3472 setenv(KRB5_ENV_CCNAME, "MEMORY:prtpub_cache", 1);
3473 SAFE_FREE(ads->auth.password);
3474 ads->auth.password = secrets_fetch_machine_password(lp_workgroup(),
3475 NULL, NULL);
3477 /* ads_connect() will find the DC for us */
3478 ads_rc = ads_connect(ads);
3479 if (!ADS_ERR_OK(ads_rc)) {
3480 DEBUG(3, ("ads_connect failed: %s\n", ads_errstr(ads_rc)));
3481 ads_destroy(&ads);
3482 ads_kdestroy("MEMORY:prtpub_cache");
3483 return WERR_ACCESS_DENIED;
3486 for (snum = 0; snum < n_services; snum++) {
3487 if (!(lp_snum_ok(snum) && lp_print_ok(snum)))
3488 continue;
3490 if (W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2,
3491 lp_servicename(snum))) &&
3492 (printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED))
3493 nt_printer_publish_ads(ads, printer);
3495 free_a_printer(&printer, 2);
3498 ads_destroy(&ads);
3499 ads_kdestroy("MEMORY:prtpub_cache");
3500 return WERR_OK;
3503 bool is_printer_published(Printer_entry *print_hnd, int snum,
3504 struct GUID *guid)
3506 NT_PRINTER_INFO_LEVEL *printer = NULL;
3507 REGVAL_CTR *ctr;
3508 REGISTRY_VALUE *guid_val;
3509 WERROR win_rc;
3510 int i;
3511 bool ret = False;
3513 win_rc = get_a_printer(print_hnd, &printer, 2, lp_servicename(snum));
3515 if (!W_ERROR_IS_OK(win_rc) ||
3516 !(printer->info_2->attributes & PRINTER_ATTRIBUTE_PUBLISHED) ||
3517 ((i = lookup_printerkey(printer->info_2->data, SPOOL_DSSPOOLER_KEY)) < 0) ||
3518 !(ctr = printer->info_2->data->keys[i].values) ||
3519 !(guid_val = regval_ctr_getvalue(ctr, "objectGUID")))
3521 free_a_printer(&printer, 2);
3522 return False;
3525 /* fetching printer guids really ought to be a separate function. */
3527 if ( guid ) {
3528 fstring guid_str;
3530 /* We used to store the guid as REG_BINARY, then swapped
3531 to REG_SZ for Vista compatibility so check for both */
3533 switch ( regval_type(guid_val) ){
3534 case REG_SZ:
3535 rpcstr_pull( guid_str, regval_data_p(guid_val),
3536 sizeof(guid_str)-1, -1, STR_TERMINATE );
3537 ret = smb_string_to_uuid( guid_str, guid );
3538 break;
3539 case REG_BINARY:
3540 if ( regval_size(guid_val) != sizeof(struct GUID) ) {
3541 ret = False;
3542 break;
3544 memcpy(guid, regval_data_p(guid_val), sizeof(struct GUID));
3545 break;
3546 default:
3547 DEBUG(0,("is_printer_published: GUID value stored as "
3548 "invaluid type (%d)\n", regval_type(guid_val) ));
3549 break;
3553 free_a_printer(&printer, 2);
3554 return ret;
3556 #else
3557 WERROR nt_printer_publish(Printer_entry *print_hnd, int snum, int action)
3559 return WERR_OK;
3562 WERROR check_published_printers(void)
3564 return WERR_OK;
3567 bool is_printer_published(Printer_entry *print_hnd, int snum,
3568 struct GUID *guid)
3570 return False;
3572 #endif /* HAVE_ADS */
3574 /****************************************************************************
3575 ***************************************************************************/
3577 WERROR delete_all_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key )
3579 NT_PRINTER_DATA *data;
3580 int i;
3581 int removed_keys = 0;
3582 int empty_slot;
3584 data = p2->data;
3585 empty_slot = data->num_keys;
3587 if ( !key )
3588 return WERR_INVALID_PARAM;
3590 /* remove all keys */
3592 if ( !strlen(key) ) {
3594 TALLOC_FREE( data );
3596 p2->data = NULL;
3598 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from printer [%s]\n",
3599 p2->printername ));
3601 return WERR_OK;
3604 /* remove a specific key (and all subkeys) */
3606 for ( i=0; i<data->num_keys; i++ ) {
3607 if ( StrnCaseCmp( data->keys[i].name, key, strlen(key)) == 0 ) {
3608 DEBUG(8,("delete_all_printer_data: Removed all Printer Data from key [%s]\n",
3609 data->keys[i].name));
3611 TALLOC_FREE( data->keys[i].name );
3612 TALLOC_FREE( data->keys[i].values );
3614 /* mark the slot as empty */
3616 ZERO_STRUCTP( &data->keys[i] );
3620 /* find the first empty slot */
3622 for ( i=0; i<data->num_keys; i++ ) {
3623 if ( !data->keys[i].name ) {
3624 empty_slot = i;
3625 removed_keys++;
3626 break;
3630 if ( i == data->num_keys )
3631 /* nothing was removed */
3632 return WERR_INVALID_PARAM;
3634 /* move everything down */
3636 for ( i=empty_slot+1; i<data->num_keys; i++ ) {
3637 if ( data->keys[i].name ) {
3638 memcpy( &data->keys[empty_slot], &data->keys[i], sizeof(NT_PRINTER_KEY) );
3639 ZERO_STRUCTP( &data->keys[i] );
3640 empty_slot++;
3641 removed_keys++;
3645 /* update count */
3647 data->num_keys -= removed_keys;
3649 /* sanity check to see if anything is left */
3651 if ( !data->num_keys ) {
3652 DEBUG(8,("delete_all_printer_data: No keys left for printer [%s]\n", p2->printername ));
3654 SAFE_FREE( data->keys );
3655 ZERO_STRUCTP( data );
3658 return WERR_OK;
3661 /****************************************************************************
3662 ***************************************************************************/
3664 WERROR delete_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3666 WERROR result = WERR_OK;
3667 int key_index;
3669 /* we must have names on non-zero length */
3671 if ( !key || !*key|| !value || !*value )
3672 return WERR_INVALID_NAME;
3674 /* find the printer key first */
3676 key_index = lookup_printerkey( p2->data, key );
3677 if ( key_index == -1 )
3678 return WERR_OK;
3680 /* make sure the value exists so we can return the correct error code */
3682 if ( !regval_ctr_getvalue( p2->data->keys[key_index].values, value ) )
3683 return WERR_BADFILE;
3685 regval_ctr_delvalue( p2->data->keys[key_index].values, value );
3687 DEBUG(8,("delete_printer_data: Removed key => [%s], value => [%s]\n",
3688 key, value ));
3690 return result;
3693 /****************************************************************************
3694 ***************************************************************************/
3696 WERROR add_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value,
3697 uint32 type, uint8 *data, int real_len )
3699 WERROR result = WERR_OK;
3700 int key_index;
3702 /* we must have names on non-zero length */
3704 if ( !key || !*key|| !value || !*value )
3705 return WERR_INVALID_NAME;
3707 /* find the printer key first */
3709 key_index = lookup_printerkey( p2->data, key );
3710 if ( key_index == -1 )
3711 key_index = add_new_printer_key( p2->data, key );
3713 if ( key_index == -1 )
3714 return WERR_NOMEM;
3716 regval_ctr_addvalue( p2->data->keys[key_index].values, value,
3717 type, (const char *)data, real_len );
3719 DEBUG(8,("add_printer_data: Added key => [%s], value => [%s], type=> [%d], size => [%d]\n",
3720 key, value, type, real_len ));
3722 return result;
3725 /****************************************************************************
3726 ***************************************************************************/
3728 REGISTRY_VALUE* get_printer_data( NT_PRINTER_INFO_LEVEL_2 *p2, const char *key, const char *value )
3730 int key_index;
3732 if ( (key_index = lookup_printerkey( p2->data, key )) == -1 )
3733 return NULL;
3735 DEBUG(8,("get_printer_data: Attempting to lookup key => [%s], value => [%s]\n",
3736 key, value ));
3738 return regval_ctr_getvalue( p2->data->keys[key_index].values, value );
3741 /****************************************************************************
3742 Unpack a list of registry values frem the TDB
3743 ***************************************************************************/
3745 static int unpack_values(NT_PRINTER_DATA *printer_data, const uint8 *buf, int buflen)
3747 int len = 0;
3748 uint32 type;
3749 fstring string;
3750 const char *valuename = NULL;
3751 const char *keyname = NULL;
3752 char *str;
3753 int size;
3754 uint8 *data_p;
3755 REGISTRY_VALUE *regval_p;
3756 int key_index;
3758 /* add the "PrinterDriverData" key first for performance reasons */
3760 add_new_printer_key( printer_data, SPOOL_PRINTERDATA_KEY );
3762 /* loop and unpack the rest of the registry values */
3764 while ( True ) {
3766 /* check to see if there are any more registry values */
3768 regval_p = NULL;
3769 len += tdb_unpack(buf+len, buflen-len, "p", &regval_p);
3770 if ( !regval_p )
3771 break;
3773 /* unpack the next regval */
3775 len += tdb_unpack(buf+len, buflen-len, "fdB",
3776 string,
3777 &type,
3778 &size,
3779 &data_p);
3781 /* lookup for subkey names which have a type of REG_NONE */
3782 /* there's no data with this entry */
3784 if ( type == REG_NONE ) {
3785 if ( (key_index=lookup_printerkey( printer_data, string)) == -1 )
3786 add_new_printer_key( printer_data, string );
3787 continue;
3791 * break of the keyname from the value name.
3792 * Valuenames can have embedded '\'s so be careful.
3793 * only support one level of keys. See the
3794 * "Konica Fiery S300 50C-K v1.1. enu" 2k driver.
3795 * -- jerry
3798 str = strchr_m( string, '\\');
3800 /* Put in "PrinterDriverData" is no key specified */
3802 if ( !str ) {
3803 keyname = SPOOL_PRINTERDATA_KEY;
3804 valuename = string;
3806 else {
3807 *str = '\0';
3808 keyname = string;
3809 valuename = str+1;
3812 /* see if we need a new key */
3814 if ( (key_index=lookup_printerkey( printer_data, keyname )) == -1 )
3815 key_index = add_new_printer_key( printer_data, keyname );
3817 if ( key_index == -1 ) {
3818 DEBUG(0,("unpack_values: Failed to allocate a new key [%s]!\n",
3819 keyname));
3820 break;
3823 DEBUG(8,("specific: [%s:%s], len: %d\n", keyname, valuename, size));
3825 /* Vista doesn't like unknown REG_BINARY values in DsSpooler.
3826 Thanks to Martin Zielinski for the hint. */
3828 if ( type == REG_BINARY &&
3829 strequal( keyname, SPOOL_DSSPOOLER_KEY ) &&
3830 strequal( valuename, "objectGUID" ) )
3832 struct GUID guid;
3833 UNISTR2 unistr_guid;
3835 ZERO_STRUCT( unistr_guid );
3837 /* convert the GUID to a UNICODE string */
3839 memcpy( &guid, data_p, sizeof(struct GUID) );
3841 init_unistr2( &unistr_guid,
3842 smb_uuid_string(talloc_tos(), guid),
3843 UNI_STR_TERMINATE );
3845 regval_ctr_addvalue( printer_data->keys[key_index].values,
3846 valuename, REG_SZ,
3847 (const char *)unistr_guid.buffer,
3848 unistr_guid.uni_str_len*2 );
3850 } else {
3851 /* add the value */
3853 regval_ctr_addvalue( printer_data->keys[key_index].values,
3854 valuename, type, (const char *)data_p,
3855 size );
3858 SAFE_FREE(data_p); /* 'B' option to tdbpack does a malloc() */
3862 return len;
3865 /****************************************************************************
3866 ***************************************************************************/
3868 static char *last_from;
3869 static char *last_to;
3871 static const char *get_last_from(void)
3873 if (!last_from) {
3874 return "";
3876 return last_from;
3879 static const char *get_last_to(void)
3881 if (!last_to) {
3882 return "";
3884 return last_to;
3887 static bool set_last_from_to(const char *from, const char *to)
3889 char *orig_from = last_from;
3890 char *orig_to = last_to;
3892 last_from = SMB_STRDUP(from);
3893 last_to = SMB_STRDUP(to);
3895 SAFE_FREE(orig_from);
3896 SAFE_FREE(orig_to);
3898 if (!last_from || !last_to) {
3899 SAFE_FREE(last_from);
3900 SAFE_FREE(last_to);
3901 return false;
3903 return true;
3906 static void map_to_os2_driver(fstring drivername)
3908 char *mapfile = lp_os2_driver_map();
3909 char **lines = NULL;
3910 int numlines = 0;
3911 int i;
3913 if (!strlen(drivername))
3914 return;
3916 if (!*mapfile)
3917 return;
3919 if (strequal(drivername,get_last_from())) {
3920 DEBUG(3,("Mapped Windows driver %s to OS/2 driver %s\n",
3921 drivername,get_last_to()));
3922 fstrcpy(drivername,get_last_to());
3923 return;
3926 lines = file_lines_load(mapfile, &numlines,0);
3927 if (numlines == 0 || lines == NULL) {
3928 DEBUG(0,("No entries in OS/2 driver map %s\n",mapfile));
3929 SAFE_FREE(lines);
3930 return;
3933 DEBUG(4,("Scanning OS/2 driver map %s\n",mapfile));
3935 for( i = 0; i < numlines; i++) {
3936 char *nt_name = lines[i];
3937 char *os2_name = strchr(nt_name,'=');
3939 if (!os2_name)
3940 continue;
3942 *os2_name++ = 0;
3944 while (isspace(*nt_name))
3945 nt_name++;
3947 if (!*nt_name || strchr("#;",*nt_name))
3948 continue;
3951 int l = strlen(nt_name);
3952 while (l && isspace(nt_name[l-1])) {
3953 nt_name[l-1] = 0;
3954 l--;
3958 while (isspace(*os2_name))
3959 os2_name++;
3962 int l = strlen(os2_name);
3963 while (l && isspace(os2_name[l-1])) {
3964 os2_name[l-1] = 0;
3965 l--;
3969 if (strequal(nt_name,drivername)) {
3970 DEBUG(3,("Mapped windows driver %s to os2 driver%s\n",drivername,os2_name));
3971 set_last_from_to(drivername,os2_name);
3972 fstrcpy(drivername,os2_name);
3973 file_lines_free(lines);
3974 return;
3978 file_lines_free(lines);
3981 /****************************************************************************
3982 Get a default printer info 2 struct.
3983 ****************************************************************************/
3985 static WERROR get_a_printer_2_default(NT_PRINTER_INFO_LEVEL_2 *info,
3986 const char *servername,
3987 const char* sharename,
3988 bool get_loc_com)
3990 int snum = lp_servicenumber(sharename);
3992 slprintf(info->servername, sizeof(info->servername)-1, "\\\\%s", servername);
3993 slprintf(info->printername, sizeof(info->printername)-1, "\\\\%s\\%s",
3994 servername, sharename);
3995 fstrcpy(info->sharename, sharename);
3996 fstrcpy(info->portname, SAMBA_PRINTER_PORT_NAME);
3998 /* by setting the driver name to an empty string, a local NT admin
3999 can now run the **local** APW to install a local printer driver
4000 for a Samba shared printer in 2.2. Without this, drivers **must** be
4001 installed on the Samba server for NT clients --jerry */
4002 #if 0 /* JERRY --do not uncomment-- */
4003 if (!*info->drivername)
4004 fstrcpy(info->drivername, "NO DRIVER AVAILABLE FOR THIS PRINTER");
4005 #endif
4008 DEBUG(10,("get_a_printer_2_default: driver name set to [%s]\n", info->drivername));
4010 strlcpy(info->comment, "", sizeof(info->comment));
4011 fstrcpy(info->printprocessor, "winprint");
4012 fstrcpy(info->datatype, "RAW");
4014 #ifdef HAVE_CUPS
4015 if (get_loc_com && (enum printing_types)lp_printing(snum) == PRINT_CUPS ) {
4016 /* Pull the location and comment strings from cups if we don't
4017 already have one */
4018 if ( !strlen(info->location) || !strlen(info->comment) )
4019 cups_pull_comment_location( info );
4021 #endif
4023 info->attributes = PRINTER_ATTRIBUTE_SAMBA;
4025 info->starttime = 0; /* Minutes since 12:00am GMT */
4026 info->untiltime = 0; /* Minutes since 12:00am GMT */
4027 info->priority = 1;
4028 info->default_priority = 1;
4029 info->setuptime = (uint32)time(NULL);
4032 * I changed this as I think it is better to have a generic
4033 * DEVMODE than to crash Win2k explorer.exe --jerry
4034 * See the HP Deskjet 990c Win2k drivers for an example.
4036 * However the default devmode appears to cause problems
4037 * with the HP CLJ 8500 PCL driver. Hence the addition of
4038 * the "default devmode" parameter --jerry 22/01/2002
4041 if (lp_default_devmode(snum)) {
4042 if ((info->devmode = construct_nt_devicemode(info->printername)) == NULL) {
4043 goto fail;
4045 } else {
4046 info->devmode = NULL;
4049 if (!nt_printing_getsec(info, sharename, &info->secdesc_buf)) {
4050 goto fail;
4053 return WERR_OK;
4055 fail:
4056 if (info->devmode)
4057 free_nt_devicemode(&info->devmode);
4059 return WERR_ACCESS_DENIED;
4062 /****************************************************************************
4063 ****************************************************************************/
4065 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info,
4066 const char *servername,
4067 const char *sharename,
4068 bool get_loc_com)
4070 int len = 0;
4071 int snum = lp_servicenumber(sharename);
4072 TDB_DATA kbuf, dbuf;
4073 fstring printername;
4074 char adevice[MAXDEVICENAME];
4075 char *comment = NULL;
4077 kbuf = make_printer_tdbkey(talloc_tos(), sharename);
4079 dbuf = tdb_fetch(tdb_printers, kbuf);
4080 if (!dbuf.dptr) {
4081 return get_a_printer_2_default(info, servername,
4082 sharename, get_loc_com);
4085 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "dddddddddddfffffPfffff",
4086 &info->attributes,
4087 &info->priority,
4088 &info->default_priority,
4089 &info->starttime,
4090 &info->untiltime,
4091 &info->status,
4092 &info->cjobs,
4093 &info->averageppm,
4094 &info->changeid,
4095 &info->c_setprinter,
4096 &info->setuptime,
4097 info->servername,
4098 info->printername,
4099 info->sharename,
4100 info->portname,
4101 info->drivername,
4102 &comment,
4103 info->location,
4104 info->sepfile,
4105 info->printprocessor,
4106 info->datatype,
4107 info->parameters);
4109 if (comment) {
4110 strlcpy(info->comment, comment, sizeof(info->comment));
4111 SAFE_FREE(comment);
4114 /* Samba has to have shared raw drivers. */
4115 info->attributes |= PRINTER_ATTRIBUTE_SAMBA;
4116 info->attributes &= ~PRINTER_ATTRIBUTE_NOT_SAMBA;
4118 /* Restore the stripped strings. */
4119 slprintf(info->servername, sizeof(info->servername)-1, "\\\\%s", servername);
4121 if ( lp_force_printername(snum) ) {
4122 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, sharename );
4123 } else {
4124 slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", servername, info->printername);
4127 fstrcpy(info->printername, printername);
4129 #ifdef HAVE_CUPS
4130 if (get_loc_com && (enum printing_types)lp_printing(snum) == PRINT_CUPS ) {
4131 /* Pull the location and comment strings from cups if we don't
4132 already have one */
4133 if ( !strlen(info->location) || !strlen(info->comment) )
4134 cups_pull_comment_location( info );
4136 #endif
4138 len += unpack_devicemode(&info->devmode,dbuf.dptr+len, dbuf.dsize-len);
4141 * Some client drivers freak out if there is a NULL devmode
4142 * (probably the driver is not checking before accessing
4143 * the devmode pointer) --jerry
4145 * See comments in get_a_printer_2_default()
4148 if (lp_default_devmode(snum) && !info->devmode) {
4149 DEBUG(8,("get_a_printer_2: Constructing a default device mode for [%s]\n",
4150 printername));
4151 info->devmode = construct_nt_devicemode(printername);
4154 slprintf( adevice, sizeof(adevice), "%s", info->printername );
4155 if (info->devmode) {
4156 fstrcpy(info->devmode->devicename, adevice);
4159 if ( !(info->data = TALLOC_ZERO_P( info, NT_PRINTER_DATA )) ) {
4160 DEBUG(0,("unpack_values: talloc() failed!\n"));
4161 SAFE_FREE(dbuf.dptr);
4162 return WERR_NOMEM;
4164 len += unpack_values( info->data, dbuf.dptr+len, dbuf.dsize-len );
4166 /* This will get the current RPC talloc context, but we should be
4167 passing this as a parameter... fixme... JRA ! */
4169 if (!nt_printing_getsec(info, sharename, &info->secdesc_buf)) {
4170 SAFE_FREE(dbuf.dptr);
4171 return WERR_NOMEM;
4174 /* Fix for OS/2 drivers. */
4176 if (get_remote_arch() == RA_OS2) {
4177 map_to_os2_driver(info->drivername);
4180 SAFE_FREE(dbuf.dptr);
4182 DEBUG(9,("Unpacked printer [%s] name [%s] running driver [%s]\n",
4183 sharename, info->printername, info->drivername));
4185 return WERR_OK;
4188 /****************************************************************************
4189 Debugging function, dump at level 6 the struct in the logs.
4190 ****************************************************************************/
4191 static uint32 dump_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4193 uint32 result;
4194 NT_PRINTER_INFO_LEVEL_2 *info2;
4196 DEBUG(106,("Dumping printer at level [%d]\n", level));
4198 switch (level) {
4199 case 2:
4201 if (printer->info_2 == NULL)
4202 result=5;
4203 else
4205 info2=printer->info_2;
4207 DEBUGADD(106,("attributes:[%d]\n", info2->attributes));
4208 DEBUGADD(106,("priority:[%d]\n", info2->priority));
4209 DEBUGADD(106,("default_priority:[%d]\n", info2->default_priority));
4210 DEBUGADD(106,("starttime:[%d]\n", info2->starttime));
4211 DEBUGADD(106,("untiltime:[%d]\n", info2->untiltime));
4212 DEBUGADD(106,("status:[%d]\n", info2->status));
4213 DEBUGADD(106,("cjobs:[%d]\n", info2->cjobs));
4214 DEBUGADD(106,("averageppm:[%d]\n", info2->averageppm));
4215 DEBUGADD(106,("changeid:[%d]\n", info2->changeid));
4216 DEBUGADD(106,("c_setprinter:[%d]\n", info2->c_setprinter));
4217 DEBUGADD(106,("setuptime:[%d]\n", info2->setuptime));
4219 DEBUGADD(106,("servername:[%s]\n", info2->servername));
4220 DEBUGADD(106,("printername:[%s]\n", info2->printername));
4221 DEBUGADD(106,("sharename:[%s]\n", info2->sharename));
4222 DEBUGADD(106,("portname:[%s]\n", info2->portname));
4223 DEBUGADD(106,("drivername:[%s]\n", info2->drivername));
4224 DEBUGADD(106,("comment:[%s]\n", info2->comment));
4225 DEBUGADD(106,("location:[%s]\n", info2->location));
4226 DEBUGADD(106,("sepfile:[%s]\n", info2->sepfile));
4227 DEBUGADD(106,("printprocessor:[%s]\n", info2->printprocessor));
4228 DEBUGADD(106,("datatype:[%s]\n", info2->datatype));
4229 DEBUGADD(106,("parameters:[%s]\n", info2->parameters));
4230 result=0;
4232 break;
4234 default:
4235 DEBUGADD(106,("dump_a_printer: Level %u not implemented\n", (unsigned int)level ));
4236 result=1;
4237 break;
4240 return result;
4243 /****************************************************************************
4244 Update the changeid time.
4245 This is SO NASTY as some drivers need this to change, others need it
4246 static. This value will change every second, and I must hope that this
4247 is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
4248 UTAH ! JRA.
4249 ****************************************************************************/
4251 static uint32 rev_changeid(void)
4253 struct timeval tv;
4255 get_process_uptime(&tv);
4257 #if 1 /* JERRY */
4258 /* Return changeid as msec since spooler restart */
4259 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
4260 #else
4262 * This setting seems to work well but is too untested
4263 * to replace the above calculation. Left in for experiementation
4264 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
4266 return tv.tv_sec * 10 + tv.tv_usec / 100000;
4267 #endif
4272 * The function below are the high level ones.
4273 * only those ones must be called from the spoolss code.
4274 * JFM.
4277 /****************************************************************************
4278 Modify a printer. This is called from SETPRINTERDATA/DELETEPRINTERDATA.
4279 ****************************************************************************/
4281 WERROR mod_a_printer(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4283 WERROR result;
4285 dump_a_printer(printer, level);
4287 switch (level) {
4288 case 2:
4291 * Update the changestamp. Emperical tests show that the
4292 * ChangeID is always updated,but c_setprinter is
4293 * global spooler variable (not per printer).
4296 /* ChangeID **must** be increasing over the lifetime
4297 of client's spoolss service in order for the
4298 client's cache to show updates */
4300 printer->info_2->changeid = rev_changeid();
4303 * Because one day someone will ask:
4304 * NT->NT An admin connection to a remote
4305 * printer show changes imeediately in
4306 * the properities dialog
4308 * A non-admin connection will only show the
4309 * changes after viewing the properites page
4310 * 2 times. Seems to be related to a
4311 * race condition in the client between the spooler
4312 * updating the local cache and the Explorer.exe GUI
4313 * actually displaying the properties.
4315 * This is fixed in Win2k. admin/non-admin
4316 * connections both display changes immediately.
4318 * 14/12/01 --jerry
4321 result=update_a_printer_2(printer->info_2);
4322 break;
4324 default:
4325 result=WERR_UNKNOWN_LEVEL;
4326 break;
4329 return result;
4332 /****************************************************************************
4333 Initialize printer devmode & data with previously saved driver init values.
4334 ****************************************************************************/
4336 static bool set_driver_init_2( NT_PRINTER_INFO_LEVEL_2 *info_ptr )
4338 int len = 0;
4339 char *key = NULL;
4340 TDB_DATA dbuf;
4341 NT_PRINTER_INFO_LEVEL_2 info;
4344 ZERO_STRUCT(info);
4347 * Delete any printer data 'values' already set. When called for driver
4348 * replace, there will generally be some, but during an add printer, there
4349 * should not be any (if there are delete them).
4352 if ( info_ptr->data )
4353 delete_all_printer_data( info_ptr, "" );
4355 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX,
4356 info_ptr->drivername) < 0) {
4357 return false;
4360 dbuf = tdb_fetch_bystring(tdb_drivers, key);
4361 if (!dbuf.dptr) {
4363 * When changing to a driver that has no init info in the tdb, remove
4364 * the previous drivers init info and leave the new on blank.
4366 free_nt_devicemode(&info_ptr->devmode);
4367 SAFE_FREE(key);
4368 return false;
4371 SAFE_FREE(key);
4373 * Get the saved DEVMODE..
4376 len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
4379 * The saved DEVMODE contains the devicename from the printer used during
4380 * the initialization save. Change it to reflect the new printer.
4383 if ( info.devmode ) {
4384 ZERO_STRUCT(info.devmode->devicename);
4385 fstrcpy(info.devmode->devicename, info_ptr->printername);
4389 * NT/2k does not change out the entire DeviceMode of a printer
4390 * when changing the driver. Only the driverextra, private, &
4391 * driverversion fields. --jerry (Thu Mar 14 08:58:43 CST 2002)
4393 * Later examination revealed that Windows NT/2k does reset the
4394 * the printer's device mode, bit **only** when you change a
4395 * property of the device mode such as the page orientation.
4396 * --jerry
4400 /* Bind the saved DEVMODE to the new the printer */
4402 free_nt_devicemode(&info_ptr->devmode);
4403 info_ptr->devmode = info.devmode;
4405 DEBUG(10,("set_driver_init_2: Set printer [%s] init %s DEVMODE for driver [%s]\n",
4406 info_ptr->printername, info_ptr->devmode?"VALID":"NULL", info_ptr->drivername));
4408 /* Add the printer data 'values' to the new printer */
4410 if ( !(info_ptr->data = TALLOC_ZERO_P( info_ptr, NT_PRINTER_DATA )) ) {
4411 DEBUG(0,("set_driver_init_2: talloc() failed!\n"));
4412 return False;
4415 len += unpack_values( info_ptr->data, dbuf.dptr+len, dbuf.dsize-len );
4417 SAFE_FREE(dbuf.dptr);
4419 return true;
4422 /****************************************************************************
4423 Initialize printer devmode & data with previously saved driver init values.
4424 When a printer is created using AddPrinter, the drivername bound to the
4425 printer is used to lookup previously saved driver initialization info, which
4426 is bound to the new printer.
4427 ****************************************************************************/
4429 bool set_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4431 bool result = False;
4433 switch (level) {
4434 case 2:
4435 result = set_driver_init_2(printer->info_2);
4436 break;
4438 default:
4439 DEBUG(0,("set_driver_init: Programmer's error! Unknown driver_init level [%d]\n",
4440 level));
4441 break;
4444 return result;
4447 /****************************************************************************
4448 Delete driver init data stored for a specified driver
4449 ****************************************************************************/
4451 bool del_driver_init(char *drivername)
4453 char *key;
4454 bool ret;
4456 if (!drivername || !*drivername) {
4457 DEBUG(3,("del_driver_init: No drivername specified!\n"));
4458 return false;
4461 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX, drivername) < 0) {
4462 return false;
4465 DEBUG(6,("del_driver_init: Removing driver init data for [%s]\n",
4466 drivername));
4468 ret = (tdb_delete_bystring(tdb_drivers, key) == 0);
4469 SAFE_FREE(key);
4470 return ret;
4473 /****************************************************************************
4474 Pack up the DEVMODE and values for a printer into a 'driver init' entry
4475 in the tdb. Note: this is different from the driver entry and the printer
4476 entry. There should be a single driver init entry for each driver regardless
4477 of whether it was installed from NT or 2K. Technically, they should be
4478 different, but they work out to the same struct.
4479 ****************************************************************************/
4481 static uint32 update_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info)
4483 char *key = NULL;
4484 uint8 *buf;
4485 int buflen, len, ret;
4486 int retlen;
4487 TDB_DATA dbuf;
4489 buf = NULL;
4490 buflen = 0;
4492 again:
4493 len = 0;
4494 len += pack_devicemode(info->devmode, buf+len, buflen-len);
4496 retlen = pack_values( info->data, buf+len, buflen-len );
4497 if (retlen == -1) {
4498 ret = -1;
4499 goto done;
4501 len += retlen;
4503 if (buflen < len) {
4504 buf = (uint8 *)SMB_REALLOC(buf, len);
4505 if (!buf) {
4506 DEBUG(0, ("update_driver_init_2: failed to enlarge buffer!\n"));
4507 ret = -1;
4508 goto done;
4510 buflen = len;
4511 goto again;
4514 SAFE_FREE(key);
4515 if (asprintf(&key, "%s%s", DRIVER_INIT_PREFIX, info->drivername) < 0) {
4516 ret = (uint32)-1;
4517 goto done;
4520 dbuf.dptr = buf;
4521 dbuf.dsize = len;
4523 ret = tdb_store_bystring(tdb_drivers, key, dbuf, TDB_REPLACE);
4525 done:
4526 if (ret == -1)
4527 DEBUG(8, ("update_driver_init_2: error updating printer init to tdb on disk\n"));
4529 SAFE_FREE(buf);
4531 DEBUG(10,("update_driver_init_2: Saved printer [%s] init DEVMODE & values for driver [%s]\n",
4532 info->sharename, info->drivername));
4534 return ret;
4537 /****************************************************************************
4538 Update (i.e. save) the driver init info (DEVMODE and values) for a printer
4539 ****************************************************************************/
4541 static uint32 update_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
4543 uint32 result;
4545 dump_a_printer(printer, level);
4547 switch (level) {
4548 case 2:
4549 result = update_driver_init_2(printer->info_2);
4550 break;
4551 default:
4552 result = 1;
4553 break;
4556 return result;
4559 /****************************************************************************
4560 Convert the printer data value, a REG_BINARY array, into an initialization
4561 DEVMODE. Note: the array must be parsed as if it was a DEVMODE in an rpc...
4562 got to keep the endians happy :).
4563 ****************************************************************************/
4565 static bool convert_driver_init( TALLOC_CTX *ctx, NT_DEVICEMODE *nt_devmode, uint8 *data, uint32 data_len )
4567 bool result = False;
4568 prs_struct ps;
4569 DEVICEMODE devmode;
4571 ZERO_STRUCT(devmode);
4573 prs_init_empty(&ps, ctx, UNMARSHALL);
4574 ps.data_p = (char *)data;
4575 ps.buffer_size = data_len;
4577 if (spoolss_io_devmode("phantom DEVMODE", &ps, 0, &devmode))
4578 result = convert_devicemode("", &devmode, &nt_devmode);
4579 else
4580 DEBUG(10,("convert_driver_init: error parsing DEVMODE\n"));
4582 return result;
4585 /****************************************************************************
4586 Set the DRIVER_INIT info in the tdb. Requires Win32 client code that:
4588 1. Use the driver's config DLL to this UNC printername and:
4589 a. Call DrvPrintEvent with PRINTER_EVENT_INITIALIZE
4590 b. Call DrvConvertDevMode with CDM_DRIVER_DEFAULT to get default DEVMODE
4591 2. Call SetPrinterData with the 'magic' key and the DEVMODE as data.
4593 The last step triggers saving the "driver initialization" information for
4594 this printer into the tdb. Later, new printers that use this driver will
4595 have this initialization information bound to them. This simulates the
4596 driver initialization, as if it had run on the Samba server (as it would
4597 have done on NT).
4599 The Win32 client side code requirement sucks! But until we can run arbitrary
4600 Win32 printer driver code on any Unix that Samba runs on, we are stuck with it.
4602 It would have been easier to use SetPrinter because all the UNMARSHALLING of
4603 the DEVMODE is done there, but 2K/XP clients do not set the DEVMODE... think
4604 about it and you will realize why. JRR 010720
4605 ****************************************************************************/
4607 static WERROR save_driver_init_2(NT_PRINTER_INFO_LEVEL *printer, uint8 *data, uint32 data_len )
4609 WERROR status = WERR_OK;
4610 TALLOC_CTX *ctx = NULL;
4611 NT_DEVICEMODE *nt_devmode = NULL;
4612 NT_DEVICEMODE *tmp_devmode = printer->info_2->devmode;
4615 * When the DEVMODE is already set on the printer, don't try to unpack it.
4617 DEBUG(8,("save_driver_init_2: Enter...\n"));
4619 if ( !printer->info_2->devmode && data_len ) {
4621 * Set devmode on printer info, so entire printer initialization can be
4622 * saved to tdb.
4625 if ((ctx = talloc_init("save_driver_init_2")) == NULL)
4626 return WERR_NOMEM;
4628 if ((nt_devmode = SMB_MALLOC_P(NT_DEVICEMODE)) == NULL) {
4629 status = WERR_NOMEM;
4630 goto done;
4633 ZERO_STRUCTP(nt_devmode);
4636 * The DEVMODE is held in the 'data' component of the param in raw binary.
4637 * Convert it to to a devmode structure
4639 if ( !convert_driver_init( ctx, nt_devmode, data, data_len )) {
4640 DEBUG(10,("save_driver_init_2: error converting DEVMODE\n"));
4641 status = WERR_INVALID_PARAM;
4642 goto done;
4645 printer->info_2->devmode = nt_devmode;
4649 * Pack up and add (or update) the DEVMODE and any current printer data to
4650 * a 'driver init' element in the tdb
4654 if ( update_driver_init(printer, 2) != 0 ) {
4655 DEBUG(10,("save_driver_init_2: error updating DEVMODE\n"));
4656 status = WERR_NOMEM;
4657 goto done;
4661 * If driver initialization info was successfully saved, set the current
4662 * printer to match it. This allows initialization of the current printer
4663 * as well as the driver.
4665 status = mod_a_printer(printer, 2);
4666 if (!W_ERROR_IS_OK(status)) {
4667 DEBUG(10,("save_driver_init_2: error setting DEVMODE on printer [%s]\n",
4668 printer->info_2->printername));
4671 done:
4672 talloc_destroy(ctx);
4673 free_nt_devicemode( &nt_devmode );
4675 printer->info_2->devmode = tmp_devmode;
4677 return status;
4680 /****************************************************************************
4681 Update the driver init info (DEVMODE and specifics) for a printer
4682 ****************************************************************************/
4684 WERROR save_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level, uint8 *data, uint32 data_len)
4686 WERROR status = WERR_OK;
4688 switch (level) {
4689 case 2:
4690 status = save_driver_init_2( printer, data, data_len );
4691 break;
4692 default:
4693 status = WERR_UNKNOWN_LEVEL;
4694 break;
4697 return status;
4700 /****************************************************************************
4701 Get a NT_PRINTER_INFO_LEVEL struct. It returns malloced memory.
4703 Previously the code had a memory allocation problem because it always
4704 used the TALLOC_CTX from the Printer_entry*. This context lasts
4705 as a long as the original handle is open. So if the client made a lot
4706 of getprinter[data]() calls, the memory usage would climb. Now we use
4707 a short lived TALLOC_CTX for printer_info_2 objects returned. We
4708 still use the Printer_entry->ctx for maintaining the cache copy though
4709 since that object must live as long as the handle by definition.
4710 --jerry
4712 ****************************************************************************/
4714 static WERROR get_a_printer_internal( Printer_entry *print_hnd, NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level,
4715 const char *sharename, bool get_loc_com)
4717 WERROR result;
4718 fstring servername;
4720 DEBUG(10,("get_a_printer: [%s] level %u\n", sharename, (unsigned int)level));
4722 if ( !(*pp_printer = TALLOC_ZERO_P(NULL, NT_PRINTER_INFO_LEVEL)) ) {
4723 DEBUG(0,("get_a_printer: talloc() fail.\n"));
4724 return WERR_NOMEM;
4727 switch (level) {
4728 case 2:
4729 if ( !((*pp_printer)->info_2 = TALLOC_ZERO_P(*pp_printer, NT_PRINTER_INFO_LEVEL_2)) ) {
4730 DEBUG(0,("get_a_printer: talloc() fail.\n"));
4731 TALLOC_FREE( *pp_printer );
4732 return WERR_NOMEM;
4735 if ( print_hnd )
4736 fstrcpy( servername, print_hnd->servername );
4737 else {
4738 fstrcpy( servername, "%L" );
4739 standard_sub_basic( "", "", servername,
4740 sizeof(servername)-1 );
4743 result = get_a_printer_2( (*pp_printer)->info_2,
4744 servername, sharename, get_loc_com);
4746 /* we have a new printer now. Save it with this handle */
4748 if ( !W_ERROR_IS_OK(result) ) {
4749 TALLOC_FREE( *pp_printer );
4750 DEBUG(10,("get_a_printer: [%s] level %u returning %s\n",
4751 sharename, (unsigned int)level, dos_errstr(result)));
4752 return result;
4755 dump_a_printer( *pp_printer, level);
4757 break;
4759 default:
4760 TALLOC_FREE( *pp_printer );
4761 return WERR_UNKNOWN_LEVEL;
4764 return WERR_OK;
4767 WERROR get_a_printer( Printer_entry *print_hnd,
4768 NT_PRINTER_INFO_LEVEL **pp_printer,
4769 uint32 level,
4770 const char *sharename)
4772 return get_a_printer_internal(print_hnd, pp_printer, level,
4773 sharename, true);
4776 WERROR get_a_printer_search( Printer_entry *print_hnd,
4777 NT_PRINTER_INFO_LEVEL **pp_printer,
4778 uint32 level,
4779 const char *sharename)
4781 return get_a_printer_internal(print_hnd, pp_printer, level,
4782 sharename, false);
4785 /****************************************************************************
4786 Deletes a NT_PRINTER_INFO_LEVEL struct.
4787 ****************************************************************************/
4789 uint32 free_a_printer(NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level)
4791 NT_PRINTER_INFO_LEVEL *printer = *pp_printer;
4793 if ( !printer )
4794 return 0;
4796 switch (level) {
4797 case 2:
4798 if ( printer->info_2 )
4799 free_nt_printer_info_level_2(&printer->info_2);
4800 break;
4802 default:
4803 DEBUG(0,("free_a_printer: unknown level! [%d]\n", level ));
4804 return 1;
4807 TALLOC_FREE(*pp_printer);
4809 return 0;
4812 /****************************************************************************
4813 ****************************************************************************/
4814 uint32 add_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4816 uint32 result;
4817 DEBUG(104,("adding a printer at level [%d]\n", level));
4818 dump_a_printer_driver(driver, level);
4820 switch (level) {
4821 case 3:
4822 result=add_a_printer_driver_3(driver.info_3);
4823 break;
4825 case 6:
4826 result=add_a_printer_driver_6(driver.info_6);
4827 break;
4829 default:
4830 result=1;
4831 break;
4834 return result;
4836 /****************************************************************************
4837 ****************************************************************************/
4839 WERROR get_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL *driver, uint32 level,
4840 fstring drivername, const char *architecture, uint32 version)
4842 WERROR result;
4844 switch (level) {
4845 case 3:
4846 /* Sometime we just want any version of the driver */
4848 if ( version == DRIVER_ANY_VERSION ) {
4849 /* look for Win2k first and then for NT4 */
4850 result = get_a_printer_driver_3(&driver->info_3, drivername,
4851 architecture, 3);
4853 if ( !W_ERROR_IS_OK(result) ) {
4854 result = get_a_printer_driver_3( &driver->info_3,
4855 drivername, architecture, 2 );
4857 } else {
4858 result = get_a_printer_driver_3(&driver->info_3, drivername,
4859 architecture, version);
4861 break;
4863 default:
4864 result=W_ERROR(1);
4865 break;
4868 if (W_ERROR_IS_OK(result))
4869 dump_a_printer_driver(*driver, level);
4871 return result;
4874 /****************************************************************************
4875 ****************************************************************************/
4876 uint32 free_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
4878 uint32 result;
4880 switch (level) {
4881 case 3:
4883 NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
4884 if (driver.info_3 != NULL)
4886 info3=driver.info_3;
4887 SAFE_FREE(info3->dependentfiles);
4888 ZERO_STRUCTP(info3);
4889 SAFE_FREE(info3);
4890 result=0;
4891 } else {
4892 result=4;
4894 break;
4896 case 6:
4898 NT_PRINTER_DRIVER_INFO_LEVEL_6 *info6;
4899 if (driver.info_6 != NULL) {
4900 info6=driver.info_6;
4901 SAFE_FREE(info6->dependentfiles);
4902 SAFE_FREE(info6->previousnames);
4903 ZERO_STRUCTP(info6);
4904 SAFE_FREE(info6);
4905 result=0;
4906 } else {
4907 result=4;
4909 break;
4911 default:
4912 result=1;
4913 break;
4915 return result;
4919 /****************************************************************************
4920 Determine whether or not a particular driver is currently assigned
4921 to a printer
4922 ****************************************************************************/
4924 bool printer_driver_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3 )
4926 int snum;
4927 int n_services = lp_numservices();
4928 NT_PRINTER_INFO_LEVEL *printer = NULL;
4929 bool in_use = False;
4931 if ( !info_3 )
4932 return False;
4934 DEBUG(10,("printer_driver_in_use: Beginning search through ntprinters.tdb...\n"));
4936 /* loop through the printers.tdb and check for the drivername */
4938 for (snum=0; snum<n_services && !in_use; snum++) {
4939 if ( !(lp_snum_ok(snum) && lp_print_ok(snum) ) )
4940 continue;
4942 if ( !W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, lp_servicename(snum))) )
4943 continue;
4945 if ( strequal(info_3->name, printer->info_2->drivername) )
4946 in_use = True;
4948 free_a_printer( &printer, 2 );
4951 DEBUG(10,("printer_driver_in_use: Completed search through ntprinters.tdb...\n"));
4953 if ( in_use ) {
4954 NT_PRINTER_DRIVER_INFO_LEVEL d;
4955 WERROR werr;
4957 DEBUG(5,("printer_driver_in_use: driver \"%s\" is currently in use\n", info_3->name));
4959 /* we can still remove the driver if there is one of
4960 "Windows NT x86" version 2 or 3 left */
4962 if ( !strequal( "Windows NT x86", info_3->environment ) ) {
4963 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", DRIVER_ANY_VERSION );
4965 else {
4966 switch ( info_3->cversion ) {
4967 case 2:
4968 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", 3 );
4969 break;
4970 case 3:
4971 werr = get_a_printer_driver( &d, 3, info_3->name, "Windows NT x86", 2 );
4972 break;
4973 default:
4974 DEBUG(0,("printer_driver_in_use: ERROR! unknown driver version (%d)\n",
4975 info_3->cversion));
4976 werr = WERR_UNKNOWN_PRINTER_DRIVER;
4977 break;
4981 /* now check the error code */
4983 if ( W_ERROR_IS_OK(werr) ) {
4984 /* it's ok to remove the driver, we have other architctures left */
4985 in_use = False;
4986 free_a_printer_driver( d, 3 );
4990 /* report that the driver is not in use by default */
4992 return in_use;
4996 /**********************************************************************
4997 Check to see if a ogiven file is in use by *info
4998 *********************************************************************/
5000 static bool drv_file_in_use( char* file, NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
5002 int i = 0;
5004 if ( !info )
5005 return False;
5007 /* mz: skip files that are in the list but already deleted */
5008 if (!file || !file[0]) {
5009 return false;
5012 if ( strequal(file, info->driverpath) )
5013 return True;
5015 if ( strequal(file, info->datafile) )
5016 return True;
5018 if ( strequal(file, info->configfile) )
5019 return True;
5021 if ( strequal(file, info->helpfile) )
5022 return True;
5024 /* see of there are any dependent files to examine */
5026 if ( !info->dependentfiles )
5027 return False;
5029 while ( *info->dependentfiles[i] ) {
5030 if ( strequal(file, info->dependentfiles[i]) )
5031 return True;
5032 i++;
5035 return False;
5039 /**********************************************************************
5040 Utility function to remove the dependent file pointed to by the
5041 input parameter from the list
5042 *********************************************************************/
5044 static void trim_dependent_file( fstring files[], int idx )
5047 /* bump everything down a slot */
5049 while( *files[idx+1] ) {
5050 fstrcpy( files[idx], files[idx+1] );
5051 idx++;
5054 *files[idx] = '\0';
5056 return;
5059 /**********************************************************************
5060 Check if any of the files used by src are also used by drv
5061 *********************************************************************/
5063 static bool trim_overlap_drv_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *src,
5064 NT_PRINTER_DRIVER_INFO_LEVEL_3 *drv )
5066 bool in_use = False;
5067 int i = 0;
5069 if ( !src || !drv )
5070 return False;
5072 /* check each file. Remove it from the src structure if it overlaps */
5074 if ( drv_file_in_use(src->driverpath, drv) ) {
5075 in_use = True;
5076 DEBUG(10,("Removing driverfile [%s] from list\n", src->driverpath));
5077 fstrcpy( src->driverpath, "" );
5080 if ( drv_file_in_use(src->datafile, drv) ) {
5081 in_use = True;
5082 DEBUG(10,("Removing datafile [%s] from list\n", src->datafile));
5083 fstrcpy( src->datafile, "" );
5086 if ( drv_file_in_use(src->configfile, drv) ) {
5087 in_use = True;
5088 DEBUG(10,("Removing configfile [%s] from list\n", src->configfile));
5089 fstrcpy( src->configfile, "" );
5092 if ( drv_file_in_use(src->helpfile, drv) ) {
5093 in_use = True;
5094 DEBUG(10,("Removing helpfile [%s] from list\n", src->helpfile));
5095 fstrcpy( src->helpfile, "" );
5098 /* are there any dependentfiles to examine? */
5100 if ( !src->dependentfiles )
5101 return in_use;
5103 while ( *src->dependentfiles[i] ) {
5104 if ( drv_file_in_use(src->dependentfiles[i], drv) ) {
5105 in_use = True;
5106 DEBUG(10,("Removing [%s] from dependent file list\n", src->dependentfiles[i]));
5107 trim_dependent_file( src->dependentfiles, i );
5108 } else
5109 i++;
5112 return in_use;
5115 /****************************************************************************
5116 Determine whether or not a particular driver files are currently being
5117 used by any other driver.
5119 Return value is True if any files were in use by other drivers
5120 and False otherwise.
5122 Upon return, *info has been modified to only contain the driver files
5123 which are not in use
5125 Fix from mz:
5127 This needs to check all drivers to ensure that all files in use
5128 have been removed from *info, not just the ones in the first
5129 match.
5130 ****************************************************************************/
5132 bool printer_driver_files_in_use ( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info )
5134 int i;
5135 int ndrivers;
5136 uint32 version;
5137 fstring *list = NULL;
5138 NT_PRINTER_DRIVER_INFO_LEVEL driver;
5139 bool in_use = false;
5141 if ( !info )
5142 return False;
5144 version = info->cversion;
5146 /* loop over all driver versions */
5148 DEBUG(5,("printer_driver_files_in_use: Beginning search through ntdrivers.tdb...\n"));
5150 /* get the list of drivers */
5152 list = NULL;
5153 ndrivers = get_ntdrivers(&list, info->environment, version);
5155 DEBUGADD(4,("we have:[%d] drivers in environment [%s] and version [%d]\n",
5156 ndrivers, info->environment, version));
5158 /* check each driver for overlap in files */
5160 for (i=0; i<ndrivers; i++) {
5161 DEBUGADD(5,("\tdriver: [%s]\n", list[i]));
5163 ZERO_STRUCT(driver);
5165 if ( !W_ERROR_IS_OK(get_a_printer_driver(&driver, 3, list[i], info->environment, version)) ) {
5166 SAFE_FREE(list);
5167 return True;
5170 /* check if d2 uses any files from d1 */
5171 /* only if this is a different driver than the one being deleted */
5173 if ( !strequal(info->name, driver.info_3->name) ) {
5174 if ( trim_overlap_drv_files(info, driver.info_3) ) {
5175 /* mz: Do not instantly return -
5176 * we need to ensure this file isn't
5177 * also in use by other drivers. */
5178 in_use = true;
5182 free_a_printer_driver(driver, 3);
5185 SAFE_FREE(list);
5187 DEBUG(5,("printer_driver_files_in_use: Completed search through ntdrivers.tdb...\n"));
5189 driver.info_3 = info;
5191 if ( DEBUGLEVEL >= 20 )
5192 dump_a_printer_driver( driver, 3 );
5194 return in_use;
5197 /****************************************************************************
5198 Actually delete the driver files. Make sure that
5199 printer_driver_files_in_use() return False before calling
5200 this.
5201 ****************************************************************************/
5203 static bool delete_driver_files( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user )
5205 int i = 0;
5206 char *s;
5207 const char *file;
5208 connection_struct *conn;
5209 DATA_BLOB null_pw;
5210 NTSTATUS nt_status;
5211 fstring res_type;
5212 SMB_STRUCT_STAT st;
5214 if ( !info_3 )
5215 return False;
5217 DEBUG(6,("delete_driver_files: deleting driver [%s] - version [%d]\n", info_3->name, info_3->cversion));
5220 * Connect to the print$ share under the same account as the
5221 * user connected to the rpc pipe. Note we must be root to
5222 * do this.
5225 null_pw = data_blob_null;
5226 fstrcpy(res_type, "A:");
5227 become_root();
5228 conn = make_connection_with_chdir( "print$", null_pw, res_type, user->vuid, &nt_status );
5229 unbecome_root();
5231 if ( !conn ) {
5232 DEBUG(0,("delete_driver_files: Unable to connect\n"));
5233 return False;
5236 if ( !CAN_WRITE(conn) ) {
5237 DEBUG(3,("delete_driver_files: Cannot delete print driver when [print$] is read-only\n"));
5238 return False;
5241 /* Save who we are - we are temporarily becoming the connection user. */
5243 if ( !become_user(conn, conn->vuid) ) {
5244 DEBUG(0,("delete_driver_files: Can't become user!\n"));
5245 return False;
5248 /* now delete the files; must strip the '\print$' string from
5249 fron of path */
5251 if ( *info_3->driverpath ) {
5252 if ( (s = strchr( &info_3->driverpath[1], '\\' )) != NULL ) {
5253 file = s;
5254 driver_unix_convert(conn,file,&st);
5255 DEBUG(10,("deleting driverfile [%s]\n", s));
5256 unlink_internals(conn, NULL, 0, file, False);
5260 if ( *info_3->configfile ) {
5261 if ( (s = strchr( &info_3->configfile[1], '\\' )) != NULL ) {
5262 file = s;
5263 driver_unix_convert(conn,file,&st);
5264 DEBUG(10,("deleting configfile [%s]\n", s));
5265 unlink_internals(conn, NULL, 0, file, False);
5269 if ( *info_3->datafile ) {
5270 if ( (s = strchr( &info_3->datafile[1], '\\' )) != NULL ) {
5271 file = s;
5272 driver_unix_convert(conn,file,&st);
5273 DEBUG(10,("deleting datafile [%s]\n", s));
5274 unlink_internals(conn, NULL, 0, file, False);
5278 if ( *info_3->helpfile ) {
5279 if ( (s = strchr( &info_3->helpfile[1], '\\' )) != NULL ) {
5280 file = s;
5281 driver_unix_convert(conn,file,&st);
5282 DEBUG(10,("deleting helpfile [%s]\n", s));
5283 unlink_internals(conn, NULL, 0, file, False);
5287 /* check if we are done removing files */
5289 if ( info_3->dependentfiles ) {
5290 while ( info_3->dependentfiles[i][0] ) {
5291 char *p;
5293 /* bypass the "\print$" portion of the path */
5295 if ( (p = strchr( info_3->dependentfiles[i]+1, '\\' )) != NULL ) {
5296 file = p;
5297 driver_unix_convert(conn,file,&st);
5298 DEBUG(10,("deleting dependent file [%s]\n", file));
5299 unlink_internals(conn, NULL, 0, file, False);
5302 i++;
5306 unbecome_user();
5308 return true;
5311 /****************************************************************************
5312 Remove a printer driver from the TDB. This assumes that the the driver was
5313 previously looked up.
5314 ***************************************************************************/
5316 WERROR delete_printer_driver( NT_PRINTER_DRIVER_INFO_LEVEL_3 *info_3, struct current_user *user,
5317 uint32 version, bool delete_files )
5319 char *key = NULL;
5320 const char *arch;
5321 TDB_DATA dbuf;
5322 NT_PRINTER_DRIVER_INFO_LEVEL ctr;
5324 /* delete the tdb data first */
5326 arch = get_short_archi(info_3->environment);
5327 if (!arch) {
5328 return WERR_UNKNOWN_PRINTER_DRIVER;
5330 if (asprintf(&key, "%s%s/%d/%s", DRIVERS_PREFIX,
5331 arch, version, info_3->name) < 0) {
5332 return WERR_NOMEM;
5335 DEBUG(5,("delete_printer_driver: key = [%s] delete_files = %s\n",
5336 key, delete_files ? "TRUE" : "FALSE" ));
5338 ctr.info_3 = info_3;
5339 dump_a_printer_driver( ctr, 3 );
5341 /* check if the driver actually exists for this environment */
5343 dbuf = tdb_fetch_bystring( tdb_drivers, key );
5344 if ( !dbuf.dptr ) {
5345 DEBUG(8,("delete_printer_driver: Driver unknown [%s]\n", key));
5346 SAFE_FREE(key);
5347 return WERR_UNKNOWN_PRINTER_DRIVER;
5350 SAFE_FREE( dbuf.dptr );
5352 /* ok... the driver exists so the delete should return success */
5354 if (tdb_delete_bystring(tdb_drivers, key) == -1) {
5355 DEBUG (0,("delete_printer_driver: fail to delete %s!\n", key));
5356 SAFE_FREE(key);
5357 return WERR_ACCESS_DENIED;
5361 * now delete any associated files if delete_files == True
5362 * even if this part failes, we return succes because the
5363 * driver doesn not exist any more
5366 if ( delete_files )
5367 delete_driver_files( info_3, user );
5369 DEBUG(5,("delete_printer_driver: driver delete successful [%s]\n", key));
5370 SAFE_FREE(key);
5372 return WERR_OK;
5375 /****************************************************************************
5376 Store a security desc for a printer.
5377 ****************************************************************************/
5379 WERROR nt_printing_setsec(const char *sharename, SEC_DESC_BUF *secdesc_ctr)
5381 SEC_DESC_BUF *new_secdesc_ctr = NULL;
5382 SEC_DESC_BUF *old_secdesc_ctr = NULL;
5383 prs_struct ps;
5384 bool prs_init_done = false;
5385 TALLOC_CTX *mem_ctx = NULL;
5386 TDB_DATA kbuf;
5387 WERROR status;
5389 mem_ctx = talloc_init("nt_printing_setsec");
5390 if (mem_ctx == NULL)
5391 return WERR_NOMEM;
5393 /* The old owner and group sids of the security descriptor are not
5394 present when new ACEs are added or removed by changing printer
5395 permissions through NT. If they are NULL in the new security
5396 descriptor then copy them over from the old one. */
5398 if (!secdesc_ctr->sd->owner_sid || !secdesc_ctr->sd->group_sid) {
5399 DOM_SID *owner_sid, *group_sid;
5400 SEC_ACL *dacl, *sacl;
5401 SEC_DESC *psd = NULL;
5402 size_t size;
5404 if (!nt_printing_getsec(mem_ctx, sharename, &old_secdesc_ctr)) {
5405 status = WERR_NOMEM;
5406 goto out;
5409 /* Pick out correct owner and group sids */
5411 owner_sid = secdesc_ctr->sd->owner_sid ?
5412 secdesc_ctr->sd->owner_sid :
5413 old_secdesc_ctr->sd->owner_sid;
5415 group_sid = secdesc_ctr->sd->group_sid ?
5416 secdesc_ctr->sd->group_sid :
5417 old_secdesc_ctr->sd->group_sid;
5419 dacl = secdesc_ctr->sd->dacl ?
5420 secdesc_ctr->sd->dacl :
5421 old_secdesc_ctr->sd->dacl;
5423 sacl = secdesc_ctr->sd->sacl ?
5424 secdesc_ctr->sd->sacl :
5425 old_secdesc_ctr->sd->sacl;
5427 /* Make a deep copy of the security descriptor */
5429 psd = make_sec_desc(mem_ctx, secdesc_ctr->sd->revision, secdesc_ctr->sd->type,
5430 owner_sid, group_sid,
5431 sacl,
5432 dacl,
5433 &size);
5435 if (!psd) {
5436 status = WERR_NOMEM;
5437 goto out;
5440 new_secdesc_ctr = make_sec_desc_buf(mem_ctx, size, psd);
5443 if (!new_secdesc_ctr) {
5444 new_secdesc_ctr = secdesc_ctr;
5447 /* Store the security descriptor in a tdb */
5449 if (!prs_init(&ps,
5450 (uint32)ndr_size_security_descriptor(new_secdesc_ctr->sd, 0)
5451 + sizeof(SEC_DESC_BUF), mem_ctx, MARSHALL) ) {
5452 status = WERR_NOMEM;
5453 goto out;
5457 prs_init_done = true;
5459 if (!sec_io_desc_buf("nt_printing_setsec", &new_secdesc_ctr,
5460 &ps, 1)) {
5461 status = WERR_BADFUNC;
5462 goto out;
5465 kbuf = make_printers_secdesc_tdbkey(mem_ctx, sharename );
5467 if (tdb_prs_store(tdb_printers, kbuf, &ps)==0) {
5468 status = WERR_OK;
5469 } else {
5470 DEBUG(1,("Failed to store secdesc for %s\n", sharename));
5471 status = WERR_BADFUNC;
5474 /* Free malloc'ed memory */
5476 out:
5478 if (prs_init_done) {
5479 prs_mem_free(&ps);
5481 if (mem_ctx)
5482 talloc_destroy(mem_ctx);
5483 return status;
5486 /****************************************************************************
5487 Construct a default security descriptor buffer for a printer.
5488 ****************************************************************************/
5490 static SEC_DESC_BUF *construct_default_printer_sdb(TALLOC_CTX *ctx)
5492 SEC_ACE ace[5]; /* max number of ace entries */
5493 int i = 0;
5494 SEC_ACCESS sa;
5495 SEC_ACL *psa = NULL;
5496 SEC_DESC_BUF *sdb = NULL;
5497 SEC_DESC *psd = NULL;
5498 DOM_SID adm_sid;
5499 size_t sd_size;
5501 /* Create an ACE where Everyone is allowed to print */
5503 init_sec_access(&sa, PRINTER_ACE_PRINT);
5504 init_sec_ace(&ace[i++], &global_sid_World, SEC_ACE_TYPE_ACCESS_ALLOWED,
5505 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5507 /* Add the domain admins group if we are a DC */
5509 if ( IS_DC ) {
5510 DOM_SID domadmins_sid;
5512 sid_copy(&domadmins_sid, get_global_sam_sid());
5513 sid_append_rid(&domadmins_sid, DOMAIN_GROUP_RID_ADMINS);
5515 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5516 init_sec_ace(&ace[i++], &domadmins_sid,
5517 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5518 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5519 init_sec_ace(&ace[i++], &domadmins_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
5520 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5522 else if (secrets_fetch_domain_sid(lp_workgroup(), &adm_sid)) {
5523 sid_append_rid(&adm_sid, DOMAIN_USER_RID_ADMIN);
5525 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5526 init_sec_ace(&ace[i++], &adm_sid,
5527 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5528 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5529 init_sec_ace(&ace[i++], &adm_sid, SEC_ACE_TYPE_ACCESS_ALLOWED,
5530 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5533 /* add BUILTIN\Administrators as FULL CONTROL */
5535 init_sec_access(&sa, PRINTER_ACE_FULL_CONTROL);
5536 init_sec_ace(&ace[i++], &global_sid_Builtin_Administrators,
5537 SEC_ACE_TYPE_ACCESS_ALLOWED, sa,
5538 SEC_ACE_FLAG_OBJECT_INHERIT | SEC_ACE_FLAG_INHERIT_ONLY);
5539 init_sec_ace(&ace[i++], &global_sid_Builtin_Administrators,
5540 SEC_ACE_TYPE_ACCESS_ALLOWED,
5541 sa, SEC_ACE_FLAG_CONTAINER_INHERIT);
5543 /* Make the security descriptor owned by the BUILTIN\Administrators */
5545 /* The ACL revision number in rpc_secdesc.h differs from the one
5546 created by NT when setting ACE entries in printer
5547 descriptors. NT4 complains about the property being edited by a
5548 NT5 machine. */
5550 if ((psa = make_sec_acl(ctx, NT4_ACL_REVISION, i, ace)) != NULL) {
5551 psd = make_sec_desc(ctx, SEC_DESC_REVISION, SEC_DESC_SELF_RELATIVE,
5552 &global_sid_Builtin_Administrators,
5553 &global_sid_Builtin_Administrators,
5554 NULL, psa, &sd_size);
5557 if (!psd) {
5558 DEBUG(0,("construct_default_printer_sd: Failed to make SEC_DESC.\n"));
5559 return NULL;
5562 sdb = make_sec_desc_buf(ctx, sd_size, psd);
5564 DEBUG(4,("construct_default_printer_sdb: size = %u.\n",
5565 (unsigned int)sd_size));
5567 return sdb;
5570 /****************************************************************************
5571 Get a security desc for a printer.
5572 ****************************************************************************/
5574 bool nt_printing_getsec(TALLOC_CTX *ctx, const char *sharename, SEC_DESC_BUF **secdesc_ctr)
5576 prs_struct ps;
5577 TDB_DATA kbuf;
5578 char *temp;
5580 if (strlen(sharename) > 2 && (temp = strchr(sharename + 2, '\\'))) {
5581 sharename = temp + 1;
5584 ZERO_STRUCT(ps);
5586 /* Fetch security descriptor from tdb */
5588 kbuf = make_printers_secdesc_tdbkey(ctx, sharename );
5590 if (tdb_prs_fetch(tdb_printers, kbuf, &ps, ctx)!=0 ||
5591 !sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
5593 prs_mem_free(&ps);
5595 DEBUG(4,("using default secdesc for %s\n", sharename));
5597 if (!(*secdesc_ctr = construct_default_printer_sdb(ctx))) {
5598 return False;
5601 /* Save default security descriptor for later */
5603 if (!prs_init(&ps, (uint32)ndr_size_security_descriptor((*secdesc_ctr)->sd, 0) +
5604 sizeof(SEC_DESC_BUF), ctx, MARSHALL))
5605 return False;
5607 if (sec_io_desc_buf("nt_printing_getsec", secdesc_ctr, &ps, 1)) {
5608 tdb_prs_store(tdb_printers, kbuf, &ps);
5611 prs_mem_free(&ps);
5613 return True;
5616 prs_mem_free(&ps);
5618 /* If security descriptor is owned by S-1-1-0 and winbindd is up,
5619 this security descriptor has been created when winbindd was
5620 down. Take ownership of security descriptor. */
5622 if (sid_equal((*secdesc_ctr)->sd->owner_sid, &global_sid_World)) {
5623 DOM_SID owner_sid;
5625 /* Change sd owner to workgroup administrator */
5627 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
5628 SEC_DESC_BUF *new_secdesc_ctr = NULL;
5629 SEC_DESC *psd = NULL;
5630 size_t size;
5632 /* Create new sd */
5634 sid_append_rid(&owner_sid, DOMAIN_USER_RID_ADMIN);
5636 psd = make_sec_desc(ctx, (*secdesc_ctr)->sd->revision, (*secdesc_ctr)->sd->type,
5637 &owner_sid,
5638 (*secdesc_ctr)->sd->group_sid,
5639 (*secdesc_ctr)->sd->sacl,
5640 (*secdesc_ctr)->sd->dacl,
5641 &size);
5643 if (!psd) {
5644 return False;
5647 new_secdesc_ctr = make_sec_desc_buf(ctx, size, psd);
5648 if (!new_secdesc_ctr) {
5649 return False;
5652 /* Swap with other one */
5654 *secdesc_ctr = new_secdesc_ctr;
5656 /* Set it */
5658 nt_printing_setsec(sharename, *secdesc_ctr);
5662 if (DEBUGLEVEL >= 10) {
5663 SEC_ACL *the_acl = (*secdesc_ctr)->sd->dacl;
5664 int i;
5666 DEBUG(10, ("secdesc_ctr for %s has %d aces:\n",
5667 sharename, the_acl->num_aces));
5669 for (i = 0; i < the_acl->num_aces; i++) {
5670 DEBUG(10, ("%s %d %d 0x%08x\n",
5671 sid_string_dbg(&the_acl->aces[i].trustee),
5672 the_acl->aces[i].type, the_acl->aces[i].flags,
5673 the_acl->aces[i].access_mask));
5677 return True;
5680 /* error code:
5681 0: everything OK
5682 1: level not implemented
5683 2: file doesn't exist
5684 3: can't allocate memory
5685 4: can't free memory
5686 5: non existant struct
5690 A printer and a printer driver are 2 different things.
5691 NT manages them separatelly, Samba does the same.
5692 Why ? Simply because it's easier and it makes sense !
5694 Now explanation: You have 3 printers behind your samba server,
5695 2 of them are the same make and model (laser A and B). But laser B
5696 has an 3000 sheet feeder and laser A doesn't such an option.
5697 Your third printer is an old dot-matrix model for the accounting :-).
5699 If the /usr/local/samba/lib directory (default dir), you will have
5700 5 files to describe all of this.
5702 3 files for the printers (1 by printer):
5703 NTprinter_laser A
5704 NTprinter_laser B
5705 NTprinter_accounting
5706 2 files for the drivers (1 for the laser and 1 for the dot matrix)
5707 NTdriver_printer model X
5708 NTdriver_printer model Y
5710 jfm: I should use this comment for the text file to explain
5711 same thing for the forms BTW.
5712 Je devrais mettre mes commentaires en francais, ca serait mieux :-)
5716 /* Convert generic access rights to printer object specific access rights.
5717 It turns out that NT4 security descriptors use generic access rights and
5718 NT5 the object specific ones. */
5720 void map_printer_permissions(SEC_DESC *sd)
5722 int i;
5724 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
5725 se_map_generic(&sd->dacl->aces[i].access_mask,
5726 &printer_generic_mapping);
5730 void map_job_permissions(SEC_DESC *sd)
5732 int i;
5734 for (i = 0; sd->dacl && i < sd->dacl->num_aces; i++) {
5735 se_map_generic(&sd->dacl->aces[i].access_mask,
5736 &job_generic_mapping);
5741 /****************************************************************************
5742 Check a user has permissions to perform the given operation. We use the
5743 permission constants defined in include/rpc_spoolss.h to check the various
5744 actions we perform when checking printer access.
5746 PRINTER_ACCESS_ADMINISTER:
5747 print_queue_pause, print_queue_resume, update_printer_sec,
5748 update_printer, spoolss_addprinterex_level_2,
5749 _spoolss_setprinterdata
5751 PRINTER_ACCESS_USE:
5752 print_job_start
5754 JOB_ACCESS_ADMINISTER:
5755 print_job_delete, print_job_pause, print_job_resume,
5756 print_queue_purge
5758 Try access control in the following order (for performance reasons):
5759 1) root ans SE_PRINT_OPERATOR can do anything (easy check)
5760 2) check security descriptor (bit comparisons in memory)
5761 3) "printer admins" (may result in numerous calls to winbind)
5763 ****************************************************************************/
5764 bool print_access_check(struct auth_serversupplied_info *server_info, int snum,
5765 int access_type)
5767 SEC_DESC_BUF *secdesc = NULL;
5768 uint32 access_granted;
5769 NTSTATUS status;
5770 bool result;
5771 const char *pname;
5772 TALLOC_CTX *mem_ctx = NULL;
5773 SE_PRIV se_printop = SE_PRINT_OPERATOR;
5775 /* If user is NULL then use the current_user structure */
5777 /* Always allow root or SE_PRINT_OPERATROR to do anything */
5779 if (server_info->utok.uid == 0
5780 || user_has_privileges(server_info->ptok, &se_printop ) ) {
5781 return True;
5784 /* Get printer name */
5786 pname = PRINTERNAME(snum);
5788 if (!pname || !*pname) {
5789 errno = EACCES;
5790 return False;
5793 /* Get printer security descriptor */
5795 if(!(mem_ctx = talloc_init("print_access_check"))) {
5796 errno = ENOMEM;
5797 return False;
5800 if (!nt_printing_getsec(mem_ctx, pname, &secdesc)) {
5801 talloc_destroy(mem_ctx);
5802 errno = ENOMEM;
5803 return False;
5806 if (access_type == JOB_ACCESS_ADMINISTER) {
5807 SEC_DESC_BUF *parent_secdesc = secdesc;
5809 /* Create a child security descriptor to check permissions
5810 against. This is because print jobs are child objects
5811 objects of a printer. */
5813 secdesc = se_create_child_secdesc(mem_ctx, parent_secdesc->sd, False);
5815 if (!secdesc) {
5816 talloc_destroy(mem_ctx);
5817 errno = ENOMEM;
5818 return False;
5821 map_job_permissions(secdesc->sd);
5822 } else {
5823 map_printer_permissions(secdesc->sd);
5826 /* Check access */
5827 result = se_access_check(secdesc->sd, server_info->ptok, access_type,
5828 &access_granted, &status);
5830 DEBUG(4, ("access check was %s\n", result ? "SUCCESS" : "FAILURE"));
5832 /* see if we need to try the printer admin list */
5834 if ((access_granted == 0) &&
5835 (token_contains_name_in_list(uidtoname(server_info->utok.uid),
5836 NULL, NULL, server_info->ptok,
5837 lp_printer_admin(snum)))) {
5838 talloc_destroy(mem_ctx);
5839 return True;
5842 talloc_destroy(mem_ctx);
5844 if (!result) {
5845 errno = EACCES;
5848 return result;
5851 /****************************************************************************
5852 Check the time parameters allow a print operation.
5853 *****************************************************************************/
5855 bool print_time_access_check(const char *servicename)
5857 NT_PRINTER_INFO_LEVEL *printer = NULL;
5858 bool ok = False;
5859 time_t now = time(NULL);
5860 struct tm *t;
5861 uint32 mins;
5863 if (!W_ERROR_IS_OK(get_a_printer(NULL, &printer, 2, servicename)))
5864 return False;
5866 if (printer->info_2->starttime == 0 && printer->info_2->untiltime == 0)
5867 ok = True;
5869 t = gmtime(&now);
5870 mins = (uint32)t->tm_hour*60 + (uint32)t->tm_min;
5872 if (mins >= printer->info_2->starttime && mins <= printer->info_2->untiltime)
5873 ok = True;
5875 free_a_printer(&printer, 2);
5877 if (!ok)
5878 errno = EACCES;
5880 return ok;
5883 /****************************************************************************
5884 Fill in the servername sent in the _spoolss_open_printer_ex() call
5885 ****************************************************************************/
5887 char* get_server_name( Printer_entry *printer )
5889 return printer->servername;