根据调用线程中的变量值关闭线程

Question

我在C中使用posix线程。我的程序中有两个线程，thread1和thread2。 线程1启动线程2。

线程1维护一个变量var1，基于该变量必须安全退出线程2。

thread2调用了许多冗长的函数（其运行时间最多为5秒）。 在thread2中有许多使用malloc的内存分配。 一旦var1变为true，我需要关闭thread2，但仅在取消分配所有已分配的内存之后。 如何做到这一点？

代码如下：

void * func1(void *arg)
{
   pthread_create(&thread2, &attr, func2, NULL);
   while(1)
   {
      // ...
      var1 = func3();
      if(var1 == true)
      {   
          // Cancel thread2
      }
      // ...
   }
}

void * func2(void *arg)
{
    // ...
    func4(); // runs for 2 sec
    char* var2 = malloc(10);
    func5(); // runs for 5 sec
    char* var3 = malloc(20);
    // ...
 cleanup:
    free(var2);
    free(var3);
    return (void*) 0;
}

Answer 1

有几种方法可以做到这一点。 这是一个：

从线程1的角度来看，您可以在时间到时在线程2上简单地使用pthread_cancel 。 但是您需要相应地重组thread2，以确保它在安全的地方具有取消点。 您还需要重新构造数据的存储方式，以允许为线程2调用取消回调：

struct func2_data {
  char *var2;
  char *var3;
};

void func2_cleanup(void *vdata) {
  struct func2_data *data = vdata;
  free(data->var2);
  free(data->var3);
}

void * func2(void *arg)
{
  struct func2_data data = {
    NULL, NULL
  };

  pthread_cleanup_push(func2_cleanup, &data);

  // ....    

  func4();
  data.var2 = malloc(10);
  pthread_testcancel();

  func5();
  data.var3 = malloc(10);
  pthread_testcancel();

  // ....        

  pthread_cleanup_pop(1);
  return NULL;
}

那么这里发生了什么？

1. 所有需要释放的指针都捆绑在一起并初始化为NULL。 如果在分配任何单个变量之前调用了清除函数，则允许以free操作的方式对其进行操作。

2. 注册了一个回调以释放指针。 该回调将在任何取消点被调用。 包括线程何时从其处理函数返回。 因此，除了调用pop之外，您无需在其中明确地处理它。

3. 取消点通过pthread_testcancel添加到各个“安全”位置。 因此，允许线程在计算之间过早结束。 根据需要添加它们。

Answer 2

即使您的问题无法正确理解，在下面的代码中，我还是给出了一个想法来处理这些问题。 使用ptread_join安全退出线程

void * func1(void *arg)
{
    pthread_create(&thread2, &attr, func2, (void*)&var1); /* Pass addr of var1 as argument to thread*/
    while(1)
    {
        ...
        var1 = func3();
        if(var1 == TRUE){
            int *ret;
            pthread_join(thread2,&ret);
        }
        ...
    }
}

void * func2(void *arg)
{
    bool *check = (bool*)arg;
    ...
    func4(); // runs for 2 sec
    if(*check == TRUE)
        goto end;
    char* var2 = malloc(10);
    func5(); // runs for 5 sec
    if(*check == TRUE)
        goto clean_var2;
    char* var3 = malloc(20);
    ...
    cleanup:
    free(var3);
    clean_var2:
    free(var2);
    end:
    return (void*) 0;
}