pktgen.c 90.6 KB
Newer Older
L
Linus Torvalds 已提交
1 2 3 4 5 6 7 8
/*
 * Authors:
 * Copyright 2001, 2002 by Robert Olsson <robert.olsson@its.uu.se>
 *                             Uppsala University and
 *                             Swedish University of Agricultural Sciences
 *
 * Alexey Kuznetsov  <kuznet@ms2.inr.ac.ru>
 * Ben Greear <greearb@candelatech.com>
9
 * Jens Låås <jens.laas@data.slu.se>
L
Linus Torvalds 已提交
10 11 12 13 14 15 16 17
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version
 * 2 of the License, or (at your option) any later version.
 *
 *
 * A tool for loading the network with preconfigurated packets.
18
 * The tool is implemented as a linux module.  Parameters are output
L
Linus Torvalds 已提交
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
 * device, delay (to hard_xmit), number of packets, and whether
 * to use multiple SKBs or just the same one.
 * pktgen uses the installed interface's output routine.
 *
 * Additional hacking by:
 *
 * Jens.Laas@data.slu.se
 * Improved by ANK. 010120.
 * Improved by ANK even more. 010212.
 * MAC address typo fixed. 010417 --ro
 * Integrated.  020301 --DaveM
 * Added multiskb option 020301 --DaveM
 * Scaling of results. 020417--sigurdur@linpro.no
 * Significant re-work of the module:
 *   *  Convert to threaded model to more efficiently be able to transmit
 *       and receive on multiple interfaces at once.
 *   *  Converted many counters to __u64 to allow longer runs.
 *   *  Allow configuration of ranges, like min/max IP address, MACs,
 *       and UDP-ports, for both source and destination, and can
 *       set to use a random distribution or sequentially walk the range.
 *   *  Can now change most values after starting.
 *   *  Place 12-byte packet in UDP payload with magic number,
 *       sequence number, and timestamp.
 *   *  Add receiver code that detects dropped pkts, re-ordered pkts, and
 *       latencies (with micro-second) precision.
 *   *  Add IOCTL interface to easily get counters & configuration.
 *   --Ben Greear <greearb@candelatech.com>
 *
47 48
 * Renamed multiskb to clone_skb and cleaned up sending core for two distinct
 * skb modes. A clone_skb=0 mode for Ben "ranges" work and a clone_skb != 0
L
Linus Torvalds 已提交
49
 * as a "fastpath" with a configurable number of clones after alloc's.
50 51
 * clone_skb=0 means all packets are allocated this also means ranges time
 * stamps etc can be used. clone_skb=100 means 1 malloc is followed by 100
L
Linus Torvalds 已提交
52 53
 * clones.
 *
54
 * Also moved to /proc/net/pktgen/
L
Linus Torvalds 已提交
55 56 57 58 59 60 61 62
 * --ro
 *
 * Sept 10:  Fixed threading/locking.  Lots of bone-headed and more clever
 *    mistakes.  Also merged in DaveM's patch in the -pre6 patch.
 * --Ben Greear <greearb@candelatech.com>
 *
 * Integrated to 2.5.x 021029 --Lucio Maciel (luciomaciel@zipmail.com.br)
 *
63
 *
L
Linus Torvalds 已提交
64 65 66 67
 * 021124 Finished major redesign and rewrite for new functionality.
 * See Documentation/networking/pktgen.txt for how to use this.
 *
 * The new operation:
68 69 70 71
 * For each CPU one thread/process is created at start. This process checks
 * for running devices in the if_list and sends packets until count is 0 it
 * also the thread checks the thread->control which is used for inter-process
 * communication. controlling process "posts" operations to the threads this
L
Linus Torvalds 已提交
72 73 74
 * way. The if_lock should be possible to remove when add/rem_device is merged
 * into this too.
 *
75 76
 * By design there should only be *one* "controlling" process. In practice
 * multiple write accesses gives unpredictable result. Understood by "write"
L
Linus Torvalds 已提交
77
 * to /proc gives result code thats should be read be the "writer".
78
 * For practical use this should be no problem.
L
Linus Torvalds 已提交
79
 *
80 81
 * Note when adding devices to a specific CPU there good idea to also assign
 * /proc/irq/XX/smp_affinity so TX-interrupts gets bound to the same CPU.
L
Linus Torvalds 已提交
82 83
 * --ro
 *
84
 * Fix refcount off by one if first packet fails, potential null deref,
L
Linus Torvalds 已提交
85 86 87 88 89 90 91
 * memleak 030710- KJP
 *
 * First "ranges" functionality for ipv6 030726 --ro
 *
 * Included flow support. 030802 ANK.
 *
 * Fixed unaligned access on IA-64 Grant Grundler <grundler@parisc-linux.org>
92
 *
L
Linus Torvalds 已提交
93 94 95
 * Remove if fix from added Harald Welte <laforge@netfilter.org> 040419
 * ia64 compilation fix from  Aron Griffis <aron@hp.com> 040604
 *
96
 * New xmit() return, do_div and misc clean up by Stephen Hemminger
L
Linus Torvalds 已提交
97 98
 * <shemminger@osdl.org> 040923
 *
99
 * Randy Dunlap fixed u64 printk compiler waring
L
Linus Torvalds 已提交
100 101 102 103
 *
 * Remove FCS from BW calculation.  Lennert Buytenhek <buytenh@wantstofly.org>
 * New time handling. Lennert Buytenhek <buytenh@wantstofly.org> 041213
 *
104
 * Corrections from Nikolai Malykh (nmalykh@bilim.com)
L
Linus Torvalds 已提交
105 106
 * Removed unused flags F_SET_SRCMAC & F_SET_SRCIP 041230
 *
107
 * interruptible_sleep_on_timeout() replaced Nishanth Aravamudan <nacc@us.ibm.com>
L
Linus Torvalds 已提交
108
 * 050103
109 110 111
 *
 * MPLS support by Steven Whitehouse <steve@chygwyn.com>
 *
F
Francesco Fondelli 已提交
112 113
 * 802.1Q/Q-in-Q support by Francesco Fondelli (FF) <francesco.fondelli@gmail.com>
 *
A
Adit Ranadive 已提交
114 115 116
 * Fixed src_mac command to set source mac of packet to value specified in
 * command by Adit Ranadive <adit.262@gmail.com>
 *
L
Linus Torvalds 已提交
117
 */
J
Joe Perches 已提交
118 119 120

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

L
Linus Torvalds 已提交
121 122 123 124 125
#include <linux/sys.h>
#include <linux/types.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/kernel.h>
126
#include <linux/mutex.h>
L
Linus Torvalds 已提交
127 128 129 130 131 132 133 134 135
#include <linux/sched.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include <linux/unistd.h>
#include <linux/string.h>
#include <linux/ptrace.h>
#include <linux/errno.h>
#include <linux/ioport.h>
#include <linux/interrupt.h>
136
#include <linux/capability.h>
S
Stephen Hemminger 已提交
137
#include <linux/hrtimer.h>
A
Andrew Morton 已提交
138
#include <linux/freezer.h>
L
Linus Torvalds 已提交
139 140
#include <linux/delay.h>
#include <linux/timer.h>
141
#include <linux/list.h>
L
Linus Torvalds 已提交
142 143 144 145 146 147 148
#include <linux/init.h>
#include <linux/skbuff.h>
#include <linux/netdevice.h>
#include <linux/inet.h>
#include <linux/inetdevice.h>
#include <linux/rtnetlink.h>
#include <linux/if_arp.h>
F
Francesco Fondelli 已提交
149
#include <linux/if_vlan.h>
L
Linus Torvalds 已提交
150 151 152 153 154
#include <linux/in.h>
#include <linux/ip.h>
#include <linux/ipv6.h>
#include <linux/udp.h>
#include <linux/proc_fs.h>
155
#include <linux/seq_file.h>
L
Linus Torvalds 已提交
156
#include <linux/wait.h>
157
#include <linux/etherdevice.h>
158
#include <linux/kthread.h>
159
#include <linux/prefetch.h>
160
#include <net/net_namespace.h>
L
Linus Torvalds 已提交
161 162 163
#include <net/checksum.h>
#include <net/ipv6.h>
#include <net/addrconf.h>
J
Jamal Hadi Salim 已提交
164 165 166
#ifdef CONFIG_XFRM
#include <net/xfrm.h>
#endif
L
Linus Torvalds 已提交
167 168
#include <asm/byteorder.h>
#include <linux/rcupdate.h>
J
Jiri Slaby 已提交
169
#include <linux/bitops.h>
170 171 172
#include <linux/io.h>
#include <linux/timex.h>
#include <linux/uaccess.h>
L
Linus Torvalds 已提交
173
#include <asm/dma.h>
L
Luiz Capitulino 已提交
174
#include <asm/div64.h>		/* do_div */
L
Linus Torvalds 已提交
175

176
#define VERSION	"2.74"
L
Linus Torvalds 已提交
177
#define IP_NAME_SZ 32
178
#define MAX_MPLS_LABELS 16 /* This is the max label stack depth */
179
#define MPLS_STACK_BOTTOM htonl(0x00000100)
L
Linus Torvalds 已提交
180

J
Joe Perches 已提交
181 182
#define func_enter() pr_debug("entering %s\n", __func__);

L
Linus Torvalds 已提交
183
/* Device flag bits */
L
Luiz Capitulino 已提交
184 185 186 187 188 189 190 191
#define F_IPSRC_RND   (1<<0)	/* IP-Src Random  */
#define F_IPDST_RND   (1<<1)	/* IP-Dst Random  */
#define F_UDPSRC_RND  (1<<2)	/* UDP-Src Random */
#define F_UDPDST_RND  (1<<3)	/* UDP-Dst Random */
#define F_MACSRC_RND  (1<<4)	/* MAC-Src Random */
#define F_MACDST_RND  (1<<5)	/* MAC-Dst Random */
#define F_TXSIZE_RND  (1<<6)	/* Transmit size is random */
#define F_IPV6        (1<<7)	/* Interface in IPV6 Mode */
192
#define F_MPLS_RND    (1<<8)	/* Random MPLS labels */
F
Francesco Fondelli 已提交
193 194
#define F_VID_RND     (1<<9)	/* Random VLAN ID */
#define F_SVID_RND    (1<<10)	/* Random SVLAN ID */
195
#define F_FLOW_SEQ    (1<<11)	/* Sequential flows */
J
Jamal Hadi Salim 已提交
196
#define F_IPSEC_ON    (1<<12)	/* ipsec on for flows */
R
Robert Olsson 已提交
197
#define F_QUEUE_MAP_RND (1<<13)	/* queue map Random */
R
Robert Olsson 已提交
198
#define F_QUEUE_MAP_CPU (1<<14)	/* queue map mirrors smp_processor_id() */
R
Robert Olsson 已提交
199
#define F_NODE          (1<<15)	/* Node memory alloc*/
L
Linus Torvalds 已提交
200 201

/* Thread control flag bits */
202 203 204 205
#define T_STOP        (1<<0)	/* Stop run */
#define T_RUN         (1<<1)	/* Start run */
#define T_REMDEVALL   (1<<2)	/* Remove all devs */
#define T_REMDEV      (1<<3)	/* Remove one dev */
L
Linus Torvalds 已提交
206 207 208 209 210 211 212

/* If lock -- can be removed after some work */
#define   if_lock(t)           spin_lock(&(t->if_lock));
#define   if_unlock(t)           spin_unlock(&(t->if_lock));

/* Used to help with determining the pkts on receive */
#define PKTGEN_MAGIC 0xbe9be955
213 214
#define PG_PROC_DIR "pktgen"
#define PGCTRL	    "pgctrl"
215
static struct proc_dir_entry *pg_proc_dir;
L
Linus Torvalds 已提交
216 217 218

#define MAX_CFLOWS  65536

F
Francesco Fondelli 已提交
219 220 221
#define VLAN_TAG_SIZE(x) ((x)->vlan_id == 0xffff ? 0 : 4)
#define SVLAN_TAG_SIZE(x) ((x)->svlan_id == 0xffff ? 0 : 4)

L
Luiz Capitulino 已提交
222
struct flow_state {
A
Al Viro 已提交
223
	__be32 cur_daddr;
L
Luiz Capitulino 已提交
224
	int count;
J
Jamal Hadi Salim 已提交
225 226 227
#ifdef CONFIG_XFRM
	struct xfrm_state *x;
#endif
228
	__u32 flags;
L
Linus Torvalds 已提交
229 230
};

231 232 233
/* flow flag bits */
#define F_INIT   (1<<0)		/* flow has been initialized */

L
Linus Torvalds 已提交
234 235 236 237
struct pktgen_dev {
	/*
	 * Try to keep frequent/infrequent used vars. separated.
	 */
238 239
	struct proc_dir_entry *entry;	/* proc file */
	struct pktgen_thread *pg_thread;/* the owner */
240
	struct list_head list;		/* chaining in the thread's run-queue */
L
Linus Torvalds 已提交
241

242
	int running;		/* if false, the test will stop */
L
Linus Torvalds 已提交
243

L
Luiz Capitulino 已提交
244 245 246 247
	/* If min != max, then we will either do a linear iteration, or
	 * we will do a random selection from within the range.
	 */
	__u32 flags;
248 249
	int removal_mark;	/* non-zero => the device is marked for
				 * removal by worker thread */
L
Linus Torvalds 已提交
250

L
Luiz Capitulino 已提交
251 252
	int min_pkt_size;	/* = ETH_ZLEN; */
	int max_pkt_size;	/* = ETH_ZLEN; */
253
	int pkt_overhead;	/* overhead for MPLS, VLANs, IPSEC etc */
L
Luiz Capitulino 已提交
254
	int nfrags;
E
Eric Dumazet 已提交
255
	struct page *page;
256 257
	u64 delay;		/* nano-seconds */

L
Luiz Capitulino 已提交
258 259 260
	__u64 count;		/* Default No packets to send */
	__u64 sofar;		/* How many pkts we've sent so far */
	__u64 tx_bytes;		/* How many bytes we've transmitted */
261
	__u64 errors;		/* Errors when trying to transmit, */
L
Luiz Capitulino 已提交
262 263 264 265 266 267

	/* runtime counters relating to clone_skb */

	__u64 allocated_skbs;
	__u32 clone_count;
	int last_ok;		/* Was last skb sent?
268 269
				 * Or a failed transmit of some sort?
				 * This will keep sequence numbers in order
L
Luiz Capitulino 已提交
270
				 */
271 272 273 274 275
	ktime_t next_tx;
	ktime_t started_at;
	ktime_t stopped_at;
	u64	idle_acc;	/* nano-seconds */

L
Luiz Capitulino 已提交
276 277
	__u32 seq_num;

278 279 280 281 282 283 284 285
	int clone_skb;		/*
				 * Use multiple SKBs during packet gen.
				 * If this number is greater than 1, then
				 * that many copies of the same packet will be
				 * sent before a new packet is allocated.
				 * If you want to send 1024 identical packets
				 * before creating a new packet,
				 * set clone_skb to 1024.
L
Luiz Capitulino 已提交
286 287 288 289 290 291 292 293 294 295 296
				 */

	char dst_min[IP_NAME_SZ];	/* IP, ie 1.2.3.4 */
	char dst_max[IP_NAME_SZ];	/* IP, ie 1.2.3.4 */
	char src_min[IP_NAME_SZ];	/* IP, ie 1.2.3.4 */
	char src_max[IP_NAME_SZ];	/* IP, ie 1.2.3.4 */

	struct in6_addr in6_saddr;
	struct in6_addr in6_daddr;
	struct in6_addr cur_in6_daddr;
	struct in6_addr cur_in6_saddr;
L
Linus Torvalds 已提交
297
	/* For ranges */
L
Luiz Capitulino 已提交
298 299 300 301 302 303 304 305
	struct in6_addr min_in6_daddr;
	struct in6_addr max_in6_daddr;
	struct in6_addr min_in6_saddr;
	struct in6_addr max_in6_saddr;

	/* If we're doing ranges, random or incremental, then this
	 * defines the min/max for those ranges.
	 */
A
Al Viro 已提交
306 307 308 309
	__be32 saddr_min;	/* inclusive, source IP address */
	__be32 saddr_max;	/* exclusive, source IP address */
	__be32 daddr_min;	/* inclusive, dest IP address */
	__be32 daddr_max;	/* exclusive, dest IP address */
L
Luiz Capitulino 已提交
310 311 312 313 314 315

	__u16 udp_src_min;	/* inclusive, source UDP port */
	__u16 udp_src_max;	/* exclusive, source UDP port */
	__u16 udp_dst_min;	/* inclusive, dest UDP port */
	__u16 udp_dst_max;	/* exclusive, dest UDP port */

F
Francesco Fondelli 已提交
316
	/* DSCP + ECN */
317 318 319 320
	__u8 tos;            /* six MSB of (former) IPv4 TOS
				are for dscp codepoint */
	__u8 traffic_class;  /* ditto for the (former) Traffic Class in IPv6
				(see RFC 3260, sec. 4) */
F
Francesco Fondelli 已提交
321

322
	/* MPLS */
323
	unsigned int nr_labels;	/* Depth of stack, 0 = no MPLS */
324 325
	__be32 labels[MAX_MPLS_LABELS];

F
Francesco Fondelli 已提交
326 327 328 329 330 331 332 333 334
	/* VLAN/SVLAN (802.1Q/Q-in-Q) */
	__u8  vlan_p;
	__u8  vlan_cfi;
	__u16 vlan_id;  /* 0xffff means no vlan tag */

	__u8  svlan_p;
	__u8  svlan_cfi;
	__u16 svlan_id; /* 0xffff means no svlan tag */

L
Luiz Capitulino 已提交
335 336 337 338 339 340 341 342
	__u32 src_mac_count;	/* How many MACs to iterate through */
	__u32 dst_mac_count;	/* How many MACs to iterate through */

	unsigned char dst_mac[ETH_ALEN];
	unsigned char src_mac[ETH_ALEN];

	__u32 cur_dst_mac_offset;
	__u32 cur_src_mac_offset;
A
Al Viro 已提交
343 344
	__be32 cur_saddr;
	__be32 cur_daddr;
E
Eric Dumazet 已提交
345
	__u16 ip_id;
L
Luiz Capitulino 已提交
346 347
	__u16 cur_udp_dst;
	__u16 cur_udp_src;
R
Robert Olsson 已提交
348
	__u16 cur_queue_map;
L
Luiz Capitulino 已提交
349
	__u32 cur_pkt_size;
350
	__u32 last_pkt_size;
L
Luiz Capitulino 已提交
351 352 353 354 355 356 357 358 359 360 361 362

	__u8 hh[14];
	/* = {
	   0x00, 0x80, 0xC8, 0x79, 0xB3, 0xCB,

	   We fill in SRC address later
	   0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
	   0x08, 0x00
	   };
	 */
	__u16 pad;		/* pad out the hh struct to an even 16 bytes */

363
	struct sk_buff *skb;	/* skb we are to transmit next, used for when we
L
Luiz Capitulino 已提交
364 365
				 * are transmitting the same one multiple times
				 */
366 367 368 369 370 371 372 373
	struct net_device *odev; /* The out-going device.
				  * Note that the device should have it's
				  * pg_info pointer pointing back to this
				  * device.
				  * Set when the user specifies the out-going
				  * device name (not when the inject is
				  * started as it used to do.)
				  */
E
Eric Dumazet 已提交
374
	char odevname[32];
L
Linus Torvalds 已提交
375
	struct flow_state *flows;
376 377 378 379
	unsigned int cflows;	/* Concurrent flows (config) */
	unsigned int lflow;		/* Flow length  (config) */
	unsigned int nflows;	/* accumulated flows (stats) */
	unsigned int curfl;		/* current sequenced flow (state)*/
R
Robert Olsson 已提交
380 381 382

	u16 queue_map_min;
	u16 queue_map_max;
J
John Fastabend 已提交
383
	__u32 skb_priority;	/* skb priority field */
R
Robert Olsson 已提交
384
	int node;               /* Memory node */
R
Robert Olsson 已提交
385

J
Jamal Hadi Salim 已提交
386 387 388 389
#ifdef CONFIG_XFRM
	__u8	ipsmode;		/* IPSEC mode (config) */
	__u8	ipsproto;		/* IPSEC type (config) */
#endif
390
	char result[512];
L
Linus Torvalds 已提交
391 392 393
};

struct pktgen_hdr {
A
Al Viro 已提交
394 395 396 397
	__be32 pgh_magic;
	__be32 seq_num;
	__be32 tv_sec;
	__be32 tv_usec;
L
Linus Torvalds 已提交
398 399
};

E
Eric Dumazet 已提交
400 401
static bool pktgen_exiting __read_mostly;

L
Linus Torvalds 已提交
402
struct pktgen_thread {
403
	spinlock_t if_lock;		/* for list of devices */
404
	struct list_head if_list;	/* All device here */
405
	struct list_head th_list;
406
	struct task_struct *tsk;
L
Luiz Capitulino 已提交
407 408
	char result[512];

409 410
	/* Field for thread to receive "posted" events terminate,
	   stop ifs etc. */
L
Luiz Capitulino 已提交
411 412

	u32 control;
L
Linus Torvalds 已提交
413 414
	int cpu;

L
Luiz Capitulino 已提交
415
	wait_queue_head_t queue;
416
	struct completion start_done;
L
Linus Torvalds 已提交
417 418 419 420 421
};

#define REMOVE 1
#define FIND   0

422
static inline ktime_t ktime_now(void)
L
Linus Torvalds 已提交
423
{
424 425 426 427
	struct timespec ts;
	ktime_get_ts(&ts);

	return timespec_to_ktime(ts);
L
Linus Torvalds 已提交
428 429
}

430 431
/* This works even if 32 bit because of careful byte order choice */
static inline int ktime_lt(const ktime_t cmp1, const ktime_t cmp2)
L
Linus Torvalds 已提交
432
{
433
	return cmp1.tv64 < cmp2.tv64;
L
Linus Torvalds 已提交
434 435
}

S
Stephen Hemminger 已提交
436
static const char version[] =
J
Joe Perches 已提交
437 438
	"Packet Generator for packet performance testing. "
	"Version: " VERSION "\n";
L
Linus Torvalds 已提交
439

L
Luiz Capitulino 已提交
440 441 442
static int pktgen_remove_device(struct pktgen_thread *t, struct pktgen_dev *i);
static int pktgen_add_device(struct pktgen_thread *t, const char *ifname);
static struct pktgen_dev *pktgen_find_dev(struct pktgen_thread *t,
E
Eric Dumazet 已提交
443
					  const char *ifname, bool exact);
L
Linus Torvalds 已提交
444 445
static int pktgen_device_event(struct notifier_block *, unsigned long, void *);
static void pktgen_run_all_threads(void);
446
static void pktgen_reset_all_threads(void);
L
Linus Torvalds 已提交
447
static void pktgen_stop_all_threads_ifs(void);
S
Stephen Hemminger 已提交
448

L
Luiz Capitulino 已提交
449
static void pktgen_stop(struct pktgen_thread *t);
L
Linus Torvalds 已提交
450
static void pktgen_clear_counters(struct pktgen_dev *pkt_dev);
451

L
Luiz Capitulino 已提交
452
static unsigned int scan_ip6(const char *s, char ip[16]);
L
Linus Torvalds 已提交
453 454

/* Module parameters, defaults. */
455 456 457 458
static int pg_count_d __read_mostly = 1000;
static int pg_delay_d __read_mostly;
static int pg_clone_skb_d  __read_mostly;
static int debug  __read_mostly;
L
Linus Torvalds 已提交
459

460
static DEFINE_MUTEX(pktgen_thread_lock);
461
static LIST_HEAD(pktgen_threads);
L
Linus Torvalds 已提交
462 463 464 465 466 467

