Update examples/auth.go

This commit is contained in:
Adnan Maolood 2021-02-16 11:26:09 -05:00
parent 779be8b95b
commit 7c1a5184c9

View File

@ -55,11 +55,11 @@ func fingerprint(cert *x509.Certificate) string {
} }
func profile(w gemini.ResponseWriter, r *gemini.Request) { func profile(w gemini.ResponseWriter, r *gemini.Request) {
if r.Certificate == nil { if len(r.TLS.PeerCertificates) == 0 {
w.Status(gemini.StatusCertificateRequired) w.Status(gemini.StatusCertificateRequired)
return return
} }
fingerprint := fingerprint(r.Certificate.Leaf) fingerprint := fingerprint(r.TLS.PeerCertificates[0])
user, ok := users[fingerprint] user, ok := users[fingerprint]
if !ok { if !ok {
user = &User{} user = &User{}
@ -70,7 +70,7 @@ func profile(w gemini.ResponseWriter, r *gemini.Request) {
} }
func changeUsername(w gemini.ResponseWriter, r *gemini.Request) { func changeUsername(w gemini.ResponseWriter, r *gemini.Request) {
if r.Certificate == nil { if len(r.TLS.PeerCertificates) == 0 {
w.Status(gemini.StatusCertificateRequired) w.Status(gemini.StatusCertificateRequired)
return return
} }
@ -80,7 +80,7 @@ func changeUsername(w gemini.ResponseWriter, r *gemini.Request) {
w.Header(gemini.StatusInput, "Username") w.Header(gemini.StatusInput, "Username")
return return
} }
fingerprint := fingerprint(r.Certificate.Leaf) fingerprint := fingerprint(r.TLS.PeerCertificates[0])
user, ok := users[fingerprint] user, ok := users[fingerprint]
if !ok { if !ok {
user = &User{} user = &User{}