1 /*
2 * Copyright 2009 Hans Leidekker for CodeWeavers
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19 #pragma once
20
21 #include "wine/debug.h"
22 #include "wine/heap.h"
23 #include "wine/list.h"
24 #ifdef __REACTOS__
25 #include <winnls.h>
26 #endif
27
28 extern IClientSecurity client_security DECLSPEC_HIDDEN;
29 extern struct list *table_list DECLSPEC_HIDDEN;
30
31 enum param_direction
32 {
33 PARAM_OUT = -1,
34 PARAM_INOUT = 0,
35 PARAM_IN = 1
36 };
37
38 #define CIM_TYPE_MASK 0x00000fff
39
40 #define COL_TYPE_MASK 0x0000ffff
41 #define COL_FLAG_DYNAMIC 0x00010000
42 #define COL_FLAG_KEY 0x00020000
43 #define COL_FLAG_METHOD 0x00040000
44
45 typedef HRESULT (class_method)(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **);
46
47 enum operator
48 {
49 OP_EQ = 1,
50 OP_AND = 2,
51 OP_OR = 3,
52 OP_GT = 4,
53 OP_LT = 5,
54 OP_LE = 6,
55 OP_GE = 7,
56 OP_NE = 8,
57 OP_ISNULL = 9,
58 OP_NOTNULL = 10,
59 OP_LIKE = 11,
60 OP_NOT = 12
61 };
62
63 struct expr;
64 struct complex_expr
65 {
66 enum operator op;
67 struct expr *left;
68 struct expr *right;
69 };
70
71 enum expr_type
72 {
73 EXPR_COMPLEX = 1,
74 EXPR_UNARY = 2,
75 EXPR_PROPVAL = 3,
76 EXPR_SVAL = 4,
77 EXPR_IVAL = 5,
78 EXPR_BVAL = 6
79 };
80
81 struct expr
82 {
83 enum expr_type type;
84 union
85 {
86 struct complex_expr expr;
87 const struct property *propval;
88 const WCHAR *sval;
89 int ival;
90 } u;
91 };
92
93 struct column
94 {
95 const WCHAR *name;
96 UINT type;
97 };
98
99 enum fill_status
100 {
101 FILL_STATUS_FAILED = -1,
102 FILL_STATUS_UNFILTERED,
103 FILL_STATUS_FILTERED
104 };
105
106 #define TABLE_FLAG_DYNAMIC 0x00000001
107
108 struct table
109 {
110 const WCHAR *name;
111 UINT num_cols;
112 const struct column *columns;
113 UINT num_rows;
114 UINT num_rows_allocated;
115 BYTE *data;
116 enum fill_status (*fill)(struct table *, const struct expr *cond);
117 UINT flags;
118 struct list entry;
119 LONG refs;
120 };
121
122 struct property
123 {
124 const WCHAR *name;
125 const WCHAR *class;
126 const struct property *next;
127 };
128
129 struct array
130 {
131 UINT elem_size;
132 UINT count;
133 void *ptr;
134 };
135
136 struct field
137 {
138 UINT type;
139 union
140 {
141 LONGLONG ival;
142 WCHAR *sval;
143 struct array *aval;
144 } u;
145 };
146
147 struct record
148 {
149 UINT count;
150 struct field *fields;
151 struct table *table;
152 };
153
154 struct keyword
155 {
156 const WCHAR *name;
157 const WCHAR *value;
158 const struct keyword *next;
159 };
160
161 enum view_type
162 {
163 VIEW_TYPE_SELECT,
164 VIEW_TYPE_ASSOCIATORS,
165 };
166
167 struct view
168 {
169 enum view_type type;
170 const WCHAR *path; /* ASSOCIATORS OF query */
171 const struct keyword *keywordlist;
172 const struct property *proplist; /* SELECT query */
173 const struct expr *cond;
174 UINT table_count;
175 struct table **table;
176 UINT result_count;
177 UINT *result;
178 };
179
180 struct query
181 {
182 LONG refs;
183 struct view *view;
184 struct list mem;
185 };
186
187 struct path
188 {
189 WCHAR *class;
190 UINT class_len;
191 WCHAR *filter;
192 UINT filter_len;
193 };
194
195 HRESULT parse_path( const WCHAR *, struct path ** ) DECLSPEC_HIDDEN;
196 void free_path( struct path * ) DECLSPEC_HIDDEN;
197 WCHAR *query_from_path( const struct path * ) DECLSPEC_HIDDEN;
198
199 struct query *create_query(void) DECLSPEC_HIDDEN;
200 void free_query( struct query * ) DECLSPEC_HIDDEN;
201 struct query *addref_query( struct query * ) DECLSPEC_HIDDEN;
202 void release_query( struct query *query ) DECLSPEC_HIDDEN;
203 HRESULT exec_query( const WCHAR *, IEnumWbemClassObject ** ) DECLSPEC_HIDDEN;
204 HRESULT parse_query( const WCHAR *, struct view **, struct list * ) DECLSPEC_HIDDEN;
205 HRESULT create_view( enum view_type, const WCHAR *, const struct keyword *, const WCHAR *, const struct property *,
206 const struct expr *, struct view ** ) DECLSPEC_HIDDEN;
207 void destroy_view( struct view * ) DECLSPEC_HIDDEN;
208 HRESULT execute_view( struct view * ) DECLSPEC_HIDDEN;
209 struct table *get_view_table( const struct view *, UINT ) DECLSPEC_HIDDEN;
210 void init_table_list( void ) DECLSPEC_HIDDEN;
211 struct table *grab_table( const WCHAR * ) DECLSPEC_HIDDEN;
212 struct table *addref_table( struct table * ) DECLSPEC_HIDDEN;
213 void release_table( struct table * ) DECLSPEC_HIDDEN;
214 struct table *create_table( const WCHAR *, UINT, const struct column *, UINT, UINT, BYTE *,
215 enum fill_status (*)(struct table *, const struct expr *) ) DECLSPEC_HIDDEN;
216 BOOL add_table( struct table * ) DECLSPEC_HIDDEN;
217 void free_columns( struct column *, UINT ) DECLSPEC_HIDDEN;
218 void free_row_values( const struct table *, UINT ) DECLSPEC_HIDDEN;
219 void clear_table( struct table * ) DECLSPEC_HIDDEN;
220 void free_table( struct table * ) DECLSPEC_HIDDEN;
221 UINT get_type_size( CIMTYPE ) DECLSPEC_HIDDEN;
222 HRESULT eval_cond( const struct table *, UINT, const struct expr *, LONGLONG *, UINT * ) DECLSPEC_HIDDEN;
223 HRESULT get_column_index( const struct table *, const WCHAR *, UINT * ) DECLSPEC_HIDDEN;
224 HRESULT get_value( const struct table *, UINT, UINT, LONGLONG * ) DECLSPEC_HIDDEN;
225 BSTR get_value_bstr( const struct table *, UINT, UINT ) DECLSPEC_HIDDEN;
226 HRESULT set_value( const struct table *, UINT, UINT, LONGLONG, CIMTYPE ) DECLSPEC_HIDDEN;
227 BOOL is_method( const struct table *, UINT ) DECLSPEC_HIDDEN;
228 HRESULT get_method( const struct table *, const WCHAR *, class_method ** ) DECLSPEC_HIDDEN;
229 HRESULT get_propval( const struct view *, UINT, const WCHAR *, VARIANT *, CIMTYPE *, LONG * ) DECLSPEC_HIDDEN;
230 HRESULT put_propval( const struct view *, UINT, const WCHAR *, VARIANT *, CIMTYPE ) DECLSPEC_HIDDEN;
231 HRESULT to_longlong( VARIANT *, LONGLONG *, CIMTYPE * ) DECLSPEC_HIDDEN;
232 SAFEARRAY *to_safearray( const struct array *, CIMTYPE ) DECLSPEC_HIDDEN;
233 VARTYPE to_vartype( CIMTYPE ) DECLSPEC_HIDDEN;
234 void destroy_array( struct array *, CIMTYPE ) DECLSPEC_HIDDEN;
235 BOOL is_result_prop( const struct view *, const WCHAR * ) DECLSPEC_HIDDEN;
236 HRESULT get_properties( const struct view *, UINT, LONG, SAFEARRAY ** ) DECLSPEC_HIDDEN;
237 HRESULT get_object( const WCHAR *, IWbemClassObject ** ) DECLSPEC_HIDDEN;
238 BSTR get_method_name( const WCHAR *, UINT ) DECLSPEC_HIDDEN;
239 void set_variant( VARTYPE, LONGLONG, void *, VARIANT * ) DECLSPEC_HIDDEN;
240 HRESULT create_signature( const WCHAR *, const WCHAR *, enum param_direction,
241 IWbemClassObject ** ) DECLSPEC_HIDDEN;
242
243 HRESULT WbemLocator_create(LPVOID *) DECLSPEC_HIDDEN;
244 HRESULT WbemServices_create(const WCHAR *, LPVOID *) DECLSPEC_HIDDEN;
245 HRESULT create_class_object(const WCHAR *, IEnumWbemClassObject *, UINT,
246 struct record *, IWbemClassObject **) DECLSPEC_HIDDEN;
247 HRESULT EnumWbemClassObject_create(struct query *, LPVOID *) DECLSPEC_HIDDEN;
248 HRESULT WbemQualifierSet_create(const WCHAR *, const WCHAR *, LPVOID *) DECLSPEC_HIDDEN;
249
250 HRESULT process_get_owner(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
251 HRESULT reg_create_key(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
252 HRESULT reg_enum_key(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
253 HRESULT reg_enum_values(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
254 HRESULT reg_get_stringvalue(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
255 HRESULT service_pause_service(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
256 HRESULT service_resume_service(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
257 HRESULT service_start_service(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
258 HRESULT service_stop_service(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
259 HRESULT security_get_sd(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
260 HRESULT security_set_sd(IWbemClassObject *, IWbemClassObject *, IWbemClassObject **) DECLSPEC_HIDDEN;
261
heap_strdupW(const WCHAR * src)262 static inline WCHAR *heap_strdupW( const WCHAR *src )
263 {
264 WCHAR *dst;
265 if (!src) return NULL;
266 if ((dst = heap_alloc( (lstrlenW( src ) + 1) * sizeof(WCHAR) ))) lstrcpyW( dst, src );
267 return dst;
268 }
269
heap_strdupAW(const char * src)270 static inline WCHAR *heap_strdupAW( const char *src )
271 {
272 int len;
273 WCHAR *dst;
274 if (!src) return NULL;
275 len = MultiByteToWideChar( CP_ACP, 0, src, -1, NULL, 0 );
276 if ((dst = heap_alloc( len * sizeof(*dst) ))) MultiByteToWideChar( CP_ACP, 0, src, -1, dst, len );
277 return dst;
278 }
279
280 static const WCHAR class_processW[] = {'W','i','n','3','2','_','P','r','o','c','e','s','s',0};
281 static const WCHAR class_serviceW[] = {'W','i','n','3','2','_','S','e','r','v','i','c','e',0};
282 static const WCHAR class_stdregprovW[] = {'S','t','d','R','e','g','P','r','o','v',0};
283 static const WCHAR class_systemsecurityW[] = {'_','_','S','y','s','t','e','m','S','e','c','u','r','i','t','y',0};
284
285 static const WCHAR prop_nameW[] = {'N','a','m','e',0};
286
287 static const WCHAR method_createkeyW[] = {'C','r','e','a','t','e','K','e','y',0};
288 static const WCHAR method_enumkeyW[] = {'E','n','u','m','K','e','y',0};
289 static const WCHAR method_enumvaluesW[] = {'E','n','u','m','V','a','l','u','e','s',0};
290 static const WCHAR method_getownerW[] = {'G','e','t','O','w','n','e','r',0};
291 static const WCHAR method_getsdW[] = {'G','e','t','S','D',0};
292 static const WCHAR method_getstringvalueW[] = {'G','e','t','S','t','r','i','n','g','V','a','l','u','e',0};
293 static const WCHAR method_pauseserviceW[] = {'P','a','u','s','e','S','e','r','v','i','c','e',0};
294 static const WCHAR method_resumeserviceW[] = {'R','e','s','u','m','e','S','e','r','v','i','c','e',0};
295 static const WCHAR method_setsdW[] = {'S','e','t','S','D',0};
296 static const WCHAR method_startserviceW[] = {'S','t','a','r','t','S','e','r','v','i','c','e',0};
297 static const WCHAR method_stopserviceW[] = {'S','t','o','p','S','e','r','v','i','c','e',0};
298
299 static const WCHAR param_defkeyW[] = {'h','D','e','f','K','e','y',0};
300 static const WCHAR param_domainW[] = {'D','o','m','a','i','n',0};
301 static const WCHAR param_namesW[] = {'s','N','a','m','e','s',0};
302 static const WCHAR param_returnvalueW[] = {'R','e','t','u','r','n','V','a','l','u','e',0};
303 static const WCHAR param_sdW[] = {'S','D',0};
304 static const WCHAR param_subkeynameW[] = {'s','S','u','b','K','e','y','N','a','m','e',0};
305 static const WCHAR param_typesW[] = {'T','y','p','e','s',0};
306 static const WCHAR param_userW[] = {'U','s','e','r',0};
307 static const WCHAR param_valueW[] = {'s','V','a','l','u','e',0};
308 static const WCHAR param_valuenameW[] = {'s','V','a','l','u','e','N','a','m','e',0};
309