1083 lines
44 KiB
Python
1083 lines
44 KiB
Python
|
# This file is part of Scapy
|
||
|
|
# Copyright (C) 2007, 2008, 2009 Arnaud Ebalard
|
||
|
|
# 2015, 2016, 2017 Maxence Tury
|
||
|
|
# 2019 Romain Perez
|
||
|
|
# This program is published under a GPLv2 license
|
||
|
|
|
||
|
|
"""
|
||
|
|
TLS session handler.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import struct
|
||
|
|
|
||
|
|
from scapy.config import conf
|
||
|
|
from scapy.compat import raw
|
||
|
|
import scapy.modules.six as six
|
||
|
|
from scapy.error import log_runtime, warning
|
||
|
|
from scapy.packet import Packet
|
||
|
|
from scapy.pton_ntop import inet_pton
|
||
|
|
from scapy.sessions import DefaultSession
|
||
|
|
from scapy.utils import repr_hex, strxor
|
||
|
|
from scapy.layers.inet import TCP
|
||
|
|
from scapy.layers.tls.crypto.compression import Comp_NULL
|
||
|
|
from scapy.layers.tls.crypto.hkdf import TLS13_HKDF
|
||
|
|
from scapy.layers.tls.crypto.prf import PRF
|
||
|
|
|
||
|
|
# Note the following import may happen inside connState.__init__()
|
||
|
|
# in order to avoid to avoid cyclical dependencies.
|
||
|
|
# from scapy.layers.tls.crypto.suites import TLS_NULL_WITH_NULL_NULL
|
||
|
|
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# Connection states #
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
class connState(object):
|
||
|
|
"""
|
||
|
|
From RFC 5246, section 6.1:
|
||
|
|
A TLS connection state is the operating environment of the TLS Record
|
||
|
|
Protocol. It specifies a compression algorithm, an encryption
|
||
|
|
algorithm, and a MAC algorithm. In addition, the parameters for
|
||
|
|
these algorithms are known: the MAC key and the bulk encryption keys
|
||
|
|
for the connection in both the read and the write directions.
|
||
|
|
Logically, there are always four connection states outstanding: the
|
||
|
|
current read and write states, and the pending read and write states.
|
||
|
|
All records are processed under the current read and write states.
|
||
|
|
The security parameters for the pending states can be set by the TLS
|
||
|
|
Handshake Protocol, and the ChangeCipherSpec can selectively make
|
||
|
|
either of the pending states current, in which case the appropriate
|
||
|
|
current state is disposed of and replaced with the pending state; the
|
||
|
|
pending state is then reinitialized to an empty state. It is illegal
|
||
|
|
to make a state that has not been initialized with security
|
||
|
|
parameters a current state. The initial current state always
|
||
|
|
specifies that no encryption, compression, or MAC will be used.
|
||
|
|
|
||
|
|
(For practical reasons, Scapy scraps these two last lines, through the
|
||
|
|
implementation of dummy ciphers and MAC with TLS_NULL_WITH_NULL_NULL.)
|
||
|
|
|
||
|
|
These attributes and behaviours are mostly mapped in this class.
|
||
|
|
Also, note that Scapy may make a current state out of a pending state
|
||
|
|
which has been initialized with dummy security parameters. We need
|
||
|
|
this in order to know when the content of a TLS message is encrypted,
|
||
|
|
whether we possess the right keys to decipher/verify it or not.
|
||
|
|
For instance, when Scapy parses a CKE without knowledge of any secret,
|
||
|
|
and then a CCS, it needs to know that the following Finished
|
||
|
|
is encrypted and signed according to a new cipher suite, even though
|
||
|
|
it cannot decipher the message nor verify its integrity.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,
|
||
|
|
connection_end="server",
|
||
|
|
read_or_write="read",
|
||
|
|
seq_num=0,
|
||
|
|
compression_alg=Comp_NULL,
|
||
|
|
ciphersuite=None,
|
||
|
|
tls_version=0x0303):
|
||
|
|
|
||
|
|
self.tls_version = tls_version
|
||
|
|
|
||
|
|
# It is the user's responsibility to keep the record seq_num
|
||
|
|
# under 2**64-1. If this value gets maxed out, the TLS class in
|
||
|
|
# record.py will crash when trying to encode it with struct.pack().
|
||
|
|
self.seq_num = seq_num
|
||
|
|
|
||
|
|
self.connection_end = connection_end
|
||
|
|
self.row = read_or_write
|
||
|
|
|
||
|
|
if ciphersuite is None:
|
||
|
|
from scapy.layers.tls.crypto.suites import TLS_NULL_WITH_NULL_NULL
|
||
|
|
ciphersuite = TLS_NULL_WITH_NULL_NULL
|
||
|
|
self.ciphersuite = ciphersuite(tls_version=tls_version)
|
||
|
|
|
||
|
|
if not self.ciphersuite.usable:
|
||
|
|
warning("TLS ciphersuite not usable. Is the cryptography Python module installed ?") # noqa: E501
|
||
|
|
return
|
||
|
|
|
||
|
|
self.compression = compression_alg()
|
||
|
|
self.key_exchange = ciphersuite.kx_alg()
|
||
|
|
self.cipher = ciphersuite.cipher_alg()
|
||
|
|
self.hash = ciphersuite.hash_alg()
|
||
|
|
|
||
|
|
if tls_version > 0x0200:
|
||
|
|
if ciphersuite.cipher_alg.type == "aead":
|
||
|
|
self.hmac = None
|
||
|
|
self.mac_len = self.cipher.tag_len
|
||
|
|
else:
|
||
|
|
self.hmac = ciphersuite.hmac_alg()
|
||
|
|
self.mac_len = self.hmac.hmac_len
|
||
|
|
else:
|
||
|
|
self.hmac = ciphersuite.hmac_alg() # should be Hmac_NULL
|
||
|
|
self.mac_len = self.hash.hash_len
|
||
|
|
|
||
|
|
if tls_version >= 0x0304:
|
||
|
|
self.hkdf = TLS13_HKDF(self.hash.name.lower())
|
||
|
|
else:
|
||
|
|
self.prf = PRF(ciphersuite.hash_alg.name, tls_version)
|
||
|
|
|
||
|
|
def debug_repr(self, name, secret):
|
||
|
|
if conf.debug_tls and secret:
|
||
|
|
log_runtime.debug("TLS: %s %s %s: %s",
|
||
|
|
self.connection_end,
|
||
|
|
self.row,
|
||
|
|
name,
|
||
|
|
repr_hex(secret))
|
||
|
|
|
||
|
|
def derive_keys(self,
|
||
|
|
client_random=b"",
|
||
|
|
server_random=b"",
|
||
|
|
master_secret=b""):
|
||
|
|
# XXX Can this be called over a non-usable suite? What happens then?
|
||
|
|
|
||
|
|
cs = self.ciphersuite
|
||
|
|
|
||
|
|
# Derive the keys according to the cipher type and protocol version
|
||
|
|
key_block = self.prf.derive_key_block(master_secret,
|
||
|
|
server_random,
|
||
|
|
client_random,
|
||
|
|
cs.key_block_len)
|
||
|
|
|
||
|
|
# When slicing the key_block, keep the right half of the material
|
||
|
|
skip_first = False
|
||
|
|
if ((self.connection_end == "client" and self.row == "read") or
|
||
|
|
(self.connection_end == "server" and self.row == "write")):
|
||
|
|
skip_first = True
|
||
|
|
|
||
|
|
pos = 0
|
||
|
|
cipher_alg = cs.cipher_alg
|
||
|
|
|
||
|
|
# MAC secret (for block and stream ciphers)
|
||
|
|
if (cipher_alg.type == "stream") or (cipher_alg.type == "block"):
|
||
|
|
start = pos
|
||
|
|
if skip_first:
|
||
|
|
start += cs.hmac_alg.key_len
|
||
|
|
end = start + cs.hmac_alg.key_len
|
||
|
|
mac_secret = key_block[start:end]
|
||
|
|
self.debug_repr("mac_secret", mac_secret)
|
||
|
|
pos += 2 * cs.hmac_alg.key_len
|
||
|
|
else:
|
||
|
|
mac_secret = None
|
||
|
|
|
||
|
|
# Cipher secret
|
||
|
|
start = pos
|
||
|
|
if skip_first:
|
||
|
|
start += cipher_alg.key_len
|
||
|
|
end = start + cipher_alg.key_len
|
||
|
|
cipher_secret = key_block[start:end]
|
||
|
|
if cs.kx_alg.export:
|
||
|
|
reqLen = cipher_alg.expanded_key_len
|
||
|
|
cipher_secret = self.prf.postprocess_key_for_export(cipher_secret,
|
||
|
|
client_random,
|
||
|
|
server_random,
|
||
|
|
self.connection_end, # noqa: E501
|
||
|
|
self.row,
|
||
|
|
reqLen)
|
||
|
|
self.debug_repr("cipher_secret", cipher_secret)
|
||
|
|
pos += 2 * cipher_alg.key_len
|
||
|
|
|
||
|
|
# Implicit IV (for block and AEAD ciphers)
|
||
|
|
start = pos
|
||
|
|
if cipher_alg.type == "block":
|
||
|
|
if skip_first:
|
||
|
|
start += cipher_alg.block_size
|
||
|
|
end = start + cipher_alg.block_size
|
||
|
|
elif cipher_alg.type == "aead":
|
||
|
|
if skip_first:
|
||
|
|
start += cipher_alg.fixed_iv_len
|
||
|
|
end = start + cipher_alg.fixed_iv_len
|
||
|
|
|
||
|
|
# Now we have the secrets, we can instantiate the algorithms
|
||
|
|
if cs.hmac_alg is None: # AEAD
|
||
|
|
self.hmac = None
|
||
|
|
self.mac_len = cipher_alg.tag_len
|
||
|
|
else:
|
||
|
|
self.hmac = cs.hmac_alg(mac_secret)
|
||
|
|
self.mac_len = self.hmac.hmac_len
|
||
|
|
|
||
|
|
if cipher_alg.type == "stream":
|
||
|
|
cipher = cipher_alg(cipher_secret)
|
||
|
|
elif cipher_alg.type == "block":
|
||
|
|
# We set an IV every time, even though it does not matter for
|
||
|
|
# TLS 1.1+ as it requires an explicit IV. Indeed the cipher.iv
|
||
|
|
# would get updated in TLS.post_build() or TLS.pre_dissect().
|
||
|
|
iv = key_block[start:end]
|
||
|
|
if cs.kx_alg.export:
|
||
|
|
reqLen = cipher_alg.block_size
|
||
|
|
iv = self.prf.generate_iv_for_export(client_random,
|
||
|
|
server_random,
|
||
|
|
self.connection_end,
|
||
|
|
self.row,
|
||
|
|
reqLen)
|
||
|
|
cipher = cipher_alg(cipher_secret, iv)
|
||
|
|
self.debug_repr("block iv", iv)
|
||
|
|
elif cipher_alg.type == "aead":
|
||
|
|
fixed_iv = key_block[start:end]
|
||
|
|
nonce_explicit_init = 0
|
||
|
|
# If you ever wanted to set a random nonce_explicit, use this:
|
||
|
|
# exp_bit_len = cipher_alg.nonce_explicit_len * 8
|
||
|
|
# nonce_explicit_init = random.randint(0, 2**exp_bit_len - 1)
|
||
|
|
cipher = cipher_alg(cipher_secret, fixed_iv, nonce_explicit_init)
|
||
|
|
self.debug_repr("aead fixed iv", fixed_iv)
|
||
|
|
self.cipher = cipher
|
||
|
|
|
||
|
|
def sslv2_derive_keys(self, key_material):
|
||
|
|
"""
|
||
|
|
There is actually only one key, the CLIENT-READ-KEY or -WRITE-KEY.
|
||
|
|
|
||
|
|
Note that skip_first is opposite from the one with SSLv3 derivation.
|
||
|
|
|
||
|
|
Also, if needed, the IV should be set elsewhere.
|
||
|
|
"""
|
||
|
|
skip_first = True
|
||
|
|
if ((self.connection_end == "client" and self.row == "read") or
|
||
|
|
(self.connection_end == "server" and self.row == "write")):
|
||
|
|
skip_first = False
|
||
|
|
|
||
|
|
cipher_alg = self.ciphersuite.cipher_alg
|
||
|
|
|
||
|
|
start = 0
|
||
|
|
if skip_first:
|
||
|
|
start += cipher_alg.key_len
|
||
|
|
end = start + cipher_alg.key_len
|
||
|
|
cipher_secret = key_material[start:end]
|
||
|
|
self.cipher = cipher_alg(cipher_secret)
|
||
|
|
self.debug_repr("cipher_secret", cipher_secret)
|
||
|
|
|
||
|
|
def tls13_derive_keys(self, key_material):
|
||
|
|
cipher_alg = self.ciphersuite.cipher_alg
|
||
|
|
key_len = cipher_alg.key_len
|
||
|
|
iv_len = cipher_alg.fixed_iv_len
|
||
|
|
write_key = self.hkdf.expand_label(key_material, b"key", b"", key_len)
|
||
|
|
write_iv = self.hkdf.expand_label(key_material, b"iv", b"", iv_len)
|
||
|
|
self.cipher = cipher_alg(write_key, write_iv)
|
||
|
|
|
||
|
|
def snapshot(self):
|
||
|
|
"""
|
||
|
|
This is used mostly as a way to keep the cipher state and the seq_num.
|
||
|
|
"""
|
||
|
|
snap = connState(connection_end=self.connection_end,
|
||
|
|
read_or_write=self.row,
|
||
|
|
seq_num=self.seq_num,
|
||
|
|
compression_alg=type(self.compression),
|
||
|
|
ciphersuite=type(self.ciphersuite),
|
||
|
|
tls_version=self.tls_version)
|
||
|
|
snap.cipher = self.cipher.snapshot()
|
||
|
|
if self.hmac:
|
||
|
|
snap.hmac.key = self.hmac.key
|
||
|
|
return snap
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
res = "Connection end : %s\n" % self.connection_end.upper()
|
||
|
|
res += "Cipher suite : %s (0x%04x)\n" % (self.ciphersuite.name,
|
||
|
|
self.ciphersuite.val)
|
||
|
|
res += "Compression : %s (0x%02x)\n" % (self.compression.name,
|
||
|
|
self.compression.val)
|
||
|
|
tabsize = 4
|
||
|
|
return res.expandtabs(tabsize)
|
||
|
|
|
||
|
|
|
||
|
|
class readConnState(connState):
|
||
|
|
def __init__(self, **kargs):
|
||
|
|
connState.__init__(self, read_or_write="read", **kargs)
|
||
|
|
|
||
|
|
|
||
|
|
class writeConnState(connState):
|
||
|
|
def __init__(self, **kargs):
|
||
|
|
connState.__init__(self, read_or_write="write", **kargs)
|
||
|
|
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# TLS session #
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
class tlsSession(object):
|
||
|
|
"""
|
||
|
|
This is our TLS context, which gathers information from both sides of the
|
||
|
|
TLS connection. These sides are represented by a readConnState instance and
|
||
|
|
a writeConnState instance. Along with overarching network attributes, a
|
||
|
|
tlsSession object also holds negotiated, shared information, such as the
|
||
|
|
key exchange parameters and the master secret (when available).
|
||
|
|
|
||
|
|
The default connection_end is "server". This corresponds to the expected
|
||
|
|
behaviour for static exchange analysis (with a ClientHello parsed first).
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self,
|
||
|
|
ipsrc=None, ipdst=None,
|
||
|
|
sport=None, dport=None, sid=None,
|
||
|
|
connection_end="server",
|
||
|
|
wcs=None, rcs=None):
|
||
|
|
|
||
|
|
# Use this switch to prevent additions to the 'handshake_messages'.
|
||
|
|
self.frozen = False
|
||
|
|
|
||
|
|
# Network settings
|
||
|
|
self.ipsrc = ipsrc
|
||
|
|
self.ipdst = ipdst
|
||
|
|
self.sport = sport
|
||
|
|
self.dport = dport
|
||
|
|
self.sid = sid
|
||
|
|
|
||
|
|
# Our TCP socket. None until we send (or receive) a packet.
|
||
|
|
self.sock = None
|
||
|
|
|
||
|
|
# Connection states
|
||
|
|
self.connection_end = connection_end
|
||
|
|
|
||
|
|
if wcs is None:
|
||
|
|
# Instantiate wcs with dummy values.
|
||
|
|
self.wcs = writeConnState(connection_end=connection_end)
|
||
|
|
self.wcs.derive_keys()
|
||
|
|
else:
|
||
|
|
self.wcs = wcs
|
||
|
|
|
||
|
|
if rcs is None:
|
||
|
|
# Instantiate rcs with dummy values.
|
||
|
|
self.rcs = readConnState(connection_end=connection_end)
|
||
|
|
self.rcs.derive_keys()
|
||
|
|
else:
|
||
|
|
self.rcs = rcs
|
||
|
|
|
||
|
|
# The pending write/read states are updated by the building/parsing
|
||
|
|
# of various TLS packets. They get committed to self.wcs/self.rcs
|
||
|
|
# once Scapy builds/parses a ChangeCipherSpec message, or for certain
|
||
|
|
# other messages in case of TLS 1.3.
|
||
|
|
self.pwcs = None
|
||
|
|
self.triggered_pwcs_commit = False
|
||
|
|
self.prcs = None
|
||
|
|
self.triggered_prcs_commit = False
|
||
|
|
|
||
|
|
# Certificates and private keys
|
||
|
|
|
||
|
|
# The server certificate chain, as a list of Cert instances.
|
||
|
|
# Either we act as server and it has to be provided, or it is expected
|
||
|
|
# to be sent by the server through a Certificate message.
|
||
|
|
# The server certificate should be self.server_certs[0].
|
||
|
|
self.server_certs = []
|
||
|
|
|
||
|
|
# The server private key, as a PrivKey instance, when acting as server.
|
||
|
|
# XXX It would be nice to be able to provide both an RSA and an ECDSA
|
||
|
|
# key in order for the same Scapy server to support both families of
|
||
|
|
# cipher suites. See INIT_TLS_SESSION() in automaton_srv.py.
|
||
|
|
# (For now server_key holds either one of both types for DHE
|
||
|
|
# authentication, while server_rsa_key is used only for RSAkx.)
|
||
|
|
self.server_key = None
|
||
|
|
self.server_rsa_key = None
|
||
|
|
# self.server_ecdsa_key = None
|
||
|
|
|
||
|
|
# Back in the dreadful EXPORT days, US servers were forbidden to use
|
||
|
|
# RSA keys longer than 512 bits for RSAkx. When their usual RSA key
|
||
|
|
# was longer than this, they had to create a new key and send it via
|
||
|
|
# a ServerRSAParams message. When receiving such a message,
|
||
|
|
# Scapy stores this key in server_tmp_rsa_key as a PubKey instance.
|
||
|
|
self.server_tmp_rsa_key = None
|
||
|
|
|
||
|
|
# When client authentication is performed, we need at least a
|
||
|
|
# client certificate chain. If we act as client, we also have
|
||
|
|
# to provide the key associated with the first certificate.
|
||
|
|
self.client_certs = []
|
||
|
|
self.client_key = None
|
||
|
|
|
||
|
|
# Ephemeral key exchange parameters
|
||
|
|
|
||
|
|
# These are the group/curve parameters, needed to hold the information
|
||
|
|
# e.g. from receiving an SKE to sending a CKE. Usually, only one of
|
||
|
|
# these attributes will be different from None.
|
||
|
|
self.client_kx_ffdh_params = None
|
||
|
|
self.client_kx_ecdh_params = None
|
||
|
|
|
||
|
|
# These are PrivateKeys and PublicKeys from the appropriate FFDH/ECDH
|
||
|
|
# cryptography module, i.e. these are not raw bytes. Usually, only one
|
||
|
|
# in two will be different from None, e.g. when being a TLS client you
|
||
|
|
# will need the client_kx_privkey (the serialized public key is not
|
||
|
|
# actually registered) and you will receive a server_kx_pubkey.
|
||
|
|
self.client_kx_privkey = None
|
||
|
|
self.client_kx_pubkey = None
|
||
|
|
self.server_kx_privkey = None
|
||
|
|
self.server_kx_pubkey = None
|
||
|
|
|
||
|
|
# When using TLS 1.3, the tls13_client_pubshares will contain every
|
||
|
|
# potential key share (equate the 'client_kx_pubkey' before) the client
|
||
|
|
# offered, indexed by the id of the FFDH/ECDH group. These dicts
|
||
|
|
# effectively replace the four previous attributes.
|
||
|
|
self.tls13_client_privshares = {}
|
||
|
|
self.tls13_client_pubshares = {}
|
||
|
|
self.tls13_server_privshare = {}
|
||
|
|
self.tls13_server_pubshare = {}
|
||
|
|
|
||
|
|
# Negotiated session parameters
|
||
|
|
|
||
|
|
# The advertised TLS version found in the ClientHello (and
|
||
|
|
# EncryptedPreMasterSecret if used). If acting as server, it is set to
|
||
|
|
# the value advertised by the client in its ClientHello.
|
||
|
|
# The default value corresponds to TLS 1.2 (and TLS 1.3, incidentally).
|
||
|
|
self.advertised_tls_version = 0x0303
|
||
|
|
|
||
|
|
# The agreed-upon TLS version found in the ServerHello.
|
||
|
|
self.tls_version = None
|
||
|
|
|
||
|
|
# These attributes should eventually be known to both sides (SSLv3-TLS 1.2). # noqa: E501
|
||
|
|
self.client_random = None
|
||
|
|
self.server_random = None
|
||
|
|
self.pre_master_secret = None
|
||
|
|
self.master_secret = None
|
||
|
|
|
||
|
|
# The agreed-upon signature algorithm (for TLS 1.2-TLS 1.3 only)
|
||
|
|
self.selected_sig_alg = None
|
||
|
|
|
||
|
|
# A session ticket received by the client.
|
||
|
|
self.client_session_ticket = None
|
||
|
|
|
||
|
|
# These attributes should only be used with SSLv2 connections.
|
||
|
|
# We need to keep the KEY-MATERIAL here because it may be reused.
|
||
|
|
self.sslv2_common_cs = []
|
||
|
|
self.sslv2_connection_id = None
|
||
|
|
self.sslv2_challenge = None
|
||
|
|
self.sslv2_challenge_clientcert = None
|
||
|
|
self.sslv2_key_material = None
|
||
|
|
|
||
|
|
# These attributes should only be used with TLS 1.3 connections.
|
||
|
|
self.tls13_psk_secret = None
|
||
|
|
self.tls13_early_secret = None
|
||
|
|
self.tls13_dhe_secret = None
|
||
|
|
self.tls13_handshake_secret = None
|
||
|
|
self.tls13_master_secret = None
|
||
|
|
self.tls13_derived_secrets = {}
|
||
|
|
self.post_handshake_auth = False
|
||
|
|
self.tls13_ticket_ciphersuite = None
|
||
|
|
self.tls13_retry = False
|
||
|
|
self.middlebox_compatibility = False
|
||
|
|
|
||
|
|
# Handshake messages needed for Finished computation/validation.
|
||
|
|
# No record layer headers, no HelloRequests, no ChangeCipherSpecs.
|
||
|
|
self.handshake_messages = []
|
||
|
|
self.handshake_messages_parsed = []
|
||
|
|
|
||
|
|
# All exchanged TLS packets.
|
||
|
|
# XXX no support for now
|
||
|
|
# self.exchanged_pkts = []
|
||
|
|
|
||
|
|
def __setattr__(self, name, val):
|
||
|
|
if name == "connection_end":
|
||
|
|
if hasattr(self, "rcs") and self.rcs:
|
||
|
|
self.rcs.connection_end = val
|
||
|
|
if hasattr(self, "wcs") and self.wcs:
|
||
|
|
self.wcs.connection_end = val
|
||
|
|
if hasattr(self, "prcs") and self.prcs:
|
||
|
|
self.prcs.connection_end = val
|
||
|
|
if hasattr(self, "pwcs") and self.pwcs:
|
||
|
|
self.pwcs.connection_end = val
|
||
|
|
super(tlsSession, self).__setattr__(name, val)
|
||
|
|
|
||
|
|
# Mirroring
|
||
|
|
|
||
|
|
def mirror(self):
|
||
|
|
"""
|
||
|
|
This function takes a tlsSession object and swaps the IP addresses,
|
||
|
|
ports, connection ends and connection states. The triggered_commit are
|
||
|
|
also swapped (though it is probably overkill, it is cleaner this way).
|
||
|
|
|
||
|
|
It is useful for static analysis of a series of messages from both the
|
||
|
|
client and the server. In such a situation, it should be used every
|
||
|
|
time the message being read comes from a different side than the one
|
||
|
|
read right before, as the reading state becomes the writing state, and
|
||
|
|
vice versa. For instance you could do:
|
||
|
|
|
||
|
|
client_hello = open('client_hello.raw').read()
|
||
|
|
<read other messages>
|
||
|
|
|
||
|
|
m1 = TLS(client_hello)
|
||
|
|
m2 = TLS(server_hello, tls_session=m1.tls_session.mirror())
|
||
|
|
m3 = TLS(server_cert, tls_session=m2.tls_session)
|
||
|
|
m4 = TLS(client_keyexchange, tls_session=m3.tls_session.mirror())
|
||
|
|
"""
|
||
|
|
|
||
|
|
self.ipdst, self.ipsrc = self.ipsrc, self.ipdst
|
||
|
|
self.dport, self.sport = self.sport, self.dport
|
||
|
|
|
||
|
|
self.rcs, self.wcs = self.wcs, self.rcs
|
||
|
|
if self.rcs:
|
||
|
|
self.rcs.row = "read"
|
||
|
|
if self.wcs:
|
||
|
|
self.wcs.row = "write"
|
||
|
|
|
||
|
|
self.prcs, self.pwcs = self.pwcs, self.prcs
|
||
|
|
if self.prcs:
|
||
|
|
self.prcs.row = "read"
|
||
|
|
if self.pwcs:
|
||
|
|
self.pwcs.row = "write"
|
||
|
|
|
||
|
|
self.triggered_prcs_commit, self.triggered_pwcs_commit = \
|
||
|
|
self.triggered_pwcs_commit, self.triggered_prcs_commit
|
||
|
|
|
||
|
|
if self.connection_end == "client":
|
||
|
|
self.connection_end = "server"
|
||
|
|
elif self.connection_end == "server":
|
||
|
|
self.connection_end = "client"
|
||
|
|
|
||
|
|
return self
|
||
|
|
|
||
|
|
# Secrets management for SSLv3 to TLS 1.2
|
||
|
|
|
||
|
|
def compute_master_secret(self):
|
||
|
|
if self.pre_master_secret is None:
|
||
|
|
warning("Missing pre_master_secret while computing master_secret!")
|
||
|
|
if self.client_random is None:
|
||
|
|
warning("Missing client_random while computing master_secret!")
|
||
|
|
if self.server_random is None:
|
||
|
|
warning("Missing server_random while computing master_secret!")
|
||
|
|
|
||
|
|
ms = self.pwcs.prf.compute_master_secret(self.pre_master_secret,
|
||
|
|
self.client_random,
|
||
|
|
self.server_random)
|
||
|
|
self.master_secret = ms
|
||
|
|
if conf.debug_tls:
|
||
|
|
log_runtime.debug("TLS: master secret: %s", repr_hex(ms))
|
||
|
|
|
||
|
|
def compute_ms_and_derive_keys(self):
|
||
|
|
self.compute_master_secret()
|
||
|
|
self.prcs.derive_keys(client_random=self.client_random,
|
||
|
|
server_random=self.server_random,
|
||
|
|
master_secret=self.master_secret)
|
||
|
|
self.pwcs.derive_keys(client_random=self.client_random,
|
||
|
|
server_random=self.server_random,
|
||
|
|
master_secret=self.master_secret)
|
||
|
|
|
||
|
|
# Secrets management for SSLv2
|
||
|
|
|
||
|
|
def compute_sslv2_key_material(self):
|
||
|
|
if self.master_secret is None:
|
||
|
|
warning("Missing master_secret while computing key_material!")
|
||
|
|
if self.sslv2_challenge is None:
|
||
|
|
warning("Missing challenge while computing key_material!")
|
||
|
|
if self.sslv2_connection_id is None:
|
||
|
|
warning("Missing connection_id while computing key_material!")
|
||
|
|
|
||
|
|
km = self.pwcs.prf.derive_key_block(self.master_secret,
|
||
|
|
self.sslv2_challenge,
|
||
|
|
self.sslv2_connection_id,
|
||
|
|
2 * self.pwcs.cipher.key_len)
|
||
|
|
self.sslv2_key_material = km
|
||
|
|
if conf.debug_tls:
|
||
|
|
log_runtime.debug("TLS: master secret: %s", repr_hex(self.master_secret)) # noqa: E501
|
||
|
|
log_runtime.debug("TLS: key material: %s", repr_hex(km))
|
||
|
|
|
||
|
|
def compute_sslv2_km_and_derive_keys(self):
|
||
|
|
self.compute_sslv2_key_material()
|
||
|
|
self.prcs.sslv2_derive_keys(key_material=self.sslv2_key_material)
|
||
|
|
self.pwcs.sslv2_derive_keys(key_material=self.sslv2_key_material)
|
||
|
|
|
||
|
|
# Secrets management for TLS 1.3
|
||
|
|
|
||
|
|
def compute_tls13_early_secrets(self, external=False):
|
||
|
|
"""
|
||
|
|
This function computes the Early Secret, the binder_key,
|
||
|
|
the client_early_traffic_secret and the
|
||
|
|
early_exporter_master_secret (See RFC8446, section 7.1).
|
||
|
|
|
||
|
|
The parameter external is used for the computation of the
|
||
|
|
binder_key:
|
||
|
|
|
||
|
|
- For external PSK provisioned outside out of TLS, the parameter
|
||
|
|
external must be True.
|
||
|
|
- For resumption PSK, the parameter external must be False.
|
||
|
|
|
||
|
|
If no argument is specified, the label "res binder" will be
|
||
|
|
used by default.
|
||
|
|
|
||
|
|
Ciphers key and IV are updated accordingly for 0-RTT data.
|
||
|
|
self.handshake_messages should be ClientHello only.
|
||
|
|
"""
|
||
|
|
|
||
|
|
# if no hash algorithm is set, default to SHA-256
|
||
|
|
if self.prcs and self.prcs.hkdf:
|
||
|
|
hkdf = self.prcs.hkdf
|
||
|
|
elif self.pwcs and self.pwcs.hkdf:
|
||
|
|
hkdf = self.pwcs.hkdf
|
||
|
|
else:
|
||
|
|
hkdf = TLS13_HKDF("sha256")
|
||
|
|
|
||
|
|
if self.tls13_early_secret is None:
|
||
|
|
self.tls13_early_secret = hkdf.extract(None,
|
||
|
|
self.tls13_psk_secret)
|
||
|
|
|
||
|
|
if "binder_key" not in self.tls13_derived_secrets:
|
||
|
|
if external:
|
||
|
|
bk = hkdf.derive_secret(self.tls13_early_secret,
|
||
|
|
b"ext binder",
|
||
|
|
b"")
|
||
|
|
else:
|
||
|
|
bk = hkdf.derive_secret(self.tls13_early_secret,
|
||
|
|
b"res binder",
|
||
|
|
b"")
|
||
|
|
|
||
|
|
self.tls13_derived_secrets["binder_key"] = bk
|
||
|
|
|
||
|
|
cets = hkdf.derive_secret(self.tls13_early_secret,
|
||
|
|
b"c e traffic",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
|
||
|
|
self.tls13_derived_secrets["client_early_traffic_secret"] = cets
|
||
|
|
ees = hkdf.derive_secret(self.tls13_early_secret,
|
||
|
|
b"e exp master",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["early_exporter_secret"] = ees
|
||
|
|
|
||
|
|
if self.connection_end == "server":
|
||
|
|
if self.prcs:
|
||
|
|
self.prcs.tls13_derive_keys(cets)
|
||
|
|
elif self.connection_end == "client":
|
||
|
|
if self.pwcs:
|
||
|
|
self.pwcs.tls13_derive_keys(cets)
|
||
|
|
|
||
|
|
def compute_tls13_handshake_secrets(self):
|
||
|
|
"""
|
||
|
|
Ciphers key and IV are updated accordingly for Handshake data.
|
||
|
|
self.handshake_messages should be ClientHello...ServerHello.
|
||
|
|
"""
|
||
|
|
if self.prcs:
|
||
|
|
hkdf = self.prcs.hkdf
|
||
|
|
elif self.pwcs:
|
||
|
|
hkdf = self.pwcs.hkdf
|
||
|
|
else:
|
||
|
|
warning("No HKDF. This is abnormal.")
|
||
|
|
return
|
||
|
|
|
||
|
|
if self.tls13_early_secret is None:
|
||
|
|
self.tls13_early_secret = hkdf.extract(None,
|
||
|
|
self.tls13_psk_secret)
|
||
|
|
|
||
|
|
secret = hkdf.derive_secret(self.tls13_early_secret, b"derived", b"")
|
||
|
|
self.tls13_handshake_secret = hkdf.extract(secret, self.tls13_dhe_secret) # noqa: E501
|
||
|
|
|
||
|
|
chts = hkdf.derive_secret(self.tls13_handshake_secret,
|
||
|
|
b"c hs traffic",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["client_handshake_traffic_secret"] = chts
|
||
|
|
|
||
|
|
shts = hkdf.derive_secret(self.tls13_handshake_secret,
|
||
|
|
b"s hs traffic",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["server_handshake_traffic_secret"] = shts
|
||
|
|
|
||
|
|
def compute_tls13_traffic_secrets(self):
|
||
|
|
"""
|
||
|
|
Ciphers key and IV are updated accordingly for Application data.
|
||
|
|
self.handshake_messages should be ClientHello...ServerFinished.
|
||
|
|
"""
|
||
|
|
if self.prcs and self.prcs.hkdf:
|
||
|
|
hkdf = self.prcs.hkdf
|
||
|
|
elif self.pwcs and self.pwcs.hkdf:
|
||
|
|
hkdf = self.pwcs.hkdf
|
||
|
|
else:
|
||
|
|
warning("No HKDF. This is abnormal.")
|
||
|
|
return
|
||
|
|
|
||
|
|
tmp = hkdf.derive_secret(self.tls13_handshake_secret,
|
||
|
|
b"derived",
|
||
|
|
b"")
|
||
|
|
self.tls13_master_secret = hkdf.extract(tmp, None)
|
||
|
|
|
||
|
|
cts0 = hkdf.derive_secret(self.tls13_master_secret,
|
||
|
|
b"c ap traffic",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["client_traffic_secrets"] = [cts0]
|
||
|
|
|
||
|
|
sts0 = hkdf.derive_secret(self.tls13_master_secret,
|
||
|
|
b"s ap traffic",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["server_traffic_secrets"] = [sts0]
|
||
|
|
|
||
|
|
es = hkdf.derive_secret(self.tls13_master_secret,
|
||
|
|
b"exp master",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["exporter_secret"] = es
|
||
|
|
|
||
|
|
if self.connection_end == "server":
|
||
|
|
# self.prcs.tls13_derive_keys(cts0)
|
||
|
|
self.pwcs.tls13_derive_keys(sts0)
|
||
|
|
elif self.connection_end == "client":
|
||
|
|
# self.pwcs.tls13_derive_keys(cts0)
|
||
|
|
self.prcs.tls13_derive_keys(sts0)
|
||
|
|
|
||
|
|
def compute_tls13_traffic_secrets_end(self):
|
||
|
|
cts0 = self.tls13_derived_secrets["client_traffic_secrets"][0]
|
||
|
|
if self.connection_end == "server":
|
||
|
|
self.prcs.tls13_derive_keys(cts0)
|
||
|
|
elif self.connection_end == "client":
|
||
|
|
self.pwcs.tls13_derive_keys(cts0)
|
||
|
|
|
||
|
|
def compute_tls13_verify_data(self, connection_end, read_or_write):
|
||
|
|
shts = "server_handshake_traffic_secret"
|
||
|
|
chts = "client_handshake_traffic_secret"
|
||
|
|
if read_or_write == "read":
|
||
|
|
hkdf = self.rcs.hkdf
|
||
|
|
if connection_end == "client":
|
||
|
|
basekey = self.tls13_derived_secrets[shts]
|
||
|
|
elif connection_end == "server":
|
||
|
|
basekey = self.tls13_derived_secrets[chts]
|
||
|
|
elif read_or_write == "write":
|
||
|
|
hkdf = self.wcs.hkdf
|
||
|
|
if connection_end == "client":
|
||
|
|
basekey = self.tls13_derived_secrets[chts]
|
||
|
|
elif connection_end == "server":
|
||
|
|
basekey = self.tls13_derived_secrets[shts]
|
||
|
|
|
||
|
|
if not hkdf or not basekey:
|
||
|
|
warning("Missing arguments for verify_data computation!")
|
||
|
|
return None
|
||
|
|
# XXX this join() works in standard cases, but does it in all of them?
|
||
|
|
handshake_context = b"".join(self.handshake_messages)
|
||
|
|
return hkdf.compute_verify_data(basekey, handshake_context)
|
||
|
|
|
||
|
|
def compute_tls13_resumption_secret(self):
|
||
|
|
"""
|
||
|
|
self.handshake_messages should be ClientHello...ClientFinished.
|
||
|
|
"""
|
||
|
|
if self.connection_end == "server":
|
||
|
|
hkdf = self.prcs.hkdf
|
||
|
|
elif self.connection_end == "client":
|
||
|
|
hkdf = self.pwcs.hkdf
|
||
|
|
rs = hkdf.derive_secret(self.tls13_master_secret,
|
||
|
|
b"res master",
|
||
|
|
b"".join(self.handshake_messages))
|
||
|
|
self.tls13_derived_secrets["resumption_secret"] = rs
|
||
|
|
|
||
|
|
def compute_tls13_next_traffic_secrets(self, connection_end, read_or_write): # noqa : E501
|
||
|
|
"""
|
||
|
|
Ciphers key and IV are updated accordingly.
|
||
|
|
"""
|
||
|
|
if self.rcs.hkdf:
|
||
|
|
hkdf = self.rcs.hkdf
|
||
|
|
hl = hkdf.hash.digest_size
|
||
|
|
elif self.wcs.hkdf:
|
||
|
|
hkdf = self.wcs.hkdf
|
||
|
|
hl = hkdf.hash.digest_size
|
||
|
|
|
||
|
|
if read_or_write == "read":
|
||
|
|
if connection_end == "client":
|
||
|
|
cts = self.tls13_derived_secrets["client_traffic_secrets"]
|
||
|
|
ctsN = cts[-1]
|
||
|
|
ctsN_1 = hkdf.expand_label(ctsN, b"traffic upd", b"", hl)
|
||
|
|
cts.append(ctsN_1)
|
||
|
|
self.prcs.tls13_derive_keys(ctsN_1)
|
||
|
|
elif connection_end == "server":
|
||
|
|
sts = self.tls13_derived_secrets["server_traffic_secrets"]
|
||
|
|
stsN = sts[-1]
|
||
|
|
stsN_1 = hkdf.expand_label(stsN, b"traffic upd", b"", hl)
|
||
|
|
sts.append(stsN_1)
|
||
|
|
|
||
|
|
self.prcs.tls13_derive_keys(stsN_1)
|
||
|
|
|
||
|
|
elif read_or_write == "write":
|
||
|
|
if connection_end == "client":
|
||
|
|
cts = self.tls13_derived_secrets["client_traffic_secrets"]
|
||
|
|
ctsN = cts[-1]
|
||
|
|
ctsN_1 = hkdf.expand_label(ctsN, b"traffic upd", b"", hl)
|
||
|
|
cts.append(ctsN_1)
|
||
|
|
self.pwcs.tls13_derive_keys(ctsN_1)
|
||
|
|
elif connection_end == "server":
|
||
|
|
sts = self.tls13_derived_secrets["server_traffic_secrets"]
|
||
|
|
stsN = sts[-1]
|
||
|
|
stsN_1 = hkdf.expand_label(stsN, b"traffic upd", b"", hl)
|
||
|
|
sts.append(stsN_1)
|
||
|
|
|
||
|
|
self.pwcs.tls13_derive_keys(stsN_1)
|
||
|
|
|
||
|
|
# Tests for record building/parsing
|
||
|
|
|
||
|
|
def consider_read_padding(self):
|
||
|
|
# Return True if padding is needed. Used by TLSPadField.
|
||
|
|
return (self.rcs.cipher.type == "block" and
|
||
|
|
not (False in six.itervalues(self.rcs.cipher.ready)))
|
||
|
|
|
||
|
|
def consider_write_padding(self):
|
||
|
|
# Return True if padding is needed. Used by TLSPadField.
|
||
|
|
return self.wcs.cipher.type == "block"
|
||
|
|
|
||
|
|
def use_explicit_iv(self, version, cipher_type):
|
||
|
|
# Return True if an explicit IV is needed. Required for TLS 1.1+
|
||
|
|
# when either a block or an AEAD cipher is used.
|
||
|
|
if cipher_type == "stream":
|
||
|
|
return False
|
||
|
|
return version >= 0x0302
|
||
|
|
|
||
|
|
# Python object management
|
||
|
|
|
||
|
|
def hash(self):
|
||
|
|
s1 = struct.pack("!H", self.sport)
|
||
|
|
s2 = struct.pack("!H", self.dport)
|
||
|
|
family = socket.AF_INET
|
||
|
|
if ':' in self.ipsrc:
|
||
|
|
family = socket.AF_INET6
|
||
|
|
s1 += inet_pton(family, self.ipsrc)
|
||
|
|
s2 += inet_pton(family, self.ipdst)
|
||
|
|
return strxor(s1, s2)
|
||
|
|
|
||
|
|
def eq(self, other):
|
||
|
|
ok = False
|
||
|
|
if (self.sport == other.sport and self.dport == other.dport and
|
||
|
|
self.ipsrc == other.ipsrc and self.ipdst == other.ipdst):
|
||
|
|
ok = True
|
||
|
|
|
||
|
|
if (not ok and
|
||
|
|
self.dport == other.sport and self.sport == other.dport and
|
||
|
|
self.ipdst == other.ipsrc and self.ipsrc == other.ipdst):
|
||
|
|
ok = True
|
||
|
|
|
||
|
|
if ok:
|
||
|
|
if self.sid and other.sid:
|
||
|
|
return self.sid == other.sid
|
||
|
|
return True
|
||
|
|
|
||
|
|
return False
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
sid = repr(self.sid)
|
||
|
|
if len(sid) > 12:
|
||
|
|
sid = sid[:11] + "..."
|
||
|
|
return "%s:%s > %s:%s" % (self.ipsrc, str(self.sport),
|
||
|
|
self.ipdst, str(self.dport))
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# Session singleton #
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
|
||
|
|
class _GenericTLSSessionInheritance(Packet):
|
||
|
|
"""
|
||
|
|
Many classes inside the TLS module need to get access to session-related
|
||
|
|
information. For instance, an encrypted TLS record cannot be parsed without
|
||
|
|
some knowledge of the cipher suite being used and the secrets which have
|
||
|
|
been negotiated. Passing information is also essential to the handshake.
|
||
|
|
To this end, various TLS objects inherit from the present class.
|
||
|
|
"""
|
||
|
|
__slots__ = ["tls_session", "rcs_snap_init", "wcs_snap_init"]
|
||
|
|
name = "Dummy Generic TLS Packet"
|
||
|
|
fields_desc = []
|
||
|
|
|
||
|
|
def __init__(self, _pkt="", post_transform=None, _internal=0,
|
||
|
|
_underlayer=None, tls_session=None, **fields):
|
||
|
|
try:
|
||
|
|
setme = self.tls_session is None
|
||
|
|
except Exception:
|
||
|
|
setme = True
|
||
|
|
|
||
|
|
newses = False
|
||
|
|
if setme:
|
||
|
|
if tls_session is None:
|
||
|
|
newses = True
|
||
|
|
self.tls_session = tlsSession()
|
||
|
|
else:
|
||
|
|
self.tls_session = tls_session
|
||
|
|
|
||
|
|
self.rcs_snap_init = self.tls_session.rcs.snapshot()
|
||
|
|
self.wcs_snap_init = self.tls_session.wcs.snapshot()
|
||
|
|
|
||
|
|
if isinstance(_underlayer, TCP):
|
||
|
|
tcp = _underlayer
|
||
|
|
self.tls_session.sport = tcp.sport
|
||
|
|
self.tls_session.dport = tcp.dport
|
||
|
|
try:
|
||
|
|
self.tls_session.ipsrc = tcp.underlayer.src
|
||
|
|
self.tls_session.ipdst = tcp.underlayer.dst
|
||
|
|
except AttributeError:
|
||
|
|
pass
|
||
|
|
if conf.tls_session_enable:
|
||
|
|
if newses:
|
||
|
|
s = conf.tls_sessions.find(self.tls_session)
|
||
|
|
if s:
|
||
|
|
if s.dport == self.tls_session.dport:
|
||
|
|
self.tls_session = s
|
||
|
|
else:
|
||
|
|
self.tls_session = s.mirror()
|
||
|
|
else:
|
||
|
|
conf.tls_sessions.add(self.tls_session)
|
||
|
|
if self.tls_session.connection_end == "server":
|
||
|
|
srk = conf.tls_sessions.server_rsa_key
|
||
|
|
if not self.tls_session.server_rsa_key and \
|
||
|
|
srk:
|
||
|
|
self.tls_session.server_rsa_key = srk
|
||
|
|
|
||
|
|
Packet.__init__(self, _pkt=_pkt, post_transform=post_transform,
|
||
|
|
_internal=_internal, _underlayer=_underlayer,
|
||
|
|
**fields)
|
||
|
|
|
||
|
|
def __getattr__(self, attr):
|
||
|
|
"""
|
||
|
|
The tls_session should be found only through the normal mechanism.
|
||
|
|
"""
|
||
|
|
if attr == "tls_session":
|
||
|
|
return None
|
||
|
|
return super(_GenericTLSSessionInheritance, self).__getattr__(attr)
|
||
|
|
|
||
|
|
def tls_session_update(self, msg_str):
|
||
|
|
"""
|
||
|
|
post_{build, dissection}_tls_session_update() are used to update the
|
||
|
|
tlsSession context. The default definitions below, along with
|
||
|
|
tls_session_update(), may prevent code duplication in some cases.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
def post_build_tls_session_update(self, msg_str):
|
||
|
|
self.tls_session_update(msg_str)
|
||
|
|
|
||
|
|
def post_dissection_tls_session_update(self, msg_str):
|
||
|
|
self.tls_session_update(msg_str)
|
||
|
|
|
||
|
|
def copy(self):
|
||
|
|
pkt = Packet.copy(self)
|
||
|
|
pkt.tls_session = self.tls_session
|
||
|
|
return pkt
|
||
|
|
|
||
|
|
def clone_with(self, payload=None, **kargs):
|
||
|
|
pkt = Packet.clone_with(self, payload=payload, **kargs)
|
||
|
|
pkt.tls_session = self.tls_session
|
||
|
|
return pkt
|
||
|
|
|
||
|
|
def raw_stateful(self):
|
||
|
|
return super(_GenericTLSSessionInheritance, self).__bytes__()
|
||
|
|
|
||
|
|
def str_stateful(self):
|
||
|
|
return self.raw_stateful()
|
||
|
|
|
||
|
|
def __bytes__(self):
|
||
|
|
"""
|
||
|
|
The __bytes__ call has to leave the connection states unchanged.
|
||
|
|
We also have to delete raw_packet_cache in order to access post_build.
|
||
|
|
|
||
|
|
For performance, the pending connStates are not snapshotted.
|
||
|
|
This should not be an issue, but maybe pay attention to this.
|
||
|
|
|
||
|
|
The previous_freeze_state prevents issues with calling a raw() calling
|
||
|
|
in turn another raw(), which would unfreeze the session too soon.
|
||
|
|
"""
|
||
|
|
s = self.tls_session
|
||
|
|
rcs_snap = s.rcs.snapshot()
|
||
|
|
wcs_snap = s.wcs.snapshot()
|
||
|
|
rpc_snap = self.raw_packet_cache
|
||
|
|
rpcf_snap = self.raw_packet_cache_fields
|
||
|
|
|
||
|
|
s.wcs = self.rcs_snap_init
|
||
|
|
|
||
|
|
self.raw_packet_cache = None
|
||
|
|
previous_freeze_state = s.frozen
|
||
|
|
s.frozen = True
|
||
|
|
built_packet = super(_GenericTLSSessionInheritance, self).__bytes__()
|
||
|
|
s.frozen = previous_freeze_state
|
||
|
|
|
||
|
|
s.rcs = rcs_snap
|
||
|
|
s.wcs = wcs_snap
|
||
|
|
self.raw_packet_cache = rpc_snap
|
||
|
|
self.raw_packet_cache_fields = rpcf_snap
|
||
|
|
|
||
|
|
return built_packet
|
||
|
|
__str__ = __bytes__
|
||
|
|
|
||
|
|
def show2(self):
|
||
|
|
"""
|
||
|
|
Rebuild the TLS packet with the same context, and then .show() it.
|
||
|
|
We need self.__class__ to call the subclass in a dynamic way.
|
||
|
|
|
||
|
|
Howether we do not want the tls_session.{r,w}cs.seq_num to be updated.
|
||
|
|
We have to bring back the init states (it's possible the cipher context
|
||
|
|
has been updated because of parsing) but also to keep the current state
|
||
|
|
and restore it afterwards (the raw() call may also update the states).
|
||
|
|
"""
|
||
|
|
s = self.tls_session
|
||
|
|
rcs_snap = s.rcs.snapshot()
|
||
|
|
wcs_snap = s.wcs.snapshot()
|
||
|
|
|
||
|
|
s.rcs = self.rcs_snap_init
|
||
|
|
|
||
|
|
built_packet = raw(self)
|
||
|
|
s.frozen = True
|
||
|
|
self.__class__(built_packet, tls_session=s).show()
|
||
|
|
s.frozen = False
|
||
|
|
|
||
|
|
s.rcs = rcs_snap
|
||
|
|
s.wcs = wcs_snap
|
||
|
|
|
||
|
|
def mysummary(self):
|
||
|
|
return "TLS %s / %s" % (repr(self.tls_session),
|
||
|
|
getattr(self, "_name", self.name))
|
||
|
|
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# Multiple TLS sessions #
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
class _tls_sessions(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.sessions = {}
|
||
|
|
self.server_rsa_key = None
|
||
|
|
|
||
|
|
def add(self, session):
|
||
|
|
s = self.find(session)
|
||
|
|
if s:
|
||
|
|
log_runtime.info("TLS: previous session shall not be overwritten")
|
||
|
|
return
|
||
|
|
|
||
|
|
h = session.hash()
|
||
|
|
if h in self.sessions:
|
||
|
|
self.sessions[h].append(session)
|
||
|
|
else:
|
||
|
|
self.sessions[h] = [session]
|
||
|
|
|
||
|
|
def rem(self, session):
|
||
|
|
s = self.find(session)
|
||
|
|
if s:
|
||
|
|
log_runtime.info("TLS: previous session shall not be overwritten")
|
||
|
|
return
|
||
|
|
|
||
|
|
h = session.hash()
|
||
|
|
self.sessions[h].remove(session)
|
||
|
|
|
||
|
|
def find(self, session):
|
||
|
|
try:
|
||
|
|
h = session.hash()
|
||
|
|
except Exception:
|
||
|
|
return None
|
||
|
|
if h in self.sessions:
|
||
|
|
for k in self.sessions[h]:
|
||
|
|
if k.eq(session):
|
||
|
|
if conf.tls_verbose:
|
||
|
|
log_runtime.info("TLS: found session matching %s", k)
|
||
|
|
return k
|
||
|
|
if conf.tls_verbose:
|
||
|
|
log_runtime.info("TLS: did not find session matching %s", session)
|
||
|
|
return None
|
||
|
|
|
||
|
|
def __repr__(self):
|
||
|
|
res = [("First endpoint", "Second endpoint", "Session ID")]
|
||
|
|
for l in six.itervalues(self.sessions):
|
||
|
|
for s in l:
|
||
|
|
src = "%s[%d]" % (s.ipsrc, s.sport)
|
||
|
|
dst = "%s[%d]" % (s.ipdst, s.dport)
|
||
|
|
sid = repr(s.sid)
|
||
|
|
if len(sid) > 12:
|
||
|
|
sid = sid[:11] + "..."
|
||
|
|
res.append((src, dst, sid))
|
||
|
|
colwidth = (max(len(y) for y in x) for x in zip(*res))
|
||
|
|
fmt = " ".join(map(lambda x: "%%-%ds" % x, colwidth))
|
||
|
|
return "\n".join(map(lambda x: fmt % x, res))
|
||
|
|
|
||
|
|
|
||
|
|
class TLSSession(DefaultSession):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
server_rsa_key = kwargs.pop("server_rsa_key", None)
|
||
|
|
super(TLSSession, self).__init__(*args, **kwargs)
|
||
|
|
self._old_conf_status = conf.tls_session_enable
|
||
|
|
conf.tls_session_enable = True
|
||
|
|
if server_rsa_key:
|
||
|
|
conf.tls_sessions.server_rsa_key = server_rsa_key
|
||
|
|
|
||
|
|
def toPacketList(self):
|
||
|
|
conf.tls_session_enable = self._old_conf_status
|
||
|
|
return super(TLSSession, self).toPacketList()
|
||
|
|
|
||
|
|
|
||
|
|
conf.tls_sessions = _tls_sessions()
|
||
|
|
conf.tls_session_enable = False
|
||
|
|
conf.tls_verbose = False
|