# coding: utf-8
# +-------------------------------------------------------------------
# | YakPanel x5
# +-------------------------------------------------------------------
# | Copyright (c) 2015-2018 YakPanel(www.yakpanel.com) All rights reserved.
# +-------------------------------------------------------------------
# | Author: 1249648969@qq.com
# +-------------------------------------------------------------------
import sys, os, public, re, firewalld, time

if sys.version_info[0] == 2:
    reload(sys)
    sys.setdefaultencoding('utf-8')


class firewalls:
    __isFirewalld = False
    __isUfw = False
    __Obj = None

    def __init__(self):
        if os.path.exists('/usr/sbin/firewalld'): self.__isFirewalld = True
        if os.path.exists('/usr/sbin/ufw'): self.__isUfw = True
        public.M('firewall').execute("alter table firewall add ports TEXT;", ())
        public.M('firewall').execute("alter table firewall add protocol TEXT;", ())
        public.M('firewall').execute("alter table firewall add address_ip TEXT;", ())
        public.M('firewall').execute("alter table firewall add types TEXT;", ())
        # 这里判断的是Centos7 的系统
        if self.__isFirewalld:
            self.__Obj = firewalld.firewalld();
            # 获取列表信息
            self.GetList();

    # 获取服务端列表
    def GetList(self, get=None):
        try:
            data = {}
            # 获取开放的端口
            data['ports'] = self.__Obj.GetAcceptPortList();
            # 当前时间
            # '2018-10-11 14:36:40'
            addtime = time.strftime('%Y-%m-%d %X', time.localtime())
            #
            for i in range(len(data['ports'])):
                #
                tmp = self.CheckDbExists(data['ports'][i]['port'], data['ports'][i]['protocol']);
                # | id | port  | ps  | addtime   | ports | protocol | address_ip | types |
                if not tmp: public.M('firewall').add('port,ps,addtime', (data['ports'][i]['port'], '', addtime))

            data['iplist'] = self.__Obj.GetDropAddressList();

            for i in range(len(data['iplist'])):
                try:
                    tmp = self.CheckDbExists(data['iplist'][i]['address']);
                    if not tmp: public.M('firewall').add('port,ps,addtime', (data['iplist'][i]['address'], '', addtime))
                except:
                    return public.get_error_info()

            # 添加到firewalls 数据表中
            data['reject'] = self.__Obj.GetrejectLIST()

            for i in range(len(data['reject'])):
                try:
                    tmp = self.CheckDbExists2(data['reject'][i]['protocol'],
                                              data['reject'][i]['type'],
                                              data['reject'][i]['port'],
                                              data['reject'][i]['address'])
                    if not tmp: public.M('firewall').add('protocol,types,ports,address_ip,addtime',
                                                         (data['reject'][i]['protocol'],
                                                          data['reject'][i]['type'],
                                                          data['reject'][i]['port'],
                                                          data['reject'][i]['address'], addtime))
                except:
                    return public.get_error_info()
            # 添加允许信息到firewalls 表中
            data['accept'] = self.__Obj.Getacceptlist()
            # return data
            for i in range(len(data['accept'])):
                try:
                    tmp = self.CheckDbExists2(data['accept'][i]['protocol'],
                                              data['accept'][i]['type'],
                                              data['accept'][i]['port'],
                                              data['accept'][i]['address'])
                    if not tmp: public.M('firewall').add('protocol,types,ports,address_ip,addtime',
                                                         (data['accept'][i]['protocol'],
                                                          data['accept'][i]['type'],
                                                          data['accept'][i]['port'],
                                                          data['accept'][i]['address'], addtime))
                except:
                    return public.get_error_info()
            count = public.M('firewall').count();
            data = {}
            data['page'] = public.get_page(count, int(get.p), 12, get.collback)
            data['data'] = public.M('firewall').limit(data['page']['shift'] + ',' + data['page']['row']).order(
                'id desc').select()
            for i in range(len(data['data'])):
                if data['data'][i]['port'].find(':') != -1 or data['data'][i]['port'].find('.') != -1 or \
                        data['data'][i]['port'].find('-') != -1:
                    data['data'][i]['status'] = -1;
                else:
                    data['data'][i]['status'] = public.check_port_stat(int(data['data'][i]['port']));

            data['page'] = data['page']['page']
            return data
        except Exception as ex:
            return public.get_error_info()

    # 检查数据库是否存在
    def CheckDbExists(self, port, type=None):
        data = public.M('firewall').field('id,port,ps,addtime,types').select();
        return data
        for dt in data:
            if dt['port'] == port and dt['type'] == type: return dt;
        return False;

    # 查看frewalls 数据库表中是否存在
    # | id | port  | ps  | addtime   | ports | protocol | address_ip | types |
    def CheckDbExists2(self, protocol, type, port, address):
        data = public.M('firewall').field('protocol,types,ports,address_ip').select()
        for dt in data:
            if dt['ports'] == port and dt['protocol'] == protocol and dt['types'] == type and dt[
                'address_ip'] == address: return dt;
        return False

    # 重载防火墙配置
    def FirewallReload(self):
        if self.__isUfw:
            public.ExecShell('/usr/sbin/ufw reload')
            return;
        if self.__isFirewalld:
            public.ExecShell('firewall-cmd --reload')
        else:
            public.ExecShell('/etc/init.d/iptables save')
            public.ExecShell('/etc/init.d/iptables restart')

    # 添加屏蔽IP
    def AddDropAddress(self, get):
        import time
        import re
        rep = r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(\/\d{1,2})?$"
        if not re.search(rep, get.port): return public.return_msg_gettext(False, public.lang("IP address youve entered is illegal!"));
        address = get.port
        if public.M('firewall').where("port=?", (address,)).count() > 0: return public.return_msg_gettext(False, public.lang("The IP exists in block list, no need to repeat processing!"))
        if self.__isUfw:
            public.ExecShell('ufw deny from ' + address + ' to any');
        else:
            if self.__isFirewalld:
                public.ExecShell(
                    'firewall-cmd --permanent --add-rich-rule=\'rule family=ipv4 source address="' + address + '" drop\'')
                ret = self.__Obj.CheckIpDrop(address)
                if not ret:
                    self.__Obj.AddDropAddress(address)

            else:
                public.ExecShell('iptables -I INPUT -s ' + address + ' -j DROP')

        public.write_log_gettext("Firewall manager", 'Successfully blocked IP [{}]!', (address,))
        addtime = time.strftime('%Y-%m-%d %X', time.localtime())
        public.M('firewall').add('port,ps,addtime', (address, get.ps, addtime))
        self.FirewallReload()
        return public.return_msg_gettext(True, public.lang("Successfully added"))

    # 删除IP屏蔽
    def DelDropAddress(self, get):
        address = get.port
        id = get.id
        if self.__isUfw:
            public.ExecShell('ufw delete deny from ' + address + ' to any');
        else:
            if self.__isFirewalld:
                public.ExecShell(
                    'firewall-cmd --permanent --remove-rich-rule=\'rule family=ipv4 source address="' + address + '" drop\'')
                ret = self.__Obj.DelDropAddress(address)
                if ret:
                    pass
            else:
                public.ExecShell('iptables -D INPUT -s ' + address + ' -j DROP')

        public.write_log_gettext("Firewall manager", 'Unblocked IP [{}]!', (address,))
        public.M('firewall').where("id=?", (id,)).delete()

        self.FirewallReload();
        return public.return_msg_gettext(True, public.lang("Successfully deleted"))

    # 添加放行端口
    def AddAcceptPort(self, get):
        flag = False
        import re
        rep = r"^\d{1,5}(:\d{1,5})?$"
        if not re.search(rep, get.port): return public.return_msg_gettext(False, public.lang("Port range is incorrect!"));
        import time
        port = get.port
        ps = get.ps
        types = get.type
        type_list = ['tcp', 'udp']
        if types not in type_list: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
        notudps = ['80', '443', '8888', '888', '39000:40000', '21', '22']
        if port in notudps: flag = True
        # return public.M('firewall').where("port=?", (port,)).count()
        if types == 'tcp':
            if flag:
                if public.M('firewall').where("port=?", (port,)).count() > 0: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
            else:
                if public.M('firewall').where("port=? and type='tcp'",
                                              (port,)).count() > 0: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
        elif types == 'udp':
            if flag:
                if public.M('firewall').where("port=?", (port,)).count() > 0: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
            else:
                if public.M('firewall').where("port=? and type='udp'",
                                              (port,)).count() > 0: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
        else:
            return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))

        if self.__isUfw:
            if port in notudps:
                public.ExecShell('ufw allow ' + port + '/tcp')
            else:
                public.ExecShell('ufw allow ' + port + '/' + type + '');
        else:
            if self.__isFirewalld:
                port = port.replace(':', '-')
                if port in notudps:
                    public.ExecShell('firewall-cmd --permanent --zone=public --add-port=' + port + '/tcp')
                else:
                    public.ExecShell('firewall-cmd --permanent --zone=public --add-port=' + port + '/' + types + '')
            else:
                if port in notudps:
                    public.ExecShell(
                        'iptables -I INPUT -p tcp -m state --state NEW -m tcp --dport ' + port + ' -j ACCEPT')
                else:
                    public.ExecShell(
                        'iptables -I INPUT -p tcp -m state --state NEW -m ' + types + ' --dport ' + port + ' -j ACCEPT')

        public.write_log_gettext("Firewall manager", 'Successfully accepted port [{}]!', (port,))
        addtime = time.strftime('%Y-%m-%d %X', time.localtime())
        result = public.M('firewall').add('port,ps,addtime,types', (port, ps, addtime, types))
        # return result
        self.FirewallReload()
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    # 删除放行端口
    def DelAcceptPort(self, get):
        port = get.port
        id = get.id
        types = get.type
        type_list = ['tcp', 'udp']
        if not types in type_list: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
        try:
            if (port == public.GetHost(True)): return public.return_msg_gettext(False, public.lang("Failed，cannot delete current port of the panel!"))
            if self.__isUfw:
                public.ExecShell('ufw delete allow ' + port + '/' + types + '');
            else:
                if self.__isFirewalld:
                    public.ExecShell('firewall-cmd --permanent --zone=public --remove-port=' + port + '/' + types + '')
                else:
                    public.ExecShell(
                        'iptables -D INPUT -p tcp -m state --state NEW -m ' + types + ' --dport ' + port + ' -j ACCEPT')
            public.write_log_gettext("Firewall manager", 'Successfully deleted accepted port [{}] on firewall!',
                                     (port,))
            public.M('firewall').where("id=?", (id,)).delete()

            self.FirewallReload()
            return public.return_msg_gettext(True, public.lang("Successfully deleted"))
        except:
            return public.return_msg_gettext(False, public.lang("Failed to delete"))

    # 设置远程端口状态
    def SetSshStatus(self, get):
        version = public.readFile('/etc/redhat-release')
        if int(get['status']) == 1:
            msg = public.get_msg_gettext('SSH service turned off')
            act = 'stop'
        else:
            msg = public.get_msg_gettext('SSH service turned on')
            act = 'start'

        if not os.path.exists('/etc/redhat-release'):
            public.ExecShell('service ssh ' + act);
        elif version.find(' 7.') != -1:
            public.ExecShell("systemctl " + act + " sshd.service")
        else:
            public.ExecShell("/etc/init.d/sshd " + act)

        public.write_log_gettext("Firewall manager", msg)
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    # 设置ping
    def SetPing(self, get):
        if get.status == '1':
            get.status = '0';
        else:
            get.status = '1';
        filename = '/etc/sysctl.conf'
        conf = public.readFile(filename)
        if conf.find('net.ipv4.icmp_echo') != -1:
            rep = r"net\.ipv4\.icmp_echo.*"
            conf = re.sub(rep, 'net.ipv4.icmp_echo_ignore_all=' + get.status, conf)
        else:
            conf += "\nnet.ipv4.icmp_echo_ignore_all=" + get.status

        public.writeFile(filename, conf)
        public.ExecShell('sysctl -p')
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    # 改远程端口
    def SetSshPort(self, get):
        # return public.returnMsg(False, public.lang("演示服务器，禁止此操作!"));
        port = get.port
        if int(port) < 22 or int(port) > 65535: return public.return_msg_gettext(False, public.lang("Port range must be between 22 and 65535!"));
        ports = ['21', '25', '80', '443', '8080', '888', '8888'];
        if port in ports: return public.return_msg_gettext(False, public.lang(""));

        file = '/etc/ssh/sshd_config'
        conf = public.readFile(file)

        rep = "#*Port\\s+([0-9]+)\\s*\n"
        conf = re.sub(rep, "Port " + port + "\n", conf)
        public.writeFile(file, conf)

        if self.__isFirewalld:
            self.__Obj.AddAcceptPort(port);
            public.ExecShell('setenforce 0');
            public.ExecShell('sed -i "s#SELINUX=enforcing#SELINUX=disabled#" /etc/selinux/config');
            public.ExecShell("systemctl restart sshd.service")
        elif self.__isUfw:
            public.ExecShell('ufw allow ' + port + '/tcp');
            public.ExecShell("service ssh restart")
        else:
            public.ExecShell('iptables -I INPUT -p tcp -m state --state NEW -m tcp --dport ' + port + ' -j ACCEPT')
            public.ExecShell("/etc/init.d/sshd restart")

        self.FirewallReload()
        public.M('firewall').where("ps=?", (public.get_msg_gettext('SSH Server'),)).setField('port', port)
        public.write_log_gettext("Firewall manager", "Successfully changed SSH port to [{}]!", (port,))
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    # 取SSH信息
    def GetSshInfo(self, get):
        file = '/etc/ssh/sshd_config'
        conf = public.readFile(file)
        rep = "#*Port\\s+([0-9]+)\\s*\n"
        port = re.search(rep, conf).groups(0)[0]
        import system
        panelsys = system.system();

        version = panelsys.GetSystemVersion();
        if os.path.exists('/usr/bin/apt-get'):
            status = public.ExecShell("service ssh status | grep -P '(dead|stop)'")
        else:
            if version.find(' 7.') != -1:
                status = public.ExecShell("systemctl status sshd.service | grep 'dead'")
            else:
                status = public.ExecShell("/etc/init.d/sshd status | grep -e 'stopped' -e '已停'")

        if len(status[0]) > 3:
            status = False
        else:
            status = True
        isPing = True
        try:
            file = '/etc/sysctl.conf'
            conf = public.readFile(file)
            rep = r"#*net\.ipv4\.icmp_echo_ignore_all\s*=\s*([0-9]+)"
            tmp = re.search(rep, conf).groups(0)[0]
            if tmp == '1': isPing = False
        except:
            isPing = True

        data = {}
        data['port'] = port
        data['status'] = status
        data['ping'] = isPing
        return data

    # 指定端口 放行IP
    def AddSpecifiesIp(self, get):
        '''
        get 里面 有  protocol type port  address ps   五个参数
        protocol == ['tcp','udp']
        types==['reject','accept'] # 放行和禁止
        port = 端口
        address  地址
        :param get :
        :return:
        '''

        # | ports | protocol | address_ip | types |
        flag = False
        import re
        # 判断端口是否正确
        rep = r"^\d{1,5}(:\d{1,5})?$"
        if not re.search(rep, get.port): return public.return_msg_gettext(False, public.lang("Port range is incorrect! should be between 100-65535"));

        # 判断IP是否正确
        rep2 = r"^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}(\/\d{1,2})?$"
        if not re.search(rep2, get.address): return public.return_msg_gettext(False, public.lang("IP address is illegal!"));
        import time
        ports = get.port
        ps = get.ps
        types = get.type
        protocol = get.protocol
        address_ip = get.address

        protocol_list = ['tcp', 'udp']
        type_list = ['reject', 'accept']
        # 判断type类型是否正确

        if types not in type_list: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))
        # 判断protocol 类型是否正确

        if protocol not in protocol_list: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))

        notudps = ['80', '443', '8888', '888', '39000:40000', '21', '22']
        if ports in notudps: flag = True

        # sql 查询
        # sql="select * from firewall where ports='%s' and address_ip='%s' and protocol='%s' and types='%s';" % (str(ports), str(address_ip), str(protocol), str(types))
        query_result = public.M('firewall').where('ports=? and address_ip=? and protocol=? and types=?',
                                                  (ports, address_ip, protocol, types)).count()
        # 这里大于0 表示存在
        if query_result > 0: return public.return_msg_gettext(False, public.lang("The port exists, no need to repeat the release!"))

        if self.__isUfw:
            if type == 'accept':
                public.ExecShell('ufw allow proto ' + protocol + ' from ' + address_ip + ' to any port ' + ports + '')

            else:
                public.ExecShell('ufw deny proto ' + protocol + ' from ' + address_ip + ' to any port ' + ports + '')

        else:
            if self.__isFirewalld:
                port = ports.replace(':', '-')
                self.__Obj.Add_Port_IP(port=ports, address=address_ip, pool=protocol, type=types)
            else:
                if type == 'accept':
                    public.ExecShell(
                        'iptables -I INPUT -s ' + address_ip + ' -p ' + protocol + ' --dport ' + ports + ' -j ACCEPT')
                else:
                    public.ExecShell(
                        'iptables -I INPUT -s ' + address_ip + ' -p ' + protocol + ' --dport ' + ports + ' -j DROP')

        public.write_log_gettext("Firewall manager", 'Successfully accepted port [{}]!', (ports,))
        addtime = time.strftime('%Y-%m-%d %X', time.localtime())
        result = public.M('firewall').add('protocol,types,port,address_ip,ps,addtime',
                                          (protocol, types, ports, address_ip, ps, addtime))
        self.FirewallReload()
        return public.return_msg_gettext(True, public.lang("Setup successfully!"))

    # 删除指定放行端口
    def DelSpecifiesIp(self, get):
        '''
        get 里面 有  protocol type port  address ps   五个参数
        protocol == ['tcp','udp']
        type==['reject','accept'] # 放行和禁止
        port = 端口
        address  地址
        :param get:
        :return:
        '''
        ports = get.port
        types = get.type
        protocol = get.protocol
        address_ip = get.address
        protocol_list = ['tcp', 'udp']
        id = get.id
        if protocol not in protocol_list: return public.return_msg_gettext(False, public.lang("Specified protocol does NOT exist!"))
        if self.__isUfw:
            if type == 'accept':
                public.ExecShell(
                    'ufw delete allow proto ' + protocol + ' from ' + address_ip + ' to any port ' + ports + '')
            else:
                public.ExecShell(
                    'ufw delete deny proto ' + protocol + ' from ' + address_ip + ' to any port ' + ports + '')
        else:
            if self.__isFirewalld:
                self.__Obj.Del_Port_IP(port=ports, address=address_ip, pool=protocol, type=types)

            else:
                if type == 'accept':
                    public.ExecShell(
                        'iptables -D INPUT -s ' + address_ip + ' -p ' + protocol + ' --dport ' + ports + ' -j ACCEPT')
                else:
                    public.ExecShell(
                        'iptables -D INPUT -s ' + address_ip + ' -p ' + protocol + ' --dport ' + ports + ' -j DROP')
        public.write_log_gettext("Firewall manager", 'FIREWALL_DROP_PORT', (ports,))
        public.M('firewall').where("id=?", (id,)).delete()

        self.FirewallReload()
        return public.return_msg_gettext(True, public.lang("Successfully deleted"))


