Line data    Source code

       1             : //===- ObjectYAML.cpp - YAML utilities for object files -------------------===//
       2             : //
       3             : //                     The LLVM Compiler Infrastructure
       4             : //
       5             : // This file is distributed under the University of Illinois Open Source
       6             : // License. See LICENSE.TXT for details.
       7             : //
       8             : //===----------------------------------------------------------------------===//
       9             : //
      10             : // This file defines a wrapper class for handling tagged YAML input
      11             : //
      12             : //===----------------------------------------------------------------------===//
      13             : 
      14             : #include "llvm/ObjectYAML/ObjectYAML.h"
      15             : #include "llvm/ADT/Twine.h"
      16             : #include "llvm/Support/YAMLParser.h"
      17             : #include "llvm/Support/YAMLTraits.h"
      18             : #include <string>
      19             : 
      20             : using namespace llvm;
      21             : using namespace yaml;
      22             : 
      23         632 : void MappingTraits<YamlObjectFile>::mapping(IO &IO,
      24             :                                             YamlObjectFile &ObjectFile) {
      25         632 :   if (IO.outputting()) {
      26          30 :     if (ObjectFile.Elf)
      27           0 :       MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
      28          30 :     if (ObjectFile.Coff)
      29           0 :       MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
      30          30 :     if (ObjectFile.MachO)
      31          30 :       MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
      32          30 :     if (ObjectFile.FatMachO)
      33           0 :       MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
      34             :                                                          *ObjectFile.FatMachO);
      35             :   } else {
      36        1204 :     if (IO.mapTag("!ELF")) {
      37         288 :       ObjectFile.Elf.reset(new ELFYAML::Object());
      38         288 :       MappingTraits<ELFYAML::Object>::mapping(IO, *ObjectFile.Elf);
      39         628 :     } else if (IO.mapTag("!COFF")) {
      40         244 :       ObjectFile.Coff.reset(new COFFYAML::Object());
      41         244 :       MappingTraits<COFFYAML::Object>::mapping(IO, *ObjectFile.Coff);
      42         140 :     } else if (IO.mapTag("!mach-o")) {
      43          37 :       ObjectFile.MachO.reset(new MachOYAML::Object());
      44          37 :       MappingTraits<MachOYAML::Object>::mapping(IO, *ObjectFile.MachO);
      45          66 :     } else if (IO.mapTag("!fat-mach-o")) {
      46           1 :       ObjectFile.FatMachO.reset(new MachOYAML::UniversalBinary());
      47           1 :       MappingTraits<MachOYAML::UniversalBinary>::mapping(IO,
      48             :                                                          *ObjectFile.FatMachO);
      49          64 :     } else if (IO.mapTag("!WASM")) {
      50          29 :       ObjectFile.Wasm.reset(new WasmYAML::Object());
      51          29 :       MappingTraits<WasmYAML::Object>::mapping(IO, *ObjectFile.Wasm);
      52             :     } else {
      53             :       Input &In = (Input &)IO;
      54           3 :       std::string Tag = In.getCurrentNode()->getRawTag();
      55           3 :       if (Tag.empty())
      56           4 :         IO.setError("YAML Object File missing document type tag!");
      57             :       else
      58           1 :         IO.setError(
      59           1 :             Twine("YAML Object File unsupported document type tag '") +
      60           2 :             Twine(Tag) + Twine("'!"));
      61             :     }
      62             :   }
      63         632 : }