1 | /*
|
---|
2 | * Unix SMB/CIFS implementation.
|
---|
3 | * Virtual Windows Registry Layer
|
---|
4 | * Copyright (C) Gerald Carter 2002-2005
|
---|
5 | * Copyright (c) Andreas Schneider <asn@samba.org> 2010
|
---|
6 | *
|
---|
7 | * This program is free software; you can redistribute it and/or modify
|
---|
8 | * it under the terms of the GNU General Public License as published by
|
---|
9 | * the Free Software Foundation; either version 3 of the License, or
|
---|
10 | * (at your option) any later version.
|
---|
11 | *
|
---|
12 | * This program is distributed in the hope that it will be useful,
|
---|
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
15 | * GNU General Public License for more details.
|
---|
16 | *
|
---|
17 | * You should have received a copy of the GNU General Public License
|
---|
18 | * along with this program; if not, see <http://www.gnu.org/licenses/>.
|
---|
19 | */
|
---|
20 |
|
---|
21 | /* Implementation of registry virtual views for printing information */
|
---|
22 |
|
---|
23 | #include "includes.h"
|
---|
24 | #include "registry.h"
|
---|
25 | #include "reg_util_internal.h"
|
---|
26 | #include "reg_backend_db.h"
|
---|
27 |
|
---|
28 | #undef DBGC_CLASS
|
---|
29 | #define DBGC_CLASS DBGC_REGISTRY
|
---|
30 |
|
---|
31 | /* registry paths used in the print_registry[] */
|
---|
32 | #define KEY_CONTROL_PRINTERS "HKLM\\SYSTEM\\CURRENTCONTROLSET\\CONTROL\\PRINT\\PRINTERS"
|
---|
33 | #define KEY_WINNT_PRINTERS "HKLM\\SOFTWARE\\MICROSOFT\\WINDOWS NT\\CURRENTVERSION\\PRINT\\PRINTERS"
|
---|
34 |
|
---|
35 | /* callback table for various registry paths below the ones we service in this module */
|
---|
36 |
|
---|
37 | struct reg_dyn_tree {
|
---|
38 | /* full key path in normalized form */
|
---|
39 | const char *path;
|
---|
40 |
|
---|
41 | /* callbscks for fetch/store operations */
|
---|
42 | int ( *fetch_subkeys) ( const char *path, struct regsubkey_ctr *subkeys );
|
---|
43 | bool (*store_subkeys) ( const char *path, struct regsubkey_ctr *subkeys );
|
---|
44 | int (*fetch_values) ( const char *path, struct regval_ctr *values );
|
---|
45 | bool (*store_values) ( const char *path, struct regval_ctr *values );
|
---|
46 | };
|
---|
47 |
|
---|
48 | /*********************************************************************
|
---|
49 | *********************************************************************
|
---|
50 | ** "HKLM/SYSTEM/CURRENTCONTROLSET/CONTROL/PRINT/PRINTERS"
|
---|
51 | ** "HKLM/SOFTWARE/MICROSOFT/WINDOWS NT/CURRENTVERSION/PRINT/PRINTERS"
|
---|
52 | *********************************************************************
|
---|
53 | *********************************************************************/
|
---|
54 |
|
---|
55 | static char *create_printer_registry_path(TALLOC_CTX *mem_ctx, const char *key) {
|
---|
56 | char *path;
|
---|
57 | char *subkey = NULL;
|
---|
58 |
|
---|
59 | path = talloc_strdup(mem_ctx, key);
|
---|
60 | if (path == NULL) {
|
---|
61 | return NULL;
|
---|
62 | }
|
---|
63 |
|
---|
64 | path = normalize_reg_path(mem_ctx, path);
|
---|
65 | if (path == NULL) {
|
---|
66 | return NULL;
|
---|
67 | }
|
---|
68 |
|
---|
69 | if (strncmp(path, KEY_CONTROL_PRINTERS, strlen(KEY_CONTROL_PRINTERS)) == 0) {
|
---|
70 | subkey = reg_remaining_path(mem_ctx, key + strlen(KEY_CONTROL_PRINTERS));
|
---|
71 | if (subkey == NULL) {
|
---|
72 | return NULL;
|
---|
73 | }
|
---|
74 | return talloc_asprintf(mem_ctx, "%s\\%s", KEY_WINNT_PRINTERS, subkey);
|
---|
75 | }
|
---|
76 |
|
---|
77 | return NULL;
|
---|
78 | }
|
---|
79 |
|
---|
80 | /*********************************************************************
|
---|
81 | *********************************************************************/
|
---|
82 |
|
---|
83 | static int key_printers_fetch_keys( const char *key, struct regsubkey_ctr *subkeys )
|
---|
84 | {
|
---|
85 | TALLOC_CTX *ctx = talloc_tos();
|
---|
86 | char *printers_key;
|
---|
87 |
|
---|
88 | printers_key = create_printer_registry_path(ctx, key);
|
---|
89 | if (printers_key == NULL) {
|
---|
90 | /* normalize on the 'HKLM\SOFTWARE\....\Print\Printers' key */
|
---|
91 | return regdb_fetch_keys(KEY_WINNT_PRINTERS, subkeys);
|
---|
92 | }
|
---|
93 |
|
---|
94 | return regdb_fetch_keys(printers_key, subkeys);
|
---|
95 | }
|
---|
96 |
|
---|
97 | /**********************************************************************
|
---|
98 | *********************************************************************/
|
---|
99 |
|
---|
100 | static bool key_printers_store_keys( const char *key, struct regsubkey_ctr *subkeys )
|
---|
101 | {
|
---|
102 | TALLOC_CTX *ctx = talloc_tos();
|
---|
103 | char *printers_key;
|
---|
104 |
|
---|
105 | printers_key = create_printer_registry_path(ctx, key);
|
---|
106 | if (printers_key == NULL) {
|
---|
107 | /* normalize on the 'HKLM\SOFTWARE\....\Print\Printers' key */
|
---|
108 | return regdb_store_keys(KEY_WINNT_PRINTERS, subkeys);
|
---|
109 | }
|
---|
110 |
|
---|
111 | return regdb_store_keys(printers_key, subkeys);
|
---|
112 | }
|
---|
113 |
|
---|
114 | /**********************************************************************
|
---|
115 | *********************************************************************/
|
---|
116 |
|
---|
117 | static int key_printers_fetch_values(const char *key, struct regval_ctr *values)
|
---|
118 | {
|
---|
119 | TALLOC_CTX *ctx = talloc_tos();
|
---|
120 | char *printers_key;
|
---|
121 |
|
---|
122 | printers_key = create_printer_registry_path(ctx, key);
|
---|
123 | if (printers_key == NULL) {
|
---|
124 | /* normalize on the 'HKLM\SOFTWARE\....\Print\Printers' key */
|
---|
125 | return regdb_fetch_values(KEY_WINNT_PRINTERS, values);
|
---|
126 | }
|
---|
127 |
|
---|
128 | return regdb_fetch_values(printers_key, values);
|
---|
129 | }
|
---|
130 |
|
---|
131 | /**********************************************************************
|
---|
132 | *********************************************************************/
|
---|
133 |
|
---|
134 | static bool key_printers_store_values(const char *key, struct regval_ctr *values)
|
---|
135 | {
|
---|
136 | TALLOC_CTX *ctx = talloc_tos();
|
---|
137 | char *printers_key;
|
---|
138 |
|
---|
139 | printers_key = create_printer_registry_path(ctx, key);
|
---|
140 | if (printers_key == NULL) {
|
---|
141 | /* normalize on the 'HKLM\SOFTWARE\....\Print\Printers' key */
|
---|
142 | return regdb_store_values(KEY_WINNT_PRINTERS, values);
|
---|
143 | }
|
---|
144 |
|
---|
145 | return regdb_store_values(printers_key, values);
|
---|
146 | }
|
---|
147 |
|
---|
148 | /**********************************************************************
|
---|
149 | *********************************************************************
|
---|
150 | ** Structure to hold dispatch table of ops for various printer keys.
|
---|
151 | ** Make sure to always store deeper keys along the same path first so
|
---|
152 | ** we ge a more specific match.
|
---|
153 | *********************************************************************
|
---|
154 | *********************************************************************/
|
---|
155 |
|
---|
156 | static struct reg_dyn_tree print_registry[] = {
|
---|
157 | { KEY_CONTROL_PRINTERS,
|
---|
158 | &key_printers_fetch_keys,
|
---|
159 | &key_printers_store_keys,
|
---|
160 | &key_printers_fetch_values,
|
---|
161 | &key_printers_store_values },
|
---|
162 |
|
---|
163 | { NULL, NULL, NULL, NULL, NULL }
|
---|
164 | };
|
---|
165 |
|
---|
166 |
|
---|
167 | /**********************************************************************
|
---|
168 | *********************************************************************
|
---|
169 | ** Main reg_printing interface functions
|
---|
170 | *********************************************************************
|
---|
171 | *********************************************************************/
|
---|
172 |
|
---|
173 | /***********************************************************************
|
---|
174 | Lookup a key in the print_registry table, returning its index.
|
---|
175 | -1 on failure
|
---|
176 | **********************************************************************/
|
---|
177 |
|
---|
178 | static int match_registry_path(const char *key)
|
---|
179 | {
|
---|
180 | int i;
|
---|
181 | char *path = NULL;
|
---|
182 | TALLOC_CTX *ctx = talloc_tos();
|
---|
183 |
|
---|
184 | if ( !key )
|
---|
185 | return -1;
|
---|
186 |
|
---|
187 | path = talloc_strdup(ctx, key);
|
---|
188 | if (!path) {
|
---|
189 | return -1;
|
---|
190 | }
|
---|
191 | path = normalize_reg_path(ctx, path);
|
---|
192 | if (!path) {
|
---|
193 | return -1;
|
---|
194 | }
|
---|
195 |
|
---|
196 | for ( i=0; print_registry[i].path; i++ ) {
|
---|
197 | if (strncmp( path, print_registry[i].path, strlen(print_registry[i].path) ) == 0 )
|
---|
198 | return i;
|
---|
199 | }
|
---|
200 |
|
---|
201 | return -1;
|
---|
202 | }
|
---|
203 |
|
---|
204 | /***********************************************************************
|
---|
205 | **********************************************************************/
|
---|
206 |
|
---|
207 | static int regprint_fetch_reg_keys( const char *key, struct regsubkey_ctr *subkeys )
|
---|
208 | {
|
---|
209 | int i = match_registry_path( key );
|
---|
210 |
|
---|
211 | if ( i == -1 )
|
---|
212 | return -1;
|
---|
213 |
|
---|
214 | if ( !print_registry[i].fetch_subkeys )
|
---|
215 | return -1;
|
---|
216 |
|
---|
217 | return print_registry[i].fetch_subkeys( key, subkeys );
|
---|
218 | }
|
---|
219 |
|
---|
220 | /**********************************************************************
|
---|
221 | *********************************************************************/
|
---|
222 |
|
---|
223 | static bool regprint_store_reg_keys( const char *key, struct regsubkey_ctr *subkeys )
|
---|
224 | {
|
---|
225 | int i = match_registry_path( key );
|
---|
226 |
|
---|
227 | if ( i == -1 )
|
---|
228 | return False;
|
---|
229 |
|
---|
230 | if ( !print_registry[i].store_subkeys )
|
---|
231 | return False;
|
---|
232 |
|
---|
233 | return print_registry[i].store_subkeys( key, subkeys );
|
---|
234 | }
|
---|
235 |
|
---|
236 | /**********************************************************************
|
---|
237 | *********************************************************************/
|
---|
238 |
|
---|
239 | static int regprint_fetch_reg_values(const char *key, struct regval_ctr *values)
|
---|
240 | {
|
---|
241 | int i = match_registry_path( key );
|
---|
242 |
|
---|
243 | if ( i == -1 )
|
---|
244 | return -1;
|
---|
245 |
|
---|
246 | /* return 0 values by default since we know the key had
|
---|
247 | to exist because the client opened a handle */
|
---|
248 |
|
---|
249 | if ( !print_registry[i].fetch_values )
|
---|
250 | return 0;
|
---|
251 |
|
---|
252 | return print_registry[i].fetch_values( key, values );
|
---|
253 | }
|
---|
254 |
|
---|
255 | /**********************************************************************
|
---|
256 | *********************************************************************/
|
---|
257 |
|
---|
258 | static bool regprint_store_reg_values(const char *key, struct regval_ctr *values)
|
---|
259 | {
|
---|
260 | int i = match_registry_path( key );
|
---|
261 |
|
---|
262 | if ( i == -1 )
|
---|
263 | return False;
|
---|
264 |
|
---|
265 | if ( !print_registry[i].store_values )
|
---|
266 | return False;
|
---|
267 |
|
---|
268 | return print_registry[i].store_values( key, values );
|
---|
269 | }
|
---|
270 |
|
---|
271 | /*
|
---|
272 | * Table of function pointers for accessing printing data
|
---|
273 | */
|
---|
274 |
|
---|
275 | struct registry_ops printing_ops = {
|
---|
276 | .fetch_subkeys = regprint_fetch_reg_keys,
|
---|
277 | .fetch_values = regprint_fetch_reg_values,
|
---|
278 | .store_subkeys = regprint_store_reg_keys,
|
---|
279 | .store_values = regprint_store_reg_values,
|
---|
280 | };
|
---|