reait 0.0.18__py3-none-any.whl → 0.0.20__py3-none-any.whl

reait/api.py

@@ -1,307 +1,561 @@
 #!/usr/bin/env python
-from __future__ import print_function
+# -*- coding: utf-8 -*-
+from __future__ import print_function, annotations
+
 from hashlib import sha256
-from rich import print_json, print as rich_print
+
 from sklearn.metrics.pairwise import cosine_similarity
-import os
-import re
-import argparse
+from os.path import basename, exists, expanduser
+from requests import request, Response, HTTPError
 import requests
-from numpy import array, vstack, mean, average, dot, arccos, pi
+from numpy import array, vstack, dot, arccos, pi
 from pandas import DataFrame
 import json
 import tomli
-from os.path import isfile
-from sys import exit
-from IPython import embed
-import lief
+import logging
+from lief import parse, ELF, PE, MachO
 
-__version__ = "0.0.18"
 
 re_conf = {
-    'apikey' : 'l1br3', 
-    'host' : 'https://api.reveng.ai',
-    'model': 'binnet-0.1'
+    "apikey": "l1br3",
+    "host": "https://api.reveng.ai",
+    "model": "binnet-0.2-x86"
 }
 
-def reveng_req(r: requests.request, end_point: str, data=None, ex_headers: dict = None, params=None):
+
+logger = logging.getLogger("REAIT")
+
+
+def reveng_req(r: request, end_point: str, data=None, ex_headers: dict = None, params=None,
+               json_data: dict = None, timeout: int = 30) -> Response:
+    """
+    Constructs and sends a Request
+    :param r: Method for the new Request
+    :param end_point: Endpoint to add to the base URL
+    :param ex_headers: Extended HTTP headers to add
+    :param data: Dictionary, list of tuples, bytes, or file-like object to send in the body
+    :param params: Dictionary, list of tuples or bytes to send in the query string for the query string
+    :param json_data: A JSON serializable Python object to send in the body
+    :param timeout: Number of seconds to stop waiting for a Response
+    """
     url = f"{re_conf['host']}/{end_point}"
-    headers = { "Authorization": f"{re_conf['apikey']}" }
+    headers = {"Authorization": re_conf["apikey"]}
+
     if ex_headers:
         headers.update(ex_headers)
-    return r(url, headers=headers, data=data, params=params)
 
+    logger.debug("Making request %s:\n  - headers: %s\n  - data: %s\n  - json_data: %s\n  - params: %s",
+                 url, headers, data, json_data, params)
+
+    response: Response = r(url, headers=headers, json=json_data, data=data, params=params, timeout=timeout)
+
+    logger.debug("Making response %s:\n  - headers: %s\n  - status_code: %d\n  - content: %s",
+                 url, response.headers, response.status_code, response.text)
+
+    return response
+
+
+def re_hash_check(bin_id: str) -> bool:
+    status = False
+    res = reveng_req(requests.get, f"search?search=sha_256_hash:{bin_id}&state=All&user_owned=true")
 
-def RE_delete(fpath: str, model_name: str):
-    """
-        Delete analysis results for Binary ID in command
-    """
-    bin_id = binary_id(fpath)
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.delete, f"/analyse/{bin_id}", params=params)
     if res.status_code == 200:
-        print(f"[+] Success. Securely deleted {fpath} analysis")
-    elif res.status_code == 404:
-        print(f"[!] Error, analysis not found for {bin_id} under {model_name}.")
+        binaries_data = res.json()["binaries"]
+        status = len(binaries_data) > 0
+    elif res.status_code == 400:
+        logger.warning("Bad Request: %s", res.text)
     else:
-        print(f"[!] Error deleteing binary {bin_id} under {model_name}. Server returned {res.status_code}.")
-    return
+        logger.error("Internal Server Error.")
+
+    res.raise_for_status()
+    return status
+
 
