static int/*bool*/ s_AddServerInfo(struct SDISPD_Data* data, SSERV_Info* info)
{
size_t i;
const char* name = SERV_NameOfInfo(info);
/* First check that the new server info updates an existing one */
for (i = 0; i < data->n_cand; i++) {
if (strcasecmp(name, SERV_NameOfInfo(data->cand[i].info)) == 0
&& SERV_EqualInfo(info, data->cand[i].info)) {
/* Replace older version */
free((void*) data->cand[i].info);
data->cand[i].info = info;
return 1;
}
}
/* Next, add new service to the list */
if (data->n_cand == data->a_cand) {
size_t n = data->a_cand + 10;
SLB_Candidate* temp = (SLB_Candidate*)
(data->cand
? realloc(data->cand, n * sizeof(*temp))
: malloc ( n * sizeof(*temp)));
if (!temp)
return 0;
data->cand = temp;
data->a_cand = n;
}
data->cand[data->n_cand++].info = info;
return 1;
}
static int/*bool*/ s_AddSkipInfo(SERV_ITER iter,
const char* name,
SSERV_Info* info)
{
size_t n;
assert(name);
for (n = 0; n < iter->n_skip; n++) {
if (strcasecmp(name, SERV_NameOfInfo(iter->skip[n])) == 0
&& (SERV_EqualInfo(info, iter->skip[n]) ||
(iter->skip[n]->type == fSERV_Firewall &&
iter->skip[n]->u.firewall.type == info->u.firewall.type))) {
/* Replace older version */
if (iter->last == iter->skip[n])
iter->last = info;
free(iter->skip[n]);
iter->skip[n] = info;
return 1;
}
}
if (iter->n_skip == iter->a_skip) {
SSERV_Info** temp;
n = iter->a_skip + 10;
temp = (SSERV_Info**)
(iter->skip
? realloc(iter->skip, n * sizeof(*temp))
: malloc ( n * sizeof(*temp)));
if (!temp)
return 0;
iter->skip = temp;
iter->a_skip = n;
}
iter->skip[iter->n_skip++] = info;
return 1;
}
static SSERV_Info* s_GetNextInfo(SERV_ITER iter, HOST_INFO* host_info)
{
struct SLOCAL_Data* data = (struct SLOCAL_Data*) iter->data;
const TSERV_Type type = iter->type & ~fSERV_Firewall;
int/*bool*/ dns_info_seen = 0/*false*/;
SSERV_Info* info;
size_t i, n;
assert(data);
if (data->reset) {
data->reset = 0/*false*/;
if (!s_LoadServices(iter))
return 0;
if (data->n_cand > 1)
qsort(data->cand, data->n_cand, sizeof(*data->cand), s_Sort);
}
i = 0;
data->i_cand = 0;
while (i < data->n_cand) {
/* NB all servers have been loaded in accordance with iter->external */
info = (SSERV_Info*) data->cand[i].info;
if (info->rate > 0.0 || iter->ok_down) {
const char* c = SERV_NameOfInfo(info);
for (n = 0; n < iter->n_skip; n++) {
const SSERV_Info* skip = iter->skip[n];
const char* s = SERV_NameOfInfo(skip);
if (*s) {
assert(iter->ismask || iter->reverse_dns);
if (strcasecmp(s, c) == 0
&& ((skip->type == fSERV_Dns && !skip->host) ||
SERV_EqualInfo(skip, info))) {
break;
}
} else if (SERV_EqualInfo(skip, info))
break;
if (iter->reverse_dns && skip->type == fSERV_Dns
&& skip->host == info->host
&& (!skip->port || skip->port == info->port)) {
break;
}
}
} else
n = 0;
if (!iter->ismask) {
if (type == fSERV_Any) {
if (iter->reverse_dns && info->type != fSERV_Dns)
dns_info_seen = 1/*true*/;
} else if ((type & info->type) && info->type == fSERV_Dns)
dns_info_seen = 1/*true*/;
}
if (n < iter->n_skip) {
if (i < --data->n_cand) {
memmove(data->cand + i, data->cand + i + 1,
(data->n_cand - i) * sizeof(*data->cand));
}
free(info);
} else {
if (type != fSERV_Any && !(type & info->type))
break;
if (type == fSERV_Any && info->type == fSERV_Dns)
break;
data->i_cand++;
data->cand[i].status = info->rate < 0.0 ? 0.0 : info->rate;
if (iter->ok_down)
break;
i++;
}
}
if (data->i_cand) {
n = LB_Select(iter, data, s_GetCandidate, 1.0);
info = (SSERV_Info*) data->cand[n].info;
if (iter->reverse_dns && info->type != fSERV_Dns) {
dns_info_seen = 0/*false*/;
for (i = 0; i < data->n_cand; i++) {
SSERV_Info* temp = (SSERV_Info*) data->cand[i].info;
if (temp->type != fSERV_Dns ||
temp->host != info->host || temp->port != info->port) {
continue;
}
if (!iter->ismask)
dns_info_seen = 1/*true*/;
if (iter->external && temp->locl)
continue; /* external mapping req'd; local server */
assert(!(temp->locl & 0xF0)); /* no private DNS */
if (temp->rate > 0.0 || iter->ok_down) {
data->cand[i].status = data->cand[n].status;
info = temp;
n = i;
break;
}
}
if (i >= data->n_cand && dns_info_seen)
info = 0;
}
if (info) {
info->rate = data->cand[n].status;
info->time += iter->time;
if (n < --data->n_cand) {
memmove(data->cand + n, data->cand + n + 1,
(data->n_cand - n) * sizeof(*data->cand));
}
}
} else if (iter->last || iter->n_skip || !dns_info_seen) {
info = 0;
} else if ((info = SERV_CreateDnsInfo(0)) != 0)
info->time = NCBI_TIME_INFINITE;
if (info && host_info)
*host_info = 0;
return info;
}
static SSERV_Info* s_GetNextInfo(SERV_ITER iter, HOST_INFO* host_info)
{
const TSERV_Type types = iter->types & ~fSERV_Firewall;
size_t i, n, idx[eHost_NoMatch], n_cand, a_cand;
EHost_Match best_match, match;
struct SLBSM_Candidate* cand;
const SLBSM_Service* svc;
TNCBI_Time dns_info_time;
const SLBSM_Host* host;
const char* env, *a;
SSERV_Info* info;
const char* name;
double status;
int standby;
HEAP heap;
char* v;
heap = (HEAP)(iter->data != iter ? iter->data : 0);
if (heap) {
#ifdef LBSM_DEBUG
CORE_LOGF(eLOG_Trace,
("LBSM heap[%p, %p, %d] for \"%s\" detected",
heap, HEAP_Base(heap), HEAP_Serial(heap), iter->name));
#endif /*LBSM_DEBUG*/
/*noop*/;
} else if (!(heap = s_GetHeapCopy(iter->time))) {
return iter->external || iter->data == iter || !(types & fSERV_Dns)
? 0 : s_FakeDnsReturn(iter, host_info, 0, NCBI_TIME_INFINITE);
}
best_match = eHost_InitMatch;
memset(idx, 0, sizeof(idx));
standby = -1/*unassigned*/;
dns_info_time = 0/*none*/;
n = n_cand = a_cand = 0;
a = v = 0;
cand = 0;
svc = 0;
name = *iter->name ? iter->name : 0;
assert(name || iter->ismask); /*NB: ismask ignored for NULL*/
while ((svc = LBSM_LookupService(heap, name, iter->ismask, svc))) {
if (svc->entry.good < iter->time)
continue; /* out-of-date entry */
if (!svc->info.time)
continue; /* off */
if (types != fSERV_Any && !(types & svc->info.type))
continue; /* type doesn't match */
if (iter->external && svc->info.locl)
continue; /* external mapping requested; local/private server */
if (svc->info.locl & 0xF0) {
/* private server */
if (svc->info.host &&
svc->info.host != s_GetLocalHostAddress(heap)) {
continue;
}
}
if (svc->info.type == fSERV_Dns) {
if (types == fSERV_Any)
continue; /* DNS entries have to be requested explicitly */
if (!iter->ismask) {
if (dns_info_time < svc->info.time)
dns_info_time = svc->info.time;
}
} else {
if (iter->stateless && svc->info.sful) {
/* Skip stateful-only non-CGI (NCBID and standalone) svc */
if (!(svc->info.type & fSERV_Http))
continue;
}
if (!iter->ismask && iter->reverse_dns) {
if (dns_info_time < svc->info.time)
dns_info_time = svc->info.time;
}
}
if (svc->info.rate > 0.0 || host_info) {
if (!(host = s_LookupHost(heap, iter, svc))
&& svc->info.rate > 0.0) {
continue; /* no host information for non-static server */
}
} else
host = 0;
for (n = 0; n < iter->n_skip; n++) {
const SSERV_Info* skip = iter->skip[n];
const char* s = SERV_NameOfInfo(skip);
if (*s) {
assert(iter->ismask || iter->reverse_dns);
if (strcasecmp(s, (const char*) svc + svc->name) == 0
&& ((skip->type == fSERV_Dns && !skip->host)
|| SERV_EqualInfo(skip, &svc->info))) {
break;
}
} else if (SERV_EqualInfo(skip, &svc->info))
break;
if (skip->type == fSERV_Firewall
&& skip->u.firewall.type == svc->info.type) {
break;
}
if (iter->reverse_dns && skip->type == fSERV_Dns
&& skip->host == svc->info.host
&& (!skip->port || skip->port == svc->info.port)) {
break;
}
}
/*FIXME*//*CORE_LOG(eLOG_Note, (char*) svc + svc->name);*/
if (n >= iter->n_skip) {
status = LBSM_CalculateStatus(svc->info.rate, svc->fine,
svc->info.flag, &host->sys.load);
if (status <= 0.0) {
if (!svc->info.rate) {
if (!iter->ok_down)
continue; /* not operational */
status = 0.0;
} else
status = copysign(svc->info.rate, -1.0);
}
} else
status = 0.0; /* dummy assignment to keep no-init warning off */
if (v) {
free(v);
v = 0;
}
a = env = 0;
if (iter->pref < 0.0 && iter->host
&& (iter->host != svc->info.host
|| (iter->port && iter->port != svc->info.port))) {
/* not a suitable fixed latching */
match = eHost_BadMatch;
} else if (iter->arglen) {
assert(iter->arg);
if (!host)
host = s_LookupHost(heap, iter, svc);
if ( host && host->env)
env = (const char*) host + host->env;
match = s_Match(env,
iter->arg, iter->arglen,
iter->val, iter->vallen, &a, &v);
assert(!a || a == iter->arg);
} else
match = eHost_GoodMatch;
if (best_match > match)
best_match = match;
if (match > eHost_NoMatch) {
assert(!v);
continue;
}
if (svc->info.rate) {
/* NB: server is _not_ down, but it may have been suppressed */
if (fabs(svc->info.rate) < 0.01) {
if (!standby) {
if (!iter->ok_suppressed)
continue;
/* this has to be given out as a suppressed one */
status = copysign(svc->info.rate, -1.0);
} else if (standby < 0)
standby = 1;
} else if (standby) {
standby = 0/*cancel*/;
if (!iter->ok_suppressed) {
memset(idx, 0, sizeof(idx));
for (i = 0; i < n_cand; i++) {
if (cand[i].val)
free((void*) cand[i].val);
}
n_cand = 0;
} else for (i = 0; i < n_cand; i++)
cand[i].cand.status = copysign(cand[i].svc->info.rate,-1.);
}
}
if (n < iter->n_skip)
continue; /* excluded/seen; NB: dummy assignment goes off here */
if (!iter->ok_suppressed && status < 0.0)
continue;
#ifdef NCBI_LB_DEBUG
if (iter->arglen) {
char* s = SERV_WriteInfo(&svc->info);
const char* m;
assert(s);
switch (match) {
case eHost_BestMatch:
m = "Best match";
break;
case eHost_GoodMatch:
m = "Good match";
break;
case eHost_FairMatch:
m = "Fair match";
break;
case eHost_PoorMatch:
m = "Poor match";
break;
case eHost_NoMatch:
m = "No match";
break;
default:
assert(0);
m = "?";
break;
}
assert(!a || *a);
assert(!v || a);
CORE_LOGF(eLOG_Note, ("%s%s%s%s: %s%s%s%s%s%s", s,
env ? " <" : "", env ? env : "",
env ? ">" : "", m,
a ? ", arg=" : "",
a ? a : "",
v ? ", val=" : "",
v ? (*v ? v : "\"\"") : "",
standby > 0 ? ", standby" : ""));
free(s);
}
#endif /*NCBI_LB_DEBUG*/
/* This server should be taken into consideration */
if (n_cand == a_cand) {
struct SLBSM_Candidate* temp;
n = a_cand + 10;
temp = (struct SLBSM_Candidate*)
(cand
? realloc(cand, n * sizeof(*temp))
: malloc ( n * sizeof(*temp)));
if (!temp)
break;
cand = temp;
a_cand = n;
}
if (match < eHost_NoMatch) {
assert((size_t) match < sizeof(idx)/sizeof(idx[0]));
n = idx[match];
if (n < n_cand)
memmove(&cand[n + 1], &cand[n], sizeof(cand[0])*(n_cand - n));
for (i = match; i < sizeof(idx)/sizeof(idx[0]); i++)
idx[i]++;
} else
n = n_cand;
cand[n].cand.info = &svc->info;
cand[n].cand.status = status;
cand[n].host = host;
cand[n].svc = svc;
cand[n].arg = a;
cand[n].val = v;
a = v = 0;
n_cand++;
}
if (v)
free(v);
if (best_match < eHost_NoMatch) {
assert(!best_match || !idx[best_match - 1]);
for (n = idx[best_match]; n < n_cand; n++) {
if (cand[n].val)
free((void*) cand[n].val);
}
n_cand = idx[best_match];
}
if (n_cand) {
assert(cand);
do {
if (standby <= 0) {
struct SLBSM_Data data;
data.cand = cand;
data.n_cand = n_cand;
n = LB_Select(iter, &data, s_GetCandidate, LBSMD_LOCAL_BONUS);
} else {
qsort(cand, n_cand, sizeof(*cand), s_SortStandbys);
status = cand[0].cand.status;
for (n = 1; n < n_cand; n++) {
if (status != cand[n].cand.status)
break;
}
n = rand() % n;
}
svc = cand[n].svc;
if (iter->reverse_dns && svc->info.type != fSERV_Dns) {
svc = 0;
dns_info_time = 0/*none*/;
while ((svc = LBSM_LookupService(heap, 0/*all*/, 0, svc)) !=0){
if (svc->info.type != fSERV_Dns || !svc->info.time ||
svc->info.host != cand[n].svc->info.host ||
svc->info.port != cand[n].svc->info.port) {
continue;
}
if (!iter->ismask) {
if (dns_info_time < svc->info.time)
dns_info_time = svc->info.time;
}
if (iter->external && svc->info.locl)
continue;/* external mapping requested; local server */
assert(!(svc->info.locl & 0xF0)); /* no private DNS */
status = LBSM_CalculateStatus(!svc->info.rate ? 0.0
: -LBSM_DEFAULT_RATE,
svc->fine, fSERV_Regular,
NULL);
if (status > 0.0)
break;
if ((!svc->info.rate && iter->ok_down) ||
( svc->info.rate && iter->ok_suppressed)) {
cand[n].cand.status = !svc->info.rate ? 0.0
: copysign(svc->info.rate, -1.0);
break;
}
}
if (!svc && !dns_info_time)
svc = cand[n].svc;
}
if (svc)
break;
if (cand[n].val)
free((void*) cand[n].val);
if (n < --n_cand)
memmove(cand + n, cand + n + 1, (n_cand - n) * sizeof(*cand));
} while (n_cand);
} else
svc = 0;
if (svc) {
const char* name = (iter->ismask || iter->reverse_dns ?
(const char*) svc + svc->name : "");
if ((info = SERV_CopyInfoEx(&svc->info, name)) != 0) {
info->rate = cand[n].cand.status;
if (info->time != NCBI_TIME_INFINITE)
info->time = cand[n].svc->entry.good;
if (host_info) {
if ((host = cand[n].host) != 0) {
*host_info =
HINFO_Create(host->addr, &host->sys, sizeof(host->sys),
host->env
? (const char*) host + host->env
: 0, cand[n].arg, cand[n].val);
} else
*host_info = 0;
}
}
} else {
info = !n_cand && dns_info_time
? s_FakeDnsReturn(iter, host_info,
best_match == eHost_InitMatch ? 0/*down*/ :
best_match != eHost_BadMatch ? -1/*busy*/ : 1,
dns_info_time)
: 0;
}
for (n = 0; n < n_cand; n++) {
if (cand[n].val)
free((void*) cand[n].val);
}
if (cand)
free(cand);
if (!s_FastHeapAccess) {
#ifdef LBSM_DEBUG
CORE_LOGF(eLOG_Trace,
("LBSM heap[%p, %p, %d] for \"%s\" released",
heap, HEAP_Base(heap), HEAP_Serial(heap), iter->name));
#endif /*LBSM_DEBUG*/
CORE_LOCK_WRITE;
HEAP_Detach(heap);
CORE_UNLOCK;
heap = 0;
}
#ifdef LBSM_DEBUG
else {
CORE_LOGF(eLOG_Trace,
("LBSM heap[%p, %p, %d] for \"%s\" retained",
heap, HEAP_Base(heap), HEAP_Serial(heap), iter->name));
}
#endif /*LBSM_DEBUG*/
iter->data = heap;
return info;
}
