Update examples/auth.go

This commit is contained in:
Adnan Maolood 2020-11-05 15:37:46 -05:00
parent b76080c863
commit 8101fbe473

View File

@ -64,8 +64,8 @@ func main() {
} }
func getSession(cert *x509.Certificate) (*session, bool) { func getSession(cert *x509.Certificate) (*session, bool) {
fingerprint := gemini.Fingerprint(cert) fingerprint := gemini.NewFingerprint(cert)
session, ok := sessions[fingerprint] session, ok := sessions[fingerprint.Hex]
return session, ok return session, ok
} }
@ -79,8 +79,8 @@ func login(w *gemini.ResponseWriter, r *gemini.Request) {
w.WriteHeader(gemini.StatusInput, "Username") w.WriteHeader(gemini.StatusInput, "Username")
return return
} }
fingerprint := gemini.Fingerprint(r.Certificate.Leaf) fingerprint := gemini.NewFingerprint(r.Certificate.Leaf)
sessions[fingerprint] = &session{ sessions[fingerprint.Hex] = &session{
username: username, username: username,
} }
w.WriteHeader(gemini.StatusRedirect, "/password") w.WriteHeader(gemini.StatusRedirect, "/password")
@ -116,8 +116,8 @@ func logout(w *gemini.ResponseWriter, r *gemini.Request) {
w.WriteStatus(gemini.StatusCertificateRequired) w.WriteStatus(gemini.StatusCertificateRequired)
return return
} }
fingerprint := gemini.Fingerprint(r.Certificate.Leaf) fingerprint := gemini.NewFingerprint(r.Certificate.Leaf)
delete(sessions, fingerprint) delete(sessions, fingerprint.Hex)
fmt.Fprintln(w, "Successfully logged out.") fmt.Fprintln(w, "Successfully logged out.")
fmt.Fprintln(w, "=> / Index") fmt.Fprintln(w, "=> / Index")
} }