lib/ecdsa/ecdsa-libcrypto.c - filogic/uboot - Gitiles

 // SPDX-License-Identifier: GPL-2.0+
 /*
  * ECDSA image signing implementation using libcrypto backend
  *
  * The signature is a binary representation of the (R, S) points, padded to the
  * key size. The signature will be (2 * key_size_bits) / 8 bytes.
  *
  * Deviations from behavior of RSA equivalent:
  *  - Verification uses private key. This is not technically required, but a
  *    limitation on how clumsy the openssl API is to use.
  *  - Handling of keys and key paths:
  *    - The '-K' key directory option must contain path to the key file,
  *      instead of the key directory.
  *    - No assumptions are made about the file extension of the key
  *    - The 'key-name-hint' property is only used for naming devicetree nodes,
  *      but is not used for looking up keys on the filesystem.
  *
  * Copyright (c) 2020,2021, Alexandru Gagniuc <mr.nuke.me@gmail.com>
  */

 #define OPENSSL_API_COMPAT 0x10101000L

 #include <u-boot/ecdsa.h>
 #include <u-boot/fdt-libcrypto.h>
 #include <openssl/ssl.h>
 #include <openssl/ec.h>
 #include <openssl/bn.h>

 /* Image signing context for openssl-libcrypto */
 struct signer {
 	EVP_PKEY *evp_key;	/* Pointer to EVP_PKEY object */
 	EC_KEY *ecdsa_key;	/* Pointer to EC_KEY object */
 	void *hash;		/* Pointer to hash used for verification */
 	void *signature;	/* Pointer to output signature. Do not free()!*/
 };

 static int alloc_ctx(struct signer *ctx, const struct image_sign_info *info)
 {
 	memset(ctx, 0, sizeof(*ctx));

 	if (!OPENSSL_init_ssl(0, NULL)) {
 		fprintf(stderr, "Failure to init SSL library\n");
 		return -1;
 	}

 	ctx->hash = malloc(info->checksum->checksum_len);
 	ctx->signature = malloc(info->crypto->key_len * 2);

 	if (!ctx->hash || !ctx->signature)
 		return -ENOMEM;

 	return 0;
 }

 static void free_ctx(struct signer *ctx)
 {
 	if (ctx->ecdsa_key)
 		EC_KEY_free(ctx->ecdsa_key);

 	if (ctx->evp_key)
 		EVP_PKEY_free(ctx->evp_key);

 	if (ctx->hash)
 		free(ctx->hash);
 }

 /*
  * Convert an ECDSA signature to raw format
  *
  * openssl DER-encodes 'binary' signatures. We want the signature in a raw
  * (R, S) point pair. So we have to dance a bit.
  */
 static void ecdsa_sig_encode_raw(void *buf, const ECDSA_SIG *sig, size_t order)
 {
 	int point_bytes = order;
 	const BIGNUM *r, *s;
 	uintptr_t s_buf;

 	ECDSA_SIG_get0(sig, &r, &s);
 	s_buf = (uintptr_t)buf + point_bytes;
 	BN_bn2binpad(r, buf, point_bytes);
 	BN_bn2binpad(s, (void *)s_buf, point_bytes);
 }

 /* Get a signature from a raw encoding */
 static ECDSA_SIG *ecdsa_sig_from_raw(void *buf, size_t order)
 {
 	int point_bytes = order;
 	uintptr_t s_buf;
 	ECDSA_SIG *sig;
 	BIGNUM *r, *s;

 	sig = ECDSA_SIG_new();
 	if (!sig)
 		return NULL;

 	s_buf = (uintptr_t)buf + point_bytes;
 	r = BN_bin2bn(buf, point_bytes, NULL);
 	s = BN_bin2bn((void *)s_buf, point_bytes, NULL);
 	ECDSA_SIG_set0(sig, r, s);

 	return sig;
 }

 /* ECDSA key size in bytes */
 static size_t ecdsa_key_size_bytes(const EC_KEY *key)
 {
 	const EC_GROUP *group;

 	group = EC_KEY_get0_group(key);
 	return EC_GROUP_order_bits(group) / 8;
 }

 static int read_key(struct signer *ctx, const char *key_name)
 {
 	FILE *f = fopen(key_name, "r");

 	if (!f) {
 		fprintf(stderr, "Can not get key file '%s'\n", key_name);
 		return -ENOENT;
 	}

 	ctx->evp_key = PEM_read_PrivateKey(f, NULL, NULL, NULL);
 	fclose(f);
 	if (!ctx->evp_key) {
 		fprintf(stderr, "Can not read key from '%s'\n", key_name);
 		return -EIO;
 	}

 	if (EVP_PKEY_id(ctx->evp_key) != EVP_PKEY_EC) {
 		fprintf(stderr, "'%s' is not an ECDSA key\n", key_name);
 		return -EINVAL;
 	}

 	ctx->ecdsa_key = EVP_PKEY_get1_EC_KEY(ctx->evp_key);
 	if (!ctx->ecdsa_key)
 		fprintf(stderr, "Can not extract ECDSA key\n");

 	return (ctx->ecdsa_key) ? 0 : -EINVAL;
 }

 /* Prepare a 'signer' context that's ready to sign and verify. */
 static int prepare_ctx(struct signer *ctx, const struct image_sign_info *info)
 {
 	int key_len_bytes, ret;
 	char kname[1024];

 	memset(ctx, 0, sizeof(*ctx));

 	if (info->keyfile) {
 		snprintf(kname,  sizeof(kname), "%s", info->keyfile);
 	} else if (info->keydir && info->keyname) {
 		snprintf(kname, sizeof(kname), "%s/%s.pem", info->keydir,
 			 info->keyname);
 	} else {
 		fprintf(stderr, "keyfile, keyname, or key-name-hint missing\n");
 		return -EINVAL;
 	}

 	ret = alloc_ctx(ctx, info);
 	if (ret)
 		return ret;

 	ret = read_key(ctx, kname);
 	if (ret)
 		return ret;

 	key_len_bytes = ecdsa_key_size_bytes(ctx->ecdsa_key);
 	if (key_len_bytes != info->crypto->key_len) {
 		fprintf(stderr, "Expected a %u-bit key, got %u-bit key\n",
 			info->crypto->key_len * 8, key_len_bytes * 8);
 		return -EINVAL;
 	}

 	return 0;
 }

 static int do_sign(struct signer *ctx, struct image_sign_info *info,
 		   const struct image_region region[], int region_count)
 {
 	const struct checksum_algo *algo = info->checksum;
 	ECDSA_SIG *sig;

 	algo->calculate(algo->name, region, region_count, ctx->hash);
 	sig = ECDSA_do_sign(ctx->hash, algo->checksum_len, ctx->ecdsa_key);

 	ecdsa_sig_encode_raw(ctx->signature, sig, info->crypto->key_len);

 	return 0;
 }

 static int ecdsa_check_signature(struct signer *ctx, struct image_sign_info *info)
 {
 	ECDSA_SIG *sig;
 	int okay;

 	sig = ecdsa_sig_from_raw(ctx->signature, info->crypto->key_len);
 	if (!sig)
 		return -ENOMEM;

 	okay = ECDSA_do_verify(ctx->hash, info->checksum->checksum_len,
 			       sig, ctx->ecdsa_key);
 	if (!okay)
 		fprintf(stderr, "WARNING: Signature is fake news!\n");

 	ECDSA_SIG_free(sig);
 	return !okay;
 }

 static int do_verify(struct signer *ctx, struct image_sign_info *info,
 		     const struct image_region region[], int region_count,
 		     uint8_t *raw_sig, uint sig_len)
 {
 	const struct checksum_algo *algo = info->checksum;

 	if (sig_len != info->crypto->key_len * 2) {
 		fprintf(stderr, "Signature has wrong length\n");
 		return -EINVAL;
 	}

 	memcpy(ctx->signature, raw_sig, sig_len);
 	algo->calculate(algo->name, region, region_count, ctx->hash);

 	return ecdsa_check_signature(ctx, info);
 }

 int ecdsa_sign(struct image_sign_info *info, const struct image_region region[],
 	       int region_count, uint8_t **sigp, uint *sig_len)
 {
 	struct signer ctx;
 	int ret;

 	ret = prepare_ctx(&ctx, info);
 	if (ret >= 0) {
 		do_sign(&ctx, info, region, region_count);
 		*sigp = ctx.signature;
 		*sig_len = info->crypto->key_len * 2;

 		ret = ecdsa_check_signature(&ctx, info);
 	}

 	free_ctx(&ctx);
 	return ret;
 }

 int ecdsa_verify(struct image_sign_info *info,
 		 const struct image_region region[], int region_count,
 		 uint8_t *sig, uint sig_len)
 {
 	struct signer ctx;
 	int ret;

 	ret = prepare_ctx(&ctx, info);
 	if (ret >= 0)
 		ret = do_verify(&ctx, info, region, region_count, sig, sig_len);

 	free_ctx(&ctx);
 	return ret;
 }

 static int do_add(struct signer *ctx, void *fdt, const char *key_node_name)
 {
 	int signature_node, key_node, ret, key_bits;
 	const char *curve_name;
 	const EC_GROUP *group;
 	const EC_POINT *point;
 	BIGNUM *x, *y;

 	signature_node = fdt_subnode_offset(fdt, 0, FIT_SIG_NODENAME);
 	if (signature_node < 0) {
 		fprintf(stderr, "Could not find 'signature node: %s\n",
 			fdt_strerror(signature_node));
 		return signature_node;
 	}

 	key_node = fdt_add_subnode(fdt, signature_node, key_node_name);
 	if (key_node < 0) {
 		fprintf(stderr, "Could not create '%s' node: %s\n",
 			key_node_name, fdt_strerror(key_node));
 		return key_node;
 	}

 	group = EC_KEY_get0_group(ctx->ecdsa_key);
 	key_bits = EC_GROUP_order_bits(group);
 	curve_name = OBJ_nid2sn(EC_GROUP_get_curve_name(group));
 	/* Let 'x' and 'y' memory leak by not BN_free()'ing them. */
 	x = BN_new();
 	y = BN_new();
 	point = EC_KEY_get0_public_key(ctx->ecdsa_key);
 	EC_POINT_get_affine_coordinates(group, point, x, y, NULL);

 	ret = fdt_setprop_string(fdt, key_node, "ecdsa,curve", curve_name);
 	if (ret < 0)
 		return ret;

 	ret = fdt_add_bignum(fdt, key_node, "ecdsa,x-point", x, key_bits);
 	if (ret < 0)
 		return ret;

 	ret = fdt_add_bignum(fdt, key_node, "ecdsa,y-point", y, key_bits);
 	if (ret < 0)
 		return ret;

 	return 0;
 }

 int ecdsa_add_verify_data(struct image_sign_info *info, void *fdt)
 {
 	const char *fdt_key_name;
 	struct signer ctx;
 	int ret;

 	fdt_key_name = info->keyname ? info->keyname : "default-key";
 	ret = prepare_ctx(&ctx, info);
 	if (ret >= 0)
 		do_add(&ctx, fdt, fdt_key_name);

 	free_ctx(&ctx);
 	return ret;
 }
	// SPDX-License-Identifier: GPL-2.0+
	/*
	* ECDSA image signing implementation using libcrypto backend
	*
	* The signature is a binary representation of the (R, S) points, padded to the
	* key size. The signature will be (2 * key_size_bits) / 8 bytes.
	*
	* Deviations from behavior of RSA equivalent:
	* - Verification uses private key. This is not technically required, but a
	* limitation on how clumsy the openssl API is to use.
	* - Handling of keys and key paths:
	* - The '-K' key directory option must contain path to the key file,
	* instead of the key directory.
	* - No assumptions are made about the file extension of the key
	* - The 'key-name-hint' property is only used for naming devicetree nodes,
	* but is not used for looking up keys on the filesystem.
	*
	* Copyright (c) 2020,2021, Alexandru Gagniuc <mr.nuke.me@gmail.com>
	*/

	#define OPENSSL_API_COMPAT 0x10101000L

	#include <u-boot/ecdsa.h>
	#include <u-boot/fdt-libcrypto.h>
	#include <openssl/ssl.h>
	#include <openssl/ec.h>
	#include <openssl/bn.h>

	/* Image signing context for openssl-libcrypto */
	struct signer {
	EVP_PKEY evp_key; / Pointer to EVP_PKEY object */
	EC_KEY ecdsa_key; / Pointer to EC_KEY object */
	void hash; / Pointer to hash used for verification */
	void signature; / Pointer to output signature. Do not free()!*/
	};

	static int alloc_ctx(struct signer ctx, const struct image_sign_info info)
	{
	memset(ctx, 0, sizeof(*ctx));

	if (!OPENSSL_init_ssl(0, NULL)) {
	fprintf(stderr, "Failure to init SSL library\n");
	return -1;
	}

	ctx->hash = malloc(info->checksum->checksum_len);
	ctx->signature = malloc(info->crypto->key_len * 2);

	if (!ctx->hash \|\| !ctx->signature)
	return -ENOMEM;

	return 0;
	}

	static void free_ctx(struct signer *ctx)
	{
	if (ctx->ecdsa_key)
	EC_KEY_free(ctx->ecdsa_key);

	if (ctx->evp_key)
	EVP_PKEY_free(ctx->evp_key);

	if (ctx->hash)
	free(ctx->hash);
	}

	/*
	* Convert an ECDSA signature to raw format
	*
	* openssl DER-encodes 'binary' signatures. We want the signature in a raw
	* (R, S) point pair. So we have to dance a bit.
	*/
	static void ecdsa_sig_encode_raw(void buf, const ECDSA_SIG sig, size_t order)
	{
	int point_bytes = order;
	const BIGNUM r, s;
	uintptr_t s_buf;

	ECDSA_SIG_get0(sig, &r, &s);
	s_buf = (uintptr_t)buf + point_bytes;
	BN_bn2binpad(r, buf, point_bytes);
	BN_bn2binpad(s, (void *)s_buf, point_bytes);
	}

	/* Get a signature from a raw encoding */
	static ECDSA_SIG ecdsa_sig_from_raw(void buf, size_t order)
	{
	int point_bytes = order;
	uintptr_t s_buf;
	ECDSA_SIG *sig;
	BIGNUM r, s;

	sig = ECDSA_SIG_new();
	if (!sig)
	return NULL;

	s_buf = (uintptr_t)buf + point_bytes;
	r = BN_bin2bn(buf, point_bytes, NULL);
	s = BN_bin2bn((void *)s_buf, point_bytes, NULL);
	ECDSA_SIG_set0(sig, r, s);

	return sig;
	}

	/* ECDSA key size in bytes */
	static size_t ecdsa_key_size_bytes(const EC_KEY *key)
	{
	const EC_GROUP *group;

	group = EC_KEY_get0_group(key);
	return EC_GROUP_order_bits(group) / 8;
	}

	static int read_key(struct signer ctx, const char key_name)
	{
	FILE *f = fopen(key_name, "r");

	if (!f) {
	fprintf(stderr, "Can not get key file '%s'\n", key_name);
	return -ENOENT;
	}

	ctx->evp_key = PEM_read_PrivateKey(f, NULL, NULL, NULL);
	fclose(f);
	if (!ctx->evp_key) {
	fprintf(stderr, "Can not read key from '%s'\n", key_name);
	return -EIO;
	}

	if (EVP_PKEY_id(ctx->evp_key) != EVP_PKEY_EC) {
	fprintf(stderr, "'%s' is not an ECDSA key\n", key_name);
	return -EINVAL;
	}

	ctx->ecdsa_key = EVP_PKEY_get1_EC_KEY(ctx->evp_key);
	if (!ctx->ecdsa_key)
	fprintf(stderr, "Can not extract ECDSA key\n");

	return (ctx->ecdsa_key) ? 0 : -EINVAL;
	}

	/* Prepare a 'signer' context that's ready to sign and verify. */
	static int prepare_ctx(struct signer ctx, const struct image_sign_info info)
	{
	int key_len_bytes, ret;
	char kname[1024];

	memset(ctx, 0, sizeof(*ctx));

	if (info->keyfile) {
	snprintf(kname, sizeof(kname), "%s", info->keyfile);
	} else if (info->keydir && info->keyname) {
	snprintf(kname, sizeof(kname), "%s/%s.pem", info->keydir,
	info->keyname);
	} else {
	fprintf(stderr, "keyfile, keyname, or key-name-hint missing\n");
	return -EINVAL;
	}

	ret = alloc_ctx(ctx, info);
	if (ret)
	return ret;

	ret = read_key(ctx, kname);
	if (ret)
	return ret;

	key_len_bytes = ecdsa_key_size_bytes(ctx->ecdsa_key);
	if (key_len_bytes != info->crypto->key_len) {
	fprintf(stderr, "Expected a %u-bit key, got %u-bit key\n",
	info->crypto->key_len * 8, key_len_bytes * 8);
	return -EINVAL;
	}

	return 0;
	}

	static int do_sign(struct signer ctx, struct image_sign_info info,
	const struct image_region region[], int region_count)
	{
	const struct checksum_algo *algo = info->checksum;
	ECDSA_SIG *sig;

	algo->calculate(algo->name, region, region_count, ctx->hash);
	sig = ECDSA_do_sign(ctx->hash, algo->checksum_len, ctx->ecdsa_key);

	ecdsa_sig_encode_raw(ctx->signature, sig, info->crypto->key_len);

	return 0;
	}

	static int ecdsa_check_signature(struct signer ctx, struct image_sign_info info)
	{
	ECDSA_SIG *sig;
	int okay;

	sig = ecdsa_sig_from_raw(ctx->signature, info->crypto->key_len);
	if (!sig)
	return -ENOMEM;

	okay = ECDSA_do_verify(ctx->hash, info->checksum->checksum_len,
	sig, ctx->ecdsa_key);
	if (!okay)
	fprintf(stderr, "WARNING: Signature is fake news!\n");

	ECDSA_SIG_free(sig);
	return !okay;
	}

	static int do_verify(struct signer ctx, struct image_sign_info info,
	const struct image_region region[], int region_count,
	uint8_t *raw_sig, uint sig_len)
	{
	const struct checksum_algo *algo = info->checksum;

	if (sig_len != info->crypto->key_len * 2) {
	fprintf(stderr, "Signature has wrong length\n");
	return -EINVAL;
	}

	memcpy(ctx->signature, raw_sig, sig_len);
	algo->calculate(algo->name, region, region_count, ctx->hash);

	return ecdsa_check_signature(ctx, info);
	}

	int ecdsa_sign(struct image_sign_info *info, const struct image_region region[],
	int region_count, uint8_t *sigp, uint sig_len)
	{
	struct signer ctx;
	int ret;

	ret = prepare_ctx(&ctx, info);
	if (ret >= 0) {
	do_sign(&ctx, info, region, region_count);
	*sigp = ctx.signature;
	sig_len = info->crypto->key_len 2;

	ret = ecdsa_check_signature(&ctx, info);
	}

	free_ctx(&ctx);
	return ret;
	}

	int ecdsa_verify(struct image_sign_info *info,
	const struct image_region region[], int region_count,
	uint8_t *sig, uint sig_len)
	{
	struct signer ctx;
	int ret;

	ret = prepare_ctx(&ctx, info);
	if (ret >= 0)
	ret = do_verify(&ctx, info, region, region_count, sig, sig_len);

	free_ctx(&ctx);
	return ret;
	}

	static int do_add(struct signer ctx, void fdt, const char *key_node_name)
	{
	int signature_node, key_node, ret, key_bits;
	const char *curve_name;
	const EC_GROUP *group;
	const EC_POINT *point;
	BIGNUM x, y;

	signature_node = fdt_subnode_offset(fdt, 0, FIT_SIG_NODENAME);
	if (signature_node < 0) {
	fprintf(stderr, "Could not find 'signature node: %s\n",
	fdt_strerror(signature_node));
	return signature_node;
	}

	key_node = fdt_add_subnode(fdt, signature_node, key_node_name);
	if (key_node < 0) {
	fprintf(stderr, "Could not create '%s' node: %s\n",
	key_node_name, fdt_strerror(key_node));
	return key_node;
	}

	group = EC_KEY_get0_group(ctx->ecdsa_key);
	key_bits = EC_GROUP_order_bits(group);
	curve_name = OBJ_nid2sn(EC_GROUP_get_curve_name(group));
	/* Let 'x' and 'y' memory leak by not BN_free()'ing them. */
	x = BN_new();
	y = BN_new();
	point = EC_KEY_get0_public_key(ctx->ecdsa_key);
	EC_POINT_get_affine_coordinates(group, point, x, y, NULL);

	ret = fdt_setprop_string(fdt, key_node, "ecdsa,curve", curve_name);
	if (ret < 0)
	return ret;

	ret = fdt_add_bignum(fdt, key_node, "ecdsa,x-point", x, key_bits);
	if (ret < 0)
	return ret;

	ret = fdt_add_bignum(fdt, key_node, "ecdsa,y-point", y, key_bits);
	if (ret < 0)
	return ret;

	return 0;
	}

	int ecdsa_add_verify_data(struct image_sign_info info, void fdt)
	{
	const char *fdt_key_name;
	struct signer ctx;
	int ret;

	fdt_key_name = info->keyname ? info->keyname : "default-key";
	ret = prepare_ctx(&ctx, info);
	if (ret >= 0)
	do_add(&ctx, fdt, fdt_key_name);

	free_ctx(&ctx);
	return ret;
	}