|
0
|
1 /*
|
|
|
2 client.go : client-program.
|
|
1
|
3 Version : 1.0
|
|
|
4 Last Change: 2018-10-01 Mon 21:45:18.
|
|
0
|
5
|
|
|
6 install to: rsearcher_root/
|
|
|
7
|
|
|
8 rsearcher_root/client.exe
|
|
|
9 + grsearcher.exe
|
|
|
10 + app.conf
|
|
|
11 + verup.bat
|
|
|
12 + auth.db ( temporary )
|
|
|
13 + hhs.db ( temporary )
|
|
|
14 + index.db ( 10 days )
|
|
|
15 + image/*.png
|
|
|
16 + doc/
|
|
|
17 + .cache/*.jpg ( temporary )
|
|
|
18
|
|
|
19 $ client.exe -r 192.168.0.1:3910 # get new release
|
|
|
20 $ client.exe -u 192.168.0.1:3910 # get new dbs
|
|
|
21 $ client.exe 192.168.0.1:3910 0800012345 20171231 # get images
|
|
|
22 */
|
|
|
23 package main
|
|
|
24
|
|
|
25 import (
|
|
|
26 "archive/tar"
|
|
|
27 "bytes"
|
|
|
28 "compress/gzip"
|
|
|
29 "flag"
|
|
|
30 "fmt"
|
|
|
31 "io"
|
|
|
32 "io/ioutil"
|
|
|
33 "log"
|
|
|
34 "net/http"
|
|
|
35 "os"
|
|
|
36 "path/filepath"
|
|
|
37 "regexp"
|
|
|
38 "time"
|
|
|
39 )
|
|
|
40
|
|
|
41 func main() {
|
|
|
42 var upgrade = flag.Bool( "r", false, "" ) // get grsearcher.exe, client.exe
|
|
|
43 var update = flag.Bool( "u", false, "" ) // get auth.db, hhs.db, index.db(5day)
|
|
|
44 flag.Parse()
|
|
|
45
|
|
|
46 if *upgrade {
|
|
|
47 if flag.NArg() != 1 {
|
|
|
48 fmt.Fprintf( os.Stderr, "no remote host.\n" )
|
|
|
49 os.Exit( 1 )
|
|
|
50 }
|
|
1
|
51 fmt.Print( "Getting grsearcher.exe..." )
|
|
0
|
52 host := flag.Args()[0]
|
|
|
53 if _, err := get_file( host, "/release/grsearcher.exe", "grsearcher.exe" ); err != nil {
|
|
|
54 log.Fatal( err )
|
|
|
55 }
|
|
1
|
56 fmt.Print( "Getting client.exe..." )
|
|
0
|
57 if _, err := get_file( host, "/release/client.exe", "client1.exe" ); err != nil {
|
|
|
58 log.Fatal( err )
|
|
|
59 }
|
|
|
60 os.Exit( 0 )
|
|
|
61 }
|
|
|
62
|
|
|
63 if *update {
|
|
|
64 if flag.NArg() != 1 {
|
|
|
65 fmt.Fprintf( os.Stderr, "no remote host.\n" )
|
|
|
66 os.Exit( 1 )
|
|
|
67 }
|
|
1
|
68 fmt.Print( "Getting auth.db..." )
|
|
0
|
69 host := flag.Args()[0]
|
|
|
70 if _, err := get_file( host, "/db/auth.db", "auth.db" ); err != nil {
|
|
|
71 log.Fatal( err )
|
|
|
72 }
|
|
1
|
73 fmt.Print( "Getting hhs.db..." )
|
|
0
|
74 if _, err := get_file( host, "/db/hhs.db", "hhs.db" ); err != nil {
|
|
|
75 log.Fatal( err )
|
|
|
76 }
|
|
|
77
|
|
|
78 if time.Now().Day() % 10 == 0 {
|
|
1
|
79 fmt.Print( "Getting index.db..." )
|
|
0
|
80 if _, err := get_file( host, "/db/index.db", "index.db" ); err != nil {
|
|
|
81 log.Fatal( err )
|
|
|
82 }
|
|
|
83 }
|
|
|
84 os.Exit( 0 )
|
|
|
85 }
|
|
|
86
|
|
|
87 // main
|
|
|
88 if flag.NArg() != 3 {
|
|
|
89 fmt.Fprintf( os.Stderr, "bad argument.\n" )
|
|
|
90 os.Exit( 1 )
|
|
|
91 }
|
|
|
92
|
|
|
93 host := flag.Args()[0]
|
|
|
94 h := flag.Args()[1]
|
|
|
95 d := flag.Args()[2]
|
|
|
96
|
|
|
97 r, err := regexp.Compile( "^0[1238][0-9]{8}$" )
|
|
|
98 if err != nil {
|
|
|
99 log.Fatal( err )
|
|
|
100 }
|
|
|
101 m := r.FindString( h )
|
|
|
102 if len( m ) == 0 {
|
|
|
103 fmt.Fprintf( os.Stderr, "bad hno.\n" )
|
|
|
104 os.Exit( 1 )
|
|
|
105 }
|
|
|
106
|
|
|
107 rfile := "/images/" + d + "/" + h + ".tgz"
|
|
|
108 b, err := get_file( host, rfile, "" )
|
|
|
109 if err != nil {
|
|
|
110 log.Fatal( err )
|
|
|
111 }
|
|
|
112 if err := untgz_images( d, b ); err != nil {
|
|
|
113 log.Fatal( err )
|
|
|
114 }
|
|
|
115 }
|
|
|
116
|
|
|
117 /* Subroutines */
|
|
|
118 // HTTP GET
|
|
|
119 func get_file( host, rfile, lfile string ) ( []byte, error ) {
|
|
|
120 res, err := http.Get( "http://" + host + "/" + rfile )
|
|
|
121 if err != nil {
|
|
|
122 return nil, err
|
|
|
123 }
|
|
1
|
124 fmt.Println( "\t[" + res.Status + "]" )
|
|
|
125 if res.StatusCode == http.StatusNotFound {
|
|
|
126 return nil, nil
|
|
|
127 }
|
|
0
|
128 b, err := ioutil.ReadAll( res.Body )
|
|
|
129 res.Body.Close()
|
|
|
130 if err != nil {
|
|
|
131 return nil, err
|
|
|
132 }
|
|
|
133 if lfile != "" {
|
|
|
134 if err := ioutil.WriteFile( lfile, b, 0644 ); err != nil {
|
|
|
135 return nil, err
|
|
|
136 }
|
|
|
137 return nil, nil
|
|
|
138 }
|
|
|
139 return b, nil
|
|
|
140 }
|
|
|
141
|
|
|
142 // Un-Tar/Gzip
|
|
|
143 func untgz_images( d string, b []byte ) error {
|
|
|
144 br := bytes.NewReader( b )
|
|
|
145 zr, err := gzip.NewReader( br )
|
|
|
146 if err != nil {
|
|
|
147 return err
|
|
|
148 }
|
|
|
149
|
|
|
150 tr := tar.NewReader( zr )
|
|
|
151 for i := 1; ; i++ {
|
|
|
152 //hdr, err := tr.Next()
|
|
|
153 _, err := tr.Next()
|
|
|
154 if err == io.EOF {
|
|
|
155 break // End of archive
|
|
|
156 }
|
|
|
157 if err != nil {
|
|
|
158 return err
|
|
|
159 }
|
|
|
160 //f, _ := os.Create( hdr.Name )
|
|
|
161 fn := fmt.Sprintf( "%s_%d.jpg", d, i )
|
|
1
|
162 fn = fmt.Sprintf( "00000000_%d", i ) // for test
|
|
0
|
163
|
|
|
164 fn = filepath.Join( ".cache", fn )
|
|
|
165 f, _ := os.Create( fn )
|
|
|
166 if _, err := io.Copy( f, tr ); err != nil {
|
|
|
167 return err
|
|
|
168 }
|
|
|
169 f.Close()
|
|
|
170 }
|
|
|
171
|
|
|
172 if err := zr.Close(); err != nil {
|
|
|
173 return err
|
|
|
174 }
|
|
|
175 return nil
|
|
|
176 }
|
|
|
177
|
