|
Packit Service |
b25606 |
/*
|
|
Packit Service |
b25606 |
* $Id: udp2.c,v 1.6 2004/03/01 20:26:12 mike Exp $
|
|
Packit Service |
b25606 |
*
|
|
Packit Service |
b25606 |
* libnet 1.1
|
|
Packit Service |
b25606 |
* Build a UDP packet using port list chains
|
|
Packit Service |
b25606 |
*
|
|
Packit Service |
b25606 |
* Copyright (c) 1998 - 2004 Mike D. Schiffman <mike@infonexus.com>
|
|
Packit Service |
b25606 |
* All rights reserved.
|
|
Packit Service |
b25606 |
*
|
|
Packit Service |
b25606 |
* Redistribution and use in source and binary forms, with or without
|
|
Packit Service |
b25606 |
* modification, are permitted provided that the following conditions
|
|
Packit Service |
b25606 |
* are met:
|
|
Packit Service |
b25606 |
* 1. Redistributions of source code must retain the above copyright
|
|
Packit Service |
b25606 |
* notice, this list of conditions and the following disclaimer.
|
|
Packit Service |
b25606 |
* 2. Redistributions in binary form must reproduce the above copyright
|
|
Packit Service |
b25606 |
* notice, this list of conditions and the following disclaimer in the
|
|
Packit Service |
b25606 |
* documentation and/or other materials provided with the distribution.
|
|
Packit Service |
b25606 |
*
|
|
Packit Service |
b25606 |
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
Packit Service |
b25606 |
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
Packit Service |
b25606 |
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
Packit Service |
b25606 |
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
Packit Service |
b25606 |
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
Packit Service |
b25606 |
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
Packit Service |
b25606 |
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
Packit Service |
b25606 |
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
Packit Service |
b25606 |
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
Packit Service |
b25606 |
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
Packit Service |
b25606 |
* SUCH DAMAGE.
|
|
Packit Service |
b25606 |
*
|
|
Packit Service |
b25606 |
*/
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
#if (HAVE_CONFIG_H)
|
|
Packit Service |
b25606 |
#include "../include/config.h"
|
|
Packit Service |
b25606 |
#endif
|
|
Packit Service |
b25606 |
#include "./libnet_test.h"
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
int
|
|
Packit Service |
b25606 |
main(int argc, char **argv)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
int c, build_ip;
|
|
Packit Service |
b25606 |
struct timeval r;
|
|
Packit Service |
b25606 |
struct timeval s;
|
|
Packit Service |
b25606 |
struct timeval e;
|
|
Packit Service |
b25606 |
libnet_t *l;
|
|
Packit Service |
b25606 |
libnet_ptag_t udp;
|
|
Packit Service |
b25606 |
char *payload;
|
|
Packit Service |
b25606 |
libnet_ptag_t t;
|
|
Packit Service |
b25606 |
struct libnet_stats ls;
|
|
Packit Service |
b25606 |
uint16_t payload_s;
|
|
Packit Service |
b25606 |
uint32_t src_ip, dst_ip;
|
|
Packit Service |
b25606 |
uint16_t bport, eport, cport;
|
|
Packit Service |
b25606 |
libnet_plist_t plist, *plist_p;
|
|
Packit Service |
b25606 |
char errbuf[LIBNET_ERRBUF_SIZE];
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
printf("libnet 1.1.2 packet shaping: UDP2[link]\n");
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
l = libnet_init(
|
|
Packit Service |
b25606 |
LIBNET_LINK, /* injection type */
|
|
Packit Service |
b25606 |
NULL, /* network interface */
|
|
Packit Service |
b25606 |
errbuf); /* errbuf */
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
if (l == NULL)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "libnet_init() failed: %s", errbuf);
|
|
Packit Service |
b25606 |
exit(EXIT_FAILURE);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
src_ip = 0;
|
|
Packit Service |
b25606 |
dst_ip = 0;
|
|
Packit Service |
b25606 |
payload = NULL;
|
|
Packit Service |
b25606 |
payload_s = 0;
|
|
Packit Service |
b25606 |
plist_p = NULL;
|
|
Packit Service |
b25606 |
while ((c = getopt(argc, argv, "d:s:p:P:")) != EOF)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
switch (c)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
case 'd':
|
|
Packit Service |
b25606 |
if ((dst_ip = libnet_name2addr4(l, optarg,
|
|
Packit Service |
b25606 |
LIBNET_RESOLVE)) == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Bad destination IP address: %s\n", optarg);
|
|
Packit Service |
b25606 |
exit(1);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
break;
|
|
Packit Service |
b25606 |
case 's':
|
|
Packit Service |
b25606 |
if ((src_ip = libnet_name2addr4(l, optarg,
|
|
Packit Service |
b25606 |
LIBNET_RESOLVE)) == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Bad source IP address: %s\n", optarg);
|
|
Packit Service |
b25606 |
exit(1);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
break;
|
|
Packit Service |
b25606 |
case 'P':
|
|
Packit Service |
b25606 |
plist_p = &plist;
|
|
Packit Service |
b25606 |
if (libnet_plist_chain_new(l, &plist_p, optarg) == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Bad token in port list: %s\n",
|
|
Packit Service |
b25606 |
libnet_geterror(l));
|
|
Packit Service |
b25606 |
exit(1);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
break;
|
|
Packit Service |
b25606 |
case 'p':
|
|
Packit Service |
b25606 |
payload = optarg;
|
|
Packit Service |
b25606 |
payload_s = strlen(payload);
|
|
Packit Service |
b25606 |
break;
|
|
Packit Service |
b25606 |
default:
|
|
Packit Service |
b25606 |
usage(argv[0]);
|
|
Packit Service |
b25606 |
exit(EXIT_FAILURE);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
if (!src_ip || !dst_ip || !plist_p)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
usage(argv[0]);
|
|
Packit Service |
b25606 |
exit(EXIT_FAILURE);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
udp = 0;
|
|
Packit Service |
b25606 |
#if !(__WIN32__)
|
|
Packit Service |
b25606 |
gettimeofday(&s, NULL);
|
|
Packit Service |
b25606 |
#else
|
|
Packit Service |
b25606 |
/* This obviously is not as good - but it compiles now. */
|
|
Packit Service |
b25606 |
s.tv_sec = time(NULL);
|
|
Packit Service |
b25606 |
s.tv_usec = 0;
|
|
Packit Service |
b25606 |
#endif
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
build_ip = 1;
|
|
Packit Service |
b25606 |
while (libnet_plist_chain_next_pair(plist_p, &bport, &eport))
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
while (!(bport > eport) && bport != 0)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
cport = bport++;
|
|
Packit Service |
b25606 |
udp = libnet_build_udp(
|
|
Packit Service |
b25606 |
1025, /* source port */
|
|
Packit Service |
b25606 |
cport, /* destination port */
|
|
Packit Service |
b25606 |
LIBNET_UDP_H + payload_s, /* packet size */
|
|
Packit Service |
b25606 |
0, /* checksum */
|
|
Packit Service |
b25606 |
payload, /* payload */
|
|
Packit Service |
b25606 |
payload_s, /* payload size */
|
|
Packit Service |
b25606 |
l, /* libnet handle */
|
|
Packit Service |
b25606 |
udp); /* libnet id */
|
|
Packit Service |
b25606 |
if (udp == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Can't build UDP header (at port %d): %s\n",
|
|
Packit Service |
b25606 |
cport, libnet_geterror(l));
|
|
Packit Service |
b25606 |
goto bad;
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
if (build_ip)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
build_ip = 0;
|
|
Packit Service |
b25606 |
t = libnet_build_ipv4(
|
|
Packit Service |
b25606 |
LIBNET_IPV4_H + LIBNET_UDP_H + payload_s, /* length */
|
|
Packit Service |
b25606 |
0, /* TOS */
|
|
Packit Service |
b25606 |
242, /* IP ID */
|
|
Packit Service |
b25606 |
0, /* IP Frag */
|
|
Packit Service |
b25606 |
64, /* TTL */
|
|
Packit Service |
b25606 |
IPPROTO_UDP, /* protocol */
|
|
Packit Service |
b25606 |
0, /* checksum */
|
|
Packit Service |
b25606 |
src_ip, /* source IP */
|
|
Packit Service |
b25606 |
dst_ip, /* destination IP */
|
|
Packit Service |
b25606 |
NULL, /* payload */
|
|
Packit Service |
b25606 |
0, /* payload size */
|
|
Packit Service |
b25606 |
l, /* libnet handle */
|
|
Packit Service |
b25606 |
0);
|
|
Packit Service |
b25606 |
if (t == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Can't build IP header: %s\n", libnet_geterror(l));
|
|
Packit Service |
b25606 |
goto bad;
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
t = libnet_build_ethernet(
|
|
Packit Service |
b25606 |
enet_dst, /* ethernet dest */
|
|
Packit Service |
b25606 |
enet_src, /* ethernet source */
|
|
Packit Service |
b25606 |
ETHERTYPE_IP, /* protocol type */
|
|
Packit Service |
b25606 |
NULL, /* payload */
|
|
Packit Service |
b25606 |
0, /* payload size */
|
|
Packit Service |
b25606 |
l, /* libnet handle */
|
|
Packit Service |
b25606 |
0);
|
|
Packit Service |
b25606 |
if (t == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "Can't build ethernet header: %s\n",
|
|
Packit Service |
b25606 |
libnet_geterror(l));
|
|
Packit Service |
b25606 |
goto bad;
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
c = libnet_write(l);
|
|
Packit Service |
b25606 |
if (c == -1)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "write error: %s\n", libnet_geterror(l));
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
else
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "wrote %d byte UDP packet to port %d\r", c,
|
|
Packit Service |
b25606 |
cport);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
fprintf(stderr, "\n");
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
#if !(__WIN32__)
|
|
Packit Service |
b25606 |
gettimeofday(&e, NULL);
|
|
Packit Service |
b25606 |
#else
|
|
Packit Service |
b25606 |
/* This obviously is not as good - but it compiles now. */
|
|
Packit Service |
b25606 |
s.tv_sec = time(NULL);
|
|
Packit Service |
b25606 |
s.tv_usec = 0;
|
|
Packit Service |
b25606 |
#endif
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
libnet_timersub(&e, &s, &r);
|
|
Packit Service |
b25606 |
fprintf(stderr, "Total time spent in loop: %d.%d\n", r.tv_sec, r.tv_usec);
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
libnet_stats(l, &ls);
|
|
Packit Service |
b25606 |
fprintf(stderr, "Packets sent: %lld\n"
|
|
Packit Service |
b25606 |
"Packet errors: %lld\n"
|
|
Packit Service |
b25606 |
"Bytes written: %lld\n",
|
|
Packit Service |
b25606 |
ls.packets_sent, ls.packet_errors, ls.bytes_written);
|
|
Packit Service |
b25606 |
libnet_destroy(l);
|
|
Packit Service |
b25606 |
return (EXIT_SUCCESS);
|
|
Packit Service |
b25606 |
bad:
|
|
Packit Service |
b25606 |
libnet_destroy(l);
|
|
Packit Service |
b25606 |
return (EXIT_FAILURE);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
void
|
|
Packit Service |
b25606 |
usage(char *name)
|
|
Packit Service |
b25606 |
{
|
|
Packit Service |
b25606 |
fprintf(stderr, "usage: %s -s s_ip -d d_ip -P port list [-p payload]\n", name);
|
|
Packit Service |
b25606 |
}
|
|
Packit Service |
b25606 |
|
|
Packit Service |
b25606 |
/* EOF */
|