ACE/ASNMP/asnmp/sagent.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    sagent.cpp
   5  *
   6  * SNMP agent class definition. The sagent class provides an object oriented
   7  * approach for creating SNMP Agents. The sagent class is an encapsulation
   8  * of SNMP sessions, gets, sets, etc.
   9  *
  10  *  @author Michael R. MacFaden
  11  */
  12 //=============================================================================
  13
  14 #include "ace/Reactor.h"
  15 #include "ace/SOCK_Dgram.h"
  16
  17 #include "asnmp/oid.h"                // snmp++ oid class
  18 #include "asnmp/vb.h"                 // snbmp++ vb class
  19 #include "asnmp/target.h"             // snmp++ target class
  20 #include "asnmp/pdu.h"                // snmp++ pdu class
  21 #include "asnmp/snmperrs.h"           // error macros and strings
  22 #include "asnmp/address.h"            // snmp++ address class defs
  23 #include "asnmp/snmp.h"               // manager snmp interface
  24 #include "asnmp/sagent.h"             // agent interface
  25 #include "asnmp/transaction.h"        // convert from wire to API
  26
  27 sagent::sagent(unsigned short port): Snmp(port)
  28 {
  29   ACE_TRACE("sagent::sagent(short)");
  30 }
  31
  32 sagent::~sagent()
  33 {
  34   ACE_TRACE("sagent::~sagent");
  35 }
  36
  37 int sagent::handle_input(ACE_HANDLE fd)
  38 {
  39   ACE_TRACE("sagent::handle_input");
  40
  41   transaction tr(iv_snmp_session_); // this section needs a better design
  42   tr.handle_input(fd);
  43   char rcv_com_str[MAX_COMM_STR_LEN];
  44   if (tr.result(pdu_, rcv_com_str) < 0)
  45      return 0;
  46   OctetStr community(rcv_com_str);
  47   const ACE_INET_Addr &ta = tr.get_from_addr();
  48   ACE_TCHAR buf_tmp[MAXHOSTNAMELEN + 1];
  49   ta.addr_to_string (buf_tmp, MAXHOSTNAMELEN);
  50   UdpAddress ra (ACE_TEXT_ALWAYS_CHAR (buf_tmp));
  51   tgt_.set_address(ra);
  52
  53
  54   // process msg here by calling subclass's implementation
  55   switch (pdu_.get_type()){
  56   case sNMP_PDU_GET:
  57     tgt_.set_read_community(community);
  58     this->handle_get(pdu_, tgt_);
  59     break;
  60
  61   case sNMP_PDU_GETNEXT:
  62     tgt_.set_read_community(community);
  63     this->handle_get_next(pdu_, tgt_);
  64     break;
  65
  66   case sNMP_PDU_SET:
  67     tgt_.set_write_community(community);
  68     this->handle_set(pdu_, tgt_);
  69     break;
  70
  71   default:
  72     ACE_ASSERT(0);
  73   }
  74   return 0;
  75 }
  76
  77 ACE_HANDLE sagent::get_handle() const
  78 {
  79   ACE_TRACE("sagent::get_handle");
  80   return iv_snmp_session_.get_handle();
  81 }
  82
  83 int sagent::respond(Pdu& pdu,UdpTarget& tgt)
  84 {
  85   pdu.set_type(sNMP_PDU_RESPONSE);
  86   transaction tr(pdu, tgt, iv_snmp_session_);
  87   tr.send();
  88   return 0;
  89 }