github/ping
1
0
Fork 0
mirror of https://github.com/go-ping/ping.git synced 2025-08-01 21:49:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ability to specifiy source IP address

Browse Source
This commit is contained in:
Alirie Gray 2019-05-29 14:47:08 -07:00 committed by Cameron Sparr
parent 5cc5e2921d
commit 3eec4c7c0d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ping.go
View File

@ -137,6 +137,9 @@ type Pinger struct {
// Tracker: Used to uniquely identify packet when non-priviledged
Tracker int64
// Source is the source IP address
Source string
// stop chan bool
done chan bool
@ -144,7 +147,6 @@ type Pinger struct {
addr string
ipv4 bool
source string
size int
id int
sequence int
@ -277,12 +279,12 @@ func (p *Pinger) Run() {
func (p *Pinger) run() {
var conn *icmp.PacketConn
if p.ipv4 {
if conn = p.listen(ipv4Proto[p.network], p.source); conn == nil {
if conn = p.listen(ipv4Proto[p.network]); conn == nil {
return
}
conn.IPv4PacketConn().SetControlMessage(ipv4.FlagTTL, true)
} else {
if conn = p.listen(ipv6Proto[p.network], p.source); conn == nil {
if conn = p.listen(ipv6Proto[p.network]); conn == nil {
return
}
conn.IPv6PacketConn().SetControlMessage(ipv6.FlagHopLimit, true)
@ -570,8 +572,8 @@ func (p *Pinger) sendICMP(conn *icmp.PacketConn) error {
return nil
}
func (p *Pinger) listen(netProto string, source string) *icmp.PacketConn {
conn, err := icmp.ListenPacket(netProto, source)
func (p *Pinger) listen(netProto string) *icmp.PacketConn {
conn, err := icmp.ListenPacket(netProto, p.Source)
if err != nil {
fmt.Printf("Error listening for ICMP packets: %s\n", err.Error())
close(p.done)