Blob type and algorithm type sanity checks
[oweals/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 <openssl/crypto.h>
58 #include <openssl/buffer.h>
59 #include <openssl/engine.h>
60 #include <openssl/rsa.h>
61 #include <openssl/bn.h>
62 #include <openssl/pem.h>
63
64 #ifdef OPENSSL_SYS_WIN32
65 #ifndef OPENSSL_NO_CAPIENG
66
67 #ifndef _WIN32_WINNT
68 #define _WIN32_WINNT 0x400
69 #endif
70
71 #include <windows.h>
72 #include <wincrypt.h>
73
74 #include "e_capi_err.h"
75 #include "e_capi_err.c"
76
77
78 static const char *engine_capi_id = "capi";
79 static const char *engine_capi_name = "CryptoAPI ENGINE";
80
81 typedef struct CAPI_CTX_st CAPI_CTX;
82 typedef struct CAPI_KEY_st CAPI_KEY;
83
84 static void capi_addlasterror(void);
85 static void capi_adderror(DWORD err);
86
87 static void CAPI_trace(CAPI_CTX *ctx, char *format, ...);
88
89 static int capi_list_providers(CAPI_CTX *ctx, BIO *out);
90 static int capi_list_containers(CAPI_CTX *ctx, BIO *out);
91 int capi_list_certs(CAPI_CTX *ctx, BIO *out, char *storename);
92 void capi_free_key(CAPI_KEY *key);
93
94 static PCCERT_CONTEXT capi_find_cert(CAPI_CTX *ctx, const char *id, HCERTSTORE hstore);
95
96 CAPI_KEY *capi_find_key(CAPI_CTX *ctx, const char *id);
97
98 static EVP_PKEY *capi_load_privkey(ENGINE *eng, const char *key_id,
99         UI_METHOD *ui_method, void *callback_data);
100 static int capi_rsa_sign(int dtype, const unsigned char *m, unsigned int m_len,
101              unsigned char *sigret, unsigned int *siglen, const RSA *rsa);
102 static int capi_rsa_priv_enc(int flen, const unsigned char *from,
103                 unsigned char *to, RSA *rsa, int padding);
104 static int capi_rsa_priv_dec(int flen, const unsigned char *from,
105                 unsigned char *to, RSA *rsa, int padding);
106 static int capi_rsa_free(RSA *rsa);
107
108 /* This structure contains CAPI ENGINE specific data:
109  * it contains various global options and affects how
110  * other functions behave.
111  */
112
113 #define CAPI_DBG_TRACE  2
114 #define CAPI_DBG_ERROR  1
115
116 struct CAPI_CTX_st {
117         int debug_level;
118         char *debug_file;
119         /* Parameters to use for container lookup */
120         DWORD keytype;
121         LPTSTR cspname;
122         DWORD csptype;
123         /* Certificate store name to use */
124         LPTSTR storename;
125
126 /* Lookup string meanings in load_private_key */
127 /* Substring of subject: uses "storename" */
128 #define CAPI_LU_SUBSTR          0
129 /* Friendly name: uses storename */
130 #define CAPI_LU_FNAME           1
131 /* Container name: uses cspname, keytype */
132 #define CAPI_LU_CONTNAME        2
133         int lookup_method;
134 /* Info to dump with dumpcerts option */
135 /* Issuer and serial name strings */
136 #define CAPI_DMP_SUMMARY        0x1
137 /* Friendly name */
138 #define CAPI_DMP_FNAME          0x2
139 /* Full X509_print dump */
140 #define CAPI_DMP_FULL           0x4
141 /* Dump PEM format certificate */
142 #define CAPI_DMP_PEM            0x8
143 /* Dump pseudo key (if possible) */
144 #define CAPI_DMP_PSKEY          0x10
145 /* Dump key info (if possible) */
146 #define CAPI_DMP_PKEYINFO       0x20
147
148         DWORD dump_flags;
149 };
150
151
152 static CAPI_CTX *capi_ctx_new();
153 static void capi_ctx_free(CAPI_CTX *ctx);
154 static int capi_ctx_set_provname(CAPI_CTX *ctx, LPSTR pname, DWORD type, int check);
155 static int capi_ctx_set_provname_idx(CAPI_CTX *ctx, int idx);
156
157 #define CAPI_CMD_LIST_CERTS                     ENGINE_CMD_BASE
158 #define CAPI_CMD_LOOKUP_CERT            (ENGINE_CMD_BASE + 1)
159 #define CAPI_CMD_DEBUG_LEVEL            (ENGINE_CMD_BASE + 2)
160 #define CAPI_CMD_DEBUG_FILE                     (ENGINE_CMD_BASE + 3)
161 #define CAPI_CMD_KEYTYPE                        (ENGINE_CMD_BASE + 4)
162 #define CAPI_CMD_LIST_CSPS                      (ENGINE_CMD_BASE + 5)
163 #define CAPI_CMD_SET_CSP_IDX            (ENGINE_CMD_BASE + 6)
164 #define CAPI_CMD_SET_CSP_NAME           (ENGINE_CMD_BASE + 7)
165 #define CAPI_CMD_SET_CSP_TYPE           (ENGINE_CMD_BASE + 8)
166 #define CAPI_CMD_LIST_CONTAINERS        (ENGINE_CMD_BASE + 9)
167 #define CAPI_CMD_LIST_OPTIONS           (ENGINE_CMD_BASE + 10)
168 #define CAPI_CMD_LOOKUP_METHOD          (ENGINE_CMD_BASE + 11)
169
170 static const ENGINE_CMD_DEFN capi_cmd_defns[] = {
171         {CAPI_CMD_LIST_CERTS,
172                 "list_certs",
173                 "List all certificates in store",
174                 ENGINE_CMD_FLAG_NO_INPUT},
175         {CAPI_CMD_LOOKUP_CERT,
176                 "lookup_cert",
177                 "Lookup and output certificates",
178                 ENGINE_CMD_FLAG_STRING},
179         {CAPI_CMD_DEBUG_LEVEL,
180                 "debug_level",
181                 "debug level (1=errors, 2=trace)",
182                 ENGINE_CMD_FLAG_NUMERIC},
183         {CAPI_CMD_DEBUG_FILE,
184                 "debug_file",
185                 "debugging filename)",
186                 ENGINE_CMD_FLAG_STRING},
187         {CAPI_CMD_KEYTYPE,
188                 "key_type",
189                 "Key type: 1=AT_KEYEXCHANGE (default), 2=AT_SIGNATURE",
190                 ENGINE_CMD_FLAG_NUMERIC},
191         {CAPI_CMD_LIST_CSPS,
192                 "list_csps",
193                 "List all CSPs",
194                 ENGINE_CMD_FLAG_NO_INPUT},
195         {CAPI_CMD_SET_CSP_IDX,
196                 "csp_idx",
197                 "Set CSP by index",
198                 ENGINE_CMD_FLAG_NUMERIC},
199         {CAPI_CMD_SET_CSP_NAME,
200                 "csp_name",
201                 "Set CSP name, (default CSP used if not specified)",
202                 ENGINE_CMD_FLAG_STRING},
203         {CAPI_CMD_SET_CSP_TYPE,
204                 "csp_type",
205                 "Set CSP type, (default RSA_PROV_FULL)",
206                 ENGINE_CMD_FLAG_NUMERIC},
207         {CAPI_CMD_LIST_CONTAINERS,
208                 "list_containers",
209                 "list container names",
210                 ENGINE_CMD_FLAG_NO_INPUT},
211         {CAPI_CMD_LIST_OPTIONS,
212                 "list_options",
213                 "Set list options (1=summary,2=friendly name, 4=full printout, 8=PEM output, 16=XXX, "
214                 "32=private key info)",
215                 ENGINE_CMD_FLAG_NUMERIC},
216         {CAPI_CMD_LOOKUP_METHOD,
217                 "lookup_method",
218                 "Set key lookup method (1=substring, 2=friendlyname, 3=container name)",
219                 ENGINE_CMD_FLAG_NUMERIC},
220
221         {0, NULL, NULL, 0}
222         };
223
224 static int capi_idx = -1;
225 static int rsa_capi_idx = -1;
226
227 static int capi_ctrl(ENGINE *e, int cmd, long i, void *p, void (*f)(void))
228 {
229         int ret = 1;
230         CAPI_CTX *ctx;
231         BIO *out;
232         if (capi_idx == -1)
233                 {
234                 CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_ENGINE_NOT_INITIALIZED);
235                 return 0;
236                 }
237         ctx = ENGINE_get_ex_data(e, capi_idx);
238         out = BIO_new_fp(stdout, BIO_NOCLOSE);
239         switch (cmd)
240                 {
241                 case CAPI_CMD_LIST_CSPS:
242                 ret = capi_list_providers(ctx, out);
243                 break;
244
245                 case CAPI_CMD_LIST_CERTS:
246                 ret = capi_list_certs(ctx, out, NULL);
247                 break;
248
249                 case CAPI_CMD_LOOKUP_CERT:
250                 ret = capi_list_certs(ctx, out, p);
251                 break;
252
253                 case CAPI_CMD_LIST_CONTAINERS:
254                 ret = capi_list_containers(ctx, out);
255                 break;
256
257                 case CAPI_CMD_DEBUG_LEVEL:
258                 ctx->debug_level = (int)i;
259                 CAPI_trace(ctx, "Setting debug level to %d\n", ctx->debug_level);
260                 break;
261
262                 case CAPI_CMD_DEBUG_FILE:
263                 ctx->debug_file = BUF_strdup(p);
264                 CAPI_trace(ctx, "Setting debug file to %s\n", ctx->debug_file);
265                 break;
266
267                 case CAPI_CMD_KEYTYPE:
268                 ctx->keytype = i;
269                 CAPI_trace(ctx, "Setting key type to %d\n", ctx->keytype);
270                 break;
271
272                 case CAPI_CMD_SET_CSP_IDX:
273                 ret = capi_ctx_set_provname_idx(ctx, i);
274                 break;
275
276                 case CAPI_CMD_LIST_OPTIONS:
277                 ctx->dump_flags = i;
278                 break;
279
280                 case CAPI_CMD_LOOKUP_METHOD:
281                 if (i < 1 || i > 3)
282                         {
283                         CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_INVALID_LOOKUP_METHOD);
284                         return 0;
285                         }
286                 ctx->lookup_method = i;
287                 break;
288
289                 case CAPI_CMD_SET_CSP_NAME:
290                 ret = capi_ctx_set_provname(ctx, p, ctx->csptype, 1);
291                 break;
292
293                 case CAPI_CMD_SET_CSP_TYPE:
294                 ctx->csptype = i;
295                 break;
296
297                 default:
298                 CAPIerr(CAPI_F_CAPI_CTRL, CAPI_R_UNKNOWN_COMMAND);
299                 ret = 0;
300         }
301
302         BIO_free(out);
303         return ret;
304
305 }
306
307 static RSA_METHOD capi_rsa_method =
308         {
309         "CryptoAPI RSA method",
310         0,                              /* pub_enc */
311         0,                              /* pub_dec */
312         capi_rsa_priv_enc,      /* priv_enc */
313         capi_rsa_priv_dec,      /* priv_dec */
314         0,                              /* rsa_mod_exp */
315         0,                              /* bn_mod_exp */
316         0,                              /* init */
317         capi_rsa_free,  /* finish */
318         RSA_FLAG_SIGN_VER, /* flags */
319         NULL,                   /* app_data */
320         capi_rsa_sign,  /* rsa_sign */
321         0                               /* rsa_verify */
322         };
323
324 static void capi_ex_free(void *obj, void *item, CRYPTO_EX_DATA *ad,
325         int ind,long argl, void *argp)
326         {
327 /*fprintf(stderr, "Called capi_ex_free obj=%lx, idx=%d, item=%lx\n", obj, ind, item);*/
328         capi_ctx_free(item);
329         }
330
331 static void capi_rsa_ex_free(void *obj, void *item, CRYPTO_EX_DATA *ad,
332         int ind,long argl, void *argp)
333         {
334 /*fprintf(stderr, "Called capi_rsa_free_key\n");*/
335
336         capi_free_key(item);
337         }
338
339 static int capi_init(ENGINE *e)
340         {
341         CAPI_CTX *ctx;
342         const RSA_METHOD *ossl_meth;
343         capi_idx = ENGINE_get_ex_new_index(0, NULL, NULL, NULL, /*capi_ex_free*/ 0);
344
345         ctx = capi_ctx_new();
346         if (!ctx || (capi_idx < 0))
347                 goto memerr;
348
349         ENGINE_set_ex_data(e, capi_idx, ctx);
350         /* Setup RSA_METHOD */
351         rsa_capi_idx = RSA_get_ex_new_index(0, NULL, NULL, NULL, /*capi_rsa_ex_free*/ 0);
352         ossl_meth = RSA_PKCS1_SSLeay();
353         capi_rsa_method.rsa_pub_enc = ossl_meth->rsa_pub_enc;
354         capi_rsa_method.rsa_pub_dec = ossl_meth->rsa_pub_dec;
355         capi_rsa_method.rsa_mod_exp = ossl_meth->rsa_mod_exp;
356         capi_rsa_method.bn_mod_exp = ossl_meth->bn_mod_exp;
357
358         return 1;
359
360         memerr:
361         CAPIerr(CAPI_F_CAPI_INIT, ERR_R_MALLOC_FAILURE);
362         return 0;
363
364         return 1;
365         }
366
367 static int capi_destroy(ENGINE *e)
368         {
369
370         ERR_unload_CAPI_strings();
371         return 1;
372         }
373
374 static int capi_finish(ENGINE *e)
375         {
376         CAPI_CTX *ctx;
377         ctx = ENGINE_get_ex_data(e, capi_idx);
378         capi_ctx_free(ctx);
379         ENGINE_set_ex_data(e, capi_idx, NULL);
380         return 1;
381         }
382
383
384 /* CryptoAPI key application data. This contains
385  * a handle to the private key container (for sign operations)
386  * and a handle to the key (for decrypt operations).
387  */
388
389 struct CAPI_KEY_st
390         {
391         HCRYPTPROV hprov;
392         HCRYPTKEY key;
393         };
394
395 static int bind_capi(ENGINE *e)
396         {
397         if (!ENGINE_set_id(e, engine_capi_id)
398                 || !ENGINE_set_name(e, engine_capi_name)
399                 || !ENGINE_set_init_function(e, capi_init)
400                 || !ENGINE_set_finish_function(e, capi_finish)
401                 || !ENGINE_set_destroy_function(e, capi_destroy)
402                 || !ENGINE_set_RSA(e, &capi_rsa_method)
403                 || !ENGINE_set_load_privkey_function(e, capi_load_privkey)
404                 || !ENGINE_set_cmd_defns(e, capi_cmd_defns)
405                 || !ENGINE_set_ctrl_function(e, capi_ctrl))
406                         return 0;
407         ERR_load_CAPI_strings();
408
409         return 1;
410
411         }
412
413 #ifndef OPENSSL_NO_DYNAMIC_ENGINE
414 static int bind_helper(ENGINE *e, const char *id)
415         {
416         if(id && (strcmp(id, engine_capi_id) != 0))
417                 return 0;
418         if(!bind_capi(e))
419                 return 0;
420         return 1;
421         }       
422 IMPLEMENT_DYNAMIC_CHECK_FN()
423 IMPLEMENT_DYNAMIC_BIND_FN(bind_helper)
424 #else
425 static ENGINE *engine_capi(void)
426         {
427         ENGINE *ret = ENGINE_new();
428         if(!ret)
429                 return NULL;
430         if(!bind_capi(ret))
431                 {
432                 ENGINE_free(ret);
433                 return NULL;
434                 }
435         return ret;
436         }
437
438 void ENGINE_load_capi(void)
439         {
440         /* Copied from eng_[openssl|dyn].c */
441         ENGINE *toadd = engine_capi();
442         if(!toadd) return;
443         ENGINE_add(toadd);
444         ENGINE_free(toadd);
445         ERR_clear_error();
446         }
447 #endif
448
449 static EVP_PKEY *capi_load_privkey(ENGINE *eng, const char *key_id,
450         UI_METHOD *ui_method, void *callback_data)
451         {
452         EVP_PKEY *ret = NULL;
453         CAPI_CTX *ctx;
454         CAPI_KEY *key;
455         unsigned char *pubkey = NULL;
456         DWORD len, i, rsa_modlen;
457         BLOBHEADER *bh;
458         RSAPUBKEY *rp;
459         unsigned char *rsa_modulus;
460         RSA *rkey = NULL;
461         ctx = ENGINE_get_ex_data(eng, capi_idx);
462
463         if (!ctx)
464                 {
465                 CAPIerr(CAPI_F_CAPI_LOAD_PRIVKEY, CAPI_R_CANT_FIND_CAPI_CONTEXT);
466                 return NULL;
467                 }
468
469         key = capi_find_key(ctx, key_id);
470
471         if (!key)
472                 return NULL;
473
474         len = 0;
475         if (!CryptExportKey(key->key, 0, PUBLICKEYBLOB, 0, NULL, &len))
476                 {
477                 CAPIerr(CAPI_F_CAPI_LOAD_PRIVKEY, CAPI_R_PUBKEY_EXPORT_LENGTH_ERROR);
478                 capi_addlasterror();
479                 return NULL;
480                 }
481
482         pubkey = OPENSSL_malloc(len);
483
484         if (!pubkey)
485                 goto memerr;
486
487         if (!CryptExportKey(key->key, 0, PUBLICKEYBLOB, 0, pubkey, &len))
488                 {
489                 CAPIerr(CAPI_F_CAPI_LOAD_PRIVKEY, CAPI_R_PUBKEY_EXPORT_ERROR);
490                 capi_addlasterror();
491                 goto err;
492                 }
493
494         bh = (BLOBHEADER *)pubkey;
495         if (bh->bType != PUBLICKEYBLOB)
496                 {
497                 /* FIXME */
498                 fprintf(stderr, "Invalid public key blob\n");
499                 goto err;
500                 }
501         if (bh->aiKeyAlg == CALG_RSA_SIGN || bh->aiKeyAlg == CALG_RSA_KEYX)
502                 {
503                 rp = (RSAPUBKEY *)(bh + 1);
504                 rsa_modulus = (unsigned char *)(rp + 1);
505                 rkey = RSA_new_method(eng);
506                 if (!rkey)
507                         goto memerr;
508
509                 rkey->e = BN_new();
510                 rkey->n = BN_new();
511
512                 if (!rkey->e || !rkey->n)
513                         goto memerr;
514
515                 if (!BN_set_word(rkey->e, rp->pubexp))
516                         goto memerr;
517
518                 rsa_modlen = rp->bitlen / 8;
519
520                 for(i = 0; i < rsa_modlen / 2; i++)
521                         {
522                         unsigned char c;
523                         c = rsa_modulus[i];
524                         rsa_modulus[i] = rsa_modulus[rsa_modlen - i - 1];
525                         rsa_modulus[rsa_modlen - i - 1] = c;
526                         }
527
528                 if (!BN_bin2bn(rsa_modulus, rsa_modlen, rkey->n))
529                         goto memerr;
530
531                 RSA_set_ex_data(rkey, rsa_capi_idx, key);
532
533                 if (!(ret = EVP_PKEY_new()))
534                         goto memerr;
535
536                 EVP_PKEY_assign_RSA(ret, rkey);
537
538
539                 }
540         else
541                 {
542                 fprintf(stderr, "Unsupported Key Algorithm %x\n",
543                                         bh->aiKeyAlg);
544                 goto err;
545                 }
546
547         err:
548         if (pubkey)
549                 OPENSSL_free(pubkey);
550         if (!ret)
551                 {
552                 if (rkey)
553                         RSA_free(rkey);
554                 if (key)
555                         capi_free_key(key);
556                 }
557
558         return ret;
559
560 memerr:
561         CAPIerr(CAPI_F_CAPI_LOAD_PRIVKEY, ERR_R_MALLOC_FAILURE);
562         goto err;
563
564         }
565
566 /* CryptoAPI RSA operations */
567
568 int capi_rsa_priv_enc(int flen, const unsigned char *from,
569                 unsigned char *to, RSA *rsa, int padding)
570         {
571         CAPIerr(CAPI_F_CAPI_RSA_PRIV_ENC, CAPI_R_FUNCTION_NOT_SUPPORTED);
572         return -1;
573         }
574
575 int capi_rsa_sign(int dtype, const unsigned char *m, unsigned int m_len,
576              unsigned char *sigret, unsigned int *siglen, const RSA *rsa)
577         {
578         ALG_ID alg;
579         HCRYPTHASH hash;
580         DWORD slen;
581         unsigned int i;
582         int ret = -1;
583         CAPI_KEY *capi_key;
584         CAPI_CTX *ctx;
585
586         ctx = ENGINE_get_ex_data(rsa->engine, capi_idx);
587
588         CAPI_trace(ctx, "Called CAPI_rsa_sign()\n");
589
590         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
591         if (!capi_key)
592                 {
593                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_GET_KEY);
594                 return -1;
595                 }
596 /* Convert the signature type to a CryptoAPI algorithm ID */
597         switch(dtype) {
598         case NID_sha1:
599                 alg = CALG_SHA1;
600                 break;
601
602         case NID_md5:
603                 alg = CALG_MD5;
604                 break;
605
606         case NID_md5_sha1:
607                 alg = CALG_SSL3_SHAMD5;
608                 break;
609         default:
610                 {
611                 char algstr[10];
612                 sprintf(algstr, "%lx", dtype);
613                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_UNSUPPORTED_ALGORITHM_NID);
614                 ERR_add_error_data(2, "NID=0x", algstr);
615                 return -1;
616                 }
617         }
618
619
620
621 /* Create the hash object */
622         if(!CryptCreateHash(capi_key->hprov, alg, 0, 0, &hash)) {
623                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_CREATE_HASH_OBJECT);
624                 capi_addlasterror();
625                 return -1;
626         }
627 /* Set the hash value to the value passed */
628
629         if(!CryptSetHashParam(hash, HP_HASHVAL, (unsigned char *)m, 0)) {
630                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_CANT_SET_HASH_VALUE);
631                 capi_addlasterror();
632                 goto err;
633         }
634
635
636 /* Finally sign it */
637         slen = RSA_size(rsa);
638         if(!CryptSignHash(hash, AT_KEYEXCHANGE, NULL, 0, sigret, &slen)) {
639                 CAPIerr(CAPI_F_CAPI_RSA_SIGN, CAPI_R_ERROR_SIGNING_HASH);
640                 capi_addlasterror();
641                 goto err;
642         } else {
643                 ret = 1;
644                 /* Inplace byte reversal of signature */
645                 for(i = 0; i < slen / 2; i++) {
646                         unsigned char c;
647                         c = sigret[i];
648                         sigret[i] = sigret[slen - i - 1];
649                         sigret[slen - i - 1] = c;
650                 }
651                 *siglen = slen;
652         }
653
654
655         /* Now cleanup */
656
657 err:
658         CryptDestroyHash(hash);
659
660         return ret;
661 }
662
663 int capi_rsa_priv_dec(int flen, const unsigned char *from,
664                 unsigned char *to, RSA *rsa, int padding)
665 {
666         int i;
667         unsigned char *tmpbuf;
668         CAPI_KEY *capi_key;
669         CAPI_CTX *ctx;
670         ctx = ENGINE_get_ex_data(rsa->engine, capi_idx);
671
672         CAPI_trace(ctx, "Called capi_rsa_priv_dec()\n");
673
674
675         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
676         if (!capi_key)
677                 {
678                 CAPIerr(CAPI_F_CAPI_RSA_DECRYPT, CAPI_R_CANT_GET_KEY);
679                 return -1;
680                 }
681
682         if(padding != RSA_PKCS1_PADDING)
683                 {
684                 char errstr[10];
685                 sprintf(errstr, "%d", padding);
686                 CAPIerr(CAPI_F_CAPI_RSA_DECRYPT, CAPI_R_UNSUPPORTED_PADDING);
687                 ERR_add_error_data(2, "padding=", errstr);
688                 return -1;
689                 }
690
691         /* Create temp reverse order version of input */
692         if(!(tmpbuf = OPENSSL_malloc(flen)) ) 
693                 {
694                 CAPIerr(CAPI_F_CAPI_RSA_DECRYPT, ERR_R_MALLOC_FAILURE);
695                 return -1;
696                 }
697         for(i = 0; i < flen; i++) tmpbuf[flen - i - 1] = from[i];
698         
699         /* Finally decrypt it */
700         if(!CryptDecrypt(capi_key->key, 0, TRUE, 0, tmpbuf, &flen))
701                 {
702                 CAPIerr(CAPI_F_CAPI_RSA_DECRYPT, CAPI_R_DECRYPT_ERROR);
703                 capi_addlasterror();
704                 OPENSSL_free(tmpbuf);
705                 return -1;
706                 } 
707         else memcpy(to, tmpbuf, flen);
708
709         OPENSSL_free(tmpbuf);
710
711         return flen;
712 }
713
714 static int capi_rsa_free(RSA *rsa)
715         {
716         CAPI_KEY *capi_key;
717         capi_key = RSA_get_ex_data(rsa, rsa_capi_idx);
718         capi_free_key(capi_key);
719         RSA_set_ex_data(rsa, rsa_capi_idx, 0);
720         return 1;
721         }
722
723 static void capi_vtrace(CAPI_CTX *ctx, int level, char *format, va_list argptr)
724         {
725         BIO *out;
726
727         if (!ctx || (ctx->debug_level < level) || (!ctx->debug_file))
728                 return;
729         out = BIO_new_file(ctx->debug_file, "a+");
730         BIO_vprintf(out, format, argptr);
731         BIO_free(out);
732         }
733
734 static void CAPI_trace(CAPI_CTX *ctx, char *format, ...)
735         {
736         va_list args;
737         va_start(args, format);
738         capi_vtrace(ctx, CAPI_DBG_TRACE, format, args);
739         va_end(args);
740         }
741
742 static void capi_addlasterror(void)
743         {
744         capi_adderror(GetLastError());
745         }
746
747 static void capi_adderror(DWORD err)
748         {
749         char errstr[10];
750         sprintf(errstr, "%lX", err);
751         ERR_add_error_data(2, "Error code= 0x", errstr);
752         }
753
754 static char *wide_to_asc(LPWSTR wstr)
755         {
756         char *str;
757         if (!wstr)
758                 return NULL;
759         str = OPENSSL_malloc(wcslen(wstr) + 1);
760         if (!str)
761                 {
762                 CAPIerr(CAPI_F_WIDE_TO_ASC, ERR_R_MALLOC_FAILURE);
763                 return NULL;
764                 }
765         sprintf(str, "%S", wstr);
766         return str;
767         }
768
769 static int capi_get_provname(CAPI_CTX *ctx, LPSTR *pname, DWORD *ptype, DWORD idx)
770         {
771         LPSTR name;
772         DWORD len, err;
773         CAPI_trace(ctx, "capi_get_provname, index=%d\n", idx);
774         if (!CryptEnumProviders(idx, NULL, 0, ptype, NULL, &len))
775                 {
776                 err = GetLastError();
777                 if (err == ERROR_NO_MORE_ITEMS)
778                         return 2;
779                 CAPIerr(CAPI_F_CAPI_GET_PROVNAME, CAPI_R_CRYPTENUMPROVIDERS_ERROR);
780                 capi_adderror(err);
781                 return 0;
782                 }
783         name = OPENSSL_malloc(len);
784                 if (!CryptEnumProviders(idx, NULL, 0, ptype, name, &len))
785                 {
786                 err = GetLastError();
787                 if (err == ERROR_NO_MORE_ITEMS)
788                         return 2;
789                 CAPIerr(CAPI_F_CAPI_GET_PROVNAME, CAPI_R_CRYPTENUMPROVIDERS_ERROR);
790                 capi_adderror(err);
791                 return 0;
792                 }
793         *pname = name;
794         CAPI_trace(ctx, "capi_get_provname, returned name=%s, type=%d\n", name, *ptype);
795
796         return 1;
797         }
798
799 static int capi_list_providers(CAPI_CTX *ctx, BIO *out)
800         {
801         DWORD idx, ptype;
802         int ret;
803         LPTSTR provname = NULL;
804         CAPI_trace(ctx, "capi_list_providers\n");
805         BIO_printf(out, "Available CSPs:\n");
806         for(idx = 0; ; idx++)
807                 {
808                 ret = capi_get_provname(ctx, &provname, &ptype, idx);
809                 if (ret == 2)
810                         break;
811                 if (ret == 0)
812                         break;
813                 BIO_printf(out, "%d. %s, type %d\n", idx, provname, ptype);
814                 OPENSSL_free(provname);
815                 }
816         return 1;
817         }
818
819 static int capi_list_containers(CAPI_CTX *ctx, BIO *out)
820         {
821         int ret = 1;
822         HCRYPTPROV hprov;
823         DWORD err, idx, flags, buflen = 0, clen;
824         LPSTR cname;
825         CAPI_trace(ctx, "Listing containers CSP=%s, type = %d\n", ctx->cspname, ctx->csptype);
826         if (!CryptAcquireContext(&hprov, NULL, ctx->cspname, ctx->csptype, CRYPT_VERIFYCONTEXT))
827                 {
828                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
829                 capi_addlasterror();
830                 return 0;
831                 }
832         if (!CryptGetProvParam(hprov, PP_ENUMCONTAINERS, NULL, &buflen, CRYPT_FIRST))
833                 {
834                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_ENUMCONTAINERS_ERROR);
835                 capi_addlasterror();
836                 return 0;
837                 }
838         CAPI_trace(ctx, "Got max container len %d\n", buflen);
839         if (buflen == 0)
840                 buflen = 1024;
841         cname = OPENSSL_malloc(buflen);
842         if (!cname)
843                 {
844                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, ERR_R_MALLOC_FAILURE);
845                 goto err;
846                 }
847
848         for (idx = 0;;idx++)
849                 {
850                         clen = buflen;
851                         cname[0] = 0;
852
853                         if (idx == 0)
854                                 flags = CRYPT_FIRST;
855                         else
856                                 flags = 0;
857                         if(!CryptGetProvParam(hprov, PP_ENUMCONTAINERS, cname, &clen, flags))
858                                 {
859                                 err = GetLastError();
860                                 if (err == ERROR_NO_MORE_ITEMS)
861                                         goto done;
862                                 CAPIerr(CAPI_F_CAPI_LIST_CONTAINERS, CAPI_R_ENUMCONTAINERS_ERROR);
863                                 capi_adderror(err);
864                                 goto err;
865                                 }
866                         CAPI_trace(ctx, "Container name %s, len=%d, index=%d, flags=%d\n", cname, clen, idx, flags);
867                         if (!cname[0] && (clen == buflen))
868                                 {
869                                 CAPI_trace(ctx, "Enumerate bug: using workaround\n");
870                                 goto done;
871                                 }
872                         BIO_printf(out, "%d. %s\n", idx, cname);
873                 }
874         err:
875
876         ret = 0;
877
878         done:
879         if (cname)
880                 OPENSSL_free(cname);
881         CryptReleaseContext(hprov, 0);
882
883         return ret;
884         }
885
886 CRYPT_KEY_PROV_INFO *capi_get_prov_info(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
887         {
888         DWORD len;
889         CRYPT_KEY_PROV_INFO *pinfo;
890         
891         if(!CertGetCertificateContextProperty(cert, CERT_KEY_PROV_INFO_PROP_ID, NULL, &len))
892                 return NULL;
893         pinfo = OPENSSL_malloc(len);
894         if (!pinfo)
895                 {
896                 CAPIerr(CAPI_F_CAPI_GET_PROV_INFO, ERR_R_MALLOC_FAILURE);
897                 return NULL;
898                 }
899         if(!CertGetCertificateContextProperty(cert, CERT_KEY_PROV_INFO_PROP_ID, pinfo, &len))
900                 {
901                 CAPIerr(CAPI_F_CAPI_GET_PROV_INFO, CAPI_R_ERROR_GETTING_KEY_PROVIDER_INFO);
902                 capi_addlasterror();
903                 OPENSSL_free(pinfo);
904                 return NULL;
905                 }
906         return pinfo;
907         }
908
909 static void capi_dump_prov_info(CAPI_CTX *ctx, BIO *out, CRYPT_KEY_PROV_INFO *pinfo)
910         {
911         char *provname = NULL, *contname = NULL;
912         if (!pinfo)
913                 {
914                 BIO_printf(out, "  No Private Key\n");
915                 return;
916                 }
917         provname = wide_to_asc(pinfo->pwszProvName);
918         contname = wide_to_asc(pinfo->pwszContainerName);
919         if (!provname || !contname)
920                 goto err;
921
922         BIO_printf(out, "  Private Key Info:\n");
923         BIO_printf(out, "    Provider Name:  %s, Provider Type %d\n", provname, pinfo->dwProvType);
924         BIO_printf(out, "    Container Name: %s, Key Type %d\n", contname, pinfo->dwKeySpec);
925         err:
926         if (provname)
927                 OPENSSL_free(provname);
928         if (contname)
929                 OPENSSL_free(contname);
930         }
931
932 char * capi_cert_get_fname(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
933         {
934         LPWSTR wfname;
935         DWORD dlen;
936
937         CAPI_trace(ctx, "capi_cert_get_fname\n");
938         if (!CertGetCertificateContextProperty(cert, CERT_FRIENDLY_NAME_PROP_ID, NULL, &dlen))
939                 return NULL;
940         wfname = OPENSSL_malloc(dlen);
941         if (CertGetCertificateContextProperty(cert, CERT_FRIENDLY_NAME_PROP_ID, wfname, &dlen))
942                 {
943                 char *fname = wide_to_asc(wfname);
944                 OPENSSL_free(wfname);
945                 return fname;
946                 }
947         CAPIerr(CAPI_F_CAPI_CERT_GET_FNAME, CAPI_R_ERROR_GETTING_FRIENDLY_NAME);
948         capi_addlasterror();
949
950         OPENSSL_free(wfname);
951         return NULL;
952         }
953
954
955 void capi_dump_cert(CAPI_CTX *ctx, BIO *out, PCCERT_CONTEXT cert)
956         {
957         X509 *x;
958         unsigned char *p;
959         unsigned long flags = ctx->dump_flags;
960         if (flags & CAPI_DMP_FNAME)
961                 {
962                 char *fname;
963                 fname = capi_cert_get_fname(ctx, cert);
964                 if (fname)
965                         {
966                         BIO_printf(out, "  Friendly Name \"%s\"\n", fname);
967                         OPENSSL_free(fname);
968                         }
969                 else
970                         BIO_printf(out, "  <No Friendly Name>\n");
971                 }
972
973         p = cert->pbCertEncoded;
974         x = d2i_X509(NULL, &p, cert->cbCertEncoded);
975         if (!x)
976                 BIO_printf(out, "  <Can't parse certificate>\n");
977         if (flags & CAPI_DMP_SUMMARY)
978                 {
979                 BIO_printf(out, "  Subject: ");
980                 X509_NAME_print_ex(out, X509_get_subject_name(x), 0, XN_FLAG_ONELINE);
981                 BIO_printf(out, "\n  Issuer: ");
982                 X509_NAME_print_ex(out, X509_get_issuer_name(x), 0, XN_FLAG_ONELINE);
983                 BIO_printf(out, "\n");
984                 }
985         if (flags & CAPI_DMP_FULL)
986                 X509_print_ex(out, x, XN_FLAG_ONELINE,0);
987
988         if (flags & CAPI_DMP_PKEYINFO)
989                 {
990                 CRYPT_KEY_PROV_INFO *pinfo;
991                 pinfo = capi_get_prov_info(ctx, cert);
992                 capi_dump_prov_info(ctx, out, pinfo);
993                 if (pinfo)
994                         OPENSSL_free(pinfo);
995                 }
996
997         if (flags & CAPI_DMP_PEM)
998                 PEM_write_bio_X509(out, x);
999         X509_free(x);
1000         }
1001
1002 HCERTSTORE capi_open_store(CAPI_CTX *ctx, char *storename)
1003         {
1004         HCERTSTORE hstore;
1005
1006         if (!storename)
1007                 storename = ctx->storename;
1008         if (!storename)
1009                 storename = "MY";
1010         CAPI_trace(ctx, "Opening certificate store %s\n", storename);
1011
1012         hstore = CertOpenSystemStore(0, storename);
1013         if (!hstore)
1014                 {
1015                 CAPIerr(CAPI_F_CAPI_OPEN_STORE, CAPI_R_ERROR_OPENING_STORE);
1016                 capi_addlasterror();
1017                 }
1018         return hstore;
1019         }
1020
1021 int capi_list_certs(CAPI_CTX *ctx, BIO *out, char *id)
1022         {
1023         char *storename;
1024         int idx;
1025         int ret = 1;
1026         HCERTSTORE hstore;
1027         PCCERT_CONTEXT cert = NULL;
1028
1029         storename = ctx->storename;
1030         if (!storename)
1031                 storename = "MY";
1032         CAPI_trace(ctx, "Listing certs for store %s\n", storename);
1033
1034         hstore = capi_open_store(ctx, storename);
1035         if (!hstore)
1036                 return 0;
1037         if (id)
1038                 {
1039                 cert = capi_find_cert(ctx, id, hstore);
1040                 if (!cert)
1041                         {
1042                         ret = 0;
1043                         goto err;
1044                         }
1045                 capi_dump_cert(ctx, out, cert);
1046                 CertFreeCertificateContext(cert);
1047                 }
1048         else
1049                 {
1050                 for(idx = 0;;idx++)
1051                         {
1052                         LPWSTR fname = NULL;
1053                         cert = CertEnumCertificatesInStore(hstore, cert);
1054                         if (!cert)
1055                                 break;
1056                         BIO_printf(out, "Certificate %d\n", idx);
1057                         capi_dump_cert(ctx, out, cert);
1058                         }
1059                 }
1060         err:
1061         CertCloseStore(hstore, 0);
1062         return ret;
1063         }
1064
1065 static PCCERT_CONTEXT capi_find_cert(CAPI_CTX *ctx, const char *id, HCERTSTORE hstore)
1066         {
1067         PCCERT_CONTEXT cert = NULL;
1068         char *fname = NULL;
1069         int match;
1070         switch(ctx->lookup_method)
1071                 {
1072                 case CAPI_LU_SUBSTR:
1073                         return CertFindCertificateInStore(hstore, X509_ASN_ENCODING, 0, 
1074                                                                                         CERT_FIND_SUBJECT_STR_A, id, NULL);
1075                 case CAPI_LU_FNAME:
1076                         for(;;)
1077                                 {
1078                                 cert = CertEnumCertificatesInStore(hstore, cert);
1079                                 if (!cert)
1080                                         return NULL;
1081                                 fname = capi_cert_get_fname(ctx, cert);
1082                                 if (fname)
1083                                         {
1084                                         if (strcmp(fname, id))
1085                                                 match = 0;
1086                                         else
1087                                                 match = 1;
1088                                         OPENSSL_free(fname);
1089                                         if (match)
1090                                                 return cert;
1091                                         }
1092                                 }
1093                 default:
1094                         return NULL;
1095                 }
1096         }
1097
1098 static CAPI_KEY *capi_get_key(CAPI_CTX *ctx, const char *contname, char *provname, DWORD ptype, DWORD keyspec)
1099         {
1100         CAPI_KEY *key;
1101         key = OPENSSL_malloc(sizeof(CAPI_KEY));
1102         CAPI_trace(ctx, "capi_get_key, contname=%s, provname=%s, type=%d\n", 
1103                                                                                                                         contname, provname, ptype);
1104         if (!CryptAcquireContext(&key->hprov, contname, provname, ptype, 0))
1105                 {
1106                 CAPIerr(CAPI_F_CAPI_GET_KEY, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
1107                 capi_addlasterror();
1108                 goto err;
1109                 }
1110         if (!CryptGetUserKey(key->hprov, keyspec, &key->key))
1111                 {
1112                 CAPIerr(CAPI_F_CAPI_GET_KEY, CAPI_R_GETUSERKEY_ERROR);
1113                 capi_addlasterror();
1114                 CryptReleaseContext(key->hprov, 0);
1115                 goto err;
1116                 }
1117         return key;
1118
1119         err:
1120         OPENSSL_free(key);
1121         return NULL;
1122         }
1123
1124 static CAPI_KEY *capi_get_cert_key(CAPI_CTX *ctx, PCCERT_CONTEXT cert)
1125         {
1126         CAPI_KEY *key;
1127         CRYPT_KEY_PROV_INFO *pinfo = NULL;
1128         char *provname = NULL, *contname = NULL;
1129         pinfo = capi_get_prov_info(ctx, cert);
1130         if (!pinfo)
1131                 goto err;
1132         provname = wide_to_asc(pinfo->pwszProvName);
1133         contname = wide_to_asc(pinfo->pwszContainerName);
1134         if (!provname || !contname)
1135                 return 0;
1136
1137         key = capi_get_key(ctx, contname, provname, pinfo->dwProvType, pinfo->dwKeySpec);
1138
1139         err:
1140         if (pinfo)
1141                 OPENSSL_free(pinfo);
1142         if (provname)
1143                 OPENSSL_free(provname);
1144         if (contname)
1145                 OPENSSL_free(contname);
1146         return key;
1147         }
1148
1149 CAPI_KEY *capi_find_key(CAPI_CTX *ctx, const char *id)
1150         {
1151         PCCERT_CONTEXT cert;
1152         HCERTSTORE hstore;
1153         CAPI_KEY *key = NULL;
1154         switch (ctx->lookup_method)
1155                 {
1156                 case CAPI_LU_SUBSTR:
1157                 case CAPI_LU_FNAME:
1158                 hstore = capi_open_store(ctx, NULL);
1159                 if (!hstore)
1160                         return NULL;
1161                 cert = capi_find_cert(ctx, id, hstore);
1162                 if (cert)
1163                         {
1164                         key = capi_get_cert_key(ctx, cert);
1165                         CertFreeCertificateContext(cert);
1166                         }
1167                 CertCloseStore(hstore, 0);
1168                 break;
1169
1170                 case CAPI_LU_CONTNAME:
1171                 key = capi_get_key(ctx, id, ctx->cspname, ctx->csptype, ctx->keytype);
1172                 break;
1173                 }
1174
1175         return key;
1176         }
1177
1178 void capi_free_key(CAPI_KEY *key)
1179         {
1180         if (!key)
1181                 return;
1182         CryptDestroyKey(key->key);
1183         CryptReleaseContext(key->hprov, 0);
1184         OPENSSL_free(key);
1185         }
1186
1187
1188 /* Initialize a CAPI_CTX structure */
1189
1190 static CAPI_CTX *capi_ctx_new()
1191         {
1192         CAPI_CTX *ctx;
1193         ctx = OPENSSL_malloc(sizeof(CAPI_CTX));
1194         if (!ctx)
1195                 {
1196                 CAPIerr(CAPI_F_CAPI_CTX_NEW, ERR_R_MALLOC_FAILURE);
1197                 return NULL;
1198                 }
1199         ctx->cspname = NULL;
1200         ctx->csptype = PROV_RSA_FULL;
1201         ctx->dump_flags = CAPI_DMP_SUMMARY|CAPI_DMP_FNAME;
1202         ctx->keytype = AT_KEYEXCHANGE;
1203         ctx->storename = NULL;
1204         ctx->lookup_method = CAPI_LU_SUBSTR;
1205         ctx->debug_level = 0;
1206         ctx->debug_file = NULL;
1207         return ctx;
1208         }
1209
1210 static void capi_ctx_free(CAPI_CTX *ctx)
1211         {
1212         CAPI_trace(ctx, "Calling capi_ctx_free with %lx\n", ctx);
1213         if (!ctx)
1214                 return;
1215         if (ctx->cspname)
1216                 OPENSSL_free(ctx->cspname);
1217         if (ctx->debug_file)
1218                 OPENSSL_free(ctx->debug_file);
1219         if (ctx->storename)
1220                 OPENSSL_free(ctx->storename);
1221         OPENSSL_free(ctx);
1222         }
1223
1224 static int capi_ctx_set_provname(CAPI_CTX *ctx, LPSTR pname, DWORD type, int check)
1225         {
1226         CAPI_trace(ctx, "capi_ctx_set_provname, name=%s, type=%d\n", pname, type);
1227         if (check)
1228                 {
1229                 HCRYPTPROV hprov;
1230                 if (!CryptAcquireContext(&hprov, NULL, pname, type, CRYPT_VERIFYCONTEXT))
1231                         {
1232                         CAPIerr(CAPI_F_CAPI_CTX_SET_PROVNAME, CAPI_R_CRYPTACQUIRECONTEXT_ERROR);
1233                         capi_addlasterror();
1234                         return 0;
1235                         }
1236                 CryptReleaseContext(hprov, 0);
1237                 }
1238         ctx->cspname = BUF_strdup(pname);
1239         ctx->csptype = type;
1240         return 1;
1241         }
1242
1243 static int capi_ctx_set_provname_idx(CAPI_CTX *ctx, int idx)
1244         {
1245         LPSTR pname;
1246         DWORD type;
1247         if (capi_get_provname(ctx, &pname, &type, idx) != 1)
1248                 return 0;
1249         return capi_ctx_set_provname(ctx, pname, type, 0);
1250         }
1251
1252 #endif
1253 #endif