linux内核中的likely和unlikely
Kernel version：2.6.14
CPU architecture：ARM920T
Author：ce123(http://blog.csdn.net/ce123)
GCC version：arm-linux-gcc-3.4.1
看内核时经常遇到if(likely( )){}或是if(unlikely( ))这样的语句，不甚了解，例如(选自kernel/fork.c中copy_process)：
SET_LINKS(p); if (unlikely(p->ptrace & PT_PTRACED)) __ptrace_link(p, current->parent);
下面详细分析一下。
likely() 与 unlikely()是内核中定义的两个宏。位于/include/linux/compiler.h中，具体定义如下：
#define likely(x) __builtin_expect(!!(x), 1) #define unlikely(x) __builtin_expect(!!(x), 0)
__builtin_expect是GCC(version>=2.9)引进的内建函数，其作用就是帮助编译器判断条件跳转的预期值，避免跳转造成时间乱费，有利于代码优化。查阅GCC手册，发现其定义如下(http://gcc.gnu.org/onlinedocs/gcc/Other-Builtins.html)：
-- Built-in Function: long __builtin_expect (long EXP, long C)
You may use `__builtin_expect' to provide the compiler with branch
prediction information. In general, you should prefer to use
actual profile feedback for this (`-fprofile-arcs'), as
programmers are notoriously bad at predicting how their programs
actually perform. However, there are applications in which this
data is hard to collect.
The return value is the value of EXP, which should be an integral
expression. The value of C must be a compile-time constant. The
semantics of the built-in are that it is expected that EXP == C.
For example:
if (__builtin_expect (x, 0))
foo ();
would indicate that we do not expect to call `foo', since we
expect `x' to be zero. Since you are limited to integral
expressions for EXP, you should use constructions such as
if (__builtin_expect (ptr != NULL, 1))
error ();
when testing pointer or floating-point values.
大致意思是：可以使用。由于大部分程序员在分支预测方面做得很糟糕，所以GCC提供了__builtin_expect这个内建函数，给编译器提供分支预测信息，以帮助程序员处理分支预测，优化程序。其第一个参数EXP为一个整型表达式，这个内建函数的返回值也是这个EXP，而C为一个编译期常量，这个函数的语义是：你期望EXP表达式的值等于常量C，从而GCC为你优化程序，将符合这个条件的分支放在合适的地方。由于该内建函数只提供了整型表达式，所以如果你要优化其他类型的表达式，可以采用指针的形式。
当GCC的版本较低时(_GNUC_MINOR__ < 96)，__builtin_expect直接返回EXP。下面的代码摘自/include/linux/compiler-gcc2.h。
/* These definitions are for GCC v2.x. */ /* Somewhere in the middle of the GCC 2.96 development cycle, we implemented a mechanism by which the user can annotate likely branch directions and expect the blocks to be reordered appropriately. Define __builtin_expect to nothing for earlier compilers. */ #include <linux/compiler-gcc.h> #if __GNUC_MINOR__ < 96 # define __builtin_expect(x, expected_value) (x) #endif
总结一下：if() 语句照常用， 和以前一样， 只是 如果你觉得if()是1 的可能性非常大的时候， 就在表达式的外面加一个likely()，如果可能性非常小(比如几率非常小)，就用unlikely()包裹上。下面我们看一个例子。
//test_builtin_expect.c
#define likely(x) __builtin_expect(!!(x), 1)
#define unlikely(x) __builtin_expect(!!(x), 0)
int test_likely(int x)
{
if(likely(x))
x = 5;
else
x = 6;
return x;
}
int test_unlikely(int x)
{
if(unlikely(x))
x = 5;
else
x = 6;
return x;
}
root@czu:~/桌面/socket# arm-linux-gcc -fprofile-arcs -O2 -c test.c
root@czu:~/桌面/socket# arm-linux-gcc -fprofile-arcs -O2 -o test test.c
root@czu:~/桌面/socket# arm-linux-objdump -D test > test.dis
000088cc <test_likely>:
88cc: e3500000 cmp r0, #0 ; 0x0
88d0: e92d4010 stmdb sp!, {r4, lr}
88d4: e59fc044 ldr ip, [pc, #68] ; 8920 <.text+0x148>
88d8: e59fe044 ldr lr, [pc, #68] ; 8924 <.text+0x14c>
88dc: e3a00005 mov r0, #5 ; 0x5
88e0: 0a000006 beq 8900 <test_likely+0x34>//前面通过cmp将r0和0进行比较，因为x=1的概率很大，优先执行不等于0的分支
88e4: e89c0018 ldmia ip, {r3, r4}
88e8: e3a02000 mov r2, #0 ; 0x0
88ec: e3a01001 mov r1, #1 ; 0x1
88f0: e0933001 adds r3, r3, r1
88f4: e0a44002 adc r4, r4, r2
88f8: e88c0018 stmia ip, {r3, r4}
88fc: e8bd8010 ldmia sp!, {r4, pc}
8900: e89e0006 ldmia lr, {r1, r2}
8904: e3a04000 mov r4, #0 ; 0x0
8908: e3a03001 mov r3, #1 ; 0x1
890c: e0911003 adds r1, r1, r3
8910: e0a22004 adc r2, r2, r4
8914: e3a00006 mov r0, #6 ; 0x6
8918: e88e0006 stmia lr, {r1, r2}
891c: e8bd8010 ldmia sp!, {r4, pc}
8920: 000121e0 andeq r2, r1, r0, ror #3
8924: 000121e8 andeq r2, r1, r8, ror #3
00008928 <test_unlikely>:
8928: e3500000 cmp r0, #0 ; 0x0
892c: e92d4010 stmdb sp!, {r4, lr}
8930: e59fc044 ldr ip, [pc, #68] ; 897c <.text+0x1a4>
8934: e59fe044 ldr lr, [pc, #68] ; 8980 <.text+0x1a8>
8938: e3a00005 mov r0, #5 ; 0x5
893c: 1a000007 bne 8960 <test_unlikely+0x38>//前面通过cmp将r0和0进行比较，因为x=0的概率很大，优先执行等于0的分支
8940: e89c0018 ldmia ip, {r3, r4}
8944: e3a02000 mov r2, #0 ; 0x0
8948: e3a01001 mov r1, #1 ; 0x1
894c: e0933001 adds r3, r3, r1
8950: e0a44002 adc r4, r4, r2
8954: e3a00006 mov r0, #6 ; 0x6
8958: e88c0018 stmia ip, {r3, r4}
895c: e8bd8010 ldmia sp!, {r4, pc}
8960: e89e0006 ldmia lr, {r1, r2}
8964: e3a04000 mov r4, #0 ; 0x0
8968: e3a03001 mov r3, #1 ; 0x1
896c: e0911003 adds r1, r1, r3
8970: e0a22004 adc r2, r2, r4
8974: e88e0006 stmia lr, {r1, r2}
8978: e8bd8010 ldmia sp!, {r4, pc}
897c: 000121f8 streqd r2, [r1], -r8
8980: 000121f0 streqd r2, [r1], -r0
如果我们将代码修改一下，不用这两个宏结果会怎样呢？
//test_builtin_expect.c
int test_likely(int x)
{
if(x)
x = 5;
else
x = 6;
return x;
}
int test_unlikely(int x)
{
if(x)
x = 5;
else
x = 6;
return x;
}
反汇编代码如下：
00008460 <test_likely>:
8460: e3500000 cmp r0, #0 ; 0x0
8464: 03a00006 moveq r0, #6 ; 0x6
8468: 13a00005 movne r0, #5 ; 0x5
846c: e1a0f00e mov pc, lr
00008470 <test_unlikely>:
8470: e3500000 cmp r0, #0 ; 0x0
8474: 03a00006 moveq r0, #6 ; 0x6
8478: 13a00005 movne r0, #5 ; 0x5
847c: e1a0f00e mov pc, lr
如上述例子分析所示，两个函数编译生成的汇编语句所使用到的跳转指令不一样，仔细分析下会发现__builtin_expect实际上是为了满足在大多数情况不执行跳转指令，__builtin_expect仅仅是告诉编译器优化，并没有改变其对真值的判断。宏likely和宏unlikely唯一的作用就是选择”将if分支还是else分支放在跳转指令之后，从而优化程序的执行效率”。 因为likely(EXP)代表条件表达式EXP很可能成立，而unlikely(EXP)代表条件表达式EXP很可能不成立，当程序员清楚EXP表达式 多数情况成立(不成立)时，就可使用likely(unlikely)，使if分支(else分支)紧跟跳转指令其后，从而在大多数情况下不用执行跳转指令，避开跳转指令所带来的开销，从而达到优化的目的。
还有一点需要注意的是，在生成汇编时用的是arm-linux-gcc -fprofile-arcs -O2 -c test_builtin_expect.c，而不是arm-linux-gcc -O2 -c test_builtin_expect.c。