1 files changed, 235 insertions, 136 deletions

diff --git a/source4/gtk/tools/gepdump.c b/source4/gtk/tools/gepdump.c
index 1ad84b99f5..6e5a9757d4 100644
--- a/source4/gtk/tools/gepdump.c
+++ b/source4/gtk/tools/gepdump.c
@@ -22,6 +22,7 @@
 #include "includes.h"
 #include "dynconfig.h"
 #include "librpc/gen_ndr/ndr_epmapper.h"
+#include "librpc/gen_ndr/ndr_mgmt.h"
 #include "gtk/common/select.h"
 #include "gtk/common/gtk-smb.h"
 
@@ -36,6 +37,10 @@
 static GtkWidget *mainwin;
 static GtkWidget *entry_binding;
 static GtkTreeStore *store_eps;
+static GtkWidget *table_statistics;
+static GtkWidget *lbl_calls_in, *lbl_calls_out, *lbl_pkts_in, *lbl_pkts_out;
+static GtkWidget *lbl_iface_version, *lbl_iface_uuid, *lbl_iface_name;
+TALLOC_CTX *eps_ctx = NULL;
 
 static void on_quit1_activate (GtkMenuItem *menuitem, gpointer user_data)
 {
@@ -84,7 +89,7 @@ static void add_epm_entry(TALLOC_CTX *mem_ctx, const char *annotation, struct ep
 	ZERO_STRUCT(bd.object);
 
 	gtk_tree_store_append(store_eps, &toweriter, NULL);
-	gtk_tree_store_set(store_eps, &toweriter, 0, strdup(annotation), 1, strdup(dcerpc_binding_string(mem_ctx, &bd)), -1);
+	gtk_tree_store_set(store_eps, &toweriter, 0, strdup(annotation), 1, strdup(dcerpc_binding_string(mem_ctx, &bd)), 2, t, -1);
 
 	for (i = 0; i < t->num_floors; i++) {
 		const char *data;
@@ -111,6 +116,8 @@ static void on_dump_clicked (GtkButton *btn, gpointer user_data)
 	struct dcerpc_pipe *p;
 	TALLOC_CTX *mem_ctx = talloc_init("dump");
 
+	talloc_destroy(eps_ctx);
+
 	status = dcerpc_pipe_connect(&p, gtk_entry_get_text(GTK_ENTRY(entry_binding)), DCERPC_EPMAPPER_UUID, DCERPC_EPMAPPER_VERSION, lp_workgroup(), "", "");
 
 	if (NT_STATUS_IS_ERR(status)) {
@@ -133,9 +140,11 @@ static void on_dump_clicked (GtkButton *btn, gpointer user_data)
 
 	gtk_tree_store_clear(store_eps);
 
+	eps_ctx = talloc_init("current endpoint list data");
+
 	do {
 		int i;
-		status = dcerpc_epm_Lookup(p, mem_ctx, &r);
+		status = dcerpc_epm_Lookup(p, eps_ctx, &r);
 		if (!NT_STATUS_IS_OK(status) || r.out.result != 0) {
 			break;
 		}
@@ -158,7 +167,6 @@ static void on_select_target_clicked(GtkButton *btn, gpointer         user_data)
 {
 	GtkRpcBindingDialog *d;
 	TALLOC_CTX *mem_ctx;
-	struct dcerpc_binding *bd;
 	gint result;
 
 	d = GTK_RPC_BINDING_DIALOG(gtk_rpc_binding_dialog_new(TRUE, NULL));
@@ -172,155 +180,246 @@ static void on_select_target_clicked(GtkButton *btn, gpointer         user_data)
 	}
 
 	mem_ctx = talloc_init("select_target");
-	bd = gtk_rpc_binding_dialog_get_binding (d, mem_ctx),
-	gtk_entry_set_text(GTK_ENTRY(entry_binding), dcerpc_binding_string(mem_ctx, bd));
+	gtk_entry_set_text(GTK_ENTRY(entry_binding), gtk_rpc_binding_dialog_get_binding_string (d, mem_ctx));
 	talloc_destroy(mem_ctx);
 	gtk_widget_destroy(GTK_WIDGET(d));
 }
 
+static gboolean on_eps_select(GtkTreeSelection *selection,
+    GtkTreeModel *model, GtkTreePath *path, gboolean path_currently_selected, gpointer data)
+{
+	struct dcerpc_pipe *p;
+	NTSTATUS status;
+	struct dcerpc_binding bd;
+	TALLOC_CTX *mem_ctx = talloc_init("eps");
+
+	ZERO_STRUCT(bd);
+	bd.host = "192.168.4.26";
+	bd.endpoint = "samr";
+	bd.transport = NCACN_NP;
+
+	status = dcerpc_pipe_connect_b(&p, &bd, DCERPC_MGMT_UUID, DCERPC_MGMT_VERSION, "", "administrator", "penguin");
+
+	if (NT_STATUS_IS_ERR(status)) {
+		gtk_show_ntstatus(NULL, status);
+		return FALSE;
+	}
+	
+	{
+		/* Do an InqStats call */
+		struct mgmt_inq_stats r;
+
+		r.in.max_count = MGMT_STATS_ARRAY_MAX_SIZE;
+		r.in.unknown = 0;
+
+		status = dcerpc_mgmt_inq_stats(p, mem_ctx, &r);
+		if (NT_STATUS_IS_ERR(status)) {
+			gtk_show_ntstatus(NULL, status);
+			return TRUE;
+		}
+
+		if (r.out.statistics.count != MGMT_STATS_ARRAY_MAX_SIZE) {
+			printf("Unexpected array size %d\n", r.out.statistics.count);
+			return False;
+		}
+
+		gtk_label_set_text(GTK_LABEL(lbl_calls_in), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_CALLS_IN]));
+		gtk_label_set_text(GTK_LABEL(lbl_calls_out), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_CALLS_OUT]));
+		gtk_label_set_text(GTK_LABEL(lbl_pkts_in), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_PKTS_IN]));
+		gtk_label_set_text(GTK_LABEL(lbl_pkts_out), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_PKTS_OUT]));
+	}
+
+	{
+		struct mgmt_inq_princ_name r;
+		int i;
+
+		for (i=0;i<100;i++) {
+			r.in.authn_proto = i;  /* DCERPC_AUTH_TYPE_* */
+			r.in.princ_name_size = 100;
+
+			status = dcerpc_mgmt_inq_princ_name(p, mem_ctx, &r);
+			if (!NT_STATUS_IS_OK(status)) {
+				continue;
+			}
+			if (W_ERROR_IS_OK(r.out.result)) {
+				const char *name = gensec_get_name_by_authtype(i);
+				if (name) {
+					printf("\tprinciple name for proto %u (%s) is '%s'\n",
+						   i, name, r.out.princ_name);
+				} else {
+					printf("\tprinciple name for proto %u is '%s'\n",
+						   i, r.out.princ_name);
+				}
+			}
+		}
+	}
+
+	return TRUE;
+}
+
+
 static GtkWidget* create_mainwindow (void)
 {
-  GtkWidget *mainwindow;
-  GtkWidget *vbox1;
-  GtkWidget *vbox2;
-  GtkWidget *menubar1;
-  GtkWidget *menuitem1;
-  GtkWidget *menuitem1_menu;
-  GtkWidget *quit1;
-  GtkWidget *menuitem4;
-  GtkWidget *menuitem4_menu;
-  GtkWidget *about1;
-  GtkWidget *handlebox1;
-  GtkWidget *hbox1;
-  GtkWidget *hbox2;
-  GtkWidget *label1;
-  GtkWidget *btn_select_target;
-  GtkWidget *btn_dump;
-  GtkWidget *scrolledwindow1;
-  GtkWidget *frame1;
-  GtkWidget *tree_eps;
-  GtkTreeViewColumn *curcol;
-  GtkCellRenderer *renderer;
-  GtkWidget *statusbar;
-  GtkAccelGroup *accel_group;
-
-  accel_group = gtk_accel_group_new ();
-
-  mainwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
-  gtk_window_set_title (GTK_WINDOW (mainwindow), "Gtk+ Endpoint Mapper Viewer");
-
-  vbox1 = gtk_vbox_new (FALSE, 0);
-  gtk_widget_show (vbox1);
-  gtk_container_add (GTK_CONTAINER (mainwindow), vbox1);
-
-  menubar1 = gtk_menu_bar_new ();
-  gtk_widget_show (menubar1);
-  gtk_box_pack_start (GTK_BOX (vbox1), menubar1, FALSE, FALSE, 0);
-
-  menuitem1 = gtk_menu_item_new_with_mnemonic ("_File");
-  gtk_widget_show (menuitem1);
-  gtk_container_add (GTK_CONTAINER (menubar1), menuitem1);
-
-  menuitem1_menu = gtk_menu_new ();
-  gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem1), menuitem1_menu);
-
-  quit1 = gtk_image_menu_item_new_from_stock ("gtk-quit", accel_group);
-  gtk_container_add (GTK_CONTAINER (menuitem1_menu), quit1);
-
-  menuitem4 = gtk_menu_item_new_with_mnemonic ("_Help");
-  gtk_container_add (GTK_CONTAINER (menubar1), menuitem4);
-
-  menuitem4_menu = gtk_menu_new ();
-  gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem4), menuitem4_menu);
-
-  about1 = gtk_menu_item_new_with_mnemonic ("_About");
-  gtk_container_add (GTK_CONTAINER (menuitem4_menu), about1);
-
-  handlebox1 = gtk_handle_box_new ();
-  gtk_box_pack_start (GTK_BOX (vbox1), handlebox1, FALSE, TRUE, 0);
-
-  hbox1 = gtk_hbox_new (FALSE, 0);
-  gtk_container_add (GTK_CONTAINER (handlebox1), hbox1);
-
-  label1 = gtk_label_new ("Location:");
-  gtk_box_pack_start (GTK_BOX (hbox1), label1, FALSE, FALSE, 0);
-
-  entry_binding = gtk_entry_new ();
-  gtk_entry_set_text(GTK_ENTRY(entry_binding), "ncalrpc:");
-  gtk_box_pack_start (GTK_BOX (hbox1), entry_binding, FALSE, FALSE, 0);
-
-  btn_select_target = gtk_button_new_with_mnemonic ("_Select Target");
-  gtk_box_pack_start (GTK_BOX (hbox1), btn_select_target, FALSE, FALSE, 0);
-
-  btn_dump = gtk_button_new_with_mnemonic ("_Dump");
-  gtk_box_pack_start (GTK_BOX (hbox1), btn_dump, FALSE, FALSE, 0);
-
-  hbox2 = gtk_hbox_new (FALSE, 0);
-  gtk_container_add (GTK_CONTAINER (vbox1), hbox2);
-
-  scrolledwindow1 = gtk_scrolled_window_new (NULL, NULL);
-  gtk_box_pack_start (GTK_BOX(hbox2), scrolledwindow1, TRUE, TRUE, 0);
-  
-  tree_eps = gtk_tree_view_new ();
-
-  curcol = gtk_tree_view_column_new ();
-  gtk_tree_view_column_set_title(curcol, "Name");
-  renderer = gtk_cell_renderer_text_new();
-  gtk_tree_view_column_pack_start(curcol, renderer, True);
-
-  gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
-  gtk_tree_view_column_add_attribute(curcol, renderer, "text", 0);
-
-  curcol = gtk_tree_view_column_new ();
-  gtk_tree_view_column_set_title(curcol, "Binding String");
-  renderer = gtk_cell_renderer_text_new();
-  gtk_tree_view_column_pack_start(curcol, renderer, True);
-  gtk_tree_view_column_add_attribute(curcol, renderer, "text", 1);
-
-
-  gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
-
-  store_eps = gtk_tree_store_new(2, G_TYPE_STRING, G_TYPE_STRING);
-  gtk_tree_view_set_model(GTK_TREE_VIEW(tree_eps), GTK_TREE_MODEL(store_eps));
-  g_object_unref(store_eps);
-  
-  gtk_container_add (GTK_CONTAINER (scrolledwindow1), tree_eps);
-
-  vbox2 = gtk_vbox_new (FALSE, 0);
-  gtk_container_add (GTK_CONTAINER (hbox2), vbox2);
+	GtkWidget *mainwindow;
+	GtkWidget *vbox1, *vbox2, *vbox3;
+	GtkWidget *menubar1;
+	GtkWidget *menuitem1;
+	GtkWidget *menuitem1_menu;
+	GtkWidget *quit1;
+	GtkWidget *menuitem4;
+	GtkWidget *menuitem4_menu;
+	GtkWidget *about1;
+	GtkWidget *handlebox1;
+	GtkWidget *hbox1;
+	GtkWidget *hbox2;
+	GtkWidget *label1;
+	GtkWidget *btn_select_target;
+	GtkWidget *btn_dump;
+	GtkWidget *scrolledwindow1;
+	GtkWidget *frame1;
+	GtkWidget *tree_eps;
+	GtkTreeViewColumn *curcol;
+	GtkCellRenderer *renderer;
+	GtkWidget *statusbar;
+	GtkAccelGroup *accel_group;
 
