Add support to customize existing users

- Randomize unused users
This commit is contained in:
Manuel Gutierrez
2020-10-22 17:08:00 +02:00
parent fe1b3d147e
commit 1aa36c15d6
2 changed files with 33 additions and 2 deletions

View File

@@ -1,6 +1,8 @@
import logging import logging
import sys import sys
import json import json
import random
import string
import os import os
# Set framework path # Set framework path
@@ -43,7 +45,9 @@ if __name__ == "__main__":
# abort if no user file detected # abort if no user file detected
sys.exit(0) sys.exit(0)
username, password = read_user_file() username, password = read_user_file()
if username not in db_users(): initial_users = db_users()
if username not in initial_users:
# create a new user
create_user(username=username, password=password) create_user(username=username, password=password)
users = db_users() users = db_users()
uid = users[username] uid = users[username]
@@ -57,3 +61,30 @@ if __name__ == "__main__":
str(rid), str(rid),
], ],
) )
else:
# modify an existing user ("wazuh" or "wazuh-wui")
uid = initial_users[username]
update_user(
user_id=[
str(uid),
],
password=password,
)
# set a random password for all other users
for name, id in initial_users.items():
if name != username:
random_pass = "".join(
random.choices(
string.ascii_uppercase
+ string.ascii_lowercase
+ string.digits
+ "@$!%*?&-_",
k=16,
)
)
update_user(
user_id=[
str(id),
],
password=random_pass,
)

View File

@@ -84,7 +84,7 @@ function_create_custom_user() {
} }
EOF EOF
# insert user into API database # create or customize API user
/var/ossec/framework/python/bin/python3 /var/ossec/framework/scripts/create_user.py /var/ossec/framework/python/bin/python3 /var/ossec/framework/scripts/create_user.py
fi fi
} }