mirror of
https://github.com/drasko/codezero.git
synced 2026-01-12 02:43:15 +01:00
221 lines
6.8 KiB
Python
221 lines
6.8 KiB
Python
#! /usr/bin/env python2.6
|
|
# -*- mode: python; coding: utf-8; -*-
|
|
import os, sys, shelve, shutil, re
|
|
from projpaths import *
|
|
from lib import *
|
|
from caps import *
|
|
from string import Template
|
|
|
|
cap_strings = { 'ipc' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_IPC | CAP_RTYPE_CONTAINER,
|
|
\t\t\t\t.access = CAP_IPC_SEND | CAP_IPC_RECV
|
|
\t\t\t\t | CAP_IPC_FULL | CAP_IPC_SHORT
|
|
\t\t\t\t | CAP_IPC_EXTENDED | CAP_CHANGEABLE
|
|
\t\t\t\t | CAP_REPLICABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0, .size = 0,
|
|
\t\t\t},
|
|
'''
|
|
, 'tctrl' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_TCTRL | CAP_RTYPE_CONTAINER,
|
|
\t\t\t\t.access = CAP_TCTRL_CREATE | CAP_TCTRL_DESTROY
|
|
\t\t\t\t | CAP_TCTRL_SUSPEND | CAP_TCTRL_RUN
|
|
\t\t\t\t | CAP_TCTRL_RECYCLE | CAP_TCTRL_WAIT
|
|
\t\t\t\t | CAP_CHANGEABLE | CAP_REPLICABLE
|
|
\t\t\t\t | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0, .size = 0,
|
|
\t\t\t},
|
|
'''
|
|
, 'exregs' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_EXREGS | CAP_RTYPE_CONTAINER,
|
|
\t\t\t\t.access = CAP_EXREGS_RW_PAGER
|
|
\t\t\t\t | CAP_EXREGS_RW_UTCB | CAP_EXREGS_RW_SP
|
|
\t\t\t\t | CAP_EXREGS_RW_PC | CAP_EXREGS_RW_REGS
|
|
\t\t\t\t | CAP_CHANGEABLE | CAP_REPLICABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0, .size = 0,
|
|
\t\t\t},
|
|
'''
|
|
, 'capctrl' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_CAP | CAP_RTYPE_CONTAINER,
|
|
\t\t\t\t.access = CAP_CAP_GRANT | CAP_CAP_READ
|
|
\t\t\t\t | CAP_CAP_SHARE | CAP_CAP_REPLICATE
|
|
\t\t\t\t | CAP_CAP_MODIFY
|
|
\t\t\t\t| CAP_CAP_READ | CAP_CAP_SHARE,
|
|
\t\t\t\t.start = 0, .end = 0, .size = 0,
|
|
\t\t\t},
|
|
'''
|
|
, 'umutex' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_UMUTEX | CAP_RTYPE_CONTAINER,
|
|
\t\t\t\t.access = CAP_UMUTEX_LOCK | CAP_UMUTEX_UNLOCK,
|
|
\t\t\t\t.start = 0, .end = 0, .size = 0,
|
|
\t\t\t},
|
|
'''
|
|
, 'threadpool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY
|
|
\t\t\t\t | CAP_RTYPE_THREADPOOL,
|
|
\t\t\t\t.access = CAP_CHANGEABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0,
|
|
\t\t\t\t.size = ${size},
|
|
\t\t\t},
|
|
'''
|
|
, 'spacepool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY | CAP_RTYPE_SPACEPOOL,
|
|
\t\t\t\t.access = CAP_CHANGEABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0,
|
|
\t\t\t\t.size = ${size},
|
|
\t\t\t},
|
|
'''
|
|
, 'cpupool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY | CAP_RTYPE_CPUPOOL,
|
|
\t\t\t\t.access = 0, .start = 0, .end = 0,
|
|
\t\t\t\t.size = ${size} /* Percentage */,
|
|
\t\t\t},
|
|
'''
|
|
, 'mutexpool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY | CAP_RTYPE_MUTEXPOOL,
|
|
\t\t\t\t.access = CAP_CHANGEABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0,
|
|
\t\t\t\t.size = ${size},
|
|
\t\t\t},
|
|
'''
|
|
, 'mappool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t/* For pmd accounting */
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY | CAP_RTYPE_MAPPOOL,
|
|
\t\t\t\t.access = CAP_CHANGEABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0,
|
|
\t\t\t\t/* Function of mem regions, nthreads etc. */
|
|
\t\t\t\t.size = ${size},
|
|
\t\t\t},
|
|
'''
|
|
, 'cappool' : \
|
|
'''
|
|
\t\t\t[${idx}] = {
|
|
\t\t\t\t/* For cap spliting, creating, etc. */
|
|
\t\t\t\t.target_type = ${target_type},
|
|
\t\t\t\t.target = ${cid},
|
|
\t\t\t\t.type = CAP_TYPE_QUANTITY | CAP_RTYPE_CAPPOOL,
|
|
\t\t\t\t.access = CAP_CHANGEABLE | CAP_TRANSFERABLE,
|
|
\t\t\t\t.start = 0, .end = 0,
|
|
\t\t\t\t/* This may be existing caps X 2 etc. */
|
|
\t\t\t\t.size = ${size},
|
|
\t\t\t},
|
|
'''
|
|
}
|
|
|
|
|
|
def prepare_custom_capability(cont, param, val):
|
|
if 'TYPE' in param:
|
|
capkey, captype, rest = param.split('_', 2)
|
|
capkey = capkey.lower()
|
|
captype = captype.lower()
|
|
cont.caps[capkey] = cap_strings[captype]
|
|
elif 'TARGET' in param:
|
|
target_parts = param.split('_', 2)
|
|
if len(target_parts) == 2:
|
|
capkey = target_parts[0].lower()
|
|
templ = Template(cont.caps[capkey])
|
|
cont.caps[capkey] = templ.safe_substitute(cid = val)
|
|
elif len(target_parts) == 3:
|
|
capkey = target_parts[0].lower()
|
|
ttype = target_parts[2]
|
|
templ = Template(cont.caps[capkey])
|
|
|
|
# Insert current container id, if target has current
|
|
if ttype[:len('CURRENT')] == 'CURRENT':
|
|
cont.caps[capkey] = templ.safe_substitute(target_type = ttype, cid = cont.id)
|
|
else:
|
|
cont.caps[capkey] = templ.safe_substitute(target_type = ttype)
|
|
else: # Ignore custom_use symbol
|
|
return
|
|
print capkey
|
|
print cont.caps[capkey]
|
|
|
|
def prepare_typed_capability(cont, param, val):
|
|
captype, params = param.split('_', 1)
|
|
captype = captype.lower()
|
|
|
|
|
|
# USE makes us assign the initial cap string with blank fields
|
|
if 'USE' in params:
|
|
cont.caps[captype] = cap_strings[captype]
|
|
|
|
# Prepare string template from capability type
|
|
templ = Template(cont.caps[captype])
|
|
|
|
# If it is a pool, amend default target type and id
|
|
if captype[-len('pool'):] == 'pool':
|
|
cont.caps[captype] = templ.safe_substitute(target_type = 'CURRENT_PAGER_SPACE',
|
|
cid = cont.id)
|
|
|
|
# Fill in the blank size field
|
|
elif 'SIZE' in params:
|
|
# Get reference to capability string template
|
|
templ = Template(cont.caps[captype])
|
|
cont.caps[captype] = templ.safe_substitute(size = val)
|
|
|
|
# Fill in capability target type and target id fields
|
|
elif 'TARGET' in params:
|
|
# Get reference to capability string template
|
|
templ = Template(cont.caps[captype])
|
|
target, ttype = params.split('_', 1)
|
|
|
|
# Target type
|
|
if ttype != None:
|
|
# Insert current container id, if target has current
|
|
if ttype[:len('CURRENT')] == 'CURRENT':
|
|
cont.caps[captype] = templ.safe_substitute(target_type = ttype, cid = cont.id)
|
|
else:
|
|
cont.caps[captype] = templ.safe_substitute(target_type = ttype)
|
|
# Get target value supplied by user
|
|
else:
|
|
cont.caps[captype] = templ.safe_substitute(cid = val)
|
|
|
|
print captype
|
|
print cont.caps[captype]
|
|
|
|
def prepare_capability(cont, param, val):
|
|
if 'CUSTOM' in param:
|
|
prepare_custom_capability(cont, param, val)
|
|
else:
|
|
prepare_typed_capability(cont, param, val)
|
|
|