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 tObjMessageHandler
=procedure(msg
:tSMsg
) of object;
33 {deliver message from peer to the object}
34 procedure SetMsgHandler(OpCode
:byte; from
:tNetAddr
; handler
:tObjMessageHandler
);
36 type tTimeVal
=UnixType
.timeval
;
41 USES SysUtils
,Sockets
,BaseUnix
45 {aim for most simple implementation, since could be extended anytime}
50 var pollArr
: packed array [tPollTop
] of tPollFd
;
52 cb
: tFDEventHandler
; {proc+object}
54 var pollHnd
: array [tPollTop
] of tFdHndDsc
;
55 var pollTop
: tPollTop
;
57 var hnd
: array [1..36] of tMessageHandler
;
58 var HiHnd
: tMessageHandler
;
60 type tSheduled_ptr
=^tSheduled
; tSheduled
=record
65 var ShedTop
: ^tSheduled
;
66 var ShedUU
: ^tSheduled
;
67 var LastShed
: UnixType
.timeval
;
68 var PollTimeout
:LongInt;
70 procedure SC(fn
:pointer; retval
:cint
);
72 if retval
< 0 then begin
73 raise eXception
.Create(Format('Socket error %d operation %P',[SocketError
,fn
]));
77 procedure s_SetupInet
;
78 var bind_addr
:tInetSockAddr
;
81 with bind_addr
do begin
85 s_inet
:=fpSocket(family
,SOCK_DGRAM
,IPPROTO_UDP
);
87 turnon
:=IP_PMTUDISC_DO
;
88 SC(@fpsetsockopt
,fpsetsockopt(s_inet
, IPPROTO_IP
, IP_MTU_DISCOVER
, @turnon
, sizeof(turnon
)));
90 SC(@fpBind
,fpBind(s_inet
,@bind_addr
,sizeof(bind_addr
)));
91 with PollArr
[0] do begin
98 var Terminated
:boolean=false;
100 procedure SendMessage(const data
; len
:word; const rcpt
:tSockAddrL
);
103 {SC(@fpsendto,}fpsendto(s_inet
,@data
,len
,0,@rcpt
,sizeof(sockaddr_in
)){)};
105 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
);
109 SendMessage(data
,len
,sa
);
111 procedure SendMessage(const data
; len
:word; const rcpt
:tNetAddr
; channel
:word );
113 SendMessage(data
,len
,rcpt
);
114 {todo: optimization??}
117 procedure SignalHandler(sig
:cint
);CDecl;
120 if terminated
then raise eControlC
.Create('CtrlC DoubleTap') ;
122 writeln('Shutdown requested');
125 {index=iphash+opcode}
126 type tPeerTableBucket
=record
129 handler
:tObjMessageHandler
;
131 var PT
:array [0..255] of ^tPeerTableBucket
;
132 var PT_opcodes
: set of 1..36;
134 function FindPT(opcode
:byte; const addr
:tNetAddr
):Word; { $FFFF=fail}
137 i
:=(addr
.hash
+opcode
) mod high(PT
); {0..63}
138 for o
:=0 to high(PT
) do begin
139 result
:=(i
+o
) mod high(PT
);
140 if not assigned(PT
[result
]) then break
;
141 if (PT
[result
]^.opcode
=opcode
) and (PT
[result
]^.remote
=addr
) then exit
;
146 procedure UnSetMsgHandler(const from
:tNetAddr
; opcode
:byte);
149 h
:=FindPT(opcode
,from
);
150 if h
=$FFFF then exit
;
153 {go reverse exit on null, hash them, match: move to H and stop}
155 while (i
<>h
)and assigned(PT
[i
]) do begin
156 if (PT
[i
]^.remote
.hash
+PT
[i
]^.opcode
)=h
then begin
161 if i
=0 then i
:=high(PT
) else dec(i
);
165 procedure SetMsgHandler(OpCode
:byte; from
:tNetAddr
; handler
:tObjMessageHandler
);
168 UnSetMsgHandler(from
,opcode
);
169 if handler
=nil then exit
;
170 h
:=(from
.hash
+opcode
) mod high(PT
);
171 for o
:=0 to high(PT
) do begin
172 i
:=(h
+o
) mod high(PT
);
173 if not assigned(PT
[i
]) then break
;
176 PT
[i
]^.opcode
:=OpCode
;
178 PT
[i
]^.handler
:=handler
;
179 Include(PT_opcodes
,opcode
);
182 {do not waste stack on statics}
183 var EventsCount
:integer;
184 var Buffer
:array [1..1024] of byte;
186 var From
:tSockAddrL
; {use larger struct so everything fits}
187 var FromLen
:LongWord
;
188 var curhnd
:tMessageHandler
;
189 var curhndo
:tObjMessageHandler
;
193 procedure DoSock(var p
:tPollFD
);
199 if (p
.revents
and pollIN
)=0 then exit
;
200 FromLen
:=sizeof(From
);
201 pkLen
:=fprecvfrom(p
.FD
,@Buffer
,sizeof(Buffer
),0,@from
,@fromlen
);
202 SC(@fprecvfrom
,pkLen
);
204 FromG
.FromSocket(from
);
205 Msg
.Source
:=@FromG
; {!thread}
207 Msg
.Data
:=@Buffer
; {!thread}
208 Msg
.stream
.Init(@Buffer
,pkLen
,sizeof(Buffer
));
209 Msg
.channel
:=0; {!multisocket}
210 if Buffer
[1]>128 then curhnd
:=HiHnd
else if Buffer
[1]<=high(hnd
) then curhnd
:=hnd
[Buffer
[1]];
211 if Buffer
[1] in PT_opcodes
then ptidx
:=FindPT(Buffer
[1],FromG
) else ptidx
:=0;
212 if ptidx
>0 then curhndo
:=PT
[ptidx
]^.handler
;
218 var now
:UnixType
.timeval
absolute iNow
;
220 var olTop
:^tSheduled
;
226 ShedTop
:=nil; {unlink the current shed list}
227 fpgettimeofday(@Now
,nil);
228 delta
:=(Now
.tv_sec
-LastShed
.tv_sec
);
229 if delta
>6 then delta
:=5000 else delta
:=(delta
*1000)+((Now
.tv_usec
-LastShed
.tv_usec
) div 1000);
231 //writeln('DeltaTime: ',delta);
232 while assigned(cur
) do begin
233 if (cur
^.left
<=delta
)or(cur
^.left
=0) then begin
240 DEC(cur
^.left
,delta
);
241 //writeln('Left: ',cur^.left);
242 if pollTimeOut
>cur
^.left
then PollTimeOut
:=cur
^.left
;
247 pcur
^:=ShedTop
; {append newly added tasks to end of untriggererd list}
248 ShedTop
:=olTop
; {link in the untriggered tasks}
250 while assigned(cur
) do begin
251 if cur
^.left
<PollTimeout
then PollTimeout
:=cur
^.left
;
254 if pollTimeout
=0 then pollTimeOut
:=1;
260 while not terminated
do begin
263 EventsCount
:=fpPoll(@PollArr
[0],PollTop
,PollTimeout
);
264 if (eventscount
=-1)and terminated
then break
;
265 if eventscount
=-1 then break
; {fixme: print error}
266 if eventscount
=0 then continue
else begin
269 if assigned(curhndo
) then curhndo(msg
)
270 else if assigned(curhnd
) then curhnd(msg
)
271 else raise eXception
.Create('No handler for opcode '+IntToStr(Buffer
[1]));
274 for tp
:=1 to pollTop
do if PollArr
[tp
].revents
>0 then begin
275 PollHnd
[tp
].CB(PollArr
[tp
].rEvents
);
276 PollArr
[tp
].revents
:=0;
280 write('Loop broken [');
285 procedure SetMsgHandler(OpCode
:byte; handler
:tMessageHandler
);
286 begin assert(hnd
[OpCode
]=nil); hnd
[OpCode
]:=handler
; end;
287 procedure SetHiMsgHandler(handler
:tMessageHandler
);
288 begin Hihnd
:=handler
; end;
290 procedure WatchFD(fd
:tHandle
; h
:tFDEventHandler
);
293 if assigned(h
) then begin
294 PollHnd
[pollTop
].CB
:=h
;
295 PollArr
[pollTop
].fd
:=fd
;
296 PollArr
[pollTop
].events
:=POLLERR
or POLLHUP
or POLLIN
or POLLPRI
or
297 POLLRDBAND
or POLLRDNORM
;
298 PollArr
[pollTop
].revents
:=0;
299 //writeln('Add watch ',pollTop,' on ',fd,' to ',IntToHex(qword(@h),8));
301 end else for opt
:=0 to high(opt
) do if PollArr
[opt
].fd
=fd
then begin
302 if (pollTop
-1)>opt
then begin
303 PollArr
[opt
]:=PollArr
[pollTop
-1];
304 PollHnd
[opt
]:=PollHnd
[pollTop
-1];
307 PollArr
[pollTop
].fd
:=-1;
308 PollArr
[pollTop
].events
:=0;
309 PollArr
[pollTop
].revents
:=0;
314 procedure Shedule(timeout
{ms}: LongWord
; h
:tOnTimer
);
318 if Assigned(ShedUU
) then begin
320 ShedUU
:=ShedUU
^.next
;
321 end else New(ShedTop
);
322 ShedTop
^.Left
:=timeout
;
327 procedure UnShedule(h
:tOnTimer
);
333 while assigned(cur
) do begin
334 if cur
^.cb
=h
then begin
335 pcur
^:=cur
^.next
; {unlink from main list}
336 cur
^.next
:=ShedUU
; ShedUU
:=cur
; {link to unused}
348 fpSignal(SigInt
,@SignalHandler
);
349 fpSignal(SigTerm
,@SignalHandler
);
350 for i
:=1 to high(hnd
) do hnd
[i
]:=nil;
351 for i
:=1 to high(PT
) do PT
[i
]:=nil;
353 pollTop
:=1; {1 for basic listen}
355 ShedUU
:=nil; {todo: allocate a few to improve paging}
356 fpgettimeofday(@LastShed
,nil);