lock mutex on reading program status

dynatrace-oss · Oct 25, 2023 · 89d7070 · 89d7070
1 parent 662f7b6
commit 89d7070
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ebpfdiscoverysrv/src/main.cpp b/ebpfdiscoverysrv/src/main.cpp
@@ -111,8 +111,13 @@ static void initLibbpf() {
 	libbpf_set_print(libbpfPrintFn);
 }
 
+bool isProgramRunning() {
+	std::lock_guard<std::mutex> lock(programStatusMutex);
+	return programStatus == ProgramStatus::Running;
+}
+
 void servicesProvidingLoop(ebpfdiscovery::Discovery& discoveryInstance, std::chrono::seconds interval) {
-	while (programStatus == ProgramStatus::Running) {
+	while (isProgramRunning()) {
 		if (auto services = discoveryInstance.popServices(); !services.empty()) {
 			auto servicesProto = proto::internalToProto(services);
 			LOG_DEBUG("Services list:\n{}\n", servicesProto.DebugString());