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 IClientSecurity client_security DECLSPEC_HIDDEN; 29 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 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 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