mirror of https://github.com/dirtbags/moth.git
54 lines
1.0 KiB
Python
Executable File
54 lines
1.0 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import fcntl
|
|
import time
|
|
import os
|
|
from urllib.parse import quote, unquote
|
|
|
|
house = 'dirtbags'
|
|
|
|
passwdfn = '/var/lib/ctf/passwd'
|
|
|
|
teams = {}
|
|
built = 0
|
|
def build_teams():
|
|
global teams, built
|
|
|
|
if not os.path.exists(passwdfn):
|
|
return
|
|
if os.path.getmtime(passwdfn) <= built:
|
|
return
|
|
|
|
teams = {}
|
|
try:
|
|
f = open(passwdfn)
|
|
for line in f:
|
|
line = line.strip()
|
|
team, passwd = [unquote(v) for v in line.strip().split('\t')]
|
|
teams[team] = passwd
|
|
except IOError:
|
|
pass
|
|
built = time.time()
|
|
|
|
def validate(team):
|
|
build_teams()
|
|
|
|
def chkpasswd(team, passwd):
|
|
validate(team)
|
|
if teams.get(team) == passwd:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def exists(team):
|
|
validate(team)
|
|
if team == house:
|
|
return True
|
|
return team in teams
|
|
|
|
def add(team, passwd):
|
|
f = open(passwdfn, 'a')
|
|
fcntl.lockf(f, fcntl.LOCK_EX)
|
|
f.seek(0, 2)
|
|
f.write('%s\t%s\n' % (quote(team), quote(passwd)))
|