]> git.ipfire.org Git - thirdparty/cups.git/blame - cups/tls-sspi.c
Add SSLOptions to enable Diffie-Hellman key exchange and disable TLS/1.0.
[thirdparty/cups.git] / cups / tls-sspi.c
CommitLineData
cc754834 1/*
321d8d57 2 * "$Id$"
cc754834 3 *
63aefcd5
MS
4 * TLS support for CUPS on Windows using the Security Support Provider
5 * Interface (SSPI).
321d8d57 6 *
3abb875b 7 * Copyright 2010-2015 by Apple Inc.
cc754834 8 *
2c85b752
MS
9 * These coded instructions, statements, and computer programs are the
10 * property of Apple Inc. and are protected by Federal copyright
11 * law. Distribution and use rights are outlined in the file "LICENSE.txt"
12 * which should have been included with this file. If this file is
13 * file is missing or damaged, see the license at "http://www.cups.org/".
cc754834 14 *
2c85b752 15 * This file is subject to the Apple OS-Developed Software exception.
cc754834
MS
16 */
17
ebb24a07
MS
18/**** This file is included from tls.c ****/
19
cc754834
MS
20/*
21 * Include necessary headers...
22 */
23
cc754834 24#include "debug-private.h"
321d8d57 25
cc754834 26
2ece34a9
MS
27/*
28 * Include necessary libraries...
29 */
30
cc754834
MS
31#pragma comment(lib, "Crypt32.lib")
32#pragma comment(lib, "Secur32.lib")
33#pragma comment(lib, "Ws2_32.lib")
34
35
2ece34a9
MS
36/*
37 * Constants...
38 */
39
40#ifndef SECURITY_FLAG_IGNORE_UNKNOWN_CA
cc754834 41# define SECURITY_FLAG_IGNORE_UNKNOWN_CA 0x00000100 /* Untrusted root */
2ece34a9 42#endif /* SECURITY_FLAG_IGNORE_UNKNOWN_CA */
cc754834 43
9041146f
MS
44#ifndef SECURITY_FLAG_IGNORE_CERT_CN_INVALID
45# define SECURITY_FLAG_IGNORE_CERT_CN_INVALID 0x00001000 /* Common name does not match */
46#endif /* !SECURITY_FLAG_IGNORE_CERT_CN_INVALID */
47
2ece34a9 48#ifndef SECURITY_FLAG_IGNORE_CERT_DATE_INVALID
cc754834 49# define SECURITY_FLAG_IGNORE_CERT_DATE_INVALID 0x00002000 /* Expired X509 Cert. */
2ece34a9
MS
50#endif /* !SECURITY_FLAG_IGNORE_CERT_DATE_INVALID */
51
63aefcd5
MS
52
53/*
54 * Local globals...
55 */
56
57static int tls_options = 0;/* Options for TLS connections */
58
59
2ece34a9
MS
60/*
61 * Local functions...
62 */
cc754834 63
2ece34a9
MS
64static _http_sspi_t *http_sspi_alloc(void);
65static int http_sspi_client(http_t *http, const char *hostname);
17eecbf1 66static PCCERT_CONTEXT http_sspi_create_credential(http_credential_t *cred);
e32497ad
MS
67static BOOL http_sspi_find_credentials(http_t *http, const LPWSTR containerName, const char *common_name);
68static void http_sspi_free(_http_sspi_t *sspi);
69static BOOL http_sspi_make_credentials(_http_sspi_t *sspi, const LPWSTR containerName, const char *common_name, _http_mode_t mode, int years);
2ece34a9 70static int http_sspi_server(http_t *http, const char *hostname);
e32497ad
MS
71static void http_sspi_set_allows_any_root(_http_sspi_t *sspi, BOOL allow);
72static void http_sspi_set_allows_expired_certs(_http_sspi_t *sspi, BOOL allow);
52770f63 73static const char *http_sspi_strerror(char *buffer, size_t bufsize, DWORD code);
2ece34a9 74static DWORD http_sspi_verify(PCCERT_CONTEXT cert, const char *common_name, DWORD dwCertFlags);
cc754834
MS
75
76
2c85b752 77/*
2ece34a9
MS
78 * 'cupsMakeServerCredentials()' - Make a self-signed certificate and private key pair.
79 *
e1f19878 80 * @since CUPS 2.0/OS 10.10@
2c85b752
MS
81 */
82
2ece34a9
MS
83int /* O - 1 on success, 0 on failure */
84cupsMakeServerCredentials(
85 const char *path, /* I - Keychain path or @code NULL@ for default */
86 const char *common_name, /* I - Common name */
87 int num_alt_names, /* I - Number of subject alternate names */
88 const char **alt_names, /* I - Subject Alternate Names */
89 time_t expiration_date) /* I - Expiration date */
2c85b752 90{
e32497ad
MS
91 _http_sspi_t *sspi; /* SSPI data */
92 int ret; /* Return value */
93
94
2ece34a9 95 DEBUG_printf(("cupsMakeServerCredentials(path=\"%s\", common_name=\"%s\", num_alt_names=%d, alt_names=%p, expiration_date=%d)", path, common_name, num_alt_names, alt_names, (int)expiration_date));
2c85b752 96
2ece34a9 97 (void)path;
2ece34a9
MS
98 (void)num_alt_names;
99 (void)alt_names;
2c85b752 100
e32497ad
MS
101 sspi = http_sspi_alloc();
102 ret = http_sspi_make_credentials(sspi, L"ServerContainer", common_name, _HTTP_MODE_SERVER, (int)((expiration_date - time(NULL) + 86399) / 86400 / 365));
103
104 http_sspi_free(sspi);
105
106 return (ret);
2ece34a9 107}
2c85b752 108
2c85b752 109
2ece34a9
MS
110/*
111 * 'cupsSetServerCredentials()' - Set the default server credentials.
112 *
113 * Note: The server credentials are used by all threads in the running process.
114 * This function is threadsafe.
115 *
e1f19878 116 * @since CUPS 2.0/OS 10.10@
2ece34a9 117 */
2c85b752 118
2ece34a9
MS
119int /* O - 1 on success, 0 on failure */
120cupsSetServerCredentials(
121 const char *path, /* I - Keychain path or @code NULL@ for default */
122 const char *common_name, /* I - Default common name for server */
123 int auto_create) /* I - 1 = automatically create self-signed certificates */
124{
125 DEBUG_printf(("cupsSetServerCredentials(path=\"%s\", common_name=\"%s\", auto_create=%d)", path, common_name, auto_create));
2c85b752 126
2ece34a9
MS
127 (void)path;
128 (void)common_name;
129 (void)auto_create;
2c85b752 130
2ece34a9 131 return (0);
2c85b752
MS
132}
133
134
2c85b752 135/*
2ece34a9
MS
136 * 'httpCopyCredentials()' - Copy the credentials associated with the peer in
137 * an encrypted connection.
138 *
139 * @since CUPS 1.5/OS X 10.7@
2c85b752
MS
140 */
141
2ece34a9
MS
142int /* O - Status of call (0 = success) */
143httpCopyCredentials(
144 http_t *http, /* I - Connection to server */
145 cups_array_t **credentials) /* O - Array of credentials */
2c85b752 146{
2ece34a9 147 DEBUG_printf(("httpCopyCredentials(http=%p, credentials=%p)", http, credentials));
2c85b752 148
9041146f
MS
149 if (!http || !http->tls || !http->tls->remoteCert || !credentials)
150 {
151 if (credentials)
152 *credentials = NULL;
153
154 return (-1);
155 }
2c85b752 156
9041146f
MS
157 *credentials = cupsArrayNew(NULL, NULL);
158 httpAddCredential(*credentials, http->tls->remoteCert->pbCertEncoded, http->tls->remoteCert->cbCertEncoded);
2c85b752 159
9041146f 160 return (0);
2ece34a9 161}
2c85b752 162
2c85b752 163
2ece34a9
MS
164/*
165 * '_httpCreateCredentials()' - Create credentials in the internal format.
166 */
2c85b752 167
2ece34a9
MS
168http_tls_credentials_t /* O - Internal credentials */
169_httpCreateCredentials(
170 cups_array_t *credentials) /* I - Array of credentials */
171{
17eecbf1 172 return (http_sspi_create_credential((http_credential_t *)cupsArrayFirst(credentials)));
2ece34a9 173}
2c85b752 174
2c85b752 175
2ece34a9
MS
176/*
177 * 'httpCredentialsAreValidForName()' - Return whether the credentials are valid for the given name.
178 *
e1f19878 179 * @since CUPS 2.0/OS 10.10@
2ece34a9 180 */
2c85b752 181
2ece34a9
MS
182int /* O - 1 if valid, 0 otherwise */
183httpCredentialsAreValidForName(
184 cups_array_t *credentials, /* I - Credentials */
185 const char *common_name) /* I - Name to check */
186{
17eecbf1
MS
187 int valid = 1; /* Valid name? */
188 PCCERT_CONTEXT cert = http_sspi_create_credential((http_credential_t *)cupsArrayFirst(credentials));
189 /* Certificate */
190 char cert_name[1024]; /* Name from certificate */
191
192
193 if (cert)
194 {
9041146f
MS
195 if (CertNameToStr(X509_ASN_ENCODING, &(cert->pCertInfo->Subject), CERT_SIMPLE_NAME_STR, cert_name, sizeof(cert_name)))
196 {
197 /*
198 * Extract common name at end...
199 */
200
201 char *ptr = strrchr(cert_name, ',');
202 if (ptr && ptr[1])
203 _cups_strcpy(cert_name, ptr + 2);
204 }
205 else
17eecbf1
MS
206 strlcpy(cert_name, "unknown", sizeof(cert_name));
207
208 CertFreeCertificateContext(cert);
209 }
210 else
211 strlcpy(cert_name, "unknown", sizeof(cert_name));
212
213 /*
214 * Compare the common names...
215 */
216
217 if (_cups_strcasecmp(common_name, cert_name))
218 {
219 /*
220 * Not an exact match for the common name, check for wildcard certs...
221 */
222
223 const char *domain = strchr(common_name, '.');
224 /* Domain in common name */
225
226 if (strncmp(cert_name, "*.", 2) || !domain || _cups_strcasecmp(domain, cert_name + 1))
227 {
228 /*
229 * Not a wildcard match.
230 */
231
232 /* TODO: Check subject alternate names */
233 valid = 0;
234 }
235 }
2c85b752 236
17eecbf1 237 return (valid);
2c85b752 238}
2c85b752
MS
239
240
2c85b752 241/*
2ece34a9
MS
242 * 'httpCredentialsGetTrust()' - Return the trust of credentials.
243 *
e1f19878 244 * @since CUPS 2.0/OS 10.10@
2c85b752
MS
245 */
246
2ece34a9
MS
247http_trust_t /* O - Level of trust */
248httpCredentialsGetTrust(
249 cups_array_t *credentials, /* I - Credentials */
250 const char *common_name) /* I - Common name for trust lookup */
2c85b752 251{
17eecbf1
MS
252 http_trust_t trust = HTTP_TRUST_OK; /* Level of trust */
253 PCCERT_CONTEXT cert = NULL; /* Certificate to validate */
254 DWORD certFlags = 0; /* Cert verification flags */
255 _cups_globals_t *cg = _cupsGlobals(); /* Per-thread global data */
2c85b752
MS
256
257
2ece34a9
MS
258 if (!common_name)
259 return (HTTP_TRUST_UNKNOWN);
2c85b752 260
17eecbf1
MS
261 cert = http_sspi_create_credential((http_credential_t *)cupsArrayFirst(credentials));
262 if (!cert)
263 return (HTTP_TRUST_UNKNOWN);
2c85b752 264
3abb875b
MS
265 if (cg->any_root < 0)
266 _cupsSetDefaults();
267
17eecbf1
MS
268 if (cg->any_root)
269 certFlags |= SECURITY_FLAG_IGNORE_UNKNOWN_CA;
2c85b752 270
17eecbf1
MS
271 if (cg->expired_certs)
272 certFlags |= SECURITY_FLAG_IGNORE_CERT_DATE_INVALID;
2c85b752 273
17eecbf1
MS
274 if (!cg->validate_certs)
275 certFlags |= SECURITY_FLAG_IGNORE_CERT_CN_INVALID;
2c85b752 276
17eecbf1 277 if (http_sspi_verify(cert, common_name, certFlags) != SEC_E_OK)
2ece34a9 278 trust = HTTP_TRUST_INVALID;
2c85b752 279
17eecbf1 280 CertFreeCertificateContext(cert);
2c85b752 281
2ece34a9
MS
282 return (trust);
283}
2c85b752 284
2c85b752 285
2ece34a9
MS
286/*
287 * 'httpCredentialsGetExpiration()' - Return the expiration date of the credentials.
288 *
e1f19878 289 * @since CUPS 2.0/OS 10.10@
2ece34a9 290 */
2c85b752 291
2ece34a9
MS
292time_t /* O - Expiration date of credentials */
293httpCredentialsGetExpiration(
294 cups_array_t *credentials) /* I - Credentials */
295{
17eecbf1
MS
296 time_t expiration_date = 0; /* Expiration data of credentials */
297 PCCERT_CONTEXT cert = http_sspi_create_credential((http_credential_t *)cupsArrayFirst(credentials));
298 /* Certificate */
2c85b752 299
17eecbf1
MS
300 if (cert)
301 {
302 SYSTEMTIME systime; /* System time */
303 struct tm tm; /* UNIX date/time */
304
9041146f 305 FileTimeToSystemTime(&(cert->pCertInfo->NotAfter), &systime);
17eecbf1
MS
306
307 tm.tm_year = systime.wYear - 1900;
308 tm.tm_mon = systime.wMonth - 1;
309 tm.tm_mday = systime.wDay;
310 tm.tm_hour = systime.wHour;
311 tm.tm_min = systime.wMinute;
312 tm.tm_sec = systime.wSecond;
313
314 expiration_date = mktime(&tm);
315
316 CertFreeCertificateContext(cert);
317 }
318
319 return (expiration_date);
2ece34a9 320}
2c85b752 321
2c85b752 322
2ece34a9
MS
323/*
324 * 'httpCredentialsString()' - Return a string representing the credentials.
325 *
e1f19878 326 * @since CUPS 2.0/OS 10.10@
2ece34a9 327 */
2c85b752 328
2ece34a9
MS
329size_t /* O - Total size of credentials string */
330httpCredentialsString(
331 cups_array_t *credentials, /* I - Credentials */
332 char *buffer, /* I - Buffer or @code NULL@ */
333 size_t bufsize) /* I - Size of buffer */
334{
17eecbf1
MS
335 http_credential_t *first = (http_credential_t *)cupsArrayFirst(credentials);
336 /* First certificate */
337 PCCERT_CONTEXT cert; /* Certificate */
338
339
2ece34a9 340 DEBUG_printf(("httpCredentialsString(credentials=%p, buffer=%p, bufsize=" CUPS_LLFMT ")", credentials, buffer, CUPS_LLCAST bufsize));
2c85b752 341
2ece34a9
MS
342 if (!buffer)
343 return (0);
2c85b752 344
2ece34a9
MS
345 if (buffer && bufsize > 0)
346 *buffer = '\0';
2c85b752 347
17eecbf1 348 cert = http_sspi_create_credential(first);
2c85b752 349
17eecbf1 350 if (cert)
2c85b752 351 {
17eecbf1
MS
352 char cert_name[256]; /* Common name */
353 SYSTEMTIME systime; /* System time */
354 struct tm tm; /* UNIX date/time */
2ece34a9
MS
355 time_t expiration; /* Expiration date of cert */
356 _cups_md5_state_t md5_state; /* MD5 state */
357 unsigned char md5_digest[16]; /* MD5 result */
2c85b752 358
9041146f 359 FileTimeToSystemTime(&(cert->pCertInfo->NotAfter), &systime);
17eecbf1
MS
360
361 tm.tm_year = systime.wYear - 1900;
362 tm.tm_mon = systime.wMonth - 1;
363 tm.tm_mday = systime.wDay;
364 tm.tm_hour = systime.wHour;
365 tm.tm_min = systime.wMinute;
366 tm.tm_sec = systime.wSecond;
367
368 expiration = mktime(&tm);
2c85b752 369
9041146f
MS
370 if (CertNameToStr(X509_ASN_ENCODING, &(cert->pCertInfo->Subject), CERT_SIMPLE_NAME_STR, cert_name, sizeof(cert_name)))
371 {
372 /*
373 * Extract common name at end...
374 */
375
376 char *ptr = strrchr(cert_name, ',');
377 if (ptr && ptr[1])
378 _cups_strcpy(cert_name, ptr + 2);
379 }
380 else
17eecbf1 381 strlcpy(cert_name, "unknown", sizeof(cert_name));
2c85b752 382
2ece34a9
MS
383 _cupsMD5Init(&md5_state);
384 _cupsMD5Append(&md5_state, first->data, (int)first->datalen);
385 _cupsMD5Finish(&md5_state, md5_digest);
2c85b752 386
17eecbf1 387 snprintf(buffer, bufsize, "%s / %s / %02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X", cert_name, httpGetDateString(expiration), md5_digest[0], md5_digest[1], md5_digest[2], md5_digest[3], md5_digest[4], md5_digest[5], md5_digest[6], md5_digest[7], md5_digest[8], md5_digest[9], md5_digest[10], md5_digest[11], md5_digest[12], md5_digest[13], md5_digest[14], md5_digest[15]);
2c85b752 388
17eecbf1 389 CertFreeCertificateContext(cert);
2c85b752
MS
390 }
391
2ece34a9 392 DEBUG_printf(("1httpCredentialsString: Returning \"%s\".", buffer));
2c85b752 393
2ece34a9
MS
394 return (strlen(buffer));
395}
2c85b752 396
2c85b752 397
2ece34a9
MS
398/*
399 * '_httpFreeCredentials()' - Free internal credentials.
400 */
2c85b752 401
2ece34a9
MS
402void
403_httpFreeCredentials(
404 http_tls_credentials_t credentials) /* I - Internal credentials */
405{
406 if (!credentials)
407 return;
17eecbf1
MS
408
409 CertFreeCertificateContext(credentials);
2ece34a9 410}
2c85b752 411
2c85b752 412
2ece34a9
MS
413/*
414 * 'httpLoadCredentials()' - Load X.509 credentials from a keychain file.
415 *
e1f19878 416 * @since CUPS 2.0/OS 10.10@
2ece34a9 417 */
2c85b752 418
2ece34a9
MS
419int /* O - 0 on success, -1 on error */
420httpLoadCredentials(
421 const char *path, /* I - Keychain path or @code NULL@ for default */
422 cups_array_t **credentials, /* IO - Credentials */
423 const char *common_name) /* I - Common name for credentials */
424{
52770f63
MS
425 HCERTSTORE store = NULL; /* Certificate store */
426 PCCERT_CONTEXT storedContext = NULL; /* Context created from the store */
427 DWORD dwSize = 0; /* 32 bit size */
428 PBYTE p = NULL; /* Temporary storage */
429 HCRYPTPROV hProv = (HCRYPTPROV)NULL;
430 /* Handle to a CSP */
431 CERT_NAME_BLOB sib; /* Arbitrary array of bytes */
432#ifdef DEBUG
433 char error[1024]; /* Error message buffer */
434#endif /* DEBUG */
435
436
2ece34a9 437 DEBUG_printf(("httpLoadCredentials(path=\"%s\", credentials=%p, common_name=\"%s\")", path, credentials, common_name));
2c85b752 438
2ece34a9 439 (void)path;
2c85b752 440
2ece34a9 441 if (credentials)
52770f63 442 {
2ece34a9 443 *credentials = NULL;
52770f63
MS
444 }
445 else
446 {
447 DEBUG_puts("1httpLoadCredentials: NULL credentials pointer, returning -1.");
448 return (-1);
449 }
450
451 if (!common_name)
452 {
453 DEBUG_puts("1httpLoadCredentials: Bad common name, returning -1.");
454 return (-1);
455 }
456
457 if (!CryptAcquireContextW(&hProv, L"RememberedContainer", MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_NEWKEYSET | CRYPT_MACHINE_KEYSET))
458 {
459 if (GetLastError() == NTE_EXISTS)
460 {
461 if (!CryptAcquireContextW(&hProv, L"RememberedContainer", MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_MACHINE_KEYSET))
462 {
463 DEBUG_printf(("1httpLoadCredentials: CryptAcquireContext failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
464 goto cleanup;
465 }
466 }
467 }
468
469 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, X509_ASN_ENCODING|PKCS_7_ASN_ENCODING, hProv, CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_STORE_NO_CRYPT_RELEASE_FLAG | CERT_STORE_OPEN_EXISTING_FLAG, L"MY");
470
471 if (!store)
472 {
473 DEBUG_printf(("1httpLoadCredentials: CertOpenSystemStore failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
474 goto cleanup;
475 }
476
477 dwSize = 0;
478
479 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, NULL, &dwSize, NULL))
480 {
481 DEBUG_printf(("1httpLoadCredentials: CertStrToName failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
482 goto cleanup;
483 }
484
485 p = (PBYTE)malloc(dwSize);
486
487 if (!p)
488 {
489 DEBUG_printf(("1httpLoadCredentials: malloc failed for %d bytes.", dwSize));
490 goto cleanup;
491 }
492
493 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, p, &dwSize, NULL))
494 {
495 DEBUG_printf(("1httpLoadCredentials: CertStrToName failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
496 goto cleanup;
497 }
498
499 sib.cbData = dwSize;
500 sib.pbData = p;
501
502 storedContext = CertFindCertificateInStore(store, X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, 0, CERT_FIND_SUBJECT_NAME, &sib, NULL);
503
504 if (!storedContext)
505 {
506 DEBUG_printf(("1httpLoadCredentials: Unable to find credentials for \"%s\".", common_name));
507 goto cleanup;
508 }
509
510 *credentials = cupsArrayNew(NULL, NULL);
511 httpAddCredential(*credentials, storedContext->pbCertEncoded, storedContext->cbCertEncoded);
512
513cleanup:
514
515 /*
516 * Cleanup
517 */
518
519 if (storedContext)
520 CertFreeCertificateContext(storedContext);
521
522 if (p)
523 free(p);
524
525 if (store)
526 CertCloseStore(store, 0);
527
528 if (hProv)
529 CryptReleaseContext(hProv, 0);
2c85b752 530
2ece34a9 531 DEBUG_printf(("1httpLoadCredentials: Returning %d.", *credentials ? 0 : -1));
2c85b752 532
2ece34a9
MS
533 return (*credentials ? 0 : -1);
534}
2c85b752 535
2c85b752 536
2ece34a9
MS
537/*
538 * 'httpSaveCredentials()' - Save X.509 credentials to a keychain file.
539 *
e1f19878 540 * @since CUPS 2.0/OS 10.10@
2ece34a9 541 */
2c85b752 542
2ece34a9
MS
543int /* O - -1 on error, 0 on success */
544httpSaveCredentials(
545 const char *path, /* I - Keychain path or @code NULL@ for default */
546 cups_array_t *credentials, /* I - Credentials */
547 const char *common_name) /* I - Common name for credentials */
548{
52770f63
MS
549 HCERTSTORE store = NULL; /* Certificate store */
550 PCCERT_CONTEXT storedContext = NULL; /* Context created from the store */
551 PCCERT_CONTEXT createdContext = NULL; /* Context created by us */
552 DWORD dwSize = 0; /* 32 bit size */
553 PBYTE p = NULL; /* Temporary storage */
554 HCRYPTPROV hProv = (HCRYPTPROV)NULL;
555 /* Handle to a CSP */
74ac5fe7 556 CRYPT_KEY_PROV_INFO ckp; /* Handle to crypto key */
52770f63
MS
557 int ret = -1; /* Return value */
558#ifdef DEBUG
559 char error[1024]; /* Error message buffer */
560#endif /* DEBUG */
561
562
2ece34a9 563 DEBUG_printf(("httpSaveCredentials(path=\"%s\", credentials=%p, common_name=\"%s\")", path, credentials, common_name));
2c85b752 564
2ece34a9 565 (void)path;
2c85b752 566
52770f63
MS
567 if (!common_name)
568 {
569 DEBUG_puts("1httpSaveCredentials: Bad common name, returning -1.");
570 return (-1);
571 }
572
74ac5fe7 573 createdContext = http_sspi_create_credential((http_credential_t *)cupsArrayFirst(credentials));
52770f63
MS
574 if (!createdContext)
575 {
576 DEBUG_puts("1httpSaveCredentials: Bad credentials, returning -1.");
577 return (-1);
578 }
579
580 if (!CryptAcquireContextW(&hProv, L"RememberedContainer", MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_NEWKEYSET | CRYPT_MACHINE_KEYSET))
581 {
582 if (GetLastError() == NTE_EXISTS)
583 {
584 if (!CryptAcquireContextW(&hProv, L"RememberedContainer", MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_MACHINE_KEYSET))
585 {
586 DEBUG_printf(("1httpSaveCredentials: CryptAcquireContext failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
587 goto cleanup;
588 }
589 }
590 }
591
592 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, X509_ASN_ENCODING|PKCS_7_ASN_ENCODING, hProv, CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_STORE_NO_CRYPT_RELEASE_FLAG | CERT_STORE_OPEN_EXISTING_FLAG, L"MY");
593
594 if (!store)
595 {
596 DEBUG_printf(("1httpSaveCredentials: CertOpenSystemStore failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
597 goto cleanup;
598 }
599
600 dwSize = 0;
601
602 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, NULL, &dwSize, NULL))
603 {
604 DEBUG_printf(("1httpSaveCredentials: CertStrToName failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
605 goto cleanup;
606 }
607
608 p = (PBYTE)malloc(dwSize);
609
610 if (!p)
611 {
612 DEBUG_printf(("1httpSaveCredentials: malloc failed for %d bytes.", dwSize));
613 goto cleanup;
614 }
615
616 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, p, &dwSize, NULL))
617 {
618 DEBUG_printf(("1httpSaveCredentials: CertStrToName failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
619 goto cleanup;
620 }
621
622 /*
623 * Add the created context to the named store, and associate it with the named
624 * container...
625 */
626
627 if (!CertAddCertificateContextToStore(store, createdContext, CERT_STORE_ADD_REPLACE_EXISTING, &storedContext))
628 {
629 DEBUG_printf(("1httpSaveCredentials: CertAddCertificateContextToStore failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
630 goto cleanup;
631 }
632
633 ZeroMemory(&ckp, sizeof(ckp));
634 ckp.pwszContainerName = L"RememberedContainer";
635 ckp.pwszProvName = MS_DEF_PROV_W;
636 ckp.dwProvType = PROV_RSA_FULL;
637 ckp.dwFlags = CRYPT_MACHINE_KEYSET;
638 ckp.dwKeySpec = AT_KEYEXCHANGE;
639
640 if (!CertSetCertificateContextProperty(storedContext, CERT_KEY_PROV_INFO_PROP_ID, 0, &ckp))
641 {
642 DEBUG_printf(("1httpSaveCredentials: CertSetCertificateContextProperty failed: %s", http_sspi_strerror(error, sizeof(error), GetLastError())));
643 goto cleanup;
644 }
645
646 ret = 0;
647
648cleanup:
649
650 /*
651 * Cleanup
652 */
653
654 if (createdContext)
655 CertFreeCertificateContext(createdContext);
656
657 if (storedContext)
658 CertFreeCertificateContext(storedContext);
659
660 if (p)
661 free(p);
662
663 if (store)
664 CertCloseStore(store, 0);
665
666 if (hProv)
667 CryptReleaseContext(hProv, 0);
668
669 DEBUG_printf(("1httpSaveCredentials: Returning %d.", ret));
670 return (ret);
2c85b752
MS
671}
672
673
674/*
2ece34a9 675 * '_httpTLSInitialize()' - Initialize the TLS stack.
2c85b752
MS
676 */
677
2ece34a9
MS
678void
679_httpTLSInitialize(void)
2c85b752 680{
2ece34a9
MS
681 /*
682 * Nothing to do...
683 */
684}
2c85b752 685
2c85b752 686
2ece34a9
MS
687/*
688 * '_httpTLSPending()' - Return the number of pending TLS-encrypted bytes.
689 */
2c85b752 690
2ece34a9
MS
691size_t /* O - Bytes available */
692_httpTLSPending(http_t *http) /* I - HTTP connection */
693{
694 if (http->tls)
695 return (http->tls->readBufferUsed);
696 else
697 return (0);
698}
2c85b752 699
2c85b752 700
2ece34a9
MS
701/*
702 * '_httpTLSRead()' - Read from a SSL/TLS connection.
703 */
2c85b752 704
2ece34a9
MS
705int /* O - Bytes read */
706_httpTLSRead(http_t *http, /* I - HTTP connection */
707 char *buf, /* I - Buffer to store data */
708 int len) /* I - Length of buffer */
709{
710 int i; /* Looping var */
e32497ad 711 _http_sspi_t *sspi = http->tls; /* SSPI data */
2ece34a9
MS
712 SecBufferDesc message; /* Array of SecBuffer struct */
713 SecBuffer buffers[4] = { 0 }; /* Security package buffer */
714 int num = 0; /* Return value */
715 PSecBuffer pDataBuffer; /* Data buffer */
716 PSecBuffer pExtraBuffer; /* Excess data buffer */
717 SECURITY_STATUS scRet; /* SSPI status */
2c85b752 718
2c85b752 719
195c1f2d
MS
720 DEBUG_printf(("4_httpTLSRead(http=%p, buf=%p, len=%d)", http, buf, len));
721
2ece34a9
MS
722 /*
723 * If there are bytes that have already been decrypted and have not yet been
724 * read, return those...
725 */
2c85b752 726
e32497ad 727 if (sspi->readBufferUsed > 0)
2ece34a9 728 {
e32497ad 729 int bytesToCopy = min(sspi->readBufferUsed, len);
2ece34a9 730 /* Number of bytes to copy */
2c85b752 731
e32497ad
MS
732 memcpy(buf, sspi->readBuffer, bytesToCopy);
733 sspi->readBufferUsed -= bytesToCopy;
2c85b752 734
e32497ad
MS
735 if (sspi->readBufferUsed > 0)
736 memmove(sspi->readBuffer, sspi->readBuffer + bytesToCopy, sspi->readBufferUsed);
2c85b752 737
195c1f2d
MS
738 DEBUG_printf(("5_httpTLSRead: Returning %d bytes previously decrypted.", bytesToCopy));
739
2ece34a9
MS
740 return (bytesToCopy);
741 }
2c85b752 742
2ece34a9
MS
743 /*
744 * Initialize security buffer structs
745 */
2c85b752 746
2ece34a9 747 message.ulVersion = SECBUFFER_VERSION;
195c1f2d
MS
748 message.cBuffers = 4;
749 message.pBuffers = buffers;
2c85b752 750
2ece34a9 751 do
2c85b752
MS
752 {
753 /*
2ece34a9 754 * If there is not enough space in the buffer, then increase its size...
2c85b752
MS
755 */
756
e32497ad 757 if (sspi->decryptBufferLength <= sspi->decryptBufferUsed)
2c85b752 758 {
2ece34a9 759 BYTE *temp; /* New buffer */
2c85b752 760
e32497ad 761 if (sspi->decryptBufferLength >= 262144)
2ece34a9
MS
762 {
763 WSASetLastError(E_OUTOFMEMORY);
764 DEBUG_puts("_httpTLSRead: Decryption buffer too large (>256k)");
765 return (-1);
766 }
2c85b752 767
e32497ad 768 if ((temp = realloc(sspi->decryptBuffer, sspi->decryptBufferLength + 4096)) == NULL)
2ece34a9 769 {
e32497ad 770 DEBUG_printf(("_httpTLSRead: Unable to allocate %d byte decryption buffer.", sspi->decryptBufferLength + 4096));
2ece34a9
MS
771 WSASetLastError(E_OUTOFMEMORY);
772 return (-1);
773 }
2c85b752 774
e32497ad
MS
775 sspi->decryptBufferLength += 4096;
776 sspi->decryptBuffer = temp;
195c1f2d 777
e32497ad 778 DEBUG_printf(("_httpTLSRead: Resized decryption buffer to %d bytes.", sspi->decryptBufferLength));
2ece34a9 779 }
2c85b752 780
e32497ad
MS
781 buffers[0].pvBuffer = sspi->decryptBuffer;
782 buffers[0].cbBuffer = (unsigned long)sspi->decryptBufferUsed;
2ece34a9
MS
783 buffers[0].BufferType = SECBUFFER_DATA;
784 buffers[1].BufferType = SECBUFFER_EMPTY;
785 buffers[2].BufferType = SECBUFFER_EMPTY;
786 buffers[3].BufferType = SECBUFFER_EMPTY;
2c85b752 787
e32497ad 788 DEBUG_printf(("5_httpTLSRead: decryptBufferUsed=%d", sspi->decryptBufferUsed));
195c1f2d 789
e32497ad 790 scRet = DecryptMessage(&sspi->context, &message, 0, NULL);
2c85b752 791
2ece34a9
MS
792 if (scRet == SEC_E_INCOMPLETE_MESSAGE)
793 {
e32497ad 794 num = recv(http->fd, sspi->decryptBuffer + sspi->decryptBufferUsed, (int)(sspi->decryptBufferLength - sspi->decryptBufferUsed), 0);
2ece34a9
MS
795 if (num < 0)
796 {
195c1f2d 797 DEBUG_printf(("5_httpTLSRead: recv failed: %d", WSAGetLastError()));
2ece34a9
MS
798 return (-1);
799 }
800 else if (num == 0)
801 {
195c1f2d 802 DEBUG_puts("5_httpTLSRead: Server disconnected.");
2ece34a9
MS
803 return (0);
804 }
2c85b752 805
195c1f2d
MS
806 DEBUG_printf(("5_httpTLSRead: Read %d bytes into decryption buffer.", num));
807
e32497ad 808 sspi->decryptBufferUsed += num;
2ece34a9 809 }
2c85b752 810 }
2ece34a9 811 while (scRet == SEC_E_INCOMPLETE_MESSAGE);
2c85b752 812
2ece34a9
MS
813 if (scRet == SEC_I_CONTEXT_EXPIRED)
814 {
195c1f2d 815 DEBUG_puts("5_httpTLSRead: Context expired.");
2ece34a9
MS
816 WSASetLastError(WSAECONNRESET);
817 return (-1);
818 }
819 else if (scRet != SEC_E_OK)
820 {
52770f63 821 DEBUG_printf(("5_httpTLSRead: DecryptMessage failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9
MS
822 WSASetLastError(WSASYSCALLFAILURE);
823 return (-1);
824 }
2c85b752 825
2ece34a9
MS
826 /*
827 * The decryption worked. Now, locate data buffer.
828 */
2c85b752 829
2ece34a9
MS
830 pDataBuffer = NULL;
831 pExtraBuffer = NULL;
2c85b752 832
2ece34a9
MS
833 for (i = 1; i < 4; i++)
834 {
835 if (buffers[i].BufferType == SECBUFFER_DATA)
836 pDataBuffer = &buffers[i];
837 else if (!pExtraBuffer && (buffers[i].BufferType == SECBUFFER_EXTRA))
838 pExtraBuffer = &buffers[i];
839 }
2c85b752 840
2ece34a9
MS
841 /*
842 * If a data buffer is found, then copy the decrypted bytes to the passed-in
843 * buffer...
844 */
2c85b752 845
2ece34a9
MS
846 if (pDataBuffer)
847 {
33c9220c 848 int bytesToCopy = min((int)pDataBuffer->cbBuffer, len);
2ece34a9
MS
849 /* Number of bytes to copy into buf */
850 int bytesToSave = pDataBuffer->cbBuffer - bytesToCopy;
851 /* Number of bytes to save in our read buffer */
2c85b752 852
2ece34a9
MS
853 if (bytesToCopy)
854 memcpy(buf, pDataBuffer->pvBuffer, bytesToCopy);
2c85b752 855
2c85b752 856 /*
2ece34a9
MS
857 * If there are more decrypted bytes than can be copied to the passed in
858 * buffer, then save them...
2c85b752
MS
859 */
860
2ece34a9 861 if (bytesToSave)
2c85b752 862 {
e32497ad 863 if ((sspi->readBufferLength - sspi->readBufferUsed) < bytesToSave)
2ece34a9
MS
864 {
865 BYTE *temp; /* New buffer pointer */
2c85b752 866
e32497ad 867 if ((temp = realloc(sspi->readBuffer, sspi->readBufferUsed + bytesToSave)) == NULL)
2ece34a9 868 {
e32497ad 869 DEBUG_printf(("_httpTLSRead: Unable to allocate %d bytes.", sspi->readBufferUsed + bytesToSave));
2ece34a9
MS
870 WSASetLastError(E_OUTOFMEMORY);
871 return (-1);
872 }
2c85b752 873
e32497ad
MS
874 sspi->readBufferLength = sspi->readBufferUsed + bytesToSave;
875 sspi->readBuffer = temp;
2ece34a9 876 }
2c85b752 877
e32497ad 878 memcpy(((BYTE *)sspi->readBuffer) + sspi->readBufferUsed, ((BYTE *)pDataBuffer->pvBuffer) + bytesToCopy, bytesToSave);
2c85b752 879
e32497ad 880 sspi->readBufferUsed += bytesToSave;
2ece34a9 881 }
2c85b752 882
adad9dd6 883 num = bytesToCopy;
2ece34a9
MS
884 }
885 else
886 {
33c9220c 887 DEBUG_puts("_httpTLSRead: Unable to find data buffer.");
2ece34a9
MS
888 WSASetLastError(WSASYSCALLFAILURE);
889 return (-1);
890 }
2c85b752 891
2ece34a9
MS
892 /*
893 * If the decryption process left extra bytes, then save those back in
894 * decryptBuffer. They will be processed the next time through the loop.
895 */
2c85b752 896
2ece34a9 897 if (pExtraBuffer)
2c85b752 898 {
e32497ad
MS
899 memmove(sspi->decryptBuffer, pExtraBuffer->pvBuffer, pExtraBuffer->cbBuffer);
900 sspi->decryptBufferUsed = pExtraBuffer->cbBuffer;
2ece34a9
MS
901 }
902 else
903 {
e32497ad 904 sspi->decryptBufferUsed = 0;
2c85b752 905 }
2c85b752 906
adad9dd6 907 return (num);
2c85b752 908}
2c85b752
MS
909
910
63aefcd5
MS
911/*
912 * '_httpTLSSetOptions()' - Set TLS protocol and cipher suite options.
913 */
914
915void
916_httpTLSSetOptions(int options) /* I - Options */
917{
918 tls_options = options;
919}
920
921
cc754834 922/*
2ece34a9 923 * '_httpTLSStart()' - Set up SSL/TLS support on a connection.
cc754834 924 */
2ece34a9
MS
925
926int /* O - 0 on success, -1 on failure */
927_httpTLSStart(http_t *http) /* I - HTTP connection */
cc754834 928{
2ece34a9
MS
929 char hostname[256], /* Hostname */
930 *hostptr; /* Pointer into hostname */
cc754834 931
cc754834 932
2ece34a9
MS
933 DEBUG_printf(("7_httpTLSStart(http=%p)", http));
934
935 if ((http->tls = http_sspi_alloc()) == NULL)
936 return (-1);
cc754834 937
2ece34a9
MS
938 if (http->mode == _HTTP_MODE_CLIENT)
939 {
940 /*
941 * Client: determine hostname...
942 */
cc754834 943
2ece34a9
MS
944 if (httpAddrLocalhost(http->hostaddr))
945 {
946 strlcpy(hostname, "localhost", sizeof(hostname));
947 }
948 else
949 {
950 /*
951 * Otherwise make sure the hostname we have does not end in a trailing dot.
952 */
cc754834 953
2ece34a9
MS
954 strlcpy(hostname, http->hostname, sizeof(hostname));
955 if ((hostptr = hostname + strlen(hostname) - 1) >= hostname &&
956 *hostptr == '.')
957 *hostptr = '\0';
958 }
cc754834 959
2ece34a9
MS
960 return (http_sspi_client(http, hostname));
961 }
962 else
cc754834 963 {
2ece34a9
MS
964 /*
965 * Server: determine hostname to use...
966 */
967
968 if (http->fields[HTTP_FIELD_HOST][0])
cc754834 969 {
2ece34a9
MS
970 /*
971 * Use hostname for TLS upgrade...
972 */
973
974 strlcpy(hostname, http->fields[HTTP_FIELD_HOST], sizeof(hostname));
975 }
976 else
977 {
978 /*
979 * Resolve hostname from connection address...
980 */
981
982 http_addr_t addr; /* Connection address */
983 socklen_t addrlen; /* Length of address */
984
985 addrlen = sizeof(addr);
986 if (getsockname(http->fd, (struct sockaddr *)&addr, &addrlen))
cc754834 987 {
2ece34a9
MS
988 DEBUG_printf(("4_httpTLSStart: Unable to get socket address: %s", strerror(errno)));
989 hostname[0] = '\0';
990 }
991 else if (httpAddrLocalhost(&addr))
992 hostname[0] = '\0';
993 else
994 {
995 httpAddrLookup(&addr, hostname, sizeof(hostname));
996 DEBUG_printf(("4_httpTLSStart: Resolved socket address to \"%s\".", hostname));
cc754834
MS
997 }
998 }
cc754834 999
2ece34a9 1000 return (http_sspi_server(http, hostname));
cc754834 1001 }
2ece34a9 1002}
cc754834 1003
321d8d57 1004
2ece34a9
MS
1005/*
1006 * '_httpTLSStop()' - Shut down SSL/TLS on a connection.
1007 */
cc754834 1008
2ece34a9
MS
1009void
1010_httpTLSStop(http_t *http) /* I - HTTP connection */
1011{
e32497ad 1012 _http_sspi_t *sspi = http->tls; /* SSPI data */
cc754834 1013
cc754834 1014
e32497ad 1015 if (sspi->contextInitialized && http->fd >= 0)
cc754834 1016 {
2ece34a9
MS
1017 SecBufferDesc message; /* Array of SecBuffer struct */
1018 SecBuffer buffers[1] = { 0 };
1019 /* Security package buffer */
1020 DWORD dwType; /* Type */
1021 DWORD status; /* Status */
cc754834 1022
2ece34a9
MS
1023 /*
1024 * Notify schannel that we are about to close the connection.
1025 */
cc754834 1026
2ece34a9 1027 dwType = SCHANNEL_SHUTDOWN;
321d8d57 1028
2ece34a9
MS
1029 buffers[0].pvBuffer = &dwType;
1030 buffers[0].BufferType = SECBUFFER_TOKEN;
1031 buffers[0].cbBuffer = sizeof(dwType);
cc754834 1032
2ece34a9
MS
1033 message.cBuffers = 1;
1034 message.pBuffers = buffers;
1035 message.ulVersion = SECBUFFER_VERSION;
cc754834 1036
e32497ad 1037 status = ApplyControlToken(&sspi->context, &message);
cc754834 1038
2ece34a9
MS
1039 if (SUCCEEDED(status))
1040 {
1041 PBYTE pbMessage; /* Message buffer */
1042 DWORD cbMessage; /* Message buffer count */
1043 DWORD cbData; /* Data count */
1044 DWORD dwSSPIFlags; /* SSL attributes we requested */
1045 DWORD dwSSPIOutFlags; /* SSL attributes we received */
1046 TimeStamp tsExpiry; /* Time stamp */
cc754834 1047
2ece34a9
MS
1048 dwSSPIFlags = ASC_REQ_SEQUENCE_DETECT |
1049 ASC_REQ_REPLAY_DETECT |
1050 ASC_REQ_CONFIDENTIALITY |
1051 ASC_REQ_EXTENDED_ERROR |
1052 ASC_REQ_ALLOCATE_MEMORY |
1053 ASC_REQ_STREAM;
321d8d57 1054
2ece34a9
MS
1055 buffers[0].pvBuffer = NULL;
1056 buffers[0].BufferType = SECBUFFER_TOKEN;
1057 buffers[0].cbBuffer = 0;
cc754834 1058
2ece34a9
MS
1059 message.cBuffers = 1;
1060 message.pBuffers = buffers;
1061 message.ulVersion = SECBUFFER_VERSION;
1062
e32497ad 1063 status = AcceptSecurityContext(&sspi->creds, &sspi->context, NULL,
2ece34a9
MS
1064 dwSSPIFlags, SECURITY_NATIVE_DREP, NULL,
1065 &message, &dwSSPIOutFlags, &tsExpiry);
1066
1067 if (SUCCEEDED(status))
1068 {
1069 pbMessage = buffers[0].pvBuffer;
1070 cbMessage = buffers[0].cbBuffer;
1071
1072 /*
1073 * Send the close notify message to the client.
1074 */
1075
1076 if (pbMessage && cbMessage)
1077 {
1078 cbData = send(http->fd, pbMessage, cbMessage, 0);
1079 if ((cbData == SOCKET_ERROR) || (cbData == 0))
1080 {
1081 status = WSAGetLastError();
1082 DEBUG_printf(("_httpTLSStop: sending close notify failed: %d", status));
1083 }
1084 else
1085 {
1086 FreeContextBuffer(pbMessage);
1087 }
1088 }
1089 }
1090 else
1091 {
52770f63 1092 DEBUG_printf(("_httpTLSStop: AcceptSecurityContext failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), status)));
2ece34a9
MS
1093 }
1094 }
1095 else
1096 {
52770f63 1097 DEBUG_printf(("_httpTLSStop: ApplyControlToken failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), status)));
cc754834 1098 }
2ece34a9 1099 }
cc754834 1100
e32497ad 1101 http_sspi_free(sspi);
cc754834 1102
2ece34a9
MS
1103 http->tls = NULL;
1104}
1105
1106
1107/*
1108 * '_httpTLSWrite()' - Write to a SSL/TLS connection.
1109 */
1110
1111int /* O - Bytes written */
1112_httpTLSWrite(http_t *http, /* I - HTTP connection */
1113 const char *buf, /* I - Buffer holding data */
1114 int len) /* I - Length of buffer */
1115{
e32497ad 1116 _http_sspi_t *sspi = http->tls; /* SSPI data */
2ece34a9
MS
1117 SecBufferDesc message; /* Array of SecBuffer struct */
1118 SecBuffer buffers[4] = { 0 }; /* Security package buffer */
1119 int bufferLen; /* Buffer length */
1120 int bytesLeft; /* Bytes left to write */
1121 const char *bufptr; /* Pointer into buffer */
1122 int num = 0; /* Return value */
1123
1124
e32497ad 1125 bufferLen = sspi->streamSizes.cbMaximumMessage + sspi->streamSizes.cbHeader + sspi->streamSizes.cbTrailer;
2ece34a9 1126
e32497ad 1127 if (bufferLen > sspi->writeBufferLength)
2ece34a9
MS
1128 {
1129 BYTE *temp; /* New buffer pointer */
1130
e32497ad 1131 if ((temp = (BYTE *)realloc(sspi->writeBuffer, bufferLen)) == NULL)
cc754834 1132 {
2ece34a9
MS
1133 DEBUG_printf(("_httpTLSWrite: Unable to allocate buffer of %d bytes.", bufferLen));
1134 WSASetLastError(E_OUTOFMEMORY);
1135 return (-1);
cc754834 1136 }
2ece34a9 1137
e32497ad
MS
1138 sspi->writeBuffer = temp;
1139 sspi->writeBufferLength = bufferLen;
cc754834
MS
1140 }
1141
2ece34a9
MS
1142 bytesLeft = len;
1143 bufptr = buf;
cc754834 1144
2ece34a9
MS
1145 while (bytesLeft)
1146 {
e32497ad 1147 int chunk = min((int)sspi->streamSizes.cbMaximumMessage, bytesLeft);
2ece34a9
MS
1148 /* Size of data to write */
1149 SECURITY_STATUS scRet; /* SSPI status */
1150
1151 /*
1152 * Copy user data into the buffer, starting just past the header...
1153 */
1154
e32497ad 1155 memcpy(sspi->writeBuffer + sspi->streamSizes.cbHeader, bufptr, chunk);
2ece34a9
MS
1156
1157 /*
1158 * Setup the SSPI buffers
1159 */
1160
1161 message.ulVersion = SECBUFFER_VERSION;
1162 message.cBuffers = 4;
1163 message.pBuffers = buffers;
1164
e32497ad
MS
1165 buffers[0].pvBuffer = sspi->writeBuffer;
1166 buffers[0].cbBuffer = sspi->streamSizes.cbHeader;
2ece34a9 1167 buffers[0].BufferType = SECBUFFER_STREAM_HEADER;
e32497ad 1168 buffers[1].pvBuffer = sspi->writeBuffer + sspi->streamSizes.cbHeader;
2ece34a9
MS
1169 buffers[1].cbBuffer = (unsigned long) chunk;
1170 buffers[1].BufferType = SECBUFFER_DATA;
e32497ad
MS
1171 buffers[2].pvBuffer = sspi->writeBuffer + sspi->streamSizes.cbHeader + chunk;
1172 buffers[2].cbBuffer = sspi->streamSizes.cbTrailer;
2ece34a9
MS
1173 buffers[2].BufferType = SECBUFFER_STREAM_TRAILER;
1174 buffers[3].BufferType = SECBUFFER_EMPTY;
1175
1176 /*
1177 * Encrypt the data
1178 */
1179
e32497ad 1180 scRet = EncryptMessage(&sspi->context, 0, &message, 0);
2ece34a9
MS
1181
1182 if (FAILED(scRet))
1183 {
52770f63 1184 DEBUG_printf(("_httpTLSWrite: EncryptMessage failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9
MS
1185 WSASetLastError(WSASYSCALLFAILURE);
1186 return (-1);
1187 }
1188
1189 /*
1190 * Send the data. Remember the size of the total data to send is the size
1191 * of the header, the size of the data the caller passed in and the size
1192 * of the trailer...
1193 */
1194
e32497ad 1195 num = send(http->fd, sspi->writeBuffer, buffers[0].cbBuffer + buffers[1].cbBuffer + buffers[2].cbBuffer, 0);
2ece34a9
MS
1196
1197 if (num <= 0)
1198 {
1199 DEBUG_printf(("_httpTLSWrite: send failed: %ld", WSAGetLastError()));
1200 return (num);
1201 }
1202
1203 bytesLeft -= chunk;
1204 bufptr += chunk;
1205 }
1206
1207 return (len);
1208}
1209
1210
33c9220c 1211#if 0
2ece34a9
MS
1212/*
1213 * 'http_setup_ssl()' - Set up SSL/TLS support on a connection.
1214 */
1215
1216static int /* O - 0 on success, -1 on failure */
1217http_setup_ssl(http_t *http) /* I - Connection to server */
1218{
1219 char hostname[256], /* Hostname */
1220 *hostptr; /* Pointer into hostname */
1221
1222 TCHAR username[256]; /* Username returned from GetUserName() */
1223 TCHAR commonName[256];/* Common name for certificate */
1224 DWORD dwSize; /* 32 bit size */
cc754834 1225
2ece34a9
MS
1226
1227 DEBUG_printf(("7http_setup_ssl(http=%p)", http));
cc754834
MS
1228
1229 /*
2ece34a9 1230 * Get the hostname to use for SSL...
cc754834 1231 */
2ece34a9
MS
1232
1233 if (httpAddrLocalhost(http->hostaddr))
cc754834 1234 {
2ece34a9 1235 strlcpy(hostname, "localhost", sizeof(hostname));
cc754834 1236 }
2ece34a9
MS
1237 else
1238 {
1239 /*
1240 * Otherwise make sure the hostname we have does not end in a trailing dot.
1241 */
cc754834 1242
2ece34a9
MS
1243 strlcpy(hostname, http->hostname, sizeof(hostname));
1244 if ((hostptr = hostname + strlen(hostname) - 1) >= hostname &&
1245 *hostptr == '.')
1246 *hostptr = '\0';
1247 }
cc754834 1248
33c9220c 1249 http->tls = http_sspi_alloc();
cc754834 1250
2ece34a9
MS
1251 if (!http->tls)
1252 {
1253 _cupsSetHTTPError(HTTP_STATUS_ERROR);
1254 return (-1);
1255 }
cc754834 1256
2ece34a9
MS
1257 dwSize = sizeof(username) / sizeof(TCHAR);
1258 GetUserName(username, &dwSize);
1259 _sntprintf_s(commonName, sizeof(commonName) / sizeof(TCHAR),
1260 sizeof(commonName) / sizeof(TCHAR), TEXT("CN=%s"), username);
cc754834 1261
2ece34a9
MS
1262 if (!_sspiGetCredentials(http->tls, L"ClientContainer",
1263 commonName, FALSE))
1264 {
1265 _sspiFree(http->tls);
1266 http->tls = NULL;
cc754834 1267
2ece34a9
MS
1268 http->error = EIO;
1269 http->status = HTTP_STATUS_ERROR;
cc754834 1270
2ece34a9
MS
1271 _cupsSetError(IPP_STATUS_ERROR_CUPS_PKI,
1272 _("Unable to establish a secure connection to host."), 1);
cc754834 1273
2ece34a9
MS
1274 return (-1);
1275 }
1276
1277 _sspiSetAllowsAnyRoot(http->tls, TRUE);
1278 _sspiSetAllowsExpiredCerts(http->tls, TRUE);
1279
1280 if (!_sspiConnect(http->tls, hostname))
1281 {
1282 _sspiFree(http->tls);
1283 http->tls = NULL;
1284
1285 http->error = EIO;
1286 http->status = HTTP_STATUS_ERROR;
1287
1288 _cupsSetError(IPP_STATUS_ERROR_CUPS_PKI,
1289 _("Unable to establish a secure connection to host."), 1);
1290
1291 return (-1);
1292 }
1293
1294 return (0);
cc754834 1295}
33c9220c 1296#endif // 0
cc754834
MS
1297
1298
1299/*
2ece34a9 1300 * 'http_sspi_alloc()' - Allocate SSPI object.
cc754834 1301 */
2ece34a9
MS
1302
1303static _http_sspi_t * /* O - New SSPI/SSL object */
1304http_sspi_alloc(void)
1305{
e32497ad 1306 return ((_http_sspi_t *)calloc(sizeof(_http_sspi_t), 1));
2ece34a9
MS
1307}
1308
1309
1310/*
1311 * 'http_sspi_client()' - Negotiate a TLS connection as a client.
1312 */
1313
1314static int /* O - 0 on success, -1 on failure */
1315http_sspi_client(http_t *http, /* I - Client connection */
1316 const char *hostname) /* I - Server hostname */
cc754834 1317{
e32497ad 1318 _http_sspi_t *sspi = http->tls; /* SSPI data */
9870d12b 1319 DWORD dwSize; /* Size for buffer */
33c9220c
MS
1320 DWORD dwSSPIFlags; /* SSL connection attributes we want */
1321 DWORD dwSSPIOutFlags; /* SSL connection attributes we got */
1322 TimeStamp tsExpiry; /* Time stamp */
1323 SECURITY_STATUS scRet; /* Status */
1324 int cbData; /* Data count */
1325 SecBufferDesc inBuffer; /* Array of SecBuffer structs */
1326 SecBuffer inBuffers[2]; /* Security package buffer */
1327 SecBufferDesc outBuffer; /* Array of SecBuffer structs */
1328 SecBuffer outBuffers[1]; /* Security package buffer */
1329 int ret = 0; /* Return value */
9870d12b
MS
1330 char username[1024], /* Current username */
1331 common_name[1024]; /* CN=username */
cc754834 1332
cc754834 1333
52770f63 1334 DEBUG_printf(("4http_sspi_client(http=%p, hostname=\"%s\")", http, hostname));
2ece34a9 1335
cc754834
MS
1336 dwSSPIFlags = ISC_REQ_SEQUENCE_DETECT |
1337 ISC_REQ_REPLAY_DETECT |
1338 ISC_REQ_CONFIDENTIALITY |
1339 ISC_RET_EXTENDED_ERROR |
1340 ISC_REQ_ALLOCATE_MEMORY |
1341 ISC_REQ_STREAM;
1342
9870d12b
MS
1343 /*
1344 * Lookup the client certificate...
1345 */
1346
1347 dwSize = sizeof(username);
1348 GetUserName(username, &dwSize);
1349 snprintf(common_name, sizeof(common_name), "CN=%s", username);
1350
e32497ad
MS
1351 if (!http_sspi_find_credentials(http, L"ClientContainer", common_name))
1352 if (!http_sspi_make_credentials(http->tls, L"ClientContainer", common_name, _HTTP_MODE_CLIENT, 10))
1353 {
52770f63 1354 DEBUG_puts("5http_sspi_client: Unable to get client credentials.");
e32497ad
MS
1355 return (-1);
1356 }
9870d12b 1357
cc754834
MS
1358 /*
1359 * Initiate a ClientHello message and generate a token.
1360 */
2ece34a9 1361
cc754834
MS
1362 outBuffers[0].pvBuffer = NULL;
1363 outBuffers[0].BufferType = SECBUFFER_TOKEN;
1364 outBuffers[0].cbBuffer = 0;
1365
2ece34a9
MS
1366 outBuffer.cBuffers = 1;
1367 outBuffer.pBuffers = outBuffers;
cc754834
MS
1368 outBuffer.ulVersion = SECBUFFER_VERSION;
1369
e32497ad 1370 scRet = InitializeSecurityContext(&sspi->creds, NULL, TEXT(""), dwSSPIFlags, 0, SECURITY_NATIVE_DREP, NULL, 0, &sspi->context, &outBuffer, &dwSSPIOutFlags, &tsExpiry);
cc754834
MS
1371
1372 if (scRet != SEC_I_CONTINUE_NEEDED)
1373 {
52770f63 1374 DEBUG_printf(("5http_sspi_client: InitializeSecurityContext(1) failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9 1375 return (-1);
cc754834
MS
1376 }
1377
1378 /*
1379 * Send response to server if there is one.
1380 */
2ece34a9 1381
cc754834
MS
1382 if (outBuffers[0].cbBuffer && outBuffers[0].pvBuffer)
1383 {
2ece34a9 1384 if ((cbData = send(http->fd, outBuffers[0].pvBuffer, outBuffers[0].cbBuffer, 0)) <= 0)
cc754834 1385 {
52770f63 1386 DEBUG_printf(("5http_sspi_client: send failed: %d", WSAGetLastError()));
cc754834 1387 FreeContextBuffer(outBuffers[0].pvBuffer);
e32497ad 1388 DeleteSecurityContext(&sspi->context);
2ece34a9 1389 return (-1);
cc754834
MS
1390 }
1391
52770f63 1392 DEBUG_printf(("5http_sspi_client: %d bytes of handshake data sent.", cbData));
cc754834 1393
cc754834
MS
1394 FreeContextBuffer(outBuffers[0].pvBuffer);
1395 outBuffers[0].pvBuffer = NULL;
1396 }
1397
1398 dwSSPIFlags = ISC_REQ_MANUAL_CRED_VALIDATION |
2ece34a9 1399 ISC_REQ_SEQUENCE_DETECT |
cc754834
MS
1400 ISC_REQ_REPLAY_DETECT |
1401 ISC_REQ_CONFIDENTIALITY |
1402 ISC_RET_EXTENDED_ERROR |
1403 ISC_REQ_ALLOCATE_MEMORY |
1404 ISC_REQ_STREAM;
1405
e32497ad 1406 sspi->decryptBufferUsed = 0;
cc754834 1407
321d8d57 1408 /*
cc754834
MS
1409 * Loop until the handshake is finished or an error occurs.
1410 */
2ece34a9 1411
cc754834
MS
1412 scRet = SEC_I_CONTINUE_NEEDED;
1413
1414 while(scRet == SEC_I_CONTINUE_NEEDED ||
1415 scRet == SEC_E_INCOMPLETE_MESSAGE ||
321d8d57 1416 scRet == SEC_I_INCOMPLETE_CREDENTIALS)
cc754834 1417 {
e32497ad 1418 if (sspi->decryptBufferUsed == 0 || scRet == SEC_E_INCOMPLETE_MESSAGE)
cc754834 1419 {
e32497ad 1420 if (sspi->decryptBufferLength <= sspi->decryptBufferUsed)
cc754834 1421 {
2ece34a9 1422 BYTE *temp; /* New buffer */
cc754834 1423
e32497ad 1424 if (sspi->decryptBufferLength >= 262144)
2ece34a9
MS
1425 {
1426 WSASetLastError(E_OUTOFMEMORY);
52770f63 1427 DEBUG_puts("5http_sspi_client: Decryption buffer too large (>256k)");
2ece34a9
MS
1428 return (-1);
1429 }
1430
e32497ad 1431 if ((temp = realloc(sspi->decryptBuffer, sspi->decryptBufferLength + 4096)) == NULL)
2ece34a9 1432 {
52770f63 1433 DEBUG_printf(("5http_sspi_client: Unable to allocate %d byte buffer.", sspi->decryptBufferLength + 4096));
2ece34a9
MS
1434 WSASetLastError(E_OUTOFMEMORY);
1435 return (-1);
1436 }
1437
e32497ad
MS
1438 sspi->decryptBufferLength += 4096;
1439 sspi->decryptBuffer = temp;
cc754834 1440 }
321d8d57 1441
e32497ad 1442 cbData = recv(http->fd, sspi->decryptBuffer + sspi->decryptBufferUsed, (int)(sspi->decryptBufferLength - sspi->decryptBufferUsed), 0);
321d8d57 1443
2ece34a9 1444 if (cbData < 0)
cc754834 1445 {
52770f63 1446 DEBUG_printf(("5http_sspi_client: recv failed: %d", WSAGetLastError()));
2ece34a9 1447 return (-1);
cc754834
MS
1448 }
1449 else if (cbData == 0)
1450 {
52770f63 1451 DEBUG_printf(("5http_sspi_client: Server unexpectedly disconnected."));
2ece34a9 1452 return (-1);
cc754834
MS
1453 }
1454
52770f63 1455 DEBUG_printf(("5http_sspi_client: %d bytes of handshake data received", cbData));
cc754834 1456
e32497ad 1457 sspi->decryptBufferUsed += cbData;
cc754834
MS
1458 }
1459
1460 /*
2ece34a9
MS
1461 * Set up the input buffers. Buffer 0 is used to pass in data received from
1462 * the server. Schannel will consume some or all of this. Leftover data
1463 * (if any) will be placed in buffer 1 and given a buffer type of
1464 * SECBUFFER_EXTRA.
cc754834 1465 */
2ece34a9 1466
e32497ad
MS
1467 inBuffers[0].pvBuffer = sspi->decryptBuffer;
1468 inBuffers[0].cbBuffer = (unsigned long)sspi->decryptBufferUsed;
cc754834
MS
1469 inBuffers[0].BufferType = SECBUFFER_TOKEN;
1470
1471 inBuffers[1].pvBuffer = NULL;
1472 inBuffers[1].cbBuffer = 0;
1473 inBuffers[1].BufferType = SECBUFFER_EMPTY;
1474
1475 inBuffer.cBuffers = 2;
1476 inBuffer.pBuffers = inBuffers;
1477 inBuffer.ulVersion = SECBUFFER_VERSION;
1478
1479 /*
2ece34a9
MS
1480 * Set up the output buffers. These are initialized to NULL so as to make it
1481 * less likely we'll attempt to free random garbage later.
cc754834 1482 */
cc754834 1483
2ece34a9
MS
1484 outBuffers[0].pvBuffer = NULL;
1485 outBuffers[0].BufferType = SECBUFFER_TOKEN;
1486 outBuffers[0].cbBuffer = 0;
1487
1488 outBuffer.cBuffers = 1;
1489 outBuffer.pBuffers = outBuffers;
1490 outBuffer.ulVersion = SECBUFFER_VERSION;
cc754834
MS
1491
1492 /*
1493 * Call InitializeSecurityContext.
1494 */
2ece34a9 1495
e32497ad 1496 scRet = InitializeSecurityContext(&sspi->creds, &sspi->context, NULL, dwSSPIFlags, 0, SECURITY_NATIVE_DREP, &inBuffer, 0, NULL, &outBuffer, &dwSSPIOutFlags, &tsExpiry);
cc754834
MS
1497
1498 /*
2ece34a9
MS
1499 * If InitializeSecurityContext was successful (or if the error was one of
1500 * the special extended ones), send the contents of the output buffer to the
1501 * server.
cc754834 1502 */
2ece34a9 1503
cc754834
MS
1504 if (scRet == SEC_E_OK ||
1505 scRet == SEC_I_CONTINUE_NEEDED ||
1506 FAILED(scRet) && (dwSSPIOutFlags & ISC_RET_EXTENDED_ERROR))
1507 {
1508 if (outBuffers[0].cbBuffer && outBuffers[0].pvBuffer)
1509 {
2ece34a9 1510 cbData = send(http->fd, outBuffers[0].pvBuffer, outBuffers[0].cbBuffer, 0);
321d8d57 1511
2ece34a9 1512 if (cbData <= 0)
cc754834 1513 {
52770f63 1514 DEBUG_printf(("5http_sspi_client: send failed: %d", WSAGetLastError()));
cc754834 1515 FreeContextBuffer(outBuffers[0].pvBuffer);
e32497ad 1516 DeleteSecurityContext(&sspi->context);
2ece34a9 1517 return (-1);
cc754834
MS
1518 }
1519
52770f63 1520 DEBUG_printf(("5http_sspi_client: %d bytes of handshake data sent.", cbData));
cc754834
MS
1521
1522 /*
1523 * Free output buffer.
1524 */
2ece34a9 1525
cc754834
MS
1526 FreeContextBuffer(outBuffers[0].pvBuffer);
1527 outBuffers[0].pvBuffer = NULL;
1528 }
1529 }
1530
1531 /*
2ece34a9
MS
1532 * If InitializeSecurityContext returned SEC_E_INCOMPLETE_MESSAGE, then we
1533 * need to read more data from the server and try again.
cc754834 1534 */
2ece34a9 1535
cc754834
MS
1536 if (scRet == SEC_E_INCOMPLETE_MESSAGE)
1537 continue;
1538
1539 /*
2ece34a9
MS
1540 * If InitializeSecurityContext returned SEC_E_OK, then the handshake
1541 * completed successfully.
cc754834 1542 */
2ece34a9 1543
cc754834
MS
1544 if (scRet == SEC_E_OK)
1545 {
1546 /*
1547 * If the "extra" buffer contains data, this is encrypted application
2ece34a9
MS
1548 * protocol layer stuff. It needs to be saved. The application layer will
1549 * later decrypt it with DecryptMessage.
cc754834 1550 */
2ece34a9 1551
52770f63 1552 DEBUG_puts("5http_sspi_client: Handshake was successful.");
cc754834
MS
1553
1554 if (inBuffers[1].BufferType == SECBUFFER_EXTRA)
1555 {
e32497ad 1556 memmove(sspi->decryptBuffer, sspi->decryptBuffer + sspi->decryptBufferUsed - inBuffers[1].cbBuffer, inBuffers[1].cbBuffer);
cc754834 1557
e32497ad 1558 sspi->decryptBufferUsed = inBuffers[1].cbBuffer;
cc754834 1559
52770f63 1560 DEBUG_printf(("5http_sspi_client: %d bytes of app data was bundled with handshake data", sspi->decryptBufferUsed));
cc754834
MS
1561 }
1562 else
e32497ad 1563 sspi->decryptBufferUsed = 0;
cc754834
MS
1564
1565 /*
1566 * Bail out to quit
1567 */
2ece34a9 1568
cc754834
MS
1569 break;
1570 }
1571
1572 /*
1573 * Check for fatal error.
1574 */
2ece34a9 1575
cc754834
MS
1576 if (FAILED(scRet))
1577 {
52770f63 1578 DEBUG_printf(("5http_sspi_client: InitializeSecurityContext(2) failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9 1579 ret = -1;
cc754834
MS
1580 break;
1581 }
1582
1583 /*
1584 * If InitializeSecurityContext returned SEC_I_INCOMPLETE_CREDENTIALS,
321d8d57 1585 * then the server just requested client authentication.
cc754834 1586 */
2ece34a9 1587
cc754834
MS
1588 if (scRet == SEC_I_INCOMPLETE_CREDENTIALS)
1589 {
1590 /*
1591 * Unimplemented
1592 */
2ece34a9 1593
52770f63 1594 DEBUG_printf(("5http_sspi_client: server requested client credentials."));
2ece34a9 1595 ret = -1;
cc754834
MS
1596 break;
1597 }
1598
1599 /*
2ece34a9 1600 * Copy any leftover data from the "extra" buffer, and go around again.
cc754834 1601 */
2ece34a9 1602
cc754834
MS
1603 if (inBuffers[1].BufferType == SECBUFFER_EXTRA)
1604 {
e32497ad 1605 memmove(sspi->decryptBuffer, sspi->decryptBuffer + sspi->decryptBufferUsed - inBuffers[1].cbBuffer, inBuffers[1].cbBuffer);
cc754834 1606
e32497ad 1607 sspi->decryptBufferUsed = inBuffers[1].cbBuffer;
cc754834
MS
1608 }
1609 else
1610 {
e32497ad 1611 sspi->decryptBufferUsed = 0;
cc754834
MS
1612 }
1613 }
1614
2ece34a9 1615 if (!ret)
cc754834 1616 {
cc754834 1617 /*
2ece34a9 1618 * Success! Get the server cert
cc754834 1619 */
2ece34a9 1620
e32497ad 1621 sspi->contextInitialized = TRUE;
2ece34a9 1622
e32497ad 1623 scRet = QueryContextAttributes(&sspi->context, SECPKG_ATTR_REMOTE_CERT_CONTEXT, (VOID *)&(sspi->remoteCert));
cc754834
MS
1624
1625 if (scRet != SEC_E_OK)
1626 {
52770f63 1627 DEBUG_printf(("5http_sspi_client: QueryContextAttributes failed(SECPKG_ATTR_REMOTE_CERT_CONTEXT): %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9 1628 return (-1);
cc754834
MS
1629 }
1630
cc754834
MS
1631 /*
1632 * Find out how big the header/trailer will be:
1633 */
2ece34a9 1634
e32497ad 1635 scRet = QueryContextAttributes(&sspi->context, SECPKG_ATTR_STREAM_SIZES, &sspi->streamSizes);
cc754834
MS
1636
1637 if (scRet != SEC_E_OK)
1638 {
52770f63 1639 DEBUG_printf(("5http_sspi_client: QueryContextAttributes failed(SECPKG_ATTR_STREAM_SIZES): %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9 1640 ret = -1;
cc754834
MS
1641 }
1642 }
1643
2ece34a9
MS
1644 return (ret);
1645}
cc754834 1646
cc754834 1647
17eecbf1
MS
1648/*
1649 * 'http_sspi_create_credential()' - Create an SSPI certificate context.
1650 */
1651
1652static PCCERT_CONTEXT /* O - Certificate context */
1653http_sspi_create_credential(
1654 http_credential_t *cred) /* I - Credential */
1655{
1656 if (cred)
1657 return (CertCreateCertificateContext(X509_ASN_ENCODING, cred->data, cred->datalen));
1658 else
1659 return (NULL);
1660}
1661
1662
2ece34a9 1663/*
e32497ad 1664 * 'http_sspi_find_credentials()' - Retrieve a TLS certificate from the system store.
2ece34a9
MS
1665 */
1666
9870d12b 1667static BOOL /* O - 1 on success, 0 on failure */
e32497ad 1668http_sspi_find_credentials(
9870d12b
MS
1669 http_t *http, /* I - HTTP connection */
1670 const LPWSTR container, /* I - Cert container name */
e32497ad 1671 const char *common_name) /* I - Common name of certificate */
cc754834 1672{
e32497ad 1673 _http_sspi_t *sspi = http->tls; /* SSPI data */
9870d12b
MS
1674 HCERTSTORE store = NULL; /* Certificate store */
1675 PCCERT_CONTEXT storedContext = NULL; /* Context created from the store */
9870d12b
MS
1676 DWORD dwSize = 0; /* 32 bit size */
1677 PBYTE p = NULL; /* Temporary storage */
1678 HCRYPTPROV hProv = (HCRYPTPROV)NULL;
2ece34a9 1679 /* Handle to a CSP */
9870d12b
MS
1680 CERT_NAME_BLOB sib; /* Arbitrary array of bytes */
1681 SCHANNEL_CRED SchannelCred; /* Schannel credential data */
1682 TimeStamp tsExpiry; /* Time stamp */
1683 SECURITY_STATUS Status; /* Status */
9870d12b 1684 BOOL ok = TRUE; /* Return value */
cc754834 1685
cc754834 1686
9870d12b 1687 if (!CryptAcquireContextW(&hProv, (LPWSTR)container, MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_NEWKEYSET | CRYPT_MACHINE_KEYSET))
cc754834 1688 {
2ece34a9 1689 if (GetLastError() == NTE_EXISTS)
cc754834 1690 {
9870d12b 1691 if (!CryptAcquireContextW(&hProv, (LPWSTR)container, MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_MACHINE_KEYSET))
cc754834 1692 {
52770f63 1693 DEBUG_printf(("5http_sspi_find_credentials: CryptAcquireContext failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
cc754834
MS
1694 ok = FALSE;
1695 goto cleanup;
1696 }
cc754834 1697 }
2ece34a9 1698 }
cc754834 1699
9870d12b 1700 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, X509_ASN_ENCODING|PKCS_7_ASN_ENCODING, hProv, CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_STORE_NO_CRYPT_RELEASE_FLAG | CERT_STORE_OPEN_EXISTING_FLAG, L"MY");
cc754834 1701
2ece34a9
MS
1702 if (!store)
1703 {
52770f63 1704 DEBUG_printf(("5http_sspi_find_credentials: CertOpenSystemStore failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
2ece34a9
MS
1705 ok = FALSE;
1706 goto cleanup;
1707 }
cc754834 1708
2ece34a9 1709 dwSize = 0;
cc754834 1710
9870d12b 1711 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, NULL, &dwSize, NULL))
2ece34a9 1712 {
52770f63 1713 DEBUG_printf(("5http_sspi_find_credentials: CertStrToName failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
2ece34a9
MS
1714 ok = FALSE;
1715 goto cleanup;
1716 }
cc754834 1717
9870d12b 1718 p = (PBYTE)malloc(dwSize);
cc754834 1719
2ece34a9
MS
1720 if (!p)
1721 {
52770f63 1722 DEBUG_printf(("5http_sspi_find_credentials: malloc failed for %d bytes.", dwSize));
2ece34a9
MS
1723 ok = FALSE;
1724 goto cleanup;
1725 }
cc754834 1726
9870d12b 1727 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, p, &dwSize, NULL))
2ece34a9 1728 {
52770f63 1729 DEBUG_printf(("5http_sspi_find_credentials: CertStrToName failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
2ece34a9
MS
1730 ok = FALSE;
1731 goto cleanup;
1732 }
cc754834 1733
2ece34a9
MS
1734 sib.cbData = dwSize;
1735 sib.pbData = p;
cc754834 1736
9870d12b 1737 storedContext = CertFindCertificateInStore(store, X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, 0, CERT_FIND_SUBJECT_NAME, &sib, NULL);
cc754834 1738
2ece34a9
MS
1739 if (!storedContext)
1740 {
52770f63 1741 DEBUG_printf(("5http_sspi_find_credentials: Unable to find credentials for \"%s\".", common_name));
e32497ad
MS
1742 ok = FALSE;
1743 goto cleanup;
cc754834
MS
1744 }
1745
2ece34a9 1746 ZeroMemory(&SchannelCred, sizeof(SchannelCred));
cc754834 1747
2ece34a9 1748 SchannelCred.dwVersion = SCHANNEL_CRED_VERSION;
9870d12b
MS
1749 SchannelCred.cCreds = 1;
1750 SchannelCred.paCred = &storedContext;
cc754834 1751
2ece34a9 1752 /*
63aefcd5 1753 * Set supported protocols (can also be overriden in the registry...)
2ece34a9 1754 */
9870d12b 1755
5c460b65 1756#ifdef SP_PROT_TLS1_2_SERVER
e32497ad 1757 if (http->mode == _HTTP_MODE_SERVER)
63aefcd5 1758 {
ee6226a5
MS
1759 if (tls_options & _HTTP_TLS_DENY_TLS10)
1760 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_SERVER | SP_PROT_TLS1_1_SERVER;
1761 else if (tls_options & _HTTP_TLS_ALLOW_SSL3)
5c460b65 1762 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_SERVER | SP_PROT_TLS1_1_SERVER | SP_PROT_TLS1_0_SERVER | SP_PROT_SSL3_SERVER;
63aefcd5 1763 else
5c460b65 1764 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_SERVER | SP_PROT_TLS1_1_SERVER | SP_PROT_TLS1_0_SERVER;
63aefcd5
MS
1765 }
1766 else
1767 {
ee6226a5
MS
1768 if (tls_options & _HTTP_TLS_DENY_TLS10)
1769 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_CLIENT | SP_PROT_TLS1_1_CLIENT;
1770 else if (tls_options & _HTTP_TLS_ALLOW_SSL3)
5c460b65 1771 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_CLIENT | SP_PROT_TLS1_1_CLIENT | SP_PROT_TLS1_0_CLIENT | SP_PROT_SSL3_CLIENT;
63aefcd5 1772 else
5c460b65 1773 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_2_CLIENT | SP_PROT_TLS1_1_CLIENT | SP_PROT_TLS1_0_CLIENT;
63aefcd5
MS
1774 }
1775
5c460b65
MS
1776#else
1777 if (http->mode == _HTTP_MODE_SERVER)
1778 {
1779 if (tls_options & _HTTP_TLS_ALLOW_SSL3)
1780 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_SERVER | SP_PROT_SSL3_SERVER;
1781 else
1782 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_SERVER;
1783 }
1784 else
1785 {
1786 if (tls_options & _HTTP_TLS_ALLOW_SSL3)
1787 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_CLIENT | SP_PROT_SSL3_CLIENT;
1788 else
1789 SchannelCred.grbitEnabledProtocols = SP_PROT_TLS1_CLIENT;
1790 }
1791#endif /* SP_PROT_TLS1_2_SERVER */
1792
ee6226a5 1793 /* TODO: Support _HTTP_TLS_ALLOW_RC4 and _HTTP_TLS_ALLOW_DH options; right now we'll rely on Windows registry to enable/disable RC4/DH... */
cc754834 1794
2ece34a9
MS
1795 /*
1796 * Create an SSPI credential.
1797 */
9870d12b 1798
e32497ad 1799 Status = AcquireCredentialsHandle(NULL, UNISP_NAME, http->mode == _HTTP_MODE_SERVER ? SECPKG_CRED_INBOUND : SECPKG_CRED_OUTBOUND, NULL, &SchannelCred, NULL, NULL, &sspi->creds, &tsExpiry);
2ece34a9 1800 if (Status != SEC_E_OK)
cc754834 1801 {
52770f63 1802 DEBUG_printf(("5http_sspi_find_credentials: AcquireCredentialsHandle failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), Status)));
2ece34a9 1803 ok = FALSE;
cc754834
MS
1804 goto cleanup;
1805 }
1806
2ece34a9 1807cleanup:
cc754834 1808
2ece34a9
MS
1809 /*
1810 * Cleanup
1811 */
9870d12b 1812
2ece34a9
MS
1813 if (storedContext)
1814 CertFreeCertificateContext(storedContext);
cc754834 1815
2ece34a9
MS
1816 if (p)
1817 free(p);
cc754834 1818
2ece34a9
MS
1819 if (store)
1820 CertCloseStore(store, 0);
cc754834 1821
2ece34a9
MS
1822 if (hProv)
1823 CryptReleaseContext(hProv, 0);
cc754834 1824
2ece34a9 1825 return (ok);
cc754834 1826}
9870d12b
MS
1827
1828
1829/*
1830 * 'http_sspi_free()' - Close a connection and free resources.
1831 */
1832
1833static void
e32497ad 1834http_sspi_free(_http_sspi_t *sspi) /* I - SSPI data */
9870d12b 1835{
e32497ad 1836 if (!sspi)
9870d12b
MS
1837 return;
1838
e32497ad
MS
1839 if (sspi->contextInitialized)
1840 DeleteSecurityContext(&sspi->context);
1841
1842 if (sspi->decryptBuffer)
1843 free(sspi->decryptBuffer);
1844
1845 if (sspi->readBuffer)
1846 free(sspi->readBuffer);
1847
1848 if (sspi->writeBuffer)
1849 free(sspi->writeBuffer);
1850
1851 if (sspi->localCert)
1852 CertFreeCertificateContext(sspi->localCert);
1853
1854 if (sspi->remoteCert)
1855 CertFreeCertificateContext(sspi->remoteCert);
1856
1857 free(sspi);
1858}
1859
1860
1861/*
1862 * 'http_sspi_make_credentials()' - Create a TLS certificate in the system store.
1863 */
1864
1865static BOOL /* O - 1 on success, 0 on failure */
1866http_sspi_make_credentials(
1867 _http_sspi_t *sspi, /* I - SSPI data */
1868 const LPWSTR container, /* I - Cert container name */
1869 const char *common_name, /* I - Common name of certificate */
1870 _http_mode_t mode, /* I - Client or server? */
1871 int years) /* I - Years until expiration */
1872{
1873 HCERTSTORE store = NULL; /* Certificate store */
1874 PCCERT_CONTEXT storedContext = NULL; /* Context created from the store */
1875 PCCERT_CONTEXT createdContext = NULL; /* Context created by us */
1876 DWORD dwSize = 0; /* 32 bit size */
1877 PBYTE p = NULL; /* Temporary storage */
1878 HCRYPTPROV hProv = (HCRYPTPROV)NULL;
1879 /* Handle to a CSP */
1880 CERT_NAME_BLOB sib; /* Arbitrary array of bytes */
1881 SCHANNEL_CRED SchannelCred; /* Schannel credential data */
1882 TimeStamp tsExpiry; /* Time stamp */
1883 SECURITY_STATUS Status; /* Status */
1884 HCRYPTKEY hKey = (HCRYPTKEY)NULL; /* Handle to crypto key */
1885 CRYPT_KEY_PROV_INFO kpi; /* Key container info */
1886 SYSTEMTIME et; /* System time */
1887 CERT_EXTENSIONS exts; /* Array of cert extensions */
1888 CRYPT_KEY_PROV_INFO ckp; /* Handle to crypto key */
1889 BOOL ok = TRUE; /* Return value */
1890
1891
1892 DEBUG_printf(("4http_sspi_make_credentials(sspi=%p, container=%p, common_name=\"%s\", mode=%d, years=%d)", sspi, container, common_name, mode, years));
1893
1894 if (!CryptAcquireContextW(&hProv, (LPWSTR)container, MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_NEWKEYSET | CRYPT_MACHINE_KEYSET))
1895 {
1896 if (GetLastError() == NTE_EXISTS)
1897 {
1898 if (!CryptAcquireContextW(&hProv, (LPWSTR)container, MS_DEF_PROV_W, PROV_RSA_FULL, CRYPT_MACHINE_KEYSET))
1899 {
52770f63 1900 DEBUG_printf(("5http_sspi_make_credentials: CryptAcquireContext failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1901 ok = FALSE;
1902 goto cleanup;
1903 }
1904 }
1905 }
1906
1907 store = CertOpenStore(CERT_STORE_PROV_SYSTEM, X509_ASN_ENCODING|PKCS_7_ASN_ENCODING, hProv, CERT_SYSTEM_STORE_LOCAL_MACHINE | CERT_STORE_NO_CRYPT_RELEASE_FLAG | CERT_STORE_OPEN_EXISTING_FLAG, L"MY");
1908
1909 if (!store)
1910 {
52770f63 1911 DEBUG_printf(("5http_sspi_make_credentials: CertOpenSystemStore failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1912 ok = FALSE;
1913 goto cleanup;
1914 }
1915
1916 dwSize = 0;
1917
1918 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, NULL, &dwSize, NULL))
1919 {
52770f63 1920 DEBUG_printf(("5http_sspi_make_credentials: CertStrToName failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1921 ok = FALSE;
1922 goto cleanup;
1923 }
1924
1925 p = (PBYTE)malloc(dwSize);
1926
1927 if (!p)
1928 {
52770f63 1929 DEBUG_printf(("5http_sspi_make_credentials: malloc failed for %d bytes", dwSize));
e32497ad
MS
1930 ok = FALSE;
1931 goto cleanup;
1932 }
1933
1934 if (!CertStrToName(X509_ASN_ENCODING, common_name, CERT_OID_NAME_STR, NULL, p, &dwSize, NULL))
1935 {
52770f63 1936 DEBUG_printf(("5http_sspi_make_credentials: CertStrToName failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1937 ok = FALSE;
1938 goto cleanup;
1939 }
1940
1941 /*
52770f63 1942 * Create a private key and self-signed certificate...
e32497ad 1943 */
9870d12b 1944
e32497ad
MS
1945 if (!CryptGenKey(hProv, AT_KEYEXCHANGE, CRYPT_EXPORTABLE, &hKey))
1946 {
52770f63 1947 DEBUG_printf(("5http_sspi_make_credentials: CryptGenKey failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1948 ok = FALSE;
1949 goto cleanup;
1950 }
1951
1952 ZeroMemory(&kpi, sizeof(kpi));
1953 kpi.pwszContainerName = (LPWSTR)container;
1954 kpi.pwszProvName = MS_DEF_PROV_W;
1955 kpi.dwProvType = PROV_RSA_FULL;
1956 kpi.dwFlags = CERT_SET_KEY_CONTEXT_PROP_ID;
1957 kpi.dwKeySpec = AT_KEYEXCHANGE;
1958
1959 GetSystemTime(&et);
1960 et.wYear += years;
9870d12b 1961
e32497ad 1962 ZeroMemory(&exts, sizeof(exts));
9870d12b 1963
e32497ad 1964 createdContext = CertCreateSelfSignCertificate(hProv, &sib, 0, &kpi, NULL, NULL, &et, &exts);
9870d12b 1965
e32497ad
MS
1966 if (!createdContext)
1967 {
52770f63 1968 DEBUG_printf(("5http_sspi_make_credentials: CertCreateSelfSignCertificate failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1969 ok = FALSE;
1970 goto cleanup;
1971 }
9870d12b 1972
52770f63
MS
1973 /*
1974 * Add the created context to the named store, and associate it with the named
1975 * container...
1976 */
1977
e32497ad
MS
1978 if (!CertAddCertificateContextToStore(store, createdContext, CERT_STORE_ADD_REPLACE_EXISTING, &storedContext))
1979 {
52770f63 1980 DEBUG_printf(("5http_sspi_make_credentials: CertAddCertificateContextToStore failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1981 ok = FALSE;
1982 goto cleanup;
1983 }
1984
1985 ZeroMemory(&ckp, sizeof(ckp));
1986 ckp.pwszContainerName = (LPWSTR) container;
1987 ckp.pwszProvName = MS_DEF_PROV_W;
1988 ckp.dwProvType = PROV_RSA_FULL;
1989 ckp.dwFlags = CRYPT_MACHINE_KEYSET;
1990 ckp.dwKeySpec = AT_KEYEXCHANGE;
1991
1992 if (!CertSetCertificateContextProperty(storedContext, CERT_KEY_PROV_INFO_PROP_ID, 0, &ckp))
1993 {
52770f63 1994 DEBUG_printf(("5http_sspi_make_credentials: CertSetCertificateContextProperty failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), GetLastError())));
e32497ad
MS
1995 ok = FALSE;
1996 goto cleanup;
1997 }
1998
52770f63
MS
1999 /*
2000 * Get a handle to use the certificate...
2001 */
e32497ad
MS
2002
2003 ZeroMemory(&SchannelCred, sizeof(SchannelCred));
2004
2005 SchannelCred.dwVersion = SCHANNEL_CRED_VERSION;
2006 SchannelCred.cCreds = 1;
2007 SchannelCred.paCred = &storedContext;
2008
2009 /*
2010 * SSPI doesn't seem to like it if grbitEnabledProtocols is set for a client.
2011 */
2012
2013 if (mode == _HTTP_MODE_SERVER)
2014 SchannelCred.grbitEnabledProtocols = SP_PROT_SSL3TLS1;
2015
2016 /*
2017 * Create an SSPI credential.
2018 */
2019
2020 Status = AcquireCredentialsHandle(NULL, UNISP_NAME, mode == _HTTP_MODE_SERVER ? SECPKG_CRED_INBOUND : SECPKG_CRED_OUTBOUND, NULL, &SchannelCred, NULL, NULL, &sspi->creds, &tsExpiry);
2021 if (Status != SEC_E_OK)
2022 {
52770f63 2023 DEBUG_printf(("5http_sspi_make_credentials: AcquireCredentialsHandle failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), Status)));
e32497ad
MS
2024 ok = FALSE;
2025 goto cleanup;
2026 }
2027
2028cleanup:
2029
2030 /*
2031 * Cleanup
2032 */
2033
2034 if (hKey)
2035 CryptDestroyKey(hKey);
2036
2037 if (createdContext)
2038 CertFreeCertificateContext(createdContext);
2039
2040 if (storedContext)
2041 CertFreeCertificateContext(storedContext);
9870d12b 2042
e32497ad
MS
2043 if (p)
2044 free(p);
2045
2046 if (store)
2047 CertCloseStore(store, 0);
2048
2049 if (hProv)
2050 CryptReleaseContext(hProv, 0);
2051
2052 return (ok);
9870d12b 2053}
cc754834
MS
2054
2055
2056/*
2ece34a9 2057 * 'http_sspi_server()' - Negotiate a TLS connection as a server.
cc754834 2058 */
2ece34a9
MS
2059
2060static int /* O - 0 on success, -1 on failure */
2061http_sspi_server(http_t *http, /* I - HTTP connection */
2062 const char *hostname) /* I - Hostname of server */
cc754834 2063{
e32497ad
MS
2064 _http_sspi_t *sspi = http->tls; /* I - SSPI data */
2065 char common_name[512]; /* Common name for cert */
2ece34a9
MS
2066 DWORD dwSSPIFlags; /* SSL connection attributes we want */
2067 DWORD dwSSPIOutFlags; /* SSL connection attributes we got */
2068 TimeStamp tsExpiry; /* Time stamp */
2069 SECURITY_STATUS scRet; /* SSPI Status */
2070 SecBufferDesc inBuffer; /* Array of SecBuffer structs */
2071 SecBuffer inBuffers[2]; /* Security package buffer */
2072 SecBufferDesc outBuffer; /* Array of SecBuffer structs */
2073 SecBuffer outBuffers[1]; /* Security package buffer */
2074 int num = 0; /* 32 bit status value */
2075 BOOL fInitContext = TRUE; /* Has the context been init'd? */
2076 int ret = 0; /* Return value */
2077
2078
52770f63 2079 DEBUG_printf(("4http_sspi_server(http=%p, hostname=\"%s\")", http, hostname));
cc754834 2080
2ece34a9
MS
2081 dwSSPIFlags = ASC_REQ_SEQUENCE_DETECT |
2082 ASC_REQ_REPLAY_DETECT |
2083 ASC_REQ_CONFIDENTIALITY |
2084 ASC_REQ_EXTENDED_ERROR |
2085 ASC_REQ_ALLOCATE_MEMORY |
2086 ASC_REQ_STREAM;
2087
e32497ad
MS
2088 sspi->decryptBufferUsed = 0;
2089
2090 /*
2091 * Lookup the server certificate...
2092 */
2093
2094 snprintf(common_name, sizeof(common_name), "CN=%s", hostname);
2095
2096 if (!http_sspi_find_credentials(http, L"ServerContainer", common_name))
2097 if (!http_sspi_make_credentials(http->tls, L"ServerContainer", common_name, _HTTP_MODE_SERVER, 10))
2098 {
52770f63 2099 DEBUG_puts("5http_sspi_server: Unable to get server credentials.");
e32497ad
MS
2100 return (-1);
2101 }
cc754834
MS
2102
2103 /*
2ece34a9 2104 * Set OutBuffer for AcceptSecurityContext call
cc754834 2105 */
cc754834 2106
2ece34a9
MS
2107 outBuffer.cBuffers = 1;
2108 outBuffer.pBuffers = outBuffers;
2109 outBuffer.ulVersion = SECBUFFER_VERSION;
cc754834 2110
2ece34a9 2111 scRet = SEC_I_CONTINUE_NEEDED;
cc754834 2112
2ece34a9
MS
2113 while (scRet == SEC_I_CONTINUE_NEEDED ||
2114 scRet == SEC_E_INCOMPLETE_MESSAGE ||
2115 scRet == SEC_I_INCOMPLETE_CREDENTIALS)
cc754834 2116 {
e32497ad 2117 if (sspi->decryptBufferUsed == 0 || scRet == SEC_E_INCOMPLETE_MESSAGE)
cc754834 2118 {
e32497ad 2119 if (sspi->decryptBufferLength <= sspi->decryptBufferUsed)
cc754834 2120 {
2ece34a9 2121 BYTE *temp; /* New buffer */
cc754834 2122
e32497ad 2123 if (sspi->decryptBufferLength >= 262144)
2ece34a9
MS
2124 {
2125 WSASetLastError(E_OUTOFMEMORY);
52770f63 2126 DEBUG_puts("5http_sspi_server: Decryption buffer too large (>256k)");
2ece34a9
MS
2127 return (-1);
2128 }
cc754834 2129
e32497ad 2130 if ((temp = realloc(sspi->decryptBuffer, sspi->decryptBufferLength + 4096)) == NULL)
2ece34a9 2131 {
52770f63 2132 DEBUG_printf(("5http_sspi_server: Unable to allocate %d byte buffer.", sspi->decryptBufferLength + 4096));
2ece34a9
MS
2133 WSASetLastError(E_OUTOFMEMORY);
2134 return (-1);
2135 }
cc754834 2136
e32497ad
MS
2137 sspi->decryptBufferLength += 4096;
2138 sspi->decryptBuffer = temp;
2ece34a9 2139 }
cc754834 2140
2ece34a9 2141 for (;;)
cc754834 2142 {
e32497ad 2143 num = recv(http->fd, sspi->decryptBuffer + sspi->decryptBufferUsed, (int)(sspi->decryptBufferLength - sspi->decryptBufferUsed), 0);
cc754834 2144
33c9220c 2145 if (num == -1 && WSAGetLastError() == WSAEWOULDBLOCK)
2ece34a9 2146 Sleep(1);
cc754834 2147 else
2ece34a9 2148 break;
cc754834 2149 }
cc754834 2150
2ece34a9
MS
2151 if (num < 0)
2152 {
52770f63 2153 DEBUG_printf(("5http_sspi_server: recv failed: %d", WSAGetLastError()));
2ece34a9
MS
2154 return (-1);
2155 }
2156 else if (num == 0)
2157 {
52770f63 2158 DEBUG_puts("5http_sspi_server: client disconnected");
2ece34a9
MS
2159 return (-1);
2160 }
2161
52770f63 2162 DEBUG_printf(("5http_sspi_server: received %d (handshake) bytes from client.", num));
e32497ad 2163 sspi->decryptBufferUsed += num;
cc754834
MS
2164 }
2165
2166 /*
2ece34a9
MS
2167 * InBuffers[1] is for getting extra data that SSPI/SCHANNEL doesn't process
2168 * on this run around the loop.
cc754834 2169 */
2ece34a9 2170
e32497ad
MS
2171 inBuffers[0].pvBuffer = sspi->decryptBuffer;
2172 inBuffers[0].cbBuffer = (unsigned long)sspi->decryptBufferUsed;
2ece34a9
MS
2173 inBuffers[0].BufferType = SECBUFFER_TOKEN;
2174
2175 inBuffers[1].pvBuffer = NULL;
2176 inBuffers[1].cbBuffer = 0;
2177 inBuffers[1].BufferType = SECBUFFER_EMPTY;
2178
2179 inBuffer.cBuffers = 2;
2180 inBuffer.pBuffers = inBuffers;
2181 inBuffer.ulVersion = SECBUFFER_VERSION;
cc754834
MS
2182
2183 /*
2ece34a9
MS
2184 * Initialize these so if we fail, pvBuffer contains NULL, so we don't try to
2185 * free random garbage at the quit.
cc754834 2186 */
cc754834 2187
2ece34a9
MS
2188 outBuffers[0].pvBuffer = NULL;
2189 outBuffers[0].BufferType = SECBUFFER_TOKEN;
2190 outBuffers[0].cbBuffer = 0;
2191
e32497ad 2192 scRet = AcceptSecurityContext(&sspi->creds, (fInitContext?NULL:&sspi->context), &inBuffer, dwSSPIFlags, SECURITY_NATIVE_DREP, (fInitContext?&sspi->context:NULL), &outBuffer, &dwSSPIOutFlags, &tsExpiry);
cc754834 2193
2ece34a9
MS
2194 fInitContext = FALSE;
2195
2196 if (scRet == SEC_E_OK ||
2197 scRet == SEC_I_CONTINUE_NEEDED ||
2198 (FAILED(scRet) && ((dwSSPIOutFlags & ISC_RET_EXTENDED_ERROR) != 0)))
2199 {
2200 if (outBuffers[0].cbBuffer && outBuffers[0].pvBuffer)
cc754834 2201 {
2ece34a9
MS
2202 /*
2203 * Send response to server if there is one.
2204 */
cc754834 2205
2ece34a9
MS
2206 num = send(http->fd, outBuffers[0].pvBuffer, outBuffers[0].cbBuffer, 0);
2207
2208 if (num <= 0)
2209 {
52770f63 2210 DEBUG_printf(("5http_sspi_server: handshake send failed: %d", WSAGetLastError()));
2ece34a9 2211 return (-1);
cc754834
MS
2212 }
2213
52770f63 2214 DEBUG_printf(("5http_sspi_server: sent %d handshake bytes to client.", outBuffers[0].cbBuffer));
cc754834 2215
2ece34a9
MS
2216 FreeContextBuffer(outBuffers[0].pvBuffer);
2217 outBuffers[0].pvBuffer = NULL;
cc754834 2218 }
2ece34a9
MS
2219 }
2220
2221 if (scRet == SEC_E_OK)
2222 {
2223 /*
2224 * If there's extra data then save it for next time we go to decrypt.
2225 */
cc754834 2226
2ece34a9
MS
2227 if (inBuffers[1].BufferType == SECBUFFER_EXTRA)
2228 {
e32497ad
MS
2229 memcpy(sspi->decryptBuffer, (LPBYTE)(sspi->decryptBuffer + sspi->decryptBufferUsed - inBuffers[1].cbBuffer), inBuffers[1].cbBuffer);
2230 sspi->decryptBufferUsed = inBuffers[1].cbBuffer;
2ece34a9
MS
2231 }
2232 else
2233 {
e32497ad 2234 sspi->decryptBufferUsed = 0;
2ece34a9
MS
2235 }
2236 break;
cc754834 2237 }
2ece34a9 2238 else if (FAILED(scRet) && scRet != SEC_E_INCOMPLETE_MESSAGE)
cc754834 2239 {
52770f63 2240 DEBUG_printf(("5http_sspi_server: AcceptSecurityContext failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9
MS
2241 ret = -1;
2242 break;
cc754834
MS
2243 }
2244
2ece34a9
MS
2245 if (scRet != SEC_E_INCOMPLETE_MESSAGE &&
2246 scRet != SEC_I_INCOMPLETE_CREDENTIALS)
cc754834 2247 {
2ece34a9
MS
2248 if (inBuffers[1].BufferType == SECBUFFER_EXTRA)
2249 {
e32497ad
MS
2250 memcpy(sspi->decryptBuffer, (LPBYTE)(sspi->decryptBuffer + sspi->decryptBufferUsed - inBuffers[1].cbBuffer), inBuffers[1].cbBuffer);
2251 sspi->decryptBufferUsed = inBuffers[1].cbBuffer;
2ece34a9
MS
2252 }
2253 else
2254 {
e32497ad 2255 sspi->decryptBufferUsed = 0;
2ece34a9 2256 }
cc754834 2257 }
2ece34a9
MS
2258 }
2259
2260 if (!ret)
2261 {
e32497ad 2262 sspi->contextInitialized = TRUE;
2ece34a9
MS
2263
2264 /*
2265 * Find out how big the header will be:
2266 */
2267
e32497ad 2268 scRet = QueryContextAttributes(&sspi->context, SECPKG_ATTR_STREAM_SIZES, &sspi->streamSizes);
2ece34a9
MS
2269
2270 if (scRet != SEC_E_OK)
cc754834 2271 {
52770f63 2272 DEBUG_printf(("5http_sspi_server: QueryContextAttributes failed: %s", http_sspi_strerror(sspi->error, sizeof(sspi->error), scRet)));
2ece34a9 2273 ret = -1;
cc754834
MS
2274 }
2275 }
2276
2ece34a9 2277 return (ret);
cc754834
MS
2278}
2279
2280
adad9dd6
MS
2281/*
2282 * 'http_sspi_strerror()' - Return a string for the specified error code.
2283 */
2284
2285static const char * /* O - String for error */
52770f63
MS
2286http_sspi_strerror(char *buffer, /* I - Error message buffer */
2287 size_t bufsize, /* I - Size of buffer */
2288 DWORD code) /* I - Error code */
adad9dd6 2289{
52770f63 2290 if (FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, code, 0, buffer, bufsize, NULL))
adad9dd6
MS
2291 {
2292 /*
2293 * Strip trailing CR + LF...
2294 */
2295
2296 char *ptr; /* Pointer into error message */
2297
52770f63 2298 for (ptr = buffer + strlen(buffer) - 1; ptr >= buffer; ptr --)
adad9dd6
MS
2299 if (*ptr == '\n' || *ptr == '\r')
2300 *ptr = '\0';
2301 else
2302 break;
2303 }
2304 else
52770f63 2305 snprintf(buffer, bufsize, "Unknown error %x", code);
cc754834 2306
52770f63 2307 return (buffer);
cc754834
MS
2308}
2309
2310
2311/*
2ece34a9 2312 * 'http_sspi_verify()' - Verify a certificate.
cc754834 2313 */
2ece34a9
MS
2314
2315static DWORD /* O - Error code (0 == No error) */
2316http_sspi_verify(
2317 PCCERT_CONTEXT cert, /* I - Server certificate */
2318 const char *common_name, /* I - Common name */
2319 DWORD dwCertFlags) /* I - Verification flags */
cc754834 2320{
2ece34a9
MS
2321 HTTPSPolicyCallbackData httpsPolicy; /* HTTPS Policy Struct */
2322 CERT_CHAIN_POLICY_PARA policyPara; /* Cert chain policy parameters */
2323 CERT_CHAIN_POLICY_STATUS policyStatus;/* Cert chain policy status */
2324 CERT_CHAIN_PARA chainPara; /* Used for searching and matching criteria */
2325 PCCERT_CHAIN_CONTEXT chainContext = NULL;
cc754834 2326 /* Certificate chain */
2ece34a9
MS
2327 PWSTR commonNameUnicode = NULL;
2328 /* Unicode common name */
2329 LPSTR rgszUsages[] = { szOID_PKIX_KP_SERVER_AUTH,
2330 szOID_SERVER_GATED_CRYPTO,
2331 szOID_SGC_NETSCAPE };
cc754834 2332 /* How are we using this certificate? */
2ece34a9 2333 DWORD cUsages = sizeof(rgszUsages) / sizeof(LPSTR);
cc754834 2334 /* Number of ites in rgszUsages */
2ece34a9
MS
2335 DWORD count; /* 32 bit count variable */
2336 DWORD status; /* Return value */
74ac5fe7
MS
2337#ifdef DEBUG
2338 char error[1024]; /* Error message string */
2339#endif /* DEBUG */
cc754834 2340
2ece34a9
MS
2341
2342 if (!cert)
2343 return (SEC_E_WRONG_PRINCIPAL);
cc754834
MS
2344
2345 /*
2ece34a9 2346 * Convert common name to Unicode.
cc754834 2347 */
cc754834 2348
2ece34a9
MS
2349 if (!common_name || !*common_name)
2350 return (SEC_E_WRONG_PRINCIPAL);
2351
2352 count = MultiByteToWideChar(CP_ACP, 0, common_name, -1, NULL, 0);
2353 commonNameUnicode = LocalAlloc(LMEM_FIXED, count * sizeof(WCHAR));
2354 if (!commonNameUnicode)
2355 return (SEC_E_INSUFFICIENT_MEMORY);
2356
2357 if (!MultiByteToWideChar(CP_ACP, 0, common_name, -1, commonNameUnicode, count))
cc754834 2358 {
2ece34a9
MS
2359 LocalFree(commonNameUnicode);
2360 return (SEC_E_WRONG_PRINCIPAL);
cc754834
MS
2361 }
2362
2363 /*
2364 * Build certificate chain.
2365 */
2ece34a9 2366
cc754834 2367 ZeroMemory(&chainPara, sizeof(chainPara));
2ece34a9 2368
cc754834
MS
2369 chainPara.cbSize = sizeof(chainPara);
2370 chainPara.RequestedUsage.dwType = USAGE_MATCH_TYPE_OR;
2371 chainPara.RequestedUsage.Usage.cUsageIdentifier = cUsages;
2372 chainPara.RequestedUsage.Usage.rgpszUsageIdentifier = rgszUsages;
2373
2ece34a9 2374 if (!CertGetCertificateChain(NULL, cert, NULL, cert->hCertStore, &chainPara, 0, NULL, &chainContext))
cc754834
MS
2375 {
2376 status = GetLastError();
52770f63 2377
52770f63 2378 DEBUG_printf(("CertGetCertificateChain returned: %s", http_sspi_strerror(error, sizeof(error), status)));
2ece34a9
MS
2379
2380 LocalFree(commonNameUnicode);
2381 return (status);
cc754834
MS
2382 }
2383
2384 /*
2385 * Validate certificate chain.
321d8d57 2386 */
2ece34a9 2387
cc754834
MS
2388 ZeroMemory(&httpsPolicy, sizeof(HTTPSPolicyCallbackData));
2389 httpsPolicy.cbStruct = sizeof(HTTPSPolicyCallbackData);
2390 httpsPolicy.dwAuthType = AUTHTYPE_SERVER;
2391 httpsPolicy.fdwChecks = dwCertFlags;
2ece34a9 2392 httpsPolicy.pwszServerName = commonNameUnicode;
cc754834
MS
2393
2394 memset(&policyPara, 0, sizeof(policyPara));
2395 policyPara.cbSize = sizeof(policyPara);
2396 policyPara.pvExtraPolicyPara = &httpsPolicy;
2397
2398 memset(&policyStatus, 0, sizeof(policyStatus));
2399 policyStatus.cbSize = sizeof(policyStatus);
2400
2ece34a9 2401 if (!CertVerifyCertificateChainPolicy(CERT_CHAIN_POLICY_SSL, chainContext, &policyPara, &policyStatus))
cc754834
MS
2402 {
2403 status = GetLastError();
52770f63 2404
52770f63 2405 DEBUG_printf(("CertVerifyCertificateChainPolicy returned %s", http_sspi_strerror(error, sizeof(error), status)));
cc754834 2406 }
2ece34a9 2407 else if (policyStatus.dwError)
cc754834 2408 status = policyStatus.dwError;
2ece34a9
MS
2409 else
2410 status = SEC_E_OK;
cc754834
MS
2411
2412 if (chainContext)
2413 CertFreeCertificateChain(chainContext);
2414
2ece34a9
MS
2415 if (commonNameUnicode)
2416 LocalFree(commonNameUnicode);
cc754834
MS
2417
2418 return (status);
2419}
e32497ad
MS
2420
2421
321d8d57
MS
2422/*
2423 * End of "$Id$".
2424 */