Tanguy Pruvot
10 years ago
6 changed files with 396 additions and 344 deletions
@ -0,0 +1,357 @@
@@ -0,0 +1,357 @@
|
||||
/**
|
||||
* Functions which handle multiple pools data |
||||
*/ |
||||
|
||||
#include <stdio.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
#include <unistd.h> |
||||
|
||||
#include "miner.h" |
||||
#include "compat.h" |
||||
|
||||
// to move in miner.h
|
||||
extern bool allow_gbt; |
||||
extern bool allow_mininginfo; |
||||
extern bool check_dups; |
||||
|
||||
extern double opt_max_diff; |
||||
extern double opt_max_rate; |
||||
extern int opt_scantime; |
||||
extern int opt_time_limit; |
||||
|
||||
extern char* rpc_url; |
||||
extern char* rpc_user; |
||||
extern char* rpc_pass; |
||||
extern char* short_url; |
||||
|
||||
extern struct work _ALIGN(64) g_work; |
||||
extern struct stratum_ctx stratum; |
||||
extern pthread_mutex_t stratum_work_lock; |
||||
extern bool get_work(struct thr_info *thr, struct work *work); |
||||
extern bool stratum_need_reset; |
||||
extern time_t firstwork_time; |
||||
|
||||
extern volatile time_t g_work_time; |
||||
extern volatile int pool_switch_count; |
||||
extern volatile bool pool_is_switching; |
||||
|
||||
extern struct option options[]; |
||||
|
||||
#define CFG_NULL 0 |
||||
#define CFG_POOL 1 |
||||
struct opt_config_array { |
||||
int cat; |
||||
const char *name; // json key
|
||||
const char *longname; // global opt name if different
|
||||
} cfg_array_keys[] = { |
||||
{ CFG_POOL, "url", NULL }, /* let this key first, increment pools */ |
||||
{ CFG_POOL, "user", NULL }, |
||||
{ CFG_POOL, "pass", NULL }, |
||||
{ CFG_POOL, "userpass", NULL }, |
||||
{ CFG_POOL, "name", "pool-name" }, |
||||
{ CFG_POOL, "scantime", "pool-scantime" }, |
||||
{ CFG_POOL, "max-diff", "pool-max-diff" }, |
||||
{ CFG_POOL, "max-rate", "pool-max-rate" }, |
||||
{ CFG_POOL, "removed", "pool-removed" }, |
||||
{ CFG_POOL, "disabled", "pool-removed" }, // sample alias
|
||||
{ CFG_POOL, "time-limit", "pool-time-limit" }, |
||||
{ CFG_NULL, NULL, NULL } |
||||
}; |
||||
|
||||
// store current credentials in pools container
|
||||
void pool_set_creds(int pooln) |
||||
{ |
||||
struct pool_infos *p = &pools[pooln]; |
||||
|
||||
snprintf(p->url, sizeof(p->url), "%s", rpc_url); |
||||
snprintf(p->short_url, sizeof(p->short_url), "%s", short_url); |
||||
snprintf(p->user, sizeof(p->user), "%s", rpc_user); |
||||
snprintf(p->pass, sizeof(p->pass), "%s", rpc_pass); |
||||
|
||||
if (!(p->status & POOL_ST_DEFINED)) { |
||||
p->id = pooln; |
||||
p->status |= POOL_ST_DEFINED; |
||||
// init pool options as "unset"
|
||||
// until cmdline is not fully parsed...
|
||||
p->max_diff = -1.; |
||||
p->max_rate = -1.; |
||||
p->scantime = -1; |
||||
p->time_limit = -1; |
||||
|
||||
p->allow_mininginfo = allow_mininginfo; |
||||
p->allow_gbt = allow_gbt; |
||||
p->check_dups = check_dups; |
||||
|
||||
p->status |= POOL_ST_DEFINED; |
||||
} |
||||
|
||||
if (strlen(rpc_url)) { |
||||
if (!strncasecmp(rpc_url, "stratum", 7)) |
||||
p->type = POOL_STRATUM; |
||||
else /* if (!strncasecmp(rpc_url, "http", 4)) */ |
||||
p->type = POOL_GETWORK; // todo: or longpoll
|
||||
p->status |= POOL_ST_VALID; |
||||
} |
||||
} |
||||
|
||||
// fill the unset pools options with cmdline ones
|
||||
void pool_init_defaults() |
||||
{ |
||||
struct pool_infos *p; |
||||
for (int i=0; i<num_pools; i++) { |
||||
p = &pools[i]; |
||||
if (p->max_diff == -1.) p->max_diff = opt_max_diff; |
||||
if (p->max_rate == -1.) p->max_rate = opt_max_rate; |
||||
if (p->scantime == -1) p->scantime = opt_scantime; |
||||
if (p->time_limit == -1) p->time_limit = opt_time_limit; |
||||
} |
||||
} |
||||
|
||||
// attributes only set by a json pools config
|
||||
void pool_set_attr(int pooln, const char* key, char* arg) |
||||
{ |
||||
struct pool_infos *p = &pools[pooln]; |
||||
|
||||
if (!strcasecmp(key, "name")) { |
||||
snprintf(p->name, sizeof(p->name), "%s", arg); |
||||
return; |
||||
} |
||||
if (!strcasecmp(key, "scantime")) { |
||||
p->scantime = atoi(arg); |
||||
return; |
||||
} |
||||
if (!strcasecmp(key, "max-diff")) { |
||||
p->max_diff = atof(arg); |
||||
return; |
||||
} |
||||
if (!strcasecmp(key, "max-rate")) { |
||||
p->max_rate = atof(arg); |
||||
return; |
||||
} |
||||
if (!strcasecmp(key, "time-limit")) { |
||||
p->time_limit = atoi(arg); |
||||
return; |
||||
} |
||||
if (!strcasecmp(key, "removed")) { |
||||
int removed = atoi(arg); |
||||
if (removed) { |
||||
p->status |= POOL_ST_REMOVED; |
||||
} |
||||
return; |
||||
} |
||||
} |
||||
|
||||
// pool switching code
|
||||
bool pool_switch(int pooln) |
||||
{ |
||||
int prevn = cur_pooln; |
||||
struct pool_infos *prev = &pools[cur_pooln]; |
||||
struct pool_infos* p = NULL; |
||||
|
||||
// save prev stratum connection infos (struct)
|
||||
if (prev->type & POOL_STRATUM) { |
||||
// may not be the right moment to free,
|
||||
// to check if required on submit...
|
||||
stratum_free_job(&stratum); |
||||
prev->stratum = stratum; |
||||
} |
||||
|
||||
if (pooln < num_pools) { |
||||
cur_pooln = pooln; |
||||
p = &pools[cur_pooln]; |
||||
} else { |
||||
applog(LOG_ERR, "Switch to inexistant pool %d!", pooln); |
||||
return false; |
||||
} |
||||
|
||||
// save global attributes
|
||||
prev->allow_mininginfo = allow_mininginfo; |
||||
prev->allow_gbt = allow_gbt; |
||||
prev->check_dups = check_dups; |
||||
|
||||
pthread_mutex_lock(&stratum_work_lock); |
||||
|
||||
free(rpc_user); rpc_user = strdup(p->user); |
||||
free(rpc_pass); rpc_pass = strdup(p->pass); |
||||
free(rpc_url); rpc_url = strdup(p->url); |
||||
|
||||
short_url = p->short_url; // just a pointer, no alloc
|
||||
|
||||
opt_scantime = p->scantime; |
||||
opt_max_diff = p->max_diff; |
||||
opt_max_rate = p->max_rate; |
||||
opt_time_limit = p->time_limit; |
||||
|
||||
want_stratum = have_stratum = (p->type & POOL_STRATUM) != 0; |
||||
|
||||
pthread_mutex_unlock(&stratum_work_lock); |
||||
|
||||
if (prevn != cur_pooln) { |
||||
|
||||
pool_switch_count++; |
||||
g_work_time = 0; |
||||
g_work.data[0] = 0; |
||||
pool_is_switching = true; |
||||
stratum_need_reset = true; |
||||
// used to get the pool uptime
|
||||
firstwork_time = time(NULL); |
||||
restart_threads(); |
||||
|
||||
// restore flags
|
||||
allow_gbt = p->allow_gbt; |
||||
allow_mininginfo = p->allow_mininginfo; |
||||
check_dups = p->check_dups; |
||||
|
||||
if (want_stratum) { |
||||
|
||||
// temporary... until stratum code cleanup
|
||||
stratum = p->stratum; |
||||
stratum.pooln = cur_pooln; |
||||
|
||||
// unlock the stratum thread
|
||||
tq_push(thr_info[stratum_thr_id].q, strdup(rpc_url)); |
||||
applog(LOG_BLUE, "Switch to stratum pool %d: %s", cur_pooln, |
||||
strlen(p->name) ? p->name : p->short_url); |
||||
} else { |
||||
applog(LOG_BLUE, "Switch to pool %d: %s", cur_pooln, |
||||
strlen(p->name) ? p->name : p->short_url); |
||||
} |
||||
|
||||
// will unlock the longpoll thread on /LP url receive
|
||||
want_longpoll = (p->type & POOL_LONGPOLL) || !(p->type & POOL_STRATUM); |
||||
if (want_longpoll) { |
||||
pthread_mutex_lock(&stratum_work_lock); |
||||
// will issue a lp_url request to unlock the longpoll thread
|
||||
have_longpoll = false; |
||||
get_work(&thr_info[0], &g_work); |
||||
pthread_mutex_unlock(&stratum_work_lock); |
||||
} |
||||
|
||||
} |
||||
return true; |
||||
} |
||||
|
||||
// search available pool
|
||||
int pool_get_first_valid(int startfrom) |
||||
{ |
||||
int next = 0; |
||||
struct pool_infos *p; |
||||
for (int i=0; i<num_pools; i++) { |
||||
int pooln = (startfrom + i) % num_pools; |
||||
p = &pools[pooln]; |
||||
if (!(p->status & POOL_ST_VALID)) |
||||
continue; |
||||
if (p->status & (POOL_ST_DISABLED | POOL_ST_REMOVED)) |
||||
continue; |
||||
next = pooln; |
||||
break; |
||||
} |
||||
return next; |
||||
} |
||||
|
||||
// switch to next available pool
|
||||
bool pool_switch_next() |
||||
{ |
||||
if (num_pools > 1) { |
||||
int pooln = pool_get_first_valid(cur_pooln+1); |
||||
return pool_switch(pooln); |
||||
} else { |
||||
// no switch possible
|
||||
if (!opt_quiet) |
||||
applog(LOG_DEBUG, "No other pools to try..."); |
||||
return false; |
||||
} |
||||
} |
||||
|
||||
// seturl from api remote (deprecated)
|
||||
bool pool_switch_url(char *params) |
||||
{ |
||||
int prevn = cur_pooln, nextn; |
||||
parse_arg('o', params); |
||||
// cur_pooln modified by parse_arg('o'), get new pool num
|
||||
nextn = cur_pooln; |
||||
// and to handle the "hot swap" from current one...
|
||||
cur_pooln = prevn; |
||||
if (nextn == prevn) |
||||
return false; |
||||
return pool_switch(nextn); |
||||
} |
||||
|
||||
// Parse pools array in json config
|
||||
bool parse_pool_array(json_t *obj) |
||||
{ |
||||
size_t idx; |
||||
json_t *p, *val; |
||||
|
||||
if (!json_is_array(obj)) |
||||
return false; |
||||
|
||||
// array of objects [ {}, {} ]
|
||||
json_array_foreach(obj, idx, p) |
||||
{ |
||||
if (!json_is_object(p)) |
||||
continue; |
||||
|
||||
for (int i = 0; i < ARRAY_SIZE(cfg_array_keys); i++) |
||||
{ |
||||
int opt = -1; |
||||
char *s = NULL; |
||||
if (cfg_array_keys[i].cat != CFG_POOL) |
||||
continue; |
||||
|
||||
val = json_object_get(p, cfg_array_keys[i].name); |
||||
if (!val) |
||||
continue; |
||||
|
||||
for (int k = 0; k < options_count(); k++) |
||||
{ |
||||
const char *alias = cfg_array_keys[i].longname; |
||||
if (alias && !strcasecmp(options[k].name, alias)) { |
||||
opt = k; |
||||
break; |
||||
} |
||||
if (!alias && !strcasecmp(options[k].name, cfg_array_keys[i].name)) { |
||||
opt = k; |
||||
break; |
||||
} |
||||
} |
||||
if (opt == -1) |
||||
continue; |
||||
|
||||
if (json_is_string(val)) { |
||||
s = strdup(json_string_value(val)); |
||||
if (!s) |
||||
continue; |
||||
|
||||
// applog(LOG_DEBUG, "pool key %s '%s'", options[opt].name, s);
|
||||
parse_arg(options[opt].val, s); |
||||
free(s); |
||||
} else { |
||||
// numeric or bool
|
||||
char buf[32] = { 0 }; |
||||
double d = 0.; |
||||
if (json_is_true(val)) d = 1.; |
||||
else if (json_is_integer(val)) |
||||
d = 1.0 * json_integer_value(val); |
||||
else if (json_is_real(val)) |
||||
d = json_real_value(val); |
||||
snprintf(buf, sizeof(buf)-1, "%f", d); |
||||
// applog(LOG_DEBUG, "pool key %s '%f'", options[opt].name, d);
|
||||
parse_arg(options[opt].val, buf); |
||||
} |
||||
} |
||||
} |
||||
return true; |
||||
} |
||||
|
||||
// debug stuff
|
||||
void pool_dump_infos() |
||||
{ |
||||
struct pool_infos *p; |
||||
for (int i=0; i<num_pools; i++) { |
||||
p = &pools[i]; |
||||
applog(LOG_DEBUG, "POOL %01d: %s USER %s -s %d", i, |
||||
p->short_url, p->user, p->scantime); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue