mirror of
https://github.com/danderson/netboot.git
synced 2025-10-16 10:01:20 +02:00
178a83f3e0
The Next/Packet/Err structure makes read loops much cleaner, and presents a more familiar interface for people familiar with the stdlib.
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package pcap
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/kr/pretty"
|
|
)
|
|
|
|
func TestFiles(t *testing.T) {
|
|
for _, fname := range []string{"usec", "nsec"} {
|
|
f, err := os.Open(fmt.Sprintf("testdata/%s.pcap", fname))
|
|
if err != nil {
|
|
t.Fatalf("Opening test input file: %s", err)
|
|
}
|
|
r, err := NewReader(f)
|
|
if err != nil {
|
|
t.Fatalf("Creating pcap reader: %s", err)
|
|
}
|
|
if r.LinkType != LinkEthernet {
|
|
t.Errorf("Expected link type %d, got %d", LinkEthernet, r.LinkType)
|
|
}
|
|
|
|
pkts := []*Packet{}
|
|
for r.Next() {
|
|
pkts = append(pkts, r.Packet())
|
|
}
|
|
if r.Err() != nil {
|
|
t.Fatalf("Reading packets from %s.pcap: %s", fname, r.Err())
|
|
}
|
|
|
|
res := pretty.Sprintf("%# v", pkts)
|
|
expectedFile := fmt.Sprintf("testdata/%s.parsed", fname)
|
|
expected, err := ioutil.ReadFile(expectedFile)
|
|
if err != nil {
|
|
t.Fatalf("Reading expected file: %s", err)
|
|
}
|
|
if res != string(expected) {
|
|
if os.Getenv("UPDATE_TESTDATA") != "" {
|
|
ioutil.WriteFile(expectedFile, []byte(res), 0644)
|
|
t.Errorf("%s.pcap didn't decode to %s.parsed (updated %s.parsed)", fname, fname, fname)
|
|
} else {
|
|
t.Fatalf("%s.pcap didn't decode to %s.parsed (rerun with UPDATE_TESTDATA=1 to get diff)", fname, fname)
|
|
}
|
|
}
|
|
}
|
|
}
|