Allow to program L4 fields in policy selector (#113)

Signed-off-by: Alessandro Boch <aboch@docker.com>

xfrm_policy.go

@@ -52,6 +52,9 @@ type XfrmPolicyTmpl struct {
 type XfrmPolicy struct {
 	Dst      *net.IPNet
 	Src      *net.IPNet
+	Proto    Proto
+	DstPort  int
+	SrcPort  int
 	Dir      Dir
 	Priority int
 	Index    int

xfrm_policy_linux.go

@@ -14,6 +14,11 @@ func selFromPolicy(sel *nl.XfrmSelector, policy *XfrmPolicy) {
 	sel.PrefixlenD = uint8(prefixlenD)
 	prefixlenS, _ := policy.Src.Mask.Size()
 	sel.PrefixlenS = uint8(prefixlenS)
+	sel.Proto = uint8(policy.Proto)
+	sel.Dport = nl.Swap16(uint16(policy.DstPort))
+	sel.Sport = nl.Swap16(uint16(policy.SrcPort))
+	sel.DportMask = ^uint16(0)
+	sel.SportMask = ^uint16(0)
 }
 
 // XfrmPolicyAdd will add an xfrm policy to the system.
@@ -160,6 +165,9 @@ func (h *Handle) XfrmPolicyList(family int) ([]XfrmPolicy, error) {
 
 		policy.Dst = msg.Sel.Daddr.ToIPNet(msg.Sel.PrefixlenD)
 		policy.Src = msg.Sel.Saddr.ToIPNet(msg.Sel.PrefixlenS)
+		policy.Proto = Proto(msg.Sel.Proto)
+		policy.DstPort = int(nl.Swap16(msg.Sel.Dport))
+		policy.SrcPort = int(nl.Swap16(msg.Sel.Sport))
 		policy.Priority = int(msg.Priority)
 		policy.Index = int(msg.Index)
 		policy.Dir = Dir(msg.Dir)

xfrm_policy_test.go

@@ -1,6 +1,7 @@
 package netlink
 
 import (
+	"bytes"
 	"net"
 	"testing"
 )
@@ -14,6 +15,9 @@ func TestXfrmPolicyAddUpdateDel(t *testing.T) {
 	policy := XfrmPolicy{
 		Src:     src,
 		Dst:     dst,
+		Proto:   17,
+		DstPort: 1234,
+		SrcPort: 5678,
 		Dir:     XFRM_DIR_OUT,
 		Mark: &XfrmMark{
 			Value: 0xabff22,
@@ -40,6 +44,16 @@ func TestXfrmPolicyAddUpdateDel(t *testing.T) {
 		t.Fatal("Policy not added properly")
 	}
 
+	// Verify Selector fields
+	if !compareIPNet(policies[0].Dst, policy.Dst) ||
+		!compareIPNet(policies[0].Src, policy.Src) ||
+		policies[0].Proto != policy.Proto ||
+		policies[0].DstPort != policy.DstPort ||
+		policies[0].SrcPort != policy.SrcPort {
+		t.Fatalf("Incorrect policy data retrieved. Expected %v. Got %v.",
+			policy, policies[0])
+	}
+
 	// Modify the policy
 	policy.Priority = 100
 	if err := XfrmPolicyUpdate(&policy); err != nil {
@@ -65,3 +79,13 @@ func TestXfrmPolicyAddUpdateDel(t *testing.T) {
 		t.Fatal("Policy not removed properly")
 	}
 }
+
+func compareIPNet(a, b *net.IPNet) bool {
+	if a == b {
+		return true
+	}
+	if a == nil || b == nil {
+		return false
+	}
+	return a.IP.Equal(b.IP) && bytes.Equal(a.Mask, b.Mask)
+}

xfrm_state_linux.go

@@ -40,7 +40,7 @@ func writeMark(m *XfrmMark) []byte {
 		Mask:  m.Mask,
 	}
 	if mark.Mask == 0 {
-		mark.Mask = 0xfffffff
+		mark.Mask = ^uint32(0)
 	}
 	return mark.Serialize()
 }