Function p2_random_limits

tests/all/cli_tests.rs:2301–2389

()

    #[test]
    fn p2_random_limits() -> Result<()> {
        println!("{P2_RANDOM_COMPONENT}");

        // By default, p2_random.rs fits within the random limits - always
        // asks for 256 bytes.
        let output = run_wasmtime(&["run", P2_RANDOM_COMPONENT])?;
        assert_eq!(output, "");

        // Lowering limit to 255 bytes should produce a trap in the first
        // call, which goes by way of the wasip1 adapter:
        let output = run_wasmtime(&["run", "-Smax-random-size=255", P2_RANDOM_COMPONENT])
            .err()
            .ok_or_else(|| format_err!("execution with max-random-size=255 should trap"))?;
        let output = format!("{output:?}");
        assert!(
            output.contains("lib_generated::random_get"),
            "expected error stack frames to contain 'wasip1::lib_generated::random_get'. Got:\n{output}"
        );
        assert!(
            output.contains("requested len 256 exceeds limit 255"),
            "expected error stack frames to contain 'requested len 256 exceeds limit 255'. Got:\n{output}"
        );

        // Lowering limit to 255 bytes and setting environment variable for
        // the first call to only request 255 bytes should be OK, and then the
        // program produce a trap in the second call, which calls the wasip2
        // random import directly:
        let output = run_wasmtime(&[
            "run",
            "-Smax-random-size=255",
            "--env",
            "TEST_P1_RANDOM_LEN=255",
            P2_RANDOM_COMPONENT,
        ])
        .err()
        .ok_or_else(|| format_err!("execution with max-random-size=255 should trap"))?;
        let output = format!("{output:?}");
        assert!(
            output.contains("random::random::get_random_bytes"),
            "expected error stack frames to contain 'wasi::random::random::get_random_bytes'. Got:\n{output}"
        );
        assert!(
            output.contains("requested len 256 exceeds limit 255"),
            "expected error stack frames to contain 'requested len 256 exceeds limit 255'. Got:\n{output}"
        );

        // Lowering limit to 255 bytes and setting environment variable for
        // the first and second calls to be at the limit should produce a trap
        // in the third call, which calls the wasip2 insecure random import:
        let output = run_wasmtime(&[
            "run",
            "-Smax-random-size=255",
            "--env",
            "TEST_P1_RANDOM_LEN=255",
            "--env",
            "TEST_P2_RANDOM_LEN=255",
            P2_RANDOM_COMPONENT,
        ])