Added support for filtering based on port ranges.

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_min 20
  dst_port_max 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 181 sec used 5 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_min 100
  dst_port_max 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 28 sec used 6 sec
        Action statistics:
        Sent 460 bytes 10 pkt (dropped 10, overlimits 0 requeues 0)
        backlog 0b 0p requeues 0

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

diff --git a/include/uapi/linux/pkt_cls.h b/include/uapi/linux/pkt_cls.h
index be382fb..3d9727f 100644
--- a/include/uapi/linux/pkt_cls.h
+++ b/include/uapi/linux/pkt_cls.h
@@ -405,6 +405,11 @@ enum {
        TCA_FLOWER_KEY_UDP_SRC,         /* be16 */
        TCA_FLOWER_KEY_UDP_DST,         /* be16 */
 
+       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_FLAGS,
        TCA_FLOWER_KEY_VLAN_ID,         /* be16 */
        TCA_FLOWER_KEY_VLAN_PRIO,       /* u8   */
diff --git a/tc/f_flower.c b/tc/f_flower.c
index 59e5f57..1a7bc80 100644
--- a/tc/f_flower.c
+++ b/tc/f_flower.c
@@ -33,6 +33,11 @@ enum flower_endpoint {
        FLOWER_ENDPOINT_DST
 };
 
+struct range_type {
+       __be16 min_port_type;
+       __be16 max_port_type;
+};
+
 enum flower_icmp_field {
        FLOWER_ICMP_FIELD_TYPE,
        FLOWER_ICMP_FIELD_CODE
@@ -493,6 +498,64 @@ 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,
+                                      struct range_type *range)
+{
+       if (ip_proto == IPPROTO_TCP || ip_proto == IPPROTO_UDP ||
+           ip_proto == IPPROTO_SCTP) {
+               if (type == FLOWER_ENDPOINT_SRC) {
+                       range->min_port_type = TCA_FLOWER_KEY_PORT_SRC_MIN;
+                       range->max_port_type = TCA_FLOWER_KEY_PORT_SRC_MAX;
+               } else {
+                       range->min_port_type = TCA_FLOWER_KEY_PORT_DST_MIN;
+                       range->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)
+{
+       struct range_type range;
+
+       flower_port_range_attr_type(ip_proto, endpoint, &range);
+       addattr16(n, MAX_MSG, range.min_port_type, *min);
+       addattr16(n, MAX_MSG, range.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,
@@ -887,20 +950,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();
@@ -1309,6 +1406,17 @@ 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 *attr)
+{
+       SPRINT_BUF(namefrm);
+
+       if (!attr)
+               return;
+
+       sprintf(namefrm, "\n  %s %%u", name);
+       print_uint(PRINT_ANY, name, namefrm, rta_getattr_be16(attr));
+}
+
 static void flower_print_tcp_flags(const char *name, struct rtattr *flags_attr,
                                   struct rtattr *mask_attr)
 {
@@ -1398,6 +1506,7 @@ static int flower_print_opt(struct filter_util *qu, FILE 
*f,
                            struct rtattr *opt, __u32 handle)
 {
        struct rtattr *tb[TCA_FLOWER_MAX + 1];
+       struct range_type range;
        int nl_type, nl_mask_type;
        __be16 eth_type = 0;
        __u8 ip_proto = 0xff;
@@ -1516,6 +1625,22 @@ 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, &range)
+           == 0) {
+               flower_print_port_range("dst_port_min",
+                                       tb[range.min_port_type]);
+               flower_print_port_range("dst_port_max",
+                                       tb[range.max_port_type]);
+       }
+
+       if (flower_port_range_attr_type(ip_proto, FLOWER_ENDPOINT_SRC, &range)
+           == 0) {
+               flower_print_port_range("src_port_min",
+                                       tb[range.min_port_type]);
+               flower_print_port_range("src_port_max",
+                                       tb[range.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