1//===------ offload2yaml.cpp - obj2yaml conversion tool ---*- C++ -------*-===//
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#include "obj2yaml.h"
10#include "llvm/BinaryFormat/Magic.h"
11#include "llvm/Object/OffloadBinary.h"
12#include "llvm/ObjectYAML/OffloadYAML.h"
13#include "llvm/Support/StringSaver.h"
14
15using namespace llvm;
16
17namespace {
18
19void populateYAML(OffloadYAML::Binary &YAMLBinary,
20 ArrayRef<std::unique_ptr<object::OffloadBinary>> OBinaries,
21 UniqueStringSaver Saver) {
22 for (const auto &OBinaryPtr : OBinaries) {
23 object::OffloadBinary &OB = *OBinaryPtr;
24
25 YAMLBinary.Members.emplace_back();
26 auto &Member = YAMLBinary.Members.back();
27 Member.ImageKind = OB.getImageKind();
28 Member.OffloadKind = OB.getOffloadKind();
29 Member.Flags = OB.getFlags();
30 if (!OB.strings().empty()) {
31 Member.StringEntries = std::vector<OffloadYAML::Binary::StringEntry>();
32 for (const auto &StringEntry : OB.strings())
33 Member.StringEntries->emplace_back(args: OffloadYAML::Binary::StringEntry(
34 {.Key: Saver.save(S: StringEntry.first), .Value: Saver.save(S: StringEntry.second)}));
35 }
36
37 if (!OB.getImage().empty())
38 Member.Content = arrayRefFromStringRef(Input: OB.getImage());
39 }
40}
41
42Expected<OffloadYAML::Binary *> dump(MemoryBufferRef Source,
43 UniqueStringSaver Saver) {
44 std::unique_ptr<OffloadYAML::Binary> YAMLBinary =
45 std::make_unique<OffloadYAML::Binary>();
46
47 YAMLBinary->Members = std::vector<OffloadYAML::Binary::Member>();
48
49 uint64_t Offset = 0;
50 while (Offset < Source.getBufferSize()) {
51 MemoryBufferRef Buffer = MemoryBufferRef(
52 Source.getBuffer().drop_front(N: Offset), Source.getBufferIdentifier());
53 auto BinariesOrErr = object::OffloadBinary::create(Buf: Buffer);
54 if (!BinariesOrErr)
55 return BinariesOrErr.takeError();
56
57 SmallVector<std::unique_ptr<object::OffloadBinary>> &Binaries =
58 *BinariesOrErr;
59 populateYAML(YAMLBinary&: *YAMLBinary, OBinaries: Binaries, Saver);
60
61 Offset += Binaries[0]->getSize();
62 }
63
64 return YAMLBinary.release();
65}
66
67} // namespace
68
69Error offload2yaml(raw_ostream &Out, MemoryBufferRef Source) {
70 BumpPtrAllocator Alloc;
71 UniqueStringSaver Saver(Alloc);
72
73 Expected<OffloadYAML::Binary *> YAMLOrErr = dump(Source, Saver);
74 if (!YAMLOrErr)
75 return YAMLOrErr.takeError();
76
77 std::unique_ptr<OffloadYAML::Binary> YAML(YAMLOrErr.get());
78 yaml::Output Yout(Out);
79 Yout << *YAML;
80
81 return Error::success();
82}
83