s4 dns: Add TSIG and TKEY records to idl
[Samba/gebeck_regimport.git] / source3 / rpc_client / cli_winreg_spoolss.c
blob49b67aa48f739879f6f27f7239b703091c974686
1 /*
2 * Unix SMB/CIFS implementation.
4 * SPOOLSS RPC Pipe server / winreg client routines
6 * Copyright (c) 2010 Andreas Schneider <asn@samba.org>
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "nt_printing.h"
24 #include "../librpc/gen_ndr/ndr_spoolss.h"
25 #include "../librpc/gen_ndr/ndr_winreg_c.h"
26 #include "../librpc/gen_ndr/ndr_security.h"
27 #include "secrets.h"
28 #include "../libcli/security/security.h"
29 #include "rpc_client/cli_winreg.h"
30 #include "../libcli/registry/util_reg.h"
31 #include "rpc_client/cli_winreg_spoolss.h"
32 #include "printing/nt_printing_os2.h"
33 #include "rpc_client/init_spoolss.h"
35 #define TOP_LEVEL_PRINT_KEY "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Print"
36 #define TOP_LEVEL_PRINT_PRINTERS_KEY TOP_LEVEL_PRINT_KEY "\\Printers"
37 #define TOP_LEVEL_CONTROL_KEY "SYSTEM\\CurrentControlSet\\Control\\Print"
38 #define TOP_LEVEL_CONTROL_FORMS_KEY TOP_LEVEL_CONTROL_KEY "\\Forms"
40 #define EMPTY_STRING ""
42 #define FILL_STRING(mem_ctx, in, out) \
43 do { \
44 if (in && strlen(in)) { \
45 out = talloc_strdup(mem_ctx, in); \
46 } else { \
47 out = talloc_strdup(mem_ctx, ""); \
48 } \
49 W_ERROR_HAVE_NO_MEMORY(out); \
50 } while (0);
52 #define CHECK_ERROR(result) \
53 if (W_ERROR_IS_OK(result)) continue; \
54 if (W_ERROR_EQUAL(result, WERR_NOT_FOUND)) result = WERR_OK; \
55 if (!W_ERROR_IS_OK(result)) break
57 /* FLAGS, NAME, with, height, left, top, right, bottom */
58 static const struct spoolss_FormInfo1 builtin_forms1[] = {
59 { SPOOLSS_FORM_BUILTIN, "10x11", {0x3e030,0x44368}, {0x0,0x0,0x3e030,0x44368} },
60 { SPOOLSS_FORM_BUILTIN, "10x14", {0x3e030,0x56d10}, {0x0,0x0,0x3e030,0x56d10} },
61 { SPOOLSS_FORM_BUILTIN, "11x17", {0x44368,0x696b8}, {0x0,0x0,0x44368,0x696b8} },
62 { SPOOLSS_FORM_BUILTIN, "12x11", {0x4a724,0x443e1}, {0x0,0x0,0x4a724,0x443e1} },
63 { SPOOLSS_FORM_BUILTIN, "15x11", {0x5d048,0x44368}, {0x0,0x0,0x5d048,0x44368} },
64 { SPOOLSS_FORM_BUILTIN, "6 3/4 Envelope", {0x167ab,0x284ec}, {0x0,0x0,0x167ab,0x284ec} },
65 { SPOOLSS_FORM_BUILTIN, "9x11", {0x37cf8,0x44368}, {0x0,0x0,0x37cf8,0x44368} },
66 { SPOOLSS_FORM_BUILTIN, "A0", {0xcd528,0x122488},{0x0,0x0,0xcd528,0x122488} },
67 { SPOOLSS_FORM_BUILTIN, "A1", {0x91050,0xcd528}, {0x0,0x0,0x91050,0xcd528} },
68 { SPOOLSS_FORM_BUILTIN, "A2", {0x668a0,0x91050}, {0x0,0x0,0x668a0,0x91050} },
69 { SPOOLSS_FORM_BUILTIN, "A3 Extra Transverse", {0x4e9d0,0x6ca48}, {0x0,0x0,0x4e9d0,0x6ca48} },
70 { SPOOLSS_FORM_BUILTIN, "A3 Extra", {0x4e9d0,0x6ca48}, {0x0,0x0,0x4e9d0,0x6ca48} },
71 { SPOOLSS_FORM_BUILTIN, "A3 Rotated", {0x668a0,0x48828}, {0x0,0x0,0x668a0,0x48828} },
72 { SPOOLSS_FORM_BUILTIN, "A3 Transverse", {0x48828,0x668a0}, {0x0,0x0,0x48828,0x668a0} },
73 { SPOOLSS_FORM_BUILTIN, "A3", {0x48828,0x668a0}, {0x0,0x0,0x48828,0x668a0} },
74 { SPOOLSS_FORM_BUILTIN, "A4 Extra", {0x397c2,0x4eb16}, {0x0,0x0,0x397c2,0x4eb16} },
75 { SPOOLSS_FORM_BUILTIN, "A4 Plus", {0x33450,0x50910}, {0x0,0x0,0x33450,0x50910} },
76 { SPOOLSS_FORM_BUILTIN, "A4 Rotated", {0x48828,0x33450}, {0x0,0x0,0x48828,0x33450} },
77 { SPOOLSS_FORM_BUILTIN, "A4 Small", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
78 { SPOOLSS_FORM_BUILTIN, "A4 Transverse", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
79 { SPOOLSS_FORM_BUILTIN, "A4", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
80 { SPOOLSS_FORM_BUILTIN, "A5 Extra", {0x2a7b0,0x395f8}, {0x0,0x0,0x2a7b0,0x395f8} },
81 { SPOOLSS_FORM_BUILTIN, "A5 Rotated", {0x33450,0x24220}, {0x0,0x0,0x33450,0x24220} },
82 { SPOOLSS_FORM_BUILTIN, "A5 Transverse", {0x24220,0x33450}, {0x0,0x0,0x24220,0x33450} },
83 { SPOOLSS_FORM_BUILTIN, "A5", {0x24220,0x33450}, {0x0,0x0,0x24220,0x33450} },
84 { SPOOLSS_FORM_BUILTIN, "A6 Rotated", {0x24220,0x19a28}, {0x0,0x0,0x24220,0x19a28} },
85 { SPOOLSS_FORM_BUILTIN, "A6", {0x19a28,0x24220}, {0x0,0x0,0x19a28,0x24220} },
86 { SPOOLSS_FORM_BUILTIN, "B4 (ISO)", {0x3d090,0x562e8}, {0x0,0x0,0x3d090,0x562e8} },
87 { SPOOLSS_FORM_BUILTIN, "B4 (JIS) Rotated", {0x58de0,0x3ebe8}, {0x0,0x0,0x58de0,0x3ebe8} },
88 { SPOOLSS_FORM_BUILTIN, "B4 (JIS)", {0x3ebe8,0x58de0}, {0x0,0x0,0x3ebe8,0x58de0} },
89 { SPOOLSS_FORM_BUILTIN, "B5 (ISO) Extra", {0x31128,0x43620}, {0x0,0x0,0x31128,0x43620} },
90 { SPOOLSS_FORM_BUILTIN, "B5 (JIS) Rotated", {0x3ebe8,0x2c6f0}, {0x0,0x0,0x3ebe8,0x2c6f0} },
91 { SPOOLSS_FORM_BUILTIN, "B5 (JIS) Transverse", {0x2c6f0,0x3ebe8}, {0x0,0x0,0x2c6f0,0x3ebe8} },
92 { SPOOLSS_FORM_BUILTIN, "B5 (JIS)", {0x2c6f0,0x3ebe8}, {0x0,0x0,0x2c6f0,0x3ebe8} },
93 { SPOOLSS_FORM_BUILTIN, "B6 (JIS) Rotated", {0x2c6f0,0x1f400}, {0x0,0x0,0x2c6f0,0x1f400} },
94 { SPOOLSS_FORM_BUILTIN, "B6 (JIS)", {0x1f400,0x2c6f0}, {0x0,0x0,0x1f400,0x2c6f0} },
95 { SPOOLSS_FORM_BUILTIN, "C size sheet", {0x696b8,0x886d0}, {0x0,0x0,0x696b8,0x886d0} },
96 { SPOOLSS_FORM_BUILTIN, "D size sheet", {0x886d0,0xd2d70}, {0x0,0x0,0x886d0,0xd2d70} },
97 { SPOOLSS_FORM_BUILTIN, "Double Japan Postcard Rotated", {0x24220,0x30d40}, {0x0,0x0,0x24220,0x30d40} },
98 { SPOOLSS_FORM_BUILTIN, "E size sheet", {0xd2d70,0x110da0},{0x0,0x0,0xd2d70,0x110da0} },
99 { SPOOLSS_FORM_BUILTIN, "Envelope #10", {0x19947,0x3ae94}, {0x0,0x0,0x19947,0x3ae94} },
100 { SPOOLSS_FORM_BUILTIN, "Envelope #11", {0x1be7c,0x40565}, {0x0,0x0,0x1be7c,0x40565} },
101 { SPOOLSS_FORM_BUILTIN, "Envelope #12", {0x1d74a,0x44368}, {0x0,0x0,0x1d74a,0x44368} },
102 { SPOOLSS_FORM_BUILTIN, "Envelope #14", {0x1f018,0x47504}, {0x0,0x0,0x1f018,0x47504} },
103 { SPOOLSS_FORM_BUILTIN, "Envelope #9", {0x18079,0x37091}, {0x0,0x0,0x18079,0x37091} },
104 { SPOOLSS_FORM_BUILTIN, "Envelope B4", {0x3d090,0x562e8}, {0x0,0x0,0x3d090,0x562e8} },
105 { SPOOLSS_FORM_BUILTIN, "Envelope B5", {0x2af80,0x3d090}, {0x0,0x0,0x2af80,0x3d090} },
106 { SPOOLSS_FORM_BUILTIN, "Envelope B6", {0x2af80,0x1e848}, {0x0,0x0,0x2af80,0x1e848} },
107 { SPOOLSS_FORM_BUILTIN, "Envelope C3", {0x4f1a0,0x6fd10}, {0x0,0x0,0x4f1a0,0x6fd10} },
108 { SPOOLSS_FORM_BUILTIN, "Envelope C4", {0x37e88,0x4f1a0}, {0x0,0x0,0x37e88,0x4f1a0} },
109 { SPOOLSS_FORM_BUILTIN, "Envelope C5", {0x278d0,0x37e88}, {0x0,0x0,0x278d0,0x37e88} },
110 { SPOOLSS_FORM_BUILTIN, "Envelope C6", {0x1bd50,0x278d0}, {0x0,0x0,0x1bd50,0x278d0} },
111 { SPOOLSS_FORM_BUILTIN, "Envelope C65", {0x1bd50,0x37e88}, {0x0,0x0,0x1bd50,0x37e88} },
112 { SPOOLSS_FORM_BUILTIN, "Envelope DL", {0x1adb0,0x35b60}, {0x0,0x0,0x1adb0,0x35b60} },
113 { SPOOLSS_FORM_BUILTIN, "Envelope Invite", {0x35b60,0x35b60}, {0x0,0x0,0x35b60,0x35b60} },
114 { SPOOLSS_FORM_BUILTIN, "Envelope Monarch", {0x18079,0x2e824}, {0x0,0x0,0x18079,0x2e824} },
115 { SPOOLSS_FORM_BUILTIN, "Envelope", {0x1adb0,0x38270}, {0x0,0x0,0x1adb0,0x38270} },
116 { SPOOLSS_FORM_BUILTIN, "Executive", {0x2cf56,0x411cc}, {0x0,0x0,0x2cf56,0x411cc} },
117 { SPOOLSS_FORM_BUILTIN, "Folio", {0x34b5c,0x509d8}, {0x0,0x0,0x34b5c,0x509d8} },
118 { SPOOLSS_FORM_BUILTIN, "German Legal Fanfold", {0x34b5c,0x509d8}, {0x0,0x0,0x34b5c,0x509d8} },
119 { SPOOLSS_FORM_BUILTIN, "German Std Fanfold", {0x34b5c,0x4a6a0}, {0x0,0x0,0x34b5c,0x4a6a0} },
120 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Chou #3 Rotated", {0x395f8,0x1d4c0}, {0x0,0x0,0x395f8,0x1d4c0} },
121 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Chou #4 Rotated", {0x320c8,0x15f90}, {0x0,0x0,0x320c8,0x15f90} },
122 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Kaku #2 Rotated", {0x510e0,0x3a980}, {0x0,0x0,0x510e0,0x3a980} },
123 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Kaku #3 Rotated", {0x43a08,0x34bc0}, {0x0,0x0,0x43a08,0x34bc0} },
124 { SPOOLSS_FORM_BUILTIN, "Japan Envelope You #4 Rotated", {0x395f8,0x19a28}, {0x0,0x0,0x395f8,0x19a28} },
125 { SPOOLSS_FORM_BUILTIN, "Japan Envelope You #4", {0x19a28,0x395f8}, {0x0,0x0,0x19a28,0x395f8} },
126 { SPOOLSS_FORM_BUILTIN, "Japanese Double Postcard", {0x30d40,0x24220}, {0x0,0x0,0x30d40,0x24220} },
127 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Chou #3", {0x1d4c0,0x395f8}, {0x0,0x0,0x1d4c0,0x395f8} },
128 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Chou #4", {0x15f90,0x320c8}, {0x0,0x0,0x15f90,0x320c8} },
129 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Kaku #2", {0x3a980,0x510e0}, {0x0,0x0,0x3a980,0x510e0} },
130 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Kaku #3", {0x34bc0,0x43a08}, {0x0,0x0,0x34bc0,0x43a08} },
131 { SPOOLSS_FORM_BUILTIN, "Japanese Postcard Rotated", {0x24220,0x186a0}, {0x0,0x0,0x24220,0x186a0} },
132 { SPOOLSS_FORM_BUILTIN, "Japanese Postcard", {0x186a0,0x24220}, {0x0,0x0,0x186a0,0x24220} },
133 { SPOOLSS_FORM_BUILTIN, "Ledger", {0x696b8,0x44368}, {0x0,0x0,0x696b8,0x44368} },
134 { SPOOLSS_FORM_BUILTIN, "Legal Extra", {0x3ae94,0x5d048}, {0x0,0x0,0x3ae94,0x5d048} },
135 { SPOOLSS_FORM_BUILTIN, "Legal", {0x34b5c,0x56d10}, {0x0,0x0,0x34b5c,0x56d10} },
136 { SPOOLSS_FORM_BUILTIN, "Letter Extra Transverse", {0x3ae94,0x4a6a0}, {0x0,0x0,0x3ae94,0x4a6a0} },
137 { SPOOLSS_FORM_BUILTIN, "Letter Extra", {0x3ae94,0x4a6a0}, {0x0,0x0,0x3ae94,0x4a6a0} },
138 { SPOOLSS_FORM_BUILTIN, "Letter Plus", {0x34b5c,0x4eb16}, {0x0,0x0,0x34b5c,0x4eb16} },
139 { SPOOLSS_FORM_BUILTIN, "Letter Rotated", {0x44368,0x34b5c}, {0x0,0x0,0x44368,0x34b5c} },
140 { SPOOLSS_FORM_BUILTIN, "Letter Small", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
141 { SPOOLSS_FORM_BUILTIN, "Letter Transverse", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
142 { SPOOLSS_FORM_BUILTIN, "Letter", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
143 { SPOOLSS_FORM_BUILTIN, "Note", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
144 { SPOOLSS_FORM_BUILTIN, "PRC 16K Rotated", {0x3f7a0,0x2de60}, {0x0,0x0,0x3f7a0,0x2de60} },
145 { SPOOLSS_FORM_BUILTIN, "PRC 16K", {0x2de60,0x3f7a0}, {0x0,0x0,0x2de60,0x3f7a0} },
146 { SPOOLSS_FORM_BUILTIN, "PRC 32K Rotated", {0x2cec0,0x1fbd0}, {0x0,0x0,0x2cec0,0x1fbd0} },
147 { SPOOLSS_FORM_BUILTIN, "PRC 32K", {0x1fbd0,0x2cec0}, {0x0,0x0,0x1fbd0,0x2cec0} },
148 { SPOOLSS_FORM_BUILTIN, "PRC 32K(Big) Rotated", {0x318f8,0x222e0}, {0x0,0x0,0x318f8,0x222e0} },
149 { SPOOLSS_FORM_BUILTIN, "PRC 32K(Big)", {0x222e0,0x318f8}, {0x0,0x0,0x222e0,0x318f8} },
150 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #1 Rotated", {0x28488,0x18e70}, {0x0,0x0,0x28488,0x18e70} },
151 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #1", {0x18e70,0x28488}, {0x0,0x0,0x18e70,0x28488} },
152 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #10 Rotated", {0x6fd10,0x4f1a0}, {0x0,0x0,0x6fd10,0x4f1a0} },
153 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #10", {0x4f1a0,0x6fd10}, {0x0,0x0,0x4f1a0,0x6fd10} },
154 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #2 Rotated", {0x2af80,0x18e70}, {0x0,0x0,0x2af80,0x18e70} },
155 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #2", {0x18e70,0x2af80}, {0x0,0x0,0x18e70,0x2af80} },
156 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #3 Rotated", {0x2af80,0x1e848}, {0x0,0x0,0x2af80,0x1e848} },
157 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #3", {0x1e848,0x2af80}, {0x0,0x0,0x1e848,0x2af80} },
158 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #4 Rotated", {0x32c80,0x1adb0}, {0x0,0x0,0x32c80,0x1adb0} },
159 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #4", {0x1adb0,0x32c80}, {0x0,0x0,0x1adb0,0x32c80} },
160 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #5 Rotated", {0x35b60,0x1adb0}, {0x0,0x0,0x35b60,0x1adb0} },
161 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #5", {0x1adb0,0x35b60}, {0x0,0x0,0x1adb0,0x35b60} },
162 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #6 Rotated", {0x38270,0x1d4c0}, {0x0,0x0,0x38270,0x1d4c0} },
163 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #6", {0x1d4c0,0x38270}, {0x0,0x0,0x1d4c0,0x38270} },
164 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #7 Rotated", {0x38270,0x27100}, {0x0,0x0,0x38270,0x27100} },
165 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #7", {0x27100,0x38270}, {0x0,0x0,0x27100,0x38270} },
166 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #8 Rotated", {0x4b708,0x1d4c0}, {0x0,0x0,0x4b708,0x1d4c0} },
167 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #8", {0x1d4c0,0x4b708}, {0x0,0x0,0x1d4c0,0x4b708} },
168 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #9 Rotated", {0x4f1a0,0x37e88}, {0x0,0x0,0x4f1a0,0x37e88} },
169 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #9", {0x37e88,0x4f1a0}, {0x0,0x0,0x37e88,0x4f1a0} },
170 { SPOOLSS_FORM_BUILTIN, "Quarto", {0x347d8,0x43238}, {0x0,0x0,0x347d8,0x43238} },
171 { SPOOLSS_FORM_BUILTIN, "Reserved48", {0x1,0x1}, {0x0,0x0,0x1,0x1} },
172 { SPOOLSS_FORM_BUILTIN, "Reserved49", {0x1,0x1}, {0x0,0x0,0x1,0x1} },
173 { SPOOLSS_FORM_BUILTIN, "Statement", {0x221b4,0x34b5c}, {0x0,0x0,0x221b4,0x34b5c} },
174 { SPOOLSS_FORM_BUILTIN, "Super A", {0x376b8,0x56ea0}, {0x0,0x0,0x376b8,0x56ea0} },
175 { SPOOLSS_FORM_BUILTIN, "Super B", {0x4a768,0x76e58}, {0x0,0x0,0x4a768,0x76e58} },
176 { SPOOLSS_FORM_BUILTIN, "Tabloid Extra", {0x4a6a0,0x6f9f0}, {0x0,0x0,0x4a6a0,0x6f9f0} },
177 { SPOOLSS_FORM_BUILTIN, "Tabloid", {0x44368,0x696b8}, {0x0,0x0,0x44368,0x696b8} },
178 { SPOOLSS_FORM_BUILTIN, "US Std Fanfold", {0x5c3e1,0x44368}, {0x0,0x0,0x5c3e1,0x44368} }
181 /********************************************************************
182 static helper functions
183 ********************************************************************/
185 /****************************************************************************
186 Update the changeid time.
187 ****************************************************************************/
189 * @internal
191 * @brief Update the ChangeID time of a printer.
193 * This is SO NASTY as some drivers need this to change, others need it
194 * static. This value will change every second, and I must hope that this
195 * is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
196 * UTAH ! JRA.
198 * @return The ChangeID.
200 static uint32_t winreg_printer_rev_changeid(void)
202 struct timeval tv;
204 get_process_uptime(&tv);
206 #if 1 /* JERRY */
207 /* Return changeid as msec since spooler restart */
208 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
209 #else
211 * This setting seems to work well but is too untested
212 * to replace the above calculation. Left in for experimentation
213 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
215 return tv.tv_sec * 10 + tv.tv_usec / 100000;
216 #endif
219 static WERROR winreg_printer_openkey(TALLOC_CTX *mem_ctx,
220 struct dcerpc_binding_handle *binding_handle,
221 const char *path,
222 const char *key,
223 bool create_key,
224 uint32_t access_mask,
225 struct policy_handle *hive_handle,
226 struct policy_handle *key_handle)
228 struct winreg_String wkey, wkeyclass;
229 char *keyname;
230 NTSTATUS status;
231 WERROR result = WERR_OK;
233 status = dcerpc_winreg_OpenHKLM(binding_handle,
234 mem_ctx,
235 NULL,
236 access_mask,
237 hive_handle,
238 &result);
239 if (!NT_STATUS_IS_OK(status)) {
240 DEBUG(0, ("winreg_printer_openkey: Could not open HKLM hive: %s\n",
241 nt_errstr(status)));
242 return ntstatus_to_werror(status);
244 if (!W_ERROR_IS_OK(result)) {
245 DEBUG(0, ("winreg_printer_openkey: Could not open HKLM hive: %s\n",
246 win_errstr(result)));
247 return result;
250 if (key && *key) {
251 keyname = talloc_asprintf(mem_ctx, "%s\\%s", path, key);
252 } else {
253 keyname = talloc_strdup(mem_ctx, path);
255 if (keyname == NULL) {
256 return WERR_NOMEM;
259 ZERO_STRUCT(wkey);
260 wkey.name = keyname;
262 if (create_key) {
263 enum winreg_CreateAction action = REG_ACTION_NONE;
265 ZERO_STRUCT(wkeyclass);
266 wkeyclass.name = "";
268 status = dcerpc_winreg_CreateKey(binding_handle,
269 mem_ctx,
270 hive_handle,
271 wkey,
272 wkeyclass,
274 access_mask,
275 NULL,
276 key_handle,
277 &action,
278 &result);
279 switch (action) {
280 case REG_ACTION_NONE:
281 DEBUG(8, ("winreg_printer_openkey:createkey did nothing -- huh?\n"));
282 break;
283 case REG_CREATED_NEW_KEY:
284 DEBUG(8, ("winreg_printer_openkey: createkey created %s\n", keyname));
285 break;
286 case REG_OPENED_EXISTING_KEY:
287 DEBUG(8, ("winreg_printer_openkey: createkey opened existing %s\n", keyname));
288 break;
290 } else {
291 status = dcerpc_winreg_OpenKey(binding_handle,
292 mem_ctx,
293 hive_handle,
294 wkey,
296 access_mask,
297 key_handle,
298 &result);
300 if (!NT_STATUS_IS_OK(status)) {
301 result = ntstatus_to_werror(status);
303 if (!W_ERROR_IS_OK(result)) {
304 WERROR ignore;
306 if (is_valid_policy_hnd(hive_handle)) {
307 dcerpc_winreg_CloseKey(binding_handle,
308 mem_ctx,
309 hive_handle,
310 &ignore);
312 ZERO_STRUCTP(hive_handle);
314 return result;
317 return WERR_OK;
321 * @brief Create the registry keyname for the given printer.
323 * @param[in] mem_ctx The memory context to use.
325 * @param[in] printer The name of the printer to get the registry key.
327 * @return The registry key or NULL on error.
329 static char *winreg_printer_data_keyname(TALLOC_CTX *mem_ctx, const char *printer) {
330 return talloc_asprintf(mem_ctx, "%s\\%s", TOP_LEVEL_PRINT_PRINTERS_KEY, printer);
333 static WERROR winreg_printer_opendriver(TALLOC_CTX *mem_ctx,
334 struct dcerpc_binding_handle *winreg_handle,
335 const char *drivername,
336 const char *architecture,
337 uint32_t version,
338 uint32_t access_mask,
339 bool create,
340 struct policy_handle *hive_hnd,
341 struct policy_handle *key_hnd)
343 WERROR result;
344 char *key_name;
346 key_name = talloc_asprintf(mem_ctx, "%s\\Environments\\%s\\Drivers\\Version-%u",
347 TOP_LEVEL_CONTROL_KEY,
348 architecture, version);
349 if (!key_name) {
350 return WERR_NOMEM;
353 result = winreg_printer_openkey(mem_ctx,
354 winreg_handle,
355 key_name,
356 drivername,
357 create,
358 access_mask,
359 hive_hnd,
360 key_hnd);
361 return result;
364 static WERROR winreg_enumval_to_dword(TALLOC_CTX *mem_ctx,
365 struct spoolss_PrinterEnumValues *v,
366 const char *valuename, uint32_t *dw)
368 /* just return if it is not the one we are looking for */
369 if (strcmp(valuename, v->value_name) != 0) {
370 return WERR_NOT_FOUND;
373 if (v->type != REG_DWORD) {
374 return WERR_INVALID_DATATYPE;
377 if (v->data_length != 4) {
378 *dw = 0;
379 return WERR_OK;
382 *dw = IVAL(v->data->data, 0);
383 return WERR_OK;
386 static WERROR winreg_enumval_to_sz(TALLOC_CTX *mem_ctx,
387 struct spoolss_PrinterEnumValues *v,
388 const char *valuename, const char **_str)
390 /* just return if it is not the one we are looking for */
391 if (strcmp(valuename, v->value_name) != 0) {
392 return WERR_NOT_FOUND;
395 if (v->type != REG_SZ) {
396 return WERR_INVALID_DATATYPE;
399 if (v->data_length == 0) {
400 *_str = talloc_strdup(mem_ctx, EMPTY_STRING);
401 if (*_str == NULL) {
402 return WERR_NOMEM;
404 return WERR_OK;
407 if (!pull_reg_sz(mem_ctx, v->data, _str)) {
408 return WERR_NOMEM;
411 return WERR_OK;
414 static WERROR winreg_enumval_to_multi_sz(TALLOC_CTX *mem_ctx,
415 struct spoolss_PrinterEnumValues *v,
416 const char *valuename,
417 const char ***array)
419 /* just return if it is not the one we are looking for */
420 if (strcmp(valuename, v->value_name) != 0) {
421 return WERR_NOT_FOUND;
424 if (v->type != REG_MULTI_SZ) {
425 return WERR_INVALID_DATATYPE;
428 if (v->data_length == 0) {
429 *array = talloc_array(mem_ctx, const char *, 1);
430 if (*array == NULL) {
431 return WERR_NOMEM;
433 *array[0] = NULL;
434 return WERR_OK;
437 if (!pull_reg_multi_sz(mem_ctx, v->data, array)) {
438 return WERR_NOMEM;
441 return WERR_OK;
444 static WERROR winreg_printer_write_date(TALLOC_CTX *mem_ctx,
445 struct dcerpc_binding_handle *winreg_handle,
446 struct policy_handle *key_handle,
447 const char *value,
448 NTTIME data)
450 struct winreg_String wvalue = { 0, };
451 DATA_BLOB blob;
452 WERROR result = WERR_OK;
453 NTSTATUS status;
454 const char *str;
455 struct tm *tm;
456 time_t t;
458 if (data == 0) {
459 str = talloc_strdup(mem_ctx, "01/01/1601");
460 } else {
461 t = nt_time_to_unix(data);
462 tm = localtime(&t);
463 if (tm == NULL) {
464 return map_werror_from_unix(errno);
466 str = talloc_asprintf(mem_ctx, "%02d/%02d/%04d",
467 tm->tm_mon + 1, tm->tm_mday, tm->tm_year + 1900);
469 if (!str) {
470 return WERR_NOMEM;
473 wvalue.name = value;
474 if (!push_reg_sz(mem_ctx, &blob, str)) {
475 return WERR_NOMEM;
477 status = dcerpc_winreg_SetValue(winreg_handle,
478 mem_ctx,
479 key_handle,
480 wvalue,
481 REG_SZ,
482 blob.data,
483 blob.length,
484 &result);
485 if (!NT_STATUS_IS_OK(status)) {
486 result = ntstatus_to_werror(status);
488 if (!W_ERROR_IS_OK(result)) {
489 DEBUG(0, ("winreg_printer_write_date: Could not set value %s: %s\n",
490 wvalue.name, win_errstr(result)));
493 return result;
496 static WERROR winreg_printer_date_to_NTTIME(const char *str, NTTIME *data)
498 struct tm tm;
499 time_t t;
501 if (strequal(str, "01/01/1601")) {
502 *data = 0;
503 return WERR_OK;
506 ZERO_STRUCT(tm);
508 if (sscanf(str, "%d/%d/%d",
509 &tm.tm_mon, &tm.tm_mday, &tm.tm_year) != 3) {
510 return WERR_INVALID_PARAMETER;
512 tm.tm_mon -= 1;
513 tm.tm_year -= 1900;
514 tm.tm_isdst = -1;
516 t = mktime(&tm);
517 unix_to_nt_time(data, t);
519 return WERR_OK;
522 static WERROR winreg_printer_write_ver(TALLOC_CTX *mem_ctx,
523 struct dcerpc_binding_handle *winreg_handle,
524 struct policy_handle *key_handle,
525 const char *value,
526 uint64_t data)
528 struct winreg_String wvalue = { 0, };
529 DATA_BLOB blob;
530 WERROR result = WERR_OK;
531 NTSTATUS status;
532 char *str;
534 /* FIXME: check format is right,
535 * this needs to be something like: 6.1.7600.16385 */
536 str = talloc_asprintf(mem_ctx, "%u.%u.%u.%u",
537 (unsigned)((data >> 48) & 0xFFFF),
538 (unsigned)((data >> 32) & 0xFFFF),
539 (unsigned)((data >> 16) & 0xFFFF),
540 (unsigned)(data & 0xFFFF));
541 if (!str) {
542 return WERR_NOMEM;
545 wvalue.name = value;
546 if (!push_reg_sz(mem_ctx, &blob, str)) {
547 return WERR_NOMEM;
549 status = dcerpc_winreg_SetValue(winreg_handle,
550 mem_ctx,
551 key_handle,
552 wvalue,
553 REG_SZ,
554 blob.data,
555 blob.length,
556 &result);
557 if (!NT_STATUS_IS_OK(status)) {
558 result = ntstatus_to_werror(status);
560 if (!W_ERROR_IS_OK(result)) {
561 DEBUG(0, ("winreg_printer_write_date: Could not set value %s: %s\n",
562 wvalue.name, win_errstr(result)));
565 return result;
568 static WERROR winreg_printer_ver_to_dword(const char *str, uint64_t *data)
570 unsigned int v1, v2, v3, v4;
572 if (sscanf(str, "%u.%u.%u.%u", &v1, &v2, &v3, &v4) != 4) {
573 return WERR_INVALID_PARAMETER;
576 *data = ((uint64_t)(v1 & 0xFFFF) << 48) +
577 ((uint64_t)(v2 & 0xFFFF) << 32) +
578 ((uint64_t)(v3 & 0xFFFF) << 16) +
579 (uint64_t)(v2 & 0xFFFF);
581 return WERR_OK;
584 /********************************************************************
585 Public winreg function for spoolss
586 ********************************************************************/
588 WERROR winreg_create_printer(TALLOC_CTX *mem_ctx,
589 struct dcerpc_binding_handle *winreg_handle,
590 const char *sharename)
592 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
593 struct policy_handle hive_hnd, key_hnd;
594 struct spoolss_SetPrinterInfo2 *info2;
595 struct security_descriptor *secdesc;
596 struct winreg_String wkey, wkeyclass;
597 const char *path;
598 const char *subkeys[] = { SPOOL_DSDRIVER_KEY, SPOOL_DSSPOOLER_KEY, SPOOL_PRINTERDATA_KEY };
599 uint32_t i, count = ARRAY_SIZE(subkeys);
600 uint32_t info2_mask = 0;
601 WERROR result = WERR_OK;
602 WERROR ignore;
603 TALLOC_CTX *tmp_ctx;
605 tmp_ctx = talloc_stackframe();
606 if (tmp_ctx == NULL) {
607 return WERR_NOMEM;
610 path = winreg_printer_data_keyname(tmp_ctx, sharename);
611 if (path == NULL) {
612 TALLOC_FREE(tmp_ctx);
613 return WERR_NOMEM;
616 ZERO_STRUCT(hive_hnd);
617 ZERO_STRUCT(key_hnd);
619 result = winreg_printer_openkey(tmp_ctx,
620 winreg_handle,
621 path,
623 false,
624 access_mask,
625 &hive_hnd,
626 &key_hnd);
627 if (W_ERROR_IS_OK(result)) {
628 DEBUG(2, ("winreg_create_printer: Skipping, %s already exists\n", path));
629 goto done;
630 } else if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
631 DEBUG(2, ("winreg_create_printer: Creating default values in %s\n", path));
632 } else if (!W_ERROR_IS_OK(result)) {
633 DEBUG(0, ("winreg_create_printer: Could not open key %s: %s\n",
634 path, win_errstr(result)));
635 goto done;
638 if (is_valid_policy_hnd(&key_hnd)) {
639 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
641 if (is_valid_policy_hnd(&hive_hnd)) {
642 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
645 /* Create the main key */
646 result = winreg_printer_openkey(tmp_ctx,
647 winreg_handle,
648 path,
650 true,
651 access_mask,
652 &hive_hnd,
653 &key_hnd);
654 if (!W_ERROR_IS_OK(result)) {
655 DEBUG(0, ("winreg_create_printer_keys: Could not create key %s: %s\n",
656 path, win_errstr(result)));
657 goto done;
660 if (is_valid_policy_hnd(&key_hnd)) {
661 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &result);
664 /* Create subkeys */
665 for (i = 0; i < count; i++) {
666 NTSTATUS status;
667 enum winreg_CreateAction action = REG_ACTION_NONE;
669 ZERO_STRUCT(key_hnd);
670 ZERO_STRUCT(wkey);
672 wkey.name = talloc_asprintf(tmp_ctx, "%s\\%s", path, subkeys[i]);
673 if (wkey.name == NULL) {
674 result = WERR_NOMEM;
675 goto done;
678 ZERO_STRUCT(wkeyclass);
679 wkeyclass.name = "";
681 status = dcerpc_winreg_CreateKey(winreg_handle,
682 tmp_ctx,
683 &hive_hnd,
684 wkey,
685 wkeyclass,
687 access_mask,
688 NULL,
689 &key_hnd,
690 &action,
691 &result);
692 if (!NT_STATUS_IS_OK(status)) {
693 result = ntstatus_to_werror(status);
695 if (!W_ERROR_IS_OK(result)) {
696 DEBUG(0, ("winreg_create_printer_keys: Could not create key %s: %s\n",
697 wkey.name, win_errstr(result)));
698 goto done;
701 if (strequal(subkeys[i], SPOOL_DSSPOOLER_KEY)) {
702 const char *dnssuffix;
703 const char *longname;
704 const char *uncname;
706 status = dcerpc_winreg_set_sz(tmp_ctx,
707 winreg_handle,
708 &key_hnd,
709 SPOOL_REG_PRINTERNAME,
710 sharename,
711 &result);
712 if (!NT_STATUS_IS_OK(status)) {
713 result = ntstatus_to_werror(status);
715 if (!W_ERROR_IS_OK(result)) {
716 goto done;
719 status = dcerpc_winreg_set_sz(tmp_ctx,
720 winreg_handle,
721 &key_hnd,
722 SPOOL_REG_SHORTSERVERNAME,
723 lp_netbios_name(),
724 &result);
725 if (!NT_STATUS_IS_OK(status)) {
726 result = ntstatus_to_werror(status);
728 if (!W_ERROR_IS_OK(result)) {
729 goto done;
732 /* We make the assumption that the netbios name
733 * is the same as the DNS name since the former
734 * will be what we used to join the domain
736 dnssuffix = get_mydnsdomname(tmp_ctx);
737 if (dnssuffix != NULL && dnssuffix[0] != '\0') {
738 longname = talloc_asprintf(tmp_ctx, "%s.%s", lp_netbios_name(), dnssuffix);
739 } else {
740 longname = talloc_strdup(tmp_ctx, lp_netbios_name());
742 if (longname == NULL) {
743 result = WERR_NOMEM;
744 goto done;
747 status = dcerpc_winreg_set_sz(tmp_ctx,
748 winreg_handle,
749 &key_hnd,
750 SPOOL_REG_SERVERNAME,
751 longname,
752 &result);
753 if (!NT_STATUS_IS_OK(status)) {
754 result = ntstatus_to_werror(status);
756 if (!W_ERROR_IS_OK(result)) {
757 goto done;
760 uncname = talloc_asprintf(tmp_ctx, "\\\\%s\\%s",
761 longname, sharename);
762 if (uncname == NULL) {
763 result = WERR_NOMEM;
764 goto done;
767 status = dcerpc_winreg_set_sz(tmp_ctx,
768 winreg_handle,
769 &key_hnd,
770 SPOOL_REG_UNCNAME,
771 uncname,
772 &result);
773 if (!NT_STATUS_IS_OK(status)) {
774 result = ntstatus_to_werror(status);
776 if (!W_ERROR_IS_OK(result)) {
777 goto done;
780 status = dcerpc_winreg_set_dword(tmp_ctx,
781 winreg_handle,
782 &key_hnd,
783 SPOOL_REG_VERSIONNUMBER,
785 &result);
786 if (!NT_STATUS_IS_OK(status)) {
787 result = ntstatus_to_werror(status);
789 if (!W_ERROR_IS_OK(result)) {
790 goto done;
793 status = dcerpc_winreg_set_dword(tmp_ctx,
794 winreg_handle,
795 &key_hnd,
796 SPOOL_REG_PRINTSTARTTIME,
798 &result);
799 if (!NT_STATUS_IS_OK(status)) {
800 result = ntstatus_to_werror(status);
802 if (!W_ERROR_IS_OK(result)) {
803 goto done;
806 status = dcerpc_winreg_set_dword(tmp_ctx,
807 winreg_handle,
808 &key_hnd,
809 SPOOL_REG_PRINTENDTIME,
811 &result);
812 if (!NT_STATUS_IS_OK(status)) {
813 result = ntstatus_to_werror(status);
815 if (!W_ERROR_IS_OK(result)) {
816 goto done;
819 status = dcerpc_winreg_set_dword(tmp_ctx,
820 winreg_handle,
821 &key_hnd,
822 SPOOL_REG_PRIORITY,
824 &result);
825 if (!NT_STATUS_IS_OK(status)) {
826 result = ntstatus_to_werror(status);
828 if (!W_ERROR_IS_OK(result)) {
829 goto done;
832 status = dcerpc_winreg_set_dword(tmp_ctx,
833 winreg_handle,
834 &key_hnd,
835 SPOOL_REG_PRINTKEEPPRINTEDJOBS,
837 &result);
838 if (!NT_STATUS_IS_OK(status)) {
839 result = ntstatus_to_werror(status);
841 if (!W_ERROR_IS_OK(result)) {
842 goto done;
846 if (is_valid_policy_hnd(&key_hnd)) {
847 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &result);
850 info2 = talloc_zero(tmp_ctx, struct spoolss_SetPrinterInfo2);
851 if (info2 == NULL) {
852 result = WERR_NOMEM;
853 goto done;
856 info2->printername = sharename;
857 if (info2->printername == NULL) {
858 result = WERR_NOMEM;
859 goto done;
861 info2_mask |= SPOOLSS_PRINTER_INFO_PRINTERNAME;
863 info2->sharename = sharename;
864 info2_mask |= SPOOLSS_PRINTER_INFO_SHARENAME;
866 info2->portname = SAMBA_PRINTER_PORT_NAME;
867 info2_mask |= SPOOLSS_PRINTER_INFO_PORTNAME;
869 info2->printprocessor = "winprint";
870 info2_mask |= SPOOLSS_PRINTER_INFO_PRINTPROCESSOR;
872 info2->datatype = "RAW";
873 info2_mask |= SPOOLSS_PRINTER_INFO_DATATYPE;
875 info2->comment = "";
876 info2_mask |= SPOOLSS_PRINTER_INFO_COMMENT;
878 info2->attributes = PRINTER_ATTRIBUTE_SAMBA;
879 info2_mask |= SPOOLSS_PRINTER_INFO_ATTRIBUTES;
881 info2->starttime = 0; /* Minutes since 12:00am GMT */
882 info2_mask |= SPOOLSS_PRINTER_INFO_STARTTIME;
884 info2->untiltime = 0; /* Minutes since 12:00am GMT */
885 info2_mask |= SPOOLSS_PRINTER_INFO_UNTILTIME;
887 info2->priority = 1;
888 info2_mask |= SPOOLSS_PRINTER_INFO_PRIORITY;
890 info2->defaultpriority = 1;
891 info2_mask |= SPOOLSS_PRINTER_INFO_DEFAULTPRIORITY;
893 result = spoolss_create_default_secdesc(tmp_ctx, &secdesc);
894 if (!W_ERROR_IS_OK(result)) {
895 goto done;
897 info2_mask |= SPOOLSS_PRINTER_INFO_SECDESC;
900 * Don't write a default Device Mode to the registry! The Device Mode is
901 * only written to disk with a SetPrinter level 2 or 8.
904 result = winreg_update_printer(tmp_ctx,
905 winreg_handle,
906 sharename,
907 info2_mask,
908 info2,
909 NULL,
910 secdesc);
912 done:
913 if (is_valid_policy_hnd(&key_hnd)) {
914 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
916 if (is_valid_policy_hnd(&hive_hnd)) {
917 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
920 talloc_free(tmp_ctx);
921 return result;
924 WERROR winreg_update_printer(TALLOC_CTX *mem_ctx,
925 struct dcerpc_binding_handle *winreg_handle,
926 const char *sharename,
927 uint32_t info2_mask,
928 struct spoolss_SetPrinterInfo2 *info2,
929 struct spoolss_DeviceMode *devmode,
930 struct security_descriptor *secdesc)
932 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
933 struct policy_handle hive_hnd, key_hnd;
934 int snum = lp_servicenumber(sharename);
935 enum ndr_err_code ndr_err;
936 DATA_BLOB blob;
937 char *path;
938 WERROR result = WERR_OK;
939 NTSTATUS status;
940 TALLOC_CTX *tmp_ctx;
942 tmp_ctx = talloc_stackframe();
943 if (tmp_ctx == NULL) {
944 return WERR_NOMEM;
947 path = winreg_printer_data_keyname(tmp_ctx, sharename);
948 if (path == NULL) {
949 TALLOC_FREE(tmp_ctx);
950 return WERR_NOMEM;
953 ZERO_STRUCT(hive_hnd);
954 ZERO_STRUCT(key_hnd);
956 result = winreg_printer_openkey(tmp_ctx,
957 winreg_handle,
958 path,
960 true,
961 access_mask,
962 &hive_hnd,
963 &key_hnd);
964 if (!W_ERROR_IS_OK(result)) {
965 DEBUG(0, ("winreg_update_printer: Could not open key %s: %s\n",
966 path, win_errstr(result)));
967 goto done;
970 if (info2_mask & SPOOLSS_PRINTER_INFO_ATTRIBUTES) {
971 status = dcerpc_winreg_set_dword(tmp_ctx,
972 winreg_handle,
973 &key_hnd,
974 "Attributes",
975 info2->attributes,
976 &result);
977 if (!NT_STATUS_IS_OK(status)) {
978 result = ntstatus_to_werror(status);
980 if (!W_ERROR_IS_OK(result)) {
981 goto done;
985 #if 0
986 if (info2_mask & SPOOLSS_PRINTER_INFO_AVERAGEPPM) {
987 status = dcerpc_winreg_set_dword(tmp_ctx,
988 winreg_handle,
989 &key_hnd,
990 "AveragePpm",
991 info2->attributes,
992 &result);
993 if (!NT_STATUS_IS_OK(status)) {
994 result = ntstatus_to_werror(status);
996 if (!W_ERROR_IS_OK(result)) {
997 goto done;
1000 #endif
1002 if (info2_mask & SPOOLSS_PRINTER_INFO_COMMENT) {
1003 status = dcerpc_winreg_set_sz(tmp_ctx,
1004 winreg_handle,
1005 &key_hnd,
1006 "Description",
1007 info2->comment,
1008 &result);
1009 if (!NT_STATUS_IS_OK(status)) {
1010 result = ntstatus_to_werror(status);
1012 if (!W_ERROR_IS_OK(result)) {
1013 goto done;
1017 if (info2_mask & SPOOLSS_PRINTER_INFO_DATATYPE) {
1018 status = dcerpc_winreg_set_sz(tmp_ctx,
1019 winreg_handle,
1020 &key_hnd,
1021 "Datatype",
1022 info2->datatype,
1023 &result);
1024 if (!NT_STATUS_IS_OK(status)) {
1025 result = ntstatus_to_werror(status);
1027 if (!W_ERROR_IS_OK(result)) {
1028 goto done;
1032 if (info2_mask & SPOOLSS_PRINTER_INFO_DEFAULTPRIORITY) {
1033 status = dcerpc_winreg_set_dword(tmp_ctx,
1034 winreg_handle,
1035 &key_hnd,
1036 "Default Priority",
1037 info2->defaultpriority,
1038 &result);
1039 if (!NT_STATUS_IS_OK(status)) {
1040 result = ntstatus_to_werror(status);
1042 if (!W_ERROR_IS_OK(result)) {
1043 goto done;
1047 if (info2_mask & SPOOLSS_PRINTER_INFO_DEVMODE) {
1049 * Some client drivers freak out if there is a NULL devmode
1050 * (probably the driver is not checking before accessing
1051 * the devmode pointer) --jerry
1053 if (devmode == NULL && lp_default_devmode(snum) && info2 != NULL) {
1054 result = spoolss_create_default_devmode(tmp_ctx,
1055 info2->printername,
1056 &devmode);
1057 if (!W_ERROR_IS_OK(result)) {
1058 goto done;
1062 if (devmode->size != (ndr_size_spoolss_DeviceMode(devmode, 0) - devmode->__driverextra_length)) {
1063 result = WERR_INVALID_PARAM;
1064 goto done;
1067 ndr_err = ndr_push_struct_blob(&blob, tmp_ctx, devmode,
1068 (ndr_push_flags_fn_t) ndr_push_spoolss_DeviceMode);
1069 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1070 DEBUG(0, ("winreg_update_printer: Failed to marshall device mode\n"));
1071 result = WERR_NOMEM;
1072 goto done;
1075 status = dcerpc_winreg_set_binary(tmp_ctx,
1076 winreg_handle,
1077 &key_hnd,
1078 "Default DevMode",
1079 &blob,
1080 &result);
1081 if (!NT_STATUS_IS_OK(status)) {
1082 result = ntstatus_to_werror(status);
1084 if (!W_ERROR_IS_OK(result)) {
1085 goto done;
1089 if (info2_mask & SPOOLSS_PRINTER_INFO_DRIVERNAME) {
1090 status = dcerpc_winreg_set_sz(tmp_ctx,
1091 winreg_handle,
1092 &key_hnd,
1093 "Printer Driver",
1094 info2->drivername,
1095 &result);
1096 if (!NT_STATUS_IS_OK(status)) {
1097 result = ntstatus_to_werror(status);
1099 if (!W_ERROR_IS_OK(result)) {
1100 goto done;
1104 if (info2_mask & SPOOLSS_PRINTER_INFO_LOCATION) {
1105 status = dcerpc_winreg_set_sz(tmp_ctx,
1106 winreg_handle,
1107 &key_hnd,
1108 "Location",
1109 info2->location,
1110 &result);
1111 if (!NT_STATUS_IS_OK(status)) {
1112 result = ntstatus_to_werror(status);
1114 if (!W_ERROR_IS_OK(result)) {
1115 goto done;
1119 if (info2_mask & SPOOLSS_PRINTER_INFO_PARAMETERS) {
1120 status = dcerpc_winreg_set_sz(tmp_ctx,
1121 winreg_handle,
1122 &key_hnd,
1123 "Parameters",
1124 info2->parameters,
1125 &result);
1126 if (!NT_STATUS_IS_OK(status)) {
1127 result = ntstatus_to_werror(status);
1129 if (!W_ERROR_IS_OK(result)) {
1130 goto done;
1134 if (info2_mask & SPOOLSS_PRINTER_INFO_PORTNAME) {
1135 status = dcerpc_winreg_set_sz(tmp_ctx,
1136 winreg_handle,
1137 &key_hnd,
1138 "Port",
1139 info2->portname,
1140 &result);
1141 if (!NT_STATUS_IS_OK(status)) {
1142 result = ntstatus_to_werror(status);
1144 if (!W_ERROR_IS_OK(result)) {
1145 goto done;
1149 if (info2_mask & SPOOLSS_PRINTER_INFO_PRINTERNAME) {
1151 * in addprinter: no servername and the printer is the name
1152 * in setprinter: servername is \\server
1153 * and printer is \\server\\printer
1155 * Samba manages only local printers.
1156 * we currently don't support things like i
1157 * path=\\other_server\printer
1159 * We only store the printername, not \\server\printername
1161 const char *p = strrchr(info2->printername, '\\');
1162 if (p == NULL) {
1163 p = info2->printername;
1164 } else {
1165 p++;
1167 status = dcerpc_winreg_set_sz(tmp_ctx,
1168 winreg_handle,
1169 &key_hnd,
1170 "Name",
1172 &result);
1173 if (!NT_STATUS_IS_OK(status)) {
1174 result = ntstatus_to_werror(status);
1176 if (!W_ERROR_IS_OK(result)) {
1177 goto done;
1181 if (info2_mask & SPOOLSS_PRINTER_INFO_PRINTPROCESSOR) {
1182 status = dcerpc_winreg_set_sz(tmp_ctx,
1183 winreg_handle,
1184 &key_hnd,
1185 "Print Processor",
1186 info2->printprocessor,
1187 &result);
1188 if (!NT_STATUS_IS_OK(status)) {
1189 result = ntstatus_to_werror(status);
1191 if (!W_ERROR_IS_OK(result)) {
1192 goto done;
1196 if (info2_mask & SPOOLSS_PRINTER_INFO_PRIORITY) {
1197 status = dcerpc_winreg_set_dword(tmp_ctx,
1198 winreg_handle,
1199 &key_hnd,
1200 "Priority",
1201 info2->priority,
1202 &result);
1203 if (!NT_STATUS_IS_OK(status)) {
1204 result = ntstatus_to_werror(status);
1206 if (!W_ERROR_IS_OK(result)) {
1207 goto done;
1211 if (info2_mask & SPOOLSS_PRINTER_INFO_SECDESC) {
1213 * We need a security descriptor, if it isn't specified by
1214 * AddPrinter{Ex} then create a default descriptor.
1216 if (secdesc == NULL) {
1217 result = spoolss_create_default_secdesc(tmp_ctx, &secdesc);
1218 if (!W_ERROR_IS_OK(result)) {
1219 goto done;
1222 result = winreg_set_printer_secdesc(tmp_ctx,
1223 winreg_handle,
1224 sharename,
1225 secdesc);
1226 if (!W_ERROR_IS_OK(result)) {
1227 goto done;
1231 if (info2_mask & SPOOLSS_PRINTER_INFO_SEPFILE) {
1232 status = dcerpc_winreg_set_sz(tmp_ctx,
1233 winreg_handle,
1234 &key_hnd,
1235 "Separator File",
1236 info2->sepfile,
1237 &result);
1238 if (!NT_STATUS_IS_OK(status)) {
1239 result = ntstatus_to_werror(status);
1241 if (!W_ERROR_IS_OK(result)) {
1242 goto done;
1246 if (info2_mask & SPOOLSS_PRINTER_INFO_SHARENAME) {
1247 status = dcerpc_winreg_set_sz(tmp_ctx,
1248 winreg_handle,
1249 &key_hnd,
1250 "Share Name",
1251 info2->sharename,
1252 &result);
1253 if (!NT_STATUS_IS_OK(status)) {
1254 result = ntstatus_to_werror(status);
1256 if (!W_ERROR_IS_OK(result)) {
1257 goto done;
1261 if (info2_mask & SPOOLSS_PRINTER_INFO_STARTTIME) {
1262 status = dcerpc_winreg_set_dword(tmp_ctx,
1263 winreg_handle,
1264 &key_hnd,
1265 "StartTime",
1266 info2->starttime,
1267 &result);
1268 if (!NT_STATUS_IS_OK(status)) {
1269 result = ntstatus_to_werror(status);
1271 if (!W_ERROR_IS_OK(result)) {
1272 goto done;
1276 if (info2_mask & SPOOLSS_PRINTER_INFO_STATUS) {
1277 status = dcerpc_winreg_set_dword(tmp_ctx,
1278 winreg_handle,
1279 &key_hnd,
1280 "Status",
1281 info2->status,
1282 &result);
1283 if (!NT_STATUS_IS_OK(status)) {
1284 result = ntstatus_to_werror(status);
1286 if (!W_ERROR_IS_OK(result)) {
1287 goto done;
1291 if (info2_mask & SPOOLSS_PRINTER_INFO_UNTILTIME) {
1292 status = dcerpc_winreg_set_dword(tmp_ctx,
1293 winreg_handle,
1294 &key_hnd,
1295 "UntilTime",
1296 info2->untiltime,
1297 &result);
1298 if (!NT_STATUS_IS_OK(status)) {
1299 result = ntstatus_to_werror(status);
1301 if (!W_ERROR_IS_OK(result)) {
1302 goto done;
1306 status = dcerpc_winreg_set_dword(tmp_ctx,
1307 winreg_handle,
1308 &key_hnd,
1309 "ChangeID",
1310 winreg_printer_rev_changeid(),
1311 &result);
1312 if (!NT_STATUS_IS_OK(status)) {
1313 result = ntstatus_to_werror(status);
1315 if (!W_ERROR_IS_OK(result)) {
1316 goto done;
1319 result = WERR_OK;
1320 done:
1321 if (winreg_handle != NULL) {
1322 WERROR ignore;
1324 if (is_valid_policy_hnd(&key_hnd)) {
1325 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
1327 if (is_valid_policy_hnd(&hive_hnd)) {
1328 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
1332 TALLOC_FREE(tmp_ctx);
1333 return result;
1336 WERROR winreg_get_printer(TALLOC_CTX *mem_ctx,
1337 struct dcerpc_binding_handle *winreg_handle,
1338 const char *printer,
1339 struct spoolss_PrinterInfo2 **pinfo2)
1341 struct spoolss_PrinterInfo2 *info2;
1342 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1343 struct policy_handle hive_hnd, key_hnd;
1344 struct spoolss_PrinterEnumValues enum_value;
1345 struct spoolss_PrinterEnumValues *v = NULL;
1346 enum ndr_err_code ndr_err;
1347 DATA_BLOB blob;
1348 int snum = lp_servicenumber(printer);
1349 uint32_t num_values = 0;
1350 uint32_t i;
1351 char *path;
1352 NTSTATUS status;
1353 WERROR result = WERR_OK;
1354 const char **enum_names = NULL;
1355 enum winreg_Type *enum_types = NULL;
1356 DATA_BLOB *enum_data_blobs = NULL;
1357 TALLOC_CTX *tmp_ctx;
1359 tmp_ctx = talloc_stackframe();
1360 if (tmp_ctx == NULL) {
1361 return WERR_NOMEM;
1364 path = winreg_printer_data_keyname(tmp_ctx, printer);
1365 if (path == NULL) {
1366 TALLOC_FREE(tmp_ctx);
1367 return WERR_NOMEM;
1370 result = winreg_printer_openkey(tmp_ctx,
1371 winreg_handle,
1372 path,
1374 false,
1375 access_mask,
1376 &hive_hnd,
1377 &key_hnd);
1378 if (!W_ERROR_IS_OK(result)) {
1379 DEBUG(2, ("winreg_get_printer: Could not open key %s: %s\n",
1380 path, win_errstr(result)));
1381 goto done;
1384 status = dcerpc_winreg_enumvals(tmp_ctx,
1385 winreg_handle,
1386 &key_hnd,
1387 &num_values,
1388 &enum_names,
1389 &enum_types,
1390 &enum_data_blobs,
1391 &result);
1392 if (!NT_STATUS_IS_OK(status)){
1393 result = ntstatus_to_werror(status);
1396 if (!W_ERROR_IS_OK(result)) {
1397 DEBUG(0, ("winreg_get_printer: Could not enumerate values in %s: %s\n",
1398 path, win_errstr(result)));
1399 goto done;
1402 info2 = talloc_zero(tmp_ctx, struct spoolss_PrinterInfo2);
1403 if (info2 == NULL) {
1404 result = WERR_NOMEM;
1405 goto done;
1408 FILL_STRING(info2, EMPTY_STRING, info2->servername);
1409 FILL_STRING(info2, EMPTY_STRING, info2->printername);
1410 FILL_STRING(info2, EMPTY_STRING, info2->sharename);
1411 FILL_STRING(info2, EMPTY_STRING, info2->portname);
1412 FILL_STRING(info2, EMPTY_STRING, info2->drivername);
1413 FILL_STRING(info2, EMPTY_STRING, info2->comment);
1414 FILL_STRING(info2, EMPTY_STRING, info2->location);
1415 FILL_STRING(info2, EMPTY_STRING, info2->sepfile);
1416 FILL_STRING(info2, EMPTY_STRING, info2->printprocessor);
1417 FILL_STRING(info2, EMPTY_STRING, info2->datatype);
1418 FILL_STRING(info2, EMPTY_STRING, info2->parameters);
1420 for (i = 0; i < num_values; i++) {
1421 enum_value.value_name = enum_names[i];
1422 enum_value.value_name_len = 2*strlen_m_term(enum_names[i]);
1423 enum_value.type = enum_types[i];
1424 enum_value.data_length = enum_data_blobs[i].length;
1425 enum_value.data = NULL;
1426 if (enum_value.data_length != 0){
1427 enum_value.data = &enum_data_blobs[i];
1429 v = &enum_value;
1431 result = winreg_enumval_to_sz(info2,
1433 "Name",
1434 &info2->printername);
1435 CHECK_ERROR(result);
1437 result = winreg_enumval_to_sz(info2,
1439 "Share Name",
1440 &info2->sharename);
1441 CHECK_ERROR(result);
1443 result = winreg_enumval_to_sz(info2,
1445 "Port",
1446 &info2->portname);
1447 CHECK_ERROR(result);
1449 result = winreg_enumval_to_sz(info2,
1451 "Description",
1452 &info2->comment);
1453 CHECK_ERROR(result);
1455 result = winreg_enumval_to_sz(info2,
1457 "Location",
1458 &info2->location);
1459 CHECK_ERROR(result);
1461 result = winreg_enumval_to_sz(info2,
1463 "Separator File",
1464 &info2->sepfile);
1465 CHECK_ERROR(result);
1467 result = winreg_enumval_to_sz(info2,
1469 "Print Processor",
1470 &info2->printprocessor);
1471 CHECK_ERROR(result);
1473 result = winreg_enumval_to_sz(info2,
1475 "Datatype",
1476 &info2->datatype);
1477 CHECK_ERROR(result);
1479 result = winreg_enumval_to_sz(info2,
1481 "Parameters",
1482 &info2->parameters);
1483 CHECK_ERROR(result);
1485 result = winreg_enumval_to_sz(info2,
1487 "Printer Driver",
1488 &info2->drivername);
1489 CHECK_ERROR(result);
1491 result = winreg_enumval_to_dword(info2,
1493 "Attributes",
1494 &info2->attributes);
1495 CHECK_ERROR(result);
1497 result = winreg_enumval_to_dword(info2,
1499 "Priority",
1500 &info2->priority);
1501 CHECK_ERROR(result);
1503 result = winreg_enumval_to_dword(info2,
1505 "Default Priority",
1506 &info2->defaultpriority);
1507 CHECK_ERROR(result);
1509 result = winreg_enumval_to_dword(info2,
1511 "StartTime",
1512 &info2->starttime);
1513 CHECK_ERROR(result);
1515 result = winreg_enumval_to_dword(info2,
1517 "UntilTime",
1518 &info2->untiltime);
1519 CHECK_ERROR(result);
1521 result = winreg_enumval_to_dword(info2,
1523 "Status",
1524 &info2->status);
1525 CHECK_ERROR(result);
1527 result = winreg_enumval_to_dword(info2,
1529 "StartTime",
1530 &info2->starttime);
1531 CHECK_ERROR(result);
1534 if (!W_ERROR_IS_OK(result)) {
1535 DEBUG(0, ("winreg_get_printer: winreg_enumval_to_TYPE() failed "
1536 "for %s: %s\n",
1537 v->value_name,
1538 win_errstr(result)));
1539 goto done;
1542 /* Construct the Device Mode */
1543 status = dcerpc_winreg_query_binary(tmp_ctx,
1544 winreg_handle,
1545 &key_hnd,
1546 "Default DevMode",
1547 &blob,
1548 &result);
1549 if (!NT_STATUS_IS_OK(status)) {
1550 result = ntstatus_to_werror(status);
1552 if (W_ERROR_IS_OK(result)) {
1553 info2->devmode = talloc_zero(info2, struct spoolss_DeviceMode);
1554 if (info2->devmode == NULL) {
1555 result = WERR_NOMEM;
1556 goto done;
1558 ndr_err = ndr_pull_struct_blob(&blob,
1559 info2->devmode,
1560 info2->devmode,
1561 (ndr_pull_flags_fn_t) ndr_pull_spoolss_DeviceMode);
1562 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1563 DEBUG(0, ("winreg_get_printer: Failed to unmarshall device mode\n"));
1564 result = WERR_NOMEM;
1565 goto done;
1569 if (info2->devmode == NULL && lp_default_devmode(snum)) {
1570 result = spoolss_create_default_devmode(info2,
1571 info2->printername,
1572 &info2->devmode);
1573 if (!W_ERROR_IS_OK(result)) {
1574 goto done;
1578 if (info2->devmode) {
1579 info2->devmode->size = ndr_size_spoolss_DeviceMode(info2->devmode, 0) - info2->devmode->driverextra_data.length;
1582 result = winreg_get_printer_secdesc(info2,
1583 winreg_handle,
1584 printer,
1585 &info2->secdesc);
1586 if (!W_ERROR_IS_OK(result)) {
1587 goto done;
1590 /* Fix for OS/2 drivers. */
1591 if (get_remote_arch() == RA_OS2) {
1592 spoolss_map_to_os2_driver(info2, &info2->drivername);
1595 if (pinfo2) {
1596 *pinfo2 = talloc_move(mem_ctx, &info2);
1599 result = WERR_OK;
1600 done:
1601 if (winreg_handle != NULL) {
1602 WERROR ignore;
1604 if (is_valid_policy_hnd(&key_hnd)) {
1605 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
1607 if (is_valid_policy_hnd(&hive_hnd)) {
1608 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
1612 TALLOC_FREE(tmp_ctx);
1613 return result;
1616 WERROR winreg_get_printer_secdesc(TALLOC_CTX *mem_ctx,
1617 struct dcerpc_binding_handle *winreg_handle,
1618 const char *sharename,
1619 struct spoolss_security_descriptor **psecdesc)
1621 struct spoolss_security_descriptor *secdesc;
1622 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1623 struct policy_handle hive_hnd, key_hnd;
1624 const char *path;
1625 TALLOC_CTX *tmp_ctx;
1626 NTSTATUS status;
1627 WERROR result;
1629 tmp_ctx = talloc_stackframe();
1630 if (tmp_ctx == NULL) {
1631 return WERR_NOMEM;
1634 path = winreg_printer_data_keyname(tmp_ctx, sharename);
1635 if (path == NULL) {
1636 talloc_free(tmp_ctx);
1637 return WERR_NOMEM;
1640 ZERO_STRUCT(hive_hnd);
1641 ZERO_STRUCT(key_hnd);
1643 result = winreg_printer_openkey(tmp_ctx,
1644 winreg_handle,
1645 path,
1647 false,
1648 access_mask,
1649 &hive_hnd,
1650 &key_hnd);
1651 if (!W_ERROR_IS_OK(result)) {
1652 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
1653 goto create_default;
1655 goto done;
1658 status = dcerpc_winreg_query_sd(tmp_ctx,
1659 winreg_handle,
1660 &key_hnd,
1661 "Security",
1662 &secdesc,
1663 &result);
1664 if (!NT_STATUS_IS_OK(status)) {
1665 result = ntstatus_to_werror(status);
1667 if (!W_ERROR_IS_OK(result)) {
1668 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
1669 WERROR ignore;
1671 if (is_valid_policy_hnd(&key_hnd)) {
1672 dcerpc_winreg_CloseKey(winreg_handle,
1673 tmp_ctx,
1674 &key_hnd,
1675 &ignore);
1678 if (is_valid_policy_hnd(&hive_hnd)) {
1679 dcerpc_winreg_CloseKey(winreg_handle,
1680 tmp_ctx,
1681 &hive_hnd,
1682 &ignore);
1684 goto create_default;
1686 goto done;
1689 if (psecdesc) {
1690 *psecdesc = talloc_move(mem_ctx, &secdesc);
1693 result = WERR_OK;
1694 goto done;
1696 create_default:
1697 result = winreg_printer_openkey(tmp_ctx,
1698 winreg_handle,
1699 path,
1701 true,
1702 access_mask,
1703 &hive_hnd,
1704 &key_hnd);
1705 if (!W_ERROR_IS_OK(result)) {
1706 goto done;
1709 result = spoolss_create_default_secdesc(tmp_ctx, &secdesc);
1710 if (!W_ERROR_IS_OK(result)) {
1711 goto done;
1714 /* If security descriptor is owned by S-1-1-0 and winbindd is up,
1715 this security descriptor has been created when winbindd was
1716 down. Take ownership of security descriptor. */
1717 if (dom_sid_equal(secdesc->owner_sid, &global_sid_World)) {
1718 struct dom_sid owner_sid;
1720 /* Change sd owner to workgroup administrator */
1722 if (secrets_fetch_domain_sid(lp_workgroup(), &owner_sid)) {
1723 struct spoolss_security_descriptor *new_secdesc;
1724 size_t size;
1726 /* Create new sd */
1727 sid_append_rid(&owner_sid, DOMAIN_RID_ADMINISTRATOR);
1729 new_secdesc = make_sec_desc(tmp_ctx,
1730 secdesc->revision,
1731 secdesc->type,
1732 &owner_sid,
1733 secdesc->group_sid,
1734 secdesc->sacl,
1735 secdesc->dacl,
1736 &size);
1738 if (new_secdesc == NULL) {
1739 result = WERR_NOMEM;
1740 goto done;
1743 /* Swap with other one */
1744 secdesc = new_secdesc;
1748 status = dcerpc_winreg_set_sd(tmp_ctx,
1749 winreg_handle,
1750 &key_hnd,
1751 "Security",
1752 secdesc,
1753 &result);
1754 if (!NT_STATUS_IS_OK(status)) {
1755 result = ntstatus_to_werror(status);
1757 if (!W_ERROR_IS_OK(result)) {
1758 return result;
1761 if (psecdesc) {
1762 *psecdesc = talloc_move(mem_ctx, &secdesc);
1765 result = WERR_OK;
1766 done:
1767 if (winreg_handle != NULL) {
1768 WERROR ignore;
1770 if (is_valid_policy_hnd(&key_hnd)) {
1771 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
1773 if (is_valid_policy_hnd(&hive_hnd)) {
1774 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
1778 talloc_free(tmp_ctx);
1779 return result;
1782 WERROR winreg_set_printer_secdesc(TALLOC_CTX *mem_ctx,
1783 struct dcerpc_binding_handle *winreg_handle,
1784 const char *sharename,
1785 const struct spoolss_security_descriptor *secdesc)
1787 const struct spoolss_security_descriptor *new_secdesc = secdesc;
1788 struct spoolss_security_descriptor *old_secdesc;
1789 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1790 struct policy_handle hive_hnd, key_hnd;
1791 const char *path;
1792 TALLOC_CTX *tmp_ctx;
1793 NTSTATUS status;
1794 WERROR result;
1796 tmp_ctx = talloc_stackframe();
1797 if (tmp_ctx == NULL) {
1798 return WERR_NOMEM;
1801 path = winreg_printer_data_keyname(tmp_ctx, sharename);
1802 if (path == NULL) {
1803 talloc_free(tmp_ctx);
1804 return WERR_NOMEM;
1808 * The old owner and group sids of the security descriptor are not
1809 * present when new ACEs are added or removed by changing printer
1810 * permissions through NT. If they are NULL in the new security
1811 * descriptor then copy them over from the old one.
1813 if (!secdesc->owner_sid || !secdesc->group_sid) {
1814 struct dom_sid *owner_sid, *group_sid;
1815 struct security_acl *dacl, *sacl;
1816 size_t size;
1818 result = winreg_get_printer_secdesc(tmp_ctx,
1819 winreg_handle,
1820 sharename,
1821 &old_secdesc);
1822 if (!W_ERROR_IS_OK(result)) {
1823 talloc_free(tmp_ctx);
1824 return result;
1827 /* Pick out correct owner and group sids */
1828 owner_sid = secdesc->owner_sid ?
1829 secdesc->owner_sid :
1830 old_secdesc->owner_sid;
1832 group_sid = secdesc->group_sid ?
1833 secdesc->group_sid :
1834 old_secdesc->group_sid;
1836 dacl = secdesc->dacl ?
1837 secdesc->dacl :
1838 old_secdesc->dacl;
1840 sacl = secdesc->sacl ?
1841 secdesc->sacl :
1842 old_secdesc->sacl;
1844 /* Make a deep copy of the security descriptor */
1845 new_secdesc = make_sec_desc(tmp_ctx,
1846 secdesc->revision,
1847 secdesc->type,
1848 owner_sid,
1849 group_sid,
1850 sacl,
1851 dacl,
1852 &size);
1853 if (new_secdesc == NULL) {
1854 talloc_free(tmp_ctx);
1855 return WERR_NOMEM;
1859 ZERO_STRUCT(hive_hnd);
1860 ZERO_STRUCT(key_hnd);
1862 result = winreg_printer_openkey(tmp_ctx,
1863 winreg_handle,
1864 path,
1866 false,
1867 access_mask,
1868 &hive_hnd,
1869 &key_hnd);
1870 if (!W_ERROR_IS_OK(result)) {
1871 goto done;
1874 status = dcerpc_winreg_set_sd(tmp_ctx,
1875 winreg_handle,
1876 &key_hnd,
1877 "Security",
1878 new_secdesc,
1879 &result);
1880 if (!NT_STATUS_IS_OK(status)) {
1881 result = ntstatus_to_werror(status);
1884 done:
1885 if (winreg_handle != NULL) {
1886 WERROR ignore;
1888 if (is_valid_policy_hnd(&key_hnd)) {
1889 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
1891 if (is_valid_policy_hnd(&hive_hnd)) {
1892 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
1896 talloc_free(tmp_ctx);
1897 return result;
1900 /* Set printer data over the winreg pipe. */
1901 WERROR winreg_set_printer_dataex(TALLOC_CTX *mem_ctx,
1902 struct dcerpc_binding_handle *winreg_handle,
1903 const char *printer,
1904 const char *key,
1905 const char *value,
1906 enum winreg_Type type,
1907 uint8_t *data,
1908 uint32_t data_size)
1910 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1911 struct policy_handle hive_hnd, key_hnd;
1912 struct winreg_String wvalue = { 0, };
1913 char *path;
1914 WERROR result = WERR_OK;
1915 NTSTATUS status;
1916 TALLOC_CTX *tmp_ctx;
1918 tmp_ctx = talloc_stackframe();
1919 if (tmp_ctx == NULL) {
1920 return WERR_NOMEM;
1923 path = winreg_printer_data_keyname(tmp_ctx, printer);
1924 if (path == NULL) {
1925 TALLOC_FREE(tmp_ctx);
1926 return WERR_NOMEM;
1929 ZERO_STRUCT(hive_hnd);
1930 ZERO_STRUCT(key_hnd);
1932 DEBUG(8, ("winreg_set_printer_dataex: Open printer key %s, value %s, access_mask: 0x%05x for [%s]\n",
1933 key, value, access_mask, printer));
1934 result = winreg_printer_openkey(tmp_ctx,
1935 winreg_handle,
1936 path,
1937 key,
1938 true,
1939 access_mask,
1940 &hive_hnd,
1941 &key_hnd);
1942 if (!W_ERROR_IS_OK(result)) {
1943 DEBUG(0, ("winreg_set_printer_dataex: Could not open key %s: %s\n",
1944 key, win_errstr(result)));
1945 goto done;
1948 wvalue.name = value;
1949 status = dcerpc_winreg_SetValue(winreg_handle,
1950 tmp_ctx,
1951 &key_hnd,
1952 wvalue,
1953 type,
1954 data,
1955 data_size,
1956 &result);
1957 if (!NT_STATUS_IS_OK(status)) {
1958 DEBUG(0, ("winreg_set_printer_dataex: Could not set value %s: %s\n",
1959 value, nt_errstr(status)));
1960 result = ntstatus_to_werror(status);
1963 done:
1964 if (winreg_handle != NULL) {
1965 WERROR ignore;
1967 if (is_valid_policy_hnd(&key_hnd)) {
1968 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
1970 if (is_valid_policy_hnd(&hive_hnd)) {
1971 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
1975 TALLOC_FREE(tmp_ctx);
1976 return result;
1979 /* Get printer data over a winreg pipe. */
1980 WERROR winreg_get_printer_dataex(TALLOC_CTX *mem_ctx,
1981 struct dcerpc_binding_handle *winreg_handle,
1982 const char *printer,
1983 const char *key,
1984 const char *value,
1985 enum winreg_Type *type,
1986 uint8_t **data,
1987 uint32_t *data_size)
1989 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1990 struct policy_handle hive_hnd, key_hnd;
1991 struct winreg_String wvalue;
1992 enum winreg_Type type_in = REG_NONE;
1993 char *path;
1994 uint8_t *data_in = NULL;
1995 uint32_t data_in_size = 0;
1996 uint32_t value_len = 0;
1997 WERROR result = WERR_OK;
1998 NTSTATUS status;
1999 TALLOC_CTX *tmp_ctx;
2001 tmp_ctx = talloc_stackframe();
2002 if (tmp_ctx == NULL) {
2003 return WERR_NOMEM;
2006 path = winreg_printer_data_keyname(tmp_ctx, printer);
2007 if (path == NULL) {
2008 TALLOC_FREE(tmp_ctx);
2009 return WERR_NOMEM;
2012 ZERO_STRUCT(hive_hnd);
2013 ZERO_STRUCT(key_hnd);
2015 result = winreg_printer_openkey(tmp_ctx,
2016 winreg_handle,
2017 path,
2018 key,
2019 false,
2020 access_mask,
2021 &hive_hnd,
2022 &key_hnd);
2023 if (!W_ERROR_IS_OK(result)) {
2024 DEBUG(2, ("winreg_get_printer_dataex: Could not open key %s: %s\n",
2025 key, win_errstr(result)));
2026 goto done;
2029 wvalue.name = value;
2032 * call QueryValue once with data == NULL to get the
2033 * needed memory size to be allocated, then allocate
2034 * data buffer and call again.
2036 status = dcerpc_winreg_QueryValue(winreg_handle,
2037 tmp_ctx,
2038 &key_hnd,
2039 &wvalue,
2040 &type_in,
2041 NULL,
2042 &data_in_size,
2043 &value_len,
2044 &result);
2045 if (!NT_STATUS_IS_OK(status)) {
2046 DEBUG(0, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
2047 value, nt_errstr(status)));
2048 result = ntstatus_to_werror(status);
2049 goto done;
2051 if (!W_ERROR_IS_OK(result)) {
2052 DEBUG(2, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
2053 value, win_errstr(result)));
2054 goto done;
2057 data_in = (uint8_t *) TALLOC(tmp_ctx, data_in_size);
2058 if (data_in == NULL) {
2059 result = WERR_NOMEM;
2060 goto done;
2062 value_len = 0;
2064 status = dcerpc_winreg_QueryValue(winreg_handle,
2065 tmp_ctx,
2066 &key_hnd,
2067 &wvalue,
2068 &type_in,
2069 data_in,
2070 &data_in_size,
2071 &value_len,
2072 &result);
2073 if (!NT_STATUS_IS_OK(status)) {
2074 DEBUG(0, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
2075 value, nt_errstr(status)));
2076 result = ntstatus_to_werror(status);
2077 goto done;
2079 if (!W_ERROR_IS_OK(result)) {
2080 DEBUG(2, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
2081 value, win_errstr(result)));
2082 goto done;
2085 *type = type_in;
2086 *data_size = data_in_size;
2087 if (data_in_size) {
2088 *data = talloc_move(mem_ctx, &data_in);
2091 result = WERR_OK;
2092 done:
2093 if (winreg_handle != NULL) {
2094 WERROR ignore;
2096 if (is_valid_policy_hnd(&key_hnd)) {
2097 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2099 if (is_valid_policy_hnd(&hive_hnd)) {
2100 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2104 TALLOC_FREE(tmp_ctx);
2105 return result;
2108 /* Enumerate on the values of a given key and provide the data. */
2109 WERROR winreg_enum_printer_dataex(TALLOC_CTX *mem_ctx,
2110 struct dcerpc_binding_handle *winreg_handle,
2111 const char *printer,
2112 const char *key,
2113 uint32_t *pnum_values,
2114 struct spoolss_PrinterEnumValues **penum_values)
2116 uint32_t i;
2117 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2118 struct policy_handle hive_hnd, key_hnd;
2120 struct spoolss_PrinterEnumValues *enum_values = NULL;
2121 uint32_t num_values = 0;
2122 char *path;
2123 WERROR result = WERR_OK;
2124 NTSTATUS status;
2125 const char **enum_names = NULL;
2126 enum winreg_Type *enum_types = NULL;
2127 DATA_BLOB *enum_data_blobs = NULL;
2129 TALLOC_CTX *tmp_ctx;
2131 tmp_ctx = talloc_stackframe();
2132 if (tmp_ctx == NULL) {
2133 return WERR_NOMEM;
2136 path = winreg_printer_data_keyname(tmp_ctx, printer);
2137 if (path == NULL) {
2138 TALLOC_FREE(tmp_ctx);
2139 return WERR_NOMEM;
2142 result = winreg_printer_openkey(tmp_ctx,
2143 winreg_handle,
2144 path,
2145 key,
2146 false,
2147 access_mask,
2148 &hive_hnd,
2149 &key_hnd);
2150 if (!W_ERROR_IS_OK(result)) {
2151 DEBUG(2, ("winreg_enum_printer_dataex: Could not open key %s: %s\n",
2152 key, win_errstr(result)));
2153 goto done;
2156 status = dcerpc_winreg_enumvals(tmp_ctx,
2157 winreg_handle,
2158 &key_hnd,
2159 &num_values,
2160 &enum_names,
2161 &enum_types,
2162 &enum_data_blobs,
2163 &result);
2164 if (!NT_STATUS_IS_OK(status)){
2165 result = ntstatus_to_werror(status);
2168 if (!W_ERROR_IS_OK(result)) {
2169 DEBUG(0, ("winreg_enum_printer_dataex: Could not enumerate values in %s: %s\n",
2170 key, win_errstr(result)));
2171 goto done;
2174 enum_values = talloc_array(tmp_ctx, struct spoolss_PrinterEnumValues, num_values);
2175 if (enum_values == NULL){
2176 result = WERR_NOMEM;
2177 DEBUG(0, ("winreg_enum_printer_dataex: Could not enumerate values in %s: %s\n",
2178 key, win_errstr(result)));
2179 goto done;
2182 for (i = 0; i < num_values; i++){
2183 enum_values[i].value_name = enum_names[i];
2184 enum_values[i].value_name_len = strlen_m_term(enum_names[i]) * 2;
2185 enum_values[i].type = enum_types[i];
2186 enum_values[i].data_length = enum_data_blobs[i].length;
2187 enum_values[i].data = NULL;
2189 if (enum_values[i].data_length != 0){
2190 enum_values[i].data = &enum_data_blobs[i];
2194 talloc_steal(enum_values, enum_names);
2195 talloc_steal(enum_values, enum_data_blobs);
2197 *pnum_values = num_values;
2198 if (penum_values) {
2199 *penum_values = talloc_move(mem_ctx, &enum_values);
2202 result = WERR_OK;
2203 done:
2204 if (winreg_handle != NULL) {
2205 WERROR ignore;
2207 if (is_valid_policy_hnd(&key_hnd)) {
2208 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2210 if (is_valid_policy_hnd(&hive_hnd)) {
2211 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2215 TALLOC_FREE(tmp_ctx);
2216 return result;
2219 /* Delete printer data over a winreg pipe. */
2220 WERROR winreg_delete_printer_dataex(TALLOC_CTX *mem_ctx,
2221 struct dcerpc_binding_handle *winreg_handle,
2222 const char *printer,
2223 const char *key,
2224 const char *value)
2226 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2227 struct policy_handle hive_hnd, key_hnd;
2228 struct winreg_String wvalue = { 0, };
2229 char *path;
2230 WERROR result = WERR_OK;
2231 NTSTATUS status;
2233 TALLOC_CTX *tmp_ctx;
2235 tmp_ctx = talloc_stackframe();
2236 if (tmp_ctx == NULL) {
2237 return WERR_NOMEM;
2240 path = winreg_printer_data_keyname(tmp_ctx, printer);
2241 if (path == NULL) {
2242 TALLOC_FREE(tmp_ctx);
2243 return WERR_NOMEM;
2246 ZERO_STRUCT(hive_hnd);
2247 ZERO_STRUCT(key_hnd);
2249 result = winreg_printer_openkey(tmp_ctx,
2250 winreg_handle,
2251 path,
2252 key,
2253 false,
2254 access_mask,
2255 &hive_hnd,
2256 &key_hnd);
2257 if (!W_ERROR_IS_OK(result)) {
2258 DEBUG(0, ("winreg_delete_printer_dataex: Could not open key %s: %s\n",
2259 key, win_errstr(result)));
2260 goto done;
2263 wvalue.name = value;
2264 status = dcerpc_winreg_DeleteValue(winreg_handle,
2265 tmp_ctx,
2266 &key_hnd,
2267 wvalue,
2268 &result);
2269 if (!NT_STATUS_IS_OK(status)) {
2270 DEBUG(0, ("winreg_delete_printer_dataex: Could not delete value %s: %s\n",
2271 value, nt_errstr(status)));
2272 result = ntstatus_to_werror(status);
2275 done:
2276 if (winreg_handle != NULL) {
2277 WERROR ignore;
2279 if (is_valid_policy_hnd(&key_hnd)) {
2280 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2282 if (is_valid_policy_hnd(&hive_hnd)) {
2283 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2287 TALLOC_FREE(tmp_ctx);
2288 return result;
2291 /* Enumerate on the subkeys of a given key and provide the data. */
2292 WERROR winreg_enum_printer_key(TALLOC_CTX *mem_ctx,
2293 struct dcerpc_binding_handle *winreg_handle,
2294 const char *printer,
2295 const char *key,
2296 uint32_t *pnum_subkeys,
2297 const char ***psubkeys)
2299 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2300 struct policy_handle hive_hnd, key_hnd;
2301 char *path;
2302 const char **subkeys = NULL;
2303 uint32_t num_subkeys = -1;
2305 WERROR result = WERR_OK;
2306 NTSTATUS status;
2308 TALLOC_CTX *tmp_ctx;
2310 tmp_ctx = talloc_stackframe();
2311 if (tmp_ctx == NULL) {
2312 return WERR_NOMEM;
2315 path = winreg_printer_data_keyname(tmp_ctx, printer);
2316 if (path == NULL) {
2317 TALLOC_FREE(tmp_ctx);
2318 return WERR_NOMEM;
2321 ZERO_STRUCT(hive_hnd);
2322 ZERO_STRUCT(key_hnd);
2324 result = winreg_printer_openkey(tmp_ctx,
2325 winreg_handle,
2326 path,
2327 key,
2328 false,
2329 access_mask,
2330 &hive_hnd,
2331 &key_hnd);
2332 if (!W_ERROR_IS_OK(result)) {
2333 DEBUG(2, ("winreg_enum_printer_key: Could not open key %s: %s\n",
2334 key, win_errstr(result)));
2335 goto done;
2338 status = dcerpc_winreg_enum_keys(tmp_ctx,
2339 winreg_handle,
2340 &key_hnd,
2341 &num_subkeys,
2342 &subkeys,
2343 &result);
2344 if (!NT_STATUS_IS_OK(status)) {
2345 result = ntstatus_to_werror(status);
2347 if (!W_ERROR_IS_OK(result)) {
2348 DEBUG(0, ("winreg_enum_printer_key: Could not enumerate subkeys in %s: %s\n",
2349 key, win_errstr(result)));
2350 goto done;
2353 *pnum_subkeys = num_subkeys;
2354 if (psubkeys) {
2355 *psubkeys = talloc_move(mem_ctx, &subkeys);
2358 result = WERR_OK;
2359 done:
2360 if (winreg_handle != NULL) {
2361 WERROR ignore;
2363 if (is_valid_policy_hnd(&key_hnd)) {
2364 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2366 if (is_valid_policy_hnd(&hive_hnd)) {
2367 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2371 TALLOC_FREE(tmp_ctx);
2372 return result;
2375 /* Delete a key with subkeys of a given printer. */
2376 WERROR winreg_delete_printer_key(TALLOC_CTX *mem_ctx,
2377 struct dcerpc_binding_handle *winreg_handle,
2378 const char *printer,
2379 const char *key)
2381 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2382 struct policy_handle hive_hnd, key_hnd;
2383 char *keyname;
2384 char *path;
2385 WERROR result;
2386 NTSTATUS status;
2387 TALLOC_CTX *tmp_ctx;
2389 tmp_ctx = talloc_stackframe();
2390 if (tmp_ctx == NULL) {
2391 return WERR_NOMEM;
2394 path = winreg_printer_data_keyname(tmp_ctx, printer);
2395 if (path == NULL) {
2396 TALLOC_FREE(tmp_ctx);
2397 return WERR_NOMEM;
2400 result = winreg_printer_openkey(tmp_ctx,
2401 winreg_handle,
2402 path,
2403 key,
2404 false,
2405 access_mask,
2406 &hive_hnd,
2407 &key_hnd);
2408 if (!W_ERROR_IS_OK(result)) {
2409 /* key doesn't exist */
2410 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
2411 result = WERR_OK;
2412 goto done;
2415 DEBUG(0, ("winreg_delete_printer_key: Could not open key %s: %s\n",
2416 key, win_errstr(result)));
2417 goto done;
2420 if (is_valid_policy_hnd(&key_hnd)) {
2421 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &result);
2424 if (key == NULL || key[0] == '\0') {
2425 keyname = path;
2426 } else {
2427 keyname = talloc_asprintf(tmp_ctx,
2428 "%s\\%s",
2429 path,
2430 key);
2431 if (keyname == NULL) {
2432 result = WERR_NOMEM;
2433 goto done;
2437 status = dcerpc_winreg_delete_subkeys_recursive(tmp_ctx,
2438 winreg_handle,
2439 &hive_hnd,
2440 access_mask,
2441 keyname,
2442 &result);
2444 if (!NT_STATUS_IS_OK(status)) {
2445 DEBUG(0, ("winreg_delete_printer_key: Could not delete key %s: %s\n",
2446 key, nt_errstr(status)));
2447 result = ntstatus_to_werror(status);
2448 goto done;
2451 if (!W_ERROR_IS_OK(result)) {
2452 DEBUG(0, ("winreg_delete_printer_key: Could not delete key %s: %s\n",
2453 key, win_errstr(result)));
2454 goto done;
2457 done:
2458 if (winreg_handle != NULL) {
2459 WERROR ignore;
2461 if (is_valid_policy_hnd(&key_hnd)) {
2462 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2464 if (is_valid_policy_hnd(&hive_hnd)) {
2465 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2469 TALLOC_FREE(tmp_ctx);
2470 return result;
2473 WERROR winreg_printer_update_changeid(TALLOC_CTX *mem_ctx,
2474 struct dcerpc_binding_handle *winreg_handle,
2475 const char *printer)
2477 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2478 struct policy_handle hive_hnd, key_hnd;
2479 char *path;
2480 NTSTATUS status;
2481 WERROR result;
2482 TALLOC_CTX *tmp_ctx;
2484 tmp_ctx = talloc_stackframe();
2485 if (tmp_ctx == NULL) {
2486 return WERR_NOMEM;
2489 path = winreg_printer_data_keyname(tmp_ctx, printer);
2490 if (path == NULL) {
2491 TALLOC_FREE(tmp_ctx);
2492 return WERR_NOMEM;
2495 ZERO_STRUCT(hive_hnd);
2496 ZERO_STRUCT(key_hnd);
2498 result = winreg_printer_openkey(tmp_ctx,
2499 winreg_handle,
2500 path,
2502 false,
2503 access_mask,
2504 &hive_hnd,
2505 &key_hnd);
2506 if (!W_ERROR_IS_OK(result)) {
2507 DEBUG(0, ("winreg_printer_update_changeid: Could not open key %s: %s\n",
2508 path, win_errstr(result)));
2509 goto done;
2512 status = dcerpc_winreg_set_dword(tmp_ctx,
2513 winreg_handle,
2514 &key_hnd,
2515 "ChangeID",
2516 winreg_printer_rev_changeid(),
2517 &result);
2518 if (!NT_STATUS_IS_OK(status)) {
2519 result = ntstatus_to_werror(status);
2521 if (!W_ERROR_IS_OK(result)) {
2522 goto done;
2525 result = WERR_OK;
2526 done:
2527 if (winreg_handle != NULL) {
2528 WERROR ignore;
2530 if (is_valid_policy_hnd(&key_hnd)) {
2531 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2533 if (is_valid_policy_hnd(&hive_hnd)) {
2534 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2538 TALLOC_FREE(tmp_ctx);
2539 return result;
2542 WERROR winreg_printer_get_changeid(TALLOC_CTX *mem_ctx,
2543 struct dcerpc_binding_handle *winreg_handle,
2544 const char *printer,
2545 uint32_t *pchangeid)
2547 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2548 struct policy_handle hive_hnd, key_hnd;
2549 uint32_t changeid = 0;
2550 char *path;
2551 NTSTATUS status;
2552 WERROR result;
2553 TALLOC_CTX *tmp_ctx;
2555 tmp_ctx = talloc_stackframe();
2556 if (tmp_ctx == NULL) {
2557 return WERR_NOMEM;
2560 path = winreg_printer_data_keyname(tmp_ctx, printer);
2561 if (path == NULL) {
2562 TALLOC_FREE(tmp_ctx);
2563 return WERR_NOMEM;
2566 ZERO_STRUCT(hive_hnd);
2567 ZERO_STRUCT(key_hnd);
2569 result = winreg_printer_openkey(tmp_ctx,
2570 winreg_handle,
2571 path,
2573 false,
2574 access_mask,
2575 &hive_hnd,
2576 &key_hnd);
2577 if (!W_ERROR_IS_OK(result)) {
2578 DEBUG(2, ("winreg_printer_get_changeid: Could not open key %s: %s\n",
2579 path, win_errstr(result)));
2580 goto done;
2583 DEBUG(10, ("winreg_printer_get_changeid: get changeid from %s\n", path));
2585 status = dcerpc_winreg_query_dword(tmp_ctx,
2586 winreg_handle,
2587 &key_hnd,
2588 "ChangeID",
2589 &changeid,
2590 &result);
2591 if (!NT_STATUS_IS_OK(status)) {
2592 result = ntstatus_to_werror(status);
2594 if (!W_ERROR_IS_OK(result)) {
2595 goto done;
2598 if (pchangeid) {
2599 *pchangeid = changeid;
2602 result = WERR_OK;
2603 done:
2604 if (winreg_handle != NULL) {
2605 WERROR ignore;
2607 if (is_valid_policy_hnd(&key_hnd)) {
2608 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2610 if (is_valid_policy_hnd(&hive_hnd)) {
2611 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2615 TALLOC_FREE(tmp_ctx);
2616 return result;
2620 * The special behaviour of the spoolss forms is documented at the website:
2622 * Managing Win32 Printserver Forms
2623 * http://unixwiz.net/techtips/winspooler-forms.html
2626 WERROR winreg_printer_addform1(TALLOC_CTX *mem_ctx,
2627 struct dcerpc_binding_handle *winreg_handle,
2628 struct spoolss_AddFormInfo1 *form)
2630 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2631 struct policy_handle hive_hnd, key_hnd;
2632 struct winreg_String wvalue = { 0, };
2633 DATA_BLOB blob;
2634 uint32_t num_info = 0;
2635 union spoolss_FormInfo *info = NULL;
2636 uint32_t i;
2637 WERROR result;
2638 NTSTATUS status;
2639 TALLOC_CTX *tmp_ctx;
2641 tmp_ctx = talloc_stackframe();
2642 if (tmp_ctx == NULL) {
2643 return WERR_NOMEM;
2646 ZERO_STRUCT(hive_hnd);
2647 ZERO_STRUCT(key_hnd);
2649 result = winreg_printer_openkey(tmp_ctx,
2650 winreg_handle,
2651 TOP_LEVEL_CONTROL_FORMS_KEY,
2653 true,
2654 access_mask,
2655 &hive_hnd,
2656 &key_hnd);
2657 if (!W_ERROR_IS_OK(result)) {
2658 DEBUG(0, ("winreg_printer_addform1: Could not open key %s: %s\n",
2659 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2660 goto done;
2663 result = winreg_printer_enumforms1(tmp_ctx, winreg_handle,
2664 &num_info, &info);
2665 if (!W_ERROR_IS_OK(result)) {
2666 DEBUG(0, ("winreg_printer_addform: Could not enum keys %s: %s\n",
2667 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2668 goto done;
2671 /* If form name already exists or is builtin return ALREADY_EXISTS */
2672 for (i = 0; i < num_info; i++) {
2673 if (strequal(info[i].info1.form_name, form->form_name)) {
2674 result = WERR_FILE_EXISTS;
2675 goto done;
2679 wvalue.name = form->form_name;
2681 blob = data_blob_talloc(tmp_ctx, NULL, 32);
2682 SIVAL(blob.data, 0, form->size.width);
2683 SIVAL(blob.data, 4, form->size.height);
2684 SIVAL(blob.data, 8, form->area.left);
2685 SIVAL(blob.data, 12, form->area.top);
2686 SIVAL(blob.data, 16, form->area.right);
2687 SIVAL(blob.data, 20, form->area.bottom);
2688 SIVAL(blob.data, 24, num_info + 1); /* FIXME */
2689 SIVAL(blob.data, 28, form->flags);
2691 status = dcerpc_winreg_SetValue(winreg_handle,
2692 tmp_ctx,
2693 &key_hnd,
2694 wvalue,
2695 REG_BINARY,
2696 blob.data,
2697 blob.length,
2698 &result);
2699 if (!NT_STATUS_IS_OK(status)) {
2700 DEBUG(0, ("winreg_printer_addform1: Could not set value %s: %s\n",
2701 wvalue.name, nt_errstr(status)));
2702 result = ntstatus_to_werror(status);
2705 done:
2706 if (winreg_handle != NULL) {
2707 WERROR ignore;
2709 if (is_valid_policy_hnd(&key_hnd)) {
2710 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2712 if (is_valid_policy_hnd(&hive_hnd)) {
2713 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2717 TALLOC_FREE(info);
2718 TALLOC_FREE(tmp_ctx);
2719 return result;
2722 WERROR winreg_printer_enumforms1(TALLOC_CTX *mem_ctx,
2723 struct dcerpc_binding_handle *winreg_handle,
2724 uint32_t *pnum_info,
2725 union spoolss_FormInfo **pinfo)
2727 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2728 struct policy_handle hive_hnd, key_hnd;
2729 union spoolss_FormInfo *info;
2730 struct spoolss_PrinterEnumValues *enum_values = NULL;
2731 uint32_t num_values = 0;
2732 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
2733 uint32_t i;
2734 WERROR result;
2735 NTSTATUS status;
2736 const char **enum_names = NULL;
2737 enum winreg_Type *enum_types = NULL;
2738 DATA_BLOB *enum_data_blobs = NULL;
2739 TALLOC_CTX *tmp_ctx;
2741 tmp_ctx = talloc_stackframe();
2742 if (tmp_ctx == NULL) {
2743 return WERR_NOMEM;
2746 ZERO_STRUCT(hive_hnd);
2747 ZERO_STRUCT(key_hnd);
2749 result = winreg_printer_openkey(tmp_ctx,
2750 winreg_handle,
2751 TOP_LEVEL_CONTROL_FORMS_KEY,
2753 true,
2754 access_mask,
2755 &hive_hnd,
2756 &key_hnd);
2757 if (!W_ERROR_IS_OK(result)) {
2758 /* key doesn't exist */
2759 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
2760 result = WERR_OK;
2761 goto done;
2764 DEBUG(0, ("winreg_printer_enumforms1: Could not open key %s: %s\n",
2765 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2766 goto done;
2769 status = dcerpc_winreg_enumvals(tmp_ctx,
2770 winreg_handle,
2771 &key_hnd,
2772 &num_values,
2773 &enum_names,
2774 &enum_types,
2775 &enum_data_blobs,
2776 &result);
2777 if (!NT_STATUS_IS_OK(status)){
2778 result = ntstatus_to_werror(status);
2781 if (!W_ERROR_IS_OK(result)) {
2782 DEBUG(0, ("winreg_printer_enumforms1: Could not enumerate values in %s: %s\n",
2783 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2784 goto done;
2787 enum_values = talloc_zero_array(tmp_ctx,
2788 struct spoolss_PrinterEnumValues,
2789 num_values);
2790 if (enum_values == NULL){
2791 result = WERR_NOMEM;
2792 goto done;
2795 for (i = 0; i < num_values; i++){
2796 enum_values[i].value_name = enum_names[i];
2797 enum_values[i].value_name_len = strlen_m_term(enum_names[i]) * 2;
2798 enum_values[i].type = enum_types[i];
2799 enum_values[i].data_length = enum_data_blobs[i].length;
2800 enum_values[i].data = NULL;
2801 if (enum_values[i].data_length != 0){
2802 enum_values[i].data = &enum_data_blobs[i];
2806 if (!W_ERROR_IS_OK(result)) {
2807 DEBUG(0, ("winreg_printer_enumforms1: Could not enumerate values in %s: %s\n",
2808 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2809 goto done;
2812 info = talloc_array(tmp_ctx, union spoolss_FormInfo, num_builtin + num_values);
2813 if (info == NULL) {
2814 result = WERR_NOMEM;
2815 goto done;
2818 /* Enumerate BUILTIN forms */
2819 for (i = 0; i < num_builtin; i++) {
2820 info[i].info1 = builtin_forms1[i];
2823 /* Enumerate registry forms */
2824 for (i = 0; i < num_values; i++) {
2825 union spoolss_FormInfo val;
2827 if (enum_values[i].type != REG_BINARY ||
2828 enum_values[i].data_length != 32) {
2829 continue;
2832 val.info1.form_name = talloc_strdup(info, enum_values[i].value_name);
2833 if (val.info1.form_name == NULL) {
2834 result = WERR_NOMEM;
2835 goto done;
2838 val.info1.size.width = IVAL(enum_values[i].data->data, 0);
2839 val.info1.size.height = IVAL(enum_values[i].data->data, 4);
2840 val.info1.area.left = IVAL(enum_values[i].data->data, 8);
2841 val.info1.area.top = IVAL(enum_values[i].data->data, 12);
2842 val.info1.area.right = IVAL(enum_values[i].data->data, 16);
2843 val.info1.area.bottom = IVAL(enum_values[i].data->data, 20);
2844 /* skip form index IVAL(enum_values[i].data->data, 24)));*/
2845 val.info1.flags = (enum spoolss_FormFlags) IVAL(enum_values[i].data->data, 28);
2847 info[i + num_builtin] = val;
2850 *pnum_info = num_builtin + num_values;
2851 if (pinfo) {
2852 *pinfo = talloc_move(mem_ctx, &info);
2855 done:
2856 if (winreg_handle != NULL) {
2857 WERROR ignore;
2859 if (is_valid_policy_hnd(&key_hnd)) {
2860 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2862 if (is_valid_policy_hnd(&hive_hnd)) {
2863 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2867 TALLOC_FREE(enum_values);
2868 TALLOC_FREE(tmp_ctx);
2869 return result;
2872 WERROR winreg_printer_deleteform1(TALLOC_CTX *mem_ctx,
2873 struct dcerpc_binding_handle *winreg_handle,
2874 const char *form_name)
2876 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2877 struct policy_handle hive_hnd, key_hnd;
2878 struct winreg_String wvalue = { 0, };
2879 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
2880 uint32_t i;
2881 WERROR result = WERR_OK;
2882 NTSTATUS status;
2883 TALLOC_CTX *tmp_ctx;
2885 for (i = 0; i < num_builtin; i++) {
2886 if (strequal(builtin_forms1[i].form_name, form_name)) {
2887 return WERR_INVALID_PARAMETER;
2891 tmp_ctx = talloc_stackframe();
2892 if (tmp_ctx == NULL) {
2893 return WERR_NOMEM;
2896 ZERO_STRUCT(hive_hnd);
2897 ZERO_STRUCT(key_hnd);
2899 result = winreg_printer_openkey(tmp_ctx,
2900 winreg_handle,
2901 TOP_LEVEL_CONTROL_FORMS_KEY,
2903 false,
2904 access_mask,
2905 &hive_hnd,
2906 &key_hnd);
2907 if (!W_ERROR_IS_OK(result)) {
2908 DEBUG(0, ("winreg_printer_deleteform1: Could not open key %s: %s\n",
2909 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2910 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
2911 result = WERR_INVALID_FORM_NAME;
2913 goto done;
2916 wvalue.name = form_name;
2917 status = dcerpc_winreg_DeleteValue(winreg_handle,
2918 tmp_ctx,
2919 &key_hnd,
2920 wvalue,
2921 &result);
2922 if (!NT_STATUS_IS_OK(status)) {
2923 /* If the value doesn't exist, return WERR_INVALID_FORM_NAME */
2924 DEBUG(0, ("winreg_printer_delteform1: Could not delete value %s: %s\n",
2925 wvalue.name, nt_errstr(status)));
2926 result = ntstatus_to_werror(status);
2927 goto done;
2930 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
2931 result = WERR_INVALID_FORM_NAME;
2934 done:
2935 if (winreg_handle != NULL) {
2936 WERROR ignore;
2938 if (is_valid_policy_hnd(&key_hnd)) {
2939 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
2941 if (is_valid_policy_hnd(&hive_hnd)) {
2942 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
2946 TALLOC_FREE(tmp_ctx);
2947 return result;
2950 WERROR winreg_printer_setform1(TALLOC_CTX *mem_ctx,
2951 struct dcerpc_binding_handle *winreg_handle,
2952 const char *form_name,
2953 struct spoolss_AddFormInfo1 *form)
2955 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2956 struct policy_handle hive_hnd, key_hnd;
2957 struct winreg_String wvalue = { 0, };
2958 DATA_BLOB blob;
2959 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
2960 uint32_t i;
2961 WERROR result;
2962 NTSTATUS status;
2963 TALLOC_CTX *tmp_ctx = NULL;
2965 for (i = 0; i < num_builtin; i++) {
2966 if (strequal(builtin_forms1[i].form_name, form->form_name)) {
2967 result = WERR_INVALID_PARAM;
2968 goto done;
2972 tmp_ctx = talloc_stackframe();
2973 if (tmp_ctx == NULL) {
2974 return WERR_NOMEM;
2977 ZERO_STRUCT(hive_hnd);
2978 ZERO_STRUCT(key_hnd);
2980 result = winreg_printer_openkey(tmp_ctx,
2981 winreg_handle,
2982 TOP_LEVEL_CONTROL_FORMS_KEY,
2984 true,
2985 access_mask,
2986 &hive_hnd,
2987 &key_hnd);
2988 if (!W_ERROR_IS_OK(result)) {
2989 DEBUG(0, ("winreg_printer_setform1: Could not open key %s: %s\n",
2990 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2991 goto done;
2994 /* If form_name != form->form_name then we renamed the form */
2995 if (strequal(form_name, form->form_name)) {
2996 result = winreg_printer_deleteform1(tmp_ctx, winreg_handle,
2997 form_name);
2998 if (!W_ERROR_IS_OK(result)) {
2999 DEBUG(0, ("winreg_printer_setform1: Could not open key %s: %s\n",
3000 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
3001 goto done;
3005 wvalue.name = form->form_name;
3007 blob = data_blob_talloc(tmp_ctx, NULL, 32);
3008 SIVAL(blob.data, 0, form->size.width);
3009 SIVAL(blob.data, 4, form->size.height);
3010 SIVAL(blob.data, 8, form->area.left);
3011 SIVAL(blob.data, 12, form->area.top);
3012 SIVAL(blob.data, 16, form->area.right);
3013 SIVAL(blob.data, 20, form->area.bottom);
3014 SIVAL(blob.data, 24, 42);
3015 SIVAL(blob.data, 28, form->flags);
3017 status = dcerpc_winreg_SetValue(winreg_handle,
3018 tmp_ctx,
3019 &key_hnd,
3020 wvalue,
3021 REG_BINARY,
3022 blob.data,
3023 blob.length,
3024 &result);
3025 if (!NT_STATUS_IS_OK(status)) {
3026 DEBUG(0, ("winreg_printer_setform1: Could not set value %s: %s\n",
3027 wvalue.name, nt_errstr(status)));
3028 result = ntstatus_to_werror(status);
3031 done:
3032 if (winreg_handle != NULL) {
3033 WERROR ignore;
3035 if (is_valid_policy_hnd(&key_hnd)) {
3036 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3038 if (is_valid_policy_hnd(&hive_hnd)) {
3039 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3043 TALLOC_FREE(tmp_ctx);
3044 return result;
3047 WERROR winreg_printer_getform1(TALLOC_CTX *mem_ctx,
3048 struct dcerpc_binding_handle *winreg_handle,
3049 const char *form_name,
3050 struct spoolss_FormInfo1 *r)
3052 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3053 struct policy_handle hive_hnd, key_hnd;
3054 struct winreg_String wvalue;
3055 enum winreg_Type type_in = REG_NONE;
3056 uint8_t *data_in = NULL;
3057 uint32_t data_in_size = 0;
3058 uint32_t value_len = 0;
3059 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
3060 uint32_t i;
3061 WERROR result;
3062 NTSTATUS status;
3063 TALLOC_CTX *tmp_ctx;
3065 /* check builtin forms first */
3066 for (i = 0; i < num_builtin; i++) {
3067 if (strequal(builtin_forms1[i].form_name, form_name)) {
3068 *r = builtin_forms1[i];
3069 return WERR_OK;
3073 tmp_ctx = talloc_stackframe();
3074 if (tmp_ctx == NULL) {
3075 return WERR_NOMEM;
3078 ZERO_STRUCT(hive_hnd);
3079 ZERO_STRUCT(key_hnd);
3081 result = winreg_printer_openkey(tmp_ctx,
3082 winreg_handle,
3083 TOP_LEVEL_CONTROL_FORMS_KEY,
3085 true,
3086 access_mask,
3087 &hive_hnd,
3088 &key_hnd);
3089 if (!W_ERROR_IS_OK(result)) {
3090 DEBUG(2, ("winreg_printer_getform1: Could not open key %s: %s\n",
3091 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
3092 goto done;
3095 wvalue.name = form_name;
3098 * call QueryValue once with data == NULL to get the
3099 * needed memory size to be allocated, then allocate
3100 * data buffer and call again.
3102 status = dcerpc_winreg_QueryValue(winreg_handle,
3103 tmp_ctx,
3104 &key_hnd,
3105 &wvalue,
3106 &type_in,
3107 NULL,
3108 &data_in_size,
3109 &value_len,
3110 &result);
3111 if (!NT_STATUS_IS_OK(status)) {
3112 DEBUG(0, ("winreg_printer_getform1: Could not query value %s: %s\n",
3113 wvalue.name, nt_errstr(status)));
3114 result = ntstatus_to_werror(status);
3115 goto done;
3117 if (!W_ERROR_IS_OK(result)) {
3118 goto done;
3121 data_in = (uint8_t *) TALLOC(tmp_ctx, data_in_size);
3122 if (data_in == NULL) {
3123 result = WERR_NOMEM;
3124 goto done;
3126 value_len = 0;
3128 status = dcerpc_winreg_QueryValue(winreg_handle,
3129 tmp_ctx,
3130 &key_hnd,
3131 &wvalue,
3132 &type_in,
3133 data_in,
3134 &data_in_size,
3135 &value_len,
3136 &result);
3137 if (!NT_STATUS_IS_OK(status)) {
3138 DEBUG(0, ("winreg_printer_getform1: Could not query value %s: %s\n",
3139 wvalue.name, nt_errstr(status)));
3140 result = ntstatus_to_werror(status);
3141 goto done;
3143 if (!W_ERROR_IS_OK(result)) {
3144 goto done;
3147 r->form_name = talloc_strdup(mem_ctx, form_name);
3148 if (r->form_name == NULL) {
3149 result = WERR_NOMEM;
3150 goto done;
3153 r->size.width = IVAL(data_in, 0);
3154 r->size.height = IVAL(data_in, 4);
3155 r->area.left = IVAL(data_in, 8);
3156 r->area.top = IVAL(data_in, 12);
3157 r->area.right = IVAL(data_in, 16);
3158 r->area.bottom = IVAL(data_in, 20);
3159 /* skip index IVAL(data_in, 24)));*/
3160 r->flags = (enum spoolss_FormFlags) IVAL(data_in, 28);
3162 result = WERR_OK;
3163 done:
3164 if (winreg_handle != NULL) {
3165 WERROR ignore;
3167 if (is_valid_policy_hnd(&key_hnd)) {
3168 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3170 if (is_valid_policy_hnd(&hive_hnd)) {
3171 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3175 TALLOC_FREE(tmp_ctx);
3176 return result;
3179 WERROR winreg_add_driver(TALLOC_CTX *mem_ctx,
3180 struct dcerpc_binding_handle *winreg_handle,
3181 struct spoolss_AddDriverInfoCtr *r,
3182 const char **driver_name,
3183 uint32_t *driver_version)
3185 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3186 struct policy_handle hive_hnd, key_hnd;
3187 struct spoolss_DriverInfo8 info8;
3188 TALLOC_CTX *tmp_ctx = NULL;
3189 NTSTATUS status;
3190 WERROR result;
3192 ZERO_STRUCT(hive_hnd);
3193 ZERO_STRUCT(key_hnd);
3194 ZERO_STRUCT(info8);
3196 if (!driver_info_ctr_to_info8(r, &info8)) {
3197 result = WERR_INVALID_PARAMETER;
3198 goto done;
3201 tmp_ctx = talloc_stackframe();
3202 if (tmp_ctx == NULL) {
3203 return WERR_NOMEM;
3206 result = winreg_printer_opendriver(tmp_ctx,
3207 winreg_handle,
3208 info8.driver_name,
3209 info8.architecture,
3210 info8.version,
3211 access_mask, true,
3212 &hive_hnd,
3213 &key_hnd);
3214 if (!W_ERROR_IS_OK(result)) {
3215 DEBUG(0, ("winreg_add_driver: "
3216 "Could not open driver key (%s,%s,%d): %s\n",
3217 info8.driver_name, info8.architecture,
3218 info8.version, win_errstr(result)));
3219 goto done;
3222 /* TODO: "Attributes" ? */
3224 status = dcerpc_winreg_set_dword(tmp_ctx,
3225 winreg_handle,
3226 &key_hnd,
3227 "Version",
3228 info8.version,
3229 &result);
3230 if (!NT_STATUS_IS_OK(status)) {
3231 result = ntstatus_to_werror(status);
3233 if (!W_ERROR_IS_OK(result)) {
3234 goto done;
3237 status = dcerpc_winreg_set_sz(tmp_ctx,
3238 winreg_handle,
3239 &key_hnd,
3240 "Driver",
3241 info8.driver_path,
3242 &result);
3243 if (!NT_STATUS_IS_OK(status)) {
3244 result = ntstatus_to_werror(status);
3246 if (!W_ERROR_IS_OK(result)) {
3247 goto done;
3250 status = dcerpc_winreg_set_sz(tmp_ctx,
3251 winreg_handle,
3252 &key_hnd,
3253 "Data File",
3254 info8.data_file,
3255 &result);
3256 if (!NT_STATUS_IS_OK(status)) {
3257 result = ntstatus_to_werror(status);
3259 if (!W_ERROR_IS_OK(result)) {
3260 goto done;
3263 status = dcerpc_winreg_set_sz(tmp_ctx,
3264 winreg_handle,
3265 &key_hnd,
3266 "Configuration File",
3267 info8.config_file,
3268 &result);
3269 if (!NT_STATUS_IS_OK(status)) {
3270 result = ntstatus_to_werror(status);
3272 if (!W_ERROR_IS_OK(result)) {
3273 goto done;
3276 status = dcerpc_winreg_set_sz(tmp_ctx,
3277 winreg_handle,
3278 &key_hnd,
3279 "Help File",
3280 info8.help_file,
3281 &result);
3282 if (!NT_STATUS_IS_OK(status)) {
3283 result = ntstatus_to_werror(status);
3285 if (!W_ERROR_IS_OK(result)) {
3286 goto done;
3289 status = dcerpc_winreg_set_multi_sz(tmp_ctx,
3290 winreg_handle,
3291 &key_hnd,
3292 "Dependent Files",
3293 info8.dependent_files,
3294 &result);
3295 if (!NT_STATUS_IS_OK(status)) {
3296 result = ntstatus_to_werror(status);
3298 if (!W_ERROR_IS_OK(result)) {
3299 goto done;
3302 status = dcerpc_winreg_set_sz(tmp_ctx,
3303 winreg_handle,
3304 &key_hnd,
3305 "Monitor",
3306 info8.monitor_name,
3307 &result);
3308 if (!NT_STATUS_IS_OK(status)) {
3309 result = ntstatus_to_werror(status);
3311 if (!W_ERROR_IS_OK(result)) {
3312 goto done;
3315 status = dcerpc_winreg_set_sz(tmp_ctx,
3316 winreg_handle,
3317 &key_hnd,
3318 "Datatype",
3319 info8.default_datatype,
3320 &result);
3321 if (!NT_STATUS_IS_OK(status)) {
3322 result = ntstatus_to_werror(status);
3324 if (!W_ERROR_IS_OK(result)) {
3325 goto done;
3328 status = dcerpc_winreg_set_multi_sz(tmp_ctx,
3329 winreg_handle,
3330 &key_hnd, "Previous Names",
3331 info8.previous_names,
3332 &result);
3333 if (!NT_STATUS_IS_OK(status)) {
3334 result = ntstatus_to_werror(status);
3336 if (!W_ERROR_IS_OK(result)) {
3337 goto done;
3340 result = winreg_printer_write_date(tmp_ctx, winreg_handle,
3341 &key_hnd, "DriverDate",
3342 info8.driver_date);
3343 if (!W_ERROR_IS_OK(result)) {
3344 goto done;
3347 result = winreg_printer_write_ver(tmp_ctx, winreg_handle,
3348 &key_hnd, "DriverVersion",
3349 info8.driver_version);
3350 if (!W_ERROR_IS_OK(result)) {
3351 goto done;
3354 status = dcerpc_winreg_set_sz(tmp_ctx,
3355 winreg_handle,
3356 &key_hnd,
3357 "Manufacturer",
3358 info8.manufacturer_name,
3359 &result);
3360 if (!NT_STATUS_IS_OK(status)) {
3361 result = ntstatus_to_werror(status);
3363 if (!W_ERROR_IS_OK(result)) {
3364 goto done;
3367 status = dcerpc_winreg_set_sz(tmp_ctx,
3368 winreg_handle,
3369 &key_hnd,
3370 "OEM URL",
3371 info8.manufacturer_url,
3372 &result);
3373 if (!NT_STATUS_IS_OK(status)) {
3374 result = ntstatus_to_werror(status);
3376 if (!W_ERROR_IS_OK(result)) {
3377 goto done;
3380 status = dcerpc_winreg_set_sz(tmp_ctx,
3381 winreg_handle,
3382 &key_hnd,
3383 "HardwareID",
3384 info8.hardware_id,
3385 &result);
3386 if (!NT_STATUS_IS_OK(status)) {
3387 result = ntstatus_to_werror(status);
3389 if (!W_ERROR_IS_OK(result)) {
3390 goto done;
3393 status = dcerpc_winreg_set_sz(tmp_ctx,
3394 winreg_handle,
3395 &key_hnd,
3396 "Provider",
3397 info8.provider,
3398 &result);
3399 if (!NT_STATUS_IS_OK(status)) {
3400 result = ntstatus_to_werror(status);
3402 if (!W_ERROR_IS_OK(result)) {
3403 goto done;
3406 status = dcerpc_winreg_set_sz(tmp_ctx,
3407 winreg_handle,
3408 &key_hnd,
3409 "Print Processor",
3410 info8.print_processor,
3411 &result);
3412 if (!NT_STATUS_IS_OK(status)) {
3413 result = ntstatus_to_werror(status);
3415 if (!W_ERROR_IS_OK(result)) {
3416 goto done;
3419 status = dcerpc_winreg_set_sz(tmp_ctx,
3420 winreg_handle,
3421 &key_hnd,
3422 "VendorSetup",
3423 info8.vendor_setup,
3424 &result);
3425 if (!NT_STATUS_IS_OK(status)) {
3426 result = ntstatus_to_werror(status);
3428 if (!W_ERROR_IS_OK(result)) {
3429 goto done;
3432 status = dcerpc_winreg_set_multi_sz(tmp_ctx,
3433 winreg_handle,
3434 &key_hnd,
3435 "Color Profiles",
3436 info8.color_profiles,
3437 &result);
3438 if (!NT_STATUS_IS_OK(status)) {
3439 result = ntstatus_to_werror(status);
3441 if (!W_ERROR_IS_OK(result)) {
3442 goto done;
3445 status = dcerpc_winreg_set_sz(tmp_ctx,
3446 winreg_handle,
3447 &key_hnd,
3448 "InfPath",
3449 info8.inf_path,
3450 &result);
3451 if (!NT_STATUS_IS_OK(status)) {
3452 result = ntstatus_to_werror(status);
3454 if (!W_ERROR_IS_OK(result)) {
3455 goto done;
3458 status = dcerpc_winreg_set_dword(tmp_ctx,
3459 winreg_handle,
3460 &key_hnd,
3461 "PrinterDriverAttributes",
3462 info8.printer_driver_attributes,
3463 &result);
3464 if (!NT_STATUS_IS_OK(status)) {
3465 result = ntstatus_to_werror(status);
3467 if (!W_ERROR_IS_OK(result)) {
3468 goto done;
3471 status = dcerpc_winreg_set_multi_sz(tmp_ctx,
3472 winreg_handle,
3473 &key_hnd,
3474 "CoreDependencies",
3475 info8.core_driver_dependencies,
3476 &result);
3477 if (!NT_STATUS_IS_OK(status)) {
3478 result = ntstatus_to_werror(status);
3480 if (!W_ERROR_IS_OK(result)) {
3481 goto done;
3484 result = winreg_printer_write_date(tmp_ctx, winreg_handle,
3485 &key_hnd, "MinInboxDriverVerDate",
3486 info8.min_inbox_driver_ver_date);
3487 if (!W_ERROR_IS_OK(result)) {
3488 goto done;
3491 result = winreg_printer_write_ver(tmp_ctx, winreg_handle, &key_hnd,
3492 "MinInboxDriverVerVersion",
3493 info8.min_inbox_driver_ver_version);
3494 if (!W_ERROR_IS_OK(result)) {
3495 goto done;
3498 *driver_name = info8.driver_name;
3499 *driver_version = info8.version;
3500 result = WERR_OK;
3501 done:
3502 if (winreg_handle != NULL) {
3503 WERROR ignore;
3505 if (is_valid_policy_hnd(&key_hnd)) {
3506 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3508 if (is_valid_policy_hnd(&hive_hnd)) {
3509 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3513 TALLOC_FREE(tmp_ctx);
3514 return result;
3517 WERROR winreg_get_driver(TALLOC_CTX *mem_ctx,
3518 struct dcerpc_binding_handle *winreg_handle,
3519 const char *architecture,
3520 const char *driver_name,
3521 uint32_t driver_version,
3522 struct spoolss_DriverInfo8 **_info8)
3524 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3525 struct policy_handle hive_hnd, key_hnd;
3526 struct spoolss_DriverInfo8 i8, *info8;
3527 struct spoolss_PrinterEnumValues *enum_values = NULL;
3528 struct spoolss_PrinterEnumValues *v;
3529 uint32_t num_values = 0;
3530 TALLOC_CTX *tmp_ctx;
3531 WERROR result;
3532 NTSTATUS status;
3533 uint32_t i;
3534 const char **enum_names = NULL;
3535 enum winreg_Type *enum_types = NULL;
3536 DATA_BLOB *enum_data_blobs = NULL;
3538 ZERO_STRUCT(hive_hnd);
3539 ZERO_STRUCT(key_hnd);
3540 ZERO_STRUCT(i8);
3542 tmp_ctx = talloc_stackframe();
3543 if (tmp_ctx == NULL) {
3544 return WERR_NOMEM;
3547 if (driver_version == DRIVER_ANY_VERSION) {
3548 /* look for Win2k first and then for NT4 */
3549 result = winreg_printer_opendriver(tmp_ctx,
3550 winreg_handle,
3551 driver_name,
3552 architecture,
3554 access_mask, false,
3555 &hive_hnd,
3556 &key_hnd);
3557 if (!W_ERROR_IS_OK(result)) {
3558 result = winreg_printer_opendriver(tmp_ctx,
3559 winreg_handle,
3560 driver_name,
3561 architecture,
3563 access_mask, false,
3564 &hive_hnd,
3565 &key_hnd);
3567 } else {
3568 /* ok normal case */
3569 result = winreg_printer_opendriver(tmp_ctx,
3570 winreg_handle,
3571 driver_name,
3572 architecture,
3573 driver_version,
3574 access_mask, false,
3575 &hive_hnd,
3576 &key_hnd);
3578 if (!W_ERROR_IS_OK(result)) {
3579 DEBUG(5, ("winreg_get_driver: "
3580 "Could not open driver key (%s,%s,%d): %s\n",
3581 driver_name, architecture,
3582 driver_version, win_errstr(result)));
3583 goto done;
3586 status = dcerpc_winreg_enumvals(tmp_ctx,
3587 winreg_handle,
3588 &key_hnd,
3589 &num_values,
3590 &enum_names,
3591 &enum_types,
3592 &enum_data_blobs,
3593 &result);
3594 if (!NT_STATUS_IS_OK(status)){
3595 result = ntstatus_to_werror(status);
3598 if (!W_ERROR_IS_OK(result)) {
3599 DEBUG(0, ("winreg_get_driver: "
3600 "Could not enumerate values for (%s,%s,%d): %s\n",
3601 driver_name, architecture,
3602 driver_version, win_errstr(result)));
3603 goto done;
3606 enum_values = talloc_zero_array(tmp_ctx,
3607 struct spoolss_PrinterEnumValues,
3608 num_values);
3609 if (enum_values == NULL){
3610 result = WERR_NOMEM;
3611 goto done;
3614 for (i = 0; i < num_values; i++){
3615 enum_values[i].value_name = enum_names[i];
3616 enum_values[i].value_name_len = strlen_m_term(enum_names[i]) * 2;
3617 enum_values[i].type = enum_types[i];
3618 enum_values[i].data_length = enum_data_blobs[i].length;
3619 enum_values[i].data = NULL;
3620 if (enum_values[i].data_length != 0){
3621 enum_values[i].data = &enum_data_blobs[i];
3625 info8 = talloc_zero(tmp_ctx, struct spoolss_DriverInfo8);
3626 if (info8 == NULL) {
3627 result = WERR_NOMEM;
3628 goto done;
3631 info8->driver_name = talloc_strdup(info8, driver_name);
3632 if (info8->driver_name == NULL) {
3633 result = WERR_NOMEM;
3634 goto done;
3637 info8->architecture = talloc_strdup(info8, architecture);
3638 if (info8->architecture == NULL) {
3639 result = WERR_NOMEM;
3640 goto done;
3643 result = WERR_OK;
3645 for (i = 0; i < num_values; i++) {
3646 const char *tmp_str;
3647 uint32_t tmp = 0;
3649 v = &enum_values[i];
3651 result = winreg_enumval_to_dword(info8, v,
3652 "Version",
3653 &tmp);
3654 if (W_ERROR_IS_OK(result)) {
3655 info8->version = (enum spoolss_DriverOSVersion) tmp;
3657 CHECK_ERROR(result);
3659 result = winreg_enumval_to_sz(info8, v,
3660 "Driver",
3661 &info8->driver_path);
3662 CHECK_ERROR(result);
3664 result = winreg_enumval_to_sz(info8, v,
3665 "Data File",
3666 &info8->data_file);
3667 CHECK_ERROR(result);
3669 result = winreg_enumval_to_sz(info8, v,
3670 "Configuration File",
3671 &info8->config_file);
3672 CHECK_ERROR(result);
3674 result = winreg_enumval_to_sz(info8, v,
3675 "Help File",
3676 &info8->help_file);
3677 CHECK_ERROR(result);
3679 result = winreg_enumval_to_multi_sz(info8, v,
3680 "Dependent Files",
3681 &info8->dependent_files);
3682 CHECK_ERROR(result);
3684 result = winreg_enumval_to_sz(info8, v,
3685 "Monitor",
3686 &info8->monitor_name);
3687 CHECK_ERROR(result);
3689 result = winreg_enumval_to_sz(info8, v,
3690 "Datatype",
3691 &info8->default_datatype);
3692 CHECK_ERROR(result);
3694 result = winreg_enumval_to_multi_sz(info8, v,
3695 "Previous Names",
3696 &info8->previous_names);
3697 CHECK_ERROR(result);
3699 result = winreg_enumval_to_sz(info8, v,
3700 "DriverDate",
3701 &tmp_str);
3702 if (W_ERROR_IS_OK(result)) {
3703 result = winreg_printer_date_to_NTTIME(tmp_str,
3704 &info8->driver_date);
3706 CHECK_ERROR(result);
3708 result = winreg_enumval_to_sz(info8, v,
3709 "DriverVersion",
3710 &tmp_str);
3711 if (W_ERROR_IS_OK(result)) {
3712 result = winreg_printer_ver_to_dword(tmp_str,
3713 &info8->driver_version);
3715 CHECK_ERROR(result);
3717 result = winreg_enumval_to_sz(info8, v,
3718 "Manufacturer",
3719 &info8->manufacturer_name);
3720 CHECK_ERROR(result);
3722 result = winreg_enumval_to_sz(info8, v,
3723 "OEM URL",
3724 &info8->manufacturer_url);
3725 CHECK_ERROR(result);
3727 result = winreg_enumval_to_sz(info8, v,
3728 "HardwareID",
3729 &info8->hardware_id);
3730 CHECK_ERROR(result);
3732 result = winreg_enumval_to_sz(info8, v,
3733 "Provider",
3734 &info8->provider);
3735 CHECK_ERROR(result);
3737 result = winreg_enumval_to_sz(info8, v,
3738 "Print Processor",
3739 &info8->print_processor);
3740 CHECK_ERROR(result);
3742 result = winreg_enumval_to_sz(info8, v,
3743 "VendorSetup",
3744 &info8->vendor_setup);
3745 CHECK_ERROR(result);
3747 result = winreg_enumval_to_multi_sz(info8, v,
3748 "Color Profiles",
3749 &info8->color_profiles);
3750 CHECK_ERROR(result);
3752 result = winreg_enumval_to_sz(info8, v,
3753 "InfPath",
3754 &info8->inf_path);
3755 CHECK_ERROR(result);
3757 result = winreg_enumval_to_dword(info8, v,
3758 "PrinterDriverAttributes",
3759 &info8->printer_driver_attributes);
3760 CHECK_ERROR(result);
3762 result = winreg_enumval_to_multi_sz(info8, v,
3763 "CoreDependencies",
3764 &info8->core_driver_dependencies);
3765 CHECK_ERROR(result);
3767 result = winreg_enumval_to_sz(info8, v,
3768 "MinInboxDriverVerDate",
3769 &tmp_str);
3770 if (W_ERROR_IS_OK(result)) {
3771 result = winreg_printer_date_to_NTTIME(tmp_str,
3772 &info8->min_inbox_driver_ver_date);
3774 CHECK_ERROR(result);
3776 result = winreg_enumval_to_sz(info8, v,
3777 "MinInboxDriverVerVersion",
3778 &tmp_str);
3779 if (W_ERROR_IS_OK(result)) {
3780 result = winreg_printer_ver_to_dword(tmp_str,
3781 &info8->min_inbox_driver_ver_version);
3783 CHECK_ERROR(result);
3786 if (!W_ERROR_IS_OK(result)) {
3787 DEBUG(0, ("winreg_enumval_to_TYPE() failed "
3788 "for %s: %s\n", v->value_name,
3789 win_errstr(result)));
3790 goto done;
3793 *_info8 = talloc_steal(mem_ctx, info8);
3794 result = WERR_OK;
3795 done:
3796 if (winreg_handle != NULL) {
3797 WERROR ignore;
3799 if (is_valid_policy_hnd(&key_hnd)) {
3800 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3802 if (is_valid_policy_hnd(&hive_hnd)) {
3803 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3807 TALLOC_FREE(tmp_ctx);
3808 return result;
3811 WERROR winreg_del_driver(TALLOC_CTX *mem_ctx,
3812 struct dcerpc_binding_handle *winreg_handle,
3813 struct spoolss_DriverInfo8 *info8,
3814 uint32_t version)
3816 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3817 struct policy_handle hive_hnd, key_hnd;
3818 TALLOC_CTX *tmp_ctx;
3819 char *key_name;
3820 WERROR result;
3821 NTSTATUS status;
3823 ZERO_STRUCT(hive_hnd);
3824 ZERO_STRUCT(key_hnd);
3826 tmp_ctx = talloc_stackframe();
3827 if (tmp_ctx == NULL) {
3828 return WERR_NOMEM;
3831 /* test that the key exists */
3832 result = winreg_printer_opendriver(tmp_ctx,
3833 winreg_handle,
3834 info8->driver_name,
3835 info8->architecture,
3836 version,
3837 access_mask, false,
3838 &hive_hnd,
3839 &key_hnd);
3840 if (!W_ERROR_IS_OK(result)) {
3841 /* key doesn't exist */
3842 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
3843 result = WERR_OK;
3844 goto done;
3847 DEBUG(5, ("winreg_del_driver: "
3848 "Could not open driver (%s,%s,%u): %s\n",
3849 info8->driver_name, info8->architecture,
3850 version, win_errstr(result)));
3851 goto done;
3855 if (is_valid_policy_hnd(&key_hnd)) {
3856 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &result);
3859 key_name = talloc_asprintf(tmp_ctx,
3860 "%s\\Environments\\%s\\Drivers\\Version-%u\\%s",
3861 TOP_LEVEL_CONTROL_KEY,
3862 info8->architecture, version,
3863 info8->driver_name);
3864 if (key_name == NULL) {
3865 result = WERR_NOMEM;
3866 goto done;
3869 status = dcerpc_winreg_delete_subkeys_recursive(tmp_ctx,
3870 winreg_handle,
3871 &hive_hnd,
3872 access_mask,
3873 key_name,
3874 &result);
3876 if (!NT_STATUS_IS_OK(status)){
3877 DEBUG(0, ("winreg_del_driver: "
3878 "Could not open driver (%s,%s,%u): %s\n",
3879 info8->driver_name, info8->architecture,
3880 version, nt_errstr(status)));
3881 goto done;
3884 if (!W_ERROR_IS_OK(result)) {
3885 DEBUG(0, ("winreg_del_driver: "
3886 "Could not open driver (%s,%s,%u): %s\n",
3887 info8->driver_name, info8->architecture,
3888 version, win_errstr(result)));
3889 goto done;
3892 result = WERR_OK;
3893 done:
3894 if (winreg_handle != NULL) {
3895 WERROR ignore;
3897 if (is_valid_policy_hnd(&key_hnd)) {
3898 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3900 if (is_valid_policy_hnd(&hive_hnd)) {
3901 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3905 TALLOC_FREE(tmp_ctx);
3906 return result;
3909 WERROR winreg_get_driver_list(TALLOC_CTX *mem_ctx,
3910 struct dcerpc_binding_handle *winreg_handle,
3911 const char *architecture,
3912 uint32_t version,
3913 uint32_t *num_drivers,
3914 const char ***drivers_p)
3916 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
3917 struct policy_handle hive_hnd, key_hnd;
3918 const char **drivers;
3919 TALLOC_CTX *tmp_ctx;
3920 WERROR result;
3921 NTSTATUS status;
3923 *num_drivers = 0;
3924 *drivers_p = NULL;
3926 ZERO_STRUCT(hive_hnd);
3927 ZERO_STRUCT(key_hnd);
3929 tmp_ctx = talloc_stackframe();
3930 if (tmp_ctx == NULL) {
3931 return WERR_NOMEM;
3934 /* use NULL for the driver name so we open the key that is
3935 * parent of all drivers for this architecture and version */
3936 result = winreg_printer_opendriver(tmp_ctx,
3937 winreg_handle,
3938 NULL,
3939 architecture,
3940 version,
3941 access_mask, false,
3942 &hive_hnd,
3943 &key_hnd);
3944 if (!W_ERROR_IS_OK(result)) {
3945 DEBUG(5, ("winreg_get_driver_list: "
3946 "Could not open key (%s,%u): %s\n",
3947 architecture, version, win_errstr(result)));
3948 result = WERR_OK;
3949 goto done;
3952 status = dcerpc_winreg_enum_keys(tmp_ctx,
3953 winreg_handle,
3954 &key_hnd,
3955 num_drivers,
3956 &drivers,
3957 &result);
3958 if (!NT_STATUS_IS_OK(status)) {
3959 result = ntstatus_to_werror(status);
3961 if (!W_ERROR_IS_OK(result)) {
3962 DEBUG(0, ("winreg_get_driver_list: "
3963 "Could not enumerate drivers for (%s,%u): %s\n",
3964 architecture, version, win_errstr(result)));
3965 goto done;
3968 *drivers_p = talloc_steal(mem_ctx, drivers);
3970 result = WERR_OK;
3971 done:
3972 if (winreg_handle != NULL) {
3973 WERROR ignore;
3975 if (is_valid_policy_hnd(&key_hnd)) {
3976 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &key_hnd, &ignore);
3978 if (is_valid_policy_hnd(&hive_hnd)) {
3979 dcerpc_winreg_CloseKey(winreg_handle, tmp_ctx, &hive_hnd, &ignore);
3983 TALLOC_FREE(tmp_ctx);
3984 return result;