Skip to content
@Enhancing-threat-intelligence-for-iot

Enhancing Threat Intelligence for Internet of Things (IoT) Systems

A full‑stack app to explore IoT devices, link them to critical infrastructure sectors, and surface known vulnerabilities with AI‑assisted insights.

Enhancing threat intelligence for internet of things (IoT) systems

A full‑stack app to explore IoT devices, link them to critical infrastructure sectors, and surface known vulnerabilities with AI‑assisted insights.

  • Browse and manage IoT devices
  • Assign devices to one or more sectors
  • Auto‑enrich devices with NIST CVE data and generate AI summaries via Ollama
  • Color‑coded risk indicators and clean UI (SvelteKit + Tailwind + DaisyUI)

🐳 Quick Start ­(Docker)

Prerequisites: Docker Desktop with Compose. For GPU, install appropriate drivers and Docker toolkit.

Using the frontend-and-backend-images.tar you can skip building the images local entirely:

First run this command to load the images:

docker load -i frontend-and-backend-images.tar

you can download the frontend-and-backend-images.tar file here

Then run one of the following commands

Go to Docker main to get all the following Compose.yaml files

  • CPU:
    • docker compose up -d --build
  • NVIDIA GPU:
    • docker compose -f Compose.yaml -f Compose.nvidia.yaml up -d
  • AMD GPU:
    • docker compose -f Compose.yaml -f Compose.amd.yaml up -d

🛑 Stop the services

  • Remove the image and stop services

    • docker compose down -v
      Note: -v will delete the database along with recently added IoT device(s)
  • Stop Services

    • docker-compose stop

📂 Accessible links

💾 Backend/Django links:

🧑‍💻Dev Notes

🌳 Architecture

📜 Key files:

⌨️ Development (without Docker)

📱 Frontend (SvelteKit):

  • cd PolySectorMap
  • yarn install
  • yarn run dev
  • Build: yarn run build

🗃️Backend (Django):

  • cd BackEnd

  • 💻 Create venv and install:

    • python -m venv venv-backend && source venv-backend/bin/activate
    • pip install -r iot_backend/requirements.txt
  • 🐛 Turn on debug mode

    • Create an .env file with the debug flag:
      • macOS/Linux: printf 'DJANGO_DEBUG="True"\n' > iot_backend/.env
      • Windows (PowerShell): Set-Content -Path iot_backend/.env -Value 'DJANGO_DEBUG="True"'
  • 🚀 Migrate and run:

    • python iot_backend/manage.py migrate
    • python iot_backend/manage.py runserver 0.0.0.0:8080

🤖 Data enrichment and AI summaries

  • NIST CVE enrichment and AI summarization are triggered on device creation or via management utilities.
  • Batch update (management command):
    • python BackEnd/iot_backend/manage.py update_threats
  • Utility script (optional): BackEnd/update_all_device_threats.py
  • Ollama endpoints and examples (generate/chat/embed, model ops): see BackEnd/iot_backend/api.md

Pinned Loading

  1. Backend Backend Public

    Python

  2. Frontend Frontend Public

    Svelte

  3. Docker-Main Docker-Main Public

    Clone this repo as well as the frontend and backend repos within this directory

  4. Report Report Public

    Report

Repositories

Showing 7 of 7 repositories

Top languages

Loading…

Most used topics

Loading…