rmem 1.0.0

data/LICENSE

@@ -0,0 +1,51 @@
+Copyright (c) 2009 Todd A Fisher (todd DOT fisher AT gmail DOT com ).
+Rmem is free software licensed under the following terms:
+
+  1. You may make and give away verbatim copies of the source form of the
+     software without restriction, provided that you duplicate all of the
+     original copyright notices and associated disclaimers.
+
+  2. You may modify your copy of the software in any way, provided that
+     you do at least ONE of the following:
+
+       a) place your modifications in the Public Domain or otherwise
+          make them Freely Available, such as by posting said
+	  modifications to Usenet or an equivalent medium, or by allowing
+	  the author to include your modifications in the software.
+
+       b) use the modified software only within your corporation or
+          organization.
+
+       c) give non-standard binaries non-standard names, with
+          instructions on where to get the original software distribution.
+
+       d) make other distribution arrangements with the author.
+
+  3. You may distribute the software in object code or binary form,
+     provided that you do at least ONE of the following:
+
+       a) distribute the binaries and library files of the software,
+	  together with instructions (in the manual page or equivalent)
+	  on where to get the original distribution.
+
+       b) accompany the distribution with the machine-readable source of
+	  the software.
+
+       c) give non-standard binaries non-standard names, with
+          instructions on where to get the original software distribution.
+
+       d) make other distribution arrangements with the author.
+
+  4. You may modify and include the part of the software into any other
+     software (possibly commercial).
+
+  5. The scripts and library files supplied as input to or produced as 
+     output from the software do not automatically fall under the
+     copyright of the software, but belong to whomever generated them, 
+     and may be sold commercially, and may be aggregated with this
+     software.
+
+  6. THIS SOFTWARE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR
+     IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
+     WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+     PURPOSE.

data/README

@@ -0,0 +1,11 @@
+Ruby Memory Measure
+-------------------
+
+Ever wonder how much memory your process is using?  Have you ever wanted to know at specific points of execution what the memory usage looks like? 
+This should give you a similar sense of the process memory usage as looking at top, task manager or activity monitor.  The advantage is you don't have
+to run an external tool to capture what's going on inside your process.
+
+ bytes = RMem::Report.memory
+ puts format("footprint: %d bytes, %.2f kbytes, and %.4f MB", bytes, bytes/1024.0, bytes/1024.0/1024.0)
+
+For ruby see test.rb

data/Rakefile

@@ -0,0 +1,61 @@
+
+require 'rake/clean'
+CURRENT_VERSION = "1.0.0"
+RMEM_SO = "ext/rmem.#{Config::MAKEFILE_CONFIG['DLEXT']}"
+# Make tasks -----------------------------------------------------
+MAKECMD = ENV['MAKE_CMD'] || 'make'
+MAKEOPTS = ENV['MAKE_OPTS'] || ''
+CLEAN.include '**/*.o'
+CLEAN.include "**/*.#{Config::MAKEFILE_CONFIG['DLEXT']}"
+CLOBBER.include '**/*.log'
+CLOBBER.include '**/Makefile'
+CLOBBER.include '**/extconf.h'
+CLOBBER.include '**/rmem_config.h'
+
+file 'ext/Makefile' => 'ext/extconf.rb' do
+  Dir.chdir('ext') do
+    ruby "extconf.rb #{ENV['EXTCONF_OPTS']}"
+  end
+end
+
+def make(target = '')
+  Dir.chdir('ext') do 
+    pid = system("#{MAKECMD} #{MAKEOPTS} #{target}")
+    $?.exitstatus
+  end    
+end
+
+# Let make handle dependencies between c/o/so - we'll just run it. 
+file RMEM_SO => (['ext/Makefile'] + Dir['ext/*.c'] + Dir['ext/*.h']) do
+  m = make
+  fail "Make failed (status #{m})" unless m == 0
+end
+
+desc "Compile the shared object"
+task :compile => [RMEM_SO]
+
+task :default => :compile
+
+desc 'Generate gem specification'
+task :gemspec => :clobber do
+  require 'erb'
+  tspec = ERB.new(File.read(File.join(File.dirname(__FILE__),'ext','rmem.gemspec.erb')))
+  File.open(File.join(File.dirname(__FILE__),'rmem.gemspec'),'wb') do|f|
+    f << tspec.result
+  end
+end
+
+if ! defined?(Gem)
+  warn "Package Target requires RubyGEMs"
+else
+  desc 'Build gem'
+  task :package => :gemspec do
+    require 'rubygems/specification'
+    spec_source = File.read File.join(File.dirname(__FILE__),'rmem.gemspec')
+    spec = nil
+    # see: http://gist.github.com/16215
+    Thread.new { spec = eval("$SAFE = 3\n#{spec_source}") }.join
+    spec.validate
+    Gem::Builder.new(spec).build
+  end
+end

data/ext/extconf.rb

