libnftnl  1.2.8
obj/synproxy.c
1 #include <stdio.h>
2 #include <stdint.h>
3 #include <arpa/inet.h>
4 #include <errno.h>
5 #include <inttypes.h>
6 
7 #include <linux/netfilter/nf_tables.h>
8 
9 #include "internal.h"
10 #include <libmnl/libmnl.h>
11 #include <libnftnl/object.h>
12 
13 #include "obj.h"
14 
15 static int nftnl_obj_synproxy_set(struct nftnl_obj *e, uint16_t type,
16  const void *data, uint32_t data_len)
17 {
18  struct nftnl_obj_synproxy *synproxy = nftnl_obj_data(e);
19 
20  switch (type) {
21  case NFTNL_OBJ_SYNPROXY_MSS:
22  memcpy(&synproxy->mss, data, data_len);
23  break;
24  case NFTNL_OBJ_SYNPROXY_WSCALE:
25  memcpy(&synproxy->wscale, data, data_len);
26  break;
27  case NFTNL_OBJ_SYNPROXY_FLAGS:
28  memcpy(&synproxy->flags, data, data_len);
29  break;
30  }
31  return 0;
32 }
33 
34 static const void *nftnl_obj_synproxy_get(const struct nftnl_obj *e,
35  uint16_t type, uint32_t *data_len)
36 {
37  struct nftnl_obj_synproxy *synproxy = nftnl_obj_data(e);
38 
39  switch (type) {
40  case NFTNL_OBJ_SYNPROXY_MSS:
41  *data_len = sizeof(synproxy->mss);
42  return &synproxy->mss;
43  case NFTNL_OBJ_SYNPROXY_WSCALE:
44  *data_len = sizeof(synproxy->wscale);
45  return &synproxy->wscale;
46  case NFTNL_OBJ_SYNPROXY_FLAGS:
47  *data_len = sizeof(synproxy->flags);
48  return &synproxy->flags;
49  }
50  return NULL;
51 }
52 
53 static int nftnl_obj_synproxy_cb(const struct nlattr *attr, void *data)
54 {
55  int type = mnl_attr_get_type(attr);
56  const struct nlattr **tb = data;
57 
58  if (mnl_attr_type_valid(attr, NFTA_SYNPROXY_MAX) < 0)
59  return MNL_CB_OK;
60 
61  switch (type) {
62  case NFTA_SYNPROXY_MSS:
63  if (mnl_attr_validate(attr, MNL_TYPE_U16) < 0)
64  abi_breakage();
65  break;
66  case NFTA_SYNPROXY_WSCALE:
67  if (mnl_attr_validate(attr, MNL_TYPE_U8) < 0)
68  abi_breakage();
69  break;
70  case NFTA_SYNPROXY_FLAGS:
71  if (mnl_attr_validate(attr, MNL_TYPE_U32) < 0)
72  abi_breakage();
73  break;
74  }
75 
76  tb[type] = attr;
77  return MNL_CB_OK;
78 }
79 
80 static void nftnl_obj_synproxy_build(struct nlmsghdr *nlh,
81  const struct nftnl_obj *e)
82 {
83  struct nftnl_obj_synproxy *synproxy = nftnl_obj_data(e);
84 
85  if (e->flags & (1 << NFTNL_OBJ_SYNPROXY_MSS))
86  mnl_attr_put_u16(nlh, NFTA_SYNPROXY_MSS, htons(synproxy->mss));
87  if (e->flags & (1 << NFTNL_OBJ_SYNPROXY_WSCALE))
88  mnl_attr_put_u8(nlh, NFTA_SYNPROXY_WSCALE, synproxy->wscale);
89  if (e->flags & (1 << NFTNL_OBJ_SYNPROXY_FLAGS))
90  mnl_attr_put_u32(nlh, NFTA_SYNPROXY_FLAGS,
91  htonl(synproxy->flags));
92 }
93 
94 static int nftnl_obj_synproxy_parse(struct nftnl_obj *e, struct nlattr *attr)
95 {
96  struct nftnl_obj_synproxy *synproxy = nftnl_obj_data(e);
97  struct nlattr *tb[NFTA_SYNPROXY_MAX + 1] = {};
98 
99  if (mnl_attr_parse_nested(attr, nftnl_obj_synproxy_cb, tb) < 0)
100  return -1;
101 
102  if (tb[NFTA_SYNPROXY_MSS]) {
103  synproxy->mss = ntohs(mnl_attr_get_u16(tb[NFTA_SYNPROXY_MSS]));
104  e->flags |= (1 << NFTNL_OBJ_SYNPROXY_MSS);
105  }
106  if (tb[NFTA_SYNPROXY_WSCALE]) {
107  synproxy->wscale = mnl_attr_get_u8(tb[NFTA_SYNPROXY_WSCALE]);
108  e->flags |= (1 << NFTNL_OBJ_SYNPROXY_WSCALE);
109  }
110  if (tb[NFTA_SYNPROXY_FLAGS]) {
111  synproxy->flags = ntohl(mnl_attr_get_u32(tb[NFTA_SYNPROXY_FLAGS]));
112  e->flags |= (1 << NFTNL_OBJ_SYNPROXY_FLAGS);
113  }
114 
115  return 0;
116 }
117 
118 static int nftnl_obj_synproxy_snprintf(char *buf, size_t len,
119  uint32_t flags,
120  const struct nftnl_obj *e)
121 {
122  struct nftnl_obj_synproxy *synproxy = nftnl_obj_data(e);
123  int ret, offset = 0;
124 
125  if (e->flags & (1 << NFTNL_OBJ_SYNPROXY_MSS) &&
126  e->flags & (1 << NFTNL_OBJ_SYNPROXY_WSCALE)) {
127  ret = snprintf(buf, len, "mss %u wscale %u ", synproxy->mss,
128  synproxy->wscale);
129  SNPRINTF_BUFFER_SIZE(ret, len, offset);
130  }
131 
132  return offset;
133 }
134 
135 static struct attr_policy obj_synproxy_attr_policy[__NFTNL_OBJ_SYNPROXY_MAX] = {
136  [NFTNL_OBJ_SYNPROXY_MSS] = { .maxlen = sizeof(uint16_t) },
137  [NFTNL_OBJ_SYNPROXY_WSCALE] = { .maxlen = sizeof(uint8_t) },
138  [NFTNL_OBJ_SYNPROXY_FLAGS] = { .maxlen = sizeof(uint32_t) },
139 };
140 
141 struct obj_ops obj_ops_synproxy = {
142  .name = "synproxy",
143  .type = NFT_OBJECT_SYNPROXY,
144  .alloc_len = sizeof(struct nftnl_obj_synproxy),
145  .nftnl_max_attr = __NFTNL_OBJ_SYNPROXY_MAX - 1,
146  .attr_policy = obj_synproxy_attr_policy,
147  .set = nftnl_obj_synproxy_set,
148  .get = nftnl_obj_synproxy_get,
149  .parse = nftnl_obj_synproxy_parse,
150  .build = nftnl_obj_synproxy_build,
151  .output = nftnl_obj_synproxy_snprintf,
152 };