s3-spoolss: Added winreg helper functions
[Samba/ekacnet.git] / source3 / rpc_server / srv_spoolss_util.c
blob79af761b0dfdeb3eab36c5c12517e4e508b9234d
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 "srv_spoolss_util.h"
24 #include "../librpc/gen_ndr/srv_winreg.h"
25 #include "../librpc/gen_ndr/cli_winreg.h"
27 #define TOP_LEVEL_PRINT_KEY "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Print"
28 #define TOP_LEVEL_PRINT_PRINTERS_KEY TOP_LEVEL_PRINT_KEY "\\Printers"
29 #define TOP_LEVEL_CONTROL_KEY "SYSTEM\\CurrentControlSet\\Control\\Print"
30 #define TOP_LEVEL_CONTROL_FORMS_KEY TOP_LEVEL_CONTROL_KEY "\\Forms"
32 /* FLAGS, NAME, with, height, left, top, right, bottom */
33 static const struct spoolss_FormInfo1 builtin_forms1[] = {
34 { SPOOLSS_FORM_BUILTIN, "Letter", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
35 { SPOOLSS_FORM_BUILTIN, "Letter Small", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
36 { SPOOLSS_FORM_BUILTIN, "Tabloid", {0x44368,0x696b8}, {0x0,0x0,0x44368,0x696b8} },
37 { SPOOLSS_FORM_BUILTIN, "Ledger", {0x696b8,0x44368}, {0x0,0x0,0x696b8,0x44368} },
38 { SPOOLSS_FORM_BUILTIN, "Legal", {0x34b5c,0x56d10}, {0x0,0x0,0x34b5c,0x56d10} },
39 { SPOOLSS_FORM_BUILTIN, "Statement", {0x221b4,0x34b5c}, {0x0,0x0,0x221b4,0x34b5c} },
40 { SPOOLSS_FORM_BUILTIN, "Executive", {0x2cf56,0x411cc}, {0x0,0x0,0x2cf56,0x411cc} },
41 { SPOOLSS_FORM_BUILTIN, "A3", {0x48828,0x668a0}, {0x0,0x0,0x48828,0x668a0} },
42 { SPOOLSS_FORM_BUILTIN, "A4", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
43 { SPOOLSS_FORM_BUILTIN, "A4 Small", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
44 { SPOOLSS_FORM_BUILTIN, "A5", {0x24220,0x33450}, {0x0,0x0,0x24220,0x33450} },
45 { SPOOLSS_FORM_BUILTIN, "B4 (JIS)", {0x3ebe8,0x58de0}, {0x0,0x0,0x3ebe8,0x58de0} },
46 { SPOOLSS_FORM_BUILTIN, "B5 (JIS)", {0x2c6f0,0x3ebe8}, {0x0,0x0,0x2c6f0,0x3ebe8} },
47 { SPOOLSS_FORM_BUILTIN, "Folio", {0x34b5c,0x509d8}, {0x0,0x0,0x34b5c,0x509d8} },
48 { SPOOLSS_FORM_BUILTIN, "Quarto", {0x347d8,0x43238}, {0x0,0x0,0x347d8,0x43238} },
49 { SPOOLSS_FORM_BUILTIN, "10x14", {0x3e030,0x56d10}, {0x0,0x0,0x3e030,0x56d10} },
50 { SPOOLSS_FORM_BUILTIN, "11x17", {0x44368,0x696b8}, {0x0,0x0,0x44368,0x696b8} },
51 { SPOOLSS_FORM_BUILTIN, "Note", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
52 { SPOOLSS_FORM_BUILTIN, "Envelope #9", {0x18079,0x37091}, {0x0,0x0,0x18079,0x37091} },
53 { SPOOLSS_FORM_BUILTIN, "Envelope #10", {0x19947,0x3ae94}, {0x0,0x0,0x19947,0x3ae94} },
54 { SPOOLSS_FORM_BUILTIN, "Envelope #11", {0x1be7c,0x40565}, {0x0,0x0,0x1be7c,0x40565} },
55 { SPOOLSS_FORM_BUILTIN, "Envelope #12", {0x1d74a,0x44368}, {0x0,0x0,0x1d74a,0x44368} },
56 { SPOOLSS_FORM_BUILTIN, "Envelope #14", {0x1f018,0x47504}, {0x0,0x0,0x1f018,0x47504} },
57 { SPOOLSS_FORM_BUILTIN, "C size sheet", {0x696b8,0x886d0}, {0x0,0x0,0x696b8,0x886d0} },
58 { SPOOLSS_FORM_BUILTIN, "D size sheet", {0x886d0,0xd2d70}, {0x0,0x0,0x886d0,0xd2d70} },
59 { SPOOLSS_FORM_BUILTIN, "E size sheet", {0xd2d70,0x110da0},{0x0,0x0,0xd2d70,0x110da0} },
60 { SPOOLSS_FORM_BUILTIN, "Envelope DL", {0x1adb0,0x35b60}, {0x0,0x0,0x1adb0,0x35b60} },
61 { SPOOLSS_FORM_BUILTIN, "Envelope C5", {0x278d0,0x37e88}, {0x0,0x0,0x278d0,0x37e88} },
62 { SPOOLSS_FORM_BUILTIN, "Envelope C3", {0x4f1a0,0x6fd10}, {0x0,0x0,0x4f1a0,0x6fd10} },
63 { SPOOLSS_FORM_BUILTIN, "Envelope C4", {0x37e88,0x4f1a0}, {0x0,0x0,0x37e88,0x4f1a0} },
64 { SPOOLSS_FORM_BUILTIN, "Envelope C6", {0x1bd50,0x278d0}, {0x0,0x0,0x1bd50,0x278d0} },
65 { SPOOLSS_FORM_BUILTIN, "Envelope C65", {0x1bd50,0x37e88}, {0x0,0x0,0x1bd50,0x37e88} },
66 { SPOOLSS_FORM_BUILTIN, "Envelope B4", {0x3d090,0x562e8}, {0x0,0x0,0x3d090,0x562e8} },
67 { SPOOLSS_FORM_BUILTIN, "Envelope B5", {0x2af80,0x3d090}, {0x0,0x0,0x2af80,0x3d090} },
68 { SPOOLSS_FORM_BUILTIN, "Envelope B6", {0x2af80,0x1e848}, {0x0,0x0,0x2af80,0x1e848} },
69 { SPOOLSS_FORM_BUILTIN, "Envelope", {0x1adb0,0x38270}, {0x0,0x0,0x1adb0,0x38270} },
70 { SPOOLSS_FORM_BUILTIN, "Envelope Monarch", {0x18079,0x2e824}, {0x0,0x0,0x18079,0x2e824} },
71 { SPOOLSS_FORM_BUILTIN, "6 3/4 Envelope", {0x167ab,0x284ec}, {0x0,0x0,0x167ab,0x284ec} },
72 { SPOOLSS_FORM_BUILTIN, "US Std Fanfold", {0x5c3e1,0x44368}, {0x0,0x0,0x5c3e1,0x44368} },
73 { SPOOLSS_FORM_BUILTIN, "German Std Fanfold", {0x34b5c,0x4a6a0}, {0x0,0x0,0x34b5c,0x4a6a0} },
74 { SPOOLSS_FORM_BUILTIN, "German Legal Fanfold", {0x34b5c,0x509d8}, {0x0,0x0,0x34b5c,0x509d8} },
75 { SPOOLSS_FORM_BUILTIN, "B4 (ISO)", {0x3d090,0x562e8}, {0x0,0x0,0x3d090,0x562e8} },
76 { SPOOLSS_FORM_BUILTIN, "Japanese Postcard", {0x186a0,0x24220}, {0x0,0x0,0x186a0,0x24220} },
77 { SPOOLSS_FORM_BUILTIN, "9x11", {0x37cf8,0x44368}, {0x0,0x0,0x37cf8,0x44368} },
78 { SPOOLSS_FORM_BUILTIN, "10x11", {0x3e030,0x44368}, {0x0,0x0,0x3e030,0x44368} },
79 { SPOOLSS_FORM_BUILTIN, "15x11", {0x5d048,0x44368}, {0x0,0x0,0x5d048,0x44368} },
80 { SPOOLSS_FORM_BUILTIN, "Envelope Invite", {0x35b60,0x35b60}, {0x0,0x0,0x35b60,0x35b60} },
81 { SPOOLSS_FORM_BUILTIN, "Reserved48", {0x1,0x1}, {0x0,0x0,0x1,0x1} },
82 { SPOOLSS_FORM_BUILTIN, "Reserved49", {0x1,0x1}, {0x0,0x0,0x1,0x1} },
83 { SPOOLSS_FORM_BUILTIN, "Letter Extra", {0x3ae94,0x4a6a0}, {0x0,0x0,0x3ae94,0x4a6a0} },
84 { SPOOLSS_FORM_BUILTIN, "Legal Extra", {0x3ae94,0x5d048}, {0x0,0x0,0x3ae94,0x5d048} },
85 { SPOOLSS_FORM_BUILTIN, "Tabloid Extra", {0x4a6a0,0x6f9f0}, {0x0,0x0,0x4a6a0,0x6f9f0} },
86 { SPOOLSS_FORM_BUILTIN, "A4 Extra", {0x397c2,0x4eb16}, {0x0,0x0,0x397c2,0x4eb16} },
87 { SPOOLSS_FORM_BUILTIN, "Letter Transverse", {0x34b5c,0x44368}, {0x0,0x0,0x34b5c,0x44368} },
88 { SPOOLSS_FORM_BUILTIN, "A4 Transverse", {0x33450,0x48828}, {0x0,0x0,0x33450,0x48828} },
89 { SPOOLSS_FORM_BUILTIN, "Letter Extra Transverse", {0x3ae94,0x4a6a0}, {0x0,0x0,0x3ae94,0x4a6a0} },
90 { SPOOLSS_FORM_BUILTIN, "Super A", {0x376b8,0x56ea0}, {0x0,0x0,0x376b8,0x56ea0} },
91 { SPOOLSS_FORM_BUILTIN, "Super B", {0x4a768,0x76e58}, {0x0,0x0,0x4a768,0x76e58} },
92 { SPOOLSS_FORM_BUILTIN, "Letter Plus", {0x34b5c,0x4eb16}, {0x0,0x0,0x34b5c,0x4eb16} },
93 { SPOOLSS_FORM_BUILTIN, "A4 Plus", {0x33450,0x50910}, {0x0,0x0,0x33450,0x50910} },
94 { SPOOLSS_FORM_BUILTIN, "A5 Transverse", {0x24220,0x33450}, {0x0,0x0,0x24220,0x33450} },
95 { SPOOLSS_FORM_BUILTIN, "B5 (JIS) Transverse", {0x2c6f0,0x3ebe8}, {0x0,0x0,0x2c6f0,0x3ebe8} },
96 { SPOOLSS_FORM_BUILTIN, "A3 Extra", {0x4e9d0,0x6ca48}, {0x0,0x0,0x4e9d0,0x6ca48} },
97 { SPOOLSS_FORM_BUILTIN, "A5 Extra", {0x2a7b0,0x395f8}, {0x0,0x0,0x2a7b0,0x395f8} },
98 { SPOOLSS_FORM_BUILTIN, "B5 (ISO) Extra", {0x31128,0x43620}, {0x0,0x0,0x31128,0x43620} },
99 { SPOOLSS_FORM_BUILTIN, "A2", {0x668a0,0x91050}, {0x0,0x0,0x668a0,0x91050} },
100 { SPOOLSS_FORM_BUILTIN, "A3 Transverse", {0x48828,0x668a0}, {0x0,0x0,0x48828,0x668a0} },
101 { SPOOLSS_FORM_BUILTIN, "A3 Extra Transverse", {0x4e9d0,0x6ca48}, {0x0,0x0,0x4e9d0,0x6ca48} },
102 { SPOOLSS_FORM_BUILTIN, "Japanese Double Postcard", {0x30d40,0x24220}, {0x0,0x0,0x30d40,0x24220} },
103 { SPOOLSS_FORM_BUILTIN, "A6", {0x19a28,0x24220}, {0x0,0x0,0x19a28,0x24220} },
104 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Kaku #2", {0x3a980,0x510e0}, {0x0,0x0,0x3a980,0x510e0} },
105 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Kaku #3", {0x34bc0,0x43a08}, {0x0,0x0,0x34bc0,0x43a08} },
106 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Chou #3", {0x1d4c0,0x395f8}, {0x0,0x0,0x1d4c0,0x395f8} },
107 { SPOOLSS_FORM_BUILTIN, "Japanese Envelope Chou #4", {0x15f90,0x320c8}, {0x0,0x0,0x15f90,0x320c8} },
108 { SPOOLSS_FORM_BUILTIN, "Letter Rotated", {0x44368,0x34b5c}, {0x0,0x0,0x44368,0x34b5c} },
109 { SPOOLSS_FORM_BUILTIN, "A3 Rotated", {0x668a0,0x48828}, {0x0,0x0,0x668a0,0x48828} },
110 { SPOOLSS_FORM_BUILTIN, "A4 Rotated", {0x48828,0x33450}, {0x0,0x0,0x48828,0x33450} },
111 { SPOOLSS_FORM_BUILTIN, "A5 Rotated", {0x33450,0x24220}, {0x0,0x0,0x33450,0x24220} },
112 { SPOOLSS_FORM_BUILTIN, "B4 (JIS) Rotated", {0x58de0,0x3ebe8}, {0x0,0x0,0x58de0,0x3ebe8} },
113 { SPOOLSS_FORM_BUILTIN, "B5 (JIS) Rotated", {0x3ebe8,0x2c6f0}, {0x0,0x0,0x3ebe8,0x2c6f0} },
114 { SPOOLSS_FORM_BUILTIN, "Japanese Postcard Rotated", {0x24220,0x186a0}, {0x0,0x0,0x24220,0x186a0} },
115 { SPOOLSS_FORM_BUILTIN, "Double Japan Postcard Rotated", {0x24220,0x30d40}, {0x0,0x0,0x24220,0x30d40} },
116 { SPOOLSS_FORM_BUILTIN, "A6 Rotated", {0x24220,0x19a28}, {0x0,0x0,0x24220,0x19a28} },
117 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Kaku #2 Rotated", {0x510e0,0x3a980}, {0x0,0x0,0x510e0,0x3a980} },
118 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Kaku #3 Rotated", {0x43a08,0x34bc0}, {0x0,0x0,0x43a08,0x34bc0} },
119 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Chou #3 Rotated", {0x395f8,0x1d4c0}, {0x0,0x0,0x395f8,0x1d4c0} },
120 { SPOOLSS_FORM_BUILTIN, "Japan Envelope Chou #4 Rotated", {0x320c8,0x15f90}, {0x0,0x0,0x320c8,0x15f90} },
121 { SPOOLSS_FORM_BUILTIN, "B6 (JIS)", {0x1f400,0x2c6f0}, {0x0,0x0,0x1f400,0x2c6f0} },
122 { SPOOLSS_FORM_BUILTIN, "B6 (JIS) Rotated", {0x2c6f0,0x1f400}, {0x0,0x0,0x2c6f0,0x1f400} },
123 { SPOOLSS_FORM_BUILTIN, "12x11", {0x4a724,0x443e1}, {0x0,0x0,0x4a724,0x443e1} },
124 { SPOOLSS_FORM_BUILTIN, "Japan Envelope You #4", {0x19a28,0x395f8}, {0x0,0x0,0x19a28,0x395f8} },
125 { SPOOLSS_FORM_BUILTIN, "Japan Envelope You #4 Rotated", {0x395f8,0x19a28}, {0x0,0x0,0x395f8,0x19a28} },
126 { SPOOLSS_FORM_BUILTIN, "PRC 16K", {0x2de60,0x3f7a0}, {0x0,0x0,0x2de60,0x3f7a0} },
127 { SPOOLSS_FORM_BUILTIN, "PRC 32K", {0x1fbd0,0x2cec0}, {0x0,0x0,0x1fbd0,0x2cec0} },
128 { SPOOLSS_FORM_BUILTIN, "PRC 32K(Big)", {0x222e0,0x318f8}, {0x0,0x0,0x222e0,0x318f8} },
129 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #1", {0x18e70,0x28488}, {0x0,0x0,0x18e70,0x28488} },
130 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #2", {0x18e70,0x2af80}, {0x0,0x0,0x18e70,0x2af80} },
131 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #3", {0x1e848,0x2af80}, {0x0,0x0,0x1e848,0x2af80} },
132 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #4", {0x1adb0,0x32c80}, {0x0,0x0,0x1adb0,0x32c80} },
133 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #5", {0x1adb0,0x35b60}, {0x0,0x0,0x1adb0,0x35b60} },
134 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #6", {0x1d4c0,0x38270}, {0x0,0x0,0x1d4c0,0x38270} },
135 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #7", {0x27100,0x38270}, {0x0,0x0,0x27100,0x38270} },
136 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #8", {0x1d4c0,0x4b708}, {0x0,0x0,0x1d4c0,0x4b708} },
137 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #9", {0x37e88,0x4f1a0}, {0x0,0x0,0x37e88,0x4f1a0} },
138 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #10", {0x4f1a0,0x6fd10}, {0x0,0x0,0x4f1a0,0x6fd10} },
139 { SPOOLSS_FORM_BUILTIN, "PRC 16K Rotated", {0x3f7a0,0x2de60}, {0x0,0x0,0x3f7a0,0x2de60} },
140 { SPOOLSS_FORM_BUILTIN, "PRC 32K Rotated", {0x2cec0,0x1fbd0}, {0x0,0x0,0x2cec0,0x1fbd0} },
141 { SPOOLSS_FORM_BUILTIN, "PRC 32K(Big) Rotated", {0x318f8,0x222e0}, {0x0,0x0,0x318f8,0x222e0} },
142 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #1 Rotated", {0x28488,0x18e70}, {0x0,0x0,0x28488,0x18e70} },
143 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #2 Rotated", {0x2af80,0x18e70}, {0x0,0x0,0x2af80,0x18e70} },
144 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #3 Rotated", {0x2af80,0x1e848}, {0x0,0x0,0x2af80,0x1e848} },
145 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #4 Rotated", {0x32c80,0x1adb0}, {0x0,0x0,0x32c80,0x1adb0} },
146 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #5 Rotated", {0x35b60,0x1adb0}, {0x0,0x0,0x35b60,0x1adb0} },
147 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #6 Rotated", {0x38270,0x1d4c0}, {0x0,0x0,0x38270,0x1d4c0} },
148 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #7 Rotated", {0x38270,0x27100}, {0x0,0x0,0x38270,0x27100} },
149 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #8 Rotated", {0x4b708,0x1d4c0}, {0x0,0x0,0x4b708,0x1d4c0} },
150 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #9 Rotated", {0x4f1a0,0x37e88}, {0x0,0x0,0x4f1a0,0x37e88} },
151 { SPOOLSS_FORM_BUILTIN, "PRC Envelope #10 Rotated", {0x6fd10,0x4f1a0}, {0x0,0x0,0x6fd10,0x4f1a0} }
154 /********************************************************************
155 static helper functions
156 ********************************************************************/
158 /****************************************************************************
159 Update the changeid time.
160 ****************************************************************************/
162 * @internal
164 * @brief Update the ChangeID time of a printer.
166 * This is SO NASTY as some drivers need this to change, others need it
167 * static. This value will change every second, and I must hope that this
168 * is enough..... DON'T CHANGE THIS CODE WITHOUT A TEST MATRIX THE SIZE OF
169 * UTAH ! JRA.
171 * @return The ChangeID.
173 static uint32_t winreg_printer_rev_changeid(void)
175 struct timeval tv;
177 get_process_uptime(&tv);
179 #if 1 /* JERRY */
180 /* Return changeid as msec since spooler restart */
181 return tv.tv_sec * 1000 + tv.tv_usec / 1000;
182 #else
184 * This setting seems to work well but is too untested
185 * to replace the above calculation. Left in for experiementation
186 * of the reader --jerry (Tue Mar 12 09:15:05 CST 2002)
188 return tv.tv_sec * 10 + tv.tv_usec / 100000;
189 #endif
193 * @internal
195 * @brief Connect to the interal winreg server and open the given printer key.
197 * The function will create the needed subkeys if they don't exist.
199 * @param[in] mem_ctx The memory context to use.
201 * @param[in] server_info The supplied server info.
203 * @param[out] winreg_pipe A pointer for the winreg rpc client pipe.
205 * @param[in] path The path to the key to open.
207 * @param[in] key The key to open.
209 * @param[in] create_key Set to true if the key should be created if it
210 * doesn't exist.
212 * @param[in] access_mask The access mask to open the key.
214 * @param[out] hive_handle A policy handle for the opened hive.
216 * @param[out] key_handle A policy handle for the opened key.
218 * @return WERR_OK on success, the corresponding DOS error
219 * code if something gone wrong.
221 static WERROR winreg_printer_openkey(TALLOC_CTX *mem_ctx,
222 struct auth_serversupplied_info *server_info,
223 struct rpc_pipe_client **winreg_pipe,
224 const char *path,
225 const char *key,
226 bool create_key,
227 uint32_t access_mask,
228 struct policy_handle *hive_handle,
229 struct policy_handle *key_handle)
231 struct rpc_pipe_client *pipe_handle;
232 struct winreg_String wkey, wkeyclass;
233 char *keyname;
234 NTSTATUS status;
235 WERROR result = WERR_OK;
237 /* create winreg connection */
238 status = rpc_pipe_open_internal(mem_ctx,
239 &ndr_table_winreg.syntax_id,
240 rpc_winreg_dispatch,
241 server_info,
242 &pipe_handle);
243 if (!NT_STATUS_IS_OK(status)) {
244 DEBUG(0, ("winreg_printer_openkey: Could not connect to winreg_pipe: %s\n",
245 nt_errstr(status)));
246 return ntstatus_to_werror(status);
249 status = rpccli_winreg_OpenHKLM(pipe_handle,
250 mem_ctx,
251 NULL,
252 access_mask,
253 hive_handle,
254 &result);
255 if (!NT_STATUS_IS_OK(status)) {
256 DEBUG(0, ("winreg_printer_openkey: Could not open HKLM hive: %s\n",
257 nt_errstr(status)));
258 talloc_free(pipe_handle);
259 if (!W_ERROR_IS_OK(result)) {
260 return result;
262 return ntstatus_to_werror(status);
265 if (key && *key) {
266 keyname = talloc_asprintf(mem_ctx, "%s\\%s", path, key);
267 } else {
268 keyname = talloc_strdup(mem_ctx, path);
270 if (keyname == NULL) {
271 talloc_free(pipe_handle);
272 return WERR_NOMEM;
275 ZERO_STRUCT(wkey);
276 wkey.name = keyname;
278 if (create_key) {
279 enum winreg_CreateAction action = REG_ACTION_NONE;
281 ZERO_STRUCT(wkeyclass);
282 wkeyclass.name = "";
284 status = rpccli_winreg_CreateKey(pipe_handle,
285 mem_ctx,
286 hive_handle,
287 wkey,
288 wkeyclass,
290 access_mask,
291 NULL,
292 key_handle,
293 &action,
294 &result);
295 switch (action) {
296 case REG_ACTION_NONE:
297 DEBUG(8, ("winreg_printer_openkey:createkey did nothing -- huh?\n"));
298 break;
299 case REG_CREATED_NEW_KEY:
300 DEBUG(8, ("winreg_printer_openkey: createkey created %s\n", keyname));
301 break;
302 case REG_OPENED_EXISTING_KEY:
303 DEBUG(8, ("winreg_printer_openkey: createkey opened existing %s\n", keyname));
304 break;
306 } else {
307 status = rpccli_winreg_OpenKey(pipe_handle,
308 mem_ctx,
309 hive_handle,
310 wkey,
312 access_mask,
313 key_handle,
314 &result);
316 if (!NT_STATUS_IS_OK(status)) {
317 talloc_free(pipe_handle);
318 if (!W_ERROR_IS_OK(result)) {
319 return result;
321 return ntstatus_to_werror(status);
324 *winreg_pipe = pipe_handle;
326 return WERR_OK;
330 * @brief Create the registry keyname for the given printer.
332 * @param[in] mem_ctx The memory context to use.
334 * @param[in] printer The name of the printer to get the registry key.
336 * @return The registry key or NULL on error.
338 static char *winreg_printer_data_keyname(TALLOC_CTX *mem_ctx, const char *printer) {
339 return talloc_asprintf(mem_ctx, "%s\\%s", TOP_LEVEL_PRINT_PRINTERS_KEY, printer);
343 * @internal
345 * @brief Enumerate values of an opened key handle and retrieve the data.
347 * @param[in] mem_ctx The memory context to use.
349 * @param[in] pipe_handle The pipe handle for the rpc connection.
351 * @param[in] key_hnd The opened key handle.
353 * @param[out] pnum_values A pointer to store he number of values found.
355 * @param[out] pnum_values A pointer to store the number of values we found.
357 * @return WERR_OK on success, the corresponding DOS error
358 * code if something gone wrong.
360 static WERROR winreg_printer_enumvalues(TALLOC_CTX *mem_ctx,
361 struct rpc_pipe_client *pipe_handle,
362 struct policy_handle *key_hnd,
363 uint32_t *pnum_values,
364 struct spoolss_PrinterEnumValues **penum_values)
366 TALLOC_CTX *tmp_ctx;
367 uint32_t num_subkeys, max_subkeylen, max_classlen;
368 uint32_t num_values, max_valnamelen, max_valbufsize;
369 uint32_t secdescsize;
370 uint32_t i;
371 NTTIME last_changed_time;
372 struct winreg_String classname;
374 struct spoolss_PrinterEnumValues *enum_values;
376 WERROR result = WERR_OK;
377 NTSTATUS status;
379 tmp_ctx = talloc_new(mem_ctx);
380 if (tmp_ctx == NULL) {
381 return WERR_NOMEM;
384 ZERO_STRUCT(classname);
386 status = rpccli_winreg_QueryInfoKey(pipe_handle,
387 tmp_ctx,
388 key_hnd,
389 &classname,
390 &num_subkeys,
391 &max_subkeylen,
392 &max_classlen,
393 &num_values,
394 &max_valnamelen,
395 &max_valbufsize,
396 &secdescsize,
397 &last_changed_time,
398 &result);
399 if (!NT_STATUS_IS_OK(status)) {
400 DEBUG(0, ("winreg_printer_enumvalues: Could not query info: %s\n",
401 nt_errstr(status)));
402 if (!W_ERROR_IS_OK(result)) {
403 goto error;
405 result = ntstatus_to_werror(status);
406 goto error;
409 if (num_values == 0) {
410 *pnum_values = 0;
411 TALLOC_FREE(tmp_ctx);
412 return WERR_OK;
415 enum_values = TALLOC_ARRAY(tmp_ctx, struct spoolss_PrinterEnumValues, num_values);
416 if (enum_values == NULL) {
417 result = WERR_NOMEM;
418 goto error;
421 for (i = 0; i < num_values; i++) {
422 struct spoolss_PrinterEnumValues val;
423 struct winreg_ValNameBuf name_buf;
424 enum winreg_Type type = REG_NONE;
425 uint8_t *data = NULL;
426 uint32_t data_size;
427 uint32_t length;
428 char n = '\0';;
430 name_buf.name = &n;
431 name_buf.size = max_valnamelen + 2;
432 name_buf.length = 0;
434 data_size = max_valbufsize;
435 data = (uint8_t *) TALLOC(tmp_ctx, data_size);
436 length = 0;
438 status = rpccli_winreg_EnumValue(pipe_handle,
439 tmp_ctx,
440 key_hnd,
442 &name_buf,
443 &type,
444 data,
445 &data_size,
446 &length,
447 &result);
448 if (W_ERROR_EQUAL(result, WERR_NO_MORE_ITEMS) ) {
449 result = WERR_OK;
450 status = NT_STATUS_OK;
451 break;
454 if (!NT_STATUS_IS_OK(status)) {
455 DEBUG(0, ("winreg_printer_enumvalues: Could not enumerate values: %s\n",
456 nt_errstr(status)));
457 if (!W_ERROR_IS_OK(result)) {
458 goto error;
460 result = ntstatus_to_werror(status);
461 goto error;
464 if (name_buf.name == NULL) {
465 result = WERR_INVALID_PARAMETER;
466 goto error;
469 val.value_name = talloc_strdup(enum_values, name_buf.name);
470 if (val.value_name == NULL) {
471 result = WERR_NOMEM;
472 goto error;
474 val.value_name_len = strlen_m_term(val.value_name) * 2;
476 val.type = type;
477 val.data_length = data_size;
478 val.data = NULL;
479 if (val.data_length) {
480 val.data = talloc(enum_values, DATA_BLOB);
481 if (val.data == NULL) {
482 result = WERR_NOMEM;
483 goto error;
485 *val.data = data_blob_talloc(enum_values, data, data_size);
488 enum_values[i] = val;
491 *pnum_values = num_values;
492 if (penum_values) {
493 *penum_values = talloc_move(mem_ctx, &enum_values);
496 result = WERR_OK;
498 error:
499 TALLOC_FREE(tmp_ctx);
500 return result;
504 * @internal
506 * @brief Enumerate subkeys of an opened key handle and get the names.
508 * @param[in] mem_ctx The memory context to use.
510 * @param[in] pipe_handle The pipe handle for the rpc connection.
512 * @param[in] key_hnd The opened key handle.
514 * @param[in] pnum_subkeys A pointer to store the number of found subkeys.
516 * @param[in] psubkeys A pointer to an array to store the found names of
517 * subkeys.
519 * @return WERR_OK on success, the corresponding DOS error
520 * code if something gone wrong.
522 static WERROR winreg_printer_enumkeys(TALLOC_CTX *mem_ctx,
523 struct rpc_pipe_client *pipe_handle,
524 struct policy_handle *key_hnd,
525 uint32_t *pnum_subkeys,
526 const char ***psubkeys)
528 TALLOC_CTX *tmp_ctx;
529 const char **subkeys;
530 uint32_t num_subkeys, max_subkeylen, max_classlen;
531 uint32_t num_values, max_valnamelen, max_valbufsize;
532 uint32_t i;
533 NTTIME last_changed_time;
534 uint32_t secdescsize;
535 struct winreg_String classname;
536 WERROR result = WERR_OK;
537 NTSTATUS status;
539 tmp_ctx = talloc_new(mem_ctx);
540 if (tmp_ctx == NULL) {
541 return WERR_NOMEM;
544 ZERO_STRUCT(classname);
546 status = rpccli_winreg_QueryInfoKey(pipe_handle,
547 tmp_ctx,
548 key_hnd,
549 &classname,
550 &num_subkeys,
551 &max_subkeylen,
552 &max_classlen,
553 &num_values,
554 &max_valnamelen,
555 &max_valbufsize,
556 &secdescsize,
557 &last_changed_time,
558 &result);
559 if (!NT_STATUS_IS_OK(status)) {
560 DEBUG(0, ("winreg_printer_enumkeys: Could not query info: %s\n",
561 nt_errstr(status)));
562 if (!W_ERROR_IS_OK(result)) {
563 goto error;
565 result = ntstatus_to_werror(status);
566 goto error;
569 subkeys = talloc_zero_array(tmp_ctx, const char *, num_subkeys + 2);
570 if (subkeys == NULL) {
571 result = WERR_NOMEM;
572 goto error;
575 if (num_subkeys == 0) {
576 subkeys[0] = talloc_strdup(subkeys, "");
577 if (subkeys[0] == NULL) {
578 result = WERR_NOMEM;
579 goto error;
581 *pnum_subkeys = 0;
582 if (psubkeys) {
583 *psubkeys = talloc_move(mem_ctx, &subkeys);
586 TALLOC_FREE(tmp_ctx);
587 return WERR_OK;
590 for (i = 0; i < num_subkeys; i++) {
591 char c = '\0';
592 char n = '\0';
593 char *name = NULL;
594 struct winreg_StringBuf class_buf;
595 struct winreg_StringBuf name_buf;
596 NTTIME modtime;
598 class_buf.name = &c;
599 class_buf.size = max_classlen + 2;
600 class_buf.length = 0;
602 name_buf.name = &n;
603 name_buf.size = max_subkeylen + 2;
604 name_buf.length = 0;
606 ZERO_STRUCT(modtime);
608 status = rpccli_winreg_EnumKey(pipe_handle,
609 tmp_ctx,
610 key_hnd,
612 &name_buf,
613 &class_buf,
614 &modtime,
615 &result);
616 if (W_ERROR_EQUAL(result, WERR_NO_MORE_ITEMS) ) {
617 result = WERR_OK;
618 status = NT_STATUS_OK;
619 break;
622 if (!NT_STATUS_IS_OK(status)) {
623 DEBUG(0, ("winreg_printer_enumkeys: Could not enumerate keys: %s\n",
624 nt_errstr(status)));
625 if (!W_ERROR_IS_OK(result)) {
626 goto error;
628 result = ntstatus_to_werror(status);
629 goto error;
632 if (name_buf.name == NULL) {
633 result = WERR_INVALID_PARAMETER;
634 goto error;
637 name = talloc_strdup(subkeys, name_buf.name);
638 if (name == NULL) {
639 result = WERR_NOMEM;
640 goto error;
643 subkeys[i] = name;
646 *pnum_subkeys = num_subkeys;
647 if (psubkeys) {
648 *psubkeys = talloc_move(mem_ctx, &subkeys);
651 error:
652 TALLOC_FREE(tmp_ctx);
653 return result;
657 * @internal
659 * @brief A function to delete a key and its subkeys recurively.
661 * @param[in] mem_ctx The memory context to use.
663 * @param[in] pipe_handle The pipe handle for the rpc connection.
665 * @param[in] hive_handle A opened hive handle to the key.
667 * @param[in] access_mask The access mask to access the key.
669 * @param[in] key The key to delete
671 * @return WERR_OK on success, the corresponding DOS error
672 * code if something gone wrong.
674 static WERROR winreg_printer_delete_subkeys(TALLOC_CTX *mem_ctx,
675 struct rpc_pipe_client *pipe_handle,
676 struct policy_handle *hive_handle,
677 uint32_t access_mask,
678 const char *key)
680 const char **subkeys = NULL;
681 uint32_t num_subkeys = 0;
682 struct policy_handle key_hnd;
683 struct winreg_String wkey;
684 WERROR result = WERR_OK;
685 NTSTATUS status;
686 uint32_t i;
688 ZERO_STRUCT(key_hnd);
689 wkey.name = key;
691 DEBUG(2, ("winreg_printer_delete_subkeys: delete key %s\n", key));
692 /* open the key */
693 status = rpccli_winreg_OpenKey(pipe_handle,
694 mem_ctx,
695 hive_handle,
696 wkey,
698 access_mask,
699 &key_hnd,
700 &result);
701 if (!NT_STATUS_IS_OK(status)) {
702 DEBUG(0, ("winreg_printer_delete_subkeys: Could not open key %s: %s\n",
703 wkey.name, nt_errstr(status)));
704 if (!W_ERROR_IS_OK(result)) {
705 return result;
707 return ntstatus_to_werror(status);
710 result = winreg_printer_enumkeys(mem_ctx,
711 pipe_handle,
712 &key_hnd,
713 &num_subkeys,
714 &subkeys);
715 if (!W_ERROR_IS_OK(result)) {
716 goto done;
719 for (i = 0; i < num_subkeys; i++) {
720 /* create key + subkey */
721 char *subkey = talloc_asprintf(mem_ctx, "%s\\%s", key, subkeys[i]);
722 if (subkey == NULL) {
723 goto done;
726 DEBUG(2, ("winreg_printer_delete_subkeys: delete subkey %s\n", subkey));
727 result = winreg_printer_delete_subkeys(mem_ctx,
728 pipe_handle,
729 hive_handle,
730 access_mask,
731 subkey);
732 if (!W_ERROR_IS_OK(result)) {
733 goto done;
737 if (is_valid_policy_hnd(&key_hnd)) {
738 rpccli_winreg_CloseKey(pipe_handle, mem_ctx, &key_hnd, NULL);
741 wkey.name = key;
743 status = rpccli_winreg_DeleteKey(pipe_handle,
744 mem_ctx,
745 hive_handle,
746 wkey,
747 &result);
749 done:
750 if (is_valid_policy_hnd(&key_hnd)) {
751 rpccli_winreg_CloseKey(pipe_handle, mem_ctx, &key_hnd, NULL);
754 return result;
757 static WERROR winreg_printer_write_sz(TALLOC_CTX *mem_ctx,
758 struct rpc_pipe_client *pipe_handle,
759 struct policy_handle *key_handle,
760 const char *value,
761 const char *data)
763 struct winreg_String wvalue;
764 DATA_BLOB blob;
765 WERROR result = WERR_OK;
766 NTSTATUS status;
768 wvalue.name = value;
769 if (data == NULL) {
770 blob = data_blob_string_const("");
771 } else {
772 if (!push_reg_sz(mem_ctx, NULL, &blob, data)) {
773 DEBUG(0, ("winreg_printer_write_sz: Could not marshall string %s for %s\n",
774 data, wvalue.name));
775 return WERR_NOMEM;
778 status = rpccli_winreg_SetValue(pipe_handle,
779 mem_ctx,
780 key_handle,
781 wvalue,
782 REG_SZ,
783 blob.data,
784 blob.length,
785 &result);
786 if (!NT_STATUS_IS_OK(status)) {
787 DEBUG(0, ("winreg_printer_write_sz: Could not set value %s: %s\n",
788 wvalue.name, win_errstr(result)));
789 if (!W_ERROR_IS_OK(result)) {
790 result = ntstatus_to_werror(status);
794 return result;
797 static WERROR winreg_printer_write_dword(TALLOC_CTX *mem_ctx,
798 struct rpc_pipe_client *pipe_handle,
799 struct policy_handle *key_handle,
800 const char *value,
801 uint32_t data)
803 struct winreg_String wvalue;
804 DATA_BLOB blob;
805 WERROR result = WERR_OK;
806 NTSTATUS status;
808 wvalue.name = value;
809 blob = data_blob_talloc(mem_ctx, NULL, 4);
810 SIVAL(blob.data, 0, data);
812 status = rpccli_winreg_SetValue(pipe_handle,
813 mem_ctx,
814 key_handle,
815 wvalue,
816 REG_DWORD,
817 blob.data,
818 blob.length,
819 &result);
820 if (!NT_STATUS_IS_OK(status)) {
821 DEBUG(0, ("winreg_printer_write_dword: Could not set value %s: %s\n",
822 wvalue.name, win_errstr(result)));
823 if (!W_ERROR_IS_OK(result)) {
824 result = ntstatus_to_werror(status);
828 return result;
831 static WERROR winreg_printer_write_binary(TALLOC_CTX *mem_ctx,
832 struct rpc_pipe_client *pipe_handle,
833 struct policy_handle *key_handle,
834 const char *value,
835 DATA_BLOB blob)
837 struct winreg_String wvalue;
838 WERROR result = WERR_OK;
839 NTSTATUS status;
841 wvalue.name = value;
842 status = rpccli_winreg_SetValue(pipe_handle,
843 mem_ctx,
844 key_handle,
845 wvalue,
846 REG_BINARY,
847 blob.data,
848 blob.length,
849 &result);
850 if (!NT_STATUS_IS_OK(status)) {
851 DEBUG(0, ("winreg_printer_write_binary: Could not set value %s: %s\n",
852 wvalue.name, win_errstr(result)));
853 if (!W_ERROR_IS_OK(result)) {
854 result = ntstatus_to_werror(status);
858 return result;
861 static WERROR winreg_printer_query_dword(TALLOC_CTX *mem_ctx,
862 struct rpc_pipe_client *pipe_handle,
863 struct policy_handle *key_handle,
864 const char *value,
865 uint32_t *data)
867 struct winreg_String wvalue;
868 enum winreg_Type type;
869 WERROR result = WERR_OK;
870 uint32_t value_len = 0;
871 NTSTATUS status;
872 DATA_BLOB blob;
874 wvalue.name = value;
875 status = rpccli_winreg_QueryValue(pipe_handle,
876 mem_ctx,
877 key_handle,
878 &wvalue,
879 &type,
880 NULL,
881 (uint32_t *) &blob.length,
882 &value_len,
883 &result);
884 if (!NT_STATUS_IS_OK(status)) {
885 DEBUG(0, ("winreg_printer_query_dword: Could not query value %s: %s\n",
886 wvalue.name, nt_errstr(status)));
887 if (!W_ERROR_IS_OK(result)) {
888 goto done;
890 result = ntstatus_to_werror(status);
891 goto done;
894 if (type != REG_DWORD) {
895 result = WERR_INVALID_DATATYPE;
896 goto done;
899 if (blob.length != 4) {
900 result = WERR_INVALID_DATA;
901 goto done;
904 blob.data = (uint8_t *) TALLOC(mem_ctx, blob.length);
905 if (blob.data == NULL) {
906 result = WERR_NOMEM;
907 goto done;
909 value_len = 0;
911 status = rpccli_winreg_QueryValue(pipe_handle,
912 mem_ctx,
913 key_handle,
914 &wvalue,
915 &type,
916 blob.data,
917 (uint32_t *) &blob.length,
918 &value_len,
919 &result);
920 if (!NT_STATUS_IS_OK(status)) {
921 DEBUG(0, ("winreg_printer_query_dword: Could not query value %s: %s\n",
922 wvalue.name, nt_errstr(status)));
923 if (!W_ERROR_IS_OK(result)) {
924 result = ntstatus_to_werror(status);
926 goto done;
929 if (data) {
930 *data = IVAL(blob.data, 0);
932 done:
933 return result;
936 static WERROR winreg_printer_write_multi_sz(TALLOC_CTX *mem_ctx,
937 struct rpc_pipe_client *pipe_handle,
938 struct policy_handle *key_handle,
939 const char *value,
940 const char **data)
942 struct winreg_String wvalue;
943 DATA_BLOB blob;
944 WERROR result = WERR_OK;
945 NTSTATUS status;
947 wvalue.name = value;
948 if (!push_reg_multi_sz(mem_ctx, NULL, &blob, data)) {
949 return WERR_NOMEM;
951 status = rpccli_winreg_SetValue(pipe_handle,
952 mem_ctx,
953 key_handle,
954 wvalue,
955 REG_MULTI_SZ,
956 blob.data,
957 blob.length,
958 &result);
959 if (!NT_STATUS_IS_OK(status)) {
960 DEBUG(0, ("winreg_printer_write_multi_sz: Could not set value %s: %s\n",
961 wvalue.name, win_errstr(result)));
962 if (!W_ERROR_IS_OK(result)) {
963 result = ntstatus_to_werror(status);
967 return result;
970 static WERROR winreg_printer_opendriver(TALLOC_CTX *mem_ctx,
971 struct auth_serversupplied_info *server_info,
972 const char *drivername,
973 const char *architecture,
974 uint32_t version,
975 uint32_t access_mask,
976 bool create,
977 struct rpc_pipe_client **winreg_pipe,
978 struct policy_handle *hive_hnd,
979 struct policy_handle *key_hnd)
981 WERROR result;
982 char *key_name;
984 key_name = talloc_asprintf(mem_ctx, "%s\\Environments\\%s\\Drivers\\Version-%u",
985 TOP_LEVEL_CONTROL_KEY,
986 architecture, version);
987 if (!key_name) {
988 return WERR_NOMEM;
991 result = winreg_printer_openkey(mem_ctx,
992 server_info,
993 winreg_pipe,
994 key_name,
995 drivername,
996 create,
997 access_mask,
998 hive_hnd,
999 key_hnd);
1000 return result;
1003 static WERROR winreg_enumval_to_dword(TALLOC_CTX *mem_ctx,
1004 struct spoolss_PrinterEnumValues *v,
1005 const char *valuename, uint32_t *dw)
1007 /* just return if it is not the one we are looking for */
1008 if (strcmp(valuename, v->value_name) != 0) {
1009 return WERR_NOT_FOUND;
1012 if (v->type != REG_DWORD) {
1013 return WERR_INVALID_DATATYPE;
1016 *dw = IVAL(v->data->data, 0);
1017 return WERR_OK;
1020 static WERROR winreg_enumval_to_sz(TALLOC_CTX *mem_ctx,
1021 struct spoolss_PrinterEnumValues *v,
1022 const char *valuename, const char **_str)
1024 /* just return if it is not the one we are looking for */
1025 if (strcmp(valuename, v->value_name) != 0) {
1026 return WERR_NOT_FOUND;
1029 if (v->type != REG_SZ) {
1030 return WERR_INVALID_DATATYPE;
1033 if (!pull_reg_sz(mem_ctx, NULL, v->data, _str)) {
1034 return WERR_NOMEM;
1037 return WERR_OK;
1040 static WERROR winreg_enumval_to_multi_sz(TALLOC_CTX *mem_ctx,
1041 struct spoolss_PrinterEnumValues *v,
1042 const char *valuename,
1043 const char ***array)
1045 /* just return if it is not the one we are looking for */
1046 if (strcmp(valuename, v->value_name) != 0) {
1047 return WERR_NOT_FOUND;
1050 if (v->type != REG_MULTI_SZ) {
1051 return WERR_INVALID_DATATYPE;
1054 if (!pull_reg_multi_sz(mem_ctx, NULL, v->data, array)) {
1055 return WERR_NOMEM;
1058 return WERR_OK;
1061 /********************************************************************
1062 Public winreg function for spoolss
1063 ********************************************************************/
1065 /* Set printer data over the winreg pipe. */
1066 WERROR winreg_set_printer_dataex(TALLOC_CTX *mem_ctx,
1067 struct auth_serversupplied_info *server_info,
1068 const char *printer,
1069 const char *key,
1070 const char *value,
1071 enum winreg_Type type,
1072 uint8_t *data,
1073 uint32_t data_size)
1075 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1076 struct rpc_pipe_client *winreg_pipe = NULL;
1077 struct policy_handle hive_hnd, key_hnd;
1078 struct winreg_String wvalue;
1079 char *path;
1080 WERROR result = WERR_OK;
1081 NTSTATUS status;
1082 TALLOC_CTX *tmp_ctx;
1084 tmp_ctx = talloc_new(mem_ctx);
1085 if (tmp_ctx == NULL) {
1086 return WERR_NOMEM;
1089 path = winreg_printer_data_keyname(tmp_ctx, printer);
1090 if (path == NULL) {
1091 TALLOC_FREE(tmp_ctx);
1092 return WERR_NOMEM;
1095 ZERO_STRUCT(hive_hnd);
1096 ZERO_STRUCT(key_hnd);
1098 DEBUG(8, ("winreg_set_printer_dataex: Open printer key %s, value %s, access_mask: 0x%05x for [%s]\n",
1099 key, value, access_mask, printer));
1100 result = winreg_printer_openkey(tmp_ctx,
1101 server_info,
1102 &winreg_pipe,
1103 path,
1104 key,
1105 true,
1106 access_mask,
1107 &hive_hnd,
1108 &key_hnd);
1109 if (!W_ERROR_IS_OK(result)) {
1110 DEBUG(0, ("winreg_set_printer_dataex: Could not open key %s: %s\n",
1111 key, win_errstr(result)));
1112 goto done;
1115 wvalue.name = value;
1116 status = rpccli_winreg_SetValue(winreg_pipe,
1117 tmp_ctx,
1118 &key_hnd,
1119 wvalue,
1120 type,
1121 data,
1122 data_size,
1123 &result);
1124 if (!NT_STATUS_IS_OK(status)) {
1125 DEBUG(0, ("winreg_set_printer_dataex: Could not set value %s: %s\n",
1126 value, nt_errstr(status)));
1127 if (!W_ERROR_IS_OK(result)) {
1128 goto done;
1130 result = ntstatus_to_werror(status);
1131 goto done;
1134 result = WERR_OK;
1135 done:
1136 if (winreg_pipe != NULL) {
1137 if (is_valid_policy_hnd(&key_hnd)) {
1138 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1140 if (is_valid_policy_hnd(&hive_hnd)) {
1141 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1145 TALLOC_FREE(tmp_ctx);
1146 return result;
1149 /* Get printer data over a winreg pipe. */
1150 WERROR winreg_get_printer_dataex(TALLOC_CTX *mem_ctx,
1151 struct auth_serversupplied_info *server_info,
1152 const char *printer,
1153 const char *key,
1154 const char *value,
1155 enum winreg_Type *type,
1156 uint8_t **data,
1157 uint32_t *data_size)
1159 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1160 struct rpc_pipe_client *winreg_pipe = NULL;
1161 struct policy_handle hive_hnd, key_hnd;
1162 struct winreg_String wvalue;
1163 enum winreg_Type type_in;
1164 char *path;
1165 uint8_t *data_in;
1166 uint32_t data_in_size = 0;
1167 uint32_t value_len = 0;
1168 WERROR result = WERR_OK;
1169 NTSTATUS status;
1170 TALLOC_CTX *tmp_ctx;
1172 tmp_ctx = talloc_new(mem_ctx);
1173 if (tmp_ctx == NULL) {
1174 return WERR_NOMEM;
1177 path = winreg_printer_data_keyname(tmp_ctx, printer);
1178 if (path == NULL) {
1179 TALLOC_FREE(tmp_ctx);
1180 return WERR_NOMEM;
1183 ZERO_STRUCT(hive_hnd);
1184 ZERO_STRUCT(key_hnd);
1186 result = winreg_printer_openkey(tmp_ctx,
1187 server_info,
1188 &winreg_pipe,
1189 path,
1190 key,
1191 false,
1192 access_mask,
1193 &hive_hnd,
1194 &key_hnd);
1195 if (!W_ERROR_IS_OK(result)) {
1196 DEBUG(0, ("winreg_get_printer_dataex: Could not open key %s: %s\n",
1197 key, win_errstr(result)));
1198 goto done;
1201 wvalue.name = value;
1204 * call QueryValue once with data == NULL to get the
1205 * needed memory size to be allocated, then allocate
1206 * data buffer and call again.
1208 status = rpccli_winreg_QueryValue(winreg_pipe,
1209 tmp_ctx,
1210 &key_hnd,
1211 &wvalue,
1212 &type_in,
1213 NULL,
1214 &data_in_size,
1215 &value_len,
1216 &result);
1217 if (!NT_STATUS_IS_OK(status)) {
1218 DEBUG(0, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
1219 value, nt_errstr(status)));
1220 if (!W_ERROR_IS_OK(result)) {
1221 goto done;
1223 result = ntstatus_to_werror(status);
1224 goto done;
1227 data_in = (uint8_t *) TALLOC(tmp_ctx, data_in_size);
1228 if (data_in == NULL) {
1229 result = WERR_NOMEM;
1230 goto done;
1232 value_len = 0;
1234 status = rpccli_winreg_QueryValue(winreg_pipe,
1235 tmp_ctx,
1236 &key_hnd,
1237 &wvalue,
1238 &type_in,
1239 data_in,
1240 &data_in_size,
1241 &value_len,
1242 &result);
1243 if (!NT_STATUS_IS_OK(status)) {
1244 DEBUG(0, ("winreg_get_printer_dataex: Could not query value %s: %s\n",
1245 value, nt_errstr(status)));
1246 if (!W_ERROR_IS_OK(result)) {
1247 result = ntstatus_to_werror(status);
1249 goto done;
1252 *type = type_in;
1253 *data_size = data_in_size;
1254 if (data_in_size) {
1255 *data = talloc_move(mem_ctx, &data_in);
1258 result = WERR_OK;
1259 done:
1260 if (winreg_pipe != NULL) {
1261 if (is_valid_policy_hnd(&key_hnd)) {
1262 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1264 if (is_valid_policy_hnd(&hive_hnd)) {
1265 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1269 TALLOC_FREE(tmp_ctx);
1270 return result;
1273 /* Enumerate on the values of a given key and provide the data. */
1274 WERROR winreg_enum_printer_dataex(TALLOC_CTX *mem_ctx,
1275 struct auth_serversupplied_info *server_info,
1276 const char *printer,
1277 const char *key,
1278 uint32_t *pnum_values,
1279 struct spoolss_PrinterEnumValues **penum_values)
1281 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1282 struct rpc_pipe_client *winreg_pipe = NULL;
1283 struct policy_handle hive_hnd, key_hnd;
1285 struct spoolss_PrinterEnumValues *enum_values = NULL;
1286 uint32_t num_values = 0;
1287 char *path;
1288 WERROR result = WERR_OK;
1290 TALLOC_CTX *tmp_ctx;
1292 tmp_ctx = talloc_new(mem_ctx);
1293 if (tmp_ctx == NULL) {
1294 return WERR_NOMEM;
1297 path = winreg_printer_data_keyname(tmp_ctx, printer);
1298 if (path == NULL) {
1299 TALLOC_FREE(tmp_ctx);
1300 return WERR_NOMEM;
1303 result = winreg_printer_openkey(tmp_ctx,
1304 server_info,
1305 &winreg_pipe,
1306 path,
1307 key,
1308 false,
1309 access_mask,
1310 &hive_hnd,
1311 &key_hnd);
1312 if (!W_ERROR_IS_OK(result)) {
1313 DEBUG(0, ("winreg_enum_printer_dataex: Could not open key %s: %s\n",
1314 key, win_errstr(result)));
1315 goto done;
1318 result = winreg_printer_enumvalues(tmp_ctx,
1319 winreg_pipe,
1320 &key_hnd,
1321 &num_values,
1322 &enum_values);
1323 if (!W_ERROR_IS_OK(result)) {
1324 DEBUG(0, ("winreg_enum_printer_dataex: Could not enumerate values in %s: %s\n",
1325 key, win_errstr(result)));
1326 goto done;
1329 *pnum_values = num_values;
1330 if (penum_values) {
1331 *penum_values = talloc_move(mem_ctx, &enum_values);
1334 result = WERR_OK;
1335 done:
1336 if (winreg_pipe != NULL) {
1337 if (is_valid_policy_hnd(&key_hnd)) {
1338 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1340 if (is_valid_policy_hnd(&hive_hnd)) {
1341 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1345 TALLOC_FREE(tmp_ctx);
1346 return result;
1349 /* Delete printer data over a winreg pipe. */
1350 WERROR winreg_delete_printer_dataex(TALLOC_CTX *mem_ctx,
1351 struct auth_serversupplied_info *server_info,
1352 const char *printer,
1353 const char *key,
1354 const char *value)
1356 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1357 struct rpc_pipe_client *winreg_pipe = NULL;
1358 struct policy_handle hive_hnd, key_hnd;
1359 struct winreg_String wvalue;
1360 char *path;
1361 WERROR result = WERR_OK;
1362 NTSTATUS status;
1364 TALLOC_CTX *tmp_ctx;
1366 tmp_ctx = talloc_new(mem_ctx);
1367 if (tmp_ctx == NULL) {
1368 return WERR_NOMEM;
1371 path = winreg_printer_data_keyname(tmp_ctx, printer);
1372 if (path == NULL) {
1373 TALLOC_FREE(tmp_ctx);
1374 return WERR_NOMEM;
1377 ZERO_STRUCT(hive_hnd);
1378 ZERO_STRUCT(key_hnd);
1380 result = winreg_printer_openkey(tmp_ctx,
1381 server_info,
1382 &winreg_pipe,
1383 path,
1384 key,
1385 false,
1386 access_mask,
1387 &hive_hnd,
1388 &key_hnd);
1389 if (!W_ERROR_IS_OK(result)) {
1390 DEBUG(0, ("winreg_delete_printer_dataex: Could not open key %s: %s\n",
1391 key, win_errstr(result)));
1392 goto done;
1395 wvalue.name = value;
1396 status = rpccli_winreg_DeleteValue(winreg_pipe,
1397 tmp_ctx,
1398 &key_hnd,
1399 wvalue,
1400 &result);
1401 if (!NT_STATUS_IS_OK(status)) {
1402 DEBUG(0, ("winreg_delete_printer_dataex: Could not delete value %s: %s\n",
1403 value, nt_errstr(status)));
1404 if (!W_ERROR_IS_OK(result)) {
1405 goto done;
1407 result = ntstatus_to_werror(status);
1408 goto done;
1411 result = WERR_OK;
1412 done:
1413 if (winreg_pipe != NULL) {
1414 if (is_valid_policy_hnd(&key_hnd)) {
1415 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1417 if (is_valid_policy_hnd(&hive_hnd)) {
1418 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1422 TALLOC_FREE(tmp_ctx);
1423 return result;
1426 /* Enumerate on the subkeys of a given key and provide the data. */
1427 WERROR winreg_enum_printer_key(TALLOC_CTX *mem_ctx,
1428 struct auth_serversupplied_info *server_info,
1429 const char *printer,
1430 const char *key,
1431 uint32_t *pnum_subkeys,
1432 const char ***psubkeys)
1434 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1435 struct rpc_pipe_client *winreg_pipe = NULL;
1436 struct policy_handle hive_hnd, key_hnd;
1437 char *path;
1438 const char **subkeys = NULL;
1439 uint32_t num_subkeys = -1;
1441 WERROR result = WERR_OK;
1443 TALLOC_CTX *tmp_ctx;
1445 tmp_ctx = talloc_new(mem_ctx);
1446 if (tmp_ctx == NULL) {
1447 return WERR_NOMEM;
1450 path = winreg_printer_data_keyname(tmp_ctx, printer);
1451 if (path == NULL) {
1452 TALLOC_FREE(tmp_ctx);
1453 return WERR_NOMEM;
1456 ZERO_STRUCT(hive_hnd);
1457 ZERO_STRUCT(key_hnd);
1459 result = winreg_printer_openkey(tmp_ctx,
1460 server_info,
1461 &winreg_pipe,
1462 path,
1463 key,
1464 false,
1465 access_mask,
1466 &hive_hnd,
1467 &key_hnd);
1468 if (!W_ERROR_IS_OK(result)) {
1469 DEBUG(0, ("winreg_enum_printer_key: Could not open key %s: %s\n",
1470 key, win_errstr(result)));
1471 goto done;
1474 result = winreg_printer_enumkeys(tmp_ctx,
1475 winreg_pipe,
1476 &key_hnd,
1477 &num_subkeys,
1478 &subkeys);
1479 if (!W_ERROR_IS_OK(result)) {
1480 DEBUG(0, ("winreg_enum_printer_key: Could not enumerate subkeys in %s: %s\n",
1481 key, win_errstr(result)));
1482 goto done;
1485 *pnum_subkeys = num_subkeys;
1486 if (psubkeys) {
1487 *psubkeys = talloc_move(mem_ctx, &subkeys);
1490 result = WERR_OK;
1491 done:
1492 if (winreg_pipe != NULL) {
1493 if (is_valid_policy_hnd(&key_hnd)) {
1494 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1496 if (is_valid_policy_hnd(&hive_hnd)) {
1497 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1501 TALLOC_FREE(tmp_ctx);
1502 return result;
1505 /* Delete a key with subkeys of a given printer. */
1506 WERROR winreg_delete_printer_key(TALLOC_CTX *mem_ctx,
1507 struct auth_serversupplied_info *server_info,
1508 const char *printer,
1509 const char *key)
1511 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1512 struct rpc_pipe_client *winreg_pipe = NULL;
1513 struct policy_handle hive_hnd, key_hnd;
1514 char *keyname;
1515 char *path;
1516 WERROR result;
1517 TALLOC_CTX *tmp_ctx;
1519 tmp_ctx = talloc_new(mem_ctx);
1520 if (tmp_ctx == NULL) {
1521 return WERR_NOMEM;
1524 path = winreg_printer_data_keyname(tmp_ctx, printer);
1525 if (path == NULL) {
1526 TALLOC_FREE(tmp_ctx);
1527 return WERR_NOMEM;
1530 result = winreg_printer_openkey(tmp_ctx,
1531 server_info,
1532 &winreg_pipe,
1533 path,
1534 key,
1535 false,
1536 access_mask,
1537 &hive_hnd,
1538 &key_hnd);
1539 if (!W_ERROR_IS_OK(result)) {
1540 /* key doesn't exist */
1541 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
1542 result = WERR_OK;
1543 goto done;
1546 DEBUG(0, ("winreg_delete_printer_key: Could not open key %s: %s\n",
1547 key, win_errstr(result)));
1548 goto done;
1551 if (is_valid_policy_hnd(&key_hnd)) {
1552 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1555 keyname = talloc_asprintf(tmp_ctx,
1556 "%s\\%s",
1557 path,
1558 key);
1559 if (keyname == NULL) {
1560 result = WERR_NOMEM;
1561 goto done;
1564 result = winreg_printer_delete_subkeys(tmp_ctx,
1565 winreg_pipe,
1566 &hive_hnd,
1567 access_mask,
1568 keyname);
1569 if (!W_ERROR_IS_OK(result)) {
1570 DEBUG(0, ("winreg_delete_printer_key: Could not delete key %s: %s\n",
1571 key, win_errstr(result)));
1572 goto done;
1575 done:
1576 if (winreg_pipe != NULL) {
1577 if (is_valid_policy_hnd(&key_hnd)) {
1578 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1580 if (is_valid_policy_hnd(&hive_hnd)) {
1581 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1585 TALLOC_FREE(tmp_ctx);
1586 return result;
1589 WERROR winreg_printer_update_changeid(TALLOC_CTX *mem_ctx,
1590 struct auth_serversupplied_info *server_info,
1591 const char *printer)
1593 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1594 struct rpc_pipe_client *winreg_pipe = NULL;
1595 struct policy_handle hive_hnd, key_hnd;
1596 char *path;
1597 WERROR result;
1598 TALLOC_CTX *tmp_ctx;
1600 tmp_ctx = talloc_new(mem_ctx);
1601 if (tmp_ctx == NULL) {
1602 return WERR_NOMEM;
1605 path = winreg_printer_data_keyname(tmp_ctx, printer);
1606 if (path == NULL) {
1607 TALLOC_FREE(tmp_ctx);
1608 return WERR_NOMEM;
1611 ZERO_STRUCT(hive_hnd);
1612 ZERO_STRUCT(key_hnd);
1614 result = winreg_printer_openkey(tmp_ctx,
1615 server_info,
1616 &winreg_pipe,
1617 path,
1619 false,
1620 access_mask,
1621 &hive_hnd,
1622 &key_hnd);
1623 if (!W_ERROR_IS_OK(result)) {
1624 DEBUG(0, ("winreg_printer_update_changeid: Could not open key %s: %s\n",
1625 path, win_errstr(result)));
1626 goto done;
1629 result = winreg_printer_write_dword(tmp_ctx,
1630 winreg_pipe,
1631 &key_hnd,
1632 "ChangeID",
1633 winreg_printer_rev_changeid());
1634 if (!W_ERROR_IS_OK(result)) {
1635 goto done;
1638 result = WERR_OK;
1639 done:
1640 if (winreg_pipe != NULL) {
1641 if (is_valid_policy_hnd(&key_hnd)) {
1642 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1644 if (is_valid_policy_hnd(&hive_hnd)) {
1645 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1649 TALLOC_FREE(tmp_ctx);
1650 return result;
1653 WERROR winreg_printer_get_changeid(TALLOC_CTX *mem_ctx,
1654 struct auth_serversupplied_info *server_info,
1655 const char *printer,
1656 uint32_t *pchangeid)
1658 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1659 struct rpc_pipe_client *winreg_pipe = NULL;
1660 struct policy_handle hive_hnd, key_hnd;
1661 uint32_t changeid = 0;
1662 char *path;
1663 WERROR result;
1664 TALLOC_CTX *tmp_ctx;
1666 tmp_ctx = talloc_new(mem_ctx);
1667 if (tmp_ctx == NULL) {
1668 return WERR_NOMEM;
1671 path = winreg_printer_data_keyname(tmp_ctx, printer);
1672 if (path == NULL) {
1673 TALLOC_FREE(tmp_ctx);
1674 return WERR_NOMEM;
1677 ZERO_STRUCT(hive_hnd);
1678 ZERO_STRUCT(key_hnd);
1680 result = winreg_printer_openkey(tmp_ctx,
1681 server_info,
1682 &winreg_pipe,
1683 path,
1685 false,
1686 access_mask,
1687 &hive_hnd,
1688 &key_hnd);
1689 if (!W_ERROR_IS_OK(result)) {
1690 DEBUG(0, ("winreg_printer_get_changeid: Could not open key %s: %s\n",
1691 path, win_errstr(result)));
1692 goto done;
1695 DEBUG(0, ("winreg_printer_get_changeid: get changeid from %s\n", path));
1696 result = winreg_printer_query_dword(tmp_ctx,
1697 winreg_pipe,
1698 &key_hnd,
1699 "ChangeID",
1700 &changeid);
1701 if (!W_ERROR_IS_OK(result)) {
1702 goto done;
1705 if (pchangeid) {
1706 *pchangeid = changeid;
1709 result = WERR_OK;
1710 done:
1711 if (winreg_pipe != NULL) {
1712 if (is_valid_policy_hnd(&key_hnd)) {
1713 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1715 if (is_valid_policy_hnd(&hive_hnd)) {
1716 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1720 TALLOC_FREE(tmp_ctx);
1721 return result;
1725 * The special behaviour of the spoolss forms is documented at the website:
1727 * Managing Win32 Printserver Forms
1728 * http://unixwiz.net/techtips/winspooler-forms.html
1731 WERROR winreg_printer_addform1(TALLOC_CTX *mem_ctx,
1732 struct auth_serversupplied_info *server_info,
1733 struct spoolss_AddFormInfo1 *form)
1735 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1736 struct rpc_pipe_client *winreg_pipe = NULL;
1737 struct policy_handle hive_hnd, key_hnd;
1738 struct winreg_String wvalue;
1739 DATA_BLOB blob;
1740 uint32_t num_info = 0;
1741 union spoolss_FormInfo *info = NULL;
1742 uint32_t i;
1743 WERROR result;
1744 NTSTATUS status;
1745 TALLOC_CTX *tmp_ctx;
1747 tmp_ctx = talloc_new(mem_ctx);
1748 if (tmp_ctx == NULL) {
1749 return WERR_NOMEM;
1752 ZERO_STRUCT(hive_hnd);
1753 ZERO_STRUCT(key_hnd);
1755 result = winreg_printer_openkey(tmp_ctx,
1756 server_info,
1757 &winreg_pipe,
1758 TOP_LEVEL_CONTROL_FORMS_KEY,
1760 true,
1761 access_mask,
1762 &hive_hnd,
1763 &key_hnd);
1764 if (!W_ERROR_IS_OK(result)) {
1765 DEBUG(0, ("winreg_printer_addform1: Could not open key %s: %s\n",
1766 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
1767 goto done;
1770 result = winreg_printer_enumforms1(tmp_ctx, server_info, &num_info, &info);
1771 if (!W_ERROR_IS_OK(result)) {
1772 DEBUG(0, ("winreg_printer_addform: Could not enum keys %s: %s\n",
1773 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
1774 goto done;
1777 /* If form name already exists or is builtin return ALREADY_EXISTS */
1778 for (i = 0; i < num_info; i++) {
1779 if (strequal(info[i].info1.form_name, form->form_name)) {
1780 result = WERR_FILE_EXISTS;
1781 goto done;
1785 wvalue.name = form->form_name;
1787 blob = data_blob_talloc(tmp_ctx, NULL, 32);
1788 SIVAL(blob.data, 0, form->size.width);
1789 SIVAL(blob.data, 4, form->size.height);
1790 SIVAL(blob.data, 8, form->area.left);
1791 SIVAL(blob.data, 12, form->area.top);
1792 SIVAL(blob.data, 16, form->area.right);
1793 SIVAL(blob.data, 20, form->area.bottom);
1794 SIVAL(blob.data, 24, num_info + 1); /* FIXME */
1795 SIVAL(blob.data, 28, form->flags);
1797 status = rpccli_winreg_SetValue(winreg_pipe,
1798 tmp_ctx,
1799 &key_hnd,
1800 wvalue,
1801 REG_BINARY,
1802 blob.data,
1803 blob.length,
1804 &result);
1805 if (!NT_STATUS_IS_OK(status)) {
1806 DEBUG(0, ("winreg_printer_addform1: Could not set value %s: %s\n",
1807 wvalue.name, nt_errstr(status)));
1808 if (!W_ERROR_IS_OK(result)) {
1809 goto done;
1811 result = ntstatus_to_werror(status);
1812 goto done;
1815 result = WERR_OK;
1816 done:
1817 if (winreg_pipe != NULL) {
1818 if (is_valid_policy_hnd(&key_hnd)) {
1819 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1821 if (is_valid_policy_hnd(&hive_hnd)) {
1822 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1826 TALLOC_FREE(info);
1827 TALLOC_FREE(tmp_ctx);
1828 return result;
1831 WERROR winreg_printer_enumforms1(TALLOC_CTX *mem_ctx,
1832 struct auth_serversupplied_info *server_info,
1833 uint32_t *pnum_info,
1834 union spoolss_FormInfo **pinfo)
1836 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1837 struct rpc_pipe_client *winreg_pipe = NULL;
1838 struct policy_handle hive_hnd, key_hnd;
1839 union spoolss_FormInfo *info;
1840 struct spoolss_PrinterEnumValues *enum_values = NULL;
1841 uint32_t num_values = 0;
1842 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
1843 uint32_t i;
1844 WERROR result;
1845 TALLOC_CTX *tmp_ctx;
1847 tmp_ctx = talloc_new(mem_ctx);
1848 if (tmp_ctx == NULL) {
1849 return WERR_NOMEM;
1852 ZERO_STRUCT(hive_hnd);
1853 ZERO_STRUCT(key_hnd);
1855 result = winreg_printer_openkey(tmp_ctx,
1856 server_info,
1857 &winreg_pipe,
1858 TOP_LEVEL_CONTROL_FORMS_KEY,
1860 true,
1861 access_mask,
1862 &hive_hnd,
1863 &key_hnd);
1864 if (!W_ERROR_IS_OK(result)) {
1865 /* key doesn't exist */
1866 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
1867 result = WERR_OK;
1868 goto done;
1871 DEBUG(0, ("winreg_printer_enumforms1: Could not open key %s: %s\n",
1872 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
1873 goto done;
1876 result = winreg_printer_enumvalues(tmp_ctx,
1877 winreg_pipe,
1878 &key_hnd,
1879 &num_values,
1880 &enum_values);
1881 if (!W_ERROR_IS_OK(result)) {
1882 DEBUG(0, ("winreg_printer_enumforms1: Could not enumerate values in %s: %s\n",
1883 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
1884 goto done;
1887 info = TALLOC_ARRAY(tmp_ctx, union spoolss_FormInfo, num_builtin + num_values);
1888 if (info == NULL) {
1889 result = WERR_NOMEM;
1890 goto done;
1893 /* Enumerate BUILTIN forms */
1894 for (i = 0; i < num_builtin; i++) {
1895 info[i].info1 = builtin_forms1[i];
1898 /* Enumerate registry forms */
1899 for (i = 0; i < num_values; i++) {
1900 union spoolss_FormInfo val;
1902 if (enum_values[i].type != REG_BINARY ||
1903 enum_values[i].data_length != 32) {
1904 continue;
1907 val.info1.form_name = talloc_strdup(info, enum_values[i].value_name);
1908 if (val.info1.form_name == NULL) {
1909 result = WERR_NOMEM;
1910 goto done;
1913 val.info1.size.width = IVAL(enum_values[i].data->data, 0);
1914 val.info1.size.height = IVAL(enum_values[i].data->data, 4);
1915 val.info1.area.left = IVAL(enum_values[i].data->data, 8);
1916 val.info1.area.top = IVAL(enum_values[i].data->data, 12);
1917 val.info1.area.right = IVAL(enum_values[i].data->data, 16);
1918 val.info1.area.bottom = IVAL(enum_values[i].data->data, 20);
1919 /* skip form index IVAL(enum_values[i].data->data, 24)));*/
1920 val.info1.flags = IVAL(enum_values[i].data->data, 28);
1922 info[i + num_builtin] = val;
1925 *pnum_info = num_builtin + num_values;
1926 if (pinfo) {
1927 *pinfo = talloc_move(mem_ctx, &info);
1930 done:
1931 if (winreg_pipe != NULL) {
1932 if (is_valid_policy_hnd(&key_hnd)) {
1933 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
1935 if (is_valid_policy_hnd(&hive_hnd)) {
1936 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
1940 TALLOC_FREE(enum_values);
1941 TALLOC_FREE(tmp_ctx);
1942 return result;
1945 WERROR winreg_printer_deleteform1(TALLOC_CTX *mem_ctx,
1946 struct auth_serversupplied_info *server_info,
1947 const char *form_name)
1949 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
1950 struct rpc_pipe_client *winreg_pipe = NULL;
1951 struct policy_handle hive_hnd, key_hnd;
1952 struct winreg_String wvalue;
1953 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
1954 uint32_t i;
1955 WERROR result = WERR_OK;
1956 NTSTATUS status;
1957 TALLOC_CTX *tmp_ctx;
1959 for (i = 0; i < num_builtin; i++) {
1960 if (strequal(builtin_forms1[i].form_name, form_name)) {
1961 return WERR_INVALID_PARAMETER;
1965 tmp_ctx = talloc_new(mem_ctx);
1966 if (tmp_ctx == NULL) {
1967 return WERR_NOMEM;
1970 ZERO_STRUCT(hive_hnd);
1971 ZERO_STRUCT(key_hnd);
1973 result = winreg_printer_openkey(tmp_ctx,
1974 server_info,
1975 &winreg_pipe,
1976 TOP_LEVEL_CONTROL_FORMS_KEY,
1978 false,
1979 access_mask,
1980 &hive_hnd,
1981 &key_hnd);
1982 if (!W_ERROR_IS_OK(result)) {
1983 DEBUG(0, ("winreg_printer_deleteform1: Could not open key %s: %s\n",
1984 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
1985 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
1986 result = WERR_INVALID_FORM_NAME;
1988 goto done;
1991 wvalue.name = form_name;
1992 status = rpccli_winreg_DeleteValue(winreg_pipe,
1993 tmp_ctx,
1994 &key_hnd,
1995 wvalue,
1996 &result);
1997 if (!NT_STATUS_IS_OK(status)) {
1998 /* If the value doesn't exist, return WERR_INVALID_FORM_NAME */
1999 if (W_ERROR_EQUAL(result, WERR_BADFILE)) {
2000 result = WERR_INVALID_FORM_NAME;
2001 goto done;
2003 DEBUG(0, ("winreg_printer_delteform1: Could not delete value %s: %s\n",
2004 wvalue.name, nt_errstr(status)));
2005 if (!W_ERROR_IS_OK(result)) {
2006 goto done;
2008 result = ntstatus_to_werror(status);
2009 goto done;
2012 result = WERR_OK;
2013 done:
2014 if (winreg_pipe != NULL) {
2015 if (is_valid_policy_hnd(&key_hnd)) {
2016 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
2018 if (is_valid_policy_hnd(&hive_hnd)) {
2019 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
2023 TALLOC_FREE(tmp_ctx);
2024 return result;
2027 WERROR winreg_printer_setform1(TALLOC_CTX *mem_ctx,
2028 struct auth_serversupplied_info *server_info,
2029 const char *form_name,
2030 struct spoolss_AddFormInfo1 *form)
2032 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2033 struct rpc_pipe_client *winreg_pipe = NULL;
2034 struct policy_handle hive_hnd, key_hnd;
2035 struct winreg_String wvalue;
2036 DATA_BLOB blob;
2037 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
2038 uint32_t i;
2039 WERROR result;
2040 NTSTATUS status;
2041 TALLOC_CTX *tmp_ctx = NULL;
2043 for (i = 0; i < num_builtin; i++) {
2044 if (strequal(builtin_forms1[i].form_name, form->form_name)) {
2045 result = WERR_INVALID_PARAM;
2046 goto done;
2050 tmp_ctx = talloc_new(mem_ctx);
2051 if (tmp_ctx == NULL) {
2052 return WERR_NOMEM;
2055 ZERO_STRUCT(hive_hnd);
2056 ZERO_STRUCT(key_hnd);
2058 result = winreg_printer_openkey(tmp_ctx,
2059 server_info,
2060 &winreg_pipe,
2061 TOP_LEVEL_CONTROL_FORMS_KEY,
2063 true,
2064 access_mask,
2065 &hive_hnd,
2066 &key_hnd);
2067 if (!W_ERROR_IS_OK(result)) {
2068 DEBUG(0, ("winreg_printer_setform1: Could not open key %s: %s\n",
2069 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2070 goto done;
2073 /* If form_name != form->form_name then we renamed the form */
2074 if (strequal(form_name, form->form_name)) {
2075 result = winreg_printer_deleteform1(tmp_ctx, server_info, form_name);
2076 if (!W_ERROR_IS_OK(result)) {
2077 DEBUG(0, ("winreg_printer_setform1: Could not open key %s: %s\n",
2078 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2079 goto done;
2083 wvalue.name = form->form_name;
2085 blob = data_blob_talloc(tmp_ctx, NULL, 32);
2086 SIVAL(blob.data, 0, form->size.width);
2087 SIVAL(blob.data, 4, form->size.height);
2088 SIVAL(blob.data, 8, form->area.left);
2089 SIVAL(blob.data, 12, form->area.top);
2090 SIVAL(blob.data, 16, form->area.right);
2091 SIVAL(blob.data, 20, form->area.bottom);
2092 SIVAL(blob.data, 24, 42);
2093 SIVAL(blob.data, 28, form->flags);
2095 status = rpccli_winreg_SetValue(winreg_pipe,
2096 tmp_ctx,
2097 &key_hnd,
2098 wvalue,
2099 REG_BINARY,
2100 blob.data,
2101 blob.length,
2102 &result);
2103 if (!NT_STATUS_IS_OK(status)) {
2104 DEBUG(0, ("winreg_printer_setform1: Could not set value %s: %s\n",
2105 wvalue.name, nt_errstr(status)));
2106 if (!W_ERROR_IS_OK(result)) {
2107 goto done;
2109 result = ntstatus_to_werror(status);
2110 goto done;
2113 result = WERR_OK;
2114 done:
2115 if (winreg_pipe != NULL) {
2116 if (is_valid_policy_hnd(&key_hnd)) {
2117 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
2119 if (is_valid_policy_hnd(&hive_hnd)) {
2120 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
2124 TALLOC_FREE(tmp_ctx);
2125 return result;
2128 WERROR winreg_printer_getform1(TALLOC_CTX *mem_ctx,
2129 struct auth_serversupplied_info *server_info,
2130 const char *form_name,
2131 struct spoolss_FormInfo1 *r)
2133 uint32_t access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
2134 struct rpc_pipe_client *winreg_pipe = NULL;
2135 struct policy_handle hive_hnd, key_hnd;
2136 struct winreg_String wvalue;
2137 enum winreg_Type type_in;
2138 uint8_t *data_in;
2139 uint32_t data_in_size = 0;
2140 uint32_t value_len = 0;
2141 uint32_t num_builtin = ARRAY_SIZE(builtin_forms1);
2142 uint32_t i;
2143 WERROR result;
2144 NTSTATUS status;
2145 TALLOC_CTX *tmp_ctx;
2147 /* check builtin forms first */
2148 for (i = 0; i < num_builtin; i++) {
2149 if (strequal(builtin_forms1[i].form_name, form_name)) {
2150 *r = builtin_forms1[i];
2151 return WERR_OK;
2155 tmp_ctx = talloc_new(mem_ctx);
2156 if (tmp_ctx == NULL) {
2157 return WERR_NOMEM;
2160 ZERO_STRUCT(hive_hnd);
2161 ZERO_STRUCT(key_hnd);
2163 result = winreg_printer_openkey(tmp_ctx,
2164 server_info,
2165 &winreg_pipe,
2166 TOP_LEVEL_CONTROL_FORMS_KEY,
2168 true,
2169 access_mask,
2170 &hive_hnd,
2171 &key_hnd);
2172 if (!W_ERROR_IS_OK(result)) {
2173 DEBUG(0, ("winreg_printer_getform1: Could not open key %s: %s\n",
2174 TOP_LEVEL_CONTROL_FORMS_KEY, win_errstr(result)));
2175 goto done;
2178 wvalue.name = form_name;
2181 * call QueryValue once with data == NULL to get the
2182 * needed memory size to be allocated, then allocate
2183 * data buffer and call again.
2185 status = rpccli_winreg_QueryValue(winreg_pipe,
2186 tmp_ctx,
2187 &key_hnd,
2188 &wvalue,
2189 &type_in,
2190 NULL,
2191 &data_in_size,
2192 &value_len,
2193 &result);
2194 if (!NT_STATUS_IS_OK(status)) {
2195 DEBUG(0, ("winreg_printer_getform1: Could not query value %s: %s\n",
2196 wvalue.name, nt_errstr(status)));
2197 if (!W_ERROR_IS_OK(result)) {
2198 goto done;
2200 result = ntstatus_to_werror(status);
2201 goto done;
2204 data_in = (uint8_t *) TALLOC(tmp_ctx, data_in_size);
2205 if (data_in == NULL) {
2206 result = WERR_NOMEM;
2207 goto done;
2209 value_len = 0;
2211 status = rpccli_winreg_QueryValue(winreg_pipe,
2212 tmp_ctx,
2213 &key_hnd,
2214 &wvalue,
2215 &type_in,
2216 data_in,
2217 &data_in_size,
2218 &value_len,
2219 &result);
2220 if (!NT_STATUS_IS_OK(status)) {
2221 DEBUG(0, ("winreg_printer_getform1: Could not query value %s: %s\n",
2222 wvalue.name, nt_errstr(status)));
2223 if (!W_ERROR_IS_OK(result)) {
2224 goto done;
2226 result = ntstatus_to_werror(status);
2227 goto done;
2230 r->form_name = talloc_strdup(mem_ctx, form_name);
2231 if (r->form_name == NULL) {
2232 result = WERR_NOMEM;
2233 goto done;
2236 r->size.width = IVAL(data_in, 0);
2237 r->size.height = IVAL(data_in, 4);
2238 r->area.left = IVAL(data_in, 8);
2239 r->area.top = IVAL(data_in, 12);
2240 r->area.right = IVAL(data_in, 16);
2241 r->area.bottom = IVAL(data_in, 20);
2242 /* skip index IVAL(data_in, 24)));*/
2243 r->flags = IVAL(data_in, 28);
2245 result = WERR_OK;
2246 done:
2247 if (winreg_pipe != NULL) {
2248 if (is_valid_policy_hnd(&key_hnd)) {
2249 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &key_hnd, NULL);
2251 if (is_valid_policy_hnd(&hive_hnd)) {
2252 rpccli_winreg_CloseKey(winreg_pipe, tmp_ctx, &hive_hnd, NULL);
2256 TALLOC_FREE(tmp_ctx);
2257 return result;