1 /* 2 * Tests for the NTLM security provider 3 * 4 * Copyright 2005, 2006 Kai Blin 5 * Copyright 2006 Dmitry Timoshkov 6 * 7 * This library is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * This library is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with this library; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA 20 * 21 * The code that tests for the behaviour of ISC_REQ_ALLOCATE_MEMORY is based 22 * on code written by Dmitry Timoshkov. 23 24 */ 25 26 #include <stdarg.h> 27 #include <stdio.h> 28 #include <assert.h> 29 #include <windef.h> 30 #include <winbase.h> 31 #define SECURITY_WIN32 32 #include <sspi.h> 33 #include <rpc.h> 34 #include <rpcdce.h> 35 #include <secext.h> 36 37 #include "wine/test.h" 38 39 #define NEGOTIATE_BASE_CAPS ( \ 40 SECPKG_FLAG_INTEGRITY | \ 41 SECPKG_FLAG_PRIVACY | \ 42 SECPKG_FLAG_CONNECTION | \ 43 SECPKG_FLAG_MULTI_REQUIRED | \ 44 SECPKG_FLAG_EXTENDED_ERROR | \ 45 SECPKG_FLAG_IMPERSONATION | \ 46 SECPKG_FLAG_ACCEPT_WIN32_NAME | \ 47 SECPKG_FLAG_NEGOTIABLE | \ 48 SECPKG_FLAG_GSS_COMPATIBLE | \ 49 SECPKG_FLAG_LOGON ) 50 51 #define NTLM_BASE_CAPS ( \ 52 SECPKG_FLAG_INTEGRITY | \ 53 SECPKG_FLAG_PRIVACY | \ 54 SECPKG_FLAG_TOKEN_ONLY | \ 55 SECPKG_FLAG_CONNECTION | \ 56 SECPKG_FLAG_MULTI_REQUIRED | \ 57 SECPKG_FLAG_IMPERSONATION | \ 58 SECPKG_FLAG_ACCEPT_WIN32_NAME | \ 59 SECPKG_FLAG_NEGOTIABLE | \ 60 SECPKG_FLAG_LOGON ) 61 62 static HMODULE secdll; 63 static PSecurityFunctionTableA (SEC_ENTRY * pInitSecurityInterfaceA)(void); 64 static SECURITY_STATUS (SEC_ENTRY * pFreeContextBuffer)(PVOID pv); 65 static SECURITY_STATUS (SEC_ENTRY * pQuerySecurityPackageInfoA)(SEC_CHAR*, PSecPkgInfoA*); 66 static SECURITY_STATUS (SEC_ENTRY * pAcquireCredentialsHandleA)(SEC_CHAR*, SEC_CHAR*, 67 ULONG, PLUID, PVOID, SEC_GET_KEY_FN, PVOID, PCredHandle, PTimeStamp); 68 static SECURITY_STATUS (SEC_ENTRY * pAcquireCredentialsHandleW)(SEC_CHAR*, SEC_WCHAR*, 69 ULONG, PLUID, void*, SEC_GET_KEY_FN, void*, CredHandle*, TimeStamp*); 70 static SECURITY_STATUS (SEC_ENTRY * pInitializeSecurityContextA)(PCredHandle, PCtxtHandle, 71 SEC_CHAR*, ULONG, ULONG, ULONG, PSecBufferDesc, ULONG, 72 PCtxtHandle, PSecBufferDesc, PULONG, PTimeStamp); 73 static SECURITY_STATUS (SEC_ENTRY * pCompleteAuthToken)(PCtxtHandle, PSecBufferDesc); 74 static SECURITY_STATUS (SEC_ENTRY * pAcceptSecurityContext)(PCredHandle, PCtxtHandle, 75 PSecBufferDesc, ULONG, ULONG, PCtxtHandle, PSecBufferDesc, 76 PULONG, PTimeStamp); 77 static SECURITY_STATUS (SEC_ENTRY * pFreeCredentialsHandle)(PCredHandle); 78 static SECURITY_STATUS (SEC_ENTRY * pDeleteSecurityContext)(PCtxtHandle); 79 static SECURITY_STATUS (SEC_ENTRY * pQueryContextAttributesA)(PCtxtHandle, ULONG, PVOID); 80 static SECURITY_STATUS (SEC_ENTRY * pMakeSignature)(PCtxtHandle, ULONG, 81 PSecBufferDesc, ULONG); 82 static SECURITY_STATUS (SEC_ENTRY * pVerifySignature)(PCtxtHandle, PSecBufferDesc, 83 ULONG, PULONG); 84 static SECURITY_STATUS (SEC_ENTRY * pEncryptMessage)(PCtxtHandle, ULONG, 85 PSecBufferDesc, ULONG); 86 static SECURITY_STATUS (SEC_ENTRY * pDecryptMessage)(PCtxtHandle, PSecBufferDesc, 87 ULONG, PULONG); 88 static BOOLEAN (WINAPI * pGetUserNameExA)(EXTENDED_NAME_FORMAT, LPSTR, PULONG); 89 90 typedef struct _SspiData { 91 CredHandle cred; 92 CtxtHandle ctxt; 93 PSecBufferDesc in_buf; 94 PSecBufferDesc out_buf; 95 PSEC_WINNT_AUTH_IDENTITY_A id; 96 ULONG max_token; 97 } SspiData; 98 99 static BYTE network_challenge[] = 100 {0x4e, 0x54, 0x4c, 0x4d, 0x53, 0x53, 0x50, 0x00, 0x02, 0x00, 101 0x00, 0x00, 0x10, 0x00, 0x10, 0x00, 0x30, 0x00, 0x00, 0x00, 102 0x05, 0x82, 0x82, 0xa0, 0xe9, 0x58, 0x7f, 0x14, 0xa2, 0x86, 103 0x3b, 0x63, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 104 0x54, 0x00, 0x54, 0x00, 0x40, 0x00, 0x00, 0x00, 0x43, 0x00, 105 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 106 0x30, 0x00, 0x31, 0x00, 0x02, 0x00, 0x10, 0x00, 0x43, 0x00, 107 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 108 0x30, 0x00, 0x31, 0x00, 0x01, 0x00, 0x10, 0x00, 0x43, 0x00, 109 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 110 0x30, 0x00, 0x31, 0x00, 0x04, 0x00, 0x10, 0x00, 0x63, 0x00, 111 0x61, 0x00, 0x73, 0x00, 0x69, 0x00, 0x6e, 0x00, 0x6f, 0x00, 112 0x30, 0x00, 0x31, 0x00, 0x03, 0x00, 0x10, 0x00, 0x63, 0x00, 113 0x61, 0x00, 0x73, 0x00, 0x69, 0x00, 0x6e, 0x00, 0x6f, 0x00, 114 0x30, 0x00, 0x31, 0x00, 0x00, 0x00, 0x00, 0x00}; 115 116 static BYTE native_challenge[] = 117 {0x4e, 0x54, 0x4c, 0x4d, 0x53, 0x53, 0x50, 0x00, 0x02, 0x00, 118 0x00, 0x00, 0x10, 0x00, 0x10, 0x00, 0x30, 0x00, 0x00, 0x00, 119 0x05, 0x82, 0x82, 0xa0, 0xb5, 0x60, 0x8e, 0x95, 0xb5, 0x3c, 120 0xee, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 121 0x54, 0x00, 0x54, 0x00, 0x40, 0x00, 0x00, 0x00, 0x43, 0x00, 122 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 123 0x30, 0x00, 0x31, 0x00, 0x02, 0x00, 0x10, 0x00, 0x43, 0x00, 124 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 125 0x30, 0x00, 0x31, 0x00, 0x01, 0x00, 0x10, 0x00, 0x43, 0x00, 126 0x41, 0x00, 0x53, 0x00, 0x49, 0x00, 0x4e, 0x00, 0x4f, 0x00, 127 0x30, 0x00, 0x31, 0x00, 0x04, 0x00, 0x10, 0x00, 0x63, 0x00, 128 0x61, 0x00, 0x73, 0x00, 0x69, 0x00, 0x6e, 0x00, 0x6f, 0x00, 129 0x30, 0x00, 0x31, 0x00, 0x03, 0x00, 0x10, 0x00, 0x63, 0x00, 130 0x61, 0x00, 0x73, 0x00, 0x69, 0x00, 0x6e, 0x00, 0x6f, 0x00, 131 0x30, 0x00, 0x31, 0x00, 0x00, 0x00, 0x00, 0x00}; 132 133 static BYTE message_signature[] = 134 {0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 135 0x00, 0x00, 0x00, 0x00, 0x00, 0x00}; 136 137 static BYTE message_binary[] = 138 {0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x2c, 0x20, 0x77, 0x6f, 0x72, 139 0x6c, 0x64, 0x21}; 140 141 static const char message[] = "Hello, world!"; 142 143 static char message_header[] = "Header Test"; 144 145 static BYTE crypt_trailer_client[] = 146 {0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xe8, 0xc7, 147 0xaa, 0x26, 0x16, 0x39, 0x07, 0x4e}; 148 149 static BYTE crypt_message_client[] = 150 {0x86, 0x9c, 0x5a, 0x10, 0x78, 0xb3, 0x30, 0x98, 0x46, 0x15, 151 0xa0, 0x31, 0xd9}; 152 153 static BYTE crypt_trailer_client2[] = 154 {0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xc5, 0xa7, 155 0xf7, 0x0f, 0x5b, 0x25, 0xbe, 0xa4}; 156 157 static BYTE crypt_message_client2[] = 158 {0x20, 0x6c, 0x01, 0xab, 0xb0, 0x4c, 0x93, 0xe4, 0x1e, 0xfc, 159 0xe1, 0xfa, 0xfe}; 160 161 static BYTE crypt_trailer_server[] = 162 {0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x1c, 0x46, 163 0x2e, 0x77, 0xeb, 0xf0, 0xf6, 0x9e}; 164 165 static BYTE crypt_message_server[] = 166 {0xf6, 0xb7, 0x92, 0x0c, 0xac, 0xea, 0x98, 0xe6, 0xef, 0xa0, 167 0x29, 0x66, 0xfd}; 168 169 static BYTE crypt_trailer_server2[] = 170 {0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xb1, 0x4e, 171 0x46, 0xb7, 0xca, 0xf7, 0x7f, 0xb3}; 172 173 static BYTE crypt_message_server2[] = 174 {0xc8, 0xf2, 0x39, 0x7f, 0x0c, 0xaf, 0xf5, 0x5d, 0xef, 0x0c, 175 0x8b, 0x5f, 0x82}; 176 177 static char test_user[] = "testuser", 178 workgroup[] = "WORKGROUP", 179 test_pass[] = "testpass", 180 sec_pkg_name[] = "NTLM"; 181 182 static void InitFunctionPtrs(void) 183 { 184 secdll = LoadLibraryA("secur32.dll"); 185 if(!secdll) 186 secdll = LoadLibraryA("security.dll"); 187 if(secdll) 188 { 189 pInitSecurityInterfaceA = (PVOID)GetProcAddress(secdll, "InitSecurityInterfaceA"); 190 pFreeContextBuffer = (PVOID)GetProcAddress(secdll, "FreeContextBuffer"); 191 pQuerySecurityPackageInfoA = (PVOID)GetProcAddress(secdll, "QuerySecurityPackageInfoA"); 192 pAcquireCredentialsHandleA = (PVOID)GetProcAddress(secdll, "AcquireCredentialsHandleA"); 193 pAcquireCredentialsHandleW = (void*)GetProcAddress(secdll, "AcquireCredentialsHandleW"); 194 pInitializeSecurityContextA = (PVOID)GetProcAddress(secdll, "InitializeSecurityContextA"); 195 pCompleteAuthToken = (PVOID)GetProcAddress(secdll, "CompleteAuthToken"); 196 pAcceptSecurityContext = (PVOID)GetProcAddress(secdll, "AcceptSecurityContext"); 197 pFreeCredentialsHandle = (PVOID)GetProcAddress(secdll, "FreeCredentialsHandle"); 198 pDeleteSecurityContext = (PVOID)GetProcAddress(secdll, "DeleteSecurityContext"); 199 pQueryContextAttributesA = (PVOID)GetProcAddress(secdll, "QueryContextAttributesA"); 200 pMakeSignature = (PVOID)GetProcAddress(secdll, "MakeSignature"); 201 pVerifySignature = (PVOID)GetProcAddress(secdll, "VerifySignature"); 202 pEncryptMessage = (PVOID)GetProcAddress(secdll, "EncryptMessage"); 203 pDecryptMessage = (PVOID)GetProcAddress(secdll, "DecryptMessage"); 204 pGetUserNameExA = (PVOID)GetProcAddress(secdll, "GetUserNameExA"); 205 } 206 } 207 208 static const char* getSecError(SECURITY_STATUS status) 209 { 210 static char buf[20]; 211 212 #define _SEC_ERR(x) case (x): return #x; 213 switch(status) 214 { 215 _SEC_ERR(SEC_E_OK); 216 _SEC_ERR(SEC_E_INSUFFICIENT_MEMORY); 217 _SEC_ERR(SEC_E_INVALID_HANDLE); 218 _SEC_ERR(SEC_E_UNSUPPORTED_FUNCTION); 219 _SEC_ERR(SEC_E_TARGET_UNKNOWN); 220 _SEC_ERR(SEC_E_INTERNAL_ERROR); 221 _SEC_ERR(SEC_E_SECPKG_NOT_FOUND); 222 _SEC_ERR(SEC_E_NOT_OWNER); 223 _SEC_ERR(SEC_E_CANNOT_INSTALL); 224 _SEC_ERR(SEC_E_INVALID_TOKEN); 225 _SEC_ERR(SEC_E_CANNOT_PACK); 226 _SEC_ERR(SEC_E_QOP_NOT_SUPPORTED); 227 _SEC_ERR(SEC_E_NO_IMPERSONATION); 228 _SEC_ERR(SEC_I_CONTINUE_NEEDED); 229 _SEC_ERR(SEC_E_BUFFER_TOO_SMALL); 230 _SEC_ERR(SEC_E_ILLEGAL_MESSAGE); 231 _SEC_ERR(SEC_E_LOGON_DENIED); 232 _SEC_ERR(SEC_E_NO_CREDENTIALS); 233 _SEC_ERR(SEC_E_OUT_OF_SEQUENCE); 234 _SEC_ERR(SEC_E_MESSAGE_ALTERED); 235 default: 236 sprintf(buf, "%08x\n", status); 237 return buf; 238 } 239 #undef _SEC_ERR 240 } 241 242 /**********************************************************************/ 243 244 static SECURITY_STATUS setupBuffers(SspiData *sspi_data, SecPkgInfoA *sec_pkg_info) 245 { 246 247 sspi_data->in_buf = HeapAlloc(GetProcessHeap(), 0, sizeof(SecBufferDesc)); 248 sspi_data->out_buf = HeapAlloc(GetProcessHeap(), 0, sizeof(SecBufferDesc)); 249 sspi_data->max_token = sec_pkg_info->cbMaxToken; 250 251 if(sspi_data->in_buf != NULL) 252 { 253 PSecBuffer sec_buffer = HeapAlloc(GetProcessHeap(), 0, 254 sizeof(SecBuffer)); 255 if(sec_buffer == NULL){ 256 trace("in_buf: sec_buffer == NULL\n"); 257 return SEC_E_INSUFFICIENT_MEMORY; 258 } 259 260 sspi_data->in_buf->ulVersion = SECBUFFER_VERSION; 261 sspi_data->in_buf->cBuffers = 1; 262 sspi_data->in_buf->pBuffers = sec_buffer; 263 264 sec_buffer->cbBuffer = sec_pkg_info->cbMaxToken; 265 sec_buffer->BufferType = SECBUFFER_TOKEN; 266 if((sec_buffer->pvBuffer = HeapAlloc(GetProcessHeap(), 0, 267 sec_pkg_info->cbMaxToken)) == NULL) 268 { 269 trace("in_buf: sec_buffer->pvBuffer == NULL\n"); 270 return SEC_E_INSUFFICIENT_MEMORY; 271 } 272 } 273 else 274 { 275 trace("HeapAlloc in_buf returned NULL\n"); 276 return SEC_E_INSUFFICIENT_MEMORY; 277 } 278 279 if(sspi_data->out_buf != NULL) 280 { 281 PSecBuffer sec_buffer = HeapAlloc(GetProcessHeap(), 0, 282 sizeof(SecBuffer)); 283 284 if(sec_buffer == NULL){ 285 trace("out_buf: sec_buffer == NULL\n"); 286 return SEC_E_INSUFFICIENT_MEMORY; 287 } 288 289 sspi_data->out_buf->ulVersion = SECBUFFER_VERSION; 290 sspi_data->out_buf->cBuffers = 1; 291 sspi_data->out_buf->pBuffers = sec_buffer; 292 293 sec_buffer->cbBuffer = sec_pkg_info->cbMaxToken; 294 sec_buffer->BufferType = SECBUFFER_TOKEN; 295 if((sec_buffer->pvBuffer = HeapAlloc(GetProcessHeap(), 0, 296 sec_pkg_info->cbMaxToken)) == NULL){ 297 trace("out_buf: sec_buffer->pvBuffer == NULL\n"); 298 return SEC_E_INSUFFICIENT_MEMORY; 299 } 300 } 301 else 302 { 303 trace("HeapAlloc out_buf returned NULL\n"); 304 return SEC_E_INSUFFICIENT_MEMORY; 305 } 306 307 return SEC_E_OK; 308 } 309 310 /**********************************************************************/ 311 312 static void cleanupBuffers(SspiData *sspi_data) 313 { 314 ULONG i; 315 316 if(sspi_data->in_buf != NULL) 317 { 318 for(i = 0; i < sspi_data->in_buf->cBuffers; ++i) 319 { 320 HeapFree(GetProcessHeap(), 0, sspi_data->in_buf->pBuffers[i].pvBuffer); 321 } 322 HeapFree(GetProcessHeap(), 0, sspi_data->in_buf->pBuffers); 323 HeapFree(GetProcessHeap(), 0, sspi_data->in_buf); 324 } 325 326 if(sspi_data->out_buf != NULL) 327 { 328 for(i = 0; i < sspi_data->out_buf->cBuffers; ++i) 329 { 330 HeapFree(GetProcessHeap(), 0, sspi_data->out_buf->pBuffers[i].pvBuffer); 331 } 332 HeapFree(GetProcessHeap(), 0, sspi_data->out_buf->pBuffers); 333 HeapFree(GetProcessHeap(), 0, sspi_data->out_buf); 334 } 335 } 336 337 /**********************************************************************/ 338 339 static SECURITY_STATUS setupClient(SspiData *sspi_data, SEC_CHAR *provider) 340 { 341 SECURITY_STATUS ret; 342 TimeStamp ttl; 343 SecPkgInfoA *sec_pkg_info; 344 345 trace("Running setupClient\n"); 346 347 ret = pQuerySecurityPackageInfoA(provider, &sec_pkg_info); 348 349 ok(ret == SEC_E_OK, "QuerySecurityPackageInfo returned %s\n", getSecError(ret)); 350 351 setupBuffers(sspi_data, sec_pkg_info); 352 pFreeContextBuffer(sec_pkg_info); 353 354 if((ret = pAcquireCredentialsHandleA(NULL, provider, SECPKG_CRED_OUTBOUND, 355 NULL, sspi_data->id, NULL, NULL, &sspi_data->cred, &ttl)) 356 != SEC_E_OK) 357 { 358 trace("AcquireCredentialsHandle() returned %s\n", getSecError(ret)); 359 } 360 361 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 362 getSecError(ret)); 363 364 return ret; 365 } 366 /**********************************************************************/ 367 368 static SECURITY_STATUS setupServer(SspiData *sspi_data, SEC_CHAR *provider) 369 { 370 SECURITY_STATUS ret; 371 TimeStamp ttl; 372 SecPkgInfoA *sec_pkg_info; 373 374 trace("Running setupServer\n"); 375 376 ret = pQuerySecurityPackageInfoA(provider, &sec_pkg_info); 377 378 ok(ret == SEC_E_OK, "QuerySecurityPackageInfo returned %s\n", getSecError(ret)); 379 380 setupBuffers(sspi_data, sec_pkg_info); 381 pFreeContextBuffer(sec_pkg_info); 382 383 if((ret = pAcquireCredentialsHandleA(NULL, provider, SECPKG_CRED_INBOUND, 384 NULL, NULL, NULL, NULL, &sspi_data->cred, &ttl)) != SEC_E_OK) 385 { 386 trace("AcquireCredentialsHandle() returned %s\n", getSecError(ret)); 387 } 388 389 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 390 getSecError(ret)); 391 392 return ret; 393 } 394 395 /**********************************************************************/ 396 397 static SECURITY_STATUS setupFakeServer(SspiData *sspi_data, SEC_CHAR *provider) 398 { 399 SECURITY_STATUS ret; 400 SecPkgInfoA *sec_pkg_info; 401 402 trace("Running setupFakeServer\n"); 403 404 ret = pQuerySecurityPackageInfoA(provider, &sec_pkg_info); 405 406 ok(ret == SEC_E_OK, "QuerySecurityPackageInfo returned %s\n", getSecError(ret)); 407 408 ret = setupBuffers(sspi_data, sec_pkg_info); 409 pFreeContextBuffer(sec_pkg_info); 410 411 return ret; 412 } 413 414 415 /**********************************************************************/ 416 417 static SECURITY_STATUS runClient(SspiData *sspi_data, BOOL first, ULONG data_rep) 418 { 419 SECURITY_STATUS ret; 420 ULONG req_attr = 0; 421 ULONG ctxt_attr; 422 TimeStamp ttl; 423 PSecBufferDesc in_buf = sspi_data->in_buf; 424 PSecBufferDesc out_buf = sspi_data->out_buf; 425 426 assert(in_buf->cBuffers >= 1); 427 assert(in_buf->pBuffers[0].pvBuffer != NULL); 428 assert(in_buf->pBuffers[0].cbBuffer != 0); 429 430 assert(out_buf->cBuffers >= 1); 431 assert(out_buf->pBuffers[0].pvBuffer != NULL); 432 assert(out_buf->pBuffers[0].cbBuffer != 0); 433 434 trace("Running the client the %s time.\n", first?"first":"second"); 435 436 /* We can either use ISC_REQ_ALLOCATE_MEMORY flag to ask the provider 437 * always allocate output buffers for us, or initialize cbBuffer 438 * before each call because the API changes it to represent actual 439 * amount of data in the buffer. 440 */ 441 442 /* test a failing call only the first time, otherwise we get 443 * SEC_E_OUT_OF_SEQUENCE 444 */ 445 if (first) 446 { 447 void *old_buf; 448 449 /* pass NULL as an output buffer */ 450 ret = pInitializeSecurityContextA(&sspi_data->cred, NULL, NULL, req_attr, 451 0, data_rep, NULL, 0, &sspi_data->ctxt, NULL, 452 &ctxt_attr, &ttl); 453 454 ok(ret == SEC_E_BUFFER_TOO_SMALL, "expected SEC_E_BUFFER_TOO_SMALL, got %s\n", getSecError(ret)); 455 456 /* pass NULL as an output buffer */ 457 old_buf = out_buf->pBuffers[0].pvBuffer; 458 out_buf->pBuffers[0].pvBuffer = NULL; 459 460 ret = pInitializeSecurityContextA(&sspi_data->cred, NULL, NULL, req_attr, 461 0, data_rep, NULL, 0, &sspi_data->ctxt, out_buf, 462 &ctxt_attr, &ttl); 463 464 ok(ret == SEC_E_INTERNAL_ERROR || ret == SEC_I_CONTINUE_NEEDED, 465 "expected SEC_E_INTERNAL_ERROR or SEC_I_CONTINUE_NEEDED, got %s\n", getSecError(ret)); 466 467 out_buf->pBuffers[0].pvBuffer = old_buf; 468 469 /* pass an output buffer of 0 size */ 470 out_buf->pBuffers[0].cbBuffer = 0; 471 472 ret = pInitializeSecurityContextA(&sspi_data->cred, NULL, NULL, req_attr, 473 0, data_rep, NULL, 0, &sspi_data->ctxt, out_buf, 474 &ctxt_attr, &ttl); 475 476 ok(ret == SEC_E_BUFFER_TOO_SMALL, "expected SEC_E_BUFFER_TOO_SMALL, got %s\n", getSecError(ret)); 477 478 ok(out_buf->pBuffers[0].cbBuffer == 0, 479 "InitializeSecurityContext set buffer size to %u\n", out_buf->pBuffers[0].cbBuffer); 480 481 out_buf->pBuffers[0].cbBuffer = sspi_data->max_token; 482 out_buf->pBuffers[0].BufferType = SECBUFFER_DATA; 483 484 ret = pInitializeSecurityContextA(&sspi_data->cred, NULL, NULL, req_attr, 485 0, data_rep, NULL, 0, &sspi_data->ctxt, out_buf, 486 &ctxt_attr, &ttl); 487 488 ok(ret == SEC_E_BUFFER_TOO_SMALL, "expected SEC_E_BUFFER_TOO_SMALL, got %s\n", getSecError(ret)); 489 out_buf->pBuffers[0].BufferType = SECBUFFER_TOKEN; 490 } 491 492 out_buf->pBuffers[0].cbBuffer = sspi_data->max_token; 493 494 ret = pInitializeSecurityContextA(first?&sspi_data->cred:NULL, first?NULL:&sspi_data->ctxt, NULL, req_attr, 495 0, data_rep, first?NULL:in_buf, 0, &sspi_data->ctxt, out_buf, 496 &ctxt_attr, &ttl); 497 498 if(ret == SEC_I_COMPLETE_AND_CONTINUE || ret == SEC_I_COMPLETE_NEEDED) 499 { 500 pCompleteAuthToken(&sspi_data->ctxt, out_buf); 501 if(ret == SEC_I_COMPLETE_AND_CONTINUE) 502 ret = SEC_I_CONTINUE_NEEDED; 503 else if(ret == SEC_I_COMPLETE_NEEDED) 504 ret = SEC_E_OK; 505 } 506 507 ok(out_buf->pBuffers[0].BufferType == SECBUFFER_TOKEN, 508 "buffer type was changed from SECBUFFER_TOKEN to %d\n", out_buf->pBuffers[0].BufferType); 509 ok(out_buf->pBuffers[0].cbBuffer < sspi_data->max_token, 510 "InitializeSecurityContext set buffer size to %u\n", out_buf->pBuffers[0].cbBuffer); 511 512 return ret; 513 } 514 515 /**********************************************************************/ 516 517 static SECURITY_STATUS runServer(SspiData *sspi_data, BOOL first, ULONG data_rep) 518 { 519 SECURITY_STATUS ret; 520 ULONG ctxt_attr; 521 TimeStamp ttl; 522 523 trace("Running the server the %s time\n", first?"first":"second"); 524 525 ret = pAcceptSecurityContext(&sspi_data->cred, first?NULL:&sspi_data->ctxt, 526 sspi_data->in_buf, 0, data_rep, &sspi_data->ctxt, 527 sspi_data->out_buf, &ctxt_attr, &ttl); 528 529 if(ret == SEC_I_COMPLETE_AND_CONTINUE || ret == SEC_I_COMPLETE_NEEDED) 530 { 531 pCompleteAuthToken(&sspi_data->ctxt, sspi_data->out_buf); 532 if(ret == SEC_I_COMPLETE_AND_CONTINUE) 533 ret = SEC_I_CONTINUE_NEEDED; 534 else if(ret == SEC_I_COMPLETE_NEEDED) 535 ret = SEC_E_OK; 536 } 537 538 return ret; 539 } 540 541 /**********************************************************************/ 542 543 static SECURITY_STATUS runFakeServer(SspiData *sspi_data, BOOL first, ULONG data_rep) 544 { 545 trace("Running the fake server the %s time\n", first?"first":"second"); 546 547 if(!first) 548 { 549 sspi_data->out_buf->pBuffers[0].cbBuffer = 0; 550 return SEC_E_OK; 551 } 552 553 if(data_rep == SECURITY_NATIVE_DREP) 554 { 555 sspi_data->out_buf->pBuffers[0].cbBuffer = sizeof(native_challenge); 556 memcpy(sspi_data->out_buf->pBuffers[0].pvBuffer, native_challenge, 557 sspi_data->out_buf->pBuffers[0].cbBuffer); 558 } 559 else 560 { 561 sspi_data->out_buf->pBuffers[0].cbBuffer = sizeof(network_challenge); 562 memcpy(sspi_data->out_buf->pBuffers[0].pvBuffer, network_challenge, 563 sspi_data->out_buf->pBuffers[0].cbBuffer); 564 } 565 566 return SEC_I_CONTINUE_NEEDED; 567 } 568 569 /**********************************************************************/ 570 571 static void communicate(SspiData *from, SspiData *to) 572 { 573 if(from->out_buf != NULL && to->in_buf != NULL) 574 { 575 trace("Running communicate.\n"); 576 if((from->out_buf->cBuffers >= 1) && (to->in_buf->cBuffers >= 1)) 577 { 578 if((from->out_buf->pBuffers[0].pvBuffer != NULL) && 579 (to->in_buf->pBuffers[0].pvBuffer != NULL)) 580 { 581 memset(to->in_buf->pBuffers[0].pvBuffer, 0, to->max_token); 582 583 memcpy(to->in_buf->pBuffers[0].pvBuffer, 584 from->out_buf->pBuffers[0].pvBuffer, 585 from->out_buf->pBuffers[0].cbBuffer); 586 587 to->in_buf->pBuffers[0].cbBuffer = from->out_buf->pBuffers[0].cbBuffer; 588 589 memset(from->out_buf->pBuffers[0].pvBuffer, 0, from->max_token); 590 } 591 } 592 } 593 } 594 595 /**********************************************************************/ 596 static void testInitializeSecurityContextFlags(void) 597 { 598 SECURITY_STATUS sec_status; 599 PSecPkgInfoA pkg_info = NULL; 600 SspiData client = {{0}}; 601 SEC_WINNT_AUTH_IDENTITY_A id; 602 ULONG req_attr, ctxt_attr; 603 TimeStamp ttl; 604 PBYTE packet; 605 606 if(pQuerySecurityPackageInfoA( sec_pkg_name, &pkg_info) != SEC_E_OK) 607 { 608 ok(0, "NTLM package not installed, skipping test.\n"); 609 return; 610 } 611 612 pFreeContextBuffer(pkg_info); 613 id.User = (unsigned char*) test_user; 614 id.UserLength = strlen((char *) id.User); 615 id.Domain = (unsigned char *) workgroup; 616 id.DomainLength = strlen((char *) id.Domain); 617 id.Password = (unsigned char*) test_pass; 618 id.PasswordLength = strlen((char *) id.Password); 619 id.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 620 621 client.id = &id; 622 623 if((sec_status = setupClient(&client, sec_pkg_name)) != SEC_E_OK) 624 { 625 skip("Setting up the client returned %s, skipping test!\n", 626 getSecError(sec_status)); 627 return; 628 } 629 630 packet = client.out_buf->pBuffers[0].pvBuffer; 631 632 /* Due to how the requesting of the flags is implemented in ntlm_auth, 633 * the tests need to be in this order, as there is no way to specify 634 * "I request no special features" in ntlm_auth */ 635 636 /* Without any flags, the lowest byte should not have bits 0x20 or 0x10 set*/ 637 req_attr = 0; 638 639 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 640 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 641 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 642 { 643 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 644 getSecError(sec_status)); 645 goto tISCFend; 646 } 647 648 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 649 "With req_attr == 0, flags are 0x%02x%02x%02x%02x.\n", 650 packet[15], packet[14], packet[13], packet[12]); 651 pDeleteSecurityContext(&client.ctxt); 652 653 /* With ISC_REQ_CONNECTION, the lowest byte should not have bits 0x20 or 0x10 set*/ 654 req_attr = ISC_REQ_CONNECTION; 655 656 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 657 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 658 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 659 { 660 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 661 getSecError(sec_status)); 662 goto tISCFend; 663 } 664 665 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 666 "For ISC_REQ_CONNECTION, flags are 0x%02x%02x%02x%02x.\n", 667 packet[15], packet[14], packet[13], packet[12]); 668 pDeleteSecurityContext(&client.ctxt); 669 670 /* With ISC_REQ_EXTENDED_ERROR, the lowest byte should not have bits 0x20 or 0x10 set*/ 671 req_attr = ISC_REQ_EXTENDED_ERROR; 672 673 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 674 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 675 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 676 { 677 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 678 getSecError(sec_status)); 679 goto tISCFend; 680 } 681 682 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 683 "For ISC_REQ_EXTENDED_ERROR, flags are 0x%02x%02x%02x%02x.\n", 684 packet[15], packet[14], packet[13], packet[12]); 685 pDeleteSecurityContext(&client.ctxt); 686 687 /* With ISC_REQ_MUTUAL_AUTH, the lowest byte should not have bits 0x20 or 0x10 set*/ 688 req_attr = ISC_REQ_MUTUAL_AUTH; 689 690 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 691 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 692 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 693 { 694 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 695 getSecError(sec_status)); 696 goto tISCFend; 697 } 698 699 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 700 "For ISC_REQ_MUTUAL_AUTH, flags are 0x%02x%02x%02x%02x.\n", 701 packet[15], packet[14], packet[13], packet[12]); 702 pDeleteSecurityContext(&client.ctxt); 703 704 /* With ISC_REQ_USE_DCE_STYLE, the lowest byte should not have bits 0x20 or 0x10 set*/ 705 req_attr = ISC_REQ_USE_DCE_STYLE; 706 707 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 708 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 709 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 710 { 711 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 712 getSecError(sec_status)); 713 goto tISCFend; 714 } 715 716 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 717 "For ISC_REQ_USE_DCE_STYLE, flags are 0x%02x%02x%02x%02x.\n", 718 packet[15], packet[14], packet[13], packet[12]); 719 pDeleteSecurityContext(&client.ctxt); 720 721 /* With ISC_REQ_DELEGATE, the lowest byte should not have bits 0x20 or 0x10 set*/ 722 req_attr = ISC_REQ_DELEGATE; 723 724 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 725 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 726 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 727 { 728 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 729 getSecError(sec_status)); 730 goto tISCFend; 731 } 732 733 ok(((packet[12] & 0x10) == 0) && ((packet[12] & 0x20) == 0), 734 "For ISC_REQ_DELEGATE, flags are 0x%02x%02x%02x%02x.\n", 735 packet[15], packet[14], packet[13], packet[12]); 736 pDeleteSecurityContext(&client.ctxt); 737 738 /* With ISC_REQ_INTEGRITY, the lowest byte should have bit 0x10 set */ 739 req_attr = ISC_REQ_INTEGRITY; 740 741 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 742 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 743 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 744 { 745 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 746 getSecError(sec_status)); 747 goto tISCFend; 748 } 749 750 ok((packet[12] & 0x10) != 0, 751 "For ISC_REQ_INTEGRITY, flags are 0x%02x%02x%02x%02x.\n", 752 packet[15], packet[14], packet[13], packet[12]); 753 pDeleteSecurityContext(&client.ctxt); 754 755 /* With ISC_REQ_REPLAY_DETECT, the lowest byte should have bit 0x10 set */ 756 req_attr = ISC_REQ_REPLAY_DETECT; 757 758 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 759 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 760 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 761 { 762 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 763 getSecError(sec_status)); 764 goto tISCFend; 765 } 766 767 ok((packet[12] & 0x10) != 0, 768 "For ISC_REQ_REPLAY_DETECT, flags are 0x%02x%02x%02x%02x.\n", 769 packet[15], packet[14], packet[13], packet[12]); 770 pDeleteSecurityContext(&client.ctxt); 771 772 /* With ISC_REQ_SEQUENCE_DETECT, the lowest byte should have bit 0x10 set */ 773 req_attr = ISC_REQ_SEQUENCE_DETECT; 774 775 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 776 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 777 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 778 { 779 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 780 getSecError(sec_status)); 781 goto tISCFend; 782 } 783 784 ok((packet[12] & 0x10) != 0, 785 "For ISC_REQ_SEQUENCE_DETECT, flags are 0x%02x%02x%02x%02x.\n", 786 packet[15], packet[14], packet[13], packet[12]); 787 pDeleteSecurityContext(&client.ctxt); 788 789 /* With ISC_REQ_CONFIDENTIALITY, the lowest byte should have bit 0x20 set */ 790 req_attr = ISC_REQ_CONFIDENTIALITY; 791 792 if((sec_status = pInitializeSecurityContextA(&client.cred, NULL, NULL, req_attr, 793 0, SECURITY_NETWORK_DREP, NULL, 0, &client.ctxt, client.out_buf, 794 &ctxt_attr, &ttl)) != SEC_I_CONTINUE_NEEDED) 795 { 796 trace("InitializeSecurityContext returned %s not SEC_I_CONTINUE_NEEDED, aborting.\n", 797 getSecError(sec_status)); 798 goto tISCFend; 799 } 800 801 ok((packet[12] & 0x20) != 0, 802 "For ISC_REQ_CONFIDENTIALITY, flags are 0x%02x%02x%02x%02x.\n", 803 packet[15], packet[14], packet[13], packet[12]); 804 pDeleteSecurityContext(&client.ctxt); 805 806 tISCFend: 807 cleanupBuffers(&client); 808 pFreeCredentialsHandle(&client.cred); 809 } 810 811 /**********************************************************************/ 812 813 static void testAuth(ULONG data_rep, BOOL fake) 814 { 815 SECURITY_STATUS client_stat = SEC_I_CONTINUE_NEEDED; 816 SECURITY_STATUS server_stat = SEC_I_CONTINUE_NEEDED; 817 SECURITY_STATUS sec_status; 818 PSecPkgInfoA pkg_info = NULL; 819 BOOL first = TRUE; 820 SspiData client = {{0}}, server = {{0}}; 821 SEC_WINNT_AUTH_IDENTITY_A id; 822 SecPkgContext_Sizes ctxt_sizes; 823 SecPkgContext_NegotiationInfoA info; 824 SecPkgInfoA *pi; 825 826 if(pQuerySecurityPackageInfoA( sec_pkg_name, &pkg_info)!= SEC_E_OK) 827 { 828 ok(0, "NTLM package not installed, skipping test.\n"); 829 return; 830 } 831 832 pFreeContextBuffer(pkg_info); 833 id.User = (unsigned char*) test_user; 834 id.UserLength = strlen((char *) id.User); 835 id.Domain = (unsigned char *) workgroup; 836 id.DomainLength = strlen((char *) id.Domain); 837 id.Password = (unsigned char*) test_pass; 838 id.PasswordLength = strlen((char *) id.Password); 839 id.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 840 841 client.id = &id; 842 843 sec_status = setupClient(&client, sec_pkg_name); 844 845 if(sec_status != SEC_E_OK) 846 { 847 skip("Error: Setting up the client returned %s, exiting test!\n", 848 getSecError(sec_status)); 849 pFreeCredentialsHandle(&client.cred); 850 return; 851 } 852 853 if(fake) 854 sec_status = setupFakeServer(&server, sec_pkg_name); 855 else 856 sec_status = setupServer(&server, sec_pkg_name); 857 858 if(sec_status != SEC_E_OK) 859 { 860 skip("Error: Setting up the server returned %s, exiting test!\n", 861 getSecError(sec_status)); 862 pFreeCredentialsHandle(&server.cred); 863 pFreeCredentialsHandle(&client.cred); 864 return; 865 } 866 867 while(client_stat == SEC_I_CONTINUE_NEEDED && server_stat == SEC_I_CONTINUE_NEEDED) 868 { 869 client_stat = runClient(&client, first, data_rep); 870 871 ok(client_stat == SEC_E_OK || client_stat == SEC_I_CONTINUE_NEEDED, 872 "Running the client returned %s, more tests will fail.\n", 873 getSecError(client_stat)); 874 875 communicate(&client, &server); 876 877 if(fake) 878 server_stat = runFakeServer(&server, first, data_rep); 879 else 880 server_stat = runServer(&server, first, data_rep); 881 882 ok(server_stat == SEC_E_OK || server_stat == SEC_I_CONTINUE_NEEDED || 883 server_stat == SEC_E_LOGON_DENIED, 884 "Running the server returned %s, more tests will fail from now.\n", 885 getSecError(server_stat)); 886 887 communicate(&server, &client); 888 trace("Looping\n"); 889 first = FALSE; 890 } 891 892 if(client_stat != SEC_E_OK) 893 { 894 skip("Authentication failed, skipping test.\n"); 895 goto tAuthend; 896 } 897 898 sec_status = pQueryContextAttributesA(&client.ctxt, 899 SECPKG_ATTR_SIZES, &ctxt_sizes); 900 901 ok(sec_status == SEC_E_OK, 902 "pQueryContextAttributesA(SECPKG_ATTR_SIZES) returned %s\n", 903 getSecError(sec_status)); 904 ok((ctxt_sizes.cbMaxToken == 1904) || (ctxt_sizes.cbMaxToken == 2888), 905 "cbMaxToken should be 1904 or 2888 but is %u\n", 906 ctxt_sizes.cbMaxToken); 907 ok(ctxt_sizes.cbMaxSignature == 16, 908 "cbMaxSignature should be 16 but is %u\n", 909 ctxt_sizes.cbMaxSignature); 910 ok(ctxt_sizes.cbSecurityTrailer == 16, 911 "cbSecurityTrailer should be 16 but is %u\n", 912 ctxt_sizes.cbSecurityTrailer); 913 ok(ctxt_sizes.cbBlockSize == 0, 914 "cbBlockSize should be 0 but is %u\n", 915 ctxt_sizes.cbBlockSize); 916 917 memset(&info, 0, sizeof(info)); 918 sec_status = QueryContextAttributesA(&client.ctxt, SECPKG_ATTR_NEGOTIATION_INFO, &info); 919 ok(sec_status == SEC_E_OK, "QueryContextAttributesA returned %08x\n", sec_status); 920 921 pi = info.PackageInfo; 922 ok(info.NegotiationState == SECPKG_NEGOTIATION_COMPLETE, "got %u\n", info.NegotiationState); 923 ok(pi != NULL, "expected non-NULL PackageInfo\n"); 924 if (pi) 925 { 926 UINT expected, got; 927 char *eob; 928 929 ok(pi->fCapabilities == NTLM_BASE_CAPS || 930 pi->fCapabilities == (NTLM_BASE_CAPS|SECPKG_FLAG_READONLY_WITH_CHECKSUM) || 931 pi->fCapabilities == (NTLM_BASE_CAPS|SECPKG_FLAG_RESTRICTED_TOKENS) || 932 pi->fCapabilities == (NTLM_BASE_CAPS|SECPKG_FLAG_RESTRICTED_TOKENS|SECPKG_FLAG_APPCONTAINER_CHECKS) || 933 pi->fCapabilities == (NTLM_BASE_CAPS|SECPKG_FLAG_RESTRICTED_TOKENS|SECPKG_FLAG_APPLY_LOOPBACK) || 934 pi->fCapabilities == (NTLM_BASE_CAPS|SECPKG_FLAG_RESTRICTED_TOKENS|SECPKG_FLAG_APPLY_LOOPBACK| 935 SECPKG_FLAG_APPCONTAINER_CHECKS), 936 "got %08x\n", pi->fCapabilities); 937 ok(pi->wVersion == 1, "got %u\n", pi->wVersion); 938 ok(pi->wRPCID == RPC_C_AUTHN_WINNT, "got %u\n", pi->wRPCID); 939 ok(!lstrcmpA( pi->Name, "NTLM" ), "got %s\n", pi->Name); 940 941 expected = sizeof(*pi) + lstrlenA(pi->Name) + 1 + lstrlenA(pi->Comment) + 1; 942 got = HeapSize(GetProcessHeap(), 0, pi); 943 ok(got == expected, "got %u, expected %u\n", got, expected); 944 eob = (char *)pi + expected; 945 ok(pi->Name + lstrlenA(pi->Name) < eob, "Name doesn't fit into allocated block\n"); 946 ok(pi->Comment + lstrlenA(pi->Comment) < eob, "Comment doesn't fit into allocated block\n"); 947 948 sec_status = FreeContextBuffer(pi); 949 ok(sec_status == SEC_E_OK, "FreeContextBuffer error %#x\n", sec_status); 950 } 951 952 tAuthend: 953 cleanupBuffers(&client); 954 cleanupBuffers(&server); 955 956 if(!fake) 957 { 958 sec_status = pDeleteSecurityContext(&server.ctxt); 959 ok(sec_status == SEC_E_OK, "DeleteSecurityContext(server) returned %s\n", 960 getSecError(sec_status)); 961 } 962 963 sec_status = pDeleteSecurityContext(&client.ctxt); 964 ok(sec_status == SEC_E_OK, "DeleteSecurityContext(client) returned %s\n", 965 getSecError(sec_status)); 966 967 if(!fake) 968 { 969 sec_status = pFreeCredentialsHandle(&server.cred); 970 ok(sec_status == SEC_E_OK, "FreeCredentialsHandle(server) returned %s\n", 971 getSecError(sec_status)); 972 } 973 974 sec_status = pFreeCredentialsHandle(&client.cred); 975 ok(sec_status == SEC_E_OK, "FreeCredentialsHandle(client) returned %s\n", 976 getSecError(sec_status)); 977 } 978 979 static void testSignSeal(void) 980 { 981 SECURITY_STATUS client_stat = SEC_I_CONTINUE_NEEDED; 982 SECURITY_STATUS server_stat = SEC_I_CONTINUE_NEEDED; 983 SECURITY_STATUS sec_status; 984 PSecPkgInfoA pkg_info = NULL; 985 BOOL first = TRUE; 986 SspiData client = {{0}}, server = {{0}}; 987 SEC_WINNT_AUTH_IDENTITY_A id; 988 static char sec_pkg_name[] = "NTLM"; 989 SecBufferDesc crypt; 990 SecBuffer data[2], fake_data[2], complex_data[4]; 991 ULONG qop = 0xdeadbeef; 992 SecPkgContext_Sizes ctxt_sizes; 993 994 complex_data[1].pvBuffer = complex_data[3].pvBuffer = NULL; 995 996 /**************************************************************** 997 * This is basically the same as in testAuth with a fake server, 998 * as we need a valid, authenticated context. 999 */ 1000 if(pQuerySecurityPackageInfoA( sec_pkg_name, &pkg_info) != SEC_E_OK) 1001 { 1002 ok(0, "NTLM package not installed, skipping test.\n"); 1003 return; 1004 } 1005 1006 pFreeContextBuffer(pkg_info); 1007 id.User = (unsigned char*) test_user; 1008 id.UserLength = strlen((char *) id.User); 1009 id.Domain = (unsigned char *) workgroup; 1010 id.DomainLength = strlen((char *) id.Domain); 1011 id.Password = (unsigned char*) test_pass; 1012 id.PasswordLength = strlen((char *) id.Password); 1013 id.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 1014 1015 client.id = &id; 1016 1017 sec_status = setupClient(&client, sec_pkg_name); 1018 1019 if(sec_status != SEC_E_OK) 1020 { 1021 skip("Error: Setting up the client returned %s, exiting test!\n", 1022 getSecError(sec_status)); 1023 pFreeCredentialsHandle(&client.cred); 1024 return; 1025 } 1026 1027 sec_status = setupFakeServer(&server, sec_pkg_name); 1028 ok(sec_status == SEC_E_OK, "setupFakeServer returned %s\n", getSecError(sec_status)); 1029 1030 while(client_stat == SEC_I_CONTINUE_NEEDED && server_stat == SEC_I_CONTINUE_NEEDED) 1031 { 1032 client_stat = runClient(&client, first, SECURITY_NETWORK_DREP); 1033 1034 ok(client_stat == SEC_E_OK || client_stat == SEC_I_CONTINUE_NEEDED, 1035 "Running the client returned %s, more tests will fail.\n", 1036 getSecError(client_stat)); 1037 1038 communicate(&client, &server); 1039 1040 server_stat = runFakeServer(&server, first, SECURITY_NETWORK_DREP); 1041 1042 communicate(&server, &client); 1043 trace("Looping\n"); 1044 first = FALSE; 1045 } 1046 1047 if(client_stat != SEC_E_OK) 1048 { 1049 skip("Authentication failed, skipping test.\n"); 1050 goto end; 1051 } 1052 1053 /******************************************** 1054 * Now start with the actual testing * 1055 ********************************************/ 1056 1057 if(pQueryContextAttributesA(&client.ctxt, SECPKG_ATTR_SIZES, 1058 &ctxt_sizes) != SEC_E_OK) 1059 { 1060 skip("Failed to get context sizes, aborting test.\n"); 1061 goto end; 1062 } 1063 1064 crypt.ulVersion = SECBUFFER_VERSION; 1065 crypt.cBuffers = 2; 1066 1067 crypt.pBuffers = fake_data; 1068 1069 fake_data[0].BufferType = SECBUFFER_DATA; 1070 fake_data[0].cbBuffer = ctxt_sizes.cbSecurityTrailer; 1071 fake_data[0].pvBuffer = HeapAlloc(GetProcessHeap(), 0, fake_data[0].cbBuffer); 1072 1073 fake_data[1].BufferType = SECBUFFER_DATA; 1074 fake_data[1].cbBuffer = lstrlenA(message); 1075 fake_data[1].pvBuffer = HeapAlloc(GetProcessHeap(), 0, fake_data[1].cbBuffer); 1076 1077 sec_status = pMakeSignature(&client.ctxt, 0, &crypt, 0); 1078 ok(sec_status == SEC_E_INVALID_TOKEN, 1079 "MakeSignature returned %s, not SEC_E_INVALID_TOKEN.\n", 1080 getSecError(sec_status)); 1081 1082 crypt.pBuffers = data; 1083 1084 data[0].BufferType = SECBUFFER_TOKEN; 1085 data[0].cbBuffer = ctxt_sizes.cbSecurityTrailer; 1086 data[0].pvBuffer = HeapAlloc(GetProcessHeap(), 0, data[0].cbBuffer); 1087 1088 data[1].BufferType = SECBUFFER_DATA; 1089 data[1].cbBuffer = lstrlenA(message); 1090 data[1].pvBuffer = HeapAlloc(GetProcessHeap(), 0, data[1].cbBuffer); 1091 memcpy(data[1].pvBuffer, message, data[1].cbBuffer); 1092 1093 /* As we forced NTLM to fall back to a password-derived session key, 1094 * we should get the same signature for our data, no matter if 1095 * it is sent by the client or the server 1096 */ 1097 sec_status = pMakeSignature(&client.ctxt, 0, &crypt, 0); 1098 ok(sec_status == SEC_E_OK, "MakeSignature returned %s, not SEC_E_OK.\n", 1099 getSecError(sec_status)); 1100 ok(!memcmp(crypt.pBuffers[0].pvBuffer, message_signature, 1101 crypt.pBuffers[0].cbBuffer), "Signature is not as expected.\n"); 1102 1103 data[0].cbBuffer = sizeof(message_signature); 1104 1105 memcpy(data[0].pvBuffer, crypt_trailer_client, data[0].cbBuffer); 1106 1107 sec_status = pVerifySignature(&client.ctxt, &crypt, 0, &qop); 1108 ok(sec_status == SEC_E_MESSAGE_ALTERED, 1109 "VerifySignature returned %s, not SEC_E_MESSAGE_ALTERED.\n", 1110 getSecError(sec_status)); 1111 ok(qop == 0xdeadbeef, "qop changed to %u\n", qop); 1112 1113 memcpy(data[0].pvBuffer, message_signature, data[0].cbBuffer); 1114 sec_status = pVerifySignature(&client.ctxt, &crypt, 0, &qop); 1115 ok(sec_status == SEC_E_OK, "VerifySignature returned %s, not SEC_E_OK.\n", 1116 getSecError(sec_status)); 1117 ok(qop == 0xdeadbeef, "qop changed to %u\n", qop); 1118 1119 sec_status = pEncryptMessage(&client.ctxt, 0, &crypt, 0); 1120 if (sec_status == SEC_E_UNSUPPORTED_FUNCTION) 1121 { 1122 skip("Encrypt message returned SEC_E_UNSUPPORTED_FUNCTION. " 1123 "Expected on Vista.\n"); 1124 goto end; 1125 } 1126 ok(sec_status == SEC_E_OK, "EncryptMessage returned %s, not SEC_E_OK.\n", 1127 getSecError(sec_status)); 1128 1129 /* first 8 bytes must always be the same */ 1130 ok(!memcmp(crypt.pBuffers[0].pvBuffer, crypt_trailer_client, 8), "Crypt trailer not as expected.\n"); 1131 1132 /* the rest depends on the session key */ 1133 if (!memcmp(crypt.pBuffers[0].pvBuffer, crypt_trailer_client, crypt.pBuffers[0].cbBuffer)) 1134 { 1135 ok(!memcmp(crypt.pBuffers[0].pvBuffer, crypt_trailer_client, 1136 crypt.pBuffers[0].cbBuffer), "Crypt trailer not as expected.\n"); 1137 ok(!memcmp(crypt.pBuffers[1].pvBuffer, crypt_message_client, 1138 crypt.pBuffers[1].cbBuffer), "Crypt message not as expected.\n"); 1139 if (memcmp(crypt.pBuffers[1].pvBuffer, crypt_message_client, 1140 crypt.pBuffers[1].cbBuffer)) 1141 { 1142 int i; 1143 for (i = 0; i < crypt.pBuffers[1].cbBuffer; i++) 1144 { 1145 if (i % 8 == 0) printf(" "); 1146 printf("0x%02x,", ((unsigned char *)crypt.pBuffers[1].pvBuffer)[i]); 1147 if (i % 8 == 7) printf("\n"); 1148 } 1149 printf("\n"); 1150 } 1151 1152 data[0].cbBuffer = sizeof(crypt_trailer_server); 1153 data[1].cbBuffer = sizeof(crypt_message_server); 1154 memcpy(data[0].pvBuffer, crypt_trailer_server, data[0].cbBuffer); 1155 memcpy(data[1].pvBuffer, crypt_message_server, data[1].cbBuffer); 1156 1157 sec_status = pDecryptMessage(&client.ctxt, &crypt, 0, &qop); 1158 1159 ok(sec_status == SEC_E_OK, "DecryptMessage returned %s, not SEC_E_OK.\n", 1160 getSecError(sec_status)); 1161 ok(!memcmp(crypt.pBuffers[1].pvBuffer, message_binary, 1162 crypt.pBuffers[1].cbBuffer), 1163 "Failed to decrypt message correctly.\n"); 1164 ok(qop == 0xdeadbeef, "qop changed to %u\n", qop); 1165 } 1166 else trace( "A different session key is being used\n" ); 1167 1168 trace("Testing with more than one buffer.\n"); 1169 1170 crypt.cBuffers = ARRAY_SIZE(complex_data); 1171 crypt.pBuffers = complex_data; 1172 1173 complex_data[0].BufferType = SECBUFFER_DATA|SECBUFFER_READONLY_WITH_CHECKSUM; 1174 complex_data[0].cbBuffer = sizeof(message_header); 1175 complex_data[0].pvBuffer = message_header; 1176 1177 complex_data[1].BufferType = SECBUFFER_DATA; 1178 complex_data[1].cbBuffer = lstrlenA(message); 1179 complex_data[1].pvBuffer = HeapAlloc(GetProcessHeap(), 0, data[1].cbBuffer); 1180 memcpy(complex_data[1].pvBuffer, message, complex_data[1].cbBuffer); 1181 1182 complex_data[2].BufferType = SECBUFFER_DATA|SECBUFFER_READONLY_WITH_CHECKSUM; 1183 complex_data[2].cbBuffer = sizeof(message_header); 1184 complex_data[2].pvBuffer = message_header; 1185 1186 complex_data[3].BufferType = SECBUFFER_TOKEN; 1187 complex_data[3].cbBuffer = ctxt_sizes.cbSecurityTrailer; 1188 complex_data[3].pvBuffer = HeapAlloc(GetProcessHeap(), 0, complex_data[3].cbBuffer); 1189 1190 /* We should get a dummy signature again. */ 1191 sec_status = pMakeSignature(&client.ctxt, 0, &crypt, 0); 1192 ok(sec_status == SEC_E_OK, "MakeSignature returned %s, not SEC_E_OK.\n", 1193 getSecError(sec_status)); 1194 ok(!memcmp(crypt.pBuffers[3].pvBuffer, message_signature, 1195 crypt.pBuffers[3].cbBuffer), "Signature is not as expected.\n"); 1196 1197 /* Being a dummy signature, it will verify right away, as if the server 1198 * sent it */ 1199 sec_status = pVerifySignature(&client.ctxt, &crypt, 0, &qop); 1200 ok(sec_status == SEC_E_OK, "VerifySignature returned %s, not SEC_E_OK\n", 1201 getSecError(sec_status)); 1202 ok(qop == 0xdeadbeef, "qop changed to %u\n", qop); 1203 1204 sec_status = pEncryptMessage(&client.ctxt, 0, &crypt, 0); 1205 ok(sec_status == SEC_E_OK, "EncryptMessage returned %s, not SEC_E_OK.\n", 1206 getSecError(sec_status)); 1207 1208 ok(!memcmp(crypt.pBuffers[3].pvBuffer, crypt_trailer_client2, 8), "Crypt trailer not as expected.\n"); 1209 1210 if (memcmp(crypt.pBuffers[3].pvBuffer, crypt_trailer_client2, 1211 crypt.pBuffers[3].cbBuffer)) goto end; 1212 1213 ok(!memcmp(crypt.pBuffers[1].pvBuffer, crypt_message_client2, 1214 crypt.pBuffers[1].cbBuffer), "Crypt message not as expected.\n"); 1215 if (memcmp(crypt.pBuffers[1].pvBuffer, crypt_message_client2, 1216 crypt.pBuffers[1].cbBuffer)) 1217 { 1218 int i; 1219 for (i = 0; i < crypt.pBuffers[1].cbBuffer; i++) 1220 { 1221 if (i % 8 == 0) printf(" "); 1222 printf("0x%02x,", ((unsigned char *)crypt.pBuffers[1].pvBuffer)[i]); 1223 if (i % 8 == 7) printf("\n"); 1224 } 1225 printf("\n"); 1226 } 1227 1228 memcpy(complex_data[1].pvBuffer, crypt_message_server2, complex_data[1].cbBuffer); 1229 memcpy(complex_data[3].pvBuffer, crypt_trailer_server2, complex_data[3].cbBuffer); 1230 1231 sec_status = pDecryptMessage(&client.ctxt, &crypt, 0, &qop); 1232 ok(sec_status == SEC_E_OK, "DecryptMessage returned %s, not SEC_E_OK.\n", 1233 getSecError(sec_status)); 1234 ok(qop == 0xdeadbeef, "qop changed to %u\n", qop); 1235 1236 1237 end: 1238 cleanupBuffers(&client); 1239 cleanupBuffers(&server); 1240 1241 pDeleteSecurityContext(&client.ctxt); 1242 pFreeCredentialsHandle(&client.cred); 1243 1244 HeapFree(GetProcessHeap(), 0, complex_data[1].pvBuffer); 1245 HeapFree(GetProcessHeap(), 0, complex_data[3].pvBuffer); 1246 } 1247 1248 static BOOL testAcquireCredentialsHandle(void) 1249 { 1250 CredHandle cred; 1251 TimeStamp ttl; 1252 SECURITY_STATUS ret; 1253 SEC_WINNT_AUTH_IDENTITY_A id; 1254 PSecPkgInfoA pkg_info = NULL; 1255 1256 if(pQuerySecurityPackageInfoA(sec_pkg_name, &pkg_info) != SEC_E_OK) 1257 { 1258 ok(0, "NTLM package not installed, skipping test\n"); 1259 return FALSE; 1260 } 1261 pFreeContextBuffer(pkg_info); 1262 1263 id.User = (unsigned char*) test_user; 1264 id.UserLength = strlen((char *) id.User); 1265 id.Domain = (unsigned char *) workgroup; 1266 id.DomainLength = strlen((char *) id.Domain); 1267 id.Password = (unsigned char*) test_pass; 1268 id.PasswordLength = strlen((char *) id.Password); 1269 id.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 1270 1271 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1272 NULL, &id, NULL, NULL, &cred, &ttl); 1273 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1274 getSecError(ret)); 1275 pFreeCredentialsHandle(&cred); 1276 1277 id.DomainLength = 0; 1278 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1279 NULL, &id, NULL, NULL, &cred, &ttl); 1280 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1281 getSecError(ret)); 1282 pFreeCredentialsHandle(&cred); 1283 1284 id.Domain = NULL; 1285 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1286 NULL, &id, NULL, NULL, &cred, &ttl); 1287 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1288 getSecError(ret)); 1289 pFreeCredentialsHandle(&cred); 1290 1291 id.Domain = (unsigned char *) workgroup; 1292 id.DomainLength = strlen((char *) id.Domain); 1293 id.UserLength = 0; 1294 id.User = NULL; 1295 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1296 NULL, &id, NULL, NULL, &cred, &ttl); 1297 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1298 getSecError(ret)); 1299 pFreeCredentialsHandle(&cred); 1300 1301 id.User = (unsigned char*) test_user; 1302 id.UserLength = strlen((char *) id.User); 1303 id.Password = NULL; 1304 id.PasswordLength = 0; 1305 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1306 NULL, &id, NULL, NULL, &cred, &ttl); 1307 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1308 getSecError(ret)); 1309 pFreeCredentialsHandle(&cred); 1310 return TRUE; 1311 } 1312 1313 static void testAcquireCredentialsHandleW(void) 1314 { 1315 CredHandle cred; 1316 TimeStamp ttl; 1317 static WCHAR sec_pkg_nameW[] = {'N','T','L','M',0 }; 1318 static WCHAR test_userW[] = {'t','e','s','t','u','s','e','r',0 }; 1319 static WCHAR workgroupW[] = {'W','O','R','K','G','R','O','U','P',0}; 1320 static WCHAR test_passW[] = {'t','e','s','t','p','a','s','s',0}; 1321 SECURITY_STATUS ret; 1322 SEC_WINNT_AUTH_IDENTITY_A idA; 1323 SEC_WINNT_AUTH_IDENTITY_W id; 1324 PSecPkgInfoA pkg_info = NULL; 1325 1326 if(!pAcquireCredentialsHandleW) 1327 { 1328 win_skip("AcquireCredentialsHandleW not available\n"); 1329 return; 1330 } 1331 1332 if(pQuerySecurityPackageInfoA(sec_pkg_name, &pkg_info) != SEC_E_OK) 1333 { 1334 ok(0, "NTLM package not installed, skipping test\n"); 1335 return; 1336 } 1337 pFreeContextBuffer(pkg_info); 1338 1339 id.User = test_userW; 1340 id.UserLength = lstrlenW(test_userW); 1341 id.Domain = workgroupW; 1342 id.DomainLength = lstrlenW(workgroupW); 1343 id.Password = test_passW; 1344 id.PasswordLength = lstrlenW(test_passW); 1345 id.Flags = SEC_WINNT_AUTH_IDENTITY_UNICODE; 1346 1347 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1348 NULL, &id, NULL, NULL, &cred, &ttl); 1349 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1350 getSecError(ret)); 1351 pFreeCredentialsHandle(&cred); 1352 1353 id.DomainLength = 0; 1354 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1355 NULL, &id, NULL, NULL, &cred, &ttl); 1356 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1357 getSecError(ret)); 1358 pFreeCredentialsHandle(&cred); 1359 1360 id.Domain = NULL; 1361 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1362 NULL, &id, NULL, NULL, &cred, &ttl); 1363 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1364 getSecError(ret)); 1365 pFreeCredentialsHandle(&cred); 1366 1367 id.Domain = workgroupW; 1368 id.DomainLength = lstrlenW(workgroupW); 1369 id.UserLength = 0; 1370 id.User = NULL; 1371 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1372 NULL, &id, NULL, NULL, &cred, &ttl); 1373 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1374 getSecError(ret)); 1375 pFreeCredentialsHandle(&cred); 1376 1377 id.User = test_userW; 1378 id.UserLength = lstrlenW(test_userW); 1379 id.Password = test_passW; /* NULL string causes a crash. */ 1380 id.PasswordLength = 0; 1381 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1382 NULL, &id, NULL, NULL, &cred, &ttl); 1383 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1384 getSecError(ret)); 1385 pFreeCredentialsHandle(&cred); 1386 1387 /* Test using the ASCII structure. */ 1388 idA.User = (unsigned char*) test_user; 1389 idA.UserLength = strlen(test_user); 1390 idA.Domain = (unsigned char *) workgroup; 1391 idA.DomainLength = strlen(workgroup); 1392 idA.Password = (unsigned char*) test_pass; 1393 idA.PasswordLength = strlen(test_pass); 1394 idA.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 1395 1396 ret = pAcquireCredentialsHandleW(NULL, sec_pkg_nameW, SECPKG_CRED_OUTBOUND, 1397 NULL, &idA, NULL, NULL, &cred, &ttl); 1398 ok(ret == SEC_E_OK, "AcquireCredentialsHandeW() returned %s\n", 1399 getSecError(ret)); 1400 pFreeCredentialsHandle(&cred); 1401 } 1402 1403 static void test_cred_multiple_use(void) 1404 { 1405 SECURITY_STATUS ret; 1406 SEC_WINNT_AUTH_IDENTITY_A id; 1407 PSecPkgInfoA pkg_info = NULL; 1408 CredHandle cred; 1409 CtxtHandle ctxt1 = {0}; 1410 CtxtHandle ctxt2 = {0}; 1411 SecBufferDesc buffer_desc; 1412 SecBuffer buffers[1]; 1413 ULONG ctxt_attr; 1414 TimeStamp ttl; 1415 1416 if(pQuerySecurityPackageInfoA(sec_pkg_name, &pkg_info) != SEC_E_OK) 1417 { 1418 ok(0, "NTLM package not installed, skipping test\n"); 1419 return; 1420 } 1421 buffers[0].cbBuffer = pkg_info->cbMaxToken; 1422 buffers[0].BufferType = SECBUFFER_TOKEN; 1423 buffers[0].pvBuffer = HeapAlloc(GetProcessHeap(), 0, buffers[0].cbBuffer); 1424 1425 pFreeContextBuffer(pkg_info); 1426 1427 id.User = (unsigned char*) test_user; 1428 id.UserLength = strlen((char *) id.User); 1429 id.Domain = (unsigned char *) workgroup; 1430 id.DomainLength = strlen((char *) id.Domain); 1431 id.Password = (unsigned char*) test_pass; 1432 id.PasswordLength = strlen((char *) id.Password); 1433 id.Flags = SEC_WINNT_AUTH_IDENTITY_ANSI; 1434 1435 ret = pAcquireCredentialsHandleA(NULL, sec_pkg_name, SECPKG_CRED_OUTBOUND, 1436 NULL, &id, NULL, NULL, &cred, &ttl); 1437 ok(ret == SEC_E_OK, "AcquireCredentialsHandle() returned %s\n", 1438 getSecError(ret)); 1439 1440 buffer_desc.ulVersion = SECBUFFER_VERSION; 1441 buffer_desc.cBuffers = ARRAY_SIZE(buffers); 1442 buffer_desc.pBuffers = buffers; 1443 1444 ret = pInitializeSecurityContextA(&cred, NULL, NULL, ISC_REQ_CONNECTION, 1445 0, SECURITY_NETWORK_DREP, NULL, 0, &ctxt1, &buffer_desc, 1446 &ctxt_attr, &ttl); 1447 ok(ret == SEC_I_CONTINUE_NEEDED, "InitializeSecurityContextA failed with error 0x%x\n", ret); 1448 1449 ret = pInitializeSecurityContextA(&cred, NULL, NULL, ISC_REQ_CONNECTION, 1450 0, SECURITY_NETWORK_DREP, NULL, 0, &ctxt2, &buffer_desc, 1451 &ctxt_attr, &ttl); 1452 ok(ret == SEC_I_CONTINUE_NEEDED, "Second InitializeSecurityContextA on cred handle failed with error 0x%x\n", ret); 1453 1454 ret = pDeleteSecurityContext(&ctxt1); 1455 ok(ret == SEC_E_OK, "DeleteSecurityContext failed with error 0x%x\n", ret); 1456 ret = pDeleteSecurityContext(&ctxt2); 1457 ok(ret == SEC_E_OK, "DeleteSecurityContext failed with error 0x%x\n", ret); 1458 ret = pFreeCredentialsHandle(&cred); 1459 ok(ret == SEC_E_OK, "FreeCredentialsHandle failed with error 0x%x\n", ret); 1460 1461 HeapFree(GetProcessHeap(), 0, buffers[0].pvBuffer); 1462 } 1463 1464 static void test_null_auth_data(void) 1465 { 1466 SECURITY_STATUS status; 1467 PSecPkgInfoA info; 1468 CredHandle cred; 1469 CtxtHandle ctx = {0}; 1470 SecBufferDesc buffer_desc; 1471 SecBuffer buffers[1]; 1472 char user[256]; 1473 TimeStamp ttl; 1474 ULONG attr, size; 1475 BOOLEAN ret; 1476 1477 if(pQuerySecurityPackageInfoA((SEC_CHAR *)"NTLM", &info) != SEC_E_OK) 1478 { 1479 ok(0, "NTLM package not installed, skipping test\n"); 1480 return; 1481 } 1482 1483 status = pAcquireCredentialsHandleA(NULL, (SEC_CHAR *)"NTLM", SECPKG_CRED_OUTBOUND, 1484 NULL, NULL, NULL, NULL, &cred, &ttl); 1485 ok(status == SEC_E_OK, "AcquireCredentialsHandle() failed %s\n", getSecError(status)); 1486 1487 buffers[0].cbBuffer = info->cbMaxToken; 1488 buffers[0].BufferType = SECBUFFER_TOKEN; 1489 buffers[0].pvBuffer = HeapAlloc(GetProcessHeap(), 0, buffers[0].cbBuffer); 1490 1491 buffer_desc.ulVersion = SECBUFFER_VERSION; 1492 buffer_desc.cBuffers = ARRAY_SIZE(buffers); 1493 buffer_desc.pBuffers = buffers; 1494 1495 size = sizeof(user); 1496 ret = pGetUserNameExA(NameSamCompatible, user, &size); 1497 ok(ret, "GetUserNameExA failed %u\n", GetLastError()); 1498 1499 status = pInitializeSecurityContextA(&cred, NULL, (SEC_CHAR *)user, 1500 ISC_REQ_CONNECTION, 0, SECURITY_NETWORK_DREP, 1501 NULL, 0, &ctx, &buffer_desc, &attr, &ttl); 1502 ok(status == SEC_I_CONTINUE_NEEDED, "InitializeSecurityContextA failed %s\n", getSecError(status)); 1503 1504 ret = pDeleteSecurityContext(&ctx); 1505 ok(ret == SEC_E_OK, "DeleteSecurityContext failed with error 0x%x\n", ret); 1506 ret = pFreeCredentialsHandle(&cred); 1507 ok(ret == SEC_E_OK, "FreeCredentialsHandle failed with error 0x%x\n", ret); 1508 1509 pFreeContextBuffer(info); 1510 HeapFree(GetProcessHeap(), 0, buffers[0].pvBuffer); 1511 } 1512 1513 START_TEST(ntlm) 1514 { 1515 InitFunctionPtrs(); 1516 1517 if(pFreeCredentialsHandle && pDeleteSecurityContext && 1518 pAcquireCredentialsHandleA && pInitializeSecurityContextA && 1519 pCompleteAuthToken && pQuerySecurityPackageInfoA) 1520 { 1521 testAcquireCredentialsHandleW(); 1522 1523 if(!testAcquireCredentialsHandle()) 1524 goto cleanup; 1525 testInitializeSecurityContextFlags(); 1526 if(pAcceptSecurityContext) 1527 { 1528 testAuth(SECURITY_NATIVE_DREP, TRUE); 1529 testAuth(SECURITY_NETWORK_DREP, TRUE); 1530 testAuth(SECURITY_NATIVE_DREP, FALSE); 1531 testAuth(SECURITY_NETWORK_DREP, FALSE); 1532 } 1533 if(pMakeSignature && pVerifySignature && pEncryptMessage && 1534 pDecryptMessage) 1535 testSignSeal(); 1536 1537 test_cred_multiple_use(); 1538 if (pGetUserNameExA) test_null_auth_data(); 1539 } 1540 else 1541 win_skip("Needed functions are not available\n"); 1542 1543 cleanup: 1544 if(secdll) 1545 FreeLibrary(secdll); 1546 } 1547