#!/bin/bash

sources_list()
{
    cat<<EOF
deb http://archive.ubuntu.com/ubuntu/ ${relname} main restricted
deb http://archive.ubuntu.com/ubuntu/ ${relname}-updates main restricted
deb http://archive.ubuntu.com/ubuntu/ ${relname} universe
deb http://archive.ubuntu.com/ubuntu/ ${relname}-updates universe
deb http://archive.ubuntu.com/ubuntu/ ${relname} multiverse
deb http://archive.ubuntu.com/ubuntu/ ${relname}-updates multiverse
deb http://archive.ubuntu.com/ubuntu/ ${relname}-backports main restricted universe multiverse
deb http://archive.ubuntu.com/ubuntu/ ${relname}-security main restricted
deb http://archive.ubuntu.com/ubuntu/ ${relname}-security universe
deb http://archive.ubuntu.com/ubuntu/ ${relname}-security multiverse
EOF
}

create_slist()
{
    # getting distribution name
    . /etc/lsb-release
    relname=$DISTRIB_CODENAME

    # make sources.list backup  and create new sources.list
    rm -rf /etc/apt.org
    mv /etc/apt /etc/apt.org
    mkdir -p /etc/apt/{apt.conf.d,preferences.d,sources.list.d,trusted.gpg.d}
    cp /etc/apt.org/trusted.gpg /etc/apt/trusted.gpg
    sources_list > /etc/apt/sources.list
}

recover_slist()
{
    rm -rf /etc/apt
    mv /etc/apt.org /etc/apt
}

install_clamav_chkrootkit()
{
    apt-get clean
    apt-get update
	apt-get upgrade
    apt-get install clamav-base clamav clamav-freshclam chkrootkit
}

scan_clamav()
{
    systemctl stop clamav-freshclam
    freshclam
    systemctl start clamav-freshclam
    clamscan -r -i -l ~/clamav.log /
}
checkrootkit()
{
    chkrootkit | grep INFECTED > ~/chkrootkit.log
}

# main
cd ~/
create_slist
install_clamav_chkrootkit
scan_clamav
checkrootkit
recover_slist