当前位置 : 主页 > 操作系统 > centos >

一个可以直接使用的可用iptables配置的stateless NAT实现

来源:互联网 收集:自由互联 发布时间:2022-06-20
使用iptables配置stateless NAT?我没有搞错。 可能你根本不知道这么多NAT的实现细节,或者说根本不在乎,那么本文就当是一个“如何编写iptables模块”的练习了。 实际上,我已经实现了一

使用iptables配置stateless NAT?我没有搞错。
可能你根本不知道这么多NAT的实现细节,或者说根本不在乎,那么本文就当是一个“如何编写iptables模块”的练习了。
实 际上,我已经实现了一个可以配置stateless NAT的内核模块了,但是它的接口是基于procfs的,并不是说这个接口不好用,而是我觉得如果能集成到iptables就更加perfect了,难道 不应该这样吗?有谁能忍受通过iptables和echo的方式配置两种NAT呢?难道不应该在一个iptables -L或者iptables-save命令中展示所有的配置吗?
       想法是一回事,实现是另一回事,二者之间需要有一个促成的动力,这个动力来自于一个陈年的帖子,大概是2004年的吧,那是我还在上学,那个帖子就是在讨 论关于Linux下stateless NAT的问题,时隔多年,如果你搜索相关的主题,依然没有太好的答案。也许是没有这方面的需求,但真的没有吗?我觉得不。在2.4内核时期,还可以通过 iproute2来配置stateless NAT,在2.6/3.X内核时期就只能用tc来配置了,不管怎么说,总是可以做到的,但是你不觉得太麻烦了吗?难道就不能像下面这样子配置一条 stateless NAT吗?
iptables -t nat -A PREROUTING -j STATIC-2-WAY-NAT --mapaddr 192.168.184.250-192.168.184.154 --type src
不 过请注意,iptables本身有一个默认的事实,那就是它是基于match和target的,它的句法是“如果...那么...”,这个句法在 static staleless NAT便不适合了,因为对于这种NAT而言,每添加一条规则,就会自动生成一条反转的规则,这样的话,“如果...那么...”就不行了,要想用 iptables配置stateless NAT,就必须仅仅将规则设置进内核,其它的都由独立的模块来做,换句话说,我们仅仅利用iptables的接口,而不使用它的match机制,因此我只 需要注册一个target,在这个target的checkentry回调中完成“设置规则(正反两个方向)”的操作,在destroy回调中完成“删除 规则”的操作,而target回调本身则什么都不做。幸亏target有这么两个回调函数,否则的话估计玩的还真的有点大。
       但是,千万别把checkentry/destroy这两个回调函数想得太简单,事实上,它们的调用机制远比你想象的要复杂得多,如果我说每添加一条规 则,该target的所有的规则都要重新checkentry一次,这可能有点抽象又不合情理,那么下面我就先说一下iptables规则的添加机制。 iptables添加/删除规则事实上操作的是两个规则集,即新的规则集replace旧的规则集,新旧规则集的区别在于对于添加操作新的规则集中多了一 条要添加的规则,对于删除操作新的规则集中少了一条要删除的规则。每一个规则集都是一块连续的内存,正是因为内存的连续性是一个要求(内存连续可以完全基 于offset来寻址,不必引入指针,完美利用局部性原理),才会出现上述那种replace机制,因为每添加/删除一条规则规则集的大小就会发生变化。 以下的文字摘自我的代码注释:
一个可以直接使用的可用iptables配置的stateless NAT实现_perfect

代码我已经放在了github上了,不过还是在这里贴一份,作为备份吧。
       给出代码前,先看一下用法,这个STATIC-2-WAY-NAT模块目前一共以下几个参数:
--mapaddr a.b.c.d-A.B.C.D
必须参数。它将地址a.b.c.d转换为A.B.C.D,至于是转换源地址还是目标地址,就看该配置的类型是src还是dst以及数据的方向了。
--type [src|dst]
必须参数。指示转换的类型,src表示源地址转换,它将--mapaddr参数中的源地址为a.b.c.d的数据包的源地址转换为A.B.C.D,目标地址为A.B.C.D的数据包的目标地址转换为a.b.c.d;dst表示目标地址转换,解释类似src。
--proto [udp|tcp]
可选参数。指示转换的数据包协议类型,如果缺失这个参数,则代表所有协议。
--dev [ethX|...]
可选参数。指示参与地址转换的数据包的接收网卡和发送网卡。
--mapport p1-p2
可选参数。指示第四层协议的端口转换规则,仅仅针对udp和tcp,即如果有这个参数,则必须指定proto为udp或者tcp。
目 前的参数就以上这么多,日后会完善。使用方法很简单,为了好看那么一点点,我将我的NAT实现放在了nat表的PREROUTING和 POSTROUTING上,在写规则的时候,随便哪个ROUTING都行,实现并不关心,matches在理论上是没有用的,但是它可以对精确匹配到的数 据包屏蔽基于conntrack的原生NAT操作,这也算是一种副作用吧。
代码一共三个文件,可用但不完美,一条日志都没有打,也算一种拆弹方式...:

xt_STATIC-2-WAY-NAT.c:这是一个内核模块,实现了NAT的核心逻辑和iptables target接口

