?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
secur32: Validate output buffer size in schan_InitializeSecurityContextW().
[wine.git] / dlls / secur32 / schannel.c
blob141a191c7c679cc57f3d90aac1e9237402490aea
1 /* Copyright (C) 2005 Juan Lang
2 * Copyright 2008 Henri Verbeet
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 *
18 * This file implements the schannel provider, or, the SSL/TLS implementations.
19 */
20
21 #include <assert.h>
22 #include <stdarg.h>
23 #include <stdlib.h>
24 #include <errno.h>
25
26 #define NONAMELESSUNION
27 #include "windef.h"
28 #include "winbase.h"
29 #include "winternl.h"
30 #include "winreg.h"
31 #include "winnls.h"
32 #include "lmcons.h"
33 #include "sspi.h"
34 #include "schannel.h"
35
36 #include "wine/unixlib.h"
37 #include "wine/debug.h"
38 #include "secur32_priv.h"
39
40 WINE_DEFAULT_DEBUG_CHANNEL(secur32);
41
42 static unixlib_handle_t gnutls_handle;
43
44 #define GNUTLS_CALL( func, params ) __wine_unix_call( gnutls_handle, unix_ ## func, params )
45
46 #define SCHAN_INVALID_HANDLE ~0UL
47
48 enum schan_handle_type
49 {
50 SCHAN_HANDLE_CRED,
51 SCHAN_HANDLE_CTX,
52 SCHAN_HANDLE_FREE
53 };
54
55 struct schan_handle
56 {
57 void *object;
58 enum schan_handle_type type;
59 };
60
61 struct schan_context
62 {
63 struct schan_transport transport;
64 ULONG req_ctx_attr;
65 const CERT_CONTEXT *cert;
66 SIZE_T header_size;
67 };
68
69 static struct schan_handle *schan_handle_table;
70 static struct schan_handle *schan_free_handles;
71 static SIZE_T schan_handle_table_size;
72 static SIZE_T schan_handle_count;
73
74 /* Protocols enabled, only those may be used for the connection. */
75 static DWORD config_enabled_protocols;
76
77 /* Protocols disabled by default. They are enabled for using, but disabled when caller asks for default settings. */
78 static DWORD config_default_disabled_protocols;
79
80 static ULONG_PTR schan_alloc_handle(void *object, enum schan_handle_type type)
81 {
82 struct schan_handle *handle;
83
84 if (schan_free_handles)
85 {
86 DWORD index = schan_free_handles - schan_handle_table;
87 /* Use a free handle */
88 handle = schan_free_handles;
89 if (handle->type != SCHAN_HANDLE_FREE)
90 {
91 ERR("Handle %d(%p) is in the free list, but has type %#x.\n", index, handle, handle->type);
92 return SCHAN_INVALID_HANDLE;
93 }
94 schan_free_handles = handle->object;
95 handle->object = object;
96 handle->type = type;
97
98 return index;
99 }
100 if (!(schan_handle_count < schan_handle_table_size))
101 {
102 /* Grow the table */
103 SIZE_T new_size = schan_handle_table_size + (schan_handle_table_size >> 1);
104 struct schan_handle *new_table = realloc(schan_handle_table, new_size * sizeof(*schan_handle_table));
105 if (!new_table)
106 {
107 ERR("Failed to grow the handle table\n");
108 return SCHAN_INVALID_HANDLE;
109 }
110 schan_handle_table = new_table;
111 schan_handle_table_size = new_size;
112 }
113
114 handle = &schan_handle_table[schan_handle_count++];
115 handle->object = object;
116 handle->type = type;
117
118 return handle - schan_handle_table;
119 }
120
121 static void *schan_free_handle(ULONG_PTR handle_idx, enum schan_handle_type type)
122 {
123 struct schan_handle *handle;
124 void *object;
125
126 if (handle_idx == SCHAN_INVALID_HANDLE) return NULL;
127 if (handle_idx >= schan_handle_count) return NULL;
128 handle = &schan_handle_table[handle_idx];
129 if (handle->type != type)
130 {
131 ERR("Handle %ld(%p) is not of type %#x\n", handle_idx, handle, type);
132 return NULL;
133 }
134
135 object = handle->object;
136 handle->object = schan_free_handles;
137 handle->type = SCHAN_HANDLE_FREE;
138 schan_free_handles = handle;
139
140 return object;
141 }
142
143 static void *schan_get_object(ULONG_PTR handle_idx, enum schan_handle_type type)
144 {
145 struct schan_handle *handle;
146
147 if (handle_idx == SCHAN_INVALID_HANDLE) return NULL;
148 if (handle_idx >= schan_handle_count) return NULL;
149 handle = &schan_handle_table[handle_idx];
150 if (handle->type != type)
151 {
152 ERR("Handle %ld(%p) is not of type %#x\n", handle_idx, handle, type);
153 return NULL;
154 }
155
156 return handle->object;
157 }
158
159 static void read_config(void)
160 {
161 DWORD enabled = 0, default_disabled = 0;
162 HKEY protocols_key, key;
163 WCHAR subkey_name[64];
164 unsigned i;
165 DWORD res;
166
167 static BOOL config_read = FALSE;
168 static const struct {
169 WCHAR key_name[20];
170 DWORD prot_client_flag;
171 BOOL enabled; /* If no config is present, enable the protocol */
172 BOOL disabled_by_default; /* Disable if caller asks for default protocol set */
173 } protocol_config_keys[] = {
174 { L"SSL 2.0", SP_PROT_SSL2_CLIENT, FALSE, TRUE }, /* NOTE: TRUE, TRUE on Windows */
175 { L"SSL 3.0", SP_PROT_SSL3_CLIENT, TRUE, FALSE },
176 { L"TLS 1.0", SP_PROT_TLS1_0_CLIENT, TRUE, FALSE },
177 { L"TLS 1.1", SP_PROT_TLS1_1_CLIENT, TRUE, FALSE /* NOTE: not enabled by default on Windows */ },
178 { L"TLS 1.2", SP_PROT_TLS1_2_CLIENT, TRUE, FALSE /* NOTE: not enabled by default on Windows */ },
179 { L"DTLS 1.0", SP_PROT_DTLS1_0_CLIENT, TRUE, TRUE },
180 { L"DTLS 1.2", SP_PROT_DTLS1_2_CLIENT, TRUE, TRUE },
181 };
182
183 /* No need for thread safety */
184 if(config_read)
185 return;
186
187 res = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
188 L"SYSTEM\\CurrentControlSet\\Control\\SecurityProviders\\SCHANNEL\\Protocols", 0, KEY_READ,
189 &protocols_key);
190 if(res == ERROR_SUCCESS) {
191 DWORD type, size, value;
192
193 for(i = 0; i < ARRAY_SIZE(protocol_config_keys); i++) {
194 wcscpy(subkey_name, protocol_config_keys[i].key_name);
195 wcscat(subkey_name, L"\\Client");
196 res = RegOpenKeyExW(protocols_key, subkey_name, 0, KEY_READ, &key);
197 if(res != ERROR_SUCCESS) {
198 if(protocol_config_keys[i].enabled)
199 enabled |= protocol_config_keys[i].prot_client_flag;
200 if(protocol_config_keys[i].disabled_by_default)
201 default_disabled |= protocol_config_keys[i].prot_client_flag;
202 continue;
203 }
204
205 size = sizeof(value);
206 res = RegQueryValueExW(key, L"enabled", NULL, &type, (BYTE *)&value, &size);
207 if(res == ERROR_SUCCESS) {
208 if(type == REG_DWORD && value)
209 enabled |= protocol_config_keys[i].prot_client_flag;
210 }else if(protocol_config_keys[i].enabled) {
211 enabled |= protocol_config_keys[i].prot_client_flag;
212 }
213
214 size = sizeof(value);
215 res = RegQueryValueExW(key, L"DisabledByDefault", NULL, &type, (BYTE *)&value, &size);
216 if(res == ERROR_SUCCESS) {
217 if(type != REG_DWORD || value)
218 default_disabled |= protocol_config_keys[i].prot_client_flag;
219 }else if(protocol_config_keys[i].disabled_by_default) {
220 default_disabled |= protocol_config_keys[i].prot_client_flag;
221 }
222
223 RegCloseKey(key);
224 }
225 }else {
226 /* No config, enable all known protocols. */
227 for(i = 0; i < ARRAY_SIZE(protocol_config_keys); i++) {
228 if(protocol_config_keys[i].enabled)
229 enabled |= protocol_config_keys[i].prot_client_flag;
230 if(protocol_config_keys[i].disabled_by_default)
231 default_disabled |= protocol_config_keys[i].prot_client_flag;
232 }
233 }
234
235 RegCloseKey(protocols_key);
236
237 config_enabled_protocols = enabled & GNUTLS_CALL( get_enabled_protocols, NULL );
238 config_default_disabled_protocols = default_disabled;
239 config_read = TRUE;
240
241 TRACE("enabled %x, disabled by default %x\n", config_enabled_protocols, config_default_disabled_protocols);
242 }
243
244 static SECURITY_STATUS schan_QueryCredentialsAttributes(
245 PCredHandle phCredential, ULONG ulAttribute, VOID *pBuffer)
246 {
247 struct schan_credentials *cred;
248 SECURITY_STATUS ret;
249
250 cred = schan_get_object(phCredential->dwLower, SCHAN_HANDLE_CRED);
251 if(!cred)
252 return SEC_E_INVALID_HANDLE;
253
254 switch (ulAttribute)
255 {
256 case SECPKG_ATTR_SUPPORTED_ALGS:
257 if (pBuffer)
258 {
259 /* FIXME: get from CryptoAPI */
260 FIXME("SECPKG_ATTR_SUPPORTED_ALGS: stub\n");
261 ret = SEC_E_UNSUPPORTED_FUNCTION;
262 }
263 else
264 ret = SEC_E_INTERNAL_ERROR;
265 break;
266 case SECPKG_ATTR_CIPHER_STRENGTHS:
267 if (pBuffer)
268 {
269 SecPkgCred_CipherStrengths *r = pBuffer;
270
271 /* FIXME: get from CryptoAPI */
272 FIXME("SECPKG_ATTR_CIPHER_STRENGTHS: semi-stub\n");
273 r->dwMinimumCipherStrength = 40;
274 r->dwMaximumCipherStrength = 168;
275 ret = SEC_E_OK;
276 }
277 else
278 ret = SEC_E_INTERNAL_ERROR;
279 break;
280 case SECPKG_ATTR_SUPPORTED_PROTOCOLS:
281 if(pBuffer) {
282 /* Regardless of MSDN documentation, tests show that this attribute takes into account
283 * what protocols are enabled for given credential. */
284 ((SecPkgCred_SupportedProtocols*)pBuffer)->grbitProtocol = cred->enabled_protocols;
285 ret = SEC_E_OK;
286 }else {
287 ret = SEC_E_INTERNAL_ERROR;
288 }
289 break;
290 default:
291 ret = SEC_E_UNSUPPORTED_FUNCTION;
292 }
293 return ret;
294 }
295
296 static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesA(
297 PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer)
298 {
299 SECURITY_STATUS ret;
300
301 TRACE("(%p, %d, %p)\n", phCredential, ulAttribute, pBuffer);
302
303 switch (ulAttribute)
304 {
305 case SECPKG_CRED_ATTR_NAMES:
306 FIXME("SECPKG_CRED_ATTR_NAMES: stub\n");
307 ret = SEC_E_UNSUPPORTED_FUNCTION;
308 break;
309 default:
310 ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute,
311 pBuffer);
312 }
313 return ret;
314 }
315
316 static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesW(
317 PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer)
318 {
319 SECURITY_STATUS ret;
320
321 TRACE("(%p, %d, %p)\n", phCredential, ulAttribute, pBuffer);
322
323 switch (ulAttribute)
324 {
325 case SECPKG_CRED_ATTR_NAMES:
326 FIXME("SECPKG_CRED_ATTR_NAMES: stub\n");
327 ret = SEC_E_UNSUPPORTED_FUNCTION;
328 break;
329 default:
330 ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute,
331 pBuffer);
332 }
333 return ret;
334 }
335
336 static SECURITY_STATUS get_cert(const SCHANNEL_CRED *cred, CERT_CONTEXT const **cert)
337 {
338 SECURITY_STATUS status;
339 DWORD i;
340
341 TRACE("dwVersion = %u\n", cred->dwVersion);
342 TRACE("cCreds = %u\n", cred->cCreds);
343 TRACE("paCred = %p\n", cred->paCred);
344 TRACE("hRootStore = %p\n", cred->hRootStore);
345 TRACE("cMappers = %u\n", cred->cMappers);
346 TRACE("cSupportedAlgs = %u:\n", cred->cSupportedAlgs);
347 for (i = 0; i < cred->cSupportedAlgs; i++) TRACE("%08x\n", cred->palgSupportedAlgs[i]);
348 TRACE("grbitEnabledProtocols = %08x\n", cred->grbitEnabledProtocols);
349 TRACE("dwMinimumCipherStrength = %u\n", cred->dwMinimumCipherStrength);
350 TRACE("dwMaximumCipherStrength = %u\n", cred->dwMaximumCipherStrength);
351 TRACE("dwSessionLifespan = %u\n", cred->dwSessionLifespan);
352 TRACE("dwFlags = %08x\n", cred->dwFlags);
353 TRACE("dwCredFormat = %u\n", cred->dwCredFormat);
354
355 switch (cred->dwVersion)
356 {
357 case SCH_CRED_V3:
358 case SCHANNEL_CRED_VERSION:
359 break;
360 default:
361 return SEC_E_INTERNAL_ERROR;
362 }
363
364 if (!cred->cCreds) status = SEC_E_NO_CREDENTIALS;
365 else if (cred->cCreds > 1) status = SEC_E_UNKNOWN_CREDENTIALS;
366 else
367 {
368 DWORD spec;
369 HCRYPTPROV prov;
370 BOOL free;
371
372 if (CryptAcquireCertificatePrivateKey(cred->paCred[0], CRYPT_ACQUIRE_CACHE_FLAG, NULL, &prov, &spec, &free))
373 {
374 if (free) CryptReleaseContext(prov, 0);
375 *cert = cred->paCred[0];
376 status = SEC_E_OK;
377 }
378 else status = SEC_E_UNKNOWN_CREDENTIALS;
379 }
380
381 return status;
382 }
383
384 static WCHAR *get_key_container_path(const CERT_CONTEXT *ctx)
385 {
386 CERT_KEY_CONTEXT keyctx;
387 DWORD size = sizeof(keyctx), prov_size = 0;
388 CRYPT_KEY_PROV_INFO *prov;
389 WCHAR username[UNLEN + 1], *ret = NULL;
390 DWORD len = ARRAY_SIZE(username);
391
392 if (CertGetCertificateContextProperty(ctx, CERT_KEY_CONTEXT_PROP_ID, &keyctx, &size))
393 {
394 char *str;
395 if (!CryptGetProvParam(keyctx.hCryptProv, PP_CONTAINER, NULL, &size, 0)) return NULL;
396 if (!(str = RtlAllocateHeap(GetProcessHeap(), 0, size))) return NULL;
397 if (!CryptGetProvParam(keyctx.hCryptProv, PP_CONTAINER, (BYTE *)str, &size, 0)) return NULL;
398
399 len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
400 if (!(ret = RtlAllocateHeap(GetProcessHeap(), 0, sizeof(L"Software\\Wine\\Crypto\\RSA\\") + len * sizeof(WCHAR))))
401 {
402 RtlFreeHeap(GetProcessHeap(), 0, str);
403 return NULL;
404 }
405 wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\");
406 MultiByteToWideChar(CP_ACP, 0, str, -1, ret + wcslen(ret), len);
407 RtlFreeHeap(GetProcessHeap(), 0, str);
408 }
409 else if (CertGetCertificateContextProperty(ctx, CERT_KEY_PROV_INFO_PROP_ID, NULL, &prov_size))
410 {
411 if (!(prov = RtlAllocateHeap(GetProcessHeap(), 0, prov_size))) return NULL;
412 if (!CertGetCertificateContextProperty(ctx, CERT_KEY_PROV_INFO_PROP_ID, prov, &prov_size))
413 {
414 RtlFreeHeap(GetProcessHeap(), 0, prov);
415 return NULL;
416 }
417 if (!(ret = RtlAllocateHeap(GetProcessHeap(), 0,
418 sizeof(L"Software\\Wine\\Crypto\\RSA\\") + wcslen(prov->pwszContainerName) * sizeof(WCHAR))))
419 {
420 RtlFreeHeap(GetProcessHeap(), 0, prov);
421 return NULL;
422 }
423 wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\");
424 wcscat(ret, prov->pwszContainerName);
425 RtlFreeHeap(GetProcessHeap(), 0, prov);
426 }
427
428 if (!ret && GetUserNameW(username, &len) &&
429 (ret = RtlAllocateHeap(GetProcessHeap(), 0, sizeof(L"Software\\Wine\\Crypto\\RSA\\") + len * sizeof(WCHAR))))
430 {
431 wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\");
432 wcscat(ret, username);
433 }
434
435 return ret;
436 }
437
438 #define MAX_LEAD_BYTES 8
439 static BYTE *get_key_blob(const CERT_CONTEXT *ctx, DWORD *size)
440 {
441 BYTE *buf, *ret = NULL;
442 DATA_BLOB blob_in, blob_out;
443 DWORD spec = 0, type, len;
444 WCHAR *path;
445 HKEY hkey;
446
447 if (!(path = get_key_container_path(ctx))) return NULL;
448 if (RegOpenKeyExW(HKEY_CURRENT_USER, path, 0, KEY_READ, &hkey))
449 {
450 RtlFreeHeap(GetProcessHeap(), 0, path);
451 return NULL;
452 }
453 RtlFreeHeap(GetProcessHeap(), 0, path);
454
455 if (!RegQueryValueExW(hkey, L"KeyExchangeKeyPair", 0, &type, NULL, &len)) spec = AT_KEYEXCHANGE;
456 else if (!RegQueryValueExW(hkey, L"SignatureKeyPair", 0, &type, NULL, &len)) spec = AT_SIGNATURE;
457 else
458 {
459 RegCloseKey(hkey);
460 return NULL;
461 }
462
463 if (!(buf = RtlAllocateHeap(GetProcessHeap(), 0, len + MAX_LEAD_BYTES)))
464 {
465 RegCloseKey(hkey);
466 return NULL;
467 }
468
469 if (!RegQueryValueExW(hkey, (spec == AT_KEYEXCHANGE) ? L"KeyExchangeKeyPair" : L"SignatureKeyPair", 0,
470 &type, buf, &len))
471 {
472 blob_in.pbData = buf;
473 blob_in.cbData = len;
474 if (CryptUnprotectData(&blob_in, NULL, NULL, NULL, NULL, 0, &blob_out))
475 {
476 assert(blob_in.cbData >= blob_out.cbData);
477 memcpy(buf, blob_out.pbData, blob_out.cbData);
478 LocalFree(blob_out.pbData);
479 *size = blob_out.cbData + MAX_LEAD_BYTES;
480 ret = buf;
481 }
482 }
483 else RtlFreeHeap(GetProcessHeap(), 0, buf);
484
485 RegCloseKey(hkey);
486 return ret;
487 }
488
489 static SECURITY_STATUS schan_AcquireClientCredentials(const SCHANNEL_CRED *schanCred,
490 PCredHandle phCredential, PTimeStamp ptsExpiry)
491 {
492 struct schan_credentials *creds;
493 unsigned enabled_protocols;
494 ULONG_PTR handle;
495 SECURITY_STATUS status = SEC_E_OK;
496 const CERT_CONTEXT *cert = NULL;
497 DATA_BLOB key_blob = {0};
498 struct allocate_certificate_credentials_params params;
499
500 TRACE("schanCred %p, phCredential %p, ptsExpiry %p\n", schanCred, phCredential, ptsExpiry);
501
502 if (schanCred)
503 {
504 const unsigned dtls_protocols = SP_PROT_DTLS_CLIENT | SP_PROT_DTLS1_2_CLIENT;
505 const unsigned tls_protocols = SP_PROT_TLS1_CLIENT | SP_PROT_TLS1_0_CLIENT | SP_PROT_TLS1_1_CLIENT |
506 SP_PROT_TLS1_2_CLIENT | SP_PROT_TLS1_3_CLIENT;
507
508 status = get_cert(schanCred, &cert);
509 if (status != SEC_E_OK && status != SEC_E_NO_CREDENTIALS)
510 return status;
511
512 if ((schanCred->grbitEnabledProtocols & tls_protocols) &&
513 (schanCred->grbitEnabledProtocols & dtls_protocols)) return SEC_E_ALGORITHM_MISMATCH;
514
515 status = SEC_E_OK;
516 }
517
518 read_config();
519 if(schanCred && schanCred->grbitEnabledProtocols)
520 enabled_protocols = schanCred->grbitEnabledProtocols & config_enabled_protocols;
521 else
522 enabled_protocols = config_enabled_protocols & ~config_default_disabled_protocols;
523 if(!enabled_protocols) {
524 ERR("Could not find matching protocol\n");
525 return SEC_E_NO_AUTHENTICATING_AUTHORITY;
526 }
527
528 if (!(creds = malloc(sizeof(*creds)))) return SEC_E_INSUFFICIENT_MEMORY;
529 creds->credential_use = SECPKG_CRED_OUTBOUND;
530 creds->enabled_protocols = enabled_protocols;
531
532 if (cert && !(key_blob.pbData = get_key_blob(cert, &key_blob.cbData))) goto fail;
533 params.c = creds;
534 params.ctx = cert;
535 params.key_blob = &key_blob;
536 if (GNUTLS_CALL( allocate_certificate_credentials, &params )) goto fail;
537 RtlFreeHeap(GetProcessHeap(), 0, key_blob.pbData);
538
539 handle = schan_alloc_handle(creds, SCHAN_HANDLE_CRED);
540 if (handle == SCHAN_INVALID_HANDLE) goto fail;
541
542 phCredential->dwLower = handle;
543 phCredential->dwUpper = 0;
544
545 /* Outbound credentials have no expiry */
546 if (ptsExpiry)
547 {
548 ptsExpiry->LowPart = 0;
549 ptsExpiry->HighPart = 0;
550 }
551
552 return status;
553
554 fail:
555 free(creds);
556 RtlFreeHeap(GetProcessHeap(), 0, key_blob.pbData);
557 return SEC_E_INTERNAL_ERROR;
558 }
559
560 static SECURITY_STATUS schan_AcquireServerCredentials(const SCHANNEL_CRED *schanCred,
561 PCredHandle phCredential, PTimeStamp ptsExpiry)
562 {
563 SECURITY_STATUS status;
564 const CERT_CONTEXT *cert = NULL;
565
566 TRACE("schanCred %p, phCredential %p, ptsExpiry %p\n", schanCred, phCredential, ptsExpiry);
567
568 if (!schanCred) return SEC_E_NO_CREDENTIALS;
569
570 status = get_cert(schanCred, &cert);
571 if (status == SEC_E_OK)
572 {
573 ULONG_PTR handle;
574 struct schan_credentials *creds;
575
576 if (!(creds = calloc(1, sizeof(*creds)))) return SEC_E_INSUFFICIENT_MEMORY;
577 creds->credential_use = SECPKG_CRED_INBOUND;
578
579 handle = schan_alloc_handle(creds, SCHAN_HANDLE_CRED);
580 if (handle == SCHAN_INVALID_HANDLE)
581 {
582 free(creds);
583 return SEC_E_INTERNAL_ERROR;
584 }
585
586 phCredential->dwLower = handle;
587 phCredential->dwUpper = 0;
588
589 /* FIXME: get expiry from cert */
590 }
591 return status;
592 }
593
594 static SECURITY_STATUS schan_AcquireCredentialsHandle(ULONG fCredentialUse,
595 const SCHANNEL_CRED *schanCred, PCredHandle phCredential, PTimeStamp ptsExpiry)
596 {
597 SECURITY_STATUS ret;
598
599 if (fCredentialUse == SECPKG_CRED_OUTBOUND)
600 ret = schan_AcquireClientCredentials(schanCred, phCredential,
601 ptsExpiry);
602 else
603 ret = schan_AcquireServerCredentials(schanCred, phCredential,
604 ptsExpiry);
605 return ret;
606 }
607
608 static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleA(
609 SEC_CHAR *pszPrincipal, SEC_CHAR *pszPackage, ULONG fCredentialUse,
610 PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn,
611 PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry)
612 {
613 TRACE("(%s, %s, 0x%08x, %p, %p, %p, %p, %p, %p)\n",
614 debugstr_a(pszPrincipal), debugstr_a(pszPackage), fCredentialUse,
615 pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry);
616 return schan_AcquireCredentialsHandle(fCredentialUse,
617 pAuthData, phCredential, ptsExpiry);
618 }
619
620 static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleW(
621 SEC_WCHAR *pszPrincipal, SEC_WCHAR *pszPackage, ULONG fCredentialUse,
622 PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn,
623 PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry)
624 {
625 TRACE("(%s, %s, 0x%08x, %p, %p, %p, %p, %p, %p)\n",
626 debugstr_w(pszPrincipal), debugstr_w(pszPackage), fCredentialUse,
627 pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry);
628 return schan_AcquireCredentialsHandle(fCredentialUse,
629 pAuthData, phCredential, ptsExpiry);
630 }
631
632 static SECURITY_STATUS SEC_ENTRY schan_FreeCredentialsHandle(
633 PCredHandle phCredential)
634 {
635 struct schan_credentials *creds;
636
637 TRACE("phCredential %p\n", phCredential);
638
639 if (!phCredential) return SEC_E_INVALID_HANDLE;
640
641 creds = schan_free_handle(phCredential->dwLower, SCHAN_HANDLE_CRED);
642 if (!creds) return SEC_E_INVALID_HANDLE;
643
644 if (creds->credential_use == SECPKG_CRED_OUTBOUND)
645 {
646 struct free_certificate_credentials_params params = { creds };
647 GNUTLS_CALL( free_certificate_credentials, &params );
648 }
649 free(creds);
650 return SEC_E_OK;
651 }
652
653 static int schan_find_sec_buffer_idx(const SecBufferDesc *desc, unsigned int start_idx, ULONG buffer_type)
654 {
655 unsigned int i;
656 PSecBuffer buffer;
657
658 for (i = start_idx; i < desc->cBuffers; ++i)
659 {
660 buffer = &desc->pBuffers[i];
661 if ((buffer->BufferType | SECBUFFER_ATTRMASK) == (buffer_type | SECBUFFER_ATTRMASK))
662 return i;
663 }
664
665 return -1;
666 }
667
668 static void dump_buffer_desc(SecBufferDesc *desc)
669 {
670 unsigned int i;
671
672 if (!desc) return;
673 TRACE("Buffer desc %p:\n", desc);
674 for (i = 0; i < desc->cBuffers; ++i)
675 {
676 SecBuffer *b = &desc->pBuffers[i];
677 TRACE("\tbuffer %u: cbBuffer %d, BufferType %#x pvBuffer %p\n", i, b->cbBuffer, b->BufferType, b->pvBuffer);
678 }
679 }
680
681 #define HEADER_SIZE_TLS 5
682 #define HEADER_SIZE_DTLS 13
683
684 static inline SIZE_T read_record_size(const BYTE *buf, SIZE_T header_size)
685 {
686 return (buf[header_size - 2] << 8) | buf[header_size - 1];
687 }
688
689 static inline BOOL is_dtls_context(const struct schan_context *ctx)
690 {
691 return ctx->header_size == HEADER_SIZE_DTLS;
692 }
693
694 /***********************************************************************
695 * InitializeSecurityContextW
696 */
697 static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextW(
698 PCredHandle phCredential, PCtxtHandle phContext, SEC_WCHAR *pszTargetName,
699 ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep,
700 PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext,
701 PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry)
702 {
703 const ULONG extra_size = 0x10000;
704 struct schan_context *ctx;
705 struct schan_buffers *out_buffers;
706 struct schan_credentials *cred;
707 SIZE_T expected_size = ~0UL;
708 SECURITY_STATUS ret;
709 SecBuffer *buffer;
710 SecBuffer alloc_buffer = { 0 };
711 struct handshake_params params;
712 int idx, i;
713
714 TRACE("%p %p %s 0x%08x %d %d %p %d %p %p %p %p\n", phCredential, phContext,
715 debugstr_w(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
716 Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
717
718 dump_buffer_desc(pInput);
719 dump_buffer_desc(pOutput);
720
721 if (ptsExpiry)
722 {
723 ptsExpiry->LowPart = 0;
724 ptsExpiry->HighPart = 0;
725 }
726
727 if (!pOutput || !pOutput->cBuffers) return SEC_E_INVALID_TOKEN;
728 for (i = 0; i < pOutput->cBuffers; i++)
729 {
730 ULONG type = pOutput->pBuffers[i].BufferType;
731
732 if (type != SECBUFFER_TOKEN && type != SECBUFFER_ALERT) continue;
733 if (!pOutput->pBuffers[i].cbBuffer && !(fContextReq & ISC_REQ_ALLOCATE_MEMORY))
734 return SEC_E_INSUFFICIENT_MEMORY;
735 }
736
737 if (!phContext)
738 {
739 ULONG_PTR handle;
740 struct create_session_params create_params;
741
742 if (!phCredential) return SEC_E_INVALID_HANDLE;
743
744 cred = schan_get_object(phCredential->dwLower, SCHAN_HANDLE_CRED);
745 if (!cred) return SEC_E_INVALID_HANDLE;
746
747 if (!(cred->credential_use & SECPKG_CRED_OUTBOUND))
748 {
749 WARN("Invalid credential use %#x\n", cred->credential_use);
750 return SEC_E_INVALID_HANDLE;
751 }
752
753 if (!(ctx = malloc(sizeof(*ctx)))) return SEC_E_INSUFFICIENT_MEMORY;
754
755 ctx->cert = NULL;
756 handle = schan_alloc_handle(ctx, SCHAN_HANDLE_CTX);
757 if (handle == SCHAN_INVALID_HANDLE)
758 {
759 free(ctx);
760 return SEC_E_INTERNAL_ERROR;
761 }
762
763 create_params.transport = &ctx->transport;
764 create_params.cred = cred;
765 if (GNUTLS_CALL( create_session, &create_params ))
766 {
767 schan_free_handle(handle, SCHAN_HANDLE_CTX);
768 free(ctx);
769 return SEC_E_INTERNAL_ERROR;
770 }
771
772 if (cred->enabled_protocols & (SP_PROT_DTLS1_0_CLIENT | SP_PROT_DTLS1_2_CLIENT))
773 ctx->header_size = HEADER_SIZE_DTLS;
774 else
775 ctx->header_size = HEADER_SIZE_TLS;
776
777 ctx->transport.ctx = ctx;
778
779 if (pszTargetName && *pszTargetName)
780 {
781 UINT len = WideCharToMultiByte( CP_UNIXCP, 0, pszTargetName, -1, NULL, 0, NULL, NULL );
782 char *target = malloc( len );
783
784 if (target)
785 {
786 struct set_session_target_params params = { ctx->transport.session, target };
787 WideCharToMultiByte( CP_UNIXCP, 0, pszTargetName, -1, target, len, NULL, NULL );
788 GNUTLS_CALL( set_session_target, &params );
789 free( target );
790 }
791 }
792
793 if (pInput && (idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_APPLICATION_PROTOCOLS)) != -1)
794 {
795 struct set_application_protocols_params params = { ctx->transport.session,
796 pInput->pBuffers[idx].pvBuffer, pInput->pBuffers[idx].cbBuffer };
797 GNUTLS_CALL( set_application_protocols, &params );
798 }
799
800 if (pInput && (idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_DTLS_MTU)) != -1)
801 {
802 buffer = &pInput->pBuffers[idx];
803 if (buffer->cbBuffer >= sizeof(WORD))
804 {
805 struct set_dtls_mtu_params params = { ctx->transport.session, *(WORD *)buffer->pvBuffer };
806 GNUTLS_CALL( set_dtls_mtu, &params );
807 }
808 else WARN("invalid buffer size %u\n", buffer->cbBuffer);
809 }
810
811 phNewContext->dwLower = handle;
812 phNewContext->dwUpper = 0;
813 }
814 else
815 {
816 SIZE_T record_size = 0;
817 unsigned char *ptr;
818
819 if (!(ctx = schan_get_object(phContext->dwLower, SCHAN_HANDLE_CTX))) return SEC_E_INVALID_HANDLE;
820 if (!pInput) return is_dtls_context(ctx) ? SEC_E_INSUFFICIENT_MEMORY : SEC_E_INCOMPLETE_MESSAGE;
821 if ((idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_TOKEN)) == -1) return SEC_E_INCOMPLETE_MESSAGE;
822
823 buffer = &pInput->pBuffers[idx];
824 ptr = buffer->pvBuffer;
825 expected_size = 0;
826
827 while (buffer->cbBuffer > expected_size + ctx->header_size)
828 {
829 record_size = ctx->header_size + read_record_size(ptr, ctx->header_size);
830
831 if (buffer->cbBuffer < expected_size + record_size) break;
832 expected_size += record_size;
833 ptr += record_size;
834 }
835
836 if (!expected_size)
837 {
838 TRACE("Expected at least %lu bytes, but buffer only contains %u bytes.\n",
839 max(ctx->header_size + 1, record_size), buffer->cbBuffer);
840 return SEC_E_INCOMPLETE_MESSAGE;
841 }
842
843 TRACE("Using expected_size %lu.\n", expected_size);
844
845 if (phNewContext) *phNewContext = *phContext;
846 }
847
848 ctx->req_ctx_attr = fContextReq;
849
850 /* Perform the TLS handshake */
851 if (fContextReq & ISC_REQ_ALLOCATE_MEMORY)
852 {
853 alloc_buffer.cbBuffer = extra_size;
854 alloc_buffer.BufferType = SECBUFFER_TOKEN;
855 alloc_buffer.pvBuffer = RtlAllocateHeap( GetProcessHeap(), 0, extra_size );
856 }
857 params.session = ctx->transport.session;
858 params.input = pInput;
859 params.input_size = expected_size;
860 params.output = pOutput;
861 params.alloc_buffer = &alloc_buffer;
862 ret = GNUTLS_CALL( handshake, &params );
863
864 out_buffers = &ctx->transport.out;
865 if (out_buffers->current_buffer_idx != -1)
866 {
867 SecBuffer *buffer = &out_buffers->desc->pBuffers[out_buffers->current_buffer_idx];
868 buffer->cbBuffer = out_buffers->offset;
869 if (buffer->pvBuffer == alloc_buffer.pvBuffer)
870 {
871 RtlReAllocateHeap( GetProcessHeap(), HEAP_REALLOC_IN_PLACE_ONLY,
872 buffer->pvBuffer, buffer->cbBuffer );
873 alloc_buffer.pvBuffer = NULL;
874 }
875 }
876 else if (out_buffers->desc && out_buffers->desc->cBuffers > 0)
877 {
878 SecBuffer *buffer = &out_buffers->desc->pBuffers[0];
879 buffer->cbBuffer = 0;
880 }
881 RtlFreeHeap( GetProcessHeap(), 0, alloc_buffer.pvBuffer );
882
883 if(ctx->transport.in.offset && ctx->transport.in.offset != pInput->pBuffers[0].cbBuffer) {
884 if(pInput->cBuffers<2 || pInput->pBuffers[1].BufferType!=SECBUFFER_EMPTY)
885 return SEC_E_INVALID_TOKEN;
886
887 pInput->pBuffers[1].BufferType = SECBUFFER_EXTRA;
888 pInput->pBuffers[1].cbBuffer = pInput->pBuffers[0].cbBuffer-ctx->transport.in.offset;
889 }
890
891 *pfContextAttr = ISC_RET_REPLAY_DETECT | ISC_RET_SEQUENCE_DETECT | ISC_RET_CONFIDENTIALITY | ISC_RET_STREAM;
892 if (ctx->req_ctx_attr & ISC_REQ_EXTENDED_ERROR) *pfContextAttr |= ISC_RET_EXTENDED_ERROR;
893 if (ctx->req_ctx_attr & ISC_REQ_DATAGRAM) *pfContextAttr |= ISC_RET_DATAGRAM;
894 if (ctx->req_ctx_attr & ISC_REQ_ALLOCATE_MEMORY) *pfContextAttr |= ISC_RET_ALLOCATED_MEMORY;
895 if (ctx->req_ctx_attr & ISC_REQ_USE_SUPPLIED_CREDS) *pfContextAttr |= ISC_RET_USED_SUPPLIED_CREDS;
896 if (ctx->req_ctx_attr & ISC_REQ_MANUAL_CRED_VALIDATION) *pfContextAttr |= ISC_RET_MANUAL_CRED_VALIDATION;
897
898 return ret;
899 }
900
901 /***********************************************************************
902 * InitializeSecurityContextA
903 */
904 static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextA(
905 PCredHandle phCredential, PCtxtHandle phContext, SEC_CHAR *pszTargetName,
906 ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep,
907 PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext,
908 PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry)
909 {
910 SECURITY_STATUS ret;
911 SEC_WCHAR *target_name = NULL;
912
913 TRACE("%p %p %s %d %d %d %p %d %p %p %p %p\n", phCredential, phContext,
914 debugstr_a(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
915 Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
916
917 if (pszTargetName)
918 {
919 INT len = MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, NULL, 0);
920 if (!(target_name = malloc(len * sizeof(*target_name)))) return SEC_E_INSUFFICIENT_MEMORY;
921 MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, target_name, len);
922 }
923
924 ret = schan_InitializeSecurityContextW(phCredential, phContext, target_name,
925 fContextReq, Reserved1, TargetDataRep, pInput, Reserved2,
926 phNewContext, pOutput, pfContextAttr, ptsExpiry);
927
928 free(target_name);
929 return ret;
930 }
931
932 static void *get_alg_name(ALG_ID id, BOOL wide)
933 {
934 static const struct {
935 ALG_ID alg_id;
936 const char* name;
937 const WCHAR nameW[8];
938 } alg_name_map[] = {
939 { CALG_ECDSA, "ECDSA", L"ECDSA" },
940 { CALG_RSA_SIGN, "RSA", L"RSA" },
941 { CALG_DES, "DES", L"DES" },
942 { CALG_RC2, "RC2", L"RC2" },
943 { CALG_3DES, "3DES", L"3DES" },
944 { CALG_AES_128, "AES", L"AES" },
945 { CALG_AES_192, "AES", L"AES" },
946 { CALG_AES_256, "AES", L"AES" },
947 { CALG_RC4, "RC4", L"RC4" },
948 };
949 unsigned i;
950
951 for (i = 0; i < ARRAY_SIZE(alg_name_map); i++)
952 if (alg_name_map[i].alg_id == id)
953 return wide ? (void*)alg_name_map[i].nameW : (void*)alg_name_map[i].name;
954
955 FIXME("Unknown ALG_ID %04x\n", id);
956 return NULL;
957 }
958
959 static SECURITY_STATUS ensure_remote_cert(struct schan_context *ctx)
960 {
961 HCERTSTORE store;
962 PCCERT_CONTEXT cert = NULL;
963 SECURITY_STATUS status;
964 CERT_BLOB *certs;
965 ULONG count, size = 0;
966 struct get_session_peer_certificate_params params = { ctx->transport.session, NULL, &size, &count };
967
968 if (ctx->cert) return SEC_E_OK;
969 if (!(store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0, CERT_STORE_CREATE_NEW_FLAG, NULL)))
970 return GetLastError();
971
972 status = GNUTLS_CALL( get_session_peer_certificate, &params );
973 if (status != SEC_E_BUFFER_TOO_SMALL) goto done;
974 if (!(certs = malloc( size )))
975 {
976 status = SEC_E_INSUFFICIENT_MEMORY;
977 goto done;
978 }
979 params.certs = certs;
980 status = GNUTLS_CALL( get_session_peer_certificate, &params );
981 if (status == SEC_E_OK)
982 {
983 unsigned int i;
984 for (i = 0; i < count; i++)
985 {
986 if (!CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING, certs[i].pbData,
987 certs[i].cbData, CERT_STORE_ADD_REPLACE_EXISTING,
988 i ? NULL : &cert))
989 {
990 if (i) CertFreeCertificateContext(cert);
991 return GetLastError();
992 }
993 }
994 }
995 free(certs);
996 done:
997 ctx->cert = cert;
998 CertCloseStore(store, 0);
999 return status;
1000 }
1001
1002 static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesW(
1003 PCtxtHandle context_handle, ULONG attribute, PVOID buffer)
1004 {
1005 struct schan_context *ctx;
1006 SECURITY_STATUS status;
1007
1008 TRACE("context_handle %p, attribute %#x, buffer %p\n",
1009 context_handle, attribute, buffer);
1010
1011 if (!context_handle) return SEC_E_INVALID_HANDLE;
1012 ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
1013
1014 switch(attribute)
1015 {
1016 case SECPKG_ATTR_STREAM_SIZES:
1017 {
1018 SecPkgContext_ConnectionInfo info;
1019 struct get_connection_info_params params = { ctx->transport.session, &info };
1020 status = GNUTLS_CALL( get_connection_info, &params );
1021 if (status == SEC_E_OK)
1022 {
1023 struct session_params params = { ctx->transport.session };
1024 SecPkgContext_StreamSizes *stream_sizes = buffer;
1025 SIZE_T mac_size = info.dwHashStrength;
1026 unsigned int block_size = GNUTLS_CALL( get_session_cipher_block_size, &params );
1027 unsigned int message_size = GNUTLS_CALL( get_max_message_size, &params );
1028
1029 TRACE("Using header size %lu mac bytes %lu, message size %u, block size %u\n",
1030 ctx->header_size, mac_size, message_size, block_size);
1031
1032 /* These are defined by the TLS RFC */
1033 stream_sizes->cbHeader = ctx->header_size;
1034 stream_sizes->cbTrailer = mac_size + 256; /* Max 255 bytes padding + 1 for padding size */
1035 stream_sizes->cbMaximumMessage = message_size;
1036 stream_sizes->cbBuffers = 4;
1037 stream_sizes->cbBlockSize = block_size;
1038 }
1039
1040 return status;
1041 }
1042 case SECPKG_ATTR_KEY_INFO:
1043 {
1044 SecPkgContext_ConnectionInfo conn_info;
1045 struct get_connection_info_params params = { ctx->transport.session, &conn_info };
1046 status = GNUTLS_CALL( get_connection_info, &params );
1047 if (status == SEC_E_OK)
1048 {
1049 struct session_params params = { ctx->transport.session };
1050 SecPkgContext_KeyInfoW *info = buffer;
1051 info->KeySize = conn_info.dwCipherStrength;
1052 info->SignatureAlgorithm = GNUTLS_CALL( get_key_signature_algorithm, &params );
1053 info->EncryptAlgorithm = conn_info.aiCipher;
1054 info->sSignatureAlgorithmName = get_alg_name(info->SignatureAlgorithm, TRUE);
1055 info->sEncryptAlgorithmName = get_alg_name(info->EncryptAlgorithm, TRUE);
1056 }
1057 return status;
1058 }
1059 case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
1060 {
1061 PCCERT_CONTEXT *cert = buffer;
1062
1063 status = ensure_remote_cert(ctx);
1064 if(status != SEC_E_OK)
1065 return status;
1066
1067 *cert = CertDuplicateCertificateContext(ctx->cert);
1068 return SEC_E_OK;
1069 }
1070 case SECPKG_ATTR_CONNECTION_INFO:
1071 {
1072 SecPkgContext_ConnectionInfo *info = buffer;
1073 struct get_connection_info_params params = { ctx->transport.session, info };
1074 return GNUTLS_CALL( get_connection_info, &params );
1075 }
1076 case SECPKG_ATTR_ENDPOINT_BINDINGS:
1077 {
1078 SecPkgContext_Bindings *bindings = buffer;
1079 CCRYPT_OID_INFO *info;
1080 ALG_ID hash_alg = CALG_SHA_256;
1081 BYTE hash[1024];
1082 DWORD hash_size;
1083 char *p;
1084 BOOL r;
1085
1086 static const char prefix[] = "tls-server-end-point:";
1087
1088 status = ensure_remote_cert(ctx);
1089 if(status != SEC_E_OK)
1090 return status;
1091
1092 /* RFC 5929 */
1093 info = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, ctx->cert->pCertInfo->SignatureAlgorithm.pszObjId, 0);
1094 if(info && info->u.Algid != CALG_SHA1 && info->u.Algid != CALG_MD5)
1095 hash_alg = info->u.Algid;
1096
1097 hash_size = sizeof(hash);
1098 r = CryptHashCertificate(0, hash_alg, 0, ctx->cert->pbCertEncoded, ctx->cert->cbCertEncoded, hash, &hash_size);
1099 if(!r)
1100 return GetLastError();
1101
1102 bindings->BindingsLength = sizeof(*bindings->Bindings) + sizeof(prefix)-1 + hash_size;
1103 /* freed with FreeContextBuffer */
1104 bindings->Bindings = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, bindings->BindingsLength);
1105 if(!bindings->Bindings)
1106 return SEC_E_INSUFFICIENT_MEMORY;
1107
1108 bindings->Bindings->cbApplicationDataLength = sizeof(prefix)-1 + hash_size;
1109 bindings->Bindings->dwApplicationDataOffset = sizeof(*bindings->Bindings);
1110
1111 p = (char*)(bindings->Bindings+1);
1112 memcpy(p, prefix, sizeof(prefix)-1);
1113 p += sizeof(prefix)-1;
1114 memcpy(p, hash, hash_size);
1115 return SEC_E_OK;
1116 }
1117 case SECPKG_ATTR_UNIQUE_BINDINGS:
1118 {
1119 static const char prefix[] = "tls-unique:";
1120 SecPkgContext_Bindings *bindings = buffer;
1121 ULONG size;
1122 char *p;
1123 struct get_unique_channel_binding_params params = { ctx->transport.session, NULL, &size };
1124
1125 if (GNUTLS_CALL( get_unique_channel_binding, &params ) != SEC_E_BUFFER_TOO_SMALL)
1126 return SEC_E_INTERNAL_ERROR;
1127
1128 bindings->BindingsLength = sizeof(*bindings->Bindings) + sizeof(prefix)-1 + size;
1129 /* freed with FreeContextBuffer */
1130 bindings->Bindings = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, bindings->BindingsLength);
1131 if(!bindings->Bindings)
1132 return SEC_E_INSUFFICIENT_MEMORY;
1133
1134 bindings->Bindings->cbApplicationDataLength = sizeof(prefix)-1 + size;
1135 bindings->Bindings->dwApplicationDataOffset = sizeof(*bindings->Bindings);
1136
1137 p = (char*)(bindings->Bindings+1);
1138 memcpy(p, prefix, sizeof(prefix)-1);
1139 p += sizeof(prefix)-1;
1140 params.buffer = p;
1141 return GNUTLS_CALL( get_unique_channel_binding, &params );
1142 }
1143 case SECPKG_ATTR_APPLICATION_PROTOCOL:
1144 {
1145 SecPkgContext_ApplicationProtocol *protocol = buffer;
1146 struct get_application_protocol_params params = { ctx->transport.session, protocol };
1147 return GNUTLS_CALL( get_application_protocol, &params );
1148 }
1149
1150 default:
1151 FIXME("Unhandled attribute %#x\n", attribute);
1152 return SEC_E_UNSUPPORTED_FUNCTION;
1153 }
1154 }
1155
1156 static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesA(
1157 PCtxtHandle context_handle, ULONG attribute, PVOID buffer)
1158 {
1159 TRACE("context_handle %p, attribute %#x, buffer %p\n",
1160 context_handle, attribute, buffer);
1161
1162 switch(attribute)
1163 {
1164 case SECPKG_ATTR_STREAM_SIZES:
1165 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1166 case SECPKG_ATTR_KEY_INFO:
1167 {
1168 SECURITY_STATUS status = schan_QueryContextAttributesW(context_handle, attribute, buffer);
1169 if (status == SEC_E_OK)
1170 {
1171 SecPkgContext_KeyInfoA *info = buffer;
1172 info->sSignatureAlgorithmName = get_alg_name(info->SignatureAlgorithm, FALSE);
1173 info->sEncryptAlgorithmName = get_alg_name(info->EncryptAlgorithm, FALSE);
1174 }
1175 return status;
1176 }
1177 case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
1178 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1179 case SECPKG_ATTR_CONNECTION_INFO:
1180 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1181 case SECPKG_ATTR_ENDPOINT_BINDINGS:
1182 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1183 case SECPKG_ATTR_UNIQUE_BINDINGS:
1184 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1185 case SECPKG_ATTR_APPLICATION_PROTOCOL:
1186 return schan_QueryContextAttributesW(context_handle, attribute, buffer);
1187
1188 default:
1189 FIXME("Unhandled attribute %#x\n", attribute);
1190 return SEC_E_UNSUPPORTED_FUNCTION;
1191 }
1192 }
1193
1194 static SECURITY_STATUS SEC_ENTRY schan_EncryptMessage(PCtxtHandle context_handle,
1195 ULONG quality, PSecBufferDesc message, ULONG message_seq_no)
1196 {
1197 struct schan_context *ctx;
1198 struct send_params params;
1199 SECURITY_STATUS status;
1200 SecBuffer *buffer;
1201 SIZE_T data_size;
1202 SIZE_T length;
1203 char *data;
1204 int idx;
1205
1206 TRACE("context_handle %p, quality %d, message %p, message_seq_no %d\n",
1207 context_handle, quality, message, message_seq_no);
1208
1209 if (!context_handle) return SEC_E_INVALID_HANDLE;
1210 ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
1211
1212 dump_buffer_desc(message);
1213
1214 idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_DATA);
1215 if (idx == -1)
1216 {
1217 WARN("No data buffer passed\n");
1218 return SEC_E_INTERNAL_ERROR;
1219 }
1220 buffer = &message->pBuffers[idx];
1221
1222 data_size = buffer->cbBuffer;
1223 data = malloc(data_size);
1224 memcpy(data, buffer->pvBuffer, data_size);
1225
1226 length = data_size;
1227 params.session = ctx->transport.session;
1228 params.output = message;
1229 params.buffer = data;
1230 params.length = &length;
1231 status = GNUTLS_CALL( send, &params );
1232
1233 TRACE("Sent %ld bytes.\n", length);
1234
1235 if (length != data_size)
1236 status = SEC_E_INTERNAL_ERROR;
1237
1238 free(data);
1239
1240 TRACE("Returning %#x.\n", status);
1241
1242 return status;
1243 }
1244
1245 static int schan_validate_decrypt_buffer_desc(PSecBufferDesc message)
1246 {
1247 int data_idx = -1;
1248 unsigned int empty_count = 0;
1249 unsigned int i;
1250
1251 if (message->cBuffers < 4)
1252 {
1253 WARN("Less than four buffers passed\n");
1254 return -1;
1255 }
1256
1257 for (i = 0; i < message->cBuffers; ++i)
1258 {
1259 SecBuffer *b = &message->pBuffers[i];
1260 if (b->BufferType == SECBUFFER_DATA)
1261 {
1262 if (data_idx != -1)
1263 {
1264 WARN("More than one data buffer passed\n");
1265 return -1;
1266 }
1267 data_idx = i;
1268 }
1269 else if (b->BufferType == SECBUFFER_EMPTY)
1270 ++empty_count;
1271 }
1272
1273 if (data_idx == -1)
1274 {
1275 WARN("No data buffer passed\n");
1276 return -1;
1277 }
1278
1279 if (empty_count < 3)
1280 {
1281 WARN("Less than three empty buffers passed\n");
1282 return -1;
1283 }
1284
1285 return data_idx;
1286 }
1287
1288 static void schan_decrypt_fill_buffer(PSecBufferDesc message, ULONG buffer_type, void *data, ULONG size)
1289 {
1290 int idx;
1291 SecBuffer *buffer;
1292
1293 idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY);
1294 buffer = &message->pBuffers[idx];
1295
1296 buffer->BufferType = buffer_type;
1297 buffer->pvBuffer = data;
1298 buffer->cbBuffer = size;
1299 }
1300
1301 static SECURITY_STATUS SEC_ENTRY schan_DecryptMessage(PCtxtHandle context_handle,
1302 PSecBufferDesc message, ULONG message_seq_no, PULONG quality)
1303 {
1304 SECURITY_STATUS status = SEC_E_OK;
1305 struct schan_context *ctx;
1306 struct recv_params params;
1307 SecBuffer *buffer;
1308 SIZE_T data_size;
1309 char *data;
1310 unsigned expected_size;
1311 SIZE_T received = 0;
1312 int idx;
1313 unsigned char *buf_ptr;
1314
1315 TRACE("context_handle %p, message %p, message_seq_no %d, quality %p\n",
1316 context_handle, message, message_seq_no, quality);
1317
1318 if (!context_handle) return SEC_E_INVALID_HANDLE;
1319 ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
1320
1321 dump_buffer_desc(message);
1322
1323 idx = schan_validate_decrypt_buffer_desc(message);
1324 if (idx == -1)
1325 return SEC_E_INVALID_TOKEN;
1326 buffer = &message->pBuffers[idx];
1327 buf_ptr = buffer->pvBuffer;
1328
1329 expected_size = ctx->header_size + read_record_size(buf_ptr, ctx->header_size);
1330 if(buffer->cbBuffer < expected_size)
1331 {
1332 TRACE("Expected %u bytes, but buffer only contains %u bytes\n", expected_size, buffer->cbBuffer);
1333 buffer->BufferType = SECBUFFER_MISSING;
1334 buffer->cbBuffer = expected_size - buffer->cbBuffer;
1335
1336 /* This is a bit weird, but windows does it too */
1337 idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY);
1338 buffer = &message->pBuffers[idx];
1339 buffer->BufferType = SECBUFFER_MISSING;
1340 buffer->cbBuffer = expected_size - buffer->cbBuffer;
1341
1342 TRACE("Returning SEC_E_INCOMPLETE_MESSAGE\n");
1343 return SEC_E_INCOMPLETE_MESSAGE;
1344 }
1345
1346 data_size = expected_size - ctx->header_size;
1347 data = malloc(data_size);
1348
1349 received = data_size;
1350
1351 params.session = ctx->transport.session;
1352 params.input = message;
1353 params.input_size = expected_size;
1354 params.buffer = data;
1355 params.length = &received;
1356 status = GNUTLS_CALL( recv, &params );
1357
1358 if (status != SEC_E_OK && status != SEC_I_RENEGOTIATE)
1359 {
1360 free(data);
1361 ERR("Returning %x\n", status);
1362 return status;
1363 }
1364
1365 TRACE("Received %ld bytes\n", received);
1366
1367 memcpy(buf_ptr + ctx->header_size, data, received);
1368 free(data);
1369
1370 schan_decrypt_fill_buffer(message, SECBUFFER_DATA,
1371 buf_ptr + ctx->header_size, received);
1372
1373 schan_decrypt_fill_buffer(message, SECBUFFER_STREAM_TRAILER,
1374 buf_ptr + ctx->header_size + received, buffer->cbBuffer - ctx->header_size - received);
1375
1376 if(buffer->cbBuffer > expected_size)
1377 schan_decrypt_fill_buffer(message, SECBUFFER_EXTRA,
1378 buf_ptr + expected_size, buffer->cbBuffer - expected_size);
1379
1380 buffer->BufferType = SECBUFFER_STREAM_HEADER;
1381 buffer->cbBuffer = ctx->header_size;
1382
1383 return status;
1384 }
1385
1386 static SECURITY_STATUS SEC_ENTRY schan_DeleteSecurityContext(PCtxtHandle context_handle)
1387 {
1388 struct schan_context *ctx;
1389 struct session_params params;
1390
1391 TRACE("context_handle %p\n", context_handle);
1392
1393 if (!context_handle) return SEC_E_INVALID_HANDLE;
1394
1395 ctx = schan_free_handle(context_handle->dwLower, SCHAN_HANDLE_CTX);
1396 if (!ctx) return SEC_E_INVALID_HANDLE;
1397
1398 if (ctx->cert) CertFreeCertificateContext(ctx->cert);
1399 params.session = ctx->transport.session;
1400 GNUTLS_CALL( dispose_session, &params );
1401 free(ctx);
1402 return SEC_E_OK;
1403 }
1404
1405 static const SecurityFunctionTableA schanTableA = {
1406 1,
1407 NULL, /* EnumerateSecurityPackagesA */
1408 schan_QueryCredentialsAttributesA,
1409 schan_AcquireCredentialsHandleA,
1410 schan_FreeCredentialsHandle,
1411 NULL, /* Reserved2 */
1412 schan_InitializeSecurityContextA,
1413 NULL, /* AcceptSecurityContext */
1414 NULL, /* CompleteAuthToken */
1415 schan_DeleteSecurityContext,
1416 NULL, /* ApplyControlToken */
1417 schan_QueryContextAttributesA,
1418 NULL, /* ImpersonateSecurityContext */
1419 NULL, /* RevertSecurityContext */
1420 NULL, /* MakeSignature */
1421 NULL, /* VerifySignature */
1422 FreeContextBuffer,
1423 NULL, /* QuerySecurityPackageInfoA */
1424 NULL, /* Reserved3 */
1425 NULL, /* Reserved4 */
1426 NULL, /* ExportSecurityContext */
1427 NULL, /* ImportSecurityContextA */
1428 NULL, /* AddCredentialsA */
1429 NULL, /* Reserved8 */
1430 NULL, /* QuerySecurityContextToken */
1431 schan_EncryptMessage,
1432 schan_DecryptMessage,
1433 NULL, /* SetContextAttributesA */
1434 };
1435
1436 static const SecurityFunctionTableW schanTableW = {
1437 1,
1438 NULL, /* EnumerateSecurityPackagesW */
1439 schan_QueryCredentialsAttributesW,
1440 schan_AcquireCredentialsHandleW,
1441 schan_FreeCredentialsHandle,
1442 NULL, /* Reserved2 */
1443 schan_InitializeSecurityContextW,
1444 NULL, /* AcceptSecurityContext */
1445 NULL, /* CompleteAuthToken */
1446 schan_DeleteSecurityContext,
1447 NULL, /* ApplyControlToken */
1448 schan_QueryContextAttributesW,
1449 NULL, /* ImpersonateSecurityContext */
1450 NULL, /* RevertSecurityContext */
1451 NULL, /* MakeSignature */
1452 NULL, /* VerifySignature */
1453 FreeContextBuffer,
1454 NULL, /* QuerySecurityPackageInfoW */
1455 NULL, /* Reserved3 */
1456 NULL, /* Reserved4 */
1457 NULL, /* ExportSecurityContext */
1458 NULL, /* ImportSecurityContextW */
1459 NULL, /* AddCredentialsW */
1460 NULL, /* Reserved8 */
1461 NULL, /* QuerySecurityContextToken */
1462 schan_EncryptMessage,
1463 schan_DecryptMessage,
1464 NULL, /* SetContextAttributesW */
1465 };
1466
1467 void SECUR32_initSchannelSP(void)
1468 {
1469 /* This is what Windows reports. This shouldn't break any applications
1470 * even though the functions are missing, because the wrapper will
1471 * return SEC_E_UNSUPPORTED_FUNCTION if our function is NULL.
1472 */
1473 static const LONG caps =
1474 SECPKG_FLAG_INTEGRITY |
1475 SECPKG_FLAG_PRIVACY |
1476 SECPKG_FLAG_CONNECTION |
1477 SECPKG_FLAG_MULTI_REQUIRED |
1478 SECPKG_FLAG_EXTENDED_ERROR |
1479 SECPKG_FLAG_IMPERSONATION |
1480 SECPKG_FLAG_ACCEPT_WIN32_NAME |
1481 SECPKG_FLAG_STREAM;
1482 static const short version = 1;
1483 static const LONG maxToken = 16384;
1484 SEC_WCHAR *uniSPName = (SEC_WCHAR *)UNISP_NAME_W,
1485 *schannel = (SEC_WCHAR *)SCHANNEL_NAME_W;
1486 const SecPkgInfoW info[] = {
1487 { caps, version, UNISP_RPC_ID, maxToken, uniSPName, uniSPName },
1488 { caps, version, UNISP_RPC_ID, maxToken, schannel, (SEC_WCHAR *)L"Schannel Security Package" },
1489 };
1490 SecureProvider *provider;
1491
1492 if (!gnutls_handle)
1493 {
1494 if (NtQueryVirtualMemory( GetCurrentProcess(), hsecur32, MemoryWineUnixFuncs,
1495 &gnutls_handle, sizeof(gnutls_handle), NULL ) ||
1496 GNUTLS_CALL( process_attach, NULL ))
1497 {
1498 ERR( "no schannel support, expect problems\n" );
1499 return;
1500 }
1501 }
1502
1503 schan_handle_table = malloc(64 * sizeof(*schan_handle_table));
1504 if (!schan_handle_table)
1505 {
1506 ERR("Failed to allocate schannel handle table.\n");
1507 goto fail;
1508 }
1509 schan_handle_table_size = 64;
1510
1511 provider = SECUR32_addProvider(&schanTableA, &schanTableW, L"schannel.dll");
1512 if (!provider)
1513 {
1514 ERR("Failed to add schannel provider.\n");
1515 goto fail;
1516 }
1517
1518 SECUR32_addPackages(provider, ARRAY_SIZE(info), NULL, info);
1519 return;
1520
1521 fail:
1522 free(schan_handle_table);
1523 schan_handle_table = NULL;
1524 return;
1525 }
1526
1527 void SECUR32_deinitSchannelSP(void)
1528 {
1529 SIZE_T i = schan_handle_count;
1530
1531 if (!schan_handle_table) return;
1532
1533 /* deinitialized sessions first because a pointer to the credentials
1534 * may be stored for the session. */
1535 while (i--)
1536 {
1537 if (schan_handle_table[i].type == SCHAN_HANDLE_CTX)
1538 {
1539 struct schan_context *ctx = schan_free_handle(i, SCHAN_HANDLE_CTX);
1540 struct session_params params = { ctx->transport.session };
1541 GNUTLS_CALL( dispose_session, &params );
1542 free(ctx);
1543 }
1544 }
1545 i = schan_handle_count;
1546 while (i--)
1547 {
1548 if (schan_handle_table[i].type != SCHAN_HANDLE_FREE)
1549 {
1550 struct schan_credentials *cred = schan_free_handle(i, SCHAN_HANDLE_CRED);
1551 struct free_certificate_credentials_params params = { cred };
1552 GNUTLS_CALL( free_certificate_credentials, &params );
1553 free(cred);
1554 }
1555 }
1556 free(schan_handle_table);
1557 GNUTLS_CALL( process_detach, NULL );
1558 gnutls_handle = 0;
1559 }
Windows API implementation