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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
|
/*
* Copyright (C) Volition, Inc. 1999. All rights reserved.
*
* All source code herein is the property of Volition, Inc. You may not sell
* or otherwise commercially exploit the source or things you created based on the
* source.
*
*/
#ifdef _WIN32
#include <winsock2.h>
#include <ws2tcpip.h>
#else
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include "pcp.h"
#include "cmdline/cmdline.h"
#include "io/timer.h"
#include "network/multi_log.h"
#include "network/multi_portfwd.h"
#include "network/psnet2.h"
// from "default_config.h" in libpcp
#ifndef PCP_SERVER_PORT
#define PCP_SERVER_PORT 5351
#endif
#ifndef PCP_MAX_SUPPORTED_VERSION
#define PCP_MAX_SUPPORTED_VERSION 2
#endif
static bool PF_initted = false;
static pcp_flow_t *PF_pcp_flow = nullptr;
static pcp_ctx_t *PF_pcp_ctx = nullptr;
static int PF_wait_timestamp = 0;
static const uint32_t PF_lifetime = 7200; // 2 hrs
static void PF_log_init();
void multi_port_forward_init()
{
SOCKADDR_STORAGE gateway_addr;
SOCKADDR_STORAGE local_addr;
int wait_ms;
bool auto_discover = true;
if (PF_initted) {
return;
}
if (Cmdline_gateway_ip) {
if ( psnet_get_addr(Cmdline_gateway_ip, PCP_SERVER_PORT, &gateway_addr) ) {
auto_discover = false;
}
}
PF_log_init(); // must call before pcp_init() to fix logging
PF_pcp_ctx = pcp_init(auto_discover ? ENABLE_AUTODISCOVERY : DISABLE_AUTODISCOVERY, nullptr);
if (PF_pcp_ctx == nullptr) {
ml_string("Port forward => Initialization failed!");
return;
}
// this needs to be after successful pcp_init() so we'll know to shutdown
PF_initted = true;
if ( !auto_discover ) {
pcp_add_server(PF_pcp_ctx, reinterpret_cast<LPSOCKADDR>(&gateway_addr), PCP_MAX_SUPPORTED_VERSION);
}
psnet_get_addr(nullptr, Psnet_default_port, &local_addr);
PF_pcp_flow = pcp_new_flow(PF_pcp_ctx, reinterpret_cast<LPSOCKADDR>(&local_addr),
nullptr, nullptr, IPPROTO_UDP, PF_lifetime, nullptr);
if (PF_pcp_flow == nullptr) {
ml_string("Port forward => Failed to init mapping!");
multi_port_forward_close();
return;
}
ml_string("Port forward => Initialized successfully");
// let's start this ball rolling...
wait_ms = pcp_pulse(PF_pcp_ctx, nullptr);
PF_wait_timestamp = timer_get_milliseconds() + wait_ms;
}
void multi_port_forward_do()
{
pcp_fstate_e state;
size_t info_count = 0;
int wait_ms;
if ( !PF_initted ) {
return;
}
if ( PF_wait_timestamp > timer_get_milliseconds() ) {
return;
}
// this ultimately does everything to maintain the connection
wait_ms = pcp_pulse(PF_pcp_ctx, nullptr);
PF_wait_timestamp = timer_get_milliseconds() + wait_ms;
// check progress and log if needed
if ( pcp_eval_flow_state(PF_pcp_flow, &state) ) {
if (state == pcp_state_failed) {
ml_string("Port forward => Mapping failed!");
} else if (state == pcp_state_succeeded) {
pcp_flow_info_t *flow_info = pcp_flow_get_info(PF_pcp_flow, &info_count);
for (size_t idx = 0; idx < info_count; ++idx) {
pcp_flow_info_t *info = &flow_info[idx];
char time_str[16];
char int_ip[INET6_ADDRSTRLEN];
char ext_ip[INET6_ADDRSTRLEN];
memset(&int_ip, 0, sizeof(int_ip));
memset(&ext_ip, 0, sizeof(ext_ip));
inet_ntop(AF_INET6, psnet_mask_addr(&info->int_ip), int_ip, sizeof(int_ip));
inet_ntop(AF_INET6, psnet_mask_addr(&info->ext_ip), ext_ip, sizeof(ext_ip));
ml_printf("Port forward => Mapping successful [%s]:%u <-> [%s]:%u",
int_ip, ntohs(info->int_port),
ext_ip, ntohs(info->ext_port));
// formatted to match what multi_log uses
strftime(time_str, sizeof(time_str), "%m/%d %H:%M:%S", localtime(&info->recv_lifetime_end));
ml_printf("Port forward => Mapping valid until %s", time_str);
}
if (flow_info) {
free(flow_info); // system call, allocated in lib ** don't change! **
}
}
}
}
void multi_port_forward_close()
{
if ( !PF_initted ) {
return;
}
pcp_terminate(PF_pcp_ctx, 1);
if (PF_pcp_flow) {
ml_string("Port forward => Mapping removed");
} else {
ml_string("Port forward => Shutdown");
}
free(PF_pcp_ctx); // system call, allocated in lib ** don't change! **
PF_pcp_ctx = nullptr;
PF_pcp_flow = nullptr;
PF_initted = false;
PF_wait_timestamp = 0;
}
#ifndef NDEBUG
static void PF_logger_fn(pcp_loglvl_e /* lvl */, const char *msg)
{
nprintf(("portfwd", "Port forward => %s", msg));
}
#endif
static void PF_log_init()
{
#ifdef NDEBUG
// log spew for libpcp is set with this
pcp_log_level = PCP_LOGLVL_NONE;
#else
pcp_log_level = PCP_LOGLVL_INFO;
pcp_set_loggerfn(PF_logger_fn);
#endif
}
|
