Extend CustomUintFormatter to support enums

Extracted by Pieter Wuille from a comment by Russ Yanofsky, see bitcoin#18317 (comment).
PIVX-Project · Jul 3, 2021 · 7344c1a · 7344c1a
1 parent c4d6228
commit 7344c1a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/serialize.h b/src/serialize.h
@@ -567,14 +567,15 @@ struct CustomUintFormatter
 
     template <typename Stream, typename I> void Unser(Stream& s, I& v)
     {
-        static_assert(std::numeric_limits<I>::max() >= MAX && std::numeric_limits<I>::min() <= 0, "CustomUintFormatter type too small");
+        using U = typename std::conditional<std::is_enum<I>::value, std::underlying_type<I>, std::common_type<I>>::type::type;
+        static_assert(std::numeric_limits<U>::max() >= MAX && std::numeric_limits<U>::min() <= 0, "Assigned type too small");
         uint64_t raw = 0;
         if (BigEndian) {
             s.read(((char*)&raw) + 8 - Bytes, Bytes);
-            v = be64toh(raw);
+            v = static_cast<I>(be64toh(raw));
         } else {
             s.read((char*)&raw, Bytes);
-            v = le64toh(raw);
+            v = static_cast<I>(le64toh(raw));
         }
     }
 };