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