123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245 |
- /* evp_test.c */
- /*
- * Written by Dr Stephen N Henson (steve@openssl.org) for the OpenSSL
- * project.
- */
- /* ====================================================================
- * Copyright (c) 2015 The OpenSSL Project. 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. All advertising materials mentioning features or use of this
- * software must display the following acknowledgment:
- * "This product includes software developed by the OpenSSL Project
- * for use in the OpenSSL Toolkit. (http://www.OpenSSL.org/)"
- *
- * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
- * endorse or promote products derived from this software without
- * prior written permission. For written permission, please contact
- * licensing@OpenSSL.org.
- *
- * 5. Products derived from this software may not be called "OpenSSL"
- * nor may "OpenSSL" appear in their names without prior written
- * permission of the OpenSSL Project.
- *
- * 6. Redistributions of any form whatsoever must retain the following
- * acknowledgment:
- * "This product includes software developed by the OpenSSL Project
- * for use in the OpenSSL Toolkit (http://www.OpenSSL.org/)"
- *
- * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
- * EXPRESSED 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 OpenSSL PROJECT OR
- * ITS 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 <stdio.h>
- #include <string.h>
- #include <stdlib.h>
- #include <ctype.h>
- #include <openssl/evp.h>
- #include <openssl/pem.h>
- #include <openssl/err.h>
- #include <openssl/x509v3.h>
- /* Remove spaces from beginning and end of a string */
- static void remove_space(char **pval)
- {
- unsigned char *p = (unsigned char *)*pval;
- while (isspace(*p))
- p++;
- *pval = (char *)p;
- p = p + strlen(*pval) - 1;
- /* Remove trailing space */
- while (isspace(*p))
- *p-- = 0;
- }
- /*
- * Given a line of the form:
- * name = value # comment
- * extract name and value. NB: modifies passed buffer.
- */
- static int parse_line(char **pkw, char **pval, char *linebuf)
- {
- char *p;
- p = linebuf + strlen(linebuf) - 1;
- if (*p != '\n') {
- fprintf(stderr, "FATAL: missing EOL\n");
- exit(1);
- }
- /* Look for # */
- p = strchr(linebuf, '#');
- if (p)
- *p = '\0';
- /* Look for = sign */
- p = strchr(linebuf, '=');
- /* If no '=' exit */
- if (!p)
- return 0;
- *p++ = '\0';
- *pkw = linebuf;
- *pval = p;
- /* Remove spaces from keyword and value */
- remove_space(pkw);
- remove_space(pval);
- return 1;
- }
- /* For a hex string "value" convert to a binary allocated buffer */
- static int test_bin(const char *value, unsigned char **buf, size_t *buflen)
- {
- long len;
- if (!*value) {
- /* Don't return NULL for zero length buffer */
- *buf = OPENSSL_malloc(1);
- if (!*buf)
- return 0;
- **buf = 0;
- *buflen = 0;
- return 1;
- }
- /* Check for string literal */
- if (value[0] == '"') {
- size_t vlen;
- value++;
- vlen = strlen(value);
- if (value[vlen - 1] != '"')
- return 0;
- vlen--;
- *buf = BUF_memdup(value, vlen);
- *buflen = vlen;
- return 1;
- }
- *buf = string_to_hex(value, &len);
- if (!*buf) {
- fprintf(stderr, "Value=%s\n", value);
- ERR_print_errors_fp(stderr);
- return -1;
- }
- /* Size of input buffer means we'll never overflow */
- *buflen = len;
- return 1;
- }
- /* Structure holding test information */
- struct evp_test {
- /* file being read */
- FILE *in;
- /* List of public and private keys */
- struct key_list *private;
- struct key_list *public;
- /* method for this test */
- const struct evp_test_method *meth;
- /* current line being processed */
- unsigned int line;
- /* start line of current test */
- unsigned int start_line;
- /* Error string for test */
- const char *err;
- /* Expected error value of test */
- char *expected_err;
- /* Number of tests */
- int ntests;
- /* Error count */
- int errors;
- /* Number of tests skipped */
- int nskip;
- /* If output mismatch expected and got value */
- unsigned char *out_got;
- unsigned char *out_expected;
- size_t out_len;
- /* test specific data */
- void *data;
- /* Current test should be skipped */
- int skip;
- };
- struct key_list {
- char *name;
- EVP_PKEY *key;
- struct key_list *next;
- };
- /* Test method structure */
- struct evp_test_method {
- /* Name of test as it appears in file */
- const char *name;
- /* Initialise test for "alg" */
- int (*init) (struct evp_test * t, const char *alg);
- /* Clean up method */
- void (*cleanup) (struct evp_test * t);
- /* Test specific name value pair processing */
- int (*parse) (struct evp_test * t, const char *name, const char *value);
- /* Run the test itself */
- int (*run_test) (struct evp_test * t);
- };
- static const struct evp_test_method digest_test_method, cipher_test_method;
- static const struct evp_test_method mac_test_method;
- static const struct evp_test_method psign_test_method, pverify_test_method;
- static const struct evp_test_method pdecrypt_test_method;
- static const struct evp_test_method pverify_recover_test_method;
- static const struct evp_test_method *evp_test_list[] = {
- &digest_test_method,
- &cipher_test_method,
- &mac_test_method,
- &psign_test_method,
- &pverify_test_method,
- &pdecrypt_test_method,
- &pverify_recover_test_method,
- NULL
- };
- static const struct evp_test_method *evp_find_test(const char *name)
- {
- const struct evp_test_method **tt;
- for (tt = evp_test_list; *tt; tt++) {
- if (strcmp(name, (*tt)->name) == 0)
- return *tt;
- }
- return NULL;
- }
- static void hex_print(const char *name, const unsigned char *buf, size_t len)
- {
- size_t i;
- fprintf(stderr, "%s ", name);
- for (i = 0; i < len; i++)
- fprintf(stderr, "%02X", buf[i]);
- fputs("\n", stderr);
- }
- static void free_expected(struct evp_test *t)
- {
- OPENSSL_free(t->expected_err);
- t->expected_err = NULL;
- OPENSSL_free(t->out_expected);
- OPENSSL_free(t->out_got);
- t->out_expected = NULL;
- t->out_got = NULL;
- }
- static void print_expected(struct evp_test *t)
- {
- if (t->out_expected == NULL)
- return;
- hex_print("Expected:", t->out_expected, t->out_len);
- hex_print("Got: ", t->out_got, t->out_len);
- free_expected(t);
- }
- static int check_test_error(struct evp_test *t)
- {
- if (!t->err && !t->expected_err)
- return 1;
- if (t->err && !t->expected_err) {
- fprintf(stderr, "Test line %d: unexpected error %s\n",
- t->start_line, t->err);
- print_expected(t);
- return 0;
- }
- if (!t->err && t->expected_err) {
- fprintf(stderr, "Test line %d: succeeded expecting %s\n",
- t->start_line, t->expected_err);
- return 0;
- }
- if (strcmp(t->err, t->expected_err) == 0)
- return 1;
- fprintf(stderr, "Test line %d: expecting %s got %s\n",
- t->start_line, t->expected_err, t->err);
- return 0;
- }
- /* Setup a new test, run any existing test */
- static int setup_test(struct evp_test *t, const struct evp_test_method *tmeth)
- {
- /* If we already have a test set up run it */
- if (t->meth) {
- t->ntests++;
- if (t->skip) {
- t->meth = tmeth;
- t->nskip++;
- return 1;
- }
- t->err = NULL;
- if (t->meth->run_test(t) != 1) {
- fprintf(stderr, "%s test error line %d\n",
- t->meth->name, t->start_line);
- return 0;
- }
- if (!check_test_error(t)) {
- if (t->err)
- ERR_print_errors_fp(stderr);
- t->errors++;
- }
- ERR_clear_error();
- t->meth->cleanup(t);
- OPENSSL_free(t->data);
- t->data = NULL;
- OPENSSL_free(t->expected_err);
- t->expected_err = NULL;
- free_expected(t);
- }
- t->meth = tmeth;
- return 1;
- }
- static int find_key(EVP_PKEY **ppk, const char *name, struct key_list *lst)
- {
- for (; lst; lst = lst->next) {
- if (strcmp(lst->name, name) == 0) {
- if (ppk)
- *ppk = lst->key;
- return 1;
- }
- }
- return 0;
- }
- static void free_key_list(struct key_list *lst)
- {
- while (lst != NULL) {
- struct key_list *ltmp;
- EVP_PKEY_free(lst->key);
- OPENSSL_free(lst->name);
- ltmp = lst->next;
- OPENSSL_free(lst);
- lst = ltmp;
- }
- }
- static int check_unsupported()
- {
- long err = ERR_peek_error();
- if (ERR_GET_LIB(err) == ERR_LIB_EVP
- && ERR_GET_REASON(err) == EVP_R_UNSUPPORTED_ALGORITHM) {
- ERR_clear_error();
- return 1;
- }
- return 0;
- }
- static int process_test(struct evp_test *t, char *buf, int verbose)
- {
- char *keyword = NULL, *value = NULL;
- int rv = 0, add_key = 0;
- long save_pos = 0;
- struct key_list **lst = NULL, *key = NULL;
- EVP_PKEY *pk = NULL;
- const struct evp_test_method *tmeth = NULL;
- if (verbose)
- fputs(buf, stdout);
- if (!parse_line(&keyword, &value, buf))
- return 1;
- if (strcmp(keyword, "PrivateKey") == 0) {
- save_pos = ftell(t->in);
- pk = PEM_read_PrivateKey(t->in, NULL, 0, NULL);
- if (pk == NULL && !check_unsupported()) {
- fprintf(stderr, "Error reading private key %s\n", value);
- ERR_print_errors_fp(stderr);
- return 0;
- }
- lst = &t->private;
- add_key = 1;
- }
- if (strcmp(keyword, "PublicKey") == 0) {
- save_pos = ftell(t->in);
- pk = PEM_read_PUBKEY(t->in, NULL, 0, NULL);
- if (pk == NULL && !check_unsupported()) {
- fprintf(stderr, "Error reading public key %s\n", value);
- ERR_print_errors_fp(stderr);
- return 0;
- }
- lst = &t->public;
- add_key = 1;
- }
- /* If we have a key add to list */
- if (add_key) {
- char tmpbuf[80];
- if (find_key(NULL, value, *lst)) {
- fprintf(stderr, "Duplicate key %s\n", value);
- return 0;
- }
- key = OPENSSL_malloc(sizeof(*key));
- if (!key)
- return 0;
- key->name = BUF_strdup(value);
- key->key = pk;
- key->next = *lst;
- *lst = key;
- /* Rewind input, read to end and update line numbers */
- fseek(t->in, save_pos, SEEK_SET);
- while (fgets(tmpbuf, sizeof(tmpbuf), t->in)) {
- t->line++;
- if (strncmp(tmpbuf, "-----END", 8) == 0)
- return 1;
- }
- fprintf(stderr, "Can't find key end\n");
- return 0;
- }
- /* See if keyword corresponds to a test start */
- tmeth = evp_find_test(keyword);
- if (tmeth) {
- if (!setup_test(t, tmeth))
- return 0;
- t->start_line = t->line;
- t->skip = 0;
- if (!tmeth->init(t, value)) {
- fprintf(stderr, "Unknown %s: %s\n", keyword, value);
- return 0;
- }
- return 1;
- } else if (t->skip) {
- return 1;
- } else if (strcmp(keyword, "Result") == 0) {
- if (t->expected_err) {
- fprintf(stderr, "Line %d: multiple result lines\n", t->line);
- return 0;
- }
- t->expected_err = BUF_strdup(value);
- if (!t->expected_err)
- return 0;
- } else {
- /* Must be test specific line: try to parse it */
- if (t->meth)
- rv = t->meth->parse(t, keyword, value);
- if (rv == 0)
- fprintf(stderr, "line %d: unexpected keyword %s\n",
- t->line, keyword);
- if (rv < 0)
- fprintf(stderr, "line %d: error processing keyword %s\n",
- t->line, keyword);
- if (rv <= 0)
- return 0;
- }
- return 1;
- }
- static int check_output(struct evp_test *t, const unsigned char *expected,
- const unsigned char *got, size_t len)
- {
- if (!memcmp(expected, got, len))
- return 0;
- t->out_expected = BUF_memdup(expected, len);
- t->out_got = BUF_memdup(got, len);
- t->out_len = len;
- if (t->out_expected == NULL || t->out_got == NULL) {
- fprintf(stderr, "Memory allocation error!\n");
- exit(1);
- }
- return 1;
- }
- int main(int argc, char **argv)
- {
- FILE *in = NULL;
- char buf[10240];
- struct evp_test t;
- if (argc != 2) {
- fprintf(stderr, "usage: evp_test testfile.txt\n");
- return 1;
- }
- CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON);
- ERR_load_crypto_strings();
- OpenSSL_add_all_algorithms();
- memset(&t, 0, sizeof(t));
- t.meth = NULL;
- t.public = NULL;
- t.private = NULL;
- t.err = NULL;
- t.line = 0;
- t.start_line = -1;
- t.errors = 0;
- t.ntests = 0;
- t.out_expected = NULL;
- t.out_got = NULL;
- t.out_len = 0;
- in = fopen(argv[1], "r");
- t.in = in;
- while (fgets(buf, sizeof(buf), in)) {
- t.line++;
- if (!process_test(&t, buf, 0))
- exit(1);
- }
- /* Run any final test we have */
- if (!setup_test(&t, NULL))
- exit(1);
- fprintf(stderr, "%d tests completed with %d errors, %d skipped\n",
- t.ntests, t.errors, t.nskip);
- free_key_list(t.public);
- free_key_list(t.private);
- fclose(in);
- EVP_cleanup();
- CRYPTO_cleanup_all_ex_data();
- ERR_remove_thread_state(NULL);
- ERR_free_strings();
- CRYPTO_mem_leaks_fp(stderr);
- if (t.errors)
- return 1;
- return 0;
- }
- static void test_free(void *d)
- {
- OPENSSL_free(d);
- }
- /* Message digest tests */
- struct digest_data {
- /* Digest this test is for */
- const EVP_MD *digest;
- /* Input to digest */
- unsigned char *input;
- size_t input_len;
- /* Repeat count for input */
- size_t nrpt;
- /* Expected output */
- unsigned char *output;
- size_t output_len;
- };
- static int digest_test_init(struct evp_test *t, const char *alg)
- {
- const EVP_MD *digest;
- struct digest_data *mdat = t->data;
- digest = EVP_get_digestbyname(alg);
- if (!digest) {
- /* If alg has an OID assume disabled algorithm */
- if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
- t->skip = 1;
- return 1;
- }
- return 0;
- }
- mdat = OPENSSL_malloc(sizeof(*mdat));
- mdat->digest = digest;
- mdat->input = NULL;
- mdat->output = NULL;
- mdat->nrpt = 1;
- t->data = mdat;
- return 1;
- }
- static void digest_test_cleanup(struct evp_test *t)
- {
- struct digest_data *mdat = t->data;
- test_free(mdat->input);
- test_free(mdat->output);
- }
- static int digest_test_parse(struct evp_test *t,
- const char *keyword, const char *value)
- {
- struct digest_data *mdata = t->data;
- if (strcmp(keyword, "Input") == 0)
- return test_bin(value, &mdata->input, &mdata->input_len);
- if (strcmp(keyword, "Output") == 0)
- return test_bin(value, &mdata->output, &mdata->output_len);
- if (strcmp(keyword, "Count") == 0) {
- long nrpt = atoi(value);
- if (nrpt <= 0)
- return 0;
- mdata->nrpt = (size_t)nrpt;
- return 1;
- }
- return 0;
- }
- static int digest_test_run(struct evp_test *t)
- {
- struct digest_data *mdata = t->data;
- size_t i;
- const char *err = "INTERNAL_ERROR";
- EVP_MD_CTX *mctx;
- unsigned char md[EVP_MAX_MD_SIZE];
- unsigned int md_len;
- mctx = EVP_MD_CTX_create();
- if (!mctx)
- goto err;
- err = "DIGESTINIT_ERROR";
- if (!EVP_DigestInit_ex(mctx, mdata->digest, NULL))
- goto err;
- err = "DIGESTUPDATE_ERROR";
- for (i = 0; i < mdata->nrpt; i++) {
- if (!EVP_DigestUpdate(mctx, mdata->input, mdata->input_len))
- goto err;
- }
- err = "DIGESTFINAL_ERROR";
- if (!EVP_DigestFinal(mctx, md, &md_len))
- goto err;
- err = "DIGEST_LENGTH_MISMATCH";
- if (md_len != mdata->output_len)
- goto err;
- err = "DIGEST_MISMATCH";
- if (check_output(t, mdata->output, md, md_len))
- goto err;
- err = NULL;
- err:
- if (mctx)
- EVP_MD_CTX_destroy(mctx);
- t->err = err;
- return 1;
- }
- static const struct evp_test_method digest_test_method = {
- "Digest",
- digest_test_init,
- digest_test_cleanup,
- digest_test_parse,
- digest_test_run
- };
- /* Cipher tests */
- struct cipher_data {
- const EVP_CIPHER *cipher;
- int enc;
- /* EVP_CIPH_GCM_MODE, EVP_CIPH_CCM_MODE or EVP_CIPH_OCB_MODE if AEAD */
- int aead;
- unsigned char *key;
- size_t key_len;
- unsigned char *iv;
- size_t iv_len;
- unsigned char *plaintext;
- size_t plaintext_len;
- unsigned char *ciphertext;
- size_t ciphertext_len;
- /* GCM, CCM only */
- unsigned char *aad;
- size_t aad_len;
- unsigned char *tag;
- size_t tag_len;
- };
- static int cipher_test_init(struct evp_test *t, const char *alg)
- {
- const EVP_CIPHER *cipher;
- struct cipher_data *cdat = t->data;
- cipher = EVP_get_cipherbyname(alg);
- if (!cipher) {
- /* If alg has an OID assume disabled algorithm */
- if (OBJ_sn2nid(alg) != NID_undef || OBJ_ln2nid(alg) != NID_undef) {
- t->skip = 1;
- return 1;
- }
- return 0;
- }
- cdat = OPENSSL_malloc(sizeof(*cdat));
- cdat->cipher = cipher;
- cdat->enc = -1;
- cdat->key = NULL;
- cdat->iv = NULL;
- cdat->ciphertext = NULL;
- cdat->plaintext = NULL;
- cdat->aad = NULL;
- cdat->tag = NULL;
- t->data = cdat;
- if (EVP_CIPHER_mode(cipher) == EVP_CIPH_GCM_MODE
- || EVP_CIPHER_mode(cipher) == EVP_CIPH_OCB_MODE
- || EVP_CIPHER_mode(cipher) == EVP_CIPH_CCM_MODE)
- cdat->aead = EVP_CIPHER_mode(cipher);
- else
- cdat->aead = 0;
- return 1;
- }
- static void cipher_test_cleanup(struct evp_test *t)
- {
- struct cipher_data *cdat = t->data;
- test_free(cdat->key);
- test_free(cdat->iv);
- test_free(cdat->ciphertext);
- test_free(cdat->plaintext);
- test_free(cdat->aad);
- test_free(cdat->tag);
- }
- static int cipher_test_parse(struct evp_test *t, const char *keyword,
- const char *value)
- {
- struct cipher_data *cdat = t->data;
- if (strcmp(keyword, "Key") == 0)
- return test_bin(value, &cdat->key, &cdat->key_len);
- if (strcmp(keyword, "IV") == 0)
- return test_bin(value, &cdat->iv, &cdat->iv_len);
- if (strcmp(keyword, "Plaintext") == 0)
- return test_bin(value, &cdat->plaintext, &cdat->plaintext_len);
- if (strcmp(keyword, "Ciphertext") == 0)
- return test_bin(value, &cdat->ciphertext, &cdat->ciphertext_len);
- if (cdat->aead) {
- if (strcmp(keyword, "AAD") == 0)
- return test_bin(value, &cdat->aad, &cdat->aad_len);
- if (strcmp(keyword, "Tag") == 0)
- return test_bin(value, &cdat->tag, &cdat->tag_len);
- }
- if (strcmp(keyword, "Operation") == 0) {
- if (strcmp(value, "ENCRYPT") == 0)
- cdat->enc = 1;
- else if (strcmp(value, "DECRYPT") == 0)
- cdat->enc = 0;
- else
- return 0;
- return 1;
- }
- return 0;
- }
- static int cipher_test_enc(struct evp_test *t, int enc)
- {
- struct cipher_data *cdat = t->data;
- unsigned char *in, *out, *tmp = NULL;
- size_t in_len, out_len;
- int tmplen, tmpflen;
- EVP_CIPHER_CTX *ctx = NULL;
- const char *err;
- err = "INTERNAL_ERROR";
- ctx = EVP_CIPHER_CTX_new();
- if (!ctx)
- goto err;
- EVP_CIPHER_CTX_set_flags(ctx, EVP_CIPHER_CTX_FLAG_WRAP_ALLOW);
- if (enc) {
- in = cdat->plaintext;
- in_len = cdat->plaintext_len;
- out = cdat->ciphertext;
- out_len = cdat->ciphertext_len;
- } else {
- in = cdat->ciphertext;
- in_len = cdat->ciphertext_len;
- out = cdat->plaintext;
- out_len = cdat->plaintext_len;
- }
- tmp = OPENSSL_malloc(in_len + 2 * EVP_MAX_BLOCK_LENGTH);
- if (!tmp)
- goto err;
- err = "CIPHERINIT_ERROR";
- if (!EVP_CipherInit_ex(ctx, cdat->cipher, NULL, NULL, NULL, enc))
- goto err;
- err = "INVALID_IV_LENGTH";
- if (cdat->iv) {
- if (cdat->aead) {
- if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_IVLEN,
- cdat->iv_len, 0))
- goto err;
- } else if (cdat->iv_len != (size_t)EVP_CIPHER_CTX_iv_length(ctx))
- goto err;
- }
- if (cdat->aead) {
- unsigned char *tag;
- /*
- * If encrypting or OCB just set tag length initially, otherwise
- * set tag length and value.
- */
- if (enc || cdat->aead == EVP_CIPH_OCB_MODE) {
- err = "TAG_LENGTH_SET_ERROR";
- tag = NULL;
- } else {
- err = "TAG_SET_ERROR";
- tag = cdat->tag;
- }
- if (tag || cdat->aead != EVP_CIPH_GCM_MODE) {
- if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_TAG,
- cdat->tag_len, tag))
- goto err;
- }
- }
- err = "INVALID_KEY_LENGTH";
- if (!EVP_CIPHER_CTX_set_key_length(ctx, cdat->key_len))
- goto err;
- err = "KEY_SET_ERROR";
- if (!EVP_CipherInit_ex(ctx, NULL, NULL, cdat->key, cdat->iv, -1))
- goto err;
- if (!enc && cdat->aead == EVP_CIPH_OCB_MODE) {
- if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_SET_TAG,
- cdat->tag_len, cdat->tag)) {
- err = "TAG_SET_ERROR";
- goto err;
- }
- }
- if (cdat->aead == EVP_CIPH_CCM_MODE) {
- if (!EVP_CipherUpdate(ctx, NULL, &tmplen, NULL, out_len)) {
- err = "CCM_PLAINTEXT_LENGTH_SET_ERROR";
- goto err;
- }
- }
- if (cdat->aad) {
- if (!EVP_CipherUpdate(ctx, NULL, &tmplen, cdat->aad, cdat->aad_len)) {
- err = "AAD_SET_ERROR";
- goto err;
- }
- }
- EVP_CIPHER_CTX_set_padding(ctx, 0);
- err = "CIPHERUPDATE_ERROR";
- if (!EVP_CipherUpdate(ctx, tmp, &tmplen, in, in_len))
- goto err;
- if (cdat->aead == EVP_CIPH_CCM_MODE)
- tmpflen = 0;
- else {
- err = "CIPHERFINAL_ERROR";
- if (!EVP_CipherFinal_ex(ctx, tmp + tmplen, &tmpflen))
- goto err;
- }
- err = "LENGTH_MISMATCH";
- if (out_len != (size_t)(tmplen + tmpflen))
- goto err;
- err = "VALUE_MISMATCH";
- if (check_output(t, out, tmp, out_len))
- goto err;
- if (enc && cdat->aead) {
- unsigned char rtag[16];
- if (cdat->tag_len > sizeof(rtag)) {
- err = "TAG_LENGTH_INTERNAL_ERROR";
- goto err;
- }
- if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_AEAD_GET_TAG,
- cdat->tag_len, rtag)) {
- err = "TAG_RETRIEVE_ERROR";
- goto err;
- }
- if (check_output(t, cdat->tag, rtag, cdat->tag_len)) {
- err = "TAG_VALUE_MISMATCH";
- goto err;
- }
- }
- err = NULL;
- err:
- OPENSSL_free(tmp);
- EVP_CIPHER_CTX_free(ctx);
- t->err = err;
- return err ? 0 : 1;
- }
- static int cipher_test_run(struct evp_test *t)
- {
- struct cipher_data *cdat = t->data;
- int rv;
- if (!cdat->key) {
- t->err = "NO_KEY";
- return 0;
- }
- if (!cdat->iv && EVP_CIPHER_iv_length(cdat->cipher)) {
- /* IV is optional and usually omitted in wrap mode */
- if (EVP_CIPHER_mode(cdat->cipher) != EVP_CIPH_WRAP_MODE) {
- t->err = "NO_IV";
- return 0;
- }
- }
- if (cdat->aead && !cdat->tag) {
- t->err = "NO_TAG";
- return 0;
- }
- if (cdat->enc) {
- rv = cipher_test_enc(t, 1);
- /* Not fatal errors: return */
- if (rv != 1) {
- if (rv < 0)
- return 0;
- return 1;
- }
- }
- if (cdat->enc != 1) {
- rv = cipher_test_enc(t, 0);
- /* Not fatal errors: return */
- if (rv != 1) {
- if (rv < 0)
- return 0;
- return 1;
- }
- }
- return 1;
- }
- static const struct evp_test_method cipher_test_method = {
- "Cipher",
- cipher_test_init,
- cipher_test_cleanup,
- cipher_test_parse,
- cipher_test_run
- };
- struct mac_data {
- /* MAC type */
- int type;
- /* Algorithm string for this MAC */
- char *alg;
- /* MAC key */
- unsigned char *key;
- size_t key_len;
- /* Input to MAC */
- unsigned char *input;
- size_t input_len;
- /* Expected output */
- unsigned char *output;
- size_t output_len;
- };
- static int mac_test_init(struct evp_test *t, const char *alg)
- {
- int type;
- struct mac_data *mdat;
- if (strcmp(alg, "HMAC") == 0)
- type = EVP_PKEY_HMAC;
- else if (strcmp(alg, "CMAC") == 0)
- type = EVP_PKEY_CMAC;
- else
- return 0;
- mdat = OPENSSL_malloc(sizeof(*mdat));
- mdat->type = type;
- mdat->alg = NULL;
- mdat->key = NULL;
- mdat->input = NULL;
- mdat->output = NULL;
- t->data = mdat;
- return 1;
- }
- static void mac_test_cleanup(struct evp_test *t)
- {
- struct mac_data *mdat = t->data;
- test_free(mdat->alg);
- test_free(mdat->key);
- test_free(mdat->input);
- test_free(mdat->output);
- }
- static int mac_test_parse(struct evp_test *t,
- const char *keyword, const char *value)
- {
- struct mac_data *mdata = t->data;
- if (strcmp(keyword, "Key") == 0)
- return test_bin(value, &mdata->key, &mdata->key_len);
- if (strcmp(keyword, "Algorithm") == 0) {
- mdata->alg = BUF_strdup(value);
- if (!mdata->alg)
- return 0;
- return 1;
- }
- if (strcmp(keyword, "Input") == 0)
- return test_bin(value, &mdata->input, &mdata->input_len);
- if (strcmp(keyword, "Output") == 0)
- return test_bin(value, &mdata->output, &mdata->output_len);
- return 0;
- }
- static int mac_test_run(struct evp_test *t)
- {
- struct mac_data *mdata = t->data;
- const char *err = "INTERNAL_ERROR";
- EVP_MD_CTX *mctx = NULL;
- EVP_PKEY_CTX *pctx = NULL, *genctx = NULL;
- EVP_PKEY *key = NULL;
- const EVP_MD *md = NULL;
- unsigned char *mac = NULL;
- size_t mac_len;
- err = "MAC_PKEY_CTX_ERROR";
- genctx = EVP_PKEY_CTX_new_id(mdata->type, NULL);
- if (!genctx)
- goto err;
- err = "MAC_KEYGEN_INIT_ERROR";
- if (EVP_PKEY_keygen_init(genctx) <= 0)
- goto err;
- if (mdata->type == EVP_PKEY_CMAC) {
- err = "MAC_ALGORITHM_SET_ERROR";
- if (EVP_PKEY_CTX_ctrl_str(genctx, "cipher", mdata->alg) <= 0)
- goto err;
- }
- err = "MAC_KEY_SET_ERROR";
- if (EVP_PKEY_CTX_set_mac_key(genctx, mdata->key, mdata->key_len) <= 0)
- goto err;
- err = "MAC_KEY_GENERATE_ERROR";
- if (EVP_PKEY_keygen(genctx, &key) <= 0)
- goto err;
- if (mdata->type == EVP_PKEY_HMAC) {
- err = "MAC_ALGORITHM_SET_ERROR";
- md = EVP_get_digestbyname(mdata->alg);
- if (!md)
- goto err;
- }
- mctx = EVP_MD_CTX_create();
- if (!mctx)
- goto err;
- err = "DIGESTSIGNINIT_ERROR";
- if (!EVP_DigestSignInit(mctx, &pctx, md, NULL, key))
- goto err;
- err = "DIGESTSIGNUPDATE_ERROR";
- if (!EVP_DigestSignUpdate(mctx, mdata->input, mdata->input_len))
- goto err;
- err = "DIGESTSIGNFINAL_LENGTH_ERROR";
- if (!EVP_DigestSignFinal(mctx, NULL, &mac_len))
- goto err;
- mac = OPENSSL_malloc(mac_len);
- if (!mac) {
- fprintf(stderr, "Error allocating mac buffer!\n");
- exit(1);
- }
- if (!EVP_DigestSignFinal(mctx, mac, &mac_len))
- goto err;
- err = "MAC_LENGTH_MISMATCH";
- if (mac_len != mdata->output_len)
- goto err;
- err = "MAC_MISMATCH";
- if (check_output(t, mdata->output, mac, mac_len))
- goto err;
- err = NULL;
- err:
- if (mctx)
- EVP_MD_CTX_destroy(mctx);
- OPENSSL_free(mac);
- EVP_PKEY_CTX_free(genctx);
- EVP_PKEY_free(key);
- t->err = err;
- return 1;
- }
- static const struct evp_test_method mac_test_method = {
- "MAC",
- mac_test_init,
- mac_test_cleanup,
- mac_test_parse,
- mac_test_run
- };
- /*
- * Public key operations. These are all very similar and can share
- * a lot of common code.
- */
- struct pkey_data {
- /* Context for this operation */
- EVP_PKEY_CTX *ctx;
- /* Key operation to perform */
- int (*keyop) (EVP_PKEY_CTX *ctx,
- unsigned char *sig, size_t *siglen,
- const unsigned char *tbs, size_t tbslen);
- /* Input to MAC */
- unsigned char *input;
- size_t input_len;
- /* Expected output */
- unsigned char *output;
- size_t output_len;
- };
- /*
- * Perform public key operation setup: lookup key, allocated ctx and call
- * the appropriate initialisation function
- */
- static int pkey_test_init(struct evp_test *t, const char *name,
- int use_public,
- int (*keyopinit) (EVP_PKEY_CTX *ctx),
- int (*keyop) (EVP_PKEY_CTX *ctx,
- unsigned char *sig, size_t *siglen,
- const unsigned char *tbs,
- size_t tbslen)
- )
- {
- struct pkey_data *kdata;
- EVP_PKEY *pkey = NULL;
- int rv = 0;
- if (use_public)
- rv = find_key(&pkey, name, t->public);
- if (!rv)
- rv = find_key(&pkey, name, t->private);
- if (!rv)
- return 0;
- if (!pkey) {
- t->skip = 1;
- return 1;
- }
- kdata = OPENSSL_malloc(sizeof(*kdata));
- if (!kdata) {
- EVP_PKEY_free(pkey);
- return 0;
- }
- kdata->ctx = NULL;
- kdata->input = NULL;
- kdata->output = NULL;
- kdata->keyop = keyop;
- t->data = kdata;
- kdata->ctx = EVP_PKEY_CTX_new(pkey, NULL);
- if (!kdata->ctx)
- return 0;
- if (keyopinit(kdata->ctx) <= 0)
- return 0;
- return 1;
- }
- static void pkey_test_cleanup(struct evp_test *t)
- {
- struct pkey_data *kdata = t->data;
- OPENSSL_free(kdata->input);
- OPENSSL_free(kdata->output);
- EVP_PKEY_CTX_free(kdata->ctx);
- }
- static int pkey_test_parse(struct evp_test *t,
- const char *keyword, const char *value)
- {
- struct pkey_data *kdata = t->data;
- if (strcmp(keyword, "Input") == 0)
- return test_bin(value, &kdata->input, &kdata->input_len);
- if (strcmp(keyword, "Output") == 0)
- return test_bin(value, &kdata->output, &kdata->output_len);
- if (strcmp(keyword, "Ctrl") == 0) {
- char *p = strchr(value, ':');
- if (p)
- *p++ = 0;
- if (EVP_PKEY_CTX_ctrl_str(kdata->ctx, value, p) <= 0)
- return 0;
- return 1;
- }
- return 0;
- }
- static int pkey_test_run(struct evp_test *t)
- {
- struct pkey_data *kdata = t->data;
- unsigned char *out = NULL;
- size_t out_len;
- const char *err = "KEYOP_LENGTH_ERROR";
- if (kdata->keyop(kdata->ctx, NULL, &out_len, kdata->input,
- kdata->input_len) <= 0)
- goto err;
- out = OPENSSL_malloc(out_len);
- if (!out) {
- fprintf(stderr, "Error allocating output buffer!\n");
- exit(1);
- }
- err = "KEYOP_ERROR";
- if (kdata->keyop
- (kdata->ctx, out, &out_len, kdata->input, kdata->input_len) <= 0)
- goto err;
- err = "KEYOP_LENGTH_MISMATCH";
- if (out_len != kdata->output_len)
- goto err;
- err = "KEYOP_MISMATCH";
- if (check_output(t, kdata->output, out, out_len))
- goto err;
- err = NULL;
- err:
- OPENSSL_free(out);
- t->err = err;
- return 1;
- }
- static int sign_test_init(struct evp_test *t, const char *name)
- {
- return pkey_test_init(t, name, 0, EVP_PKEY_sign_init, EVP_PKEY_sign);
- }
- static const struct evp_test_method psign_test_method = {
- "Sign",
- sign_test_init,
- pkey_test_cleanup,
- pkey_test_parse,
- pkey_test_run
- };
- static int verify_recover_test_init(struct evp_test *t, const char *name)
- {
- return pkey_test_init(t, name, 1, EVP_PKEY_verify_recover_init,
- EVP_PKEY_verify_recover);
- }
- static const struct evp_test_method pverify_recover_test_method = {
- "VerifyRecover",
- verify_recover_test_init,
- pkey_test_cleanup,
- pkey_test_parse,
- pkey_test_run
- };
- static int decrypt_test_init(struct evp_test *t, const char *name)
- {
- return pkey_test_init(t, name, 0, EVP_PKEY_decrypt_init,
- EVP_PKEY_decrypt);
- }
- static const struct evp_test_method pdecrypt_test_method = {
- "Decrypt",
- decrypt_test_init,
- pkey_test_cleanup,
- pkey_test_parse,
- pkey_test_run
- };
- static int verify_test_init(struct evp_test *t, const char *name)
- {
- return pkey_test_init(t, name, 1, EVP_PKEY_verify_init, 0);
- }
- static int verify_test_run(struct evp_test *t)
- {
- struct pkey_data *kdata = t->data;
- if (EVP_PKEY_verify(kdata->ctx, kdata->output, kdata->output_len,
- kdata->input, kdata->input_len) <= 0)
- t->err = "VERIFY_ERROR";
- return 1;
- }
- static const struct evp_test_method pverify_test_method = {
- "Verify",
- verify_test_init,
- pkey_test_cleanup,
- pkey_test_parse,
- verify_test_run
- };
|