+# Bin_id is referred to as hash in this program - to maintain usage BID = id of a binary bin_id = hash
+# Assumes a file has been passed, correct hash only
+# Returns the BID of the binary_id (hash)
+def re_bid_search(bin_id: str) -> int:
+    res = reveng_req(requests.get, f"search?search=sha_256_hash:{bin_id}&state=All")
+
+    bid = -1
+
+    # Valid request
+    if res.status_code == 200:
+        # Check only one record is returned
+        binaries_data = res.json()["binaries"]
+
+        if len(binaries_data) > 1:
+            logger.info("%d matches found for hash: %s.", len(binaries_data), bin_id)
+
+            if len(binaries_data) > 1:
+                options_dict = {}
+
+                for idx, binary in enumerate(binaries_data):
+                    logger.info("[%d] - ID: {}, Name: %s, Creation: %s, Model: %s, Owner: %s, Status: %s",
+                                idx, binary["binary_id"], binary["binary_name"], binary["creation"],
+                                binary["model_name"], binary["owner"], binary["status"])
+
+                    options_dict[idx] = binary["binary_id"]
+
+                user_input = input("[+] Please enter the option you want to use for this operation:")
+
+                try:
+                    option_number = int(user_input)
+
+                    bid = options_dict.get(option_number, -1)
+
+                    if bid == -1:
+                        logger.warning("Invalid option.")
+                except Exception:
+                    bid = -1
+                    logger.warning("Invalid option.")
+            # Only 1 match found
+            elif len(binaries_data) == 1:
+                binary = binaries_data[0]
+                bid = binary["binary_id"]
+            else:
+                logger.warning("No matches found for hash: %s.", bin_id)
+        elif len(binaries_data) == 1:
+            binary = binaries_data[0]
+            bid = binary["binary_id"]
+
+            logger.info("Only one record exists, selecting - ID: %d, Name: %s, "
+                        "Creation: %s, Model: %s, Owner: %s, Status: %s",
+                        bid, binary["binary_name"], binary["creation"],
+                        binary["model_name"], binary["owner"], binary["status"])
+        else:
+            logger.warning("No matches found for hash: %s.", bin_id)
+    elif res.status_code == 400:
+        logger.warning("Bad Request: %s", res.text)
+    else:
+        logger.error("Internal Server Error.")
 
-def RE_analyse(fpath: str, model: str = None, isa_options: str = None, platform_options: str = None, file_options: str = None, dynamic_execution: bool = False, command_line_args: str = None, scope: str = None, tags: str = None):
+    res.raise_for_status()
+    return bid
+
+
+def RE_delete(fpath: str, binary_id: int = 0) -> Response:
     """
-        Start analysis job for binary file
+    Delete analysis results for Binary ID in command
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    filename = os.path.basename(fpath)
-    params={ 'file_name': filename }
-    for p_name in ('model', 'isa_options', 'platform_options', 'file_options', 'dynamic_execution', 'command_line_args', 'scope', 'tags'):
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.delete, f"analyse/{bid}")
+
+    if res.status_code == 200:
+        logger.info("Securely deleted %s analysis.", bin_id)
+    elif res.status_code == 404:
+        logger.warning("Error analysis not found for %s.", bin_id)
+    else:
+        logger.error("Error deleting binary %s under. Server returned %d.", bin_id, res.status_code)
+
+    res.raise_for_status()
+    return res
+
+
+def RE_analyse(fpath: str, model_name: str = None, isa_options: str = None, platform_options: str = None,
+               file_options: str = None, dynamic_execution: bool = False, command_line_args: str = None,
+               scope: str = None, tags: list = None, priority: int = 0,
+               duplicate: bool = False, symbols: dict = None) -> Response:
+    """
+    Start analysis job for binary file
+    :param fpath: File path for binary to analyse
+    :param model_name: Binary model name
+    :param isa_options: Executable ISA
+    :param file_options: File options
+    :param platform_options: OS platform
+    :param dynamic_execution: Enable dynamic execution in sandbox during analysis
+    :param command_line_args: Command line arguments to pass when running binary sample in the sandbox
+    :param scope: Analysis visibility
+    :param tags: Assign tags to an analysis
+    :param priority: Priority to processing queue
+    :param duplicate: Duplicate an existing binary
+    :param symbols: List of functions
+    """
+    bin_id = re_binary_id(fpath)
+    result = re_hash_check(bin_id)
+
+    if result and duplicate is False:
+        logger.error("Error, duplicate analysis for %s. To upload again, use the --duplicate flag.",
+                     bin_id)
+        raise HTTPError(f"Duplicate analysis for hash: {bin_id}")
+
+    filename = basename(fpath)
+
+    params = {"file_name": filename, "sha_256_hash": bin_id}
+
+    for p_name in ("model_name", "isa_options", "platform_options", "file_options",
+                   "dynamic_execution", "command_line_args", "scope", "tags", "priority", "symbols"):
         p_value = locals()[p_name]
+
         if p_value:
             params[p_name] = p_value
 
-    res = reveng_req(requests.post, f"analyse", data=open(fpath, 'rb').read(), params=params)
+    res = reveng_req(requests.post, f"analyse", json_data=params)
+
     if res.status_code == 200:
-        print("[+] Successfully submitted binary for analysis.")
-        print(f"[+] {fpath} - {binary_id(fpath)}")
-        return res
+        logger.info("Successfully submitted binary for analysis. %s - %s", fpath, re_binary_id(fpath))
+    elif res.status_code == 400:
+        response = res.json()
 
-    if res.status_code == 400:
-        response = json.loads(res.text)
-        if 'error' in response.keys():
-            print(f"[-] Error analysing {fpath} - {response['error']}. Please check the results log file for {binary_id(fpath)}")
-            return res
+        if "error" in response.keys():
+            logger.warning("Error analysing %s - %s", fpath, response["error"])
 
     res.raise_for_status()
+    return res
 
 
-def RE_upload(fpath: str):
+def RE_upload(fpath: str) -> Response | bool:
     """
