70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
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 content %}
|
|
<ul id="list" data-role="listview" style="list-style-type:disc;">
|
|
{% for handshake in handshakes %}
|
|
<li class="file">
|
|
<a href="/plugins/handshakes-dl/{{handshake}}">{{handshake}}</a>
|
|
</li>
|
|
{% endfor %}
|
|
</ul>
|
|
{% endblock %}
|
|
"""
|
|
|
|
class HandshakesDL(plugins.Plugin):
|
|
__author__ = 'me@sayakb.com'
|
|
__version__ = '0.1.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_ready(self, agent):
|
|
self.config = agent.config()
|
|
self.ready = True
|
|
|
|
def on_internet_available(self, agent):
|
|
self.config = agent.config()
|
|
self.ready = True
|
|
|
|
def on_webhook(self, path, request):
|
|
if not self.ready:
|
|
return "Plugin not ready"
|
|
|
|
if path == "/" or not path:
|
|
handshakes = glob.glob(os.path.join(self.config['bettercap']['handshakes'], "*.pcap"))
|
|
handshakes = [os.path.basename(path)[:-5] for path in handshakes]
|
|
return render_template_string(TEMPLATE,
|
|
title="Handshakes | " + pwnagotchi.name(),
|
|
handshakes=handshakes)
|
|
|
|
else:
|
|
dir = self.config['bettercap']['handshakes']
|
|
try:
|
|
logging.info(f"[HandshakesDL] serving {dir}/{path}.pcap")
|
|
return send_from_directory(directory=dir, filename=path+'.pcap', as_attachment=True)
|
|
except FileNotFoundError:
|
|
abort(404)
|