2 Unix SMB/CIFS implementation.
3 test suite for atsvc rpc operations
5 Copyright (C) Tim Potter 2003
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.
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.
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.
23 #include "torture/torture.h"
24 #include "librpc/gen_ndr/ndr_atsvc.h"
25 #include "librpc/gen_ndr/ndr_atsvc_c.h"
26 #include "torture/rpc/rpc.h"
28 static BOOL test_JobGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t job_id)
31 struct atsvc_JobGetInfo r;
33 r.in.servername = dcerpc_server_name(p);
36 status = dcerpc_atsvc_JobGetInfo(p, mem_ctx, &r);
38 if (!NT_STATUS_IS_OK(status)) {
39 printf("JobGetInfo failed - %s\n", nt_errstr(status));
46 static BOOL test_JobDel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t min_job_id,
50 struct atsvc_JobDel r;
52 r.in.servername = dcerpc_server_name(p);
53 r.in.min_job_id = min_job_id;
54 r.in.max_job_id = max_job_id;
56 status = dcerpc_atsvc_JobDel(p, mem_ctx, &r);
58 if (!NT_STATUS_IS_OK(status)) {
59 printf("JobDel failed - %s\n", nt_errstr(status));
66 static BOOL test_JobEnum(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
69 struct atsvc_JobEnum r;
70 struct atsvc_enum_ctr ctr;
71 uint32_t resume_handle = 0, i;
74 printf("\ntesting JobEnum\n");
76 r.in.servername = dcerpc_server_name(p);
78 ctr.first_entry = NULL;
79 r.in.ctr = r.out.ctr = &ctr;
80 r.in.preferred_max_len = 0xffffffff;
81 r.in.resume_handle = r.out.resume_handle = &resume_handle;
83 status = dcerpc_atsvc_JobEnum(p, mem_ctx, &r);
85 if (!NT_STATUS_IS_OK(status)) {
86 printf("JobEnum failed - %s\n", nt_errstr(status));
90 for (i = 0; r.out.ctr && i < r.out.ctr->entries_read; i++) {
91 if (!test_JobGetInfo(p, mem_ctx, r.out.ctr->first_entry[i].job_id)) {
99 static BOOL test_JobAdd(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
102 struct atsvc_JobAdd r;
103 struct atsvc_JobInfo info;
105 printf("\ntesting JobAdd\n");
107 r.in.servername = dcerpc_server_name(p);
108 info.job_time = 0x050ae4c0; /* 11:30pm */
109 info.days_of_month = 0; /* n/a */
110 info.days_of_week = 0x02; /* Tuesday */
111 info.flags = 0x11; /* periodic, non-interactive */
112 info.command = "foo.exe";
113 r.in.job_info = &info;
115 status = dcerpc_atsvc_JobAdd(p, mem_ctx, &r);
117 if (!NT_STATUS_IS_OK(status)) {
118 printf("JobAdd failed - %s\n", nt_errstr(status));
122 /* Run EnumJobs again in case there were no jobs to begin with */
124 if (!test_JobEnum(p, mem_ctx)) {
128 if (!test_JobGetInfo(p, mem_ctx, r.out.job_id)) {
132 if (!test_JobDel(p, mem_ctx, r.out.job_id, r.out.job_id)) {
139 BOOL torture_rpc_atsvc(void)
142 struct dcerpc_pipe *p;
146 mem_ctx = talloc_init("torture_rpc_atsvc");
148 status = torture_rpc_connection(mem_ctx, &p, &dcerpc_table_atsvc);
149 if (!NT_STATUS_IS_OK(status)) {
150 talloc_free(mem_ctx);
154 if (!test_JobEnum(p, mem_ctx)) {
155 talloc_free(mem_ctx);
159 if (!test_JobAdd(p, mem_ctx)) {
160 talloc_free(mem_ctx);
164 talloc_free(mem_ctx);