@@ -0,0 +1,13 @@
+require 'mkmf'
+
+dir_config('rmem')
+checking_for '/proc file system' do
+  $defs.push( "-D HAVE_PROC" ) if File.exist?('/proc')
+end
+have_header('libproc.h')
+have_header('windows.h')
+have_header('psapi.h')
+have_library('psapi')
+
+create_header('rmem_config.h')
+create_makefile('rmem')

data/ext/rmem.c

@@ -0,0 +1,118 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include "rmem_config.h"
+#if HAVE_LIBPROC_H
+// see: http://stackoverflow.com/questions/220323/determine-process-info-programmatically-in-darwin-osx
+#  include <unistd.h>
+#  include <libproc.h>
+#elif HAVE_PROC
+// see: http://www.linuxforums.org/forum/linux-programming-scripting/11703-c-function-returns-cpu-memory-usage.html
+#  include <unistd.h>
+#elif HAVE_WINDOWS_H
+#  include <windows.h>
+#  include <psapi.h>
+#else
+  #error "No /proc or libproc.h"
+#endif
+
+#include <ruby.h>
+
+static int report_memory_usage_by_pid(unsigned long pid, unsigned long *size)
+{
+#ifdef HAVE_PROC
+  char buf[30];
+  snprintf(buf, 30, "/proc/%lu/statm", pid);
+  FILE* pf = fopen(buf, "r");
+  if (pf) {
+    int r = fscanf(pf, "%lu", size);
+    fclose(pf);
+    return (r == 1) ? 0 : 1;
+  }
+  return 1;
+#endif
+
+#ifdef HAVE_LIBPROC_H
+  struct proc_taskinfo pt;
+  int ret = proc_pidinfo(pid, PROC_PIDTASKINFO, 0, &pt, sizeof(pt));
+  if( ret <= 0 ) {
+    return 1;
+  }
+  *size = pt.pti_resident_size;
+  return 0;
+
+#endif
+#ifdef HAVE_WINDOWS_H
+   HANDLE hProcess;
+   PROCESS_MEMORY_COUNTERS pmc;
+
+   hProcess = OpenProcess( PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, pid );
+   if (NULL == hProcess)
+     return 1;
+   if ( GetProcessMemoryInfo( hProcess, &pmc, sizeof(pmc)) ) {
+     *size = pmc.PagefileUsage;
+   }
+
+   CloseHandle( hProcess );
+   return 0;
+#endif
+}
+
+static int report_memory_usage(unsigned long *size)
+{
+  unsigned long pid;
+#ifdef HAVE_WINDOWS_H
+  pid = GetCurrentProcessId();
+#else
+  pid = getpid();
+#endif
+  return report_memory_usage_by_pid(pid, size);
+}
+
+static VALUE rb_rmem;
+static VALUE rb_rmem_report;
+
+static VALUE ruby_memory_report(VALUE self)
+{
+  unsigned long bytes;
+  if( report_memory_usage(&bytes) ) {
+    rb_raise(rb_eRuntimeError, "Error detecting runtime memory usage");
+  }
+  return rb_int_new(bytes);
+}
+
+static VALUE ruby_memory_report_for(VALUE self, VALUE pid)
+{
+  unsigned long bytes;
+  if( report_memory_usage_by_pid(FIX2LONG(pid), &bytes) ) {
+    rb_raise(rb_eRuntimeError, "Error detecting runtime memory usage");
+  }
+  return rb_int_new(bytes);
+}
+
+
+void Init_rmem()
+{
+  rb_rmem = rb_define_module("RMem");
+  rb_rmem_report = rb_define_class_under( rb_rmem, "Report", rb_cObject );
+  rb_define_singleton_method(rb_rmem_report, "memory", ruby_memory_report, 0);
+  rb_define_singleton_method(rb_rmem_report, "memory_for", ruby_memory_report_for, 1);
+}
+
+
+#ifdef TEST_ENABLED
+
+int main( int argc, char **argv )
+{
+  char buffer[1024];
+  unsigned long memory_usage;
+
+  memset(buffer,0,sizeof(buffer));
+
+  report_memory_usage(&memory_usage);
+
+  printf("mem: %lu bytes\n", memory_usage);
+  return 0;
+}
+
+#endif

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification 
+name: rmem
+version: !ruby/object:Gem::Version 
+  version: 1.0.0
+platform: ruby
+authors: 
+- Todd A. Fisher
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-12-26 00:00:00 -05:00
+default_executable: 
+dependencies: []
+
+description: Measure approximate process memory
+email: todd.fisher@gmail.com
+executables: []
+
+extensions: 
+- ext/extconf.rb
+extra_rdoc_files: 
+- LICENSE
+- README
+files: 
+- LICENSE
+- README
+- Rakefile
+- ext/extconf.rb
+- ext/rmem.c
+has_rdoc: true
+homepage: http://rmem.rubyforge.org/
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --main
+- README
+require_paths: 
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: rmem
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Ruby Memory Measurebindings
+test_files: []
+