小菜写的shell自动生成shell的注释

wsleichao3

小菜练习写的一个脚本，用于生成shell脚本中的注释

1. #!/bin/bash
   
2. 
   
3. #功能：自动生成脚本的格式
   
4. #时间：2013-03-28
   
5. #版本：1.0
   
6. #历史：
   
7. #       2013-03-28      第一版
   
8. 
   
9. #生成的脚本格式内容：
   
10. #########################################################
    
11. #       功能：……                                                                        #
    
12. #       作者：……                                                                        #
    
13. #       时间：……                                                                        #
    
14. #       版本：……                                                                        #
    
15. #       历史：……                                                                        #
    
16. #########################################################
    
17. 
    
18. #声明使用的变量
    
19. PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/user/local/sbin
    
20. script_name=''
    
21. script_function=''
    
22. script_author=''
    
23. script_version=''
    
24. script_histoy=''
    
25. script_time=$(date +%F)
    
26. 
    
27. #引导用户设置相关变量
    
28. read -p "输入生成脚本的名称(默认用时间命名，如${script_time}.sh):" script_name
    
29. read -p "输入描述脚本的功能:" script_function
    
30. read -p "输入该脚本的作者:" script_author
    
31. read -p "输入该脚本的版本:" script_version
    
32. read -p "输入改脚本的历史信息:" script_histoy
    
33. 
    
34. #变量内容的替换,如果$script_name为空，则用时间填充（在这使用的变量的替换中变量为空和没有设置是不同的）。
    
35. script_name=${script_name:-$script_time}\.sh
    
36. 
    
37. script_histoy_fill="$script_time        第一版"
    
38. 
    
39. if [ -f "$script_name" ];then
    
40.         read -p "脚本名称：${script_name}以存在，是否覆盖？[y/n]" script_exist
    
41.         if [ "$script_exist" == "n" ];then
    
42.                 read -p "请重新输入脚本的名称：" script_name
    
43.         fi
    
44. fi
    
45. 
    
46. script_histoy=${scipt_histoy:-$script_histoy_fill}
    
47. 
    
48. cat >$script_name << END
    
49. #!/bin/bash
    
50. 
    
51. #功能：$script_function
    
52. #时间：$script_time
    
53. #作者：$script_author
    
54. #版本：$script_version
    
55. #历史信息：
    
56.        $script_histoy
    
57. 
    
58. PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/user/local/sbin
    
59. END
    

复制代码

经cao627指点，对文件名称加个保险，请看注释

1. #!/bin/bash
   
2. 
   
3. #功能：自动生成脚本的格式
   
4. #时间：2013-03-28
   
5. #版本：1.0
   
6. #历史：
   
7. #        2013-03-28        第一版
   
8. 
   
9. #生成的脚本格式内容：
   
10. #########################################################
    
11. #        功能：……                                        #
    
12. #        作者：……                                        #
    
13. #        时间：……                                        #
    
14. #        版本：……                                        #
    
15. #        历史：……                                        #
    
16. #########################################################
    
17. 
    
18. #声明使用的变量
    
19. PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/user/local/sbin
    
20. script_name=''
    
21. script_function=''
    
22. script_author=''
    
23. script_version=''
    
24. script_histoy=''
    
25. script_time=$(date +%F)
    
26. 
    
27. #写成函数调用自己，解决文件命名重复问题
    
28. function script_exist(){
    
29. 
    
30.         #这里原来是script_name=${script_name:-$script_time}\.sh,如果输入的名称多次存在，就会在名称后累加.sh，例如*.sh,第二次就成了*.sh.sh
    
31.         script_name=${script_name:-$script_time\.sh}
    
32. 
    
33.         if [ -f "$script_name" ];then
    
34.                 read -p "脚本名称：${script_name}以存在，是否覆盖？[y/n]" script_exist
    
35.                 case $script_exist in
    
36.                         "y")
    
37.                                 #这里实在不知道怎么写好，目的就是检测到y时，程序继续往下走
    
38.                                 script_name=$script_name
    
39.                         ;;
    
40.                         "n")
    
41.                                 read -p "请重新输入脚本的名称：" script_name
    
42.                                 #调用自己
    
43.                                 script_exist
    
44.                         ;;
    
45.                         #默认动作和其他输入时进行调用自己
    
46.                         *)
    
47.                                 script_exist
    
48.                         ;;
    
49.                 esac
    
50.         fi
    
51. }
    
52. 
    
