C语言用 setjmp/longjmp 实现“闭包”

最新推荐文章于 2023-03-20 11:22:27 发布

weixin_33901641

最新推荐文章于 2023-03-20 11:22:27 发布

阅读量153

点赞数

文章标签： c/c++ python

原文链接：https://my.oschina.net/backbye/blog/1922955

版权

2019独角兽企业重金招聘Python工程师标准>>>

C语言本身并不支持闭包，也就是说一个函数的局部变量在下一次调用的时候就会被重置（被重新赋初值，或者重新分配存储空间而是随机值）；我在学习到setjmp用法的时候想到：通过在函数体中设置一个入口点，就可以避免局部变量消失的问题。如下：

void function（void）{

int a1;

char a2

......

//以上定义变量

设置断点（）

函数逻辑语句；

}

于是每次我们只需要从断点进入，就可以保证局部变量不消失，也就是我们追求的让函数拥有记忆的能力，而不是通过外部变量，这种能力就是闭包的作用。

但是有一个问题，longjmp()函数并不能进入一个已经返回过的函数，因为返回过的函数的栈帧已经退栈。于是委屈求全，在函数体中“调用”自身。注意调用打了引号，如果是递归调用，那么是不存在断点的，也就说函数又会从头开始执行一次，包括局部变量的初始化语句。所以通过longjmp()调用实现了断点继续，变量记忆的功能；虽然不能想真正的闭包函数那样可以从外面随意调用，但是在内部的调用在功能上和闭包毫无差别。可以理解为内部闭包。

PS: 因为是函数内部的转移，所以和goto差不多。。。真正要实现闭包我觉得需要有调度机制，比如说再开一个线程。。。

下面给出示例代码，自己去体会。

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

static int state=0;
jmp_buf buf;
void print(int i){
    state=i;
    printf("run step[%d]...\n",i);
}

void tellnum(int num){
    printf("count is now: %d\n",num);
}
void coroutine(void){
    int count=0;
    setjmp(buf);
    switch(state){
        case 0:     do{
                    print(1);
                    tellnum(count++);
                    longjmp(buf,1);
                    break;
        case 1:     print(2);
                    tellnum(count++);
                    longjmp(buf,1);
                    break;
        case 2:     print(3);
                    tellnum(count++);
        }while(0);
    }
    tellnum(count++);
}

void main(){
    coroutine();
}

再给一个协程示例：发现setjmp并不能保护局部变量，它只是切换了上下文，并强行把命令指针发生跳转。A函数中longjum 到B, 其实并没有产生一个新的调用栈，而只是给A的修改了调用栈的方向和上下文（其实就是修改了寄存器）。所以在下面的示例中，consumer 和producers虽然都定义了各自的局部变量num, 但是它们修改和获取的是同一个（因为偏移是一样的，如果改变定义语句，如新增一个变量，就会完全不同）。

我尝试过起不同的名字，但是被调用（longjmp）的那个函数中的局部变量会成为随机值，因为在新的上下文中，原本的指针失去了意义，通过指针名找不到值，于是被设置为某个固定值（相对栈顶的偏移），而且那个值对应为内存位置并不被保护。

#include<setjmp.h>
#include<stdio.h>
#include<stdlib.h>
#include<time.h>
#include<unistd.h>
static int state=0;
jmp_buf producer_buf, consumer_buf;
void producer(void);
void consumer(void);
typedef struct{
	int num;
}Cakes;
static Cakes cakes={0};
static int first = 1;

void cooking(void){
	printf("\033[33m\tproducer is making cake[%d]\n\033[0m",++cakes.num);
	sleep(1);
}
	
void eating(void){
	printf("\033[32m\tconsumer is eating cake[%d]\n\033[0m", cakes.num--);
	sleep(2);
}
void producer(void){
	int num=0;
	setjmp(producer_buf);
	while(srand(time(NULL)),cakes.num<(rand()%5)){
		printf("producer[%d]:\n",num++);
		if(num>30){
			printf("\033[35mproducer is making more than 30 cakes,he quited!\n\033[0m");
			exit(0);
		}
		cooking();
	}
	longjmp(consumer_buf,num);
}

void consumer(void){
	int num=0;
	setjmp(consumer_buf);
	while(srand(time(NULL)),cakes.num>(rand()%5)){
		printf("consumer[%d]:\n",num++);
		eating();
	}
	switch(first){
		case 1: do{first=0;break;
		case 0: longjmp(producer_buf,1);	
				}while(0);
	}
	//longjmp(producer_buf,1);
}

void main(){
	consumer();
	producer();
}

转载于:https://my.oschina.net/backbye/blog/1922955