2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2002 Mike McCormack for CodeWeavers
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 #include "wine/debug.h"
27 #include "wine/unicode.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(msi);
40 /* below is the query interface to a table */
42 typedef struct tagMSIWHEREVIEW
52 static UINT WHERE_fetch_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT *val )
54 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
56 TRACE("%p %d %d %p\n", wv, row, col, val );
59 return ERROR_FUNCTION_FAILED;
61 if( row > wv->row_count )
62 return ERROR_NO_MORE_ITEMS;
64 row = wv->reorder[ row ];
66 return wv->table->ops->fetch_int( wv->table, row, col, val );
69 static UINT WHERE_fetch_stream( struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm )
71 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
73 TRACE("%p %d %d %p\n", wv, row, col, stm );
76 return ERROR_FUNCTION_FAILED;
78 if( row > wv->row_count )
79 return ERROR_NO_MORE_ITEMS;
81 row = wv->reorder[ row ];
83 return wv->table->ops->fetch_stream( wv->table, row, col, stm );
86 static UINT WHERE_set_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT val )
88 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
90 TRACE("%p %d %d %04x\n", wv, row, col, val );
93 return ERROR_FUNCTION_FAILED;
95 if( row > wv->row_count )
96 return ERROR_NO_MORE_ITEMS;
98 row = wv->reorder[ row ];
100 return wv->table->ops->set_int( wv->table, row, col, val );
103 static UINT INT_evaluate( UINT lval, UINT op, UINT rval )
108 return ( lval == rval );
110 return ( lval && rval );
112 return ( lval || rval );
114 return ( lval > rval );
116 return ( lval < rval );
118 return ( lval <= rval );
120 return ( lval >= rval );
122 return ( lval != rval );
128 ERR("Unknown operator %d\n", op );
133 static const WCHAR *STRING_evaluate( string_table *st,
134 MSIVIEW *table, UINT row, struct expr *expr, MSIRECORD *record )
140 case EXPR_COL_NUMBER:
141 r = table->ops->fetch_int( table, row, expr->u.col_number, &val );
142 if( r != ERROR_SUCCESS )
144 return msi_string_lookup_id( st, val );
150 return MSI_RecordGetString( record, 1 );
153 ERR("Invalid expression type\n");
159 static UINT STRCMP_Evaluate( string_table *st, MSIVIEW *table, UINT row,
160 struct expr *cond, UINT *val, MSIRECORD *record )
163 const WCHAR *l_str, *r_str;
165 l_str = STRING_evaluate( st, table, row, cond->u.expr.left, record );
166 r_str = STRING_evaluate( st, table, row, cond->u.expr.right, record );
169 else if( l_str && ! r_str )
171 else if( r_str && ! l_str )
174 sr = strcmpW( l_str, r_str );
176 *val = ( cond->u.expr.op == OP_EQ && ( sr == 0 ) ) ||
177 ( cond->u.expr.op == OP_LT && ( sr < 0 ) ) ||
178 ( cond->u.expr.op == OP_GT && ( sr > 0 ) );
180 return ERROR_SUCCESS;
183 static UINT WHERE_evaluate( MSIDATABASE *db, MSIVIEW *table, UINT row,
184 struct expr *cond, UINT *val, MSIRECORD *record )
189 return ERROR_SUCCESS;
193 case EXPR_COL_NUMBER:
194 return table->ops->fetch_int( table, row, cond->u.col_number, val );
198 return ERROR_SUCCESS;
201 r = WHERE_evaluate( db, table, row, cond->u.expr.left, &lval, record );
202 if( r != ERROR_SUCCESS )
204 r = WHERE_evaluate( db, table, row, cond->u.expr.right, &rval, record );
205 if( r != ERROR_SUCCESS )
207 *val = INT_evaluate( lval, cond->u.expr.op, rval );
208 return ERROR_SUCCESS;
211 return STRCMP_Evaluate( db->strings, table, row, cond, val, record );
214 *val = MSI_RecordGetInteger( record, 1 );
215 return ERROR_SUCCESS;
218 ERR("Invalid expression type\n");
222 return ERROR_SUCCESS;
226 static UINT WHERE_execute( struct tagMSIVIEW *view, MSIRECORD *record )
228 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
229 UINT count = 0, r, val, i;
230 MSIVIEW *table = wv->table;
232 TRACE("%p %p\n", wv, record);
235 return ERROR_FUNCTION_FAILED;
237 r = table->ops->execute( table, record );
238 if( r != ERROR_SUCCESS )
241 r = table->ops->get_dimensions( table, &count, NULL );
242 if( r != ERROR_SUCCESS )
245 wv->reorder = HeapAlloc( GetProcessHeap(), 0, count*sizeof(UINT) );
247 return ERROR_FUNCTION_FAILED;
249 for( i=0; i<count; i++ )
252 r = WHERE_evaluate( wv->db, table, i, wv->cond, &val, record );
253 if( r != ERROR_SUCCESS )
256 wv->reorder[ wv->row_count ++ ] = i;
259 return ERROR_SUCCESS;
262 static UINT WHERE_close( struct tagMSIVIEW *view )
264 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
269 return ERROR_FUNCTION_FAILED;
272 HeapFree( GetProcessHeap(), 0, wv->reorder );
275 return wv->table->ops->close( wv->table );
278 static UINT WHERE_get_dimensions( struct tagMSIVIEW *view, UINT *rows, UINT *cols )
280 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
282 TRACE("%p %p %p\n", wv, rows, cols );
285 return ERROR_FUNCTION_FAILED;
290 return ERROR_FUNCTION_FAILED;
291 *rows = wv->row_count;
294 return wv->table->ops->get_dimensions( wv->table, NULL, cols );
297 static UINT WHERE_get_column_info( struct tagMSIVIEW *view,
298 UINT n, LPWSTR *name, UINT *type )
300 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
302 TRACE("%p %d %p %p\n", wv, n, name, type );
305 return ERROR_FUNCTION_FAILED;
307 return wv->table->ops->get_column_info( wv->table, n, name, type );
310 static UINT WHERE_modify( struct tagMSIVIEW *view, MSIMODIFY eModifyMode, MSIHANDLE hrec)
312 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
314 TRACE("%p %d %ld\n", wv, eModifyMode, hrec );
317 return ERROR_FUNCTION_FAILED;
319 return wv->table->ops->modify( wv->table, eModifyMode, hrec );
322 static UINT WHERE_delete( struct tagMSIVIEW *view )
324 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
329 wv->table->ops->delete( wv->table );
332 HeapFree( GetProcessHeap(), 0, wv->reorder );
337 delete_expr( wv->cond );
339 msiobj_release( &wv->db->hdr );
340 HeapFree( GetProcessHeap(), 0, wv );
342 return ERROR_SUCCESS;
346 MSIVIEWOPS where_ops =
354 WHERE_get_dimensions,
355 WHERE_get_column_info,
360 static UINT WHERE_VerifyCondition( MSIDATABASE *db, MSIVIEW *table, struct expr *cond,
368 r = VIEW_find_column( table, cond->u.column, &val );
369 if( r == ERROR_SUCCESS )
372 cond->type = EXPR_COL_NUMBER;
373 cond->u.col_number = val;
378 ERR("Couldn't find column %s\n", debugstr_w( cond->u.column ) );
382 r = WHERE_VerifyCondition( db, table, cond->u.expr.left, valid );
383 if( r != ERROR_SUCCESS )
386 return ERROR_SUCCESS;
387 r = WHERE_VerifyCondition( db, table, cond->u.expr.right, valid );
388 if( r != ERROR_SUCCESS )
391 /* check the type of the comparison */
392 if( ( cond->u.expr.left->type == EXPR_SVAL ) ||
393 ( cond->u.expr.right->type == EXPR_SVAL ) )
395 switch( cond->u.expr.op )
403 return ERROR_INVALID_PARAMETER;
406 /* FIXME: check we're comparing a string to a column */
408 cond->type = EXPR_STRCMP;
414 cond->type = EXPR_UVAL;
415 cond->u.uval = cond->u.ival + (1<<15);
424 ERR("Invalid expression type\n");
429 return ERROR_SUCCESS;
432 UINT WHERE_CreateView( MSIDATABASE *db, MSIVIEW **view, MSIVIEW *table,
435 MSIWHEREVIEW *wv = NULL;
436 UINT count = 0, r, valid = 0;
440 r = table->ops->get_dimensions( table, NULL, &count );
441 if( r != ERROR_SUCCESS )
443 ERR("can't get table dimensions\n");
449 r = WHERE_VerifyCondition( db, table, cond, &valid );
450 if( r != ERROR_SUCCESS )
453 return ERROR_FUNCTION_FAILED;
456 wv = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof *wv );
458 return ERROR_FUNCTION_FAILED;
460 /* fill the structure */
461 wv->view.ops = &where_ops;
462 msiobj_addref( &db->hdr );
468 *view = (MSIVIEW*) wv;
470 return ERROR_SUCCESS;