dlls/secur32/schannel.c

   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 #include "config.h"
  21 #include "wine/port.h"
  22
  23 #include <stdarg.h>
  24 #include <errno.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winnls.h"
  29 #include "sspi.h"
  30 #include "schannel.h"
  31 #include "secur32_priv.h"
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(secur32);
  35
  36 #if defined(SONAME_LIBGNUTLS) || defined (HAVE_SECURITY_SECURITY_H)
  37
  38 #define SCHAN_INVALID_HANDLE ~0UL
  39
  40 enum schan_handle_type
  41 {
  42     SCHAN_HANDLE_CRED,
  43     SCHAN_HANDLE_CTX,
  44     SCHAN_HANDLE_FREE
  45 };
  46
  47 struct schan_handle
  48 {
  49     void *object;
  50     enum schan_handle_type type;
  51 };
  52
  53 struct schan_credentials
  54 {
  55     ULONG credential_use;
  56     schan_imp_certificate_credentials credentials;
  57 };
  58
  59 struct schan_context
  60 {
  61     schan_imp_session session;
  62     ULONG req_ctx_attr;
  63 };
  64
  65 static struct schan_handle *schan_handle_table;
  66 static struct schan_handle *schan_free_handles;
  67 static SIZE_T schan_handle_table_size;
  68 static SIZE_T schan_handle_count;
  69
  70 static ULONG_PTR schan_alloc_handle(void *object, enum schan_handle_type type)
  71 {
  72     struct schan_handle *handle;
  73
  74     if (schan_free_handles)
  75     {
  76         DWORD index = schan_free_handles - schan_handle_table;
  77         /* Use a free handle */
  78         handle = schan_free_handles;
  79         if (handle->type != SCHAN_HANDLE_FREE)
  80         {
  81             ERR("Handle %d(%p) is in the free list, but has type %#x.\n", index, handle, handle->type);
  82             return SCHAN_INVALID_HANDLE;
  83         }
  84         schan_free_handles = handle->object;
  85         handle->object = object;
  86         handle->type = type;
  87
  88         return index;
  89     }
  90     if (!(schan_handle_count < schan_handle_table_size))
  91     {
  92         /* Grow the table */
  93         SIZE_T new_size = schan_handle_table_size + (schan_handle_table_size >> 1);
  94         struct schan_handle *new_table = HeapReAlloc(GetProcessHeap(), 0, schan_handle_table, new_size * sizeof(*schan_handle_table));
  95         if (!new_table)
  96         {
  97             ERR("Failed to grow the handle table\n");
  98             return SCHAN_INVALID_HANDLE;
  99         }
 100         schan_handle_table = new_table;
 101         schan_handle_table_size = new_size;
 102     }
 103
 104     handle = &schan_handle_table[schan_handle_count++];
 105     handle->object = object;
 106     handle->type = type;
 107
 108     return handle - schan_handle_table;
 109 }
 110
 111 static void *schan_free_handle(ULONG_PTR handle_idx, enum schan_handle_type type)
 112 {
 113     struct schan_handle *handle;
 114     void *object;
 115
 116     if (handle_idx == SCHAN_INVALID_HANDLE) return NULL;
 117     if (handle_idx >= schan_handle_count) return NULL;
 118     handle = &schan_handle_table[handle_idx];
 119     if (handle->type != type)
 120     {
 121         ERR("Handle %ld(%p) is not of type %#x\n", handle_idx, handle, type);
 122         return NULL;
 123     }
 124
 125     object = handle->object;
 126     handle->object = schan_free_handles;
 127     handle->type = SCHAN_HANDLE_FREE;
 128     schan_free_handles = handle;
 129
 130     return object;
 131 }
 132
 133 static void *schan_get_object(ULONG_PTR handle_idx, enum schan_handle_type type)
 134 {
 135     struct schan_handle *handle;
 136
 137     if (handle_idx == SCHAN_INVALID_HANDLE) return NULL;
 138     if (handle_idx >= schan_handle_count) return NULL;
 139     handle = &schan_handle_table[handle_idx];
 140     if (handle->type != type)
 141     {
 142         ERR("Handle %ld(%p) is not of type %#x\n", handle_idx, handle, type);
 143         return NULL;
 144     }
 145
 146     return handle->object;
 147 }
 148
 149 static SECURITY_STATUS schan_QueryCredentialsAttributes(
 150  PCredHandle phCredential, ULONG ulAttribute, VOID *pBuffer)
 151 {
 152     SECURITY_STATUS ret;
 153
 154     switch (ulAttribute)
 155     {
 156     case SECPKG_ATTR_SUPPORTED_ALGS:
 157         if (pBuffer)
 158         {
 159             /* FIXME: get from CryptoAPI */
 160             FIXME("SECPKG_ATTR_SUPPORTED_ALGS: stub\n");
 161             ret = SEC_E_UNSUPPORTED_FUNCTION;
 162         }
 163         else
 164             ret = SEC_E_INTERNAL_ERROR;
 165         break;
 166     case SECPKG_ATTR_CIPHER_STRENGTHS:
 167         if (pBuffer)
 168         {
 169             SecPkgCred_CipherStrengths *r = pBuffer;
 170
 171             /* FIXME: get from CryptoAPI */
 172             FIXME("SECPKG_ATTR_CIPHER_STRENGTHS: semi-stub\n");
 173             r->dwMinimumCipherStrength = 40;
 174             r->dwMaximumCipherStrength = 168;
 175             ret = SEC_E_OK;
 176         }
 177         else
 178             ret = SEC_E_INTERNAL_ERROR;
 179         break;
 180     case SECPKG_ATTR_SUPPORTED_PROTOCOLS:
 181         if (pBuffer)
 182         {
 183             /* FIXME: get from OpenSSL? */
 184             FIXME("SECPKG_ATTR_SUPPORTED_PROTOCOLS: stub\n");
 185             ret = SEC_E_UNSUPPORTED_FUNCTION;
 186         }
 187         else
 188             ret = SEC_E_INTERNAL_ERROR;
 189         break;
 190     default:
 191         ret = SEC_E_UNSUPPORTED_FUNCTION;
 192     }
 193     return ret;
 194 }
 195
 196 static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesA(
 197  PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer)
 198 {
 199     SECURITY_STATUS ret;
 200
 201     TRACE("(%p, %d, %p)\n", phCredential, ulAttribute, pBuffer);
 202
 203     switch (ulAttribute)
 204     {
 205     case SECPKG_CRED_ATTR_NAMES:
 206         FIXME("SECPKG_CRED_ATTR_NAMES: stub\n");
 207         ret = SEC_E_UNSUPPORTED_FUNCTION;
 208         break;
 209     default:
 210         ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute,
 211          pBuffer);
 212     }
 213     return ret;
 214 }
 215
 216 static SECURITY_STATUS SEC_ENTRY schan_QueryCredentialsAttributesW(
 217  PCredHandle phCredential, ULONG ulAttribute, PVOID pBuffer)
 218 {
 219     SECURITY_STATUS ret;
 220
 221     TRACE("(%p, %d, %p)\n", phCredential, ulAttribute, pBuffer);
 222
 223     switch (ulAttribute)
 224     {
 225     case SECPKG_CRED_ATTR_NAMES:
 226         FIXME("SECPKG_CRED_ATTR_NAMES: stub\n");
 227         ret = SEC_E_UNSUPPORTED_FUNCTION;
 228         break;
 229     default:
 230         ret = schan_QueryCredentialsAttributes(phCredential, ulAttribute,
 231          pBuffer);
 232     }
 233     return ret;
 234 }
 235
 236 static SECURITY_STATUS schan_CheckCreds(const SCHANNEL_CRED *schanCred)
 237 {
 238     SECURITY_STATUS st;
 239     DWORD i;
 240
 241     TRACE("dwVersion = %d\n", schanCred->dwVersion);
 242     TRACE("cCreds = %d\n", schanCred->cCreds);
 243     TRACE("hRootStore = %p\n", schanCred->hRootStore);
 244     TRACE("cMappers = %d\n", schanCred->cMappers);
 245     TRACE("cSupportedAlgs = %d:\n", schanCred->cSupportedAlgs);
 246     for (i = 0; i < schanCred->cSupportedAlgs; i++)
 247         TRACE("%08x\n", schanCred->palgSupportedAlgs[i]);
 248     TRACE("grbitEnabledProtocols = %08x\n", schanCred->grbitEnabledProtocols);
 249     TRACE("dwMinimumCipherStrength = %d\n", schanCred->dwMinimumCipherStrength);
 250     TRACE("dwMaximumCipherStrength = %d\n", schanCred->dwMaximumCipherStrength);
 251     TRACE("dwSessionLifespan = %d\n", schanCred->dwSessionLifespan);
 252     TRACE("dwFlags = %08x\n", schanCred->dwFlags);
 253     TRACE("dwCredFormat = %d\n", schanCred->dwCredFormat);
 254
 255     switch (schanCred->dwVersion)
 256     {
 257     case SCH_CRED_V3:
 258     case SCHANNEL_CRED_VERSION:
 259         break;
 260     default:
 261         return SEC_E_INTERNAL_ERROR;
 262     }
 263
 264     if (schanCred->cCreds == 0)
 265         st = SEC_E_NO_CREDENTIALS;
 266     else if (schanCred->cCreds > 1)
 267         st = SEC_E_UNKNOWN_CREDENTIALS;
 268     else
 269     {
 270         DWORD keySpec;
 271         HCRYPTPROV csp;
 272         BOOL ret, freeCSP;
 273
 274         ret = CryptAcquireCertificatePrivateKey(schanCred->paCred[0],
 275          0, /* FIXME: what flags to use? */ NULL,
 276          &csp, &keySpec, &freeCSP);
 277         if (ret)
 278         {
 279             st = SEC_E_OK;
 280             if (freeCSP)
 281                 CryptReleaseContext(csp, 0);
 282         }
 283         else
 284             st = SEC_E_UNKNOWN_CREDENTIALS;
 285     }
 286     return st;
 287 }
 288
 289 static SECURITY_STATUS schan_AcquireClientCredentials(const SCHANNEL_CRED *schanCred,
 290  PCredHandle phCredential, PTimeStamp ptsExpiry)
 291 {
 292     struct schan_credentials *creds;
 293     SECURITY_STATUS st = SEC_E_OK;
 294
 295     TRACE("schanCred %p, phCredential %p, ptsExpiry %p\n", schanCred, phCredential, ptsExpiry);
 296
 297     if (schanCred)
 298     {
 299         st = schan_CheckCreds(schanCred);
 300         if (st == SEC_E_NO_CREDENTIALS)
 301             st = SEC_E_OK;
 302     }
 303
 304     /* For now, the only thing I'm interested in is the direction of the
 305      * connection, so just store it.
 306      */
 307     if (st == SEC_E_OK)
 308     {
 309         ULONG_PTR handle;
 310
 311         creds = HeapAlloc(GetProcessHeap(), 0, sizeof(*creds));
 312         if (!creds) return SEC_E_INSUFFICIENT_MEMORY;
 313
 314         handle = schan_alloc_handle(creds, SCHAN_HANDLE_CRED);
 315         if (handle == SCHAN_INVALID_HANDLE) goto fail;
 316
 317         creds->credential_use = SECPKG_CRED_OUTBOUND;
 318         if (!schan_imp_allocate_certificate_credentials(&creds->credentials))
 319         {
 320             schan_free_handle(handle, SCHAN_HANDLE_CRED);
 321             goto fail;
 322         }
 323
 324         phCredential->dwLower = handle;
 325         phCredential->dwUpper = 0;
 326
 327         /* Outbound credentials have no expiry */
 328         if (ptsExpiry)
 329         {
 330             ptsExpiry->LowPart = 0;
 331             ptsExpiry->HighPart = 0;
 332         }
 333     }
 334     return st;
 335
 336 fail:
 337     HeapFree(GetProcessHeap(), 0, creds);
 338     return SEC_E_INTERNAL_ERROR;
 339 }
 340
 341 static SECURITY_STATUS schan_AcquireServerCredentials(const SCHANNEL_CRED *schanCred,
 342  PCredHandle phCredential, PTimeStamp ptsExpiry)
 343 {
 344     SECURITY_STATUS st;
 345
 346     TRACE("schanCred %p, phCredential %p, ptsExpiry %p\n", schanCred, phCredential, ptsExpiry);
 347
 348     if (!schanCred) return SEC_E_NO_CREDENTIALS;
 349
 350     st = schan_CheckCreds(schanCred);
 351     if (st == SEC_E_OK)
 352     {
 353         ULONG_PTR handle;
 354         struct schan_credentials *creds;
 355
 356         creds = HeapAlloc(GetProcessHeap(), 0, sizeof(*creds));
 357         if (!creds) return SEC_E_INSUFFICIENT_MEMORY;
 358         creds->credential_use = SECPKG_CRED_INBOUND;
 359
 360         handle = schan_alloc_handle(creds, SCHAN_HANDLE_CRED);
 361         if (handle == SCHAN_INVALID_HANDLE)
 362         {
 363             HeapFree(GetProcessHeap(), 0, creds);
 364             return SEC_E_INTERNAL_ERROR;
 365         }
 366
 367         phCredential->dwLower = handle;
 368         phCredential->dwUpper = 0;
 369
 370         /* FIXME: get expiry from cert */
 371     }
 372     return st;
 373 }
 374
 375 static SECURITY_STATUS schan_AcquireCredentialsHandle(ULONG fCredentialUse,
 376  const SCHANNEL_CRED *schanCred, PCredHandle phCredential, PTimeStamp ptsExpiry)
 377 {
 378     SECURITY_STATUS ret;
 379
 380     if (fCredentialUse == SECPKG_CRED_OUTBOUND)
 381         ret = schan_AcquireClientCredentials(schanCred, phCredential,
 382          ptsExpiry);
 383     else
 384         ret = schan_AcquireServerCredentials(schanCred, phCredential,
 385          ptsExpiry);
 386     return ret;
 387 }
 388
 389 static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleA(
 390  SEC_CHAR *pszPrincipal, SEC_CHAR *pszPackage, ULONG fCredentialUse,
 391  PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn,
 392  PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry)
 393 {
 394     TRACE("(%s, %s, 0x%08x, %p, %p, %p, %p, %p, %p)\n",
 395      debugstr_a(pszPrincipal), debugstr_a(pszPackage), fCredentialUse,
 396      pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry);
 397     return schan_AcquireCredentialsHandle(fCredentialUse,
 398      pAuthData, phCredential, ptsExpiry);
 399 }
 400
 401 static SECURITY_STATUS SEC_ENTRY schan_AcquireCredentialsHandleW(
 402  SEC_WCHAR *pszPrincipal, SEC_WCHAR *pszPackage, ULONG fCredentialUse,
 403  PLUID pLogonID, PVOID pAuthData, SEC_GET_KEY_FN pGetKeyFn,
 404  PVOID pGetKeyArgument, PCredHandle phCredential, PTimeStamp ptsExpiry)
 405 {
 406     TRACE("(%s, %s, 0x%08x, %p, %p, %p, %p, %p, %p)\n",
 407      debugstr_w(pszPrincipal), debugstr_w(pszPackage), fCredentialUse,
 408      pLogonID, pAuthData, pGetKeyFn, pGetKeyArgument, phCredential, ptsExpiry);
 409     return schan_AcquireCredentialsHandle(fCredentialUse,
 410      pAuthData, phCredential, ptsExpiry);
 411 }
 412
 413 static SECURITY_STATUS SEC_ENTRY schan_FreeCredentialsHandle(
 414  PCredHandle phCredential)
 415 {
 416     struct schan_credentials *creds;
 417
 418     TRACE("phCredential %p\n", phCredential);
 419
 420     if (!phCredential) return SEC_E_INVALID_HANDLE;
 421
 422     creds = schan_free_handle(phCredential->dwLower, SCHAN_HANDLE_CRED);
 423     if (!creds) return SEC_E_INVALID_HANDLE;
 424
 425     if (creds->credential_use == SECPKG_CRED_OUTBOUND)
 426         schan_imp_free_certificate_credentials(creds->credentials);
 427     HeapFree(GetProcessHeap(), 0, creds);
 428
 429     return SEC_E_OK;
 430 }
 431
 432 static void init_schan_buffers(struct schan_buffers *s, const PSecBufferDesc desc,
 433         int (*get_next_buffer)(const struct schan_transport *, struct schan_buffers *))
 434 {
 435     s->offset = 0;
 436     s->limit = ~0UL;
 437     s->desc = desc;
 438     s->current_buffer_idx = -1;
 439     s->allow_buffer_resize = FALSE;
 440     s->get_next_buffer = get_next_buffer;
 441 }
 442
 443 static int schan_find_sec_buffer_idx(const SecBufferDesc *desc, unsigned int start_idx, ULONG buffer_type)
 444 {
 445     unsigned int i;
 446     PSecBuffer buffer;
 447
 448     for (i = start_idx; i < desc->cBuffers; ++i)
 449     {
 450         buffer = &desc->pBuffers[i];
 451         if (buffer->BufferType == buffer_type) return i;
 452     }
 453
 454     return -1;
 455 }
 456
 457 static void schan_resize_current_buffer(const struct schan_buffers *s, SIZE_T min_size)
 458 {
 459     SecBuffer *b = &s->desc->pBuffers[s->current_buffer_idx];
 460     SIZE_T new_size = b->cbBuffer ? b->cbBuffer * 2 : 128;
 461     void *new_data;
 462
 463     if (b->cbBuffer >= min_size || !s->allow_buffer_resize || min_size > UINT_MAX / 2) return;
 464
 465     while (new_size < min_size) new_size *= 2;
 466
 467     if (b->pvBuffer)
 468         new_data = HeapReAlloc(GetProcessHeap(), 0, b->pvBuffer, new_size);
 469     else
 470         new_data = HeapAlloc(GetProcessHeap(), 0, new_size);
 471
 472     if (!new_data)
 473     {
 474         TRACE("Failed to resize %p from %d to %ld\n", b->pvBuffer, b->cbBuffer, new_size);
 475         return;
 476     }
 477
 478     b->cbBuffer = new_size;
 479     b->pvBuffer = new_data;
 480 }
 481
 482 char *schan_get_buffer(const struct schan_transport *t, struct schan_buffers *s, SIZE_T *count)
 483 {
 484     SIZE_T max_count;
 485     PSecBuffer buffer;
 486
 487     if (!s->desc)
 488     {
 489         TRACE("No desc\n");
 490         return NULL;
 491     }
 492
 493     if (s->current_buffer_idx == -1)
 494     {
 495         /* Initial buffer */
 496         int buffer_idx = s->get_next_buffer(t, s);
 497         if (buffer_idx == -1)
 498         {
 499             TRACE("No next buffer\n");
 500             return NULL;
 501         }
 502         s->current_buffer_idx = buffer_idx;
 503     }
 504
 505     buffer = &s->desc->pBuffers[s->current_buffer_idx];
 506     TRACE("Using buffer %d: cbBuffer %d, BufferType %#x, pvBuffer %p\n", s->current_buffer_idx, buffer->cbBuffer, buffer->BufferType, buffer->pvBuffer);
 507
 508     schan_resize_current_buffer(s, s->offset + *count);
 509     max_count = buffer->cbBuffer - s->offset;
 510     if (s->limit != ~0UL && s->limit < max_count)
 511         max_count = s->limit;
 512     if (!max_count)
 513     {
 514         int buffer_idx;
 515
 516         s->allow_buffer_resize = FALSE;
 517         buffer_idx = s->get_next_buffer(t, s);
 518         if (buffer_idx == -1)
 519         {
 520             TRACE("No next buffer\n");
 521             return NULL;
 522         }
 523         s->current_buffer_idx = buffer_idx;
 524         s->offset = 0;
 525         return schan_get_buffer(t, s, count);
 526     }
 527
 528     if (*count > max_count)
 529         *count = max_count;
 530     if (s->limit != ~0UL)
 531         s->limit -= *count;
 532
 533     return (char *)buffer->pvBuffer + s->offset;
 534 }
 535
 536 /* schan_pull
 537  *      Read data from the transport input buffer.
 538  *
 539  * t - The session transport object.
 540  * buff - The buffer into which to store the read data.  Must be at least
 541  *        *buff_len bytes in length.
 542  * buff_len - On input, *buff_len is the desired length to read.  On successful
 543  *            return, *buff_len is the number of bytes actually read.
 544  *
 545  * Returns:
 546  *  0 on success, in which case:
 547  *      *buff_len == 0 indicates end of file.
 548  *      *buff_len > 0 indicates that some data was read.  May be less than
 549  *          what was requested, in which case the caller should call again if/
 550  *          when they want more.
 551  *  EAGAIN when no data could be read without blocking
 552  *  another errno-style error value on failure
 553  *
 554  */
 555 int schan_pull(struct schan_transport *t, void *buff, size_t *buff_len)
 556 {
 557     char *b;
 558     SIZE_T local_len = *buff_len;
 559
 560     TRACE("Pull %lu bytes\n", local_len);
 561
 562     *buff_len = 0;
 563
 564     b = schan_get_buffer(t, &t->in, &local_len);
 565     if (!b)
 566         return EAGAIN;
 567
 568     memcpy(buff, b, local_len);
 569     t->in.offset += local_len;
 570
 571     TRACE("Read %lu bytes\n", local_len);
 572
 573     *buff_len = local_len;
 574     return 0;
 575 }
 576
 577 /* schan_push
 578  *      Write data to the transport output buffer.
 579  *
 580  * t - The session transport object.
 581  * buff - The buffer of data to write.  Must be at least *buff_len bytes in length.
 582  * buff_len - On input, *buff_len is the desired length to write.  On successful
 583  *            return, *buff_len is the number of bytes actually written.
 584  *
 585  * Returns:
 586  *  0 on success
 587  *      *buff_len will be > 0 indicating how much data was written.  May be less
 588  *          than what was requested, in which case the caller should call again
 589             if/when they want to write more.
 590  *  EAGAIN when no data could be written without blocking
 591  *  another errno-style error value on failure
 592  *
 593  */
 594 int schan_push(struct schan_transport *t, const void *buff, size_t *buff_len)
 595 {
 596     char *b;
 597     SIZE_T local_len = *buff_len;
 598
 599     TRACE("Push %lu bytes\n", local_len);
 600
 601     *buff_len = 0;
 602
 603     b = schan_get_buffer(t, &t->out, &local_len);
 604     if (!b)
 605         return EAGAIN;
 606
 607     memcpy(b, buff, local_len);
 608     t->out.offset += local_len;
 609
 610     TRACE("Wrote %lu bytes\n", local_len);
 611
 612     *buff_len = local_len;
 613     return 0;
 614 }
 615
 616 schan_imp_session schan_session_for_transport(struct schan_transport* t)
 617 {
 618     return t->ctx->session;
 619 }
 620
 621 static int schan_init_sec_ctx_get_next_buffer(const struct schan_transport *t, struct schan_buffers *s)
 622 {
 623     if (s->current_buffer_idx == -1)
 624     {
 625         int idx = schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_TOKEN);
 626         if (t->ctx->req_ctx_attr & ISC_REQ_ALLOCATE_MEMORY)
 627         {
 628             if (idx == -1)
 629             {
 630                 idx = schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_EMPTY);
 631                 if (idx != -1) s->desc->pBuffers[idx].BufferType = SECBUFFER_TOKEN;
 632             }
 633             if (idx != -1 && !s->desc->pBuffers[idx].pvBuffer)
 634             {
 635                 s->desc->pBuffers[idx].cbBuffer = 0;
 636                 s->allow_buffer_resize = TRUE;
 637             }
 638         }
 639         return idx;
 640     }
 641
 642     return -1;
 643 }
 644
 645 static void dump_buffer_desc(SecBufferDesc *desc)
 646 {
 647     unsigned int i;
 648
 649     if (!desc) return;
 650     TRACE("Buffer desc %p:\n", desc);
 651     for (i = 0; i < desc->cBuffers; ++i)
 652     {
 653         SecBuffer *b = &desc->pBuffers[i];
 654         TRACE("\tbuffer %u: cbBuffer %d, BufferType %#x pvBuffer %p\n", i, b->cbBuffer, b->BufferType, b->pvBuffer);
 655     }
 656 }
 657
 658 /***********************************************************************
 659  *              InitializeSecurityContextW
 660  */
 661 static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextW(
 662  PCredHandle phCredential, PCtxtHandle phContext, SEC_WCHAR *pszTargetName,
 663  ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep,
 664  PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext,
 665  PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry)
 666 {
 667     struct schan_context *ctx;
 668     struct schan_buffers *out_buffers;
 669     struct schan_credentials *cred;
 670     struct schan_transport transport;
 671     SIZE_T expected_size = ~0UL;
 672     SECURITY_STATUS ret;
 673
 674     TRACE("%p %p %s 0x%08x %d %d %p %d %p %p %p %p\n", phCredential, phContext,
 675      debugstr_w(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
 676      Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
 677
 678     dump_buffer_desc(pInput);
 679     dump_buffer_desc(pOutput);
 680
 681     if (!phContext)
 682     {
 683         ULONG_PTR handle;
 684
 685         if (!phCredential) return SEC_E_INVALID_HANDLE;
 686
 687         cred = schan_get_object(phCredential->dwLower, SCHAN_HANDLE_CRED);
 688         if (!cred) return SEC_E_INVALID_HANDLE;
 689
 690         if (!(cred->credential_use & SECPKG_CRED_OUTBOUND))
 691         {
 692             WARN("Invalid credential use %#x\n", cred->credential_use);
 693             return SEC_E_INVALID_HANDLE;
 694         }
 695
 696         ctx = HeapAlloc(GetProcessHeap(), 0, sizeof(*ctx));
 697         if (!ctx) return SEC_E_INSUFFICIENT_MEMORY;
 698
 699         handle = schan_alloc_handle(ctx, SCHAN_HANDLE_CTX);
 700         if (handle == SCHAN_INVALID_HANDLE)
 701         {
 702             HeapFree(GetProcessHeap(), 0, ctx);
 703             return SEC_E_INTERNAL_ERROR;
 704         }
 705
 706         if (!schan_imp_create_session(&ctx->session, FALSE, cred->credentials))
 707         {
 708             schan_free_handle(handle, SCHAN_HANDLE_CTX);
 709             HeapFree(GetProcessHeap(), 0, ctx);
 710             return SEC_E_INTERNAL_ERROR;
 711         }
 712
 713         phNewContext->dwLower = handle;
 714         phNewContext->dwUpper = 0;
 715     }
 716     else
 717     {
 718         SIZE_T record_size = 0;
 719         unsigned char *ptr;
 720         SecBuffer *buffer;
 721         int idx;
 722
 723         if (!pInput)
 724             return SEC_E_INCOMPLETE_MESSAGE;
 725
 726         idx = schan_find_sec_buffer_idx(pInput, 0, SECBUFFER_TOKEN);
 727         if (idx == -1)
 728             return SEC_E_INCOMPLETE_MESSAGE;
 729
 730         buffer = &pInput->pBuffers[idx];
 731         ptr = buffer->pvBuffer;
 732         expected_size = 0;
 733
 734         while (buffer->cbBuffer > expected_size + 5)
 735         {
 736             record_size = 5 + ((ptr[3] << 8) | ptr[4]);
 737
 738             if (buffer->cbBuffer < expected_size + record_size)
 739                 break;
 740
 741             expected_size += record_size;
 742             ptr += record_size;
 743         }
 744
 745         if (!expected_size)
 746         {
 747             TRACE("Expected at least %lu bytes, but buffer only contains %u bytes.\n",
 748                     max(6, record_size), buffer->cbBuffer);
 749             return SEC_E_INCOMPLETE_MESSAGE;
 750         }
 751
 752         TRACE("Using expected_size %lu.\n", expected_size);
 753
 754         ctx = schan_get_object(phContext->dwLower, SCHAN_HANDLE_CTX);
 755     }
 756
 757     ctx->req_ctx_attr = fContextReq;
 758
 759     transport.ctx = ctx;
 760     init_schan_buffers(&transport.in, pInput, schan_init_sec_ctx_get_next_buffer);
 761     transport.in.limit = expected_size;
 762     init_schan_buffers(&transport.out, pOutput, schan_init_sec_ctx_get_next_buffer);
 763     schan_imp_set_session_transport(ctx->session, &transport);
 764
 765     /* Perform the TLS handshake */
 766     ret = schan_imp_handshake(ctx->session);
 767
 768     if(transport.in.offset && transport.in.offset != pInput->pBuffers[0].cbBuffer) {
 769         if(pInput->cBuffers<2 || pInput->pBuffers[1].BufferType!=SECBUFFER_EMPTY)
 770             return SEC_E_INVALID_TOKEN;
 771
 772         pInput->pBuffers[1].BufferType = SECBUFFER_EXTRA;
 773         pInput->pBuffers[1].cbBuffer = pInput->pBuffers[0].cbBuffer-transport.in.offset;
 774     }
 775
 776     out_buffers = &transport.out;
 777     if (out_buffers->current_buffer_idx != -1)
 778     {
 779         SecBuffer *buffer = &out_buffers->desc->pBuffers[out_buffers->current_buffer_idx];
 780         buffer->cbBuffer = out_buffers->offset;
 781     }
 782
 783     *pfContextAttr = 0;
 784     if (ctx->req_ctx_attr & ISC_REQ_ALLOCATE_MEMORY)
 785         *pfContextAttr |= ISC_RET_ALLOCATED_MEMORY;
 786
 787     return ret;
 788 }
 789
 790 /***********************************************************************
 791  *              InitializeSecurityContextA
 792  */
 793 static SECURITY_STATUS SEC_ENTRY schan_InitializeSecurityContextA(
 794  PCredHandle phCredential, PCtxtHandle phContext, SEC_CHAR *pszTargetName,
 795  ULONG fContextReq, ULONG Reserved1, ULONG TargetDataRep,
 796  PSecBufferDesc pInput, ULONG Reserved2, PCtxtHandle phNewContext,
 797  PSecBufferDesc pOutput, ULONG *pfContextAttr, PTimeStamp ptsExpiry)
 798 {
 799     SECURITY_STATUS ret;
 800     SEC_WCHAR *target_name = NULL;
 801
 802     TRACE("%p %p %s %d %d %d %p %d %p %p %p %p\n", phCredential, phContext,
 803      debugstr_a(pszTargetName), fContextReq, Reserved1, TargetDataRep, pInput,
 804      Reserved1, phNewContext, pOutput, pfContextAttr, ptsExpiry);
 805
 806     if (pszTargetName)
 807     {
 808         INT len = MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, NULL, 0);
 809         target_name = HeapAlloc(GetProcessHeap(), 0, len * sizeof(*target_name));
 810         MultiByteToWideChar(CP_ACP, 0, pszTargetName, -1, target_name, len);
 811     }
 812
 813     ret = schan_InitializeSecurityContextW(phCredential, phContext, target_name,
 814             fContextReq, Reserved1, TargetDataRep, pInput, Reserved2,
 815             phNewContext, pOutput, pfContextAttr, ptsExpiry);
 816
 817     HeapFree(GetProcessHeap(), 0, target_name);
 818
 819     return ret;
 820 }
 821
 822 static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesW(
 823         PCtxtHandle context_handle, ULONG attribute, PVOID buffer)
 824 {
 825     struct schan_context *ctx;
 826
 827     TRACE("context_handle %p, attribute %#x, buffer %p\n",
 828             context_handle, attribute, buffer);
 829
 830     if (!context_handle) return SEC_E_INVALID_HANDLE;
 831     ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
 832
 833     switch(attribute)
 834     {
 835         case SECPKG_ATTR_STREAM_SIZES:
 836         {
 837             SecPkgContext_ConnectionInfo info;
 838             SECURITY_STATUS status = schan_imp_get_connection_info(ctx->session, &info);
 839             if (status == SEC_E_OK)
 840             {
 841                 SecPkgContext_StreamSizes *stream_sizes = buffer;
 842                 SIZE_T mac_size = info.dwHashStrength;
 843                 unsigned int block_size = schan_imp_get_session_cipher_block_size(ctx->session);
 844                 unsigned int message_size = schan_imp_get_max_message_size(ctx->session);
 845
 846                 TRACE("Using %lu mac bytes, message size %u, block size %u\n",
 847                         mac_size, message_size, block_size);
 848
 849                 /* These are defined by the TLS RFC */
 850                 stream_sizes->cbHeader = 5;
 851                 stream_sizes->cbTrailer = mac_size + 256; /* Max 255 bytes padding + 1 for padding size */
 852                 stream_sizes->cbMaximumMessage = message_size;
 853                 stream_sizes->cbBuffers = 4;
 854                 stream_sizes->cbBlockSize = block_size;
 855             }
 856
 857             return status;
 858         }
 859         case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
 860         {
 861             PCCERT_CONTEXT *cert = buffer;
 862             return schan_imp_get_session_peer_certificate(ctx->session, cert);
 863         }
 864         case SECPKG_ATTR_CONNECTION_INFO:
 865         {
 866             SecPkgContext_ConnectionInfo *info = buffer;
 867             return schan_imp_get_connection_info(ctx->session, info);
 868         }
 869
 870         default:
 871             FIXME("Unhandled attribute %#x\n", attribute);
 872             return SEC_E_UNSUPPORTED_FUNCTION;
 873     }
 874 }
 875
 876 static SECURITY_STATUS SEC_ENTRY schan_QueryContextAttributesA(
 877         PCtxtHandle context_handle, ULONG attribute, PVOID buffer)
 878 {
 879     TRACE("context_handle %p, attribute %#x, buffer %p\n",
 880             context_handle, attribute, buffer);
 881
 882     switch(attribute)
 883     {
 884         case SECPKG_ATTR_STREAM_SIZES:
 885             return schan_QueryContextAttributesW(context_handle, attribute, buffer);
 886         case SECPKG_ATTR_REMOTE_CERT_CONTEXT:
 887             return schan_QueryContextAttributesW(context_handle, attribute, buffer);
 888         case SECPKG_ATTR_CONNECTION_INFO:
 889             return schan_QueryContextAttributesW(context_handle, attribute, buffer);
 890
 891         default:
 892             FIXME("Unhandled attribute %#x\n", attribute);
 893             return SEC_E_UNSUPPORTED_FUNCTION;
 894     }
 895 }
 896
 897 static int schan_encrypt_message_get_next_buffer(const struct schan_transport *t, struct schan_buffers *s)
 898 {
 899     SecBuffer *b;
 900
 901     if (s->current_buffer_idx == -1)
 902         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_STREAM_HEADER);
 903
 904     b = &s->desc->pBuffers[s->current_buffer_idx];
 905
 906     if (b->BufferType == SECBUFFER_STREAM_HEADER)
 907         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_DATA);
 908
 909     if (b->BufferType == SECBUFFER_DATA)
 910         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_STREAM_TRAILER);
 911
 912     return -1;
 913 }
 914
 915 static int schan_encrypt_message_get_next_buffer_token(const struct schan_transport *t, struct schan_buffers *s)
 916 {
 917     SecBuffer *b;
 918
 919     if (s->current_buffer_idx == -1)
 920         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_TOKEN);
 921
 922     b = &s->desc->pBuffers[s->current_buffer_idx];
 923
 924     if (b->BufferType == SECBUFFER_TOKEN)
 925     {
 926         int idx = schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_TOKEN);
 927         if (idx != s->current_buffer_idx) return -1;
 928         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_DATA);
 929     }
 930
 931     if (b->BufferType == SECBUFFER_DATA)
 932     {
 933         int idx = schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_TOKEN);
 934         if (idx != -1)
 935             idx = schan_find_sec_buffer_idx(s->desc, idx + 1, SECBUFFER_TOKEN);
 936         return idx;
 937     }
 938
 939     return -1;
 940 }
 941
 942 static SECURITY_STATUS SEC_ENTRY schan_EncryptMessage(PCtxtHandle context_handle,
 943         ULONG quality, PSecBufferDesc message, ULONG message_seq_no)
 944 {
 945     struct schan_transport transport;
 946     struct schan_context *ctx;
 947     struct schan_buffers *b;
 948     SECURITY_STATUS status;
 949     SecBuffer *buffer;
 950     SIZE_T data_size;
 951     SIZE_T length;
 952     char *data;
 953     int idx;
 954
 955     TRACE("context_handle %p, quality %d, message %p, message_seq_no %d\n",
 956             context_handle, quality, message, message_seq_no);
 957
 958     if (!context_handle) return SEC_E_INVALID_HANDLE;
 959     ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
 960
 961     dump_buffer_desc(message);
 962
 963     idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_DATA);
 964     if (idx == -1)
 965     {
 966         WARN("No data buffer passed\n");
 967         return SEC_E_INTERNAL_ERROR;
 968     }
 969     buffer = &message->pBuffers[idx];
 970
 971     data_size = buffer->cbBuffer;
 972     data = HeapAlloc(GetProcessHeap(), 0, data_size);
 973     memcpy(data, buffer->pvBuffer, data_size);
 974
 975     transport.ctx = ctx;
 976     init_schan_buffers(&transport.in, NULL, NULL);
 977     if (schan_find_sec_buffer_idx(message, 0, SECBUFFER_STREAM_HEADER) != -1)
 978         init_schan_buffers(&transport.out, message, schan_encrypt_message_get_next_buffer);
 979     else
 980         init_schan_buffers(&transport.out, message, schan_encrypt_message_get_next_buffer_token);
 981     schan_imp_set_session_transport(ctx->session, &transport);
 982
 983     length = data_size;
 984     status = schan_imp_send(ctx->session, data, &length);
 985
 986     TRACE("Sent %ld bytes.\n", length);
 987
 988     if (length != data_size)
 989         status = SEC_E_INTERNAL_ERROR;
 990
 991     b = &transport.out;
 992     b->desc->pBuffers[b->current_buffer_idx].cbBuffer = b->offset;
 993     HeapFree(GetProcessHeap(), 0, data);
 994
 995     TRACE("Returning %#x.\n", status);
 996
 997     return status;
 998 }
 999
