Update centos7_wireguard_install.sh

centos7_wireguard_install.sh

@@ -1,7 +1,6 @@
 #!/bin/bash
 
-# wireguard For CentOS
-# https://github.com/yobabyshark/wireguard/edit/master/wireguard_install.sh
+https://github.com/atrandys/wireguard/edit/master/wireguard_install.sh
 
 #判断系统
 if [ ! -e '/etc/redhat-release' ]; then
@@ -46,6 +45,17 @@ rand(){
     echo $(($num%$max+$min))  
 }
 
+wireguard_update(){
+    yum update -y wireguard-dkms wireguard-tools
+    echo "更新完成"
+}
+
+wireguard_remove(){
+    yum remove -y wireguard-dkms wireguard-tools
+    rm -rf /etc/wireguard/
+    echo "卸载完成"
+}
+
 config_client(){
 cat > /etc/wireguard/client.conf <<-EOF
 [Interface]
@@ -77,7 +87,7 @@ wireguard_install(){
     s2=$(cat spublickey)
     c1=$(cat cprivatekey)
     c2=$(cat cpublickey)
-    serverip=$(curl icanhazip.com)
+    serverip=$(curl ipv4.icanhazip.com)
     port=$(rand 10000 60000)
     chmod 777 -R /etc/wireguard
     systemctl stop firewalld
@@ -127,7 +137,9 @@ start_menu(){
     echo "========================="
     echo "1. 升级系统内核"
     echo "2. 安装wireguard"
-    echo "3. 退出脚本"
+    echo "3. 升级wireguard"
+    echo "4. 卸载wireguard"
+    echo "0. 退出脚本"
     echo
     read -p "请输入数字:" num
     case "$num" in
@@ -138,6 +150,12 @@ start_menu(){
 	wireguard_install
 	;;
 	3)
+	wireguard_update
+	;;
+	4)
+	wireguard_remove
+	;;
+	0)
 	exit 1
 	;;
 	*)
@@ -150,6 +168,3 @@ start_menu(){
 }
 
 start_menu
-
-
-