/
githubmirror
/
metasploit-framework
Обзор
Документация
Войти
/
githubmirror
/
metasploit-framework
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
spec/support/shared/contexts/msf/ui_driver.rb
60 строк
2 KB
adfoster-r7
Improve kerberos file load error messages
15 авг 2025, 14:38
15 авг 2025, 14:38
bebb43f
Код
Авторство
О чём код?
RSpec.shared_context 'Msf::UIDriver' do let(:driver) do instance = double('Driver', framework: framework) allow(instance).to receive(:on_command_proc=).with(kind_of(Proc)) capture_logging(instance) instance end let(:driver_input) do double(Rex::Ui::Text::Input) end let(:driver_output) do instance = double( Rex::Ui::Text::Output, prompting?: false, prompting: false ) capture_logging(instance) instance end def reset_logging! @output = [] @error = [] @combined_output = [] end def capture_logging(target, capture_verbose: false) append_output = proc do |string = ''| lines = string.split("\n") @output ||= [] @output.concat(lines) @combined_output ||= [] @combined_output.concat(lines) end append_error = proc do |string = ''| lines = string.split("\n") @error ||= [] @error.concat(lines) @combined_output ||= [] @combined_output.concat(lines) end allow(target).to receive(:print, &append_output) allow(target).to receive(:print_line, &append_output) allow(target).to receive(:print_status, &append_output) allow(target).to receive(:print_good, &append_output) if capture_verbose allow(target).to receive(:vprint_status, &append_output) allow(target).to receive(:vprint_error, &append_error) end allow(target).to receive(:print_warning, &append_error) allow(target).to receive(:print_error, &append_error) allow(target).to receive(:print_bad, &append_error) end end