PlaintextRecordLayer.h

Go to the documentation of this file.

/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <fizz/record/RecordLayer.h>

namespace fizz {

class PlaintextReadRecordLayer : public ReadRecordLayer {
 public:
  ~PlaintextReadRecordLayer() override = default;

  virtual void setSkipEncryptedRecords(bool enabled) {
    skipEncryptedRecords_ = enabled;
  }

  folly::Optional<TLSMessage> read(folly::IOBufQueue& buf) override;

  folly::Optional<ProtocolVersion> getReceivedRecordVersion() const {
    return receivedRecordVersion_;
  }

  EncryptionLevel getEncryptionLevel() const override;

 private:
  bool skipEncryptedRecords_{false};

  folly::Optional<ProtocolVersion> receivedRecordVersion_;
};

class PlaintextWriteRecordLayer : public WriteRecordLayer {
 public:
  ~PlaintextWriteRecordLayer() override = default;

  TLSContent write(TLSMessage&& msg) const override;

  virtual TLSContent writeInitialClientHello(Buf encodedClientHello) const;

  EncryptionLevel getEncryptionLevel() const override;

 private:
  TLSContent write(TLSMessage msg, ProtocolVersion recordVersion) const;
};
} // namespace fizz