| 96 | exit(-1) |
| 97 | |
| 98 | class CommandStreamReader(object): |
| 99 | |
| 100 | def __init__(self, stream, target, tag, patterns=None, outfile=None): |
| 101 | self.stream = stream |
| 102 | self.target = target |
| 103 | self.tag = tag |
| 104 | self.lines = [] |
| 105 | self.patterns = patterns or [] |
| 106 | self.outfile = outfile |
| 107 | self.ended = False |
| 108 | |
| 109 | # Empty files that already exist. |
| 110 | if self.outfile != None: |
| 111 | with open(self.outfile, 'w'): pass |
| 112 | |
| 113 | # Read lines from the stream until it ends. |
| 114 | async def _read(self): |
| 115 | while True: |
| 116 | if self.stream.at_eof(): |
| 117 | break |
| 118 | try: |
| 119 | line = (await self.stream.readline()).decode('utf8').rstrip() |
| 120 | except ValueError: |
| 121 | error('{bright}[{yellow}' + self.target.address + '{crst}/{bgreen}' + self.tag + '{crst}]{rst} A line was longer than 64 KiB and cannot be processed. Ignoring.') |
| 122 | continue |
| 123 | |
| 124 | if line != '': |
| 125 | info('{bright}[{yellow}' + self.target.address + '{crst}/{bgreen}' + self.tag + '{crst}]{rst} ' + line.strip().replace('{', '{{').replace('}', '}}'), verbosity=3) |
| 126 | |
| 127 | # Check lines for pattern matches. |
| 128 | for p in self.patterns: |
| 129 | description = '' |
| 130 | |
| 131 | # Match and replace entire pattern. |
| 132 | match = p.pattern.search(line) |
| 133 | if match: |
| 134 | if p.description: |
| 135 | description = p.description.replace('{match}', line[match.start():match.end()]) |
| 136 | |
| 137 | # Match and replace substrings. |
| 138 | matches = p.pattern.findall(line) |
| 139 | if len(matches) > 0 and isinstance(matches[0], tuple): |
| 140 | matches = list(matches[0]) |
| 141 | |
| 142 | match_count = 1 |
| 143 | for match in matches: |
| 144 | if p.description: |
| 145 | description = description.replace('{match' + str(match_count) + '}', match) |
| 146 | match_count += 1 |
| 147 | |
| 148 | async with self.target.lock: |
| 149 | with open(os.path.join(self.target.scandir, '_patterns.log'), 'a') as file: |
| 150 | info('{bright}[{yellow}' + self.target.address + '{crst}/{bgreen}' + self.tag + '{crst}]{rst} {bmagenta}' + description + '{rst}', verbosity=2) |
| 151 | file.writelines(description + '\n\n') |
| 152 | else: |
| 153 | info('{bright}[{yellow}' + self.target.address + '{crst}/{bgreen}' + self.tag + '{crst}]{rst} {bmagenta}Matched Pattern: ' + line[match.start():match.end()] + '{rst}', verbosity=2) |
| 154 | async with self.target.lock: |
| 155 | with open(os.path.join(self.target.scandir, '_patterns.log'), 'a') as file: |