SignedSafeMath库就是直接对solidity内置的int256类型的加减乘除运算的函数封装。
[openzeppelin]:v4.8.3,[forge-std]:v1.5.6
SignedSafeMath库就是直接对solidity内置的int256类型的加减乘除运算的函数封装。
注:从solidity 0.8版本开始,SignedSafeMath已经不再被广泛使用。因为从0.8版本开始,solidity编译器已经内置了运算符的溢出检查。
封装SignedSafeMath library成为一个可调用合约:
// SPDX-License-Identifier: UNLICENSED
pragma solidity ^0.8.0;
import "openzeppelin-contracts/contracts/utils/math/SignedSafeMath.sol";
contract MockSignedSafeMath {
using SignedSafeMath for int;
function mul(int a, int b) external pure returns (int) {
return a.mul(b);
}
function div(int a, int b) external pure returns (int){
return a.div(b);
}
function sub(int a, int b) external pure returns (int){
return a.sub(b);
}
function add(int a, int b) external pure returns (int) {
return a.add(b);
}
}
全部foundry测试合约:
返回两个int256的乘积,如果产生溢出会revert。
function mul(int256 a, int256 b) internal pure returns (int256) {
// 是直接对a*b的函数封装,返回a与b的乘积
return a * b;
}
foundry代码验证
contract SignedSafeMathTest is Test {
MockSignedSafeMath mssm = new MockSignedSafeMath();
function test_Mul() external {
assertEq(mssm.mul(3, 2), 6);
assertEq(mssm.mul(3, - 2), - 6);
assertEq(mssm.mul(- 3, 2), - 6);
assertEq(mssm.mul(- 3, - 2), 6);
// overflow
vm.expectRevert();
mssm.mul(type(int).max, 2);
vm.expectRevert();
mssm.mul(type(int).min, 2);
}
}
返回两个int256的商,如果除数为0会revert。
注:计算结果向0取整。
function div(int256 a, int256 b) internal pure returns (int256) {
// 是直接对a/b的函数封装,返回a与b的商
return a / b;
}
foundry代码验证
contract SignedSafeMathTest is Test {
MockSignedSafeMath mssm = new MockSignedSafeMath();
function test_Div() external {
assertEq(mssm.div(3, 2), 1);
assertEq(mssm.div(3, - 2), - 1);
assertEq(mssm.div(- 3, 2), - 1);
assertEq(mssm.div(- 3, - 2), 1);
// overflow
vm.expectRevert();
mssm.div(1, 0);
}
}
返回两个int256的差,如果产生溢出会revert
function sub(int256 a, int256 b) internal pure returns (int256) {
// 是直接对a-b的函数封装,返回a与b的差
return a - b;
}
foundry代码验证
contract SignedSafeMathTest is Test {
MockSignedSafeMath mssm = new MockSignedSafeMath();
function test_Sub() external {
assertEq(mssm.sub(3, 2), 1);
assertEq(mssm.sub(3, - 2), 5);
assertEq(mssm.sub(- 3, 2), - 5);
assertEq(mssm.sub(- 3, - 2), - 1);
// overflow
vm.expectRevert();
mssm.sub(type(int).min, 1);
}
}
返回两个int256的和,如果产生溢出会revert。
function add(int256 a, int256 b) internal pure returns (int256) {
// 是直接对a+b的函数封装,返回a与b的和
return a + b;
}
foundry代码验证
contract SignedSafeMathTest is Test {
MockSignedSafeMath mssm = new MockSignedSafeMath();
function test_Add() external {
assertEq(mssm.add(1, 2), 3);
assertEq(mssm.add(- 1, 2), 1);
assertEq(mssm.add(- 1, - 2), - 3);
// overflow
vm.expectRevert();
mssm.add(type(int).max, 1);
}
}
ps:\ 本人热爱图灵,热爱中本聪,热爱V神。 以下是我个人的公众号,如果有技术问题可以关注我的公众号来跟我交流。 同时我也会在这个公众号上每周更新我的原创文章,喜欢的小伙伴或者老伙计可以支持一下! 如果需要转发,麻烦注明作者。十分感谢!
公众号名称:后现代泼痞浪漫主义奠基人
如果觉得我的文章对您有用,请随意打赏。你的支持将鼓励我继续创作!