mirror of
https://bitbucket.org/jsuto/piler.git
synced 2024-11-08 00:41:59 +01:00
added smtp testing utility
This commit is contained in:
parent
f70a69679d
commit
6e069f422f
349900
contrib/smtp/dictionary.txt
Normal file
349900
contrib/smtp/dictionary.txt
Normal file
File diff suppressed because it is too large
Load Diff
131
contrib/smtp/smtp-source.py
Executable file
131
contrib/smtp/smtp-source.py
Executable file
@ -0,0 +1,131 @@
|
|||||||
|
#!/usr/bin/python
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import smtplib
|
||||||
|
import random
|
||||||
|
import string
|
||||||
|
import quopri
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
import email.utils
|
||||||
|
|
||||||
|
from email.mime.text import MIMEText
|
||||||
|
from email.mime.multipart import MIMEMultipart
|
||||||
|
|
||||||
|
"""
|
||||||
|
check out http://blog.magiksys.net/generate-and-send-mail-with-python-tutorial
|
||||||
|
for a real professional solution
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
eol = "\r\n"
|
||||||
|
dictionary = "dictionary.txt"
|
||||||
|
checkpoint_for_newline = 300
|
||||||
|
|
||||||
|
|
||||||
|
def create_random_string(len=60):
|
||||||
|
return ''.join(random.choice(string.uppercase + string.digits) for i in range(len))
|
||||||
|
|
||||||
|
|
||||||
|
def create_msg(words, size=10000):
|
||||||
|
msg_text = ""
|
||||||
|
msg_size = 0
|
||||||
|
new_line_checkpoint = checkpoint_for_newline
|
||||||
|
|
||||||
|
while msg_size < size:
|
||||||
|
choice = random.SystemRandom().choice
|
||||||
|
s = str(choice(words)).replace('\n', ' ')
|
||||||
|
msg_text += s
|
||||||
|
msg_size += len(s)
|
||||||
|
if msg_size > new_line_checkpoint:
|
||||||
|
new_line_checkpoint += checkpoint_for_newline
|
||||||
|
msg_text += eol + eol
|
||||||
|
|
||||||
|
return msg_text
|
||||||
|
|
||||||
|
|
||||||
|
def create_text_mime_part(text, charset, encoding, boundary):
|
||||||
|
part = ""
|
||||||
|
|
||||||
|
part += "--" + boundary + eol
|
||||||
|
part += "Content-Type: text/plain; charset=\"" + charset + "\"" + eol
|
||||||
|
part += "Content-Transfer-Encoding: " + encoding + eol
|
||||||
|
part += eol
|
||||||
|
part += text + eol + eol
|
||||||
|
|
||||||
|
return part
|
||||||
|
|
||||||
|
|
||||||
|
def create_headers(args, boundary):
|
||||||
|
hdr = ""
|
||||||
|
|
||||||
|
hdr += "Content-Type: multipart/alternative;" + eol
|
||||||
|
hdr += " boundary=\"" + boundary + "\"" + eol
|
||||||
|
hdr += "MIME-Version: 1.0" + eol
|
||||||
|
hdr += "Subject: " + args.subject + eol
|
||||||
|
hdr += "From: " + args.sender + eol
|
||||||
|
hdr += "To: " + ', ' . join(args.rcpt) + eol
|
||||||
|
hdr += "Date: " + email.utils.formatdate(time.time(), localtime=True) + eol
|
||||||
|
hdr += "Message-ID: " + "<" + create_random_string(40) + "@" + args.helo + ">" + eol
|
||||||
|
hdr += eol + eol
|
||||||
|
|
||||||
|
return hdr
|
||||||
|
|
||||||
|
|
||||||
|
def create_message(words, args):
|
||||||
|
msg = ""
|
||||||
|
|
||||||
|
boundary = '_' + create_random_string() + '_'
|
||||||
|
|
||||||
|
msg += create_headers(args, boundary)
|
||||||
|
|
||||||
|
the_text = create_msg(words, args.msglen)
|
||||||
|
the_text = quopri.encodestring(the_text)
|
||||||
|
|
||||||
|
msg += create_text_mime_part(the_text, "iso-8859-2", "quoted-printable", boundary)
|
||||||
|
msg += "--" + boundary + eol
|
||||||
|
|
||||||
|
return msg
|
||||||
|
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
|
parser.add_argument("--helo", type=str, help="ehlo hostname", default="myhost.aaa.fu")
|
||||||
|
parser.add_argument("-f", "--sender", type=str, help="sender email", default="sender@aaa.fu")
|
||||||
|
parser.add_argument("-r", "--rcpt", type=str, nargs='+', help="recipient email(s)", default=["archive@aaa.fu"])
|
||||||
|
parser.add_argument("-c", "--count", type=int, help="number of emails to send", default=1)
|
||||||
|
parser.add_argument("-d", "--debug", type=int, help="debug level", default=0)
|
||||||
|
parser.add_argument("--session", type=int, help="number of emails to send in one smtp transaction", default=1)
|
||||||
|
parser.add_argument("-s", "--server", type=str, help="smtp server", required=True)
|
||||||
|
parser.add_argument("-p", "--port", type=int, help="smtp port", default=25)
|
||||||
|
parser.add_argument("--subject", type=str, help="subject", default="This is test subject")
|
||||||
|
parser.add_argument("-l", "--msglen", type=int, help="message length (approx.)", default=20000)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
|
with open(dictionary) as f:
|
||||||
|
words = f.readlines()
|
||||||
|
|
||||||
|
i = 0
|
||||||
|
while i < args.count:
|
||||||
|
server = smtplib.SMTP(args.server, args.port, args.helo, 10)
|
||||||
|
|
||||||
|
k = 0
|
||||||
|
|
||||||
|
while i < args.count and k < args.session:
|
||||||
|
message = create_message(words, args)
|
||||||
|
server.set_debuglevel(args.debug)
|
||||||
|
server.sendmail(args.sender, args.rcpt, message)
|
||||||
|
|
||||||
|
k += 1
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
if args.debug == 0:
|
||||||
|
sys.stdout.write('%s\r' % str(i))
|
||||||
|
sys.stdout.flush()
|
||||||
|
|
||||||
|
server.quit()
|
||||||
|
|
||||||
|
if args.debug == 0:
|
||||||
|
print
|
Loading…
Reference in New Issue
Block a user