Make sure we don't return 0 on error.
[openssl.git] / crypto / ui / ui_lib.c
index 132236fe65095edaa3fce4e18328ee1ac54f4268..932658e2b711ffc4297ab7f296550ec3662d93b0 100644 (file)
@@ -57,7 +57,8 @@
  */
 
 #include <string.h>
-
+#include <openssl/e_os2.h>
+#include <openssl/buffer.h>
 #include <openssl/ui.h>
 #include <openssl/err.h>
 #include "ui_locl.h"
@@ -89,19 +90,34 @@ UI *UI_new_method(const UI_METHOD *method)
                ret->meth=method;
 
        ret->strings=NULL;
+       ret->user_data=NULL;
+       CRYPTO_new_ex_data(ui_meth,ret,&ret->ex_data);
        return ret;
        }
 
-static void free_string(void *data)
+static void free_string(UI_STRING *uis)
        {
-       UI_STRING *uis = (UI_STRING *)data;
        if (uis->flags & OUT_STRING_FREEABLE)
+               {
                OPENSSL_free((char *)uis->out_string);
+               switch(uis->type)
+                       {
+               case UIT_BOOLEAN:
+                       OPENSSL_free((char *)uis->_.boolean_data.action_desc);
+                       OPENSSL_free((char *)uis->_.boolean_data.ok_chars);
+                       OPENSSL_free((char *)uis->_.boolean_data.cancel_chars);
+                       break;
+               default:
+                       break;
+                       }
+               }
        OPENSSL_free(uis);
        }
 
 void UI_free(UI *ui)
        {
+       if (ui == NULL)
+               return;
        sk_UI_STRING_pop_free(ui->strings,free_string);
        OPENSSL_free(ui);
        }
@@ -119,50 +135,125 @@ static int allocate_string_stack(UI *ui)
        return 0;
        }
 
+static UI_STRING *general_allocate_prompt(UI *ui, const char *prompt,
+       int prompt_freeable, enum UI_string_types type, int input_flags,
+       char *result_buf)
+       {
+       UI_STRING *ret = NULL;
+
+       if (prompt == NULL)
+               {
+               UIerr(UI_F_GENERAL_ALLOCATE_PROMPT,ERR_R_PASSED_NULL_PARAMETER);
+               }
+       else if (result_buf == NULL)
+               {
+               UIerr(UI_F_GENERAL_ALLOCATE_PROMPT,UI_R_NO_RESULT_BUFFER);
+               }
+       else if ((ret = (UI_STRING *)OPENSSL_malloc(sizeof(UI_STRING))))
+               {
+               ret->out_string=prompt;
+               ret->flags=prompt_freeable ? OUT_STRING_FREEABLE : 0;
+               ret->input_flags=input_flags;
+               ret->type=type;
+               ret->result_buf=result_buf;
+               }
+       return ret;
+       }
+
 static int general_allocate_string(UI *ui, const char *prompt,
-       int prompt_freeable, enum UI_string_types type,
+       int prompt_freeable, enum UI_string_types type, int input_flags,
        char *result_buf, int minsize, int maxsize, const char *test_buf)
        {
-       int ret=-1;
+       int ret = -1;
+       UI_STRING *s = general_allocate_prompt(ui, prompt, prompt_freeable,
+               type, input_flags, result_buf);
 
-       if (prompt == NULL)
+       if (s)
                {
-               UIerr(UI_F_GENERAL_ALLOCATE_STRING,ERR_R_PASSED_NULL_PARAMETER);
+               if (allocate_string_stack(ui) >= 0)
+                       {
+                       s->_.string_data.result_minsize=minsize;
+                       s->_.string_data.result_maxsize=maxsize;
+                       s->_.string_data.test_buf=test_buf;
+                       ret=sk_UI_STRING_push(ui->strings, s);
+                       /* sk_push() returns 0 on error.  Let's addapt that */
+                       if (ret <= 0) ret--;
+                       }
+               else
+                       free_string(s);
                }
-       else if (allocate_string_stack(ui) >= 0)
+       return ret;
+       }
+
+static int general_allocate_boolean(UI *ui,
+       const char *prompt, const char *action_desc,
+       const char *ok_chars, const char *cancel_chars,
+       int prompt_freeable, enum UI_string_types type, int input_flags,
+       char *result_buf)
+       {
+       int ret = -1;
+       UI_STRING *s;
+       const char *p;
+
+       if (ok_chars == NULL)
+               {
+               UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN,ERR_R_PASSED_NULL_PARAMETER);
+               }
+       else if (cancel_chars == NULL)
+               {
+               UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN,ERR_R_PASSED_NULL_PARAMETER);
+               }
+       else
                {
-               UI_STRING *s=(UI_STRING *)OPENSSL_malloc(sizeof(UI_STRING));
-               s->out_string=prompt;
-               s->flags=prompt_freeable ? OUT_STRING_FREEABLE : 0;
-               s->type=type;
-               s->result_buf=result_buf;
-               s->result_minsize=minsize;
-               s->result_maxsize=maxsize;
-               s->test_buf=test_buf;
-               ret=sk_UI_STRING_push(ui->strings, s);
+               for(p = ok_chars; *p; p++)
+                       {
+                       if (strchr(cancel_chars, *p))
+                               {
+                               UIerr(UI_F_GENERAL_ALLOCATE_BOOLEAN,
+                                       UI_R_COMMON_OK_AND_CANCEL_CHARACTERS);
+                               }
+                       }
+
+               s = general_allocate_prompt(ui, prompt, prompt_freeable,
+                       type, input_flags, result_buf);
+
+               if (s)
+                       {
+                       if (allocate_string_stack(ui) >= 0)
+                               {
+                               s->_.boolean_data.action_desc = action_desc;
+                               s->_.boolean_data.ok_chars = ok_chars;
+                               s->_.boolean_data.cancel_chars = cancel_chars;
+                               ret=sk_UI_STRING_push(ui->strings, s);
+                               /* sk_push() returns 0 on error.
+                                  Let's addapt that */
+                               if (ret <= 0) ret--;
+                               }
+                       else
+                               free_string(s);
+                       }
                }
        return ret;
        }
 
 /* Returns the index to the place in the stack or 0 for error.  Uses a
    direct reference to the prompt.  */
