blob: de2556f5763b4db6758323ceb8aa750a85abf433 (
plain)
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
|
//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/net/TransparentProxy.h"
#include "td/utils/logging.h"
#include "td/utils/port/detail/PollableFd.h"
namespace td {
int VERBOSITY_NAME(proxy) = VERBOSITY_NAME(DEBUG);
TransparentProxy::TransparentProxy(SocketFd socket_fd, IPAddress ip_address, string username, string password,
unique_ptr<Callback> callback, ActorShared<> parent)
: fd_(std::move(socket_fd))
, ip_address_(std::move(ip_address))
, username_(std::move(username))
, password_(std::move(password))
, callback_(std::move(callback))
, parent_(std::move(parent)) {
}
void TransparentProxy::on_error(Status status) {
CHECK(status.is_error());
VLOG(proxy) << "Receive " << status;
if (callback_) {
callback_->set_result(std::move(status));
callback_.reset();
}
stop();
}
void TransparentProxy::tear_down() {
VLOG(proxy) << "Finish to connect to proxy";
Scheduler::unsubscribe(fd_.get_poll_info().get_pollable_fd_ref());
if (callback_) {
if (!fd_.input_buffer().empty()) {
LOG(ERROR) << "Have " << fd_.input_buffer().size() << " unread bytes";
callback_->set_result(Status::Error("Proxy has sent too many data"));
} else {
callback_->set_result(std::move(fd_));
}
callback_.reset();
}
}
void TransparentProxy::hangup() {
on_error(Status::Error("Canceled"));
}
void TransparentProxy::start_up() {
VLOG(proxy) << "Begin to connect to proxy";
Scheduler::subscribe(fd_.get_poll_info().extract_pollable_fd(this));
set_timeout_in(10);
sync_with_poll(fd_);
if (can_write_local(fd_)) {
loop();
}
}
void TransparentProxy::loop() {
sync_with_poll(fd_);
auto status = [&] {
TRY_STATUS(fd_.flush_read());
TRY_STATUS(loop_impl());
TRY_STATUS(fd_.flush_write());
if (can_close_local(fd_)) {
return Status::Error("Connection closed");
}
return Status::OK();
}();
if (status.is_error()) {
on_error(std::move(status));
}
}
void TransparentProxy::timeout_expired() {
on_error(Status::Error("Connection timeout expired"));
}
} // namespace td
|
