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  import os 
23  from twisted.internet import defer 
24   
25  from flumotion.component.plugs import base, lifecycle 
26  from flumotion.common.manhole import openSSHManhole 
27  from flumotion.common.manhole import openAnonymousTelnetManhole 
28   
29  __version__ = "$Rev: 6125 $" 
30   
31   
32 -class ManagerManholePlug(lifecycle.ManagerLifecycle): 
33      """ 
34      """ 
35 -    def __init__(self, args): 
36          lifecycle.ManagerLifecycle.__init__(self, args) 
37          self.useSSH = False 
38          self.authorizedKeysFile = None 
39          self.port = None 
40          self.requestedPortNum = None 
41   
42 -    def start(self, vishnu): 
43          self.namespace = {'vishnu': vishnu} 
44   
45          props = self.args['properties'] 
46          if 'ssh-authorized-keys-file' in props: 
47              self.useSSH = True 
48              self.authorizedKeysFile = os.path.expanduser( 
49                  props['ssh-authorized-keys-file']) 
50   
51          self.requestedPortNum = props.get('port', -1) 
52   
53          self._insinuate() 
54   
55          if props.get('initially-open', False): 
56              self.openManhole() 
57   
58 -    def stop(self, vishnu): 
59          self.closeManhole() 
60   
61 -    def _insinuate(self): 
62          # "And I wish you didn't have the devil's curly hair!" 
63          from flumotion.manager.admin import AdminAvatar 
64          AdminAvatar.perspective_openManhole = self.openManhole 
65          AdminAvatar.perspective_closeManhole = self.closeManhole 
66   
67 -    def openManhole(self): 
68          if not self.port: 
69              if self.useSSH: 
70                  self.port = openSSHManhole(self.authorizedKeysFile, 
71                                             self.namespace, 
72                                             self.requestedPortNum) 
73              else: 
74                  self.port = openAnonymousTelnetManhole(self.namespace, 
75                                                         self.requestedPortNum) 
76   
77          return self.port.getHost().port 
78   
79 -    def closeManhole(self): 
80          if self.port: 
81              ret = self.port.loseConnection() 
82          else: 
83              ret = defer.succeed(None) 
84          self.port = None 
85          return ret 
86