97 lines
3.5 KiB
Go
97 lines
3.5 KiB
Go
|
// Copyright 2016 Mikio Hara. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
// +build freebsd netbsd
|
||
|
|
|
||
|
|
package tcpinfo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"runtime"
|
||
|
|
"time"
|
||
|
|
"unsafe"
|
||
|
|
|
||
|
|
"github.com/mikioh/tcpopt"
|
||
|
|
)
|
||
|
|
|
||
|
|
var options = [soMax]option{
|
||
|
|
soInfo: {ianaProtocolTCP, sysTCP_INFO, parseInfo},
|
||
|
|
}
|
||
|
|
|
||
|
|
// Marshal implements the Marshal method of tcpopt.Option interface.
|
||
|
|
func (i *Info) Marshal() ([]byte, error) { return (*[sizeofTCPInfo]byte)(unsafe.Pointer(i))[:], nil }
|
||
|
|
|
||
|
|
// A SysInfo represents platform-specific information.
|
||
|
|
type SysInfo struct {
|
||
|
|
SenderWindowBytes uint `json:"snd_wnd_bytes"` // advertised sender window in bytes [FreeBSD]
|
||
|
|
SenderWindowSegs uint `json:"snd_wnd_segs"` // advertised sender window in # of segments [NetBSD]
|
||
|
|
NextEgressSeq uint `json:"egress_seq"` // next egress seq. number
|
||
|
|
NextIngressSeq uint `json:"ingress_seq"` // next ingress seq. number
|
||
|
|
RetransSegs uint `json:"retrans_segs"` // # of retransmit segments sent
|
||
|
|
OutOfOrderSegs uint `json:"ooo_segs"` // # of out-of-order segments received
|
||
|
|
ZeroWindowUpdates uint `json:"zerownd_updates"` // # of zero-window updates sent
|
||
|
|
Offloading bool `json:"offloading"` // TCP offload processing
|
||
|
|
}
|
||
|
|
|
||
|
|
var sysStates = [11]State{Closed, Listen, SynSent, SynReceived, Established, CloseWait, FinWait1, Closing, LastAck, FinWait2, TimeWait}
|
||
|
|
|
||
|
|
func parseInfo(b []byte) (tcpopt.Option, error) {
|
||
|
|
if len(b) < sizeofTCPInfo {
|
||
|
|
return nil, errors.New("short buffer")
|
||
|
|
}
|
||
|
|
ti := (*tcpInfo)(unsafe.Pointer(&b[0]))
|
||
|
|
i := &Info{State: sysStates[ti.State]}
|
||
|
|
if ti.Options&sysTCPI_OPT_WSCALE != 0 {
|
||
|
|
i.Options = append(i.Options, WindowScale(ti.Pad_cgo_0[0]>>4))
|
||
|
|
i.PeerOptions = append(i.PeerOptions, WindowScale(ti.Pad_cgo_0[0]&0x0f))
|
||
|
|
}
|
||
|
|
if ti.Options&sysTCPI_OPT_SACK != 0 {
|
||
|
|
i.Options = append(i.Options, SACKPermitted(true))
|
||
|
|
i.PeerOptions = append(i.PeerOptions, SACKPermitted(true))
|
||
|
|
}
|
||
|
|
if ti.Options&sysTCPI_OPT_TIMESTAMPS != 0 {
|
||
|
|
i.Options = append(i.Options, Timestamps(true))
|
||
|
|
i.PeerOptions = append(i.PeerOptions, Timestamps(true))
|
||
|
|
}
|
||
|
|
i.SenderMSS = MaxSegSize(ti.Snd_mss)
|
||
|
|
i.ReceiverMSS = MaxSegSize(ti.Rcv_mss)
|
||
|
|
i.RTT = time.Duration(ti.Rtt) * time.Microsecond
|
||
|
|
i.RTTVar = time.Duration(ti.Rttvar) * time.Microsecond
|
||
|
|
i.RTO = time.Duration(ti.Rto) * time.Microsecond
|
||
|
|
i.ATO = time.Duration(ti.X__tcpi_ato) * time.Microsecond
|
||
|
|
i.LastDataSent = time.Duration(ti.X__tcpi_last_data_sent) * time.Microsecond
|
||
|
|
i.LastDataReceived = time.Duration(ti.Last_data_recv) * time.Microsecond
|
||
|
|
i.LastAckReceived = time.Duration(ti.X__tcpi_last_ack_recv) * time.Microsecond
|
||
|
|
i.FlowControl = &FlowControl{
|
||
|
|
ReceiverWindow: uint(ti.Rcv_space),
|
||
|
|
}
|
||
|
|
i.CongestionControl = &CongestionControl{
|
||
|
|
SenderSSThreshold: uint(ti.Snd_ssthresh),
|
||
|
|
ReceiverSSThreshold: uint(ti.X__tcpi_rcv_ssthresh),
|
||
|
|
}
|
||
|
|
i.Sys = &SysInfo{
|
||
|
|
NextEgressSeq: uint(ti.Snd_nxt),
|
||
|
|
NextIngressSeq: uint(ti.Rcv_nxt),
|
||
|
|
RetransSegs: uint(ti.Snd_rexmitpack),
|
||
|
|
OutOfOrderSegs: uint(ti.Rcv_ooopack),
|
||
|
|
ZeroWindowUpdates: uint(ti.Snd_zerowin),
|
||
|
|
}
|
||
|
|
if ti.Options&sysTCPI_OPT_TOE != 0 {
|
||
|
|
i.Sys.Offloading = true
|
||
|
|
}
|
||
|
|
switch runtime.GOOS {
|
||
|
|
case "freebsd":
|
||
|
|
i.CongestionControl.SenderWindowBytes = uint(ti.Snd_cwnd)
|
||
|
|
i.Sys.SenderWindowBytes = uint(ti.Snd_wnd)
|
||
|
|
case "netbsd":
|
||
|
|
i.CongestionControl.SenderWindowSegs = uint(ti.Snd_cwnd)
|
||
|
|
i.Sys.SenderWindowSegs = uint(ti.Snd_wnd)
|
||
|
|
}
|
||
|
|
return i, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseCCAlgorithmInfo(name string, b []byte) (CCAlgorithmInfo, error) {
|
||
|
|
return nil, errors.New("operation not supported")
|
||
|
|
}
|