pwnv 0.4.4

CLI tool for CTF management

pwnv: A CTF Workspace Management Tool 🛠️

pwnv is a Command-Line Interface (CLI) utility designed to optimize and organize CTF workflows. It facilitates challenge management, environment setup, and integration with remote CTF platforms, providing a structured approach to CTF participation.

---

🎯 Overview

pwnv addresses common challenges in CTF participation, such as disorganized challenge files and manual platform interaction. It provides a standardized framework to structure CTF events, automate setup procedures, and interface with platforms like CTFd, enabling participants to concentrate on problem-solving and enhancing overall efficiency.

---

✨ Key Features

Feature	Description
🗂️ Structured Workspace	Establishes a consistent and organized directory structure for CTFs and their associated challenges.
📦 Virtual Environments	Manages isolated Python virtual environments for CTF workspaces, utilizing uv for rapid setup.
🔄 Remote Synchronization	Enables fetching challenges, descriptions, and attachments from CTFd instances using the ctfbridge library.
🚀 Remote Flag Submission	Allows direct submission of flags to remote CTF platforms via the command line.
🔌 Plugin Architecture	Supports custom Python plugins for automating challenge setup based on predefined categories (e.g., pwn, web).
🏷️ Challenge Tagging	Provides functionality to tag solved challenges with relevant keywords for efficient searching and retrieval.
✨ Interactive Interface	Employs fuzzy finders and interactive prompts for intuitive navigation and user input.

---

🏗️ Installation Guide

Prerequisites

• Python 3.12 or higher.
• uv: Ensure uv is installed and accessible via the system PATH.

Option 1: Via pip

pip install pwnv

Option 2: From Source (Development)

git clone https://github.com/CarixoHD/pwnv
cd pwnv
pip install --editable .

---

🚀 Quickstart Guide

1. Initialize the workspace:

   pwnv init --ctfs-folder ~/CTFs
   source ~/CTFs/.pwnvenv/bin/activate
   

2. Add a CTF event:

   # Add a local event
   pwnv ctf add ExampleCTF_Local
   
   # Add a remote event (prompts for URL and credentials)
   pwnv ctf add ExampleCTF_Remote
   

3. Add a challenge:

   pwnv challenge add RopMaster # Select category when prompted
   

4. Navigate to the challenge directory and begin work:

   cd ~/CTFs/ExampleCTF_Local/pwn/RopMaster/
   # Begin solving the challenge.
   

5. Mark the challenge as solved:

   pwnv solve --flag "FLAG{example_flag}"
   # Enter tags when prompted (e.g., "buffer-overflow, ROP").
   

---

🧠 Core Concepts

Workspace Organization

pwnv enforces a hierarchical directory structure. A primary CTF folder contains individual CTF event directories, which in turn contain challenges categorized by type:

~/CTFs/
├── .pwnvenv/
├── ExampleCTF_Local/
│   ├── pwn/
│   │   └── RopMaster/
│   │       └── solve.py
│   └── web/
│       └── WebChallenge/
└── ExampleCTF_Remote/
    ├── .env
    ├── .session
    ├── crypto/
    │   └── CryptoChallenge/
    └── ...

Remote Platform Integration

Leveraging ctfbridge, pwnv interacts with remote CTF platforms to:

• Retrieve challenge data (descriptions, values, categories, tags).
• Download associated attachments.
• Handle authentication via credentials or API tokens.
• Maintain session state.
• Submit flags programmatically via pwnv solve.

Plugin System

The plugin system allows for the execution of category-specific Python scripts during challenge creation, automating setup tasks like generating boilerplate solver scripts or setting up tools.

---

🧩 Plugin Architecture

pwnv features an extensible plugin system that allows users to define custom actions executed automatically during challenge creation (pwnv challenge add). This enables the automation of boilerplate setup, tool integration, and other category-specific tasks.

Plugin Location

• Plugin Scripts: Reside within the plugins folder in your pwnv configuration directory (typically ~/.config/pwnv/plugins/). Each .py file represents a potential plugin.
• Template Files: Associated template files (e.g., solve.py skeletons) are stored in the templates folder, organized by category (e.g., ~/.config/pwnv/templates/pwn/).

Plugin Structure

A pwnv plugin is a Python class that inherits from pwnv.plugins.ChallengePlugin. It must be decorated with @register_plugin to be discoverable.

Key components include:

• @register_plugin: Decorator that makes the plugin available to pwnv.
• category(self) -> Category:: Abstract method that must return the pwnv.models.challenge.Category for which this plugin should be considered.
• logic(self, challenge: Challenge) -> None:: Abstract method containing the core custom logic to be executed.
• templates_to_copy: Dict[str, str | None]: A class attribute specifying which files from the templates directory should be copied into the new challenge directory.

Example Plugin (pwn_plugin.py)

from pwnv.core import register_plugin
from pwnv.models.challenge import Category
from pwnv.plugins.plugin import ChallengePlugin
from pwnv.models import Challenge
from pwnv.utils.ui import info

@register_plugin
class BasicPwnPlugin(ChallengePlugin):
    # Copy 'solve.py' and 'gdbinit' from templates/pwn/ to the challenge dir.
    templates_to_copy = {
        "solve.py": None,
        "gdbinit": "gdbinit_rop" # save as gdbinit_rop
    }

    def category(self) -> Category:
        return Category.pwn

    def logic(self, challenge: Challenge) -> None:
        # Custom logic for pwn challenges
        info(f"Set up basic pwn environment for {challenge.name}")

---

⌨️ Command Reference

The following table summarizes the available commands. For detailed usage, append --help to any command or subcommand.

Command	Description
pwnv init	Initializes the pwnv environment and workspace.
pwnv reset	Removes all pwnv configurations and CTF data (exercise caution).
pwnv ctf add <name>	Adds a new CTF event (local or remote).
pwnv ctf remove	Deletes a CTF event and its challenges.
pwnv ctf info	Displays metadata for a selected CTF.
pwnv ctf start	Sets a CTF's status to 'running'.
pwnv ctf stop	Sets a CTF's status to 'stopped'.
pwnv challenge add <name>	Adds a new challenge, triggering relevant plugins.
pwnv challenge remove	Deletes a specific challenge.
pwnv challenge info	Displays metadata for a selected challenge.
pwnv challenge filter	Lists solved challenges based on specified tags.
pwnv solve	Marks a challenge as solved and handles flag submission/tagging.
pwnv plugin add <name>	Creates a new plugin and its associated template.
pwnv plugin remove	Deletes an existing plugin file.
pwnv plugin info	Displays information about registered plugins.
pwnv plugin select	Assigns a specific plugin to a challenge category.

---

🤝 Contributing

Contributions to pwnv are welcome. Please refer to the GitHub repository to report issues, propose features, or submit pull requests.

---

📄 License

pwnv is distributed under the MIT License. See the LICENSE file for further details.

MIT © Shayan Alinejad