s4:rpc_server: only use context within op_bind() hooks and dcesrv_interface_bind_...
[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 3 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, see <http://www.gnu.org/licenses/>.
21 */
22
23 #include "includes.h"
24 #include "system/filesys.h"
25 #include "rpc_server/dcerpc_server.h"
26 #include "librpc/gen_ndr/ndr_echo.h"
27 #include "lib/events/events.h"
28
29 #define DCESRV_INTERFACE_RPCECHO_BIND(call, iface) \
30        dcesrv_interface_rpcecho_bind(call, iface)
31 static NTSTATUS dcesrv_interface_rpcecho_bind(struct dcesrv_call_state *dce_call,
32                                               const struct dcesrv_interface *iface)
33 {
34         struct dcesrv_connection_context *context = dce_call->context;
35         return dcesrv_interface_bind_allow_connect(context, iface);
36 }
37
38 static NTSTATUS dcesrv_echo_AddOne(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_AddOne *r)
39 {
40         *r->out.out_data = r->in.in_data + 1;
41         return NT_STATUS_OK;
42 }
43
44 static NTSTATUS dcesrv_echo_EchoData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_EchoData *r)
45 {
46         if (!r->in.len) {
47                 return NT_STATUS_OK;
48         }
49
50         r->out.out_data = (uint8_t *)talloc_memdup(mem_ctx, r->in.in_data, r->in.len);
51         if (!r->out.out_data) {
52                 return NT_STATUS_NO_MEMORY;
53         }
54
55         return NT_STATUS_OK;
56 }
57
58 static NTSTATUS dcesrv_echo_SinkData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_SinkData *r)
59 {
60         return NT_STATUS_OK;
61 }
62
63 static NTSTATUS dcesrv_echo_SourceData(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_SourceData *r)
64 {
65         unsigned int i;
66
67         r->out.data = talloc_array(mem_ctx, uint8_t, r->in.len);
68         if (!r->out.data) {
69                 return NT_STATUS_NO_MEMORY;
70         }
71         
72         for (i=0;i<r->in.len;i++) {
73                 r->out.data[i] = i;
74         }
75
76         return NT_STATUS_OK;
77 }
78
79 static NTSTATUS dcesrv_echo_TestCall(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall *r)
80 {
81         *r->out.s2 = talloc_strdup(mem_ctx, r->in.s1);
82         if (r->in.s1 && !*r->out.s2) {
83                 return NT_STATUS_NO_MEMORY;
84         }
85         return NT_STATUS_OK;
86 }
87
88 static NTSTATUS dcesrv_echo_TestCall2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall2 *r)
89 {
90         r->out.info = talloc(mem_ctx, union echo_Info);
91         if (!r->out.info) {
92                 return NT_STATUS_NO_MEMORY;
93         }
94
95         switch (r->in.level) {
96         case 1:
97                 r->out.info->info1.v = 10;
98                 break;
99         case 2:
100                 r->out.info->info2.v = 20;
101                 break;
102         case 3:
103                 r->out.info->info3.v = 30;
104                 break;
105         case 4:
106                 r->out.info->info4.v = 40;
107                 break;
108         case 5:
109                 r->out.info->info5.v1 = 50;
110                 r->out.info->info5.v2 = 60;
111                 break;
112         case 6:
113                 r->out.info->info6.v1 = 70;
114                 r->out.info->info6.info1.v= 80;
115                 break;
116         case 7:
117                 r->out.info->info7.v1 = 80;
118                 r->out.info->info7.info4.v = 90;
119                 break;
120         default:
121                 return NT_STATUS_INVALID_LEVEL;
122         }
123
124         return NT_STATUS_OK;
125 }
126
127 static NTSTATUS dcesrv_echo_TestEnum(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestEnum *r)
128 {
129         r->out.foo2->e1 = ECHO_ENUM2;
130         return NT_STATUS_OK;
131 }
132
133 static NTSTATUS dcesrv_echo_TestSurrounding(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSurrounding *r)
134 {
135         if (!r->in.data) {
136                 r->out.data = NULL;
137                 return NT_STATUS_OK;
138         }
139
140         r->out.data = talloc(mem_ctx, struct echo_Surrounding);
141         if (!r->out.data) {
142                 return NT_STATUS_NO_MEMORY;
143         }
144         r->out.data->x = 2 * r->in.data->x;
145         r->out.data->surrounding = talloc_zero_array(mem_ctx, uint16_t, r->out.data->x);
146         if (!r->out.data->surrounding) {
147                 return NT_STATUS_NO_MEMORY;
148         }
149
150         return NT_STATUS_OK;
151 }
152
153 static uint16_t dcesrv_echo_TestDoublePointer(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestDoublePointer *r) 
154 {
155         if (!*r->in.data) 
156                 return 0;
157         if (!**r->in.data)
158                 return 0;
159         return ***r->in.data;
160 }
161
162 struct echo_TestSleep_private {
163         struct dcesrv_call_state *dce_call;
164         struct echo_TestSleep *r;
165 };
166
167 static void echo_TestSleep_handler(struct tevent_context *ev, struct tevent_timer *te, 
168                                    struct timeval t, void *private_data)
169 {
170         struct echo_TestSleep_private *p = talloc_get_type(private_data,
171                                                            struct echo_TestSleep_private);
172         struct echo_TestSleep *r = p->r;
173         NTSTATUS status;
174
175         r->out.result = r->in.seconds;
176
177         status = dcesrv_reply(p->dce_call);
178         if (!NT_STATUS_IS_OK(status)) {
179                 DEBUG(0,("echo_TestSleep_handler: dcesrv_reply() failed - %s\n",
180                         nt_errstr(status)));
181         }
182 }
183
184 static long dcesrv_echo_TestSleep(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSleep *r)
185 {
186         struct echo_TestSleep_private *p;
187
188         if (!(dce_call->state_flags & DCESRV_CALL_STATE_FLAG_MAY_ASYNC)) {
189                 /* we're not allowed to reply async */
190                 sleep(r->in.seconds);
191                 return r->in.seconds;
192         }
193
194         /* we're allowed to reply async */
195         p = talloc(mem_ctx, struct echo_TestSleep_private);
196         if (!p) {
197                 return 0;
198         }
199
200         p->dce_call     = dce_call;
201         p->r            = r;
202
203         tevent_add_timer(dce_call->event_ctx, p,
204                         timeval_add(&dce_call->time, r->in.seconds, 0),
205                         echo_TestSleep_handler, p);
206
207         dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
208         return 0;
209 }
210
211 /* include the generated boilerplate */
212 #include "librpc/gen_ndr/ndr_echo_s.c"