[SHLWAPI_WINETEST] Addendum to r65749.

CORE-7556

svn path=/trunk/; revision=65750
This commit is contained in:
Amine Khaldi
2014-12-19 13:22:15 +00:00
parent cfcca647bc
commit 736374efcd
+122
View File
@@ -53,6 +53,7 @@ static DWORD (WINAPI *pSHAnsiToAnsi)(LPCSTR,LPSTR,int);
static DWORD (WINAPI *pSHUnicodeToUnicode)(LPCWSTR,LPWSTR,int);
static LPSTR (WINAPI *pStrCatBuffA)(LPSTR,LPCSTR,INT);
static LPWSTR (WINAPI *pStrCatBuffW)(LPWSTR,LPCWSTR,INT);
static DWORD (WINAPI *pStrCatChainW)(LPWSTR,DWORD,DWORD,LPCWSTR);
static LPSTR (WINAPI *pStrCpyNXA)(LPSTR,LPCSTR,int);
static LPWSTR (WINAPI *pStrCpyNXW)(LPWSTR,LPCWSTR,int);
static LPSTR (WINAPI *pStrFormatByteSize64A)(LONGLONG,LPSTR,UINT);
@@ -1486,6 +1487,125 @@ static void test_StrStrNIW(void)
}
}
static void test_StrCatChainW(void)
{
static const WCHAR deadbeefW[] = {'D','e','A','d','B','e','E','f',0};
static const WCHAR deadW[] = {'D','e','A','d',0};
static const WCHAR beefW[] = {'B','e','E','f',0};
WCHAR buf[32 + 1];
DWORD ret;
if (!pStrCatChainW)
{
win_skip("StrCatChainW is not available\n");
return;
}
/* Test with NULL buffer */
ret = pStrCatChainW(NULL, 0, 0, beefW);
ok(ret == 0, "Expected StrCatChainW to return 0, got %u\n", ret);
/* Test with empty buffer */
memset(buf, 0x11, sizeof(buf));
ret = pStrCatChainW(buf, 0, 0, beefW);
ok(ret == 0, "Expected StrCatChainW to return 0, got %u\n", ret);
ok(buf[0] == 0x1111, "Expected buf[0] = 0x1111, got %x\n", buf[0]);
memcpy(buf, deadbeefW, sizeof(deadbeefW));
ret = pStrCatChainW(buf, 0, -1, beefW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
/* Append data to existing string with offset = -1 */
memset(buf, 0x11, sizeof(buf));
ret = pStrCatChainW(buf, 32, 0, deadW);
ok(ret == 4, "Expected StrCatChainW to return 4, got %u\n", ret);
ok(!memcmp(buf, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
ret = pStrCatChainW(buf, 32, -1, beefW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
/* Append data at a fixed offset */
memset(buf, 0x11, sizeof(buf));
ret = pStrCatChainW(buf, 32, 0, deadW);
ok(ret == 4, "Expected StrCatChainW to return 4, got %u\n", ret);
ok(!memcmp(buf, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
ret = pStrCatChainW(buf, 32, 4, beefW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
/* Buffer exactly sufficient for string + terminating null */
memset(buf, 0x11, sizeof(buf));
ret = pStrCatChainW(buf, 5, 0, deadW);
ok(ret == 4, "Expected StrCatChainW to return 4, got %u\n", ret);
ok(!memcmp(buf, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
/* Buffer too small, string will be truncated */
memset(buf, 0x11, sizeof(buf));
ret = pStrCatChainW(buf, 4, 0, deadW);
if (ret == 4)
{
/* Windows 2000 and XP uses a slightly different implementation
* for StrCatChainW, which doesn't ensure that strings are null-
* terminated. Skip test if we detect such an implementation. */
win_skip("Windows2000/XP behaviour detected for StrCatChainW, skipping tests\n");
return;
}
ok(ret == 3, "Expected StrCatChainW to return 3, got %u\n", ret);
ok(!memcmp(buf, deadW, 3 * sizeof(WCHAR)), "Buffer contains wrong data\n");
ok(!buf[3], "String is not nullterminated\n");
ok(buf[4] == 0x1111, "Expected buf[4] = 0x1111, got %x\n", buf[4]);
/* Overwrite part of an existing string */
ret = pStrCatChainW(buf, 4, 1, beefW);
ok(ret == 3, "Expected StrCatChainW to return 3, got %u\n", ret);
ok(buf[0] == 'D', "Expected buf[0] = 'D', got %x\n", buf[0]);
ok(buf[1] == 'B', "Expected buf[1] = 'B', got %x\n", buf[1]);
ok(buf[2] == 'e', "Expected buf[2] = 'e', got %x\n", buf[2]);
ok(!buf[3], "String is not nullterminated\n");
ok(buf[4] == 0x1111, "Expected buf[4] = 0x1111, got %x\n", buf[4]);
/* Test appending to full buffer */
memset(buf, 0x11, sizeof(buf));
memcpy(buf, deadbeefW, sizeof(deadbeefW));
memcpy(buf + 9, deadW, sizeof(deadW));
ret = pStrCatChainW(buf, 9, 8, beefW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
ok(!memcmp(buf + 9, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
/* Offset points at the end of the buffer */
ret = pStrCatChainW(buf, 9, 9, beefW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
ok(!memcmp(buf + 9, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
/* Offset points outside of the buffer */
ret = pStrCatChainW(buf, 9, 10, beefW);
ok(ret == 10, "Expected StrCatChainW to return 10, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
ok(!memcmp(buf + 9, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
/* The same but without nullterminated string */
memcpy(buf, deadbeefW, sizeof(deadbeefW));
ret = pStrCatChainW(buf, 5, -1, deadW);
ok(ret == 8, "Expected StrCatChainW to return 8, got %u\n", ret);
ok(!memcmp(buf, deadbeefW, sizeof(deadbeefW)), "Buffer contains wrong data\n");
ret = pStrCatChainW(buf, 5, 5, deadW);
ok(ret == 4, "Expected StrCatChainW to return 4, got %u\n", ret);
ok(!memcmp(buf, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
ok(buf[5] == 'e', "Expected buf[5] = 'e', got %x\n", buf[5]);
ret = pStrCatChainW(buf, 5, 6, deadW);
ok(ret == 6, "Expected StrCatChainW to return 6, got %u\n", ret);
ok(!memcmp(buf, deadW, sizeof(deadW)), "Buffer contains wrong data\n");
ok(buf[5] == 'e', "Expected buf[5] = 'e', got %x\n", buf[5]);
}
START_TEST(string)
{
HMODULE hShlwapi;
@@ -1505,6 +1625,7 @@ START_TEST(string)
pSHUnicodeToUnicode = (void *)GetProcAddress(hShlwapi, (LPSTR)346);
pStrCatBuffA = (void *)GetProcAddress(hShlwapi, "StrCatBuffA");
pStrCatBuffW = (void *)GetProcAddress(hShlwapi, "StrCatBuffW");
pStrCatChainW = (void *)GetProcAddress(hShlwapi, "StrCatChainW");
pStrCpyNXA = (void *)GetProcAddress(hShlwapi, (LPSTR)399);
pStrCpyNXW = (void *)GetProcAddress(hShlwapi, (LPSTR)400);
pStrChrNW = (void *)GetProcAddress(hShlwapi, "StrChrNW");
@@ -1570,6 +1691,7 @@ START_TEST(string)
test_StrStrIW();
test_StrStrNW();
test_StrStrNIW();
test_StrCatChainW();
CoUninitialize();
}