forked from Qortal/Brooklyn
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
399 lines
9.8 KiB
399 lines
9.8 KiB
/* Sign a module file using the given key. |
|
* |
|
* Copyright © 2014-2016 Red Hat, Inc. All Rights Reserved. |
|
* Copyright © 2015 Intel Corporation. |
|
* Copyright © 2016 Hewlett Packard Enterprise Development LP |
|
* |
|
* Authors: David Howells <[email protected]> |
|
* David Woodhouse <[email protected]> |
|
* Juerg Haefliger <[email protected]> |
|
* |
|
* This program is free software; you can redistribute it and/or |
|
* modify it under the terms of the GNU Lesser General Public License |
|
* as published by the Free Software Foundation; either version 2.1 |
|
* of the licence, or (at your option) any later version. |
|
*/ |
|
#define _GNU_SOURCE |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include <stdint.h> |
|
#include <stdbool.h> |
|
#include <string.h> |
|
#include <getopt.h> |
|
#include <err.h> |
|
#include <arpa/inet.h> |
|
#include <openssl/opensslv.h> |
|
#include <openssl/bio.h> |
|
#include <openssl/evp.h> |
|
#include <openssl/pem.h> |
|
#include <openssl/err.h> |
|
#include <openssl/engine.h> |
|
|
|
/* |
|
* Use CMS if we have openssl-1.0.0 or newer available - otherwise we have to |
|
* assume that it's not available and its header file is missing and that we |
|
* should use PKCS#7 instead. Switching to the older PKCS#7 format restricts |
|
* the options we have on specifying the X.509 certificate we want. |
|
* |
|
* Further, older versions of OpenSSL don't support manually adding signers to |
|
* the PKCS#7 message so have to accept that we get a certificate included in |
|
* the signature message. Nor do such older versions of OpenSSL support |
|
* signing with anything other than SHA1 - so we're stuck with that if such is |
|
* the case. |
|
*/ |
|
#if defined(LIBRESSL_VERSION_NUMBER) || \ |
|
OPENSSL_VERSION_NUMBER < 0x10000000L || \ |
|
defined(OPENSSL_NO_CMS) |
|
#define USE_PKCS7 |
|
#endif |
|
#ifndef USE_PKCS7 |
|
#include <openssl/cms.h> |
|
#else |
|
#include <openssl/pkcs7.h> |
|
#endif |
|
|
|
struct module_signature { |
|
uint8_t algo; /* Public-key crypto algorithm [0] */ |
|
uint8_t hash; /* Digest algorithm [0] */ |
|
uint8_t id_type; /* Key identifier type [PKEY_ID_PKCS7] */ |
|
uint8_t signer_len; /* Length of signer's name [0] */ |
|
uint8_t key_id_len; /* Length of key identifier [0] */ |
|
uint8_t __pad[3]; |
|
uint32_t sig_len; /* Length of signature data */ |
|
}; |
|
|
|
#define PKEY_ID_PKCS7 2 |
|
|
|
static char magic_number[] = "~Module signature appended~\n"; |
|
|
|
static __attribute__((noreturn)) |
|
void format(void) |
|
{ |
|
fprintf(stderr, |
|
"Usage: scripts/sign-file [-dp] <hash algo> <key> <x509> <module> [<dest>]\n"); |
|
fprintf(stderr, |
|
" scripts/sign-file -s <raw sig> <hash algo> <x509> <module> [<dest>]\n"); |
|
exit(2); |
|
} |
|
|
|
static void display_openssl_errors(int l) |
|
{ |
|
const char *file; |
|
char buf[120]; |
|
int e, line; |
|
|
|
if (ERR_peek_error() == 0) |
|
return; |
|
fprintf(stderr, "At main.c:%d:\n", l); |
|
|
|
while ((e = ERR_get_error_line(&file, &line))) { |
|
ERR_error_string(e, buf); |
|
fprintf(stderr, "- SSL %s: %s:%d\n", buf, file, line); |
|
} |
|
} |
|
|
|
static void drain_openssl_errors(void) |
|
{ |
|
const char *file; |
|
int line; |
|
|
|
if (ERR_peek_error() == 0) |
|
return; |
|
while (ERR_get_error_line(&file, &line)) {} |
|
} |
|
|
|
#define ERR(cond, fmt, ...) \ |
|
do { \ |
|
bool __cond = (cond); \ |
|
display_openssl_errors(__LINE__); \ |
|
if (__cond) { \ |
|
err(1, fmt, ## __VA_ARGS__); \ |
|
} \ |
|
} while(0) |
|
|
|
static const char *key_pass; |
|
|
|
static int pem_pw_cb(char *buf, int len, int w, void *v) |
|
{ |
|
int pwlen; |
|
|
|
if (!key_pass) |
|
return -1; |
|
|
|
pwlen = strlen(key_pass); |
|
if (pwlen >= len) |
|
return -1; |
|
|
|
strcpy(buf, key_pass); |
|
|
|
/* If it's wrong, don't keep trying it. */ |
|
key_pass = NULL; |
|
|
|
return pwlen; |
|
} |
|
|
|
static EVP_PKEY *read_private_key(const char *private_key_name) |
|
{ |
|
EVP_PKEY *private_key; |
|
|
|
if (!strncmp(private_key_name, "pkcs11:", 7)) { |
|
ENGINE *e; |
|
|
|
ENGINE_load_builtin_engines(); |
|
drain_openssl_errors(); |
|
e = ENGINE_by_id("pkcs11"); |
|
ERR(!e, "Load PKCS#11 ENGINE"); |
|
if (ENGINE_init(e)) |
|
drain_openssl_errors(); |
|
else |
|
ERR(1, "ENGINE_init"); |
|
if (key_pass) |
|
ERR(!ENGINE_ctrl_cmd_string(e, "PIN", key_pass, 0), |
|
"Set PKCS#11 PIN"); |
|
private_key = ENGINE_load_private_key(e, private_key_name, |
|
NULL, NULL); |
|
ERR(!private_key, "%s", private_key_name); |
|
} else { |
|
BIO *b; |
|
|
|
b = BIO_new_file(private_key_name, "rb"); |
|
ERR(!b, "%s", private_key_name); |
|
private_key = PEM_read_bio_PrivateKey(b, NULL, pem_pw_cb, |
|
NULL); |
|
ERR(!private_key, "%s", private_key_name); |
|
BIO_free(b); |
|
} |
|
|
|
return private_key; |
|
} |
|
|
|
static X509 *read_x509(const char *x509_name) |
|
{ |
|
unsigned char buf[2]; |
|
X509 *x509; |
|
BIO *b; |
|
int n; |
|
|
|
b = BIO_new_file(x509_name, "rb"); |
|
ERR(!b, "%s", x509_name); |
|
|
|
/* Look at the first two bytes of the file to determine the encoding */ |
|
n = BIO_read(b, buf, 2); |
|
if (n != 2) { |
|
if (BIO_should_retry(b)) { |
|
fprintf(stderr, "%s: Read wanted retry\n", x509_name); |
|
exit(1); |
|
} |
|
if (n >= 0) { |
|
fprintf(stderr, "%s: Short read\n", x509_name); |
|
exit(1); |
|
} |
|
ERR(1, "%s", x509_name); |
|
} |
|
|
|
ERR(BIO_reset(b) != 0, "%s", x509_name); |
|
|
|
if (buf[0] == 0x30 && buf[1] >= 0x81 && buf[1] <= 0x84) |
|
/* Assume raw DER encoded X.509 */ |
|
x509 = d2i_X509_bio(b, NULL); |
|
else |
|
/* Assume PEM encoded X.509 */ |
|
x509 = PEM_read_bio_X509(b, NULL, NULL, NULL); |
|
|
|
BIO_free(b); |
|
ERR(!x509, "%s", x509_name); |
|
|
|
return x509; |
|
} |
|
|
|
int main(int argc, char **argv) |
|
{ |
|
struct module_signature sig_info = { .id_type = PKEY_ID_PKCS7 }; |
|
char *hash_algo = NULL; |
|
char *private_key_name = NULL, *raw_sig_name = NULL; |
|
char *x509_name, *module_name, *dest_name; |
|
bool save_sig = false, replace_orig; |
|
bool sign_only = false; |
|
bool raw_sig = false; |
|
unsigned char buf[4096]; |
|
unsigned long module_size, sig_size; |
|
unsigned int use_signed_attrs; |
|
const EVP_MD *digest_algo; |
|
EVP_PKEY *private_key; |
|
#ifndef USE_PKCS7 |
|
CMS_ContentInfo *cms = NULL; |
|
unsigned int use_keyid = 0; |
|
#else |
|
PKCS7 *pkcs7 = NULL; |
|
#endif |
|
X509 *x509; |
|
BIO *bd, *bm; |
|
int opt, n; |
|
OpenSSL_add_all_algorithms(); |
|
ERR_load_crypto_strings(); |
|
ERR_clear_error(); |
|
|
|
key_pass = getenv("KBUILD_SIGN_PIN"); |
|
|
|
#ifndef USE_PKCS7 |
|
use_signed_attrs = CMS_NOATTR; |
|
#else |
|
use_signed_attrs = PKCS7_NOATTR; |
|
#endif |
|
|
|
do { |
|
opt = getopt(argc, argv, "sdpk"); |
|
switch (opt) { |
|
case 's': raw_sig = true; break; |
|
case 'p': save_sig = true; break; |
|
case 'd': sign_only = true; save_sig = true; break; |
|
#ifndef USE_PKCS7 |
|
case 'k': use_keyid = CMS_USE_KEYID; break; |
|
#endif |
|
case -1: break; |
|
default: format(); |
|
} |
|
} while (opt != -1); |
|
|
|
argc -= optind; |
|
argv += optind; |
|
if (argc < 4 || argc > 5) |
|
format(); |
|
|
|
if (raw_sig) { |
|
raw_sig_name = argv[0]; |
|
hash_algo = argv[1]; |
|
} else { |
|
hash_algo = argv[0]; |
|
private_key_name = argv[1]; |
|
} |
|
x509_name = argv[2]; |
|
module_name = argv[3]; |
|
if (argc == 5 && strcmp(argv[3], argv[4]) != 0) { |
|
dest_name = argv[4]; |
|
replace_orig = false; |
|
} else { |
|
ERR(asprintf(&dest_name, "%s.~signed~", module_name) < 0, |
|
"asprintf"); |
|
replace_orig = true; |
|
} |
|
|
|
#ifdef USE_PKCS7 |
|
if (strcmp(hash_algo, "sha1") != 0) { |
|
fprintf(stderr, "sign-file: %s only supports SHA1 signing\n", |
|
OPENSSL_VERSION_TEXT); |
|
exit(3); |
|
} |
|
#endif |
|
|
|
/* Open the module file */ |
|
bm = BIO_new_file(module_name, "rb"); |
|
ERR(!bm, "%s", module_name); |
|
|
|
if (!raw_sig) { |
|
/* Read the private key and the X.509 cert the PKCS#7 message |
|
* will point to. |
|
*/ |
|
private_key = read_private_key(private_key_name); |
|
x509 = read_x509(x509_name); |
|
|
|
/* Digest the module data. */ |
|
OpenSSL_add_all_digests(); |
|
display_openssl_errors(__LINE__); |
|
digest_algo = EVP_get_digestbyname(hash_algo); |
|
ERR(!digest_algo, "EVP_get_digestbyname"); |
|
|
|
#ifndef USE_PKCS7 |
|
/* Load the signature message from the digest buffer. */ |
|
cms = CMS_sign(NULL, NULL, NULL, NULL, |
|
CMS_NOCERTS | CMS_PARTIAL | CMS_BINARY | |
|
CMS_DETACHED | CMS_STREAM); |
|
ERR(!cms, "CMS_sign"); |
|
|
|
ERR(!CMS_add1_signer(cms, x509, private_key, digest_algo, |
|
CMS_NOCERTS | CMS_BINARY | |
|
CMS_NOSMIMECAP | use_keyid | |
|
use_signed_attrs), |
|
"CMS_add1_signer"); |
|
ERR(CMS_final(cms, bm, NULL, CMS_NOCERTS | CMS_BINARY) < 0, |
|
"CMS_final"); |
|
|
|
#else |
|
pkcs7 = PKCS7_sign(x509, private_key, NULL, bm, |
|
PKCS7_NOCERTS | PKCS7_BINARY | |
|
PKCS7_DETACHED | use_signed_attrs); |
|
ERR(!pkcs7, "PKCS7_sign"); |
|
#endif |
|
|
|
if (save_sig) { |
|
char *sig_file_name; |
|
BIO *b; |
|
|
|
ERR(asprintf(&sig_file_name, "%s.p7s", module_name) < 0, |
|
"asprintf"); |
|
b = BIO_new_file(sig_file_name, "wb"); |
|
ERR(!b, "%s", sig_file_name); |
|
#ifndef USE_PKCS7 |
|
ERR(i2d_CMS_bio_stream(b, cms, NULL, 0) < 0, |
|
"%s", sig_file_name); |
|
#else |
|
ERR(i2d_PKCS7_bio(b, pkcs7) < 0, |
|
"%s", sig_file_name); |
|
#endif |
|
BIO_free(b); |
|
} |
|
|
|
if (sign_only) { |
|
BIO_free(bm); |
|
return 0; |
|
} |
|
} |
|
|
|
/* Open the destination file now so that we can shovel the module data |
|
* across as we read it. |
|
*/ |
|
bd = BIO_new_file(dest_name, "wb"); |
|
ERR(!bd, "%s", dest_name); |
|
|
|
/* Append the marker and the PKCS#7 message to the destination file */ |
|
ERR(BIO_reset(bm) < 0, "%s", module_name); |
|
while ((n = BIO_read(bm, buf, sizeof(buf))), |
|
n > 0) { |
|
ERR(BIO_write(bd, buf, n) < 0, "%s", dest_name); |
|
} |
|
BIO_free(bm); |
|
ERR(n < 0, "%s", module_name); |
|
module_size = BIO_number_written(bd); |
|
|
|
if (!raw_sig) { |
|
#ifndef USE_PKCS7 |
|
ERR(i2d_CMS_bio_stream(bd, cms, NULL, 0) < 0, "%s", dest_name); |
|
#else |
|
ERR(i2d_PKCS7_bio(bd, pkcs7) < 0, "%s", dest_name); |
|
#endif |
|
} else { |
|
BIO *b; |
|
|
|
/* Read the raw signature file and write the data to the |
|
* destination file |
|
*/ |
|
b = BIO_new_file(raw_sig_name, "rb"); |
|
ERR(!b, "%s", raw_sig_name); |
|
while ((n = BIO_read(b, buf, sizeof(buf))), n > 0) |
|
ERR(BIO_write(bd, buf, n) < 0, "%s", dest_name); |
|
BIO_free(b); |
|
} |
|
|
|
sig_size = BIO_number_written(bd) - module_size; |
|
sig_info.sig_len = htonl(sig_size); |
|
ERR(BIO_write(bd, &sig_info, sizeof(sig_info)) < 0, "%s", dest_name); |
|
ERR(BIO_write(bd, magic_number, sizeof(magic_number) - 1) < 0, "%s", dest_name); |
|
|
|
ERR(BIO_free(bd) < 0, "%s", dest_name); |
|
|
|
/* Finally, if we're signing in place, replace the original. */ |
|
if (replace_orig) |
|
ERR(rename(dest_name, module_name) < 0, "%s", dest_name); |
|
|
|
return 0; |
|
}
|
|
|