advapi32: Add tests for credential management functions.

dlls/advapi32/tests/Makefile.in

@@ -7,6 +7,7 @@ IMPORTS   = ole32 advapi32 kernel32
 EXTRALIBS = -luuid
 
 CTESTS = \
+	cred.c \
 	crypt.c \
 	crypt_lmhash.c \
 	crypt_md4.c \

dlls/advapi32/tests/cred.c

+/*
+ * Credential Function Tests
+ *
+ * Copyright 2007 Robert Shearman
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <stdarg.h>
+#include <stdio.h>
+
+#include "windef.h"
+#include "winbase.h"
+#include "wincred.h"
+
+#include "wine/test.h"
+
+static BOOL (WINAPI *pCredDeleteA)(LPCSTR,DWORD,DWORD);
+static BOOL (WINAPI *pCredEnumerateA)(LPCSTR,DWORD,DWORD *,PCREDENTIALA **);
+static VOID (WINAPI *pCredFree)(PVOID);
+static BOOL (WINAPI *pCredReadA)(LPCSTR,DWORD,DWORD,PCREDENTIALA *);
+static BOOL (WINAPI *pCredRenameA)(LPCSTR,LPCSTR,DWORD,DWORD);
+static BOOL (WINAPI *pCredWriteA)(PCREDENTIALA,DWORD);
+
+#define TEST_TARGET_NAME  "credtest.winehq.org"
+#define TEST_TARGET_NAME2 "credtest2.winehq.org"
+
+static void test_CredReadA(void)
+{
+    BOOL ret;
+    PCREDENTIALA cred;
+
+    SetLastError(0xdeadbeef);
+    ret = pCredReadA(TEST_TARGET_NAME, -1, 0, &cred);
+    ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
+        "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
+        GetLastError());
+
+    SetLastError(0xdeadbeef);
+    ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef, &cred);
+    ok(!ret && GetLastError() == ERROR_INVALID_FLAGS,
+        "CredReadA should have failed with ERROR_INVALID_FLAGS instead of %d\n",
+        GetLastError());
+
+    SetLastError(0xdeadbeef);
+    ret = pCredReadA(NULL, CRED_TYPE_GENERIC, 0, &cred);
+    ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
+        "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
+        GetLastError());
+}
+
+static void test_CredWriteA(void)
+{
+    CREDENTIALA new_cred;
+    BOOL ret;
+
+    SetLastError(0xdeadbeef);
+    ret = pCredWriteA(NULL, 0);
+    ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
+        "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
+        GetLastError());
+
+    new_cred.Flags = 0;
+    new_cred.Type = CRED_TYPE_GENERIC;
+    new_cred.TargetName = NULL;
+    new_cred.Comment = (char *)"Comment";
+    new_cred.CredentialBlobSize = 0;
+    new_cred.CredentialBlob = NULL;
+    new_cred.Persist = CRED_PERSIST_ENTERPRISE;
+    new_cred.AttributeCount = 0;
+    new_cred.Attributes = NULL;
+    new_cred.TargetAlias = NULL;
+    new_cred.UserName = (char *)"winetest";
+
+    SetLastError(0xdeadbeef);
+    ret = pCredWriteA(&new_cred, 0);
+    ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
+        "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
+        GetLastError());
+
+    new_cred.TargetName = (char *)TEST_TARGET_NAME;
+    new_cred.Type = CRED_TYPE_DOMAIN_PASSWORD;
+
+    SetLastError(0xdeadbeef);
+    ret = pCredWriteA(&new_cred, 0);
+    ok(!ret && GetLastError() == ERROR_BAD_USERNAME,
+        "CredWrite with username without domain should return ERROR_BAD_USERNAME not %d\n", GetLastError());
+
+    new_cred.UserName = NULL;
+    SetLastError(0xdeadbeef);
+    ret = pCredWriteA(&new_cred, 0);
+    ok(!ret && GetLastError() == ERROR_BAD_USERNAME,
+        "CredWriteA with NULL username should have failed with ERROR_BAD_USERNAME instead of %d\n",
+        GetLastError());
+}
+
+static void test_CredDeleteA(void)
+{
+    BOOL ret;
+
+    SetLastError(0xdeadbeef);
+    ret = pCredDeleteA(TEST_TARGET_NAME, -1, 0);
+    ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
+        "CredDeleteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
+        GetLastError());
+
+    SetLastError(0xdeadbeef);
+    ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef);
+    ok(!ret && GetLastError() == ERROR_INVALID_FLAGS,
+        "CredDeleteA should have failed with ERROR_INVALID_FLAGS instead of %d\n",
+        GetLastError());
+}
+
+static void test_generic(void)
+{
+    BOOL ret;
+    DWORD count, i;
+    PCREDENTIALA *creds;
+    CREDENTIALA new_cred;
+    PCREDENTIALA cred;
+    static const WCHAR password[] = {'p','4','$','$','w','0','r','d','!',0};
+    BOOL found = FALSE;
+
+    new_cred.Flags = 0;
+    new_cred.Type = CRED_TYPE_GENERIC;
+    new_cred.TargetName = (char *)TEST_TARGET_NAME;
+    new_cred.Comment = (char *)"Comment";
+    new_cred.CredentialBlobSize = sizeof(password);
+    new_cred.CredentialBlob = (LPBYTE)password;
+    new_cred.Persist = CRED_PERSIST_ENTERPRISE;
+    new_cred.AttributeCount = 0;
+    new_cred.Attributes = NULL;
+    new_cred.TargetAlias = NULL;
+    new_cred.UserName = (char *)"winetest";
+
+    ret = pCredWriteA(&new_cred, 0);
+    ok(ret, "CredWriteA failed with error %d\n", GetLastError());
+
+    ret = pCredEnumerateA(NULL, 0, &count, &creds);
+    ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
+
+    for (i = 0; i < count; i++)
+    {
+        if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
+        {
+            ok(creds[i]->Type == CRED_TYPE_GENERIC, "expected creds[%d]->Type CRED_TYPE_GENERIC but got %d\n", i, creds[i]->Type);
+            ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
+            ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
+            ok(creds[i]->CredentialBlobSize == sizeof(password), "wrong CredentialBlobSize %d\n", creds[i]->CredentialBlobSize);
+            ok(creds[i]->CredentialBlob != NULL, "CredentialBlob should be present\n");
+            if (creds[i]->CredentialBlob)
+                ok(!memcmp(creds[i]->CredentialBlob, password, sizeof(password)), "credentials don't match\n");
+            ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
+            ok(!strcmp(creds[i]->UserName, "winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
+            found = TRUE;
+        }
+    }
+    pCredFree(creds);
+    ok(found, "credentials not found\n");
+
+    ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0, &cred);
+    ok(ret, "CredReadA failed with error %d\n", GetLastError());
+    pCredFree(cred);
+
+    ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0);
+    ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
+}
+
+static void test_domain_password(void)
+{
+    BOOL ret;
+    DWORD count, i;
+    PCREDENTIALA *creds;
+    CREDENTIALA new_cred;
+    PCREDENTIALA cred;
+    static const WCHAR password[] = {'p','4','$','$','w','0','r','d','!',0};
+    BOOL found = FALSE;
+
+    new_cred.Flags = 0;
+    new_cred.Type = CRED_TYPE_DOMAIN_PASSWORD;
+    new_cred.TargetName = (char *)TEST_TARGET_NAME;
+    new_cred.Comment = (char *)"Comment";
+    new_cred.CredentialBlobSize = sizeof(password);
+    new_cred.CredentialBlob = (LPBYTE)password;
+    new_cred.Persist = CRED_PERSIST_ENTERPRISE;
+    new_cred.AttributeCount = 0;
+    new_cred.Attributes = NULL;
+    new_cred.TargetAlias = NULL;
+    new_cred.UserName = (char *)"test\\winetest";
+    ret = pCredWriteA(&new_cred, 0);
+    ok(ret, "CredWriteA failed with error %d\n", GetLastError());
+
+    ret = pCredEnumerateA(NULL, 0, &count, &creds);
+    ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
+
+    for (i = 0; i < count; i++)
+    {
+        if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
+        {
+            ok(creds[i]->Type == CRED_TYPE_DOMAIN_PASSWORD, "expected creds[%d]->Type CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
+            ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
+            ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
+            todo_wine
+            ok(creds[i]->CredentialBlobSize == 0, "expected CredentialBlobSize of 0 but got %d\n", creds[i]->CredentialBlobSize);
+            todo_wine
+            ok(!creds[i]->CredentialBlob, "expected NULL credentials but got %p\n", creds[i]->CredentialBlob);
+            ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
+            ok(!strcmp(creds[i]->UserName, "test\\winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
+            found = TRUE;
+        }
+    }
+    pCredFree(creds);
+    ok(found, "credentials not found\n");
+
+    ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_DOMAIN_PASSWORD, 0, &cred);
+    ok(ret, "CredReadA failed with error %d\n", GetLastError());
+    todo_wine
+    ok(cred->CredentialBlobSize == 0, "expected CredentialBlobSize of 0 but got %d\n", cred->CredentialBlobSize);
+    todo_wine
+    ok(!cred->CredentialBlob, "expected NULL credentials but got %p\n", cred->CredentialBlob);
+    pCredFree(cred);
+
+    ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_DOMAIN_PASSWORD, 0);
+    ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
+}
+
+START_TEST(cred)
+{
+    pCredEnumerateA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredEnumerateA");
+    pCredFree = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredFree");
+    pCredWriteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredWriteA");
+    pCredDeleteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredDeleteA");
+    pCredReadA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadA");
+    pCredRenameA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredRenameA");
+
+    if (!pCredEnumerateA || !pCredFree || !pCredWriteA || !pCredDeleteA ||
+        !pCredReadA)
+    {
+        skip("credentials functions not present in advapi32.dll\n");
+        return;
+    }
+
+    test_CredReadA();
+    test_CredWriteA();
+    test_CredDeleteA();
+
+    trace("generic:\n");
+    test_generic();
+    trace("domain password:\n");
+    test_domain_password();
+}