xref: /third_party/musl/libc-test/src/functional/test-malloc-disable.c

/*
 * Copyright (C) 2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <stdio.h>
#include <malloc.h>
#include <stdlib.h>
#include <pthread.h>
#include <stdint.h>
#include <memory.h>
#include <unistd.h>

#include "test-malloc-api-common.h"

#define BARRIER_HEIGHT 2

pthread_barrier_t routine_disabled;
pthread_barrier_t routine_allocated;

const size_t SLEEP_TIME_SECONDS = 2;

void *disable_routine(void *vargp)
{
	malloc_disable();
	pthread_barrier_wait(&routine_disabled);
	sleep(SLEEP_TIME_SECONDS);
	malloc_enable();
	pthread_barrier_wait(&routine_allocated);
	return NULL;
}

int test_malloc_while_disabled(void)
{
	int ret = 0;
	pthread_barrier_init(&routine_disabled, NULL, BARRIER_HEIGHT);
	pthread_barrier_init(&routine_allocated, NULL, BARRIER_HEIGHT);
	pthread_t thread_id;
	pthread_create(&thread_id, NULL, disable_routine, NULL);
	pthread_barrier_wait(&routine_disabled);
	time_t start = time(0);
	int *x = malloc(sizeof(int));
	time_t end = time(0);
	pthread_barrier_wait(&routine_allocated);
	size_t seconds = end - start;
	if (seconds < SLEEP_TIME_SECONDS) {
		ret = -1;
	}
	free(x);
	pthread_join(thread_id, NULL);

	return ret;
}

int main()
{
	int ret = 0;

	ret = check_and_report("Testing malloc while disabled", test_malloc_while_disabled);

	return ret;
}