commit 76a36a28b352916746d6f489e72b2fac69522d6d
author Alice Wang <aliceywang@google.com> Thu Jul 27 12:11:01 2023 +0000
committer Alice Wang <aliceywang@google.com> Thu Aug 24 14:45:01 2023 +0000
tree be9758e20934628fc9980e8a0455354ce7254278
parent 748b032fb17d902378d523505413885943cc93dc

[rialto] Move request processing to a separate module

To keep the RKP VM payload execution code isolated from the
rest of the code.

Bug: 291732060
Test: atest rialto_test
Change-Id: Ia242c5050e3d681116c3871533b1baf2ff14d37b

rialto/src/main.rs

diff --git a/rialto/src/main.rs b/rialto/src/main.rs
index 1c5090a..a8338ca 100644
--- a/rialto/src/main.rs
+++ b/rialto/src/main.rs
@@ -20,6 +20,7 @@
 mod communication;
 mod error;
 mod exceptions;
+mod requests;
 
 extern crate alloc;
 
@@ -31,7 +32,6 @@
 use hyp::{get_mem_sharer, get_mmio_guard};
 use libfdt::FdtError;
 use log::{debug, error, info};
-use service_vm_comm::{Request, Response};
 use virtio_drivers::{
     device::socket::VsockAddr,
     transport::{pci::bus::PciRoot, DeviceType, Transport},
@@ -139,9 +139,7 @@
     debug!("Found socket device: guest cid = {:?}", socket_device.guest_cid());
 
     let mut vsock_stream = VsockStream::new(socket_device, host_addr())?;
-    let response = match vsock_stream.read_request()? {
-        Request::Reverse(v) => Response::Reverse(v.into_iter().rev().collect()),
-    };
+    let response = requests::process_request(vsock_stream.read_request()?);
     vsock_stream.write_response(&response)?;
     vsock_stream.shutdown()?;