|
0
|
1 /*
|
|
|
2 client.go : client-program.
|
|
13
|
3 Version : 1.3a
|
|
|
4 Last Change: 2018-11-14 水 16:58:29.
|
|
0
|
5
|
|
|
6 install to: rsearcher_root/
|
|
|
7
|
|
|
8 rsearcher_root/client.exe
|
|
|
9 + grsearcher.exe
|
|
|
10 + app.conf
|
|
12
|
11 + upgrade.bat
|
|
0
|
12 + auth.db ( temporary )
|
|
|
13 + hhs.db ( temporary )
|
|
12
|
14 + index.db ( 14 days )
|
|
3
|
15 + image/*.png, *.jpg
|
|
13
|
16 + manual.pdf
|
|
4
|
17 + .cache/* ( temporary )
|
|
0
|
18
|
|
8
|
19 $ client.exe -r 192.168.0.1:3910 # get new release
|
|
|
20 $ client.exe -u 192.168.0.1:3910 # get new dbs
|
|
0
|
21 $ client.exe 192.168.0.1:3910 0800012345 20171231 # get images
|
|
12
|
22 $ client.exe -a 192.168.0.1:3910 upfile.txt # upload file to /upload
|
|
|
23 $ client.exe -m 192.168.0.1:3910 hhs.db # upload file to /db
|
|
0
|
24 */
|
|
|
25 package main
|
|
|
26
|
|
|
27 import (
|
|
|
28 "archive/tar"
|
|
|
29 "bytes"
|
|
|
30 "compress/gzip"
|
|
|
31 "flag"
|
|
|
32 "fmt"
|
|
|
33 "io"
|
|
|
34 "io/ioutil"
|
|
|
35 "log"
|
|
|
36 "net/http"
|
|
|
37 "os"
|
|
|
38 "path/filepath"
|
|
|
39 "regexp"
|
|
|
40 "time"
|
|
|
41 )
|
|
|
42
|
|
4
|
43 var version string
|
|
|
44
|
|
|
45 func init() {
|
|
13
|
46 version = "1.3a" // manual.pdf version
|
|
4
|
47 }
|
|
|
48
|
|
0
|
49 func main() {
|
|
13
|
50 var upgrade = flag.Bool( "r", false, "get new release" ) // get grsearcher.exe, crypto.exe client.exe manual.pdf
|
|
12
|
51 var update = flag.Bool( "u", false, "get new DBs" ) // get auth.db, hhs.db, index.db(14,28)
|
|
|
52 var managedb = flag.Bool( "m", false, "upload to /db" ) // management db
|
|
|
53 var upload = flag.Bool( "a", false, "upload to /upload" ) // post somefile
|
|
|
54 var cnttime = flag.Bool( "t", false, "measure work time" ) // count time
|
|
|
55 var printver = flag.Bool( "v", false, "print version" ) // print version
|
|
0
|
56 flag.Parse()
|
|
|
57
|
|
4
|
58 // print version
|
|
|
59 if *printver {
|
|
8
|
60 fmt.Println( "client [ ver", version, "]" )
|
|
4
|
61 os.Exit( 0 )
|
|
|
62 }
|
|
|
63
|
|
8
|
64 st := time.Now() // start time
|
|
|
65
|
|
4
|
66 // upload
|
|
|
67 if *upload {
|
|
|
68 if flag.NArg() != 2 {
|
|
12
|
69 fmt.Fprint( os.Stderr, "bad host or file.\n" )
|
|
4
|
70 os.Exit( 1 )
|
|
|
71 }
|
|
|
72 host := flag.Args()[0]
|
|
|
73 file := flag.Args()[1]
|
|
|
74 fmt.Printf( "Sending %s...", file )
|
|
13
|
75 if message, err := post_file( host, "/upload/", file ); err != nil {
|
|
12
|
76 log.Fatal( err )
|
|
|
77 } else {
|
|
|
78 fmt.Println( string( message ) )
|
|
|
79 }
|
|
|
80
|
|
|
81 if *cnttime {
|
|
|
82 fmt.Println( "---", time.Now().Sub( st ) )
|
|
|
83 }
|
|
|
84 os.Exit( 0 )
|
|
|
85 }
|
|
|
86
|
|
|
87 // management db
|
|
|
88 if *managedb {
|
|
|
89 if flag.NArg() != 2 {
|
|
|
90 fmt.Fprint( os.Stderr, "bad remote host.\n" )
|
|
|
91 os.Exit( 1 )
|
|
|
92 }
|
|
|
93 host := flag.Args()[0]
|
|
|
94 file := flag.Args()[1]
|
|
|
95 fmt.Print( "Management DB..." )
|
|
|
96 if _, err := post_file( host, "/mngdb/", file ); err != nil {
|
|
4
|
97 log.Fatal( err )
|
|
|
98 }
|
|
8
|
99 if *cnttime {
|
|
|
100 fmt.Println( "---", time.Now().Sub( st ) )
|
|
|
101 }
|
|
4
|
102 os.Exit( 0 )
|
|
|
103 }
|
|
|
104
|
|
|
105 // software upgrade
|
|
0
|
106 if *upgrade {
|
|
|
107 if flag.NArg() != 1 {
|
|
12
|
108 fmt.Fprint( os.Stderr, "bad remote host.\n" )
|
|
0
|
109 os.Exit( 1 )
|
|
|
110 }
|
|
|
111 host := flag.Args()[0]
|
|
13
|
112
|
|
|
113 fmt.Print( "Getting client.exe..." )
|
|
|
114 if _, err := get_file( host, "/release/client.exe", "client1.exe" ); err != nil {
|
|
0
|
115 log.Fatal( err )
|
|
|
116 }
|
|
4
|
117 fmt.Print( "Getting crypto.exe..." )
|
|
|
118 if _, err := get_file( host, "/release/crypto.exe", "crypto.exe" ); err != nil {
|
|
|
119 log.Fatal( err )
|
|
|
120 }
|
|
13
|
121 fmt.Print( "Getting grsearcher.exe..." )
|
|
|
122 if _, err := get_file( host, "/release/grsearcher.exe", "grsearcher.exe" ); err != nil {
|
|
0
|
123 log.Fatal( err )
|
|
|
124 }
|
|
13
|
125 fmt.Print( "Getting manual.pdf..." )
|
|
|
126 if _, err := get_file( host, "/release/manual.pdf", "manual.pdf" ); err != nil {
|
|
|
127 log.Fatal( err )
|
|
|
128 }
|
|
|
129
|
|
8
|
130 if *cnttime {
|
|
|
131 fmt.Println( "---", time.Now().Sub( st ) )
|
|
|
132 }
|
|
0
|
133 os.Exit( 0 )
|
|
|
134 }
|
|
|
135
|
|
4
|
136 // database update
|
|
0
|
137 if *update {
|
|
|
138 if flag.NArg() != 1 {
|
|
12
|
139 fmt.Fprint( os.Stderr, "bad remote host.\n" )
|
|
0
|
140 os.Exit( 1 )
|
|
|
141 }
|
|
1
|
142 fmt.Print( "Getting auth.db..." )
|
|
0
|
143 host := flag.Args()[0]
|
|
|
144 if _, err := get_file( host, "/db/auth.db", "auth.db" ); err != nil {
|
|
|
145 log.Fatal( err )
|
|
|
146 }
|
|
1
|
147 fmt.Print( "Getting hhs.db..." )
|
|
0
|
148 if _, err := get_file( host, "/db/hhs.db", "hhs.db" ); err != nil {
|
|
|
149 log.Fatal( err )
|
|
|
150 }
|
|
|
151
|
|
4
|
152 if time.Now().Day() % 14 == 0 {
|
|
1
|
153 fmt.Print( "Getting index.db..." )
|
|
0
|
154 if _, err := get_file( host, "/db/index.db", "index.db" ); err != nil {
|
|
|
155 log.Fatal( err )
|
|
|
156 }
|
|
|
157 }
|
|
8
|
158
|
|
|
159 if *cnttime {
|
|
|
160 fmt.Println( "---", time.Now().Sub( st ) )
|
|
|
161 }
|
|
0
|
162 os.Exit( 0 )
|
|
|
163 }
|
|
|
164
|
|
4
|
165 // main : image download
|
|
0
|
166 if flag.NArg() != 3 {
|
|
12
|
167 fmt.Fprint( os.Stderr, "bad argument.\n" )
|
|
0
|
168 os.Exit( 1 )
|
|
|
169 }
|
|
|
170
|
|
|
171 host := flag.Args()[0]
|
|
|
172 h := flag.Args()[1]
|
|
|
173 d := flag.Args()[2]
|
|
|
174
|
|
|
175 r, err := regexp.Compile( "^0[1238][0-9]{8}$" )
|
|
|
176 if err != nil {
|
|
|
177 log.Fatal( err )
|
|
|
178 }
|
|
|
179 m := r.FindString( h )
|
|
|
180 if len( m ) == 0 {
|
|
12
|
181 fmt.Fprint( os.Stderr, "bad hno.\n" )
|
|
0
|
182 os.Exit( 1 )
|
|
|
183 }
|
|
|
184
|
|
|
185 rfile := "/images/" + d + "/" + h + ".tgz"
|
|
12
|
186 fmt.Print( "Getting", rfile )
|
|
0
|
187 b, err := get_file( host, rfile, "" )
|
|
|
188 if err != nil {
|
|
|
189 log.Fatal( err )
|
|
|
190 }
|
|
|
191 if err := untgz_images( d, b ); err != nil {
|
|
|
192 log.Fatal( err )
|
|
|
193 }
|
|
8
|
194
|
|
|
195 if *cnttime {
|
|
|
196 fmt.Println( "---", time.Now().Sub( st ) )
|
|
|
197 }
|
|
0
|
198 }
|
|
|
199
|
|
|
200 /* Subroutines */
|
|
12
|
201 // HTTP POST
|
|
|
202 func post_file( host, path, file string ) ( []byte, error ) {
|
|
|
203 f, err := os.Open( file )
|
|
|
204 if err != nil {
|
|
|
205 return nil, err
|
|
|
206 }
|
|
|
207 defer f.Close()
|
|
|
208
|
|
|
209 res, err := http.Post( "http://" + host + path + file, "binary/octet-stream", f )
|
|
|
210 if err != nil {
|
|
|
211 return nil, err
|
|
|
212 }
|
|
|
213 defer res.Body.Close()
|
|
|
214 return ioutil.ReadAll( res.Body )
|
|
|
215 }
|
|
|
216
|
|
0
|
217 // HTTP GET
|
|
|
218 func get_file( host, rfile, lfile string ) ( []byte, error ) {
|
|
|
219 res, err := http.Get( "http://" + host + "/" + rfile )
|
|
|
220 if err != nil {
|
|
|
221 return nil, err
|
|
|
222 }
|
|
1
|
223 fmt.Println( "\t[" + res.Status + "]" )
|
|
|
224 if res.StatusCode == http.StatusNotFound {
|
|
|
225 return nil, nil
|
|
|
226 }
|
|
0
|
227 b, err := ioutil.ReadAll( res.Body )
|
|
|
228 res.Body.Close()
|
|
|
229 if err != nil {
|
|
|
230 return nil, err
|
|
|
231 }
|
|
|
232 if lfile != "" {
|
|
|
233 if err := ioutil.WriteFile( lfile, b, 0644 ); err != nil {
|
|
|
234 return nil, err
|
|
|
235 }
|
|
|
236 return nil, nil
|
|
|
237 }
|
|
|
238 return b, nil
|
|
|
239 }
|
|
|
240
|
|
|
241 // Un-Tar/Gzip
|
|
|
242 func untgz_images( d string, b []byte ) error {
|
|
|
243 br := bytes.NewReader( b )
|
|
|
244 zr, err := gzip.NewReader( br )
|
|
|
245 if err != nil {
|
|
|
246 return err
|
|
|
247 }
|
|
|
248
|
|
|
249 tr := tar.NewReader( zr )
|
|
|
250 for i := 1; ; i++ {
|
|
|
251 //hdr, err := tr.Next()
|
|
|
252 _, err := tr.Next()
|
|
|
253 if err == io.EOF {
|
|
|
254 break // End of archive
|
|
|
255 }
|
|
|
256 if err != nil {
|
|
|
257 return err
|
|
|
258 }
|
|
|
259 //f, _ := os.Create( hdr.Name )
|
|
3
|
260 fn := fmt.Sprintf( "%s_%d", d, i ) // 1.0a
|
|
0
|
261
|
|
|
262 fn = filepath.Join( ".cache", fn )
|
|
|
263 f, _ := os.Create( fn )
|
|
|
264 if _, err := io.Copy( f, tr ); err != nil {
|
|
|
265 return err
|
|
|
266 }
|
|
|
267 f.Close()
|
|
|
268 }
|
|
|
269
|
|
|
270 if err := zr.Close(); err != nil {
|
|
|
271 return err
|
|
|
272 }
|
|
|
273 return nil
|
|
|
274 }
|
|
|
275
|
