Support matching on ICMP type and code. Example usage:
tc qdisc add dev eth0 ingress tc filter add dev eth0 protocol ip parent ffff: flower \ indev eth0 ip_proto icmp type 8 code 0 action drop tc filter add dev eth0 protocol ipv6 parent ffff: flower \ indev eth0 ip_proto icmpv6 type 128 code 0 action drop Signed-off-by: Simon Horman <simon.hor...@netronome.com> --- v2 * Rebase * Do not run noths() on u8 entity --- man/man8/tc-flower.8 | 20 ++++++++--- tc/f_flower.c | 96 +++++++++++++++++++++++++++++++++++++++++++++++++--- 2 files changed, 106 insertions(+), 10 deletions(-) diff --git a/man/man8/tc-flower.8 b/man/man8/tc-flower.8 index 744c450b4531..90fdfbafc75b 100644 --- a/man/man8/tc-flower.8 +++ b/man/man8/tc-flower.8 @@ -29,7 +29,7 @@ flower \- flow based traffic control filter .IR PRIORITY " | " .BR vlan_eth_type " { " ipv4 " | " ipv6 " | " .IR ETH_TYPE " } | " -.BR ip_proto " { " tcp " | " udp " | " sctp " | " +.BR ip_proto " { " tcp " | " udp " | " sctp " | " icmp " | " icmpv6 " | " .IR IP_PROTO " } | { " .BR dst_ip " | " src_ip " } { " .IR ipv4_address " | " ipv6_address " } | { " @@ -98,7 +98,7 @@ or an unsigned 16bit value in hexadecimal format. Match on layer four protocol. .I IP_PROTO may be -.BR tcp ", " udp ", " sctp +.BR tcp ", " udp ", " sctp ", " icmp ", " icmpv6 or an unsigned 8bit value in hexadecimal format. .TP .BI dst_ip " ADDRESS" @@ -117,6 +117,13 @@ Match on layer 4 protocol source or destination port number. Only available for .BR ip_proto " values " udp ", " tcp " and " sctp which have to be specified in beforehand. .TP +.BI type " NUMBER" +.TQ +.BI code " NUMBER" +Match on ICMP type or code. Only available for +.BR ip_proto " values " icmp " and " icmpv6 +which have to be specified in beforehand. +.TP .BI enc_key_id " NUMBER" .TQ .BI enc_dst_ip " ADDRESS" @@ -135,13 +142,16 @@ have no dependency, layer three matches (\fBip_proto\fR, \fBdst_ip\fR and \fBsrc_ip\fR) depend on the .B protocol -option of tc filter -and finally layer four matches +option of tc filter, layer four port matches (\fBdst_port\fR and \fBsrc_port\fR) depend on .B ip_proto being set to -.BR tcp ", " udp " or " sctp. +.BR tcp ", " udp " or " sctp, +and finally ICMP matches (\fBcode\fR and \fBtype\fR) depend on +.B ip_proto +being set to +.BR icmp " or " icmpv6. .P There can be only used one mask per one prio. If user needs to specify different mask, he has to use different prio. diff --git a/tc/f_flower.c b/tc/f_flower.c index 9899b0175a6a..5dac427e5454 100644 --- a/tc/f_flower.c +++ b/tc/f_flower.c @@ -28,6 +28,11 @@ enum flower_endpoint { FLOWER_ENDPOINT_DST }; +enum flower_icmp_field { + FLOWER_ICMP_FIELD_TYPE, + FLOWER_ICMP_FIELD_CODE +}; + static void explain(void) { fprintf(stderr, @@ -42,11 +47,13 @@ static void explain(void) " vlan_ethtype [ ipv4 | ipv6 | ETH-TYPE ] |\n" " dst_mac MAC-ADDR |\n" " src_mac MAC-ADDR |\n" - " ip_proto [tcp | udp | sctp | IP-PROTO ] |\n" + " ip_proto [tcp | udp | sctp | icmp | icmpv6 | IP-PROTO ] |\n" " dst_ip [ IPV4-ADDR | IPV6-ADDR ] |\n" " src_ip [ IPV4-ADDR | IPV6-ADDR ] |\n" " dst_port PORT-NUMBER |\n" " src_port PORT-NUMBER |\n" + " type ICMP-TYPE |\n" + " code ICMP-CODE }\n" " enc_dst_ip [ IPV4-ADDR | IPV6-ADDR ] |\n" " enc_src_ip [ IPV4-ADDR | IPV6-ADDR ] |\n" " enc_key_id [ KEY-ID ] }\n" @@ -98,16 +105,23 @@ static int flower_parse_ip_proto(char *str, __be16 eth_type, int type, int ret; __u8 ip_proto; - if (eth_type != htons(ETH_P_IP) && eth_type != htons(ETH_P_IPV6)) { - fprintf(stderr, "Illegal \"eth_type\" for ip proto\n"); - return -1; - } + if (eth_type != htons(ETH_P_IP) && eth_type != htons(ETH_P_IPV6)) + goto err; + if (matches(str, "tcp") == 0) { ip_proto = IPPROTO_TCP; } else if (matches(str, "udp") == 0) { ip_proto = IPPROTO_UDP; } else if (matches(str, "sctp") == 0) { ip_proto = IPPROTO_SCTP; + } else if (matches(str, "icmp") == 0) { + if (eth_type != htons(ETH_P_IP)) + goto err; + ip_proto = IPPROTO_ICMP; + } else if (matches(str, "icmpv6") == 0) { + if (eth_type != htons(ETH_P_IPV6)) + goto err; + ip_proto = IPPROTO_ICMPV6; } else { ret = get_u8(&ip_proto, str, 16); if (ret) @@ -116,6 +130,10 @@ static int flower_parse_ip_proto(char *str, __be16 eth_type, int type, addattr8(n, MAX_MSG, type, ip_proto); *p_ip_proto = ip_proto; return 0; + +err: + fprintf(stderr, "Illegal \"eth_type\" for ip proto\n"); + return -1; } static int flower_parse_ip_addr(char *str, __be16 eth_type, @@ -171,6 +189,41 @@ static int flower_parse_ip_addr(char *str, __be16 eth_type, return 0; } +static int flower_icmp_attr_type(__be16 eth_type, __u8 ip_proto, + enum flower_icmp_field field) +{ + if (eth_type == htons(ETH_P_IP) && ip_proto == IPPROTO_ICMP) + return field == FLOWER_ICMP_FIELD_CODE ? + TCA_FLOWER_KEY_ICMPV4_CODE : + TCA_FLOWER_KEY_ICMPV4_TYPE; + else if (eth_type == htons(ETH_P_IPV6) && ip_proto == IPPROTO_ICMPV6) + return field == FLOWER_ICMP_FIELD_CODE ? + TCA_FLOWER_KEY_ICMPV6_CODE : + TCA_FLOWER_KEY_ICMPV6_TYPE; + + return -1; +} + +static int flower_parse_icmp(char *str, __u16 eth_type, __u8 ip_proto, + enum flower_icmp_field field, struct nlmsghdr *n) +{ + int ret; + int type; + uint8_t value; + + type = flower_icmp_attr_type(eth_type, ip_proto, field); + if (type < 0) + return -1; + + ret = get_u8(&value, str, 10); + if (ret) + return -1; + + addattr8(n, MAX_MSG, type, value); + + return 0; +} + static int flower_port_attr_type(__u8 ip_proto, enum flower_endpoint endpoint) { if (ip_proto == IPPROTO_TCP) @@ -381,6 +434,22 @@ static int flower_parse_opt(struct filter_util *qu, char *handle, fprintf(stderr, "Illegal \"src_port\"\n"); return -1; } + } else if (matches(*argv, "type") == 0) { + NEXT_ARG(); + ret = flower_parse_icmp(*argv, eth_type, ip_proto, + FLOWER_ICMP_FIELD_TYPE, n); + if (ret < 0) { + fprintf(stderr, "Illegal \"icmp type\"\n"); + return -1; + } + } else if (matches(*argv, "code") == 0) { + NEXT_ARG(); + ret = flower_parse_icmp(*argv, eth_type, ip_proto, + FLOWER_ICMP_FIELD_CODE, n); + if (ret < 0) { + fprintf(stderr, "Illegal \"icmp code\"\n"); + return -1; + } } else if (matches(*argv, "enc_dst_ip") == 0) { NEXT_ARG(); ret = flower_parse_ip_addr(*argv, 0, @@ -526,6 +595,10 @@ static void flower_print_ip_proto(FILE *f, __u8 *p_ip_proto, fprintf(f, "udp"); else if (ip_proto == IPPROTO_SCTP) fprintf(f, "sctp"); + else if (ip_proto == IPPROTO_ICMP) + fprintf(f, "icmp"); + else if (ip_proto == IPPROTO_ICMPV6) + fprintf(f, "icmpv6"); else fprintf(f, "%02x", ip_proto); *p_ip_proto = ip_proto; @@ -581,6 +654,12 @@ static void flower_print_key_id(FILE *f, const char *name, fprintf(f, "\n %s %d", name, rta_getattr_be32(attr)); } +static void flower_print_icmp(FILE *f, char *name, struct rtattr *attr) +{ + if (attr) + fprintf(f, "\n %s %d", name, rta_getattr_u8(attr)); +} + static int flower_print_opt(struct filter_util *qu, FILE *f, struct rtattr *opt, __u32 handle) { @@ -649,6 +728,13 @@ static int flower_print_opt(struct filter_util *qu, FILE *f, if (nl_type >= 0) flower_print_port(f, "src_port", tb[nl_type]); + nl_type = flower_icmp_attr_type(eth_type, ip_proto, false); + if (nl_type >= 0) + flower_print_icmp(f, "icmp_type", tb[nl_type]); + nl_type = flower_icmp_attr_type(eth_type, ip_proto, true); + if (nl_type >= 0) + flower_print_icmp(f, "icmp_code", tb[nl_type]); + flower_print_ip_addr(f, "enc_dst_ip", tb[TCA_FLOWER_KEY_ENC_IPV4_DST_MASK] ? htons(ETH_P_IP) : htons(ETH_P_IPV6), -- 2.7.0.rc3.207.g0ac5344