// SPDX-License-Identifier: GPL-2.0
/*
 * hugetlb-shm:
 *
 * Example of using huge page memory in a user application using Sys V shared
 * memory system calls.  In this example the app is requesting 256MB of
 * memory that is backed by huge pages.  The application uses the flag
 * SHM_HUGETLB in the shmget system call to inform the kernel that it is
 * requesting huge pages.
 *
 * Note: The default shared memory limit is quite low on many kernels,
 * you may need to increase it via:
 *
 * echo 268435456 > /proc/sys/kernel/shmmax
 *
 * This will increase the maximum size per shared memory segment to 256MB.
 * The other limit that you will hit eventually is shmall which is the
 * total amount of shared memory in pages. To set it to 16GB on a system
 * with a 4kB pagesize do:
 *
 * echo 4194304 > /proc/sys/kernel/shmall
 */

#include <stdlib.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <sys/mman.h>

#include "vm_util.h"
#include "hugepage_settings.h"

#define LENGTH (256UL*1024*1024)

static void prepare(void)
{
        unsigned long length, hugepage_size, nr;

        hugepage_size = default_huge_page_size();
        if (!hugepage_size)
                ksft_exit_skip("Unable to determine huge page size\n");

        length = (LENGTH + hugepage_size - 1) & ~(hugepage_size - 1);
        nr = length / hugepage_size;

        if (!hugetlb_setup_default(nr))
                ksft_exit_skip("Not enough free huge pages\n");

        shm_limits_prepare(length);
}

int main(void)
{
        int shmid;
        unsigned long i;
        char *shmaddr;

        ksft_print_header();
        ksft_set_plan(1);

        prepare();

        shmid = shmget(2, LENGTH, SHM_HUGETLB | IPC_CREAT | SHM_R | SHM_W);
        if (shmid < 0)
                ksft_exit_fail_perror("shmget");

        ksft_print_msg("shmid: 0x%x\n", shmid);

        shmaddr = shmat(shmid, NULL, 0);
        if (shmaddr == (char *)-1) {
                ksft_perror("Shared memory attach failure");
                shmctl(shmid, IPC_RMID, NULL);
                ksft_exit_fail();
        }
        ksft_print_msg("shmaddr: %p\n", shmaddr);

        ksft_print_msg("Starting the writes:\n");
        for (i = 0; i < LENGTH; i++)
                shmaddr[i] = (char)(i);

        ksft_print_msg("Starting the Check...");
        for (i = 0; i < LENGTH; i++)
                if (shmaddr[i] != (char)i)
                        ksft_exit_fail_msg("Data mismatch at index %lu\n", i);
        ksft_print_msg("Done.\n");

        if (shmdt((const void *)shmaddr) != 0) {
                ksft_perror("Detach failure");
                shmctl(shmid, IPC_RMID, NULL);
                ksft_exit_fail();
        }

        shmctl(shmid, IPC_RMID, NULL);

        ksft_test_result_pass("hugepage using SysV shmget/shmat\n");
        ksft_finished();
}

SHM_LIMITS_RESTORE()