virtual void send(nrpe::packet &packet, boost::posix_time::seconds timeout) {
std::vector<char> buf = packet.get_buffer();
write_with_timeout(buf, timeout);
}
virtual nrpe::packet recv(const nrpe::packet &packet, boost::posix_time::seconds timeout) {
std::vector<char> buf(packet.get_packet_length());
read_with_timeout(buf, timeout);
return nrpe::packet(&buf[0], buf.size(), packet.get_payload_length());
}
nrpe::packet handler_impl::handle(nrpe::packet p) {
strEx::s::token cmd = strEx::s::getToken(p.getPayload(), '!');
if (cmd.first == "_NRPE_CHECK") {
return nrpe::packet::create_response(NSCAPI::returnOK, "I (" + utf8::cvt<std::string>(nscapi::plugin_singleton->get_core()->getApplicationVersionString()) + ") seem to be doing fine...", p.get_payload_length());
}
if (!allowArgs_) {
if (!cmd.second.empty()) {
NSC_LOG_ERROR("Request contained arguments (not currently allowed, check the allow arguments option).");
throw nrpe::nrpe_exception("Request contained arguments (not currently allowed, check the allow arguments option).");
}
}
if (!allowNasty_) {
if (cmd.first.find_first_of(NASTY_METACHARS) != std::wstring::npos) {
NSC_LOG_ERROR("Request command contained illegal metachars!");
throw nrpe::nrpe_exception("Request command contained illegal metachars!");
}
if (cmd.second.find_first_of(NASTY_METACHARS) != std::wstring::npos) {
NSC_LOG_ERROR("Request arguments contained illegal metachars!");
throw nrpe::nrpe_exception("Request command contained illegal metachars!");
}
}
std::string wmsg, wperf;
NSCAPI::nagiosReturn ret = -3;
try {
if (encoding_.empty()) {
ret = nscapi::core_helper::simple_query_from_nrpe(utf8::cvt<std::string>(utf8::to_unicode(cmd.first)), utf8::cvt<std::string>(utf8::to_unicode(cmd.second)), wmsg, wperf);
} else {
ret = nscapi::core_helper::simple_query_from_nrpe(utf8::cvt<std::string>(utf8::from_encoding(cmd.first, encoding_)), utf8::cvt<std::string>(utf8::from_encoding(cmd.second, encoding_)), wmsg, wperf);
}
} catch (...) {
return nrpe::packet::create_response(NSCAPI::returnUNKNOWN, "UNKNOWN: Internal exception", p.get_payload_length());
}
switch (ret) {
case NSCAPI::returnInvalidBufferLen:
return nrpe::packet::create_response(NSCAPI::returnUNKNOWN, "UNKNOWN: Return buffer to small to handle this command.", p.get_payload_length());
case NSCAPI::returnIgnored:
return nrpe::packet::create_response(NSCAPI::returnUNKNOWN, "UNKNOWN: No handler for that command.", p.get_payload_length());
case NSCAPI::returnOK:
case NSCAPI::returnWARN:
case NSCAPI::returnCRIT:
case NSCAPI::returnUNKNOWN:
break;
default:
return nrpe::packet::create_response(NSCAPI::returnUNKNOWN, "UNKNOWN: Internal error.", p.get_payload_length());
}
std::string data,msg, perf;
if (encoding_.empty()) {
msg = utf8::to_system(utf8::cvt<std::wstring>(wmsg));
perf = utf8::to_system(utf8::cvt<std::wstring>(wperf));
} else {
msg = utf8::to_encoding(utf8::cvt<std::wstring>(wmsg), encoding_);
perf = utf8::to_encoding(utf8::cvt<std::wstring>(wperf), encoding_);
}
const unsigned int max_len = p.get_payload_length()-1;
if (msg.length() >= max_len) {
data = msg.substr(0, max_len);
} else if (perf.empty() || noPerfData_) {
data = msg;
} else if (msg.length() + perf.length() + 1 > max_len) {
data = msg;
} else {
data = msg + "|" + perf;
}
return nrpe::packet::create_response(ret, data, p.get_payload_length());
}
