infosecn1nja
/
Egress-Assess
mirror of https://github.com/infosecn1nja/Egress-Assess.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed issue #51

master
Chris Truncer 2018-02-21 09:41:31 -07:00
parent 62ab4d30cd
commit afe7cba203
No known key found for this signature in database
GPG Key ID: 9715AC2B40A5A64A
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
protocols/clients/smb_client.py
View File

@ -27,9 +27,10 @@ class Client:
self.file_transfer = False self.file_transfer = False
else: else:
if "/" in cli_object.file: if "/" in cli_object.file:
self.file_transfer = cli_object.file.split("/")[-1]
else:
self.file_transfer = cli_object.file self.file_transfer = cli_object.file
self.file_name = cli_object.file.split("/")[-1]
else:
self.file_name = cli_object.file
# transmit is the only required function within the object. It is what # transmit is the only required function within the object. It is what
# called by the framework to transmit data. However, you can create as # called by the framework to transmit data. However, you can create as
@ -49,7 +50,7 @@ class Client:
os.makedirs(mount_path) os.makedirs(mount_path)
# Base command to copy file over smb # Base command to copy file over smb
smb_command = "smbclient \\\\\\\\" + self.remote_server + "\\\\\\DATA -N -c \"put " smb_command = "smbclient \\\\\\\\" + self.remote_server + "\\\\DATA -N -c \"put "
# If using a file, copy it, else write to disk and then copy # If using a file, copy it, else write to disk and then copy
if not self.file_transfer: if not self.file_transfer:
@ -59,7 +60,7 @@ class Client:
smb_command += smb_file_name + "\"" smb_command += smb_file_name + "\""
else: else:
smb_command += self.file_transfer + "\"" smb_command += self.file_transfer + " " + self.file_name + "\""
smb_file_name = self.file_transfer smb_file_name = self.file_transfer
print smb_command print smb_command