setjmp和longjmp

原创已于 2024-01-03 14:47:59 修改 · 588 阅读

12 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#服务器 #linux #unix #c语言 #c++

于 2024-01-02 21:43:50 首次发布

本文详细介绍了C语言中的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的返回值

二、注意事项

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

三、应用场景

异常捕获、协程切换

简单的异常捕获实现

#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;
}

简单协程实现

#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();
}