/* This file is part of solidity. solidity is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. solidity is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with solidity. If not, see . */ // SPDX-License-Identifier: GPL-3.0 /** * Component that can generate various useful Yul functions. */ #include #include #include #include #include #include #include #include using namespace std; using namespace solidity; using namespace solidity::util; using namespace solidity::frontend; string YulUtilFunctions::combineExternalFunctionIdFunction() { string functionName = "combine_external_function_id"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (addr, selector) -> combined { combined := (or((addr), and(selector, 0xffffffff))) } )") ("functionName", functionName) ("shl32", shiftLeftFunction(32)) ("shl64", shiftLeftFunction(64)) .render(); }); } string YulUtilFunctions::splitExternalFunctionIdFunction() { string functionName = "split_external_function_id"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (combined) -> addr, selector { combined := (combined) selector := and(combined, 0xffffffff) addr := (combined) } )") ("functionName", functionName) ("shr32", shiftRightFunction(32)) ("shr64", shiftRightFunction(64)) .render(); }); } string YulUtilFunctions::copyToMemoryFunction(bool _fromCalldata) { string functionName = "copy_" + string(_fromCalldata ? "calldata" : "memory") + "_to_memory"; return m_functionCollector.createFunction(functionName, [&]() { if (_fromCalldata) { return Whiskers(R"( function (src, dst, length) { calldatacopy(dst, src, length) // clear end mstore(add(dst, length), 0) } )") ("functionName", functionName) .render(); } else { return Whiskers(R"( function (src, dst, length) { let i := 0 for { } lt(i, length) { i := add(i, 32) } { mstore(add(dst, i), mload(add(src, i))) } if gt(i, length) { // clear end mstore(add(dst, length), 0) } } )") ("functionName", functionName) .render(); } }); } string YulUtilFunctions::requireOrAssertFunction(bool _assert, Type const* _messageType) { string functionName = string(_assert ? "assert_helper" : "require_helper") + (_messageType ? ("_" + _messageType->identifier()) : ""); solAssert(!_assert || !_messageType, "Asserts can't have messages!"); return m_functionCollector.createFunction(functionName, [&]() { if (!_messageType) return Whiskers(R"( function (condition) { if iszero(condition) { } } )") ("error", _assert ? panicFunction() + "()" : "revert(0, 0)") ("functionName", functionName) .render(); int const hashHeaderSize = 4; u256 const errorHash = util::selectorFromSignature("Error(string)"); string const encodeFunc = ABIFunctions(m_evmVersion, m_revertStrings, m_functionCollector) .tupleEncoder( {_messageType}, {TypeProvider::stringMemory()} ); return Whiskers(R"( function (condition ) { if iszero(condition) { let fmp := mload() mstore(fmp, ) let end := (add(fmp, ) ) revert(fmp, sub(end, fmp)) } } )") ("functionName", functionName) ("freeMemPointer", to_string(CompilerUtils::freeMemoryPointer)) ("errorHash", formatNumber(errorHash)) ("abiEncodeFunc", encodeFunc) ("hashHeaderSize", to_string(hashHeaderSize)) ("messageVars", (_messageType->sizeOnStack() > 0 ? ", " : "") + suffixedVariableNameList("message_", 1, 1 + _messageType->sizeOnStack()) ) .render(); }); } string YulUtilFunctions::leftAlignFunction(Type const& _type) { string functionName = string("leftAlign_") + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (value) -> aligned { } )"); templ("functionName", functionName); switch (_type.category()) { case Type::Category::Address: templ("body", "aligned := " + leftAlignFunction(IntegerType(160)) + "(value)"); break; case Type::Category::Integer: { IntegerType const& type = dynamic_cast(_type); if (type.numBits() == 256) templ("body", "aligned := value"); else templ("body", "aligned := " + shiftLeftFunction(256 - type.numBits()) + "(value)"); break; } case Type::Category::RationalNumber: solAssert(false, "Left align requested for rational number."); break; case Type::Category::Bool: templ("body", "aligned := " + leftAlignFunction(IntegerType(8)) + "(value)"); break; case Type::Category::FixedPoint: solUnimplemented("Fixed point types not implemented."); break; case Type::Category::Array: case Type::Category::Struct: solAssert(false, "Left align requested for non-value type."); break; case Type::Category::FixedBytes: templ("body", "aligned := value"); break; case Type::Category::Contract: templ("body", "aligned := " + leftAlignFunction(*TypeProvider::address()) + "(value)"); break; case Type::Category::Enum: { unsigned storageBytes = dynamic_cast(_type).storageBytes(); templ("body", "aligned := " + leftAlignFunction(IntegerType(8 * storageBytes)) + "(value)"); break; } case Type::Category::InaccessibleDynamic: solAssert(false, "Left align requested for inaccessible dynamic type."); break; default: solAssert(false, "Left align of type " + _type.identifier() + " requested."); } return templ.render(); }); } string YulUtilFunctions::shiftLeftFunction(size_t _numBits) { solAssert(_numBits < 256, ""); string functionName = "shift_left_" + to_string(_numBits); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value) -> newValue { newValue := shl(, value) mul(value, ) } )") ("functionName", functionName) ("numBits", to_string(_numBits)) ("hasShifts", m_evmVersion.hasBitwiseShifting()) ("multiplier", toCompactHexWithPrefix(u256(1) << _numBits)) .render(); }); } string YulUtilFunctions::shiftLeftFunctionDynamic() { string functionName = "shift_left_dynamic"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (bits, value) -> newValue { newValue := shl(bits, value) mul(value, exp(2, bits)) } )") ("functionName", functionName) ("hasShifts", m_evmVersion.hasBitwiseShifting()) .render(); }); } string YulUtilFunctions::shiftRightFunction(size_t _numBits) { solAssert(_numBits < 256, ""); // Note that if this is extended with signed shifts, // the opcodes SAR and SDIV behave differently with regards to rounding! string functionName = "shift_right_" + to_string(_numBits) + "_unsigned"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value) -> newValue { newValue := shr(, value) div(value, ) } )") ("functionName", functionName) ("hasShifts", m_evmVersion.hasBitwiseShifting()) ("numBits", to_string(_numBits)) ("multiplier", toCompactHexWithPrefix(u256(1) << _numBits)) .render(); }); } string YulUtilFunctions::shiftRightFunctionDynamic() { string const functionName = "shift_right_unsigned_dynamic"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (bits, value) -> newValue { newValue := shr(bits, value) div(value, exp(2, bits)) } )") ("functionName", functionName) ("hasShifts", m_evmVersion.hasBitwiseShifting()) .render(); }); } string YulUtilFunctions::shiftRightSignedFunctionDynamic() { string const functionName = "shift_right_signed_dynamic"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (bits, value) -> result { result := sar(bits, value) let divisor := exp(2, bits) let xor_mask := sub(0, slt(value, 0)) result := xor(div(xor(value, xor_mask), divisor), xor_mask) // combined version of // switch slt(value, 0) // case 0 { result := div(value, divisor) } // default { result := not(div(not(value), divisor)) } } )") ("functionName", functionName) ("hasShifts", m_evmVersion.hasBitwiseShifting()) .render(); }); } string YulUtilFunctions::typedShiftLeftFunction(Type const& _type, Type const& _amountType) { solAssert(_type.category() == Type::Category::FixedBytes || _type.category() == Type::Category::Integer, ""); solAssert(_amountType.category() == Type::Category::Integer, ""); solAssert(!dynamic_cast(_amountType).isSigned(), ""); string const functionName = "shift_left_" + _type.identifier() + "_" + _amountType.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value, bits) -> result { bits := (bits) result := ((bits, value)) } )") ("functionName", functionName) ("cleanAmount", cleanupFunction(_amountType)) ("shift", shiftLeftFunctionDynamic()) ("cleanup", cleanupFunction(_type)) .render(); }); } string YulUtilFunctions::typedShiftRightFunction(Type const& _type, Type const& _amountType) { solAssert(_type.category() == Type::Category::FixedBytes || _type.category() == Type::Category::Integer, ""); solAssert(_amountType.category() == Type::Category::Integer, ""); solAssert(!dynamic_cast(_amountType).isSigned(), ""); IntegerType const* integerType = dynamic_cast(&_type); bool valueSigned = integerType && integerType->isSigned(); string const functionName = "shift_right_" + _type.identifier() + "_" + _amountType.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value, bits) -> result { bits := (bits) result := ((bits, (value))) } )") ("functionName", functionName) ("cleanAmount", cleanupFunction(_amountType)) ("shift", valueSigned ? shiftRightSignedFunctionDynamic() : shiftRightFunctionDynamic()) ("cleanup", cleanupFunction(_type)) .render(); }); } string YulUtilFunctions::updateByteSliceFunction(size_t _numBytes, size_t _shiftBytes) { solAssert(_numBytes <= 32, ""); solAssert(_shiftBytes <= 32, ""); size_t numBits = _numBytes * 8; size_t shiftBits = _shiftBytes * 8; string functionName = "update_byte_slice_" + to_string(_numBytes) + "_shift_" + to_string(_shiftBytes); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value, toInsert) -> result { let mask := toInsert := (toInsert) value := and(value, not(mask)) result := or(value, and(toInsert, mask)) } )") ("functionName", functionName) ("mask", formatNumber(((bigint(1) << numBits) - 1) << shiftBits)) ("shl", shiftLeftFunction(shiftBits)) .render(); }); } string YulUtilFunctions::updateByteSliceFunctionDynamic(size_t _numBytes) { solAssert(_numBytes <= 32, ""); size_t numBits = _numBytes * 8; string functionName = "update_byte_slice_dynamic" + to_string(_numBytes); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value, shiftBytes, toInsert) -> result { let shiftBits := mul(shiftBytes, 8) let mask := (shiftBits, ) toInsert := (shiftBits, toInsert) value := and(value, not(mask)) result := or(value, and(toInsert, mask)) } )") ("functionName", functionName) ("mask", formatNumber((bigint(1) << numBits) - 1)) ("shl", shiftLeftFunctionDynamic()) .render(); }); } string YulUtilFunctions::maskBytesFunctionDynamic() { string functionName = "mask_bytes_dynamic"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (data, bytes) -> result { let mask := not((mul(8, bytes), not(0))) result := and(data, mask) })") ("functionName", functionName) ("shr", shiftRightFunctionDynamic()) .render(); }); } string YulUtilFunctions::maskLowerOrderBytesFunction(size_t _bytes) { string functionName = "mask_lower_order_bytes_" + to_string(_bytes); solAssert(_bytes <= 32, ""); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (data) -> result { result := and(data, ) })") ("functionName", functionName) ("mask", formatNumber((~u256(0)) >> (256 - 8 * _bytes))) .render(); }); } string YulUtilFunctions::maskLowerOrderBytesFunctionDynamic() { string functionName = "mask_lower_order_bytes_dynamic"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (data, bytes) -> result { let mask := not((mul(8, bytes), not(0))) result := and(data, mask) })") ("functionName", functionName) ("shl", shiftLeftFunctionDynamic()) .render(); }); } string YulUtilFunctions::roundUpFunction() { string functionName = "round_up_to_mul_of_32"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value) -> result { result := and(add(value, 31), not(31)) } )") ("functionName", functionName) .render(); }); } string YulUtilFunctions::overflowCheckedIntAddFunction(IntegerType const& _type) { string functionName = "checked_add_" + _type.identifier(); // TODO: Consider to add a special case for unsigned 256-bit integers // and use the following instead: // sum := add(x, y) if lt(sum, x) { () } return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (x, y) -> sum { x := (x) y := (y) // overflow, if x >= 0 and y > (maxValue - x) if and(iszero(slt(x, 0)), sgt(y, sub(, x))) { () } // underflow, if x < 0 and y < (minValue - x) if and(slt(x, 0), slt(y, sub(, x))) { () } // overflow, if x > (maxValue - y) if gt(x, sub(, y)) { () } sum := add(x, y) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("maxValue", toCompactHexWithPrefix(u256(_type.maxValue()))) ("minValue", toCompactHexWithPrefix(u256(_type.minValue()))) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::overflowCheckedIntMulFunction(IntegerType const& _type) { string functionName = "checked_mul_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return // Multiplication by zero could be treated separately and directly return zero. Whiskers(R"( function (x, y) -> product { x := (x) y := (y) // overflow, if x > 0, y > 0 and x > (maxValue / y) if and(and(sgt(x, 0), sgt(y, 0)), gt(x, div(, y))) { () } // underflow, if x > 0, y < 0 and y < (minValue / x) if and(and(sgt(x, 0), slt(y, 0)), slt(y, sdiv(, x))) { () } // underflow, if x < 0, y > 0 and x < (minValue / y) if and(and(slt(x, 0), sgt(y, 0)), slt(x, sdiv(, y))) { () } // overflow, if x < 0, y < 0 and x < (maxValue / y) if and(and(slt(x, 0), slt(y, 0)), slt(x, sdiv(, y))) { () } // overflow, if x != 0 and y > (maxValue / x) if and(iszero(iszero(x)), gt(y, div(, x))) { () } product := mul(x, y) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("maxValue", toCompactHexWithPrefix(u256(_type.maxValue()))) ("minValue", toCompactHexWithPrefix(u256(_type.minValue()))) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::overflowCheckedIntDivFunction(IntegerType const& _type) { string functionName = "checked_div_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (x, y) -> r { x := (x) y := (y) if iszero(y) { () } // overflow for minVal / -1 if and( eq(x, ), eq(y, sub(0, 1)) ) { () } r := sdiv(x, y) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("minVal", toCompactHexWithPrefix(u256(_type.minValue()))) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::checkedIntModFunction(IntegerType const& _type) { string functionName = "checked_mod_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (x, y) -> r { x := (x) y := (y) if iszero(y) { () } r := smod(x, y) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::overflowCheckedIntSubFunction(IntegerType const& _type) { string functionName = "checked_sub_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { return Whiskers(R"( function (x, y) -> diff { x := (x) y := (y) // underflow, if y >= 0 and x < (minValue + y) if and(iszero(slt(y, 0)), slt(x, add(, y))) { () } // overflow, if y < 0 and x > (maxValue + y) if and(slt(y, 0), sgt(x, add(, y))) { () } if lt(x, y) { () } diff := sub(x, y) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("maxValue", toCompactHexWithPrefix(u256(_type.maxValue()))) ("minValue", toCompactHexWithPrefix(u256(_type.minValue()))) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::overflowCheckedIntExpFunction( IntegerType const& _type, IntegerType const& _exponentType ) { solAssert(!_exponentType.isSigned(), ""); string functionName = "checked_exp_" + _type.identifier() + "_" + _exponentType.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (base, exponent) -> power { base := (base) exponent := (exponent) power := (base, exponent, , ) power := (base, exponent, ) } )") ("functionName", functionName) ("signed", _type.isSigned()) ("exp", _type.isSigned() ? overflowCheckedSignedExpFunction() : overflowCheckedUnsignedExpFunction()) ("maxValue", toCompactHexWithPrefix(_type.max())) ("minValue", toCompactHexWithPrefix(_type.min())) ("baseCleanupFunction", cleanupFunction(_type)) ("exponentCleanupFunction", cleanupFunction(_exponentType)) .render(); }); } string YulUtilFunctions::overflowCheckedIntLiteralExpFunction( RationalNumberType const& _baseType, IntegerType const& _exponentType, IntegerType const& _commonType ) { solAssert(!_exponentType.isSigned(), ""); solAssert(_baseType.isNegative() == _commonType.isSigned(), ""); solAssert(_commonType.numBits() == 256, ""); string functionName = "checked_exp_" + _baseType.richIdentifier() + "_" + _exponentType.identifier(); return m_functionCollector.createFunction(functionName, [&]() { // Converts a bigint number into u256 (negative numbers represented in two's complement form.) // We assume that `_v` fits in 256 bits. auto bigint2u = [&](bigint const& _v) -> u256 { if (_v < 0) return s2u(s256(_v)); return u256(_v); }; // Calculates the upperbound for exponentiation, that is, calculate `b`, such that // _base**b <= _maxValue and _base**(b + 1) > _maxValue auto findExponentUpperbound = [](bigint const _base, bigint const _maxValue) -> unsigned { // There is no overflow for these cases if (_base == 0 || _base == -1 || _base == 1) return 0; unsigned first = 0; unsigned last = 255; unsigned middle; while (first < last) { middle = (first + last) / 2; if ( // The condition on msb is a shortcut that avoids computing large powers in // arbitrary precision. boost::multiprecision::msb(_base) * middle <= boost::multiprecision::msb(_maxValue) && boost::multiprecision::pow(_base, middle) <= _maxValue ) { if (boost::multiprecision::pow(_base, middle + 1) > _maxValue) return middle; else first = middle + 1; } else last = middle; } return last; }; bigint baseValue = _baseType.isNegative() ? u2s(_baseType.literalValue(nullptr)) : _baseType.literalValue(nullptr); bool needsOverflowCheck = !((baseValue == 0) || (baseValue == -1) || (baseValue == 1)); unsigned exponentUpperbound; if (_baseType.isNegative()) { // Only checks for underflow. The only case where this can be a problem is when, for a // negative base, say `b`, and an even exponent, say `e`, `b**e = 2**255` (which is an // overflow.) But this never happens because, `255 = 3*5*17`, and therefore there is no even // number `e` such that `b**e = 2**255`. exponentUpperbound = findExponentUpperbound(abs(baseValue), abs(_commonType.minValue())); bigint power = boost::multiprecision::pow(baseValue, exponentUpperbound); bigint overflowedPower = boost::multiprecision::pow(baseValue, exponentUpperbound + 1); if (needsOverflowCheck) solAssert( (power <= _commonType.maxValue()) && (power >= _commonType.minValue()) && !((overflowedPower <= _commonType.maxValue()) && (overflowedPower >= _commonType.minValue())), "Incorrect exponent upper bound calculated." ); } else { exponentUpperbound = findExponentUpperbound(baseValue, _commonType.maxValue()); if (needsOverflowCheck) solAssert( boost::multiprecision::pow(baseValue, exponentUpperbound) <= _commonType.maxValue() && boost::multiprecision::pow(baseValue, exponentUpperbound + 1) > _commonType.maxValue(), "Incorrect exponent upper bound calculated." ); } return Whiskers(R"( function (exponent) -> power { exponent := (exponent) if gt(exponent, ) { () } power := exp(, exponent) } )") ("functionName", functionName) ("exponentCleanupFunction", cleanupFunction(_exponentType)) ("needsOverflowCheck", needsOverflowCheck) ("exponentUpperbound", to_string(exponentUpperbound)) ("panic", panicFunction()) ("base", bigint2u(baseValue).str()) .render(); }); } string YulUtilFunctions::overflowCheckedUnsignedExpFunction() { // Checks for the "small number specialization" below. using namespace boost::multiprecision; solAssert(pow(bigint(10), 77) < pow(bigint(2), 256), ""); solAssert(pow(bigint(11), 77) >= pow(bigint(2), 256), ""); solAssert(pow(bigint(10), 78) >= pow(bigint(2), 256), ""); solAssert(pow(bigint(306), 31) < pow(bigint(2), 256), ""); solAssert(pow(bigint(307), 31) >= pow(bigint(2), 256), ""); solAssert(pow(bigint(306), 32) >= pow(bigint(2), 256), ""); string functionName = "checked_exp_unsigned"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (base, exponent, max) -> power { // This function currently cannot be inlined because of the // "leave" statements. We have to improve the optimizer. // Note that 0**0 == 1 if iszero(exponent) { power := 1 leave } if iszero(base) { power := 0 leave } // Specializations for small bases switch base // 0 is handled above case 1 { power := 1 leave } case 2 { if gt(exponent, 255) { () } power := exp(2, exponent) if gt(power, max) { () } leave } if or( and(lt(base, 11), lt(exponent, 78)), and(lt(base, 307), lt(exponent, 32)) ) { power := exp(base, exponent) if gt(power, max) { () } leave } power, base := (1, base, exponent, max) if gt(power, div(max, base)) { () } power := mul(power, base) } )") ("functionName", functionName) ("panic", panicFunction()) ("expLoop", overflowCheckedExpLoopFunction()) ("shr_1", shiftRightFunction(1)) .render(); }); } string YulUtilFunctions::overflowCheckedSignedExpFunction() { string functionName = "checked_exp_signed"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (base, exponent, min, max) -> power { // Currently, `leave` avoids this function being inlined. // We have to improve the optimizer. // Note that 0**0 == 1 switch exponent case 0 { power := 1 leave } case 1 { power := base leave } if iszero(base) { power := 0 leave } power := 1 // We pull out the first iteration because it is the only one in which // base can be negative. // Exponent is at least 2 here. // overflow check for base * base switch sgt(base, 0) case 1 { if gt(base, div(max, base)) { () } } case 0 { if slt(base, sdiv(max, base)) { () } } if and(exponent, 1) { power := base } base := mul(base, base) exponent := (exponent) // Below this point, base is always positive. power, base := (power, base, exponent, max) if and(sgt(power, 0), gt(power, div(max, base))) { () } if and(slt(power, 0), slt(power, sdiv(min, base))) { () } power := mul(power, base) } )") ("functionName", functionName) ("panic", panicFunction()) ("expLoop", overflowCheckedExpLoopFunction()) ("shr_1", shiftRightFunction(1)) .render(); }); } string YulUtilFunctions::overflowCheckedExpLoopFunction() { // We use this loop for both signed and unsigned exponentiation // because we pull out the first iteration in the signed case which // results in the base always being positive. // This function does not include the final multiplication. string functionName = "checked_exp_helper"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (_power, _base, exponent, max) -> power, base { power := _power base := _base for { } gt(exponent, 1) {} { // overflow check for base * base if gt(base, div(max, base)) { () } if and(exponent, 1) { // No checks for power := mul(power, base) needed, because the check // for base * base above is sufficient, since: // |power| <= base (proof by induction) and thus: // |power * base| <= base * base <= max <= |min| (for signed) // (this is equally true for signed and unsigned exp) power := mul(power, base) } base := mul(base, base) exponent := (exponent) } } )") ("functionName", functionName) ("panic", panicFunction()) ("shr_1", shiftRightFunction(1)) .render(); }); } string YulUtilFunctions::extractByteArrayLengthFunction() { string functionName = "extract_byte_array_length"; return m_functionCollector.createFunction(functionName, [&]() { Whiskers w(R"( function (data) -> length { // Retrieve length both for in-place strings and off-place strings: // Computes (x & (0x100 * (ISZERO (x & 1)) - 1)) / 2 // i.e. for short strings (x & 1 == 0) it does (x & 0xff) / 2 and for long strings it // computes (x & (-1)) / 2, which is equivalent to just x / 2. let mask := sub(mul(0x100, iszero(and(data, 1))), 1) length := div(and(data, mask), 2) } )"); w("functionName", functionName); return w.render(); }); } string YulUtilFunctions::arrayLengthFunction(ArrayType const& _type) { string functionName = "array_length_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers w(R"( function (value, len) -> length { length := mload(value) length := sload(value) length := (length) length := len length := } )"); w("functionName", functionName); w("dynamic", _type.isDynamicallySized()); if (!_type.isDynamicallySized()) w("length", toCompactHexWithPrefix(_type.length())); w("memory", _type.location() == DataLocation::Memory); w("storage", _type.location() == DataLocation::Storage); w("calldata", _type.location() == DataLocation::CallData); if (_type.location() == DataLocation::Storage) { w("byteArray", _type.isByteArray()); if (_type.isByteArray()) w("extractByteArrayLength", extractByteArrayLengthFunction()); } return w.render(); }); } std::string YulUtilFunctions::resizeDynamicArrayFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); solAssert(_type.isDynamicallySized(), ""); solUnimplementedAssert(_type.baseType()->storageBytes() <= 32, "..."); if (_type.isByteArray()) return resizeDynamicByteArrayFunction(_type); string functionName = "resize_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, newLen) { if gt(newLen, ) { () } let oldLen := (array) // Store new length sstore(array, newLen) // Size was reduced, clear end of array if lt(newLen, oldLen) { let oldSlotCount := (oldLen) let newSlotCount := (newLen) let arrayDataStart := (array) let deleteStart := add(arrayDataStart, newSlotCount) let deleteEnd := add(arrayDataStart, oldSlotCount) // if we are dealing with packed array and offset is greater than zero // we have to partially clear last slot that is still used, so decreasing start by one let offset := mul(mod(newLen, ), ) if gt(offset, 0) { (sub(deleteStart, 1), offset) } (deleteStart, deleteEnd) } })") ("functionName", functionName) ("panic", panicFunction()) ("fetchLength", arrayLengthFunction(_type)) ("convertToSize", arrayConvertLengthToSize(_type)) ("dataPosition", arrayDataAreaFunction(_type)) ("clearStorageRange", clearStorageRangeFunction(*_type.baseType())) ("maxArrayLength", (u256(1) << 64).str()) ("packed", _type.baseType()->storageBytes() <= 16) ("itemsPerSlot", to_string(32 / _type.baseType()->storageBytes())) ("storageBytes", to_string(_type.baseType()->storageBytes())) ("partialClearStorageSlot", partialClearStorageSlotFunction()) .render(); }); } string YulUtilFunctions::resizeDynamicByteArrayFunction(ArrayType const& _type) { string functionName = "resize_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, newLen) { if gt(newLen, ) { () } let data := sload(array) let oldLen := (data) if gt(newLen, oldLen) { (array, data, oldLen, newLen) } if lt(newLen, oldLen) { (array, data, oldLen, newLen) } })") ("functionName", functionName) ("panic", panicFunction()) ("extractLength", extractByteArrayLengthFunction()) ("maxArrayLength", (u256(1) << 64).str()) ("decreaseSize", decreaseByteArraySizeFunction(_type)) ("increaseSize", increaseByteArraySizeFunction(_type)) .render(); }); } string YulUtilFunctions::decreaseByteArraySizeFunction(ArrayType const& _type) { string functionName = "byte_array_decrease_size_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, data, oldLen, newLen) { switch lt(newLen, 32) case 0 { let arrayDataStart := (array) let deleteStart := add(arrayDataStart, div(add(newLen, 31), 32)) // we have to partially clear last slot that is still used let offset := and(newLen, 0x1f) if offset { (sub(deleteStart, 1), offset) } (deleteStart, add(arrayDataStart, div(add(oldLen, 31), 32))) sstore(array, or(mul(2, newLen), 1)) } default { switch gt(oldLen, 31) case 1 { let arrayDataStart := (array) // clear whole old array, as we are transforming to short bytes array (add(arrayDataStart, 1), add(arrayDataStart, div(add(oldLen, 31), 32))) (array, newLen) } default { sstore(array, (data, newLen)) } } })") ("functionName", functionName) ("dataPosition", arrayDataAreaFunction(_type)) ("partialClearStorageSlot", partialClearStorageSlotFunction()) ("clearStorageRange", clearStorageRangeFunction(*_type.baseType())) ("transitLongToShort", byteArrayTransitLongToShortFunction(_type)) ("encodeUsedSetLen", shortByteArrayEncodeUsedAreaSetLengthFunction()) .render(); }); } string YulUtilFunctions::increaseByteArraySizeFunction(ArrayType const& _type) { string functionName = "byte_array_increase_size_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, data, oldLen, newLen) { switch lt(oldLen, 32) case 0 { // in this case array stays unpacked, so we just set new length sstore(array, add(mul(2, newLen), 1)) } default { switch lt(newLen, 32) case 0 { // we need to copy elements to data area as we changed array from packed to unpacked data := and(not(0xff), data) sstore((array), data) sstore(array, add(mul(2, newLen), 1)) } default { // here array stays packed, we just need to increase length sstore(array, (data, newLen)) } } })") ("functionName", functionName) ("dataPosition", arrayDataAreaFunction(_type)) ("encodeUsedSetLen", shortByteArrayEncodeUsedAreaSetLengthFunction()) .render(); }); } string YulUtilFunctions::byteArrayTransitLongToShortFunction(ArrayType const& _type) { string functionName = "transit_byte_array_long_to_short_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, len) { // we need to copy elements from old array to new // we want to copy only elements that are part of the array after resizing let dataPos := (array) let data := (sload(dataPos), len) sstore(array, data) sstore(dataPos, 0) })") ("functionName", functionName) ("dataPosition", arrayDataAreaFunction(_type)) ("extractUsedApplyLen", shortByteArrayEncodeUsedAreaSetLengthFunction()) ("shl", shiftLeftFunctionDynamic()) ("ones", formatNumber((bigint(1) << 256) - 1)) .render(); }); } string YulUtilFunctions::shortByteArrayEncodeUsedAreaSetLengthFunction() { string functionName = "extract_used_part_and_set_length_of_short_byte_array"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (data, len) -> used { // we want to save only elements that are part of the array after resizing // others should be set to zero data := (data, len) used := or(data, mul(2, len)) })") ("functionName", functionName) ("maskBytes", maskBytesFunctionDynamic()) .render(); }); } string YulUtilFunctions::storageArrayPopFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); solAssert(_type.isDynamicallySized(), ""); solUnimplementedAssert(_type.baseType()->storageBytes() <= 32, "Base type is not yet implemented."); if (_type.isByteArray()) return storageByteArrayPopFunction(_type); string functionName = "array_pop_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array) { let oldLen := (array) if iszero(oldLen) { () } let newLen := sub(oldLen, 1) let slot, offset := (array, newLen) (slot, offset) sstore(array, newLen) })") ("functionName", functionName) ("panic", panicFunction()) ("fetchLength", arrayLengthFunction(_type)) ("indexAccess", storageArrayIndexAccessFunction(_type)) ("setToZero", storageSetToZeroFunction(*_type.baseType())) .render(); }); } string YulUtilFunctions::storageByteArrayPopFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); solAssert(_type.isDynamicallySized(), ""); solAssert(_type.isByteArray(), ""); string functionName = "byte_array_pop_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array) { let data := sload(array) let oldLen := (data) if iszero(oldLen) { () } switch oldLen case 32 { // Here we have a special case where array transitions to shorter than 32 // So we need to copy data (array, 31) } default { let newLen := sub(oldLen, 1) switch lt(oldLen, 32) case 1 { sstore(array, (data, newLen)) } default { let slot, offset := (array, newLen) (slot, offset) sstore(array, sub(data, 2)) } } })") ("functionName", functionName) ("panic", panicFunction()) ("extractByteArrayLength", extractByteArrayLengthFunction()) ("transitLongToShort", byteArrayTransitLongToShortFunction(_type)) ("encodeUsedSetLen", shortByteArrayEncodeUsedAreaSetLengthFunction()) ("indexAccess", storageArrayIndexAccessFunction(_type)) ("setToZero", storageSetToZeroFunction(*_type.baseType())) ("shl", shiftLeftFunctionDynamic()) .render(); }); } string YulUtilFunctions::storageArrayPushFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); solAssert(_type.isDynamicallySized(), ""); string functionName = "array_push_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, value) { let data := sload(array) let oldLen := (data) if iszero(lt(oldLen, )) { () } switch gt(oldLen, 31) case 0 { value := byte(0, value) switch oldLen case 31 { // Here we have special case when array switches from short array to long array // We need to copy data let dataArea := (array) data := and(data, not(0xff)) sstore(dataArea, or(and(0xff, value), data)) // New length is 32, encoded as (32 * 2 + 1) sstore(array, 65) } default { data := add(data, 2) let shiftBits := mul(8, sub(31, oldLen)) let valueShifted := (shiftBits, and(0xff, value)) let mask := (shiftBits, 0xff) data := or(and(data, not(mask)), valueShifted) sstore(array, data) } } default { sstore(array, add(data, 2)) let slot, offset := (array, oldLen) (slot, offset, value) } let oldLen := sload(array) if iszero(lt(oldLen, )) { () } sstore(array, add(oldLen, 1)) let slot, offset := (array, oldLen) (slot, offset, value) })") ("functionName", functionName) ("panic", panicFunction()) ("extractByteArrayLength", _type.isByteArray() ? extractByteArrayLengthFunction() : "") ("dataAreaFunction", arrayDataAreaFunction(_type)) ("isByteArray", _type.isByteArray()) ("indexAccess", storageArrayIndexAccessFunction(_type)) ("storeValue", updateStorageValueFunction(*_type.baseType(), *_type.baseType())) ("maxArrayLength", (u256(1) << 64).str()) ("shl", shiftLeftFunctionDynamic()) ("shr", shiftRightFunction(248)) .render(); }); } string YulUtilFunctions::storageArrayPushZeroFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); solAssert(_type.isDynamicallySized(), ""); solUnimplementedAssert(!_type.isByteArray(), "Byte Arrays not yet implemented!"); solUnimplementedAssert(_type.baseType()->storageBytes() <= 32, "Base type is not yet implemented."); string functionName = "array_push_zero_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array) -> slot, offset { let oldLen := (array) if iszero(lt(oldLen, )) { () } sstore(array, add(oldLen, 1)) slot, offset := (array, oldLen) })") ("functionName", functionName) ("panic", panicFunction()) ("fetchLength", arrayLengthFunction(_type)) ("indexAccess", storageArrayIndexAccessFunction(_type)) ("maxArrayLength", (u256(1) << 64).str()) .render(); }); } string YulUtilFunctions::partialClearStorageSlotFunction() { string functionName = "partial_clear_storage_slot"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (slot, offset) { let mask := (mul(8, sub(32, offset)), ) sstore(slot, and(mask, sload(slot))) } )") ("functionName", functionName) ("ones", formatNumber((bigint(1) << 256) - 1)) ("shr", shiftRightFunctionDynamic()) .render(); }); } string YulUtilFunctions::clearStorageRangeFunction(Type const& _type) { if (_type.storageBytes() < 32) solAssert(_type.isValueType(), ""); string functionName = "clear_storage_range_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (start, end) { for {} lt(start, end) { start := add(start, ) } { (start, 0) } } )") ("functionName", functionName) ("setToZero", storageSetToZeroFunction(_type.storageBytes() < 32 ? *TypeProvider::uint256() : _type)) ("increment", _type.storageSize().str()) .render(); }); } string YulUtilFunctions::clearStorageArrayFunction(ArrayType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); if (_type.baseType()->storageBytes() < 32) { solAssert(_type.baseType()->isValueType(), "Invalid storage size for non-value type."); solAssert(_type.baseType()->storageSize() <= 1, "Invalid storage size for type."); } if (_type.baseType()->isValueType()) solAssert(_type.baseType()->storageSize() <= 1, "Invalid size for value type."); string functionName = "clear_storage_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (slot) { (slot, 0) (slot, add(slot, ())) } )") ("functionName", functionName) ("dynamic", _type.isDynamicallySized()) ("resizeArray", _type.isDynamicallySized() ? resizeDynamicArrayFunction(_type) : "") ( "clearRange", clearStorageRangeFunction( (_type.baseType()->storageBytes() < 32) ? *TypeProvider::uint256() : *_type.baseType() ) ) ("lenToSize", arrayConvertLengthToSize(_type)) ("len", _type.length().str()) .render(); }); } string YulUtilFunctions::clearStorageStructFunction(StructType const& _type) { solAssert(_type.location() == DataLocation::Storage, ""); string functionName = "clear_struct_storage_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { MemberList::MemberMap structMembers = _type.nativeMembers(nullptr); vector> memberSetValues; set slotsCleared; for (auto const& member: structMembers) if (member.type->storageBytes() < 32) { auto const& slotDiff = _type.storageOffsetsOfMember(member.name).first; if (!slotsCleared.count(slotDiff)) { memberSetValues.emplace_back().emplace("clearMember", "sstore(add(slot, " + slotDiff.str() + "), 0)"); slotsCleared.emplace(slotDiff); } } else { auto const& [memberSlotDiff, memberStorageOffset] = _type.storageOffsetsOfMember(member.name); solAssert(memberStorageOffset == 0, ""); memberSetValues.emplace_back().emplace("clearMember", Whiskers(R"( (add(slot, ), ) )") ("setZero", storageSetToZeroFunction(*member.type)) ("memberSlotDiff", memberSlotDiff.str()) ("memberStorageOffset", to_string(memberStorageOffset)) .render() ); } return Whiskers(R"( function (slot) { <#member> } )") ("functionName", functionName) ("allocStruct", allocateMemoryStructFunction(_type)) ("storageSize", _type.storageSize().str()) ("member", memberSetValues) .render(); }); } string YulUtilFunctions::copyArrayToStorageFunction(ArrayType const& _fromType, ArrayType const& _toType) { solAssert( *_fromType.copyForLocation(_toType.location(), _toType.isPointer()) == dynamic_cast(_toType), "" ); if (_fromType.isByteArray()) return copyByteArrayToStorageFunction(_fromType, _toType); if (_fromType.dataStoredIn(DataLocation::Storage) && _toType.baseType()->isValueType()) return copyValueArrayStorageToStorageFunction(_fromType, _toType); string functionName = "copy_array_to_storage_from_" + _fromType.identifier() + "_to_" + _toType.identifier(); return m_functionCollector.createFunction(functionName, [&](){ Whiskers templ(R"( function (slot, value, len) { if eq(slot, value) { leave } let length := (value, len) (slot, length) let srcPtr := (value) let elementSlot := (slot) let elementOffset := 0 for { let i := 0 } lt(i, length) {i := add(i, 1)} { let := (value, srcPtr) srcPtr := () let := (srcPtr) let := srcPtr (elementSlot, elementOffset, ) srcPtr := add(srcPtr, ) elementOffset := add(elementOffset, ) if gt(elementOffset, sub(32, )) { elementOffset := 0 elementSlot := add(elementSlot, 1) } elementSlot := add(elementSlot, ) } } )"); if (_fromType.dataStoredIn(DataLocation::Storage)) solAssert(!_fromType.isValueType(), ""); templ("functionName", functionName); bool fromCalldata = _fromType.dataStoredIn(DataLocation::CallData); templ("isFromDynamicCalldata", _fromType.isDynamicallySized() && fromCalldata); templ("fromStorage", _fromType.dataStoredIn(DataLocation::Storage)); bool fromMemory = _fromType.dataStoredIn(DataLocation::Memory); templ("fromMemory", fromMemory); templ("fromCalldata", fromCalldata); templ("isToDynamic", _toType.isDynamicallySized()); templ("srcDataLocation", arrayDataAreaFunction(_fromType)); if (fromCalldata) { templ("dynamicallySizedBase", _fromType.baseType()->isDynamicallySized()); templ("dynamicallyEncodedBase", _fromType.baseType()->isDynamicallyEncoded()); if (_fromType.baseType()->isDynamicallyEncoded()) templ("accessCalldataTail", accessCalldataTailFunction(*_fromType.baseType())); } if (_toType.isDynamicallySized()) templ("resizeArray", resizeDynamicArrayFunction(_toType)); templ("arrayLength",arrayLengthFunction(_fromType)); templ("isValueType", _fromType.baseType()->isValueType()); templ("dstDataLocation", arrayDataAreaFunction(_toType)); if (fromMemory || (fromCalldata && _fromType.baseType()->isValueType())) templ("readFromCalldataOrMemory", readFromMemoryOrCalldata(*_fromType.baseType(), fromCalldata)); templ("elementValues", suffixedVariableNameList( "elementValue_", 0, _fromType.baseType()->stackItems().size() )); templ("updateStorageValue", updateStorageValueFunction(*_fromType.baseType(), *_toType.baseType())); templ("srcStride", fromCalldata ? to_string(_fromType.calldataStride()) : fromMemory ? to_string(_fromType.memoryStride()) : formatNumber(_fromType.baseType()->storageSize()) ); templ("multipleItemsPerSlot", _toType.storageStride() <= 16); templ("storageStride", to_string(_toType.storageStride())); templ("storageSize", _toType.baseType()->storageSize().str()); return templ.render(); }); } string YulUtilFunctions::copyByteArrayToStorageFunction(ArrayType const& _fromType, ArrayType const& _toType) { solAssert( *_fromType.copyForLocation(_toType.location(), _toType.isPointer()) == dynamic_cast(_toType), "" ); solAssert(_fromType.isByteArray(), ""); solAssert(_toType.isByteArray(), ""); string functionName = "copy_byte_array_to_storage_from_" + _fromType.identifier() + "_to_" + _toType.identifier(); return m_functionCollector.createFunction(functionName, [&](){ Whiskers templ(R"( function (slot, src, len) { if eq(slot, src) { leave } let newLen := (src, len) // Make sure array length is sane if gt(newLen, 0xffffffffffffffff) { () } let oldLen := (sload(slot)) src := add(src, 0x20) // This is not needed in all branches. let dstDataArea if or(gt(oldLen, 31), gt(newLen, 31)) { dstDataArea := (slot) } if gt(oldLen, 31) { // potentially truncate data let deleteStart := add(dstDataArea, div(add(newLen, 31), 32)) if lt(newLen, 32) { deleteStart := dstDataArea } (deleteStart, add(dstDataArea, div(add(oldLen, 31), 32))) } switch gt(newLen, 31) case 1 { let loopEnd := and(newLen, not(0x1f)) let dstPtr := dstDataArea let i := 0 for { } lt(i, loopEnd) { i := add(i, 32) } { sstore(dstPtr, (add(src, i))) dstPtr := add(dstPtr, 1) } if lt(loopEnd, newLen) { let lastValue := (add(src, i)) sstore(dstPtr, (lastValue, and(newLen, 0x1f))) } sstore(slot, add(mul(newLen, 2), 1)) } default { let value := 0 if newLen { value := (src) } sstore(slot, (value, newLen)) } } )"); templ("functionName", functionName); bool fromStorage = _fromType.dataStoredIn(DataLocation::Storage); templ("fromStorage", fromStorage); bool fromCalldata = _fromType.dataStoredIn(DataLocation::CallData); templ("fromMemory", _fromType.dataStoredIn(DataLocation::Memory)); templ("fromCalldata", fromCalldata); templ("arrayLength", arrayLengthFunction(_fromType)); templ("panic", panicFunction()); templ("byteArrayLength", extractByteArrayLengthFunction()); templ("dstDataLocation", arrayDataAreaFunction(_toType)); templ("clearStorageRange", clearStorageRangeFunction(*_toType.baseType())); templ("read", fromStorage ? "sload" : fromCalldata ? "calldataload" : "mload"); templ("maskBytes", maskBytesFunctionDynamic()); templ("byteArrayCombineShort", shortByteArrayEncodeUsedAreaSetLengthFunction()); return templ.render(); }); } string YulUtilFunctions::copyValueArrayStorageToStorageFunction(ArrayType const& _fromType, ArrayType const& _toType) { solAssert( *_fromType.copyForLocation(_toType.location(), _toType.isPointer()) == dynamic_cast(_toType), "" ); solAssert(!_fromType.isByteArray(), ""); solAssert(_fromType.dataStoredIn(DataLocation::Storage) && _toType.baseType()->isValueType(), ""); solAssert(_toType.dataStoredIn(DataLocation::Storage), ""); string functionName = "copy_array_to_storage_from_" + _fromType.identifier() + "_to_" + _toType.identifier(); return m_functionCollector.createFunction(functionName, [&](){ Whiskers templ(R"( function (dst, src) { if eq(dst, src) { leave } let length := (src) // Make sure array length is sane if gt(length, 0xffffffffffffffff) { () } (dst, length) let srcPtr := (src) let dstPtr := (dst) let fullSlots := div(length, ) let i := 0 for { } lt(i, fullSlots) { i := add(i, 1) } { sstore(add(dstPtr, i), (sload(add(srcPtr, i)))) } let spill := sub(length, mul(i, )) if gt(spill, 0) { sstore(add(dstPtr, i), (sload(add(srcPtr, i)), mul(spill, ))) } } )"); if (_fromType.dataStoredIn(DataLocation::Storage)) solAssert(!_fromType.isValueType(), ""); templ("functionName", functionName); templ("isToDynamic", _toType.isDynamicallySized()); if (_toType.isDynamicallySized()) templ("resizeArray", resizeDynamicArrayFunction(_toType)); templ("arrayLength",arrayLengthFunction(_fromType)); templ("panic", panicFunction()); templ("srcDataLocation", arrayDataAreaFunction(_fromType)); templ("dstDataLocation", arrayDataAreaFunction(_toType)); unsigned itemsPerSlot = 32 / _toType.storageStride(); templ("itemsPerSlot", to_string(itemsPerSlot)); templ("bytesPerItem", to_string(_toType.storageStride())); templ("maskFull", maskLowerOrderBytesFunction(itemsPerSlot * _toType.storageStride())); templ("maskBytes", maskLowerOrderBytesFunctionDynamic()); return templ.render(); }); } string YulUtilFunctions::arrayConvertLengthToSize(ArrayType const& _type) { string functionName = "array_convert_length_to_size_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Type const& baseType = *_type.baseType(); switch (_type.location()) { case DataLocation::Storage: { unsigned const baseStorageBytes = baseType.storageBytes(); solAssert(baseStorageBytes > 0, ""); solAssert(32 / baseStorageBytes > 0, ""); return Whiskers(R"( function (length) -> size { size := length size := (, length) // Number of slots rounded up size := div(add(length, sub(, 1)), ) })") ("functionName", functionName) ("multiSlot", baseType.storageSize() > 1) ("itemsPerSlot", to_string(32 / baseStorageBytes)) ("storageSize", baseType.storageSize().str()) ("mul", overflowCheckedIntMulFunction(*TypeProvider::uint256())) .render(); } case DataLocation::CallData: // fallthrough case DataLocation::Memory: return Whiskers(R"( function (length) -> size { size := length size := (length, ) })") ("functionName", functionName) ("stride", to_string(_type.location() == DataLocation::Memory ? _type.memoryStride() : _type.calldataStride())) ("byteArray", _type.isByteArray()) ("mul", overflowCheckedIntMulFunction(*TypeProvider::uint256())) .render(); default: solAssert(false, ""); } }); } string YulUtilFunctions::arrayAllocationSizeFunction(ArrayType const& _type) { solAssert(_type.dataStoredIn(DataLocation::Memory), ""); string functionName = "array_allocation_size_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers w(R"( function (length) -> size { // Make sure we can allocate memory without overflow if gt(length, 0xffffffffffffffff) { () } // round up size := and(add(length, 0x1f), not(0x1f)) size := mul(length, 0x20) // add length slot size := add(size, 0x20) } )"); w("functionName", functionName); w("panic", panicFunction()); w("byteArray", _type.isByteArray()); w("dynamic", _type.isDynamicallySized()); return w.render(); }); } string YulUtilFunctions::arrayDataAreaFunction(ArrayType const& _type) { string functionName = "array_dataslot_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { // No special processing for calldata arrays, because they are stored as // offset of the data area and length on the stack, so the offset already // points to the data area. // This might change, if calldata arrays are stored in a single // stack slot at some point. return Whiskers(R"( function (ptr) -> data { data := ptr data := add(ptr, 0x20) mstore(0, ptr) data := keccak256(0, 0x20) } )") ("functionName", functionName) ("dynamic", _type.isDynamicallySized()) ("memory", _type.location() == DataLocation::Memory) ("storage", _type.location() == DataLocation::Storage) .render(); }); } string YulUtilFunctions::storageArrayIndexAccessFunction(ArrayType const& _type) { string functionName = "storage_array_index_access_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (array, index) -> slot, offset { let arrayLength := (array) if iszero(lt(index, arrayLength)) { () } offset := sub(31, mod(index, 0x20)) switch lt(arrayLength, 0x20) case 0 { let dataArea := (array) slot := add(dataArea, div(index, 0x20)) } default { slot := array } let dataArea := (array) slot := add(dataArea, div(index, )) offset := mul(mod(index, ), ) let dataArea := (array) slot := add(dataArea, mul(index, )) offset := 0 } )") ("functionName", functionName) ("panic", panicFunction()) ("arrayLen", arrayLengthFunction(_type)) ("dataAreaFunc", arrayDataAreaFunction(_type)) ("multipleItemsPerSlot", _type.baseType()->storageBytes() <= 16) ("isBytesArray", _type.isByteArray()) ("storageSize", _type.baseType()->storageSize().str()) ("storageBytes", toString(_type.baseType()->storageBytes())) ("itemsPerSlot", to_string(32 / _type.baseType()->storageBytes())) .render(); }); } string YulUtilFunctions::memoryArrayIndexAccessFunction(ArrayType const& _type) { string functionName = "memory_array_index_access_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (baseRef, index) -> addr { if iszero(lt(index, (baseRef))) { () } let offset := mul(index, ) offset := add(offset, 32) addr := add(baseRef, offset) } )") ("functionName", functionName) ("panic", panicFunction()) ("arrayLen", arrayLengthFunction(_type)) ("stride", to_string(_type.memoryStride())) ("dynamicallySized", _type.isDynamicallySized()) .render(); }); } string YulUtilFunctions::calldataArrayIndexAccessFunction(ArrayType const& _type) { solAssert(_type.dataStoredIn(DataLocation::CallData), ""); string functionName = "calldata_array_index_access_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (base_ref, length, index) -> addr, len { if iszero(lt(index, length)) { () } addr := add(base_ref, mul(index, )) addr, len := (base_ref, addr) } )") ("functionName", functionName) ("panic", panicFunction()) ("stride", to_string(_type.calldataStride())) ("dynamicallySized", _type.isDynamicallySized()) ("dynamicallyEncodedBase", _type.baseType()->isDynamicallyEncoded()) ("dynamicallySizedBase", _type.baseType()->isDynamicallySized()) ("arrayLen", toCompactHexWithPrefix(_type.length())) ("accessCalldataTail", _type.baseType()->isDynamicallyEncoded() ? accessCalldataTailFunction(*_type.baseType()): "") .render(); }); } string YulUtilFunctions::calldataArrayIndexRangeAccess(ArrayType const& _type) { solAssert(_type.dataStoredIn(DataLocation::CallData), ""); solAssert(_type.isDynamicallySized(), ""); string functionName = "calldata_array_index_range_access_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (offset, length, startIndex, endIndex) -> offsetOut, lengthOut { if gt(startIndex, endIndex) { } if gt(endIndex, length) { } offsetOut := add(offset, mul(startIndex, )) lengthOut := sub(endIndex, startIndex) } )") ("functionName", functionName) ("stride", to_string(_type.calldataStride())) ("revertSliceStartAfterEnd", revertReasonIfDebug("Slice starts after end")) ("revertSliceGreaterThanLength", revertReasonIfDebug("Slice is greater than length")) .render(); }); } string YulUtilFunctions::accessCalldataTailFunction(Type const& _type) { solAssert(_type.isDynamicallyEncoded(), ""); solAssert(_type.dataStoredIn(DataLocation::CallData), ""); string functionName = "access_calldata_tail_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (base_ref, ptr_to_tail) -> addr, length { let rel_offset_of_tail := calldataload(ptr_to_tail) if iszero(slt(rel_offset_of_tail, sub(sub(calldatasize(), base_ref), sub(, 1)))) { } addr := add(base_ref, rel_offset_of_tail) length := calldataload(addr) if gt(length, 0xffffffffffffffff) { } addr := add(addr, 32) if sgt(addr, sub(calldatasize(), mul(length, ))) { } } )") ("functionName", functionName) ("dynamicallySized", _type.isDynamicallySized()) ("neededLength", toCompactHexWithPrefix(_type.calldataEncodedTailSize())) ("calldataStride", toCompactHexWithPrefix(_type.isDynamicallySized() ? dynamic_cast(_type).calldataStride() : 0)) ("invalidCalldataTailOffset", revertReasonIfDebug("Invalid calldata tail offset")) ("invalidCalldataTailLength", revertReasonIfDebug("Invalid calldata tail length")) ("shortCalldataTail", revertReasonIfDebug("Calldata tail too short")) .render(); }); } string YulUtilFunctions::nextArrayElementFunction(ArrayType const& _type) { solAssert(!_type.isByteArray(), ""); if (_type.dataStoredIn(DataLocation::Storage)) solAssert(_type.baseType()->storageBytes() > 16, ""); string functionName = "array_nextElement_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (ptr) -> next { next := add(ptr, ) } )"); templ("functionName", functionName); switch (_type.location()) { case DataLocation::Memory: templ("advance", "0x20"); break; case DataLocation::Storage: { u256 size = _type.baseType()->storageSize(); solAssert(size >= 1, ""); templ("advance", toCompactHexWithPrefix(size)); break; } case DataLocation::CallData: { u256 size = _type.calldataStride(); solAssert(size >= 32 && size % 32 == 0, ""); templ("advance", toCompactHexWithPrefix(size)); break; } } return templ.render(); }); } string YulUtilFunctions::copyArrayFromStorageToMemoryFunction(ArrayType const& _from, ArrayType const& _to) { solAssert(_from.dataStoredIn(DataLocation::Storage), ""); solAssert(_to.dataStoredIn(DataLocation::Memory), ""); solAssert(_from.isDynamicallySized() == _to.isDynamicallySized(), ""); if (!_from.isDynamicallySized()) solAssert(_from.length() == _to.length(), ""); string functionName = "copy_array_from_storage_to_memory_" + _from.identifier(); return m_functionCollector.createFunction(functionName, [&]() { if (_from.baseType()->isValueType()) { solAssert(_from.baseType() == _to.baseType(), ""); ABIFunctions abi(m_evmVersion, m_revertStrings, m_functionCollector); return Whiskers(R"( function (slot) -> memptr { memptr := () let end := (slot, memptr) mstore(, end) } )") ("functionName", functionName) ("allocateTemp", allocationTemporaryMemoryFunction()) ( "encode", abi.abiEncodeAndReturnUpdatedPosFunction(_from, _to, ABIFunctions::EncodingOptions{}) ) ("freeMemoryPointer", to_string(CompilerUtils::freeMemoryPointer)) .render(); } else { solAssert(_to.memoryStride() == 32, ""); solAssert(_to.baseType()->dataStoredIn(DataLocation::Memory), ""); solAssert(_from.baseType()->dataStoredIn(DataLocation::Storage), ""); solAssert(!_from.isByteArray(), ""); solAssert(*_to.withLocation(DataLocation::Storage, _from.isPointer()) == _from, ""); return Whiskers(R"( function (slot) -> memptr { let length := (slot) memptr := (length) let mpos := memptr mpos := add(mpos, 0x20) let spos := (slot) for { let i := 0 } lt(i, length) { i := add(i, 1) } { mstore(mpos, (spos)) mpos := add(mpos, 0x20) spos := add(spos, ) } } )") ("functionName", functionName) ("lengthFunction", arrayLengthFunction(_from)) ("allocateArray", allocateMemoryArrayFunction(_to)) ("arrayDataArea", arrayDataAreaFunction(_from)) ("dynamic", _to.isDynamicallySized()) ("convert", conversionFunction(*_from.baseType(), *_to.baseType())) ("baseStorageSize", _from.baseType()->storageSize().str()) .render(); } }); } string YulUtilFunctions::mappingIndexAccessFunction(MappingType const& _mappingType, Type const& _keyType) { solAssert(_keyType.sizeOnStack() <= 1, ""); string functionName = "mapping_index_access_" + _mappingType.identifier() + "_of_" + _keyType.identifier(); return m_functionCollector.createFunction(functionName, [&]() { if (_mappingType.keyType()->isDynamicallySized()) return Whiskers(R"( function (slot , ) -> dataSlot { dataSlot := ( , slot) } )") ("functionName", functionName) ("key", _keyType.sizeOnStack() > 0 ? "key" : "") ("hash", packedHashFunction( {&_keyType, TypeProvider::uint256()}, {_mappingType.keyType(), TypeProvider::uint256()} )) .render(); else { solAssert(CompilerUtils::freeMemoryPointer >= 0x40, ""); solAssert(!_mappingType.keyType()->isDynamicallyEncoded(), ""); solAssert(_mappingType.keyType()->calldataEncodedSize(false) <= 0x20, ""); Whiskers templ(R"( function (slot ) -> dataSlot { mstore(0, ) mstore(0x20, slot) dataSlot := keccak256(0, 0x40) } )"); templ("functionName", functionName); templ("key", _keyType.sizeOnStack() == 1 ? ", key" : ""); if (_keyType.sizeOnStack() == 0) templ("convertedKey", conversionFunction(_keyType, *_mappingType.keyType()) + "()"); else templ("convertedKey", conversionFunction(_keyType, *_mappingType.keyType()) + "(key)"); return templ.render(); } }); } string YulUtilFunctions::readFromStorage(Type const& _type, size_t _offset, bool _splitFunctionTypes) { if (_type.isValueType()) return readFromStorageValueType(_type, _offset, _splitFunctionTypes); else { solAssert(_offset == 0, ""); return readFromStorageReferenceType(_type); } } string YulUtilFunctions::readFromStorageDynamic(Type const& _type, bool _splitFunctionTypes) { solAssert(_type.isValueType(), ""); return readFromStorageValueType(_type, {}, _splitFunctionTypes); } string YulUtilFunctions::readFromStorageValueType(Type const& _type, optional _offset, bool _splitFunctionTypes) { solAssert(_type.isValueType(), ""); string functionName = "read_from_storage_" + string(_splitFunctionTypes ? "split_" : "") + ( _offset.has_value() ? "offset_" + to_string(*_offset) : "dynamic" ) + "_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { Whiskers templ(R"( function (slot, offset) -> addr, selectorvalue { let value := (sload(slot), offset) addr, selector := (value) } )"); templ("functionName", functionName); templ("dynamic", !_offset.has_value()); if (_offset.has_value()) templ("extract", extractFromStorageValue(_type, *_offset)); else templ("extract", extractFromStorageValueDynamic(_type)); auto const* funType = dynamic_cast(&_type); bool split = _splitFunctionTypes && funType && funType->kind() == FunctionType::Kind::External; templ("split", split); if (split) templ("splitFunction", splitExternalFunctionIdFunction()); return templ.render(); }); } string YulUtilFunctions::readFromStorageReferenceType(Type const& _type) { solUnimplementedAssert(_type.category() == Type::Category::Struct, ""); string functionName = "read_from_storage_reference_type_" + _type.identifier(); auto const& structType = dynamic_cast(_type); solAssert(structType.location() == DataLocation::Memory, ""); MemberList::MemberMap structMembers = structType.nativeMembers(nullptr); vector> memberSetValues(structMembers.size()); for (size_t i = 0; i < structMembers.size(); ++i) { auto const& [memberSlotDiff, memberStorageOffset] = structType.storageOffsetsOfMember(structMembers[i].name); memberSetValues[i]["setMember"] = Whiskers(R"( { let := (add(slot, ), ) (add(value, ), ) } )") ("memberValues", suffixedVariableNameList("memberValue_", 0, structMembers[i].type->stackItems().size())) ("memberMemoryOffset", structType.memoryOffsetOfMember(structMembers[i].name).str()) ("memberSlotDiff", memberSlotDiff.str()) ("memberStorageOffset", to_string(memberStorageOffset)) ("readFromStorage", structMembers[i].type->isValueType() ? readFromStorageDynamic(*structMembers[i].type, true) : readFromStorage(*structMembers[i].type, memberStorageOffset, true) ) ("writeToMemory", writeToMemoryFunction(*structMembers[i].type)) ("hasOffset", structMembers[i].type->isValueType()) .render(); } return m_functionCollector.createFunction(functionName, [&] { return Whiskers(R"( function (slot) -> value { value := () <#member> } )") ("functionName", functionName) ("allocStruct", allocateMemoryStructFunction(structType)) ("member", memberSetValues) .render(); }); } string YulUtilFunctions::readFromMemory(Type const& _type) { return readFromMemoryOrCalldata(_type, false); } string YulUtilFunctions::readFromCalldata(Type const& _type) { return readFromMemoryOrCalldata(_type, true); } string YulUtilFunctions::updateStorageValueFunction( Type const& _fromType, Type const& _toType, std::optional const& _offset ) { string const functionName = "update_storage_value_" + (_offset.has_value() ? ("offset_" + to_string(*_offset)) : "") + _fromType.identifier() + "_to_" + _toType.identifier(); return m_functionCollector.createFunction(functionName, [&] { if (_toType.isValueType()) { solAssert(_fromType.isImplicitlyConvertibleTo(_toType), ""); solAssert(_toType.storageBytes() <= 32, "Invalid storage bytes size."); solAssert(_toType.storageBytes() > 0, "Invalid storage bytes size."); return Whiskers(R"( function (slot, ) { let := () sstore(slot, (sload(slot), ())) } )") ("functionName", functionName) ("update", _offset.has_value() ? updateByteSliceFunction(_toType.storageBytes(), *_offset) : updateByteSliceFunctionDynamic(_toType.storageBytes()) ) ("offset", _offset.has_value() ? "" : "offset, ") ("convert", conversionFunction(_fromType, _toType)) ("fromValues", suffixedVariableNameList("value_", 0, _fromType.sizeOnStack())) ("toValues", suffixedVariableNameList("convertedValue_", 0, _toType.sizeOnStack())) ("prepare", prepareStoreFunction(_toType)) .render(); } auto const* toReferenceType = dynamic_cast(&_toType); auto const* fromReferenceType = dynamic_cast(&_fromType); solAssert(fromReferenceType && toReferenceType, ""); solAssert(*toReferenceType->copyForLocation( fromReferenceType->location(), fromReferenceType->isPointer() ).get() == *fromReferenceType, ""); solAssert(toReferenceType->category() == fromReferenceType->category(), ""); if (_toType.category() == Type::Category::Array) { solAssert(_offset.value_or(0) == 0, ""); Whiskers templ(R"( function (slot, offset, ) { if offset { () } (slot, ) } )"); templ("functionName", functionName); templ("dynamicOffset", !_offset.has_value()); templ("panic", panicFunction()); templ("value", suffixedVariableNameList("value_", 0, _fromType.sizeOnStack())); templ("copyArrayToStorage", copyArrayToStorageFunction( dynamic_cast(_fromType), dynamic_cast(_toType) )); return templ.render(); } else { solAssert(_toType.category() == Type::Category::Struct, ""); auto const& fromStructType = dynamic_cast(_fromType); auto const& toStructType = dynamic_cast(_toType); solAssert(fromStructType.structDefinition() == toStructType.structDefinition(), ""); solAssert(_offset.value_or(0) == 0, ""); Whiskers templ(R"( function (slot, offset, value) { if offset { () } if eq(slot, value) { leave } <#member> { } } )"); templ("functionName", functionName); templ("dynamicOffset", !_offset.has_value()); templ("panic", panicFunction()); templ("fromStorage", fromStructType.dataStoredIn(DataLocation::Storage)); MemberList::MemberMap structMembers = fromStructType.nativeMembers(nullptr); MemberList::MemberMap toStructMembers = toStructType.nativeMembers(nullptr); vector> memberParams(structMembers.size()); for (size_t i = 0; i < structMembers.size(); ++i) { Type const& memberType = *structMembers[i].type; solAssert(memberType.memoryHeadSize() == 32, ""); auto const& [slotDiff, offset] = toStructType.storageOffsetsOfMember(structMembers[i].name); Whiskers t(R"( let memberSlot := add(slot, ) let memberSrcPtr := add(value, ) let := (value, memberSrcPtr) memberSrcPtr := () let := (memberSrcPtr) let := (memberSrcPtr) memberSrcPtr (memberSlot, ) )"); bool fromCalldata = fromStructType.location() == DataLocation::CallData; t("fromCalldata", fromCalldata); bool fromMemory = fromStructType.location() == DataLocation::Memory; t("fromMemory", fromMemory); bool fromStorage = fromStructType.location() == DataLocation::Storage; t("fromStorage", fromStorage); t("isValueType", memberType.isValueType()); t("memberValues", suffixedVariableNameList("memberValue_", 0, memberType.stackItems().size())); t("memberStorageSlotDiff", slotDiff.str()); if (fromCalldata) { t("memberOffset", to_string(fromStructType.calldataOffsetOfMember(structMembers[i].name))); t("dynamicallyEncodedMember", memberType.isDynamicallyEncoded()); if (memberType.isDynamicallyEncoded()) t("accessCalldataTail", accessCalldataTailFunction(memberType)); if (memberType.isValueType()) t("read", readFromCalldata(memberType)); } else if (fromMemory) { t("memberOffset", fromStructType.memoryOffsetOfMember(structMembers[i].name).str()); t("read", readFromMemory(memberType)); } else if (fromStorage) { auto [srcSlotOffset, srcOffset] = fromStructType.storageOffsetsOfMember(structMembers[i].name); t("memberOffset", formatNumber(srcSlotOffset)); if (memberType.isValueType()) t("read", readFromStorageValueType(memberType, srcOffset, false)); else solAssert(srcOffset == 0, ""); } t("memberStorageSlotOffset", to_string(offset)); t("updateStorageValue", updateStorageValueFunction( memberType, *toStructMembers[i].type, optional{offset} )); memberParams[i]["updateMemberCall"] = t.render(); } templ("member", memberParams); return templ.render(); } }); } string YulUtilFunctions::writeToMemoryFunction(Type const& _type) { string const functionName = "write_to_memory_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { solAssert(!dynamic_cast(&_type), ""); if (auto ref = dynamic_cast(&_type)) { solAssert( ref->location() == DataLocation::Memory, "Can only update types with location memory." ); return Whiskers(R"( function (memPtr, value) { mstore(memPtr, value) } )") ("functionName", functionName) .render(); } else if ( _type.category() == Type::Category::Function && dynamic_cast(_type).kind() == FunctionType::Kind::External ) { return Whiskers(R"( function (memPtr, addr, selector) { mstore(memPtr, (addr, selector)) } )") ("functionName", functionName) ("combine", combineExternalFunctionIdFunction()) .render(); } else if (_type.isValueType()) { return Whiskers(R"( function (memPtr, value) { mstore(memPtr, (value)) } )") ("functionName", functionName) ("cleanup", cleanupFunction(_type)) .render(); } else // Should never happen { solAssert( false, "Memory store of type " + _type.toString(true) + " not allowed." ); } }); } string YulUtilFunctions::extractFromStorageValueDynamic(Type const& _type) { string functionName = "extract_from_storage_value_dynamic" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { return Whiskers(R"( function (slot_value, offset) -> value { value := ((mul(offset, 8), slot_value)) } )") ("functionName", functionName) ("shr", shiftRightFunctionDynamic()) ("cleanupStorage", cleanupFromStorageFunction(_type)) .render(); }); } string YulUtilFunctions::extractFromStorageValue(Type const& _type, size_t _offset) { string functionName = "extract_from_storage_value_offset_" + to_string(_offset) + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { return Whiskers(R"( function (slot_value) -> value { value := ((slot_value)) } )") ("functionName", functionName) ("shr", shiftRightFunction(_offset * 8)) ("cleanupStorage", cleanupFromStorageFunction(_type)) .render(); }); } string YulUtilFunctions::cleanupFromStorageFunction(Type const& _type) { solAssert(_type.isValueType(), ""); string functionName = string("cleanup_from_storage_") + _type.identifier(); return m_functionCollector.createFunction(functionName, [&] { Whiskers templ(R"( function (value) -> cleaned { cleaned := } )"); templ("functionName", functionName); unsigned storageBytes = _type.storageBytes(); if (IntegerType const* type = dynamic_cast(&_type)) if (type->isSigned() && storageBytes != 32) { templ("cleaned", "signextend(" + to_string(storageBytes - 1) + ", value)"); return templ.render(); } bool leftAligned = false; if ( _type.category() != Type::Category::Function || dynamic_cast(_type).kind() == FunctionType::Kind::External ) leftAligned = _type.leftAligned(); if (storageBytes == 32) templ("cleaned", "value"); else if (leftAligned) templ("cleaned", shiftLeftFunction(256 - 8 * storageBytes) + "(value)"); else templ("cleaned", "and(value, " + toCompactHexWithPrefix((u256(1) << (8 * storageBytes)) - 1) + ")"); return templ.render(); }); } string YulUtilFunctions::prepareStoreFunction(Type const& _type) { string functionName = "prepare_store_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { solAssert(_type.isValueType(), ""); auto const* funType = dynamic_cast(&_type); if (funType && funType->kind() == FunctionType::Kind::External) { Whiskers templ(R"( function (addr, selector) -> ret { ret := ((addr, selector)) } )"); templ("functionName", functionName); templ("prepareBytes", prepareStoreFunction(*TypeProvider::fixedBytes(24))); templ("combine", combineExternalFunctionIdFunction()); return templ.render(); } else { solAssert(_type.sizeOnStack() == 1, ""); Whiskers templ(R"( function (value) -> ret { ret := } )"); templ("functionName", functionName); if (_type.category() == Type::Category::FixedBytes) templ("actualPrepare", shiftRightFunction(256 - 8 * _type.storageBytes()) + "(value)"); else templ("actualPrepare", "value"); return templ.render(); } }); } string YulUtilFunctions::allocationFunction() { string functionName = "allocateMemory"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (size) -> memPtr { memPtr := mload() let newFreePtr := add(memPtr, size) // protect against overflow if or(gt(newFreePtr, 0xffffffffffffffff), lt(newFreePtr, memPtr)) { () } mstore(, newFreePtr) } )") ("functionName", functionName) ("freeMemoryPointer", to_string(CompilerUtils::freeMemoryPointer)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::allocationTemporaryMemoryFunction() { string functionName = "allocateTemporaryMemory"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function () -> memPtr { memPtr := mload() } )") ("freeMemoryPointer", to_string(CompilerUtils::freeMemoryPointer)) ("functionName", functionName) .render(); }); } string YulUtilFunctions::releaseTemporaryMemoryFunction() { string functionName = "releaseTemporaryMemory"; return m_functionCollector.createFunction(functionName, [&](){ return Whiskers(R"( function () { } )") ("functionName", functionName) .render(); }); } string YulUtilFunctions::zeroMemoryArrayFunction(ArrayType const& _type) { if (_type.baseType()->hasSimpleZeroValueInMemory()) return zeroMemoryFunction(*_type.baseType()); return zeroComplexMemoryArrayFunction(_type); } string YulUtilFunctions::zeroMemoryFunction(Type const& _type) { solAssert(_type.hasSimpleZeroValueInMemory(), ""); string functionName = "zero_memory_chunk_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (dataStart, dataSizeInBytes) { calldatacopy(dataStart, calldatasize(), dataSizeInBytes) } )") ("functionName", functionName) .render(); }); } string YulUtilFunctions::zeroComplexMemoryArrayFunction(ArrayType const& _type) { solAssert(!_type.baseType()->hasSimpleZeroValueInMemory(), ""); string functionName = "zero_complex_memory_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { solAssert(_type.memoryStride() == 32, ""); return Whiskers(R"( function (dataStart, dataSizeInBytes) { for {let i := 0} lt(i, dataSizeInBytes) { i := add(i, ) } { mstore(add(dataStart, i), ()) } } )") ("functionName", functionName) ("stride", to_string(_type.memoryStride())) ("zeroValue", zeroValueFunction(*_type.baseType(), false)) .render(); }); } string YulUtilFunctions::allocateMemoryArrayFunction(ArrayType const& _type) { string functionName = "allocate_memory_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (length) -> memPtr { let allocSize := (length) memPtr := (allocSize) mstore(memPtr, length) } )") ("functionName", functionName) ("alloc", allocationFunction()) ("allocSize", arrayAllocationSizeFunction(_type)) ("dynamic", _type.isDynamicallySized()) .render(); }); } string YulUtilFunctions::allocateAndInitializeMemoryArrayFunction(ArrayType const& _type) { string functionName = "allocate_and_zero_memory_array_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (length) -> memPtr { memPtr := (length) let dataStart := memPtr let dataSize := (length) dataStart := add(dataStart, 32) dataSize := sub(dataSize, 32) (dataStart, dataSize) } )") ("functionName", functionName) ("allocArray", allocateMemoryArrayFunction(_type)) ("allocSize", arrayAllocationSizeFunction(_type)) ("zeroArrayFunction", zeroMemoryArrayFunction(_type)) ("dynamic", _type.isDynamicallySized()) .render(); }); } string YulUtilFunctions::allocateMemoryStructFunction(StructType const& _type) { string functionName = "allocate_memory_struct_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function () -> memPtr { memPtr := () } )"); templ("functionName", functionName); templ("alloc", allocationFunction()); templ("allocSize", _type.memoryDataSize().str()); return templ.render(); }); } string YulUtilFunctions::allocateAndInitializeMemoryStructFunction(StructType const& _type) { string functionName = "allocate_and_zero_memory_struct_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function () -> memPtr { memPtr := () let offset := memPtr <#member> mstore(offset, ()) offset := add(offset, 32) } )"); templ("functionName", functionName); templ("allocStruct", allocateMemoryStructFunction(_type)); TypePointers const& members = _type.memoryMemberTypes(); vector> memberParams(members.size()); for (size_t i = 0; i < members.size(); ++i) { solAssert(members[i]->memoryHeadSize() == 32, ""); memberParams[i]["zeroValue"] = zeroValueFunction( *TypeProvider::withLocationIfReference(DataLocation::Memory, members[i]), false ); } templ("member", memberParams); return templ.render(); }); } string YulUtilFunctions::conversionFunction(Type const& _from, Type const& _to) { if (_from.category() == Type::Category::Function) { solAssert(_to.category() == Type::Category::Function, ""); FunctionType const& fromType = dynamic_cast(_from); FunctionType const& targetType = dynamic_cast(_to); solAssert( fromType.isImplicitlyConvertibleTo(targetType) && fromType.sizeOnStack() == targetType.sizeOnStack() && (fromType.kind() == FunctionType::Kind::Internal || fromType.kind() == FunctionType::Kind::External) && fromType.kind() == targetType.kind(), "Invalid function type conversion requested." ); string const functionName = "convert_" + _from.identifier() + "_to_" + _to.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (addr, functionId) -> outAddr, outFunctionId { outAddr := addr outFunctionId := functionId } )") ("functionName", functionName) ("external", fromType.kind() == FunctionType::Kind::External) .render(); }); } else if (_from.category() == Type::Category::ArraySlice) { solAssert(_from.isDynamicallySized(), ""); solAssert(_from.dataStoredIn(DataLocation::CallData), ""); solAssert(_to.category() == Type::Category::Array, ""); ArraySliceType const& fromType = dynamic_cast(_from); ArrayType const& targetType = dynamic_cast(_to); solAssert(!fromType.arrayType().baseType()->isDynamicallyEncoded(), ""); solAssert( *fromType.arrayType().baseType() == *targetType.baseType(), "Converting arrays of different type is not possible" ); string const functionName = "convert_" + _from.identifier() + "_to_" + _to.identifier(); return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (offset, length) -> outOffset, outLength { outOffset := offset outLength := length } )") ("functionName", functionName) .render(); }); } else if (_from.category() == Type::Category::Array) { solAssert(_to.category() == Type::Category::Array, ""); return arrayConversionFunction( dynamic_cast(_from), dynamic_cast(_to) ); } if (_from.sizeOnStack() != 1 || _to.sizeOnStack() != 1) return conversionFunctionSpecial(_from, _to); string functionName = "convert_" + _from.identifier() + "_to_" + _to.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (value) -> converted { } )"); templ("functionName", functionName); string body; auto toCategory = _to.category(); auto fromCategory = _from.category(); switch (fromCategory) { case Type::Category::Address: body = Whiskers("converted := (value)") ("convert", conversionFunction(IntegerType(160), _to)) .render(); break; case Type::Category::Integer: case Type::Category::RationalNumber: case Type::Category::Contract: { if (RationalNumberType const* rational = dynamic_cast(&_from)) solUnimplementedAssert(!rational->isFractional(), "Not yet implemented - FixedPointType."); if (toCategory == Type::Category::FixedBytes) { solAssert( fromCategory == Type::Category::Integer || fromCategory == Type::Category::RationalNumber, "Invalid conversion to FixedBytesType requested." ); FixedBytesType const& toBytesType = dynamic_cast(_to); body = Whiskers("converted := ((value))") ("shiftLeft", shiftLeftFunction(256 - toBytesType.numBytes() * 8)) ("clean", cleanupFunction(_from)) .render(); } else if (toCategory == Type::Category::Enum) { solAssert(_from.mobileType(), ""); body = Whiskers("converted := ((value))") ("cleanEnum", cleanupFunction(_to)) // "mobileType()" returns integer type for rational ("cleanInt", cleanupFunction(*_from.mobileType())) .render(); } else if (toCategory == Type::Category::FixedPoint) solUnimplemented("Not yet implemented - FixedPointType."); else if (toCategory == Type::Category::Address) body = Whiskers("converted := (value)") ("convert", conversionFunction(_from, IntegerType(160))) .render(); else { solAssert( toCategory == Type::Category::Integer || toCategory == Type::Category::Contract, ""); IntegerType const addressType(160); IntegerType const& to = toCategory == Type::Category::Integer ? dynamic_cast(_to) : addressType; // Clean according to the "to" type, except if this is // a widening conversion. IntegerType const* cleanupType = &to; if (fromCategory != Type::Category::RationalNumber) { IntegerType const& from = fromCategory == Type::Category::Integer ? dynamic_cast(_from) : addressType; if (to.numBits() > from.numBits()) cleanupType = &from; } body = Whiskers("converted := (value)") ("cleanInt", cleanupFunction(*cleanupType)) .render(); } break; } case Type::Category::Bool: { solAssert(_from == _to, "Invalid conversion for bool."); body = Whiskers("converted := (value)") ("clean", cleanupFunction(_from)) .render(); break; } case Type::Category::FixedPoint: solUnimplemented("Fixed point types not implemented."); break; case Type::Category::Struct: { solAssert(toCategory == Type::Category::Struct, ""); auto const& fromStructType = dynamic_cast(_from); auto const& toStructType = dynamic_cast(_to); solAssert(fromStructType.structDefinition() == toStructType.structDefinition(), ""); if (fromStructType.location() == toStructType.location() && toStructType.isPointer()) body = "converted := value"; else { solUnimplementedAssert(toStructType.location() == DataLocation::Memory, ""); solUnimplementedAssert(fromStructType.location() != DataLocation::Memory, ""); if (fromStructType.location() == DataLocation::CallData) { solUnimplementedAssert(!fromStructType.isDynamicallyEncoded(), ""); body = Whiskers(R"( converted := (value, calldatasize()) )")("abiDecode", ABIFunctions(m_evmVersion, m_revertStrings, m_functionCollector).tupleDecoder( {&toStructType} )).render(); } else { solAssert(fromStructType.location() == DataLocation::Storage, ""); body = Whiskers(R"( converted := (value) )") ("readFromStorage", readFromStorage(toStructType, 0, true)) .render(); } } break; } case Type::Category::FixedBytes: { FixedBytesType const& from = dynamic_cast(_from); if (toCategory == Type::Category::Integer) body = Whiskers("converted := ((value))") ("shift", shiftRightFunction(256 - from.numBytes() * 8)) ("convert", conversionFunction(IntegerType(from.numBytes() * 8), _to)) .render(); else if (toCategory == Type::Category::Address) body = Whiskers("converted := (value)") ("convert", conversionFunction(_from, IntegerType(160))) .render(); else { // clear for conversion to longer bytes solAssert(toCategory == Type::Category::FixedBytes, "Invalid type conversion requested."); body = Whiskers("converted := (value)") ("clean", cleanupFunction(from)) .render(); } break; } case Type::Category::Function: { solAssert(false, "Conversion should not be called for function types."); break; } case Type::Category::Enum: { solAssert(toCategory == Type::Category::Integer || _from == _to, ""); EnumType const& enumType = dynamic_cast(_from); body = Whiskers("converted := (value)") ("clean", cleanupFunction(enumType)) .render(); break; } case Type::Category::Tuple: { solUnimplementedAssert(false, "Tuple conversion not implemented."); break; } case Type::Category::TypeType: { TypeType const& typeType = dynamic_cast(_from); if ( auto const* contractType = dynamic_cast(typeType.actualType()); contractType->contractDefinition().isLibrary() && _to == *TypeProvider::address() ) body = "converted := value"; else solAssert(false, "Invalid conversion from " + _from.canonicalName() + " to " + _to.canonicalName()); break; } case Type::Category::Mapping: { solAssert(_from == _to, ""); body = "converted := value"; break; } default: solAssert(false, "Invalid conversion from " + _from.canonicalName() + " to " + _to.canonicalName()); } solAssert(!body.empty(), _from.canonicalName() + " to " + _to.canonicalName()); templ("body", body); return templ.render(); }); } string YulUtilFunctions::arrayConversionFunction(ArrayType const& _from, ArrayType const& _to) { solAssert(_to.location() != DataLocation::CallData, ""); // Other cases are done explicitly in LValue::storeValue, and only possible by assignment. if (_to.location() == DataLocation::Storage) solAssert( (_to.isPointer() || (_from.isByteArray() && _to.isByteArray())) && _from.location() == DataLocation::Storage, "Invalid conversion to storage type." ); string functionName = "convert_array_" + _from.identifier() + "_to_" + _to.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (value, length) -> converted { } )"); templ("functionName", functionName); templ("fromCalldataDynamic", _from.dataStoredIn(DataLocation::CallData) && _from.isDynamicallySized()); if ( _from == _to || (_from.dataStoredIn(DataLocation::Memory) && _to.dataStoredIn(DataLocation::Memory)) || _to.dataStoredIn(DataLocation::Storage) ) templ("body", "converted := value"); else if (_to.dataStoredIn(DataLocation::Memory)) templ( "body", Whiskers(R"( // Copy the array to a free position in memory converted := (value) (value, , calldatasize()) )") ("fromStorage", _from.dataStoredIn(DataLocation::Storage)) ("fromCalldata", _from.dataStoredIn(DataLocation::CallData)) ("length", _from.isDynamicallySized() ? "length" : _from.length().str()) ( "abiDecode", _from.dataStoredIn(DataLocation::CallData) ? ABIFunctions( m_evmVersion, m_revertStrings, m_functionCollector ).abiDecodingFunctionArrayAvailableLength(_to, false) : "" ) ( "arrayStorageToMem", _from.dataStoredIn(DataLocation::Storage) ? copyArrayFromStorageToMemoryFunction(_from, _to) : "" ) .render() ); else solAssert(false, ""); return templ.render(); }); } string YulUtilFunctions::cleanupFunction(Type const& _type) { string functionName = string("cleanup_") + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (value) -> cleaned { } )"); templ("functionName", functionName); switch (_type.category()) { case Type::Category::Address: templ("body", "cleaned := " + cleanupFunction(IntegerType(160)) + "(value)"); break; case Type::Category::Integer: { IntegerType const& type = dynamic_cast(_type); if (type.numBits() == 256) templ("body", "cleaned := value"); else if (type.isSigned()) templ("body", "cleaned := signextend(" + to_string(type.numBits() / 8 - 1) + ", value)"); else templ("body", "cleaned := and(value, " + toCompactHexWithPrefix((u256(1) << type.numBits()) - 1) + ")"); break; } case Type::Category::RationalNumber: templ("body", "cleaned := value"); break; case Type::Category::Bool: templ("body", "cleaned := iszero(iszero(value))"); break; case Type::Category::FixedPoint: solUnimplemented("Fixed point types not implemented."); break; case Type::Category::Function: switch (dynamic_cast(_type).kind()) { case FunctionType::Kind::External: templ("body", "cleaned := " + cleanupFunction(FixedBytesType(24)) + "(value)"); break; case FunctionType::Kind::Internal: templ("body", "cleaned := value"); break; default: solAssert(false, ""); break; } break; case Type::Category::Array: case Type::Category::Struct: case Type::Category::Mapping: solAssert(_type.dataStoredIn(DataLocation::Storage), "Cleanup requested for non-storage reference type."); templ("body", "cleaned := value"); break; case Type::Category::FixedBytes: { FixedBytesType const& type = dynamic_cast(_type); if (type.numBytes() == 32) templ("body", "cleaned := value"); else if (type.numBytes() == 0) // This is disallowed in the type system. solAssert(false, ""); else { size_t numBits = type.numBytes() * 8; u256 mask = ((u256(1) << numBits) - 1) << (256 - numBits); templ("body", "cleaned := and(value, " + toCompactHexWithPrefix(mask) + ")"); } break; } case Type::Category::Contract: { AddressType addressType(dynamic_cast(_type).isPayable() ? StateMutability::Payable : StateMutability::NonPayable ); templ("body", "cleaned := " + cleanupFunction(addressType) + "(value)"); break; } case Type::Category::Enum: { // Out of range enums cannot be truncated unambigiously and therefore it should be an error. templ("body", "cleaned := value " + validatorFunction(_type, false) + "(value)"); break; } case Type::Category::InaccessibleDynamic: templ("body", "cleaned := 0"); break; default: solAssert(false, "Cleanup of type " + _type.identifier() + " requested."); } return templ.render(); }); } string YulUtilFunctions::validatorFunction(Type const& _type, bool _revertOnFailure) { string functionName = string("validator_") + (_revertOnFailure ? "revert_" : "assert_") + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function (value) { if iszero() { } } )"); templ("functionName", functionName); if (_revertOnFailure) templ("failure", "revert(0, 0)"); else templ("failure", panicFunction() + "()"); switch (_type.category()) { case Type::Category::Address: case Type::Category::Integer: case Type::Category::RationalNumber: case Type::Category::Bool: case Type::Category::FixedPoint: case Type::Category::Function: case Type::Category::Array: case Type::Category::Struct: case Type::Category::Mapping: case Type::Category::FixedBytes: case Type::Category::Contract: { templ("condition", "eq(value, " + cleanupFunction(_type) + "(value))"); break; } case Type::Category::Enum: { size_t members = dynamic_cast(_type).numberOfMembers(); solAssert(members > 0, "empty enum should have caused a parser error."); templ("condition", "lt(value, " + to_string(members) + ")"); break; } case Type::Category::InaccessibleDynamic: templ("condition", "1"); break; default: solAssert(false, "Validation of type " + _type.identifier() + " requested."); } return templ.render(); }); } string YulUtilFunctions::packedHashFunction( vector const& _givenTypes, vector const& _targetTypes ) { string functionName = string("packed_hashed_"); for (auto const& t: _givenTypes) functionName += t->identifier() + "_"; functionName += "_to_"; for (auto const& t: _targetTypes) functionName += t->identifier() + "_"; size_t sizeOnStack = 0; for (Type const* t: _givenTypes) sizeOnStack += t->sizeOnStack(); return m_functionCollector.createFunction(functionName, [&]() { Whiskers templ(R"( function () -> hash { let pos := mload() let end := (pos ) hash := keccak256(pos, sub(end, pos)) } )"); templ("functionName", functionName); templ("variables", suffixedVariableNameList("var_", 1, 1 + sizeOnStack)); templ("comma", sizeOnStack > 0 ? "," : ""); templ("freeMemoryPointer", to_string(CompilerUtils::freeMemoryPointer)); templ("packedEncode", ABIFunctions(m_evmVersion, m_revertStrings, m_functionCollector).tupleEncoderPacked(_givenTypes, _targetTypes)); return templ.render(); }); } string YulUtilFunctions::forwardingRevertFunction() { bool forward = m_evmVersion.supportsReturndata(); string functionName = "revert_forward_" + to_string(forward); return m_functionCollector.createFunction(functionName, [&]() { if (forward) return Whiskers(R"( function () { returndatacopy(0, 0, returndatasize()) revert(0, returndatasize()) } )") ("functionName", functionName) .render(); else return Whiskers(R"( function () { revert(0, 0) } )") ("functionName", functionName) .render(); }); } std::string YulUtilFunctions::decrementCheckedFunction(Type const& _type) { IntegerType const& type = dynamic_cast(_type); string const functionName = "decrement_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { u256 minintval; // Smallest admissible value to decrement if (type.isSigned()) minintval = 0 - (u256(1) << (type.numBits() - 1)) + 1; else minintval = 1; return Whiskers(R"( function (value) -> ret { value := (value) if (value, ) { () } ret := sub(value, 1) } )") ("functionName", functionName) ("panic", panicFunction()) ("minval", toCompactHexWithPrefix(minintval)) ("lt", type.isSigned() ? "slt" : "lt") ("cleanupFunction", cleanupFunction(_type)) .render(); }); } std::string YulUtilFunctions::incrementCheckedFunction(Type const& _type) { IntegerType const& type = dynamic_cast(_type); string const functionName = "increment_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { u256 maxintval; // Biggest admissible value to increment if (type.isSigned()) maxintval = (u256(1) << (type.numBits() - 1)) - 2; else maxintval = (u256(1) << type.numBits()) - 2; return Whiskers(R"( function (value) -> ret { value := (value) if (value, ) { () } ret := add(value, 1) } )") ("functionName", functionName) ("maxval", toCompactHexWithPrefix(maxintval)) ("gt", type.isSigned() ? "sgt" : "gt") ("panic", panicFunction()) ("cleanupFunction", cleanupFunction(_type)) .render(); }); } string YulUtilFunctions::negateNumberCheckedFunction(Type const& _type) { IntegerType const& type = dynamic_cast(_type); solAssert(type.isSigned(), "Expected signed type!"); string const functionName = "negate_" + _type.identifier(); u256 const minintval = 0 - (u256(1) << (type.numBits() - 1)) + 1; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function (value) -> ret { value := (value) if slt(value, ) { () } ret := sub(0, value) } )") ("functionName", functionName) ("minval", toCompactHexWithPrefix(minintval)) ("cleanupFunction", cleanupFunction(_type)) ("panic", panicFunction()) .render(); }); } string YulUtilFunctions::zeroValueFunction(Type const& _type, bool _splitFunctionTypes) { solAssert(_type.category() != Type::Category::Mapping, ""); string const functionName = "zero_value_for_" + string(_splitFunctionTypes ? "split_" : "") + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { FunctionType const* fType = dynamic_cast(&_type); if (fType && fType->kind() == FunctionType::Kind::External && _splitFunctionTypes) return Whiskers(R"( function () -> retAddress, retFunction { retAddress := 0 retFunction := 0 } )") ("functionName", functionName) .render(); if (_type.dataStoredIn(DataLocation::CallData)) { solAssert( _type.category() == Type::Category::Struct || _type.category() == Type::Category::Array, ""); Whiskers templ(R"( function () -> offset, length { offset := calldatasize() length := 0 } )"); templ("functionName", functionName); templ("hasLength", _type.category() == Type::Category::Array && dynamic_cast(_type).isDynamicallySized() ); return templ.render(); } Whiskers templ(R"( function () -> ret { ret := } )"); templ("functionName", functionName); if (_type.isValueType()) { solAssert(( _type.hasSimpleZeroValueInMemory() || (fType && (fType->kind() == FunctionType::Kind::Internal || fType->kind() == FunctionType::Kind::External)) ), ""); templ("zeroValue", "0"); } else { solAssert(_type.dataStoredIn(DataLocation::Memory), ""); if (auto const* arrayType = dynamic_cast(&_type)) { if (_type.isDynamicallySized()) templ("zeroValue", to_string(CompilerUtils::zeroPointer)); else templ("zeroValue", allocateAndInitializeMemoryArrayFunction(*arrayType) + "(" + to_string(unsigned(arrayType->length())) + ")"); } else if (auto const* structType = dynamic_cast(&_type)) templ("zeroValue", allocateAndInitializeMemoryStructFunction(*structType) + "()"); else solUnimplementedAssert(false, ""); } return templ.render(); }); } string YulUtilFunctions::storageSetToZeroFunction(Type const& _type) { string const functionName = "storage_set_to_zero_" + _type.identifier(); return m_functionCollector.createFunction(functionName, [&]() { if (_type.isValueType()) return Whiskers(R"( function (slot, offset) { (slot, offset, ()) } )") ("functionName", functionName) ("store", updateStorageValueFunction(_type, _type)) ("zeroValue", zeroValueFunction(_type)) .render(); else if (_type.category() == Type::Category::Array) return Whiskers(R"( function (slot, offset) { if iszero(eq(offset, 0)) { () } (slot) } )") ("functionName", functionName) ("clearArray", clearStorageArrayFunction(dynamic_cast(_type))) ("panic", panicFunction()) .render(); else if (_type.category() == Type::Category::Struct) return Whiskers(R"( function (slot, offset) { if iszero(eq(offset, 0)) { () } (slot) } )") ("functionName", functionName) ("clearStruct", clearStorageStructFunction(dynamic_cast(_type))) ("panic", panicFunction()) .render(); else solUnimplemented("setToZero for type " + _type.identifier() + " not yet implemented!"); }); } string YulUtilFunctions::conversionFunctionSpecial(Type const& _from, Type const& _to) { string functionName = "convert_" + _from.identifier() + "_to_" + _to.identifier(); return m_functionCollector.createFunction(functionName, [&]() { if ( auto fromTuple = dynamic_cast(&_from), toTuple = dynamic_cast(&_to); fromTuple && toTuple && fromTuple->components().size() == toTuple->components().size() ) { size_t sourceStackSize = 0; size_t destStackSize = 0; std::string conversions; for (size_t i = 0; i < fromTuple->components().size(); ++i) { auto fromComponent = fromTuple->components()[i]; auto toComponent = toTuple->components()[i]; solAssert(fromComponent, ""); if (toComponent) { conversions += suffixedVariableNameList("converted", destStackSize, destStackSize + toComponent->sizeOnStack()) + (toComponent->sizeOnStack() > 0 ? " := " : "") + conversionFunction(*fromComponent, *toComponent) + "(" + suffixedVariableNameList("value", sourceStackSize, sourceStackSize + fromComponent->sizeOnStack()) + ")\n"; destStackSize += toComponent->sizeOnStack(); } sourceStackSize += fromComponent->sizeOnStack(); } return Whiskers(R"( function () { } )") ("functionName", functionName) ("values", suffixedVariableNameList("value", 0, sourceStackSize)) ("arrow", destStackSize > 0 ? "->" : "") ("converted", suffixedVariableNameList("converted", 0, destStackSize)) ("conversions", conversions) .render(); } solUnimplementedAssert( _from.category() == Type::Category::StringLiteral, "Type conversion " + _from.toString() + " -> " + _to.toString() + " not yet implemented." ); string const& data = dynamic_cast(_from).value(); if (_to.category() == Type::Category::FixedBytes) { unsigned const numBytes = dynamic_cast(_to).numBytes(); solAssert(data.size() <= 32, ""); Whiskers templ(R"( function () -> converted { converted := } )"); templ("functionName", functionName); templ("data", formatNumber( h256::Arith(h256(data, h256::AlignLeft)) & (~(u256(-1) >> (8 * numBytes))) )); return templ.render(); } else if (_to.category() == Type::Category::Array) { auto const& arrayType = dynamic_cast(_to); solAssert(arrayType.isByteArray(), ""); size_t words = (data.size() + 31) / 32; size_t storageSize = 32 + words * 32; Whiskers templ(R"( function () -> converted { converted := () mstore(converted, ) <#word> mstore(add(converted, ), ) } )"); templ("functionName", functionName); templ("allocate", allocationFunction()); templ("storageSize", to_string(storageSize)); templ("size", to_string(data.size())); vector> wordParams(words); for (size_t i = 0; i < words; ++i) { wordParams[i]["offset"] = to_string(32 + i * 32); wordParams[i]["wordValue"] = formatAsStringOrNumber(data.substr(32 * i, 32)); } templ("word", wordParams); return templ.render(); } else solAssert( false, "Invalid conversion from string literal to " + _to.toString() + " requested." ); }); } string YulUtilFunctions::readFromMemoryOrCalldata(Type const& _type, bool _fromCalldata) { string functionName = string("read_from_") + (_fromCalldata ? "calldata" : "memory") + _type.identifier(); // TODO use ABI functions for handling calldata if (_fromCalldata) solAssert(!_type.isDynamicallyEncoded(), ""); return m_functionCollector.createFunction(functionName, [&] { if (auto refType = dynamic_cast(&_type)) { solAssert(refType->sizeOnStack() == 1, ""); solAssert(!_fromCalldata, ""); return Whiskers(R"( function (memPtr) -> value { value := mload(memPtr) } )") ("functionName", functionName) .render(); } solAssert(_type.isValueType(), ""); Whiskers templ(R"( function (ptr) -> { let value := calldataload(ptr) (value) let value := (mload(ptr)) := (value) value } )"); templ("functionName", functionName); templ("fromCalldata", _fromCalldata); if (_fromCalldata) templ("validate", validatorFunction(_type, true)); auto const* funType = dynamic_cast(&_type); if (funType && funType->kind() == FunctionType::Kind::External) { templ("externalFunction", true); templ("splitFunction", splitExternalFunctionIdFunction()); templ("returnVariables", "addr, selector"); } else { templ("externalFunction", false); templ("returnVariables", "returnValue"); } // Byte array elements generally need cleanup. // Other types are cleaned as well to account for dirty memory e.g. due to inline assembly. templ("cleanup", cleanupFunction(_type)); return templ.render(); }); } string YulUtilFunctions::revertReasonIfDebug(RevertStrings revertStrings, string const& _message) { if (revertStrings >= RevertStrings::Debug && !_message.empty()) { Whiskers templ(R"({ mstore(0, ) mstore(4, 0x20) mstore(add(4, 0x20), ) let reasonPos := add(4, 0x40) <#word> mstore(add(reasonPos, ), ) revert(0, add(reasonPos, )) })"); templ("sig", util::selectorFromSignature("Error(string)").str()); templ("length", to_string(_message.length())); size_t words = (_message.length() + 31) / 32; vector> wordParams(words); for (size_t i = 0; i < words; ++i) { wordParams[i]["offset"] = to_string(i * 32); wordParams[i]["wordValue"] = formatAsStringOrNumber(_message.substr(32 * i, 32)); } templ("word", wordParams); templ("end", to_string(words * 32)); return templ.render(); } else return "revert(0, 0)"; } string YulUtilFunctions::revertReasonIfDebug(string const& _message) { return revertReasonIfDebug(m_revertStrings, _message); } string YulUtilFunctions::panicFunction() { string functionName = "panic_error"; return m_functionCollector.createFunction(functionName, [&]() { return Whiskers(R"( function () { invalid() } )") ("functionName", functionName) .render(); }); } string YulUtilFunctions::tryDecodeErrorMessageFunction() { string const functionName = "try_decode_error_message"; return m_functionCollector.createFunction(functionName, [&]() { return util::Whiskers(R"( function () -> ret { if lt(returndatasize(), 0x44) { leave } returndatacopy(0, 0, 4) let sig := (mload(0)) if iszero(eq(sig, 0x)) { leave } let data := mload() returndatacopy(data, 4, sub(returndatasize(), 4)) let offset := mload(data) if or( gt(offset, 0xffffffffffffffff), gt(add(offset, 0x24), returndatasize()) ) { leave } let msg := add(data, offset) let length := mload(msg) if gt(length, 0xffffffffffffffff) { leave } let end := add(add(msg, 0x20), length) if gt(end, add(data, returndatasize())) { leave } mstore(, add(add(msg, 0x20), (length))) ret := msg } )") ("functionName", functionName) ("shr224", shiftRightFunction(224)) ("ErrorSignature", FixedHash<4>(util::keccak256("Error(string)")).hex()) ("freeMemoryPointer", to_string(CompilerUtils::freeMemoryPointer)) ("roundUp", roundUpFunction()) .render(); }); } string YulUtilFunctions::extractReturndataFunction() { string const functionName = "extract_returndata"; return m_functionCollector.createFunction(functionName, [&]() { return util::Whiskers(R"( function () -> data { switch returndatasize() case 0 { data := () } default { // allocate some memory into data of size returndatasize() + PADDING data := ((add(returndatasize(), 0x20))) // store array length into the front mstore(data, returndatasize()) // append to data returndatacopy(add(data, 0x20), 0, returndatasize()) } data := () } )") ("functionName", functionName) ("supportsReturndata", m_evmVersion.supportsReturndata()) ("allocate", allocationFunction()) ("roundUp", roundUpFunction()) ("emptyArray", zeroValueFunction(*TypeProvider::bytesMemory())) .render(); }); } string YulUtilFunctions::copyConstructorArgumentsToMemoryFunction( ContractDefinition const& _contract, string const& _creationObjectName ) { string functionName = "copy_arguments_for_constructor_" + toString(_contract.constructor()->id()) + "_object_" + _contract.name() + "_" + toString(_contract.id()); return m_functionCollector.createFunction(functionName, [&]() { string returnParams = suffixedVariableNameList("ret_param_",0, _contract.constructor()->parameters().size()); ABIFunctions abiFunctions(m_evmVersion, m_revertStrings, m_functionCollector); return util::Whiskers(R"( function () -> { let programSize := datasize("") let argSize := sub(codesize(), programSize) let memoryDataOffset := (argSize) codecopy(memoryDataOffset, programSize, argSize) := (memoryDataOffset, add(memoryDataOffset, argSize)) } )") ("functionName", functionName) ("retParams", returnParams) ("object", _creationObjectName) ("allocate", allocationFunction()) ("abiDecode", abiFunctions.tupleDecoder(FunctionType(*_contract.constructor()).parameterTypes(), true)) .render(); }); }