static struct notifier_block pktgen_notifier_block = {
	.notifier_call = pktgen_device_event,
};

/*
468
 * /proc handling functions
L
Linus Torvalds 已提交
469 470 471
 *
 */

472
static int pgctrl_show(struct seq_file *seq, void *v)
L
Luiz Capitulino 已提交
473
{
S
Stephen Hemminger 已提交
474
	seq_puts(seq, version);
475
	return 0;
L
Linus Torvalds 已提交
476 477
}

478 479
static ssize_t pgctrl_write(struct file *file, const char __user *buf,
			    size_t count, loff_t *ppos)
L
Linus Torvalds 已提交
480 481
{
	int err = 0;
482
	char data[128];
L
Linus Torvalds 已提交
483

L
Luiz Capitulino 已提交
484 485
	if (!capable(CAP_NET_ADMIN)) {
		err = -EPERM;
L
Linus Torvalds 已提交
486
		goto out;
L
Luiz Capitulino 已提交
487
	}
L
Linus Torvalds 已提交
488

489 490
	if (count > sizeof(data))
		count = sizeof(data);
L
Linus Torvalds 已提交
491 492

	if (copy_from_user(data, buf, count)) {
493
		err = -EFAULT;
494
		goto out;
L
Luiz Capitulino 已提交
495 496
	}
	data[count - 1] = 0;	/* Make string */
L
Linus Torvalds 已提交
497

L
Luiz Capitulino 已提交
498
	if (!strcmp(data, "stop"))
L
Linus Torvalds 已提交
499 500
		pktgen_stop_all_threads_ifs();

L
Luiz Capitulino 已提交
501
	else if (!strcmp(data, "start"))
L
Linus Torvalds 已提交
502 503
		pktgen_run_all_threads();

504 505 506
	else if (!strcmp(data, "reset"))
		pktgen_reset_all_threads();

L
Luiz Capitulino 已提交
507
	else
J
Joe Perches 已提交
508
		pr_warning("Unknown command: %s\n", data);
L
Linus Torvalds 已提交
509 510 511

	err = count;

L
Luiz Capitulino 已提交
512 513
out:
	return err;
L
Linus Torvalds 已提交
514 515
}

516 517 518 519 520
static int pgctrl_open(struct inode *inode, struct file *file)
{
	return single_open(file, pgctrl_show, PDE(inode)->data);
}

521
static const struct file_operations pktgen_fops = {
L
Luiz Capitulino 已提交
522 523 524 525 526 527
	.owner   = THIS_MODULE,
	.open    = pgctrl_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
	.write   = pgctrl_write,
	.release = single_release,
528 529 530
};

static int pktgen_if_show(struct seq_file *seq, void *v)
L
Linus Torvalds 已提交
531
{
S
Stephen Hemminger 已提交
532
	const struct pktgen_dev *pkt_dev = seq->private;
533 534
	ktime_t stopped;
	u64 idle;
L
Luiz Capitulino 已提交
535 536 537 538 539 540 541

	seq_printf(seq,
		   "Params: count %llu  min_pkt_size: %u  max_pkt_size: %u\n",
		   (unsigned long long)pkt_dev->count, pkt_dev->min_pkt_size,
		   pkt_dev->max_pkt_size);

	seq_printf(seq,
542 543
		   "     frags: %d  delay: %llu  clone_skb: %d  ifname: %s\n",
		   pkt_dev->nfrags, (unsigned long long) pkt_dev->delay,
E
Eric Dumazet 已提交
544
		   pkt_dev->clone_skb, pkt_dev->odevname);
L
Luiz Capitulino 已提交
545 546 547 548

	seq_printf(seq, "     flows: %u flowlen: %u\n", pkt_dev->cflows,
		   pkt_dev->lflow);

R
Robert Olsson 已提交
549 550 551 552 553
	seq_printf(seq,
		   "     queue_map_min: %u  queue_map_max: %u\n",
		   pkt_dev->queue_map_min,
		   pkt_dev->queue_map_max);

J
John Fastabend 已提交
554 555 556 557
	if (pkt_dev->skb_priority)
		seq_printf(seq, "     skb_priority: %u\n",
			   pkt_dev->skb_priority);

L
Luiz Capitulino 已提交
558 559
	if (pkt_dev->flags & F_IPV6) {
		seq_printf(seq,
560 561 562 563 564 565
			   "     saddr: %pI6c  min_saddr: %pI6c  max_saddr: %pI6c\n"
			   "     daddr: %pI6c  min_daddr: %pI6c  max_daddr: %pI6c\n",
			   &pkt_dev->in6_saddr,
			   &pkt_dev->min_in6_saddr, &pkt_dev->max_in6_saddr,
			   &pkt_dev->in6_daddr,
			   &pkt_dev->min_in6_daddr, &pkt_dev->max_in6_daddr);
566 567 568 569
	} else {
		seq_printf(seq,
			   "     dst_min: %s  dst_max: %s\n",
			   pkt_dev->dst_min, pkt_dev->dst_max);
L
Luiz Capitulino 已提交
570
		seq_printf(seq,
571 572 573
			   "        src_min: %s  src_max: %s\n",
			   pkt_dev->src_min, pkt_dev->src_max);
	}
L
Linus Torvalds 已提交
574

L
Luiz Capitulino 已提交
575
	seq_puts(seq, "     src_mac: ");
L
Linus Torvalds 已提交
576

J
Johannes Berg 已提交
577 578 579
	seq_printf(seq, "%pM ",
		   is_zero_ether_addr(pkt_dev->src_mac) ?
			     pkt_dev->odev->dev_addr : pkt_dev->src_mac);
L
Luiz Capitulino 已提交
580 581

	seq_printf(seq, "dst_mac: ");
J
Johannes Berg 已提交
582
	seq_printf(seq, "%pM\n", pkt_dev->dst_mac);
L
Luiz Capitulino 已提交
583 584

	seq_printf(seq,
585 586
		   "     udp_src_min: %d  udp_src_max: %d"
		   "  udp_dst_min: %d  udp_dst_max: %d\n",
L
Luiz Capitulino 已提交
587 588 589 590
		   pkt_dev->udp_src_min, pkt_dev->udp_src_max,
		   pkt_dev->udp_dst_min, pkt_dev->udp_dst_max);

	seq_printf(seq,
591
		   "     src_mac_count: %d  dst_mac_count: %d\n",
L
Luiz Capitulino 已提交
592
		   pkt_dev->src_mac_count, pkt_dev->dst_mac_count);
L
Linus Torvalds 已提交
593

594
	if (pkt_dev->nr_labels) {
595
		unsigned int i;
596
		seq_printf(seq, "     mpls: ");
597
		for (i = 0; i < pkt_dev->nr_labels; i++)
598 599 600 601
			seq_printf(seq, "%08x%s", ntohl(pkt_dev->labels[i]),
				   i == pkt_dev->nr_labels-1 ? "\n" : ", ");
	}

602
	if (pkt_dev->vlan_id != 0xffff)
F
Francesco Fondelli 已提交
603
		seq_printf(seq, "     vlan_id: %u  vlan_p: %u  vlan_cfi: %u\n",
604 605
			   pkt_dev->vlan_id, pkt_dev->vlan_p,
			   pkt_dev->vlan_cfi);
F
Francesco Fondelli 已提交
606

607
	if (pkt_dev->svlan_id != 0xffff)
F
Francesco Fondelli 已提交
608
		seq_printf(seq, "     svlan_id: %u  vlan_p: %u  vlan_cfi: %u\n",
609 610
			   pkt_dev->svlan_id, pkt_dev->svlan_p,
			   pkt_dev->svlan_cfi);
F
Francesco Fondelli 已提交
611

612
	if (pkt_dev->tos)
F
Francesco Fondelli 已提交
613 614
		seq_printf(seq, "     tos: 0x%02x\n", pkt_dev->tos);

615
	if (pkt_dev->traffic_class)
F
Francesco Fondelli 已提交
616 617
		seq_printf(seq, "     traffic_class: 0x%02x\n", pkt_dev->traffic_class);

R
Robert Olsson 已提交
618 619 620
	if (pkt_dev->node >= 0)
		seq_printf(seq, "     node: %d\n", pkt_dev->node);

621 622
	seq_printf(seq, "     Flags: ");

L
Luiz Capitulino 已提交
623 624
	if (pkt_dev->flags & F_IPV6)
		seq_printf(seq, "IPV6  ");
L
Linus Torvalds 已提交
625

L
Luiz Capitulino 已提交
626 627
	if (pkt_dev->flags & F_IPSRC_RND)
		seq_printf(seq, "IPSRC_RND  ");
L
Linus Torvalds 已提交
628

L
Luiz Capitulino 已提交
629 630
	if (pkt_dev->flags & F_IPDST_RND)
		seq_printf(seq, "IPDST_RND  ");
L
Linus Torvalds 已提交
631

L
Luiz Capitulino 已提交
632 633
	if (pkt_dev->flags & F_TXSIZE_RND)
		seq_printf(seq, "TXSIZE_RND  ");
L
Linus Torvalds 已提交
634

L
Luiz Capitulino 已提交
635 636
	if (pkt_dev->flags & F_UDPSRC_RND)
		seq_printf(seq, "UDPSRC_RND  ");
L
Linus Torvalds 已提交
637

L
Luiz Capitulino 已提交
638 639
	if (pkt_dev->flags & F_UDPDST_RND)
		seq_printf(seq, "UDPDST_RND  ");
L
Linus Torvalds 已提交
640

641 642 643
	if (pkt_dev->flags & F_MPLS_RND)
		seq_printf(seq,  "MPLS_RND  ");

R
Robert Olsson 已提交
644 645 646
	if (pkt_dev->flags & F_QUEUE_MAP_RND)
		seq_printf(seq,  "QUEUE_MAP_RND  ");

R
Robert Olsson 已提交
647 648 649
	if (pkt_dev->flags & F_QUEUE_MAP_CPU)
		seq_printf(seq,  "QUEUE_MAP_CPU  ");

650 651 652 653 654 655 656
	if (pkt_dev->cflows) {
		if (pkt_dev->flags & F_FLOW_SEQ)
			seq_printf(seq,  "FLOW_SEQ  "); /*in sequence flows*/
		else
			seq_printf(seq,  "FLOW_RND  ");
	}

J
Jamal Hadi Salim 已提交
657 658 659 660 661
#ifdef CONFIG_XFRM
	if (pkt_dev->flags & F_IPSEC_ON)
		seq_printf(seq,  "IPSEC  ");
#endif

L
Luiz Capitulino 已提交
662 663
	if (pkt_dev->flags & F_MACSRC_RND)
		seq_printf(seq, "MACSRC_RND  ");
L
Linus Torvalds 已提交
664

L
Luiz Capitulino 已提交
665 666
	if (pkt_dev->flags & F_MACDST_RND)
		seq_printf(seq, "MACDST_RND  ");
L
Linus Torvalds 已提交
667

F
Francesco Fondelli 已提交
668 669 670 671 672 673
	if (pkt_dev->flags & F_VID_RND)
		seq_printf(seq, "VID_RND  ");

	if (pkt_dev->flags & F_SVID_RND)
		seq_printf(seq, "SVID_RND  ");

R
Robert Olsson 已提交
674 675 676
	if (pkt_dev->flags & F_NODE)
		seq_printf(seq, "NODE_ALLOC  ");

L
Luiz Capitulino 已提交
677 678
	seq_puts(seq, "\n");

679 680 681 682
	/* not really stopped, more like last-running-at */
	stopped = pkt_dev->running ? ktime_now() : pkt_dev->stopped_at;
	idle = pkt_dev->idle_acc;
	do_div(idle, NSEC_PER_USEC);
L
Linus Torvalds 已提交
683

L
Luiz Capitulino 已提交
684
	seq_printf(seq,
685
		   "Current:\n     pkts-sofar: %llu  errors: %llu\n",
L
Luiz Capitulino 已提交
686
		   (unsigned long long)pkt_dev->sofar,
687 688 689 690 691 692 693
		   (unsigned long long)pkt_dev->errors);

	seq_printf(seq,
		   "     started: %lluus  stopped: %lluus idle: %lluus\n",
		   (unsigned long long) ktime_to_us(pkt_dev->started_at),
		   (unsigned long long) ktime_to_us(stopped),
		   (unsigned long long) idle);
L
Linus Torvalds 已提交
694

L
Luiz Capitulino 已提交
695 696
	seq_printf(seq,
		   "     seq_num: %d  cur_dst_mac_offset: %d  cur_src_mac_offset: %d\n",
697 698
		   pkt_dev->seq_num, pkt_dev->cur_dst_mac_offset,
		   pkt_dev->cur_src_mac_offset);
L
Linus Torvalds 已提交
699

L
Luiz Capitulino 已提交
700
	if (pkt_dev->flags & F_IPV6) {
701 702 703
		seq_printf(seq, "     cur_saddr: %pI6c  cur_daddr: %pI6c\n",
				&pkt_dev->cur_in6_saddr,
				&pkt_dev->cur_in6_daddr);
L
Luiz Capitulino 已提交
704 705
	} else
		seq_printf(seq, "     cur_saddr: 0x%x  cur_daddr: 0x%x\n",
706
			   pkt_dev->cur_saddr, pkt_dev->cur_daddr);
L
Linus Torvalds 已提交
707

L
Luiz Capitulino 已提交
708
	seq_printf(seq, "     cur_udp_dst: %d  cur_udp_src: %d\n",
709
		   pkt_dev->cur_udp_dst, pkt_dev->cur_udp_src);
L
Linus Torvalds 已提交
710

R
Robert Olsson 已提交
711 712
	seq_printf(seq, "     cur_queue_map: %u\n", pkt_dev->cur_queue_map);

L
Luiz Capitulino 已提交
713
	seq_printf(seq, "     flows: %u\n", pkt_dev->nflows);
L
Linus Torvalds 已提交
714 715

	if (pkt_dev->result[0])
L
Luiz Capitulino 已提交
716
		seq_printf(seq, "Result: %s\n", pkt_dev->result);
L
Linus Torvalds 已提交
717
	else
L
Luiz Capitulino 已提交
718
		seq_printf(seq, "Result: Idle\n");
L
Linus Torvalds 已提交
719

720
	return 0;
L
Linus Torvalds 已提交
721 722
}

723

724 725
static int hex32_arg(const char __user *user_buffer, unsigned long maxlen,
		     __u32 *num)
726 727 728 729
{
	int i = 0;
	*num = 0;

730
	for (; i < maxlen; i++) {
731
		int value;
732 733 734 735
		char c;
		*num <<= 4;
		if (get_user(c, &user_buffer[i]))
			return -EFAULT;
736 737 738
		value = hex_to_bin(c);
		if (value >= 0)
			*num |= value;
739 740 741 742 743 744
		else
			break;
	}
	return i;
}

L
Luiz Capitulino 已提交
745 746
static int count_trail_chars(const char __user * user_buffer,
			     unsigned int maxlen)
L
Linus Torvalds 已提交
747 748 749 750
{
	int i;

	for (i = 0; i < maxlen; i++) {
L
Luiz Capitulino 已提交
751 752 753 754
		char c;
		if (get_user(c, &user_buffer[i]))
			return -EFAULT;
		switch (c) {
L
Linus Torvalds 已提交
755 756 757 758 759 760 761 762 763
		case '\"':
		case '\n':
		case '\r':
		case '\t':
		case ' ':
		case '=':
			break;
		default:
			goto done;
764
		}
L
Linus Torvalds 已提交
765 766 767 768 769
	}
done:
	return i;
}

770 771
static long num_arg(const char __user *user_buffer, unsigned long maxlen,
				unsigned long *num)
L
Linus Torvalds 已提交
772
{
773
	int i;
L
Linus Torvalds 已提交
774
	*num = 0;
L
Luiz Capitulino 已提交
775

776
	for (i = 0; i < maxlen; i++) {
L
Luiz Capitulino 已提交
777 778 779 780
		char c;
		if (get_user(c, &user_buffer[i]))
			return -EFAULT;
		if ((c >= '0') && (c <= '9')) {
L
Linus Torvalds 已提交
781
			*num *= 10;
L
Luiz Capitulino 已提交
782
			*num += c - '0';
L
Linus Torvalds 已提交
783 784 785 786 787 788
		} else
			break;
	}
	return i;
}

L
Luiz Capitulino 已提交
789
static int strn_len(const char __user * user_buffer, unsigned int maxlen)
L
Linus Torvalds 已提交
790
{
791
	int i;
L
Linus Torvalds 已提交
792

793
	for (i = 0; i < maxlen; i++) {
L
Luiz Capitulino 已提交
794 795 796 797
		char c;
		if (get_user(c, &user_buffer[i]))
			return -EFAULT;
		switch (c) {
L
Linus Torvalds 已提交
798 799 800 801 802 803 804 805 806
		case '\"':
		case '\n':
		case '\r':
		case '\t':
		case ' ':
			goto done_str;
			break;
		default:
			break;
807
		}
L
Linus Torvalds 已提交
808 809 810 811 812
	}
done_str:
	return i;
}

813 814
static ssize_t get_labels(const char __user *buffer, struct pktgen_dev *pkt_dev)
{
815
	unsigned int n = 0;
816 817 818 819 820 821 822
	char c;
	ssize_t i = 0;
	int len;

	pkt_dev->nr_labels = 0;
	do {
		__u32 tmp;
F
Francesco Fondelli 已提交
823
		len = hex32_arg(&buffer[i], 8, &tmp);
824 825 826 827 828 829 830 831 832 833 834 835
		if (len <= 0)
			return len;
		pkt_dev->labels[n] = htonl(tmp);
		if (pkt_dev->labels[n] & MPLS_STACK_BOTTOM)
			pkt_dev->flags |= F_MPLS_RND;
		i += len;
		if (get_user(c, &buffer[i]))
			return -EFAULT;
		i++;
		n++;
		if (n >= MAX_MPLS_LABELS)
			return -E2BIG;
836
	} while (c == ',');
837 838 839 840 841

	pkt_dev->nr_labels = n;
	return i;
}

L
Luiz Capitulino 已提交
842 843 844
static ssize_t pktgen_if_write(struct file *file,
			       const char __user * user_buffer, size_t count,
			       loff_t * offset)
