【OP-TEE】 TA的签名与验签 基于optee 3.11

TA的签名

以optee-os 3.11版本为例。在optee_os目录下，存放着签名的私钥和签名脚本。
工程目录/optee_os/keys/default_ta.pem
工程目录/optee_os/scripts/sign_encrypt.py
编译TA时会先将TA编译为elf文件。此时执行签名脚本，对elf文件签名并生成.ta文件。
签名使用了default_ta.pem私钥，此私钥通过编译时makefile调用python脚本时传入。
此脚本还会放置头部数据，shdr，放在TA镜像的头部。这是与签名信息相关的数据，其为shdr类型的结构体。

头部放置了magic number，需要和optee-os代码里的magic number匹配。其他的就是签名相关的数据。
magic：需要与optee-os里定义的magic匹配。
img_type：TA镜像类型，是否加密。
img_siez：TA镜像大小。
alg：签名算法，默认RSA。
hash_size:：TA签名的摘要大小。此摘要是除去sig和hash后对整个ta文件的摘要
sig_size：TA签名的大小。签名是对前面摘要的签名。

拿optee_example里的aes的TA举个例子：
编译出的elf头部如下：

签名后生成的ta文件

红色框里的就是头部shdr的信息，对应为
4字节的magic：0x4f545348；
4字节的img_tpye：0x01；
4字节的img_size：0x0139a8；
4字节的alg：0x70414930，对应optee定义的TEE_ALG_RSASSA_PKCS1_PSS_MGF1_SHA256宏；
2字节的hash_size：0x20；
2字节的sig_size：0x100。
在这之后的0x20字节的内容就是digest，在之后的0x100字节就是签名的内容。

此图片就是shdr的内存布局。

TA的验签

optee验签TA时使用的公钥哪儿来的？
编译时调用的pem_to_pub_c.py脚本，从makefile给出的密钥路径default_ta.pem中解析出rsa_pub公钥，生成ta_pub_key.c文件，放到了ta_pub_key.c文件的ta_pub_key_modulus数组中。所以ta_pub_key_modulus就是验签TA用到的公钥。

如何验签？
放置公钥的数组在optee-os加载TA时shdr_verify_signature函数会从此数组里获取公钥用于验签TA。验签时取出镜像头部的shdr，通过rpc调用到ree侧的tee_supplicant，此时会将TA的镜像加载到REE与TEE的共享内存中，因为共享内存是非安全的内存，所以此时不能加载全部TA，所以先加载shdr头部到安全内存中来，然后调用shdr_verify_signature读取公钥和shdr的签名信息验签。

/* 存放公钥的数组 */
const unit8_t ta_pub_key_modulus[];

/* Validate header signature */
res = shdr_verify_signature(shdr);

校验流程：
先比较magic

if (shdr->magic != SHDR_MAGIC)
	return TEE_ERROR_SECURITY;

判断alg是否是RSA

if (TEE_ALG_GET_MAIN_ALG(shdr->algo) != TEE_MAIN_ALGO_RSA)
	return TEE_ERROR_SECURITY;

从生成的ta_pub_key_modulus数组里取出公钥

res = crypto_bignum_bin2bn(ta_pub_key_modulus, ta_pub_key_modulus_size, key.n);

使用RSA算法验签

res = crypto_acipher_rsassa_verify(shdr->algo, &key, shdr->hash_size,
					SHDR_GET_HASH(shdr), shdr->hash_size,
					SHDR_GET_SIG(shdr), shdr->sig_size);

验签成功后会校验hash。TA是分段加载的，在加载TA的时候会调用ree_fs_ta_read。每加载一次就update一次hash_ctx，

/*更新hash_ctx，以便后面计算出TA的hash值使用*/
res = crypto_hash_update(handle->hash_ctx, dst, len);

/*
		 * Last read: time to check if our digest matches the expected
		 * one (from the signed header)
		 */
		res = check_digest(handle);
		if (res != TEE_SUCCESS)
			return res;

加载完所有TA的段后校验hash,与shdr里的hash比较

static TEE_Result check_digest(struct ree_fs_ta_handle *h)
{
	void *digest = NULL;
	TEE_Result res;

	digest = malloc(h->shdr->hash_size);
	if (!digest)
		return TEE_ERROR_OUT_OF_MEMORY;
	/*加载完成后进行最后计算hash值*/
	res = crypto_hash_final(h->hash_ctx, digest, h->shdr->hash_size);
	if (res != TEE_SUCCESS) {
		res = TEE_ERROR_SECURITY;
		goto out;
	}
	/*与shdr头里存放的hash比较*/
	if (memcmp(digest, SHDR_GET_HASH(h->shdr), h->shdr->hash_size))
		res = TEE_ERROR_SECURITY;
out:
	free(digest);
	return res;
}