d4221cbae3f24e6a18a38f5ea4db7815e7cc71d6
[openssl.git] / engines / e_capi.c
1 /* engines/e_capi.c */
2 /* Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
3  * project.
4  */
5 /* ====================================================================
6  * Copyright (c) 2008 The OpenSSL Project.  All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer. 
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in
17  *    the documentation and/or other materials provided with the
18  *    distribution.
19  *
20  * 3. All advertising materials mentioning features or use of this
21  *    software must display the following acknowledgment:
22  *    "This product includes software developed by the OpenSSL Project
23  *    for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
24  *
25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
26  *    endorse or promote products derived from this software without
27  *    prior written permission. For written permission, please contact
28  *    licensing@OpenSSL.org.
29  *
30  * 5. Products derived from this software may not be called "OpenSSL"
31  *    nor may "OpenSSL" appear in their names without prior written
32  *    permission of the OpenSSL Project.
33  *
34  * 6. Redistributions of any form whatsoever must retain the following
35  *    acknowledgment:
36  *    "This product includes software developed by the OpenSSL Project
37  *    for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
38  *
39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
50  * OF THE POSSIBILITY OF SUCH DAMAGE.
51  * ====================================================================
52  */
53
54
55 #include <stdio.h>
56 #include <string.h>
57 #include <stdlib.h>
58
59 #include <openssl/crypto.h>
60
61 #ifdef OPENSSL_SYS_WIN32
62 #ifndef OPENSSL_NO_CAPIENG
63
64 #include <openssl/buffer.h>
65 #include <openssl/bn.h>
66 #include <openssl/rsa.h>
67
68 #ifndef _WIN32_WINNT
69 #define _WIN32_WINNT 0x0400
70 #endif
71
72 #include <windows.h>
73 #include <wincrypt.h>
74 #include <malloc.h>
75 #ifndef alloca
76 # define alloca _alloca
77 #endif
78
79 /*
80  * This module uses several "new" interfaces, among which is
81  * CertGetCertificateContextProperty. CERT_KEY_PROV_INFO_PROP_ID is
82  * one of possible values you can pass to function in question. By
83  * checking if it's defined we can see if wincrypt.h and accompanying
84  * crypt32.lib are in shape. The native MingW32 headers up to and
85  * including __W32API_VERSION 3.14 lack of struct DSSPUBKEY and the
86  * defines CERT_STORE_PROV_SYSTEM_A and CERT_STORE_READONLY_FLAG,
87  * so we check for these too and avoid compiling.
88  * Yes, it's rather "weak" test and if compilation fails,
89  * then re-configure with -DOPENSSL_NO_CAPIENG.
90  */
91 #if defined(CERT_KEY_PROV_INFO_PROP_ID) && \
92     defined(CERT_STORE_PROV_SYSTEM_A) && \
93     defined(CERT_STORE_READONLY_FLAG)
94 # define __COMPILE_CAPIENG
95 #endif /* CERT_KEY_PROV_INFO_PROP_ID */
96 #endif /* OPENSSL_NO_CAPIENG */
97 #endif /* OPENSSL_SYS_WIN32 */
98
99 #ifdef __COMPILE_CAPIENG
100
101 #undef X509_EXTENSIONS
102 #undef X509_CERT_PAIR
103
104 /* Definitions which may be missing from earlier version of headers */
105 #ifndef CERT_STORE_OPEN_EXISTING_FLAG
106 #define CERT_STORE_OPEN_EXISTING_FLAG                   0x00004000
107 #endif
108
109 #ifndef CERT_STORE_CREATE_NEW_FLAG
110 #define CERT_STORE_CREATE_NEW_FLAG                      0x00002000
111 #endif
112
113 #ifndef CERT_SYSTEM_STORE_CURRENT_USER
114 #define CERT_SYSTEM_STORE_CURRENT_USER                  0x00010000
115 #endif 
116
117 #ifndef ALG_SID_SHA_256
118         #define ALG_SID_SHA_256                 12
119 #endif
120 #ifndef ALG_SID_SHA_384
121         #define ALG_SID_SHA_384                 13
122 #endif
123 #ifndef ALG_SID_SHA_512
124         #define ALG_SID_SHA_512                 14
125 #endif
126
127 #ifndef CALG_SHA_256
128         #define CALG_SHA_256            (ALG_CLASS_HASH | ALG_TYPE_ANY | ALG_SID_SHA_256)
129 #endif
130 #ifndef CALG_SHA_384
131         #define CALG_SHA_384            (ALG_CLASS_HASH | ALG_TYPE_ANY | ALG_SID_SHA_384)
132 #endif
133 #ifndef CALG_SHA_512
134         #define CALG_SHA_512            (ALG_CLASS_HASH | ALG_TYPE_ANY | ALG_SID_SHA_512)
135 #endif
136
137 #include <openssl/engine.h>
138 #include <openssl/pem.h>
139 #include <openssl/x509v3.h>
140
141 #include "e_capi_err.h"
142 #include "e_capi_err.c"
143
144
145 static const char *engine_capi_id = "capi";
146 static const char *engine_capi_name = "CryptoAPI ENGINE";
147
148 typedef struct CAPI_CTX_st CAPI_CTX;
149 typedef struct CAPI_KEY_st CAPI_KEY;
150
151 static void capi_addlasterror(void);
152 static void capi_adderror(DWORD err);
153
154 static void CAPI_trace(CAPI_CTX *ctx, char *format, ...);
155
156 static int capi_list_providers(CAPI_CTX *ctx, BIO *out);
157 static int capi_list_containers(CAPI_CTX *ctx, BIO *out);
158 int capi_list_certs(CAPI_CTX *ctx, BIO *out, char *storename);
159 void capi_free_key(CAPI_KEY *key);
160
161 static PCCERT_CONTEXT capi_find_cert(CAPI_CTX *ctx, const char *id, HCERTSTORE hstore);
162
163 CAPI_KEY *capi_find_key(CAPI_CTX *ctx, const char *id);
164
165 static EVP_PKEY *capi_load_privkey(ENGINE *eng, const char *key_id,
166         UI_METHOD *ui_method, void *callback_data);
167 static int capi_rsa_sign(int dtype, const unsigned char *m, unsigned int m_len,
168              unsigned char *sigret, unsigned int *siglen, const RSA *rsa);
169 static int capi_rsa_priv_enc(int flen, const unsigned char *from,
170                 unsigned char *to, RSA *rsa, int padding);
171 static int capi_rsa_priv_dec(int flen, const unsigned char *from,
172                 unsigned char *to, RSA *rsa, int padding);
173 static int capi_rsa_free(RSA *rsa);
174
175 static DSA_SIG *capi_dsa_do_sign(const unsigned char *digest, int dlen,
176                                                         DSA *dsa);
177 static int capi_dsa_free(DSA *dsa);
178
179 static int capi_load_ssl_client_cert(ENGINE *e, SSL *ssl,
180         STACK_OF(X509_NAME) *ca_dn, X509 **pcert, EVP_PKEY **pkey,
181         STACK_OF(X509) **pother, UI_METHOD *ui_method, void *callback_data);
182
183 static int cert_select_simple(ENGINE *e, SSL *ssl, STACK_OF(X509) *certs);
184 #ifdef OPENSSL_CAPIENG_DIALOG
185 static int cert_select_dialog(ENGINE *e, SSL *ssl, STACK_OF(X509) *certs);
186 #endif
187
188 typedef PCCERT_CONTEXT (WINAPI *CERTDLG)(HCERTSTORE, HWND, LPCWSTR,
189                                                 LPCWSTR, DWORD, DWORD,
190                                                 void *);
191 typedef HWND (WINAPI *GETCONSWIN)(void);
192
193 /* This structure contains CAPI ENGINE specific data:
194  * it contains various global options and affects how
195  * other functions behave.
196  */
197
198 #define CAPI_DBG_TRACE  2
199 #define CAPI_DBG_ERROR  1
200
201 struct CAPI_CTX_st {
202         int debug_level;
203         char *debug_file;
204         /* Parameters to use for container lookup */
205         DWORD keytype;
206         LPSTR cspname;
207         DWORD csptype;
208         /* Certificate store name to use */
209         LPSTR storename;
210         LPSTR ssl_client_store;
211         /* System store flags */
212         DWORD store_flags;
213
214 /* Lookup string meanings in load_private_key */
215 /* Substring of subject: uses "storename" */
216 #define CAPI_LU_SUBSTR          1
217 /* Friendly name: uses storename */
218 #define CAPI_LU_FNAME           2
219 /* Container name: uses cspname, keytype */
220 #define CAPI_LU_CONTNAME        3
221         int lookup_method;
222 /* Info to dump with dumpcerts option */
223 /* Issuer and serial name strings */
224 #define CAPI_DMP_SUMMARY        0x1
225 /* Friendly name */
226 #define CAPI_DMP_FNAME          0x2
227 /* Full X509_print dump */
228 #define CAPI_DMP_FULL           0x4
229 /* Dump PEM format certificate */
230 #define CAPI_DMP_PEM            0x8
231 /* Dump pseudo key (if possible) */
232 #define CAPI_DMP_PSKEY          0x10
233 /* Dump key info (if possible) */
234 #define CAPI_DMP_PKEYINFO       0x20
235
236         DWORD dump_flags;
237         int (*client_cert_select)(ENGINE *e, SSL *ssl, STACK_OF(X509) *certs);
238
239         CERTDLG certselectdlg;
240         GETCONSWIN getconswindow;
241 };
242
243
244 static CAPI_CTX *capi_ctx_new();
245 static void capi_ctx_free(CAPI_CTX *ctx);
246 static int capi_ctx_set_provname(CAPI_CTX *ctx, LPSTR pname, DWORD type, int check);
247 static int capi_ctx_set_provname_idx(CAPI_CTX *ctx, int idx);
248
249 #define CAPI_CMD_LIST_CERTS             ENGINE_CMD_BASE
250 #define CAPI_CMD_LOOKUP_CERT            (ENGINE_CMD_BASE + 1)
251 #define CAPI_CMD_DEBUG_LEVEL            (ENGINE_CMD_BASE + 2)
252 #define CAPI_CMD_DEBUG_FILE             (ENGINE_CMD_BASE + 3)
253 #define CAPI_CMD_KEYTYPE                (ENGINE_CMD_BASE + 4)
254 #define CAPI_CMD_LIST_CSPS              (ENGINE_CMD_BASE + 5)
255 #define CAPI_CMD_SET_CSP_IDX            (ENGINE_CMD_BASE + 6)
256 #define CAPI_CMD_SET_CSP_NAME           (ENGINE_CMD_BASE + 7)
257 #define CAPI_CMD_SET_CSP_TYPE           (ENGINE_CMD_BASE + 8)
258 #define CAPI_CMD_LIST_CONTAINERS        (ENGINE_CMD_BASE + 9)
259 #define CAPI_CMD_LIST_OPTIONS           (ENGINE_CMD_BASE + 10)
260 #define CAPI_CMD_LOOKUP_METHOD          (ENGINE_CMD_BASE + 11)
261 #define CAPI_CMD_STORE_NAME             (ENGINE_CMD_BASE + 12)
262 #define CAPI_CMD_STORE_FLAGS            (ENGINE_CMD_BASE + 13)
263
264 static const ENGINE_CMD_DEFN capi_cmd_defns[] = {
265         {CAPI_CMD_LIST_CERTS,
266                 "list_certs",
267                 "List all certificates in store",
268                 ENGINE_CMD_FLAG_NO_INPUT},
269         {CAPI_CMD_LOOKUP_CERT,
270                 "lookup_cert",
271                 "Lookup and output certificates",
272                 ENGINE_CMD_FLAG_STRING},
273         {CAPI_CMD_DEBUG_LEVEL,
274                 "debug_level",
275                 "debug level (1=errors, 2=trace)",
276                 ENGINE_CMD_FLAG_NUMERIC},
277         {CAPI_CMD_DEBUG_FILE,
278                 "debug_file",
279                 "debugging filename)",
280                 ENGINE_CMD_FLAG_STRING},
281         {CAPI_CMD_KEYTYPE,
282                 "key_type",
283                 "Key type: 1=AT_KEYEXCHANGE (default), 2=AT_SIGNATURE",
284                 ENGINE_CMD_FLAG_NUMERIC},
285         {CAPI_CMD_LIST_CSPS,
286                 "list_csps",
287                 "List all CSPs",
288                 ENGINE_CMD_FLAG_NO_INPUT},
289         {CAPI_CMD_SET_CSP_IDX,
290                 "csp_idx",
291                 "Set CSP by index",
292                 ENGINE_CMD_FLAG_NUMERIC},
293         {CAPI_CMD_SET_CSP_NAME,
294                 "csp_name",
295                 "Set CSP name, (default CSP used if not specified)",
296                 ENGINE_CMD_FLAG_STRING},
297         {CAPI_CMD_SET_CSP_TYPE,
298                 "csp_type",
299                 "Set CSP type, (default RSA_PROV_FULL)",
300                 ENGINE_CMD_FLAG_NUMERIC},
301         {CAPI_CMD_LIST_CONTAINERS,
302                 "list_containers",
303                 "list container names",
304                 ENGINE_CMD_FLAG_NO_INPUT},
305         {CAPI_CMD_LIST_OPTIONS,
306                 "list_options",
307                 "Set list options (1=summary,2=friendly name, 4=full printout, 8=PEM output, 16=XXX, "
308                 "32=private key info)",
309                 ENGINE_CMD_FLAG_NUMERIC},
310         {CAPI_CMD_LOOKUP_METHOD,
311                 "lookup_method",
312                 "Set key lookup method (1=substring, 2=friendlyname, 3=container name)",
313                 ENGINE_CMD_FLAG_NUMERIC},
314         {CAPI_CMD_STORE_NAME,
315                 "store_name",
316                 "certificate store name, default \"MY\"",
317                 ENGINE_CMD_FLAG_STRING},
318         {CAPI_CMD_STORE_FLAGS,
319                 "store_flags",
320                 "Certificate store flags: 1 = system store",
321                 ENGINE_CMD_FLAG_NUMERIC},
322
323         {0, NULL, NULL, 0}
324         };
325
326 static int capi_idx = -1;
327 static int rsa_capi_idx = -1;
328 static int dsa_capi_idx = -1;
329 static int cert_capi_idx = -1;
330
331 static int capi_ctrl(ENGINE *e, int cmd, long i, void *p, void (*f)(void))
332         {
333         int ret = 1;
334         CAPI_CTX *ctx;
335         BIO *out;
336         if (capi_idx == -1)
337                 {
338                 CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_ENGINE_NOT_INITIALIZED);
339                 return 0;
340                 }
341         ctx = ENGINE_get_ex_data(e, capi_idx);
342         out = BIO_new_fp(stdout, BIO_NOCLOSE);
343         switch (cmd)
344                 {
345                 case CAPI_CMD_LIST_CSPS:
346                 ret = capi_list_providers(ctx, out);
347                 break;
348
349                 case CAPI_CMD_LIST_CERTS:
350                 ret = capi_list_certs(ctx, out, NULL);
351                 break;
352
353                 case CAPI_CMD_LOOKUP_CERT:
354                 ret = capi_list_certs(ctx, out, p);
355                 break;
356
357                 case CAPI_CMD_LIST_CONTAINERS:
358                 ret = capi_list_containers(ctx, out);
359                 break;
360
361                 case CAPI_CMD_STORE_NAME:
362                 if (ctx->storename)
363                         OPENSSL_free(ctx->storename);
364                 ctx->storename = BUF_strdup(p);
365                 CAPI_trace(ctx, "Setting store name to %s\n", p);
366                 break;
367
368                 case CAPI_CMD_STORE_FLAGS:
369                 if (i & 1)
370                         {
371                         ctx->store_flags |= CERT_SYSTEM_STORE_LOCAL_MACHINE;
372                         ctx->store_flags &= ~CERT_SYSTEM_STORE_CURRENT_USER;
373                         }
374                 else
375                         {
376                         ctx->store_flags |= CERT_SYSTEM_STORE_CURRENT_USER;
377                         ctx->store_flags &= ~CERT_SYSTEM_STORE_LOCAL_MACHINE;
378                         }
379                 CAPI_trace(ctx, "Setting flags to %d\n", i);
380                 break;
381
382                 case CAPI_CMD_DEBUG_LEVEL:
383                 ctx->debug_level = (int)i;
384                 CAPI_trace(ctx, "Setting debug level to %d\n", ctx->debug_level);
385                 break;
386
387                 case CAPI_CMD_DEBUG_FILE:
388                 ctx->debug_file = BUF_strdup(p);
389                 CAPI_trace(ctx, "Setting debug file to %s\n", ctx->debug_file);
390                 break;
391
392                 case CAPI_CMD_KEYTYPE:
393                 ctx->keytype = i;
394                 CAPI_trace(ctx, "Setting key type to %d\n", ctx->keytype);
395                 break;
396
397                 case CAPI_CMD_SET_CSP_IDX:
398                 ret = capi_ctx_set_provname_idx(ctx, i);
399                 break;
400
401                 case CAPI_CMD_LIST_OPTIONS:
402                 ctx->dump_flags = i;
403                 break;
404
405                 case CAPI_CMD_LOOKUP_METHOD:
406                 if (i < 1 || i > 3)
407                         {
408                         CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_INVALID_LOOKUP_METHOD);
409                         return 0;
410                         }
411                 ctx->lookup_method = i;
412                 break;
413
414                 case CAPI_CMD_SET_CSP_NAME:
415                 ret = capi_ctx_set_provname(ctx, p, ctx->csptype, 1);
416                 break;
417
418                 case CAPI_CMD_SET_CSP_TYPE:
419                 ctx->csptype = i;
420                 break;
421
422                 default:
423                 CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_UNKNOWN_COMMAND);
424                 ret = 0;
425         }
426
427         BIO_free(out);
428         return ret;
429
430         }
431
432 static RSA_METHOD capi_rsa_method =
433         {
434         "CryptoAPI RSA method",
435         0,                              /* pub_enc */
436         0,                              /* pub_dec */
437         capi_rsa_priv_enc,              /* priv_enc */
438         capi_rsa_priv_dec,              /* priv_dec */
439         0,                              /* rsa_mod_exp */
440         0,                              /* bn_mod_exp */
441         0,                              /* init */
442         capi_rsa_free,                  /* finish */
443         RSA_FLAG_SIGN_VER,              /* flags */
444         NULL,                           /* app_data */
445         capi_rsa_sign,                  /* rsa_sign */
446         0                               /* rsa_verify */
447         };
448
449 static DSA_METHOD capi_dsa_method =
450         {
451         "CryptoAPI DSA method",
452         capi_dsa_do_sign,               /* dsa_do_sign */
453         0,                              /* dsa_sign_setup */
454         0,                              /* dsa_do_verify */
455         0,                              /* dsa_mod_exp */
456         0,                              /* bn_mod_exp */
457         0,                              /* init */
458         capi_dsa_free,                  /* finish */
459         0,                              /* flags */
460         NULL,                           /* app_data */
461         0,                              /* dsa_paramgen */
462         0                               /* dsa_keygen */
463         };
464
465 static int capi_init(ENGINE *e)
466         {
467         CAPI_CTX *ctx;
468         const RSA_METHOD *ossl_rsa_meth;
469         const DSA_METHOD *ossl_dsa_meth;
470
471         if (capi_idx < 0)
472                 {
473                 capi_idx = ENGINE_get_ex_new_index(0, NULL, NULL, NULL, 0);
474                 if (capi_idx < 0)
475                         goto memerr;
476
477                 cert_capi_idx = X509_get_ex_new_index(0, NULL, NULL, NULL, 0);
478
479                 /* Setup RSA_METHOD */
480                 rsa_capi_idx = RSA_get_ex_new_index(0, NULL, NULL, NULL, 0);
481                 ossl_rsa_meth = RSA_PKCS1_SSLeay();
482                 capi_rsa_method.rsa_pub_enc = ossl_rsa_meth->rsa_pub_enc;
483                 capi_rsa_method.rsa_pub_dec = ossl_rsa_meth->rsa_pub_dec;
484                 capi_rsa_method.rsa_mod_exp = ossl_rsa_meth->rsa_mod_exp;
485                 capi_rsa_method.bn_mod_exp = ossl_rsa_meth->bn_mod_exp;
486
487                 /* Setup DSA Method */
488                 dsa_capi_idx = DSA_get_ex_new_index(0, NULL, NULL, NULL, 0);
489                 ossl_dsa_meth = DSA_OpenSSL();
490                 capi_dsa_method.dsa_do_verify = ossl_dsa_meth->dsa_do_verify;
491                 capi_dsa_method.dsa_mod_exp = ossl_dsa_meth->dsa_mod_exp;
492                 capi_dsa_method.bn_mod_exp = ossl_dsa_meth->bn_mod_exp;
493                 }
494
495         ctx = capi_ctx_new();
496         if (!ctx)
497                 goto memerr;
498
499         ENGINE_set_ex_data(e, capi_idx, ctx);
500
501 #ifdef OPENSSL_CAPIENG_DIALOG
502         {
503         HMODULE cryptui = LoadLibrary(TEXT("CRYPTUI.DLL"));
504         HMODULE kernel = GetModuleHandle(TEXT("KERNEL32.DLL"));
505         if (cryptui)
506                 ctx->certselectdlg = (CERTDLG)GetProcAddress(cryptui, "CryptUIDlgSelectCertificateFromStore");
507         if (kernel)
508                 ctx->getconswindow = (GETCONSWIN)GetProcAddress(kernel, "GetConsoleWindow");
509         if (cryptui && !OPENSSL_isservice())
510                 ctx->client_cert_select = cert_select_dialog;
511         }
512 #endif
513                 
514
515         return 1;
516
517         memerr:
518         CAPIerr(CAPI_F_CAPI_INIT, ERR_R_MALLOC_FAILURE);
519         return 0;
520
521         return 1;
522         }
523
524 static int capi_destroy(ENGINE *e)
525         {
526         ERR_unload_CAPI_strings();
527         return 1;
528         }
529
530 static int capi_finish(ENGINE *e)
531         {
532         CAPI_CTX *ctx;
533         ctx = ENGINE_get_ex_data(e, capi_idx);
534         capi_ctx_free(ctx);
535         ENGINE_set_ex_data(e, capi_idx, NULL);
536         return 1;
537         }
538
539
540 /* CryptoAPI key application data. This contains
541  * a handle to the private key container (for sign operations)
542  * and a handle to the key (for decrypt operations).
543  */
544
545 struct CAPI_KEY_st
546         {
547         /* Associated certificate context (if any) */
548         PCCERT_CONTEXT pcert;
549         HCRYPTPROV hprov;
550         HCRYPTKEY key;
551         DWORD keyspec;
552         };
553
554 static int bind_capi(ENGINE *e)
555         {
556         if (!ENGINE_set_id(e, engine_capi_id)
557                 || !ENGINE_set_name(e, engine_capi_name)
558                 || !ENGINE_set_flags(e, ENGINE_FLAGS_NO_REGISTER_ALL)
559                 || !ENGINE_set_init_function(e, capi_init)
560                 || !ENGINE_set_finish_function(e, capi_finish)
561                 || !ENGINE_set_destroy_function(e, capi_destroy)
562                 || !ENGINE_set_RSA(e, &capi_rsa_method)
563                 || !ENGINE_set_DSA(e, &capi_dsa_method)
564                 || !ENGINE_set_load_privkey_function(e, capi_load_privkey)
565                 || !ENGINE_set_load_ssl_client_cert_function(e,
566                                                 capi_load_ssl_client_cert)
567                 || !ENGINE_set_cmd_defns(e, capi_cmd_defns)
568                 || !ENGINE_set_ctrl_function(e, capi_ctrl))
569                         return 0;
570         ERR_load_CAPI_strings();
571
572         return 1;
573
574         }
575
576 #ifndef OPENSSL_NO_DYNAMIC_ENGINE
577 static int bind_helper(ENGINE *e, const char *id)
578         {
579         if(id && (strcmp(id, engine_capi_id) != 0))
580                 return 0;
581         if(!bind_capi(e))
582                 return 0;
583         return 1;
584         }       
585 IMPLEMENT_DYNAMIC_CHECK_FN()
586 IMPLEMENT_DYNAMIC_BIND_FN(bind_helper)
587 #else
588 static ENGINE *engine_capi(void)
589         {
590         ENGINE *ret = ENGINE_new();
591         if(!ret)
592                 return NULL;
593         if(!bind_capi(ret))
594                 {
595                 ENGINE_free(ret);
596                 return NULL;
597                 }
598         return ret;
599         }
600
601 void ENGINE_load_capi(void)
602         {
603         /* Copied from eng_[openssl|dyn].c */
604         ENGINE *toadd = engine_capi();
605         if(!toadd) return;
606         ENGINE_add(toadd);
607         ENGINE_free(toadd);
608         ERR_clear_error();
609         }
610 #endif
611
612
613 static int lend_tobn(BIGNUM *bn, unsigned char *bin, int binlen)
614         {
615         int i;
616         /* Reverse buffer in place: since this is a keyblob structure
617          * that will be freed up after conversion anyway it doesn't 
618          * matter if we change it.
619          */
620         for(i = 0; i < binlen / 2; i++)
621                 {
622                 unsigned char c;
623                 c = bin[i];
624                 bin[i] = bin[binlen - i - 1];
625                 bin[binlen - i - 1] = c;
626                 }
627
628         if (!BN_bin2bn(bin, binlen, bn))
629                 return 0;
630         return 1;
631         }
632
633 /* Given a CAPI_KEY get an EVP_PKEY structure */
634
635 static EVP_PKEY *capi_get_pkey(ENGINE *eng, CAPI_KEY *key)
636         {
637         unsigned char *pubkey = NULL;
638         DWORD len;
639         BLOBHEADER *bh;
640         RSA *rkey = NULL;
641         DSA *dkey = NULL;
642         EVP_PKEY *ret = NULL;
643         if (!CryptExportKey(key->key, 0, PUBLICKEYBLOB, 0, NULL, &len))
644                 {
645                 CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_PUBKEY_EXPORT_LENGTH_ERROR);
646                 capi_addlasterror();
647                 return NULL;
648                 }
649
650         pubkey = OPENSSL_malloc(len);
651
652         if (!pubkey)
653                 goto memerr;
654
655         if (!CryptExportKey(key->key, 0, PUBLICKEYBLOB, 0, pubkey, &len))
656                 {
657                 CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_PUBKEY_EXPORT_ERROR);
658                 capi_addlasterror();
659                 goto err;
660                 }
661
662         bh = (BLOBHEADER *)pubkey;
663         if (bh->bType != PUBLICKEYBLOB)
664                 {
665                 CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_INVALID_PUBLIC_KEY_BLOB);
666                 goto err;
667                 }
668         if (bh->aiKeyAlg == CALG_RSA_SIGN || bh->aiKeyAlg == CALG_RSA_KEYX)
669                 {
670                 RSAPUBKEY *rp;
671                 DWORD rsa_modlen;
672                 unsigned char *rsa_modulus;
673                 rp = (RSAPUBKEY *)(bh + 1);
674                 if (rp->magic != 0x31415352)
675                         {
676                         char magstr[10];
677                         BIO_snprintf(magstr, 10, "%lx", rp->magic);
678                         CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_INVALID_RSA_PUBLIC_KEY_BLOB_MAGIC_NUMBER);
679                         ERR_add_error_data(2, "magic=0x", magstr);
680                         goto err;
681                         }
682                 rsa_modulus = (unsigned char *)(rp + 1);
683                 rkey = RSA_new_method(eng);
684                 if (!rkey)
685                         goto memerr;
686
687                 rkey->e = BN_new();
688                 rkey->n = BN_new();
689
690                 if (!rkey->e || !rkey->n)
691                         goto memerr;
692
693                 if (!BN_set_word(rkey->e, rp->pubexp))
694                         goto memerr;
695
696                 rsa_modlen = rp->bitlen / 8;
697                 if (!lend_tobn(rkey->n, rsa_modulus, rsa_modlen))
698                         goto memerr;
699
700                 RSA_set_ex_data(rkey, rsa_capi_idx, key);
701
702                 if (!(ret = EVP_PKEY_new()))
703                         goto memerr;
704
705                 EVP_PKEY_assign_RSA(ret, rkey);
706                 rkey = NULL;
707
708                 }
709         else if (bh->aiKeyAlg == CALG_DSS_SIGN)
710                 {
711                 DSSPUBKEY *dp;
712                 DWORD dsa_plen;
713                 unsigned char *btmp;
714                 dp = (DSSPUBKEY *)(bh + 1);
715                 if (dp->magic != 0x31535344)
716                         {
717                         char magstr[10];
718                         BIO_snprintf(magstr, 10, "%lx", dp->magic);
719                         CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_INVALID_DSA_PUBLIC_KEY_BLOB_MAGIC_NUMBER);
720                         ERR_add_error_data(2, "magic=0x", magstr);
721                         goto err;
722                         }
723                 dsa_plen = dp->bitlen / 8;
724                 btmp = (unsigned char *)(dp + 1);
725                 dkey = DSA_new_method(eng);
726                 if (!dkey)
727                         goto memerr;
728                 dkey->p = BN_new();
729                 dkey->q = BN_new();
730                 dkey->g = BN_new();
731                 dkey->pub_key = BN_new();
732                 if (!dkey->p || !dkey->q || !dkey->g || !dkey->pub_key)
733                         goto memerr;
734                 if (!lend_tobn(dkey->p, btmp, dsa_plen))
735                         goto memerr;
736                 btmp += dsa_plen;
737                 if (!lend_tobn(dkey->q, btmp, 20))
738                         goto memerr;
739                 btmp += 20;
740                 if (!lend_tobn(dkey->g, btmp, dsa_plen))
741                         goto memerr;
742                 btmp += dsa_plen;
743                 if (!lend_tobn(dkey->pub_key, btmp, dsa_plen))
744                         goto memerr;
745                 btmp += dsa_plen;
746
747                 DSA_set_ex_data(dkey, dsa_capi_idx, key);
748
749                 if (!(ret = EVP_PKEY_new()))
750                         goto memerr;
751
752                 EVP_PKEY_assign_DSA(ret, dkey);
753                 dkey = NULL;
754                 }
755         else
756                 {
757                 char algstr[10];
758                 BIO_snprintf(algstr, 10, "%lx", bh->aiKeyAlg);
759                 CAPIerr(CAPI_F_CAPI_GET_PKEY, CAPI_R_UNSUPPORTED_PUBLIC_KEY_ALGORITHM);
760                 ERR_add_error_data(2, "aiKeyAlg=0x", algstr);
761                 goto err;
762                 }
763
764
765         err:
766         if (pubkey)
767                 OPENSSL_free(pubkey);
768         if (!ret)
769                 {
770                 if (rkey)
771                         RSA_free(rkey);
772                 if (dkey)
773                         DSA_free(dkey);
774                 }
775
776         return ret;
777
778 memerr:
779         CAPIerr(CAPI_F_CAPI_GET_PKEY, ERR_R_MALLOC_FAILURE);
780         goto err;
781
782         }
783
784 static EVP_PKEY *capi_load_privkey(ENGINE *eng, const char *key_id,
785         UI_METHOD *ui_method, void *callback_data)
786         {
787         CAPI_CTX *ctx;
788         CAPI_KEY *key;
789         EVP_PKEY *ret;
790         ctx = ENGINE_get_ex_data(eng, capi_idx);
791
792         if (!ctx)
793                 {
794                 CAPIerr(CAPI_F_CAPI_LOAD_PRIVKEY, CAPI_R_CANT_FIND_CAPI_CONTEXT);
795                 return NULL;
796                 }
797
798         key = capi_find_key(ctx, key_id);
799
800         if (!key)
801                 return NULL;
802
803         ret = capi_get_pkey(eng, key);
804
805         if (!ret)
806                 capi_free_key(key);
807         return ret;
808
809         }
810
811 /* CryptoAPI RSA operations */
812
813 int capi_rsa_priv_enc(int flen, const unsigned char *from,
814                 unsigned char *to, RSA *rsa, int padding)
815         {
816         CAPIerr(CAPI_F_CAPI_RSA_PRIV_ENC, CAPI_R_FUNCTION_NOT_SUPPORTED);
817         return -1;
818         }
819
820 int capi_rsa_sign(int dtype, const unsigned char *m, unsigned int m_len,
821              unsigned char *sigret, unsigned int *siglen, const RSA *rsa)
822         {
823         ALG_ID alg;
824         HCRYPTHASH hash;
825         DWORD slen;
826         unsigned int i;
827         int ret = -1;
828         CAPI_KEY *capi_key;
829         CAPI_CTX *ctx;
830
831         ctx = ENGINE_get_ex_data(rsa->engine, capi_idx);
832
833         CAPI_trace(ctx, "Called CAPI_rsa_sign()\n");
834
835         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
836         if (!capi_key)
837                 {
838                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_GET_KEY);
839                 return -1;
840                 }
841 /* Convert the signature type to a CryptoAPI algorithm ID */
842         switch(dtype)
843                 {
844         case NID_sha256:
845                 alg = CALG_SHA_256;
846                 break;
847
848         case NID_sha384:
849                 alg = CALG_SHA_384;
850                 break;
851
852         case NID_sha512:
853                 alg = CALG_SHA_512;
854                 break;
855
856         case NID_sha1:
857                 alg = CALG_SHA1;
858                 break;
859
860         case NID_md5:
861                 alg = CALG_MD5;
862                 break;
863
864         case NID_md5_sha1:
865                 alg = CALG_SSL3_SHAMD5;
866                 break;
867         default:
868                 {
869                 char algstr[10];
870                 BIO_snprintf(algstr, 10, "%lx", dtype);
871                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_UNSUPPORTED_ALGORITHM_NID);
872                 ERR_add_error_data(2, "NID=0x", algstr);
873                 return -1;
874                 }
875         }
876
877
878
879 /* Create the hash object */
880         if(!CryptCreateHash(capi_key->hprov, alg, 0, 0, &hash))
881                 {
882                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_CREATE_HASH_OBJECT);
883                 capi_addlasterror();
884                 return -1;
885                 }
886 /* Set the hash value to the value passed */
887
888         if(!CryptSetHashParam(hash, HP_HASHVAL, (unsigned char *)m, 0))
889                 {
890                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_SET_HASH_VALUE);
891                 capi_addlasterror();
892                 goto err;
893                 }
894
895
896 /* Finally sign it */
897         slen = RSA_size(rsa);
898         if(!CryptSignHash(hash, capi_key->keyspec, NULL, 0, sigret, &slen))
899                 {
900                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_ERROR_SIGNING_HASH);
901                 capi_addlasterror();
902                 goto err;
903                 }
904         else
905                 {
906                 ret = 1;
907                 /* Inplace byte reversal of signature */
908                 for(i = 0; i < slen / 2; i++)
909                         {
910                         unsigned char c;
911                         c = sigret[i];
912                         sigret[i] = sigret[slen - i - 1];
913                         sigret[slen - i - 1] = c;
914                         }
915                 *siglen = slen;
916                 }
917
918         /* Now cleanup */
919
920 err:
921         CryptDestroyHash(hash);
922
923         return ret;
924         }
925
926 int capi_rsa_priv_dec(int flen, const unsigned char *from,
927                 unsigned char *to, RSA *rsa, int padding)
928         {
929         int i;
930         unsigned char *tmpbuf;
931         CAPI_KEY *capi_key;
932         CAPI_CTX *ctx;
933         ctx = ENGINE_get_ex_data(rsa->engine, capi_idx);
934
935         CAPI_trace(ctx, "Called capi_rsa_priv_dec()\n");
936
937
938         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
939         if (!capi_key)
940                 {
941                 CAPIerr(CAPI_F_CAPI_RSA_PRIV_DEC, CAPI_R_CANT_GET_KEY);
942                 return -1;
943                 }
944
945         if(padding != RSA_PKCS1_PADDING)
946                 {
947                 char errstr[10];
948                 BIO_snprintf(errstr, 10, "%d", padding);
949                 CAPIerr(CAPI_F_CAPI_RSA_PRIV_DEC, CAPI_R_UNSUPPORTED_PADDING);
950                 ERR_add_error_data(2, "padding=", errstr);
951                 return -1;
952                 }
953
954         /* Create temp reverse order version of input */
955         if(!(tmpbuf = OPENSSL_malloc(flen)) ) 
956                 {
957                 CAPIerr(CAPI_F_CAPI_RSA_PRIV_DEC, ERR_R_MALLOC_FAILURE);
958                 return -1;
959                 }
960         for(i = 0; i < flen; i++)
961                 tmpbuf[flen - i - 1] = from[i];
962         
963         /* Finally decrypt it */
964         if(!CryptDecrypt(capi_key->key, 0, TRUE, 0, tmpbuf, &flen))
965                 {
966                 CAPIerr(CAPI_F_CAPI_RSA_PRIV_DEC, CAPI_R_DECRYPT_ERROR);
967                 capi_addlasterror();
968                 OPENSSL_free(tmpbuf);
969                 return -1;
970                 } 
971         else memcpy(to, tmpbuf, flen);
972
973         OPENSSL_free(tmpbuf);
974
975         return flen;
976         }
977
978 static int capi_rsa_free(RSA *rsa)
979         {
980         CAPI_KEY *capi_key;
981         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
982         capi_free_key(capi_key);
983         RSA_set_ex_data(rsa, rsa_capi_idx, 0);
984         return 1;
985         }
986
987 /* CryptoAPI DSA operations */
988
989 static DSA_SIG *capi_dsa_do_sign(const unsigned char *digest, int dlen,
990                                                                 DSA *dsa)
991         {
992         HCRYPTHASH hash;
993         DWORD slen;
994         DSA_SIG *ret = NULL;
995         CAPI_KEY *capi_key;
996         CAPI_CTX *ctx;
997         unsigned char csigbuf[40];
998
999         ctx = ENGINE_get_ex_data(dsa->engine, capi_idx);
1000
1001         CAPI_trace(ctx, "Called CAPI_dsa_do_sign()\n");
1002
1003         capi_key = DSA_get_ex_data(dsa, dsa_capi_idx);
1004
1005         if (!capi_key)
1006                 {
1007                 CAPIerr(CAPI_F_CAPI_DSA_DO_SIGN, CAPI_R_CANT_GET_KEY);
1008                 return NULL;
1009                 }
1010
1011         if (dlen != 20)
1012                 {
1013                 CAPIerr(CAPI_F_CAPI_DSA_DO_SIGN, CAPI_R_INVALID_DIGEST_LENGTH);
1014                 return NULL;
1015                 }
1016
1017         /* Create the hash object */
1018         if(!CryptCreateHash(capi_key->hprov, CALG_SHA1, 0, 0, &hash))
1019                 {
1020                 CAPIerr(CAPI_F_CAPI_DSA_DO_SIGN, CAPI_R_CANT_CREATE_HASH_OBJECT);
1021                 capi_addlasterror();
1022                 return NULL;
1023                 }
1024
1025         /* Set the hash value to the value passed */
1026         if(!CryptSetHashParam(hash, HP_HASHVAL, (unsigned char *)digest, 0))
1027                 {
1028                 CAPIerr(CAPI_F_CAPI_DSA_DO_SIGN, CAPI_R_CANT_SET_HASH_VALUE);
1029                 capi_addlasterror();
1030                 goto err;
1031                 }
1032
1033
1034         /* Finally sign it */
1035         slen = sizeof(csigbuf);
1036         if(!CryptSignHash(hash, capi_key->keyspec, NULL, 0, csigbuf, &slen))
1037                 {
1038                 CAPIerr(CAPI_F_CAPI_DSA_DO_SIGN, CAPI_R_ERROR_SIGNING_HASH);
1039                 capi_addlasterror();
1040                 goto err;
1041                 }
1042         else
1043                 {
1044                 ret = DSA_SIG_new();
1045                 if (!ret)
1046                         goto err;
1047                 ret->r = BN_new();
1048                 ret->s = BN_new();
1049                 if (!ret->r || !ret->s)
1050                         goto err;
1051                 if (!lend_tobn(ret->r, csigbuf, 20)
1052                         || !lend_tobn(ret->s, csigbuf + 20, 20))
1053                         {
1054                         DSA_SIG_free(ret);
1055                         ret = NULL;
1056                         goto err;
1057                         }
1058                 }
1059
1060         /* Now cleanup */
1061
1062 err:
1063         OPENSSL_cleanse(csigbuf, 40);
1064         CryptDestroyHash(hash);
1065         return ret;
1066         }
1067
1068 static int capi_dsa_free(DSA *dsa)
1069         {
1070         CAPI_KEY *capi_key;
1071         capi_key = DSA_get_ex_data(dsa, dsa_capi_idx);
1072         capi_free_key(capi_key);
1073         DSA_set_ex_data(dsa, dsa_capi_idx, 0);
1074         return 1;
1075         }
1076
1077 static void capi_vtrace(CAPI_CTX *ctx, int level, char *format, va_list argptr)
1078         {
1079         BIO *out;
1080
1081         if (!ctx || (ctx->debug_level < level) || (!ctx->debug_file))
1082                 return;
1083         out = BIO_new_file(ctx->debug_file, "a+");
1084         BIO_vprintf(out, format, argptr);
1085         BIO_free(out);
1086         }
1087
1088 static void CAPI_trace(CAPI_CTX *ctx, char *format, ...)
1089         {
1090         va_list args;
1091         va_start(args, format);
1092         capi_vtrace(ctx, CAPI_DBG_TRACE, format, args);
1093         va_end(args);
1094         }
1095
1096 static void capi_addlasterror(void)
1097         {
1098         capi_adderror(GetLastError());
1099         }
1100
1101 static void capi_adderror(DWORD err)
1102         {
1103         char errstr[10];
1104         BIO_snprintf(errstr, 10, "%lX", err);
1105         ERR_add_error_data(2, "Error code= 0x", errstr);
1106         }
1107
1108 static char *wide_to_asc(LPCWSTR wstr)
1109         {
1110         char *str;
1111         int len_0,sz;
1112
1113         if (!wstr)
1114                 return NULL;
1115         len_0 = (int)wcslen(wstr)+1;    /* WideCharToMultiByte expects int */
1116         sz = WideCharToMultiByte(CP_ACP,0,wstr,len_0,NULL,0,NULL,NULL);
1117         if (!sz)
1118                 {
1119                 CAPIerr(CAPI_F_WIDE_TO_ASC, CAPI_R_WIN32_ERROR);
1120                 return NULL;
1121                 }
1122         str = OPENSSL_malloc(sz);
1123         if (!str)
1124                 {
1125                 CAPIerr(CAPI_F_WIDE_TO_ASC, ERR_R_MALLOC_FAILURE);
1126                 return NULL;
1127                 }
1128         if (!WideCharToMultiByte(CP_ACP,0,wstr,len_0,str,sz,NULL,NULL))
1129                 {
1130                 OPENSSL_free(str);
1131                 CAPIerr(CAPI_F_WIDE_TO_ASC, CAPI_R_WIN32_ERROR);
1132                 return NULL;
1133                 }
1134         return str;
1135         }
1136
1137 static int capi_get_provname(CAPI_CTX *ctx, LPSTR *pname, DWORD *ptype, DWORD idx)
1138         {
1139         DWORD len, err;
1140         LPTSTR name;
1141         CAPI_trace(ctx, "capi_get_provname, index=%d\n", idx);
1142         if (!CryptEnumProviders(idx, NULL, 0, ptype, NULL, &len))
1143                 {
1144                 err = GetLastError();
1145                 if (err == ERROR_NO_MORE_ITEMS)
1146                         return 2;
1147                 CAPIerr(CAPI_F_CAPI_GET_PROVNAME, CAPI_R_CRYPTENUMPROVIDERS_ERROR);
1148                 capi_adderror(err);
1149                 return 0;
1150                 }
1151         if (sizeof(TCHAR) != sizeof(char))
1152                 name = alloca(len);
1153         else
1154                 name = OPENSSL_malloc(len);
1155         if (!CryptEnumProviders(idx, NULL, 0, ptype, name, &len))
1156                 {
1157                 err = GetLastError();
1158                 if (err == ERROR_NO_MORE_ITEMS)
1159                         return 2;
1160                 CAPIerr(CAPI_F_CAPI_GET_PROVNAME, CAPI_R_CRYPTENUMPROVIDERS_ERROR);
1161                 capi_adderror(err);
1162                 return 0;
1163                 }
1164         if (sizeof(TCHAR) != sizeof(char))
1165                 *pname = wide_to_asc((WCHAR *)name);
1166         else
1167                 *pname = (char *)name;
1168         CAPI_trace(ctx, "capi_get_provname, returned name=%s, type=%d\n", *pname, *ptype);
1169
1170         return 1;
1171         }
1172
1173 static int capi_list_providers(CAPI_CTX *ctx, BIO *out)
1174         {
1175         DWORD idx, ptype;
1176         int ret;
1177         LPSTR provname = NULL;
1178         CAPI_trace(ctx, "capi_list_providers\n");
1179         BIO_printf(out, "Available CSPs:\n");
1180         for(idx = 0; ; idx++)
1181                 {
1182                 ret = capi_get_provname(ctx, &provname, &ptype, idx);
1183                 if (ret == 2)
1184                         break;
1185                 if (ret == 0)
1186                         break;
1187                 BIO_printf(out, "%d. %s, type %d\n", idx, provname, ptype);
1188                 OPENSSL_free(provname);
1189                 }
1190         return 1;
1191         }
1192
1193 static int capi_list_containers(CAPI_CTX *ctx, BIO *out)
1194         {
1195         int ret = 1;
1196         HCRYPTPROV hprov;
1197         DWORD err, idx, flags, buflen = 0, clen;
1198         LPSTR cname;
1199         LPTSTR cspname = NULL;
1200
1201         CAPI_trace(ctx, "Listing containers CSP=%s, type = %d\n", ctx->cspname, ctx->csptype);
1202         if (ctx->cspname && sizeof(TCHAR)!=sizeof(char))
1203                 {
1204                 if ((clen=MultiByteToWideChar(CP_ACP,0,ctx->cspname,-1,NULL,0)))
1205                         {
1206                         cspname = alloca(clen*sizeof(WCHAR));
1207                         MultiByteToWideChar(CP_ACP,0,ctx->cspname,-1,(WCHAR *)cspname,clen);
1208                         }
1209                 if (!cspname)
1210                         {
1211                         CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, ERR_R_MALLOC_FAILURE);
1212                         capi_addlasterror();
1213                         return 0;
1214                         }
1215                 }
1216         else
1217                 cspname = (TCHAR *)ctx->cspname;
1218         if (!CryptAcquireContext(&hprov, NULL, cspname, ctx->csptype, CRYPT_VERIFYCONTEXT))
1219                 {
1220                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
1221                 capi_addlasterror();
1222                 return 0;
1223                 }
1224         if (!CryptGetProvParam(hprov, PP_ENUMCONTAINERS, NULL, &buflen, CRYPT_FIRST))
1225                 {
1226                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_ENUMCONTAINERS_ERROR);
1227                 capi_addlasterror();
1228                 CryptReleaseContext(hprov, 0);
1229                 return 0;
1230                 }
1231         CAPI_trace(ctx, "Got max container len %d\n", buflen);
1232         if (buflen == 0)
1233                 buflen = 1024;
1234         cname = OPENSSL_malloc(buflen);
1235         if (!cname)
1236                 {
1237                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, ERR_R_MALLOC_FAILURE);
1238                 goto err;
1239                 }
1240
1241         for (idx = 0;;idx++)
1242                 {
1243                 clen = buflen;
1244                 cname[0] = 0;
1245
1246                 if (idx == 0)
1247                         flags = CRYPT_FIRST;
1248                 else
1249                         flags = 0;
1250                 if(!CryptGetProvParam(hprov, PP_ENUMCONTAINERS, (BYTE *)cname, &clen, flags))
1251                         {
1252                         err = GetLastError();
1253                         if (err == ERROR_NO_MORE_ITEMS)
1254                                 goto done;
1255                         CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_ENUMCONTAINERS_ERROR);
1256                         capi_adderror(err);
1257                         goto err;
1258                         }
1259                 CAPI_trace(ctx, "Container name %s, len=%d, index=%d, flags=%d\n", cname, clen, idx, flags);
1260                 if (!cname[0] && (clen == buflen))
1261                         {
1262                         CAPI_trace(ctx, "Enumerate bug: using workaround\n");
1263                         goto done;
1264                         }
1265                 BIO_printf(out, "%d. %s\n", idx, cname);
1266                 }
1267         err:
1268
1269         ret = 0;
1270
1271         done:
1272         if (cname)
1273                 OPENSSL_free(cname);
1274         CryptReleaseContext(hprov, 0);
1275
1276         return ret;
1277         }
1278
1279 CRYPT_KEY_PROV_INFO *capi_get_prov_info(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
1280         {
1281         DWORD len;
1282         CRYPT_KEY_PROV_INFO *pinfo;
1283         
1284         if(!CertGetCertificateContextProperty(cert, CERT_KEY_PROV_INFO_PROP_ID, NULL, &len))
1285                 return NULL;
1286         pinfo = OPENSSL_malloc(len);
1287         if (!pinfo)
1288                 {
1289                 CAPIerr(CAPI_F_CAPI_GET_PROV_INFO, ERR_R_MALLOC_FAILURE);
1290                 return NULL;
1291                 }
1292         if(!CertGetCertificateContextProperty(cert, CERT_KEY_PROV_INFO_PROP_ID, pinfo, &len))
1293                 {
1294                 CAPIerr(CAPI_F_CAPI_GET_PROV_INFO, CAPI_R_ERROR_GETTING_KEY_PROVIDER_INFO);
1295                 capi_addlasterror();
1296                 OPENSSL_free(pinfo);
1297                 return NULL;
1298                 }
1299         return pinfo;
1300         }
1301
1302 static void capi_dump_prov_info(CAPI_CTX *ctx, BIO *out, CRYPT_KEY_PROV_INFO *pinfo)
1303         {
1304         char *provname = NULL, *contname = NULL;
1305         if (!pinfo)
1306                 {
1307                 BIO_printf(out, "  No Private Key\n");
1308                 return;
1309                 }
1310         provname = wide_to_asc(pinfo->pwszProvName);
1311         contname = wide_to_asc(pinfo->pwszContainerName);
1312         if (!provname || !contname)
1313                 goto err;
1314
1315         BIO_printf(out, "  Private Key Info:\n");
1316         BIO_printf(out, "    Provider Name:  %s, Provider Type %d\n", provname, pinfo->dwProvType);
1317         BIO_printf(out, "    Container Name: %s, Key Type %d\n", contname, pinfo->dwKeySpec);
1318         err:
1319         if (provname)
1320                 OPENSSL_free(provname);
1321         if (contname)
1322                 OPENSSL_free(contname);
1323         }
1324
1325 char * capi_cert_get_fname(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
1326         {
1327         LPWSTR wfname;
1328         DWORD dlen;
1329
1330         CAPI_trace(ctx, "capi_cert_get_fname\n");
1331         if (!CertGetCertificateContextProperty(cert, CERT_FRIENDLY_NAME_PROP_ID, NULL, &dlen))
1332                 return NULL;
1333         wfname = OPENSSL_malloc(dlen);
1334         if (CertGetCertificateContextProperty(cert, CERT_FRIENDLY_NAME_PROP_ID, wfname, &dlen))
1335                 {
1336                 char *fname = wide_to_asc(wfname);
1337                 OPENSSL_free(wfname);
1338                 return fname;
1339                 }
1340         CAPIerr(CAPI_F_CAPI_CERT_GET_FNAME, CAPI_R_ERROR_GETTING_FRIENDLY_NAME);
1341         capi_addlasterror();
1342
1343         OPENSSL_free(wfname);
1344         return NULL;
1345         }
1346
1347
1348 void capi_dump_cert(CAPI_CTX *ctx, BIO *out, PCCERT_CONTEXT cert)
1349         {
1350         X509 *x;
1351         unsigned char *p;
1352         unsigned long flags = ctx->dump_flags;
1353         if (flags & CAPI_DMP_FNAME)
1354                 {
1355                 char *fname;
1356                 fname = capi_cert_get_fname(ctx, cert);
1357                 if (fname)
1358                         {
1359                         BIO_printf(out, "  Friendly Name \"%s\"\n", fname);
1360                         OPENSSL_free(fname);
1361                         }
1362                 else
1363                         BIO_printf(out, "  <No Friendly Name>\n");
1364                 }
1365
1366         p = cert->pbCertEncoded;
1367         x = d2i_X509(NULL, &p, cert->cbCertEncoded);
1368         if (!x)
1369                 BIO_printf(out, "  <Can't parse certificate>\n");
1370         if (flags & CAPI_DMP_SUMMARY)
1371                 {
1372                 BIO_printf(out, "  Subject: ");
1373                 X509_NAME_print_ex(out, X509_get_subject_name(x), 0, XN_FLAG_ONELINE);
1374                 BIO_printf(out, "\n  Issuer: ");
1375                 X509_NAME_print_ex(out, X509_get_issuer_name(x), 0, XN_FLAG_ONELINE);
1376                 BIO_printf(out, "\n");
1377                 }
1378         if (flags & CAPI_DMP_FULL)
1379                 X509_print_ex(out, x, XN_FLAG_ONELINE,0);
1380
1381         if (flags & CAPI_DMP_PKEYINFO)
1382                 {
1383                 CRYPT_KEY_PROV_INFO *pinfo;
1384                 pinfo = capi_get_prov_info(ctx, cert);
1385                 capi_dump_prov_info(ctx, out, pinfo);
1386                 if (pinfo)
1387                         OPENSSL_free(pinfo);
1388                 }
1389
1390         if (flags & CAPI_DMP_PEM)
1391                 PEM_write_bio_X509(out, x);
1392         X509_free(x);
1393         }
1394
1395 HCERTSTORE capi_open_store(CAPI_CTX *ctx, char *storename)
1396         {
1397         HCERTSTORE hstore;
1398
1399         if (!storename)
1400                 storename = ctx->storename;
1401         if (!storename)
1402                 storename = "MY";
1403         CAPI_trace(ctx, "Opening certificate store %s\n", storename);
1404
1405         hstore = CertOpenStore(CERT_STORE_PROV_SYSTEM_A, 0, 0, 
1406                                 ctx->store_flags, storename);
1407         if (!hstore)
1408                 {
1409                 CAPIerr(CAPI_F_CAPI_OPEN_STORE, CAPI_R_ERROR_OPENING_STORE);
1410                 capi_addlasterror();
1411                 }
1412         return hstore;
1413         }
1414
1415 int capi_list_certs(CAPI_CTX *ctx, BIO *out, char *id)
1416         {
1417         char *storename;
1418         int idx;
1419         int ret = 1;
1420         HCERTSTORE hstore;
1421         PCCERT_CONTEXT cert = NULL;
1422
1423         storename = ctx->storename;
1424         if (!storename)
1425                 storename = "MY";
1426         CAPI_trace(ctx, "Listing certs for store %s\n", storename);
1427
1428         hstore = capi_open_store(ctx, storename);
1429         if (!hstore)
1430                 return 0;
1431         if (id)
1432                 {
1433                 cert = capi_find_cert(ctx, id, hstore);
1434                 if (!cert)
1435                         {
1436                         ret = 0;
1437                         goto err;
1438                         }
1439                 capi_dump_cert(ctx, out, cert);
1440                 CertFreeCertificateContext(cert);
1441                 }
1442         else
1443                 {
1444                 for(idx = 0;;idx++)
1445                         {
1446                         cert = CertEnumCertificatesInStore(hstore, cert);
1447                         if (!cert)
1448                                 break;
1449                         BIO_printf(out, "Certificate %d\n", idx);
1450                         capi_dump_cert(ctx, out, cert);
1451                         }
1452                 }
1453         err:
1454         CertCloseStore(hstore, 0);
1455         return ret;
1456         }
1457
1458 static PCCERT_CONTEXT capi_find_cert(CAPI_CTX *ctx, const char *id, HCERTSTORE hstore)
1459         {
1460         PCCERT_CONTEXT cert = NULL;
1461         char *fname = NULL;
1462         int match;
1463         switch(ctx->lookup_method)
1464                 {
1465                 case CAPI_LU_SUBSTR:
1466                         return CertFindCertificateInStore(hstore,
1467                                         X509_ASN_ENCODING, 0,
1468                                         CERT_FIND_SUBJECT_STR_A, id, NULL);
1469                 case CAPI_LU_FNAME:
1470                         for(;;)
1471                                 {
1472                                 cert = CertEnumCertificatesInStore(hstore, cert);
1473                                 if (!cert)
1474                                         return NULL;
1475                                 fname = capi_cert_get_fname(ctx, cert);
1476                                 if (fname)
1477                                         {
1478                                         if (strcmp(fname, id))
1479                                                 match = 0;
1480                                         else
1481                                                 match = 1;
1482                                         OPENSSL_free(fname);
1483                                         if (match)
1484                                                 return cert;
1485                                         }
1486                                 }
1487                 default:
1488                         return NULL;
1489                 }
1490         }
1491
1492 static CAPI_KEY *capi_get_key(CAPI_CTX *ctx, const TCHAR *contname, TCHAR *provname, DWORD ptype, DWORD keyspec)
1493         {
1494         CAPI_KEY *key;
1495         DWORD dwFlags = 0; 
1496         key = OPENSSL_malloc(sizeof(CAPI_KEY));
1497         if (sizeof(TCHAR)==sizeof(char))
1498                 CAPI_trace(ctx, "capi_get_key, contname=%s, provname=%s, type=%d\n",
1499                                                 contname, provname, ptype);
1500         else if (ctx && ctx->debug_level>=CAPI_DBG_TRACE && ctx->debug_file)
1501                 {
1502                 /* above 'if' is optimization to minimize malloc-ations */
1503                 char *_contname = wide_to_asc((WCHAR *)contname);
1504                 char *_provname = wide_to_asc((WCHAR *)provname);
1505
1506                 CAPI_trace(ctx, "capi_get_key, contname=%s, provname=%s, type=%d\n", 
1507                                                 _contname, _provname, ptype);
1508                 if (_provname) OPENSSL_free(_provname);
1509                 if (_contname) OPENSSL_free(_contname);
1510                 }
1511         if(ctx->store_flags & CERT_SYSTEM_STORE_LOCAL_MACHINE)
1512                 dwFlags = CRYPT_MACHINE_KEYSET;
1513         if (!CryptAcquireContext(&key->hprov, contname, provname, ptype, dwFlags))
1514                 {
1515                 CAPIerr(CAPI_F_CAPI_GET_KEY, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
1516                 capi_addlasterror();
1517                 goto err;
1518                 }
1519         if (!CryptGetUserKey(key->hprov, keyspec, &key->key))
1520                 {
1521                 CAPIerr(CAPI_F_CAPI_GET_KEY, CAPI_R_GETUSERKEY_ERROR);
1522                 capi_addlasterror();
1523                 CryptReleaseContext(key->hprov, 0);
1524                 goto err;
1525                 }
1526         key->keyspec = keyspec;
1527         key->pcert = NULL;
1528         return key;
1529
1530         err:
1531         OPENSSL_free(key);
1532         return NULL;
1533         }
1534
1535 static CAPI_KEY *capi_get_cert_key(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
1536         {
1537         CAPI_KEY *key = NULL;
1538         CRYPT_KEY_PROV_INFO *pinfo = NULL;
1539         char *provname = NULL, *contname = NULL;
1540         pinfo = capi_get_prov_info(ctx, cert);
1541         if (!pinfo)
1542                 goto err;
1543         if (sizeof(TCHAR) != sizeof(char))
1544                 key = capi_get_key(ctx, (TCHAR *)pinfo->pwszContainerName,
1545                                 (TCHAR *)pinfo->pwszProvName,
1546                                 pinfo->dwProvType, pinfo->dwKeySpec);
1547         else
1548                 {
1549                 provname = wide_to_asc(pinfo->pwszProvName);
1550                 contname = wide_to_asc(pinfo->pwszContainerName);
1551                 if (!provname || !contname)
1552                         goto err;
1553                 key = capi_get_key(ctx, (TCHAR *)contname, (TCHAR *)provname,
1554                                 pinfo->dwProvType, pinfo->dwKeySpec);
1555                 }
1556
1557         err:
1558         if (pinfo)
1559                 OPENSSL_free(pinfo);
1560         if (provname)
1561                 OPENSSL_free(provname);
1562         if (contname)
1563                 OPENSSL_free(contname);
1564         return key;
1565         }
1566
1567 CAPI_KEY *capi_find_key(CAPI_CTX *ctx, const char *id)
1568         {
1569         PCCERT_CONTEXT cert;
1570         HCERTSTORE hstore;
1571         CAPI_KEY *key = NULL;
1572         switch (ctx->lookup_method)
1573                 {
1574                 case CAPI_LU_SUBSTR:
1575                 case CAPI_LU_FNAME:
1576                 hstore = capi_open_store(ctx, NULL);
1577                 if (!hstore)
1578                         return NULL;
1579                 cert = capi_find_cert(ctx, id, hstore);
1580                 if (cert)
1581                         {
1582                         key = capi_get_cert_key(ctx, cert);
1583                         CertFreeCertificateContext(cert);
1584                         }
1585                 CertCloseStore(hstore, 0);
1586                 break;
1587
1588                 case CAPI_LU_CONTNAME:
1589                 if (sizeof(TCHAR)!=sizeof(char))
1590                         {
1591                         WCHAR *contname, *provname;
1592                         DWORD len;
1593
1594                         if ((len=MultiByteToWideChar(CP_ACP,0,id,-1,NULL,0)) &&
1595                             (contname=alloca(len*sizeof(WCHAR)),
1596                              MultiByteToWideChar(CP_ACP,0,id,-1,contname,len)) &&
1597                             (len=MultiByteToWideChar(CP_ACP,0,ctx->cspname,-1,NULL,0)) &&
1598                             (provname=alloca(len*sizeof(WCHAR)),
1599                              MultiByteToWideChar(CP_ACP,0,ctx->cspname,-1,provname,len)))
1600                                 key = capi_get_key(ctx,(TCHAR *)contname,
1601                                                 (TCHAR *)provname,
1602                                                 ctx->csptype,ctx->keytype);
1603                         }
1604                 else
1605                         key = capi_get_key(ctx, (TCHAR *)id,
1606                                                 (TCHAR *)ctx->cspname,
1607                                                 ctx->csptype,ctx->keytype);
1608                 break;
1609                 }
1610
1611         return key;
1612         }
1613
1614 void capi_free_key(CAPI_KEY *key)
1615         {
1616         if (!key)
1617                 return;
1618         CryptDestroyKey(key->key);
1619         CryptReleaseContext(key->hprov, 0);
1620         if (key->pcert)
1621                 CertFreeCertificateContext(key->pcert);
1622         OPENSSL_free(key);
1623         }
1624
1625
1626 /* Initialize a CAPI_CTX structure */
1627
1628 static CAPI_CTX *capi_ctx_new()
1629         {
1630         CAPI_CTX *ctx;
1631         ctx = OPENSSL_malloc(sizeof(CAPI_CTX));
1632         if (!ctx)
1633                 {
1634                 CAPIerr(CAPI_F_CAPI_CTX_NEW, ERR_R_MALLOC_FAILURE);
1635                 return NULL;
1636                 }
1637         ctx->cspname = NULL;
1638         ctx->csptype = PROV_RSA_FULL;
1639         ctx->dump_flags = CAPI_DMP_SUMMARY|CAPI_DMP_FNAME;
1640         ctx->keytype = AT_KEYEXCHANGE;
1641         ctx->storename = NULL;
1642         ctx->ssl_client_store = NULL;
1643         ctx->store_flags = CERT_STORE_OPEN_EXISTING_FLAG |
1644                                 CERT_STORE_READONLY_FLAG |
1645                                 CERT_SYSTEM_STORE_CURRENT_USER;
1646         ctx->lookup_method = CAPI_LU_SUBSTR;
1647         ctx->debug_level = 0;
1648         ctx->debug_file = NULL;
1649         ctx->client_cert_select = cert_select_simple;
1650         return ctx;
1651         }
1652
1653 static void capi_ctx_free(CAPI_CTX *ctx)
1654         {
1655         CAPI_trace(ctx, "Calling capi_ctx_free with %lx\n", ctx);
1656         if (!ctx)
1657                 return;
1658         if (ctx->cspname)
1659                 OPENSSL_free(ctx->cspname);
1660         if (ctx->debug_file)
1661                 OPENSSL_free(ctx->debug_file);
1662         if (ctx->storename)
1663                 OPENSSL_free(ctx->storename);
1664         if (ctx->ssl_client_store)
1665                 OPENSSL_free(ctx->ssl_client_store);
1666         OPENSSL_free(ctx);
1667         }
1668
1669 static int capi_ctx_set_provname(CAPI_CTX *ctx, LPSTR pname, DWORD type, int check)
1670         {
1671         CAPI_trace(ctx, "capi_ctx_set_provname, name=%s, type=%d\n", pname, type);
1672         if (check)
1673                 {
1674                 HCRYPTPROV hprov;
1675                 LPTSTR name=NULL;
1676
1677                 if (sizeof(TCHAR)!=sizeof(char))
1678                         {
1679                         DWORD len;
1680                         if ((len=MultiByteToWideChar(CP_ACP,0,pname,-1,NULL,0)))
1681                                 {
1682                                 name = alloca(len*sizeof(WCHAR));
1683                                 MultiByteToWideChar(CP_ACP,0,pname,-1,(WCHAR *)name,len);
1684                                 }
1685                         }
1686                 else
1687                         name = (TCHAR *)pname;
1688
1689                 if (!name || !CryptAcquireContext(&hprov, NULL, name, type,
1690                                                 CRYPT_VERIFYCONTEXT))
1691                         {
1692                         CAPIerr(CAPI_F_CAPI_CTX_SET_PROVNAME, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
1693                         capi_addlasterror();
1694                         return 0;
1695                         }
1696                 CryptReleaseContext(hprov, 0);
1697                 }
1698         if (ctx->cspname)
1699                 OPENSSL_free(ctx->cspname);
1700         ctx->cspname = BUF_strdup(pname);
1701         ctx->csptype = type;
1702         return 1;
1703         }
1704
1705 static int capi_ctx_set_provname_idx(CAPI_CTX *ctx, int idx)
1706         {
1707         LPSTR pname;
1708         DWORD type;
1709         int res;
1710         if (capi_get_provname(ctx, &pname, &type, idx) != 1)
1711                 return 0;
1712         res = capi_ctx_set_provname(ctx, pname, type, 0);
1713         OPENSSL_free(pname);
1714         return res;
1715         }
1716
1717 static int cert_issuer_match(STACK_OF(X509_NAME) *ca_dn, X509 *x)
1718         {
1719         int i;
1720         X509_NAME *nm;
1721         /* Special case: empty list: match anything */
1722         if (sk_X509_NAME_num(ca_dn) <= 0)
1723                 return 1;
1724         for (i = 0; i < sk_X509_NAME_num(ca_dn); i++)
1725                 {
1726                 nm = sk_X509_NAME_value(ca_dn, i);
1727                 if (!X509_NAME_cmp(nm, X509_get_issuer_name(x)))
1728                                 return 1;
1729                 }
1730         return 0;
1731         }
1732
1733
1734
1735 static int capi_load_ssl_client_cert(ENGINE *e, SSL *ssl,
1736         STACK_OF(X509_NAME) *ca_dn, X509 **pcert, EVP_PKEY **pkey,
1737         STACK_OF(X509) **pother, UI_METHOD *ui_method, void *callback_data)
1738         {
1739         STACK_OF(X509) *certs = NULL;
1740         X509 *x;
1741         char *storename;
1742         const char *p;
1743         int i, client_cert_idx;
1744         HCERTSTORE hstore;
1745         PCCERT_CONTEXT cert = NULL, excert = NULL;
1746         CAPI_CTX *ctx;
1747         CAPI_KEY *key;
1748         ctx = ENGINE_get_ex_data(e, capi_idx);
1749
1750         *pcert = NULL;
1751         *pkey = NULL;
1752
1753         storename = ctx->ssl_client_store;
1754         if (!storename)
1755                 storename = "MY";
1756
1757         hstore = capi_open_store(ctx, storename);
1758         if (!hstore)
1759                 return 0;
1760         /* Enumerate all certificates collect any matches */
1761         for(i = 0;;i++)
1762                 {
1763                 cert = CertEnumCertificatesInStore(hstore, cert);
1764                 if (!cert)
1765                         break;
1766                 p = cert->pbCertEncoded;
1767                 x = d2i_X509(NULL, &p, cert->cbCertEncoded);
1768                 if (!x)
1769                         {
1770                         CAPI_trace(ctx, "Can't Parse Certificate %d\n", i);
1771                         continue;
1772                         }
1773                 if (cert_issuer_match(ca_dn, x)
1774                         && X509_check_purpose(x, X509_PURPOSE_SSL_CLIENT, 0))
1775                         {
1776                         key = capi_get_cert_key(ctx, cert);
1777                         if (!key)
1778                                 {
1779                                 X509_free(x);
1780                                 continue;
1781                                 }
1782                         /* Match found: attach extra data to it so
1783                          * we can retrieve the key later.
1784                          */
1785                         excert = CertDuplicateCertificateContext(cert);
1786                         key->pcert = excert;
1787                         X509_set_ex_data(x, cert_capi_idx, key);
1788
1789                         if (!certs)
1790                                 certs = sk_X509_new_null();
1791
1792                         sk_X509_push(certs, x);
1793                         }
1794                 else
1795                         X509_free(x);
1796
1797                 }
1798
1799         if (cert)
1800                 CertFreeCertificateContext(cert);
1801         if (hstore)
1802                 CertCloseStore(hstore, 0);
1803
1804         if (!certs)
1805                 return 0;
1806
1807
1808         /* Select the appropriate certificate */
1809
1810         client_cert_idx = ctx->client_cert_select(e, ssl, certs);
1811
1812         /* Set the selected certificate and free the rest */
1813
1814         for(i = 0; i < sk_X509_num(certs); i++)
1815                 {
1816                 x = sk_X509_value(certs, i);
1817                 if (i == client_cert_idx)
1818                         *pcert = x;
1819                 else
1820                         {
1821                         key = X509_get_ex_data(x, cert_capi_idx);
1822                         capi_free_key(key);
1823                         X509_free(x);
1824                         }
1825                 }
1826
1827         sk_X509_free(certs);
1828
1829         if (!*pcert)
1830                 return 0;
1831
1832         /* Setup key for selected certificate */
1833
1834         key = X509_get_ex_data(*pcert, cert_capi_idx);
1835         *pkey = capi_get_pkey(e, key);
1836         X509_set_ex_data(*pcert, cert_capi_idx, NULL);
1837
1838         return 1;
1839
1840         }
1841
1842
1843 /* Simple client cert selection function: always select first */
1844
1845 static int cert_select_simple(ENGINE *e, SSL *ssl, STACK_OF(X509) *certs)
1846         {
1847         return 0;
1848         }
1849
1850 #ifdef OPENSSL_CAPIENG_DIALOG
1851
1852 /* More complex cert selection function, using standard function
1853  * CryptUIDlgSelectCertificateFromStore() to produce a dialog box.
1854  */
1855
1856 /* Definitions which are in cryptuiapi.h but this is not present in older
1857  * versions of headers.
1858  */
1859
1860 #ifndef CRYPTUI_SELECT_LOCATION_COLUMN
1861 #define CRYPTUI_SELECT_LOCATION_COLUMN                   0x000000010
1862 #define CRYPTUI_SELECT_INTENDEDUSE_COLUMN                0x000000004
1863 #endif
1864
1865 #define dlg_title L"OpenSSL Application SSL Client Certificate Selection"
1866 #define dlg_prompt L"Select a certificate to use for authentication"
1867 #define dlg_columns      CRYPTUI_SELECT_LOCATION_COLUMN \
1868                         |CRYPTUI_SELECT_INTENDEDUSE_COLUMN
1869
1870 static int cert_select_dialog(ENGINE *e, SSL *ssl, STACK_OF(X509) *certs)
1871         {
1872         X509 *x;
1873         HCERTSTORE dstore;
1874         PCCERT_CONTEXT cert;
1875         CAPI_CTX *ctx;
1876         CAPI_KEY *key;
1877         HWND hwnd;
1878         int i, idx = -1;
1879         if (sk_X509_num(certs) == 1)
1880                 return 0;
1881         ctx = ENGINE_get_ex_data(e, capi_idx);
1882         /* Create an in memory store of certificates */
1883         dstore = CertOpenStore(CERT_STORE_PROV_MEMORY, 0, 0,
1884                                         CERT_STORE_CREATE_NEW_FLAG, NULL);
1885         if (!dstore)
1886                 {
1887                 CAPIerr(CAPI_F_CERT_SELECT_DIALOG, CAPI_R_ERROR_CREATING_STORE);
1888                 capi_addlasterror();
1889                 goto err;
1890                 }
1891         /* Add all certificates to store */
1892         for(i = 0; i < sk_X509_num(certs); i++)
1893                 {
1894                 x = sk_X509_value(certs, i);
1895                 key = X509_get_ex_data(x, cert_capi_idx);
1896
1897                 if (!CertAddCertificateContextToStore(dstore, key->pcert,
1898                                                 CERT_STORE_ADD_NEW, NULL))
1899                         {
1900                         CAPIerr(CAPI_F_CERT_SELECT_DIALOG, CAPI_R_ERROR_ADDING_CERT);
1901                         capi_addlasterror();
1902                         goto err;
1903                         }
1904
1905                 }
1906         hwnd = GetForegroundWindow();
1907         if (!hwnd)
1908                 hwnd = GetActiveWindow();
1909         if (!hwnd && ctx->getconswindow)
1910                 hwnd = ctx->getconswindow();
1911         /* Call dialog to select one */
1912         cert = ctx->certselectdlg(dstore, hwnd, dlg_title, dlg_prompt,
1913                                                 dlg_columns, 0, NULL);
1914
1915         /* Find matching cert from list */
1916         if (cert)
1917                 {
1918                 for(i = 0; i < sk_X509_num(certs); i++)
1919                         {
1920                         x = sk_X509_value(certs, i);
1921                         key = X509_get_ex_data(x, cert_capi_idx);
1922                         if (CertCompareCertificate(
1923                                 X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
1924                                         cert->pCertInfo,
1925                                         key->pcert->pCertInfo))
1926                                 {
1927                                 idx = i;
1928                                 break;
1929                                 }
1930                         }
1931                 }
1932
1933         err:
1934         if (dstore)
1935                 CertCloseStore(dstore, 0);
1936         return idx;
1937
1938         }
1939 #endif
1940
1941 #else /* !__COMPILE_CAPIENG */
1942 #include <openssl/engine.h>
1943 #ifndef OPENSSL_NO_DYNAMIC_ENGINE
1944 OPENSSL_EXPORT
1945 int bind_engine(ENGINE *e, const char *id, const dynamic_fns *fns);
1946 OPENSSL_EXPORT
1947 int bind_engine(ENGINE *e, const char *id, const dynamic_fns *fns) { return 0; }
1948 IMPLEMENT_DYNAMIC_CHECK_FN()
1949 #else
1950 void ENGINE_load_capi(void){}
1951 #endif
1952 #endif