def command(self):
"command"
self.init()
days_to_retain = int(self.conf.get('ms.quarantine.days_to_keep', 0))
quarantine_dir = get_config_option('QuarantineDir')
if (quarantine_dir.startswith('/etc')
or quarantine_dir.startswith('/lib')
or quarantine_dir.startswith('/home')
or quarantine_dir.startswith('/bin')
or quarantine_dir.startswith('..')):
return False
if (not os.path.exists(quarantine_dir)) or (days_to_retain == 0):
return False
ignore_dirs = ['spam', 'mcp', 'nonspam']
def process_dir(dirs, process_path, direc):
"process dirs"
if os.path.exists(os.path.join(process_path, direc)):
dirs.extend(
[f for f in os.listdir(os.path.join(process_path, direc))])
dirs = [
f for f in os.listdir(quarantine_dir)
if os.path.isdir(os.path.join(quarantine_dir, f)) and QDIR.match(f)
and should_be_pruned(f, days_to_retain)
]
dirs.sort()
for direc in dirs:
process_path = os.path.join(quarantine_dir, direc)
ids = [f for f in os.listdir(process_path) if f not in ignore_dirs]
for ignore_dir in ignore_dirs:
process_dir(ids, process_path, ignore_dir)
sql = Message.__table__.update()\
.where(Message.messageid.in_(ids))\
.values(isquarantined=0)
Session.bind.execute(sql)
if (os.path.isabs(process_path)
and (not os.path.islink(process_path))):
try:
shutil.rmtree(process_path)
except shutil.Error:
print >> sys.stderr, ("Failed to remove %(path)s" %
dict(path=process_path))
else:
print >> sys.stderr, ("%(path)s is a symlink skipping" %
dict(path=process_path))
def should_be_pruned(direc, days_to_retain):
"""
Returns true or false :
if the directory is older than days_to_retain
returns true
else
returns false
"""
if (not days_to_retain) or (not QDIR.match(direc)):
return False
interval = datetime.timedelta(days=days_to_retain)
last_date = datetime.date.today() - interval
year = int(direc[0:4])
mon = int(direc[4:-2])
day = int(direc[6:])
dir_date = datetime.date(year, mon, day)
return dir_date < last_date
def command(self):
"command"
self.init()
if asbool(self.conf.get('ms.quarantine.shared')):
lock_name = 'cleanquarantine'
else:
lock_name = 'cleanquarantine-%s' % system_hostname()
if acquire_lock(lock_name, self.conf):
try:
days_to_retain = int(
self.conf.get('ms.quarantine.days_to_keep', 0))
quarantine_dir = get_config_option('QuarantineDir')
if (quarantine_dir.startswith(('/etc', '/lib', '/home',
'/bin', '/sbin', '..'))):
return False
if ((not os.path.exists(quarantine_dir)) or
(days_to_retain == 0)):
return False
ignore_dirs = ['spam', 'mcp', 'nonspam']
def process_dir(dirs, process_path, direc):
"process dirs"
if os.path.exists(os.path.join(process_path, direc)):
dirs.extend([f for f in os.listdir(
os.path.join(process_path, direc))])
dirs = [f for f in os.listdir(quarantine_dir)
if os.path.isdir(os.path.join(quarantine_dir, f)) and
QDIR.match(f) and should_be_pruned(f, days_to_retain)]
dirs.sort()
for direc in dirs:
process_path = os.path.join(quarantine_dir, direc)
ids = [f for f in os.listdir(process_path)
if f not in ignore_dirs]
for ignore_dir in ignore_dirs:
process_dir(ids, process_path, ignore_dir)
year, month, day = (int(direc[:4]),
int(direc[4:-2]),
int(direc[6:]))
startdate = datetime.datetime(year, month, day, 00, 00, 00)
enddate = datetime.datetime(year, month, day, 23, 59, 59)
localzone = make_tz(self.conf['baruwa.timezone'])
startdate = localzone.localize(startdate)
enddate = localzone.localize(enddate)
startdate = pytz.utc.normalize(
startdate.astimezone(pytz.utc))
enddate = pytz.utc.normalize(enddate.astimezone(pytz.utc))
sql = Message.__table__.update().where(and_(
Message.messageid.in_(ids),
Message.timestamp.between(startdate, enddate)
)).values(isquarantined=0)
Session.bind.execute(sql)
if (os.path.isabs(process_path) and
(not os.path.islink(process_path))):
try:
shutil.rmtree(process_path)
except shutil.Error:
print >> sys.stderr, ("Failed to remove %(path)s"
% dict(path=process_path))
else:
print >> sys.stderr, ("%(path)s is a symlink skipping"
% dict(path=process_path))
finally:
Session.close()
release_lock(lock_name, self.conf)
