system_cmds-880.40.5.tar.gz

[apple/system_cmds.git] / zprint.tproj / zprint.lua

diff --git a/zprint.tproj/zprint.lua b/zprint.tproj/zprint.lua
index 3c7d5fc9ebc7e336af6196b571b2b6ea5f053058..a5ca245e7150c9d20e044605dccaa9d3d2f8eb68 100644 (file)
--- a/zprint.tproj/zprint.lua
+++ b/zprint.tproj/zprint.lua
@@ -33,7 +33,7 @@ function zprint.zones(zpout)
   local zones = lines_inside_dashes(zpout, 1)
 
   -- Create an iterator for each line, for use in our own iteration function.
   local zones = lines_inside_dashes(zpout, 1)
 
   -- Create an iterator for each line, for use in our own iteration function.

-  local lines = zones:gmatch('([^\n]+)\n')
+  local lines = zones:gmatch('([^\n]+)')

 
   return function ()
     -- Grab the next line.
 
   return function ()
     -- Grab the next line.


@@ -65,12 +65,35 @@ function zprint.zones(zpout)
   end
 end
 
   end
 end
 

+-- Match the output of a vm_tag line
+-- This line has a variable number of columns.
+-- This function returns the name and a table containing each numeric column's
+-- value.
+local function match_tag(line, ncols)
+  -- First try to match names with C++ symbol names.
+  -- These can have whitespace in the argument list.
+  local name_pattern = '^(%S+%b()%S*)'
+  local name = line:match(name_pattern)
+  if not name then
+    name = line:match('(%S+)')
+    if not name then
+      return nil
+    end
+  end
+  local after_name = line:sub(#name)
+  local t = {}
+  for v in line:gmatch('%s+(%d+)K?') do
+    table.insert(t, v)
+  end   
+  return name, t
+end
+

 -- Iterate through the tags listed in the given zprint(1) output `zpout`.
 function zprint.tags(zpout)
   -- Get to the third zone delimited by dashes, where the tags are recorded.
   local tags = lines_inside_dashes(zpout, 3)
 
 -- Iterate through the tags listed in the given zprint(1) output `zpout`.
 function zprint.tags(zpout)
   -- Get to the third zone delimited by dashes, where the tags are recorded.
   local tags = lines_inside_dashes(zpout, 3)
 

-  local lines = tags:gmatch('([^\n]+)\n')
+  local lines = tags:gmatch('([^\n]+)')

 
   return function ()
     local line = lines()
 
   return function ()
     local line = lines()


@@ -87,24 +110,18 @@ function zprint.tags(zpout)
       return nil
     end
 
       return nil
     end
 

-    -- The line representing a region can take 4 different forms, depending on
-    -- the type of region.  Check for each of them.
+    local name, matches = match_tag(line)
+    if not name or #matches == 0 then
+      return nil
+    end

 
 

-    -- Check for 4 columns.
-    local name, maxsz_kb, cursz_kb = line:match(
-        '(%S+)%s+%d+%s+%d+%s+(%d+)K%s+(%d+)K$')
-    if not name then
-      -- Check for 3 columns.
-      name, maxsz_kb, cursz_kb = line:match('(%S+)%s+%d+%s+(%d+)K%s+(%d+)K$')
-      if not name then
-        -- Check for a two columns.
-        name, cursz_kb = line:match('(%S+)%s+%d+%s+(%d+)K')
-        if not name then
-          -- Check for a single column.
-          name, cursz_kb = line:match('(%S+)%s+(%d+)K')
-        end
-      end
+    local cursz_kb = matches[#matches]
+    -- If there are fewer than 3 numeric columns, there's no reported peak size
+    local maxsz_kb = nil
+    if #matches > 3 then
+      maxsz_kb = matches[#matches - 1]

     end
     end

+

     -- Convert numeric fields to numbers and then into bytes.
     local cursz = tonumber(cursz_kb) * 1024
     local maxsz = maxsz_kb and (tonumber(maxsz_kb) * 1024)
     -- Convert numeric fields to numbers and then into bytes.
     local cursz = tonumber(cursz_kb) * 1024
     local maxsz = maxsz_kb and (tonumber(maxsz_kb) * 1024)


@@ -118,10 +135,147 @@ function zprint.tags(zpout)
   end
 end
 
   end
 end
 

+-- Iterate through the maps listed in the given zprint(1) output `zpout`.
+function zprint.maps(zpout)
+  local maps = lines_inside_dashes(zpout, 5)
+  local lines = maps:gmatch('([^\n]+)')
+
+  return function()
+    -- Grab the next line.
+    local line = lines()
+    if not line then
+      return nil
+    end
+
+    -- The line can take on 3 different forms. Check for each of them
+
+    -- Check for 3 columns
+    local name, free_kb, largest_free_kb, curr_size_kb = line:match(
+        '(%S+)%s+(%d+)K%s+(%d+)K%s+(%d+)K')
+    local free, largest_free, peak_size_kb, peak_size, size
+    if not name then
+      -- Check for 2 columns
+      name, peak_size_kb, curr_size_kb = line:match('(%S+)%s+(%d+)K%s+(%d+)K')
+      if not name then
+        -- Check for a single column
+        name, curr_size_kb = line:match('(%S+)%s+(%d+)K')
+        assert(name)
+      else
+        peak_size = tonumber(peak_size_kb) * 1024
+      end
+    else
+      free = tonumber(free_kb) * 1024
+      largest_free = tonumber(largest_free_kb) * 1024
+    end
+    size = tonumber(curr_size_kb) * 1024
+
+    return {
+      name = name,
+      size = size,
+      max_size = peak_size,
+      free = free,
+      largest_free = largest_free
+    }
+  end
+end
+
+-- Iterate through the zone views listed in the given zprint(1) output `zpout`.
+function zprint.zone_views(zpout)
+  -- Skip to the zone views
+  local prev_pos = 1
+  -- Look for a line that starts with "zone views" and is followed by a -- line.
+  while true do
+    local start_pos, end_pos = zpout:find('\n[-]+\n', prev_pos)
+    if start_pos == nil then
+      return nil
+    end
+    local before = zpout:sub(prev_pos, start_pos)
+    local zone_views_index = zpout:find('\n%s*zone views%s+[^\n]+\n', prev_pos + 1)
+    prev_pos = end_pos
+    if  zone_views_index and zone_views_index < end_pos then
+      break
+    end
+  end
+
+  local zone_views
+  local zone_totals_index = zpout:find("\nZONE TOTALS")
+  if zone_totals_index then
+    zone_views = zpout:sub(prev_pos + 1, zone_totals_index)
+  else
+    zone_views = zpout:sub(prev_pos+ 1)
+  end
+
+  local lines = zone_views:gmatch('([^\n]+)')
+
+  return function()
+    -- Grab the next line.
+    local line = lines()
+    if not line then
+      return nil
+    end
+
+    local name, curr_size_kb = line:match('(%S+)%s+(%d+)')
+    local size = tonumber(curr_size_kb) * 1024
+
+    return {
+      name = name,
+      size = size,
+    }
+  end
+end
+

 function zprint.total(zpout)
   local total = zpout:match('total[^%d]+(%d+.%d+)M of')
   local bytes = tonumber(total) * 1024 * 1024
   return bytes
 end
 
 function zprint.total(zpout)
   local total = zpout:match('total[^%d]+(%d+.%d+)M of')
   local bytes = tonumber(total) * 1024 * 1024
   return bytes
 end
 

-return zprint
+-- Return a library object, if called from require or dofile.
+local calling_func = debug.getinfo(2).func
+if calling_func == require or calling_func == dofile then
+  return zprint
+end
+
+-- Otherwise, 'recon zprint.lua ...' runs as a script.
+
+local cjson = require 'cjson'
+
+if not arg[1] then
+  io.stderr:write('usage: ', arg[0], ' <zprint-output-path>\n')
+  os.exit(1)
+end
+
+local file
+if arg[1] == '-' then
+  file = io.stdin
+else
+  local err
+  file, err = io.open(arg[1])
+  if not file then
+    io.stderr:write('zprint.lua: ', arg[1], ': open failed: ', err, '\n')
+    os.exit(1)
+  end
+end
+
+local zpout = file:read('all')
+file:close()
+
+local function collect(iter, arg)
+  local tbl = {}
+  for elt in iter(arg) do
+    tbl[#tbl + 1] = elt
+  end
+  return tbl
+end
+
+local zones = collect(zprint.zones, zpout)
+local tags = collect(zprint.tags, zpout)
+local maps = collect(zprint.maps, zpout)
+local zone_views = collect(zprint.zone_views, zpout)
+
+print(cjson.encode({
+  zones = zones,
+  tags = tags,
+  maps = maps,
+  zone_views = zone_views,
+}))