mirror of
https://bitbucket.org/jsuto/piler.git
synced 2024-12-24 18:50:12 +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