838 lines
37 KiB
Python
838 lines
37 KiB
Python
# Author: Hubert Kario, (c) 2016, 2019
|
|
# Released under Gnu GPL v2.0, see LICENSE file for details
|
|
|
|
from __future__ import print_function
|
|
import traceback
|
|
import sys
|
|
import getopt
|
|
from itertools import chain
|
|
from random import sample
|
|
|
|
from tlsfuzzer.runner import Runner
|
|
from tlsfuzzer.messages import Connect, ClientHelloGenerator, \
|
|
ClientKeyExchangeGenerator, ChangeCipherSpecGenerator, \
|
|
FinishedGenerator, ApplicationDataGenerator, AlertGenerator, \
|
|
fuzz_message, ResetHandshakeHashes, Close, ResetRenegotiationInfo
|
|
from tlsfuzzer.expect import ExpectServerHello, ExpectCertificate, \
|
|
ExpectServerHelloDone, ExpectChangeCipherSpec, ExpectFinished, \
|
|
ExpectAlert, ExpectApplicationData, ExpectClose, \
|
|
ExpectServerKeyExchange
|
|
|
|
from tlslite.constants import CipherSuite, AlertLevel, AlertDescription, \
|
|
GroupName, ExtensionType, SignatureScheme
|
|
from tlslite.extensions import ALPNExtension, TLSExtension, \
|
|
SupportedGroupsExtension, \
|
|
SignatureAlgorithmsExtension, SignatureAlgorithmsCertExtension
|
|
from tlsfuzzer.utils.lists import natural_sort_keys
|
|
from tlsfuzzer.helpers import RSA_SIG_ALL
|
|
from tlsfuzzer.utils.ordered_dict import OrderedDict
|
|
|
|
|
|
version = 5
|
|
|
|
|
|
def help_msg():
|
|
print("Usage: <script-name> [-h hostname] [-p port] [[probe-name] ...]")
|
|
print(" -h hostname name of the host to run the test against")
|
|
print(" localhost by default")
|
|
print(" -p port port number to use for connection, 4433 by default")
|
|
print(" probe-name if present, will run only the probes with given")
|
|
print(" names and not all of them, e.g \"sanity\"")
|
|
print(" -e probe-name exclude the probe from the list of the ones run")
|
|
print(" may be specified multiple times")
|
|
print(" -x probe-name expect the probe to fail. When such probe passes despite being marked like this")
|
|
print(" it will be reported in the test summary and the whole script will fail.")
|
|
print(" May be specified multiple times.")
|
|
print(" -X message expect the `message` substring in exception raised during")
|
|
print(" execution of preceding expected failure probe")
|
|
print(" usage: [-x probe-name] [-X exception], order is compulsory!")
|
|
print(" -n num run 'num' or all(if 0) tests instead of default(all)")
|
|
print(" (\"sanity\" tests are always executed)")
|
|
print(" -d negotiate (EC)DHE instead of RSA key exchange")
|
|
print(" --help this message")
|
|
|
|
|
|
def add_dhe_extensions(extensions):
|
|
groups = [GroupName.secp256r1,
|
|
GroupName.ffdhe2048]
|
|
extensions[ExtensionType.supported_groups] = SupportedGroupsExtension()\
|
|
.create(groups)
|
|
extensions[ExtensionType.signature_algorithms] = \
|
|
SignatureAlgorithmsExtension().create(RSA_SIG_ALL)
|
|
extensions[ExtensionType.signature_algorithms_cert] = \
|
|
SignatureAlgorithmsCertExtension().create(RSA_SIG_ALL)
|
|
|
|
|
|
def main():
|
|
host = "localhost"
|
|
port = 4433
|
|
num_limit = None
|
|
run_exclude = set()
|
|
expected_failures = {}
|
|
last_exp_tmp = None
|
|
dhe = False
|
|
|
|
argv = sys.argv[1:]
|
|
opts, args = getopt.getopt(argv, "h:p:e:x:X:n:d", ["help"])
|
|
for opt, arg in opts:
|
|
if opt == '-h':
|
|
host = arg
|
|
elif opt == '-p':
|
|
port = int(arg)
|
|
elif opt == '-e':
|
|
run_exclude.add(arg)
|
|
elif opt == '-x':
|
|
expected_failures[arg] = None
|
|
last_exp_tmp = str(arg)
|
|
elif opt == '-X':
|
|
if not last_exp_tmp:
|
|
raise ValueError("-x has to be specified before -X")
|
|
expected_failures[last_exp_tmp] = str(arg)
|
|
elif opt == '-n':
|
|
num_limit = int(arg)
|
|
elif opt == '-d':
|
|
dhe = True
|
|
elif opt == '--help':
|
|
help_msg()
|
|
sys.exit(0)
|
|
else:
|
|
raise ValueError("Unknown option: {0}".format(opt))
|
|
|
|
if args:
|
|
run_only = set(args)
|
|
else:
|
|
run_only = None
|
|
|
|
conversations = {}
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
if dhe:
|
|
ext = {}
|
|
groups = [GroupName.secp256r1,
|
|
GroupName.ffdhe2048]
|
|
ext[ExtensionType.supported_groups] = SupportedGroupsExtension()\
|
|
.create(groups)
|
|
ext[ExtensionType.signature_algorithms] = \
|
|
SignatureAlgorithmsExtension().create(
|
|
[SignatureScheme.rsa_pkcs1_sha256,
|
|
SignatureScheme.rsa_pss_rsae_sha256])
|
|
ext[ExtensionType.signature_algorithms_cert] = \
|
|
SignatureAlgorithmsCertExtension().create(RSA_SIG_ALL)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ext = None
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectServerHello())
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ApplicationDataGenerator(
|
|
bytearray(b"GET / HTTP/1.0\r\n\r\n")))
|
|
node = node.add_child(ExpectApplicationData())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["sanity"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ApplicationDataGenerator(
|
|
bytearray(b"GET / HTTP/1.0\n\n")))
|
|
node = node.add_child(ExpectApplicationData())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["only http/1.1"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.')])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.no_application_protocol))
|
|
node.add_child(ExpectClose())
|
|
conversations["only http/1."] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.X')])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.no_application_protocol))
|
|
node.add_child(ExpectClose())
|
|
conversations["only http/1.X"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.'),
|
|
bytearray(b'http/1.1')])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ApplicationDataGenerator(
|
|
bytearray(b"GET / HTTP/1.0\n\n")))
|
|
node = node.add_child(ExpectApplicationData())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["http/1. and http/1.1"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'')])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node.add_child(ExpectClose())
|
|
conversations["empty element"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([])}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node.add_child(ExpectClose())
|
|
conversations["empty list"] = conversation
|
|
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: TLSExtension(extType=ExtensionType.alpn).create(bytearray())}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node.add_child(ExpectClose())
|
|
conversations["empty extension"] = conversation
|
|
|
|
# underflow length of "protocol_name_list" test
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
# the ALPN extension needs to be the last one for fuzz_message to work
|
|
# correctly
|
|
ext = OrderedDict()
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
ext[ExtensionType.alpn] = ALPNExtension().create([bytearray(b'http/1.1'),
|
|
bytearray(b'http/2')])
|
|
msg = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
# -17 is position of second byte in 2 byte long length of "protocol_name_list"
|
|
# setting it to value of 9 (bytes) will hide the second item in the "protocol_name_list"
|
|
node = node.add_child(fuzz_message(msg, substitutions={-17: 9}))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node = node.add_child(ExpectClose())
|
|
conversations["underflow length of protocol_name_list"] = conversation
|
|
|
|
# overflow length of "protocol_name_list" test
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
# the ALPN extension needs to be the last one for fuzz_message to work
|
|
# correctly
|
|
ext = OrderedDict()
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
ext[ExtensionType.alpn] = ALPNExtension().create([bytearray(b'http/1.1'),
|
|
bytearray(b'http/2')])
|
|
msg = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
# -17 is position of second byte in 2 byte long length of "protocol_name_list"
|
|
# setting it to value of 18 (bytes) will raise the length value for 2 more bytes
|
|
node = node.add_child(fuzz_message(msg, substitutions={-17: 18}))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node = node.add_child(ExpectClose())
|
|
conversations["overflow length of protocol_name_list"] = conversation
|
|
|
|
# overflow length of last item in "protocol_name_list" test
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
# the ALPN extension needs to be the last one for fuzz_message to work
|
|
# correctly
|
|
ext = OrderedDict()
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
ext[ExtensionType.alpn] = ALPNExtension().create([bytearray(b'http/1.1'),
|
|
bytearray(b'http/2')])
|
|
msg = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
# -7 is position of a length (1 byte long) for the last item in "protocol_name_list"
|
|
# setting it to value of 8 (bytes) will raise the length value for 2 more bytes
|
|
node = node.add_child(fuzz_message(msg, substitutions={-7: 8}))
|
|
node = node.add_child(ExpectAlert(AlertLevel.fatal,
|
|
AlertDescription.decode_error))
|
|
node = node.add_child(ExpectClose())
|
|
conversations["overflow length of last item"] = conversation
|
|
|
|
# renegotiation with protocol change
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
# 2nd handshake
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/2')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, session_id=bytearray(0), extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/2')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["renegotiation with protocol change"] = conversation
|
|
|
|
# renegotiation without protocol change
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
# 2nd handshake
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, session_id=bytearray(0), extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["renegotiation without protocol change"] = conversation
|
|
|
|
# renegotiation 2nd handshake alpn
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
# 2nd handshake
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, session_id=bytearray(0), extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["renegotiation 2nd handshake alpn"] = conversation
|
|
|
|
# resumption without alpn change
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
close = ExpectClose()
|
|
node.next_sibling = close
|
|
node = node.add_child(ExpectClose())
|
|
node = node.add_child(Close())
|
|
node = node.add_child(Connect(host, port))
|
|
close.add_child(node)
|
|
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
node = node.add_child(ResetRenegotiationInfo())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info:None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext, resume=True))
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
node = node.add_child(ExpectClose())
|
|
|
|
conversations["resumption without alpn change"] = conversation
|
|
|
|
# resumption with alpn change
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
close = ExpectClose()
|
|
node.next_sibling = close
|
|
node = node.add_child(ExpectClose())
|
|
node = node.add_child(Close())
|
|
node = node.add_child(Connect(host, port))
|
|
close.add_child(node)
|
|
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
node = node.add_child(ResetRenegotiationInfo())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'h2')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info:None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'h2')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext, resume=True))
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
node = node.add_child(ExpectClose())
|
|
|
|
conversations["resumption with alpn change"] = conversation
|
|
|
|
# resumption with alpn
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
ext = {ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
close = ExpectClose()
|
|
node.next_sibling = close
|
|
node = node.add_child(ExpectClose())
|
|
node = node.add_child(Close())
|
|
node = node.add_child(Connect(host, port))
|
|
close.add_child(node)
|
|
|
|
node = node.add_child(ResetHandshakeHashes())
|
|
node = node.add_child(ResetRenegotiationInfo())
|
|
ext = {ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')]),
|
|
ExtensionType.renegotiation_info:None}
|
|
if dhe:
|
|
add_dhe_extensions(ext)
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info:None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext, resume=True))
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
node = node.add_child(ExpectClose())
|
|
|
|
conversations["resumption with alpn"] = conversation
|
|
|
|
# 16269 byte long array and 255 byte long items test
|
|
# Client Hello longer than 2^14 bytes
|
|
conversation = Connect(host, port)
|
|
node = conversation
|
|
proto = bytearray(b"A" * 255)
|
|
lista = []
|
|
lista.append(proto)
|
|
# 63 items 255 bytes long + 1 item 195 bytes long + 1 item 8 byte long (http/1.1)
|
|
for p in range(1, 63):
|
|
lista.append(proto)
|
|
if dhe:
|
|
# (in DHE we send more extensions and longer cipher suite list, so the
|
|
# extension has to be shorter)
|
|
# 145 + 1 byte to reproduce the issue
|
|
lista.append(bytearray(b'B' * 146))
|
|
else:
|
|
# 195 + 1 byte to reproduce the issue
|
|
lista.append(bytearray(b'B' * 196))
|
|
lista.append(bytearray(b'http/1.1'))
|
|
ext = {ExtensionType.alpn: ALPNExtension().create(lista)}
|
|
if dhe:
|
|
groups = [GroupName.secp256r1,
|
|
GroupName.ffdhe2048]
|
|
ext[ExtensionType.supported_groups] = SupportedGroupsExtension()\
|
|
.create(groups)
|
|
ext[ExtensionType.signature_algorithms] = \
|
|
SignatureAlgorithmsExtension().create(
|
|
[SignatureScheme.rsa_pkcs1_sha256,
|
|
SignatureScheme.rsa_pss_rsae_sha256])
|
|
ext[ExtensionType.signature_algorithms_cert] = \
|
|
SignatureAlgorithmsCertExtension().create(RSA_SIG_ALL)
|
|
ciphers = [CipherSuite.TLS_ECDHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_DHE_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
else:
|
|
ciphers = [CipherSuite.TLS_RSA_WITH_AES_128_CBC_SHA,
|
|
CipherSuite.TLS_EMPTY_RENEGOTIATION_INFO_SCSV]
|
|
node = node.add_child(ClientHelloGenerator(ciphers, extensions=ext))
|
|
ext = {ExtensionType.renegotiation_info: None,
|
|
ExtensionType.alpn: ALPNExtension().create([bytearray(b'http/1.1')])}
|
|
node = node.add_child(ExpectServerHello(extensions=ext))
|
|
node = node.add_child(ExpectCertificate())
|
|
if dhe:
|
|
node = node.add_child(ExpectServerKeyExchange())
|
|
node = node.add_child(ExpectServerHelloDone())
|
|
node = node.add_child(ClientKeyExchangeGenerator())
|
|
node = node.add_child(ChangeCipherSpecGenerator())
|
|
node = node.add_child(FinishedGenerator())
|
|
node = node.add_child(ExpectChangeCipherSpec())
|
|
node = node.add_child(ExpectFinished())
|
|
node = node.add_child(AlertGenerator(AlertLevel.warning,
|
|
AlertDescription.close_notify))
|
|
node = node.add_child(ExpectAlert())
|
|
node.next_sibling = ExpectClose()
|
|
conversations["16269 byte long array"] = conversation
|
|
|
|
# run the conversation
|
|
good = 0
|
|
bad = 0
|
|
xfail = 0
|
|
xpass = 0
|
|
failed = []
|
|
xpassed = []
|
|
if not num_limit:
|
|
num_limit = len(conversations)
|
|
|
|
# make sure that sanity test is run first and last
|
|
# to verify that server was running and kept running throughout
|
|
sanity_tests = [('sanity', conversations['sanity'])]
|
|
if run_only:
|
|
if num_limit > len(run_only):
|
|
num_limit = len(run_only)
|
|
regular_tests = [(k, v) for k, v in conversations.items() if
|
|
k in run_only]
|
|
else:
|
|
regular_tests = [(k, v) for k, v in conversations.items() if
|
|
(k != 'sanity') and k not in run_exclude]
|
|
sampled_tests = sample(regular_tests, min(num_limit, len(regular_tests)))
|
|
ordered_tests = chain(sanity_tests, sampled_tests, sanity_tests)
|
|
|
|
for c_name, c_test in ordered_tests:
|
|
if run_only and c_name not in run_only or c_name in run_exclude:
|
|
continue
|
|
print("{0} ...".format(c_name))
|
|
|
|
runner = Runner(c_test)
|
|
|
|
res = True
|
|
exception = None
|
|
try:
|
|
runner.run()
|
|
except Exception as exp:
|
|
exception = exp
|
|
print("Error while processing")
|
|
print(traceback.format_exc())
|
|
res = False
|
|
|
|
if c_name in expected_failures:
|
|
if res:
|
|
xpass += 1
|
|
xpassed.append(c_name)
|
|
print("XPASS-expected failure but test passed\n")
|
|
else:
|
|
if expected_failures[c_name] is not None and \
|
|
expected_failures[c_name] not in str(exception):
|
|
bad += 1
|
|
failed.append(c_name)
|
|
print("Expected error message: {0}\n"
|
|
.format(expected_failures[c_name]))
|
|
else:
|
|
xfail += 1
|
|
print("OK-expected failure\n")
|
|
else:
|
|
if res:
|
|
good += 1
|
|
print("OK\n")
|
|
else:
|
|
bad += 1
|
|
failed.append(c_name)
|
|
|
|
print("ALPN extension tests")
|
|
print("Verify that the ALPN extenion is supported in the server and has")
|
|
print("correct error handling.")
|
|
|
|
print("Test end")
|
|
print(20 * '=')
|
|
print("version: {0}".format(version))
|
|
print(20 * '=')
|
|
print("TOTAL: {0}".format(len(sampled_tests) + 2*len(sanity_tests)))
|
|
print("SKIP: {0}".format(len(run_exclude.intersection(conversations.keys()))))
|
|
print("PASS: {0}".format(good))
|
|
print("XFAIL: {0}".format(xfail))
|
|
print("FAIL: {0}".format(bad))
|
|
print("XPASS: {0}".format(xpass))
|
|
print(20 * '=')
|
|
sort = sorted(xpassed ,key=natural_sort_keys)
|
|
if len(sort):
|
|
print("XPASSED:\n\t{0}".format('\n\t'.join(repr(i) for i in sort)))
|
|
sort = sorted(failed, key=natural_sort_keys)
|
|
if len(sort):
|
|
print("FAILED:\n\t{0}".format('\n\t'.join(repr(i) for i in sort)))
|
|
|
|
if bad or xpass:
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|