src/cli_udp.c

   1 /*
   2  * Mausezahn - A fast versatile traffic generator
   3  * Copyright (C) 2008-2010 Herbert Haas
   4  *
   5  * This program is free software; you can redistribute it and/or modify it under
   6  * the terms of the GNU General Public License version 2 as published by the
   7  * Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  11  * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
  12  * details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, see http://www.gnu.org/licenses/gpl-2.0.html
  16  *
  17 */
  18
  19
  20 #include "mz.h"
  21 #include "cli.h"
  22 #include "mops.h"
  23
  24
  25
  26 int cmd_port_source (struct cli_def *cli, char *command, char *argv[], int argc)
  27 {
  28         u_int32_t t32=0;
  29         int validport=0;
  30
  31         if ( (strcmp(argv[argc-1],"?")==0) || (argc>2) ) {
  32                 cli_print(cli, "Specify the source port number:\n");
  33                 cli_print(cli, " <port> [<end-port>]\r");
  34                 cli_print(cli, " random [norandom]\r");
  35                 cli_print(cli, "\n");
  36                 return CLI_OK;
  37         }
  38
  39         if (mz_strcmp(argv[0], "random",1)==0) {
  40                 clipkt->sp_isrand = 1;
  41                 clipkt->sp_isrange = 0;
  42         } else if  (mz_strcmp(argv[0], "norandom",1)==0) {
  43                 clipkt->sp_isrand = 0;
  44         } else {
  45                 if (!mz_strisnum(argv[0])) {
  46                         cli_print(cli, "Unknown keyword\n");
  47                         return CLI_OK;
  48                 }
  49                 t32 = str2int(argv[0]);
  50                 if (t32>65535) {
  51                         cli_print(cli, "Port number cannot exceed 65535\n");
  52                         return CLI_OK;
  53                 } else {
  54                         clipkt->sp= (u_int16_t) t32;
  55                         validport=1;
  56                         clipkt->sp_isrange = 0;
  57                 }
  58         }
  59
  60         if ((argc==2) && (validport)) {
  61                 if (!mz_strisnum(argv[1])) {
  62                         cli_print(cli, "Invalid number\n");
  63                         return CLI_OK;
  64                 }
  65                 t32 = str2int(argv[1]);
  66                 if (t32>65535) {
  67                         cli_print(cli, "Port number cannot exceed 65535\n");
  68                 } else {
  69                         clipkt->sp_start = clipkt->sp;
  70                         clipkt->sp_stop  = (u_int16_t) t32;
  71                         clipkt->sp_isrange = 1;
  72                 }
  73         }
  74
  75         return CLI_OK;
  76 }
  77
  78
  79
  80
  81 int cmd_port_destination (struct cli_def *cli, char *command, char *argv[], int argc)
  82 {
  83         u_int32_t t32=0;
  84         int validport=0;
  85
  86         if ( (strcmp(argv[argc-1],"?")==0) || (argc>2) ) {
  87                 cli_print(cli, "Specify the destination port number\r");
  88                 cli_print(cli, " <port> [<end-port>]\r");
  89                 cli_print(cli, " random [norandom]\r");
  90                 cli_print(cli, "\n");
  91                 return CLI_OK;
  92         }
  93
  94         if (mz_strcmp(argv[0], "random",1)==0) {
  95                 clipkt->dp_isrand = 1;
  96                 clipkt->dp_isrange = 0;
  97         } else if  (mz_strcmp(argv[0], "norandom",1)==0) {
  98                 clipkt->dp_isrand = 0;
  99         } else {
 100                 if (!mz_strisnum(argv[0])) {
 101                         cli_print(cli, "Unknown keyword\n");
 102                         return CLI_OK;
 103
 104                 }
 105                 t32 = str2int(argv[0]);
 106                 if (t32>65535) {
 107                         cli_print(cli, "Port number cannot exceed 65535\n");
 108                         return CLI_OK;
 109                 } else {
 110                         clipkt->dp= (u_int16_t) t32;
 111                         validport=1;
 112                         clipkt->dp_isrange = 0;
 113                 }
 114         }
 115
 116         if ((argc==2) && (validport)) {
 117                 if (!mz_strisnum(argv[1])) {
 118                         cli_print(cli, "Invalid number\n");
 119                         return CLI_OK;
 120                 }
 121                 t32 = str2int(argv[1]);
 122                 if (t32>65535) {
 123                         cli_print(cli, "Port number cannot exceed 65535\n");
 124                 } else {
 125                         clipkt->dp_start = clipkt->dp;
 126                         clipkt->dp_stop  = (u_int16_t) t32;
 127                         clipkt->dp_isrange = 1;
 128                 }
 129         }
 130
 131         return CLI_OK;
 132 }
 133
 134
 135
 136 int cmd_udp_sum (struct cli_def *cli, char *command, char *argv[], int argc)
 137 {
 138    int sum;
 139
 140    if ( (strcmp(argv[argc-1],"?")==0) || (argc>1) )
 141      {
 142         cli_print(cli, "Specify the UDP checksum:\n");
 143         cli_print(cli, " - either in hexadecimal format (0-ffff)\r");
 144         cli_print(cli, " - or use the keyword 'auto' (default)\r");
 145         cli_print(cli, " - or use the keyword 'unset'\r");
 146         cli_print(cli, "\r");
 147         cli_print(cli, "By default, the checksum is computed automatically. The keyword\r");
 148         cli_print(cli, "'unset' signals the receiver that the checksum has not be computed\r");
 149         cli_print(cli, "and should be ignored.\n");
 150         return CLI_OK;
 151      }
 152
 153    if (mz_strcmp(argv[0], "auto", 2)==0)
 154      {
 155         clipkt->udp_sum_false=0;
 156         return CLI_OK;
 157      }
 158
 159    if (mz_strcmp(argv[0], "unset", 2)==0)
 160      {
 161         clipkt->udp_sum_false=1;
 162         clipkt->udp_sum = 0xffff;
 163         return CLI_OK;
 164      }
 165
 166    sum = (int) xstr2int(argv[0]);
 167
 168    if (sum>0xffff)
 169      {
 170         cli_print(cli, "The checksum must be within range 0..ffff\n");
 171         return CLI_OK;
 172      }
 173
 174    clipkt->udp_sum = (u_int16_t) sum;
 175    clipkt->udp_sum_false=1;
 176
 177    return CLI_OK;
 178 }
 179
 180
 181
 182 int cmd_udp_len (struct cli_def *cli, char *command, char *argv[], int argc)
 183 {
 184
 185    if ( (strcmp(argv[argc-1],"?")==0) || (argc>1) )
 186      {
 187         cli_print(cli, "Specify the UDP length\r");
 188         cli_print(cli, "\n");
 189         return CLI_OK;
 190      }
 191
 192    cli_print(cli, "Not supported in this version.\n");
 193
 194    return CLI_OK;
 195 }
 196
 197
 198 int cmd_udp_end(struct cli_def *cli, char *command, char *argv[], int argc)
 199 {
 200    char prompt[16];
 201    sprintf(prompt, "pkt-%i",clipkt->id);
 202    cli_set_configmode(cli, MZ_MODE_PACKET, prompt);
 203    return CLI_OK;
 204 }