Add utils scripts
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
nemunaire 2022-07-05 10:24:37 +02:00
parent a0af9f1289
commit 3e46eb68f7
2 changed files with 90 additions and 0 deletions

56
challenge-receive-email.py Executable file
View File

@ -0,0 +1,56 @@
#!/usr/bin/env python3
import base64
import email
import email.policy
import hashlib
import hmac
import json
import os
import re
import sys
import time
import urllib.request
def readmail(fp):
theEMail = fp.read()
msg = email.message_from_bytes(theEMail, policy=email.policy.default)
rp = msg.get("Return-Path") or "someone"
cnt = msg.get_content()
return msg, rp, cnt
if __name__ == '__main__':
# Parse command line arguments
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--url-token-validator', default="https://adlin.nemunai.re/api/recv-mail",
help="URL to token-validator")
parser.add_argument('--secret', default="adelina",
help="Secret used in token HMAC")
args = parser.parse_args()
msg, rp, cnt = readmail(sys.stdin.buffer)
hostnames = re.findall(r"<[^@]+@adlin-([^>]+)>", rp)
if len(hostnames) == 1:
h = hashlib.sha512()
h.update(cnt.encode())
req = urllib.request.Request(
url=args.url_token_validator,
method='POST',
headers={
"X-ADLIN-Authentication": base64.b64encode(hmac.digest(args.secret.encode(), str(int(time.mktime(time.localtime())/10)).encode(), hashlib.sha512)),
},
data=json.dumps({
"login": hostnames[0],
"token": h.hexdigest(),
}).encode(),
)
print("validating challenge for %s" % hostnames[0])
with urllib.request.urlopen(req) as f:
print(f.read().decode('utf-8'))

34
get_grades.py Executable file
View File

@ -0,0 +1,34 @@
#!/usr/bin/env python3
# Usage : ./get_grades.py | jq -r 'to_entries | map([.key, .value.TP1, .value.TP2, .value.TP3]) | .[] | @csv'
import base64
import hashlib
import hmac
import os
import sys
import time
import urllib.request
if __name__ == '__main__':
# Parse command line arguments
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--url-token-validator', default="https://adlin.nemunai.re/api/",
help="URL to token-validator")
parser.add_argument('--secret', default="adelina",
help="Secret used in token HMAC")
args = parser.parse_args()
req = urllib.request.Request(
url=args.url_token_validator + "grades",
method='GET',
headers={
"X-ADLIN-Authentication": base64.b64encode(hmac.digest(args.secret.encode(), str(int(time.mktime(time.localtime())/10)).encode(), hashlib.sha512)),
},
)
with urllib.request.urlopen(req) as f:
sys.stdout.write(f.read().decode('utf-8'))