Question

背景：我试图通过提出这个玩具问题来弄清楚如何实现continuation / coroutines / generator（无论以下是什么）。环境是gcc 4.6和linux 3.0 x86_64上的C ++ 11。非便携式很好，但不允许使用外部库（boost.coroutine，COROUTINE等）。我认为longjmp(3)和/或makecontext(2)和朋友可能有所帮助但不确定。

说明

以下玩具解析器应该解析as和bs等长的序列。即

((a+)(b+))+

使得第二次包围生产的长度等于第三次。

当它找到一个产品（例如aaabbb）时，它会输出它找到的a个数（例如3个）。

代码：

#include <stdlib.h>
#include <iostream>
using namespace std;

const char* s;

void yield()
{
        // TODO: no data, return from produce
        abort();
}

void advance()
{
        s++;
        if (*s == 0)
                yield();
}

void consume()
{
        while (true)
        {
                int i = 0;

                while (*s == 'a')
                {
                        i++;
                        advance();
                }

                cout << i << " ";

                while (i-- > 0)
                {
                    if (*s != 'b')
                        abort();
                    advance();
                }
        }
}

void produce(const char* s_)
{
        s = s_;

        // TODO: data available, continue into consume()
        consume();
}

int main()
{
        produce("aaab");
        produce("bba");
        produce("baa");
        produce("aabbb");
        produce("b");

        // should print: 3 1 4

        return 0;
}

问题：

如您所见，调用consume时必须保存yield调用堆栈的状态，然后produce返回。再次调用produce时，必须从consume返回重新启动yield。挑战在于修改produce调用consume的方式，并实施yield，以便它们按预期运行。

（显然重新实现消费，以便它保存和重建其状态会破坏练习的目的。）

我认为需要做的是类似于makecontext手册页底部的示例：http://www.kernel.org/doc/man-pages/online/pages/man3/makecontext.3.html，但不清楚如何将其转换为此问题。（我需要睡觉）

解决方案：

（感谢Chris Dodd的设计）

#include <stdlib.h>
#include <iostream>
#include <ucontext.h>
using namespace std;

const char* s;
ucontext_t main_context, consume_context;

void yield()
{
    swapcontext(&consume_context, &main_context);
}

void advance()
{
    s++;
    if (*s == 0)
            yield();
}

void consume()
{
    while (true)
    {
            int i = 0;

            while (*s == 'a')
            {
                    i++;
                    advance();
            }

            cout << i << " ";

            while (i-- > 0)
            {
                    advance();
            }
    }
}

void produce(const char* s_)
{
    s = s_;

    swapcontext(&main_context, &consume_context);
}

int main()
{
    char consume_stack[4096];

    getcontext(&consume_context);
    consume_context.uc_stack.ss_sp = consume_stack;
    consume_context.uc_stack.ss_size = sizeof(consume_stack);
    makecontext(&consume_context, consume, 0);

    produce("aaab");
    produce("bba");
    produce("baa");
    produce("aabbb");
    produce("b");

    // should print: 3 1 4

    return 0;
}

Answer 1

使用makecontext / swapcontext非常直接 - 你使用makecontext创建一个新的协程上下文和swapcontext来交换它们。在您的情况下，您需要一个额外的协同程序来运行consume无限循环，并在主要上下文中运行main和produce。

所以main应调用getcontext + makecontext来创建一个运行消耗循环的新上下文：

getcontext(&consume_ctxt);
// set up stack in consume_context
makecontext(&consume_ctxt, consume, 0);

然后produce将切换到它而不是直接调用consume：

void produce(const char* s_)
{
    s = s_; 
    swapcontext(&main_ctxt, &consume_ctxt);
}

最后yield只是调用swapcontext(&consume_ctxt, &main_ctxt);切换回主要上下文（将继续produce并立即返回）。

请注意，由于consume是一个无限循环，因此您无需过多担心返回时会发生什么（因此永远不会使用该链接）

C ++ / gcc / linux中的Continuations / Coroutines / Generators

1 个答案: