From 5bcad333b7b63f85ff8d69a7d0d7eb7466704533 Mon Sep 17 00:00:00 2001
From: Dmitry Kislyuk <dimaki at rocketmail.com>
Date: Wed, 6 May 2009 23:11:35 -0300
Subject: kernel32/tests: Show NULLs need to be preserved in data before first
 section

---
 dlls/kernel32/tests/profile.c |   36 +++++++++++++++++++++++++++++-------
 1 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/dlls/kernel32/tests/profile.c b/dlls/kernel32/tests/profile.c
index 15a3902..3df6637 100644
--- a/dlls/kernel32/tests/profile.c
+++ b/dlls/kernel32/tests/profile.c
@@ -824,11 +824,10 @@ static void test_GetPrivateProfileString(const char *content, const char *descri
 
 static DWORD timeout = 0;
 
-static BOOL check_file_data(LPCSTR path, LPCSTR data)
+static BOOL check_binary_file_data(LPCSTR path, const VOID *data, DWORD size)
 {
     HANDLE file;
     CHAR buf[MAX_PATH];
-    DWORD size;
     BOOL ret;
 
     /* Sleep() is needed on Win9x and WinME */
@@ -839,17 +838,23 @@ static BOOL check_file_data(LPCSTR path, LPCSTR data)
     if (file == INVALID_HANDLE_VALUE)
       return FALSE;
 
-    size = GetFileSize(file, NULL);
-    buf[size] = '\0';
+    if(size != GetFileSize(file, NULL) )
+    {
+        CloseHandle(file);
+        return FALSE;
+    }
+
     ret = ReadFile(file, buf, size, &size, NULL);
     CloseHandle(file);
     if (!ret)
       return FALSE;
 
-    if (!*data && size != 0)
-        return FALSE;
+    return !memcmp(buf, data, size);
+}
 
-    return !lstrcmpA(buf, data);
+static BOOL check_file_data(LPCSTR path, LPCSTR data)
+{
+    return check_binary_file_data(path, data, lstrlenA(data));
 }
 
 static void test_WritePrivateProfileString(void)
@@ -1039,8 +1044,25 @@ static void test_WritePrivateProfileString(void)
     ret = WritePrivateProfileStringA("App3", "key5", NULL, path);
     ok(ret == TRUE, "Expected TRUE, got %d\n", ret);
     ok(check_file_data(path, data), "File doesn't match\n");
+    DeleteFileA(path);
 
+    /* NULLs in file before first section. Should be preserved in output */
+    data = "Data \0 before \0 first \0 section"    /* 31 bytes */
+           "\r\n[section1]\r\n"                    /* 14 bytes */
+           "key1=string1\r\n";                     /* 14 bytes */
+    GetTempFileNameA(temp, "wine", 0, path);
+    create_test_file(path, data, 31);
+    ret = WritePrivateProfileStringA("section1", "key1", "string1", path);
+    ok(ret == TRUE, "Expected TRUE, got %d\n", ret);
+    todo_wine
+    ok( check_binary_file_data(path, data, 59) ||
+        broken( check_binary_file_data(path,        /* Windows 9x */
+            "Data \0 before \0 first \0 section"    /* 31 bytes */
+            "\r\n\r\n[section1]\r\n"                /* 14 bytes */
+            "key1=string1"                          /* 14 bytes */
+            , 59)), "File doesn't match\n"); 
     DeleteFileA(path);
+
 }
 
 START_TEST(profile)
-- 
1.5.6.3