[SampleProfileLoader] Fix integer overflow in generateMDProfMetadata (#90217)
[llvm-project.git] / llvm / lib / ObjectYAML / ArchiveYAML.cpp
blobd2ea1eaf5210540b9b01d71a8fc3f99acd04e175
1 //===- ArchiveYAML.cpp - ELF YAMLIO implementation -------------------- ----===//
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 classes for handling the YAML representation of archives.
11 //===----------------------------------------------------------------------===//
13 #include "llvm/ObjectYAML/ArchiveYAML.h"
15 namespace llvm {
17 namespace yaml {
19 void MappingTraits<ArchYAML::Archive>::mapping(IO &IO, ArchYAML::Archive &A) {
20 assert(!IO.getContext() && "The IO context is initialized already");
21 IO.setContext(&A);
22 IO.mapTag("!Arch", true);
23 IO.mapOptional("Magic", A.Magic, "!<arch>\n");
24 IO.mapOptional("Members", A.Members);
25 IO.mapOptional("Content", A.Content);
26 IO.setContext(nullptr);
29 std::string MappingTraits<ArchYAML::Archive>::validate(IO &,
30 ArchYAML::Archive &A) {
31 if (A.Members && A.Content)
32 return "\"Content\" and \"Members\" cannot be used together";
33 return "";
36 void MappingTraits<ArchYAML::Archive::Child>::mapping(
37 IO &IO, ArchYAML::Archive::Child &E) {
38 assert(IO.getContext() && "The IO context is not initialized");
39 for (auto &P : E.Fields)
40 IO.mapOptional(P.first.data(), P.second.Value, P.second.DefaultValue);
41 IO.mapOptional("Content", E.Content);
42 IO.mapOptional("PaddingByte", E.PaddingByte);
45 std::string
46 MappingTraits<ArchYAML::Archive::Child>::validate(IO &,
47 ArchYAML::Archive::Child &C) {
48 for (auto &P : C.Fields)
49 if (P.second.Value.size() > P.second.MaxLength)
50 return ("the maximum length of \"" + P.first + "\" field is " +
51 Twine(P.second.MaxLength))
52 .str();
53 return "";
56 } // end namespace yaml
58 } // end namespace llvm