|
3 | 3 | from os.path import abspath, join, split, exists, splitext, getsize, sep |
4 | 4 | from os import makedirs, remove, stat |
5 | 5 | import time |
6 | | -import traceback |
7 | 6 | from nxc.paths import TMP_PATH |
8 | 7 | from nxc.protocols.smb.remotefile import RemoteFile |
9 | 8 | from impacket.smb3structs import FILE_READ_DATA |
@@ -159,8 +158,8 @@ def read_chunk(self, remote_file, chunk_size=CHUNK_SIZE): |
159 | 158 | remote_file.__smbConnection = self.smb.conn |
160 | 159 | return self.read_chunk(remote_file) |
161 | 160 |
|
162 | | - except Exception: |
163 | | - traceback.print_exc() |
| 161 | + except Exception as e: |
| 162 | + self.logger.exception(e) |
164 | 163 | break |
165 | 164 |
|
166 | 165 | return chunk |
@@ -214,13 +213,13 @@ def spider_shares(self): |
214 | 213 | # Start the spider at the root of the share folder |
215 | 214 | self.results[share_name] = {} |
216 | 215 | self.spider_folder(share_name, "") |
217 | | - except SessionError: |
218 | | - traceback.print_exc() |
| 216 | + except SessionError as e: |
| 217 | + self.logger.exception(e) |
219 | 218 | self.logger.fail("Got a session error while spidering.") |
220 | 219 | self.reconnect() |
221 | 220 |
|
222 | 221 | except Exception as e: |
223 | | - traceback.print_exc() |
| 222 | + self.logger.exception(e) |
224 | 223 | self.logger.fail(f"Error enumerating shares: {e!s}") |
225 | 224 |
|
226 | 225 | # Save the metadata. |
@@ -412,8 +411,7 @@ def print_stats(self): |
412 | 411 | self.logger.display(f"Total folders found: {num_folders}") |
413 | 412 | num_folders_filtered = self.stats.get("num_folders_filtered", 0) |
414 | 413 | if num_folders_filtered: |
415 | | - num_filtered_folders = len(num_folders_filtered) |
416 | | - self.logger.display(f"Folders Filtered: {num_filtered_folders}") |
| 414 | + self.logger.display(f"Folders Filtered: {num_folders_filtered}") |
417 | 415 |
|
418 | 416 | # File statistics. |
419 | 417 | num_files = self.stats.get("num_files", 0) |
|
0 commit comments