Update metering data on return

wasmerio · May 16, 2019 · b830f10 · b830f10
1 parent 14fcd78
commit b830f10
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/middleware-common/src/metering.rs b/lib/middleware-common/src/metering.rs
@@ -43,7 +43,8 @@ impl FunctionMiddleware for Metering {
                     | Operator::BrTable { .. }
                     | Operator::BrIf { .. }
                     | Operator::Call { .. }
-                    | Operator::CallIndirect { .. } => {
+                    | Operator::CallIndirect { .. }
+                    | Operator::Return => {
                         sink.push(Event::Internal(InternalEvent::GetInternal(0)));
                         sink.push(Event::WasmOwned(Operator::I64Const {
                             value: self.current_block as i64,