X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/4b8db6bde20164a3603d2800858696b2bf01900e..0a8aa4fffee15d7e175d857b7248d690081f41fd:/projects/stargazer/plugins/other/snmp/snmp.cpp?ds=inline diff --git a/projects/stargazer/plugins/other/snmp/snmp.cpp b/projects/stargazer/plugins/other/snmp/snmp.cpp index 8c026128..2884e6fc 100644 --- a/projects/stargazer/plugins/other/snmp/snmp.cpp +++ b/projects/stargazer/plugins/other/snmp/snmp.cpp @@ -211,31 +211,51 @@ else return 0; } -int SendSetResponsePDU(int fd, const PDU_t * pdu) +int SendGetResponsePDU(int fd, GetResponse_PDU_t * getResponse) +{ +asn_enc_rval_t error; + +char buffer[1024]; +error = der_encode_to_buffer(&asn_DEF_GetResponse_PDU, getResponse, buffer, sizeof(buffer)); + +if (error.encoded == -1) + { + printfd(__FILE__, "Could not encode GetResponsePDU (at %s)\n", + error.failed_type ? error.failed_type->name : "unknown"); + return -1; + } +else + { + write(fd, buffer, error.encoded); + printfd(__FILE__, "GetResponsePDU encoded successfully to %d bytes\n", error.encoded); + } +return 0; +} + +int SendGetResponseErrorPDU(int fd, const PDU_t * getRequest, int errorStatus, int errorIndex) { -//int oid[] = {1, 3, 6, 1, 4, 1, 9363, 1}; asn_enc_rval_t error; GetResponse_PDU_t msg; memset(&msg, 0, sizeof(msg)); -msg.request_id = pdu->request_id; -asn_long2INTEGER(&msg.error_status, PDU__error_status_readOnly); -asn_long2INTEGER(&msg.error_index, PDU__error_status_readOnly); +msg.request_id = getRequest->request_id; +asn_long2INTEGER(&msg.error_status, errorStatus); +asn_long2INTEGER(&msg.error_index, errorIndex); char buffer[1024]; error = der_encode_to_buffer(&asn_DEF_GetResponse_PDU, &msg, buffer, sizeof(buffer)); if (error.encoded == -1) { - printfd(__FILE__, "Could not encode GetResponsePDU (at %s)\n", + printfd(__FILE__, "Could not encode GetResponsePDU for error (at %s)\n", error.failed_type ? error.failed_type->name : "unknown"); return -1; } else { write(fd, buffer, error.encoded); - printfd(__FILE__, "GetResponsePDU encoded successfully to %d bytes\n", error.encoded); + printfd(__FILE__, "GetResponsePDU for error encoded successfully to %d bytes\n", error.encoded); } return 0; } @@ -332,6 +352,8 @@ return 0; SNMP_AGENT::SNMP_AGENT() : PLUGIN(), + users(NULL), + tariffs(NULL), running(false), stopped(true), sock(-1) @@ -350,6 +372,9 @@ pdusHandlers[PDUs_PR_set_request] = &SNMP_AGENT::SetRequestHandler; SNMP_AGENT::~SNMP_AGENT() { +Sensors::iterator it; +for (it = sensors.begin(); it != sensors.end(); ++it) + delete it->second; printfd(__FILE__, "SNMP_AGENT::~SNMP_AGENT()\n"); pthread_mutex_destroy(&mutex); } @@ -364,6 +389,20 @@ int SNMP_AGENT::Start() if (PrepareNet()) return -1; +// Users +sensors[".1.3.6.1.4.1.9363.1.1.1"] = new TotalUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.2"] = new ConnectedUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.3"] = new AuthorizedUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.4"] = new AlwaysOnlineUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.5"] = new NoCashUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.7"] = new DisabledDetailStatsUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.8"] = new DisabledUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.9"] = new PassiveUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.10"] = new CreditUsersSensor(*users); +sensors[".1.3.6.1.4.1.9363.1.1.11"] = new FreeMbUsersSensor(*users); +// Tariffs +sensors[".1.3.6.1.4.1.9363.1.2.1"] = new TotalTariffsSensor(*tariffs); + if (!running) { if (pthread_create(&thread, NULL, Runner, this)) @@ -584,12 +623,38 @@ bool SNMP_AGENT::GetRequestHandler(const PDUs_t * pdus) { printfd(__FILE__, "SNMP_AGENT::GetRequestHandler()\n"); asn_fprint(stderr, &asn_DEF_PDUs, pdus); -const VarBindList_t * vbl = &pdus->choice.get_request.variable_bindings; +const GetRequest_PDU_t * getRequest = &pdus->choice.get_request; +GetResponse_PDU_t msg; +VarBindList_t * varBindList = &msg.variable_bindings; +memset(&msg, 0, sizeof(msg)); + +msg.request_id = getRequest->request_id; +asn_long2INTEGER(&msg.error_status, 0); +asn_long2INTEGER(&msg.error_index, 0); + +const VarBindList_t * vbl = &getRequest->variable_bindings; for (int i = 0; i < vbl->list.count; ++i) { - VarBind_t * vb = pdus->choice.get_request.variable_bindings.list.array[i]; - printfd(__FILE__, "OID: %s\n", OI2String(&vb->name).c_str()); + VarBind_t * vb = getRequest->variable_bindings.list.array[i]; + Sensors::iterator it; + it = sensors.find(OI2String(&vb->name)); + if (it == sensors.end()) + { + SendGetResponseErrorPDU(sock, getRequest, PDU__error_status_noSuchName, i); + return true; + } + + VarBind_t newVb; + memset(&newVb, 0, sizeof(newVb)); + + newVb.name = vb->name; + it->second->GetValue(&newVb.value); + + ASN_SEQUENCE_ADD(varBindList, &newVb); } + +SendGetResponsePDU(sock, &msg); +asn_fprint(stderr, &asn_DEF_PDU, &msg); return false; } @@ -604,7 +669,7 @@ bool SNMP_AGENT::SetRequestHandler(const PDUs_t * pdus) { printfd(__FILE__, "SNMP_AGENT::SetRequestHandler()\n"); asn_fprint(stderr, &asn_DEF_PDUs, pdus); -SendSetResponsePDU(sock, &pdus->choice.set_request); +SendGetResponseErrorPDU(sock, &pdus->choice.set_request, PDU__error_status_readOnly, 0); return false; }