main.c

#include <linux/version.h>
#include <linux/module.h>
#include "vq.h"
#include "rx.h"
#include "tx.h"
#include "stats.h"

#ifdef QDISC
int eyeq_qdisc_register(void);
void eyeq_qdisc_unregister(void);
#endif

MODULE_AUTHOR("Vimal <j.vimal@gmail.com>");
MODULE_DESCRIPTION("Perf Isolation");
MODULE_VERSION("1");
MODULE_LICENSE("GPL");

char *iso_param_dev;
MODULE_PARM_DESC(iso_param_dev, "Interface to operate perfiso.");
module_param(iso_param_dev, charp, 0);

/* This is the main device (should be in the root ns) to be used for
 * packet tx.  We need it irrespective of how packets are classified.
 * It will be used to directly transmit feedback packets, without any
 * routing.  Eventually, we may want a kernel UDP socket, but that's
 * for later.
 */
struct net_device *iso_netdev;
static int iso_init(void);
static void iso_exit(void);

void iso_rx_hook_exit(struct iso_rx_context *);
void iso_tx_hook_exit(struct iso_tx_context *);

int iso_exiting;
#ifndef QDISC
/* Current device's GSO size */
static int __prev__ISO_GSO_MAX_SIZE;
#endif

static int iso_init() {
	int i, ret = -1;
	iso_exiting = 0;

	if(iso_param_dev == NULL) {
		/*
		  printk(KERN_INFO "perfiso: need iso_param_dev, the interface to protect.\n");
		  goto err;
		*/
		iso_param_dev = "peth2\0";
	}

	/* trim */
	for(i = 0; i < 32 && iso_param_dev[i] != '\0'; i++) {
		if(iso_param_dev[i] == '\n') {
			iso_param_dev[i] = '\0';
			break;
		}
	}

	INIT_LIST_HEAD(&rxctx_list);
	INIT_LIST_HEAD(&txctx_list);

	if(iso_params_init())
		goto out;

	if(iso_stats_init())
		goto out_1;

#ifdef QDISC
	if (eyeq_qdisc_register())
		goto out_2;
#else
	rcu_read_lock();
	iso_netdev = dev_get_by_name(&init_net, iso_param_dev);
	rcu_read_unlock();

	if(iso_netdev == NULL) {
		printk(KERN_INFO "perfiso: device %s not found", iso_param_dev);
		goto out_3;
	}

	global_rxcontext.netdev = iso_netdev;
	global_txcontext.netdev = iso_netdev;

	if (iso_rx_init(&global_rxcontext))
		goto out_4;

	if (iso_tx_init(&global_txcontext))
		goto out_5;

	printk(KERN_INFO "perfiso: operating on %s (%p)\n",
	       iso_param_dev, iso_netdev);

	__prev__ISO_GSO_MAX_SIZE = iso_netdev->gso_max_size;
	netif_set_gso_max_size(iso_netdev, ISO_GSO_MAX_SIZE);
#endif

	ret = 0;
	goto out;

#ifndef QDISC
	/* Free up resources */
	iso_tx_exit(&global_txcontext);
out_5:
	iso_rx_exit(&global_rxcontext);
out_4:
	dev_put(iso_netdev);
out_3:
#else
	eyeq_qdisc_unregister();
out_2:
#endif
	iso_stats_exit();
out_1:
	iso_params_exit();
 out:
	return ret;
}

static void iso_exit() {
	iso_exiting = 1;
	mb();

	iso_stats_exit();
	iso_params_exit();
#ifdef QDISC
	eyeq_qdisc_unregister();
#else
	iso_tx_hook_exit(&global_txcontext);
	iso_rx_hook_exit(&global_rxcontext);
	iso_tx_exit(&global_txcontext);
	iso_rx_exit(&global_rxcontext);

	netif_set_gso_max_size(iso_netdev, __prev__ISO_GSO_MAX_SIZE);
	dev_put(iso_netdev);
#endif
	printk(KERN_INFO "perfiso: goodbye.\n");
}

module_init(iso_init);
module_exit(iso_exit);

/* Local Variables: */
/* indent-tabs-mode:t */
/* End: */