static PyObject *smt2lib_bvmul(PyObject *self, PyObject *args) {
PyObject *op1 = nullptr;
PyObject *op2 = nullptr;
/* Extract arguments */
PyArg_ParseTuple(args, "O|O", &op1, &op2);
if (op1 == nullptr || !PySmtAstNode_Check(op1))
return PyErr_Format(PyExc_TypeError, "bvmul(): expected a SmtAstNode as first argument");
if (op2 == nullptr || !PySmtAstNode_Check(op2))
return PyErr_Format(PyExc_TypeError, "bvmul(): expected a SmtAstNode as second argument");
return PySmtAstNode(smt2lib::bvmul(PySmtAstNode_AsSmtAstNode(op1), PySmtAstNode_AsSmtAstNode(op2)));
}
static PyObject *Triton_evaluateAST(PyObject *self, PyObject *smtAST) {
uint512 value = 0;
if (!PySmtAstNode_Check(smtAST))
return PyErr_Format(PyExc_TypeError, "evaluateAST(): expected an SmtAstNode as argument");
value = ap.evaluateAST(PySmtAstNode_AsSmtAstNode(smtAST));
return uint512ToPyLongObject(value);
}
static PyObject *Triton_getFullExpression(PyObject *self, PyObject *node)
{
smt2lib::smtAstAbstractNode *fullExpr;
if (!PySmtAstNode_Check(node))
return PyErr_Format(PyExc_TypeError, "getFullSymExpr(): expected an SmtAstNode node as argument");
fullExpr = ap.getFullExpression(PySmtAstNode_AsSmtAstNode(node));
if (fullExpr == nullptr)
return Py_None;
return PySmtAstNode(fullExpr);
}
static PyObject *smt2lib_zx(PyObject *self, PyObject *args) {
PyObject *op1;
PyObject *op2 = nullptr;
/* Extract arguments */
PyArg_ParseTuple(args, "O|O", &op1, &op2);
if (!PyLong_Check(op1) && !PyInt_Check(op1))
return PyErr_Format(PyExc_TypeError, "zx(): expected an integer as first argument");
if (op2 == nullptr || !PySmtAstNode_Check(op2))
return PyErr_Format(PyExc_TypeError, "zx(): expected a SmtAstNode as second argument");
return PySmtAstNode(smt2lib::zx(PyLong_AsLong(op2), PySmtAstNode_AsSmtAstNode(op2)));
}
static PyObject *Triton_getModel(PyObject *self, PyObject *node) {
std::list<Smodel>::iterator it;
std::list<Smodel> model;
if (!PySmtAstNode_Check(node))
return PyErr_Format(PyExc_TypeError, "getModel(): expected a SmtAstNode as argument");
/* Get model */
model = ap.getModel(PySmtAstNode_AsSmtAstNode(node));
/* Craft the model dictionary */
PyObject *modelDict = xPyDict_New();
for (it = model.begin() ; it != model.end(); it++)
PyDict_SetItemString(modelDict, it->getName().c_str(), Py_BuildValue("k", it->getValue()));
return modelDict;
}
static PyObject *smt2lib_compound(PyObject *self, PyObject *exprsList) {
std::vector<smt2lib::smtAstAbstractNode *> exprs;
if (exprsList == nullptr || !PyList_Check(exprsList))
return PyErr_Format(PyExc_TypeError, "compound(): expected a list of SmtAstNodes as first argument");
/* Check if the mems list contains only integer item and craft a std::list */
for (Py_ssize_t i = 0; i < PyList_Size(exprsList); i++){
PyObject *item = PyList_GetItem(exprsList, i);
if (!PySmtAstNode_Check(item))
return PyErr_Format(PyExc_TypeError, "compound(): Each element from the list must be a SmtAstNode");
exprs.push_back(PySmtAstNode_AsSmtAstNode(item));
}
return PySmtAstNode(smt2lib::compound(exprs));
}
static PyObject *smt2lib_extract(PyObject *self, PyObject *args) {
PyObject *op1 = nullptr;
PyObject *op2 = nullptr;
PyObject *op3 = nullptr;
/* Extract arguments */
PyArg_ParseTuple(args, "|OOO", &op1, &op2, &op3);
if (op1 == nullptr || (!PyLong_Check(op1) && !PyInt_Check(op1)))
return PyErr_Format(PyExc_TypeError, "extract(): expected an integer as first argument");
if (op2 == nullptr || (!PyLong_Check(op2) && !PyInt_Check(op2)))
return PyErr_Format(PyExc_TypeError, "extract(): expected an integer as second argument");
if (op3 == nullptr || !PySmtAstNode_Check(op3))
return PyErr_Format(PyExc_TypeError, "extract(): expected a SmtAstNode as third argument");
return PySmtAstNode(smt2lib::extract(PyLong_AsUint(op1), PyLong_AsUint(op2), PySmtAstNode_AsSmtAstNode(op3)));
}
static PyObject *Triton_getModels(PyObject *self, PyObject *args)
{
PyObject *node = nullptr;
PyObject *limit;
PyObject *modelsList;
std::vector<std::list<Smodel>> models;
uint64 limit_c = 0;
uint64 modelsSize = 0;
/* Extract arguments */
PyArg_ParseTuple(args, "O|O", &node, &limit);
if (node == nullptr || !PySmtAstNode_Check(node))
return PyErr_Format(PyExc_TypeError, "getModels(): expected a SmtAstNode as first argument");
if (!PyLong_Check(limit) && !PyInt_Check(limit))
return PyErr_Format(PyExc_TypeError, "getModels(): expected a limit (integer) as second argument");
limit_c = PyLong_AsLong(limit);
if (limit_c == 0)
return PyErr_Format(PyExc_TypeError, "getModels(): The limit must be greater than 0");
/* Get models */
models = ap.getModels(PySmtAstNode_AsSmtAstNode(node), limit_c);
modelsSize = models.size();
modelsList = xPyList_New(modelsSize);
for (uint64 index = 0; index < modelsSize; index++){
std::list<Smodel> model = models[index];
std::list<Smodel>::iterator it;
/* Craft the model dictionary */
PyObject *modelDict = xPyDict_New();
for (it = model.begin() ; it != model.end(); it++)
PyDict_SetItemString(modelDict, it->getName().c_str(), Py_BuildValue("k", it->getValue()));
PyList_SetItem(modelsList, index, modelDict);
}
return modelsList;
}
static PyObject *SmtAstNode_setChild(PyObject *self, PyObject *args) {
PyObject *index;
PyObject *node;
uint64 i;
smt2lib::smtAstAbstractNode *dst, *src;
PyArg_ParseTuple(args, "O|O", &index, &node);
if (!PyLong_Check(index) && !PyInt_Check(index))
return PyErr_Format(PyExc_TypeError, "setChild(): expected an index (integer) as first argument");
if (!PySmtAstNode_Check(node))
return PyErr_Format(PyExc_TypeError, "setChild(): expected a SmtAstNode as second argument");
i = PyLong_AsLong(index);
src = PySmtAstNode_AsSmtAstNode(node);
dst = PySmtAstNode_AsSmtAstNode(self);
if (i >= dst->getChilds().size())
return PyErr_Format(PyExc_TypeError, "setChild(): index out-of-range");
dst->getChilds()[i] = src;
Py_RETURN_TRUE;
}
static PyObject *smt2lib_smtAssert(PyObject *self, PyObject *expr) {
if (!PySmtAstNode_Check(expr))
return PyErr_Format(PyExc_TypeError, "smtAssert(): expected a SmtAstNode as first argument");
return PySmtAstNode(smt2lib::smtAssert(PySmtAstNode_AsSmtAstNode(expr)));
}
static PyObject *smt2lib_bvnot(PyObject *self, PyObject *op1) {
if (!PySmtAstNode_Check(op1))
return PyErr_Format(PyExc_TypeError, "bvnot(): expected a SmtAstNode as first argument");
return PySmtAstNode(smt2lib::bvnot(PySmtAstNode_AsSmtAstNode(op1)));
}
