path: root/test/liblll/EndToEndTest.cpp

/*
    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 <http://www.gnu.org/licenses/>.
*/
/**
 * @author Alex Beregszaszi
 * @date 2016
 * End to end tests for LLL.
 */

#include <string>
#include <memory>
#include <boost/test/unit_test.hpp>
#include <test/liblll/ExecutionFramework.h>

using namespace std;

namespace dev
{
namespace lll
{
namespace test
{

BOOST_FIXTURE_TEST_SUITE(LLLEndToEndTest, LLLExecutionFramework)

BOOST_AUTO_TEST_CASE(smoke_test)
{
        char const* sourceCode = "(returnlll { (return \"test\") })";
        compileAndRun(sourceCode);
        BOOST_CHECK(callFallback() == encodeArgs(string("test", 4)));
}

BOOST_AUTO_TEST_CASE(bare_panic)
{
    char const* sourceCode = "(panic)";
    compileAndRunWithoutCheck(sourceCode);
    BOOST_REQUIRE(m_output.empty());
}

BOOST_AUTO_TEST_CASE(panic)
{
    char const* sourceCode = "{ (panic) }";
    compileAndRunWithoutCheck(sourceCode);
    BOOST_REQUIRE(m_output.empty());
}

BOOST_AUTO_TEST_CASE(when)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (when (= (calldatasize) 0) (return 1))
                (when (!= (calldatasize) 0) (return 2))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
    BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
}

BOOST_AUTO_TEST_CASE(unless)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (unless (!= (calldatasize) 0) (return 1))
                (unless (= (calldatasize) 0) (return 2))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
    BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
}

BOOST_AUTO_TEST_CASE(conditional_literal)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (return (if (= (calldatasize) 0) 1 2))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
    BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
}

BOOST_AUTO_TEST_CASE(conditional)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (if (= (calldatasize) 0) (return 1) (return 2))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
    BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
}

BOOST_AUTO_TEST_CASE(conditional_seq)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (return (if (= (calldatasize) 0) { 0 2 1 } { 0 1 2 }))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
    BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
}

BOOST_AUTO_TEST_CASE(exp_operator_const)
{
    char const* sourceCode = R"(
        (returnlll
            (return (exp 2 3)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == toBigEndian(u256(8)));
}

BOOST_AUTO_TEST_CASE(exp_operator_const_signed)
{
    char const* sourceCode = R"(
        (returnlll
            (return (exp (- 0 2) 3)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == toBigEndian(u256(-8)));
}

BOOST_AUTO_TEST_CASE(exp_operator_on_range)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (when (= (div (calldataload 0x00) (exp 2 224)) 0xb3de648b)
                    (return (exp 2 (calldataload 0x04))))
                (jump 0x02)))
    )";
    compileAndRun(sourceCode);
    testContractAgainstCppOnRange("f(uint256)", [](u256 const& a) -> u256 { return u256(1 << a.convert_to<int>()); }, 0, 16);
}

BOOST_AUTO_TEST_CASE(constructor_argument_internal_numeric)
{
    char const* sourceCode = R"(
        (seq
            (sstore 0x00 65535)
            (returnlll
                (return @@0x00)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(65535)));
}

BOOST_AUTO_TEST_CASE(constructor_argument_internal_string)
{
    char const* sourceCode = R"(
        (seq
            (sstore 0x00 "test")
            (returnlll
                (return @@0x00)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs("test"));
}

BOOST_AUTO_TEST_CASE(constructor_arguments_external)
{
    char const* sourceCode = R"(
        (seq
            (codecopy 0x00 (bytecodesize) 64)
            (sstore 0x00 @0x00)
            (sstore 0x01 @0x20)
            (returnlll
                (seq
                    (when (= (div (calldataload 0x00) (exp 2 224)) 0xf2c9ecd8)
                        (return @@0x00))
                    (when (= (div (calldataload 0x00) (exp 2 224)) 0x89ea642f)
                        (return @@0x01)))))
    )";
    compileAndRun(sourceCode, 0, "", encodeArgs(u256(65535), "test"));
    BOOST_CHECK(callContractFunction("getNumber()") == encodeArgs(u256(65535)));
    BOOST_CHECK(callContractFunction("getString()") == encodeArgs("test"));
}

BOOST_AUTO_TEST_CASE(fallback_and_invalid_function)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (when (= (div (calldataload 0x00) (exp 2 224)) 0xab5ed150)
                    (return "one"))
                (when (= (div (calldataload 0x00) (exp 2 224)) 0xee784123)
                    (return "two"))
                (return "three")))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callContractFunction("getOne()") == encodeArgs("one"));
    BOOST_CHECK(callContractFunction("getTwo()") == encodeArgs("two"));
    BOOST_CHECK(callContractFunction("invalidFunction()") == encodeArgs("three"));
    BOOST_CHECK(callFallback() == encodeArgs("three"));
}

BOOST_AUTO_TEST_CASE(lit_string)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (lit 0x00 "abcdef")
                (return 0x00 0x20)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(string("abcdef")));
}

