C++程序  |  49行  |  1.48 KB

/*
 * Copyright (C) 2017  Red Hat, Inc.
 *
 * 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.
 */

 /*
  * Parse the ksm0* test options in funcion parse_ksm_options().
  */

#include "tst_test.h"

int merge_across_nodes;

int size = 128, num = 3, unit = 1;
char *opt_sizestr, *opt_numstr, *opt_unitstr;

struct tst_option ksm_options[] = {
	{"n:", &opt_numstr,  "-n       Number of processes"},
	{"s:", &opt_sizestr, "-s       Memory allocation size in MB"},
	{"u:", &opt_unitstr, "-u       Memory allocation unit in MB"},
	{NULL, NULL, NULL}
};

static inline void parse_ksm_options(char *str_size, int *size,
		char *str_num, int *num, char *str_unit, int *unit)
{
	if(tst_parse_int(str_size, size, 1, INT_MAX))
		tst_brk(TBROK, "Invalid size '%s'", str_size);

	if(tst_parse_int(str_num, num, 3, INT_MAX))
		tst_brk(TBROK, "Invalid num '%s'", str_num);

	if(tst_parse_int(str_unit, unit, 1, *size))
		tst_brk(TBROK, "Invalid unit '%s'", str_unit);
	if (*size % *unit != 0)
		tst_brk(TBROK,
				"the remainder of division of size by unit is "
				"not zero.");
}