1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#! /usr/bin/env python
"""This script displays locally the graphs that are built by dotviewer
on remote machines.
Usage:
sshgraphserver.py hostname [more args for ssh...]
sshgraphserver.py LOCAL
This logs in to 'hostname' by passing the arguments on the command-line
to ssh. No further configuration is required: it works for all programs
using the dotviewer library as long as they run on 'hostname' under the
same username as the one sshgraphserver logs as.
If 'hostname' is the string 'LOCAL', then it starts locally without ssh.
"""
import graphserver, socket, subprocess, random
def ssh_graph_server(sshargs):
s1 = socket.socket()
s1.bind(('127.0.0.1', socket.INADDR_ANY))
localhost, localport = s1.getsockname()
if sshargs[0] != 'LOCAL':
remoteport = random.randrange(10000, 20000)
# ^^^ and just hope there is no conflict
args = ['ssh', '-S', 'none', '-C', '-R%d:127.0.0.1:%d' % (
remoteport, localport)]
args = args + sshargs + ['python -u -c "exec input()"']
else:
remoteport = localport
args = ['python', '-u', '-c', 'exec input()']
print ' '.join(args)
p = subprocess.Popen(args, bufsize=0,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE)
p.stdin.write(repr('port=%d\n%s' % (remoteport, REMOTE_SOURCE)) + '\n')
line = p.stdout.readline()
assert line == 'OK\n'
graphserver.listen_server(None, s1=s1)
REMOTE_SOURCE = r"""
import tempfile, getpass, os, sys
def main(port):
tmpdir = tempfile.gettempdir()
user = getpass.getuser()
fn = os.path.join(tmpdir, 'dotviewer-sshgraphsrv-%s' % user)
try:
os.unlink(fn)
except OSError:
pass
f = open(fn, 'w')
print >> f, port
f.close()
try:
sys.stdout.write('OK\n')
# just wait for the loss of the remote link, ignoring any data
while sys.stdin.read(1024):
pass
finally:
try:
os.unlink(fn)
except OSError:
pass
main(port)
"""
if __name__ == '__main__':
import sys
if len(sys.argv) <= 1:
print __doc__
sys.exit(2)
ssh_graph_server(sys.argv[1:])
|
