linux_stat 1.2.1 → 1.4.0

data/lib/linux_stat/process.rb

@@ -1,4 +1,7 @@
 module LinuxStat
+	# Shows various information about a process that is either
+	# running, sleeping, idle or a zombie.
+
 	module Process
 		class << self
 			##
@@ -6,10 +9,17 @@ module LinuxStat
 			#
 			# The return type is an Array of Integers.
 			def list
-				Dir['/proc/*'].select { |x|
-					pid = File.split(x)[1]
-					pid.to_i.to_s == pid
-				}.map! { |x| File.split(x)[-1].to_i }
+				d = Dir['/proc/*'.freeze]
+				ret, i = [], -1
+				count = d.length
+
+				while(i += 1) < count
+					pid = File.split(d[i])[1]
+					pid_i = pid.to_i
+					ret << pid_i if pid_i.to_s == pid
+				end
+
+				ret
 			end
 
 			##
@@ -17,39 +27,82 @@ module LinuxStat
 			#
 			# The return type is Integer.
 			def count
-				list.count
+				list.length
 			end
 
 			##
-			# Returns all the id of processes mapped with their names as a Hash.
+			# Returns all the id of processes mapped with their executable names (comm) as a Hash.
+			# The names can be truncated to TASK_COMM_LEN or (16 - 1 = 15) places.
 			def names
-				list.reduce({}) { |h, x|
+				h, i = {}, -1
+
+				l = list
+				count = l.length
+
+				while(i += 1) < count
+					x = l[i]
+
 					begin
-						h.merge!( x => IO.foreach(File.join('/proc', x.to_s, 'status')).first.split[1] )
-					rescue Exception
-						h
+						h.merge!( x => IO.read("/proc/#{x}/comm").strip)
+					rescue StandardError
 					end
-				}
+				end
+				h
+			end
+
+			##
+			# Returns all the id of processes mapped with their cmdline info as a Hash.
+			# The cmdlines aren't usually truncated like names, but they can contain
+			# arguments with the command.
+			def cmdlines
+				h, i = {}, -1
+
+				l = list
+				count = l.length
+
+				while(i += 1) < count
+					x = l[i]
+
+					begin
+						cmdlines = IO.read("/proc/#{x}/cmdline").strip
+						cmdlines.gsub!(?\u0000.freeze, ?\s.freeze)
+						h.merge!( x => cmdlines)
+					rescue StandardError
+					end
+				end
+				h
 			end
 
 			##
 			# Returns all the id of processes mapped with their status as a Hash.
 			def types
-				list.reduce({}) { |h, x|
+				h, i = {}, -1
+
+				l = list
+				count = l.length
+
+				while(i += 1) < count
+					x = l[i]
+
 					begin
 						h.merge!(x =>
-							case IO.read(File.join('/proc', x.to_s, 'stat')).split[2]
+							case IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip
 								when ?S.freeze then :sleeping
 								when ?I.freeze then :idle
 								when ?Z.freeze then :zombie
 								when ?R.freeze then :running
+								when ?T.freeze then :stopped
+								when ?X.freeze then :dead
+								when ?D.freeze then :sleeping
+								when ?t.freeze then :stopped
 								else :unknown
 							end
 						)
-					rescue Exception
-						h
+					rescue StandardError
 					end
-				}
+				end
+
+				h
 			end
 
 			##
@@ -58,9 +111,9 @@ module LinuxStat
 			def sleeping
 				list.select { |x|
 					begin
-						IO.read(File.join('/proc', x.to_s, 'stat')).split[2] == ?S
-					rescue Exception
-						nil
+						IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip == ?S.freeze
+					rescue StandardError
+						false
 					end
 				}
 			end
@@ -71,9 +124,9 @@ module LinuxStat
 			def idle
 				list.select { |x|
 					begin
-						IO.read(File.join('/proc', x.to_s, 'stat')).split[2] == ?I
-					rescue Exception
-						nil
+						IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip == ?I.freeze
+					rescue StandardError
+						false
 					end
 				}
 			end
@@ -84,9 +137,9 @@ module LinuxStat
 			def zombie
 				list.select { |x|
 					begin
-						IO.read(File.join('/proc', x.to_s, 'stat')).split[2] == ?Z
-					rescue Exception
-						nil
+						IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip == ?Z.freeze
+					rescue StandardError
+						false
 					end
 				}
 			end
@@ -97,9 +150,23 @@ module LinuxStat
 			def running
 				list.select { |x|
 					begin
-						IO.read(File.join('/proc', x.to_s, 'stat')).split[2] == ?R
-					rescue Exception
-						nil
+						IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip == ?R.freeze
+					rescue StandardError
+						false
+					end
+				}
+			end
+
+			##
+			# Returns all the id of processes that are stopped.
+			# The return type is an Array of Integers.
+			def stopped
+				list.select { |x|
+					begin
+						v = IO.read("/proc/#{x}/stat").split(/(\(.*\))/)[-1][/\s.+?/].strip
+						v == ?T.freeze || v == ?t.freeze
+					rescue StandardError
+						false
 					end
 				}
 			end

