Модульное тестирование для неудачного malloc()

Я увидел крутое решение этой проблемы, которое мне представил С. Пааволайнен. Идея состоит в том, чтобы переопределить стандартный malloc(), который вы можете сделать только в компоновщике, с помощью пользовательского распределителя, который

1. читает текущий стек выполнения потока, вызывающего malloc()
2. checks if the stack exists in a database that is stored on hard disk
   1. if the stack does not exist, adds the stack to the database and returns NULL
   2. если стек уже существует, обычно выделяет память и возвращает

Затем вы просто запускаете свой модульный тест много раз: эта система автоматически перебирает различные пути управления до malloc() отказа и намного эффективнее и надежнее, чем, например. случайное тестирование.

Я предлагаю создать специальную функцию для вашего специального кода malloc, который, как вы ожидаете, может дать сбой, и вы могли бы изящно его обработать. Например:

void* special_malloc(size_t bytes) {
  void* ptr = malloc(bytes);
  if(ptr == NULL) {
    /* Do something crafty */
  } else {
    return ptr;
  }
}

Затем вы можете провести модульное тестирование этого хитрого дела, передав некоторые неверные значения для байтов. Вы можете поместить это в отдельную библиотеку и создать фиктивную библиотеку, которая ведет себя специально для тестирования функций, вызывающих эту.

Это довольно грубо, но если вам действительно нужно модульное тестирование, вы можете сделать это с помощью #ifdefs:

thingy *my_thingy = malloc(sizeof(thingy));
#ifdef MALLOC_UNIT_TEST_1
my_thingy = NULL;
#endif
if (my_thingy == NULL) {
  fprintf(stderr, "We're so screwed!\n");
  exit(EXIT_FAILURE);
}

К сожалению, вам придется многое перекомпилировать с этим решением.

Если вы используете Linux, вы также можете рассмотреть возможность запуска кода в условиях нехватки памяти, используя ulimit, но будьте осторожны.

напишите свою собственную библиотеку, которая реализует malloc путем случайного сбоя или вызова реального malloc (статически связанного или явно dlopened)

затем LD_PRELOAD его

Во FreeBSD я однажды просто перегрузил модуль malloc.o библиотеки C (символы там были слабыми) и заменил реализацию malloc() на такую, которая контролировала вероятность сбоя. Поэтому я подключился статически и начал выполнять тестирование. srandom() завершил изображение контролируемой псевдослучайной последовательностью.

Также посмотрите здесь набор хороших инструментов, которые, по моему мнению, вам понадобятся. По крайней мере, они перегружают malloc() / free() для отслеживания утечек, так что кажется удобной точкой для добавления всего, что вы хотите.

Вы можете захватить malloc, используя некоторые определения и глобальные параметры для управления им... Это немного хакерски, но, похоже, работает.

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

#define malloc(x) fake_malloc(x)

struct {
  size_t last_request;
  int should_fail;
  void *(*real_malloc)(size_t);
} fake_malloc_params;

void *fake_malloc(size_t size) {
  fake_malloc_params.last_request = size;
  if (fake_malloc_params.should_fail) {
    return NULL;
  }
  return (fake_malloc_params.real_malloc)(size);;
}

int main(void) {
  fake_malloc_params.real_malloc = malloc;
  void *ptr = NULL;
  ptr = malloc(1);
  printf("last: %d\n", (int) fake_malloc_params.last_request);
  printf("ptr: 0x%p\n", ptr);
  return 0;
}