moved scripts to separate modules

Author: avneesh-akto

Dockerfile

@@ -12,13 +12,7 @@ RUN go mod download
 COPY *.go ./
 COPY db ./db
 COPY utils ./utils
-COPY tcpdump.sh ./
-COPY cleanup.sh ./
 
 RUN go build -o /mirroring-api-logging
-RUN chmod +x tcpdump.sh;
-RUN chmod +x cleanup.sh
-
-EXPOSE 4789/udp
 
 CMD ["/bin/sh", "-c", "mkdir -p /app/files && /mirroring-api-logging"]

main.go

@@ -615,7 +615,7 @@ func readTCPFileAndProcess(file fs.FileInfo) {
 	}
 
 	log.Println("Begin reading file: " + file.Name() + " size (" + strconv.FormatInt(file.Size(), 10) + ")")
-	start := time.Now().Unix()
+	start := time.Now().UnixMilli()
 
 	assemblerMap = make(map[int]*tcpassembly.Assembler)
 
@@ -627,6 +627,6 @@ func readTCPFileAndProcess(file fs.FileInfo) {
 		handle.Close()
 	}
 
-	end := time.Now().Unix()
+	end := time.Now().UnixMilli()
 	log.Println("Finished reading file: "+file.Name()+" size ("+strconv.FormatInt(file.Size(), 10)+")"+" in ", end-start, "ms")
 }

scripts/cleanup/Dockerfile

@@ -0,0 +1,14 @@
+# Use an official lightweight base image with tcpdump installed
+FROM alpine:latest
+
+# Install necessary packages
+RUN apk add --no-cache tcpdump
+
+# Copy your script to the container
+COPY cleanup.sh /usr/local/bin/cleanup.sh
+
+# Make the script executable
+RUN chmod +x /usr/local/bin/cleanup.sh
+
+# Set the default command to run your script
+CMD ["/usr/local/bin/cleanup.sh"]

cleanup.sh renamed to scripts/cleanup/cleanup.sh

@@ -1,7 +1,7 @@
 #!/bin/sh
 
 max=80
-base_dir="/app/files"  # Base directory for all files
+base_dir="/files"  # Base directory for all files
 
 while true
 do

scripts/tcpdump/Dockerfile

@@ -0,0 +1,14 @@
+# Use an official lightweight base image with tcpdump installed
+FROM alpine:latest
+
+# Install necessary packages
+RUN apk add --no-cache tcpdump
+
+# Copy your script to the container
+COPY tcpdump.sh /usr/local/bin/tcpdump.sh
+
+# Make the script executable
+RUN chmod +x /usr/local/bin/tcpdump.sh
+
+# Set the default command to run your script
+CMD ["/usr/local/bin/tcpdump.sh"]

tcpdump.sh renamed to scripts/tcpdump/tcpdump.sh

@@ -15,7 +15,7 @@ modules=$(echo "$modules" | tr -d -c 0-9)
 # Create directories if they don't exist
 i=1
 while [ $i -le $modules ]; do
-  dir="/app/files/files_$i"
+  dir="/files/files_$i"
 
   # Check if the directory exists, and create it only if it doesn't exist
   if [ ! -d "$dir" ]; then
@@ -34,7 +34,7 @@ while true; do
   index=$((counter % modules + 1))
 
   # Set the directory based on the index
-  dir="/app/files/files_$index"
+  dir="/files/files_$index"
 
   # Write to the corresponding directory
   tcpdump -i "$interface" port not 22 -w "$dir/%s.pcap" -G 30 -W 1 -K -n