tools.go_stun

Yohn Y. 2020-04-06 Parent:4b125c1ef311 Child:102fbdab34d8

2:6455d3067db1 Go to Latest

tools.go_stun/main.go

+ Добавил разрешение имён. Просто так

Download raw file
View source Diff to previous Annotate
History
1 package main
2
3 import (
4 "errors"
5 "fmt"
6 "log/syslog"
7 "net"
8 "os"
9 "strings"
10 "time"
11 )
12
13 const (
14 LOG_IDENT string = "net-a0fs-gostun"
15 )
16
17 func getPort() (res string, err error) {
18 if len(os.Args) < 2 {
19 err = errors.New("No port specify")
20 } else {
21 res = fmt.Sprintf(":%s", os.Args[1])
22 }
23 return
24 }
25
26 func fail(code int, msg string) {
27 usage := "Usage: cmd port"
28 usage += "\n port: port number for listening clients"
29
30 fmt.Fprintln(os.Stderr, msg)
31 fmt.Fprintln(os.Stderr, usage)
32 os.Exit(code)
33 return
34 }
35
36 func handler(conn net.Conn) {
37 log, err := syslog.New(syslog.LOG_USER, LOG_IDENT)
38 if err != nil {
39 panic(err)
40 }
41
42 _s_tm := time.Now()
43 remoteAddr := conn.RemoteAddr().String()
44
45 defer func() {
46 var err error
47 _s_tmd := time.Now().Sub(_s_tm)
48 if r_err := recover(); r_err != nil {
49 err = log.Err(fmt.Sprintf("Error on service for %s (service time: %v): %v", remoteAddr, _s_tmd, r_err))
50 if err != nil {
51 panic(err.Error())
52 }
53 } else {
54 err = conn.Close()
55 if err != nil {
56 err = log.Err(fmt.Sprintf("Error on closing socket when service %s (service time: %v): %v",
57 remoteAddr, _s_tmd, err))
58
59 if err != nil {
60 panic(err.Error())
61 }
62 } else {
63 _ = log.Info(fmt.Sprintf("Ok service %s in %v", remoteAddr, _s_tmd))
64 }
65 }
66 }()
67
68 addr_spit := strings.Split(remoteAddr, ":")
69 addr_last := len(addr_spit) - 1
70 addr := strings.Join(addr_spit[:addr_last], ":")
71
72 var dns string
73 dns_names, err := net.LookupAddr(strings.Trim(addr, "[] "))
74 if err != nil {
75 log.Warning(fmt.Sprintf("Fail to resolve dns name for %s: %v", remoteAddr, err))
76 } else {
77 for _, dns_name := range dns_names {
78 dns += "Name: " + dns_name + "\n"
79 }
80 }
81 addr = "IP: " + addr + "\n"
82 port := "Port: " + addr_spit[addr_last] + "\n"
83
84 if _, err := conn.Write([]byte(addr)); err != nil {
85 panic(err.Error())
86 }
87 if _, err := conn.Write([]byte(dns)); err != nil {
88 panic(err.Error())
89 }
90 if _, err := conn.Write([]byte(port)); err != nil {
91 panic(err.Error())
92 }
93 }
94
95 func main() {
96 port, err := getPort()
97 if err != nil {
98 fail(1, fmt.Sprint("Error in parsing port number: ", err))
99 }
100
101 svc, err := net.Listen("tcp", port)
102 if err != nil {
103 fail(2, fmt.Sprint("Fail to open socket: ", err))
104 }
105
106 for {
107 conn, err := svc.Accept()
108 if err != nil {
109 fail(1, fmt.Sprint("Error accepting connections: ", err))
110 }
111 go handler(conn)
112 }
113 }