mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2024-12-14 19:49:36 +00:00
45d042ed96
yaml2obj might crash on invalid input when unable to parse the YAML. Recently a crash with a very similar nature was fixed for an empty files. This patch revisits the fix and does it in yaml::Input instead. It seems to be more correct way to handle such situation. With that crash for invalid inputs is also fixed now. Differential revision: https://reviews.llvm.org/D61059 llvm-svn: 359178
64 lines
2.7 KiB
C++
64 lines
2.7 KiB
C++
//===- ObjectYAML.cpp - YAML utilities for object files -------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines a wrapper class for handling tagged YAML input
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ObjectYAML/ObjectYAML.h"
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/Support/YAMLParser.h"
|
|
#include "llvm/Support/YAMLTraits.h"
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
using namespace yaml;
|
|
|
|
void MappingTraits<YamlObjectFile>::mapping(IO &IO,
|
|
YamlObjectFile &ObjectFile) {
|
|
if (IO.outputting()) {
|
|
if (ObjectFile.Elf)
|
|
MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
|
|
if (ObjectFile.Coff)
|
|
MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
|
|
if (ObjectFile.MachO)
|
|
MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
|
|
if (ObjectFile.FatMachO)
|
|
MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
|
|
*ObjectFile.FatMachO);
|
|
} else {
|
|
Input &In = (Input &)IO;
|
|
if (IO.mapTag("!ELF")) {
|
|
ObjectFile.Elf.reset(new ELFYAML::Object());
|
|
MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
|
|
} else if (IO.mapTag("!COFF")) {
|
|
ObjectFile.Coff.reset(new COFFYAML::Object());
|
|
MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
|
|
} else if (IO.mapTag("!mach-o")) {
|
|
ObjectFile.MachO.reset(new MachOYAML::Object());
|
|
MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
|
|
} else if (IO.mapTag("!fat-mach-o")) {
|
|
ObjectFile.FatMachO.reset(new MachOYAML::UniversalBinary());
|
|
MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
|
|
*ObjectFile.FatMachO);
|
|
} else if (IO.mapTag("!minidump")) {
|
|
ObjectFile.Minidump.reset(new MinidumpYAML::Object());
|
|
MappingTraits<MinidumpYAML::Object>::mapping(IO, *ObjectFile.Minidump);
|
|
} else if (IO.mapTag("!WASM")) {
|
|
ObjectFile.Wasm.reset(new WasmYAML::Object());
|
|
MappingTraits<WasmYAML::Object>::mapping(IO, *ObjectFile.Wasm);
|
|
} else if (const Node *N = In.getCurrentNode()) {
|
|
if (N->getRawTag().empty())
|
|
IO.setError("YAML Object File missing document type tag!");
|
|
else
|
|
IO.setError("YAML Object File unsupported document type tag '" +
|
|
N->getRawTag() + "'!");
|
|
}
|
|
}
|
|
}
|