/*   *  * procfs接口的用法(已经禁用):  * 对目标地址为1.2.1.2的数据包做目标地址转换,目标转为192.168.1.8  * echo +1.2.1.2 192.168.1.8 dst >/proc/net/static_nat  * 上述命令会同时添加一条反向的SNAT映射  *  * 上述命令添加协议支持:  * echo +1.2.1.2 192.168.1.8 dst tcp >/proc/net/static_nat  *  * 继续增加端口映射的支持:  * echo +1.2.1.2 192.168.1.8 dst tcp port-map 1234 80 >/proc/net/static_nat  *  * 请解释:  * echo +192.168.184.250 192.168.184.154 src >/proc/net/static_nat  *  * iptables接口的用法:  * 见下面的详细注释。  *  */ #include <linux/module.h> #include <linux/skbuff.h> #include <linux/ctype.h> #include <net/ip.h> #include <net/netfilter/nf_conntrack.h> #include <linux/netfilter/x_tables.h> #include "xt_STATIC-2-WAY-NAT.h" static __be16 skip_atos(const char *s) {     __be16 i = 0;     while (isdigit(*s)) {         i = i * 10 + *((s)++) - '0';     }     return i; } #define DIRMASK 0x11 #define BUCKETS 1024 #define ENTRY_ADD   0x10 #define ENTRY_DEL   0x20 #define NAT_OPT_DEL         0x01 #define NAT_OPT_FIND        0x04 #define NAT_OPT_ACCT_BIT    0x02 /*  * 记录统计信息  */ struct nat_account {     u32 nat_packets;     u32 nat_bytes; }; /*  * 这个entry数据结构的设计有两个选择:  * 1.复杂模式:  *      即将所有的信息都展现于entry本身,skb匹配过程根据skb本身的IP地址信息作为key,  *      找到entry后再匹配第四层协议。  * 2.简单模式:  *      即设置多张保存简单entry(仅仅保存IP地址映射信息)的hlist,每个协议一张表,匹配  *      时仅仅需要匹配从skb中取出的protocol相关的hlist即可。  * 很明显,第二种效率更高,但是也更松散,本实现采用第一种,日后补充实现第二种。  */ struct static_nat_entry {     __be32 addr[DIR_NUM];     /* 该entry适用的第四层协议 */     u_int8_t proto;     union {         __be16 all[DIR_NUM];         struct {             __be16 port[DIR_NUM];         } tcp;         struct {             __be16 port[DIR_NUM];         } udp;         /* ...... */     } u;     enum nat_dir type;     struct net_device *dev;     struct nat_account acct[DIR_NUM];     struct hlist_node node[DIR_NUM]; }; /*  * 返回查询结果  */ struct map_result {     __be32 addr;     __be16 port; }; static DEFINE_SPINLOCK(nat_lock); /* 保存SNAT映射 */ struct hlist_head *src_list; /* 保存DNAT映射 */ struct hlist_head *dst_list; /*  * 用一个IP地址(对于PREROUTING是daddr,对于POSTROUTING是saddr)作为key来获取value。  */ static unsigned int get_address_from_map(unsigned int dir,                                             __be32 addr_key,                                             __be16 port_key,                                             u8 proto_key,                                             unsigned int opt,                                             struct map_result *res,                                             unsigned int datalen,                                             const struct net_device *dev) {     unsigned int ret = -1;     /* 首先匹配明细协议,然后再匹配通配协议 */     int try = 1;     __be32 cmp_key, ret_addr;     u32 hash;     struct hlist_head *list;     struct hlist_node *iter, *tmp;     struct static_nat_entry *ent;     /* 将协议作为hash计算的一部分,可以让来自同一地址的连接散列效果更好,      * 但是代价就是如果匹配不成功,则需要将协议设置为缺省通配协议,重新      * 匹配一次。      *      * 我的第一个版本在计算hash时并没有指定proto字段,第二个版本指定了,      * 因为经过我模拟的压力测试表明,对于配置明细协议的规则而言,匹配      * 会更加迅速。也就是说,这个效率和iptables是一样的,和你配置的规则      * 有关,即:最好配置带有明细协议的规则,而不是配置通配规则(你考虑      * 的少了,机器就要多考虑,反过来想让机器不考虑那么多,你自己就要      * 多考虑!)      */ try_agin:     hash = jhash_2words(addr_key, (__be32)proto_key, 1);     hash = hash%BUCKETS;     spin_lock(&nat_lock);     if (dir == DIR_DNAT) {         list = &dst_list[hash];     } else if (dir == DIR_SNAT) {         list = &src_list[hash];     } else {         spin_unlock(&nat_lock);         goto out;     }     hlist_for_each_safe(iter, tmp, list) {         ent = hlist_entry(iter, struct static_nat_entry, node[dir]);         /* 注意反转 */         cmp_key = (ent->type == dir) ?                             ent->addr[0]:ent->addr[1];         ret_addr = (ent->type == dir) ?                             ent->addr[1]:ent->addr[0];         if (addr_key == cmp_key) {             __be16 cmp_port = (ent->type == dir) ?                                 ent->u.all[0]:ent->u.all[1];             __be16 ret_port = (ent->type == dir) ?                                 ent->u.all[1]:ent->u.all[0];             if (ent->proto != IPPROTO_MAX - 1 && ent->proto != proto_key) {                 continue;             }             /* 如果addr比较不通过,就无需下面的了 */             if (cmp_port) {                 /* 如果port不参与比较,就无需下面的了 */                 if (cmp_port != port_key) {                     continue;                 }             }             if (dev && ent->dev && dev != ent->dev) {                 continue;             }             /* 实际上,上面的嵌套if完全可以用C的布尔逻辑搞定,但是那样的话if条件会很长 */             ret = 0;             res->addr = ret_addr;             res->port = ret_port;             try = try - 1;             if (opt == NAT_OPT_DEL) {                 if (dir == ent->type) {                     hlist_del(&ent->node[0]);                     hlist_del(&ent->node[1]);                     if (ent->dev) {                         dev_put(ent->dev);                     }                     kfree(ent);                 } else {                     ret = -1;                 }             }             if (opt & NAT_OPT_ACCT_BIT) {                 ent->acct[dir].nat_packets ++;                 ent->acct[dir].nat_bytes += datalen;             }             break;         }     }     spin_unlock(&nat_lock);     if (try > 0) {         try = try - 1;         proto_key = IPPROTO_MAX - 1;         goto try_agin;     } out:     return ret; } /*  * 处理第七层的函数  * 它处理第七层协议中携带地址端口信息的情况。  * 作为一个简单的例子,我用一个自定义的协议来举例:  * ...||layer3||layer4||saddr|daddr||  */ static int process_l7(struct sk_buff *skb, unsigned int dir, __be32 newaddr, __be16 newport) {     int ret = 0;     goto out; out:     return ret; } static u8 get_l4_proto(struct sk_buff *skb) {     u8 ret = IPPROTO_MAX - 1;     struct iphdr *iph = ip_hdr(skb);     switch (iph->protocol) {     case IPPROTO_TCP:         ret = IPPROTO_TCP;         break;     case IPPROTO_UDP:         ret = IPPROTO_UDP;         break;     default:         ret= IPPROTO_MAX - 1;         break;     }     return ret; } static s16 get_l4_port(struct sk_buff *skb, unsigned int dir) {     struct iphdr *iph = ip_hdr(skb);     void *transport_hdr = (void *)iph + ip_hdrlen(skb);     struct tcphdr *tcph;     struct udphdr *udph;     __be16 ret = 0;     switch (iph->protocol) {     case IPPROTO_TCP:         tcph = transport_hdr;         if (dir == DIR_SNAT) {             ret = tcph->source;         } else if(dir == DIR_DNAT) {             ret = tcph->dest;         } else {             ret = 0;         }         break;     case IPPROTO_UDP:     case IPPROTO_UDPLITE:         udph = transport_hdr;         if (dir == DIR_SNAT) {             ret = udph->source;         } else if(dir == DIR_DNAT) {             ret = udph->dest;         } else {             ret = 0;         }         break;     default:         ret = 0;     }     return ret; } static void set_l4_port(struct sk_buff *skb, unsigned int dir, __be16 port) {     struct iphdr *iph = ip_hdr(skb);     void *transport_hdr = (void *)iph + ip_hdrlen(skb);     struct tcphdr *tcph;     struct udphdr *udph;     switch (iph->protocol) {     case IPPROTO_TCP:         tcph = transport_hdr;         if (dir == DIR_SNAT) {             tcph->source = port;         } else if(dir == DIR_DNAT) {             tcph->dest = port;         }         break;     case IPPROTO_UDP:     case IPPROTO_UDPLITE:         udph = transport_hdr;         if (dir == DIR_SNAT) {             udph->source = port;         } else if(dir == DIR_DNAT) {             udph->dest = port;         }         break;     } } /*  * 更新第四层的校验码信息  */ static void nat4_update_l4(struct sk_buff *skb,                             __be32 oldip, __be32 newip,                             __be16 oldport, __be16 newport) {     struct iphdr *iph = ip_hdr(skb);     void *transport_hdr = (void *)iph + ip_hdrlen(skb);     struct tcphdr *tcph;     struct udphdr *udph;     bool cond;     switch (iph->protocol) {     case IPPROTO_TCP:         tcph = transport_hdr;         inet_proto_csum_replace4(&tcph->check, skb, oldip, newip, 1);         if (newport) {             inet_proto_csum_replace2(&tcph->check, skb, oldport, newport, 0);         }         break;     case IPPROTO_UDP:     case IPPROTO_UDPLITE:         udph = transport_hdr;         cond = udph->check != 0;         cond |= skb->ip_summed == CHECKSUM_PARTIAL;         if (cond) {             inet_proto_csum_replace4(&udph->check, skb, oldip, newip, 1);             if (newport) {                 inet_proto_csum_replace2(&udph->check, skb, oldport, newport, 0);             }             if (udph->check == 0) {                 udph->check = CSUM_MANGLED_0;             }         }         break;     } } /*  * 在POSTROUTING上执行源地址转换:  * 1.正向源地址转换;  * 2.目标地址转换的逆向源地址转换  */ static unsigned int ipv4_nat_out(unsigned int hooknum,                  struct sk_buff *skb,                  const struct net_device *in,                  const struct net_device *out,                  int (*okfn)(struct sk_buff *)) {     unsigned int ret = NF_ACCEPT, res = 0;     struct map_result mres;     struct iphdr *hdr = ip_hdr(skb);     __be16 port = 0;     u8 proto;     port = get_l4_port(skb, DIR_SNAT);     proto = get_l4_proto(skb);     memset(&mres, 0, sizeof(mres));     res = get_address_from_map(DIR_SNAT, hdr->saddr, port, proto, NAT_OPT_FIND|NAT_OPT_ACCT_BIT, &mres, skb->len, out);     if (res) {         goto out;     }     if (hdr->saddr == mres.addr) {         goto out;     }     /* 执行SNAT */     if (process_l7(skb, DIR_DNAT, mres.addr, mres.port)) {         /* 如果第七层发生了改变,则重新计算相关校验码 */     }     csum_replace4(&hdr->check, hdr->saddr, mres.addr);     nat4_update_l4(skb, hdr->saddr, mres.addr, port, mres.port);     if (mres.port) {         set_l4_port(skb, DIR_SNAT, mres.port);     }     hdr->saddr = mres.addr; out:     return ret; } /*  * 在PREROUTING上执行目标地址转换:  * 1.正向目标地址转换;  * 2.源地址转换的逆向目标地址转换  */ static unsigned int ipv4_nat_in(unsigned int hooknum,                       struct sk_buff *skb,                       const struct net_device *in,                       const struct net_device *out,                       int (*okfn)(struct sk_buff *)) {     unsigned int ret = NF_ACCEPT, res = 0;     struct map_result mres;     struct iphdr *hdr = ip_hdr(skb);     __be16 port = 0;     u8 proto;     port = get_l4_port(skb, DIR_DNAT);     proto = get_l4_proto(skb);     if (skb->nfct && skb->nfct != &nf_conntrack_untracked.ct_general) {         goto out;     }     memset(&mres, 0, sizeof(mres));     res = get_address_from_map(DIR_DNAT, hdr->daddr, port, proto, NAT_OPT_FIND|NAT_OPT_ACCT_BIT, &mres, skb->len, in);     if (res) {         goto out;     }     if (hdr->daddr == mres.addr) {         goto out;     }     /* 执行DNAT */     if (process_l7(skb, DIR_DNAT, mres.addr, mres.port)) {         /* 如果第七层发生了改变,则重新计算相关校验码 */     }     csum_replace4(&hdr->check, hdr->daddr, mres.addr);     nat4_update_l4(skb, hdr->daddr, mres.addr, port, mres.port);     if (mres.port) {         set_l4_port(skb, DIR_DNAT, mres.port);     }     hdr->daddr = mres.addr;     /*      *  设置一个notrack 防止其被track以及nat.      *  这是绝对合适的,因为既然是static的stateless NAT      *  我们就不希望它被状态左右      **/     /*      * 其实,并不是主要避开基于conntrack的NAT就可以了,因为      * conntrack本身就不容你对两个方向的tuple进行随意修改      */     if (!skb->nfct) {         skb->nfct = &nf_conntrack_untracked.ct_general;         skb->nfctinfo = IP_CT_NEW;         nf_conntrack_get(skb->nfct);     } out:     return ret; } static struct nf_hook_ops ipv4_nat_ops[] __read_mostly = {     {         .hook       = ipv4_nat_in,         .owner      = THIS_MODULE,         .pf     = NFPROTO_IPV4,         .hooknum    = NF_INET_PRE_ROUTING,         .priority   = NF_IP_PRI_RAW + 1,     },     {         .hook       = ipv4_nat_out,         .owner      = THIS_MODULE,         .pf     = NFPROTO_IPV4,         .hooknum    = NF_INET_POST_ROUTING,         .priority   = NF_IP_PRI_CONNTRACK + 1,     }, }; static char *parse_addr(const char *input, __be32 *from, __be32 *to) {     char *p1, *p2;     size_t length = strlen(input);     if (!(p1 = memchr(input, ' ', length))) {         return NULL;     }     if (!(p2 = memchr(p1 + 1, ' ', length - (p1 + 1 - input)))) {         return NULL;     }     if (!(in4_pton(input, p1 - input, (u8 *)from, ' ', NULL))             || !(in4_pton(p1 + 1, p2 - p1 - 1, (u8 *)to, ' ', NULL))) {         return NULL;     }     return ++p2; } static char *parse_port(char *input, __be16 *from, __be16 *to) {     char *p1;     size_t length = strlen(input);     size_t delta = 0;     char tmp[8] = {0};     __be16 res;     /*portfrom portto*/     if (!(p1 = memchr(input, ' ', length))) {         return NULL;     }     delta = p1 - input;     memcpy(tmp, input, delta);     res = skip_atos(tmp);     if (!res) {         return NULL;     }     *from = htons(res);     memset(tmp, 0, sizeof(tmp));     memcpy(tmp, p1+1, length-delta);     res = skip_atos(tmp);     if (!res) {         return NULL;     }     *to = htons(res);     return p1; } static int add_remove_nat_entry(struct static_nat_entry *ent,                                 __be32 from, __be32 to,                                 __be16 from_port, __be16 to_port,                                 u8 dir,                                 u8 proto,                                 struct net_device *dev,                                 u8 opt) {     int ret = 0;     __be32 normal, reverse;     struct map_result mres;     if (opt == ENTRY_ADD) {         if (!ent) {             ret = -EINVAL;             goto out;         }         ent->proto = proto;         /* 计算原始项的hash桶位置 */         normal = jhash_2words(from, (__be32)proto, 1);         normal = normal%BUCKETS;         /* 计算反转位置的hash桶位置 */         reverse = jhash_2words(to, (__be32)proto, 1);         reverse = reverse%BUCKETS;         /*          *  设置key/value对          *  注意,反转类型的hnode其key/value也要反转          */         ent->addr[0] = from;         ent->addr[1] = to;         ent->u.all[0] = from_port;         ent->u.all[1] = to_port;         /* 这是这个entry的type,用来区分生成的两条配置项 */         ent->type = dir;         ent->dev = dev;         /* 初始化链表节点 */         INIT_HLIST_NODE(&ent->node[DIR_SNAT]);         INIT_HLIST_NODE(&ent->node[DIR_DNAT]);         if (dir == DIR_SNAT) { /* 添加SNAT项,自动生成DNAT项 */             /* 首先判断是否已经存在了 */             if (!get_address_from_map(DIR_SNAT, from, from_port, proto, NAT_OPT_FIND, &mres, 0, dev) ||                     !get_address_from_map(DIR_SNAT, to, to_port, proto, NAT_OPT_FIND, &mres, 0, dev)) {                 ret = -EEXIST;                 goto out;             }             /* 落实到链表 */             spin_lock(&nat_lock);             hlist_add_head(&ent->node[DIR_SNAT], &src_list[normal]);             hlist_add_head(&ent->node[DIR_DNAT], &dst_list[reverse]);             spin_unlock(&nat_lock);         } else if(dir == DIR_DNAT) { /* 添加DNAT项,自动生成SNAT项 */             /* 首先判断是否已经存在了 */             struct map_result mres;             if (!get_address_from_map(DIR_DNAT, from, from_port, proto, NAT_OPT_FIND, &mres, 0, dev) ||                     !get_address_from_map(DIR_DNAT, to, to_port, proto, NAT_OPT_FIND, &mres, 0, dev)){                 ret = -EEXIST;                 goto out;             }             /* 落实到链表 */             spin_lock(&nat_lock);             hlist_add_head(&ent->node[DIR_DNAT], &dst_list[normal]);             hlist_add_head(&ent->node[DIR_SNAT], &src_list[reverse]);             spin_unlock(&nat_lock);         } else {             ret = -EINVAL;             goto out;         }     } else if (opt == ENTRY_DEL) {         u32 r1;         if (dir == DIR_SNAT) {             r1 = get_address_from_map(DIR_SNAT, from, from_port, proto, NAT_OPT_DEL, &mres, 0, dev);             if (r1) {                 ret = -ENOENT;                 goto out;             }         } else if(dir == DIR_DNAT) {             r1 = get_address_from_map(DIR_DNAT, from, from_port, proto, NAT_OPT_DEL, &mres, 0, dev);             if (r1) {                 ret = -ENOENT;                 goto out;             }         } else {             ret = -EINVAL;             goto out;         }     } else {         ret = -EINVAL;         goto out;     } out:         return ret; } static ssize_t static_nat_config_write(struct file *file, const char *buffer, size_t count, loff_t *unused) {     int ret = 0;     size_t length = count;     __be32 from, to;     __be16 from_port = 0, to_port = 0;     u8 proto = 0;     char *buf = NULL;     char *p, *pport, *last;     struct static_nat_entry *ent;     struct net_device *dev = NULL;     if (length) {         char *pp = (char *)(buffer + (length - 1));         for (; (*pp < (char)32) || (*pp > (char)126); pp--) {             if (length <= 0) {                 ret = -EINVAL;                 goto out;             }             length--;         }     } else {         goto out;     }     buf = kzalloc((length + 1), GFP_ATOMIC);     if (!buf) {         ret = -ENOMEM;         goto out;     }     memcpy(buf, buffer, length);     if (!(p = parse_addr(buf + 1, &from, &to))) {         ret = -EINVAL;         goto out;     }     /*      * dev = dev_get_by_name(&init_net, $待解析的dev名字);      * */     if (strstr(p, "tcp") && !strstr(p, "udp")) {         proto = IPPROTO_TCP;     } else if (strstr(p, "udp") && !strstr(p, "tcp")) {         proto = IPPROTO_UDP;     } else {         /* 支持IPIP以及所有协议 */         proto = IPPROTO_MAX - 1;     }     if (((pport = strstr(p, "port-map")) != NULL) &&             ((last = parse_port(pport + strlen("port-map") + 1, &from_port, &to_port)) == NULL)) {         ret = -EINVAL;         goto out;     }     if ('+' == *buf) {         ent = (struct static_nat_entry *)kzalloc(sizeof(struct static_nat_entry), GFP_KERNEL);         if (!ent) {             ret = -EFAULT;             goto out;         }         if (strstr(p, "src")) { /* 添加SNAT项,自动生成DNAT项 */             ret = add_remove_nat_entry(ent, from, to, from_port, to_port, DIR_SNAT, proto, dev, ENTRY_ADD);             if (ret) {                 kfree(ent);                 goto out;             }         } else if(strstr(p, "dst")) { /* 添加DNAT项,自动生成SNAT项 */             ret = add_remove_nat_entry(ent, from, to, from_port, to_port, DIR_DNAT, proto, dev, ENTRY_ADD);             if (ret) {                 kfree(ent);                 goto out;             }         } else {             ret = -EFAULT;             kfree(ent);             goto out;         }     } else if ('-' ==*buf) {         if (strstr(p, "src")) {             ret = add_remove_nat_entry(NULL, from, to, from_port, to_port, DIR_SNAT, proto, dev, ENTRY_DEL);             if (ret) {                 goto out;             }         } else if(strstr(p, "dst")) {             ret = add_remove_nat_entry(NULL, from, to, from_port, to_port, DIR_DNAT, proto, dev, ENTRY_DEL);             if (ret) {                 goto out;             }         } else {             ret = -EINVAL;             goto out;         }     } else {         ret = -EINVAL;         goto out;     }     ret = count; out:     kfree(buf);     return ret; } static ssize_t static_nat_config_read(struct file *file, char __user *buf, size_t count, loff_t *ppos) {     int len = 0;     static int done = 0;     int i;     char from[15], to[15];     char from_port[8], to_port[8];     char *kbuf_to_avoid_user_space_memory_page_fault = NULL; /* 每一行的最大长度 */ #define MAX_LINE_CHARS  128     if (done) {         done = 0;         goto out;     }     /*      * 分配一块内核内存,为了避免直接操作用户内存而引发页面调度,      * 页面调度会导致睡眠切换,而我们操作的内容处在自旋锁的保护      * 下,所以不能切换!      */     /*      * 问题:      * 我这里仅仅分配count大小的内存,是因为这个版本不支持多次读,      * 只能一次读完。也许我应该学学seq read的方法。      */     kbuf_to_avoid_user_space_memory_page_fault = kzalloc(count, GFP_KERNEL);     if (!kbuf_to_avoid_user_space_memory_page_fault) {         len = -ENOMEM;         done = 1;         goto out;     }     spin_lock(&nat_lock);     len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len, "Source trans table:\n");     if (len + MAX_LINE_CHARS > count) {         goto copy_now;     }     for (i = 0; i < BUCKETS; i++) {         struct hlist_node *iter, *tmp;         struct static_nat_entry *ent;         hlist_for_each_safe(iter, tmp, &src_list[i]) {             ent = hlist_entry(iter, struct static_nat_entry, node[DIR_SNAT]);             sprintf(from, "%pI4", (ent->type == DIR_SNAT)? &ent->addr[0]:&ent->addr[1]);             sprintf(to, "%pI4", (ent->type == DIR_SNAT)? &ent->addr[1]:&ent->addr[0]);             if (ent->u.all[0] && ent->u.all[1]) {                 sprintf(from_port, "%d", ntohs((ent->type == DIR_SNAT)? ent->u.all[0]:ent->u.all[1]));                 sprintf(to_port, "%d", ntohs((ent->type == DIR_SNAT)? ent->u.all[1]:ent->u.all[0]));                 len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len,                             "From:%-15s To:%-15s    [%s %s] Port map[From:%-5s To:%-5s] [%s] [Bytes:%u  Packet:%u]\n",                                 from,                                 to,                                 (ent->proto == IPPROTO_TCP)?                                             "TCP":                                             (ent->proto == IPPROTO_UDP)?"UDP":"ALL",                                 (ent->type == DIR_SNAT)?"STATIC":"AUTO",                                 from_port,                                 to_port,                                 (ent->dev == NULL)?"all":ent->dev->name,                                 ent->acct[DIR_SNAT].nat_bytes,                                 ent->acct[DIR_SNAT].nat_packets);             } else {                 len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len,                             "From:%-15s To:%-15s    [%s %s] [%s] [Bytes:%u  Packet:%u]\n",                                 from,                                 to,                                 (ent->proto == IPPROTO_TCP)?                                             "TCP":                                             (ent->proto == IPPROTO_UDP)?"UDP":"ALL",                                 (ent->type == DIR_SNAT)?"STATIC":"AUTO",                                 (ent->dev == NULL)?"all":ent->dev->name,                                 ent->acct[DIR_SNAT].nat_bytes,                                 ent->acct[DIR_SNAT].nat_packets);             }             if (len + MAX_LINE_CHARS > count) {                 goto copy_now;             }         }     }     len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len, "\nDestination trans table:\n");     if (len + MAX_LINE_CHARS > count) {         goto copy_now;     }     for (i = 0; i < BUCKETS; i++) {         struct hlist_node *iter, *tmp;         struct static_nat_entry *ent;         hlist_for_each_safe(iter, tmp, &dst_list[i]) {             ent = hlist_entry(iter, struct static_nat_entry, node[DIR_DNAT]);             sprintf(from, "%pI4", (ent->type == DIR_DNAT)? &ent->addr[0]:&ent->addr[1]);             sprintf(to, "%pI4", (ent->type == DIR_DNAT)? &ent->addr[1]:&ent->addr[0]);             if (ent->u.all[0] && ent->u.all[1]) {                 sprintf(from_port, "%d", ntohs((ent->type == DIR_DNAT)? ent->u.all[0]:ent->u.all[1]));                 sprintf(to_port, "%d", ntohs((ent->type == DIR_DNAT)? ent->u.all[1]:ent->u.all[0]));                 len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len,                             "From:%-15s To:%-15s    [%s %s] Port map[From:%-5s To:%-5s] [%s] [Bytes:%u  Packet:%u]\n",                                 from,                                 to,                                 (ent->proto == IPPROTO_TCP)?                                             "TCP":                                             (ent->proto == IPPROTO_UDP)?"UDP":"ALL",                                 (ent->type == DIR_DNAT)?"STATIC":"AUTO",                                 from_port,                                 to_port,                                 (ent->dev == NULL)?"all":ent->dev->name,                                 ent->acct[DIR_DNAT].nat_bytes,                                 ent->acct[DIR_DNAT].nat_packets);             } else {                 len += sprintf(kbuf_to_avoid_user_space_memory_page_fault + len,                             "From:%-15s To:%-15s    [%s %s] [%s] [Bytes:%u  Packet:%u]\n",                                 from,                                 to,                                 (ent->proto == IPPROTO_TCP)?                                             "TCP":                                             (ent->proto == IPPROTO_UDP)?"UDP":"ALL",                                 (ent->type == DIR_DNAT)?"STATIC":"AUTO",                                 (ent->dev == NULL)?"all":ent->dev->name,                                 ent->acct[DIR_DNAT].nat_bytes,                                 ent->acct[DIR_DNAT].nat_packets);             }             if (len + MAX_LINE_CHARS > count) {                 goto copy_now;             }         }     } copy_now:     spin_unlock(&nat_lock);     done = 1;     /* 这里已经解除自旋锁 */     if (copy_to_user(buf, kbuf_to_avoid_user_space_memory_page_fault, len))  {         len = EFAULT;         goto out;     } out:     if (kbuf_to_avoid_user_space_memory_page_fault) {         kfree(kbuf_to_avoid_user_space_memory_page_fault);     }     return len; } static const struct file_operations static_nat_file_ops = {     .owner      = THIS_MODULE,     .read       = static_nat_config_read, /* 由于有了iptables接口,为了不产生两个写接口的联动问题,特意封掉了procfs的write接口   *  .write      = static_nat_config_write,  */ }; /***************************************************************************************************************************/ /* 以下就是iptables接口了,我只是为了迎合它的语法而已,事实上static 2-way nat的规则中,matches完全不起作用,  * 甚至就连PREROUTING/POSTROUTING都完全不起作用,我的目的仅仅是将其设置进内核而已。因此起作用的只有target。  * 这是有原因的,因为我只是做一个纯粹的,无状态德,理所当然的,匹配地址端口即无条件转换的NAT,如果使用match  * 将达不到这个要求,试想,如果有-i参数匹配网卡,那么反向的包怎么匹配,你不得不写两条规则,xtables-addons里面  * 的RAWNAT似乎不是很完全,因此我要自己搞一个。  * 事实上,我只是利用了target结构体的checkentry/destroy回调函数,在checkentry中添加规则,在destroy中删除规则,  * 而所谓的规则也并不是指iptables规则,iptables只是起到一个stub的作用。  * 以下是一些规则样例:  * 1.做源地址转换,将源IP地址为192.168.184.1的包的源地址转换为192.168.184.2,反方向的包自动完成目标地址转换:  *   iptables -t nat -A PREROUTING -j STATIC-2-WAY-NAT --mapaddr 192.168.184.1-192.168.184.2 --type src   * 2.做目标地址转换,仅限于UDP协议,将目标IP地址为192.168.184.3的包的目标地址转换为192.168.184.4,反方向包自动做源地址转换:  *   iptables -t nat -A PREROUTING -j STATIC-2-WAY-NAT --mapaddr 192.168.184.3-192.168.184.4 --type dst --proto udp  * 3.做目标地址转换,仅限于TCP协议,解释同上,添加了一个端口映射:  *   iptables -t nat -A PREROUTING -j STATIC-2-WAY-NAT --mapaddr 1.1.1.5-1.6.8.6 --type dst --proto tcp \  * --mapport 1234-80  * 4.以下规则的所有matches无效,起不到限制规则匹配的作用:  *   iptables -t nat -A PREROUTING -i eth0 -p icmp -d 1.2.3.4 -j STATIC-2-WAY-NAT ...  */ /***************************************************************************************************************************/ /* 该链表保存了所有的当前iptales static nat规则 */ LIST_HEAD(curr_entrys); DEFINE_SPINLOCK(curr_entrys_lock); /*  * 引入下面的数据结构加入curr_entrys是有超级原因的。这是因为我必须维护两个链表。  * src_list/dst_list维护的是查找node,而entrys维护的则是iptables的规则node,为什么  * iptables的规则node不能重用src_list/dst_list呢?因为iptables允许添加两条多条相同  * 的规则。  * 因此必须采用引用计数的方式。关于这么做的原因还有一个因素,那就是iptables添加删除  * 规则时背后的操作:  * 添加规则:  *      1.开辟一块新的可以容纳新规则的内存空间(比原来的同一target空间大一个entry);  *      2.将老的规则全部copy到新的内存空间,新规则append到最后或者insert到中间(copy老规则时需预留间隙);  *      3.依次调用新内存空间所有entry的checkentry回调函数(失败则回退,略);  *      4.如果成功则调用老规则内存空间所有entry的destroy回调函数;  *      5.释放老规则的内存空间;  *  *      各个步骤示意图如下所示(o:old  n:new):  *  *          老规则空间:            o -1-  -2-  -3-  *  *          新规则空间:            ALLOC  *                                  n ---  ---  ---  ---  *  *          拷贝老规则到新空间:    n -1-  -2-  -3-  ---  *  *          设置新规则:            n -1-  -2-  -3-  -4-  *  *          调用1-4的checkentry:   ->chk->chk->chk->chk  *                                  n -1-  -2-  -3-  -4-  *  *          destroy老空间的规则:   ->dsy->dsy->dsy  *                                  o -1-  -2-  -3-   *  *          释放老规则空间:        FREE      *                                  o -1-  -2-  -3-   *  删除规则:  *      和添加规则一样。  *  *  之所以每次添加规则都要触动统一target所有的既有规则,是因为iptables规则在内存中是连续存放的,  *  一开始的时候并不知道数量,因此只能在每次添加新规则的时候重新分配大一个entry的空间,然后拷贝,  *  最终释放老地址空间。  */ struct entry_node {     struct list_head list;     atomic_t refcnt;     struct net_device *dev;     __be32 from, to;     __be16 port_from, port_to;     u8 proto;     u8 dir; }; static void entry_insert(struct entry_node *enode) {     spin_lock_bh(&curr_entrys_lock);     list_add_tail(&enode->list, &curr_entrys);     spin_unlock_bh(&curr_entrys_lock); } static struct entry_node *entry_alloc(__be32 from, __be32 to,                                         __be16 port_from, __be16 port_to,                                         u8 proto,                                         u8 dir,                                         struct net_device *dev) {     struct entry_node *node = kzalloc(sizeof(struct entry_node), GFP_KERNEL);     if (!node) {         return NULL;     }     if (dev) {         dev_hold(dev);     }     node->dev = dev;     node->from = from;     node->to = to;     node->port_from = port_from;     node->port_to = port_to;     node->proto = proto;     node->dir = dir;     INIT_LIST_HEAD(&node->list);     atomic_set(&node->refcnt, 1);     return node; } static bool check_and_use(__be32 from, __be32 to,                             __be16 port_from, __be16 port_to,                             u8 proto,                             u8 dir,                             struct net_device *dev) {     bool ret = false;     struct entry_node *i;     spin_lock_bh(&curr_entrys_lock);     if (!list_empty(&curr_entrys)) {         list_for_each_entry(i, &curr_entrys, list) {             if (i->from == from &&                 i->to == to &&                 i->port_from == port_from &&                 i->port_to == port_to &&                 i->proto == proto &&                 i->dir == dir &&                 i->dev == dev) {                 atomic_inc(&i->refcnt);                 ret = true;             }         }     }     spin_unlock_bh(&curr_entrys_lock);     return ret; } static bool check_and_put(__be32 from, __be32 to,                             __be16 port_from, __be16 port_to,                             u8 proto,                             u8 dir,                             struct net_device *dev) {     bool ret = false;     struct entry_node *i, *tmp;     spin_lock_bh(&curr_entrys_lock);     if (!list_empty(&curr_entrys)) {         list_for_each_entry_safe(i, tmp, &curr_entrys, list) {             if (i->from == from &&                 i->to == to &&                 i->port_from == port_from &&                 i->port_to == port_to &&                 i->proto == proto &&                 i->dir == dir &&                 i->dev == dev) {                 if (atomic_dec_and_test(&i->refcnt)) {                     list_del(&i->list);                     if (dev) {                         dev_put(dev);                     }                     kfree(i);                     ret = true;                 }             }         }     }     spin_unlock_bh(&curr_entrys_lock);     return ret; } /*   * 理论上讲,下面这个函数应该是iptables的一条规则中所有match都匹配到之后要调用的函数,  * 但是,你可以看到,在我的static 2-way nat中,它并不起任何作用。  * 但是等等!  * 它事实上取消了排在这个规则后面的statefull NAT的执行,因为它直接在nat表中ACCEPT了,  * 这是什么,这是一个副作用,这个副作用竟然如此有用,以至于它已经可以模拟Cisco/H3C设备的  * NAT了:静态NAT优先执行!  *  * 看来这是iptables相比procfs接口的一个特别有用的副作用了!  */ static unsigned int do_nothing(struct sk_buff *skb, const struct xt_target_param *par) {     return NF_ACCEPT; } static bool do_all_things_add(const struct xt_tgchk_param *par) {     int ret = true;     __be32 from = 0, to = 0;     __be16 from_port = 0, to_port = 0;     u8 proto;     u8 dir;     struct static_nat_entry *ent = NULL;     const struct xt_static_nat_tginfo *info = par->targinfo;     struct net_device *dev; /*  * struct xt_static_nat_tginfo {  *  __be32 addr[DIR_NUM];  *  __be16 port[DIR_NUM];  *  u_int8_t proto;  *  u_int8_t dir;  * };  */     from = info->addr[0];     to = info->addr[1];     from_port = info->port[0];     to_port = info->port[1];     proto = info->proto;     dir = info->dir;     dev = dev_get_by_name(&init_net, info->dev);     /* 只有在链表中没有该entry的情况下才添加 */     if (!check_and_use(from, to, from_port, to_port, proto, dir, dev)) {         struct entry_node *enode = entry_alloc(from, to, from_port, to_port, proto, dir, dev);         if (enode) {             entry_insert(enode);             ent = (struct static_nat_entry *)kzalloc(sizeof(struct static_nat_entry), GFP_KERNEL);             if (!ent) {                 ret = false;                 goto out;             }             ret = add_remove_nat_entry(ent, from, to, from_port, to_port, dir, proto, dev, ENTRY_ADD);             if (ret) {                 ret = false;                 kfree(ent);                 check_and_put(from, to, from_port, to_port, proto, dir, dev);                 goto out;             }         } else {             ret = false;             goto out;         }     } else if (dev) {         dev_put(dev);     }     ret = true; out:     return ret; } static void do_all_things_del(const struct xt_tgdtor_param *par) {     int ret = 0;     __be32 from = 0, to = 0;     __be16 from_port = 0, to_port = 0;     u8 proto;     u8 dir;     const struct xt_static_nat_tginfo *info = par->targinfo;     struct net_device *dev = NULL;     from = info->addr[0];     to = info->addr[1];     from_port = info->port[0];     to_port = info->port[1];     proto = info->proto;     dir = info->dir;     dev = dev_get_by_name(&init_net, info->dev);     if (check_and_put(from, to, from_port, to_port, proto, dir, dev)) {         ret = add_remove_nat_entry(NULL, from, to, from_port, to_port, dir, proto, dev, ENTRY_DEL);         if (ret) {             goto out;         }     } out:     if (dev) {         dev_put(dev);     }     return; } static struct xt_target static_nat_tg_reg[] __read_mostly = {     {         .name       = "STATIC-2-WAY-NAT",         .family     = NFPROTO_IPV4,         .target     = do_nothing,         .table      = "nat",         .targetsize = XT_ALIGN(sizeof(struct xt_static_nat_tginfo)),         .checkentry = do_all_things_add,         .destroy    = do_all_things_del,         .hooks      = (1 << NF_INET_POST_ROUTING) |                               (1 << NF_INET_PRE_ROUTING),         .me         = THIS_MODULE,     }, }; static int __init nf_static_nat_init(void) {     int ret = 0;     int i;     src_list = kzalloc(sizeof(struct hlist_head) * BUCKETS, GFP_KERNEL);     if (!src_list) {         ret = -ENOMEM;         goto out;     }     dst_list = kzalloc(sizeof(struct hlist_head) * BUCKETS, GFP_KERNEL);     if (!dst_list) {         ret = -ENOMEM;         goto out;     }     ret = nf_register_hooks(ipv4_nat_ops, ARRAY_SIZE(ipv4_nat_ops));     if (ret < 0) {         printk("nf_nat_ipv4: can't register hooks.\n");         goto out;     }     ret = xt_register_targets(static_nat_tg_reg, ARRAY_SIZE(static_nat_tg_reg));     if (ret < 0) {         printk("nf_nat_ipv4: can't register targets.\n");         goto out;     }     if (!proc_create("static_nat", 0644, init_net.proc_net, &static_nat_file_ops)) {         ret = -ENOMEM;         goto out;     }     for (i = 0; i < BUCKETS; i++) {         INIT_HLIST_HEAD(&src_list[i]);         INIT_HLIST_HEAD(&dst_list[i]);     }     return ret; out:     if (src_list) {         kfree(src_list);     }     if (dst_list) {         kfree(dst_list);     }     return ret; } static void __exit nf_static_nat_fini(void) {     int i;     remove_proc_entry("static_nat", init_net.proc_net);     xt_unregister_targets(static_nat_tg_reg, ARRAY_SIZE(static_nat_tg_reg));     nf_unregister_hooks(ipv4_nat_ops, ARRAY_SIZE(ipv4_nat_ops));     spin_lock(&nat_lock);     for (i = 0; i < BUCKETS; i++) {         struct hlist_node *iter, *tmp;         struct static_nat_entry *ent;         hlist_for_each_safe(iter, tmp, &src_list[i]) {             ent = hlist_entry(iter, struct static_nat_entry, node[0]);             hlist_del(&ent->node[DIR_SNAT]);             hlist_del(&ent->node[DIR_DNAT]);             kfree(ent);         }     }     spin_unlock(&nat_lock);     if (src_list) {         kfree(src_list);     }     if (dst_list) {         kfree(dst_list);     } } module_init(nf_static_nat_init); module_exit(nf_static_nat_fini); MODULE_DESCRIPTION("STATIC two-way NAT"); MODULE_AUTHOR("marywangran@126.com"); MODULE_LICENSE("GPL");