L
Linus Torvalds 已提交
845
{
846
	struct seq_file *seq = file->private_data;
L
Luiz Capitulino 已提交
847
	struct pktgen_dev *pkt_dev = seq->private;
848
	int i, max, len;
L
Linus Torvalds 已提交
849 850
	char name[16], valstr[32];
	unsigned long value = 0;
L
Luiz Capitulino 已提交
851 852
	char *pg_result = NULL;
	int tmp = 0;
L
Linus Torvalds 已提交
853
	char buf[128];
L
Luiz Capitulino 已提交
854 855 856

	pg_result = &(pkt_dev->result[0]);

L
Linus Torvalds 已提交
857
	if (count < 1) {
J
Joe Perches 已提交
858
		pr_warning("wrong command format\n");
L
Linus Torvalds 已提交
859 860
		return -EINVAL;
	}
L
Luiz Capitulino 已提交
861

862 863
	max = count;
	tmp = count_trail_chars(user_buffer, max);
L
Luiz Capitulino 已提交
864
	if (tmp < 0) {
J
Joe Perches 已提交
865
		pr_warning("illegal format\n");
L
Luiz Capitulino 已提交
866
		return tmp;
L
Linus Torvalds 已提交
867
	}
868
	i = tmp;
L
Luiz Capitulino 已提交
869

L
Linus Torvalds 已提交
870 871 872
	/* Read variable name */

	len = strn_len(&user_buffer[i], sizeof(name) - 1);
873
	if (len < 0)
L
Luiz Capitulino 已提交
874
		return len;
875

L
Linus Torvalds 已提交
876
	memset(name, 0, sizeof(name));
L
Luiz Capitulino 已提交
877
	if (copy_from_user(name, &user_buffer[i], len))
L
Linus Torvalds 已提交
878 879
		return -EFAULT;
	i += len;
L
Luiz Capitulino 已提交
880 881

	max = count - i;
L
Linus Torvalds 已提交
882
	len = count_trail_chars(&user_buffer[i], max);
L
Luiz Capitulino 已提交
883 884 885
	if (len < 0)
		return len;

L
Linus Torvalds 已提交
886 887 888
	i += len;

	if (debug) {
889
		size_t copy = min_t(size_t, count, 1023);
890 891
		char tb[copy + 1];
		if (copy_from_user(tb, user_buffer, copy))
L
Linus Torvalds 已提交
892
			return -EFAULT;
893
		tb[copy] = 0;
J
Joe Perches 已提交
894 895
		pr_debug("%s,%lu  buffer -:%s:-\n",
			 name, (unsigned long)count, tb);
L
Luiz Capitulino 已提交
896
	}
L
Linus Torvalds 已提交
897 898 899

	if (!strcmp(name, "min_pkt_size")) {
		len = num_arg(&user_buffer[i], 10, &value);
900
		if (len < 0)
L
Luiz Capitulino 已提交
901
			return len;
902

L
Linus Torvalds 已提交
903
		i += len;
L
Luiz Capitulino 已提交
904 905 906 907 908 909 910 911
		if (value < 14 + 20 + 8)
			value = 14 + 20 + 8;
		if (value != pkt_dev->min_pkt_size) {
			pkt_dev->min_pkt_size = value;
			pkt_dev->cur_pkt_size = value;
		}
		sprintf(pg_result, "OK: min_pkt_size=%u",
			pkt_dev->min_pkt_size);
L
Linus Torvalds 已提交
912 913 914
		return count;
	}

L
Luiz Capitulino 已提交
915
	if (!strcmp(name, "max_pkt_size")) {
L
Linus Torvalds 已提交
916
		len = num_arg(&user_buffer[i], 10, &value);
917
		if (len < 0)
L
Luiz Capitulino 已提交
918
			return len;
919

L
Linus Torvalds 已提交
920
		i += len;
L
Luiz Capitulino 已提交
921 922 923 924 925 926 927 928
		if (value < 14 + 20 + 8)
			value = 14 + 20 + 8;
		if (value != pkt_dev->max_pkt_size) {
			pkt_dev->max_pkt_size = value;
			pkt_dev->cur_pkt_size = value;
		}
		sprintf(pg_result, "OK: max_pkt_size=%u",
			pkt_dev->max_pkt_size);
L
Linus Torvalds 已提交
929 930 931
		return count;
	}

L
Luiz Capitulino 已提交
932
	/* Shortcut for min = max */
L
Linus Torvalds 已提交
933 934 935

	if (!strcmp(name, "pkt_size")) {
		len = num_arg(&user_buffer[i], 10, &value);
936
		if (len < 0)
L
Luiz Capitulino 已提交
937
			return len;
938

L
Linus Torvalds 已提交
939
		i += len;
L
Luiz Capitulino 已提交
940 941 942 943 944 945 946
		if (value < 14 + 20 + 8)
			value = 14 + 20 + 8;
		if (value != pkt_dev->min_pkt_size) {
			pkt_dev->min_pkt_size = value;
			pkt_dev->max_pkt_size = value;
			pkt_dev->cur_pkt_size = value;
		}
L
Linus Torvalds 已提交
947 948 949 950
		sprintf(pg_result, "OK: pkt_size=%u", pkt_dev->min_pkt_size);
		return count;
	}

L
Luiz Capitulino 已提交
951
	if (!strcmp(name, "debug")) {
L
Linus Torvalds 已提交
952
		len = num_arg(&user_buffer[i], 10, &value);
953
		if (len < 0)
L
Luiz Capitulino 已提交
954
			return len;
955

L
Linus Torvalds 已提交
956
		i += len;
L
Luiz Capitulino 已提交
957
		debug = value;
L
Linus Torvalds 已提交
958 959 960 961
		sprintf(pg_result, "OK: debug=%u", debug);
		return count;
	}

L
Luiz Capitulino 已提交
962
	if (!strcmp(name, "frags")) {
L
Linus Torvalds 已提交
963
		len = num_arg(&user_buffer[i], 10, &value);
964
		if (len < 0)
L
Luiz Capitulino 已提交
965
			return len;
966

L
Linus Torvalds 已提交
967 968 969 970 971 972 973
		i += len;
		pkt_dev->nfrags = value;
		sprintf(pg_result, "OK: frags=%u", pkt_dev->nfrags);
		return count;
	}
	if (!strcmp(name, "delay")) {
		len = num_arg(&user_buffer[i], 10, &value);
974
		if (len < 0)
L
Luiz Capitulino 已提交
975
			return len;
976

L
Linus Torvalds 已提交
977
		i += len;
978 979 980
		if (value == 0x7FFFFFFF)
			pkt_dev->delay = ULLONG_MAX;
		else
E
Eric Dumazet 已提交
981
			pkt_dev->delay = (u64)value;
982 983 984

		sprintf(pg_result, "OK: delay=%llu",
			(unsigned long long) pkt_dev->delay);
L
Linus Torvalds 已提交
985 986
		return count;
	}
987 988 989 990 991 992 993 994 995 996
	if (!strcmp(name, "rate")) {
		len = num_arg(&user_buffer[i], 10, &value);
		if (len < 0)
			return len;

		i += len;
		if (!value)
			return len;
		pkt_dev->delay = pkt_dev->min_pkt_size*8*NSEC_PER_USEC/value;
		if (debug)
J
Joe Perches 已提交
997
			pr_info("Delay set at: %llu ns\n", pkt_dev->delay);
998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011

		sprintf(pg_result, "OK: rate=%lu", value);
		return count;
	}
	if (!strcmp(name, "ratep")) {
		len = num_arg(&user_buffer[i], 10, &value);
		if (len < 0)
			return len;

		i += len;
		if (!value)
			return len;
		pkt_dev->delay = NSEC_PER_SEC/value;
		if (debug)
J
Joe Perches 已提交
1012
			pr_info("Delay set at: %llu ns\n", pkt_dev->delay);
1013 1014 1015 1016

		sprintf(pg_result, "OK: rate=%lu", value);
		return count;
	}
L
Luiz Capitulino 已提交
1017
	if (!strcmp(name, "udp_src_min")) {
L
Linus Torvalds 已提交
1018
		len = num_arg(&user_buffer[i], 10, &value);
1019
		if (len < 0)
L
Luiz Capitulino 已提交
1020
			return len;
1021

L
Linus Torvalds 已提交
1022
		i += len;
L
Luiz Capitulino 已提交
1023 1024 1025 1026
		if (value != pkt_dev->udp_src_min) {
			pkt_dev->udp_src_min = value;
			pkt_dev->cur_udp_src = value;
		}
L
Linus Torvalds 已提交
1027 1028 1029
		sprintf(pg_result, "OK: udp_src_min=%u", pkt_dev->udp_src_min);
		return count;
	}
L
Luiz Capitulino 已提交
1030
	if (!strcmp(name, "udp_dst_min")) {
L
Linus Torvalds 已提交
1031
		len = num_arg(&user_buffer[i], 10, &value);
1032
		if (len < 0)
L
Luiz Capitulino 已提交
1033
			return len;
1034

L
Linus Torvalds 已提交
1035
		i += len;
L
Luiz Capitulino 已提交
1036 1037 1038 1039
		if (value != pkt_dev->udp_dst_min) {
			pkt_dev->udp_dst_min = value;
			pkt_dev->cur_udp_dst = value;
		}
L
Linus Torvalds 已提交
1040 1041 1042
		sprintf(pg_result, "OK: udp_dst_min=%u", pkt_dev->udp_dst_min);
		return count;
	}
L
Luiz Capitulino 已提交
1043
	if (!strcmp(name, "udp_src_max")) {
L
Linus Torvalds 已提交
1044
		len = num_arg(&user_buffer[i], 10, &value);
1045
		if (len < 0)
L
Luiz Capitulino 已提交
1046
			return len;
1047

L
Linus Torvalds 已提交
1048
		i += len;
L
Luiz Capitulino 已提交
1049 1050 1051 1052
		if (value != pkt_dev->udp_src_max) {
			pkt_dev->udp_src_max = value;
			pkt_dev->cur_udp_src = value;
		}
L
Linus Torvalds 已提交
1053 1054 1055
		sprintf(pg_result, "OK: udp_src_max=%u", pkt_dev->udp_src_max);
		return count;
	}
L
Luiz Capitulino 已提交
1056
	if (!strcmp(name, "udp_dst_max")) {
L
Linus Torvalds 已提交
1057
		len = num_arg(&user_buffer[i], 10, &value);
1058
		if (len < 0)
L
Luiz Capitulino 已提交
1059
			return len;
1060

L
Linus Torvalds 已提交
1061
		i += len;
L
Luiz Capitulino 已提交
1062 1063 1064 1065
		if (value != pkt_dev->udp_dst_max) {
			pkt_dev->udp_dst_max = value;
			pkt_dev->cur_udp_dst = value;
		}
L
Linus Torvalds 已提交
1066 1067 1068 1069 1070
		sprintf(pg_result, "OK: udp_dst_max=%u", pkt_dev->udp_dst_max);
		return count;
	}
	if (!strcmp(name, "clone_skb")) {
		len = num_arg(&user_buffer[i], 10, &value);
1071
		if (len < 0)
L
Luiz Capitulino 已提交
1072
			return len;
1073 1074 1075
		if ((value > 0) &&
		    (!(pkt_dev->odev->priv_flags & IFF_TX_SKB_SHARING)))
			return -ENOTSUPP;
L
Linus Torvalds 已提交
1076
		i += len;
L
Luiz Capitulino 已提交
1077 1078
		pkt_dev->clone_skb = value;

L
Linus Torvalds 已提交
1079 1080 1081 1082 1083
		sprintf(pg_result, "OK: clone_skb=%d", pkt_dev->clone_skb);
		return count;
	}
	if (!strcmp(name, "count")) {
		len = num_arg(&user_buffer[i], 10, &value);
1084
		if (len < 0)
L
Luiz Capitulino 已提交
1085
			return len;
1086

L
Linus Torvalds 已提交
1087 1088 1089
		i += len;
		pkt_dev->count = value;
		sprintf(pg_result, "OK: count=%llu",
L
Luiz Capitulino 已提交
1090
			(unsigned long long)pkt_dev->count);
L
Linus Torvalds 已提交
1091 1092 1093 1094
		return count;
	}
	if (!strcmp(name, "src_mac_count")) {
		len = num_arg(&user_buffer[i], 10, &value);
1095
		if (len < 0)
L
Luiz Capitulino 已提交
1096
			return len;
1097

L
Linus Torvalds 已提交
1098 1099
		i += len;
		if (pkt_dev->src_mac_count != value) {
L
Luiz Capitulino 已提交
1100 1101 1102 1103 1104
			pkt_dev->src_mac_count = value;
			pkt_dev->cur_src_mac_offset = 0;
		}
		sprintf(pg_result, "OK: src_mac_count=%d",
			pkt_dev->src_mac_count);
L
Linus Torvalds 已提交
1105 1106 1107 1108
		return count;
	}
	if (!strcmp(name, "dst_mac_count")) {
		len = num_arg(&user_buffer[i], 10, &value);
1109
		if (len < 0)
L
Luiz Capitulino 已提交
1110
			return len;
1111

L
Linus Torvalds 已提交
1112 1113
		i += len;
		if (pkt_dev->dst_mac_count != value) {
L
Luiz Capitulino 已提交
1114 1115 1116 1117 1118
			pkt_dev->dst_mac_count = value;
			pkt_dev->cur_dst_mac_offset = 0;
		}
		sprintf(pg_result, "OK: dst_mac_count=%d",
			pkt_dev->dst_mac_count);
L
Linus Torvalds 已提交
1119 1120
		return count;
	}
R
Robert Olsson 已提交
1121 1122 1123 1124 1125 1126 1127 1128 1129 1130
	if (!strcmp(name, "node")) {
		len = num_arg(&user_buffer[i], 10, &value);
		if (len < 0)
			return len;

		i += len;

		if (node_possible(value)) {
			pkt_dev->node = value;
			sprintf(pg_result, "OK: node=%d", pkt_dev->node);
E
Eric Dumazet 已提交
1131 1132 1133 1134
			if (pkt_dev->page) {
				put_page(pkt_dev->page);
				pkt_dev->page = NULL;
			}
R
Robert Olsson 已提交
1135 1136 1137 1138 1139
		}
		else
			sprintf(pg_result, "ERROR: node not possible");
		return count;
	}
L
Linus Torvalds 已提交
1140
	if (!strcmp(name, "flag")) {
L
Luiz Capitulino 已提交
1141 1142
		char f[32];
		memset(f, 0, 32);
L
Linus Torvalds 已提交
1143
		len = strn_len(&user_buffer[i], sizeof(f) - 1);
1144
		if (len < 0)
L
Luiz Capitulino 已提交
1145
			return len;
1146

L
Linus Torvalds 已提交
1147 1148 1149
		if (copy_from_user(f, &user_buffer[i], len))
			return -EFAULT;
		i += len;
L
Luiz Capitulino 已提交
1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191
		if (strcmp(f, "IPSRC_RND") == 0)
			pkt_dev->flags |= F_IPSRC_RND;

		else if (strcmp(f, "!IPSRC_RND") == 0)
			pkt_dev->flags &= ~F_IPSRC_RND;

		else if (strcmp(f, "TXSIZE_RND") == 0)
			pkt_dev->flags |= F_TXSIZE_RND;

		else if (strcmp(f, "!TXSIZE_RND") == 0)
			pkt_dev->flags &= ~F_TXSIZE_RND;

		else if (strcmp(f, "IPDST_RND") == 0)
			pkt_dev->flags |= F_IPDST_RND;

		else if (strcmp(f, "!IPDST_RND") == 0)
			pkt_dev->flags &= ~F_IPDST_RND;

		else if (strcmp(f, "UDPSRC_RND") == 0)
			pkt_dev->flags |= F_UDPSRC_RND;

		else if (strcmp(f, "!UDPSRC_RND") == 0)
			pkt_dev->flags &= ~F_UDPSRC_RND;

		else if (strcmp(f, "UDPDST_RND") == 0)
			pkt_dev->flags |= F_UDPDST_RND;

		else if (strcmp(f, "!UDPDST_RND") == 0)
			pkt_dev->flags &= ~F_UDPDST_RND;

		else if (strcmp(f, "MACSRC_RND") == 0)
			pkt_dev->flags |= F_MACSRC_RND;

		else if (strcmp(f, "!MACSRC_RND") == 0)
			pkt_dev->flags &= ~F_MACSRC_RND;

		else if (strcmp(f, "MACDST_RND") == 0)
			pkt_dev->flags |= F_MACDST_RND;

		else if (strcmp(f, "!MACDST_RND") == 0)
			pkt_dev->flags &= ~F_MACDST_RND;

1192 1193 1194 1195 1196 1197
		else if (strcmp(f, "MPLS_RND") == 0)
			pkt_dev->flags |= F_MPLS_RND;

		else if (strcmp(f, "!MPLS_RND") == 0)
			pkt_dev->flags &= ~F_MPLS_RND;

F
Francesco Fondelli 已提交
1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209
		else if (strcmp(f, "VID_RND") == 0)
			pkt_dev->flags |= F_VID_RND;

		else if (strcmp(f, "!VID_RND") == 0)
			pkt_dev->flags &= ~F_VID_RND;

		else if (strcmp(f, "SVID_RND") == 0)
			pkt_dev->flags |= F_SVID_RND;

		else if (strcmp(f, "!SVID_RND") == 0)
			pkt_dev->flags &= ~F_SVID_RND;

1210 1211 1212
		else if (strcmp(f, "FLOW_SEQ") == 0)
			pkt_dev->flags |= F_FLOW_SEQ;

R
Robert Olsson 已提交
1213 1214 1215 1216 1217
		else if (strcmp(f, "QUEUE_MAP_RND") == 0)
			pkt_dev->flags |= F_QUEUE_MAP_RND;

		else if (strcmp(f, "!QUEUE_MAP_RND") == 0)
			pkt_dev->flags &= ~F_QUEUE_MAP_RND;
R
Robert Olsson 已提交
1218 1219 1220 1221 1222 1223

		else if (strcmp(f, "QUEUE_MAP_CPU") == 0)
			pkt_dev->flags |= F_QUEUE_MAP_CPU;

		else if (strcmp(f, "!QUEUE_MAP_CPU") == 0)
			pkt_dev->flags &= ~F_QUEUE_MAP_CPU;
J
Jamal Hadi Salim 已提交
1224 1225 1226 1227 1228
#ifdef CONFIG_XFRM
		else if (strcmp(f, "IPSEC") == 0)
			pkt_dev->flags |= F_IPSEC_ON;
#endif

F
Francesco Fondelli 已提交
1229 1230 1231
		else if (strcmp(f, "!IPV6") == 0)
			pkt_dev->flags &= ~F_IPV6;

R
Robert Olsson 已提交
1232 1233 1234 1235 1236 1237
		else if (strcmp(f, "NODE_ALLOC") == 0)
			pkt_dev->flags |= F_NODE;

		else if (strcmp(f, "!NODE_ALLOC") == 0)
			pkt_dev->flags &= ~F_NODE;

L
Luiz Capitulino 已提交
1238 1239 1240 1241
		else {
			sprintf(pg_result,
				"Flag -:%s:- unknown\nAvailable flags, (prepend ! to un-set flag):\n%s",
				f,
F
Francesco Fondelli 已提交
1242
				"IPSRC_RND, IPDST_RND, UDPSRC_RND, UDPDST_RND, "
R
Robert Olsson 已提交
1243
				"MACSRC_RND, MACDST_RND, TXSIZE_RND, IPV6, MPLS_RND, VID_RND, SVID_RND, FLOW_SEQ, IPSEC, NODE_ALLOC\n");
L
Luiz Capitulino 已提交
1244 1245
			return count;
		}
L
Linus Torvalds 已提交
1246 1247 1248 1249 1250
		sprintf(pg_result, "OK: flags=0x%x", pkt_dev->flags);
		return count;
	}
	if (!strcmp(name, "dst_min") || !strcmp(name, "dst")) {
		len = strn_len(&user_buffer[i], sizeof(pkt_dev->dst_min) - 1);
1251
		if (len < 0)
L
Luiz Capitulino 已提交
1252
			return len;
L
Linus Torvalds 已提交
1253

L
Luiz Capitulino 已提交
1254
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1255
			return -EFAULT;
L
Luiz Capitulino 已提交
1256 1257 1258 1259 1260 1261 1262 1263
		buf[len] = 0;
		if (strcmp(buf, pkt_dev->dst_min) != 0) {
			memset(pkt_dev->dst_min, 0, sizeof(pkt_dev->dst_min));
			strncpy(pkt_dev->dst_min, buf, len);
			pkt_dev->daddr_min = in_aton(pkt_dev->dst_min);
			pkt_dev->cur_daddr = pkt_dev->daddr_min;
		}
		if (debug)
J
Joe Perches 已提交
1264
			pr_debug("dst_min set to: %s\n", pkt_dev->dst_min);
L
Luiz Capitulino 已提交
1265
		i += len;
L
Linus Torvalds 已提交
1266 1267 1268 1269 1270
		sprintf(pg_result, "OK: dst_min=%s", pkt_dev->dst_min);
		return count;
	}
	if (!strcmp(name, "dst_max")) {
		len = strn_len(&user_buffer[i], sizeof(pkt_dev->dst_max) - 1);
1271
		if (len < 0)
L
Luiz Capitulino 已提交
1272
			return len;
1273

L
Linus Torvalds 已提交
1274

L
Luiz Capitulino 已提交
1275
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1276 1277
			return -EFAULT;

L
Luiz Capitulino 已提交
1278 1279 1280 1281 1282 1283 1284 1285
		buf[len] = 0;
		if (strcmp(buf, pkt_dev->dst_max) != 0) {
			memset(pkt_dev->dst_max, 0, sizeof(pkt_dev->dst_max));
			strncpy(pkt_dev->dst_max, buf, len);
			pkt_dev->daddr_max = in_aton(pkt_dev->dst_max);
			pkt_dev->cur_daddr = pkt_dev->daddr_max;
		}
		if (debug)
J
Joe Perches 已提交
1286
			pr_debug("dst_max set to: %s\n", pkt_dev->dst_max);
L
Linus Torvalds 已提交
1287 1288 1289 1290 1291 1292
		i += len;
		sprintf(pg_result, "OK: dst_max=%s", pkt_dev->dst_max);
		return count;
	}
	if (!strcmp(name, "dst6")) {
		len = strn_len(&user_buffer[i], sizeof(buf) - 1);
L
Luiz Capitulino 已提交
1293 1294
		if (len < 0)
			return len;
L
Linus Torvalds 已提交
1295 1296 1297

		pkt_dev->flags |= F_IPV6;

L
Luiz Capitulino 已提交
1298
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1299
			return -EFAULT;
L
Luiz Capitulino 已提交
1300
		buf[len] = 0;
L
Linus Torvalds 已提交
1301 1302

		scan_ip6(buf, pkt_dev->in6_daddr.s6_addr);
1303
		snprintf(buf, sizeof(buf), "%pI6c", &pkt_dev->in6_daddr);
L
Linus Torvalds 已提交
1304

A
Alexey Dobriyan 已提交
1305
		pkt_dev->cur_in6_daddr = pkt_dev->in6_daddr;
L
Linus Torvalds 已提交
1306

L
Luiz Capitulino 已提交
1307
		if (debug)
J
Joe Perches 已提交
1308
			pr_debug("dst6 set to: %s\n", buf);
L
Linus Torvalds 已提交
1309

L
Luiz Capitulino 已提交
1310
		i += len;
L
Linus Torvalds 已提交
1311 1312 1313 1314 1315
		sprintf(pg_result, "OK: dst6=%s", buf);
		return count;
	}
	if (!strcmp(name, "dst6_min")) {
		len = strn_len(&user_buffer[i], sizeof(buf) - 1);
L
Luiz Capitulino 已提交
1316 1317
		if (len < 0)
			return len;
L
Linus Torvalds 已提交
1318 1319 1320

		pkt_dev->flags |= F_IPV6;

L
Luiz Capitulino 已提交
1321
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1322
			return -EFAULT;
L
Luiz Capitulino 已提交
1323
		buf[len] = 0;
L
Linus Torvalds 已提交
1324 1325

		scan_ip6(buf, pkt_dev->min_in6_daddr.s6_addr);
1326
		snprintf(buf, sizeof(buf), "%pI6c", &pkt_dev->min_in6_daddr);
L
Linus Torvalds 已提交
1327

A
Alexey Dobriyan 已提交
1328
		pkt_dev->cur_in6_daddr = pkt_dev->min_in6_daddr;
L
Luiz Capitulino 已提交
1329
		if (debug)
J
Joe Perches 已提交
1330
			pr_debug("dst6_min set to: %s\n", buf);
L
Linus Torvalds 已提交
1331

L
Luiz Capitulino 已提交
1332
		i += len;
L
Linus Torvalds 已提交
1333 1334 1335 1336 1337
		sprintf(pg_result, "OK: dst6_min=%s", buf);
		return count;
	}
	if (!strcmp(name, "dst6_max")) {
		len = strn_len(&user_buffer[i], sizeof(buf) - 1);
L
Luiz Capitulino 已提交
1338 1339
		if (len < 0)
			return len;
L
Linus Torvalds 已提交
1340 1341 1342

		pkt_dev->flags |= F_IPV6;

L
Luiz Capitulino 已提交
1343
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1344
			return -EFAULT;
L
Luiz Capitulino 已提交
1345
		buf[len] = 0;
L
Linus Torvalds 已提交
1346 1347

		scan_ip6(buf, pkt_dev->max_in6_daddr.s6_addr);
1348
		snprintf(buf, sizeof(buf), "%pI6c", &pkt_dev->max_in6_daddr);
L
Linus Torvalds 已提交
1349

L
Luiz Capitulino 已提交
1350
		if (debug)
J
Joe Perches 已提交
1351
			pr_debug("dst6_max set to: %s\n", buf);
L
Linus Torvalds 已提交
1352

L
Luiz Capitulino 已提交
1353
		i += len;
L
Linus Torvalds 已提交
1354 1355 1356 1357 1358
		sprintf(pg_result, "OK: dst6_max=%s", buf);
		return count;
	}
	if (!strcmp(name, "src6")) {
		len = strn_len(&user_buffer[i], sizeof(buf) - 1);
L
Luiz Capitulino 已提交
1359 1360
		if (len < 0)
			return len;
L
Linus Torvalds 已提交
1361 1362 1363

		pkt_dev->flags |= F_IPV6;

L
Luiz Capitulino 已提交
1364
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1365
			return -EFAULT;
L
Luiz Capitulino 已提交
1366
		buf[len] = 0;
L
Linus Torvalds 已提交
1367 1368

		scan_ip6(buf, pkt_dev->in6_saddr.s6_addr);
1369
		snprintf(buf, sizeof(buf), "%pI6c", &pkt_dev->in6_saddr);
L
Linus Torvalds 已提交
1370

A
Alexey Dobriyan 已提交
1371
		pkt_dev->cur_in6_saddr = pkt_dev->in6_saddr;
L
Linus Torvalds 已提交
1372

L
Luiz Capitulino 已提交
1373
		if (debug)
J
Joe Perches 已提交
1374
			pr_debug("src6 set to: %s\n", buf);
L
Luiz Capitulino 已提交
1375 1376

		i += len;
L
Linus Torvalds 已提交
1377 1378 1379 1380 1381
		sprintf(pg_result, "OK: src6=%s", buf);
		return count;
	}
	if (!strcmp(name, "src_min")) {
		len = strn_len(&user_buffer[i], sizeof(pkt_dev->src_min) - 1);
1382
		if (len < 0)
L
Luiz Capitulino 已提交
1383
			return len;
1384

L
Luiz Capitulino 已提交
1385
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1386
			return -EFAULT;
L
Luiz Capitulino 已提交
1387 1388 1389 1390 1391 1392 1393 1394
		buf[len] = 0;
		if (strcmp(buf, pkt_dev->src_min) != 0) {
			memset(pkt_dev->src_min, 0, sizeof(pkt_dev->src_min));
			strncpy(pkt_dev->src_min, buf, len);
			pkt_dev->saddr_min = in_aton(pkt_dev->src_min);
			pkt_dev->cur_saddr = pkt_dev->saddr_min;
		}
		if (debug)
J
Joe Perches 已提交
1395
			pr_debug("src_min set to: %s\n", pkt_dev->src_min);
L
Linus Torvalds 已提交
1396 1397 1398 1399 1400 1401
		i += len;
		sprintf(pg_result, "OK: src_min=%s", pkt_dev->src_min);
		return count;
	}
	if (!strcmp(name, "src_max")) {
		len = strn_len(&user_buffer[i], sizeof(pkt_dev->src_max) - 1);
1402
		if (len < 0)
L
Luiz Capitulino 已提交
1403
			return len;
1404

L
Luiz Capitulino 已提交
1405
		if (copy_from_user(buf, &user_buffer[i], len))
L
Linus Torvalds 已提交
1406
			return -EFAULT;
L
Luiz Capitulino 已提交
1407 1408 1409 1410 1411 1412 1413 1414
		buf[len] = 0;
		if (strcmp(buf, pkt_dev->src_max) != 0) {
			memset(pkt_dev->src_max, 0, sizeof(pkt_dev->src_max));
			strncpy(pkt_dev->src_max, buf, len);
			pkt_dev->saddr_max = in_aton(pkt_dev->src_max);
			pkt_dev->cur_saddr = pkt_dev->saddr_max;
		}
		if (debug)
J
Joe Perches 已提交
1415
			pr_debug("src_max set to: %s\n", pkt_dev->src_max);
L
Linus Torvalds 已提交
1416 1417 1418 1419 1420 1421
		i += len;
		sprintf(pg_result, "OK: src_max=%s", pkt_dev->src_max);
		return count;
	}
	if (!strcmp(name, "dst_mac")) {
		len = strn_len(&user_buffer[i], sizeof(valstr) - 1);
1422
		if (len < 0)
L
Luiz Capitulino 已提交
1423
			return len;
1424

L
Linus Torvalds 已提交
1425
		memset(valstr, 0, sizeof(valstr));
L
Luiz Capitulino 已提交
1426
		if (copy_from_user(valstr, &user_buffer[i], len))
L
Linus Torvalds 已提交
1427 1428
			return -EFAULT;

1429 1430
		if (!mac_pton(valstr, pkt_dev->dst_mac))
			return -EINVAL;
L
Linus Torvalds 已提交
1431
		/* Set up Dest MAC */
1432
		memcpy(&pkt_dev->hh[0], pkt_dev->dst_mac, ETH_ALEN);
L
Luiz Capitulino 已提交
1433

1434
		sprintf(pg_result, "OK: dstmac %pM", pkt_dev->dst_mac);
L
Linus Torvalds 已提交
1435 1436 1437 1438
		return count;
	}
	if (!strcmp(name, "src_mac")) {
		len = strn_len(&user_buffer[i], sizeof(valstr) - 1);
1439
		if (len < 0)
L
Luiz Capitulino 已提交
1440
			return len;
1441

L
Linus Torvalds 已提交
1442
		memset(valstr, 0, sizeof(valstr));
L
Luiz Capitulino 已提交
1443
		if (copy_from_user(valstr, &user_buffer[i], len))
L
Linus Torvalds 已提交
1444 1445
			return -EFAULT;

1446 1447
		if (!mac_pton(valstr, pkt_dev->src_mac))
			return -EINVAL;
A
Adit Ranadive 已提交
1448
		/* Set up Src MAC */
1449
		memcpy(&pkt_dev->hh[6], pkt_dev->src_mac, ETH_ALEN);
A
Adit Ranadive 已提交
1450

1451
		sprintf(pg_result, "OK: srcmac %pM", pkt_dev->src_mac);
L
Linus Torvalds 已提交
1452 1453 1454
		return count;
	}

L
Luiz Capitulino 已提交
1455 1456 1457 1458 1459
	if (!strcmp(name, "clear_counters")) {
		pktgen_clear_counters(pkt_dev);
		sprintf(pg_result, "OK: Clearing counters.\n");
		return count;
	}
L
Linus Torvalds 已提交
1460 1461 1462

	if (!strcmp(name, "flows")) {
		len = num_arg(&user_buffer[i], 10, &value);
1463
		if (len < 0)
L
Luiz Capitulino 已提交
1464
			return len;
1465

L
Linus Torvalds 已提交
1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476
		i += len;
		if (value > MAX_CFLOWS)
			value = MAX_CFLOWS;

		pkt_dev->cflows = value;
		sprintf(pg_result, "OK: flows=%u", pkt_dev->cflows);
		return count;
	}

	if (!strcmp(name, "flowlen")) {
		len = num_arg(&user_buffer[i], 10, &value);
1477
		if (len < 0)
L
Luiz Capitulino 已提交
1478
			return len;
1479

L
Linus Torvalds 已提交
1480 1481 1482 1483 1484
		i += len;
		pkt_dev->lflow = value;
		sprintf(pg_result, "OK: flowlen=%u", pkt_dev->lflow);
		return count;
	}
L
Luiz Capitulino 已提交
1485

R
Robert Olsson 已提交
1486 1487
	if (!strcmp(name, "queue_map_min")) {
		len = num_arg(&user_buffer[i], 5, &value);
1488
		if (len < 0)
R
Robert Olsson 已提交
1489
			return len;
1490

R
Robert Olsson 已提交
1491 1492 1493 1494 1495 1496 1497 1498
		i += len;
		pkt_dev->queue_map_min = value;
		sprintf(pg_result, "OK: queue_map_min=%u", pkt_dev->queue_map_min);
		return count;
	}

	if (!strcmp(name, "queue_map_max")) {
		len = num_arg(&user_buffer[i], 5, &value);
1499
		if (len < 0)
R
Robert Olsson 已提交
1500
			return len;
1501

R
Robert Olsson 已提交
1502 1503 1504 1505 1506 1507
		i += len;
		pkt_dev->queue_map_max = value;
		sprintf(pg_result, "OK: queue_map_max=%u", pkt_dev->queue_map_max);
		return count;
	}

1508
	if (!strcmp(name, "mpls")) {
1509
		unsigned int n, cnt;
S
Stephen Hemminger 已提交
1510

1511
		len = get_labels(&user_buffer[i], pkt_dev);
S
Stephen Hemminger 已提交
1512 1513
		if (len < 0)
			return len;
1514
		i += len;
S
Stephen Hemminger 已提交
1515
		cnt = sprintf(pg_result, "OK: mpls=");
1516
		for (n = 0; n < pkt_dev->nr_labels; n++)
S
Stephen Hemminger 已提交
1517 1518 1519
			cnt += sprintf(pg_result + cnt,
				       "%08x%s", ntohl(pkt_dev->labels[n]),
				       n == pkt_dev->nr_labels-1 ? "" : ",");
F
Francesco Fondelli 已提交
1520 1521 1522 1523 1524 1525

		if (pkt_dev->nr_labels && pkt_dev->vlan_id != 0xffff) {
			pkt_dev->vlan_id = 0xffff; /* turn off VLAN/SVLAN */
			pkt_dev->svlan_id = 0xffff;

			if (debug)
J
Joe Perches 已提交
1526
				pr_debug("VLAN/SVLAN auto turned off\n");
F
Francesco Fondelli 已提交
1527 1528 1529 1530 1531 1532
		}
		return count;
	}

	if (!strcmp(name, "vlan_id")) {
		len = num_arg(&user_buffer[i], 4, &value);
1533
		if (len < 0)
F
Francesco Fondelli 已提交
1534
			return len;
1535

F
Francesco Fondelli 已提交
1536 1537 1538 1539 1540
		i += len;
		if (value <= 4095) {
			pkt_dev->vlan_id = value;  /* turn on VLAN */

			if (debug)
J
Joe Perches 已提交
1541
				pr_debug("VLAN turned on\n");
F
Francesco Fondelli 已提交
1542 1543

			if (debug && pkt_dev->nr_labels)
J
Joe Perches 已提交
1544
				pr_debug("MPLS auto turned off\n");
F
Francesco Fondelli 已提交
1545 1546 1547 1548 1549 1550 1551 1552

			pkt_dev->nr_labels = 0;    /* turn off MPLS */
			sprintf(pg_result, "OK: vlan_id=%u", pkt_dev->vlan_id);
		} else {
			pkt_dev->vlan_id = 0xffff; /* turn off VLAN/SVLAN */
			pkt_dev->svlan_id = 0xffff;

			if (debug)
J
Joe Perches 已提交
1553
				pr_debug("VLAN/SVLAN turned off\n");
F
Francesco Fondelli 已提交
1554 1555 1556 1557 1558 1559
		}
		return count;
	}

	if (!strcmp(name, "vlan_p")) {
		len = num_arg(&user_buffer[i], 1, &value);
1560
		if (len < 0)
F
Francesco Fondelli 已提交
1561
			return len;
1562

F
Francesco Fondelli 已提交
1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574
		i += len;
		if ((value <= 7) && (pkt_dev->vlan_id != 0xffff)) {
			pkt_dev->vlan_p = value;
			sprintf(pg_result, "OK: vlan_p=%u", pkt_dev->vlan_p);
		} else {
			sprintf(pg_result, "ERROR: vlan_p must be 0-7");
		}
		return count;
	}

	if (!strcmp(name, "vlan_cfi")) {
		len = num_arg(&user_buffer[i], 1, &value);
1575
		if (len < 0)
F
Francesco Fondelli 已提交
1576
			return len;
1577

F
Francesco Fondelli 已提交
1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589
		i += len;
		if ((value <= 1) && (pkt_dev->vlan_id != 0xffff)) {
			pkt_dev->vlan_cfi = value;
			sprintf(pg_result, "OK: vlan_cfi=%u", pkt_dev->vlan_cfi);
		} else {
			sprintf(pg_result, "ERROR: vlan_cfi must be 0-1");
		}
		return count;
	}

	if (!strcmp(name, "svlan_id")) {
		len = num_arg(&user_buffer[i], 4, &value);
1590
		if (len < 0)
F
Francesco Fondelli 已提交
1591
			return len;
1592

F
Francesco Fondelli 已提交
1593 1594 1595 1596 1597
		i += len;
		if ((value <= 4095) && ((pkt_dev->vlan_id != 0xffff))) {
			pkt_dev->svlan_id = value;  /* turn on SVLAN */

			if (debug)
J
Joe Perches 已提交
1598
				pr_debug("SVLAN turned on\n");
F
Francesco Fondelli 已提交
1599 1600

			if (debug && pkt_dev->nr_labels)
J
Joe Perches 已提交
1601
				pr_debug("MPLS auto turned off\n");
F
Francesco Fondelli 已提交
1602 1603 1604 1605 1606 1607 1608 1609

			pkt_dev->nr_labels = 0;    /* turn off MPLS */
			sprintf(pg_result, "OK: svlan_id=%u", pkt_dev->svlan_id);
		} else {
			pkt_dev->vlan_id = 0xffff; /* turn off VLAN/SVLAN */
			pkt_dev->svlan_id = 0xffff;

			if (debug)
J
Joe Perches 已提交
1610
				pr_debug("VLAN/SVLAN turned off\n");
F
Francesco Fondelli 已提交
1611 1612 1613 1614 1615 1616
		}
		return count;
	}

	if (!strcmp(name, "svlan_p")) {
		len = num_arg(&user_buffer[i], 1, &value);
1617
		if (len < 0)
F
Francesco Fondelli 已提交
1618
			return len;
1619

F
Francesco Fondelli 已提交
1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631
		i += len;
		if ((value <= 7) && (pkt_dev->svlan_id != 0xffff)) {
			pkt_dev->svlan_p = value;
			sprintf(pg_result, "OK: svlan_p=%u", pkt_dev->svlan_p);
		} else {
			sprintf(pg_result, "ERROR: svlan_p must be 0-7");
		}
		return count;
	}

	if (!strcmp(name, "svlan_cfi")) {
		len = num_arg(&user_buffer[i], 1, &value);
1632
		if (len < 0)
F
Francesco Fondelli 已提交
1633
			return len;
1634

F
Francesco Fondelli 已提交
1635 1636 1637 1638 1639 1640 1641
		i += len;
		if ((value <= 1) && (pkt_dev->svlan_id != 0xffff)) {
			pkt_dev->svlan_cfi = value;
			sprintf(pg_result, "OK: svlan_cfi=%u", pkt_dev->svlan_cfi);
		} else {
			sprintf(pg_result, "ERROR: svlan_cfi must be 0-1");
		}
1642 1643 1644
		return count;
	}

F
Francesco Fondelli 已提交
1645 1646 1647
	if (!strcmp(name, "tos")) {
		__u32 tmp_value = 0;
		len = hex32_arg(&user_buffer[i], 2, &tmp_value);
1648
		if (len < 0)
F
Francesco Fondelli 已提交
1649
			return len;
1650

F
Francesco Fondelli 已提交
1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663
		i += len;
		if (len == 2) {
			pkt_dev->tos = tmp_value;
			sprintf(pg_result, "OK: tos=0x%02x", pkt_dev->tos);
		} else {
			sprintf(pg_result, "ERROR: tos must be 00-ff");
		}
		return count;
	}

	if (!strcmp(name, "traffic_class")) {
		__u32 tmp_value = 0;
		len = hex32_arg(&user_buffer[i], 2, &tmp_value);
1664
		if (len < 0)
F
Francesco Fondelli 已提交
1665
			return len;
1666

F
Francesco Fondelli 已提交
1667 1668 1669 1670 1671 1672 1673 1674 1675 1676
		i += len;
		if (len == 2) {
			pkt_dev->traffic_class = tmp_value;
			sprintf(pg_result, "OK: traffic_class=0x%02x", pkt_dev->traffic_class);
		} else {
			sprintf(pg_result, "ERROR: traffic_class must be 00-ff");
		}
		return count;
	}

J
John Fastabend 已提交
1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688
	if (!strcmp(name, "skb_priority")) {
		len = num_arg(&user_buffer[i], 9, &value);
		if (len < 0)
			return len;

		i += len;
		pkt_dev->skb_priority = value;
		sprintf(pg_result, "OK: skb_priority=%i",
			pkt_dev->skb_priority);
		return count;
	}

L
Linus Torvalds 已提交
1689 1690 1691 1692
	sprintf(pkt_dev->result, "No such parameter \"%s\"", name);
	return -EINVAL;
}

