浏览代码

Check for empty data in more cases.

Andrew Walbran 2 年之前
父节点
当前提交
daac39791b
共有 1 个文件被更改,包括 3 次插入0 次删除
  1. 3 0
      src/device/socket/vsock.rs

+ 3 - 0
src/device/socket/vsock.rs

@@ -397,9 +397,11 @@ impl<H: Hal, T: Transport> VirtIOSocket<H, T> {
 
             match op {
                 VirtioVsockOp::Request => {
+                    header.check_data_is_empty()?;
                     // TODO: Send a Rst, or support listening.
                 }
                 VirtioVsockOp::Response => {
+                    header.check_data_is_empty()?;
                     return Ok(Some(VsockEvent {
                         source: connection_info.dst,
                         destination: VsockAddr {
@@ -456,6 +458,7 @@ impl<H: Hal, T: Transport> VirtIOSocket<H, T> {
                     }));
                 }
                 VirtioVsockOp::CreditRequest => {
+                    header.check_data_is_empty()?;
                     // TODO: Send a credit update.
                 }
                 VirtioVsockOp::Invalid => {