|
16
|
1 /*
|
|
24
|
2 UT : Unix like utility Tool.
|
|
16
|
3
|
|
|
4 TODO:
|
|
|
5 password
|
|
|
6 delete file on server
|
|
|
7 */
|
|
|
8 package main
|
|
|
9
|
|
|
10 import (
|
|
|
11 "bufio"
|
|
|
12 "bytes"
|
|
|
13 "flag"
|
|
|
14 "fmt"
|
|
|
15 "io"
|
|
|
16 "io/ioutil"
|
|
|
17 "log"
|
|
|
18 "mime/multipart"
|
|
|
19 "net/http"
|
|
|
20 "os"
|
|
|
21 "runtime"
|
|
|
22 "strconv"
|
|
|
23 "strings"
|
|
|
24 "time"
|
|
|
25
|
|
|
26 "./lu"
|
|
20
|
27 "./nw"
|
|
16
|
28 )
|
|
|
29
|
|
21
|
30 var version = "1.06"
|
|
16
|
31
|
|
|
32 func main() {
|
|
|
33
|
|
|
34 var getfile = flag.Bool( "g", false, "get a file from site." ) // ok
|
|
|
35 var putfile = flag.Bool( "G", false, "put a file to site." ) // ok
|
|
17
|
36 var compress = flag.Bool( "z", false, "compress files. (.tgz)" ) // ok
|
|
16
|
37 var uncmprss = flag.Bool( "x", false, "uncompress files. (.tgz )" ) // ok
|
|
|
38 var noprefix = flag.Bool( "n", false, "not add prefix(YYYYMMDDhhmm_) to filename." ) // ok
|
|
|
39
|
|
|
40 var wget = flag.Bool( "w", false, "little wget tool." ) // ok
|
|
|
41 var wc_l = flag.Bool( "l", false, "little wc -l tool." ) // ok
|
|
|
42 var catn = flag.Bool( "C", false, "little cat -n tool." ) // ok
|
|
20
|
43 var reverse = flag.Bool( "R", false, "little tac tool. ( reverse file )" ) // ok
|
|
16
|
44 var uniq = flag.Bool( "u", false, "little uniq tool." ) // ok
|
|
|
45 var uniqc = flag.Bool( "uc", false, "little uniq tool.( count )" ) // ok
|
|
|
46 var uniqd = flag.Bool( "ud", false, "little uniq tool.( only dup )" ) // ok
|
|
|
47 var uniqdc = flag.Bool( "udc", false, "little uniq tool.( count dup )" ) // ok
|
|
|
48 var md5 = flag.Bool( "md5", false, "md5." ) // ok
|
|
|
49 var sha256 = flag.Bool( "sha256", false, "sha256." ) // ok
|
|
18
|
50 var grep = flag.String( "a", "", "little grep tool." ) // ok
|
|
|
51 var grepv = flag.String( "b", "", "little grep -v tool." ) // ok
|
|
|
52 var orgrep = flag.String( "A", "", "little grep tool. ( from list file )" ) // ok
|
|
|
53 var orgrepv = flag.String( "B", "", "little grep -v tool. ( from list file )" ) // ok
|
|
19
|
54 var tee = flag.String( "e", "", "little tee tool." ) // ok
|
|
21
|
55 var replace = flag.Bool( "r", false, "replace strings. ( s/arg1/arg2/g )" ) // ok
|
|
20
|
56 var nc = flag.Bool( "nc", false, "netcat." ) // ok
|
|
|
57 var ncl = flag.Bool( "ncl", false, "netcat listen-mode." ) // ok
|
|
23
|
58 var fserver = flag.Int( "fs", 0, "file server" ) // ok
|
|
|
59 var wserver = flag.Int( "ws", 0, "web server" ) // ok
|
|
16
|
60
|
|
|
61 var msleep = flag.Bool( "M", false, "sleep." )
|
|
|
62 //var swatch = flag.Bool( "W", false, "stop watch." )
|
|
|
63 //var bccalc = flag.Bool( "j", false, "calculator." )
|
|
|
64 //var bhole = flag.Bool( "k", false, "black hole. ( /dev/null )" )
|
|
|
65 var cmdtime = flag.Bool( "T", false, "measure program running time." )
|
|
|
66 var touch = flag.Bool( "t", false, "little touch tool." )
|
|
|
67 var sort = flag.Bool( "s", false, "little sort tool." )
|
|
|
68 var nsort = flag.Bool( "ns", false, "little number sort tool." )
|
|
|
69 var tree = flag.Bool( "f", false, "little tree tool." )
|
|
|
70 var head = flag.Int( "d", 0, "little head/tail tool.( head:n, tail:-n )" )
|
|
|
71 //var cut = flag.String( "c", "", "little cut tool ( only csv )." )
|
|
|
72 var migemo = flag.Bool( "m", false, "migemo tool." )
|
|
|
73 //var split = flag.Int( "s", 100, "split file ( default 100 MB )." )
|
|
|
74 var ipscan = flag.Bool( "p", false, "ping." )
|
|
|
75 var minfo = flag.Bool( "i", false, "print machine infomation." ) // half-ok
|
|
|
76
|
|
|
77 //var imgsz = flag.String( "is", "", "resize image" )
|
|
|
78 //var imgconv = flag.String( "ic", "", "convert image type" )
|
|
|
79
|
|
|
80 var pversion = flag.Bool( "v", false, "print version." ) // ok
|
|
|
81 var help = flag.Bool( "h", false, "help." ) // ok
|
|
|
82 flag.Parse()
|
|
|
83
|
|
|
84 if *pversion {
|
|
|
85 fmt.Fprintf( os.Stderr, "%s [ ver %s ]\n", os.Args[0], version )
|
|
|
86 os.Exit( 0 )
|
|
|
87 }
|
|
|
88
|
|
|
89 if *help {
|
|
|
90 print_help()
|
|
|
91 os.Exit( 0 )
|
|
|
92 }
|
|
|
93
|
|
|
94 if *minfo {
|
|
|
95 // IP, mask, gw, dns
|
|
|
96 host, _ := os.Hostname()
|
|
|
97 fmt.Fprintf( os.Stderr, "OS = %s\n", runtime.GOOS )
|
|
|
98 fmt.Fprintf( os.Stderr, "ARCH = %s\n", runtime.GOARCH )
|
|
|
99 fmt.Fprintf( os.Stderr, "HOSTNAME = %s", host )
|
|
20
|
100 fmt.Fprintf( os.Stderr, " [%s]\n", nw.GetLocalIP() )
|
|
16
|
101 os.Exit( 0 )
|
|
|
102 }
|
|
|
103
|
|
23
|
104 /* Here Servers */
|
|
|
105 if *fserver != 0 {
|
|
|
106 addr := nw.GetLocalIP() + ":" + fmt.Sprintf( "%d", *fserver )
|
|
|
107 log.Fatal( http.ListenAndServe( addr, http.FileServer( http.Dir( ".") ) ) )
|
|
|
108 }
|
|
|
109
|
|
|
110 if *wserver != 0 {
|
|
|
111 addr := nw.GetLocalIP() + ":" + fmt.Sprintf( "%d", *wserver )
|
|
|
112 http.HandleFunc( "/", func( w http.ResponseWriter, r *http.Request ) {
|
|
|
113 if flag.NArg() == 0 {
|
|
|
114 fmt.Fprintf( w, "WebServer\n" )
|
|
|
115 return
|
|
|
116 }
|
|
|
117 file := flag.Args()[0]
|
|
|
118 f, err := os.Open( file )
|
|
|
119 if err != nil {
|
|
|
120 fmt.Fprintf( w, "WebServer\n" )
|
|
|
121 }
|
|
|
122 if _, err = io.Copy( w, f ); err != nil {
|
|
|
123 log.Fatal( err )
|
|
|
124 }
|
|
|
125 f.Close()
|
|
|
126 } )
|
|
|
127 log.Fatal( http.ListenAndServe( addr, nil ) )
|
|
|
128 }
|
|
|
129
|
|
16
|
130 /* Here Utility Function ( unix tool like tools ) */
|
|
|
131 if *msleep {
|
|
|
132 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
133 os.Exit( 3 )
|
|
|
134 }
|
|
|
135
|
|
|
136 if *cmdtime {
|
|
|
137 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
138 os.Exit( 3 )
|
|
|
139 }
|
|
|
140
|
|
|
141 if *wget { // done.
|
|
|
142 if flag.NArg() == 0 {
|
|
|
143 fmt.Fprintln( os.Stderr, "no url" )
|
|
|
144 os.Exit( 1 )
|
|
|
145 }
|
|
|
146 url := flag.Args()[0]
|
|
|
147 if !strings.Contains( url, "://" ) {
|
|
|
148 url = "http://" + url
|
|
|
149 }
|
|
|
150
|
|
|
151 fmt.Printf( "%s", url_get( url ) )
|
|
|
152 os.Exit( 0 )
|
|
|
153 }
|
|
|
154
|
|
|
155 /*
|
|
|
156 if *cut {
|
|
|
157 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
158 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
159 log.Fatal( err )
|
|
|
160 }
|
|
|
161 lu.CutCsv( flag.Args() )
|
|
|
162 os.Exit( 3 )
|
|
|
163 }
|
|
|
164 */
|
|
|
165
|
|
19
|
166 if *tee != "" { // done.
|
|
|
167 lu.Tee( *tee )
|
|
|
168 os.Exit( 0 )
|
|
17
|
169 }
|
|
|
170
|
|
16
|
171 if *sort {
|
|
|
172 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
173 os.Exit( 3 )
|
|
|
174 }
|
|
|
175
|
|
|
176 if *nsort {
|
|
|
177 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
178 os.Exit( 3 )
|
|
|
179 }
|
|
|
180
|
|
|
181 if *tree {
|
|
|
182 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
183 os.Exit( 3 )
|
|
|
184 }
|
|
|
185
|
|
|
186 if *catn { // done.
|
|
|
187 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
188 log.Fatal( err )
|
|
|
189 }
|
|
|
190 lu.Cat_n( flag.Args() )
|
|
|
191 os.Exit( 0 )
|
|
|
192 }
|
|
|
193
|
|
|
194 if *wc_l { // done.
|
|
|
195 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
196 log.Fatal( err )
|
|
|
197 }
|
|
|
198 lu.Wc_l( flag.Args() )
|
|
|
199 os.Exit( 0 )
|
|
|
200 }
|
|
|
201
|
|
|
202 if *uniq || *uniqc || *uniqd || *uniqdc { // done.
|
|
|
203 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
204 log.Fatal( err )
|
|
|
205 }
|
|
|
206 if *uniqdc {
|
|
|
207 *uniqc, *uniqd = true, true
|
|
|
208 }
|
|
|
209 lu.Uniq( *uniqc, *uniqd, flag.Args() )
|
|
|
210 os.Exit( 0 )
|
|
|
211 }
|
|
|
212
|
|
|
213 if *head != 0 {
|
|
|
214 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
215 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
216 log.Fatal( err )
|
|
|
217 }
|
|
|
218 lu.Head( *head, flag.Args() )
|
|
|
219 os.Exit( 0 )
|
|
|
220 }
|
|
|
221
|
|
|
222 if *reverse { // done.
|
|
|
223 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
224 log.Fatal( err )
|
|
|
225 }
|
|
|
226 lu.Reverse( flag.Args() )
|
|
|
227 os.Exit( 0 )
|
|
|
228 }
|
|
|
229
|
|
18
|
230 if *grep != "" { // done.
|
|
|
231 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
232 log.Fatal( err )
|
|
16
|
233 }
|
|
18
|
234 lu.Grep( *grep, false, flag.Args() )
|
|
|
235 os.Exit( 0 )
|
|
|
236 }
|
|
|
237
|
|
|
238 if *grepv != "" { // done.
|
|
|
239 if err := is_exist_files( flag.Args() ); err != nil {
|
|
16
|
240 log.Fatal( err )
|
|
|
241 }
|
|
18
|
242 lu.Grep( *grepv, true, flag.Args() )
|
|
|
243 os.Exit( 0 )
|
|
|
244 }
|
|
|
245
|
|
|
246 if *orgrep != "" {
|
|
|
247 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
248 log.Fatal( err )
|
|
|
249 }
|
|
|
250 lu.OrGrep( *orgrep, false, flag.Args() )
|
|
|
251 os.Exit( 0 )
|
|
|
252 }
|
|
|
253
|
|
|
254 if *orgrepv != "" {
|
|
|
255 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
256 log.Fatal( err )
|
|
|
257 }
|
|
|
258 lu.OrGrep( *orgrepv, true, flag.Args() )
|
|
16
|
259 os.Exit( 0 )
|
|
|
260 }
|
|
|
261
|
|
|
262 if *migemo {
|
|
|
263 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
264 os.Exit( 3 )
|
|
|
265 }
|
|
|
266
|
|
21
|
267 if *replace { // done.
|
|
|
268 if flag.NArg() < 2 {
|
|
|
269 fmt.Fprintf( os.Stderr, "too few argument.\n" )
|
|
|
270 os.Exit( 1 )
|
|
|
271 }
|
|
16
|
272 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
273 log.Fatal( err )
|
|
|
274 }
|
|
21
|
275 lu.Replace( flag.Args()[0], flag.Args()[1], flag.Args()[2:] )
|
|
16
|
276 os.Exit( 0 )
|
|
|
277 }
|
|
|
278
|
|
|
279 if *touch {
|
|
|
280 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
281 log.Fatal( err )
|
|
|
282 }
|
|
|
283 lu.Touch( flag.Args() )
|
|
|
284 os.Exit( 0 )
|
|
|
285 }
|
|
|
286
|
|
|
287 if *md5 { // done.
|
|
|
288 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
289 log.Fatal( err )
|
|
|
290 }
|
|
|
291 lu.Md5( flag.Args() )
|
|
|
292 os.Exit( 0 )
|
|
|
293 }
|
|
|
294
|
|
|
295 if *sha256 { // done.
|
|
|
296 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
297 log.Fatal( err )
|
|
|
298 }
|
|
|
299 lu.Sha256( flag.Args() )
|
|
|
300 os.Exit( 0 )
|
|
|
301 }
|
|
|
302
|
|
20
|
303 if *nc { // done.
|
|
|
304 if flag.NArg() != 2 {
|
|
|
305 fmt.Fprintf( os.Stderr, "$ ut -nc 192.168.0.55 8080\n" )
|
|
|
306 os.Exit( 1 )
|
|
|
307 }
|
|
|
308 nw.Netcat( flag.Args()[0], flag.Args()[1] )
|
|
|
309 os.Exit( 0 )
|
|
|
310 }
|
|
|
311
|
|
|
312 if *ncl { // done.
|
|
|
313 if flag.NArg() == 0 {
|
|
|
314 fmt.Fprintf( os.Stderr, "$ ut -ncl 8080\n" )
|
|
|
315 os.Exit( 1 )
|
|
|
316 }
|
|
|
317 nw.Netcat_l( flag.Args()[0] )
|
|
|
318 os.Exit( 0 )
|
|
|
319 }
|
|
|
320
|
|
16
|
321 if *ipscan {
|
|
|
322 fmt.Fprintln( os.Stderr, "not implement" )
|
|
|
323 os.Exit( 2 )
|
|
|
324 }
|
|
|
325
|
|
17
|
326 if *compress && !*putfile { // done.
|
|
|
327 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
328 log.Fatal( err )
|
|
|
329 }
|
|
|
330 tgz := time.Now().Format( "200601021504.tgz" )
|
|
|
331 lu.Tgz( tgz, flag.Args() )
|
|
|
332 os.Exit( 0 )
|
|
|
333 }
|
|
|
334
|
|
16
|
335 if *uncmprss { // done.
|
|
|
336 if err := is_exist_files( flag.Args() ); err != nil {
|
|
|
337 log.Fatal( err )
|
|
|
338 }
|
|
|
339 lu.UnTgz( flag.Args() )
|
|
|
340 os.Exit( 0 )
|
|
|
341 }
|
|
|
342
|
|
|
343 /* Here My Uplaad / Download Function */
|
|
|
344 host := "http://777.nazo.cc"
|
|
17
|
345
|
|
16
|
346 if *putfile {
|
|
|
347 if flag.NArg() > 0 {
|
|
|
348
|
|
|
349 // アップロードには，このプログラムで認証が必要:未実装
|
|
|
350 if !is_valid_user() {
|
|
|
351 fmt.Println( "wrong password." )
|
|
|
352 os.Exit( 0 )
|
|
|
353 }
|
|
|
354
|
|
|
355 for _, file := range flag.Args() {
|
|
|
356 if _, err := os.Stat( file ); os.IsNotExist( err ) {
|
|
|
357 fmt.Fprintf( os.Stderr, "%s: not exist\n", file )
|
|
|
358 }
|
|
|
359 }
|
|
|
360
|
|
|
361 url := host + "/uploader/index.cgi/goup/post"
|
|
|
362 //url = "http://localhost:3000/gofile"
|
|
|
363
|
|
|
364 var prefix string
|
|
|
365 if !*noprefix {
|
|
|
366 prefix = time.Now().Format( "200601021504_" )
|
|
|
367 }
|
|
|
368
|
|
|
369 // 圧縮
|
|
17
|
370 if *compress {
|
|
16
|
371 tmpfile, err := ioutil.TempFile( ".", "muxx" )
|
|
|
372 if err != nil {
|
|
|
373 log.Fatal( err )
|
|
|
374 }
|
|
|
375
|
|
|
376 tgz := prefix + tmpfile.Name() + ".tgz"
|
|
|
377 os.Rename( tmpfile.Name(), tgz )
|
|
17
|
378 defer os.Remove( tgz )
|
|
|
379 lu.Tgz( tgz, flag.Args() )
|
|
16
|
380 fmt.Printf( "archive & compress ...done [%s].\n", tgz )
|
|
|
381
|
|
17
|
382 if err := post_file( url, tgz ); err != nil {
|
|
|
383 log.Fatal( err )
|
|
16
|
384 }
|
|
17
|
385 fmt.Println( "upload ...done." )
|
|
|
386 // 非圧縮
|
|
16
|
387 } else {
|
|
|
388 for _, file := range flag.Args() {
|
|
|
389 if !*noprefix {
|
|
|
390 c, err := ioutil.ReadFile( file )
|
|
|
391 if err != nil {
|
|
|
392 log.Fatal( err )
|
|
|
393 }
|
|
|
394 if err := ioutil.WriteFile( prefix + file, c, 0644 ); err != nil {
|
|
|
395 log.Fatal( err )
|
|
|
396 }
|
|
|
397 }
|
|
|
398 if err := post_file( url, prefix + file ); err != nil {
|
|
|
399 log.Fatal( err )
|
|
|
400 }
|
|
|
401 fmt.Println( file + "\tupload ...done." )
|
|
|
402 }
|
|
|
403 }
|
|
|
404 }
|
|
17
|
405 os.Exit( 0 )
|
|
16
|
406 }
|
|
|
407
|
|
|
408 // ファイル一覧を取得
|
|
|
409 if *getfile {
|
|
|
410 url := host + "/uploader/index.cgi/goup/"
|
|
|
411 list := url_get( url )
|
|
|
412 files := strings.Split( string( list ), "\n" )
|
|
|
413
|
|
|
414 var filename [100]string
|
|
|
415 for i, file := range files {
|
|
|
416 info := strings.Split( file, "#" )
|
|
|
417 fmt.Printf( "[%3d] %-28s %10s %s\n", i + 1, info[0], info[1], info[2] )
|
|
|
418 filename[i] = info[0]
|
|
|
419 }
|
|
|
420
|
|
|
421 fmt.Print( "> " )
|
|
|
422 scanner := bufio.NewScanner( os.Stdin )
|
|
|
423 scanner.Scan()
|
|
|
424 input := scanner.Text()
|
|
|
425
|
|
|
426 if strings.Compare( input, "q" ) == 0 || strings.Compare( input, "bye" ) == 0 {
|
|
|
427 fmt.Println( "bye." )
|
|
|
428 os.Exit( 0 )
|
|
|
429 }
|
|
|
430
|
|
|
431 no, err := strconv.Atoi( input )
|
|
|
432 if err != nil || no < 1 || no > len( files ) {
|
|
|
433 fmt.Fprint( os.Stderr, "bad input\n" )
|
|
|
434 os.Exit( 0 )
|
|
|
435 }
|
|
|
436
|
|
|
437 url = host + "/uploader/files/goup/" + filename[ no - 1 ]
|
|
|
438 fmt.Printf( "downloading %s...\n", filename[ no - 1 ] )
|
|
|
439 bytes := url_get( url )
|
|
|
440 if err := ioutil.WriteFile( filename[ no - 1 ], bytes, 0644 ); err != nil {
|
|
|
441 log.Fatal( err )
|
|
|
442 }
|
|
|
443 os.Exit( 0 )
|
|
|
444 }
|
|
|
445
|
|
|
446 print_help()
|
|
|
447 }
|
|
|
448
|
|
|
449 // Subroutines
|
|
|
450 func url_get( url string ) []byte {
|
|
|
451 res, err := http.Get( url )
|
|
|
452 if err != nil {
|
|
|
453 log.Fatal( err )
|
|
|
454 }
|
|
|
455 b, err := ioutil.ReadAll( res.Body )
|
|
|
456 res.Body.Close()
|
|
|
457 if err != nil {
|
|
|
458 log.Fatal( err )
|
|
|
459 }
|
|
|
460 return b
|
|
|
461 }
|
|
|
462
|
|
|
463 func post_file( url, file string ) error {
|
|
|
464
|
|
|
465 f, err := os.Open( file )
|
|
|
466 if err != nil {
|
|
|
467 return err
|
|
|
468 }
|
|
|
469 defer f.Close()
|
|
|
470
|
|
|
471 buf := &bytes.Buffer{}
|
|
|
472 writer := multipart.NewWriter( buf )
|
|
|
473 fw, err := writer.CreateFormFile( "filename", file )
|
|
|
474 if err != nil {
|
|
|
475 return err
|
|
|
476 }
|
|
|
477
|
|
|
478 if _, err = io.Copy( fw, f ); err != nil {
|
|
|
479 return err
|
|
|
480 }
|
|
|
481
|
|
|
482 ct := writer.FormDataContentType()
|
|
|
483 writer.Close()
|
|
|
484
|
|
|
485 resp, err := http.Post( url, ct, buf )
|
|
|
486 if err != nil {
|
|
|
487 fmt.Fprintf( os.Stderr, "%s", resp.Status )
|
|
|
488 return err
|
|
|
489 }
|
|
|
490
|
|
|
491 return nil
|
|
|
492 }
|
|
|
493
|
|
|
494 func is_valid_user() bool {
|
|
|
495 return true
|
|
|
496 return false
|
|
|
497 }
|
|
|
498
|
|
|
499 func is_exist_files( files []string ) error {
|
|
|
500 for _, file := range files {
|
|
|
501 if _, err := os.Stat( file ); os.IsNotExist( err ) {
|
|
|
502 return fmt.Errorf( "%s: not exist\n", file )
|
|
|
503 }
|
|
|
504 }
|
|
|
505 return nil
|
|
|
506 }
|
|
|
507
|
|
|
508 func print_help() {
|
|
|
509 fmt.Fprintf( os.Stderr, "Usage of %s [%s]:\n", os.Args[0], version )
|
|
|
510 flag.PrintDefaults()
|
|
|
511 }
|
|
17
|
512
|
