Gecko [ 167.86.127.34 ]

Name	Size	Permission
http	[ DIR ]	drwxr-xr-x
internal	[ DIR ]	drwxr-xr-x
mail	[ DIR ]	drwxr-xr-x
rpc	[ DIR ]	drwxr-xr-x
smtp	[ DIR ]	drwxr-xr-x
testdata	[ DIR ]	drwxr-xr-x
textproto	[ DIR ]	drwxr-xr-x
url	[ DIR ]	drwxr-xr-x
addrselect.go	9.51 KB	-rw-r--r--
addrselect_test.go	6.2 KB	-rw-r--r--
cgo_aix.go	578 B	-rw-r--r--
cgo_android.go	268 B	-rw-r--r--
cgo_bsd.go	348 B	-rw-r--r--
cgo_linux.go	635 B	-rw-r--r--
cgo_netbsd.go	272 B	-rw-r--r--
cgo_openbsd.go	272 B	-rw-r--r--
cgo_resnew.go	577 B	-rw-r--r--
cgo_resold.go	571 B	-rw-r--r--
cgo_socknew.go	748 B	-rw-r--r--
cgo_sockold.go	838 B	-rw-r--r--
cgo_solaris.go	339 B	-rw-r--r--
cgo_stub.go	1.06 KB	-rw-r--r--
cgo_unix.go	9.53 KB	-rw-r--r--
cgo_unix_test.go	1.89 KB	-rw-r--r--
cgo_windows.go	398 B	-rw-r--r--
conf.go	8.88 KB	-rw-r--r--
conf_netcgo.go	369 B	-rw-r--r--
conf_test.go	9.59 KB	-rw-r--r--
conn_test.go	1.77 KB	-rw-r--r--
dial.go	22.44 KB	-rw-r--r--
dial_test.go	27 KB	-rw-r--r--
dial_unix_test.go	2.74 KB	-rw-r--r--
dnsclient.go	5.34 KB	-rw-r--r--
dnsclient_test.go	1.54 KB	-rw-r--r--
dnsclient_unix.go	21.86 KB	-rw-r--r--
dnsclient_unix_test.go	45.17 KB	-rw-r--r--
dnsconfig_unix.go	5.47 KB	-rw-r--r--
dnsconfig_unix_test.go	6.73 KB	-rw-r--r--
dnsname_test.go	1.98 KB	-rw-r--r--
error_plan9.go	224 B	-rw-r--r--
error_plan9_test.go	398 B	-rw-r--r--
error_posix.go	583 B	-rw-r--r--
error_posix_test.go	980 B	-rw-r--r--
error_test.go	19.39 KB	-rw-r--r--
error_unix.go	421 B	-rw-r--r--
error_unix_test.go	672 B	-rw-r--r--
error_windows.go	355 B	-rw-r--r--
error_windows_test.go	483 B	-rw-r--r--
example_test.go	3.16 KB	-rw-r--r--
external_test.go	4.07 KB	-rw-r--r--
fd_plan9.go	3.56 KB	-rw-r--r--
fd_unix.go	7.36 KB	-rw-r--r--
fd_windows.go	6.49 KB	-rw-r--r--
file.go	1.69 KB	-rw-r--r--
file_plan9.go	2.68 KB	-rw-r--r--
file_stub.go	485 B	-rw-r--r--
file_test.go	6.53 KB	-rw-r--r--
file_unix.go	2.51 KB	-rw-r--r--
file_windows.go	521 B	-rw-r--r--
hook.go	628 B	-rw-r--r--
hook_plan9.go	283 B	-rw-r--r--
hook_unix.go	734 B	-rw-r--r--
hook_windows.go	875 B	-rw-r--r--
hosts.go	3.1 KB	-rw-r--r--
hosts_test.go	4.65 KB	-rw-r--r--
interface.go	7.14 KB	-rw-r--r--
interface_aix.go	4.41 KB	-rw-r--r--
interface_bsd.go	2.74 KB	-rw-r--r--
interface_bsd_test.go	1.42 KB	-rw-r--r--
interface_bsdvar.go	711 B	-rw-r--r--
interface_darwin.go	1.29 KB	-rw-r--r--
interface_freebsd.go	1.42 KB	-rw-r--r--
interface_linux.go	6.9 KB	-rw-r--r--
interface_linux_test.go	3.65 KB	-rw-r--r--
interface_plan9.go	4.59 KB	-rw-r--r--
interface_solaris.go	2.46 KB	-rw-r--r--
interface_stub.go	808 B	-rw-r--r--
interface_test.go	9.67 KB	-rw-r--r--
interface_unix_test.go	4.69 KB	-rw-r--r--
interface_windows.go	5.41 KB	-rw-r--r--
ip.go	16.61 KB	-rw-r--r--
ip_test.go	23.96 KB	-rw-r--r--
iprawsock.go	7.08 KB	-rw-r--r--
iprawsock_plan9.go	874 B	-rw-r--r--
iprawsock_posix.go	3.45 KB	-rw-r--r--
iprawsock_test.go	5.62 KB	-rw-r--r--
ipsock.go	8.77 KB	-rw-r--r--
ipsock_plan9.go	6.25 KB	-rw-r--r--
ipsock_posix.go	6.16 KB	-rw-r--r--
ipsock_test.go	6.81 KB	-rw-r--r--
listen_test.go	21.32 KB	-rw-r--r--
lookup.go	15.45 KB	-rw-r--r--
lookup_fake.go	1.64 KB	-rw-r--r--
lookup_plan9.go	7.68 KB	-rw-r--r--
lookup_test.go	29.14 KB	-rw-r--r--
lookup_unix.go	8.78 KB	-rw-r--r--
lookup_windows.go	11.74 KB	-rw-r--r--
lookup_windows_test.go	8 KB	-rw-r--r--
mac.go	1.88 KB	-rw-r--r--
mac_test.go	3.26 KB	-rw-r--r--
main_cloexec_test.go	669 B	-rw-r--r--
main_conf_test.go	915 B	-rw-r--r--
main_noconf_test.go	493 B	-rw-r--r--
main_plan9_test.go	392 B	-rw-r--r--
main_posix_test.go	1.37 KB	-rw-r--r--
main_test.go	7.11 KB	-rw-r--r--
main_unix_test.go	1.24 KB	-rw-r--r--
main_windows_test.go	1.08 KB	-rw-r--r--
mockserver_test.go	11.23 KB	-rw-r--r--
net.go	20.9 KB	-rw-r--r--
net_fake.go	5.34 KB	-rw-r--r--
net_test.go	12.1 KB	-rw-r--r--
net_windows_test.go	15.94 KB	-rw-r--r--
netgo_unix_test.go	631 B	-rw-r--r--
nss.go	3.71 KB	-rw-r--r--
nss_test.go	3.33 KB	-rw-r--r--
packetconn_test.go	3.56 KB	-rw-r--r--
parse.go	7.82 KB	-rw-r--r--
parse_test.go	2.14 KB	-rw-r--r--
pipe.go	5.59 KB	-rw-r--r--
pipe_test.go	1.2 KB	-rw-r--r--
platform_test.go	4.59 KB	-rw-r--r--
port.go	1.46 KB	-rw-r--r--
port_test.go	1.34 KB	-rw-r--r--
port_unix.go	1.28 KB	-rw-r--r--
protoconn_test.go	7.22 KB	-rw-r--r--
rawconn.go	1.89 KB	-rw-r--r--
rawconn_stub_test.go	622 B	-rw-r--r--
rawconn_test.go	4.52 KB	-rw-r--r--
rawconn_unix_test.go	3.15 KB	-rw-r--r--
rawconn_windows_test.go	3.26 KB	-rw-r--r--
sendfile_linux.go	1.1 KB	-rw-r--r--
sendfile_stub.go	329 B	-rw-r--r--
sendfile_test.go	5.69 KB	-rw-r--r--
sendfile_unix_alt.go	1.98 KB	-rw-r--r--
sendfile_windows.go	1.02 KB	-rw-r--r--
server_test.go	11.55 KB	-rw-r--r--
sock_bsd.go	898 B	-rw-r--r--
sock_cloexec.go	1.46 KB	-rw-r--r--
sock_linux.go	644 B	-rw-r--r--
sock_plan9.go	262 B	-rw-r--r--
sock_posix.go	6.83 KB	-rw-r--r--
sock_stub.go	378 B	-rw-r--r--
sock_windows.go	1.16 KB	-rw-r--r--
sockaddr_posix.go	955 B	-rw-r--r--
sockopt_aix.go	1.32 KB	-rw-r--r--
sockopt_bsd.go	2.09 KB	-rw-r--r--
sockopt_linux.go	1.14 KB	-rw-r--r--
sockopt_plan9.go	406 B	-rw-r--r--
sockopt_posix.go	2.67 KB	-rw-r--r--
sockopt_solaris.go	1.14 KB	-rw-r--r--
sockopt_stub.go	738 B	-rw-r--r--
sockopt_windows.go	1.34 KB	-rw-r--r--
sockoptip_bsdvar.go	848 B	-rw-r--r--
sockoptip_linux.go	735 B	-rw-r--r--
sockoptip_posix.go	1.43 KB	-rw-r--r--
sockoptip_stub.go	949 B	-rw-r--r--
sockoptip_windows.go	819 B	-rw-r--r--
splice_linux.go	1.09 KB	-rw-r--r--
splice_stub.go	286 B	-rw-r--r--
splice_test.go	10.51 KB	-rw-r--r--
sys_cloexec.go	977 B	-rw-r--r--
tcpsock.go	9.85 KB	-rw-r--r--
tcpsock_plan9.go	2.13 KB	-rw-r--r--
tcpsock_posix.go	5.27 KB	-rw-r--r--
tcpsock_test.go	17.51 KB	-rw-r--r--
tcpsock_unix_test.go	2.38 KB	-rw-r--r--
tcpsockopt_darwin.go	770 B	-rw-r--r--
tcpsockopt_dragonfly.go	698 B	-rw-r--r--
tcpsockopt_openbsd.go	365 B	-rw-r--r--
tcpsockopt_plan9.go	503 B	-rw-r--r--
tcpsockopt_posix.go	491 B	-rw-r--r--
tcpsockopt_solaris.go	1.15 KB	-rw-r--r--
tcpsockopt_stub.go	393 B	-rw-r--r--
tcpsockopt_unix.go	712 B	-rw-r--r--
tcpsockopt_windows.go	741 B	-rw-r--r--
timeout_test.go	23.18 KB	-rw-r--r--
udpsock.go	8.7 KB	-rw-r--r--
udpsock_plan9.go	3.77 KB	-rw-r--r--
udpsock_plan9_test.go	1.31 KB	-rw-r--r--
udpsock_posix.go	4.09 KB	-rw-r--r--
udpsock_test.go	11.18 KB	-rw-r--r--
unixsock.go	10.17 KB	-rw-r--r--
unixsock_linux_test.go	2.29 KB	-rw-r--r--
unixsock_plan9.go	1.24 KB	-rw-r--r--
unixsock_posix.go	5.81 KB	-rw-r--r--
unixsock_test.go	10.58 KB	-rw-r--r--
unixsock_windows_test.go	2.07 KB	-rw-r--r--
write_unix_test.go	1.59 KB	-rw-r--r--
writev_test.go	5.01 KB	-rw-r--r--
writev_unix.go	706 B	-rw-r--r--

