lib/mbedtls
1
0
Fork 0
mirror of https://github.com/Mbed-TLS/mbedtls.git synced 2025-07-29 11:41:15 +03:00
Code Activity

PSA wrappers: don't poison buffers when buffer copying is disabled

Browse Source 
Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine
2024-01-04 20:33:29 +01:00
parent 7c7b7d5db3
commit b3d457ce2f
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/scripts/generate_psa_wrappers.py
View File

@ -129,6 +129,20 @@ class PSAWrapperGenerator(c_wrapper_generator.Base):
param.buffer_name, param.size_name
))
def _write_poison_buffer_parameters(self, out: typing_util.Writable,
buffer_parameters: List[BufferParameter],
poison: bool) -> None:
"""Write poisoning or unpoisoning code for the buffer parameters.
Write poisoning code if poison is true, unpoisoning code otherwise.
"""
if not buffer_parameters:
return
out.write('#if defined(MBEDTLS_PSA_COPY_CALLER_BUFFERS)\n')
for param in buffer_parameters:
self._write_poison_buffer_parameter(out, param, poison)
out.write('#endif /* defined(MBEDTLS_PSA_COPY_CALLER_BUFFERS) */\n')
@staticmethod
def _parameter_should_be_copied(function_name: str,
_buffer_name: Optional[str]) -> bool:
@ -148,11 +162,9 @@ class PSAWrapperGenerator(c_wrapper_generator.Base):
argument_names)
if self._parameter_should_be_copied(function.name,
function.arguments[param.index].name))
for param in buffer_parameters:
self._write_poison_buffer_parameter(out, param, True)
self._write_poison_buffer_parameters(out, buffer_parameters, True)
super()._write_function_call(out, function, argument_names)
for param in buffer_parameters:
self._write_poison_buffer_parameter(out, param, False)
self._write_poison_buffer_parameters(out, buffer_parameters, False)
def _write_prologue(self, out: typing_util.Writable, header: bool) -> None:
super()._write_prologue(out, header)