xref: /rk3399_ARM-atf/tools/cert_create/src/cert.c (revision 6311f63de02ee04d93016242977ade4727089de8)
1 /*
2  * Copyright (c) 2015-2017, ARM Limited and Contributors. All rights reserved.
3  *
4  * SPDX-License-Identifier: BSD-3-Clause
5  */
6 
7 #include <stdio.h>
8 #include <stdlib.h>
9 #include <string.h>
10 
11 #include <openssl/conf.h>
12 #include <openssl/err.h>
13 #include <openssl/opensslv.h>
14 #include <openssl/pem.h>
15 #include <openssl/sha.h>
16 #include <openssl/x509v3.h>
17 
18 #if USE_TBBR_DEFS
19 #include <tbbr_oid.h>
20 #else
21 #include <platform_oid.h>
22 #endif
23 
24 #include "cert.h"
25 #include "cmd_opt.h"
26 #include "debug.h"
27 #include "key.h"
28 #include "sha.h"
29 
30 #define SERIAL_RAND_BITS	64
31 #define RSA_SALT_LEN		32
32 
33 int rand_serial(BIGNUM *b, ASN1_INTEGER *ai)
34 {
35 	BIGNUM *btmp;
36 	int ret = 0;
37 	if (b)
38 		btmp = b;
39 	else
40 		btmp = BN_new();
41 
42 	if (!btmp)
43 		return 0;
44 
45 	if (!BN_pseudo_rand(btmp, SERIAL_RAND_BITS, 0, 0))
46 		goto error;
47 	if (ai && !BN_to_ASN1_INTEGER(btmp, ai))
48 		goto error;
49 
50 	ret = 1;
51 
52 error:
53 
54 	if (!b)
55 		BN_free(btmp);
56 
57 	return ret;
58 }
59 
60 int cert_add_ext(X509 *issuer, X509 *subject, int nid, char *value)
61 {
62 	X509_EXTENSION *ex;
63 	X509V3_CTX ctx;
64 
65 	/* No configuration database */
66 	X509V3_set_ctx_nodb(&ctx);
67 
68 	/* Set issuer and subject certificates in the context */
69 	X509V3_set_ctx(&ctx, issuer, subject, NULL, NULL, 0);
70 	ex = X509V3_EXT_conf_nid(NULL, &ctx, nid, value);
71 	if (!ex) {
72 		ERR_print_errors_fp(stdout);
73 		return 0;
74 	}
75 
76 	X509_add_ext(subject, ex, -1);
77 	X509_EXTENSION_free(ex);
78 
79 	return 1;
80 }
81 
82 int cert_new(cert_t *cert, int days, int ca, STACK_OF(X509_EXTENSION) * sk)
83 {
84 	EVP_PKEY *pkey = keys[cert->key].key;
85 	cert_t *issuer_cert = &certs[cert->issuer];
86 	EVP_PKEY *ikey = keys[issuer_cert->key].key;
87 	X509 *issuer = issuer_cert->x;
88 	X509 *x;
89 	X509_EXTENSION *ex;
90 	X509_NAME *name;
91 	ASN1_INTEGER *sno;
92 	int i, num, rc = 0;
93 	EVP_MD_CTX  mdCtx;
94 	EVP_PKEY_CTX *pKeyCtx = NULL;
95 
96 	/* Create the certificate structure */
97 	x = X509_new();
98 	if (!x) {
99 		return 0;
100 	}
101 
102 	/* If we do not have a key, use the issuer key (the certificate will
103 	 * become self signed). This happens in content certificates. */
104 	if (!pkey) {
105 		pkey = ikey;
106 	}
107 
108 	/* If we do not have an issuer certificate, use our own (the certificate
109 	 * will become self signed) */
110 	if (!issuer) {
111 		issuer = x;
112 	}
113 
114 	EVP_MD_CTX_init(&mdCtx);
115 	if (!EVP_DigestSignInit(&mdCtx, &pKeyCtx, EVP_sha256(), NULL, ikey)) {
116 		ERR_print_errors_fp(stdout);
117 		goto END;
118 	}
119 
120 	if (!EVP_PKEY_CTX_set_rsa_padding(pKeyCtx, RSA_PKCS1_PSS_PADDING)) {
121 		ERR_print_errors_fp(stdout);
122 		goto END;
123 	}
124 
125 	if (!EVP_PKEY_CTX_set_rsa_pss_saltlen(pKeyCtx, RSA_SALT_LEN)) {
126 		ERR_print_errors_fp(stdout);
127 		goto END;
128 	}
129 
130 	if (!EVP_PKEY_CTX_set_rsa_mgf1_md(pKeyCtx, EVP_sha256())) {
131 		ERR_print_errors_fp(stdout);
132 		goto END;
133 	}
134 
135 	/* x509.v3 */
136 	X509_set_version(x, 2);
137 
138 	/* Random serial number */
139 	sno = ASN1_INTEGER_new();
140 	rand_serial(NULL, sno);
141 	X509_set_serialNumber(x, sno);
142 	ASN1_INTEGER_free(sno);
143 
144 	X509_gmtime_adj(X509_get_notBefore(x), 0);
145 	X509_gmtime_adj(X509_get_notAfter(x), (long)60*60*24*days);
146 	X509_set_pubkey(x, pkey);
147 
148 	/* Subject name */
149 	name = X509_get_subject_name(x);
150 	X509_NAME_add_entry_by_txt(name, "CN", MBSTRING_ASC,
151 			(const unsigned char *)cert->cn, -1, -1, 0);
152 	X509_set_subject_name(x, name);
153 
154 	/* Issuer name */
155 	name = X509_get_issuer_name(x);
156 	X509_NAME_add_entry_by_txt(name, "CN", MBSTRING_ASC,
157 			(const unsigned char *)issuer_cert->cn, -1, -1, 0);
158 	X509_set_issuer_name(x, name);
159 
160 	/* Add various extensions: standard extensions */
161 	cert_add_ext(issuer, x, NID_subject_key_identifier, "hash");
162 	cert_add_ext(issuer, x, NID_authority_key_identifier, "keyid:always");
163 	if (ca) {
164 		cert_add_ext(issuer, x, NID_basic_constraints, "CA:TRUE");
165 		cert_add_ext(issuer, x, NID_key_usage, "keyCertSign");
166 	} else {
167 		cert_add_ext(issuer, x, NID_basic_constraints, "CA:FALSE");
168 	}
169 
170 	/* Add custom extensions */
171 	if (sk != NULL) {
172 		num = sk_X509_EXTENSION_num(sk);
173 		for (i = 0; i < num; i++) {
174 			ex = sk_X509_EXTENSION_value(sk, i);
175 			X509_add_ext(x, ex, -1);
176 		}
177 	}
178 
179 	if (!X509_sign_ctx(x, &mdCtx)) {
180 		ERR_print_errors_fp(stdout);
181 		goto END;
182 	}
183 
184 	/* X509 certificate signed successfully */
185 	rc = 1;
186 	cert->x = x;
187 
188 END:
189 	EVP_MD_CTX_cleanup(&mdCtx);
190 	return rc;
191 }
192 
193 int cert_init(void)
194 {
195 	cmd_opt_t cmd_opt;
196 	cert_t *cert;
197 	unsigned int i;
198 
199 	for (i = 0; i < num_certs; i++) {
200 		cert = &certs[i];
201 		cmd_opt.long_opt.name = cert->opt;
202 		cmd_opt.long_opt.has_arg = required_argument;
203 		cmd_opt.long_opt.flag = NULL;
204 		cmd_opt.long_opt.val = CMD_OPT_CERT;
205 		cmd_opt.help_msg = cert->help_msg;
206 		cmd_opt_add(&cmd_opt);
207 	}
208 
209 	return 0;
210 }
211 
212 cert_t *cert_get_by_opt(const char *opt)
213 {
214 	cert_t *cert;
215 	unsigned int i;
216 
217 	for (i = 0; i < num_certs; i++) {
218 		cert = &certs[i];
219 		if (0 == strcmp(cert->opt, opt)) {
220 			return cert;
221 		}
222 	}
223 
224 	return NULL;
225 }
226