From 5d57d7965beede91e0519f618c0d3baf19ec4cfa Mon Sep 17 00:00:00 2001
From: chenxinquan <chenxinquan@kylinos.cn>
Date: Thu, 16 Nov 2023 15:44:44 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E5=9F=BA=E7=BA=BF=E6=89=AB?=
 =?UTF-8?q?=20=E6=8F=8F=E9=A1=B9'99=E6=A3=80=E6=9F=A5ssh=E5=AE=89=E5=85=A8?=
 =?UTF-8?q?=E5=8D=8F=E8=AE=AE'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../99/99检查ssh安全协议.py                   | 71 +++++++++++++++++++
 .../99/99检查ssh安全协议.yaml                 | 22 ++++++
 2 files changed, 93 insertions(+)
 create mode 100755 data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.py
 create mode 100755 data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.yaml

diff --git a/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.py b/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.py
new file mode 100755
index 0000000..6e15326
--- /dev/null
+++ b/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.py
@@ -0,0 +1,71 @@
+
+import os
+import sys
+
+################################
+# 常量
+
+# for get_env_lang()
+STR_GET_ENV_LANG_ZH     = "语言环境为中文"
+STR_GET_ENV_LANG_EN     = "语言环境为英文"
+STR_GET_ENV_LANG_UNKNOW = "语言环境未知"
+
+################################
+# 环境检查函数
+
+def get_env_lang():
+#    lang = os.getenv("LANG")
+#    if lang.startswith("zh"):
+#        return STR_GET_ENV_LANG_ZH
+#    elif lang.startswith("en"):
+#        return STR_GET_ENV_LANG_EN
+#    else:
+#        return STR_GET_ENV_LANG_UNKNOW
+#
+    if arg_lang == "zh":
+        return STR_GET_ENV_LANG_ZH
+    elif arg_lang == "en":
+        return STR_GET_ENV_LANG_EN
+    else:
+        return STR_GET_ENV_LANG_UNKNOW
+
+def is_root():
+    if os.geteuid() == 0:
+        print(STR_IS_ROOT_TRUE)
+        return True
+    else:
+        print(STR_IS_ROOT_FALSE)
+        return False
+
+################################
+# 辅助函数
+def l_print(zh_str, en_str) :
+    if STR_GET_ENV_LANG_ZH == get_env_lang() :
+        print(zh_str);
+    else :
+        print(en_str);
+
+
+################################
+# 功能函数
+def portocol():
+
+    output = os.popen('cat /etc/ssh/sshd_config | grep Protocol | grep -v ^#  ').read().strip() 
+
+    if "Protocol 2" in output :
+        l_print("[OK] 测试通过",
+                "[OK] pass")
+    else:
+        l_print("[ERROR] 测试未通过",
+                "[ERROR] fail")
+
+################################
+# main
+if __name__ == "__main__":
+    if len(sys.argv)>1:
+        arg_lang = sys.argv[1]
+    else:
+        arg_lang = 'zh'
+
+    portocol()
+    exit(0)
diff --git a/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.yaml b/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.yaml
new file mode 100755
index 0000000..f708146
--- /dev/null
+++ b/data/BaseLine/LocalServices/TestingDepartmentPrivateUseCases/99/99检查ssh安全协议.yaml
@@ -0,0 +1,22 @@
+FormatVer: 20230623
+Id: portocol
+Belong: baseline
+SiteInfo:
+    Name: 99检查ssh安全协议
+Power : "root"
+SiteRequests:
+    Implement:
+        ImArray:
+            - Inter : python3
+              InterArgs :
+              Exec : 99检查ssh安全协议.py
+              Args :
+        Inter:
+            - "[ERROR]"
+        Condition: None
+RepairArgs:
+    - Inter : python3
+      InterArgs :
+      Exec :
+      Args :
+      RepairPower: #root # root权限或者普通用户权限