3

考虑以下通过alloca()函数在堆栈上分配内存的玩具示例:

#include <alloca.h>

void foo() {
    volatile int *p = alloca(4);
    *p = 7;
}

使用 gcc 8.2 编译上述函数,-O3得到以下汇编代码:

foo:
   pushq   %rbp
   movq    %rsp, %rbp
   subq    $16, %rsp
   leaq    15(%rsp), %rax
   andq    $-16, %rax
   movl    $7, (%rax)
   leave
   ret

老实说,我本来希望有更紧凑的汇编代码。


分配内存的 16 字节对齐

上面代码中的指令andq $-16, %rax导致在地址和(包括两者)之间rax包含(仅)16 字节对齐的地址。rsprsp + 15

这种对齐强制是我不明白的第一件事:为什么alloca()将分配的内存对齐到 16 字节边界?


可能错过优化?

无论如何,让我们考虑一下我们希望分配的内存alloca()是 16 字节对齐的。即便如此,在上面的汇编代码中,请记住call foo,如果我们注意堆栈内部的状态,GCC 在执行函数调用时假定堆栈与 16 字节边界对齐(即 )foo() 就在推送rbp寄存器之后:

Size          Stack          RSP mod 16      Description
-----------------------------------------------------------------------------------
        ------------------
        |       .        |
        |       .        | 
        |       .        |            
        ------------------........0          at "call foo" (stack 16-byte aligned)
8 bytes | return address |
        ------------------........8          at foo entry
8 bytes |   saved RBP    |
        ------------------........0  <-----  RSP is 16-byte aligned!!!

我认为通过利用红色区域(即无需修改)和已经包含16 字节对齐地址rsp的事实,可以使用以下代码:rsp

foo:
   pushq   %rbp
   movq    %rsp, %rbp
   movl    $7, -16(%rbp)
   leave
   ret

寄存器中包含的地址rbp是 16 字节对齐的,因此rbp - 16也将对齐到 16 字节边界。

更好的是,新堆栈帧的创建可以被优化掉,因为rsp它没有被修改:

foo:
   movl    $7, -8(%rsp)
   ret

这只是错过的优化还是我在这里遗漏了其他东西?

4

2 回答 2

6

这是 gcc 中(部分)错过的优化。Clang 按预期做到了。

我说部分是因为如果你知道你将使用 gcc,你可以使用内置函数(对 gcc 和其他编译器使用条件编译来获得可移植代码)。

__builtin_alloca_with_align是你的朋友;)

这是一个示例(已更改,因此编译器不会将函数调用减少到单个 ret):

#include <alloca.h>

volatile int* p;

void foo() 
{
    p = alloca(4) ;
    *p = 7;
}

void zoo() 
{
    // aligment is 16 bits, not bytes
    p = __builtin_alloca_with_align(4,16) ;
    *p = 7;
}

int main()
{
  foo();
  zoo();
}

反汇编代码(带objdump -d -w --insn-width=12 -M intel

Clang 将生成以下代码 ( clang -O3 test.c) - 两个函数看起来很相似

0000000000400480 <foo>:
  400480:       48 8d 44 24 f8                          lea    rax,[rsp-0x8]
  400485:       48 89 05 a4 0b 20 00                    mov    QWORD PTR [rip+0x200ba4],rax        # 601030 <p>
  40048c:       c7 44 24 f8 07 00 00 00                 mov    DWORD PTR [rsp-0x8],0x7
  400494:       c3                                      ret    

00000000004004a0 <zoo>:
  4004a0:       48 8d 44 24 fc                          lea    rax,[rsp-0x4]
  4004a5:       48 89 05 84 0b 20 00                    mov    QWORD PTR [rip+0x200b84],rax        # 601030 <p>
  4004ac:       c7 44 24 fc 07 00 00 00                 mov    DWORD PTR [rsp-0x4],0x7
  4004b4:       c3                                      ret    

GCC 这个 ( gcc -g -O3 -fno-stack-protector)

0000000000000620 <foo>:
 620:   55                                      push   rbp
 621:   48 89 e5                                mov    rbp,rsp
 624:   48 83 ec 20                             sub    rsp,0x20
 628:   48 8d 44 24 0f                          lea    rax,[rsp+0xf]
 62d:   48 83 e0 f0                             and    rax,0xfffffffffffffff0
 631:   48 89 05 e0 09 20 00                    mov    QWORD PTR [rip+0x2009e0],rax        # 201018 <p>
 638:   c7 00 07 00 00 00                       mov    DWORD PTR [rax],0x7
 63e:   c9                                      leave  
 63f:   c3                                      ret    

0000000000000640 <zoo>:
 640:   48 8d 44 24 fc                          lea    rax,[rsp-0x4]
 645:   c7 44 24 fc 07 00 00 00                 mov    DWORD PTR [rsp-0x4],0x7
 64d:   48 89 05 c4 09 20 00                    mov    QWORD PTR [rip+0x2009c4],rax        # 201018 <p>
 654:   c3                                      ret    

如您所见,zoo 现在看起来像预期的那样,并且类似于 clang 代码。

于 2018-09-26T22:12:30.213 回答
4

x86-64 System V ABI 要求 VLA(C99 可变长度数组)以 16 字节对齐,对于 >= 16 字节的自动/静态数组也是如此。

看起来 gcc 被alloca视为 VLA,并且未能将常量传播到alloca每个函数调用仅运行一次的中。(或者它在内部alloca用于 VLA。)

通用alloca/VLA 不能使用红色区域,以防运行时值大于 128 字节。GCC 还使用 RBP 创建一个堆栈帧,而不是保存分配大小并add rsp, rdx稍后再做。

因此,如果大小是函数 arg 或其他运行时变量而不是常量,则 asm 看起来与它完全一样。 这就是导致我得出这个结论的原因。


此外alignof(maxalign_t) == 16, but allocaandmalloc可以满足返回可用于任何对象的内存的要求,而对于小于 16 字节的对象,没有 16 字节对齐。没有一个标准类型的对齐要求比 x86-64 SysV 中的大小更宽。


你是对的,它应该能够优化它:

void foo() {
    alignas(16) int dummy[1];
    volatile int *p = dummy;   // alloca(4)
    *p = 7;
}

并将其编译到movl $7, -8(%rsp); ret你建议。

这里alignas(16)对于 alloca 可能是可选的。


如果您真的需要 gcc 在常量传播使 argalloca成为编译时常量时发出更好的代码,您可以考虑首先使用VLA。GNU C++ 在 C++ 模式下支持 C99 风格的 VLA,但 ISO C++(和 MSVC)不支持。

或者可能使用if(__builtin_constant_p(size)) { VLA version } else { alloca version },但 VLA 的范围意味着您不能从if检测到我们正在与编译时常量内联的范围内返回 VLA size。所以你必须复制需要指针的代码。

于 2018-09-26T20:56:45.393 回答