-        Upload binary to Server
+    Upload binary to Server
+    :param fpath: File path for binary to analyse
     """
-    res = reveng_req(requests.post, f"upload", data=open(fpath, 'rb').read())
+    bin_id = re_binary_id(fpath)
+    result = re_hash_check(bin_id)
+
+    if result:
+        logger.info("File %s - %s already exists. Skipping upload...", basename(fpath), re_binary_id(fpath))
+        return True
+
+    res = reveng_req(requests.post, f"upload", data=open(fpath, "rb").read())
+
     if res.status_code == 200:
-        print("[+] Successfully uploaded binary to your account.")
-        print(f"[+] {fpath} - {binary_id(fpath)}")
-        return res
+        logger.info("Successfully uploaded binary to your account. %s - %s", fpath, re_binary_id(fpath))
+    elif res.status_code == 400:
+        response = res.json()
 
-    if res.status_code == 400:
-        if 'already exists' in json.loads(res.text)['reason']:
-            print(f"[-] {fpath} already exists. Please check the results log file for {binary_id(fpath)}")
-            return True
+        if "error" in response.keys():
+            logger.warning("Error uploading %s - %s", fpath, response["error"])
+    elif res.status_code == 413:
+        logger.warning("File too large. Please upload files under 100MB.")
+    elif res.status_code == 500:
+        logger.error("Internal Server Error. Please contact support. Skipping upload...")
 
     res.raise_for_status()
+    return res
 
 
-def RE_embeddings(fpath: str, model_name: str):
+def RE_embeddings(fpath: str, binary_id: int = 0) -> Response:
     """
-        Fetch symbol embeddings
+    Fetch symbol embeddings
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.get, f"embeddings/{binary_id(fpath)}", params=params)
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"embeddings/{bid}")
+
     if res.status_code == 400:
-        print(f"[-] Analysis for {binary_id(fpath)} still in progress. Please check the logs (-l) and try again later.")
+        logger.warning("Analysis for %s still in progress. Please check the logs (-l) and try again later.",
+                       bin_id)
 
     res.raise_for_status()
