119c402afSSimon Glass /* 219c402afSSimon Glass * Copyright (c) 2013, Google Inc. 319c402afSSimon Glass * 41a459660SWolfgang Denk * SPDX-License-Identifier: GPL-2.0+ 519c402afSSimon Glass */ 619c402afSSimon Glass 719c402afSSimon Glass #include "mkimage.h" 819c402afSSimon Glass #include <stdio.h> 919c402afSSimon Glass #include <string.h> 1019c402afSSimon Glass #include <image.h> 1119c402afSSimon Glass #include <time.h> 12c3b43281SJelle van der Waa #include <openssl/bn.h> 1319c402afSSimon Glass #include <openssl/rsa.h> 1419c402afSSimon Glass #include <openssl/pem.h> 1519c402afSSimon Glass #include <openssl/err.h> 1619c402afSSimon Glass #include <openssl/ssl.h> 1719c402afSSimon Glass #include <openssl/evp.h> 18f1ca1fdeSGeorge McCollister #include <openssl/engine.h> 1919c402afSSimon Glass 2019c402afSSimon Glass #if OPENSSL_VERSION_NUMBER >= 0x10000000L 2119c402afSSimon Glass #define HAVE_ERR_REMOVE_THREAD_STATE 2219c402afSSimon Glass #endif 2319c402afSSimon Glass 24c3b43281SJelle van der Waa #if OPENSSL_VERSION_NUMBER < 0x10100000L 25c3b43281SJelle van der Waa static void RSA_get0_key(const RSA *r, 26c3b43281SJelle van der Waa const BIGNUM **n, const BIGNUM **e, const BIGNUM **d) 27c3b43281SJelle van der Waa { 28c3b43281SJelle van der Waa if (n != NULL) 29c3b43281SJelle van der Waa *n = r->n; 30c3b43281SJelle van der Waa if (e != NULL) 31c3b43281SJelle van der Waa *e = r->e; 32c3b43281SJelle van der Waa if (d != NULL) 33c3b43281SJelle van der Waa *d = r->d; 34c3b43281SJelle van der Waa } 35c3b43281SJelle van der Waa #endif 36c3b43281SJelle van der Waa 3719c402afSSimon Glass static int rsa_err(const char *msg) 3819c402afSSimon Glass { 3919c402afSSimon Glass unsigned long sslErr = ERR_get_error(); 4019c402afSSimon Glass 4119c402afSSimon Glass fprintf(stderr, "%s", msg); 4219c402afSSimon Glass fprintf(stderr, ": %s\n", 4319c402afSSimon Glass ERR_error_string(sslErr, 0)); 4419c402afSSimon Glass 4519c402afSSimon Glass return -1; 4619c402afSSimon Glass } 4719c402afSSimon Glass 4819c402afSSimon Glass /** 49f1ca1fdeSGeorge McCollister * rsa_pem_get_pub_key() - read a public key from a .crt file 5019c402afSSimon Glass * 5119c402afSSimon Glass * @keydir: Directory containins the key 5219c402afSSimon Glass * @name Name of key file (will have a .crt extension) 5319c402afSSimon Glass * @rsap Returns RSA object, or NULL on failure 5419c402afSSimon Glass * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 5519c402afSSimon Glass */ 56f1ca1fdeSGeorge McCollister static int rsa_pem_get_pub_key(const char *keydir, const char *name, RSA **rsap) 5719c402afSSimon Glass { 5819c402afSSimon Glass char path[1024]; 5919c402afSSimon Glass EVP_PKEY *key; 6019c402afSSimon Glass X509 *cert; 6119c402afSSimon Glass RSA *rsa; 6219c402afSSimon Glass FILE *f; 6319c402afSSimon Glass int ret; 6419c402afSSimon Glass 6519c402afSSimon Glass *rsap = NULL; 6619c402afSSimon Glass snprintf(path, sizeof(path), "%s/%s.crt", keydir, name); 6719c402afSSimon Glass f = fopen(path, "r"); 6819c402afSSimon Glass if (!f) { 6919c402afSSimon Glass fprintf(stderr, "Couldn't open RSA certificate: '%s': %s\n", 7019c402afSSimon Glass path, strerror(errno)); 7119c402afSSimon Glass return -EACCES; 7219c402afSSimon Glass } 7319c402afSSimon Glass 7419c402afSSimon Glass /* Read the certificate */ 7519c402afSSimon Glass cert = NULL; 7619c402afSSimon Glass if (!PEM_read_X509(f, &cert, NULL, NULL)) { 7719c402afSSimon Glass rsa_err("Couldn't read certificate"); 7819c402afSSimon Glass ret = -EINVAL; 7919c402afSSimon Glass goto err_cert; 8019c402afSSimon Glass } 8119c402afSSimon Glass 8219c402afSSimon Glass /* Get the public key from the certificate. */ 8319c402afSSimon Glass key = X509_get_pubkey(cert); 8419c402afSSimon Glass if (!key) { 8519c402afSSimon Glass rsa_err("Couldn't read public key\n"); 8619c402afSSimon Glass ret = -EINVAL; 8719c402afSSimon Glass goto err_pubkey; 8819c402afSSimon Glass } 8919c402afSSimon Glass 9019c402afSSimon Glass /* Convert to a RSA_style key. */ 9119c402afSSimon Glass rsa = EVP_PKEY_get1_RSA(key); 9219c402afSSimon Glass if (!rsa) { 9319c402afSSimon Glass rsa_err("Couldn't convert to a RSA style key"); 9454267162SSimon Glass ret = -EINVAL; 9519c402afSSimon Glass goto err_rsa; 9619c402afSSimon Glass } 9719c402afSSimon Glass fclose(f); 9819c402afSSimon Glass EVP_PKEY_free(key); 9919c402afSSimon Glass X509_free(cert); 10019c402afSSimon Glass *rsap = rsa; 10119c402afSSimon Glass 10219c402afSSimon Glass return 0; 10319c402afSSimon Glass 10419c402afSSimon Glass err_rsa: 10519c402afSSimon Glass EVP_PKEY_free(key); 10619c402afSSimon Glass err_pubkey: 10719c402afSSimon Glass X509_free(cert); 10819c402afSSimon Glass err_cert: 10919c402afSSimon Glass fclose(f); 11019c402afSSimon Glass return ret; 11119c402afSSimon Glass } 11219c402afSSimon Glass 11319c402afSSimon Glass /** 114f1ca1fdeSGeorge McCollister * rsa_engine_get_pub_key() - read a public key from given engine 11519c402afSSimon Glass * 116f1ca1fdeSGeorge McCollister * @keydir: Key prefix 117f1ca1fdeSGeorge McCollister * @name Name of key 118f1ca1fdeSGeorge McCollister * @engine Engine to use 119f1ca1fdeSGeorge McCollister * @rsap Returns RSA object, or NULL on failure 120f1ca1fdeSGeorge McCollister * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 121f1ca1fdeSGeorge McCollister */ 122f1ca1fdeSGeorge McCollister static int rsa_engine_get_pub_key(const char *keydir, const char *name, 123f1ca1fdeSGeorge McCollister ENGINE *engine, RSA **rsap) 124f1ca1fdeSGeorge McCollister { 125f1ca1fdeSGeorge McCollister const char *engine_id; 126f1ca1fdeSGeorge McCollister char key_id[1024]; 127f1ca1fdeSGeorge McCollister EVP_PKEY *key; 128f1ca1fdeSGeorge McCollister RSA *rsa; 129f1ca1fdeSGeorge McCollister int ret; 130f1ca1fdeSGeorge McCollister 131f1ca1fdeSGeorge McCollister *rsap = NULL; 132f1ca1fdeSGeorge McCollister 133f1ca1fdeSGeorge McCollister engine_id = ENGINE_get_id(engine); 134f1ca1fdeSGeorge McCollister 135f1ca1fdeSGeorge McCollister if (engine_id && !strcmp(engine_id, "pkcs11")) { 136f1ca1fdeSGeorge McCollister if (keydir) 137f1ca1fdeSGeorge McCollister snprintf(key_id, sizeof(key_id), 138f1ca1fdeSGeorge McCollister "pkcs11:%s;object=%s;type=public", 139f1ca1fdeSGeorge McCollister keydir, name); 140f1ca1fdeSGeorge McCollister else 141f1ca1fdeSGeorge McCollister snprintf(key_id, sizeof(key_id), 142f1ca1fdeSGeorge McCollister "pkcs11:object=%s;type=public", 143f1ca1fdeSGeorge McCollister name); 144f1ca1fdeSGeorge McCollister } else { 145f1ca1fdeSGeorge McCollister fprintf(stderr, "Engine not supported\n"); 146f1ca1fdeSGeorge McCollister return -ENOTSUP; 147f1ca1fdeSGeorge McCollister } 148f1ca1fdeSGeorge McCollister 149f1ca1fdeSGeorge McCollister key = ENGINE_load_public_key(engine, key_id, NULL, NULL); 150f1ca1fdeSGeorge McCollister if (!key) 151f1ca1fdeSGeorge McCollister return rsa_err("Failure loading public key from engine"); 152f1ca1fdeSGeorge McCollister 153f1ca1fdeSGeorge McCollister /* Convert to a RSA_style key. */ 154f1ca1fdeSGeorge McCollister rsa = EVP_PKEY_get1_RSA(key); 155f1ca1fdeSGeorge McCollister if (!rsa) { 156f1ca1fdeSGeorge McCollister rsa_err("Couldn't convert to a RSA style key"); 157f1ca1fdeSGeorge McCollister ret = -EINVAL; 158f1ca1fdeSGeorge McCollister goto err_rsa; 159f1ca1fdeSGeorge McCollister } 160f1ca1fdeSGeorge McCollister 161f1ca1fdeSGeorge McCollister EVP_PKEY_free(key); 162f1ca1fdeSGeorge McCollister *rsap = rsa; 163f1ca1fdeSGeorge McCollister 164f1ca1fdeSGeorge McCollister return 0; 165f1ca1fdeSGeorge McCollister 166f1ca1fdeSGeorge McCollister err_rsa: 167f1ca1fdeSGeorge McCollister EVP_PKEY_free(key); 168f1ca1fdeSGeorge McCollister return ret; 169f1ca1fdeSGeorge McCollister } 170f1ca1fdeSGeorge McCollister 171f1ca1fdeSGeorge McCollister /** 172f1ca1fdeSGeorge McCollister * rsa_get_pub_key() - read a public key 173f1ca1fdeSGeorge McCollister * 174f1ca1fdeSGeorge McCollister * @keydir: Directory containing the key (PEM file) or key prefix (engine) 175f1ca1fdeSGeorge McCollister * @name Name of key file (will have a .crt extension) 176f1ca1fdeSGeorge McCollister * @engine Engine to use 177f1ca1fdeSGeorge McCollister * @rsap Returns RSA object, or NULL on failure 178f1ca1fdeSGeorge McCollister * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 179f1ca1fdeSGeorge McCollister */ 180f1ca1fdeSGeorge McCollister static int rsa_get_pub_key(const char *keydir, const char *name, 181f1ca1fdeSGeorge McCollister ENGINE *engine, RSA **rsap) 182f1ca1fdeSGeorge McCollister { 183f1ca1fdeSGeorge McCollister if (engine) 184f1ca1fdeSGeorge McCollister return rsa_engine_get_pub_key(keydir, name, engine, rsap); 185f1ca1fdeSGeorge McCollister return rsa_pem_get_pub_key(keydir, name, rsap); 186f1ca1fdeSGeorge McCollister } 187f1ca1fdeSGeorge McCollister 188f1ca1fdeSGeorge McCollister /** 189f1ca1fdeSGeorge McCollister * rsa_pem_get_priv_key() - read a private key from a .key file 190f1ca1fdeSGeorge McCollister * 191f1ca1fdeSGeorge McCollister * @keydir: Directory containing the key 19219c402afSSimon Glass * @name Name of key file (will have a .key extension) 19319c402afSSimon Glass * @rsap Returns RSA object, or NULL on failure 19419c402afSSimon Glass * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 19519c402afSSimon Glass */ 196f1ca1fdeSGeorge McCollister static int rsa_pem_get_priv_key(const char *keydir, const char *name, 197f1ca1fdeSGeorge McCollister RSA **rsap) 19819c402afSSimon Glass { 19919c402afSSimon Glass char path[1024]; 20019c402afSSimon Glass RSA *rsa; 20119c402afSSimon Glass FILE *f; 20219c402afSSimon Glass 20319c402afSSimon Glass *rsap = NULL; 20419c402afSSimon Glass snprintf(path, sizeof(path), "%s/%s.key", keydir, name); 20519c402afSSimon Glass f = fopen(path, "r"); 20619c402afSSimon Glass if (!f) { 20719c402afSSimon Glass fprintf(stderr, "Couldn't open RSA private key: '%s': %s\n", 20819c402afSSimon Glass path, strerror(errno)); 20919c402afSSimon Glass return -ENOENT; 21019c402afSSimon Glass } 21119c402afSSimon Glass 21219c402afSSimon Glass rsa = PEM_read_RSAPrivateKey(f, 0, NULL, path); 21319c402afSSimon Glass if (!rsa) { 21419c402afSSimon Glass rsa_err("Failure reading private key"); 21519c402afSSimon Glass fclose(f); 21619c402afSSimon Glass return -EPROTO; 21719c402afSSimon Glass } 21819c402afSSimon Glass fclose(f); 21919c402afSSimon Glass *rsap = rsa; 22019c402afSSimon Glass 22119c402afSSimon Glass return 0; 22219c402afSSimon Glass } 22319c402afSSimon Glass 224f1ca1fdeSGeorge McCollister /** 225f1ca1fdeSGeorge McCollister * rsa_engine_get_priv_key() - read a private key from given engine 226f1ca1fdeSGeorge McCollister * 227f1ca1fdeSGeorge McCollister * @keydir: Key prefix 228f1ca1fdeSGeorge McCollister * @name Name of key 229f1ca1fdeSGeorge McCollister * @engine Engine to use 230f1ca1fdeSGeorge McCollister * @rsap Returns RSA object, or NULL on failure 231f1ca1fdeSGeorge McCollister * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 232f1ca1fdeSGeorge McCollister */ 233f1ca1fdeSGeorge McCollister static int rsa_engine_get_priv_key(const char *keydir, const char *name, 234f1ca1fdeSGeorge McCollister ENGINE *engine, RSA **rsap) 235f1ca1fdeSGeorge McCollister { 236f1ca1fdeSGeorge McCollister const char *engine_id; 237f1ca1fdeSGeorge McCollister char key_id[1024]; 238f1ca1fdeSGeorge McCollister EVP_PKEY *key; 239f1ca1fdeSGeorge McCollister RSA *rsa; 240f1ca1fdeSGeorge McCollister int ret; 241f1ca1fdeSGeorge McCollister 242f1ca1fdeSGeorge McCollister *rsap = NULL; 243f1ca1fdeSGeorge McCollister 244f1ca1fdeSGeorge McCollister engine_id = ENGINE_get_id(engine); 245f1ca1fdeSGeorge McCollister 246f1ca1fdeSGeorge McCollister if (engine_id && !strcmp(engine_id, "pkcs11")) { 247f1ca1fdeSGeorge McCollister if (keydir) 248f1ca1fdeSGeorge McCollister snprintf(key_id, sizeof(key_id), 249f1ca1fdeSGeorge McCollister "pkcs11:%s;object=%s;type=private", 250f1ca1fdeSGeorge McCollister keydir, name); 251f1ca1fdeSGeorge McCollister else 252f1ca1fdeSGeorge McCollister snprintf(key_id, sizeof(key_id), 253f1ca1fdeSGeorge McCollister "pkcs11:object=%s;type=private", 254f1ca1fdeSGeorge McCollister name); 255f1ca1fdeSGeorge McCollister } else { 256f1ca1fdeSGeorge McCollister fprintf(stderr, "Engine not supported\n"); 257f1ca1fdeSGeorge McCollister return -ENOTSUP; 258f1ca1fdeSGeorge McCollister } 259f1ca1fdeSGeorge McCollister 260f1ca1fdeSGeorge McCollister key = ENGINE_load_private_key(engine, key_id, NULL, NULL); 261f1ca1fdeSGeorge McCollister if (!key) 262f1ca1fdeSGeorge McCollister return rsa_err("Failure loading private key from engine"); 263f1ca1fdeSGeorge McCollister 264f1ca1fdeSGeorge McCollister /* Convert to a RSA_style key. */ 265f1ca1fdeSGeorge McCollister rsa = EVP_PKEY_get1_RSA(key); 266f1ca1fdeSGeorge McCollister if (!rsa) { 267f1ca1fdeSGeorge McCollister rsa_err("Couldn't convert to a RSA style key"); 268f1ca1fdeSGeorge McCollister ret = -EINVAL; 269f1ca1fdeSGeorge McCollister goto err_rsa; 270f1ca1fdeSGeorge McCollister } 271f1ca1fdeSGeorge McCollister 272f1ca1fdeSGeorge McCollister EVP_PKEY_free(key); 273f1ca1fdeSGeorge McCollister *rsap = rsa; 274f1ca1fdeSGeorge McCollister 275f1ca1fdeSGeorge McCollister return 0; 276f1ca1fdeSGeorge McCollister 277f1ca1fdeSGeorge McCollister err_rsa: 278f1ca1fdeSGeorge McCollister EVP_PKEY_free(key); 279f1ca1fdeSGeorge McCollister return ret; 280f1ca1fdeSGeorge McCollister } 281f1ca1fdeSGeorge McCollister 282f1ca1fdeSGeorge McCollister /** 283f1ca1fdeSGeorge McCollister * rsa_get_priv_key() - read a private key 284f1ca1fdeSGeorge McCollister * 285f1ca1fdeSGeorge McCollister * @keydir: Directory containing the key (PEM file) or key prefix (engine) 286f1ca1fdeSGeorge McCollister * @name Name of key 287f1ca1fdeSGeorge McCollister * @engine Engine to use for signing 288f1ca1fdeSGeorge McCollister * @rsap Returns RSA object, or NULL on failure 289f1ca1fdeSGeorge McCollister * @return 0 if ok, -ve on error (in which case *rsap will be set to NULL) 290f1ca1fdeSGeorge McCollister */ 291f1ca1fdeSGeorge McCollister static int rsa_get_priv_key(const char *keydir, const char *name, 292f1ca1fdeSGeorge McCollister ENGINE *engine, RSA **rsap) 293f1ca1fdeSGeorge McCollister { 294f1ca1fdeSGeorge McCollister if (engine) 295f1ca1fdeSGeorge McCollister return rsa_engine_get_priv_key(keydir, name, engine, rsap); 296f1ca1fdeSGeorge McCollister return rsa_pem_get_priv_key(keydir, name, rsap); 297f1ca1fdeSGeorge McCollister } 298f1ca1fdeSGeorge McCollister 29919c402afSSimon Glass static int rsa_init(void) 30019c402afSSimon Glass { 30119c402afSSimon Glass int ret; 30219c402afSSimon Glass 303c3b43281SJelle van der Waa #if OPENSSL_VERSION_NUMBER < 0x10100000L 30419c402afSSimon Glass ret = SSL_library_init(); 305c3b43281SJelle van der Waa #else 306c3b43281SJelle van der Waa ret = OPENSSL_init_ssl(0, NULL); 307c3b43281SJelle van der Waa #endif 30819c402afSSimon Glass if (!ret) { 30919c402afSSimon Glass fprintf(stderr, "Failure to init SSL library\n"); 31019c402afSSimon Glass return -1; 31119c402afSSimon Glass } 312c3b43281SJelle van der Waa #if OPENSSL_VERSION_NUMBER < 0x10100000L 31319c402afSSimon Glass SSL_load_error_strings(); 31419c402afSSimon Glass 31519c402afSSimon Glass OpenSSL_add_all_algorithms(); 31619c402afSSimon Glass OpenSSL_add_all_digests(); 31719c402afSSimon Glass OpenSSL_add_all_ciphers(); 318c3b43281SJelle van der Waa #endif 31919c402afSSimon Glass 32019c402afSSimon Glass return 0; 32119c402afSSimon Glass } 32219c402afSSimon Glass 323f1ca1fdeSGeorge McCollister static int rsa_engine_init(const char *engine_id, ENGINE **pe) 324f1ca1fdeSGeorge McCollister { 325f1ca1fdeSGeorge McCollister ENGINE *e; 326f1ca1fdeSGeorge McCollister int ret; 327f1ca1fdeSGeorge McCollister 328f1ca1fdeSGeorge McCollister ENGINE_load_builtin_engines(); 329f1ca1fdeSGeorge McCollister 330f1ca1fdeSGeorge McCollister e = ENGINE_by_id(engine_id); 331f1ca1fdeSGeorge McCollister if (!e) { 332f1ca1fdeSGeorge McCollister fprintf(stderr, "Engine isn't available\n"); 333f1ca1fdeSGeorge McCollister ret = -1; 334f1ca1fdeSGeorge McCollister goto err_engine_by_id; 335f1ca1fdeSGeorge McCollister } 336f1ca1fdeSGeorge McCollister 337f1ca1fdeSGeorge McCollister if (!ENGINE_init(e)) { 338f1ca1fdeSGeorge McCollister fprintf(stderr, "Couldn't initialize engine\n"); 339f1ca1fdeSGeorge McCollister ret = -1; 340f1ca1fdeSGeorge McCollister goto err_engine_init; 341f1ca1fdeSGeorge McCollister } 342f1ca1fdeSGeorge McCollister 343f1ca1fdeSGeorge McCollister if (!ENGINE_set_default_RSA(e)) { 344f1ca1fdeSGeorge McCollister fprintf(stderr, "Couldn't set engine as default for RSA\n"); 345f1ca1fdeSGeorge McCollister ret = -1; 346f1ca1fdeSGeorge McCollister goto err_set_rsa; 347f1ca1fdeSGeorge McCollister } 348f1ca1fdeSGeorge McCollister 349f1ca1fdeSGeorge McCollister *pe = e; 350f1ca1fdeSGeorge McCollister 351f1ca1fdeSGeorge McCollister return 0; 352f1ca1fdeSGeorge McCollister 353f1ca1fdeSGeorge McCollister err_set_rsa: 354f1ca1fdeSGeorge McCollister ENGINE_finish(e); 355f1ca1fdeSGeorge McCollister err_engine_init: 356f1ca1fdeSGeorge McCollister ENGINE_free(e); 357f1ca1fdeSGeorge McCollister err_engine_by_id: 358c3b43281SJelle van der Waa #if OPENSSL_VERSION_NUMBER < 0x10100000L 359f1ca1fdeSGeorge McCollister ENGINE_cleanup(); 360c3b43281SJelle van der Waa #endif 361f1ca1fdeSGeorge McCollister return ret; 362f1ca1fdeSGeorge McCollister } 363f1ca1fdeSGeorge McCollister 36419c402afSSimon Glass static void rsa_remove(void) 36519c402afSSimon Glass { 366c3b43281SJelle van der Waa #if OPENSSL_VERSION_NUMBER < 0x10100000L 36719c402afSSimon Glass CRYPTO_cleanup_all_ex_data(); 36819c402afSSimon Glass ERR_free_strings(); 36919c402afSSimon Glass #ifdef HAVE_ERR_REMOVE_THREAD_STATE 37019c402afSSimon Glass ERR_remove_thread_state(NULL); 37119c402afSSimon Glass #else 37219c402afSSimon Glass ERR_remove_state(0); 37319c402afSSimon Glass #endif 37419c402afSSimon Glass EVP_cleanup(); 375c3b43281SJelle van der Waa #endif 37619c402afSSimon Glass } 37719c402afSSimon Glass 378f1ca1fdeSGeorge McCollister static void rsa_engine_remove(ENGINE *e) 379f1ca1fdeSGeorge McCollister { 380f1ca1fdeSGeorge McCollister if (e) { 381f1ca1fdeSGeorge McCollister ENGINE_finish(e); 382f1ca1fdeSGeorge McCollister ENGINE_free(e); 383f1ca1fdeSGeorge McCollister } 384f1ca1fdeSGeorge McCollister } 385f1ca1fdeSGeorge McCollister 386d46373c1SJoseph Chen /* 387d46373c1SJoseph Chen * With this data2sign.bin, we can provide it to who real holds the RAS-private 388d46373c1SJoseph Chen * key to sign current fit image. Then we replace the signature in fit image 389d46373c1SJoseph Chen * with a valid one. 390d46373c1SJoseph Chen */ 391242e56faSJoseph Chen static int gen_data2sign(const struct image_region region[], int region_count) 392d46373c1SJoseph Chen { 393d46373c1SJoseph Chen char *file = "data2sign.bin"; 394d46373c1SJoseph Chen FILE *fd; 395d46373c1SJoseph Chen int i; 396d46373c1SJoseph Chen 397d46373c1SJoseph Chen fd = fopen(file, "wb"); 398d46373c1SJoseph Chen if (!fd) { 399d46373c1SJoseph Chen fprintf(stderr, "Failed to create %s: %s\n", 400d46373c1SJoseph Chen file, strerror(errno)); 401d46373c1SJoseph Chen return -ENOENT; 402d46373c1SJoseph Chen } 403d46373c1SJoseph Chen 404d46373c1SJoseph Chen for (i = 0; i < region_count; i++) 405d46373c1SJoseph Chen fwrite(region[i].data, region[i].size, 1, fd); 406d46373c1SJoseph Chen 407d46373c1SJoseph Chen fclose(fd); 408242e56faSJoseph Chen 409242e56faSJoseph Chen return 0; 410d46373c1SJoseph Chen } 411d46373c1SJoseph Chen 412219050bfSPhilippe Reynes static int rsa_sign_with_key(RSA *rsa, struct padding_algo *padding_algo, 413219050bfSPhilippe Reynes struct checksum_algo *checksum_algo, 414646257d1SHeiko Schocher const struct image_region region[], int region_count, 415646257d1SHeiko Schocher uint8_t **sigp, uint *sig_size) 41619c402afSSimon Glass { 41719c402afSSimon Glass EVP_PKEY *key; 418219050bfSPhilippe Reynes EVP_PKEY_CTX *ckey; 41919c402afSSimon Glass EVP_MD_CTX *context; 420624f7c07SPhilippe Reynes int ret = 0; 421624f7c07SPhilippe Reynes size_t size; 42219c402afSSimon Glass uint8_t *sig; 42319c402afSSimon Glass int i; 42419c402afSSimon Glass 42519c402afSSimon Glass key = EVP_PKEY_new(); 42619c402afSSimon Glass if (!key) 42719c402afSSimon Glass return rsa_err("EVP_PKEY object creation failed"); 42819c402afSSimon Glass 42919c402afSSimon Glass if (!EVP_PKEY_set1_RSA(key, rsa)) { 43019c402afSSimon Glass ret = rsa_err("EVP key setup failed"); 43119c402afSSimon Glass goto err_set; 43219c402afSSimon Glass } 43319c402afSSimon Glass 43419c402afSSimon Glass size = EVP_PKEY_size(key); 43519c402afSSimon Glass sig = malloc(size); 43619c402afSSimon Glass if (!sig) { 437624f7c07SPhilippe Reynes fprintf(stderr, "Out of memory for signature (%zu bytes)\n", 43819c402afSSimon Glass size); 43919c402afSSimon Glass ret = -ENOMEM; 44019c402afSSimon Glass goto err_alloc; 44119c402afSSimon Glass } 44219c402afSSimon Glass 44319c402afSSimon Glass context = EVP_MD_CTX_create(); 44419c402afSSimon Glass if (!context) { 44519c402afSSimon Glass ret = rsa_err("EVP context creation failed"); 44619c402afSSimon Glass goto err_create; 44719c402afSSimon Glass } 44819c402afSSimon Glass EVP_MD_CTX_init(context); 449219050bfSPhilippe Reynes 450219050bfSPhilippe Reynes ckey = EVP_PKEY_CTX_new(key, NULL); 451219050bfSPhilippe Reynes if (!ckey) { 452219050bfSPhilippe Reynes ret = rsa_err("EVP key context creation failed"); 453219050bfSPhilippe Reynes goto err_create; 454219050bfSPhilippe Reynes } 455219050bfSPhilippe Reynes 456219050bfSPhilippe Reynes if (EVP_DigestSignInit(context, &ckey, 457624f7c07SPhilippe Reynes checksum_algo->calculate_sign(), 458624f7c07SPhilippe Reynes NULL, key) <= 0) { 45919c402afSSimon Glass ret = rsa_err("Signer setup failed"); 46019c402afSSimon Glass goto err_sign; 46119c402afSSimon Glass } 46219c402afSSimon Glass 46385289e9dSPhilippe Reynes #ifdef CONFIG_FIT_ENABLE_RSASSA_PSS_SUPPORT 46485289e9dSPhilippe Reynes if (padding_algo && !strcmp(padding_algo->name, "pss")) { 46585289e9dSPhilippe Reynes if (EVP_PKEY_CTX_set_rsa_padding(ckey, 46685289e9dSPhilippe Reynes RSA_PKCS1_PSS_PADDING) <= 0) { 46785289e9dSPhilippe Reynes ret = rsa_err("Signer padding setup failed"); 46885289e9dSPhilippe Reynes goto err_sign; 46985289e9dSPhilippe Reynes } 47085289e9dSPhilippe Reynes } 47185289e9dSPhilippe Reynes #endif /* CONFIG_FIT_ENABLE_RSASSA_PSS_SUPPORT */ 47285289e9dSPhilippe Reynes 47319c402afSSimon Glass for (i = 0; i < region_count; i++) { 474624f7c07SPhilippe Reynes if (!EVP_DigestSignUpdate(context, region[i].data, 475624f7c07SPhilippe Reynes region[i].size)) { 47619c402afSSimon Glass ret = rsa_err("Signing data failed"); 47719c402afSSimon Glass goto err_sign; 47819c402afSSimon Glass } 47919c402afSSimon Glass } 48019c402afSSimon Glass 481624f7c07SPhilippe Reynes if (!EVP_DigestSignFinal(context, sig, &size)) { 48219c402afSSimon Glass ret = rsa_err("Could not obtain signature"); 48319c402afSSimon Glass goto err_sign; 48419c402afSSimon Glass } 485624f7c07SPhilippe Reynes 486624f7c07SPhilippe Reynes #if OPENSSL_VERSION_NUMBER < 0x10100000L || \ 487624f7c07SPhilippe Reynes (defined(LIBRESSL_VERSION_NUMBER) && LIBRESSL_VERSION_NUMBER < 0x02070000fL) 48819c402afSSimon Glass EVP_MD_CTX_cleanup(context); 489c3b43281SJelle van der Waa #else 490c3b43281SJelle van der Waa EVP_MD_CTX_reset(context); 491c3b43281SJelle van der Waa #endif 49219c402afSSimon Glass EVP_MD_CTX_destroy(context); 49319c402afSSimon Glass EVP_PKEY_free(key); 49419c402afSSimon Glass 495624f7c07SPhilippe Reynes debug("Got signature: %d bytes, expected %zu\n", *sig_size, size); 49619c402afSSimon Glass *sigp = sig; 49719c402afSSimon Glass *sig_size = size; 49819c402afSSimon Glass 499d46373c1SJoseph Chen gen_data2sign(region, region_count); 500d46373c1SJoseph Chen 50119c402afSSimon Glass return 0; 50219c402afSSimon Glass 50319c402afSSimon Glass err_sign: 50419c402afSSimon Glass EVP_MD_CTX_destroy(context); 50519c402afSSimon Glass err_create: 50619c402afSSimon Glass free(sig); 50719c402afSSimon Glass err_alloc: 50819c402afSSimon Glass err_set: 50919c402afSSimon Glass EVP_PKEY_free(key); 51019c402afSSimon Glass return ret; 51119c402afSSimon Glass } 51219c402afSSimon Glass 51319c402afSSimon Glass int rsa_sign(struct image_sign_info *info, 51419c402afSSimon Glass const struct image_region region[], int region_count, 51519c402afSSimon Glass uint8_t **sigp, uint *sig_len) 51619c402afSSimon Glass { 51719c402afSSimon Glass RSA *rsa; 518f1ca1fdeSGeorge McCollister ENGINE *e = NULL; 51919c402afSSimon Glass int ret; 52019c402afSSimon Glass 52119c402afSSimon Glass ret = rsa_init(); 52219c402afSSimon Glass if (ret) 52319c402afSSimon Glass return ret; 52419c402afSSimon Glass 525f1ca1fdeSGeorge McCollister if (info->engine_id) { 526f1ca1fdeSGeorge McCollister ret = rsa_engine_init(info->engine_id, &e); 527f1ca1fdeSGeorge McCollister if (ret) 528f1ca1fdeSGeorge McCollister goto err_engine; 529f1ca1fdeSGeorge McCollister } 530f1ca1fdeSGeorge McCollister 531f1ca1fdeSGeorge McCollister ret = rsa_get_priv_key(info->keydir, info->keyname, e, &rsa); 53219c402afSSimon Glass if (ret) 53319c402afSSimon Glass goto err_priv; 534219050bfSPhilippe Reynes ret = rsa_sign_with_key(rsa, info->padding, info->checksum, region, 535646257d1SHeiko Schocher region_count, sigp, sig_len); 53619c402afSSimon Glass if (ret) 53719c402afSSimon Glass goto err_sign; 53819c402afSSimon Glass 53919c402afSSimon Glass RSA_free(rsa); 540f1ca1fdeSGeorge McCollister if (info->engine_id) 541f1ca1fdeSGeorge McCollister rsa_engine_remove(e); 54219c402afSSimon Glass rsa_remove(); 54319c402afSSimon Glass 54419c402afSSimon Glass return ret; 54519c402afSSimon Glass 54619c402afSSimon Glass err_sign: 54719c402afSSimon Glass RSA_free(rsa); 54819c402afSSimon Glass err_priv: 549f1ca1fdeSGeorge McCollister if (info->engine_id) 550f1ca1fdeSGeorge McCollister rsa_engine_remove(e); 551f1ca1fdeSGeorge McCollister err_engine: 55219c402afSSimon Glass rsa_remove(); 55319c402afSSimon Glass return ret; 55419c402afSSimon Glass } 55519c402afSSimon Glass 55619c402afSSimon Glass /* 557e0f2f155SMichael van der Westhuizen * rsa_get_exponent(): - Get the public exponent from an RSA key 558e0f2f155SMichael van der Westhuizen */ 559e0f2f155SMichael van der Westhuizen static int rsa_get_exponent(RSA *key, uint64_t *e) 560e0f2f155SMichael van der Westhuizen { 561e0f2f155SMichael van der Westhuizen int ret; 562e0f2f155SMichael van der Westhuizen BIGNUM *bn_te; 563c3b43281SJelle van der Waa const BIGNUM *key_e; 564e0f2f155SMichael van der Westhuizen uint64_t te; 565e0f2f155SMichael van der Westhuizen 566e0f2f155SMichael van der Westhuizen ret = -EINVAL; 567e0f2f155SMichael van der Westhuizen bn_te = NULL; 568e0f2f155SMichael van der Westhuizen 569e0f2f155SMichael van der Westhuizen if (!e) 570e0f2f155SMichael van der Westhuizen goto cleanup; 571e0f2f155SMichael van der Westhuizen 572c3b43281SJelle van der Waa RSA_get0_key(key, NULL, &key_e, NULL); 573c3b43281SJelle van der Waa if (BN_num_bits(key_e) > 64) 574e0f2f155SMichael van der Westhuizen goto cleanup; 575e0f2f155SMichael van der Westhuizen 576c3b43281SJelle van der Waa *e = BN_get_word(key_e); 577e0f2f155SMichael van der Westhuizen 578c3b43281SJelle van der Waa if (BN_num_bits(key_e) < 33) { 579e0f2f155SMichael van der Westhuizen ret = 0; 580e0f2f155SMichael van der Westhuizen goto cleanup; 581e0f2f155SMichael van der Westhuizen } 582e0f2f155SMichael van der Westhuizen 583c3b43281SJelle van der Waa bn_te = BN_dup(key_e); 584e0f2f155SMichael van der Westhuizen if (!bn_te) 585e0f2f155SMichael van der Westhuizen goto cleanup; 586e0f2f155SMichael van der Westhuizen 587e0f2f155SMichael van der Westhuizen if (!BN_rshift(bn_te, bn_te, 32)) 588e0f2f155SMichael van der Westhuizen goto cleanup; 589e0f2f155SMichael van der Westhuizen 590e0f2f155SMichael van der Westhuizen if (!BN_mask_bits(bn_te, 32)) 591e0f2f155SMichael van der Westhuizen goto cleanup; 592e0f2f155SMichael van der Westhuizen 593e0f2f155SMichael van der Westhuizen te = BN_get_word(bn_te); 594e0f2f155SMichael van der Westhuizen te <<= 32; 595e0f2f155SMichael van der Westhuizen *e |= te; 596e0f2f155SMichael van der Westhuizen ret = 0; 597e0f2f155SMichael van der Westhuizen 598e0f2f155SMichael van der Westhuizen cleanup: 599e0f2f155SMichael van der Westhuizen if (bn_te) 600e0f2f155SMichael van der Westhuizen BN_free(bn_te); 601e0f2f155SMichael van der Westhuizen 602e0f2f155SMichael van der Westhuizen return ret; 603e0f2f155SMichael van der Westhuizen } 604e0f2f155SMichael van der Westhuizen 605e0f2f155SMichael van der Westhuizen /* 60619c402afSSimon Glass * rsa_get_params(): - Get the important parameters of an RSA public key 60719c402afSSimon Glass */ 608e0f2f155SMichael van der Westhuizen int rsa_get_params(RSA *key, uint64_t *exponent, uint32_t *n0_invp, 609008ec9b4SJoseph Chen BIGNUM **modulusp, BIGNUM **exponent_BN, BIGNUM **r_squaredp, 610008ec9b4SJoseph Chen BIGNUM **c_factorp, BIGNUM **np_factorp) 61119c402afSSimon Glass { 612008ec9b4SJoseph Chen BIGNUM *big1, *big2, *big32, *big2_32, *big4100, *big2180; 613008ec9b4SJoseph Chen BIGNUM *n, *e, *r, *r_squared, *tmp, *c_factor, *np_factor; 614008ec9b4SJoseph Chen const BIGNUM *key_n, *key_e; 61519c402afSSimon Glass BN_CTX *bn_ctx = BN_CTX_new(); 61619c402afSSimon Glass int ret = 0; 61719c402afSSimon Glass 61819c402afSSimon Glass /* Initialize BIGNUMs */ 61919c402afSSimon Glass big1 = BN_new(); 62019c402afSSimon Glass big2 = BN_new(); 62119c402afSSimon Glass big32 = BN_new(); 622008ec9b4SJoseph Chen big4100 = BN_new(); 623008ec9b4SJoseph Chen big2180 = BN_new(); 624008ec9b4SJoseph Chen 62519c402afSSimon Glass r = BN_new(); 62619c402afSSimon Glass r_squared = BN_new(); 627008ec9b4SJoseph Chen c_factor = BN_new(); 628008ec9b4SJoseph Chen np_factor = BN_new(); 62919c402afSSimon Glass tmp = BN_new(); 63019c402afSSimon Glass big2_32 = BN_new(); 63119c402afSSimon Glass n = BN_new(); 632008ec9b4SJoseph Chen e = BN_new(); 633008ec9b4SJoseph Chen if (!big1 || !big2 || !big32 || !big4100 || !big2180 || !r || 634008ec9b4SJoseph Chen !r_squared || !tmp || !big2_32 || !n || !e || 635008ec9b4SJoseph Chen !c_factor || !np_factor) { 63619c402afSSimon Glass fprintf(stderr, "Out of memory (bignum)\n"); 63719c402afSSimon Glass return -ENOMEM; 63819c402afSSimon Glass } 63919c402afSSimon Glass 640e0f2f155SMichael van der Westhuizen if (0 != rsa_get_exponent(key, exponent)) 641e0f2f155SMichael van der Westhuizen ret = -1; 642e0f2f155SMichael van der Westhuizen 643008ec9b4SJoseph Chen RSA_get0_key(key, &key_n, &key_e, NULL); 644008ec9b4SJoseph Chen if (!BN_copy(n, key_n) || !BN_copy(e, key_e) || 645008ec9b4SJoseph Chen !BN_set_word(big1, 1L) || 646008ec9b4SJoseph Chen !BN_set_word(big2, 2L) || !BN_set_word(big32, 32L) || 647008ec9b4SJoseph Chen !BN_set_word(big4100, 4100L) || !BN_set_word(big2180, 2180L)) 64819c402afSSimon Glass ret = -1; 64919c402afSSimon Glass 65019c402afSSimon Glass /* big2_32 = 2^32 */ 65119c402afSSimon Glass if (!BN_exp(big2_32, big2, big32, bn_ctx)) 65219c402afSSimon Glass ret = -1; 65319c402afSSimon Glass 65419c402afSSimon Glass /* Calculate n0_inv = -1 / n[0] mod 2^32 */ 65519c402afSSimon Glass if (!BN_mod_inverse(tmp, n, big2_32, bn_ctx) || 65619c402afSSimon Glass !BN_sub(tmp, big2_32, tmp)) 65719c402afSSimon Glass ret = -1; 65819c402afSSimon Glass *n0_invp = BN_get_word(tmp); 65919c402afSSimon Glass 66019c402afSSimon Glass /* Calculate R = 2^(# of key bits) */ 66119c402afSSimon Glass if (!BN_set_word(tmp, BN_num_bits(n)) || 66219c402afSSimon Glass !BN_exp(r, big2, tmp, bn_ctx)) 66319c402afSSimon Glass ret = -1; 66419c402afSSimon Glass 66519c402afSSimon Glass /* Calculate r_squared = R^2 mod n */ 66619c402afSSimon Glass if (!BN_copy(r_squared, r) || 66719c402afSSimon Glass !BN_mul(tmp, r_squared, r, bn_ctx) || 66819c402afSSimon Glass !BN_mod(r_squared, tmp, n, bn_ctx)) 66919c402afSSimon Glass ret = -1; 67019c402afSSimon Glass 671008ec9b4SJoseph Chen /* Calculate c_factor = 2^4100 mod n */ 672008ec9b4SJoseph Chen if (!BN_exp(tmp, big2, big4100, bn_ctx) || 673008ec9b4SJoseph Chen !BN_mod(c_factor, tmp, n, bn_ctx)) 674008ec9b4SJoseph Chen ret = -1; 675008ec9b4SJoseph Chen 676008ec9b4SJoseph Chen /* Calculate np_factor = 2^2180 div n */ 677008ec9b4SJoseph Chen if (!BN_exp(tmp, big2, big2180, bn_ctx) || 678008ec9b4SJoseph Chen !BN_div(np_factor, NULL, tmp, n, bn_ctx)) 679008ec9b4SJoseph Chen ret = -1; 680008ec9b4SJoseph Chen 68119c402afSSimon Glass *modulusp = n; 682008ec9b4SJoseph Chen *exponent_BN = e; 68319c402afSSimon Glass *r_squaredp = r_squared; 684008ec9b4SJoseph Chen *c_factorp = c_factor; 685008ec9b4SJoseph Chen *np_factorp = np_factor; 68619c402afSSimon Glass 68719c402afSSimon Glass BN_free(big1); 68819c402afSSimon Glass BN_free(big2); 68919c402afSSimon Glass BN_free(big32); 690008ec9b4SJoseph Chen BN_free(big4100); 691008ec9b4SJoseph Chen BN_free(big2180); 69219c402afSSimon Glass BN_free(r); 69319c402afSSimon Glass BN_free(tmp); 69419c402afSSimon Glass BN_free(big2_32); 69519c402afSSimon Glass if (ret) { 69619c402afSSimon Glass fprintf(stderr, "Bignum operations failed\n"); 69719c402afSSimon Glass return -ENOMEM; 69819c402afSSimon Glass } 69919c402afSSimon Glass 70019c402afSSimon Glass return ret; 70119c402afSSimon Glass } 70219c402afSSimon Glass 703*b6ea0cb4SJoseph Chen static void rsa_convert_big_endian(uint32_t *dst, const uint32_t *src, int len) 704*b6ea0cb4SJoseph Chen { 705*b6ea0cb4SJoseph Chen int i; 706*b6ea0cb4SJoseph Chen 707*b6ea0cb4SJoseph Chen for (i = 0; i < len; i++) 708*b6ea0cb4SJoseph Chen dst[i] = fdt32_to_cpu(src[len - 1 - i]); 709*b6ea0cb4SJoseph Chen } 710*b6ea0cb4SJoseph Chen 711*b6ea0cb4SJoseph Chen static int rsa_set_key_hash(void *keydest, int key_node, 712*b6ea0cb4SJoseph Chen int key_len, const char *csum_algo) 713*b6ea0cb4SJoseph Chen { 714*b6ea0cb4SJoseph Chen const void *rsa_n, *rsa_e, *rsa_c, *rsa_np; 715*b6ea0cb4SJoseph Chen void *n, *e, *c, *np; 716*b6ea0cb4SJoseph Chen uint8_t value[FIT_MAX_HASH_LEN]; 717*b6ea0cb4SJoseph Chen char hash_c[] = "hash@c"; 718*b6ea0cb4SJoseph Chen char hash_np[] = "hash@np"; 719*b6ea0cb4SJoseph Chen char *rsa_key; 720*b6ea0cb4SJoseph Chen int key_word; 721*b6ea0cb4SJoseph Chen int hash_node; 722*b6ea0cb4SJoseph Chen int value_len; 723*b6ea0cb4SJoseph Chen int ret = -ENOSPC; 724*b6ea0cb4SJoseph Chen 725*b6ea0cb4SJoseph Chen rsa_key = malloc(key_len * 3); 726*b6ea0cb4SJoseph Chen if (!rsa_key) 727*b6ea0cb4SJoseph Chen return -ENOSPC; 728*b6ea0cb4SJoseph Chen 729*b6ea0cb4SJoseph Chen rsa_n = fdt_getprop(keydest, key_node, "rsa,modulus", NULL); 730*b6ea0cb4SJoseph Chen rsa_e = fdt_getprop(keydest, key_node, "rsa,exponent-BN", NULL); 731*b6ea0cb4SJoseph Chen rsa_c = fdt_getprop(keydest, key_node, "rsa,c", NULL); 732*b6ea0cb4SJoseph Chen rsa_np = fdt_getprop(keydest, key_node, "rsa,np", NULL); 733*b6ea0cb4SJoseph Chen if (!rsa_c || !rsa_np || !rsa_n || !rsa_e) 734*b6ea0cb4SJoseph Chen goto err_nospc; 735*b6ea0cb4SJoseph Chen 736*b6ea0cb4SJoseph Chen n = rsa_key; 737*b6ea0cb4SJoseph Chen e = rsa_key + key_len; 738*b6ea0cb4SJoseph Chen key_word = key_len / sizeof(uint32_t); 739*b6ea0cb4SJoseph Chen rsa_convert_big_endian(n, rsa_n, key_word); 740*b6ea0cb4SJoseph Chen rsa_convert_big_endian(e, rsa_e, key_word); 741*b6ea0cb4SJoseph Chen 742*b6ea0cb4SJoseph Chen /* hash@c node: n, e, c */ 743*b6ea0cb4SJoseph Chen c = rsa_key + key_len * 2; 744*b6ea0cb4SJoseph Chen rsa_convert_big_endian(c, rsa_c, key_word); 745*b6ea0cb4SJoseph Chen hash_node = fdt_add_subnode(keydest, key_node, hash_c); 746*b6ea0cb4SJoseph Chen if (hash_node < 0) 747*b6ea0cb4SJoseph Chen goto err_nospc; 748*b6ea0cb4SJoseph Chen ret = calculate_hash(rsa_key, key_len * 3, csum_algo, value, &value_len); 749*b6ea0cb4SJoseph Chen if (ret) 750*b6ea0cb4SJoseph Chen goto err_nospc; 751*b6ea0cb4SJoseph Chen ret = fdt_setprop(keydest, hash_node, FIT_VALUE_PROP, value, value_len); 752*b6ea0cb4SJoseph Chen if (ret) 753*b6ea0cb4SJoseph Chen goto err_nospc; 754*b6ea0cb4SJoseph Chen ret = fdt_setprop_string(keydest, hash_node, FIT_ALGO_PROP, csum_algo); 755*b6ea0cb4SJoseph Chen if (ret < 0) 756*b6ea0cb4SJoseph Chen goto err_nospc; 757*b6ea0cb4SJoseph Chen 758*b6ea0cb4SJoseph Chen /* hash@np node: n, e, np */ 759*b6ea0cb4SJoseph Chen np = rsa_key + key_len * 2; 760*b6ea0cb4SJoseph Chen rsa_convert_big_endian(np, rsa_np, key_word); 761*b6ea0cb4SJoseph Chen hash_node = fdt_add_subnode(keydest, key_node, hash_np); 762*b6ea0cb4SJoseph Chen if (hash_node < 0) 763*b6ea0cb4SJoseph Chen goto err_nospc; 764*b6ea0cb4SJoseph Chen 765*b6ea0cb4SJoseph Chen ret = calculate_hash(rsa_key, key_len * 2 + 20, csum_algo, value, &value_len); 766*b6ea0cb4SJoseph Chen if (ret) 767*b6ea0cb4SJoseph Chen goto err_nospc; 768*b6ea0cb4SJoseph Chen ret = fdt_setprop(keydest, hash_node, FIT_VALUE_PROP, value, value_len); 769*b6ea0cb4SJoseph Chen if (ret < 0) 770*b6ea0cb4SJoseph Chen goto err_nospc; 771*b6ea0cb4SJoseph Chen ret = fdt_setprop_string(keydest, hash_node, FIT_ALGO_PROP, csum_algo); 772*b6ea0cb4SJoseph Chen 773*b6ea0cb4SJoseph Chen err_nospc: 774*b6ea0cb4SJoseph Chen if (rsa_key) 775*b6ea0cb4SJoseph Chen free(rsa_key); 776*b6ea0cb4SJoseph Chen 777*b6ea0cb4SJoseph Chen return ret ? -ENOSPC : 0; 778*b6ea0cb4SJoseph Chen } 779*b6ea0cb4SJoseph Chen 78019c402afSSimon Glass static int fdt_add_bignum(void *blob, int noffset, const char *prop_name, 78119c402afSSimon Glass BIGNUM *num, int num_bits) 78219c402afSSimon Glass { 78319c402afSSimon Glass int nwords = num_bits / 32; 78419c402afSSimon Glass int size; 78519c402afSSimon Glass uint32_t *buf, *ptr; 78619c402afSSimon Glass BIGNUM *tmp, *big2, *big32, *big2_32; 78719c402afSSimon Glass BN_CTX *ctx; 78819c402afSSimon Glass int ret; 78919c402afSSimon Glass 79019c402afSSimon Glass tmp = BN_new(); 79119c402afSSimon Glass big2 = BN_new(); 79219c402afSSimon Glass big32 = BN_new(); 79319c402afSSimon Glass big2_32 = BN_new(); 79419c402afSSimon Glass if (!tmp || !big2 || !big32 || !big2_32) { 79519c402afSSimon Glass fprintf(stderr, "Out of memory (bignum)\n"); 79619c402afSSimon Glass return -ENOMEM; 79719c402afSSimon Glass } 79819c402afSSimon Glass ctx = BN_CTX_new(); 79919c402afSSimon Glass if (!tmp) { 80019c402afSSimon Glass fprintf(stderr, "Out of memory (bignum context)\n"); 80119c402afSSimon Glass return -ENOMEM; 80219c402afSSimon Glass } 80319c402afSSimon Glass BN_set_word(big2, 2L); 80419c402afSSimon Glass BN_set_word(big32, 32L); 80519c402afSSimon Glass BN_exp(big2_32, big2, big32, ctx); /* B = 2^32 */ 80619c402afSSimon Glass 80719c402afSSimon Glass size = nwords * sizeof(uint32_t); 80819c402afSSimon Glass buf = malloc(size); 80919c402afSSimon Glass if (!buf) { 81019c402afSSimon Glass fprintf(stderr, "Out of memory (%d bytes)\n", size); 81119c402afSSimon Glass return -ENOMEM; 81219c402afSSimon Glass } 81319c402afSSimon Glass 81419c402afSSimon Glass /* Write out modulus as big endian array of integers */ 81519c402afSSimon Glass for (ptr = buf + nwords - 1; ptr >= buf; ptr--) { 81619c402afSSimon Glass BN_mod(tmp, num, big2_32, ctx); /* n = N mod B */ 81719c402afSSimon Glass *ptr = cpu_to_fdt32(BN_get_word(tmp)); 81819c402afSSimon Glass BN_rshift(num, num, 32); /* N = N/B */ 81919c402afSSimon Glass } 82019c402afSSimon Glass 821713fb2dcSmario.six@gdsys.cc /* 822713fb2dcSmario.six@gdsys.cc * We try signing with successively increasing size values, so this 823713fb2dcSmario.six@gdsys.cc * might fail several times 824713fb2dcSmario.six@gdsys.cc */ 82519c402afSSimon Glass ret = fdt_setprop(blob, noffset, prop_name, buf, size); 8262b9ec762Smario.six@gdsys.cc if (ret) 8272b9ec762Smario.six@gdsys.cc return -FDT_ERR_NOSPACE; 82819c402afSSimon Glass free(buf); 82919c402afSSimon Glass BN_free(tmp); 83019c402afSSimon Glass BN_free(big2); 83119c402afSSimon Glass BN_free(big32); 83219c402afSSimon Glass BN_free(big2_32); 83319c402afSSimon Glass 83419c402afSSimon Glass return ret; 83519c402afSSimon Glass } 83619c402afSSimon Glass 83719c402afSSimon Glass int rsa_add_verify_data(struct image_sign_info *info, void *keydest) 83819c402afSSimon Glass { 839008ec9b4SJoseph Chen BIGNUM *modulus, *exponent_BN, *r_squared, *c_factor, *np_factor; 840e0f2f155SMichael van der Westhuizen uint64_t exponent; 84119c402afSSimon Glass uint32_t n0_inv; 84219c402afSSimon Glass int parent, node; 84319c402afSSimon Glass char name[100]; 84419c402afSSimon Glass int ret; 84519c402afSSimon Glass int bits; 84619c402afSSimon Glass RSA *rsa; 847f1ca1fdeSGeorge McCollister ENGINE *e = NULL; 84819c402afSSimon Glass 84919c402afSSimon Glass debug("%s: Getting verification data\n", __func__); 850f1ca1fdeSGeorge McCollister if (info->engine_id) { 851f1ca1fdeSGeorge McCollister ret = rsa_engine_init(info->engine_id, &e); 85219c402afSSimon Glass if (ret) 85319c402afSSimon Glass return ret; 854f1ca1fdeSGeorge McCollister } 855f1ca1fdeSGeorge McCollister ret = rsa_get_pub_key(info->keydir, info->keyname, e, &rsa); 856f1ca1fdeSGeorge McCollister if (ret) 857f1ca1fdeSGeorge McCollister goto err_get_pub_key; 858008ec9b4SJoseph Chen ret = rsa_get_params(rsa, &exponent, &n0_inv, &modulus, 859008ec9b4SJoseph Chen &exponent_BN, &r_squared, &c_factor, &np_factor); 86019c402afSSimon Glass if (ret) 861f1ca1fdeSGeorge McCollister goto err_get_params; 86219c402afSSimon Glass bits = BN_num_bits(modulus); 86319c402afSSimon Glass parent = fdt_subnode_offset(keydest, 0, FIT_SIG_NODENAME); 86419c402afSSimon Glass if (parent == -FDT_ERR_NOTFOUND) { 86519c402afSSimon Glass parent = fdt_add_subnode(keydest, 0, FIT_SIG_NODENAME); 86619c402afSSimon Glass if (parent < 0) { 867597a8b2cSSimon Glass ret = parent; 868597a8b2cSSimon Glass if (ret != -FDT_ERR_NOSPACE) { 86919c402afSSimon Glass fprintf(stderr, "Couldn't create signature node: %s\n", 87019c402afSSimon Glass fdt_strerror(parent)); 87119c402afSSimon Glass } 87219c402afSSimon Glass } 873597a8b2cSSimon Glass } 874597a8b2cSSimon Glass if (ret) 875597a8b2cSSimon Glass goto done; 87619c402afSSimon Glass 87719c402afSSimon Glass /* Either create or overwrite the named key node */ 87819c402afSSimon Glass snprintf(name, sizeof(name), "key-%s", info->keyname); 87919c402afSSimon Glass node = fdt_subnode_offset(keydest, parent, name); 88019c402afSSimon Glass if (node == -FDT_ERR_NOTFOUND) { 88119c402afSSimon Glass node = fdt_add_subnode(keydest, parent, name); 88219c402afSSimon Glass if (node < 0) { 883597a8b2cSSimon Glass ret = node; 884597a8b2cSSimon Glass if (ret != -FDT_ERR_NOSPACE) { 88519c402afSSimon Glass fprintf(stderr, "Could not create key subnode: %s\n", 88619c402afSSimon Glass fdt_strerror(node)); 887597a8b2cSSimon Glass } 88819c402afSSimon Glass } 88919c402afSSimon Glass } else if (node < 0) { 89019c402afSSimon Glass fprintf(stderr, "Cannot select keys parent: %s\n", 89119c402afSSimon Glass fdt_strerror(node)); 892597a8b2cSSimon Glass ret = node; 89319c402afSSimon Glass } 89419c402afSSimon Glass 895597a8b2cSSimon Glass if (!ret) { 89619c402afSSimon Glass ret = fdt_setprop_string(keydest, node, "key-name-hint", 89719c402afSSimon Glass info->keyname); 898597a8b2cSSimon Glass } 8994f427a42SSimon Glass if (!ret) 9004f427a42SSimon Glass ret = fdt_setprop_u32(keydest, node, "rsa,num-bits", bits); 9014f427a42SSimon Glass if (!ret) 9024f427a42SSimon Glass ret = fdt_setprop_u32(keydest, node, "rsa,n0-inverse", n0_inv); 9034f427a42SSimon Glass if (!ret) { 904e0f2f155SMichael van der Westhuizen ret = fdt_setprop_u64(keydest, node, "rsa,exponent", exponent); 905e0f2f155SMichael van der Westhuizen } 906e0f2f155SMichael van der Westhuizen if (!ret) { 907008ec9b4SJoseph Chen ret = fdt_add_bignum(keydest, node, "rsa,exponent-BN", 908008ec9b4SJoseph Chen exponent_BN, bits); 909008ec9b4SJoseph Chen } 910008ec9b4SJoseph Chen if (!ret) { 9114f427a42SSimon Glass ret = fdt_add_bignum(keydest, node, "rsa,modulus", modulus, 9124f427a42SSimon Glass bits); 9134f427a42SSimon Glass } 9144f427a42SSimon Glass if (!ret) { 9154f427a42SSimon Glass ret = fdt_add_bignum(keydest, node, "rsa,r-squared", r_squared, 9164f427a42SSimon Glass bits); 9174f427a42SSimon Glass } 9184f427a42SSimon Glass if (!ret) { 919008ec9b4SJoseph Chen ret = fdt_add_bignum(keydest, node, "rsa,c", c_factor, 920008ec9b4SJoseph Chen bits); 921008ec9b4SJoseph Chen } 922008ec9b4SJoseph Chen if (!ret) { 923008ec9b4SJoseph Chen ret = fdt_add_bignum(keydest, node, "rsa,np", np_factor, 924008ec9b4SJoseph Chen bits); 925008ec9b4SJoseph Chen } 926008ec9b4SJoseph Chen if (!ret) { 9274f427a42SSimon Glass ret = fdt_setprop_string(keydest, node, FIT_ALGO_PROP, 92883dd98e0SAndrew Duda info->name); 9294f427a42SSimon Glass } 9302b9ec762Smario.six@gdsys.cc if (!ret && info->require_keys) { 9314f427a42SSimon Glass ret = fdt_setprop_string(keydest, node, "required", 93219c402afSSimon Glass info->require_keys); 93319c402afSSimon Glass } 934*b6ea0cb4SJoseph Chen if (!ret) { 935*b6ea0cb4SJoseph Chen ret = rsa_set_key_hash(keydest, node, info->crypto->key_len, 936*b6ea0cb4SJoseph Chen info->checksum->name); 937*b6ea0cb4SJoseph Chen } 938597a8b2cSSimon Glass done: 93919c402afSSimon Glass BN_free(modulus); 94019c402afSSimon Glass BN_free(r_squared); 94119c402afSSimon Glass if (ret) 942f1ca1fdeSGeorge McCollister ret = ret == -FDT_ERR_NOSPACE ? -ENOSPC : -EIO; 943f1ca1fdeSGeorge McCollister err_get_params: 944f1ca1fdeSGeorge McCollister RSA_free(rsa); 945f1ca1fdeSGeorge McCollister err_get_pub_key: 946f1ca1fdeSGeorge McCollister if (info->engine_id) 947f1ca1fdeSGeorge McCollister rsa_engine_remove(e); 94819c402afSSimon Glass 949f1ca1fdeSGeorge McCollister return ret; 95019c402afSSimon Glass } 951