data/lib/linux_stat/process_info.rb

@@ -1,4 +1,11 @@
 module LinuxStat
+	# Shows various information about a process that is either
+	# running, sleeping, idle or a zombie.
+	# Most methods can take a PID, but some uses polling to calculate
+	# something, they can accept options instead of arguments.
+	# Consult the documentation on the specific methods
+	# for more details on that.
+
 	module ProcessInfo
 		class << self
 			##
@@ -62,6 +69,8 @@ module LinuxStat
 
 			##
 			# = command_name(pid = $$)
+			# Not to be confused with process_name
+			# It just splits the cmdline to show the command name
 			#
 			# Where pid is the process ID.
 			#
@@ -77,12 +86,40 @@ module LinuxStat
 			# If the info isn't available it will return an empty frozen String.
 			def command_name(pid = $$)
 				# Do note that the /proc/ppid/comm may not contain the full name
-				file = "/proc/#{pid}/cmdline".freeze
-				return ''.freeze unless File.readable?(file)
+				# It's limited by TASK_COMM_LEN (16) characters
+				c = cmdline(pid).split[0]
+				return ''.freeze unless c
+				File.split(c)[-1]
+			end
 
-				_cmdline = IO.read(file)
-				_cmdline.gsub!(?\u0000, ?\s)
-				File.split(_cmdline.tap(&:strip!).split[0])[-1]
+			##
+			# = process_name(pid = $$)
+			# It shows the filename of the command
+			# Sometimes the filename is stripped
+			#
+			# Where pid is the process ID.
+			#
+			# By default it is the id of the current process ($$)
+			#
+			# It retuns the total command name of the process.
+			#
+			# The output is String. For example:
+			#    LinuxStat::ProcessInfo.process_name
+			#
+			#    "ruby"
+			#
+			# If the info isn't available it will return an empty frozen String.
+			def process_name(pid = $$)
+				file = "/proc/#{pid}/stat".freeze
+				return command_name unless File.readable?(file)
+
+				name = IO.read(file).split(/(\(.*\))/) &.[](1) &.[](1..-2)
+
+				if name && name.length > 0 && name.length < 15
+					name
+				else
+					command_name
+				end
 			end
 
 			##
@@ -249,8 +286,11 @@ module LinuxStat
 				ticks = get_ticks
 
 				return {} unless File.readable?(file)
-				utime, stime, starttime = IO.read(file)
-					.split.values_at(13, 14, 21).map(&:to_f)
+
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return {} unless stat
+
+				utime, stime, starttime = *stat.values_at(11, 12, 19).map(&:to_f)
 
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
@@ -260,9 +300,12 @@ module LinuxStat
 				sleep(sleep)
 
 				return {} unless File.readable?(file)
-				stat = IO.read(file).split
 
-				utime2, stime2, starttime2 = stat.values_at(13, 14, 21).map(&:to_f)
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return {} unless stat
+
+				utime2, stime2, starttime2 = *stat.values_at(11, 12, 19).map(&:to_f)
+
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
 				total_time2 = utime2 + stime2
@@ -273,8 +316,8 @@ module LinuxStat
 
 				{
 					cpu_usage: cpu_u > 100 ? 100.0 : cpu_u.round(2),
-					threads: stat[19].to_i,
-					last_executed_cpu: stat[38].to_i
+					threads: stat[17].to_i,
+					last_executed_cpu: stat[36].to_i
 				}
 			end
 
@@ -309,8 +352,11 @@ module LinuxStat
 				ticks = get_ticks
 
 				return nil unless File.readable?(file)
-				utime, stime, starttime = IO.read(file)
-					.split.values_at(13, 14, 21).map(&:to_f)
+
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
+				utime, stime, starttime = *stat.values_at(11, 12, 19).map(&:to_f)
 
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
@@ -320,9 +366,11 @@ module LinuxStat
 				sleep(sleep)
 
 				return nil unless File.readable?(file)
-				utime2, stime2, starttime2 = IO.read(file)
-					.split.values_at(13, 14, 21).map(&:to_f)
 
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
+				utime2, stime2, starttime2 = *stat.values_at(11, 12, 19).map(&:to_f)
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
 				total_time2 = utime2 + stime2
@@ -361,8 +409,11 @@ module LinuxStat
 				ticks = get_ticks
 
 				return nil unless File.readable?(file)
