meshcore-cli 1.2.1__tar.gz → 1.2.2__tar.gz

{meshcore_cli-1.2.1 → meshcore_cli-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: meshcore-cli
-Version: 1.2.1
+Version: 1.2.2
 Summary: Command line interface to meshcore companion radios
 Project-URL: Homepage, https://github.com/fdlamotte/meshcore-cli
 Project-URL: Issues, https://github.com/fdlamotte/meshcore-cli/issues
@@ -12,6 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Requires-Dist: meshcore>=2.1.19
 Requires-Dist: prompt-toolkit>=3.0.50
+Requires-Dist: pycryptodome
 Requires-Dist: requests>=2.28.0
 Description-Content-Type: text/markdown
 

{meshcore_cli-1.2.1 → meshcore_cli-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "meshcore-cli"
-version = "1.2.1"
+version = "1.2.2"
 authors = [
   { name="Florent de Lamotte", email="florent@frizoncorrea.fr" },
 ]
@@ -17,7 +17,7 @@ classifiers = [
 ]
 license = "MIT"
 license-files = ["LICEN[CS]E*"]
-dependencies = [ "meshcore >= 2.1.19", "prompt_toolkit >= 3.0.50", "requests >= 2.28.0" ]
+dependencies = [ "meshcore >= 2.1.19", "prompt_toolkit >= 3.0.50", "requests >= 2.28.0", "pycryptodome" ]
 
 [project.urls]
 Homepage = "https://github.com/fdlamotte/meshcore-cli"

{meshcore_cli-1.2.1 → meshcore_cli-1.2.2}/src/meshcore_cli/meshcore_cli.py

@@ -24,13 +24,15 @@ from prompt_toolkit.key_binding import KeyBindings
 from prompt_toolkit.shortcuts import radiolist_dialog
 from prompt_toolkit.completion.word_completer import WordCompleter
 from prompt_toolkit.document import Document
+from hashlib import sha256
+from Crypto.Cipher import AES
 
 import re
 
 from meshcore import MeshCore, EventType, logger
 
 # Version
-VERSION = "v1.2.1"
+VERSION = "v1.2.2"
 
 # default ble address is stored in a config file
 MCCLI_CONFIG_DIR = str(Path.home()) + "/.config/meshcore/"
@@ -193,6 +195,41 @@ process_event_message.print_snr=False
 process_event_message.color=True
 process_event_message.last_node=None
 
+async def handle_log_rx(event):
+    mc = handle_log_rx.mc
+    if handle_log_rx.json_log_rx: # json mode ... raw dump
+        msg = json.dumps(event.payload)
+        if handle_message.above:
+            print_above(msg)
+        else :
+            print(msg)
+        return
+
+    pkt = bytes().fromhex(event.payload["payload"])
+
+    if handle_log_rx.log_channels:
+        if pkt[0] == 0x15: 
+            path_len = pkt[1]
+            path = pkt[2:path_len+2].hex()
+            chan_hash = pkt[path_len+2:path_len+3].hex()
+            cipher_mac = int.from_bytes(pkt[path_len+3:path_len+5], byteorder="little")
+            msg = pkt[path_len+5:]
+            channel = await get_channel_by_hash(mc, chan_hash)
+            if channel is None :
+                chan_name = chan_hash
+                message = msg.hex()
+            else:
+                chan_name = channel["channel_name"]
+                aes_key = bytes.fromhex(channel["channel_secret"])
+                cipher = AES.new(aes_key, AES.MODE_ECB)
+                message = cipher.decrypt(msg)[5:].decode("utf-8").strip("\x00")
+            
+            print_above(f"{ANSI_LIGHT_GRAY}{chan_name:>10} {ANSI_GREEN}{message[0:25]:25} {ANSI_LIGHT_GRAY}({event.payload['snr']:6,.2f},{event.payload['rssi']:4}){ANSI_YELLOW} [{path}]{ANSI_END}")
+            
+handle_log_rx.json_log_rx = False
+handle_log_rx.log_channels = False
+handle_log_rx.mc = None
+
 async def handle_advert(event):
     if not handle_advert.print_adverts:
         return
@@ -320,7 +357,7 @@ class MyNestedCompleter(NestedCompleter):
                 opts = self.options.keys()
             completer = WordCompleter(
                 opts, ignore_case=self.ignore_case,
-                pattern=re.compile(r"([a-zA-Z0-9_\\/]+|[^a-zA-Z0-9_\s]+)"))
+                pattern=re.compile(r"([a-zA-Z0-9_\\/\#]+|[^a-zA-Z0-9_\s\#]+)"))
             yield from completer.get_completions(document, complete_event)
         else: # normal behavior for remainder
             yield from super().get_completions(document, complete_event)
@@ -417,6 +454,8 @@ def make_completion_dict(contacts, pending={}, to=None, channels=None):
             "color" : {"on":None, "off":None},
             "print_name" : {"on":None, "off":None},
             "print_adverts" : {"on":None, "off":None},
+            "json_log_rx" : {"on":None, "off":None},
+            "log_channels" : {"on":None, "off":None},
             "print_new_contacts" : {"on": None, "off":None},
             "print_path_updates" : {"on":None,"off":None},
             "classic_prompt" : {"on" : None, "off":None},
@@ -444,6 +483,8 @@ def make_completion_dict(contacts, pending={}, to=None, channels=None):
             "color":None,
             "print_name":None,
             "print_adverts":None,
+            "json_log_rx":None,
+            "log_channels":None,
             "print_path_updates":None,
             "print_new_contacts":None,
             "classic_prompt":None,
@@ -604,6 +645,14 @@ def make_completion_dict(contacts, pending={}, to=None, channels=None):
 
     completion_list.update(slash_contacts_completion_list)
 
+    slash_chan_completion_list = {}
+    if not channels is None:
+        for c in channels :
+            if c["channel_name"] != "":
+                slash_chan_completion_list["/" + c["channel_name"]] = None
+
+    completion_list.update(slash_chan_completion_list)
+
     completion_list.update({
         "script" : None,
         "quit" : None
@@ -675,7 +724,7 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
                 if classic :
                     prompt = prompt + " > "
                 else :
-                    prompt = prompt + "🭨"
+                    prompt = prompt + f"{ANSI_NORMAL}🭬{ANSI_INVERT}"
 
             if not contact is None :
                 if not last_ack:
@@ -696,7 +745,7 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
                     prompt = prompt + f"{ANSI_INVERT}"
 
                 if print_name and not classic :
-                    prompt = prompt + "🭬"
+                    prompt = prompt + f"{ANSI_NORMAL}🭨{ANSI_INVERT}"
 
                 prompt = prompt + f"{contact['adv_name']}"
                 if classic :
@@ -734,8 +783,19 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
             elif line.startswith("/") :
                 path = line.split(" ", 1)[0]
                 if path.count("/") == 1:
-                    args = shlex.split(line[1:])
-                    await process_cmds(mc, args)
+                    args = line[1:].split(" ")
+                    tct = mc.get_contact_by_name(args[0])
+                    if len(args)>1 and not tct is None: # a contact, send a message
+                        if tct["type"] == 1 or tct["type"] == 3: # client or room
+                            last_ack = await msg_ack(mc, tct, line.split(" ", 1)[1])
+                        else:
+                            print("Can only send msg to chan, client or room")
+                    else :
+                        ch = await get_channel_by_name(mc, args[0])
+                        if len(args)>1 and not ch is None: # a channel, send message
+                            await send_chan_msg(mc, ch["channel_idx"], line.split(" ", 1)[1])
+                        else :
+                            await process_cmds(mc, shlex.split(line[1:]))
                 else:
                     cmdline = line[1:].split("/",1)[1]
                     contact_name = path[1:].split("/",1)[0]
@@ -744,10 +804,11 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
                         print(f"{contact_name} is not a contact")
                     else:
                         if not await process_contact_chat_line(mc, tct, cmdline):
-                            if tct["type"] == 1:
-                                last_ack = await msg_ack(mc, tct, cmdline)
-                            else :
-                                await process_cmds(mc, ["cmd", tct["adv_name"], cmdline])
+                            if cmdline != "":
+                                if tct["type"] == 1:
+                                    last_ack = await msg_ack(mc, tct, cmdline)
+                                else :
+                                    await process_cmds(mc, ["cmd", tct["adv_name"], cmdline])
 
             elif line.startswith("to ") : # dest
                 dest = line[3:]
@@ -805,7 +866,7 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
                 if ln is None :
                     print("No received msg yet !")
                 elif ln["type"] == 0 :
-                    await process_cmds(mc, ["chan", str(contact["chan_nb"]), line]  )
+                    await send_chan_msg(mc, ln["chan_nb"], line[1:])
                 else :
                     last_ack = await msg_ack(mc, ln, line[1:])
                     if last_ack == False :
@@ -837,7 +898,7 @@ Line starting with \"$\" or \".\" will issue a meshcli command.
                 last_ack = await msg_ack(mc, contact, line)
 
             elif contact["type"] == 0 : # channel, send msg to channel
-                await process_cmds(mc, ["chan", str(contact["chan_nb"]), line]  )
+                await send_chan_msg(mc, contact["chan_nb"], line)
 
             elif contact["type"] == 1 : # chat, send to recipient and wait ack
                 last_ack = await msg_ack(mc, contact, line)
@@ -970,7 +1031,18 @@ async def process_contact_chat_line(mc, contact, line):
         password_file = ""
         password = ""
         if os.path.isdir(MCCLI_CONFIG_DIR) :
+            # if a password file exists with node name open it and destroy it
             password_file = MCCLI_CONFIG_DIR + contact['adv_name'] + ".pass"
+            if os.path.exists(password_file) :
+                with open(password_file, "r", encoding="utf-8") as f :
+                    password=f.readline().strip()
+                os.remove(password_file)
+                password_file = MCCLI_CONFIG_DIR + contact["public_key"] + ".pass"
+                with open(password_file, "w", encoding="utf-8") as f :
+                    f.write(password)
+
+            # this is the new correct password file, using pubkey
+            password_file = MCCLI_CONFIG_DIR + contact["public_key"] + ".pass"
             if os.path.exists(password_file) :
                 with open(password_file, "r", encoding="utf-8") as f :
                     password=f.readline().strip()
@@ -993,6 +1065,9 @@ async def process_contact_chat_line(mc, contact, line):
 
     if line.startswith("forget_password") or line.startswith("fp"):
         password_file = MCCLI_CONFIG_DIR + contact['adv_name'] + ".pass"
+        if os.path.exists(password_file):
+            os.remove(password_file)
+        password_file = MCCLI_CONFIG_DIR + contact['public_key'] + ".pass"
         if os.path.exists(password_file):
             os.remove(password_file)
         try:
@@ -1112,6 +1187,7 @@ async def set_channel (mc, chan, name, key=None):
         return None
 
     info = res.payload
+    info["channel_hash"] = sha256(info["channel_secret"]).digest()[0:1].hex()
     info["channel_secret"] = info["channel_secret"].hex()
 
     if hasattr(mc,'channels') :
@@ -1129,6 +1205,16 @@ async def get_channel_by_name (mc, name):
 
     return None
 
+async def get_channel_by_hash (mc, hash):
+    if not hasattr(mc, 'channels') :
+        await_get_channels(mc)
+
+    for c in mc.channels:
+        if c['channel_hash'] == hash:
+            return c
+
+    return None
+
 async def get_contacts (mc, anim=False, lastomod=0, timeout=5) :
     if mc._contacts:
         return
@@ -1200,6 +1286,7 @@ async def get_channels (mc, anim=False) :
         if res.type == EventType.ERROR:
             break
         info = res.payload
+        info["channel_hash"] = sha256(info["channel_secret"]).digest()[0:1].hex()
         info["channel_secret"] = info["channel_secret"].hex()
         mc.channels.append(info)
         ch = ch + 1
@@ -1411,6 +1498,14 @@ async def next_cmd(mc, cmds, json_output=False):
                         process_event_message.print_snr = (cmds[2] == "on")
                         if json_output :
                             print(json.dumps({"cmd" : cmds[1], "param" : cmds[2]}))
+                    case "json_log_rx" :
+                        handle_log_rx.json_log_rx = (cmds[2] == "on")
+                        if json_output :
+                            print(json.dumps({"cmd" : cmds[1], "param" : cmds[2]}))
+                    case "log_channels" :
+                        handle_log_rx.log_channels = (cmds[2] == "on")
+                        if json_output :
+                            print(json.dumps({"cmd" : cmds[1], "param" : cmds[2]}))
                     case "print_adverts" :
                         handle_advert.print_adverts = (cmds[2] == "on")
                         if json_output :
@@ -1641,6 +1736,16 @@ async def next_cmd(mc, cmds, json_output=False):
                             print(json.dumps({"color" : process_event_message.color}))
                         else:
                             print(f"{'on' if process_event_message.color else 'off'}")
+                    case "json_log_rx":
+                        if json_output :
+                            print(json.dumps({"json_log_rx" : handle_log_rx.json_log_rx}))
+                        else:
+                            print(f"{'on' if handle_log_rx.json_log_rx else 'off'}")
+                    case "log_channels":
+                        if json_output :
+                            print(json.dumps({"log_channels" : handle_log_rx.log_channels}))
+                        else:
+                            print(f"{'on' if handle_log_rx.log_channels else 'off'}")
                     case "print_adverts":
                         if json_output :
                             print(json.dumps({"print_adverts" : handle_advert.print_adverts}))
@@ -2881,10 +2986,12 @@ async def main(argv):
     handle_message.mc = mc # connect meshcore to handle_message
     handle_advert.mc = mc
     handle_path_update.mc = mc
+    handle_log_rx.mc = mc
 
     mc.subscribe(EventType.ADVERTISEMENT, handle_advert)
     mc.subscribe(EventType.PATH_UPDATE, handle_path_update)
     mc.subscribe(EventType.NEW_CONTACT, handle_new_contact)
+    mc.subscribe(EventType.RX_LOG_DATA, handle_log_rx)
 
     mc.auto_update_contacts = True