两学一做考学网站,网站建设的费用记什么科目,网站免费维护建立网站,免费做电子目录的网站注意#xff1a;该脚本会清空linux防火墙的filter表的规则和用户自定义链路
脚本的效果是将端口限制为仅服务器内部访问
可以提供ip地址白名单 具体脚本#xff1a;
#!/usr/bin/python3
import argparse, subprocess, sys, redef popen(cmd):global resulttry:result su…注意该脚本会清空linux防火墙的filter表的规则和用户自定义链路
脚本的效果是将端口限制为仅服务器内部访问
可以提供ip地址白名单 具体脚本
#!/usr/bin/python3
import argparse, subprocess, sys, redef popen(cmd):global resulttry:result subprocess.run(cmd, shellTrue, stdoutsubprocess.PIPE, stderrsubprocess.PIPE, encodingutf-8, checkTrue)except Exception as e:raise Exception(e.stderr)file_count result.stdout.strip() # 假设grep -c 的输出只有一行且没有额外的空格或换行符return file_countdef assignment():parser argparse.ArgumentParser(descriptionmanual to this script)parser.add_argument(--ip_white_list, typestr, default)parser.add_argument(--port_list, typestr, default)args parser.parse_args()ip_white_list args.ip_white_listport_list args.port_listif not port_list :raise Exception(--port_list不能为空)ip_white_list f{ip_white_list},.split(,)port_list f{port_list},.split(,)ip_white_list set(filter(lambda x:x!,ip_white_list))port_list set(filter(lambda x:x!,port_list))for port in port_list:if re.search(r[^0-9], port):raise Exception(--port_list 不是合法的端口)passif ip_white_list :for ip_white in ip_white_list:if re.search(r[^0-9.], ip_white):raise Exception(--ip_white_list 不是合法的ip地址)passreturn ip_white_list, port_listdef clear_iptables_rule():popen(iptables -F )popen(iptables -X )passdef ip_white(ip_white_list, port_list):popen(iptables -t filter -P INPUT ACCEPT)popen(iptables -t filter -P OUTPUT ACCEPT)popen(iptables -A INPUT -p tcp -s 127.0.0.1 -j ACCEPT)popen(iptables -A INPUT -p tcp -s $(ip r g 1|cut -d -f 7) -j ACCEPT)if ip_white_list:for ip in ip_white_list:for port in port_list:popen(fiptables -A INPUT -p tcp -s {ip} --dport {port} -j ACCEPT)def port_disables(port_list):if port_list:for port in port_list:popen(fiptables -A INPUT -p tcp --dport {port} -j REJECT )passdef clear_script():popen(fecho {sys.argv[0]})passif __name__ __main__:ip_white_list, port_list assignment()clear_iptables_rule()ip_white(ip_white_list, port_list)port_disables(port_list)clear_script()pass执行脚本示例选项
./ta --port_list 8081,8080 --ip_white_list 192.168.0.1,120.0.0.2
