s3-rpcclient: add object_uuid argument to cmd_epmapper_map()
[samba.git] / source3 / rpcclient / cmd_epmapper.c
index cb334165f26d3fe492ca9a51a9271222cafc1da7..289b086718fa6366b75cfd4a87aa5a46be164dc5 100644 (file)
 
 #include "includes.h"
 #include "rpcclient.h"
+#include "../librpc/gen_ndr/ndr_epmapper_c.h"
+#include "librpc/ndr/ndr_table.h"
 
 static NTSTATUS cmd_epmapper_map(struct rpc_pipe_client *p,
                                 TALLOC_CTX *mem_ctx,
                                 int argc, const char **argv)
 {
-       struct dcerpc_binding map_binding;
+       struct dcerpc_binding_handle *b = p->binding_handle;
+       struct dcerpc_binding *map_binding;
        struct epm_twr_t map_tower;
-       struct epm_twr_t res_tower;
-       struct epm_twr_p_t towers;
+       struct epm_twr_p_t towers[500];
        struct policy_handle entry_handle;
        struct ndr_syntax_id abstract_syntax;
        uint32_t num_towers;
        TALLOC_CTX *tmp_ctx = talloc_stackframe();
        NTSTATUS status;
+       uint32_t result;
+       uint32_t i;
+       const struct ndr_interface_list *l;
+       const char *interface_name = "lsarpc";
+       enum dcerpc_transport_t transport = NCACN_NP;
+       bool ok = false;
+       struct GUID object_uuid = GUID_zero();
+
+       if (argc > 4) {
+               d_fprintf(stderr,
+                         "Usage: %s [interface_name] [transport] "
+                         "[object_uuid]\n",
+                         argv[0]);
+               return NT_STATUS_OK;
+       }
+
+       if (argc >= 2) {
+               interface_name = argv[1];
+       }
+
+       for (l = ndr_table_list(); l != NULL; l = l->next) {
+
+               ok = strequal(interface_name, l->table->name);
+               if (ok) {
+                       abstract_syntax = l->table->syntax_id;
+                       break;
+               }
+       }
+
+       if (!ok) {
+               d_fprintf(stderr, "unknown interface: %s\n",
+                       interface_name);
+               status = NT_STATUS_UNSUCCESSFUL;
+               goto done;
+       }
+
+       if (argc >= 3) {
+               transport = dcerpc_transport_by_name(argv[2]);
+               if (transport == NCA_UNKNOWN) {
+                       d_fprintf(stderr, "unknown transport: %s\n",
+                               argv[2]);
+                       status = NT_STATUS_UNSUCCESSFUL;
+                       goto done;
+               }
+       }
 
-       abstract_syntax = ndr_table_lsarpc.syntax_id;
+       if (argc >= 4) {
+               status = GUID_from_string(argv[3], &object_uuid);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
+               }
+       }
+
+       /* 127.0.0.1[0] => correct? needed? */
+       status = dcerpc_parse_binding(tmp_ctx, "ncacn_np:127.0.0.1[0]",
+                                     &map_binding);
+       if (!NT_STATUS_IS_OK(status)) {
+               d_fprintf(stderr, "dcerpc_parse_binding returned %s\n",
+                         nt_errstr(status));
+               goto done;
+       }
 
-       map_binding.transport = NCACN_NP;
-        map_binding.object = abstract_syntax;
-        map_binding.host = "127.0.0.1"; /* needed? */
-        map_binding.endpoint = "0"; /* correct? needed? */
+       status = dcerpc_binding_set_transport(map_binding, transport);
+       if (!NT_STATUS_IS_OK(status)) {
+               d_fprintf(stderr, "dcerpc_binding_set_transport returned %s\n",
+                         nt_errstr(status));
+               goto done;
+       }
+
+       status = dcerpc_binding_set_abstract_syntax(map_binding,
+                                                   &abstract_syntax);
+       if (!NT_STATUS_IS_OK(status)) {
+               d_fprintf(stderr, "dcerpc_binding_set_abstract_syntax returned %s\n",
+                         nt_errstr(status));
+               goto done;
+       }
 
