2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2002-2004 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
26 #include "wine/debug.h"
36 WINE_DEFAULT_DEBUG_CHANNEL(msidb
);
39 /* below is the query interface to a table */
41 typedef struct tagMSISELECTVIEW
51 static UINT
SELECT_fetch_int( struct tagMSIVIEW
*view
, UINT row
, UINT col
, UINT
*val
)
53 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
55 TRACE("%p %d %d %p\n", sv
, row
, col
, val
);
58 return ERROR_FUNCTION_FAILED
;
60 if( (col
==0) || (col
>sv
->num_cols
) )
61 return ERROR_FUNCTION_FAILED
;
63 col
= sv
->cols
[ col
- 1 ];
65 return sv
->table
->ops
->fetch_int( sv
->table
, row
, col
, val
);
68 static UINT
SELECT_fetch_stream( struct tagMSIVIEW
*view
, UINT row
, UINT col
, IStream
**stm
)
70 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
72 TRACE("%p %d %d %p\n", sv
, row
, col
, stm
);
75 return ERROR_FUNCTION_FAILED
;
77 if( (col
==0) || (col
>sv
->num_cols
) )
78 return ERROR_FUNCTION_FAILED
;
80 col
= sv
->cols
[ col
- 1 ];
82 return sv
->table
->ops
->fetch_stream( sv
->table
, row
, col
, stm
);
85 static UINT
SELECT_get_row( struct tagMSIVIEW
*view
, UINT row
, MSIRECORD
**rec
)
87 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
89 TRACE("%p %d %p\n", sv
, row
, rec
);
92 return ERROR_FUNCTION_FAILED
;
94 return msi_view_get_row(sv
->db
, view
, row
, rec
);
97 static UINT
SELECT_set_row( struct tagMSIVIEW
*view
, UINT row
, MSIRECORD
*rec
, UINT mask
)
99 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
100 UINT i
, expanded_mask
= 0, r
= ERROR_SUCCESS
, col_count
= 0;
103 TRACE("%p %d %p %08x\n", sv
, row
, rec
, mask
);
106 return ERROR_FUNCTION_FAILED
;
108 /* test if any of the mask bits are invalid */
109 if ( mask
>= (1<<sv
->num_cols
) )
110 return ERROR_INVALID_PARAMETER
;
112 /* find the number of columns in the table below */
113 r
= sv
->table
->ops
->get_dimensions( sv
->table
, NULL
, &col_count
);
117 /* expand the record to the right size for the underlying table */
118 expanded
= MSI_CreateRecord( col_count
);
120 return ERROR_FUNCTION_FAILED
;
122 /* move the right fields across */
123 for ( i
=0; i
<sv
->num_cols
; i
++ )
125 r
= MSI_RecordCopyField( rec
, i
+1, expanded
, sv
->cols
[ i
] );
126 if (r
!= ERROR_SUCCESS
)
128 expanded_mask
|= (1<<(sv
->cols
[i
]-1));
131 /* set the row in the underlying table */
132 if (r
== ERROR_SUCCESS
)
133 r
= sv
->table
->ops
->set_row( sv
->table
, row
, expanded
, expanded_mask
);
135 msiobj_release( &expanded
->hdr
);
139 static UINT
SELECT_insert_row( struct tagMSIVIEW
*view
, MSIRECORD
*record
, UINT row
, BOOL temporary
)
141 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
142 UINT i
, table_cols
, r
;
145 TRACE("%p %p\n", sv
, record
);
148 return ERROR_FUNCTION_FAILED
;
150 /* rearrange the record to suit the table */
151 r
= sv
->table
->ops
->get_dimensions( sv
->table
, NULL
, &table_cols
);
152 if (r
!= ERROR_SUCCESS
)
155 outrec
= MSI_CreateRecord( table_cols
+ 1 );
157 for (i
=0; i
<sv
->num_cols
; i
++)
159 r
= MSI_RecordCopyField( record
, i
+1, outrec
, sv
->cols
[i
] );
160 if (r
!= ERROR_SUCCESS
)
164 r
= sv
->table
->ops
->insert_row( sv
->table
, outrec
, row
, temporary
);
167 msiobj_release( &outrec
->hdr
);
172 static UINT
SELECT_execute( struct tagMSIVIEW
*view
, MSIRECORD
*record
)
174 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
176 TRACE("%p %p\n", sv
, record
);
179 return ERROR_FUNCTION_FAILED
;
181 return sv
->table
->ops
->execute( sv
->table
, record
);
184 static UINT
SELECT_close( struct tagMSIVIEW
*view
)
186 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
191 return ERROR_FUNCTION_FAILED
;
193 return sv
->table
->ops
->close( sv
->table
);
196 static UINT
SELECT_get_dimensions( struct tagMSIVIEW
*view
, UINT
*rows
, UINT
*cols
)
198 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
200 TRACE("%p %p %p\n", sv
, rows
, cols
);
203 return ERROR_FUNCTION_FAILED
;
206 *cols
= sv
->num_cols
;
208 return sv
->table
->ops
->get_dimensions( sv
->table
, rows
, NULL
);
211 static UINT
SELECT_get_column_info( struct tagMSIVIEW
*view
, UINT n
, LPCWSTR
*name
,
212 UINT
*type
, BOOL
*temporary
, LPCWSTR
*table_name
)
214 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
216 TRACE("%p %d %p %p %p %p\n", sv
, n
, name
, type
, temporary
, table_name
);
219 return ERROR_FUNCTION_FAILED
;
221 if( (n
==0) || (n
>sv
->num_cols
) )
222 return ERROR_FUNCTION_FAILED
;
224 n
= sv
->cols
[ n
- 1 ];
226 return sv
->table
->ops
->get_column_info( sv
->table
, n
, name
,
227 type
, temporary
, table_name
);
230 static UINT
msi_select_update(struct tagMSIVIEW
*view
, MSIRECORD
*rec
, UINT row
)
232 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
233 UINT r
, i
, num_columns
, col
, type
, val
;
237 r
= SELECT_get_dimensions(view
, NULL
, &num_columns
);
238 if (r
!= ERROR_SUCCESS
)
241 r
= sv
->table
->ops
->get_row(sv
->table
, row
- 1, &mod
);
242 if (r
!= ERROR_SUCCESS
)
245 for (i
= 0; i
< num_columns
; i
++)
249 r
= SELECT_get_column_info(view
, i
+ 1, NULL
, &type
, NULL
, NULL
);
250 if (r
!= ERROR_SUCCESS
)
252 ERR("Failed to get column information: %d\n", r
);
256 if (MSITYPE_IS_BINARY(type
))
258 ERR("Cannot modify binary data!\n");
259 r
= ERROR_FUNCTION_FAILED
;
262 else if (type
& MSITYPE_STRING
)
264 str
= MSI_RecordGetString(rec
, i
+ 1);
265 r
= MSI_RecordSetStringW(mod
, col
, str
);
269 val
= MSI_RecordGetInteger(rec
, i
+ 1);
270 r
= MSI_RecordSetInteger(mod
, col
, val
);
273 if (r
!= ERROR_SUCCESS
)
275 ERR("Failed to modify record: %d\n", r
);
280 r
= sv
->table
->ops
->modify(sv
->table
, MSIMODIFY_UPDATE
, mod
, row
);
283 msiobj_release(&mod
->hdr
);
287 static UINT
SELECT_modify( struct tagMSIVIEW
*view
, MSIMODIFY eModifyMode
,
288 MSIRECORD
*rec
, UINT row
)
290 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
292 TRACE("%p %d %p %d\n", sv
, eModifyMode
, rec
, row
);
295 return ERROR_FUNCTION_FAILED
;
297 if (eModifyMode
== MSIMODIFY_UPDATE
)
298 return msi_select_update(view
, rec
, row
);
300 return sv
->table
->ops
->modify( sv
->table
, eModifyMode
, rec
, row
);
303 static UINT
SELECT_delete( struct tagMSIVIEW
*view
)
305 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
310 sv
->table
->ops
->delete( sv
->table
);
315 return ERROR_SUCCESS
;
318 static UINT
SELECT_find_matching_rows( struct tagMSIVIEW
*view
, UINT col
,
319 UINT val
, UINT
*row
, MSIITERHANDLE
*handle
)
321 MSISELECTVIEW
*sv
= (MSISELECTVIEW
*)view
;
323 TRACE("%p, %d, %u, %p\n", view
, col
, val
, *handle
);
326 return ERROR_FUNCTION_FAILED
;
328 if( (col
==0) || (col
>sv
->num_cols
) )
329 return ERROR_FUNCTION_FAILED
;
331 col
= sv
->cols
[ col
- 1 ];
333 return sv
->table
->ops
->find_matching_rows( sv
->table
, col
, val
, row
, handle
);
337 static const MSIVIEWOPS select_ops
=
347 SELECT_get_dimensions
,
348 SELECT_get_column_info
,
351 SELECT_find_matching_rows
,
360 static UINT
SELECT_AddColumn( MSISELECTVIEW
*sv
, LPCWSTR name
,
366 TRACE("%p adding %s.%s\n", sv
, debugstr_w( table_name
),
369 if( sv
->view
.ops
!= &select_ops
)
370 return ERROR_FUNCTION_FAILED
;
374 return ERROR_FUNCTION_FAILED
;
375 if( !table
->ops
->get_dimensions
)
376 return ERROR_FUNCTION_FAILED
;
377 if( !table
->ops
->get_column_info
)
378 return ERROR_FUNCTION_FAILED
;
380 if( sv
->num_cols
>= sv
->max_cols
)
381 return ERROR_FUNCTION_FAILED
;
383 r
= VIEW_find_column( table
, name
, table_name
, &n
);
384 if( r
!= ERROR_SUCCESS
)
387 sv
->cols
[sv
->num_cols
] = n
;
388 TRACE("Translating column %s from %d -> %d\n",
389 debugstr_w( name
), sv
->num_cols
, n
);
393 return ERROR_SUCCESS
;
396 static int select_count_columns( const column_info
*col
)
399 for (n
= 0; col
; col
= col
->next
)
404 UINT
SELECT_CreateView( MSIDATABASE
*db
, MSIVIEW
**view
, MSIVIEW
*table
,
405 const column_info
*columns
)
407 MSISELECTVIEW
*sv
= NULL
;
408 UINT count
= 0, r
= ERROR_SUCCESS
;
412 count
= select_count_columns( columns
);
414 sv
= msi_alloc_zero( sizeof *sv
+ count
*sizeof (UINT
) );
416 return ERROR_FUNCTION_FAILED
;
418 /* fill the structure */
419 sv
->view
.ops
= &select_ops
;
423 sv
->max_cols
= count
;
427 r
= SELECT_AddColumn( sv
, columns
->column
, columns
->table
);
430 columns
= columns
->next
;
433 if( r
== ERROR_SUCCESS
)