10b57cec5SDimitry Andric //===- ObjectYAML.cpp - YAML utilities for object files -------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric //
90b57cec5SDimitry Andric // This file defines a wrapper class for handling tagged YAML input
100b57cec5SDimitry Andric //
110b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
120b57cec5SDimitry Andric
130b57cec5SDimitry Andric #include "llvm/ObjectYAML/ObjectYAML.h"
140b57cec5SDimitry Andric #include "llvm/ADT/Twine.h"
150b57cec5SDimitry Andric #include "llvm/Support/YAMLParser.h"
160b57cec5SDimitry Andric #include "llvm/Support/YAMLTraits.h"
170b57cec5SDimitry Andric #include <string>
180b57cec5SDimitry Andric
190b57cec5SDimitry Andric using namespace llvm;
200b57cec5SDimitry Andric using namespace yaml;
210b57cec5SDimitry Andric
mapping(IO & IO,YamlObjectFile & ObjectFile)220b57cec5SDimitry Andric void MappingTraits<YamlObjectFile>::mapping(IO &IO,
230b57cec5SDimitry Andric YamlObjectFile &ObjectFile) {
240b57cec5SDimitry Andric if (IO.outputting()) {
250b57cec5SDimitry Andric if (ObjectFile.Elf)
260b57cec5SDimitry Andric MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
270b57cec5SDimitry Andric if (ObjectFile.Coff)
280b57cec5SDimitry Andric MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
295f757f3fSDimitry Andric if (ObjectFile.Goff)
305f757f3fSDimitry Andric MappingTraits<GOFFYAML::Object>::mapping(IO, *ObjectFile.Goff);
310b57cec5SDimitry Andric if (ObjectFile.MachO)
320b57cec5SDimitry Andric MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
330b57cec5SDimitry Andric if (ObjectFile.FatMachO)
340b57cec5SDimitry Andric MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
350b57cec5SDimitry Andric *ObjectFile.FatMachO);
360b57cec5SDimitry Andric } else {
370b57cec5SDimitry Andric Input &In = (Input &)IO;
38e8d8bef9SDimitry Andric if (IO.mapTag("!Arch")) {
39e8d8bef9SDimitry Andric ObjectFile.Arch.reset(new ArchYAML::Archive());
40e8d8bef9SDimitry Andric MappingTraits<ArchYAML::Archive>::mapping(IO, *ObjectFile.Arch);
41e8d8bef9SDimitry Andric std::string Err =
42e8d8bef9SDimitry Andric MappingTraits<ArchYAML::Archive>::validate(IO, *ObjectFile.Arch);
43e8d8bef9SDimitry Andric if (!Err.empty())
44e8d8bef9SDimitry Andric IO.setError(Err);
45e8d8bef9SDimitry Andric } else if (IO.mapTag("!ELF")) {
460b57cec5SDimitry Andric ObjectFile.Elf.reset(new ELFYAML::Object());
470b57cec5SDimitry Andric MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
480b57cec5SDimitry Andric } else if (IO.mapTag("!COFF")) {
490b57cec5SDimitry Andric ObjectFile.Coff.reset(new COFFYAML::Object());
500b57cec5SDimitry Andric MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
515f757f3fSDimitry Andric } else if (IO.mapTag("!GOFF")) {
525f757f3fSDimitry Andric ObjectFile.Goff.reset(new GOFFYAML::Object());
535f757f3fSDimitry Andric MappingTraits<GOFFYAML::Object>::mapping(IO, *ObjectFile.Goff);
540b57cec5SDimitry Andric } else if (IO.mapTag("!mach-o")) {
550b57cec5SDimitry Andric ObjectFile.MachO.reset(new MachOYAML::Object());
560b57cec5SDimitry Andric MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
570b57cec5SDimitry Andric } else if (IO.mapTag("!fat-mach-o")) {
580b57cec5SDimitry Andric ObjectFile.FatMachO.reset(new MachOYAML::UniversalBinary());
590b57cec5SDimitry Andric MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
600b57cec5SDimitry Andric *ObjectFile.FatMachO);
610b57cec5SDimitry Andric } else if (IO.mapTag("!minidump")) {
620b57cec5SDimitry Andric ObjectFile.Minidump.reset(new MinidumpYAML::Object());
630b57cec5SDimitry Andric MappingTraits<MinidumpYAML::Object>::mapping(IO, *ObjectFile.Minidump);
6481ad6265SDimitry Andric } else if (IO.mapTag("!Offload")) {
6581ad6265SDimitry Andric ObjectFile.Offload.reset(new OffloadYAML::Binary());
6681ad6265SDimitry Andric MappingTraits<OffloadYAML::Binary>::mapping(IO, *ObjectFile.Offload);
670b57cec5SDimitry Andric } else if (IO.mapTag("!WASM")) {
680b57cec5SDimitry Andric ObjectFile.Wasm.reset(new WasmYAML::Object());
690b57cec5SDimitry Andric MappingTraits<WasmYAML::Object>::mapping(IO, *ObjectFile.Wasm);
70fe6060f1SDimitry Andric } else if (IO.mapTag("!XCOFF")) {
71fe6060f1SDimitry Andric ObjectFile.Xcoff.reset(new XCOFFYAML::Object());
72fe6060f1SDimitry Andric MappingTraits<XCOFFYAML::Object>::mapping(IO, *ObjectFile.Xcoff);
7381ad6265SDimitry Andric } else if (IO.mapTag("!dxcontainer")) {
7481ad6265SDimitry Andric ObjectFile.DXContainer.reset(new DXContainerYAML::Object());
7581ad6265SDimitry Andric MappingTraits<DXContainerYAML::Object>::mapping(IO,
7681ad6265SDimitry Andric *ObjectFile.DXContainer);
770b57cec5SDimitry Andric } else if (const Node *N = In.getCurrentNode()) {
780b57cec5SDimitry Andric if (N->getRawTag().empty())
790b57cec5SDimitry Andric IO.setError("YAML Object File missing document type tag!");
800b57cec5SDimitry Andric else
810b57cec5SDimitry Andric IO.setError("YAML Object File unsupported document type tag '" +
820b57cec5SDimitry Andric N->getRawTag() + "'!");
830b57cec5SDimitry Andric }
840b57cec5SDimitry Andric }
850b57cec5SDimitry Andric }
86