ceph/src/common/code_environment.cc

   1 // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
   2 // vim: ts=8 sw=2 smarttab
   3 /*
   4  * Ceph - scalable distributed file system
   5  *
   6  * Copyright (C) 2011 New Dream Network
   7  *
   8  * This is free software; you can redistribute it and/or
   9  * modify it under the terms of the GNU Lesser General Public
  10  * License version 2.1, as published by the Free Software
  11  * Foundation.  See file COPYING.
  12  *
  13  */
  14
  15 #include "common/code_environment.h"
  16
  17 #include <iostream>
  18
  19 #include "acconfig.h"
  20
  21 #ifdef HAVE_SYS_PRCTL_H
  22 #include <sys/prctl.h>
  23 #endif
  24
  25 #include <string.h>
  26
  27 code_environment_t g_code_env = CODE_ENVIRONMENT_UTILITY;
  28
  29 extern "C" const char *code_environment_to_str(enum code_environment_t e)
  30 {
  31   switch (e) {
  32     case CODE_ENVIRONMENT_UTILITY:
  33       return "CODE_ENVIRONMENT_UTILITY";
  34     case CODE_ENVIRONMENT_DAEMON:
  35       return "CODE_ENVIRONMENT_DAEMON";
  36     case CODE_ENVIRONMENT_LIBRARY:
  37       return "CODE_ENVIRONMENT_LIBRARY";
  38     default:
  39       return NULL;
  40   }
  41 }
  42
  43 std::ostream &operator<<(std::ostream &oss, const enum code_environment_t e)
  44 {
  45   oss << code_environment_to_str(e);
  46   return oss;
  47 }
  48
  49 #if defined(HAVE_SYS_PRCTL_H) && defined(PR_GET_NAME) /* Since 2.6.11 */
  50
  51 int get_process_name(char *buf, int len)
  52 {
  53   if (len <= 16) {
  54     /* The man page discourages using this prctl with a buffer shorter
  55      * than 16 bytes. With a 16-byte buffer, it might not be
  56      * null-terminated. */
  57     return -ENAMETOOLONG;
  58   }
  59   memset(buf, 0, len);
  60   return prctl(PR_GET_NAME, buf);
  61 }
  62
  63 #elif defined(HAVE_GETPROGNAME)
  64
  65 int get_process_name(char *buf, int len)
  66 {
  67   if (len <= 0) {
  68     return -EINVAL;
  69   }
  70
  71   const char *progname = getprogname();
  72   if (progname == nullptr || *progname == '\0') {
  73     return -ENOSYS;
  74   }
  75
  76   strncpy(buf, progname, len - 1);
  77   buf[len - 1] = '\0';
  78   return 0;
  79 }
  80
  81 #else
  82
  83 int get_process_name(char *buf, int len)
  84 {
  85   return -ENOSYS;
  86 }
  87
  88 #endif
  89
  90 std::string get_process_name_cpp()
  91 {
  92   char buf[32];
  93   if (get_process_name(buf, sizeof(buf))) {
  94     return "(unknown)";
  95   }
  96   return std::string(buf);
  97 }