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 uint16_t echo_TestDoublePointer(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestDoublePointer *r)
 155 {
 156         if (!*r->in.data)
 157                 return 0;
 158         if (!**r->in.data)
 159                 return 0;
 160         return ***r->in.data;
 161 }
 162
 163 static long echo_TestSleep(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct echo_TestSleep *r)
 164 {
 165         struct echo_TestSleep_private *p;
 166
 167         if (!(dce_call->state_flags & DCESRV_CALL_STATE_FLAG_MAY_ASYNC)) {
 168                 /* we're not allowed to reply async */
 169                 sleep(r->in.seconds);
 170                 return r->in.seconds;
 171         }
 172
 173         /* we're allowed to reply async */
 174         p = talloc(mem_ctx, struct echo_TestSleep_private);
 175         if (!p) {
 176                 return 0;
 177         }
 178
 179         p->dce_call     = dce_call;
 180         p->r            = r;
 181
 182         event_add_timed(dce_call->event_ctx, p,
 183                         timeval_add(&dce_call->time, r->in.seconds, 0),
 184                         echo_TestSleep_handler, p);
 185
 186         dce_call->state_flags |= DCESRV_CALL_STATE_FLAG_ASYNC;
 187         return 0;
 188 }
 189
 190 /* include the generated boilerplate */
 191 #include "librpc/gen_ndr/ndr_echo_s.c"