diff --git a/src/traits.rs b/src/traits.rs
index 268ef39c..10b24514 100644
--- a/src/traits.rs
+++ b/src/traits.rs
@@ -136,6 +136,10 @@ pub trait StreamContext: Context {
         hostcalls::get_buffer(BufferType::DownstreamData, start, max_size).unwrap()
     }
 
+    fn set_downstream_data(&self, start: usize, size: usize, value: &[u8]) {
+        hostcalls::set_buffer(BufferType::DownstreamData, start, size, value).unwrap()
+    }
+
     fn on_downstream_close(&mut self, _peer_type: PeerType) {}
 
     fn on_upstream_data(&mut self, _data_size: usize, _end_of_stream: bool) -> Action {
@@ -146,6 +150,10 @@ pub trait StreamContext: Context {
         hostcalls::get_buffer(BufferType::UpstreamData, start, max_size).unwrap()
     }
 
+    fn set_upstream_data(&self, start: usize, size: usize, value: &[u8]) {
+        hostcalls::set_buffer(BufferType::UpstreamData, start, size, value).unwrap()
+    }
+
     fn on_upstream_close(&mut self, _peer_type: PeerType) {}
 
     fn on_log(&mut self) {}