From: Álvaro Neira Ayuso <alvaroneay@xxxxxxxxx> This patch allows to use DSCP support in nftables. Signed-off-by: Alvaro Neira Ayuso <alvaroneay@xxxxxxxxx> --- include/datatype.h | 2 ++ include/proto.h | 1 + src/parser.y | 2 ++ src/proto.c | 73 ++++++++++++++++++++++++++++++++++++++++++++++++++++ src/scanner.l | 1 + 5 files changed, 79 insertions(+) diff --git a/include/datatype.h b/include/datatype.h index 2ddab7d..987e95e 100644 --- a/include/datatype.h +++ b/include/datatype.h @@ -36,6 +36,7 @@ * @TYPE_ICMP6_TYPE: ICMPv6 type codes (integer subtype) * @TYPE_CT_LABEL: Conntrack Label (bitmask subtype) * @TYPE_TOS: Type of service (integer subtype) + * @TYPE_DSCP: Differentiated Services Field (integer subtype) */ enum datatypes { TYPE_INVALID, @@ -70,6 +71,7 @@ enum datatypes { TYPE_ICMP6_TYPE, TYPE_CT_LABEL, TYPE_TOS, + TYPE_DSCP, __TYPE_MAX }; #define TYPE_MAX (__TYPE_MAX - 1) diff --git a/include/proto.h b/include/proto.h index bd3701e..233a482 100644 --- a/include/proto.h +++ b/include/proto.h @@ -169,6 +169,7 @@ enum ip_hdr_fields { IPHDR_INVALID, IPHDR_VERSION, IPHDR_HDRLENGTH, + IPHDR_DSCP, IPHDR_TOS, IPHDR_LENGTH, IPHDR_ID, diff --git a/src/parser.y b/src/parser.y index 9c20737..1000cad 100644 --- a/src/parser.y +++ b/src/parser.y @@ -229,6 +229,7 @@ static void location_update(struct location *loc, struct location *rhs, int n) %token IP "ip" %token VERSION "version" %token HDRLENGTH "hdrlength" +%token DSCP "dscp" %token TOS "tos" %token LENGTH "length" %token FRAG_OFF "frag-off" @@ -1793,6 +1794,7 @@ ip_hdr_expr : IP ip_hdr_field ip_hdr_field : VERSION { $$ = IPHDR_VERSION; } | HDRLENGTH { $$ = IPHDR_HDRLENGTH; } | TOS { $$ = IPHDR_TOS; } + | DSCP { $$ = IPHDR_DSCP; } | LENGTH { $$ = IPHDR_LENGTH; } | ID { $$ = IPHDR_ID; } | FRAG_OFF { $$ = IPHDR_FRAG_OFF; } diff --git a/src/proto.c b/src/proto.c index cb27495..492e8f3 100644 --- a/src/proto.c +++ b/src/proto.c @@ -479,6 +479,77 @@ const struct proto_desc proto_sctp = { #include <netinet/ip.h> +static const struct symbol_table dscp_type_tbl = { + .symbols = { + SYMBOL("CS0", 0x00), + SYMBOL("CS1", 0x08), + SYMBOL("CS2", 0x10), + SYMBOL("CS3", 0x18), + SYMBOL("CS4", 0x20), + SYMBOL("CS5", 0x28), + SYMBOL("CS6", 0x30), + SYMBOL("CS7", 0x38), + SYMBOL("BE", 0x00), + SYMBOL("AF11", 0x0a), + SYMBOL("AF12", 0x0c), + SYMBOL("AF13", 0x0e), + SYMBOL("AF21", 0x12), + SYMBOL("AF22", 0x14), + SYMBOL("AF23", 0x16), + SYMBOL("AF31", 0x1a), + SYMBOL("AF32", 0x1c), + SYMBOL("AF33", 0x1e), + SYMBOL("AF41", 0x22), + SYMBOL("AF42", 0x24), + SYMBOL("AF43", 0x26), + SYMBOL("EF", 0x2e), + SYMBOL_LIST_END + }, +}; + +static struct error_record *dscp_type_parse(const struct expr *sym, + struct expr **res) +{ + struct error_record *erec; + const struct symbolic_constant *s; + + for (s = dscp_type_tbl.symbols; s->identifier != NULL; s++) { + if (!strcmp(sym->identifier, s->identifier)) { + *res = constant_expr_alloc(&sym->location, sym->dtype, + sym->dtype->byteorder, + sym->dtype->size, + &s->value); + return NULL; + } + } + + *res = NULL; + erec = sym->dtype->basetype->parse(sym, res); + if (erec != NULL) + return erec; + if (*res) + return NULL; + + return symbolic_constant_parse(sym, &dscp_type_tbl, res); +} + +static void dscp_type_print(const struct expr *expr) +{ + return symbolic_constant_print(&dscp_type_tbl, expr); +} + +static const struct datatype dscp_type = { + .type = TYPE_DSCP, + .name = "dscp_type", + .desc = "Differentiated Services Field", + .byteorder = BYTEORDER_BIG_ENDIAN, + .size = 6, + .basetype = &integer_type, + .basefmt = "0x%.2Zx", + .print = dscp_type_print, + .parse = dscp_type_parse, +}; + static const struct symbol_table tos_type_tbl = { .symbols = { SYMBOL("minimize-delay", IPTOS_LOWDELAY), @@ -558,6 +629,7 @@ const struct proto_desc proto_ip = { .templates = { [IPHDR_VERSION] = HDR_BITFIELD("version", &integer_type, 0, 4), [IPHDR_HDRLENGTH] = HDR_BITFIELD("hdrlength", &integer_type, 4, 4), + [IPHDR_DSCP] = IPHDR_TOS("dscp", &dscp_type), [IPHDR_TOS] = IPHDR_TOS("tos", &tos_type), [IPHDR_LENGTH] = IPHDR_FIELD("length", tot_len), [IPHDR_ID] = IPHDR_FIELD("id", id), @@ -868,5 +940,6 @@ static void __init proto_init(void) datatype_register(&arpop_type); datatype_register(ðertype_type); datatype_register(&icmp6_type_type); + datatype_register(&dscp_type); datatype_register(&tos_type); } diff --git a/src/scanner.l b/src/scanner.l index 801c030..c70012a 100644 --- a/src/scanner.l +++ b/src/scanner.l @@ -329,6 +329,7 @@ addrstring ({macaddr}|{ip4addr}|{ip6addr}) "ip" { return IP; } "version" { return VERSION; } "hdrlength" { return HDRLENGTH; } +"dscp" { return DSCP; } "tos" { return TOS; } "length" { return LENGTH; } "frag-off" { return FRAG_OFF; } -- 1.7.10.4 -- To unsubscribe from this list: send the line "unsubscribe netfilter-devel" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html