fixup! Add javascript challenge sample

luainkernel · Aug 11, 2020 · bdd9779 · bdd9779
1 parent b6ca132
commit bdd9779
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/samples/bpf/checkcookie.lua b/samples/bpf/checkcookie.lua
@@ -45,7 +45,12 @@ function checkcookieport(pkt, ip, port)
 end
 
 function checkcookie(pkt, ip)
-	local cookiepkt = tonumber(string.match(tostring(pkt), "Cookie:%s__xdp=(.-)\r\n"))
+	local pattern = 'Cookie:%s*=__xdp=(%d+)%s*'
+	local cookiepkt = tonumber(string.match(tostring(pkt), pattern))
+
+	if not cookies[ip] then
+		return true
+	end
 
 	return cookies[ip] == cookiepkt and true or false
 end