Code Editor : splice_test.go

// Copyright 2018 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.

// +build linux

package net

import (
	"io"
	"io/ioutil"
	"log"
	"os"
	"os/exec"
	"strconv"
	"sync"
	"testing"
	"time"
)

func TestSplice(t *testing.T) {
	t.Run("tcp-to-tcp", func(t *testing.T) { testSplice(t, "tcp", "tcp") })
	if !testableNetwork("unixgram") {
		t.Skip("skipping unix-to-tcp tests")
	}
	t.Run("unix-to-tcp", func(t *testing.T) { testSplice(t, "unix", "tcp") })
	t.Run("no-unixpacket", testSpliceNoUnixpacket)
	t.Run("no-unixgram", testSpliceNoUnixgram)
}

func testSplice(t *testing.T, upNet, downNet string) {
	t.Run("simple", spliceTestCase{upNet, downNet, 128, 128, 0}.test)
	t.Run("multipleWrite", spliceTestCase{upNet, downNet, 4096, 1 << 20, 0}.test)
	t.Run("big", spliceTestCase{upNet, downNet, 5 << 20, 1 << 30, 0}.test)
	t.Run("honorsLimitedReader", spliceTestCase{upNet, downNet, 4096, 1 << 20, 1 << 10}.test)
	t.Run("updatesLimitedReaderN", spliceTestCase{upNet, downNet, 1024, 4096, 4096 + 100}.test)
	t.Run("limitedReaderAtLimit", spliceTestCase{upNet, downNet, 32, 128, 128}.test)
	t.Run("readerAtEOF", func(t *testing.T) { testSpliceReaderAtEOF(t, upNet, downNet) })
	t.Run("issue25985", func(t *testing.T) { testSpliceIssue25985(t, upNet, downNet) })
}

