利用ruby简洁的数组对比来完成的检查服务器开启关闭端口的程序

laohuanggua

shell配合ruby完成的。纯ruby的我还不会。。。


shell部分

1. #!/bin/bash
   
2. #date:2010-11-13
   
3. #purpose:check new service
   
4. 
   
5. work_dir="/root/tools/develop"
   
6. ports_orig_file="${work_dir}/ports_orig"
   
7. ports_current_file="${work_dir}/ports_current"
   
8. 
   
9. if [[ -d ${work_dir} ]];then
   
10.     mkdir -p ${work_dir}
    
11. fi
    
12. #首先判断文件ports_orig是否存在
    
13. #如果没有就用netstat -ntlp来生成一个。
    
14. 
    
15. touch ${ports_orig_file}
    
16. touch ${ports_current_file}
    
17. 
    
18. if [[ -e ${ports_orig_file} ]];then
    
19. :
    
20. else
    
21.     touch ${ports_orig_file}
    
22.     `netstat -ntlp > "${ports_orig_file}"`
    
23. fi
    
24. 
    
25. #生成一个ports_current文件
    
26. netstat -ntlp > "${ports_current_file}"
    
27. ./duibi.rb
    
28. mv -f ${ports_current_file} ${ports_orig_file}
    
29. #END
    

复制代码

duibi.rb

1. #!/usr/bin/ruby
   
2. 
   
3. old_array=File.open("ports_orig").readlines
   
4. current_array=File.open("ports_current").readlines
   
5. 
   
6. puts "closed ports:"
   
7. puts old_array-current_array
   
8. 
   
9. puts "new open ports:"
   
10. puts current_array-old_array
    
11. #END

复制代码

执行的时候运行./1.sh即可。