ton/adnl/adnl-proxy-types.hpp

/*
    This file is part of TON Blockchain Library.

    TON Blockchain Library is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 2 of the License, or
    (at your option) any later version.

    TON Blockchain Library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.

    Copyright 2017-2019 Telegram Systems LLP
*/
#pragma once

#include "adnl-proxy-types.h"

#include "common/checksum.h"

namespace ton {

namespace adnl {

class AdnlProxyNone : public AdnlProxy {
 public:
  AdnlProxyNone() {
  }
  td::BufferSlice encrypt(Packet packet) const override {
    return std::move(packet.data);
  }
  td::Result<Packet> decrypt(td::BufferSlice packet) const override {
    return Packet{0, 0, std::move(packet)};
  }
  tl_object_ptr<ton_api::adnl_Proxy> tl() const override {
    return create_tl_object<ton_api::adnl_proxy_none>();
  }
};

class AdnlProxyFast : public AdnlProxy {
 public:
  AdnlProxyFast(td::Slice shared_secret)
      : shared_secret_(sha256_bits256(shared_secret)), shared_secret_raw_(shared_secret) {
  }
  td::BufferSlice encrypt(Packet packet) const override;
  td::Result<Packet> decrypt(td::BufferSlice packet) const override;
  tl_object_ptr<ton_api::adnl_Proxy> tl() const override {
    return create_tl_object<ton_api::adnl_proxy_fast>(shared_secret_raw_.clone_as_buffer_slice());
  }

 private:
  td::Bits256 shared_secret_;
  td::SharedSlice shared_secret_raw_;
};

}  // namespace adnl

}  // namespace ton