Fixed zero length allocation.

diff --git a/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c b/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
index 6da2034..3a516b3 100644 (file)
--- a/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
+++ b/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
@@ -98,8 +98,11 @@ int TestEnvironmentGetSetEB(int argc, char* argv[])
                }
        }
        free(lpszEnvironmentBlockNew);
+       lpszEnvironmentBlockNew = NULL;
+
+       if (length > 0)
+               lpszEnvironmentBlockNew = (LPTCH) malloc(length);
 
-       lpszEnvironmentBlockNew = (LPTCH) malloc(length);
        if (!lpszEnvironmentBlockNew)
                return -1;