2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2002,2003,2004,2005 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
24 #define NONAMELESSUNION
30 #include "wine/debug.h"
31 #include "wine/unicode.h"
37 #include "msiserver.h"
41 WINE_DEFAULT_DEBUG_CHANNEL(msi
);
43 DEFINE_GUID( CLSID_MsiDatabase
, 0x000c1084, 0x0000, 0x0000,
44 0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
45 DEFINE_GUID( CLSID_MsiPatch
, 0x000c1086, 0x0000, 0x0000,
46 0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
51 * An .msi file is a structured storage file.
52 * It contains a number of streams.
53 * A stream for each table in the database.
54 * Two streams for the string table in the database.
55 * Any binary data in a table is a reference to a stream.
58 static VOID
MSI_CloseDatabase( MSIOBJECTHDR
*arg
)
60 MSIDATABASE
*db
= (MSIDATABASE
*) arg
;
63 free_cached_tables( db
);
64 msi_free_transforms( db
);
65 msi_destroy_stringtable( db
->strings
);
66 IStorage_Release( db
->storage
);
69 DeleteFileW( db
->deletefile
);
70 msi_free( db
->deletefile
);
74 UINT
MSI_OpenDatabaseW(LPCWSTR szDBPath
, LPCWSTR szPersist
, MSIDATABASE
**pdb
)
78 MSIDATABASE
*db
= NULL
;
79 UINT ret
= ERROR_FUNCTION_FAILED
;
80 LPCWSTR szMode
, save_path
;
85 static const WCHAR backslash
[] = {'\\',0};
86 static const WCHAR szTables
[] = { '_','T','a','b','l','e','s',0 };
88 TRACE("%s %s\n",debugstr_w(szDBPath
),debugstr_w(szPersist
) );
91 return ERROR_INVALID_PARAMETER
;
93 if (szPersist
- MSIDBOPEN_PATCHFILE
>= MSIDBOPEN_READONLY
&&
94 szPersist
- MSIDBOPEN_PATCHFILE
<= MSIDBOPEN_CREATEDIRECT
)
96 TRACE("Database is a patch\n");
97 szPersist
-= MSIDBOPEN_PATCHFILE
;
100 save_path
= szDBPath
;
102 if( HIWORD( szPersist
) )
104 if (!CopyFileW( szDBPath
, szPersist
, FALSE
))
105 return ERROR_OPEN_FAILED
;
107 szDBPath
= szPersist
;
108 szPersist
= MSIDBOPEN_TRANSACT
;
112 if( szPersist
== MSIDBOPEN_READONLY
)
114 r
= StgOpenStorage( szDBPath
, NULL
,
115 STGM_DIRECT
|STGM_READ
|STGM_SHARE_DENY_WRITE
, NULL
, 0, &stg
);
117 else if( szPersist
== MSIDBOPEN_CREATE
|| szPersist
== MSIDBOPEN_CREATEDIRECT
)
119 /* FIXME: MSIDBOPEN_CREATE should case STGM_TRANSACTED flag to be
121 r
= StgCreateDocfile( szDBPath
,
122 STGM_CREATE
|STGM_DIRECT
|STGM_READWRITE
|STGM_SHARE_EXCLUSIVE
, 0, &stg
);
123 if( r
== ERROR_SUCCESS
)
125 IStorage_SetClass( stg
, &CLSID_MsiDatabase
);
126 /* create the _Tables stream */
127 r
= write_stream_data(stg
, szTables
, NULL
, 0, TRUE
);
129 r
= msi_init_string_table( stg
);
133 else if( szPersist
== MSIDBOPEN_TRANSACT
)
135 /* FIXME: MSIDBOPEN_TRANSACT should case STGM_TRANSACTED flag to be
137 r
= StgOpenStorage( szDBPath
, NULL
,
138 STGM_DIRECT
|STGM_READWRITE
|STGM_SHARE_EXCLUSIVE
, NULL
, 0, &stg
);
140 else if( szPersist
== MSIDBOPEN_DIRECT
)
142 r
= StgOpenStorage( szDBPath
, NULL
,
143 STGM_DIRECT
|STGM_READWRITE
|STGM_SHARE_EXCLUSIVE
, NULL
, 0, &stg
);
147 ERR("unknown flag %p\n",szPersist
);
148 return ERROR_INVALID_PARAMETER
;
151 if( FAILED( r
) || !stg
)
153 FIXME("open failed r = %08x for %s\n", r
, debugstr_w(szDBPath
));
154 return ERROR_FUNCTION_FAILED
;
157 r
= IStorage_Stat( stg
, &stat
, STATFLAG_NONAME
);
160 FIXME("Failed to stat storage\n");
164 if ( !IsEqualGUID( &stat
.clsid
, &CLSID_MsiDatabase
) &&
165 !IsEqualGUID( &stat
.clsid
, &CLSID_MsiPatch
) )
167 ERR("storage GUID is not a MSI database GUID %s\n",
168 debugstr_guid(&stat
.clsid
) );
172 db
= alloc_msiobject( MSIHANDLETYPE_DATABASE
, sizeof (MSIDATABASE
),
176 FIXME("Failed to allocate a handle\n");
180 if (!strchrW( save_path
, '\\' ))
182 GetCurrentDirectoryW( MAX_PATH
, path
);
183 lstrcatW( path
, backslash
);
184 lstrcatW( path
, save_path
);
187 lstrcpyW( path
, save_path
);
189 db
->path
= strdupW( path
);
191 if( TRACE_ON( msi
) )
192 enum_stream_names( stg
);
197 db
->deletefile
= strdupW( szDBPath
);
199 db
->deletefile
= NULL
;
200 list_init( &db
->tables
);
201 list_init( &db
->transforms
);
203 db
->strings
= msi_load_string_table( stg
, &db
->bytes_per_strref
);
207 msi_table_set_strref( db
->bytes_per_strref
);
210 msiobj_addref( &db
->hdr
);
211 IStorage_AddRef( stg
);
216 msiobj_release( &db
->hdr
);
218 IStorage_Release( stg
);
223 UINT WINAPI
MsiOpenDatabaseW(LPCWSTR szDBPath
, LPCWSTR szPersist
, MSIHANDLE
*phDB
)
228 TRACE("%s %s %p\n",debugstr_w(szDBPath
),debugstr_w(szPersist
), phDB
);
230 ret
= MSI_OpenDatabaseW( szDBPath
, szPersist
, &db
);
231 if( ret
== ERROR_SUCCESS
)
233 *phDB
= alloc_msihandle( &db
->hdr
);
235 ret
= ERROR_NOT_ENOUGH_MEMORY
;
236 msiobj_release( &db
->hdr
);
242 UINT WINAPI
MsiOpenDatabaseA(LPCSTR szDBPath
, LPCSTR szPersist
, MSIHANDLE
*phDB
)
244 HRESULT r
= ERROR_FUNCTION_FAILED
;
245 LPWSTR szwDBPath
= NULL
, szwPersist
= NULL
;
247 TRACE("%s %s %p\n", debugstr_a(szDBPath
), debugstr_a(szPersist
), phDB
);
251 szwDBPath
= strdupAtoW( szDBPath
);
256 if( HIWORD(szPersist
) )
258 szwPersist
= strdupAtoW( szPersist
);
263 szwPersist
= (LPWSTR
)(DWORD_PTR
)szPersist
;
265 r
= MsiOpenDatabaseW( szwDBPath
, szwPersist
, phDB
);
268 if( HIWORD(szPersist
) )
269 msi_free( szwPersist
);
270 msi_free( szwDBPath
);
275 static LPWSTR
msi_read_text_archive(LPCWSTR path
)
280 DWORD read
, size
= 0;
282 file
= CreateFileW( path
, GENERIC_READ
, FILE_SHARE_READ
, NULL
, OPEN_EXISTING
, 0, NULL
);
283 if (file
== INVALID_HANDLE_VALUE
)
286 size
= GetFileSize( file
, NULL
);
287 data
= msi_alloc( size
+ 1 );
291 if (!ReadFile( file
, data
, size
, &read
, NULL
))
295 wdata
= strdupAtoW( data
);
303 static void msi_parse_line(LPWSTR
*line
, LPWSTR
**entries
, DWORD
*num_entries
)
305 LPWSTR ptr
= *line
, save
;
310 /* stay on this line */
311 while (*ptr
&& *ptr
!= '\n')
313 /* entries are separated by tabs */
320 *entries
= msi_alloc(count
* sizeof(LPWSTR
));
324 /* store pointers into the data */
325 for (i
= 0, ptr
= *line
; i
< count
; i
++)
327 while (*ptr
&& *ptr
== '\r') ptr
++;
330 while (*ptr
&& *ptr
!= '\t' && *ptr
!= '\n' && *ptr
!= '\r') ptr
++;
332 /* NULL-separate the data */
333 if (*ptr
== '\n' || *ptr
== '\r')
335 while (*ptr
== '\n' || *ptr
== '\r')
341 (*entries
)[i
] = save
;
344 /* move to the next line if there's more, else EOF */
348 *num_entries
= count
;
351 static LPWSTR
msi_build_createsql_prelude(LPWSTR table
)
356 static const WCHAR create_fmt
[] = {'C','R','E','A','T','E',' ','T','A','B','L','E',' ','`','%','s','`',' ','(',' ',0};
358 size
= sizeof(create_fmt
)/sizeof(create_fmt
[0]) + lstrlenW(table
) - 2;
359 prelude
= msi_alloc(size
* sizeof(WCHAR
));
363 sprintfW(prelude
, create_fmt
, table
);
367 static LPWSTR
msi_build_createsql_columns(LPWSTR
*columns_data
, LPWSTR
*types
, DWORD num_columns
)
371 DWORD sql_size
= 1, i
, len
;
372 WCHAR expanded
[128], *ptr
;
373 WCHAR size
[10], comma
[2], extra
[30];
375 static const WCHAR column_fmt
[] = {'`','%','s','`',' ','%','s','%','s','%','s','%','s',' ',0};
376 static const WCHAR size_fmt
[] = {'(','%','s',')',0};
377 static const WCHAR type_char
[] = {'C','H','A','R',0};
378 static const WCHAR type_int
[] = {'I','N','T',0};
379 static const WCHAR type_long
[] = {'L','O','N','G',0};
380 static const WCHAR type_notnull
[] = {' ','N','O','T',' ','N','U','L','L',0};
381 static const WCHAR localizable
[] = {' ','L','O','C','A','L','I','Z','A','B','L','E',0};
383 columns
= msi_alloc_zero(sql_size
* sizeof(WCHAR
));
387 for (i
= 0; i
< num_columns
; i
++)
390 comma
[1] = size
[0] = extra
[0] = '\0';
392 if (i
== num_columns
- 1)
404 lstrcpyW(extra
, type_notnull
);
406 lstrcatW(extra
, localizable
);
408 sprintfW(size
, size_fmt
, ptr
);
411 lstrcpyW(extra
, type_notnull
);
414 sprintfW(size
, size_fmt
, ptr
);
417 lstrcpyW(extra
, type_notnull
);
425 ERR("Unknown type: %c\n", types
[i
][0]);
430 sprintfW(expanded
, column_fmt
, columns_data
[i
], type
, size
, extra
, comma
);
431 sql_size
+= lstrlenW(expanded
);
433 p
= msi_realloc(columns
, sql_size
* sizeof(WCHAR
));
441 lstrcatW(columns
, expanded
);
447 static LPWSTR
msi_build_createsql_postlude(LPWSTR
*primary_keys
, DWORD num_keys
)
449 LPWSTR postlude
, keys
, ptr
;
450 DWORD size
, key_size
, i
;
452 static const WCHAR key_fmt
[] = {'`','%','s','`',',',' ',0};
453 static const WCHAR postlude_fmt
[] = {'P','R','I','M','A','R','Y',' ','K','E','Y',' ','%','s',')',0};
455 for (i
= 0, size
= 1; i
< num_keys
; i
++)
456 size
+= lstrlenW(key_fmt
) + lstrlenW(primary_keys
[i
]) - 2;
458 keys
= msi_alloc(size
* sizeof(WCHAR
));
462 for (i
= 0, ptr
= keys
; i
< num_keys
; i
++)
464 key_size
= lstrlenW(key_fmt
) + lstrlenW(primary_keys
[i
]) -2;
465 sprintfW(ptr
, key_fmt
, primary_keys
[i
]);
469 /* remove final ', ' */
472 size
= lstrlenW(postlude_fmt
) + size
- 1;
473 postlude
= msi_alloc(size
* sizeof(WCHAR
));
477 sprintfW(postlude
, postlude_fmt
, keys
);
484 static UINT
msi_add_table_to_db(MSIDATABASE
*db
, LPWSTR
*columns
, LPWSTR
*types
, LPWSTR
*labels
, DWORD num_labels
, DWORD num_columns
)
490 LPWSTR prelude
, columns_sql
, postlude
;
492 prelude
= msi_build_createsql_prelude(labels
[0]);
493 columns_sql
= msi_build_createsql_columns(columns
, types
, num_columns
);
494 postlude
= msi_build_createsql_postlude(labels
+ 1, num_labels
- 1); /* skip over table name */
496 if (!prelude
|| !columns_sql
|| !postlude
)
497 return ERROR_OUTOFMEMORY
;
499 size
= lstrlenW(prelude
) + lstrlenW(columns_sql
) + lstrlenW(postlude
) + 1;
500 create_sql
= msi_alloc(size
* sizeof(WCHAR
));
502 return ERROR_OUTOFMEMORY
;
504 lstrcpyW(create_sql
, prelude
);
505 lstrcatW(create_sql
, columns_sql
);
506 lstrcatW(create_sql
, postlude
);
509 msi_free(columns_sql
);
512 r
= MSI_DatabaseOpenViewW( db
, create_sql
, &view
);
513 msi_free(create_sql
);
515 if (r
!= ERROR_SUCCESS
)
518 r
= MSI_ViewExecute(view
, NULL
);
520 msiobj_release(&view
->hdr
);
525 static UINT
construct_record(DWORD num_columns
, LPWSTR
*types
,
526 LPWSTR
*data
, MSIRECORD
**rec
)
530 *rec
= MSI_CreateRecord(num_columns
);
532 return ERROR_OUTOFMEMORY
;
534 for (i
= 0; i
< num_columns
; i
++)
538 case 'L': case 'l': case 'S': case 's':
539 MSI_RecordSetStringW(*rec
, i
+ 1, data
[i
]);
543 MSI_RecordSetInteger(*rec
, i
+ 1, atoiW(data
[i
]));
546 ERR("Unhandled column type: %c\n", types
[i
][0]);
547 msiobj_release(&(*rec
)->hdr
);
548 return ERROR_FUNCTION_FAILED
;
552 return ERROR_SUCCESS
;
555 static UINT
msi_add_records_to_table(MSIDATABASE
*db
, LPWSTR
*columns
, LPWSTR
*types
,
556 LPWSTR
*labels
, LPWSTR
**records
,
557 int num_columns
, int num_records
)
565 static const WCHAR select
[] = {
566 'S','E','L','E','C','T',' ','*',' ',
567 'F','R','O','M',' ','`','%','s','`',0
570 size
= lstrlenW(select
) + lstrlenW(labels
[0]) - 1;
571 query
= msi_alloc(size
* sizeof(WCHAR
));
573 return ERROR_OUTOFMEMORY
;
575 sprintfW(query
, select
, labels
[0]);
577 r
= MSI_DatabaseOpenViewW(db
, query
, &view
);
579 if (r
!= ERROR_SUCCESS
)
582 while (MSI_ViewFetch(view
, &rec
) != ERROR_NO_MORE_ITEMS
)
584 r
= MSI_ViewModify(view
, MSIMODIFY_DELETE
, rec
);
585 if (r
!= ERROR_SUCCESS
)
589 for (i
= 0; i
< num_records
; i
++)
591 r
= construct_record(num_columns
, types
, records
[i
], &rec
);
592 if (r
!= ERROR_SUCCESS
)
595 r
= MSI_ViewModify(view
, MSIMODIFY_INSERT
, rec
);
596 if (r
!= ERROR_SUCCESS
)
598 msiobj_release(&rec
->hdr
);
602 msiobj_release(&rec
->hdr
);
606 msiobj_release(&view
->hdr
);
610 UINT
MSI_DatabaseImport(MSIDATABASE
*db
, LPCWSTR folder
, LPCWSTR file
)
614 DWORD num_labels
, num_types
;
615 DWORD num_columns
, num_records
= 0;
616 LPWSTR
*columns
, *types
, *labels
;
617 LPWSTR path
, ptr
, data
;
618 LPWSTR
**records
= NULL
;
619 LPWSTR
**temp_records
;
621 static const WCHAR backslash
[] = {'\\',0};
623 TRACE("%p %s %s\n", db
, debugstr_w(folder
), debugstr_w(file
) );
625 if( folder
== NULL
|| file
== NULL
)
626 return ERROR_INVALID_PARAMETER
;
628 len
= lstrlenW(folder
) + lstrlenW(backslash
) + lstrlenW(file
) + 1;
629 path
= msi_alloc( len
* sizeof(WCHAR
) );
631 return ERROR_OUTOFMEMORY
;
633 lstrcpyW( path
, folder
);
634 lstrcatW( path
, backslash
);
635 lstrcatW( path
, file
);
637 data
= msi_read_text_archive( path
);
640 msi_parse_line( &ptr
, &columns
, &num_columns
);
641 msi_parse_line( &ptr
, &types
, &num_types
);
642 msi_parse_line( &ptr
, &labels
, &num_labels
);
644 if (num_columns
!= num_types
)
646 r
= ERROR_FUNCTION_FAILED
;
650 records
= msi_alloc(sizeof(LPWSTR
*));
653 r
= ERROR_OUTOFMEMORY
;
657 /* read in the table records */
660 msi_parse_line( &ptr
, &records
[num_records
], NULL
);
663 temp_records
= msi_realloc(records
, (num_records
+ 1) * sizeof(LPWSTR
*));
666 r
= ERROR_OUTOFMEMORY
;
669 records
= temp_records
;
672 if (!TABLE_Exists(db
, labels
[0]))
674 r
= msi_add_table_to_db( db
, columns
, types
, labels
, num_labels
, num_columns
);
675 if (r
!= ERROR_SUCCESS
)
677 r
= ERROR_FUNCTION_FAILED
;
682 r
= msi_add_records_to_table( db
, columns
, types
, labels
, records
, num_columns
, num_records
);
691 for (i
= 0; i
< num_records
; i
++)
692 msi_free(records
[i
]);
699 UINT WINAPI
MsiDatabaseImportW(MSIHANDLE handle
, LPCWSTR szFolder
, LPCWSTR szFilename
)
704 TRACE("%lx %s %s\n",handle
,debugstr_w(szFolder
), debugstr_w(szFilename
));
706 db
= msihandle2msiinfo( handle
, MSIHANDLETYPE_DATABASE
);
709 IWineMsiRemoteDatabase
*remote_database
;
711 remote_database
= (IWineMsiRemoteDatabase
*)msi_get_remote( handle
);
712 if ( !remote_database
)
713 return ERROR_INVALID_HANDLE
;
715 IWineMsiRemoteDatabase_Release( remote_database
);
716 WARN("MsiDatabaseImport not allowed during a custom action!\n");
718 return ERROR_SUCCESS
;
721 r
= MSI_DatabaseImport( db
, szFolder
, szFilename
);
722 msiobj_release( &db
->hdr
);
726 UINT WINAPI
MsiDatabaseImportA( MSIHANDLE handle
,
727 LPCSTR szFolder
, LPCSTR szFilename
)
729 LPWSTR path
= NULL
, file
= NULL
;
730 UINT r
= ERROR_OUTOFMEMORY
;
732 TRACE("%lx %s %s\n", handle
, debugstr_a(szFolder
), debugstr_a(szFilename
));
736 path
= strdupAtoW( szFolder
);
743 file
= strdupAtoW( szFilename
);
748 r
= MsiDatabaseImportW( handle
, path
, file
);
757 static UINT
msi_export_record( HANDLE handle
, MSIRECORD
*row
, UINT start
)
759 UINT i
, count
, len
, r
= ERROR_SUCCESS
;
765 buffer
= msi_alloc( len
);
767 return ERROR_OUTOFMEMORY
;
769 count
= MSI_RecordGetFieldCount( row
);
770 for ( i
=start
; i
<=count
; i
++ )
773 r
= MSI_RecordGetStringA( row
, i
, buffer
, &sz
);
774 if (r
== ERROR_MORE_DATA
)
776 char *p
= msi_realloc( buffer
, sz
+ 1 );
783 r
= MSI_RecordGetStringA( row
, i
, buffer
, &sz
);
784 if (r
!= ERROR_SUCCESS
)
787 if (!WriteFile( handle
, buffer
, sz
, &sz
, NULL
))
789 r
= ERROR_FUNCTION_FAILED
;
793 sep
= (i
< count
) ? "\t" : "\r\n";
794 if (!WriteFile( handle
, sep
, strlen(sep
), &sz
, NULL
))
796 r
= ERROR_FUNCTION_FAILED
;
804 static UINT
msi_export_row( MSIRECORD
*row
, void *arg
)
806 return msi_export_record( arg
, row
, 1 );
809 static UINT
msi_export_forcecodepage( HANDLE handle
)
813 static const char data
[] = "\r\n\r\n0\t_ForceCodepage\r\n";
815 FIXME("Read the codepage from the strings table!\n");
817 sz
= lstrlenA(data
) + 1;
818 if (!WriteFile(handle
, data
, sz
, &sz
, NULL
))
819 return ERROR_FUNCTION_FAILED
;
821 return ERROR_SUCCESS
;
824 UINT
MSI_DatabaseExport( MSIDATABASE
*db
, LPCWSTR table
,
825 LPCWSTR folder
, LPCWSTR file
)
827 static const WCHAR query
[] = {
828 's','e','l','e','c','t',' ','*',' ','f','r','o','m',' ','%','s',0 };
829 static const WCHAR szbs
[] = { '\\', 0 };
830 static const WCHAR forcecodepage
[] = {
831 '_','F','o','r','c','e','C','o','d','e','p','a','g','e',0 };
832 MSIRECORD
*rec
= NULL
;
833 MSIQUERY
*view
= NULL
;
838 TRACE("%p %s %s %s\n", db
, debugstr_w(table
),
839 debugstr_w(folder
), debugstr_w(file
) );
841 if( folder
== NULL
|| file
== NULL
)
842 return ERROR_INVALID_PARAMETER
;
844 len
= lstrlenW(folder
) + lstrlenW(file
) + 2;
845 filename
= msi_alloc(len
* sizeof (WCHAR
));
847 return ERROR_OUTOFMEMORY
;
849 lstrcpyW( filename
, folder
);
850 lstrcatW( filename
, szbs
);
851 lstrcatW( filename
, file
);
853 handle
= CreateFileW( filename
, GENERIC_READ
| GENERIC_WRITE
, 0,
854 NULL
, CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
855 msi_free( filename
);
856 if (handle
== INVALID_HANDLE_VALUE
)
857 return ERROR_FUNCTION_FAILED
;
859 if (!lstrcmpW( table
, forcecodepage
))
861 r
= msi_export_forcecodepage( handle
);
865 r
= MSI_OpenQuery( db
, &view
, query
, table
);
866 if (r
== ERROR_SUCCESS
)
868 /* write out row 1, the column names */
869 r
= MSI_ViewGetColumnInfo(view
, MSICOLINFO_NAMES
, &rec
);
870 if (r
== ERROR_SUCCESS
)
872 msi_export_record( handle
, rec
, 1 );
873 msiobj_release( &rec
->hdr
);
876 /* write out row 2, the column types */
877 r
= MSI_ViewGetColumnInfo(view
, MSICOLINFO_TYPES
, &rec
);
878 if (r
== ERROR_SUCCESS
)
880 msi_export_record( handle
, rec
, 1 );
881 msiobj_release( &rec
->hdr
);
884 /* write out row 3, the table name + keys */
885 r
= MSI_DatabaseGetPrimaryKeys( db
, table
, &rec
);
886 if (r
== ERROR_SUCCESS
)
888 MSI_RecordSetStringW( rec
, 0, table
);
889 msi_export_record( handle
, rec
, 0 );
890 msiobj_release( &rec
->hdr
);
893 /* write out row 4 onwards, the data */
894 r
= MSI_IterateRecords( view
, 0, msi_export_row
, handle
);
895 msiobj_release( &view
->hdr
);
899 CloseHandle( handle
);
903 /***********************************************************************
904 * MsiExportDatabaseW [MSI.@]
906 * Writes a file containing the table data as tab separated ASCII.
908 * The format is as follows:
910 * row1 : colname1 <tab> colname2 <tab> .... colnameN <cr> <lf>
911 * row2 : coltype1 <tab> coltype2 <tab> .... coltypeN <cr> <lf>
912 * row3 : tablename <tab> key1 <tab> key2 <tab> ... keyM <cr> <lf>
914 * Followed by the data, starting at row 1 with one row per line
916 * row4 : data <tab> data <tab> data <tab> ... data <cr> <lf>
918 UINT WINAPI
MsiDatabaseExportW( MSIHANDLE handle
, LPCWSTR szTable
,
919 LPCWSTR szFolder
, LPCWSTR szFilename
)
924 TRACE("%lx %s %s %s\n", handle
, debugstr_w(szTable
),
925 debugstr_w(szFolder
), debugstr_w(szFilename
));
927 db
= msihandle2msiinfo( handle
, MSIHANDLETYPE_DATABASE
);
930 IWineMsiRemoteDatabase
*remote_database
;
932 remote_database
= (IWineMsiRemoteDatabase
*)msi_get_remote( handle
);
933 if ( !remote_database
)
934 return ERROR_INVALID_HANDLE
;
936 IWineMsiRemoteDatabase_Release( remote_database
);
937 WARN("MsiDatabaseExport not allowed during a custom action!\n");
939 return ERROR_SUCCESS
;
942 r
= MSI_DatabaseExport( db
, szTable
, szFolder
, szFilename
);
943 msiobj_release( &db
->hdr
);
947 UINT WINAPI
MsiDatabaseExportA( MSIHANDLE handle
, LPCSTR szTable
,
948 LPCSTR szFolder
, LPCSTR szFilename
)
950 LPWSTR path
= NULL
, file
= NULL
, table
= NULL
;
951 UINT r
= ERROR_OUTOFMEMORY
;
953 TRACE("%lx %s %s %s\n", handle
, debugstr_a(szTable
),
954 debugstr_a(szFolder
), debugstr_a(szFilename
));
958 table
= strdupAtoW( szTable
);
965 path
= strdupAtoW( szFolder
);
972 file
= strdupAtoW( szFilename
);
977 r
= MsiDatabaseExportW( handle
, table
, path
, file
);
987 MSIDBSTATE WINAPI
MsiGetDatabaseState( MSIHANDLE handle
)
989 MSIDBSTATE ret
= MSIDBSTATE_READ
;
992 TRACE("%ld\n", handle
);
994 db
= msihandle2msiinfo( handle
, MSIHANDLETYPE_DATABASE
);
997 IWineMsiRemoteDatabase
*remote_database
;
999 remote_database
= (IWineMsiRemoteDatabase
*)msi_get_remote( handle
);
1000 if ( !remote_database
)
1001 return MSIDBSTATE_ERROR
;
1003 IWineMsiRemoteDatabase_Release( remote_database
);
1004 WARN("MsiGetDatabaseState not allowed during a custom action!\n");
1006 return MSIDBSTATE_READ
;
1009 if (db
->mode
!= MSIDBOPEN_READONLY
)
1010 ret
= MSIDBSTATE_WRITE
;
1011 msiobj_release( &db
->hdr
);
1016 typedef struct _msi_remote_database_impl
{
1017 const IWineMsiRemoteDatabaseVtbl
*lpVtbl
;
1020 } msi_remote_database_impl
;
1022 static inline msi_remote_database_impl
* mrd_from_IWineMsiRemoteDatabase( IWineMsiRemoteDatabase
* iface
)
1024 return (msi_remote_database_impl
*)iface
;
1027 static HRESULT WINAPI
mrd_QueryInterface( IWineMsiRemoteDatabase
*iface
,
1028 REFIID riid
,LPVOID
*ppobj
)
1030 if( IsEqualCLSID( riid
, &IID_IUnknown
) ||
1031 IsEqualCLSID( riid
, &IID_IWineMsiRemoteDatabase
) )
1033 IUnknown_AddRef( iface
);
1038 return E_NOINTERFACE
;
1041 static ULONG WINAPI
mrd_AddRef( IWineMsiRemoteDatabase
*iface
)
1043 msi_remote_database_impl
* This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1045 return InterlockedIncrement( &This
->refs
);
1048 static ULONG WINAPI
mrd_Release( IWineMsiRemoteDatabase
*iface
)
1050 msi_remote_database_impl
* This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1053 r
= InterlockedDecrement( &This
->refs
);
1056 MsiCloseHandle( This
->database
);
1062 static HRESULT WINAPI
mrd_IsTablePersistent( IWineMsiRemoteDatabase
*iface
,
1063 BSTR table
, MSICONDITION
*persistent
)
1065 msi_remote_database_impl
*This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1066 *persistent
= MsiDatabaseIsTablePersistentW(This
->database
, (LPWSTR
)table
);
1070 static HRESULT WINAPI
mrd_GetPrimaryKeys( IWineMsiRemoteDatabase
*iface
,
1071 BSTR table
, MSIHANDLE
*keys
)
1073 msi_remote_database_impl
*This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1074 UINT r
= MsiDatabaseGetPrimaryKeysW(This
->database
, (LPWSTR
)table
, keys
);
1075 return HRESULT_FROM_WIN32(r
);
1078 static HRESULT WINAPI
mrd_GetSummaryInformation( IWineMsiRemoteDatabase
*iface
,
1079 UINT updatecount
, MSIHANDLE
*suminfo
)
1081 msi_remote_database_impl
*This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1082 UINT r
= MsiGetSummaryInformationW(This
->database
, NULL
, updatecount
, suminfo
);
1083 return HRESULT_FROM_WIN32(r
);
1086 static HRESULT WINAPI
mrd_OpenView( IWineMsiRemoteDatabase
*iface
,
1087 BSTR query
, MSIHANDLE
*view
)
1089 msi_remote_database_impl
*This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1090 UINT r
= MsiDatabaseOpenViewW(This
->database
, (LPWSTR
)query
, view
);
1091 return HRESULT_FROM_WIN32(r
);
1094 static HRESULT WINAPI
mrd_SetMsiHandle( IWineMsiRemoteDatabase
*iface
, MSIHANDLE handle
)
1096 msi_remote_database_impl
* This
= mrd_from_IWineMsiRemoteDatabase( iface
);
1097 This
->database
= handle
;
1101 static const IWineMsiRemoteDatabaseVtbl msi_remote_database_vtbl
=
1106 mrd_IsTablePersistent
,
1108 mrd_GetSummaryInformation
,
1113 HRESULT
create_msi_remote_database( IUnknown
*pOuter
, LPVOID
*ppObj
)
1115 msi_remote_database_impl
*This
;
1117 This
= msi_alloc( sizeof *This
);
1119 return E_OUTOFMEMORY
;
1121 This
->lpVtbl
= &msi_remote_database_vtbl
;