-  frame1 = gtk_frame_new("Interface");
-  gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+	accel_group = gtk_accel_group_new ();
 
-  frame1 = gtk_frame_new("Statistics");
-  gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+	mainwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
+	gtk_window_set_title (GTK_WINDOW (mainwindow), "Gtk+ Endpoint Mapper Viewer");
 
-  frame1 = gtk_frame_new("Authentication");
-  gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+	vbox1 = gtk_vbox_new (FALSE, 0);
+	gtk_widget_show (vbox1);
+	gtk_container_add (GTK_CONTAINER (mainwindow), vbox1);
 
-  statusbar = gtk_statusbar_new ();
-  gtk_box_pack_start (GTK_BOX (vbox1), statusbar, FALSE, FALSE, 0);
+	menubar1 = gtk_menu_bar_new ();
+	gtk_widget_show (menubar1);
+	gtk_box_pack_start (GTK_BOX (vbox1), menubar1, FALSE, FALSE, 0);
 
-  g_signal_connect ((gpointer) quit1, "activate",
-                    G_CALLBACK (on_quit1_activate),
-                    NULL);
-  g_signal_connect ((gpointer) about1, "activate",
-                    G_CALLBACK (on_about1_activate),
-                    NULL);
-  g_signal_connect ((gpointer) btn_select_target, "clicked",
-                    G_CALLBACK (on_select_target_clicked),
-                    NULL);
-  g_signal_connect ((gpointer) btn_dump, "clicked",
-                    G_CALLBACK (on_dump_clicked),
-                    NULL);
+	menuitem1 = gtk_menu_item_new_with_mnemonic ("_File");
+	gtk_widget_show (menuitem1);
+	gtk_container_add (GTK_CONTAINER (menubar1), menuitem1);
 