libxt_STATIC-2-WAY-NAT.c:这是用户态的iptables target模块的实现文件

/*  *  根据Netfilter/iptables社区的友好相权,逼出来一个iptables接口。  *  相比procfs要友好多了  *  *  起初,我就是嫌iptables模块写起来太麻烦,后来写过procfs接口之后才发现,  *  原来最麻烦的不是例行公事的调用,而是字符串解析。使用iptables框架的好处  *  在于它已经有了很多可重用的字符串解析以及到IP地址,端口的转换接口  */ #include <getopt.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <xtables.h> #include "xt_STATIC-2-WAY-NAT.h" #undef XT_ALIGN #define XT_ALIGN(s) (((s) + (__alignof__(struct _xt_align)-1))  \                     & ~(__alignof__(struct _xt_align)-1)) /* 枚举规则解析状态机的当前状态 */ enum {     ADDR_OK = (1<<0),     TYPE_OK = (1<<1),     PROTO_OK = (1<<2),     PORT_OK = (1<<3),     DEV_OK  = (1<<4), }; /* 规则命令定义 */ static const struct option static_2_way_nat_tg_opts[] = {     {.name = "mapaddr", .has_arg = true, .val = 'a'},     {.name = "mapport", .has_arg = true, .val = 'o'},     {.name = "proto", .has_arg = true, .val = 'p'},     /* 本来想将type设置成source/destination的,但UNIX短名称传统更适合Linux */     {.name = "type", .has_arg = true, .val = 't'},     /* 本来想为dev设置inside/outside属性的,但是还是用Linux术语吧 */     {.name = "dev", .has_arg = true, .val = 'd'},     {}, }; static void static_2_way_nat_tg_help(void) {     printf( "STATIC-2-WAY-NAT target options:\n" "    --mapaddr from-to --type [src|dst] --dev [ethX] --proto [tcp|udp|all] --mapport from_port-to_port\n" ); } /* 从字符串解析IP地址 */ static void parse_addr(const char *orig_arg, struct xt_static_nat_tginfo *info) {     char *arg, *dash;     size_t delta, len;     u_int32_t from, to;     arg = strdup(orig_arg);     if (arg == NULL) {         xtables_error(RESOURCE_PROBLEM, "strdup");     }     len = strlen(arg);     dash = strchr(arg, '-');     if (!dash) {         xtables_error(RESOURCE_PROBLEM, "invalid argument.");     }     delta = dash - arg;     arg[delta] = 0;     from = inet_addr(arg);     if (from == INADDR_NONE){         xtables_error(RESOURCE_PROBLEM, "invalid from address.");     }     info->addr[0] = from;     arg += delta + 1;     to = inet_addr(arg);     if (to == INADDR_NONE){         xtables_error(RESOURCE_PROBLEM, "invalid to address.");     }     info->addr[1] = to; } /* 从字符串解析端口信息 */ static void parse_port(const char *orig_arg, struct xt_static_nat_tginfo *info) {     char *arg, *dash;     size_t delta, len;     u_int16_t from, to;     arg = strdup(orig_arg);     if (arg == NULL) {         xtables_error(RESOURCE_PROBLEM, "strdup");     }     len = strlen(arg);     dash = strchr(arg, '-');     if (!dash) {         xtables_error(RESOURCE_PROBLEM, "invalid argument.");     }     delta = dash - arg;     arg[delta] = 0;     from = atoi(arg);     if (from == 0){         xtables_error(RESOURCE_PROBLEM, "invalid from port.");     }     info->port[0] = htons(from);     arg += delta + 1;     to = atoi(arg);     if (to == 0){         xtables_error(RESOURCE_PROBLEM, "invalid to port.");     }     info->port[1] = htons(to); } /* iptables框架内的规则命令解析回调函数实现 */ static int static_2_way_nat_tg_parse(int c, char **argv, int invert, unsigned int *flags,                   const void *entry, struct xt_entry_target **target) {     int ret = false;     struct xt_static_nat_tginfo *info = (void *)(*target)->data;     switch (c) {     case 'a':         if (*flags & ADDR_OK) {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: multi addrmap.");         }         parse_addr(optarg, info);         /* 如果没有携带或者还没有解析到协议参数,则设置默认值 */         if (!(*flags & PROTO_OK)) {             info->proto = IPPROTO_MAX - 1;         }         /* 如果没有携带或者还没有解析到端口参数,则设置默认值 */         if (!(*flags & PORT_OK)) {             info->port[0] = 0;             info->port[1] = 0;         }         /* 如果没有携带或者还没有解析到网卡参数,则设置默认值 */         if (!(*flags & DEV_OK)) {             memset(&info->dev[0], 0, MAX_DEV_NAME);             strcpy(info->dev, "");         }         *flags |= ADDR_OK;         ret = true;         break;     case 'o':         if (*flags & PORT_OK) {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: multi portmap.");         }         parse_port(optarg, info);         *flags |= PORT_OK;         ret = true;         break;     case 'p':         if (*flags & PROTO_OK) {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: multi protocol.");         }         if (!strcmp (optarg, "tcp")) {             info->proto = IPPROTO_TCP;         } else if (!strcmp (optarg, "udp")) {             info->proto = IPPROTO_UDP;         } else {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: invalid type.");         }         *flags |= PROTO_OK;         ret = true;         break;     case 't':         if (*flags & TYPE_OK) {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: multi type.");         }         if (!strcmp (optarg, "src")) {             info->dir = DIR_SNAT;         } else if (!strcmp (optarg, "dst")) {             info->dir = DIR_DNAT;         } else {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: invalid type.");         }         *flags |= TYPE_OK;         ret = true;         break;     case 'd':         if (*flags & DEV_OK) {             xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: multi device.");         }         strncpy(info->dev, optarg, MAX_DEV_NAME);         *flags |= DEV_OK;         ret = true;         break;     }     return ret; } /* iptables框架内的规则命令解析完毕检查回调函数实现 */ static void static_2_way_nat_tg_check(unsigned int flags) {     /* 地址转换信息和类型[SNAT|DNAT]是必须要设置的 */     if (!(flags & ADDR_OK) || !(flags & TYPE_OK)) {         xtables_error(PARAMETER_PROBLEM, "STATIC-2-WAY-NAT: "             "\"--mapaddr a.b.c.d-A.B.C.D and --type [src|dst]\" is required.");     } } static void static_2_way_nat_tg_print(const void *entry, const struct xt_entry_target *target,                   int numeric) {     u_int32_t addr;     const struct xt_static_nat_tginfo *info = (const void *)target->data;     if (!info) {         return;     }     addr = info->addr[0];     printf(" --mapaddr %s-", xtables_ipaddr_to_numeric((struct in_addr *)&addr));     addr = info->addr[1];     printf("%s --type %s --proto %s --mapport %d-%d --dev %s",                 xtables_ipaddr_to_numeric((struct in_addr *)&addr),                 (info->dir == DIR_SNAT)?"src":"dst",                 (info->proto == IPPROTO_TCP)?"tcp":                                             ((info->proto == IPPROTO_UDP)?"udp":"all"),                 ntohs(info->port[0]),                 ntohs(info->port[1]),                 !strcmp(info->dev, "")?"all":info->dev); } /* iptables-save将按照该函数的实现来罗列规则 */ static void static_2_way_nat_tg_save(const void *entry, const struct xt_entry_target *target) {     u_int32_t addr;     const struct xt_static_nat_tginfo *info = (const void *)target->data;     if (!info) {         return;     }     addr = info->addr[0];     printf(" --mapaddr %s-", xtables_ipaddr_to_numeric((struct in_addr *)&addr));     addr = info->addr[1];     printf("%s --type %s --proto %s --mapport %d-%d --dev %s",                 xtables_ipaddr_to_numeric((struct in_addr *)&addr),                 (info->dir == DIR_SNAT)?"src":"dst",                 (info->proto == IPPROTO_TCP)?"tcp":                                             ((info->proto == IPPROTO_UDP)?"udp":"all"),                 ntohs(info->port[0]),                 ntohs(info->port[1]),                 !strcmp(info->dev, "")?"all":info->dev); } static struct xtables_target static_2_way_nat_tg_reg = {         .version       = XTABLES_VERSION,         .name          = "STATIC-2-WAY-NAT",         .family        = NFPROTO_IPV4,         .size          = XT_ALIGN(sizeof(struct xt_static_nat_tginfo)),         .userspacesize = XT_ALIGN(sizeof(struct xt_static_nat_tginfo)),         .help          = static_2_way_nat_tg_help,         .parse         = static_2_way_nat_tg_parse,         .final_check   = static_2_way_nat_tg_check,         .print         = static_2_way_nat_tg_print,         .save          = static_2_way_nat_tg_save,         .extra_opts    = static_2_way_nat_tg_opts, }; static __attribute__((constructor)) void static_2_way_nat_tg_ldr(void) {     xtables_register_target(&static_2_way_nat_tg_reg); }



