Home | History | Annotate | Line # | Download | only in test 
      1 /*
      2  * Copyright 2002-2022 The OpenSSL Project Authors. All Rights Reserved.
      3  *
      4  * Licensed under the Apache License 2.0 (the "License").  You may not use
      5  * this file except in compliance with the License.  You can obtain a copy
      6  * in the file LICENSE in the source distribution or at
      7  * https://www.openssl.org/source/license.html
      8  */
      9 
     10 #include <stdio.h>
     11 #include <string.h>
     12 #include <openssl/opensslconf.h>
     13 #include <openssl/err.h>
     14 #include "apps_ui.h"
     15 #include "testutil.h"
     16 
     17 
     18 #include <openssl/ui.h>
     19 
     20 /* Old style PEM password callback */
     21 static int test_pem_password_cb(char *buf, int size, int rwflag, void *userdata)
     22 {
     23     OPENSSL_strlcpy(buf, (char *)userdata, (size_t)size);
     24     return strlen(buf);
     25 }
     26 
     27 /*
     28  * Test wrapping old style PEM password callback in a UI method through the
     29  * use of UI utility functions
     30  */
     31 static int test_old(void)
     32 {
     33     UI_METHOD *ui_method = NULL;
     34     UI *ui = NULL;
     35     char defpass[] = "password";
     36     char pass[16];
     37     int ok = 0;
     38 
     39     if (!TEST_ptr(ui_method =
     40                   UI_UTIL_wrap_read_pem_callback( test_pem_password_cb, 0))
     41             || !TEST_ptr(ui = UI_new_method(ui_method)))
     42         goto err;
     43 
     44     /* The wrapper passes the UI userdata as the callback userdata param */
     45     UI_add_user_data(ui, defpass);
     46 
     47     if (UI_add_input_string(ui, "prompt", UI_INPUT_FLAG_DEFAULT_PWD,
     48                              pass, 0, sizeof(pass) - 1) <= 0)
     49         goto err;
     50 
     51     switch (UI_process(ui)) {
     52     case -2:
     53         TEST_info("test_old: UI process interrupted or cancelled");
     54         /* fall through */
     55     case -1:
     56         goto err;
     57     default:
     58         break;
     59     }
     60 
     61     if (TEST_str_eq(pass, defpass))
     62         ok = 1;
     63 
     64  err:
     65     UI_free(ui);
     66     UI_destroy_method(ui_method);
     67 
     68     return ok;
     69 }
     70 
     71 /* Test of UI.  This uses the UI method defined in apps/apps.c */
     72 static int test_new_ui(void)
     73 {
     74     PW_CB_DATA cb_data = {
     75         "password",
     76         "prompt"
     77     };
     78     char pass[16];
     79     int ok = 0;
     80 
     81     (void)setup_ui_method();
     82     if (TEST_int_gt(password_callback(pass, sizeof(pass), 0, &cb_data), 0)
     83             && TEST_str_eq(pass, cb_data.password))
     84         ok = 1;
     85     destroy_ui_method();
     86     return ok;
     87 }
     88 
     89 int setup_tests(void)
     90 {
     91     ADD_TEST(test_old);
     92     ADD_TEST(test_new_ui);
     93     return 1;
     94 }
     95