mirror of
https://github.com/Adaptix-Framework/AdaptixC2
synced 2026-06-08 10:20:39 +00:00
Ralf
99 lines
3.1 KiB
C++
99 lines
3.1 KiB
C++
#ifndef AXSCRIPTMANAGER_H
|
|
#define AXSCRIPTMANAGER_H
|
|
|
|
#include <QObject>
|
|
#include <QMenu>
|
|
#include <QJSValue>
|
|
#include <Agent/Commander.h>
|
|
#include <main.h>
|
|
|
|
struct ExtensionFile;
|
|
struct AxMenuItem;
|
|
struct AxEvent;
|
|
class AxScriptEngine;
|
|
class AdaptixWidget;
|
|
class Agent;
|
|
|
|
struct DataMenuFileBrowser {
|
|
QString agentId;
|
|
QString path;
|
|
QString name;
|
|
QString type;
|
|
};
|
|
|
|
struct DataMenuProcessBrowser {
|
|
QString agentId;
|
|
QString pid;
|
|
QString ppid;
|
|
QString arch;
|
|
QString session_id;
|
|
QString context;
|
|
QString process;
|
|
};
|
|
|
|
struct DataMenuDownload {
|
|
QString agentId;
|
|
QString fileId;
|
|
QString path;
|
|
QString state;
|
|
};
|
|
|
|
class AxScriptManager : public QObject {
|
|
Q_OBJECT
|
|
AdaptixWidget* adaptixWidget = nullptr;
|
|
AxScriptEngine* mainScript = nullptr;
|
|
QMap<QString, AxScriptEngine*> scripts;
|
|
QMap<QString, AxScriptEngine*> listeners_scripts;
|
|
QMap<QString, AxScriptEngine*> agents_scripts;
|
|
|
|
public:
|
|
AxScriptManager(AdaptixWidget* main_widget, QObject *parent = nullptr);
|
|
~AxScriptManager() override;
|
|
|
|
QJSEngine* MainScriptEngine();
|
|
void ResetMain();
|
|
void Clear();
|
|
|
|
AdaptixWidget* GetAdaptix() const;
|
|
QMap<QString, Agent*> GetAgents() const;
|
|
QJSEngine* GetEngine(const QString &name);
|
|
|
|
QStringList ListenerScriptList();
|
|
void ListenerScriptAdd(const QString &name, const QString &ax_script);
|
|
QJSEngine* ListenerScriptEngine(const QString &name);
|
|
|
|
QStringList AgentScriptList();
|
|
void AgentScriptAdd(const QString &name, const QString &ax_script);
|
|
QJSEngine* AgentScriptEngine(const QString &name);
|
|
QJSValue AgentScriptExecute(const QString &name, const QString &code);
|
|
|
|
QStringList ScriptList();
|
|
bool ScriptAdd(ExtensionFile* ext);
|
|
void ScriptRemove(const ExtensionFile &ext);
|
|
|
|
void GlobalScriptLoad(const QString &path);
|
|
void GlobalScriptUnload(const QString &path);
|
|
|
|
void RegisterCommandsGroup(const CommandsGroup &group, const QStringList &listeners, const QStringList &agents, const QList<int> &os);
|
|
void RemoveEvent(const QString &event_id);
|
|
QList<AxMenuItem> FilterMenuItems(const QStringList &agentIds, const QString &menuType);
|
|
QList<AxEvent> FilterEvents(const QString &agentId, const QString &eventType);
|
|
|
|
int AddMenuSession(QMenu* menu, const QString &menuType, QStringList agentIds);
|
|
int AddMenuFileBrowser(QMenu* menu, QVector<DataMenuFileBrowser> files);
|
|
int AddMenuProcessBrowser(QMenu* menu, QVector<DataMenuProcessBrowser> processes);
|
|
int AddMenuDownload(QMenu* menu, const QString &menuType, QVector<DataMenuDownload> files);
|
|
int AddMenuTask(QMenu* menu, const QString &menuType, const QStringList &tasks);
|
|
|
|
public slots:
|
|
void consolePrintMessage(const QString &message);
|
|
void consolePrintError(const QString &message);
|
|
|
|
void emitFileBrowserDisks(const QString &agentId);
|
|
void emitFileBrowserList(const QString &agentId, const QString &path);
|
|
void emitFileBrowserUpload(const QString &agentId, const QString &path, const QString &localFilename);
|
|
void emitProcessBrowserList(const QString &agentId);
|
|
};
|
|
|
|
#endif
|