crypto/ui/ui_lib.c

   1 /* crypto/ui/ui_lib.c -*- mode:C; c-file-style: "eay" -*- */
   2 /* Written by Richard Levitte (levitte@stacken.kth.se) for the OpenSSL
   3  * project 2000.
   4  */
   5 /* ====================================================================
   6  * Copyright (c) 1998-2000 The OpenSSL Project.  All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  *
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in
  17  *    the documentation and/or other materials provided with the
  18  *    distribution.
  19  *
  20  * 3. All advertising materials mentioning features or use of this
  21  *    software must display the following acknowledgment:
  22  *    "This product includes software developed by the OpenSSL Project
  23  *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
  24  *
  25  * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
  26  *    endorse or promote products derived from this software without
  27  *    prior written permission. For written permission, please contact
  28  *    openssl-core@openssl.org.
  29  *
  30  * 5. Products derived from this software may not be called "OpenSSL"
  31  *    nor may "OpenSSL" appear in their names without prior written
  32  *    permission of the OpenSSL Project.
  33  *
  34  * 6. Redistributions of any form whatsoever must retain the following
  35  *    acknowledgment:
  36  *    "This product includes software developed by the OpenSSL Project
  37  *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
  38  *
  39  * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
  40  * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  41  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  42  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
  43  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  44  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  45  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  46  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  48  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  49  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  50  * OF THE POSSIBILITY OF SUCH DAMAGE.
  51  * ====================================================================
  52  *
  53  * This product includes cryptographic software written by Eric Young
  54  * (eay@cryptsoft.com).  This product includes software written by Tim
  55  * Hudson (tjh@cryptsoft.com).
  56  *
  57  */
  58
  59 #include <openssl/e_os2.h>
  60 /* The following defines enable the declaration of strdup(), which is an
  61    extended function according to X/Open. */
  62 #ifdef OPENSSL_SYS_VMS_DECC
  63 # define _XOPEN_SOURCE_EXTENDED
  64 #endif
  65 #ifdef OPENSSL_SYS_UNIX
  66 # define __USE_XOPEN_EXTENDED   /* For Linux and probably anything GNU */
  67 #endif
  68 #include <string.h>
  69
  70 #include <openssl/ui.h>
  71 #include <openssl/err.h>
  72 #include "ui_locl.h"
  73
  74 IMPLEMENT_STACK_OF(UI_STRING_ST)
  75
  76 static const UI_METHOD *default_UI_meth=NULL;
  77 static int ui_meth_num=0;
  78 static STACK_OF(CRYPTO_EX_DATA_FUNCS) *ui_meth=NULL;
  79
  80 UI *UI_new(void)
  81         {
  82         return(UI_new_method(NULL));
  83         }
  84
  85 UI *UI_new_method(const UI_METHOD *method)
  86         {
  87         UI *ret;
  88
  89         ret=(UI *)OPENSSL_malloc(sizeof(UI));
  90         if (ret == NULL)
  91                 {
  92                 UIerr(UI_F_UI_NEW_METHOD,ERR_R_MALLOC_FAILURE);
  93                 return NULL;
  94                 }
  95         if (method == NULL)
  96                 ret->meth=UI_get_default_method();
  97         else
  98                 ret->meth=method;
  99
 100         ret->strings=NULL;
 101         return ret;
 102         }
 103
 104 static void free_string(UI_STRING *uis)
 105         {
 106         if (uis->flags & OUT_STRING_FREEABLE)
 107                 OPENSSL_free((char *)uis->out_string);
 108         OPENSSL_free(uis);
 109         }
 110
 111 void UI_free(UI *ui)
 112         {
 113         sk_UI_STRING_pop_free(ui->strings,free_string);
 114         OPENSSL_free(ui);
 115         }
 116
 117 static int allocate_string_stack(UI *ui)
 118         {
 119         if (ui->strings == NULL)
 120                 {
 121                 ui->strings=sk_UI_STRING_new_null();
 122                 if (ui->strings == NULL)
 123                         {
 124                         return -1;
 125                         }
 126                 }
 127         return 0;
 128         }
 129
 130 static int general_allocate_string(UI *ui, const char *prompt,
 131         int prompt_freeable, enum UI_string_types type,
 132         char *result_buf, int minsize, int maxsize, const char *test_buf)
 133         {
 134         int ret=-1;
 135
 136         if (prompt == NULL)
 137                 {
 138                 UIerr(UI_F_GENERAL_ALLOCATE_STRING,ERR_R_PASSED_NULL_PARAMETER);
 139                 }
 140         else if (allocate_string_stack(ui) >= 0)
 141                 {
 142                 UI_STRING *s=(UI_STRING *)OPENSSL_malloc(sizeof(UI_STRING));
 143                 s->out_string=prompt;
 144                 s->flags=prompt_freeable ? OUT_STRING_FREEABLE : 0;
 145                 s->type=type;
 146                 s->result_buf=result_buf;
 147                 s->result_minsize=minsize;
 148                 s->result_maxsize=maxsize;
 149                 s->test_buf=test_buf;
 150                 ret=sk_UI_STRING_push(ui->strings, s);
 151                 }
 152         return ret;
 153         }
 154
 155 /* Returns the index to the place in the stack or 0 for error.  Uses a
 156    direct reference to the prompt.  */
 157 int UI_add_input_string(UI *ui, const char *prompt, int echo_p,
 158         char *result_buf, int minsize, int maxsize)
 159         {
 160         return general_allocate_string(ui, prompt, 0,
 161                 echo_p?UI_STRING_ECHO:UI_STRING_NOECHO,
 162                 result_buf, minsize, maxsize, NULL);
 163         }
 164
 165 /* Same as UI_add_input_string(), excepts it takes a copy of the prompt */
 166 int UI_dup_input_string(UI *ui, const char *prompt, int echo_p,
 167         char *result_buf, int minsize, int maxsize)
 168         {
 169         char *prompt_copy=NULL;
 170
 171         if (prompt)
 172                 {
 173                 prompt_copy=strdup(prompt);
 174                 if (prompt_copy == NULL)
 175                         {
 176                         UIerr(UI_F_UI_DUP_INPUT_STRING,ERR_R_MALLOC_FAILURE);
 177                         return 0;
 178                         }
 179                 }
 180
 181         return general_allocate_string(ui, prompt, 1,
 182                 echo_p?UI_STRING_ECHO:UI_STRING_NOECHO,
 183                 result_buf, minsize, maxsize, NULL);
 184         }
 185
 186 int UI_add_verify_string(UI *ui, const char *prompt, int echo_p,
 187         char *result_buf, int minsize, int maxsize, const char *test_buf)
 188         {
 189         return general_allocate_string(ui, prompt, 0,
 190                 echo_p?UI_VERIFY_ECHO:UI_VERIFY_NOECHO,
 191                 result_buf, minsize, maxsize, test_buf);
 192         }
 193
 194 int UI_dup_verify_string(UI *ui, const char *prompt, int echo_p,
 195         char *result_buf, int minsize, int maxsize, const char *test_buf)
 196         {
 197         char *prompt_copy=NULL;
 198
 199         if (prompt)
 200                 {
 201                 prompt_copy=strdup(prompt);
 202                 if (prompt_copy == NULL)
 203                         {
 204                         UIerr(UI_F_UI_DUP_VERIFY_STRING,ERR_R_MALLOC_FAILURE);
 205                         return -1;
 206                         }
 207                 }
 208
 209         return general_allocate_string(ui, prompt, 1,
 210                 echo_p?UI_VERIFY_ECHO:UI_VERIFY_NOECHO,
 211                 result_buf, minsize, maxsize, test_buf);
 212         }
 213
 214 int UI_add_info_string(UI *ui, const char *text)
 215         {
 216         return general_allocate_string(ui, text, 0, UI_INFO, NULL, 0, 0, NULL);
 217         }
 218
 219 int UI_dup_info_string(UI *ui, const char *text)
 220         {
 221         char *text_copy=NULL;
 222
 223         if (text)
 224                 {
 225                 text_copy=strdup(text);
 226                 if (text_copy == NULL)
 227                         {
 228                         UIerr(UI_F_UI_DUP_INFO_STRING,ERR_R_MALLOC_FAILURE);
 229                         return -1;
 230                         }
 231                 }
 232
 233         return general_allocate_string(ui, text, 1, UI_INFO, NULL, 0, 0, NULL);
 234         }
 235
 236 int UI_add_error_string(UI *ui, const char *text)
 237         {
 238         return general_allocate_string(ui, text, 0, UI_ERROR, NULL, 0, 0,
 239                 NULL);
 240         }
 241
 242 int UI_dup_error_string(UI *ui, const char *text)
 243         {
 244         char *text_copy=NULL;
 245
 246         if (text)
 247                 {
 248                 text_copy=strdup(text);
 249                 if (text_copy == NULL)
 250                         {
 251                         UIerr(UI_F_UI_DUP_ERROR_STRING,ERR_R_MALLOC_FAILURE);
 252                         return -1;
 253                         }
 254                 }
 255         return general_allocate_string(ui, text_copy, 1, UI_ERROR, NULL, 0, 0,
 256                 NULL);
 257         }
 258
 259 const char *UI_get0_result(UI *ui, int i)
 260         {
 261         if (i < 0)
 262                 {
 263                 UIerr(UI_F_UI_GET0_RESULT,UI_R_INDEX_TOO_SMALL);
 264                 return NULL;
 265                 }
 266         if (i >= sk_UI_STRING_num(ui->strings))
 267                 {
 268                 UIerr(UI_F_UI_GET0_RESULT,UI_R_INDEX_TOO_LARGE);
 269                 return NULL;
 270                 }
 271         return UI_get0_result_string(sk_UI_STRING_value(ui->strings, i));
 272         }
 273
 274 int UI_process(UI *ui)
 275         {
 276         int i, ok=0;
 277
 278         if (ui->meth->ui_open_session && !ui->meth->ui_open_session(ui))
 279                 return -1;
 280
 281         for(i=0; i<sk_UI_STRING_num(ui->strings); i++)
 282                 {
 283                 if (ui->meth->ui_write_string
 284                         && !ui->meth->ui_write_string(ui,
 285                                 sk_UI_STRING_value(ui->strings, i)))
 286                         {
 287                         ok=-1;
 288                         goto err;
 289                         }
 290                 }
 291
 292         for(i=0; i<sk_UI_STRING_num(ui->strings); i++)
 293                 {
 294                 if (ui->meth->ui_read_string
 295                         && !ui->meth->ui_read_string(ui,
 296                                 sk_UI_STRING_value(ui->strings, i)))
 297                         {
 298                         ok=-1;
 299                         goto err;
 300                         }
 301                 }
 302  err:
 303         if (ui->meth->ui_close_session && !ui->meth->ui_close_session(ui))
 304                 return -1;
 305         return ok;
 306         }
 307
 308 int UI_get_ex_new_index(long argl, void *argp, CRYPTO_EX_new *new_func,
 309              CRYPTO_EX_dup *dup_func, CRYPTO_EX_free *free_func)
 310         {
 311         ui_meth_num++;
 312         return(CRYPTO_get_ex_new_index(ui_meth_num-1,
 313                 &ui_meth,argl,argp,new_func,dup_func,free_func));
 314         }
 315
 316 int UI_set_ex_data(UI *r, int idx, void *arg)
 317         {
 318         return(CRYPTO_set_ex_data(&r->ex_data,idx,arg));
 319         }
 320
 321 void *UI_get_ex_data(UI *r, int idx)
 322         {
 323         return(CRYPTO_get_ex_data(&r->ex_data,idx));
 324         }
 325
 326 void UI_set_default_method(const UI_METHOD *meth)
 327         {
 328         default_UI_meth=meth;
 329         }
 330
 331 const UI_METHOD *UI_get_default_method(void)
 332         {
 333         if (default_UI_meth == NULL)
 334                 {
 335                 default_UI_meth=UI_OpenSSL();
 336                 }
 337         return default_UI_meth;
 338         }
 339
 340 const UI_METHOD *UI_get_method(UI *ui)
 341         {
 342         return ui->meth;
 343         }
 344
 345 const UI_METHOD *UI_set_method(UI *ui, const UI_METHOD *meth)
 346         {
 347         ui->meth=meth;
 348         return ui->meth;
 349         }
 350
 351
 352 UI_METHOD *UI_create_method(void)
 353         {
 354         return (UI_METHOD *)OPENSSL_malloc(sizeof(UI_METHOD));
 355         }
 356
 357 int UI_method_set_opener(UI_METHOD *method, int (*opener)(UI *ui))
 358         {
 359         if (method)
 360                 {
 361                 method->ui_open_session = opener;
 362                 return 0;
 363                 }
 364         else
 365                 return -1;
 366         }
 367
 368 int UI_method_set_writer(UI_METHOD *method, int (*writer)(UI *ui, UI_STRING *uis))
 369         {
 370         if (method)
 371                 {
 372                 method->ui_write_string = writer;
 373                 return 0;
 374                 }
 375         else
 376                 return -1;
 377         }
 378
 379 int UI_method_set_reader(UI_METHOD *method, int (*reader)(UI *ui, UI_STRING *uis))
 380         {
 381         if (method)
 382                 {
 383                 method->ui_read_string = reader;
 384                 return 0;
 385                 }
 386         else
 387                 return -1;
 388         }
 389
 390 int UI_method_set_closer(UI_METHOD *method, int (*closer)(UI *ui))
 391         {
 392         if (method)
 393                 {
 394                 method->ui_close_session = closer;
 395                 return 0;
 396                 }
 397         else
 398                 return -1;
 399         }
 400
 401 int (*UI_method_get_opener(UI_METHOD *method))(UI*)
 402         {
 403         if (method)
 404                 return method->ui_open_session;
 405         else
 406                 return NULL;
 407         }
 408
 409 int (*UI_method_get_writer(UI_METHOD *method))(UI*,UI_STRING*)
 410         {
 411         if (method)
 412                 return method->ui_write_string;
 413         else
 414                 return NULL;
 415         }
 416
 417 int (*UI_method_get_reader(UI_METHOD *method))(UI*,UI_STRING*)
 418         {
 419         if (method)
 420                 return method->ui_read_string;
 421         else
 422                 return NULL;
 423         }
 424
 425 int (*UI_method_get_closer(UI_METHOD *method))(UI*)
 426         {
 427         if (method)
 428                 return method->ui_close_session;
 429         else
 430                 return NULL;
 431         }
 432
 433 enum UI_string_types UI_get_string_type(UI_STRING *uis)
 434         {
 435         if (!uis)
 436                 return UI_NONE;
 437         return uis->type;
 438         }
 439
 440 const char *UI_get0_output_string(UI_STRING *uis)
 441         {
 442         if (!uis)
 443                 return NULL;
 444         return uis->out_string;
 445         }
 446
 447 const char *UI_get0_result_string(UI_STRING *uis)
 448         {
 449         if (!uis)
 450                 return NULL;
 451         switch(uis->type)
 452                 {
 453         case UI_STRING_ECHO:
 454         case UI_STRING_NOECHO:
 455         case UI_VERIFY_ECHO:
 456         case UI_VERIFY_NOECHO:
 457                 return uis->result_buf;
 458         default:
 459                 return NULL;
 460                 }
 461         }
 462
 463 const char *UI_get0_test_string(UI_STRING *uis)
 464         {
 465         if (!uis)
 466                 return NULL;
 467         return uis->test_buf;
 468         }
 469
 470 int UI_get_result_minsize(UI_STRING *uis)
 471         {
 472         if (!uis)
 473                 return -1;
 474         return uis->result_minsize;
 475         }
 476
 477 int UI_get_result_maxsize(UI_STRING *uis)
 478         {
 479         if (!uis)
 480                 return -1;
 481         return uis->result_maxsize;
 482         }
 483
 484 int UI_set_result(UI_STRING *uis, char *result)
 485         {
 486         int l = strlen(result);
 487
 488         if (!uis)
 489                 return -1;
 490         if (l < uis->result_minsize)
 491                 {
 492                 UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_SMALL);
 493                 return -1;
 494                 }
 495         if (l > uis->result_maxsize)
 496                 {
 497                 UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_LARGE);
 498                 return -1;
 499                 }
 500
 501         if (!uis->result_buf)
 502                 {
 503                 uis->result_buf = OPENSSL_malloc(uis->result_maxsize+1);
 504                 }
 505
 506         if (!uis->result_buf)
 507                 {
 508                 UIerr(UI_F_UI_NEW_METHOD,ERR_R_MALLOC_FAILURE);
 509                 return -1;
 510                 }
 511
 512         strcpy(uis->result_buf, result);
 513         return 0;
 514         }