dsr-module.c

LINUX下的mesh路由协议dsr的源代码包

/* Copyright (C) Uppsala University
 *
 * This file is distributed under the terms of the GNU general Public
 * License (GPL), see the file LICENSE
 *
 * Author: Erik Nordström, <[e-mail]>
 */
#include <linux/config.h>
#include <linux/version.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/skbuff.h>
#include <net/protocol.h>
#include <linux/netdevice.h>
#include <linux/if_ether.h>
#include <linux/proc_fs.h>
#include <linux/socket.h>
#include <net/arp.h>
#include <net/ip.h>
#include <net/dst.h>
#include <net/neighbour.h>
#include <asm/uaccess.h>
#include <linux/netfilter_ipv4.h>
#ifdef KERNEL26
#include <linux/moduleparam.h>
#endif
#include <net/icmp.h>
#include <linux/ctype.h>

#include "dsr.h"
#include "dsr-dev.h"
#include "dsr-io.h"
#include "dsr-pkt.h"
#include "debug.h"
#include "neigh.h"
#include "dsr-rreq.h"
#include "maint-buf.h"
#include "send-buf.h"
#include "link-cache.h"

static char *ifname = NULL;
static char *mackill = NULL;

MODULE_AUTHOR("[e-mail]");
MODULE_DESCRIPTION("Dynamic Source Routing (DSR) protocol stack");
MODULE_LICENSE("GPL");

#ifdef KERNEL26
module_param(ifname, charp, 0);
module_param(mackill, charp, 0);
#else
MODULE_PARM(ifname, "s");
MODULE_PARM(mackill, "s");
#endif

#define CONFIG_PROC_NAME "dsr_config"

#define MAX_MACKILL 10

static unsigned char mackill_list[MAX_MACKILL][ETH_ALEN];
static int mackill_len = 0;

static char *confval_names[CONFVAL_TYPE_MAX] = { 
	"Seconds (s)",
	"Milliseconds (ms)",
	"Microseconds (us)",
	"Nanoseconds (ns)",
	"Quanta",
	"Binary",
	"Command"
};

/* Shamelessly stolen from LUNAR <[e-mail]> */
static int parse_mackill(void)
{
	char *pa[MAX_MACKILL], *cp;
	int i, j;		// , ia[ETH_ALEN];

	cp = mackill;
	while (cp && mackill_len < MAX_MACKILL) {
		pa[mackill_len] = strsep(&cp, ",");
		if (!pa[mackill_len])
			break;
		mackill_len++;
	}
	for (i = 0; i < mackill_len; i++) {
		// lnx kernel bug in 2.4.X: sscanf format "%x" does not work ....
		cp = pa[i];
		for (j = 0; j < ETH_ALEN; j++, cp++) {
			mackill_list[i][j] = 0;
			for (; isxdigit(*cp); cp++) {
				mackill_list[i][j] =
				    (mackill_list[i][j] << 4) |
				    (*cp <= '9' ? (*cp - '0') :
				     ((*cp & 0x07) + 9));
			}
			if (*cp && *cp != ':')
				break;
		}
		if (j != ETH_ALEN) {
			DEBUG("mackill: error in MAC addr %s\n", pa[i]);
			mackill_len--;
			return -1;
		}

		DEBUG("mackill +%s\n", print_eth(mackill_list[i]));
	}
	return 0;
}

int do_mackill(char *mac)
{
	int i;

	for (i = 0; i < mackill_len; i++) {
		if (memcmp(mac, mackill_list[i], ETH_ALEN) == 0)
			return 1;
	}
	return 0;
}

int dsr_ip_recv(struct sk_buff *skb)
{
	struct dsr_pkt *dp;
#ifdef DEBUG
	atomic_inc(&num_pkts);
#endif
	DEBUG("Received DSR packet\n");

	dp = dsr_pkt_alloc(skb);

	if (!dp) {
		DEBUG("Could not allocate DSR packet\n");
		kfree_skb(skb);
		return 0;
	}

	if (skb->pkt_type == PACKET_OTHERHOST) {
		dp->flags |= PKT_PROMISC_RECV;
	}
	if ((skb->len + (dp->nh.iph->ihl << 2)) < ntohs(dp->nh.iph->tot_len)) {
		DEBUG("Data to short! IP header len=%d tot_len=%d!\n", 
		      skb->len + (dp->nh.iph->ihl << 2), 
		      ntohs(dp->nh.iph->tot_len));
		dsr_pkt_free(dp);
		return 0;
	}

/* 	DEBUG("iph_len=%d iph_totlen=%d dsr_opts_len=%d data_len=%d\n", */
/* 	      (dp->nh.iph->ihl << 2), ntohs(dp->nh.iph->tot_len), */
/* 	      dsr_pkt_opts_len(dp), dp->payload_len); */

	/* Add mac address of previous hop to the arp table */
	dsr_recv(dp);

	return 0;
};

