aboutsummaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
authoradnano <[email protected]>2020-09-27 20:20:59 -0400
committeradnano <[email protected]>2020-09-27 20:20:59 -0400
commit38fe1f21dc8d356157845bf7b930291d54879def (patch)
tree45024a1f6f5e95952a30f7111bc8e40d9208ebe7 /examples
parentAdd helper handler functions (diff)
downloadgo-gemini-38fe1f21dc8d356157845bf7b930291d54879def.tar.xz
go-gemini-38fe1f21dc8d356157845bf7b930291d54879def.zip
Change package name to gmi
Diffstat (limited to 'examples')
-rw-r--r--examples/auth/auth.go62
-rw-r--r--examples/client/client.go30
-rw-r--r--examples/server/server.go6
3 files changed, 47 insertions, 51 deletions
diff --git a/examples/auth/auth.go b/examples/auth/auth.go
index 0c99535..6abd0c8 100644
--- a/examples/auth/auth.go
+++ b/examples/auth/auth.go
@@ -39,12 +39,12 @@ func main() {
//
// go run -tags=example ../cert
//
- cert, err := tls.LoadX509KeyPair("examples/client/localhost.crt", "examples/client/localhost.key")
+ cert, err := tls.LoadX509KeyPair("examples/server/localhost.crt", "examples/server/localhost.key")
if err != nil {
log.Fatal(err)
}
- handler := &gemini.ServeMux{}
+ handler := &gmi.ServeMux{}
handler.HandleFunc("", welcome)
handler.HandleFunc("/login", login)
handler.HandleFunc("/login/password", loginPassword)
@@ -52,7 +52,7 @@ func main() {
handler.HandleFunc("/admin", admin)
handler.HandleFunc("/logout", logout)
- server := &gemini.Server{
+ server := &gmi.Server{
Certificate: cert,
Handler: handler,
}
@@ -63,102 +63,98 @@ func main() {
}
func getSession(crt *x509.Certificate) (*session, bool) {
- fingerprint := gemini.Fingerprint(crt)
+ fingerprint := gmi.Fingerprint(crt)
session, ok := sessions[fingerprint]
return session, ok
}
-func welcome(rw *gemini.ResponseWriter, req *gemini.Request) {
- rw.WriteHeader(gemini.StatusSuccess, "text/gemini")
+func welcome(rw *gmi.ResponseWriter, req *gmi.Request) {
+ rw.WriteHeader(gmi.StatusSuccess, "text/gemini")
rw.Write([]byte("Welcome to this example.\n=> /login Login\n"))
}
-func login(rw *gemini.ResponseWriter, req *gemini.Request) {
+func login(rw *gmi.ResponseWriter, req *gmi.Request) {
if len(req.TLS.PeerCertificates) > 0 {
if username := req.URL.RawQuery; username == "" {
- rw.WriteHeader(gemini.StatusInput, "Username")
+ rw.WriteHeader(gmi.StatusInput, "Username")
} else {
- fingerprint := gemini.Fingerprint(req.TLS.PeerCertificates[0])
+ fingerprint := gmi.Fingerprint(req.TLS.PeerCertificates[0])
sessions[fingerprint] = &session{
username: username,
}
// TODO: Remove scheme and host once example client can handle relative redirects
- rw.WriteHeader(gemini.StatusRedirectTemporary, "gemini://localhost/login/password")
+ rw.WriteHeader(gmi.StatusRedirect, "gemini://localhost/login/password")
}
} else {
- rw.WriteHeader(gemini.StatusClientCertificateRequired, "Certificate required")
+ rw.WriteHeader(gmi.StatusClientCertificateRequired, "Certificate required")
}
}
-func loginPassword(rw *gemini.ResponseWriter, req *gemini.Request) {
+func loginPassword(rw *gmi.ResponseWriter, req *gmi.Request) {
if len(req.TLS.PeerCertificates) > 0 {
session, ok := getSession(req.TLS.PeerCertificates[0])
if !ok {
- rw.WriteHeader(gemini.StatusCertificateNotAuthorised, "Not authorized")
+ rw.WriteHeader(gmi.StatusCertificateNotAuthorised, "Not authorized")
return
}
if password := req.URL.RawQuery; password == "" {
- rw.WriteHeader(gemini.StatusInput, "Password")
+ rw.WriteHeader(gmi.StatusInput, "Password")
} else {
expected := logins[session.username].password
if password == expected {
// TODO: Remove scheme and host once example client can handle relative redirects
session.authorized = true
- rw.WriteHeader(gemini.StatusRedirectTemporary, "gemini://localhost/profile")
+ rw.WriteHeader(gmi.StatusRedirect, "gemini://localhost/profile")
} else {
- rw.WriteHeader(gemini.StatusInput, "Wrong password. Please try again")
+ rw.WriteHeader(gmi.StatusInput, "Wrong password. Please try again")
}
}
} else {
- rw.WriteHeader(gemini.StatusClientCertificateRequired, "Certificate required")
+ rw.WriteHeader(gmi.StatusClientCertificateRequired, "Certificate required")
}
}
-func logout(rw *gemini.ResponseWriter, req *gemini.Request) {
+func logout(rw *gmi.ResponseWriter, req *gmi.Request) {
if len(req.TLS.PeerCertificates) > 0 {
- fingerprint := gemini.Fingerprint(req.TLS.PeerCertificates[0])
+ fingerprint := gmi.Fingerprint(req.TLS.PeerCertificates[0])
delete(sessions, fingerprint)
}
- rw.WriteHeader(gemini.StatusSuccess, "text/gemini")
+ rw.WriteHeader(gmi.StatusSuccess, "text/gemini")
rw.Write([]byte("Successfully logged out.\n"))
}
-func badLogin(rw *gemini.ResponseWriter, req *gemini.Request) {
-
-}
-
-func profile(rw *gemini.ResponseWriter, req *gemini.Request) {
+func profile(rw *gmi.ResponseWriter, req *gmi.Request) {
if len(req.TLS.PeerCertificates) > 0 {
session, ok := getSession(req.TLS.PeerCertificates[0])
if !ok {
- rw.WriteHeader(gemini.StatusCertificateNotAuthorised, "Certificate not authorized")
+ rw.WriteHeader(gmi.StatusCertificateNotAuthorised, "Certificate not authorized")
return
}
user := logins[session.username]
profile := fmt.Sprintf("Username: %s\nAdmin: %t\n=> /logout Logout", session.username, user.admin)
- rw.WriteHeader(gemini.StatusSuccess, "text/gemini")
+ rw.WriteHeader(gmi.StatusSuccess, "text/gemini")
rw.Write([]byte(profile))
} else {
- rw.WriteHeader(gemini.StatusClientCertificateRequired, "Certificate required")
+ rw.WriteHeader(gmi.StatusClientCertificateRequired, "Certificate required")
}
}
-func admin(rw *gemini.ResponseWriter, req *gemini.Request) {
+func admin(rw *gmi.ResponseWriter, req *gmi.Request) {
if len(req.TLS.PeerCertificates) > 0 {
session, ok := getSession(req.TLS.PeerCertificates[0])
if !ok {
- rw.WriteHeader(gemini.StatusCertificateNotAuthorised, "Certificate not authorized")
+ rw.WriteHeader(gmi.StatusCertificateNotAuthorised, "Certificate not authorized")
return
}
user := logins[session.username]
if !user.admin {
- rw.WriteHeader(gemini.StatusCertificateNotAuthorised, "Admins only!")
+ rw.WriteHeader(gmi.StatusCertificateNotAuthorised, "Admins only!")
return
}
- rw.WriteHeader(gemini.StatusSuccess, "text/gemini")
+ rw.WriteHeader(gmi.StatusSuccess, "text/gemini")
rw.Write([]byte("Welcome to the admin portal.\n"))
} else {
- rw.WriteHeader(gemini.StatusClientCertificateRequired, "Certificate required")
+ rw.WriteHeader(gmi.StatusClientCertificateRequired, "Certificate required")
}
}
diff --git a/examples/client/client.go b/examples/client/client.go
index 4c924a6..b209d0e 100644
--- a/examples/client/client.go
+++ b/examples/client/client.go
@@ -13,22 +13,22 @@ import (
var (
scanner = bufio.NewScanner(os.Stdin)
- client *gemini.Client
+ client *gmi.Client
)
func init() {
// Initialize the client
- client = &gemini.Client{}
+ client = &gmi.Client{}
client.KnownHosts.Load() // Load known hosts
- client.TrustCertificate = func(hostname string, cert *x509.Certificate, knownHosts *gemini.KnownHosts) error {
+ client.TrustCertificate = func(hostname string, cert *x509.Certificate, knownHosts *gmi.KnownHosts) error {
err := knownHosts.Lookup(hostname, cert)
if err != nil {
switch err {
- case gemini.ErrCertificateNotTrusted:
+ case gmi.ErrCertificateNotTrusted:
// Alert the user that the certificate is not trusted
fmt.Printf("Warning: Certificate for %s is not trusted!\n", hostname)
fmt.Println("This could indicate a Man-in-the-Middle attack.")
- case gemini.ErrUnknownCertificate:
+ case gmi.ErrUnknownCertificate:
// Prompt the user to trust the certificate
trust := trustCertificate(cert)
switch trust {
@@ -48,33 +48,33 @@ func init() {
}
// sendRequest sends a request to the given url.
-func sendRequest(req *gemini.Request) error {
+func sendRequest(req *gmi.Request) error {
resp, err := client.Send(req)
if err != nil {
return err
}
switch resp.Status / 10 {
- case gemini.StatusClassInput:
+ case gmi.StatusClassInput:
fmt.Printf("%s: ", resp.Meta)
scanner.Scan()
req.URL.RawQuery = scanner.Text()
return sendRequest(req)
- case gemini.StatusClassSuccess:
+ case gmi.StatusClassSuccess:
fmt.Print(string(resp.Body))
return nil
- case gemini.StatusClassRedirect:
+ case gmi.StatusClassRedirect:
fmt.Println("Redirecting to ", resp.Meta)
- req, err := gemini.NewRequest(resp.Meta)
+ req, err := gmi.NewRequest(resp.Meta)
if err != nil {
return err
}
return sendRequest(req)
- case gemini.StatusClassTemporaryFailure:
+ case gmi.StatusClassTemporaryFailure:
return fmt.Errorf("Temporary failure: %s", resp.Meta)
- case gemini.StatusClassPermanentFailure:
+ case gmi.StatusClassPermanentFailure:
return fmt.Errorf("Permanent failure: %s", resp.Meta)
- case gemini.StatusClassClientCertificateRequired:
+ case gmi.StatusClassClientCertificateRequired:
fmt.Println("Generating client certificate for", req.Hostname())
return nil // TODO: Generate and store client certificate
}
@@ -99,7 +99,7 @@ Otherwise, this should be safe to trust.
=> `
func trustCertificate(cert *x509.Certificate) trust {
- fmt.Printf(trustPrompt, gemini.Fingerprint(cert))
+ fmt.Printf(trustPrompt, gmi.Fingerprint(cert))
scanner.Scan()
switch scanner.Text() {
case "t":
@@ -118,7 +118,7 @@ func main() {
}
url := os.Args[1]
- req, err := gemini.NewRequest(url)
+ req, err := gmi.NewRequest(url)
if err != nil {
fmt.Println(err)
os.Exit(1)
diff --git a/examples/server/server.go b/examples/server/server.go
index e761cd0..09abf50 100644
--- a/examples/server/server.go
+++ b/examples/server/server.go
@@ -20,10 +20,10 @@ func main() {
log.Fatal(err)
}
- mux := &gemini.ServeMux{}
- mux.Handle("/", gemini.FileServer(gemini.Dir("/var/www")))
+ mux := &gmi.ServeMux{}
+ mux.Handle("/", gmi.FileServer(gmi.Dir("/var/www")))
- server := gemini.Server{
+ server := gmi.Server{
Handler: mux,
Certificate: cert,
}