-  gtk_window_add_accel_group (GTK_WINDOW (mainwindow), accel_group);
+	menuitem1_menu = gtk_menu_new ();
+	gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem1), menuitem1_menu);
 
-  return mainwindow;
+	quit1 = gtk_image_menu_item_new_from_stock ("gtk-quit", accel_group);
+	gtk_container_add (GTK_CONTAINER (menuitem1_menu), quit1);
+
+	menuitem4 = gtk_menu_item_new_with_mnemonic ("_Help");
+	gtk_container_add (GTK_CONTAINER (menubar1), menuitem4);
+
+	menuitem4_menu = gtk_menu_new ();
+	gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem4), menuitem4_menu);
+
+	about1 = gtk_menu_item_new_with_mnemonic ("_About");
+	gtk_container_add (GTK_CONTAINER (menuitem4_menu), about1);
+
+	handlebox1 = gtk_handle_box_new ();
+	gtk_box_pack_start (GTK_BOX (vbox1), handlebox1, FALSE, TRUE, 0);
+
+	hbox1 = gtk_hbox_new (FALSE, 0);
+	gtk_container_add (GTK_CONTAINER (handlebox1), hbox1);
+
+	label1 = gtk_label_new ("Location:");
+	gtk_box_pack_start (GTK_BOX (hbox1), label1, FALSE, FALSE, 0);
+
+	entry_binding = gtk_entry_new ();
+	gtk_entry_set_text(GTK_ENTRY(entry_binding), "ncalrpc:");
+	gtk_box_pack_start (GTK_BOX (hbox1), entry_binding, FALSE, FALSE, 0);
+
+	btn_select_target = gtk_button_new_with_mnemonic ("_Select Target");
+	gtk_box_pack_start (GTK_BOX (hbox1), btn_select_target, FALSE, FALSE, 0);
+
+	btn_dump = gtk_button_new_with_mnemonic ("_Dump");
+	gtk_box_pack_start (GTK_BOX (hbox1), btn_dump, FALSE, FALSE, 0);
+
+	hbox2 = gtk_hbox_new (FALSE, 0);
+	gtk_container_add (GTK_CONTAINER (vbox1), hbox2);
+
+	scrolledwindow1 = gtk_scrolled_window_new (NULL, NULL);
+	gtk_box_pack_start (GTK_BOX(hbox2), scrolledwindow1, TRUE, TRUE, 0);
+
+	tree_eps = gtk_tree_view_new ();
+
+	curcol = gtk_tree_view_column_new ();
+	gtk_tree_view_column_set_title(curcol, "Name");
+	renderer = gtk_cell_renderer_text_new();
+	gtk_tree_view_column_pack_start(curcol, renderer, True);
+
+	gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
+	gtk_tree_view_column_add_attribute(curcol, renderer, "text", 0);
+
+	curcol = gtk_tree_view_column_new ();
+	gtk_tree_view_column_set_title(curcol, "Binding String");
+	renderer = gtk_cell_renderer_text_new();
+	gtk_tree_view_column_pack_start(curcol, renderer, True);
+	gtk_tree_view_column_add_attribute(curcol, renderer, "text", 1);
+
+
+	gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
+
+	store_eps = gtk_tree_store_new(3, GTK_TYPE_STRING, GTK_TYPE_STRING, GTK_TYPE_POINTER);
+	gtk_tree_view_set_model(GTK_TREE_VIEW(tree_eps), GTK_TREE_MODEL(store_eps));
+	g_object_unref(store_eps);
+
+	gtk_container_add (GTK_CONTAINER (scrolledwindow1), tree_eps);
+
+	gtk_tree_selection_set_select_function (gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_eps)), on_eps_select, NULL, NULL);
+
+	vbox2 = gtk_vbox_new (FALSE, 0);
+	gtk_container_add (GTK_CONTAINER (hbox2), vbox2);
+
+	frame1 = gtk_frame_new("Interface");
+	gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+
+	vbox3 = gtk_vbox_new (FALSE, 0);
+	gtk_container_add (GTK_CONTAINER (frame1), vbox3);
+	gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_uuid = gtk_label_new(""));
+	gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_version = gtk_label_new(""));
+	gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_name = gtk_label_new(""));
+
+	frame1 = gtk_frame_new("Statistics");
+	gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+
+	table_statistics = gtk_table_new(4, 2, TRUE);
+	gtk_container_add (GTK_CONTAINER(frame1), table_statistics);
+
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Calls In: "), 0, 1, 0, 1);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_calls_in = gtk_label_new(""), 1, 2, 0, 1);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Calls Out: "), 0, 1, 1, 2);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_calls_out = gtk_label_new(""), 1, 2, 1, 2);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Packets In: "), 0, 1, 2, 3);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_pkts_in = gtk_label_new(""), 1, 2, 2, 3);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Packets Out: "), 0, 1, 3, 4);
+	gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_pkts_out = gtk_label_new(""), 1, 2, 3, 4);
+	
+	frame1 = gtk_frame_new("Authentication");
+	gtk_container_add (GTK_CONTAINER(vbox2), frame1);
+
+	statusbar = gtk_statusbar_new ();
+	gtk_box_pack_start (GTK_BOX (vbox1), statusbar, FALSE, FALSE, 0);
+
+	g_signal_connect ((gpointer) quit1, "activate",
+					  G_CALLBACK (on_quit1_activate),
+					  NULL);
+	g_signal_connect ((gpointer) about1, "activate",
+					  G_CALLBACK (on_about1_activate),
+					  NULL);
+	g_signal_connect ((gpointer) btn_select_target, "clicked",
+					  G_CALLBACK (on_select_target_clicked),
+					  NULL);
+	g_signal_connect ((gpointer) btn_dump, "clicked",
+					  G_CALLBACK (on_dump_clicked),
+					  NULL);
+
+	gtk_window_add_accel_group (GTK_WINDOW (mainwindow), accel_group);
+
+	return mainwindow;
 }
 
 
- int main(int argc, char **argv)
+int main(int argc, char **argv)
 {
 	gtk_init(&argc, &argv);
 	lp_load(dyn_CONFIGFILE,True,False,False);