xref: /openssl/test/uitest.c (revision 5755c11f)
1 /*
2  * Copyright 2002-2020 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 */
test_pem_password_cb(char * buf,int size,int rwflag,void * userdata)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  */
test_old(void)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 */
test_new_ui(void)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 
setup_tests(void)89 int setup_tests(void)
90 {
91     ADD_TEST(test_old);
92     ADD_TEST(test_new_ui);
93     return 1;
94 }
95