wbemprox: Support overriding the CIM to VARIANT type mapping for integer properties.
[wine] / dlls / wbemprox / wbemprox_private.h
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 #include "wine/list.h"
20 #include "wine/unicode.h"
21
22 IClientSecurity client_security;
23
24 #define SIZEOF(array) (sizeof(array)/sizeof((array)[0]))
25
26 #define COL_TYPE_MASK    0x0000ffff
27 #define COL_FLAG_DYNAMIC 0x00010000
28 #define COL_FLAG_KEY     0x00020000
29
30 struct column
31 {
32     const WCHAR *name;
33     UINT type;
34     VARTYPE vartype; /* 0 for default mapping */
35 };
36
37 struct table
38 {
39     const WCHAR *name;
40     UINT num_cols;
41     const struct column *columns;
42     UINT num_rows;
43     BYTE *data;
44     void (*fill)(struct table *);
45 };
46
47 struct property
48 {
49     const WCHAR *name;
50     const WCHAR *class;
51     const struct property *next;
52 };
53
54 enum operator
55 {
56     OP_EQ      = 1,
57     OP_AND     = 2,
58     OP_OR      = 3,
59     OP_GT      = 4,
60     OP_LT      = 5,
61     OP_LE      = 6,
62     OP_GE      = 7,
63     OP_NE      = 8,
64     OP_ISNULL  = 9,
65     OP_NOTNULL = 10,
66     OP_LIKE    = 11
67 };
68
69 struct expr;
70 struct complex_expr
71 {
72     enum operator op;
73     struct expr *left;
74     struct expr *right;
75 };
76
77 enum expr_type
78 {
79     EXPR_COMPLEX = 1,
80     EXPR_UNARY   = 2,
81     EXPR_PROPVAL = 3,
82     EXPR_SVAL    = 4,
83     EXPR_IVAL    = 5,
84     EXPR_BVAL    = 6
85 };
86
87 struct expr
88 {
89     enum expr_type type;
90     union
91     {
92         struct complex_expr expr;
93         const struct property *propval;
94         const WCHAR *sval;
95         int ival;
96     } u;
97 };
98
99 struct view
100 {
101     const struct property *proplist;
102     struct table *table;
103     const struct expr *cond;
104     UINT *result;
105     UINT  count;
106 };
107
108 struct query
109 {
110     LONG refs;
111     struct view *view;
112     struct list mem;
113 };
114
115 void addref_query( struct query * ) DECLSPEC_HIDDEN;
116 void release_query( struct query *query ) DECLSPEC_HIDDEN;
117 HRESULT exec_query( const WCHAR *, IEnumWbemClassObject ** ) DECLSPEC_HIDDEN;
118 HRESULT parse_query( const WCHAR *, struct view **, struct list * ) DECLSPEC_HIDDEN;
119 HRESULT create_view( const struct property *, const WCHAR *, const struct expr *,
120                      struct view ** ) DECLSPEC_HIDDEN;
121 void destroy_view( struct view * ) DECLSPEC_HIDDEN;
122 struct table *get_table( const WCHAR * ) DECLSPEC_HIDDEN;
123 HRESULT get_propval( const struct view *, UINT, const WCHAR *, VARIANT *,
124                      CIMTYPE *, LONG * ) DECLSPEC_HIDDEN;
125 HRESULT get_properties( const struct view *, SAFEARRAY ** ) DECLSPEC_HIDDEN;
126
127 HRESULT WbemLocator_create(IUnknown *, LPVOID *) DECLSPEC_HIDDEN;
128 HRESULT WbemServices_create(IUnknown *, const WCHAR *, LPVOID *) DECLSPEC_HIDDEN;
129 HRESULT WbemClassObject_create(IUnknown *, IEnumWbemClassObject *, UINT, LPVOID *) DECLSPEC_HIDDEN;
130 HRESULT EnumWbemClassObject_create(IUnknown *, struct query *, LPVOID *) DECLSPEC_HIDDEN;
131
132 static void *heap_alloc( size_t len ) __WINE_ALLOC_SIZE(1);
133 static inline void *heap_alloc( size_t len )
134 {
135     return HeapAlloc( GetProcessHeap(), 0, len );
136 }
137
138 static void *heap_realloc( void *mem, size_t len ) __WINE_ALLOC_SIZE(2);
139 static inline void *heap_realloc( void *mem, size_t len )
140 {
141     return HeapReAlloc( GetProcessHeap(), 0, mem, len );
142 }
143
144 static inline BOOL heap_free( void *mem )
145 {
146     return HeapFree( GetProcessHeap(), 0, mem );
147 }
148
149 static inline WCHAR *heap_strdupW( const WCHAR *src )
150 {
151     WCHAR *dst;
152     if (!src) return NULL;
153     if ((dst = heap_alloc( (strlenW( src ) + 1) * sizeof(WCHAR) ))) strcpyW( dst, src );
154     return dst;
155 }