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
|
#!/usr/bin/env python
# -*- coding: ISO-8859-15 -*-
#
# Copyright (C) 2005-2007 David Guerizec <david@guerizec.net>
#
# Last modified: 2007 Oct 14, 05:43:25 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
import os.path
from sshproxy import get_class
from sshproxy.config import get_config, ConfigSection, path
from sshproxy import keys
from sshproxy.proxy import ProxyShell
from sshproxy.backend import Backend
class LogUsersConfigSection(ConfigSection):
section_id = 'logusers'
section_defaults = {
'logdir': '@logusers',
}
types = {
'logdir': path,
}
LogUsersConfigSection.register()
ProxyShell = get_class('ProxyShell')
class LoggedProxyShell(ProxyShell):
tr_table = {}
_tr_table = {
'\r\n': '\n',
'\r': '\n',
'\n': '\n',
'<': '<INF>',
'>': '<SUP>',
}
def __reginit__(self, *args, **kw):
conf = get_config('logusers')
if not os.path.isdir(conf['logdir']):
os.makedirs(conf['logdir'])
self.logdir = conf['logdir']
# fill our translation table
for key in dir(keys):
if key[0] == '_' or not isinstance(getattr(keys, key), str):
continue
self.tr_table[getattr(keys, key)] = '<%s>' % key
for key, value in self._tr_table.items():
self.tr_table[key] = value
ProxyShell.__reginit__(self, *args, **kw)
user = self.ipc_chan.call('get_ns_tag', namespace='client',
tag='username')
path = os.path.join(self.logdir, user)
if not os.path.isdir(path):
os.makedirs(path)
site_login = self.ipc_chan.call('get_ns_tag', namespace='site',
tag='login')
site_name = self.ipc_chan.call('get_ns_tag', namespace='site',
tag='name')
site = '%s@%s' % (site_login, site_name)
logfile = os.path.join(path, site)
self.log = open(logfile, 'a')
def client_recv_data(self, source, name):
data = ProxyShell.recv_data(self, source, name)
#if name == 'client_chan':
if True:
for x in data:
self.log.write(self.translate(x))
self.log.flush()
return data
def copy_client(self, source, event, destination,
recv_data=None, send_data=None):
return self.channel_copy(source, event, destination,
recv_data=LoggedProxyShell.client_recv_data)
def __del__(self):
self.log.close()
ProxyShell.__del__(self)
def translate(self, char):
return self.tr_table.get(char, char)
|
