extern int location_fini(void)
{
int rc;
if (!plugin_context)
return SLURM_SUCCESS;
init_run = false;
rc = plugin_context_destroy(plugin_context);
plugin_context = NULL;
return rc;
}
extern int jobacct_gather_fini(void)
{
int rc = SLURM_SUCCESS;
slurm_mutex_lock(&g_context_lock);
if (g_context) {
init_run = false;
rc = plugin_context_destroy(g_context);
g_context = NULL;
}
slurm_mutex_unlock(&g_context_lock);
return rc;
}
extern int slurm_acct_storage_fini(void)
{
int rc;
if (!plugin_context)
return SLURM_SUCCESS;
init_run = false;
// (*(ops.acct_storage_fini))();
rc = plugin_context_destroy(plugin_context);
plugin_context = NULL;
return rc;
}
extern int acct_gather_energy_fini(void)
{
int rc;
if (!g_context)
return SLURM_SUCCESS;
init_run = false;
rc = plugin_context_destroy(g_context);
g_context = NULL;
return rc;
}
/* *********************************************************************** */
extern int
slurm_topo_fini( void )
{
int rc;
if (!g_context)
return SLURM_SUCCESS;
init_run = false;
rc = plugin_context_destroy(g_context);
g_context = NULL;
return rc;
}
/* shutdown checkpoint plugin */
extern int
checkpoint_fini(void)
{
int rc;
if (!g_context)
return SLURM_SUCCESS;
slurm_mutex_lock(&context_lock);
init_run = false;
rc = plugin_context_destroy(g_context);
slurm_mutex_unlock(&context_lock);
return rc;
}
extern int slurm_mcs_fini(void)
{
int rc = SLURM_SUCCESS;
if (!g_mcs_context)
return rc;
init_run = false;
rc = plugin_context_destroy(g_mcs_context);
g_mcs_context = NULL;
xfree(mcs_params_common);
xfree(mcs_params_specific);
xfree(mcs_params);
return rc;
}
extern int other_select_fini(void)
{
int rc = SLURM_SUCCESS;
slurm_mutex_lock(&g_context_lock);
init_run = false;
if (!g_context)
goto fini;
rc = plugin_context_destroy(g_context);
g_context = NULL;
fini:
slurm_mutex_unlock(&g_context_lock);
return rc;
}
extern int
g_slurm_jobcomp_fini(void)
{
slurm_mutex_lock( &context_lock );
if ( !g_context)
goto done;
init_run = false;
plugin_context_destroy ( g_context );
g_context = NULL;
done:
slurm_mutex_unlock( &context_lock );
return SLURM_SUCCESS;
}
/* ************************************************************************** */
extern int route_fini(void)
{
int rc;
if (!g_context)
return SLURM_SUCCESS;
init_run = false;
rc = plugin_context_destroy(g_context);
g_context = NULL;
xfree(msg_collect_node);
xfree(msg_collect_backup);
return rc;
}
extern int acct_gather_profile_fini(void)
{
int rc = SLURM_SUCCESS, i;
if (!g_context)
return SLURM_SUCCESS;
slurm_mutex_lock(&g_context_lock);
if (!g_context)
goto done;
init_run = false;
for (i=0; i < PROFILE_CNT; i++) {
switch (i) {
case PROFILE_ENERGY:
acct_gather_energy_fini();
break;
case PROFILE_TASK:
jobacct_gather_fini();
break;
case PROFILE_FILESYSTEM:
acct_gather_filesystem_fini();
break;
case PROFILE_NETWORK:
acct_gather_infiniband_fini();
break;
default:
fatal("Unhandled profile option %d please update "
"slurm_acct_gather_profile.c "
"(acct_gather_profile_fini)", i);
}
}
if (timer_thread_id) {
pthread_cancel(timer_thread_id);
pthread_join(timer_thread_id, NULL);
}
rc = plugin_context_destroy(g_context);
g_context = NULL;
done:
slurm_mutex_unlock(&g_context_lock);
return rc;
}
/* *********************************************************************** */
extern int
slurm_sched_fini( void )
{
int rc;
if (!g_context)
return SLURM_SUCCESS;
init_run = false;
rc = plugin_context_destroy(g_context);
g_context = NULL;
if ( (slurm_get_preempt_mode() & PREEMPT_MODE_GANG) &&
(gs_fini() != SLURM_SUCCESS))
error( "cannot stop gang scheduler" );
return rc;
}
extern int switch_fini(void)
{
int rc = SLURM_SUCCESS, i;
slurm_mutex_lock( &context_lock );
if (!switch_context)
goto fini;
init_run = false;
for (i = 0; i < switch_context_cnt; i++) {
rc |= plugin_context_destroy(switch_context[i]);
}
xfree(switch_context);
xfree(ops);
switch_context_cnt = -1;
fini:
slurm_mutex_unlock( &context_lock );
return rc;
}
extern int acct_gather_infiniband_fini(void)
{
int rc = SLURM_SUCCESS;
slurm_mutex_lock(&g_context_lock);
if (g_context) {
init_run = false;
if (watch_node_thread_id) {
pthread_cancel(watch_node_thread_id);
pthread_join(watch_node_thread_id, NULL);
}
rc = plugin_context_destroy(g_context);
g_context = NULL;
}
slurm_mutex_unlock(&g_context_lock);
return rc;
}
extern int slurm_select_fini(void)
{
int rc = SLURM_SUCCESS, i, j;
slurm_mutex_lock(&select_context_lock);
if (!select_context)
goto fini;
init_run = false;
for (i=0; i<select_context_cnt; i++) {
j = plugin_context_destroy(select_context[i]);
if (j != SLURM_SUCCESS)
rc = j;
}
xfree(ops);
xfree(select_context);
select_context_cnt = -1;
fini: slurm_mutex_unlock(&select_context_lock);
return rc;
}
/* Terminate the power plugin and free all memory */
extern void power_g_fini(void)
{
int i;
slurm_mutex_lock(&g_context_lock);
if (g_context_cnt < 0)
goto fini;
init_run = false;
for (i = 0; i < g_context_cnt; i++) {
if (g_context[i])
plugin_context_destroy(g_context[i]);
}
xfree(ops);
xfree(g_context);
xfree(power_plugin_list);
g_context_cnt = -1;
fini: slurm_mutex_unlock(&g_context_lock);
return;
}
extern int acct_gather_interconnect_fini(void)
{
int rc2, rc = SLURM_SUCCESS;
int i;
slurm_mutex_lock(&g_context_lock);
init_run = false;
if (watch_node_thread_id) {
slurm_mutex_unlock(&g_context_lock);
slurm_mutex_lock(&profile_timer->notify_mutex);
slurm_cond_signal(&profile_timer->notify);
slurm_mutex_unlock(&profile_timer->notify_mutex);
pthread_join(watch_node_thread_id, NULL);
slurm_mutex_lock(&g_context_lock);
}
for (i = 0; i < g_context_num; i++) {
if (!g_context[i])
continue;
rc2 = plugin_context_destroy(g_context[i]);
if (rc2 != SLURM_SUCCESS) {
debug("%s: %s: %s", __func__,
g_context[i]->type,
slurm_strerror(rc2));
rc = SLURM_ERROR;
}
}
xfree(ops);
xfree(g_context);
g_context_num = -1;
slurm_mutex_unlock(&g_context_lock);
return rc;
}
/*
* Terminate the job submit plugin. Free memory.
*
* Returns a SLURM errno.
*/
extern int job_submit_plugin_fini(void)
{
int i, j, rc = SLURM_SUCCESS;
slurm_mutex_lock(&g_context_lock);
if (g_context_cnt < 0)
goto fini;
init_run = false;
for (i=0; i<g_context_cnt; i++) {
if (g_context[i]) {
j = plugin_context_destroy(g_context[i]);
if (j != SLURM_SUCCESS)
rc = j;
}
}
xfree(ops);
xfree(g_context);
xfree(submit_plugin_list);
g_context_cnt = -1;
fini: slurm_mutex_unlock(&g_context_lock);
return rc;
}
/*
* Create a priority context
*/
extern plugin_context_t *plugin_context_create(
const char *plugin_type, const char *uler_type,
void *ptrs[], const char *names[], size_t names_size)
{
plugin_context_t *c;
int n_names;
if (!uler_type) {
debug3("plugin_context_create: no uler type");
return NULL;
} else if (!plugin_type) {
debug3("plugin_context_create: no plugin type");
return NULL;
} else if (!names) {
error("plugin_context_create: no symbols given for plugin %s",
plugin_type);
return NULL;
} else if (!ptrs) {
error("plugin_context_create: no ptrs given for plugin %s",
plugin_type);
return NULL;
}
c = xmalloc(sizeof(plugin_context_t));
c->type = xstrdup(uler_type);
c->cur_plugin = PLUGIN_INVALID_HANDLE;
n_names = names_size / sizeof(char *);
/* Find the correct plugin. */
c->cur_plugin = plugin_load_and_link(c->type, n_names, names, ptrs);
if (c->cur_plugin != PLUGIN_INVALID_HANDLE)
return c;
if (errno != EPLUGIN_NOTFOUND) {
error("Couldn't load specified plugin name for %s: %s",
c->type, plugin_strerror(errno));
goto fail;
}
error("Couldn't find the specified plugin name for %s "
"looking at all files",
c->type);
/* Get plugin list. */
if (!c->plugin_list) {
char *plugin_dir;
c->plugin_list = plugrack_create();
if (!c->plugin_list) {
error("cannot create plugin manager");
goto fail;
}
plugrack_set_major_type(c->plugin_list, plugin_type);
plugrack_set_paranoia(
c->plugin_list, PLUGRACK_PARANOIA_NONE, 0);
plugin_dir = slurm_get_plugin_dir();
plugrack_read_dir(c->plugin_list, plugin_dir);
xfree(plugin_dir);
}
c->cur_plugin = plugrack_use_by_type(c->plugin_list, c->type);
if (c->cur_plugin == PLUGIN_INVALID_HANDLE) {
error("cannot find %s plugin for %s", plugin_type, c->type);
goto fail;
}
/* Dereference the API. */
if (plugin_get_syms(c->cur_plugin, n_names, names, ptrs) < n_names) {
error("incomplete %s plugin detected", plugin_type);
goto fail;
}
return c;
fail:
plugin_context_destroy(c);
return NULL;
}
/* Get this plugin's sequence number in Slurm's internal tables */
extern int select_get_plugin_id_pos(uint32_t plugin_id)
{
int i;
static bool cray_other_cons_res = false;
if (slurm_select_init(0) < 0)
return SLURM_ERROR;
again:
for (i = 0; i < select_context_cnt; i++) {
if (*(ops[i].plugin_id) == plugin_id)
break;
}
if (i >= select_context_cnt) {
/*
* Put on the extra Cray select plugins that do not get
* generated automatically.
*/
if (!cray_other_cons_res &&
((plugin_id == SELECT_PLUGIN_CRAY_CONS_RES) ||
(plugin_id == SELECT_PLUGIN_CRAY_CONS_TRES) ||
(plugin_id == SELECT_PLUGIN_CRAY_LINEAR))) {
char *type = "select", *name = "select/cray";
uint16_t save_params = slurm_get_select_type_param();
uint16_t params[2];
int cray_plugin_id[2], cray_offset;
cray_other_cons_res = true;
if (plugin_id == SELECT_PLUGIN_CRAY_LINEAR) {
params[0] = save_params & ~CR_OTHER_CONS_RES;
cray_plugin_id[0] = SELECT_PLUGIN_CRAY_CONS_RES;
params[1] = save_params & ~CR_OTHER_CONS_TRES;
cray_plugin_id[1] = SELECT_PLUGIN_CRAY_CONS_TRES;
} else if (plugin_id == SELECT_PLUGIN_CRAY_CONS_RES) {
params[0] = save_params | CR_OTHER_CONS_RES;
cray_plugin_id[0] = SELECT_PLUGIN_CRAY_LINEAR;
params[1] = save_params & ~CR_OTHER_CONS_RES;
cray_plugin_id[1] = SELECT_PLUGIN_CRAY_CONS_TRES;
} else { /* SELECT_PLUGIN_CRAY_CONS_TRES */
params[0] = save_params | CR_OTHER_CONS_TRES;
cray_plugin_id[0] = SELECT_PLUGIN_CRAY_LINEAR;
params[1] = save_params & ~CR_OTHER_CONS_RES;
cray_plugin_id[1] = SELECT_PLUGIN_CRAY_CONS_RES;
}
for (cray_offset = 0; cray_offset < 2; cray_offset++) {
for (i = 0; i < select_context_cnt; i++) {
if (*(ops[i].plugin_id) ==
cray_plugin_id[cray_offset])
break;
}
if (i < select_context_cnt)
break; /* Found match */
}
if (i >= select_context_cnt)
goto end_it; /* No match */
slurm_mutex_lock(&select_context_lock);
slurm_set_select_type_param(params[cray_offset]);
plugin_context_destroy(select_context[i]);
select_context[i] =
plugin_context_create(type, name,
(void **)&ops[i],
node_select_syms,
sizeof(node_select_syms));
slurm_set_select_type_param(save_params);
slurm_mutex_unlock(&select_context_lock);
goto again;
}
end_it:
return SLURM_ERROR;
}
return i;
}
