fix(wiggle_abi): write the result's length, not the guest buffer's

fastly · Jun 13, 2024 · b40369b · b40369b
1 parent e95fbbf
commit b40369b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/src/wiggle_abi/device_detection_impl.rs b/lib/src/wiggle_abi/device_detection_impl.rs
@@ -44,7 +44,7 @@ impl FastlyDeviceDetection for Session {
         };
 
         if result.len() > buf_len as usize {
-            nwritten_out.write(buf_len)?;
+            nwritten_out.write(result.len())?;
             return Err(Error::BufferLengthError {
                 buf: "device_detection_lookup",
                 len: "device_detection_lookup_max_len",

diff --git a/lib/src/wiggle_abi/geo_impl.rs b/lib/src/wiggle_abi/geo_impl.rs
@@ -39,7 +39,7 @@ impl FastlyGeo for Session {
         let result = self.geolocation_lookup(&ip_addr).unwrap_or_default();
 
         if result.len() > buf_len as usize {
-            nwritten_out.write(buf_len)?;
+            nwritten_out.write(result.len())?;
             return Err(Error::BufferLengthError {
                 buf: "geolocation_lookup",
                 len: "geolocation_lookup_max_len",