memusage.cc

Go to the documentation of this file.

/*
 * Copyright 2009 Savarese Software Research Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     https://www.savarese.com/software/ApacheLicense-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <ssrc/wispers/utility/memusage.h>

// TODO: Implement for FreeBSD.
// TODO: exceptions?  errors?  scale all values to bytes?
#if defined(WISPERS_HAVE_GET_MEMUSAGE)

#include <sstream>

#  if defined(linux) || defined(__linux) || defined(__linux__)

#  include <fstream>

__BEGIN_NS_SSRC_WSPR_UTILITY

bool get_memusage(const unsigned int pid, memusage & usage) {
  std::ostringstream buf;

  buf << "/proc/" << pid << "/statm";

  std::ifstream in(buf.str().c_str());

  if(in) {
    in >> usage.size >> usage.resident >> usage.share
       >> usage.text >> usage.lib >> usage.data >> usage.dt;

    usage.page_size = ::sysconf(_SC_PAGE_SIZE);

    return true;
  }

  return false;
}

__END_NS_SSRC_WSPR_UTILITY

#  elif defined(sun) || defined(__sun) || defined(__sun__)

#  include <fcntl.h>
#  include <procfs.h>

__BEGIN_NS_SSRC_WSPR_UTILITY

bool get_memusage(const unsigned int pid, memusage & usage) {
  ::psinfo_t psinfo;
  ::pstatus_t pstatus;
  std::ostringstream buf;

  buf << "/proc/" << pid << "/status";

  int fd = ::open(buf.str().c_str(), O_RDONLY);

  if(fd < 0)
    return false;

  if(::read(fd, &pstatus, sizeof(pstatus_t)) != sizeof(pstatus_t))
    return false;

  ::close(fd);

  buf.str("");
  buf << "/proc/" << pid << "/psinfo";

  fd = ::open(buf.str().c_str(), O_RDONLY);

  if(fd < 0)
    return false;

  if(::read(fd, &psinfo, sizeof(psinfo_t)) != sizeof(psinfo_t))
    return false;

  ::close(fd);

  usage.size     = psinfo.pr_size;
  usage.resident = psinfo.pr_rssize;
  usage.heap     = pstatus.pr_brksize;
  usage.stack    = pstatus.pr_stksize;

  return true;
}

__END_NS_SSRC_WSPR_UTILITY

#  endif

#else

__BEGIN_NS_SSRC_WSPR_UTILITY

// Dummy function to allow module to compile.
bool get_memusage(const unsigned int pid) { return false; }

__END_NS_SSRC_WSPR_UTILITY

#endif