[PATCH 1/13] msi: Add support for non-persistent strings.
Robert Shearman
rob at codeweavers.com
Mon Apr 23 02:17:00 CDT 2007
Store the loaded refcount in the string table.
---
dlls/msi/msipriv.h | 12 +++++++---
dlls/msi/string.c | 63
++++++++++++++++++++++++++++++++++++----------------
dlls/msi/table.c | 9 +++++--
3 files changed, 59 insertions(+), 25 deletions(-)
-------------- next part --------------
diff --git a/dlls/msi/msipriv.h b/dlls/msi/msipriv.h
index 55dd2fa..3b5b69f 100644
--- a/dlls/msi/msipriv.h
+++ b/dlls/msi/msipriv.h
@@ -546,8 +546,14 @@ extern HRESULT init_string_table( IStora
/* string table functions */
-extern BOOL msi_addstring( string_table *st, UINT string_no, const CHAR *data, int len, UINT refcount );
-extern BOOL msi_addstringW( string_table *st, UINT string_no, const WCHAR *data, int len, UINT refcount );
+enum StringPersistence
+{
+ StringPersistent = 0,
+ StringNonPersistent = 1
+};
+
+extern BOOL msi_addstring( string_table *st, UINT string_no, const CHAR *data, int len, UINT refcount, enum StringPersistence persistence );
+extern BOOL msi_addstringW( string_table *st, UINT string_no, const WCHAR *data, int len, UINT refcount, enum StringPersistence persistence );
extern UINT msi_id2stringW( string_table *st, UINT string_no, LPWSTR buffer, UINT *sz );
extern UINT msi_id2stringA( string_table *st, UINT string_no, LPSTR buffer, UINT *sz );
@@ -556,7 +562,7 @@ extern UINT msi_string2idA( string_table
extern string_table *msi_init_stringtable( int entries, UINT codepage );
extern VOID msi_destroy_stringtable( string_table *st );
extern UINT msi_string_count( string_table *st );
-extern UINT msi_id_refcount( string_table *st, UINT i );
+extern UINT msi_id_persistent_refcount( string_table *st, UINT i );
extern UINT msi_string_totalsize( string_table *st, UINT *datasize, UINT *poolsize );
extern UINT msi_strcmp( string_table *st, UINT lval, UINT rval, UINT *res );
extern const WCHAR *msi_string_lookup_id( string_table *st, UINT id );
diff --git a/dlls/msi/string.c b/dlls/msi/string.c
index 7a80663..16ed478 100644
--- a/dlls/msi/string.c
+++ b/dlls/msi/string.c
@@ -42,7 +42,8 @@ #define HASH_SIZE 0x101
typedef struct _msistring
{
int hash_next;
- UINT refcount;
+ UINT persistent_refcount;
+ UINT nonpersistent_refcount;
LPWSTR str;
} msistring;
@@ -103,7 +104,8 @@ VOID msi_destroy_stringtable( string_tab
for( i=0; i<st->maxcount; i++ )
{
- if( st->strings[i].refcount )
+ if( st->strings[i].persistent_refcount ||
+ st->strings[i].nonpersistent_refcount )
msi_free( st->strings[i].str );
}
msi_free( st->strings );
@@ -120,11 +122,13 @@ static int st_find_free_entry( string_ta
if( st->freeslot )
{
for( i = st->freeslot; i < st->maxcount; i++ )
- if( !st->strings[i].refcount )
+ if( !st->strings[i].persistent_refcount &&
+ !st->strings[i].nonpersistent_refcount )
return i;
}
for( i = 1; i < st->maxcount; i++ )
- if( !st->strings[i].refcount )
+ if( !st->strings[i].persistent_refcount &&
+ !st->strings[i].nonpersistent_refcount )
return i;
/* dynamically resize */
@@ -135,16 +139,27 @@ static int st_find_free_entry( string_ta
st->strings = p;
st->freeslot = st->maxcount;
st->maxcount = sz;
- if( st->strings[st->freeslot].refcount )
+ if( st->strings[st->freeslot].persistent_refcount ||
+ st->strings[st->freeslot].nonpersistent_refcount )
ERR("oops. expected freeslot to be free...\n");
return st->freeslot;
}
-static void set_st_entry( string_table *st, UINT n, LPWSTR str )
+static void set_st_entry( string_table *st, UINT n, LPWSTR str, UINT refcount, enum StringPersistence persistence )
{
UINT hash = msistring_makehash( str );
- st->strings[n].refcount = 1;
+ if (persistence == StringPersistent)
+ {
+ st->strings[n].persistent_refcount = refcount;
+ st->strings[n].nonpersistent_refcount = 0;
+ }
+ else
+ {
+ st->strings[n].persistent_refcount = 0;
+ st->strings[n].nonpersistent_refcount = refcount;
+ }
+
st->strings[n].str = str;
st->strings[n].hash_next = st->hash[hash];
@@ -154,7 +169,7 @@ static void set_st_entry( string_table *
st->freeslot = n + 1;
}
-int msi_addstring( string_table *st, UINT n, const CHAR *data, int len, UINT refcount )
+int msi_addstring( string_table *st, UINT n, const CHAR *data, int len, UINT refcount, enum StringPersistence persistence )
{
LPWSTR str;
int sz;
@@ -165,14 +180,18 @@ int msi_addstring( string_table *st, UIN
return 0;
if( n > 0 )
{
- if( st->strings[n].refcount )
+ if( st->strings[n].persistent_refcount ||
+ st->strings[n].nonpersistent_refcount )
return -1;
}
else
{
if( ERROR_SUCCESS == msi_string2idA( st, data, &n ) )
{
- st->strings[n].refcount++;
+ if (persistence == StringPersistent)
+ st->strings[n].persistent_refcount += refcount;
+ else
+ st->strings[n].nonpersistent_refcount += refcount;
return n;
}
n = st_find_free_entry( st );
@@ -196,12 +215,12 @@ int msi_addstring( string_table *st, UIN
MultiByteToWideChar( st->codepage, 0, data, len, str, sz );
str[sz] = 0;
- set_st_entry( st, n, str );
+ set_st_entry( st, n, str, refcount, persistence );
return n;
}
-int msi_addstringW( string_table *st, UINT n, const WCHAR *data, int len, UINT refcount )
+int msi_addstringW( string_table *st, UINT n, const WCHAR *data, int len, UINT refcount, enum StringPersistence persistence )
{
LPWSTR str;
@@ -213,14 +232,18 @@ int msi_addstringW( string_table *st, UI
return 0;
if( n > 0 )
{
- if( st->strings[n].refcount )
+ if( st->strings[n].persistent_refcount ||
+ st->strings[n].nonpersistent_refcount )
return -1;
}
else
{
if( ERROR_SUCCESS == msi_string2idW( st, data, &n ) )
{
- st->strings[n].refcount++;
+ if (persistence == StringPersistent)
+ st->strings[n].persistent_refcount += refcount;
+ else
+ st->strings[n].nonpersistent_refcount += refcount;
return n;
}
n = st_find_free_entry( st );
@@ -246,7 +269,7 @@ int msi_addstringW( string_table *st, UI
memcpy( str, data, len*sizeof(WCHAR) );
str[len] = 0;
- set_st_entry( st, n, str );
+ set_st_entry( st, n, str, refcount, persistence );
return n;
}
@@ -261,7 +284,7 @@ const WCHAR *msi_string_lookup_id( strin
if( id >= st->maxcount )
return NULL;
- if( id && !st->strings[id].refcount )
+ if( id && !st->strings[id].persistent_refcount && !st->strings[id].nonpersistent_refcount)
return NULL;
return st->strings[id].str;
@@ -428,18 +451,18 @@ UINT msi_string_count( string_table *st
return st->maxcount;
}
-UINT msi_id_refcount( string_table *st, UINT i )
+UINT msi_id_persistent_refcount( string_table *st, UINT i )
{
if( i >= st->maxcount )
return 0;
- return st->strings[i].refcount;
+ return st->strings[i].persistent_refcount;
}
UINT msi_string_totalsize( string_table *st, UINT *datasize, UINT *poolsize )
{
UINT i, len, max, holesize;
- if( st->strings[0].str || st->strings[0].refcount )
+ if( st->strings[0].str || st->strings[0].persistent_refcount || st->strings[0].nonpersistent_refcount)
ERR("oops. element 0 has a string\n");
*poolsize = 4;
@@ -448,6 +471,8 @@ UINT msi_string_totalsize( string_table
holesize = 0;
for( i=1; i<st->maxcount; i++ )
{
+ if( !st->strings[i].persistent_refcount )
+ continue;
if( st->strings[i].str )
{
TRACE("[%u] = %s\n", i, debugstr_w(st->strings[i].str));
diff --git a/dlls/msi/table.c b/dlls/msi/table.c
index 420519b..1f820dc 100644
--- a/dlls/msi/table.c
+++ b/dlls/msi/table.c
@@ -743,7 +743,7 @@ string_table *load_string_table( IStorag
break;
}
- r = msi_addstring( st, n, data+offset, len, refs );
+ r = msi_addstring( st, n, data+offset, len, refs, StringPersistent );
if( r != n )
ERR("Failed to add string %d\n", n );
n++;
@@ -796,6 +796,9 @@ static UINT save_string_table( MSIDATABA
n = 1;
for( i=1; i<count; i++ )
{
+ UINT refcount = msi_id_persistent_refcount( db->strings, i );
+ if( !refcount )
+ continue;
sz = datasize - used;
r = msi_id2stringA( db->strings, i, data+used, &sz );
if( r != ERROR_SUCCESS )
@@ -807,7 +810,7 @@ static UINT save_string_table( MSIDATABA
sz--;
if (sz)
- pool[ n*2 + 1 ] = msi_id_refcount( db->strings, i );
+ pool[ n*2 + 1 ] = refcount;
else
pool[ n*2 + 1 ] = 0;
if (sz < 0x10000)
@@ -1258,7 +1261,7 @@ static UINT TABLE_set_row( struct tagMSI
else if ( tv->columns[i].type & MSITYPE_STRING )
{
LPCWSTR sval = MSI_RecordGetString( rec, i + 1 );
- val = msi_addstringW( tv->db->strings, 0, sval, -1, 1 );
+ val = msi_addstringW( tv->db->strings, 0, sval, -1, 1, StringPersistent );
}
else if ( 2 == bytes_per_column( &tv->columns[ i ] ) )
{
More information about the wine-patches
mailing list