Winny/Share監視制御

-- ルーターの LAN インターフェースの数
num_lan = 1

-- パケット検出時にメールを送信するか否か（on..送信する、off..送信しない)
mail_send = "off"

-- メールの設定
mail_tbl = {
　　smtp_address = "(SMTPサーバーのアドレス)",
　　from = "(送信元のメールアドレス)",
　　to = "(宛先のメールアドレス)"
}

-- メールの送信に失敗したときに出力する SYSLOG のレベル (info, debug, notice)
log_level = "info"

-- ポートの遮断からのタイムアウト
timeout = 300

-- 遮断しているポートの数
cnt = 0

-- ポートを遮断されているホスト情報のテーブル
host_tbl = {}

function search(mac, addr, lan)
　　-- ホスト検索結果の経路を格納
　　route = nil
　　-- ホスト検索結果の経路から一つ前の経路を格納
　　switchroute = nil
　　-- ホストが接続されているポート番号を格納
　　hostport = nil

　　-- LAN1のMACアドレステーブルでホストが見つかった場合
　　if (lan == 1) then
　　　　rtn, str = rt.command("show status switching-hub macaddress " ..mac)
　　　　port = string.match(str,"port (%d):")
　　　　if (port ~= nil) then
　　　　　　route = "LAN1:" ..port
　　　　　　switchroute = "RTX1200"
　　　　　　hostport = port
　　　　else
　　　　　　route = nil
　　　　　　switchroute = nil
　　　　　　hostport = addr
　　　　end
　　else
　　　　-- LAN2またはLAN3でホストが見つかった場合
　　　　lan = tostring(lan)
　　　　route = "LAN"..lan
　　　　switchroute = "RTX1200"
　　　　hostport = "LAN"..lan
　　end

　　--[[
　　上記いずれかの方法でホストが見つかった場合、routeの値を基点とした経路で
　　ホストが見つかるまで繰り返し検索する
　　]]
　　while (true) do
　　　　if (route ~= nil) then
　　　　　　rtn, str = rt.command("switch control function get status-macaddress-addr "
　　　　　　　　　　　　　　　　　　　　　..mac .." " ..route)
　　　　　　if (rtn) and (str ~= "0 entry\r\n") then
　　　　　　　　switchroute = route
　　　　　　　　hostport = string.match(str,"%d+")
　　　　　　　　route = route .."-" ..string.match(str,"%d+")
　　　　　　else
　　　　　　　　route = nil
　　　　　　end
　　　　else
　　　　　　break
　　　　end
　　end
　　return switchroute, hostport
end

function send_mail(apl, addr, macaddr, route, port)
　　local rtn, str

　　mail_tbl.text = string.format("%sパケットを検出したため、switch(%s)の"..
　　　　　　　　　　　　　　　　　　　　"ポート%dを閉じました。\r\n\r\n", apl, route, port)
　　mail_tbl.text = mail_tbl.text .. "送信元ホスト情報\r\n\tIPアドレス：" .. addr .. "\r\n"
　　mail_tbl.text = mail_tbl.text .. "\tMACアドレス：" .. macaddr
　　mail_tbl.subject = string.format("%s packet detected", apl)
　　rtn = rt.mail(mail_tbl)

　　if (not rtn) then
　　　　rt.syslog(log_level, "failed to send mail. swx2200_lua_winny_share_rtx1200.lua")
　　end
end

function exec_sw_cmd(route, cmd)
　　rtn, str = rt.command("switch select "..route)
　　if (rtn) then
　　　　rtn, str = rt.command(cmd)
　　end
　　return rtn
end

function update_host_info()
　　local host_info = {}
　　local i = #host_tbl

　　while (i >= 1) do
　　　　host_info = host_tbl[i]
　　　　host_info.timeout = host_info.timeout - 1

