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 "librpc/gen_ndr/ndr_atsvc.h"
25 static BOOL test_JobGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t job_id)
28 struct atsvc_JobGetInfo r;
30 r.in.servername = dcerpc_server_name(p);
33 status = dcerpc_atsvc_JobGetInfo(p, mem_ctx, &r);
35 if (!NT_STATUS_IS_OK(status)) {
36 printf("JobGetInfo failed - %s\n", nt_errstr(status));
43 static BOOL test_JobDel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, uint32_t min_job_id,
47 struct atsvc_JobDel r;
49 r.in.servername = dcerpc_server_name(p);
50 r.in.min_job_id = min_job_id;
51 r.in.max_job_id = max_job_id;
53 status = dcerpc_atsvc_JobDel(p, mem_ctx, &r);
55 if (!NT_STATUS_IS_OK(status)) {
56 printf("JobDel failed - %s\n", nt_errstr(status));
63 static BOOL test_JobEnum(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
66 struct atsvc_JobEnum r;
67 struct atsvc_enum_ctr ctr;
68 uint32_t resume_handle = 0, i;
71 printf("\ntesting JobEnum\n");
73 r.in.servername = dcerpc_server_name(p);
75 ctr.first_entry = NULL;
76 r.in.ctr = r.out.ctr = &ctr;
77 r.in.preferred_max_len = 0xffffffff;
78 r.in.resume_handle = r.out.resume_handle = &resume_handle;
80 status = dcerpc_atsvc_JobEnum(p, mem_ctx, &r);
82 if (!NT_STATUS_IS_OK(status)) {
83 printf("JobEnum failed - %s\n", nt_errstr(status));
87 for (i = 0; r.out.ctr && i < r.out.ctr->entries_read; i++) {
88 if (!test_JobGetInfo(p, mem_ctx, r.out.ctr->first_entry[i].job_id)) {
96 static BOOL test_JobAdd(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
99 struct atsvc_JobAdd r;
100 struct atsvc_JobInfo info;
102 printf("\ntesting JobAdd\n");
104 r.in.servername = dcerpc_server_name(p);
105 info.job_time = 0x050ae4c0; /* 11:30pm */
106 info.days_of_month = 0; /* n/a */
107 info.days_of_week = 0x02; /* Tuesday */
108 info.flags = 0x11; /* periodic, non-interactive */
109 info.command = "foo.exe";
110 r.in.job_info = &info;
112 status = dcerpc_atsvc_JobAdd(p, mem_ctx, &r);
114 if (!NT_STATUS_IS_OK(status)) {
115 printf("JobAdd failed - %s\n", nt_errstr(status));
119 /* Run EnumJobs again in case there were no jobs to begin with */
121 if (!test_JobEnum(p, mem_ctx)) {
125 if (!test_JobGetInfo(p, mem_ctx, r.out.job_id)) {
129 if (!test_JobDel(p, mem_ctx, r.out.job_id, r.out.job_id)) {
136 BOOL torture_rpc_atsvc(void)
139 struct dcerpc_pipe *p;
143 mem_ctx = talloc_init("torture_rpc_atsvc");
145 status = torture_rpc_connection(mem_ctx, &p, &dcerpc_table_atsvc);
146 if (!NT_STATUS_IS_OK(status)) {
147 talloc_free(mem_ctx);
151 if (!test_JobEnum(p, mem_ctx)) {
152 talloc_free(mem_ctx);
156 if (!test_JobAdd(p, mem_ctx)) {
157 talloc_free(mem_ctx);
161 talloc_free(mem_ctx);