REDROOM

Navigation

Breadcrumbs

/ / proc / thread-self / root / opt / golang / 1.22.0 / src / os

Create

Create File | Create Folder | Upload Files | Run Command

Edit File Size: 4.21 KB

/proc/thread-self/root/opt/golang/1.22.0/src/os/writeto_linux_test.go

Text Base64

// Copyright 2023 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package os_test

import (
	"bytes"
	"internal/poll"
	"io"
	"math/rand"
	"net"
	. "os"
	"strconv"
	"syscall"
	"testing"
	"time"
)

func TestSendFile(t *testing.T) {
	sizes := []int{
		1,
		42,
		1025,
		syscall.Getpagesize() + 1,
		32769,
	}
	t.Run("sendfile-to-unix", func(t *testing.T) {
		for _, size := range sizes {
			t.Run(strconv.Itoa(size), func(t *testing.T) {
				testSendFile(t, "unix", int64(size))
			})
		}
	})
	t.Run("sendfile-to-tcp", func(t *testing.T) {
		for _, size := range sizes {
			t.Run(strconv.Itoa(size), func(t *testing.T) {
				testSendFile(t, "tcp", int64(size))
			})
		}
	})
}

func testSendFile(t *testing.T, proto string, size int64) {
	dst, src, recv, data, hook := newSendFileTest(t, proto, size)

// Now call WriteTo (through io.Copy), which will hopefully call poll.SendFile
	n, err := io.Copy(dst, src)
	if err != nil {
		t.Fatalf("io.Copy error: %v", err)
	}

// We should have called poll.Splice with the right file descriptor arguments.
	if n > 0 && !hook.called {
		t.Fatal("expected to called poll.SendFile")
	}
	if hook.called && hook.srcfd != int(src.Fd()) {
		t.Fatalf("wrong source file descriptor: got %d, want %d", hook.srcfd, src.Fd())
	}
	sc, ok := dst.(syscall.Conn)
	if !ok {
		t.Fatalf("destination is not a syscall.Conn")
	}
	rc, err := sc.SyscallConn()
	if err != nil {
		t.Fatalf("destination SyscallConn error: %v", err)
	}
	if err = rc.Control(func(fd uintptr) {
		if hook.called && hook.dstfd != int(fd) {
			t.Fatalf("wrong destination file descriptor: got %d, want %d", hook.dstfd, int(fd))
		}
	}); err != nil {
		t.Fatalf("destination Conn Control error: %v", err)
	}

// Verify the data size and content.
	dataSize := len(data)
	dstData := make([]byte, dataSize)
	m, err := io.ReadFull(recv, dstData)
	if err != nil {
		t.Fatalf("server Conn Read error: %v", err)
	}
	if n != int64(dataSize) {
		t.Fatalf("data length mismatch for io.Copy, got %d, want %d", n, dataSize)
	}
	if m != dataSize {
		t.Fatalf("data length mismatch for net.Conn.Read, got %d, want %d", m, dataSize)
	}
	if !bytes.Equal(dstData, data) {
		t.Errorf("data mismatch, got %s, want %s", dstData, data)
	}
}

// newSendFileTest initializes a new test for sendfile.
//
// It creates source file and destination sockets, and populates the source file
// with random data of the specified size. It also hooks package os' call
// to poll.Sendfile and returns the hook so it can be inspected.
func newSendFileTest(t *testing.T, proto string, size int64) (net.Conn, *File, net.Conn, []byte, *sendFileHook) {
	t.Helper()

hook := hookSendFile(t)

client, server := createSocketPair(t, proto)
	tempFile, data := createTempFile(t, size)

return client, tempFile, server, data, hook
}

func hookSendFile(t *testing.T) *sendFileHook {
	h := new(sendFileHook)
	h.install()
	t.Cleanup(h.uninstall)
	return h
}

type sendFileHook struct {
	called bool
	dstfd  int
	srcfd  int
	remain int64

written int64
	handled bool
	err     error

original func(dst *poll.FD, src int, remain int64) (int64, error, bool)
}

func (h *sendFileHook) install() {
	h.original = *PollSendFile
	*PollSendFile = func(dst *poll.FD, src int, remain int64) (int64, error, bool) {
		h.called = true
		h.dstfd = dst.Sysfd
		h.srcfd = src
		h.remain = remain
		h.written, h.err, h.handled = h.original(dst, src, remain)
		return h.written, h.err, h.handled
	}
}

func (h *sendFileHook) uninstall() {
	*PollSendFile = h.original
}

