57 lines
1.0 KiB
Go
57 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
|
|
"github.com/crewjam/go-xmlsec/xmldsig"
|
|
)
|
|
|
|
func main() {
|
|
doVerify := flag.Bool("v", false, "verify the document")
|
|
doSign := flag.Bool("s", false, "sign the document")
|
|
keyPath := flag.String("k", "", "the path to the key")
|
|
flag.Parse()
|
|
|
|
if !*doVerify && !*doSign {
|
|
fmt.Println("you must specify -v to verify or -s to sign")
|
|
os.Exit(1)
|
|
}
|
|
if *keyPath == "" {
|
|
fmt.Println("you must specify a key file")
|
|
os.Exit(1)
|
|
}
|
|
|
|
key, err := ioutil.ReadFile(*keyPath)
|
|
if err != nil {
|
|
fmt.Printf("%s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
buf, err := ioutil.ReadAll(os.Stdin)
|
|
|
|
if *doSign {
|
|
signedBuf, err := xmldsig.Sign(key, buf)
|
|
if err != nil {
|
|
fmt.Printf("%s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
os.Stdout.Write(signedBuf)
|
|
}
|
|
|
|
if *doVerify {
|
|
err := xmldsig.Verify(key, buf)
|
|
if err == xmldsig.ErrVerificationFailed {
|
|
fmt.Println("signature is not correct")
|
|
os.Exit(1)
|
|
}
|
|
if err != nil {
|
|
fmt.Printf("error: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Println("signature is correct")
|
|
}
|
|
}
|