123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425 |
- /** @file
- IA32-specific functions for unit-testing INTN and UINTN functions in
- SafeIntLib.
- Copyright (c) Microsoft Corporation.<BR>
- Copyright (c) 2019 - 2020, Intel Corporation. All rights reserved.<BR>
- SPDX-License-Identifier: BSD-2-Clause-Patent
- **/
- #include <gtest/gtest.h>
- extern "C" {
- #include <Base.h>
- #include <Library/SafeIntLib.h>
- }
- TEST(ConversionTestSuite, TestSafeInt32ToUintn) {
- RETURN_STATUS Status;
- INT32 Operand;
- UINTN Result;
- //
- // If Operand is non-negative, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeInt32ToUintn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((UINTN)0x5bababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (-1537977259);
- Status = SafeInt32ToUintn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeUint32ToIntn) {
- RETURN_STATUS Status;
- UINT32 Operand;
- INTN Result;
- //
- // If Operand is <= MAX_INTN, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeUint32ToIntn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x5bababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0xabababab);
- Status = SafeUint32ToIntn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeIntnToInt32) {
- RETURN_STATUS Status;
- INTN Operand;
- INT32 Result;
- //
- // INTN is same as INT32 in IA32, so this is just a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeIntnToInt32 (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x5bababab, Result);
- }
- TEST(ConversionTestSuite, TestSafeIntnToUint32) {
- RETURN_STATUS Status;
- INTN Operand;
- UINT32 Result;
- //
- // If Operand is non-negative, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeIntnToUint32 (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((UINT32)0x5bababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (-1537977259);
- Status = SafeIntnToUint32 (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeUintnToUint32) {
- RETURN_STATUS Status;
- UINTN Operand;
- UINT32 Result;
- //
- // UINTN is same as UINT32 in IA32, so this is just a cast
- //
- Operand = 0xabababab;
- Result = 0;
- Status = SafeUintnToUint32 (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0xabababab, Result);
- }
- TEST(ConversionTestSuite, TestSafeUintnToIntn) {
- RETURN_STATUS Status;
- UINTN Operand;
- INTN Result;
- //
- // If Operand is <= MAX_INTN, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeUintnToIntn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x5bababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0xabababab);
- Status = SafeUintnToIntn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeUintnToInt64) {
- RETURN_STATUS Status;
- UINTN Operand;
- INT64 Result;
- //
- // UINTN is same as UINT32 in IA32, and UINT32 is a subset of
- // INT64, so this is just a cast
- //
- Operand = 0xabababab;
- Result = 0;
- Status = SafeUintnToInt64 (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0xabababab, Result);
- }
- TEST(ConversionTestSuite, TestSafeInt64ToIntn) {
- RETURN_STATUS Status;
- INT64 Operand;
- INTN Result;
- //
- // If Operand is between MIN_INTN and MAX_INTN2 inclusive, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeInt64ToIntn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x5bababab, Result);
- Operand = (-1537977259);
- Status = SafeInt64ToIntn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((-1537977259), Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0x5babababefefefef);
- Status = SafeInt64ToIntn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- Operand = (-6605562033422200815);
- Status = SafeInt64ToIntn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeInt64ToUintn) {
- RETURN_STATUS Status;
- INT64 Operand;
- UINTN Result;
- //
- // If Operand is between 0 and MAX_UINTN inclusive, then it's a cast
- //
- Operand = 0xabababab;
- Result = 0;
- Status = SafeInt64ToUintn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0xabababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0x5babababefefefef);
- Status = SafeInt64ToUintn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- Operand = (-6605562033422200815);
- Status = SafeInt64ToUintn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeUint64ToIntn) {
- RETURN_STATUS Status;
- UINT64 Operand;
- INTN Result;
- //
- // If Operand is <= MAX_INTN, then it's a cast
- //
- Operand = 0x5bababab;
- Result = 0;
- Status = SafeUint64ToIntn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x5bababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0xababababefefefef);
- Status = SafeUint64ToIntn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(ConversionTestSuite, TestSafeUint64ToUintn) {
- RETURN_STATUS Status;
- UINT64 Operand;
- UINTN Result;
- //
- // If Operand is <= MAX_UINTN, then it's a cast
- //
- Operand = 0xabababab;
- Result = 0;
- Status = SafeUint64ToUintn (Operand, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0xabababab, Result);
- //
- // Otherwise should result in an error status
- //
- Operand = (0xababababefefefef);
- Status = SafeUint64ToUintn (Operand, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(AdditionSubtractionTestSuite, TestSafeUintnAdd) {
- RETURN_STATUS Status;
- UINTN Augend;
- UINTN Addend;
- UINTN Result;
- //
- // If the result of addition doesn't overflow MAX_UINTN, then it's addition
- //
- Augend = 0x3a3a3a3a;
- Addend = 0x3a3a3a3a;
- Result = 0;
- Status = SafeUintnAdd (Augend, Addend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((UINTN)0x74747474, Result);
- //
- // Otherwise should result in an error status
- //
- Augend = 0xabababab;
- Addend = 0xbcbcbcbc;
- Status = SafeUintnAdd (Augend, Addend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(AdditionSubtractionTestSuite, TestSafeIntnAdd) {
- RETURN_STATUS Status;
- INTN Augend;
- INTN Addend;
- INTN Result;
- //
- // If the result of addition doesn't overflow MAX_INTN
- // and doesn't underflow MIN_INTN, then it's addition
- //
- Augend = 0x3a3a3a3a;
- Addend = 0x3a3a3a3a;
- Result = 0;
- Status = SafeIntnAdd (Augend, Addend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x74747474, Result);
- Augend = (-976894522);
- Addend = (-976894522);
- Status = SafeIntnAdd (Augend, Addend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((-1953789044), Result);
- //
- // Otherwise should result in an error status
- //
- Augend = 0x5a5a5a5a;
- Addend = 0x5a5a5a5a;
- Status = SafeIntnAdd (Augend, Addend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- Augend = (-1515870810);
- Addend = (-1515870810);
- Status = SafeIntnAdd (Augend, Addend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(AdditionSubtractionTestSuite, TestSafeUintnSub) {
- RETURN_STATUS Status;
- UINTN Minuend;
- UINTN Subtrahend;
- UINTN Result;
- //
- // If Minuend >= Subtrahend, then it's subtraction
- //
- Minuend = 0x5a5a5a5a;
- Subtrahend = 0x3b3b3b3b;
- Result = 0;
- Status = SafeUintnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((UINTN)0x1f1f1f1f, Result);
- //
- // Otherwise should result in an error status
- //
- Minuend = 0x5a5a5a5a;
- Subtrahend = 0x6d6d6d6d;
- Status = SafeUintnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(AdditionSubtractionTestSuite, TestSafeIntnSub) {
- RETURN_STATUS Status;
- INTN Minuend;
- INTN Subtrahend;
- INTN Result;
- //
- // If the result of subtractions doesn't overflow MAX_INTN or
- // underflow MIN_INTN, then it's subtraction
- //
- Minuend = 0x5a5a5a5a;
- Subtrahend = 0x3a3a3a3a;
- Result = 0;
- Status = SafeIntnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x20202020, Result);
- Minuend = 0x3a3a3a3a;
- Subtrahend = 0x5a5a5a5a;
- Status = SafeIntnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ ((-538976288), Result);
- //
- // Otherwise should result in an error status
- //
- Minuend = (-2054847098);
- Subtrahend = 2054847098;
- Status = SafeIntnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- Minuend = (2054847098);
- Subtrahend = (-2054847098);
- Status = SafeIntnSub (Minuend, Subtrahend, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(MultiplicationTestSuite, TestSafeUintnMult) {
- RETURN_STATUS Status;
- UINTN Multiplicand;
- UINTN Multiplier;
- UINTN Result;
- //
- // If the result of multiplication doesn't overflow MAX_UINTN, it will succeed
- //
- Multiplicand = 0xa122a;
- Multiplier = 0xd23;
- Result = 0;
- Status = SafeUintnMult (Multiplicand, Multiplier, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x844c9dbe, Result);
- //
- // Otherwise should result in an error status
- //
- Multiplicand = 0xa122a;
- Multiplier = 0xed23;
- Status = SafeUintnMult (Multiplicand, Multiplier, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
- TEST(MultiplicationTestSuite, TestSafeIntnMult) {
- RETURN_STATUS Status;
- INTN Multiplicand;
- INTN Multiplier;
- INTN Result;
- //
- // If the result of multiplication doesn't overflow MAX_INTN and doesn't
- // underflow MIN_UINTN, it will succeed
- //
- Multiplicand = 0x123456;
- Multiplier = 0x678;
- Result = 0;
- Status = SafeIntnMult (Multiplicand, Multiplier, &Result);
- ASSERT_EQ (Status, RETURN_SUCCESS);
- ASSERT_EQ (0x75c28c50, Result);
- //
- // Otherwise should result in an error status
- //
- Multiplicand = 0x123456;
- Multiplier = 0xabc;
- Status = SafeIntnMult (Multiplicand, Multiplier, &Result);
- ASSERT_EQ (RETURN_BUFFER_TOO_SMALL, Status);
- }
|