/* Copyright (C) 2005 Juan Lang * Copyright 2008 Henri Verbeet * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA * * This file implements the schannel provider, or, the SSL/TLS implementations. */ #include #include #include #include #define NONAMELESSUNION #include "windef.h" #include "winbase.h" #include "winternl.h" #include "winreg.h" #include "winnls.h" #include "lmcons.h" #include "sspi.h" #define SCHANNEL_USE_BLACKLISTS #include "schannel.h" #include "wine/unixlib.h" #include "wine/debug.h" #include "secur32_priv.h" WINE_DEFAULT_DEBUG_CHANNEL(secur32); #define GNUTLS_CALL( func, params ) WINE_UNIX_CALL( unix_ ## func, params ) #define SCHAN_INVALID_HANDLE ~0UL enum schan_handle_type { SCHAN_HANDLE_CRED, SCHAN_HANDLE_CTX, SCHAN_HANDLE_FREE }; struct schan_handle { void *object; enum schan_handle_type type; }; struct schan_context { schan_session session; ULONG req_ctx_attr; const CERT_CONTEXT *cert; SIZE_T header_size; enum control_token control_token; unsigned int alert_type; unsigned int alert_number; }; static struct schan_handle *schan_handle_table; static struct schan_handle *schan_free_handles; static SIZE_T schan_handle_table_size; static SIZE_T schan_handle_count; /* Protocols enabled, only those may be used for the connection. */ static DWORD config_enabled_protocols; /* Protocols disabled by default. They are enabled for using, but disabled when caller asks for default settings. */ static DWORD config_default_disabled_protocols; static ULONG_PTR schan_alloc_handle(void *object, enum schan_handle_type type) { struct schan_handle *handle; if (schan_free_handles) { DWORD index = schan_free_handles - schan_handle_table; /* Use a free handle */ handle = schan_free_handles; if (handle->type != SCHAN_HANDLE_FREE) { ERR("Handle %ld(%p) is in the free list, but has type %#x.\n", index, handle, handle->type); return SCHAN_INVALID_HANDLE; } schan_free_handles = handle->object; handle->object = object; handle->type = type; return index; } if (!(schan_handle_count < schan_handle_table_size)) { /* Grow the table */ SIZE_T new_size = schan_handle_table_size + (schan_handle_table_size >> 1); struct schan_handle *new_table = realloc(schan_handle_table, new_size * sizeof(*schan_handle_table)); if (!new_table) { ERR("Failed to grow the handle table\n"); return SCHAN_INVALID_HANDLE; } schan_handle_table = new_table; schan_handle_table_size = new_size; } handle = &schan_handle_table[schan_handle_count++]; handle->object = object; handle->type = type; return handle - schan_handle_table; } static void *schan_free_handle(ULONG_PTR handle_idx, enum schan_handle_type type) { struct schan_handle *handle; void *object; if (handle_idx == SCHAN_INVALID_HANDLE) return NULL; if (handle_idx >= schan_handle_count) return NULL; handle = &schan_handle_table[handle_idx]; if (handle->type != type) { ERR("Handle %Id(%p) is not of type %#x\n", handle_idx, handle, type); return NULL; } object = handle->object; handle->object = schan_free_handles; handle->type = SCHAN_HANDLE_FREE; schan_free_handles = handle; return object; } static void *schan_get_object(ULONG_PTR handle_idx, enum schan_handle_type type) { struct schan_handle *handle; if (handle_idx == SCHAN_INVALID_HANDLE) return NULL; if (handle_idx >= schan_handle_count) return NULL; handle = &schan_handle_table[handle_idx]; if (handle->type != type) { ERR("Handle %Id(%p) is not of type %#x\n", handle_idx, handle, type); return NULL; } return handle->object; } static void read_config(void) { DWORD enabled = 0, default_disabled = 0; HKEY protocols_key, key; WCHAR subkey_name[64]; unsigned i, server; DWORD res; static BOOL config_read = FALSE; static const struct { WCHAR key_name[20]; DWORD prot_client_flag; DWORD prot_server_flag; BOOL enabled; /* If no config is present, enable the protocol */ BOOL disabled_by_default; /* Disable if caller asks for default protocol set */ } protocol_config_keys[] = { { L"SSL 2.0", SP_PROT_SSL2_CLIENT, SP_PROT_SSL2_SERVER, FALSE, TRUE }, /* NOTE: TRUE, TRUE on Windows */ { L"SSL 3.0", SP_PROT_SSL3_CLIENT, SP_PROT_SSL3_SERVER, TRUE, FALSE }, { L"TLS 1.0", SP_PROT_TLS1_0_CLIENT, SP_PROT_TLS1_0_SERVER, TRUE, FALSE }, { L"TLS 1.1", SP_PROT_TLS1_1_CLIENT, SP_PROT_TLS1_1_SERVER, TRUE, FALSE /* NOTE: not enabled by default on Windows */ }, { L"TLS 1.2", SP_PROT_TLS1_2_CLIENT, SP_PROT_TLS1_2_SERVER, TRUE, FALSE /* NOTE: not enabled by default on Windows */ }, { L"DTLS 1.0", SP_PROT_DTLS1_0_CLIENT, SP_PROT_DTLS1_0_SERVER, TRUE, TRUE }, { L"DTLS 1.2", SP_PROT_DTLS1_2_CLIENT, SP_PROT_DTLS1_2_SERVER, TRUE, TRUE }, }; /* No need for thread safety */ if(config_read) return; res = RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\SecurityProviders\\SCHANNEL\\Protocols", 0, KEY_READ, &protocols_key); if(res == ERROR_SUCCESS) { DWORD type, size, value; for(i = 0; i < ARRAY_SIZE(protocol_config_keys); i++) { for (server = 0; server < 2; server++) { DWORD flag = server ? protocol_config_keys[i].prot_server_flag : protocol_config_keys[i].prot_client_flag; wcscpy(subkey_name, protocol_config_keys[i].key_name); wcscat(subkey_name, server ? L"\\Server" : L"\\Client"); res = RegOpenKeyExW(protocols_key, subkey_name, 0, KEY_READ, &key); if(res != ERROR_SUCCESS) { if(protocol_config_keys[i].enabled) enabled |= flag; if(protocol_config_keys[i].disabled_by_default) default_disabled |= flag; continue; } size = sizeof(value); res = RegQueryValueExW(key, L"enabled", NULL, &type, (BYTE *)&value, &size); if(res == ERROR_SUCCESS) { if(type == REG_DWORD && value) enabled |= flag; }else if(protocol_config_keys[i].enabled) { enabled |= flag; } size = sizeof(value); res = RegQueryValueExW(key, L"DisabledByDefault", NULL, &type, (BYTE *)&value, &size); if(res == ERROR_SUCCESS) { if(type != REG_DWORD || value) default_disabled |= flag; }else if(protocol_config_keys[i].disabled_by_default) { default_disabled |= flag; } RegCloseKey(key); } } }else { /* No config, enable all known protocols. */ for(i = 0; i < ARRAY_SIZE(protocol_config_keys); i++) { DWORD flag = protocol_config_keys[i].prot_client_flag | protocol_config_keys[i].prot_server_flag; if(protocol_config_keys[i].enabled) enabled |= flag; if(protocol_config_keys[i].disabled_by_default) default_disabled |= flag; } } RegCloseKey(protocols_key); config_enabled_protocols = enabled & GNUTLS_CALL( get_enabled_protocols, NULL ); config_default_disabled_protocols = default_disabled; config_read = TRUE; TRACE("enabled %lx, disabled by default %lx\n", config_enabled_protocols, config_default_disabled_protocols); } static SECURITY_STATUS schan_QueryCredentialsAttributes( PCredHandle phCredential, ULONG ulAttribute, VOID *pBuffer) { struct schan_credentials *cred; SECURITY_STATUS ret; cred = schan_get_object(phCredential->dwLower, SCHAN_HANDLE_CRED); if(!cred) return SEC_E_INVALID_HANDLE; switch (ulAttribute) { case SECPKG_ATTR_SUPPORTED_ALGS: if (pBuffer) { /* FIXME: get from CryptoAPI */ FIXME("SECPKG_ATTR_SUPPORTED_ALGS: stub\n"); ret = SEC_E_UNSUPPORTED_FUNCTION; } else ret = SEC_E_INTERNAL_ERROR; break; case SECPKG_ATTR_CIPHER_STRENGTHS: if (pBuffer) { SecPkgCred_CipherStrengths *r = pBuffer; /* FIXME: get from CryptoAPI */ FIXME("SECPKG_ATTR_CIPHER_STRENGTHS: semi-stub\n"); r->dwMinimumCipherStrength = 40; r->dwMaximumCipherStrength = 168; ret = SEC_E_OK; } else ret = SEC_E_INTERNAL_ERROR; break; case SECPKG_ATTR_SUPPORTED_PROTOCOLS: if(pBuffer) { /* Regardless of MSDN documentation, tests show that this attribute takes into account * what protocols are enabled for given credential. */ ((SecPkgCred_SupportedProtocols*)pBuffer)->grbitProtocol = cred->enabled_protocols; ret = SEC_E_OK; }else { ret = SEC_E_INTERNAL_ERROR; } break; default: ret = SEC_E_UNSUPPORTED_FUNCTION; } return ret; } static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesA( PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer) { SECURITY_STATUS ret; TRACE("(%p, %ld, %p)\n", phCredential, ulAttribute, pBuffer); switch (ulAttribute) { case SECPKG_CRED_ATTR_NAMES: FIXME("SECPKG_CRED_ATTR_NAMES: stub\n"); ret = SEC_E_NO_CREDENTIALS; break; default: ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute, pBuffer); } return ret; } static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesW( PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer) { SECURITY_STATUS ret; TRACE("(%p, %ld, %p)\n", phCredential, ulAttribute, pBuffer); switch (ulAttribute) { case SECPKG_CRED_ATTR_NAMES: FIXME("SECPKG_CRED_ATTR_NAMES: stub\n"); ret = SEC_E_UNSUPPORTED_FUNCTION; break; default: ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute, pBuffer); } return ret; } static SECURITY_STATUS get_cert(const void *credentials, CERT_CONTEXT const **cert) { SECURITY_STATUS status; const SCHANNEL_CRED *cred_old; const SCH_CREDENTIALS *cred = credentials; PCCERT_CONTEXT *cert_list; DWORD i, cert_count; switch (cred->dwVersion) { case SCH_CRED_V3: case SCHANNEL_CRED_VERSION: cred_old = credentials; TRACE("dwVersion = %lu\n", cred_old->dwVersion); TRACE("cCreds = %lu\n", cred_old->cCreds); TRACE("paCred = %p\n", cred_old->paCred); TRACE("hRootStore = %p\n", cred_old->hRootStore); TRACE("cMappers = %lu\n", cred_old->cMappers); TRACE("cSupportedAlgs = %lu:\n", cred_old->cSupportedAlgs); for (i = 0; i < cred_old->cSupportedAlgs; i++) TRACE("%08x\n", cred_old->palgSupportedAlgs[i]); TRACE("grbitEnabledProtocols = %08lx\n", cred_old->grbitEnabledProtocols); TRACE("dwMinimumCipherStrength = %lu\n", cred_old->dwMinimumCipherStrength); TRACE("dwMaximumCipherStrength = %lu\n", cred_old->dwMaximumCipherStrength); TRACE("dwSessionLifespan = %lu\n", cred_old->dwSessionLifespan); TRACE("dwFlags = %08lx\n", cred_old->dwFlags); TRACE("dwCredFormat = %lu\n", cred_old->dwCredFormat); cert_list = cred_old->paCred; cert_count = cred_old->cCreds; break; case SCH_CREDENTIALS_VERSION: TRACE("dwVersion = %lu\n", cred->dwVersion); TRACE("dwCredFormat = %lu\n", cred->dwCredFormat); TRACE("cCreds = %lu\n", cred->cCreds); TRACE("paCred = %p\n", cred->paCred); TRACE("hRootStore = %p\n", cred->hRootStore); TRACE("cMappers = %lu\n", cred->cMappers); TRACE("dwSessionLifespan = %lu\n", cred->dwSessionLifespan); TRACE("dwFlags = %08lx\n", cred->dwFlags); TRACE("cTlsParameters = %lu:\n", cred->cTlsParameters); for (i = 0; i < cred->cTlsParameters; i++) { TRACE(" cAlpnIds %lu\n", cred->pTlsParameters[i].cAlpnIds); TRACE(" grbitDisabledProtocols %08lx\n", cred->pTlsParameters[i].grbitDisabledProtocols); TRACE(" cDisabledCrypto %lu\n", cred->pTlsParameters[i].cDisabledCrypto); TRACE(" dwFlags %08lx\n", cred->pTlsParameters[i].dwFlags); } cert_list = cred->paCred; cert_count = cred->cCreds; break; default: FIXME("unhandled version %lu\n", cred->dwVersion); return SEC_E_UNKNOWN_CREDENTIALS; } if (!cert_count) status = SEC_E_NO_CREDENTIALS; else if (cert_count > 1) status = SEC_E_UNKNOWN_CREDENTIALS; else { DWORD spec; HCRYPTPROV prov; BOOL free; if (CryptAcquireCertificatePrivateKey(cert_list[0], CRYPT_ACQUIRE_CACHE_FLAG, NULL, &prov, &spec, &free)) { if (free) CryptReleaseContext(prov, 0); *cert = cert_list[0]; status = SEC_E_OK; } else status = SEC_E_UNKNOWN_CREDENTIALS; } return status; } static DWORD get_enabled_protocols(const void *credentials) { const SCHANNEL_CRED *cred_old; const SCH_CREDENTIALS *cred = credentials; switch (cred->dwVersion) { case SCH_CRED_V3: case SCHANNEL_CRED_VERSION: cred_old = credentials; return cred_old->grbitEnabledProtocols; case SCH_CREDENTIALS_VERSION: if (cred->cTlsParameters) FIXME("handle TLS parameters\n"); return 0; default: FIXME("unhandled version %lu\n", cred->dwVersion); return 0; } } static WCHAR *get_key_container_path(const CERT_CONTEXT *ctx) { CERT_KEY_CONTEXT keyctx; DWORD size = sizeof(keyctx), prov_size = 0; CRYPT_KEY_PROV_INFO *prov; WCHAR username[UNLEN + 1], *ret = NULL; DWORD len = ARRAY_SIZE(username); if (CertGetCertificateContextProperty(ctx, CERT_KEY_CONTEXT_PROP_ID, &keyctx, &size)) { char *str; if (!CryptGetProvParam(keyctx.hCryptProv, PP_CONTAINER, NULL, &size, 0)) return NULL; if (!(str = malloc(size))) return NULL; if (!CryptGetProvParam(keyctx.hCryptProv, PP_CONTAINER, (BYTE *)str, &size, 0)) return NULL; len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0); if (!(ret = malloc(sizeof(L"Software\\Wine\\Crypto\\RSA\\") + len * sizeof(WCHAR)))) { free(str); return NULL; } wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\"); MultiByteToWideChar(CP_ACP, 0, str, -1, ret + wcslen(ret), len); free(str); } else if (CertGetCertificateContextProperty(ctx, CERT_KEY_PROV_INFO_PROP_ID, NULL, &prov_size)) { if (!(prov = malloc(prov_size))) return NULL; if (!CertGetCertificateContextProperty(ctx, CERT_KEY_PROV_INFO_PROP_ID, prov, &prov_size)) { free(prov); return NULL; } if (!(ret = malloc(sizeof(L"Software\\Wine\\Crypto\\RSA\\") + wcslen(prov->pwszContainerName) * sizeof(WCHAR)))) { free(prov); return NULL; } wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\"); wcscat(ret, prov->pwszContainerName); free(prov); } if (!ret && GetUserNameW(username, &len) && (ret = malloc(sizeof(L"Software\\Wine\\Crypto\\RSA\\") + len * sizeof(WCHAR)))) { wcscpy(ret, L"Software\\Wine\\Crypto\\RSA\\"); wcscat(ret, username); } return ret; } #define MAX_LEAD_BYTES 8 static BYTE *get_key_blob(const CERT_CONTEXT *ctx, DWORD *size) { BYTE *buf, *ret = NULL; DATA_BLOB blob_in, blob_out; DWORD spec = 0, type, len; LSTATUS retval; WCHAR *path; HKEY hkey; if (!(path = get_key_container_path(ctx))) return NULL; retval = RegOpenKeyExW(HKEY_CURRENT_USER, path, 0, KEY_READ, &hkey); free(path); if (retval) return NULL; if (!RegQueryValueExW(hkey, L"KeyExchangeKeyPair", 0, &type, NULL, &len)) spec = AT_KEYEXCHANGE; else if (!RegQueryValueExW(hkey, L"SignatureKeyPair", 0, &type, NULL, &len)) spec = AT_SIGNATURE; else { RegCloseKey(hkey); return NULL; } if (!(buf = malloc(len + MAX_LEAD_BYTES))) { RegCloseKey(hkey); return NULL; } if (!RegQueryValueExW(hkey, (spec == AT_KEYEXCHANGE) ? L"KeyExchangeKeyPair" : L"SignatureKeyPair", 0, &type, buf, &len)) { blob_in.pbData = buf; blob_in.cbData = len; if (CryptUnprotectData(&blob_in, NULL, NULL, NULL, NULL, 0, &blob_out)) { assert(blob_in.cbData >= blob_out.cbData); memcpy(buf, blob_out.pbData, blob_out.cbData); LocalFree(blob_out.pbData); *size = blob_out.cbData + MAX_LEAD_BYTES; ret = buf; } } else free(buf); RegCloseKey(hkey); return ret; } static SECURITY_STATUS acquire_credentials_handle(ULONG fCredentialUse, const SCHANNEL_CRED *schanCred, PCredHandle phCredential, PTimeStamp ptsExpiry) { struct schan_credentials *creds; DWORD enabled_protocols, cred_enabled_protocols; ULONG_PTR handle; SECURITY_STATUS status = SEC_E_OK; const CERT_CONTEXT *cert = NULL; struct allocate_certificate_credentials_params params = { 0 }; BYTE *key_blob = NULL; ULONG key_size = 0; TRACE("fCredentialUse %#lx, schanCred %p, phCredential %p, ptsExpiry %p\n", fCredentialUse, schanCred, phCredential, ptsExpiry); if (schanCred) { const unsigned dtls_protocols = SP_PROT_DTLS1_X; const unsigned non_dtls_protocols = (SP_PROT_X_CLIENTS | SP_PROT_X_SERVERS) & ~SP_PROT_DTLS1_X; status = get_cert(schanCred, &cert); if (status != SEC_E_OK && status != SEC_E_NO_CREDENTIALS) return status; cred_enabled_protocols = get_enabled_protocols(schanCred); if ((cred_enabled_protocols & non_dtls_protocols) && (cred_enabled_protocols & dtls_protocols)) return SEC_E_ALGORITHM_MISMATCH; status = SEC_E_OK; } else if (!fCredentialUse || (fCredentialUse & SECPKG_CRED_INBOUND)) { return SEC_E_NO_CREDENTIALS; } read_config(); if(schanCred && cred_enabled_protocols) enabled_protocols = cred_enabled_protocols & config_enabled_protocols; else enabled_protocols = config_enabled_protocols & ~config_default_disabled_protocols; if (!(fCredentialUse & SECPKG_CRED_OUTBOUND)) enabled_protocols &= ~SP_PROT_X_CLIENTS; if (!(fCredentialUse & SECPKG_CRED_INBOUND)) enabled_protocols &= ~SP_PROT_X_SERVERS; if(!enabled_protocols) { ERR("Could not find matching protocol\n"); return SEC_E_ALGORITHM_MISMATCH; } if (!(creds = malloc(sizeof(*creds)))) return SEC_E_INSUFFICIENT_MEMORY; creds->credential_use = fCredentialUse; creds->enabled_protocols = enabled_protocols; if (cert && !(key_blob = get_key_blob(cert, &key_size))) goto fail; params.c = creds; if (cert) { params.cert_encoding = cert->dwCertEncodingType; params.cert_size = cert->cbCertEncoded; params.cert_blob = cert->pbCertEncoded; } params.key_size = key_size; params.key_blob = key_blob; status = GNUTLS_CALL( allocate_certificate_credentials, ¶ms ); free(key_blob); if (status) goto fail; handle = schan_alloc_handle(creds, SCHAN_HANDLE_CRED); if (handle == SCHAN_INVALID_HANDLE) goto fail; phCredential->dwLower = handle; phCredential->dwUpper = 0; if (ptsExpiry) { if (fCredentialUse & SECPKG_CRED_INBOUND) { /* FIXME: get expiry from cert */ } else { /* Outbound credentials have no expiry */ ptsExpiry->LowPart = 0; ptsExpiry->HighPart = 0; } } return status; fail: free(creds); return SEC_E_INTERNAL_ERROR; } static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleA( SEC_CHAR *pszPrincipal, SEC_CHAR *pszPackage, ULONG fCredentialUse, PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn, PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry) { TRACE("(%s, %s, 0x%08lx, %p, %p, %p, %p, %p, %p)\n", debugstr_a(pszPrincipal), debugstr_a(pszPackage), fCredentialUse, pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry); return acquire_credentials_handle(fCredentialUse, pAuthData, phCredential, ptsExpiry); } static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleW( SEC_WCHAR *pszPrincipal, SEC_WCHAR *pszPackage, ULONG fCredentialUse, PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn, PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry) { TRACE("(%s, %s, 0x%08lx, %p, %p, %p, %p, %p, %p)\n", debugstr_w(pszPrincipal), debugstr_w(pszPackage), fCredentialUse, pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry); return acquire_credentials_handle(fCredentialUse, pAuthData, phCredential, ptsExpiry); } static SECURITY_STATUS SEC_ENTRY schan_FreeCredentialsHandle( PCredHandle phCredential) { struct schan_credentials *creds; TRACE("phCredential %p\n", phCredential); if (!phCredential) return SEC_E_INVALID_HANDLE; creds = schan_free_handle(phCredential->dwLower, SCHAN_HANDLE_CRED); if (!creds) return SEC_E_INVALID_HANDLE; if (creds->credential_use == SECPKG_CRED_OUTBOUND) { struct free_certificate_credentials_params params = { creds }; GNUTLS_CALL( free_certificate_credentials, ¶ms ); } free(creds); return SEC_E_OK; } static int schan_find_sec_buffer_idx(const SecBufferDesc *desc, unsigned int start_idx, ULONG buffer_type) { unsigned int i; PSecBuffer buffer; for (i = start_idx; i < desc->cBuffers; ++i) { buffer = &desc->pBuffers[i]; if ((buffer->BufferType | SECBUFFER_ATTRMASK) == (buffer_type | SECBUFFER_ATTRMASK)) return i; } return -1; } static void dump_buffer_desc(SecBufferDesc *desc) { unsigned int i; if (!desc) return; TRACE("Buffer desc %p:\n", desc); for (i = 0; i < desc->cBuffers; ++i) { SecBuffer *b = &desc->pBuffers[i]; TRACE("\tbuffer %u: cbBuffer %ld, BufferType %#lx pvBuffer %p\n", i, b->cbBuffer, b->BufferType, b->pvBuffer); } } #define HEADER_SIZE_TLS 5 #define HEADER_SIZE_DTLS 13 static inline SIZE_T read_record_size(const BYTE *buf, SIZE_T header_size) { return (buf[header_size - 2] << 8) | buf[header_size - 1]; } static inline BOOL is_dtls_context(const struct schan_context *ctx) { return ctx->header_size == HEADER_SIZE_DTLS; } static void fill_missing_sec_buffer(SecBufferDesc *input, DWORD size) { int idx = schan_find_sec_buffer_idx(input, 0, SECBUFFER_EMPTY); if (idx == -1) WARN("no empty buffer\n"); else { SecBuffer *buffer = &input->pBuffers[idx]; buffer->BufferType = SECBUFFER_MISSING; buffer->cbBuffer = size; } } static BOOL validate_input_buffers(SecBufferDesc *desc) { int i; for (i = 0; i < desc->cBuffers; i++) { SecBuffer *buffer = &desc->pBuffers[i]; if (buffer->BufferType == SECBUFFER_EMPTY && buffer->cbBuffer) return FALSE; } return TRUE; } static SECURITY_STATUS establish_context( PCredHandle phCredential, PCtxtHandle phContext, SEC_WCHAR *pszTargetName, PSecBufferDesc pInput, ULONG fContextReq, ULONG TargetDataRep, PCtxtHandle phNewContext, PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsTimeStamp, BOOL bIsServer) { const ULONG extra_size = 0x10000; struct schan_context *ctx; struct schan_credentials *cred; SIZE_T expected_size = 0; SECURITY_STATUS ret; SecBuffer *buffer; SecBuffer alloc_buffer = { 0 }; struct handshake_params params = { 0 }; int output_buffer_idx = -1; int idx, i; ULONG input_offset = 0, output_offset = 0; SecBufferDesc input_desc, output_desc; if (ptsTimeStamp) { ptsTimeStamp->LowPart = 0; ptsTimeStamp->HighPart = 0; } if (!pOutput || !pOutput->cBuffers) return SEC_E_INVALID_TOKEN; for (i = 0; i < pOutput->cBuffers; i++) { ULONG type = pOutput->pBuffers[i].BufferType; ULONG allocate_memory_flag = bIsServer ? ASC_REQ_ALLOCATE_MEMORY : ISC_REQ_ALLOCATE_MEMORY; if (type != SECBUFFER_TOKEN && type != SECBUFFER_ALERT) continue; if (!pOutput->pBuffers[i].cbBuffer && !(fContextReq & allocate_memory_flag)) return SEC_E_INSUFFICIENT_MEMORY; } if (!phContext) { ULONG_PTR handle; struct create_session_params create_params; ULONG credential_use = bIsServer ? SECPKG_CRED_INBOUND : SECPKG_CRED_OUTBOUND; if (!phCredential) return SEC_E_INVALID_HANDLE; cred = schan_get_object(phCredential->dwLower, SCHAN_HANDLE_CRED); if (!cred) return SEC_E_INVALID_HANDLE; if (!(cred->credential_use & credential_use)) { WARN("Invalid credential use %#lx, expected %#lx\n", cred->credential_use, credential_use); return SEC_E_INVALID_HANDLE; } if (!(ctx = calloc(1, sizeof(*ctx)))) return SEC_E_INSUFFICIENT_MEMORY; handle = schan_alloc_handle(ctx, SCHAN_HANDLE_CTX); if (handle == SCHAN_INVALID_HANDLE) { free(ctx); return SEC_E_INTERNAL_ERROR; } create_params.cred = cred; create_params.session = &ctx->session; if (GNUTLS_CALL( create_session, &create_params )) { schan_free_handle(handle, SCHAN_HANDLE_CTX); free(ctx); return SEC_E_INTERNAL_ERROR; } if (cred->enabled_protocols & SP_PROT_DTLS1_X) ctx->header_size = HEADER_SIZE_DTLS; else ctx->header_size = HEADER_SIZE_TLS; if (pszTargetName && *pszTargetName) { UINT len = WideCharToMultiByte( CP_UNIXCP, 0, pszTargetName, -1, NULL, 0, NULL, NULL ); char *target = malloc( len ); if (target) { struct set_session_target_params params = { ctx->session, target }; WideCharToMultiByte( CP_UNIXCP, 0, pszTargetName, -1, target, len, NULL, NULL ); GNUTLS_CALL( set_session_target, ¶ms ); free( target ); } } if (pInput && (idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_APPLICATION_PROTOCOLS)) != -1) { struct set_application_protocols_params params = { ctx->session, pInput->pBuffers[idx].pvBuffer, pInput->pBuffers[idx].cbBuffer }; GNUTLS_CALL( set_application_protocols, ¶ms ); } if (pInput && (idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_DTLS_MTU)) != -1) { buffer = &pInput->pBuffers[idx]; if (buffer->cbBuffer >= sizeof(WORD)) { struct set_dtls_mtu_params params = { ctx->session, *(WORD *)buffer->pvBuffer }; GNUTLS_CALL( set_dtls_mtu, ¶ms ); } else WARN("invalid buffer size %lu\n", buffer->cbBuffer); } if (is_dtls_context(ctx)) { struct set_dtls_timeouts_params params = { ctx->session, 0, 60000 }; GNUTLS_CALL( set_dtls_timeouts, ¶ms ); } phNewContext->dwLower = handle; phNewContext->dwUpper = 0; } if (bIsServer || phContext) { SIZE_T record_size = 0; unsigned char *ptr; if (phContext && !(ctx = schan_get_object(phContext->dwLower, SCHAN_HANDLE_CTX))) return SEC_E_INVALID_HANDLE; if (!pInput && !ctx->control_token && !is_dtls_context(ctx)) return SEC_E_INCOMPLETE_MESSAGE; if (!ctx->control_token && pInput) { if (!validate_input_buffers(pInput)) return SEC_E_INVALID_TOKEN; if ((idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_TOKEN)) == -1) return SEC_E_INCOMPLETE_MESSAGE; buffer = &pInput->pBuffers[idx]; ptr = buffer->pvBuffer; if (buffer->cbBuffer < ctx->header_size) { TRACE("Expected at least %Iu bytes, but buffer only contains %lu bytes.\n", ctx->header_size, buffer->cbBuffer); fill_missing_sec_buffer(pInput, ctx->header_size - buffer->cbBuffer); return SEC_E_INCOMPLETE_MESSAGE; } while (buffer->cbBuffer >= expected_size + ctx->header_size) { record_size = ctx->header_size + read_record_size(ptr, ctx->header_size); if (buffer->cbBuffer < expected_size + record_size) break; expected_size += record_size; ptr += record_size; } if (!expected_size) { TRACE("Expected at least %Iu bytes, but buffer only contains %lu bytes.\n", max(ctx->header_size, record_size), buffer->cbBuffer); fill_missing_sec_buffer(pInput, record_size - buffer->cbBuffer); return SEC_E_INCOMPLETE_MESSAGE; } TRACE("Using expected_size %Iu.\n", expected_size); } if (phNewContext && phContext) *phNewContext = *phContext; } ctx->req_ctx_attr = fContextReq; /* Perform the TLS handshake */ memset(&input_desc, 0, sizeof(input_desc)); if (pInput && (idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_TOKEN)) != -1) { input_desc.cBuffers = 1; input_desc.pBuffers = &pInput->pBuffers[idx]; } memset(&output_desc, 0, sizeof(output_desc)); idx = schan_find_sec_buffer_idx(pOutput, 0, SECBUFFER_TOKEN); if (idx == -1) idx = schan_find_sec_buffer_idx(pOutput, 0, SECBUFFER_EMPTY); if (idx != -1) { output_desc.cBuffers = 1; output_desc.pBuffers = &pOutput->pBuffers[idx]; if (!output_desc.pBuffers->pvBuffer || (fContextReq & ISC_REQ_ALLOCATE_MEMORY)) { alloc_buffer.cbBuffer = extra_size; alloc_buffer.BufferType = SECBUFFER_TOKEN; alloc_buffer.pvBuffer = RtlAllocateHeap( GetProcessHeap(), 0, extra_size ); output_desc.pBuffers = &alloc_buffer; } } params.session = ctx->session; params.input = pInput ? &input_desc : NULL; params.input_size = expected_size; params.output = &output_desc; params.input_offset = &input_offset; params.output_buffer_idx = &output_buffer_idx; params.output_offset = &output_offset; params.control_token = ctx->control_token; params.alert_type = ctx->alert_type; params.alert_number = ctx->alert_number; ctx->control_token = CONTROL_TOKEN_NONE; ret = GNUTLS_CALL( handshake, ¶ms ); if (output_buffer_idx != -1) { SecBuffer *buffer = &pOutput->pBuffers[idx]; buffer->BufferType = SECBUFFER_TOKEN; buffer->cbBuffer = output_offset; if (output_desc.pBuffers == &alloc_buffer) { RtlReAllocateHeap( GetProcessHeap(), HEAP_REALLOC_IN_PLACE_ONLY, alloc_buffer.pvBuffer, buffer->cbBuffer ); buffer->pvBuffer = alloc_buffer.pvBuffer; alloc_buffer.pvBuffer = NULL; } } else { pOutput->pBuffers[0].cbBuffer = 0; } RtlFreeHeap( GetProcessHeap(), 0, alloc_buffer.pvBuffer ); if (input_offset && input_offset != pInput->pBuffers[0].cbBuffer) { if(pInput->cBuffers<2 || pInput->pBuffers[1].BufferType!=SECBUFFER_EMPTY) return SEC_E_INVALID_TOKEN; pInput->pBuffers[1].BufferType = SECBUFFER_EXTRA; pInput->pBuffers[1].cbBuffer = pInput->pBuffers[0].cbBuffer - input_offset; } for (i = 0; i < pOutput->cBuffers; i++) { SecBuffer *buffer = &pOutput->pBuffers[i]; if (buffer->BufferType == SECBUFFER_ALERT) buffer->cbBuffer = 0; } if (bIsServer) { *pfContextAttr = ASC_RET_REPLAY_DETECT | ASC_RET_SEQUENCE_DETECT | ASC_RET_CONFIDENTIALITY | ASC_RET_STREAM; if (ctx->req_ctx_attr & ASC_REQ_EXTENDED_ERROR) *pfContextAttr |= ASC_RET_EXTENDED_ERROR; if (ctx->req_ctx_attr & ASC_REQ_DATAGRAM) *pfContextAttr |= ASC_RET_DATAGRAM; if (ctx->req_ctx_attr & ASC_REQ_ALLOCATE_MEMORY) *pfContextAttr |= ASC_RET_ALLOCATED_MEMORY; } else { *pfContextAttr = ISC_RET_REPLAY_DETECT | ISC_RET_SEQUENCE_DETECT | ISC_RET_CONFIDENTIALITY | ISC_RET_STREAM; if (ctx->req_ctx_attr & ISC_REQ_EXTENDED_ERROR) *pfContextAttr |= ISC_RET_EXTENDED_ERROR; if (ctx->req_ctx_attr & ISC_REQ_DATAGRAM) *pfContextAttr |= ISC_RET_DATAGRAM; if (ctx->req_ctx_attr & ISC_REQ_ALLOCATE_MEMORY) *pfContextAttr |= ISC_RET_ALLOCATED_MEMORY; if (ctx->req_ctx_attr & ISC_REQ_USE_SUPPLIED_CREDS) *pfContextAttr |= ISC_RET_USED_SUPPLIED_CREDS; if (ctx->req_ctx_attr & ISC_REQ_MANUAL_CRED_VALIDATION) *pfContextAttr |= ISC_RET_MANUAL_CRED_VALIDATION; } return ret; } /*********************************************************************** * InitializeSecurityContextW */ static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextW( PCredHandle phCredential, PCtxtHandle phContext, SEC_WCHAR *pszTargetName, ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep, PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext, PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry) { TRACE("%p %p %s 0x%08lx %ld %ld %p %ld %p %p %p %p\n", phCredential, phContext, debugstr_w(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput, Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry); dump_buffer_desc(pInput); dump_buffer_desc(pOutput); return establish_context(phCredential, phContext, pszTargetName, pInput, fContextReq, TargetDataRep, phNewContext, pOutput, pfContextAttr, ptsExpiry, FALSE); } /*********************************************************************** * InitializeSecurityContextA */ static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextA( PCredHandle phCredential, PCtxtHandle phContext, SEC_CHAR *pszTargetName, ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep, PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext, PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry) { SECURITY_STATUS ret; SEC_WCHAR *target_name = NULL; TRACE("%p %p %s 0x%08lx %ld %ld %p %ld %p %p %p %p\n", phCredential, phContext, debugstr_a(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput, Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry); if (pszTargetName) { INT len = MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, NULL, 0); if (!(target_name = malloc(len * sizeof(*target_name)))) return SEC_E_INSUFFICIENT_MEMORY; MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, target_name, len); } ret = schan_InitializeSecurityContextW(phCredential, phContext, target_name, fContextReq, Reserved1, TargetDataRep, pInput, Reserved2, phNewContext, pOutput, pfContextAttr, ptsExpiry); free(target_name); return ret; } /*********************************************************************** * AcceptSecurityContext */ static SECURITY_STATUS SEC_ENTRY schan_AcceptSecurityContext( PCredHandle phCredential, PCtxtHandle phContext, PSecBufferDesc pInput, ULONG fContextReq, ULONG TargetDataRep, PCtxtHandle phNewContext, PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsTimeStamp) { TRACE("%p %p %p 0x%08lx %ld %p %p %p %p\n", phCredential, phContext, pInput, fContextReq, TargetDataRep, phNewContext, pOutput, pfContextAttr, ptsTimeStamp); dump_buffer_desc(pInput); dump_buffer_desc(pOutput); return establish_context(phCredential, phContext, NULL, pInput, fContextReq, TargetDataRep, phNewContext, pOutput, pfContextAttr, ptsTimeStamp, TRUE); } static void *get_alg_name(ALG_ID id, BOOL wide) { static const struct { ALG_ID alg_id; const char* name; const WCHAR nameW[8]; } alg_name_map[] = { { CALG_ECDSA, "ECDSA", L"ECDSA" }, { CALG_RSA_SIGN, "RSA", L"RSA" }, { CALG_DES, "DES", L"DES" }, { CALG_RC2, "RC2", L"RC2" }, { CALG_3DES, "3DES", L"3DES" }, { CALG_AES_128, "AES", L"AES" }, { CALG_AES_192, "AES", L"AES" }, { CALG_AES_256, "AES", L"AES" }, { CALG_RC4, "RC4", L"RC4" }, }; unsigned i; for (i = 0; i < ARRAY_SIZE(alg_name_map); i++) if (alg_name_map[i].alg_id == id) return wide ? (void*)alg_name_map[i].nameW : (void*)alg_name_map[i].name; FIXME("Unknown ALG_ID %04x\n", id); return NULL; } static SECURITY_STATUS ensure_remote_cert(struct schan_context *ctx) { HCERTSTORE store; PCCERT_CONTEXT cert = NULL; SECURITY_STATUS status; ULONG count, size = 0; struct get_session_peer_certificate_params params = { ctx->session, NULL, &size, &count }; if (ctx->cert) return SEC_E_OK; if (!(store = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0, CERT_STORE_CREATE_NEW_FLAG, NULL))) return GetLastError(); status = GNUTLS_CALL( get_session_peer_certificate, ¶ms ); if (status != SEC_E_BUFFER_TOO_SMALL) goto done; if (!(params.buffer = malloc( size ))) { status = SEC_E_INSUFFICIENT_MEMORY; goto done; } status = GNUTLS_CALL( get_session_peer_certificate, ¶ms ); if (status == SEC_E_OK) { unsigned int i; ULONG *sizes; BYTE *blob; sizes = (ULONG *)params.buffer; blob = params.buffer + count * sizeof(*sizes); for (i = 0; i < count; i++) { if (!CertAddEncodedCertificateToStore(store, X509_ASN_ENCODING, blob, sizes[i], CERT_STORE_ADD_REPLACE_EXISTING, i ? NULL : &cert)) { if (i) CertFreeCertificateContext(cert); return GetLastError(); } blob += sizes[i]; } } free(params.buffer); done: ctx->cert = cert; CertCloseStore(store, 0); return status; } static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesW( PCtxtHandle context_handle, ULONG attribute, PVOID buffer) { struct schan_context *ctx; SECURITY_STATUS status; TRACE("context_handle %p, attribute %#lx, buffer %p\n", context_handle, attribute, buffer); if (!context_handle || !(ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX))) return SEC_E_INVALID_HANDLE; switch (attribute) { case SECPKG_ATTR_STREAM_SIZES: { SecPkgContext_ConnectionInfo info; struct get_connection_info_params params = { ctx->session, &info }; status = GNUTLS_CALL( get_connection_info, ¶ms ); if (status == SEC_E_OK) { struct session_params params = { ctx->session }; SecPkgContext_StreamSizes *stream_sizes = buffer; SIZE_T mac_size = info.dwHashStrength; unsigned int block_size = GNUTLS_CALL( get_session_cipher_block_size, ¶ms ); unsigned int message_size = GNUTLS_CALL( get_max_message_size, ¶ms ); TRACE("Using header size %Iu mac bytes %Iu, message size %u, block size %u\n", ctx->header_size, mac_size, message_size, block_size); /* These are defined by the TLS RFC */ stream_sizes->cbHeader = ctx->header_size; stream_sizes->cbTrailer = mac_size + 256; /* Max 255 bytes padding + 1 for padding size */ stream_sizes->cbMaximumMessage = message_size; stream_sizes->cBuffers = 4; stream_sizes->cbBlockSize = block_size; } return status; } case SECPKG_ATTR_KEY_INFO: { SecPkgContext_ConnectionInfo conn_info; struct get_connection_info_params params = { ctx->session, &conn_info }; status = GNUTLS_CALL( get_connection_info, ¶ms ); if (status == SEC_E_OK) { struct session_params params = { ctx->session }; SecPkgContext_KeyInfoW *info = buffer; info->KeySize = conn_info.dwCipherStrength; info->SignatureAlgorithm = GNUTLS_CALL( get_key_signature_algorithm, ¶ms ); info->EncryptAlgorithm = conn_info.aiCipher; info->sSignatureAlgorithmName = get_alg_name(info->SignatureAlgorithm, TRUE); info->sEncryptAlgorithmName = get_alg_name(info->EncryptAlgorithm, TRUE); } return status; } case SECPKG_ATTR_REMOTE_CERT_CONTEXT: { PCCERT_CONTEXT *cert = buffer; if ((status = ensure_remote_cert(ctx)) != SEC_E_OK) return status; *cert = CertDuplicateCertificateContext(ctx->cert); return SEC_E_OK; } case SECPKG_ATTR_CONNECTION_INFO: { SecPkgContext_ConnectionInfo *info = buffer; struct get_connection_info_params params = { ctx->session, info }; return GNUTLS_CALL( get_connection_info, ¶ms ); } case SECPKG_ATTR_ENDPOINT_BINDINGS: { static const char prefix[] = "tls-server-end-point:"; SecPkgContext_Bindings *bindings = buffer; CCRYPT_OID_INFO *info; ALG_ID hash_alg = CALG_SHA_256; BYTE hash[1024]; DWORD hash_size; char *p; BOOL ret; if ((status = ensure_remote_cert(ctx)) != SEC_E_OK) return status; /* RFC 5929 */ info = CryptFindOIDInfo(CRYPT_OID_INFO_OID_KEY, ctx->cert->pCertInfo->SignatureAlgorithm.pszObjId, 0); if (info && info->u.Algid != CALG_SHA1 && info->u.Algid != CALG_MD5) hash_alg = info->u.Algid; hash_size = sizeof(hash); ret = CryptHashCertificate(0, hash_alg, 0, ctx->cert->pbCertEncoded, ctx->cert->cbCertEncoded, hash, &hash_size); if (!ret) return GetLastError(); bindings->BindingsLength = sizeof(*bindings->Bindings) + sizeof(prefix) - 1 + hash_size; /* freed with FreeContextBuffer */ bindings->Bindings = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, bindings->BindingsLength); if (!bindings->Bindings) return SEC_E_INSUFFICIENT_MEMORY; bindings->Bindings->cbApplicationDataLength = sizeof(prefix) - 1 + hash_size; bindings->Bindings->dwApplicationDataOffset = sizeof(*bindings->Bindings); p = (char *)(bindings->Bindings + 1); memcpy(p, prefix, sizeof(prefix) - 1); p += sizeof(prefix) - 1; memcpy(p, hash, hash_size); return SEC_E_OK; } case SECPKG_ATTR_UNIQUE_BINDINGS: { static const char prefix[] = "tls-unique:"; SecPkgContext_Bindings *bindings = buffer; ULONG size; char *p; struct get_unique_channel_binding_params params = { ctx->session, NULL, &size }; if (GNUTLS_CALL( get_unique_channel_binding, ¶ms ) != SEC_E_BUFFER_TOO_SMALL) return SEC_E_INTERNAL_ERROR; bindings->BindingsLength = sizeof(*bindings->Bindings) + sizeof(prefix) - 1 + size; /* freed with FreeContextBuffer */ bindings->Bindings = RtlAllocateHeap(GetProcessHeap(), HEAP_ZERO_MEMORY, bindings->BindingsLength); if (!bindings->Bindings) return SEC_E_INSUFFICIENT_MEMORY; bindings->Bindings->cbApplicationDataLength = sizeof(prefix) - 1 + size; bindings->Bindings->dwApplicationDataOffset = sizeof(*bindings->Bindings); p = (char *)(bindings->Bindings + 1); memcpy(p, prefix, sizeof(prefix) - 1); p += sizeof(prefix) - 1; params.buffer = p; return GNUTLS_CALL( get_unique_channel_binding, ¶ms ); } case SECPKG_ATTR_APPLICATION_PROTOCOL: { SecPkgContext_ApplicationProtocol *protocol = buffer; struct get_application_protocol_params params = { ctx->session, protocol }; return GNUTLS_CALL( get_application_protocol, ¶ms ); } case SECPKG_ATTR_CIPHER_INFO: { SecPkgContext_CipherInfo *info = buffer; struct get_cipher_info_params params = { ctx->session, info }; return GNUTLS_CALL( get_cipher_info, ¶ms ); } default: FIXME("Unhandled attribute %#lx\n", attribute); return SEC_E_UNSUPPORTED_FUNCTION; } } static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesA( PCtxtHandle context_handle, ULONG attribute, PVOID buffer) { TRACE("context_handle %p, attribute %#lx, buffer %p\n", context_handle, attribute, buffer); switch(attribute) { case SECPKG_ATTR_KEY_INFO: { SECURITY_STATUS status = schan_QueryContextAttributesW(context_handle, attribute, buffer); if (status == SEC_E_OK) { SecPkgContext_KeyInfoA *info = buffer; info->sSignatureAlgorithmName = get_alg_name(info->SignatureAlgorithm, FALSE); info->sEncryptAlgorithmName = get_alg_name(info->EncryptAlgorithm, FALSE); } return status; } case SECPKG_ATTR_STREAM_SIZES: case SECPKG_ATTR_REMOTE_CERT_CONTEXT: case SECPKG_ATTR_CONNECTION_INFO: case SECPKG_ATTR_ENDPOINT_BINDINGS: case SECPKG_ATTR_UNIQUE_BINDINGS: case SECPKG_ATTR_APPLICATION_PROTOCOL: case SECPKG_ATTR_CIPHER_INFO: return schan_QueryContextAttributesW(context_handle, attribute, buffer); default: FIXME("Unhandled attribute %#lx\n", attribute); return SEC_E_UNSUPPORTED_FUNCTION; } } static SECURITY_STATUS SEC_ENTRY schan_EncryptMessage(PCtxtHandle context_handle, ULONG quality, PSecBufferDesc message, ULONG message_seq_no) { struct schan_context *ctx; struct send_params params; SECURITY_STATUS status; SecBuffer *buffer; SIZE_T data_size; char *data; int output_buffer_idx = -1; ULONG output_offset = 0; SecBufferDesc output_desc = { 0 }; SecBuffer output_buffers[3]; int header_idx, data_idx, trailer_idx = -1; int buffer_index[3]; TRACE("context_handle %p, quality %ld, message %p, message_seq_no %ld\n", context_handle, quality, message, message_seq_no); if (!context_handle) return SEC_E_INVALID_HANDLE; ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX); dump_buffer_desc(message); data_idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_DATA); if (data_idx == -1) { WARN("No data buffer passed\n"); return SEC_E_INTERNAL_ERROR; } buffer = &message->pBuffers[data_idx]; data_size = buffer->cbBuffer; data = malloc(data_size); memcpy(data, buffer->pvBuffer, data_size); /* Use { STREAM_HEADER, DATA, STREAM_TRAILER } or { TOKEN, DATA, TOKEN } buffers. */ output_desc.pBuffers = output_buffers; if ((header_idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_STREAM_HEADER)) == -1) { if ((header_idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_TOKEN)) != -1) { output_buffers[output_desc.cBuffers++] = message->pBuffers[header_idx]; output_buffers[output_desc.cBuffers++] = message->pBuffers[data_idx]; trailer_idx = schan_find_sec_buffer_idx(message, header_idx + 1, SECBUFFER_TOKEN); if (trailer_idx != -1) output_buffers[output_desc.cBuffers++] = message->pBuffers[trailer_idx]; } } else { output_buffers[output_desc.cBuffers++] = message->pBuffers[header_idx]; output_buffers[output_desc.cBuffers++] = message->pBuffers[data_idx]; trailer_idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_STREAM_TRAILER); if (trailer_idx != -1) output_buffers[output_desc.cBuffers++] = message->pBuffers[trailer_idx]; } buffer_index[0] = header_idx; buffer_index[1] = data_idx; buffer_index[2] = trailer_idx; params.session = ctx->session; params.output = &output_desc; params.buffer = data; params.length = data_size; params.output_buffer_idx = &output_buffer_idx; params.output_offset = &output_offset; status = GNUTLS_CALL( send, ¶ms ); if (!status) message->pBuffers[buffer_index[output_buffer_idx]].cbBuffer = output_offset; free(data); TRACE("Returning %#lx.\n", status); return status; } static int schan_validate_decrypt_buffer_desc(PSecBufferDesc message) { int data_idx = -1; unsigned int empty_count = 0; unsigned int i; if (message->cBuffers < 4) { WARN("Less than four buffers passed\n"); return -1; } for (i = 0; i < message->cBuffers; ++i) { SecBuffer *b = &message->pBuffers[i]; if (b->BufferType == SECBUFFER_DATA) { if (data_idx != -1) { WARN("More than one data buffer passed\n"); return -1; } data_idx = i; } else if (b->BufferType == SECBUFFER_EMPTY) ++empty_count; } if (data_idx == -1) { WARN("No data buffer passed\n"); return -1; } if (empty_count < 3) { WARN("Less than three empty buffers passed\n"); return -1; } return data_idx; } static void schan_decrypt_fill_buffer(PSecBufferDesc message, ULONG buffer_type, void *data, ULONG size) { int idx; SecBuffer *buffer; idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY); buffer = &message->pBuffers[idx]; buffer->BufferType = buffer_type; buffer->pvBuffer = data; buffer->cbBuffer = size; } static SECURITY_STATUS SEC_ENTRY schan_DecryptMessage(PCtxtHandle context_handle, PSecBufferDesc message, ULONG message_seq_no, PULONG quality) { SECURITY_STATUS status = SEC_E_OK; struct schan_context *ctx; struct recv_params params; SecBuffer *buffer; SIZE_T data_size; char *data; unsigned expected_size; ULONG received = 0; int idx; unsigned char *buf_ptr; SecBufferDesc input_desc = { 0 }; TRACE("context_handle %p, message %p, message_seq_no %ld, quality %p\n", context_handle, message, message_seq_no, quality); if (!context_handle) return SEC_E_INVALID_HANDLE; ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX); dump_buffer_desc(message); idx = schan_validate_decrypt_buffer_desc(message); if (idx == -1) return SEC_E_INVALID_TOKEN; buffer = &message->pBuffers[idx]; buf_ptr = buffer->pvBuffer; expected_size = ctx->header_size + read_record_size(buf_ptr, ctx->header_size); if(buffer->cbBuffer < expected_size) { TRACE("Expected %u bytes, but buffer only contains %lu bytes\n", expected_size, buffer->cbBuffer); buffer->BufferType = SECBUFFER_MISSING; buffer->cbBuffer = expected_size - buffer->cbBuffer; /* This is a bit weird, but windows does it too */ idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY); buffer = &message->pBuffers[idx]; buffer->BufferType = SECBUFFER_MISSING; buffer->cbBuffer = expected_size - buffer->cbBuffer; TRACE("Returning SEC_E_INCOMPLETE_MESSAGE\n"); return SEC_E_INCOMPLETE_MESSAGE; } data_size = expected_size - ctx->header_size; data = malloc(data_size); received = data_size; input_desc.cBuffers = 1; input_desc.pBuffers = &message->pBuffers[idx]; params.session = ctx->session; params.input = &input_desc; params.input_size = expected_size; params.buffer = data; params.length = &received; status = GNUTLS_CALL( recv, ¶ms ); if (status != SEC_E_OK && status != SEC_I_RENEGOTIATE) { free(data); ERR("Returning %lx\n", status); return status; } TRACE("Received %lu bytes\n", received); memcpy(buf_ptr + ctx->header_size, data, received); free(data); schan_decrypt_fill_buffer(message, SECBUFFER_DATA, buf_ptr + ctx->header_size, received); schan_decrypt_fill_buffer(message, SECBUFFER_STREAM_TRAILER, buf_ptr + ctx->header_size + received, buffer->cbBuffer - ctx->header_size - received); if(buffer->cbBuffer > expected_size) schan_decrypt_fill_buffer(message, SECBUFFER_EXTRA, buf_ptr + expected_size, buffer->cbBuffer - expected_size); buffer->BufferType = SECBUFFER_STREAM_HEADER; buffer->cbBuffer = ctx->header_size; return status; } static SECURITY_STATUS SEC_ENTRY schan_DeleteSecurityContext(PCtxtHandle context_handle) { struct schan_context *ctx; struct session_params params; TRACE("context_handle %p\n", context_handle); if (!context_handle) return SEC_E_INVALID_HANDLE; ctx = schan_free_handle(context_handle->dwLower, SCHAN_HANDLE_CTX); if (!ctx) return SEC_E_INVALID_HANDLE; if (ctx->cert) CertFreeCertificateContext(ctx->cert); params.session = ctx->session; GNUTLS_CALL( dispose_session, ¶ms ); free(ctx); return SEC_E_OK; } static SECURITY_STATUS SEC_ENTRY schan_ApplyControlToken(PCtxtHandle context_handle, PSecBufferDesc input) { struct schan_context *ctx; DWORD type; TRACE("%p %p\n", context_handle, input); dump_buffer_desc(input); if (!context_handle || !(ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX))) return SEC_E_INVALID_HANDLE; if (!input) return SEC_E_INTERNAL_ERROR; if (input->cBuffers != 1) return SEC_E_INVALID_TOKEN; if (input->pBuffers[0].BufferType != SECBUFFER_TOKEN) return SEC_E_INVALID_TOKEN; if (input->pBuffers[0].cbBuffer < sizeof(type)) return SEC_E_UNSUPPORTED_FUNCTION; type = *(DWORD *)input->pBuffers[0].pvBuffer; switch (type) { case SCHANNEL_SHUTDOWN: ctx->control_token = CONTROL_TOKEN_SHUTDOWN; ctx->alert_type = TLS1_ALERT_WARNING; ctx->alert_number = TLS1_ALERT_CLOSE_NOTIFY; return SEC_E_OK; case SCHANNEL_ALERT: { SCHANNEL_ALERT_TOKEN *alert = input->pBuffers[0].pvBuffer; if (input->pBuffers[0].cbBuffer < sizeof(*alert)) return SEC_E_INVALID_TOKEN; ctx->control_token = CONTROL_TOKEN_ALERT; ctx->alert_type = alert->dwAlertType; ctx->alert_number = alert->dwAlertNumber; return SEC_E_OK; } default: FIXME("token type %lu not supported\n", type); return SEC_E_UNSUPPORTED_FUNCTION; } } static const SecurityFunctionTableA schanTableA = { 1, NULL, /* EnumerateSecurityPackagesA */ schan_QueryCredentialsAttributesA, schan_AcquireCredentialsHandleA, schan_FreeCredentialsHandle, NULL, /* Reserved2 */ schan_InitializeSecurityContextA, schan_AcceptSecurityContext, NULL, /* CompleteAuthToken */ schan_DeleteSecurityContext, schan_ApplyControlToken, /* ApplyControlToken */ schan_QueryContextAttributesA, NULL, /* ImpersonateSecurityContext */ NULL, /* RevertSecurityContext */ NULL, /* MakeSignature */ NULL, /* VerifySignature */ FreeContextBuffer, NULL, /* QuerySecurityPackageInfoA */ NULL, /* Reserved3 */ NULL, /* Reserved4 */ NULL, /* ExportSecurityContext */ NULL, /* ImportSecurityContextA */ NULL, /* AddCredentialsA */ NULL, /* Reserved8 */ NULL, /* QuerySecurityContextToken */ schan_EncryptMessage, schan_DecryptMessage, NULL, /* SetContextAttributesA */ }; static const SecurityFunctionTableW schanTableW = { 1, NULL, /* EnumerateSecurityPackagesW */ schan_QueryCredentialsAttributesW, schan_AcquireCredentialsHandleW, schan_FreeCredentialsHandle, NULL, /* Reserved2 */ schan_InitializeSecurityContextW, schan_AcceptSecurityContext, NULL, /* CompleteAuthToken */ schan_DeleteSecurityContext, schan_ApplyControlToken, /* ApplyControlToken */ schan_QueryContextAttributesW, NULL, /* ImpersonateSecurityContext */ NULL, /* RevertSecurityContext */ NULL, /* MakeSignature */ NULL, /* VerifySignature */ FreeContextBuffer, NULL, /* QuerySecurityPackageInfoW */ NULL, /* Reserved3 */ NULL, /* Reserved4 */ NULL, /* ExportSecurityContext */ NULL, /* ImportSecurityContextW */ NULL, /* AddCredentialsW */ NULL, /* Reserved8 */ NULL, /* QuerySecurityContextToken */ schan_EncryptMessage, schan_DecryptMessage, NULL, /* SetContextAttributesW */ }; void SECUR32_initSchannelSP(void) { /* This is what Windows reports. This shouldn't break any applications * even though the functions are missing, because the wrapper will * return SEC_E_UNSUPPORTED_FUNCTION if our function is NULL. */ static const LONG caps = SECPKG_FLAG_INTEGRITY | SECPKG_FLAG_PRIVACY | SECPKG_FLAG_CONNECTION | SECPKG_FLAG_MULTI_REQUIRED | SECPKG_FLAG_EXTENDED_ERROR | SECPKG_FLAG_IMPERSONATION | SECPKG_FLAG_ACCEPT_WIN32_NAME | SECPKG_FLAG_STREAM; static const short version = 1; static const LONG maxToken = 16384; SEC_WCHAR *uniSPName = (SEC_WCHAR *)UNISP_NAME_W, *schannel = (SEC_WCHAR *)SCHANNEL_NAME_W; const SecPkgInfoW info[] = { { caps, version, UNISP_RPC_ID, maxToken, uniSPName, uniSPName }, { caps, version, UNISP_RPC_ID, maxToken, schannel, (SEC_WCHAR *)L"Schannel Security Package" }, }; SecureProvider *provider; if (__wine_init_unix_call() || GNUTLS_CALL( process_attach, NULL )) { ERR( "no schannel support, expect problems\n" ); return; } schan_handle_table = malloc(64 * sizeof(*schan_handle_table)); if (!schan_handle_table) { ERR("Failed to allocate schannel handle table.\n"); goto fail; } schan_handle_table_size = 64; provider = SECUR32_addProvider(&schanTableA, &schanTableW, L"schannel.dll"); if (!provider) { ERR("Failed to add schannel provider.\n"); goto fail; } SECUR32_addPackages(provider, ARRAY_SIZE(info), NULL, info); return; fail: free(schan_handle_table); schan_handle_table = NULL; return; } void SECUR32_deinitSchannelSP(void) { SIZE_T i = schan_handle_count; if (!schan_handle_table) return; /* deinitialized sessions first because a pointer to the credentials * may be stored for the session. */ while (i--) { if (schan_handle_table[i].type == SCHAN_HANDLE_CTX) { struct schan_context *ctx = schan_free_handle(i, SCHAN_HANDLE_CTX); struct session_params params = { ctx->session }; GNUTLS_CALL( dispose_session, ¶ms ); free(ctx); } } i = schan_handle_count; while (i--) { if (schan_handle_table[i].type != SCHAN_HANDLE_FREE) { struct schan_credentials *cred = schan_free_handle(i, SCHAN_HANDLE_CRED); struct free_certificate_credentials_params params = { cred }; GNUTLS_CALL( free_certificate_credentials, ¶ms ); free(cred); } } free(schan_handle_table); GNUTLS_CALL( process_detach, NULL ); }