77 lines
2.3 KiB
Python
77 lines
2.3 KiB
Python
"""ClamAV related tools."""
|
|
|
|
from .. import package
|
|
from .. import utils
|
|
from .. import system
|
|
|
|
from . import base
|
|
|
|
|
|
class Clamav(base.Installer):
|
|
|
|
"""ClamAV installer."""
|
|
|
|
appname = "clamav"
|
|
packages = {
|
|
"deb": ["clamav-daemon"],
|
|
"rpm": [
|
|
"clamav", "clamav-update", "clamav-server", "clamav-server-systemd"
|
|
],
|
|
}
|
|
|
|
def get_daemon_name(self):
|
|
"""Return appropriate daemon name."""
|
|
if package.backend.FORMAT == "rpm":
|
|
return "clamd@amavisd"
|
|
return "clamav-daemon"
|
|
|
|
@property
|
|
def config_dir(self):
|
|
"""Return appropriate config dir."""
|
|
if package.backend.FORMAT == "rpm":
|
|
return "/etc"
|
|
return ""
|
|
|
|
def get_config_files(self):
|
|
"""Return appropriate config files."""
|
|
if package.backend.FORMAT == "rpm":
|
|
return ["sysconfig/clamd.amavisd", "tmpfiles.d/clamd.amavisd.conf"]
|
|
return []
|
|
|
|
def post_run(self):
|
|
"""Additional tasks."""
|
|
if package.backend.FORMAT == "deb":
|
|
user = self.config.get(self.appname, "user")
|
|
system.add_user_to_group(
|
|
user, self.config.get("amavis", "user")
|
|
)
|
|
pattern = (
|
|
"s/^AllowSupplementaryGroups false/"
|
|
"AllowSupplementaryGroups true/")
|
|
utils.exec_cmd(
|
|
"perl -pi -e '{}' /etc/clamav/clamd.conf".format(pattern))
|
|
else:
|
|
user = "clamupdate"
|
|
utils.exec_cmd(
|
|
"perl -pi -e 's/^Example/#Example/' /etc/freshclam.conf")
|
|
# Check if not present before
|
|
path = "/usr/lib/systemd/system/clamd@.service"
|
|
code, output = utils.exec_cmd(
|
|
"grep 'WantedBy=multi-user.target' {}".format(path))
|
|
if code:
|
|
utils.exec_cmd(
|
|
"""cat <<EOM >> {}
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
EOM
|
|
""".format(path))
|
|
|
|
if utils.dist_name() in ["debian", "ubuntu"]:
|
|
# Stop freshclam daemon to allow manual download
|
|
utils.exec_cmd("service clamav-freshclam stop")
|
|
utils.exec_cmd("freshclam", sudo_user=user, login=False)
|
|
utils.exec_cmd("service clamav-freshclam start")
|
|
else:
|
|
utils.exec_cmd("freshclam", sudo_user=user, login=False)
|