12 print 'Content-Type: text/html\n'
13 sys.stderr = sys.stdout
14 sys.path.append('/home/ecprice/.local/lib/python2.5/site-packages')
16 from Cheetah.Template import Template
17 from sipb_xen_database import *
20 # ... and stolen from xend/uuid.py
22 """Generate a random UUID."""
24 return [ random.randint(0, 255) for _ in range(0, 16) ]
27 return "-".join(["%02x" * 4, "%02x" * 2, "%02x" * 2, "%02x" * 2,
28 "%02x" * 6]) % tuple(u)
34 def haveAccess(user, machine):
38 def error(op, user, fields, errorMessage):
41 errorMessage=errorMessage)
42 print Template(file='error.tmpl',
45 def validMachineName(name):
48 charset = string.ascii_letters + string.digits + '-'
49 if name[0] == '-' or len(name) > 22:
51 return all(x in charset for x in name)
54 keytab = '/etc/tabbott.keytab'
55 username = 'tabbott/extra'
56 p = subprocess.Popen(['kinit', "-k", "-t", keytab,
61 p = subprocess.Popen(['klist', '-s'])
67 p = subprocess.Popen(['remctl', 'black-mesa.mit.edu']
69 stdout=subprocess.PIPE,
70 stderr=subprocess.PIPE)
72 print >> sys.stderr, 'ERROR on remctl ', args
73 print >> sys.stderr, p.stderr.read()
76 remctl('lvcreate','all')
78 def bootMachine(machine, cdtype):
79 if cdtype is not None:
80 remctl('vmboot', 'cdrom', str(machine.name),
83 remctl('vmboot', 'cdrom', str(machine.name))
85 def createVm(user, name, memory, disk, is_hvm, cdrom):
86 # put stuff in the table
87 transaction = ctx.current.create_transaction()
89 res = meta.engine.execute('select nextval(\'"machines_machine_id_seq"\')')
90 id = res.fetchone()[0]
92 machine.machine_id = id
94 machine.memory = memory
95 machine.owner = user.username
96 machine.contact = user.email
97 machine.uuid = uuidToString(randomUUID())
98 machine.boot_off_cd = True
99 machine_type = Type.get_by(hvm=is_hvm)
100 machine.type_id = machine_type.type_id
101 ctx.current.save(machine)
102 disk = Disk(machine.machine_id,
104 open = NIC.select_by(machine_id=None)
105 if not open: #No IPs left!
106 return "No IP addresses left! Contact sipb-xen-dev@mit.edu"
108 nic.machine_id = machine.machine_id
110 ctx.current.save(nic)
111 ctx.current.save(disk)
114 transaction.rollback()
117 # tell it to boot with cdrom
118 bootMachine(machine, cdrom)
122 def create(user, fields):
123 name = fields.getfirst('name')
124 if not validMachineName(name):
125 return error('create', user, fields,
126 "Invalid name '%s'" % name)
129 if Machine.get_by(name=name):
130 return error('create', user, fields,
131 "A machine named '%s' already exists" % name)
133 memory = fields.getfirst('memory')
139 return error('create', user, fields,
140 "Invalid memory amount")
141 if memory > maxMemory(user):
142 return error('create', user, fields,
143 "Too much memory requested")
145 disk = fields.getfirst('disk')
148 disk = int(disk * 1024)
152 return error('create', user, fields,
153 "Invalid disk amount")
155 vm_type = fields.getfirst('vmtype')
156 if vm_type not in ('hvm', 'paravm'):
157 return error('create', user, fields,
158 "Invalid vm type '%s'" % vm_type)
159 is_hvm = (vm_type == 'hvm')
161 cdrom = fields.getfirst('cdrom')
162 if cdrom is not None and not CDROM.get(cdrom):
163 return error('create', user, fields,
164 "Invalid cdrom type '%s'" % cdrom)
166 machine = createVm(user, name, memory, disk, is_hvm, cdrom)
167 if isinstance(machine, basestring):
168 return error('create', user, fields,
172 print Template(file='create.tmpl',
175 def listVms(user, fields):
176 machines = Machine.select()
179 cdroms=CDROM.select())
181 print Template(file='list.tmpl', searchList=d)
183 def testMachineId(user, machineId, exists=True):
184 if machineId is None:
185 error('vnc', user, fields,
186 "No machine ID specified")
189 machineId = int(machineId)
191 error('vnc', user, fields,
192 "Invalid machine ID '%s'"
195 machine = Machine.get(machineId)
196 if exists and machine is None:
197 error('vnc', user, fields,
198 "No such machine ID '%s'"
201 if not haveAccess(user, machine):
202 error('vnc', user, fields,
203 "No access to machine ID '%s'"
208 def vnc(user, fields):
209 machine = testMachineId(user, fields.getfirst('machine_id'))
210 if machine is None: #gave error page already
213 TOKEN_KEY = "0M6W0U1IXexThi5idy8mnkqPKEq1LtEnlK/pZSn0cDrN"
217 data["machine"]=machine
218 data["expires"]=time.time()+(5*60)
219 pickledData = cPickle.dumps(data)
220 m = hmac.new(TOKEN_KEY, digestmod=sha)
221 m.update(pickledData)
222 token = {'data': pickledData, 'digest': m.digest()}
223 token = cPickle.dumps(token)
224 token = base64.urlsafe_b64encode(token)
228 hostname='localhost',
230 print Template(file='vnc.tmpl',
233 def info(user, fields):
234 machine = testMachineId(user, fields.getfirst('machine_id'))
235 if machine is None: #gave error page already
240 print Template(file='info.tmpl',
243 mapping = dict(list=listVms,
248 if __name__ == '__main__':
249 fields = cgi.FieldStorage()
252 email = 'moo@cow.com'
254 connect('postgres://sipb-xen@sipb-xen-dev/sipb_xen')
255 operation = os.environ.get('PATH_INFO', '')
256 if operation.startswith('/'):
257 operation = operation[1:]
261 fun = mapping.get(operation,
263 error(operation, u, e,
264 "Invalid operation '%'" % operation))