-    return res.json()
+    return res
 
 
-def RE_signature(fpath: str, model_name: str):
+def RE_signature(fpath: str, binary_id: int = 0) -> Response:
     """
-        Fetch binary BinNet signature
+    Fetch binary BinNet signature
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.get, f"signature/{binary_id(fpath)}", params=params)
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"signature/{bid}")
+
     if res.status_code == 425:
-        print(f"[-] Analysis for {binary_id(fpath)} still in progress. Please check the logs (-l) and try again later.")
+        logger.warning("Analysis for %s still in progress. Please check the logs (-l) and try again later.",
+                       bin_id)
 
     res.raise_for_status()
-    return res.json()
+    return res
 
 
-def RE_embedding(fpath: str, start_vaddr: int, end_vaddr: int = None, base_vaddr: int = None, model: str = None):
+def RE_embedding(fpath: str, start_vaddr: int, end_vaddr: int = None, base_vaddr: int = None,
+                 model: str = None) -> Response:
     """
-        Fetch embedding for custom symbol range
+    Fetch embedding for custom symbol range
+    :param fpath: File path for binary to analyse
+    :param start_vaddr: Start virtual address of the function to extract embeddings
+    :param end_vaddr: End virtual address of the function to extract embeddings
+    :param base_vaddr: Base address of the binary
+    :param model: Binary model name
     """
     params = {}
 
     if end_vaddr:
-        params['end_vaddr']: end_vaddr
+        params["end_vaddr"] = end_vaddr
     if base_vaddr:
-        params['base_vaddr']: base_vaddr
+        params["base_vaddr"] = base_vaddr
     if model:
-        params['model']: model
+        params["models"] = model
+
+    bin_id = re_binary_id(fpath)
+
+    res = reveng_req(requests.get, f"embedding/{bin_id}/{start_vaddr}", params=params)
 
-    res = reveng_req(requests.get, f"embedding/{binary_id(fpath)}/{start_vaddr}", params=params)
     if res.status_code == 425:
-        print(f"[-] Analysis for {binary_id(fpath)} still in progress. Please check the logs (-l) and try again later.")
-        return
+        logger.warning("Analysis for %s still in progress. Please check the logs (-l) and try again later.",
+                       bin_id)
 
     res.raise_for_status()
-    return res.json()
+    return res
 
 
-def RE_logs(fpath: str, model_name: str):
+def RE_logs(fpath: str, binary_id: int = 0, console: bool = True) -> Response:
     """
-        Delete analysis results for Binary ID in command
+    Get the logs for an analysis associated to Binary ID in command
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
+    :param console: Show response in console
     """
-    bin_id = binary_id(fpath)
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.get, f"/logs/{bin_id}", params=params)
-    if res.status_code == 200:
-        print(res.text)
-        return
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"logs/{bid}")
+
+    if res.status_code == 200 and console:
+        logger.info("Logs found for %s:\n%s", bin_id, res.text)
     elif res.status_code == 404:
-        print(f"[!] Error, binary analysis for {bin_id} under {model_name} not found.")
-        return
+        logger.warning("Error, logs not found for %s.", bin_id)
 
     res.raise_for_status()
+    return res
 
 
-def RE_cves(fpath: str, model_name: str):
+def RE_cves(fpath: str, binary_id: int = 0) -> Response:
     """
-        Check for known CVEs in Binary 
+    Check for known CVEs in Binary
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    bin_id = binary_id(fpath)
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.get, f"/cves/{bin_id}", params)
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"cves/{bid}")
+
     if res.status_code == 200:
         cves = json.loads(res.text)
-        rich_print(f"[bold blue]Checking for known CVEs embedded inside [/bold blue] [bold bright_green]{fpath}[/bold bright_green]:")
+        logger.info("Checking for known CVEs embedded inside %s", fpath)
+
         if len(cves) == 0:
-            rich_print(f"[bold bright_green]0 CVEs found.[/bold bright_green]")
+            logger.info("0 CVEs found.")
         else:
-            rich_print(f"[bold red]Warning CVEs found![/bold red]")
-            print_json(data=cves)
-        return
+            logger.warning("Warning CVEs found!\n%s", res.text)
     elif res.status_code == 404:
