mirror of
https://github.com/RPCS3/llvm-mirror.git
synced 2024-11-22 18:54:02 +01:00
ed47bccbaf
Multiple times we faced an issue of huge outputs due to unexpected behavior or incorrect test cases. The last one was https://reviews.llvm.org/D80629#2073066. This patch limits the output to 10 Mb for ELF and introduces the --max-size to change this limit. I've tried to keep the implementation non-intrusive. The current logic we have is that we prepare section content in a buffer first and write it to the output later. This patch checks the available limit on each writing attempt to this buffer and stops writing when the limit is reached and raises the internal error flag. Later, this flag is is checked before the actual writing to a file happens and an error is reported. Differential revision: https://reviews.llvm.org/D81258
78 lines
2.2 KiB
C++
78 lines
2.2 KiB
C++
//===-- yaml2obj.cpp ------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ObjectYAML/yaml2obj.h"
|
|
#include "llvm/ADT/StringExtras.h"
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/Object/ObjectFile.h"
|
|
#include "llvm/ObjectYAML/ObjectYAML.h"
|
|
#include "llvm/Support/Errc.h"
|
|
#include "llvm/Support/WithColor.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
|
|
namespace llvm {
|
|
namespace yaml {
|
|
|
|
bool convertYAML(yaml::Input &YIn, raw_ostream &Out, ErrorHandler ErrHandler,
|
|
unsigned DocNum, uint64_t MaxSize) {
|
|
unsigned CurDocNum = 0;
|
|
do {
|
|
if (++CurDocNum != DocNum)
|
|
continue;
|
|
|
|
yaml::YamlObjectFile Doc;
|
|
YIn >> Doc;
|
|
if (std::error_code EC = YIn.error()) {
|
|
ErrHandler("failed to parse YAML input: " + EC.message());
|
|
return false;
|
|
}
|
|
|
|
if (Doc.Elf)
|
|
return yaml2elf(*Doc.Elf, Out, ErrHandler, MaxSize);
|
|
if (Doc.Coff)
|
|
return yaml2coff(*Doc.Coff, Out, ErrHandler);
|
|
if (Doc.MachO || Doc.FatMachO)
|
|
return yaml2macho(Doc, Out, ErrHandler);
|
|
if (Doc.Minidump)
|
|
return yaml2minidump(*Doc.Minidump, Out, ErrHandler);
|
|
if (Doc.Wasm)
|
|
return yaml2wasm(*Doc.Wasm, Out, ErrHandler);
|
|
|
|
ErrHandler("unknown document type");
|
|
return false;
|
|
|
|
} while (YIn.nextDocument());
|
|
|
|
ErrHandler("cannot find the " + Twine(DocNum) +
|
|
getOrdinalSuffix(DocNum).data() + " document");
|
|
return false;
|
|
}
|
|
|
|
std::unique_ptr<object::ObjectFile>
|
|
yaml2ObjectFile(SmallVectorImpl<char> &Storage, StringRef Yaml,
|
|
ErrorHandler ErrHandler) {
|
|
Storage.clear();
|
|
raw_svector_ostream OS(Storage);
|
|
|
|
yaml::Input YIn(Yaml);
|
|
if (!convertYAML(YIn, OS, ErrHandler))
|
|
return {};
|
|
|
|
Expected<std::unique_ptr<object::ObjectFile>> ObjOrErr =
|
|
object::ObjectFile::createObjectFile(
|
|
MemoryBufferRef(OS.str(), "YamlObject"));
|
|
if (ObjOrErr)
|
|
return std::move(*ObjOrErr);
|
|
|
|
ErrHandler(toString(ObjOrErr.takeError()));
|
|
return {};
|
|
}
|
|
|
|
} // namespace yaml
|
|
} // namespace llvm
|