X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/911b6d1f5643bf4d353774f82a8b742bec4e5677..78bad7b5d52407128d5069cb4003f7d6d0d10dd1:/projects/stargazer/plugins/other/snmp/snmp.cpp diff --git a/projects/stargazer/plugins/other/snmp/snmp.cpp b/projects/stargazer/plugins/other/snmp/snmp.cpp index 3ea7d70a..7e961208 100644 --- a/projects/stargazer/plugins/other/snmp/snmp.cpp +++ b/projects/stargazer/plugins/other/snmp/snmp.cpp @@ -13,6 +13,10 @@ #include "asn1/OpenPDU.h" #include "asn1/ClosePDU.h" #include "asn1/RReqPDU.h" +#include "asn1/GetRequest-PDU.h" +#include "asn1/GetResponse-PDU.h" +#include "asn1/VarBindList.h" +#include "asn1/VarBind.h" #include "asn1/OBJECT_IDENTIFIER.h" #include "asn1/ber_decoder.h" #include "asn1/der_encoder.h" @@ -22,6 +26,62 @@ bool WaitPackets(int sd); +std::string OI2String(OBJECT_IDENTIFIER_t * oi) +{ +std::string res; + +int arcs[1024]; +int count = OBJECT_IDENTIFIER_get_arcs(oi, arcs, sizeof(arcs[0]), 1024); + +if (count > 1024) + return ""; + +for (int i = 0; i < count; ++i) + { + res += "."; + std::string arc; + strprintf(&arc, "%d", arcs[i]); + res += arc; + } + +return res; +} + +bool String2OI(const std::string & str, OBJECT_IDENTIFIER_t * oi) +{ +size_t left = 0, pos = 0, arcPos = 0; +int arcs[1024]; +pos = str.find_first_of('.', left); +if (pos == 0) + { + left = 1; + pos = str.find_first_of('.', left); + } +while (pos != std::string::npos) + { + int arc = 0; + if (str2x(str.substr(left, left - pos), arc)) + { + return false; + } + arcs[arcPos++] = arc; + left = pos + 1; + pos = str.find_first_of('.', left); + } +if (left < str.length()) + { + int arc = 0; + if (str2x(str.substr(left, left - pos), arc)) + { + return false; + } + arcs[arcPos++] = arc; + } +printfd(__FILE__, "String2OI() - arcPos: %d\n", arcPos); +OBJECT_IDENTIFIER_set_arcs(oi, arcs, sizeof(arcs[0]), arcPos); +return true; +} + class SNMP_AGENT_CREATOR { private: @@ -54,7 +114,7 @@ return sac.GetPlugin(); int SendOpenPDU(int fd) { const char * description = "Stg SNMP Agent"; -int oid[] = {1, 3, 6, 1, 4, 1, 9363, 1, 5, 2, 1, 1}; +//int oid[] = {1, 3, 6, 1, 4, 1, 9363, 1, 5, 2, 1, 1}; asn_enc_rval_t error; OpenPDU_t msg; @@ -62,10 +122,15 @@ memset(&msg, 0, sizeof(msg)); msg.present = OpenPDU_PR_simple; asn_long2INTEGER(&msg.choice.simple.version, SimpleOpen__version_version_1); -OBJECT_IDENTIFIER_set_arcs(&msg.choice.simple.identity, +/*OBJECT_IDENTIFIER_set_arcs(&msg.choice.simple.identity, oid, sizeof(oid[0]), - 7); + 7);*/ +if (!String2OI(".1.3.6.1.4.1.9363", &msg.choice.simple.identity)) + { + printfd(__FILE__, "SendOpenPDU() - failed to convert string to OBJECT_IDENTIFIER\n"); + return -1; + } OCTET_STRING_fromString(&msg.choice.simple.description, description); OCTET_STRING_fromString(&msg.choice.simple.password, @@ -146,6 +211,90 @@ else return 0; } +int SendGetResponsePDU(int fd, const PDU_t * pdu) +{ +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, 0); +asn_long2INTEGER(&msg.error_index, 0); + +VarBind_t vb; + +memset(&vb, 0, sizeof(vb)); + +if (!String2OI(".1.3.6.1.4.1.9363.1", &vb.name)) + { + printfd(__FILE__, "SendSetResponsePDU() - failed to convert string to OBJECT_IDENTIFIER\n"); + return -1; + } + +ObjectSyntax_t * objectSyntax = &vb.value; +objectSyntax->present = ObjectSyntax_PR_simple; +SimpleSyntax_t * simpleSyntax = &objectSyntax->choice.simple; +simpleSyntax->present = SimpleSyntax_PR_number; +asn_long2INTEGER(&simpleSyntax->choice.number, 1); + + +ASN_SEQUENCE_ADD(&msg.variable_bindings, &vb); + +/*const VarBindList_t * vbl = &pdu->variable_bindings; +for (int i = 0; i < vbl->list.count; ++i) + { + VarBind_t * vb = pdu->variable_bindings.list.array[i]; + printfd(__FILE__, "OID: %s\n", OI2String(&vb->name).c_str()); + asn_fprint(stderr, &asn_DEF_ObjectSyntax, &vb->value); + }*/ + +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", + 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 SendSetResponsePDU(int fd, const PDU_t * pdu) +{ +//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, 0); + +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", + 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; +} + SMUX_PDUs_t * RecvSMUXPDUs(int fd) { char buffer[1024]; @@ -244,10 +393,14 @@ SNMP_AGENT::SNMP_AGENT() { pthread_mutex_init(&mutex, NULL); -handlers[SMUX_PDUs_PR_close] = &SNMP_AGENT::CloseHandler; -handlers[SMUX_PDUs_PR_registerResponse] = &SNMP_AGENT::RegisterResponseHandler; -handlers[SMUX_PDUs_PR_pdus] = &SNMP_AGENT::PDUsHandler; -handlers[SMUX_PDUs_PR_commitOrRollback] = &SNMP_AGENT::CommitOrRollbackHandler; +smuxHandlers[SMUX_PDUs_PR_close] = &SNMP_AGENT::CloseHandler; +smuxHandlers[SMUX_PDUs_PR_registerResponse] = &SNMP_AGENT::RegisterResponseHandler; +smuxHandlers[SMUX_PDUs_PR_pdus] = &SNMP_AGENT::PDUsHandler; +smuxHandlers[SMUX_PDUs_PR_commitOrRollback] = &SNMP_AGENT::CommitOrRollbackHandler; + +pdusHandlers[PDUs_PR_get_request] = &SNMP_AGENT::GetRequestHandler; +pdusHandlers[PDUs_PR_get_next_request] = &SNMP_AGENT::GetNextRequestHandler; +pdusHandlers[PDUs_PR_set_request] = &SNMP_AGENT::SetRequestHandler; } SNMP_AGENT::~SNMP_AGENT() @@ -404,9 +557,9 @@ return true; bool SNMP_AGENT::DispatchPDUs(const SMUX_PDUs_t * pdus) { -std::map::iterator it; -it = handlers.find(pdus->present); -if (it != handlers.end()) +SMUXHandlers::iterator it; +it = smuxHandlers.find(pdus->present); +if (it != smuxHandlers.end()) { return (this->*(it->second))(pdus); } @@ -449,6 +602,29 @@ bool SNMP_AGENT::PDUsHandler(const SMUX_PDUs_t * pdus) { printfd(__FILE__, "SNMP_AGENT::PDUsHandler()\n"); asn_fprint(stderr, &asn_DEF_SMUX_PDUs, pdus); +PDUsHandlers::iterator it; +it = pdusHandlers.find(pdus->choice.pdus.present); +if (it != pdusHandlers.end()) + { + return (this->*(it->second))(&pdus->choice.pdus); + } +else + { + switch (pdus->present) + { + case PDUs_PR_NOTHING: + printfd(__FILE__, "SNMP_AGENT::PDUsHandler() - nothing\n"); + break; + case PDUs_PR_get_response: + printfd(__FILE__, "SNMP_AGENT::PDUsHandler() - get response\n"); + break; + case PDUs_PR_trap: + printfd(__FILE__, "SNMP_AGENT::PDUsHandler() - trap\n"); + break; + default: + printfd(__FILE__, "SNMP_AGENT::PDUsHandler() - undefined\n"); + } + } return false; } @@ -458,3 +634,27 @@ printfd(__FILE__, "SNMP_AGENT::CommitOrRollbackHandler()\n"); asn_fprint(stderr, &asn_DEF_SMUX_PDUs, pdus); return false; } + +bool SNMP_AGENT::GetRequestHandler(const PDUs_t * pdus) +{ +printfd(__FILE__, "SNMP_AGENT::GetRequestHandler()\n"); +asn_fprint(stderr, &asn_DEF_PDUs, pdus); +SendGetResponsePDU(sock, &pdus->choice.get_request); +return false; +} + +bool SNMP_AGENT::GetNextRequestHandler(const PDUs_t * pdus) +{ +printfd(__FILE__, "SNMP_AGENT::GetNextRequestHandler()\n"); +asn_fprint(stderr, &asn_DEF_PDUs, pdus); +return false; +} + +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); +return false; +} +