Rename Project to Open5GS

- Change BuildTool to Meson - Many BUGS Fixed
2026-05-01 21:00:19 +00:00 · 2019-10-27 17:41:14 +09:00 · 2019-10-27 17:41:14 +09:00 · 142763c280
commit 142763c280
parent cb447af93a
919 changed files with 30459 additions and 6284 deletions
--- a/lib/diameter/common/logger.c
+++ b/lib/diameter/common/logger.c
@ -0,0 +1,177 @@
+/*
+ * Copyright (C) 2019 by Sukchan Lee <acetcom@gmail.com>
+ *
+ * This file is part of Open5GS.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include "ogs-diameter-common.h"
+
+static struct ogs_diam_logger_t self;
+
+static struct fd_hook_hdl *logger_hdl = NULL;
+static pthread_mutex_t mtx = PTHREAD_MUTEX_INITIALIZER;
+static pthread_t fd_stats_th = (pthread_t)NULL;
+
+static ogs_diam_logger_user_handler user_handler = NULL;
+
+static void ogs_diam_logger_cb(enum fd_hook_type type, struct msg * msg, 
+    struct peer_hdr * peer, void * other, struct fd_hook_permsgdata *pmd, 
+    void * regdata);
+static void * diam_stats_worker(void * arg);
+
+int ogs_diam_logger_init(int mode)
+{
+	uint32_t mask_peers = HOOK_MASK( HOOK_PEER_CONNECT_SUCCESS );
+
+    memset(&self, 0, sizeof(struct ogs_diam_logger_t));
+
+    self.mode = mode;
+	self.duration = 60;       /* 60 seconds */
+	
+    CHECK_FCT( fd_hook_register( 
+            mask_peers, ogs_diam_logger_cb, NULL, NULL, &logger_hdl) );
+
+	CHECK_POSIX( pthread_mutex_init(&self.stats_lock, NULL) );
+
+	return 0;
+}
+
+void ogs_diam_logger_final()
+{
+	CHECK_FCT_DO( fd_thr_term(&fd_stats_th), );
+	CHECK_POSIX_DO( pthread_mutex_destroy(&self.stats_lock), );
+
+	if (logger_hdl) { CHECK_FCT_DO( fd_hook_unregister( logger_hdl ), ); }
+}
+
+struct ogs_diam_logger_t* ogs_diam_logger_self()
+{
+    return &self;
+}
+
+int ogs_diam_logger_stats_start()
+{
+	/* Start the statistics thread */
+	CHECK_POSIX( pthread_create(&fd_stats_th, NULL, diam_stats_worker, NULL) );
+
+    return 0;
+}
+
+void ogs_diam_logger_register(ogs_diam_logger_user_handler instance)
+{
+    user_handler = instance;
+}
+
+void ogs_diam_logger_unregister(void)
+{
+    user_handler = NULL;
+}
+
+/* The callback called when messages are received and sent */
+static void ogs_diam_logger_cb(enum fd_hook_type type, struct msg * msg, 
+    struct peer_hdr * peer, void * other, struct fd_hook_permsgdata *pmd, 
+    void * regdata)
+{
+	const char * peer_name = peer ? peer->info.pi_diamid : "<unknown peer>";
+	
+	CHECK_POSIX_DO( pthread_mutex_lock(&mtx), );
+
+    if (user_handler)
+    {
+        user_handler(type, msg, peer, other, pmd, regdata);
+    }
+
+	switch (type) {
+/* peers */
+	case HOOK_PEER_CONNECT_SUCCESS:
+		{
+			char protobuf[40];
+			if (peer) {
+				CHECK_FCT_DO(fd_peer_cnx_proto_info(peer, protobuf, sizeof(protobuf)), break );
+			} else {
+				protobuf[0] = '-';
+				protobuf[1] = '\0';
+			}
+			ogs_info("CONNECTED TO '%s' (%s):", peer_name, protobuf);
+		}
+		break;
+    default:
+        ogs_warn("Unknown type(%d)", type);
+		break;
+	}
+	
+	CHECK_POSIX_DO( pthread_mutex_unlock(&mtx), );
+}
+
+/* Function to display statistics periodically */
+static void * diam_stats_worker(void * arg) 
+{
+	struct timespec start, now;
+	struct fd_stats copy;
+	
+	/* Get the start time */
+	CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &start), );
+	
+	/* Now, loop until canceled */
+	while (1) {
+		/* Display statistics every XX seconds */
+		sleep(self.duration);
+		
+		/* Now, get the current stats */
+		CHECK_POSIX_DO( pthread_mutex_lock(&self.stats_lock), );
+		memcpy(&copy, &self.stats, sizeof(struct fd_stats));
+		CHECK_POSIX_DO( pthread_mutex_unlock(&self.stats_lock), );
+		
+		/* Get the current execution time */
+		CHECK_SYS_DO( clock_gettime(CLOCK_REALTIME, &now), );
+		
+		/* Now, display everything */
+		ogs_trace("------- fd statistics ---------");
+		if (now.tv_nsec >= start.tv_nsec) 
+        {
+			ogs_trace(" Executing for: %d.%06ld sec",
+					(int)(now.tv_sec - start.tv_sec),
+					(long)(now.tv_nsec - start.tv_nsec) / 1000);
+		} 
+        else 
+        {
+			ogs_trace(" Executing for: %d.%06ld sec",
+					(int)(now.tv_sec - 1 - start.tv_sec),
+					(long)(now.tv_nsec + 1000000000 - start.tv_nsec) / 1000);
+		}
+		
+        if (self.mode & FD_MODE_SERVER) {
+            ogs_trace(" Server: %llu message(s) echoed", 
+                    copy.nb_echoed);
+        }
+        if (self.mode & FD_MODE_CLIENT) {
+            ogs_trace(" Client:");
+            ogs_trace("   %llu message(s) sent", copy.nb_sent);
+            ogs_trace("   %llu error(s) received", copy.nb_errs);
+            ogs_trace("   %llu answer(s) received", copy.nb_recv);
+            ogs_trace("     fastest: %ld.%06ld sec.", 
+                    copy.shortest / 1000000, copy.shortest % 1000000);
+            ogs_trace("     slowest: %ld.%06ld sec.", 
+                    copy.longest / 1000000, copy.longest % 1000000);
+            ogs_trace("     Average: %ld.%06ld sec.", 
+                    copy.avg / 1000000, copy.avg % 1000000);
+        }
+		ogs_trace("-------------------------------------");
+	}
+	
+	return NULL; /* never called */
+}
+