bash socket写的wget程序

W.Z.T

unix的一条哲学是每个命令只完成一个功能， bash却可以把他们像拼积木一样组合起来， 比如这个纯bash的http downlaoder， 可以下载binary和asiic文本， 带进度条， 目前还不支持302格式： https://github.com/cloudsec/brootkit/blob/master/brget.sh

1. #!/bin/bash
   
2. 
   
3. declare remote_host
   
4. declare remote_port
   
5. declare remote_file
   
6. declare local_file
   
7. declare remote_file_len=0
   
8. declare curr_file_len=0
   
9. declare max_col_num=64
   
10. declare total_run_time
    
11. 
    
12. function br_run_init()
    
13. {
    
14.         local i
    
15.        
    
16.         echo -ne "["
    
17.         for ((i = 1; i < $max_col_num; i++))
    
18.         do
    
19.                 echo -ne " "
    
20.         done
    
21.         echo -ne "]\r"
    
22. }
    
23. 
    
24. function br_run_play()
    
25. {
    
26.         local i x y tmp_col
    
27. 
    
28.         tmp_col=$((curr_file_len * max_col_num / remote_file_len))
    
29. 
    
30.         echo -ne "["
    
31.         for ((i = 1; i < $tmp_col; i++))
    
32.         do
    
33.                 echo -ne ">"
    
34.         done
    
35.         echo -ne "\r"
    
36. }
    
37. 
    
38. function br_run_finsh()
    
39. {
    
40.         echo -ne "\033[?25h"
    
41. }
    
42. 
    
43. function compute_run_time()
    
44. {
    
45.         local day hour min rtime
    
46. 
    
47.         day=$(($1/3600/24))
    
48.         hour=$(($1/3600))
    
49.         min=$(($1/60))
    
50. 
    
51.         if [ $min -eq 0 ]; then
    
52.                 sec=$(($1%60))
    
53.                 total_run_time="$sec s"
    
54.         else
    
55.                 if [ $hour -eq 0 ]; then
    
56.                         sec=$(($1%60))
    
57.                         total_run_time="$min m $sec s"
    
58.                 else
    
59.                         if [ $day -eq 0 ]; then
    
60.                                 tmp=$(($1%3600))
    
61.                                 min=$(($tmp/60))
    
62.                                 sec=$(($tmp%60))
    
63.                                 total_run_time="$hour h $min m $sec s"
    
64.                         else
    
65.                                 # 86400 = 3600 * 24
    
66.                                 tmp=$(($1%86400))
    
67.                                 hour=$(($tmp/3600))
    
68.                                 tmp1=$(($tmp%3600))
    
69.                                 min=$(($tmp1/60))
    
70.                                 sec=$(($tmp1%60))
    
71.                                 total_run_time="$day d $hour h $min m $sec s"
    
72.                         fi
    
73. 
    
74. 
    
75.                 fi
    
76.         fi
    
77. }
    
78. 
    
79. function get_run_time()
    
80. {
    
81.         local run_count local_hz run_time
    
82.         local start_time curr_time
    
83. 
    
84.         if [ -d "/proc/$1" ]; then
    
85.                 run_count=`cat /proc/$1/stat | cut -d " " -f 22`
    
86.         else
    
87.                 return 0
    
88.         fi
    
89. 
    
90.         local_hz=`getconf CLK_TCK`
    
91.         start_time=$(($run_count/$local_hz))
    
92. 
    
93.         curr_time=`cat /proc/uptime | cut -d " " -f 1 | cut -d "." -f 1`
    
94.         run_time=$((curr_time-start_time))
    
95. 
    
96.         return $run_time
    
97. }
    
98. 
    
99. function sock_read()
    
