mirror of
https://github.com/Adaptix-Framework/AdaptixC2
synced 2026-06-08 10:20:39 +00:00
Ralf
95 lines
2.6 KiB
C++
95 lines
2.6 KiB
C++
#ifndef AXSCRIPTENGINE_H
|
|
#define AXSCRIPTENGINE_H
|
|
|
|
#include <QAction>
|
|
#include <QJSValue>
|
|
#include <QJSEngine>
|
|
#include <QString>
|
|
#include <QObject>
|
|
#include <QWidget>
|
|
|
|
class BridgeApp;
|
|
class BridgeForm;
|
|
class BridgeEvent;
|
|
class BridgeMenu;
|
|
class AbstractAxMenuItem;
|
|
class AxScriptManager;
|
|
|
|
struct AxEvent {
|
|
QJSValue handler;
|
|
QString event_id;
|
|
QSet<QString> agents;
|
|
QSet<QString> listenerts;
|
|
QSet<int> os;
|
|
QJSEngine* jsEngine;
|
|
};
|
|
|
|
struct AxMenuItem {
|
|
AbstractAxMenuItem* menu;
|
|
QSet<QString> agents;
|
|
QSet<QString> listenerts;
|
|
QSet<int> os;
|
|
};
|
|
|
|
struct ScriptContext {
|
|
QString name;
|
|
QJSValue scriptObject;
|
|
QList<QObject*> objects;
|
|
QList<QAction*> actions;
|
|
|
|
QList<AxEvent> eventFileBroserDisks;
|
|
QList<AxEvent> eventFileBroserList;
|
|
QList<AxEvent> eventFileBroserUpload;
|
|
QList<AxEvent> eventProcessBrowserList;
|
|
|
|
QList<AxMenuItem> menuSessionMain;
|
|
QList<AxMenuItem> menuSessionAgent;
|
|
QList<AxMenuItem> menuSessionBrowser;
|
|
QList<AxMenuItem> menuSessionAccess;
|
|
QList<AxMenuItem> menuFileBrowser;
|
|
QList<AxMenuItem> menuDownloadRunning;
|
|
QList<AxMenuItem> menuDownloadFinished;
|
|
QList<AxMenuItem> menuTasks;
|
|
QList<AxMenuItem> menuTasksJob;
|
|
};
|
|
|
|
class AxScriptEngine : public QObject {
|
|
Q_OBJECT
|
|
AxScriptManager* scriptManager;
|
|
|
|
std::unique_ptr<QJSEngine> jsEngine;
|
|
std::unique_ptr<BridgeApp> bridgeApp;
|
|
std::unique_ptr<BridgeForm> bridgeForm;
|
|
std::unique_ptr<BridgeEvent> bridgeEvent;
|
|
std::unique_ptr<BridgeMenu> bridgeMenu;
|
|
|
|
public:
|
|
ScriptContext context;
|
|
|
|
explicit AxScriptEngine(AxScriptManager* script_manager, const QString &name = "", QObject *parent = nullptr);
|
|
~AxScriptEngine() override;
|
|
|
|
QJSEngine* engine() const;
|
|
BridgeApp* app() const;
|
|
BridgeForm* form() const;
|
|
BridgeEvent* event() const;
|
|
BridgeMenu* menu() const;
|
|
|
|
AxScriptManager* manager() const;
|
|
|
|
void registerObject(QObject* obj);
|
|
void registerAction(QAction* action);
|
|
void registerEvent(const QString &type, const QJSValue &handler, const QSet<QString> &list_agents, const QSet<QString> &list_os, const QSet<QString> &list_listeners, const QString &id);
|
|
void removeEvent(const QString &id);
|
|
void registerMenu(const QString &type, AbstractAxMenuItem* menu, const QSet<QString> &list_agents, const QSet<QString> &list_os, const QSet<QString> &list_listeners);
|
|
bool execute(const QString &code);
|
|
|
|
QList<AxEvent> getEvents(const QString &type);
|
|
QList<AxMenuItem> getMenuItems(const QString &type);
|
|
|
|
public slots:
|
|
void engineError(const QString &message);
|
|
};
|
|
|
|
#endif
|