r3211: Add a graphical endpoint mapper view tool

(This used to be commit 71dfc27b8efb3057c4b7383feca0ba35ab5768a4)

5 files changed, 334 insertions, 8 deletions

diff --git a/source4/gtk/common/gtk-smb.c b/source4/gtk/common/gtk-smb.c
index 2bc14d93bf..6a581beec7 100644
--- a/source4/gtk/common/gtk-smb.c
+++ b/source4/gtk/common/gtk-smb.c
@@ -342,6 +342,7 @@ struct dcerpc_binding *gtk_rpc_binding_dialog_get_binding(GtkRpcBindingDialog *d
 
 	binding->options = NULL;
 	binding->flags = 0;
+	binding->endpoint = NULL;
 
 	if(gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(d->chk_seal))) {
 		binding->flags |= DCERPC_SEAL;
diff --git a/source4/gtk/config.m4 b/source4/gtk/config.m4
index 0518f3af55..6be4bf11bc 100644
--- a/source4/gtk/config.m4
+++ b/source4/gtk/config.m4
@@ -5,12 +5,14 @@ SMB_SUBSYSTEM_ENABLE(GTKSMB, NO)
 SMB_BINARY_ENABLE(gregedit, NO)
 SMB_BINARY_ENABLE(gwcrontab, NO)
 SMB_BINARY_ENABLE(gwsam, NO)
+SMB_BINARY_ENABLE(gepdump, NO)
 
 if test t$SMB_EXT_LIB_ENABLE_gtk = tYES; then
 	SMB_SUBSYSTEM_ENABLE(GTKSMB, YES)
 	SMB_BINARY_ENABLE(gregedit, YES)
 	SMB_BINARY_ENABLE(gwcrontab, YES)
 	SMB_BINARY_ENABLE(gwsam, YES)
+	SMB_BINARY_ENABLE(gepdump, YES)
 	AC_DEFINE(HAVE_GTK, 1, [Whether GTK+ is available])
 fi
 
@@ -18,3 +20,4 @@ SMB_SUBSYSTEM_MK(GTKSMB,gtk/config.mk)
 SMB_BINARY_MK(gregedit,gtk/config.mk)
 SMB_BINARY_MK(gwcrontab,gtk/config.mk)
 SMB_BINARY_MK(gwsam,gtk/config.mk)
+SMB_BINARY_MK(gepdump,gtk/config.mk)
diff --git a/source4/gtk/config.mk b/source4/gtk/config.mk
index 5ff34e7d09..a8fe27e1a5 100644
--- a/source4/gtk/config.mk
+++ b/source4/gtk/config.mk
@@ -19,6 +19,14 @@ REQUIRED_SUBSYSTEMS = CONFIG LIBCMDLINE REGISTRY GTKSMB
 ################################################
 
 ################################################
+# Start BINARY gepdump 
+[BINARY::gepdump]
+OBJ_FILES = gtk/tools/gepdump.o
+REQUIRED_SUBSYSTEMS = CONFIG LIBCMDLINE GTKSMB LIBRPC LIBSMB
+# End BINARY gepdump 
+################################################
+
+################################################
 # Start BINARY gwcrontab
 [BINARY::gwcrontab]
 OBJ_FILES = gtk/tools/gwcrontab.o
diff --git a/source4/gtk/tools/gepdump.c b/source4/gtk/tools/gepdump.c
new file mode 100644
index 0000000000..05d3ed2425
--- /dev/null
+++ b/source4/gtk/tools/gepdump.c
@@ -0,0 +1,314 @@
+/* 
+   Unix SMB/CIFS implementation.
+   GTK+ Endpoint Mapper frontend
+   
+   Copyright (C) Jelmer Vernooij 2004
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+#include "gtk/common/gtk-smb.h"
+
+GtkWidget *mainwin;
+GtkWidget *entry_binding;
+GtkTreeStore *store_eps;
+
+static void on_quit1_activate                      (GtkMenuItem     *menuitem,
+                                        gpointer         user_data)
+{
+	gtk_main_quit();
+}
+
+
+static void on_about1_activate                     (GtkMenuItem     *menuitem,
+                                        gpointer         user_data)
+{
+	GtkDialog *aboutwin = GTK_DIALOG(create_gtk_samba_about_dialog("gsmbtorture"));
+	gtk_dialog_run(aboutwin);
+	gtk_widget_destroy(GTK_WIDGET(aboutwin));
+}
+
+static const char *get_protocol_name(enum epm_protocols protocol)
+{
+	switch (protocol) {
+	case EPM_PROTOCOL_UUID: return "UUID";
+	case EPM_PROTOCOL_NCACN: return "NCACN";
+	case EPM_PROTOCOL_NCALRPC: return "NCALRPC";
+	case EPM_PROTOCOL_NCADG: return "NCADG";
+	case EPM_PROTOCOL_IP: return "IP";
+	case EPM_PROTOCOL_TCP: return "TCP";
+	case EPM_PROTOCOL_NETBIOS: return "NetBIOS";
+	case EPM_PROTOCOL_SMB: return "SMB";
+	case EPM_PROTOCOL_PIPE: return "PIPE";
+	default: return "Unknown";
+	}
+}
+
+static void add_epm_entry(TALLOC_CTX *mem_ctx, const char *annotation, struct epm_tower *t)
+{
+	struct dcerpc_binding bd;
+	int i;
+	NTSTATUS status;
+	GtkTreeIter toweriter;
+
+	status = dcerpc_binding_from_tower(mem_ctx, t, &bd);
+	if (!NT_STATUS_IS_OK(status)) {
+		gtk_show_ntstatus(mainwin, status);
+		return;
+	}
+	
+	/* Don't show UUID's */
+	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);
+
+	for (i = 0; i < t->num_floors; i++) {
+		const char *data;
+		GtkTreeIter iter;
+		gtk_tree_store_append(store_eps, &iter, &toweriter);
+
+		if (t->floors[i].lhs.protocol == EPM_PROTOCOL_UUID) {
+			data = GUID_string(mem_ctx, &t->floors[i].lhs.info.uuid.uuid);
+		} else {
+			data = dcerpc_floor_get_rhs_data(mem_ctx, &t->floors[i]);
+		}
+		
+		gtk_tree_store_set(store_eps, &iter, 0, get_protocol_name(t->floors[i].lhs.protocol), 1, data, -1);
+	}
+}
+
+static void on_dump_clicked                     (GtkButton *btn, gpointer         user_data)
+{
+	NTSTATUS status;
+	struct epm_Lookup r;
+	struct GUID uuid;
+	struct rpc_if_id_t iface;
+	struct policy_handle handle;
+	struct dcerpc_pipe *p;
+	TALLOC_CTX *mem_ctx = talloc_init("dump");
+
+	status = dcerpc_pipe_connect(&p, gtk_entry_get_text(GTK_ENTRY(entry_binding)), DCERPC_EPMAPPER_UUID, DCERPC_EPMAPPER_VERSION, lp_workgroup(), NULL, NULL);
+
+	if (NT_STATUS_IS_ERR(status)) {
+		gtk_show_ntstatus(mainwin, status);
+		talloc_destroy(mem_ctx);
+		return;
+	}
+
+	ZERO_STRUCT(uuid);
+	ZERO_STRUCT(iface);
+	ZERO_STRUCT(handle);
+
+	r.in.inquiry_type = 0;
+	r.in.object = &uuid;
+	r.in.interface_id = &iface;
+	r.in.vers_option = 0;
+	r.in.entry_handle = &handle;
+	r.out.entry_handle = &handle;
+	r.in.max_ents = 10;
+
+	gtk_tree_store_clear(store_eps);
+
+	do {
+		int i;
+		status = dcerpc_epm_Lookup(p, mem_ctx, &r);
+		if (!NT_STATUS_IS_OK(status) || r.out.result != 0) {
+			break;
+		}
+		for (i=0;i<r.out.num_ents;i++) {
+			add_epm_entry(mem_ctx, r.out.entries[i].annotation, &r.out.entries[i].tower->tower);
+		}
+	} while (NT_STATUS_IS_OK(status) && 
+		 r.out.result == 0 && 
+		 r.out.num_ents == r.in.max_ents);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		gtk_show_ntstatus(mainwin, status);
+		talloc_destroy(mem_ctx);
+		return;
+	}
+	talloc_destroy(mem_ctx);
+}
+
+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(FALSE, NULL));
+	result = gtk_dialog_run(GTK_DIALOG(d));
+	switch(result) {
+	case GTK_RESPONSE_ACCEPT:
+		break;
+	default:
+		gtk_widget_destroy(GTK_WIDGET(d));
+		return;
+	}
+
+	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));
+	talloc_destroy(mem_ctx);
+	gtk_widget_destroy(GTK_WIDGET(d));
+}
+
+static GtkWidget* create_mainwindow (void)
+{
+  GtkWidget *mainwindow;
+  GtkWidget *vbox1;
+  GtkWidget *menubar1;
+  GtkWidget *menuitem1;
+  GtkWidget *menuitem1_menu;
+  GtkWidget *quit1;
+  GtkWidget *menuitem4;
+  GtkWidget *menuitem4_menu;
+  GtkWidget *about1;
+  GtkWidget *handlebox1;
+  GtkWidget *hbox1;
+  GtkWidget *label1;
+  GtkWidget *btn_select_target;
+  GtkWidget *btn_dump;
+  GtkWidget *scrolledwindow1;
+  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_widget_show (quit1);
+  gtk_container_add (GTK_CONTAINER (menuitem1_menu), quit1);
+
+  menuitem4 = gtk_menu_item_new_with_mnemonic ("_Help");
+  gtk_widget_show (menuitem4);
+  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_widget_show (about1);
+  gtk_container_add (GTK_CONTAINER (menuitem4_menu), about1);
+
+  handlebox1 = gtk_handle_box_new ();
+  gtk_widget_show (handlebox1);
+  gtk_box_pack_start (GTK_BOX (vbox1), handlebox1, FALSE, TRUE, 0);
+
+  hbox1 = gtk_hbox_new (FALSE, 0);
+  gtk_widget_show (hbox1);
+  gtk_container_add (GTK_CONTAINER (handlebox1), hbox1);
+
+  label1 = gtk_label_new ("Location:");
+  gtk_widget_show (label1);
+  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_widget_show (entry_binding);
+  gtk_box_pack_start (GTK_BOX (hbox1), entry_binding, FALSE, FALSE, 0);
+
+  btn_select_target = gtk_button_new_with_mnemonic ("_Select Target");
+  gtk_widget_show (btn_select_target);
+  gtk_box_pack_start (GTK_BOX (hbox1), btn_select_target, FALSE, FALSE, 0);
+
+  btn_dump = gtk_button_new_with_mnemonic ("_Dump");
+  gtk_widget_show (btn_dump);
+  gtk_box_pack_start (GTK_BOX (hbox1), btn_dump, FALSE, FALSE, 0);
+
+  scrolledwindow1 = gtk_scrolled_window_new (NULL, NULL);
+  gtk_widget_show (scrolledwindow1);
+  gtk_box_pack_start (GTK_BOX (vbox1), 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, GTK_TYPE_STRING, GTK_TYPE_STRING);
+  gtk_tree_view_set_model(GTK_TREE_VIEW(tree_eps), GTK_TREE_MODEL(store_eps));
+  g_object_unref(store_eps);
+  
+  gtk_widget_show (tree_eps);
+  gtk_container_add (GTK_CONTAINER (scrolledwindow1), tree_eps);
+
+  statusbar = gtk_statusbar_new ();
+  gtk_widget_show (statusbar);
+  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)
+{
+	gtk_init(&argc, &argv);
+	mainwin = create_mainwindow();
+	gtk_widget_show(mainwin);
+	gtk_main();
+	return 0;
+}
diff --git a/source4/librpc/rpc/dcerpc_util.c b/source4/librpc/rpc/dcerpc_util.c
index d0ecf3a7a8..89a4366b96 100644
--- a/source4/librpc/rpc/dcerpc_util.c
+++ b/source4/librpc/rpc/dcerpc_util.c
@@ -357,7 +357,7 @@ NTSTATUS dcerpc_parse_binding(TALLOC_CTX *mem_ctx, const char *s, struct dcerpc_
 	return NT_STATUS_OK;
 }
 
