rldp2 support in rldp-http-proxy (#608)

Co-authored-by: SpyCheese <[email protected]>

Author: EmelyanenkoK

rldp-http-proxy/CMakeLists.txt

@@ -2,4 +2,4 @@ cmake_minimum_required(VERSION 3.0.2 FATAL_ERROR)
 
 add_executable(rldp-http-proxy rldp-http-proxy.cpp DNSResolver.h DNSResolver.cpp)
 target_include_directories(rldp-http-proxy PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/..>)
-target_link_libraries(rldp-http-proxy PRIVATE tonhttp rldp dht tonlib git)
+target_link_libraries(rldp-http-proxy PRIVATE tonhttp rldp rldp2 dht tonlib git)

rldp-http-proxy/rldp-http-proxy.cpp

@@ -92,6 +92,8 @@ class RldpIn : public RldpImpl {
 
   void get_conn_ip_str(adnl::AdnlNodeIdShort l_id, adnl::AdnlNodeIdShort p_id, td::Promise<td::string> promise) override;
 
+  void set_default_mtu(td::uint64 mtu) override;
+
   RldpIn(td::actor::ActorId<adnl::AdnlPeerTable> adnl) : adnl_(adnl) {
   }
 
@@ -107,6 +109,8 @@ class RldpIn : public RldpImpl {
 
   std::set<adnl::AdnlNodeIdShort> local_ids_;
 
+  td::optional<td::uint64> custom_default_mtu_;
+
   td::actor::ActorId<RldpConnectionActor> create_connection(adnl::AdnlNodeIdShort src, adnl::AdnlNodeIdShort dst);
 };
 

rldp2/rldp-in.hpp

@@ -44,6 +44,9 @@ class RldpConnectionActor : public td::actor::Actor, private ConnectionCallback
     connection_.receive_raw(std::move(data));
     yield();
   }
+  void set_default_mtu(td::uint64 mtu) {
+    connection_.set_default_mtu(mtu);
+  }
 
  private:
   td::actor::ActorId<RldpIn> rldp_;
@@ -129,6 +132,9 @@ td::actor::ActorId<RldpConnectionActor> RldpIn::create_connection(adnl::AdnlNode
     return it->second.get();
   }
   auto connection = td::actor::create_actor<RldpConnectionActor>("RldpConnection", actor_id(this), src, dst, adnl_);
+  if (custom_default_mtu_) {
+    td::actor::send_closure(connection, &RldpConnectionActor::set_default_mtu, custom_default_mtu_.value());
+  }
   auto res = connection.get();
   connections_[std::make_pair(src, dst)] = std::move(connection);
   return res;
@@ -221,6 +227,13 @@ void RldpIn::get_conn_ip_str(adnl::AdnlNodeIdShort l_id, adnl::AdnlNodeIdShort p
   td::actor::send_closure(adnl_, &adnl::AdnlPeerTable::get_conn_ip_str, l_id, p_id, std::move(promise));
 }
 
+void RldpIn::set_default_mtu(td::uint64 mtu) {
+  custom_default_mtu_ = mtu;
+  for (auto &connection : connections_) {
+    td::actor::send_closure(connection.second, &RldpConnectionActor::set_default_mtu, mtu);
+  }
+}
+
 std::unique_ptr<adnl::Adnl::Callback> RldpIn::make_adnl_callback() {
   class Callback : public adnl::Adnl::Callback {
    private:

rldp2/rldp.cpp

@@ -37,6 +37,8 @@ class Rldp : public adnl::AdnlSenderInterface {
   virtual void send_message_ex(adnl::AdnlNodeIdShort src, adnl::AdnlNodeIdShort dst, td::Timestamp timeout,
                                td::BufferSlice data) = 0;
 
+  virtual void set_default_mtu(td::uint64 mtu) = 0;
+
   static td::actor::ActorOwn<Rldp> create(td::actor::ActorId<adnl::Adnl> adnl);
 };
 

rldp2/rldp.h

@@ -720,11 +720,13 @@ storage.getPiece piece_id:int = storage.Piece;
 http.header name:string value:string = http.Header;
 http.payloadPart data:bytes trailer:(vector http.header) last:Bool = http.PayloadPart;
 http.response http_version:string status_code:int reason:string headers:(vector http.header) no_payload:Bool = http.Response;
+http.proxy.capabilities capabilities:long = http.proxy.Capabilities;
 
 ---functions---
 
 http.request id:int256 method:string url:string http_version:string headers:(vector http.header) = http.Response;
 http.getNextPayloadPart id:int256 seqno:int max_chunk_size:int = http.PayloadPart;
+http.proxy.getCapabilities capabilities:long = http.proxy.Capabilities;
 
 ---types---