HPACKEncodeBuffer.h

Go to the documentation of this file.

/*
 *  Copyright (c) 2015-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. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */
#pragma once

#include <folly/FBString.h>
#include <folly/io/Cursor.h>
#include <folly/io/IOBufQueue.h>
#include <proxygen/lib/http/codec/compress/HPACKConstants.h>
#include <proxygen/lib/http/codec/compress/Huffman.h>

namespace proxygen {

class HPACKEncodeBuffer {

 public:
  HPACKEncodeBuffer(
    uint32_t growthSize,
    bool huffmanEnabled);

  explicit HPACKEncodeBuffer(uint32_t growthSize);

  ~HPACKEncodeBuffer() {}

  std::unique_ptr<folly::IOBuf> release() {
    return bufQueue_.move();
  }

  void clear() {
    bufQueue_.clear();
  }

  void addHeadroom(uint32_t bytes);

  uint32_t encodeInteger(uint64_t value, uint8_t instruction, uint8_t nbit);

  uint32_t encodeInteger(uint64_t value, const HPACK::Instruction& instruction);

  uint32_t encodeInteger(uint64_t value);

  uint32_t encodeLiteral(folly::StringPiece literal);

  uint32_t encodeLiteral(uint8_t instruction, uint8_t nbit,
                         folly::StringPiece literal);

  uint32_t encodeHuffman(folly::StringPiece literal);

  uint32_t encodeHuffman(uint8_t instruction, uint8_t nbit,
                         folly::StringPiece literal);

  std::string toBin();

 private:

  void append(uint8_t byte);

  folly::IOBufQueue bufQueue_;
  folly::io::QueueAppender buf_;
  uint32_t growthSize_;
  bool huffmanEnabled_;
};

}