Question

我有问题。我正在开发一个预期会出现分段错误的项目，我必须使用setjmp和ljmp机制来捕获分段错误并更改默认行为（即应用程序崩溃）。当代码抛出分段错误时，已经完成了内存分配。有没有什么办法可以在调用setjmp后释放内存分配？简单的说法：我想在初始调用setjmp和从ljmp调用的调用之间释放内存分配。这是一个很小但工作正常的代码，我们可以讨论......

由于

#include <stdio.h>
#include <setjmp.h>
#include <signal.h>
#include <stdlib.h>

static jmp_buf buf;



static void 
addSignalToMask()
{
  sigset_t sigset;
  if ( 0 > sigprocmask(0,NULL,&sigset)){
    fprintf(stderr," error while retreiving procmask \n");
    exit(1);
  }

  if ( -1 == sigaddset(&sigset,SIGSEGV) ){
    /* on error errno is set */
    perror(" SIGSEGV cannot be added to signal set \n ");
    exit(1);
  }

  if ( 0 > sigprocmask(SIG_UNBLOCK,&sigset,NULL)){
    fprintf(stderr," error while retreiving procmask \n");
    exit(1);
  }

  fprintf(stdout, " SIG_SEGV added to signal set \n");

  return;
}

static void 
signalHandler(int arg)
{
  fprintf(stdout ," I am in signal handler \n");
  longjmp(buf,1);             // jumps back to where setjmp was called - making setjmp now return 1
}


static void
throwSegmentation()
{
  int * ptr = 0x0;

  /* a leak of 10x4 = 40 bytes */
  void * leak = malloc(sizeof(int) * 10);

  /* now attempting to write on address 0x0 */
  *ptr = 0xdeadbeef;
}

static void
setUpAndThrowSeg()
{
 if ( 0 == setjmp(buf) ) {
    fprintf(stdout," jump set \n");
    throwSegmentation();
 }else {                    
    /* remove blocked SIGSEGV from procmask */
    fprintf(stdout," returned from segmentation\n");
    addSignalToMask();
 }
}

int main() {
  int count = 10;
  if (SIG_ERR == signal(SIGSEGV, signalHandler)){
    fprintf(stderr," signal handler not set up \n");
    exit(1);
  }

  while ( 0 != count --) {
    fprintf(stdout, " throwing again \n");
    setUpAndThrowSeg();
  }

  return 0;
}

Answer 1

使用库存malloc无法实现此目的。没有便携式工具可以跟踪自某个时间点以来已完成的分配。

你可以做的是这样的：创建一个包装malloc的函数，并将每个新分配的内存区域添加到链表中。当您需要清除内存泄漏时，请遍历此列表并free在那里找到的所有块。当然，这只有在你可以保证没有指向已分配区域的指针的情况下才有效。

重新设计控制流可能是一个好主意，因此在您想要中止控制的时间内不会发生内存分配。

ljmp和setjmp之间的内存泄漏

1 个答案: