diff --git a/balancer.py b/balancer.py index e9f48ad..de63d79 100644 --- a/balancer.py +++ b/balancer.py @@ -5,25 +5,73 @@ from itertools import cycle, izip rDownloadURL = "https://bitbucket.org/xoceunder/xtream-ui-install/raw/main/sub_xui_xoceunder.zip" rPackages = ["python", "libcurl4", "libxslt1-dev", "libgeoip-dev", "libonig-dev", "e2fsprogs", "wget", "mcrypt", "nscd", "htop", "zip", "unzip", "mc"] +rRemove = ["mysql-server"] +rSysCtl = "# XtreamCodes\n\nnet.ipv4.tcp_congestion_control = bbr\nnet.core.default_qdisc = fq\nnet.ipv4.tcp_rmem = 8192 87380 134217728\nnet.ipv4.udp_rmem_min = 16384\nnet.core.rmem_default = 262144\nnet.core.rmem_max = 268435456\nnet.ipv4.tcp_wmem = 8192 65536 134217728\nnet.ipv4.udp_wmem_min = 16384\nnet.core.wmem_default = 262144\nnet.core.wmem_max = 268435456\nnet.core.somaxconn = 1000000\nnet.core.netdev_max_backlog = 250000\nnet.core.optmem_max = 65535\nnet.ipv4.tcp_max_tw_buckets = 1440000\nnet.ipv4.tcp_max_orphans = 16384\nnet.ipv4.ip_local_port_range = 2000 65000\nnet.ipv4.tcp_no_metrics_save = 1\nnet.ipv4.tcp_slow_start_after_idle = 0\nnet.ipv4.tcp_fin_timeout = 15\nnet.ipv4.tcp_keepalive_time = 300\nnet.ipv4.tcp_keepalive_probes = 5\nnet.ipv4.tcp_keepalive_intvl = 15\nfs.file-max=20970800\nfs.nr_open=20970800\nfs.aio-max-nr=20970800\nnet.ipv4.tcp_timestamps = 1\nnet.ipv4.tcp_window_scaling = 1\nnet.ipv4.tcp_mtu_probing = 1\nnet.ipv4.route.flush = 1\nnet.ipv6.route.flush = 1" + +initd_script = """#!/bin/sh +SCRIPT=/home/xtreamcodes/iptv_xtream_codes/xtreamcodes +NAME=xtreamcodes + +case "$1" in + start) + $SCRIPT start + ;; + stop) + $SCRIPT stop + ;; + restart) + $SCRIPT restart + ;; + reload) + $SCRIPT reload + ;; + status) + pids=$(pgrep -u xtreamcodes nginx | wc -l) + if [ "$pids" -gt 0 ]; then + echo "$NAME is running" + else + echo "$NAME is stopped" + fi + ;; + *) + echo "Usage: /etc/init.d/$NAME {start|stop|restart|reload|status}" + exit 1 + ;; +esac + +exit 0 + +""" def getVersion(): try: return subprocess.check_output("lsb_release -d".split()).split(":")[-1].strip() except: return "" + +def is_installed(package_name): + try: + subprocess.run(['dpkg', '-s', package_name], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE) + return True + except subprocess.CalledProcessError: + return False def prepare(): - global rPackages - for rFile in ["/var/lib/dpkg/lock-frontend", "/var/cache/apt/archives/lock", "/var/lib/dpkg/lock"]: - try: os.remove(rFile) - except: pass - os.system("apt-get update > /dev/null") - os.system("apt-get remove --auto-remove libcurl4 -y > /dev/null") - for rPackage in rPackages: os.system("apt-get install %s -y > /dev/null" % rPackage) - os.system("wget -q -O /tmp/libpng12.deb https://xtreamtools.org/XCodes/libpng12-0_1.2.54-1ubuntu1_amd64.deb") - os.system("dpkg -i /tmp/libpng12.deb > /dev/null") + global rPackages, rRemove + for rFile in ["/var/lib/dpkg/lock-frontend","/var/cache/apt/archives/lock","/var/lib/dpkg/lock"]: + if os.path.exists(rFile): + try: + os.remove(rFile) + except: + pass + os.system("apt-get update > /dev/null 2>&1") + os.system("apt-get -y full-upgrade > /dev/null 2>&1") + for rPackage in rRemove: + if is_installed(rPackage): + os.system("sudo DEBIAN_FRONTEND=noninteractive apt-get remove %s > /dev/null" % rPackage) + for rPackage in rPackages: + if not is_installed(rPackage): + subprocess.run(f"sudo DEBIAN_FRONTEND=noninteractive apt-get install {rPackage} -yq > /dev/null 2>&1", shell=True) os.system("apt-get install -y > /dev/null") # Clean up above - try: os.remove("/tmp/libpng12.deb") - except: pass - os.system("adduser --system --shell /bin/false --group --disabled-login xtreamcodes 2> /dev/null") + os.system("adduser --system --shell /bin/false --group --disabled-login xtreamcodes > /dev/null") if not os.path.exists("/home/xtreamcodes"): os.mkdir("/home/xtreamcodes") return True @@ -46,57 +94,67 @@ def encrypt(rHost="127.0.0.1", rUsername="user_iptvpro", rPassword="", rDatabase rf.close() def configure(): - rYou = "https://yt-dl.org/downloads/latest/youtube-dl" if not "/home/xtreamcodes/iptv_xtream_codes/" in open("/etc/fstab").read(): rFile = open("/etc/fstab", "a") rFile.write("tmpfs /home/xtreamcodes/iptv_xtream_codes/streams tmpfs defaults,noatime,nosuid,nodev,noexec,mode=1777,size=90% 0 0\ntmpfs /home/xtreamcodes/iptv_xtream_codes/tmp tmpfs defaults,noatime,nosuid,nodev,noexec,mode=1777,size=2G 0 0") rFile.close() - if not "/sbin/iptables" in open("/etc/sudoers").read(): os.system('sed -i "s|xtreamcodes|#xtreamcodes|g" /etc/sudoers && echo "xtreamcodes ALL=(root) NOPASSWD: /sbin/iptables" >> /etc/sudoers') - if not os.path.exists("/etc/init.d/xtreamcodes"): os.system("touch /etc/init.d/xtreamcodes") - if not "Provides" in open("/etc/init.d/xtreamcodes").read(): - os.system("rm /etc/init.d/xtreamcodes") - rStart = open("/etc/init.d/xtreamcodes", "w") - rStart.write("#!/bin/bash\n### BEGIN INIT INFO\n# Provides: xtreamcodes\n# Required-Start: $all\n# Required-Stop:\n# Default-Start: 2 3 4 5\n# Default-Stop:\n# Short-Description: Run /etc/init.d/xtreamcodes if it exist\n### END INIT INFO\n/home/xtreamcodes/iptv_xtream_codes/start_services.sh > /dev/null") - rStart.close() - os.system("chmod 755 /etc/init.d/xtreamcodes 2>/dev/null") - os.system("update-rc.d xtreamcodes defaults 2>/dev/null") - try: os.remove("/usr/bin/ffmpeg 2>/dev/null || /usr/bin/ffmpeg 2>/dev/null") + if not "xtreamcodes" in open("/etc/sudoers").read(): os.system('echo "xtreamcodes ALL = (root) NOPASSWD: /sbin/iptables, /usr/bin/chattr, /usr/bin/python3, /usr/bin/python" >> /etc/sudoers') + if not os.path.exists("/home/xtreamcodes/iptv_xtream_codes/xtreamcodes"): + os.system("wget -q https://github.com/xoceunder/Xtream-UI-install/raw/main/xtreamcodes -O /home/xtreamcodes/iptv_xtream_codes/xtreamcodes") + os.system("sudo chmod +x /home/xtreamcodes/iptv_xtream_codes/xtreamcodes") + if not os.path.exists("/etc/init.d/xtreamcodes"): + with io.open("/etc/init.d/xtreamcodes", "w", encoding="utf-8") as f: + f.write(initd_script) + os.system("sudo chmod +x /etc/init.d/xtreamcodes") + os.system("sudo update-rc.d xtreamcodes defaults") + os.system("sudo modprobe ip_conntrack") + rFile = open("/etc/sysctl.conf", "w", encoding="utf-8") + rFile.write(rSysCtl) + rFile.close() + os.system("sudo sysctl -p >/dev/null 2>&1") + rFile = open("/home/xtreamcodes/iptv_xtream_codes/sysctl.on", "w") + rFile.close() + if not "DefaultLimitNOFILE=655350" in open("/etc/systemd/system.conf").read(): + os.system('sudo echo "\nDefaultLimitNOFILE=655350" >> "/etc/systemd/system.conf"') + os.system('sudo echo "\nDefaultLimitNOFILE=655350" >> "/etc/systemd/user.conf"') + try: os.remove("/usr/bin/ffmpeg") except: pass if not os.path.exists("/home/xtreamcodes/iptv_xtream_codes/tv_archive"): os.mkdir("/home/xtreamcodes/iptv_xtream_codes/tv_archive/") os.system("ln -s /home/xtreamcodes/iptv_xtream_codes/bin/ffmpeg /usr/bin/") - os.system("chattr -i /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb") - os.system("rm /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb") - os.system("wget -q https://bitbucket.org/xoceunder/xtream-ui-install/raw/main/GeoLite2.mmdb -O /home/xtreamcodes/iptv_xtream_codes/XCodes/GeoLite2.mmdb") + os.system("wget -q https://bitbucket.org/xoceunder/xtream-ui-install/raw/main/GeoLite2.mmdb -O /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb") + os.system("chattr -ai /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb > /dev/null") + os.system("sudo chmod 0777 /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb > /dev/null") os.system("wget -q https://bitbucket.org/xoceunder/xtream-ui-install/raw/main/pid_monitor.php -O /home/xtreamcodes/iptv_xtream_codes/crons/pid_monitor.php") - os.system("rm /usr/local/bin/youtube-dl 2>/dev/null") - os.system('wget -q -O "/usr/local/bin/youtube-dl" "%s"' % rYou) - os.system("sudo chmod a+rx /usr/local/bin/youtube-dl") - if not "/home/xtreamcodes 2>/dev/null" in open("/home/xtreamcodes/iptv_xtream_codes/start_services.sh").read(): - os.system("sed -i 's|chown -R xtreamcodes:xtreamcodes /home/xtreamcodes|chown -R xtreamcodes:xtreamcodes /home/xtreamcodes 2>/dev/null|g' /home/xtreamcodes/iptv_xtream_codes/start_services.sh") - os.system("sed -i 's|chmod -R 777 /home/xtreamcodes|chmod -R 777 /home/xtreamcodes 2>/dev/null|g' /home/xtreamcodes/iptv_xtream_codes/start_services.sh") - os.system("chmod +x /home/xtreamcodes/iptv_xtream_codes/start_services.sh") + os.system("chown xtreamcodes:xtreamcodes -R /home/xtreamcodes > /dev/null") + os.system("chmod -R 0777 /home/xtreamcodes > /dev/null") + os.system("sed -i 's|chown -R xtreamcodes:xtreamcodes /home/xtreamcodes|chown -R xtreamcodes:xtreamcodes /home/xtreamcodes 2>/dev/null|g' /home/xtreamcodes/iptv_xtream_codes/start_services.sh") + os.system("chmod +x /home/xtreamcodes/iptv_xtream_codes/start_services.sh > /dev/null") os.system("mount -a") - os.system("chown -R xtreamcodes:xtreamcodes /home/xtreamcodes 2>/dev/null") - os.system("chmod -R 0777 /home/xtreamcodes 2>/dev/null") - os.system("chattr +i /home/xtreamcodes/iptv_xtream_codes/GeoLite2.mmdb") - os.system("sed -i 's|echo \"Xtream Codes Reborn\";|header(\"Location: https://www.google.com/\");|g' /home/xtreamcodes/iptv_xtream_codes/wwwdir/index.php 2>/dev/null") - if not "xtream-codes.com" in open("/etc/hosts").read(): os.system('echo "127.0.0.1 xtream-codes.com" >> /etc/hosts') + os.system("chown xtreamcodes:xtreamcodes /home/xtreamcodes/iptv_xtream_codes/config") + os.system("chmod 0700 /home/xtreamcodes/iptv_xtream_codes/config > /dev/null") + os.system("sed -i 's|echo \"Xtream Codes Reborn\";|header(\"Location: https://www.google.com/\");|g' /home/xtreamcodes/iptv_xtream_codes/wwwdir/index.php") if not "api.xtream-codes.com" in open("/etc/hosts").read(): os.system('echo "127.0.0.1 api.xtream-codes.com" >> /etc/hosts') if not "downloads.xtream-codes.com" in open("/etc/hosts").read(): os.system('echo "127.0.0.1 downloads.xtream-codes.com" >> /etc/hosts') if not "xtream-codes.com" in open("/etc/hosts").read(): os.system('echo "127.0.0.1 xtream-codes.com" >> /etc/hosts') + if not "@reboot root /etc/init.d/xtreamcodes start" in open("/etc/crontab").read(): os.system('echo "@reboot root /etc/init.d/xtreamcodes start" >> /etc/crontab') + +def start(): os.system("sudo /etc/init.d/xtreamcodes start > /dev/null") def setPorts(rPorts): os.system("sed -i 's/listen 25461;/listen %d;/g' /home/xtreamcodes/iptv_xtream_codes/nginx/conf/nginx.conf" % rPorts[0]) os.system("sed -i 's/:25461/:%d/g' /home/xtreamcodes/iptv_xtream_codes/nginx_rtmp/conf/nginx.conf" % rPorts[0]) os.system("sed -i 's/listen 25463 ssl;/listen %d ssl;/g' /home/xtreamcodes/iptv_xtream_codes/nginx/conf/nginx.conf" % rPorts[1]) os.system("sed -i 's/listen 25462;/listen %d;/g' /home/xtreamcodes/iptv_xtream_codes/nginx_rtmp/conf/nginx.conf" % rPorts[2]) - -def start(): - os.system("chown xtreamcodes:xtreamcodes /home/xtreamcodes/iptv_xtream_codes/config") - os.system("chmod 777 /home/xtreamcodes/iptv_xtream_codes/config") - os.system("/home/xtreamcodes/iptv_xtream_codes/start_services.sh 2>/dev/null") - + if __name__ == "__main__": + try: rVersion = os.popen('lsb_release -sr').read().strip() + except: rVersion = None + if rVersion == "20.04": + rPackages.append("libzip5") + elif rVersion == "18.04": + rPackages.append("libzip4") + else: + rPackages.append("libzip5") rHost = sys.argv[1] rPort = int(sys.argv[2]) rUsername = sys.argv[3]