From d67ff8ac9b5156f81f9efa7f6bd8d16c163b1661 Mon Sep 17 00:00:00 2001
From: Kevin Krakauer <krakauer@google.com>
Date: Wed, 13 Jul 2022 10:17:57 -0700
Subject: [PATCH] use PACKET_MMAP for rx when supported

This was fixed in 5.6-rc7, so we enable PACKET_MMAP for all kernels >= 5.6.

Downloads in iperf go from ~90MB/s to ~102MB/s.

PiperOrigin-RevId: 460746326
---
 runsc/boot/BUILD      |  1 +
 runsc/boot/network.go | 20 +++++++++++++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/runsc/boot/BUILD b/runsc/boot/BUILD
index 1b6331022..53030ad61 100644
--- a/runsc/boot/BUILD
+++ b/runsc/boot/BUILD
@@ -119,6 +119,7 @@ go_library(
         "//runsc/specutils/seccomp",
         "@com_github_opencontainers_runtime_spec//specs-go:go_default_library",
         "@org_golang_google_protobuf//proto:go_default_library",
+        "@org_golang_x_mod//semver:go_default_library",
         "@org_golang_x_sys//unix:go_default_library",
     ],
 )
diff --git a/runsc/boot/network.go b/runsc/boot/network.go
index 7094632c9..d524a9765 100644
--- a/runsc/boot/network.go
+++ b/runsc/boot/network.go
@@ -17,9 +17,11 @@ package boot
 import (
 	"fmt"
 	"net"
+	"regexp"
 	"runtime"
 	"strings"
 
+	"golang.org/x/mod/semver"
 	"golang.org/x/sys/unix"
 	"gvisor.dev/gvisor/pkg/log"
 	"gvisor.dev/gvisor/pkg/tcpip"
@@ -194,6 +196,22 @@ func (n *Network) CreateLinksAndRoutes(args *CreateLinksAndRoutesArgs, _ *struct
 		}
 	}
 
+	// Choose a dispatch mode.
+	// Get the Linux kernel version. Uname will return something like
+	// "5.17.7-1distro-amd64", from which we'd want "5.17.7".
+	var uname unix.Utsname
+	if err := unix.Uname(&uname); err != nil {
+		return err
+	}
+	re := regexp.MustCompile(`[0-9]+\.[0-9]+(\.[0-9]+)?`)
+	version := "v" + string(re.Find(uname.Release[:]))
+	dispatchMode := fdbased.RecvMMsg
+	if semver.IsValid(version) && semver.Compare(version, "v5.6") >= 0 {
+		dispatchMode = fdbased.PacketMMap
+	} else {
+		log.Infof("Host kernel version < 5.6, falling back to RecvMMsg dispatch")
+	}
+
 	fdOffset := 0
 	for _, link := range args.FDBasedLinks {
 		nicID++
@@ -219,7 +237,7 @@ func (n *Network) CreateLinksAndRoutes(args *CreateLinksAndRoutesArgs, _ *struct
 			MTU:                uint32(link.MTU),
 			EthernetHeader:     mac != "",
 			Address:            mac,
-			PacketDispatchMode: fdbased.RecvMMsg,
+			PacketDispatchMode: dispatchMode,
 			GSOMaxSize:         link.GSOMaxSize,
 			GvisorGSOEnabled:   link.GvisorGSOEnabled,
 			TXChecksumOffload:  link.TXChecksumOffload,