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>