InPanel 1.2.1__py3-none-any.whl

inpanel/mod/firewall_macos.py

@@ -0,0 +1,213 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2020-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of The New BSD License.
+# The full license can be found in 'LICENSE'.
+'''Module for macOS Firewall Management'''
+
+import shutil
+from typing import Tuple, List, Dict
+
+from .firewall_base import FirewallManager
+from .system import get_os_family
+
+
+class MacosFirewallPM(FirewallManager):
+    """macOS 防火墙管理器"""
+    
+    SOCKETFILTERFW = '/usr/libexec/ApplicationFirewall/socketfilterfw'
+    PFCTL = '/sbin/pfctl'
+    
+    def detect(self) -> bool:
+        """检测当前系统是否为 macOS"""
+        return get_os_family() == 'darwin'
+    
+    def status(self) -> Tuple[bool, str]:
+        """获取防火墙状态"""
+        if shutil.which(self.SOCKETFILTERFW):
+            success, output = self._run_cmd([self.SOCKETFILTERFW, '--getglobalstate'])
+            if success:
+                if 'enabled' in output.lower():
+                    return (True, '运行中')
+                elif 'disabled' in output.lower():
+                    return (True, '已停止')
+        return (False, '未知')
+    
+    def enable(self) -> Tuple[bool, str]:
+        """启用防火墙（开机自启）"""
+        if shutil.which(self.SOCKETFILTERFW):
+            success, output = self._run_cmd(['sudo', self.SOCKETFILTERFW, '--setglobalstate', 'on'])
+            if success:
+                return (True, '防火墙已启用')
+        return (False, '无法启用防火墙')
+    
+    def disable(self) -> Tuple[bool, str]:
+        """禁用防火墙（关闭开机自启）"""
+        if shutil.which(self.SOCKETFILTERFW):
+            success, output = self._run_cmd(['sudo', self.SOCKETFILTERFW, '--setglobalstate', 'off'])
+            if success:
+                return (True, '防火墙已禁用')
+        return (False, '无法禁用防火墙')
+    
+    def start(self) -> Tuple[bool, str]:
+        """启动防火墙服务"""
+        return self.enable()
+    
+    def stop(self) -> Tuple[bool, str]:
+        """停止防火墙服务"""
+        return self.disable()
+    
+    def restart(self) -> Tuple[bool, str]:
+        """重启防火墙服务"""
+        self.disable()
+        return self.enable()
+    
+    def add_rule(self, port: int, protocol: str = 'tcp', zone: str = '') -> Tuple[bool, str]:
+        """添加端口规则 - macOS 使用 pf"""
+        if shutil.which(self.PFCTL):
+            rule = f'pass in proto {protocol} from any to any port {port}'
+            success, output = self._run_cmd(['sudo', self.PFCTL, '-t', 'InPanel', '-T', 'add', rule])
+            if not success:
+                success, output = self._run_cmd(['sudo', self.PFCTL, '-f', '/dev/stdin'], input=rule)
+                if success:
+                    return (True, f'已添加端口规则: {port}/{protocol}')
+        return (False, '无法添加端口规则')
+    
+    def remove_rule(self, port: int, protocol: str = 'tcp', zone: str = '') -> Tuple[bool, str]:
+        """移除端口规则"""
+        if shutil.which(self.PFCTL):
+            success, output = self._run_cmd(['sudo', self.PFCTL, '-t', 'InPanel', '-T', 'delete', f'pass in proto {protocol} from any to any port {port}'])
+            if success:
+                return (True, f'已移除端口规则: {port}/{protocol}')
+        return (False, '无法移除端口规则')
+    
+    def add_ip_rule(self, ip: str, action: str = 'allow') -> Tuple[bool, str]:
+        """添加IP规则"""
+        if shutil.which(self.PFCTL):
+            if action == 'allow':
+                rule = f'pass in from {ip} to any'
+            else:
+                rule = f'block in from {ip} to any'
+            success, output = self._run_cmd(['sudo', self.PFCTL, '-t', 'InPanel', '-T', 'add', rule])
+            if success:
+                return (True, f'已添加IP规则: {action} {ip}')
+        return (False, '无法添加IP规则')
+    
+    def remove_ip_rule(self, ip: str) -> Tuple[bool, str]:
+        """移除IP规则"""
+        if shutil.which(self.PFCTL):
+            success, output = self._run_cmd(['sudo', self.PFCTL, '-t', 'InPanel', '-T', 'delete', f'pass in from {ip} to any'])
+            if not success:
+                success, output = self._run_cmd(['sudo', self.PFCTL, '-t', 'InPanel', '-T', 'delete', f'block in from {ip} to any'])
+            if success:
+                return (True, f'已移除IP规则: {ip}')
+        return (False, '无法移除IP规则')
+    
+    def remove_app_rule(self, app_path: str) -> Tuple[bool, str]:
+        """移除应用规则"""
+        if shutil.which(self.SOCKETFILTERFW):
+            success, output = self._run_cmd(['sudo', self.SOCKETFILTERFW, '--remove', app_path])
+            if success:
+                return (True, f'已移除应用规则: {app_path}')
+        return (False, '无法移除应用规则')
+    
+    def list_rules(self) -> Tuple[bool, str]:
+        """列出所有规则"""
+        rules = []
+        
+        if shutil.which(self.SOCKETFILTERFW):
+            success, output = self._run_cmd([self.SOCKETFILTERFW, '--listapps'])
+            if success:
+                rules.append("=== 应用规则 ===")
+                rules.append(output)
+        
+        if shutil.which(self.PFCTL):
+            success, output = self._run_cmd(['sudo', self.PFCTL, '-s', 'rules'])
+            if success:
+                rules.append("\n=== PF 端口规则 ===")
+                rules.append(output)
+        
+        return (True, '\n'.join(rules) if rules else '无规则')
+    
+    def list_zones(self) -> Tuple[bool, List[str]]:
+        """macOS 不使用区域概念"""
+        return (True, [])
+    
+    def get_default_zone(self) -> Tuple[bool, str]:
+        """macOS 不使用区域概念"""
+        return (True, '')
+    
+    def parse_rules(self) -> List[Dict]:
+        """解析规则列表为结构化数据"""
+        rules = []
+        success, output = self.list_rules()
+        
+        if not success:
+            return rules
+        
+        lines = output.split('\n')
+        app_section = False
+        pf_section = False
+        current_app_path = ''
+        
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+            
+            if '=== 应用规则 ===' in line:
+                app_section = True
+                pf_section = False
+                continue
+            if '=== PF 端口规则 ===' in line:
+                app_section = False
+                pf_section = True
+                continue
+            
+            if app_section and line:
+                if 'total number of apps' in line:
+                    continue
+                
+                import re
+                app_match = re.match(r'\d+\s*:\s*(/.+)', line)
+                if app_match:
+                    current_app_path = app_match.group(1).strip()
+                    app_name = current_app_path.split('/')[-1].replace('.app', '')
+                    rules.append({
+                        'type': 'app',
+                        'port': '',
+                        'protocol': '',
+                        'action': 'allow',
+                        'ip': '',
+                        'zone': '',
+                        'description': app_name,
+                        'app_path': current_app_path
+                    })
+                elif 'Allow incoming connections' in line and current_app_path:
+                    pass
+            
+            if pf_section and line:
+                import re
+                port_match = re.search(r'port\s+(\d+)', line)
+                proto_match = re.search(r'proto\s+(\w+)', line)
+                ip_match = re.search(r'from\s+(\d+\.\d+\.\d+\.\d+)', line)
+                
+                rule_type = 'port' if port_match else 'ip'
+                port = port_match.group(1) if port_match else ''
+                protocol = proto_match.group(1) if proto_match else 'tcp'
+                ip = ip_match.group(1) if ip_match else ''
+                action = 'allow' if 'pass' in line else 'deny'
+                
+                rules.append({
+                    'type': rule_type,
+                    'port': port,
+                    'protocol': protocol,
+                    'action': action,
+                    'ip': ip,
+                    'zone': '',
+                    'description': line
+                })
+        
+        return rules

