bivashy/Blitz-Proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(bulk_users): replace password generation method with secrets module

Browse Source
This commit is contained in:
ReturnFI
2025-12-12 09:15:55 +00:00
parent 889b48e4bd
commit 826a9bc137
1 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
core/scripts/hysteria2/bulk_users.py
View File

@ -2,10 +2,10 @@
import init_paths import init_paths
import sys import sys
import os
import subprocess
import argparse import argparse
import re import re
import secrets
import string
from db.database import db from db.database import db
def add_bulk_users(traffic_gb, expiration_days, count, prefix, start_number, unlimited_user): def add_bulk_users(traffic_gb, expiration_days, count, prefix, start_number, unlimited_user):
@ -44,16 +44,8 @@ def add_bulk_users(traffic_gb, expiration_days, count, prefix, start_number, unl
if count > new_users_count: if count > new_users_count:
print(f"Warning: {count - new_users_count} user(s) already exist. Skipping them.") print(f"Warning: {count - new_users_count} user(s) already exist. Skipping them.")
try: alphabet = string.ascii_letters + string.digits
password_process = subprocess.run(['pwgen', '-s', '32', str(new_users_count)], capture_output=True, text=True, check=True) passwords = [''.join(secrets.choice(alphabet) for _ in range(32)) for _ in range(new_users_count)]
passwords = password_process.stdout.strip().split('\n')
except (FileNotFoundError, subprocess.CalledProcessError):
print("Warning: 'pwgen' not found or failed. Falling back to UUID for password generation.")
passwords = [subprocess.check_output(['cat', '/proc/sys/kernel/random/uuid'], text=True).strip() for _ in range(new_users_count)]
if len(passwords) < new_users_count:
print("Error: Could not generate enough passwords.")
return 1
users_to_insert = [] users_to_insert = []
for i, username in enumerate(new_usernames): for i, username in enumerate(new_usernames):