fix: tracing::instrument

Adds function instrumentation.

Signed-off-by: Jonathan Woollett-Light <jcawl@amazon.co.uk>

Author: Jonathan Woollett-Light

src/api_server/src/lib.rs

@@ -63,6 +63,7 @@ impl ApiServer {
     /// Constructor for `ApiServer`.
     ///
     /// Returns the newly formed `ApiServer`.
+    #[tracing::instrument(level = "trace", ret)]
     pub fn new(
         api_request_sender: mpsc::Sender<ApiRequest>,
         vmm_response_receiver: mpsc::Receiver<ApiResponse>,
@@ -147,6 +148,7 @@ impl ApiServer {
     /// let mut buf: [u8; 100] = [0; 100];
     /// assert!(sock.read(&mut buf[..]).unwrap() > 0);
     /// ```
+    #[tracing::instrument(level = "trace", ret)]
     pub fn bind_and_run(
         &mut self,
         path: &PathBuf,
@@ -223,6 +225,7 @@ impl ApiServer {
     }
 
     /// Handles an API request received through the associated socket.
+    #[tracing::instrument(level = "trace", ret)]
     pub fn handle_request(
         &mut self,
         request: &Request,
@@ -252,6 +255,7 @@ impl ApiServer {
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     fn serve_vmm_action_request(
         &mut self,
         vmm_action: Box<VmmAction>,
@@ -294,12 +298,14 @@ impl ApiServer {
     }
 
     /// An HTTP response which also includes a body.
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn json_response<T: Into<String> + Debug>(status: StatusCode, body: T) -> Response {
         let mut response = Response::new(Version::Http11, status);
         response.set_body(Body::new(body.into()));
         response
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     fn json_fault_message<T: AsRef<str> + serde::Serialize + Debug>(msg: T) -> String {
         json!({ "fault_message": msg }).to_string()
     }

src/api_server/src/parsed_request.rs

@@ -42,13 +42,15 @@ pub(crate) struct ParsingInfo {
 }
 
 impl ParsingInfo {
+    #[tracing::instrument(level = "trace", ret)]
     pub fn append_deprecation_message(&mut self, message: &str) {
         match self.deprecation_message.as_mut() {
             None => self.deprecation_message = Some(message.to_owned()),
             Some(s) => (*s).push_str(message),
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub fn take_deprecation_message(&mut self) -> Option<String> {
         self.deprecation_message.take()
     }
@@ -122,21 +124,25 @@ impl TryFrom<&Request> for ParsedRequest {
 }
 
 impl ParsedRequest {
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn new(action: RequestAction) -> Self {
         Self {
             action,
             parsing_info: Default::default(),
         }
     }
-
+    
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn into_parts(self) -> (RequestAction, ParsingInfo) {
         (self.action, self.parsing_info)
     }
 
+    #[tracing::instrument(level = "trace")]
     pub(crate) fn parsing_info(&mut self) -> &mut ParsingInfo {
         &mut self.parsing_info
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn success_response_with_data<T>(body_data: &T) -> Response
     where
         T: ?Sized + Serialize + Debug,
@@ -147,6 +153,7 @@ impl ParsedRequest {
         response
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn success_response_with_mmds_value(body_data: &Value) -> Response {
         info!("The request was executed successfully. Status code: 200 OK.");
         let mut response = Response::new(Version::Http11, StatusCode::OK);
@@ -158,6 +165,7 @@ impl ParsedRequest {
         response
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn convert_to_response(
         request_outcome: &std::result::Result<VmmData, VmmActionError>,
     ) -> Response {
@@ -207,6 +215,7 @@ impl ParsedRequest {
     }
 
     /// Helper function to avoid boiler-plate code.
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn new_sync(vmm_action: VmmAction) -> ParsedRequest {
         ParsedRequest::new(RequestAction::Sync(Box::new(vmm_action)))
     }
@@ -216,6 +225,7 @@ impl ParsedRequest {
 ///
 /// The `info` macro is used for logging.
 #[inline]
+#[tracing::instrument(level = "trace", ret)]
 fn log_received_api_request(api_description: String) {
     info!("The API server received a {}.", api_description);
 }
@@ -227,6 +237,7 @@ fn log_received_api_request(api_description: String) {
 /// * `method` - one of `GET`, `PATCH`, `PUT`
 /// * `path` - path of the API request
 /// * `body` - body of the API request
+#[tracing::instrument(level = "trace", ret)]
 fn describe(method: Method, path: &str, body: Option<&Body>) -> String {
     match (path, body) {
         ("/mmds", Some(_)) | (_, None) => format!("{:?} request on {:?}", method, path),
@@ -242,6 +253,7 @@ fn describe(method: Method, path: &str, body: Option<&Body>) -> String {
 }
 
 /// Generates a `GenericError` for each request method.
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn method_to_error(method: Method) -> Result<ParsedRequest, Error> {
     match method {
         Method::Get => Err(Error::Generic(
@@ -280,6 +292,7 @@ pub(crate) enum Error {
 
 // It's convenient to turn errors into HTTP responses directly.
 impl From<Error> for Response {
+    #[tracing::instrument(level = "trace", ret)]
     fn from(err: Error) -> Self {
         let msg = ApiServer::json_fault_message(format!("{}", err));
         match err {
@@ -293,6 +306,7 @@ impl From<Error> for Response {
 }
 
 // This function is supposed to do id validation for requests.
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn checked_id(id: &str) -> Result<&str, Error> {
     // todo: are there any checks we want to do on id's?
     // not allow them to be empty strings maybe?
@@ -325,6 +339,7 @@ pub mod tests {
     use super::*;
 
     impl PartialEq for ParsedRequest {
+        #[tracing::instrument(level = "trace", ret)]
         fn eq(&self, other: &ParsedRequest) -> bool {
             if self.parsing_info.deprecation_message != other.parsing_info.deprecation_message {
                 return false;
@@ -339,13 +354,15 @@ pub mod tests {
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn vmm_action_from_request(req: ParsedRequest) -> VmmAction {
         match req.action {
             RequestAction::Sync(vmm_action) => *vmm_action,
             _ => panic!("Invalid request"),
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     pub(crate) fn depr_action_from_req(req: ParsedRequest, msg: Option<String>) -> VmmAction {
         let (action_req, mut parsing_info) = req.into_parts();
         match action_req {
@@ -359,6 +376,7 @@ pub mod tests {
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     fn http_response(body: &str, status_code: i32) -> String {
         let header = format!(
             "HTTP/1.1 {} \r\nServer: Firecracker API\r\nConnection: keep-alive\r\n",
@@ -378,6 +396,7 @@ pub mod tests {
         }
     }
 
+    #[tracing::instrument(level = "trace", ret)]
     fn http_request(request_type: &str, endpoint: &str, body: Option<&str>) -> String {
         let req_no_body = format!(
             "{} {} HTTP/1.1\r\nContent-Type: application/json\r\n",

src/api_server/src/request/actions.rs

@@ -28,6 +28,7 @@ struct ActionBody {
     action_type: ActionType,
 }
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_actions(body: &Body) -> Result<ParsedRequest, Error> {
     METRICS.put_api_requests.actions_count.inc();
     let action_body = serde_json::from_slice::<ActionBody>(body.raw()).map_err(|err| {

src/api_server/src/request/balloon.rs

@@ -10,6 +10,7 @@ use super::super::VmmAction;
 use crate::parsed_request::{Error, ParsedRequest};
 use crate::request::Body;
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_get_balloon(path_second_token: Option<&str>) -> Result<ParsedRequest, Error> {
     match path_second_token {
         Some(stats_path) => match stats_path {
@@ -23,12 +24,14 @@ pub(crate) fn parse_get_balloon(path_second_token: Option<&str>) -> Result<Parse
     }
 }
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_balloon(body: &Body) -> Result<ParsedRequest, Error> {
     Ok(ParsedRequest::new_sync(VmmAction::SetBalloonDevice(
         serde_json::from_slice::<BalloonDeviceConfig>(body.raw())?,
     )))
 }
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_patch_balloon(
     body: &Body,
     path_second_token: Option<&str>,

src/api_server/src/request/boot_source.rs

@@ -8,6 +8,7 @@ use super::super::VmmAction;
 use crate::parsed_request::{Error, ParsedRequest};
 use crate::request::Body;
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_boot_source(body: &Body) -> Result<ParsedRequest, Error> {
     METRICS.put_api_requests.boot_source_count.inc();
     Ok(ParsedRequest::new_sync(VmmAction::ConfigureBootSource(

src/api_server/src/request/cpu_configuration.rs

@@ -8,6 +8,7 @@ use super::super::VmmAction;
 use crate::parsed_request::{Error, ParsedRequest};
 use crate::request::Body;
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_cpu_config(body: &Body) -> Result<ParsedRequest, Error> {
     METRICS.put_api_requests.cpu_cfg_count.inc();
 

src/api_server/src/request/drive.rs

@@ -8,6 +8,7 @@ use super::super::VmmAction;
 use crate::parsed_request::{checked_id, Error, ParsedRequest};
 use crate::request::{Body, StatusCode};
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_drive(
     body: &Body,
     id_from_path: Option<&str>,
@@ -38,6 +39,7 @@ pub(crate) fn parse_put_drive(
     }
 }
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_patch_drive(
     body: &Body,
     id_from_path: Option<&str>,

src/api_server/src/request/entropy.rs

@@ -7,6 +7,7 @@ use vmm::vmm_config::entropy::EntropyDeviceConfig;
 use crate::parsed_request::{Error, ParsedRequest};
 use crate::request::Body;
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_entropy(body: &Body) -> Result<ParsedRequest, Error> {
     let cfg = serde_json::from_slice::<EntropyDeviceConfig>(body.raw())?;
     Ok(ParsedRequest::new_sync(VmmAction::SetEntropyDevice(cfg)))

src/api_server/src/request/instance_info.rs

@@ -6,6 +6,7 @@ use vmm::rpc_interface::VmmAction;
 
 use crate::parsed_request::{Error, ParsedRequest};
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_get_instance_info() -> Result<ParsedRequest, Error> {
     METRICS.get_api_requests.instance_info_count.inc();
     Ok(ParsedRequest::new_sync(VmmAction::GetVmInstanceInfo))

src/api_server/src/request/logger.rs

@@ -8,6 +8,7 @@ use super::super::VmmAction;
 use crate::parsed_request::{Error, ParsedRequest};
 use crate::request::Body;
 
+#[tracing::instrument(level = "trace", ret)]
 pub(crate) fn parse_put_logger(body: &Body) -> Result<ParsedRequest, Error> {
     METRICS.put_api_requests.logger_count.inc();
     let res = serde_json::from_slice::<LoggerConfig>(body.raw());