　　　　if (host_info.timeout <= 0) then
　　　　　　str = "no switch control function set port-use " ..host_info.port
　　　　　　rtn = exec_sw_cmd(host_info.route, str)
　　　　　　if (rtn) then
　　　　　　　　rt.syslog(log_level, "switch("..host_info.route..") port"..host_info.port.." UP")
　　　　　　　　while (true) do
　　　　　　　　　　table.remove(host_info)
　　　　　　　　　　if (#host_info < 1) then
　　　　　　　　　　　　break
　　　　　　　　　　end
　　　　　　　　end
　　　　　　　　table.remove(host_tbl, i)
　　　　　　　　cnt = cnt - 1
　　　　　　else
　　　　　　　　host_tbl[i] = host_info
　　　　　　end
　　　　end
　　　　i = i - 1
　　end
end

function add_host(ipaddr, switchroute, hostport)
　　local host_info = {}
　　local add_info = {}
　　local find = 0

　　for j = 1, #host_tbl do
　　　　host_info = host_tbl[j]
　　　　if (host_info.ipaddr == ipaddr) then
　　　　　　find = 1
　　　　　　host_info.timeout = timeout
　　　　　　host_info.route = switchroute
　　　　　　host_info.port = hostport
　　　　　　host_tbl[j] = host_info
　　　　　　break
　　　　end
　　end

　　if (find == 0) then
　　　　j = #host_tbl + 1
　　　　add_info.ipaddr = ipaddr
　　　　add_info.timeout = timeout
　　　　add_info.route = switchroute
　　　　add_info.port = hostport
　　　　host_tbl[j] = add_info
　　　　cnt = cnt + 1
　　end
end

local rtn, str, app, ipaddr, mac
local j = 0
local find = 0
-- Winny/Share パケットを検出した際に出力される SYSLOG の文字列パターン
ptn = "([SW][hi][an][rn][ey]) version .-"

while (true) do
　　rtn, str = rt.syslogwatch(ptn, 1,1)
　　if (cnt > 0) then
　　　　update_host_info()
　　end

　　if (rtn > 0) and (str) then
　　　　app = string.match(str[1], ptn)
　　　　if (app == "Winny") or (app == "Share") then
　　　　　　if (app == "Winny") then
　　　　　　　　ipaddr = string.match(str[1], "(%d+%.%d+%.%d+%.%d+)")
　　　　　　else
　　　　　　　　ipaddr = string.match(str[1], "> (%d+%.%d+%.%d+%.%d+)")
　　　　　　end

　　　　　　for i = 1, num_lan do
　　　　　　　　rtn, str = rt.command("show arp lan" .. i .. " | grep " .. ipaddr)
　　　　　　　　if (rtn) and (str) and (string.match(str, ipaddr) ~= nil) then
　　　　　　　　　　mac = string.match(str, "(%x%x:%x%x:%x%x:%x%x:%x%x:%x%x)")
　　　　　　　　　　if (mac ~= nil) then
　　　　　　　　　　　　switchroute, hostport = search(mac, ipaddr, i)
　　　　　　　　　　　　if (switchroute ~= nil) then
　　　　　　　　　　　　　　str = "switch control function set port-use " ..hostport .." off"
　　　　　　　　　　　　　　rtn = exec_sw_cmd(switchroute, str)
　　　　　　　　　　　　　　if (rtn) then
　　　　　　　　　　　　　　　　if (mail_send == "on") then
　　　　　　　　　　　　　　　　　　send_mail(app, ipaddr, mac, switchroute, hostport)
　　　　　　　　　　　　　　　　end
　　　　　　　　　　　　　　　　rt.syslog(log_level, app.." packet detected from "..ipaddr..
　　　　　　　　　　　　　　　　　　　　　　" swx2200_lua_winny_share_rtx1200.lua")
　　　　　　　　　　　　　　　　add_host(ipaddr, switchroute, hostport)
　　　　　　　　　　　　　　　　break
　　　　　　　　　　　　　　end
　　　　　　　　　　　　end
　　　　　　　　　　else
　　　　　　　　　　　　rt.syslog(log_level, "[ERROR] not found MAC Address "..
　　　　　　　　　　　　　　　　　　"swx2200_lua_winny_share_rtx1200.lua")
　　　　　　　　　　end
　　　　　　　　end
　　　　　　end
　　　　end
　　end
end