commit 1a101d30728d7de12e6b2976914c8cc5a8bfe7d8
author Weining Lu <luweining@loongson.cn> Tue Mar 05 23:19:16 2024 +0800
committer Copybara-Service <copybara-worker@google.com> Tue Mar 05 07:42:53 2024 -0800
tree e4e46e99e22dac4cf8b0318d9996045f6fec9730
parent c7f9f7972edabc2200626be0d1c8641f2c8a839e

[lld][LoongArch] Fix handleUleb128

GitOrigin-RevId: a41bcb3930534ef1525b4fc30e53e818b39e2b60

ELF/Arch/LoongArch.cpp

diff --git a/ELF/Arch/LoongArch.cpp b/ELF/Arch/LoongArch.cpp
index 5f0b73d..2c5d5df 100644
--- a/ELF/Arch/LoongArch.cpp
+++ b/ELF/Arch/LoongArch.cpp
@@ -159,8 +159,9 @@
 static void handleUleb128(uint8_t *loc, uint64_t val) {
   const uint32_t maxcount = 1 + 64 / 7;
   uint32_t count;
-  uint64_t orig = decodeULEB128(loc, &count);
-  if (count > maxcount)
+  const char *error = nullptr;
+  uint64_t orig = decodeULEB128(loc, &count, nullptr, &error);
+  if (count > maxcount || (count == maxcount && error))
     errorOrWarn(getErrorLocation(loc) + "extra space for uleb128");
   uint64_t mask = count < maxcount ? (1ULL << 7 * count) - 1 : -1ULL;
   encodeULEB128((orig + val) & mask, loc, count);