lets use mdaxfr.poy for faster processing

This commit is contained in:
Dionysus 2024-11-30 02:06:10 -05:00
parent fd44e0bdf2
commit 434a62f49d
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE

View File

@ -155,16 +155,14 @@ if __name__ == '__main__':
parser.add_argument('-t', '--tlds', action='store_true', help='Perform AXFR on all TLDs') parser.add_argument('-t', '--tlds', action='store_true', help='Perform AXFR on all TLDs')
parser.add_argument('-p', '--psl', action='store_true', help='use the Public Suffix List') parser.add_argument('-p', '--psl', action='store_true', help='use the Public Suffix List')
parser.add_argument('-c', '--concurrency', type=int, default=30, help='maximum concurrent tasks') parser.add_argument('-c', '--concurrency', type=int, default=30, help='maximum concurrent tasks')
parser.add_argument('-o', '--output', type=str, default='axfrout', help='output directory') parser.add_argument('-o', '--output', dest='output_dir', type=str, default='axfrout', help='output directory')
args = parser.parse_args() args = parser.parse_args()
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s') logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
output_dir = args.output # Store the output directory
root_dir = os.path.join(output_dir, 'root') # Use the stored output directory
# Create output directories # Create output directories
os.makedirs(output_dir, exist_ok=True) os.makedirs(args.output_dir, exist_ok=True)
root_dir = os.path.join(args.output_dir, 'root')
os.makedirs(root_dir, exist_ok=True) os.makedirs(root_dir, exist_ok=True)
# Set DNS timeout # Set DNS timeout
@ -221,7 +219,7 @@ if __name__ == '__main__':
elif args.psl: elif args.psl:
# PSL mode # PSL mode
logging.info('Fetching PSL domains...') logging.info('Fetching PSL domains...')
psl_dir = os.path.join(output_dir, 'psl') # Use output_dir here too psl_dir = os.path.join(args.output_dir, 'psl')
os.makedirs(psl_dir, exist_ok=True) os.makedirs(psl_dir, exist_ok=True)
domains = get_psl_tlds() domains = get_psl_tlds()