1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
| // 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.
|
| // +build dragonfly freebsd linux netbsd openbsd
|
| 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
| }
|
|
