Annotation of embedaddon/libnet/sample/udp1.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  *  $Id: udp1.c,v 1.6 2004/03/01 20:26:12 mike Exp $
        !             3:  *
        !             4:  *  libnet 1.1
        !             5:  *  Build a UDP packet
        !             6:  *
        !             7:  *  Copyright (c) 1998 - 2004 Mike D. Schiffman <mike@infonexus.com>
        !             8:  *  All rights reserved.
        !             9:  *
        !            10:  * Redistribution and use in source and binary forms, with or without
        !            11:  * modification, are permitted provided that the following conditions
        !            12:  * are met:
        !            13:  * 1. Redistributions of source code must retain the above copyright
        !            14:  *    notice, this list of conditions and the following disclaimer.
        !            15:  * 2. Redistributions in binary form must reproduce the above copyright
        !            16:  *    notice, this list of conditions and the following disclaimer in the
        !            17:  *    documentation and/or other materials provided with the distribution.
        !            18:  *
        !            19:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
        !            20:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
        !            21:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
        !            22:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
        !            23:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
        !            24:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
        !            25:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
        !            26:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
        !            27:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
        !            28:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
        !            29:  * SUCH DAMAGE.
        !            30:  *
        !            31:  */
        !            32: 
        !            33: #if (HAVE_CONFIG_H)
        !            34: #include "../include/config.h"
        !            35: #endif
        !            36: #include "./libnet_test.h"
        !            37: #ifdef __WIN32__
        !            38: #include "../include/win32/getopt.h"
        !            39: #endif
        !            40: 
        !            41: int
        !            42: main(int argc, char *argv[])
        !            43: {
        !            44:     int c, i, j, build_ip;
        !            45:     u_char *cp;
        !            46:     libnet_t *l;
        !            47:     libnet_ptag_t ip, ipo;
        !            48:     libnet_ptag_t udp;
        !            49:     char *payload;
        !            50:     u_short payload_s;
        !            51:     struct libnet_stats ls;
        !            52:     u_long src_ip, dst_ip;
        !            53:     u_short src_prt, dst_prt;
        !            54:     u_char opt[20];
        !            55:     char errbuf[LIBNET_ERRBUF_SIZE];
        !            56: 
        !            57:     printf("libnet 1.1 packet shaping: UDP + IP options[raw]\n"); 
        !            58: 
        !            59:     /*
        !            60:      *  Initialize the library.  Root priviledges are required.
        !            61:      */
        !            62:     l = libnet_init(
        !            63:             LIBNET_RAW4,                            /* injection type */
        !            64:             NULL,                                   /* network interface */
        !            65:             errbuf);                                /* errbuf */
        !            66: 
        !            67:     if (l == NULL)
        !            68:     {
        !            69:         fprintf(stderr, "libnet_init() failed: %s\n", errbuf);
        !            70:         exit(EXIT_FAILURE);
        !            71:     }
        !            72: 
        !            73:     src_ip  = 0;
        !            74:     dst_ip  = 0;
        !            75:     src_prt = 0;
        !            76:     dst_prt = 0;
        !            77:     payload = NULL;
        !            78:     payload_s = 0;
        !            79:     ip = ipo = udp = 0;
        !            80:     while ((c = getopt(argc, argv, "d:s:p:")) != EOF)
        !            81:     {
        !            82:         switch (c)
        !            83:         {
        !            84:             /*
        !            85:              *  We expect the input to be of the form `ip.ip.ip.ip.port`.  We
        !            86:              *  point cp to the last dot of the IP address/port string and
        !            87:              *  then seperate them with a NULL byte.  The optarg now points to
        !            88:              *  just the IP address, and cp points to the port.
        !            89:              */
        !            90:             case 'd':
        !            91:                 if (!(cp = strrchr(optarg, '.')))
        !            92:                 {
        !            93:                     usage(argv[0]);
        !            94:                 }
        !            95:                 *cp++ = 0;
        !            96:                 dst_prt = (u_short)atoi(cp);
        !            97:                 if ((dst_ip = libnet_name2addr4(l, optarg, LIBNET_RESOLVE)) == -1)
        !            98:                 {
        !            99:                     fprintf(stderr, "Bad destination IP address: %s\n", optarg);                    exit(EXIT_FAILURE);
        !           100:                 }
        !           101:                 break;
        !           102:             case 's':
        !           103:                 if (!(cp = strrchr(optarg, '.')))
        !           104:                 {
        !           105:                     usage(argv[0]);
        !           106:                 }
        !           107:                 *cp++ = 0;
        !           108:                 src_prt = (u_short)atoi(cp);
        !           109:                 if ((src_ip = libnet_name2addr4(l, optarg, LIBNET_RESOLVE)) == -1)
        !           110:                 {
        !           111:                     fprintf(stderr, "Bad source IP address: %s\n", optarg);
        !           112:                     exit(EXIT_FAILURE);
        !           113:                 }
        !           114:                 break;
        !           115:             case 'p':
        !           116:                 payload = optarg;
        !           117:                 payload_s = strlen(payload);
        !           118:                 break;
        !           119:             default:
        !           120:                 exit(EXIT_FAILURE);
        !           121:         }
        !           122:     }
        !           123: 
        !           124:     if (!src_ip || !src_prt || !dst_ip || !dst_prt)
        !           125:     {
        !           126:         usage(argv[0]);
        !           127:         exit(EXIT_FAILURE);
        !           128:     }
        !           129: 
        !           130:     for (build_ip = 0, i = 0; i < 10; i++)
        !           131:     {
        !           132:         udp = libnet_build_udp(
        !           133:             src_prt,                                /* source port */
        !           134:             dst_prt + i,                            /* destination port */
        !           135:             LIBNET_UDP_H + payload_s,               /* packet length */
        !           136:             0,                                      /* checksum */
        !           137:             payload,                                /* payload */
        !           138:             payload_s,                              /* payload size */
        !           139:             l,                                      /* libnet handle */
        !           140:             udp);                                   /* libnet id */
        !           141:         if (udp == -1)
        !           142:         {
        !           143:             fprintf(stderr, "Can't build UDP header: %s\n", libnet_geterror(l));
        !           144:             goto bad;
        !           145:         }
        !           146: 
        !           147:         if (1)
        !           148:         {
        !           149:             build_ip = 0;
        !           150:             /* this is not a legal options string */
        !           151:             for (j = 0; j < 20; j++)
        !           152:             {
        !           153:                 opt[j] = libnet_get_prand(LIBNET_PR2);
        !           154:             }
        !           155:             ipo = libnet_build_ipv4_options(
        !           156:                 opt,
        !           157:                 20,
        !           158:                 l,
        !           159:                 ipo);
        !           160:             if (ipo == -1)
        !           161:             {
        !           162:                 fprintf(stderr, "Can't build IP options: %s\n", libnet_geterror(l));
        !           163:                 goto bad;
        !           164:             }
        !           165: 
        !           166:             ip = libnet_build_ipv4(
        !           167:                 LIBNET_IPV4_H + 20 + payload_s + LIBNET_UDP_H, /* length */
        !           168:                 0,                                          /* TOS */
        !           169:                 242,                                        /* IP ID */
        !           170:                 0,                                          /* IP Frag */
        !           171:                 64,                                         /* TTL */
        !           172:                 IPPROTO_UDP,                                /* protocol */
        !           173:                 0,                                          /* checksum */
        !           174:                 src_ip,
        !           175:                 dst_ip,
        !           176:                 NULL,                                       /* payload */
        !           177:                 0,                                          /* payload size */
        !           178:                 l,                                          /* libnet handle */
        !           179:                ip);                                         /* libnet id */
        !           180:             if (ip == -1)
        !           181:             {
        !           182:                 fprintf(stderr, "Can't build IP header: %s\n", libnet_geterror(l));
        !           183:                 goto bad;
        !           184:             }
        !           185:         }
        !           186: 
        !           187:         /*
        !           188:          *  Write it to the wire.
        !           189:          */
        !           190:         fprintf(stderr, "%d byte packet, ready to go\n",
        !           191:             libnet_getpacket_size(l));
        !           192:         c = libnet_write(l);
        !           193:         if (c == -1)
        !           194:         {
        !           195:             fprintf(stderr, "Write error: %s\n", libnet_geterror(l));
        !           196:             goto bad;
        !           197:         }
        !           198:         else
        !           199:         {
        !           200:             fprintf(stderr, "Wrote %d byte UDP packet; check the wire.\n", c);
        !           201:         }
        !           202:     }
        !           203:     libnet_stats(l, &ls);
        !           204:     fprintf(stderr, "Packets sent:  %lld\n"
        !           205:                     "Packet errors: %lld\n"
        !           206:                     "Bytes written: %lld\n",
        !           207:                     ls.packets_sent, ls.packet_errors, ls.bytes_written);
        !           208:     libnet_destroy(l);
        !           209:     return (EXIT_SUCCESS);
        !           210: bad:
        !           211:     libnet_destroy(l);
        !           212:     return (EXIT_FAILURE);
        !           213: }
        !           214: 
        !           215: void
        !           216: usage(char *name)
        !           217: {
        !           218:     fprintf(stderr,
        !           219:         "usage: %s -s source_ip.source_port -d destination_ip.destination_port"
        !           220:         " [-p payload]\n",
        !           221:         name);
        !           222: }
        !           223: /* EOF */

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>