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 <stdio.h>
#include <ucontext.h>
#include <unistd.h>

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 <stdio.h>
#include <ucontext.h>
#include <unistd.h>


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

  • 8
    点赞
  • 45
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
以下是一个简单的基于C语言和Linux的协程示例: ```c #include <stdio.h> #include <stdlib.h> #include <ucontext.h> ucontext_t main_context, coroutine_context; void coroutine() { printf("Coroutine\n"); swapcontext(&coroutine_context, &main_context); printf("Coroutine end\n"); swapcontext(&coroutine_context, &main_context); } int main() { char coroutine_stack[4096]; getcontext(&coroutine_context); coroutine_context.uc_stack.ss_sp = coroutine_stack; coroutine_context.uc_stack.ss_size = sizeof(coroutine_stack); coroutine_context.uc_link = &main_context; makecontext(&coroutine_context, (void (*) (void)) coroutine, 0); printf("Main start\n"); swapcontext(&main_context, &coroutine_context); printf("Main end\n"); swapcontext(&main_context, &coroutine_context); printf("Main end 2\n"); return 0; } ``` 以上代码定义了两个上下文 `main_context` 和 `coroutine_context`,以及一个协程函数 `coroutine()`,该函数在执行过程中交换了主上下文和协程上下文,从而实现协程的效果。 在 `main()` 函数中,我们首先创建了一个协程上下文 `coroutine_context`,并为其分配一个 4096 字节大小的栈空间。然后我们通过 `makecontext()` 函数将 `coroutine()` 函数设置为该上下文的执行函数。接着,在第一次交换上下文之前,我们调用了 `getcontext()` 函数来获取主上下文 `main_context`。然后我们交换上下文,执行协程函数,打印出 "Coroutine",并将控制权还回给主上下文。此时,主上下文恢复,继续执行,打印出 "Main end"。接着,我们再次交换上下文,回到协程上下文中,继续执行,打印出 "Coroutine end"。最后,我们再次交换上下文,回到主上下文,再次打印出 "Main end"。 以上就是一个简单的基于C语言和Linux的协程示例。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值