r24557: rename 'dcerpc_table_' -> 'ndr_table_'
[ira/wip.git] / source / torture / rpc / mgmt.c
index 45a96cf9f973809c0f4c45e6209b24977b233f65..475e933bd27da27e39886811630d2d2a60c8cb1e 100644 (file)
@@ -6,7 +6,7 @@
    
    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
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    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.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
-#include "librpc/gen_ndr/ndr_mgmt.h"
-#include "librpc/gen_ndr/tables.h"
+#include "torture/torture.h"
+#include "librpc/gen_ndr/ndr_mgmt_c.h"
+#include "auth/gensec/gensec.h"
+#include "librpc/rpc/dcerpc_table.h"
+#include "torture/rpc/rpc.h"
 
 
 /*
   ask the server what interface IDs are available on this endpoint
 */
-static BOOL test_inq_if_ids(struct dcerpc_pipe *p, 
-                           TALLOC_CTX *mem_ctx)
+BOOL test_inq_if_ids(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+                    BOOL (*per_id_test)(const struct ndr_interface_table *iface,
+                                        TALLOC_CTX *mem_ctx,
+                                        struct ndr_syntax_id *id),
+                    const void *priv)
 {
        NTSTATUS status;
        struct mgmt_inq_if_ids r;
+       struct rpc_if_id_vector_t *vector;
        int i;
+
+       vector = talloc(mem_ctx, struct rpc_if_id_vector_t);
+       r.out.if_id_vector = &vector;
        
        status = dcerpc_mgmt_inq_if_ids(p, mem_ctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
@@ -45,21 +54,22 @@ static BOOL test_inq_if_ids(struct dcerpc_pipe *p,
                return False;
        }
 
-       if (!r.out.if_id_vector) {
+       if (!vector) {
                printf("inq_if_ids gave NULL if_id_vector\n");
                return False;
        }
 
-       for (i=0;i<r.out.if_id_vector->count;i++) {
-               const char *uuid;
-               struct dcerpc_syntax_id *id = r.out.if_id_vector->if_id[i].id;
+       for (i=0;i<vector->count;i++) {
+               struct ndr_syntax_id *id = vector->if_id[i].id;
                if (!id) continue;
 
-               uuid = GUID_string(mem_ctx, &id->uuid);
-
                printf("\tuuid %s  version 0x%08x  '%s'\n",
-                      uuid,
-                      id->if_version, idl_pipe_name(uuid, id->if_version));
+                      GUID_string(mem_ctx, &id->uuid),
+                      id->if_version, idl_pipe_name(&id->uuid, id->if_version));
+
+               if (per_id_test) {
+                       per_id_test(priv, mem_ctx, id);
+               }
        }
 
        return True;
@@ -70,9 +80,11 @@ static BOOL test_inq_stats(struct dcerpc_pipe *p,
 {
        NTSTATUS status;
        struct mgmt_inq_stats r;
+       struct mgmt_statistics statistics;
 
        r.in.max_count = MGMT_STATS_ARRAY_MAX_SIZE;
        r.in.unknown = 0;
+       r.out.statistics = &statistics;
 
        status = dcerpc_mgmt_inq_stats(p, mem_ctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
@@ -80,16 +92,16 @@ static BOOL test_inq_stats(struct dcerpc_pipe *p,
                return False;
        }
 
-       if (r.out.statistics.count != MGMT_STATS_ARRAY_MAX_SIZE) {
-               printf("Unexpected array size %d\n", r.out.statistics.count);
+       if (statistics.count != MGMT_STATS_ARRAY_MAX_SIZE) {
+               printf("Unexpected array size %d\n", statistics.count);
                return False;
        }
 
        printf("\tcalls_in %6d  calls_out %6d\n\tpkts_in  %6d  pkts_out  %6d\n",
-              r.out.statistics.statistics[MGMT_STATS_CALLS_IN],
-              r.out.statistics.statistics[MGMT_STATS_CALLS_OUT],
-              r.out.statistics.statistics[MGMT_STATS_PKTS_IN],
-              r.out.statistics.statistics[MGMT_STATS_PKTS_OUT]);
+              statistics.statistics[MGMT_STATS_CALLS_IN],
+              statistics.statistics[MGMT_STATS_CALLS_OUT],
+              statistics.statistics[MGMT_STATS_PKTS_IN],
+              statistics.statistics[MGMT_STATS_PKTS_OUT]);
 
        return True;
 }
@@ -135,6 +147,7 @@ static BOOL test_is_server_listening(struct dcerpc_pipe *p,
 {
        NTSTATUS status;
        struct mgmt_is_server_listening r;
+       r.out.status = talloc(mem_ctx, uint32_t);
 
        status = dcerpc_mgmt_is_server_listening(p, mem_ctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
@@ -142,7 +155,7 @@ static BOOL test_is_server_listening(struct dcerpc_pipe *p,
                return False;
        }
 
-       if (r.out.status != 0 || r.out.result == 0) {
+       if (*r.out.status != 0 || r.out.result == 0) {
                printf("\tserver is NOT listening\n");
        } else {
                printf("\tserver is listening\n");
@@ -174,15 +187,15 @@ static BOOL test_stop_server_listening(struct dcerpc_pipe *p,
 }
 
 
-BOOL torture_rpc_mgmt(void)
+BOOL torture_rpc_mgmt(struct torture_context *torture)
 {
         NTSTATUS status;
         struct dcerpc_pipe *p;
-       TALLOC_CTX *mem_ctx;
+       TALLOC_CTX *mem_ctx, *loop_ctx;
        BOOL ret = True;
-       int i;
-       const char *binding = lp_parm_string(-1, "torture", "binding");
-       struct dcerpc_binding b;
+       const char *binding = torture_setting_string(torture, "binding", NULL);
+       const struct ndr_interface_list *l;
+       struct dcerpc_binding *b;
 
        mem_ctx = talloc_init("torture_rpc_mgmt");
 
@@ -193,63 +206,66 @@ BOOL torture_rpc_mgmt(void)
        
        status = dcerpc_parse_binding(mem_ctx, binding, &b);
        if (!NT_STATUS_IS_OK(status)) {
+               talloc_free(mem_ctx);
                printf("Failed to parse binding '%s'\n", binding);
                return False;
        }
 
-       for (i=0;dcerpc_pipes[i];i++) {         
+       for (l=librpc_dcerpc_pipes();l;l=l->next) {             
+               loop_ctx = talloc_named(mem_ctx, 0, "torture_rpc_mgmt loop context");
+               
                /* some interfaces are not mappable */
-               if (dcerpc_pipes[i]->num_calls == 0 ||
-                   strcmp(dcerpc_pipes[i]->name, "mgmt") == 0) {
+               if (l->table->num_calls == 0 ||
+                   strcmp(l->table->name, "mgmt") == 0) {
+                       talloc_free(loop_ctx);
                        continue;
                }
 
-               printf("\nTesting pipe '%s'\n", dcerpc_pipes[i]->name);
+               printf("\nTesting pipe '%s'\n", l->table->name);
 
-               if (b.transport == NCACN_IP_TCP) {
-                       status = dcerpc_epm_map_binding(mem_ctx, &b, 
-                                                        dcerpc_pipes[i]->uuid,
-                                                        dcerpc_pipes[i]->if_version);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               printf("Failed to map port for uuid %s\n", dcerpc_pipes[i]->uuid);
-                               continue;
-                       }
-               } else {
-                       b.endpoint = dcerpc_pipes[i]->name;
+               status = dcerpc_epm_map_binding(loop_ctx, b, l->table, NULL);
+               if (!NT_STATUS_IS_OK(status)) {
+                       printf("Failed to map port for uuid %s\n", 
+                                  GUID_string(loop_ctx, &l->table->syntax_id.uuid));
+                       talloc_free(loop_ctx);
+                       continue;
                }
 
-               lp_set_cmdline("torture:binding", dcerpc_binding_string(mem_ctx, &b));
+               lp_set_cmdline("torture:binding", dcerpc_binding_string(loop_ctx, b));
+
+               status = torture_rpc_connection(loop_ctx, &p, &ndr_table_mgmt);
+               if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
+                       printf("Interface not available - skipping\n");
+                       talloc_free(loop_ctx);
+                       continue;
+               }
 
-               status = torture_rpc_connection(&p, 
-                                               dcerpc_pipes[i]->name,
-                                               DCERPC_MGMT_UUID,
-                                               DCERPC_MGMT_VERSION);
                if (!NT_STATUS_IS_OK(status)) {
+                       talloc_free(loop_ctx);
                        ret = False;
                        continue;
                }
-       
-               if (!test_is_server_listening(p, mem_ctx)) {
+
+               if (!test_is_server_listening(p, loop_ctx)) {
                        ret = False;
                }
 
-               if (!test_stop_server_listening(p, mem_ctx)) {
+               if (!test_stop_server_listening(p, loop_ctx)) {
                        ret = False;
                }
 
-               if (!test_inq_stats(p, mem_ctx)) {
+               if (!test_inq_stats(p, loop_ctx)) {
                        ret = False;
                }
 
-               if (!test_inq_princ_name(p, mem_ctx)) {
+               if (!test_inq_princ_name(p, loop_ctx)) {
                        ret = False;
                }
 
-               if (!test_inq_if_ids(p, mem_ctx)) {
+               if (!test_inq_if_ids(p, loop_ctx, NULL, NULL)) {
                        ret = False;
                }
 
-               torture_rpc_close(p);
        }
 
        return ret;