53. #引导用户设置相关变量
    
54. read -p "输入生成脚本的名称(默认用时间命名，如${script_time}.sh):" script_name
    
55. read -p "输入描述脚本的功能:" script_function
    
56. read -p "输入该脚本的作者:" script_author
    
57. read -p "输入该脚本的版本:" script_version
    
58. read -p "输入改脚本的历史信息:" script_histoy
    
59. 
    
60. #进行文件名的检测，自定义函数
    
61. script_exist
    
62. 
    
63. 
    
64.        
    
65. script_histoy_fill="$script_time        第一版"
    
66. script_histoy=${script_histoy:-$script_histoy_fill}
    
67. 
    
68. cat >$script_name << END
    
69. #!/bin/bash
    
70. 
    
71. #功能：$script_function
    
72. #时间：$script_time
    
73. #作者：$script_author
    
74. #版本：$script_version
    
75. #历史信息：
    
76.        $script_histoy
    
77. 
    
78. PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/user/local/sbin
    
79. END

复制代码

wsleichao3

还有一个脚本是根据抚琴煮酒大侠写的脚本稍微改动了下
这个脚本使用的sudo，没有直接使用root,是为了使用权限较大的命令时能够在日志中记录，sudo配置的时无密码的sudo。

1. #!/bin/bash
   
2. #脚本模仿“抚琴煮酒”的http://andrewyu.blog.51cto.com/1604432/1074650
   
3. #脚本描述：
   
4. #       对SSH和FTP的恶意连接的IP尝试次数大于10次以上进行屏蔽。
   
5. #时间:2013-03-23
   
6. 
   
7. 
   
8. #awk展开是
   
9. #for(i=1;i<NF;i++){
   
10. #       if($i~/rhost/&&$i!~/\//){
    
11. #               ip=substr($i,7);
    
12. #               if(ip~/^:/){
    
13. #                       print substr(ip,8)
    
14. #               }else{
    
15. #                       print print ip
    
16. #               }
    
17. #       }
    
18. #}
    
19. sudo awk '{for(i=1;i<NF;i++){if($i~/rhost/&&$i!~/\//){ip=substr($i,7);if(ip~/^:/){print substr(ip,8)}else print ip}}}' /var/log/secure|sort|uniq -c >drop.ip
    
20. 
    
21. 
    
22. cat drop.ip|while read A
    
23. do
    
24.         num=$(echo $A|awk '{print $1}');
    
25.         ip=$(echo $A|awk '{print $2}');
    
26. 
    
27. 
    
28.         if [ "$num" -ge "10" ];then
    
29.                 sudo iptables -L -n|grep "$ip" >/dev/null
    
30.                 if [ "$?" -ne "0" ];then
    
31.                         echo "sudo iptables -I INPUT -s $ip -j DROP"
    
32.                 fi
    
33.         fi
    
34. done

复制代码

Shell_HAT

是否考虑在第40行前面加个判断，如果存在$script_name同名文件，则提示用户是否覆盖。

wsleichao3

回复 3# Shell_HAT

谢谢大侠指导，我修改下～
   

cao627

回复 1# wsleichao3


    问一下
第一楼的39到44行

1. if [ -f "$script_name" ];then
   
2.         read -p "脚本名称：${script_name}以存在，是否覆盖？[y/n]" script_exist
   
3.         if [ "$script_exist" == "n" ];then
   
4.                 read -p "请重新输入脚本的名称：" script_name
   
5.         fi
   
6. fi

复制代码

如果

read -p "请重新输入脚本的名称：" script_name

读到的”script_name“还是存在的怎么办？

是否有必要把外的层if 语句换成 wile语句?

davidbeckham921

学习！~~~:wink:

死皮赖狗

学习了，谢谢分享

wsleichao3

回复 5# cao627
大侠说的有道理，是我想的不够全面。我修改了下，麻烦过目下。还是在一楼


   

cao627

回复 9# wsleichao3

我觉得没必要用函数自我调用的方式来实习这个功能，直接写个while循环不是简单点吗？

1. while [  -f "$script_name" ]
   
2. do
   
3. read -p "脚本名称：${script_name}以存在，是否覆盖？[y/n]" script_exist
   
4. if [ "$script_exist" == "n" ];then
   
5.   read -p "请重新输入脚本的名称：" script_name
   
6. elif [ "$script_exist" == "y" ];then
   
7.   break
   
8. fi
   
9. done
   

复制代码