tdlight/tdnet/td/net/TransparentProxy.cpp

85 lines
2.3 KiB
C++
Raw Normal View History

//
2021-01-01 13:57:46 +01:00
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2021
//
// 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 to much data"));
} else {
callback_->set_result(std::move(fd_));
}
callback_.reset();
}
}
void TransparentProxy::hangup() {
on_error(Status::Error("Cancelled"));
}
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());
return Status::OK();
}();
if (status.is_error()) {
on_error(std::move(status));
}
if (can_close_local(fd_)) {
on_error(Status::Error("Connection closed"));
}
}
void TransparentProxy::timeout_expired() {
on_error(Status::Error("Connection timeout expired"));
}
} // namespace td