reait 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

reait/api.py

@@ -17,7 +17,7 @@ from numpy import array, vstack, dot, arccos, pi
 from pandas import DataFrame
 from lief import parse, Binary, ELF, PE, MachO
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"
 
 re_conf = {
     "apikey": "l1br3",
@@ -387,7 +387,7 @@ def RE_compute_distance(embedding: list, embeddings: list, nns: int = 5) -> list
     json_sims = [{"similaritiy": float(d[0]),
                   "vaddr": int(df.iloc[v]["vaddr"]),
                   "name": str(df.iloc[v]["name"]),
-                  "size": int(df.iloc[v]["size"])
+                  "size": int(df.iloc[v]["size"]),
                  } for d, v in similarities]
     return json_sims
 
@@ -405,7 +405,8 @@ def RE_nearest_symbols_batch(function_ids: list[int], nns: int = 5, collections:
     params = {"function_id_list": function_ids,
               "result_per_function": nns,
               "debug_mode": debug_enabled,
-              "distance": distance,}
+              "distance": distance,
+              }
 
     if collections:
         # api param is collection, not collections
@@ -437,7 +438,8 @@ def RE_nearest_functions(fpath: str, binary_id: int = 0, nns: int = 5,
 
     params = {"result_per_function": nns,
               "debug_mode": debug_enabled,
-              "distance": distance, }
+              "distance": distance,
+              }
 
     res: Response = reveng_req(requests.post, end_point, json_data=params)
 
@@ -500,6 +502,22 @@ def RE_functions_rename(function_id: int, new_name: str) -> Response:
     return res
 
 
+def RE_functions_rename_batch(mapping: dict[int, str]) -> Response:
+    """
+    Send a list of dictionaries, with a corresponding key as function ID and the desired function_name
+    :param mapping: dictionary containing the function_id as key and function_name as value
+    """
+    params = {"new_name_mapping":
+                  [{"function_id": func_id,
+                    "function_name": func_name,
+                    } for func_id, func_name in mapping.items()]
+              }
+
+    res: Response = reveng_req(requests.post, "v1/functions/batch/rename", json_data=params)
+
+    res.raise_for_status()
+    return res
+
 def RE_settings() -> Response:
     """
     Get the configuration settings
@@ -551,7 +569,7 @@ def re_binary_id(fpath: str) -> str:
     else:
         logger.error("File '%s' doesn't exist or isn't readable", fpath)
 
-    return "undefined"
+    return "Undefined"
 
 
 def _binary_isa(binary: Binary, exec_type: str) -> str:

{reait-1.0.0.dist-info → reait-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: reait
-Version: 1.0.0
+Version: 1.0.1
 Home-page: https://github.com/RevEng-AI/reait
 Author: James Patrick-Evans
 Author-email: James Patrick-Evans <james@reveng.ai>
@@ -692,15 +692,15 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: argparse
 Requires-Dist: requests
 Requires-Dist: rich
-Requires-Dist: argparse
 Requires-Dist: tomli
 Requires-Dist: pandas
 Requires-Dist: numpy
 Requires-Dist: scipy
-Requires-Dist: lief
 Requires-Dist: scikit-learn
+Requires-Dist: lief
 
 # reait
 

reait-1.0.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+reait/__init__.py,sha256=U3Y2DK1m2kht0sK_mBvvQskWRM5PwL5JtIDWurZOyBw,66
+reait/api.py,sha256=RqyC2wROiTTSyMwXl3hkyz97Jwr4Y_dxN4np-fdhdfg,22596
+reait/main.py,sha256=IDfmtQxiA-lHb1Xcp90oCP9wqRnPZCnmLpMxkRaZxiU,17066
+reait-1.0.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+reait-1.0.1.dist-info/METADATA,sha256=98HnmSd_jb8AsWzCzHDsFQKtghk3u_uE7_NarkEMcWA,47455
+reait-1.0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+reait-1.0.1.dist-info/entry_points.txt,sha256=8Ek11o7a6O8hjBFw6-1vmkBfbv_45O2vOKj5CDUB1e4,42
+reait-1.0.1.dist-info/top_level.txt,sha256=EnJssmctKe3Ugjcmu66L9_Q4elLdAwaXK6M8E6E8f_M,6
+reait-1.0.1.dist-info/RECORD,,

{reait-1.0.0.dist-info → reait-1.0.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

reait-1.0.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-reait/__init__.py,sha256=U3Y2DK1m2kht0sK_mBvvQskWRM5PwL5JtIDWurZOyBw,66
-reait/api.py,sha256=W2FDqKYzLk_ZRf6MueXCWXx1p9tuXg0ALftU8ak7JNg,21942
-reait/main.py,sha256=IDfmtQxiA-lHb1Xcp90oCP9wqRnPZCnmLpMxkRaZxiU,17066
-reait-1.0.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-reait-1.0.0.dist-info/METADATA,sha256=mxr_fzWU6gWMrQtGfsBKTl-s33So9-vBLOozN7FAiQY,47455
-reait-1.0.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-reait-1.0.0.dist-info/entry_points.txt,sha256=8Ek11o7a6O8hjBFw6-1vmkBfbv_45O2vOKj5CDUB1e4,42
-reait-1.0.0.dist-info/top_level.txt,sha256=EnJssmctKe3Ugjcmu66L9_Q4elLdAwaXK6M8E6E8f_M,6
-reait-1.0.0.dist-info/RECORD,,