1
0
mirror of https://github.com/golang/go synced 2024-10-05 11:31:22 -06:00
go/src/net/main_windows_test.go
Mikio Hara 92b74d0940 net: add missing aborted connection handling on accept test
This change adds TestAcceptIgnoreAbortedConnRequest to test accepting
aborted connection requests on all supported platforms except Plan 9.

Change-Id: I5936b04085184ff348539962289b1167ec4ac619
Reviewed-on: https://go-review.googlesource.com/19707
Reviewed-by: Ian Lance Taylor <iant@golang.org>
Run-TryBot: Mikio Hara <mikioh.mikioh@gmail.com>
TryBot-Result: Gobot Gobot <gobot@golang.org>
2016-02-21 03:32:36 +00:00

40 lines
881 B
Go

// Copyright 2015 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 net
var (
// Placeholders for saving original socket system calls.
origSocket = socketFunc
origClosesocket = closeFunc
origConnect = connectFunc
origConnectEx = connectExFunc
origListen = listenFunc
origAccept = acceptFunc
)
func installTestHooks() {
socketFunc = sw.Socket
closeFunc = sw.Closesocket
connectFunc = sw.Connect
connectExFunc = sw.ConnectEx
listenFunc = sw.Listen
acceptFunc = sw.AcceptEx
}
func uninstallTestHooks() {
socketFunc = origSocket
closeFunc = origClosesocket
connectFunc = origConnect
connectExFunc = origConnectEx
listenFunc = origListen
acceptFunc = origAccept
}
func forceCloseSockets() {
for s := range sw.Sockets() {
closeFunc(s)
}
}