4 uses MemStream
,NetAddr
,UnixType
;
13 stream
: tMemoryStream
;
16 type tMessageHandler
=procedure(msg
:tSMsg
);
17 procedure SetMsgHandler(OpCode
:byte; handler
:tMessageHandler
);
18 procedure SetHiMsgHandler(handler
:tMessageHandler
);
20 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
);
21 {procedure SendReply(const data; len:word; const rcpt:tSMsg );}
22 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
; channel
:word );
24 {#Sheduling and watching#}
25 type tFDEventHandler
=procedure(ev
:Word) of object;
26 type tOnTimer
=procedure of object;
27 procedure WatchFD(fd
:tHandle
; h
:tFDEventHandler
);
28 procedure Shedule(timeout
{ms}: LongWord
; h
:tOnTimer
);
29 procedure UnShedule(h
:tOnTimer
);
30 {note unshed will fail when called from OnTimer proc}
32 type tTimeVal
=UnixType
.timeval
;
37 USES SysUtils
,Sockets
,BaseUnix
41 {aim for most simple implementation, since could be extended anytime}
46 var pollArr
: packed array [tPollTop
] of tPollFd
;
48 cb
: tFDEventHandler
; {proc+object}
50 var pollHnd
: array [tPollTop
] of tFdHndDsc
;
51 var pollTop
: tPollTop
;
53 var hnd
: array [1..36] of tMessageHandler
;
54 var HiHnd
: tMessageHandler
;
56 type tSheduled_ptr
=^tSheduled
; tSheduled
=record
61 var ShedTop
: ^tSheduled
;
62 var ShedUU
: ^tSheduled
;
63 var LastShed
: UnixType
.timeval
;
64 var PollTimeout
:LongInt;
66 procedure SC(fn
:pointer; retval
:cint
);
68 if retval
< 0 then begin
69 raise eXception
.Create(Format('Socket error %d operation %P',[SocketError
,fn
]));
73 procedure s_SetupInet
;
74 var bind_addr
:tInetSockAddr
;
76 with bind_addr
do begin
80 s_inet
:=fpSocket(family
,SOCK_DGRAM
,IPPROTO_UDP
);
83 SC(@fpBind
,fpBind(s_inet
,@bind_addr
,sizeof(bind_addr
)));
84 with PollArr
[0] do begin
91 var Terminated
:boolean=false;
93 procedure SendMessage(const data
; len
:word; const rcpt
:tSockAddrL
);
96 SC(@fpsendto
,fpsendto(s_inet
,@data
,len
,0,@rcpt
,sizeof(sockaddr_in
)));
98 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
);
102 SendMessage(data
,len
,sa
);
104 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
; channel
:word );
106 SendMessage(data
,len
,rcpt
);
107 {todo: optimization??}
110 procedure SignalHandler(sig
:cint
);CDecl;
113 if terminated
then raise eControlC
.Create('CtrlC DoubleTap') ;
115 writeln('Shutdown requested');
118 {do not waste stack on statics}
119 var EventsCount
:integer;
120 var Buffer
:array [1..1024] of byte;
122 var From
:tSockAddrL
; {use larger struct so everything fits}
123 var FromLen
:LongWord
;
125 var curhnd
:tMessageHandler
;
129 procedure PrepareHandler
;
131 FromG
.FromSocket(from
);
132 if Buffer
[1]>128 then curhnd
:=HiHnd
else curhnd
:=hnd
[Buffer
[1]];
133 if not assigned(curhnd
) then raise eXception
.Create('No handler for opcode '+IntToStr(Buffer
[1]));
134 Msg
.Source
:=@FromG
; {!thread}
136 Msg
.Data
:=@Buffer
; {!thread}
137 Msg
.stream
.Init(@Buffer
,pkLen
,sizeof(Buffer
));
138 Msg
.channel
:=0; {!multisocket}
144 var now
:UnixType
.timeval
absolute iNow
;
146 var olTop
:^tSheduled
;
152 ShedTop
:=nil; {unlink the current shed list}
153 fpgettimeofday(@Now
,nil);
154 delta
:=(Now
.tv_sec
-LastShed
.tv_sec
);
155 if delta
>6 then delta
:=5000 else delta
:=(delta
*1000)+((Now
.tv_usec
-LastShed
.tv_usec
) div 1000);
157 //writeln('DeltaTime: ',delta);
158 while assigned(cur
) do begin
159 if (cur
^.left
<=delta
)or(cur
^.left
=0) then begin
166 DEC(cur
^.left
,delta
);
167 //writeln('Left: ',cur^.left);
168 if pollTimeOut
>cur
^.left
then PollTimeOut
:=cur
^.left
;
173 pcur
^:=ShedTop
; {append newly added tasks to end of untriggererd list}
174 ShedTop
:=olTop
; {link in the untriggered tasks}
176 while assigned(cur
) do begin
177 if cur
^.left
<PollTimeout
then PollTimeout
:=cur
^.left
;
180 if pollTimeout
=0 then pollTimeOut
:=1;
186 while not terminated
do begin
189 EventsCount
:=fpPoll(@PollArr
[0],PollTop
,PollTimeout
);
190 if (eventscount
=-1)and terminated
then break
;
191 if eventscount
=-1 then break
; {fixme: print error}
192 if eventscount
=0 then continue
else begin
194 with PollArr
[0] do begin
195 if (revents
and pollIN
)>0 then begin
196 FromLen
:=sizeof(From
);
197 pkLen
:=fprecvfrom(s_inet
,@Buffer
,sizeof(Buffer
),0,@from
,@fromlen
);
198 SC(@fprecvfrom
,pkLen
);
206 for tp
:=1 to pollTop
do if PollArr
[tp
].revents
>0 then begin
207 PollHnd
[tp
].CB(PollArr
[tp
].rEvents
);
208 PollArr
[tp
].revents
:=0;
212 write('Loop broken [');
217 procedure SetMsgHandler(OpCode
:byte; handler
:tMessageHandler
);
218 begin assert(hnd
[OpCode
]=nil); hnd
[OpCode
]:=handler
; end;
219 procedure SetHiMsgHandler(handler
:tMessageHandler
);
220 begin Hihnd
:=handler
; end;
222 procedure WatchFD(fd
:tHandle
; h
:tFDEventHandler
);
225 if assigned(h
) then begin
226 PollHnd
[pollTop
].CB
:=h
;
227 PollArr
[pollTop
].fd
:=fd
;
228 PollArr
[pollTop
].events
:=POLLERR
or POLLHUP
or POLLIN
or POLLPRI
or
229 POLLRDBAND
or POLLRDNORM
;
230 PollArr
[pollTop
].revents
:=0;
231 //writeln('Add watch ',pollTop,' on ',fd,' to ',IntToHex(qword(@h),8));
233 end else for opt
:=0 to high(opt
) do if PollArr
[opt
].fd
=fd
then begin
234 if (pollTop
-1)>opt
then begin
235 PollArr
[opt
]:=PollArr
[pollTop
-1];
236 PollHnd
[opt
]:=PollHnd
[pollTop
-1];
239 PollArr
[pollTop
].fd
:=-1;
240 PollArr
[pollTop
].events
:=0;
241 PollArr
[pollTop
].revents
:=0;
246 procedure Shedule(timeout
{ms}: LongWord
; h
:tOnTimer
);
250 if Assigned(ShedUU
) then begin
252 ShedUU
:=ShedUU
^.next
;
253 end else New(ShedTop
);
254 ShedTop
^.Left
:=timeout
;
259 procedure UnShedule(h
:tOnTimer
);
265 while assigned(cur
) do begin
266 if cur
^.cb
=h
then begin
267 pcur
^:=cur
^.next
; {unlink from main list}
268 cur
^.next
:=ShedUU
; ShedUU
:=cur
; {link to unused}
280 fpSignal(SigInt
,@SignalHandler
);
281 fpSignal(SigTerm
,@SignalHandler
);
282 for i
:=1 to high(hnd
) do hnd
[i
]:=nil;
283 pollTop
:=1; {1 for basic listen}
285 ShedUU
:=nil; {todo: allocate a few to improve paging}
286 fpgettimeofday(@LastShed
,nil);