// Copyright 2016 Google Inc. All Rights Reserved.
//
// 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.

#include "bloaty.h"
#include "strarr.h"

namespace bloaty {

class StringPieceInputFile : public InputFile {
 public:
  StringPieceInputFile(StringPiece data)
      : InputFile("fake_StringPieceInputFile_file") {
    data_ = data;
  }
};

class StringPieceInputFileFactory : public InputFileFactory {
 public:
  StringPieceInputFileFactory(StringPiece data) : data_(data) {}
 private:
  StringPiece data_;
  std::unique_ptr<InputFile> TryOpenFile(
      const std::string& filename) const override {
    return std::unique_ptr<InputFile>(new StringPieceInputFile(data_));
  }
};

void RunBloaty(const InputFileFactory& factory,
               const std::string& data_source) {
  bloaty::RollupOutput output;
  StrArr args({"bloaty", "-d", data_source, "dummy_filename"});
  bloaty::BloatyMain(args.size(), args.get(), factory, &output);
}

}  // namespace bloaty

extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
  const char *data2 = reinterpret_cast<const char*>(data);
  bloaty::StringPieceInputFileFactory factory(bloaty::StringPiece(data2, size));

  // Try all of the data sources.
  RunBloaty(factory, "segments");
  RunBloaty(factory, "sections");
  RunBloaty(factory, "symbols");
  RunBloaty(factory, "compileunits");
  RunBloaty(factory, "inlines");
  RunBloaty(factory, "armembers");

  return 0;
}