From 134d923775c47f3a044dbd97756598b3eac0f1fa Mon Sep 17 00:00:00 2001
From: Neale Pickett <neale@lanl.gov>
Date: Mon, 11 Aug 2014 16:00:14 -0600
Subject: [PATCH] Pcap passes its tests

---
 src/pcap/pcap.go      |  8 ++++++--
 src/pcap/pcap_test.go | 26 +++++++++++++++++---------
 2 files changed, 23 insertions(+), 11 deletions(-)

diff --git a/src/pcap/pcap.go b/src/pcap/pcap.go
index 6078039..e08e5ba 100644
--- a/src/pcap/pcap.go
+++ b/src/pcap/pcap.go
@@ -2,9 +2,9 @@ package pcap
 
 import (
 	"time"
-	"fmt"
 	"io"
 	"encoding/binary"
+	"fmt"
 )
 
 const MAXFRAME = 9000
@@ -95,7 +95,7 @@ func NewReader(r io.Reader) (*Reader, error) {
 	return ret, nil
 }
 
-func (r *Reader) Read() (*Frame, error) {
+func (r *Reader) ReadFrame() (*Frame, error) {
 	var h FrameHeader
 
 	err := binary.Read(r.r, r.order, &h)
@@ -141,6 +141,10 @@ func NewWriter(w io.Writer) (*Writer, error) {
 }
 
 func (w *Writer) WriteFrame(frame Frame) (error) {
+	if frame.Header.Caplen != uint32(len(frame.Payload)) {
+		return fmt.Errorf("Caplen != len(Payload)")
+	}
+	
 	if err := binary.Write(w.w, w.order, frame.Header); err != nil {
 		return err
 	}
diff --git a/src/pcap/pcap_test.go b/src/pcap/pcap_test.go
index 1602bcf..209592e 100644
--- a/src/pcap/pcap_test.go
+++ b/src/pcap/pcap_test.go
@@ -6,7 +6,6 @@ import (
 	"math/rand"
 	"os"
 	"testing"
-	"time"
 )
 
 func randomString(prng *rand.Rand, length int) string {
@@ -32,7 +31,7 @@ func TestReader(t *testing.T) {
 	}
 	
 	for {
-		frame, err := pin.Read()
+		frame, err := pin.ReadFrame()
 		if err == io.EOF {
 			break
 		}
@@ -47,7 +46,7 @@ func TestReader(t *testing.T) {
 }
 
 func TestReadWrite(t *testing.T) {
-	prng := rand.New(rand.NewSource(time.Now().Unix()))
+	prng := rand.New(rand.NewSource(58))
 	frames := make([]Frame, 20)
 	
 	outf := new(bytes.Buffer)
@@ -56,28 +55,37 @@ func TestReadWrite(t *testing.T) {
 		t.Fatal(err)
 	}
 	
-	for _, f := range frames {
+	for i, f := range frames {
 		f.Header.Sec = prng.Uint32()
 		f.Header.Usec = prng.Uint32()
-		f.Header.Caplen = prng.Uint32()
 		f.Header.Framelen = prng.Uint32()
 		f.Payload = randomString(prng, prng.Intn(400))
+		f.Header.Caplen = uint32(len(f.Payload))
 		
-		pout.WriteFrame(f)
+		err := pout.WriteFrame(f); if err != nil {
+			t.Fatal(err)
+		}
+		
+		frames[i] = f
 	}
 	
 	pin, err := NewReader(outf)
 	if err != nil {
 		t.Fatal(err)
 	}
-	for _, f := range frames {
-		ff, err := pin.Read()
+	
+	if pout.Header != pin.Header {
+		t.Fatal("Headers differ")
+	}
+	
+	for i, f := range frames {
+		ff, err := pin.ReadFrame()
 		if err != nil {
 			t.Fatal(err)
 		}
 		
 		if *ff != f {
-			t.Fatal(ff, f)
+			t.Fatalf("Frame %d compare failed", i)
 		}
 	}		
 }