-       status = dcerpc_binding_build_tower(tmp_ctx, &map_binding,
+       status = dcerpc_binding_build_tower(tmp_ctx, map_binding,
                                            &map_tower.tower);
        if (!NT_STATUS_IS_OK(status)) {
                d_fprintf(stderr, "dcerpc_binding_build_tower returned %s\n",
                          nt_errstr(status));
-               return status;
+               goto done;
        }
 
-       towers.twr = &res_tower;
-
+       ZERO_STRUCT(towers);
        ZERO_STRUCT(entry_handle);
-       status = rpccli_epm_Map(
-               p, tmp_ctx, &abstract_syntax.uuid,
-               &map_tower, &entry_handle, 1,
-               &num_towers, &towers);
 
+       status = dcerpc_epm_Map(
+               b, tmp_ctx, &object_uuid,
+               &map_tower, &entry_handle, ARRAY_SIZE(towers),
+               &num_towers, towers, &result);
+       if (!NT_STATUS_IS_OK(status)) {
+               d_fprintf(stderr, "dcerpc_epm_Map returned %s\n",
+                         nt_errstr(status));
+               goto done;
+       }
+
+       if (result != EPMAPPER_STATUS_OK) {
+               d_fprintf(stderr, "epm_Map returned %u (0x%08X)\n",
+                         result, result);
+               status = NT_STATUS_UNSUCCESSFUL;
+               goto done;
+       }
+
+       d_printf("num_tower[%u]\n", num_towers);
+
+       for (i=0; i < num_towers; i++) {
+               struct dcerpc_binding *binding;
+
+               if (towers[i].twr == NULL) {
+                       d_fprintf(stderr, "tower[%u] NULL\n", i);
+                       break;
+               }
+
+               status = dcerpc_binding_from_tower(tmp_ctx, &towers[i].twr->tower,
+                                                  &binding);
+               if (!NT_STATUS_IS_OK(status)) {
+                       break;
+               }
+
+               d_printf("tower[%u] %s\n", i, dcerpc_binding_string(tmp_ctx, binding));
+       }
+done:
+       TALLOC_FREE(tmp_ctx);
        return status;
 }
 
@@ -65,6 +168,7 @@ static NTSTATUS cmd_epmapper_lookup(struct rpc_pipe_client *p,
                                    TALLOC_CTX *mem_ctx,
                                    int argc, const char **argv)
 {
+       struct dcerpc_binding_handle *b = p->binding_handle;
        struct policy_handle entry_handle;
 
        ZERO_STRUCT(entry_handle);
@@ -76,23 +180,36 @@ static NTSTATUS cmd_epmapper_lookup(struct rpc_pipe_client *p,
                NTSTATUS status;
                char *guid_string;
                struct dcerpc_binding *binding;
+               uint32_t result;
 
-               status = rpccli_epm_Lookup(p, tmp_ctx,
+               status = dcerpc_epm_Lookup(b, tmp_ctx,
                                   0, /* rpc_c_ep_all */
                                   NULL,
                                   NULL,
                                   0, /* rpc_c_vers_all */
                                   &entry_handle,
                                   1, /* max_ents */
-                                  &num_entries, &entry);
+                                  &num_entries, &entry,
+                                  &result);
                if (!NT_STATUS_IS_OK(status)) {
-                       d_fprintf(stderr, "rpccli_epm_Lookup returned %s\n",
+                       d_fprintf(stderr, "dcerpc_epm_Lookup returned %s\n",
                                  nt_errstr(status));
                        break;
                }
 
+               if (result == EPMAPPER_STATUS_NO_MORE_ENTRIES) {
+                       d_fprintf(stderr, "epm_Lookup no more entries\n");
+                       break;
+               }
+
+               if (result != EPMAPPER_STATUS_OK) {
+                       d_fprintf(stderr, "epm_Lookup returned %u (0x%08X)\n",
+                                 result, result);
+                       break;
+               }
+
                if (num_entries != 1) {
-                       d_fprintf(stderr, "rpccli_epm_Lookup returned %d "
+                       d_fprintf(stderr, "epm_Lookup returned %d "
                                  "entries, expected one\n", (int)num_entries);
                        break;
                }
@@ -126,8 +243,8 @@ struct cmd_set epmapper_commands[] = {
        { "EPMAPPER" },
 
        { "epmmap", RPC_RTYPE_NTSTATUS, cmd_epmapper_map,     NULL,
-         &ndr_table_epmapper.syntax_id, NULL, "Map a binding", "" },
+         &ndr_table_epmapper, NULL, "Map a binding", "" },
        { "epmlookup", RPC_RTYPE_NTSTATUS, cmd_epmapper_lookup,     NULL,
-         &ndr_table_epmapper.syntax_id, NULL, "Lookup bindings", "" },
+         &ndr_table_epmapper, NULL, "Lookup bindings", "" },
        { NULL }
 };