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