示例#1
0
class AVScanner(object):
    def __init__(self):
        self.multi_av = CMultiAV("%s/cmultiav.cfg" % os.path.dirname(__file__))

    def has_virus(self, file, parallel=True):
        with tempfile.NamedTemporaryFile(delete=False) as temporaryfile:
            temporaryfile.write(file.read())
            temporaryfile.close()
            permission = 0o664  # PEP 3127: octal literals
            os.chmod(temporaryfile.name, permission)  # clamav needs permission to scan
            if parallel:
                # TODO: Handle the case for when no scanner is installed ...
                # Currently we say that nothing was found...
                # The single core case works as expected
                ret = self.multi_av.scan(temporaryfile.name, AV_SPEED_MEDIUM)
            else:

                try:
                    ret = self.multi_av.single_scan(temporaryfile.name, AV_SPEED_MEDIUM)
                except OSError:
                    # It would seem a scanner is not installed...
                    return True, "No virus scanner was found on the system..."

            os.unlink(temporaryfile.name)
            for x in list(ret.values()):
                if x != {}:
                    # all is lost as soon as one scanner finds something
                    return True, ret
            return False, None
示例#2
0
def main(path):
    multi_av = CMultiAV()
    ret = multi_av.scan(path, AV_SPEED_ALL)

    import pprint
    pprint.pprint(ret)