Merge pull request #24 from p0dalirius/add-lcat-lbat-commands

[enhancement] Add lcat lbat commands, fixing #22 #23
p0dalirius · Jun 7, 2024 · bd0b28b · bd0b28b
2 parents 8136239 + 44a55e5
commit bd0b28b
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 2 deletions.
diff --git a/smbclientng/core/CommandCompleter.py b/smbclientng/core/CommandCompleter.py
@@ -28,14 +28,14 @@ class CommandCompleter(object):
     commands = {
         "bat": {
             "description": [
-                "Pretty prints the contents of a file.", 
+                "Pretty prints the contents of a remote file.", 
                 "Syntax: 'bat <file>'"
             ], 
             "subcommands": []
         },
         "cat": {
             "description": [
-                "Get the contents of a file.", 
+                "Get the contents of a remote file.", 
                 "Syntax: 'cat <file>'"
             ], 
             "subcommands": []
@@ -103,6 +103,20 @@ class CommandCompleter(object):
             ], 
             "subcommands": ["server", "share"]
         },
+        "lbat": {
+            "description": [
+                "Pretty prints the contents of a local file.", 
+                "Syntax: 'lbat <file>'"
+            ], 
+            "subcommands": []
+        },
+        "lcat": {
+            "description": [
+                "Get the contents of a local file.", 
+                "Syntax: 'lcat <file>'"
+            ], 
+            "subcommands": []
+        },
         "lcd": {
             "description": [
                 "Changes the current local directory.",

diff --git a/smbclientng/core/InteractiveShell.py b/smbclientng/core/InteractiveShell.py
@@ -174,6 +174,10 @@ def process_command(self, command, arguments=[]):
         elif command == "put":
             self.command_put(arguments, command)
 
+        # Shows the content of a local file
+        elif command == "lcat":
+            self.command_lcat(arguments, command)
+
         # Changes the current local directory
         elif command == "lcd":
             self.command_lcd(arguments, command)
@@ -182,6 +186,10 @@ def process_command(self, command, arguments=[]):
         elif command == "lcp":
             self.command_lcp(arguments, command)
 
+        # Pretty prints the content of a local file
+        elif command == "lbat":
+            self.command_lbat(arguments, command)
+
         # Lists the contents of the current local directory
         elif command == "lls":
             self.command_lls(arguments, command)
@@ -383,6 +391,29 @@ def command_info(self, arguments, command):
         except impacket.smbconnection.SessionError as e:
             print("[!] SMB Error: %s" % e)
 
+    @command_arguments_required
+    def command_lcat(self, arguments, command):
+        # Command arguments required   : Yes
+        # Active SMB connection needed : No
+        # SMB share needed             : No
+
+        path = ' '.join(arguments)
+        try:
+            if os.path.exists(path=path):
+                f = open(path, 'rb')
+                rawcontents = f.read()
+                if rawcontents is not None:
+                    encoding = charset_normalizer.detect(rawcontents)["encoding"]
+                    if encoding is not None:
+                        filecontent = rawcontents.decode(encoding).rstrip()
+                        print(filecontent)
+                    else:
+                        print("[!] Could not detect charset of '%s'." % path)
+            else:
+                print("[!] Local file '%s' does not exist." % path)
+        except impacket.smbconnection.SessionError as e:
+            print("[!] SMB Error: %s" % e)
+
     @command_arguments_required
     def command_lcd(self, arguments, command):
         # Command arguments required   : Yes
@@ -417,6 +448,37 @@ def command_lcp(self, arguments, command):
         else:
             self.commandCompleterObject.print_help(command=command)
 
+    @command_arguments_required
+    def command_lbat(self, arguments, command):
+        # Command arguments required   : Yes
+        # Active SMB connection needed : No
+        # SMB share needed             : No
+
+        path = ' '.join(arguments)
+        try:
+            if os.path.exists(path=path):
+                f = open(path, 'rb')
+                rawcontents = f.read()
+                if rawcontents is not None:
+                    encoding = charset_normalizer.detect(rawcontents)["encoding"]
+                    if encoding is not None:
+                        filecontent = rawcontents.decode(encoding).rstrip()
+                        lexer = Syntax.guess_lexer(path=ntpath.basename(path), code=filecontent)
+                        # Some trickery for the files undetected by the lexer
+                        if lexer == "default":
+                            if '<?xml' in filecontent:
+                                lexer = "xml"
+                            elif '<html>' in filecontent:
+                                lexer = "html"
+                        syntax = Syntax(code=filecontent, line_numbers=True, lexer=lexer)
+                        Console().print(syntax)
+                    else:
+                        print("[!] Could not detect charset of '%s'." % path)
+            else:
+                print("[!] Local file '%s' does not exist." % path)
+        except impacket.smbconnection.SessionError as e:
+            print("[!] SMB Error: %s" % e)
+
     def command_lls(self, arguments, command):
         # Command arguments required   : No
         # Active SMB connection needed : No