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
;
77 with bind_addr
do begin
81 s_inet
:=fpSocket(family
,SOCK_DGRAM
,IPPROTO_UDP
);
83 turnon
:=IP_PMTUDISC_DO
;
84 SC(@fpsetsockopt
,fpsetsockopt(s_inet
, IPPROTO_IP
, IP_MTU_DISCOVER
, @turnon
, sizeof(turnon
)));
86 SC(@fpBind
,fpBind(s_inet
,@bind_addr
,sizeof(bind_addr
)));
87 with PollArr
[0] do begin
94 var Terminated
:boolean=false;
96 procedure SendMessage(const data
; len
:word; const rcpt
:tSockAddrL
);
99 {SC(@fpsendto,}fpsendto(s_inet
,@data
,len
,0,@rcpt
,sizeof(sockaddr_in
)){)};
101 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
);
105 SendMessage(data
,len
,sa
);
107 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
; channel
:word );
109 SendMessage(data
,len
,rcpt
);
110 {todo: optimization??}
113 procedure SignalHandler(sig
:cint
);CDecl;
116 if terminated
then raise eControlC
.Create('CtrlC DoubleTap') ;
118 writeln('Shutdown requested');
121 {do not waste stack on statics}
122 var EventsCount
:integer;
123 var Buffer
:array [1..1024] of byte;
125 var From
:tSockAddrL
; {use larger struct so everything fits}
126 var FromLen
:LongWord
;
128 var curhnd
:tMessageHandler
;
132 procedure PrepareHandler
;
134 FromG
.FromSocket(from
);
135 if Buffer
[1]>128 then curhnd
:=HiHnd
else curhnd
:=hnd
[Buffer
[1]];
136 if not assigned(curhnd
) then raise eXception
.Create('No handler for opcode '+IntToStr(Buffer
[1]));
137 Msg
.Source
:=@FromG
; {!thread}
139 Msg
.Data
:=@Buffer
; {!thread}
140 Msg
.stream
.Init(@Buffer
,pkLen
,sizeof(Buffer
));
141 Msg
.channel
:=0; {!multisocket}
147 var now
:UnixType
.timeval
absolute iNow
;
149 var olTop
:^tSheduled
;
155 ShedTop
:=nil; {unlink the current shed list}
156 fpgettimeofday(@Now
,nil);
157 delta
:=(Now
.tv_sec
-LastShed
.tv_sec
);
158 if delta
>6 then delta
:=5000 else delta
:=(delta
*1000)+((Now
.tv_usec
-LastShed
.tv_usec
) div 1000);
160 //writeln('DeltaTime: ',delta);
161 while assigned(cur
) do begin
162 if (cur
^.left
<=delta
)or(cur
^.left
=0) then begin
169 DEC(cur
^.left
,delta
);
170 //writeln('Left: ',cur^.left);
171 if pollTimeOut
>cur
^.left
then PollTimeOut
:=cur
^.left
;
176 pcur
^:=ShedTop
; {append newly added tasks to end of untriggererd list}
177 ShedTop
:=olTop
; {link in the untriggered tasks}
179 while assigned(cur
) do begin
180 if cur
^.left
<PollTimeout
then PollTimeout
:=cur
^.left
;
183 if pollTimeout
=0 then pollTimeOut
:=1;
189 while not terminated
do begin
192 EventsCount
:=fpPoll(@PollArr
[0],PollTop
,PollTimeout
);
193 if (eventscount
=-1)and terminated
then break
;
194 if eventscount
=-1 then break
; {fixme: print error}
195 if eventscount
=0 then continue
else begin
197 with PollArr
[0] do begin
198 if (revents
and pollIN
)>0 then begin
199 FromLen
:=sizeof(From
);
200 pkLen
:=fprecvfrom(s_inet
,@Buffer
,sizeof(Buffer
),0,@from
,@fromlen
);
201 SC(@fprecvfrom
,pkLen
);
209 for tp
:=1 to pollTop
do if PollArr
[tp
].revents
>0 then begin
210 PollHnd
[tp
].CB(PollArr
[tp
].rEvents
);
211 PollArr
[tp
].revents
:=0;
215 write('Loop broken [');
220 procedure SetMsgHandler(OpCode
:byte; handler
:tMessageHandler
);
221 begin assert(hnd
[OpCode
]=nil); hnd
[OpCode
]:=handler
; end;
222 procedure SetHiMsgHandler(handler
:tMessageHandler
);
223 begin Hihnd
:=handler
; end;
225 procedure WatchFD(fd
:tHandle
; h
:tFDEventHandler
);
228 if assigned(h
) then begin
229 PollHnd
[pollTop
].CB
:=h
;
230 PollArr
[pollTop
].fd
:=fd
;
231 PollArr
[pollTop
].events
:=POLLERR
or POLLHUP
or POLLIN
or POLLPRI
or
232 POLLRDBAND
or POLLRDNORM
;
233 PollArr
[pollTop
].revents
:=0;
234 //writeln('Add watch ',pollTop,' on ',fd,' to ',IntToHex(qword(@h),8));
236 end else for opt
:=0 to high(opt
) do if PollArr
[opt
].fd
=fd
then begin
237 if (pollTop
-1)>opt
then begin
238 PollArr
[opt
]:=PollArr
[pollTop
-1];
239 PollHnd
[opt
]:=PollHnd
[pollTop
-1];
242 PollArr
[pollTop
].fd
:=-1;
243 PollArr
[pollTop
].events
:=0;
244 PollArr
[pollTop
].revents
:=0;
249 procedure Shedule(timeout
{ms}: LongWord
; h
:tOnTimer
);
253 if Assigned(ShedUU
) then begin
255 ShedUU
:=ShedUU
^.next
;
256 end else New(ShedTop
);
257 ShedTop
^.Left
:=timeout
;
262 procedure UnShedule(h
:tOnTimer
);
268 while assigned(cur
) do begin
269 if cur
^.cb
=h
then begin
270 pcur
^:=cur
^.next
; {unlink from main list}
271 cur
^.next
:=ShedUU
; ShedUU
:=cur
; {link to unused}
283 fpSignal(SigInt
,@SignalHandler
);
284 fpSignal(SigTerm
,@SignalHandler
);
285 for i
:=1 to high(hnd
) do hnd
[i
]:=nil;
286 pollTop
:=1; {1 for basic listen}
288 ShedUU
:=nil; {todo: allocate a few to improve paging}
289 fpgettimeofday(@LastShed
,nil);