🐛 use standard icmp package

Author: catusax

.gitignore

@@ -1,5 +1,5 @@
 # Binaries for programs and plugins
-lmap
+/lmap
 *.exe
 *.exe~
 *.dll

go.mod

@@ -1,3 +1,8 @@
 module github.com/LinuxHub-Group/lmap
 
 go 1.16
+
+require (
+	github.com/ysmood/got v0.30.0
+	golang.org/x/net v0.0.0-20220607020251-c690dde0001d
+)

go.sum

@@ -0,0 +1,10 @@
+github.com/ysmood/got v0.30.0 h1:n6KVknQ2gjU2FsEVvMrpEw20uoqkhRDvAoNK69bOV/U=
+github.com/ysmood/got v0.30.0/go.mod h1:pE1l4LOwOBhQg6A/8IAatkGp7uZjnalzrZolnlhhMgY=
+golang.org/x/net v0.0.0-20220607020251-c690dde0001d h1:4SFsTMi4UahlKoloni7L4eYzhFRifURQLw+yv0QDCx8=
+golang.org/x/net v0.0.0-20220607020251-c690dde0001d/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a h1:dGzPydgVsqGcTRVwiLJ1jVbufYwmzD3LfVPLKsKg+0k=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=

pkg/lmap/check_ip.go

@@ -32,7 +32,7 @@ func CheckIP(subnet string, isVerbose bool) {
 	t := time.Now()
 	hosts, _ := GetAllIPsFromCIDR(subnet)
 	for index := range hosts {
-		time.Sleep(500)
+		//time.Sleep(500)
 		checkerGroup.Add(1)
 
 		go func(index int) {

pkg/lmap/common.go

@@ -20,14 +20,6 @@ package lmap
 
 import "net"
 
-type ICMP struct {
-	Type        uint8
-	Code        uint8
-	Checksum    uint16
-	Identifier  uint16
-	SequenceNum uint16
-}
-
 type HostInfo struct {
 	host   net.IP
 	isUsed bool

pkg/lmap/ping.go

@@ -19,63 +19,73 @@
 package lmap
 
 import (
-	"bytes"
 	"encoding/binary"
+	"golang.org/x/net/icmp"
 	"log"
 	"net"
 	"time"
 )
 
 func Ping(ip net.IP) bool {
-	var icmp ICMP
-	//开始填充数据包
-	icmp.Type = 8 //8->echo message  0->reply message
+	recvBuf := make([]byte, 8)
 
-	recvBuf := make([]byte, 32)
-	var buffer bytes.Buffer
+	// Start listening for icmp replies
+	conn, err := icmp.ListenPacket("ip4:icmp", "0.0.0.0")
 
-	//先在buffer中写入icmp数据报求去校验和
-	_ = binary.Write(&buffer, binary.BigEndian, icmp)
-	icmp.Checksum = checkSum(buffer.Bytes())
-	//然后清空buffer并把求完校验和的icmp数据报写入其中准备发送
-	buffer.Reset()
-	_ = binary.Write(&buffer, binary.BigEndian, icmp)
-
-	conn, err := net.DialTimeout("ip4:icmp", ip.String(), 1*time.Second)
 	if err != nil {
+		log.Println("dial error:", err)
 		return false
 	}
-	_, err = conn.Write(buffer.Bytes())
+	defer conn.Close()
+
+	sendBytes := []byte{8, 0, 247, 255, 0, 0, 0, 0}
+	//expectCheckSum := int(checkSum([]byte{0, 0, 0, 0, 0, 0, 0, 0}))
+	expectCheckSum := 65535
+
+	_, err = conn.WriteTo(sendBytes, &net.IPAddr{
+		IP: ip,
+	})
 	if err != nil {
-		log.Println("conn.Write error:", err)
 		return false
 	}
 	_ = conn.SetReadDeadline(time.Now().Add(time.Second * 2))
-	num, err := conn.Read(recvBuf)
+
+	_, _, err = conn.ReadFrom(recvBuf)
 	if err != nil {
 		return false
 	}
 
-	_ = conn.SetReadDeadline(time.Time{})
+	recvType := recvBuf[0]
+	recvCheckSum := int(binary.BigEndian.Uint16(recvBuf[2:4]))
 
-	return string(recvBuf[0:num]) != ""
-}
+	_ = conn.SetReadDeadline(time.Time{})
 
-func checkSum(data []byte) uint16 {
-	var (
-		sum    uint32
-		length = len(data)
-		index  int
-	)
-	for length > 1 {
-		sum += uint32(data[index])<<8 + uint32(data[index+1])
-		index += 2
-		length -= 2
+	if recvType != 0 {
+		return false
 	}
-	if length > 0 {
-		sum += uint32(data[index])
+
+	if recvCheckSum != expectCheckSum {
+		return false
 	}
-	sum += sum >> 16
 
-	return uint16(^sum)
+	return true
 }
+
+//func checkSum(data []byte) uint16 {
+//	var (
+//		sum    uint32
+//		length = len(data)
+//		index  int
+//	)
+//	for length > 1 {
+//		sum += uint32(data[index])<<8 + uint32(data[index+1])
+//		index += 2
+//		length -= 2
+//	}
+//	if length > 0 {
+//		sum += uint32(data[index])
+//	}
+//	sum += sum >> 16
+//
+//	return uint16(^sum)
+//}

pkg/lmap/ping_test.go

@@ -0,0 +1,31 @@
+/*
+ *     lmap (LinuxHub's Nmap) is the nmap next generation pro plus max.
+ *     Copyright (C) <2022>  <LinuxHub-Group>
+ *
+ *     This program is free software: you can redistribute it and/or modify
+ *     it under the terms of the GNU General Public License as published by
+ *     the Free Software Foundation, either version 3 of the License, or
+ *     (at your option) any later version.
+ *
+ *     This program is distributed in the hope that it will be useful,
+ *     but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *     GNU General Public License for more details.
+ *
+ *     You should have received a copy of the GNU General Public License
+ *     along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package lmap
+
+import (
+	"github.com/ysmood/got"
+	"net"
+	"testing"
+)
+
+func TestPing(t *testing.T) {
+	ip := net.ParseIP("192.168.0.106")
+	res := Ping(ip)
+	got.T(t).True(res)
+}