INNER CODE UNIT · Python
main
gendigitalinc/ioc · AuraStealer/extras/aurastealer_config_extractor.py:42
def main():
parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("-s", help="sample file")
group.add_argument("-d", help="directory of samples")
args = parser.parse_args()
targets = []
if args.s:
targets.append(args.s)
elif args.d:
try:
for f in os.listdir(args.d):
path = os.path.join(args.d, f)
if os.path.isfile(path):
targets.append(path)
except OSError as e:
print(f"Error reading directory: {e}")