#include <ctype.h>
#include <inttypes.h>
+#include <limits.h>
+#include <pwd.h>
#include <strings.h>
+#include <sys/types.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+
#include "ms_setting.h"
#include "ms_conn.h"
#define RAND_CHAR_SIZE (10 * 1024 * 1024) /* 10M character table */
#define RESERVED_RAND_CHAR_SIZE (2 * 1024 * 1024) /* reserved 2M to avoid pointer sloping over */
+#define DEFAULT_CONFIG_NAME ".memslap.cnf"
+
#define DEFAULT_THREADS_NUM 1 /* default start one thread */
#define DEFAULT_CONNS_NUM 16 /* default each thread with 16 connections */
#define DEFAULT_EXE_NUM 0 /* default execute number is 0 */
for (;;)
{
- char i;
-
- i = getc(fp);
+ int i= getc(fp);
if (i == EOF)
{
result = -1;
*line_size= needed;
}
- (*line)[cur_len]= i;
+ (*line)[cur_len]= (char)i;
cur_len++;
if (i == delim)
{
srvs= (ms_mcd_server_t *)realloc(
ms_setting.servers,
- (uint64_t)ms_setting.total_srv_cnt
+ (size_t)ms_setting.total_srv_cnt
* sizeof(ms_mcd_server_t) * 2);
if (srvs == NULL)
{
*/
static void ms_no_config_file()
{
- FILE *fd= fopen("config", "w+");
+ char userpath[PATH_MAX];
+ struct passwd *usr= NULL;
+ FILE *fd;
+
+ usr= getpwuid(getuid());
+ snprintf(userpath, PATH_MAX, "%s/%s", usr->pw_dir, DEFAULT_CONFIG_NAME);
+
+ if (access (userpath, F_OK | R_OK) == 0)
+ goto exit;
+
+ fd= fopen(userpath, "w+");
+
+ if (fd == NULL)
+ {
+ fprintf(stderr, "Could not create default configure file %s\n", userpath);
+ perror(strerror(errno));
+ exit(1);
+ }
fprintf(fd, "%s", DEFAULT_CONGIF_STR);
fclose(fd);
- ms_setting.cfg_file= strdup("config");
+exit:
+ ms_setting.cfg_file= strdup(userpath);
} /* ms_no_config_file */
int end_of_file= 0;
ms_key_distr_t *key_distr= NULL;
ms_value_distr_t *val_distr= NULL;
- bool no_cfg= false;
if (cfg_file == NULL)
{
ms_no_config_file();
cfg_file= ms_setting.cfg_file;
- no_cfg= true;
}
/*read key value configure file*/
if ((((nread= getline(&line, &read_len, f)) == 1)
|| ! ms_read_is_data(line, nread)) && (nread != EOF)) /* bypass blank line */
continue;
+
if (nread == EOF)
{
fprintf(stderr, "Bad configuration file, no configuration find.\n");
if (nread != EOF)
{
- if (sscanf(line, "%lu %lu %lf ", &start_len,
+ if (sscanf(line, "%zu %zu %lf ", &start_len,
&end_len, &proportion) != 3)
{
conf_type= ms_get_conf_type(line);
{
key_distr= (ms_key_distr_t *)realloc(
ms_setting.key_distr,
- (uint64_t)ms_setting.
+ (size_t)ms_setting.
total_key_rng_cnt * sizeof(ms_key_distr_t) * 2);
if (key_distr == NULL)
{
if (nread != EOF)
{
- if (sscanf(line, "%lu %lu %lf %lu", &start_len, &end_len,
+ if (sscanf(line, "%zu %zu %lf %zu", &start_len, &end_len,
&proportion, &frequence) != 3)
{
conf_type= ms_get_conf_type(line);
{
val_distr= (ms_value_distr_t *)realloc(
ms_setting.value_distr,
- (uint64_t)ms_setting.
+ (size_t)ms_setting.
total_val_rng_cnt * sizeof(ms_value_distr_t) * 2);
if (val_distr == NULL)
{
fclose(f);
- if (no_cfg)
- {
- remove(ms_setting.cfg_file);
- }
-
if (line != NULL)
+ {
free(line);
+ }
} /* ms_parse_cfg_file */
ms_setting.run_time= DEFAULT_RUN_TIME;
ms_setting.total_srv_cnt= MCD_SRVS_NUM_INIT;
ms_setting.servers= (ms_mcd_server_t *)malloc(
- (uint64_t)ms_setting.total_srv_cnt
+ (size_t)ms_setting.total_srv_cnt
* sizeof(ms_mcd_server_t));
if (ms_setting.servers == NULL)
{
static void ms_setting_slapmode_init_post()
{
ms_setting.total_key_rng_cnt= KEY_RANGE_COUNT_INIT;
- ms_setting.key_distr= (ms_key_distr_t *)malloc(
- (uint64_t)ms_setting.total_key_rng_cnt
- * sizeof(ms_key_distr_t));
+ ms_setting.key_distr=
+ (ms_key_distr_t *)malloc((size_t)ms_setting.total_key_rng_cnt * sizeof(ms_key_distr_t));
+
if (ms_setting.key_distr == NULL)
{
fprintf(stderr, "Can't allocate key distribution structure.\n");
}
ms_setting.total_val_rng_cnt= VALUE_RANGE_COUNT_INIT;
- ms_setting.value_distr= (ms_value_distr_t *)malloc(
- (uint64_t)ms_setting.total_val_rng_cnt
- * sizeof(
- ms_value_distr_t));
+
+ ms_setting.value_distr=
+ (ms_value_distr_t *)malloc((size_t)ms_setting.total_val_rng_cnt * sizeof( ms_value_distr_t));
+
if (ms_setting.value_distr == NULL)
{
fprintf(stderr, "Can't allocate value distribution structure.\n");