Update ABI to Proxy-Wasm ABI v0.2.1. (#64)

Signed-off-by: Piotr Sikora <piotrsikora@google.com>

Author: PiotrSikora

CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
+## [Unreleased]
+
+### Changed
+
+- Updated ABI to Proxy-Wasm ABI v0.2.1.
+
 ## [0.1.4] - 2021-07-01
 
 ### Added
@@ -53,6 +59,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 - Initial release.
 
 
+[Unreleased]: https://github.com/proxy-wasm/proxy-wasm-rust-sdk/compare/v0.1.4...HEAD
 [0.1.4]: https://github.com/proxy-wasm/proxy-wasm-rust-sdk/compare/v0.1.3...v0.1.4
 [0.1.3]: https://github.com/proxy-wasm/proxy-wasm-rust-sdk/compare/v0.1.2...v0.1.3
 [0.1.2]: https://github.com/proxy-wasm/proxy-wasm-rust-sdk/compare/v0.1.1...v0.1.2

examples/http_auth_random.rs

@@ -26,7 +26,7 @@ pub fn _start() {
 struct HttpAuthRandom;
 
 impl HttpContext for HttpAuthRandom {
-    fn on_http_request_headers(&mut self, _: usize) -> Action {
+    fn on_http_request_headers(&mut self, _: usize, _: bool) -> Action {
         self.dispatch_http_call(
             "httpbin",
             vec![
@@ -42,7 +42,7 @@ impl HttpContext for HttpAuthRandom {
         Action::Pause
     }
 
-    fn on_http_response_headers(&mut self, _: usize) -> Action {
+    fn on_http_response_headers(&mut self, _: usize, _: bool) -> Action {
         self.set_http_response_header("Powered-By", Some("proxy-wasm"));
         Action::Continue
     }

examples/http_body.rs

@@ -40,7 +40,7 @@ struct HttpBody;
 impl Context for HttpBody {}
 
 impl HttpContext for HttpBody {
-    fn on_http_response_headers(&mut self, _: usize) -> Action {
+    fn on_http_response_headers(&mut self, _: usize, _: bool) -> Action {
         // If there is a Content-Length header and we change the length of
         // the body later, then clients will break. So remove it.
         // We must do this here, because once we exit this function we

examples/http_config.rs

@@ -32,7 +32,7 @@ struct HttpConfigHeader {
 impl Context for HttpConfigHeader {}
 
 impl HttpContext for HttpConfigHeader {
-    fn on_http_response_headers(&mut self, _: usize) -> Action {
+    fn on_http_response_headers(&mut self, _: usize, _: bool) -> Action {
         self.add_http_response_header("custom-header", self.header_content.as_str());
         Action::Continue
     }
@@ -46,7 +46,7 @@ impl Context for HttpConfigHeaderRoot {}
 
 impl RootContext for HttpConfigHeaderRoot {
     fn on_configure(&mut self, _: usize) -> bool {
-        if let Some(config_bytes) = self.get_configuration() {
+        if let Some(config_bytes) = self.get_plugin_configuration() {
             self.header_content = String::from_utf8(config_bytes).unwrap()
         }
         true

examples/http_headers.rs

@@ -43,7 +43,7 @@ struct HttpHeaders {
 impl Context for HttpHeaders {}
 
 impl HttpContext for HttpHeaders {
-    fn on_http_request_headers(&mut self, _: usize) -> Action {
+    fn on_http_request_headers(&mut self, _: usize, _: bool) -> Action {
         for (name, value) in &self.get_http_request_headers() {
             trace!("#{} -> {}: {}", self.context_id, name, value);
         }
@@ -61,7 +61,7 @@ impl HttpContext for HttpHeaders {
         }
     }
 
-    fn on_http_response_headers(&mut self, _: usize) -> Action {
+    fn on_http_response_headers(&mut self, _: usize, _: bool) -> Action {
         for (name, value) in &self.get_http_response_headers() {
             trace!("#{} <- {}: {}", self.context_id, name, value);
         }

src/dispatcher.rs

@@ -324,10 +324,15 @@ impl Dispatcher {
         }
     }
 
-    fn on_http_request_headers(&self, context_id: u32, num_headers: usize) -> Action {
+    fn on_http_request_headers(
+        &self,
+        context_id: u32,
+        num_headers: usize,
+        end_of_stream: bool,
+    ) -> Action {
         if let Some(http_stream) = self.http_streams.borrow_mut().get_mut(&context_id) {
             self.active_id.set(context_id);
-            http_stream.on_http_request_headers(num_headers)
+            http_stream.on_http_request_headers(num_headers, end_of_stream)
         } else {
             panic!("invalid context_id")
         }
@@ -356,10 +361,15 @@ impl Dispatcher {
         }
     }
 
-    fn on_http_response_headers(&self, context_id: u32, num_headers: usize) -> Action {
+    fn on_http_response_headers(
+        &self,
+        context_id: u32,
+        num_headers: usize,
+        end_of_stream: bool,
+    ) -> Action {
         if let Some(http_stream) = self.http_streams.borrow_mut().get_mut(&context_id) {
             self.active_id.set(context_id);
-            http_stream.on_http_response_headers(num_headers)
+            http_stream.on_http_response_headers(num_headers, end_of_stream)
         } else {
             panic!("invalid context_id")
         }
@@ -605,8 +615,14 @@ pub extern "C" fn proxy_on_upstream_connection_close(context_id: u32, peer_type:
 }
 
 #[no_mangle]
-pub extern "C" fn proxy_on_request_headers(context_id: u32, num_headers: usize) -> Action {
-    DISPATCHER.with(|dispatcher| dispatcher.on_http_request_headers(context_id, num_headers))
+pub extern "C" fn proxy_on_request_headers(
+    context_id: u32,
+    num_headers: usize,
+    end_of_stream: bool,
+) -> Action {
+    DISPATCHER.with(|dispatcher| {
+        dispatcher.on_http_request_headers(context_id, num_headers, end_of_stream)
+    })
 }
 
 #[no_mangle]
@@ -625,8 +641,14 @@ pub extern "C" fn proxy_on_request_trailers(context_id: u32, num_trailers: usize
 }
 
 #[no_mangle]
-pub extern "C" fn proxy_on_response_headers(context_id: u32, num_headers: usize) -> Action {
-    DISPATCHER.with(|dispatcher| dispatcher.on_http_response_headers(context_id, num_headers))
+pub extern "C" fn proxy_on_response_headers(
+    context_id: u32,
+    num_headers: usize,
+    end_of_stream: bool,
+) -> Action {
+    DISPATCHER.with(|dispatcher| {
+        dispatcher.on_http_response_headers(context_id, num_headers, end_of_stream)
+    })
 }
 
 #[no_mangle]

src/hostcalls.rs

@@ -31,55 +31,41 @@ pub fn log(level: LogLevel, message: &str) -> Result<(), Status> {
 }
 
 extern "C" {
-    fn proxy_get_current_time_nanoseconds(return_time: *mut u64) -> Status;
+    fn proxy_get_log_level(return_level: *mut LogLevel) -> Status;
 }
 
-pub fn get_current_time() -> Result<SystemTime, Status> {
-    let mut return_time: u64 = 0;
+pub fn get_log_level() -> Result<LogLevel, Status> {
+    let mut return_level: LogLevel = LogLevel::Trace;
     unsafe {
-        match proxy_get_current_time_nanoseconds(&mut return_time) {
-            Status::Ok => Ok(UNIX_EPOCH + Duration::from_nanos(return_time)),
+        match proxy_get_log_level(&mut return_level) {
+            Status::Ok => Ok(return_level),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
 }
 
 extern "C" {
-    fn proxy_set_tick_period_milliseconds(period: u32) -> Status;
+    fn proxy_get_current_time_nanoseconds(return_time: *mut u64) -> Status;
 }
 
-pub fn set_tick_period(period: Duration) -> Result<(), Status> {
+pub fn get_current_time() -> Result<SystemTime, Status> {
+    let mut return_time: u64 = 0;
     unsafe {
-        match proxy_set_tick_period_milliseconds(period.as_millis() as u32) {
-            Status::Ok => Ok(()),
+        match proxy_get_current_time_nanoseconds(&mut return_time) {
+            Status::Ok => Ok(UNIX_EPOCH + Duration::from_nanos(return_time)),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
 }
 
 extern "C" {
-    fn proxy_get_configuration(
-        return_buffer_data: *mut *mut u8,
-        return_buffer_size: *mut usize,
-    ) -> Status;
+    fn proxy_set_tick_period_milliseconds(period: u32) -> Status;
 }
 
-pub fn get_configuration() -> Result<Option<Bytes>, Status> {
-    let mut return_data: *mut u8 = null_mut();
-    let mut return_size: usize = 0;
+pub fn set_tick_period(period: Duration) -> Result<(), Status> {
     unsafe {
-        match proxy_get_configuration(&mut return_data, &mut return_size) {
-            Status::Ok => {
-                if !return_data.is_null() {
-                    Ok(Some(Vec::from_raw_parts(
-                        return_data,
-                        return_size,
-                        return_size,
-                    )))
-                } else {
-                    Ok(None)
-                }
-            }
+        match proxy_set_tick_period_milliseconds(period.as_millis() as u32) {
+            Status::Ok => Ok(()),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
@@ -258,6 +244,7 @@ pub fn get_map_value(map_type: MapType, key: &str) -> Result<Option<String>, Sta
                     Ok(None)
                 }
             }
+            Status::NotFound => Ok(None),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
@@ -285,6 +272,7 @@ pub fn get_map_value_bytes(map_type: MapType, key: &str) -> Result<Option<Bytes>
                     Ok(None)
                 }
             }
+            Status::NotFound => Ok(None),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
@@ -624,25 +612,78 @@ pub fn enqueue_shared_queue(queue_id: u32, value: Option<&[u8]>) -> Result<(), S
 }
 
 extern "C" {
-    fn proxy_continue_request() -> Status;
+    fn proxy_continue_stream(stream_type: StreamType) -> Status;
+}
+
+pub fn resume_downstream() -> Result<(), Status> {
+    unsafe {
+        match proxy_continue_stream(StreamType::Downstream) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
+}
+
+pub fn resume_upstream() -> Result<(), Status> {
+    unsafe {
+        match proxy_continue_stream(StreamType::Upstream) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
 }
 
 pub fn resume_http_request() -> Result<(), Status> {
     unsafe {
-        match proxy_continue_request() {
+        match proxy_continue_stream(StreamType::HttpRequest) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
+}
+
+pub fn resume_http_response() -> Result<(), Status> {
+    unsafe {
+        match proxy_continue_stream(StreamType::HttpResponse) {
             Status::Ok => Ok(()),
             status => panic!("unexpected status: {}", status as u32),
         }
     }
 }
 
 extern "C" {
-    fn proxy_continue_response() -> Status;
+    fn proxy_close_stream(stream_type: StreamType) -> Status;
 }
 
-pub fn resume_http_response() -> Result<(), Status> {
+pub fn close_downstream() -> Result<(), Status> {
+    unsafe {
+        match proxy_close_stream(StreamType::Downstream) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
+}
+pub fn close_upstream() -> Result<(), Status> {
+    unsafe {
+        match proxy_close_stream(StreamType::Upstream) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
+}
+
+pub fn reset_http_request() -> Result<(), Status> {
+    unsafe {
+        match proxy_close_stream(StreamType::HttpRequest) {
+            Status::Ok => Ok(()),
+            status => panic!("unexpected status: {}", status as u32),
+        }
+    }
+}
+
+pub fn reset_http_response() -> Result<(), Status> {
     unsafe {
-        match proxy_continue_response() {
+        match proxy_close_stream(StreamType::HttpResponse) {
             Status::Ok => Ok(()),
             status => panic!("unexpected status: {}", status as u32),
         }
@@ -685,19 +726,6 @@ pub fn send_http_response(
     }
 }
 
-extern "C" {
-    fn proxy_clear_route_cache() -> Status;
-}
-
-pub fn clear_http_route_cache() -> Result<(), Status> {
-    unsafe {
-        match proxy_clear_route_cache() {
-            Status::Ok => Ok(()),
-            status => panic!("unexpected status: {}", status as u32),
-        }
-    }
-}
-
 extern "C" {
     fn proxy_http_call(
         upstream_data: *const u8,

src/lib.rs

@@ -37,4 +37,4 @@ pub fn set_http_context(callback: types::NewHttpContext) {
 }
 
 #[no_mangle]
-pub extern "C" fn proxy_abi_version_0_1_0() {}
+pub extern "C" fn proxy_abi_version_0_2_1() {}