| ''' |
| run the command under test, under valgrind and collect memory leak info |
| as a separate test. |
| ''' |
| |
| |
| import os |
| import re |
| import signal |
| from string import Template |
| import subprocess |
| import time |
| from TdcPlugin import TdcPlugin |
| |
| from tdc_config import * |
| |
| def vp_extract_num_from_string(num_as_string_maybe_with_commas): |
| return int(num_as_string_maybe_with_commas.replace(',','')) |
| |
| class SubPlugin(TdcPlugin): |
| def __init__(self): |
| self.sub_class = 'valgrind/SubPlugin' |
| self.tap = '' |
| super().__init__() |
| |
| def pre_suite(self, testcount, testidlist): |
| '''run commands before test_runner goes into a test loop''' |
| super().pre_suite(testcount, testidlist) |
| if self.args.verbose > 1: |
| print('{}.pre_suite'.format(self.sub_class)) |
| if self.args.valgrind: |
| self._add_to_tap('1..{}\n'.format(self.testcount)) |
| |
| def post_suite(self, index): |
| '''run commands after test_runner goes into a test loop''' |
| super().post_suite(index) |
| self._add_to_tap('\n|---\n') |
| if self.args.verbose > 1: |
| print('{}.post_suite'.format(self.sub_class)) |
| print('{}'.format(self.tap)) |
| if self.args.verbose < 4: |
| subprocess.check_output('rm -f vgnd-*.log', shell=True) |
| |
| def add_args(self, parser): |
| super().add_args(parser) |
| self.argparser_group = self.argparser.add_argument_group( |
| 'valgrind', |
| 'options for valgrindPlugin (run command under test under Valgrind)') |
| |
| self.argparser_group.add_argument( |
| '-V', '--valgrind', action='store_true', |
| help='Run commands under valgrind') |
| |
| return self.argparser |
| |
| def adjust_command(self, stage, command): |
| super().adjust_command(stage, command) |
| cmdform = 'list' |
| cmdlist = list() |
| |
| if not self.args.valgrind: |
| return command |
| |
| if self.args.verbose > 1: |
| print('{}.adjust_command'.format(self.sub_class)) |
| |
| if not isinstance(command, list): |
| cmdform = 'str' |
| cmdlist = command.split() |
| else: |
| cmdlist = command |
| |
| if stage == 'execute': |
| if self.args.verbose > 1: |
| print('adjust_command: stage is {}; inserting valgrind stuff in command [{}] list [{}]'. |
| format(stage, command, cmdlist)) |
| cmdlist.insert(0, '--track-origins=yes') |
| cmdlist.insert(0, '--show-leak-kinds=definite,indirect') |
| cmdlist.insert(0, '--leak-check=full') |
| cmdlist.insert(0, '--log-file=vgnd-{}.log'.format(self.args.testid)) |
| cmdlist.insert(0, '-v') # ask for summary of non-leak errors |
| cmdlist.insert(0, ENVIR['VALGRIND_BIN']) |
| else: |
| pass |
| |
| if cmdform == 'str': |
| command = ' '.join(cmdlist) |
| else: |
| command = cmdlist |
| |
| if self.args.verbose > 1: |
| print('adjust_command: return command [{}]'.format(command)) |
| return command |
| |
| def post_execute(self): |
| if not self.args.valgrind: |
| return |
| |
| self.definitely_lost_re = re.compile( |
| r'definitely lost:\s+([,0-9]+)\s+bytes in\s+([,0-9]+)\sblocks', re.MULTILINE | re.DOTALL) |
| self.indirectly_lost_re = re.compile( |
| r'indirectly lost:\s+([,0-9]+)\s+bytes in\s+([,0-9]+)\s+blocks', re.MULTILINE | re.DOTALL) |
| self.possibly_lost_re = re.compile( |
| r'possibly lost:\s+([,0-9]+)bytes in\s+([,0-9]+)\s+blocks', re.MULTILINE | re.DOTALL) |
| self.non_leak_error_re = re.compile( |
| r'ERROR SUMMARY:\s+([,0-9]+) errors from\s+([,0-9]+)\s+contexts', re.MULTILINE | re.DOTALL) |
| |
| def_num = 0 |
| ind_num = 0 |
| pos_num = 0 |
| nle_num = 0 |
| |
| # what about concurrent test runs? Maybe force them to be in different directories? |
| with open('vgnd-{}.log'.format(self.args.testid)) as vfd: |
| content = vfd.read() |
| def_mo = self.definitely_lost_re.search(content) |
| ind_mo = self.indirectly_lost_re.search(content) |
| pos_mo = self.possibly_lost_re.search(content) |
| nle_mo = self.non_leak_error_re.search(content) |
| |
| if def_mo: |
| def_num = int(def_mo.group(2)) |
| if ind_mo: |
| ind_num = int(ind_mo.group(2)) |
| if pos_mo: |
| pos_num = int(pos_mo.group(2)) |
| if nle_mo: |
| nle_num = int(nle_mo.group(1)) |
| |
| mem_results = '' |
| if (def_num > 0) or (ind_num > 0) or (pos_num > 0) or (nle_num > 0): |
| mem_results += 'not ' |
| |
| mem_results += 'ok {} - {}-mem # {}\n'.format( |
| self.args.test_ordinal, self.args.testid, 'memory leak check') |
| self._add_to_tap(mem_results) |
| if mem_results.startswith('not '): |
| print('{}'.format(content)) |
| self._add_to_tap(content) |
| |
| def _add_to_tap(self, more_tap_output): |
| self.tap += more_tap_output |
