summaryrefslogtreecommitdiff
path: root/cmd
diff options
context:
space:
mode:
Diffstat (limited to 'cmd')
-rw-r--r--cmd/ioworkload/scenario_socket.go54
-rw-r--r--cmd/ioworkload/scenarios.go1
2 files changed, 55 insertions, 0 deletions
diff --git a/cmd/ioworkload/scenario_socket.go b/cmd/ioworkload/scenario_socket.go
index db95ad4..420cdfb 100644
--- a/cmd/ioworkload/scenario_socket.go
+++ b/cmd/ioworkload/scenario_socket.go
@@ -120,3 +120,57 @@ func socketAcceptLifecyclePlain() error {
}
return nil
}
+
+func socketIntrospection() error {
+ dir, cleanup, err := makeTempDir("socket-introspection")
+ if err != nil {
+ return err
+ }
+ defer cleanup()
+
+ socketPath := filepath.Join(dir, "introspection.sock")
+
+ listenerFD, err := syscall.Socket(syscall.AF_UNIX, syscall.SOCK_STREAM, 0)
+ if err != nil {
+ return fmt.Errorf("listener socket: %w", err)
+ }
+ defer syscall.Close(listenerFD) //nolint:errcheck
+
+ if err := syscall.Bind(listenerFD, &syscall.SockaddrUnix{Name: socketPath}); err != nil {
+ return fmt.Errorf("bind: %w", err)
+ }
+ if err := syscall.Listen(listenerFD, 1); err != nil {
+ return fmt.Errorf("listen: %w", err)
+ }
+
+ clientFD, err := syscall.Socket(syscall.AF_UNIX, syscall.SOCK_STREAM, 0)
+ if err != nil {
+ return fmt.Errorf("client socket: %w", err)
+ }
+ defer syscall.Close(clientFD) //nolint:errcheck
+
+ if err := syscall.Connect(clientFD, &syscall.SockaddrUnix{Name: socketPath}); err != nil {
+ return fmt.Errorf("connect: %w", err)
+ }
+
+ acceptedFD, _, err := syscall.Accept4(listenerFD, 0)
+ if err != nil {
+ return fmt.Errorf("accept4: %w", err)
+ }
+ defer syscall.Close(acceptedFD) //nolint:errcheck
+
+ if _, err := syscall.Getsockname(acceptedFD); err != nil {
+ return fmt.Errorf("getsockname: %w", err)
+ }
+ if _, err := syscall.Getpeername(acceptedFD); err != nil {
+ return fmt.Errorf("getpeername: %w", err)
+ }
+ if err := syscall.SetsockoptInt(acceptedFD, syscall.SOL_SOCKET, syscall.SO_RCVBUF, 32768); err != nil {
+ return fmt.Errorf("setsockopt: %w", err)
+ }
+ if _, err := syscall.GetsockoptInt(acceptedFD, syscall.SOL_SOCKET, syscall.SO_RCVBUF); err != nil {
+ return fmt.Errorf("getsockopt: %w", err)
+ }
+
+ return nil
+}
diff --git a/cmd/ioworkload/scenarios.go b/cmd/ioworkload/scenarios.go
index 5648801..9700d85 100644
--- a/cmd/ioworkload/scenarios.go
+++ b/cmd/ioworkload/scenarios.go
@@ -29,6 +29,7 @@ var scenarios = map[string]func() error{
"socketpair-basic": socketpairBasic,
"socket-accept-lifecycle": socketAcceptLifecycle,
"socket-accept-lifecycle-plain": socketAcceptLifecyclePlain,
+ "socket-introspection": socketIntrospection,
"family-mixed": familyMixed,
"close-basic": closeBasic,
"close-range": closeRange,