116 lines
2.3 KiB
C
116 lines
2.3 KiB
C
/** @file
|
|
|
|
OcGuardLib
|
|
|
|
Copyright (c) 2018, vit9696
|
|
|
|
All rights reserved.
|
|
|
|
This program and the accompanying materials
|
|
are licensed and made available under the terms and conditions of the BSD License
|
|
which accompanies this distribution. The full text of the license may be found at
|
|
http://opensource.org/licenses/bsd-license.php
|
|
|
|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
|
|
|
|
**/
|
|
|
|
#include <Library/OcGuardLib.h>
|
|
|
|
//
|
|
// Currently no architectures provide UINTN and INTN different from 32-bit or 64-bit
|
|
// integers. For this reason they are the only variants supported and are enforced here.
|
|
// The lib must be C99-compliant, thus no _Static_assert.
|
|
//
|
|
|
|
typedef BOOLEAN SignedIntnMustBe32or64[sizeof (INTN) == sizeof (INT64)
|
|
|| sizeof (INTN) == sizeof (INT32) ? 1 : -1];
|
|
|
|
typedef BOOLEAN UnsignedIntnMustBe32or64[sizeof (UINTN) == sizeof (UINT64)
|
|
|| sizeof (UINTN) == sizeof (UINT32) ? 1 : -1];
|
|
|
|
BOOLEAN
|
|
(OcOverflowAddUN) (
|
|
UINTN A,
|
|
UINTN B,
|
|
UINTN *Result
|
|
)
|
|
{
|
|
if (sizeof (UINTN) == sizeof (UINT64)) {
|
|
return OcOverflowAddU64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowAddU32 ((UINT32)A, (UINT32)B, (UINT32 *)Result);
|
|
}
|
|
|
|
BOOLEAN
|
|
(OcOverflowSubUN) (
|
|
UINTN A,
|
|
UINTN B,
|
|
UINTN *Result
|
|
)
|
|
{
|
|
if (sizeof (UINTN) == sizeof (UINT64)) {
|
|
return OcOverflowSubU64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowSubU32 ((UINT32)A, (UINT32)B, (UINT32 *)Result);
|
|
}
|
|
|
|
BOOLEAN
|
|
(OcOverflowMulUN) (
|
|
UINTN A,
|
|
UINTN B,
|
|
UINTN *Result
|
|
)
|
|
{
|
|
if (sizeof (UINTN) == sizeof (UINT64)) {
|
|
return OcOverflowMulU64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowMulU32 ((UINT32)A, (UINT32)B, (UINT32 *)Result);
|
|
}
|
|
|
|
BOOLEAN
|
|
(OcOverflowAddSN) (
|
|
INTN A,
|
|
INTN B,
|
|
INTN *Result
|
|
)
|
|
{
|
|
if (sizeof (INTN) == sizeof (INT64)) {
|
|
return OcOverflowAddS64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowAddS32 ((INT32)A, (INT32)B, (INT32 *)Result);
|
|
}
|
|
|
|
BOOLEAN
|
|
(OcOverflowSubSN) (
|
|
INTN A,
|
|
INTN B,
|
|
INTN *Result
|
|
)
|
|
{
|
|
if (sizeof (INTN) == sizeof (INT64)) {
|
|
return OcOverflowSubS64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowSubS32 ((INT32)A, (INT32)B, (INT32 *)Result);
|
|
}
|
|
|
|
BOOLEAN
|
|
(OcOverflowMulSN) (
|
|
INTN A,
|
|
INTN B,
|
|
INTN *Result
|
|
)
|
|
{
|
|
if (sizeof (INTN) == sizeof (INT64)) {
|
|
return OcOverflowMulS64 (A, B, Result);
|
|
}
|
|
|
|
return OcOverflowMulS32 ((INT32)A, (INT32)B, (INT32 *)Result);
|
|
}
|