inpanel/mod/firewall_ufw.py

@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2020-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of The New BSD License.
+# The full license can be found in 'LICENSE'.
+'''Module for UFW (Uncomplicated Firewall) Management'''
+
+import shutil
+from typing import Tuple, List, Dict
+
+from .firewall_base import FirewallManager
+from .system import is_debian_family
+
+
+class UfwPM(FirewallManager):
+    """Debian 9-13 / Ubuntu 18.04+ / Linux Mint：ufw"""
+    
+    def detect(self) -> bool:
+        if not shutil.which("ufw"):
+            return False
+        return is_debian_family()
+    
+    def status(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "status", "verbose"])
+    
+    def enable(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "enable"])
+    
+    def disable(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "disable"])
+    
+    def start(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "enable"])
+    
+    def stop(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "disable"])
+    
+    def restart(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "reload"])
+    
+    def add_rule(self, port: int, protocol: str = 'tcp', zone: str = '') -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "allow", f"{port}/{protocol}"])
+    
+    def remove_rule(self, port: int, protocol: str = 'tcp', zone: str = '') -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "delete", "allow", f"{port}/{protocol}"])
+    
+    def add_ip_rule(self, ip: str, action: str = 'allow') -> Tuple[bool, str]:
+        if action == 'allow':
+            return self._run_cmd(["ufw", "allow", "from", ip])
+        else:
+            return self._run_cmd(["ufw", "deny", "from", ip])
+    
+    def remove_ip_rule(self, ip: str) -> Tuple[bool, str]:
+        success1, output1 = self._run_cmd(["ufw", "delete", "allow", "from", ip])
+        if not success1:
+            success2, output2 = self._run_cmd(["ufw", "delete", "deny", "from", ip])
+            return (success2, output2)
+        return (success1, output1)
+    
+    def list_rules(self) -> Tuple[bool, str]:
+        return self._run_cmd(["ufw", "status"])
+    
+    def list_zones(self) -> Tuple[bool, List[str]]:
+        return (False, [])
+    
+    def get_default_zone(self) -> Tuple[bool, str]:
+        return (False, "Not supported")
+    
+    def parse_rules(self) -> List[Dict]:
+        rules = []
+        success, output = self._run_cmd(["ufw", "status", "numbered"])
+        if not success:
+            return rules
+        
+        lines = output.split('\n')
+        for line in lines:
+            line = line.strip()
+            if not line or line.startswith('Status:') or line.startswith('To') or line.startswith('--'):
+                continue
+            
+            parts = line.split()
+            if len(parts) < 4:
+                continue
+            
+            num = parts[0].rstrip('[')
+            target = parts[1]
+            action = parts[2]
+            source = parts[3]
+            
+            protocol = 'tcp'
+            port = ''
+            ip = ''
+            
+            if '/' in target:
+                port, proto = target.split('/', 1)
+                protocol = proto
+            elif target.isdigit():
+                port = int(target)
+            else:
+                rules.append({
+                    'type': 'service',
+                    'service': target,
+                    'port': '',
+                    'protocol': '',
+                    'action': action.lower(),
+                    'ip': '',
+                    'zone': '',
+                    'description': f"{action}: {target} from {source}"
+                })
+                continue
+            
+            if source != 'Anywhere':
+                ip = source
+            
+            rules.append({
+                'type': 'port',
+                'service': '',
+                'port': int(port) if port.isdigit() else port,
+                'protocol': protocol,
+                'action': action.lower(),
+                'ip': ip,
+                'zone': '',
+                'description': f"{action}: {port}/{protocol} from {source}"
+            })
+        
+        return rules

