在我们使用多线程进程进行编程的时候,有的时候也会创建多进程,这样的话,因为在fork的时候, 子进程也会把父进程的加锁状态复制过去,这样的话就会出现死锁的情况,这样的话我们就需要使用到pthread_atfork函数了。
首先我先介绍一下,fork的时候并不会把父进程中的线程也复制过去,
《POSIX多线程程序设计》中的原话:“当多线程进程调用fork创造子进程时,Pthreads指定只用那个调用fork的线程在子进程中存在。”换句话说,只会复制调用fork的那个执行流。
API
#include <pthread.h>
int pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void));
pthread_atfork()在fork()之前调用,当调用fork时,内部创建子进程前在父进程中会调用prepare,内部创建子进程成功后,父进程会调用parent ,子进程会调用child
案例
#include<iostream>
#include<unistd.h>
#include<pthread.h>
#include<sys/wait.h>
pthread_mutex_t g_mutex;
void * Fun(void * arg)
{
pthread_mutex_lock(&g_mutex);
std::cout<<"Fun lock"<<std::endl;
sleep(5);
pthread_mutex_unlock(&g_mutex);
std::cout<<"Fun unlock"<<std::endl;
return NULL;
}
int main()
{
pthread_t threadpid ;
pthread_mutex_init(&g_mutex,NULL);
pthread_create(&threadpid,NULL,Fun,NULL);
sleep(1);
pid_t pid = fork();
if(-1 == pid)
{
std::cout<<"fork erro"<<std::endl;
exit(-1);
}else if(0 == pid)
{
std::cout<<"Child will lock"<<std::endl;
pthread_mutex_lock(&g_mutex);
std::cout<<"Child lock"<<std::endl;
pthread_mutex_unlock(&g_mutex);
std::cout<<"Child unlock"<<std::endl;
}else
{
wait(NULL);
exit(0);
}
}
我们在线程中加锁,通过在fork()前sleep1秒来让线程加锁,然后再进行fork,此时,子进程中的锁的状态已经是被获取了。
运行结果
这个时候,我们可以看到,子进程想要获取锁,但是因为此时锁的状态已经是被获取了,此时出现了死锁。
使用pthread_atfork函数来解决这个问题
#include<iostream>
#include<unistd.h>
#include<pthread.h>
#include<sys/wait.h>
pthread_mutex_t g_mutex;
void BeforeFork()
{
pthread_mutex_lock(&g_mutex);
}
void AfterFork()
{
pthread_mutex_unlock(&g_mutex);
}
void * Fun(void * arg)
{
pthread_mutex_lock(&g_mutex);
std::cout<<"Fun lock"<<std::endl;
sleep(5);
pthread_mutex_unlock(&g_mutex);
std::cout<<"Fun unlock"<<std::endl;
return NULL;
}
int main()
{
pthread_atfork(BeforeFork,AfterFork,AfterFork);
pthread_t threadpid ;
pthread_mutex_init(&g_mutex,NULL);
pthread_create(&threadpid,NULL,Fun,NULL);
sleep(1);
pid_t pid = fork();
if(-1 == pid)
{
std::cout<<"fork erro"<<std::endl;
exit(-1);
}else if(0 == pid)
{
std::cout<<"Child will lock"<<std::endl;
pthread_mutex_lock(&g_mutex);
std::cout<<"Child lock"<<std::endl;
pthread_mutex_unlock(&g_mutex);
std::cout<<"Child unlock"<<std::endl;
}else
{
wait(NULL);
exit(0);
}
}
在fork前先依次对父进程加锁,然后再fork后依次在父进程和子进程中解锁就可以了。
结果