Clippy guests

Author: ludfjig

.github/workflows/dep_rust.yml

@@ -48,7 +48,9 @@ jobs:
         run: just fmt-check
 
       - name: clippy
-        run: just clippy ${{ matrix.config }}
+        run: |
+          just clippy ${{ matrix.config }}
+          just clippy-guests
 
       - name: Ensure up-to-date Cargo.lock
         run: |

Justfile

@@ -133,6 +133,10 @@ fmt-apply:
 clippy target=default-target:
     cargo clippy --all-targets --all-features --profile={{ if target == "debug" { "dev" } else { target } }} -- -D warnings
 
+clippy-guests target=default-target:
+    cd src/tests/rust_guests/simpleguest && cargo clippy --profile={{ if target == "debug" { "dev" } else { target } }} -- -D warnings
+    cd src/tests/rust_guests/callbackguest && cargo clippy --profile={{ if target == "debug" { "dev" } else { target } }} -- -D warnings
+
 clippy-apply-fix-unix:
     cargo clippy --fix --all 
 

src/hyperlight_guest/src/guest_function_call.rs

@@ -28,7 +28,7 @@ use crate::shared_input_data::try_pop_shared_input_data_into;
 use crate::shared_output_data::push_shared_output_data;
 use crate::REGISTERED_GUEST_FUNCTIONS;
 
-type GuestFunc = fn(&FunctionCall) -> Result<Vec<u8>>;
+pub(crate) type GuestFunc = fn(&FunctionCall) -> Result<Vec<u8>>;
 
 pub(crate) fn call_guest_function(function_call: FunctionCall) -> Result<Vec<u8>> {
     // Validate this is a Guest Function Call
@@ -56,10 +56,7 @@ pub(crate) fn call_guest_function(function_call: FunctionCall) -> Result<Vec<u8>
         // Verify that the function call has the correct parameter types and length.
         registered_function_definition.verify_parameters(&function_call_parameter_types)?;
 
-        let p_function = unsafe {
-            let function_pointer = registered_function_definition.function_pointer;
-            core::mem::transmute::<i64, GuestFunc>(function_pointer)
-        };
+        let p_function = registered_function_definition.function_pointer;
 
         p_function(&function_call)
     } else {

src/hyperlight_guest/src/guest_function_definition.rs

@@ -22,6 +22,7 @@ use hyperlight_common::flatbuffer_wrappers::function_types::{ParameterType, Retu
 use hyperlight_common::flatbuffer_wrappers::guest_error::ErrorCode;
 
 use crate::error::{HyperlightGuestError, Result};
+use crate::guest_function_call::GuestFunc;
 
 /// The definition of a function exposed from the guest to the host
 #[derive(Debug, Clone, PartialEq, Eq)]
@@ -33,7 +34,7 @@ pub struct GuestFunctionDefinition {
     /// The type of the return value from the host function call
     pub return_type: ReturnType,
     /// The function pointer to the guest function
-    pub function_pointer: i64,
+    pub function_pointer: GuestFunc,
 }
 
 impl GuestFunctionDefinition {
@@ -42,7 +43,7 @@ impl GuestFunctionDefinition {
         function_name: String,
         parameter_types: Vec<ParameterType>,
         return_type: ReturnType,
-        function_pointer: i64,
+        function_pointer: GuestFunc,
     ) -> Self {
         Self {
             function_name,

src/tests/rust_guests/callbackguest/src/main.rs

@@ -62,47 +62,47 @@ fn guest_function(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
         send_message_to_host_method("HostMethod", "Hello from GuestFunction, ", message)
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to guest_function".to_string(),
-        ));
+        ))
     }
 }
 
 fn guest_function1(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
         send_message_to_host_method("HostMethod1", "Hello from GuestFunction1, ", message)
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to guest_function1".to_string(),
-        ));
+        ))
     }
 }
 
 fn guest_function2(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
         send_message_to_host_method("HostMethod1", "Hello from GuestFunction2, ", message)
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to guest_function2".to_string(),
-        ));
+        ))
     }
 }
 
 fn guest_function3(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
         send_message_to_host_method("HostMethod1", "Hello from GuestFunction3, ", message)
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to guest_function3".to_string(),
-        ));
+        ))
     }
 }
 
