moth/mkpuzzles.py

115 lines
3.5 KiB
Python
Raw Normal View History

2009-09-29 15:36:25 -06:00
#! /usr/bin/env python3
import os
import shutil
import optparse
from ctf import config
2009-09-29 15:36:25 -06:00
p = optparse.OptionParser()
p.add_option('-p', '--puzzles', dest='puzzles', default='puzzles',
help='Directory containing puzzles')
p.add_option('-w', '--htmldir', dest='htmldir', default='puzzler',
help='Directory to write HTML puzzle tree')
p.add_option('-k', '--keyfile', dest='keyfile', default='puzzler.keys',
help='Where to write keys')
opts, args = p.parse_args()
keys = []
2009-10-08 16:42:26 -06:00
js = '''
<script type="text/javascript">
function readCookie(key) {
var s = key + '=';
var toks = document.cookie.split(';');
for (var i = 0; i < toks.length; i++) {
var tok = toks[i];
while (tok.charAt(0) == ' ') {
tok = tok.substring(1, tok.length);
}
if (tok.indexOf(s) == 0) {
return tok.substring(s.length, tok.length);
}
}
return null;
}
function getTeamInfo() {
team = readCookie('team');
passwd = readCookie('passwd');
if (team != null) {
document.getElementById("form").t.value = team;
}
if (passwd != null) {
document.getElementById("form").w.value = passwd;
}
}
window.onload = getTeamInfo;
</script>
'''
2009-09-29 15:36:25 -06:00
for cat in os.listdir(opts.puzzles):
dirname = os.path.join(opts.puzzles, cat)
for points in os.listdir(dirname):
pointsdir = os.path.join(dirname, points)
if not os.path.isdir(pointsdir):
continue
2009-09-29 15:36:25 -06:00
outdir = os.path.join(opts.htmldir, cat, points)
try:
os.makedirs(outdir)
except OSError:
pass
2009-09-29 15:36:25 -06:00
readme = ''
files = []
for fn in os.listdir(pointsdir):
path = os.path.join(pointsdir, fn)
if fn == 'key':
key = open(path, encoding='utf-8').readline().strip()
keys.append((cat, points, key))
elif fn == 'index.html':
readme = open(path, encoding='utf-8').read()
elif fn.endswith('~'):
pass
2009-09-29 15:36:25 -06:00
else:
files.append((fn, path))
title = '%s for %s points' % (cat, points)
f = open(os.path.join(outdir, 'index.html'), 'w', encoding='utf-8')
2009-10-08 16:42:26 -06:00
f.write(config.start_html(title, js))
2009-09-29 15:36:25 -06:00
if readme:
f.write('<div class="readme">%s</div>\n' % readme)
if files:
f.write('<ul>\n')
for fn, path in files:
if os.path.isdir(path):
shutil.copytree(path, os.path.join(outdir, fn))
else:
shutil.copy(path, outdir)
2009-10-05 11:03:27 -06:00
if not fn.startswith(','):
f.write('<li><a href="%s">%s</a></li>\n' % (fn, fn))
2009-09-29 15:36:25 -06:00
f.write('</ul>\n')
f.write('''
<form id="form" action="%(cgi)s" method="post">
2009-09-30 13:58:16 -06:00
<fieldset>
<legend>Your answer:</legend>
<input type="hidden" name="c" value="%(cat)s" />
<input type="hidden" name="p" value="%(points)s" />
Team: <input name="t" /><br />
Password: <input type="password" name="w" /><br />
Key: <input name="k" /><br />
<input type="submit" />
</fieldset>
2009-09-29 15:36:25 -06:00
</form>
''' % {'cgi': config.get('puzzler', 'cgi_url'),
'cat': cat,
'points': points})
2009-10-08 16:42:26 -06:00
f.write(config.end_html())
2009-09-29 15:36:25 -06:00
f = open(opts.keyfile, 'w', encoding='utf-8')
for key in keys:
f.write('%s\t%s\t%s\n' % key)