static PyObject *smt2lib_bvlshr(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, "bvlshr(): expected a SmtAstNode as first argument"); if (op2 == nullptr || !PySmtAstNode_Check(op2)) return PyErr_Format(PyExc_TypeError, "bvlshr(): expected a SmtAstNode as second argument"); return PySmtAstNode(smt2lib::bvlshr(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 *SmtAstNode_getChilds(PyObject *self, PyObject *noarg) { PyObject *childs; smt2lib::smtAstAbstractNode *node = PySmtAstNode_AsSmtAstNode(self); uint64 size = node->getChilds().size(); childs = xPyList_New(size); uint64 index = 0; for ( ; index < size; index++) PyList_SetItem(childs, index, PySmtAstNode(node->getChilds()[index])); return childs; }
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 *SmtAstNode_getValue(PyObject *self, PyObject *noarg) { smt2lib::smtAstAbstractNode *node = PySmtAstNode_AsSmtAstNode(self); if (node->getKind() == smt2lib::DECIMAL_NODE) return uint128ToPyLongObject(reinterpret_cast<smt2lib::smtAstDecimalNode *>(node)->getValue()); else if (node->getKind() == smt2lib::STRING_NODE) return Py_BuildValue("s", reinterpret_cast<smt2lib::smtAstStringNode *>(node)->getValue().c_str()); else if (node->getKind() == smt2lib::REFERENCE_NODE) return Py_BuildValue("k", reinterpret_cast<smt2lib::smtAstReferenceNode *>(node)->getValue()); return PyErr_Format(PyExc_TypeError, "SmtAstNode.getValue() - Cannot use getValue() on this kind of node"); }
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 *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 *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 *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))); }
static PyObject *SmtAstNode_getKind(PyObject *self, PyObject *noarg) { return Py_BuildValue("k", PySmtAstNode_AsSmtAstNode(self)->getKind()); }