BOOST_AUTO_TEST_CASE(arithmetic)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (mstore8 0x00 (+ 160 22))
                (mstore8 0x01 (- 223 41))
                (mstore8 0x02 (* 33 2))
                (mstore8 0x03 (/ 10 2))
                (mstore8 0x04 (% 67 2))
                (mstore8 0x05 (& 15 8))
                (mstore8 0x06 (| 18 8))
                (mstore8 0x07 (^ 26 6))
                (return 0x00 0x20)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("b6b6420501081a1c000000000000000000000000000000000000000000000000")));
}

BOOST_AUTO_TEST_CASE(binary)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (mstore8 0x00 (< 53 87))
                (mstore8 0x01 (< 73 42))
                (mstore8 0x02 (<= 37 94))
                (mstore8 0x03 (<= 37 37))
                (mstore8 0x04 (<= 183 34))
                (mstore8 0x05 (S< (- 0 53) 87))
                (mstore8 0x06 (S< 73 (- 0 42)))
                (mstore8 0x07 (S<= (- 0 37) 94))
                (mstore8 0x08 (S<= (- 0 37) (- 0 37)))
                (mstore8 0x09 (S<= 183 (- 0 34)))
                (mstore8 0x0a (> 73 42))
                (mstore8 0x0b (> 53 87))
                (mstore8 0x0c (>= 94 37))
                (mstore8 0x0d (>= 94 94))
                (mstore8 0x0e (>= 34 183))
                (mstore8 0x0f (S> 73 (- 0 42)))
                (mstore8 0x10 (S> (- 0 53) 87))
                (mstore8 0x11 (S>= 94 (- 0 37)))
                (mstore8 0x12 (S>= (- 0 94) (- 0 94)))
                (mstore8 0x13 (S>= (- 0 34) 183))
                (mstore8 0x14 (= 53 53))
                (mstore8 0x15 (= 73 42))
                (mstore8 0x16 (!= 37 94))
                (mstore8 0x17 (!= 37 37))
                (return 0x00 0x20)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("0100010100010001010001000101000100010100010001000000000000000000")));
}

BOOST_AUTO_TEST_CASE(unary)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (mstore8 0x00 (! (< 53 87)))
                (mstore8 0x01 (! (>= 42 73)))
                (mstore8 0x02 (~ 0x7f))
                (mstore8 0x03 (~ 0xaa))
                (return 0x00 0x20)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("0001805500000000000000000000000000000000000000000000000000000000")));
}

BOOST_AUTO_TEST_CASE(assembly_mload_mstore)
{
    char const* sourceCode = R"(
        (returnlll
            (asm
                0x07 0x00 mstore
                "abcdef" 0x20 mstore
                0x00 mload 0x40 mstore
                0x20 mload 0x60 mstore
                0x40 0x40 return))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(7), string("abcdef")));
}

BOOST_AUTO_TEST_CASE(assembly_sload_sstore)
{
    char const* sourceCode = R"(
        (returnlll
            (asm
                0x07 0x00 sstore
                "abcdef" 0x01 sstore
                0x00 sload 0x00 mstore
                0x01 sload 0x20 mstore
                0x40 0x00 return))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(7), string("abcdef")));
}

BOOST_AUTO_TEST_CASE(assembly_codecopy)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (lit 0x00 "abcdef")
                (asm
                    0x06 6 codesize sub 0x20 codecopy
                    0x20 0x20 return)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(string("abcdef")));
}

BOOST_AUTO_TEST_CASE(zeroarg_macro)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (def 'zeroarg () (seq (mstore 0 0x1234) (return 0 32)))
                (zeroarg)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(0x1234)));
}

BOOST_AUTO_TEST_CASE(keccak256_32bytes)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (mstore 0x00 0x01)
                (return (keccak256 0x00 0x20))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("b10e2d527612073b26eecdfd717e6a320cf44b4afac2b0732d9fcbe2b7fa0cf6")));
}

BOOST_AUTO_TEST_CASE(sha3_two_args)
{
    char const* sourceCode = R"(
        (returnlll
            (seq
                (mstore 0x00 0x01)
                (return (sha3 0x00 0x20))))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("b10e2d527612073b26eecdfd717e6a320cf44b4afac2b0732d9fcbe2b7fa0cf6")));
}

BOOST_AUTO_TEST_CASE(sha3_one_arg)
{
    char const* sourceCode = R"(
        (returnlll
            (return (sha3 0x01)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(
        fromHex("b10e2d527612073b26eecdfd717e6a320cf44b4afac2b0732d9fcbe2b7fa0cf6")));
}

BOOST_AUTO_TEST_CASE(shift_left)
{
    char const* sourceCode = R"(
        (returnlll
            (return (shl 1 8)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(256)));
}

BOOST_AUTO_TEST_CASE(shift_right)
{
    char const* sourceCode = R"(
        (returnlll
            (return (shr 65536 8)))
    )";
    compileAndRun(sourceCode);
    BOOST_CHECK(callFallback() == encodeArgs(u256(256)));
}

BOOST_AUTO_TEST_SUITE_END()

}
}
} // end namespaces