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