source4/gtk/tools/gregedit.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    GTK+ registry frontend
   4
   5    Copyright (C) Jelmer Vernooij 2004-2005
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23 #include "lib/registry/registry.h"
  24 #include "gtk/common/gtk-smb.h"
  25 #include "smb_build.h"
  26
  27 static GtkTreeStore *store_keys;
  28 static GtkListStore *store_vals;
  29 static GtkWidget *tree_keys;
  30 static GtkWidget *tree_vals;
  31 static GtkWidget *mainwin;
  32 static GtkWidget *mnu_add_key, *mnu_set_value, *mnu_del_key, *mnu_del_value, *mnu_find;
  33 static TALLOC_CTX *mem_ctx; /* FIXME: Split up */
  34
  35 static GtkWidget *save;
  36 static GtkWidget *save_as;
  37 static GtkWidget* create_openfilewin (void);
  38 static GtkWidget* create_savefilewin (void);
  39 struct registry_context *registry = NULL;
  40 struct registry_key *current_key = NULL;
  41
  42 static GtkWidget* create_FindDialog (void)
  43 {
  44   GtkWidget *FindDialog;
  45   GtkWidget *dialog_vbox2;
  46   GtkWidget *vbox1;
  47   GtkWidget *hbox1;
  48   GtkWidget *label6;
  49   GtkWidget *entry_pattern;
  50   GtkWidget *frame3;
  51   GtkWidget *alignment3;
  52   GtkWidget *vbox2;
  53   GtkWidget *checkbutton1;
  54   GtkWidget *checkbutton2;
  55   GtkWidget *checkbutton3;
  56   GtkWidget *label7;
  57   GtkWidget *dialog_action_area2;
  58   GtkWidget *cancelbutton2;
  59   GtkWidget *okbutton2;
  60
  61   FindDialog = gtk_dialog_new ();
  62   gtk_window_set_title (GTK_WINDOW (FindDialog), "Find Key or Value");
  63   gtk_window_set_resizable (GTK_WINDOW (FindDialog), FALSE);
  64   gtk_window_set_type_hint (GTK_WINDOW (FindDialog), GDK_WINDOW_TYPE_HINT_DIALOG);
  65
  66   dialog_vbox2 = GTK_DIALOG (FindDialog)->vbox;
  67
  68   vbox1 = gtk_vbox_new (FALSE, 0);
  69   gtk_box_pack_start (GTK_BOX (dialog_vbox2), vbox1, TRUE, TRUE, 0);
  70
  71   hbox1 = gtk_hbox_new (FALSE, 0);
  72   gtk_box_pack_start (GTK_BOX (vbox1), hbox1, TRUE, TRUE, 0);
  73
  74   label6 = gtk_label_new ("Find String");
  75   gtk_box_pack_start (GTK_BOX (hbox1), label6, FALSE, FALSE, 0);
  76
  77   entry_pattern = gtk_entry_new ();
  78   gtk_box_pack_start (GTK_BOX (hbox1), entry_pattern, TRUE, TRUE, 0);
  79
  80   frame3 = gtk_frame_new (NULL);
  81   gtk_box_pack_start (GTK_BOX (vbox1), frame3, TRUE, TRUE, 0);
  82   gtk_frame_set_shadow_type (GTK_FRAME (frame3), GTK_SHADOW_NONE);
  83
  84   alignment3 = gtk_alignment_new (0.5, 0.5, 1, 1);
  85   gtk_container_add (GTK_CONTAINER (frame3), alignment3);
  86   gtk_alignment_set_padding (GTK_ALIGNMENT (alignment3), 0, 0, 12, 0);
  87
  88   vbox2 = gtk_vbox_new (FALSE, 0);
  89   gtk_container_add (GTK_CONTAINER (alignment3), vbox2);
  90
  91   checkbutton1 = gtk_check_button_new_with_mnemonic ("_Key Names");
  92   gtk_box_pack_start (GTK_BOX (vbox2), checkbutton1, FALSE, FALSE, 0);
  93
  94   checkbutton2 = gtk_check_button_new_with_mnemonic ("_Value Names");
  95   gtk_box_pack_start (GTK_BOX (vbox2), checkbutton2, FALSE, FALSE, 0);
  96
  97   checkbutton3 = gtk_check_button_new_with_mnemonic ("Value _Data");
  98   gtk_box_pack_start (GTK_BOX (vbox2), checkbutton3, FALSE, FALSE, 0);
  99
 100   label7 = gtk_label_new ("<b>Search in</b>");
 101   gtk_frame_set_label_widget (GTK_FRAME (frame3), label7);
 102   gtk_label_set_use_markup (GTK_LABEL (label7), TRUE);
 103
 104   dialog_action_area2 = GTK_DIALOG (FindDialog)->action_area;
 105   gtk_button_box_set_layout (GTK_BUTTON_BOX (dialog_action_area2), GTK_BUTTONBOX_END);
 106
 107   cancelbutton2 = gtk_button_new_from_stock ("gtk-cancel");
 108   gtk_dialog_add_action_widget (GTK_DIALOG (FindDialog), cancelbutton2, GTK_RESPONSE_CANCEL);
 109   GTK_WIDGET_SET_FLAGS (cancelbutton2, GTK_CAN_DEFAULT);
 110
 111   okbutton2 = gtk_button_new_from_stock ("gtk-ok");
 112   gtk_dialog_add_action_widget (GTK_DIALOG (FindDialog), okbutton2, GTK_RESPONSE_OK);
 113   GTK_WIDGET_SET_FLAGS (okbutton2, GTK_CAN_DEFAULT);
 114
 115   gtk_widget_show_all (dialog_vbox2);
 116
 117   return FindDialog;
 118 }
 119
 120 static GtkWidget* create_SetValueDialog (GtkWidget **entry_name, GtkWidget **entry_type, GtkWidget **entry_data)
 121 {
 122   GtkWidget *SetValueDialog;
 123   GtkWidget *dialog_vbox1;
 124   GtkWidget *table1;
 125   GtkWidget *label3;
 126   GtkWidget *label4;
 127   GtkWidget *label5;
 128   GtkWidget *entry_value_name;
 129   GtkWidget *value_data;
 130   GtkWidget *combo_data_type;
 131   GtkWidget *dialog_action_area1;
 132   GtkWidget *cancelbutton1;
 133   GtkWidget *okbutton1;
 134
 135   SetValueDialog = gtk_dialog_new ();
 136   gtk_window_set_title (GTK_WINDOW (SetValueDialog), "Set Registry Value");
 137   gtk_window_set_position (GTK_WINDOW (SetValueDialog), GTK_WIN_POS_CENTER);
 138   gtk_window_set_resizable (GTK_WINDOW (SetValueDialog), FALSE);
 139   gtk_window_set_type_hint (GTK_WINDOW (SetValueDialog), GDK_WINDOW_TYPE_HINT_DIALOG);
 140
 141   dialog_vbox1 = GTK_DIALOG (SetValueDialog)->vbox;
 142
 143   table1 = gtk_table_new (3, 2, FALSE);
 144   gtk_box_pack_start (GTK_BOX (dialog_vbox1), table1, TRUE, TRUE, 0);
 145
 146   label3 = gtk_label_new ("Value name:");
 147   gtk_table_attach (GTK_TABLE (table1), label3, 0, 1, 0, 1,
 148                     (GtkAttachOptions) (GTK_FILL),
 149                     (GtkAttachOptions) (0), 0, 0);
 150   gtk_misc_set_alignment (GTK_MISC (label3), 0, 0.5);
 151
 152   label4 = gtk_label_new ("Data Type:");
 153   gtk_table_attach (GTK_TABLE (table1), label4, 0, 1, 1, 2,
 154                     (GtkAttachOptions) (GTK_FILL),
 155                     (GtkAttachOptions) (0), 0, 0);
 156   gtk_misc_set_alignment (GTK_MISC (label4), 0, 0.5);
 157
 158   label5 = gtk_label_new ("Data:");
 159   gtk_table_attach (GTK_TABLE (table1), label5, 0, 1, 2, 3,
 160                     (GtkAttachOptions) (GTK_FILL),
 161                     (GtkAttachOptions) (0), 0, 0);
 162   gtk_misc_set_alignment (GTK_MISC (label5), 0, 0.5);
 163
 164   *entry_name = entry_value_name = gtk_entry_new ();
 165   gtk_table_attach (GTK_TABLE (table1), entry_value_name, 1, 2, 0, 1,
 166                     (GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
 167                     (GtkAttachOptions) (0), 0, 0);
 168
 169   *entry_data = value_data = gtk_entry_new ();
 170   gtk_table_attach (GTK_TABLE (table1), value_data, 1, 2, 2, 3,
 171                     (GtkAttachOptions) (GTK_EXPAND | GTK_FILL),
 172                     (GtkAttachOptions) (0), 0, 0);
 173
 174   *entry_type = combo_data_type = gtk_combo_box_new_text ();
 175
 176   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_NONE");
 177   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_SZ");
 178   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_EXPAND_SZ");
 179   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_BINARY");
 180   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_DWORD_LE");
 181   gtk_combo_box_append_text(GTK_COMBO_BOX(combo_data_type), "REG_DWORD_BE");
 182
 183   gtk_table_attach (GTK_TABLE (table1), combo_data_type, 1, 2, 1, 2,
 184                     (GtkAttachOptions) (GTK_FILL),
 185                     (GtkAttachOptions) (GTK_FILL), 0, 0);
 186
 187   dialog_action_area1 = GTK_DIALOG (SetValueDialog)->action_area;
 188   gtk_button_box_set_layout (GTK_BUTTON_BOX (dialog_action_area1), GTK_BUTTONBOX_END);
 189
 190   cancelbutton1 = gtk_button_new_from_stock ("gtk-cancel");
 191   gtk_dialog_add_action_widget (GTK_DIALOG (SetValueDialog), cancelbutton1, GTK_RESPONSE_CANCEL);
 192   GTK_WIDGET_SET_FLAGS (cancelbutton1, GTK_CAN_DEFAULT);
 193
 194   okbutton1 = gtk_button_new_from_stock ("gtk-ok");
 195   gtk_dialog_add_action_widget (GTK_DIALOG (SetValueDialog), okbutton1, GTK_RESPONSE_OK);
 196   GTK_WIDGET_SET_FLAGS (okbutton1, GTK_CAN_DEFAULT);
 197
 198   gtk_widget_show_all (dialog_vbox1);
 199
 200   return SetValueDialog;
 201 }
 202
 203 static GtkWidget* create_NewKeyDialog (GtkWidget **name_entry)
 204 {
 205   GtkWidget *NewKeyDialog;
 206   GtkWidget *dialog_vbox2;
 207   GtkWidget *hbox1;
 208   GtkWidget *label6;
 209   GtkWidget *entry_key_name;
 210   GtkWidget *dialog_action_area2;
 211   GtkWidget *cancelbutton2;
 212   GtkWidget *okbutton2;
 213
 214   NewKeyDialog = gtk_dialog_new ();
 215   gtk_window_set_title (GTK_WINDOW (NewKeyDialog), "New Registry Key");
 216   gtk_window_set_position (GTK_WINDOW (NewKeyDialog), GTK_WIN_POS_CENTER);
 217   gtk_window_set_resizable (GTK_WINDOW (NewKeyDialog), FALSE);
 218   gtk_window_set_type_hint (GTK_WINDOW (NewKeyDialog), GDK_WINDOW_TYPE_HINT_DIALOG);
 219
 220   dialog_vbox2 = GTK_DIALOG (NewKeyDialog)->vbox;
 221
 222   hbox1 = gtk_hbox_new (FALSE, 0);
 223   gtk_box_pack_start (GTK_BOX (dialog_vbox2), hbox1, TRUE, TRUE, 0);
 224
 225   label6 = gtk_label_new ("Name:");
 226   gtk_box_pack_start (GTK_BOX (hbox1), label6, FALSE, FALSE, 0);
 227
 228   entry_key_name = gtk_entry_new ();
 229   gtk_box_pack_start (GTK_BOX (hbox1), entry_key_name, TRUE, TRUE, 0);
 230
 231   dialog_action_area2 = GTK_DIALOG (NewKeyDialog)->action_area;
 232   gtk_button_box_set_layout (GTK_BUTTON_BOX (dialog_action_area2), GTK_BUTTONBOX_END);
 233
 234   *name_entry = entry_key_name;
 235
 236   cancelbutton2 = gtk_button_new_from_stock ("gtk-cancel");
 237   gtk_dialog_add_action_widget (GTK_DIALOG (NewKeyDialog), cancelbutton2, GTK_RESPONSE_CANCEL);
 238   GTK_WIDGET_SET_FLAGS (cancelbutton2, GTK_CAN_DEFAULT);
 239
 240   okbutton2 = gtk_button_new_from_stock ("gtk-ok");
 241   gtk_dialog_add_action_widget (GTK_DIALOG (NewKeyDialog), okbutton2, GTK_RESPONSE_OK);
 242   GTK_WIDGET_SET_FLAGS (okbutton2, GTK_CAN_DEFAULT);
 243
 244   gtk_widget_show_all (dialog_vbox2);
 245
 246   return NewKeyDialog;
 247 }
 248
 249 static void expand_key(GtkTreeView *treeview, GtkTreeIter *parent, GtkTreePath *arg2)
 250 {
 251         GtkTreeIter firstiter, iter, tmpiter;
 252         struct registry_key *k, *sub;
 253         char *name;
 254         WERROR error;
 255         int i;
 256
 257         gtk_tree_model_iter_children(GTK_TREE_MODEL(store_keys), &firstiter, parent);
 258
 259         /* See if this row has ever had a name gtk_tree_store_set()'ed to it.
 260            If not, read the directory contents */
 261         gtk_tree_model_get(GTK_TREE_MODEL(store_keys), &firstiter, 0, &name, -1);
 262
 263         if(name) return;
 264
 265         gtk_tree_model_get(GTK_TREE_MODEL(store_keys), parent, 1, &k, -1);
 266
 267         g_assert(k);
 268
 269         for(i = 0; W_ERROR_IS_OK(error = reg_key_get_subkey_by_index(mem_ctx, k, i, &sub)); i++) {
 270                 uint32_t count;
 271                 /* Replace the blank child with the first directory entry
 272            You may be tempted to remove the blank child node and then
 273            append a new one.  Don't.  If you remove the blank child
 274            node GTK gets confused and won't expand the parent row. */
 275
 276                 if(i == 0) {
 277                         iter = firstiter;
 278                 } else {
 279                         gtk_tree_store_append(store_keys, &iter, parent);
 280                 }
 281                 gtk_tree_store_set (store_keys,
 282                                             &iter,
 283                                                 0,
 284                                                 sub->name,
 285                                                 1,
 286                                                 sub,
 287                                                 -1);
 288
 289                 if(W_ERROR_IS_OK(reg_key_num_subkeys(sub, &count)) && count > 0)
 290                         gtk_tree_store_append(store_keys, &tmpiter, &iter);
 291         }
 292
 293         if(!W_ERROR_EQUAL(error, WERR_NO_MORE_ITEMS)) {
 294                 gtk_show_werror(mainwin, "While enumerating subkeys", error);
 295         }
 296 }
 297
 298 static void registry_load_hive(struct registry_key *root)
 299 {
 300         GtkTreeIter iter, tmpiter;
 301         gtk_list_store_clear(store_vals);
 302         /* Add the root */
 303         gtk_tree_store_append(store_keys, &iter, NULL);
 304         gtk_tree_store_set (store_keys,
 305                                     &iter,
 306                                         0,
 307                                         root->name?root->name:"",
 308                                         1,
 309                                         root,
 310                                         -1);
 311
 312         gtk_tree_store_append(store_keys, &tmpiter, &iter);
 313
 314         gtk_widget_set_sensitive( save, True );
 315         gtk_widget_set_sensitive( save_as, True );
 316 }
 317
 318 static void registry_load_root(void)
 319 {
 320         struct registry_key *root;
 321         uint32_t i = 0;
 322         if(!registry) return;
 323
 324         gtk_list_store_clear(store_vals);
 325         gtk_tree_store_clear(store_keys);
 326
 327         for(i = HKEY_CLASSES_ROOT; i <= HKEY_PERFORMANCE_NLSTEXT; i++)
 328         {
 329                 if (!W_ERROR_IS_OK(reg_get_predefined_key(registry, i, &root))) { continue; }
 330
 331                 registry_load_hive(root);
 332         }
 333 }
 334
 335 static void on_open_file_activate (GtkMenuItem *menuitem, gpointer user_data)
 336 {
 337         GtkWidget *openfilewin;
 338         gint result;
 339         char *filename, *tmp;
 340         struct registry_key *root;
 341         WERROR error;
 342
 343         openfilewin = create_openfilewin();
 344
 345         result = gtk_dialog_run(GTK_DIALOG(openfilewin));
 346
 347         switch(result) {
 348         case GTK_RESPONSE_OK:
 349                 filename = strdup(gtk_file_selection_get_filename(GTK_FILE_SELECTION(openfilewin)));
 350                 error = reg_open_hive(NULL, user_data, filename, NULL, &root);
 351                 if(!W_ERROR_IS_OK(error)) {
 352                         gtk_show_werror(mainwin, "Error while opening hive", error);
 353                         break;
 354                 }
 355
 356                 tmp = g_strdup_printf("Registry Editor - %s", filename);
 357                 gtk_window_set_title (GTK_WINDOW (mainwin), tmp);
 358                 g_free(tmp);
 359                 gtk_tree_store_clear(store_keys);
 360                 registry_load_hive(root);
 361                 break;
 362         default:
 363                 break;
 364         }
 365
 366         gtk_widget_destroy(openfilewin);
 367 }
 368
 369 static void on_open_gconf_activate(GtkMenuItem *menuitem, gpointer user_data)
 370 {
 371         struct registry_key *root;
 372         WERROR error = reg_open_hive(NULL, "gconf", NULL, NULL, &root);
 373         if(!W_ERROR_IS_OK(error)) {
 374                 gtk_show_werror(mainwin, "Error while opening GConf", error);
 375                 return;
 376         }
 377
 378         gtk_window_set_title (GTK_WINDOW (mainwin), "Registry Editor - GConf");
 379
 380         gtk_tree_store_clear(store_keys);
 381         registry_load_hive(root);
 382 }
 383
 384 static void on_open_local_activate(GtkMenuItem *menuitem, gpointer user_data)
 385 {
 386         WERROR error = reg_open_local(&registry);
 387         if(!W_ERROR_IS_OK(error)) {
 388                 gtk_show_werror(mainwin, "Error while opening local registry", error);
 389                 return;
 390         }
 391         registry_load_root();
 392 }
 393
 394 static void on_open_remote_activate(GtkMenuItem *menuitem, gpointer user_data)
 395 {
 396         char *tmp;
 397         GtkWidget *rpcwin = GTK_WIDGET(gtk_rpc_binding_dialog_new(NULL));
 398         gint result = gtk_dialog_run(GTK_DIALOG(rpcwin));
 399         WERROR error;
 400         struct cli_credentials *creds;
 401
 402         if(result != GTK_RESPONSE_ACCEPT)
 403         {
 404                 gtk_widget_destroy(rpcwin);
 405                 return;
 406         }
 407
 408         creds = cli_credentials_init(mem_ctx);
 409         cli_credentials_guess(creds);
 410         cli_credentials_set_gtk_callbacks(creds);
 411
 412         error = reg_open_remote(&registry,
 413                                 creds,
 414                                 gtk_rpc_binding_dialog_get_binding_string(GTK_RPC_BINDING_DIALOG(rpcwin), mem_ctx),
 415                                 NULL);
 416
 417         if(!W_ERROR_IS_OK(error)) {
 418                 gtk_show_werror(mainwin, "Error while opening remote registry", error);
 419                 gtk_widget_destroy(rpcwin);
 420                 return;
 421         }
 422
 423         tmp = g_strdup_printf("Registry Editor - Remote Registry at %s", gtk_rpc_binding_dialog_get_host(GTK_RPC_BINDING_DIALOG(rpcwin)));
 424         gtk_window_set_title (GTK_WINDOW (mainwin), tmp);
 425         g_free(tmp);
 426
 427         registry_load_root();
 428
 429
 430         gtk_widget_destroy(rpcwin);
 431 }
 432
 433
 434 static void on_save_as_activate(GtkMenuItem *menuitem, gpointer user_data)
 435 {
 436         gint result;
 437         WERROR error;
 438         GtkWidget *savefilewin = create_savefilewin();
 439         result = gtk_dialog_run(GTK_DIALOG(savefilewin));
 440         switch(result) {
 441         case GTK_RESPONSE_OK:
 442         /* FIXME:               error = reg_dump(registry, gtk_file_selection_get_filename(GTK_FILE_SELECTION(savefilewin))); */
 443                 if(!W_ERROR_IS_OK(error)) {
 444                         gtk_show_werror(mainwin, "Error while saving as", error);
 445                 }
 446                 break;
 447
 448         default:
 449                 break;
 450
 451         }
 452         gtk_widget_destroy(savefilewin);
 453 }
 454
 455
 456 static void on_quit_activate(GtkMenuItem *menuitem, gpointer user_data)
 457 {
 458         gtk_main_quit();
 459 }
 460
 461
 462 static void on_delete_value_activate(GtkMenuItem *menuitem, gpointer user_data)
 463 {
 464         WERROR error;
 465         GtkTreeIter iter;
 466         const char *value;
 467
 468         if (!gtk_tree_selection_get_selected (gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_vals)), NULL, &iter)) {
 469                 return;
 470         }
 471
 472         gtk_tree_model_get(GTK_TREE_MODEL(store_vals), &iter, 0, &value, -1);
 473
 474         error = reg_del_value(current_key, value);
 475
 476         if (!W_ERROR_IS_OK(error)) {
 477                 gtk_show_werror(NULL, "Error while deleting value", error);
 478                 return;
 479         }
 480 }
 481
 482 static void on_delete_key_activate(GtkMenuItem *menuitem, gpointer user_data)
 483 {
 484         WERROR error;
 485         GtkTreeIter iter, parentiter;
 486         struct registry_key *parent_key;
 487
 488         if (!gtk_tree_selection_get_selected (gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_keys)), NULL, &iter)) {
 489                 return;
 490         }
 491
 492         if (!gtk_tree_model_iter_parent(GTK_TREE_MODEL(store_keys), &parentiter, &iter)) {
 493                 return;
 494         }
 495
 496         gtk_tree_model_get(GTK_TREE_MODEL(store_keys), &parentiter, 1, &parent_key, -1);
 497
 498         error = reg_key_del(parent_key, current_key->name);
 499
 500         if (!W_ERROR_IS_OK(error)) {
 501                 gtk_show_werror(NULL, "Error while deleting key", error);
 502                 return;
 503         }
 504 }
 505
 506 static void on_add_key_activate(GtkMenuItem *menuitem, gpointer user_data)
 507 {
 508         GtkWidget *entry;
 509         GtkDialog *addwin = GTK_DIALOG(create_NewKeyDialog(&entry));
 510         gint result = gtk_dialog_run(addwin);
 511
 512         if (result == GTK_RESPONSE_OK)
 513         {
 514                 struct registry_key *newkey;
 515                 WERROR error = reg_key_add_name(mem_ctx, current_key, gtk_entry_get_text(GTK_ENTRY(entry)), 0, NULL, &newkey);
 516
 517                 if (!W_ERROR_IS_OK(error)) {
 518                         gtk_show_werror(NULL, "Error while adding key", error);
 519                 }
 520         }
 521
 522         gtk_widget_destroy(GTK_WIDGET(addwin));
 523 }
 524
 525 static void on_value_activate(GtkTreeView *treeview, GtkTreePath *arg1,
 526          GtkTreeViewColumn *arg2, gpointer user_data)
 527 {
 528         GtkWidget *entry_name, *entry_type, *entry_value;
 529         GtkDialog *addwin = GTK_DIALOG(create_SetValueDialog(&entry_name, &entry_type, &entry_value));
 530         GtkTreeIter iter;
 531         struct registry_value *value;
 532         gint result;
 533
 534         gtk_tree_model_get_iter(GTK_TREE_MODEL(store_vals), &iter, arg1);
 535
 536         gtk_tree_model_get(GTK_TREE_MODEL(store_vals), &iter, 3, &value, -1);
 537
 538         gtk_widget_set_sensitive(entry_name, FALSE);
 539         gtk_entry_set_text(GTK_ENTRY(entry_name), value->name);
 540         gtk_entry_set_text(GTK_ENTRY(entry_value), reg_val_data_string(mem_ctx, value->data_type, &value->data));
 541         gtk_combo_box_set_active(GTK_COMBO_BOX(entry_type), value->data_type);
 542
 543         result = gtk_dialog_run(addwin);
 544         if (result == GTK_RESPONSE_OK)
 545         {
 546                 WERROR error;
 547                 DATA_BLOB data;
 548                 uint32_t data_type;
 549
 550                 reg_string_to_val(mem_ctx,str_regtype(gtk_combo_box_get_active(GTK_COMBO_BOX(entry_type))), gtk_entry_get_text(GTK_ENTRY(entry_value)), &data_type, &data);
 551
 552                 error = reg_val_set(current_key, gtk_entry_get_text(GTK_ENTRY(entry_name)), data_type, data);
 553
 554                 if (!W_ERROR_IS_OK(error)) {
 555                         gtk_show_werror(NULL, "Error while setting value", error);
 556                 }
 557         }
 558         gtk_widget_destroy(GTK_WIDGET(addwin));
 559 }
 560
 561 static void on_set_value_activate(GtkMenuItem *menuitem, gpointer user_data)
 562 {
 563         GtkWidget *entry_name, *entry_type, *entry_value;
 564         GtkDialog *addwin = GTK_DIALOG(create_SetValueDialog(&entry_name, &entry_type, &entry_value));
 565         gint result = gtk_dialog_run(addwin);
 566         if (result == GTK_RESPONSE_OK)
 567         {
 568                 WERROR error;
 569                 uint32_t data_type;
 570                 DATA_BLOB data;
 571
 572                 reg_string_to_val(mem_ctx,str_regtype(gtk_combo_box_get_active(GTK_COMBO_BOX(entry_type))), gtk_entry_get_text(GTK_ENTRY(entry_value)), &data_type, &data);
 573
 574                 error = reg_val_set(current_key, gtk_entry_get_text(GTK_ENTRY(entry_name)), data_type, data);
 575
 576                 if (!W_ERROR_IS_OK(error)) {
 577                         gtk_show_werror(NULL, "Error while setting value", error);
 578                 }
 579         }
 580         gtk_widget_destroy(GTK_WIDGET(addwin));
 581 }
 582
 583 static void on_find_activate(GtkMenuItem *menuitem, gpointer user_data)
 584 {
 585         GtkDialog *findwin = GTK_DIALOG(create_FindDialog());
 586         /*gint result = gtk_dialog_run(findwin);
 587         FIXME */
 588         gtk_widget_destroy(GTK_WIDGET(findwin));
 589 }
 590
 591 static void on_about_activate (GtkMenuItem *menuitem, gpointer user_data)
 592 {
 593     GtkDialog *aboutwin = GTK_DIALOG(create_gtk_samba_about_dialog("gregedit"));
 594     gtk_dialog_run(aboutwin);
 595     gtk_widget_destroy(GTK_WIDGET(aboutwin));
 596 }
 597
 598 static gboolean on_key_activate(GtkTreeSelection *selection,
 599                                              GtkTreeModel *model,
 600                                              GtkTreePath *path,
 601                                              gboolean path_currently_selected,
 602                                              gpointer data)
 603 {
 604         int i;
 605         struct registry_key *k;
 606         struct registry_value *val;
 607         WERROR error;
 608         GtkTreeIter parent;
 609
 610         gtk_widget_set_sensitive(mnu_add_key, !path_currently_selected);
 611         gtk_widget_set_sensitive(mnu_set_value, !path_currently_selected);
 612         gtk_widget_set_sensitive(mnu_del_key, !path_currently_selected);
 613         gtk_widget_set_sensitive(mnu_del_value, !path_currently_selected);
 614         gtk_widget_set_sensitive(mnu_find, !path_currently_selected);
 615
 616         if(path_currently_selected) {
 617                 current_key = NULL;
 618                 return TRUE;
 619         }
 620
 621         gtk_tree_model_get_iter(GTK_TREE_MODEL(store_keys), &parent, path);
 622         gtk_tree_model_get(GTK_TREE_MODEL(store_keys), &parent, 1, &k, -1);
 623
 624         current_key = k;
 625
 626         if (!k) return FALSE;
 627
 628         gtk_list_store_clear(store_vals);
 629
 630         for(i = 0; W_ERROR_IS_OK(error = reg_key_get_value_by_index(mem_ctx, k, i, &val)); i++) {
 631                 GtkTreeIter iter;
 632                 gtk_list_store_append(store_vals, &iter);
 633                 gtk_list_store_set (store_vals,
 634                                             &iter,
 635                                                 0,
 636                                                 val->name,
 637                                                 1,
 638                                                 str_regtype(val->data_type),
 639                                                 2,
 640                                                 reg_val_data_string(mem_ctx, val->data_type, &val->data),
 641                                                 3,
 642                                                 val,
 643                                                 -1);
 644         }
 645
 646         if(!W_ERROR_EQUAL(error, WERR_NO_MORE_ITEMS)) {
 647                  gtk_show_werror(mainwin, "Error while enumerating values",  error);
 648                  return FALSE;
 649         }
 650         return TRUE;
 651 }
 652
 653 static GtkWidget* create_mainwindow(void)
 654 {
 655         GtkWidget *vbox1;
 656         GtkWidget *menubar;
 657         GtkWidget *menu_file;
 658         GtkWidget *menu_file_menu;
 659         GtkWidget *open_nt4;
 660         GtkWidget *open_ldb;
 661         GtkWidget *open_w95;
 662         GtkWidget *open_gconf;
 663         GtkWidget *open_remote;
 664         GtkWidget *open_local;
 665         GtkWidget *separatormenuitem1;
 666         GtkWidget *quit;
 667         GtkWidget *men_key;
 668         GtkWidget *men_key_menu;
 669         GtkCellRenderer *renderer;
 670         GtkTreeViewColumn *curcol;
 671         GtkWidget *help;
 672         GtkWidget *help_menu;
 673         GtkWidget *about;
 674         GtkWidget *hbox1;
 675         GtkWidget *scrolledwindow1;
 676         GtkWidget *scrolledwindow2;
 677         GtkWidget *statusbar;
 678         GtkAccelGroup *accel_group;
 679
 680         accel_group = gtk_accel_group_new ();
 681
 682         mainwin = gtk_window_new (GTK_WINDOW_TOPLEVEL);
 683         gtk_window_set_title (GTK_WINDOW (mainwin), "Registry editor");
 684         gtk_window_set_default_size (GTK_WINDOW (mainwin), 642, 562);
 685
 686         vbox1 = gtk_vbox_new (FALSE, 0);
 687         gtk_container_add (GTK_CONTAINER (mainwin), vbox1);
 688
 689         menubar = gtk_menu_bar_new ();
 690         gtk_box_pack_start (GTK_BOX (vbox1), menubar, FALSE, FALSE, 0);
 691
 692         menu_file = gtk_menu_item_new_with_mnemonic ("_File");
 693         gtk_container_add (GTK_CONTAINER (menubar), menu_file);
 694
 695         menu_file_menu = gtk_menu_new ();
 696         gtk_menu_item_set_submenu (GTK_MENU_ITEM (menu_file), menu_file_menu);
 697
 698         open_local = gtk_menu_item_new_with_mnemonic ("Open _Local");
 699         gtk_container_add (GTK_CONTAINER (menu_file_menu), open_local);
 700         g_signal_connect ((gpointer) open_local, "activate",
 701                                                   G_CALLBACK (on_open_local_activate), NULL);
 702
 703         if(reg_has_backend("rpc")) {
 704                 open_remote = gtk_menu_item_new_with_mnemonic ("Open _Remote");
 705                 gtk_container_add (GTK_CONTAINER (menu_file_menu), open_remote);
 706
 707                 g_signal_connect ((gpointer) open_remote, "activate",
 708                                                   G_CALLBACK (on_open_remote_activate),
 709                                                   NULL);
 710         }
 711
 712         separatormenuitem1 = gtk_menu_item_new ();
 713         gtk_container_add (GTK_CONTAINER (menu_file_menu), separatormenuitem1);
 714         gtk_widget_set_sensitive (separatormenuitem1, FALSE);
 715
 716
 717         if(reg_has_backend("nt4")) {
 718                 open_nt4 = gtk_image_menu_item_new_with_mnemonic("Open _NT4 file");
 719                 gtk_container_add (GTK_CONTAINER (menu_file_menu), open_nt4);
 720
 721                 g_signal_connect(open_nt4, "activate",
 722                                  G_CALLBACK (on_open_file_activate),
 723                                  discard_const_p(char, "nt4"));
 724         }
 725
 726         if(reg_has_backend("w95")) {
 727                 open_w95 = gtk_image_menu_item_new_with_mnemonic("Open Win_9x file");
 728                 gtk_container_add (GTK_CONTAINER (menu_file_menu), open_w95);
 729
 730                 g_signal_connect (open_w95, "activate",
 731                                   G_CALLBACK (on_open_file_activate),
 732                                   discard_const_p(char, "w95"));
 733         }
 734
 735         if(reg_has_backend("gconf")) {
 736                 open_gconf = gtk_image_menu_item_new_with_mnemonic ("Open _GConf");
 737                 gtk_container_add (GTK_CONTAINER (menu_file_menu), open_gconf);
 738
 739                 g_signal_connect ((gpointer) open_gconf, "activate",
 740                                                   G_CALLBACK (on_open_gconf_activate),
 741                                                   NULL);
 742         }
 743
 744         if(reg_has_backend("ldb")) {
 745                 open_ldb = gtk_image_menu_item_new_with_mnemonic("Open _LDB file");
 746                 gtk_container_add (GTK_CONTAINER (menu_file_menu), open_ldb);
 747
 748                 g_signal_connect(open_ldb, "activate",
 749                                  G_CALLBACK (on_open_file_activate),
 750                                  discard_const_p(char, "ldb"));
 751         }
 752
 753         separatormenuitem1 = gtk_menu_item_new ();
 754         gtk_container_add (GTK_CONTAINER (menu_file_menu), separatormenuitem1);
 755         gtk_widget_set_sensitive (separatormenuitem1, FALSE);
 756
 757         save = gtk_image_menu_item_new_from_stock ("gtk-save", accel_group);
 758         gtk_widget_set_sensitive( save, False );
 759         gtk_container_add (GTK_CONTAINER (menu_file_menu), save);
 760
 761         save_as = gtk_image_menu_item_new_from_stock ("gtk-save-as", accel_group);
 762         gtk_widget_set_sensitive( save_as, False );
 763         gtk_container_add (GTK_CONTAINER (menu_file_menu), save_as);
 764
 765         separatormenuitem1 = gtk_menu_item_new ();
 766         gtk_container_add (GTK_CONTAINER (menu_file_menu), separatormenuitem1);
 767         gtk_widget_set_sensitive (separatormenuitem1, FALSE);
 768
 769         quit = gtk_image_menu_item_new_from_stock ("gtk-quit", accel_group);
 770         gtk_container_add (GTK_CONTAINER (menu_file_menu), quit);
 771
 772         men_key = gtk_menu_item_new_with_mnemonic ("_Key");
 773         gtk_container_add (GTK_CONTAINER (menubar), men_key);
 774
 775         men_key_menu = gtk_menu_new ();
 776         gtk_menu_item_set_submenu (GTK_MENU_ITEM (men_key), men_key_menu);
 777
 778         mnu_add_key = gtk_image_menu_item_new_with_mnemonic("Add _Subkey");
 779         gtk_image_menu_item_set_image (GTK_IMAGE_MENU_ITEM (mnu_add_key), gtk_image_new_from_stock ("gtk-add", GTK_ICON_SIZE_MENU));
 780
 781         gtk_widget_set_sensitive(mnu_add_key, False);
 782         gtk_container_add (GTK_CONTAINER (men_key_menu), mnu_add_key);
 783
 784         mnu_set_value = gtk_image_menu_item_new_with_mnemonic("Set _Value");
 785         gtk_widget_set_sensitive(mnu_set_value, False);
 786         gtk_image_menu_item_set_image (GTK_IMAGE_MENU_ITEM (mnu_set_value), gtk_image_new_from_stock ("gtk-add", GTK_ICON_SIZE_MENU));
 787         gtk_container_add (GTK_CONTAINER (men_key_menu), mnu_set_value);
 788
 789         mnu_find = gtk_image_menu_item_new_from_stock ("gtk-find", accel_group);
 790         gtk_widget_set_sensitive(mnu_find, False);
 791         gtk_container_add (GTK_CONTAINER (men_key_menu), mnu_find);
 792
 793         mnu_del_key = gtk_image_menu_item_new_with_mnemonic ("Delete Key");
 794         gtk_widget_set_sensitive(mnu_del_key, False);
 795         gtk_image_menu_item_set_image (GTK_IMAGE_MENU_ITEM (mnu_del_value), gtk_image_new_from_stock ("gtk-delete", GTK_ICON_SIZE_MENU));
 796         gtk_container_add (GTK_CONTAINER (men_key_menu), mnu_del_key);
 797
 798         mnu_del_value = gtk_image_menu_item_new_with_mnemonic ("Delete Value");
 799         gtk_widget_set_sensitive(mnu_del_value, False);
 800         gtk_image_menu_item_set_image (GTK_IMAGE_MENU_ITEM (mnu_del_value), gtk_image_new_from_stock ("gtk-delete", GTK_ICON_SIZE_MENU));
 801         gtk_container_add (GTK_CONTAINER (men_key_menu), mnu_del_value);
 802
 803
 804         help = gtk_menu_item_new_with_mnemonic ("_Help");
 805         gtk_container_add (GTK_CONTAINER (menubar), help);
 806
 807         help_menu = gtk_menu_new ();
 808         gtk_menu_item_set_submenu (GTK_MENU_ITEM (help), help_menu);
 809
 810         about = gtk_menu_item_new_with_mnemonic ("_About");
 811         gtk_container_add (GTK_CONTAINER (help_menu), about);
 812
 813         hbox1 = gtk_hbox_new (FALSE, 0);
 814         gtk_box_pack_start (GTK_BOX (vbox1), hbox1, TRUE, TRUE, 0);
 815
 816         scrolledwindow1 = gtk_scrolled_window_new (NULL, NULL);
 817         gtk_box_pack_start (GTK_BOX (hbox1), scrolledwindow1, TRUE, TRUE, 0);
 818
 819         tree_keys = gtk_tree_view_new ();
 820
 821         /* Column names */
 822         curcol = gtk_tree_view_column_new ();
 823         gtk_tree_view_column_set_title(curcol, "Name");
 824         renderer = gtk_cell_renderer_text_new();
 825         gtk_tree_view_column_pack_start(curcol, renderer, True);
 826
 827         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_keys), curcol);
 828
 829         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 0);
 830         gtk_container_add (GTK_CONTAINER (scrolledwindow1), tree_keys);
 831         store_keys = gtk_tree_store_new(2, G_TYPE_STRING, G_TYPE_POINTER);
 832         gtk_tree_view_set_model(GTK_TREE_VIEW(tree_keys), GTK_TREE_MODEL(store_keys));
 833         g_object_unref(store_keys);
 834
 835         gtk_tree_selection_set_select_function (gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_keys)), on_key_activate, NULL, NULL);
 836
 837         g_signal_connect ((gpointer) tree_keys, "row-expanded",
 838                                           G_CALLBACK (expand_key),
 839                                           NULL);
 840
 841
 842         scrolledwindow2 = gtk_scrolled_window_new (NULL, NULL);
 843         gtk_box_pack_start (GTK_BOX (hbox1), scrolledwindow2, TRUE, TRUE, 0);
 844
 845         tree_vals = gtk_tree_view_new ();
 846         /* Column names */
 847
 848         curcol = gtk_tree_view_column_new ();
 849         gtk_tree_view_column_set_title(curcol, "Name");
 850         renderer = gtk_cell_renderer_text_new();
 851         gtk_tree_view_column_pack_start(curcol, renderer, True);
 852         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_vals), curcol);
 853         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 0);
 854
 855         curcol = gtk_tree_view_column_new ();
 856         gtk_tree_view_column_set_title(curcol, "Type");
 857         renderer = gtk_cell_renderer_text_new();
 858         gtk_tree_view_column_pack_start(curcol, renderer, True);
 859         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_vals), curcol);
 860         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 1);
 861
 862         curcol = gtk_tree_view_column_new ();
 863         gtk_tree_view_column_set_title(curcol, "Value");
 864         renderer = gtk_cell_renderer_text_new();
 865         gtk_tree_view_column_pack_start(curcol, renderer, True);
 866         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_vals), curcol);
 867         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 2);
 868
 869
 870         gtk_container_add (GTK_CONTAINER (scrolledwindow2), tree_vals);
 871
 872         store_vals = gtk_list_store_new(4, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_POINTER);
 873         gtk_tree_view_set_model(GTK_TREE_VIEW(tree_vals), GTK_TREE_MODEL(store_vals));
 874         g_object_unref(store_vals);
 875
 876         statusbar = gtk_statusbar_new ();
 877         gtk_box_pack_start (GTK_BOX (vbox1), statusbar, FALSE, FALSE, 0);
 878         gtk_statusbar_set_has_resize_grip (GTK_STATUSBAR (statusbar), FALSE);
 879
 880         g_signal_connect ((gpointer) save_as, "activate",
 881                                           G_CALLBACK (on_save_as_activate),
 882                                           NULL);
 883         g_signal_connect ((gpointer) quit, "activate",
 884                                           G_CALLBACK (on_quit_activate),
 885                                           NULL);
 886         g_signal_connect ((gpointer) mnu_add_key, "activate",
 887                                           G_CALLBACK (on_add_key_activate),
 888                                           NULL);
 889         g_signal_connect ((gpointer) mnu_set_value, "activate",
 890                                           G_CALLBACK (on_set_value_activate),
 891                                           NULL);
 892         g_signal_connect ((gpointer) mnu_find, "activate",
 893                                           G_CALLBACK (on_find_activate),
 894                                           NULL);
 895         g_signal_connect ((gpointer) mnu_del_key, "activate",
 896                                           G_CALLBACK (on_delete_key_activate),
 897                                           NULL);
 898         g_signal_connect ((gpointer) mnu_del_value, "activate",
 899                                           G_CALLBACK (on_delete_value_activate),
 900                                           NULL);
 901         g_signal_connect ((gpointer) about, "activate",
 902                                           G_CALLBACK (on_about_activate),
 903                                           NULL);
 904
 905         g_signal_connect ((gpointer) tree_vals, "row-activated",
 906                                           G_CALLBACK (on_value_activate),
 907                                           NULL);
 908
 909
 910         gtk_window_add_accel_group (GTK_WINDOW (mainwin), accel_group);
 911
 912         return mainwin;
 913 }
 914
 915 static GtkWidget* create_openfilewin (void)
 916 {
 917         GtkWidget *openfilewin;
 918         GtkWidget *ok_button;
 919         GtkWidget *cancel_button;
 920
 921         openfilewin = gtk_file_selection_new ("Select File");
 922         gtk_container_set_border_width (GTK_CONTAINER (openfilewin), 10);
 923
 924         ok_button = GTK_FILE_SELECTION (openfilewin)->ok_button;
 925         GTK_WIDGET_SET_FLAGS (ok_button, GTK_CAN_DEFAULT);
 926
 927         cancel_button = GTK_FILE_SELECTION (openfilewin)->cancel_button;
 928         GTK_WIDGET_SET_FLAGS (cancel_button, GTK_CAN_DEFAULT);
 929
 930         return openfilewin;
 931 }
 932
 933 static GtkWidget* create_savefilewin (void)
 934 {
 935         GtkWidget *savefilewin;
 936         GtkWidget *ok_button;
 937         GtkWidget *cancel_button;
 938
 939         savefilewin = gtk_file_selection_new ("Select File");
 940         gtk_container_set_border_width (GTK_CONTAINER (savefilewin), 10);
 941
 942         ok_button = GTK_FILE_SELECTION (savefilewin)->ok_button;
 943         GTK_WIDGET_SET_FLAGS (ok_button, GTK_CAN_DEFAULT);
 944
 945         cancel_button = GTK_FILE_SELECTION (savefilewin)->cancel_button;
 946         GTK_WIDGET_SET_FLAGS (cancel_button, GTK_CAN_DEFAULT);
 947
 948         return savefilewin;
 949 }
 950
 951 static int gregedit_load_defaults(void)
 952 {
 953         WERROR error = reg_open_local(&registry);
 954         if(!W_ERROR_IS_OK(error)) {
 955                 gtk_show_werror(mainwin, "Error while loading local registry", error);
 956                 return -1;
 957         }
 958         registry_load_root();
 959
 960         return 0;
 961 }
 962
 963 int main(int argc, char *argv[])
 964 {
 965         int ret;
 966
 967         gregedit_init_subsystems;
 968         registry_init();
 969         lp_load();
 970         setup_logging(argv[0], DEBUG_STDERR);
 971
 972         mem_ctx = talloc_init("gregedit");
 973
 974         gtk_init(&argc, &argv);
 975         mainwin = create_mainwindow();
 976         gtk_widget_show_all(mainwin);
 977
 978         ret = gregedit_load_defaults();
 979         if (ret != 0) goto failed;
 980
 981         ret = gtk_event_loop();
 982
 983 failed:
 984         talloc_free(mem_ctx);
 985         return ret;
 986 }