add private keygen endpoint, minor refactor, add first test
This commit is contained in:
parent
2fe128a017
commit
153851b41d
|
@ -69,7 +69,7 @@ func main() {
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
mockid.Route(jwksPrefix, privkey)
|
mux := mockid.Route(jwksPrefix, privkey)
|
||||||
|
|
||||||
fs := http.FileServer(http.Dir("./public"))
|
fs := http.FileServer(http.Dir("./public"))
|
||||||
http.Handle("/", fs)
|
http.Handle("/", fs)
|
||||||
|
@ -82,7 +82,7 @@ func main() {
|
||||||
|
|
||||||
fmt.Printf("Serving on port %d\n", port)
|
fmt.Printf("Serving on port %d\n", port)
|
||||||
go func() {
|
go func() {
|
||||||
log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), nil))
|
log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), mux))
|
||||||
done <- true
|
done <- true
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,56 @@
|
||||||
|
package mockid
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/ecdsa"
|
||||||
|
"crypto/rsa"
|
||||||
|
"encoding/base64"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"math/big"
|
||||||
|
|
||||||
|
"git.rootprojects.org/root/keypairs"
|
||||||
|
)
|
||||||
|
|
||||||
|
// MarshalJWKPrivateKey outputs the given private key as JWK
|
||||||
|
func MarshalJWKPrivateKey(privkey keypairs.PrivateKey) []byte {
|
||||||
|
// thumbprint keys are alphabetically sorted and only include the necessary public parts
|
||||||
|
switch k := privkey.(type) {
|
||||||
|
case *rsa.PrivateKey:
|
||||||
|
return MarshalRSAPrivateKey(k)
|
||||||
|
case *ecdsa.PrivateKey:
|
||||||
|
return MarshalECPrivateKey(k)
|
||||||
|
default:
|
||||||
|
// this is unreachable because we know the types that we pass in
|
||||||
|
log.Printf("keytype: %t, %+v\n", privkey, privkey)
|
||||||
|
panic(keypairs.ErrInvalidPublicKey)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarshalECPrivateKey will output the given private key as JWK
|
||||||
|
func MarshalECPrivateKey(k *ecdsa.PrivateKey) []byte {
|
||||||
|
crv := k.Curve.Params().Name
|
||||||
|
d := base64.RawURLEncoding.EncodeToString(k.D.Bytes())
|
||||||
|
x := base64.RawURLEncoding.EncodeToString(k.X.Bytes())
|
||||||
|
y := base64.RawURLEncoding.EncodeToString(k.Y.Bytes())
|
||||||
|
return []byte(fmt.Sprintf(
|
||||||
|
`{"crv":%q,"d":%q,"kty":"EC","x":%q,"y":%q}`,
|
||||||
|
crv, d, x, y,
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarshalRSAPrivateKey will output the given private key as JWK
|
||||||
|
func MarshalRSAPrivateKey(pk *rsa.PrivateKey) []byte {
|
||||||
|
e := base64.RawURLEncoding.EncodeToString(big.NewInt(int64(pk.E)).Bytes())
|
||||||
|
n := base64.RawURLEncoding.EncodeToString(pk.N.Bytes())
|
||||||
|
d := base64.RawURLEncoding.EncodeToString(pk.D.Bytes())
|
||||||
|
p := base64.RawURLEncoding.EncodeToString(pk.Primes[0].Bytes())
|
||||||
|
q := base64.RawURLEncoding.EncodeToString(pk.Primes[1].Bytes())
|
||||||
|
dp := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Dp.Bytes())
|
||||||
|
dq := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Dq.Bytes())
|
||||||
|
qi := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Qinv.Bytes())
|
||||||
|
return []byte(fmt.Sprintf(
|
||||||
|
`{"d":%q,"dp":%q,"dq":%q,"e":%q,"kty":"RSA","n":%q,"p":%q,"q":%q,"qi":%q}`,
|
||||||
|
d, dp, dq, e, n, p, q, qi,
|
||||||
|
))
|
||||||
|
}
|
570
mockid/mockid.go
570
mockid/mockid.go
|
@ -5,30 +5,20 @@ import (
|
||||||
"crypto/ecdsa"
|
"crypto/ecdsa"
|
||||||
"crypto/rand"
|
"crypto/rand"
|
||||||
"crypto/rsa"
|
"crypto/rsa"
|
||||||
"crypto/sha1"
|
|
||||||
"crypto/sha256"
|
"crypto/sha256"
|
||||||
"crypto/sha512"
|
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
|
||||||
"log"
|
|
||||||
"math/big"
|
"math/big"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"git.rootprojects.org/root/keypairs"
|
"git.rootprojects.org/root/keypairs"
|
||||||
"git.rootprojects.org/root/keypairs/keyfetch"
|
|
||||||
|
|
||||||
//jwt "github.com/dgrijalva/jwt-go"
|
//jwt "github.com/dgrijalva/jwt-go"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type PublicJWK struct {
|
type PublicJWK struct {
|
||||||
|
@ -67,7 +57,7 @@ var defaultReplyTo string
|
||||||
var nonces sync.Map
|
var nonces sync.Map
|
||||||
var salt []byte
|
var salt []byte
|
||||||
|
|
||||||
func init() {
|
func Init() {
|
||||||
var err error
|
var err error
|
||||||
salt64 := os.Getenv("SALT")
|
salt64 := os.Getenv("SALT")
|
||||||
salt, err = base64.RawURLEncoding.DecodeString(salt64)
|
salt, err = base64.RawURLEncoding.DecodeString(salt64)
|
||||||
|
@ -89,507 +79,6 @@ func init() {
|
||||||
*/
|
*/
|
||||||
}
|
}
|
||||||
|
|
||||||
func Route(jwksPrefix string, privkey keypairs.PrivateKey) {
|
|
||||||
// TODO get from main()
|
|
||||||
tokPrefix := jwksPrefix
|
|
||||||
pubkey := keypairs.NewPublicKey(privkey.Public())
|
|
||||||
|
|
||||||
http.HandleFunc("/api/new-nonce", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
baseURL := getBaseURL(r)
|
|
||||||
/*
|
|
||||||
res.statusCode = 200;
|
|
||||||
res.setHeader("Cache-Control", "max-age=0, no-cache, no-store");
|
|
||||||
// TODO
|
|
||||||
//res.setHeader("Date", "Sun, 10 Mar 2019 08:04:45 GMT");
|
|
||||||
// is this the expiration of the nonce itself? methinks maybe so
|
|
||||||
//res.setHeader("Expires", "Sun, 10 Mar 2019 08:04:45 GMT");
|
|
||||||
// TODO use one of the registered domains
|
|
||||||
//var indexUrl = "https://acme-staging-v02.api.letsencrypt.org/index"
|
|
||||||
*/
|
|
||||||
//var port = (state.config.ipc && state.config.ipc.port || state._ipc.port || undefined);
|
|
||||||
//var indexUrl = "http://localhost:" + port + "/index";
|
|
||||||
indexUrl := baseURL + "/index"
|
|
||||||
w.Header().Set("Link", "<"+indexUrl+">;rel=\"index\"")
|
|
||||||
w.Header().Set("Cache-Control", "max-age=0, no-cache, no-store")
|
|
||||||
w.Header().Set("Pragma", "no-cache")
|
|
||||||
//res.setHeader("Strict-Transport-Security", "max-age=604800");
|
|
||||||
|
|
||||||
w.Header().Set("X-Frame-Options", "DENY")
|
|
||||||
issueNonce(w, r)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/api/new-account", requireNonce(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
// Try to decode the request body into the struct. If there is an error,
|
|
||||||
// respond to the client with the error message and a 400 status code.
|
|
||||||
data := map[string]string{}
|
|
||||||
err := json.NewDecoder(r.Body).Decode(&data)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO check DNS for MX records
|
|
||||||
parts := strings.Split(data["to"], ", <>\n\r\t")
|
|
||||||
to := parts[0]
|
|
||||||
if len(parts) > 1 || !strings.Contains(to, "@") {
|
|
||||||
http.Error(w, "invalid email address", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
token, err := uuid.NewRandom()
|
|
||||||
if nil != err {
|
|
||||||
// nothing else to do if we run out of random
|
|
||||||
// or are on a platform that doesn't support random
|
|
||||||
panic(fmt.Errorf("random bytes read failure: %w", err))
|
|
||||||
}
|
|
||||||
token64 := base64.RawURLEncoding.EncodeToString([]byte(token[:]))
|
|
||||||
// hash token to prevent fs read timing attacks
|
|
||||||
hash := sha1.Sum(append(token[:], salt...))
|
|
||||||
tokname := base64.RawURLEncoding.EncodeToString(hash[:])
|
|
||||||
if err := ioutil.WriteFile(
|
|
||||||
filepath.Join(tokPrefix, tokname+".tok.txt"),
|
|
||||||
[]byte(`{"comment":"I have no idea..."}`),
|
|
||||||
os.FileMode(0600),
|
|
||||||
); nil != err {
|
|
||||||
http.Error(w, "database connection failed when writing verification token", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
subject := "Verify New Account"
|
|
||||||
// TODO go tpl
|
|
||||||
// TODO determine OS and Browser from user agent
|
|
||||||
baseURL := getBaseURL(r)
|
|
||||||
text := fmt.Sprintf(
|
|
||||||
"It looks like you just tried to register a new Pocket ID account.\n\n Verify account: %s/verify/%s\n\nNot you? Just ignore this message.",
|
|
||||||
baseURL, token64,
|
|
||||||
)
|
|
||||||
_, err = SendSimpleMessage(to, defaultFrom, subject, text, defaultReplyTo)
|
|
||||||
if nil != err {
|
|
||||||
// TODO neuter mailgun output
|
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Fprintf(w, `{ "success": true, "error": "" }%s`, "\n")
|
|
||||||
}))
|
|
||||||
|
|
||||||
// TODO use chi
|
|
||||||
http.HandleFunc("/verify/", requireNonce(func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
parts := strings.Split(r.URL.Path, "/")
|
|
||||||
if 3 != len(parts) {
|
|
||||||
http.Error(w, "invalid url path", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
token64 := parts[2]
|
|
||||||
token, err := base64.RawURLEncoding.DecodeString(token64)
|
|
||||||
if err != nil || 0 == len(token) {
|
|
||||||
http.Error(w, "invalid url path", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// hash token to prevent fs read timing attacks
|
|
||||||
hash := sha1.Sum(append(token, salt...))
|
|
||||||
tokname := base64.RawURLEncoding.EncodeToString(hash[:])
|
|
||||||
tokfile := filepath.Join(tokPrefix, tokname+".tok.txt")
|
|
||||||
_, err = ioutil.ReadFile(tokfile)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "database connection failed when reading verification token", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
os.Remove(tokfile)
|
|
||||||
|
|
||||||
fmt.Fprintf(w, `{ "success": true, "error": "" }%s`, "\n")
|
|
||||||
}))
|
|
||||||
|
|
||||||
http.HandleFunc("/api/jwks", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Printf("%s %s %s", r.Method, r.Host, r.URL.Path)
|
|
||||||
if "POST" != r.Method {
|
|
||||||
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tok := make(map[string]interface{})
|
|
||||||
decoder := json.NewDecoder(r.Body)
|
|
||||||
err := decoder.Decode(&tok)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Request: invalid json", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
defer r.Body.Close()
|
|
||||||
|
|
||||||
// TODO better, JSON error messages
|
|
||||||
if _, ok := tok["d"]; ok {
|
|
||||||
http.Error(w, "Bad Request: private key", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
kty, _ := tok["kty"].(string)
|
|
||||||
switch kty {
|
|
||||||
case "EC":
|
|
||||||
postEC(jwksPrefix, tok, w, r)
|
|
||||||
case "RSA":
|
|
||||||
postRSA(jwksPrefix, tok, w, r)
|
|
||||||
default:
|
|
||||||
http.Error(w, "Bad Request: only EC and RSA keys are supported", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/access_token", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
|
||||||
_, _, token := GenToken(getBaseURL(r), privkey, r.URL.Query())
|
|
||||||
fmt.Fprintf(w, token)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/inspect_token", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
token := r.Header.Get("Authorization")
|
|
||||||
log.Printf("%s %s %s\n", r.Method, r.URL.Path, token)
|
|
||||||
|
|
||||||
if "" == token {
|
|
||||||
token = r.URL.Query().Get("access_token")
|
|
||||||
if "" == token {
|
|
||||||
http.Error(w, "Bad Format: missing Authorization header and 'access_token' query", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
parts := strings.Split(token, " ")
|
|
||||||
if 2 != len(parts) {
|
|
||||||
http.Error(w, "Bad Format: expected Authorization header to be in the format of 'Bearer <Token>'", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
token = parts[1]
|
|
||||||
}
|
|
||||||
|
|
||||||
parts := strings.Split(token, ".")
|
|
||||||
if 3 != len(parts) {
|
|
||||||
http.Error(w, "Bad Format: token should be in the format of <protected-header>.<payload>.<signature>", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
protected64 := parts[0]
|
|
||||||
payload64 := parts[1]
|
|
||||||
signature64 := parts[2]
|
|
||||||
|
|
||||||
protectedB, err := base64.RawURLEncoding.DecodeString(protected64)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: token's header should be URL-safe base64 encoded", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
payloadB, err := base64.RawURLEncoding.DecodeString(payload64)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: token's payload should be URL-safe base64 encoded", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// TODO verify signature
|
|
||||||
sig, err := base64.RawURLEncoding.DecodeString(signature64)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: token's signature should be URL-safe base64 encoded", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
errors := []string{}
|
|
||||||
|
|
||||||
protected := map[string]interface{}{}
|
|
||||||
err = json.Unmarshal(protectedB, &protected)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: token's header should be URL-safe base64-encoded JSON", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
kid, kidOK := protected["kid"].(string)
|
|
||||||
// TODO parse jwkM
|
|
||||||
_, jwkOK := protected["jwk"]
|
|
||||||
if !kidOK && !jwkOK {
|
|
||||||
errors = append(errors, "must have either header.kid or header.jwk")
|
|
||||||
}
|
|
||||||
|
|
||||||
data := map[string]interface{}{}
|
|
||||||
err = json.Unmarshal(payloadB, &data)
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: token's payload should be URL-safe base64-encoded JSON", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
iss, issOK := data["iss"].(string)
|
|
||||||
if !jwkOK && !issOK {
|
|
||||||
errors = append(errors, "payload.iss must exist to complement header.kid")
|
|
||||||
}
|
|
||||||
|
|
||||||
pub, err := keyfetch.OIDCJWK(kid, iss)
|
|
||||||
if nil != err {
|
|
||||||
fmt.Println("couldn't fetch pub key:")
|
|
||||||
fmt.Println(err)
|
|
||||||
}
|
|
||||||
fmt.Println("fetched pub key:")
|
|
||||||
fmt.Println(pub)
|
|
||||||
|
|
||||||
hash := sha256.Sum256([]byte(fmt.Sprintf("%s.%s", protected64, payload64)))
|
|
||||||
verified := JOSEVerify(pub, hash[:], sig)
|
|
||||||
|
|
||||||
inspected := &InspectableToken{
|
|
||||||
Public: pub,
|
|
||||||
Protected: protected,
|
|
||||||
Payload: data,
|
|
||||||
Signature: signature64,
|
|
||||||
Verified: verified,
|
|
||||||
Errors: errors,
|
|
||||||
}
|
|
||||||
|
|
||||||
tokenB, _ := json.MarshalIndent(inspected, "", " ")
|
|
||||||
if nil != err {
|
|
||||||
http.Error(w, "Bad Format: malformed token, or malformed jwk at issuer url", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
fmt.Fprintf(w, string(tokenB))
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/authorization_header", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
|
||||||
|
|
||||||
var header string
|
|
||||||
headers, _ := r.URL.Query()["header"]
|
|
||||||
if 0 == len(headers) {
|
|
||||||
header = "Authorization"
|
|
||||||
} else {
|
|
||||||
header = headers[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
var prefix string
|
|
||||||
prefixes, _ := r.URL.Query()["prefix"]
|
|
||||||
if 0 == len(prefixes) {
|
|
||||||
prefix = "Bearer "
|
|
||||||
} else {
|
|
||||||
prefix = prefixes[0]
|
|
||||||
}
|
|
||||||
|
|
||||||
_, _, token := GenToken(getBaseURL(r), privkey, r.URL.Query())
|
|
||||||
fmt.Fprintf(w, "%s: %s%s", header, prefix, token)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/key.jwk.json", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Printf("%s %s", r.Method, r.URL.Path)
|
|
||||||
jwk := string(MarshalJWKPrivateKey(privkey))
|
|
||||||
jwk = strings.Replace(jwk, `{"`, `{ "`, 1)
|
|
||||||
jwk = strings.Replace(jwk, `",`, `", `, -1)
|
|
||||||
jwk = jwk[0 : len(jwk)-1]
|
|
||||||
jwk = jwk + `, "ext": true , "key_ops": ["sign"] }`
|
|
||||||
// `{ "kty": "EC" , "crv": %q , "d": %q , "x": %q , "y": %q }`, jwk.Crv, jwk.D, jwk.X, jwk.Y
|
|
||||||
fmt.Fprintf(w, jwk)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/.well-known/openid-configuration", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
baseURL := getBaseURL(r)
|
|
||||||
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
|
||||||
fmt.Fprintf(w, `{ "issuer": "%s", "jwks_uri": "%s/.well-known/jwks.json" }`, baseURL, baseURL)
|
|
||||||
})
|
|
||||||
|
|
||||||
http.HandleFunc("/.well-known/jwks.json", func(w http.ResponseWriter, r *http.Request) {
|
|
||||||
log.Printf("%s %s %s", r.Method, r.Host, r.URL.Path)
|
|
||||||
parts := strings.Split(r.Host, ".")
|
|
||||||
kid := parts[0]
|
|
||||||
|
|
||||||
b, err := ioutil.ReadFile(filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"))
|
|
||||||
if nil != err {
|
|
||||||
//http.Error(w, "Not Found", http.StatusNotFound)
|
|
||||||
exp := strconv.FormatInt(time.Now().Add(15*time.Minute).Unix(), 10)
|
|
||||||
jwk := string(keypairs.MarshalJWKPublicKey(pubkey))
|
|
||||||
jwk = strings.Replace(jwk, `{"`, `{ "`, 1)
|
|
||||||
jwk = strings.Replace(jwk, `",`, `" ,`, -1)
|
|
||||||
jwk = jwk[0 : len(jwk)-1]
|
|
||||||
jwk = jwk + fmt.Sprintf(`, "ext": true , "key_ops": ["verify"], "exp": %s }`, exp)
|
|
||||||
// { "kty": "EC" , "crv": %q , "x": %q , "y": %q , "kid": %q , "ext": true , "key_ops": ["verify"] , "exp": %s }
|
|
||||||
jwkstr := fmt.Sprintf(`{ "keys": [ %s ] }`, jwk)
|
|
||||||
fmt.Println(jwkstr)
|
|
||||||
fmt.Fprintf(w, jwkstr)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tok := &PublicJWK{}
|
|
||||||
err = json.Unmarshal(b, tok)
|
|
||||||
if nil != err {
|
|
||||||
// TODO delete the bad file?
|
|
||||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
jwkstr := fmt.Sprintf(
|
|
||||||
`{ "keys": [ { "kty": "EC", "crv": %q, "x": %q, "y": %q, "kid": %q,`+
|
|
||||||
` "ext": true, "key_ops": ["verify"], "exp": %s } ] }`,
|
|
||||||
tok.Crv, tok.X, tok.Y, tok.KeyID, strconv.FormatInt(time.Now().Add(15*time.Minute).Unix(), 10),
|
|
||||||
)
|
|
||||||
fmt.Println(jwkstr)
|
|
||||||
fmt.Fprintf(w, jwkstr)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseExp(exp string) (int, error) {
|
|
||||||
if "" == exp {
|
|
||||||
exp = "15m"
|
|
||||||
}
|
|
||||||
mult := 1
|
|
||||||
switch exp[len(exp)-1] {
|
|
||||||
case 'w':
|
|
||||||
mult *= 7
|
|
||||||
fallthrough
|
|
||||||
case 'd':
|
|
||||||
mult *= 24
|
|
||||||
fallthrough
|
|
||||||
case 'h':
|
|
||||||
mult *= 60
|
|
||||||
fallthrough
|
|
||||||
case 'm':
|
|
||||||
mult *= 60
|
|
||||||
fallthrough
|
|
||||||
case 's':
|
|
||||||
// no fallthrough
|
|
||||||
default:
|
|
||||||
// could be 'k' or 'z', but we assume its empty
|
|
||||||
exp += "s"
|
|
||||||
}
|
|
||||||
|
|
||||||
num, err := strconv.Atoi(exp[:len(exp)-1])
|
|
||||||
if nil != err {
|
|
||||||
return 0, err
|
|
||||||
}
|
|
||||||
return num * mult, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func postEC(jwksPrefix string, tok map[string]interface{}, w http.ResponseWriter, r *http.Request) {
|
|
||||||
crv, ok := tok["crv"].(string)
|
|
||||||
if 5 != len(crv) || "P-" != crv[:2] {
|
|
||||||
http.Error(w, "Bad Request: bad curve", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
x, ok := tok["x"].(string)
|
|
||||||
if !ok {
|
|
||||||
http.Error(w, "Bad Request: missing 'x'", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
y, ok := tok["y"].(string)
|
|
||||||
if !ok {
|
|
||||||
http.Error(w, "Bad Request: missing 'y'", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
thumbprintable := []byte(
|
|
||||||
fmt.Sprintf(`{"crv":%q,"kty":"EC","x":%q,"y":%q}`, crv, x, y),
|
|
||||||
)
|
|
||||||
alg := crv[2:]
|
|
||||||
|
|
||||||
var thumb []byte
|
|
||||||
switch alg {
|
|
||||||
case "256":
|
|
||||||
hash := sha256.Sum256(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
case "384":
|
|
||||||
hash := sha512.Sum384(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
case "521":
|
|
||||||
fallthrough
|
|
||||||
case "512":
|
|
||||||
hash := sha512.Sum512(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
default:
|
|
||||||
http.Error(w, "Bad Request: bad key length or curve", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
kid := base64.RawURLEncoding.EncodeToString(thumb)
|
|
||||||
if kid2, _ := tok["kid"].(string); "" != kid2 && kid != kid2 {
|
|
||||||
http.Error(w, "Bad Request: kid should be "+kid, http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
pub := []byte(fmt.Sprintf(
|
|
||||||
`{"crv":%q,"kid":%q,"kty":"EC","x":%q,"y":%q}`, crv, kid, x, y,
|
|
||||||
))
|
|
||||||
|
|
||||||
// TODO allow posting at the top-level?
|
|
||||||
// TODO support a group of keys by PPID
|
|
||||||
// (right now it's only by KID)
|
|
||||||
if !strings.HasPrefix(r.Host, strings.ToLower(kid)+".") {
|
|
||||||
http.Error(w, "Bad Request: prefix should be "+kid, http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := ioutil.WriteFile(
|
|
||||||
filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"),
|
|
||||||
pub,
|
|
||||||
0644,
|
|
||||||
); nil != err {
|
|
||||||
fmt.Println("can't write file")
|
|
||||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
baseURL := getBaseURL(r)
|
|
||||||
w.Write([]byte(fmt.Sprintf(
|
|
||||||
`{ "iss":%q, "jwks_url":%q }`, baseURL+"/", baseURL+"/.well-known/jwks.json",
|
|
||||||
)))
|
|
||||||
}
|
|
||||||
|
|
||||||
func postRSA(jwksPrefix string, tok map[string]interface{}, w http.ResponseWriter, r *http.Request) {
|
|
||||||
e, ok := tok["e"].(string)
|
|
||||||
if !ok {
|
|
||||||
http.Error(w, "Bad Request: missing 'e'", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
n, ok := tok["n"].(string)
|
|
||||||
if !ok {
|
|
||||||
http.Error(w, "Bad Request: missing 'n'", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
thumbprintable := []byte(
|
|
||||||
fmt.Sprintf(`{"e":%q,"kty":"RSA","n":%q}`, e, n),
|
|
||||||
)
|
|
||||||
|
|
||||||
var thumb []byte
|
|
||||||
// TODO handle bit lengths well
|
|
||||||
switch 3 * (len(n) / 4.0) {
|
|
||||||
case 256:
|
|
||||||
hash := sha256.Sum256(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
case 384:
|
|
||||||
hash := sha512.Sum384(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
case 512:
|
|
||||||
hash := sha512.Sum512(thumbprintable)
|
|
||||||
thumb = hash[:]
|
|
||||||
default:
|
|
||||||
http.Error(w, "Bad Request: only standard RSA key lengths (2048, 3072, 4096) are supported", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
kid := base64.RawURLEncoding.EncodeToString(thumb)
|
|
||||||
if kid2, _ := tok["kid"].(string); "" != kid2 && kid != kid2 {
|
|
||||||
http.Error(w, "Bad Request: kid should be "+kid, http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
pub := []byte(fmt.Sprintf(
|
|
||||||
`{"e":%q,"kid":%q,"kty":"EC","n":%q}`, e, kid, n,
|
|
||||||
))
|
|
||||||
|
|
||||||
// TODO allow posting at the top-level?
|
|
||||||
// TODO support a group of keys by PPID
|
|
||||||
// (right now it's only by KID)
|
|
||||||
if !strings.HasPrefix(r.Host, strings.ToLower(kid)+".") {
|
|
||||||
http.Error(w, "Bad Request: prefix should be "+kid, http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := ioutil.WriteFile(
|
|
||||||
filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"),
|
|
||||||
pub,
|
|
||||||
0644,
|
|
||||||
); nil != err {
|
|
||||||
fmt.Println("can't write file")
|
|
||||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
baseURL := getBaseURL(r)
|
|
||||||
w.Write([]byte(fmt.Sprintf(
|
|
||||||
`{ "iss":%q, "jwks_url":%q }`, baseURL+"/", baseURL+"/.well-known/jwks.json",
|
|
||||||
)))
|
|
||||||
}
|
|
||||||
|
|
||||||
func GenToken(host string, privkey keypairs.PrivateKey, query url.Values) (string, string, string) {
|
func GenToken(host string, privkey keypairs.PrivateKey, query url.Values) (string, string, string) {
|
||||||
thumbprint := keypairs.ThumbprintPublicKey(keypairs.NewPublicKey(privkey.Public()))
|
thumbprint := keypairs.ThumbprintPublicKey(keypairs.NewPublicKey(privkey.Public()))
|
||||||
// TODO keypairs.Alg(key)
|
// TODO keypairs.Alg(key)
|
||||||
|
@ -708,60 +197,3 @@ func requireNonce(next http.HandlerFunc) http.HandlerFunc {
|
||||||
next(w, r)
|
next(w, r)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func getBaseURL(r *http.Request) string {
|
|
||||||
var scheme string
|
|
||||||
if nil != r.TLS || "https" == r.Header.Get("X-Forwarded-Proto") {
|
|
||||||
scheme = "https:"
|
|
||||||
} else {
|
|
||||||
scheme = "http:"
|
|
||||||
}
|
|
||||||
return fmt.Sprintf(
|
|
||||||
"%s//%s",
|
|
||||||
scheme,
|
|
||||||
r.Host,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// MarshalJWKPrivateKey outputs the given private key as JWK
|
|
||||||
func MarshalJWKPrivateKey(privkey keypairs.PrivateKey) []byte {
|
|
||||||
// thumbprint keys are alphabetically sorted and only include the necessary public parts
|
|
||||||
switch k := privkey.(type) {
|
|
||||||
case *rsa.PrivateKey:
|
|
||||||
return MarshalRSAPrivateKey(k)
|
|
||||||
case *ecdsa.PrivateKey:
|
|
||||||
return MarshalECPrivateKey(k)
|
|
||||||
default:
|
|
||||||
// this is unreachable because we know the types that we pass in
|
|
||||||
log.Printf("keytype: %t, %+v\n", privkey, privkey)
|
|
||||||
panic(keypairs.ErrInvalidPublicKey)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MarshalECPrivateKey will output the given private key as JWK
|
|
||||||
func MarshalECPrivateKey(k *ecdsa.PrivateKey) []byte {
|
|
||||||
crv := k.Curve.Params().Name
|
|
||||||
d := base64.RawURLEncoding.EncodeToString(k.D.Bytes())
|
|
||||||
x := base64.RawURLEncoding.EncodeToString(k.X.Bytes())
|
|
||||||
y := base64.RawURLEncoding.EncodeToString(k.Y.Bytes())
|
|
||||||
return []byte(fmt.Sprintf(
|
|
||||||
`{"crv":%q,"d":%q,"kty":"EC","x":%q,"y":%q}`,
|
|
||||||
crv, d, x, y,
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
// MarshalRSAPrivateKey will output the given private key as JWK
|
|
||||||
func MarshalRSAPrivateKey(pk *rsa.PrivateKey) []byte {
|
|
||||||
e := base64.RawURLEncoding.EncodeToString(big.NewInt(int64(pk.E)).Bytes())
|
|
||||||
n := base64.RawURLEncoding.EncodeToString(pk.N.Bytes())
|
|
||||||
d := base64.RawURLEncoding.EncodeToString(pk.D.Bytes())
|
|
||||||
p := base64.RawURLEncoding.EncodeToString(pk.Primes[0].Bytes())
|
|
||||||
q := base64.RawURLEncoding.EncodeToString(pk.Primes[1].Bytes())
|
|
||||||
dp := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Dp.Bytes())
|
|
||||||
dq := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Dq.Bytes())
|
|
||||||
qi := base64.RawURLEncoding.EncodeToString(pk.Precomputed.Qinv.Bytes())
|
|
||||||
return []byte(fmt.Sprintf(
|
|
||||||
`{"d":%q,"dp":%q,"dq":%q,"e":%q,"kty":"RSA","n":%q,"p":%q,"q":%q,"qi":%q}`,
|
|
||||||
d, dp, dq, e, n, p, q, qi,
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,12 +1,71 @@
|
||||||
package mockid
|
package mockid
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"crypto/ecdsa"
|
||||||
|
"crypto/elliptic"
|
||||||
|
"crypto/rand"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
|
||||||
//keypairs "github.com/big-squid/go-keypairs"
|
//keypairs "github.com/big-squid/go-keypairs"
|
||||||
//"github.com/big-squid/go-keypairs/keyfetch/uncached"
|
//"github.com/big-squid/go-keypairs/keyfetch/uncached"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestTest(t *testing.T) {
|
var srv *httptest.Server
|
||||||
t.Fatal("no test")
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
os.Setenv("SALT", "XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX")
|
||||||
|
jwksPrefix := "public-jwks"
|
||||||
|
err := os.MkdirAll(jwksPrefix, 0755)
|
||||||
|
if nil != err {
|
||||||
|
fmt.Fprintf(os.Stderr, "couldn't write %q: %s", jwksPrefix, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
privkey, _ := ecdsa.GenerateKey(elliptic.P256(), rand.Reader)
|
||||||
|
mux := Route(jwksPrefix, privkey)
|
||||||
|
|
||||||
|
srv = httptest.NewServer(mux)
|
||||||
|
|
||||||
|
//fs := http.FileServer(http.Dir("public"))
|
||||||
|
//http.Handle("/", fs)
|
||||||
|
|
||||||
|
os.Exit(m.Run())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTest(t *testing.T) {
|
||||||
|
client := srv.Client()
|
||||||
|
urlstr, _ := url.Parse(srv.URL + "/private.jwk.json")
|
||||||
|
//fmt.Println("URL:", srv.URL, urlstr)
|
||||||
|
res, err := client.Do(&http.Request{
|
||||||
|
Method: "POST",
|
||||||
|
URL: urlstr,
|
||||||
|
})
|
||||||
|
if nil != err {
|
||||||
|
//t.Fatal(err)
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := ioutil.ReadAll(res.Body)
|
||||||
|
if nil != err {
|
||||||
|
//t.Fatal(err)
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
jwk := map[string]string{}
|
||||||
|
err = json.Unmarshal(data, &jwk)
|
||||||
|
if nil != err {
|
||||||
|
//t.Fatal(err)
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if "" == jwk["d"] {
|
||||||
|
t.Fatal("Missing key 'd' from supposed private key")
|
||||||
|
}
|
||||||
|
//fmt.Printf("%#v\n", jwk)
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,35 @@
|
||||||
|
package mockid
|
||||||
|
|
||||||
|
import "strconv"
|
||||||
|
|
||||||
|
func parseExp(exp string) (int, error) {
|
||||||
|
if "" == exp {
|
||||||
|
exp = "15m"
|
||||||
|
}
|
||||||
|
mult := 1
|
||||||
|
switch exp[len(exp)-1] {
|
||||||
|
case 'w':
|
||||||
|
mult *= 7
|
||||||
|
fallthrough
|
||||||
|
case 'd':
|
||||||
|
mult *= 24
|
||||||
|
fallthrough
|
||||||
|
case 'h':
|
||||||
|
mult *= 60
|
||||||
|
fallthrough
|
||||||
|
case 'm':
|
||||||
|
mult *= 60
|
||||||
|
fallthrough
|
||||||
|
case 's':
|
||||||
|
// no fallthrough
|
||||||
|
default:
|
||||||
|
// could be 'k' or 'z', but we assume its empty
|
||||||
|
exp += "s"
|
||||||
|
}
|
||||||
|
|
||||||
|
num, err := strconv.Atoi(exp[:len(exp)-1])
|
||||||
|
if nil != err {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return num * mult, nil
|
||||||
|
}
|
|
@ -0,0 +1,539 @@
|
||||||
|
package mockid
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/rsa"
|
||||||
|
"crypto/sha1"
|
||||||
|
"crypto/sha256"
|
||||||
|
"crypto/sha512"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"git.rootprojects.org/root/keypairs"
|
||||||
|
"git.rootprojects.org/root/keypairs/keyfetch"
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Route(jwksPrefix string, privkey keypairs.PrivateKey) http.Handler {
|
||||||
|
Init()
|
||||||
|
|
||||||
|
// TODO get from main()
|
||||||
|
tokPrefix := jwksPrefix
|
||||||
|
pubkey := keypairs.NewPublicKey(privkey.Public())
|
||||||
|
|
||||||
|
http.HandleFunc("/api/new-nonce", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
baseURL := getBaseURL(r)
|
||||||
|
/*
|
||||||
|
res.statusCode = 200;
|
||||||
|
res.setHeader("Cache-Control", "max-age=0, no-cache, no-store");
|
||||||
|
// TODO
|
||||||
|
//res.setHeader("Date", "Sun, 10 Mar 2019 08:04:45 GMT");
|
||||||
|
// is this the expiration of the nonce itself? methinks maybe so
|
||||||
|
//res.setHeader("Expires", "Sun, 10 Mar 2019 08:04:45 GMT");
|
||||||
|
// TODO use one of the registered domains
|
||||||
|
//var indexUrl = "https://acme-staging-v02.api.letsencrypt.org/index"
|
||||||
|
*/
|
||||||
|
//var port = (state.config.ipc && state.config.ipc.port || state._ipc.port || undefined);
|
||||||
|
//var indexUrl = "http://localhost:" + port + "/index";
|
||||||
|
indexUrl := baseURL + "/index"
|
||||||
|
w.Header().Set("Link", "<"+indexUrl+">;rel=\"index\"")
|
||||||
|
w.Header().Set("Cache-Control", "max-age=0, no-cache, no-store")
|
||||||
|
w.Header().Set("Pragma", "no-cache")
|
||||||
|
//res.setHeader("Strict-Transport-Security", "max-age=604800");
|
||||||
|
|
||||||
|
w.Header().Set("X-Frame-Options", "DENY")
|
||||||
|
issueNonce(w, r)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/api/new-account", requireNonce(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// Try to decode the request body into the struct. If there is an error,
|
||||||
|
// respond to the client with the error message and a 400 status code.
|
||||||
|
data := map[string]string{}
|
||||||
|
err := json.NewDecoder(r.Body).Decode(&data)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO check DNS for MX records
|
||||||
|
parts := strings.Split(data["to"], ", <>\n\r\t")
|
||||||
|
to := parts[0]
|
||||||
|
if len(parts) > 1 || !strings.Contains(to, "@") {
|
||||||
|
http.Error(w, "invalid email address", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO expect JWK in JWS/JWT
|
||||||
|
// TODO place validated JWK into file with token
|
||||||
|
|
||||||
|
token, err := uuid.NewRandom()
|
||||||
|
if nil != err {
|
||||||
|
// nothing else to do if we run out of random
|
||||||
|
// or are on a platform that doesn't support random
|
||||||
|
panic(fmt.Errorf("random bytes read failure: %w", err))
|
||||||
|
}
|
||||||
|
token64 := base64.RawURLEncoding.EncodeToString([]byte(token[:]))
|
||||||
|
// hash token to prevent fs read timing attacks
|
||||||
|
hash := sha1.Sum(append(token[:], salt...))
|
||||||
|
tokname := base64.RawURLEncoding.EncodeToString(hash[:])
|
||||||
|
if err := ioutil.WriteFile(
|
||||||
|
filepath.Join(tokPrefix, tokname+".tok.txt"),
|
||||||
|
[]byte(`{"comment":"I have no idea..."}`),
|
||||||
|
os.FileMode(0600),
|
||||||
|
); nil != err {
|
||||||
|
http.Error(w, "database connection failed when writing verification token", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
subject := "Verify New Account"
|
||||||
|
// TODO go tpl
|
||||||
|
// TODO determine OS and Browser from user agent
|
||||||
|
baseURL := getBaseURL(r)
|
||||||
|
text := fmt.Sprintf(
|
||||||
|
"It looks like you just tried to register a new Pocket ID account.\n\n Verify account: %s/verify/%s\n\nNot you? Just ignore this message.",
|
||||||
|
baseURL, token64,
|
||||||
|
)
|
||||||
|
_, err = SendSimpleMessage(to, defaultFrom, subject, text, defaultReplyTo)
|
||||||
|
if nil != err {
|
||||||
|
// TODO neuter mailgun output
|
||||||
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Fprintf(w, `{ "success": true, "error": "" }%s`, "\n")
|
||||||
|
}))
|
||||||
|
|
||||||
|
// TODO use chi
|
||||||
|
http.HandleFunc("/verify/", requireNonce(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
parts := strings.Split(r.URL.Path, "/")
|
||||||
|
if 3 != len(parts) {
|
||||||
|
http.Error(w, "invalid url path", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
token64 := parts[2]
|
||||||
|
token, err := base64.RawURLEncoding.DecodeString(token64)
|
||||||
|
if err != nil || 0 == len(token) {
|
||||||
|
http.Error(w, "invalid url path", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// hash token to prevent fs read timing attacks
|
||||||
|
hash := sha1.Sum(append(token, salt...))
|
||||||
|
tokname := base64.RawURLEncoding.EncodeToString(hash[:])
|
||||||
|
tokfile := filepath.Join(tokPrefix, tokname+".tok.txt")
|
||||||
|
_, err = ioutil.ReadFile(tokfile)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "database connection failed when reading verification token", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO promote JWK to public... and related to an ID or email??
|
||||||
|
os.Remove(tokfile)
|
||||||
|
|
||||||
|
fmt.Fprintf(w, `{ "success": true, "error": "" }%s`, "\n")
|
||||||
|
}))
|
||||||
|
|
||||||
|
http.HandleFunc("/api/jwks", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s %s", r.Method, r.Host, r.URL.Path)
|
||||||
|
if "POST" != r.Method {
|
||||||
|
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tok := make(map[string]interface{})
|
||||||
|
decoder := json.NewDecoder(r.Body)
|
||||||
|
err := decoder.Decode(&tok)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Request: invalid json", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer r.Body.Close()
|
||||||
|
|
||||||
|
// TODO better, JSON error messages
|
||||||
|
if _, ok := tok["d"]; ok {
|
||||||
|
http.Error(w, "Bad Request: private key", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
kty, _ := tok["kty"].(string)
|
||||||
|
switch kty {
|
||||||
|
case "EC":
|
||||||
|
postEC(jwksPrefix, tok, w, r)
|
||||||
|
case "RSA":
|
||||||
|
postRSA(jwksPrefix, tok, w, r)
|
||||||
|
default:
|
||||||
|
http.Error(w, "Bad Request: only EC and RSA keys are supported", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/access_token", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
||||||
|
_, _, token := GenToken(getBaseURL(r), privkey, r.URL.Query())
|
||||||
|
fmt.Fprintf(w, token)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/private.jwk.json", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
||||||
|
if "POST" != r.Method {
|
||||||
|
http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
keylen := 2048
|
||||||
|
privkey, _ := rsa.GenerateKey(rand.Reader, keylen)
|
||||||
|
jwk := string(MarshalJWKPrivateKey(privkey))
|
||||||
|
fmt.Fprintf(w, jwk)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/inspect_token", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
token := r.Header.Get("Authorization")
|
||||||
|
log.Printf("%s %s %s\n", r.Method, r.URL.Path, token)
|
||||||
|
|
||||||
|
if "" == token {
|
||||||
|
token = r.URL.Query().Get("access_token")
|
||||||
|
if "" == token {
|
||||||
|
http.Error(w, "Bad Format: missing Authorization header and 'access_token' query", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
parts := strings.Split(token, " ")
|
||||||
|
if 2 != len(parts) {
|
||||||
|
http.Error(w, "Bad Format: expected Authorization header to be in the format of 'Bearer <Token>'", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
token = parts[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
parts := strings.Split(token, ".")
|
||||||
|
if 3 != len(parts) {
|
||||||
|
http.Error(w, "Bad Format: token should be in the format of <protected-header>.<payload>.<signature>", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
protected64 := parts[0]
|
||||||
|
payload64 := parts[1]
|
||||||
|
signature64 := parts[2]
|
||||||
|
|
||||||
|
protectedB, err := base64.RawURLEncoding.DecodeString(protected64)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: token's header should be URL-safe base64 encoded", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
payloadB, err := base64.RawURLEncoding.DecodeString(payload64)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: token's payload should be URL-safe base64 encoded", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// TODO verify signature
|
||||||
|
sig, err := base64.RawURLEncoding.DecodeString(signature64)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: token's signature should be URL-safe base64 encoded", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
errors := []string{}
|
||||||
|
|
||||||
|
protected := map[string]interface{}{}
|
||||||
|
err = json.Unmarshal(protectedB, &protected)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: token's header should be URL-safe base64-encoded JSON", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
kid, kidOK := protected["kid"].(string)
|
||||||
|
// TODO parse jwkM
|
||||||
|
_, jwkOK := protected["jwk"]
|
||||||
|
if !kidOK && !jwkOK {
|
||||||
|
errors = append(errors, "must have either header.kid or header.jwk")
|
||||||
|
}
|
||||||
|
|
||||||
|
data := map[string]interface{}{}
|
||||||
|
err = json.Unmarshal(payloadB, &data)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: token's payload should be URL-safe base64-encoded JSON", http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
iss, issOK := data["iss"].(string)
|
||||||
|
if !jwkOK && !issOK {
|
||||||
|
errors = append(errors, "payload.iss must exist to complement header.kid")
|
||||||
|
}
|
||||||
|
|
||||||
|
pub, err := keyfetch.OIDCJWK(kid, iss)
|
||||||
|
if nil != err {
|
||||||
|
fmt.Println("couldn't fetch pub key:")
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
fmt.Println("fetched pub key:")
|
||||||
|
fmt.Println(pub)
|
||||||
|
|
||||||
|
hash := sha256.Sum256([]byte(fmt.Sprintf("%s.%s", protected64, payload64)))
|
||||||
|
verified := JOSEVerify(pub, hash[:], sig)
|
||||||
|
|
||||||
|
inspected := &InspectableToken{
|
||||||
|
Public: pub,
|
||||||
|
Protected: protected,
|
||||||
|
Payload: data,
|
||||||
|
Signature: signature64,
|
||||||
|
Verified: verified,
|
||||||
|
Errors: errors,
|
||||||
|
}
|
||||||
|
|
||||||
|
tokenB, _ := json.MarshalIndent(inspected, "", " ")
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, "Bad Format: malformed token, or malformed jwk at issuer url", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprintf(w, string(tokenB))
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/authorization_header", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
||||||
|
|
||||||
|
var header string
|
||||||
|
headers, _ := r.URL.Query()["header"]
|
||||||
|
if 0 == len(headers) {
|
||||||
|
header = "Authorization"
|
||||||
|
} else {
|
||||||
|
header = headers[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
var prefix string
|
||||||
|
prefixes, _ := r.URL.Query()["prefix"]
|
||||||
|
if 0 == len(prefixes) {
|
||||||
|
prefix = "Bearer "
|
||||||
|
} else {
|
||||||
|
prefix = prefixes[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
_, _, token := GenToken(getBaseURL(r), privkey, r.URL.Query())
|
||||||
|
fmt.Fprintf(w, "%s: %s%s", header, prefix, token)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/key.jwk.json", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s", r.Method, r.URL.Path)
|
||||||
|
jwk := string(MarshalJWKPrivateKey(privkey))
|
||||||
|
jwk = strings.Replace(jwk, `{"`, `{ "`, 1)
|
||||||
|
jwk = strings.Replace(jwk, `",`, `", `, -1)
|
||||||
|
jwk = jwk[0 : len(jwk)-1]
|
||||||
|
jwk = jwk + `, "ext": true , "key_ops": ["sign"] }`
|
||||||
|
// `{ "kty": "EC" , "crv": %q , "d": %q , "x": %q , "y": %q }`, jwk.Crv, jwk.D, jwk.X, jwk.Y
|
||||||
|
fmt.Fprintf(w, jwk)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/.well-known/openid-configuration", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
baseURL := getBaseURL(r)
|
||||||
|
log.Printf("%s %s\n", r.Method, r.URL.Path)
|
||||||
|
fmt.Fprintf(w, `{ "issuer": "%s", "jwks_uri": "%s/.well-known/jwks.json" }`, baseURL, baseURL)
|
||||||
|
})
|
||||||
|
|
||||||
|
http.HandleFunc("/.well-known/jwks.json", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
log.Printf("%s %s %s", r.Method, r.Host, r.URL.Path)
|
||||||
|
parts := strings.Split(r.Host, ".")
|
||||||
|
kid := parts[0]
|
||||||
|
|
||||||
|
b, err := ioutil.ReadFile(filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"))
|
||||||
|
if nil != err {
|
||||||
|
//http.Error(w, "Not Found", http.StatusNotFound)
|
||||||
|
exp := strconv.FormatInt(time.Now().Add(15*time.Minute).Unix(), 10)
|
||||||
|
jwk := string(keypairs.MarshalJWKPublicKey(pubkey))
|
||||||
|
jwk = strings.Replace(jwk, `{"`, `{ "`, 1)
|
||||||
|
jwk = strings.Replace(jwk, `",`, `" ,`, -1)
|
||||||
|
jwk = jwk[0 : len(jwk)-1]
|
||||||
|
jwk = jwk + fmt.Sprintf(`, "ext": true , "key_ops": ["verify"], "exp": %s }`, exp)
|
||||||
|
// { "kty": "EC" , "crv": %q , "x": %q , "y": %q , "kid": %q , "ext": true , "key_ops": ["verify"] , "exp": %s }
|
||||||
|
jwkstr := fmt.Sprintf(`{ "keys": [ %s ] }`, jwk)
|
||||||
|
fmt.Println(jwkstr)
|
||||||
|
fmt.Fprintf(w, jwkstr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tok := &PublicJWK{}
|
||||||
|
err = json.Unmarshal(b, tok)
|
||||||
|
if nil != err {
|
||||||
|
// TODO delete the bad file?
|
||||||
|
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
jwkstr := fmt.Sprintf(
|
||||||
|
`{ "keys": [ { "kty": "EC", "crv": %q, "x": %q, "y": %q, "kid": %q,`+
|
||||||
|
` "ext": true, "key_ops": ["verify"], "exp": %s } ] }`,
|
||||||
|
tok.Crv, tok.X, tok.Y, tok.KeyID, strconv.FormatInt(time.Now().Add(15*time.Minute).Unix(), 10),
|
||||||
|
)
|
||||||
|
fmt.Println(jwkstr)
|
||||||
|
fmt.Fprintf(w, jwkstr)
|
||||||
|
})
|
||||||
|
|
||||||
|
return http.DefaultServeMux
|
||||||
|
}
|
||||||
|
|
||||||
|
func getBaseURL(r *http.Request) string {
|
||||||
|
var scheme string
|
||||||
|
if nil != r.TLS || "https" == r.Header.Get("X-Forwarded-Proto") {
|
||||||
|
scheme = "https:"
|
||||||
|
} else {
|
||||||
|
scheme = "http:"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(
|
||||||
|
"%s//%s",
|
||||||
|
scheme,
|
||||||
|
r.Host,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
type HTTPError struct {
|
||||||
|
message string
|
||||||
|
code int
|
||||||
|
}
|
||||||
|
|
||||||
|
func postEC(jwksPrefix string, tok map[string]interface{}, w http.ResponseWriter, r *http.Request) {
|
||||||
|
data, err := postECHelper(r.Host, jwksPrefix, tok, r)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, err.message, err.code)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.Write(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func postECHelper(hostname, jwksPrefix string, tok map[string]interface{}, r *http.Request) ([]byte, *HTTPError) {
|
||||||
|
crv, ok := tok["crv"].(string)
|
||||||
|
if 5 != len(crv) || "P-" != crv[:2] {
|
||||||
|
return nil, &HTTPError{"Bad Request: bad curve", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
x, ok := tok["x"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, &HTTPError{"Bad Request: missing 'x'", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
y, ok := tok["y"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, &HTTPError{"Bad Request: missing 'y'", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
thumbprintable := []byte(
|
||||||
|
fmt.Sprintf(`{"crv":%q,"kty":"EC","x":%q,"y":%q}`, crv, x, y),
|
||||||
|
)
|
||||||
|
alg := crv[2:]
|
||||||
|
|
||||||
|
var thumb []byte
|
||||||
|
switch alg {
|
||||||
|
case "256":
|
||||||
|
hash := sha256.Sum256(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
case "384":
|
||||||
|
hash := sha512.Sum384(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
case "521":
|
||||||
|
fallthrough
|
||||||
|
case "512":
|
||||||
|
hash := sha512.Sum512(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
default:
|
||||||
|
return nil, &HTTPError{"Bad Request: bad key length or curve", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
kid := base64.RawURLEncoding.EncodeToString(thumb)
|
||||||
|
if kid2, _ := tok["kid"].(string); "" != kid2 && kid != kid2 {
|
||||||
|
return nil, &HTTPError{"Bad Request: kid should be " + kid, http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub := []byte(fmt.Sprintf(
|
||||||
|
`{"crv":%q,"kid":%q,"kty":"EC","x":%q,"y":%q}`, crv, kid, x, y,
|
||||||
|
))
|
||||||
|
|
||||||
|
// TODO allow posting at the top-level?
|
||||||
|
// TODO support a group of keys by PPID
|
||||||
|
// (right now it's only by KID)
|
||||||
|
if !strings.HasPrefix(hostname, strings.ToLower(kid)+".") {
|
||||||
|
return nil, &HTTPError{"Bad Request: prefix should be " + kid, http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := ioutil.WriteFile(
|
||||||
|
filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"),
|
||||||
|
pub,
|
||||||
|
0644,
|
||||||
|
); nil != err {
|
||||||
|
fmt.Println("can't write file")
|
||||||
|
return nil, &HTTPError{"Internal Server Error", http.StatusInternalServerError}
|
||||||
|
}
|
||||||
|
|
||||||
|
baseURL := getBaseURL(r)
|
||||||
|
return []byte(fmt.Sprintf(
|
||||||
|
`{ "iss":%q, "jwks_url":%q }`, baseURL+"/", baseURL+"/.well-known/jwks.json",
|
||||||
|
)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func postRSA(jwksPrefix string, tok map[string]interface{}, w http.ResponseWriter, r *http.Request) {
|
||||||
|
data, err := postRSAHelper(r.Host, jwksPrefix, tok, r)
|
||||||
|
if nil != err {
|
||||||
|
http.Error(w, err.message, err.code)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.Write(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func postRSAHelper(hostname, jwksPrefix string, tok map[string]interface{}, r *http.Request) ([]byte, *HTTPError) {
|
||||||
|
e, ok := tok["e"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, &HTTPError{"Bad Request: missing 'e'", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
n, ok := tok["n"].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, &HTTPError{"Bad Request: missing 'n'", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
thumbprintable := []byte(
|
||||||
|
fmt.Sprintf(`{"e":%q,"kty":"RSA","n":%q}`, e, n),
|
||||||
|
)
|
||||||
|
|
||||||
|
var thumb []byte
|
||||||
|
// TODO handle bit lengths well
|
||||||
|
switch 3 * (len(n) / 4.0) {
|
||||||
|
case 256:
|
||||||
|
hash := sha256.Sum256(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
case 384:
|
||||||
|
hash := sha512.Sum384(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
case 512:
|
||||||
|
hash := sha512.Sum512(thumbprintable)
|
||||||
|
thumb = hash[:]
|
||||||
|
default:
|
||||||
|
return nil, &HTTPError{"Bad Request: only standard RSA key lengths (2048, 3072, 4096) are supported", http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
kid := base64.RawURLEncoding.EncodeToString(thumb)
|
||||||
|
if kid2, _ := tok["kid"].(string); "" != kid2 && kid != kid2 {
|
||||||
|
return nil, &HTTPError{"Bad Request: kid should be " + kid, http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub := []byte(fmt.Sprintf(
|
||||||
|
`{"e":%q,"kid":%q,"kty":"EC","n":%q}`, e, kid, n,
|
||||||
|
))
|
||||||
|
|
||||||
|
// TODO allow posting at the top-level?
|
||||||
|
// TODO support a group of keys by PPID
|
||||||
|
// (right now it's only by KID)
|
||||||
|
if !strings.HasPrefix(hostname, strings.ToLower(kid)+".") {
|
||||||
|
return nil, &HTTPError{"Bad Request: prefix should be " + kid, http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := ioutil.WriteFile(
|
||||||
|
filepath.Join(jwksPrefix, strings.ToLower(kid)+".jwk.json"),
|
||||||
|
pub,
|
||||||
|
0644,
|
||||||
|
); nil != err {
|
||||||
|
fmt.Println("can't write file")
|
||||||
|
return nil, &HTTPError{"Internal Server Error", http.StatusInternalServerError}
|
||||||
|
}
|
||||||
|
|
||||||
|
baseURL := getBaseURL(r)
|
||||||
|
return []byte(fmt.Sprintf(
|
||||||
|
`{ "iss":%q, "jwks_url":%q }`, baseURL+"/", baseURL+"/.well-known/jwks.json",
|
||||||
|
)), nil
|
||||||
|
}
|
Loading…
Reference in New Issue