Caught one problem where a BOOL fn was returning 3 different values :-).
[ira/wip.git] / source3 / printing / nt_printing.c
1 /*
2  *  Unix SMB/Netbios implementation.
3  *  Version 1.9.
4  *  RPC Pipe client / server routines
5  *  Copyright (C) Andrew Tridgell              1992-2000,
6  *  Copyright (C) Jean François Micouleau      1998-2000.
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
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.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21  */
22
23 #include "includes.h"
24
25 extern DOM_SID global_sid_World;
26
27 static TDB_CONTEXT *tdb_forms; /* used for forms files */
28 static TDB_CONTEXT *tdb_drivers; /* used for driver files */
29 static TDB_CONTEXT *tdb_printers; /* used for printers files */
30
31 #define FORMS_PREFIX "FORMS/"
32 #define DRIVERS_PREFIX "DRIVERS/"
33 #define DRIVER_INIT_PREFIX "DRIVER_INIT/"
34 #define PRINTERS_PREFIX "PRINTERS/"
35 #define SECDESC_PREFIX "SECDESC/"
36  
37 #define NTDRIVERS_DATABASE_VERSION_1 1
38 #define NTDRIVERS_DATABASE_VERSION_2 2
39  
40 #define NTDRIVERS_DATABASE_VERSION NTDRIVERS_DATABASE_VERSION_2
41
42 /* Map generic permissions to printer object specific permissions */
43
44 struct generic_mapping printer_generic_mapping = {
45         PRINTER_READ,
46         PRINTER_WRITE,
47         PRINTER_EXECUTE,
48         PRINTER_ALL_ACCESS
49 };
50
51 /* We need one default form to support our default printer. Msoft adds the
52 forms it wants and in the ORDER it wants them (note: DEVMODE papersize is an
53 array index). Letter is always first, so (for the current code) additions
54 always put things in the correct order. */
55 static nt_forms_struct default_forms[] = {
56         {"Letter",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
57         {"Letter Small",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
58         {"Tabloid",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
59         {"Ledger",0x1,0x696b8,0x44368,0x0,0x0,0x696b8,0x44368},
60         {"Legal",0x1,0x34b5c,0x56d10,0x0,0x0,0x34b5c,0x56d10},
61         {"Statement",0x1,0x221b4,0x34b5c,0x0,0x0,0x221b4,0x34b5c},
62         {"Executive",0x1,0x2cf56,0x411cc,0x0,0x0,0x2cf56,0x411cc},
63         {"A3",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
64         {"A4",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
65         {"A4 Small",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
66         {"A5",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
67         {"B4 (JIS)",0x1,0x3ebe8,0x58de0,0x0,0x0,0x3ebe8,0x58de0},
68         {"B5 (JIS)",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
69         {"Folio",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
70         {"Quarto",0x1,0x347d8,0x43238,0x0,0x0,0x347d8,0x43238},
71         {"10x14",0x1,0x3e030,0x56d10,0x0,0x0,0x3e030,0x56d10},
72         {"11x17",0x1,0x44368,0x696b8,0x0,0x0,0x44368,0x696b8},
73         {"Note",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
74         {"Envelope #9",0x1,0x18079,0x37091,0x0,0x0,0x18079,0x37091},
75         {"Envelope #10",0x1,0x19947,0x3ae94,0x0,0x0,0x19947,0x3ae94},
76         {"Envelope #11",0x1,0x1be7c,0x40565,0x0,0x0,0x1be7c,0x40565},
77         {"Envelope #12",0x1,0x1d74a,0x44368,0x0,0x0,0x1d74a,0x44368},
78         {"Envelope #14",0x1,0x1f018,0x47504,0x0,0x0,0x1f018,0x47504},
79         {"C size sheet",0x1,0x696b8,0x886d0,0x0,0x0,0x696b8,0x886d0},
80         {"D size sheet",0x1,0x886d0,0xd2d70,0x0,0x0,0x886d0,0xd2d70},
81         {"E size sheet",0x1,0xd2d70,0x110da0,0x0,0x0,0xd2d70,0x110da0},
82         {"Envelope DL",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
83         {"Envelope C5",0x1,0x278d0,0x37e88,0x0,0x0,0x278d0,0x37e88},
84         {"Envelope C3",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
85         {"Envelope C4",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
86         {"Envelope C6",0x1,0x1bd50,0x278d0,0x0,0x0,0x1bd50,0x278d0},
87         {"Envelope C65",0x1,0x1bd50,0x37e88,0x0,0x0,0x1bd50,0x37e88},
88         {"Envelope B4",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
89         {"Envelope B5",0x1,0x2af80,0x3d090,0x0,0x0,0x2af80,0x3d090},
90         {"Envelope B6",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
91         {"Envelope",0x1,0x1adb0,0x38270,0x0,0x0,0x1adb0,0x38270},
92         {"Envelope Monarch",0x1,0x18079,0x2e824,0x0,0x0,0x18079,0x2e824},
93         {"6 3/4 Envelope",0x1,0x167ab,0x284ec,0x0,0x0,0x167ab,0x284ec},
94         {"US Std Fanfold",0x1,0x5c3e1,0x44368,0x0,0x0,0x5c3e1,0x44368},
95         {"German Std Fanfold",0x1,0x34b5c,0x4a6a0,0x0,0x0,0x34b5c,0x4a6a0},
96         {"German Legal Fanfold",0x1,0x34b5c,0x509d8,0x0,0x0,0x34b5c,0x509d8},
97         {"B4 (ISO)",0x1,0x3d090,0x562e8,0x0,0x0,0x3d090,0x562e8},
98         {"Japanese Postcard",0x1,0x186a0,0x24220,0x0,0x0,0x186a0,0x24220},
99         {"9x11",0x1,0x37cf8,0x44368,0x0,0x0,0x37cf8,0x44368},
100         {"10x11",0x1,0x3e030,0x44368,0x0,0x0,0x3e030,0x44368},
101         {"15x11",0x1,0x5d048,0x44368,0x0,0x0,0x5d048,0x44368},
102         {"Envelope Invite",0x1,0x35b60,0x35b60,0x0,0x0,0x35b60,0x35b60},
103         {"Reserved48",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
104         {"Reserved49",0x1,0x1,0x1,0x0,0x0,0x1,0x1},
105         {"Letter Extra",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
106         {"Legal Extra",0x1,0x3ae94,0x5d048,0x0,0x0,0x3ae94,0x5d048},
107         {"Tabloid Extra",0x1,0x4a6a0,0x6f9f0,0x0,0x0,0x4a6a0,0x6f9f0},
108         {"A4 Extra",0x1,0x397c2,0x4eb16,0x0,0x0,0x397c2,0x4eb16},
109         {"Letter Transverse",0x1,0x34b5c,0x44368,0x0,0x0,0x34b5c,0x44368},
110         {"A4 Transverse",0x1,0x33450,0x48828,0x0,0x0,0x33450,0x48828},
111         {"Letter Extra Transverse",0x1,0x3ae94,0x4a6a0,0x0,0x0,0x3ae94,0x4a6a0},
112         {"Super A",0x1,0x376b8,0x56ea0,0x0,0x0,0x376b8,0x56ea0},
113         {"Super B",0x1,0x4a768,0x76e58,0x0,0x0,0x4a768,0x76e58},
114         {"Letter Plus",0x1,0x34b5c,0x4eb16,0x0,0x0,0x34b5c,0x4eb16},
115         {"A4 Plus",0x1,0x33450,0x50910,0x0,0x0,0x33450,0x50910},
116         {"A5 Transverse",0x1,0x24220,0x33450,0x0,0x0,0x24220,0x33450},
117         {"B5 (JIS) Transverse",0x1,0x2c6f0,0x3ebe8,0x0,0x0,0x2c6f0,0x3ebe8},
118         {"A3 Extra",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
119         {"A5 Extra",0x1,0x2a7b0,0x395f8,0x0,0x0,0x2a7b0,0x395f8},
120         {"B5 (ISO) Extra",0x1,0x31128,0x43620,0x0,0x0,0x31128,0x43620},
121         {"A2",0x1,0x668a0,0x91050,0x0,0x0,0x668a0,0x91050},
122         {"A3 Transverse",0x1,0x48828,0x668a0,0x0,0x0,0x48828,0x668a0},
123         {"A3 Extra Transverse",0x1,0x4e9d0,0x6ca48,0x0,0x0,0x4e9d0,0x6ca48},
124         {"Japanese Double Postcard",0x1,0x30d40,0x24220,0x0,0x0,0x30d40,0x24220},
125         {"A6",0x1,0x19a28,0x24220,0x0,0x0,0x19a28,0x24220},
126         {"Japanese Envelope Kaku #2",0x1,0x3a980,0x510e0,0x0,0x0,0x3a980,0x510e0},
127         {"Japanese Envelope Kaku #3",0x1,0x34bc0,0x43a08,0x0,0x0,0x34bc0,0x43a08},
128         {"Japanese Envelope Chou #3",0x1,0x1d4c0,0x395f8,0x0,0x0,0x1d4c0,0x395f8},
129         {"Japanese Envelope Chou #4",0x1,0x15f90,0x320c8,0x0,0x0,0x15f90,0x320c8},
130         {"Letter Rotated",0x1,0x44368,0x34b5c,0x0,0x0,0x44368,0x34b5c},
131         {"A3 Rotated",0x1,0x668a0,0x48828,0x0,0x0,0x668a0,0x48828},
132         {"A4 Rotated",0x1,0x48828,0x33450,0x0,0x0,0x48828,0x33450},
133         {"A5 Rotated",0x1,0x33450,0x24220,0x0,0x0,0x33450,0x24220},
134         {"B4 (JIS) Rotated",0x1,0x58de0,0x3ebe8,0x0,0x0,0x58de0,0x3ebe8},
135         {"B5 (JIS) Rotated",0x1,0x3ebe8,0x2c6f0,0x0,0x0,0x3ebe8,0x2c6f0},
136         {"Japanese Postcard Rotated",0x1,0x24220,0x186a0,0x0,0x0,0x24220,0x186a0},
137         {"Double Japan Postcard Rotated",0x1,0x24220,0x30d40,0x0,0x0,0x24220,0x30d40},
138         {"A6 Rotated",0x1,0x24220,0x19a28,0x0,0x0,0x24220,0x19a28},
139         {"Japan Envelope Kaku #2 Rotated",0x1,0x510e0,0x3a980,0x0,0x0,0x510e0,0x3a980},
140         {"Japan Envelope Kaku #3 Rotated",0x1,0x43a08,0x34bc0,0x0,0x0,0x43a08, 0x34bc0},
141         {"Japan Envelope Chou #3 Rotated",0x1,0x395f8,0x1d4c0,0x0,0x0,0x395f8,0x1d4c0},
142         {"Japan Envelope Chou #4 Rotated",0x1,0x320c8,0x15f90,0x0,0x0,0x320c8,0x15f90},
143         {"B6 (JIS)",0x1,0x1f400,0x2c6f0,0x0,0x0,0x1f400,0x2c6f0},
144         {"B6 (JIS) Rotated",0x1,0x2c6f0,0x1f400,0x0,0x0,0x2c6f0,0x1f400},
145         {"12x11",0x1,0x4a724,0x443e1,0x0,0x0,0x4a724,0x443e1},
146         {"Japan Envelope You #4",0x1,0x19a28,0x395f8,0x0,0x0,0x19a28,0x395f8},
147         {"Japan Envelope You #4 Rotated",0x1,0x395f8,0x19a28,0x0,0x0,0x395f8,0x19a28},
148         {"PRC 16K",0x1,0x2de60,0x3f7a0,0x0,0x0,0x2de60,0x3f7a0},
149         {"PRC 32K",0x1,0x1fbd0,0x2cec0,0x0,0x0,0x1fbd0,0x2cec0},
150         {"PRC 32K(Big)",0x1,0x222e0,0x318f8,0x0,0x0,0x222e0,0x318f8},
151         {"PRC Envelope #1",0x1,0x18e70,0x28488,0x0,0x0,0x18e70,0x28488},
152         {"PRC Envelope #2",0x1,0x18e70,0x2af80,0x0,0x0,0x18e70,0x2af80},
153         {"PRC Envelope #3",0x1,0x1e848,0x2af80,0x0,0x0,0x1e848,0x2af80},
154         {"PRC Envelope #4",0x1,0x1adb0,0x32c80,0x0,0x0,0x1adb0,0x32c80},
155         {"PRC Envelope #5",0x1,0x1adb0,0x35b60,0x0,0x0,0x1adb0,0x35b60},
156         {"PRC Envelope #6",0x1,0x1d4c0,0x38270,0x0,0x0,0x1d4c0,0x38270},
157         {"PRC Envelope #7",0x1,0x27100,0x38270,0x0,0x0,0x27100,0x38270},
158         {"PRC Envelope #8",0x1,0x1d4c0,0x4b708,0x0,0x0,0x1d4c0,0x4b708},
159         {"PRC Envelope #9",0x1,0x37e88,0x4f1a0,0x0,0x0,0x37e88,0x4f1a0},
160         {"PRC Envelope #10",0x1,0x4f1a0,0x6fd10,0x0,0x0,0x4f1a0,0x6fd10},
161         {"PRC 16K Rotated",0x1,0x3f7a0,0x2de60,0x0,0x0,0x3f7a0,0x2de60},
162         {"PRC 32K Rotated",0x1,0x2cec0,0x1fbd0,0x0,0x0,0x2cec0,0x1fbd0},
163         {"PRC 32K(Big) Rotated",0x1,0x318f8,0x222e0,0x0,0x0,0x318f8,0x222e0},
164         {"PRC Envelope #1 Rotated",0x1,0x28488,0x18e70,0x0,0x0,0x28488,0x18e70},
165         {"PRC Envelope #2 Rotated",0x1,0x2af80,0x18e70,0x0,0x0,0x2af80,0x18e70},
166         {"PRC Envelope #3 Rotated",0x1,0x2af80,0x1e848,0x0,0x0,0x2af80,0x1e848},
167         {"PRC Envelope #4 Rotated",0x1,0x32c80,0x1adb0,0x0,0x0,0x32c80,0x1adb0},
168         {"PRC Envelope #5 Rotated",0x1,0x35b60,0x1adb0,0x0,0x0,0x35b60,0x1adb0},
169         {"PRC Envelope #6 Rotated",0x1,0x38270,0x1d4c0,0x0,0x0,0x38270,0x1d4c0},
170         {"PRC Envelope #7 Rotated",0x1,0x38270,0x27100,0x0,0x0,0x38270,0x27100},
171         {"PRC Envelope #8 Rotated",0x1,0x4b708,0x1d4c0,0x0,0x0,0x4b708,0x1d4c0},
172         {"PRC Envelope #9 Rotated",0x1,0x4f1a0,0x37e88,0x0,0x0,0x4f1a0,0x37e88},
173         {"PRC Envelope #10 Rotated",0x1,0x6fd10,0x4f1a0,0x0,0x0,0x6fd10,0x4f1a0}
174 };
175
176 static BOOL upgrade_to_version_2(void)
177 {
178         TDB_DATA kbuf, newkey, dbuf;
179  
180         DEBUG(0,("upgrade_to_version_2: upgrading print tdb's to version 2\n"));
181  
182         for (kbuf = tdb_firstkey(tdb_drivers); kbuf.dptr;
183                         newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
184
185                 dbuf = tdb_fetch(tdb_drivers, kbuf);
186
187                 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) == 0) {
188                         DEBUG(0,("upgrade_to_version_2:moving form\n"));
189                         if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) {
190                                 DEBUG(0,("upgrade_to_version_2: failed to move form. Error (%s).\n", tdb_errorstr(tdb_forms)));
191                                 return False;
192                         }
193                         if (tdb_delete(tdb_drivers, kbuf) != 0) {
194                                 DEBUG(0,("upgrade_to_version_2: failed to delete form. Error (%s)\n", tdb_errorstr(tdb_drivers)));
195                                 return False;
196                         }
197                 }
198  
199                 if (strncmp(kbuf.dptr, PRINTERS_PREFIX, strlen(PRINTERS_PREFIX)) == 0) {
200                         DEBUG(0,("upgrade_to_version_2:moving printer\n"));
201                         if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
202                                 DEBUG(0,("upgrade_to_version_2: failed to move printer. Error (%s)\n", tdb_errorstr(tdb_printers)));
203                                 return False;
204                         }
205                         if (tdb_delete(tdb_drivers, kbuf) != 0) {
206                                 DEBUG(0,("upgrade_to_version_2: failed to delete printer. Error (%s)\n", tdb_errorstr(tdb_drivers)));
207                                 return False;
208                         }
209                 }
210  
211                 if (strncmp(kbuf.dptr, SECDESC_PREFIX, strlen(SECDESC_PREFIX)) == 0) {
212                         DEBUG(0,("upgrade_to_version_2:moving secdesc\n"));
213                         if (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) != 0) {
214                                 DEBUG(0,("upgrade_to_version_2: failed to move secdesc. Error (%s)\n", tdb_errorstr(tdb_printers)));
215                                 return False;
216                         }
217                         if (tdb_delete(tdb_drivers, kbuf) != 0) {
218                                 DEBUG(0,("upgrade_to_version_2: failed to delete secdesc. Error (%s)\n", tdb_errorstr(tdb_drivers)));
219                                 return False;
220                         }
221                 }
222  
223                 SAFE_FREE(dbuf.dptr);
224         }
225
226         return True;
227 }
228
229 /****************************************************************************
230 open the NT printing tdb
231 ****************************************************************************/
232 BOOL nt_printing_init(void)
233 {
234         static pid_t local_pid;
235         char *vstring = "INFO/version";
236
237         if (tdb_drivers && tdb_printers && tdb_forms && local_pid == sys_getpid())
238                 return True;
239  
240         tdb_drivers = tdb_open_log(lock_path("ntdrivers.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
241         if (!tdb_drivers) {
242                 DEBUG(0,("nt_printing_init: Failed to open nt drivers database %s (%s)\n",
243                         lock_path("ntdrivers.tdb"), strerror(errno) ));
244                 return False;
245         }
246  
247         tdb_printers = tdb_open_log(lock_path("ntprinters.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
248         if (!tdb_printers) {
249                 DEBUG(0,("nt_printing_init: Failed to open nt printers database %s (%s)\n",
250                         lock_path("ntprinters.tdb"), strerror(errno) ));
251                 return False;
252         }
253  
254         tdb_forms = tdb_open_log(lock_path("ntforms.tdb"), 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0600);
255         if (!tdb_forms) {
256                 DEBUG(0,("nt_printing_init: Failed to open nt forms database %s (%s)\n",
257                         lock_path("ntforms.tdb"), strerror(errno) ));
258                 return False;
259         }
260  
261         local_pid = sys_getpid();
262  
263         /* handle a Samba upgrade */
264         tdb_lock_bystring(tdb_drivers, vstring);
265         if (tdb_fetch_int(tdb_drivers, vstring) != NTDRIVERS_DATABASE_VERSION) {
266  
267                 if (tdb_fetch_int(tdb_drivers, vstring) == NTDRIVERS_DATABASE_VERSION_1) {
268                         if (!upgrade_to_version_2())
269                                 return False;
270                 } else
271                         tdb_traverse(tdb_drivers, tdb_traverse_delete_fn, NULL);
272  
273                 tdb_store_int(tdb_drivers, vstring, NTDRIVERS_DATABASE_VERSION);
274         }
275         tdb_unlock_bystring(tdb_drivers, vstring);
276
277         return True;
278 }
279
280 /****************************************************************************
281  get builtin form struct list
282 ****************************************************************************/
283 int get_builtin_ntforms(nt_forms_struct **list)
284 {
285         *list = (nt_forms_struct *)memdup(&default_forms[0], sizeof(default_forms));
286         return sizeof(default_forms) / sizeof(default_forms[0]);
287 }
288
289 /****************************************************************************
290  get a builtin form struct
291 ****************************************************************************/
292
293 BOOL get_a_builtin_ntform(UNISTR2 *uni_formname,nt_forms_struct *form)
294 {
295         int i,count;
296         fstring form_name;
297         unistr2_to_ascii(form_name, uni_formname, sizeof(form_name)-1);
298         DEBUGADD(6,("Looking for builtin form %s \n", form_name));
299         count = sizeof(default_forms) / sizeof(default_forms[0]);
300         for (i=0;i<count;i++) {
301                 if (strequal(form_name,default_forms[i].name)) {
302                         DEBUGADD(6,("Found builtin form %s \n", form_name));
303                         memcpy(form,&default_forms[i],sizeof(*form));
304                         break;
305                 }
306         }
307
308         return (i !=count);
309 }
310
311 /****************************************************************************
312 get a form struct list
313 ****************************************************************************/
314 int get_ntforms(nt_forms_struct **list)
315 {
316         TDB_DATA kbuf, newkey, dbuf;
317         nt_forms_struct *tl;
318         nt_forms_struct form;
319         int ret;
320         int i;
321         int n = 0;
322
323         for (kbuf = tdb_firstkey(tdb_forms);
324              kbuf.dptr;
325              newkey = tdb_nextkey(tdb_forms, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
326                 if (strncmp(kbuf.dptr, FORMS_PREFIX, strlen(FORMS_PREFIX)) != 0) continue;
327                 
328                 dbuf = tdb_fetch(tdb_forms, kbuf);
329                 if (!dbuf.dptr) continue;
330
331                 fstrcpy(form.name, kbuf.dptr+strlen(FORMS_PREFIX));
332                 ret = tdb_unpack(dbuf.dptr, dbuf.dsize, "dddddddd",
333                                  &i, &form.flag, &form.width, &form.length, &form.left,
334                                  &form.top, &form.right, &form.bottom);
335                 SAFE_FREE(dbuf.dptr);
336                 if (ret != dbuf.dsize) continue;
337
338                 tl = Realloc(*list, sizeof(nt_forms_struct)*(n+1));
339                 if (!tl) {
340                         DEBUG(0,("get_ntforms: Realloc fail.\n"));
341                         return 0;
342                 }
343                 *list = tl;
344                 (*list)[n] = form;
345                 n++;
346         }
347         
348
349         return n;
350 }
351
352 /****************************************************************************
353 write a form struct list
354 ****************************************************************************/
355 int write_ntforms(nt_forms_struct **list, int number)
356 {
357         pstring buf, key;
358         int len;
359         TDB_DATA kbuf,dbuf;
360         int i;
361
362         for (i=0;i<number;i++) {
363                 /* save index, so list is rebuilt in correct order */
364                 len = tdb_pack(buf, sizeof(buf), "dddddddd",
365                                i, (*list)[i].flag, (*list)[i].width, (*list)[i].length,
366                                (*list)[i].left, (*list)[i].top, (*list)[i].right,
367                                (*list)[i].bottom);
368                 if (len > sizeof(buf)) break;
369                 slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[i].name);
370                 kbuf.dsize = strlen(key)+1;
371                 kbuf.dptr = key;
372                 dbuf.dsize = len;
373                 dbuf.dptr = buf;
374                 if (tdb_store(tdb_forms, kbuf, dbuf, TDB_REPLACE) != 0) break;
375        }
376
377        return i;
378 }
379
380 /****************************************************************************
381 add a form struct at the end of the list
382 ****************************************************************************/
383 BOOL add_a_form(nt_forms_struct **list, const FORM *form, int *count)
384 {
385         int n=0;
386         BOOL update;
387         fstring form_name;
388         nt_forms_struct *tl;
389
390         /*
391          * NT tries to add forms even when
392          * they are already in the base
393          * only update the values if already present
394          */
395
396         update=False;
397         
398         unistr2_to_ascii(form_name, &form->name, sizeof(form_name)-1);
399         for (n=0; n<*count; n++) {
400                 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
401                         DEBUG(103, ("NT workaround, [%s] already exists\n", form_name));
402                         update=True;
403                         break;
404                 }
405         }
406
407         if (update==False) {
408                 if((tl=Realloc(*list, (n+1)*sizeof(nt_forms_struct))) == NULL) {
409                         DEBUG(0,("add_a_form: failed to enlarge forms list!\n"));
410                         return False;
411                 }
412                 *list = tl;
413                 unistr2_to_ascii((*list)[n].name, &form->name, sizeof((*list)[n].name)-1);
414                 (*count)++;
415         }
416         
417         (*list)[n].flag=form->flags;
418         (*list)[n].width=form->size_x;
419         (*list)[n].length=form->size_y;
420         (*list)[n].left=form->left;
421         (*list)[n].top=form->top;
422         (*list)[n].right=form->right;
423         (*list)[n].bottom=form->bottom;
424
425         return True;
426 }
427
428 /****************************************************************************
429  delete a named form struct
430 ****************************************************************************/
431 BOOL delete_a_form(nt_forms_struct **list, UNISTR2 *del_name, int *count, WERROR *ret)
432 {
433         pstring key;
434         TDB_DATA kbuf;
435         int n=0;
436         fstring form_name;
437
438         *ret = WERR_OK;
439
440         unistr2_to_ascii(form_name, del_name, sizeof(form_name)-1);
441
442         for (n=0; n<*count; n++) {
443                 if (!strncmp((*list)[n].name, form_name, strlen(form_name))) {
444                         DEBUG(103, ("delete_a_form, [%s] in list\n", form_name));
445                         break;
446                 }
447         }
448
449         if (n == *count) {
450                 DEBUG(10,("delete_a_form, [%s] not found\n", form_name));
451                 *ret = WERR_INVALID_PARAM;
452                 return False;
453         }
454
455         slprintf(key, sizeof(key)-1, "%s%s", FORMS_PREFIX, (*list)[n].name);
456         kbuf.dsize = strlen(key)+1;
457         kbuf.dptr = key;
458         if (tdb_delete(tdb_forms, kbuf) != 0) {
459                 *ret = WERR_NOMEM;
460                 return False;
461         }
462
463         return True;
464 }
465
466 /****************************************************************************
467 update a form struct
468 ****************************************************************************/
469 void update_a_form(nt_forms_struct **list, const FORM *form, int count)
470 {
471         int n=0;
472         fstring form_name;
473         unistr2_to_ascii(form_name, &(form->name), sizeof(form_name)-1);
474
475         DEBUG(106, ("[%s]\n", form_name));
476         for (n=0; n<count; n++)
477         {
478                 DEBUGADD(106, ("n [%d]:[%s]\n", n, (*list)[n].name));
479                 if (!strncmp((*list)[n].name, form_name, strlen(form_name)))
480                         break;
481         }
482
483         if (n==count) return;
484
485         (*list)[n].flag=form->flags;
486         (*list)[n].width=form->size_x;
487         (*list)[n].length=form->size_y;
488         (*list)[n].left=form->left;
489         (*list)[n].top=form->top;
490         (*list)[n].right=form->right;
491         (*list)[n].bottom=form->bottom;
492 }
493
494 /****************************************************************************
495 get the nt drivers list
496
497 traverse the database and look-up the matching names
498 ****************************************************************************/
499 int get_ntdrivers(fstring **list, char *architecture, uint32 version)
500 {
501         int total=0;
502         fstring short_archi;
503         fstring *fl;
504         pstring key;
505         TDB_DATA kbuf, newkey;
506
507         get_short_archi(short_archi, architecture);
508         slprintf(key, sizeof(key)-1, "%s%s/%d/", DRIVERS_PREFIX, short_archi, version);
509
510         for (kbuf = tdb_firstkey(tdb_drivers);
511              kbuf.dptr;
512              newkey = tdb_nextkey(tdb_drivers, kbuf), safe_free(kbuf.dptr), kbuf=newkey) {
513                 if (strncmp(kbuf.dptr, key, strlen(key)) != 0) continue;
514                 
515                 if((fl = Realloc(*list, sizeof(fstring)*(total+1))) == NULL) {
516                         DEBUG(0,("get_ntdrivers: failed to enlarge list!\n"));
517                         return -1;
518                 }
519                 else *list = fl;
520
521                 fstrcpy((*list)[total], kbuf.dptr+strlen(key));
522                 total++;
523         }
524
525         return(total);
526 }
527
528 /****************************************************************************
529 function to do the mapping between the long architecture name and
530 the short one.
531 ****************************************************************************/
532 BOOL get_short_archi(char *short_archi, char *long_archi)
533 {
534         struct table {
535                 char *long_archi;
536                 char *short_archi;
537         };
538         
539         struct table archi_table[]=
540         {
541                 {"Windows 4.0",          "WIN40"    },
542                 {"Windows NT x86",       "W32X86"   },
543                 {"Windows NT R4000",     "W32MIPS"  },
544                 {"Windows NT Alpha_AXP", "W32ALPHA" },
545                 {"Windows NT PowerPC",   "W32PPC"   },
546                 {NULL,                   ""         }
547         };
548         
549         int i=-1;
550
551         DEBUG(107,("Getting architecture dependant directory\n"));
552         do {
553                 i++;
554         } while ( (archi_table[i].long_archi!=NULL ) &&
555                   StrCaseCmp(long_archi, archi_table[i].long_archi) );
556
557         if (archi_table[i].long_archi==NULL) {
558                 DEBUGADD(107,("Unknown architecture [%s] !\n", long_archi));
559                 return False;
560         }
561
562         StrnCpy (short_archi, archi_table[i].short_archi, strlen(archi_table[i].short_archi));
563
564         DEBUGADD(108,("index: [%d]\n", i));
565         DEBUGADD(108,("long architecture: [%s]\n", long_archi));
566         DEBUGADD(108,("short architecture: [%s]\n", short_archi));
567         
568         return True;
569 }
570
571 /****************************************************************************
572  Version information in Microsoft files is held in a VS_VERSION_INFO structure.
573  There are two case to be covered here: PE (Portable Executable) and NE (New
574  Executable) files. Both files support the same INFO structure, but PE files
575  store the signature in unicode, and NE files store it as !unicode.
576  returns -1 on error, 1 on version info found, and 0 on no version info found.
577 ****************************************************************************/
578
579 static int get_file_version(files_struct *fsp, char *fname,uint32 *major, uint32 *minor)
580 {
581         int     i;
582         char    *buf;
583         ssize_t byte_count;
584
585         if ((buf=malloc(PE_HEADER_SIZE)) == NULL) {
586                 DEBUG(0,("get_file_version: PE file [%s] PE Header malloc failed bytes = %d\n",
587                                 fname, PE_HEADER_SIZE));
588                 goto error_exit;
589         }
590
591         /* Note: DOS_HEADER_SIZE < malloc'ed PE_HEADER_SIZE */
592         if ((byte_count = vfs_read_data(fsp, buf, DOS_HEADER_SIZE)) < DOS_HEADER_SIZE) {
593                 DEBUG(3,("get_file_version: File [%s] DOS header too short, bytes read = %d\n",
594                                 fname, byte_count));
595                 goto no_version_info;
596         }
597
598         /* Is this really a DOS header? */
599         if (SVAL(buf,DOS_HEADER_MAGIC_OFFSET) != DOS_HEADER_MAGIC) {
600                 DEBUG(6,("get_file_version: File [%s] bad DOS magic = 0x%x\n",
601                                 fname, SVAL(buf,DOS_HEADER_MAGIC_OFFSET)));
602                 goto no_version_info;
603         }
604
605         /* Skip OEM header (if any) and the DOS stub to start of Windows header */
606         if (fsp->conn->vfs_ops.lseek(fsp, fsp->fd, SVAL(buf,DOS_HEADER_LFANEW_OFFSET), SEEK_SET) == (SMB_OFF_T)-1) {
607                 DEBUG(3,("get_file_version: File [%s] too short, errno = %d\n",
608                                 fname, errno));
609                 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
610                 goto no_version_info;
611         }
612
613         if ((byte_count = vfs_read_data(fsp, buf, PE_HEADER_SIZE)) < PE_HEADER_SIZE) {
614                 DEBUG(3,("get_file_version: File [%s] Windows header too short, bytes read = %d\n",
615                                 fname, byte_count));
616                 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
617                 goto no_version_info;
618         }
619
620         /* The header may be a PE (Portable Executable) or an NE (New Executable) */
621         if (IVAL(buf,PE_HEADER_SIGNATURE_OFFSET) == PE_HEADER_SIGNATURE) {
622                 int num_sections;
623                 int section_table_bytes;
624                 
625                 if (SVAL(buf,PE_HEADER_MACHINE_OFFSET) != PE_HEADER_MACHINE_I386) {
626                         DEBUG(3,("get_file_version: PE file [%s] wrong machine = 0x%x\n",
627                                         fname, SVAL(buf,PE_HEADER_MACHINE_OFFSET)));
628                         /* At this point, we assume the file is in error. It still could be somthing
629                          * else besides a PE file, but it unlikely at this point.
630                          */
631                         goto error_exit;
632                 }
633
634                 /* get the section table */
635                 num_sections        = SVAL(buf,PE_HEADER_NUMBER_OF_SECTIONS);
636                 section_table_bytes = num_sections * PE_HEADER_SECT_HEADER_SIZE;
637                 SAFE_FREE(buf);
638                 if ((buf=malloc(section_table_bytes)) == NULL) {
639                         DEBUG(0,("get_file_version: PE file [%s] section table malloc failed bytes = %d\n",
640                                         fname, section_table_bytes));
641                         goto error_exit;
642                 }
643
644                 if ((byte_count = vfs_read_data(fsp, buf, section_table_bytes)) < section_table_bytes) {
645                         DEBUG(3,("get_file_version: PE file [%s] Section header too short, bytes read = %d\n",
646                                         fname, byte_count));
647                         goto error_exit;
648                 }
649
650                 /* Iterate the section table looking for the resource section ".rsrc" */
651                 for (i = 0; i < num_sections; i++) {
652                         int sec_offset = i * PE_HEADER_SECT_HEADER_SIZE;
653
654                         if (strcmp(".rsrc", &buf[sec_offset+PE_HEADER_SECT_NAME_OFFSET]) == 0) {
655                                 int section_pos   = IVAL(buf,sec_offset+PE_HEADER_SECT_PTR_DATA_OFFSET);
656                                 int section_bytes = IVAL(buf,sec_offset+PE_HEADER_SECT_SIZE_DATA_OFFSET);
657
658                                 SAFE_FREE(buf);
659                                 if ((buf=malloc(section_bytes)) == NULL) {
660                                         DEBUG(0,("get_file_version: PE file [%s] version malloc failed bytes = %d\n",
661                                                         fname, section_bytes));
662                                         goto error_exit;
663                                 }
664
665                                 /* Seek to the start of the .rsrc section info */
666                                 if (fsp->conn->vfs_ops.lseek(fsp, fsp->fd, section_pos, SEEK_SET) == (SMB_OFF_T)-1) {
667                                         DEBUG(3,("get_file_version: PE file [%s] too short for section info, errno = %d\n",
668                                                         fname, errno));
669                                         goto error_exit;
670                                 }
671
672                                 if ((byte_count = vfs_read_data(fsp, buf, section_bytes)) < section_bytes) {
673                                         DEBUG(3,("get_file_version: PE file [%s] .rsrc section too short, bytes read = %d\n",
674                                                         fname, byte_count));
675                                         goto error_exit;
676                                 }
677
678                                 for (i=0; i<section_bytes-VS_VERSION_INFO_UNICODE_SIZE; i++) {
679                                         /* Scan for 1st 3 unicoded bytes followed by word aligned magic value */
680                                         if (buf[i] == 'V' && buf[i+1] == '\0' && buf[i+2] == 'S') {
681                                                 /* Align to next long address */
682                                                 int pos = (i + sizeof(VS_SIGNATURE)*2 + 3) & 0xfffffffc;
683
684                                                 if (IVAL(buf,pos) == VS_MAGIC_VALUE) {
685                                                         *major = IVAL(buf,pos+VS_MAJOR_OFFSET);
686                                                         *minor = IVAL(buf,pos+VS_MINOR_OFFSET);
687                                                         
688                                                         DEBUG(6,("get_file_version: PE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
689                                                                           fname, *major, *minor,
690                                                                           (*major>>16)&0xffff, *major&0xffff,
691                                                                           (*minor>>16)&0xffff, *minor&0xffff));
692                                                         SAFE_FREE(buf);
693                                                         return 1;
694                                                 }
695                                         }
696                                 }
697                         }
698                 }
699
700                 /* Version info not found, fall back to origin date/time */
701                 DEBUG(10,("get_file_version: PE file [%s] has no version info\n", fname));
702                 SAFE_FREE(buf);
703                 return 0;
704
705         } else if (SVAL(buf,NE_HEADER_SIGNATURE_OFFSET) == NE_HEADER_SIGNATURE) {
706                 if (CVAL(buf,NE_HEADER_TARGET_OS_OFFSET) != NE_HEADER_TARGOS_WIN ) {
707                         DEBUG(3,("get_file_version: NE file [%s] wrong target OS = 0x%x\n",
708                                         fname, CVAL(buf,NE_HEADER_TARGET_OS_OFFSET)));
709                         /* At this point, we assume the file is in error. It still could be somthing
710                          * else besides a NE file, but it unlikely at this point. */
711                         goto error_exit;
712                 }
713
714                 /* Allocate a bit more space to speed up things */
715                 SAFE_FREE(buf);
716                 if ((buf=malloc(VS_NE_BUF_SIZE)) == NULL) {
717                         DEBUG(0,("get_file_version: NE file [%s] malloc failed bytes  = %d\n",
718                                         fname, PE_HEADER_SIZE));
719                         goto error_exit;
720                 }
721
722                 /* This is a HACK! I got tired of trying to sort through the messy
723                  * 'NE' file format. If anyone wants to clean this up please have at
724                  * it, but this works. 'NE' files will eventually fade away. JRR */
725                 while((byte_count = vfs_read_data(fsp, buf, VS_NE_BUF_SIZE)) > 0) {
726                         /* Cover case that should not occur in a well formed 'NE' .dll file */
727                         if (byte_count-VS_VERSION_INFO_SIZE <= 0) break;
728
729                         for(i=0; i<byte_count; i++) {
730                                 /* Fast skip past data that can't possibly match */
731                                 if (buf[i] != 'V') continue;
732
733                                 /* Potential match data crosses buf boundry, move it to beginning
734                                  * of buf, and fill the buf with as much as it will hold. */
735                                 if (i>byte_count-VS_VERSION_INFO_SIZE) {
736                                         int bc;
737
738                                         memcpy(buf, &buf[i], byte_count-i);
739                                         if ((bc = vfs_read_data(fsp, &buf[byte_count-i], VS_NE_BUF_SIZE-
740                                                                    (byte_count-i))) < 0) {
741
742                                                 DEBUG(0,("get_file_version: NE file [%s] Read error, errno=%d\n",
743                                                                  fname, errno));
744                                                 goto error_exit;
745                                         }
746
747                                         byte_count = bc + (byte_count - i);
748                                         if (byte_count<VS_VERSION_INFO_SIZE) break;
749
750                                         i = 0;
751                                 }
752
753                                 /* Check that the full signature string and the magic number that
754                                  * follows exist (not a perfect solution, but the chances that this
755                                  * occurs in code is, well, remote. Yes I know I'm comparing the 'V'
756                                  * twice, as it is simpler to read the code. */
757                                 if (strcmp(&buf[i], VS_SIGNATURE) == 0) {
758                                         /* Compute skip alignment to next long address */
759                                         int skip = -(fsp->conn->vfs_ops.lseek(fsp, fsp->fd, 0, SEEK_CUR) - (byte_count - i) +
760                                                                  sizeof(VS_SIGNATURE)) & 3;
761                                         if (IVAL(buf,i+sizeof(VS_SIGNATURE)+skip) != 0xfeef04bd) continue;
762
763                                         *major = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MAJOR_OFFSET);
764                                         *minor = IVAL(buf,i+sizeof(VS_SIGNATURE)+skip+VS_MINOR_OFFSET);
765                                         DEBUG(6,("get_file_version: NE file [%s] Version = %08x:%08x (%d.%d.%d.%d)\n",
766                                                           fname, *major, *minor,
767                                                           (*major>>16)&0xffff, *major&0xffff,
768                                                           (*minor>>16)&0xffff, *minor&0xffff));
769                                         SAFE_FREE(buf);
770                                         return 1;
771                                 }
772                         }
773                 }
774
775                 /* Version info not found, fall back to origin date/time */
776                 DEBUG(0,("get_file_version: NE file [%s] Version info not found\n", fname));
777                 SAFE_FREE(buf);
778                 return 0;
779
780         } else
781                 /* Assume this isn't an error... the file just looks sort of like a PE/NE file */
782                 DEBUG(3,("get_file_version: File [%s] unknown file format, signature = 0x%x\n",
783                                 fname, IVAL(buf,PE_HEADER_SIGNATURE_OFFSET)));
784
785         no_version_info:
786                 SAFE_FREE(buf);
787                 return 0;
788
789         error_exit:
790                 SAFE_FREE(buf);
791                 return -1;
792 }
793
794 /****************************************************************************
795 Drivers for Microsoft systems contain multiple files. Often, multiple drivers
796 share one or more files. During the MS installation process files are checked
797 to insure that only a newer version of a shared file is installed over an
798 older version. There are several possibilities for this comparison. If there
799 is no previous version, the new one is newer (obviously). If either file is
800 missing the version info structure, compare the creation date (on Unix use
801 the modification date). Otherwise chose the numerically larger version number.
802 ****************************************************************************/
803 static int file_version_is_newer(connection_struct *conn, fstring new_file,
804                                                                 fstring old_file)
805 {
806         BOOL   use_version = True;
807         pstring filepath;
808
809         uint32 new_major;
810         uint32 new_minor;
811         time_t new_create_time;
812
813         uint32 old_major;
814         uint32 old_minor;
815         time_t old_create_time;
816
817         int access_mode;
818         int action;
819         files_struct    *fsp = NULL;
820         SMB_STRUCT_STAT st;
821         SMB_STRUCT_STAT stat_buf;
822         BOOL bad_path;
823
824         ZERO_STRUCT(st);
825         ZERO_STRUCT(stat_buf);
826         new_create_time = (time_t)0;
827         old_create_time = (time_t)0;
828
829         /* Get file version info (if available) for previous file (if it exists) */
830         pstrcpy(filepath, old_file);
831
832         unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
833
834         fsp = open_file_shared(conn, filepath, &stat_buf,
835                                                    SET_OPEN_MODE(DOS_OPEN_RDONLY),
836                                                    (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
837                                                    0, 0, &access_mode, &action);
838         if (!fsp) {
839                 /* Old file not found, so by definition new file is in fact newer */
840                 DEBUG(10,("file_version_is_newer: Can't open old file [%s], errno = %d\n",
841                                 filepath, errno));
842                 return True;
843
844         } else {
845                 int ret = get_file_version(fsp, old_file, &old_major, &old_minor);
846                 if (ret == -1) goto error_exit;
847
848                 if (!ret) {
849                         DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
850                                          old_file));
851                         use_version = False;
852                         if (fsp->conn->vfs_ops.fstat(fsp, fsp->fd, &st) == -1) goto error_exit;
853                         old_create_time = st.st_mtime;
854                         DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", old_create_time));
855                 }
856         }
857         close_file(fsp, True);
858
859         /* Get file version info (if available) for new file */
860         pstrcpy(filepath, new_file);
861         unix_convert(filepath,conn,NULL,&bad_path,&stat_buf);
862
863         fsp = open_file_shared(conn, filepath, &stat_buf,
864                                                    SET_OPEN_MODE(DOS_OPEN_RDONLY),
865                                                    (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
866                                                    0, 0, &access_mode, &action);
867         if (!fsp) {
868                 /* New file not found, this shouldn't occur if the caller did its job */
869                 DEBUG(3,("file_version_is_newer: Can't open new file [%s], errno = %d\n",
870                                 filepath, errno));
871                 goto error_exit;
872
873         } else {
874                 int ret = get_file_version(fsp, new_file, &new_major, &new_minor);
875                 if (ret == -1) goto error_exit;
876
877                 if (!ret) {
878                         DEBUG(6,("file_version_is_newer: Version info not found [%s], use mod time\n",
879                                          new_file));
880                         use_version = False;
881                         if (fsp->conn->vfs_ops.fstat(fsp, fsp->fd, &st) == -1) goto error_exit;
882                         new_create_time = st.st_mtime;
883                         DEBUGADD(6,("file_version_is_newer: mod time = %ld sec\n", new_create_time));
884                 }
885         }
886         close_file(fsp, True);
887
888         if (use_version) {
889                 /* Compare versions and choose the larger version number */
890                 if (new_major > old_major ||
891                         (new_major == old_major && new_minor > old_minor)) {
892                         
893                         DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
894                         return True;
895                 }
896                 else {
897                         DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
898                         return False;
899                 }
900
901         } else {
902                 /* Compare modification time/dates and choose the newest time/date */
903                 if (new_create_time > old_create_time) {
904                         DEBUG(6,("file_version_is_newer: Replacing [%s] with [%s]\n", old_file, new_file));
905                         return True;
906                 }
907                 else {
908                         DEBUG(6,("file_version_is_newer: Leaving [%s] unchanged\n", old_file));
909                         return False;
910                 }
911         }
912
913         error_exit:
914                 if(fsp)
915                         close_file(fsp, True);
916                 return -1;
917 }
918
919 /****************************************************************************
920 Determine the correct cVersion associated with an architecture and driver
921 ****************************************************************************/
922 static uint32 get_correct_cversion(fstring architecture, fstring driverpath_in,
923                                    struct current_user *user, WERROR *perr)
924 {
925         int               cversion;
926         int               access_mode;
927         int               action;
928         NTSTATUS          nt_status;
929         pstring           driverpath;
930         DATA_BLOB         null_pw;
931         files_struct      *fsp = NULL;
932         BOOL              bad_path;
933         SMB_STRUCT_STAT   st;
934         connection_struct *conn;
935
936         ZERO_STRUCT(st);
937
938         *perr = WERR_INVALID_PARAM;
939
940         /* If architecture is Windows 95/98/ME, the version is always 0. */
941         if (strcmp(architecture, "WIN40") == 0) {
942                 DEBUG(10,("get_correct_cversion: Driver is Win9x, cversion = 0\n"));
943                 *perr = WERR_OK;
944                 return 0;
945         }
946
947         /* connect to the print$ share under the same account as the user connected to the rpc pipe */  
948         /* Null password is ok - we are already an authenticated user... */
949         null_pw = data_blob(NULL, 0);
950
951         become_root();
952         conn = make_connection("print$", null_pw, "A:", user->vuid, &nt_status);
953         unbecome_root();
954
955         if (conn == NULL) {
956                 DEBUG(0,("get_correct_cversion: Unable to connect\n"));
957                 *perr = ntstatus_to_werror(nt_status);
958                 return -1;
959         }
960
961         /* We are temporarily becoming the connection user. */
962         if (!become_user(conn, conn->vuid)) {
963                 DEBUG(0,("get_correct_cversion: Can't become user!\n"));
964                 *perr = WERR_ACCESS_DENIED;
965                 return -1;
966         }
967
968         /* Open the driver file (Portable Executable format) and determine the
969          * deriver the cversion. */
970         slprintf(driverpath, sizeof(driverpath)-1, "%s/%s", architecture, driverpath_in);
971
972         unix_convert(driverpath,conn,NULL,&bad_path,&st);
973
974         fsp = open_file_shared(conn, driverpath, &st,
975                                                    SET_OPEN_MODE(DOS_OPEN_RDONLY),
976                                                    (FILE_FAIL_IF_NOT_EXIST|FILE_EXISTS_OPEN),
977                                                    0, 0, &access_mode, &action);
978         if (!fsp) {
979                 DEBUG(3,("get_correct_cversion: Can't open file [%s], errno = %d\n",
980                                 driverpath, errno));
981                 *perr = WERR_ACCESS_DENIED;
982                 goto error_exit;
983         }
984         else {
985                 uint32 major;
986                 uint32 minor;
987                 int    ret = get_file_version(fsp, driverpath, &major, &minor);
988                 if (ret == -1) goto error_exit;
989
990                 if (!ret) {
991                         DEBUG(6,("get_correct_cversion: Version info not found [%s]\n", driverpath));
992                         goto error_exit;
993                 }
994
995                 /*
996                  * This is a Microsoft'ism. See references in MSDN to VER_FILEVERSION
997                  * for more details. Version in this case is not just the version of the 
998                  * file, but the version in the sense of kernal mode (2) vs. user mode
999                  * (3) drivers. Other bits of the version fields are the version info. 
1000                  * JRR 010716
1001                 */
1002                 cversion = major & 0x0000ffff;
1003                 switch (cversion) {
1004                         case 2: /* WinNT drivers */
1005                         case 3: /* Win2K drivers */
1006                                 break;
1007                         
1008                         default:
1009                                 DEBUG(6,("get_correct_cversion: cversion invalid [%s]  cversion = %d\n", 
1010                                         driverpath, cversion));
1011                                 goto error_exit;
1012                 }
1013
1014                 DEBUG(10,("get_correct_cversion: Version info found [%s]  major = 0x%x  minor = 0x%x\n",
1015                                   driverpath, major, minor));
1016         }
1017
1018         DEBUG(10,("get_correct_cversion: Driver file [%s] cversion = %d\n",
1019                   driverpath, cversion));
1020
1021         close_file(fsp, True);
1022         close_cnum(conn, user->vuid);
1023         unbecome_user();
1024         *perr = WERR_OK;
1025         return cversion;
1026
1027   error_exit:
1028
1029         if(fsp)
1030                 close_file(fsp, True);
1031         
1032         close_cnum(conn, user->vuid);
1033         unbecome_user();
1034         return -1;
1035 }
1036
1037 /****************************************************************************
1038 ****************************************************************************/
1039 static WERROR clean_up_driver_struct_level_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver,
1040                                              struct current_user *user)
1041 {
1042         fstring architecture;
1043         fstring new_name;
1044         char *p;
1045         int i;
1046         WERROR err;
1047
1048         /* clean up the driver name.
1049          * we can get .\driver.dll
1050          * or worse c:\windows\system\driver.dll !
1051          */
1052         /* using an intermediate string to not have overlaping memcpy()'s */
1053         if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1054                 fstrcpy(new_name, p+1);
1055                 fstrcpy(driver->driverpath, new_name);
1056         }
1057
1058         if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1059                 fstrcpy(new_name, p+1);
1060                 fstrcpy(driver->datafile, new_name);
1061         }
1062
1063         if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1064                 fstrcpy(new_name, p+1);
1065                 fstrcpy(driver->configfile, new_name);
1066         }
1067
1068         if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1069                 fstrcpy(new_name, p+1);
1070                 fstrcpy(driver->helpfile, new_name);
1071         }
1072
1073         if (driver->dependentfiles) {
1074                 for (i=0; *driver->dependentfiles[i]; i++) {
1075                         if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1076                                 fstrcpy(new_name, p+1);
1077                                 fstrcpy(driver->dependentfiles[i], new_name);
1078                         }
1079                 }
1080         }
1081
1082         get_short_archi(architecture, driver->environment);
1083         
1084         /* jfm:7/16/2000 the client always sends the cversion=0.
1085          * The server should check which version the driver is by reading
1086          * the PE header of driver->driverpath.
1087          *
1088          * For Windows 95/98 the version is 0 (so the value sent is correct)
1089          * For Windows NT (the architecture doesn't matter)
1090          *      NT 3.1: cversion=0
1091          *      NT 3.5/3.51: cversion=1
1092          *      NT 4: cversion=2
1093          *      NT2K: cversion=3
1094          */
1095         if ((driver->cversion = get_correct_cversion( architecture,
1096                                                       driver->driverpath, user, &err)) == -1)
1097                 return err;
1098
1099         return WERR_OK;
1100 }
1101         
1102 /****************************************************************************
1103 ****************************************************************************/
1104 static WERROR clean_up_driver_struct_level_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver,
1105                                              struct current_user *user)
1106 {
1107         fstring architecture;
1108         fstring new_name;
1109         char *p;
1110         int i;
1111         WERROR err;
1112
1113         /* clean up the driver name.
1114          * we can get .\driver.dll
1115          * or worse c:\windows\system\driver.dll !
1116          */
1117         /* using an intermediate string to not have overlaping memcpy()'s */
1118         if ((p = strrchr(driver->driverpath,'\\')) != NULL) {
1119                 fstrcpy(new_name, p+1);
1120                 fstrcpy(driver->driverpath, new_name);
1121         }
1122
1123         if ((p = strrchr(driver->datafile,'\\')) != NULL) {
1124                 fstrcpy(new_name, p+1);
1125                 fstrcpy(driver->datafile, new_name);
1126         }
1127
1128         if ((p = strrchr(driver->configfile,'\\')) != NULL) {
1129                 fstrcpy(new_name, p+1);
1130                 fstrcpy(driver->configfile, new_name);
1131         }
1132
1133         if ((p = strrchr(driver->helpfile,'\\')) != NULL) {
1134                 fstrcpy(new_name, p+1);
1135                 fstrcpy(driver->helpfile, new_name);
1136         }
1137
1138         if (driver->dependentfiles) {
1139                 for (i=0; *driver->dependentfiles[i]; i++) {
1140                         if ((p = strrchr(driver->dependentfiles[i],'\\')) != NULL) {
1141                                 fstrcpy(new_name, p+1);
1142                                 fstrcpy(driver->dependentfiles[i], new_name);
1143                         }
1144                 }
1145         }
1146
1147         get_short_archi(architecture, driver->environment);
1148
1149         /* jfm:7/16/2000 the client always sends the cversion=0.
1150          * The server should check which version the driver is by reading
1151          * the PE header of driver->driverpath.
1152          *
1153          * For Windows 95/98 the version is 0 (so the value sent is correct)
1154          * For Windows NT (the architecture doesn't matter)
1155          *      NT 3.1: cversion=0
1156          *      NT 3.5/3.51: cversion=1
1157          *      NT 4: cversion=2
1158          *      NT2K: cversion=3
1159          */
1160         if ((driver->version = get_correct_cversion(architecture,
1161                                                     driver->driverpath, user, &err)) == -1)
1162                 return err;
1163
1164         return WERR_OK;
1165 }
1166
1167 /****************************************************************************
1168 ****************************************************************************/
1169 WERROR clean_up_driver_struct(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract,
1170                                                           uint32 level, struct current_user *user)
1171 {
1172         switch (level) {
1173                 case 3:
1174                 {
1175                         NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1176                         driver=driver_abstract.info_3;
1177                         return clean_up_driver_struct_level_3(driver, user);
1178                 }
1179                 case 6:
1180                 {
1181                         NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver;
1182                         driver=driver_abstract.info_6;
1183                         return clean_up_driver_struct_level_6(driver, user);
1184                 }
1185                 default:
1186                         return WERR_INVALID_PARAM;
1187         }
1188 }
1189
1190 /****************************************************************************
1191  This function sucks and should be replaced. JRA.
1192 ****************************************************************************/
1193
1194 static void convert_level_6_to_level3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *dst, NT_PRINTER_DRIVER_INFO_LEVEL_6 *src)
1195 {
1196     dst->cversion  = src->version;
1197
1198     fstrcpy( dst->name, src->name);
1199     fstrcpy( dst->environment, src->environment);
1200     fstrcpy( dst->driverpath, src->driverpath);
1201     fstrcpy( dst->datafile, src->datafile);
1202     fstrcpy( dst->configfile, src->configfile);
1203     fstrcpy( dst->helpfile, src->helpfile);
1204     fstrcpy( dst->monitorname, src->monitorname);
1205     fstrcpy( dst->defaultdatatype, src->defaultdatatype);
1206     dst->dependentfiles = src->dependentfiles;
1207 }
1208
1209 #if 0 /* Debugging function */
1210
1211 static char* ffmt(unsigned char *c){
1212         int i;
1213         static char ffmt_str[17];
1214
1215         for (i=0; i<16; i++) {
1216                 if ((c[i] < ' ') || (c[i] > '~'))
1217                         ffmt_str[i]='.';
1218                 else
1219                         ffmt_str[i]=c[i];
1220         }
1221     ffmt_str[16]='\0';
1222         return ffmt_str;
1223 }
1224
1225 #endif
1226
1227 /****************************************************************************
1228 ****************************************************************************/
1229 BOOL move_driver_to_download_area(NT_PRINTER_DRIVER_INFO_LEVEL driver_abstract, uint32 level, 
1230                                   struct current_user *user, WERROR *perr)
1231 {
1232         NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver;
1233         NT_PRINTER_DRIVER_INFO_LEVEL_3 converted_driver;
1234         fstring architecture;
1235         pstring new_dir;
1236         pstring old_name;
1237         pstring new_name;
1238         DATA_BLOB null_pw;
1239         connection_struct *conn;
1240         NTSTATUS nt_status;
1241         int ver = 0;
1242         int i;
1243
1244         *perr = WERR_OK;
1245
1246         if (level==3)
1247                 driver=driver_abstract.info_3;
1248         else if (level==6) {
1249                 convert_level_6_to_level3(&converted_driver, driver_abstract.info_6);
1250                 driver = &converted_driver;
1251         } else {
1252                 DEBUG(0,("move_driver_to_download_area: Unknown info level (%u)\n", (unsigned int)level ));
1253                 return False;
1254         }
1255
1256         get_short_archi(architecture, driver->environment);
1257
1258         /* connect to the print$ share under the same account as the user connected to the rpc pipe */  
1259         /* Null password is ok - we are already an authenticated user... */
1260         null_pw = data_blob(NULL, 0);
1261         conn = make_connection("print$", null_pw, "A:", user->vuid, &nt_status);
1262
1263         if (conn == NULL) {
1264                 DEBUG(0,("move_driver_to_download_area: Unable to connect\n"));
1265                 *perr = ntstatus_to_werror(nt_status);
1266                 return False;
1267         }
1268
1269         /*
1270          * Save who we are - we are temporarily becoming the connection user.
1271          */
1272
1273         push_sec_ctx();
1274
1275         if (!become_user(conn, conn->vuid)) {
1276                 DEBUG(0,("move_driver_to_download_area: Can't become user!\n"));
1277                 pop_sec_ctx();
1278                 return False;
1279         }
1280
1281         /*
1282          * make the directories version and version\driver_name
1283          * under the architecture directory.
1284          */
1285         DEBUG(5,("Creating first directory\n"));
1286         slprintf(new_dir, sizeof(new_dir)-1, "%s/%d", architecture, driver->cversion);
1287         mkdir_internal(conn, new_dir);
1288
1289         /* For each driver file, archi\filexxx.yyy, if there is a duplicate file
1290          * listed for this driver which has already been moved, skip it (note:
1291          * drivers may list the same file name several times. Then check if the
1292          * file already exists in archi\cversion\, if so, check that the version
1293          * info (or time stamps if version info is unavailable) is newer (or the
1294          * date is later). If it is, move it to archi\cversion\filexxx.yyy.
1295          * Otherwise, delete the file.
1296          *
1297          * If a file is not moved to archi\cversion\ because of an error, all the
1298          * rest of the 'unmoved' driver files are removed from archi\. If one or
1299          * more of the driver's files was already moved to archi\cversion\, it
1300          * potentially leaves the driver in a partially updated state. Version
1301          * trauma will most likely occur if an client attempts to use any printer
1302          * bound to the driver. Perhaps a rewrite to make sure the moves can be
1303          * done is appropriate... later JRR
1304          */
1305
1306         DEBUG(5,("Moving files now !\n"));
1307
1308         if (driver->driverpath && strlen(driver->driverpath)) {
1309                 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->driverpath);      
1310                 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->driverpath);   
1311                 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1312                         NTSTATUS status;
1313                         status = rename_internals(conn, new_name, old_name, True);
1314                         if (!NT_STATUS_IS_OK(status)) {
1315                                 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1316                                                 new_name, old_name));
1317                                 *perr = ntstatus_to_werror(status);
1318                                 unlink_internals(conn, 0, new_name);
1319                                 ver = -1;
1320                         }
1321                 }
1322                 else
1323                         unlink_internals(conn, 0, new_name);
1324         }
1325
1326         if (driver->datafile && strlen(driver->datafile)) {
1327                 if (!strequal(driver->datafile, driver->driverpath)) {
1328                         slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->datafile);        
1329                         slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->datafile);     
1330                         if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1331                                 NTSTATUS status;
1332                                 status = rename_internals(conn, new_name, old_name, True);
1333                                 if (!NT_STATUS_IS_OK(status)) {
1334                                         DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1335                                                         new_name, old_name));
1336                                         *perr = ntstatus_to_werror(status);
1337                                         unlink_internals(conn, 0, new_name);
1338                                         ver = -1;
1339                                 }
1340                         }
1341                         else
1342                                 unlink_internals(conn, 0, new_name);
1343                 }
1344         }
1345
1346         if (driver->configfile && strlen(driver->configfile)) {
1347                 if (!strequal(driver->configfile, driver->driverpath) &&
1348                         !strequal(driver->configfile, driver->datafile)) {
1349                         slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->configfile);      
1350                         slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->configfile);   
1351                         if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1352                                 NTSTATUS status;
1353                                 status = rename_internals(conn, new_name, old_name, True);
1354                                 if (!NT_STATUS_IS_OK(status)) {
1355                                         DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1356                                                         new_name, old_name));
1357                                         *perr = ntstatus_to_werror(status);
1358                                         unlink_internals(conn, 0, new_name);
1359                                         ver = -1;
1360                                 }
1361                         }
1362                         else
1363                                 unlink_internals(conn, 0, new_name);
1364                 }
1365         }
1366
1367         if (driver->helpfile && strlen(driver->helpfile)) {
1368                 if (!strequal(driver->helpfile, driver->driverpath) &&
1369                         !strequal(driver->helpfile, driver->datafile) &&
1370                         !strequal(driver->helpfile, driver->configfile)) {
1371                         slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->helpfile);        
1372                         slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->helpfile);     
1373                         if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1374                                 NTSTATUS status;
1375                                 status = rename_internals(conn, new_name, old_name, True);
1376                                 if (!NT_STATUS_IS_OK(status)) {
1377                                         DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1378                                                         new_name, old_name));
1379                                         *perr = ntstatus_to_werror(status);
1380                                         unlink_internals(conn, 0, new_name);
1381                                         ver = -1;
1382                                 }
1383                         }
1384                         else
1385                                 unlink_internals(conn, 0, new_name);
1386                 }
1387         }
1388
1389         if (driver->dependentfiles) {
1390                 for (i=0; *driver->dependentfiles[i]; i++) {
1391                         if (!strequal(driver->dependentfiles[i], driver->driverpath) &&
1392                                 !strequal(driver->dependentfiles[i], driver->datafile) &&
1393                                 !strequal(driver->dependentfiles[i], driver->configfile) &&
1394                                 !strequal(driver->dependentfiles[i], driver->helpfile)) {
1395                                 int j;
1396                                 for (j=0; j < i; j++) {
1397                                         if (strequal(driver->dependentfiles[i], driver->dependentfiles[j])) {
1398                                                 goto NextDriver;
1399                                         }
1400                                 }
1401
1402                                 slprintf(new_name, sizeof(new_name)-1, "%s/%s", architecture, driver->dependentfiles[i]);       
1403                                 slprintf(old_name, sizeof(old_name)-1, "%s/%s", new_dir, driver->dependentfiles[i]);    
1404                                 if (ver != -1 && (ver=file_version_is_newer(conn, new_name, old_name)) > 0) {
1405                                         NTSTATUS status;
1406                                         status = rename_internals(conn, new_name, old_name, True);
1407                                         if (!NT_STATUS_IS_OK(status)) {
1408                                                 DEBUG(0,("move_driver_to_download_area: Unable to rename [%s] to [%s]\n",
1409                                                                 new_name, old_name));
1410                                                 *perr = ntstatus_to_werror(status);
1411                                                 unlink_internals(conn, 0, new_name);
1412                                                 ver = -1;
1413                                         }
1414                                 }
1415                                 else
1416                                         unlink_internals(conn, 0, new_name);
1417                         }
1418                 NextDriver: ;
1419                 }
1420         }
1421
1422         close_cnum(conn, user->vuid);
1423         pop_sec_ctx();
1424
1425         return ver == -1 ? False : True;
1426 }
1427
1428 /****************************************************************************
1429 ****************************************************************************/
1430 static uint32 add_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 *driver)
1431 {
1432         int len, buflen;
1433         fstring architecture;
1434         pstring directory;
1435         pstring temp_name;
1436         pstring key;
1437         char *buf;
1438         int i, ret;
1439         TDB_DATA kbuf, dbuf;
1440
1441         get_short_archi(architecture, driver->environment);
1442
1443         /* The names are relative. We store them in the form: \print$\arch\version\driver.xxx
1444          * \\server is added in the rpc server layer.
1445          * It does make sense to NOT store the server's name in the printer TDB.
1446          */
1447
1448         slprintf(directory, sizeof(directory)-1, "\\print$\\%s\\%d\\", architecture, driver->cversion);
1449
1450     /* .inf files do not always list a file for each of the four standard files. 
1451      * Don't prepend a path to a null filename, or client claims:
1452      *   "The server on which the printer resides does not have a suitable 
1453      *   <printer driver name> printer driver installed. Click OK if you 
1454      *   wish to install the driver on your local machine."
1455      */
1456         if (strlen(driver->driverpath)) {
1457         fstrcpy(temp_name, driver->driverpath);
1458         slprintf(driver->driverpath, sizeof(driver->driverpath)-1, "%s%s", directory, temp_name);
1459     }
1460
1461         if (strlen(driver->datafile)) {
1462         fstrcpy(temp_name, driver->datafile);
1463         slprintf(driver->datafile, sizeof(driver->datafile)-1, "%s%s", directory, temp_name);
1464     }
1465
1466         if (strlen(driver->configfile)) {
1467         fstrcpy(temp_name, driver->configfile);
1468         slprintf(driver->configfile, sizeof(driver->configfile)-1, "%s%s", directory, temp_name);
1469     }
1470
1471         if (strlen(driver->helpfile)) {
1472         fstrcpy(temp_name, driver->helpfile);
1473         slprintf(driver->helpfile, sizeof(driver->helpfile)-1, "%s%s", directory, temp_name);
1474     }
1475
1476         if (driver->dependentfiles) {
1477                 for (i=0; *driver->dependentfiles[i]; i++) {
1478             fstrcpy(temp_name, driver->dependentfiles[i]);
1479             slprintf(driver->dependentfiles[i], sizeof(driver->dependentfiles[i])-1, "%s%s", directory, temp_name);
1480                 }
1481         }
1482
1483         slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, driver->cversion, driver->name);
1484
1485         DEBUG(5,("add_a_printer_driver_3: Adding driver with key %s\n", key ));
1486
1487         buf = NULL;
1488         len = buflen = 0;
1489
1490  again:
1491         len = 0;
1492         len += tdb_pack(buf+len, buflen-len, "dffffffff",
1493                         driver->cversion,
1494                         driver->name,
1495                         driver->environment,
1496                         driver->driverpath,
1497                         driver->datafile,
1498                         driver->configfile,
1499                         driver->helpfile,
1500                         driver->monitorname,
1501                         driver->defaultdatatype);
1502
1503         if (driver->dependentfiles) {
1504                 for (i=0; *driver->dependentfiles[i]; i++) {
1505                         len += tdb_pack(buf+len, buflen-len, "f",
1506                                         driver->dependentfiles[i]);
1507                 }
1508         }
1509
1510         if (len != buflen) {
1511                 char *tb;
1512                 
1513                 tb = (char *)Realloc(buf, len);
1514                 if (!tb) {
1515                         DEBUG(0,("add_a_printer_driver_3: failed to enlarge buffer\n!"));
1516                         ret = -1;
1517                         goto done;
1518                 }
1519                 else buf = tb;
1520                 buflen = len;
1521                 goto again;
1522         }
1523
1524
1525         kbuf.dptr = key;
1526         kbuf.dsize = strlen(key)+1;
1527         dbuf.dptr = buf;
1528         dbuf.dsize = len;
1529         
1530         ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
1531
1532 done:
1533         if (ret)
1534                 DEBUG(0,("add_a_printer_driver_3: Adding driver with key %s failed.\n", key ));
1535
1536         SAFE_FREE(buf);
1537         return ret;
1538 }
1539
1540 /****************************************************************************
1541 ****************************************************************************/
1542 static uint32 add_a_printer_driver_6(NT_PRINTER_DRIVER_INFO_LEVEL_6 *driver)
1543 {
1544         NT_PRINTER_DRIVER_INFO_LEVEL_3 info3;
1545
1546         ZERO_STRUCT(info3);
1547         info3.cversion = driver->version;
1548         fstrcpy(info3.name,driver->name);
1549         fstrcpy(info3.environment,driver->environment);
1550         fstrcpy(info3.driverpath,driver->driverpath);
1551         fstrcpy(info3.datafile,driver->datafile);
1552         fstrcpy(info3.configfile,driver->configfile);
1553         fstrcpy(info3.helpfile,driver->helpfile);
1554         fstrcpy(info3.monitorname,driver->monitorname);
1555         fstrcpy(info3.defaultdatatype,driver->defaultdatatype);
1556         info3.dependentfiles = driver->dependentfiles;
1557
1558         return add_a_printer_driver_3(&info3);
1559 }
1560
1561
1562 /****************************************************************************
1563 ****************************************************************************/
1564 static WERROR get_a_printer_driver_3_default(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring in_prt, fstring in_arch)
1565 {
1566         NT_PRINTER_DRIVER_INFO_LEVEL_3 info;
1567
1568         ZERO_STRUCT(info);
1569
1570         fstrcpy(info.name, in_prt);
1571         fstrcpy(info.defaultdatatype, "RAW");
1572         
1573         fstrcpy(info.driverpath, "");
1574         fstrcpy(info.datafile, "");
1575         fstrcpy(info.configfile, "");
1576         fstrcpy(info.helpfile, "");
1577
1578         if ((info.dependentfiles=(fstring *)malloc(2*sizeof(fstring))) == NULL)
1579                 return WERR_NOMEM;
1580
1581         memset(info.dependentfiles, '\0', 2*sizeof(fstring));
1582         fstrcpy(info.dependentfiles[0], "");
1583
1584         *info_ptr = memdup(&info, sizeof(info));
1585         
1586         return WERR_OK;
1587 }
1588
1589 /****************************************************************************
1590 ****************************************************************************/
1591 static WERROR get_a_printer_driver_3(NT_PRINTER_DRIVER_INFO_LEVEL_3 **info_ptr, fstring in_prt, fstring in_arch, uint32 version)
1592 {
1593         NT_PRINTER_DRIVER_INFO_LEVEL_3 driver;
1594         TDB_DATA kbuf, dbuf;
1595         fstring architecture;
1596         int len = 0;
1597         int i;
1598         pstring key;
1599
1600         ZERO_STRUCT(driver);
1601
1602         get_short_archi(architecture, in_arch);
1603
1604         DEBUG(8,("get_a_printer_driver_3: [%s%s/%d/%s]\n", DRIVERS_PREFIX, architecture, version, in_prt));
1605
1606         slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, architecture, version, in_prt);
1607
1608         kbuf.dptr = key;
1609         kbuf.dsize = strlen(key)+1;
1610         
1611         dbuf = tdb_fetch(tdb_drivers, kbuf);
1612 #if 0
1613         if (!dbuf.dptr) return get_a_printer_driver_3_default(info_ptr, in_prt, in_arch);
1614 #else
1615         if (!dbuf.dptr) return WERR_ACCESS_DENIED;
1616 #endif
1617         len += tdb_unpack(dbuf.dptr, dbuf.dsize, "dffffffff",
1618                           &driver.cversion,
1619                           driver.name,
1620                           driver.environment,
1621                           driver.driverpath,
1622                           driver.datafile,
1623                           driver.configfile,
1624                           driver.helpfile,
1625                           driver.monitorname,
1626                           driver.defaultdatatype);
1627
1628         i=0;
1629         while (len < dbuf.dsize) {
1630                 fstring *tddfs;
1631         
1632                 tddfs = (fstring *)Realloc(driver.dependentfiles,
1633                                                          sizeof(fstring)*(i+2));
1634                 if (tddfs == NULL) {
1635                         DEBUG(0,("get_a_printer_driver_3: failed to enlarge buffer!\n"));
1636                         break;
1637                 }
1638                 else driver.dependentfiles = tddfs;
1639
1640                 len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "f",
1641                                   &driver.dependentfiles[i]);
1642                 i++;
1643         }
1644         if (driver.dependentfiles != NULL)
1645                 fstrcpy(driver.dependentfiles[i], "");
1646
1647         SAFE_FREE(dbuf.dptr);
1648
1649         if (len != dbuf.dsize) {
1650                 SAFE_FREE(driver.dependentfiles);
1651
1652                 return get_a_printer_driver_3_default(info_ptr, in_prt, in_arch);
1653         }
1654
1655         *info_ptr = (NT_PRINTER_DRIVER_INFO_LEVEL_3 *)memdup(&driver, sizeof(driver));
1656
1657         return WERR_OK;
1658 }
1659
1660 /****************************************************************************
1661 ****************************************************************************/
1662 uint32 get_a_printer_driver_9x_compatible(pstring line, fstring model)
1663 {
1664         NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
1665         TDB_DATA kbuf;
1666         pstring key;
1667         int i;
1668         line[0] = '\0';
1669
1670         slprintf(key, sizeof(key)-1, "%s%s/%d/%s", DRIVERS_PREFIX, "WIN40", 0, model);
1671         DEBUG(10,("driver key: [%s]\n", key));
1672         
1673         kbuf.dptr = key;
1674         kbuf.dsize = strlen(key)+1;
1675         if (!tdb_exists(tdb_drivers, kbuf)) return False;
1676
1677         ZERO_STRUCT(info3);
1678         get_a_printer_driver_3(&info3, model, "Windows 4.0", 0);
1679         
1680     DEBUGADD(10,("info3->name            [%s]\n", info3->name));
1681     DEBUGADD(10,("info3->datafile        [%s]\n", info3->datafile));
1682     DEBUGADD(10,("info3->helpfile        [%s]\n", info3->helpfile));
1683     DEBUGADD(10,("info3->monitorname     [%s]\n", info3->monitorname));
1684     DEBUGADD(10,("info3->defaultdatatype [%s]\n", info3->defaultdatatype));
1685         for (i=0; info3->dependentfiles && *info3->dependentfiles[i]; i++) {
1686     DEBUGADD(10,("info3->dependentfiles  [%s]\n", info3->dependentfiles[i]));
1687     }
1688     DEBUGADD(10,("info3->environment     [%s]\n", info3->environment));
1689     DEBUGADD(10,("info3->driverpath      [%s]\n", info3->driverpath));
1690     DEBUGADD(10,("info3->configfile      [%s]\n", info3->configfile));
1691
1692         /*pstrcat(line, info3->name);             pstrcat(line, ":");*/
1693         trim_string(info3->configfile, "\\print$\\WIN40\\0\\", 0);
1694         pstrcat(line, info3->configfile);
1695     pstrcat(line, ":");
1696         trim_string(info3->datafile, "\\print$\\WIN40\\0\\", 0);
1697         pstrcat(line, info3->datafile);
1698     pstrcat(line, ":");
1699         trim_string(info3->helpfile, "\\print$\\WIN40\\0\\", 0);
1700         pstrcat(line, info3->helpfile);
1701     pstrcat(line, ":");
1702         trim_string(info3->monitorname, "\\print$\\WIN40\\0\\", 0);
1703         pstrcat(line, info3->monitorname);
1704     pstrcat(line, ":");
1705         pstrcat(line, "RAW");                /*info3->defaultdatatype);*/
1706     pstrcat(line, ":");
1707
1708         for (i=0; info3->dependentfiles &&
1709                  *info3->dependentfiles[i]; i++) {
1710                 if (i) pstrcat(line, ",");               /* don't end in a "," */
1711                 trim_string(info3->dependentfiles[i], "\\print$\\WIN40\\0\\", 0);
1712                 pstrcat(line, info3->dependentfiles[i]);
1713         }
1714         
1715         SAFE_FREE(info3);
1716
1717         return True;    
1718 }
1719
1720 /****************************************************************************
1721 debugging function, dump at level 6 the struct in the logs
1722 ****************************************************************************/
1723 static uint32 dump_a_printer_driver(NT_PRINTER_DRIVER_INFO_LEVEL driver, uint32 level)
1724 {
1725         uint32 result;
1726         NT_PRINTER_DRIVER_INFO_LEVEL_3 *info3;
1727         int i;
1728         
1729         DEBUG(106,("Dumping printer driver at level [%d]\n", level));
1730         
1731         switch (level)
1732         {
1733                 case 3:
1734                 {
1735                         if (driver.info_3 == NULL)
1736                                 result=5;
1737                         else {
1738                                 info3=driver.info_3;
1739                         
1740                                 DEBUGADD(106,("version:[%d]\n",         info3->cversion));
1741                                 DEBUGADD(106,("name:[%s]\n",            info3->name));
1742                                 DEBUGADD(106,("environment:[%s]\n",     info3->environment));
1743                                 DEBUGADD(106,("driverpath:[%s]\n",      info3->driverpath));
1744                                 DEBUGADD(106,("datafile:[%s]\n",        info3->datafile));
1745                                 DEBUGADD(106,("configfile:[%s]\n",      info3->configfile));
1746                                 DEBUGADD(106,("helpfile:[%s]\n",        info3->helpfile));
1747                                 DEBUGADD(106,("monitorname:[%s]\n",     info3->monitorname));
1748                                 DEBUGADD(106,("defaultdatatype:[%s]\n", info3->defaultdatatype));
1749                                 
1750                                 for (i=0; info3->dependentfiles &&
1751                                           *info3->dependentfiles[i]; i++) {
1752                                         DEBUGADD(106,("dependentfile:[%s]\n",
1753                                                       info3->dependentfiles[i]));
1754                                 }
1755                                 result=0;
1756                         }
1757                         break;
1758                 }
1759                 default:
1760                         DEBUGADD(1,("Level not implemented\n"));
1761                         result=1;
1762                         break;
1763         }
1764         
1765         return result;
1766 }
1767
1768 /****************************************************************************
1769 ****************************************************************************/
1770 static int pack_devicemode(NT_DEVICEMODE *nt_devmode, char *buf, int buflen)
1771 {
1772         int len = 0;
1773
1774         len += tdb_pack(buf+len, buflen-len, "p", nt_devmode);
1775
1776         if (!nt_devmode) return len;
1777
1778         len += tdb_pack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
1779                         nt_devmode->devicename,
1780                         nt_devmode->formname,
1781
1782                         nt_devmode->specversion,
1783                         nt_devmode->driverversion,
1784                         nt_devmode->size,
1785                         nt_devmode->driverextra,
1786                         nt_devmode->orientation,
1787                         nt_devmode->papersize,
1788                         nt_devmode->paperlength,
1789                         nt_devmode->paperwidth,
1790                         nt_devmode->scale,
1791                         nt_devmode->copies,
1792                         nt_devmode->defaultsource,
1793                         nt_devmode->printquality,
1794                         nt_devmode->color,
1795                         nt_devmode->duplex,
1796                         nt_devmode->yresolution,
1797                         nt_devmode->ttoption,
1798                         nt_devmode->collate,
1799                         nt_devmode->logpixels,
1800                         
1801                         nt_devmode->fields,
1802                         nt_devmode->bitsperpel,
1803                         nt_devmode->pelswidth,
1804                         nt_devmode->pelsheight,
1805                         nt_devmode->displayflags,
1806                         nt_devmode->displayfrequency,
1807                         nt_devmode->icmmethod,
1808                         nt_devmode->icmintent,
1809                         nt_devmode->mediatype,
1810                         nt_devmode->dithertype,
1811                         nt_devmode->reserved1,
1812                         nt_devmode->reserved2,
1813                         nt_devmode->panningwidth,
1814                         nt_devmode->panningheight,
1815                         nt_devmode->private);
1816
1817         
1818         if (nt_devmode->private) {
1819                 len += tdb_pack(buf+len, buflen-len, "B",
1820                                 nt_devmode->driverextra,
1821                                 nt_devmode->private);
1822         }
1823
1824         DEBUG(8,("Packed devicemode [%s]\n", nt_devmode->formname));
1825
1826         return len;
1827 }
1828
1829 /****************************************************************************
1830 ****************************************************************************/
1831 static int pack_specifics(NT_PRINTER_PARAM *param, char *buf, int buflen)
1832 {
1833         int len = 0;
1834
1835         while (param != NULL) {
1836                 len += tdb_pack(buf+len, buflen-len, "pfdB",
1837                                 param,
1838                                 param->value,
1839                                 param->type,
1840                                 param->data_len,
1841                                 param->data);
1842                 param=param->next;      
1843         }
1844
1845         len += tdb_pack(buf+len, buflen-len, "p", param);
1846
1847         return len;
1848 }
1849
1850
1851 /****************************************************************************
1852 delete a printer - this just deletes the printer info file, any open
1853 handles are not affected
1854 ****************************************************************************/
1855 uint32 del_a_printer(char *sharename)
1856 {
1857         pstring key;
1858         TDB_DATA kbuf;
1859
1860         slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
1861
1862         kbuf.dptr=key;
1863         kbuf.dsize=strlen(key)+1;
1864
1865         tdb_delete(tdb_printers, kbuf);
1866         return 0;
1867 }
1868
1869 /* FIXME!!!  Reorder so this forward declaration is not necessary --jerry */
1870 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 **, fstring);
1871 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **);
1872 /****************************************************************************
1873 ****************************************************************************/
1874 static WERROR update_a_printer_2(NT_PRINTER_INFO_LEVEL_2 *info)
1875 {
1876         pstring key;
1877         char *buf;
1878         int buflen, len;
1879         WERROR ret;
1880         TDB_DATA kbuf, dbuf;
1881         
1882         /*
1883          * in addprinter: no servername and the printer is the name
1884          * in setprinter: servername is \\server
1885          *                and printer is \\server\\printer
1886          *
1887          * Samba manages only local printers.
1888          * we currently don't support things like path=\\other_server\printer
1889          */
1890
1891         if (info->servername[0]!='\0') {
1892                 trim_string(info->printername, info->servername, NULL);
1893                 trim_string(info->printername, "\\", NULL);
1894                 info->servername[0]='\0';
1895         }
1896
1897         /*
1898          * JFM: one day I'll forget.
1899          * below that's info->portname because that's the SAMBA sharename
1900          * and I made NT 'thinks' it's the portname
1901          * the info->sharename is the thing you can name when you add a printer
1902          * that's the short-name when you create shared printer for 95/98
1903          * So I've made a limitation in SAMBA: you can only have 1 printer model
1904          * behind a SAMBA share.
1905          */
1906
1907         buf = NULL;
1908         buflen = 0;
1909
1910  again: 
1911         len = 0;
1912         len += tdb_pack(buf+len, buflen-len, "dddddddddddfffffPfffff",
1913                         info->attributes,
1914                         info->priority,
1915                         info->default_priority,
1916                         info->starttime,
1917                         info->untiltime,
1918                         info->status,
1919                         info->cjobs,
1920                         info->averageppm,
1921                         info->changeid,
1922                         info->c_setprinter,
1923                         info->setuptime,
1924                         info->servername,
1925                         info->printername,
1926                         info->sharename,
1927                         info->portname,
1928                         info->drivername,
1929                         info->comment,
1930                         info->location,
1931                         info->sepfile,
1932                         info->printprocessor,
1933                         info->datatype,
1934                         info->parameters);
1935
1936         len += pack_devicemode(info->devmode, buf+len, buflen-len);
1937         
1938         len += pack_specifics(info->specific, buf+len, buflen-len);
1939
1940         if (buflen != len) {
1941                 char *tb;
1942                 
1943                 tb = (char *)Realloc(buf, len);
1944                 if (!tb) {
1945                         DEBUG(0,("update_a_printer_2: failed to enlarge buffer!\n"));
1946                         ret = WERR_NOMEM;
1947                         goto done;
1948                 }
1949                 else buf = tb;
1950                 buflen = len;
1951                 goto again;
1952         }
1953         
1954
1955         slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, info->sharename);
1956
1957         kbuf.dptr = key;
1958         kbuf.dsize = strlen(key)+1;
1959         dbuf.dptr = buf;
1960         dbuf.dsize = len;
1961
1962         ret = (tdb_store(tdb_printers, kbuf, dbuf, TDB_REPLACE) == 0? WERR_OK : WERR_NOMEM);
1963
1964 done:
1965         if (!W_ERROR_IS_OK(ret))
1966                 DEBUG(8, ("error updating printer to tdb on disk\n"));
1967
1968         SAFE_FREE(buf);
1969
1970         DEBUG(8,("packed printer [%s] with driver [%s] portname=[%s] len=%d\n",
1971                  info->sharename, info->drivername, info->portname, len));
1972
1973         return ret;
1974 }
1975
1976
1977 /****************************************************************************
1978 ****************************************************************************/
1979 void add_a_specific_param(NT_PRINTER_INFO_LEVEL_2 *info_2, NT_PRINTER_PARAM **param)
1980 {
1981         NT_PRINTER_PARAM *current;
1982         
1983         DEBUG(108,("add_a_specific_param\n"));  
1984
1985         (*param)->next=NULL;
1986         
1987         if (info_2->specific == NULL)
1988         {
1989                 info_2->specific=*param;
1990         }
1991         else
1992         {
1993                 current=info_2->specific;               
1994                 while (current->next != NULL) {
1995                         current=current->next;
1996                 }               
1997                 current->next=*param;
1998         }
1999
2000         *param = NULL;
2001 }
2002
2003 /****************************************************************************
2004 ****************************************************************************/
2005 BOOL unlink_specific_param_if_exist(NT_PRINTER_INFO_LEVEL_2 *info_2, NT_PRINTER_PARAM *param)
2006 {
2007         NT_PRINTER_PARAM *current;
2008         NT_PRINTER_PARAM *previous;
2009         
2010         current=info_2->specific;
2011         previous=current;
2012         
2013         if (current==NULL) return (False);
2014         
2015         if ( !strcmp(current->value, param->value) &&
2016             (strlen(current->value)==strlen(param->value)) ) {
2017                 DEBUG(109,("deleting first value\n"));
2018                 info_2->specific=current->next;
2019                 SAFE_FREE(current->data);
2020                 SAFE_FREE(current);
2021                 DEBUG(109,("deleted first value\n"));
2022                 return (True);
2023         }
2024
2025         current=previous->next;
2026                 
2027         while ( current!=NULL ) {
2028                 if (!strcmp(current->value, param->value) &&
2029                     strlen(current->value)==strlen(param->value) ) {
2030                         DEBUG(109,("deleting current value\n"));
2031                         previous->next=current->next;
2032                         SAFE_FREE(current->data);
2033                         SAFE_FREE(current);
2034                         DEBUG(109,("deleted current value\n"));
2035                         return(True);
2036                 }
2037                 
2038                 previous=previous->next;
2039                 current=current->next;
2040         }
2041         return (False);
2042 }
2043
2044 /****************************************************************************
2045  Clean up and deallocate a (maybe partially) allocated NT_PRINTER_PARAM.
2046 ****************************************************************************/
2047 void free_nt_printer_param(NT_PRINTER_PARAM **param_ptr)
2048 {
2049         NT_PRINTER_PARAM *param = *param_ptr;
2050
2051         if(param == NULL)
2052                 return;
2053
2054         DEBUG(106,("free_nt_printer_param: deleting param [%s]\n", param->value));
2055
2056         SAFE_FREE(param->data);
2057         SAFE_FREE(*param_ptr);
2058 }
2059
2060 /****************************************************************************
2061  Malloc and return an NT devicemode.
2062 ****************************************************************************/
2063
2064 NT_DEVICEMODE *construct_nt_devicemode(const fstring default_devicename)
2065 {
2066 /*
2067  * should I init this ones ???
2068         nt_devmode->devicename
2069 */
2070
2071         char adevice[32];
2072         NT_DEVICEMODE *nt_devmode = (NT_DEVICEMODE *)malloc(sizeof(NT_DEVICEMODE));
2073
2074         if (nt_devmode == NULL) {
2075                 DEBUG(0,("construct_nt_devicemode: malloc fail.\n"));
2076                 return NULL;
2077         }
2078
2079         ZERO_STRUCTP(nt_devmode);
2080
2081         safe_strcpy(adevice, default_devicename, sizeof(adevice));
2082         fstrcpy(nt_devmode->devicename, adevice);       
2083         
2084         fstrcpy(nt_devmode->formname, "Letter");
2085
2086         nt_devmode->specversion      = 0x0401;
2087         nt_devmode->driverversion    = 0x0400;
2088         nt_devmode->size             = 0x00DC;
2089         nt_devmode->driverextra      = 0x0000;
2090         nt_devmode->fields           = FORMNAME | TTOPTION | PRINTQUALITY |
2091                                        DEFAULTSOURCE | COPIES | SCALE |
2092                                        PAPERSIZE | ORIENTATION;
2093         nt_devmode->orientation      = 1;
2094         nt_devmode->papersize        = PAPER_LETTER;
2095         nt_devmode->paperlength      = 0;
2096         nt_devmode->paperwidth       = 0;
2097         nt_devmode->scale            = 0x64;
2098         nt_devmode->copies           = 1;
2099         nt_devmode->defaultsource    = BIN_FORMSOURCE;
2100         nt_devmode->printquality     = RES_HIGH;           /* 0x0258 */
2101         nt_devmode->color            = COLOR_MONOCHROME;
2102         nt_devmode->duplex           = DUP_SIMPLEX;
2103         nt_devmode->yresolution      = 0;
2104         nt_devmode->ttoption         = TT_SUBDEV;
2105         nt_devmode->collate          = COLLATE_FALSE;
2106         nt_devmode->icmmethod        = 0;
2107         nt_devmode->icmintent        = 0;
2108         nt_devmode->mediatype        = 0;
2109         nt_devmode->dithertype       = 0;
2110
2111         /* non utilisés par un driver d'imprimante */
2112         nt_devmode->logpixels        = 0;
2113         nt_devmode->bitsperpel       = 0;
2114         nt_devmode->pelswidth        = 0;
2115         nt_devmode->pelsheight       = 0;
2116         nt_devmode->displayflags     = 0;
2117         nt_devmode->displayfrequency = 0;
2118         nt_devmode->reserved1        = 0;
2119         nt_devmode->reserved2        = 0;
2120         nt_devmode->panningwidth     = 0;
2121         nt_devmode->panningheight    = 0;
2122         
2123         nt_devmode->private = NULL;
2124         return nt_devmode;
2125 }
2126
2127 /****************************************************************************
2128  Deepcopy an NT devicemode.
2129 ****************************************************************************/
2130
2131 NT_DEVICEMODE *dup_nt_devicemode(NT_DEVICEMODE *nt_devicemode)
2132 {
2133         NT_DEVICEMODE *new_nt_devicemode = NULL;
2134
2135         if ((new_nt_devicemode = (NT_DEVICEMODE *)memdup(nt_devicemode, sizeof(NT_DEVICEMODE))) == NULL) {
2136                 DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2137                 return NULL;
2138         }
2139
2140         new_nt_devicemode->private = NULL;
2141         if (nt_devicemode->private != NULL) {
2142                 if ((new_nt_devicemode->private = memdup(nt_devicemode->private, nt_devicemode->driverextra)) == NULL) {
2143                         SAFE_FREE(new_nt_devicemode);
2144                         DEBUG(0,("dup_nt_devicemode: malloc fail.\n"));
2145                         return NULL;
2146         }
2147         }
2148
2149         return new_nt_devicemode;
2150 }
2151
2152 /****************************************************************************
2153  Clean up and deallocate a (maybe partially) allocated NT_DEVICEMODE.
2154 ****************************************************************************/
2155
2156 void free_nt_devicemode(NT_DEVICEMODE **devmode_ptr)
2157 {
2158         NT_DEVICEMODE *nt_devmode = *devmode_ptr;
2159
2160         if(nt_devmode == NULL)
2161                 return;
2162
2163         DEBUG(106,("free_nt_devicemode: deleting DEVMODE\n"));
2164
2165         SAFE_FREE(nt_devmode->private);
2166         SAFE_FREE(*devmode_ptr);
2167 }
2168
2169 /****************************************************************************
2170  Clean up and deallocate a (maybe partially) allocated NT_PRINTER_INFO_LEVEL_2.
2171 ****************************************************************************/
2172 static void free_nt_printer_info_level_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr)
2173 {
2174         NT_PRINTER_INFO_LEVEL_2 *info = *info_ptr;
2175         NT_PRINTER_PARAM *param_ptr;
2176
2177         if(info == NULL)
2178                 return;
2179
2180         DEBUG(106,("free_nt_printer_info_level_2: deleting info\n"));
2181
2182         free_nt_devicemode(&info->devmode);
2183
2184         for(param_ptr = info->specific; param_ptr; ) {
2185                 NT_PRINTER_PARAM *tofree = param_ptr;
2186
2187                 param_ptr = param_ptr->next;
2188                 free_nt_printer_param(&tofree);
2189         }
2190
2191         SAFE_FREE(*info_ptr);
2192 }
2193
2194
2195 /****************************************************************************
2196 ****************************************************************************/
2197 static int unpack_devicemode(NT_DEVICEMODE **nt_devmode, char *buf, int buflen)
2198 {
2199         int len = 0;
2200         int extra_len = 0;
2201         NT_DEVICEMODE devmode;
2202
2203         ZERO_STRUCT(devmode);
2204
2205         len += tdb_unpack(buf+len, buflen-len, "p", nt_devmode);
2206
2207         if (!*nt_devmode) return len;
2208
2209         len += tdb_unpack(buf+len, buflen-len, "ffwwwwwwwwwwwwwwwwwwddddddddddddddp",
2210                           devmode.devicename,
2211                           devmode.formname,
2212
2213                           &devmode.specversion,
2214                           &devmode.driverversion,
2215                           &devmode.size,
2216                           &devmode.driverextra,
2217                           &devmode.orientation,
2218                           &devmode.papersize,
2219                           &devmode.paperlength,
2220                           &devmode.paperwidth,
2221                           &devmode.scale,
2222                           &devmode.copies,
2223                           &devmode.defaultsource,
2224                           &devmode.printquality,
2225                           &devmode.color,
2226                           &devmode.duplex,
2227                           &devmode.yresolution,
2228                           &devmode.ttoption,
2229                           &devmode.collate,
2230                           &devmode.logpixels,
2231                         
2232                           &devmode.fields,
2233                           &devmode.bitsperpel,
2234                           &devmode.pelswidth,
2235                           &devmode.pelsheight,
2236                           &devmode.displayflags,
2237                           &devmode.displayfrequency,
2238                           &devmode.icmmethod,
2239                           &devmode.icmintent,
2240                           &devmode.mediatype,
2241                           &devmode.dithertype,
2242                           &devmode.reserved1,
2243                           &devmode.reserved2,
2244                           &devmode.panningwidth,
2245                           &devmode.panningheight,
2246                           &devmode.private);
2247         
2248         if (devmode.private) {
2249                 /* the len in tdb_unpack is an int value and
2250                  * devmode.driverextra is only a short
2251                  */
2252                 len += tdb_unpack(buf+len, buflen-len, "B", &extra_len, &devmode.private);
2253                 devmode.driverextra=(uint16)extra_len;
2254                 
2255                 /* check to catch an invalid TDB entry so we don't segfault */
2256                 if (devmode.driverextra == 0) {
2257                         devmode.private = NULL;
2258                 }
2259         }
2260
2261         *nt_devmode = (NT_DEVICEMODE *)memdup(&devmode, sizeof(devmode));
2262
2263         DEBUG(8,("Unpacked devicemode [%s](%s)\n", devmode.devicename, devmode.formname));
2264         if (devmode.private)
2265                 DEBUG(8,("with a private section of %d bytes\n", devmode.driverextra));
2266
2267         return len;
2268 }
2269
2270 /****************************************************************************
2271 ****************************************************************************/
2272 static int unpack_specifics(NT_PRINTER_PARAM **list, char *buf, int buflen)
2273 {
2274         int len = 0;
2275         NT_PRINTER_PARAM param, *p;
2276
2277         *list = NULL;
2278
2279         while (1) {
2280                 len += tdb_unpack(buf+len, buflen-len, "p", &p);
2281                 if (!p) break;
2282
2283                 len += tdb_unpack(buf+len, buflen-len, "fdB",
2284                                   param.value,
2285                                   &param.type,
2286                                   &param.data_len,
2287                                   &param.data);
2288                 param.next = *list;
2289                 *list = memdup(&param, sizeof(param));
2290
2291                 DEBUG(8,("specific: [%s], len: %d\n", param.value, param.data_len));
2292         }
2293
2294         return len;
2295 }
2296
2297 static void map_to_os2_driver(fstring drivername)
2298 {
2299         static BOOL initialised=False;
2300         static fstring last_from,last_to;
2301         char *mapfile = lp_os2_driver_map();
2302         char **lines = NULL;
2303         int numlines = 0;
2304         int i;
2305
2306         if (!strlen(drivername))
2307                 return;
2308
2309         if (!*mapfile)
2310                 return;
2311
2312         if (!initialised) {
2313                 *last_from = *last_to = 0;
2314                 initialised = True;
2315         }
2316
2317         if (strequal(drivername,last_from)) {
2318                 DEBUG(3,("Mapped Windows driver %s to OS/2 driver %s\n",drivername,last_to));
2319                 fstrcpy(drivername,last_to);
2320                 return;
2321         }
2322
2323         lines = file_lines_load(mapfile, &numlines);
2324         if (numlines == 0) {
2325                 DEBUG(0,("No entries in OS/2 driver map %s\n",mapfile));
2326                 return;
2327         }
2328
2329         DEBUG(4,("Scanning OS/2 driver map %s\n",mapfile));
2330
2331         for( i = 0; i < numlines; i++) {
2332                 char *nt_name = lines[i];
2333                 char *os2_name = strchr(nt_name,'=');
2334
2335                 if (!os2_name)
2336                         continue;
2337
2338                 *os2_name++ = 0;
2339
2340                 while (isspace(*nt_name))
2341                         nt_name++;
2342
2343                 if (!*nt_name || strchr("#;",*nt_name))
2344                         continue;
2345
2346                 {
2347                         int l = strlen(nt_name);
2348                         while (l && isspace(nt_name[l-1])) {
2349                                 nt_name[l-1] = 0;
2350                                 l--;
2351                         }
2352                 }
2353
2354                 while (isspace(*os2_name))
2355                         os2_name++;
2356
2357                 {
2358                         int l = strlen(os2_name);
2359                         while (l && isspace(os2_name[l-1])) {
2360                                 os2_name[l-1] = 0;
2361                                 l--;
2362                         }
2363                 }
2364
2365                 if (strequal(nt_name,drivername)) {
2366                         DEBUG(3,("Mapped windows driver %s to os2 driver%s\n",drivername,os2_name));
2367                         fstrcpy(last_from,drivername);
2368                         fstrcpy(last_to,os2_name);
2369                         fstrcpy(drivername,os2_name);
2370                         file_lines_free(lines);
2371                         return;
2372                 }
2373         }
2374
2375         file_lines_free(lines);
2376 }
2377
2378 /****************************************************************************
2379 get a default printer info 2 struct
2380 ****************************************************************************/
2381 static WERROR get_a_printer_2_default(NT_PRINTER_INFO_LEVEL_2 **info_ptr, fstring sharename)
2382 {
2383         int snum;
2384         NT_PRINTER_INFO_LEVEL_2 info;
2385
2386         ZERO_STRUCT(info);
2387
2388         snum = lp_servicenumber(sharename);
2389
2390         slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", get_called_name());
2391         slprintf(info.printername, sizeof(info.printername)-1, "\\\\%s\\%s", 
2392                  get_called_name(), sharename);
2393         fstrcpy(info.sharename, sharename);
2394         fstrcpy(info.portname, SAMBA_PRINTER_PORT_NAME);
2395         fstrcpy(info.drivername, lp_printerdriver(snum));
2396
2397         /* by setting the driver name to an empty string, a local NT admin
2398            can now run the **local** APW to install a local printer driver
2399            for a Samba shared printer in 2.2.  Without this, drivers **must** be 
2400            installed on the Samba server for NT clients --jerry */
2401 #if 0   /* JERRY --do not uncomment-- */
2402         if (!*info.drivername)
2403                 fstrcpy(info.drivername, "NO DRIVER AVAILABLE FOR THIS PRINTER");
2404 #endif
2405
2406
2407         DEBUG(10,("get_a_printer_2_default: driver name set to [%s]\n", info.drivername));
2408
2409         pstrcpy(info.comment, "");
2410         fstrcpy(info.printprocessor, "winprint");
2411         fstrcpy(info.datatype, "RAW");
2412
2413         info.attributes = PRINTER_ATTRIBUTE_SHARED   \
2414                          | PRINTER_ATTRIBUTE_LOCAL  \
2415                          | PRINTER_ATTRIBUTE_RAW_ONLY \
2416                          | PRINTER_ATTRIBUTE_QUEUED ;            /* attributes */
2417
2418         info.starttime = 0; /* Minutes since 12:00am GMT */
2419         info.untiltime = 0; /* Minutes since 12:00am GMT */
2420         info.priority = 1;
2421         info.default_priority = 1;
2422         info.setuptime = (uint32)time(NULL) - 86400;    /* minus 1 day */
2423
2424         /*
2425          * I changed this as I think it is better to have a generic
2426          * DEVMODE than to crash Win2k explorer.exe   --jerry
2427          * See the HP Deskjet 990c Win2k drivers for an example.
2428          */
2429
2430 #if 0 /* JRA - NO NOT CHANGE ! */
2431         info.devmode = NULL;
2432 #else
2433         /*
2434          * We should not return a default devicemode, as this causes
2435          * Win2K to not send the correct one on PCL drivers. It needs to
2436          * see a null devicemode so it can then overwrite the devicemode
2437          * on OpenPrinterEx. Yes this *is* insane :-). JRA.
2438          */
2439         if ((info.devmode = construct_nt_devicemode(info.printername)) == NULL)
2440                 goto fail;
2441 #endif
2442
2443         /* This will get the current RPC talloc context, but we should be
2444            passing this as a parameter... fixme... JRA ! */
2445
2446         if (!nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf))
2447                 goto fail;
2448
2449         *info_ptr = (NT_PRINTER_INFO_LEVEL_2 *)memdup(&info, sizeof(info));
2450         if (! *info_ptr) {
2451                 DEBUG(0,("get_a_printer_2_default: malloc fail.\n"));
2452                 goto fail;
2453         }
2454
2455         return WERR_OK;
2456
2457   fail:
2458
2459         if (info.devmode)
2460                 free_nt_devicemode(&info.devmode);
2461         return WERR_ACCESS_DENIED;
2462 }
2463
2464 /****************************************************************************
2465 ****************************************************************************/
2466 static WERROR get_a_printer_2(NT_PRINTER_INFO_LEVEL_2 **info_ptr, fstring sharename)
2467 {
2468         pstring key;
2469         NT_PRINTER_INFO_LEVEL_2 info;
2470         int             len = 0;
2471         TDB_DATA kbuf, dbuf;
2472         fstring printername;
2473                 
2474         ZERO_STRUCT(info);
2475
2476         slprintf(key, sizeof(key)-1, "%s%s", PRINTERS_PREFIX, sharename);
2477
2478         kbuf.dptr = key;
2479         kbuf.dsize = strlen(key)+1;
2480
2481         dbuf = tdb_fetch(tdb_printers, kbuf);
2482         if (!dbuf.dptr)
2483                 return get_a_printer_2_default(info_ptr, sharename);
2484
2485         len += tdb_unpack(dbuf.dptr+len, dbuf.dsize-len, "dddddddddddfffffPfffff",
2486                         &info.attributes,
2487                         &info.priority,
2488                         &info.default_priority,
2489                         &info.starttime,
2490                         &info.untiltime,
2491                         &info.status,
2492                         &info.cjobs,
2493                         &info.averageppm,
2494                         &info.changeid,
2495                         &info.c_setprinter,
2496                         &info.setuptime,
2497                         info.servername,
2498                         info.printername,
2499                         info.sharename,
2500                         info.portname,
2501                         info.drivername,
2502                         info.comment,
2503                         info.location,
2504                         info.sepfile,
2505                         info.printprocessor,
2506                         info.datatype,
2507                         info.parameters);
2508
2509         /* Samba has to have shared raw drivers. */
2510         info.attributes |= (PRINTER_ATTRIBUTE_SHARED|PRINTER_ATTRIBUTE_RAW_ONLY);
2511
2512         /* Restore the stripped strings. */
2513         slprintf(info.servername, sizeof(info.servername)-1, "\\\\%s", get_called_name());
2514         slprintf(printername, sizeof(printername)-1, "\\\\%s\\%s", get_called_name(),
2515                         info.printername);
2516         fstrcpy(info.printername, printername);
2517
2518         len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
2519 #if 1
2520         /*
2521          * Some client drivers freak out if there is a NULL devmode
2522          * (probably the driver is not checking before accessing 
2523          * the devmode pointer)   --jerry
2524          */
2525         if (!info.devmode)
2526         {
2527                 DEBUG(8,("get_a_printer_2: Constructing a default device mode for [%s]\n",
2528                         printername));
2529                 info.devmode = construct_nt_devicemode(printername);
2530         }
2531 #endif
2532         len += unpack_specifics(&info.specific,dbuf.dptr+len, dbuf.dsize-len);
2533
2534         /* This will get the current RPC talloc context, but we should be
2535        passing this as a parameter... fixme... JRA ! */
2536
2537         nt_printing_getsec(get_talloc_ctx(), sharename, &info.secdesc_buf);
2538
2539         /* Fix for OS/2 drivers. */
2540
2541         if (get_remote_arch() == RA_OS2)
2542                 map_to_os2_driver(info.drivername);
2543
2544         SAFE_FREE(dbuf.dptr);
2545         *info_ptr=memdup(&info, sizeof(info));
2546
2547         DEBUG(9,("Unpacked printer [%s] name [%s] running driver [%s]\n",
2548                  sharename, info.printername, info.drivername));
2549
2550         return WERR_OK; 
2551 }
2552
2553 /****************************************************************************
2554 debugging function, dump at level 6 the struct in the logs
2555 ****************************************************************************/
2556 static uint32 dump_a_printer(NT_PRINTER_INFO_LEVEL printer, uint32 level)
2557 {
2558         uint32 result;
2559         NT_PRINTER_INFO_LEVEL_2 *info2;
2560         
2561         DEBUG(106,("Dumping printer at level [%d]\n", level));
2562         
2563         switch (level)
2564         {
2565                 case 2:
2566                 {
2567                         if (printer.info_2 == NULL)
2568                                 result=5;
2569                         else
2570                         {
2571                                 info2=printer.info_2;
2572                         
2573                                 DEBUGADD(106,("attributes:[%d]\n", info2->attributes));
2574                                 DEBUGADD(106,("priority:[%d]\n", info2->priority));
2575                                 DEBUGADD(106,("default_priority:[%d]\n", info2->default_priority));
2576                                 DEBUGADD(106,("starttime:[%d]\n", info2->starttime));
2577                                 DEBUGADD(106,("untiltime:[%d]\n", info2->untiltime));
2578                                 DEBUGADD(106,("status:[%d]\n", info2->status));
2579                                 DEBUGADD(106,("cjobs:[%d]\n", info2->cjobs));
2580                                 DEBUGADD(106,("averageppm:[%d]\n", info2->averageppm));
2581                                 DEBUGADD(106,("changeid:[%d]\n", info2->changeid));
2582                                 DEBUGADD(106,("c_setprinter:[%d]\n", info2->c_setprinter));
2583                                 DEBUGADD(106,("setuptime:[%d]\n", info2->setuptime));
2584
2585                                 DEBUGADD(106,("servername:[%s]\n", info2->servername));
2586                                 DEBUGADD(106,("printername:[%s]\n", info2->printername));
2587                                 DEBUGADD(106,("sharename:[%s]\n", info2->sharename));
2588                                 DEBUGADD(106,("portname:[%s]\n", info2->portname));
2589                                 DEBUGADD(106,("drivername:[%s]\n", info2->drivername));
2590                                 DEBUGADD(106,("comment:[%s]\n", info2->comment));
2591                                 DEBUGADD(106,("location:[%s]\n", info2->location));
2592                                 DEBUGADD(106,("sepfile:[%s]\n", info2->sepfile));
2593                                 DEBUGADD(106,("printprocessor:[%s]\n", info2->printprocessor));
2594                                 DEBUGADD(106,("datatype:[%s]\n", info2->datatype));
2595                                 DEBUGADD(106,("parameters:[%s]\n", info2->parameters));
2596                                 result=0;
2597                         }
2598                         break;
2599                 }
2600                 default:
2601                         DEBUGADD(1,("Level not implemented\n"));
2602                         result=1;
2603                         break;
2604         }
2605         
2606         return result;
2607 }
2608
2609 /****************************************************************************
2610  Get the parameters we can substitute in an NT print job.
2611 ****************************************************************************/
2612
2613 void get_printer_subst_params(int snum, fstring *printername, fstring *sharename, fstring *portname)
2614 {
2615         NT_PRINTER_INFO_LEVEL *printer = NULL;
2616         
2617         **printername = **sharename = **portname = '\0';
2618
2619         if (!W_ERROR_IS_OK(get_a_printer(&printer, 2, lp_servicename(snum))))
2620                 return;
2621
2622         fstrcpy(*printername, printer->info_2->printername);
2623         fstrcpy(*sharename, printer->info_2->sharename);
2624         fstrcpy(*portname, printer->info_2->portname);
2625
2626         free_a_printer(&printer, 2);
2627 }
2628
2629 /*
2630  * The function below are the high level ones.
2631  * only those ones must be called from the spoolss code.
2632  * JFM.
2633  */
2634
2635 /****************************************************************************
2636  Modify a printer. This is called from SETPRINTERDATA/DELETEPRINTERDATA.
2637 ****************************************************************************/
2638
2639 WERROR mod_a_printer(NT_PRINTER_INFO_LEVEL printer, uint32 level)
2640 {
2641         WERROR result;
2642         
2643         dump_a_printer(printer, level); 
2644         
2645         switch (level)
2646         {
2647                 case 2:
2648                 {
2649                         /*
2650                          * Update the changestamp.  Emperical tests show that the
2651                          * ChangeID is always updated,but c_setprinter is only 
2652                          * incremented on a SetPrinter() call.
2653                          */
2654
2655                         time_t time_unix = time(NULL);
2656
2657                         /* ChangeID **must** be increasing over the lifetime
2658                            of client's spoolss service in order for the 
2659                            client's cache to show updates */
2660
2661                         printer.info_2->changeid = time_unix * 100;
2662
2663                         /*
2664                          * Because one day someone will ask:
2665                          * NT->NT       An admin connection to a remote
2666                          *              printer show changes imeediately in
2667                          *              the properities dialog
2668                          *      
2669                          *              A non-admin connection will only show the
2670                          *              changes after viewing the properites page
2671                          *              2 times.  Seems to be related to a
2672                          *              race condition in the client between the spooler
2673                          *              updating the local cache and the Explorer.exe GUI
2674                          *              actually displaying the properties.
2675                          *
2676                          *              This is fixed in Win2k.  admin/non-admin
2677                          *              connections both display changes immediately.
2678                          *
2679                          * 14/12/01     --jerry
2680                          */
2681
2682                         result=update_a_printer_2(printer.info_2);
2683                         break;
2684                 }
2685                 default:
2686                         result=WERR_UNKNOWN_LEVEL;
2687                         break;
2688         }
2689         
2690         return result;
2691 }
2692
2693 /****************************************************************************
2694  Add a printer. This is called from ADDPRINTER(EX) and also SETPRINTER.
2695  We split this out from mod_a_printer as it updates the id's and timestamps.
2696 ****************************************************************************/
2697
2698 WERROR add_a_printer(NT_PRINTER_INFO_LEVEL printer, uint32 level)
2699 {
2700         WERROR result;
2701         
2702         dump_a_printer(printer, level); 
2703         
2704         switch (level)
2705         {
2706                 case 2:
2707                 {
2708                         /*
2709                          * Update the changestamp.  See comments in mod_a_printer()
2710                          * --jerry
2711                          */
2712
2713                         time_t time_unix = time(NULL);
2714
2715                         printer.info_2->changeid = time_unix * 100;
2716                         printer.info_2->c_setprinter++;
2717
2718                         result=update_a_printer_2(printer.info_2);
2719                         break;
2720                 }
2721                 default:
2722                         result=WERR_UNKNOWN_LEVEL;
2723                         break;
2724         }
2725         
2726         return result;
2727 }
2728
2729 /****************************************************************************
2730  Initialize printer devmode & data with previously saved driver init values.
2731 ****************************************************************************/
2732 static uint32 set_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info_ptr)
2733 {
2734         int                     len = 0;
2735         pstring                 key;
2736         TDB_DATA                kbuf, dbuf;
2737         NT_PRINTER_PARAM        *current;
2738         NT_PRINTER_INFO_LEVEL_2 info;
2739
2740         ZERO_STRUCT(info);
2741
2742         slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info_ptr->drivername);
2743
2744         kbuf.dptr = key;
2745         kbuf.dsize = strlen(key)+1;
2746
2747         dbuf = tdb_fetch(tdb_drivers, kbuf);
2748         if (!dbuf.dptr)
2749                 return False;
2750
2751         /*
2752          * Get the saved DEVMODE..
2753          */
2754         len += unpack_devicemode(&info.devmode,dbuf.dptr+len, dbuf.dsize-len);
2755
2756         /*
2757          * The saved DEVMODE contains the devicename from the printer used during
2758          * the initialization save. Change it to reflect the new printer.
2759          */
2760         ZERO_STRUCT(info.devmode->devicename);
2761         fstrcpy(info.devmode->devicename, info_ptr->printername);
2762
2763         /* 
2764          *      Bind the saved DEVMODE to the new the printer.
2765          */
2766         free_nt_devicemode(&info_ptr->devmode);
2767         info_ptr->devmode = info.devmode;
2768
2769         DEBUG(10,("set_driver_init_2: Set printer [%s] init DEVMODE for driver [%s]\n",
2770                         info_ptr->printername, info_ptr->drivername));
2771
2772         /* 
2773          * There should not be any printer data 'specifics' already set during the
2774          * add printer operation, if there are delete them. 
2775          */
2776         while ( (current=info_ptr->specific) != NULL ) {
2777                 info_ptr->specific=current->next;
2778                 SAFE_FREE(current->data);
2779                 SAFE_FREE(current);
2780         }
2781
2782         /* 
2783          * Add the printer data 'specifics' to the new printer
2784          */
2785         len += unpack_specifics(&info_ptr->specific,dbuf.dptr+len, dbuf.dsize-len);
2786
2787         SAFE_FREE(dbuf.dptr);
2788
2789         return True;    
2790 }
2791
2792 /****************************************************************************
2793  Initialize printer devmode & data with previously saved driver init values.
2794  When a printer is created using AddPrinter, the drivername bound to the
2795  printer is used to lookup previously saved driver initialization info, which
2796  is bound to the new printer.
2797 ****************************************************************************/
2798
2799 uint32 set_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level)
2800 {
2801         uint32 result;
2802         
2803         switch (level)
2804         {
2805                 case 2:
2806                 {
2807                         result=set_driver_init_2(printer->info_2);
2808                         break;
2809                 }
2810                 default:
2811                         result=1;
2812                         break;
2813         }
2814         
2815         return result;
2816 }
2817
2818 /****************************************************************************
2819  Pack up the DEVMODE and specifics for a printer into a 'driver init' entry 
2820  in the tdb. Note: this is different from the driver entry and the printer
2821  entry. There should be a single driver init entry for each driver regardless
2822  of whether it was installed from NT or 2K. Technically, they should be
2823  different, but they work out to the same struct.
2824 ****************************************************************************/
2825 static uint32 update_driver_init_2(NT_PRINTER_INFO_LEVEL_2 *info)
2826 {
2827         pstring key;
2828         char *buf;
2829         int buflen, len, ret;
2830         TDB_DATA kbuf, dbuf;
2831
2832         buf = NULL;
2833         buflen = 0;
2834
2835  again: 
2836         len = 0;
2837         len += pack_devicemode(info->devmode, buf+len, buflen-len);
2838
2839         len += pack_specifics(info->specific, buf+len, buflen-len);
2840
2841         if (buflen != len) {
2842                 char *tb;
2843                 
2844                 tb = (char *)Realloc(buf, len);
2845                 if (!tb) {
2846                         DEBUG(0, ("update_driver_init_2: failed to enlarge buffer!\n"));
2847                         ret = -1;
2848                         goto done;
2849                 }
2850                 else buf = tb;
2851                 buflen = len;
2852                 goto again;
2853         }
2854
2855         slprintf(key, sizeof(key)-1, "%s%s", DRIVER_INIT_PREFIX, info->drivername);
2856
2857         kbuf.dptr = key;
2858         kbuf.dsize = strlen(key)+1;
2859         dbuf.dptr = buf;
2860         dbuf.dsize = len;
2861
2862         ret = tdb_store(tdb_drivers, kbuf, dbuf, TDB_REPLACE);
2863
2864 done:
2865         if (ret == -1)
2866                 DEBUG(8, ("update_driver_init_2: error updating printer init to tdb on disk\n"));
2867
2868         SAFE_FREE(buf);
2869
2870         DEBUG(10,("update_driver_init_2: Saved printer [%s] init DEVMODE & specifics for driver [%s]\n",
2871                  info->sharename, info->drivername));
2872
2873         return ret;
2874 }
2875
2876 /****************************************************************************
2877  Update (i.e. save) the driver init info (DEVMODE and specifics) for a printer
2878 ****************************************************************************/
2879
2880 static uint32 update_driver_init(NT_PRINTER_INFO_LEVEL printer, uint32 level)
2881 {
2882         uint32 result;
2883         
2884         dump_a_printer(printer, level); 
2885         
2886         switch (level)
2887         {
2888                 case 2:
2889                 {
2890                         result=update_driver_init_2(printer.info_2);
2891                         break;
2892                 }
2893                 default:
2894                         result=1;
2895                         break;
2896         }
2897         
2898         return result;
2899 }
2900
2901 /****************************************************************************
2902  Convert the printer data value, a REG_BINARY array, into an initialization 
2903  DEVMODE. Note: the array must be parsed as if it was a DEVMODE in an rpc...
2904  got to keep the endians happy :).
2905 ****************************************************************************/
2906
2907 static BOOL convert_driver_init(NT_PRINTER_PARAM *param, TALLOC_CTX *ctx, NT_DEVICEMODE *nt_devmode)
2908 {
2909         BOOL       result = False;
2910         prs_struct ps;
2911         DEVICEMODE devmode;
2912
2913         ZERO_STRUCT(devmode);
2914
2915         prs_init(&ps, 0, ctx, UNMARSHALL);
2916         ps.data_p      = (char *)param->data;
2917         ps.buffer_size = param->data_len;
2918
2919         if (spoolss_io_devmode("phantom DEVMODE", &ps, 0, &devmode))
2920                 result = convert_devicemode("", &devmode, &nt_devmode);
2921         else
2922                 DEBUG(10,("convert_driver_init: error parsing DEVMODE\n"));
2923
2924         return result;
2925 }
2926
2927 /****************************************************************************
2928  Set the DRIVER_INIT info in the tdb. Requires Win32 client code that:
2929
2930  1. Use the driver's config DLL to this UNC printername and:
2931     a. Call DrvPrintEvent with PRINTER_EVENT_INITIALIZE
2932     b. Call DrvConvertDevMode with CDM_DRIVER_DEFAULT to get default DEVMODE
2933  2. Call SetPrinterData with the 'magic' key and the DEVMODE as data.
2934
2935  The last step triggers saving the "driver initialization" information for
2936  this printer into the tdb. Later, new printers that use this driver will
2937  have this initialization information bound to them. This simulates the
2938  driver initialization, as if it had run on the Samba server (as it would
2939  have done on NT).
2940
2941  The Win32 client side code requirement sucks! But until we can run arbitrary
2942  Win32 printer driver code on any Unix that Samba runs on, we are stuck with it.
2943  
2944  It would have been easier to use SetPrinter because all the UNMARSHALLING of
2945  the DEVMODE is done there, but 2K/XP clients do not set the DEVMODE... think
2946  about it and you will realize why.  JRR 010720
2947 ****************************************************************************/
2948
2949 static WERROR save_driver_init_2(NT_PRINTER_INFO_LEVEL *printer, NT_PRINTER_PARAM *param)
2950 {
2951         WERROR        status       = WERR_OK;
2952         TALLOC_CTX    *ctx         = NULL;
2953         NT_DEVICEMODE *nt_devmode  = NULL;
2954         NT_DEVICEMODE *tmp_devmode = printer->info_2->devmode;
2955         
2956         /*
2957          * Set devmode on printer info, so entire printer initialization can be 
2958          * saved to tdb.
2959          */
2960         if ((ctx = talloc_init()) == NULL)
2961                 return WERR_NOMEM;
2962
2963         if ((nt_devmode = (NT_DEVICEMODE*)malloc(sizeof(NT_DEVICEMODE))) == NULL) {
2964                 status = WERR_NOMEM;
2965                 goto done;
2966         }
2967         
2968         ZERO_STRUCTP(nt_devmode);
2969
2970         /*
2971          * The DEVMODE is held in the 'data' component of the param in raw binary.
2972          * Convert it to to a devmode structure
2973          */
2974         if (!convert_driver_init(param, ctx, nt_devmode)) {
2975                 DEBUG(10,("save_driver_init_2: error converting DEVMODE\n"));
2976                 status = WERR_INVALID_PARAM;
2977                 goto done;
2978         }
2979
2980         /*
2981          * Pack up and add (or update) the DEVMODE and any current printer data to
2982          * a 'driver init' element in the tdb
2983          * 
2984          */
2985         printer->info_2->devmode = nt_devmode;
2986         if (update_driver_init(*printer, 2)!=0) {
2987                 DEBUG(10,("save_driver_init_2: error updating DEVMODE\n"));
2988                 status = WERR_NOMEM;
2989                 goto done;
2990         }
2991         
2992         /*
2993          * If driver initialization info was successfully saved, set the current 
2994          * printer to match it. This allows initialization of the current printer 
2995          * as well as the driver.
2996          */
2997         status = mod_a_printer(*printer, 2);
2998         if (!W_ERROR_IS_OK(status)) {
2999                 DEBUG(10,("save_driver_init_2: error setting DEVMODE on printer [%s]\n",
3000                                   printer->info_2->printername));
3001         }
3002
3003   done:
3004         talloc_destroy(ctx);
3005         if (nt_devmode)
3006                 SAFE_FREE(nt_devmode->private);
3007         SAFE_FREE(nt_devmode);
3008         printer->info_2->devmode = tmp_devmode;
3009
3010         return status;
3011 }
3012
3013 /****************************************************************************
3014  Update the driver init info (DEVMODE and specifics) for a printer
3015 ****************************************************************************/
3016
3017 WERROR save_driver_init(NT_PRINTER_INFO_LEVEL *printer, uint32 level, NT_PRINTER_PARAM *param)
3018 {
3019         WERROR status = WERR_OK;
3020         
3021         switch (level)
3022         {
3023                 case 2:
3024                 {
3025                         status=save_driver_init_2(printer, param);
3026                         break;
3027                 }
3028                 default:
3029                         status=WERR_UNKNOWN_LEVEL;
3030                         break;
3031         }
3032         
3033         return status;
3034 }
3035
3036 /****************************************************************************
3037  Get a NT_PRINTER_INFO_LEVEL struct. It returns malloced memory.
3038 ****************************************************************************/
3039
3040 WERROR get_a_printer(NT_PRINTER_INFO_LEVEL **pp_printer, uint32 level, fstring sharename)
3041 {
3042         WERROR result;
3043         NT_PRINTER_INFO_LEVEL *printer = NULL;
3044         
3045         *pp_printer = NULL;
3046
3047         DEBUG(10,("get_a_printer: [%s] level %u\n", sharename, (unsigned int)level));
3048
3049         switch (level)
3050         {
3051                 case 2:
3052                 {
3053                         if ((printer = (NT_PRINTER_INFO_LEVEL *)malloc(sizeof(NT_PRINTER_INFO_LEVEL))) == NULL) {
3054                                 DEBUG(0,("get_a_printer: malloc fail.\n"));
3055                                 return WERR_NOMEM;
3056                         }
3057                         ZERO_STRUCTP(printer);
3058                         result=get_a_printer_2(&printer->info_2, sharename);
3059                         if (W_ERROR_IS_OK(result)) {
3060                                 dump_a_printer(*printer, level);
3061                                 *pp_printer = printer;
3062                         } else {
3063                                 SAFE_FREE(printer);
3064                         }
3065                         break;
3066                 }
3067                 default:
3068                         result=WERR_UNKNOWN_LEVEL;
3069                         break;
3070         }
3071         
3072         DEBUG(10,("get_a_printer: [%s] level %u returning %s\n", sharename, (unsigned int)level, werror_str(result)));
3073
3074         return result;
3075 }