Merge pull request #13 from extism/handle-errors

handle allocation errors

Author: mhmd-azeez

Cargo.lock

@@ -4,6 +4,11 @@ use extism_pdk::*;
 pub fn http_get(Json(input): Json<HttpRequest>) -> FnResult<Memory> {
     let res = http::request::<()>(&input, None)?;
     let res = res.to_memory()?;
+
+    if res.offset() == 0 {
+        return Err(WithReturnCode::new(Error::msg("Failed to allocate memory"), 1));
+    }
+
     Ok(res)
 }
 
@@ -18,5 +23,10 @@ struct HttpRequestWithBody {
 pub fn http_post(Json(input): Json<HttpRequestWithBody>) -> FnResult<Memory> {
     let res = http::request::<&str>(&input.req, Some(&input.data))?;
     let res = res.into_memory();
+
+    if res.offset() == 0 {
+        return Err(WithReturnCode::new(Error::msg("Failed to allocate memory"), 1));
+    }
+
     Ok(res)
 }

http/src/lib.rs

@@ -8,15 +8,21 @@ struct AllocRequest {
 
 #[plugin_fn]
 pub unsafe fn alloc_memory(Json(input): Json<AllocRequest>) -> FnResult<()> {
-    _ = extism::alloc(input.bytes);
+    let offs = extism::alloc(input.bytes);
+    if offs == 0 {
+        return Err(WithReturnCode::new(Error::msg("Failed to allocate memory"), 1));
+    }
 
     Ok(())
 }
 
 #[plugin_fn]
 pub unsafe fn alloc_var(Json(input): Json<AllocRequest>) -> FnResult<()> {
     let buffer = vec![0u8; input.bytes as usize];
-    _ = var::set("buffer", buffer);
+
+    if let Err(e) = var::set("buffer", buffer) {
+       return Err(WithReturnCode::from(e));
+    }
 
     Ok(())
 }