1693
static int pktgen_if_open(struct inode *inode, struct file *file)
L
Linus Torvalds 已提交
1694
{
1695 1696
	return single_open(file, pktgen_if_show, PDE(inode)->data);
}
L
Linus Torvalds 已提交
1697

1698
static const struct file_operations pktgen_if_fops = {
L
Luiz Capitulino 已提交
1699 1700 1701 1702 1703 1704
	.owner   = THIS_MODULE,
	.open    = pktgen_if_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
	.write   = pktgen_if_write,
	.release = single_release,
1705
};
L
Linus Torvalds 已提交
1706

1707 1708
static int pktgen_thread_show(struct seq_file *seq, void *v)
{
L
Luiz Capitulino 已提交
1709
	struct pktgen_thread *t = seq->private;
S
Stephen Hemminger 已提交
1710
	const struct pktgen_dev *pkt_dev;
1711 1712

	BUG_ON(!t);
L
Linus Torvalds 已提交
1713

L
Luiz Capitulino 已提交
1714
	seq_printf(seq, "Running: ");
L
Linus Torvalds 已提交
1715

L
Luiz Capitulino 已提交
1716
	if_lock(t);
1717
	list_for_each_entry(pkt_dev, &t->if_list, list)
L
Luiz Capitulino 已提交
1718
		if (pkt_dev->running)
E
Eric Dumazet 已提交
1719
			seq_printf(seq, "%s ", pkt_dev->odevname);
L
Linus Torvalds 已提交
1720

L
Luiz Capitulino 已提交
1721 1722
	seq_printf(seq, "\nStopped: ");

1723
	list_for_each_entry(pkt_dev, &t->if_list, list)
L
Luiz Capitulino 已提交
1724
		if (!pkt_dev->running)
E
Eric Dumazet 已提交
1725
			seq_printf(seq, "%s ", pkt_dev->odevname);
L
Linus Torvalds 已提交
1726 1727

	if (t->result[0])
1728
		seq_printf(seq, "\nResult: %s\n", t->result);
L
Linus Torvalds 已提交
1729
	else
1730
		seq_printf(seq, "\nResult: NA\n");
L
Linus Torvalds 已提交
1731

L
Luiz Capitulino 已提交
1732
	if_unlock(t);
L
Linus Torvalds 已提交
1733

1734
	return 0;
L
Linus Torvalds 已提交
1735 1736
}

1737
static ssize_t pktgen_thread_write(struct file *file,
L
Luiz Capitulino 已提交
1738 1739
				   const char __user * user_buffer,
				   size_t count, loff_t * offset)
L
Linus Torvalds 已提交
1740
{
1741
	struct seq_file *seq = file->private_data;
L
Luiz Capitulino 已提交
1742
	struct pktgen_thread *t = seq->private;
1743
	int i, max, len, ret;
L
Linus Torvalds 已提交
1744
	char name[40];
L
Luiz Capitulino 已提交
1745
	char *pg_result;
1746

L
Linus Torvalds 已提交
1747
	if (count < 1) {
L
Luiz Capitulino 已提交
1748
		//      sprintf(pg_result, "Wrong command format");
L
Linus Torvalds 已提交
1749 1750
		return -EINVAL;
	}
1751

1752 1753
	max = count;
	len = count_trail_chars(user_buffer, max);
L
Luiz Capitulino 已提交
1754
	if (len < 0)
1755 1756
		return len;

1757
	i = len;
1758

L
Linus Torvalds 已提交
1759 1760 1761
	/* Read variable name */

	len = strn_len(&user_buffer[i], sizeof(name) - 1);
L
Luiz Capitulino 已提交
1762
	if (len < 0)
1763
		return len;
L
Luiz Capitulino 已提交
1764

L
Linus Torvalds 已提交
1765 1766 1767 1768
	memset(name, 0, sizeof(name));
	if (copy_from_user(name, &user_buffer[i], len))
		return -EFAULT;
	i += len;
1769

L
Luiz Capitulino 已提交
1770
	max = count - i;
L
Linus Torvalds 已提交
1771
	len = count_trail_chars(&user_buffer[i], max);
L
Luiz Capitulino 已提交
1772
	if (len < 0)
1773 1774
		return len;

L
Linus Torvalds 已提交
1775 1776
	i += len;

1777
	if (debug)
J
Joe Perches 已提交
1778
		pr_debug("t=%s, count=%lu\n", name, (unsigned long)count);
L
Linus Torvalds 已提交
1779

L
Luiz Capitulino 已提交
1780
	if (!t) {
J
Joe Perches 已提交
1781
		pr_err("ERROR: No thread\n");
L
Linus Torvalds 已提交
1782 1783 1784 1785 1786 1787
		ret = -EINVAL;
		goto out;
	}

	pg_result = &(t->result[0]);

L
Luiz Capitulino 已提交
1788 1789 1790
	if (!strcmp(name, "add_device")) {
		char f[32];
		memset(f, 0, 32);
L
Linus Torvalds 已提交
1791
		len = strn_len(&user_buffer[i], sizeof(f) - 1);
L
Luiz Capitulino 已提交
1792 1793
		if (len < 0) {
			ret = len;
L
Linus Torvalds 已提交
1794 1795
			goto out;
		}
L
Luiz Capitulino 已提交
1796
		if (copy_from_user(f, &user_buffer[i], len))
L
Linus Torvalds 已提交
1797 1798
			return -EFAULT;
		i += len;
1799
		mutex_lock(&pktgen_thread_lock);
L
Luiz Capitulino 已提交
1800
		pktgen_add_device(t, f);
1801
		mutex_unlock(&pktgen_thread_lock);
L
Luiz Capitulino 已提交
1802 1803
		ret = count;
		sprintf(pg_result, "OK: add_device=%s", f);
L
Linus Torvalds 已提交
1804 1805 1806
		goto out;
	}

L
Luiz Capitulino 已提交
1807
	if (!strcmp(name, "rem_device_all")) {
1808
		mutex_lock(&pktgen_thread_lock);
1809
		t->control |= T_REMDEVALL;
1810
		mutex_unlock(&pktgen_thread_lock);
L
Luiz Capitulino 已提交
1811
		schedule_timeout_interruptible(msecs_to_jiffies(125));	/* Propagate thread->control  */
L
Linus Torvalds 已提交
1812
		ret = count;
L
Luiz Capitulino 已提交
1813
		sprintf(pg_result, "OK: rem_device_all");
L
Linus Torvalds 已提交
1814 1815 1816
		goto out;
	}

L
Luiz Capitulino 已提交
1817
	if (!strcmp(name, "max_before_softirq")) {
1818
		sprintf(pg_result, "OK: Note! max_before_softirq is obsoleted -- Do not use");
L
Luiz Capitulino 已提交
1819
		ret = count;
L
Linus Torvalds 已提交
1820 1821 1822 1823
		goto out;
	}

	ret = -EINVAL;
L
Luiz Capitulino 已提交
1824
out:
L
Linus Torvalds 已提交
1825 1826 1827
	return ret;
}

1828
static int pktgen_thread_open(struct inode *inode, struct file *file)
L
Linus Torvalds 已提交
1829
{
1830
	return single_open(file, pktgen_thread_show, PDE(inode)->data);
L
Linus Torvalds 已提交
1831 1832
}

1833
static const struct file_operations pktgen_thread_fops = {
L
Luiz Capitulino 已提交
1834 1835 1836 1837 1838 1839
	.owner   = THIS_MODULE,
	.open    = pktgen_thread_open,
	.read    = seq_read,
	.llseek  = seq_lseek,
	.write   = pktgen_thread_write,
	.release = single_release,
1840
};
L
Linus Torvalds 已提交
1841 1842

/* Think find or remove for NN */
L
Luiz Capitulino 已提交
1843
static struct pktgen_dev *__pktgen_NN_threads(const char *ifname, int remove)
L
Linus Torvalds 已提交
1844 1845 1846
{
	struct pktgen_thread *t;
	struct pktgen_dev *pkt_dev = NULL;
E
Eric Dumazet 已提交
1847
	bool exact = (remove == FIND);
L
Linus Torvalds 已提交
1848

1849
	list_for_each_entry(t, &pktgen_threads, th_list) {
E
Eric Dumazet 已提交
1850
		pkt_dev = pktgen_find_dev(t, ifname, exact);
L
Linus Torvalds 已提交
1851
		if (pkt_dev) {
L
Luiz Capitulino 已提交
1852 1853 1854 1855 1856 1857
			if (remove) {
				if_lock(t);
				pkt_dev->removal_mark = 1;
				t->control |= T_REMDEV;
				if_unlock(t);
			}
L
Linus Torvalds 已提交
1858 1859 1860
			break;
		}
	}
L
Luiz Capitulino 已提交
1861
	return pkt_dev;
L
Linus Torvalds 已提交
1862 1863
}

1864 1865 1866
/*
 * mark a device for removal
 */
1867
static void pktgen_mark_device(const char *ifname)
L
Linus Torvalds 已提交
1868 1869
{
	struct pktgen_dev *pkt_dev = NULL;
1870 1871 1872
	const int max_tries = 10, msec_per_try = 125;
	int i = 0;

1873
	mutex_lock(&pktgen_thread_lock);
J
Joe Perches 已提交
1874
	pr_debug("%s: marking %s for removal\n", __func__, ifname);
1875

L
Luiz Capitulino 已提交
1876
	while (1) {
1877 1878

		pkt_dev = __pktgen_NN_threads(ifname, REMOVE);
L
Luiz Capitulino 已提交
1879 1880
		if (pkt_dev == NULL)
			break;	/* success */
1881

1882
		mutex_unlock(&pktgen_thread_lock);
J
Joe Perches 已提交
1883 1884
		pr_debug("%s: waiting for %s to disappear....\n",
			 __func__, ifname);
1885
		schedule_timeout_interruptible(msecs_to_jiffies(msec_per_try));
1886
		mutex_lock(&pktgen_thread_lock);
1887 1888

		if (++i >= max_tries) {
J
Joe Perches 已提交
1889 1890
			pr_err("%s: timed out after waiting %d msec for device %s to be removed\n",
			       __func__, msec_per_try * i, ifname);
1891 1892 1893 1894 1895
			break;
		}

	}

1896
	mutex_unlock(&pktgen_thread_lock);
1897
}
1898

1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911
static void pktgen_change_name(struct net_device *dev)
{
	struct pktgen_thread *t;

	list_for_each_entry(t, &pktgen_threads, th_list) {
		struct pktgen_dev *pkt_dev;

		list_for_each_entry(pkt_dev, &t->if_list, list) {
			if (pkt_dev->odev != dev)
				continue;

			remove_proc_entry(pkt_dev->entry->name, pg_proc_dir);

A
Alexey Dobriyan 已提交
1912 1913 1914 1915
			pkt_dev->entry = proc_create_data(dev->name, 0600,
							  pg_proc_dir,
							  &pktgen_if_fops,
							  pkt_dev);
1916
			if (!pkt_dev->entry)
J
Joe Perches 已提交
1917 1918
				pr_err("can't move proc entry for '%s'\n",
				       dev->name);
1919 1920 1921
			break;
		}
	}
L
Linus Torvalds 已提交
1922 1923
}

