-
Notifications
You must be signed in to change notification settings - Fork 485
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
yangchangpei
committed
Dec 24, 2020
1 parent
f54488c
commit 7dc76e8
Showing
5 changed files
with
180 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,134 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "crypto/md5" | ||
| "crypto/sha1" | ||
| "errors" | ||
| "hash" | ||
| "io" | ||
| "os" | ||
| "os/exec" | ||
| "path/filepath" | ||
| ) | ||
|
|
||
| var xJar = XJar{ | ||
| md5: []byte{#{xJar.md5}}, | ||
| sha1: []byte{#{xJar.sha1}}, | ||
| } | ||
|
|
||
| var xKey = XKey{ | ||
| algorithm: []byte{#{xKey.algorithm}}, | ||
| keysize: []byte{#{xKey.keysize}}, | ||
| ivsize: []byte{#{xKey.ivsize}}, | ||
| password: []byte{#{xKey.password}}, | ||
| } | ||
|
|
||
| func main() { | ||
| // search the jar to start | ||
| jar, err := JAR(os.Args) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| // parse jar name to absolute path | ||
| path, err := filepath.Abs(jar) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| // verify jar with MD5 | ||
| MD5, err := MD5(path) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| if bytes.Compare(MD5, xJar.md5) != 0 { | ||
| panic(errors.New("invalid jar with MD5")) | ||
| } | ||
|
|
||
| // verify jar with SHA-1 | ||
| SHA1, err := SHA1(path) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| if bytes.Compare(SHA1, xJar.sha1) != 0 { | ||
| panic(errors.New("invalid jar with SHA-1")) | ||
| } | ||
|
|
||
| // start java application | ||
| java := os.Args[1] | ||
| args := os.Args[2:] | ||
| key := bytes.Join([][]byte{ | ||
| xKey.algorithm, {13, 10}, | ||
| xKey.keysize, {13, 10}, | ||
| xKey.ivsize, {13, 10}, | ||
| xKey.password, {13, 10}, | ||
| }, []byte{}) | ||
| cmd := exec.Command(java, args...) | ||
| cmd.Stdin = bytes.NewReader(key) | ||
| cmd.Stdout = os.Stdout | ||
| cmd.Stderr = os.Stderr | ||
| err = cmd.Run() | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| } | ||
|
|
||
| // find jar name from args | ||
| func JAR(args []string) (string, error) { | ||
| var jar string | ||
|
|
||
| l := len(args) | ||
| for i := 1; i < l-1; i++ { | ||
| arg := args[i] | ||
| if arg == "-jar" { | ||
| jar = args[i+1] | ||
| } | ||
| } | ||
|
|
||
| if jar == "" { | ||
| return "", errors.New("unspecified jar name") | ||
| } | ||
|
|
||
| return jar, nil | ||
| } | ||
|
|
||
| // calculate file's MD5 | ||
| func MD5(path string) ([]byte, error) { | ||
| return HASH(path, md5.New()) | ||
| } | ||
|
|
||
| // calculate file's SHA-1 | ||
| func SHA1(path string) ([]byte, error) { | ||
| return HASH(path, sha1.New()) | ||
| } | ||
|
|
||
| // calculate file's HASH value with specified HASH Algorithm | ||
| func HASH(path string, hash hash.Hash) ([]byte, error) { | ||
| file, err := os.Open(path) | ||
|
|
||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| _, _err := io.Copy(hash, file) | ||
| if _err != nil { | ||
| return nil, _err | ||
| } | ||
|
|
||
| sum := hash.Sum(nil) | ||
|
|
||
| return sum, nil | ||
| } | ||
|
|
||
| type XJar struct { | ||
| md5 []byte | ||
| sha1 []byte | ||
| } | ||
|
|
||
| type XKey struct { | ||
| algorithm []byte | ||
| keysize []byte | ||
| ivsize []byte | ||
| password []byte | ||
| } |