-int UI_add_input_string(UI *ui, const char *prompt, int echo_p,
+int UI_add_input_string(UI *ui, const char *prompt, int flags,
        char *result_buf, int minsize, int maxsize)
        {
        return general_allocate_string(ui, prompt, 0,
-               echo_p?UI_STRING_ECHO:UI_STRING_NOECHO,
-               result_buf, minsize, maxsize, NULL);
+               UIT_PROMPT, flags, result_buf, minsize, maxsize, NULL);
        }
 
 /* Same as UI_add_input_string(), excepts it takes a copy of the prompt */
-int UI_dup_input_string(UI *ui, const char *prompt, int echo_p,
+int UI_dup_input_string(UI *ui, const char *prompt, int flags,
        char *result_buf, int minsize, int maxsize)
        {
        char *prompt_copy=NULL;
 
        if (prompt)
                {
-               prompt_copy=strdup(prompt);
+               prompt_copy=BUF_strdup(prompt);
                if (prompt_copy == NULL)
                        {
                        UIerr(UI_F_UI_DUP_INPUT_STRING,ERR_R_MALLOC_FAILURE);
@@ -170,27 +261,25 @@ int UI_dup_input_string(UI *ui, const char *prompt, int echo_p,
                        }
                }
        
-       return general_allocate_string(ui, prompt, 1,
-               echo_p?UI_STRING_ECHO:UI_STRING_NOECHO,
-               result_buf, minsize, maxsize, NULL);
+       return general_allocate_string(ui, prompt_copy, 1,
+               UIT_PROMPT, flags, result_buf, minsize, maxsize, NULL);
        }
 
-int UI_add_verify_string(UI *ui, const char *prompt, int echo_p,
+int UI_add_verify_string(UI *ui, const char *prompt, int flags,
        char *result_buf, int minsize, int maxsize, const char *test_buf)
        {
        return general_allocate_string(ui, prompt, 0,
-               echo_p?UI_VERIFY_ECHO:UI_VERIFY_NOECHO,
-               result_buf, minsize, maxsize, test_buf);
+               UIT_VERIFY, flags, result_buf, minsize, maxsize, test_buf);
        }
 
-int UI_dup_verify_string(UI *ui, const char *prompt, int echo_p,
+int UI_dup_verify_string(UI *ui, const char *prompt, int flags,
        char *result_buf, int minsize, int maxsize, const char *test_buf)
        {
        char *prompt_copy=NULL;
 
        if (prompt)
                {
-               prompt_copy=strdup(prompt);
+               prompt_copy=BUF_strdup(prompt);
                if (prompt_copy == NULL)
                        {
                        UIerr(UI_F_UI_DUP_VERIFY_STRING,ERR_R_MALLOC_FAILURE);
@@ -198,14 +287,82 @@ int UI_dup_verify_string(UI *ui, const char *prompt, int echo_p,
                        }
                }
        
-       return general_allocate_string(ui, prompt, 1,
-               echo_p?UI_VERIFY_ECHO:UI_VERIFY_NOECHO,
-               result_buf, minsize, maxsize, test_buf);
+       return general_allocate_string(ui, prompt_copy, 1,
+               UIT_VERIFY, flags, result_buf, minsize, maxsize, test_buf);
+       }
+
+int UI_add_input_boolean(UI *ui, const char *prompt, const char *action_desc,
+       const char *ok_chars, const char *cancel_chars,
+       int flags, char *result_buf)
+       {
+       return general_allocate_boolean(ui, prompt, action_desc,
+               ok_chars, cancel_chars, 0, UIT_BOOLEAN, flags, result_buf);
+       }
+
+int UI_dup_input_boolean(UI *ui, const char *prompt, const char *action_desc,
+       const char *ok_chars, const char *cancel_chars,
+       int flags, char *result_buf)
+       {
+       char *prompt_copy = NULL;
+       char *action_desc_copy = NULL;
+       char *ok_chars_copy = NULL;
+       char *cancel_chars_copy = NULL;
+
+       if (prompt)
+               {
+               prompt_copy=BUF_strdup(prompt);
+               if (prompt_copy == NULL)
+                       {
+                       UIerr(UI_F_UI_DUP_INPUT_BOOLEAN,ERR_R_MALLOC_FAILURE);
+                       goto err;
+                       }
+               }
+       
+       if (action_desc)
+               {
+               action_desc_copy=BUF_strdup(action_desc);
+               if (action_desc_copy == NULL)
+                       {
+                       UIerr(UI_F_UI_DUP_INPUT_BOOLEAN,ERR_R_MALLOC_FAILURE);
+                       goto err;
+                       }
+               }
+       
+       if (ok_chars)
+               {
+               ok_chars_copy=BUF_strdup(ok_chars);
+               if (ok_chars_copy == NULL)
+                       {
+                       UIerr(UI_F_UI_DUP_INPUT_BOOLEAN,ERR_R_MALLOC_FAILURE);
+                       goto err;
+                       }
+               }
+       
+       if (cancel_chars)
+               {
+               cancel_chars_copy=BUF_strdup(cancel_chars);
+               if (cancel_chars_copy == NULL)
+                       {
+                       UIerr(UI_F_UI_DUP_INPUT_BOOLEAN,ERR_R_MALLOC_FAILURE);
+                       goto err;
+                       }
+               }
+       
+       return general_allocate_boolean(ui, prompt_copy, action_desc_copy,
+               ok_chars_copy, cancel_chars_copy, 1, UIT_BOOLEAN, flags,
+               result_buf);
+ err:
+       if (prompt_copy) OPENSSL_free(prompt_copy);
+       if (action_desc_copy) OPENSSL_free(action_desc_copy);
+       if (ok_chars_copy) OPENSSL_free(ok_chars_copy);
+       if (cancel_chars_copy) OPENSSL_free(cancel_chars_copy);
+       return -1;
        }
 
 int UI_add_info_string(UI *ui, const char *text)
        {
-       return general_allocate_string(ui, text, 0, UI_INFO, NULL, 0, 0, NULL);
+       return general_allocate_string(ui, text, 0, UIT_INFO, 0, NULL, 0, 0,
+               NULL);
        }
 
 int UI_dup_info_string(UI *ui, const char *text)
@@ -214,7 +371,7 @@ int UI_dup_info_string(UI *ui, const char *text)
 
        if (text)
                {
-               text_copy=strdup(text);
+               text_copy=BUF_strdup(text);
                if (text_copy == NULL)
                        {
                        UIerr(UI_F_UI_DUP_INFO_STRING,ERR_R_MALLOC_FAILURE);
@@ -222,12 +379,13 @@ int UI_dup_info_string(UI *ui, const char *text)
                        }
                }
 
-       return general_allocate_string(ui, text, 1, UI_INFO, NULL, 0, 0, NULL);
+       return general_allocate_string(ui, text_copy, 1, UIT_INFO, 0, NULL,
+               0, 0, NULL);
        }
 
 int UI_add_error_string(UI *ui, const char *text)
        {
-       return general_allocate_string(ui, text, 0, UI_ERROR, NULL, 0, 0,
+       return general_allocate_string(ui, text, 0, UIT_ERROR, 0, NULL, 0, 0,
                NULL);
        }
 
@@ -237,15 +395,62 @@ int UI_dup_error_string(UI *ui, const char *text)
 
        if (text)
                {
-               text_copy=strdup(text);
+               text_copy=BUF_strdup(text);
                if (text_copy == NULL)
                        {
                        UIerr(UI_F_UI_DUP_ERROR_STRING,ERR_R_MALLOC_FAILURE);
                        return -1;
                        }
                }
-       return general_allocate_string(ui, text_copy, 1, UI_ERROR, NULL, 0, 0,
-               NULL);
+       return general_allocate_string(ui, text_copy, 1, UIT_ERROR, 0, NULL,
+               0, 0, NULL);
+       }
+
+char *UI_construct_prompt(UI *ui, const char *object_desc,
+       const char *object_name)
+       {
+       char *prompt = NULL;
+
+       if (ui->meth->ui_construct_prompt)
+               prompt = ui->meth->ui_construct_prompt(ui,
+                       object_desc, object_name);
+       else
+               {
+               char prompt1[] = "Enter ";
+               char prompt2[] = " for ";
+               char prompt3[] = ":";
+               int len = 0;
+
+               if (object_desc == NULL)
+                       return NULL;
+               len = sizeof(prompt1) - 1 + strlen(object_desc);
+               if (object_name)
+                       len += sizeof(prompt2) - 1 + strlen(object_name);
+               len += sizeof(prompt3) - 1;
+
+               prompt = (char *)OPENSSL_malloc(len + 1);
+               strcpy(prompt, prompt1);
+               strcat(prompt, object_desc);
+               if (object_name)
+                       {
+                       strcat(prompt, prompt2);
+                       strcat(prompt, object_name);
+                       }
+               strcat(prompt, prompt3);
+               }
+       return prompt;
+       }
+
+void *UI_add_user_data(UI *ui, void *user_data)
+       {
+       void *old_data = ui->user_data;
+       ui->user_data = user_data;
+       return old_data;
+       }
+
+void *UI_get0_user_data(UI *ui)
+       {
+       return ui->user_data;
        }
 
 const char *UI_get0_result(UI *ui, int i)
@@ -263,6 +468,20 @@ const char *UI_get0_result(UI *ui, int i)
        return UI_get0_result_string(sk_UI_STRING_value(ui->strings, i));
        }
 
+static int print_error(const char *str, size_t len, UI *ui)
+       {
+       UI_STRING uis;
+
+       memset(&uis, 0, sizeof(uis));
+       uis.type = UIT_ERROR;
+       uis.out_string = str;
+
+       if (ui->meth->ui_write_string
+               && !ui->meth->ui_write_string(ui, &uis))
+               return -1;
+       return 0;
+       }
+
 int UI_process(UI *ui)
        {
        int i, ok=0;
@@ -270,6 +489,11 @@ int UI_process(UI *ui)
        if (ui->meth->ui_open_session && !ui->meth->ui_open_session(ui))
                return -1;
 
+       if (ui->flags & UI_FLAG_PRINT_ERRORS)
+               ERR_print_errors_cb(
+                       (int (*)(const char *, size_t, void *))print_error,
+                       (void *)ui);
+
        for(i=0; i<sk_UI_STRING_num(ui->strings); i++)
                {
                if (ui->meth->ui_write_string
@@ -281,14 +505,37 @@ int UI_process(UI *ui)
                        }
                }
 
+       if (ui->meth->ui_flush)
+               switch(ui->meth->ui_flush(ui))
+                       {
+               case -1: /* Interrupt/Cancel/something... */
+                       ok = -2;
+                       goto err;
+               case 0: /* Errors */
+                       ok = -1;
+                       goto err;
+               default: /* Success */
+                       ok = 0;
+                       break;
+                       }
+
        for(i=0; i<sk_UI_STRING_num(ui->strings); i++)
                {
-               if (ui->meth->ui_read_string
-                       && !ui->meth->ui_read_string(ui,
-                               sk_UI_STRING_value(ui->strings, i)))
+               if (ui->meth->ui_read_string)
                        {
-                       ok=-1;
-                       goto err;
+                       switch(ui->meth->ui_read_string(ui,
+                               sk_UI_STRING_value(ui->strings, i)))
+                               {
+                       case -1: /* Interrupt/Cancel/something... */
+                               ok = -2;
+                               goto err;
+                       case 0: /* Errors */
+                               ok = -1;
+                               goto err;
+                       default: /* Success */
+                               ok = 0;
+                               break;
+                               }
                        }
                }
  err:
@@ -297,6 +544,33 @@ int UI_process(UI *ui)
        return ok;
        }
 
+int UI_ctrl(UI *ui, int cmd, long i, void *p, void (*f)())
+       {
+       if (ui == NULL)
+               {
+               UIerr(UI_F_UI_CTRL,ERR_R_PASSED_NULL_PARAMETER);
+               return -1;
+               }
+       switch(cmd)
+               {
+       case UI_CTRL_PRINT_ERRORS:
+               {
+               int save_flag = !!(ui->flags & UI_FLAG_PRINT_ERRORS);
+               if (i)
+                       ui->flags |= UI_FLAG_PRINT_ERRORS;
+               else
+                       ui->flags &= ~UI_FLAG_PRINT_ERRORS;
+               return save_flag;
+               }
+       case UI_CTRL_IS_REDOABLE:
+               return !!(ui->flags & UI_FLAG_REDOABLE);
+       default:
+               break;
+               }
+       UIerr(UI_F_UI_CTRL,UI_R_UNKNOWN_CONTROL_COMMAND);
+       return -1;
+       }
+
 int UI_get_ex_new_index(long argl, void *argp, CRYPTO_EX_new *new_func,
             CRYPTO_EX_dup *dup_func, CRYPTO_EX_free *free_func)
         {
@@ -341,9 +615,24 @@ const UI_METHOD *UI_set_method(UI *ui, const UI_METHOD *meth)
        }
 
 
-UI_METHOD *UI_create_method(void)
+UI_METHOD *UI_create_method(char *name)
        {
-       return (UI_METHOD *)OPENSSL_malloc(sizeof(UI_METHOD));
+       UI_METHOD *ui_method = (UI_METHOD *)OPENSSL_malloc(sizeof(UI_METHOD));
+
+       if (ui_method)
+               memset(ui_method, 0, sizeof(*ui_method));
+       ui_method->name = BUF_strdup(name);
+       return ui_method;
+       }
+
+/* BIG FSCKING WARNING!!!!  If you use this on a statically allocated method
+   (that is, it hasn't been allocated using UI_create_method(), you deserve
+   anything Murphy can throw at you and more!  You have been warned. */
+void UI_destroy_method(UI_METHOD *ui_method)
+       {
+       OPENSSL_free(ui_method->name);
+       ui_method->name = NULL;
+       OPENSSL_free(ui_method);
        }
 
 int UI_method_set_opener(UI_METHOD *method, int (*opener)(UI *ui))
@@ -368,6 +657,17 @@ int UI_method_set_writer(UI_METHOD *method, int (*writer)(UI *ui, UI_STRING *uis
                return -1;
        }
 
+int UI_method_set_flusher(UI_METHOD *method, int (*flusher)(UI *ui))
+       {
+       if (method)
+               {
+               method->ui_flush = flusher;
+               return 0;
+               }
+       else
+               return -1;
+       }
+
 int UI_method_set_reader(UI_METHOD *method, int (*reader)(UI *ui, UI_STRING *uis))
        {
        if (method)
@@ -425,10 +725,17 @@ int (*UI_method_get_closer(UI_METHOD *method))(UI*)
 enum UI_string_types UI_get_string_type(UI_STRING *uis)
        {
        if (!uis)
-               return UI_NONE;
+               return UIT_NONE;
        return uis->type;
        }
 
+int UI_get_input_flags(UI_STRING *uis)
+       {
+       if (!uis)
+               return 0;
+       return uis->input_flags;
+       }
+
 const char *UI_get0_output_string(UI_STRING *uis)
        {
        if (!uis)
@@ -436,16 +743,28 @@ const char *UI_get0_output_string(UI_STRING *uis)
        return uis->out_string;
        }
 
+const char *UI_get0_action_string(UI_STRING *uis)
+       {
+       if (!uis)
+               return NULL;
+       switch(uis->type)
+               {
+       case UIT_PROMPT:
+       case UIT_BOOLEAN:
+               return uis->_.boolean_data.action_desc;
+       default:
+               return NULL;
+               }
+       }
+
 const char *UI_get0_result_string(UI_STRING *uis)
        {
        if (!uis)
                return NULL;
        switch(uis->type)
                {
-       case UI_STRING_ECHO:
-       case UI_STRING_NOECHO:
-       case UI_VERIFY_ECHO:
-       case UI_VERIFY_NOECHO:
+       case UIT_PROMPT:
+       case UIT_VERIFY:
                return uis->result_buf;
        default:
                return NULL;
@@ -456,51 +775,119 @@ const char *UI_get0_test_string(UI_STRING *uis)
        {
        if (!uis)
                return NULL;
-       return uis->test_buf;
+       switch(uis->type)
+               {
+       case UIT_VERIFY:
+               return uis->_.string_data.test_buf;
+       default:
+               return NULL;
+               }
        }
 
 int UI_get_result_minsize(UI_STRING *uis)
        {
        if (!uis)
                return -1;
-       return uis->result_minsize;
+       switch(uis->type)
+               {
+       case UIT_PROMPT:
+       case UIT_VERIFY:
+               return uis->_.string_data.result_minsize;
+       default:
+               return -1;
+               }
        }
 
 int UI_get_result_maxsize(UI_STRING *uis)
        {
        if (!uis)
                return -1;
-       return uis->result_maxsize;
+       switch(uis->type)
+               {
+       case UIT_PROMPT:
+       case UIT_VERIFY:
+               return uis->_.string_data.result_maxsize;
+       default:
+               return -1;
+               }
        }
 
-int UI_set_result(UI_STRING *uis, char *result)
+int UI_set_result(UI *ui, UI_STRING *uis, const char *result)
        {
        int l = strlen(result);
 
+       ui->flags &= ~UI_FLAG_REDOABLE;
+
        if (!uis)
                return -1;
-       if (l < uis->result_minsize)
+       switch (uis->type)
                {
-               UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_SMALL);
-               return -1;
-               }
-       if (l > uis->result_maxsize)
+       case UIT_PROMPT:
+       case UIT_VERIFY:
                {
-               UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_LARGE);
-               return -1;
-               }
+               char number1[20];
+               char number2[20];
 
-       if (!uis->result_buf)
-               {
-               uis->result_buf = OPENSSL_malloc(uis->result_maxsize+1);
+               BIO_snprintf(number1, sizeof(number1), "%d",
+                       uis->_.string_data.result_minsize);
+               BIO_snprintf(number2, sizeof(number2), "%d",
+                       uis->_.string_data.result_maxsize);
+
+               if (l < uis->_.string_data.result_minsize)
+                       {
+                       ui->flags |= UI_FLAG_REDOABLE;
+                       UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_SMALL);
+                       ERR_add_error_data(5,"You must type in ",
+                               number1," to ",number2," characters");
+                       return -1;
+                       }
+               if (l > uis->_.string_data.result_maxsize)
+                       {
+                       ui->flags |= UI_FLAG_REDOABLE;
+                       UIerr(UI_F_UI_SET_RESULT,UI_R_RESULT_TOO_LARGE);
+                       ERR_add_error_data(5,"You must type in ",
+                               number1," to ",number2," characters");
+                       return -1;
+                       }
                }
 
-       if (!uis->result_buf)
+               if (!uis->result_buf)
+                       {
+                       UIerr(UI_F_UI_SET_RESULT,UI_R_NO_RESULT_BUFFER);
+                       return -1;
+                       }
+
+               strcpy(uis->result_buf, result);
+               break;
+       case UIT_BOOLEAN:
                {
-               UIerr(UI_F_UI_NEW_METHOD,ERR_R_MALLOC_FAILURE);
-               return -1;
-               }
+               const char *p;
+
+               if (!uis->result_buf)
+                       {
+                       UIerr(UI_F_UI_SET_RESULT,UI_R_NO_RESULT_BUFFER);
+                       return -1;
+                       }
 
-       strcpy(uis->result_buf, result);
+               uis->result_buf[0] = '\0';
+               for(p = result; *p; p++)
+                       {
+                       if (strchr(uis->_.boolean_data.ok_chars, *p))
+                               {
+                               uis->result_buf[0] =
+                                       uis->_.boolean_data.ok_chars[0];
+                               break;
+                               }
+                       if (strchr(uis->_.boolean_data.cancel_chars, *p))
+                               {
+                               uis->result_buf[0] =
+                                       uis->_.boolean_data.cancel_chars[0];
+                               break;
+                               }
+                       }
+       default:
+               break;
+               }
+               }
        return 0;
        }