staging/cli_arp.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
  21 #include "mz.h"
  22 #include "cli.h"
  23 #include "mops.h"
  24
  25
  26 // ISSUES:
  27 //
  28 // - Currently only IP/MAC resolution supported (i.e. hw_size=6, pr_size=4)
  29 // - Add macro support: commands like request/response should set all params correctly
  30
  31
  32
  33
  34 int cmd_arp_hwtype (struct cli_def *cli, const char *command, char *argv[], int argc)
  35 {
  36    struct mops_ext_arp * pd = clipkt->p_desc;
  37
  38    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
  39      {
  40         cli_print(cli,"Specify the hardware type (0-ffff, default 1=Eth)\n");
  41      }
  42    else if (mops_pdesc_2byte(&pd->hw_type, argv[0], 1, 0, 0xffff))
  43      {
  44         cli_print(cli, "Hardware type must be between 0 and ffff\n");
  45      }
  46
  47    return CLI_OK;
  48 }
  49
  50
  51
  52 int cmd_arp_prtype (struct cli_def *cli, const char *command, char *argv[], int argc)
  53 {
  54    struct mops_ext_arp * pd = clipkt->p_desc;
  55
  56    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
  57      {
  58         cli_print(cli,"Specify the protocol type (0-ffff, default=800=IP)\n");
  59      }
  60    else if (mops_pdesc_2byte(&pd->pr_type, argv[0], 1, 0, 0xffff))
  61      {
  62         cli_print(cli, "Protocol type must be between 0 and ffff\n");
  63      }
  64
  65    return CLI_OK;
  66 }
  67
  68
  69
  70 int cmd_arp_hwaddrsize (struct cli_def *cli, const char *command, char *argv[], int argc)
  71 {
  72    struct mops_ext_arp * pd = clipkt->p_desc;
  73
  74    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
  75      {
  76         cli_print(cli,"Specify the hardware address size (0-255, default=6)\n");
  77      }
  78    else if (mops_pdesc_1byte(&pd->hw_size, argv[0], 0, 0, 255))
  79      {
  80         cli_print(cli, "Hardware size must be between 0 and 255\n");
  81      }
  82
  83      return CLI_OK;
  84 }
  85
  86
  87 int cmd_arp_praddrsize (struct cli_def *cli, const char *command, char *argv[], int argc)
  88 {
  89    struct mops_ext_arp * pd = clipkt->p_desc;
  90
  91    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
  92      {
  93         cli_print(cli,"Specify the protocol address size (0-255, default=4)\n");
  94      }
  95    else if (mops_pdesc_1byte(&pd->pr_size, argv[0], 0, 0, 255))
  96      {
  97         cli_print(cli, "Protocol size must be between 0 and 255\n");
  98      }
  99
 100      return CLI_OK;
 101 }
 102
 103
 104 int cmd_arp_opcode (struct cli_def *cli, const char *command, char *argv[], int argc)
 105 {
 106    struct mops_ext_arp * pd = clipkt->p_desc;
 107
 108    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 109      {
 110         cli_print(cli,"Specify the ARP operation code (0-ffff)\n");
 111         cli_print(cli,"Optional keywords: 'request' (default) or 'response'\n");
 112      }
 113    else if (mz_strcmp(argv[0],"request", 3)==0)
 114      {
 115         cli_print(cli, "Set ARP mode to request\n");
 116         pd->opcode = 1;
 117      }
 118    else if (mz_strcmp(argv[0],"response", 3)==0)
 119      {
 120         cli_print(cli, "Set ARP mode to response\n");
 121         pd->opcode = 2;
 122      }
 123    else
 124      {
 125         cli_print(cli, "Invalid ARP mode\n");
 126      }
 127
 128    return CLI_OK;
 129 }
 130
 131
 132
 133 int cmd_arp_smac (struct cli_def *cli, const char *command, char *argv[], int argc)
 134 {
 135    struct mops_ext_arp * pd = clipkt->p_desc;
 136
 137    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 138      {
 139         cli_print(cli,"Specify a source MAC address (format: XX:XX:XX:XX:XX:XX)\n");
 140      }
 141    else
 142      {
 143         if (mops_pdesc_mac(pd->sender_mac, argv[0]))
 144           {
 145              cli_print(cli,"Invalid MAC address (use format: XX:XX:XX:XX:XX:XX)\n");
 146           }
 147      }
 148    return CLI_OK;
 149 }
 150
 151
 152
 153 int cmd_arp_sip (struct cli_def *cli, const char *command, char *argv[], int argc)
 154 {
 155    struct mops_ext_arp * pd = clipkt->p_desc;
 156
 157    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 158      {
 159         cli_print(cli,"Specify a source IP address (format: A.B.C.D)\n");
 160      }
 161    else if (mops_pdesc_ip (pd->sender_ip, argv[0]))
 162      {
 163         cli_print(cli,"Invalid IP address (use format: A.B.C.D)\n");
 164      }
 165
 166    return CLI_OK;
 167 }
 168
 169
 170
 171 int cmd_arp_tmac (struct cli_def *cli, const char *command, char *argv[], int argc)
 172 {
 173    struct mops_ext_arp * pd = clipkt->p_desc;
 174
 175    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 176      {
 177         cli_print(cli,"Specify a target MAC address (format: XX:XX:XX:XX:XX:XX)\n");
 178      }
 179    else if (mops_pdesc_mac(pd->target_mac, argv[0]))
 180      {
 181         cli_print(cli,"Invalid MAC address (use format: XX:XX:XX:XX:XX:XX)\n");
 182      }
 183
 184    return CLI_OK;
 185 }
 186
 187
 188
 189 int cmd_arp_tip (struct cli_def *cli, const char *command, char *argv[], int argc)
 190 {
 191    struct mops_ext_arp * pd = clipkt->p_desc;
 192
 193    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 194      {
 195         cli_print(cli,"Specify a target IP address (format: A.B.C.D)\n");
 196      }
 197    else if (mops_pdesc_ip (pd->target_ip, argv[0]))
 198      {
 199         cli_print(cli,"Invalid IP address (use format: A.B.C.D)\n");
 200      }
 201
 202    return CLI_OK;
 203 }
 204
 205
 206
 207 int cmd_arp_trailer (struct cli_def *cli, const char *command, char *argv[], int argc)
 208 {
 209    struct mops_ext_arp * pd = (MOPS_EXT_ARP) clipkt->p_desc;
 210
 211    if ( (strncmp(argv[argc-1],"?",1)==0) || (argc!=1) )
 212      {
 213         cli_print(cli,"Specify the trailer length (0-2000, default=18)\n");
 214      }
 215    else if (mops_pdesc_2byte(&pd->trailer, argv[0], 0, 0, 2000))
 216      {
 217         cli_print(cli, "Trailer must be between 0 and 2000\n");
 218      }
 219
 220    return CLI_OK;
 221 }
 222
 223
 224
 225 int cmd_arp_end(struct cli_def *cli, const char *command, char *argv[], int argc)
 226 {
 227    char prompt[16];
 228    sprintf(prompt, "pkt-%i",clipkt->id);
 229    cli_set_configmode(cli, MZ_MODE_PACKET, prompt);
 230    return CLI_OK;
 231 }
 232