Update the packets seen count in the database when added a new packet

main.go

@@ -134,14 +134,12 @@ func capture(dataset string, allowBlock bool, port *string) {
 }
 
 func usage() {
-	fmt.Println("client-cli capture [protocol] [dataset] <port>")
-	fmt.Println("Example: client-cli capture testing allow")
-	fmt.Println("Example: client-cli capture testing allow 80")
-	fmt.Println("Example: client-cli capture testing block")
-	fmt.Println("Example: client-cli capture testing block 443")
+	fmt.Println("AdversaryLabClient capture [protocol] [dataset] <port>")
+	fmt.Println("Example: AdversaryLabClient capture testing allow")
+	fmt.Println("Example: AdversaryLabClient capture testing allow 80")
+	fmt.Println("Example: AdversaryLabClient capture testing block")
+	fmt.Println("Example: AdversaryLabClient capture testing block 443")
 	fmt.Println()
-	fmt.Println("client-cli rules [protocol]")
-	fmt.Println("Example: client-client rules HTTP")
 	os.Exit(1)
 }
 

protocol/client.go

@@ -7,19 +7,25 @@ import (
 )
 
 const (
+	packetStatsKey = "Packet:Stats"
+
 	allowedConnectionsKey = "Allowed:Connections"
 	allowedIncomingKey    = "Allowed:Incoming:Packets"
 	allowedOutgoingKey    = "Allowed:Outgoing:Packets"
 
 	allowedIncomingDatesKey = "Allowed:Incoming:Dates"
 	allowedOutgoingDatesKey = "Allowed:Outgoing:Dates"
 
+	allowedPacketsSeenKey = "Allowed:Connections:Seen"
+
 	blockedConnectionsKey = "Blocked:Connections"
 	blockedIncomingKey    = "Blocked:Incoming:Packets"
 	blockedOutgoingKey    = "Blocked:Outgoing:Packets"
 
 	blockedIncomingDatesKey = "Blocked:Incoming:Dates"
 	blockedOutgoingDatesKey = "Blocked:Outgoing:Dates"
+
+	blockedPacketsSeenKey = "Blocked:Connections:Seen"
 )
 
 // Client holds the connection to the Redis database
@@ -77,12 +83,14 @@ func (client Client) AddTrainPacket(dataset string, allowBlock bool, conn Connec
 		client.conn.Do("hset", allowedIncomingDatesKey, connectionIDString, incomingTime)
 		client.conn.Do("hset", allowedOutgoingDatesKey, connectionIDString, outgoingTime)
 		client.conn.Do("rpush", allowedConnectionsKey, connectionIDString)
+		client.conn.Do("hincrby", packetStatsKey, allowedPacketsSeenKey, "1")
 	} else {
 		client.conn.Do("hset", blockedIncomingKey, connectionIDString, incomingPayload)
 		client.conn.Do("hset", blockedOutgoingKey, connectionIDString, outgoingPayload)
 		client.conn.Do("hset", blockedIncomingDatesKey, connectionIDString, incomingTime)
 		client.conn.Do("hset", blockedOutgoingDatesKey, connectionIDString, outgoingTime)
 		client.conn.Do("rpush", blockedConnectionsKey, connectionIDString)
+		client.conn.Do("hincrby", packetStatsKey, blockedPacketsSeenKey, "1")
 	}
 }