/***************************************************************************
 *   User front end for using huge pages Copyright (C) 2008, IBM           *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the Lesser GNU General Public License as        *
 *   published by the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.                   *
 *                                                                         *
 *   You should have received a copy of the Lesser GNU General Public      *
 *   License along with this program; if not, write to the                 *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

/*
 * pagesize exposes the available and hardware supported page sizes on
 * the system.
 *
 * This program should be treated as an ABI for using libhugetlbfs.
 */

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <limits.h>

#define _GNU_SOURCE /* for getopt_long */
#include <unistd.h>
#include <getopt.h>

#define REPORT_UTIL "pagesize"
#include "libhugetlbfs_internal.h"
#include "hugetlbfs.h"

extern int errno;
extern int optind;
extern char *optarg;

#define OPTION(opts, text)	fprintf(stderr, " %-25s  %s\n", opts, text)
#define CONT(text) 		fprintf(stderr, " %-25s  %s\n", "", text)

void print_usage()
{
	fprintf(stderr, "pagesize [options] target\n");
	fprintf(stderr, "options:\n");

	OPTION("--help, -h", "Prints this message");

	OPTION("--all, -a", "show all supported page sizes");
	OPTION("--huge-only, -H", "show only huge page sizes");
}

static int cmpsizes(const void *p1, const void *p2)
{
	return *((long *)p1) > *((long *)p2);
}

#define MAX_PAGESIZES 32

int main(int argc, char** argv)
{
	int opt_all = 0;
	int opt_huge = 0;

	char opts[] = "+haH";
	int ret = 0, index = 0;
	struct option long_opts[] = {
		{"all",       no_argument, NULL, 'a'},
		{"huge-only", no_argument, NULL, 'H'},

		{0},
	};

	long pagesizes[MAX_PAGESIZES];
	int i;

	hugetlbfs_setup_debug();

	while (ret != -1) {
		ret = getopt_long(argc, argv, opts, long_opts, &index);
		switch (ret) {
		case '?':
			print_usage();
			exit(EXIT_FAILURE);

		case 'h':
			print_usage();
			exit(EXIT_SUCCESS);

		case 'a':
			opt_all = 1;
			INFO("selecting all page sizes\n");
			break;

		case 'H':
			opt_huge = 1;
			opt_all = 1;
			INFO("selecting only huge page sizes\n");
			break;

		case -1:
			break;

		default:
			WARNING("unparsed option %08x\n", ret);
			ret = -1;
			break;
		}
	}
	index = optind;
	if ((argc - index) != 0) {
		print_usage();
		exit(EXIT_FAILURE);
	}

	if (!opt_all) {
		pagesizes[0] = sysconf(_SC_PAGESIZE);
		ret = 1;
	} else if (opt_huge)
		ret = gethugepagesizes(pagesizes, MAX_PAGESIZES);
	else
		ret = getpagesizes(pagesizes, MAX_PAGESIZES);
	if (ret < 0) {
		ERROR("failed to get list of supported page sizes\n");
		exit(EXIT_FAILURE);
	}

	qsort(pagesizes, ret, sizeof(long), cmpsizes);
	for (i = 0; i < ret; i++) {
		printf("%ld\n", pagesizes[i]);
	}

	exit(EXIT_SUCCESS);
}