xt_STATIC-2-WAY.h:这是一个头文件,内核模块和用户态文件公用

#ifndef _LINUX_NETFILTER_XT_TARGET_STATICNAT #define _LINUX_NETFILTER_XT_TARGET_STATICNAT 1 #define MAX_DEV_NAME    8 enum nat_dir {     DIR_SNAT,     DIR_DNAT,     DIR_NUM = 2 }; struct xt_static_nat_tginfo {     u_int32_t addr[DIR_NUM];     u_int16_t port[DIR_NUM];     u_int8_t proto;     u_int8_t dir;     char dev[MAX_DEV_NAME]; }; #endif /* _LINUX_NETFILTER_XT_TARGET_STATICNAT */



最后是一个Makefile:


obj-m += xt_STATIC-2-WAY-NAT.o all: module xtlib         @true module:         make -C /lib/modules/`uname -r`/build SUBDIRS=`pwd` modules         @rm -rf *.o .tmp_versions .*.mod.o .*.o.cmd *.mod.c .*.ko.cmd Module.symvers modules.order *.oo  xtlib:         gcc libxt_STATIC-2-WAY-NAT.c -fPIC -shared -o libxt_STATIC-2-WAY-NAT.so  -L/usr/local/lib -lxtables         @rm -rf *.o .tmp_versions .*.mod.o .*.o.cmd *.mod.c .*.ko.cmd Module.symvers modules.order *.oo  clean:         rm -rf *.ko *.o .tmp_versions .*.mod.o .*.o.cmd *.mod.c .*.ko.cmd Module.symvers modules.order *.so *.oo




