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