summaryrefslogtreecommitdiffstats
path: root/libtransport/src/implementation/p2psecure_socket_producer.h
blob: f9434725812dbabc178d9d32eb36542b62848576 (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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
 * Copyright (c) 2017-2019 Cisco and/or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <hicn/transport/auth/identity.h>
#include <hicn/transport/auth/signer.h>
#include <implementation/socket_producer.h>
// #include <implementation/tls_rtc_socket_producer.h>
#include <implementation/tls_socket_producer.h>
#include <openssl/ssl.h>
#include <utils/content_store.h>

#include <condition_variable>
#include <forward_list>
#include <mutex>

namespace transport {
namespace implementation {

class P2PSecureProducerSocket : public ProducerSocket {
  friend class TLSProducerSocket;
  // TODO
  //   friend class TLSRTCProducerSocket;

 public:
  explicit P2PSecureProducerSocket(interface::ProducerSocket *producer_socket);

  explicit P2PSecureProducerSocket(
      interface::ProducerSocket *producer_socket, bool rtc,
      const std::shared_ptr<auth::Identity> &identity);

  ~P2PSecureProducerSocket();

  uint32_t produceDatagram(const Name &content_name,
                           std::unique_ptr<utils::MemBuf> &&buffer) override;

  uint32_t produceStream(const Name &content_name, const uint8_t *buffer,
                         size_t buffer_size, bool is_last = true,
                         uint32_t start_offset = 0) override;

  uint32_t produceStream(const Name &content_name,
                         std::unique_ptr<utils::MemBuf> &&buffer,
                         bool is_last = true,
                         uint32_t start_offset = 0) override;

  void asyncProduce(Name content_name, std::unique_ptr<utils::MemBuf> &&buffer,
                    bool is_last, uint32_t offset,
                    uint32_t **last_segment = nullptr) override;

  int setSocketOption(int socket_option_key,
                      ProducerInterestCallback socket_option_value) override;

  int setSocketOption(
      int socket_option_key,
      const std::shared_ptr<auth::Signer> &socket_option_value) override;

  int setSocketOption(int socket_option_key,
                      uint32_t socket_option_value) override;

  int setSocketOption(int socket_option_key, bool socket_option_value) override;

  int setSocketOption(int socket_option_key,
                      Name *socket_option_value) override;

  int setSocketOption(
      int socket_option_key,
      ProducerContentObjectCallback socket_option_value) override;

  int setSocketOption(int socket_option_key,
                      ProducerContentCallback socket_option_value) override;

  int setSocketOption(int socket_option_key,
                      auth::CryptoHashType socket_option_value) override;

  int setSocketOption(int socket_option_key,
                      const std::string &socket_option_value) override;

  using ProducerSocket::getSocketOption;
  //   using ProducerSocket::onInterest;

 protected:
  /* Callback invoked once an interest has been received and its payload
   * decrypted */
  ProducerInterestCallback on_interest_input_decrypted_;
  ProducerInterestCallback on_interest_process_decrypted_;
  ProducerContentCallback on_content_produced_application_;

 private:
  bool rtc_;
  std::mutex mtx_;
  /* Condition variable for the wait */
  std::condition_variable cv_;
  X509 *cert_509_;
  EVP_PKEY *pkey_rsa_;
  std::unordered_map<core::Name, std::unique_ptr<TLSProducerSocket>,
                     core::hash<core::Name>, core::compare2<core::Name>>
      map_producers;
  std::list<std::unique_ptr<TLSProducerSocket>> list_producers;

  void onInterestCallback(interface::ProducerSocket &p, Interest &interest);

  void initSessionSocket(std::unique_ptr<TLSProducerSocket> &producer);
};

}  // namespace implementation
}  // namespace transport