-static const char *floor_get_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *floor)
+const char *dcerpc_floor_get_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *floor)
 {
 	switch (floor->lhs.protocol) {
 	case EPM_PROTOCOL_TCP:
@@ -424,7 +424,7 @@ static const char *floor_get_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *flo
 	return NULL;
 }
 
-static NTSTATUS floor_set_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *floor,  const char *data)
+static NTSTATUS dcerpc_floor_set_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *floor,  const char *data)
 {
 	switch (floor->lhs.protocol) {
 	case EPM_PROTOCOL_TCP:
@@ -576,14 +576,14 @@ NTSTATUS dcerpc_binding_from_tower(TALLOC_CTX *mem_ctx, struct epm_tower *tower,
 
 	/* Set endpoint */
 	if (tower->num_floors >= 4) {
-		binding->endpoint = floor_get_rhs_data(mem_ctx, &tower->floors[3]);
+		binding->endpoint = dcerpc_floor_get_rhs_data(mem_ctx, &tower->floors[3]);
 	} else {
 		binding->endpoint = NULL;
 	}
 
 	/* Set network address */
 	if (tower->num_floors >= 5) {
-		binding->host = floor_get_rhs_data(mem_ctx, &tower->floors[4]);
+		binding->host = dcerpc_floor_get_rhs_data(mem_ctx, &tower->floors[4]);
 	}
 	return NT_STATUS_OK;
 }
@@ -631,12 +631,12 @@ NTSTATUS dcerpc_binding_build_tower(TALLOC_CTX *mem_ctx, struct dcerpc_binding *
 		tower->floors[2 + i].lhs.protocol = protseq[i];
 		tower->floors[2 + i].lhs.info.lhs_data = data_blob_talloc(mem_ctx, NULL, 0);
 		ZERO_STRUCT(tower->floors[2 + i].rhs);
-		floor_set_rhs_data(mem_ctx, &tower->floors[2 + i], "");
+		dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[2 + i], "");
 	}
 
 	/* The 4th floor contains the endpoint */
 	if (num_protocols >= 2 && binding->endpoint) {
-		status = floor_set_rhs_data(mem_ctx, &tower->floors[3], binding->endpoint);
+		status = dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[3], binding->endpoint);
 		if (NT_STATUS_IS_ERR(status)) {
 			return status;
 		}
@@ -644,7 +644,7 @@ NTSTATUS dcerpc_binding_build_tower(TALLOC_CTX *mem_ctx, struct dcerpc_binding *
 	
 	/* The 5th contains the network address */
 	if (num_protocols >= 3 && binding->host) {
-		status = floor_set_rhs_data(mem_ctx, &tower->floors[4], binding->host);
+		status = dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[4], binding->host);
 		if (NT_STATUS_IS_ERR(status)) {
 			return status;
 		}
@@ -735,7 +735,7 @@ NTSTATUS dcerpc_epm_map_binding(TALLOC_CTX *mem_ctx, struct dcerpc_binding *bind
 		return NT_STATUS_PORT_UNREACHABLE;
 	}
 
-	binding->endpoint = floor_get_rhs_data(mem_ctx, &twr_r->tower.floors[3]);
+	binding->endpoint = dcerpc_floor_get_rhs_data(mem_ctx, &twr_r->tower.floors[3]);
 
 	dcerpc_pipe_close(p);