neale
/
fluffy
mirror of https://github.com/dirtbags/fluffy.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Work on pcap.go

This commit is contained in:
Neale Pickett 2014-03-17 14:32:54 -06:00
parent ceae73c1bb
commit 6d3d01630b
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
pcap.go
View File

@ -9,8 +9,9 @@ import (
"encoding/binary" "encoding/binary"
) )
const MAGIC = 0xa1b2c3d4
const MAXFRAME = 9000 const MAXFRAME = 9000
const MAGIC_BE = "\xa1\xb2\xc3\xd4"
const MAGIC_LE = "\xd4\xc3\xb2\xa1"
const LINKTYPE_ETHERNET = 1 const LINKTYPE_ETHERNET = 1
const LINKTYPE_RAW = 101 const LINKTYPE_RAW = 101
@ -57,16 +58,22 @@ type Writer struct {
func NewReader(r io.Reader) (*Reader, error) { func NewReader(r io.Reader) (*Reader, error) {
var h FileHeader var h FileHeader
var magic uint32 var order binary.ByteOrder
var order binary.ByteOrder = binary.BigEndian magic := make([]byte, 4)
if err := binary.Read(r, order, &magic); err != nil { if err := r.Read(&magic); err != nil {
return nil, err return nil, err
} }
if magic != MAGIC { switch (magic) {
case MAGIC_BE:
order = binary.BigEndian
case MAGIC_LE:
order = binary.LittleEndian order = binary.LittleEndian
default:
return nil, fmt.Errorf("Cannot determine endianness")
} }
if err := binary.Read(r, order, &h); err != nil { if err := binary.Read(r, order, &h); err != nil {
return nil, err return nil, err
} }