《一个“蝇量级” c 语言协程库》由会员分享，可在线阅读，更多相关《一个“蝇量级” c 语言协程库（15页珍藏版）》请在金锄头文库上搜索。
1、一个轻量级的协助程框架协程(coroutine)顾名思义就是“协作的例程”（co-operative routines）。跟具有操作系统概念的线程不一样，协程是在用户空间利用程序语言的语法语义就能实现逻辑上类似多任务的编程技巧。实际上协程的概念比线程还要早，按照 Knuth 的说法“子例程是协程的特例子例程是协程的特例”，一个子例程就是一次子函数调用，那么实际上协程就是类函数一样的程序组件，你可以在一个线程里面轻松创建数十万个协程，就像数十万次函数调用一样。只不过子例程只有一个调用入口起始点，返回之后就结束了，而协程入口既可以是起始点，又可以从上一个返回点继续执行，也就是说协程之间可以通过 y
2、ield 方式转移执行权，对称对称（symmetric）、平级）、平级地调用对方，而不是像例程那样上下级调用关系。当然 Knuth 的“特例”指的是协程也可以模拟例程那样实现上下级调用关系，这就叫非对称协程非对称协程（asymmetric coroutines）。基于事件驱动模型我们举一个例子来看看一种对称协程对称协程调用场景，大家最熟悉的“生产者-消费者”事件驱动模型，一个协程负责生产产品并将它们加入队列，另一个负责从队列中取出产品并使用它。为了提高效率，你想一次增加或删除多个产品。伪代码可以是这样的：123456789101112# producer coroutineloopwhile
3、queue is not fullcreate some new itemsadd the items to queueyield to consumer# consumer coroutineloopwhile queue is not emptyremove some items from queueuse the items13yield to producer大多数教材上拿这种模型作为多线程的例子，实际上多线程在此的应用还是显得有点“重量级”，由于缺乏 yield 语义，线程之间不得不使用同步机制来避免产生全局资源的竟态，这就不可避免产生了休眠、调度、切换上下文一类的系统开销，而且线程
4、调度还会产生时序上的不确定性。而对于协程来说，“挂起”的概念只不过是转让代码执行权并调用另外的协程，待到转让的协程告一段落后重新得到调用并从挂起点“唤醒”，这种协程间的调用是逻辑上可控的，时序上确定的，可谓一切尽在掌握中。当今一些具备协程语义的语言，比较重量级的如 C#、erlang、golang，以及轻量级的 python、lua、javascript、ruby，还有函数式的 scala、scheme 等。相比之下，作为原生态语言的 C 反而处于尴尬的地位，原因在于 C 依赖于一种叫做栈帧栈帧的例程调用，例程内部的状态量和返回值都保留在堆栈上，这意味着生产者和消费者相互之间无法实现平级调用，
5、当然你可以改写成把生产者作为主例程然后将产品作为传递参数调用消费者例程，这样的代码写起来费力不讨好而且看起来会很难受，特别当协程数目达到十万数量级，这种写法就过于僵化了。这就引出了协程的概念，如果将每个协程的上下文（比如程序计数器）保存在如果将每个协程的上下文（比如程序计数器）保存在其它地方而不是堆栈上，协程之间相互调用时，被调用的协程只要从堆栈以外其它地方而不是堆栈上，协程之间相互调用时，被调用的协程只要从堆栈以外的地方恢复上次出让点之前的上下文即可，这有点类似于的地方恢复上次出让点之前的上下文即可，这有点类似于 CPU 的上下文切换，的上下文切换，遗憾的是似乎只有更底层的汇编语言才能做到这
6、一点。难道 C 语言只能用多线程吗？幸运的是，C 标准库给我们提供了两种协程调度原语：一种是 setjmp/longjmp，另一种是 ucontext 组件，它们内部（当然是用汇编语言）实现了协程的上下文切换，相较之下前者在应用上会产生相当的不确定性（比如不好封装，具体说明参考联机文档），所以后者应用更广泛一些，网上绝大多数 C 协程库也是基于 ucontext 组件实现的。“蝇量级”的协程库在此，我来介绍一种“蝇量级”的开源 C 协程库 protothreads。这是一个全部用 ANSI C 写成的库，之所以称为“蝇量级”的，就是说，实现已经不能再精简了，几乎就是原语级别。事实上 proto
7、threads 整个库不需要链接加载，因为所有源码都是头文件，类似于 STL 这样不依赖任何第三方库，在任何平台上可移植；总共也就 5 个头文件，有效代码量不足 100 行；API 都是宏定义的，所以不存在调用开销；最后，每个协程的空间开销是 2 个字节（是的，你没有看错，就是一个 short 单位的“栈”！）当然这种精简是要以使用上的局限为代价的，接下来的分析会说明这一点。先来看看 protothreads 作者，Adam Dunkels，一位来自瑞典皇家理工学院的计算机天才帅哥。话说这哥们挺有意思的，写了好多轻量级的作品，都是 BSD 许可证。顺便说一句，轻量级开源软件全世界多如牛毛，可像
8、这位哥们写得如此出名的并不多。比如嵌入式网络操作系统 Contiki，国人耳熟能详的 TCP/IP 协议栈 uIP 和 lwIP 也是出自其手。上述这些软件都是经过数十年企业级应用的考验，质量之高可想而知。很多人会好奇如此“蝇量级”的代码究竟是怎么实现的呢？在分析 protothreads 源码之前，我先来给大家补一补 C 语言的基础课;-)简而言之，这利用了 C 语言特性上的一个“奇技淫巧”，而且这种技巧恐怕连许多具备十年以上经验的 C 程序员老手都不见得知晓。当然这里先要声明我不是推荐大家都这么用，实际上这是以破坏语言的代码规范为代价，在一些严肃的项目工程中需要谨慎对待，除非你想被炒鱿鱼。
9、C 语言的“yield 语义”下面的教程来自于一位 ARM 工程师、天才黑客 Simon Tatham（开源 Telnet/SSH 客户端 PuTTY 和汇编器 NASM 的作者，吐槽一句，PuTTY 的源码号称是所有正式项目里最难 hack 的 C，你应该猜到作者是什么语言出身）的博文：Coroutines in C。中文译文在这里。我们知道 python 的 yield 语义功能类似于一种迭代生成器，函数会保留上次的调用状态，并在下次调用时会从上个返回点继续执行。用 C 语言来写就像这样：12345int function(void) int i;for (i = 0; i lc)/* 声
10、明一个函数，返回值为 char 即退出码，表示函数体内使用了 proto thread，（个人觉得有些多此一举） */#define PT_THREAD(name_args) char name_args/* 协程入口点， PT_YIELD_FLAG=0 表示出让，=1 表示不出让，放在 switch 语句前面，下次调用的时候可以跳转到上次出让点继续执行 */#define PT_BEGIN(pt) char PT_YIELD_FLAG = 1; LC_RESUME(pt)-lc)/* 协程退出点，至此一个协程算是终止了，清空所有上下文和标志 */#define PT_END(pt) LC_E
11、ND(pt)-lc); PT_YIELD_FLAG = 0; PT_INIT(pt); return PT_ENDED; /* 协程出让点，如果此时协程状态变量 lc 已经变为 _LINE_ 跳转过来的，那么 PT_YIELD_FLAG = 1，表示从出让点继续执行。 */#define PT_YIELD(pt) do PT_YIELD_FLAG = 0; LC_SET(pt)-lc); if(PT_YIELD_FLAG = 0) return PT_YIELDED; while(0)/* 附加出让条件 */#define PT_YIELD_UNTIL(pt, cond) do PT_YIEL
12、D_FLAG = 0; LC_SET(pt)-lc); if(PT_YIELD_FLAG = 0) | !(cond) return PT_YIELDED; 313233343536373839404142434445464748495051525354555657 while(0)/* 协程阻塞点(blocking),本质上等同于 PT_YIELD_UNTIL，只不过退出码是 PT_WAITING，用来模拟信号量同步 */#define PT_WAIT_UNTIL(pt, condition) do LC_SET(pt)-lc); if(!(condition) return PT_WAIT
13、ING; while(0)/* 同 PT_WAIT_UNTIL 条件反转 */#define PT_WAIT_WHILE(pt, cond) PT_WAIT_UNTIL(pt), !(cond)/* 协程调度，调用协程 f 并检查它的退出码，直到协程终止返回 0，否则返回 1。 */#define PT_SCHEDULE(f) (f) count = c#define PT_SEM_WAIT(pt, s) do PT_WAIT_UNTIL(pt, (s)-count 0); -(s)-count; while(0)#define PT_SEM_SIGNAL(pt, s) +(s)-count这
14、些应该不需要我多说了吧，呵呵，让我们回到最初例举的生产者-消费者模型，看看 protothreads 表现怎样。Protothreads 实战1234567891011#include “pt-sem.h“#define NUM_ITEMS 32#define BUFSIZE 8static struct pt_sem mutex, full, empty;PT_THREAD(producer(struct pt *pt)static int produced;1213141516171819202122232425262728293031323334353637383940PT_BEGIN(
15、pt);for (produced = 0; produced NUM_ITEMS; +produced) PT_SEM_WAIT(pt, PT_SEM_WAIT(pt, add_to_buffer(produce_item();PT_SEM_SIGNAL(pt, PT_SEM_SIGNAL(pt, PT_END(pt);PT_THREAD(consumer(struct pt *pt)static int consumed;PT_BEGIN(pt);for (consumed = 0; consumed NUM_ITEMS; +consumed) PT_SEM_WAIT(pt, PT_SEM_WAIT(pt, consume_item(get_from_buffer();PT_SEM_SIGNAL(pt, PT_SEM_SIGNAL(pt, PT_END(pt);PT_THREAD(driver_thread(struct pt *pt)static struct pt pt_producer, pt_consumer;41424344454647484950PT_BEGIN(pt);PT_SEM_INIT(PT_SEM_INIT(PT_SEM
