Root Explorer PHP

[ Root System Explorer ]

Editing: writeto_linux_test.go

// 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
}

[ CANCEL ]

Name	Type	Actions
.. (Parent Directory)
📄 dir.go	FILE	[EDIT]
📄 dir_darwin.go	FILE	[EDIT]
📄 dir_plan9.go	FILE	[EDIT]
📄 dir_unix.go	FILE	[EDIT]
📄 dir_windows.go	FILE	[EDIT]
📄 dirent_aix.go	FILE	[EDIT]
📄 dirent_dragonfly.go	FILE	[EDIT]
📄 dirent_freebsd.go	FILE	[EDIT]
📄 dirent_js.go	FILE	[EDIT]
📄 dirent_linux.go	FILE	[EDIT]
📄 dirent_netbsd.go	FILE	[EDIT]
📄 dirent_openbsd.go	FILE	[EDIT]
📄 dirent_solaris.go	FILE	[EDIT]
📄 dirent_wasip1.go	FILE	[EDIT]
📄 endian_big.go	FILE	[EDIT]
📄 endian_little.go	FILE	[EDIT]
📄 env.go	FILE	[EDIT]
📄 env_test.go	FILE	[EDIT]
📄 env_unix_test.go	FILE	[EDIT]
📄 error.go	FILE	[EDIT]
📄 error_errno.go	FILE	[EDIT]
📄 error_plan9.go	FILE	[EDIT]
📄 error_posix.go	FILE	[EDIT]
📄 error_test.go	FILE	[EDIT]
📄 error_unix_test.go	FILE	[EDIT]
📄 error_windows_test.go	FILE	[EDIT]
📄 example_test.go	FILE	[EDIT]
📁 exec/	DIR
📄 exec.go	FILE	[EDIT]
📄 exec_plan9.go	FILE	[EDIT]
📄 exec_posix.go	FILE	[EDIT]
📄 exec_unix.go	FILE	[EDIT]
📄 exec_unix_test.go	FILE	[EDIT]
📄 exec_windows.go	FILE	[EDIT]
📄 exec_windows_test.go	FILE	[EDIT]
📄 executable.go	FILE	[EDIT]
📄 executable_darwin.go	FILE	[EDIT]
📄 executable_dragonfly.go	FILE	[EDIT]
📄 executable_freebsd.go	FILE	[EDIT]
📄 executable_path.go	FILE	[EDIT]
📄 executable_plan9.go	FILE	[EDIT]
📄 executable_procfs.go	FILE	[EDIT]
📄 executable_solaris.go	FILE	[EDIT]
📄 executable_sysctl.go	FILE	[EDIT]
📄 executable_test.go	FILE	[EDIT]
📄 executable_wasm.go	FILE	[EDIT]
📄 executable_windows.go	FILE	[EDIT]
📄 export_linux_test.go	FILE	[EDIT]
📄 export_test.go	FILE	[EDIT]
📄 export_unix_test.go	FILE	[EDIT]
📄 export_windows_test.go	FILE	[EDIT]
📄 fifo_test.go	FILE	[EDIT]
📄 file.go	FILE	[EDIT]
📄 file_mutex_plan9.go	FILE	[EDIT]
📄 file_open_unix.go	FILE	[EDIT]
📄 file_open_wasip1.go	FILE	[EDIT]
📄 file_plan9.go	FILE	[EDIT]
📄 file_posix.go	FILE	[EDIT]
📄 file_unix.go	FILE	[EDIT]
📄 file_wasip1.go	FILE	[EDIT]
📄 file_windows.go	FILE	[EDIT]
📄 getwd.go	FILE	[EDIT]
📄 os_test.go	FILE	[EDIT]
📄 os_unix_test.go	FILE	[EDIT]
📄 os_windows_test.go	FILE	[EDIT]
📄 path.go	FILE	[EDIT]
📄 path_plan9.go	FILE	[EDIT]
📄 path_test.go	FILE	[EDIT]
📄 path_unix.go	FILE	[EDIT]
📄 path_windows.go	FILE	[EDIT]
📄 path_windows_test.go	FILE	[EDIT]
📄 pipe2_unix.go	FILE	[EDIT]
📄 pipe_test.go	FILE	[EDIT]
📄 pipe_unix.go	FILE	[EDIT]
📄 pipe_wasm.go	FILE	[EDIT]
📄 proc.go	FILE	[EDIT]
📄 rawconn.go	FILE	[EDIT]
📄 rawconn_test.go	FILE	[EDIT]
📄 read_test.go	FILE	[EDIT]
📄 readfrom_linux_test.go	FILE	[EDIT]
📄 removeall_at.go	FILE	[EDIT]
📄 removeall_noat.go	FILE	[EDIT]
📄 removeall_test.go	FILE	[EDIT]
📁 signal/	DIR
📄 stat.go	FILE	[EDIT]
📄 stat_aix.go	FILE	[EDIT]
📄 stat_darwin.go	FILE	[EDIT]
📄 stat_dragonfly.go	FILE	[EDIT]
📄 stat_freebsd.go	FILE	[EDIT]
📄 stat_js.go	FILE	[EDIT]
📄 stat_linux.go	FILE	[EDIT]
📄 stat_netbsd.go	FILE	[EDIT]
📄 stat_openbsd.go	FILE	[EDIT]
📄 stat_plan9.go	FILE	[EDIT]
📄 stat_solaris.go	FILE	[EDIT]
📄 stat_test.go	FILE	[EDIT]
📄 stat_unix.go	FILE	[EDIT]
📄 stat_wasip1.go	FILE	[EDIT]
📄 stat_windows.go	FILE	[EDIT]
📄 sticky_bsd.go	FILE	[EDIT]
📄 sticky_notbsd.go	FILE	[EDIT]
📄 sys.go	FILE	[EDIT]
📄 sys_aix.go	FILE	[EDIT]
📄 sys_bsd.go	FILE	[EDIT]
📄 sys_js.go	FILE	[EDIT]
📄 sys_linux.go	FILE	[EDIT]
📄 sys_plan9.go	FILE	[EDIT]
📄 sys_solaris.go	FILE	[EDIT]
📄 sys_unix.go	FILE	[EDIT]
📄 sys_wasip1.go	FILE	[EDIT]
📄 sys_windows.go	FILE	[EDIT]
📄 tempfile.go	FILE	[EDIT]
📄 tempfile_test.go	FILE	[EDIT]
📁 testdata/	DIR
📄 timeout_test.go	FILE	[EDIT]
📄 types.go	FILE	[EDIT]
📄 types_plan9.go	FILE	[EDIT]
📄 types_unix.go	FILE	[EDIT]
📄 types_windows.go	FILE	[EDIT]
📁 user/	DIR
📄 wait6_dragonfly.go	FILE	[EDIT]
📄 wait6_freebsd64.go	FILE	[EDIT]
📄 wait6_freebsd_386.go	FILE	[EDIT]
📄 wait6_freebsd_arm.go	FILE	[EDIT]
📄 wait6_netbsd.go	FILE	[EDIT]
📄 wait_unimp.go	FILE	[EDIT]
📄 wait_wait6.go	FILE	[EDIT]
📄 wait_waitid.go	FILE	[EDIT]
📄 writeto_linux_test.go	FILE	[EDIT]
📄 zero_copy_linux.go	FILE	[EDIT]
📄 zero_copy_stub.go	FILE	[EDIT]