【Linux pthread_atfork 函数】

在我们使用多线程进程进行编程的时候，有的时候也会创建多进程，这样的话，因为在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后依次在父进程和子进程中解锁就可以了。
结果