github.com/c9s/go@v0.0.0-20180120015821-984e81f64e0c/src/net/main_unix_test.go (about)

     1  // Copyright 2015 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  // +build darwin dragonfly freebsd linux nacl netbsd openbsd solaris
     6  
     7  package net
     8  
     9  import "internal/poll"
    10  
    11  var (
    12  	// Placeholders for saving original socket system calls.
    13  	origSocket        = socketFunc
    14  	origClose         = poll.CloseFunc
    15  	origConnect       = connectFunc
    16  	origListen        = listenFunc
    17  	origAccept        = poll.AcceptFunc
    18  	origGetsockoptInt = getsockoptIntFunc
    19  
    20  	extraTestHookInstallers   []func()
    21  	extraTestHookUninstallers []func()
    22  )
    23  
    24  func installTestHooks() {
    25  	socketFunc = sw.Socket
    26  	poll.CloseFunc = sw.Close
    27  	connectFunc = sw.Connect
    28  	listenFunc = sw.Listen
    29  	poll.AcceptFunc = sw.Accept
    30  	getsockoptIntFunc = sw.GetsockoptInt
    31  
    32  	for _, fn := range extraTestHookInstallers {
    33  		fn()
    34  	}
    35  }
    36  
    37  func uninstallTestHooks() {
    38  	socketFunc = origSocket
    39  	poll.CloseFunc = origClose
    40  	connectFunc = origConnect
    41  	listenFunc = origListen
    42  	poll.AcceptFunc = origAccept
    43  	getsockoptIntFunc = origGetsockoptInt
    44  
    45  	for _, fn := range extraTestHookUninstallers {
    46  		fn()
    47  	}
    48  }
    49  
    50  // forceCloseSockets must be called only from TestMain.
    51  func forceCloseSockets() {
    52  	for s := range sw.Sockets() {
    53  		poll.CloseFunc(s)
    54  	}
    55  }