type spliceTestCase struct {
	upNet, downNet string

chunkSize, totalSize int
	limitReadSize        int
}

func (tc spliceTestCase) test(t *testing.T) {
	clientUp, serverUp, err := spliceTestSocketPair(tc.upNet)
	if err != nil {
		t.Fatal(err)
	}
	defer serverUp.Close()
	cleanup, err := startSpliceClient(clientUp, "w", tc.chunkSize, tc.totalSize)
	if err != nil {
		t.Fatal(err)
	}
	defer cleanup()
	clientDown, serverDown, err := spliceTestSocketPair(tc.downNet)
	if err != nil {
		t.Fatal(err)
	}
	defer serverDown.Close()
	cleanup, err = startSpliceClient(clientDown, "r", tc.chunkSize, tc.totalSize)
	if err != nil {
		t.Fatal(err)
	}
	defer cleanup()
	var (
		r    io.Reader = serverUp
		size           = tc.totalSize
	)
	if tc.limitReadSize > 0 {
		if tc.limitReadSize < size {
			size = tc.limitReadSize
		}

r = &io.LimitedReader{
			N: int64(tc.limitReadSize),
			R: serverUp,
		}
		defer serverUp.Close()
	}
	n, err := io.Copy(serverDown, r)
	serverDown.Close()
	if err != nil {
		t.Fatal(err)
	}
	if want := int64(size); want != n {
		t.Errorf("want %d bytes spliced, got %d", want, n)
	}

if tc.limitReadSize > 0 {
		wantN := 0
		if tc.limitReadSize > size {
			wantN = tc.limitReadSize - size
		}

if n := r.(*io.LimitedReader).N; n != int64(wantN) {
			t.Errorf("r.N = %d, want %d", n, wantN)
		}
	}
}

