Add validate_region

Author: webmaster128

packages/vm/src/errors.rs

@@ -245,6 +245,26 @@ pub enum CommunicationError {
         max_length: usize,
         backtrace: snafu::Backtrace,
     },
+    #[snafu(display(
+        "Region length exceeds capacity. Length {}, capacity {}",
+        length,
+        capacity
+    ))]
+    RegionLengthExceedsCapacity {
+        length: u32,
+        capacity: u32,
+        backtrace: snafu::Backtrace,
+    },
+    #[snafu(display(
+        "Region exceeds address space. Offset {}, capacity {}",
+        offset,
+        capacity
+    ))]
+    RegionOutOfRange {
+        offset: u32,
+        capacity: u32,
+        backtrace: snafu::Backtrace,
+    },
     #[snafu(display("Region too small. Got {}, required {}", size, required))]
     RegionTooSmall {
         size: usize,
@@ -281,6 +301,14 @@ impl CommunicationError {
         RegionLengthTooBig { length, max_length }.build()
     }
 
+    pub(crate) fn region_length_exceeds_capacity(length: u32, capacity: u32) -> Self {
+        RegionLengthExceedsCapacity { length, capacity }.build()
+    }
+
+    pub(crate) fn region_out_of_range(offset: u32, capacity: u32) -> Self {
+        RegionOutOfRange { offset, capacity }.build()
+    }
+
     pub(crate) fn region_too_small(size: usize, required: usize) -> Self {
         RegionTooSmall { size, required }.build()
     }
@@ -547,6 +575,34 @@ mod test {
         }
     }
 
+    #[test]
+    fn communication_error_region_length_exceeds_capacity_works() {
+        let error = CommunicationError::region_length_exceeds_capacity(50, 20);
+        match error {
+            CommunicationError::RegionLengthExceedsCapacity {
+                length, capacity, ..
+            } => {
+                assert_eq!(length, 50);
+                assert_eq!(capacity, 20);
+            }
+            e => panic!("Unexpected error: {:?}", e),
+        }
+    }
+
+    #[test]
+    fn communication_error_region_out_of_range_works() {
+        let error = CommunicationError::region_out_of_range(u32::MAX, 1);
+        match error {
+            CommunicationError::RegionOutOfRange {
+                offset, capacity, ..
+            } => {
+                assert_eq!(offset, u32::MAX);
+                assert_eq!(capacity, 1);
+            }
+            e => panic!("Unexpected error: {:?}", e),
+        }
+    }
+
     #[test]
     fn communication_error_region_too_small_works() {
         let error = CommunicationError::region_too_small(12, 33);

packages/vm/src/memory.rs

@@ -145,6 +145,27 @@ fn get_region(ctx: &Ctx, ptr: u32) -> CommunicationResult<Region> {
     }
 }
 
+/// Performs plausibility checks in the given Region. Regions are always created by the
+/// contract and this can be used to detect problems in the standard library of the contract.
+fn validate_region(region: &Region) -> CommunicationResult<()> {
+    if region.offset == 0 {
+        return Err(CommunicationError::zero_address());
+    }
+    if region.length > region.capacity {
+        return Err(CommunicationError::region_length_exceeds_capacity(
+            region.length,
+            region.capacity,
+        ));
+    }
+    if region.capacity > (u32::MAX - region.offset) {
+        return Err(CommunicationError::region_out_of_range(
+            region.offset,
+            region.capacity,
+        ));
+    }
+    Ok(())
+}
+
 /// Overrides a Region at ptr in wasm memory with data
 fn set_region(ctx: &Ctx, ptr: u32, data: Region) -> CommunicationResult<()> {
     let memory = ctx.memory(0);
@@ -161,3 +182,119 @@ fn set_region(ctx: &Ctx, ptr: u32, data: Region) -> CommunicationResult<()> {
         )),
     }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn validate_region_passes_for_valid_region() {
+        // empty
+        let region = Region {
+            offset: 23,
+            capacity: 500,
+            length: 0,
+        };
+        validate_region(&region).unwrap();
+
+        // half full
+        let region = Region {
+            offset: 23,
+            capacity: 500,
+            length: 250,
+        };
+        validate_region(&region).unwrap();
+
+        // full
+        let region = Region {
+            offset: 23,
+            capacity: 500,
+            length: 500,
+        };
+        validate_region(&region).unwrap();
+
+        // at end of linear memory (1)
+        let region = Region {
+            offset: u32::MAX,
+            capacity: 0,
+            length: 0,
+        };
+        validate_region(&region).unwrap();
+
+        // at end of linear memory (2)
+        let region = Region {
+            offset: 1,
+            capacity: u32::MAX - 1,
+            length: 0,
+        };
+        validate_region(&region).unwrap();
+    }
+
+    #[test]
+    fn validate_region_fails_for_zero_offset() {
+        let region = Region {
+            offset: 0,
+            capacity: 500,
+            length: 250,
+        };
+        let result = validate_region(&region);
+        match result.unwrap_err() {
+            CommunicationError::ZeroAddress { .. } => {}
+            e => panic!("Got unexpected error: {:?}", e),
+        }
+    }
+
+    #[test]
+    fn validate_region_fails_for_length_exceeding_capacity() {
+        let region = Region {
+            offset: 23,
+            capacity: 500,
+            length: 501,
+        };
+        let result = validate_region(&region);
+        match result.unwrap_err() {
+            CommunicationError::RegionLengthExceedsCapacity {
+                length, capacity, ..
+            } => {
+                assert_eq!(length, 501);
+                assert_eq!(capacity, 500);
+            }
+            e => panic!("Got unexpected error: {:?}", e),
+        }
+    }
+
+    #[test]
+    fn validate_region_fails_when_exceeding_address_space() {
+        let region = Region {
+            offset: 23,
+            capacity: u32::MAX,
+            length: 501,
+        };
+        let result = validate_region(&region);
+        match result.unwrap_err() {
+            CommunicationError::RegionOutOfRange {
+                offset, capacity, ..
+            } => {
+                assert_eq!(offset, 23);
+                assert_eq!(capacity, u32::MAX);
+            }
+            e => panic!("Got unexpected error: {:?}", e),
+        }
+
+        let region = Region {
+            offset: u32::MAX,
+            capacity: 1,
+            length: 0,
+        };
+        let result = validate_region(&region);
+        match result.unwrap_err() {
+            CommunicationError::RegionOutOfRange {
+                offset, capacity, ..
+            } => {
+                assert_eq!(offset, u32::MAX);
+                assert_eq!(capacity, 1);
+            }
+            e => panic!("Got unexpected error: {:?}", e),
+        }
+    }
+}