2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2007 James Hawkins
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
34 #include "wine/debug.h"
35 #include "wine/unicode.h"
37 WINE_DEFAULT_DEBUG_CHANNEL(msidb);
39 #define NUM_STREAMS_COLS 2
40 #define MAX_STREAM_NAME_LEN 62
42 typedef struct tabSTREAM
48 typedef struct tagMSISTREAMSVIEW
58 static BOOL streams_set_table_size(MSISTREAMSVIEW *sv, UINT size)
60 if (size >= sv->max_streams)
63 sv->streams = msi_realloc_zero(sv->streams, sv->max_streams * sizeof(STREAM *));
71 static STREAM *create_stream(MSISTREAMSVIEW *sv, LPCWSTR name, BOOL encoded, IStream *stm)
74 WCHAR decoded[MAX_STREAM_NAME_LEN];
76 stream = msi_alloc(sizeof(STREAM));
82 decode_streamname(name, decoded);
83 TRACE("stream -> %s %s\n", debugstr_w(name), debugstr_w(decoded));
87 stream->str_index = msi_addstringW(sv->db->strings, name, -1, 1, StringNonPersistent);
92 static UINT STREAMS_fetch_int(struct tagMSIVIEW *view, UINT row, UINT col, UINT *val)
94 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
96 TRACE("(%p, %d, %d, %p)\n", view, row, col, val);
99 return ERROR_INVALID_PARAMETER;
101 if (row >= sv->num_rows)
102 return ERROR_NO_MORE_ITEMS;
104 *val = sv->streams[row]->str_index;
106 return ERROR_SUCCESS;
109 static UINT STREAMS_fetch_stream(struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm)
111 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
113 TRACE("(%p, %d, %d, %p)\n", view, row, col, stm);
115 if (row >= sv->num_rows)
116 return ERROR_FUNCTION_FAILED;
118 IStream_AddRef(sv->streams[row]->stream);
119 *stm = sv->streams[row]->stream;
121 return ERROR_SUCCESS;
124 static UINT STREAMS_get_row( struct tagMSIVIEW *view, UINT row, MSIRECORD **rec )
126 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
128 TRACE("%p %d %p\n", sv, row, rec);
130 return msi_view_get_row( sv->db, view, row, rec );
133 static UINT STREAMS_set_row(struct tagMSIVIEW *view, UINT row, MSIRECORD *rec, UINT mask)
135 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
139 LPWSTR encname = NULL, name = NULL;
143 UINT r = ERROR_FUNCTION_FAILED;
145 TRACE("(%p, %d, %p, %08x)\n", view, row, rec, mask);
147 if (row > sv->num_rows)
148 return ERROR_FUNCTION_FAILED;
150 r = MSI_RecordGetIStream(rec, 2, &stm);
151 if (r != ERROR_SUCCESS)
154 hr = IStream_Stat(stm, &stat, STATFLAG_NONAME);
157 WARN("failed to stat stream: %08x\n", hr);
161 if (stat.cbSize.QuadPart >> 32)
163 WARN("stream too large\n");
167 data = msi_alloc(stat.cbSize.QuadPart);
171 hr = IStream_Read(stm, data, stat.cbSize.QuadPart, &count);
172 if (FAILED(hr) || count != stat.cbSize.QuadPart)
174 WARN("failed to read stream: %08x\n", hr);
178 name = strdupW(MSI_RecordGetString(rec, 1));
181 WARN("failed to retrieve stream name\n");
185 encname = encode_streamname(FALSE, name);
186 db_destroy_stream(sv->db, encname);
188 r = write_stream_data(sv->db->storage, name, data, count, FALSE);
189 if (r != ERROR_SUCCESS)
191 WARN("failed to write stream data: %d\n", r);
195 stream = create_stream(sv, name, FALSE, NULL);
199 hr = IStorage_OpenStream(sv->db->storage, encname, 0,
200 STGM_READ | STGM_SHARE_EXCLUSIVE, 0, &stream->stream);
203 WARN("failed to open stream: %08x\n", hr);
207 sv->streams[row] = stream;
214 IStream_Release(stm);
219 static UINT STREAMS_insert_row(struct tagMSIVIEW *view, MSIRECORD *rec, UINT row, BOOL temporary)
221 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
224 TRACE("(%p, %p, %d, %d)\n", view, rec, row, temporary);
226 if (!streams_set_table_size(sv, ++sv->num_rows))
227 return ERROR_FUNCTION_FAILED;
230 row = sv->num_rows - 1;
232 /* shift the rows to make room for the new row */
233 for (i = sv->num_rows - 1; i > row; i--)
235 sv->streams[i] = sv->streams[i - 1];
238 return STREAMS_set_row(view, row, rec, 0);
241 static UINT STREAMS_delete_row(struct tagMSIVIEW *view, UINT row)
243 FIXME("(%p %d): stub!\n", view, row);
244 return ERROR_SUCCESS;
247 static UINT STREAMS_execute(struct tagMSIVIEW *view, MSIRECORD *record)
249 TRACE("(%p, %p)\n", view, record);
250 return ERROR_SUCCESS;
253 static UINT STREAMS_close(struct tagMSIVIEW *view)
255 TRACE("(%p)\n", view);
256 return ERROR_SUCCESS;
259 static UINT STREAMS_get_dimensions(struct tagMSIVIEW *view, UINT *rows, UINT *cols)
261 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
263 TRACE("(%p, %p, %p)\n", view, rows, cols);
265 if (cols) *cols = NUM_STREAMS_COLS;
266 if (rows) *rows = sv->num_rows;
268 return ERROR_SUCCESS;
271 static UINT STREAMS_get_column_info(struct tagMSIVIEW *view, UINT n,
272 LPWSTR *name, UINT *type, BOOL *temporary,
275 LPCWSTR name_ptr = NULL;
277 static const WCHAR Name[] = {'N','a','m','e',0};
278 static const WCHAR Data[] = {'D','a','t','a',0};
279 static const WCHAR _Streams[] = {'_','S','t','r','e','a','m','s',0};
281 TRACE("(%p, %d, %p, %p, %p, %p)\n", view, n, name, type, temporary,
284 if (n == 0 || n > NUM_STREAMS_COLS)
285 return ERROR_INVALID_PARAMETER;
291 if (type) *type = MSITYPE_STRING | MSITYPE_VALID | MAX_STREAM_NAME_LEN;
296 if (type) *type = MSITYPE_STRING | MSITYPE_VALID | MSITYPE_NULLABLE;
302 *name = strdupW(name_ptr);
303 if (!*name) return ERROR_FUNCTION_FAILED;
308 *table_name = strdupW(_Streams);
312 return ERROR_FUNCTION_FAILED;
319 return ERROR_SUCCESS;
322 static UINT streams_find_row(MSISTREAMSVIEW *sv, MSIRECORD *rec, UINT *row)
327 str = MSI_RecordGetString(rec, 1);
328 msi_string2idW(sv->db->strings, str, &id);
330 for (i = 0; i < sv->num_rows; i++)
332 STREAMS_fetch_int(&sv->view, i, 1, &data);
337 return ERROR_SUCCESS;
341 return ERROR_FUNCTION_FAILED;
344 static UINT streams_modify_update(struct tagMSIVIEW *view, MSIRECORD *rec)
346 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
349 r = streams_find_row(sv, rec, &row);
350 if (r != ERROR_SUCCESS)
351 return ERROR_FUNCTION_FAILED;
353 return STREAMS_set_row(view, row, rec, 0);
356 static UINT streams_modify_assign(struct tagMSIVIEW *view, MSIRECORD *rec)
358 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
361 r = streams_find_row(sv, rec, &row);
362 if (r == ERROR_SUCCESS)
363 return streams_modify_update(view, rec);
365 return STREAMS_insert_row(view, rec, -1, FALSE);
368 static UINT STREAMS_modify(struct tagMSIVIEW *view, MSIMODIFY eModifyMode, MSIRECORD *rec, UINT row)
372 TRACE("%p %d %p\n", view, eModifyMode, rec);
376 case MSIMODIFY_ASSIGN:
377 r = streams_modify_assign(view, rec);
380 case MSIMODIFY_INSERT:
381 r = STREAMS_insert_row(view, rec, -1, FALSE);
384 case MSIMODIFY_UPDATE:
385 r = streams_modify_update(view, rec);
388 case MSIMODIFY_VALIDATE_NEW:
389 case MSIMODIFY_INSERT_TEMPORARY:
390 case MSIMODIFY_REFRESH:
391 case MSIMODIFY_REPLACE:
392 case MSIMODIFY_MERGE:
393 case MSIMODIFY_DELETE:
394 case MSIMODIFY_VALIDATE:
395 case MSIMODIFY_VALIDATE_FIELD:
396 case MSIMODIFY_VALIDATE_DELETE:
397 FIXME("%p %d %p - mode not implemented\n", view, eModifyMode, rec );
398 r = ERROR_CALL_NOT_IMPLEMENTED;
402 r = ERROR_INVALID_DATA;
408 static UINT STREAMS_delete(struct tagMSIVIEW *view)
410 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
413 TRACE("(%p)\n", view);
415 for (i = 0; i < sv->num_rows; i++)
419 if (sv->streams[i]->stream)
420 IStream_Release(sv->streams[i]->stream);
421 msi_free(sv->streams[i]);
425 msi_free(sv->streams);
428 return ERROR_SUCCESS;
431 static UINT STREAMS_find_matching_rows(struct tagMSIVIEW *view, UINT col,
432 UINT val, UINT *row, MSIITERHANDLE *handle)
434 MSISTREAMSVIEW *sv = (MSISTREAMSVIEW *)view;
435 UINT index = PtrToUlong(*handle);
437 TRACE("(%p, %d, %d, %p, %p)\n", view, col, val, row, handle);
439 if (col == 0 || col > NUM_STREAMS_COLS)
440 return ERROR_INVALID_PARAMETER;
442 while (index < sv->num_rows)
444 if (sv->streams[index]->str_index == val)
453 *handle = UlongToPtr(++index);
455 if (index > sv->num_rows)
456 return ERROR_NO_MORE_ITEMS;
458 return ERROR_SUCCESS;
461 static const MSIVIEWOPS streams_ops =
464 STREAMS_fetch_stream,
471 STREAMS_get_dimensions,
472 STREAMS_get_column_info,
475 STREAMS_find_matching_rows,
484 static INT add_streams_to_table(MSISTREAMSVIEW *sv)
486 IEnumSTATSTG *stgenum = NULL;
488 STREAM *stream = NULL;
490 UINT r, count = 0, size;
493 hr = IStorage_EnumElements(sv->db->storage, 0, NULL, 0, &stgenum);
498 sv->streams = msi_alloc_zero(sizeof(STREAM *));
505 hr = IEnumSTATSTG_Next(stgenum, 1, &stat, &size);
506 if (FAILED(hr) || !size)
509 if (stat.type != STGTY_STREAM)
511 CoTaskMemFree(stat.pwcsName);
515 /* table streams are not in the _Streams table */
516 if (*stat.pwcsName == 0x4840)
518 CoTaskMemFree(stat.pwcsName);
522 stream = create_stream(sv, stat.pwcsName, TRUE, NULL);
526 CoTaskMemFree(stat.pwcsName);
530 /* these streams appear to be unencoded */
531 if (*stat.pwcsName == 0x0005)
533 r = db_get_raw_stream(sv->db, stat.pwcsName, &stream->stream);
537 encname = encode_streamname(FALSE, stat.pwcsName);
538 r = db_get_raw_stream(sv->db, encname, &stream->stream);
541 CoTaskMemFree(stat.pwcsName);
543 if (r != ERROR_SUCCESS)
545 WARN("unable to get stream %u\n", r);
550 if (!streams_set_table_size(sv, ++count))
556 sv->streams[count - 1] = stream;
559 IEnumSTATSTG_Release(stgenum);
563 UINT STREAMS_CreateView(MSIDATABASE *db, MSIVIEW **view)
568 TRACE("(%p, %p)\n", db, view);
570 sv = msi_alloc(sizeof(MSISTREAMSVIEW));
572 return ERROR_FUNCTION_FAILED;
574 sv->view.ops = &streams_ops;
576 rows = add_streams_to_table(sv);
580 return ERROR_FUNCTION_FAILED;
584 *view = (MSIVIEW *)sv;
586 return ERROR_SUCCESS;