C语言实现协程(一)

目录

  • 基本原理
  • 协程管理
  • 协程实现

引言

在使用socket编程时,我们会用到accept、connect、recv、send等函数,这些函数在没有数据到达时,会阻塞等待IO数据的到达。这不利于我们处理多个连接并快速响应。一种方案是,服务端每accept一个连接,就创建一个新的线程用来处理这个连接。这会导致线程过多,而且线程之前切换开销很大。这就可以使用到协程了。当然不止socket这种可以使用协程,IO密集型都可以使用协程,无论是网络IO还是其他IO。

协程

协程可以理解为用户态的轻量级的非抢占式的线程。

特点

用户态:协程是在用户态实现调度。
轻量级:协程不用内核调度,内核态与用户态之间切换。
非抢占:协程是由用户自己实现调度,并且同一时间只能有一个协程在执行,协程自己主动交出CPU的。

优缺点

优点:

  1. 协程切换的时候开销小,用户态且轻量
  2. 非抢占式,不用加很多锁,减小复杂度,不用很复杂的处理线程同步问题。
    缺点:
    协程不能利用多核,只能使用单核,因为同时只有一个协程在运行。

适用场景

IO密集型。
在IO密集的情况下,协程的开销比线程小,能快速实现调度。
协程不适用于计算密集型,协程不能很好的利用多核cpu。

ucontext组件

linux下在头文件< ucontext.h >提供了getcontext(),setcontext(),makecontext(),swapcontext()四个函数和mcontext_t和ucontext_t结构体。

其中mcontext_t与机器相关。ucontext_t结构体如下(一般在/usr/include下):

typedef struct ucontext
{
    unsigned long int uc_flags;
    struct ucontext *uc_link;
    stack_t uc_stack;
    mcontext_t uc_mcontext;
    __sigset_t uc_sigmask;
    struct _libc_fpstate __fpregs_mem;
} ucontext_t;

其中uc_link指向下文,及当前上下文(可以理解为执行状态)执行完了,恢复运行的上下文;
uc_sigmask为该上下文中的阻塞信号集合;
uc_stack为该上下文中使用的栈;
uc_mcontext保存的上下文的特定机器表示,包括调用线程的特定寄存器等。
通过栈式计算机原理,我们可以知道,保存栈区和所有通用寄存器的值就可以保存程序运行的状态。这里uc_mcontext就是用来保存所有的寄存器的值的。而我们把栈设置到uc_stack所指向的内存,
uc_stack就保存了栈的状态。

ucontext的4个函数介绍

int getcontext(ucontext_t *ucp);

获取当前上下文,初始化ucp结构体,将当前的上下文保存到ucp中。如果执行成功,返回0。执行失败返回-1。

int setcontext(const ucontext_t *ucp);

设置当前上下文,设置当前的上下文为ucp, 恢复ucp的执行状态。如果ucp执行完了,会恢复到uc_link所指向的上下文,若uc_link为NULL,则线程退出。如果执行成功,不返回。执行失败返回-1。

void makecontext(ucontext_t *ucp, void (*func)(), int argc, ...);

创建上下文,修改通过getcontext取得的上下文ucp, 然后给该上下文指定一个栈空间ucp->stack,设置后继的上下文ucp->uc_link。

int swapcontext(ucontext_t *oucp, ucontext_t *ucp);

切换上下文,保存当前上下文到oucp结构体中,然后激活upc上下文。 如果执行成功,不返回。执行失败返回-1。

ucontext简单试用

下面一段简单代码体验一下获取和恢复上下文:

#include 
#include 
#include 

int main(int argc, const char *argv[]){
	ucontext_t context;
	
	getcontext(&context);
	puts("Hello world");
	sleep(1);
	setcontext(&context);
	return 0;
}

保存并编译,执行结果:

Hello world  
Hello world  
Hello world  
Hello world  
Hello world  
Hello world  
Hello world  
^C  

如果不主动ctrl+c结束程序,会不断执行下去。第8行哪里getcontext保存了上下文,相当于按了一个暂停并保存起来放到了context变量里面,程序继续执行,到第11行的时候,恢复了原来的上下文,相当于把context变量里面保存的东西恢复了,恢复到第8行的状态。所以程序会不断执行下去。

函数之前切换

下面一段简单的代码体验一下不同函数间切换和恢复上下文:

#include 
#include 
#include 


void fun1()
{
    printf("func1\n");
}
void fun2(ucontext_t *ctx)
{
    ucontext_t t_ctx;
    printf("func2\n");
    swapcontext(&t_ctx, ctx);
}

int main(int argc, char *argv[]) {
    ucontext_t context, context2, main_ctx;
    char stack[1024];                                                                                                                         
    getcontext(&context);
    getcontext(&context2);
    context.uc_stack.ss_sp = stack;
    context.uc_stack.ss_size = 1024;
    context.uc_link = &main_ctx;
    context2.uc_stack.ss_sp = stack;
    context2.uc_stack.ss_size = 1024;
    makecontext(&context,(void (*)(void))fun1, 0);
    swapcontext(&main_ctx, &context);
    puts("main");
    makecontext(&context2,(void (*)(void))fun2,1,&main_ctx);
    swapcontext(&main_ctx, &context2);
    puts("main over");
    return 0;
}

执行结果:

func1
main
func2
main over

这里第28行,先保存上下文到main_ctx,切换到context上下文执行,就调用了fun1函数,fun1函数执行完后恢复uc_link指向的上下文main_ctx。然后接着执行输出了main。第31行保存上下文到main_ctx恢复context2,执行fun2函数,并传一个函数main_ctx,执行到14行后保存上下文到gCtx,然后恢复ctx(ctx是通过指针传过来的mainc_ctx);即恢复到了32行的状态最后main函数执行完成。

总结

linux系统,为我们提供了获取当前执行状态的函数、恢复当前上下文的函数和构造当前上下文的函数,所有栈的区域我们都可以自己分配一段空间来做协程的栈,每个协程都有自己独立的空间,我们就可以自己在用户态实现上线文切换,实现一个“用户态的线程”,即协程。

你可能感兴趣的:(协程实现)