simulate gettimeofday on windows
[lcore.git] / lsocket.pas
1 {lsocket.pas}\r
2 \r
3 {socket code by plugwash}\r
4 \r
5 { Copyright (C) 2005 Bas Steendijk and Peter Green\r
6   For conditions of distribution and use, see copyright notice in zlib_license.txt\r
7   which is included in the package\r
8   ----------------------------------------------------------------------------- }\r
9 {\r
10 changes by plugwash (20030728)\r
11 * created handlefdtrigger virtual method in tlasio (overridden in tlsocket) and moved a lot of code from messageloop into it\r
12 * changed tlasio to tlasio\r
13 * split fdhandle into fdhandlein and fdhandleout\r
14 * i now use fdsrmaster and fdswmaster instead of rebuilding the lists every loop\r
15 * split lsocket.pas into lsocket.pas and lcore.pas\r
16 \r
17 \r
18 changes by beware (20030903)\r
19 * added getxaddr, getxport (local addr, port, as string)\r
20 * added getpeername, remote addr+port as binary\r
21 * added htons and htonl functions (endian swap, same interface as windows API)\r
22 \r
23 beware (20030905)\r
24 * if connect failed (conn refused) set state to connected and call internalclose, to get closed handler (instead of fdclose)\r
25 * (lcore) if closing the fd's in internalcose, set fd's to -1 because closing an fd makes it invalid\r
26 \r
27 beware (20030927)\r
28 * fixed: on connect failed, tried to close fdhandle's which were already set to -1, added check\r
29 \r
30 beware (20031017)\r
31 * added getpeeraddr, getpeerport, remote addr+port as string\r
32 }\r
33 \r
34 \r
35 unit lsocket;\r
36 {$ifdef fpc}\r
37   {$mode delphi}\r
38 {$endif}\r
39 \r
40 {$include lcoreconfig.inc}\r
41 \r
42 interface\r
43   uses\r
44     sysutils,\r
45     {$ifdef win32}\r
46       windows,winsock,\r
47     {$else}\r
48 \r
49       {$ifdef VER1_0}\r
50         linux,\r
51       {$else}\r
52         baseunix,unix,unixutil,\r
53       {$endif}\r
54       sockets,\r
55     {$endif}\r
56     classes,{pgdebugout,}pgtypes,lcore,fd_utils,binipstuff,dnssync;\r
57 \r
58 {$ifdef ipv6}\r
59 const\r
60   v4listendefault:boolean=false;\r
61 {$endif}\r
62 \r
63 \r
64 type\r
65   sunB = packed record\r
66     s_b1, s_b2, s_b3, s_b4: byte;\r
67   end;\r
68 \r
69   SunW = packed record\r
70     s_w1, s_w2: word;\r
71   end;\r
72 \r
73   TInAddr = packed record\r
74     case integer of\r
75       0: (S_un_b: SunB);\r
76       1: (S_un_w: SunW);\r
77       2: (S_addr: cardinal);\r
78   end;\r
79 \r
80   type\r
81     TLsocket = class(tlasio)\r
82     public\r
83       //a: string;\r
84 \r
85       inAddr             : TInetSockAddrV;\r
86 \r
87       biniplist:tbiniplist;\r
88       trymoreips:boolean;\r
89       currentip:integer;\r
90       connecttimeout:tltimer;\r
91 \r
92 {      inAddrSize:integer;}\r
93 \r
94       //host               : THostentry      ;\r
95 \r
96       //mainthread         : boolean         ; //for debuggin only\r
97       addr:thostname;\r
98       port:ansistring;\r
99       localaddr:thostname;\r
100       localport:ansistring;\r
101       proto:ansistring;\r
102       udp,dgram:boolean;\r
103       listenqueue:integer;\r
104       {$ifdef secondlistener}\r
105       secondlistener:tlsocket;\r
106       lastsessionfromsecond:boolean;\r
107       procedure secondaccepthandler(sender:tobject;error:word);\r
108       procedure internalclose(error:word);override;\r
109       {$endif}\r
110       function getaddrsize:integer;\r
111       procedure connect; virtual;\r
112       procedure realconnect;\r
113       procedure bindsocket;\r
114       procedure listen;\r
115       function accept : longint;\r
116       function sendto(dest:TInetSockAddrV;destlen:integer;data:pointer;len:integer):integer; virtual;\r
117       function receivefrom(data:pointer;len:integer;var src:TInetSockAddrV;var srclen:integer):integer; virtual;\r
118 \r
119       procedure handlefdtrigger(readtrigger,writetrigger:boolean); override;\r
120       function send(data:pointer;len:integer):integer;override;\r
121       procedure sendstr(const str : tbufferstring);override;\r
122       function Receive(Buf:Pointer;BufSize:integer):integer; override;\r
123       function getpeername(var addr:tsockaddrin;addrlen:integer):integer; virtual;\r
124       procedure getXaddrbin(var binip:tbinip); virtual;\r
125       procedure getpeeraddrbin(var binip:tbinip); virtual;\r
126       function getXaddr:thostname; virtual;\r
127       function getpeeraddr:thostname; virtual;\r
128       function getXport:ansistring; virtual;\r
129       function getpeerport:ansistring; virtual;\r
130       constructor Create(AOwner: TComponent); override;\r
131 \r
132       //this one has to be kept public for now because lcorewsaasyncselect calls it\r
133       procedure connectionfailedhandler(error:word);\r
134     private\r
135       {$ifdef ipv6}\r
136         isv6socket : boolean; //identifies if the socket is v6, set by bindsocket\r
137       {$endif}\r
138       procedure taskcallconnectionfailedhandler(wparam,lparam : longint);\r
139 \r
140       procedure connecttimeouthandler(sender:tobject);\r
141       procedure connectsuccesshandler;\r
142       {$ifdef win32}\r
143         procedure myfdclose(fd : integer); override;\r
144         function myfdwrite(fd: LongInt;const buf;size: LongInt):LongInt; override;\r
145         function myfdread(fd: LongInt;var buf;size: LongInt):LongInt; override;\r
146       {$endif}\r
147     end;\r
148     tsocket=longint; // for compatibility with twsocket\r
149 \r
150   twsocket=tlsocket; {easy}\r
151 \r
152 \r
153 const\r
154   TCP_NODELAY=1;\r
155   IPPROTO_TCP=6;\r
156 \r
157 implementation\r
158 {$include unixstuff.inc}\r
159 \r
160 \r
161 function tlsocket.getaddrsize:integer;\r
162 begin\r
163   result := inaddrsize(inaddr);\r
164 end;\r
165 \r
166 //I used to use the system versions of these from within lsocket (which has\r
167 //functions whose name clashes with them) by using sockets.* and but I can't do\r
168 //that anymore since in some cases connect is now provided by unixstuff.inc\r
169 //hence these wrapper functions --plugwash\r
170 {$ifndef win32}\r
171   function system_Connect(Sock: LongInt;const Addr;Addrlen: LongInt):Boolean;\r
172   begin\r
173     result := connect(sock,addr,addrlen);\r
174   end;\r
175   function system_SendTo(Sock: LongInt; const Buf;BufLen: LongInt;Flags: LongInt;var Addr;AddrLen: LongInt):LongInt;\r
176   begin\r
177     result := sendto(sock,buf,buflen,flags,addr,addrlen);\r
178   end;\r
179   function system_getpeername(Sock: LongInt;var Addr;var Addrlen: LongInt):LongInt;\r
180   begin\r
181     result := getpeername(sock,addr,addrlen);\r
182   end;\r
183   function system_listen(Sock: LongInt; MaxConnect: LongInt):Boolean;\r
184   begin\r
185     result := listen(sock,maxconnect);\r
186   end;\r
187   function system_Accept(Sock: LongInt;var Addr;var Addrlen: LongInt):LongInt;\r
188   begin\r
189     result := accept(sock,addr,addrlen);\r
190   end;\r
191 {$endif}\r
192 \r
193 procedure tlsocket.realconnect;\r
194 var\r
195   a,b:integer;\r
196 begin\r
197   //writeln('trying to connect to ',ipbintostr(biniplist_get(biniplist,currentip)),'#',port);\r
198   makeinaddrv(biniplist_get(biniplist,currentip),port,inaddr);\r
199   inc(currentip);\r
200   if (currentip >= biniplist_getcount(biniplist)) then trymoreips := false;\r
201 \r
202   udp := false;\r
203   if (uppercase(proto) = 'UDP') then begin\r
204     b := IPPROTO_UDP;\r
205     a := SOCK_DGRAM;\r
206     udp := true;\r
207     dgram := true;\r
208   end else if (uppercase(proto) = 'TCP') or (uppercase(proto) = '') then begin\r
209     b := IPPROTO_TCP;\r
210     a := SOCK_STREAM;\r
211     dgram := false;\r
212   end else if (uppercase(proto) = 'ICMP') or (strtointdef(proto,256) < 256) then begin\r
213     b := strtointdef(proto,IPPROTO_ICMP);\r
214     a := SOCK_RAW;\r
215     dgram := true;\r
216   end else begin\r
217     raise ESocketException.create('unrecognised protocol');\r
218   end;\r
219 \r
220   a := Socket(inaddr.inaddr.family,a,b);\r
221   //writeln(ord(inaddr.inaddr.family));\r
222   if a = -1 then begin\r
223     //unable to create socket, fire an error event (better to use an error event\r
224     //to avoid poor interaction with multilistener stuff.\r
225     //a socket value of -2 is a special value to say there is no socket but\r
226     //we want internalclose to act as if there was\r
227     fdhandlein := -2;\r
228     fdhandleout := -2;\r
229     tltask.create(taskcallconnectionfailedhandler,self,{$ifdef win32}wsagetlasterror{$else}socketerror{$endif},0);\r
230     exit;\r
231   end;\r
232   try\r
233     dup(a);\r
234     bindsocket;\r
235     if dgram then begin\r
236       {$ifndef win32}\r
237         SetSocketOptions(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
238       {$else}\r
239         SetSockOpt(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
240       {$endif}\r
241       state := wsconnected;\r
242       if assigned(onsessionconnected) then onsessionconnected(self,0);\r
243 \r
244       eventcore.rmasterset(fdhandlein,false);\r
245       eventcore.wmasterclr(fdhandleout);\r
246     end else begin\r
247       state :=wsconnecting;\r
248       {$ifdef win32}\r
249         //writeln(inaddr.inaddr.port);\r
250         winsock.Connect(fdhandlein,winsock.psockaddr(@inADDR)^,getaddrsize);\r
251       {$else}\r
252         system_Connect(fdhandlein,inADDR,getaddrsize);\r
253       {$endif}\r
254       eventcore.rmasterset(fdhandlein,false);\r
255       eventcore.wmasterset(fdhandleout);\r
256       if trymoreips then connecttimeout.enabled := true;\r
257     end;\r
258     //sendq := '';\r
259   except\r
260     on e: exception do begin\r
261       fdcleanup;\r
262       raise; //reraise the exception\r
263     end;\r
264   end;\r
265 \r
266 end;\r
267 \r
268 procedure tlsocket.connecttimeouthandler(sender:tobject);\r
269 begin\r
270   connecttimeout.enabled := false;\r
271   destroying := true; //hack to not cause handler to trigger\r
272   internalclose(0);\r
273   destroying := false;\r
274   realconnect;\r
275 end;\r
276 \r
277 \r
278 \r
279 \r
280 procedure tlsocket.connect;\r
281 var\r
282   a:integer;\r
283   ip:tbinip;\r
284 begin\r
285   if state <> wsclosed then close;\r
286   //prevtime := 0;\r
287   if isbiniplist(addr) then biniplist := addr else biniplist := forwardlookuplist(addr,0);\r
288   if biniplist_getcount(biniplist) = 0 then raise exception.create('unable to resolve '+addr);\r
289 \r
290   //makeinaddrv(addr,port,inaddr);\r
291 \r
292   currentip := 0;\r
293   if not assigned(connecttimeout) then begin\r
294     connecttimeout := tltimer.create(self);\r
295     connecttimeout.ontimer := connecttimeouthandler;\r
296     connecttimeout.interval := 2500;\r
297     connecttimeout.enabled := false;\r
298   end;\r
299   realconnect;\r
300 end;\r
301 \r
302 procedure tlsocket.sendstr(const str : tbufferstring);\r
303 begin\r
304   if dgram then begin\r
305     send(@str[1],length(str))\r
306   end else begin\r
307     inherited sendstr(str);\r
308   end;\r
309 end;\r
310 \r
311 function tlsocket.send(data:pointer;len:integer):integer;\r
312 begin\r
313   if dgram then begin\r
314 //    writeln('sending to '+ipbintostr(inaddrvtobinip(inaddr)),' ',htons(inaddr.inaddr.port),' ',len,' bytes');\r
315     result := sendto(inaddr,getaddrsize,data,len);\r
316 \r
317 //    writeln('send result ',result);\r
318 //    writeln('errno',errno);\r
319   end else begin\r
320     result := inherited send(data,len);\r
321   end;\r
322 end;\r
323 \r
324 \r
325 function tlsocket.receive(Buf:Pointer;BufSize:integer):integer;\r
326 begin\r
327   if dgram then begin\r
328     {$ifdef secondlistener}\r
329     if lastsessionfromsecond then begin\r
330       result := secondlistener.receive(buf,bufsize);\r
331       lastsessionfromsecond := false;\r
332     end else\r
333     {$endif}\r
334       result := myfdread(self.fdhandlein,buf^,bufsize);\r
335   end else begin\r
336     result := inherited receive(buf,bufsize);\r
337   end;\r
338 end;\r
339 \r
340 procedure tlsocket.bindsocket;\r
341 var\r
342   a:integer;\r
343   inAddrtemp:TInetSockAddrV;\r
344   inAddrtempx:{$ifdef win32}winsock.TSockaddr{$else}TInetSockAddrV{$endif} absolute inaddrtemp;\r
345   inaddrtempsize:integer;\r
346 begin\r
347   try\r
348     if (localaddr <> '') or (localport <> '') then begin\r
349       if localaddr = '' then begin\r
350         {$ifdef ipv6}\r
351         if inaddr.inaddr.family = AF_INET6 then localaddr := '::' else\r
352         {$endif}\r
353         localaddr := '0.0.0.0';\r
354       end;\r
355       //gethostbyname(localaddr,host);\r
356       inaddrtempsize := makeinaddrv(forwardlookup(localaddr,0),localport,inaddrtemp);\r
357       {$ifdef ipv6}\r
358         isv6socket := (inaddrtemp.inaddr.family = AF_INET6);\r
359       {$endif}\r
360       If Bind(fdhandlein,inaddrtempx,inaddrtempsize)<> {$ifdef win32}0{$else}true{$endif} Then begin\r
361         state := wsclosed;\r
362         lasterror := {$ifdef win32}getlasterror{$else}socketerror{$endif};\r
363         raise ESocketException.create('unable to bind on address '+localaddr+'#'+localport+', error '+inttostr(lasterror));\r
364       end;\r
365       state := wsbound;\r
366     end;\r
367   except\r
368     on e: exception do begin\r
369       fdcleanup;\r
370       raise; //reraise the exception\r
371     end;\r
372   end;\r
373 end;\r
374 \r
375 procedure tlsocket.listen;\r
376 var\r
377   yes,no:longint;\r
378   socktype:integer;\r
379   biniptemp:tbinip;\r
380   origaddr:thostname;\r
381 begin\r
382   if state <> wsclosed then close;\r
383   udp := uppercase(proto) = 'UDP';\r
384   if udp then begin\r
385     socktype := SOCK_DGRAM;\r
386     dgram := true;\r
387   end else socktype := SOCK_STREAM;\r
388   origaddr := addr;\r
389 \r
390   if addr = '' then begin\r
391     {$ifdef ipv6}\r
392     //writeln('ipv6 is defined');\r
393     if not v4listendefault then begin\r
394       //writeln('setting addr to ::');\r
395       addr := '::';\r
396     end else\r
397     {$endif}\r
398     addr := '0.0.0.0';\r
399   end;\r
400   if isbiniplist(addr) then biniptemp := biniplist_get(addr,0) else biniptemp := forwardlookup(addr,10);\r
401   addr := ipbintostr(biniptemp);\r
402   //writeln('after ipbintostr call addr =',addr);\r
403   //writeln('biniptemp.family =',biniptemp.family);\r
404   //writeln('AF_INET6=',AF_INET6);\r
405   //writeln('PF_INET6=',PF_INET6);\r
406   //writeln('AF_INET=',AF_INET);\r
407   //writeln('PF_INET=',PF_INET);\r
408   \r
409   fdhandlein := socket(biniptemp.family,socktype,0);\r
410   {$ifdef ipv6}\r
411   if (addr = '::') and (origaddr = '') and (fdhandlein < 0) then begin\r
412     {writeln('failed to create an IPV6 socket with error ',socketerror,'. trying to create an IPV4 one instead');}\r
413     addr := '0.0.0.0';\r
414     fdhandlein := socket(PF_INET,socktype,0);\r
415   end;\r
416   {$endif}\r
417 \r
418   if fdhandlein = -1 then raise ESocketException.create('unable to create socket'{$ifdef win32}+' error='+inttostr(wsagetlasterror){$endif});\r
419   dupnowatch(fdhandlein); // sets up maxs and copies handle to fdhandleout among other things\r
420   //eventcore.setfdreverse(fdhandlein,self); //already taken care of by dup\r
421   state := wsclosed; // then set this back as it was an undesired side effect of dup\r
422 \r
423   try\r
424     yes := $01010101;  {Copied this from existing code. Value is empiric,\r
425                     but works. (yes=true<>0) }\r
426     no := 0;\r
427     {$ifndef win32}\r
428       if SetSocketOptions(fdhandlein, SOL_SOCKET, SO_REUSEADDR,yes,sizeof(yes))=-1 then begin\r
429         raise ESocketException.create('unable to set SO_REUSEADDR socket option');\r
430       end;\r
431       //writeln('addr=',addr);\r
432       //writeln('setting IPV6_V6ONLY option to 0');\r
433       //allow v4 connections on v6 sockets\r
434       if biniptemp.family = af_inet6 then begin\r
435         if SetSocketOptions(fdhandlein, IPPROTO_IPV6,IPV6_V6ONLY,no,sizeof(no))=-1 then begin\r
436           writeln(IPPROTO_IPV6);\r
437           writeln(IPV6_V6ONLY);\r
438           raise ESocketException.create('unable to set IPV6_V6ONLY socket option error='+inttostr(socketerror));\r
439           \r
440         end;\r
441       end;\r
442     {$endif}\r
443     localaddr := addr;\r
444     localport := port;\r
445     bindsocket;\r
446 \r
447     if not udp then begin\r
448       {!!! allow custom queue length? default 5}\r
449       if listenqueue = 0 then listenqueue := 5;\r
450       If {$ifdef win32}winsock.listen{$else}system_listen{$endif}(fdhandlein,listenqueue)<>{$ifdef win32}0{$else}true{$endif} Then raise\r
451 esocketexception.create('unable to listen');\r
452       state := wsListening;\r
453     end else begin\r
454       {$ifndef win32}\r
455         SetSocketOptions(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
456       {$else}\r
457         SetSockOpt(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
458       {$endif}\r
459       state := wsconnected;\r
460     end;\r
461 \r
462     {$ifdef secondlistener}\r
463     //listening on ::. try to listen on 0.0.0.0 as well for platforms which don't already do that\r
464     if addr = '::' then begin\r
465       secondlistener := tlsocket.create(nil);\r
466       secondlistener.proto := proto;\r
467       secondlistener.addr := '0.0.0.0';\r
468       secondlistener.port := port;\r
469       if udp then begin\r
470         secondlistener.ondataavailable := secondaccepthandler;\r
471       end else begin\r
472         secondlistener.onsessionAvailable := secondaccepthandler;\r
473       end;\r
474       try\r
475         secondlistener.listen;\r
476       except\r
477         secondlistener.destroy;\r
478         secondlistener := nil;\r
479       end;\r
480     end;\r
481     {$endif}\r
482   finally\r
483     if state = wsclosed then begin\r
484       if fdhandlein >= 0 then begin\r
485         {one *can* get here without fd -beware}\r
486         eventcore.rmasterclr(fdhandlein);\r
487         myfdclose(fdhandlein); // we musnt leak file discriptors\r
488         eventcore.setfdreverse(fdhandlein,nil);\r
489         fdhandlein := -1;\r
490       end;\r
491     end else begin\r
492       eventcore.rmasterset(fdhandlein,not udp);\r
493     end;\r
494     if fdhandleout >= 0 then eventcore.wmasterclr(fdhandleout);\r
495   end;\r
496   //writeln('listened on addr '+addr+':'+port+' '+proto+' using socket number ',fdhandlein);\r
497 end;\r
498 \r
499 {$ifdef secondlistener}\r
500 procedure tlsocket.internalclose(error:word);\r
501 begin\r
502   if assigned(secondlistener) then begin\r
503     secondlistener.destroy;\r
504     secondlistener := nil;\r
505   end;\r
506   inherited internalclose(error);\r
507 end;\r
508 \r
509 procedure tlsocket.secondaccepthandler;\r
510 begin\r
511   lastsessionfromsecond := true;\r
512   if udp then begin\r
513     ondataavailable(self,error);\r
514   end else begin\r
515     if assigned(onsessionavailable) then onsessionavailable(self,error);\r
516   end;\r
517 end;\r
518 {$endif}\r
519 \r
520 function tlsocket.accept : longint;\r
521 var\r
522   FromAddrSize     : LongInt;        // i don't realy know what to do with these at this\r
523   FromAddr         : TInetSockAddrV;  // at this point time will tell :)\r
524   a:integer;\r
525 begin\r
526   {$ifdef secondlistener}\r
527   if (lastsessionfromsecond) then begin\r
528     lastsessionfromsecond := false;\r
529     result := secondlistener.accept;\r
530     exit;\r
531   end;\r
532   {$endif}\r
533 \r
534   FromAddrSize := Sizeof(FromAddr);\r
535   {$ifdef win32}\r
536     result := winsock.accept(fdhandlein,@fromaddr,@fromaddrsize);\r
537   {$else}\r
538     result := system_accept(fdhandlein,fromaddr,fromaddrsize);\r
539   {$endif}\r
540   //now we have accepted one request start monitoring for more again\r
541   eventcore.rmasterset(fdhandlein,true);\r
542 \r
543   if result = -1 then begin\r
544     raise esocketexception.create('error '+inttostr({$ifdef win32}getlasterror{$else}socketerror{$endif})+' while accepting');\r
545   end;\r
546   if result > absoloutemaxs then begin\r
547     myfdclose(result);\r
548     a := result;\r
549     result := -1;\r
550     raise esocketexception.create('file discriptor out of range: '+inttostr(a));\r
551   end;\r
552 end;\r
553 \r
554 \r
555 function tlsocket.sendto(dest:TInetSockAddrV;destlen:integer;data:pointer;len:integer):integer;\r
556 var\r
557   {$ifdef ipv6}\r
558     realdest : tinetsockaddrv;\r
559     biniptemp : tbinip;\r
560   {$endif}\r
561   destx : {$ifdef win32}winsock.pSockAddr{$else}pInetSockAddrV{$endif};\r
562 \r
563 begin\r
564   {$ifdef secondlistener}\r
565     if assigned(secondlistener) then if (dest.inaddr.family = AF_INET) then begin\r
566       result := secondlistener.sendto(dest,destlen,data,len);\r
567       exit;\r
568     end;\r
569   {$endif}\r
570   {$ifdef ipv6}\r
571     if isv6socket then begin\r
572       biniptemp := inaddrvtobinip(dest);\r
573       converttov6(biniptemp);\r
574       destlen := makeinaddrv(biniptemp,inttostr(ntohs(dest.InAddr.port)),realdest);\r
575       destx := {$ifdef win32}winsock.pSockAddr{$else}pInetSockAddrV{$endif}(@realdest)\r
576     end else begin\r
577       destx := {$ifdef win32}winsock.pSockAddr{$else}pInetSockAddrV{$endif}(@dest)\r
578     end;\r
579   {$else}\r
580     destx := {$ifdef win32}winsock.pSockAddr{$else}pInetSockAddrV{$endif}(@dest)\r
581   {$endif}\r
582 \r
583   result := {$ifdef win32}winsock.sendto{$else}system_sendto{$endif}(self.fdhandleout,data^,len,0,destx^,destlen);\r
584 end;\r
585 \r
586 \r
587 function tlsocket.receivefrom(data:pointer;len:integer;var src:TInetSockAddrV;var srclen:integer):integer;\r
588 var\r
589   tempsrc:TInetSockAddrV;\r
590   tempsrclen:integer;\r
591   srcx : {$ifdef win32}winsock.TSockAddr{$else}TInetSockAddrV{$endif} absolute tempsrc;\r
592   biniptemp:tbinip;\r
593 begin\r
594   {$ifdef secondlistener}\r
595   if assigned(secondlistener) then if lastsessionfromsecond then begin\r
596     lastsessionfromsecond := false;\r
597     result := secondlistener.receivefrom(data,len,src,srclen);\r
598     exit;\r
599   end;\r
600   {$endif}\r
601   tempsrclen := sizeof(tempsrc);\r
602   result := recvfrom(self.fdhandlein,data^,len,0,srcx,tempsrclen);\r
603 \r
604   {$ifdef ipv6}\r
605   biniptemp := inaddrvtobinip(tempsrc);\r
606   if needconverttov4(biniptemp) then begin\r
607     converttov4(biniptemp);\r
608     tempsrclen := makeinaddrv(biniptemp,inttostr(ntohs(tempsrc.InAddr.port)),tempsrc);\r
609   end;\r
610   {$endif}\r
611 \r
612   move(tempsrc,src,srclen);\r
613   srclen := tempsrclen;\r
614 end;\r
615 \r
616 procedure tlsocket.taskcallconnectionfailedhandler(wparam,lparam : longint);\r
617 begin\r
618   connectionfailedhandler(wparam);\r
619 end;\r
620 \r
621 procedure tlsocket.connectionfailedhandler(error:word);\r
622 begin\r
623    if trymoreips then begin\r
624 //     writeln('failed with error ',error);\r
625      connecttimeout.enabled := false;\r
626      destroying := true;\r
627      state := wsconnected;\r
628      self.internalclose(0);\r
629      destroying := false;\r
630      realconnect;\r
631    end else begin\r
632      state := wsconnected;\r
633      if assigned(onsessionconnected) then onsessionconnected(self,error);\r
634      self.internalclose(0);\r
635      recvq.del(maxlongint);\r
636    end;\r
637 end;\r
638 \r
639 procedure tlsocket.connectsuccesshandler;\r
640 begin\r
641    trymoreips := false;\r
642    connecttimeout.enabled := false;\r
643    if assigned(onsessionconnected) then onsessionconnected(self,0);\r
644 end;\r
645 \r
646 \r
647 procedure tlsocket.handlefdtrigger(readtrigger,writetrigger:boolean);\r
648 var\r
649   tempbuf:array[0..receivebufsize-1] of byte;\r
650 begin\r
651 //  writeln('got a fd trigger, readtrigger=',readtrigger,' writetrigger=',writetrigger,' state=',integer(state));\r
652   if (state =wslistening) and readtrigger then begin\r
653 {    debugout('listening socket triggered on read');}\r
654     eventcore.rmasterclr(fdhandlein);\r
655     if assigned(onsessionAvailable) then onsessionAvailable(self,0);\r
656   end;\r
657   if dgram and readtrigger then begin\r
658     if assigned(ondataAvailable) then ondataAvailable(self,0);\r
659     {!!!test}\r
660     exit;\r
661   end;\r
662   if (state =wsconnecting) and writetrigger then begin\r
663     // code for dealing with the reults of a non-blocking connect is\r
664     // rather complex\r
665     // if just write is triggered it means connect suceeded\r
666     // if both read and write are triggered it can mean 2 things\r
667     // 1: connect ok and data availible\r
668     // 2: connect fail\r
669     // to find out which you must read from the socket and look for errors\r
670     // there if we read successfully we drop through into the code for fireing\r
671     // the read event\r
672     if not readtrigger then begin\r
673       state := wsconnected;\r
674       connectsuccesshandler;\r
675     end else begin\r
676       numread := myfdread(fdhandlein,tempbuf,sizeof(tempbuf));\r
677       if numread <> -1 then begin\r
678         state := wsconnected;\r
679         connectsuccesshandler;\r
680         //connectread := true;\r
681         recvq.add(@tempbuf,numread);\r
682       end else begin\r
683         connectionfailedhandler({$ifdef win32}wsagetlasterror{$else}linuxerror{$endif});\r
684         exit;\r
685       end;\r
686       // if things went well here we are now in the state wsconnected with data sitting in our receive buffer\r
687       // so we drop down into the processing for data availible\r
688     end;\r
689     if fdhandlein >= 0 then begin\r
690       if state = wsconnected then begin\r
691         eventcore.rmasterset(fdhandlein,false);\r
692       end else begin\r
693         eventcore.rmasterclr(fdhandlein);\r
694       end;\r
695     end;\r
696     if fdhandleout >= 0 then begin\r
697       if sendq.size = 0 then begin\r
698         //don't clear the bit in fdswmaster if data is in the sendq\r
699         eventcore.wmasterclr(fdhandleout);\r
700       end;\r
701     end;\r
702 \r
703   end;\r
704   inherited handlefdtrigger(readtrigger,writetrigger);\r
705 end;\r
706 \r
707 constructor tlsocket.Create(AOwner: TComponent);\r
708 begin\r
709   inherited create(aowner);\r
710   closehandles := true;\r
711   trymoreips := true;\r
712 end;\r
713 \r
714 \r
715 \r
716 function tlsocket.getpeername(var addr:tsockaddrin;addrlen:integer):integer;\r
717 var\r
718   addrx : {$ifdef win32}winsock.tsockaddr{$else}tsockaddrin{$endif} absolute addr;\r
719 begin\r
720   result := {$ifdef win32}winsock.getpeername{$else}system_getpeername{$endif}(self.fdhandlein,addrx,addrlen);\r
721 end;\r
722 \r
723 procedure tlsocket.getxaddrbin(var binip:tbinip);\r
724 var\r
725   addr:tinetsockaddrv;\r
726   i:integer;\r
727 begin\r
728   i := sizeof(addr);\r
729   fillchar(addr,sizeof(addr),0);\r
730 \r
731   {$ifdef win32}\r
732     winsock.getsockname(self.fdhandlein,psockaddr(@addr)^,i);\r
733   {$else}\r
734     getsocketname(self.fdhandlein,addr,i);\r
735   {$endif}\r
736   binip := inaddrvtobinip(addr);\r
737   converttov4(binip);\r
738 end;\r
739 \r
740 procedure tlsocket.getpeeraddrbin(var binip:tbinip);\r
741 var\r
742   addr:tinetsockaddrv;\r
743   i:integer;\r
744 begin\r
745   i := sizeof(addr);\r
746   fillchar(addr,sizeof(addr),0);\r
747   {$ifdef win32}\r
748     winsock.getpeername(self.fdhandlein,psockaddr(@addr)^,i);\r
749   {$else}\r
750     system_getpeername(self.fdhandlein,addr,i);\r
751   {$endif}\r
752 \r
753   binip := inaddrvtobinip(addr);\r
754   converttov4(binip);\r
755 end;\r
756 \r
757 function tlsocket.getXaddr:thostname;\r
758 var\r
759   biniptemp:tbinip;\r
760 begin\r
761   getxaddrbin(biniptemp);\r
762   result := ipbintostr(biniptemp);\r
763   if result = '' then result := 'error';\r
764 end;\r
765 \r
766 function tlsocket.getpeeraddr:thostname;\r
767 var\r
768   biniptemp:tbinip;\r
769 begin\r
770   getpeeraddrbin(biniptemp);\r
771   result := ipbintostr(biniptemp);\r
772   if result = '' then result := 'error';\r
773 end;\r
774 \r
775 function tlsocket.getXport:ansistring;\r
776 var\r
777   addr:tinetsockaddrv;\r
778   i:integer;\r
779 begin\r
780   i := sizeof(addr);\r
781   {$ifdef win32}\r
782     winsock.getsockname(self.fdhandlein,psockaddrin(@addr)^,i);\r
783 \r
784   {$else}\r
785     getsocketname(self.fdhandlein,addr,i);\r
786 \r
787   {$endif}\r
788   result := inttostr(htons(addr.InAddr.port));\r
789 end;\r
790 \r
791 function tlsocket.getpeerport:ansistring;\r
792 var\r
793   addr:tinetsockaddrv;\r
794   i:integer;\r
795 begin\r
796   i := sizeof(addr);\r
797   {$ifdef win32}\r
798     winsock.getpeername(self.fdhandlein,psockaddrin(@addr)^,i);\r
799 \r
800   {$else}\r
801     system_getpeername(self.fdhandlein,addr,i);\r
802 \r
803   {$endif}\r
804   result := inttostr(htons(addr.InAddr.port));\r
805 end;\r
806 \r
807 {$ifdef win32}\r
808   procedure tlsocket.myfdclose(fd : integer);\r
809   begin\r
810     closesocket(fd);\r
811   end;\r
812   function tlsocket.myfdwrite(fd: LongInt;const buf;size: LongInt):LongInt;\r
813   begin\r
814     result := winsock.send(fd,(@buf)^,size,0);\r
815   end;\r
816   function tlsocket.myfdread(fd: LongInt;var buf;size: LongInt):LongInt;\r
817   begin\r
818     result := winsock.recv(fd,buf,size,0);\r
819   end;\r
820 {$endif}\r
821 \r
822 end.\r
823 \r