Added support for filtering based on port ranges.
UAPI changes have been accepted into net-next.

Example:
1. Match on a port range:
-------------------------
$ tc filter add dev enp4s0 protocol ip parent ffff:\
  prio 1 flower ip_proto tcp dst_port range 20-30 skip_hw\
  action drop

$ tc -s filter show dev enp4s0 parent ffff:
filter protocol ip pref 1 flower chain 0
filter protocol ip pref 1 flower chain 0 handle 0x1
  eth_type ipv4
  ip_proto tcp
  dst_port range 20-30
  skip_hw
  not_in_hw
        action order 1: gact action drop
         random type none pass val 0
         index 1 ref 1 bind 1 installed 85 sec used 3 sec
        Action statistics:
        Sent 460 bytes 10 pkt (dropped 10, overlimits 0 requeues 0)
        backlog 0b 0p requeues 0

2. Match on IP address and port range:
--------------------------------------
$ tc filter add dev enp4s0 protocol ip parent ffff:\
  prio 1 flower dst_ip 192.168.1.1 ip_proto tcp dst_port range 100-200\
  skip_hw action drop

$ tc -s filter show dev enp4s0 parent ffff:
filter protocol ip pref 1 flower chain 0 handle 0x2
  eth_type ipv4
  ip_proto tcp
  dst_ip 192.168.1.1
  dst_port range 100-200
  skip_hw
  not_in_hw
        action order 1: gact action drop
         random type none pass val 0
         index 2 ref 1 bind 1 installed 58 sec used 2 sec
        Action statistics:
        Sent 920 bytes 20 pkt (dropped 20, overlimits 0 requeues 0)
        backlog 0b 0p requeues 0

v3:
Modified flower_port_range_attr_type calls.

v2:
Addressed Jiri's comment to sync output format with input

Signed-off-by: Amritha Nambiar <amritha.namb...@intel.com>
---
 include/uapi/linux/pkt_cls.h |    7 ++
 tc/f_flower.c                |  143 +++++++++++++++++++++++++++++++++++++++---
 2 files changed, 140 insertions(+), 10 deletions(-)

diff --git a/include/uapi/linux/pkt_cls.h b/include/uapi/linux/pkt_cls.h
index 401d0c1..95d0db2 100644
--- a/include/uapi/linux/pkt_cls.h
+++ b/include/uapi/linux/pkt_cls.h
@@ -485,6 +485,11 @@ enum {
 
        TCA_FLOWER_IN_HW_COUNT,
 
+       TCA_FLOWER_KEY_PORT_SRC_MIN,    /* be16 */
+       TCA_FLOWER_KEY_PORT_SRC_MAX,    /* be16 */
+       TCA_FLOWER_KEY_PORT_DST_MIN,    /* be16 */
+       TCA_FLOWER_KEY_PORT_DST_MAX,    /* be16 */
+
        __TCA_FLOWER_MAX,
 };
 
@@ -518,6 +523,8 @@ enum {
        TCA_FLOWER_KEY_FLAGS_FRAG_IS_FIRST = (1 << 1),
 };
 
