diff options
Diffstat (limited to 'src/build.c')
-rw-r--r-- | src/build.c | 33 |
1 files changed, 28 insertions, 5 deletions
diff --git a/src/build.c b/src/build.c index 9ba8b57..5403300 100644 --- a/src/build.c +++ b/src/build.c @@ -66,9 +66,16 @@ ct_build_u32(const struct nf_conntrack *ct, int a, struct nethdr *n, int b) } static inline void -ct_build_str(const struct nf_conntrack *ct, int a, struct nethdr *n, int b) +ct_build_u128(const struct nf_conntrack *ct, int a, struct nethdr *n, int b) { const char *data = nfct_get_attr(ct, a); + addattr(n, b, data, sizeof(uint32_t) * 4); +} + +static inline void +ct_build_str(const struct nf_conntrack *ct, int a, struct nethdr *n, int b) +{ + const void *data = nfct_get_attr(ct, a); addattr(n, b, data, strlen(data)+1); } @@ -258,10 +265,26 @@ void ct2msg(const struct nf_conntrack *ct, struct nethdr *n) } /* NAT */ - if (nfct_getobjopt(ct, NFCT_GOPT_IS_SNAT)) - ct_build_u32(ct, ATTR_REPL_IPV4_DST, n, NTA_SNAT_IPV4); - if (nfct_getobjopt(ct, NFCT_GOPT_IS_DNAT)) - ct_build_u32(ct, ATTR_REPL_IPV4_SRC, n, NTA_DNAT_IPV4); + switch (nfct_get_attr_u8(ct, ATTR_ORIG_L3PROTO)) { + case AF_INET: + if (nfct_getobjopt(ct, NFCT_GOPT_IS_SNAT)) + ct_build_u32(ct, ATTR_REPL_IPV4_DST, n, NTA_SNAT_IPV4); + if (nfct_getobjopt(ct, NFCT_GOPT_IS_DNAT)) + ct_build_u32(ct, ATTR_REPL_IPV4_SRC, n, NTA_DNAT_IPV4); + break; + case AF_INET6: + if (nfct_getobjopt(ct, NFCT_GOPT_IS_SNAT)) { + ct_build_u128(ct, ATTR_REPL_IPV6_DST, n, + NTA_SNAT_IPV6); + } + if (nfct_getobjopt(ct, NFCT_GOPT_IS_DNAT)) { + ct_build_u128(ct, ATTR_REPL_IPV6_SRC, n, + NTA_DNAT_IPV6); + } + break; + default: + break; + } if (nfct_getobjopt(ct, NFCT_GOPT_IS_SPAT)) ct_build_u16(ct, ATTR_REPL_PORT_DST, n, NTA_SPAT_PORT); if (nfct_getobjopt(ct, NFCT_GOPT_IS_DPAT)) |