-
Notifications
You must be signed in to change notification settings - Fork 11
/
Copy pathhi.go
48 lines (45 loc) · 1.03 KB
/
hi.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package main
import (
"log"
"net"
"net/http"
"os"
"strconv"
)
func main() {
//Set listening port
port := 9898
if len(os.Args) > 1 {
var err error
port, err = strconv.Atoi(os.Args[1])
if err != nil {
log.Fatal(err)
}
}
//Find default output IP and hostname
conn, err := net.Dial("udp", "8.8.8.8:80")
if err != nil {
log.Fatal(err)
}
defer conn.Close()
localAddr := conn.LocalAddr().(*net.UDPAddr)
hostname, err := os.Hostname()
if err != nil {
log.Fatal(err)
}
//Register handler
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
ClientAddress := r.Header.Get("X-Real-Ip")
if ClientAddress == "" {
ClientAddress = r.Header.Get("X-Forwarded-For")
}
if ClientAddress == "" {
ClientAddress = r.RemoteAddr
}
log.Println("Request from " + ClientAddress)
w.Write([]byte(hostname + " " + localAddr.IP.String() + "\n"))
})
//Serve
log.Println(hostname + " " + localAddr.IP.String() + " " + strconv.Itoa(port))
log.Fatal(http.ListenAndServe("0.0.0.0:"+strconv.Itoa(port), nil))
}