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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
package server
import (
"strconv"
"fmt"
)
const MAXPORT = 1 << 16 - 1
type tcpListen struct {
addr string
}
type tcpOut struct {
addr string
}
func (t *tcpListen) Open() (proto, error) {
return nil, errNotImplemented
}
func (t *tcpListen) Close() error {
return nil
}
func (t *tcpListen) String() string {
return fmt.Sprintf("tcp <- %s", t.addr)
}
func (t *tcpOut) Open() (proto, error) {
return nil, errNotImplemented
}
func (t *tcpOut) Close() error {
return nil
}
func (t *tcpOut) String() string {
return fmt.Sprintf("tcp -> %s", t.addr)
}
func parsePort(r *request, s string) int {
port, err := strconv.Atoi(s)
if err != nil {
r.Fatalf("bad port value: %v", err)
}
if port <= 0 || port > MAXPORT {
r.Fatal("bad port value")
}
return port
}
func addTcpListen(r *request) {
r.expect(1, 2)
var host string
var n int
if r.argc > 1 {
host = r.args[n]
n++
}
port := parsePort(r, r.args[n])
t := &tcpListen{addr: fmt.Sprintf("%s:%d", host, port)}
r.c.s.proto.add(t)
}
func addTcpOut(r *request) {
r.expect(2)
host := r.args[0]
port := parsePort(r, r.args[1])
t := &tcpOut{addr: fmt.Sprintf("%s:%d", host, port)}
r.c.s.proto.add(t)
}
func init() {
newCmd(addTcpListen, "add", "tcp", "listen")
newCmd(addTcpOut, "add", "tcp", "out")
}
|
