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
, LPWSTR 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
, 0, 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 IStorage_DestroyElement(sv
->db
->storage
, 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
;