2019-04-21 08:34:25 +08:00
|
|
|
package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"crypto/hmac"
|
|
|
|
|
"crypto/sha256"
|
|
|
|
|
"encoding/hex"
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"fmt"
|
|
|
|
|
"net/http"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func ssoCallbackHandler(w http.ResponseWriter, r *http.Request) {
|
|
|
|
|
payloadHex := r.FormValue("payload")
|
|
|
|
|
signature := r.FormValue("hmac")
|
|
|
|
|
|
|
|
|
|
payloadBytes, err := hex.DecodeString(payloadHex)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Fprintf(w, "Error: invalid JSON payload hex encoding: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
signatureBytes, err := hex.DecodeString(signature)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Fprintf(w, "Error: invalid HMAC signature hex encoding: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
payload := ssoPayload{}
|
|
|
|
|
err = json.Unmarshal(payloadBytes, &payload)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Fprintf(w, "Error: cannot unmarshal JSON payload: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 11:25:35 +08:00
|
|
|
if payload.Token == "" || payload.Email == "" || payload.Name == "" {
|
2019-04-21 08:34:25 +08:00
|
|
|
fmt.Fprintf(w, "Error: %s\n", errorMissingField.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if payload.Link == "" {
|
|
|
|
|
payload.Link = "undefined"
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if payload.Photo == "" {
|
|
|
|
|
payload.Photo = "undefined"
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 11:25:35 +08:00
|
|
|
domain, commenterToken, err := ssoTokenExtract(payload.Token)
|
|
|
|
|
if err != nil {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
d, err := domainGet(domain)
|
2019-04-21 08:34:25 +08:00
|
|
|
if err != nil {
|
|
|
|
|
if err == errorNoSuchDomain {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
} else {
|
|
|
|
|
logger.Errorf("cannot get domain for SSO: %v", err)
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", errorInternal.Error())
|
|
|
|
|
}
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if d.SsoSecret == "" || d.SsoUrl == "" {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", errorMissingConfig.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
key, err := hex.DecodeString(d.SsoSecret)
|
|
|
|
|
if err != nil {
|
|
|
|
|
logger.Errorf("cannot decode SSO secret as hex: %v", err)
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
h := hmac.New(sha256.New, key)
|
|
|
|
|
h.Write(payloadBytes)
|
|
|
|
|
expectedSignatureBytes := h.Sum(nil)
|
|
|
|
|
if !hmac.Equal(expectedSignatureBytes, signatureBytes) {
|
|
|
|
|
fmt.Fprintf(w, "Error: HMAC signature verification failed\n")
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 11:25:35 +08:00
|
|
|
_, err = commenterGetByCommenterToken(commenterToken)
|
2019-04-21 08:34:25 +08:00
|
|
|
if err != nil && err != errorNoSuchToken {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 11:25:35 +08:00
|
|
|
c, err := commenterGetByEmail("sso:"+domain, payload.Email)
|
2019-04-21 08:34:25 +08:00
|
|
|
if err != nil && err != errorNoSuchCommenter {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var commenterHex string
|
|
|
|
|
|
|
|
|
|
if err == errorNoSuchCommenter {
|
2019-04-21 11:25:35 +08:00
|
|
|
commenterHex, err = commenterNew(payload.Email, payload.Name, payload.Link, payload.Photo, "sso:"+domain, "")
|
2019-04-21 08:34:25 +08:00
|
|
|
if err != nil {
|
|
|
|
|
fmt.Fprintf(w, "Error: %s", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
} else {
|
2019-08-12 06:52:56 +08:00
|
|
|
if err = commenterUpdate(c.CommenterHex, payload.Email, payload.Name, payload.Link, payload.Photo, "sso:"+domain); err != nil {
|
|
|
|
|
logger.Warningf("cannot update commenter: %s", err)
|
|
|
|
|
// not a serious enough to exit with an error
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 08:34:25 +08:00
|
|
|
commenterHex = c.CommenterHex
|
|
|
|
|
}
|
|
|
|
|
|
2019-04-21 11:25:35 +08:00
|
|
|
if err = commenterSessionUpdate(commenterToken, commenterHex); err != nil {
|
2019-04-21 08:34:25 +08:00
|
|
|
fmt.Fprintf(w, "Error: %s\n", err.Error())
|
|
|
|
|
return
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fmt.Fprintf(w, "<html><script>window.parent.close()</script></html>")
|
|
|
|
|
}
|
