Move invirt-console/files to invirt-console/server
[invirt/packages/invirt-console.git] / server / usr / bin / invirt-consolefs
1 #!/usr/bin/python
2
3 import routefs
4 from routes import Mapper
5
6 from syslog import *
7 from time import time
8
9 import os
10 import errno
11
12 from invirt.config import structs as config
13 from invirt import database
14
15 realpath = "/home/machines/"
16
17 class ConsoleFS(routefs.RouteFS):
18     """
19     ConsoleFS creates a series of subdirectories each mirroring the same real
20     directory, except for a single file - the .k5login - which is dynamically
21     generated for each subdirectory
22     """
23     
24     def __init__(self, *args, **kw):
25         """Initialize the filesystem and set it to allow_other access besides
26         the user who mounts the filesystem (i.e. root)
27         """
28         super(ConsoleFS, self).__init__(*args, **kw)
29         self.lasttime = 0
30         self.machines = []
31         self.fuse_args.add("allow_other", True)
32         
33         openlog('invirt-consolefs ', LOG_PID, LOG_DAEMON)
34         
35         syslog(LOG_DEBUG, 'Init complete.')
36     
37     def make_map(self):
38         m = Mapper()
39         m.connect('', controller='getMachines')
40         m.connect(':machine', controller='getMirror')
41         m.connect(':machine/.k5login', controller='getK5login')
42         m.connect(':machine/*(path)', controller='getMirror')
43         return m
44     
45     def recache(self):
46         if time() - self.lasttime > 5:
47             self.lasttime = time()
48             database.clear_cache()
49             self.machines = dict((machine.name, machine) for machine in database.session.query(database.Machine).all())
50
51     def getMachines(self, **kw):
52         self.recache()
53         return self.machines.keys()
54     
55     def getMirror(self, machine, path='', **kw):
56         """Translate the path into its realpath equivalent, and return that
57         """
58         real = realpath + path
59         if os.path.isdir(real):
60             # The list is converted to a set so that we can handle the case 
61             # where there is already a .k5login in the realpath gracefully    
62             return routefs.Directory(set(os.listdir(real) + ['.k5login']))
63         elif os.path.islink(real):
64             return routefs.Symlink(os.readlink(real))
65         elif os.path.isfile(real):
66             return open(real).read()
67         else:
68             return -errno.EINVAL
69     
70     def getK5login(self, machine, **kw):
71         self.recache()
72         machine = self.machines[machine]
73         users = [acl.user for acl in machine.acl]
74         return "\n".join(map(self.userToPrinc, users) + [''])
75     
76     def mirrorPath(self, path):
77         """Translate a virtual path to its real path counterpart"""
78         return realpath + "/".join(getParts(path)[1:])
79     
80     def userToPrinc(self, user):
81         """Convert Kerberos v4-style names to v5-style and append a default
82         realm if none is specified
83         """
84         if '@' in user:
85             (princ, realm) = user.split('@')
86         else:
87             princ = user
88             realm = config.authn[0].realm
89         
90         return princ.replace('.', '/') + '@' + realm
91
92 if __name__ == '__main__':
93     database.connect()
94     routefs.main(ConsoleFS)