mirror of
https://github.com/inspircd/inspircd.git
synced 2025-03-09 10:39:02 -04:00
130 lines
4.1 KiB
Python
Executable File
130 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# InspIRCd -- Internet Relay Chat Daemon
|
|
#
|
|
# Copyright (C) 2024 Val Lorentz <progval+git@progval.net>
|
|
# Copyright (C) 2024 Sadie Powell <sadie@witchery.services>
|
|
#
|
|
# This file is part of InspIRCd. InspIRCd is free software: you can
|
|
# redistribute it and/or modify it under the terms of the GNU General Public
|
|
# License as published by the Free Software Foundation, version 2.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
|
|
import io
|
|
import os
|
|
import sys
|
|
|
|
|
|
class UnrealDB:
|
|
def __init__(self, path):
|
|
self.error = None
|
|
try:
|
|
with open(sys.argv[1], mode="rb") as fh:
|
|
data = fh.read()
|
|
if data.startswith(b"UnrealIRCd-DB-v1"):
|
|
self.data = io.BytesIO(data[40:])
|
|
elif data.startswith(b"UnrealIRCd-DB"):
|
|
self.error = f"Unsupported database version: {data[0:32]}"
|
|
else:
|
|
self.data = io.BytesIO(data)
|
|
except OSError as e:
|
|
self.error = f"Read error: {e}"
|
|
|
|
def read_char(self):
|
|
return self.data.read(1).decode("utf-8")
|
|
|
|
def read_i16(self):
|
|
return int.from_bytes(self.data.read(2), byteorder="little")
|
|
|
|
def read_i32(self):
|
|
return int.from_bytes(self.data.read(4), byteorder="little")
|
|
|
|
def read_i64(self):
|
|
return int.from_bytes(self.data.read(8), byteorder="little")
|
|
|
|
def read_str(self):
|
|
len = self.read_i16()
|
|
if len == 0 or len == 0xFFFF:
|
|
return ""
|
|
return self.data.read(len).decode("utf-8")
|
|
|
|
|
|
def error(msg):
|
|
print(msg, file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
|
|
if len(sys.argv) < 3:
|
|
program = os.path.basename(__file__)
|
|
error(f"Usage: {program} <input-file> <output-file>")
|
|
|
|
db = UnrealDB(sys.argv[1])
|
|
if db.error:
|
|
error(db.error)
|
|
|
|
magic = db.read_i32()
|
|
if magic != 0x10101010:
|
|
error(f"Unrecognised database magic: {magic}")
|
|
|
|
version = db.read_i32()
|
|
if version != 4999:
|
|
error(f"Unrecognised TKL database version: {version}")
|
|
|
|
count = db.read_i64()
|
|
print(f"Converting {count} TKLs ...")
|
|
|
|
try:
|
|
with open(sys.argv[2], mode="w") as fh:
|
|
print("VERSION 1", file=fh)
|
|
for i in range(count):
|
|
type = db.read_char()
|
|
set_by = db.read_str()
|
|
set_at = db.read_i64()
|
|
expire_at = db.read_i64()
|
|
duration = expire_at - set_at
|
|
|
|
if type in "Ee":
|
|
username = db.read_str()
|
|
hostname = db.read_str()
|
|
etype = db.read_str() # unused
|
|
reason = db.read_str()
|
|
print(
|
|
f"LINE E {username}@{hostname} {set_by} {set_at} {duration} :{reason}",
|
|
file=fh,
|
|
)
|
|
elif type in "Ff":
|
|
method = db.read_str() # unused
|
|
pattern = db.read_str() # unused
|
|
target = db.read_str() # unused
|
|
action = db.read_char() # unused
|
|
reason = db.read_str() # unused
|
|
duration = db.read_i64() # unused
|
|
elif type in "GksZz":
|
|
username = db.read_str()
|
|
hostname = db.read_str()
|
|
reason = db.read_str()
|
|
line = "SHUN" if type == "s" else type.upper()
|
|
print(
|
|
f"LINE {line} {username}@{hostname} {set_by} {set_at} {duration} :{reason}",
|
|
file=fh,
|
|
)
|
|
elif type in "Qq":
|
|
etype = db.read_str() # unused
|
|
name = db.read_str()
|
|
reason = db.read_str()
|
|
print(f"LINE Q {name} {set_by} {set_at} {duration} :{reason}", file=fh)
|
|
else:
|
|
error(f"Unsupported row type: {type}")
|
|
|
|
except OSError as e:
|
|
error(f"Write error: {e}")
|