add -r option

zaftzaft · Nov 12, 2016 · 9e68102 · 9e68102
1 parent 8f76940
commit 9e68102
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/dcatch.go b/dcatch.go
@@ -12,12 +12,15 @@ import (
 	"time"
 )
 
-var version = "0.0.1"
+var version = "0.0.2"
 
 func main() {
 	var device string
 	flag.StringVar(&device, "i", "any", "interface name")
 
+	var filename string
+	flag.StringVar(&filename, "r", "", "pcap file")
+
 	var showVersion bool
 	flag.BoolVar(&showVersion, "v", false, "show version")
 	flag.BoolVar(&showVersion, "version", false, "show version")
@@ -29,9 +32,18 @@ func main() {
 		return
 	}
 
-	handle, err := pcap.OpenLive(device, 65536, true, time.Second)
-	if err != nil {
-		log.Fatal("pcap openlive err:", err)
+	var handle *pcap.Handle
+	var err error
+	if filename != "" {
+		handle, err = pcap.OpenOffline(filename)
+		if err != nil {
+			log.Fatal("[Error] pcap OpenOffline err", err)
+		}
+	} else {
+		handle, err = pcap.OpenLive(device, 65536, true, time.Second)
+		if err != nil {
+			log.Fatal("pcap openlive err:", err)
+		}
 	}
 
 	defer handle.Close()