[ros-diffs] [reactos] 327/360: [WINESYNC] msi: Check for a NULL return from deformat_string() when checking for failure.

[winesync]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=de80f2a3c11778003e075…

commit de80f2a3c11778003e075a4cb655d82f45b5208b
Author:     winesync <ros-dev(a)reactos.org>
AuthorDate: Sun Mar 13 23:41:13 2022 +0100
Commit:     Mark Jansen <mark.jansen(a)reactos.org>
CommitDate: Sun Mar 20 19:28:39 2022 +0100

    [WINESYNC] msi: Check for a NULL return from deformat_string() when checking for
failure.
    
    An empty string is valid, and can occur if e.g. a nonexistent property is
    expanded.
    
    Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=51091
    Signed-off-by: Zebediah Figura <z.figura12(a)gmail.com>
    Signed-off-by: Hans Leidekker <hans(a)codeweavers.com>
    Signed-off-by: Alexandre Julliard <julliard(a)winehq.org>
    
    wine commit id 814f18ab756cd69158ce893c4f7898e14d3e46dc by Zebediah Figura
<z.figura12(a)gmail.com>
---
 dll/win32/msi/action.c                  | 69 +++++++++++++++++++++------------
 modules/rostests/winetests/msi/action.c |  8 +++-
 2 files changed, 52 insertions(+), 25 deletions(-)

diff --git a/dll/win32/msi/action.c b/dll/win32/msi/action.c
index 40d0596a717..f7186446203 100644
--- a/dll/win32/msi/action.c
+++ b/dll/win32/msi/action.c
@@ -6818,25 +6818,34 @@ static UINT ITERATE_WriteEnvironmentString( MSIRECORD *rec, LPVOID
param )
     if (res != ERROR_SUCCESS || !value)
        goto done;
 
-    if (value && !deformat_string(package, value, &deformatted))
+    if (value)
     {
-        res = ERROR_OUTOFMEMORY;
-        goto done;
-    }
+        DWORD len = deformat_string( package, value, &deformatted );
+        if (!deformatted)
+        {
+            res = ERROR_OUTOFMEMORY;
+            goto done;
+        }
 
-    if ((value = deformatted))
-    {
-        if (flags & ENV_MOD_PREFIX)
+        if (len)
         {
-            p = wcsrchr( value, ';' );
-            len_value = p - value;
+            value = deformatted;
+            if (flags & ENV_MOD_PREFIX)
+            {
+                p = wcsrchr( value, ';' );
+                len_value = p - value;
+            }
+            else if (flags & ENV_MOD_APPEND)
+            {
+                value = wcschr( value, ';' ) + 1;
+                len_value = lstrlenW( value );
+            }
+            else len_value = lstrlenW( value );
         }
-        else if (flags & ENV_MOD_APPEND)
+        else
         {
-            value = wcschr( value, ';' ) + 1;
-            len_value = lstrlenW( value );
+            value = NULL;
         }
-        else len_value = lstrlenW( value );
     }
 
     res = open_env_key( flags, &env );
@@ -7025,22 +7034,34 @@ static UINT ITERATE_RemoveEnvironmentString( MSIRECORD *rec,
LPVOID param )
         return ERROR_SUCCESS;
     }
 
-    if (value && !deformat_string( package, value, &deformatted ))
-        return ERROR_OUTOFMEMORY;
-
-    if ((value = deformatted))
+    if (value)
     {
-        if (flags & ENV_MOD_PREFIX)
+        DWORD len = deformat_string( package, value, &deformatted );
+        if (!deformatted)
+        {
+            res = ERROR_OUTOFMEMORY;
+            goto done;
+        }
+
+        if (len)
         {
-            p = wcschr( value, ';' );
-            len_value = p - value;
+            value = deformatted;
+            if (flags & ENV_MOD_PREFIX)
+            {
+                p = wcsrchr( value, ';' );
+                len_value = p - value;
+            }
+            else if (flags & ENV_MOD_APPEND)
+            {
+                value = wcschr( value, ';' ) + 1;
+                len_value = lstrlenW( value );
+            }
+            else len_value = lstrlenW( value );
         }
-        else if (flags & ENV_MOD_APPEND)
+        else
         {
-            value = wcschr( value, ';' ) + 1;
-            len_value = lstrlenW( value );
+            value = NULL;
         }
-        else len_value = lstrlenW( value );
     }
 
     r = open_env_key( flags, &env );
diff --git a/modules/rostests/winetests/msi/action.c
b/modules/rostests/winetests/msi/action.c
index ce3b47cfe02..2c5f5b0dc40 100644
--- a/modules/rostests/winetests/msi/action.c
+++ b/modules/rostests/winetests/msi/action.c
@@ -252,7 +252,9 @@ static const char env_environment_dat[] =
     "Var27\t+-MSITESTVAR21\t[~];1\tOne\n"
     "Var28\t-MSITESTVAR22\t1\tOne\n"
     "Var29\t-MSITESTVAR23\t2\tOne\n"
-    "Var30\t*MSITESTVAR100\t1\tOne\n";
+    "Var30\t*MSITESTVAR100\t1\tOne\n"
+    "Var31\t-=MSITESTVAR24\t[SERVNAME]\tOne\n"
+    "Var32\t-=MSITESTVAR25\t[bogus_prop]\tOne\n";
 
 static const char service_install_dat[] =
    
"ServiceInstall\tName\tDisplayName\tServiceType\tStartType\tErrorControl\t"
@@ -5047,6 +5049,7 @@ static void test_envvar(void)
     CHECK_REG_STR(env, "MSITESTVAR19", "1");
     CHECK_REG_STR(env, "MSITESTVAR20", "1");
     CHECK_REG_STR(env, "MSITESTVAR21", "1");
+    CHECK_REG_STR(env, "MSITESTVAR24", "TestService");
     CHECK_REG_STR(env2, "MSITESTVAR100", "1");
 
     res = RegSetValueExA(env, "MSITESTVAR22", 0, REG_SZ, (const BYTE
*)"1", 2);
@@ -5055,6 +5058,9 @@ static void test_envvar(void)
     res = RegSetValueExA(env, "MSITESTVAR23", 0, REG_SZ, (const BYTE
*)"1", 2);
     ok(res == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %d\n", res);
 
+    res = RegDeleteValueA(env, "MSITESTVAR25");
+    ok(res == ERROR_FILE_NOT_FOUND, "Expected ERROR_FILE_NOT_FOUND, got %d\n",
res);
+
     r = MsiInstallProductA(msifile, "REMOVE=ALL");
     ok(!r, "got %u\n", r);