nathanwoodburn/woodburn-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bot: Add try for TLSA missing
All checks were successful
Build Docker / Build Docker (push) Successful in 16s
Details

Browse Source
This commit is contained in:
Nathan Woodburn 2023-09-27 18:11:27 +10:00
parent 1b9f4eafc8
commit 2ac733f33d
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
bot.py
View File

@ -196,7 +196,11 @@ async def ssl(ctx, domain: str):
message = message + "\n## TLSA Record from webserver:\n`" + tlsa_server + "`\n"
else:
message = message + "\n## Website Certificate:\n:x: No certificate found\n"
message = message + "\n## TLSA Record from webserver:\n:x: No certificate found\n"
try:
# Check for TLSA record
response = resolver.resolve("_443._tcp."+domain, "TLSA")
tlsa_records = []
@ -213,24 +217,25 @@ async def ssl(ctx, domain: str):
else:
message = message + "\n## Result:\n:x: TLSA record does not match certificate\n"
# If message is too long, send it in 2 messages
if (len(message) > 2000):
# Split on the last line under 2000 characters
message1 = message[:2000]
message1 = message1[:message1.rfind("\n")]
message2 = message[len(message1):]
except:
message = message + "\n## TLSA Records from DNS:\n:x: No TLSA record found\n"
message = message + "\n## Result:\n:x: No TLSA record found\n"
await ctx.response.send_message(message1)
await ctx.channel.send(message2)
# If message is too long, send it in 2 messages
if (len(message) > 2000):
# Split on the last line under 2000 characters
message1 = message[:2000]
message1 = message1[:message1.rfind("\n")]
message2 = message[len(message1):]
await ctx.response.send_message(message1)
await ctx.channel.send(message2)
else:
await ctx.response.send_message(message)
else:
ctx.response.send_message(f"No certificate found for {domain}")
return
await ctx.response.send_message(message)
# Catch all exceptions
except Exception as e: