bin/http/upload_element_reader.h

// Copyright 2016 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef GARNET_BIN_HTTP_UPLOAD_ELEMENT_READER_H_
#define GARNET_BIN_HTTP_UPLOAD_ELEMENT_READER_H_

#include <array>
#include <limits>

#include <lib/fxl/macros.h>
#include <lib/zx/socket.h>
#include <lib/zx/vmo.h>

namespace http {

class UploadElementReader {
 public:
  static constexpr size_t BUFSIZE = 1024;
  static constexpr size_t kUnknownSize = std::numeric_limits<size_t>::max();

  UploadElementReader();
  virtual ~UploadElementReader();

  zx_status_t err() const { return err_; }

  // may produce kUnknownSize
  virtual size_t size() = 0;

  // returns true if and only if content was read or there is potentially more
  // to read
  virtual bool ReadAvailable(std::ostream* os) = 0;

 protected:
  zx_status_t err_;

 private:
  FXL_DISALLOW_COPY_AND_ASSIGN(UploadElementReader);
};

class SocketUploadElementReader : public UploadElementReader {
 public:
  SocketUploadElementReader(zx::socket socket);
  ~SocketUploadElementReader() override;

  // always kUnknownSize
  size_t size() override;
  bool ReadAvailable(std::ostream* os) override;

 private:
  zx::socket socket_;
  std::array<char, BUFSIZE> buf_;
};

class VmoUploadElementReader : public UploadElementReader {
 public:
  VmoUploadElementReader(zx::vmo vmo);
  VmoUploadElementReader(zx::vmo vmo, uint64_t size);
  ~VmoUploadElementReader() override;

  size_t size() override;
  bool ReadAvailable(std::ostream* os) override;

 private:
  zx::vmo vmo_;
  uint64_t size_;
  uint64_t offset_;
  std::array<char, BUFSIZE> buf_;
};

}  // namespace http

#endif  // GARNET_BIN_HTTP_UPLOAD_ELEMENT_READER_H_