func testSpliceReaderAtEOF(t *testing.T, upNet, downNet string) {
	clientUp, serverUp, err := spliceTestSocketPair(upNet)
	if err != nil {
		t.Fatal(err)
	}
	defer clientUp.Close()
	clientDown, serverDown, err := spliceTestSocketPair(downNet)
	if err != nil {
		t.Fatal(err)
	}
	defer clientDown.Close()

serverUp.Close()

// We'd like to call net.splice here and check the handled return
	// value, but we disable splice on old Linux kernels.
	//
	// In that case, poll.Splice and net.splice return a non-nil error
	// and handled == false. We'd ideally like to see handled == true
	// because the source reader is at EOF, but if we're running on an old
	// kernel, and splice is disabled, we won't see EOF from net.splice,
	// because we won't touch the reader at all.
	//
	// Trying to untangle the errors from net.splice and match them
	// against the errors created by the poll package would be brittle,
	// so this is a higher level test.
	//
	// The following ReadFrom should return immediately, regardless of
	// whether splice is disabled or not. The other side should then
	// get a goodbye signal. Test for the goodbye signal.
	msg := "bye"
	go func() {
		serverDown.(io.ReaderFrom).ReadFrom(serverUp)
		io.WriteString(serverDown, msg)
		serverDown.Close()
	}()

buf := make([]byte, 3)
	_, err = io.ReadFull(clientDown, buf)
	if err != nil {
		t.Errorf("clientDown: %v", err)
	}
	if string(buf) != msg {
		t.Errorf("clientDown got %q, want %q", buf, msg)
	}
}

