github.com/code-reading/golang@v0.0.0-20220303082512-ba5bc0e589a3/go/src/net/main_posix_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  //go:build !js && !plan9
     6  // +build !js,!plan9
     7  
     8  package net
     9  
    10  import (
    11  	"net/internal/socktest"
    12  	"strings"
    13  	"syscall"
    14  )
    15  
    16  func enableSocketConnect() {
    17  	sw.Set(socktest.FilterConnect, nil)
    18  }
    19  
    20  func disableSocketConnect(network string) {
    21  	ss := strings.Split(network, ":")
    22  	sw.Set(socktest.FilterConnect, func(so *socktest.Status) (socktest.AfterFilter, error) {
    23  		switch ss[0] {
    24  		case "tcp4":
    25  			if so.Cookie.Family() == syscall.AF_INET && so.Cookie.Type() == syscall.SOCK_STREAM {
    26  				return nil, syscall.EHOSTUNREACH
    27  			}
    28  		case "udp4":
    29  			if so.Cookie.Family() == syscall.AF_INET && so.Cookie.Type() == syscall.SOCK_DGRAM {
    30  				return nil, syscall.EHOSTUNREACH
    31  			}
    32  		case "ip4":
    33  			if so.Cookie.Family() == syscall.AF_INET && so.Cookie.Type() == syscall.SOCK_RAW {
    34  				return nil, syscall.EHOSTUNREACH
    35  			}
    36  		case "tcp6":
    37  			if so.Cookie.Family() == syscall.AF_INET6 && so.Cookie.Type() == syscall.SOCK_STREAM {
    38  				return nil, syscall.EHOSTUNREACH
    39  			}
    40  		case "udp6":
    41  			if so.Cookie.Family() == syscall.AF_INET6 && so.Cookie.Type() == syscall.SOCK_DGRAM {
    42  				return nil, syscall.EHOSTUNREACH
    43  			}
    44  		case "ip6":
    45  			if so.Cookie.Family() == syscall.AF_INET6 && so.Cookie.Type() == syscall.SOCK_RAW {
    46  				return nil, syscall.EHOSTUNREACH
    47  			}
    48  		}
    49  		return nil, nil
    50  	})
    51  }