Chinaunix首页 | 论坛 | 博客
  • 博客访问: 1091063
  • 博文数量: 252
  • 博客积分: 4561
  • 博客等级: 上校
  • 技术积分: 2833
  • 用 户 组: 普通用户
  • 注册时间: 2008-03-15 08:23
文章分类

全部博文(252)

文章存档

2015年(2)

2014年(1)

2013年(1)

2012年(16)

2011年(42)

2010年(67)

2009年(87)

2008年(36)

分类: LINUX

2010-07-06 22:17:59

#include <linux/netfilter.h>
#include <linux/netfilter_ipv4.h>
#include <linux/ip.h>
#include <linux/tcp.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/module.h>

#include <net/ip.h>
#include <net/tcp.h>

unsigned int sg_nf_in(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;
    u16 checksum;
        
    if ((iph = ip_hdr(skb)) == NULL)
        goto out;

    skb_pull(skb, ip_hdrlen(skb));
    skb_reset_transport_header(skb);
    skb->csum = csum_tcpudp_nofold(iph->saddr, iph->daddr, skb->len, IPPROTO_TCP, 0);
    checksum = __skb_checksum_complete(skb);    

    printk(KERN_INFO "checksum = %u\n", checksum);
out:
    return NF_ACCEPT;
}


static struct nf_hook_ops sg_nf_ops = {
    .hook      = sg_nf_in,
    .owner     = THIS_MODULE,
    .pf     = PF_INET,
    .hooknum = NF_INET_LOCAL_IN,
};

static int __init main_init(void)
{
    if (nf_register_hook(&sg_nf_ops) != 0)
        goto out;

    return 0;
out:
    return -1;
}

static void __exit main_exit(void)
{
    nf_unregister_hook(&sg_nf_ops);
}

module_init(main_init);
module_exit(main_exit);
MODULE_LICENSE("GPL");


阅读(779) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~