Merge pull request #74 from si74/new_match_ipfrag

ovs ipFrag match

Author: si74

ovs/match.go

@@ -56,6 +56,7 @@ const (
 	ipv6DST   = "ipv6_dst"
 	ipv6SRC   = "ipv6_src"
 	ipv6Label = "ipv6_label"
+	ipFrag    = "ip_frag"
 	metadata  = "metadata"
 	ndSLL     = "nd_sll"
 	ndTLL     = "nd_tll"
@@ -1309,3 +1310,39 @@ func matchTransportPort(srcdst string, port uint16, mask uint16) ([]byte, error)
 
 	return bprintf("tp_%s=0x%04x/0x%04x", srcdst, port, mask), nil
 }
+
+// IPFragFlag is a string type which can be used with the IPFragMatch.
+type IPFragFlag string
+
+// OvS IP frag flags.
+// Source: http://www.openvswitch.org/support/dist-docs-2.5/ovs-ofctl.8.txt
+const (
+	IPFragFlagYes      IPFragFlag = "yes"
+	IPFragFlagNo       IPFragFlag = "no"
+	IPFragFlagFirst    IPFragFlag = "first"
+	IPFragFlagLater    IPFragFlag = "later"
+	IPFragFlagNotLater IPFragFlag = "not_later"
+)
+
+// IPFrag returns an ipFragMatch.
+func IPFrag(flag IPFragFlag) Match {
+	return &ipFragMatch{flag: flag}
+}
+
+// ipFragMatch implements the Match interface and is a match against
+// a packet fragmentation value.
+type ipFragMatch struct {
+	flag IPFragFlag
+}
+
+var _ Match = &ipFragMatch{}
+
+// GoString implements Match.
+func (m *ipFragMatch) GoString() string {
+	return fmt.Sprintf("ovs.IpFrag(%v)", m.flag)
+}
+
+// MarshalText implements Match.
+func (m *ipFragMatch) MarshalText() ([]byte, error) {
+	return bprintf("%s=%s", ipFrag, m.flag), nil
+}

ovs/match_test.go

@@ -1415,6 +1415,54 @@ func TestMatchARPOperation(t *testing.T) {
 	}
 }
 
+func TestMatchIPFrag(t *testing.T) {
+	var tests = []struct {
+		desc string
+		m    Match
+		out  string
+	}{
+		{
+			desc: "ip frag is no",
+			m:    IPFrag(IPFragFlagNo),
+			out:  "ip_frag=no",
+		},
+		{
+			desc: "ip frag is yes",
+			m:    IPFrag(IPFragFlagYes),
+			out:  "ip_frag=yes",
+		},
+		{
+			desc: "ip frag is first",
+			m:    IPFrag(IPFragFlagFirst),
+			out:  "ip_frag=first",
+		},
+		{
+			desc: "ip frag is later",
+			m:    IPFrag(IPFragFlagLater),
+			out:  "ip_frag=later",
+		},
+		{
+			desc: "ip frag is not later",
+			m:    IPFrag(IPFragFlagNotLater),
+			out:  "ip_frag=not_later",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.desc, func(t *testing.T) {
+			out, err := tt.m.MarshalText()
+			if err != nil {
+				t.Fatalf("unexpected error: %v", err)
+			}
+
+			if want, got := tt.out, string(out); want != got {
+				t.Fatalf("unexpected Match output:\n- want: %q\n-  got: %q",
+					want, got)
+			}
+		})
+	}
+}
+
 // mustParseMAC is a helper to parse a hardware address from a string using
 // net.ParseMAC, that panic on failure.
 func mustParseMAC(addr string) net.HardwareAddr {