swisskyrepo
/
NetExec
mirror of https://github.com/swisskyrepo/NetExec.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Closes Issue #697. Added a flag to append self.host to the filename of the retrieved file from the get-file option.

main
twosevenzero 2023-03-21 12:53:54 -04:00 committed by Marshall Hallenbeck
parent 433107ba6d
commit 8233c5bf48
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
cme/protocols/smb.py
View File

@ -227,10 +227,9 @@ class smb(connection):
sgroup.add_argument("--only-files", action='store_true', help='only spider files')
tgroup = smb_parser.add_argument_group("Files", "Options for put and get remote files")
tgroup.add_argument("--put-file", nargs=2, metavar="FILE",
help='Put a local file into remote target, ex: whoami.txt \\\\Windows\\\\Temp\\\\whoami.txt')
tgroup.add_argument("--get-file", nargs=2, metavar="FILE",
help='Get a remote file, ex: \\\\Windows\\\\Temp\\\\whoami.txt whoami.txt')
tgroup.add_argument("--put-file", nargs=2, metavar="FILE", help='Put a local file into remote target, ex: whoami.txt \\\\Windows\\\\Temp\\\\whoami.txt')
tgroup.add_argument("--get-file", nargs=2, metavar="FILE", help='Get a remote file, ex: \\\\Windows\\\\Temp\\\\whoami.txt whoami.txt')
tgroup.add_argument("--append-host", action='store_true', help='append the host to the get-file filename')
cgroup = smb_parser.add_argument_group("Command Execution", "Options for executing commands")
cgroup.add_argument('--exec-method', choices={"wmiexec", "mmcexec", "smbexec", "atexec"}, default=None,
@ -1410,10 +1409,13 @@ class smb(connection):
def get_file(self):
self.logger.info('Copy {} to {}'.format(self.args.get_file[0], self.args.get_file[1]))
with open(self.args.get_file[1], 'wb+') as file:
file_handle = self.args.get_file[1]
if self.args.append_host:
file_handle = self.args.get_file[1] + "_" + self.host
with open(file_handle, 'wb+') as file:
try:
self.conn.getFile(self.args.share, self.args.get_file[0], file.write)
self.logger.success(f"File {self.args.get_file[0]} was transferred to {self.args.get_file[1]}")
self.logger.success('File {} was transferred to {}'.format(self.args.get_file[0], file_handle))
except Exception as e:
self.logger.error('Error reading file {}: {}'.format(self.args.share, e))