summaryrefslogtreecommitdiffstats
path: root/src/dcae_snmptrapd.py
diff options
context:
space:
mode:
authorLadue, David (dl3158) <dl3158@att.com>2018-03-14 14:01:36 -0400
committerLadue, David (dl3158) <dl3158@att.com>2018-03-18 02:42:23 -0400
commitabb283c5a3b72eb2d91e6adb4eda68184f040f35 (patch)
tree4765c8d6cb54a2ddb2052c0955425604bc265230 /src/dcae_snmptrapd.py
parenta6f91efda578965710630154b173cf1ee9b64727 (diff)
copyright and controller ENV changes
Change-Id: Ic12aa439a03f19c7a06a536368a8d2a1f26855aa Issue-ID: DCAEGEN2-271 Signed-off-by: Ladue, David (dl3158) <dl3158@att.com>
Diffstat (limited to 'src/dcae_snmptrapd.py')
-rw-r--r--src/dcae_snmptrapd.py832
1 files changed, 0 insertions, 832 deletions
diff --git a/src/dcae_snmptrapd.py b/src/dcae_snmptrapd.py
deleted file mode 100644
index 77043d8..0000000
--- a/src/dcae_snmptrapd.py
+++ /dev/null
@@ -1,832 +0,0 @@
-# ============LICENSE_START=======================================================)
-# org.onap.dcae
-# ================================================================================
-# Copyright (c) 2017-2018 AT&T Intellectual Property. All rights reserved.
-# ================================================================================
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============LICENSE_END=========================================================
-#
-# ECOMP is a trademark and service mark of AT&T Intellectual Property.
-#
-"""
-dcae_snmptrapd is responsible for SNMP trap receipt and publishing activities.
-It's behavior is controlled by several configs, the primary being:
-
- ../etc/trapd.yaml
-
-As traps arrive they are decomposed and transformed into a JSON message which
-is published to a dmaap instance that has been defined by controller.
-
-:Parameters:
- usage: dcae_snmptrapd.py -c <yaml_conf_file_name> [-v]
-:Keywords:
- onap dcae snmp trap publish dmaap
-"""
-
-__docformat__ = 'restructuredtext'
-
-# basics
-import argparse
-import array
-import asyncio
-from collections import Counter
-import datetime
-import errno
-import json
-import logging
-import logging.handlers
-import os
-import pprint
-import requests
-import re
-import sys
-import signal
-import string
-import socket
-import time
-import traceback
-import unicodedata
-import uuid as uuid_mod
-import yaml
-
-# pysnmp
-from pysnmp.entity import engine, config
-from pysnmp.carrier.asyncio.dgram import udp, udp6
-# from pysnmp.carrier.asyncore.dgram import udp
-from pysnmp.entity.rfc3413 import ntfrcv
-from pysnmp.proto.api import v2c
-
-# gen2 controller
-from onap_dcae_cbs_docker_client.client import get_config
-
-# dcae_snmptrap
-from trapd_runtime_pid import save_pid, rm_pid
-from trapd_yaml_config import read_yaml_config
-from trapd_trap_config import read_trap_config
-from trapd_dmaap_config import read_dmaap_config
-from trapd_exit import cleanup_and_exit
-from trapd_http_session import init_session_obj
-from trapd_perm_status import log_to_perm_status
-
-prog_name = os.path.basename(__file__)
-
-traps_in_second = 0
-last_epoch_second = 0
-
-# <dmaap.conf>
-dmaap_url = ""
-dmaap_user_name = ""
-dmaap_p_var = ""
-dmaap_stream_id = ""
-dmaap_host = ""
-# </dmaap.conf>
-
-# Requests session object
-dmaap_requests_session = None
-http_headers = {"Content-type": "application/json"}
-
-# <DNS cache>
-#
-# dns_cache_ip_to_name
-# key [ip address] -> fqdn
-# dns_cache_ip_expires
-# key [ip address] -> epoch time this entry expires and must be reloaded
-dns_cache_ip_to_name = {}
-dns_cache_ip_expires = {}
-# </DNS cache>
-
-# <trap config>
-num_trap_conf_entries = 0
-trap_conf_dict = {}
-# </trap config>
-
-pid_file_name = ""
-
-# logging
-dcae_logger = logging.getLogger('dcae_logger')
-handler = ""
-dcae_logger_max_bytes = 60000000
-dcae_logger_num_archives = 10
-
-undefined = "undefined"
-rc = 0
-usage_requested = False
-
-json_fd = None
-last_hour = -1
-
-verbose = False
-
-trap_dict = {}
-
-# # # # # # # # # # #
-# fx: usage_err
-# # # # # # # # # # #
-def usage_err():
- """
- Notify of incorrect (argument) usage
- :Parameters:
- none
- :Exceptions:
- none
- :Keywords:
- usage args
- """
-
- print('Correct usage:')
- print(' %s -c <yaml_conf_file_name> [-v]' % prog_name)
- cleanup_and_exit(1, "undefined")
-
-
-# # # # # # # # # # # # # # # # # # #
-# fx: setup dcae_logger custom logger
-# # # # # # # # # # ## # # # # # # #
-def setup_dcae_logger(_yc_trapd_diag):
- """
- Setup custom logger for dcae_snmptrapd that incorporates
- a rotating file handler with 10 backups of diagnostic messages
- :Parameters:
- _yc_trapd_diag - the full path output filename
- :Exceptions:
- none
- :Keywords:
- logging rotation
- """
-
- global dcae_logger
- global handler
-
- date_fmt = '%m/%d/%Y %H:%M:%S'
-
- _yc_trapd_diag_bak = "%s.bak" % (_yc_trapd_diag)
- if os.path.isfile(_yc_trapd_diag):
- os.rename(_yc_trapd_diag, _yc_trapd_diag_bak)
-
- # handler = logging.handlers.RotatingFileHandler(yc_trapd_diag, maxBytes=60000000, backupCount=10)
- handler = logging.handlers.RotatingFileHandler(_yc_trapd_diag,
- maxBytes=dcae_logger_max_bytes,
- backupCount=dcae_logger_num_archives)
-
- # set logLevel - valid values NOTSET, DEBUG, INFO, WARNING, ERROR, CRITICAL
- handler.setLevel(logging.DEBUG)
- dcae_logger.setLevel(logging.DEBUG)
-
- log_fmt = '%(levelname)s|%(asctime)s|%(name)s|%(process)d|%(funcName)s|'\
- '%(message)s'
- formatter = logging.Formatter(log_fmt)
- handler.setFormatter(formatter)
- dcae_logger.addHandler(handler)
-
- if os.path.isfile(_yc_trapd_diag):
- os.chmod(_yc_trapd_diag, 0o640)
-
- if os.path.isfile(_yc_trapd_diag_bak):
- os.chmod(_yc_trapd_diag_bak, 0o640)
-
-
-# # # # # # # # # # # # # # # # # # #
-# fx: load_all_configs
-# FIXME: currently on hold for load and signal handling convergence
-# # # # # # # # # # ## # # # # # # #
-def load_all_configs(_signum, _frame):
- """
- Calls individual functions to read various config files required. This
- function is called directly (e.g. at startup) and is also registered
- with signal handling (e.g. kill -sigusr1 <pid>)
-
- :Parameters:
- signum and frame (only present when called via signal to running process)
- :Exceptions:
- none
- :Keywords:
- config files
- :Variables:
- yaml_conf_file
- dmaap_requests_session
- """
-
- if int(_signum) != 0:
- dcae_logger.info("%s Received signal %s at frame %s; re-reading config file"
- % (prog_name, _signum, _frame))
- else:
- dcae_logger("Reading config files")
-
- # FIXME: should be re-reading all configs here
-
- # Initialize dmaap requests session object. Close existing session
- # if applicable.
- if dmaap_requests_session != None:
- dmaap_requests_session.close()
- dmaap_requests_session = init_session_obj(dcae_logger)
-
- return _yaml_config_values
-
-
-# # # # # # # # # # # # #
-# fx: rename_json_log
-# # # # # # # # # # # # #
-def rename_json_log(_outputFname):
- """
- Renames JSON output file to include ISO-formatted date suffix
- :Parameters:
- signum and frame (only present when called via signal to running process)
- :Exceptions:
- none
- :Keywords:
- json log
- :Variables:
- json log filename
- """
-
- global json_fd
-
- # check if outputfile exists; if it does, move to timestamped version
- _outputFnameBak = "%s.%s" % (_outputFname,
- datetime.datetime.fromtimestamp(time.time()).
- fromtimestamp(time.time()).
- strftime('%Y-%m-%dT%H:%M:%S'))
-
- # close existing file
- close_json_log()
-
- if os.path.isfile(_outputFname):
- dcae_logger.debug('Renaming %s to %s' %
- (_outputFname, _outputFnameBak))
- os.rename(_outputFname, _outputFnameBak)
- else:
- dcae_logger.error("Unable to move %s to %s - source file does not exist" %
- (_outputFname, _outputFnameBak))
- # open new (empty) log file
- try:
- json_fd = open_json_log()
- except:
- dcae_logger.exception(
- "Error opening new json log file %s - exiting " % _outputFname)
- sys.exit(1)
-
-# # # # # # # # # # # # #
-# fx: open_json_log
-# # # # # # # # # # # # #
-def open_json_log():
-
- try:
- # open append mode just in case so nothing is lost, but should be
- # non-existent file
- _json_fd = open(json_log_filename, 'a')
- dcae_logger.exception("Opened %s append mode: " % json_log_filename)
- return _json_fd
- except:
- dcae_logger.exception(
- "Error opening %s append mode: " % json_log_filename)
- sys.exit(1)
-
-
-# # # # # # # # # # # # #
-# fx: close_json_log
-# # # # # # # # # # # # #
-def close_json_log():
-
- global json_fd
-
- try:
- json_fd.close()
- except:
- dcae_logger.error("ERROR closing json audit file %s - results "
- "indeterminate" % (json_log_filename))
-
-# # # # # # # # # # # # #
-# fx: log_published_messages
-# # # # # # # # # # # # #
-
-
-def log_published_messages(loc_post_data_enclosed):
-
- # FIXME: should keep data dictionary of Fd's open, and reference those vs.
- # repeatedly opening append-mode
- # open json audit log file
-
- global json_fd, last_hour
-
- # close output file, backup current and move new one into place on day change
- dcae_logger.info('%.4f adding %s to json log' %
- (time.time(), trap_dict["uuid"]))
- curr_hour = datetime.datetime.now().hour
- if curr_hour < last_hour:
- rename_json_log(json_log_filename)
- json_fd = open_json_log(json_log_filename)
-
- try:
- m = loc_post_data_enclosed + '\n'
- json_fd.write('%s' % str(m))
- except Exception as e:
- dcae_logger.error("ERROR writing json audit file %s - message NOT LOGGED: %s"
- % (json_log_filename, str(e)))
-
- last_hour = curr_hour
- dcae_logger.info('%.4f logged %s' % (time.time(), trap_dict["uuid"]))
-
-# # # # # # # # # # # # #
-# fx: post_dmaap
-# # # # # # # # # # # # #
-
-
-def post_dmaap(dmaap_url, dmaap_user_name, dmaap_p_var, dmaap_stream_id, dmaap_host, uuid, traps_json_string):
- """
- Publish trap daata in json format to dmaap
- :Parameters:
- dmaap_url
- base url for http post
- dmaap_user_name
- username for http post
- dmaap_p_var
- access credential for http post
- dmaap_stream_id
- appended to dmaap_url, equiv to "topic"
- dmaap_host
- target dmaap server to submit http post
- uuid
- unique ID associated with this trap
- traps_json_string
- json format string to include in http post
- :Exceptions:
- none
- :Keywords:
- http post dmaap json message
- :Variables:
- """
-
- global http_resp, dmaap_requests_session, last_pub_time
-
- if dmaap_requests_session == None:
- dmaap_requests_session = init_session_obj(dcae_logger)
-
- post_data_enclosed = '[' + traps_json_string + ']'
-
- k = 0
- dmaap_pub_success = False
-
- if verbose:
- print('%.4f starting publish of %s' % (time.time(), trap_dict["uuid"]))
- dcae_logger.info('%.4f starting publish of %s' %
- (time.time(), trap_dict["uuid"]))
- while not dmaap_pub_success and k < yaml_config_values.yc_http_retries:
- try:
- dcae_logger.debug("Attempt %d to %s dmaap_url: "
- "%s dmaap_user_name: %s post_data: %s"
- % (k, dmaap_host,
- dmaap_url,
- dmaap_user_name,
- post_data_enclosed))
-
- # below disable_warnings required until python updated:
- # https://github.com/shazow/urllib3/issues/497
- # requests.packages.urllib3.disable_warnings()
- http_resp = dmaap_requests_session.post(dmaap_url, post_data_enclosed,
- auth=(dmaap_user_name,
- dmaap_p_var),
- headers=http_headers,
- timeout=yaml_config_values.yc_http_timeout)
- dcae_logger.debug("Response from %s on stream %s: %s dmaap_requests_session: %s"
- % (dmaap_host, dmaap_stream_id, http_resp.status_code, dmaap_requests_session))
- if verbose:
- print('%.4f published %s successfully' %
- (time.time(), trap_dict["uuid"]))
- dcae_logger.info('%.4f published %s successfully' %
- (time.time(), trap_dict["uuid"]))
- if http_resp.status_code == requests.codes.ok:
- dcae_logger.debug("Response from %s: %s dmaap_request_sesson: %s" % (
- dmaap_url, http_resp.status_code, dmaap_requests_session))
- log_published_messages(post_data_enclosed)
- last_pub_time = time.time()
- dmaap_pub_success = True
- break
- else:
- dcae_logger.debug("Response (non-200) detail from %s on stream "
- "%s: %s" % (dmaap_host, dmaap_stream_id, http_resp.text))
-
- except OSError as e:
- dcae_logger.debug("Exception while posting message to host: %s, stream: %s, dmaap_requests_session: %s, exception: %s %s"
- % (dmaap_host, dmaap_stream_id, dmaap_requests_session, e.errno, e.strerror))
- except requests.exceptions.RequestException as e:
- dcae_logger.error("Exception while posting to %s topic %s: -->%s<--"
- % (dmaap_host, dmaap_stream_id, e))
-
- k += 1
-
- if k < yaml_config_values.yc_http_retries:
- dcae_logger.error("sleeping %s and retrying" %
- yaml_config_values.yc_http_secs_between_retries)
- time.sleep(yaml_config_values.yc_http_secs_between_retries)
- else:
- dcae_logger.error("exhausted all attempts - giving up")
- break
-
- if verbose:
- print('%.4f exiting post_dmaap for %s' %
- (time.time(), trap_dict["uuid"]))
- dcae_logger.info('%.4f exiting post_dmaap for %s' %
- (time.time(), trap_dict["uuid"]))
- if not dmaap_pub_success:
- # uuid = uuid_mod.uuid1()
- perm_msg = "CRITICAL: publish failure to DMAAP server: "\
- "%s, stream: %s trap: %s" % (
- dmaap_host, dmaap_stream_id, uuid)
- dcae_logger.error(perm_msg)
- dcae_logger.error("SEND-TO-PERM-STATUS: %s" % perm_msg)
- log_to_perm_status(
- yaml_config_values.yc_perm_status_file, perm_msg, dcae_logger)
- dcae_logger.info("%.4f %s" % (time.time(), perm_msg))
- if verbose:
- print("%.4f %s" % (time.time(), perm_msg))
-
-
-# # # # # # # # # # # # # # # # # # #
-# fx: request_observer for community string rewrite
-# # # # # # # # # # # # # # # # # # #
-def comm_string_rewrite_observer(snmpEngine, execpoint, variables, cbCtx):
-
- # match ALL community strings
- if re.match('.*', str(variables['communityName'])):
- dcae_logger.debug('Rewriting communityName \'%s\' from %s into \'public\'' % (variables['communityName'], ':'.join([str(x) for x in
- variables['transportInformation'][1]])))
- variables['communityName'] = variables['communityName'].clone('public')
-
-# # # # # # # # # # # # # # # # # # #
-# fx: snmp_engine_observer_cb
-# callback for when trap is received
-# # # # # # # # # # # # # # # # # # #
-
-
-def snmp_engine_observer_cb(snmp_engine, execpoint, variables, cbCtx):
- """
- Decompose trap attributes and load in dictionary which is later used to
- create json string for publishing to dmaap.
- :Parameters:
- snmp_engine
- snmp engine created to listen for arriving traps
- execpoint
- point in code that snmp_engine_observer_cb was invoked
- variables
- trap attributes
- cbCtx
- callback context
- :Exceptions:
- none
- :Keywords:
- UEB non-AAF legacy http post
- :Variables:
- """
-
- global trap_dict, last_epoch_second, traps_in_epoch
-
- # init dictionary on new trap
- trap_dict = {}
-
- # assign uuid to trap
- trap_dict["uuid"] = str(uuid_mod.uuid1())
-
- if verbose:
- print('%.4f snmp trap arrived from %s, assigned uuid: %s' %
- (time.time(), variables['transportAddress'][0], trap_dict["uuid"]))
- dcae_logger.info('%.4f snmp trap arrived from %s, assigned uuid: %s' % (
- time.time(), variables['transportAddress'][0], trap_dict["uuid"]))
-
- # if re.match('.*', str(variables['communityName'])):
- # print('Rewriting communityName \'%s\' from %s into \'public\'' % (variables['communityName'], ':'.join([str(x) for x in variables['transportInformation'][1]])))
- # variables['communityName'] = variables['communityName'].clone('public')
-
- # ip and hostname
- ip_addr_str = str(variables['transportAddress'][0])
- trap_dict["agent address"] = ip_addr_str
- try:
- if int(dns_cache_ip_expires[ip_addr_str] < int(time.time())):
- dcae_logger.debug('dns cache expired for %s' % ip_addr_str)
- raise Exception('cache expired for %s at %d - updating value' %
- (ip_addr_str, (dns_cache_ip_expires[ip_addr_str])))
- else:
- trap_dict["agent name"] = dns_cache_ip_to_name[ip_addr_str]
- except:
- if verbose:
- print('%.4f dns cache expired for %s' % (time.time(), ip_addr_str))
- dcae_logger.debug(
- 'dns cache expired or missing for %s - reloading' % ip_addr_str)
- host_addr_info = socket.gethostbyaddr(ip_addr_str)
- agent_fqdn = str(host_addr_info[0])
- trap_dict["agent name"] = agent_fqdn
-
- dns_cache_ip_to_name[ip_addr_str] = agent_fqdn
- dns_cache_ip_expires[ip_addr_str] = (
- time.time() + yaml_config_values.yc_dns_cache_ttl_seconds)
- dcae_logger.debug('cache for %s (%s) updated - set to expire at %d' %
- (agent_fqdn, ip_addr_str, dns_cache_ip_expires[ip_addr_str]))
-
- dns_cache_ip_to_name[str(trap_dict["agent address"])]
-
- trap_dict["cambria.partition"] = str(trap_dict["agent name"])
- trap_dict["community"] = "" # do not include cleartext community in pub
- # do not include cleartext community in pub
- trap_dict["community len"] = 0
-
- # FIXME.CHECK_WITH_DOWNSTREAM_CONSUMERS: get rid of round for millisecond val
- # epoch_second = int(round(time.time()))
- epoch_msecond = time.time()
- epoch_second = int(round(epoch_msecond))
- if epoch_second == last_epoch_second:
- traps_in_epoch += 1
- else:
- traps_in_epoch = 0
- last_epoch_second = epoch_second
- traps_in_epoch_04d = format(traps_in_epoch, '04d')
- trap_dict['epoch_serno'] = int(
- (str(epoch_second) + str(traps_in_epoch_04d)))
-
- snmp_version = variables['securityModel']
- if snmp_version == 1:
- trap_dict["protocol version"] = "v1"
- else:
- if snmp_version == 2:
- trap_dict["protocol version"] = "v2c"
- else:
- if snmp_version == 3:
- trap_dict["protocol version"] = "v3"
- else:
- trap_dict["protocol version"] = "unknown"
-
- if snmp_version == 3:
- trap_dict["protocol version"] = "v3"
- trap_dict["security level"] = str(variables['securityLevel'])
- trap_dict["context name"] = str(variables['contextName'].prettyPrint())
- trap_dict["security name"] = str(variables['securityName'])
- trap_dict["security engine"] = str(
- variables['contextEngineId'].prettyPrint())
- trap_dict['time received'] = epoch_msecond
- # get this from dmaap_url when ready
- trap_dict['trap category'] = "DCAE-COLLECTOR-UCSNMP"
-
-
-# Callback function for receiving notifications
-# noinspection PyUnusedLocal,PyUnusedLocal,PyUnusedLocal
-def notif_receiver_cb(snmp_engine, stateReference, contextEngineId, contextName,
- varBinds, cbCtx):
- """
- Callback executed when trap arrives
- :Parameters:
- snmp_engine
- snmp engine created to listen for arriving traps
- stateReference
- contextEngineId
- contextName
- varBinds
- trap varbinds
- cbCtx
- callback context
- :Exceptions:
- none
- :Keywords:
- callback trap arrival
- :Variables:
- """
-
- global trap_dict
-
- if verbose:
- print('%.4f processing varbinds for %s' %
- (time.time(), trap_dict["uuid"]))
- dcae_logger.info('%.4f processing varbinds for %s' %
- (time.time(), trap_dict["uuid"]))
-
- # FIXME: add conversion from v1 to v2 prior to below? or special handling for v1?
-
- # FIXME update reset location when batching publishes
- vb_dict = {}
-
- vb_idx = 0
- k1 = ""
- k2 = ""
-
- # FIXME: Note that the vb type is present, just need to extract it efficiently somehow
- # print('\nvarBinds ==> %s' % (varBinds))
- #
- # varBinds ==> [(ObjectName('1.3.6.1.2.1.1.3.0'), TimeTicks(1243175676)),
- # (ObjectName('1.3.6.1.6.3.1.1.4.1.0'), ObjectIdentifier('1.3.6.1.4.1.74.2.46.12.1.1')),
- # (ObjectName('1.3.6.1.4.1.74.2.46.12.1.1.1'), OctetString(b'ucsnmp heartbeat - ignore')),
- # (ObjectName('1.3.6.1.4.1.74.2.46.12.1.1.2'), OctetString(b'Fri Aug 11 17:46:01 EDT 2017'))]
- #
- # This does NOT work:
- # for name, typ, val in varBinds:
- # print('name = %s' % (name))
- # print('typ = %s' % (typ))
- # print('val = %s\n' % (val))
-
- vb_all_string = ""
- for name, val in varBinds:
- vb_dict = {}
- if vb_idx == 0:
- vb_sys_uptime_oid = name
- vb_sys_uptime = val
- trap_dict["sysUptime"] = str(val)
- # print('vb_sys_uptime = %s' % (vb_sys_uptime))
- else:
- if vb_idx == 1:
- trap_dict["notify OID"] = str(val)
- trap_dict["notify OID len"] = (
- trap_dict["notify OID"].count('.') + 1)
- # print('vb_notify_oid = %s' % (vb_notify_oid))
- # else:
- # vb_idx_02d = format((vb_idx - 2), '02d')
- vb_idx_02d = format((vb_idx), '02d')
-
- k1 = "varbind_oid_" + str(vb_idx_02d)
- k2 = "varbind_value_" + str(vb_idx_02d)
- # vb_dict[k1] = name.prettyPrint()
- # vb_dict[k2] = val.prettyPrint()
- vb_dict["varbind_type"] = "tbd"
- vb_dict["varbind_oid"] = name.prettyPrint()
- vb_dict["varbind_value"] = val.prettyPrint()
- vb_json = json.dumps(vb_dict)
- vb_all_string += vb_json
-
- vb_idx += 1
-
- trap_dict["num varbinds"] = vb_idx
-
- # add varbind dict to trap dict
- # trap_dict["varbinds"] = vb_dict
- trap_dict["varbinds"] = vb_all_string
-
- dcae_logger.debug("vb_dict json-ized: %s" % (json.dumps(vb_dict)))
- trap_json_msg = json.dumps(trap_dict)
-
- # publish to dmaap after last varbind is processed
- post_dmaap(dmaap_config_values.dmaap_url, dmaap_config_values.dmaap_user_name, dmaap_config_values.dmaap_p_var,
- dmaap_config_values.dmaap_stream_id, dmaap_config_values.dmaap_host, trap_dict["uuid"], trap_json_msg)
-
-
-# # # # # # # # # # # # #
-# Main MAIN Main MAIN
-# # # # # # # # # # # # #
-# parse command line args
-parser = argparse.ArgumentParser(description='Post SNMP traps '
- 'to DCAE DMaap MR')
-parser.add_argument('-c', action="store", dest="yaml_conf_file", type=str,
- help="yaml config file name")
-parser.add_argument('-v', action="store_true", dest="verbose",
- help="verbose logging")
-parser.add_argument('-?', action="store_true", dest="usage_requested",
- help="show command line use")
-
-# set vars from args
-parser.set_defaults(yaml_conf_file="")
-
-# parse args
-args = parser.parse_args()
-
-# set vars from args
-yaml_conf_file = args.yaml_conf_file
-verbose = args.verbose
-usage_requested = args.usage_requested
-
-# if usage, just display and exit
-if usage_requested:
- usage_err()
-
-# Get non-ENV settings from config file; spoof 2 params
-# so same fx can be used for signal handling
-if yaml_conf_file == "":
- usage_err()
-
-# always get yaml config values
-yaml_config_values = read_yaml_config(yaml_conf_file)
-
-# setup custom logger
-setup_dcae_logger(yaml_config_values.yc_trapd_diag)
-
-# bump up logging level if overridden at command line
-if verbose:
- dcae_logger.setLevel(logging.DEBUG)
- handler.setLevel(logging.DEBUG)
- dcae_logger.debug("log level increased to DEBUG")
-
-dcae_logger.info("log will include info level messages")
-dcae_logger.error("log will include error level messages")
-dcae_logger.debug("log will include debug level messages")
-dcae_logger.info("Runtime PID file: %s" % pid_file_name)
-
-# setup signal handling for config file reload
-# FIXME: need to have signal handler return all tuples for configs
-# signal.signal(signal.SIGUSR1, load_all_configs)
-
-# save current PID for future/external reference
-pid_file_name = '%s/%s.pid' % (yaml_config_values.yc_pid_dir, prog_name)
-rc = save_pid(pid_file_name)
-
-# always get trap configs
-trap_config_values = read_trap_config(
- yaml_config_values.yc_trap_conf, dcae_logger)
-
-# Set initial startup hour for rolling logfile
-last_hour = datetime.datetime.now().hour
-
-#make sure my env is set properly
-try:
- c = get_config()
- if c == {}:
- msg = "Unable to fetch configuration or it is erroneously empty - fatal ONAP controller error, trying OpenDCAE config"
- dcae_logger.error(msg)
- print('%s' % msg)
-
-#if new controller not present, try dmaap.conf
-except:
- msg = "ONAP controller not present, attempting OpenDCAE dmaap.conf config"
- dcae_logger.error(msg)
- dmaap_config_values = read_dmaap_config(
- yaml_config_values.yc_dmaap_conf, dcae_logger)
-
- # get the topic from the url
- dmaap_topic = dmaap_config_values.dmaap_url.split('.')[-1]
- dcae_logger.info("Topic: %s" % dmaap_topic)
- json_log_filename = yaml_config_values.yc_published_traps_dir + '/' + 'DMAAP' + '_' \
- + dmaap_topic + '.json'
- json_fd = open_json_log()
- msg = "Using OpenDCAE dmaap.conf config"
-
-# Get the event loop for this thread
-loop = asyncio.get_event_loop()
-
-# Create SNMP engine with autogenernated engineID pre-bound
-# to socket transport dispatcher
-snmp_engine = engine.SnmpEngine()
-
-# # # # # # # # # # # #
-# Transport setup
-# # # # # # # # # # # #
-
-# UDP over IPv4
-# FIXME: add check for presense of ipv4_interface prior to attempting add OR just put entire thing in try/except clause
-config.addTransport(
- snmp_engine,
- udp.domainName + (1,),
- udp.UdpTransport().openServerMode(
- (yaml_config_values.yc_ipv4_interface, yaml_config_values.yc_ipv4_port))
-)
-
-# UDP over IPv6
-# FIXME: add check for presense of ipv6_interface prior to attempting add OR just put entire thing in try/except clause
-config.addTransport(
- snmp_engine,
- udp6.domainName,
- udp6.Udp6Transport().openServerMode(
- (yaml_config_values.yc_ipv6_interface, yaml_config_values.yc_ipv6_port))
-)
-
-# UDP over IPv4, second listening interface/port
-# config.addTransport(
-# snmp_engine,
-# udp.domainName + (2,),
-# udp.UdpTransport().openServerMode(('127.0.0.1', 2162))
-# )
-
-# # # # # # # # # # # #
-# SNMPv1/2c setup
-# # # # # # # # # # # #
-
-# SecurityName <-> CommunityName mapping
-# to restrict trap reception to only those with specific community
-# strings
-config.addV1System(snmp_engine, 'my-area', 'public')
-
-# register comm_string_rewrite_observer for message arrival
-snmp_engine.observer.registerObserver(
- comm_string_rewrite_observer,
- 'rfc2576.processIncomingMsg:writable'
-)
-
-# register snmp_engine_observer_cb for message arrival
-snmp_engine.observer.registerObserver(
- snmp_engine_observer_cb,
- 'rfc3412.receiveMessage:request',
- 'rfc3412.returnResponsePdu',
-)
-
-# Register SNMP Application at the SNMP engine
-ntfrcv.NotificationReceiver(snmp_engine, notif_receiver_cb)
-
-snmp_engine.transportDispatcher.jobStarted(1) # loop forever
-
-# Run I/O dispatcher which would receive queries and send confirmations
-try:
- snmp_engine.transportDispatcher.runDispatcher()
-except:
- snmp_engine.observer.unregisterObserver()
- snmp_engine.transportDispatcher.closeDispatcher()
- cleanup_and_exit(1, pid_file_name)