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"
36 WINE_DEFAULT_DEBUG_CHANNEL(msidb);
39 /* below is the query interface to a table */
41 typedef struct tagMSIWHEREVIEW
51 static UINT WHERE_fetch_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT *val )
53 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
55 TRACE("%p %d %d %p\n", wv, row, col, val );
58 return ERROR_FUNCTION_FAILED;
60 if( row > wv->row_count )
61 return ERROR_NO_MORE_ITEMS;
63 row = wv->reorder[ row ];
65 return wv->table->ops->fetch_int( wv->table, row, col, val );
68 static UINT WHERE_fetch_stream( struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm )
70 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
72 TRACE("%p %d %d %p\n", wv, row, col, stm );
75 return ERROR_FUNCTION_FAILED;
77 if( row > wv->row_count )
78 return ERROR_NO_MORE_ITEMS;
80 row = wv->reorder[ row ];
82 return wv->table->ops->fetch_stream( wv->table, row, col, stm );
85 static UINT WHERE_set_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT val )
87 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
89 TRACE("%p %d %d %04x\n", wv, row, col, val );
92 return ERROR_FUNCTION_FAILED;
94 if( row > wv->row_count )
95 return ERROR_NO_MORE_ITEMS;
97 row = wv->reorder[ row ];
99 return wv->table->ops->set_int( wv->table, row, col, val );
102 static UINT INT_evaluate( UINT lval, UINT op, UINT rval )
107 return ( lval == rval );
109 return ( lval && rval );
111 return ( lval || rval );
113 return ( lval > rval );
115 return ( lval < rval );
117 return ( lval <= rval );
119 return ( lval >= rval );
121 return ( lval != rval );
127 ERR("Unknown operator %d\n", op );
132 static const WCHAR *STRING_evaluate( string_table *st,
133 MSIVIEW *table, UINT row, struct expr *expr, MSIRECORD *record )
139 case EXPR_COL_NUMBER_STRING:
140 r = table->ops->fetch_int( table, row, expr->u.col_number, &val );
141 if( r != ERROR_SUCCESS )
143 return msi_string_lookup_id( st, val );
149 return MSI_RecordGetString( record, 1 );
152 ERR("Invalid expression type\n");
158 static UINT STRCMP_Evaluate( string_table *st, MSIVIEW *table, UINT row,
159 struct expr *cond, UINT *val, MSIRECORD *record )
162 const WCHAR *l_str, *r_str;
164 l_str = STRING_evaluate( st, table, row, cond->u.expr.left, record );
165 r_str = STRING_evaluate( st, table, row, cond->u.expr.right, record );
168 else if( l_str && ! r_str )
170 else if( r_str && ! l_str )
173 sr = lstrcmpW( l_str, r_str );
175 *val = ( cond->u.expr.op == OP_EQ && ( sr == 0 ) ) ||
176 ( cond->u.expr.op == OP_LT && ( sr < 0 ) ) ||
177 ( cond->u.expr.op == OP_GT && ( sr > 0 ) );
179 return ERROR_SUCCESS;
182 static UINT WHERE_evaluate( MSIDATABASE *db, MSIVIEW *table, UINT row,
183 struct expr *cond, UINT *val, MSIRECORD *record )
188 return ERROR_SUCCESS;
192 case EXPR_COL_NUMBER_STRING:
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 msi_free( wv->reorder );
246 wv->reorder = msi_alloc( count*sizeof(UINT) );
248 return ERROR_FUNCTION_FAILED;
251 if (wv->cond->type == EXPR_STRCMP)
253 MSIITERHANDLE handle = NULL;
254 UINT row, value, col;
255 struct expr *col_cond = wv->cond->u.expr.left;
256 struct expr *val_cond = wv->cond->u.expr.right;
258 /* swap conditionals */
259 if (col_cond->type != EXPR_COL_NUMBER_STRING)
261 val_cond = wv->cond->u.expr.left;
262 col_cond = wv->cond->u.expr.right;
265 if ((col_cond->type == EXPR_COL_NUMBER_STRING) && (val_cond->type == EXPR_SVAL))
267 col = col_cond->u.col_number;
268 r = msi_string2idW(wv->db->strings, val_cond->u.sval, &value);
269 if (r != ERROR_SUCCESS)
271 TRACE("no id for %s, assuming it doesn't exist in the table\n", debugstr_w(wv->cond->u.expr.right->u.sval));
272 return ERROR_SUCCESS;
277 r = table->ops->find_matching_rows(table, col, value, &row, &handle);
278 if (r == ERROR_SUCCESS)
279 wv->reorder[ wv->row_count ++ ] = row;
280 } while (r == ERROR_SUCCESS);
282 if (r == ERROR_NO_MORE_ITEMS)
283 return ERROR_SUCCESS;
287 /* else fallback to slow case */
290 for( i=0; i<count; i++ )
293 r = WHERE_evaluate( wv->db, table, i, wv->cond, &val, record );
294 if( r != ERROR_SUCCESS )
297 wv->reorder[ wv->row_count ++ ] = i;
300 return ERROR_SUCCESS;
303 static UINT WHERE_close( struct tagMSIVIEW *view )
305 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
310 return ERROR_FUNCTION_FAILED;
312 msi_free( wv->reorder );
315 return wv->table->ops->close( wv->table );
318 static UINT WHERE_get_dimensions( struct tagMSIVIEW *view, UINT *rows, UINT *cols )
320 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
322 TRACE("%p %p %p\n", wv, rows, cols );
325 return ERROR_FUNCTION_FAILED;
330 return ERROR_FUNCTION_FAILED;
331 *rows = wv->row_count;
334 return wv->table->ops->get_dimensions( wv->table, NULL, cols );
337 static UINT WHERE_get_column_info( struct tagMSIVIEW *view,
338 UINT n, LPWSTR *name, UINT *type )
340 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
342 TRACE("%p %d %p %p\n", wv, n, name, type );
345 return ERROR_FUNCTION_FAILED;
347 return wv->table->ops->get_column_info( wv->table, n, name, type );
350 static UINT WHERE_modify( struct tagMSIVIEW *view, MSIMODIFY eModifyMode,
353 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
355 TRACE("%p %d %p\n", wv, eModifyMode, rec );
358 return ERROR_FUNCTION_FAILED;
360 return wv->table->ops->modify( wv->table, eModifyMode, rec );
363 static UINT WHERE_delete( struct tagMSIVIEW *view )
365 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
370 wv->table->ops->delete( wv->table );
373 msi_free( wv->reorder );
377 msiobj_release( &wv->db->hdr );
380 return ERROR_SUCCESS;
383 static UINT WHERE_find_matching_rows( struct tagMSIVIEW *view, UINT col,
384 UINT val, UINT *row, MSIITERHANDLE *handle )
386 MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
389 TRACE("%p, %d, %u, %p\n", view, col, val, *handle);
392 return ERROR_FUNCTION_FAILED;
394 r = wv->table->ops->find_matching_rows( wv->table, col, val, row, handle );
396 if( *row > wv->row_count )
397 return ERROR_NO_MORE_ITEMS;
399 *row = wv->reorder[ *row ];
405 MSIVIEWOPS where_ops =
413 WHERE_get_dimensions,
414 WHERE_get_column_info,
417 WHERE_find_matching_rows
420 static UINT WHERE_VerifyCondition( MSIDATABASE *db, MSIVIEW *table, struct expr *cond,
428 r = VIEW_find_column( table, cond->u.column, &val );
429 if( r == ERROR_SUCCESS )
432 r = table->ops->get_column_info( table, val, NULL, &type );
433 if( r == ERROR_SUCCESS )
435 if (type&MSITYPE_STRING)
436 cond->type = EXPR_COL_NUMBER_STRING;
438 cond->type = EXPR_COL_NUMBER;
439 cond->u.col_number = val;
448 ERR("Couldn't find column %s\n", debugstr_w( cond->u.column ) );
452 r = WHERE_VerifyCondition( db, table, cond->u.expr.left, valid );
453 if( r != ERROR_SUCCESS )
456 return ERROR_SUCCESS;
457 r = WHERE_VerifyCondition( db, table, cond->u.expr.right, valid );
458 if( r != ERROR_SUCCESS )
461 /* check the type of the comparison */
462 if( ( cond->u.expr.left->type == EXPR_SVAL ) ||
463 ( cond->u.expr.left->type == EXPR_COL_NUMBER_STRING ) ||
464 ( cond->u.expr.right->type == EXPR_SVAL ) ||
465 ( cond->u.expr.right->type == EXPR_COL_NUMBER_STRING ) )
467 switch( cond->u.expr.op )
475 return ERROR_INVALID_PARAMETER;
478 /* FIXME: check we're comparing a string to a column */
480 cond->type = EXPR_STRCMP;
486 cond->type = EXPR_UVAL;
487 cond->u.uval = cond->u.ival + (1<<15);
496 ERR("Invalid expression type\n");
501 return ERROR_SUCCESS;
504 UINT WHERE_CreateView( MSIDATABASE *db, MSIVIEW **view, MSIVIEW *table,
507 MSIWHEREVIEW *wv = NULL;
508 UINT count = 0, r, valid = 0;
510 TRACE("%p\n", table );
512 r = table->ops->get_dimensions( table, NULL, &count );
513 if( r != ERROR_SUCCESS )
515 ERR("can't get table dimensions\n");
521 r = WHERE_VerifyCondition( db, table, cond, &valid );
522 if( r != ERROR_SUCCESS )
525 return ERROR_FUNCTION_FAILED;
528 wv = msi_alloc_zero( sizeof *wv );
530 return ERROR_FUNCTION_FAILED;
532 /* fill the structure */
533 wv->view.ops = &where_ops;
534 msiobj_addref( &db->hdr );
540 *view = (MSIVIEW*) wv;
542 return ERROR_SUCCESS;