• Home
  • Categories
  • WIKI
  • LANGUAGE: en
    • bash脚本 --- ssh登录弹窗警告
    社区开发 695 views · 0 replies ·
    Tofloor
    poster avatar
    151******18
    deepin
    2018-11-27 07:13
    Author
    > 大佬不要嘲笑我，我还是个新手

    * 在纠结了半个多月，测试了好久，最后决定还是使用死循环实现吧，内存占用四百多k，还是可以接受的。
    * 功能：当有人使用ssh登录你的电脑，因情况而定最迟不超过10秒，会有右上角提示，登出也会有。
    * 优化与新增功能：每次循环睡眠的时间根据内存占用情况而定，安装后开机自启动（参考了随机更换壁纸脚本lswc的安装与卸载脚本）。
    * 功能脚本代码：
    1. #!/bin/bash

    3. #===================================================Description and processing of Global variables. START

    5. # 获取脚本所在位置
    6. path="$(dirname $0)";

    8. # 跳转到脚本所在位置
    9. cd ${path};

    11. # 信息存储文件 : logFile

    13. # 当前登陆者信息 : nowInfo

    15. # 登录者信息 : nowLogin

    17. # 登出者信息 : nowLogout

    19. #===================================================Description and processing of Global variables. END

    21. #===================================================personal function START

    23. # Determine whether the information storage file exists, if it does not exist, create the file START
    24. function touchInfofile() {
    25.     logFile="${path}/log_monitor.info";
    26.     if [[ -e ${logFile} ]];then
    27.         return 0;
    28.     fi

    30.     # 创建信息存储文件
    31.     touch ${logFile};

    33.     # 获取登陆者信息
    34.     echo "$(who | tr -s ' ')" > ${logFile};
    35.     return 1;
    36. }
    37. # Determine whether the information storage file exists, if it does not exist, create the file END

    39. # send information function START
    40. function send_info() {
    41.     # ${1} : nowLogin / nowLogout
    42.     # ${2} : 1 : log in , 2 : log out.

    44.     local LogTTY=$(echo "${1}" | cut -d ' ' -f 2);
    45.     local LogTime=$(echo "${1}" | cut -d ' ' -f 4);
    46.     local LogerIP=$(echo "${1}" | cut -d ' ' -f 5);
    47.     local str1;
    48.     if [[ ${2} -eq 0 ]];then
    49.         if [[ "${LogerIP}" == "(:0)" || "${LogerIP}" == "(127.0.0.1)" ]];then
    50.             str1="本机登出${LogTTY}"
    51.         else
    52.             str1="来自${LogerIP}的登陆者从${LogTTY}登出";
    53.         fi
    54.         LogTime="$(date +"%H:%M")";
    55.     else
    56.         if [[ "${LogerIP}" == "(:0)" || "${LogerIP}" == "(127.0.0.1)" ]];then
    57.             str1="本机登入${LogTTY}"
    58.         else
    59.             str1="来自${LogerIP}的登录者登入${LogTTY}";
    60.         fi
    61.     fi

    63.     notify-send -i gtk-dialog-info "${str1}" "时间 : ${LogTime}";
    64.     return $?;
    65. }
    66. # send information function END

    68. # Let the program sleep for a while START
    69. function sleepSometime() {
    70.     local sltime=$(cat /proc/meminfo | grep -E '^Mem([T]|[A])' | grep -Eo '[0-9]*' | tr -s '\n' ' ' | awk '{
    71.         mem = $2 / $1 * 100;
    72.         if (mem > 75) {
    73.             printf("1");
    74.         } else if (mem > 55) {
    75.             printf("3");
    76.         } else {
    77.             printf("5");
    78.         }
    79.     }');
    80.     sleep ${sltime};
    81.     return 0;
    82. }
    83. # Let the program sleep for a while END

    85. # Determine if the login information has changed START
    86. function ifContinue() {
    87.     local now_info="$(who | tr -s ' ')";
    88.     if [[ "${now_info}" == "$(cat ${logFile})" ]];then
    89.         sleepSometime;
    90.     else
    91.         nowInfo=${now_info};
    92.         return 0;
    93.     fi
    94. }
    95. # Determine if the login information has changed END

    97. # Determine if someone is logged in START
    98. function haveLogin() {
    99.     local now_info=${1};
    100.     local info_file="$(cat ${logFile})";
    101.     local login_info="$(echo "${now_info}" | grep -v "${info_file}")";
    102.     if [[ "x${login_info}" != "x" ]];then
    103.         echo "${login_info}" >> ${logFile};
    104.         nowLogin=${login_info};
    105.         return 1;
    106.     else
    107.         # no one.
    108.         return 0;
    109.     fi
    110. }
    111. # Determine if someone is logged in END

    113. # Determine if someone is logged out START
    114. function haveLogout() {
    115.     local now_info=${1};
    116.     local logout_info="$(grep -vn "${now_info}" ${logFile})";
    117.     local line=$(echo "${logout_info}" | cut -d ':' -f 1);
    118.     local logout_info=$(echo "${logout_info}" | sed 's/^[0-9]*://g');
    119.     if [[ "x${logout_info}" != "x" ]];then
    120.         echo "$(cat ${logFile} | sed "${line}d")" > ${logFile};
    121.         nowLogout=${logout_info};
    122.         return 1;
    123.     else
    124.         # no one.
    125.         return 0;
    126.     fi
    127. }
    128. # Determine if someone is logged out END

    130. #===================================================personal function END

    132. function main() {
    133.     while true;
    134.     do
    135.         # Determine if the information storage file exists.
    136.         # Create this file if it does not exist.
    137.         # prevent the file from being deleted.

    139.         touchInfofile;
    140.    
    141.         # Determine the sleep time based on the memory usage.
    142.         
    143.         sleepSometime;
    144.    
    145.         # Determine whether the login information has changed.
    146.         # If there is no change, it will not return directly.
    147.    
    148.         ifContinue;
    149.    
    150.         haveLogin "${nowInfo}";
    151.         if [[ ${?} -eq 1 ]];then
    152.             send_info "${nowLogin}" 1;
    153.         fi
    154.    
    155.         haveLogout "${nowInfo}";
    156.         if [[ ${?} -eq 1 ]];then
    157.             send_info "${nowLogout}" 0;
    158.         fi
    159.    
    160.     done
    161.     return 0;
    162. }

    164. main;
    165. exit 0;

    Copy the Code


    * 安装脚本代码：
    1. #!/bin/bash

    3. #        File Name:    install.sh
    4. #        Author:       sunowsir
    5. #        Mail:         sunowsir@protonmail.com
    6. #        Created Time: 2018年10月22日 星期一 21时40分13秒

    8. mkdir -p ~/.local/scripts/log_monitor

    10. cp ./log_monitor.sh ~/.local/scripts/log_monitor/

    12. chmod u+x ~/.local/scripts/log_monitor/log_monitor.sh

    14. cat <
      >  ${HOME}/.config/autostart/log_monitor.desktop
    15. [Desktop Entry]
    16. Name=log_monitor
    17. Exec=${HOME}/.local/scripts/log_monitor/log_monitor.sh
    18. Icon=bash
    19. Terminal=false
    20. Type=Application
    21. Categories=GNOME;GTK;System;Utility
    22. Comment[en_US.UTF-8]=Ssh stalker monitoring
    23. DT

    25. bash ~/.local/scripts/log_monitor/log_monitor.sh &

    27. echo -e "\033[1;32mSuccessful installation! Self-starting every time.\033[0m"

    Copy the Code
    * 卸载脚本代码：
    1. #!/bin/bash

    3. #        File Name:    uninstall.sh
    4. #        Author:       sunowsir
    5. #        Mail:         sunowsir@protonmail.com
    6. #        Created Time: 2018年10月22日 星期一 21时44分23秒


    9. ps -aux | grep -E '\S*bash\S*\s\S*\/log_monitor.sh' | tr -s ' ' | cut -d ' ' -f 2 | xargs -I {} kill -9 {} 2> /dev/null
    10. ps -aux | grep -E '\S*bash\S*\s\S*\/log_monitor.sh' | tr -s ' ' | cut -d ' ' -f 2 | xargs -I {} kill -9 {} 2> /dev/null
    11. rm -rf ~/.local/scripts/log_monitor/
    12. rm -rf ~/.config/autostart/log_monitor.desktop
    13. echo -e "\033[1;32mBye. \033[0m"
    Copy the Code
    https://github.com/sunowsir/SSH_log_reminder

    Reply Favorite View the author
    All Replies

    No replies yet