From 883fdf1a17f6aac44fb201c52e239810144b22e2 Mon Sep 17 00:00:00 2001 From: Jelmer Vernooij Date: Mon, 25 Oct 2004 11:28:09 +0000 Subject: r3211: Add a graphical endpoint mapper view tool (This used to be commit 71dfc27b8efb3057c4b7383feca0ba35ab5768a4) --- source4/gtk/common/gtk-smb.c | 1 + source4/gtk/config.m4 | 3 + source4/gtk/config.mk | 8 + source4/gtk/tools/gepdump.c | 314 +++++++++++++++++++++++++++++++++++++++ source4/librpc/rpc/dcerpc_util.c | 16 +- 5 files changed, 334 insertions(+), 8 deletions(-) create mode 100644 source4/gtk/tools/gepdump.c 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 @@ -18,6 +18,14 @@ REQUIRED_SUBSYSTEMS = CONFIG LIBCMDLINE REGISTRY GTKSMB # End BINARY gregedit ################################################ +################################################ +# 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] 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;itower); + } + } 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); -- cgit