Fixed array out of bound access.

diff --git a/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c b/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
index 1134140..6da2034 100644 (file)
--- a/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
+++ b/winpr/libwinpr/environment/test/TestEnvironmentGetSetEB.c
@@ -99,11 +99,11 @@ int TestEnvironmentGetSetEB(int argc, char* argv[])
        }
        free(lpszEnvironmentBlockNew);
 
-       lpszEnvironmentBlockNew = (LPTCH) malloc(1024);
+       lpszEnvironmentBlockNew = (LPTCH) malloc(length);
        if (!lpszEnvironmentBlockNew)
                return -1;
 
-       memcpy(lpszEnvironmentBlockNew,lpszEnvironmentBlock,56);
+       memcpy(lpszEnvironmentBlockNew,lpszEnvironmentBlock,length);
 
        /* Set variable in empty environment block */
        if (SetEnvironmentVariableEBA(&lpszEnvironmentBlockNew, "test", "5"))