ce606a6a6e64276fb2bca9eb4208a09d80b7f665
[samba.git] / source4 / gtk / tools / gepdump.c
1 /* 
2    Unix SMB/CIFS implementation.
3    GTK+ Endpoint Mapper frontend
4    
5    Copyright (C) Jelmer Vernooij 2004
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 "dynconfig.h"
24 #include "librpc/gen_ndr/ndr_epmapper.h"
25 #include "librpc/gen_ndr/ndr_mgmt.h"
26 #include "gtk/common/select.h"
27 #include "gtk/common/gtk-smb.h"
28
29 /* 
30  * Show: 
31  *  - RPC statistics
32  *  - Available interfaces
33  *   - Per interface: available endpoints
34  *   - Per interface auth details
35  */
36
37 static GtkWidget *mainwin;
38 static GtkWidget *entry_binding;
39 static GtkTreeStore *store_eps;
40 static GtkWidget *table_statistics;
41 static GtkWidget *lbl_calls_in, *lbl_calls_out, *lbl_pkts_in, *lbl_pkts_out;
42 static GtkWidget *lbl_iface_version, *lbl_iface_uuid, *lbl_iface_name;
43 TALLOC_CTX *eps_ctx = NULL;
44
45 static void on_quit1_activate (GtkMenuItem *menuitem, gpointer user_data)
46 {
47         gtk_main_quit();
48 }
49
50
51 static void on_about1_activate (GtkMenuItem *menuitem, gpointer user_data)
52 {
53         GtkDialog *aboutwin = GTK_DIALOG(create_gtk_samba_about_dialog("gepdump"));
54         gtk_dialog_run(aboutwin);
55         gtk_widget_destroy(GTK_WIDGET(aboutwin));
56 }
57
58 static const char *get_protocol_name(enum epm_protocols protocol)
59 {
60         switch (protocol) {
61         case EPM_PROTOCOL_UUID: return "UUID";
62         case EPM_PROTOCOL_NCACN: return "NCACN";
63         case EPM_PROTOCOL_NCALRPC: return "NCALRPC";
64         case EPM_PROTOCOL_NCADG: return "NCADG";
65         case EPM_PROTOCOL_IP: return "IP";
66         case EPM_PROTOCOL_TCP: return "TCP";
67         case EPM_PROTOCOL_NETBIOS: return "NetBIOS";
68         case EPM_PROTOCOL_SMB: return "SMB";
69         case EPM_PROTOCOL_PIPE: return "PIPE";
70         case EPM_PROTOCOL_UNIX_DS: return "Unix";
71         default: return "Unknown";
72         }
73 }
74
75 static void add_epm_entry(TALLOC_CTX *mem_ctx, const char *annotation, struct epm_tower *t)
76 {
77         struct dcerpc_binding bd;
78         int i;
79         NTSTATUS status;
80         GtkTreeIter toweriter;
81
82         status = dcerpc_binding_from_tower(mem_ctx, t, &bd);
83         if (!NT_STATUS_IS_OK(status)) {
84                 gtk_show_ntstatus(mainwin, status);
85                 return;
86         }
87         
88         /* Don't show UUID's */
89         ZERO_STRUCT(bd.object);
90
91         gtk_tree_store_append(store_eps, &toweriter, NULL);
92         gtk_tree_store_set(store_eps, &toweriter, 0, strdup(annotation), 1, strdup(dcerpc_binding_string(mem_ctx, &bd)), 2, t, -1);
93
94         for (i = 0; i < t->num_floors; i++) {
95                 const char *data;
96                 GtkTreeIter iter;
97                 gtk_tree_store_append(store_eps, &iter, &toweriter);
98
99                 if (t->floors[i].lhs.protocol == EPM_PROTOCOL_UUID) {
100                         data = GUID_string(mem_ctx, &t->floors[i].lhs.info.uuid.uuid);
101                 } else {
102                         data = dcerpc_floor_get_rhs_data(mem_ctx, &t->floors[i]);
103                 }
104                 
105                 gtk_tree_store_set(store_eps, &iter, 0, get_protocol_name(t->floors[i].lhs.protocol), 1, data, -1);
106         }
107 }
108
109 static void on_dump_clicked (GtkButton *btn, gpointer user_data)
110 {
111         NTSTATUS status;
112         struct epm_Lookup r;
113         struct GUID uuid;
114         struct rpc_if_id_t iface;
115         struct policy_handle handle;
116         struct dcerpc_pipe *p;
117         TALLOC_CTX *mem_ctx = talloc_init("dump");
118
119         talloc_destroy(eps_ctx);
120
121         status = dcerpc_pipe_connect(&p, gtk_entry_get_text(GTK_ENTRY(entry_binding)), DCERPC_EPMAPPER_UUID, DCERPC_EPMAPPER_VERSION, lp_workgroup(), "", "");
122
123         if (NT_STATUS_IS_ERR(status)) {
124                 gtk_show_ntstatus(mainwin, status);
125                 talloc_destroy(mem_ctx);
126                 return;
127         }
128
129         ZERO_STRUCT(uuid);
130         ZERO_STRUCT(iface);
131         ZERO_STRUCT(handle);
132
133         r.in.inquiry_type = 0;
134         r.in.object = &uuid;
135         r.in.interface_id = &iface;
136         r.in.vers_option = 0;
137         r.in.entry_handle = &handle;
138         r.out.entry_handle = &handle;
139         r.in.max_ents = 10;
140
141         gtk_tree_store_clear(store_eps);
142
143         eps_ctx = talloc_init("current endpoint list data");
144
145         do {
146                 int i;
147                 status = dcerpc_epm_Lookup(p, eps_ctx, &r);
148                 if (!NT_STATUS_IS_OK(status) || r.out.result != 0) {
149                         break;
150                 }
151                 for (i=0;i<r.out.num_ents;i++) {
152                         add_epm_entry(mem_ctx, r.out.entries[i].annotation, &r.out.entries[i].tower->tower);
153                 }
154         } while (NT_STATUS_IS_OK(status) && 
155                  r.out.result == 0 && 
156                  r.out.num_ents == r.in.max_ents);
157
158         if (!NT_STATUS_IS_OK(status)) {
159                 gtk_show_ntstatus(mainwin, status);
160                 talloc_destroy(mem_ctx);
161                 return;
162         }
163         talloc_destroy(mem_ctx);
164 }
165
166 static void on_select_target_clicked(GtkButton *btn, gpointer         user_data)
167 {
168         GtkRpcBindingDialog *d;
169         TALLOC_CTX *mem_ctx;
170         gint result;
171
172         d = GTK_RPC_BINDING_DIALOG(gtk_rpc_binding_dialog_new(TRUE, NULL));
173         result = gtk_dialog_run(GTK_DIALOG(d));
174         switch(result) {
175         case GTK_RESPONSE_ACCEPT:
176                 break;
177         default:
178                 gtk_widget_destroy(GTK_WIDGET(d));
179                 return;
180         }
181
182         mem_ctx = talloc_init("select_target");
183         gtk_entry_set_text(GTK_ENTRY(entry_binding), gtk_rpc_binding_dialog_get_binding_string (d, mem_ctx));
184         talloc_destroy(mem_ctx);
185         gtk_widget_destroy(GTK_WIDGET(d));
186 }
187
188 static gboolean on_eps_select(GtkTreeSelection *selection,
189     GtkTreeModel *model, GtkTreePath *path, gboolean path_currently_selected, gpointer data)
190 {
191         struct dcerpc_pipe *p;
192         NTSTATUS status;
193         struct dcerpc_binding bd;
194         TALLOC_CTX *mem_ctx = talloc_init("eps");
195
196         ZERO_STRUCT(bd);
197         bd.host = "192.168.4.26";
198         bd.endpoint = "samr";
199         bd.transport = NCACN_NP;
200
201         status = dcerpc_pipe_connect_b(&p, &bd, DCERPC_MGMT_UUID, DCERPC_MGMT_VERSION, "", "administrator", "penguin");
202
203         if (NT_STATUS_IS_ERR(status)) {
204                 gtk_show_ntstatus(NULL, status);
205                 return FALSE;
206         }
207         
208         {
209                 /* Do an InqStats call */
210                 struct mgmt_inq_stats r;
211
212                 r.in.max_count = MGMT_STATS_ARRAY_MAX_SIZE;
213                 r.in.unknown = 0;
214
215                 status = dcerpc_mgmt_inq_stats(p, mem_ctx, &r);
216                 if (NT_STATUS_IS_ERR(status)) {
217                         gtk_show_ntstatus(NULL, status);
218                         return TRUE;
219                 }
220
221                 if (r.out.statistics.count != MGMT_STATS_ARRAY_MAX_SIZE) {
222                         printf("Unexpected array size %d\n", r.out.statistics.count);
223                         return False;
224                 }
225
226                 gtk_label_set_text(GTK_LABEL(lbl_calls_in), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_CALLS_IN]));
227                 gtk_label_set_text(GTK_LABEL(lbl_calls_out), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_CALLS_OUT]));
228                 gtk_label_set_text(GTK_LABEL(lbl_pkts_in), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_PKTS_IN]));
229                 gtk_label_set_text(GTK_LABEL(lbl_pkts_out), talloc_asprintf(mem_ctx, "%6d", r.out.statistics.statistics[MGMT_STATS_PKTS_OUT]));
230         }
231
232         {
233                 struct mgmt_inq_princ_name r;
234                 int i;
235
236                 for (i=0;i<100;i++) {
237                         r.in.authn_proto = i;  /* DCERPC_AUTH_TYPE_* */
238                         r.in.princ_name_size = 100;
239
240                         status = dcerpc_mgmt_inq_princ_name(p, mem_ctx, &r);
241                         if (!NT_STATUS_IS_OK(status)) {
242                                 continue;
243                         }
244                         if (W_ERROR_IS_OK(r.out.result)) {
245                                 const char *name = gensec_get_name_by_authtype(i);
246                                 if (name) {
247                                         printf("\tprinciple name for proto %u (%s) is '%s'\n",
248                                                    i, name, r.out.princ_name);
249                                 } else {
250                                         printf("\tprinciple name for proto %u is '%s'\n",
251                                                    i, r.out.princ_name);
252                                 }
253                         }
254                 }
255         }
256
257         return TRUE;
258 }
259
260
261 static GtkWidget* create_mainwindow (void)
262 {
263         GtkWidget *mainwindow;
264         GtkWidget *vbox1, *vbox2, *vbox3;
265         GtkWidget *menubar1;
266         GtkWidget *menuitem1;
267         GtkWidget *menuitem1_menu;
268         GtkWidget *quit1;
269         GtkWidget *menuitem4;
270         GtkWidget *menuitem4_menu;
271         GtkWidget *about1;
272         GtkWidget *handlebox1;
273         GtkWidget *hbox1;
274         GtkWidget *hbox2;
275         GtkWidget *label1;
276         GtkWidget *btn_select_target;
277         GtkWidget *btn_dump;
278         GtkWidget *scrolledwindow1;
279         GtkWidget *frame1;
280         GtkWidget *tree_eps;
281         GtkTreeViewColumn *curcol;
282         GtkCellRenderer *renderer;
283         GtkWidget *statusbar;
284         GtkAccelGroup *accel_group;
285
286         accel_group = gtk_accel_group_new ();
287
288         mainwindow = gtk_window_new (GTK_WINDOW_TOPLEVEL);
289         gtk_window_set_title (GTK_WINDOW (mainwindow), "Gtk+ Endpoint Mapper Viewer");
290
291         vbox1 = gtk_vbox_new (FALSE, 0);
292         gtk_widget_show (vbox1);
293         gtk_container_add (GTK_CONTAINER (mainwindow), vbox1);
294
295         menubar1 = gtk_menu_bar_new ();
296         gtk_widget_show (menubar1);
297         gtk_box_pack_start (GTK_BOX (vbox1), menubar1, FALSE, FALSE, 0);
298
299         menuitem1 = gtk_menu_item_new_with_mnemonic ("_File");
300         gtk_widget_show (menuitem1);
301         gtk_container_add (GTK_CONTAINER (menubar1), menuitem1);
302
303         menuitem1_menu = gtk_menu_new ();
304         gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem1), menuitem1_menu);
305
306         quit1 = gtk_image_menu_item_new_from_stock ("gtk-quit", accel_group);
307         gtk_container_add (GTK_CONTAINER (menuitem1_menu), quit1);
308
309         menuitem4 = gtk_menu_item_new_with_mnemonic ("_Help");
310         gtk_container_add (GTK_CONTAINER (menubar1), menuitem4);
311
312         menuitem4_menu = gtk_menu_new ();
313         gtk_menu_item_set_submenu (GTK_MENU_ITEM (menuitem4), menuitem4_menu);
314
315         about1 = gtk_menu_item_new_with_mnemonic ("_About");
316         gtk_container_add (GTK_CONTAINER (menuitem4_menu), about1);
317
318         handlebox1 = gtk_handle_box_new ();
319         gtk_box_pack_start (GTK_BOX (vbox1), handlebox1, FALSE, TRUE, 0);
320
321         hbox1 = gtk_hbox_new (FALSE, 0);
322         gtk_container_add (GTK_CONTAINER (handlebox1), hbox1);
323
324         label1 = gtk_label_new ("Location:");
325         gtk_box_pack_start (GTK_BOX (hbox1), label1, FALSE, FALSE, 0);
326
327         entry_binding = gtk_entry_new ();
328         gtk_entry_set_text(GTK_ENTRY(entry_binding), "ncalrpc:");
329         gtk_box_pack_start (GTK_BOX (hbox1), entry_binding, FALSE, FALSE, 0);
330
331         btn_select_target = gtk_button_new_with_mnemonic ("_Select Target");
332         gtk_box_pack_start (GTK_BOX (hbox1), btn_select_target, FALSE, FALSE, 0);
333
334         btn_dump = gtk_button_new_with_mnemonic ("_Dump");
335         gtk_box_pack_start (GTK_BOX (hbox1), btn_dump, FALSE, FALSE, 0);
336
337         hbox2 = gtk_hbox_new (FALSE, 0);
338         gtk_container_add (GTK_CONTAINER (vbox1), hbox2);
339
340         scrolledwindow1 = gtk_scrolled_window_new (NULL, NULL);
341         gtk_box_pack_start (GTK_BOX(hbox2), scrolledwindow1, TRUE, TRUE, 0);
342
343         tree_eps = gtk_tree_view_new ();
344
345         curcol = gtk_tree_view_column_new ();
346         gtk_tree_view_column_set_title(curcol, "Name");
347         renderer = gtk_cell_renderer_text_new();
348         gtk_tree_view_column_pack_start(curcol, renderer, True);
349
350         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
351         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 0);
352
353         curcol = gtk_tree_view_column_new ();
354         gtk_tree_view_column_set_title(curcol, "Binding String");
355         renderer = gtk_cell_renderer_text_new();
356         gtk_tree_view_column_pack_start(curcol, renderer, True);
357         gtk_tree_view_column_add_attribute(curcol, renderer, "text", 1);
358
359
360         gtk_tree_view_append_column(GTK_TREE_VIEW(tree_eps), curcol);
361
362         store_eps = gtk_tree_store_new(3, G_TYPE_STRING, G_TYPE_STRING, G_TYPE_POINTER);
363         gtk_tree_view_set_model(GTK_TREE_VIEW(tree_eps), GTK_TREE_MODEL(store_eps));
364         g_object_unref(store_eps);
365
366         gtk_container_add (GTK_CONTAINER (scrolledwindow1), tree_eps);
367
368         gtk_tree_selection_set_select_function (gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_eps)), on_eps_select, NULL, NULL);
369
370         vbox2 = gtk_vbox_new (FALSE, 0);
371         gtk_container_add (GTK_CONTAINER (hbox2), vbox2);
372
373         frame1 = gtk_frame_new("Interface");
374         gtk_container_add (GTK_CONTAINER(vbox2), frame1);
375
376         vbox3 = gtk_vbox_new (FALSE, 0);
377         gtk_container_add (GTK_CONTAINER (frame1), vbox3);
378         gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_uuid = gtk_label_new(""));
379         gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_version = gtk_label_new(""));
380         gtk_container_add (GTK_CONTAINER (vbox3), lbl_iface_name = gtk_label_new(""));
381
382         frame1 = gtk_frame_new("Statistics");
383         gtk_container_add (GTK_CONTAINER(vbox2), frame1);
384
385         table_statistics = gtk_table_new(4, 2, TRUE);
386         gtk_container_add (GTK_CONTAINER(frame1), table_statistics);
387
388         gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Calls In: "), 0, 1, 0, 1);
389         gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_calls_in = gtk_label_new(""), 1, 2, 0, 1);
390         gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Calls Out: "), 0, 1, 1, 2);
391         gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_calls_out = gtk_label_new(""), 1, 2, 1, 2);
392         gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Packets In: "), 0, 1, 2, 3);
393         gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_pkts_in = gtk_label_new(""), 1, 2, 2, 3);
394         gtk_table_attach_defaults (GTK_TABLE(table_statistics), gtk_label_new("Packets Out: "), 0, 1, 3, 4);
395         gtk_table_attach_defaults (GTK_TABLE(table_statistics), lbl_pkts_out = gtk_label_new(""), 1, 2, 3, 4);
396         
397         frame1 = gtk_frame_new("Authentication");
398         gtk_container_add (GTK_CONTAINER(vbox2), frame1);
399
400         statusbar = gtk_statusbar_new ();
401         gtk_box_pack_start (GTK_BOX (vbox1), statusbar, FALSE, FALSE, 0);
402
403         g_signal_connect ((gpointer) quit1, "activate",
404                                           G_CALLBACK (on_quit1_activate),
405                                           NULL);
406         g_signal_connect ((gpointer) about1, "activate",
407                                           G_CALLBACK (on_about1_activate),
408                                           NULL);
409         g_signal_connect ((gpointer) btn_select_target, "clicked",
410                                           G_CALLBACK (on_select_target_clicked),
411                                           NULL);
412         g_signal_connect ((gpointer) btn_dump, "clicked",
413                                           G_CALLBACK (on_dump_clicked),
414                                           NULL);
415
416         gtk_window_add_accel_group (GTK_WINDOW (mainwindow), accel_group);
417
418         return mainwindow;
419 }
420
421
422 int main(int argc, char **argv)
423 {
424         gtk_init(&argc, &argv);
425         gepdump_init_subsystems;
426         lp_load(dyn_CONFIGFILE,True,False,False);
427         load_interfaces();
428         setup_logging("gepdump", True);
429         mainwin = create_mainwindow();
430         gtk_widget_show_all(mainwin);
431         gtk_main();
432         return 0;
433 }