func createTempFile(t *testing.T, size int64) (*File, []byte) {
	f, err := CreateTemp(t.TempDir(), "writeto-sendfile-to-socket")
	if err != nil {
		t.Fatalf("failed to create temporary file: %v", err)
	}
	t.Cleanup(func() {
		f.Close()
	})

randSeed := time.Now().Unix()
	t.Logf("random data seed: %d\n", randSeed)
	prng := rand.New(rand.NewSource(randSeed))
	data := make([]byte, size)
	prng.Read(data)
	if _, err := f.Write(data); err != nil {
		t.Fatalf("failed to create and feed the file: %v", err)
	}
	if err := f.Sync(); err != nil {
		t.Fatalf("failed to save the file: %v", err)
	}
	if _, err := f.Seek(0, io.SeekStart); err != nil {
		t.Fatalf("failed to rewind the file: %v", err)
	}

return f, data
}

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Directory Contents

Dirs: 4 × Files: 127

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
exec DIR	-	drwxr-xr-x	2024-02-02 18:09:55	Edit Download recursive recursive
signal DIR	-	drwxr-xr-x	2024-02-02 18:09:55	Edit Download recursive recursive
testdata DIR	-	drwxr-xr-x	2024-02-02 18:09:55	Edit Download recursive recursive
user DIR	-	drwxr-xr-x	2024-02-02 18:09:55	Edit Download recursive recursive
dir.go	4.40 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_aix.go	759 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_dragonfly.go	1.28 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_freebsd.go	1.16 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_js.go	678 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_linux.go	1.18 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_netbsd.go	1.16 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_openbsd.go	1.16 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_solaris.go	759 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dirent_wasip1.go	1.36 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dir_darwin.go	3.34 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dir_plan9.go	2.10 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dir_unix.go	5.24 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
dir_windows.go	7.17 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
endian_big.go	244 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
endian_little.go	304 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
env.go	3.85 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
env_test.go	5.04 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
env_unix_test.go	1.22 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error.go	4.76 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_errno.go	247 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_plan9.go	234 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_posix.go	538 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_test.go	4.92 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_unix_test.go	1.49 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
error_windows_test.go	1.72 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
example_test.go	8.39 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec.go	6.01 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable.go	774 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_darwin.go	613 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_dragonfly.go	293 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_freebsd.go	292 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_path.go	2.31 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_plan9.go	427 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_procfs.go	904 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_solaris.go	695 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_sysctl.go	891 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_test.go	3.40 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_wasm.go	333 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
executable_windows.go	641 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_plan9.go	3.27 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_posix.go	3.44 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_unix.go	2.09 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_unix_test.go	999 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_windows.go	4.57 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
exec_windows_test.go	1.78 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
export_linux_test.go	337 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
export_test.go	433 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
export_unix_test.go	241 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
export_windows_test.go	395 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
fifo_test.go	4.62 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file.go	24.91 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_mutex_plan9.go	1.81 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_open_unix.go	397 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_open_wasip1.go	818 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_plan9.go	15.99 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_posix.go	7.10 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_unix.go	14.05 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_wasip1.go	633 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
file_windows.go	12.65 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
getwd.go	2.52 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
os_test.go	77.07 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
os_unix_test.go	11.47 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
os_windows_test.go	41.83 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path.go	2.27 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path_plan9.go	492 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path_test.go	2.96 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path_unix.go	1.56 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path_windows.go	5.66 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
path_windows_test.go	4.03 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
pipe2_unix.go	640 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
pipe_test.go	12.41 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
pipe_unix.go	760 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
pipe_wasm.go	488 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
proc.go	2.27 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
rawconn.go	993 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
rawconn_test.go	1.15 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
readfrom_linux_test.go	20.30 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
read_test.go	3.20 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
removeall_at.go	4.91 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
removeall_noat.go	3.13 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
removeall_test.go	11.96 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat.go	965 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_aix.go	1.18 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_darwin.go	1.09 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_dragonfly.go	1.06 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_freebsd.go	1.07 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_js.go	1.11 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_linux.go	1.06 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_netbsd.go	1.07 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_openbsd.go	1.06 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_plan9.go	2.37 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_solaris.go	1.30 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_test.go	6.46 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_unix.go	1.23 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_wasip1.go	956 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
stat_windows.go	4.93 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sticky_bsd.go	425 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sticky_notbsd.go	320 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys.go	294 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_aix.go	682 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_bsd.go	466 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_js.go	313 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_linux.go	1.04 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_plan9.go	453 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_solaris.go	265 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_unix.go	493 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_wasip1.go	309 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
sys_windows.go	874 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
tempfile.go	3.80 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
tempfile_test.go	5.49 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
timeout_test.go	17.09 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
types.go	2.79 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
types_plan9.go	797 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
types_unix.go	776 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
types_windows.go	10.01 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait6_dragonfly.go	496 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait6_freebsd64.go	547 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait6_freebsd_386.go	544 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait6_freebsd_arm.go	549 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait6_netbsd.go	534 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait_unimp.go	831 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait_wait6.go	781 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
wait_waitid.go	1.32 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
writeto_linux_test.go	4.21 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
zero_copy_linux.go	4.38 KB	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive
zero_copy_stub.go	406 B	lrw-r--r--	2024-02-02 18:09:55	Edit Download recursive recursive