cutter/src/qrcore.h

262 lines
6.1 KiB
C
Raw Normal View History

#ifndef QRCORE_H
#define QRCORE_H
#include <QMap>
#include <QDebug>
#include <QObject>
#include <QStringList>
#include <QMessageBox>
#include <QJsonDocument>
2017-03-30 03:07:34 +00:00
//Workaround for compile errors on Windows
#ifdef _WIN32
#include <r2hacks.h>
#endif //_WIN32
#include "r_core.h"
2017-03-30 03:07:34 +00:00
//Workaround for compile errors on Windows.
#ifdef _WIN32
#undef min
#endif //_WIN32
#define HAVE_LATEST_LIBR2 false
#define QRListForeach(list, it, type, x) \
if (list) for (it = list->head; it && ((x=(type*)it->data)); it = it->n)
#define __alert(x) QMessageBox::question (this, "Alert", QString(x), QMessageBox::Ok)
#define __question(x) (QMessageBox::Yes==QMessageBox::question (this, "Alert", QString(x), QMessageBox::Yes| QMessageBox::No))
2017-04-09 18:42:45 +00:00
class RCoreLocked
{
2017-04-09 19:55:06 +00:00
RCore *core;
2017-04-09 18:42:45 +00:00
public:
2017-04-09 19:55:06 +00:00
explicit RCoreLocked(RCore *core);
RCoreLocked(const RCoreLocked &) = delete;
RCoreLocked &operator=(const RCoreLocked &) = delete;
RCoreLocked(RCoreLocked &&);
~RCoreLocked();
2017-04-09 19:55:06 +00:00
operator RCore *() const;
RCore *operator->() const;
};
#define QNOTUSED(x) do { (void)(x); } while ( 0 );
typedef ut64 RVA;
inline QString RAddressString(RVA addr)
{
return QString::asprintf("%#010llx", addr);
}
inline QString RSizeString(RVA size)
{
return QString::asprintf("%lld", size);
}
struct FunctionDescription
{
RVA offset;
RVA size;
QString name;
bool contains(RVA addr) const { return addr >= offset && addr < offset + size; }
};
struct ImportDescription
{
RVA plt;
int ordinal;
QString bind;
QString type;
QString name;
};
struct ExportDescription
{
RVA vaddr;
RVA paddr;
RVA size;
QString type;
QString name;
QString flag_name;
};
struct SymbolDescription
{
RVA vaddr;
QString bind;
QString type;
QString name;
};
struct CommentDescription
{
RVA offset;
QString name;
};
struct RelocDescription
{
RVA vaddr;
RVA paddr;
QString type;
QString name;
};
struct StringDescription
{
RVA vaddr;
QString string;
};
struct FlagspaceDescription
{
QString name;
};
struct FlagDescription
{
RVA offset;
RVA size;
QString name;
};
struct SectionDescription
{
RVA vaddr;
RVA paddr;
RVA size;
RVA vsize;
QString name;
QString flags;
};
2017-06-07 10:56:55 +00:00
struct XRefDescription
{
RVA from;
RVA to;
QString type;
};
Q_DECLARE_METATYPE(FunctionDescription)
Q_DECLARE_METATYPE(ImportDescription)
Q_DECLARE_METATYPE(ExportDescription)
Q_DECLARE_METATYPE(SymbolDescription)
Q_DECLARE_METATYPE(CommentDescription)
Q_DECLARE_METATYPE(RelocDescription)
Q_DECLARE_METATYPE(StringDescription)
Q_DECLARE_METATYPE(FlagspaceDescription)
Q_DECLARE_METATYPE(FlagDescription)
2017-06-07 15:48:36 +00:00
Q_DECLARE_METATYPE(XRefDescription)
class QRCore : public QObject
{
Q_OBJECT
public:
QString projectPath;
explicit QRCore(QObject *parent = 0);
~QRCore();
RVA getOffset() const { return core_->offset; }
int getCycloComplex(ut64 addr);
int getFcnSize(ut64 addr);
int fcnCyclomaticComplexity(ut64 addr);
int fcnBasicBlockCount(ut64 addr);
int fcnEndBbs(QString addr);
QString cmd(const QString &str);
QJsonDocument cmdj(const QString &str);
void renameFunction(QString prev_name, QString new_name);
void setComment(RVA addr, QString cmt);
void setComment(QString addr, QString cmt);
void delComment(ut64 addr);
2017-04-09 19:55:06 +00:00
QMap<QString, QList<QList<QString>>> getNestedComments();
void setOptions(QString key);
bool loadFile(QString path, uint64_t loadaddr = 0LL, uint64_t mapaddr = 0LL, bool rw = false, int va = 0, int idx = 0, bool loadbin = false);
bool tryFile(QString path, bool rw);
void analyze(int level);
void seek(QString addr);
void seek(ut64 offset);
ut64 math(const QString &expr);
2017-04-09 19:55:06 +00:00
QString itoa(ut64 num, int rdx = 16);
QString config(const QString &k, const QString &v = NULL);
int config(const QString &k, int v);
QString assemble(const QString &code);
QString disassemble(const QString &hex);
2017-06-07 15:48:36 +00:00
QString disassembleSingleInstruction(RVA addr);
void setDefaultCPU();
2017-04-13 14:24:23 +00:00
void setCPU(QString arch, QString cpu, int bits, bool temporary = false);
2017-04-09 19:55:06 +00:00
RAnalFunction *functionAt(ut64 addr);
QString cmdFunctionAt(QString addr);
QString cmdFunctionAt(RVA addr);
/* sdb */
QList<QString> sdbList(QString path);
QList<QString> sdbListKeys(QString path);
QString sdbGet(QString path, QString key);
bool sdbSet(QString path, QString key, QString val);
int get_size();
ulong get_baddr();
QList<QList<QString>> get_exec_sections();
QString getOffsetInfo(QString addr);
QString getOffsetJump(QString addr);
QString getDecompiledCode(QString addr);
QString getFileInfo();
QStringList getStats();
QString getSimpleGraph(QString function);
QString binStart;
QString binEnd;
void getOpcodes();
QList<QString> opcodes;
QList<QString> regs;
void setSettings();
QList<QString> getList(const QString &type, const QString &subtype = "");
QList<RVA> getSeekHistory();
QStringList getAsmPluginNames();
QStringList getAnalPluginNames();
QStringList getProjectNames();
QList<FunctionDescription> getAllFunctions();
QList<ImportDescription> getAllImports();
QList<ExportDescription> getAllExports();
QList<SymbolDescription> getAllSymbols();
QList<CommentDescription> getAllComments(const QString &filterType);
QList<RelocDescription> getAllRelocs();
QList<StringDescription> getAllStrings();
QList<FlagspaceDescription> getAllFlagspaces();
QList<FlagDescription> getAllFlags(QString flagspace = NULL);
QList<SectionDescription> getAllSections();
2017-06-07 10:56:55 +00:00
QList<QString> getFunctionXrefs(ut64 addr);
QList<QString> getFunctionRefs(ut64 addr, char type);
QList<XRefDescription> getXRefs(RVA addr, bool to, const QString &filterType = QString::null);
RCoreLocked core() const;
/* fields */
Sdb *db;
signals:
void functionRenamed(QString prev_name, QString new_name);
public slots:
private:
QString default_arch;
QString default_cpu;
int default_bits;
RCore *core_;
};
#endif // QRCORE_H