mirror of https://github.com/jumpserver/jumpserver
140 lines
3.6 KiB
Python
Executable File
140 lines
3.6 KiB
Python
Executable File
#coding: utf-8
|
|
|
|
import socket
|
|
import sys
|
|
import os
|
|
import select
|
|
import time
|
|
import paramiko
|
|
import sys
|
|
import struct
|
|
import fcntl
|
|
import signal
|
|
|
|
try:
|
|
import termios
|
|
import tty
|
|
except ImportError:
|
|
print 'Only postfix supported.'
|
|
sys.exit()
|
|
|
|
|
|
CURRENT_DIR = os.path.abspath('.')
|
|
LOG_DIR = os.path.join(CURRENT_DIR, 'logs')
|
|
|
|
|
|
def get_win_size():
|
|
"""This function use to get the size of the windows!"""
|
|
if 'TIOCGWINSZ' in dir(termios):
|
|
TIOCGWINSZ = termios.TIOCGWINSZ
|
|
else:
|
|
TIOCGWINSZ = 1074295912L # Assume
|
|
s = struct.pack('HHHH', 0, 0, 0, 0)
|
|
x = fcntl.ioctl(sys.stdout.fileno(), TIOCGWINSZ, s)
|
|
return struct.unpack('HHHH', x)[0:2]
|
|
|
|
|
|
def set_win_size(sig, data):
|
|
"""This function use to set the window size of the terminal!"""
|
|
try:
|
|
win_size = get_win_size()
|
|
channel.resize_pty(height=win_size[0], width=win_size[1])
|
|
except:
|
|
pass
|
|
|
|
|
|
def posix_shell(chan, user, host):
|
|
"""
|
|
Use paramiko channel connect server and logging.
|
|
"""
|
|
connect_log_dir = os.path.join(LOG_DIR, 'connect')
|
|
today = time.strftime('%Y%m%d')
|
|
date_now = time.strftime('%Y%m%d%H%M%S')
|
|
today_connect_log_dir = os.path.join(connect_log_dir, today)
|
|
log_filename = '%s_%s_%s.log' % (user, host, date_now)
|
|
log_file_path = os.path.join(today_connect_log_dir, log_filename)
|
|
|
|
if not os.path.isdir(today_connect_log_dir):
|
|
try:
|
|
os.makedirs(today_connect_log_dir)
|
|
except OSError:
|
|
print 'Create %s failed, Please modify %s permission.' % (today_connect_log_dir, connect_log_dir)
|
|
sys.exit()
|
|
|
|
try:
|
|
log = open(log_file_path, 'a')
|
|
except IOError:
|
|
print 'Create logfile failed, Please modify %s permission.' % today_connect_log_dir
|
|
sys.exit()
|
|
|
|
old_tty = termios.tcgetattr(sys.stdin)
|
|
try:
|
|
tty.setraw(sys.stdin.fileno())
|
|
tty.setcbreak(sys.stdin.fileno())
|
|
chan.settimeout(0.0)
|
|
|
|
while True:
|
|
try:
|
|
r, w, e = select.select([chan, sys.stdin], [], [])
|
|
except:
|
|
pass
|
|
|
|
if chan in r:
|
|
try:
|
|
x = chan.recv(1024)
|
|
if len(x) == 0:
|
|
break
|
|
sys.stdout.write(x)
|
|
sys.stdout.flush()
|
|
log.write(x)
|
|
log.flush()
|
|
except socket.timeout:
|
|
pass
|
|
|
|
if sys.stdin in r:
|
|
x = os.read(sys.stdin.fileno(), 1)
|
|
if len(x) == 0:
|
|
break
|
|
chan.send(x)
|
|
|
|
finally:
|
|
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_tty)
|
|
|
|
|
|
def connect(username, password, host, port):
|
|
"""
|
|
Connect server.
|
|
"""
|
|
|
|
ps1 = '[\u@%s \W]\$' % host
|
|
|
|
# Make a ssh connection
|
|
ssh = paramiko.SSHClient()
|
|
ssh.load_system_host_keys()
|
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
ssh.connect(host, port=port, username=username, password=password, compress=True)
|
|
|
|
# Make a channel
|
|
global channel
|
|
channel = ssh.invoke_shell()
|
|
try:
|
|
signal.signal(signal.SIGWINCH, set_win_size)
|
|
except:
|
|
pass
|
|
|
|
# Modify PS1
|
|
channel.send('PS1=%s' % ps1)
|
|
|
|
# Make ssh interactive tunnel
|
|
posix_shell(channel, username, host)
|
|
|
|
# shutdown channel socket
|
|
ssh.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
connect('guanghongwei', 'Lov@j1ax1n', '172.16.1.122', 2001)
|
|
|
|
|
|
|