1//===- ObjectYAML.cpp - YAML utilities for object files -------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file defines a wrapper class for handling tagged YAML input
10//
11//===----------------------------------------------------------------------===//
12
13#include "llvm/ObjectYAML/ObjectYAML.h"
14#include "llvm/ADT/Twine.h"
15#include "llvm/Support/YAMLParser.h"
16#include "llvm/Support/YAMLTraits.h"
17#include <string>
18
19using namespace llvm;
20using namespace yaml;
21
22void MappingTraits<YamlObjectFile>::mapping(IO &IO,
23                                            YamlObjectFile &ObjectFile) {
24  if (IO.outputting()) {
25    if (ObjectFile.Elf)
26      MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
27    if (ObjectFile.Coff)
28      MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
29    if (ObjectFile.MachO)
30      MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
31    if (ObjectFile.FatMachO)
32      MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
33                                                         *ObjectFile.FatMachO);
34  } else {
35    Input &In = (Input &)IO;
36    if (IO.mapTag("!Arch")) {
37      ObjectFile.Arch.reset(new ArchYAML::Archive());
38      MappingTraits<ArchYAML::Archive>::mapping(IO, *ObjectFile.Arch);
39      std::string Err =
40          MappingTraits<ArchYAML::Archive>::validate(IO, *ObjectFile.Arch);
41      if (!Err.empty())
42        IO.setError(Err);
43    } else if (IO.mapTag("!ELF")) {
44      ObjectFile.Elf.reset(new ELFYAML::Object());
45      MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
46    } else if (IO.mapTag("!COFF")) {
47      ObjectFile.Coff.reset(new COFFYAML::Object());
48      MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
49    } else if (IO.mapTag("!mach-o")) {
50      ObjectFile.MachO.reset(new MachOYAML::Object());
51      MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
52    } else if (IO.mapTag("!fat-mach-o")) {
53      ObjectFile.FatMachO.reset(new MachOYAML::UniversalBinary());
54      MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
55                                                         *ObjectFile.FatMachO);
56    } else if (IO.mapTag("!minidump")) {
57      ObjectFile.Minidump.reset(new MinidumpYAML::Object());
58      MappingTraits<MinidumpYAML::Object>::mapping(IO, *ObjectFile.Minidump);
59    } else if (IO.mapTag("!WASM")) {
60      ObjectFile.Wasm.reset(new WasmYAML::Object());
61      MappingTraits<WasmYAML::Object>::mapping(IO, *ObjectFile.Wasm);
62    } else if (const Node *N = In.getCurrentNode()) {
63      if (N->getRawTag().empty())
64        IO.setError("YAML Object File missing document type tag!");
65      else
66        IO.setError("YAML Object File unsupported document type tag '" +
67                    N->getRawTag() + "'!");
68    }
69  }
70}
71