L
Luiz Capitulino 已提交
1924 1925
static int pktgen_device_event(struct notifier_block *unused,
			       unsigned long event, void *ptr)
L
Linus Torvalds 已提交
1926
{
1927
	struct net_device *dev = ptr;
L
Linus Torvalds 已提交
1928

1929
	if (!net_eq(dev_net(dev), &init_net) || pktgen_exiting)
1930 1931
		return NOTIFY_DONE;

L
Linus Torvalds 已提交
1932 1933 1934 1935 1936
	/* It is OK that we do not hold the group lock right now,
	 * as we run under the RTNL lock.
	 */

	switch (event) {
1937 1938
	case NETDEV_CHANGENAME:
		pktgen_change_name(dev);
L
Linus Torvalds 已提交
1939
		break;
L
Luiz Capitulino 已提交
1940

L
Linus Torvalds 已提交
1941
	case NETDEV_UNREGISTER:
L
Luiz Capitulino 已提交
1942
		pktgen_mark_device(dev->name);
L
Linus Torvalds 已提交
1943
		break;
1944
	}
L
Linus Torvalds 已提交
1945 1946 1947 1948

	return NOTIFY_DONE;
}

1949 1950
static struct net_device *pktgen_dev_get_by_name(struct pktgen_dev *pkt_dev,
						 const char *ifname)
R
Robert Olsson 已提交
1951 1952
{
	char b[IFNAMSIZ+5];
1953
	int i;
R
Robert Olsson 已提交
1954

1955 1956
	for (i = 0; ifname[i] != '@'; i++) {
		if (i == IFNAMSIZ)
R
Robert Olsson 已提交
1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
			break;

		b[i] = ifname[i];
	}
	b[i] = 0;

	return dev_get_by_name(&init_net, b);
}


L
Linus Torvalds 已提交
1967 1968
/* Associate pktgen_dev with a device. */

1969
static int pktgen_setup_dev(struct pktgen_dev *pkt_dev, const char *ifname)
L
Luiz Capitulino 已提交
1970
{
L
Linus Torvalds 已提交
1971
	struct net_device *odev;
1972
	int err;
L
Linus Torvalds 已提交
1973 1974 1975 1976

	/* Clean old setups */
	if (pkt_dev->odev) {
		dev_put(pkt_dev->odev);
L
Luiz Capitulino 已提交
1977 1978
		pkt_dev->odev = NULL;
	}
L
Linus Torvalds 已提交
1979

R
Robert Olsson 已提交
1980
	odev = pktgen_dev_get_by_name(pkt_dev, ifname);
L
Linus Torvalds 已提交
1981
	if (!odev) {
J
Joe Perches 已提交
1982
		pr_err("no such netdevice: \"%s\"\n", ifname);
1983
		return -ENODEV;
L
Linus Torvalds 已提交
1984
	}
1985

L
Linus Torvalds 已提交
1986
	if (odev->type != ARPHRD_ETHER) {
J
Joe Perches 已提交
1987
		pr_err("not an ethernet device: \"%s\"\n", ifname);
1988 1989
		err = -EINVAL;
	} else if (!netif_running(odev)) {
J
Joe Perches 已提交
1990
		pr_err("device is down: \"%s\"\n", ifname);
1991 1992 1993 1994
		err = -ENETDOWN;
	} else {
		pkt_dev->odev = odev;
		return 0;
L
Linus Torvalds 已提交
1995 1996 1997
	}

	dev_put(odev);
1998
	return err;
L
Linus Torvalds 已提交
1999 2000 2001 2002 2003 2004 2005
}

/* Read pkt_dev from the interface and set up internal pktgen_dev
 * structure to have the right information to create/send packets
 */
static void pktgen_setup_inject(struct pktgen_dev *pkt_dev)
{
2006 2007
	int ntxq;

L
Luiz Capitulino 已提交
2008
	if (!pkt_dev->odev) {
J
Joe Perches 已提交
2009
		pr_err("ERROR: pkt_dev->odev == NULL in setup_inject\n");
L
Luiz Capitulino 已提交
2010 2011 2012 2013 2014
		sprintf(pkt_dev->result,
			"ERROR: pkt_dev->odev == NULL in setup_inject.\n");
		return;
	}

2015 2016
	/* make sure that we don't pick a non-existing transmit queue */
	ntxq = pkt_dev->odev->real_num_tx_queues;
R
Robert Olsson 已提交
2017

2018
	if (ntxq <= pkt_dev->queue_map_min) {
J
Joe Perches 已提交
2019 2020 2021
		pr_warning("WARNING: Requested queue_map_min (zero-based) (%d) exceeds valid range [0 - %d] for (%d) queues on %s, resetting\n",
			   pkt_dev->queue_map_min, (ntxq ?: 1) - 1, ntxq,
			   pkt_dev->odevname);
2022
		pkt_dev->queue_map_min = (ntxq ?: 1) - 1;
2023
	}
2024
	if (pkt_dev->queue_map_max >= ntxq) {
J
Joe Perches 已提交
2025 2026 2027
		pr_warning("WARNING: Requested queue_map_max (zero-based) (%d) exceeds valid range [0 - %d] for (%d) queues on %s, resetting\n",
			   pkt_dev->queue_map_max, (ntxq ?: 1) - 1, ntxq,
			   pkt_dev->odevname);
2028
		pkt_dev->queue_map_max = (ntxq ?: 1) - 1;
2029 2030
	}

L
Luiz Capitulino 已提交
2031
	/* Default to the interface's mac if not explicitly set. */
L
Linus Torvalds 已提交
2032

2033
	if (is_zero_ether_addr(pkt_dev->src_mac))
L
Luiz Capitulino 已提交
2034
		memcpy(&(pkt_dev->hh[6]), pkt_dev->odev->dev_addr, ETH_ALEN);
L
Linus Torvalds 已提交
2035

L
Luiz Capitulino 已提交
2036
	/* Set up Dest MAC */
2037
	memcpy(&(pkt_dev->hh[0]), pkt_dev->dst_mac, ETH_ALEN);
L
Linus Torvalds 已提交
2038

L
Luiz Capitulino 已提交
2039 2040 2041 2042
	/* Set up pkt size */
	pkt_dev->cur_pkt_size = pkt_dev->min_pkt_size;

	if (pkt_dev->flags & F_IPV6) {
L
Linus Torvalds 已提交
2043
		/*
2044
		 * Skip this automatic address setting until locks or functions
L
Linus Torvalds 已提交
2045 2046 2047 2048
		 * gets exported
		 */

#ifdef NOTNOW
L
Luiz Capitulino 已提交
2049
		int i, set = 0, err = 1;
L
Linus Torvalds 已提交
2050 2051
		struct inet6_dev *idev;

L
Luiz Capitulino 已提交
2052 2053
		for (i = 0; i < IN6_ADDR_HSIZE; i++)
			if (pkt_dev->cur_in6_saddr.s6_addr[i]) {
L
Linus Torvalds 已提交
2054 2055 2056 2057
				set = 1;
				break;
			}

L
Luiz Capitulino 已提交
2058 2059
		if (!set) {

L
Linus Torvalds 已提交
2060 2061 2062 2063 2064 2065
			/*
			 * Use linklevel address if unconfigured.
			 *
			 * use ipv6_get_lladdr if/when it's get exported
			 */

2066
			rcu_read_lock();
2067 2068
			idev = __in6_dev_get(pkt_dev->odev);
			if (idev) {
L
Linus Torvalds 已提交
2069 2070 2071
				struct inet6_ifaddr *ifp;

				read_lock_bh(&idev->lock);
L
Luiz Capitulino 已提交
2072 2073
				for (ifp = idev->addr_list; ifp;
				     ifp = ifp->if_next) {
2074 2075
					if (ifp->scope == IFA_LINK &&
					    !(ifp->flags & IFA_F_TENTATIVE)) {
A
Alexey Dobriyan 已提交
2076
						pkt_dev->cur_in6_saddr = ifp->addr;
L
Linus Torvalds 已提交
2077 2078 2079 2080 2081 2082
						err = 0;
						break;
					}
				}
				read_unlock_bh(&idev->lock);
			}
2083
			rcu_read_unlock();
L
Luiz Capitulino 已提交
2084
			if (err)
J
Joe Perches 已提交
2085
				pr_err("ERROR: IPv6 link address not available\n");
L
Linus Torvalds 已提交
2086 2087
		}
#endif
L
Luiz Capitulino 已提交
2088
	} else {
L
Linus Torvalds 已提交
2089 2090 2091
		pkt_dev->saddr_min = 0;
		pkt_dev->saddr_max = 0;
		if (strlen(pkt_dev->src_min) == 0) {
L
Luiz Capitulino 已提交
2092 2093

			struct in_device *in_dev;
L
Linus Torvalds 已提交
2094 2095

			rcu_read_lock();
2096
			in_dev = __in_dev_get_rcu(pkt_dev->odev);
L
Linus Torvalds 已提交
2097 2098
			if (in_dev) {
				if (in_dev->ifa_list) {
L
Luiz Capitulino 已提交
2099 2100
					pkt_dev->saddr_min =
					    in_dev->ifa_list->ifa_address;
L
Linus Torvalds 已提交
2101 2102 2103 2104
					pkt_dev->saddr_max = pkt_dev->saddr_min;
				}
			}
			rcu_read_unlock();
L
Luiz Capitulino 已提交
2105
		} else {
L
Linus Torvalds 已提交
2106 2107 2108 2109 2110 2111 2112
			pkt_dev->saddr_min = in_aton(pkt_dev->src_min);
			pkt_dev->saddr_max = in_aton(pkt_dev->src_max);
		}

		pkt_dev->daddr_min = in_aton(pkt_dev->dst_min);
		pkt_dev->daddr_max = in_aton(pkt_dev->dst_max);
	}
L
Luiz Capitulino 已提交
2113 2114 2115 2116 2117 2118 2119
	/* Initialize current values. */
	pkt_dev->cur_dst_mac_offset = 0;
	pkt_dev->cur_src_mac_offset = 0;
	pkt_dev->cur_saddr = pkt_dev->saddr_min;
	pkt_dev->cur_daddr = pkt_dev->daddr_min;
	pkt_dev->cur_udp_dst = pkt_dev->udp_dst_min;
	pkt_dev->cur_udp_src = pkt_dev->udp_src_min;
L
Linus Torvalds 已提交
2120 2121 2122
	pkt_dev->nflows = 0;
}

2123 2124 2125

static void spin(struct pktgen_dev *pkt_dev, ktime_t spin_until)
{
2126
	ktime_t start_time, end_time;
E
Eric Dumazet 已提交
2127
	s64 remaining;
S
Stephen Hemminger 已提交
2128 2129 2130 2131 2132
	struct hrtimer_sleeper t;

	hrtimer_init_on_stack(&t.timer, CLOCK_MONOTONIC, HRTIMER_MODE_ABS);
	hrtimer_set_expires(&t.timer, spin_until);

2133
	remaining = ktime_to_ns(hrtimer_expires_remaining(&t.timer));
E
Eric Dumazet 已提交
2134 2135
	if (remaining <= 0) {
		pkt_dev->next_tx = ktime_add_ns(spin_until, pkt_dev->delay);
S
Stephen Hemminger 已提交
2136
		return;
E
Eric Dumazet 已提交
2137
	}
S
Stephen Hemminger 已提交
2138

2139
	start_time = ktime_now();
E
Eric Dumazet 已提交
2140 2141 2142 2143 2144 2145
	if (remaining < 100000) {
		/* for small delays (<100us), just loop until limit is reached */
		do {
			end_time = ktime_now();
		} while (ktime_lt(end_time, spin_until));
	} else {
S
Stephen Hemminger 已提交
2146 2147 2148 2149 2150 2151 2152 2153 2154
		/* see do_nanosleep */
		hrtimer_init_sleeper(&t, current);
		do {
			set_current_state(TASK_INTERRUPTIBLE);
			hrtimer_start_expires(&t.timer, HRTIMER_MODE_ABS);
			if (!hrtimer_active(&t.timer))
				t.task = NULL;

			if (likely(t.task))
L
Linus Torvalds 已提交
2155 2156
				schedule();

S
Stephen Hemminger 已提交
2157 2158 2159
			hrtimer_cancel(&t.timer);
		} while (t.task && pkt_dev->running && !signal_pending(current));
		__set_current_state(TASK_RUNNING);
E
Eric Dumazet 已提交
2160
		end_time = ktime_now();
L
Linus Torvalds 已提交
2161
	}
2162 2163

	pkt_dev->idle_acc += ktime_to_ns(ktime_sub(end_time, start_time));
2164
	pkt_dev->next_tx = ktime_add_ns(spin_until, pkt_dev->delay);
L
Linus Torvalds 已提交
2165 2166
}

2167 2168
static inline void set_pkt_overhead(struct pktgen_dev *pkt_dev)
{
J
Jamal Hadi Salim 已提交
2169
	pkt_dev->pkt_overhead = 0;
2170 2171 2172 2173 2174
	pkt_dev->pkt_overhead += pkt_dev->nr_labels*sizeof(u32);
	pkt_dev->pkt_overhead += VLAN_TAG_SIZE(pkt_dev);
	pkt_dev->pkt_overhead += SVLAN_TAG_SIZE(pkt_dev);
}

S
Stephen Hemminger 已提交
2175
static inline int f_seen(const struct pktgen_dev *pkt_dev, int flow)
2176
{
S
Stephen Hemminger 已提交
2177
	return !!(pkt_dev->flows[flow].flags & F_INIT);
2178 2179 2180 2181 2182 2183 2184 2185 2186 2187
}

static inline int f_pick(struct pktgen_dev *pkt_dev)
{
	int flow = pkt_dev->curfl;

	if (pkt_dev->flags & F_FLOW_SEQ) {
		if (pkt_dev->flows[flow].count >= pkt_dev->lflow) {
			/* reset time */
			pkt_dev->flows[flow].count = 0;
R
Robert Olsson 已提交
2188
			pkt_dev->flows[flow].flags = 0;
2189 2190 2191 2192 2193 2194
			pkt_dev->curfl += 1;
			if (pkt_dev->curfl >= pkt_dev->cflows)
				pkt_dev->curfl = 0; /*reset */
		}
	} else {
		flow = random32() % pkt_dev->cflows;
R
Robert Olsson 已提交
2195
		pkt_dev->curfl = flow;
2196

R
Robert Olsson 已提交
2197
		if (pkt_dev->flows[flow].count > pkt_dev->lflow) {
2198
			pkt_dev->flows[flow].count = 0;
R
Robert Olsson 已提交
2199 2200
			pkt_dev->flows[flow].flags = 0;
		}
2201 2202 2203 2204 2205
	}

	return pkt_dev->curfl;
}

J
Jamal Hadi Salim 已提交
2206 2207 2208 2209 2210

#ifdef CONFIG_XFRM
/* If there was already an IPSEC SA, we keep it as is, else
 * we go look for it ...
*/
2211
#define DUMMY_MARK 0
2212
static void get_ipsec_sa(struct pktgen_dev *pkt_dev, int flow)
J
Jamal Hadi Salim 已提交
2213 2214 2215 2216
{
	struct xfrm_state *x = pkt_dev->flows[flow].x;
	if (!x) {
		/*slow path: we dont already have xfrm_state*/
2217
		x = xfrm_stateonly_find(&init_net, DUMMY_MARK,
2218
					(xfrm_address_t *)&pkt_dev->cur_daddr,
J
Jamal Hadi Salim 已提交
2219 2220 2221 2222 2223 2224 2225
					(xfrm_address_t *)&pkt_dev->cur_saddr,
					AF_INET,
					pkt_dev->ipsmode,
					pkt_dev->ipsproto, 0);
		if (x) {
			pkt_dev->flows[flow].x = x;
			set_pkt_overhead(pkt_dev);
2226
			pkt_dev->pkt_overhead += x->props.header_len;
J
Jamal Hadi Salim 已提交
2227 2228 2229 2230 2231
		}

	}
}
#endif
2232 2233
static void set_cur_queue_map(struct pktgen_dev *pkt_dev)
{
R
Robert Olsson 已提交
2234 2235 2236 2237

	if (pkt_dev->flags & F_QUEUE_MAP_CPU)
		pkt_dev->cur_queue_map = smp_processor_id();

E
Eric Dumazet 已提交
2238
	else if (pkt_dev->queue_map_min <= pkt_dev->queue_map_max) {
2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251
		__u16 t;
		if (pkt_dev->flags & F_QUEUE_MAP_RND) {
			t = random32() %
				(pkt_dev->queue_map_max -
				 pkt_dev->queue_map_min + 1)
				+ pkt_dev->queue_map_min;
		} else {
			t = pkt_dev->cur_queue_map + 1;
			if (t > pkt_dev->queue_map_max)
				t = pkt_dev->queue_map_min;
		}
		pkt_dev->cur_queue_map = t;
	}
R
Robert Olsson 已提交
2252
	pkt_dev->cur_queue_map  = pkt_dev->cur_queue_map % pkt_dev->odev->real_num_tx_queues;
2253 2254
}

L
Linus Torvalds 已提交
2255 2256 2257
/* Increment/randomize headers according to flags and current values
 * for IP src/dest, UDP src/dst port, MAC-Addr src/dst
 */
