serial: buffer limit regression test

Signed-off-by: Iulian Barbu <[email protected]>

Author: Iulian Barbu

tests/integration_tests/functional/test_serial_io.py

@@ -1,12 +1,18 @@
 # Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 # SPDX-License-Identifier: Apache-2.0
 """Tests scenario for the Firecracker serial console."""
+
+import fcntl
+import os
+import termios
 import time
+
 from framework.microvm import Serial
 from framework.state_machine import TestState
+import framework.utils as utils
 
 
-class WaitLogin(TestState):
+class WaitLogin(TestState):  # pylint: disable=too-few-public-methods
     """Initial state when we wait for the login prompt."""
 
     def handle_input(self, serial, input_char) -> TestState:
@@ -18,7 +24,7 @@ def handle_input(self, serial, input_char) -> TestState:
         return self
 
 
-class WaitPasswordPrompt(TestState):
+class WaitPasswordPrompt(TestState):  # pylint: disable=too-few-public-methods
     """Wait for the password prompt to be shown."""
 
     def handle_input(self, serial, input_char) -> TestState:
@@ -32,7 +38,7 @@ def handle_input(self, serial, input_char) -> TestState:
         return self
 
 
-class WaitIDResult(TestState):
+class WaitIDResult(TestState):  # pylint: disable=too-few-public-methods
     """Wait for the console to show the result of the 'id' shell command."""
 
     def handle_input(self, unused_serial, input_char) -> TestState:
@@ -42,10 +48,10 @@ def handle_input(self, unused_serial, input_char) -> TestState:
         return self
 
 
-class TestFinished(TestState):
+class TestFinished(TestState):  # pylint: disable=too-few-public-methods
     """Test complete and successful."""
 
-    def handle_input(self, unused_serial, input_char) -> TestState:
+    def handle_input(self, unused_serial, _) -> TestState:
         """Return self since the test is about to end."""
         return self
 
@@ -68,11 +74,57 @@ def test_serial_console_login(test_microvm_with_ssh):
 
     serial = Serial(microvm)
     serial.open()
-
-    # Set initial state - wait for 'login:' prompt
     current_state = WaitLogin("login:")
 
     while not isinstance(current_state, TestFinished):
         output_char = serial.rx_char()
         current_state = current_state.handle_input(
             serial, output_char)
+
+
+def get_total_mem_size(pid):
+    """Get total memory usage for a process."""
+    cmd = f"pmap {pid} | tail -n 1 | sed 's/^ //' | tr -s ' ' | cut -d' ' -f2"
+    rc, stdout, stderr = utils.run_cmd(cmd)
+    assert rc == 0
+    assert stderr == ""
+
+    return stdout
+
+
+def send_bytes(tty, bytes_count, timeout=60):
+    """Send data to the terminal."""
+    start = time.time()
+    for _ in range(bytes_count):
+        fcntl.ioctl(tty, termios.TIOCSTI, '\n')
+        current = time.time()
+        if current - start > timeout:
+            break
+
+
+def test_serial_dos(test_microvm_with_ssh):
+    """Test serial console behavior under DoS."""
+    microvm = test_microvm_with_ssh
+    microvm.jailer.daemonize = False
+    microvm.spawn()
+    microvm.memory_events_queue = None
+
+    # Set up the microVM with 1 vCPU and a serial console.
+    microvm.basic_config(vcpu_count=1,
+                         add_root_device=False,
+                         boot_args='console=ttyS0 reboot=k panic=1 pci=off')
+    microvm.start()
+
+    # Open an fd for firecracker process terminal.
+    tty_path = f"/proc/{microvm.jailer_clone_pid}/fd/0"
+    tty_fd = os.open(tty_path, os.O_RDWR)
+
+    # Check if the total memory size changed.
+    before_size = get_total_mem_size(microvm.jailer_clone_pid)
+    send_bytes(tty_fd, 100000000, timeout=1)
+    after_size = get_total_mem_size(microvm.jailer_clone_pid)
+    assert before_size == after_size, "The memory size of the " \
+                                      "Firecracker process " \
+                                      "changed from {} to {}." \
+                                      .format(before_size,
+                                              after_size)