source3/rpcclient/cmd_echo.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    RPC pipe client
   4
   5    Copyright (C) Tim Potter 2003
   6
   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 3 of the License, or
  10    (at your option) any later version.
  11
  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.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "rpcclient.h"
  23 #include "../librpc/gen_ndr/cli_echo.h"
  24
  25 static NTSTATUS cmd_echo_add_one(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
  26                                  int argc, const char **argv)
  27 {
  28         uint32 request = 1, response;
  29         NTSTATUS result;
  30
  31         if (argc > 2) {
  32                 printf("Usage: %s [num]\n", argv[0]);
  33                 return NT_STATUS_OK;
  34         }
  35
  36         if (argc == 2)
  37                 request = atoi(argv[1]);
  38
  39         result = rpccli_echo_AddOne(cli, mem_ctx, request, &response);
  40
  41         if (!NT_STATUS_IS_OK(result))
  42                 goto done;
  43
  44         printf("%d + 1 = %d\n", request, response);
  45
  46 done:
  47         return result;
  48 }
  49
  50 static NTSTATUS cmd_echo_data(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
  51                               int argc, const char **argv)
  52 {
  53         uint32 size, i;
  54         NTSTATUS result;
  55         uint8_t *in_data = NULL, *out_data = NULL;
  56
  57         if (argc != 2) {
  58                 printf("Usage: %s num\n", argv[0]);
  59                 return NT_STATUS_OK;
  60         }
  61
  62         size = atoi(argv[1]);
  63         if ( (in_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
  64                 printf("Failure to allocate buff of %d bytes\n",
  65                        size);
  66                 result = NT_STATUS_NO_MEMORY;
  67                 goto done;
  68         }
  69         if ( (out_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
  70                 printf("Failure to allocate buff of %d bytes\n",
  71                        size);
  72                 result = NT_STATUS_NO_MEMORY;
  73                 goto done;
  74         }
  75
  76         for (i = 0; i < size; i++)
  77                 in_data[i] = i & 0xff;
  78
  79         result = rpccli_echo_EchoData(cli, mem_ctx, size, in_data, out_data);
  80
  81         if (!NT_STATUS_IS_OK(result))
  82                 goto done;
  83
  84         for (i = 0; i < size; i++) {
  85                 if (in_data[i] != out_data[i]) {
  86                         printf("mismatch at offset %d, %d != %d\n",
  87                                i, in_data[i], out_data[i]);
  88                         result = NT_STATUS_UNSUCCESSFUL;
  89                 }
  90         }
  91
  92 done:
  93         SAFE_FREE(in_data);
  94         SAFE_FREE(out_data);
  95
  96         return result;
  97 }
  98
  99 static NTSTATUS cmd_echo_source_data(struct rpc_pipe_client *cli,
 100                                      TALLOC_CTX *mem_ctx, int argc,
 101                                      const char **argv)
 102 {
 103         uint32 size, i;
 104         NTSTATUS result;
 105         uint8_t *out_data = NULL;
 106
 107         if (argc != 2) {
 108                 printf("Usage: %s num\n", argv[0]);
 109                 return NT_STATUS_OK;
 110         }
 111
 112         size = atoi(argv[1]);
 113         if ( (out_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
 114                 printf("Failure to allocate buff of %d bytes\n",
 115                        size);
 116                 result = NT_STATUS_NO_MEMORY;
 117                 goto done;
 118         }
 119
 120
 121         result = rpccli_echo_SourceData(cli, mem_ctx, size, out_data);
 122
 123         if (!NT_STATUS_IS_OK(result))
 124                 goto done;
 125
 126         for (i = 0; i < size; i++) {
 127                 if (out_data && out_data[i] != (i & 0xff)) {
 128                         printf("mismatch at offset %d, %d != %d\n",
 129                                i, out_data[i], i & 0xff);
 130                         result = NT_STATUS_UNSUCCESSFUL;
 131                 }
 132         }
 133
 134 done:
 135
 136         SAFE_FREE(out_data);
 137         return result;
 138 }
 139
 140 static NTSTATUS cmd_echo_sink_data(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
 141                                    int argc, const char **argv)
 142 {
 143         uint32 size, i;
 144         NTSTATUS result;
 145         uint8_t *in_data = NULL;
 146
 147         if (argc != 2) {
 148                 printf("Usage: %s num\n", argv[0]);
 149                 return NT_STATUS_OK;
 150         }
 151
 152         size = atoi(argv[1]);
 153         if ( (in_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
 154                 printf("Failure to allocate buff of %d bytes\n",
 155                        size);
 156                 result = NT_STATUS_NO_MEMORY;
 157                 goto done;
 158         }
 159
 160         for (i = 0; i < size; i++)
 161                 in_data[i] = i & 0xff;
 162
 163         result = rpccli_echo_SinkData(cli, mem_ctx, size, in_data);
 164
 165         if (!NT_STATUS_IS_OK(result))
 166                 goto done;
 167
 168 done:
 169         SAFE_FREE(in_data);
 170
 171         return result;
 172 }
 173
 174 /* List of commands exported by this module */
 175
 176 struct cmd_set echo_commands[] = {
 177
 178         { "ECHO" },
 179
 180         { "echoaddone", RPC_RTYPE_NTSTATUS, cmd_echo_add_one,     NULL, &ndr_table_rpcecho.syntax_id, NULL, "Add one to a number", "" },
 181         { "echodata",   RPC_RTYPE_NTSTATUS, cmd_echo_data,        NULL, &ndr_table_rpcecho.syntax_id, NULL, "Echo data",           "" },
 182         { "sinkdata",   RPC_RTYPE_NTSTATUS, cmd_echo_sink_data,   NULL, &ndr_table_rpcecho.syntax_id, NULL, "Sink data",           "" },
 183         { "sourcedata", RPC_RTYPE_NTSTATUS, cmd_echo_source_data, NULL, &ndr_table_rpcecho.syntax_id, NULL, "Source data",         "" },
 184         { NULL }
 185 };