static void
_handle_msg(slurm_msg_t *msg)
{
static uint32_t slurm_uid = NO_VAL;
uid_t req_uid = g_slurm_auth_get_uid(msg->auth_cred,
slurm_get_auth_info());
uid_t uid = getuid();
job_step_kill_msg_t *ss;
srun_user_msg_t *um;
if (slurm_uid == NO_VAL)
slurm_uid = slurm_get_slurm_user_id();
if ((req_uid != slurm_uid) && (req_uid != 0) && (req_uid != uid)) {
error ("Security violation, slurm message from uid %u",
(unsigned int) req_uid);
return;
}
switch (msg->msg_type) {
case SRUN_PING:
debug3("slurmctld ping received");
slurm_send_rc_msg(msg, SLURM_SUCCESS);
slurm_free_srun_ping_msg(msg->data);
break;
case SRUN_JOB_COMPLETE:
debug("received job step complete message");
runjob_signal(SIGKILL);
slurm_free_srun_job_complete_msg(msg->data);
break;
case SRUN_USER_MSG:
um = msg->data;
info("%s", um->msg);
slurm_free_srun_user_msg(msg->data);
break;
case SRUN_TIMEOUT:
debug("received job step timeout message");
_handle_timeout(msg->data);
slurm_free_srun_timeout_msg(msg->data);
break;
case SRUN_STEP_SIGNAL:
ss = msg->data;
debug("received step signal %u RPC", ss->signal);
if (ss->signal)
runjob_signal(ss->signal);
slurm_free_job_step_kill_msg(msg->data);
break;
default:
debug("received spurious message type: %u",
msg->msg_type);
break;
}
return;
}
static void _purge_agent_args(agent_arg_t *agent_arg_ptr)
{
if (agent_arg_ptr == NULL)
return;
hostlist_destroy(agent_arg_ptr->hostlist);
xfree(agent_arg_ptr->addr);
if (agent_arg_ptr->msg_args) {
if (agent_arg_ptr->msg_type == REQUEST_BATCH_JOB_LAUNCH)
slurmctld_free_batch_job_launch_msg(agent_arg_ptr->
msg_args);
else if (agent_arg_ptr->msg_type ==
RESPONSE_RESOURCE_ALLOCATION)
slurm_free_resource_allocation_response_msg(
agent_arg_ptr->msg_args);
else if ((agent_arg_ptr->msg_type == REQUEST_ABORT_JOB) ||
(agent_arg_ptr->msg_type == REQUEST_TERMINATE_JOB) ||
(agent_arg_ptr->msg_type == REQUEST_KILL_PREEMPTED) ||
(agent_arg_ptr->msg_type == REQUEST_KILL_TIMELIMIT))
slurm_free_kill_job_msg(agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == SRUN_USER_MSG)
slurm_free_srun_user_msg(agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == SRUN_EXEC)
slurm_free_srun_exec_msg(agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == SRUN_NODE_FAIL)
slurm_free_srun_node_fail_msg(agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == SRUN_STEP_MISSING)
slurm_free_srun_step_missing_msg(
agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == SRUN_STEP_SIGNAL)
slurm_free_job_step_kill_msg(
agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == REQUEST_JOB_NOTIFY)
slurm_free_job_notify_msg(agent_arg_ptr->msg_args);
else if (agent_arg_ptr->msg_type == REQUEST_SUSPEND_INT)
slurm_free_suspend_int_msg(agent_arg_ptr->msg_args);
else
xfree(agent_arg_ptr->msg_args);
}
xfree(agent_arg_ptr);
}