-        print(f"[!] Error, binary analysis for {bin_id} not found.")
-        return
+        logger.warning("Error, binary analysis not found for %s.", bin_id)
 
     res.raise_for_status()
+    return res
+
 
-def RE_status(fpath: str, model_name: str):
+def RE_status(fpath: str, binary_id: int = 0) -> Response:
     """
-        Check for known CVEs in Binary 
+    Get the status of an ongoing binary analysis
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    bin_id = binary_id(fpath)
-    params = { 'model_name': model_name }
-    res = reveng_req(requests.get, f"/analyse/status/{bin_id}", params)
-    if res.status_code == 200:
-        return res.json()
-    elif res.status_code == 400:
-        print(f"[!] Error, status not found for {bin_id}:{model_name} not found.")
-        return res.json()
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"analyse/status/{bid}")
+
+    if res.status_code == 400:
+        logger.warning(" Error, status not found for %s.", bin_id)
 
     res.raise_for_status()
+    return res
 
 
-def RE_compute_distance(embedding: list, embeddings: list, nns: int = 5):
+def RE_compute_distance(embedding: list, embeddings: list, nns: int = 5) -> list:
     """
-        Compute the cosine distance between source embedding and embeddinsg from binary
+    Compute the cosine distance between source embedding and embedding from binary
+    :param embedding: Embedding vector as python list
+    :param embeddings: Symbol embeddings
+    :param nns: Number of nearest neighbors
     """
     df = DataFrame(data=embeddings)
     np_embedding = array(embedding).reshape(1, -1)
-    source_embeddings = vstack(df['embedding'].values)
+    source_embeddings = vstack(df["embedding"].values)
     closest = cosine_similarity(source_embeddings, np_embedding).squeeze().argsort()[::-1][:nns]
     distances = cosine_similarity(source_embeddings[closest], np_embedding)
+
     # match closest embeddings with similarity
     closest_df = df.iloc[closest]
+
     # create json similarity object
     similarities = list(zip(distances, closest_df.index.tolist()))
-    json_sims = [{'similaritiy': float(d[0]), 'vaddr': int(df.iloc[v]['vaddr']), 'name': str(df.iloc[v]['name']), 'size': int(df.iloc[v]['size'])} for d, v in similarities]
+    json_sims = [{"similaritiy": float(d[0]), "vaddr": int(df.iloc[v]["vaddr"]), "name": str(df.iloc[v]["name"]),
+                  "size": int(df.iloc[v]["size"])} for d, v in similarities]
     return json_sims
 
 
-def RE_nearest_symbols(embedding: list, model_name, nns: int = 5, collections : list = None):
+def RE_nearest_symbols(embedding: list, model_name: str, nns: int = 5,
+                       collections: list = None, ignore_hashes: list = None,
+                       distance: float = 0.0, debug_enabled: bool = False) -> Response:
     """
-        Get function name suggestions for an embedding
-        :param embedding: embedding vector as python list
-        :param nns: Number of nearest neighbors
-        :param collections: str RegEx to search through RevEng.AI collections
+    Get function name suggestions for an embedding
+    :param embedding: Embedding vector as python list
+    :param model_name: Binary model name
+    :param nns: Number of nearest neighbors
+    :param collections: List of collections RevEng.AI collection names to search through
+    :param ignore_hashes: List[str] SHA-256 hash of binary file to ignore symbols from (usually the current binary)
+    :param distance: How close we want the ANN search to filter for
+    :param debug_enabled: ANN Symbol Search, only perform ANN on debug symbols if set
     """
-    params={'nns': nns, 'model_name': model_name }
+    params = {"nns": nns, "model_name": model_name, "debug_enabled": debug_enabled}
+
+    if collections and len(collections) > 0:
+        # api param is collection, not collections
+        params["collection"] = "|".join(collections)
+
+    if ignore_hashes and len(ignore_hashes) > 0:
+        params["ignore_hashes"] = ignore_hashes
 
-    if collections:
-        params['collections'] = collections
+    if distance > 0.0:
+        params["distance"] = distance
 
     res = reveng_req(requests.post, "ann/symbol", data=json.dumps(embedding), params=params)
+
     res.raise_for_status()
-    f_suggestions = res.json()
-    print_json(data=f_suggestions)
+    return res
 
 
-def RE_nearest_binaries(embedding: list, model_name, nns: int = 5, collections : list = None):
+def RE_nearest_binaries(embedding: list, model_name: str, nns: int = 5,
+                        collections: list = None, ignore_hashes: list = None) -> Response:
     """