L
Luiz Capitulino 已提交
2258 2259 2260 2261 2262
static void mod_cur_headers(struct pktgen_dev *pkt_dev)
{
	__u32 imn;
	__u32 imx;
	int flow = 0;
L
Linus Torvalds 已提交
2263

2264 2265
	if (pkt_dev->cflows)
		flow = f_pick(pkt_dev);
L
Linus Torvalds 已提交
2266 2267

	/*  Deal with source MAC */
L
Luiz Capitulino 已提交
2268 2269 2270 2271 2272
	if (pkt_dev->src_mac_count > 1) {
		__u32 mc;
		__u32 tmp;

		if (pkt_dev->flags & F_MACSRC_RND)
S
Stephen Hemminger 已提交
2273
			mc = random32() % pkt_dev->src_mac_count;
L
Luiz Capitulino 已提交
2274 2275
		else {
			mc = pkt_dev->cur_src_mac_offset++;
R
Robert Olsson 已提交
2276
			if (pkt_dev->cur_src_mac_offset >=
L
Luiz Capitulino 已提交
2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298
			    pkt_dev->src_mac_count)
				pkt_dev->cur_src_mac_offset = 0;
		}

		tmp = pkt_dev->src_mac[5] + (mc & 0xFF);
		pkt_dev->hh[11] = tmp;
		tmp = (pkt_dev->src_mac[4] + ((mc >> 8) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[10] = tmp;
		tmp = (pkt_dev->src_mac[3] + ((mc >> 16) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[9] = tmp;
		tmp = (pkt_dev->src_mac[2] + ((mc >> 24) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[8] = tmp;
		tmp = (pkt_dev->src_mac[1] + (tmp >> 8));
		pkt_dev->hh[7] = tmp;
	}

	/*  Deal with Destination MAC */
	if (pkt_dev->dst_mac_count > 1) {
		__u32 mc;
		__u32 tmp;

		if (pkt_dev->flags & F_MACDST_RND)
S
Stephen Hemminger 已提交
2299
			mc = random32() % pkt_dev->dst_mac_count;
L
Luiz Capitulino 已提交
2300 2301 2302

		else {
			mc = pkt_dev->cur_dst_mac_offset++;
R
Robert Olsson 已提交
2303
			if (pkt_dev->cur_dst_mac_offset >=
L
Luiz Capitulino 已提交
2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
			    pkt_dev->dst_mac_count) {
				pkt_dev->cur_dst_mac_offset = 0;
			}
		}

		tmp = pkt_dev->dst_mac[5] + (mc & 0xFF);
		pkt_dev->hh[5] = tmp;
		tmp = (pkt_dev->dst_mac[4] + ((mc >> 8) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[4] = tmp;
		tmp = (pkt_dev->dst_mac[3] + ((mc >> 16) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[3] = tmp;
		tmp = (pkt_dev->dst_mac[2] + ((mc >> 24) & 0xFF) + (tmp >> 8));
		pkt_dev->hh[2] = tmp;
		tmp = (pkt_dev->dst_mac[1] + (tmp >> 8));
		pkt_dev->hh[1] = tmp;
	}

2321
	if (pkt_dev->flags & F_MPLS_RND) {
2322
		unsigned int i;
2323
		for (i = 0; i < pkt_dev->nr_labels; i++)
2324 2325
			if (pkt_dev->labels[i] & MPLS_STACK_BOTTOM)
				pkt_dev->labels[i] = MPLS_STACK_BOTTOM |
S
Stephen Hemminger 已提交
2326
					     ((__force __be32)random32() &
2327 2328 2329
						      htonl(0x000fffff));
	}

F
Francesco Fondelli 已提交
2330
	if ((pkt_dev->flags & F_VID_RND) && (pkt_dev->vlan_id != 0xffff)) {
S
Stephen Hemminger 已提交
2331
		pkt_dev->vlan_id = random32() & (4096-1);
F
Francesco Fondelli 已提交
2332 2333 2334
	}

	if ((pkt_dev->flags & F_SVID_RND) && (pkt_dev->svlan_id != 0xffff)) {
S
Stephen Hemminger 已提交
2335
		pkt_dev->svlan_id = random32() & (4096 - 1);
F
Francesco Fondelli 已提交
2336 2337
	}

L
Luiz Capitulino 已提交
2338 2339
	if (pkt_dev->udp_src_min < pkt_dev->udp_src_max) {
		if (pkt_dev->flags & F_UDPSRC_RND)
S
Stephen Hemminger 已提交
2340 2341 2342
			pkt_dev->cur_udp_src = random32() %
				(pkt_dev->udp_src_max - pkt_dev->udp_src_min)
				+ pkt_dev->udp_src_min;
L
Luiz Capitulino 已提交
2343 2344

		else {
L
Linus Torvalds 已提交
2345 2346 2347
			pkt_dev->cur_udp_src++;
			if (pkt_dev->cur_udp_src >= pkt_dev->udp_src_max)
				pkt_dev->cur_udp_src = pkt_dev->udp_src_min;
L
Luiz Capitulino 已提交
2348 2349 2350 2351 2352
		}
	}

	if (pkt_dev->udp_dst_min < pkt_dev->udp_dst_max) {
		if (pkt_dev->flags & F_UDPDST_RND) {
S
Stephen Hemminger 已提交
2353 2354 2355
			pkt_dev->cur_udp_dst = random32() %
				(pkt_dev->udp_dst_max - pkt_dev->udp_dst_min)
				+ pkt_dev->udp_dst_min;
L
Luiz Capitulino 已提交
2356
		} else {
L
Linus Torvalds 已提交
2357
			pkt_dev->cur_udp_dst++;
L
Luiz Capitulino 已提交
2358
			if (pkt_dev->cur_udp_dst >= pkt_dev->udp_dst_max)
L
Linus Torvalds 已提交
2359
				pkt_dev->cur_udp_dst = pkt_dev->udp_dst_min;
L
Luiz Capitulino 已提交
2360 2361
		}
	}
L
Linus Torvalds 已提交
2362 2363 2364

	if (!(pkt_dev->flags & F_IPV6)) {

2365 2366 2367
		imn = ntohl(pkt_dev->saddr_min);
		imx = ntohl(pkt_dev->saddr_max);
		if (imn < imx) {
L
Linus Torvalds 已提交
2368
			__u32 t;
L
Luiz Capitulino 已提交
2369
			if (pkt_dev->flags & F_IPSRC_RND)
S
Stephen Hemminger 已提交
2370
				t = random32() % (imx - imn) + imn;
L
Linus Torvalds 已提交
2371 2372 2373
			else {
				t = ntohl(pkt_dev->cur_saddr);
				t++;
2374
				if (t > imx)
L
Linus Torvalds 已提交
2375
					t = imn;
2376

L
Linus Torvalds 已提交
2377 2378 2379
			}
			pkt_dev->cur_saddr = htonl(t);
		}
L
Luiz Capitulino 已提交
2380

2381
		if (pkt_dev->cflows && f_seen(pkt_dev, flow)) {
L
Linus Torvalds 已提交
2382 2383
			pkt_dev->cur_daddr = pkt_dev->flows[flow].cur_daddr;
		} else {
A
Al Viro 已提交
2384 2385 2386
			imn = ntohl(pkt_dev->daddr_min);
			imx = ntohl(pkt_dev->daddr_max);
			if (imn < imx) {
L
Linus Torvalds 已提交
2387
				__u32 t;
A
Al Viro 已提交
2388
				__be32 s;
L
Linus Torvalds 已提交
2389 2390
				if (pkt_dev->flags & F_IPDST_RND) {

S
Stephen Hemminger 已提交
2391
					t = random32() % (imx - imn) + imn;
A
Al Viro 已提交
2392
					s = htonl(t);
L
Linus Torvalds 已提交
2393

2394 2395
					while (ipv4_is_loopback(s) ||
					       ipv4_is_multicast(s) ||
2396
					       ipv4_is_lbcast(s) ||
2397 2398
					       ipv4_is_zeronet(s) ||
					       ipv4_is_local_multicast(s)) {
S
Stephen Hemminger 已提交
2399
						t = random32() % (imx - imn) + imn;
A
Al Viro 已提交
2400
						s = htonl(t);
L
Linus Torvalds 已提交
2401
					}
A
Al Viro 已提交
2402 2403
					pkt_dev->cur_daddr = s;
				} else {
L
Linus Torvalds 已提交
2404 2405 2406 2407 2408 2409 2410 2411
					t = ntohl(pkt_dev->cur_daddr);
					t++;
					if (t > imx) {
						t = imn;
					}
					pkt_dev->cur_daddr = htonl(t);
				}
			}
L
Luiz Capitulino 已提交
2412
			if (pkt_dev->cflows) {
2413
				pkt_dev->flows[flow].flags |= F_INIT;
L
Luiz Capitulino 已提交
2414 2415
				pkt_dev->flows[flow].cur_daddr =
				    pkt_dev->cur_daddr;
J
Jamal Hadi Salim 已提交
2416 2417 2418 2419
#ifdef CONFIG_XFRM
				if (pkt_dev->flags & F_IPSEC_ON)
					get_ipsec_sa(pkt_dev, flow);
#endif
L
Linus Torvalds 已提交
2420 2421 2422
				pkt_dev->nflows++;
			}
		}
L
Luiz Capitulino 已提交
2423 2424 2425 2426 2427 2428
	} else {		/* IPV6 * */

		if (pkt_dev->min_in6_daddr.s6_addr32[0] == 0 &&
		    pkt_dev->min_in6_daddr.s6_addr32[1] == 0 &&
		    pkt_dev->min_in6_daddr.s6_addr32[2] == 0 &&
		    pkt_dev->min_in6_daddr.s6_addr32[3] == 0) ;
L
Linus Torvalds 已提交
2429 2430 2431 2432 2433
		else {
			int i;

			/* Only random destinations yet */

L
Luiz Capitulino 已提交
2434
			for (i = 0; i < 4; i++) {
L
Linus Torvalds 已提交
2435
				pkt_dev->cur_in6_daddr.s6_addr32[i] =
S
Stephen Hemminger 已提交
2436
				    (((__force __be32)random32() |
L
Luiz Capitulino 已提交
2437 2438
				      pkt_dev->min_in6_daddr.s6_addr32[i]) &
				     pkt_dev->max_in6_daddr.s6_addr32[i]);
L
Linus Torvalds 已提交
2439
			}
L
Luiz Capitulino 已提交
2440
		}
L
Linus Torvalds 已提交
2441 2442
	}

L
Luiz Capitulino 已提交
2443 2444 2445
	if (pkt_dev->min_pkt_size < pkt_dev->max_pkt_size) {
		__u32 t;
		if (pkt_dev->flags & F_TXSIZE_RND) {
S
Stephen Hemminger 已提交
2446 2447 2448
			t = random32() %
				(pkt_dev->max_pkt_size - pkt_dev->min_pkt_size)
				+ pkt_dev->min_pkt_size;
L
Luiz Capitulino 已提交
2449
		} else {
L
Linus Torvalds 已提交
2450
			t = pkt_dev->cur_pkt_size + 1;
L
Luiz Capitulino 已提交
2451
			if (t > pkt_dev->max_pkt_size)
L
Linus Torvalds 已提交
2452
				t = pkt_dev->min_pkt_size;
L
Luiz Capitulino 已提交
2453 2454 2455
		}
		pkt_dev->cur_pkt_size = t;
	}
L
Linus Torvalds 已提交
2456

2457
	set_cur_queue_map(pkt_dev);
R
Robert Olsson 已提交
2458

L
Linus Torvalds 已提交
2459 2460 2461
	pkt_dev->flows[flow].count++;
}

J
Jamal Hadi Salim 已提交
2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477

#ifdef CONFIG_XFRM
static int pktgen_output_ipsec(struct sk_buff *skb, struct pktgen_dev *pkt_dev)
{
	struct xfrm_state *x = pkt_dev->flows[pkt_dev->curfl].x;
	int err = 0;

	if (!x)
		return 0;
	/* XXX: we dont support tunnel mode for now until
	 * we resolve the dst issue */
	if (x->props.mode != XFRM_MODE_TRANSPORT)
		return 0;

	spin_lock(&x->lock);

2478
	err = x->outer_mode->output(x, skb);
J
Jamal Hadi Salim 已提交
2479 2480 2481 2482 2483 2484
	if (err)
		goto error;
	err = x->type->output(x, skb);
	if (err)
		goto error;

2485
	x->curlft.bytes += skb->len;
J
Jamal Hadi Salim 已提交
2486 2487 2488 2489 2490 2491
	x->curlft.packets++;
error:
	spin_unlock(&x->lock);
	return err;
}

S
Stephen Hemminger 已提交
2492
static void free_SAs(struct pktgen_dev *pkt_dev)
J
Jamal Hadi Salim 已提交
2493 2494 2495
{
	if (pkt_dev->cflows) {
		/* let go of the SAs if we have them */
2496 2497
		int i;
		for (i = 0; i < pkt_dev->cflows; i++) {
J
Jamal Hadi Salim 已提交
2498 2499 2500 2501 2502 2503 2504 2505 2506
			struct xfrm_state *x = pkt_dev->flows[i].x;
			if (x) {
				xfrm_state_put(x);
				pkt_dev->flows[i].x = NULL;
			}
		}
	}
}

S
Stephen Hemminger 已提交
2507
static int process_ipsec(struct pktgen_dev *pkt_dev,
J
Jamal Hadi Salim 已提交
2508 2509 2510 2511 2512 2513 2514 2515 2516
			      struct sk_buff *skb, __be16 protocol)
{
	if (pkt_dev->flags & F_IPSEC_ON) {
		struct xfrm_state *x = pkt_dev->flows[pkt_dev->curfl].x;
		int nhead = 0;
		if (x) {
			int ret;
			__u8 *eth;
			nhead = x->props.header_len - skb_headroom(skb);
2517
			if (nhead > 0) {
J
Jamal Hadi Salim 已提交
2518 2519
				ret = pskb_expand_head(skb, nhead, 0, GFP_ATOMIC);
				if (ret < 0) {
J
Joe Perches 已提交
2520 2521
					pr_err("Error expanding ipsec packet %d\n",
					       ret);
2522
					goto err;
J
Jamal Hadi Salim 已提交
2523 2524 2525 2526 2527 2528 2529
				}
			}

			/* ipsec is not expecting ll header */
			skb_pull(skb, ETH_HLEN);
			ret = pktgen_output_ipsec(skb, pkt_dev);
			if (ret) {
J
Joe Perches 已提交
2530
				pr_err("Error creating ipsec packet %d\n", ret);
2531
				goto err;
J
Jamal Hadi Salim 已提交
2532 2533 2534 2535
			}
			/* restore ll */
			eth = (__u8 *) skb_push(skb, ETH_HLEN);
			memcpy(eth, pkt_dev->hh, 12);
2536
			*(u16 *) &eth[12] = protocol;
J
Jamal Hadi Salim 已提交
2537 2538 2539
		}
	}
	return 1;
2540 2541 2542
err:
	kfree_skb(skb);
	return 0;
J
Jamal Hadi Salim 已提交
2543 2544 2545
}
#endif

2546 2547
static void mpls_push(__be32 *mpls, struct pktgen_dev *pkt_dev)
{
2548
	unsigned int i;
2549
	for (i = 0; i < pkt_dev->nr_labels; i++)
2550
		*mpls++ = pkt_dev->labels[i] & ~MPLS_STACK_BOTTOM;
2551

2552 2553 2554 2555
	mpls--;
	*mpls |= MPLS_STACK_BOTTOM;
}

A
Al Viro 已提交
2556 2557 2558 2559 2560 2561
static inline __be16 build_tci(unsigned int id, unsigned int cfi,
			       unsigned int prio)
{
	return htons(id | (cfi << 12) | (prio << 13));
}

E
Eric Dumazet 已提交
2562 2563 2564 2565 2566 2567 2568 2569 2570 2571
static void pktgen_finalize_skb(struct pktgen_dev *pkt_dev, struct sk_buff *skb,
				int datalen)
{
	struct timeval timestamp;
	struct pktgen_hdr *pgh;

	pgh = (struct pktgen_hdr *)skb_put(skb, sizeof(*pgh));
	datalen -= sizeof(*pgh);

	if (pkt_dev->nfrags <= 0) {
2572
		memset(skb_put(skb, datalen), 0, datalen);
E
Eric Dumazet 已提交
2573 2574 2575
	} else {
		int frags = pkt_dev->nfrags;
		int i, len;
2576
		int frag_len;
E
Eric Dumazet 已提交
2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587


		if (frags > MAX_SKB_FRAGS)
			frags = MAX_SKB_FRAGS;
		len = datalen - frags * PAGE_SIZE;
		if (len > 0) {
			memset(skb_put(skb, len), 0, len);
			datalen = frags * PAGE_SIZE;
		}

		i = 0;
2588 2589
		frag_len = (datalen/frags) < PAGE_SIZE ?
			   (datalen/frags) : PAGE_SIZE;
E
Eric Dumazet 已提交
2590 2591 2592 2593 2594 2595 2596 2597 2598 2599
		while (datalen > 0) {
			if (unlikely(!pkt_dev->page)) {
				int node = numa_node_id();

				if (pkt_dev->node >= 0 && (pkt_dev->flags & F_NODE))
					node = pkt_dev->node;
				pkt_dev->page = alloc_pages_node(node, GFP_KERNEL | __GFP_ZERO, 0);
				if (!pkt_dev->page)
					break;
			}
2600
			get_page(pkt_dev->page);
2601
			skb_frag_set_page(skb, i, pkt_dev->page);
E
Eric Dumazet 已提交
2602
			skb_shinfo(skb)->frags[i].page_offset = 0;
2603 2604
			/*last fragment, fill rest of data*/
			if (i == (frags - 1))
E
Eric Dumazet 已提交
2605 2606
				skb_frag_size_set(&skb_shinfo(skb)->frags[i],
				    (datalen < PAGE_SIZE ? datalen : PAGE_SIZE));
2607
			else
E
Eric Dumazet 已提交
2608 2609 2610 2611
				skb_frag_size_set(&skb_shinfo(skb)->frags[i], frag_len);
			datalen -= skb_frag_size(&skb_shinfo(skb)->frags[i]);
			skb->len += skb_frag_size(&skb_shinfo(skb)->frags[i]);
			skb->data_len += skb_frag_size(&skb_shinfo(skb)->frags[i]);
E
Eric Dumazet 已提交
2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627
			i++;
			skb_shinfo(skb)->nr_frags = i;
		}
	}

	/* Stamp the time, and sequence number,
	 * convert them to network byte order
	 */
	pgh->pgh_magic = htonl(PKTGEN_MAGIC);
	pgh->seq_num = htonl(pkt_dev->seq_num);

	do_gettimeofday(&timestamp);
	pgh->tv_sec = htonl(timestamp.tv_sec);
	pgh->tv_usec = htonl(timestamp.tv_usec);
}

L
Luiz Capitulino 已提交
2628 2629
static struct sk_buff *fill_packet_ipv4(struct net_device *odev,
					struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
2630 2631 2632 2633 2634 2635
{
	struct sk_buff *skb = NULL;
	__u8 *eth;
	struct udphdr *udph;
	int datalen, iplen;
	struct iphdr *iph;
2636
	__be16 protocol = htons(ETH_P_IP);
2637
	__be32 *mpls;
F
Francesco Fondelli 已提交
2638 2639 2640 2641
	__be16 *vlan_tci = NULL;                 /* Encapsulates priority and VLAN ID */
	__be16 *vlan_encapsulated_proto = NULL;  /* packet type ID field (or len) for VLAN tag */
	__be16 *svlan_tci = NULL;                /* Encapsulates priority and SVLAN ID */
	__be16 *svlan_encapsulated_proto = NULL; /* packet type ID field (or len) for SVLAN tag */
2642
	u16 queue_map;
2643 2644

	if (pkt_dev->nr_labels)
2645
		protocol = htons(ETH_P_MPLS_UC);
L
Luiz Capitulino 已提交
2646

F
Francesco Fondelli 已提交
2647
	if (pkt_dev->vlan_id != 0xffff)
2648
		protocol = htons(ETH_P_8021Q);
F
Francesco Fondelli 已提交
2649

2650 2651 2652 2653
	/* Update any of the values, used when we're incrementing various
	 * fields.
	 */
	mod_cur_headers(pkt_dev);
2654
	queue_map = pkt_dev->cur_queue_map;
2655

2656
	datalen = (odev->hard_header_len + 16) & ~0xf;
R
Robert Olsson 已提交
2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677

	if (pkt_dev->flags & F_NODE) {
		int node;

		if (pkt_dev->node >= 0)
			node = pkt_dev->node;
		else
			node =  numa_node_id();

		skb = __alloc_skb(NET_SKB_PAD + pkt_dev->cur_pkt_size + 64
				  + datalen + pkt_dev->pkt_overhead, GFP_NOWAIT, 0, node);
		if (likely(skb)) {
			skb_reserve(skb, NET_SKB_PAD);
			skb->dev = odev;
		}
	}
	else
	  skb = __netdev_alloc_skb(odev,
				   pkt_dev->cur_pkt_size + 64
				   + datalen + pkt_dev->pkt_overhead, GFP_NOWAIT);

L
Linus Torvalds 已提交
2678 2679 2680 2681
	if (!skb) {
		sprintf(pkt_dev->result, "No memory");
		return NULL;
	}
J
Junchang Wang 已提交
2682
	prefetchw(skb->data);
L
Linus Torvalds 已提交
2683

2684
	skb_reserve(skb, datalen);
L
Linus Torvalds 已提交
2685 2686 2687

	/*  Reserve for ethernet and IP header  */
	eth = (__u8 *) skb_push(skb, 14);
2688 2689 2690
	mpls = (__be32 *)skb_put(skb, pkt_dev->nr_labels*sizeof(__u32));
	if (pkt_dev->nr_labels)
		mpls_push(mpls, pkt_dev);
F
Francesco Fondelli 已提交
2691 2692

	if (pkt_dev->vlan_id != 0xffff) {
2693
		if (pkt_dev->svlan_id != 0xffff) {
F
Francesco Fondelli 已提交
2694
			svlan_tci = (__be16 *)skb_put(skb, sizeof(__be16));
A
Al Viro 已提交
2695 2696 2697
			*svlan_tci = build_tci(pkt_dev->svlan_id,
					       pkt_dev->svlan_cfi,
					       pkt_dev->svlan_p);
F
Francesco Fondelli 已提交
2698
			svlan_encapsulated_proto = (__be16 *)skb_put(skb, sizeof(__be16));
2699
			*svlan_encapsulated_proto = htons(ETH_P_8021Q);
F
Francesco Fondelli 已提交
2700 2701
		}
		vlan_tci = (__be16 *)skb_put(skb, sizeof(__be16));
A
Al Viro 已提交
2702 2703 2704
		*vlan_tci = build_tci(pkt_dev->vlan_id,
				      pkt_dev->vlan_cfi,
				      pkt_dev->vlan_p);
F
Francesco Fondelli 已提交
2705
		vlan_encapsulated_proto = (__be16 *)skb_put(skb, sizeof(__be16));
2706
		*vlan_encapsulated_proto = htons(ETH_P_IP);
F
Francesco Fondelli 已提交
2707 2708
	}

2709
	skb->network_header = skb->tail;
2710
	skb->transport_header = skb->network_header + sizeof(struct iphdr);
2711
	skb_put(skb, sizeof(struct iphdr) + sizeof(struct udphdr));
2712
	skb_set_queue_mapping(skb, queue_map);
J
John Fastabend 已提交
2713 2714
	skb->priority = pkt_dev->skb_priority;

2715 2716
	iph = ip_hdr(skb);
	udph = udp_hdr(skb);
L
Linus Torvalds 已提交
2717 2718

	memcpy(eth, pkt_dev->hh, 12);
A
Al Viro 已提交
2719
	*(__be16 *) & eth[12] = protocol;
L
Linus Torvalds 已提交
2720

2721 2722
	/* Eth + IPh + UDPh + mpls */
	datalen = pkt_dev->cur_pkt_size - 14 - 20 - 8 -
2723
		  pkt_dev->pkt_overhead;
2724
	if (datalen < 0 || datalen < sizeof(struct pktgen_hdr))
L
Linus Torvalds 已提交
2725
		datalen = sizeof(struct pktgen_hdr);
L
Luiz Capitulino 已提交
2726

L
Linus Torvalds 已提交
2727 2728
	udph->source = htons(pkt_dev->cur_udp_src);
	udph->dest = htons(pkt_dev->cur_udp_dst);
L
Luiz Capitulino 已提交
2729 2730
	udph->len = htons(datalen + 8);	/* DATA + udphdr */
	udph->check = 0;	/* No checksum */
L
Linus Torvalds 已提交
2731 2732 2733 2734

	iph->ihl = 5;
	iph->version = 4;
	iph->ttl = 32;
F
Francesco Fondelli 已提交
2735
	iph->tos = pkt_dev->tos;
L
Luiz Capitulino 已提交
2736
	iph->protocol = IPPROTO_UDP;	/* UDP */
L
Linus Torvalds 已提交
2737 2738
	iph->saddr = pkt_dev->cur_saddr;
	iph->daddr = pkt_dev->cur_daddr;
E
Eric Dumazet 已提交
2739 2740
	iph->id = htons(pkt_dev->ip_id);
	pkt_dev->ip_id++;
L
Linus Torvalds 已提交
2741 2742 2743 2744
	iph->frag_off = 0;
	iplen = 20 + 8 + datalen;
	iph->tot_len = htons(iplen);
	iph->check = 0;
L
Luiz Capitulino 已提交
2745
	iph->check = ip_fast_csum((void *)iph, iph->ihl);
2746
	skb->protocol = protocol;
2747
	skb->mac_header = (skb->network_header - ETH_HLEN -
2748
			   pkt_dev->pkt_overhead);
L
Linus Torvalds 已提交
2749 2750
	skb->dev = odev;
	skb->pkt_type = PACKET_HOST;
E
Eric Dumazet 已提交
2751
	pktgen_finalize_skb(pkt_dev, skb, datalen);
L
Linus Torvalds 已提交
2752

J
Jamal Hadi Salim 已提交
2753 2754 2755 2756 2757
#ifdef CONFIG_XFRM
	if (!process_ipsec(pkt_dev, skb, protocol))
		return NULL;
#endif

L
Linus Torvalds 已提交
2758 2759 2760 2761
	return skb;
}

/*
2762
 * scan_ip6, fmt_ip taken from dietlibc-0.21
L
Linus Torvalds 已提交
2763 2764
 * Author Felix von Leitner <felix-dietlibc@fefe.de>
 *
2765
 * Slightly modified for kernel.
L
Linus Torvalds 已提交
2766 2767 2768 2769
 * Should be candidate for net/ipv4/utils.c
 * --ro
 */

L
Luiz Capitulino 已提交
2770
static unsigned int scan_ip6(const char *s, char ip[16])
L
Linus Torvalds 已提交
2771 2772
{
	unsigned int i;
L
Luiz Capitulino 已提交
2773
	unsigned int len = 0;
L
Linus Torvalds 已提交
2774 2775
	unsigned long u;
	char suffix[16];
L
Luiz Capitulino 已提交
2776 2777
	unsigned int prefixlen = 0;
	unsigned int suffixlen = 0;
A
Al Viro 已提交
2778
	__be32 tmp;
S
Stephen Hemminger 已提交
2779
	char *pos;
L
Linus Torvalds 已提交
2780

L
Luiz Capitulino 已提交
2781 2782
	for (i = 0; i < 16; i++)
		ip[i] = 0;
L
Linus Torvalds 已提交
2783 2784 2785 2786

	for (;;) {
		if (*s == ':') {
			len++;
L
Luiz Capitulino 已提交
2787 2788
			if (s[1] == ':') {	/* Found "::", skip to part 2 */
				s += 2;
L
Linus Torvalds 已提交
2789 2790 2791 2792 2793 2794
				len++;
				break;
			}
			s++;
		}

S
Stephen Hemminger 已提交
2795 2796
		u = simple_strtoul(s, &pos, 16);
		i = pos - s;
L
Luiz Capitulino 已提交
2797 2798 2799
		if (!i)
			return 0;
		if (prefixlen == 12 && s[i] == '.') {
L
Linus Torvalds 已提交
2800 2801 2802 2803

			/* the last 4 bytes may be written as IPv4 address */

			tmp = in_aton(s);
L
Luiz Capitulino 已提交
2804 2805
			memcpy((struct in_addr *)(ip + 12), &tmp, sizeof(tmp));
			return i + len;
L
Linus Torvalds 已提交
2806 2807 2808
		}
		ip[prefixlen++] = (u >> 8);
		ip[prefixlen++] = (u & 255);
L
Luiz Capitulino 已提交
2809 2810 2811
		s += i;
		len += i;
		if (prefixlen == 16)
L
Linus Torvalds 已提交
2812 2813 2814 2815 2816 2817
			return len;
	}

/* part 2, after "::" */
	for (;;) {
		if (*s == ':') {
L
Luiz Capitulino 已提交
2818
			if (suffixlen == 0)
L
Linus Torvalds 已提交
2819 2820 2821
				break;
			s++;
			len++;
L
Luiz Capitulino 已提交
2822
		} else if (suffixlen != 0)
L
Linus Torvalds 已提交
2823
			break;
S
Stephen Hemminger 已提交
2824 2825 2826

		u = simple_strtol(s, &pos, 16);
		i = pos - s;
L
Linus Torvalds 已提交
2827
		if (!i) {
L
Luiz Capitulino 已提交
2828 2829
			if (*s)
				len--;
L
Linus Torvalds 已提交
2830 2831
			break;
		}
L
Luiz Capitulino 已提交
2832
		if (suffixlen + prefixlen <= 12 && s[i] == '.') {
L
Linus Torvalds 已提交
2833
			tmp = in_aton(s);
L
Luiz Capitulino 已提交
2834 2835 2836 2837
			memcpy((struct in_addr *)(suffix + suffixlen), &tmp,
			       sizeof(tmp));
			suffixlen += 4;
			len += strlen(s);
L
Linus Torvalds 已提交
2838 2839 2840 2841
			break;
		}
		suffix[suffixlen++] = (u >> 8);
		suffix[suffixlen++] = (u & 255);
L
Luiz Capitulino 已提交
2842 2843 2844
		s += i;
		len += i;
		if (prefixlen + suffixlen == 16)
L
Linus Torvalds 已提交
2845 2846
			break;
	}
L
Luiz Capitulino 已提交
2847 2848
	for (i = 0; i < suffixlen; i++)
		ip[16 - suffixlen + i] = suffix[i];
L
Linus Torvalds 已提交
2849 2850 2851
	return len;
}

L
Luiz Capitulino 已提交
2852 2853
static struct sk_buff *fill_packet_ipv6(struct net_device *odev,
					struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
2854 2855 2856 2857 2858 2859
{
	struct sk_buff *skb = NULL;
	__u8 *eth;
	struct udphdr *udph;
	int datalen;
	struct ipv6hdr *iph;
2860
	__be16 protocol = htons(ETH_P_IPV6);
2861
	__be32 *mpls;
F
Francesco Fondelli 已提交
2862 2863 2864 2865
	__be16 *vlan_tci = NULL;                 /* Encapsulates priority and VLAN ID */
	__be16 *vlan_encapsulated_proto = NULL;  /* packet type ID field (or len) for VLAN tag */
	__be16 *svlan_tci = NULL;                /* Encapsulates priority and SVLAN ID */
	__be16 *svlan_encapsulated_proto = NULL; /* packet type ID field (or len) for SVLAN tag */
2866
	u16 queue_map;
2867 2868

	if (pkt_dev->nr_labels)
2869
		protocol = htons(ETH_P_MPLS_UC);
2870

F
Francesco Fondelli 已提交
2871
	if (pkt_dev->vlan_id != 0xffff)
2872
		protocol = htons(ETH_P_8021Q);
F
Francesco Fondelli 已提交
2873

2874 2875 2876 2877
	/* Update any of the values, used when we're incrementing various
	 * fields.
	 */
	mod_cur_headers(pkt_dev);
2878
	queue_map = pkt_dev->cur_queue_map;
2879

S
Stephen Hemminger 已提交
2880 2881 2882
	skb = __netdev_alloc_skb(odev,
				 pkt_dev->cur_pkt_size + 64
				 + 16 + pkt_dev->pkt_overhead, GFP_NOWAIT);
L
Linus Torvalds 已提交
2883 2884 2885 2886
	if (!skb) {
		sprintf(pkt_dev->result, "No memory");
		return NULL;
	}
J
Junchang Wang 已提交
2887
	prefetchw(skb->data);
L
Linus Torvalds 已提交
2888 2889 2890 2891 2892

	skb_reserve(skb, 16);

	/*  Reserve for ethernet and IP header  */
	eth = (__u8 *) skb_push(skb, 14);
2893 2894 2895
	mpls = (__be32 *)skb_put(skb, pkt_dev->nr_labels*sizeof(__u32));
	if (pkt_dev->nr_labels)
		mpls_push(mpls, pkt_dev);
F
Francesco Fondelli 已提交
2896 2897

	if (pkt_dev->vlan_id != 0xffff) {
2898
		if (pkt_dev->svlan_id != 0xffff) {
F
Francesco Fondelli 已提交
2899
			svlan_tci = (__be16 *)skb_put(skb, sizeof(__be16));
A
Al Viro 已提交
2900 2901 2902
			*svlan_tci = build_tci(pkt_dev->svlan_id,
					       pkt_dev->svlan_cfi,
					       pkt_dev->svlan_p);
F
Francesco Fondelli 已提交
2903
			svlan_encapsulated_proto = (__be16 *)skb_put(skb, sizeof(__be16));
2904
			*svlan_encapsulated_proto = htons(ETH_P_8021Q);
F
Francesco Fondelli 已提交
2905 2906
		}
		vlan_tci = (__be16 *)skb_put(skb, sizeof(__be16));
A
Al Viro 已提交
2907 2908 2909
		*vlan_tci = build_tci(pkt_dev->vlan_id,
				      pkt_dev->vlan_cfi,
				      pkt_dev->vlan_p);
F
Francesco Fondelli 已提交
2910
		vlan_encapsulated_proto = (__be16 *)skb_put(skb, sizeof(__be16));
2911
		*vlan_encapsulated_proto = htons(ETH_P_IPV6);
F
Francesco Fondelli 已提交
2912 2913
	}

2914
	skb->network_header = skb->tail;
2915
	skb->transport_header = skb->network_header + sizeof(struct ipv6hdr);
2916
	skb_put(skb, sizeof(struct ipv6hdr) + sizeof(struct udphdr));
2917
	skb_set_queue_mapping(skb, queue_map);
J
John Fastabend 已提交
2918
	skb->priority = pkt_dev->skb_priority;
2919 2920
	iph = ipv6_hdr(skb);
	udph = udp_hdr(skb);
L
Linus Torvalds 已提交
2921 2922

	memcpy(eth, pkt_dev->hh, 12);
2923
	*(__be16 *) &eth[12] = protocol;
2924

2925 2926 2927
	/* Eth + IPh + UDPh + mpls */
	datalen = pkt_dev->cur_pkt_size - 14 -
		  sizeof(struct ipv6hdr) - sizeof(struct udphdr) -
2928
		  pkt_dev->pkt_overhead;
L
Linus Torvalds 已提交
2929

L
Luiz Capitulino 已提交
2930
	if (datalen < sizeof(struct pktgen_hdr)) {
L
Linus Torvalds 已提交
2931
		datalen = sizeof(struct pktgen_hdr);
2932
		net_info_ratelimited("increased datalen to %d\n", datalen);
L
Linus Torvalds 已提交
2933 2934 2935 2936
	}

	udph->source = htons(pkt_dev->cur_udp_src);
	udph->dest = htons(pkt_dev->cur_udp_dst);
L
Luiz Capitulino 已提交
2937 2938
	udph->len = htons(datalen + sizeof(struct udphdr));
	udph->check = 0;	/* No checksum */
L
Linus Torvalds 已提交
2939

2940
	*(__be32 *) iph = htonl(0x60000000);	/* Version + flow */
L
Linus Torvalds 已提交
2941

F
Francesco Fondelli 已提交
2942 2943
	if (pkt_dev->traffic_class) {
		/* Version + traffic class + flow (0) */
A
Al Viro 已提交
2944
		*(__be32 *)iph |= htonl(0x60000000 | (pkt_dev->traffic_class << 20));
F
Francesco Fondelli 已提交
2945 2946
	}

L
Linus Torvalds 已提交
2947 2948 2949 2950 2951
	iph->hop_limit = 32;

	iph->payload_len = htons(sizeof(struct udphdr) + datalen);
	iph->nexthdr = IPPROTO_UDP;

A
Alexey Dobriyan 已提交
2952 2953
	iph->daddr = pkt_dev->cur_in6_daddr;
	iph->saddr = pkt_dev->cur_in6_saddr;
L
Linus Torvalds 已提交
2954

2955
	skb->mac_header = (skb->network_header - ETH_HLEN -
2956
			   pkt_dev->pkt_overhead);
2957
	skb->protocol = protocol;
L
Linus Torvalds 已提交
2958 2959 2960
	skb->dev = odev;
	skb->pkt_type = PACKET_HOST;

E
Eric Dumazet 已提交
2961
	pktgen_finalize_skb(pkt_dev, skb, datalen);
L
Luiz Capitulino 已提交
2962

L
Linus Torvalds 已提交
2963 2964 2965
	return skb;
}

S
Stephen Hemminger 已提交
2966 2967
static struct sk_buff *fill_packet(struct net_device *odev,
				   struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
2968
{
L
Luiz Capitulino 已提交
2969
	if (pkt_dev->flags & F_IPV6)
L
Linus Torvalds 已提交
2970 2971 2972 2973 2974
		return fill_packet_ipv6(odev, pkt_dev);
	else
		return fill_packet_ipv4(odev, pkt_dev);
}

L
Luiz Capitulino 已提交
2975
static void pktgen_clear_counters(struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
2976
{
L
Luiz Capitulino 已提交
2977 2978
	pkt_dev->seq_num = 1;
	pkt_dev->idle_acc = 0;
L
Linus Torvalds 已提交
2979
	pkt_dev->sofar = 0;
L
Luiz Capitulino 已提交
2980 2981
	pkt_dev->tx_bytes = 0;
	pkt_dev->errors = 0;
L
Linus Torvalds 已提交
2982 2983 2984 2985 2986 2987
}

/* Set up structure for sending pkts, clear counters */

static void pktgen_run(struct pktgen_thread *t)
{
2988
	struct pktgen_dev *pkt_dev;
L
Linus Torvalds 已提交
2989 2990
	int started = 0;

J
Joe Perches 已提交
2991
	func_enter();
L
Linus Torvalds 已提交
2992 2993

	if_lock(t);
2994
	list_for_each_entry(pkt_dev, &t->if_list, list) {
L
Linus Torvalds 已提交
2995 2996 2997 2998 2999 3000

		/*
		 * setup odev and create initial packet.
		 */
		pktgen_setup_inject(pkt_dev);

L
Luiz Capitulino 已提交
3001
		if (pkt_dev->odev) {
L
Linus Torvalds 已提交
3002
			pktgen_clear_counters(pkt_dev);
L
Luiz Capitulino 已提交
3003
			pkt_dev->running = 1;	/* Cranke yeself! */
L
Linus Torvalds 已提交
3004
			pkt_dev->skb = NULL;
3005 3006 3007
			pkt_dev->started_at =
				pkt_dev->next_tx = ktime_now();

3008
			set_pkt_overhead(pkt_dev);
L
Luiz Capitulino 已提交
3009

L
Linus Torvalds 已提交
3010 3011
			strcpy(pkt_dev->result, "Starting");
			started++;
L
Luiz Capitulino 已提交
3012
		} else
L
Linus Torvalds 已提交
3013 3014 3015
			strcpy(pkt_dev->result, "Error starting");
	}
	if_unlock(t);
L
Luiz Capitulino 已提交
3016 3017
	if (started)
		t->control &= ~(T_STOP);
L
Linus Torvalds 已提交
3018 3019 3020 3021
}

static void pktgen_stop_all_threads_ifs(void)
{
3022
	struct pktgen_thread *t;
L
Linus Torvalds 已提交
3023

J
Joe Perches 已提交
3024
	func_enter();
L
Linus Torvalds 已提交
3025

3026
	mutex_lock(&pktgen_thread_lock);
3027 3028

	list_for_each_entry(t, &pktgen_threads, th_list)
3029
		t->control |= T_STOP;
3030

3031
	mutex_unlock(&pktgen_thread_lock);
L
Linus Torvalds 已提交
3032 3033
}

S
Stephen Hemminger 已提交
3034
static int thread_is_running(const struct pktgen_thread *t)
L
Linus Torvalds 已提交
3035
{
S
Stephen Hemminger 已提交
3036
	const struct pktgen_dev *pkt_dev;
L
Linus Torvalds 已提交
3037

3038
	list_for_each_entry(pkt_dev, &t->if_list, list)
S
Stephen Hemminger 已提交
3039 3040 3041
		if (pkt_dev->running)
			return 1;
	return 0;
L
Linus Torvalds 已提交
3042 3043
}

L
Luiz Capitulino 已提交
3044
static int pktgen_wait_thread_run(struct pktgen_thread *t)
L
Linus Torvalds 已提交
3045
{
L
Luiz Capitulino 已提交
3046
	if_lock(t);
L
Linus Torvalds 已提交
3047

L
Luiz Capitulino 已提交
3048
	while (thread_is_running(t)) {
L
Linus Torvalds 已提交
3049

L
Luiz Capitulino 已提交
3050
		if_unlock(t);
L
Linus Torvalds 已提交
3051

L
Luiz Capitulino 已提交
3052
		msleep_interruptible(100);
L
Linus Torvalds 已提交
3053

L
Luiz Capitulino 已提交
3054 3055 3056 3057 3058 3059 3060 3061
		if (signal_pending(current))
			goto signal;
		if_lock(t);
	}
	if_unlock(t);
	return 1;
signal:
	return 0;
L
Linus Torvalds 已提交
3062 3063 3064 3065
}

static int pktgen_wait_all_threads_run(void)
{
3066
	struct pktgen_thread *t;
L
Linus Torvalds 已提交
3067
	int sig = 1;
L
Luiz Capitulino 已提交
3068

3069
	mutex_lock(&pktgen_thread_lock);
3070 3071

	list_for_each_entry(t, &pktgen_threads, th_list) {
L
Linus Torvalds 已提交
3072
		sig = pktgen_wait_thread_run(t);
L
Luiz Capitulino 已提交
3073 3074
		if (sig == 0)
			break;
L
Linus Torvalds 已提交
3075
	}
3076 3077 3078

	if (sig == 0)
		list_for_each_entry(t, &pktgen_threads, th_list)
L
Linus Torvalds 已提交
3079
			t->control |= (T_STOP);
3080

3081
	mutex_unlock(&pktgen_thread_lock);
L
Linus Torvalds 已提交
3082 3083 3084 3085 3086
	return sig;
}

static void pktgen_run_all_threads(void)
{
3087
	struct pktgen_thread *t;
L
Linus Torvalds 已提交
3088

J
Joe Perches 已提交
3089
	func_enter();
L
Linus Torvalds 已提交
3090

3091
	mutex_lock(&pktgen_thread_lock);
L
Linus Torvalds 已提交
3092

3093
	list_for_each_entry(t, &pktgen_threads, th_list)
L
Linus Torvalds 已提交
3094
		t->control |= (T_RUN);
3095

3096
	mutex_unlock(&pktgen_thread_lock);
L
Linus Torvalds 已提交
3097

3098 3099
	/* Propagate thread->control  */
	schedule_timeout_interruptible(msecs_to_jiffies(125));
L
Luiz Capitulino 已提交
3100

L
Linus Torvalds 已提交
3101 3102 3103
	pktgen_wait_all_threads_run();
}

3104 3105 3106 3107
static void pktgen_reset_all_threads(void)
{
	struct pktgen_thread *t;

J
Joe Perches 已提交
3108
	func_enter();
3109 3110 3111 3112 3113 3114 3115 3116

	mutex_lock(&pktgen_thread_lock);

	list_for_each_entry(t, &pktgen_threads, th_list)
		t->control |= (T_REMDEVALL);

	mutex_unlock(&pktgen_thread_lock);

3117 3118
	/* Propagate thread->control  */
	schedule_timeout_interruptible(msecs_to_jiffies(125));
3119 3120 3121 3122

	pktgen_wait_all_threads_run();
}

L
Linus Torvalds 已提交
3123 3124
static void show_results(struct pktgen_dev *pkt_dev, int nr_frags)
{
3125
	__u64 bps, mbps, pps;
L
Luiz Capitulino 已提交
3126
	char *p = pkt_dev->result;
3127 3128 3129 3130
	ktime_t elapsed = ktime_sub(pkt_dev->stopped_at,
				    pkt_dev->started_at);
	ktime_t idle = ns_to_ktime(pkt_dev->idle_acc);

3131
	p += sprintf(p, "OK: %llu(c%llu+d%llu) usec, %llu (%dbyte,%dfrags)\n",
3132 3133 3134
		     (unsigned long long)ktime_to_us(elapsed),
		     (unsigned long long)ktime_to_us(ktime_sub(elapsed, idle)),
		     (unsigned long long)ktime_to_us(idle),
L
Luiz Capitulino 已提交
3135 3136 3137
		     (unsigned long long)pkt_dev->sofar,
		     pkt_dev->cur_pkt_size, nr_frags);

3138 3139
	pps = div64_u64(pkt_dev->sofar * NSEC_PER_SEC,
			ktime_to_ns(elapsed));
L
Luiz Capitulino 已提交
3140 3141 3142 3143 3144 3145 3146 3147 3148 3149

	bps = pps * 8 * pkt_dev->cur_pkt_size;

	mbps = bps;
	do_div(mbps, 1000000);
	p += sprintf(p, "  %llupps %lluMb/sec (%llubps) errors: %llu",
		     (unsigned long long)pps,
		     (unsigned long long)mbps,
		     (unsigned long long)bps,
		     (unsigned long long)pkt_dev->errors);
L
Linus Torvalds 已提交
3150 3151 3152
}

/* Set stopped-at timer, remove from running list, do counters & statistics */
L
Luiz Capitulino 已提交
3153
static int pktgen_stop_device(struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
3154
{
L
Luiz Capitulino 已提交
3155
	int nr_frags = pkt_dev->skb ? skb_shinfo(pkt_dev->skb)->nr_frags : -1;
3156

L
Luiz Capitulino 已提交
3157
	if (!pkt_dev->running) {
J
Joe Perches 已提交
3158 3159
		pr_warning("interface: %s is already stopped\n",
			   pkt_dev->odevname);
L
Luiz Capitulino 已提交
3160 3161
		return -EINVAL;
	}
L
Linus Torvalds 已提交
3162

S
Stephen Hemminger 已提交
3163 3164
	kfree_skb(pkt_dev->skb);
	pkt_dev->skb = NULL;
3165
	pkt_dev->stopped_at = ktime_now();
L
Luiz Capitulino 已提交
3166
	pkt_dev->running = 0;
L
Linus Torvalds 已提交
3167

3168
	show_results(pkt_dev, nr_frags);
L
Linus Torvalds 已提交
3169

L
Luiz Capitulino 已提交
3170
	return 0;
L
Linus Torvalds 已提交
3171 3172
}

L
Luiz Capitulino 已提交
3173
static struct pktgen_dev *next_to_run(struct pktgen_thread *t)
L
Linus Torvalds 已提交
3174
{
3175
	struct pktgen_dev *pkt_dev, *best = NULL;
L
Luiz Capitulino 已提交
3176

L
Linus Torvalds 已提交
3177 3178
	if_lock(t);

3179 3180
	list_for_each_entry(pkt_dev, &t->if_list, list) {
		if (!pkt_dev->running)
L
Luiz Capitulino 已提交
3181 3182
			continue;
		if (best == NULL)
3183
			best = pkt_dev;
3184
		else if (ktime_lt(pkt_dev->next_tx, best->next_tx))
3185
			best = pkt_dev;
L
Linus Torvalds 已提交
3186 3187
	}
	if_unlock(t);
L
Luiz Capitulino 已提交
3188
	return best;
L
Linus Torvalds 已提交
3189 3190
}

L
Luiz Capitulino 已提交
3191 3192
static void pktgen_stop(struct pktgen_thread *t)
{
3193
	struct pktgen_dev *pkt_dev;
L
Linus Torvalds 已提交
3194

J
Joe Perches 已提交
3195
	func_enter();
L
Linus Torvalds 已提交
3196

L
Luiz Capitulino 已提交
3197
	if_lock(t);
L
Linus Torvalds 已提交
3198

3199 3200
	list_for_each_entry(pkt_dev, &t->if_list, list) {
		pktgen_stop_device(pkt_dev);
3201
	}
L
Linus Torvalds 已提交
3202

L
Luiz Capitulino 已提交
3203
	if_unlock(t);
L
Linus Torvalds 已提交
3204 3205
}

3206 3207 3208 3209 3210 3211
/*
 * one of our devices needs to be removed - find it
 * and remove it
 */
static void pktgen_rem_one_if(struct pktgen_thread *t)
{
3212 3213
	struct list_head *q, *n;
	struct pktgen_dev *cur;
3214

J
Joe Perches 已提交
3215
	func_enter();
3216 3217 3218

	if_lock(t);

3219 3220
	list_for_each_safe(q, n, &t->if_list) {
		cur = list_entry(q, struct pktgen_dev, list);
3221

L
Luiz Capitulino 已提交
3222 3223
		if (!cur->removal_mark)
			continue;
3224

3225
		kfree_skb(cur->skb);
3226 3227 3228 3229 3230 3231 3232 3233 3234 3235
		cur->skb = NULL;

		pktgen_remove_device(t, cur);

		break;
	}

	if_unlock(t);
}

L
Luiz Capitulino 已提交
3236
static void pktgen_rem_all_ifs(struct pktgen_thread *t)
L
Linus Torvalds 已提交
3237
{
3238 3239
	struct list_head *q, *n;
	struct pktgen_dev *cur;
L
Luiz Capitulino 已提交
3240

J
Joe Perches 已提交
3241 3242
	func_enter();

L
Luiz Capitulino 已提交
3243
	/* Remove all devices, free mem */
3244

L
Luiz Capitulino 已提交
3245
	if_lock(t);
L
Linus Torvalds 已提交
3246

3247 3248
	list_for_each_safe(q, n, &t->if_list) {
		cur = list_entry(q, struct pktgen_dev, list);
3249

3250
		kfree_skb(cur->skb);
3251 3252
		cur->skb = NULL;

L
Linus Torvalds 已提交
3253 3254 3255
		pktgen_remove_device(t, cur);
	}

L
Luiz Capitulino 已提交
3256
	if_unlock(t);
L
Linus Torvalds 已提交
3257 3258
}

L
Luiz Capitulino 已提交
3259
static void pktgen_rem_thread(struct pktgen_thread *t)
L
Linus Torvalds 已提交
3260
{
L
Luiz Capitulino 已提交
3261
	/* Remove from the thread list */
L
Linus Torvalds 已提交
3262

3263
	remove_proc_entry(t->tsk->comm, pg_proc_dir);
L
Linus Torvalds 已提交
3264 3265 3266

}

3267
static void pktgen_resched(struct pktgen_dev *pkt_dev)
3268
{
3269
	ktime_t idle_start = ktime_now();
3270 3271 3272
	schedule();
	pkt_dev->idle_acc += ktime_to_ns(ktime_sub(ktime_now(), idle_start));
}
3273

3274 3275 3276
static void pktgen_wait_for_skb(struct pktgen_dev *pkt_dev)
{
	ktime_t idle_start = ktime_now();
3277

3278 3279 3280 3281 3282 3283 3284 3285 3286
	while (atomic_read(&(pkt_dev->skb->users)) != 1) {
		if (signal_pending(current))
			break;

		if (need_resched())
			pktgen_resched(pkt_dev);
		else
			cpu_relax();
	}
3287
	pkt_dev->idle_acc += ktime_to_ns(ktime_sub(ktime_now(), idle_start));
3288 3289
}

S
Stephen Hemminger 已提交
3290
static void pktgen_xmit(struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
3291
{
3292
	struct net_device *odev = pkt_dev->odev;
3293
	netdev_tx_t (*xmit)(struct sk_buff *, struct net_device *)
3294
		= odev->netdev_ops->ndo_start_xmit;
3295 3296
	struct netdev_queue *txq;
	u16 queue_map;
L
Linus Torvalds 已提交
3297 3298
	int ret;

3299 3300 3301 3302
	/* If device is offline, then don't send */
	if (unlikely(!netif_running(odev) || !netif_carrier_ok(odev))) {
		pktgen_stop_device(pkt_dev);
		return;
3303 3304
	}

3305 3306 3307 3308 3309
	/* This is max DELAY, this has special meaning of
	 * "never transmit"
	 */
	if (unlikely(pkt_dev->delay == ULLONG_MAX)) {
		pkt_dev->next_tx = ktime_add_ns(ktime_now(), ULONG_MAX);
3310
		return;
L
Linus Torvalds 已提交
3311
	}
L
Luiz Capitulino 已提交
3312

3313
	/* If no skb or clone count exhausted then get new one */
3314 3315 3316 3317 3318 3319 3320
	if (!pkt_dev->skb || (pkt_dev->last_ok &&
			      ++pkt_dev->clone_count >= pkt_dev->clone_skb)) {
		/* build a new pkt */
		kfree_skb(pkt_dev->skb);

		pkt_dev->skb = fill_packet(odev, pkt_dev);
		if (pkt_dev->skb == NULL) {
J
Joe Perches 已提交
3321
			pr_err("ERROR: couldn't allocate skb in fill_packet\n");
3322 3323 3324
			schedule();
			pkt_dev->clone_count--;	/* back out increment, OOM */
			return;
L
Linus Torvalds 已提交
3325
		}
3326
		pkt_dev->last_pkt_size = pkt_dev->skb->len;
3327 3328
		pkt_dev->allocated_skbs++;
		pkt_dev->clone_count = 0;	/* reset counter */
L
Linus Torvalds 已提交
3329
	}
3330

3331 3332 3333
	if (pkt_dev->delay && pkt_dev->last_ok)
		spin(pkt_dev, pkt_dev->next_tx);

3334 3335 3336 3337
	queue_map = skb_get_queue_mapping(pkt_dev->skb);
	txq = netdev_get_tx_queue(odev, queue_map);

	__netif_tx_lock_bh(txq);
3338

3339
	if (unlikely(netif_xmit_frozen_or_stopped(txq))) {
3340
		ret = NETDEV_TX_BUSY;
3341 3342 3343 3344 3345
		pkt_dev->last_ok = 0;
		goto unlock;
	}
	atomic_inc(&(pkt_dev->skb->users));
	ret = (*xmit)(pkt_dev->skb, odev);
3346 3347 3348 3349 3350 3351 3352

	switch (ret) {
	case NETDEV_TX_OK:
		txq_trans_update(txq);
		pkt_dev->last_ok = 1;
		pkt_dev->sofar++;
		pkt_dev->seq_num++;
3353
		pkt_dev->tx_bytes += pkt_dev->last_pkt_size;
3354
		break;
3355 3356 3357 3358 3359 3360
	case NET_XMIT_DROP:
	case NET_XMIT_CN:
	case NET_XMIT_POLICED:
		/* skb has been consumed */
		pkt_dev->errors++;
		break;
3361
	default: /* Drivers are not supposed to return other values! */
3362 3363
		net_info_ratelimited("%s xmit error: %d\n",
				     pkt_dev->odevname, ret);
3364 3365 3366 3367 3368 3369 3370
		pkt_dev->errors++;
		/* fallthru */
	case NETDEV_TX_LOCKED:
	case NETDEV_TX_BUSY:
		/* Retry it next time */
		atomic_dec(&(pkt_dev->skb->users));
		pkt_dev->last_ok = 0;
L
Luiz Capitulino 已提交
3371
	}
3372
unlock:
3373
	__netif_tx_unlock_bh(txq);
L
Luiz Capitulino 已提交
3374

L
Linus Torvalds 已提交
3375 3376
	/* If pkt_dev->count is zero, then run forever */
	if ((pkt_dev->count != 0) && (pkt_dev->sofar >= pkt_dev->count)) {
3377
		pktgen_wait_for_skb(pkt_dev);
L
Luiz Capitulino 已提交
3378

L
Linus Torvalds 已提交
3379 3380
		/* Done with this */
		pktgen_stop_device(pkt_dev);
L
Luiz Capitulino 已提交
3381 3382
	}
}
L
Linus Torvalds 已提交
3383

3384
/*
L
Linus Torvalds 已提交
3385 3386 3387
 * Main loop of the thread goes here
 */

3388
static int pktgen_thread_worker(void *arg)
L
Linus Torvalds 已提交
3389 3390
{
	DEFINE_WAIT(wait);
3391
	struct pktgen_thread *t = arg;
L
Luiz Capitulino 已提交
3392
	struct pktgen_dev *pkt_dev = NULL;
L
Linus Torvalds 已提交
3393 3394
	int cpu = t->cpu;

3395
	BUG_ON(smp_processor_id() != cpu);
L
Linus Torvalds 已提交
3396 3397

	init_waitqueue_head(&t->queue);
3398
	complete(&t->start_done);
L
Linus Torvalds 已提交
3399

J
Joe Perches 已提交
3400
	pr_debug("starting pktgen/%d:  pid=%d\n", cpu, task_pid_nr(current));
L
Linus Torvalds 已提交
3401

3402
	set_current_state(TASK_INTERRUPTIBLE);
L
Luiz Capitulino 已提交
3403

3404 3405
	set_freezable();

3406 3407
	while (!kthread_should_stop()) {
		pkt_dev = next_to_run(t);
L
Linus Torvalds 已提交
3408

3409
		if (unlikely(!pkt_dev && t->control == 0)) {
E
Eric Dumazet 已提交
3410 3411
			if (pktgen_exiting)
				break;
3412 3413 3414
			wait_event_interruptible_timeout(t->queue,
							 t->control != 0,
							 HZ/10);
R
Rafael J. Wysocki 已提交
3415
			try_to_freeze();
3416
			continue;
3417
		}
L
Linus Torvalds 已提交
3418

3419
		__set_current_state(TASK_RUNNING);
L
Luiz Capitulino 已提交
3420

3421
		if (likely(pkt_dev)) {
L
Linus Torvalds 已提交
3422 3423
			pktgen_xmit(pkt_dev);

3424 3425 3426 3427 3428 3429
			if (need_resched())
				pktgen_resched(pkt_dev);
			else
				cpu_relax();
		}

L
Luiz Capitulino 已提交
3430
		if (t->control & T_STOP) {
L
Linus Torvalds 已提交
3431 3432 3433 3434
			pktgen_stop(t);
			t->control &= ~(T_STOP);
		}

L
Luiz Capitulino 已提交
3435
		if (t->control & T_RUN) {
L
Linus Torvalds 已提交
3436 3437 3438 3439
			pktgen_run(t);
			t->control &= ~(T_RUN);
		}

L
Luiz Capitulino 已提交
3440
		if (t->control & T_REMDEVALL) {
L
Linus Torvalds 已提交
3441
			pktgen_rem_all_ifs(t);
3442 3443 3444
			t->control &= ~(T_REMDEVALL);
		}

L
Luiz Capitulino 已提交
3445
		if (t->control & T_REMDEV) {
3446
			pktgen_rem_one_if(t);
L
Linus Torvalds 已提交
3447 3448 3449
			t->control &= ~(T_REMDEV);
		}

A
Andrew Morton 已提交
3450 3451
		try_to_freeze();

3452
		set_current_state(TASK_INTERRUPTIBLE);
L
Luiz Capitulino 已提交
3453
	}
L
Linus Torvalds 已提交
3454

J
Joe Perches 已提交
3455
	pr_debug("%s stopping all device\n", t->tsk->comm);
L
Luiz Capitulino 已提交
3456
	pktgen_stop(t);
L
Linus Torvalds 已提交
3457

J
Joe Perches 已提交
3458
	pr_debug("%s removing all device\n", t->tsk->comm);
L
Luiz Capitulino 已提交
3459
	pktgen_rem_all_ifs(t);
L
Linus Torvalds 已提交
3460

J
Joe Perches 已提交
3461
	pr_debug("%s removing thread\n", t->tsk->comm);
L
Luiz Capitulino 已提交
3462
	pktgen_rem_thread(t);
3463

E
Eric Dumazet 已提交
3464 3465 3466 3467 3468 3469 3470
	/* Wait for kthread_stop */
	while (!kthread_should_stop()) {
		set_current_state(TASK_INTERRUPTIBLE);
		schedule();
	}
	__set_current_state(TASK_RUNNING);

3471
	return 0;
L
Linus Torvalds 已提交
3472 3473
}

L
Luiz Capitulino 已提交
3474
static struct pktgen_dev *pktgen_find_dev(struct pktgen_thread *t,
E
Eric Dumazet 已提交
3475
					  const char *ifname, bool exact)
L
Linus Torvalds 已提交
3476
{
3477
	struct pktgen_dev *p, *pkt_dev = NULL;
E
Eric Dumazet 已提交
3478
	size_t len = strlen(ifname);
L
Luiz Capitulino 已提交
3479

E
Eric Dumazet 已提交
3480
	if_lock(t);
3481
	list_for_each_entry(p, &t->if_list, list)
E
Eric Dumazet 已提交
3482 3483 3484 3485 3486
		if (strncmp(p->odevname, ifname, len) == 0) {
			if (p->odevname[len]) {
				if (exact || p->odevname[len] != '@')
					continue;
			}
3487
			pkt_dev = p;
L
Luiz Capitulino 已提交
3488 3489 3490 3491
			break;
		}

	if_unlock(t);
J
Joe Perches 已提交
3492
	pr_debug("find_dev(%s) returning %p\n", ifname, pkt_dev);
L
Luiz Capitulino 已提交
3493
	return pkt_dev;
L
Linus Torvalds 已提交
3494 3495
}

3496 3497
/*
 * Adds a dev at front of if_list.
L
Linus Torvalds 已提交
3498 3499
 */

L
Luiz Capitulino 已提交
3500 3501
static int add_dev_to_thread(struct pktgen_thread *t,
			     struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
3502 3503
{
	int rv = 0;
L
Luiz Capitulino 已提交
3504 3505 3506 3507

	if_lock(t);

	if (pkt_dev->pg_thread) {
J
Joe Perches 已提交
3508
		pr_err("ERROR: already assigned to a thread\n");
L
Luiz Capitulino 已提交
3509 3510 3511
		rv = -EBUSY;
		goto out;
	}
3512 3513

	list_add(&pkt_dev->list, &t->if_list);
L
Luiz Capitulino 已提交
3514
	pkt_dev->pg_thread = t;
L
Linus Torvalds 已提交
3515 3516
	pkt_dev->running = 0;

L
Luiz Capitulino 已提交
3517 3518 3519
out:
	if_unlock(t);
	return rv;
L
Linus Torvalds 已提交
3520 3521 3522 3523
}

/* Called under thread lock */

L
Luiz Capitulino 已提交
3524
static int pktgen_add_device(struct pktgen_thread *t, const char *ifname)
L
Linus Torvalds 已提交
3525
{
L
Luiz Capitulino 已提交
3526
	struct pktgen_dev *pkt_dev;
3527
	int err;
E
Eric Dumazet 已提交
3528
	int node = cpu_to_node(t->cpu);
L
Luiz Capitulino 已提交
3529

L
Linus Torvalds 已提交
3530 3531
	/* We don't allow a device to be on several threads */

3532 3533
	pkt_dev = __pktgen_NN_threads(ifname, FIND);
	if (pkt_dev) {
J
Joe Perches 已提交
3534
		pr_err("ERROR: interface already used\n");
L
Luiz Capitulino 已提交
3535 3536
		return -EBUSY;
	}
3537

E
Eric Dumazet 已提交
3538
	pkt_dev = kzalloc_node(sizeof(struct pktgen_dev), GFP_KERNEL, node);
3539 3540 3541
	if (!pkt_dev)
		return -ENOMEM;

E
Eric Dumazet 已提交
3542
	strcpy(pkt_dev->odevname, ifname);
3543
	pkt_dev->flows = vzalloc_node(MAX_CFLOWS * sizeof(struct flow_state),
E
Eric Dumazet 已提交
3544
				      node);
3545 3546 3547 3548 3549
	if (pkt_dev->flows == NULL) {
		kfree(pkt_dev);
		return -ENOMEM;
	}

3550
	pkt_dev->removal_mark = 0;
3551 3552 3553
	pkt_dev->min_pkt_size = ETH_ZLEN;
	pkt_dev->max_pkt_size = ETH_ZLEN;
	pkt_dev->nfrags = 0;
3554
	pkt_dev->delay = pg_delay_d;
3555 3556
	pkt_dev->count = pg_count_d;
	pkt_dev->sofar = 0;
L
Luiz Capitulino 已提交
3557
	pkt_dev->udp_src_min = 9;	/* sink port */
3558 3559 3560
	pkt_dev->udp_src_max = 9;
	pkt_dev->udp_dst_min = 9;
	pkt_dev->udp_dst_max = 9;
F
Francesco Fondelli 已提交
3561 3562 3563 3564 3565 3566
	pkt_dev->vlan_p = 0;
	pkt_dev->vlan_cfi = 0;
	pkt_dev->vlan_id = 0xffff;
	pkt_dev->svlan_p = 0;
	pkt_dev->svlan_cfi = 0;
	pkt_dev->svlan_id = 0xffff;
R
Robert Olsson 已提交
3567
	pkt_dev->node = -1;
F
Francesco Fondelli 已提交
3568

3569 3570 3571
	err = pktgen_setup_dev(pkt_dev, ifname);
	if (err)
		goto out1;
3572 3573
	if (pkt_dev->odev->priv_flags & IFF_TX_SKB_SHARING)
		pkt_dev->clone_skb = pg_clone_skb_d;
3574

3575 3576
	pkt_dev->entry = proc_create_data(ifname, 0600, pg_proc_dir,
					  &pktgen_if_fops, pkt_dev);
3577
	if (!pkt_dev->entry) {
J
Joe Perches 已提交
3578
		pr_err("cannot create %s/%s procfs entry\n",
3579
		       PG_PROC_DIR, ifname);
3580 3581
		err = -EINVAL;
		goto out2;
3582
	}
J
Jamal Hadi Salim 已提交
3583 3584 3585 3586
#ifdef CONFIG_XFRM
	pkt_dev->ipsmode = XFRM_MODE_TRANSPORT;
	pkt_dev->ipsproto = IPPROTO_ESP;
#endif
3587 3588

	return add_dev_to_thread(t, pkt_dev);
3589 3590 3591
out2:
	dev_put(pkt_dev->odev);
out1:
J
Jamal Hadi Salim 已提交
3592 3593 3594
#ifdef CONFIG_XFRM
	free_SAs(pkt_dev);
#endif
F
Figo.zhang 已提交
3595
	vfree(pkt_dev->flows);
3596 3597
	kfree(pkt_dev);
	return err;
L
Linus Torvalds 已提交
3598 3599
}

3600
static int __init pktgen_create_thread(int cpu)
L
Linus Torvalds 已提交
3601
{
3602
	struct pktgen_thread *t;
3603
	struct proc_dir_entry *pe;
3604
	struct task_struct *p;
L
Luiz Capitulino 已提交
3605

E
Eric Dumazet 已提交
3606 3607
	t = kzalloc_node(sizeof(struct pktgen_thread), GFP_KERNEL,
			 cpu_to_node(cpu));
L
Luiz Capitulino 已提交
3608
	if (!t) {
J
Joe Perches 已提交
3609
		pr_err("ERROR: out of memory, can't create new thread\n");
L
Luiz Capitulino 已提交
3610 3611 3612 3613
		return -ENOMEM;
	}

	spin_lock_init(&t->if_lock);
L
Linus Torvalds 已提交
3614
	t->cpu = cpu;
L
Luiz Capitulino 已提交
3615

3616 3617 3618
	INIT_LIST_HEAD(&t->if_list);

	list_add_tail(&t->th_list, &pktgen_threads);
3619
	init_completion(&t->start_done);
3620

3621 3622 3623 3624
	p = kthread_create_on_node(pktgen_thread_worker,
				   t,
				   cpu_to_node(cpu),
				   "kpktgend_%d", cpu);
3625
	if (IS_ERR(p)) {
J
Joe Perches 已提交
3626
		pr_err("kernel_thread() failed for cpu %d\n", t->cpu);
3627 3628 3629 3630 3631 3632 3633
		list_del(&t->th_list);
		kfree(t);
		return PTR_ERR(p);
	}
	kthread_bind(p, cpu);
	t->tsk = p;

3634 3635
	pe = proc_create_data(t->tsk->comm, 0600, pg_proc_dir,
			      &pktgen_thread_fops, t);
L
Luiz Capitulino 已提交
3636
	if (!pe) {
J
Joe Perches 已提交
3637
		pr_err("cannot create %s/%s procfs entry\n",
3638 3639 3640
		       PG_PROC_DIR, t->tsk->comm);
		kthread_stop(p);
		list_del(&t->th_list);
L
Luiz Capitulino 已提交
3641 3642 3643
		kfree(t);
		return -EINVAL;
	}
3644

3645
	wake_up_process(p);
3646
	wait_for_completion(&t->start_done);
L
Linus Torvalds 已提交
3647 3648 3649 3650

	return 0;
}

3651 3652
/*
 * Removes a device from the thread if_list.
L
Linus Torvalds 已提交
3653
 */
L
Luiz Capitulino 已提交
3654 3655
static void _rem_dev_from_if_list(struct pktgen_thread *t,
				  struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
3656
{
3657 3658
	struct list_head *q, *n;
	struct pktgen_dev *p;
L
Linus Torvalds 已提交
3659

3660 3661 3662 3663
	list_for_each_safe(q, n, &t->if_list) {
		p = list_entry(q, struct pktgen_dev, list);
		if (p == pkt_dev)
			list_del(&p->list);
L
Linus Torvalds 已提交
3664 3665 3666
	}
}

L
Luiz Capitulino 已提交
3667 3668
static int pktgen_remove_device(struct pktgen_thread *t,
				struct pktgen_dev *pkt_dev)
L
Linus Torvalds 已提交
3669 3670
{

J
Joe Perches 已提交
3671
	pr_debug("remove_device pkt_dev=%p\n", pkt_dev);
L
Linus Torvalds 已提交
3672

L
Luiz Capitulino 已提交
3673
	if (pkt_dev->running) {
J
Joe Perches 已提交
3674
		pr_warning("WARNING: trying to remove a running interface, stopping it now\n");
L
Luiz Capitulino 已提交
3675 3676 3677 3678
		pktgen_stop_device(pkt_dev);
	}

	/* Dis-associate from the interface */
L
Linus Torvalds 已提交
3679 3680 3681

	if (pkt_dev->odev) {
		dev_put(pkt_dev->odev);
L
Luiz Capitulino 已提交
3682 3683 3684
		pkt_dev->odev = NULL;
	}

L
Linus Torvalds 已提交
3685 3686 3687 3688
	/* And update the thread if_list */

	_rem_dev_from_if_list(t, pkt_dev);

3689 3690
	if (pkt_dev->entry)
		remove_proc_entry(pkt_dev->entry->name, pg_proc_dir);
L
Linus Torvalds 已提交
3691

J
Jamal Hadi Salim 已提交
3692 3693 3694
#ifdef CONFIG_XFRM
	free_SAs(pkt_dev);
#endif
F
Figo.zhang 已提交
3695
	vfree(pkt_dev->flows);
E
Eric Dumazet 已提交
3696 3697
	if (pkt_dev->page)
		put_page(pkt_dev->page);
L
Linus Torvalds 已提交
3698
	kfree(pkt_dev);
L
Luiz Capitulino 已提交
3699
	return 0;
L
Linus Torvalds 已提交
3700 3701
}

L
Luiz Capitulino 已提交
3702
static int __init pg_init(void)
L
Linus Torvalds 已提交
3703 3704
{
	int cpu;
3705
	struct proc_dir_entry *pe;
W
WANG Cong 已提交
3706
	int ret = 0;
L
Linus Torvalds 已提交
3707

J
Joe Perches 已提交
3708
	pr_info("%s", version);
L
Linus Torvalds 已提交
3709

3710
	pg_proc_dir = proc_mkdir(PG_PROC_DIR, init_net.proc_net);
3711 3712
	if (!pg_proc_dir)
		return -ENODEV;
L
Linus Torvalds 已提交
3713

3714
	pe = proc_create(PGCTRL, 0600, pg_proc_dir, &pktgen_fops);
L
Luiz Capitulino 已提交
3715
	if (pe == NULL) {
J
Joe Perches 已提交
3716
		pr_err("ERROR: cannot create %s procfs entry\n", PGCTRL);
W
WANG Cong 已提交
3717 3718
		ret = -EINVAL;
		goto remove_dir;
L
Luiz Capitulino 已提交
3719
	}
L
Linus Torvalds 已提交
3720 3721

	register_netdevice_notifier(&pktgen_notifier_block);
L
Luiz Capitulino 已提交
3722

J
John Hawkes 已提交
3723
	for_each_online_cpu(cpu) {
3724
		int err;
L
Linus Torvalds 已提交
3725

3726
		err = pktgen_create_thread(cpu);
3727
		if (err)
J
Joe Perches 已提交
3728 3729
			pr_warning("WARNING: Cannot create thread for cpu %d (%d)\n",
				   cpu, err);
L
Luiz Capitulino 已提交
3730
	}
3731 3732

	if (list_empty(&pktgen_threads)) {
J
Joe Perches 已提交
3733
		pr_err("ERROR: Initialization failed for all threads\n");
W
WANG Cong 已提交
3734 3735
		ret = -ENODEV;
		goto unregister;
3736 3737
	}

L
Luiz Capitulino 已提交
3738
	return 0;
W
WANG Cong 已提交
3739 3740 3741 3742 3743 3744 3745

 unregister:
	unregister_netdevice_notifier(&pktgen_notifier_block);
	remove_proc_entry(PGCTRL, pg_proc_dir);
 remove_dir:
	proc_net_remove(&init_net, PG_PROC_DIR);
	return ret;
L
Linus Torvalds 已提交
3746 3747 3748 3749
}

static void __exit pg_cleanup(void)
{
3750 3751
	struct pktgen_thread *t;
	struct list_head *q, *n;
3752
	LIST_HEAD(list);
L
Linus Torvalds 已提交
3753

L
Luiz Capitulino 已提交
3754
	/* Stop all interfaces & threads */
E
Eric Dumazet 已提交
3755
	pktgen_exiting = true;
L
Linus Torvalds 已提交
3756

3757
	mutex_lock(&pktgen_thread_lock);
3758
	list_splice_init(&pktgen_threads, &list);
3759 3760 3761
	mutex_unlock(&pktgen_thread_lock);

	list_for_each_safe(q, n, &list) {
3762
		t = list_entry(q, struct pktgen_thread, th_list);
3763
		list_del(&t->th_list);
3764 3765
		kthread_stop(t->tsk);
		kfree(t);
L
Luiz Capitulino 已提交
3766
	}
L
Linus Torvalds 已提交
3767

L
Luiz Capitulino 已提交
3768
	/* Un-register us from receiving netdevice events */
L
Linus Torvalds 已提交
3769 3770
	unregister_netdevice_notifier(&pktgen_notifier_block);

L
Luiz Capitulino 已提交
3771
	/* Clean up proc file system */
3772
	remove_proc_entry(PGCTRL, pg_proc_dir);
3773
	proc_net_remove(&init_net, PG_PROC_DIR);
L
Linus Torvalds 已提交
3774 3775 3776 3777 3778
}

module_init(pg_init);
module_exit(pg_cleanup);

S
Stephen Hemminger 已提交
3779
MODULE_AUTHOR("Robert Olsson <robert.olsson@its.uu.se>");
L
Linus Torvalds 已提交
3780 3781
MODULE_DESCRIPTION("Packet Generator tool");
MODULE_LICENSE("GPL");
S
Stephen Hemminger 已提交
3782
MODULE_VERSION(VERSION);
L
Linus Torvalds 已提交
3783
module_param(pg_count_d, int, 0);
S
Stephen Hemminger 已提交
3784
MODULE_PARM_DESC(pg_count_d, "Default number of packets to inject");
L
Linus Torvalds 已提交
3785
module_param(pg_delay_d, int, 0);
S
Stephen Hemminger 已提交
3786
MODULE_PARM_DESC(pg_delay_d, "Default delay between packets (nanoseconds)");
L
Linus Torvalds 已提交
3787
module_param(pg_clone_skb_d, int, 0);
S
Stephen Hemminger 已提交
3788
MODULE_PARM_DESC(pg_clone_skb_d, "Default number of copies of the same packet");
L
Linus Torvalds 已提交
3789
module_param(debug, int, 0);
S
Stephen Hemminger 已提交
3790
MODULE_PARM_DESC(debug, "Enable debugging of pktgen module");