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_c.h"
25 #include "torture/rpc/rpc.h"
27 static BOOL test_JobGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t job_id)
30 struct atsvc_JobGetInfo r;
32 r.in.servername = dcerpc_server_name(p);
35 status = dcerpc_atsvc_JobGetInfo(p, mem_ctx, &r);
37 if (!NT_STATUS_IS_OK(status)) {
38 printf("JobGetInfo failed - %s\n", nt_errstr(status));
45 static BOOL test_JobDel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t min_job_id,
49 struct atsvc_JobDel r;
51 r.in.servername = dcerpc_server_name(p);
52 r.in.min_job_id = min_job_id;
53 r.in.max_job_id = max_job_id;
55 status = dcerpc_atsvc_JobDel(p, mem_ctx, &r);
57 if (!NT_STATUS_IS_OK(status)) {
58 printf("JobDel failed - %s\n", nt_errstr(status));
65 static BOOL test_JobEnum(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
68 struct atsvc_JobEnum r;
69 struct atsvc_enum_ctr ctr;
70 uint32_t resume_handle = 0, i;
73 printf("\ntesting JobEnum\n");
75 r.in.servername = dcerpc_server_name(p);
77 ctr.first_entry = NULL;
78 r.in.ctr = r.out.ctr = &ctr;
79 r.in.preferred_max_len = 0xffffffff;
80 r.in.resume_handle = r.out.resume_handle = &resume_handle;
82 status = dcerpc_atsvc_JobEnum(p, mem_ctx, &r);
84 if (!NT_STATUS_IS_OK(status)) {
85 printf("JobEnum failed - %s\n", nt_errstr(status));
89 for (i = 0; r.out.ctr && i < r.out.ctr->entries_read; i++) {
90 if (!test_JobGetInfo(p, mem_ctx, r.out.ctr->first_entry[i].job_id)) {
98 static BOOL test_JobAdd(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
101 struct atsvc_JobAdd r;
102 struct atsvc_JobInfo info;
104 printf("\ntesting JobAdd\n");
106 r.in.servername = dcerpc_server_name(p);
107 info.job_time = 0x050ae4c0; /* 11:30pm */
108 info.days_of_month = 0; /* n/a */
109 info.days_of_week = 0x02; /* Tuesday */
110 info.flags = 0x11; /* periodic, non-interactive */
111 info.command = "foo.exe";
112 r.in.job_info = &info;
114 status = dcerpc_atsvc_JobAdd(p, mem_ctx, &r);
116 if (!NT_STATUS_IS_OK(status)) {
117 printf("JobAdd failed - %s\n", nt_errstr(status));
121 /* Run EnumJobs again in case there were no jobs to begin with */
123 if (!test_JobEnum(p, mem_ctx)) {
127 if (!test_JobGetInfo(p, mem_ctx, r.out.job_id)) {
131 if (!test_JobDel(p, mem_ctx, r.out.job_id, r.out.job_id)) {
138 BOOL torture_rpc_atsvc(void)
141 struct dcerpc_pipe *p;
145 mem_ctx = talloc_init("torture_rpc_atsvc");
147 status = torture_rpc_connection(mem_ctx, &p, &dcerpc_table_atsvc);
148 if (!NT_STATUS_IS_OK(status)) {
149 talloc_free(mem_ctx);
153 if (!test_JobEnum(p, mem_ctx)) {
154 talloc_free(mem_ctx);
158 if (!test_JobAdd(p, mem_ctx)) {
159 talloc_free(mem_ctx);
163 talloc_free(mem_ctx);