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.out_data = r->in.in_data + 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
  60         r->out.data = talloc_array(mem_ctx, uint8_t, r->in.len);
  61
  62         for (i=0;i<r->in.len;i++) {
  63                 r->out.data[i] = i;
  64         }
  65
  66         return NT_STATUS_OK;
  67 }
  68
  69 static NTSTATUS echo_TestCall(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall *r)
  70 {
  71         r->out.s2 = talloc_strdup(mem_ctx, "this is a test string");
  72
  73         return NT_STATUS_OK;
  74 }
  75
  76 static NTSTATUS echo_TestCall2(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestCall2 *r)
  77 {
  78         r->out.info = talloc(mem_ctx, union echo_Info);
  79         if (!r->out.info) {
  80                 return NT_STATUS_NO_MEMORY;
  81         }
  82
  83         switch (r->in.level) {
  84         case 1:
  85                 r->out.info->info1.v = 10;
  86                 break;
  87         case 2:
  88                 r->out.info->info2.v = 20;
  89                 break;
  90         case 3:
  91                 r->out.info->info3.v = 30;
  92                 break;
  93         case 4:
  94                 r->out.info->info4.v = 40;
  95                 break;
  96         case 5:
  97                 r->out.info->info5.v1 = 50;
  98                 r->out.info->info5.v2 = 60;
  99                 break;
 100         case 6:
 101                 r->out.info->info6.v1 = 70;
 102                 r->out.info->info6.info1.v= 80;
 103                 break;
 104         case 7:
 105                 r->out.info->info7.v1 = 80;
 106                 r->out.info->info7.info4.v = 90;
 107                 break;
 108         default:
 109                 return NT_STATUS_INVALID_LEVEL;
 110         }
 111
 112         return NT_STATUS_OK;
 113 }
 114
 115 static NTSTATUS echo_TestEnum(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestEnum *r)
 116 {
 117         r->out.foo2->e1 = ECHO_ENUM2;
 118         return NT_STATUS_OK;
 119 }
 120
 121 static NTSTATUS echo_TestSurrounding(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSurrounding *r)
 122 {
 123         if (!r->in.data) {
 124                 r->out.data = NULL;
 125                 return NT_STATUS_OK;
 126         }
 127
 128         r->out.data = talloc(mem_ctx, struct echo_Surrounding);
 129         r->out.data->x = 2 * r->in.data->x;
 130         r->out.data->surrounding = talloc_zero_array(mem_ctx, uint16_t, r->out.data->x);
 131
 132         return NT_STATUS_OK;
 133 }
 134
 135 static uint16_t echo_TestDoublePointer(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestDoublePointer *r)
 136 {
 137         if (!*r->in.data)
 138                 return 0;
 139         if (!**r->in.data)
 140                 return 0;
 141         return ***r->in.data;
 142 }
 143
 144 struct echo_TestSleep_private {
 145         struct dcesrv_call_state *dce_call;
 146         struct echo_TestSleep *r;
 147 };
 148
 149 static void echo_TestSleep_handler(struct event_context *ev, struct timed_event *te,
 150                                    struct timeval t, void *private)
 151 {
 152         struct echo_TestSleep_private *p = talloc_get_type(private,
 153                                                            struct echo_TestSleep_private);
 154         struct echo_TestSleep *r = p->r;
 155         NTSTATUS status;
 156
 157         r->out.result = r->in.seconds;
 158
 159         status = dcesrv_reply(p->dce_call);
 160         if (!NT_STATUS_IS_OK(status)) {
 161                 DEBUG(0,("echo_TestSleep_handler: dcesrv_reply() failed - %s\n",
 162                         nt_errstr(status)));
 163         }
 164 }
 165
 166 static long echo_TestSleep(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSleep *r)
 167 {
 168         struct echo_TestSleep_private *p;
 169
 170         if (!(dce_call->state_flags & DCESRV_CALL_STATE_FLAG_MAY_ASYNC)) {
 171                 /* we're not allowed to reply async */
 172                 sleep(r->in.seconds);
 173                 return r->in.seconds;
 174         }
 175
 176         /* we're allowed to reply async */
 177         p = talloc(mem_ctx, struct echo_TestSleep_private);
 178         if (!p) {
 179                 return 0;
 180         }
 181
 182         p->dce_call     = dce_call;
 183         p->r            = r;
 184
 185         event_add_timed(dce_call->event_ctx, p,
 186                         timeval_add(&dce_call->time, r->in.seconds, 0),
 187                         echo_TestSleep_handler, p);
 188
 189         dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
 190         return 0;
 191 }
 192
 193 /* include the generated boilerplate */
 194 #include "librpc/gen_ndr/ndr_echo_s.c"