From d097c2a15a859de50fb756750e418a4e854303b8 Mon Sep 17 00:00:00 2001
From: HITSword <admin@huayizhiyun.com>
Date: Thu, 5 Mar 2020 17:07:06 +0800
Subject: [PATCH] =?UTF-8?q?=E5=86=99=E5=85=A5=E8=84=9A=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: HITSword <admin@huayizhiyun.com>
---
 udptools.sh | 41 +++++++++++++++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/udptools.sh b/udptools.sh
index 2a6bd38..5abb92f 100644
--- a/udptools.sh
+++ b/udptools.sh
@@ -51,7 +51,40 @@ buildServer()
 -k $PASSWD
 # 密码
 --cipher-mode xor
-# 简单xor加密" > /usr/local/udptools/conf/udp2raw-s${MPORT}.conf
+# 简单xor加密
+--fix-gro
+# 修复粘包" > /usr/local/udptools/conf/udp2raw-s${MPORT}.conf
+    echo "#!/bin/bash
+PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
+export PATH
+#进程名
+PROG=Udp2Raw-Server-${MPORT}
+#BIN路径
+BIN_FILE=/usr/local/udptools/bin/udp2raw
+#配置路径
+CONFIG_FILE=/usr/local/udptools/conf/udp2raw-s${MPORT}.conf
+#日志路径
+LOG_FILE=/usr/local/udptools/log/udp2raw-s${MPORT}.log
+#PID路径
+PID_FILE=/usr/local/udptools/pid/udp2raw-s${MPORT}.pid
+    " > /usr/local/udptools/udp2raw-s${MPORT}.sh
+    cat >> /usr/local/udptools/udp2raw-s${MPORT}.sh <<EOF
+checkSet(){
+  #获取监听端口
+  SERVER_PORT=`cat \$CONFIG_FILE | grep '\-l ' | awk -F ":" '{print $2}'`
+  #检查iptables规则
+  IPTALBES=`iptables -nvL | grep DROP | grep tcp | grep \$SERVER_PORT`
+  if [ ! -n "\$IPTALBES" ]; then
+    echo "Adding iptables rules."
+    #添加iptables规则
+    RULES=`\$BIN_FILE --conf-file \$CONFIG_FILE -g | grep iptables |grep -v rule`
+    \$RULES
+  fi
+  #赋权
+  setcap cap_net_raw+ep \$BIN_FILE
+}
+    EOF
+    chmod +x /usr/local/udptools/udp2raw-s${MPORT}.sh
 
   #判断服务模式
   if pgrep systemd-journal > /dev/null; then
@@ -76,7 +109,9 @@ buildClient()
 -k $PASSWD
 # 密码
 --cipher-mode xor
-# 简单xor加密" > /usr/local/udptools/conf/udp2raw-c${MPORT}.conf
+# 简单xor加密
+--fix-gro
+# 修复粘包" > /usr/local/udptools/conf/udp2raw-c${MPORT}.conf
 
   #判断服务模式
   if pgrep systemd-journal > /dev/null; then
@@ -130,6 +165,7 @@ case $RUNMODE in
       fi
       echo""
     done
+    
     buildClient
   ;;
   2)
@@ -165,6 +201,7 @@ case $RUNMODE in
       fi
       echo""
     done
+    
     buildServer
   ;;
 esac
\ No newline at end of file