-				utime, stime, starttime = IO.read(file)
-					.split.values_at(13, 14, 21).map(&:to_f)
+
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
+				utime, stime, starttime = *stat.values_at(11, 12, 19).map(&:to_f)
 
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
@@ -372,8 +423,11 @@ module LinuxStat
 				sleep(sleep)
 
 				return nil unless File.readable?(file)
-				utime2, stime2, starttime2 = IO.read(file)
-					.split.values_at(13, 14, 21).map(&:to_f)
+
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
+				utime2, stime2, starttime2 = *stat.values_at(11, 12, 19).map(&:to_f)
 
 				uptime = IO.read('/proc/uptime'.freeze).to_f * ticks
 
@@ -409,7 +463,7 @@ module LinuxStat
 				file = "/proc/#{pid}/stat".freeze
 				return nil unless File.readable?(file)
 
-				data = IO.foreach(file, ' '.freeze).first(20)[-1]
+				data = IO.read(file).split(/(\(.*\))/)[-1] &.split &.at(17)
 				data ? data.to_i : nil
 			end
 
@@ -434,7 +488,8 @@ module LinuxStat
 				file = "/proc/#{pid}/stat".freeze
 				return nil unless File.readable?(file)
 
-				IO.read(file).split[38].to_i
+				data = IO.read(file).split(/(\(.*\))/)[-1] &.split &.at(36)
+				data ? data.to_i : nil
 			end
 
 			##
@@ -517,8 +572,11 @@ module LinuxStat
 				@@u_readable ||= File.readable?(uptime)
 				return nil unless @@u_readable && File.readable?(stat_file)
 
+				stat = IO.read(stat_file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
 				u = IO.foreach(uptime, ' '.freeze).next.to_f
-				st = (IO.foreach(stat_file, ' '.freeze).first(22)[-1].to_f / get_ticks)
+				st = stat[19].to_f / get_ticks
 
 				# Getting two Time objects and dealing with floating point numbers
 				# Just to make sure the time goes monotonically
@@ -567,8 +625,11 @@ module LinuxStat
 				@@u_readable ||= File.readable?(uptime)
 				return nil unless @@u_readable && File.readable?(stat_file)
 
+				stat = IO.read(stat_file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
 				IO.foreach(uptime, ' '.freeze).next.to_f
-					.-(IO.read(stat_file).split[21].to_f / get_ticks).round(2)
+					.-(stat[19].to_f / get_ticks).round(2)
 			end
 
 			##
@@ -592,7 +653,11 @@ module LinuxStat
 			def state(pid = $$)
 				file = "/proc/#{pid}/stat".freeze
 				return ''.freeze unless File.readable?(file)
-				IO.foreach(file, ' '.freeze).first(3)[-1].tap(&:rstrip!).freeze
+
+				stat = IO.read(file).split(/(\(.*\))/)[-1]
+				return '' unless stat
+
+				stat[/\s.+?/].strip
 			end
 
 			##
@@ -608,7 +673,10 @@ module LinuxStat
 				file = "/proc/#{pid}/stat"
 				return nil unless File.readable?(file)
 
-				IO.foreach(file, ' ').first(19)[-1].to_i
+				stat = IO.read(file).split(/(\(.*\))/)[-1] &.split
+				return nil unless stat
+
+				stat[16].to_i
 			end
 
 			##

data/lib/linux_stat/swap.rb

@@ -1,4 +1,6 @@
 module LinuxStat
+	# Shows various Swap devices related information of the current system.
+
 	module Swap
 		class << self
 			##
@@ -8,7 +10,7 @@ module LinuxStat
 			def list
 				return {} unless swaps_readable?
 
-				file = IO.readlines('/proc/swaps').drop(1)
+				file = IO.readlines('/proc/swaps'.freeze).drop(1)
 				file.reduce({}) do |h, x|
 					name, *stats = x.strip.split
 					h.merge!(name => stats.map! { |v| v.to_i.to_s == v ? v.to_i : v.to_sym })
@@ -20,7 +22,7 @@ module LinuxStat
 			#
 			# If the info isn't available, it will return an empty Hash.
 			def any?
-				!!IO.foreach('/proc/swaps').drop(1).first
+				!!IO.foreach('/proc/swaps'.freeze).first(2)[1]
 			end
 
 			# Show aggregated used and available swap.
@@ -114,7 +116,7 @@ module LinuxStat
 			def read_usage
 				return [[], []] unless swaps_readable?
 
-				val = IO.readlines('/proc/swaps').drop(1)
+				val = IO.readlines('/proc/swaps'.freeze).drop(1)
 				return [[], []] if val.empty?
 
 				val.map! { |x|
@@ -123,7 +125,7 @@ module LinuxStat
 			end
 
 			def swaps_readable?
-				@@swaps_readable ||= File.readable?('/proc/swaps')
+				@@swaps_readable ||= File.readable?('/proc/swaps'.freeze)
 			end
 		end
 	end