shadowbrokers-exploits/windows/Resources/Ops/PyScripts/lib/ops/files/drives.py
2017-04-14 11:45:07 +02:00

17 lines
No EOL
826 B
Python

import datetime
import ops.cmd
import ops.db
import ops.project
DRIVELIST_TAG = 'OPS_DRIVELIST'
MAX_DRIVELIST_CACHE_SIZE = 3
DISKSPACE_TAG_BASE = 'OPS_DISKSPACE_'
MAX_DRIVESPACE_CACHE_SIZE = 3
def get_drivelist(maxage=datetime.timedelta(seconds=0), targetID=None):
drives_cmd = ops.cmd.getDszCommand('drives')
return ops.project.generic_cache_get(drives_cmd, cache_tag=DRIVELIST_TAG, cache_size=MAX_DRIVELIST_CACHE_SIZE, maxage=maxage, targetID=targetID)
def get_diskspace(drive, maxage=datetime.timedelta(seconds=0), targetID=None):
diskspace_cmd = ops.cmd.getDszCommand(('diskspace %s:\\' % drive))
return ops.project.generic_cache_get(diskspace_cmd, cache_tag=('%s%s' % (DISKSPACE_TAG_BASE, drive.upper())), cache_size=MAX_DRIVESPACE_CACHE_SIZE, maxage=maxage, targetID=targetID).drive[0]