kopano-scan-attachments.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
import kopano
from MAPICore import *
from six import BytesIO

import clamd


def opt_args():
    parser = kopano.parser('skpcubeC')
    parser.add_option("--all", dest="all", action="store_true",
                      default=False, help="run program for all users")
    parser.add_option("--autoremove", dest="autoremove", action="store_true", default=False,
                      help="remove infected attachments")
    return parser.parse_args()


def scanmail(clam, email, autoremove):
    for attachment in email.attachments():
        try:
            result = clam.instream(BytesIO(attachment.data))
            if result['stream'][0] == 'FOUND':
                print ('\t\tVirus found: [%s] [%s] [%s]' % (email.subject, result['stream'][0], result['stream'][1]))
                if autoremove:
                    delete(email, attachment, str(result['stream'][1]))
        except Exception as e:
            print ("\tUnable to scan attachment: [%s] [%s] [%s]" % (email.subject, attachment.name, e))


def delete(email, attachment, virus):
    print ('\t\tAutoremoving attachment: [%s] [%s]' % (email.subject, attachment.name))
    email.delete(attachment)
    message = 'The attachment with the name %s has been removed because it was infected with %s' % (
        attachment.name, virus)
    email.create_attachment('virus-removed.txt', message)


def main():
    options, args = opt_args()
    if not (len(options.users) or options.all):
        print ('Run `zarafa-scan-attachments.py --help` for parameters')
        return

    clam = clamd.ClamdUnixSocket()

    if not clam.ping():
        print ('No ClamAV Daemon connection')
    else:
        conn = kopano.Server(options=options)
        for user in conn.users():
            print ('Scanning user: [%s]' % user.name)
            for folder in user.store.folders():
                print ('\tScanning folder: [%s]' % folder.name)
                for item in folder.items():
                    scanmail(clam, item, options.autoremove)


if __name__ == '__main__':
    main()