static void randomize_buf(void *buf, unsigned int size, int seed)
{
struct frand_state state;
init_rand_seed(&state, seed);
fill_random_buf(&state, buf, size);
}
void machdep()
{
long seed ;
static int first=1 ;
if( !first ) return ; else first = 0 ;
/*-- force use of mcw_malloc.c functions - 05 Nov 2001 --*/
#ifdef USING_MCW_MALLOC
if( AFNI_yesenv("AFNI_FORCE_MCW_MALLOC") ) enable_mcw_malloc();
#endif
/*-- disable mmap() in malloc() [21 Aug 2002: mostly] --*/
#if defined(LINUX) && defined(M_MMAP_MAX)
mallopt( M_MMAP_MAX , 1 ) ;
#endif
seed = (long)AFNI_numenv("AFNI_RANDOM_SEEDVAL") ;
init_rand_seed(seed) ;
be_quiet = AFNI_yesenv("AFNI_QUIET_STARTUP") ; /* 08 Dec 2010 */
if( AFNI_yesenv("AFNI_USE_FGETS") ) afni_fgets_setskip(1) ; /* 21 Dec 2011 */
AFNI_do_nothing() ; /* 02 Oct 2012 */
return ;
}
void init_rand()
{
// seed 0: random device
unsigned int uiSeed0 = 0;
try
{
std::random_device rnd;
if(rnd.entropy() == 0)
log_debug("Random seed entropy is zero!");
uiSeed0 = rnd();
}
catch(const std::exception& ex)
{
log_debug(ex.what());
uiSeed0 = 0;
}
// seed 1: time based
struct timeval timev;
gettimeofday(&timev, 0);
unsigned int uiSeed1 = timev.tv_sec ^ timev.tv_usec;
// total seed
unsigned int uiSeed = uiSeed0 ^ uiSeed1;
log_debug("Random seed: ", uiSeed0, ", time seed: ", uiSeed1, ", total seed: ", uiSeed, ".");
init_rand_seed(uiSeed);
}
int main(int argc, char **argv)
{
int c;
unsigned long checksum = 0;
char *str;
char *buf;
int length = 10;
u64 seed = 0;
int loop = 0;
int i;
while ((c = getopt(argc, argv, "l:c:s:h")) != -1) {
switch (c) {
case 'l':
length = atol(optarg);
break;
case 'c':
sscanf(optarg, "%li", &checksum);
loop = 1;
break;
case 's':
seed = atoll(optarg);
break;
case 'h':
print_usage(1);
case '?':
print_usage(255);
}
}
set_argv0(argv);
str = argv[optind];
if (!loop) {
if (check_argc_exact(argc - optind, 1))
print_usage(255);
printf("%12u - %s\n", crc32c(~1, str, strlen(str)), str);
return 0;
}
if (check_argc_exact(argc - optind, 0))
print_usage(255);
buf = malloc(length);
if (!buf)
return -ENOMEM;
if (seed)
init_rand_seed(seed);
while (1) {
for (i = 0; i < length; i++)
buf[i] = rand_range(94) + 33;
if (crc32c(~1, buf, length) == checksum)
printf("%12lu - %.*s\n", checksum, length, buf);
}
return 0;
}
void td_fill_rand_seeds(struct thread_data *td)
{
if (td->o.use_os_rand)
td_fill_rand_seeds_os(td);
else
td_fill_rand_seeds_internal(td);
init_rand_seed(&td->buf_state, td->rand_seeds[7]);
}
static void shared_rand_init(struct zipf_state *zs, unsigned long nranges,
unsigned int seed)
{
memset(zs, 0, sizeof(*zs));
zs->nranges = nranges;
init_rand_seed(&zs->rand, seed, 0);
zs->rand_off = __rand(&zs->rand);
}
void gauss_init(struct gauss_state *gs, unsigned long nranges, double dev,
unsigned int seed)
{
memset(gs, 0, sizeof(*gs));
init_rand_seed(&gs->r, seed, 0);
gs->nranges = nranges;
if (dev != 0.0) {
gs->stddev = ceil((double) (nranges * 100.0) / dev);
if (gs->stddev > nranges / 2)
gs->stddev = nranges / 2;
}
}
static void td_fill_rand_seeds_internal(struct thread_data *td)
{
init_rand_seed(&td->__bsrange_state, td->rand_seeds[FIO_RAND_BS_OFF]);
init_rand_seed(&td->__verify_state, td->rand_seeds[FIO_RAND_VER_OFF]);
init_rand_seed(&td->__rwmix_state, td->rand_seeds[FIO_RAND_MIX_OFF]);
if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
init_rand_seed(&td->__next_file_state, td->rand_seeds[FIO_RAND_FILE_OFF]);
init_rand_seed(&td->__file_size_state, td->rand_seeds[FIO_RAND_FILE_SIZE_OFF]);
init_rand_seed(&td->__trim_state, td->rand_seeds[FIO_RAND_TRIM_OFF]);
if (!td_random(td))
return;
if (td->o.rand_repeatable)
td->rand_seeds[FIO_RAND_BLOCK_OFF] = FIO_RANDSEED * td->thread_number;
init_rand_seed(&td->__random_state, td->rand_seeds[FIO_RAND_BLOCK_OFF]);
init_rand_seed(&td->__seq_rand_state[DDIR_READ], td->rand_seeds[FIO_RAND_SEQ_RAND_READ_OFF]);
init_rand_seed(&td->__seq_rand_state[DDIR_WRITE], td->rand_seeds[FIO_RAND_SEQ_RAND_WRITE_OFF]);
init_rand_seed(&td->__seq_rand_state[DDIR_TRIM], td->rand_seeds[FIO_RAND_SEQ_RAND_TRIM_OFF]);
}
void td_fill_rand_seeds(struct thread_data *td)
{
if (td->o.allrand_repeatable) {
for (int i = 0; i < FIO_RAND_NR_OFFS; i++)
td->rand_seeds[i] = FIO_RANDSEED * td->thread_number
+ i;
}
if (td->o.use_os_rand)
td_fill_rand_seeds_os(td);
else
td_fill_rand_seeds_internal(td);
init_rand_seed(&td->buf_state, td->rand_seeds[FIO_RAND_BUF_OFF]);
}
static void td_fill_rand_seeds_internal(struct thread_data *td)
{
init_rand_seed(&td->__bsrange_state, td->rand_seeds[0]);
init_rand_seed(&td->__verify_state, td->rand_seeds[1]);
init_rand_seed(&td->__rwmix_state, td->rand_seeds[2]);
if (td->o.file_service_type == FIO_FSERVICE_RANDOM)
init_rand_seed(&td->__next_file_state, td->rand_seeds[3]);
init_rand_seed(&td->__file_size_state, td->rand_seeds[5]);
init_rand_seed(&td->__trim_state, td->rand_seeds[6]);
if (!td_random(td))
return;
if (td->o.rand_repeatable)
td->rand_seeds[4] = FIO_RANDSEED * td->thread_number;
init_rand_seed(&td->__random_state, td->rand_seeds[4]);
}
int main(int argc, char **argv)
{
(void)argc;
(void)argv;
if (init_events_init())
return 1;
start_logging();
if (init_rand_seed())
return 1;
if (ensure_single_instance())
return 1;
if (conf_load())
return 1;
if (cl_init())
return 1;
if (start_task_queue())
return 1;
if (start_state_manager())
return 1;
if (start_debug_console())
return 1;
if (enqueue_task(main2, NULL))
return 1;
if (start_rpc_svc())
return 1;
return 0;
}
int main(int argc, char *argv[]) {
int r = 255;
int wrote_pid_file = 0;
avahi_set_log_function(log_function);
init_rand_seed();
avahi_server_config_init(&config.server_config);
config.command = DAEMON_RUN;
config.daemonize = 0;
config.config_file = NULL;
#ifdef HAVE_DBUS
config.enable_dbus = 1;
config.fail_on_missing_dbus = 1;
config.n_clients_max = 0;
config.n_objects_per_client_max = 0;
config.n_entries_per_entry_group_max = 0;
#endif
config.drop_root = 1;
config.set_rlimits = 1;
#ifdef ENABLE_CHROOT
config.use_chroot = 1;
#endif
config.modify_proc_title = 1;
config.disable_user_service_publishing = 0;
config.publish_dns_servers = NULL;
config.publish_resolv_conf = 0;
config.use_syslog = 0;
config.debug = 0;
config.rlimit_as_set = 0;
config.rlimit_core_set = 0;
config.rlimit_data_set = 0;
config.rlimit_fsize_set = 0;
config.rlimit_nofile_set = 0;
config.rlimit_stack_set = 0;
#ifdef RLIMIT_NPROC
config.rlimit_nproc_set = 0;
#endif
if ((argv0 = strrchr(argv[0], '/')))
argv0 = avahi_strdup(argv0 + 1);
else
argv0 = avahi_strdup(argv[0]);
daemon_pid_file_ident = (const char *) argv0;
daemon_log_ident = (char*) argv0;
daemon_pid_file_proc = pid_file_proc;
if (parse_command_line(&config, argc, argv) < 0)
goto finish;
if (config.modify_proc_title)
avahi_init_proc_title(argc, argv);
#ifdef ENABLE_CHROOT
config.use_chroot = config.use_chroot && config.drop_root;
#endif
if (config.command == DAEMON_HELP) {
help(stdout);
r = 0;
} else if (config.command == DAEMON_VERSION) {
printf("%s "PACKAGE_VERSION"\n", argv0);
r = 0;
} else if (config.command == DAEMON_KILL) {
if (daemon_pid_file_kill_wait(SIGTERM, 5) < 0) {
avahi_log_warn("Failed to kill daemon: %s", strerror(errno));
goto finish;
}
r = 0;
} else if (config.command == DAEMON_RELOAD) {
if (daemon_pid_file_kill(SIGHUP) < 0) {
avahi_log_warn("Failed to kill daemon: %s", strerror(errno));
goto finish;
}
r = 0;
} else if (config.command == DAEMON_CHECK)
r = (daemon_pid_file_is_running() >= 0) ? 0 : 1;
else if (config.command == DAEMON_RUN) {
pid_t pid;
if (getuid() != 0 && config.drop_root) {
avahi_log_error("This program is intended to be run as root.");
goto finish;
}
if ((pid = daemon_pid_file_is_running()) >= 0) {
avahi_log_error("Daemon already running on PID %u", pid);
goto finish;
}
if (load_config_file(&config) < 0)
goto finish;
if (config.daemonize) {
daemon_retval_init();
if ((pid = daemon_fork()) < 0)
goto finish;
else if (pid != 0) {
int ret;
/** Parent **/
if ((ret = daemon_retval_wait(20)) < 0) {
avahi_log_error("Could not receive return value from daemon process.");
goto finish;
}
r = ret;
goto finish;
}
/* Child */
}
if (config.use_syslog || config.daemonize)
daemon_log_use = DAEMON_LOG_SYSLOG;
if (sd_listen_fds(0) <= 0)
if (daemon_close_all(-1) < 0)
avahi_log_warn("Failed to close all remaining file descriptors: %s", strerror(errno));
daemon_reset_sigs(-1);
daemon_unblock_sigs(-1);
if (make_runtime_dir() < 0)
goto finish;
if (config.drop_root) {
#ifdef ENABLE_CHROOT
if (config.use_chroot)
if (avahi_caps_reduce() < 0)
goto finish;
#endif
if (drop_root() < 0)
goto finish;
#ifdef ENABLE_CHROOT
if (config.use_chroot)
if (avahi_caps_reduce2() < 0)
goto finish;
#endif
}
if (daemon_pid_file_create() < 0) {
avahi_log_error("Failed to create PID file: %s", strerror(errno));
if (config.daemonize)
daemon_retval_send(1);
goto finish;
} else
wrote_pid_file = 1;
if (config.set_rlimits)
enforce_rlimits();
chdir("/");
#ifdef ENABLE_CHROOT
if (config.drop_root && config.use_chroot)
if (avahi_chroot_helper_start(argv0) < 0) {
avahi_log_error("failed to start chroot() helper daemon.");
goto finish;
}
#endif
avahi_log_info("%s "PACKAGE_VERSION" starting up.", argv0);
sd_notifyf(0, "STATUS=%s "PACKAGE_VERSION" starting up.", argv0);
avahi_set_proc_title(argv0, "%s: starting up", argv0);
if (run_server(&config) == 0)
r = 0;
avahi_log_info("%s "PACKAGE_VERSION" exiting.", argv0);
sd_notifyf(0, "STATUS=%s "PACKAGE_VERSION" exiting.", argv0);
}
finish:
if (config.daemonize)
daemon_retval_done();
avahi_server_config_free(&config.server_config);
avahi_free(config.config_file);
avahi_strfreev(config.publish_dns_servers);
avahi_strfreev(resolv_conf_name_servers);
avahi_strfreev(resolv_conf_search_domains);
if (wrote_pid_file) {
#ifdef ENABLE_CHROOT
avahi_chroot_helper_unlink(pid_file_proc());
#else
daemon_pid_file_remove();
#endif
}
#ifdef ENABLE_CHROOT
avahi_chroot_helper_shutdown();
#endif
avahi_free(argv0);
return r;
}
