bam/auto-boot-ollama-host
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enhance Wake-on-LAN functionality by forcing IPv4 UDP socket and adding source IP binding

Browse Source
This commit is contained in:
Bastian (BaM)
2025-09-14 11:08:40 +02:00
parent 8b32856c0d
commit c57359daef
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
scripts/auto-boot-ollama-host.lua
View File

@@ -47,25 +47,33 @@ local function send_wol(mac_str, bcast_ip, port)
if not mb then return false, "invalid MAC" end if not mb then return false, "invalid MAC" end
local packet = string.rep(string.char(0xFF), 6) .. mb:rep(16) local packet = string.rep(string.char(0xFF), 6) .. mb:rep(16)
local udp = assert(socket.udp()) -- 1) Force an IPv4 UDP socket (fallback to udp() if udp4() is not available)
local udp_ctor = socket.udp4 or socket.udp
local udp = assert(udp_ctor())
udp:settimeout(2) udp:settimeout(2)
-- Optional: bind to a specific source IP (helps in host network with multi-NIC) -- 2) Bind to an IPv4 source to lock the address family
local WOL_SRC_IP = os.getenv("WOL_SRC_IP") -- If you have WOL_SRC_IP set, bind to that; otherwise 0.0.0.0.
if WOL_SRC_IP and WOL_SRC_IP ~= "" then local src_ip = os.getenv("WOL_SRC_IP")
assert(udp:setsockname(WOL_SRC_IP, 0)) if src_ip and src_ip ~= "" then
assert(udp:setsockname(src_ip, 0))
else
assert(udp:setsockname("0.0.0.0", 0))
end end
-- IMPORTANT: must be true or Linux will return EACCES on broadcast sendto() -- 3) Enable broadcast. This will fail on AF_INET6, so do it *after* binding IPv4.
assert(udp:setoption("broadcast", true)) assert(udp:setoption("broadcast", true))
-- 4) Send to directed broadcast (recommended) or 255.255.255.255 if you must.
local ok, err = udp:sendto(packet, bcast_ip, port) local ok, err = udp:sendto(packet, bcast_ip, port)
udp:close() udp:close()
if not ok then return false, err end if not ok then return false, err end
return true return true
end end
-- Kept for reference; not used to keep parity with your minimal bash -- Kept for reference; not used to keep parity with your minimal bash
-- local function port_is_up(host, port, timeout_sec) -- local function port_is_up(host, port, timeout_sec)
-- local deadline = socket.gettime() + (timeout_sec or 1) -- local deadline = socket.gettime() + (timeout_sec or 1)