diff options
Diffstat (limited to 'libtransport/src/protocols/verification_manager.h')
-rw-r--r-- | libtransport/src/protocols/verification_manager.h | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/libtransport/src/protocols/verification_manager.h b/libtransport/src/protocols/verification_manager.h new file mode 100644 index 000000000..7d8a00a65 --- /dev/null +++ b/libtransport/src/protocols/verification_manager.h @@ -0,0 +1,71 @@ +/* + * 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/interfaces/callbacks.h> +#include <hicn/transport/interfaces/verification_policy.h> + +#include <hicn/transport/core/content_object.h> + +#include <protocols/errors.h> + +#include <queue> + +namespace transport { + +namespace interface { +class ConsumerSocket; +} + +namespace protocol { + +using Packet = core::Packet; +using interface::VerificationPolicy; +using ContentObjectPtr = std::shared_ptr<core::ContentObject>; + +class VerificationManager { + public: + virtual ~VerificationManager() = default; + virtual VerificationPolicy onPacketToVerify(const Packet& packet) = 0; + virtual bool onKeyToVerify() { return false; } +}; + +class SignatureVerificationManager : public VerificationManager { + public: + SignatureVerificationManager(implementation::ConsumerSocket* icn_socket) + : icn_socket_(icn_socket), key_packets_() {} + + interface::VerificationPolicy onPacketToVerify(const Packet& packet) override; + bool onKeyToVerify() override; + + private: + implementation::ConsumerSocket* icn_socket_; + std::queue<ContentObjectPtr> key_packets_; + + ContentObjectPtr copyPacket(const Packet& packet) { + std::shared_ptr<utils::MemBuf> packet_copy = + packet.acquireMemBufReference(); + ContentObjectPtr content_object_copy = + std::make_shared<core::ContentObject>(std::move(packet_copy)); + std::unique_ptr<utils::MemBuf> payload_copy = packet.getPayload(); + content_object_copy->appendPayload(std::move(payload_copy)); + return content_object_copy; + } +}; + +} // end namespace protocol + +} // end namespace transport |