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