Elixir实现JavaScript Math.imul结果不一致问题求助

我在开发Elixir项目时，需要实现JavaScript标准方法Math.imul的功能。参考Math.imul的shim实现完成Elixir版本后，发现返回结果和JavaScript的Math.imul差异极大，无法保证值一致性。比如调用imul(-5, 12)时，JavaScript返回-60，但Elixir返回51539607492。检查后发现过程中各中间值完全一致，仅在最终左移运算后的结果计算环节出现差异。我知道Elixir采用任意精度整数，无需担心32位限制，但位运算语义应该类似，实在困惑问题出在哪里。

问题代码对比 #

JavaScript实现：

const imul = (x, y) => {
    const a = ToUint32(x); // 4294967291
    const b = ToUint32(y); // 12
    const ah = (a >>> 16) & 0xffff; // 65535
    const al = a & 0xffff; // 65531
    const bh = (b >>> 16) & 0xffff; // 0
    const bl = b & 0xffff; // 12

    return (al * bl) + (((ah * bl) + (al * bh)) << 16); // -60
}

Elixir初始实现：

def imul(x, y) do
    a = toUInt32(x) # 4294967291
    b = toUInt32(y) # 12
    ah = (a >>> 16) &&& 0xffff # 65535
    al = a &&& 0xffff # 65531
    bh = (b >>> 16) &&& 0xffff # 0
    bl = b &&& 0xffff # 12

    (al * bl) + (((ah * bl) + (al * bh)) <<< 16) # 51539607492
end

def toUInt32(num) do
    <<num :: integer-unsigned-32>>
    |> :binary.decode_unsigned(:big)
end

问题根源 #

JavaScript的Math.imul返回的是32位有符号整数：即使计算过程中数值超出32位范围，最终会自动截断到32位，并按有符号整数规则解析（最高位为1时表示负数）。而Elixir的整数是任意精度的，不会自动截断，所以直接计算得到的是完整的大整数，没有做符号转换和位数限制。

解决方案 #

在Elixir中计算完原始结果后，需要手动将其转换为32位有符号整数。可以通过二进制编码再解码的方式实现这一逻辑：

def imul(x, y) do
    a = toUInt32(x)
    b = toUInt32(y)
    ah = (a >>> 16) &&& 0xffff
    al = a &&& 0xffff
    bh = (b >>> 16) &&& 0xffff
    bl = b &&& 0xffff

    raw_result = (al * bl) + (((ah * bl) + (al * bh)) <<< 16)
    # 将原始结果转换为32位有符号整数
    <<result :: integer-signed-32>> = <<raw_result :: integer-unsigned-32>>
    result
end

def toUInt32(num) do
    <<num :: integer-unsigned-32>>
    |> :binary.decode_unsigned(:big)
end

原理说明 #

1. 先计算出无符号的原始结果raw_result，这一步和JS的计算逻辑完全一致；
2. 将raw_result编码为32位无符号二进制数据；
3. 再将该二进制数据解码为32位有符号整数，此时超出32位的部分会被自动截断，并且最高位会被当作符号位处理，完美复现JSMath.imul的行为。

测试imul(-5, 12)时，raw_result为51539607492，转换后得到-60，与JS结果完全一致。

内容的提问来源于stack exchange，提问作者Jon Girard