| // 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. | |
| //go:build dragonfly || freebsd || linux || netbsd || openbsd || solaris | |
| package socktest | |
| import "syscall" | |
| // Accept4 wraps syscall.Accept4. | |
| func (sw *Switch) Accept4(s, flags int) (ns int, sa syscall.Sockaddr, err error) { | |
| so := sw.sockso(s) | |
| if so == nil { | |
| return syscall.Accept4(s, flags) | |
| } | |
| sw.fmu.RLock() | |
| f := sw.fltab[FilterAccept] | |
| sw.fmu.RUnlock() | |
| af, err := f.apply(so) | |
| if err != nil { | |
| return -1, nil, err | |
| } | |
| ns, sa, so.Err = syscall.Accept4(s, flags) | |
| if err = af.apply(so); err != nil { | |
| if so.Err == nil { | |
| syscall.Close(ns) | |
| } | |
| return -1, nil, err | |
| } | |
| sw.smu.Lock() | |
| defer sw.smu.Unlock() | |
| if so.Err != nil { | |
| sw.stats.getLocked(so.Cookie).AcceptFailed++ | |
| return -1, nil, so.Err | |
| } | |
| nso := sw.addLocked(ns, so.Cookie.Family(), so.Cookie.Type(), so.Cookie.Protocol()) | |
| sw.stats.getLocked(nso.Cookie).Accepted++ | |
| return ns, sa, nil | |
| } | |