1 files changed, 150 insertions, 0 deletions
diff --git a/accel-pppd/extra/net-snmp/agent.c b/accel-pppd/extra/net-snmp/agent.c
new file mode 100644
index 0000000..dd38b44
--- /dev/null
+++ b/accel-pppd/extra/net-snmp/agent.c
@@ -0,0 +1,150 @@
+#include <pthread.h>
+#include <signal.h>
+
+#include <net-snmp/net-snmp-config.h>
+#include <net-snmp/net-snmp-includes.h>
+#include <net-snmp/agent/net-snmp-agent-includes.h>
+
+#include "log.h"
+#include "triton.h"
+
+#include "statCore.h"
+#include "statPPP.h"
+#include "statPPTP.h"
+#include "statL2TP.h"
+#include "statPPPOE.h"
+#include "terminate.h"
+#include "shutdown.h"
+#include "sessionTable.h"
+
+static const char *conf_agent_name = "accel-ppp";
+static int conf_master = 0;
+/*static const char *conf_oid_prefix = "1.3.6.1.4.1.8072.100";
+
+static oid* oid_prefix;
+static size_t oid_prefix_size;*/
+
+static pthread_t snmp_thr;
+static int snmp_term = 0;
+
+/*int accel_ppp_alloc_oid(oid tail, size_t size, oid **oid)
+{
+	*oid = malloc(sizeof(oid) * (oid_prefix_size + size));
+
+	memcpy(*oid, oid_prefix, oid_prefix_size);
+	memcpy((*oid) + oid_prefix_size, tail, size);
+
+	return oid_prefix_size + size;
+}*/
+
+static int agent_log(int major, int minor, void *serv_arg, void *cl_arg)
+{
+	struct snmp_log_message *m = serv_arg;
+	
+	switch (m->priority) {
+		case LOG_EMERG:
+			log_emerg("net-snmp: %s", m->msg);
+			break;
+		case LOG_ALERT:
+		case LOG_CRIT:
+		case LOG_ERR:
+			log_error("net-snmp: %s", m->msg);
+			break;
+		case LOG_WARNING:
+			log_warn("net-snmp: %s", m->msg);
+			break;
+		case LOG_NOTICE:
+			log_info1("net-snmp: %s", m->msg);
+			break;
+		case LOG_INFO:
+			log_info2("net-snmp: %s", m->msg);
+			break;
+		case LOG_DEBUG:
+			log_debug("net-snmp: %s", m->msg);
+			break;
+		default:
+			log_msg("net-snmp: %s", m->msg);
+	}
+	return 0;
+}
+
+static void *snmp_thread(void *a)
+{
+	sigset_t set;
+
+	sigfillset(&set);
+	sigdelset(&set, SIGKILL);
+	sigdelset(&set, SIGSTOP);
+	pthread_sigmask(SIG_BLOCK, &set, NULL);
+
+	snmp_register_callback(SNMP_CALLBACK_LIBRARY, SNMP_CALLBACK_LOGGING, agent_log, NULL);
+  snmp_disable_log();
+	snmp_enable_calllog();
+	//snmp_set_do_debugging(1);
+	//netsnmp_ds_set_boolean(NETSNMP_DS_APPLICATION_ID, NETSNMP_DS_AGENT_ROLE, 1);
+	
+	if (!conf_master)
+		netsnmp_enable_subagent();
+
+	init_agent(conf_agent_name);
+
+	init_statCore();
+	init_statPPP();
+	init_statPPTP();
+	init_statL2TP();
+	init_statPPPOE();
+	init_terminate();
+	init_shutdown();
+	init_sessionTable();
+
+	init_snmp(conf_agent_name);
+
+	if (conf_master)
+		init_master_agent();
+	
+	while (!snmp_term) {
+    agent_check_and_process(1);
+	}
+	
+	snmp_shutdown(conf_agent_name);
+
+  SOCK_CLEANUP;
+
+	return NULL;
+}
+
+static void snmp_ctx_close(struct triton_context_t *ctx)
+{
+	snmp_term = 1;
+	snmp_shutdown(conf_agent_name);
+	triton_context_unregister(ctx);
+}
+
+static struct triton_context_t ctx = {
+	.close = snmp_ctx_close,
+};
+
+static void init(void)
+{
+	const char *opt;
+
+	opt = conf_get_opt("snmp", "master");
+	if (opt)
+		conf_master = atoi(opt);
+
+	opt = conf_get_opt("snmp", "agent-name");
+	if (opt)
+		conf_agent_name = opt;
+	
+	/*opt = conf_get_opt("snmp", "oid-prefix")
+	if (opt)
+		conf_oid_prefix = opt;*/
+	
+	pthread_create(&snmp_thr, NULL, snmp_thread, NULL);
+	triton_context_register(&ctx, NULL);
+	triton_context_wakeup(&ctx);
+	triton_collect_cpu_usage();
+}
+
+DEFINE_INIT(100, init);
+