gtp-rtnl: Split TID handling for GTPv0 and GTPv1
GTPv1 tunnel use separate 32bit Tunnel Endpoint Identifier's for each
direction while GTPv0 uses only one 64bit Tunnel IDentifier.
Signed-off-by: Andreas Schultz <aschultz@tpip.net>
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
diff --git a/src/gtp-genl.c b/src/gtp-genl.c
index ddb7cc7..3c7deed 100644
--- a/src/gtp-genl.c
+++ b/src/gtp-genl.c
@@ -49,8 +49,13 @@
mnl_attr_put_u32(nlh, GTPA_LINK, t->ifidx);
mnl_attr_put_u32(nlh, GTPA_SGSN_ADDRESS, t->sgsn_addr.s_addr);
mnl_attr_put_u32(nlh, GTPA_MS_ADDRESS, t->ms_addr.s_addr);
- mnl_attr_put_u64(nlh, GTPA_TID, t->tid);
- mnl_attr_put_u16(nlh, GTPA_FLOW, t->flowid);
+ if (t->gtp_version == GTP_V0) {
+ mnl_attr_put_u64(nlh, GTPA_TID, t->u.v0.tid);
+ mnl_attr_put_u16(nlh, GTPA_FLOW, t->u.v0.flowid);
+ } else if (t->gtp_version == GTP_V1) {
+ mnl_attr_put_u32(nlh, GTPA_I_TEI, t->u.v1.i_tei);
+ mnl_attr_put_u32(nlh, GTPA_O_TEI, t->u.v1.o_tei);
+ }
}
int gtp_add_tunnel(int genl_id, struct mnl_socket *nl, struct gtp_tunnel *t)
@@ -95,7 +100,15 @@
struct gtp_pdp {
uint32_t version;
- uint64_t tid;
+ union {
+ struct {
+ uint64_t tid;
+ } v0;
+ struct {
+ uint32_t i_tei;
+ uint32_t o_tei;
+ } v1;
+ } u;
struct in_addr sgsn_addr;
struct in_addr ms_addr;
};
@@ -115,6 +128,8 @@
return MNL_CB_ERROR;
}
break;
+ case GTPA_O_TEI:
+ case GTPA_I_TEI:
case GTPA_SGSN_ADDRESS:
case GTPA_MS_ADDRESS:
case GTPA_VERSION:
@@ -138,7 +153,11 @@
mnl_attr_parse(nlh, sizeof(*genl), genl_gtp_validate_cb, tb);
if (tb[GTPA_TID])
- pdp.tid = mnl_attr_get_u64(tb[GTPA_TID]);
+ pdp.u.v0.tid = mnl_attr_get_u64(tb[GTPA_TID]);
+ if (tb[GTPA_I_TEI])
+ pdp.u.v1.i_tei = mnl_attr_get_u32(tb[GTPA_I_TEI]);
+ if (tb[GTPA_O_TEI])
+ pdp.u.v1.o_tei = mnl_attr_get_u32(tb[GTPA_O_TEI]);
if (tb[GTPA_SGSN_ADDRESS]) {
pdp.sgsn_addr.s_addr =
mnl_attr_get_u32(tb[GTPA_SGSN_ADDRESS]);
@@ -151,7 +170,12 @@
}
printf("version %u ", pdp.version);
- printf("tid %"PRIu64" ms_addr %s ", pdp.tid, inet_ntoa(pdp.sgsn_addr));
+ if (pdp.version == GTP_V0)
+ printf("tid %"PRIu64" ms_addr %s ",
+ pdp.u.v0.tid, inet_ntoa(pdp.sgsn_addr));
+ else if (pdp.version == GTP_V1)
+ printf("tei %u/%u ms_addr %s ", pdp.u.v1.i_tei,
+ pdp.u.v1.o_tei, inet_ntoa(pdp.sgsn_addr));
printf("sgsn_addr %s\n", inet_ntoa(pdp.ms_addr));
return MNL_CB_OK;
diff --git a/src/gtp.c b/src/gtp.c
index 6e3d473..bb5eaaf 100644
--- a/src/gtp.c
+++ b/src/gtp.c
@@ -71,16 +71,28 @@
void gtp_tunnel_set_tid(struct gtp_tunnel *t, uint64_t tid)
{
- t->tid = tid;
+ t->u.v0.tid = tid;
}
EXPORT_SYMBOL(gtp_tunnel_set_tid);
void gtp_tunnel_set_flowid(struct gtp_tunnel *t, uint16_t flowid)
{
- t->flowid = flowid;
+ t->u.v0.flowid = flowid;
}
EXPORT_SYMBOL(gtp_tunnel_set_flowid);
+void gtp_tunnel_set_i_tei(struct gtp_tunnel *t, uint32_t i_tei)
+{
+ t->u.v1.i_tei = i_tei;
+}
+EXPORT_SYMBOL(gtp_tunnel_set_i_tei);
+
+void gtp_tunnel_set_o_tei(struct gtp_tunnel *t, uint32_t o_tei)
+{
+ t->u.v1.o_tei = o_tei;
+}
+EXPORT_SYMBOL(gtp_tunnel_set_o_tei);
+
const int gtp_tunnel_get_ifns(struct gtp_tunnel *t)
{
return t->ifns;
@@ -113,12 +125,24 @@
uint64_t gtp_tunnel_get_tid(struct gtp_tunnel *t)
{
- return t->tid;
+ return t->u.v0.tid;
}
EXPORT_SYMBOL(gtp_tunnel_get_tid);
uint16_t gtp_tunnel_get_flowid(struct gtp_tunnel *t)
{
- return t->flowid;
+ return t->u.v0.flowid;
}
EXPORT_SYMBOL(gtp_tunnel_get_flowid);
+
+uint32_t gtp_tunnel_get_i_tei(struct gtp_tunnel *t)
+{
+ return t->u.v1.i_tei;
+}
+EXPORT_SYMBOL(gtp_tunnel_get_i_tei);
+
+uint32_t gtp_tunnel_get_o_tei(struct gtp_tunnel *t)
+{
+ return t->u.v1.o_tei;
+}
+EXPORT_SYMBOL(gtp_tunnel_get_o_tei);
diff --git a/src/internal.h b/src/internal.h
index 68f0135..1754e3b 100644
--- a/src/internal.h
+++ b/src/internal.h
@@ -17,9 +17,17 @@
uint32_t ifidx;
struct in_addr ms_addr;
struct in_addr sgsn_addr;
- uint64_t tid;
- uint16_t flowid;
int gtp_version;
+ union {
+ struct {
+ uint64_t tid;
+ uint16_t flowid;
+ } v0;
+ struct {
+ uint32_t i_tei;
+ uint32_t o_tei;
+ } v1;
+ } u;
};
#endif
diff --git a/src/libgtpnl.map b/src/libgtpnl.map
index 2368467..9ca83c7 100644
--- a/src/libgtpnl.map
+++ b/src/libgtpnl.map
@@ -22,6 +22,8 @@
gtp_tunnel_set_version;
gtp_tunnel_set_tid;
gtp_tunnel_set_flowid;
+ gtp_tunnel_set_i_tei;
+ gtp_tunnel_set_o_tei;
gtp_tunnel_get_ifns;
gtp_tunnel_get_ifidx;
gtp_tunnel_get_ms_ip4;
@@ -29,6 +31,8 @@
gtp_tunnel_get_version;
gtp_tunnel_get_tid;
gtp_tunnel_get_flowid;
+ gtp_tunnel_get_i_tei;
+ gtp_tunnel_get_o_tei;
local: *;
};