Implement "pause" method

lib/pwnlib/pwn.rb

@@ -17,7 +17,7 @@
 require 'pwnlib/tubes/process'
 require 'pwnlib/tubes/serialtube'
 require 'pwnlib/tubes/sock'
-
+require 'pwnlib/ui'
 require 'pwnlib/util/cyclic'
 require 'pwnlib/util/fiddling'
 require 'pwnlib/util/getdents'
@@ -31,6 +31,7 @@ module Pwn
   include ::Pwnlib::Context
   include ::Pwnlib::Logger
   include ::Pwnlib::Runner
+  include ::Pwnlib::UI
   include ::Pwnlib::Util::Cyclic
   include ::Pwnlib::Util::Fiddling
   include ::Pwnlib::Util::HexDump

lib/pwnlib/ui.rb

@@ -0,0 +1,21 @@
+# encoding: ASCII-8BIT
+# frozen_string_literal: true
+
+require 'pwnlib/logger'
+
+module Pwnlib
+  # This module collects utilities that need user interactions.
+  module UI
+    module_function
+
+    # Waits for user input.
+    #
+    # @return [void]
+    def pause
+      log.info('Paused (press enter to continue)')
+      $stdin.gets
+    end
+
+    include ::Pwnlib::Logger
+  end
+end

test/test_helper.rb

@@ -51,5 +51,22 @@ def log_hook(obj)
         ::Pwnlib::Logger.log.instance_variable_set(:@logdev, old)
       end
     end
+
+    # Hooks +$stdin+.
+    #
+    # @param [IO] io
+    #   IO object to be used as +$stdin+.
+    #
+    # @return [Object]
+    #   Returns what the block returned.
+    def hook_stdin(io)
+      org_stdin = $stdin
+      $stdin = io
+      begin
+        yield
+      ensure
+        $stdin = org_stdin
+      end
+    end
   end
 end

test/ui_test.rb

@@ -0,0 +1,18 @@
+# encoding: ASCII-8BIT
+# frozen_string_literal: true
+
+require 'stringio'
+
+require 'test_helper'
+
+require 'pwnlib/ui'
+
+class UITest < MiniTest::Test
+  def test_pause
+    hook_stdin(StringIO.new("\n")) do
+      assert_output(<<-EOS) { log_stdout { ::Pwnlib::UI.pause } }
+[INFO] Paused (press enter to continue)
+      EOS
+    end
+  end
+end