mirror of
https://github.com/chatmail/relay.git
synced 2026-05-12 00:54:37 +00:00
62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
import base64
|
|
import sys
|
|
|
|
|
|
def get_user_data(user):
|
|
if user == b"link2xt@instant2.testrun.org":
|
|
return dict(
|
|
homedir="/home/vmail/link2xt",
|
|
uid="vmail",
|
|
gid="vmail",
|
|
password=b"Ahyei6ie",
|
|
)
|
|
return {}
|
|
|
|
|
|
def create_user(user, password):
|
|
assert isinstance(password, bytes)
|
|
return dict(
|
|
homedir=f"/home/vmail/{user}", uid="vmail", gid="vmail", password=password
|
|
)
|
|
|
|
|
|
def verify_user(user, password):
|
|
userdata = get_user_data(user)
|
|
if userdata:
|
|
if userdata.get("password") == password:
|
|
userdata["status"] = "ok"
|
|
else:
|
|
userdata["status"] = "fail"
|
|
else:
|
|
userdata = create_user(user, password)
|
|
userdata["status"] = "ok"
|
|
|
|
return userdata
|
|
|
|
|
|
def lookup_user(user):
|
|
userdata = get_user_data(user)
|
|
if userdata:
|
|
userdata["status"] = "ok"
|
|
else:
|
|
userdata["status"] = "fail"
|
|
return userdata
|
|
|
|
|
|
def dump_result(res):
|
|
for key, value in res.items():
|
|
print(f"{key}={value}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if sys.argv[1] == "hexauth":
|
|
login = base64.b16decode(sys.argv[2])
|
|
password = base64.b16decode(sys.argv[3])
|
|
res = verify_user(login, password)
|
|
dump_result(res)
|
|
elif sys.argv[1] == "hexlookup":
|
|
login = base64.b16decode(sys.argv[2])
|
|
res = lookup_user(login)
|
|
dump_result(res)
|