|
| 1 | +#!/bin/bash |
| 2 | +IMAGE_NAME="docker_ipfixprobe" |
| 3 | + |
| 4 | +# Run the ipfixprobe on the input pcap file with defined script, and save the output CSV file to the output path. |
| 5 | +PROCESS_SCRIPT_PATH=$1 |
| 6 | +INPUT_FILE_PATH=$2 |
| 7 | +OUTPUT_CSV_PATH=$3 |
| 8 | + |
| 9 | +if [ -z "$PROCESS_SCRIPT_PATH" ] || [ -z "$INPUT_FILE_PATH" ] || [ -z "$OUTPUT_CSV_PATH" ] ; then |
| 10 | + echo "Usage: $0 <process_script> <input_file_path> <output_csv_path>" |
| 11 | + exit 1 |
| 12 | +fi |
| 13 | + |
| 14 | +CONT_BIN="$(which podman 2>/dev/null)" |
| 15 | +if [ -z "$CONT_BIN" ]; then |
| 16 | + CONT_BIN="$(which docker 2>/dev/null)" |
| 17 | +fi |
| 18 | +if [ -z "$CONT_BIN" ]; then |
| 19 | + echo "Missing podman or docker." |
| 20 | + exit 2 |
| 21 | +fi |
| 22 | + |
| 23 | +# Check if the Docker image exists |
| 24 | +if ! $CONT_BIN image inspect "$IMAGE_NAME" >/dev/null 2>&1; then |
| 25 | + echo "Docker image '$IMAGE_NAME' not found. Attempting to build it..." |
| 26 | + |
| 27 | + # Determine the script directory |
| 28 | + SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| 29 | + DOCKERFILE_PATH="$SCRIPT_DIR/Dockerfile" |
| 30 | + |
| 31 | + if [ ! -f "$DOCKERFILE_PATH" ]; then |
| 32 | + echo "Dockerfile not found at $DOCKERFILE_PATH" |
| 33 | + exit 3 |
| 34 | + fi |
| 35 | + |
| 36 | + # Build the Docker image |
| 37 | + echo "Building Docker image '$IMAGE_NAME'..." |
| 38 | + $CONT_BIN build -t "$IMAGE_NAME" -f "$DOCKERFILE_PATH" "$SCRIPT_DIR" |
| 39 | + |
| 40 | + if [ $? -ne 0 ]; then |
| 41 | + echo "Failed to build Docker image." |
| 42 | + exit 4 |
| 43 | + fi |
| 44 | +fi |
| 45 | + |
| 46 | + |
| 47 | +INPUT_FILE=$(basename "$INPUT_FILE_PATH") |
| 48 | +PROCESS_SCRIPT=$(basename "$PROCESS_SCRIPT_PATH") |
| 49 | +TMP_FOLDER="$(mktemp -d)" |
| 50 | + |
| 51 | +cp "$INPUT_FILE_PATH" "$TMP_FOLDER/$INPUT_FILE" |
| 52 | +cp "$PROCESS_SCRIPT_PATH" "$TMP_FOLDER/$PROCESS_SCRIPT" |
| 53 | +chmod +x "$TMP_FOLDER/$PROCESS_SCRIPT" |
| 54 | + |
| 55 | +"$CONT_BIN" run --privileged --rm -v $TMP_FOLDER:/output "$IMAGE_NAME" "/output/$PROCESS_SCRIPT \"$INPUT_FILE\"" |
| 56 | +[ -f "$TMP_FOLDER/$INPUT_FILE.csv" ] && cp "$TMP_FOLDER/$INPUT_FILE.csv" "$OUTPUT_CSV_PATH" || echo "No output CSV file found." |
| 57 | + |
| 58 | +# Clean up |
| 59 | +rm "$TMP_FOLDER/$INPUT_FILE" |
| 60 | +rm "$TMP_FOLDER/$PROCESS_SCRIPT" |
| 61 | +[ -f "$TMP_FOLDER/$INPUT_FILE.csv" ] && rm "$TMP_FOLDER/$INPUT_FILE.csv" |
| 62 | +rm -rf "$TMP_FOLDER" |
0 commit comments