]> git.kernelconcepts.de Git - karo-tx-linux.git/blobdiff - scripts/sign-file.c
net: revert "net_sched: move tp->root allocation into fw_init()"
[karo-tx-linux.git] / scripts / sign-file.c
index 5b8a6dda3235ab59e1634ca19bfb6a60c66942bd..058bba3103e2707911cd493f70a1b39015f9d697 100755 (executable)
@@ -20,8 +20,9 @@
 #include <openssl/bio.h>
 #include <openssl/evp.h>
 #include <openssl/pem.h>
-#include <openssl/pkcs7.h>
+#include <openssl/cms.h>
 #include <openssl/err.h>
+#include <openssl/engine.h>
 
 struct module_signature {
        uint8_t         algo;           /* Public-key crypto algorithm [0] */
@@ -80,28 +81,56 @@ static void drain_openssl_errors(void)
                }                                       \
        } 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;
+}
+
 int main(int argc, char **argv)
 {
        struct module_signature sig_info = { .id_type = PKEY_ID_PKCS7 };
        char *hash_algo = NULL;
        char *private_key_name, *x509_name, *module_name, *dest_name;
-       bool save_pkcs7 = false, replace_orig;
+       bool save_cms = false, replace_orig;
+       bool sign_only = false;
        unsigned char buf[4096];
-       unsigned long module_size, pkcs7_size;
+       unsigned long module_size, cms_size;
+       unsigned int use_keyid = 0, use_signed_attrs = CMS_NOATTR;
        const EVP_MD *digest_algo;
        EVP_PKEY *private_key;
-       PKCS7 *pkcs7;
+       CMS_ContentInfo *cms;
        X509 *x509;
-       BIO *b, *bd, *bm;
+       BIO *b, *bd = NULL, *bm;
        int opt, n;
 
+       OpenSSL_add_all_algorithms();
        ERR_load_crypto_strings();
        ERR_clear_error();
 
+       key_pass = getenv("KBUILD_SIGN_PIN");
+
        do {
-               opt = getopt(argc, argv, "dp");
+               opt = getopt(argc, argv, "dpk");
                switch (opt) {
-               case 'p': save_pkcs7 = true; break;
+               case 'p': save_cms = true; break;
+               case 'd': sign_only = true; save_cms = true; break;
+               case 'k': use_keyid = CMS_USE_KEYID; break;
                case -1: break;
                default: format();
                }
@@ -128,16 +157,35 @@ int main(int argc, char **argv)
        /* Read the private key and the X.509 cert the PKCS#7 message
         * will point to.
         */
-       b = BIO_new_file(private_key_name, "rb");
-       ERR(!b, "%s", private_key_name);
-        private_key = PEM_read_bio_PrivateKey(b, NULL, NULL, NULL);
-       BIO_free(b);
+       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 {
+               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);
+       }
 
        b = BIO_new_file(x509_name, "rb");
        ERR(!b, "%s", x509_name);
        x509 = d2i_X509_bio(b, NULL); /* Binary encoded X.509 */
        if (!x509) {
-               BIO_reset(b);
+               ERR(BIO_reset(b) != 1, "%s", x509_name);
                x509 = PEM_read_bio_X509(b, NULL, NULL, NULL); /* PEM encoded X.509 */
                if (x509)
                        drain_openssl_errors();
@@ -148,8 +196,10 @@ int main(int argc, char **argv)
        /* 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);
+       if (!sign_only) {
+               bd = BIO_new_file(dest_name, "wb");
+               ERR(!bd, "%s", dest_name);
+       }
 
        /* Digest the module data. */
        OpenSSL_add_all_digests();
@@ -160,26 +210,31 @@ int main(int argc, char **argv)
        bm = BIO_new_file(module_name, "rb");
        ERR(!bm, "%s", module_name);
 
-       /* Load the PKCS#7 message from the digest buffer. */
-       pkcs7 = PKCS7_sign(NULL, NULL, NULL, NULL,
-                          PKCS7_NOCERTS | PKCS7_PARTIAL | PKCS7_BINARY | PKCS7_DETACHED | PKCS7_STREAM);
-       ERR(!pkcs7, "PKCS7_sign");
+       /* Load the CMS 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(!PKCS7_sign_add_signer(pkcs7, x509, private_key, digest_algo, PKCS7_NOCERTS | PKCS7_BINARY),
-           "PKCS7_sign_add_signer");
-       ERR(PKCS7_final(pkcs7, bm, PKCS7_NOCERTS | PKCS7_BINARY) < 0,
-           "PKCS7_final");
+       ERR(!CMS_add1_signer(cms, x509, private_key, digest_algo,
+                            CMS_NOCERTS | CMS_BINARY | CMS_NOSMIMECAP |
+                            use_keyid | use_signed_attrs),
+           "CMS_sign_add_signer");
+       ERR(CMS_final(cms, bm, NULL, CMS_NOCERTS | CMS_BINARY) < 0,
+           "CMS_final");
 
-       if (save_pkcs7) {
-               char *pkcs7_name;
+       if (save_cms) {
+               char *cms_name;
 
-               ERR(asprintf(&pkcs7_name, "%s.pkcs7", module_name) < 0, "asprintf");
-               b = BIO_new_file(pkcs7_name, "wb");
-               ERR(!b, "%s", pkcs7_name);
-               ERR(i2d_PKCS7_bio_stream(b, pkcs7, NULL, 0) < 0, "%s", pkcs7_name);
+               ERR(asprintf(&cms_name, "%s.p7s", module_name) < 0, "asprintf");
+               b = BIO_new_file(cms_name, "wb");
+               ERR(!b, "%s", cms_name);
+               ERR(i2d_CMS_bio_stream(b, cms, NULL, 0) < 0, "%s", cms_name);
                BIO_free(b);
        }
 
+       if (sign_only)
+               return 0;
+
        /* 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))),
@@ -189,9 +244,9 @@ int main(int argc, char **argv)
        ERR(n < 0, "%s", module_name);
        module_size = BIO_number_written(bd);
 
-       ERR(i2d_PKCS7_bio_stream(bd, pkcs7, NULL, 0) < 0, "%s", dest_name);
-       pkcs7_size = BIO_number_written(bd) - module_size;
-       sig_info.sig_len = htonl(pkcs7_size);
+       ERR(i2d_CMS_bio_stream(bd, cms, NULL, 0) < 0, "%s", dest_name);
+       cms_size = BIO_number_written(bd) - module_size;
+       sig_info.sig_len = htonl(cms_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);