-        Get executable suggestions for a binary embedding
-        :param embedding: embedding vector as python list
-        :param nns: Number of nearest neighbors
-        :param collections: str RegEx to search through RevEng.AI collections
+    Get executable suggestions for a binary embedding
+    :param embedding: Embedding vector as python list
+    :param model_name: Binary model name
+    :param nns: Number of nearest neighbors
+    :param collections: List of collections RevEng.AI collection names to search through
+    :param ignore_hashes: List[str] SHA-256 hash of binary files to ignore symbols from (usually the current binary)
     """
-    params={'nns': nns, 'model_name': model_name }
+    params = {"nns": nns, "model_name": model_name}
+
+    if collections and len(collections) > 0:
+        # api param is collection, not collections
+        params["collection"] = "|".join(collections)
 
-    if collections:
-        params['collections'] = collections
+    if ignore_hashes and len(ignore_hashes) > 0:
+        params["ignore_hashes"] = ignore_hashes
 
     res = reveng_req(requests.post, "ann/binary", data=json.dumps(embedding), params=params)
+
     res.raise_for_status()
-    f_suggestions = res.json()
-    print_json(data=f_suggestions)
+    return res
 
 
-def RE_SBOM(fpath: str, model_name: str):
+def RE_SBOM(fpath: str, binary_id: int = 0) -> Response:
     """
-        Get Software Bill Of Materials for binary
-        :param fpath: File path for binaty to analyse
-        :param model_name: str model name of RevEng.AI AI model
+    Get Software Bill Of Materials for binary
+    :param fpath: File path for binary to analyse
+    :param binary_id: ID of binary
     """
-    params={'model_name': model_name }
+    bin_id = re_binary_id(fpath)
+    bid = re_bid_search(bin_id) if binary_id == 0 else binary_id
+
+    if bid == -1:
+        raise HTTPError(f"No matches found for hash: {bin_id}")
+
+    res = reveng_req(requests.get, f"sboms/{bid}")
+
+    logger.info("SBOM for %s:\n%s", fpath, res.text)
 
-    res = reveng_req(requests.get, f"sboms/{binary_id(fpath)}", params=params)
     res.raise_for_status()
-    sbom = res.json()
-    print_json(data=sbom)
+    return res
 
 
-def binary_id(path: str):
-    """Take the SHA-256 hash of binary file"""
+def RE_functions_rename(function_id: int, new_name: str) -> Response:
+    """
+    Send the new name of a function to C2
+    :param function_id: ID of a function
+    :param new_name: New function name
+    """
+    res = reveng_req(requests.post, f"functions/rename/{function_id}", json_data={"new_name": new_name})
+
+    if res.status_code == 200:
+        logger.info("FunctionId %d has been renamed with '%s'.", function_id, new_name)
+    else:
+        logger.warning("Error, cannot rename FunctionId %d. %s", function_id, res.text)
+
+    res.raise_for_status()
+    return res
+
+
+def re_binary_id(fpath: str) -> str:
+    """
+    Take the SHA-256 hash of binary file
+    :param fpath: File path for binary to analyse
+    """
+    if not fpath or not exists(fpath):
+        return "undefined"
+
     hf = sha256()
-    with open(path, "rb") as f:
+
+    with open(fpath, "rb") as f:
         c = f.read()
         hf.update(c)
+
     return hf.hexdigest()
 
 
-def _binary_isa(lief_hdlr, exec_type):
+def _binary_isa(lief_hdlr, exec_type: str) -> str:
     """
