2 * Implementation of the Microsoft Installer (msi.dll)
4 * Copyright 2002-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
25 #define NONAMELESSUNION
26 #define NONAMELESSSTRUCT
39 #include "wine/debug.h"
40 #include "wine/unicode.h"
42 WINE_DEFAULT_DEBUG_CHANNEL(msidb
);
44 #define MSITABLE_HASH_TABLE_SIZE 37
46 typedef struct tagMSICOLUMNHASHENTRY
48 struct tagMSICOLUMNHASHENTRY
*next
;
53 typedef struct tagMSICOLUMNINFO
60 MSICOLUMNHASHENTRY
**hash_table
;
71 typedef struct tagMSITRANSFORM
{
76 static const WCHAR szStringData
[] = {
77 '_','S','t','r','i','n','g','D','a','t','a',0 };
78 static const WCHAR szStringPool
[] = {
79 '_','S','t','r','i','n','g','P','o','o','l',0 };
81 #define MAX_STREAM_NAME 0x1f
83 static UINT
table_get_column_info( MSIDATABASE
*db
, LPCWSTR name
,
84 MSICOLUMNINFO
**pcols
, UINT
*pcount
);
85 static UINT
get_tablecolumns( MSIDATABASE
*db
,
86 LPCWSTR szTableName
, MSICOLUMNINFO
*colinfo
, UINT
*sz
);
87 static void msi_free_colinfo( MSICOLUMNINFO
*colinfo
, UINT count
);
89 static inline UINT
bytes_per_column( const MSICOLUMNINFO
*col
)
91 if( col
->type
& MSITYPE_STRING
)
93 if( (col
->type
& 0xff) > 4 )
94 ERR("Invalid column size!\n");
95 return col
->type
& 0xff;
98 static int utf2mime(int x
)
100 if( (x
>='0') && (x
<='9') )
102 if( (x
>='A') && (x
<='Z') )
104 if( (x
>='a') && (x
<='z') )
113 static LPWSTR
encode_streamname(BOOL bTable
, LPCWSTR in
)
115 DWORD count
= MAX_STREAM_NAME
;
120 count
= lstrlenW( in
)+2;
121 out
= msi_alloc( count
*sizeof(WCHAR
) );
137 if( ( ch
< 0x80 ) && ( utf2mime(ch
) >= 0 ) )
139 ch
= utf2mime(ch
) + 0x4800;
141 if( next
&& (next
<0x80) )
143 next
= utf2mime(next
);
154 ERR("Failed to encode stream name (%s)\n",debugstr_w(in
));
159 static int mime2utf(int x
)
166 return x
- 10 - 26 + 'a';
167 if( x
== (10+26+26) )
172 static BOOL
decode_streamname(LPWSTR in
, LPWSTR out
)
177 while ( (ch
= *in
++) )
179 if( (ch
>= 0x3800 ) && (ch
< 0x4840 ) )
182 ch
= mime2utf(ch
-0x4800);
186 *out
++ = mime2utf(ch
&0x3f);
188 ch
= mime2utf((ch
>>6)&0x3f);
198 void enum_stream_names( IStorage
*stg
)
200 IEnumSTATSTG
*stgenum
= NULL
;
206 r
= IStorage_EnumElements( stg
, 0, NULL
, 0, &stgenum
);
214 r
= IEnumSTATSTG_Next( stgenum
, 1, &stat
, &count
);
215 if( FAILED( r
) || !count
)
217 decode_streamname( stat
.pwcsName
, name
);
218 TRACE("stream %2d -> %s %s\n", n
,
219 debugstr_w(stat
.pwcsName
), debugstr_w(name
) );
223 IEnumSTATSTG_Release( stgenum
);
226 static UINT
read_stream_data( IStorage
*stg
, LPCWSTR stname
,
227 USHORT
**pdata
, UINT
*psz
)
230 UINT ret
= ERROR_FUNCTION_FAILED
;
237 encname
= encode_streamname(TRUE
, stname
);
239 TRACE("%s -> %s\n",debugstr_w(stname
),debugstr_w(encname
));
241 r
= IStorage_OpenStream(stg
, encname
, NULL
,
242 STGM_READ
| STGM_SHARE_EXCLUSIVE
, 0, &stm
);
246 WARN("open stream failed r = %08x - empty table?\n", r
);
250 r
= IStream_Stat(stm
, &stat
, STATFLAG_NONAME
);
253 WARN("open stream failed r = %08x!\n", r
);
257 if( stat
.cbSize
.QuadPart
>> 32 )
263 sz
= stat
.cbSize
.QuadPart
;
264 data
= msi_alloc( sz
);
267 WARN("couldn't allocate memory r=%08x!\n", r
);
268 ret
= ERROR_NOT_ENOUGH_MEMORY
;
272 r
= IStream_Read(stm
, data
, sz
, &count
);
273 if( FAILED( r
) || ( count
!= sz
) )
276 WARN("read stream failed r = %08x!\n", r
);
285 IStream_Release( stm
);
290 UINT
db_get_raw_stream( MSIDATABASE
*db
, LPCWSTR stname
, IStream
**stm
)
295 encname
= encode_streamname(FALSE
, stname
);
297 TRACE("%s -> %s\n",debugstr_w(stname
),debugstr_w(encname
));
299 r
= IStorage_OpenStream(db
->storage
, encname
, NULL
,
300 STGM_READ
| STGM_SHARE_EXCLUSIVE
, 0, stm
);
303 MSITRANSFORM
*transform
;
305 LIST_FOR_EACH_ENTRY( transform
, &db
->transforms
, MSITRANSFORM
, entry
)
307 TRACE("looking for %s in transform storage\n", debugstr_w(stname
) );
308 r
= IStorage_OpenStream( transform
->stg
, encname
, NULL
,
309 STGM_READ
| STGM_SHARE_EXCLUSIVE
, 0, stm
);
317 return SUCCEEDED(r
) ? ERROR_SUCCESS
: ERROR_FUNCTION_FAILED
;
320 UINT
read_raw_stream_data( MSIDATABASE
*db
, LPCWSTR stname
,
321 USHORT
**pdata
, UINT
*psz
)
324 UINT ret
= ERROR_FUNCTION_FAILED
;
330 r
= db_get_raw_stream( db
, stname
, &stm
);
331 if( r
!= ERROR_SUCCESS
)
333 r
= IStream_Stat(stm
, &stat
, STATFLAG_NONAME
);
336 WARN("open stream failed r = %08x!\n", r
);
340 if( stat
.cbSize
.QuadPart
>> 32 )
346 sz
= stat
.cbSize
.QuadPart
;
347 data
= msi_alloc( sz
);
350 WARN("couldn't allocate memory r=%08x!\n", r
);
351 ret
= ERROR_NOT_ENOUGH_MEMORY
;
355 r
= IStream_Read(stm
, data
, sz
, &count
);
356 if( FAILED( r
) || ( count
!= sz
) )
359 WARN("read stream failed r = %08x!\n", r
);
368 IStream_Release( stm
);
373 static UINT
write_stream_data( IStorage
*stg
, LPCWSTR stname
,
374 LPVOID data
, UINT sz
)
377 UINT ret
= ERROR_FUNCTION_FAILED
;
384 encname
= encode_streamname(TRUE
, stname
);
385 r
= IStorage_OpenStream( stg
, encname
, NULL
,
386 STGM_WRITE
| STGM_SHARE_EXCLUSIVE
, 0, &stm
);
389 r
= IStorage_CreateStream( stg
, encname
,
390 STGM_WRITE
| STGM_SHARE_EXCLUSIVE
, 0, 0, &stm
);
395 WARN("open stream failed r = %08x\n", r
);
400 r
= IStream_SetSize( stm
, size
);
403 WARN("Failed to SetSize\n");
408 r
= IStream_Seek( stm
, pos
, STREAM_SEEK_SET
, NULL
);
411 WARN("Failed to Seek\n");
415 r
= IStream_Write(stm
, data
, sz
, &count
);
416 if( FAILED( r
) || ( count
!= sz
) )
418 WARN("Failed to Write\n");
425 IStream_Release( stm
);
430 static void free_table( MSITABLE
*table
)
433 for( i
=0; i
<table
->row_count
; i
++ )
434 msi_free( table
->data
[i
] );
435 msi_free( table
->data
);
439 static UINT
msi_table_get_row_size( const MSICOLUMNINFO
*cols
, UINT count
)
441 const MSICOLUMNINFO
*last_col
= &cols
[count
-1];
444 return last_col
->offset
+ bytes_per_column( last_col
);
447 /* add this table to the list of cached tables in the database */
448 static MSITABLE
*read_table_from_storage( IStorage
*stg
, LPCWSTR name
,
449 const MSICOLUMNINFO
*cols
, UINT num_cols
)
452 USHORT
*rawdata
= NULL
;
453 UINT rawsize
= 0, i
, j
, row_size
= 0;
455 TRACE("%s\n",debugstr_w(name
));
457 /* nonexistent tables should be interpreted as empty tables */
458 t
= msi_alloc( sizeof (MSITABLE
) + lstrlenW(name
)*sizeof (WCHAR
) );
462 row_size
= msi_table_get_row_size( cols
, num_cols
);
466 lstrcpyW( t
->name
, name
);
468 /* if we can't read the table, just assume that it's empty */
469 read_stream_data( stg
, name
, &rawdata
, &rawsize
);
473 TRACE("Read %d bytes\n", rawsize
);
475 if( rawsize
% row_size
)
477 WARN("Table size is invalid %d/%d\n", rawsize
, row_size
);
481 t
->row_count
= rawsize
/ row_size
;
482 t
->data
= msi_alloc_zero( t
->row_count
* sizeof (USHORT
*) );
486 /* transpose all the data */
487 TRACE("Transposing data from %d rows\n", t
->row_count
);
488 for( i
=0; i
<t
->row_count
; i
++ )
490 t
->data
[i
] = msi_alloc( row_size
);
494 for( j
=0; j
<num_cols
; j
++ )
496 UINT ofs
= cols
[j
].offset
/2;
497 UINT n
= bytes_per_column( &cols
[j
] );
502 t
->data
[i
][ofs
] = rawdata
[ofs
*t
->row_count
+ i
];
505 t
->data
[i
][ofs
] = rawdata
[ofs
*t
->row_count
+ i
*2 ];
506 t
->data
[i
][ofs
+1] = rawdata
[ofs
*t
->row_count
+ i
*2 + 1];
509 ERR("oops - unknown column width %d\n", n
);
523 void free_cached_tables( MSIDATABASE
*db
)
525 while( !list_empty( &db
->tables
) )
527 MSITABLE
*t
= LIST_ENTRY( list_head( &db
->tables
), MSITABLE
, entry
);
529 list_remove( &t
->entry
);
534 static MSITABLE
*find_cached_table( MSIDATABASE
*db
, LPCWSTR name
)
538 LIST_FOR_EACH_ENTRY( t
, &db
->tables
, MSITABLE
, entry
)
539 if( !lstrcmpW( name
, t
->name
) )
545 static UINT
table_get_column_info( MSIDATABASE
*db
, LPCWSTR name
, MSICOLUMNINFO
**pcols
, UINT
*pcount
)
547 UINT r
, column_count
= 0;
548 MSICOLUMNINFO
*columns
;
550 /* get the number of columns in this table */
552 r
= get_tablecolumns( db
, name
, NULL
, &column_count
);
553 if( r
!= ERROR_SUCCESS
)
556 /* if there's no columns, there's no table */
557 if( column_count
== 0 )
558 return ERROR_INVALID_PARAMETER
;
560 TRACE("Table %s found\n", debugstr_w(name
) );
562 columns
= msi_alloc( column_count
*sizeof (MSICOLUMNINFO
) );
564 return ERROR_FUNCTION_FAILED
;
566 r
= get_tablecolumns( db
, name
, columns
, &column_count
);
567 if( r
!= ERROR_SUCCESS
)
570 return ERROR_FUNCTION_FAILED
;
574 *pcount
= column_count
;
579 static MSITABLE
*get_table( MSIDATABASE
*db
, LPCWSTR name
,
580 const MSICOLUMNINFO
*cols
, UINT num_cols
)
584 /* first, see if the table is cached */
585 table
= find_cached_table( db
, name
);
589 table
= read_table_from_storage( db
->storage
, name
, cols
, num_cols
);
591 list_add_head( &db
->tables
, &table
->entry
);
596 static UINT
save_table( MSIDATABASE
*db
, MSITABLE
*t
)
598 USHORT
*rawdata
= NULL
, *p
;
599 UINT rawsize
, r
, i
, j
, row_size
, num_cols
= 0;
600 MSICOLUMNINFO
*cols
= NULL
;
602 TRACE("Saving %s\n", debugstr_w( t
->name
) );
604 r
= table_get_column_info( db
, t
->name
, &cols
, &num_cols
);
605 if( r
!= ERROR_SUCCESS
)
608 row_size
= msi_table_get_row_size( cols
, num_cols
);
610 rawsize
= t
->row_count
* row_size
;
611 rawdata
= msi_alloc_zero( rawsize
);
614 r
= ERROR_NOT_ENOUGH_MEMORY
;
619 for( i
=0; i
<num_cols
; i
++ )
621 for( j
=0; j
<t
->row_count
; j
++ )
623 UINT offset
= cols
[i
].offset
;
625 *p
++ = t
->data
[j
][offset
/2];
626 if( 4 == bytes_per_column( &cols
[i
] ) )
627 *p
++ = t
->data
[j
][offset
/2+1];
631 TRACE("writing %d bytes\n", rawsize
);
632 r
= write_stream_data( db
->storage
, t
->name
, rawdata
, rawsize
);
635 msi_free_colinfo( cols
, num_cols
);
642 HRESULT
init_string_table( IStorage
*stg
)
645 USHORT zero
[2] = { 0, 0 };
650 encname
= encode_streamname(TRUE
, szStringPool
);
652 /* create the StringPool stream... add the zero string to it*/
653 r
= IStorage_CreateStream( stg
, encname
,
654 STGM_WRITE
| STGM_SHARE_EXCLUSIVE
, 0, 0, &stm
);
662 r
= IStream_Write(stm
, zero
, sizeof zero
, &count
);
663 IStream_Release( stm
);
665 if( FAILED( r
) || ( count
!= sizeof zero
) )
671 /* create the StringData stream... make it zero length */
672 encname
= encode_streamname(TRUE
, szStringData
);
673 r
= IStorage_CreateStream( stg
, encname
,
674 STGM_WRITE
| STGM_SHARE_EXCLUSIVE
, 0, 0, &stm
);
681 IStream_Release( stm
);
686 string_table
*load_string_table( IStorage
*stg
)
688 string_table
*st
= NULL
;
691 UINT r
, datasize
= 0, poolsize
= 0, codepage
;
692 DWORD i
, count
, offset
, len
, n
, refs
;
694 r
= read_stream_data( stg
, szStringPool
, &pool
, &poolsize
);
695 if( r
!= ERROR_SUCCESS
)
697 r
= read_stream_data( stg
, szStringData
, (USHORT
**)&data
, &datasize
);
698 if( r
!= ERROR_SUCCESS
)
703 codepage
= pool
[0] | ( pool
[1] << 16 );
706 st
= msi_init_stringtable( count
, codepage
);
713 /* the string reference count is always the second word */
716 /* empty entries have two zeros, still have a string id */
717 if (pool
[i
*2] == 0 && refs
== 0)
725 * If a string is over 64k, the previous string entry is made null
726 * and its the high word of the length is inserted in the null string's
727 * reference count field.
731 len
= (pool
[i
*2+3] << 16) + pool
[i
*2+2];
740 if ( (offset
+ len
) > datasize
)
742 ERR("string table corrupt?\n");
746 r
= msi_addstring( st
, n
, data
+offset
, len
, refs
);
748 ERR("Failed to add string %d\n", n
);
753 if ( datasize
!= offset
)
754 ERR("string table load failed! (%08x != %08x), please report\n", datasize
, offset
);
756 TRACE("Loaded %d strings\n", count
);
765 static UINT
save_string_table( MSIDATABASE
*db
)
767 UINT i
, count
, datasize
= 0, poolsize
= 0, sz
, used
, r
, codepage
, n
;
768 UINT ret
= ERROR_FUNCTION_FAILED
;
774 /* construct the new table in memory first */
775 count
= msi_string_totalsize( db
->strings
, &datasize
, &poolsize
);
777 TRACE("%u %u %u\n", count
, datasize
, poolsize
);
779 pool
= msi_alloc( poolsize
);
782 WARN("Failed to alloc pool %d bytes\n", poolsize
);
785 data
= msi_alloc( datasize
);
788 WARN("Failed to alloc data %d bytes\n", poolsize
);
793 codepage
= msi_string_get_codepage( db
->strings
);
794 pool
[0]=codepage
&0xffff;
795 pool
[1]=(codepage
>>16);
797 for( i
=1; i
<count
; i
++ )
799 sz
= datasize
- used
;
800 r
= msi_id2stringA( db
->strings
, i
, data
+used
, &sz
);
801 if( r
!= ERROR_SUCCESS
)
803 ERR("failed to fetch string\n");
806 if( sz
&& (sz
< (datasize
- used
) ) )
810 pool
[ n
*2 + 1 ] = msi_id_refcount( db
->strings
, i
);
821 pool
[ n
*2 + 2 ] = sz
&0xffff;
822 pool
[ n
*2 + 3 ] = (sz
>>16);
826 if( used
> datasize
)
828 ERR("oops overran %d >= %d\n", used
, datasize
);
833 if( used
!= datasize
)
835 ERR("oops used %d != datasize %d\n", used
, datasize
);
839 /* write the streams */
840 r
= write_stream_data( db
->storage
, szStringData
, data
, datasize
);
841 TRACE("Wrote StringData r=%08x\n", r
);
844 r
= write_stream_data( db
->storage
, szStringPool
, pool
, poolsize
);
845 TRACE("Wrote StringPool r=%08x\n", r
);
858 /* information for default tables */
859 static const WCHAR szTables
[] = { '_','T','a','b','l','e','s',0 };
860 static const WCHAR szTable
[] = { 'T','a','b','l','e',0 };
861 static const WCHAR szName
[] = { 'N','a','m','e',0 };
862 static const WCHAR szColumns
[] = { '_','C','o','l','u','m','n','s',0 };
863 static const WCHAR szNumber
[] = { 'N','u','m','b','e','r',0 };
864 static const WCHAR szType
[] = { 'T','y','p','e',0 };
866 static const MSICOLUMNINFO _Columns_cols
[4] = {
867 { szColumns
, 1, szTable
, MSITYPE_VALID
| MSITYPE_STRING
| 64, 0 },
868 { szColumns
, 2, szNumber
, MSITYPE_VALID
| 2, 2 },
869 { szColumns
, 3, szName
, MSITYPE_VALID
| MSITYPE_STRING
| 64, 4 },
870 { szColumns
, 4, szType
, MSITYPE_VALID
| 2, 6 },
872 static const MSICOLUMNINFO _Tables_cols
[1] = {
873 { szTables
, 1, szName
, MSITYPE_VALID
| MSITYPE_STRING
| 64, 0 },
876 static void table_calc_column_offsets( MSICOLUMNINFO
*colinfo
, DWORD count
)
880 for( i
=0; colinfo
&& (i
<count
); i
++ )
882 assert( (i
+1) == colinfo
[ i
].number
);
884 colinfo
[i
].offset
= colinfo
[ i
- 1 ].offset
885 + bytes_per_column( &colinfo
[ i
- 1 ] );
887 colinfo
[i
].offset
= 0;
888 TRACE("column %d is [%s] with type %08x ofs %d\n",
889 colinfo
[i
].number
, debugstr_w(colinfo
[i
].colname
),
890 colinfo
[i
].type
, colinfo
[i
].offset
);
894 static UINT
get_defaulttablecolumns( LPCWSTR name
, MSICOLUMNINFO
*colinfo
, UINT
*sz
)
896 const MSICOLUMNINFO
*p
;
899 TRACE("%s\n", debugstr_w(name
));
901 if (!lstrcmpW( name
, szTables
))
906 else if (!lstrcmpW( name
, szColumns
))
912 return ERROR_FUNCTION_FAILED
;
914 /* duplicate the string data so we can free it in msi_free_colinfo */
917 if (colinfo
&& (i
< *sz
) )
919 memcpy( &colinfo
[i
], &p
[i
], sizeof(MSICOLUMNINFO
) );
920 colinfo
[i
].tablename
= strdupW( p
[i
].tablename
);
921 colinfo
[i
].colname
= strdupW( p
[i
].colname
);
923 if( colinfo
&& (i
>= *sz
) )
926 table_calc_column_offsets( colinfo
, n
);
928 return ERROR_SUCCESS
;
931 static void msi_free_colinfo( MSICOLUMNINFO
*colinfo
, UINT count
)
935 for( i
=0; i
<count
; i
++ )
937 msi_free( (LPWSTR
) colinfo
[i
].tablename
);
938 msi_free( (LPWSTR
) colinfo
[i
].colname
);
939 msi_free( colinfo
[i
].hash_table
);
943 static LPWSTR
msi_makestring( MSIDATABASE
*db
, UINT stringid
)
945 return strdupW(msi_string_lookup_id( db
->strings
, stringid
));
948 static UINT
get_tablecolumns( MSIDATABASE
*db
,
949 LPCWSTR szTableName
, MSICOLUMNINFO
*colinfo
, UINT
*sz
)
951 UINT r
, i
, n
=0, table_id
, count
, maxcount
= *sz
;
952 MSITABLE
*table
= NULL
;
954 TRACE("%s\n", debugstr_w(szTableName
));
956 /* first check if there is a default table with that name */
957 r
= get_defaulttablecolumns( szTableName
, colinfo
, sz
);
958 if( ( r
== ERROR_SUCCESS
) && *sz
)
961 table
= get_table( db
, szColumns
, _Columns_cols
, 4 );
964 ERR("couldn't load _Columns table\n");
965 return ERROR_FUNCTION_FAILED
;
968 /* convert table and column names to IDs from the string table */
969 r
= msi_string2idW( db
->strings
, szTableName
, &table_id
);
970 if( r
!= ERROR_SUCCESS
)
972 WARN("Couldn't find id for %s\n", debugstr_w(szTableName
));
976 TRACE("Table id is %d, row count is %d\n", table_id
, table
->row_count
);
978 /* if maxcount is non-zero, assume it's exactly right for this table */
979 memset( colinfo
, 0, maxcount
*sizeof(*colinfo
) );
980 count
= table
->row_count
;
981 for( i
=0; i
<count
; i
++ )
983 if( table
->data
[ i
][ 0 ] != table_id
)
987 UINT id
= table
->data
[ i
] [ 2 ];
988 UINT col
= table
->data
[ i
][ 1 ] - (1<<15);
990 /* check the column number is in range */
991 if (col
<1 || col
>maxcount
)
993 ERR("column %d out of range\n", col
);
997 /* check if this column was already set */
998 if (colinfo
[ col
- 1 ].number
)
1000 ERR("duplicate column %d\n", col
);
1004 colinfo
[ col
- 1 ].tablename
= msi_makestring( db
, table_id
);
1005 colinfo
[ col
- 1 ].number
= col
;
1006 colinfo
[ col
- 1 ].colname
= msi_makestring( db
, id
);
1007 colinfo
[ col
- 1 ].type
= table
->data
[ i
] [ 3 ] - (1<<15);
1008 colinfo
[ col
- 1 ].offset
= 0;
1009 colinfo
[ col
- 1 ].hash_table
= NULL
;
1014 TRACE("%s has %d columns\n", debugstr_w(szTableName
), n
);
1016 if (colinfo
&& n
!= maxcount
)
1018 ERR("missing column in table %s\n", debugstr_w(szTableName
));
1019 msi_free_colinfo(colinfo
, maxcount
);
1020 return ERROR_FUNCTION_FAILED
;
1023 table_calc_column_offsets( colinfo
, n
);
1026 return ERROR_SUCCESS
;
1029 /* try to find the table name in the _Tables table */
1030 BOOL
TABLE_Exists( MSIDATABASE
*db
, LPWSTR name
)
1032 UINT r
, table_id
= 0, i
, count
;
1033 MSITABLE
*table
= NULL
;
1035 if( !lstrcmpW( name
, szTables
) )
1037 if( !lstrcmpW( name
, szColumns
) )
1040 r
= msi_string2idW( db
->strings
, name
, &table_id
);
1041 if( r
!= ERROR_SUCCESS
)
1043 TRACE("Couldn't find id for %s\n", debugstr_w(name
));
1047 table
= get_table( db
, szTables
, _Tables_cols
, 1 );
1050 TRACE("table %s not available\n", debugstr_w(szTables
));
1054 /* count = table->size/2; */
1055 count
= table
->row_count
;
1056 for( i
=0; i
<count
; i
++ )
1057 if( table
->data
[ i
][ 0 ] == table_id
)
1063 TRACE("Searched %d tables, but %d was not found\n", count
, table_id
);
1068 /* below is the query interface to a table */
1070 typedef struct tagMSITABLEVIEW
1075 MSICOLUMNINFO
*columns
;
1081 static UINT
TABLE_fetch_int( struct tagMSIVIEW
*view
, UINT row
, UINT col
, UINT
*val
)
1083 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1084 UINT offset
, num_rows
, n
;
1087 return ERROR_INVALID_PARAMETER
;
1089 if( (col
==0) || (col
>tv
->num_cols
) )
1090 return ERROR_INVALID_PARAMETER
;
1092 /* how many rows are there ? */
1093 num_rows
= tv
->table
->row_count
;
1094 if( row
>= num_rows
)
1095 return ERROR_NO_MORE_ITEMS
;
1097 if( tv
->columns
[col
-1].offset
>= tv
->row_size
)
1099 ERR("Stuffed up %d >= %d\n", tv
->columns
[col
-1].offset
, tv
->row_size
);
1100 ERR("%p %p\n", tv
, tv
->columns
);
1101 return ERROR_FUNCTION_FAILED
;
1104 offset
= row
+ (tv
->columns
[col
-1].offset
/2) * num_rows
;
1105 n
= bytes_per_column( &tv
->columns
[col
-1] );
1109 offset
= tv
->columns
[col
-1].offset
/2;
1110 *val
= tv
->table
->data
[row
][offset
] +
1111 (tv
->table
->data
[row
][offset
+ 1] << 16);
1114 offset
= tv
->columns
[col
-1].offset
/2;
1115 *val
= tv
->table
->data
[row
][offset
];
1118 ERR("oops! what is %d bytes per column?\n", n
);
1119 return ERROR_FUNCTION_FAILED
;
1122 /* TRACE("Data [%d][%d] = %d\n", row, col, *val ); */
1124 return ERROR_SUCCESS
;
1128 * We need a special case for streams, as we need to reference column with
1129 * the name of the stream in the same table, and the table name
1130 * which may not be available at higher levels of the query
1132 static UINT
TABLE_fetch_stream( struct tagMSIVIEW
*view
, UINT row
, UINT col
, IStream
**stm
)
1134 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1135 UINT ival
= 0, refcol
= 0, r
;
1139 static const WCHAR szDot
[] = { '.', 0 };
1142 if( !view
->ops
->fetch_int
)
1143 return ERROR_INVALID_PARAMETER
;
1146 * The column marked with the type stream data seems to have a single number
1147 * which references the column containing the name of the stream data
1149 * Fetch the column to reference first.
1151 r
= view
->ops
->fetch_int( view
, row
, col
, &ival
);
1152 if( r
!= ERROR_SUCCESS
)
1155 /* check the column value is in range */
1156 if (ival
< 0 || ival
> tv
->num_cols
|| ival
== col
)
1158 ERR("bad column ref (%u) for stream\n", ival
);
1159 return ERROR_FUNCTION_FAILED
;
1162 if ( tv
->columns
[ival
- 1].type
& MSITYPE_STRING
)
1164 /* now get the column with the name of the stream */
1165 r
= view
->ops
->fetch_int( view
, row
, ival
, &refcol
);
1166 if ( r
!= ERROR_SUCCESS
)
1169 /* lookup the string value from the string table */
1170 sval
= msi_string_lookup_id( tv
->db
->strings
, refcol
);
1172 return ERROR_INVALID_PARAMETER
;
1176 static const WCHAR fmt
[] = { '%','d',0 };
1177 sprintfW( number
, fmt
, ival
);
1181 len
= lstrlenW( tv
->name
) + 2 + lstrlenW( sval
);
1182 full_name
= msi_alloc( len
*sizeof(WCHAR
) );
1183 lstrcpyW( full_name
, tv
->name
);
1184 lstrcatW( full_name
, szDot
);
1185 lstrcatW( full_name
, sval
);
1187 r
= db_get_raw_stream( tv
->db
, full_name
, stm
);
1189 ERR("fetching stream %s, error = %d\n",debugstr_w(full_name
), r
);
1190 msi_free( full_name
);
1195 static UINT
TABLE_set_int( MSITABLEVIEW
*tv
, UINT row
, UINT col
, UINT val
)
1200 return ERROR_INVALID_PARAMETER
;
1202 if( (col
==0) || (col
>tv
->num_cols
) )
1203 return ERROR_INVALID_PARAMETER
;
1205 if( tv
->columns
[col
-1].offset
>= tv
->row_size
)
1207 ERR("Stuffed up %d >= %d\n", tv
->columns
[col
-1].offset
, tv
->row_size
);
1208 ERR("%p %p\n", tv
, tv
->columns
);
1209 return ERROR_FUNCTION_FAILED
;
1212 n
= bytes_per_column( &tv
->columns
[col
-1] );
1216 offset
= tv
->columns
[col
-1].offset
/2;
1217 tv
->table
->data
[row
][offset
] = val
& 0xffff;
1218 tv
->table
->data
[row
][offset
+ 1] = (val
>>16)&0xffff;
1221 offset
= tv
->columns
[col
-1].offset
/2;
1222 tv
->table
->data
[row
][offset
] = val
;
1225 ERR("oops! what is %d bytes per column?\n", n
);
1226 return ERROR_FUNCTION_FAILED
;
1228 return ERROR_SUCCESS
;
1231 static UINT
TABLE_set_row( struct tagMSIVIEW
*view
, UINT row
, MSIRECORD
*rec
, UINT mask
)
1233 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1234 UINT i
, val
, r
= ERROR_SUCCESS
;
1237 return ERROR_INVALID_PARAMETER
;
1239 /* test if any of the mask bits are invalid */
1240 if ( mask
>= (1<<tv
->num_cols
) )
1241 return ERROR_INVALID_PARAMETER
;
1243 for ( i
= 0; i
< tv
->num_cols
; i
++ )
1245 /* only update the fields specified in the mask */
1246 if ( !(mask
&(1<<i
)) )
1249 /* FIXME: should we allow updating keys? */
1252 if ( !MSI_RecordIsNull( rec
, i
+ 1 ) )
1254 if ( MSITYPE_IS_BINARY(tv
->columns
[ i
].type
) )
1256 val
= 1; /* refers to the first key column */
1258 else if ( tv
->columns
[i
].type
& MSITYPE_STRING
)
1260 LPCWSTR sval
= MSI_RecordGetString( rec
, i
+ 1 );
1261 val
= msi_addstringW( tv
->db
->strings
, 0, sval
, -1, 1 );
1263 else if ( 2 == bytes_per_column( &tv
->columns
[ i
] ) )
1265 val
= 0x8000 + MSI_RecordGetInteger( rec
, i
+ 1 );
1266 if ( val
& 0xffff0000 )
1268 ERR("field %u value %d out of range\n", i
+1, val
- 0x8000 );
1269 return ERROR_FUNCTION_FAILED
;
1274 INT ival
= MSI_RecordGetInteger( rec
, i
+ 1 );
1275 val
= ival
^ 0x80000000;
1279 r
= TABLE_set_int( tv
, row
, i
+1, val
);
1280 if ( r
!= ERROR_SUCCESS
)
1286 static UINT
table_create_new_row( struct tagMSIVIEW
*view
, UINT
*num
)
1288 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1292 TRACE("%p\n", view
);
1295 return ERROR_INVALID_PARAMETER
;
1297 row
= msi_alloc_zero( tv
->row_size
);
1299 return ERROR_NOT_ENOUGH_MEMORY
;
1301 sz
= (tv
->table
->row_count
+ 1) * sizeof (UINT
*);
1302 if( tv
->table
->data
)
1303 p
= msi_realloc( tv
->table
->data
, sz
);
1305 p
= msi_alloc( sz
);
1309 return ERROR_NOT_ENOUGH_MEMORY
;
1312 tv
->table
->data
= p
;
1313 tv
->table
->data
[tv
->table
->row_count
] = row
;
1314 *num
= tv
->table
->row_count
;
1315 tv
->table
->row_count
++;
1317 return ERROR_SUCCESS
;
1320 static UINT
TABLE_execute( struct tagMSIVIEW
*view
, MSIRECORD
*record
)
1322 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1324 TRACE("%p %p\n", tv
, record
);
1326 TRACE("There are %d columns\n", tv
->num_cols
);
1327 tv
->table
= get_table( tv
->db
, tv
->name
, tv
->columns
, tv
->num_cols
);
1329 return ERROR_FUNCTION_FAILED
;
1331 return ERROR_SUCCESS
;
1334 static UINT
TABLE_close( struct tagMSIVIEW
*view
)
1336 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1338 TRACE("%p\n", view
);
1341 return ERROR_FUNCTION_FAILED
;
1345 return ERROR_SUCCESS
;
1348 static UINT
TABLE_get_dimensions( struct tagMSIVIEW
*view
, UINT
*rows
, UINT
*cols
)
1350 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1352 TRACE("%p %p %p\n", view
, rows
, cols
);
1355 *cols
= tv
->num_cols
;
1359 return ERROR_INVALID_PARAMETER
;
1360 *rows
= tv
->table
->row_count
;
1363 return ERROR_SUCCESS
;
1366 static UINT
TABLE_get_column_info( struct tagMSIVIEW
*view
,
1367 UINT n
, LPWSTR
*name
, UINT
*type
)
1369 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1371 TRACE("%p %d %p %p\n", tv
, n
, name
, type
);
1373 if( ( n
== 0 ) || ( n
> tv
->num_cols
) )
1374 return ERROR_INVALID_PARAMETER
;
1378 *name
= strdupW( tv
->columns
[n
-1].colname
);
1380 return ERROR_FUNCTION_FAILED
;
1383 *type
= tv
->columns
[n
-1].type
;
1385 return ERROR_SUCCESS
;
1388 static UINT
msi_table_find_row( MSITABLEVIEW
*tv
, MSIRECORD
*rec
, UINT
*row
);
1390 static UINT
table_validate_new( MSITABLEVIEW
*tv
, MSIRECORD
*rec
)
1394 /* check there's no null values where they're not allowed */
1395 for( i
= 0; i
< tv
->num_cols
; i
++ )
1397 if ( tv
->columns
[i
].type
& MSITYPE_NULLABLE
)
1400 if ( MSITYPE_IS_BINARY(tv
->columns
[i
].type
) )
1401 TRACE("skipping binary column\n");
1402 else if ( tv
->columns
[i
].type
& MSITYPE_STRING
)
1406 str
= MSI_RecordGetString( rec
, i
+1 );
1407 if (str
== NULL
|| str
[0] == 0)
1408 return ERROR_INVALID_DATA
;
1414 n
= MSI_RecordGetInteger( rec
, i
+1 );
1415 if (n
== MSI_NULL_INTEGER
)
1416 return ERROR_INVALID_DATA
;
1420 /* check there's no duplicate keys */
1421 r
= msi_table_find_row( tv
, rec
, &row
);
1422 if (r
== ERROR_SUCCESS
)
1423 return ERROR_INVALID_DATA
;
1425 return ERROR_SUCCESS
;
1428 static UINT
TABLE_insert_row( struct tagMSIVIEW
*view
, MSIRECORD
*rec
)
1430 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1433 TRACE("%p %p\n", tv
, rec
);
1435 /* check that the key is unique - can we find a matching row? */
1436 r
= table_validate_new( tv
, rec
);
1437 if( r
!= ERROR_SUCCESS
)
1438 return ERROR_FUNCTION_FAILED
;
1440 r
= table_create_new_row( view
, &row
);
1441 TRACE("insert_row returned %08x\n", r
);
1442 if( r
!= ERROR_SUCCESS
)
1445 return TABLE_set_row( view
, row
, rec
, (1<<tv
->num_cols
) - 1 );
1448 static UINT
TABLE_modify( struct tagMSIVIEW
*view
, MSIMODIFY eModifyMode
,
1451 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1454 TRACE("%p %d %p\n", view
, eModifyMode
, rec
);
1458 r
= TABLE_execute( view
, NULL
);
1463 switch (eModifyMode
)
1465 case MSIMODIFY_VALIDATE_NEW
:
1466 r
= table_validate_new( tv
, rec
);
1469 case MSIMODIFY_INSERT_TEMPORARY
:
1470 r
= table_validate_new( tv
, rec
);
1471 if (r
!= ERROR_SUCCESS
)
1473 r
= TABLE_insert_row( view
, rec
);
1476 case MSIMODIFY_REFRESH
:
1477 case MSIMODIFY_INSERT
:
1478 case MSIMODIFY_UPDATE
:
1479 case MSIMODIFY_ASSIGN
:
1480 case MSIMODIFY_REPLACE
:
1481 case MSIMODIFY_MERGE
:
1482 case MSIMODIFY_DELETE
:
1483 case MSIMODIFY_VALIDATE
:
1484 case MSIMODIFY_VALIDATE_FIELD
:
1485 case MSIMODIFY_VALIDATE_DELETE
:
1486 FIXME("%p %d %p - mode not implemented\n", view
, eModifyMode
, rec
);
1487 r
= ERROR_CALL_NOT_IMPLEMENTED
;
1491 r
= ERROR_INVALID_DATA
;
1497 static UINT
TABLE_delete( struct tagMSIVIEW
*view
)
1499 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1501 TRACE("%p\n", view
);
1507 msi_free_colinfo( tv
->columns
, tv
->num_cols
);
1508 msi_free( tv
->columns
);
1514 return ERROR_SUCCESS
;
1517 static UINT
TABLE_find_matching_rows( struct tagMSIVIEW
*view
, UINT col
,
1518 UINT val
, UINT
*row
, MSIITERHANDLE
*handle
)
1520 MSITABLEVIEW
*tv
= (MSITABLEVIEW
*)view
;
1521 const MSICOLUMNHASHENTRY
*entry
;
1523 TRACE("%p, %d, %u, %p\n", view
, col
, val
, *handle
);
1526 return ERROR_INVALID_PARAMETER
;
1528 if( (col
==0) || (col
> tv
->num_cols
) )
1529 return ERROR_INVALID_PARAMETER
;
1531 if( !tv
->columns
[col
-1].hash_table
)
1534 UINT num_rows
= tv
->table
->row_count
;
1535 MSICOLUMNHASHENTRY
**hash_table
;
1536 MSICOLUMNHASHENTRY
*new_entry
;
1538 if( tv
->columns
[col
-1].offset
>= tv
->row_size
)
1540 ERR("Stuffed up %d >= %d\n", tv
->columns
[col
-1].offset
, tv
->row_size
);
1541 ERR("%p %p\n", tv
, tv
->columns
);
1542 return ERROR_FUNCTION_FAILED
;
1545 /* allocate contiguous memory for the table and its entries so we
1546 * don't have to do an expensive cleanup */
1547 hash_table
= msi_alloc(MSITABLE_HASH_TABLE_SIZE
* sizeof(MSICOLUMNHASHENTRY
*) +
1548 num_rows
* sizeof(MSICOLUMNHASHENTRY
));
1550 return ERROR_OUTOFMEMORY
;
1552 memset(hash_table
, 0, MSITABLE_HASH_TABLE_SIZE
* sizeof(MSICOLUMNHASHENTRY
*));
1553 tv
->columns
[col
-1].hash_table
= hash_table
;
1555 new_entry
= (MSICOLUMNHASHENTRY
*)(hash_table
+ MSITABLE_HASH_TABLE_SIZE
);
1557 for (i
= 0; i
< num_rows
; i
++, new_entry
++)
1560 UINT offset
= i
+ (tv
->columns
[col
-1].offset
/2) * num_rows
;
1561 n
= bytes_per_column( &tv
->columns
[col
-1] );
1565 offset
= tv
->columns
[col
-1].offset
/2;
1566 row_value
= tv
->table
->data
[i
][offset
] +
1567 (tv
->table
->data
[i
][offset
+ 1] << 16);
1570 offset
= tv
->columns
[col
-1].offset
/2;
1571 row_value
= tv
->table
->data
[i
][offset
];
1574 ERR("oops! what is %d bytes per column?\n", n
);
1578 new_entry
->next
= NULL
;
1579 new_entry
->value
= row_value
;
1581 if (hash_table
[row_value
% MSITABLE_HASH_TABLE_SIZE
])
1583 MSICOLUMNHASHENTRY
*prev_entry
= hash_table
[row_value
% MSITABLE_HASH_TABLE_SIZE
];
1584 while (prev_entry
->next
)
1585 prev_entry
= prev_entry
->next
;
1586 prev_entry
->next
= new_entry
;
1589 hash_table
[row_value
% MSITABLE_HASH_TABLE_SIZE
] = new_entry
;
1594 entry
= tv
->columns
[col
-1].hash_table
[val
% MSITABLE_HASH_TABLE_SIZE
];
1596 entry
= (*handle
)->next
;
1598 while (entry
&& entry
->value
!= val
)
1599 entry
= entry
->next
;
1603 return ERROR_NO_MORE_ITEMS
;
1606 return ERROR_SUCCESS
;
1610 static const MSIVIEWOPS table_ops
=
1618 TABLE_get_dimensions
,
1619 TABLE_get_column_info
,
1622 TABLE_find_matching_rows
1625 UINT
TABLE_CreateView( MSIDATABASE
*db
, LPCWSTR name
, MSIVIEW
**view
)
1628 UINT r
, sz
, column_count
;
1629 MSICOLUMNINFO
*columns
;
1631 TRACE("%p %s %p\n", db
, debugstr_w(name
), view
);
1633 /* get the number of columns in this table */
1635 r
= get_tablecolumns( db
, name
, NULL
, &column_count
);
1636 if( r
!= ERROR_SUCCESS
)
1639 /* if there's no columns, there's no table */
1640 if( column_count
== 0 )
1641 return ERROR_INVALID_PARAMETER
;
1643 TRACE("Table found\n");
1645 sz
= sizeof *tv
+ lstrlenW(name
)*sizeof name
[0] ;
1646 tv
= msi_alloc_zero( sz
);
1648 return ERROR_FUNCTION_FAILED
;
1650 columns
= msi_alloc( column_count
*sizeof (MSICOLUMNINFO
));
1654 return ERROR_FUNCTION_FAILED
;
1657 r
= get_tablecolumns( db
, name
, columns
, &column_count
);
1658 if( r
!= ERROR_SUCCESS
)
1660 msi_free( columns
);
1662 return ERROR_FUNCTION_FAILED
;
1665 TRACE("Table has %d columns\n", column_count
);
1667 /* fill the structure */
1668 tv
->view
.ops
= &table_ops
;
1670 tv
->columns
= columns
;
1671 tv
->num_cols
= column_count
;
1673 tv
->row_size
= msi_table_get_row_size( columns
, column_count
);
1675 TRACE("%s one row is %d bytes\n", debugstr_w(name
), tv
->row_size
);
1677 *view
= (MSIVIEW
*) tv
;
1678 lstrcpyW( tv
->name
, name
);
1680 return ERROR_SUCCESS
;
1683 UINT
MSI_CommitTables( MSIDATABASE
*db
)
1686 MSITABLE
*table
= NULL
;
1690 r
= save_string_table( db
);
1691 if( r
!= ERROR_SUCCESS
)
1693 WARN("failed to save string table r=%08x\n",r
);
1697 LIST_FOR_EACH_ENTRY( table
, &db
->tables
, MSITABLE
, entry
)
1699 r
= save_table( db
, table
);
1700 if( r
!= ERROR_SUCCESS
)
1702 WARN("failed to save table %s (r=%08x)\n",
1703 debugstr_w(table
->name
), r
);
1708 /* force everything to reload next time */
1709 free_cached_tables( db
);
1711 return ERROR_SUCCESS
;
1714 MSICONDITION
MSI_DatabaseIsTablePersistent( MSIDATABASE
*db
, LPCWSTR table
)
1717 return MSICONDITION_ERROR
;
1719 return MSICONDITION_FALSE
;
1722 static MSIRECORD
*msi_get_transform_record( MSITABLEVIEW
*tv
, string_table
*st
, USHORT
*rawdata
)
1724 UINT i
, val
, ofs
= 0;
1725 USHORT mask
= *rawdata
++;
1726 MSICOLUMNINFO
*columns
= tv
->columns
;
1729 rec
= MSI_CreateRecord( tv
->num_cols
);
1734 for( i
=0; i
<tv
->num_cols
; i
++ )
1736 UINT n
= bytes_per_column( &columns
[i
] );
1738 if ( (mask
&1) && (i
>=(mask
>>8)) )
1740 /* all keys must be present */
1741 if ( (~mask
&1) && (~columns
[i
].type
& MSITYPE_KEY
) && ((1<<i
) & ~mask
) )
1748 if( (columns
[i
].type
& MSITYPE_STRING
) &&
1749 ! MSITYPE_IS_BINARY(tv
->columns
[i
].type
) )
1751 LPCWSTR sval
= msi_string_lookup_id( st
, val
);
1752 MSI_RecordSetStringW( rec
, i
+1, sval
);
1753 TRACE(" field %d [%s]\n", i
+1, debugstr_w(sval
));
1758 MSI_RecordSetInteger( rec
, i
+1, val
^0x8000 );
1759 TRACE(" field %d [0x%04x]\n", i
+1, val
);
1763 val
= (rawdata
[ofs
] + (rawdata
[ofs
+ 1]<<16));
1765 MSI_RecordSetInteger( rec
, i
+1, val
^0x80000000 );
1766 TRACE(" field %d [0x%08x]\n", i
+1, val
);
1769 ERR("oops - unknown column width %d\n", n
);
1777 static void dump_record( MSIRECORD
*rec
)
1781 n
= MSI_RecordGetFieldCount( rec
);
1782 for( i
=1; i
<=n
; i
++ )
1784 LPCWSTR sval
= MSI_RecordGetString( rec
, i
);
1786 if( MSI_RecordIsNull( rec
, i
) )
1787 TRACE("row -> []\n");
1788 else if( (sval
= MSI_RecordGetString( rec
, i
)) )
1789 TRACE("row -> [%s]\n", debugstr_w(sval
));
1791 TRACE("row -> [0x%08x]\n", MSI_RecordGetInteger( rec
, i
) );
1795 static void dump_table( string_table
*st
, USHORT
*rawdata
, UINT rawsize
)
1800 for( i
=0; i
<(rawsize
/2); i
++ )
1802 sval
= msi_string_lookup_id( st
, rawdata
[i
] );
1803 MESSAGE(" %04x %s\n", rawdata
[i
], debugstr_w(sval
) );
1807 static UINT
* msi_record_to_row( MSITABLEVIEW
*tv
, MSIRECORD
*rec
)
1812 data
= msi_alloc( tv
->num_cols
*sizeof (UINT
) );
1813 for( i
=0; i
<tv
->num_cols
; i
++ )
1817 if ( ~tv
->columns
[i
].type
& MSITYPE_KEY
)
1820 /* turn the transform column value into a row value */
1821 if ( ( tv
->columns
[i
].type
& MSITYPE_STRING
) &&
1822 ! MSITYPE_IS_BINARY(tv
->columns
[i
].type
) )
1824 str
= MSI_RecordGetString( rec
, i
+1 );
1825 r
= msi_string2idW( tv
->db
->strings
, str
, &data
[i
] );
1827 /* if there's no matching string in the string table,
1828 these keys can't match any record, so fail now. */
1829 if( ERROR_SUCCESS
!= r
)
1837 data
[i
] = MSI_RecordGetInteger( rec
, i
+1 );
1838 if ((tv
->columns
[i
].type
&0xff) == 2)
1841 data
[i
] += 0x80000000;
1847 static UINT
msi_row_matches( MSITABLEVIEW
*tv
, UINT row
, UINT
*data
)
1849 UINT i
, r
, x
, ret
= ERROR_FUNCTION_FAILED
;
1851 for( i
=0; i
<tv
->num_cols
; i
++ )
1853 if ( ~tv
->columns
[i
].type
& MSITYPE_KEY
)
1856 /* turn the transform column value into a row value */
1857 r
= TABLE_fetch_int( &tv
->view
, row
, i
+1, &x
);
1858 if ( r
!= ERROR_SUCCESS
)
1860 ERR("TABLE_fetch_int shouldn't fail here\n");
1864 /* if this key matches, move to the next column */
1867 ret
= ERROR_FUNCTION_FAILED
;
1871 ret
= ERROR_SUCCESS
;
1877 static UINT
msi_table_find_row( MSITABLEVIEW
*tv
, MSIRECORD
*rec
, UINT
*row
)
1879 UINT i
, r
= ERROR_FUNCTION_FAILED
, *data
;
1881 data
= msi_record_to_row( tv
, rec
);
1884 for( i
=0; i
<tv
->table
->row_count
; i
++ )
1886 r
= msi_row_matches( tv
, i
, data
);
1887 if( r
== ERROR_SUCCESS
)
1897 static UINT
msi_delete_row( MSITABLEVIEW
*tv
, UINT row
)
1901 for( i
=1; i
<=tv
->num_cols
; i
++ )
1902 TABLE_set_int( tv
, row
, i
, 0 );
1903 return ERROR_SUCCESS
;
1906 static UINT
msi_table_load_transform( MSIDATABASE
*db
, IStorage
*stg
,
1907 string_table
*st
, LPCWSTR name
)
1910 USHORT
*rawdata
= NULL
;
1911 MSITABLEVIEW
*tv
= NULL
;
1912 UINT r
, n
, sz
, i
, mask
;
1913 MSIRECORD
*rec
= NULL
;
1918 TRACE("%p %p %p %s\n", db
, stg
, st
, debugstr_w(name
) );
1920 /* read the transform data */
1921 read_stream_data( stg
, name
, &rawdata
, &rawsize
);
1924 TRACE("table %s empty\n", debugstr_w(name
) );
1925 return ERROR_INVALID_TABLE
;
1928 /* create a table view */
1929 r
= TABLE_CreateView( db
, name
, (MSIVIEW
**) &tv
);
1930 if( r
!= ERROR_SUCCESS
)
1933 r
= tv
->view
.ops
->execute( &tv
->view
, NULL
);
1934 if( r
!= ERROR_SUCCESS
)
1937 TRACE("name = %s columns = %u row_size = %u raw size = %u\n",
1938 debugstr_w(name
), tv
->num_cols
, tv
->row_size
, rawsize
);
1940 /* interpret the data */
1942 for( n
=0; n
< (rawsize
/2); )
1949 * if the low bit is set, columns are continuous and
1950 * the number of columns is specified in the high byte
1952 sz
= 2 + tv
->row_size
;
1957 * If the low bit is not set, rowdata[n] is a bitmask.
1958 * Excepting for key fields, which are always present,
1959 * each bit indicates that a field is present in the transform record.
1961 * rawdata[n] == 0 is a special case ... only the keys will be present
1962 * and it means that this row should be deleted.
1965 for( i
=0; i
<tv
->num_cols
; i
++ )
1967 if( (tv
->columns
[i
].type
& MSITYPE_KEY
) || ((1<<i
)&mask
))
1968 sz
+= bytes_per_column( &tv
->columns
[i
] );
1972 /* check we didn't run of the end of the table */
1973 if ( (n
+sz
) > rawsize
)
1976 dump_table( st
, rawdata
, rawsize
);
1980 rec
= msi_get_transform_record( tv
, st
, &rawdata
[n
] );
1985 TRACE("inserting record\n");
1988 * Native msi seems writes nul into the
1989 * Number (2nd) column of the _Columns table.
1990 * Not sure that it's deliberate...
1992 if (!lstrcmpW(name
, szColumns
))
1997 MSI_RecordGetStringW( rec
, 1, table
, &sz
);
1999 /* reset the column number on a new table */
2000 if ( lstrcmpW(coltable
, table
) )
2003 lstrcpyW( coltable
, table
);
2006 /* fix nul column numbers */
2007 MSI_RecordSetInteger( rec
, 2, ++colcol
);
2010 r
= TABLE_insert_row( &tv
->view
, rec
);
2011 if (r
!= ERROR_SUCCESS
)
2012 ERR("insert row failed\n");
2018 r
= msi_table_find_row( tv
, rec
, &row
);
2019 if (r
!= ERROR_SUCCESS
)
2020 ERR("no matching row to transform\n");
2023 TRACE("modifying row [%d]:\n", row
);
2024 TABLE_set_row( &tv
->view
, row
, rec
, mask
);
2028 TRACE("deleting row [%d]:\n", row
);
2029 msi_delete_row( tv
, row
);
2032 if( TRACE_ON(msidb
) ) dump_record( rec
);
2033 msiobj_release( &rec
->hdr
);
2040 /* no need to free the table, it's associated with the database */
2041 msi_free( rawdata
);
2043 tv
->view
.ops
->delete( &tv
->view
);
2045 return ERROR_SUCCESS
;
2049 * msi_table_apply_transform
2051 * Enumerate the table transforms in a transform storage and apply each one.
2053 UINT
msi_table_apply_transform( MSIDATABASE
*db
, IStorage
*stg
)
2055 IEnumSTATSTG
*stgenum
= NULL
;
2060 string_table
*strings
;
2061 UINT ret
= ERROR_FUNCTION_FAILED
;
2063 TRACE("%p %p\n", db
, stg
);
2065 strings
= load_string_table( stg
);
2069 r
= IStorage_EnumElements( stg
, 0, NULL
, 0, &stgenum
);
2074 * Apply _Tables and _Columns transforms first so that
2075 * the table metadata is correct, and empty tables exist.
2077 ret
= msi_table_load_transform( db
, stg
, strings
, szTables
);
2078 if (ret
!= ERROR_SUCCESS
&& ret
!= ERROR_INVALID_TABLE
)
2081 ret
= msi_table_load_transform( db
, stg
, strings
, szColumns
);
2082 if (ret
!= ERROR_SUCCESS
&& ret
!= ERROR_INVALID_TABLE
)
2085 ret
= ERROR_SUCCESS
;
2087 while( r
== ERROR_SUCCESS
)
2090 r
= IEnumSTATSTG_Next( stgenum
, 1, &stat
, &count
);
2091 if( FAILED( r
) || !count
)
2094 decode_streamname( stat
.pwcsName
, name
);
2095 if ( name
[0] != 0x4840 )
2098 TRACE("transform contains stream %s\n", debugstr_w(name
));
2100 if ( !lstrcmpW( name
+1, szStringPool
) ||
2101 !lstrcmpW( name
+1, szStringData
) ||
2102 !lstrcmpW( name
+1, szColumns
) ||
2103 !lstrcmpW( name
+1, szTables
) )
2106 ret
= msi_table_load_transform( db
, stg
, strings
, name
+1 );
2109 if ( ret
== ERROR_SUCCESS
)
2110 append_storage_to_db( db
, stg
);
2114 IEnumSTATSTG_Release( stgenum
);
2116 msi_destroy_stringtable( strings
);
2121 void append_storage_to_db( MSIDATABASE
*db
, IStorage
*stg
)
2125 t
= msi_alloc( sizeof *t
);
2127 IStorage_AddRef( stg
);
2128 list_add_tail( &db
->transforms
, &t
->entry
);
2131 void msi_free_transforms( MSIDATABASE
*db
)
2133 while( !list_empty( &db
->transforms
) )
2135 MSITRANSFORM
*t
= LIST_ENTRY( list_head( &db
->transforms
),
2136 MSITRANSFORM
, entry
);
2137 list_remove( &t
->entry
);
2138 IStorage_Release( t
->stg
);