2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2008 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
36 #include "wine/debug.h"
38 WINE_DEFAULT_DEBUG_CHANNEL(msidb
);
40 #define NUM_STORAGES_COLS 2
41 #define MAX_STORAGES_NAME_LEN 62
43 typedef struct tabSTORAGE
50 typedef struct tagMSISTORAGESVIEW
60 static BOOL
storages_set_table_size(MSISTORAGESVIEW
*sv
, UINT size
)
62 if (size
>= sv
->max_storages
)
64 sv
->max_storages
*= 2;
65 sv
->storages
= msi_realloc(sv
->storages
, sv
->max_storages
* sizeof(STORAGE
*));
73 static STORAGE
*create_storage(MSISTORAGESVIEW
*sv
, LPWSTR name
, IStorage
*stg
)
77 storage
= msi_alloc(sizeof(STORAGE
));
81 storage
->name
= strdupW(name
);
88 storage
->str_index
= msi_addstringW(sv
->db
->strings
, 0, storage
->name
, -1, 1, StringNonPersistent
);
89 storage
->storage
= stg
;
92 IStorage_AddRef(storage
->storage
);
97 static UINT
STORAGES_fetch_int(struct tagMSIVIEW
*view
, UINT row
, UINT col
, UINT
*val
)
99 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
101 TRACE("(%p, %d, %d, %p)\n", view
, row
, col
, val
);
104 return ERROR_INVALID_PARAMETER
;
106 if (row
>= sv
->num_rows
)
107 return ERROR_NO_MORE_ITEMS
;
109 *val
= sv
->storages
[row
]->str_index
;
111 return ERROR_SUCCESS
;
114 static UINT
STORAGES_fetch_stream(struct tagMSIVIEW
*view
, UINT row
, UINT col
, IStream
**stm
)
116 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
118 TRACE("(%p, %d, %d, %p)\n", view
, row
, col
, stm
);
120 if (row
>= sv
->num_rows
)
121 return ERROR_FUNCTION_FAILED
;
123 return ERROR_INVALID_DATA
;
126 static UINT
STORAGES_get_row( struct tagMSIVIEW
*view
, UINT row
, MSIRECORD
**rec
)
128 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
130 FIXME("%p %d %p\n", sv
, row
, rec
);
132 return ERROR_CALL_NOT_IMPLEMENTED
;
135 static HRESULT
stream_to_storage(IStream
*stm
, IStorage
**stg
)
137 ILockBytes
*lockbytes
= NULL
;
142 ULARGE_INTEGER offset
;
144 hr
= IStream_Stat(stm
, &stat
, STATFLAG_NONAME
);
148 if (stat
.cbSize
.QuadPart
>> 32)
150 ERR("Storage is too large\n");
154 size
= stat
.cbSize
.QuadPart
;
155 data
= msi_alloc(size
);
157 return E_OUTOFMEMORY
;
159 hr
= IStream_Read(stm
, data
, size
, &read
);
160 if (FAILED(hr
) || read
!= size
)
163 hr
= CreateILockBytesOnHGlobal(NULL
, TRUE
, &lockbytes
);
167 ZeroMemory(&offset
, sizeof(ULARGE_INTEGER
));
168 hr
= ILockBytes_WriteAt(lockbytes
, offset
, data
, size
, &read
);
169 if (FAILED(hr
) || read
!= size
)
172 hr
= StgOpenStorageOnILockBytes(lockbytes
, NULL
,
173 STGM_READWRITE
| STGM_SHARE_DENY_NONE
,
180 if (lockbytes
) ILockBytes_Release(lockbytes
);
184 static UINT
STORAGES_set_row(struct tagMSIVIEW
*view
, UINT row
, MSIRECORD
*rec
, UINT mask
)
186 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
187 IStorage
*stg
, *substg
= NULL
;
191 UINT r
= ERROR_FUNCTION_FAILED
;
193 TRACE("(%p, %p)\n", view
, rec
);
195 if (row
> sv
->num_rows
)
196 return ERROR_FUNCTION_FAILED
;
198 r
= MSI_RecordGetIStream(rec
, 2, &stm
);
199 if (r
!= ERROR_SUCCESS
)
202 r
= stream_to_storage(stm
, &stg
);
203 if (r
!= ERROR_SUCCESS
)
205 IStream_Release(stm
);
209 name
= strdupW(MSI_RecordGetString(rec
, 1));
212 r
= ERROR_OUTOFMEMORY
;
216 hr
= IStorage_CreateStorage(sv
->db
->storage
, name
,
217 STGM_WRITE
| STGM_SHARE_EXCLUSIVE
,
221 r
= ERROR_FUNCTION_FAILED
;
225 hr
= IStorage_CopyTo(stg
, 0, NULL
, NULL
, substg
);
228 r
= ERROR_FUNCTION_FAILED
;
232 sv
->storages
[row
] = create_storage(sv
, name
, stg
);
233 if (!sv
->storages
[row
])
234 r
= ERROR_FUNCTION_FAILED
;
239 if (substg
) IStorage_Release(substg
);
240 IStorage_Release(stg
);
241 IStream_Release(stm
);
246 static UINT
STORAGES_insert_row(struct tagMSIVIEW
*view
, MSIRECORD
*rec
, UINT row
, BOOL temporary
)
248 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
250 if (!storages_set_table_size(sv
, ++sv
->num_rows
))
251 return ERROR_FUNCTION_FAILED
;
254 row
= sv
->num_rows
- 1;
256 /* FIXME have to readjust rows */
258 return STORAGES_set_row(view
, row
, rec
, 0);
261 static UINT
STORAGES_delete_row(struct tagMSIVIEW
*view
, UINT row
)
263 FIXME("(%p %d): stub!\n", view
, row
);
264 return ERROR_SUCCESS
;
267 static UINT
STORAGES_execute(struct tagMSIVIEW
*view
, MSIRECORD
*record
)
269 TRACE("(%p, %p)\n", view
, record
);
270 return ERROR_SUCCESS
;
273 static UINT
STORAGES_close(struct tagMSIVIEW
*view
)
275 TRACE("(%p)\n", view
);
276 return ERROR_SUCCESS
;
279 static UINT
STORAGES_get_dimensions(struct tagMSIVIEW
*view
, UINT
*rows
, UINT
*cols
)
281 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
283 TRACE("(%p, %p, %p)\n", view
, rows
, cols
);
285 if (cols
) *cols
= NUM_STORAGES_COLS
;
286 if (rows
) *rows
= sv
->num_rows
;
288 return ERROR_SUCCESS
;
291 static UINT
STORAGES_get_column_info(struct tagMSIVIEW
*view
, UINT n
,
292 LPWSTR
*name
, UINT
*type
, BOOL
*temporary
,
295 LPCWSTR name_ptr
= NULL
;
297 static const WCHAR Name
[] = {'N','a','m','e',0};
298 static const WCHAR Data
[] = {'D','a','t','a',0};
300 TRACE("(%p, %d, %p, %p, %p, %p)\n", view
, n
, name
, type
, temporary
,
303 if (n
== 0 || n
> NUM_STORAGES_COLS
)
304 return ERROR_INVALID_PARAMETER
;
310 if (type
) *type
= MSITYPE_STRING
| MAX_STORAGES_NAME_LEN
;
315 if (type
) *type
= MSITYPE_STRING
| MSITYPE_VALID
| MSITYPE_NULLABLE
;
321 *name
= strdupW(name_ptr
);
322 if (!*name
) return ERROR_FUNCTION_FAILED
;
328 return ERROR_SUCCESS
;
331 static UINT
storages_find_row(MSISTORAGESVIEW
*sv
, MSIRECORD
*rec
, UINT
*row
)
336 str
= MSI_RecordGetString(rec
, 1);
337 msi_string2idW(sv
->db
->strings
, str
, &id
);
339 for (i
= 0; i
< sv
->num_rows
; i
++)
341 STORAGES_fetch_int(&sv
->view
, i
, 1, &data
);
346 return ERROR_SUCCESS
;
350 return ERROR_FUNCTION_FAILED
;
353 static UINT
storages_modify_update(struct tagMSIVIEW
*view
, MSIRECORD
*rec
)
355 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
358 r
= storages_find_row(sv
, rec
, &row
);
359 if (r
!= ERROR_SUCCESS
)
360 return ERROR_FUNCTION_FAILED
;
362 return STORAGES_set_row(view
, row
, rec
, 0);
365 static UINT
storages_modify_assign(struct tagMSIVIEW
*view
, MSIRECORD
*rec
)
367 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
370 r
= storages_find_row(sv
, rec
, &row
);
371 if (r
== ERROR_SUCCESS
)
372 return storages_modify_update(view
, rec
);
374 return STORAGES_insert_row(view
, rec
, -1, FALSE
);
377 static UINT
STORAGES_modify(struct tagMSIVIEW
*view
, MSIMODIFY eModifyMode
, MSIRECORD
*rec
, UINT row
)
381 TRACE("%p %d %p\n", view
, eModifyMode
, rec
);
385 case MSIMODIFY_ASSIGN
:
386 r
= storages_modify_assign(view
, rec
);
389 case MSIMODIFY_INSERT
:
390 r
= STORAGES_insert_row(view
, rec
, -1, FALSE
);
393 case MSIMODIFY_UPDATE
:
394 r
= storages_modify_update(view
, rec
);
397 case MSIMODIFY_VALIDATE_NEW
:
398 case MSIMODIFY_INSERT_TEMPORARY
:
399 case MSIMODIFY_REFRESH
:
400 case MSIMODIFY_REPLACE
:
401 case MSIMODIFY_MERGE
:
402 case MSIMODIFY_DELETE
:
403 case MSIMODIFY_VALIDATE
:
404 case MSIMODIFY_VALIDATE_FIELD
:
405 case MSIMODIFY_VALIDATE_DELETE
:
406 FIXME("%p %d %p - mode not implemented\n", view
, eModifyMode
, rec
);
407 r
= ERROR_CALL_NOT_IMPLEMENTED
;
411 r
= ERROR_INVALID_DATA
;
417 static UINT
STORAGES_delete(struct tagMSIVIEW
*view
)
419 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
422 TRACE("(%p)\n", view
);
424 for (i
= 0; i
< sv
->num_rows
; i
++)
426 if (sv
->storages
[i
]->storage
)
427 IStorage_Release(sv
->storages
[i
]->storage
);
429 msi_free(sv
->storages
[i
]);
432 msi_free(sv
->storages
);
435 return ERROR_SUCCESS
;
438 static UINT
STORAGES_find_matching_rows(struct tagMSIVIEW
*view
, UINT col
,
439 UINT val
, UINT
*row
, MSIITERHANDLE
*handle
)
441 MSISTORAGESVIEW
*sv
= (MSISTORAGESVIEW
*)view
;
442 UINT index
= (UINT
)*handle
;
444 TRACE("(%d, %d): %d\n", *row
, col
, val
);
446 if (col
== 0 || col
> NUM_STORAGES_COLS
)
447 return ERROR_INVALID_PARAMETER
;
449 while (index
< sv
->num_rows
)
451 if (sv
->storages
[index
]->str_index
== val
)
460 *handle
= (MSIITERHANDLE
)++index
;
461 if (index
>= sv
->num_rows
)
462 return ERROR_NO_MORE_ITEMS
;
464 return ERROR_SUCCESS
;
467 static const MSIVIEWOPS storages_ops
=
470 STORAGES_fetch_stream
,
477 STORAGES_get_dimensions
,
478 STORAGES_get_column_info
,
481 STORAGES_find_matching_rows
,
490 static INT
add_storages_to_table(MSISTORAGESVIEW
*sv
)
492 STORAGE
*storage
= NULL
;
493 IEnumSTATSTG
*stgenum
= NULL
;
496 UINT count
= 0, size
;
498 hr
= IStorage_EnumElements(sv
->db
->storage
, 0, NULL
, 0, &stgenum
);
502 sv
->max_storages
= 1;
503 sv
->storages
= msi_alloc(sizeof(STORAGE
*));
510 hr
= IEnumSTATSTG_Next(stgenum
, 1, &stat
, &size
);
511 if (FAILED(hr
) || !size
)
514 if (stat
.type
!= STGTY_STORAGE
)
516 CoTaskMemFree(stat
.pwcsName
);
520 TRACE("enumerated storage %s\n", debugstr_w(stat
.pwcsName
));
522 storage
= create_storage(sv
, stat
.pwcsName
, NULL
);
526 CoTaskMemFree(stat
.pwcsName
);
530 IStorage_OpenStorage(sv
->db
->storage
, stat
.pwcsName
, NULL
,
531 STGM_READ
| STGM_SHARE_EXCLUSIVE
, NULL
, 0,
533 CoTaskMemFree(stat
.pwcsName
);
535 if (!storages_set_table_size(sv
, ++count
))
541 sv
->storages
[count
- 1] = storage
;
544 IEnumSTATSTG_Release(stgenum
);
548 UINT
STORAGES_CreateView(MSIDATABASE
*db
, MSIVIEW
**view
)
553 TRACE("(%p, %p)\n", db
, view
);
555 sv
= msi_alloc(sizeof(MSISTORAGESVIEW
));
557 return ERROR_FUNCTION_FAILED
;
559 sv
->view
.ops
= &storages_ops
;
562 rows
= add_storages_to_table(sv
);
566 return ERROR_FUNCTION_FAILED
;
570 *view
= (MSIVIEW
*)sv
;
572 return ERROR_SUCCESS
;