//===- 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() + "'!"); } } } |