static void dsr_ip_recv_err(struct sk_buff *skb, u32 info)
{
	DEBUG("received error, info=%u\n", info);

	kfree_skb(skb);
}

static int dsr_config_proc_read(char *buffer, char **start, 
				off_t offset, int length,
				int *eof, void *data)
{
	int len = 0;
	int i;

	len += sprintf(buffer + len, 
		       "# %-23s %-6s %s\n", "Name", "Value", "Type");

	for (i = 0; i < CONFVAL_MAX; i++)
		len += sprintf(buffer + len, "%-25s %-6u %s\n",
			       confvals_def[i].name,
			       get_confval(i),
			       confval_names[confvals_def[i].type]);

	*start = buffer + offset;
	len -= offset;
	if (len > length)
		len = length;
	else if (len < 0)
		len = 0;
	return len;
}
static int dsr_config_proc_write(struct file *file, const char *buffer,
				 unsigned long count, void *data)
{
#define CMD_MAX_LEN 256
	char cmd[CMD_MAX_LEN];
	int i;

	memset(cmd, '\0', CMD_MAX_LEN);

	if (count > CMD_MAX_LEN)
		return -EINVAL;

	/* Don't read the '\n' or '\0' character... */
	if (copy_from_user(cmd, buffer, count))
		return -EFAULT;

	for (i = 0; i < CONFVAL_MAX; i++) {
		int n = strlen(confvals_def[i].name);

		if (strncmp(cmd, confvals_def[i].name, n) == 0) {
			char *from, *to;
			unsigned int val, val_prev;

			if (confvals_def[i].type == COMMAND) {
				if (i == FlushLinkCache)
					lc_flush();
				break;
			}

			if (strlen(cmd) - 2 <= n)
				continue;

			from = strstr(cmd, "=");
			from++;	/* Exclude '=' */

			val_prev = ConfVal(i);
			val = simple_strtol(from, &to, 10);

			if (confvals_def[i].type == BINARY)
				val = (val ? 1 : 0);

			set_confval(i, val);

			if (i == PromiscOperation && val_prev != val
			    && dsr_node) {
				if (val)
					DEBUG
					    ("Setting promiscuous operation\n");
				else
					DEBUG
					    ("Disabling promiscuous operation\n");

				dsr_node_lock(dsr_node);
				dev_set_promiscuity(dsr_node->slave_dev,
						    val ? 1 : -1);
				dsr_node_unlock(dsr_node);
			}
			if (i == RequestTableSize)
				rreq_tbl_set_max_len(val);

			if (i == RexmtBufferSize)
				maint_buf_set_max_len(val);

			if (i == SendBufferSize)
				send_buf_set_max_len(val);

			DEBUG("Setting %s to %d\n", confvals_def[i].name, val);
		}
	}
	return count;
}

/* This hook is used to do mac filtering or to receive promiscuously snooped
 * packets */
static unsigned int dsr_pre_routing_recv(unsigned int hooknum,
					 struct sk_buff **skb,
					 const struct net_device *in,
					 const struct net_device *out,
					 int (*okfn) (struct sk_buff *))
{
	if (in && in->ifindex == get_slave_dev_ifindex() &&
	    (*skb)->protocol == htons(ETH_P_IP)) {

		if (do_mackill((*skb)->mac.raw + ETH_ALEN))
			return NF_DROP;
	}
	return NF_ACCEPT;
}

/* This hook is the architecturally correct place to look at DSR packets that
 * are to be forwarded. This enables you to, for example, disable forwarding by
 * setting "/proc/sys/net/ipv4/conf/<eth*>/forwarding" to 0. */
static unsigned int dsr_ip_forward_recv(unsigned int hooknum,
					struct sk_buff **skb,
					const struct net_device *in,
					const struct net_device *out,
					int (*okfn) (struct sk_buff *))
{
	struct iphdr *iph = (*skb)->nh.iph;
	struct in_addr myaddr = my_addr();

	if (in && in->ifindex == get_slave_dev_ifindex() &&
	    (*skb)->protocol == htons(ETH_P_IP)) {

		if (iph && iph->protocol == IPPROTO_DSR &&
		    iph->daddr != myaddr.s_addr &&
		    iph->daddr != DSR_BROADCAST) {

			(*skb)->data = (*skb)->nh.raw + (iph->ihl << 2);
			(*skb)->len = (*skb)->tail - (*skb)->data;
			dsr_ip_recv(*skb);

			return NF_STOLEN;
		}
	}
	return NF_ACCEPT;
}