-fn guest_function4() -> Result<Vec<u8>> {
+fn guest_function4(_: &FunctionCall) -> Result<Vec<u8>> {
     call_host_function(
         "HostMethod4",
         Some(Vec::from(&[ParameterValue::String(
@@ -125,7 +125,7 @@ fn guest_log_message(function_call: &FunctionCall) -> Result<Vec<u8>> {
         &function_call.parameters.as_ref().unwrap()[2],
     ) {
         let mut log_level = *level;
-        if log_level < 0 || log_level > 6 {
+        if !(0..=6).contains(&log_level) {
             log_level = 0;
         }
 
@@ -140,25 +140,25 @@ fn guest_log_message(function_call: &FunctionCall) -> Result<Vec<u8>> {
 
         Ok(get_flatbuffer_result_from_int(message.len() as i32))
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to guest_log_message".to_string(),
-        ));
+        ))
     }
 }
 
 fn call_error_method(function_call: &FunctionCall) -> Result<Vec<u8>> {
     if let ParameterValue::String(message) = &function_call.parameters.as_ref().unwrap()[0] {
         send_message_to_host_method("ErrorMethod", "Error From Host: ", message)
     } else {
-        return Err(HyperlightGuestError::new(
+        Err(HyperlightGuestError::new(
             ErrorCode::GuestFunctionParameterTypeMismatch,
             "Invalid parameters passed to call_error_method".to_string(),
-        ));
+        ))
     }
 }
 
-fn call_host_spin() -> Result<Vec<u8>> {
+fn call_host_spin(_: &FunctionCall) -> Result<Vec<u8>> {
     call_host_function("Spin", None, ReturnType::Void)?;
     Ok(get_flatbuffer_result_from_void())
 }
@@ -169,47 +169,47 @@ pub extern "C" fn hyperlight_main() {
         "PrintOutput".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        print_output_as_guest_function as i64,
+        print_output_as_guest_function,
     );
     register_function(print_output_def);
 
     let guest_function_def = GuestFunctionDefinition::new(
         "GuestMethod".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        guest_function as i64,
+        guest_function,
     );
     register_function(guest_function_def);
 
     let guest_function1_def = GuestFunctionDefinition::new(
         "GuestMethod1".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        guest_function1 as i64,
+        guest_function1,
     );
     register_function(guest_function1_def);
 
     let guest_function2_def = GuestFunctionDefinition::new(
         "GuestMethod2".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        guest_function2 as i64,
+        guest_function2,
     );
     register_function(guest_function2_def);
 
     let guest_function3_def = GuestFunctionDefinition::new(
         "GuestMethod3".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        guest_function3 as i64,
+        guest_function3,
     );
     register_function(guest_function3_def);
 
     let guest_function4_def = GuestFunctionDefinition::new(
         "GuestMethod4".to_string(),
         Vec::new(),
         ReturnType::Int,
-        guest_function4 as i64,
+        guest_function4,
     );
     register_function(guest_function4_def);
 
@@ -221,23 +221,23 @@ pub extern "C" fn hyperlight_main() {
             ParameterType::Int,
         ]),
         ReturnType::Int,
-        guest_log_message as i64,
+        guest_log_message,
     );
     register_function(guest_log_message_def);
 
     let call_error_method_def = GuestFunctionDefinition::new(
         "CallErrorMethod".to_string(),
         Vec::from(&[ParameterType::String]),
         ReturnType::Int,
-        call_error_method as i64,
+        call_error_method,
     );
     register_function(call_error_method_def);
 
     let call_host_spin_def = GuestFunctionDefinition::new(
         "CallHostSpin".to_string(),
         Vec::new(),
         ReturnType::Int,
-        call_host_spin as i64,
+        call_host_spin,
     );
     register_function(call_host_spin_def);
 }