vault: Read stdin data as binary on python3 (#52229)

On python3 sys.stdin is an encoded file object that does not support
reading raw binary data. Use the supplied buffer object to do so.

Signed-off-by: Sven Wegener <sven.wegener@inovex.de>

Co-authored-by: Sven Wegener <sven.wegener@inovex.de>
pull/72933/head
Sven Wegener 4 years ago committed by GitHub
parent e5ccd18be4
commit 08cc6edc64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -0,0 +1,2 @@
bugfixes:
- vault - Support reading raw binary data from stdin under python3

@ -1038,7 +1038,10 @@ class VaultEditor:
try:
if filename == '-':
data = sys.stdin.read()
if PY3:
data = sys.stdin.buffer.read()
else:
data = sys.stdin.read()
else:
with open(filename, "rb") as fh:
data = fh.read()

@ -22,6 +22,7 @@ __metaclass__ = type
import os
import tempfile
from io import BytesIO, StringIO
import pytest
@ -32,6 +33,7 @@ from ansible import errors
from ansible.parsing import vault
from ansible.parsing.vault import VaultLib, VaultEditor, match_encrypt_secret
from ansible.module_utils.six import PY3
from ansible.module_utils._text import to_bytes, to_text
from units.mock.vault_helper import TextVaultSecret
@ -113,6 +115,21 @@ class TestVaultEditor(unittest.TestCase):
self.assertNotEqual(src_contents, b_ciphertext)
def test_stdin_binary(self):
stdin_data = '\0'
if PY3:
fake_stream = StringIO(stdin_data)
fake_stream.buffer = BytesIO(to_bytes(stdin_data))
else:
fake_stream = BytesIO(to_bytes(stdin_data))
with patch('sys.stdin', fake_stream):
ve = self._vault_editor()
data = ve.read_data('-')
self.assertEqual(data, b'\0')
@patch('ansible.parsing.vault.subprocess.call')
def test_edit_file_helper_call_exception(self, mock_sp_call):
self._test_dir = self._create_test_dir()

Loading…
Cancel
Save