mirror of
https://github.com/golang/go
synced 2024-10-05 16:41:21 -06:00
34 lines
721 B
Go
34 lines
721 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
|
||
|
)
|
||
|
|
||
|
func installTestHooks() {
|
||
|
socketFunc = sw.Socket
|
||
|
closeFunc = sw.Closesocket
|
||
|
connectFunc = sw.Connect
|
||
|
connectExFunc = sw.ConnectEx
|
||
|
}
|
||
|
|
||
|
func uninstallTestHooks() {
|
||
|
socketFunc = origSocket
|
||
|
closeFunc = origClosesocket
|
||
|
connectFunc = origConnect
|
||
|
connectExFunc = origConnectEx
|
||
|
}
|
||
|
|
||
|
func forceCloseSockets() {
|
||
|
for s := range sw.Sockets() {
|
||
|
closeFunc(s)
|
||
|
}
|
||
|
}
|