Fix small UI issues
[openssl.git] / crypto / ui / ui_lib.c
1 /*
2  * Copyright 2001-2016 The OpenSSL Project Authors. All Rights Reserved.
3  *
4  * Licensed under the OpenSSL license (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 <string.h>
11 #include "internal/cryptlib.h"
12 #include <openssl/e_os2.h>
13 #include <openssl/buffer.h>
14 #include <openssl/ui.h>
15 #include <openssl/err.h>
16 #include "ui_locl.h"
17
18 UI *UI_new(void)
19 {
20     return (UI_new_method(NULL));
21 }
22
23 UI *UI_new_method(const UI_METHOD *method)
24 {
25     UI *ret = OPENSSL_zalloc(sizeof(*ret));
26
27     if (ret == NULL) {
28         UIerr(UI_F_UI_NEW_METHOD, ERR_R_MALLOC_FAILURE);
29         return NULL;
30     }
31
32     ret->lock = CRYPTO_THREAD_lock_new();
33     if (ret->lock == NULL) {
34         UIerr(UI_F_UI_NEW_METHOD, ERR_R_MALLOC_FAILURE);
35         OPENSSL_free(ret);
36         return NULL;
37     }
38
39     if (method == NULL)
40         method = UI_get_default_method();
41     if (method == NULL)
42         method = UI_null();
43     ret->meth = method;
44
45     if (!CRYPTO_new_ex_data(CRYPTO_EX_INDEX_UI, ret, &ret->ex_data)) {
46         OPENSSL_free(ret);
47         return NULL;
48     }
49     return ret;
50 }
51
52 static void free_string(UI_STRING *uis)
53 {
54     if (uis->flags & OUT_STRING_FREEABLE) {
55         OPENSSL_free((char *)uis->out_string);
56         switch (uis->type) {
57         case UIT_BOOLEAN:
58             OPENSSL_free((char *)uis->_.boolean_data.action_desc);
59             OPENSSL_free((char *)uis->_.boolean_data.ok_chars);
60             OPENSSL_free((char *)uis->_.boolean_data.cancel_chars);
61             break;
62         case UIT_NONE:
63         case UIT_PROMPT:
64         case UIT_VERIFY:
65         case UIT_ERROR:
66         case UIT_INFO:
67             break;
68         }
69     }
70     OPENSSL_free(uis);
71 }
72
73 void UI_free(UI *ui)
74 {
75     if (ui == NULL)
76         return;
77     if ((ui->flags & UI_FLAG_DUPL_DATA) != 0) {
78         ui->meth->ui_destroy_data(ui, ui->user_data);
79     }
80     sk_UI_STRING_pop_free(ui->strings, free_string);
81     CRYPTO_free_ex_data(CRYPTO_EX_INDEX_UI, ui, &ui->ex_data);
82     CRYPTO_THREAD_lock_free(ui->lock);
83     OPENSSL_free(ui);
84 }
85
86 static int allocate_string_stack(UI *ui)
87 {
88     if (ui->strings == NULL) {
89         ui->strings = sk_UI_STRING_new_null();
90         if (ui->strings == NULL) {
91             return -1;
92         }
93     }
94     return 0;
95 }
96
97 static UI_STRING *general_allocate_prompt(UI *ui, const char *prompt,
98                                           int prompt_freeable,
99                                           enum UI_string_types type,
100                                           int input_flags, char *result_buf)
101 {
102     UI_STRING *ret = NULL;
103
104     if (prompt == NULL) {
105         UIerr(UI_F_GENERAL_ALLOCATE_PROMPT, ERR_R_PASSED_NULL_PARAMETER);
106     } else if ((type == UIT_PROMPT || type == UIT_VERIFY
107                 || type == UIT_BOOLEAN) && result_buf == NULL) {
108         UIerr(UI_F_GENERAL_ALLOCATE_PROMPT, UI_R_NO_RESULT_BUFFER);
109     } else if ((ret = OPENSSL_malloc(sizeof(*ret))) != NULL) {
110         ret->out_string = prompt;
111         ret->flags = prompt_freeable ? OUT_STRING_FREEABLE : 0;
112         ret->input_flags = input_flags;
113         ret->type = type;
114         ret->result_buf = result_buf;
115     }
116     return ret;
117 }
118
119 static int general_allocate_string(UI *ui, const char *prompt,
120                                    int prompt_freeable,
121                                    enum UI_string_types type, int input_flags,
122                                    char *result_buf, int minsize, int maxsize,
123                                    const char *test_buf)
124 {
125     int ret = -1;
126     UI_STRING *s = general_allocate_prompt(ui, prompt, prompt_freeable,
127                                            type, input_flags, result_buf);
128
129     if (s != NULL) {
130         if (allocate_string_stack(ui) >= 0) {
131             s->_.string_data.result_minsize = minsize;
132             s->_.string_data.result_maxsize = maxsize;
133             s->_.string_data.test_buf = test_buf;
134             ret = sk_UI_STRING_push(ui->strings, s);
135             /* sk_push() returns 0 on error.  Let's adapt that */
136             if (ret <= 0) {
137                 ret--;
138                 free_string(s);
139             }
140         } else
141             free_string(s);
142     }
143     return ret;
144 }
145
146 static int general_allocate_boolean(UI *ui,
147                                     const char *prompt,
148                                     const char *action_desc,
149                                     const char *ok_chars,
150                                     const char *cancel_chars,
151                                     int prompt_freeable,
152                                     enum UI_string_types type,
153                                     int input_flags, char *result_buf)
154 {
155     int ret = -1;
156     UI_STRING *s;
157     const char *p;
158
159     if (ok_chars == NULL) {
160         UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN, ERR_R_PASSED_NULL_PARAMETER);
161     } else if (cancel_chars == NULL) {
162         UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN, ERR_R_PASSED_NULL_PARAMETER);
163     } else {
164         for (p = ok_chars; *p != '\0'; p++) {
165             if (strchr(cancel_chars, *p) != NULL) {
166                 UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN,
167                       UI_R_COMMON_OK_AND_CANCEL_CHARACTERS);
168             }
169         }
170
171         s = general_allocate_prompt(ui, prompt, prompt_freeable,
172                                     type, input_flags, result_buf);
173
174         if (s != NULL) {
175             if (allocate_string_stack(ui) >= 0) {
176                 s->_.boolean_data.action_desc = action_desc;
177                 s->_.boolean_data.ok_chars = ok_chars;
178                 s->_.boolean_data.cancel_chars = cancel_chars;
179                 ret = sk_UI_STRING_push(ui->strings, s);
180                 /*
181                  * sk_push() returns 0 on error. Let's adapt that
182                  */
183                 if (ret <= 0) {
184                     ret--;
185                     free_string(s);
186                 }
187             } else
188                 free_string(s);
189         }
190     }
191     return ret;
192 }
193
194 /*
195  * Returns the index to the place in the stack or -1 for error.  Uses a
196  * direct reference to the prompt.
197  */
198 int UI_add_input_string(UI *ui, const char *prompt, int flags,
199                         char *result_buf, int minsize, int maxsize)
200 {
201     return general_allocate_string(ui, prompt, 0,
202                                    UIT_PROMPT, flags, result_buf, minsize,
203                                    maxsize, NULL);
204 }
205
206 /* Same as UI_add_input_string(), excepts it takes a copy of the prompt */
207 int UI_dup_input_string(UI *ui, const char *prompt, int flags,
208                         char *result_buf, int minsize, int maxsize)
209 {
210     char *prompt_copy = NULL;
211
212     if (prompt != NULL) {
213         prompt_copy = OPENSSL_strdup(prompt);
214         if (prompt_copy == NULL) {
215             UIerr(UI_F_UI_DUP_INPUT_STRING, ERR_R_MALLOC_FAILURE);
216             return 0;
217         }
218     }
219
220     return general_allocate_string(ui, prompt_copy, 1,
221                                    UIT_PROMPT, flags, result_buf, minsize,
222                                    maxsize, NULL);
223 }
224
225 int UI_add_verify_string(UI *ui, const char *prompt, int flags,
226                          char *result_buf, int minsize, int maxsize,
227                          const char *test_buf)
228 {
229     return general_allocate_string(ui, prompt, 0,
230                                    UIT_VERIFY, flags, result_buf, minsize,
231                                    maxsize, test_buf);
232 }
233
234 int UI_dup_verify_string(UI *ui, const char *prompt, int flags,
235                          char *result_buf, int minsize, int maxsize,
236                          const char *test_buf)
237 {
238     char *prompt_copy = NULL;
239
240     if (prompt != NULL) {
241         prompt_copy = OPENSSL_strdup(prompt);
242         if (prompt_copy == NULL) {
243             UIerr(UI_F_UI_DUP_VERIFY_STRING, ERR_R_MALLOC_FAILURE);
244             return -1;
245         }
246     }
247
248     return general_allocate_string(ui, prompt_copy, 1,
249                                    UIT_VERIFY, flags, result_buf, minsize,
250                                    maxsize, test_buf);
251 }
252
253 int UI_add_input_boolean(UI *ui, const char *prompt, const char *action_desc,
254                          const char *ok_chars, const char *cancel_chars,
255                          int flags, char *result_buf)
256 {
257     return general_allocate_boolean(ui, prompt, action_desc,
258                                     ok_chars, cancel_chars, 0, UIT_BOOLEAN,
259                                     flags, result_buf);
260 }
261
262 int UI_dup_input_boolean(UI *ui, const char *prompt, const char *action_desc,
263                          const char *ok_chars, const char *cancel_chars,
264                          int flags, char *result_buf)
265 {
266     char *prompt_copy = NULL;
267     char *action_desc_copy = NULL;
268     char *ok_chars_copy = NULL;
269     char *cancel_chars_copy = NULL;
270
271     if (prompt != NULL) {
272         prompt_copy = OPENSSL_strdup(prompt);
273         if (prompt_copy == NULL) {
274             UIerr(UI_F_UI_DUP_INPUT_BOOLEAN, ERR_R_MALLOC_FAILURE);
275             goto err;
276         }
277     }
278
279     if (action_desc != NULL) {
280         action_desc_copy = OPENSSL_strdup(action_desc);
281         if (action_desc_copy == NULL) {
282             UIerr(UI_F_UI_DUP_INPUT_BOOLEAN, ERR_R_MALLOC_FAILURE);
283             goto err;
284         }
285     }
286
287     if (ok_chars != NULL) {
288         ok_chars_copy = OPENSSL_strdup(ok_chars);
289         if (ok_chars_copy == NULL) {
290             UIerr(UI_F_UI_DUP_INPUT_BOOLEAN, ERR_R_MALLOC_FAILURE);
291             goto err;
292         }
293     }
294
295     if (cancel_chars != NULL) {
296         cancel_chars_copy = OPENSSL_strdup(cancel_chars);
297         if (cancel_chars_copy == NULL) {
298             UIerr(UI_F_UI_DUP_INPUT_BOOLEAN, ERR_R_MALLOC_FAILURE);
299             goto err;
300         }
301     }
302
303     return general_allocate_boolean(ui, prompt_copy, action_desc_copy,
304                                     ok_chars_copy, cancel_chars_copy, 1,
305                                     UIT_BOOLEAN, flags, result_buf);
306  err:
307     OPENSSL_free(prompt_copy);
308     OPENSSL_free(action_desc_copy);
309     OPENSSL_free(ok_chars_copy);
310     OPENSSL_free(cancel_chars_copy);
311     return -1;
312 }
313
314 int UI_add_info_string(UI *ui, const char *text)
315 {
316     return general_allocate_string(ui, text, 0, UIT_INFO, 0, NULL, 0, 0,
317                                    NULL);
318 }
319
320 int UI_dup_info_string(UI *ui, const char *text)
321 {
322     char *text_copy = NULL;
323
324     if (text != NULL) {
325         text_copy = OPENSSL_strdup(text);
326         if (text_copy == NULL) {
327             UIerr(UI_F_UI_DUP_INFO_STRING, ERR_R_MALLOC_FAILURE);
328             return -1;
329         }
330     }
331
332     return general_allocate_string(ui, text_copy, 1, UIT_INFO, 0, NULL,
333                                    0, 0, NULL);
334 }
335
336 int UI_add_error_string(UI *ui, const char *text)
337 {
338     return general_allocate_string(ui, text, 0, UIT_ERROR, 0, NULL, 0, 0,
339                                    NULL);
340 }
341
342 int UI_dup_error_string(UI *ui, const char *text)
343 {
344     char *text_copy = NULL;
345
346     if (text != NULL) {
347         text_copy = OPENSSL_strdup(text);
348         if (text_copy == NULL) {
349             UIerr(UI_F_UI_DUP_ERROR_STRING, ERR_R_MALLOC_FAILURE);
350             return -1;
351         }
352     }
353     return general_allocate_string(ui, text_copy, 1, UIT_ERROR, 0, NULL,
354                                    0, 0, NULL);
355 }
356
357 char *UI_construct_prompt(UI *ui, const char *object_desc,
358                           const char *object_name)
359 {
360     char *prompt = NULL;
361
362     if (ui->meth->ui_construct_prompt != NULL)
363         prompt = ui->meth->ui_construct_prompt(ui, object_desc, object_name);
364     else {
365         char prompt1[] = "Enter ";
366         char prompt2[] = " for ";
367         char prompt3[] = ":";
368         int len = 0;
369
370         if (object_desc == NULL)
371             return NULL;
372         len = sizeof(prompt1) - 1 + strlen(object_desc);
373         if (object_name != NULL)
374             len += sizeof(prompt2) - 1 + strlen(object_name);
375         len += sizeof(prompt3) - 1;
376
377         prompt = OPENSSL_malloc(len + 1);
378         if (prompt == NULL)
379             return NULL;
380         OPENSSL_strlcpy(prompt, prompt1, len + 1);
381         OPENSSL_strlcat(prompt, object_desc, len + 1);
382         if (object_name != NULL) {
383             OPENSSL_strlcat(prompt, prompt2, len + 1);
384             OPENSSL_strlcat(prompt, object_name, len + 1);
385         }
386         OPENSSL_strlcat(prompt, prompt3, len + 1);
387     }
388     return prompt;
389 }
390
391 void *UI_add_user_data(UI *ui, void *user_data)
392 {
393     void *old_data = ui->user_data;
394
395     if ((ui->flags & UI_FLAG_DUPL_DATA) != 0) {
396         ui->meth->ui_destroy_data(ui, old_data);
397         old_data = NULL;
398     }
399     ui->user_data = user_data;
400     ui->flags &= ~UI_FLAG_DUPL_DATA;
401     return old_data;
402 }
403
404 int UI_dup_user_data(UI *ui, void *user_data)
405 {
406     void *duplicate = NULL;
407
408     if (ui->meth->ui_duplicate_data == NULL
409         || ui->meth->ui_destroy_data == NULL) {
410         UIerr(UI_F_UI_DUP_USER_DATA, UI_R_USER_DATA_DUPLICATION_UNSUPPORTED);
411         return -1;
412     }
413
414     duplicate = ui->meth->ui_duplicate_data(ui, user_data);
415     if (duplicate == NULL) {
416         UIerr(UI_F_UI_DUP_USER_DATA, ERR_R_MALLOC_FAILURE);
417         return -1;
418     }
419
420     (void)UI_add_user_data(ui, duplicate);
421     ui->flags |= UI_FLAG_DUPL_DATA;
422
423     return 0;
424 }
425
426 void *UI_get0_user_data(UI *ui)
427 {
428     return ui->user_data;
429 }
430
431 const char *UI_get0_result(UI *ui, int i)
432 {
433     if (i < 0) {
434         UIerr(UI_F_UI_GET0_RESULT, UI_R_INDEX_TOO_SMALL);
435         return NULL;
436     }
437     if (i >= sk_UI_STRING_num(ui->strings)) {
438         UIerr(UI_F_UI_GET0_RESULT, UI_R_INDEX_TOO_LARGE);
439         return NULL;
440     }
441     return UI_get0_result_string(sk_UI_STRING_value(ui->strings, i));
442 }
443
444 static int print_error(const char *str, size_t len, UI *ui)
445 {
446     UI_STRING uis;
447
448     memset(&uis, 0, sizeof(uis));
449     uis.type = UIT_ERROR;
450     uis.out_string = str;
451
452     if (ui->meth->ui_write_string != NULL
453         && ui->meth->ui_write_string(ui, &uis) <= 0)
454         return -1;
455     return 0;
456 }
457
458 int UI_process(UI *ui)
459 {
460     int i, ok = 0;
461     const char *state = "processing";
462
463     if (ui->meth->ui_open_session != NULL
464         && ui->meth->ui_open_session(ui) <= 0) {
465         state = "opening session";
466         ok = -1;
467         goto err;
468     }
469
470     if (ui->flags & UI_FLAG_PRINT_ERRORS)
471         ERR_print_errors_cb((int (*)(const char *, size_t, void *))
472                             print_error, (void *)ui);
473
474     for (i = 0; i < sk_UI_STRING_num(ui->strings); i++) {
475         if (ui->meth->ui_write_string != NULL
476             && (ui->meth->ui_write_string(ui,
477                                           sk_UI_STRING_value(ui->strings, i))
478                 <= 0))
479         {
480             state = "writing strings";
481             ok = -1;
482             goto err;
483         }
484     }
485
486     if (ui->meth->ui_flush != NULL)
487         switch (ui->meth->ui_flush(ui)) {
488         case -1:               /* Interrupt/Cancel/something... */
489             ok = -2;
490             goto err;
491         case 0:                /* Errors */
492             state = "flushing";
493             ok = -1;
494             goto err;
495         default:               /* Success */
496             ok = 0;
497             break;
498         }
499
500     for (i = 0; i < sk_UI_STRING_num(ui->strings); i++) {
501         if (ui->meth->ui_read_string != NULL) {
502             switch (ui->meth->ui_read_string(ui,
503                                              sk_UI_STRING_value(ui->strings,
504                                                                 i))) {
505             case -1:           /* Interrupt/Cancel/something... */
506                 ok = -2;
507                 goto err;
508             case 0:            /* Errors */
509                 state = "reading strings";
510                 ok = -1;
511                 goto err;
512             default:           /* Success */
513                 ok = 0;
514                 break;
515             }
516         }
517     }
518
519     state = NULL;
520  err:
521     if (ui->meth->ui_close_session != NULL
522         && ui->meth->ui_close_session(ui) <= 0) {
523         if (state == NULL)
524             state = "closing session";
525         ok = -1;
526     }
527
528     if (ok == -1) {
529         UIerr(UI_F_UI_PROCESS, UI_R_PROCESSING_ERROR);
530         ERR_add_error_data(2, "while ", state);
531     }
532     return ok;
533 }
534
535 int UI_ctrl(UI *ui, int cmd, long i, void *p, void (*f) (void))
536 {
537     if (ui == NULL) {
538         UIerr(UI_F_UI_CTRL, ERR_R_PASSED_NULL_PARAMETER);
539         return -1;
540     }
541     switch (cmd) {
542     case UI_CTRL_PRINT_ERRORS:
543         {
544             int save_flag = ! !(ui->flags & UI_FLAG_PRINT_ERRORS);
545             if (i)
546                 ui->flags |= UI_FLAG_PRINT_ERRORS;
547             else
548                 ui->flags &= ~UI_FLAG_PRINT_ERRORS;
549             return save_flag;
550         }
551     case UI_CTRL_IS_REDOABLE:
552         return ! !(ui->flags & UI_FLAG_REDOABLE);
553     default:
554         break;
555     }
556     UIerr(UI_F_UI_CTRL, UI_R_UNKNOWN_CONTROL_COMMAND);
557     return -1;
558 }
559
560 int UI_set_ex_data(UI *r, int idx, void *arg)
561 {
562     return (CRYPTO_set_ex_data(&r->ex_data, idx, arg));
563 }
564
565 void *UI_get_ex_data(UI *r, int idx)
566 {
567     return (CRYPTO_get_ex_data(&r->ex_data, idx));
568 }
569
570 const UI_METHOD *UI_get_method(UI *ui)
571 {
572     return ui->meth;
573 }
574
575 const UI_METHOD *UI_set_method(UI *ui, const UI_METHOD *meth)
576 {
577     ui->meth = meth;
578     return ui->meth;
579 }
580
581 UI_METHOD *UI_create_method(const char *name)
582 {
583     UI_METHOD *ui_method = NULL;
584
585     if ((ui_method = OPENSSL_zalloc(sizeof(*ui_method))) == NULL
586         || (ui_method->name = OPENSSL_strdup(name)) == NULL
587         || !CRYPTO_new_ex_data(CRYPTO_EX_INDEX_UI_METHOD, ui_method,
588                                &ui_method->ex_data)) {
589         if (ui_method)
590             OPENSSL_free(ui_method->name);
591         OPENSSL_free(ui_method);
592         UIerr(UI_F_UI_CREATE_METHOD, ERR_R_MALLOC_FAILURE);
593         return NULL;
594     }
595     return ui_method;
596 }
597
598 /*
599  * BIG FSCKING WARNING!!!! If you use this on a statically allocated method
600  * (that is, it hasn't been allocated using UI_create_method(), you deserve
601  * anything Murphy can throw at you and more! You have been warned.
602  */
603 void UI_destroy_method(UI_METHOD *ui_method)
604 {
605     if (ui_method == NULL)
606         return;
607     CRYPTO_free_ex_data(CRYPTO_EX_INDEX_UI_METHOD, ui_method,
608                         &ui_method->ex_data);
609     OPENSSL_free(ui_method->name);
610     ui_method->name = NULL;
611     OPENSSL_free(ui_method);
612 }
613
614 int UI_method_set_opener(UI_METHOD *method, int (*opener) (UI *ui))
615 {
616     if (method != NULL) {
617         method->ui_open_session = opener;
618         return 0;
619     }
620     return -1;
621 }
622
623 int UI_method_set_writer(UI_METHOD *method,
624                          int (*writer) (UI *ui, UI_STRING *uis))
625 {
626     if (method != NULL) {
627         method->ui_write_string = writer;
628         return 0;
629     }
630     return -1;
631 }
632
633 int UI_method_set_flusher(UI_METHOD *method, int (*flusher) (UI *ui))
634 {
635     if (method != NULL) {
636         method->ui_flush = flusher;
637         return 0;
638     }
639     return -1;
640 }
641
642 int UI_method_set_reader(UI_METHOD *method,
643                          int (*reader) (UI *ui, UI_STRING *uis))
644 {
645     if (method != NULL) {
646         method->ui_read_string = reader;
647         return 0;
648     }
649     return -1;
650 }
651
652 int UI_method_set_closer(UI_METHOD *method, int (*closer) (UI *ui))
653 {
654     if (method != NULL) {
655         method->ui_close_session = closer;
656         return 0;
657     }
658     return -1;
659 }
660
661 int UI_method_set_data_duplicator(UI_METHOD *method,
662                                   void *(*duplicator) (UI *ui, void *ui_data),
663                                   void (*destructor)(UI *ui, void *ui_data))
664 {
665     if (method != NULL) {
666         method->ui_duplicate_data = duplicator;
667         method->ui_destroy_data = destructor;
668         return 0;
669     }
670     return -1;
671 }
672
673 int UI_method_set_prompt_constructor(UI_METHOD *method,
674                                      char *(*prompt_constructor) (UI *ui,
675                                                                   const char
676                                                                   *object_desc,
677                                                                   const char
678                                                                   *object_name))
679 {
680     if (method != NULL) {
681         method->ui_construct_prompt = prompt_constructor;
682         return 0;
683     }
684     return -1;
685 }
686
687 int UI_method_set_ex_data(UI_METHOD *method, int idx, void *data)
688 {
689     return CRYPTO_set_ex_data(&method->ex_data, idx, data);
690 }
691
692 int (*UI_method_get_opener(const UI_METHOD *method)) (UI *)
693 {
694     if (method != NULL)
695         return method->ui_open_session;
696     return NULL;
697 }
698
699 int (*UI_method_get_writer(const UI_METHOD *method)) (UI *, UI_STRING *)
700 {
701     if (method != NULL)
702         return method->ui_write_string;
703     return NULL;
704 }
705
706 int (*UI_method_get_flusher(const UI_METHOD *method)) (UI *)
707 {
708     if (method != NULL)
709         return method->ui_flush;
710     return NULL;
711 }
712
713 int (*UI_method_get_reader(const UI_METHOD *method)) (UI *, UI_STRING *)
714 {
715     if (method != NULL)
716         return method->ui_read_string;
717     return NULL;
718 }
719
720 int (*UI_method_get_closer(const UI_METHOD *method)) (UI *)
721 {
722     if (method != NULL)
723         return method->ui_close_session;
724     return NULL;
725 }
726
727 char *(*UI_method_get_prompt_constructor(const UI_METHOD *method))
728     (UI *, const char *, const char *)
729 {
730     if (method != NULL)
731         return method->ui_construct_prompt;
732     return NULL;
733 }
734
735 void *(*UI_method_get_data_duplicator(const UI_METHOD *method)) (UI *, void *)
736 {
737     if (method != NULL)
738         return method->ui_duplicate_data;
739     return NULL;
740 }
741
742 void (*UI_method_get_data_destructor(const UI_METHOD *method)) (UI *, void *)
743 {
744     if (method != NULL)
745         return method->ui_destroy_data;
746     return NULL;
747 }
748
749 const void *UI_method_get_ex_data(const UI_METHOD *method, int idx)
750 {
751     return CRYPTO_get_ex_data(&method->ex_data, idx);
752 }
753
754 enum UI_string_types UI_get_string_type(UI_STRING *uis)
755 {
756     return uis->type;
757 }
758
759 int UI_get_input_flags(UI_STRING *uis)
760 {
761     return uis->input_flags;
762 }
763
764 const char *UI_get0_output_string(UI_STRING *uis)
765 {
766     return uis->out_string;
767 }
768
769 const char *UI_get0_action_string(UI_STRING *uis)
770 {
771     switch (uis->type) {
772     case UIT_BOOLEAN:
773         return uis->_.boolean_data.action_desc;
774     case UIT_PROMPT:
775     case UIT_NONE:
776     case UIT_VERIFY:
777     case UIT_INFO:
778     case UIT_ERROR:
779         break;
780     }
781     return NULL;
782 }
783
784 const char *UI_get0_result_string(UI_STRING *uis)
785 {
786     switch (uis->type) {
787     case UIT_PROMPT:
788     case UIT_VERIFY:
789         return uis->result_buf;
790     case UIT_NONE:
791     case UIT_BOOLEAN:
792     case UIT_INFO:
793     case UIT_ERROR:
794         break;
795     }
796     return NULL;
797 }
798
799 const char *UI_get0_test_string(UI_STRING *uis)
800 {
801     switch (uis->type) {
802     case UIT_VERIFY:
803         return uis->_.string_data.test_buf;
804     case UIT_NONE:
805     case UIT_BOOLEAN:
806     case UIT_INFO:
807     case UIT_ERROR:
808     case UIT_PROMPT:
809         break;
810     }
811     return NULL;
812 }
813
814 int UI_get_result_minsize(UI_STRING *uis)
815 {
816     switch (uis->type) {
817     case UIT_PROMPT:
818     case UIT_VERIFY:
819         return uis->_.string_data.result_minsize;
820     case UIT_NONE:
821     case UIT_INFO:
822     case UIT_ERROR:
823     case UIT_BOOLEAN:
824         break;
825     }
826     return -1;
827 }
828
829 int UI_get_result_maxsize(UI_STRING *uis)
830 {
831     switch (uis->type) {
832     case UIT_PROMPT:
833     case UIT_VERIFY:
834         return uis->_.string_data.result_maxsize;
835     case UIT_NONE:
836     case UIT_INFO:
837     case UIT_ERROR:
838     case UIT_BOOLEAN:
839         break;
840     }
841     return -1;
842 }
843
844 int UI_set_result(UI *ui, UI_STRING *uis, const char *result)
845 {
846     int l = strlen(result);
847
848     ui->flags &= ~UI_FLAG_REDOABLE;
849
850     switch (uis->type) {
851     case UIT_PROMPT:
852     case UIT_VERIFY:
853         {
854             char number1[DECIMAL_SIZE(uis->_.string_data.result_minsize) + 1];
855             char number2[DECIMAL_SIZE(uis->_.string_data.result_maxsize) + 1];
856
857             BIO_snprintf(number1, sizeof(number1), "%d",
858                          uis->_.string_data.result_minsize);
859             BIO_snprintf(number2, sizeof(number2), "%d",
860                          uis->_.string_data.result_maxsize);
861
862             if (l < uis->_.string_data.result_minsize) {
863                 ui->flags |= UI_FLAG_REDOABLE;
864                 UIerr(UI_F_UI_SET_RESULT, UI_R_RESULT_TOO_SMALL);
865                 ERR_add_error_data(5, "You must type in ",
866                                    number1, " to ", number2, " characters");
867                 return -1;
868             }
869             if (l > uis->_.string_data.result_maxsize) {
870                 ui->flags |= UI_FLAG_REDOABLE;
871                 UIerr(UI_F_UI_SET_RESULT, UI_R_RESULT_TOO_LARGE);
872                 ERR_add_error_data(5, "You must type in ",
873                                    number1, " to ", number2, " characters");
874                 return -1;
875             }
876         }
877
878         if (uis->result_buf == NULL) {
879             UIerr(UI_F_UI_SET_RESULT, UI_R_NO_RESULT_BUFFER);
880             return -1;
881         }
882
883         OPENSSL_strlcpy(uis->result_buf, result,
884                     uis->_.string_data.result_maxsize + 1);
885         break;
886     case UIT_BOOLEAN:
887         {
888             const char *p;
889
890             if (uis->result_buf == NULL) {
891                 UIerr(UI_F_UI_SET_RESULT, UI_R_NO_RESULT_BUFFER);
892                 return -1;
893             }
894
895             uis->result_buf[0] = '\0';
896             for (p = result; *p; p++) {
897                 if (strchr(uis->_.boolean_data.ok_chars, *p)) {
898                     uis->result_buf[0] = uis->_.boolean_data.ok_chars[0];
899                     break;
900                 }
901                 if (strchr(uis->_.boolean_data.cancel_chars, *p)) {
902                     uis->result_buf[0] = uis->_.boolean_data.cancel_chars[0];
903                     break;
904                 }
905             }
906         }
907     case UIT_NONE:
908     case UIT_INFO:
909     case UIT_ERROR:
910         break;
911     }
912     return 0;
913 }