aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/ObjCopy/DXContainer/DXContainerReader.cpp
blob: 61ee94c5cae957251a81d9ae9666743ab279e549 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
//===- DXContainerReader.cpp ----------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "DXContainerReader.h"

namespace llvm {
namespace objcopy {
namespace dxbc {

using namespace object;

Expected<std::unique_ptr<Object>> DXContainerReader::create() const {
  auto Obj = std::make_unique<Object>();
  Obj->Header = DXContainerObj.getHeader();
  for (const SectionRef &Part : DXContainerObj.sections()) {
    DataRefImpl PartDRI = Part.getRawDataRefImpl();
    Expected<StringRef> Name = DXContainerObj.getSectionName(PartDRI);
    if (auto E = Name.takeError())
      return E;
    assert(Name->size() == 4 &&
           "Valid DXIL Part name consists of 4 characters");
    Expected<ArrayRef<uint8_t>> Data =
        DXContainerObj.getSectionContents(PartDRI);
    if (auto E = Data.takeError())
      return E;
    Obj->Parts.push_back({*Name, *Data});
  }
  return std::move(Obj);
}

} // end namespace dxbc
} // end namespace objcopy
} // end namespace llvm