go test: properly pad message

Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
This commit is contained in:
Jason A. Donenfeld 2017-06-01 06:31:26 +02:00
parent 32afe0e220
commit a1e931f9dc
1 changed files with 2 additions and 2 deletions

View File

@ -127,7 +127,7 @@ func main() {
Dst: net.IPv4(10, 189, 129, 1), Dst: net.IPv4(10, 189, 129, 1),
}).Marshal() }).Marshal()
binary.BigEndian.PutUint16(pingHeader[2:], uint16(ipv4.HeaderLen+len(pingMessage))) // fix the length endianness on BSDs binary.BigEndian.PutUint16(pingHeader[2:], uint16(ipv4.HeaderLen+len(pingMessage))) // fix the length endianness on BSDs
pingData := append(pingHeader, pingMessage...) pingData := append(append(pingHeader, pingMessage...), 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
binary.BigEndian.PutUint16(pingData[10:], ipChecksum(pingData)) binary.BigEndian.PutUint16(pingData[10:], ipChecksum(pingData))
pingPacket := make([]byte, 16) pingPacket := make([]byte, 16)
pingPacket[0] = 4 // Type: Data pingPacket[0] = 4 // Type: Data
@ -142,7 +142,7 @@ func main() {
} }
// read ICMP Echo Reply packet // read ICMP Echo Reply packet
replyPacket := make([]byte, 128) replyPacket := make([]byte, 80)
n, err = conn.Read(replyPacket) n, err = conn.Read(replyPacket)
if err != nil { if err != nil {
log.Fatalf("error reading ping reply message: %s", err) log.Fatalf("error reading ping reply message: %s", err)