func testSpliceIssue25985(t *testing.T, upNet, downNet string) {
	front, err := newLocalListener(upNet)
	if err != nil {
		t.Fatal(err)
	}
	defer front.Close()
	back, err := newLocalListener(downNet)
	if err != nil {
		t.Fatal(err)
	}
	defer back.Close()

var wg sync.WaitGroup
	wg.Add(2)

proxy := func() {
		src, err := front.Accept()
		if err != nil {
			return
		}
		dst, err := Dial(downNet, back.Addr().String())
		if err != nil {
			return
		}
		defer dst.Close()
		defer src.Close()
		go func() {
			io.Copy(src, dst)
			wg.Done()
		}()
		go func() {
			io.Copy(dst, src)
			wg.Done()
		}()
	}

go proxy()

toFront, err := Dial(upNet, front.Addr().String())
	if err != nil {
		t.Fatal(err)
	}

io.WriteString(toFront, "foo")
	toFront.Close()

fromProxy, err := back.Accept()
	if err != nil {
		t.Fatal(err)
	}
	defer fromProxy.Close()

_, err = ioutil.ReadAll(fromProxy)
	if err != nil {
		t.Fatal(err)
	}

wg.Wait()
}

func testSpliceNoUnixpacket(t *testing.T) {
	clientUp, serverUp, err := spliceTestSocketPair("unixpacket")
	if err != nil {
		t.Fatal(err)
	}
	defer clientUp.Close()
	defer serverUp.Close()
	clientDown, serverDown, err := spliceTestSocketPair("tcp")
	if err != nil {
		t.Fatal(err)
	}
	defer clientDown.Close()
	defer serverDown.Close()
	// If splice called poll.Splice here, we'd get err == syscall.EINVAL
	// and handled == false.  If poll.Splice gets an EINVAL on the first
	// try, it assumes the kernel it's running on doesn't support splice
	// for unix sockets and returns handled == false. This works for our
	// purposes by somewhat of an accident, but is not entirely correct.
	//
	// What we want is err == nil and handled == false, i.e. we never
	// called poll.Splice, because we know the unix socket's network.
	_, err, handled := splice(serverDown.(*TCPConn).fd, serverUp)
	if err != nil || handled != false {
		t.Fatalf("got err = %v, handled = %t, want nil error, handled == false", err, handled)
	}
}

func testSpliceNoUnixgram(t *testing.T) {
	addr, err := ResolveUnixAddr("unixgram", testUnixAddr())
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(addr.Name)
	up, err := ListenUnixgram("unixgram", addr)
	if err != nil {
		t.Fatal(err)
	}
	defer up.Close()
	clientDown, serverDown, err := spliceTestSocketPair("tcp")
	if err != nil {
		t.Fatal(err)
	}
	defer clientDown.Close()
	defer serverDown.Close()
	// Analogous to testSpliceNoUnixpacket.
	_, err, handled := splice(serverDown.(*TCPConn).fd, up)
	if err != nil || handled != false {
		t.Fatalf("got err = %v, handled = %t, want nil error, handled == false", err, handled)
	}
}

func BenchmarkSplice(b *testing.B) {
	testHookUninstaller.Do(uninstallTestHooks)

b.Run("tcp-to-tcp", func(b *testing.B) { benchSplice(b, "tcp", "tcp") })
	b.Run("unix-to-tcp", func(b *testing.B) { benchSplice(b, "unix", "tcp") })
}

func benchSplice(b *testing.B, upNet, downNet string) {
	for i := 0; i <= 10; i++ {
		chunkSize := 1 << uint(i+10)
		tc := spliceTestCase{
			upNet:     upNet,
			downNet:   downNet,
			chunkSize: chunkSize,
		}

b.Run(strconv.Itoa(chunkSize), tc.bench)
	}
}

