X-Git-Url: http://git.bitcoin.ninja/index.cgi?p=flowspec-xdp;a=blobdiff_plain;f=genrules.py;h=4468c1e00e52d58fb12c9d4769ae12c95c7d2704;hp=edef6175e264f03f6edec910069c55ea92ae1c41;hb=a8be3b609743627f58a464d7b75217a778eb4fa9;hpb=827e945d6361b743f2b9c765758e914ccc2a689e

diff --git a/genrules.py b/genrules.py
index edef617..4468c1e 100755
--- a/genrules.py
+++ b/genrules.py
@@ -232,10 +232,10 @@ def dscp_to_rule(proto, rules):
 def port_to_rule(ty, rules):
     if ty == "port" :
         ast = parse_ast(rules, parse_numbers_expr)
-        return "if (tcp == NULL && udp == NULL) break;\nif (!( " + ast.write("sport", "dport") + " )) break;"
+        return "if (!ports_valid) break;\nif (!( " + ast.write("sport", "dport") + " )) break;"
 
     ast = parse_ast(rules, parse_numbers_expr)
-    return "if (tcp == NULL && udp == NULL) break;\nif (!( " + ast.write(ty) + " )) break;"
+    return "if (!ports_valid) break;\nif (!( " + ast.write(ty) + " )) break;"
 
 def tcp_flags_to_rule(rules):
     ast = parse_ast(rules, parse_bit_expr)