First commit

server/frontend/app/blueprints/analysis.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import re
+import os
+import json
+import sys
+from flask import Blueprint, jsonify
+from app.classes.analysis import Analysis
+import subprocess as sp
+import json
+
+analysis_bp = Blueprint("analysis", __name__)
+
+
+@analysis_bp.route("/start/<token>", methods=["GET"])
+def api_start_analysis(token):
+    """ 
+        Start an analysis
+    """
+    return jsonify(Analysis(token).start())
+
+
+@analysis_bp.route("/report/<token>", methods=["GET"])
+def api_report_analysis(token):
+    """ 
+        Get the report of an analysis
+    """
+    return jsonify(Analysis(token).get_report())

server/frontend/app/blueprints/capture.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from flask import jsonify, Blueprint
+from app.classes.capture import Capture
+
+capture = Capture()
+capture_bp = Blueprint("capture", __name__)
+
+
+@capture_bp.route("/start", methods=["GET"])
+def api_capture_start():
+    """ Start the capture """
+    return jsonify(capture.start_capture())
+
+
+@capture_bp.route("/stop", methods=["GET"])
+def api_capture_stop():
+    """ Stop the capture """
+    return jsonify(capture.stop_capture())
+
+
+@capture_bp.route("/stats", methods=["GET"])
+def api_capture_stats():
+    """ Stop the capture """
+    return jsonify(capture.get_capture_stats())

server/frontend/app/blueprints/device.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from flask import jsonify, Blueprint
+from app.classes.device import Device
+
+device_bp = Blueprint("device", __name__)
+
+
+@device_bp.route("/get/<token>", methods=["GET"])
+def api_device_get(token):
+    """ Get device assets """
+    return jsonify(Device(token).get())

server/frontend/app/blueprints/misc.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import subprocess as sp
+from flask import Blueprint, jsonify
+from app.utils import read_config
+
+misc_bp = Blueprint("misc", __name__)
+
+
+@misc_bp.route("/reboot", methods=["GET"])
+def api_reboot():
+    """ 
+        Reboot the device 
+    """
+    sp.Popen("reboot", shell=True)
+    return jsonify({"mesage": "Let's reboot."})
+
+
+@misc_bp.route("/config", methods=["GET"])
+def get_config():
+    """ 
+        Get configuration keys relative to the GUI 
+    """
+    return jsonify({
+        "virtual_keyboard": read_config(("frontend", "virtual_keyboard")),
+        "hide_mouse": read_config(("frontend", "hide_mouse")),
+        "download_links": read_config(("frontend", "download_links")),
+        "sparklines": read_config(("frontend", "sparklines")),
+    })

server/frontend/app/blueprints/network.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from flask import Blueprint, jsonify, request
+from app.classes.network import Network
+
+network = Network()
+network_bp = Blueprint("network", __name__)
+
+
+@network_bp.route("/status", methods=["GET"])
+def api_network_status():
+    """ Get the network status of eth0, wlan0 """
+    return jsonify(network.check_status())
+
+
+@network_bp.route("/wifi/list", methods=["GET"])
+def api_get_wifi_list():
+    """ List available WIFI networks """
+    return jsonify(network.wifi_list_networks())
+
+
+@network_bp.route("/wifi/setup", methods=["POST", "OPTIONS"])
+def api_set_wifi():
+    """ Set an access point and a password """
+    if request.method == "POST":
+        data = request.get_json()
+        res = network.wifi_setup(data["ssid"], data["password"])
+        return jsonify(res)
+    else:
+        return ""
+
+
+@network_bp.route("/wifi/connect", methods=["GET"])
+def api_connect_wifi():
+    """ Connect to the specified wifi network """
+    res = network.wifi_connect()
+    return jsonify(res)
+
+
+@network_bp.route("/ap/start", methods=["GET"])
+def api_start_ap():
+    """ Start an access point """
+    return jsonify(network.start_ap())
+
+
+@network_bp.route("/ap/stop", methods=["GET"])
+def api_stop_ap():
+    """ Generate an access point """
+    return jsonify(network.stop_hostapd())

server/frontend/app/blueprints/save.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from flask import Blueprint, jsonify, request
+from app.classes.save import Save
+from app.classes.device import Device
+
+save = Save()
+save_bp = Blueprint("save", __name__)
+
+
+@save_bp.route("/usb-check", methods=["GET"])
+def api_usb_list():
+    """ List connected usb devices """
+    return save.usb_check()
+
+
+@save_bp.route("/save-capture/<token>/<method>", methods=["GET"])
+def api_save_capture(token, method):
+    """ Save the capture on the USB or for download """
+    return save.save_capture(token, method)