Robert Reif : advapi32: LookupAccountSid buffer size query tests.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: ca80c55ae89aa3ae049af86e6de0754b1b4d89a0
URL:    http://source.winehq.org/git/?p=wine.git;a=commit;h=ca80c55ae89aa3ae049af86e...

Author: Robert Reif <reif(a)earthlink.net>
Date:   Sat Aug 19 09:17:30 2006 -0400

advapi32: LookupAccountSid buffer size query tests.

---

 dlls/advapi32/tests/security.c |   56 +++++++++++++++++++++++++++++++++++++++-
 1 files changed, 54 insertions(+), 2 deletions(-)

diff --git a/dlls/advapi32/tests/security.c b/dlls/advapi32/tests/security.c
index 9cf2af6..0a6a193 100644
--- a/dlls/advapi32/tests/security.c
+++ b/dlls/advapi32/tests/security.c
@@ -880,6 +880,7 @@ static void test_LookupAccountSid(void)
     SID_IDENTIFIER_AUTHORITY SIDAuthNT = { SECURITY_NT_AUTHORITY };
     char account[MAX_PATH], domain[MAX_PATH];
     DWORD acc_size, dom_size;
+    DWORD real_acc_size, real_dom_size;
     PSID pUsersSid = NULL;
     SID_NAME_USE use;
     BOOL ret;
@@ -899,18 +900,69 @@ static void test_LookupAccountSid(void)
     if (!ret && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
         return;
 
+    real_acc_size = MAX_PATH;
+    real_dom_size = MAX_PATH;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &real_acc_size, domain, &real_dom_size, &use);
+    ok(ret, "LookupAccountSid() Expected TRUE, got FALSE\n");
+
     /* try NULL account */
     acc_size = MAX_PATH;
     dom_size = MAX_PATH;
-    ret = LookupAccountSid(NULL, pUsersSid, NULL, &acc_size, domain, &dom_size, &use);
+    ret = LookupAccountSidA(NULL, pUsersSid, NULL, &acc_size, domain, &dom_size, &use);
     ok(ret, "LookupAccountSid() Expected TRUE, got FALSE\n");
 
     /* try NULL domain */
     acc_size = MAX_PATH;
     dom_size = MAX_PATH;
-    ret = LookupAccountSid(NULL, pUsersSid, account, &acc_size, NULL, &dom_size, &use);
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, NULL, &dom_size, &use);
     ok(ret, "LookupAccountSid() Expected TRUE, got FALSE\n");
 
+    /* try a small account buffer */
+    acc_size = 1;
+    dom_size = MAX_PATH;
+    account[0] = 0;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, domain, &dom_size, &use);
+    ok(!ret, "LookupAccountSid() Expected FALSE got TRUE\n");
+    ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY, "LookupAccountSid() Expected ERROR_NOT_ENOUGH_MEMORY, got %lu\n", GetLastError());
+
+    /* try a 0 sized account buffer */
+    acc_size = 0;
+    dom_size = MAX_PATH;
+    account[0] = 0;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, domain, &dom_size, &use);
+    /* this can fail or succeed depending on OS version but the size will always be returned */
+    ok(acc_size == real_acc_size, "LookupAccountSid() Expected acc_size = %lu, got %lu\n", real_acc_size, acc_size);
+
+    /* try a 0 sized account buffer */
+    acc_size = 0;
+    dom_size = MAX_PATH;
+    ret = LookupAccountSidA(NULL, pUsersSid, NULL, &acc_size, domain, &dom_size, &use);
+    /* this can fail or succeed depending on OS version but the size will always be returned */
+    ok(acc_size == real_acc_size, "LookupAccountSid() Expected acc_size = %lu, got %lu\n", real_acc_size, acc_size);
+
+    /* try a small domain buffer */
+    dom_size = 1;
+    acc_size = MAX_PATH;
+    account[0] = 0;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, domain, &dom_size, &use);
+    ok(!ret, "LookupAccountSid() Expected FALSE got TRUE\n");
+    ok(GetLastError() == ERROR_NOT_ENOUGH_MEMORY, "LookupAccountSid() Expected ERROR_NOT_ENOUGH_MEMORY, got %lu\n", GetLastError());
+
+    /* try a 0 sized domain buffer */
+    dom_size = 0;
+    acc_size = MAX_PATH;
+    account[0] = 0;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, domain, &dom_size, &use);
+    /* this can fail or succeed depending on OS version but the size will always be returned */
+    ok(dom_size == real_dom_size, "LookupAccountSid() Expected dom_size = %lu, got %lu\n", real_dom_size, dom_size);
+
+    /* try a 0 sized domain buffer */
+    dom_size = 0;
+    acc_size = MAX_PATH;
+    ret = LookupAccountSidA(NULL, pUsersSid, account, &acc_size, NULL, &dom_size, &use);
+    /* this can fail or succeed depending on OS version but the size will always be returned */
+    ok(acc_size == real_acc_size, "LookupAccountSid() Expected dom_size = %lu, got %lu\n", real_dom_size, dom_size);
+
     pCreateWellKnownSid = (fnCreateWellKnownSid)GetProcAddress( hmod, "CreateWellKnownSid" );
 
     if (pCreateWellKnownSid && pConvertSidToStringSidA)