1 files changed, 21 insertions, 15 deletions

diff --git a/source4/torture/rpc/wkssvc.c b/source4/torture/rpc/wkssvc.c
index 6b09565bda..92d1063f3b 100644
--- a/source4/torture/rpc/wkssvc.c
+++ b/source4/torture/rpc/wkssvc.c
@@ -22,11 +22,11 @@
 #include "includes.h"
 
 
-static BOOL test_QueryInfo(struct dcerpc_pipe *p, 
+static BOOL test_NetWkstaGetInfo(struct dcerpc_pipe *p, 
 			   TALLOC_CTX *mem_ctx)
 {
 	NTSTATUS status;
-	struct wkssvc_QueryInfo r;
+	struct wkssvc_NetWkstaGetInfo r;
 	uint16 levels[] = {100, 101, 102, 502};
 	int i;
 	BOOL ret = True;
@@ -35,42 +35,48 @@ static BOOL test_QueryInfo(struct dcerpc_pipe *p,
 
 	for (i=0;i<ARRAY_SIZE(levels);i++) {
 		r.in.level = levels[i];
-		printf("testing QueryInfo level %u\n", r.in.level);
-		status = dcerpc_wkssvc_QueryInfo(p, mem_ctx, &r);
+		printf("testing NetWkstaGetInfo level %u\n", r.in.level);
+		status = dcerpc_wkssvc_NetWkstaGetInfo(p, mem_ctx, &r);
 		if (!NT_STATUS_IS_OK(status)) {
-			printf("QueryInfo level %u failed - %s\n", r.in.level, nt_errstr(status));
+			printf("NetWkstaGetInfo level %u failed - %s\n", r.in.level, nt_errstr(status));
 			ret = False;
 		}
+		if (!W_ERROR_IS_OK(r.out.result)) {
+			printf("NetWkstaGetInfo level %u failed - %s\n", r.in.level, win_errstr(r.out.result));
+		}
 	}
 
 	return ret;
 }
 
 
-static BOOL test_TransportEnum(struct dcerpc_pipe *p, 
+static BOOL test_NetWkstaTransportEnum(struct dcerpc_pipe *p, 
 			       TALLOC_CTX *mem_ctx)
 {
 	NTSTATUS status;
-	struct wkssvc_TransportEnum r;
+	struct wkssvc_NetWkstaTransportEnum r;
 	BOOL ret = True;
 	uint32 resume_handle = 0;
-	struct wkssvc_TransportInfoArray array;
+	struct wkssvc_NetWkstaTransportCtr0 ctr0;
 
-	ZERO_STRUCT(array);
+	ZERO_STRUCT(ctr0);
 
 	r.in.server_name = dcerpc_server_name(p);
 	r.in.level = 0;
-	r.in.info.array = &array;
+	r.in.ctr.ctr0 = &ctr0;
 	r.in.max_buffer = (uint32)-1;
 	r.in.resume_handle = &resume_handle;
 	r.out.resume_handle = &resume_handle;
 
-	printf("testing TransportEnum\n");
-	status = dcerpc_wkssvc_TransportEnum(p, mem_ctx, &r);
+	printf("testing NetWkstaTransportEnum\n");
+	status = dcerpc_wkssvc_NetWkstaTransportEnum(p, mem_ctx, &r);
 	if (!NT_STATUS_IS_OK(status)) {
-		printf("TransportEnum failed - %s\n", nt_errstr(status));
+		printf("NetWkstaTransportEnum failed - %s\n", nt_errstr(status));
 		ret = False;
 	}
+	if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("NetWkstaTransportEnum level %u failed - %s\n", r.in.level, win_errstr(r.out.result));
+	}
 
 	return ret;
 }
@@ -96,11 +102,11 @@ BOOL torture_rpc_wkssvc(int dummy)
 
 	p->flags |= DCERPC_DEBUG_PRINT_BOTH;
 	
-	if (!test_QueryInfo(p, mem_ctx)) {
+	if (!test_NetWkstaGetInfo(p, mem_ctx)) {
 		ret = False;
 	}
 
-	if (!test_TransportEnum(p, mem_ctx)) {
+	if (!test_NetWkstaTransportEnum(p, mem_ctx)) {
 		ret = False;
 	}