network sniffer

sys/apps/Network.lua

@@ -128,6 +128,13 @@ local function sendCommand(host, command)
 	end
 end
 
+function page.ports:eventHandler(event)
+	if event.type == 'grid_select' then
+		shell.openForegroundTab('sniff ' .. event.selected.port)
+	end
+	return UI.SlideOut.eventHandler(self, event)
+end
+
 function page.ports.grid:update()
 	local function findConnection(port)
 		for _,socket in pairs(_G.transport.sockets) do

sys/apps/sniff.lua

@@ -0,0 +1,69 @@
+local Event     = require('event')
+local Terminal  = require('terminal')
+local Util      = require('util')
+
+local colors    = _G.colors
+local modem     = _G.device.wireless_modem
+local term      = _G.term
+local textutils = _G.textutils
+
+local terminal = Terminal.window(term.current())
+terminal.setMaxScroll(300)
+local oldTerm = term.redirect(terminal)
+
+local function syntax()
+  error('Syntax: sniff [port]')
+end
+
+local port = ({ ... })[1] or syntax()
+port = tonumber(port) or syntax()
+
+Event.on('modem_message',
+  function(_, _, dport, _, data, _)
+    if dport == port then
+      terminal.scrollBottom()
+      terminal.setTextColor(colors.white)
+      print(textutils.serialize(data))
+    end
+  end)
+
+Event.on('mouse_scroll', function(_, direction)
+  if direction == -1 then
+    terminal.scrollUp()
+  else
+    terminal.scrollDown()
+  end
+end)
+
+local function sniffer(_, _, data)
+  terminal.scrollBottom()
+  terminal.setTextColor(colors.yellow)
+  local ot = term.redirect(terminal)
+  print(textutils.serialize(data))
+  term.redirect(ot)
+end
+
+local socket = _G.transport.sockets[port]
+if socket then
+  if not socket.sniffers then
+    socket.sniffers = { modem.transmit }
+    socket.transmit = function(...)
+      for _,v in pairs(socket.sniffers) do
+        v(...)
+      end
+    end
+  end
+  table.insert(socket.sniffers, sniffer)
+end
+
+local s, m = pcall(Event.pullEvents)
+
+if socket then
+  Util.removeByValue(socket.sniffers, sniffer)
+end
+
+term.redirect(oldTerm)
+
+if not s and m then
+  error(m)
+end