1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
if not (Spring.GetConfigInt("LuaSocketEnabled", 0) == 1) then
Spring.Echo("LuaSocketEnabled is disabled")
return false
end
function widget:GetInfo()
return {
name = "Test-Widget for luasocket",
desc = "a simple test widget to show capabilities of luasocket",
author = "abma",
date = "Feb. 2012",
license = "GNU GPL, v2 or later",
layer = 0,
enabled = true,
}
end
local socket = socket
local client
local set
local headersent
local host = "springrts.com"
local port = 80
local file = "/dl/buildbot/default/master/LATEST"
local function dumpConfig()
-- dump all luasocket related config settings to console
for _, conf in ipairs({"TCPAllowConnect", "TCPAllowListen", "UDPAllowConnect", "UDPAllowListen" }) do
Spring.Echo(conf .. " = " .. Spring.GetConfigString(conf, ""))
end
end
local function newset()
local reverse = {}
local set = {}
return setmetatable(set, {__index = {
insert = function(set, value)
if not reverse[value] then
table.insert(set, value)
reverse[value] = table.getn(set)
end
end,
remove = function(set, value)
local index = reverse[value]
if index then
reverse[value] = nil
local top = table.remove(set)
if top ~= value then
reverse[top] = index
set[index] = top
end
end
end
}})
end
-- initiates a connection to host:port, returns true on success
local function SocketConnect(host, port)
client=socket.tcp()
client:settimeout(0)
res, err = client:connect(host, port)
if not res and not res=="timeout" then
Spring.Echo("Error in connect: "..err)
return false
end
set = newset()
set:insert(client)
return true
end
function widget:Initialize()
dumpConfig()
--Spring.Echo(socket.dns.toip("localhost"))
--FIXME dns-request seems to block
SocketConnect(host, port)
end
-- called when data was received through a connection
local function SocketDataReceived(sock, str)
Spring.Echo(str)
end
local headersent
-- called when data can be written to a socket
local function SocketWriteAble(sock)
if headersent==nil then
-- socket is writeable
headersent=1
Spring.Echo("sending http request")
sock:send("GET " .. file .. " HTTP/1.0\r\nHost: " .. host .. " \r\n\r\n")
end
end
-- called when a connection is closed
local function SocketClosed(sock)
Spring.Echo("closed connection")
end
function widget:Update()
if set==nil or #set<=0 then
return
end
-- get sockets ready for read
local readable, writeable, err = socket.select(set, set, 0)
if err~=nil then
-- some error happened in select
if err=="timeout" then
-- nothing to do, return
return
end
Spring.Echo("Error in select: " .. error)
end
for _, input in ipairs(readable) do
local s, status, partial = input:receive('*a') --try to read all data
if status == "timeout" or status == nil then
SocketDataReceived(input, s or partial)
elseif status == "closed" then
SocketClosed(input)
input:close()
set:remove(input)
end
end
for __, output in ipairs(writeable) do
SocketWriteAble(output)
end
end
|
