mirror of
https://github.com/chatmail/relay.git
synced 2026-05-11 16:34:39 +00:00
Compare commits
1 Commits
link2xt/py
...
link2xt/cr
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d1ddf4d9e0 |
@@ -33,13 +33,6 @@ class Connection:
|
|||||||
def cursor(self):
|
def cursor(self):
|
||||||
return self._sqlconn.cursor()
|
return self._sqlconn.cursor()
|
||||||
|
|
||||||
def create_user(self, addr: str, password: str):
|
|
||||||
"""Create a row in the users table."""
|
|
||||||
self.execute("PRAGMA foreign_keys=on")
|
|
||||||
q = """INSERT INTO users (addr, password, last_login)
|
|
||||||
VALUES (?, ?, ?)"""
|
|
||||||
self.execute(q, (addr, password, int(time.time())))
|
|
||||||
|
|
||||||
def get_user(self, addr: str) -> {}:
|
def get_user(self, addr: str) -> {}:
|
||||||
"""Get a row from the users table."""
|
"""Get a row from the users table."""
|
||||||
q = "SELECT addr, password, last_login from users WHERE addr = ?"
|
q = "SELECT addr, password, last_login from users WHERE addr = ?"
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
|
import time
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
import crypt
|
import crypt
|
||||||
@@ -46,17 +47,6 @@ def is_allowed_to_create(user, cleartext_password) -> bool:
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
def create_user(db, user, encrypted_password):
|
|
||||||
with db.write_transaction() as conn:
|
|
||||||
conn.create_user(user, encrypted_password)
|
|
||||||
return dict(
|
|
||||||
home=f"/home/vmail/{user}",
|
|
||||||
uid="vmail",
|
|
||||||
gid="vmail",
|
|
||||||
password=encrypted_password,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def get_user_data(db, user):
|
def get_user_data(db, user):
|
||||||
with db.read_connection() as conn:
|
with db.read_connection() as conn:
|
||||||
result = conn.get_user(user)
|
result = conn.get_user(user)
|
||||||
@@ -71,14 +61,30 @@ def lookup_userdb(db, user):
|
|||||||
|
|
||||||
|
|
||||||
def lookup_passdb(db, user, cleartext_password):
|
def lookup_passdb(db, user, cleartext_password):
|
||||||
userdata = get_user_data(db, user)
|
with db.write_transaction() as conn:
|
||||||
if not userdata:
|
userdata = conn.get_user(user)
|
||||||
|
if userdata:
|
||||||
|
# Update last login time.
|
||||||
|
conn.execute(
|
||||||
|
"UPDATE users SET last_login=? WHERE addr=?", (int(time.time()), user)
|
||||||
|
)
|
||||||
|
|
||||||
|
userdata["uid"] = "vmail"
|
||||||
|
userdata["gid"] = "vmail"
|
||||||
|
return userdata
|
||||||
if not is_allowed_to_create(user, cleartext_password):
|
if not is_allowed_to_create(user, cleartext_password):
|
||||||
return
|
return
|
||||||
|
|
||||||
encrypted_password = encrypt_password(cleartext_password)
|
encrypted_password = encrypt_password(cleartext_password)
|
||||||
userdata = create_user(db=db, user=user, encrypted_password=encrypted_password)
|
q = """INSERT INTO users (addr, password, last_login)
|
||||||
userdata["password"] = userdata["password"].strip()
|
VALUES (?, ?, ?)"""
|
||||||
return userdata
|
conn.execute(q, (user, encrypted_password, int(time.time())))
|
||||||
|
return dict(
|
||||||
|
home=f"/home/vmail/{user}",
|
||||||
|
uid="vmail",
|
||||||
|
gid="vmail",
|
||||||
|
password=encrypted_password,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def handle_dovecot_request(msg, db, mail_domain):
|
def handle_dovecot_request(msg, db, mail_domain):
|
||||||
|
|||||||
Reference in New Issue
Block a user