mesh.cat/meshname
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add ip=true optional parameter

Browse Source
This commit is contained in:
cynic 2024-08-08 14:54:19 +00:00
parent 46b3388b83
commit 507b16fbd0
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
cmd/whoami/main.go
View File

@ -6,6 +6,7 @@ import (
"encoding/base32" "encoding/base32"
"strings" "strings"
"flag" "flag"
"log"
) )
// define tld // define tld
@ -14,12 +15,17 @@ var tld = flag.String("tld", "mesh.cat", "The top level domain for our network")
// domainFromIP derives a meshname subdomain for the authoritative DNS server address // domainFromIP derives a meshname subdomain for the authoritative DNS server address
func domainFromIP(target net.IP) string { func domainFromIP(target net.IP) string {
host := strings.ToLower(base32.StdEncoding.EncodeToString(target)[0:26]) host := strings.ToLower(base32.StdEncoding.EncodeToString(target)[0:26])
domain := strings.Join([]string{host, ".", *tld, "\n"}, "") domain := host + "." + *tld
return domain return domain
} }
// handle HTTP requests // handle HTTP requests
func handler(w http.ResponseWriter, r *http.Request) { func handler(w http.ResponseWriter, r *http.Request) {
// parse query parameters
params := r.URL.Query()
ipParam := params.Get("ip")
// check if the "ip" query parameter is set to "true"
showIP := ipParam == "true"
// get client's ip address // get client's ip address
ip, port, err := net.SplitHostPort(r.RemoteAddr) ip, port, err := net.SplitHostPort(r.RemoteAddr)
_, _ = port, err _, _ = port, err
@ -31,7 +37,13 @@ func handler(w http.ResponseWriter, r *http.Request) {
parsedIP := net.ParseIP(ip) parsedIP := net.ParseIP(ip)
// return domain for IPv6 only // return domain for IPv6 only
if parsedIP.To4() == nil { if parsedIP.To4() == nil {
w.Write([]byte(domainFromIP(parsedIP))) if (showIP) {
w.Write([]byte(ip + "\n"))
} else {
w.Write([]byte(domainFromIP(parsedIP) + "\n"))
}
log.Println(showIP)
log.Println(ip)
} else { } else {
// set the response status code to 422 // set the response status code to 422
w.WriteHeader(http.StatusUnprocessableEntity) w.WriteHeader(http.StatusUnprocessableEntity)
@ -49,7 +61,7 @@ func main() {
// parse the command-line arguments // parse the command-line arguments
flag.Parse() flag.Parse()
// construct listening address // construct listening address
listenAddr := strings.Join([]string{*address, ":", *port}, "") listenAddr := *address + ":" + *port
print("Listening on address: ", listenAddr, "\n") print("Listening on address: ", listenAddr, "\n")
// start server // start server
http.ListenAndServe(listenAddr, nil) http.ListenAndServe(listenAddr, nil)