Added shr & srl implementations for addresses

Author: jounathaen

src/lib.rs

@@ -5,8 +5,8 @@
 use core::fmt;
 use core::fmt::Debug;
 use core::ops::{
-    Add, AddAssign, BitAnd, BitAndAssign, BitOr, BitOrAssign, BitXor, BitXorAssign, Not, Sub,
-    SubAssign,
+    Add, AddAssign, BitAnd, BitAndAssign, BitOr, BitOrAssign, BitXor, BitXorAssign, Not, Shl,
+    ShlAssign, Shr, ShrAssign, Sub, SubAssign,
 };
 
 pub mod arch;
@@ -48,6 +48,10 @@ pub trait MemoryAddress:
     + Sub<Self, Output = Self::RAW>
     + Sub<<Self>::RAW, Output = Self>
     + SubAssign<<Self>::RAW>
+    + Shr<usize, Output = Self>
+    + ShrAssign<usize>
+    + Shl<usize, Output = Self>
+    + ShlAssign<usize>
     + fmt::Binary
     + fmt::LowerHex
     + fmt::UpperHex

src/macros.rs

@@ -131,6 +131,36 @@ macro_rules! impl_address {
             }
         }
 
+        impl core::ops::Shl<usize> for $T {
+            type Output = $T;
+            #[inline]
+            fn shl(self, rhs: usize) -> Self::Output {
+                Self::new_truncate(self.0 << rhs)
+            }
+        }
+
+        impl core::ops::Shr<usize> for $T {
+            type Output = $T;
+            #[inline]
+            fn shr(self, rhs: usize) -> Self::Output {
+                Self::new_truncate(self.0 >> rhs)
+            }
+        }
+
+        impl core::ops::ShlAssign<usize> for $T {
+            #[inline]
+            fn shl_assign(&mut self, rhs: usize) {
+                *self = Self::new_truncate(self.0 << rhs);
+            }
+        }
+
+        impl core::ops::ShrAssign<usize> for $T {
+            #[inline]
+            fn shr_assign(&mut self, rhs: usize) {
+                *self = Self::new_truncate(self.0 >> rhs);
+            }
+        }
+
         impl From<$V> for $T {
             #[inline]
             fn from(addr: $V) -> $T {