95 lines
2.9 KiB
Swift
95 lines
2.9 KiB
Swift
|
import XCTest
|
||
|
|
import libtommath
|
||
|
|
|
||
|
|
/* ---> Basic Manipulations <--- */
|
||
|
|
|
||
|
|
extension mp_int {
|
||
|
|
var isZero: Bool { used == 0 }
|
||
|
|
var isNeg: Bool { sign == MP_NEG }
|
||
|
|
var isEven: Bool { used == 0 || (dp[0] & 1) == 0 }
|
||
|
|
var isOdd: Bool { !isEven }
|
||
|
|
}
|
||
|
|
|
||
|
|
func mp_get_u32(_ a: UnsafePointer<mp_int>) -> UInt32 {
|
||
|
|
return UInt32(bitPattern: mp_get_i32(a))
|
||
|
|
}
|
||
|
|
|
||
|
|
class LibTommathTests: XCTestCase {
|
||
|
|
|
||
|
|
override func setUpWithError() throws {
|
||
|
|
// Put setup code here. This method is called before the invocation of each test method in the class.
|
||
|
|
}
|
||
|
|
|
||
|
|
override func tearDownWithError() throws {
|
||
|
|
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
||
|
|
}
|
||
|
|
|
||
|
|
func testTrivialStuff() throws {
|
||
|
|
var a = mp_int()
|
||
|
|
var b = mp_int()
|
||
|
|
var c = mp_int()
|
||
|
|
var d = mp_int()
|
||
|
|
|
||
|
|
XCTAssertEqual(mp_init(&a), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_init(&b), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_init(&c), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_init(&d), MP_OKAY)
|
||
|
|
|
||
|
|
defer {
|
||
|
|
mp_clear(&a)
|
||
|
|
mp_clear(&b)
|
||
|
|
mp_clear(&c)
|
||
|
|
mp_clear(&d)
|
||
|
|
}
|
||
|
|
|
||
|
|
XCTAssert(mp_error_to_string(MP_OKAY) != nil)
|
||
|
|
|
||
|
|
/* a: 0->5 */
|
||
|
|
mp_set(&a, 5)
|
||
|
|
/* a: 5-> b: -5 */
|
||
|
|
XCTAssertEqual(mp_neg(&a, &b), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_cmp(&a, &b), MP_GT)
|
||
|
|
XCTAssertEqual(mp_cmp(&b, &a), MP_LT)
|
||
|
|
XCTAssertTrue(b.isNeg)
|
||
|
|
/* a: 5-> a: -5 */
|
||
|
|
var t = a // Fix compiler error: Overlapping accesses to 'a', but modification requires exclusive access; consider copying to a local variable
|
||
|
|
XCTAssertEqual(mp_neg(&t, &a), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_cmp(&b, &a), MP_EQ)
|
||
|
|
XCTAssertTrue(a.isNeg)
|
||
|
|
/* a: -5-> b: 5 */
|
||
|
|
XCTAssertEqual(mp_abs(&a, &b), MP_OKAY)
|
||
|
|
XCTAssertTrue(!b.isNeg)
|
||
|
|
/* a: -5-> b: -4 */
|
||
|
|
XCTAssertEqual(mp_add_d(&a, 1, &b), MP_OKAY)
|
||
|
|
XCTAssertTrue(b.isNeg)
|
||
|
|
XCTAssertEqual(mp_get_i32(&b), -4)
|
||
|
|
XCTAssertEqual(mp_get_u32(&b), UInt32(bitPattern: -4))
|
||
|
|
XCTAssertEqual(mp_get_mag_u32(&b), 4)
|
||
|
|
/* a: -5-> b: 1 */
|
||
|
|
XCTAssertEqual(mp_add_d(&a, 6, &b), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_get_u32(&b), 1)
|
||
|
|
/* a: -5-> a: 1 */
|
||
|
|
t = a
|
||
|
|
XCTAssertEqual(mp_add_d(&t, 6, &a), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_get_u32(&a), 1)
|
||
|
|
mp_zero(&a);
|
||
|
|
/* a: 0-> a: 6 */
|
||
|
|
t = a
|
||
|
|
XCTAssertEqual(mp_add_d(&t, 6, &a), MP_OKAY)
|
||
|
|
XCTAssertEqual(mp_get_u32(&a), 6)
|
||
|
|
|
||
|
|
mp_set(&a, 42)
|
||
|
|
mp_set(&b, 1)
|
||
|
|
t = b
|
||
|
|
XCTAssertEqual(mp_neg(&t, &b), MP_OKAY)
|
||
|
|
mp_set(&c, 1)
|
||
|
|
XCTAssertEqual(mp_exptmod(&a, &b, &c, &d), MP_OKAY)
|
||
|
|
|
||
|
|
mp_set(&c, 7)
|
||
|
|
/* same here */
|
||
|
|
XCTAssertTrue(mp_exptmod(&a, &b, &c, &d) != MP_OKAY)
|
||
|
|
|
||
|
|
XCTAssertTrue(a.isEven != a.isOdd)
|
||
|
|
}
|
||
|
|
}
|