- Merge from trunk
[reactos.git] / dll / win32 / msi / join.c
1 /*
2 * Implementation of the Microsoft Installer (msi.dll)
3 *
4 * Copyright 2006 Mike McCormack for CodeWeavers
5 *
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.
10 *
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.
15 *
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
19 */
20
21 #include <stdarg.h>
22
23 #include "windef.h"
24 #include "winbase.h"
25 #include "winerror.h"
26 #include "msi.h"
27 #include "msiquery.h"
28 #include "objbase.h"
29 #include "objidl.h"
30 #include "msipriv.h"
31 #include "query.h"
32
33 #include "wine/debug.h"
34 #include "wine/unicode.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(msidb);
37
38 typedef struct tagJOINTABLE
39 {
40 struct list entry;
41 MSIVIEW *view;
42 UINT columns;
43 UINT rows;
44 UINT next_rows;
45 } JOINTABLE;
46
47 typedef struct tagMSIJOINVIEW
48 {
49 MSIVIEW view;
50 MSIDATABASE *db;
51 struct list tables;
52 UINT columns;
53 UINT rows;
54 } MSIJOINVIEW;
55
56 static UINT JOIN_fetch_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT *val )
57 {
58 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
59 JOINTABLE *table;
60 UINT cols = 0;
61 UINT prev_rows = 1;
62
63 if (col == 0 || col > jv->columns)
64 return ERROR_FUNCTION_FAILED;
65
66 if (row >= jv->rows)
67 return ERROR_FUNCTION_FAILED;
68
69 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
70 {
71 if (col <= cols + table->columns)
72 {
73 row = (row % (jv->rows / table->next_rows)) / prev_rows;
74 col -= cols;
75 break;
76 }
77
78 prev_rows *= table->rows;
79 cols += table->columns;
80 }
81
82 return table->view->ops->fetch_int( table->view, row, col, val );
83 }
84
85 static UINT JOIN_fetch_stream( struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm)
86 {
87 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
88 JOINTABLE *table;
89 UINT cols = 0;
90 UINT prev_rows = 1;
91
92 TRACE("%p %d %d %p\n", jv, row, col, stm );
93
94 if (col == 0 || col > jv->columns)
95 return ERROR_FUNCTION_FAILED;
96
97 if (row >= jv->rows)
98 return ERROR_FUNCTION_FAILED;
99
100 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
101 {
102 if (col <= cols + table->columns)
103 {
104 row = (row % (jv->rows / table->next_rows)) / prev_rows;
105 col -= cols;
106 break;
107 }
108
109 prev_rows *= table->rows;
110 cols += table->columns;
111 }
112
113 return table->view->ops->fetch_stream( table->view, row, col, stm );
114 }
115
116 static UINT JOIN_get_row( struct tagMSIVIEW *view, UINT row, MSIRECORD **rec )
117 {
118 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
119
120 TRACE("%p %d %p\n", jv, row, rec);
121
122 return msi_view_get_row( jv->db, view, row, rec );
123 }
124
125 static UINT JOIN_execute( struct tagMSIVIEW *view, MSIRECORD *record )
126 {
127 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
128 JOINTABLE *table;
129 UINT r, rows;
130
131 TRACE("%p %p\n", jv, record);
132
133 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
134 {
135 table->view->ops->execute(table->view, NULL);
136
137 r = table->view->ops->get_dimensions(table->view, &table->rows, NULL);
138 if (r != ERROR_SUCCESS)
139 {
140 ERR("failed to get table dimensions\n");
141 return r;
142 }
143
144 /* each table must have at least one row */
145 if (table->rows == 0)
146 {
147 jv->rows = 0;
148 return ERROR_SUCCESS;
149 }
150
151 if (jv->rows == 0)
152 jv->rows = table->rows;
153 else
154 jv->rows *= table->rows;
155 }
156
157 rows = jv->rows;
158 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
159 {
160 rows /= table->rows;
161 table->next_rows = rows;
162 }
163
164 return ERROR_SUCCESS;
165 }
166
167 static UINT JOIN_close( struct tagMSIVIEW *view )
168 {
169 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
170 JOINTABLE *table;
171
172 TRACE("%p\n", jv );
173
174 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
175 {
176 table->view->ops->close(table->view);
177 }
178
179 return ERROR_SUCCESS;
180 }
181
182 static UINT JOIN_get_dimensions( struct tagMSIVIEW *view, UINT *rows, UINT *cols )
183 {
184 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
185
186 TRACE("%p %p %p\n", jv, rows, cols );
187
188 if (cols)
189 *cols = jv->columns;
190
191 if (rows)
192 *rows = jv->rows;
193
194 return ERROR_SUCCESS;
195 }
196
197 static UINT JOIN_get_column_info( struct tagMSIVIEW *view,
198 UINT n, LPWSTR *name, UINT *type, BOOL *temporary,
199 LPWSTR *table_name )
200 {
201 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
202 JOINTABLE *table;
203 UINT cols = 0;
204
205 TRACE("%p %d %p %p %p %p\n", jv, n, name, type, temporary, table_name );
206
207 if (n == 0 || n > jv->columns)
208 return ERROR_FUNCTION_FAILED;
209
210 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
211 {
212 if (n <= cols + table->columns)
213 return table->view->ops->get_column_info(table->view, n - cols,
214 name, type, temporary,
215 table_name);
216
217 cols += table->columns;
218 }
219
220 return ERROR_FUNCTION_FAILED;
221 }
222
223 static UINT join_find_row( MSIJOINVIEW *jv, MSIRECORD *rec, UINT *row )
224 {
225 LPCWSTR str;
226 UINT r, i, id, data;
227
228 str = MSI_RecordGetString( rec, 1 );
229 r = msi_string2idW( jv->db->strings, str, &id );
230 if (r != ERROR_SUCCESS)
231 return r;
232
233 for (i = 0; i < jv->rows; i++)
234 {
235 JOIN_fetch_int( &jv->view, i, 1, &data );
236
237 if (data == id)
238 {
239 *row = i;
240 return ERROR_SUCCESS;
241 }
242 }
243
244 return ERROR_FUNCTION_FAILED;
245 }
246
247 static UINT JOIN_set_row( struct tagMSIVIEW *view, UINT row, MSIRECORD *rec, UINT mask )
248 {
249 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
250 JOINTABLE *table;
251 UINT i, reduced_mask = 0, r = ERROR_SUCCESS, offset = 0, col_count;
252 MSIRECORD *reduced;
253
254 TRACE("%p %d %p %u %08x\n", jv, row, rec, rec->count, mask );
255
256 if (mask >= 1 << jv->columns)
257 return ERROR_INVALID_PARAMETER;
258
259 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
260 {
261 r = table->view->ops->get_dimensions( table->view, NULL, &col_count );
262 if (r != ERROR_SUCCESS)
263 return r;
264
265 reduced = MSI_CreateRecord( col_count );
266 if (!reduced)
267 return ERROR_FUNCTION_FAILED;
268
269 for (i = 0; i < col_count; i++)
270 {
271 r = MSI_RecordCopyField( rec, i + offset + 1, reduced, i + 1 );
272 if (r != ERROR_SUCCESS)
273 break;
274 }
275
276 offset += col_count;
277 reduced_mask = mask >> (jv->columns - offset) & ((1 << col_count) - 1);
278
279 if (r == ERROR_SUCCESS)
280 r = table->view->ops->set_row( table->view, row, reduced, reduced_mask );
281
282 msiobj_release( &reduced->hdr );
283 }
284
285 return r;
286 }
287
288 static UINT join_modify_update( struct tagMSIVIEW *view, MSIRECORD *rec )
289 {
290 MSIJOINVIEW *jv = (MSIJOINVIEW *)view;
291 UINT r, row;
292
293 r = join_find_row( jv, rec, &row );
294 if (r != ERROR_SUCCESS)
295 return r;
296
297 return JOIN_set_row( view, row, rec, (1 << jv->columns) - 1 );
298 }
299
300 static UINT JOIN_modify( struct tagMSIVIEW *view, MSIMODIFY mode, MSIRECORD *rec, UINT row )
301 {
302 UINT r;
303
304 TRACE("%p %d %p %u\n", view, mode, rec, row);
305
306 switch (mode)
307 {
308 case MSIMODIFY_UPDATE:
309 return join_modify_update( view, rec );
310
311 case MSIMODIFY_ASSIGN:
312 case MSIMODIFY_DELETE:
313 case MSIMODIFY_INSERT:
314 case MSIMODIFY_INSERT_TEMPORARY:
315 case MSIMODIFY_MERGE:
316 case MSIMODIFY_REPLACE:
317 case MSIMODIFY_SEEK:
318 case MSIMODIFY_VALIDATE:
319 case MSIMODIFY_VALIDATE_DELETE:
320 case MSIMODIFY_VALIDATE_FIELD:
321 case MSIMODIFY_VALIDATE_NEW:
322 r = ERROR_FUNCTION_FAILED;
323 break;
324
325 case MSIMODIFY_REFRESH:
326 r = ERROR_CALL_NOT_IMPLEMENTED;
327 break;
328
329 default:
330 WARN("%p %d %p %u - unknown mode\n", view, mode, rec, row );
331 r = ERROR_INVALID_PARAMETER;
332 break;
333 }
334
335 return r;
336 }
337
338 static UINT JOIN_delete( struct tagMSIVIEW *view )
339 {
340 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
341 struct list *item, *cursor;
342
343 TRACE("%p\n", jv );
344
345 LIST_FOR_EACH_SAFE(item, cursor, &jv->tables)
346 {
347 JOINTABLE* table = LIST_ENTRY(item, JOINTABLE, entry);
348
349 list_remove(&table->entry);
350 table->view->ops->delete(table->view);
351 table->view = NULL;
352 msi_free(table);
353 }
354
355 msi_free(jv);
356
357 return ERROR_SUCCESS;
358 }
359
360 static UINT JOIN_find_matching_rows( struct tagMSIVIEW *view, UINT col,
361 UINT val, UINT *row, MSIITERHANDLE *handle )
362 {
363 MSIJOINVIEW *jv = (MSIJOINVIEW*)view;
364 UINT i, row_value;
365
366 TRACE("%p, %d, %u, %p\n", view, col, val, *handle);
367
368 if (col == 0 || col > jv->columns)
369 return ERROR_INVALID_PARAMETER;
370
371 for (i = PtrToUlong(*handle); i < jv->rows; i++)
372 {
373 if (view->ops->fetch_int( view, i, col, &row_value ) != ERROR_SUCCESS)
374 continue;
375
376 if (row_value == val)
377 {
378 *row = i;
379 (*(UINT *)handle) = i + 1;
380 return ERROR_SUCCESS;
381 }
382 }
383
384 return ERROR_NO_MORE_ITEMS;
385 }
386
387 static UINT JOIN_sort(struct tagMSIVIEW *view, column_info *columns)
388 {
389 MSIJOINVIEW *jv = (MSIJOINVIEW *)view;
390 JOINTABLE *table;
391 UINT r;
392
393 TRACE("%p %p\n", view, columns);
394
395 LIST_FOR_EACH_ENTRY(table, &jv->tables, JOINTABLE, entry)
396 {
397 r = table->view->ops->sort(table->view, columns);
398 if (r != ERROR_SUCCESS)
399 return r;
400 }
401
402 return ERROR_SUCCESS;
403 }
404
405 static const MSIVIEWOPS join_ops =
406 {
407 JOIN_fetch_int,
408 JOIN_fetch_stream,
409 JOIN_get_row,
410 NULL,
411 NULL,
412 NULL,
413 JOIN_execute,
414 JOIN_close,
415 JOIN_get_dimensions,
416 JOIN_get_column_info,
417 JOIN_modify,
418 JOIN_delete,
419 JOIN_find_matching_rows,
420 NULL,
421 NULL,
422 NULL,
423 NULL,
424 JOIN_sort,
425 NULL,
426 };
427
428 UINT JOIN_CreateView( MSIDATABASE *db, MSIVIEW **view, LPWSTR tables )
429 {
430 MSIJOINVIEW *jv = NULL;
431 UINT r = ERROR_SUCCESS;
432 JOINTABLE *table;
433 LPWSTR ptr;
434
435 TRACE("%p (%s)\n", jv, debugstr_w(tables) );
436
437 jv = msi_alloc_zero( sizeof *jv );
438 if( !jv )
439 return ERROR_FUNCTION_FAILED;
440
441 /* fill the structure */
442 jv->view.ops = &join_ops;
443 jv->db = db;
444 jv->columns = 0;
445 jv->rows = 0;
446
447 list_init(&jv->tables);
448
449 while (*tables)
450 {
451 if ((ptr = strchrW(tables, ' ')))
452 *ptr = '\0';
453
454 table = msi_alloc(sizeof(JOINTABLE));
455 if (!table)
456 {
457 r = ERROR_OUTOFMEMORY;
458 goto end;
459 }
460
461 r = TABLE_CreateView( db, tables, &table->view );
462 if( r != ERROR_SUCCESS )
463 {
464 WARN("can't create table: %s\n", debugstr_w(tables));
465 msi_free(table);
466 r = ERROR_BAD_QUERY_SYNTAX;
467 goto end;
468 }
469
470 r = table->view->ops->get_dimensions( table->view, NULL,
471 &table->columns );
472 if( r != ERROR_SUCCESS )
473 {
474 ERR("can't get table dimensions\n");
475 goto end;
476 }
477
478 jv->columns += table->columns;
479
480 list_add_head( &jv->tables, &table->entry );
481
482 if (!ptr)
483 break;
484
485 tables = ptr + 1;
486 }
487
488 *view = &jv->view;
489 return ERROR_SUCCESS;
490
491 end:
492 jv->view.ops->delete( &jv->view );
493
494 return r;
495 }