static struct nf_hook_ops dsr_pre_routing_hook = {

	.hook = dsr_pre_routing_recv,
#ifdef KERNEL26
	.owner = THIS_MODULE,
#endif
	.pf = PF_INET,
	.hooknum = NF_IP_PRE_ROUTING,
	.priority = NF_IP_PRI_FIRST,
};

static struct nf_hook_ops dsr_ip_forward_hook = {

	.hook = dsr_ip_forward_recv,
#ifdef KERNEL26
	.owner = THIS_MODULE,
#endif
	.pf = PF_INET,
	.hooknum = NF_IP_FORWARD,
	.priority = NF_IP_PRI_FIRST,
};

/* This is kind of a mess */
#if LINUX_VERSION_CODE < KERNEL_VERSION(2,6,7)
static struct inet_protocol dsr_inet_prot = {
#else
static struct net_protocol dsr_inet_prot = {
#endif
	.handler = dsr_ip_recv,
	.err_handler = dsr_ip_recv_err,
#ifdef KERNEL26
	.no_policy = 1,
#else
	.protocol = IPPROTO_DSR,
	.name = "DSR"
#endif
};

static int __init dsr_module_init(void)
{
	int res = -EAGAIN;
	struct proc_dir_entry *proc;

#ifdef DEBUG
	dbg_init();
#endif
	parse_mackill();

	res = dsr_dev_init(ifname);

	if (res < 0) {
		DEBUG("dsr-dev init failed\n");
		return -EAGAIN;
	}
	res = send_buf_init();

	if (res < 0)
		goto cleanup_dsr_dev;

	res = rreq_tbl_init();

	if (res < 0)
		goto cleanup_send_buf;

	res = neigh_tbl_init();

	if (res < 0)
		goto cleanup_rreq_tbl;

	res = nf_register_hook(&dsr_pre_routing_hook);

	if (res < 0)
		goto cleanup_neigh_tbl;

	res = nf_register_hook(&dsr_ip_forward_hook);

	if (res < 0)
		goto cleanup_nf_hook2;

	res = maint_buf_init();

	if (res < 0)
		goto cleanup_nf_hook1;

	proc = create_proc_entry(CONFIG_PROC_NAME, S_IRUGO | S_IWUSR, proc_net);

	if (!proc)
		goto cleanup_maint_buf;

	proc->owner = THIS_MODULE;
	proc->read_proc = dsr_config_proc_read;
	proc->write_proc = dsr_config_proc_write;

#ifndef KERNEL26
	inet_add_protocol(&dsr_inet_prot);
	DEBUG("Setup finished\n");
	return 0;
#else
	res = inet_add_protocol(&dsr_inet_prot, IPPROTO_DSR);

	if (res < 0) {
		DEBUG("Could not register inet protocol\n");
		goto cleanup_proc;
	}

	DEBUG("Setup finished res=%d\n", res);

	return 0;
      cleanup_proc:
	proc_net_remove(CONFIG_PROC_NAME);
#endif

      cleanup_maint_buf:
	maint_buf_cleanup();
      cleanup_nf_hook1:
	nf_unregister_hook(&dsr_ip_forward_hook);
      cleanup_nf_hook2:
	nf_unregister_hook(&dsr_pre_routing_hook);
      cleanup_neigh_tbl:
	neigh_tbl_cleanup();
      cleanup_rreq_tbl:
	rreq_tbl_cleanup();
      cleanup_send_buf:
	send_buf_cleanup();
      cleanup_dsr_dev:
	dsr_dev_cleanup();
#ifdef DEBUG
	dbg_cleanup();
#endif
	return res;
}

static void __exit dsr_module_cleanup(void)
{
#ifdef KERNEL26
	inet_del_protocol(&dsr_inet_prot, IPPROTO_DSR);
#else
	inet_del_protocol(&dsr_inet_prot);
#endif
	proc_net_remove(CONFIG_PROC_NAME);
	nf_unregister_hook(&dsr_pre_routing_hook);
	nf_unregister_hook(&dsr_ip_forward_hook);
	send_buf_cleanup();
	rreq_tbl_cleanup();
	neigh_tbl_cleanup();
	maint_buf_cleanup();
	dsr_dev_cleanup();
#ifdef DEBUG
	dbg_cleanup();
#endif
}

module_init(dsr_module_init);
module_exit(dsr_module_cleanup);