External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-16 04:10:44 -03:30
Code Issues Packages Projects Releases Wiki Activity

deal with case that can cause memory errors

Browse Source
This commit is contained in:
AlanCoding 2017-01-31 15:30:39 -05:00
parent f377da0ecb
commit bdc04723b0
3 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
awx/main/models/jobs.py
View File

@ -609,6 +609,8 @@ class Job(UnifiedJob, JobOptions, SurveyJobMixin, JobNotificationMixin):
if k in job_extra_vars]
return_content = content
for val in password_list:
if len(val) == 0:
continue # avoids memory errors
return_content = PlainTextCleaner.remove_sensitive(return_content, val)
return return_content

2
awx/main/redact.py
View File

@ -58,4 +58,6 @@ class PlainTextCleaner(object):
@staticmethod
def remove_sensitive(cleartext, sensitive):
if sensitive == '':
return cleartext
return re.sub(r'%s' % re.escape(sensitive), '$encrypted$', cleartext)

6
awx/main/tests/unit/models/test_survey_models.py
View File

@ -61,6 +61,11 @@ def test_survey_passwords_not_in_extra_vars():
@pytest.mark.survey
def test_survey_passwords_not_in_stdout(job_with_survey):
job_with_survey.survey_passwords['has_blank_value'] = '$encrypted$'
job_with_survey.extra_vars = json.dumps({
'has_blank_value': '',
'secret_key': '6kQngg3h8lgiSTvIEb21',
'SSN': '123-45-6789'})
example_stdout = '''
PLAY [all] *********************************************************************
@ -77,6 +82,7 @@ TASK [debug] *******************************************************************
display_stdout = job_with_survey._survey_search_and_replace(example_stdout)
assert display_stdout == example_stdout.replace(
'6kQngg3h8lgiSTvIEb21', '$encrypted$').replace('123-45-6789', '$encrypted$')
assert type(display_stdout) == type(example_stdout)