tests: add kasan test
-u: This option is used to test the use of freed memory -h: This option is used to test heap memory out-of-bounds access errors. -g: This option is used to test global variables for out-of-bounds access errors. -s: This option is used to test the Memory Stampede phenomenon. Signed-off-by: wangmingrong <wangmingrong@xiaomi.com>
This commit is contained in:
parent
8ad39537dd
commit
396290962f
10
testing/kasantest/Kconfig
Normal file
10
testing/kasantest/Kconfig
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
#
|
||||||
|
# For a description of the syntax of this configuration file,
|
||||||
|
# see the file kconfig-language.txt in the NuttX tools repository.
|
||||||
|
#
|
||||||
|
|
||||||
|
config TESTING_KASAN
|
||||||
|
tristate "KASan validity test"
|
||||||
|
default n
|
||||||
|
---help---
|
||||||
|
Enable the KASan validity test
|
23
testing/kasantest/Make.defs
Normal file
23
testing/kasantest/Make.defs
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
############################################################################
|
||||||
|
# apps/testing/kasantest/Make.defs
|
||||||
|
#
|
||||||
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
# contributor license agreements. See the NOTICE file distributed with
|
||||||
|
# this work for additional information regarding copyright ownership. The
|
||||||
|
# ASF licenses this file to you 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.
|
||||||
|
#
|
||||||
|
############################################################################
|
||||||
|
|
||||||
|
ifneq ($(CONFIG_TESTING_KASAN),)
|
||||||
|
CONFIGURED_APPS += $(APPDIR)/testing/kasantest
|
||||||
|
endif
|
32
testing/kasantest/Makefile
Normal file
32
testing/kasantest/Makefile
Normal file
@ -0,0 +1,32 @@
|
|||||||
|
############################################################################
|
||||||
|
# apps/testing/kasantest/Makefile
|
||||||
|
#
|
||||||
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
# contributor license agreements. See the NOTICE file distributed with
|
||||||
|
# this work for additional information regarding copyright ownership. The
|
||||||
|
# ASF licenses this file to you 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 $(APPDIR)/Make.defs
|
||||||
|
|
||||||
|
# KASan validity test
|
||||||
|
|
||||||
|
MAINSRC = kasantest.c
|
||||||
|
PROGNAME = kasantest
|
||||||
|
PRIORITY = SCHED_PRIORITY_DEFAULT
|
||||||
|
STACKSIZE = $(CONFIG_DEFAULT_TASK_STACKSIZE)
|
||||||
|
|
||||||
|
CFLAGS += -Wno-error -Wno-use-after-free
|
||||||
|
|
||||||
|
include $(APPDIR)/Application.mk
|
175
testing/kasantest/kasantest.c
Normal file
175
testing/kasantest/kasantest.c
Normal file
@ -0,0 +1,175 @@
|
|||||||
|
/****************************************************************************
|
||||||
|
* apps/testing/kasantest/kasantest.c
|
||||||
|
*
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
||||||
|
* contributor license agreements. See the NOTICE file distributed with
|
||||||
|
* this work for additional information regarding copyright ownership. The
|
||||||
|
* ASF licenses this file to you 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.
|
||||||
|
*
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Included Files
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
#include <nuttx/config.h>
|
||||||
|
#include <nuttx/mm/kasan.h>
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <syslog.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Pre-processor Definitions
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
#define KASAN_TEST_MEM_SIZE 128
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Private Data
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
static char g_kasan_test_buffer[KASAN_TEST_MEM_SIZE];
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Private Functions
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
static void kasan_test(char *p, size_t size)
|
||||||
|
{
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
for (i = 0; i < size + 64; i++)
|
||||||
|
{
|
||||||
|
syslog(LOG_SYSLOG,
|
||||||
|
"Access Buffer[%zu] : %zu address: %p", size, i, &p[i]);
|
||||||
|
p[i]++;
|
||||||
|
syslog(LOG_SYSLOG, "read: %02x -- Successful\n", p[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void kasan_test_use_after_free(void)
|
||||||
|
{
|
||||||
|
char *ptr = malloc(KASAN_TEST_MEM_SIZE);
|
||||||
|
|
||||||
|
if (ptr == NULL)
|
||||||
|
{
|
||||||
|
syslog(LOG_SYSLOG, "Failed to allocate memory\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
syslog(LOG_SYSLOG, "KASan test use after free\n");
|
||||||
|
strcpy(ptr, "kasan test use after free");
|
||||||
|
free(ptr);
|
||||||
|
printf("%s\n", ptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void kasan_test_heap_memory_out_of_bounds(char *str)
|
||||||
|
{
|
||||||
|
char *endptr;
|
||||||
|
size_t size;
|
||||||
|
char *ptr;
|
||||||
|
|
||||||
|
size = strtoul(str, &endptr, 0);
|
||||||
|
if (*endptr != '\0')
|
||||||
|
{
|
||||||
|
printf("Conversion failed: Not a valid integer.\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
ptr = zalloc(size);
|
||||||
|
if (ptr == NULL)
|
||||||
|
{
|
||||||
|
syslog(LOG_SYSLOG, "Failed to allocate memory\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
syslog(LOG_SYSLOG,
|
||||||
|
"KASan test accessing heap memory out of bounds completed\n");
|
||||||
|
kasan_test(ptr, size);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void kasan_test_global_variable_out_of_bounds(void)
|
||||||
|
{
|
||||||
|
syslog(LOG_SYSLOG,
|
||||||
|
"KASan test accessing global variable out of bounds\n");
|
||||||
|
kasan_test(g_kasan_test_buffer, KASAN_TEST_MEM_SIZE);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void *mm_stampede_thread(void *arg)
|
||||||
|
{
|
||||||
|
char *p = (char *)arg;
|
||||||
|
|
||||||
|
syslog(LOG_SYSLOG, "Child thread is running");
|
||||||
|
kasan_test(p, KASAN_TEST_MEM_SIZE);
|
||||||
|
pthread_exit(NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void kasan_test_memory_stampede(void)
|
||||||
|
{
|
||||||
|
pthread_t thread;
|
||||||
|
char array[KASAN_TEST_MEM_SIZE];
|
||||||
|
|
||||||
|
syslog(LOG_SYSLOG, "KASan test accessing memory stampede\n");
|
||||||
|
pthread_create(&thread, NULL, mm_stampede_thread, kasan_reset_tag(&array));
|
||||||
|
pthread_join(thread, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Public Functions
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
/****************************************************************************
|
||||||
|
* Name: kasantest_main
|
||||||
|
****************************************************************************/
|
||||||
|
|
||||||
|
int main(int argc, char *argv[])
|
||||||
|
{
|
||||||
|
/* NutttX cannot check the secondary release
|
||||||
|
* because the mm module has closed kasan instrumentation
|
||||||
|
*/
|
||||||
|
|
||||||
|
if (argc < 2)
|
||||||
|
{
|
||||||
|
printf("Usage: %s <test_option>\n", argv[0]);
|
||||||
|
printf("Available test options:\n");
|
||||||
|
printf(" -u : Test use after free\n");
|
||||||
|
printf(" -h <arg> : Test heap memory out of bounds (provide size)\n");
|
||||||
|
printf(" -g : Test global variable out of bounds\n");
|
||||||
|
printf(" -s : Test memory stampede\n");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
else if (strncmp(argv[1], "-u", 2) == 0)
|
||||||
|
{
|
||||||
|
kasan_test_use_after_free();
|
||||||
|
}
|
||||||
|
else if (strncmp(argv[1], "-h", 2) == 0 && argc == 3)
|
||||||
|
{
|
||||||
|
kasan_test_heap_memory_out_of_bounds(argv[2]);
|
||||||
|
}
|
||||||
|
else if (strncmp(argv[1], "-g", 2) == 0)
|
||||||
|
{
|
||||||
|
kasan_test_global_variable_out_of_bounds();
|
||||||
|
}
|
||||||
|
else if (strncmp(argv[1], "-s", 2) == 0)
|
||||||
|
{
|
||||||
|
kasan_test_memory_stampede();
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
printf("Unknown test option: %s\n", argv[1]);
|
||||||
|
}
|
||||||
|
|
||||||
|
syslog(LOG_SYSLOG, "KASan test failed, please check\n");
|
||||||
|
return 0;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user