taigrr/go-fastping
1
0
Fork 0
mirror of https://github.com/taigrr/go-fastping synced 2025-01-18 05:03:15 -08:00
Code Issues Projects Releases Wiki Activity

Remove packet struct

Browse Source 
The recv channel isn't used anymore so it doesn't need to pack received
packet bytes and a remote address into a packet struct.

This removes packet struct, fixes procRecv arguments and so on.
This commit is contained in:
Tatsushi Demachi 2015-10-25 14:47:07 +09:00
parent d7610ec331
commit 022e2c3afe
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
fastping.go
View File

@ -110,11 +110,6 @@ func numGoRoutines() int {
return runtime.NumCPU() * 4
}
type packet struct {
bytes []byte
addr net.Addr
}
type context struct {
stop chan bool
done chan bool
@ -642,13 +637,13 @@ func (p *Pinger) recvICMP(conn *icmp.PacketConn, ctx *context, wg *sync.WaitGrou
}
}
}
p.procRecv(&packet{bytes: bytes, addr: ra})
p.procRecv(bytes, ra)
}
}
func (p *Pinger) procRecv(recv *packet) {
func (p *Pinger) procRecv(bytes []byte, ra net.Addr) {
var ipaddr *net.IPAddr
switch adr := recv.addr.(type) {
switch adr := ra.(type) {
case *net.IPAddr:
ipaddr = adr
case *net.UDPAddr:
@ -665,17 +660,13 @@ func (p *Pinger) procRecv(recv *packet) {
}
p.mu.Unlock()
var bytes []byte
var proto int
if isIPv4(ipaddr.IP) {
if p.network == "ip" {
bytes = ipv4Payload(recv.bytes)
} else {
bytes = recv.bytes
bytes = ipv4Payload(bytes)
}
proto = ProtocolICMP
} else if isIPv6(ipaddr.IP) {
bytes = recv.bytes
proto = ProtocolIPv6ICMP
} else {
return