github/ping
1
0
Fork 0
mirror of https://github.com/go-ping/ping.git synced 2025-06-25 05:31:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

make InflightPackets private

Browse Source
This commit is contained in:
thatmattlove 2022-02-18 07:01:55 -07:00
parent 7d19def034
commit 035eb2feee
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ping.go
View File

@ -108,7 +108,7 @@ func New(addr string) *Pinger {
ipv4: false,
network: "ip",
protocol: "udp",
InFlightPackets: firstSequence,
inFlightPackets: firstSequence,
TTL: 64,
logger: StdLogger{Logger: log.New(log.Writer(), log.Prefix(), log.Flags())},
}
@ -206,7 +206,7 @@ type Pinger struct {
id int
sequence int
// awaitingSequences are in-flight sequence numbers we keep track of to help remove duplicate receipts
InFlightPackets map[uuid.UUID]map[int]InFlightPacket
inFlightPackets map[uuid.UUID]map[int]InFlightPacket
// network is one of "ip", "ip4", or "ip6".
network string
// protocol is "icmp" or "udp".
@ -526,10 +526,10 @@ func (p *Pinger) CheckInFlightPackets() {
return
}
currentTime := time.Now()
for id, inflight := range p.InFlightPackets {
for id, inflight := range p.inFlightPackets {
for seq, pkt := range inflight {
if pkt.DispatchedTime.Add(p.PacketTimeout).Before(currentTime) {
delete(p.InFlightPackets[id], seq)
delete(p.inFlightPackets[id], seq)
if p.OnTimeout != nil {
p.OnTimeout(seq, &pkt)
}
@ -712,7 +712,7 @@ func (p *Pinger) processPacket(recv *packet) error {
inPkt.Rtt = receivedAt.Sub(timestamp)
inPkt.Seq = pkt.Seq
// If we've already received this sequence, ignore it.
if _, inflight := p.InFlightPackets[*pktUUID][pkt.Seq]; !inflight {
if _, inflight := p.inFlightPackets[*pktUUID][pkt.Seq]; !inflight {
p.PacketsRecvDuplicates++
if p.OnDuplicateRecv != nil {
p.OnDuplicateRecv(inPkt)
@ -720,7 +720,7 @@ func (p *Pinger) processPacket(recv *packet) error {
return nil
}
// remove it from the list of sequences we're waiting for so we don't get duplicates.
delete(p.InFlightPackets[*pktUUID], pkt.Seq)
delete(p.inFlightPackets[*pktUUID], pkt.Seq)
p.updateStatistics(inPkt)
default:
// Very bad, not sure how this can happen
@ -789,7 +789,7 @@ func (p *Pinger) sendICMP(conn packetConn) error {
handler(outPkt)
}
// mark this sequence as in-flight
p.InFlightPackets[currentUUID][p.sequence] = InFlightPacket{
p.inFlightPackets[currentUUID][p.sequence] = InFlightPacket{
DispatchedTime: time.Now(),
}
p.PacketsSent++
@ -797,7 +797,7 @@ func (p *Pinger) sendICMP(conn packetConn) error {
if p.sequence > 65535 {
newUUID := uuid.New()
p.trackerUUIDs = append(p.trackerUUIDs, newUUID)
p.InFlightPackets[newUUID] = make(map[int]InFlightPacket)
p.inFlightPackets[newUUID] = make(map[int]InFlightPacket)
p.sequence = 0
}
break

4
ping_test.go
View File

@ -38,7 +38,7 @@ func TestProcessPacket(t *testing.T) {
Seq: pinger.sequence,
Data: data,
}
pinger.InFlightPackets[currentUUID][pinger.sequence] = InFlightPacket{}
pinger.inFlightPackets[currentUUID][pinger.sequence] = InFlightPacket{}
msg := &icmp.Message{
Type: ipv4.ICMPTypeEchoReply,
@ -608,7 +608,7 @@ func TestProcessPacket_IgnoresDuplicateSequence(t *testing.T) {
Data: data,
}
// register the sequence as sent
pinger.InFlightPackets[currentUUID][0] = InFlightPacket{}
pinger.inFlightPackets[currentUUID][0] = InFlightPacket{}
msg := &icmp.Message{
Type: ipv4.ICMPTypeEchoReply,