1 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/http/mock_sspi_library_win.h"
6
7 #include "testing/gtest/include/gtest/gtest.h"
8
9 namespace net {
10
MockSSPILibrary()11 MockSSPILibrary::MockSSPILibrary() {
12 }
13
~MockSSPILibrary()14 MockSSPILibrary::~MockSSPILibrary() {
15 EXPECT_TRUE(expected_package_queries_.empty());
16 EXPECT_TRUE(expected_freed_packages_.empty());
17 }
18
AcquireCredentialsHandle(LPWSTR pszPrincipal,LPWSTR pszPackage,unsigned long fCredentialUse,void * pvLogonId,void * pvAuthData,SEC_GET_KEY_FN pGetKeyFn,void * pvGetKeyArgument,PCredHandle phCredential,PTimeStamp ptsExpiry)19 SECURITY_STATUS MockSSPILibrary::AcquireCredentialsHandle(
20 LPWSTR pszPrincipal,
21 LPWSTR pszPackage,
22 unsigned long fCredentialUse,
23 void* pvLogonId,
24 void* pvAuthData,
25 SEC_GET_KEY_FN pGetKeyFn,
26 void* pvGetKeyArgument,
27 PCredHandle phCredential,
28 PTimeStamp ptsExpiry) {
29 // Fill in phCredential with arbitrary value.
30 phCredential->dwLower = phCredential->dwUpper = ((ULONG_PTR) ((INT_PTR)0));
31 return SEC_E_OK;
32 }
33
InitializeSecurityContext(PCredHandle phCredential,PCtxtHandle phContext,SEC_WCHAR * pszTargetName,unsigned long fContextReq,unsigned long Reserved1,unsigned long TargetDataRep,PSecBufferDesc pInput,unsigned long Reserved2,PCtxtHandle phNewContext,PSecBufferDesc pOutput,unsigned long * contextAttr,PTimeStamp ptsExpiry)34 SECURITY_STATUS MockSSPILibrary::InitializeSecurityContext(
35 PCredHandle phCredential,
36 PCtxtHandle phContext,
37 SEC_WCHAR* pszTargetName,
38 unsigned long fContextReq,
39 unsigned long Reserved1,
40 unsigned long TargetDataRep,
41 PSecBufferDesc pInput,
42 unsigned long Reserved2,
43 PCtxtHandle phNewContext,
44 PSecBufferDesc pOutput,
45 unsigned long* contextAttr,
46 PTimeStamp ptsExpiry) {
47 // Fill in the outbound buffer with garbage data.
48 PSecBuffer out_buffer = pOutput->pBuffers;
49 out_buffer->cbBuffer = 2;
50 uint8* buf = reinterpret_cast<uint8 *>(out_buffer->pvBuffer);
51 buf[0] = 0xAB;
52 buf[1] = 0xBA;
53
54 // Fill in phNewContext with arbitrary value if it's invalid.
55 if (phNewContext != phContext)
56 phNewContext->dwLower = phNewContext->dwUpper = ((ULONG_PTR) ((INT_PTR)0));
57 return SEC_E_OK;
58 }
59
QuerySecurityPackageInfo(LPWSTR pszPackageName,PSecPkgInfoW * pkgInfo)60 SECURITY_STATUS MockSSPILibrary::QuerySecurityPackageInfo(
61 LPWSTR pszPackageName, PSecPkgInfoW *pkgInfo) {
62 EXPECT_TRUE(!expected_package_queries_.empty());
63 PackageQuery package_query = expected_package_queries_.front();
64 expected_package_queries_.pop_front();
65 std::wstring actual_package(pszPackageName);
66 EXPECT_EQ(package_query.expected_package, actual_package);
67 *pkgInfo = package_query.package_info;
68 if (package_query.response_code == SEC_E_OK)
69 expected_freed_packages_.insert(package_query.package_info);
70 return package_query.response_code;
71 }
72
FreeCredentialsHandle(PCredHandle phCredential)73 SECURITY_STATUS MockSSPILibrary::FreeCredentialsHandle(
74 PCredHandle phCredential) {
75 EXPECT_TRUE(phCredential->dwLower == ((ULONG_PTR) ((INT_PTR) 0)));
76 EXPECT_TRUE(phCredential->dwLower == ((ULONG_PTR) ((INT_PTR) 0)));
77 SecInvalidateHandle(phCredential);
78 return SEC_E_OK;
79 }
80
DeleteSecurityContext(PCtxtHandle phContext)81 SECURITY_STATUS MockSSPILibrary::DeleteSecurityContext(PCtxtHandle phContext) {
82 EXPECT_TRUE(phContext->dwLower == ((ULONG_PTR) ((INT_PTR) 0)));
83 EXPECT_TRUE(phContext->dwLower == ((ULONG_PTR) ((INT_PTR) 0)));
84 SecInvalidateHandle(phContext);
85 return SEC_E_OK;
86 }
87
FreeContextBuffer(PVOID pvContextBuffer)88 SECURITY_STATUS MockSSPILibrary::FreeContextBuffer(PVOID pvContextBuffer) {
89 PSecPkgInfoW package_info = static_cast<PSecPkgInfoW>(pvContextBuffer);
90 std::set<PSecPkgInfoW>::iterator it = expected_freed_packages_.find(
91 package_info);
92 EXPECT_TRUE(it != expected_freed_packages_.end());
93 expected_freed_packages_.erase(it);
94 return SEC_E_OK;
95 }
96
ExpectQuerySecurityPackageInfo(const std::wstring & expected_package,SECURITY_STATUS response_code,PSecPkgInfoW package_info)97 void MockSSPILibrary::ExpectQuerySecurityPackageInfo(
98 const std::wstring& expected_package,
99 SECURITY_STATUS response_code,
100 PSecPkgInfoW package_info) {
101 PackageQuery package_query = {expected_package, response_code,
102 package_info};
103 expected_package_queries_.push_back(package_query);
104 }
105
106 } // namespace net
107