a2d2e552d6979440f53ebc03fd9095d2bb0ce62f
[samba.git] / source4 / rpc_server / echo / rpc_echo.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    endpoint server for the echo pipe
5
6    Copyright (C) Andrew Tridgell 2003
7    Copyright (C) Stefan (metze) Metzmacher 2005
8    
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 2 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program; if not, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24 #include "includes.h"
25 #include "system/filesys.h"
26 #include "rpc_server/dcerpc_server.h"
27 #include "librpc/gen_ndr/ndr_echo.h"
28 #include "lib/events/events.h"
29
30
31 static NTSTATUS echo_AddOne(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_AddOne *r)
32 {
33         *r->out.v = *r->in.v + 1;
34         return NT_STATUS_OK;
35 }
36
37 static NTSTATUS echo_EchoData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_EchoData *r)
38 {
39         if (!r->in.len) {
40                 return NT_STATUS_OK;
41         }
42
43         r->out.out_data = talloc_memdup(mem_ctx, r->in.in_data, r->in.len);
44         if (!r->out.out_data) {
45                 return NT_STATUS_NO_MEMORY;
46         }
47
48         return NT_STATUS_OK;
49 }
50
51 static NTSTATUS echo_SinkData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_SinkData *r)
52 {
53         return NT_STATUS_OK;
54 }
55
56 static NTSTATUS echo_SourceData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_SourceData *r)
57 {
58         int i;
59         for (i=0;i<r->in.len;i++) {
60                 r->out.data[i] = i;
61         }
62
63         return NT_STATUS_OK;
64 }
65
66 static NTSTATUS echo_TestCall(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall *r)
67 {
68         r->out.s2 = "this is a test string";
69         
70         return NT_STATUS_OK;
71 }
72
73 static NTSTATUS echo_TestCall2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall2 *r)
74 {
75         r->out.info = talloc(mem_ctx, union echo_Info);
76         if (!r->out.info) {
77                 return NT_STATUS_NO_MEMORY;
78         }
79
80         switch (r->in.level) {
81         case 1:
82                 r->out.info->info1.v = 10;
83                 break;
84         case 2:
85                 r->out.info->info2.v = 20;
86                 break;
87         case 3:
88                 r->out.info->info3.v = 30;
89                 break;
90         case 4:
91                 r->out.info->info4.v = 40;
92                 break;
93         case 5:
94                 r->out.info->info5.v1 = 50;
95                 r->out.info->info5.v2 = 60;
96                 break;
97         case 6:
98                 r->out.info->info6.v1 = 70;
99                 r->out.info->info6.info1.v= 80;
100                 break;
101         case 7:
102                 r->out.info->info7.v1 = 80;
103                 r->out.info->info7.info4.v = 90;
104                 break;
105         default:
106                 return NT_STATUS_INVALID_LEVEL;
107         }
108
109         return NT_STATUS_OK;
110 }
111
112 static NTSTATUS echo_TestEnum(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestEnum *r)
113 {
114         r->out.foo2->e1 = ECHO_ENUM2;
115         return NT_STATUS_OK;
116 }
117
118 struct echo_TestSleep_private {
119         struct dcesrv_call_state *dce_call;
120         struct echo_TestSleep *r;
121 };
122
123 static void echo_TestSleep_handler(struct event_context *ev, struct timed_event *te, 
124                                    struct timeval t, void *private)
125 {
126         struct echo_TestSleep_private *p = talloc_get_type(private, 
127                                                            struct echo_TestSleep_private);
128         struct echo_TestSleep *r = p->r;
129         NTSTATUS status;
130
131         r->out.result = r->in.seconds;
132
133         status = dcesrv_reply(p->dce_call);
134         if (!NT_STATUS_IS_OK(status)) {
135                 DEBUG(0,("echo_TestSleep_handler: dcesrv_reply() failed - %s\n",
136                         nt_errstr(status)));
137         }
138 }
139
140 static NTSTATUS echo_TestSurrounding(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSurrounding *r)
141 {
142         if (!r->in.data) {
143                 r->out.data = NULL;
144                 return NT_STATUS_OK;
145         }
146
147         r->out.data = talloc(mem_ctx, struct echo_Surrounding);
148         r->out.data->x = 2 * r->in.data->x;
149         r->out.data->surrounding = talloc_zero_array(mem_ctx, uint16_t, r->out.data->x);
150
151         return NT_STATUS_OK;
152 }
153
154 static long echo_TestSleep(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSleep *r)
155 {
156         struct echo_TestSleep_private *p;
157
158         if (!(dce_call->state_flags & DCESRV_CALL_STATE_FLAG_MAY_ASYNC)) {
159                 /* we're not allowed to reply async */
160                 sleep(r->in.seconds);
161                 return r->in.seconds;
162         }
163
164         /* we're allowed to reply async */
165         p = talloc(mem_ctx, struct echo_TestSleep_private);
166         if (!p) {
167                 return 0;
168         }
169
170         p->dce_call     = dce_call;
171         p->r            = r;
172
173         event_add_timed(dce_call->event_ctx, p, 
174                         timeval_add(&dce_call->time, r->in.seconds, 0),
175                         echo_TestSleep_handler, p);
176
177         dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
178         return 0;
179 }
180
181 /* include the generated boilerplate */
182 #include "librpc/gen_ndr/ndr_echo_s.c"