inpanel/mod/ftp.py

@@ -0,0 +1,231 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2019-2026 Jackson Dou
+# All rights reserved.
+#
+# InPanel is distributed under the terms of the (new) BSD License.
+# The full license can be found in 'LICENSE'.
+
+'''Module for FTP'''
+
+import time
+from pathlib import Path
+from ftplib import FTP
+
+
+class FTPClient():
+    '''FTP Module'''
+
+    def __init__(self, host, user, pswd):
+        self.ftp = None
+        self.host = host
+        self.user = user
+        self.pswd = pswd
+        self.init_client()
+
+    def __delete__(self, instance):
+        self.quit()
+
+    def init_client(self):
+        '''init instance'''
+        try:
+            self.ftp = FTP(self.host)
+            self.ftp.login(self.user, self.pswd)
+            print('connected: "%s"' % self.host)
+            return self.ftp
+        except:
+            print('connect failed.')
+            print('Please check if the host, user and password is correct.')
+            return False
+
+    def reconnect(self):
+        '''reconnect to ftp'''
+        try:
+            self.ftp.login(self.user, self.pswd)
+            print('reconnected: "%s"' % self.host)
+        except:
+            print('reconnect failed')
+            return False
+        return True
+
+    def disconnect(self):
+        '''close connection'''
+        try:
+            self.ftp.close()
+            print('disconnect successful')
+        except:
+            print('disconnect failed')
+            return False
+        return True
+
+    def quit(self):
+        '''quit and close connection'''
+        try:
+            self.ftp.quit()
+            print('quit successful')
+        except:
+            print('quit failed')
+            return False
+        return True
+
+    def cd(self, path):
+        '''change directory'''
+        if not path:
+            return False
+        try:
+            return self.ftp.cwd(path)
+        except:
+            print('change directory failed: "%s"' % path)
+            return False
+
+    def pwd(self):
+        '''get current path'''
+        try:
+            return self.ftp.pwd()
+        except:
+            print('get path failed')
+            return None
+
+    def ls(self, path='', onlyname=False):
+        '''get list of files and directories'''
+        try:
+            if onlyname is True:
+                return self.ftp.nlst()
+            return self.ftp.dir(path)
+        except:
+            print('get list failed: "%s"' % path)
+            return None
+
+    def mkd(self, path):
+        '''create a directory, return its full pathname.
+        '''
+        if not path:
+            return False
+        try:
+            return self.ftp.mkd(path)
+        except:
+            print('create failed: "%s"' % path)
+            return False
+
+    def rename(self, oldname, newname):
+        '''rename a file or directory.'''
+        try:
+            self.ftp.rename(oldname, newname)
+            print('rename successful: from "%s" to "%s"' % (oldname, newname))
+        except:
+            print('rename failed: from "%s" to "%s"' % (oldname, newname))
+            return False
+        return True
+
+    def rm(self, filename):
+        '''delete a file or directory.'''
+        try:
+            if self.ftp.delete(filename):
+                print('delete successful: "%s"' % filename)
+                return True
+            else:
+                print('delete failed: "%s"' % filename)
+                return False
+        except:
+            print('delete error: "%s"' % filename)
+            return False
+
+    def rmd(self, filename):
+        '''delete a file or directory.'''
+        try:
+            if self.ftp.delete(filename):
+                print('delete successful: "%s"' % filename)
+                return True
+            else:
+                print('delete failed: "%s"' % filename)
+                return False
+        except:
+            print('delete failed: "%s"' % filename)
+            return False
+
+    def upload(self, filepath):
+        '''upload local file to FTP'''
+        try:
+            with open(filepath, 'rb') as source:
+                target = Path(filepath).name
+                self.ftp.storbinary('STOR %s' % target, source)
+        except:
+            print('upload failed: "%s"' % filepath)
+            return False
+        print('upload successful: "%s"' % filepath)
+        return True
+
+    def download(self, filename, localfile=''):
+        '''download FTP file to local'''
+        localfile = str(Path(filename if not localfile else localfile))
+        try:
+            with open(localfile, 'wb') as target:
+                self.ftp.retrbinary("RETR %s" % filename, target.write)
+        except:
+            print('download failed: from "%s" to "%s"' % (filename, localfile))
+            return False
+        print('download successful: from "%s" to "%s"' % (filename, localfile))
+        return True
+
+    def find(self, filename):
+        '''find file on FTP'''
+        files = self.ftp.nlst()
+        if filename in files:
+            return True
+        else:
+            return False
+
+def uploadtoftp(address, account, password, source, target):
+    try:
+        ftp = FTPClient(address, account, password)
+        # print('1')
+        ftp.cd(str(Path(target)))
+        # print('2')
+        ftp.upload(source)
+        # print('3')
+        ftp.quit()
+        # print('4')
+        return True
+    except:
+        return False
+
+def uploadtoftpa(*args, **kwds):
+    time.sleep(5)
+    return True
+
+def test():
+    host = ''
+    user = ''
+    pswd = ''
+    ftp = FTPClient(host, user, pswd)
+    # ftp.ls()
+    ftp.cd('/Backup')
+    # print(ftp.pwd())
+    # ftp.ls()
+    # ftp.rename('1234567/cc.txt', 'ccaa.txt')
+    # ftp.rename('ccaa.txt', '1234567/afasff.txt')
+    # ftp.mkd('abc/aaa/ccc')  # 创建文件夹
+    # ftp.mkd('ccc')  # 创建文件夹
+    # ftp.upload('./app.icns')  # 上传文件
+    # ftp.upload('./bbb.txt')  # 上传文件
+
+    # ftp.upload('111.txt')  # 上传文件
+    # ftp.upload('222.txt')  # 上传文件
+    # ftp.upload('bbb.txt')  # 上传文件
+    # ftp.download('111.txt', '111aaa.txt')  # 下载文件重命名
+    # ftp.download('222.txt', '222aaa.txt')  # 下载文件重命名
+    # ftp.download('111.txt')  # 下载文件覆盖
+    # ftp.download('222.txt')  # 下载文件覆盖
+    # ftp.download('bbb.txt')  # 下载文件覆盖
+
+    # ftp.rm('bbb.txt')  # 删除文件
+    # ftp.rmd("ccc")  # 删除目录
+
+    print(ftp.find('aaa.txt'))
+
+
+    ftp.quit()  # 退出
+
+
+if __name__ == '__main__':
+    test()