func (tc spliceTestCase) bench(b *testing.B) {
	// To benchmark the genericReadFrom code path, set this to false.
	useSplice := true

clientUp, serverUp, err := spliceTestSocketPair(tc.upNet)
	if err != nil {
		b.Fatal(err)
	}
	defer serverUp.Close()

cleanup, err := startSpliceClient(clientUp, "w", tc.chunkSize, tc.chunkSize*b.N)
	if err != nil {
		b.Fatal(err)
	}
	defer cleanup()

clientDown, serverDown, err := spliceTestSocketPair(tc.downNet)
	if err != nil {
		b.Fatal(err)
	}
	defer serverDown.Close()

cleanup, err = startSpliceClient(clientDown, "r", tc.chunkSize, tc.chunkSize*b.N)
	if err != nil {
		b.Fatal(err)
	}
	defer cleanup()

b.SetBytes(int64(tc.chunkSize))
	b.ResetTimer()

if useSplice {
		_, err := io.Copy(serverDown, serverUp)
		if err != nil {
			b.Fatal(err)
		}
	} else {
		type onlyReader struct {
			io.Reader
		}
		_, err := io.Copy(serverDown, onlyReader{serverUp})
		if err != nil {
			b.Fatal(err)
		}
	}
}

func spliceTestSocketPair(net string) (client, server Conn, err error) {
	ln, err := newLocalListener(net)
	if err != nil {
		return nil, nil, err
	}
	defer ln.Close()
	var cerr, serr error
	acceptDone := make(chan struct{})
	go func() {
		server, serr = ln.Accept()
		acceptDone <- struct{}{}
	}()
	client, cerr = Dial(ln.Addr().Network(), ln.Addr().String())
	<-acceptDone
	if cerr != nil {
		if server != nil {
			server.Close()
		}
		return nil, nil, cerr
	}
	if serr != nil {
		if client != nil {
			client.Close()
		}
		return nil, nil, serr
	}
	return client, server, nil
}

func startSpliceClient(conn Conn, op string, chunkSize, totalSize int) (func(), error) {
	f, err := conn.(interface{ File() (*os.File, error) }).File()
	if err != nil {
		return nil, err
	}

cmd := exec.Command(os.Args[0], os.Args[1:]...)
	cmd.Env = []string{
		"GO_NET_TEST_SPLICE=1",
		"GO_NET_TEST_SPLICE_OP=" + op,
		"GO_NET_TEST_SPLICE_CHUNK_SIZE=" + strconv.Itoa(chunkSize),
		"GO_NET_TEST_SPLICE_TOTAL_SIZE=" + strconv.Itoa(totalSize),
		"TMPDIR=" + os.Getenv("TMPDIR"),
	}
	cmd.ExtraFiles = append(cmd.ExtraFiles, f)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

if err := cmd.Start(); err != nil {
		return nil, err
	}

donec := make(chan struct{})
	go func() {
		cmd.Wait()
		conn.Close()
		f.Close()
		close(donec)
	}()

return func() {
		select {
		case <-donec:
		case <-time.After(5 * time.Second):
			log.Printf("killing splice client after 5 second shutdown timeout")
			cmd.Process.Kill()
			select {
			case <-donec:
			case <-time.After(5 * time.Second):
				log.Printf("splice client didn't die after 10 seconds")
			}
		}
	}, nil
}

func init() {
	if os.Getenv("GO_NET_TEST_SPLICE") == "" {
		return
	}
	defer os.Exit(0)

f := os.NewFile(uintptr(3), "splice-test-conn")
	defer f.Close()

conn, err := FileConn(f)
	if err != nil {
		log.Fatal(err)
	}

var chunkSize int
	if chunkSize, err = strconv.Atoi(os.Getenv("GO_NET_TEST_SPLICE_CHUNK_SIZE")); err != nil {
		log.Fatal(err)
	}
	buf := make([]byte, chunkSize)

var totalSize int
	if totalSize, err = strconv.Atoi(os.Getenv("GO_NET_TEST_SPLICE_TOTAL_SIZE")); err != nil {
		log.Fatal(err)
	}

var fn func([]byte) (int, error)
	switch op := os.Getenv("GO_NET_TEST_SPLICE_OP"); op {
	case "r":
		fn = conn.Read
	case "w":
		defer conn.Close()

fn = conn.Write
	default:
		log.Fatalf("unknown op %q", op)
	}

var n int
	for count := 0; count < totalSize; count += n {
		if count+chunkSize > totalSize {
			buf = buf[:totalSize-count]
		}

var err error
		if n, err = fn(buf); err != nil {
			return
		}
	}
}

${this.title}

Code Editor : splice_test.go