+#define TCA_FLOWER_MASK_FLAGS_RANGE    (1 << 0) /* Range-based match */
+
 /* Match-all classifier */
 
 enum {
diff --git a/tc/f_flower.c b/tc/f_flower.c
index 65fca04..9bddf7b 100644
--- a/tc/f_flower.c
+++ b/tc/f_flower.c
@@ -494,6 +494,68 @@ static int flower_parse_port(char *str, __u8 ip_proto,
        return 0;
 }
 
+static int flower_port_range_attr_type(__u8 ip_proto, enum flower_endpoint 
type,
+                                      __be16 *min_port_type,
+                                      __be16 *max_port_type)
+{
+       if (ip_proto == IPPROTO_TCP || ip_proto == IPPROTO_UDP ||
+           ip_proto == IPPROTO_SCTP) {
+               if (type == FLOWER_ENDPOINT_SRC) {
+                       *min_port_type = TCA_FLOWER_KEY_PORT_SRC_MIN;
+                       *max_port_type = TCA_FLOWER_KEY_PORT_SRC_MAX;
+               } else {
+                       *min_port_type = TCA_FLOWER_KEY_PORT_DST_MIN;
+                       *max_port_type = TCA_FLOWER_KEY_PORT_DST_MAX;
+               }
+       } else {
+               return -1;
+       }
+
+       return 0;
+}
+
+static int flower_parse_port_range(__be16 *min, __be16 *max, __u8 ip_proto,
+                                  enum flower_endpoint endpoint,
+                                  struct nlmsghdr *n)
+{
+       __be16 min_port_type, max_port_type;
+
+       if (flower_port_range_attr_type(ip_proto, endpoint, &min_port_type,
+                                       &max_port_type))
+               return -1;
+
+       addattr16(n, MAX_MSG, min_port_type, *min);
+       addattr16(n, MAX_MSG, max_port_type, *max);
+
+       return 0;
+}
+
+static int get_range(__be16 *min, __be16 *max, char *argv)
+{
+       char *r;
+
+       r = strchr(argv, '-');
+       if (r) {
+               *r = '\0';
+               if (get_be16(min, argv, 10)) {
+                       fprintf(stderr, "invalid min range\n");
+                       return -1;
+               }
+               if (get_be16(max, r + 1, 10)) {
+                       fprintf(stderr, "invalid max range\n");
+                       return -1;
+               }
+               if (htons(*max) <= htons(*min)) {
+                       fprintf(stderr, "max value should be greater than min 
value\n");
+                       return -1;
+               }
+       } else {
+               fprintf(stderr, "Illegal range format\n");
+               return -1;
+       }
+       return 0;
+}
+
 #define TCP_FLAGS_MAX_MASK 0xfff
 
 static int flower_parse_tcp_flags(char *str, int flags_type, int mask_type,
@@ -1061,20 +1123,54 @@ static int flower_parse_opt(struct filter_util *qu, 
char *handle,
                                return -1;
                        }
                } else if (matches(*argv, "dst_port") == 0) {
+                       __be16 min, max;
+
                        NEXT_ARG();
-                       ret = flower_parse_port(*argv, ip_proto,
-                                               FLOWER_ENDPOINT_DST, n);
-                       if (ret < 0) {
-                               fprintf(stderr, "Illegal \"dst_port\"\n");
-                               return -1;
+                       if (matches(*argv, "range") == 0) {
+                               NEXT_ARG();
+                               ret = get_range(&min, &max, *argv);
+                               if (ret < 0)
+                                       return -1;
+                               ret = flower_parse_port_range(&min, &max,
+                                                             ip_proto,
+                                                             
FLOWER_ENDPOINT_DST,
+                                                             n);
+                               if (ret < 0) {
+                                       fprintf(stderr, "Illegal \"dst_port 
range\"\n");
+                                       return -1;
+                               }
+                       } else {
+                               ret = flower_parse_port(*argv, ip_proto,
+                                                       FLOWER_ENDPOINT_DST, n);
+                               if (ret < 0) {
+                                       fprintf(stderr, "Illegal 
\"dst_port\"\n");
+                                       return -1;
+                               }
                        }
                } else if (matches(*argv, "src_port") == 0) {
+                       __be16 min, max;
+
                        NEXT_ARG();
-                       ret = flower_parse_port(*argv, ip_proto,
-                                               FLOWER_ENDPOINT_SRC, n);
-                       if (ret < 0) {
-                               fprintf(stderr, "Illegal \"src_port\"\n");
-                               return -1;
+                       if (matches(*argv, "range") == 0) {
+                               NEXT_ARG();
+                               ret = get_range(&min, &max, *argv);
+                               if (ret < 0)
+                                       return -1;
+                               ret = flower_parse_port_range(&min, &max,
+                                                             ip_proto,
+                                                             
FLOWER_ENDPOINT_SRC,
+                                                             n);
+                               if (ret < 0) {
+                                       fprintf(stderr, "Illegal \"src_port 
range\"\n");
+                                       return -1;
+                               }
+                       } else {
+                               ret = flower_parse_port(*argv, ip_proto,
+                                                       FLOWER_ENDPOINT_SRC, n);
+                               if (ret < 0) {
+                                       fprintf(stderr, "Illegal 
\"src_port\"\n");
+                                       return -1;
+                               }
                        }
                } else if (matches(*argv, "tcp_flags") == 0) {
                        NEXT_ARG();
@@ -1490,6 +1586,22 @@ static void flower_print_port(char *name, struct rtattr 
*attr)
        print_hu(PRINT_ANY, name, namefrm, rta_getattr_be16(attr));
 }
 
+static void flower_print_port_range(char *name, struct rtattr *min_attr,
+                                   struct rtattr *max_attr)
+{
+       SPRINT_BUF(namefrm);
+       SPRINT_BUF(out);
+       size_t done;
+
+       if (!min_attr || !max_attr)
+               return;
+
+       done = sprintf(out, "%u", rta_getattr_be16(min_attr));
+       sprintf(out + done, "-%u", rta_getattr_be16(max_attr));
+       sprintf(namefrm, "\n  %s %%s", name);
+       print_string(PRINT_ANY, name, namefrm, out);
+}
+
 static void flower_print_tcp_flags(const char *name, struct rtattr *flags_attr,
                                   struct rtattr *mask_attr)
 {
@@ -1678,6 +1790,7 @@ static int flower_print_opt(struct filter_util *qu, FILE 
*f,
                            struct rtattr *opt, __u32 handle)
 {
        struct rtattr *tb[TCA_FLOWER_MAX + 1];
+       __be16 min_port_type, max_port_type;
        int nl_type, nl_mask_type;
        __be16 eth_type = 0;
        __u8 ip_proto = 0xff;
@@ -1796,6 +1909,16 @@ static int flower_print_opt(struct filter_util *qu, FILE 
*f,
        if (nl_type >= 0)
                flower_print_port("src_port", tb[nl_type]);
 
+       if (!flower_port_range_attr_type(ip_proto, FLOWER_ENDPOINT_DST,
+                                        &min_port_type, &max_port_type))
+               flower_print_port_range("dst_port range",
+                                       tb[min_port_type], tb[max_port_type]);
+
+       if (!flower_port_range_attr_type(ip_proto, FLOWER_ENDPOINT_SRC,
+                                        &min_port_type, &max_port_type))
+               flower_print_port_range("src_port range",
+                                       tb[min_port_type], tb[max_port_type]);
+
        flower_print_tcp_flags("tcp_flags", tb[TCA_FLOWER_KEY_TCP_FLAGS],
                               tb[TCA_FLOWER_KEY_TCP_FLAGS_MASK]);
 

Reply via email to