136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
|
|
||
|
|
import ops.cmd
|
||
|
|
import dsz
|
||
|
|
from ops.cmd import OpsCommandException
|
||
|
|
VALID_OPTIONS = ['hive', 'key', 'value', 'recursive', 'target', 'wow64', 'wow32', 'chunksize']
|
||
|
|
HIVES = ['u', 'l', 'c', 'g', 'r']
|
||
|
|
|
||
|
|
class RegistryQueryCommand(ops.cmd.DszCommand, ):
|
||
|
|
|
||
|
|
def __init__(self, plugin='registryquery', prefixes=[], arglist=None, dszquiet=True, hive='l', **optdict):
|
||
|
|
ops.cmd.DszCommand.__init__(self, plugin=plugin, dszquiet=dszquiet, **optdict)
|
||
|
|
self.hive = hive
|
||
|
|
if ('key' in optdict):
|
||
|
|
self.key = optdict['key']
|
||
|
|
if ('value' in optdict):
|
||
|
|
self.value = optdict['value']
|
||
|
|
|
||
|
|
def _getHive(self):
|
||
|
|
return self.optdict['hive']
|
||
|
|
|
||
|
|
def _setHive(self, val):
|
||
|
|
if (val is None):
|
||
|
|
raise OpsCommandException('You must set hive, hive cannot be None')
|
||
|
|
if (val.lower() in HIVES):
|
||
|
|
self.optdict['hive'] = val.lower()
|
||
|
|
else:
|
||
|
|
raise OpsCommandException(('Invalid hive %s' % val))
|
||
|
|
hive = property(_getHive, _setHive)
|
||
|
|
|
||
|
|
def _getKey(self):
|
||
|
|
if ('key' in self.optdict):
|
||
|
|
return self.optdict['key']
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def _setKey(self, val):
|
||
|
|
if ((val is None) or (val.strip() == '')):
|
||
|
|
if ('key' in self.optdict):
|
||
|
|
del self.optdict['key']
|
||
|
|
return
|
||
|
|
if ((val.find(' ') > (-1)) and (val[0] != '"')):
|
||
|
|
val = ('"%s"' % val)
|
||
|
|
if (val.find('""') > (-1)):
|
||
|
|
val = val.replace('""', '"')
|
||
|
|
self.optdict['key'] = val
|
||
|
|
key = property(_getKey, _setKey)
|
||
|
|
|
||
|
|
def _getValue(self):
|
||
|
|
if ('value' in self.optdict):
|
||
|
|
return self.optdict['value']
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def _setValue(self, val):
|
||
|
|
if (val is None):
|
||
|
|
if ('value' in self.optdict):
|
||
|
|
del self.optdict['value']
|
||
|
|
return
|
||
|
|
if ((val.find(' ') > (-1)) and (val[0] != '"')):
|
||
|
|
val = ('"%s"' % val)
|
||
|
|
if (val.find('""') > (-1)):
|
||
|
|
val = val.replace('""', '"')
|
||
|
|
self.optdict['value'] = val
|
||
|
|
value = property(_getValue, _setValue)
|
||
|
|
|
||
|
|
def _getRecursive(self):
|
||
|
|
if (('recursive' in self.optdict) and self.optdict['recursive']):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def _setRecursive(self, val):
|
||
|
|
if val:
|
||
|
|
self.optdict['recursive'] = True
|
||
|
|
elif ('recursive' in self.optdict):
|
||
|
|
del self.optdict['recursive']
|
||
|
|
recursive = property(_getRecursive, _setRecursive)
|
||
|
|
|
||
|
|
def _getTarget(self):
|
||
|
|
if ('target' in self.optdict):
|
||
|
|
return self.optdict['target']
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def _setTarget(self, val):
|
||
|
|
if (val is None):
|
||
|
|
if ('target' in self.optdict):
|
||
|
|
del self.optdict['target']
|
||
|
|
return
|
||
|
|
self.optdict['target'] = val
|
||
|
|
target = property(_getTarget, _setTarget)
|
||
|
|
|
||
|
|
def _getWow64(self):
|
||
|
|
if (('wow64' in self.optdict) and self.optdict['wow64']):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def _setWow64(self, val):
|
||
|
|
if val:
|
||
|
|
self.optdict['wow64'] = val
|
||
|
|
elif ('wow64' in self.optdict):
|
||
|
|
del self.optdict['wow64']
|
||
|
|
wow64 = property(_getWow64, _setWow64)
|
||
|
|
|
||
|
|
def _getWow32(self):
|
||
|
|
if (('wow32' in self.optdict) and self.optdict['wow32']):
|
||
|
|
return True
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
def _setWow32(self, val):
|
||
|
|
if val:
|
||
|
|
self.optdict['wow32'] = val
|
||
|
|
elif ('wow32' in self.optdict):
|
||
|
|
del self.optdict['wow32']
|
||
|
|
wow32 = property(_getWow32, _setWow32)
|
||
|
|
|
||
|
|
def _getChunksize(self):
|
||
|
|
if ('chunksize' in self.optdict):
|
||
|
|
return self.optdict['chunksize']
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def _setChunksize(self, val):
|
||
|
|
if (val is None):
|
||
|
|
if ('chunksize' in self.optdict):
|
||
|
|
del self.optdict['chunksize']
|
||
|
|
return
|
||
|
|
if (type(val) is int):
|
||
|
|
self.optdict['chunksize'] = val
|
||
|
|
else:
|
||
|
|
raise OpsCommandException('chunksize is required to be an integer')
|
||
|
|
chunksize = property(_getChunksize, _setChunksize)
|
||
|
|
ops.cmd.command_classes['registryquery'] = RegistryQueryCommand
|
||
|
|
ops.cmd.aliasoptions['registyquery'] = VALID_OPTIONS
|