2017-03-29 10:18:37 +00:00
|
|
|
#ifndef QRCORE_H
|
|
|
|
#define QRCORE_H
|
|
|
|
|
|
|
|
#include <QMap>
|
|
|
|
#include <QDebug>
|
|
|
|
#include <QObject>
|
|
|
|
#include <QStringList>
|
|
|
|
#include <QMessageBox>
|
2017-04-18 08:33:35 +00:00
|
|
|
#include <QJsonDocument>
|
2017-03-30 03:07:34 +00:00
|
|
|
|
|
|
|
//Workaround for compile errors on Windows
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include <r2hacks.h>
|
|
|
|
#endif //_WIN32
|
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
#include "r_core.h"
|
|
|
|
|
2017-03-30 03:07:34 +00:00
|
|
|
//Workaround for compile errors on Windows.
|
|
|
|
#ifdef _WIN32
|
|
|
|
#undef min
|
|
|
|
#endif //_WIN32
|
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
#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 17:12:36 +00:00
|
|
|
{
|
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 &&);
|
2017-04-09 17:12:36 +00:00
|
|
|
~RCoreLocked();
|
2017-04-09 19:55:06 +00:00
|
|
|
operator RCore *() const;
|
|
|
|
RCore *operator->() const;
|
2017-04-09 17:12:36 +00:00
|
|
|
};
|
|
|
|
|
2017-04-09 17:09:35 +00:00
|
|
|
#define QNOTUSED(x) do { (void)(x); } while ( 0 );
|
2017-03-29 10:18:37 +00:00
|
|
|
|
2017-04-28 13:09:40 +00:00
|
|
|
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;
|
|
|
|
};
|
|
|
|
|
2017-05-19 07:45:26 +00:00
|
|
|
struct ExportDescription
|
|
|
|
{
|
|
|
|
RVA vaddr;
|
|
|
|
RVA paddr;
|
|
|
|
RVA size;
|
|
|
|
QString type;
|
|
|
|
QString name;
|
|
|
|
QString flag_name;
|
|
|
|
};
|
|
|
|
|
2017-04-28 13:09:40 +00:00
|
|
|
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;
|
|
|
|
};
|
|
|
|
|
2017-05-03 09:09:57 +00:00
|
|
|
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;
|
|
|
|
};
|
|
|
|
|
2017-04-28 13:09:40 +00:00
|
|
|
Q_DECLARE_METATYPE(FunctionDescription)
|
|
|
|
Q_DECLARE_METATYPE(ImportDescription)
|
2017-05-19 07:45:26 +00:00
|
|
|
Q_DECLARE_METATYPE(ExportDescription)
|
2017-04-28 13:09:40 +00:00
|
|
|
Q_DECLARE_METATYPE(SymbolDescription)
|
|
|
|
Q_DECLARE_METATYPE(CommentDescription)
|
|
|
|
Q_DECLARE_METATYPE(RelocDescription)
|
|
|
|
Q_DECLARE_METATYPE(StringDescription)
|
2017-05-03 09:09:57 +00:00
|
|
|
Q_DECLARE_METATYPE(FlagspaceDescription)
|
|
|
|
Q_DECLARE_METATYPE(FlagDescription)
|
2017-06-07 15:48:36 +00:00
|
|
|
Q_DECLARE_METATYPE(XRefDescription)
|
2017-04-28 13:09:40 +00:00
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
class QRCore : public QObject
|
|
|
|
{
|
|
|
|
Q_OBJECT
|
2017-04-28 13:09:40 +00:00
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
public:
|
|
|
|
QString projectPath;
|
2017-04-28 13:09:40 +00:00
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
explicit QRCore(QObject *parent = 0);
|
|
|
|
~QRCore();
|
2017-04-28 13:09:40 +00:00
|
|
|
|
|
|
|
RVA getOffset() const { return core_->offset; }
|
2017-03-29 10:18:37 +00:00
|
|
|
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);
|
2017-04-18 08:33:35 +00:00
|
|
|
QJsonDocument cmdj(const QString &str);
|
2017-03-29 10:18:37 +00:00
|
|
|
void renameFunction(QString prev_name, QString new_name);
|
2017-04-28 13:09:40 +00:00
|
|
|
void setComment(RVA addr, QString cmt);
|
2017-03-29 10:18:37 +00:00
|
|
|
void setComment(QString addr, QString cmt);
|
|
|
|
void delComment(ut64 addr);
|
2017-04-09 19:55:06 +00:00
|
|
|
QMap<QString, QList<QList<QString>>> getNestedComments();
|
2017-03-29 10:18:37 +00:00
|
|
|
void setOptions(QString key);
|
2017-05-03 09:09:57 +00:00
|
|
|
bool loadFile(QString path, uint64_t loadaddr = 0LL, uint64_t mapaddr = 0LL, bool rw = false, int va = 0, int idx = 0, bool loadbin = false);
|
2017-03-29 10:18:37 +00:00
|
|
|
bool tryFile(QString path, bool rw);
|
|
|
|
void analyze(int level);
|
|
|
|
void seek(QString addr);
|
2017-04-28 13:09:40 +00:00
|
|
|
void seek(ut64 offset);
|
2017-03-29 10:18:37 +00:00
|
|
|
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);
|
2017-03-29 10:18:37 +00:00
|
|
|
int config(const QString &k, int v);
|
|
|
|
QString assemble(const QString &code);
|
2017-04-12 20:44:05 +00:00
|
|
|
QString disassemble(const QString &hex);
|
2017-06-07 15:48:36 +00:00
|
|
|
QString disassembleSingleInstruction(RVA addr);
|
2017-03-29 10:18:37 +00:00
|
|
|
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);
|
2017-03-29 10:18:37 +00:00
|
|
|
QString cmdFunctionAt(QString addr);
|
2017-04-28 13:09:40 +00:00
|
|
|
QString cmdFunctionAt(RVA addr);
|
2017-03-29 10:18:37 +00:00
|
|
|
/* 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;
|
2017-03-30 16:47:50 +00:00
|
|
|
void setSettings();
|
2017-03-29 10:18:37 +00:00
|
|
|
|
2017-04-28 13:09:40 +00:00
|
|
|
QList<QString> getList(const QString &type, const QString &subtype = "");
|
|
|
|
|
|
|
|
QList<RVA> getSeekHistory();
|
2017-05-03 09:09:57 +00:00
|
|
|
|
|
|
|
QStringList getAsmPluginNames();
|
|
|
|
QStringList getAnalPluginNames();
|
|
|
|
|
2017-05-13 18:09:36 +00:00
|
|
|
QStringList getProjectNames();
|
|
|
|
|
2017-04-28 13:09:40 +00:00
|
|
|
QList<FunctionDescription> getAllFunctions();
|
|
|
|
QList<ImportDescription> getAllImports();
|
2017-05-19 07:45:26 +00:00
|
|
|
QList<ExportDescription> getAllExports();
|
2017-04-28 13:09:40 +00:00
|
|
|
QList<SymbolDescription> getAllSymbols();
|
2017-04-28 13:38:01 +00:00
|
|
|
QList<CommentDescription> getAllComments(const QString &filterType);
|
2017-04-28 13:09:40 +00:00
|
|
|
QList<RelocDescription> getAllRelocs();
|
|
|
|
QList<StringDescription> getAllStrings();
|
2017-05-03 09:09:57 +00:00
|
|
|
QList<FlagspaceDescription> getAllFlagspaces();
|
|
|
|
QList<FlagDescription> getAllFlags(QString flagspace = NULL);
|
|
|
|
QList<SectionDescription> getAllSections();
|
2017-04-28 13:09:40 +00:00
|
|
|
|
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);
|
|
|
|
|
2017-04-09 17:12:36 +00:00
|
|
|
RCoreLocked core() const;
|
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
/* fields */
|
2017-04-09 17:12:36 +00:00
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
Sdb *db;
|
2017-04-28 13:09:40 +00:00
|
|
|
|
2017-03-29 10:18:37 +00:00
|
|
|
signals:
|
2017-04-28 13:09:40 +00:00
|
|
|
void functionRenamed(QString prev_name, QString new_name);
|
2017-03-29 10:18:37 +00:00
|
|
|
|
|
|
|
public slots:
|
|
|
|
|
|
|
|
private:
|
|
|
|
QString default_arch;
|
|
|
|
QString default_cpu;
|
|
|
|
int default_bits;
|
2017-04-09 17:12:36 +00:00
|
|
|
|
|
|
|
RCore *core_;
|
2017-03-29 10:18:37 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
#endif // QRCORE_H
|