Package flumotion :: Package component :: Package plugs :: Module manhole
[hide private]

Source Code for Module flumotion.component.plugs.manhole

 1  # -*- Mode: Python -*- 
 2  # vi:si:et:sw=4:sts=4:ts=4 
 3  # 
 4  # Flumotion - a streaming media server 
 5  # Copyright (C) 2006 Fluendo, S.L. (www.fluendo.com). 
 6  # All rights reserved. 
 7   
 8  # This file may be distributed and/or modified under the terms of 
 9  # the GNU General Public License version 2 as published by 
10  # the Free Software Foundation. 
11  # This file is distributed without any warranty; without even the implied 
12  # warranty of merchantability or fitness for a particular purpose. 
13  # See "LICENSE.GPL" in the source distribution for more information. 
14   
15  # Licensees having purchased or holding a valid Flumotion Advanced 
16  # Streaming Server license may use this file in accordance with the 
17  # Flumotion Advanced Streaming Server Commercial License Agreement. 
18  # See "LICENSE.Flumotion" in the source distribution for more information. 
19   
20  # Headers in this file shall remain intact. 
21   
22   
23  import os 
24  from twisted.internet import defer 
25   
26  from flumotion.component.plugs import base, lifecycle 
27  from flumotion.common.manhole import openSSHManhole 
28  from flumotion.common.manhole import openAnonymousTelnetManhole 
29   
30   
31 -class ManagerManholePlug(lifecycle.ManagerLifecycle):
32 """ 33 """
34 - def __init__(self, args):
35 lifecycle.ManagerLifecycle.__init__(self, args) 36 self.useSSH = False 37 self.authorizedKeysFile = None 38 self.port = None 39 self.requestedPortNum = None
40
41 - def start(self, vishnu):
42 self.namespace = {'vishnu': vishnu} 43 44 props = self.args['properties'] 45 if 'ssh-authorized-keys-file' in props: 46 self.useSSH = True 47 self.authorizedKeysFile = os.path.expanduser( 48 props['ssh-authorized-keys-file']) 49 50 self.requestedPortNum = props.get('port', -1) 51 52 self._insinuate() 53 54 if props.get('initially-open', False): 55 self.openManhole()
56
57 - def stop(self, vishnu):
58 self.closeManhole()
59
60 - def _insinuate(self):
61 # "And I wish you didn't have the devil's curly hair!" 62 from flumotion.manager.admin import AdminAvatar 63 AdminAvatar.perspective_openManhole = self.openManhole 64 AdminAvatar.perspective_closeManhole = self.closeManhole
65
66 - def openManhole(self):
67 if not self.port: 68 if self.useSSH: 69 self.port = openSSHManhole(self.authorizedKeysFile, 70 self.namespace, 71 self.requestedPortNum) 72 else: 73 self.port = openAnonymousTelnetManhole(self.namespace, 74 self.requestedPortNum) 75 76 return self.port.getHost().port
77
78 - def closeManhole(self):
79 if self.port: 80 ret = self.port.loseConnection() 81 else: 82 ret = defer.succeed(None) 83 self.port = None 84 return ret
85