关 于备份不得不多说几句,我现在有时候在工作中碰到问题的时候,参考的最多的就是我自己的博客,因为总是隐隐约约觉得自己曾经搞定过某件事,但只是曾经而 已,只要有迹可寻,找到那个曾经的方案即可,当然,如果现在重新从零开始最终也是可以搞定的,但是那将浪费很多时间。以前我喜欢在纸上做笔记,但是几乎不 做索引,随着本子越来越多越来越厚,就很难找到要找的东西了,后来就改成了在电脑里用Word,OneNote甚至记事本做笔记,可是最终的结果和用纸和 笔的效果一样,后来我觉得互联网上现成的索引做的不错,为何不让搜索引擎替我搜索呢?于是就改成博客的方式了,至于代码,我倒不是很看重,自娱自乐而已, 我主要想记录的是当时想了些什么而不是怎么做的。其实你有没有想过,你参考的最多还是自己以往的经验,而不是别人的,因此干嘛不把自己以往的想法录下来 呢?以前是写日记,现在是写博客,零散的想法以前可以随身带个小本子写随笔,现在有微博和朋友圈,其实万变不离其中。


【本文来源:韩国服务器 http://www.yidunidc.com欢迎留下您的宝贵建议】
上一篇:iptables日志探秘
下一篇:没有了
网友评论