summaryrefslogtreecommitdiffstats
path: root/src/common/code_environment.cc
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/common/code_environment.cc126
1 files changed, 126 insertions, 0 deletions
diff --git a/src/common/code_environment.cc b/src/common/code_environment.cc
new file mode 100644
index 000000000..14d55f60c
--- /dev/null
+++ b/src/common/code_environment.cc
@@ -0,0 +1,126 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph - scalable distributed file system
+ *
+ * Copyright (C) 2011 New Dream Network
+ *
+ * This is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License version 2.1, as published by the Free Software
+ * Foundation. See file COPYING.
+ *
+ */
+
+#include "common/code_environment.h"
+
+#include <iostream>
+
+#include "acconfig.h"
+
+#ifdef HAVE_PTHREAD_GETNAME_NP
+#include <pthread.h>
+#endif
+
+#include <string.h>
+
+code_environment_t g_code_env = CODE_ENVIRONMENT_UTILITY;
+
+extern "C" const char *code_environment_to_str(enum code_environment_t e)
+{
+ switch (e) {
+ case CODE_ENVIRONMENT_UTILITY:
+ return "CODE_ENVIRONMENT_UTILITY";
+ case CODE_ENVIRONMENT_DAEMON:
+ return "CODE_ENVIRONMENT_DAEMON";
+ case CODE_ENVIRONMENT_LIBRARY:
+ return "CODE_ENVIRONMENT_LIBRARY";
+ default:
+ return NULL;
+ }
+}
+
+std::ostream &operator<<(std::ostream &oss, const enum code_environment_t e)
+{
+ oss << code_environment_to_str(e);
+ return oss;
+}
+
+#if defined(HAVE_PTHREAD_GETNAME_NP) && !defined(_WIN32)
+
+int get_process_name(char *buf, int len)
+{
+ if (len <= 16) {
+ // The man page discourages using pthread_getname_np() with a buffer shorter
+ // than 16 bytes. With a 16-byte buffer, it might not be null-terminated.
+ return -ENAMETOOLONG;
+ }
+ // FIPS zeroization audit 20191115: this memset is not security related.
+ memset(buf, 0, len);
+ return pthread_getname_np(pthread_self(), buf, len);
+}
+
+#elif defined(HAVE_GETPROGNAME)
+
+int get_process_name(char *buf, int len)
+{
+ if (len <= 0) {
+ return -EINVAL;
+ }
+
+ const char *progname = getprogname();
+ if (progname == nullptr || *progname == '\0') {
+ return -ENOSYS;
+ }
+
+ strncpy(buf, progname, len - 1);
+ buf[len - 1] = '\0';
+ return 0;
+}
+
+#elif defined(_WIN32)
+
+int get_process_name(char *buf, int len)
+{
+ if (len <= 0) {
+ return -EINVAL;
+ }
+
+ char full_path[MAX_PATH];
+ int length = GetModuleFileNameA(nullptr, full_path, sizeof(full_path));
+ if (length <= 0)
+ return -ENOSYS;
+
+ char* start = strrchr(full_path, '\\');
+ if (!start)
+ return -ENOSYS;
+ start++;
+ char* end = strstr(start, ".exe");
+ if (!end)
+ return -ENOSYS;
+ if (len <= end - start) {
+ return -ENAMETOOLONG;
+ }
+
+ memcpy(buf, start, end - start);
+ buf[end - start] = '\0';
+ return 0;
+}
+
+#else
+
+int get_process_name(char *buf, int len)
+{
+ return -ENOSYS;
+}
+
+#endif
+
+std::string get_process_name_cpp()
+{
+ char buf[32];
+ if (get_process_name(buf, sizeof(buf))) {
+ return "(unknown)";
+ }
+ return std::string(buf);
+}