4 files changed, 45 insertions, 3 deletions
diff --git a/pkg/unet/BUILD b/pkg/unet/BUILD
index 5e177e78e..769509e80 100644
--- a/pkg/unet/BUILD
+++ b/pkg/unet/BUILD
@@ -8,7 +8,7 @@ go_library(
         "unet.go",
         "unet_unsafe.go",
     ],
-    importpath = "gvisor.googlesource.com/gvisor/pkg/unet",
+    importpath = "gvisor.dev/gvisor/pkg/unet",
     visibility = ["//visibility:public"],
     deps = [
         "//pkg/abi/linux",
diff --git a/pkg/unet/unet.go b/pkg/unet/unet.go
index 2aa1af4ff..d843f19cf 100644
--- a/pkg/unet/unet.go
+++ b/pkg/unet/unet.go
@@ -23,7 +23,7 @@ import (
 	"sync/atomic"
 	"syscall"
 
-	"gvisor.googlesource.com/gvisor/pkg/gate"
+	"gvisor.dev/gvisor/pkg/gate"
 )
 
 // backlog is used for the listen request.
diff --git a/pkg/unet/unet_test.go b/pkg/unet/unet_test.go
index 763b23c7c..a3cc6f5d3 100644
--- a/pkg/unet/unet_test.go
+++ b/pkg/unet/unet_test.go
@@ -691,3 +691,45 @@ func TestControlMessage(t *testing.T) {
 		}
 	}
 }
+
+func benchmarkSendRecv(b *testing.B, packet bool) {
+	server, client, err := SocketPair(packet)
+	if err != nil {
+		b.Fatalf("SocketPair: got %v, wanted nil", err)
+	}
+	defer server.Close()
+	defer client.Close()
+	go func() {
+		buf := make([]byte, 1)
+		for i := 0; i < b.N; i++ {
+			n, err := server.Read(buf)
+			if n != 1 || err != nil {
+				b.Fatalf("server.Read: got (%d, %v), wanted (1, nil)", n, err)
+			}
+			n, err = server.Write(buf)
+			if n != 1 || err != nil {
+				b.Fatalf("server.Write: got (%d, %v), wanted (1, nil)", n, err)
+			}
+		}
+	}()
+	buf := make([]byte, 1)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		n, err := client.Write(buf)
+		if n != 1 || err != nil {
+			b.Fatalf("client.Write: got (%d, %v), wanted (1, nil)", n, err)
+		}
+		n, err = client.Read(buf)
+		if n != 1 || err != nil {
+			b.Fatalf("client.Read: got (%d, %v), wanted (1, nil)", n, err)
+		}
+	}
+}
+
+func BenchmarkSendRecvStream(b *testing.B) {
+	benchmarkSendRecv(b, false)
+}
+
+func BenchmarkSendRecvPacket(b *testing.B) {
+	benchmarkSendRecv(b, true)
+}
diff --git a/pkg/unet/unet_unsafe.go b/pkg/unet/unet_unsafe.go
index fa0916439..f8a42c914 100644
--- a/pkg/unet/unet_unsafe.go
+++ b/pkg/unet/unet_unsafe.go
@@ -21,7 +21,7 @@ import (
 	"syscall"
 	"unsafe"
 
-	"gvisor.googlesource.com/gvisor/pkg/abi/linux"
+	"gvisor.dev/gvisor/pkg/abi/linux"
 )
 
 // wait blocks until the socket FD is ready for reading or writing, depending