You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.5 KiB
44 lines
1.5 KiB
import logging
|
|
import clamd
|
|
import sys
|
|
|
|
from django.core.exceptions import ValidationError
|
|
from django.conf import settings
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def file_scan_validation(file):
|
|
"""
|
|
This validator sends the file to ClamAV for scanning and returns returns to the form. By default, if antivirus
|
|
service is not available or there are errors, the validation will fail.
|
|
|
|
Usage:
|
|
class UploadForm(forms.Form):
|
|
file = forms.FileField(validators=[file_scan_validation])
|
|
:param file:
|
|
:return:
|
|
"""
|
|
logger.debug("starting file scanning with clamav")
|
|
if not settings.CLAMAV_ENABLED:
|
|
logger.warning('File scanning has been disabled.')
|
|
return
|
|
|
|
# make sure we're at the beginning of the file stream
|
|
file.seek(0)
|
|
|
|
# we're just going to assume a network connection to clamav here .. no local unix socket support
|
|
scanner = clamd.ClamdNetworkSocket(settings.CLAMAV_TCP_ADDR, settings.CLAMAV_TCP_PORT)
|
|
try:
|
|
result = scanner.instream(file)
|
|
except:
|
|
# it doesn't really matter what the actual error is .. log it and raise validation error
|
|
logger.error('Error occurred while trying to scan file. "{}"'.format(sys.exc_info()[0]))
|
|
raise ValidationError('Unable to scan file.', code='scanerror')
|
|
finally:
|
|
# reset file stream
|
|
file.seek(0)
|
|
|
|
if result and result['stream'][0] == 'FOUND':
|
|
logger.warning('Virus found: {}'.format(file.name))
|
|
raise ValidationError('Infected file found.', code='infected')
|