1557f75e5SDag-Erling Smørgrav /* $OpenBSD: test_sshkey.c,v 1.4 2015/04/22 01:38:36 djm Exp $ */ 2a0ee8cc6SDag-Erling Smørgrav /* 3a0ee8cc6SDag-Erling Smørgrav * Regress test for sshkey.h key management API 4a0ee8cc6SDag-Erling Smørgrav * 5a0ee8cc6SDag-Erling Smørgrav * Placed in the public domain 6a0ee8cc6SDag-Erling Smørgrav */ 7a0ee8cc6SDag-Erling Smørgrav 8a0ee8cc6SDag-Erling Smørgrav #include "includes.h" 9a0ee8cc6SDag-Erling Smørgrav 10a0ee8cc6SDag-Erling Smørgrav #include <sys/types.h> 11a0ee8cc6SDag-Erling Smørgrav #include <sys/param.h> 12a0ee8cc6SDag-Erling Smørgrav #include <stdio.h> 13a0ee8cc6SDag-Erling Smørgrav #ifdef HAVE_STDINT_H 14a0ee8cc6SDag-Erling Smørgrav #include <stdint.h> 15a0ee8cc6SDag-Erling Smørgrav #endif 16a0ee8cc6SDag-Erling Smørgrav #include <stdlib.h> 17a0ee8cc6SDag-Erling Smørgrav #include <string.h> 18a0ee8cc6SDag-Erling Smørgrav 19a0ee8cc6SDag-Erling Smørgrav #include <openssl/bn.h> 20a0ee8cc6SDag-Erling Smørgrav #include <openssl/rsa.h> 21a0ee8cc6SDag-Erling Smørgrav #include <openssl/dsa.h> 22bc5531deSDag-Erling Smørgrav #if defined(OPENSSL_HAS_ECC) && defined(OPENSSL_HAS_NISTP256) 23a0ee8cc6SDag-Erling Smørgrav # include <openssl/ec.h> 24a0ee8cc6SDag-Erling Smørgrav #endif 25a0ee8cc6SDag-Erling Smørgrav 26a0ee8cc6SDag-Erling Smørgrav #include "../test_helper/test_helper.h" 27a0ee8cc6SDag-Erling Smørgrav 28a0ee8cc6SDag-Erling Smørgrav #include "ssherr.h" 29a0ee8cc6SDag-Erling Smørgrav #include "sshbuf.h" 30a0ee8cc6SDag-Erling Smørgrav #define SSHBUF_INTERNAL 1 /* access internals for testing */ 31a0ee8cc6SDag-Erling Smørgrav #include "sshkey.h" 32a0ee8cc6SDag-Erling Smørgrav 33a0ee8cc6SDag-Erling Smørgrav #include "authfile.h" 34a0ee8cc6SDag-Erling Smørgrav #include "common.h" 35a0ee8cc6SDag-Erling Smørgrav #include "ssh2.h" 36a0ee8cc6SDag-Erling Smørgrav 37a0ee8cc6SDag-Erling Smørgrav void sshkey_tests(void); 38a0ee8cc6SDag-Erling Smørgrav 39a0ee8cc6SDag-Erling Smørgrav static void 40bc5531deSDag-Erling Smørgrav put_opt(struct sshbuf *b, const char *name, const char *value) 41bc5531deSDag-Erling Smørgrav { 42bc5531deSDag-Erling Smørgrav struct sshbuf *sect; 43bc5531deSDag-Erling Smørgrav 44bc5531deSDag-Erling Smørgrav sect = sshbuf_new(); 45bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(sect, NULL); 46bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(b, name), 0); 47bc5531deSDag-Erling Smørgrav if (value != NULL) 48bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(sect, value), 0); 49bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_stringb(b, sect), 0); 50bc5531deSDag-Erling Smørgrav sshbuf_free(sect); 51bc5531deSDag-Erling Smørgrav } 52bc5531deSDag-Erling Smørgrav 53bc5531deSDag-Erling Smørgrav static void 54a0ee8cc6SDag-Erling Smørgrav build_cert(struct sshbuf *b, const struct sshkey *k, const char *type, 55a0ee8cc6SDag-Erling Smørgrav const struct sshkey *sign_key, const struct sshkey *ca_key) 56a0ee8cc6SDag-Erling Smørgrav { 57a0ee8cc6SDag-Erling Smørgrav struct sshbuf *ca_buf, *pk, *principals, *critopts, *exts; 58a0ee8cc6SDag-Erling Smørgrav u_char *sigblob; 59a0ee8cc6SDag-Erling Smørgrav size_t siglen; 60a0ee8cc6SDag-Erling Smørgrav 61a0ee8cc6SDag-Erling Smørgrav ca_buf = sshbuf_new(); 62bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(ca_buf, NULL); 63bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_putb(ca_key, ca_buf), 0); 64a0ee8cc6SDag-Erling Smørgrav 65a0ee8cc6SDag-Erling Smørgrav /* 66a0ee8cc6SDag-Erling Smørgrav * Get the public key serialisation by rendering the key and skipping 67a0ee8cc6SDag-Erling Smørgrav * the type string. This is a bit of a hack :/ 68a0ee8cc6SDag-Erling Smørgrav */ 69a0ee8cc6SDag-Erling Smørgrav pk = sshbuf_new(); 70bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(pk, NULL); 71bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_putb_plain(k, pk), 0); 72a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_skip_string(pk), 0); 73a0ee8cc6SDag-Erling Smørgrav 74a0ee8cc6SDag-Erling Smørgrav principals = sshbuf_new(); 75bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(principals, NULL); 76a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(principals, "gsamsa"), 0); 77a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(principals, "gregor"), 0); 78a0ee8cc6SDag-Erling Smørgrav 79a0ee8cc6SDag-Erling Smørgrav critopts = sshbuf_new(); 80bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(critopts, NULL); 81bc5531deSDag-Erling Smørgrav put_opt(critopts, "force-command", "/usr/local/bin/nethack"); 82bc5531deSDag-Erling Smørgrav put_opt(critopts, "source-address", "192.168.0.0/24,127.0.0.1,::1"); 83a0ee8cc6SDag-Erling Smørgrav 84a0ee8cc6SDag-Erling Smørgrav exts = sshbuf_new(); 85bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(exts, NULL); 86bc5531deSDag-Erling Smørgrav put_opt(critopts, "permit-X11-forwarding", NULL); 87a0ee8cc6SDag-Erling Smørgrav 88a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(b, type), 0); 89a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(b, "noncenoncenonce!"), 0); /* nonce */ 90a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_putb(b, pk), 0); /* public key serialisation */ 91a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_u64(b, 1234), 0); /* serial */ 92a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_u32(b, SSH2_CERT_TYPE_USER), 0); /* type */ 93a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_cstring(b, "gregor"), 0); /* key ID */ 94a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_stringb(b, principals), 0); /* principals */ 95a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_u64(b, 0), 0); /* start */ 96a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_u64(b, 0xffffffffffffffffULL), 0); /* end */ 97a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_stringb(b, critopts), 0); /* options */ 98a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_stringb(b, exts), 0); /* extensions */ 99a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_string(b, NULL, 0), 0); /* reserved */ 100a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_stringb(b, ca_buf), 0); /* signature key */ 101a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_sign(sign_key, &sigblob, &siglen, 102a0ee8cc6SDag-Erling Smørgrav sshbuf_ptr(b), sshbuf_len(b), 0), 0); 103a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshbuf_put_string(b, sigblob, siglen), 0); /* signature */ 104a0ee8cc6SDag-Erling Smørgrav 105a0ee8cc6SDag-Erling Smørgrav free(sigblob); 106a0ee8cc6SDag-Erling Smørgrav sshbuf_free(ca_buf); 107a0ee8cc6SDag-Erling Smørgrav sshbuf_free(exts); 108a0ee8cc6SDag-Erling Smørgrav sshbuf_free(critopts); 109a0ee8cc6SDag-Erling Smørgrav sshbuf_free(principals); 110a0ee8cc6SDag-Erling Smørgrav sshbuf_free(pk); 111a0ee8cc6SDag-Erling Smørgrav } 112a0ee8cc6SDag-Erling Smørgrav 113bc5531deSDag-Erling Smørgrav static void 114bc5531deSDag-Erling Smørgrav signature_test(struct sshkey *k, struct sshkey *bad, const u_char *d, size_t l) 115bc5531deSDag-Erling Smørgrav { 116bc5531deSDag-Erling Smørgrav size_t len; 117bc5531deSDag-Erling Smørgrav u_char *sig; 118bc5531deSDag-Erling Smørgrav 119bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_sign(k, &sig, &len, d, l, 0), 0); 120bc5531deSDag-Erling Smørgrav ASSERT_SIZE_T_GT(len, 8); 121bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(sig, NULL); 122bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_verify(k, sig, len, d, l, 0), 0); 123bc5531deSDag-Erling Smørgrav ASSERT_INT_NE(sshkey_verify(bad, sig, len, d, l, 0), 0); 124bc5531deSDag-Erling Smørgrav /* Fuzz test is more comprehensive, this is just a smoke test */ 125bc5531deSDag-Erling Smørgrav sig[len - 5] ^= 0x10; 126bc5531deSDag-Erling Smørgrav ASSERT_INT_NE(sshkey_verify(k, sig, len, d, l, 0), 0); 127bc5531deSDag-Erling Smørgrav free(sig); 128bc5531deSDag-Erling Smørgrav } 129bc5531deSDag-Erling Smørgrav 130bc5531deSDag-Erling Smørgrav static void 131bc5531deSDag-Erling Smørgrav banana(u_char *s, size_t l) 132bc5531deSDag-Erling Smørgrav { 133bc5531deSDag-Erling Smørgrav size_t o; 134bc5531deSDag-Erling Smørgrav const u_char the_banana[] = { 'b', 'a', 'n', 'a', 'n', 'a' }; 135bc5531deSDag-Erling Smørgrav 136bc5531deSDag-Erling Smørgrav for (o = 0; o < l; o += sizeof(the_banana)) { 137bc5531deSDag-Erling Smørgrav if (l - o < sizeof(the_banana)) { 138bc5531deSDag-Erling Smørgrav memcpy(s + o, "nanananana", l - o); 139bc5531deSDag-Erling Smørgrav break; 140bc5531deSDag-Erling Smørgrav } 141bc5531deSDag-Erling Smørgrav memcpy(s + o, banana, sizeof(the_banana)); 142bc5531deSDag-Erling Smørgrav } 143bc5531deSDag-Erling Smørgrav } 144bc5531deSDag-Erling Smørgrav 145bc5531deSDag-Erling Smørgrav static void 146bc5531deSDag-Erling Smørgrav signature_tests(struct sshkey *k, struct sshkey *bad) 147bc5531deSDag-Erling Smørgrav { 148bc5531deSDag-Erling Smørgrav u_char i, buf[2049]; 149bc5531deSDag-Erling Smørgrav size_t lens[] = { 150bc5531deSDag-Erling Smørgrav 1, 2, 7, 8, 9, 15, 16, 17, 31, 32, 33, 127, 128, 129, 151bc5531deSDag-Erling Smørgrav 255, 256, 257, 1023, 1024, 1025, 2047, 2048, 2049 152bc5531deSDag-Erling Smørgrav }; 153bc5531deSDag-Erling Smørgrav 154bc5531deSDag-Erling Smørgrav for (i = 0; i < (sizeof(lens)/sizeof(lens[0])); i++) { 155bc5531deSDag-Erling Smørgrav test_subtest_info("%s key, banana length %zu", 156bc5531deSDag-Erling Smørgrav sshkey_type(k), lens[i]); 157bc5531deSDag-Erling Smørgrav banana(buf, lens[i]); 158bc5531deSDag-Erling Smørgrav signature_test(k, bad, buf, lens[i]); 159bc5531deSDag-Erling Smørgrav } 160bc5531deSDag-Erling Smørgrav } 161bc5531deSDag-Erling Smørgrav 162bc5531deSDag-Erling Smørgrav static struct sshkey * 163bc5531deSDag-Erling Smørgrav get_private(const char *n) 164bc5531deSDag-Erling Smørgrav { 165bc5531deSDag-Erling Smørgrav struct sshbuf *b; 166bc5531deSDag-Erling Smørgrav struct sshkey *ret; 167bc5531deSDag-Erling Smørgrav 168bc5531deSDag-Erling Smørgrav b = load_file(n); 169bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_parse_private_fileblob(b, "", n, &ret, NULL), 0); 170bc5531deSDag-Erling Smørgrav sshbuf_free(b); 171bc5531deSDag-Erling Smørgrav return ret; 172bc5531deSDag-Erling Smørgrav } 173bc5531deSDag-Erling Smørgrav 174a0ee8cc6SDag-Erling Smørgrav void 175a0ee8cc6SDag-Erling Smørgrav sshkey_tests(void) 176a0ee8cc6SDag-Erling Smørgrav { 177bc5531deSDag-Erling Smørgrav struct sshkey *k1, *k2, *k3, *k4, *kr, *kd, *kf; 178bc5531deSDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 179bc5531deSDag-Erling Smørgrav struct sshkey *ke; 180bc5531deSDag-Erling Smørgrav #endif 181a0ee8cc6SDag-Erling Smørgrav struct sshbuf *b; 182a0ee8cc6SDag-Erling Smørgrav 183a0ee8cc6SDag-Erling Smørgrav TEST_START("new invalid"); 184a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(-42); 185a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1, NULL); 186a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 187a0ee8cc6SDag-Erling Smørgrav 188a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_UNSPEC"); 189a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_UNSPEC); 190a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 191a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 192a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 193a0ee8cc6SDag-Erling Smørgrav 194a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_RSA1"); 195a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_RSA1); 196a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 197a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa, NULL); 198a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->n, NULL); 199a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->e, NULL); 200a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->rsa->p, NULL); 201a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 202a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 203a0ee8cc6SDag-Erling Smørgrav 204a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_RSA"); 205a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_RSA); 206a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 207a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa, NULL); 208a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->n, NULL); 209a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->e, NULL); 210a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->rsa->p, NULL); 211a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 212a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 213a0ee8cc6SDag-Erling Smørgrav 214a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_DSA"); 215a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_DSA); 216a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 217a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa, NULL); 218a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa->g, NULL); 219a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->dsa->priv_key, NULL); 220a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 221a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 222a0ee8cc6SDag-Erling Smørgrav 223bc5531deSDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 224a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_ECDSA"); 225a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_ECDSA); 226a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 227a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->ecdsa, NULL); /* Can't allocate without NID */ 228a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 229a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 230bc5531deSDag-Erling Smørgrav #endif 231a0ee8cc6SDag-Erling Smørgrav 232a0ee8cc6SDag-Erling Smørgrav TEST_START("new/free KEY_ED25519"); 233a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new(KEY_ED25519); 234a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 235a0ee8cc6SDag-Erling Smørgrav /* These should be blank until key loaded or generated */ 236a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->ed25519_sk, NULL); 237a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->ed25519_pk, NULL); 238a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 239a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 240a0ee8cc6SDag-Erling Smørgrav 241a0ee8cc6SDag-Erling Smørgrav TEST_START("new_private KEY_RSA"); 242a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new_private(KEY_RSA); 243a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 244a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa, NULL); 245a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->n, NULL); 246a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->e, NULL); 247a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->p, NULL); 248a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_add_private(k1), 0); 249a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 250a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 251a0ee8cc6SDag-Erling Smørgrav 252a0ee8cc6SDag-Erling Smørgrav TEST_START("new_private KEY_DSA"); 253a0ee8cc6SDag-Erling Smørgrav k1 = sshkey_new_private(KEY_DSA); 254a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 255a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa, NULL); 256a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa->g, NULL); 257a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa->priv_key, NULL); 258a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_add_private(k1), 0); 259a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 260a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 261a0ee8cc6SDag-Erling Smørgrav 262a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_RSA too small modulus"); 263a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_RSA, 128, &k1), 264a0ee8cc6SDag-Erling Smørgrav SSH_ERR_INVALID_ARGUMENT); 265a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1, NULL); 266a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 267a0ee8cc6SDag-Erling Smørgrav 268a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_RSA too large modulus"); 269a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_RSA, 1 << 20, &k1), 270a0ee8cc6SDag-Erling Smørgrav SSH_ERR_INVALID_ARGUMENT); 271a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1, NULL); 272a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 273a0ee8cc6SDag-Erling Smørgrav 274a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_DSA wrong bits"); 275a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_DSA, 2048, &k1), 276a0ee8cc6SDag-Erling Smørgrav SSH_ERR_INVALID_ARGUMENT); 277a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1, NULL); 278a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 279a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 280a0ee8cc6SDag-Erling Smørgrav 281bc5531deSDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 282a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_ECDSA wrong bits"); 283a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_ECDSA, 42, &k1), 284a0ee8cc6SDag-Erling Smørgrav SSH_ERR_INVALID_ARGUMENT); 285a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1, NULL); 286a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 287a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 288bc5531deSDag-Erling Smørgrav #endif 289a0ee8cc6SDag-Erling Smørgrav 290a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_RSA"); 291a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_RSA, 768, &kr), 0); 292a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr, NULL); 293a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr->rsa, NULL); 294a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr->rsa->n, NULL); 295a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr->rsa->e, NULL); 296a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr->rsa->p, NULL); 297a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(BN_num_bits(kr->rsa->n), 768); 298a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 299a0ee8cc6SDag-Erling Smørgrav 300a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_DSA"); 301a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_DSA, 1024, &kd), 0); 302a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kd, NULL); 303a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kd->dsa, NULL); 304a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kd->dsa->g, NULL); 305a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kd->dsa->priv_key, NULL); 306a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 307a0ee8cc6SDag-Erling Smørgrav 308a0ee8cc6SDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 309a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_ECDSA"); 310a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_ECDSA, 256, &ke), 0); 311a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(ke, NULL); 312a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(ke->ecdsa, NULL); 313a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(EC_KEY_get0_public_key(ke->ecdsa), NULL); 314a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(EC_KEY_get0_private_key(ke->ecdsa), NULL); 315a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 316a0ee8cc6SDag-Erling Smørgrav #endif 317a0ee8cc6SDag-Erling Smørgrav 318a0ee8cc6SDag-Erling Smørgrav TEST_START("generate KEY_ED25519"); 319a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_ED25519, 256, &kf), 0); 320a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kf, NULL); 321a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(kf->type, KEY_ED25519); 322a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kf->ed25519_pk, NULL); 323a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kf->ed25519_sk, NULL); 324a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 325a0ee8cc6SDag-Erling Smørgrav 326a0ee8cc6SDag-Erling Smørgrav TEST_START("demote KEY_RSA"); 327a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_demote(kr, &k1), 0); 328a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 329a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kr, k1); 330a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(k1->type, KEY_RSA); 331a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa, NULL); 332a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->n, NULL); 333a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->rsa->e, NULL); 334a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->rsa->p, NULL); 335a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 336a0ee8cc6SDag-Erling Smørgrav 337a0ee8cc6SDag-Erling Smørgrav TEST_START("equal KEY_RSA/demoted KEY_RSA"); 338a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kr, k1), 1); 339a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 340a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 341a0ee8cc6SDag-Erling Smørgrav 342a0ee8cc6SDag-Erling Smørgrav TEST_START("demote KEY_DSA"); 343a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_demote(kd, &k1), 0); 344a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 345a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kd, k1); 346a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(k1->type, KEY_DSA); 347a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa, NULL); 348a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->dsa->g, NULL); 349a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->dsa->priv_key, NULL); 350a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 351a0ee8cc6SDag-Erling Smørgrav 352a0ee8cc6SDag-Erling Smørgrav TEST_START("equal KEY_DSA/demoted KEY_DSA"); 353a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kd, k1), 1); 354a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 355a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 356a0ee8cc6SDag-Erling Smørgrav 357a0ee8cc6SDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 358a0ee8cc6SDag-Erling Smørgrav TEST_START("demote KEY_ECDSA"); 359a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_demote(ke, &k1), 0); 360a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 361a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(ke, k1); 362a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(k1->type, KEY_ECDSA); 363a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->ecdsa, NULL); 364a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(k1->ecdsa_nid, ke->ecdsa_nid); 365a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(EC_KEY_get0_public_key(ke->ecdsa), NULL); 366a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(EC_KEY_get0_private_key(k1->ecdsa), NULL); 367a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 368a0ee8cc6SDag-Erling Smørgrav 369a0ee8cc6SDag-Erling Smørgrav TEST_START("equal KEY_ECDSA/demoted KEY_ECDSA"); 370a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(ke, k1), 1); 371a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 372a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 373a0ee8cc6SDag-Erling Smørgrav #endif 374a0ee8cc6SDag-Erling Smørgrav 375a0ee8cc6SDag-Erling Smørgrav TEST_START("demote KEY_ED25519"); 376a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_demote(kf, &k1), 0); 377a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1, NULL); 378a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(kf, k1); 379a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(k1->type, KEY_ED25519); 380a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_NE(k1->ed25519_pk, NULL); 381a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k1->ed25519_sk, NULL); 382a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 383a0ee8cc6SDag-Erling Smørgrav 384a0ee8cc6SDag-Erling Smørgrav TEST_START("equal KEY_ED25519/demoted KEY_ED25519"); 385a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kf, k1), 1); 386a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 387a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 388a0ee8cc6SDag-Erling Smørgrav 389a0ee8cc6SDag-Erling Smørgrav TEST_START("equal mismatched key types"); 390a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kd, kr), 0); 391a0ee8cc6SDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 392a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kd, ke), 0); 393a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kr, ke), 0); 394a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(ke, kf), 0); 395a0ee8cc6SDag-Erling Smørgrav #endif 396a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kd, kf), 0); 397a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 398a0ee8cc6SDag-Erling Smørgrav 399a0ee8cc6SDag-Erling Smørgrav TEST_START("equal different keys"); 400a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_RSA, 768, &k1), 0); 401a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kr, k1), 0); 402a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 403a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_DSA, 1024, &k1), 0); 404a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kd, k1), 0); 405a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 406a0ee8cc6SDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 407a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_ECDSA, 256, &k1), 0); 408a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(ke, k1), 0); 409a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 410a0ee8cc6SDag-Erling Smørgrav #endif 411a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_generate(KEY_ED25519, 256, &k1), 0); 412a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_equal(kf, k1), 0); 413a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 414a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 415a0ee8cc6SDag-Erling Smørgrav 416a0ee8cc6SDag-Erling Smørgrav sshkey_free(kr); 417a0ee8cc6SDag-Erling Smørgrav sshkey_free(kd); 418a0ee8cc6SDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 419a0ee8cc6SDag-Erling Smørgrav sshkey_free(ke); 420a0ee8cc6SDag-Erling Smørgrav #endif 421a0ee8cc6SDag-Erling Smørgrav sshkey_free(kf); 422a0ee8cc6SDag-Erling Smørgrav 423bc5531deSDag-Erling Smørgrav TEST_START("certify key"); 424bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("ed25519_1.pub"), 425bc5531deSDag-Erling Smørgrav &k1, NULL), 0); 426bc5531deSDag-Erling Smørgrav k2 = get_private("ed25519_2"); 427bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_to_certified(k1, 0), 0); 428bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert, NULL); 429bc5531deSDag-Erling Smørgrav k1->cert->type = SSH2_CERT_TYPE_USER; 430bc5531deSDag-Erling Smørgrav k1->cert->serial = 1234; 431bc5531deSDag-Erling Smørgrav k1->cert->key_id = strdup("estragon"); 432bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->key_id, NULL); 433bc5531deSDag-Erling Smørgrav k1->cert->principals = calloc(4, sizeof(*k1->cert->principals)); 434bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->principals, NULL); 435bc5531deSDag-Erling Smørgrav k1->cert->principals[0] = strdup("estragon"); 436bc5531deSDag-Erling Smørgrav k1->cert->principals[1] = strdup("vladimir"); 437bc5531deSDag-Erling Smørgrav k1->cert->principals[2] = strdup("pozzo"); 438bc5531deSDag-Erling Smørgrav k1->cert->principals[3] = strdup("lucky"); 439bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->principals[0], NULL); 440bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->principals[1], NULL); 441bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->principals[2], NULL); 442bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->principals[3], NULL); 443bc5531deSDag-Erling Smørgrav k1->cert->valid_after = 0; 444bc5531deSDag-Erling Smørgrav k1->cert->valid_before = (u_int64_t)-1; 445bc5531deSDag-Erling Smørgrav k1->cert->critical = sshbuf_new(); 446bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->critical, NULL); 447bc5531deSDag-Erling Smørgrav k1->cert->extensions = sshbuf_new(); 448bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(k1->cert->extensions, NULL); 449bc5531deSDag-Erling Smørgrav put_opt(k1->cert->critical, "force-command", "/usr/bin/true"); 450bc5531deSDag-Erling Smørgrav put_opt(k1->cert->critical, "source-address", "127.0.0.1"); 451bc5531deSDag-Erling Smørgrav put_opt(k1->cert->extensions, "permit-X11-forwarding", NULL); 452bc5531deSDag-Erling Smørgrav put_opt(k1->cert->extensions, "permit-agent-forwarding", NULL); 453bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_from_private(k2, &k1->cert->signature_key), 0); 454bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_certify(k1, k2), 0); 455bc5531deSDag-Erling Smørgrav b = sshbuf_new(); 456bc5531deSDag-Erling Smørgrav ASSERT_PTR_NE(b, NULL); 457bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_putb(k1, b), 0); 458bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_from_blob(sshbuf_ptr(b), sshbuf_len(b), &k3), 0); 459bc5531deSDag-Erling Smørgrav 460bc5531deSDag-Erling Smørgrav sshkey_free(k1); 461bc5531deSDag-Erling Smørgrav sshkey_free(k2); 462bc5531deSDag-Erling Smørgrav sshkey_free(k3); 463bc5531deSDag-Erling Smørgrav sshbuf_reset(b); 464bc5531deSDag-Erling Smørgrav TEST_DONE(); 465bc5531deSDag-Erling Smørgrav 466bc5531deSDag-Erling Smørgrav TEST_START("sign and verify RSA"); 467bc5531deSDag-Erling Smørgrav k1 = get_private("rsa_1"); 468bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("rsa_2.pub"), &k2, 469bc5531deSDag-Erling Smørgrav NULL), 0); 470bc5531deSDag-Erling Smørgrav signature_tests(k1, k2); 471bc5531deSDag-Erling Smørgrav sshkey_free(k1); 472bc5531deSDag-Erling Smørgrav sshkey_free(k2); 473bc5531deSDag-Erling Smørgrav TEST_DONE(); 474bc5531deSDag-Erling Smørgrav 475bc5531deSDag-Erling Smørgrav TEST_START("sign and verify DSA"); 476bc5531deSDag-Erling Smørgrav k1 = get_private("dsa_1"); 477bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("dsa_2.pub"), &k2, 478bc5531deSDag-Erling Smørgrav NULL), 0); 479bc5531deSDag-Erling Smørgrav signature_tests(k1, k2); 480bc5531deSDag-Erling Smørgrav sshkey_free(k1); 481bc5531deSDag-Erling Smørgrav sshkey_free(k2); 482bc5531deSDag-Erling Smørgrav TEST_DONE(); 483bc5531deSDag-Erling Smørgrav 484bc5531deSDag-Erling Smørgrav #ifdef OPENSSL_HAS_ECC 485bc5531deSDag-Erling Smørgrav TEST_START("sign and verify ECDSA"); 486bc5531deSDag-Erling Smørgrav k1 = get_private("ecdsa_1"); 487bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("ecdsa_2.pub"), &k2, 488bc5531deSDag-Erling Smørgrav NULL), 0); 489bc5531deSDag-Erling Smørgrav signature_tests(k1, k2); 490bc5531deSDag-Erling Smørgrav sshkey_free(k1); 491bc5531deSDag-Erling Smørgrav sshkey_free(k2); 492bc5531deSDag-Erling Smørgrav TEST_DONE(); 493bc5531deSDag-Erling Smørgrav #endif 494bc5531deSDag-Erling Smørgrav 495bc5531deSDag-Erling Smørgrav TEST_START("sign and verify ED25519"); 496bc5531deSDag-Erling Smørgrav k1 = get_private("ed25519_1"); 497bc5531deSDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("ed25519_2.pub"), &k2, 498bc5531deSDag-Erling Smørgrav NULL), 0); 499bc5531deSDag-Erling Smørgrav signature_tests(k1, k2); 500bc5531deSDag-Erling Smørgrav sshkey_free(k1); 501bc5531deSDag-Erling Smørgrav sshkey_free(k2); 502bc5531deSDag-Erling Smørgrav TEST_DONE(); 503a0ee8cc6SDag-Erling Smørgrav 504a0ee8cc6SDag-Erling Smørgrav TEST_START("nested certificate"); 505a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_cert(test_data_file("rsa_1"), &k1), 0); 506a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_load_public(test_data_file("rsa_1.pub"), &k2, 507a0ee8cc6SDag-Erling Smørgrav NULL), 0); 508557f75e5SDag-Erling Smørgrav k3 = get_private("rsa_1"); 509a0ee8cc6SDag-Erling Smørgrav build_cert(b, k2, "ssh-rsa-cert-v01@openssh.com", k3, k1); 510a0ee8cc6SDag-Erling Smørgrav ASSERT_INT_EQ(sshkey_from_blob(sshbuf_ptr(b), sshbuf_len(b), &k4), 511a0ee8cc6SDag-Erling Smørgrav SSH_ERR_KEY_CERT_INVALID_SIGN_KEY); 512a0ee8cc6SDag-Erling Smørgrav ASSERT_PTR_EQ(k4, NULL); 513a0ee8cc6SDag-Erling Smørgrav sshkey_free(k1); 514a0ee8cc6SDag-Erling Smørgrav sshkey_free(k2); 515a0ee8cc6SDag-Erling Smørgrav sshkey_free(k3); 516bc5531deSDag-Erling Smørgrav sshbuf_free(b); 517a0ee8cc6SDag-Erling Smørgrav TEST_DONE(); 518a0ee8cc6SDag-Erling Smørgrav 519a0ee8cc6SDag-Erling Smørgrav } 520