drunkendog
/
lxd-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add MITM ssh logic

os-repl
root 2022-11-25 16:08:10 +00:00
parent c87598f737
commit 18908d3f64
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
sshim_patch.py
View File

@ -82,17 +82,21 @@ class Runner(threading.Thread):
def __init__(self, client, channel: paramiko.Channel): def __init__(self, client, channel: paramiko.Channel):
threading.Thread.__init__(self, name='sshim.Runner(%s)' % channel.chanid) threading.Thread.__init__(self, name='sshim.Runner(%s)' % channel.chanid)
self.instance_name = "instance-" + str(uuid.uuid4()) self.instance_name = "instance-" + str(uuid.uuid4())
self.instance_password = str(uuid.uuid4()) # TODO: secure password generation
self.daemon = True self.daemon = True
self.client = client self.client = client
self.channel = channel self.channel = channel
self.channel.settimeout(None) self.channel.settimeout(None)
def run(self) -> None: def run(self) -> None:
lxd_interface.create_instance(self.instance_name) vm_ip = lxd_interface.create_instance(self.instance_name)
with paramiko.ProxyCommand(command_line=f'lxc exec {self.instance_name} -- /bin/bash') as proxy_command: with paramiko.SSHClient() as ssh_client:
self.channel.recv = proxy_command.recv ssh_client.connect(vm_ip, username='root', passphrase=self.instance_password)
self.channel.send = proxy_command.send tmp_channel = ssh_client.invoke_shell()
for attribute in dir(tmp_channel):
setattr(self.channel, attribute, getattr(tmp_channel, attribute))
Script.expect = expect Script.expect = expect