/* * Copyright (c) Linux Test Project, 2014-2017 * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See * the GNU General Public License for more details. */ /* * DESCRIPTION * hugeshmat04 - test for hugepage leak inspection. * * It is a regression test for shared hugepage leak, when over 1GB * shered memory was alocated in hugepage, the hugepage is not released * though process finished. * * You need more than 2GB memory in test job * * HISTORY * 05/2014 - Written by Fujistu Corp. * 12/2014 - Port to LTP by Li Wang. * * RESTRICTIONS * test must be run at root */ #include "hugetlb.h" #define SIZE (1024 * 1024 * 1024) #define BOUNDARY (1024 * 1024 * 1024) static long huge_free; static long huge_free2; static long hugepages; static long orig_shmmax, new_shmmax; static void shared_hugepage(void); static void test_hugeshmat(unsigned int i LTP_ATTRIBUTE_UNUSED) { huge_free = SAFE_READ_MEMINFO("HugePages_Free:"); shared_hugepage(); huge_free2 = SAFE_READ_MEMINFO("HugePages_Free:"); if (huge_free2 != huge_free) tst_brk(TFAIL, "Test failed. Hugepage leak inspection."); else tst_res(TPASS, "No regression found."); } static void shared_hugepage(void) { pid_t pid; int status, shmid; size_t size = (size_t)SIZE; void *buf; shmid = shmget(IPC_PRIVATE, size, SHM_HUGETLB | IPC_CREAT | 0777); if (shmid < 0) tst_brk(TBROK | TERRNO, "shmget"); buf = shmat(shmid, (void *)BOUNDARY, SHM_RND | 0777); if (buf == (void *)-1) { shmctl(shmid, IPC_RMID, NULL); tst_brk(TBROK | TERRNO, "shmat"); } memset(buf, 2, size); pid = SAFE_FORK(); if (pid == 0) exit(1); wait(&status); shmdt(buf); shmctl(shmid, IPC_RMID, NULL); } static void setup(void) { long mem_total, hpage_size, orig_hugepages; orig_hugepages = save_nr_hugepages(); mem_total = SAFE_READ_MEMINFO("MemTotal:"); SAFE_FILE_SCANF(PATH_SHMMAX, "%ld", &orig_shmmax); SAFE_FILE_PRINTF(PATH_SHMMAX, "%ld", (long)SIZE); SAFE_FILE_SCANF(PATH_SHMMAX, "%ld", &new_shmmax); if (mem_total < 2L*1024*1024) tst_brk(TCONF, "Needed > 2GB RAM, have: %ld", mem_total); if (new_shmmax < SIZE) tst_brk(TCONF, "shmmax too low, have: %ld", new_shmmax); hpage_size = SAFE_READ_MEMINFO("Hugepagesize:") * 1024; hugepages = (orig_hugepages * hpage_size + SIZE) / hpage_size; set_sys_tune("nr_hugepages", hugepages, 1); } static void cleanup(void) { restore_nr_hugepages(); SAFE_FILE_PRINTF(PATH_SHMMAX, "%ld", orig_shmmax); } static struct tst_test test = { .needs_root = 1, .forks_child = 1, .needs_tmpdir = 1, .tcnt = 3, .test = test_hugeshmat, .setup = setup, .cleanup = cleanup, };