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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/usr/bin/env python
# -*- coding: ISO-8859-15 -*-
#
# Copyright (C) 2005-2007 David Guerizec <david@guerizec.net>
#
# Last modified: 2007 Oct 14, 01:00:05 by david
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
from sshproxy import get_class
from sshproxy.util import chanfmt
pssh = """#!/bin/bash
PROXY_PORT=${SSHPROXY_PORT:-%(port)d}
PROXY_HOST=${SSHPROXY_HOST:-%(ip_addr)s}
PROXY_USER=${SSHPROXY_USER:-%(user)s}
[ -n "$verbose" ] && echo ssh -tp $PROXY_PORT $PROXY_USER@$PROXY_HOST -- "$@"
exec ssh -tp $PROXY_PORT $PROXY_USER@$PROXY_HOST -- "$@"
"""
pscp = """#!/bin/bash
PROXY_PORT=${SSHPROXY_PORT:-%(port)s}
PROXY_HOST=${SSHPROXY_HOST:-%(ip_addr)s}
PROXY_USER=${SSHPROXY_USER:-%(user)s}
REMOTE=$PROXY_USER@$PROXY_HOST
OPTS=( )
remote_set=
args=( )
while [ $# -gt 0 ]; do
case "$1" in
-*)
if [ "$1" = "-d" ]; then
verbose=true
else
OPTS=( "${OPTS[@]}" "$1" )
fi
;;
*:*)
if [ -z "$remote_set" ]; then
args=( "${args[@]}" "$REMOTE:$1" )
remote_set=yes
else
echo "Cannot have two remote locations"
exit 1
fi
;;
*)
args=( "${args[@]}" "$1" )
;;
esac
shift
done
[ -n "$verbose" ] && echo scp -oPort=$PROXY_PORT "${OPTS[@]}" "${args[@]}"
exec scp -oPort=$PROXY_PORT "${OPTS[@]}" "${args[@]}"
"""
base_class = get_class('Server')
class Server(base_class):
def add_cmdline_options(self, parser):
base_class.add_cmdline_options(self, parser)
parser.add_option("", "--get-pssh", dest="action",
help="display pssh client script.",
action="store_const",
const="get_pssh",
)
parser.add_option("", "--get-pscp", dest="action",
help="display pscp client script.",
action="store_const",
const="get_pscp",
)
def opt_get_pssh(self, options, *args):
user = self.pwdb.get_client().username
ip_addr, port = self.ip_addr, self.port
self.chan.send(pssh % locals())
def opt_get_pscp(self, options, *args):
user = self.pwdb.get_client().username
ip_addr, port = self.ip_addr, self.port
self.chan.send(pscp % locals())
Server.register()
|
