feat: Add AES ready-to-use attack selection functions. (1859977e) · Commits · eshard / scared

scared/selection_functions/init.py

0 → 100644

+5 −0

Original line number	Diff line number	Diff line
		from .selection_functions import ( # noqa: F401
		attack_selection_function, reverse_selection_function, selection_function,
		SelectionFunction
		)
		from . import aes # noqa: F401

scared/selection_functions/aes/init.py

0 → 100644

+1 −0

Original line number	Diff line number	Diff line
		from . import encrypt, decrypt # noqa: F401

scared/selection_functions/aes/decrypt.py

0 → 100644

+52 −0

Original line number	Diff line number	Diff line
		from . import encrypt

		first_add_round_key = encrypt.last_add_round_key
		first_add_round_key.__doc__ = """Build an attack selection function which computes intermediate values after AES decrypt round key operation at first round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.

		"""

		last_add_round_key = encrypt.first_add_round_key
		last_add_round_key.__doc__ = """Build an attack selection function which computes intermediate values after AES decrypt round key operation at last round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		plaintext_tag (str, default='plaintext'): tag (key value) of the plaintext metadata to use to retrieve plaintext
		values from the metadata dict when selection function is called.

		"""
		first_sub_bytes = encrypt.last_sub_bytes
		first_sub_bytes.__doc__ = """Build an attack selection function which computes intermediate values after AES decrypt sub bytes (S-box) operation at first round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.

		"""
		last_sub_bytes = encrypt.first_sub_bytes
		last_sub_bytes.__doc__ = """Build an attack selection function which computes intermediate values after AES decrypt sub bytes (S-box) operation at last round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		plaintext_tag (str, default='plaintext'): tag (key value) of the plaintext metadata to use to retrieve plaintext
		values from the metadata dict when selection function is called.

		"""
		delta_r_first_rounds = encrypt.delta_r_last_rounds
		delta_r_first_rounds.__doc__ = """Build an attack selection function which computes delta intermediate values between AES decrypt first two rounds, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.
		"""

scared/selection_functions/aes/encrypt.py

0 → 100644

+102 −0

Original line number	Diff line number	Diff line
		from ..selection_functions import _decorated_selection_function, _AttackSelectionFunctionWrapped
		from ... import aes
		import numpy as _np


		def _add_round_key(data, guesses):
		res = _np.empty((data.shape[0], len(guesses), data.shape[1]), dtype='uint8')
		data = data.astype('uint8')
		for g in guesses:
		res[:, g, :] = _np.bitwise_xor(data, g)
		return res


		def _sub_bytes(data, guesses):
		return aes.sub_bytes(_add_round_key(data=data, guesses=guesses))


		def _inv_sub_bytes(data, guesses):
		return aes.inv_sub_bytes(_add_round_key(data=data, guesses=guesses))


		def _delta_last_rounds(data, guesses):
		return _np.bitwise_xor(
		aes.shift_rows(data),
		aes.inv_sub_bytes(
		_add_round_key(data=data, guesses=guesses)
		).swapaxes(0, 1)
		).swapaxes(0, 1)


		def first_add_round_key(guesses=_np.arange(256, dtype='uint8'), words=None, plaintext_tag='plaintext'):
		"""Build an attack selection function which computes intermediate values after AES encrypt round key operation at first round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		plaintext_tag (str, default='plaintext'): tag (key value) of the plaintext metadata to use to retrieve plaintext
		values from the metadata dict when selection function is called.

		"""
		sf = _decorated_selection_function(_AttackSelectionFunctionWrapped, _add_round_key, words=words, guesses=guesses)
		sf.target_tag = plaintext_tag
		return sf


		def last_add_round_key(guesses=_np.arange(256, dtype='uint8'), words=None, ciphertext_tag='ciphertext'):
		"""Build an attack selection function which computes intermediate values after AES encrypt round key operation at last round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.

		"""
		sf = _decorated_selection_function(_AttackSelectionFunctionWrapped, _add_round_key, words=words, guesses=guesses)
		sf.target_tag = ciphertext_tag
		return sf


		def first_sub_bytes(guesses=_np.arange(256, dtype='uint8'), words=None, plaintext_tag='plaintext'):
		"""Build an attack selection function which computes intermediate values after AES encrypt sub bytes (S-box) operation at first round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		plaintext_tag (str, default='plaintext'): tag (key value) of the plaintext metadata to use to retrieve plaintext
		values from the metadata dict when selection function is called.

		"""
		sf = _decorated_selection_function(_AttackSelectionFunctionWrapped, _sub_bytes, words=words, guesses=guesses)
		sf.target_tag = plaintext_tag
		return sf


		def last_sub_bytes(guesses=_np.arange(256, dtype='uint8'), words=None, ciphertext_tag='ciphertext'):
		"""Build an attack selection function which computes intermediate values after AES encrypt sub bytes (S-box) operation at last round, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.

		"""
		sf = _decorated_selection_function(_AttackSelectionFunctionWrapped, _inv_sub_bytes, words=words, guesses=guesses)
		sf.target_tag = ciphertext_tag
		return sf


		def delta_r_last_rounds(guesses=_np.arange(256, dtype='uint8'), words=None, ciphertext_tag='ciphertext'):
		"""Build an attack selection function which computes delta intermediate values between AES encrypt last two rounds, for guesses values.

		Args:
		guesses (numpy.array, default=numpy.arange(256)): default guesses value used for key hypothesis.
		words (:class:`scared.traces.Samples.SUPPORTED_INDICES_TYPES`, default=None): selection of key words computed.
		ciphertext_tag (str, default='ciphertext'): tag (key value) of the ciphertext metadata to use to retrieve ciphertext
		values from the metadata dict when selection function is called.
		"""
		sf = _decorated_selection_function(_AttackSelectionFunctionWrapped, _delta_last_rounds, words=words, guesses=guesses)
		sf.target_tag = ciphertext_tag
		return sf

scared/selection_functions.py→scared/selection_functions/selection_functions.py

+13 −1

Original line number	Diff line number	Diff line
		import logging
		import numpy as _np
		import inspect
		from ._utils import _is_bytes_array
		from .._utils import _is_bytes_array

		logger = logging.getLogger(__name__)

		@@ -90,3 +90,15 @@ def attack_selection_function(function=None, guesses=None, words=None):
		def reverse_selection_function(function=None, words=None):
		"""Decorator that wraps provided selection function as a reverse selection function."""
		return selection_function(function, words=words)


		class _AttackSelectionFunctionWrapped(_AttackSelectionFunction):

		def __init__(self, function, guesses, words, target_tag=None, target_name=None):
		super().__init__(function=function, words=words, guesses=guesses)
		self.target_tag = target_tag
		self.target_name = 'data'

		def __call__(self, **kwargs):
		kwargs[self.target_name] = kwargs[self.target_tag]
		return super().__call__(**kwargs)