1000 static int schan_decrypt_message_get_next_buffer(const struct schan_transport *t, struct schan_buffers *s)
1001 {
1002     if (s->current_buffer_idx == -1)
1003         return schan_find_sec_buffer_idx(s->desc, 0, SECBUFFER_DATA);
1004
1005     return -1;
1006 }
1007
1008 static int schan_validate_decrypt_buffer_desc(PSecBufferDesc message)
1009 {
1010     int data_idx = -1;
1011     unsigned int empty_count = 0;
1012     unsigned int i;
1013
1014     if (message->cBuffers < 4)
1015     {
1016         WARN("Less than four buffers passed\n");
1017         return -1;
1018     }
1019
1020     for (i = 0; i < message->cBuffers; ++i)
1021     {
1022         SecBuffer *b = &message->pBuffers[i];
1023         if (b->BufferType == SECBUFFER_DATA)
1024         {
1025             if (data_idx != -1)
1026             {
1027                 WARN("More than one data buffer passed\n");
1028                 return -1;
1029             }
1030             data_idx = i;
1031         }
1032         else if (b->BufferType == SECBUFFER_EMPTY)
1033             ++empty_count;
1034     }
1035
1036     if (data_idx == -1)
1037     {
1038         WARN("No data buffer passed\n");
1039         return -1;
1040     }
1041
1042     if (empty_count < 3)
1043     {
1044         WARN("Less than three empty buffers passed\n");
1045         return -1;
1046     }
1047
1048     return data_idx;
1049 }
1050
1051 static void schan_decrypt_fill_buffer(PSecBufferDesc message, ULONG buffer_type, void *data, ULONG size)
1052 {
1053     int idx;
1054     SecBuffer *buffer;
1055
1056     idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY);
1057     buffer = &message->pBuffers[idx];
1058
1059     buffer->BufferType = buffer_type;
1060     buffer->pvBuffer = data;
1061     buffer->cbBuffer = size;
1062 }
1063
1064 static SECURITY_STATUS SEC_ENTRY schan_DecryptMessage(PCtxtHandle context_handle,
1065         PSecBufferDesc message, ULONG message_seq_no, PULONG quality)
1066 {
1067     struct schan_transport transport;
1068     struct schan_context *ctx;
1069     SecBuffer *buffer;
1070     SIZE_T data_size;
1071     char *data;
1072     unsigned expected_size;
1073     SSIZE_T received = 0;
1074     int idx;
1075     unsigned char *buf_ptr;
1076
1077     TRACE("context_handle %p, message %p, message_seq_no %d, quality %p\n",
1078             context_handle, message, message_seq_no, quality);
1079
1080     if (!context_handle) return SEC_E_INVALID_HANDLE;
1081     ctx = schan_get_object(context_handle->dwLower, SCHAN_HANDLE_CTX);
1082
1083     dump_buffer_desc(message);
1084
1085     idx = schan_validate_decrypt_buffer_desc(message);
1086     if (idx == -1)
1087         return SEC_E_INVALID_TOKEN;
1088     buffer = &message->pBuffers[idx];
1089     buf_ptr = buffer->pvBuffer;
1090
1091     expected_size = 5 + ((buf_ptr[3] << 8) | buf_ptr[4]);
1092     if(buffer->cbBuffer < expected_size)
1093     {
1094         TRACE("Expected %u bytes, but buffer only contains %u bytes\n", expected_size, buffer->cbBuffer);
1095         buffer->BufferType = SECBUFFER_MISSING;
1096         buffer->cbBuffer = expected_size - buffer->cbBuffer;
1097
1098         /* This is a bit weird, but windows does it too */
1099         idx = schan_find_sec_buffer_idx(message, 0, SECBUFFER_EMPTY);
1100         buffer = &message->pBuffers[idx];
1101         buffer->BufferType = SECBUFFER_MISSING;
1102         buffer->cbBuffer = expected_size - buffer->cbBuffer;
1103
1104         TRACE("Returning SEC_E_INCOMPLETE_MESSAGE\n");
1105         return SEC_E_INCOMPLETE_MESSAGE;
1106     }
1107
1108     data_size = expected_size - 5;
1109     data = HeapAlloc(GetProcessHeap(), 0, data_size);
1110
1111     transport.ctx = ctx;
1112     init_schan_buffers(&transport.in, message, schan_decrypt_message_get_next_buffer);
1113     transport.in.limit = expected_size;
1114     init_schan_buffers(&transport.out, NULL, NULL);
1115     schan_imp_set_session_transport(ctx->session, &transport);
1116
1117     while (received < data_size)
1118     {
1119         SIZE_T length = data_size - received;
1120         SECURITY_STATUS status = schan_imp_recv(ctx->session, data + received, &length);
1121
1122         if (status == SEC_I_CONTINUE_NEEDED)
1123             break;
1124
1125         if (status != SEC_E_OK)
1126         {
1127             HeapFree(GetProcessHeap(), 0, data);
1128             ERR("Returning %d\n", status);
1129             return status;
1130         }
1131
1132         if (!length)
1133             break;
1134
1135         received += length;
1136     }
1137
1138     TRACE("Received %ld bytes\n", received);
1139
1140     memcpy(buf_ptr + 5, data, received);
1141     HeapFree(GetProcessHeap(), 0, data);
1142
1143     schan_decrypt_fill_buffer(message, SECBUFFER_DATA,
1144         buf_ptr + 5, received);
1145
1146     schan_decrypt_fill_buffer(message, SECBUFFER_STREAM_TRAILER,
1147         buf_ptr + 5 + received, buffer->cbBuffer - 5 - received);
1148
1149     if(buffer->cbBuffer > expected_size)
1150         schan_decrypt_fill_buffer(message, SECBUFFER_EXTRA,
1151             buf_ptr + expected_size, buffer->cbBuffer - expected_size);
1152
1153     buffer->BufferType = SECBUFFER_STREAM_HEADER;
1154     buffer->cbBuffer = 5;
1155
1156     return SEC_E_OK;
1157 }
1158
1159 static SECURITY_STATUS SEC_ENTRY schan_DeleteSecurityContext(PCtxtHandle context_handle)
1160 {
1161     struct schan_context *ctx;
1162
1163     TRACE("context_handle %p\n", context_handle);
1164
1165     if (!context_handle) return SEC_E_INVALID_HANDLE;
1166
1167     ctx = schan_free_handle(context_handle->dwLower, SCHAN_HANDLE_CTX);
1168     if (!ctx) return SEC_E_INVALID_HANDLE;
1169
1170     schan_imp_dispose_session(ctx->session);
1171     HeapFree(GetProcessHeap(), 0, ctx);
1172
1173     return SEC_E_OK;
1174 }
1175
1176 static const SecurityFunctionTableA schanTableA = {
1177     1,
1178     NULL, /* EnumerateSecurityPackagesA */
1179     schan_QueryCredentialsAttributesA,
1180     schan_AcquireCredentialsHandleA,
1181     schan_FreeCredentialsHandle,
1182     NULL, /* Reserved2 */
1183     schan_InitializeSecurityContextA,
1184     NULL, /* AcceptSecurityContext */
1185     NULL, /* CompleteAuthToken */
1186     schan_DeleteSecurityContext,
1187     NULL, /* ApplyControlToken */
1188     schan_QueryContextAttributesA,
1189     NULL, /* ImpersonateSecurityContext */
1190     NULL, /* RevertSecurityContext */
1191     NULL, /* MakeSignature */
1192     NULL, /* VerifySignature */
1193     FreeContextBuffer,
1194     NULL, /* QuerySecurityPackageInfoA */
1195     NULL, /* Reserved3 */
1196     NULL, /* Reserved4 */
1197     NULL, /* ExportSecurityContext */
1198     NULL, /* ImportSecurityContextA */
1199     NULL, /* AddCredentialsA */
1200     NULL, /* Reserved8 */
1201     NULL, /* QuerySecurityContextToken */
1202     schan_EncryptMessage,
1203     schan_DecryptMessage,
1204     NULL, /* SetContextAttributesA */
1205 };
1206
1207 static const SecurityFunctionTableW schanTableW = {
1208     1,
1209     NULL, /* EnumerateSecurityPackagesW */
1210     schan_QueryCredentialsAttributesW,
1211     schan_AcquireCredentialsHandleW,
1212     schan_FreeCredentialsHandle,
1213     NULL, /* Reserved2 */
1214     schan_InitializeSecurityContextW,
1215     NULL, /* AcceptSecurityContext */
1216     NULL, /* CompleteAuthToken */
1217     schan_DeleteSecurityContext,
1218     NULL, /* ApplyControlToken */
1219     schan_QueryContextAttributesW,
1220     NULL, /* ImpersonateSecurityContext */
1221     NULL, /* RevertSecurityContext */
1222     NULL, /* MakeSignature */
1223     NULL, /* VerifySignature */
1224     FreeContextBuffer,
1225     NULL, /* QuerySecurityPackageInfoW */
1226     NULL, /* Reserved3 */
1227     NULL, /* Reserved4 */
1228     NULL, /* ExportSecurityContext */
1229     NULL, /* ImportSecurityContextW */
1230     NULL, /* AddCredentialsW */
1231     NULL, /* Reserved8 */
1232     NULL, /* QuerySecurityContextToken */
1233     schan_EncryptMessage,
1234     schan_DecryptMessage,
1235     NULL, /* SetContextAttributesW */
1236 };
1237
1238 static const WCHAR schannelComment[] = { 'S','c','h','a','n','n','e','l',' ',
1239  'S','e','c','u','r','i','t','y',' ','P','a','c','k','a','g','e',0 };
1240 static const WCHAR schannelDllName[] = { 's','c','h','a','n','n','e','l','.','d','l','l',0 };
1241
1242 void SECUR32_initSchannelSP(void)
1243 {
1244     /* This is what Windows reports.  This shouldn't break any applications
1245      * even though the functions are missing, because the wrapper will
1246      * return SEC_E_UNSUPPORTED_FUNCTION if our function is NULL.
1247      */
1248     static const LONG caps =
1249         SECPKG_FLAG_INTEGRITY |
1250         SECPKG_FLAG_PRIVACY |
1251         SECPKG_FLAG_CONNECTION |
1252         SECPKG_FLAG_MULTI_REQUIRED |
1253         SECPKG_FLAG_EXTENDED_ERROR |
1254         SECPKG_FLAG_IMPERSONATION |
1255         SECPKG_FLAG_ACCEPT_WIN32_NAME |
1256         SECPKG_FLAG_STREAM;
1257     static const short version = 1;
1258     static const LONG maxToken = 16384;
1259     SEC_WCHAR *uniSPName = (SEC_WCHAR *)UNISP_NAME_W,
1260               *schannel = (SEC_WCHAR *)SCHANNEL_NAME_W;
1261     const SecPkgInfoW info[] = {
1262         { caps, version, UNISP_RPC_ID, maxToken, uniSPName, uniSPName },
1263         { caps, version, UNISP_RPC_ID, maxToken, schannel,
1264             (SEC_WCHAR *)schannelComment },
1265     };
1266     SecureProvider *provider;
1267
1268     if (!schan_imp_init())
1269         return;
1270
1271     schan_handle_table = HeapAlloc(GetProcessHeap(), 0, 64 * sizeof(*schan_handle_table));
1272     if (!schan_handle_table)
1273     {
1274         ERR("Failed to allocate schannel handle table.\n");
1275         goto fail;
1276     }
1277     schan_handle_table_size = 64;
1278
1279     provider = SECUR32_addProvider(&schanTableA, &schanTableW, schannelDllName);
1280     if (!provider)
1281     {
1282         ERR("Failed to add schannel provider.\n");
1283         goto fail;
1284     }
1285
1286     SECUR32_addPackages(provider, sizeof(info) / sizeof(info[0]), NULL, info);
1287
1288     return;
1289
1290 fail:
1291     HeapFree(GetProcessHeap(), 0, schan_handle_table);
1292     schan_handle_table = NULL;
1293     schan_imp_deinit();
1294     return;
1295 }
1296
1297 void SECUR32_deinitSchannelSP(void)
1298 {
1299     SIZE_T i = schan_handle_count;
1300
1301     if (!schan_handle_table) return;
1302
1303     /* deinitialized sessions first because a pointer to the credentials
1304      * may be stored for the session. */
1305     while (i--)
1306     {
1307         if (schan_handle_table[i].type == SCHAN_HANDLE_CTX)
1308         {
1309             struct schan_context *ctx = schan_free_handle(i, SCHAN_HANDLE_CTX);
1310             schan_imp_dispose_session(ctx->session);
1311             HeapFree(GetProcessHeap(), 0, ctx);
1312         }
1313     }
1314     i = schan_handle_count;
1315     while (i--)
1316     {
1317         if (schan_handle_table[i].type != SCHAN_HANDLE_FREE)
1318         {
1319             struct schan_credentials *cred;
1320             cred = schan_free_handle(i, SCHAN_HANDLE_CRED);
1321             schan_imp_free_certificate_credentials(cred->credentials);
1322             HeapFree(GetProcessHeap(), 0, cred);
1323         }
1324     }
1325     HeapFree(GetProcessHeap(), 0, schan_handle_table);
1326     schan_imp_deinit();
1327 }
1328
1329 #else /* SONAME_LIBGNUTLS || HAVE_SECURITY_SECURITY_H */
1330
1331 void SECUR32_initSchannelSP(void)
1332 {
1333     ERR("TLS library not found, SSL connections will fail\n");
1334 }
1335
1336 void SECUR32_deinitSchannelSP(void) {}
1337
1338 #endif /* SONAME_LIBGNUTLS || HAVE_SECURITY_SECURITY_H */