handshakes-dl-hashie.py

import logging
import json
import os
import glob

import pwnagotchi
import pwnagotchi.plugins as plugins

from flask import abort
from flask import send_from_directory
from flask import render_template_string

TEMPLATE = """
{% extends "base.html" %}
{% set active_page = "handshakes" %}
{% block title %}
    {{ title }}
{% endblock %}
{% block styles %}
    {{ super() }}
    <style>
        #filter {
            width: 100%;
            font-size: 16px;
            padding: 12px 20px 12px 40px;
            border: 1px solid #ddd;
            margin-bottom: 12px;
        }
    </style>
{% endblock %}
{% block script %}
    var shakeList = document.getElementById('list');
    var filter = document.getElementById('filter');
    var filterVal = filter.value.toUpperCase();
    filter.onkeyup = function() {
        document.body.style.cursor = 'progress';
        var table, tr, tds, td, i, txtValue;
        filterVal = filter.value.toUpperCase();
        li = shakeList.getElementsByTagName("li");
        for (i = 0; i < li.length; i++) {
            txtValue = li[i].textContent || li[i].innerText;
            if (txtValue.toUpperCase().indexOf(filterVal) > -1) {
                li[i].style.display = "list-item";
            } else {
                li[i].style.display = "none";
            }
        }
        document.body.style.cursor = 'default';
    }
{% endblock %}
{% block content %}
    <input type="text" id="filter" placeholder="Search for ..." title="Type in a filter">
    <ul id="list" data-role="listview" style="list-style-type:disc;">
        {% for handshake in handshakes %}
            {% for ext in handshake.ext %}
                <li class="file">
                    <a href="/plugins/handshakes-dl-hashie/{{handshake.name}}{{ext}}">{{handshake.name}}{{ext}}</a>
                </li>
            {% endfor %}
        {% endfor %}
    </ul>
{% endblock %}
"""

class handshakes:  
    def __init__(self, name, path, ext):  
        self.name = name  
        self.path = path  
        self.ext = ext 

class HandshakesDL(plugins.Plugin):
    __author__ = 'me@sayakb.com'
    __version__ = '1.0.0'
    __license__ = 'GPL3'
    __description__ = 'Download handshake captures from web-ui.'

    def __init__(self):
        self.ready = False

    def on_loaded(self):
        logging.info("[HandshakesDL] plugin loaded")

    def on_config_changed(self, config):
        self.config = config
        self.ready = True

    def on_webhook(self, path, request):
        if not self.ready:
            return "Plugin not ready"

        if path == "/" or not path:
            pcapfiles = glob.glob(os.path.join(self.config['bettercap']['handshakes'], "*.pcap"))
            
            data = []
            for path in pcapfiles:
                name = os.path.basename(path)[:-5]
                fullpathNoExt = path[:-5]
                possibleExt = ['.2500', '.16800', '.22000']
                foundExt = ['.pcap']
                for ext in possibleExt: 
                    if os.path.isfile(fullpathNoExt +  ext):
                        foundExt.append(ext)
                data.append(handshakes(name, fullpathNoExt, foundExt)) 
            return render_template_string(TEMPLATE,
                                    title="Handshakes | " + pwnagotchi.name(),
                                    handshakes=data)
        else:
            dir = self.config['bettercap']['handshakes']
            try:
                logging.info(f"[HandshakesDL] serving {dir}/{path}")
                return send_from_directory(directory=dir, filename=path, as_attachment=True)
            except FileNotFoundError:
                abort(404)