-        Get executable file format
+    Get ISA format
     """
     if exec_type == "elf":
         machine_type = lief_hdlr.header.machine_type
-        if machine_type == lief.ELF.ARCH.i386:
+
+        if machine_type == ELF.ARCH.i386:
             return "x86"
-        elif machine_type == lief.ELF.ARCH.x86_64:
+        elif machine_type == ELF.ARCH.x86_64:
             return "x86_64"
-
     elif exec_type == "pe":
         machine_type = lief_hdlr.header.machine
-        if machine_type == lief.PE.MACHINE_TYPES.I386:
+
+        if machine_type == PE.MACHINE_TYPES.I386:
             return "x86"
-        elif machine_type == lief.PE.MACHINE_TYPES.AMD64:
+        elif machine_type == PE.MACHINE_TYPES.AMD64:
             return "x86_64"
-
     elif exec_type == "macho":
         machine_type = lief_hdlr.header.cpu_type
-        if machine_type == lief.MachO.CPU_TYPES.x86:
+
+        if machine_type == MachO.CPU_TYPES.x86:
             return "x86"
-        elif machine_type == lief.MachO.CPU_TYPES.x86_64:
+        elif machine_type == MachO.CPU_TYPES.x86_64:
             return "x86_64"
-    
-    raise RuntimeError(f"Error, failed to determine or unsupported ISA for exec_type:{exec_type}")
 
+    logger.error("Error, failed to determine or unsupported ISA for exec_type: %s.", exec_type)
+    raise RuntimeError(f"Error, failed to determine or unsupported ISA for exec_type:{exec_type}.")
 
-def _binary_format(lief_hdlr):
+
+def _binary_format(lief_hdlr) -> str:
     """
-        Get executable file format
+    Get executable file format
     """
     if lief_hdlr.format == lief_hdlr.format.PE:
         return "pe"
@@ -309,41 +563,42 @@ def _binary_format(lief_hdlr):
         return "elf"
     if lief_hdlr.format == lief_hdlr.format.MACHO:
         return "macho"
-    
-    raise RuntimeError("Error, could not determine binary format")
 
+    logger.error("Error, could not determine binary format: %s.", lief_hdlr.format)
+    raise RuntimeError("Error, could not determine binary format.")
 
 
-def file_type(fpath: str):
+def file_type(fpath: str) -> tuple[str, str]:
     """
-        Determine ISA for binary
+    Determine ISA for binary
+    :param fpath: File path for binary to analyse
     """
-    binary = lief.parse(fpath)  
+    binary = parse(fpath)
 
     # handle PE and ELF files
     file_format = _binary_format(binary)
-    isa         = _binary_isa(binary, file_format)
+    isa = _binary_isa(binary, file_format)
+
     return file_format, isa
 
 
-def parse_config():
+def parse_config() -> None:
+    """
+    Parse ~/.reait.toml config file
     """
-        Parse ~/.reait.toml config file
-    """     
-    if not os.path.exists(os.path.expanduser("~/.reait.toml")):
-        return
+    if exists(expanduser("~/.reait.toml")):
+        with open(expanduser("~/.reait.toml"), "r") as file:
+            config = tomli.loads(file.read())
 
-    with open(os.path.expanduser("~/.reait.toml"), "r") as file:
-        config = tomli.loads(file.read())
-        for key in ('apikey', 'host'):
-            if key in config:
-                re_conf[key] = config[key]
+            for key in ("apikey", "host", "model"):
+                if key in config:
+                    re_conf[key] = config[key]
 
 
-def angular_distance(x, y):
+def angular_distance(x, y) -> float:
     """
     Compute angular distance between two embedding vectors
-    Normalised euclidean distance
+    Normalised euclidian distance
     """
     cos = dot(x, y) / ((dot(x, x) * dot(y, y)) ** 0.5)
-    return 1.0 - arccos(cos)/pi
+    return 1.0 - arccos(cos) / pi