1 files changed, 84 insertions, 0 deletions
diff --git a/protocols/Telegram/tdlib/td/tdnet/td/net/TransparentProxy.cpp b/protocols/Telegram/tdlib/td/tdnet/td/net/TransparentProxy.cpp
new file mode 100644
index 0000000000..b5102a37b4
--- /dev/null
+++ b/protocols/Telegram/tdlib/td/tdnet/td/net/TransparentProxy.cpp
@@ -0,0 +1,84 @@
+//
+// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
+//
+// 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());
+    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