dtls_utils.h - mozsearch

Enable keyboard shortcuts

/*

 *  Copyright 2024 The WebRTC Project Authors. All rights reserved.

 *  Use of this source code is governed by a BSD-style license

 *  that can be found in the LICENSE file in the root of the source

 *  tree. An additional intellectual property rights grant can be found

 *  in the file PATENTS.  All contributing project authors may

 *  be found in the AUTHORS file in the root of the source tree.

*/

#ifndef P2P_DTLS_DTLS_UTILS_H_

#define P2P_DTLS_DTLS_UTILS_H_

#include <cstddef>

#include <cstdint>

#include <memory>

#include <optional>

#include <vector>

#include "absl/container/flat_hash_set.h"

#include "api/array_view.h"

#include "rtc_base/buffer.h"

namespace webrtc {

const size_t kDtlsRecordHeaderLen = 13;

const size_t kMaxDtlsPacketLen = 2048;

bool IsDtlsPacket(ArrayView<const uint8_t> payload);

bool IsDtlsClientHelloPacket(ArrayView<const uint8_t> payload);

bool IsDtlsHandshakePacket(ArrayView<const uint8_t> payload);

std::optional<std::vector<uint16_t>> GetDtlsHandshakeAcks(

    ArrayView<const uint8_t> dtls_packet);

uint32_t ComputeDtlsPacketHash(ArrayView<const uint8_t> dtls_packet);

class PacketStash {

 public:

  PacketStash() {}

  void Add(ArrayView<const uint8_t> packet);

  bool AddIfUnique(ArrayView<const uint8_t> packet);

  void Prune(const absl::flat_hash_set<uint32_t>& packet_hashes);

  void Prune(uint32_t max_size);

  ArrayView<const uint8_t> GetNext();

  void clear() {

    packets_.clear();

    pos_ = 0;

  bool empty() const { return packets_.empty(); }

  int size() const { return packets_.size(); }

  static uint32_t Hash(ArrayView<const uint8_t> packet) {

    return ComputeDtlsPacketHash(packet);

 private:

  struct StashedPacket {

    uint32_t hash;

    std::unique_ptr<Buffer> buffer;

};

  // This vector will only contain very few items,

  // so it is appropriate to use a vector rather than

  // e.g. a hash map.

  uint32_t pos_ = 0;

  std::vector<StashedPacket> packets_;

};

}  //  namespace webrtc

// Re-export symbols from the webrtc namespace for backwards compatibility.

// TODO(bugs.webrtc.org/4222596): Remove once all references are updated.

#ifdef WEBRTC_ALLOW_DEPRECATED_NAMESPACES

namespace cricket {

using ::webrtc::GetDtlsHandshakeAcks;

using ::webrtc::IsDtlsClientHelloPacket;

using ::webrtc::IsDtlsHandshakePacket;

using ::webrtc::IsDtlsPacket;

using ::webrtc::kDtlsRecordHeaderLen;

using ::webrtc::kMaxDtlsPacketLen;

}  // namespace cricket

#endif  // WEBRTC_ALLOW_DEPRECATED_NAMESPACES

#endif  // P2P_DTLS_DTLS_UTILS_H_