100. {
     
101.         local line tmp len=0 idx=0
     
102. 
     
103.         read -u 9 -t 5 line
     
104.         if ! echo $line|grep -e "200 OK" >/dev/null; then
     
105.                 echo $line
     
106.                 rm -f $remote_file
     
107.                 socket_close
     
108.                 exit
     
109.         else
     
110.                 echo "response 200 ok."
     
111.         fi
     
112. 
     
113.         while read -u 9 -t 5 line
     
114.         do
     
115.                 if [ ${#line} -eq 1 ]; then
     
116.                         break
     
117.                 fi
     
118. 
     
119.                 tmp=`echo $line|cut -d " " -f 1`
     
120.                 if [ "$tmp" == "Content-Length:" ]; then
     
121.                         remote_file_len=`echo $line|cut -d " " -f 2`
     
122.                 fi
     
123.         done
     
124. 
     
125.         echo -e "length: $remote_file_len\n"
     
126. 
     
127.         br_run_init
     
128. 
     
129.         tmp=${#remote_file_len}
     
130.         ((tmp--))
     
131.         remote_file_len=${remote_file_len:0:$tmp}
     
132. 
     
133.         while [ $curr_file_len -le $remote_file_len ]
     
134.         do
     
135.                 `dd bs=1024 count=1 of=$local_file seek=$idx <&9 2>/dev/null`
     
136.                 ((idx++))
     
137.                 curr_file_len=$((idx*1024))
     
138.                 br_run_play
     
139.         done
     
140. 
     
141.         #get_run_time $
     
142.         #compute_run_time $?
     
143.         #echo -ne "\n$total_run_time"
     
144. }
     
145. 
     
146. function sock_write()
     
147. {
     
148.         local buf
     
149. 
     
150.         buf="GET /$3 http/1.0\r\nHost: $1:$2\r\n"
     
151.         echo -e $buf >&9
     
152.         [ $? -eq 0 ] && echo "send http request ok." || echo "send http request failed."
     
153. }
     
154. 
     
155. function socket_create()
     
156. {
     
157.         exec 9<> /dev/tcp/$1/$2
     
158.         [ $? -eq 0 ] && echo "connect to $1:$2 ok." || echo "connect to $1:$2 failed."
     
159. }
     
160. 
     
161. function socket_close()
     
162. {
     
163.         exec >&9-
     
164.         [ $? -ne 0 ] && echo "close socket failed."
     
165. }
     
166. 
     
167. function parse_url()
     
168. {
     
169.         local url=$1
     
170. 
     
171.         url=${url#http://}
     
172.         remote_file=${url#*/}
     
173. 
     
174.         [ -n "$2" ] && local_file=$2 || local_file=${url##*/}
     
175. 
     
176.         remote_host=`echo $url | awk -F '/' '{print $1}'`
     
177.         remote_port=`echo $remote_host | awk -F ':' '{print $2}'`
     
178.         remote_host=`echo $remote_host | awk -F ':' '{print $1}'`
     
179.        
     
180.         [ "$remote_port" == "" ] && remote_port=80
     
181. }
     
182. 
     
183. function file_init()
     
184. {
     
185.         [ -f $local_file ] && rm -f $local_file || touch $local_file
     
186. }
     
187. 
     
188. function display_start()
     
189. {
     
190.         local tmp
     
191. 
     
192.         tmp=`date +'%F %T'`
     
193.         tmp="--$tmp-- $1"
     
194.         echo -e $tmp
     
195. }
     
196. 
     
197. function display_finsh()
     
198. {
     
199.         local tmp
     
200. 
     
201.         tmp=`date +'%F %T'`
     
202.         tmp="\n\n--$tmp-- - $local_file saved $remote_file_len"
     
203.         echo -e "$tmp"
     
204. }
     
205. 
     
206. function brget_usage()
     
207. {
     
208.         echo -e "$0 <http_url> [local_file]\n"
     
209.         echo "exp:"
     
210.         echo "$0 http://www.baidu.com/index.html"
     
211.         echo "$0 http://www.baidu.com:80/index.html"
     
212. }
     
213. 
     
214. function main()
     
215. {
     
216.         if [ $# -eq 0 ]; then
     
217.                 brget_usage $1
     
218.                 exit
     
219.         fi
     
220. 
     
221.         parse_url $@
     
222. 
     
223.         file_init
     
224.         display_start $1
     
225.         socket_create $remote_host $remote_port
     
226.         sock_write $remote_host $remote_port $remote_file
     
227.         sock_read
     
228.         display_finsh
     
229.         socket_close
     
230. }
     
231. 
     
232. main $@

复制代码

xinglp

没有/dev/tcp肿麽办

ziyunfei

495616的文件下下来成了496640, 你这可执行程序能执行起来?

qq5910225

用200多行写了一个wget类似效果的shell。怎么说呢，都说一个管道符顶的上万行C代码，用shell就是图省事呀。楼主有这个雅兴不如学个C吧，这种纯花拳绣腿的脚本没有什么实际意义，仅仅满足了你的一点点虚荣心而已。就好像有人用bash写个游戏，除了秀一些技巧，写的代码，几乎对你工作一点帮助都没有。

W.Z.T

回复 4# qq5910225

我觉得你好像还没有资格教育我学不学c，我写os，jvm的时候， 你还不知道在干嘛， 你的评论说明你还不懂unix，更不懂c。

bikong0411

bash写的不错

bikkuri

楼主有好东西拿出来给大家分享，精神可嘉，大家应该有感恩之心，互相学习互相帮助。

jixuuse

你试试curl这么命令好么？

qq5910225

回复 5# W.Z.T

嗯嗯 你说你会开航天飞机我都信。
   

W.Z.T

回复 9# qq5910225

王军别闹了行么，os和jvm源码：https://github.com/cloudsec