llvm/lib/DebugInfo/GSYM/Range.cpp

   1 //===- Range.cpp ------------------------------------------------*- 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 "llvm/DebugInfo/GSYM/Range.h"
  10 #include "llvm/DebugInfo/GSYM/FileWriter.h"
  11 #include "llvm/Support/DataExtractor.h"
  12 #include <algorithm>
  13 #include <inttypes.h>
  14
  15 using namespace llvm;
  16 using namespace gsym;
  17
  18
  19 void AddressRanges::insert(AddressRange Range) {
  20   if (Range.size() == 0)
  21     return;
  22
  23   auto It = llvm::upper_bound(Ranges, Range);
  24   auto It2 = It;
  25   while (It2 != Ranges.end() && It2->Start < Range.End)
  26     ++It2;
  27   if (It != It2) {
  28     Range.End = std::max(Range.End, It2[-1].End);
  29     It = Ranges.erase(It, It2);
  30   }
  31   if (It != Ranges.begin() && Range.Start < It[-1].End)
  32     It[-1].End = std::max(It[-1].End, Range.End);
  33   else
  34     Ranges.insert(It, Range);
  35 }
  36
  37 bool AddressRanges::contains(uint64_t Addr) const {
  38   auto It = std::partition_point(
  39       Ranges.begin(), Ranges.end(),
  40       [=](const AddressRange &R) { return R.Start <= Addr; });
  41   return It != Ranges.begin() && Addr < It[-1].End;
  42 }
  43
  44 bool AddressRanges::contains(AddressRange Range) const {
  45   if (Range.size() == 0)
  46     return false;
  47   auto It = std::partition_point(
  48       Ranges.begin(), Ranges.end(),
  49       [=](const AddressRange &R) { return R.Start <= Range.Start; });
  50   if (It == Ranges.begin())
  51     return false;
  52   return Range.End <= It[-1].End;
  53 }
  54
  55 Optional<AddressRange>
  56 AddressRanges::getRangeThatContains(uint64_t Addr) const {
  57   auto It = std::partition_point(
  58       Ranges.begin(), Ranges.end(),
  59       [=](const AddressRange &R) { return R.Start <= Addr; });
  60   if (It != Ranges.begin() && Addr < It[-1].End)
  61     return It[-1];
  62   return llvm::None;
  63 }
  64
  65 raw_ostream &llvm::gsym::operator<<(raw_ostream &OS, const AddressRange &R) {
  66   return OS << '[' << HEX64(R.Start) << " - " << HEX64(R.End) << ")";
  67 }
  68
  69 raw_ostream &llvm::gsym::operator<<(raw_ostream &OS, const AddressRanges &AR) {
  70   size_t Size = AR.size();
  71   for (size_t I = 0; I < Size; ++I) {
  72     if (I)
  73       OS << ' ';
  74     OS << AR[I];
  75   }
  76   return OS;
  77 }
  78
  79 void AddressRange::encode(FileWriter &O, uint64_t BaseAddr) const {
  80   assert(Start >= BaseAddr);
  81   O.writeULEB(Start - BaseAddr);
  82   O.writeULEB(size());
  83 }
  84
  85 void AddressRange::decode(DataExtractor &Data, uint64_t BaseAddr,
  86                           uint64_t &Offset) {
  87   const uint64_t AddrOffset = Data.getULEB128(&Offset);
  88   const uint64_t Size = Data.getULEB128(&Offset);
  89   const uint64_t StartAddr = BaseAddr + AddrOffset;
  90   Start = StartAddr;
  91   End = StartAddr + Size;
  92 }
  93
  94 void AddressRanges::encode(FileWriter &O, uint64_t BaseAddr) const {
  95   O.writeULEB(Ranges.size());
  96   if (Ranges.empty())
  97     return;
  98   for (auto Range : Ranges)
  99     Range.encode(O, BaseAddr);
 100 }
 101
 102 void AddressRanges::decode(DataExtractor &Data, uint64_t BaseAddr,
 103                            uint64_t &Offset) {
 104   clear();
 105   uint64_t NumRanges = Data.getULEB128(&Offset);
 106   if (NumRanges == 0)
 107     return;
 108   Ranges.resize(NumRanges);
 109   for (auto &Range : Ranges)
 110     Range.decode(Data, BaseAddr, Offset);
 111 }
 112
 113 void AddressRange::skip(DataExtractor &Data, uint64_t &Offset) {
 114   Data.getULEB128(&Offset);
 115   Data.getULEB128(&Offset);
 116 }
 117
 118 uint64_t AddressRanges::skip(DataExtractor &Data, uint64_t &Offset) {
 119   uint64_t NumRanges = Data.getULEB128(&Offset);
 120   for (uint64_t I=0; I<NumRanges; ++I)
 121     AddressRange::skip(Data, Offset);
 122   return NumRanges;
 123 }