forked from Firepup650/python-talk
Compare commits
28 commits
Author | SHA1 | Date | |
---|---|---|---|
21bbba74c0 | |||
6376818b85 | |||
e71b256a3e | |||
f5e4726c8e | |||
8714a1bec8 | |||
d39f8224d4 | |||
5805d5efcc | |||
9ff812a7b3 | |||
f90c4f40c8 | |||
426fa341ce | |||
a44e7c29fd | |||
aaeb313181 | |||
6d1ff52b02 | |||
8bc876d7af | |||
a245acc770 | |||
a6a9ba0754 | |||
a6ecd9a129 | |||
d84aab7feb | |||
95e1c3f38d | |||
85f4b289e8 | |||
6ab9c030ba | |||
284c3932dd | |||
0994f92669 | |||
16b74ece80 | |||
7c0c4b0875 | |||
1714010182 | |||
60098ddfd4 | |||
53e57b9dc6 |
7 changed files with 440 additions and 240 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -2,3 +2,4 @@ __pycache__/**
|
||||||
.env
|
.env
|
||||||
cache.py
|
cache.py
|
||||||
*.swp
|
*.swp
|
||||||
|
bugs.txt
|
||||||
|
|
35
IRC-spec.txt
Normal file
35
IRC-spec.txt
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
# On-connect
|
||||||
|
:<servername> CAP * LS :
|
||||||
|
:<servername> 001 <client> :Welcome to the python-talk Network, <client>
|
||||||
|
:<servername> 002 <client> :Your host is <servername>, running version 0.1
|
||||||
|
:<servername> 003 <client> :This server was created 0
|
||||||
|
:<servername> 004 <client> python-talk 0.1 r t
|
||||||
|
:<servername> 005 <client> AWAYLEN=200,CASEMAPPING=ascii,CHANTYPES=#,ELIST=U,HOSTLEN=16 :are supported by this server
|
||||||
|
:<servername> 005 <client> CHANLIMIT=#:1,CHANMODES=b,k,l,t,CHANNELLEN=4,KICKLEN=1,MAXLIST=b:1 :are supported by this server
|
||||||
|
:<servername> 005 <client> MODES=4,NETWORK=talknet,NICKLEN=20,TOPICLEN=30,USERLEN=1 :are supported by this server
|
||||||
|
:<servername> 251 <client> :There are <u> users and 0 invisible on <s> servers
|
||||||
|
:<servername> 252 <client> 0 :operator(s) online
|
||||||
|
:<servername> 253 <client> 0 :unknown connection(s)
|
||||||
|
:<servername> 254 <client> 1 :channel formed
|
||||||
|
:<servername> 255 <client> :I have <c> clients and <s> servers
|
||||||
|
:<servername> 265 <client> [<u> <m>] :Current local users <u>, max <m>
|
||||||
|
:<servername> 266 <client> [<u> <m>] :Current global users <u>, max <m>
|
||||||
|
:<servername> 422 <client> :This server does not support MOTDs
|
||||||
|
:<client> JOIN #main
|
||||||
|
:<servername> 324 <client> #main t
|
||||||
|
:<servername> 332 <client> #main :Welcome to the main channel!
|
||||||
|
:<servername> 353 <client> = #main :<client>, ...other clients here...
|
||||||
|
:<servername> 366 <client> #main :End of /NAMES list
|
||||||
|
# On Attempting to join/part a channel, or anything else I don't want to deal with
|
||||||
|
:<servername> 421 <client> <command> :Unknown command
|
||||||
|
# I should probably return the "NOSUCHCHANNEL" thing for joins
|
||||||
|
|
||||||
|
#PING PONG
|
||||||
|
:<servername> PONG <servername> :<data>
|
||||||
|
|
||||||
|
# If the user tries to kill us or another server, respond with an easter egg
|
||||||
|
:<servername> 483 <client> :You can't kill me, I'm Skynet!
|
||||||
|
# Otherwise, if the user exists:
|
||||||
|
:<servername> 481 <client> :Permission Denied- There are no IRC operators here.
|
||||||
|
# Otherwise:
|
||||||
|
:<servername> 401 <client> <nickname> :No such nick/channel
|
13
init-scripts/freebsd-rc
Normal file
13
init-scripts/freebsd-rc
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
#!/bin/sh
|
||||||
|
. /etc/rc.subr
|
||||||
|
name="python_talk"
|
||||||
|
start_cmd="${name}_start"
|
||||||
|
stop_cmd=":"
|
||||||
|
rcvar=python_talk_enable
|
||||||
|
load_rc_config $name
|
||||||
|
: ${python_talk_enable:=no}
|
||||||
|
python_talk_start() {
|
||||||
|
cd /usr/src/python-talk
|
||||||
|
(while true; do python3 server.py --link=37.27.51.34:65048 --no-log --no-cache --cwlgd 2>&1 >> /var/log/python-talk.log; done) &
|
||||||
|
}
|
||||||
|
run_rc_command "$1"
|
38
init-scripts/systemd-system.service
Normal file
38
init-scripts/systemd-system.service
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
[Unit]
|
||||||
|
Description=firepup's python-talk
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
DynamicUser=yes
|
||||||
|
WorkingDirectory=/opt/python-talk
|
||||||
|
ExecStart=/usr/bin/python3 server.py --link=37.27.51.34:65048 --no-logs --no-cache --cwlgd
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
NoNewPrivileges=true
|
||||||
|
RestrictSUIDSGID=true
|
||||||
|
ProtectSystem=strict
|
||||||
|
ProtectHome=true
|
||||||
|
ProtectProc=ptraceable
|
||||||
|
ProcSubset=pid
|
||||||
|
PrivateTmp=true
|
||||||
|
CapabilityBoundingSet=
|
||||||
|
PrivateDevices=true
|
||||||
|
ProtectClock=true
|
||||||
|
ProtectKernelLogs=true
|
||||||
|
ProtectControlGroups=true
|
||||||
|
ProtectKernelModules=true
|
||||||
|
SystemCallArchitectures=native
|
||||||
|
# python.
|
||||||
|
MemoryDenyWriteExecute=false
|
||||||
|
RestrictNamespaces=true
|
||||||
|
ProtectHostname=true
|
||||||
|
LockPersonality=true
|
||||||
|
ProtectKernelTunables=true
|
||||||
|
RestrictAddressFamilies=AF_INET
|
||||||
|
RestrictRealtime=true
|
||||||
|
PrivateUsers=true
|
||||||
|
SystemCallFilter=@system-service
|
||||||
|
SystemCallFilter=~@privileged @resources
|
||||||
|
UMask=0077
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target
|
38
init-scripts/systemd-user.service
Normal file
38
init-scripts/systemd-user.service
Normal file
|
@ -0,0 +1,38 @@
|
||||||
|
[Unit]
|
||||||
|
Description=firepup's python-talk
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
WorkingDirectory=%h/python-talk
|
||||||
|
ExecStart=/usr/bin/python3 server.py --link=37.27.51.34:65048 --no-logs --no-cache --cwlgd
|
||||||
|
Restart=always
|
||||||
|
RestartSec=5
|
||||||
|
NoNewPrivileges=true
|
||||||
|
RestrictSUIDSGID=true
|
||||||
|
ProtectSystem=strict
|
||||||
|
ProtectHome=tmpfs
|
||||||
|
BindReadOnlyPaths=%h/python-talk
|
||||||
|
ProtectProc=ptraceable
|
||||||
|
ProcSubset=pid
|
||||||
|
PrivateTmp=true
|
||||||
|
CapabilityBoundingSet=
|
||||||
|
PrivateDevices=true
|
||||||
|
ProtectClock=true
|
||||||
|
ProtectKernelLogs=true
|
||||||
|
ProtectControlGroups=true
|
||||||
|
ProtectKernelModules=true
|
||||||
|
SystemCallArchitectures=native
|
||||||
|
# python.
|
||||||
|
MemoryDenyWriteExecute=false
|
||||||
|
RestrictNamespaces=true
|
||||||
|
ProtectHostname=true
|
||||||
|
LockPersonality=true
|
||||||
|
ProtectKernelTunables=true
|
||||||
|
RestrictAddressFamilies=AF_INET
|
||||||
|
RestrictRealtime=true
|
||||||
|
PrivateUsers=true
|
||||||
|
SystemCallFilter=@system-service
|
||||||
|
SystemCallFilter=~@privileged @resources
|
||||||
|
UMask=0077
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=default.target
|
8
logs.py
8
logs.py
|
@ -30,12 +30,16 @@ def log(
|
||||||
print(
|
print(
|
||||||
f"[{level}]{'['+origin+']' if origin else ''}[{time}] {message}",
|
f"[{level}]{'['+origin+']' if origin else ''}[{time}] {message}",
|
||||||
file=stream,
|
file=stream,
|
||||||
|
flush=True,
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
for line in message.split("\n"):
|
for line in message.split("\n"):
|
||||||
log = log + f"\n[{level}]{'['+origin+']' if origin else ''}[{time}] {line}"
|
log = (
|
||||||
|
log + f"\r\n[{level}]{'['+origin+']' if origin else ''}[{time}] {line}"
|
||||||
|
)
|
||||||
print(
|
print(
|
||||||
f"[{level}]{'['+origin+']' if origin else ''}[{time}] {line}",
|
f"[{level}]{'['+origin+']' if origin else ''}[{time}] {line}",
|
||||||
file=stream,
|
file=stream,
|
||||||
|
flush=True,
|
||||||
)
|
)
|
||||||
return (log[5:] + "\n").encode("utf8")
|
return (log[5:] + "\r\n").encode("utf8")
|
||||||
|
|
313
server.py
313
server.py
|
@ -1,19 +1,27 @@
|
||||||
#!/usr/bin/python3
|
#!/usr/bin/python3
|
||||||
import os, sys, asyncio, re, signal
|
import os, sys, asyncio, re, signal, socket, random
|
||||||
from platform import uname
|
from platform import uname
|
||||||
from traceback import format_exc
|
from traceback import format_exc
|
||||||
from logs import log
|
from logs import log
|
||||||
|
|
||||||
|
|
||||||
class LinksDownException(Exception): ...
|
class LinkDownError(Exception): ...
|
||||||
|
|
||||||
|
|
||||||
|
class FailedLinkError(Exception): ...
|
||||||
|
|
||||||
|
|
||||||
class Globals: ...
|
class Globals: ...
|
||||||
|
|
||||||
|
|
||||||
# The two below lines are a hacky fix for python 3.10 asyncio
|
TimeoutErrors = (TimeoutError, asyncio.exceptions.TimeoutError)
|
||||||
TimoutError = TimeoutError
|
DisconnectErrors = (
|
||||||
TimeoutError = (TimoutError, asyncio.exceptions.TimeoutError)
|
ConnectionResetError,
|
||||||
|
BrokenPipeError,
|
||||||
|
IndexError,
|
||||||
|
UnicodeDecodeError,
|
||||||
|
*TimeoutErrors,
|
||||||
|
)
|
||||||
G = Globals()
|
G = Globals()
|
||||||
G.uniqueClients = 0
|
G.uniqueClients = 0
|
||||||
G.servers = {}
|
G.servers = {}
|
||||||
|
@ -28,8 +36,11 @@ G.killList = {}
|
||||||
G.outboundLinks = []
|
G.outboundLinks = []
|
||||||
G.S2SLogs = []
|
G.S2SLogs = []
|
||||||
G.cwlgd = False
|
G.cwlgd = False
|
||||||
|
G.NUL = "\x1e"
|
||||||
|
G.IRCStats = {"users": {"localMax": 0, "localCurrent": 0, "globalMax": 0, "globalCurrent": 0}, "servers": 0}
|
||||||
|
G.queue = []
|
||||||
saveLogs = True
|
saveLogs = True
|
||||||
address = "0.0.0.0"
|
address = "::"
|
||||||
# Try to load a message log, if one exists
|
# Try to load a message log, if one exists
|
||||||
try:
|
try:
|
||||||
G.msgs = __import__("cache").msgs
|
G.msgs = __import__("cache").msgs
|
||||||
|
@ -65,6 +76,7 @@ Accepted arguments:
|
||||||
--hostname=<string> - Sets the hostname for this server to use when talking to other servers. Defaults to the current system's hostname (limit of 16 chars)
|
--hostname=<string> - Sets the hostname for this server to use when talking to other servers. Defaults to the current system's hostname (limit of 16 chars)
|
||||||
--address=<IP> - Sets the IP to listen on, defaults to all addresses (0.0.0.0)
|
--address=<IP> - Sets the IP to listen on, defaults to all addresses (0.0.0.0)
|
||||||
--cwlgd - Crashes the server when outbound links go down
|
--cwlgd - Crashes the server when outbound links go down
|
||||||
|
-r, --random-port - Randomizes the port to listen on
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
{filename} --hostname=Fun-chat --link=chat.example.com:65048 --port=92628
|
{filename} --hostname=Fun-chat --link=chat.example.com:65048 --port=92628
|
||||||
|
@ -75,17 +87,20 @@ Examples:
|
||||||
elif arg in ["-l", "--no-logs"]:
|
elif arg in ["-l", "--no-logs"]:
|
||||||
log("Explicitly disabling saving of logs!")
|
log("Explicitly disabling saving of logs!")
|
||||||
saveLogs = False
|
saveLogs = False
|
||||||
elif arg.startswith("--link"):
|
elif arg.startswith("--link="):
|
||||||
G.outboundLinks.append((arg[7:].split(":")[0], int(arg.split(":")[1])))
|
G.outboundLinks.append((arg[7:].split(":")[0], int(arg.split(":")[1])))
|
||||||
elif arg.startswith("--hostname"):
|
elif arg.startswith("--hostname="):
|
||||||
G.remoteID = arg[11:]
|
G.remoteID = arg[11:]
|
||||||
if not G.remoteID or len(G.remoteID) > 16:
|
if not G.remoteID or len(G.remoteID) > 16:
|
||||||
raise ValueError
|
raise ValueError
|
||||||
elif arg.startswith("--address"):
|
elif arg.startswith("--address="):
|
||||||
address = arg[9:]
|
address = arg[10:]
|
||||||
elif arg.startswith("--cwlgd"):
|
elif arg in ["--cwlgd"]:
|
||||||
G.cwlgd = True
|
G.cwlgd = True
|
||||||
log("Server will crasg when outbound links go down!", "WARN")
|
log("Server will crash when outbound links go down!", "WARN")
|
||||||
|
elif arg in ["-r", "--random-port"]:
|
||||||
|
random.seed()
|
||||||
|
port = random.randint(10000, 60000)
|
||||||
else:
|
else:
|
||||||
log(f"Unrecognized argument {arg}!", "WARN")
|
log(f"Unrecognized argument {arg}!", "WARN")
|
||||||
except Exception:
|
except Exception:
|
||||||
|
@ -94,12 +109,31 @@ except Exception:
|
||||||
|
|
||||||
if not saveLogs:
|
if not saveLogs:
|
||||||
G.msgs.append(
|
G.msgs.append(
|
||||||
b"[00-00-0000 00:00:00] Notice: Logging is disabled on this server instance!\n"
|
b"[00-00-0000 00:00:00] Notice: Logging is disabled on this server instance!\r\n"
|
||||||
)
|
)
|
||||||
if len(G.remoteID) > 16:
|
if len(G.remoteID) > 16:
|
||||||
G.remoteID = G.remoteID[:15]
|
G.remoteID = G.remoteID[:15]
|
||||||
|
|
||||||
|
|
||||||
|
async def getMessage(reader, localQueue: list, size: int = 0, timeout: float = 0, sanitize: bool = False) -> str:
|
||||||
|
if not localQueue:
|
||||||
|
data = ""
|
||||||
|
if timeout and timeout > 0:
|
||||||
|
data = await asyncio.wait_for(reader.read(size), timeout)
|
||||||
|
else:
|
||||||
|
data = await reader.read(size)
|
||||||
|
localQueue = data.decode("utf8").strip().split("\r\n")
|
||||||
|
tmp = localQueue.pop(0)
|
||||||
|
if sanitize:
|
||||||
|
tmp = raw(tmp)
|
||||||
|
if size and size > 0:
|
||||||
|
other = tmp[size:]
|
||||||
|
tmp = tmp[:size]
|
||||||
|
if other:
|
||||||
|
localQueue.insert(0, other)
|
||||||
|
return tmp, localQueue
|
||||||
|
|
||||||
|
|
||||||
def raw(string: str) -> str:
|
def raw(string: str) -> str:
|
||||||
s = string.strip()
|
s = string.strip()
|
||||||
s = f"{s!r}"[1:-1].replace("\\\\", "\\")
|
s = f"{s!r}"[1:-1].replace("\\\\", "\\")
|
||||||
|
@ -118,33 +152,31 @@ def fmt(msg: str, name: str = "", action: bool = False) -> str:
|
||||||
|
|
||||||
async def handleClient(reader, writer):
|
async def handleClient(reader, writer):
|
||||||
try:
|
try:
|
||||||
|
name = ""
|
||||||
global G
|
global G
|
||||||
writer.write(b"Please identify yourself. Nick limit is 20 chars.\n")
|
writer.write(b"Please identify yourself. Nick limit is 20 chars.\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
name = raw((await reader.read(20)).decode("utf8"))
|
name, localQueue = await getMessage(reader, [], 20, 0, True)
|
||||||
|
localQueue = []
|
||||||
if len(name) > 20:
|
if len(name) > 20:
|
||||||
name = name[
|
name = name[
|
||||||
:19
|
:19
|
||||||
] # Really this is only possible if someone passes raw unicode as a nick, but let's clean it up anyways.
|
] # Really this is only possible if someone passes raw unicode as a nick, but let's clean it up anyways.
|
||||||
try:
|
|
||||||
await asyncio.wait_for(
|
|
||||||
reader.read(), 0.01
|
|
||||||
) # Silently consume the excess username data
|
|
||||||
except TimeoutError:
|
|
||||||
pass
|
|
||||||
if not name:
|
if not name:
|
||||||
writer.write(b"Nice try. Actually set a nick.\n")
|
writer.write(b"Nice try. Actually set a nick.\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return
|
return
|
||||||
if name.lower() in G.clientsConnected:
|
if name.lower() in G.clientsConnected:
|
||||||
writer.write(f"Nick ({name}) in use\n".encode("utf8"))
|
writer.write(f"Nick ({name}) in use\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return
|
return
|
||||||
if not name.startswith("S2S-"):
|
if name == "CAP LS 302":
|
||||||
|
pass # TODO: IRC Logic
|
||||||
|
elif not name.startswith("S2S-"):
|
||||||
G.clientsConnected[name.lower()] = G.remoteID
|
G.clientsConnected[name.lower()] = G.remoteID
|
||||||
msgIndex = 0
|
msgIndex = 0
|
||||||
G.uniqueClients += 1
|
G.uniqueClients += 1
|
||||||
|
@ -152,8 +184,7 @@ async def handleClient(reader, writer):
|
||||||
G.S2SLogs.append(("+", name, G.remoteID))
|
G.S2SLogs.append(("+", name, G.remoteID))
|
||||||
while 1:
|
while 1:
|
||||||
try:
|
try:
|
||||||
buffer = await asyncio.wait_for(reader.read(967), 0.1)
|
request, localQueue = await getMessage(reader, localQueue, 967, 0.1, True)
|
||||||
request = raw(buffer.decode("utf8"))
|
|
||||||
response = None
|
response = None
|
||||||
if request.startswith("/mes "):
|
if request.startswith("/mes "):
|
||||||
response = log(f"* {name}'s {request[5:]}")
|
response = log(f"* {name}'s {request[5:]}")
|
||||||
|
@ -163,25 +194,25 @@ async def handleClient(reader, writer):
|
||||||
G.S2SLogs.append(("A", (name, request[4:]), G.remoteID))
|
G.S2SLogs.append(("A", (name, request[4:]), G.remoteID))
|
||||||
elif request.startswith("/h"):
|
elif request.startswith("/h"):
|
||||||
writer.write(
|
writer.write(
|
||||||
b"""Command List:
|
b"""Command List:\r
|
||||||
/me <action> - Sends a special message so it looks like you did <action>
|
/me <action> - Sends a special message so it looks like you did <action>\r
|
||||||
/mes <action> - Same as /me, but adds a 's onto your nick
|
/mes <action> - Same as /me, but adds a 's onto your nick\r
|
||||||
/afk [reason] - Optional reason, short hand for `/me is afk [to [reason]]`
|
/afk [reason] - Optional reason, short hand for `/me is afk [to [reason]]`\r
|
||||||
/h, /help - Triggers this command listing
|
/h, /help - Triggers this command listing\r
|
||||||
/back - Shorthand for `/me is back`
|
/back - Shorthand for `/me is back`\r
|
||||||
/stats - Sends you some server stats
|
/stat - Sends you some server stats\r
|
||||||
/quit - Disconnects you from the server\n"""
|
/quit - Disconnects you from the server\r\n"""
|
||||||
)
|
)
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
elif request.startswith("/stats"):
|
elif request.startswith("/stat"):
|
||||||
writer.write(
|
writer.write(
|
||||||
f"""Server stats:
|
f"""Server stats:\r
|
||||||
Linked servers: {len(G.servers)}
|
Linked servers: {len(G.servers)}\r
|
||||||
Connected clients: {G.uniqueClients}
|
Connected clients: {G.uniqueClients}\r
|
||||||
Total Clients: {len(G.clientsConnected)}
|
Total Clients: {len(G.clientsConnected)}\r
|
||||||
Messages sent: {len(G.msgs)}
|
Messages sent: {len(G.msgs)}\r
|
||||||
|
\r
|
||||||
Please note that this is not network level statistics.\n""".encode(
|
Please note that this is not network level statistics.\r\n""".encode(
|
||||||
"utf8"
|
"utf8"
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
@ -205,14 +236,14 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
G.S2SLogs.append(("M", (name, request), G.remoteID))
|
G.S2SLogs.append(("M", (name, request), G.remoteID))
|
||||||
if response:
|
if response:
|
||||||
G.msgs.append(response)
|
G.msgs.append(response)
|
||||||
except TimeoutError:
|
except TimeoutErrors:
|
||||||
pass
|
pass
|
||||||
if msgIndex < len(G.msgs):
|
if msgIndex < len(G.msgs):
|
||||||
writer.writelines(G.msgs[msgIndex:])
|
writer.writelines(G.msgs[msgIndex:])
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
msgIndex = len(G.msgs)
|
msgIndex = len(G.msgs)
|
||||||
if name.lower() in G.killList and G.killList[name.lower()]:
|
if name.lower() in G.killList and G.killList[name.lower()]:
|
||||||
writer.write(b"Your client has been killed by the server\n")
|
writer.write(b"Your client has been killed by the server\r\n")
|
||||||
G.killList[name.lower()] = False
|
G.killList[name.lower()] = False
|
||||||
break
|
break
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
|
@ -230,55 +261,54 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return # Server is already "linked", drop the connection
|
return # Server is already "linked", drop the connection
|
||||||
if G.remoteID == sName: # Hey! you can't *also* be ***me***!
|
if G.remoteID == sName or not sName: # Hey! you can't *also* be ***me***!
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return # drop "us"
|
return # drop "us"
|
||||||
G.msgs.append(log(f"{sName} has linked to the network"))
|
G.msgs.append(log(f"{sName} has linked to the network"))
|
||||||
G.servers[sName] = {}
|
G.servers[sName] = {}
|
||||||
msgIndex = 0
|
msgIndex = 0
|
||||||
writer.write(b"I am awaiting your client listing.\n")
|
writer.write(b"I am awaiting your client listing.\r\n")
|
||||||
while 1:
|
while 1:
|
||||||
client = raw((await reader.read(1024)).decode("utf8"))
|
client, localQueue = await getMessage(reader, localQueue, 1024, 0, True)
|
||||||
if client == f"END OF CLIENT LISTING FROM {sName}":
|
if client == f"END OF CLIENT LISTING FROM {sName}":
|
||||||
break
|
break
|
||||||
if (
|
if (
|
||||||
client.lower() in G.servers[sName]
|
client.lower() in G.servers[sName]
|
||||||
or client.lower() in G.clientsConnected
|
or client.lower() in G.clientsConnected
|
||||||
):
|
):
|
||||||
writer.write(b"K Client rejected: Already exists\n")
|
writer.write(b"K Client rejected: Already exists\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
continue
|
continue
|
||||||
writer.write(b"I added that client.\n")
|
writer.write(b"I added that client.\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
G.msgs.append(log(f"{client} has connected from {sName}"))
|
G.msgs.append(log(f"{client} has connected from {sName}"))
|
||||||
G.servers[sName][client.lower()] = False
|
G.servers[sName][client.lower()] = False
|
||||||
G.S2SLogs.append(("+", client, sName))
|
G.S2SLogs.append(("+", client, sName))
|
||||||
writer.write(f"{G.remoteID}\n".encode("utf8"))
|
writer.write(f"{G.remoteID}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
await reader.read(1024)
|
await reader.read(1024)
|
||||||
for client in G.clientsConnected:
|
for client in G.clientsConnected:
|
||||||
writer.write(f"{client}\n".encode("utf8"))
|
writer.write(f"{client}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
resp = raw((await reader.read(1024)).decode("utf8"))
|
resp, localQueue = await getMessage(reader, localQueue, 1024, 0, True)
|
||||||
if resp.startswith("K"):
|
if resp.startswith("K"):
|
||||||
if G.clientsConnected[client] == G.remoteID:
|
if G.clientsConnected[client] == G.remoteID:
|
||||||
G.killList[client] = True
|
G.killList[client] = True
|
||||||
else:
|
else:
|
||||||
G.servers[G.clientsConnected[client]][client] = True
|
G.servers[G.clientsConnected[client]][client] = True
|
||||||
writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\n".encode("utf8"))
|
writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
for client in G.servers[sName]:
|
for client in G.servers[sName]:
|
||||||
G.clientsConnected[client] = sName
|
G.clientsConnected[client] = sName
|
||||||
msgInd = len(G.S2SLogs)
|
msgInd = len(G.S2SLogs)
|
||||||
while 1:
|
while 1:
|
||||||
try:
|
try:
|
||||||
rawMsg = await asyncio.wait_for(reader.read(967), 0.1)
|
buffer, localQueue = await getMessage(reader, localQueue, 1024, 0.1, False)
|
||||||
buffer = raw(rawMsg.decode("utf8"))
|
|
||||||
match buffer[0]:
|
match buffer[0]:
|
||||||
case "S": # Server notice
|
case "S": # Server notice
|
||||||
G.msgs.extend([log(buffer[2:])])
|
G.msgs.extend([log(buffer[2:])])
|
||||||
writer.write(b"I Mmm... Blueberries\n")
|
writer.write(b"I Mmm... Blueberries\r\n")
|
||||||
case "I":
|
case "I":
|
||||||
pass
|
pass
|
||||||
case "+":
|
case "+":
|
||||||
|
@ -290,9 +320,9 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
G.servers[sName][cName.lower()] = False
|
G.servers[sName][cName.lower()] = False
|
||||||
G.clientsConnected[cName.lower()] = sName
|
G.clientsConnected[cName.lower()] = sName
|
||||||
G.S2SLogs.append(("+", cName, sName))
|
G.S2SLogs.append(("+", cName, sName))
|
||||||
writer.write(b"I Mmm... Pineapples\n")
|
writer.write(b"I Mmm... Pineapples\r\n")
|
||||||
else:
|
else:
|
||||||
writer.write(f"K {cName}\n".encode("utf8"))
|
writer.write(f"K {cName}\r\n".encode("utf8"))
|
||||||
case "-":
|
case "-":
|
||||||
cName = buffer[2:]
|
cName = buffer[2:]
|
||||||
if G.clientsConnected.get(cName.lower(), None) == sName:
|
if G.clientsConnected.get(cName.lower(), None) == sName:
|
||||||
|
@ -302,41 +332,43 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
del G.servers[sName][cName.lower()]
|
del G.servers[sName][cName.lower()]
|
||||||
del G.clientsConnected[cName.lower()]
|
del G.clientsConnected[cName.lower()]
|
||||||
G.S2SLogs.append(("-", cName, sName))
|
G.S2SLogs.append(("-", cName, sName))
|
||||||
writer.write(b"I Mmm... Bananas\n")
|
writer.write(b"I Mmm... Bananas\r\n")
|
||||||
else:
|
else:
|
||||||
writer.write(
|
writer.write(
|
||||||
f"S Your server is LYING about who is connected to it. - {G.remoteID}, a fellow server\n".encode(
|
f"S Your server is LYING about who is connected to it. - {G.remoteID}, a fellow server\r\n".encode(
|
||||||
"utf8"
|
"utf8"
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
case "M":
|
case "M":
|
||||||
cName = buffer[2:].split("|", 1)[0]
|
cName = buffer[2:].split(G.NUL, 1)[0]
|
||||||
message = buffer[2:].split("|", 1)[1]
|
message = buffer[2:].split(G.NUL, 1)[1]
|
||||||
G.msgs.append(log(f" {cName}: {message}"))
|
G.msgs.append(log(f" {cName}: {message}"))
|
||||||
G.S2SLogs.append(("M", (cName, message), sName))
|
G.S2SLogs.append(("M", (cName, message), sName))
|
||||||
writer.write(
|
writer.write(
|
||||||
b"I Get these damn heretic ghost clients out of my store so i can buy my cult candles in peace.\n"
|
b"I Get these damn heretic ghost clients out of my store so i can buy my cult candles in peace.\r\n"
|
||||||
)
|
)
|
||||||
case "A":
|
case "A":
|
||||||
cName = buffer[2:].split("|", 1)[0]
|
cName = buffer[2:].split(G.NUL, 1)[0]
|
||||||
message = buffer[2:].split("|", 1)[1]
|
message = buffer[2:].split(G.NUL, 1)[1]
|
||||||
G.msgs.append(log(f"* {cName} {message}"))
|
G.msgs.append(log(f"* {cName} {message}"))
|
||||||
G.S2SLogs.append(("A", (cName, message), sName))
|
G.S2SLogs.append(("A", (cName, message), sName))
|
||||||
writer.write(b"I Mmm... Strawberries\n")
|
writer.write(b"I Mmm... Strawberries\r\n")
|
||||||
case "Q":
|
case "Q":
|
||||||
break
|
break
|
||||||
case "K":
|
case "K":
|
||||||
cName = buffer[2:]
|
cName = buffer[2:]
|
||||||
if G.clientsConnected[cName.lower()] == G.remoteID:
|
if not G.clientsConnected.get(cName.lower(), False):
|
||||||
|
pass # They don't exist, safe to ignore
|
||||||
|
elif G.clientsConnected[cName.lower()] == G.remoteID:
|
||||||
G.killList[cName.lower()] = True
|
G.killList[cName.lower()] = True
|
||||||
else:
|
else:
|
||||||
G.servers[G.clientsConnected[cName.lower()]][
|
G.servers[G.clientsConnected[cName.lower()]][
|
||||||
cName.lower()
|
cName.lower()
|
||||||
] = True
|
] = True
|
||||||
writer.write(b"I Mmm... Blood\n")
|
writer.write(b"I Mmm... Blood\r\n")
|
||||||
case _:
|
case _:
|
||||||
writer.write(
|
writer.write(
|
||||||
f"S Your server is doing drugs over here, sending me bullshit messages man - {G.remoteID}, A fellow server\n".encode(
|
f"S Your server is doing drugs over here, sending me bullshit messages man - {G.remoteID}, A fellow server\r\n".encode(
|
||||||
"utf8"
|
"utf8"
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
@ -345,12 +377,12 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
"WARN",
|
"WARN",
|
||||||
)
|
)
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
except TimeoutError:
|
except TimeoutErrors:
|
||||||
pass
|
pass
|
||||||
if any(G.servers[sName].values()):
|
if any(G.servers[sName].values()):
|
||||||
for name in G.servers[sName]:
|
for name in G.servers[sName]:
|
||||||
if G.servers[sName][name]:
|
if G.servers[sName][name]:
|
||||||
writer.write(f"K {name}\n".encode("utf8"))
|
writer.write(f"K {name}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
G.servers[sName][name] = False
|
G.servers[sName][name] = False
|
||||||
while msgInd < len(G.S2SLogs):
|
while msgInd < len(G.S2SLogs):
|
||||||
|
@ -360,14 +392,14 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
match type:
|
match type:
|
||||||
case "A":
|
case "A":
|
||||||
nick, msg = data
|
nick, msg = data
|
||||||
writer.write(f"A {nick}|{msg}\n".encode("utf8"))
|
writer.write(f"A {nick}{G.NUL}{msg}\r\n".encode("utf8"))
|
||||||
case "M":
|
case "M":
|
||||||
nick, msg = data
|
nick, msg = data
|
||||||
writer.write(f"M {nick}|{msg}\n".encode("utf8"))
|
writer.write(f"M {nick}{G.NUL}{msg}\r\n".encode("utf8"))
|
||||||
case "+":
|
case "+":
|
||||||
writer.write(f"+ {data}\n".encode("utf8"))
|
writer.write(f"+ {data}\r\n".encode("utf8"))
|
||||||
case "-":
|
case "-":
|
||||||
writer.write(f"- {data}\n".encode("utf8"))
|
writer.write(f"- {data}\r\n".encode("utf8"))
|
||||||
case _:
|
case _:
|
||||||
pass
|
pass
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
|
@ -381,11 +413,9 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
del G.clientsConnected[cName.lower()]
|
del G.clientsConnected[cName.lower()]
|
||||||
del G.servers[sName]
|
del G.servers[sName]
|
||||||
G.msgs.append(log(f"{sName} has de-linked from the network"))
|
G.msgs.append(log(f"{sName} has de-linked from the network"))
|
||||||
except (
|
except DisconnectErrors:
|
||||||
ConnectionResetError,
|
if not name:
|
||||||
BrokenPipeError,
|
return
|
||||||
IndexError,
|
|
||||||
): # Don't ask. IndexError needs to be caught here too.
|
|
||||||
if not name.startswith("S2S-"):
|
if not name.startswith("S2S-"):
|
||||||
G.uniqueClients -= 1
|
G.uniqueClients -= 1
|
||||||
G.msgs.append(log(f"{name} has disconnected from the server."))
|
G.msgs.append(log(f"{name} has disconnected from the server."))
|
||||||
|
@ -401,53 +431,58 @@ Please note that this is not network level statistics.\n""".encode(
|
||||||
pass
|
pass
|
||||||
del G.servers[name[4:]]
|
del G.servers[name[4:]]
|
||||||
G.msgs.append(log(f"{name[4:]} has de-linked from the network"))
|
G.msgs.append(log(f"{name[4:]} has de-linked from the network"))
|
||||||
|
except Exception:
|
||||||
|
Err = format_exc()
|
||||||
|
for line in Err.split("\r\n"):
|
||||||
|
log(line, "ERROR")
|
||||||
|
|
||||||
|
|
||||||
async def connectServer(hostname: str, port: int):
|
async def connectServer(hostname: str, port: int):
|
||||||
global G
|
global G
|
||||||
|
try:
|
||||||
reader, writer = await asyncio.open_connection(hostname, port)
|
reader, writer = await asyncio.open_connection(hostname, port)
|
||||||
await reader.read(1024)
|
ext, localQueue = await getMessage(reader, [], 1024)
|
||||||
writer.write(f"S2S-{G.remoteID}\n".encode("utf8"))
|
writer.write(f"S2S-{G.remoteID}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
await reader.read(1024)
|
ext, localQueue = await getMessage(reader, [], 1024)
|
||||||
for client in G.clientsConnected:
|
for client in G.clientsConnected:
|
||||||
writer.write(f"{client}\n".encode("utf8"))
|
writer.write(f"{client}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
resp = raw((await reader.read(1024)).decode("utf8"))
|
resp, localQueue = await getMessage(reader, localQueue, 1024, 0, True)
|
||||||
if resp.startswith("K"):
|
if resp.startswith("K"):
|
||||||
if G.clientsConnected[client] == G.remoteID:
|
if G.clientsConnected[client] == G.remoteID:
|
||||||
G.killList[client] = True
|
G.killList[client] = True
|
||||||
else:
|
else:
|
||||||
G.servers[G.clientsConnected[client]][client] = True
|
G.servers[G.clientsConnected[client]][client] = True
|
||||||
writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\n".encode("utf8"))
|
writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
rID = raw((await reader.read(16)).decode("utf8"))
|
rID = raw((await reader.read(16)).decode("utf8"))
|
||||||
try:
|
try:
|
||||||
await asyncio.wait_for(reader.read(1024), 0.1)
|
await asyncio.wait_for(reader.read(1024), 0.1)
|
||||||
except TimeoutError:
|
except TimeoutErrors:
|
||||||
pass
|
pass
|
||||||
if G.servers.get(rID, False) != False:
|
if G.servers.get(rID, False) != False:
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return
|
return
|
||||||
if G.remoteID == rID:
|
if G.remoteID == rID or not rID:
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
return
|
return
|
||||||
G.msgs.append(log(f"{rID} has linked to the network"))
|
G.msgs.append(log(f"{rID} has linked to the network"))
|
||||||
G.servers[rID] = {}
|
G.servers[rID] = {}
|
||||||
writer.write(b"I recieved your remote ID, now awaiting client listing\n")
|
writer.write(b"I recieved your remote ID, now awaiting client listing\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
# recieve client list from the other server
|
# recieve client list from the other server
|
||||||
while 1:
|
while 1:
|
||||||
client = raw((await reader.read(1024)).decode("utf8"))
|
client, localQueue = await getMessage(reader, localQueue, 1024, 0, True)
|
||||||
if client == f"END OF CLIENT LISTING FROM {rID}":
|
if client == f"END OF CLIENT LISTING FROM {rID}":
|
||||||
break
|
break
|
||||||
if client.lower() in G.servers[rID] or client.lower() in G.clientsConnected:
|
if client.lower() in G.servers[rID] or client.lower() in G.clientsConnected:
|
||||||
writer.write(b"K Client rejected: Already exists\n")
|
writer.write(b"K Client rejected: Already exists\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
continue
|
continue
|
||||||
writer.write(b"I added that client.\n")
|
writer.write(b"I added that client.\r\n")
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
G.msgs.append(log(f"{client} has connected from {rID}"))
|
G.msgs.append(log(f"{client} has connected from {rID}"))
|
||||||
G.servers[rID][client.lower()] = False
|
G.servers[rID][client.lower()] = False
|
||||||
|
@ -457,12 +492,11 @@ async def connectServer(hostname: str, port: int):
|
||||||
try:
|
try:
|
||||||
while 1:
|
while 1:
|
||||||
try:
|
try:
|
||||||
rawMsg = await asyncio.wait_for(reader.read(967), 0.1)
|
buffer, localQueue = await getMessage(reader, localQueue, 1024, 0.1, False)
|
||||||
buffer = raw(rawMsg.decode("utf8"))
|
|
||||||
match buffer[0]:
|
match buffer[0]:
|
||||||
case "S":
|
case "S":
|
||||||
G.msgs.extend([log(buffer[2:])])
|
G.msgs.extend([log(buffer[2:])])
|
||||||
writer.write(b"I Mmm... Blueberries\n")
|
writer.write(b"I Mmm... Blueberries\r\n")
|
||||||
case "I":
|
case "I":
|
||||||
pass
|
pass
|
||||||
case "+":
|
case "+":
|
||||||
|
@ -472,61 +506,68 @@ async def connectServer(hostname: str, port: int):
|
||||||
G.S2SLogs.append(("+", cName, rID))
|
G.S2SLogs.append(("+", cName, rID))
|
||||||
G.servers[rID][cName.lower()] = False
|
G.servers[rID][cName.lower()] = False
|
||||||
G.clientsConnected[cName.lower()] = rID
|
G.clientsConnected[cName.lower()] = rID
|
||||||
writer.write(b"I Mmm... Pineapples\n")
|
writer.write(b"I Mmm... Pineapples\r\n")
|
||||||
else:
|
else:
|
||||||
writer.write(f"K {cName}\n".encode("utf8"))
|
writer.write(f"K {cName}\r\n".encode("utf8"))
|
||||||
case "-":
|
case "-":
|
||||||
cName = buffer[2:]
|
cName = buffer[2:]
|
||||||
if G.clientsConnected.get(cName.lower(), None) == rID:
|
if G.clientsConnected.get(cName.lower(), None) == rID:
|
||||||
G.msgs.append(log(f"{cName} has disconnected from {rID}"))
|
G.msgs.append(
|
||||||
|
log(f"{cName} has disconnected from {rID}")
|
||||||
|
)
|
||||||
del G.servers[rID][cName.lower()]
|
del G.servers[rID][cName.lower()]
|
||||||
del G.clientsConnected[cName.lower()]
|
del G.clientsConnected[cName.lower()]
|
||||||
G.S2SLogs.append(("-", cName, rID))
|
G.S2SLogs.append(("-", cName, rID))
|
||||||
else:
|
else:
|
||||||
writer.write(
|
writer.write(
|
||||||
f"S Your server is LYING about who is connected to it. - {G.remoteID}, a fellow server\n".encode(
|
f"S Your server is LYING about who is connected to it. - {G.remoteID}, a fellow server\r\n".encode(
|
||||||
"utf8"
|
"utf8"
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
case "M":
|
case "M":
|
||||||
cName = buffer[2:].split("|", 1)[0]
|
cName = buffer[2:].split(G.NUL, 1)[0]
|
||||||
message = buffer[2:].split("|", 1)[1]
|
message = buffer[2:].split(G.NUL, 1)[1]
|
||||||
G.msgs.append(log(f" {cName}: {message}"))
|
G.msgs.append(log(f" {cName}: {message}"))
|
||||||
G.S2SLogs.append(("M", (cName, message), rID))
|
G.S2SLogs.append(("M", (cName, message), rID))
|
||||||
writer.write(
|
writer.write(
|
||||||
b"I Get these damn heretic ghost clients out of my store so i can buy my cult candles in peace.\n"
|
b"I Get these damn heretic ghost clients out of my store so i can buy my cult candles in peace.\r\n"
|
||||||
)
|
)
|
||||||
case "A":
|
case "A":
|
||||||
cName = buffer[2:].split("|", 1)[0]
|
cName = buffer[2:].split(G.NUL, 1)[0]
|
||||||
message = buffer[2:].split("|", 1)[1]
|
message = buffer[2:].split(G.NUL, 1)[1]
|
||||||
G.S2SLogs.append(("A", (cName, message), rID))
|
G.S2SLogs.append(("A", (cName, message), rID))
|
||||||
G.msgs.append(log(f"* {cName} {message}"))
|
G.msgs.append(log(f"* {cName} {message}"))
|
||||||
writer.write(b"I Mmm... Strawberries\n")
|
writer.write(b"I Mmm... Strawberries\r\n")
|
||||||
case "Q":
|
case "Q":
|
||||||
break
|
break
|
||||||
case "K":
|
case "K":
|
||||||
cName = buffer[2:]
|
cName = buffer[2:]
|
||||||
if G.clientsConnected[cName.lower()] == G.remoteID:
|
if not G.clientsConnected.get(cName.lower(), False):
|
||||||
|
pass # They don't exist, safe to ignore
|
||||||
|
elif G.clientsConnected[cName.lower()] == G.remoteID:
|
||||||
G.killList[cName.lower()] = True
|
G.killList[cName.lower()] = True
|
||||||
else:
|
else:
|
||||||
G.servers[G.clientsConnected[cName.lower()]][
|
G.servers[G.clientsConnected[cName.lower()]][
|
||||||
cName.lower()
|
cName.lower()
|
||||||
] = True
|
] = True
|
||||||
writer.write(b"I Mmm... Blood\n")
|
writer.write(b"I Mmm... Blood\r\n")
|
||||||
case _:
|
case _:
|
||||||
writer.write(
|
writer.write(
|
||||||
f"S Your server is doing drugs over here, sending me bullshit messages man - {G.remoteID}, A fellow server\n".encode(
|
f"S Your server is doing drugs over here, sending me bullshit messages man - {G.remoteID}, A fellow server\r\n".encode(
|
||||||
"utf8"
|
"utf8"
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
log(f"Recieved invalid message ({buffer}) from {sName}", "WARN")
|
log(
|
||||||
|
f"Recieved invalid message ({buffer}) from {rID}",
|
||||||
|
"WARN",
|
||||||
|
)
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
except TimeoutError:
|
except TimeoutErrors:
|
||||||
pass
|
pass
|
||||||
if any(G.servers[rID].values()):
|
if any(G.servers[rID].values()):
|
||||||
for name in G.servers[rID]:
|
for name in G.servers[rID]:
|
||||||
if G.servers[rID][name]:
|
if G.servers[rID][name]:
|
||||||
writer.write(f"K {name}\n".encode("utf8"))
|
writer.write(f"K {name}\r\n".encode("utf8"))
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
G.servers[rID][name] = False
|
G.servers[rID][name] = False
|
||||||
while msgInd < len(G.S2SLogs):
|
while msgInd < len(G.S2SLogs):
|
||||||
|
@ -535,20 +576,20 @@ async def connectServer(hostname: str, port: int):
|
||||||
match type:
|
match type:
|
||||||
case "A":
|
case "A":
|
||||||
nick, msg = data
|
nick, msg = data
|
||||||
writer.write(f"A {nick}|{msg}\n".encode("utf8"))
|
writer.write(f"A {nick}{G.NUL}{msg}\r\n".encode("utf8"))
|
||||||
case "M":
|
case "M":
|
||||||
nick, msg = data
|
nick, msg = data
|
||||||
writer.write(f"M {nick}|{msg}\n".encode("utf8"))
|
writer.write(f"M {nick}{G.NUL}{msg}\r\n".encode("utf8"))
|
||||||
case "+":
|
case "+":
|
||||||
writer.write(f"+ {data}\n".encode("utf8"))
|
writer.write(f"+ {data}\r\n".encode("utf8"))
|
||||||
case "-":
|
case "-":
|
||||||
writer.write(f"- {data}\n".encode("utf8"))
|
writer.write(f"- {data}\r\n".encode("utf8"))
|
||||||
case _:
|
case _:
|
||||||
pass
|
pass
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
msgInd += 1
|
msgInd += 1
|
||||||
if G.cwlgd:
|
if G.cwlgd:
|
||||||
raise LinksDownException
|
raise LinkDownError
|
||||||
await writer.drain()
|
await writer.drain()
|
||||||
writer.close()
|
writer.close()
|
||||||
await writer.wait_closed()
|
await writer.wait_closed()
|
||||||
|
@ -557,9 +598,9 @@ async def connectServer(hostname: str, port: int):
|
||||||
del G.clientsConnected[cName.lower()]
|
del G.clientsConnected[cName.lower()]
|
||||||
del G.servers[rID]
|
del G.servers[rID]
|
||||||
G.msgs.append(log(f"{rID} has de-linked from the network"))
|
G.msgs.append(log(f"{rID} has de-linked from the network"))
|
||||||
except (ConnectionResetError, BrokenPipeError, IndexError):
|
except DisconnectErrors:
|
||||||
if G.cwlgd:
|
if G.cwlgd:
|
||||||
raise LinksDownException
|
raise LinkDownError
|
||||||
for cName in G.servers[rID]:
|
for cName in G.servers[rID]:
|
||||||
G.msgs.append(log(f"{cName}'s server is going down"))
|
G.msgs.append(log(f"{cName}'s server is going down"))
|
||||||
try:
|
try:
|
||||||
|
@ -568,11 +609,35 @@ async def connectServer(hostname: str, port: int):
|
||||||
pass
|
pass
|
||||||
del G.servers[rID]
|
del G.servers[rID]
|
||||||
G.msgs.append(log(f"{rID} has de-linked from the network"))
|
G.msgs.append(log(f"{rID} has de-linked from the network"))
|
||||||
|
except OSError as E:
|
||||||
|
log("OSError: " + str(E), "ERROR")
|
||||||
|
if G.cwlgd:
|
||||||
|
raise FailedLinkError
|
||||||
|
except Exception:
|
||||||
|
Err = format_exc()
|
||||||
|
for line in Err.split("\r\n"):
|
||||||
|
log(line, "ERROR")
|
||||||
|
raise FailedLinkError
|
||||||
|
|
||||||
|
|
||||||
async def runServer(address: str, port: int):
|
async def runServer(address: str, port: int):
|
||||||
global G
|
global G
|
||||||
server = await asyncio.start_server(handleClient, address, port)
|
sock = None
|
||||||
|
try:
|
||||||
|
if ":" in address:
|
||||||
|
INET = socket.AF_INET6
|
||||||
|
else:
|
||||||
|
INET = socket.AF_INET
|
||||||
|
sock = socket.socket(INET, socket.SOCK_STREAM)
|
||||||
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # Reuse socket
|
||||||
|
sock.bind((address, port))
|
||||||
|
except OSError as E:
|
||||||
|
if E.errno != 97 or address != '::':
|
||||||
|
raise E
|
||||||
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||||
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) # Reuse socket
|
||||||
|
sock.bind(('0.0.0.0', port))
|
||||||
|
server = await asyncio.start_server(handleClient, sock=sock)
|
||||||
log(f"Listening on port {port}...")
|
log(f"Listening on port {port}...")
|
||||||
G.msgs.append(log("Server startup"))
|
G.msgs.append(log("Server startup"))
|
||||||
links = []
|
links = []
|
||||||
|
@ -582,12 +647,17 @@ async def runServer(address: str, port: int):
|
||||||
try:
|
try:
|
||||||
links.append(G.event.wait())
|
links.append(G.event.wait())
|
||||||
await asyncio.gather(*links)
|
await asyncio.gather(*links)
|
||||||
|
except LinkDownError:
|
||||||
|
G.msgs.append(log("Lost a server link, going down", "FATAL")[1:])
|
||||||
|
crash = True
|
||||||
|
except FailedLinkError:
|
||||||
|
G.msgs.append(log("Failed to establish a server link, going down", "FATAL")[1:])
|
||||||
except Exception:
|
except Exception:
|
||||||
crash = True
|
crash = True
|
||||||
G.msgs.append(log("Server crash", "FATAL")[1:])
|
G.msgs.append(log("Server crash", "FATAL")[1:])
|
||||||
log("Shutting down from Exception", "FATAL")
|
log("Shutting down from Exception", "FATAL")
|
||||||
Err = format_exc()
|
Err = format_exc()
|
||||||
for line in Err.split("\n"):
|
for line in Err.split("\r\n"):
|
||||||
log(line, "ERROR")
|
log(line, "ERROR")
|
||||||
finally:
|
finally:
|
||||||
if not crash:
|
if not crash:
|
||||||
|
@ -597,10 +667,11 @@ async def runServer(address: str, port: int):
|
||||||
# server.abort_clients()
|
# server.abort_clients()
|
||||||
if saveLogs:
|
if saveLogs:
|
||||||
with open("cache.py", "w") as cache:
|
with open("cache.py", "w") as cache:
|
||||||
cache.write(f"msgs = {G.msgs}\n")
|
cache.write(f"msgs = {G.msgs}\r\n")
|
||||||
log("Saved logs, exiting now.")
|
log("Saved logs, exiting now.")
|
||||||
else:
|
else:
|
||||||
log("Not saving logs, exiting now.")
|
log("Not saving logs, exiting now.")
|
||||||
|
sock.close()
|
||||||
|
|
||||||
|
|
||||||
class ServerInterruptException(KeyboardInterrupt): ...
|
class ServerInterruptException(KeyboardInterrupt): ...
|
||||||
|
|
Loading…
Reference in a new issue