gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-03-12 00:45:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update smtp.py

Browse Source
This commit is contained in:
Namhyeon Go 2024-03-01 00:53:33 +09:00 committed by GitHub
parent 11aa65e53d
commit 39a8012323
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 34 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
smtp.py
View File

@ -11,9 +11,37 @@
import asyncore import asyncore
from smtpd import SMTPServer from smtpd import SMTPServer
import re
import json import json
import requests import requests
from decouple import config
def extract_credentials(url):
pattern = re.compile(r'(?P<scheme>\w+://)?(?P<username>[^:/]+):(?P<password>[^@]+)@(?P<url>.+)')
match = pattern.match(url)
if match:
scheme = match.group('scheme') if match.group('scheme') else 'https://'
username = match.group('username')
password = match.group('password')
url = match.group('url')
return username, password, scheme + url
else:
return None, None, url
try:
smtp_host = config('SMTP_HOST', default='127.0.0.1')
smtp_port = config('SMTP_PORT', default=25, cast=int)
_username, _password, server_url = extract_credentials(config('SERVER_URL'))
except KeyboardInterrupt:
print("\n[*] User has requested an interrupt")
print("[*] Application Exiting.....")
sys.exit()
auth = None
if _username:
auth = HTTPBasicAuth(_username, _password)
def jsonrpc2_create_id(data): def jsonrpc2_create_id(data):
return hashlib.sha1(json.dumps(data).encode(client_encoding)).hexdigest() return hashlib.sha1(json.dumps(data).encode(client_encoding)).hexdigest()
@ -64,7 +92,7 @@ class CaterpillarSMTPServer(SMTPServer):
to = v to = v
# build a data # build a data
_, raw_data = jsonrpc2_encode("relay_sendmail", { _, raw_data = jsonrpc2_encode('relay_sendmail', {
"to": to, "to": to,
"from": mailfrom, "from": mailfrom,
"subject": subject, "subject": subject,
@ -73,18 +101,18 @@ class CaterpillarSMTPServer(SMTPServer):
# send HTTP POST request # send HTTP POST request
try: try:
response = requests.post('https://example.org', data=raw_data) response = requests.post(server_url, data=raw_data, auth=auth)
response_json = response.json() response_json = response.json()
success = response_json.get('result', {}).get('success', False) success = response_json.get('result', {}).get('success', False)
if success: if success:
print("Email sent successfully.") print("[*] Email sent successfully.")
else: else:
print("Failed to send email.") print("[*] Failed to send email.")
except Exception as e: except Exception as e:
print("Failed to send email:", str(e)) print("[*] Failed to send email:", str(e))
# Start SMTP server # Start SMTP server
smtp_server = CaterpillarSMTPServer(('127.0.0.1', 25), None) smtp_server = CaterpillarSMTPServer((smtp_host, smtp_port), None)
# Start asynchronous event loop # Start asynchronous event loop
asyncore.loop() asyncore.loop()