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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
// Copyright 2024 Konrad Zemek <konrad.zemek@gmail.com>
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package tests
import (
"net/netip"
"reflect"
"testing"
"github.com/kzemek/go-mmproxy/proxyprotocol"
"github.com/kzemek/go-mmproxy/utils"
)
func TestProxyProtocolV1(t *testing.T) {
buf := []byte("PROXY TCP4 192.168.0.1 192.168.0.11 56324 443\r\nmoredata")
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.TCP)
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
if saddr.String() != "192.168.0.1:56324" {
t.Errorf("Unexpected source address: %v", saddr)
}
if daddr.String() != "192.168.0.11:443" {
t.Errorf("Unexpected destination address: %v", daddr)
}
if !reflect.DeepEqual(rest, []byte("moredata")) {
t.Errorf("Unexpected rest: %v", rest)
}
}
func TestProxyProtocolV1_nontcp(t *testing.T) {
buf := []byte("PROXY UDP4 192.168.0.1 192.168.0.11 56324 443\r\nmoredata")
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.TCP)
if err == nil {
t.Errorf("Error was expected, yet returned %v %v %v", saddr, daddr, rest)
}
}
func TestProxyProtocolV1_Unknown(t *testing.T) {
buf := []byte("PROXY UNKNOWN\r\nmoredata")
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.TCP)
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
if saddr.IsValid() {
t.Errorf("Unexpected source address: %v", saddr)
}
if daddr.IsValid() {
t.Errorf("Unexpected destination address: %v", daddr)
}
if !reflect.DeepEqual(rest, []byte("moredata")) {
t.Errorf("Unexpected rest: %v", rest)
}
}
func TestProxyProtocolV1_UnknownWithAddrs(t *testing.T) {
buf := []byte("PROXY UNKNOWN ffff::1 ffff::1 1234 1234\r\nmoredata")
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.TCP)
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
if saddr.IsValid() {
t.Errorf("Unexpected source address: %v", saddr)
}
if daddr.IsValid() {
t.Errorf("Unexpected destination address: %v", daddr)
}
if !reflect.DeepEqual(rest, []byte("moredata")) {
t.Errorf("Unexpected rest: %v", rest)
}
}
func TestProxyProtocolV2(t *testing.T) {
buf := []byte{0x0D, 0x0A, 0x0D, 0x0A, 0x00, 0x0D, 0x0A, 0x51, 0x55, 0x49, 0x54, 0x0A}
buf = append(buf, 0x21) // PROXY
buf = append(buf, 0x11) // TCP4
buf = append(buf, 0x00, 0x0C) // 12 bytes
buf = append(buf, 192, 168, 0, 1) // saddr
buf = append(buf, 192, 168, 0, 11) // daddr
buf = append(buf, 0xDC, 0x04) // sport 56324
buf = append(buf, 0x01, 0xBB) // dport 443
buf = append(buf, []byte("moredata")...)
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.TCP)
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
if saddr.String() != "192.168.0.1:56324" {
t.Errorf("Unexpected source address: %v", saddr)
}
if daddr.String() != "192.168.0.11:443" {
t.Errorf("Unexpected destination address: %v", daddr)
}
if !reflect.DeepEqual(rest, []byte("moredata")) {
t.Errorf("Unexpected rest: %v", rest)
}
}
func TestProxyProtocolV2_udp6(t *testing.T) {
expectedSaddr := netip.MustParseAddr("2001:db8::1")
expectedDaddr := netip.MustParseAddr("2001:db8::2")
buf := []byte{0x0D, 0x0A, 0x0D, 0x0A, 0x00, 0x0D, 0x0A, 0x51, 0x55, 0x49, 0x54, 0x0A}
buf = append(buf, 0x21) // PROXY
buf = append(buf, 0x22) // UDP6
buf = append(buf, 0x00, 0x24) // 36 bytes
buf = append(buf, expectedSaddr.AsSlice()...)
buf = append(buf, expectedDaddr.AsSlice()...)
buf = append(buf, 0xDC, 0x04) // sport 56324
buf = append(buf, 0x01, 0xBB) // dport 443
buf = append(buf, []byte("moredata")...)
saddr, daddr, rest, err := proxyprotocol.ReadRemoteAddr(buf, utils.UDP)
if err != nil {
t.Errorf("Unexpected error: %v", err)
}
if saddr != netip.AddrPortFrom(expectedSaddr, 56324) {
t.Errorf("Unexpected source address: %v", saddr)
}
if daddr != netip.AddrPortFrom(expectedDaddr, 443) {
t.Errorf("Unexpected destination address: %v", daddr)
}
if !reflect.DeepEqual(rest, []byte("moredata")) {
t.Errorf("Unexpected rest: %v", rest)
}
}
|
