2 * Samba Unix/Linux SMB client library
4 * Copyright (C) Christopher Davis 2012
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "regedit_dialog.h"
22 #include "regedit_valuelist.h"
24 #include "lib/registry/registry.h"
28 static int dialog_free(struct dialog
*dia
)
33 if (dia
->sub_window
) {
34 delwin(dia
->sub_window
);
37 del_panel(dia
->panel
);
40 unpost_menu(dia
->choices
);
41 free_menu(dia
->choices
);
43 if (dia
->choice_items
) {
45 for (it
= dia
->choice_items
; *it
!= NULL
; ++it
) {
53 struct dialog
*dialog_new(TALLOC_CTX
*ctx
, const char *title
, int nlines
,
54 int ncols
, int y
, int x
)
58 dia
= talloc_zero(ctx
, struct dialog
);
63 talloc_set_destructor(dia
, dialog_free
);
65 dia
->window
= newwin(nlines
, ncols
, y
, x
);
66 if (dia
->window
== NULL
) {
70 box(dia
->window
, 0, 0);
71 mvwaddstr(dia
->window
, 0, 1, title
);
73 /* body of the dialog within the box outline */
74 dia
->sub_window
= derwin(dia
->window
, nlines
- 2, ncols
- 2, 1, 1);
75 if (dia
->sub_window
== NULL
) {
79 dia
->panel
= new_panel(dia
->window
);
80 if (dia
->panel
== NULL
) {
93 static void center_dialog_above_window(WINDOW
*below
, int *nlines
, int *ncols
,
99 getmaxyx(below
, maxy
, maxx
);
106 if (*nlines
> maxy
) {
113 if (*nlines
< centery
) {
114 *y
= centery
- *nlines
;
116 if (*ncols
< centerx
) {
117 *x
= centerx
- *ncols
;
121 struct dialog
*dialog_center_new(TALLOC_CTX
*ctx
, const char *title
, int nlines
,
122 int ncols
, WINDOW
*below
)
126 center_dialog_above_window(below
, &nlines
, &ncols
, &y
, &x
);
128 return dialog_new(ctx
, title
, nlines
, ncols
, y
, x
);
131 struct dialog
*dialog_choice_new(TALLOC_CTX
*ctx
, const char *title
,
132 const char **choices
, int nlines
,
133 int ncols
, int y
, int x
)
138 dia
= dialog_new(ctx
, title
, nlines
, ncols
, y
, x
);
143 dia
->menu_window
= derwin(dia
->sub_window
, 1, ncols
- 3,
145 if (dia
->menu_window
== NULL
) {
149 for (nchoices
= 0; choices
[nchoices
] != NULL
; ++nchoices
)
151 dia
->choice_items
= talloc_zero_array(dia
, ITEM
*, nchoices
+ 1);
152 if (dia
->choice_items
== NULL
) {
155 for (i
= 0; i
< nchoices
; ++i
) {
156 char *desc
= talloc_strdup(dia
, choices
[i
]);
160 dia
->choice_items
[i
] = new_item(desc
, desc
);
161 if (dia
->choice_items
[i
] == NULL
) {
164 /* store choice index */
165 set_item_userptr(dia
->choice_items
[i
], (void*)(uintptr_t)i
);
168 dia
->choices
= new_menu(dia
->choice_items
);
169 if (dia
->choices
== NULL
) {
173 set_menu_format(dia
->choices
, 1, ncols
);
174 set_menu_win(dia
->choices
, dia
->sub_window
);
175 set_menu_sub(dia
->choices
, dia
->menu_window
);
176 menu_opts_off(dia
->choices
, O_SHOWDESC
);
177 set_menu_mark(dia
->choices
, "* ");
178 post_menu(dia
->choices
);
179 wmove(dia
->sub_window
, 0, 0);
189 struct dialog
*dialog_choice_center_new(TALLOC_CTX
*ctx
, const char *title
,
190 const char **choices
, int nlines
,
191 int ncols
, WINDOW
*below
)
195 center_dialog_above_window(below
, &nlines
, &ncols
, &y
, &x
);
197 return dialog_choice_new(ctx
, title
, choices
, nlines
, ncols
, y
, x
);
200 static int handle_menu_input(MENU
*menu
, int c
)
206 menu_driver(menu
, REQ_LEFT_ITEM
);
209 menu_driver(menu
, REQ_RIGHT_ITEM
);
213 item
= current_item(menu
);
214 return (int)(uintptr_t)item_userptr(item
);
220 static int modal_loop(struct dialog
*dia
)
225 keypad(dia
->window
, true);
229 while (selection
== -1) {
230 c
= wgetch(dia
->window
);
231 selection
= handle_menu_input(dia
->choices
, c
);
241 int dialog_notice(TALLOC_CTX
*ctx
, enum dialog_type type
,
242 const char *title
, WINDOW
*below
,
243 const char *msg
, ...)
248 const char *choices
[] = {
256 str
= talloc_vasprintf(ctx
, msg
, ap
);
262 width
= strlen(str
) + 2;
264 if (type
== DIA_ALERT
) {
267 dia
= dialog_choice_center_new(ctx
, title
, choices
, 5, width
, below
);
272 waddstr(dia
->sub_window
, str
);
275 return modal_loop(dia
);
278 static void handle_form_input(FORM
*frm
, int c
)
282 form_driver(frm
, REQ_NEW_LINE
);
285 form_driver(frm
, REQ_UP_CHAR
);
288 form_driver(frm
, REQ_DOWN_CHAR
);
292 form_driver(frm
, REQ_DEL_PREV
);
295 form_driver(frm
, REQ_LEFT_CHAR
);
298 form_driver(frm
, REQ_RIGHT_CHAR
);
318 FIELD
*field
[MAX_FIELDS
];
319 enum input_section section
;
322 static int edit_dialog_free(struct edit_dialog
*edit
)
325 unpost_form(edit
->input
);
327 if (edit
->field
[0]) {
328 free_field(edit
->field
[0]);
330 if (edit
->field
[1]) {
331 free_field(edit
->field
[1]);
333 delwin(edit
->input_win
);
338 static WERROR
fill_value_buffer(struct edit_dialog
*edit
,
339 const struct value_item
*vitem
)
343 switch (vitem
->type
) {
346 if (vitem
->data
.length
>= 4) {
347 v
= IVAL(vitem
->data
.data
, 0);
349 tmp
= talloc_asprintf(edit
, "0x%x", v
);
353 set_field_buffer(edit
->field
[1], 0, tmp
);
358 case REG_EXPAND_SZ
: {
361 if (!pull_reg_sz(edit
, &vitem
->data
, &s
)) {
364 set_field_buffer(edit
->field
[1], 0, s
);
371 if (!pull_reg_multi_sz(edit
, &vitem
->data
, &a
)) {
374 for (p
= a
; *p
!= NULL
; ++p
) {
376 buf
= talloc_asprintf(edit
, "%s\n", *p
);
378 buf
= talloc_asprintf_append(buf
, "%s\n", *p
);
384 set_field_buffer(edit
->field
[1], 0, buf
);
393 static char *string_trim_n(TALLOC_CTX
*ctx
, const char *buf
, size_t n
)
397 str
= talloc_strndup(ctx
, buf
, n
);
400 trim_string(str
, " ", " ");
406 static char *string_trim(TALLOC_CTX
*ctx
, const char *buf
)
410 str
= talloc_strdup(ctx
, buf
);
413 trim_string(str
, " ", " ");
419 static bool value_exists(TALLOC_CTX
*ctx
, const struct registry_key
*key
,
426 rv
= reg_key_get_value_by_name(ctx
, key
, name
, &type
, &blob
);
428 return W_ERROR_IS_OK(rv
);
431 static WERROR
set_value(struct edit_dialog
*edit
, struct registry_key
*key
,
432 uint32_t type
, bool new_value
)
436 const char *buf
= field_buffer(edit
->field
[1], 0);
437 char *name
= string_trim(edit
, field_buffer(edit
->field
[0], 0));
442 if (!new_value
&& !field_status(edit
->field
[1])) {
445 if (new_value
&& value_exists(edit
, key
, name
)) {
446 return WERR_FILE_EXISTS
;
454 if (buf
[0] == '0' && tolower(buf
[1]) == 'x') {
458 val
= strtoul(buf
, NULL
, base
);
459 blob
= data_blob_talloc(edit
, NULL
, sizeof(val
));
460 SIVAL(blob
.data
, 0, val
);
465 case REG_EXPAND_SZ
: {
466 char *str
= string_trim(edit
, buf
);
467 if (!str
|| !push_reg_sz(edit
, &blob
, str
)) {
477 dynamic_field_info(edit
->field
[1], &rows
, &cols
, &max
);
479 arr
= talloc_zero_array(edit
, const char *, rows
+ 1);
483 for (i
= 0; *buf
; ++i
, buf
+= cols
) {
484 SMB_ASSERT(i
< rows
);
485 arr
[i
] = string_trim_n(edit
, buf
, cols
);
487 if (!push_reg_multi_sz(edit
, &blob
, arr
)) {
494 rv
= reg_val_set(key
, name
, type
, blob
);
499 static void section_down(struct edit_dialog
*edit
)
501 switch (edit
->section
) {
503 if (form_driver(edit
->input
, REQ_VALIDATION
) == E_OK
) {
504 edit
->section
= IN_DATA
;
505 set_current_field(edit
->input
, edit
->field
[1]);
509 if (form_driver(edit
->input
, REQ_VALIDATION
) == E_OK
) {
510 edit
->section
= IN_MENU
;
511 menu_driver(edit
->dia
->choices
, REQ_FIRST_ITEM
);
515 edit
->section
= IN_NAME
;
516 set_current_field(edit
->input
, edit
->field
[0]);
521 static void section_up(struct edit_dialog
*edit
)
523 switch (edit
->section
) {
525 if (form_driver(edit
->input
, REQ_VALIDATION
) == E_OK
) {
526 edit
->section
= IN_MENU
;
527 menu_driver(edit
->dia
->choices
, REQ_FIRST_ITEM
);
531 if (form_driver(edit
->input
, REQ_VALIDATION
) == E_OK
) {
532 edit
->section
= IN_NAME
;
533 set_current_field(edit
->input
, edit
->field
[0]);
537 edit
->section
= IN_DATA
;
538 set_current_field(edit
->input
, edit
->field
[1]);
543 WERROR
dialog_edit_value(TALLOC_CTX
*ctx
, struct registry_key
*key
, uint32_t type
,
544 const struct value_item
*vitem
, WINDOW
*below
)
546 struct edit_dialog
*edit
;
547 const char *choices
[] = {
553 int nlines
, ncols
, val_rows
;
554 WERROR rv
= WERR_NOMEM
;
557 edit
= talloc_zero(ctx
, struct edit_dialog
);
561 talloc_set_destructor(edit
, edit_dialog_free
);
563 title
= talloc_asprintf(edit
, "Edit %s value", str_regtype(type
));
569 if (type
== REG_MULTI_SZ
) {
573 edit
->dia
= dialog_choice_center_new(edit
, title
, choices
, nlines
,
576 if (edit
->dia
== NULL
) {
581 edit
->field
[0] = new_field(1, ncols
- 4, 1, 1, 0, 0);
582 if (edit
->field
[0] == NULL
) {
588 if (type
== REG_MULTI_SZ
) {
591 edit
->field
[1] = new_field(val_rows
, ncols
- 4, 4, 1, 0, 0);
592 if (edit
->field
[1] == NULL
) {
595 set_field_back(edit
->field
[0], A_REVERSE
);
596 set_field_back(edit
->field
[1], A_REVERSE
);
597 field_opts_off(edit
->field
[0], O_BLANK
| O_AUTOSKIP
| O_STATIC
);
598 field_opts_off(edit
->field
[1], O_BLANK
| O_AUTOSKIP
| O_STATIC
| O_WRAP
);
599 if (type
== REG_DWORD
) {
600 set_field_type(edit
->field
[1], TYPE_REGEXP
,
601 "^ *([0-9]+|0[xX][0-9a-fA-F]+) *$");
605 set_field_buffer(edit
->field
[0], 0, vitem
->value_name
);
606 field_opts_off(edit
->field
[0], O_EDIT
);
607 fill_value_buffer(edit
, vitem
);
610 edit
->input
= new_form(edit
->field
);
611 if (edit
->input
== NULL
) {
614 form_opts_off(edit
->input
, O_NL_OVERLOAD
| O_BS_OVERLOAD
);
616 edit
->input_win
= derwin(edit
->dia
->sub_window
, nlines
- 3, ncols
- 3,
618 if (edit
->input_win
== NULL
) {
622 set_form_win(edit
->input
, edit
->dia
->sub_window
);
623 set_form_sub(edit
->input
, edit
->input_win
);
624 post_form(edit
->input
);
625 mvwprintw(edit
->dia
->sub_window
, 0, 0, "Name");
626 mvwprintw(edit
->dia
->sub_window
, 3, 0, "Data");
628 keypad(edit
->dia
->window
, true);
632 edit
->section
= IN_NAME
;
635 int c
= wgetch(edit
->dia
->window
);
639 } else if (c
== KEY_BTAB
) {
644 if (edit
->section
== IN_NAME
|| edit
->section
== IN_DATA
) {
645 handle_form_input(edit
->input
, c
);
647 selection
= handle_menu_input(edit
->dia
->choices
, c
);
648 if (selection
== DIALOG_OK
) {
649 rv
= set_value(edit
, key
, type
, vitem
== NULL
);
650 if (W_ERROR_EQUAL(rv
, WERR_FILE_EXISTS
)) {
651 dialog_notice(edit
, DIA_ALERT
,
652 "Value exists", below
,
653 "Value name already exists.");
658 } else if (selection
== DIALOG_CANCEL
) {
674 int dialog_select_type(TALLOC_CTX
*ctx
, int *type
, WINDOW
*below
)
677 const char *choices
[] = {
682 const char *reg_types
[] = {
688 #define NTYPES (sizeof(reg_types) / sizeof(const char*))
695 dia
= dialog_choice_center_new(ctx
, "New Value", choices
, 10, 20,
701 mvwprintw(dia
->sub_window
, 0, 0, "Choose type:");
702 type_win
= derwin(dia
->sub_window
, 6, 18, 1, 0);
703 if (type_win
== NULL
) {
707 item
= talloc_zero_array(dia
, ITEM
*, NTYPES
+ 1);
712 for (i
= 0; i
< NTYPES
; ++i
) {
713 int t
= regtype_by_string(reg_types
[i
]);
715 item
[i
] = new_item(reg_types
[i
], reg_types
[i
]);
716 if (item
[i
] == NULL
) {
719 set_item_userptr(item
[i
], (void*)(uintptr_t)t
);
722 list
= new_menu(item
);
727 set_menu_format(list
, 7, 1);
728 set_menu_win(list
, dia
->sub_window
);
729 set_menu_sub(list
, type_win
);
730 menu_opts_off(list
, O_SHOWDESC
);
731 set_menu_mark(list
, "* ");
734 keypad(dia
->window
, true);
740 int c
= wgetch(dia
->window
);
744 menu_driver(list
, REQ_UP_ITEM
);
747 menu_driver(list
, REQ_DOWN_ITEM
);
750 menu_driver(dia
->choices
, REQ_LEFT_ITEM
);
753 menu_driver(dia
->choices
, REQ_RIGHT_ITEM
);
757 it
= current_item(list
);
758 *type
= (int)(uintptr_t)item_userptr(it
);
759 it
= current_item(dia
->choices
);
760 sel
= (int)(uintptr_t)item_userptr(it
);
772 for (it
= item
; *it
; ++it
) {