setjmp和longjmp

一、setjmp和longjmp函数

1.setjmp
功能：把执行这个函数时的各种上下文信息保存起来，主要就是一些寄存器的值

int setjmp(jmp_buf env);

jmp_buf env : 存储上下文的结构体变量
返回值 : setjmp返回0，但通过longjmp恢复上下文时返回值为longjmp的参数val

2.longjmp
从 jmp_buf 结构体中恢复上下文，该函数不返回，而是将val作为恢复上下文后setjmp的返回信息

void longjmp (struct jmp_buf env, int val)

struct jmp_buf env : 存储setjmp保存过的上下文的结构体变量
int val : 恢复上下文后setjmp的返回值

二、注意事项

1. longjmp 一定要在 setjmp 之后，这很好理解，没有存储过上下文就没法恢复…
2. longjmp一定要在setjmp作用域内。因为当 setjmp 函数调用时，它保存的上下文环境，只在作用域以内才会有效。如果函数返回到了上层（或更上层）的函数环境中，那么setjmp保存的程序的环境也将会无效，因为堆栈中的数据此时将可能发生覆盖，所以当然会导致不可预料的执行后果。

三、应用场景

异常捕获、协程切换

1. 简单的异常捕获实现

#include <setjmp.h>
#include <stdio.h>
#include <unistd.h>
#include <vector>
using namespace std;

jmp_buf env;

int func(vector<int>& nums, int i){
    if(i < 0){
        longjmp(env, 1);
    }
    else if(i >= nums.size()){
        longjmp(env, 2);
    }
    return nums[i];
}

int main(){
    int key = 5;
    vector<int> nums = {9, 8, 7, 6, 5};
    int val = 0;

    int ret = setjmp(env);
    
    if(ret == 0){
        val = func(nums, key);
        printf("nums[%d]为%d\n", key, val);
    }
    else if(ret == 1){
        printf("错误类型1: 索引小于0\n");
    }
    else if(ret == 2){
        printf("错误类型2: 索引大于数组长度\n");
    }
    else {
        printf("其他错误类型\n");
    }

    return 0;
}

2. 简单协程实现

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

jmp_buf env;

void func1(int row){
    while(row < 10){
        printf("row %d: func 1\n", row);
        longjmp(env, ++row);
    }
    longjmp(env, -1);
}

void func2(int row){
    while(row < 10){
        printf("row %d: func 2\n", row);
        longjmp(env, ++row);
    }
    longjmp(env, -1);
}

void func3(int row){
    while(row < 10){
        printf("row %d: func 3\n", row);
        longjmp(env, ++row);
    }
    longjmp(env, -1);
}

void schedule(){
    int ret = setjmp(env);
    if(ret == -1){
        printf("end\n");
    }
    else if(ret % 3 == 0){
        func1(ret);
    }
    else if(ret % 3 == 1){
        func2(ret);
    }
    else if(ret % 3 == 2){
        func3(ret);
    }

}


int main(){
    schedule();
}