Linux下C库学习 - setjmp.h

<setjmp.h>是C提供的用来完成非局部跳转的函数库，说到‘跳转’，大部分可能会先想到goto，但是使用goto的时候只能在函数内部进行使用，而使用<setjmp.h>中的函数可以在不同地方进行跳转，突破了这个限制，使用起来更便捷方便，那我们一起来看看<setjmp.h>中都有哪些函数吧

1. setjmp

   • 函数原型

     int setjmp(jmp_buf env);</br>

   • 参数jmp_buf

     参数jmp_buf，这个参数是用来保存堆栈信息的，以便在跳转的时候能够恢复原来的堆栈

   • 返回值

     如果是直接调用setjmp，则返回值是0，如果是跳转过来的则返回值是非0

2. longjmp

   • 函数原型

     void longjmp(jmp_buf env, int val);

   • 参数jmp_buf，val

     jmp_buf和之前一样，这个val必须非0，val的值是多少调用longjmp之后，setjmp的返回值就是多少
     
     
     

举个简单的例子方便大家理解

#include <stdio.h>
#include <setjmp.h>

jmp_buf env_buf;

void func1(void)
{
    printf("Into func1\n");
    longjmp(env_buf, 1);   
    printf("Exit func1\n");
}

int main(void)
{
    int result;

    result = setjmp(env_buf);

    if (result == 0) {
        printf("First call setjmp\n");
        func1();
        return 0;
    } else if (result == 1 ) {
        printf("Long jmp result is 1\n");
        return 0;
    }

    return 0;
}

这里输出结果为

First call setjmp
Into func1
Long jmp result is 1

如果只是这样简单的使用，看上去好像并没有什么有用的地方，但是如果稍微改进一下，你就会发现大不一样了，改进的函数如下

#include <stdio.h>
#include <setjmp.h>
#include <stdbool.h>

jmp_buf env_buf;

void func1(void)
{
    printf("Into func1\n");

    if (true) {
        printf("func1 true\n");
    } else {
        printf("func2 false\n");
        longjmp(env_buf, 1);   
    }

    printf("Exit func1\n");
}

void func2(void)
{
    printf("Into func2\n");

    if (true) {
        printf("func2 true\n");
    } else {
        printf("func2 false\n");
        longjmp(env_buf, 2);
    }

    printf("Exit func2\n");
}

int main(void)
{
    int result;

    result = setjmp(env_buf);

    if (result == 0) {
        printf("First call setjmp\n");
        func1();
        func2();
        return 0;
    } else if (result == 1 ) {
        printf("Long jmp result is 1\n");
        return 0;
    } else if (result == 2) {
        printf("Long jmp result is 2\n");
        return 0;
    }

    return 0;
}

仔细看看main函数，是不是和C++以及Java里面的try/catch语句块有异曲同工之妙，没错，使用跳转函数，可以方便你在异常情况下的处理，用C模拟了简单的try/catch，代码一下就会简洁很多，方便归方便，使用setjmp的时候有一点需要注意的，那就是包含setjmp()宏调用的函数一定不能终止，为什么会这样呢，因为C语言在调用函数的时候，系统会记录这个函数的一些信息，比如，参数类型，函数地址，返回值地址等等，然后将这些信息入栈，当函数返回的时候再将这些信息出栈，除了函数本身，更为重要的一个参数就是栈顶指针(SP)，而jmp_buf恰恰保存的就是这些信息，所以，如果调用setjmp的函数退出了，调用longjmp时，原来保存的信息和当前运行的环境不符，可能就会引起一些莫名其妙的错误，

上面大家也看到了，使用longjmp的时候程序并不会从函数返回，而是直接跳转到了主函数，那这里就可能会有一个问题，如果longjmp的函数是信号处理函数，当进入信号处理函数的时候，该信号自动被加入到信号屏蔽字当中，那什么时候恢复呢？举个简单的例子

#include <stdio.h>
#include <setjmp.h>
#include <signal.h>

jmp_buf env_buf;

void sig_fpe(int signo)
{
    longjmp(env_buf, 1);
}

int main(int argc, char *argv[])
{
    signal(SIGFPE, sig_fpe); 
    if (setjmp(env_buf) == 0) {
        int ret = 10 / 0;
    } else {
        printf("catch exception\n");
        int ret = 10 / 0;
    }
}

这里我两次对0进行整除，但是输出结果是

catch exception
Floating point exception (core dumped)

可见，信号处理函数值进行了一次，也就是说信号屏蔽字并没有恢复，导致第二次信号没有进入信号处理函数，事实上POSIX并没有规定到底是什么时候恢复信号屏蔽字，是程序返回的时候还是程序退出的时候并没有明确的定义，为了解决这个问题，提供了两个函数

1. sigsetjmp

   • 函数原型

     int sigsetjmp(sigjmp_buf env, int savesigs);

   • 参数sigjmp_buf，savesigs

     sigjmp_buf和之前一样，是用来保存堆栈信息和信号信息的
     savesigs如果savemask非0，则sigsetjmp在env中保存进程的当前信号屏蔽字。调用siglongjmp时，如果带 非0 savemask的sigsetjmp调用已经保存了env，则siglongjmp从其中恢复保存的信号屏蔽字。

   • 返回值

     如果是第一次调用，则返回0，如果是从`siglongjmp`返回则为非0
     

2. siglongjmp

   • 函数原型

     void siglongjmp(sigjmp_buf env, int val);

   • 参数env，val

     `env`和之前一样，表示要恢复的环境
     `val`也和之前一样，数值是多少`sigsetjmp`的返回值就是多少
     

还是刚才的例子，我们使用这两个函数改编下

#include <stdio.h>
#include <setjmp.h>
#include <signal.h>

static sigjmp_buf jmpbuf;

void sig_fpe(int signo)
{
    siglongjmp(jmpbuf, 1);
}

int main(int argc, char *argv[])
{
    signal(SIGFPE, sig_fpe); 
    if (sigsetjmp(jmpbuf, 1) == 0) {
        int ret = 10 / 0;
    } else {
        printf("catch exception\n");
        int ret = 10 / 0;
    }
} 

大家可以试验下，这里是会无限打印catch exception的，如果把sigsetjmp(jmpbuf, 1)改成sigsetjmp(jmpbu, 0)，那就和上面的是一样的了

参考文献
http://www.cnblogs.com/lienhua34/archive/2012/04/22/2464859.html
http://blog.csdn.net/smstong/article/details/50728022
http://blog.chinaunix.net/xmlrpc.php?r=blog/article&uid=14327709&id=2978705
http://www.cnblogs.com/nufangrensheng/p/3516134.html