125 lines
3.3 KiB
Python
125 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
import logging
|
|
import os
|
|
|
|
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'core.settings')
|
|
|
|
import django
|
|
import uvicorn
|
|
|
|
django.setup()
|
|
|
|
from core.globals import init_loop
|
|
from mail.protocol import LMTPHandler
|
|
from mail.socket import UnixSocketLMTPController
|
|
|
|
|
|
|
|
# async def handle_echo(reader, writer):
|
|
# # session = Session()
|
|
# data = await reader.read(100)
|
|
# message = data.decode()
|
|
# addr = writer.get_extra_info('peername')
|
|
#
|
|
# print(f"Received {message!r} from {addr!r}")
|
|
#
|
|
# print(f"Send: {message!r}")
|
|
# writer.write(data)
|
|
# await writer.drain()
|
|
#
|
|
# print("Close the connection")
|
|
# writer.close()
|
|
# await writer.wait_closed()
|
|
|
|
|
|
class UvicornServer(uvicorn.Server):
|
|
def install_signal_handlers(self):
|
|
pass
|
|
|
|
|
|
async def web(loop):
|
|
log_config = uvicorn.config.LOGGING_CONFIG
|
|
log_config["handlers"]["default"] = {"class": "logging.FileHandler", "filename": "web.log", "formatter": "default"}
|
|
log_config["handlers"]["access"] = {"class": "logging.FileHandler", "filename": "web-access.log",
|
|
"formatter": "access"}
|
|
config = uvicorn.Config("core.asgi:application", uds="web.sock", log_config=log_config)
|
|
server = UvicornServer(config=config)
|
|
await server.serve()
|
|
|
|
|
|
# async def tcp(loop):
|
|
# log = logging.getLogger('test.log')
|
|
# log.addHandler(logging.FileHandler('test.log'))
|
|
# log.setLevel(logging.DEBUG)
|
|
# log.info("Starting TCP server")
|
|
# server = await asyncio.start_unix_server(handle_echo, path='test.sock')
|
|
#
|
|
# addrs = ', '.join(str(sock.getsockname()) for sock in server.sockets)
|
|
# log.info(f'Serving on {addrs}')
|
|
#
|
|
# async with server:
|
|
# await server.serve_forever()
|
|
# log.info("TCP done")
|
|
|
|
|
|
async def lmtp(loop):
|
|
import grp
|
|
log = logging.getLogger('mail.log')
|
|
log.addHandler(logging.FileHandler('mail.log'))
|
|
# log.setLevel(logging.WARNING)
|
|
log.setLevel(logging.INFO)
|
|
log.info("Starting LMTP server")
|
|
server = await UnixSocketLMTPController(LMTPHandler(), unix_socket='lmtp.sock', loop=loop).serve()
|
|
|
|
addrs = ', '.join(str(sock.getsockname()) for sock in server.sockets)
|
|
log.info(f'Serving on {addrs}')
|
|
|
|
try:
|
|
os.chmod('lmtp.sock', 0o775)
|
|
current_uid = os.getuid()
|
|
posix_gid = grp.getgrnam('postfix').gr_gid
|
|
os.chown('lmtp.sock', current_uid, posix_gid)
|
|
except Exception as e:
|
|
log.error(e)
|
|
|
|
async with server:
|
|
await server.serve_forever()
|
|
log.info("LMTP done")
|
|
|
|
|
|
def main():
|
|
import sdnotify
|
|
import setproctitle
|
|
import os
|
|
setproctitle.setproctitle("c3lf-sys3")
|
|
log = logging.getLogger('server.log')
|
|
log.addHandler(logging.FileHandler('server.log'))
|
|
log.setLevel(logging.DEBUG)
|
|
log.info("Starting server")
|
|
loop = init_loop()
|
|
loop.create_task(web(loop))
|
|
# loop.create_task(tcp(loop))
|
|
loop.create_task(lmtp(loop))
|
|
n = sdnotify.SystemdNotifier()
|
|
n.notify("READY=1")
|
|
log.info("Server ready")
|
|
try:
|
|
loop.run_forever()
|
|
finally:
|
|
loop.close()
|
|
try:
|
|
os.remove("lmtp.sock")
|
|
except Exception as e:
|
|
log.error(e)
|
|
try:
|
|
os.remove("web.sock")
|
|
except Exception as e:
|
|
log.error(e)
|
|
log.error(e)
|
|
logging.info("Server stopped")
|
|
|
|
logging.shutdown()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|