dorian
/
paramiko
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Post-import edits

This commit is contained in:
Jeff Forcier 2012-11-05 17:26:47 -08:00
parent 7cd2f2715b
commit 27271fa455
1 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
paramiko/proxy.py
View File

@ -1,46 +1,48 @@
# Copyright (C) 2012 Yipit, Inc <coders@yipit.com> # Copyright (C) 2012 Yipit, Inc <coders@yipit.com>
# #
# This file is part of ssh. # This file is part of paramiko.
# #
# Paramiko is free software; you can redistribute it and/or modify it under the # Paramiko is free software; you can redistribute it and/or modify it under the
# terms of the GNU Lesser General Public License as published by the Free # terms of the GNU Lesser General Public License as published by the Free
# Software Foundation; either version 2.1 of the License, or (at your option) # Software Foundation; either version 2.1 of the License, or (at your option)
# any later version. # any later version.
# #
# 'ssh' is distrubuted in the hope that it will be useful, but WITHOUT ANY # Paramiko is distrubuted in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more # A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
# details. # details.
# #
# You should have received a copy of the GNU Lesser General Public License # You should have received a copy of the GNU Lesser General Public License
# along with 'ssh'; if not, write to the Free Software Foundation, Inc., # along with Paramiko; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Suite 500, Boston, MA 02110-1335 USA. # 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
"""
L{ProxyCommand}.
"""
import os import os
from shlex import split as shlsplit from shlex import split as shlsplit
from subprocess import Popen, PIPE from subprocess import Popen, PIPE
from ssh.ssh_exception import BadProxyCommand from ssh.ssh_exception import ProxyCommandFailure
class ProxyCommand(object): class ProxyCommand(object):
""" """
A proxy based on the program informed in the ProxyCommand config. Wraps a subprocess running ProxyCommand-driven programs.
This class implements a the interface needed by both L{Transport} This class implements a the socket-like interface needed by the
and L{Packetizer} classes. Using this class instead of a regular L{Transport} and L{Packetizer} classes. Using this class instead of a
socket makes it possible to talk with a popened command that will regular socket makes it possible to talk with a Popen'd command that will
proxy all the conversation between the client and a server hosted in proxy traffic between the client and a server hosted in another machine.
another machine.
""" """
def __init__(self, command_line): def __init__(self, command_line):
""" """
Create a new CommandProxy instance. The instance created by this Create a new CommandProxy instance. The instance created by this
class should be passed as argument to the L{Transport} class. class can be passed as an argument to the L{Transport} class.
@param command_line: the command that should be executed and @param command_line: the command that should be executed and
used as the proxy. used as the proxy.
@type command_line: str @type command_line: str
""" """
self.cmd = shlsplit(command_line) self.cmd = shlsplit(command_line)
@ -56,31 +58,33 @@ class ProxyCommand(object):
""" """
try: try:
self.process.stdin.write(content) self.process.stdin.write(content)
except IOError, exc: except IOError, e:
# There was a problem with the child process. It probably # There was a problem with the child process. It probably
# died and we can't proceed. The best option here is to # died and we can't proceed. The best option here is to
# raise an exception informing the user that the informed # raise an exception informing the user that the informed
# ProxyCommand is not working. # ProxyCommand is not working.
raise BadProxyCommand(' '.join(self.cmd), exc.strerror) raise BadProxyCommand(' '.join(self.cmd), e.strerror)
return len(content) return len(content)
def recv(self, size): def recv(self, size):
""" """
Read from the standard output of the forked program Read from the standard output of the forked program.
@param size: how many chars should be read @param size: how many chars should be read
@type size: int @type size: int
@return: the length of the readed content @return: the length of the read content
@rtype: int @rtype: int
""" """
try: try:
return os.read(self.process.stdout.fileno(), size) return os.read(self.process.stdout.fileno(), size)
except IOError, exc: except IOError, e:
raise BadProxyCommand(' '.join(self.cmd), exc.strerror) raise BadProxyCommand(' '.join(self.cmd), e.strerror)
def close(self): def close(self):
self.process.terminate() self.process.terminate()
def settimeout(self, timeout): def settimeout(self, timeout):
# Timeouts are meaningless for this implementation, but are part of the
# spec, so must be present.
pass pass