[ELF] Refactor

rui314 · Sep 29, 2021 · a6c75e9 · a6c75e9
1 parent a2c9d0a
commit a6c75e9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/mold.h b/mold.h
@@ -329,7 +329,8 @@ class ConcurrentMap {
 //
 
 class BitVector {
-  struct BitRef {
+  class BitRef {
+  public:
     BitRef(u8 &byte, u8 bitpos) : byte(byte), bitpos(bitpos) {}
 
     BitRef &operator=(bool val) {
@@ -340,10 +341,16 @@ class BitVector {
       return *this;
     }
 
+    BitRef &operator=(const BitRef &other) {
+      *this = (bool)other;
+      return *this;
+    }
+
     operator bool() const {
       return byte & (1 << bitpos);
     }
 
+  private:
     u8 &byte;
     u8 bitpos;
   };