summaryrefslogtreecommitdiffstats
path: root/third_party/heimdal/lib/hcrypto/test_rsa.c
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-05 17:47:29 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-05 17:47:29 +0000
commit4f5791ebd03eaec1c7da0865a383175b05102712 (patch)
tree8ce7b00f7a76baa386372422adebbe64510812d4 /third_party/heimdal/lib/hcrypto/test_rsa.c
parentInitial commit. (diff)
downloadsamba-4f5791ebd03eaec1c7da0865a383175b05102712.tar.xz
samba-4f5791ebd03eaec1c7da0865a383175b05102712.zip
Adding upstream version 2:4.17.12+dfsg.upstream/2%4.17.12+dfsgupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/heimdal/lib/hcrypto/test_rsa.c')
-rw-r--r--third_party/heimdal/lib/hcrypto/test_rsa.c397
1 files changed, 397 insertions, 0 deletions
diff --git a/third_party/heimdal/lib/hcrypto/test_rsa.c b/third_party/heimdal/lib/hcrypto/test_rsa.c
new file mode 100644
index 0000000..c9766b8
--- /dev/null
+++ b/third_party/heimdal/lib/hcrypto/test_rsa.c
@@ -0,0 +1,397 @@
+/*
+ * Copyright (c) 2006 - 2007 Kungliga Tekniska Högskolan
+ * (Royal Institute of Technology, Stockholm, Sweden).
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * 3. Neither the name of the Institute nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ */
+
+#include <config.h>
+#include <roken.h>
+#include <getarg.h>
+
+#include <engine.h>
+#include <evp.h>
+
+/*
+ *
+ */
+
+static int version_flag;
+static int help_flag;
+static int time_keygen;
+static char *time_key;
+static int key_blinding = 1;
+static char *rsa_key;
+static char *id_flag;
+static int loops = 1;
+
+static struct getargs args[] = {
+ { "loops", 0, arg_integer, &loops,
+ "number of loops", "loops" },
+ { "id", 0, arg_string, &id_flag,
+ "selects the engine id", "engine-id" },
+ { "time-keygen", 0, arg_flag, &time_keygen,
+ "time rsa generation", NULL },
+ { "time-key", 0, arg_string, &time_key,
+ "rsa key file", NULL },
+ { "key-blinding", 0, arg_negative_flag, &key_blinding,
+ "key blinding", NULL },
+ { "key", 0, arg_string, &rsa_key,
+ "rsa key file", NULL },
+ { "version", 0, arg_flag, &version_flag,
+ "print version", NULL },
+ { "help", 0, arg_flag, &help_flag,
+ NULL, NULL }
+};
+
+/*
+ *
+ */
+
+static void
+check_rsa(const unsigned char *in, size_t len, RSA *rsa, int padding)
+{
+ unsigned char *res, *res2;
+ unsigned int len2;
+ int keylen;
+
+ res = malloc(RSA_size(rsa));
+ if (res == NULL)
+ errx(1, "res: ENOMEM");
+
+ res2 = malloc(RSA_size(rsa));
+ if (res2 == NULL)
+ errx(1, "res2: ENOMEM");
+
+ /* signing */
+
+ keylen = RSA_private_encrypt(len, in, res, rsa, padding);
+ if (keylen <= 0)
+ errx(1, "failed to private encrypt: %d %d", (int)len, (int)keylen);
+
+ if (keylen > RSA_size(rsa))
+ errx(1, "keylen > RSA_size(rsa)");
+
+ keylen = RSA_public_decrypt(keylen, res, res2, rsa, padding);
+ if (keylen <= 0)
+ errx(1, "failed to public decrypt: %d", (int)keylen);
+
+ if (keylen != len)
+ errx(1, "output buffer not same length: %d", (int)keylen);
+
+ if (memcmp(res2, in, len) != 0)
+ errx(1, "string not the same after decryption");
+
+ /* encryption */
+
+ keylen = RSA_public_encrypt(len, in, res, rsa, padding);
+ if (keylen <= 0)
+ errx(1, "failed to public encrypt: %d", (int)keylen);
+
+ if (keylen > RSA_size(rsa))
+ errx(1, "keylen > RSA_size(rsa)");
+
+ keylen = RSA_private_decrypt(keylen, res, res2, rsa, padding);
+ if (keylen <= 0)
+ errx(1, "failed to private decrypt: %d", (int)keylen);
+
+ if (keylen != len)
+ errx(1, "output buffer not same length: %d", (int)keylen);
+
+ if (memcmp(res2, in, len) != 0)
+ errx(1, "string not the same after decryption");
+
+ len2 = keylen;
+
+ if (RSA_sign(NID_sha1, in, len, res, &len2, rsa) != 1)
+ errx(1, "RSA_sign failed");
+
+ if (RSA_verify(NID_sha1, in, len, res, len2, rsa) != 1)
+ errx(1, "RSA_verify failed");
+
+ free(res);
+ free(res2);
+}
+
+static int
+cb_func(int a, int b, BN_GENCB *c)
+{
+ return 1;
+}
+
+static RSA *
+read_key(ENGINE *engine, const char *keyfile)
+{
+ unsigned char buf[1024 * 4];
+ const unsigned char *p;
+ size_t size;
+ RSA *rsa;
+ FILE *f;
+
+ f = fopen(keyfile, "rb");
+ if (f == NULL)
+ err(1, "could not open file %s", keyfile);
+ rk_cloexec_file(f);
+
+ size = fread(buf, 1, sizeof(buf), f);
+ fclose(f);
+ if (size == 0)
+ err(1, "failed to read file %s", keyfile);
+ if (size == sizeof(buf))
+ err(1, "key too long in file %s!", keyfile);
+
+ p = buf;
+ rsa = d2i_RSAPrivateKey(NULL, &p, size);
+ if (rsa == NULL)
+ err(1, "failed to parse key in file %s", keyfile);
+
+ RSA_set_method(rsa, ENGINE_get_RSA(engine));
+
+ if (!key_blinding)
+ rsa->flags |= RSA_FLAG_NO_BLINDING;
+
+ return rsa;
+}
+
+/*
+ *
+ */
+
+static void
+usage (int ret)
+{
+ arg_printusage (args,
+ sizeof(args)/sizeof(*args),
+ NULL,
+ "filename.so");
+ exit (ret);
+}
+
+int
+main(int argc, char **argv)
+{
+ ENGINE *engine = NULL;
+ int i, j, idx = 0;
+ RSA *rsa;
+
+ setprogname(argv[0]);
+
+ if(getarg(args, sizeof(args) / sizeof(args[0]), argc, argv, &idx))
+ usage(1);
+
+ if (help_flag)
+ usage(0);
+
+ if(version_flag){
+ print_version(NULL);
+ exit(0);
+ }
+
+ argc -= idx;
+ argv += idx;
+
+ OpenSSL_add_all_algorithms();
+#ifdef OPENSSL
+ ENGINE_load_openssl();
+#endif
+ ENGINE_load_builtin_engines();
+
+ if (argc == 0) {
+ engine = ENGINE_by_id("builtin");
+ } else {
+ engine = ENGINE_by_id(argv[0]);
+ if (engine == NULL)
+ engine = ENGINE_by_dso(argv[0], id_flag);
+ }
+ if (engine == NULL)
+ errx(1, "ENGINE_by_dso failed");
+
+ if (ENGINE_get_RSA(engine) == NULL)
+ return 77;
+
+ printf("rsa %s\n", ENGINE_get_RSA(engine)->name);
+
+ if (RAND_status() != 1)
+ errx(77, "no functional random device, refusing to run tests");
+
+ if (time_keygen) {
+ struct timeval tv1, tv2;
+ BIGNUM *e;
+
+ rsa = RSA_new_method(engine);
+ if (!key_blinding)
+ rsa->flags |= RSA_FLAG_NO_BLINDING;
+
+ e = BN_new();
+ BN_set_word(e, 0x10001);
+
+ printf("running keygen with %d loops\n", loops);
+
+ gettimeofday(&tv1, NULL);
+
+ for (i = 0; i < loops; i++) {
+ rsa = RSA_new_method(engine);
+ if (RSA_generate_key_ex(rsa, 1024, e, NULL) != 1)
+ errx(1, "RSA_generate_key_ex");
+ RSA_free(rsa);
+ }
+
+ gettimeofday(&tv2, NULL);
+ timevalsub(&tv2, &tv1);
+
+ printf("time %lu.%06lu\n",
+ (unsigned long)tv2.tv_sec,
+ (unsigned long)tv2.tv_usec);
+
+ BN_free(e);
+ ENGINE_finish(engine);
+
+ return 0;
+ }
+
+ if (time_key) {
+ const int size = 20;
+ struct timeval tv1, tv2;
+ unsigned char *p;
+
+ if (strcmp(time_key, "generate") == 0) {
+ BIGNUM *e;
+
+ rsa = RSA_new_method(engine);
+ if (!key_blinding)
+ rsa->flags |= RSA_FLAG_NO_BLINDING;
+
+ e = BN_new();
+ BN_set_word(e, 0x10001);
+
+ if (RSA_generate_key_ex(rsa, 1024, e, NULL) != 1)
+ errx(1, "RSA_generate_key_ex");
+ BN_free(e);
+ } else {
+ rsa = read_key(engine, time_key);
+ }
+
+ p = emalloc(loops * size);
+
+ RAND_bytes(p, loops * size);
+
+ gettimeofday(&tv1, NULL);
+ for (i = 0; i < loops; i++)
+ check_rsa(p + (i * size), size, rsa, RSA_PKCS1_PADDING);
+ gettimeofday(&tv2, NULL);
+
+ timevalsub(&tv2, &tv1);
+
+ printf("time %lu.%06lu\n",
+ (unsigned long)tv2.tv_sec,
+ (unsigned long)tv2.tv_usec);
+
+ RSA_free(rsa);
+ ENGINE_finish(engine);
+
+ free(p);
+ return 0;
+ }
+
+ if (rsa_key) {
+ rsa = read_key(engine, rsa_key);
+
+ /*
+ * Assuming that you use the RSA key in the distribution, this
+ * test will generate a signature have a starting zero and thus
+ * will generate a checksum that is 127 byte instead of the
+ * checksum that is 128 byte (like the key).
+ */
+ {
+ const unsigned char sha1[20] = {
+ 0x6d, 0x33, 0xf9, 0x40, 0x75, 0x5b, 0x4e, 0xc5, 0x90, 0x35,
+ 0x48, 0xab, 0x75, 0x02, 0x09, 0x76, 0x9a, 0xb4, 0x7d, 0x6b
+ };
+
+ check_rsa(sha1, sizeof(sha1), rsa, RSA_PKCS1_PADDING);
+ }
+
+ for (i = 0; i < 128; i++) {
+ unsigned char sha1[20];
+
+ RAND_bytes(sha1, sizeof(sha1));
+ check_rsa(sha1, sizeof(sha1), rsa, RSA_PKCS1_PADDING);
+ }
+ for (i = 0; i < 128; i++) {
+ unsigned char des3[21];
+
+ RAND_bytes(des3, sizeof(des3));
+ check_rsa(des3, sizeof(des3), rsa, RSA_PKCS1_PADDING);
+ }
+ for (i = 0; i < 128; i++) {
+ unsigned char aes[32];
+
+ RAND_bytes(aes, sizeof(aes));
+ check_rsa(aes, sizeof(aes), rsa, RSA_PKCS1_PADDING);
+ }
+
+ RSA_free(rsa);
+ }
+
+ for (i = 0; i < loops; i++) {
+ BN_GENCB cb;
+ BIGNUM *e;
+ unsigned int n;
+
+ rsa = RSA_new_method(engine);
+ if (!key_blinding)
+ rsa->flags |= RSA_FLAG_NO_BLINDING;
+
+ e = BN_new();
+ BN_set_word(e, 0x10001);
+
+ BN_GENCB_set(&cb, cb_func, NULL);
+
+ RAND_bytes(&n, sizeof(n));
+ n &= 0x1ff;
+ n += 1024;
+
+ if (RSA_generate_key_ex(rsa, n, e, &cb) != 1)
+ errx(1, "RSA_generate_key_ex");
+
+ BN_free(e);
+
+ for (j = 0; j < 8; j++) {
+ unsigned char sha1[20];
+ RAND_bytes(sha1, sizeof(sha1));
+ check_rsa(sha1, sizeof(sha1), rsa, RSA_PKCS1_PADDING);
+ }
+
+ RSA_free(rsa);
+ }
+
+ ENGINE_finish(engine);
+
+ return 0;
+}