2 * Copyright 2012 Hans Leidekker for CodeWeavers
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.
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.
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
28 #include "wine/debug.h"
29 #include "wbemprox_private.h"
31 WINE_DEFAULT_DEBUG_CHANNEL(wbemprox
);
33 HRESULT
get_column_index( const struct table
*table
, const WCHAR
*name
, UINT
*column
)
36 for (i
= 0; i
< table
->num_cols
; i
++)
38 if (!strcmpiW( table
->columns
[i
].name
, name
))
44 return WBEM_E_INVALID_QUERY
;
47 UINT
get_type_size( CIMTYPE type
)
49 if (type
& CIM_FLAG_ARRAY
) return sizeof(void *);
66 return sizeof(WCHAR
*);
68 ERR("unhandled type %u\n", type
);
71 return sizeof(LONGLONG
);
74 static UINT
get_column_size( const struct table
*table
, UINT column
)
76 return get_type_size( table
->columns
[column
].type
& COL_TYPE_MASK
);
79 static UINT
get_column_offset( const struct table
*table
, UINT column
)
82 for (i
= 0; i
< column
; i
++) offset
+= get_column_size( table
, i
);
86 static UINT
get_row_size( const struct table
*table
)
88 return get_column_offset( table
, table
->num_cols
- 1 ) + get_column_size( table
, table
->num_cols
- 1 );
91 HRESULT
get_value( const struct table
*table
, UINT row
, UINT column
, LONGLONG
*val
)
93 UINT col_offset
, row_size
;
96 col_offset
= get_column_offset( table
, column
);
97 row_size
= get_row_size( table
);
98 ptr
= table
->data
+ row
* row_size
+ col_offset
;
100 if (table
->columns
[column
].type
& CIM_FLAG_ARRAY
)
102 *val
= (LONGLONG
)(INT_PTR
)*(const void **)ptr
;
105 switch (table
->columns
[column
].type
& COL_TYPE_MASK
)
108 *val
= *(const int *)ptr
;
112 *val
= (LONGLONG
)(INT_PTR
)*(const WCHAR
**)ptr
;
115 *val
= *(const INT16
*)ptr
;
118 *val
= *(const UINT16
*)ptr
;
121 *val
= *(const INT32
*)ptr
;
124 *val
= *(const UINT32
*)ptr
;
127 *val
= *(const INT64
*)ptr
;
130 *val
= *(const UINT64
*)ptr
;
133 ERR("invalid column type %u\n", table
->columns
[column
].type
& COL_TYPE_MASK
);
140 BSTR
get_value_bstr( const struct table
*table
, UINT row
, UINT column
)
142 static const WCHAR fmt_signedW
[] = {'%','d',0};
143 static const WCHAR fmt_unsignedW
[] = {'%','u',0};
144 static const WCHAR fmt_signed64W
[] = {'%','I','6','4','d',0};
145 static const WCHAR fmt_unsigned64W
[] = {'%','I','6','4','u',0};
146 static const WCHAR fmt_strW
[] = {'\"','%','s','\"',0};
152 if (table
->columns
[column
].type
& CIM_FLAG_ARRAY
)
154 FIXME("array to string conversion not handled\n");
157 if (get_value( table
, row
, column
, &val
) != S_OK
) return NULL
;
159 switch (table
->columns
[column
].type
& COL_TYPE_MASK
)
163 len
= strlenW( (const WCHAR
*)(INT_PTR
)val
) + 2;
164 if (!(ret
= SysAllocStringLen( NULL
, len
))) return NULL
;
165 sprintfW( ret
, fmt_strW
, (const WCHAR
*)(INT_PTR
)val
);
170 sprintfW( number
, fmt_signedW
, val
);
171 return SysAllocString( number
);
175 sprintfW( number
, fmt_unsignedW
, val
);
176 return SysAllocString( number
);
179 wsprintfW( number
, fmt_signed64W
, val
);
180 return SysAllocString( number
);
183 wsprintfW( number
, fmt_unsigned64W
, val
);
184 return SysAllocString( number
);
187 FIXME("unhandled column type %u\n", table
->columns
[column
].type
& COL_TYPE_MASK
);
193 HRESULT
set_value( const struct table
*table
, UINT row
, UINT column
, LONGLONG val
,
196 UINT col_offset
, row_size
;
199 if ((table
->columns
[column
].type
& COL_TYPE_MASK
) != type
) return WBEM_E_TYPE_MISMATCH
;
201 col_offset
= get_column_offset( table
, column
);
202 row_size
= get_row_size( table
);
203 ptr
= table
->data
+ row
* row_size
+ col_offset
;
205 switch (table
->columns
[column
].type
& COL_TYPE_MASK
)
209 *(WCHAR
**)ptr
= (WCHAR
*)(INT_PTR
)val
;
215 *(UINT16
*)ptr
= val
;
221 *(UINT32
*)ptr
= val
;
227 *(UINT64
*)ptr
= val
;
230 FIXME("unhandled column type %u\n", type
);
231 return WBEM_E_FAILED
;
236 static void clear_table( struct table
*table
)
241 if (!table
->data
) return;
243 for (i
= 0; i
< table
->num_rows
; i
++)
245 for (j
= 0; j
< table
->num_cols
; j
++)
247 if (!(table
->columns
[j
].type
& COL_FLAG_DYNAMIC
)) continue;
249 type
= table
->columns
[j
].type
& COL_TYPE_MASK
;
250 if (type
== CIM_STRING
|| type
== CIM_DATETIME
|| (type
& CIM_FLAG_ARRAY
))
252 if (get_value( table
, i
, j
, &val
) == S_OK
) heap_free( (void *)(INT_PTR
)val
);
259 heap_free( table
->data
);
264 void free_columns( struct column
*columns
, UINT num_cols
)
268 for (i
= 0; i
< num_cols
; i
++)
270 heap_free( (WCHAR
*)columns
[i
].name
);
272 heap_free( columns
);
275 void free_table( struct table
*table
)
279 clear_table( table
);
280 if (table
->flags
& TABLE_FLAG_DYNAMIC
)
282 heap_free( (WCHAR
*)table
->name
);
283 free_columns( (struct column
*)table
->columns
, table
->num_cols
);
288 struct table
*get_table( const WCHAR
*name
)
292 LIST_FOR_EACH_ENTRY( table
, table_list
, struct table
, entry
)
294 if (!strcmpiW( table
->name
, name
))
296 if (table
->fill
&& !table
->data
) table
->fill( table
);
303 struct table
*create_table( const WCHAR
*name
, UINT num_cols
, const struct column
*columns
,
304 UINT num_rows
, BYTE
*data
, void (*fill
)(struct table
*) )
308 if (!(table
= heap_alloc( sizeof(*table
) ))) return NULL
;
310 table
->num_cols
= num_cols
;
311 table
->columns
= columns
;
312 table
->num_rows
= num_rows
;
315 table
->flags
= TABLE_FLAG_DYNAMIC
;
319 BOOL
add_table( struct table
*table
)
323 LIST_FOR_EACH_ENTRY( iter
, table_list
, struct table
, entry
)
325 if (!strcmpiW( iter
->name
, table
->name
))
327 TRACE("table %s already exists\n", debugstr_w(table
->name
));
331 list_add_tail( table_list
, &table
->entry
);
335 const WCHAR
*get_method_name( const WCHAR
*class, UINT index
)
340 if (!(table
= get_table( class ))) return NULL
;
342 for (i
= 0; i
< table
->num_cols
; i
++)
344 if (table
->columns
[i
].type
& COL_FLAG_METHOD
)
346 if (index
== count
) return table
->columns
[i
].name
;
353 const WCHAR
*get_property_name( const WCHAR
*class, UINT index
)
358 if (!(table
= get_table( class ))) return NULL
;
360 for (i
= 0; i
< table
->num_cols
; i
++)
362 if (!(table
->columns
[i
].type
& COL_FLAG_METHOD
))
364 if (index
== count
) return table
->columns
[i
].name
;