ICMP does work, removing comment
[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:string;\r
98       port:string;\r
99       localaddr:string;\r
100       localport:string;\r
101       proto:string;\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 : string);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:string; virtual;\r
127       function getpeeraddr:string; virtual;\r
128       function getXport:string; virtual;\r
129       function getpeerport:string; 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       procedure taskcallconnectionfailedhandler(wparam,lparam : longint);\r
136 \r
137       procedure connecttimeouthandler(sender:tobject);\r
138       procedure connectsuccesshandler;\r
139       {$ifdef win32}\r
140         procedure myfdclose(fd : integer); override;\r
141         function myfdwrite(fd: LongInt;const buf;size: LongInt):LongInt; override;\r
142         function myfdread(fd: LongInt;var buf;size: LongInt):LongInt; override;\r
143       {$endif}\r
144     end;\r
145     tsocket=longint; // for compatibility with twsocket\r
146 \r
147   twsocket=tlsocket; {easy}\r
148 \r
149 \r
150 const\r
151   TCP_NODELAY=1;\r
152   IPPROTO_TCP=6;\r
153 \r
154 implementation\r
155 {$include unixstuff.inc}\r
156 \r
157 \r
158 function tlsocket.getaddrsize:integer;\r
159 begin\r
160   result := inaddrsize(inaddr);\r
161 end;\r
162 \r
163 \r
164 procedure tlsocket.realconnect;\r
165 var\r
166   a,b:integer;\r
167 begin\r
168   //writeln('trying to connect to ',ipbintostr(biniplist_get(biniplist,currentip)),'#',port);\r
169   makeinaddrv(biniplist_get(biniplist,currentip),port,inaddr);\r
170   inc(currentip);\r
171   if (currentip >= biniplist_getcount(biniplist)) then trymoreips := false;\r
172 \r
173   udp := false;\r
174   if (uppercase(proto) = 'UDP') then begin\r
175     b := IPPROTO_UDP;\r
176     a := SOCK_DGRAM;\r
177     udp := true;\r
178     dgram := true;\r
179   end else if (uppercase(proto) = 'TCP') or (uppercase(proto) = '') then begin\r
180     b := IPPROTO_TCP;\r
181     a := SOCK_STREAM;\r
182     dgram := false;\r
183   end else if (uppercase(proto) = 'ICMP') or (strtointdef(proto,256) < 256) then begin\r
184     b := strtointdef(proto,IPPROTO_ICMP);\r
185     a := SOCK_RAW;\r
186     dgram := true;\r
187   end else begin\r
188     raise ESocketException.create('unrecognised protocol');\r
189   end;\r
190 \r
191   a := Socket(inaddr.inaddr.family,a,b);\r
192   //writeln(ord(inaddr.inaddr.family));\r
193   if a = -1 then begin\r
194     //unable to create socket, fire an error event (better to use an error event\r
195     //to avoid poor interaction with multilistener stuff.\r
196     //a socket value of -2 is a special value to say there is no socket but\r
197     //we want internalclose to act as if there was\r
198     fdhandlein := -2;\r
199     fdhandleout := -2;\r
200     tltask.create(taskcallconnectionfailedhandler,self,{$ifdef win32}wsagetlasterror{$else}socketerror{$endif},0);\r
201     exit;\r
202   end;\r
203   try\r
204     dup(a);\r
205     bindsocket;\r
206     if dgram then begin\r
207       {$ifndef win32}\r
208         SetSocketOptions(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
209       {$else}\r
210         SetSockOpt(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
211       {$endif}\r
212       state := wsconnected;\r
213       if assigned(onsessionconnected) then onsessionconnected(self,0);\r
214 \r
215       eventcore.rmasterset(fdhandlein,false);\r
216       eventcore.wmasterclr(fdhandleout);\r
217     end else begin\r
218       state :=wsconnecting;\r
219       {$ifdef win32}\r
220         //writeln(inaddr.inaddr.port);\r
221         winsock.Connect(fdhandlein,winsock.psockaddr(@inADDR)^,getaddrsize);\r
222       {$else}\r
223         sockets.Connect(fdhandlein,inADDR,getaddrsize);\r
224       {$endif}\r
225       eventcore.rmasterset(fdhandlein,false);\r
226       eventcore.wmasterset(fdhandleout);\r
227       if trymoreips then connecttimeout.enabled := true;\r
228     end;\r
229     //sendq := '';\r
230   except\r
231     on e: exception do begin\r
232       fdcleanup;\r
233       raise; //reraise the exception\r
234     end;\r
235   end;\r
236 \r
237 end;\r
238 \r
239 procedure tlsocket.connecttimeouthandler(sender:tobject);\r
240 begin\r
241   connecttimeout.enabled := false;\r
242   destroying := true; //hack to not cause handler to trigger\r
243   internalclose(0);\r
244   destroying := false;\r
245   realconnect;\r
246 end;\r
247 \r
248 procedure tlsocket.connect;\r
249 var\r
250   a:integer;\r
251   ip:tbinip;\r
252 begin\r
253   if state <> wsclosed then close;\r
254   //prevtime := 0;\r
255   if isbiniplist(addr) then biniplist := addr else biniplist := forwardlookuplist(addr,0);\r
256   if biniplist_getcount(biniplist) = 0 then raise exception.create('unable to resolve '+addr);\r
257 \r
258   //makeinaddrv(addr,port,inaddr);\r
259 \r
260   currentip := 0;\r
261   if not assigned(connecttimeout) then begin\r
262     connecttimeout := tltimer.create(self);\r
263     connecttimeout.Tag := integer(self);\r
264     connecttimeout.ontimer := connecttimeouthandler;\r
265     connecttimeout.interval := 2500;\r
266     connecttimeout.enabled := false;\r
267   end;\r
268   realconnect;\r
269 end;\r
270 \r
271 procedure tlsocket.sendstr(const str : string);\r
272 begin\r
273   if dgram then begin\r
274     send(@str[1],length(str))\r
275   end else begin\r
276     inherited sendstr(str);\r
277   end;\r
278 end;\r
279 \r
280 function tlsocket.send(data:pointer;len:integer):integer;\r
281 begin\r
282   if dgram then begin\r
283 //    writeln('sending to '+ipbintostr(inaddrvtobinip(inaddr)),' ',htons(inaddr.inaddr.port),' ',len,' bytes');\r
284     result := sendto(inaddr,getaddrsize,data,len);\r
285 \r
286 //    writeln('send result ',result);\r
287 //    writeln('errno',errno);\r
288   end else begin\r
289     result := inherited send(data,len);\r
290   end;\r
291 end;\r
292 \r
293 \r
294 function tlsocket.receive(Buf:Pointer;BufSize:integer):integer;\r
295 begin\r
296   if dgram then begin\r
297     {$ifdef secondlistener}\r
298     if lastsessionfromsecond then begin\r
299       result := secondlistener.receive(buf,bufsize);\r
300       lastsessionfromsecond := false;\r
301     end else\r
302     {$endif}\r
303       result := myfdread(self.fdhandlein,buf^,bufsize);\r
304   end else begin\r
305     result := inherited receive(buf,bufsize);\r
306   end;\r
307 end;\r
308 \r
309 procedure tlsocket.bindsocket;\r
310 var\r
311   a:integer;\r
312   inAddrtemp:TInetSockAddrV;\r
313   inAddrtempx:{$ifdef win32}winsock.TSockaddr{$else}TInetSockAddrV{$endif} absolute inaddrtemp;\r
314   inaddrtempsize:integer;\r
315 begin\r
316   try\r
317     if (localaddr <> '') or (localport <> '') then begin\r
318       if localaddr = '' then begin\r
319         {$ifdef ipv6}\r
320         if inaddr.inaddr.family = AF_INET6 then localaddr := '::' else\r
321         {$endif}\r
322         localaddr := '0.0.0.0';\r
323       end;\r
324       //gethostbyname(localaddr,host);\r
325       inaddrtempsize := makeinaddrv(forwardlookup(localaddr,0),localport,inaddrtemp);\r
326 \r
327       If Bind(fdhandlein,inaddrtempx,inaddrtempsize)<> {$ifdef win32}0{$else}true{$endif} Then begin\r
328         state := wsclosed;\r
329         lasterror := {$ifdef win32}getlasterror{$else}socketerror{$endif};\r
330         raise ESocketException.create('unable to bind on address '+localaddr+'#'+localport+', error '+inttostr(lasterror));\r
331       end;\r
332       state := wsbound;\r
333     end;\r
334   except\r
335     on e: exception do begin\r
336       fdcleanup;\r
337       raise; //reraise the exception\r
338     end;\r
339   end;\r
340 end;\r
341 \r
342 procedure tlsocket.listen;\r
343 var\r
344   yes:longint;\r
345   socktype:integer;\r
346   biniptemp:tbinip;\r
347   origaddr:string;\r
348 begin\r
349   if state <> wsclosed then close;\r
350   udp := uppercase(proto) = 'UDP';\r
351   if udp then begin\r
352     socktype := SOCK_DGRAM;\r
353     dgram := true;\r
354   end else socktype := SOCK_STREAM;\r
355   origaddr := addr;\r
356 \r
357   if addr = '' then begin\r
358     {$ifdef ipv6}\r
359     if not v4listendefault then begin\r
360       addr := '::';\r
361     end else\r
362     {$endif}\r
363     addr := '0.0.0.0';\r
364   end;\r
365   if isbiniplist(addr) then biniptemp := biniplist_get(addr,0) else biniptemp := forwardlookup(addr,10);\r
366   addr := ipbintostr(biniptemp);\r
367   fdhandlein := socket(biniptemp.family,socktype,0);\r
368   {$ifdef ipv6}\r
369   if (addr = '::') and (origaddr = '') and (fdhandlein < 0) then begin\r
370     addr := '0.0.0.0';\r
371     fdhandlein := socket(AF_INET,socktype,0);\r
372   end;\r
373   {$endif}\r
374 \r
375   if fdhandlein = -1 then raise ESocketException.create('unable to create socket'{$ifdef win32}+' error='+inttostr(wsagetlasterror){$endif});\r
376   dupnowatch(fdhandlein); // sets up maxs and copies handle to fdhandleout among other things\r
377   //eventcore.setfdreverse(fdhandlein,self); //already taken care of by dup\r
378   state := wsclosed; // then set this back as it was an undesired side effect of dup\r
379 \r
380   try\r
381     yes := $01010101;  {Copied this from existing code. Value is empiric,\r
382                     but works. (yes=true<>0) }\r
383     {$ifndef win32}\r
384       if SetSocketOptions(fdhandlein, SOL_SOCKET, SO_REUSEADDR,yes,sizeof(yes))=-1 then begin\r
385         raise ESocketException.create('unable to set socket options');\r
386       end;\r
387     {$endif}\r
388     localaddr := addr;\r
389     localport := port;\r
390     bindsocket;\r
391 \r
392     if not udp then begin\r
393       {!!! allow custom queue length? default 5}\r
394       if listenqueue = 0 then listenqueue := 5;\r
395       If {$ifdef win32}winsock{$else}sockets{$endif}.Listen(fdhandlein,listenqueue)<>{$ifdef win32}0{$else}true{$endif} Then raise\r
396 esocketexception.create('unable to listen');\r
397       state := wsListening;\r
398     end else begin\r
399       {$ifndef win32}\r
400         SetSocketOptions(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
401       {$else}\r
402         SetSockOpt(fdhandleout, SOL_SOCKET, SO_BROADCAST, 'TRUE', Length('TRUE'));\r
403       {$endif}\r
404       state := wsconnected;\r
405     end;\r
406 \r
407     {$ifdef secondlistener}\r
408     //listening on ::. try to listen on 0.0.0.0 as well for platforms which don't already do that\r
409     if addr = '::' then begin\r
410       secondlistener := tlsocket.create(nil);\r
411       secondlistener.proto := proto;\r
412       secondlistener.addr := '0.0.0.0';\r
413       secondlistener.port := port;\r
414       if udp then begin\r
415         secondlistener.ondataavailable := secondaccepthandler;\r
416       end else begin\r
417         secondlistener.onsessionAvailable := secondaccepthandler;\r
418       end;\r
419       try\r
420         secondlistener.listen;\r
421       except\r
422         secondlistener.destroy;\r
423         secondlistener := nil;\r
424       end;\r
425     end;\r
426     {$endif}\r
427   finally\r
428     if state = wsclosed then begin\r
429       if fdhandlein >= 0 then begin\r
430         {one *can* get here without fd -beware}\r
431         eventcore.rmasterclr(fdhandlein);\r
432         myfdclose(fdhandlein); // we musnt leak file discriptors\r
433         eventcore.setfdreverse(fdhandlein,nil);\r
434         fdhandlein := -1;\r
435       end;\r
436     end else begin\r
437       eventcore.rmasterset(fdhandlein,not udp);\r
438     end;\r
439     if fdhandleout >= 0 then eventcore.wmasterclr(fdhandleout);\r
440   end;\r
441   //writeln('listened on addr '+addr+':'+port+' '+proto+' using socket number ',fdhandlein);\r
442 end;\r
443 \r
444 {$ifdef secondlistener}\r
445 procedure tlsocket.internalclose(error:word);\r
446 begin\r
447   if assigned(secondlistener) then begin\r
448     secondlistener.destroy;\r
449     secondlistener := nil;\r
450   end;\r
451   inherited internalclose(error);\r
452 end;\r
453 \r
454 procedure tlsocket.secondaccepthandler;\r
455 begin\r
456   lastsessionfromsecond := true;\r
457   if udp then begin\r
458     ondataavailable(self,error);\r
459   end else begin\r
460     if assigned(onsessionavailable) then onsessionavailable(self,error);\r
461   end;\r
462 end;\r
463 {$endif}\r
464 \r
465 function tlsocket.accept : longint;\r
466 var\r
467   FromAddrSize     : LongInt;        // i don't realy know what to do with these at this\r
468   FromAddr         : TInetSockAddrV;  // at this point time will tell :)\r
469   a:integer;\r
470 begin\r
471   {$ifdef secondlistener}\r
472   if (lastsessionfromsecond) then begin\r
473     lastsessionfromsecond := false;\r
474     result := secondlistener.accept;\r
475     exit;\r
476   end;\r
477   {$endif}\r
478 \r
479   FromAddrSize := Sizeof(FromAddr);\r
480   {$ifdef win32}\r
481     result := winsock.accept(fdhandlein,@fromaddr,@fromaddrsize);\r
482   {$else}\r
483     result := sockets.accept(fdhandlein,fromaddr,fromaddrsize);\r
484   {$endif}\r
485   //now we have accepted one request start monitoring for more again\r
486   eventcore.rmasterset(fdhandlein,true);\r
487 \r
488   if result = -1 then begin\r
489     raise esocketexception.create('error '+inttostr({$ifdef win32}getlasterror{$else}socketerror{$endif})+' while accepting');\r
490   end;\r
491   if result > absoloutemaxs then begin\r
492     myfdclose(result);\r
493     a := result;\r
494     result := -1;\r
495     raise esocketexception.create('file discriptor out of range: '+inttostr(a));\r
496   end;\r
497 end;\r
498 \r
499 function tlsocket.sendto(dest:TInetSockAddrV;destlen:integer;data:pointer;len:integer):integer;\r
500 var\r
501   destx : {$ifdef win32}winsock.TSockAddr{$else}TInetSockAddrV{$endif} absolute dest;\r
502 begin\r
503   {$ifdef secondlistener}\r
504   if assigned(secondlistener) then if (dest.inaddr.family = AF_INET) then begin\r
505     result := secondlistener.sendto(dest,destlen,data,len);\r
506     exit;\r
507   end;\r
508   {$endif}\r
509   result := {$ifdef win32}winsock{$else}sockets{$endif}.sendto(self.fdhandleout,data^,len,0,destx,destlen);\r
510 end;\r
511 \r
512 function tlsocket.receivefrom(data:pointer;len:integer;var src:TInetSockAddrV;var srclen:integer):integer;\r
513 var\r
514   tempsrc:TInetSockAddrV;\r
515   tempsrclen:integer;\r
516   srcx : {$ifdef win32}winsock.TSockAddr{$else}TInetSockAddrV{$endif} absolute tempsrc;\r
517   biniptemp:tbinip;\r
518 begin\r
519   {$ifdef secondlistener}\r
520   if assigned(secondlistener) then if lastsessionfromsecond then begin\r
521     lastsessionfromsecond := false;\r
522     result := secondlistener.receivefrom(data,len,src,srclen);\r
523     exit;\r
524   end;\r
525   {$endif}\r
526   tempsrclen := sizeof(tempsrc);\r
527   result := {$ifdef win32}winsock{$else}sockets{$endif}.recvfrom(self.fdhandlein,data^,len,0,srcx,tempsrclen);\r
528 \r
529   {$ifdef ipv6}\r
530   biniptemp := inaddrvtobinip(tempsrc);\r
531   if needconverttov4(biniptemp) then begin\r
532     converttov4(biniptemp);\r
533     tempsrclen := makeinaddrv(biniptemp,inttostr(ntohs(tempsrc.InAddr.port)),tempsrc);\r
534   end;\r
535   {$endif}\r
536 \r
537   move(tempsrc,src,srclen);\r
538   srclen := tempsrclen;\r
539 end;\r
540 \r
541 procedure tlsocket.taskcallconnectionfailedhandler(wparam,lparam : longint);\r
542 begin\r
543   connectionfailedhandler(wparam);\r
544 end;\r
545 \r
546 procedure tlsocket.connectionfailedhandler(error:word);\r
547 begin\r
548    if trymoreips then begin\r
549 //     writeln('failed with error ',error);\r
550      connecttimeout.enabled := false;\r
551      destroying := true;\r
552      state := wsconnected;\r
553      self.internalclose(0);\r
554      destroying := false;\r
555      realconnect;\r
556    end else begin\r
557      state := wsconnected;\r
558      if assigned(onsessionconnected) then onsessionconnected(self,error);\r
559      self.internalclose(0);\r
560      recvq.del(maxlongint);\r
561    end;\r
562 end;\r
563 \r
564 procedure tlsocket.connectsuccesshandler;\r
565 begin\r
566    trymoreips := false;\r
567    connecttimeout.enabled := false;\r
568    if assigned(onsessionconnected) then onsessionconnected(self,0);\r
569 end;\r
570 \r
571 \r
572 procedure tlsocket.handlefdtrigger(readtrigger,writetrigger:boolean);\r
573 var\r
574   tempbuf:array[0..receivebufsize-1] of byte;\r
575 begin\r
576 //  writeln('got a fd trigger, readtrigger=',readtrigger,' writetrigger=',writetrigger,' state=',integer(state));\r
577   if (state =wslistening) and readtrigger then begin\r
578 {    debugout('listening socket triggered on read');}\r
579     eventcore.rmasterclr(fdhandlein);\r
580     if assigned(onsessionAvailable) then onsessionAvailable(self,0);\r
581   end;\r
582   if dgram and readtrigger then begin\r
583     if assigned(ondataAvailable) then ondataAvailable(self,0);\r
584     {!!!test}\r
585     exit;\r
586   end;\r
587   if (state =wsconnecting) and writetrigger then begin\r
588     // code for dealing with the reults of a non-blocking connect is\r
589     // rather complex\r
590     // if just write is triggered it means connect suceeded\r
591     // if both read and write are triggered it can mean 2 things\r
592     // 1: connect ok and data availible\r
593     // 2: connect fail\r
594     // to find out which you must read from the socket and look for errors\r
595     // there if we read successfully we drop through into the code for fireing\r
596     // the read event\r
597     if not readtrigger then begin\r
598       state := wsconnected;\r
599       connectsuccesshandler;\r
600     end else begin\r
601       numread := myfdread(fdhandlein,tempbuf,sizeof(tempbuf));\r
602       if numread <> -1 then begin\r
603         state := wsconnected;\r
604         connectsuccesshandler;\r
605         //connectread := true;\r
606         recvq.add(@tempbuf,numread);\r
607       end else begin\r
608         connectionfailedhandler({$ifdef win32}wsagetlasterror{$else}linuxerror{$endif});\r
609         exit;\r
610       end;\r
611       // if things went well here we are now in the state wsconnected with data sitting in our receive buffer\r
612       // so we drop down into the processing for data availible\r
613     end;\r
614     if fdhandlein >= 0 then begin\r
615       if state = wsconnected then begin\r
616         eventcore.rmasterset(fdhandlein,false);\r
617       end else begin\r
618         eventcore.rmasterclr(fdhandlein);\r
619       end;\r
620     end;\r
621     if fdhandleout >= 0 then begin\r
622       if sendq.size = 0 then begin\r
623         //don't clear the bit in fdswmaster if data is in the sendq\r
624         eventcore.wmasterclr(fdhandleout);\r
625       end;\r
626     end;\r
627 \r
628   end;\r
629   inherited handlefdtrigger(readtrigger,writetrigger);\r
630 end;\r
631 \r
632 constructor tlsocket.Create(AOwner: TComponent);\r
633 begin\r
634   inherited create(aowner);\r
635   closehandles := true;\r
636   trymoreips := true;\r
637 end;\r
638 \r
639 \r
640 function tlsocket.getpeername(var addr:tsockaddrin;addrlen:integer):integer;\r
641 var\r
642   addrx : {$ifdef win32}winsock.tsockaddr{$else}tsockaddrin{$endif} absolute addr;\r
643 begin\r
644   result := {$ifdef win32}winsock{$else}sockets{$endif}.getpeername(self.fdhandlein,addrx,addrlen);\r
645 end;\r
646 \r
647 procedure tlsocket.getxaddrbin(var binip:tbinip);\r
648 var\r
649   addr:tinetsockaddrv;\r
650   i:integer;\r
651 begin\r
652   i := sizeof(addr);\r
653   fillchar(addr,sizeof(addr),0);\r
654 \r
655   {$ifdef win32}\r
656     winsock.getsockname(self.fdhandlein,psockaddr(@addr)^,i);\r
657   {$else}\r
658     sockets.getsocketname(self.fdhandlein,addr,i);\r
659   {$endif}\r
660   binip := inaddrvtobinip(addr);\r
661   converttov4(binip);\r
662 end;\r
663 \r
664 procedure tlsocket.getpeeraddrbin(var binip:tbinip);\r
665 var\r
666   addr:tinetsockaddrv;\r
667   i:integer;\r
668 begin\r
669   i := sizeof(addr);\r
670   fillchar(addr,sizeof(addr),0);\r
671   {$ifdef win32}\r
672     winsock.getpeername(self.fdhandlein,psockaddr(@addr)^,i);\r
673   {$else}\r
674     sockets.getpeername(self.fdhandlein,addr,i);\r
675   {$endif}\r
676 \r
677   binip := inaddrvtobinip(addr);\r
678   converttov4(binip);\r
679 end;\r
680 \r
681 function tlsocket.getXaddr:string;\r
682 var\r
683   biniptemp:tbinip;\r
684 begin\r
685   getxaddrbin(biniptemp);\r
686   result := ipbintostr(biniptemp);\r
687   if result = '' then result := 'error';\r
688 end;\r
689 \r
690 function tlsocket.getpeeraddr:string;\r
691 var\r
692   biniptemp:tbinip;\r
693 begin\r
694   getpeeraddrbin(biniptemp);\r
695   result := ipbintostr(biniptemp);\r
696   if result = '' then result := 'error';\r
697 end;\r
698 \r
699 function tlsocket.getXport:string;\r
700 var\r
701   addr:tinetsockaddrv;\r
702   i:integer;\r
703 begin\r
704   i := sizeof(addr);\r
705   {$ifdef win32}\r
706     winsock.getsockname(self.fdhandlein,psockaddrin(@addr)^,i);\r
707 \r
708   {$else}\r
709     sockets.getsocketname(self.fdhandlein,addr,i);\r
710 \r
711   {$endif}\r
712   result := inttostr(htons(addr.InAddr.port));\r
713 end;\r
714 \r
715 function tlsocket.getpeerport:string;\r
716 var\r
717   addr:tinetsockaddrv;\r
718   i:integer;\r
719 begin\r
720   i := sizeof(addr);\r
721   {$ifdef win32}\r
722     winsock.getpeername(self.fdhandlein,psockaddrin(@addr)^,i);\r
723 \r
724   {$else}\r
725     sockets.getpeername(self.fdhandlein,addr,i);\r
726 \r
727   {$endif}\r
728   result := inttostr(htons(addr.InAddr.port));\r
729 end;\r
730 \r
731 {$ifdef win32}\r
732   procedure tlsocket.myfdclose(fd : integer);\r
733   begin\r
734     closesocket(fd);\r
735   end;\r
736   function tlsocket.myfdwrite(fd: LongInt;const buf;size: LongInt):LongInt;\r
737   begin\r
738     result := winsock.send(fd,(@buf)^,size,0);\r
739   end;\r
740   function tlsocket.myfdread(fd: LongInt;var buf;size: LongInt):LongInt;\r
741   begin\r
742     result := winsock.recv(fd,buf,size,0);\r
743   end;\r
744 {$endif}\r
745 \r
746 end.\r
747 \r