| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379 |
- {
- $Project$
- $Workfile$
- $Revision$
- $DateUTC$
- $Id$
- This file is part of the Indy (Internet Direct) project, and is offered
- under the dual-licensing agreement described on the Indy website.
- (http://www.indyproject.org/)
- Copyright:
- (c) 1993-2005, Chad Z. Hower and the Indy Pit Crew. All rights reserved.
- $Log$
- Rev 1.7 10/26/2004 8:20:04 PM JPMugaas
- Fixed some oversights with conversion. OOPS!!!
- Rev 1.6 10/26/2004 8:12:32 PM JPMugaas
- Now uses TIdStrings and TIdStringList for portability.
- Rev 1.5 12/06/2004 15:17:20 CCostelloe
- Restructured to correspond with IdStackWindows, now works.
- Rev 1.4 07/06/2004 21:31:02 CCostelloe
- Kylix 3 changes
- Rev 1.3 4/18/04 10:43:22 PM RLebeau
- Fixed syntax error
- Rev 1.2 4/18/04 10:29:46 PM RLebeau
- Renamed Int64Parts structure to TIdInt64Parts
- Rev 1.1 4/18/04 2:47:28 PM RLebeau
- Conversion support for Int64 values
- Removed WSHToNs(), WSNToHs(), WSHToNL(), and WSNToHL() methods, obsolete
- Rev 1.0 2004.02.03 3:14:48 PM czhower
- Move and updates
- Rev 1.3 10/19/2003 5:35:14 PM BGooijen
- SetSocketOption
- Rev 1.2 2003.10.01 9:11:24 PM czhower
- .Net
- Rev 1.1 7/5/2003 07:25:50 PM JPMugaas
- Added functions to the Linux stack which use the new TIdIPAddress record type
- for IP address parameters. I also fixed a compile bug.
- Rev 1.0 11/13/2002 08:59:24 AM JPMugaas
- }
- unit IdStackUnix;
- interface
- {$i IdCompilerDefines.inc}
- {$IFNDEF FPC}
- {$Message Fatal 'IdStackUnix is only for FreePascal.'}
- {$ENDIF}
- uses
- Classes,
- sockets,
- baseunix,
- IdStack,
- IdStackConsts,
- IdGlobal,
- IdStackBSDBase;
- {$IFDEF FREEBSD}
- {$DEFINE SOCK_HAS_SINLEN}
- {$ENDIF}
- {$IFDEF DARWIN}
- {$DEFINE SOCK_HAS_SINLEN}
- {$ENDIF}
- type
- {$IFNDEF NO_REDECLARE}
- Psockaddr = ^sockaddr;
- {$ENDIF}
- // TODO: move this class into the implementation section! It is not used outside of this unit
- TIdSocketListUnix = class (TIdSocketList)
- protected
- FCount: Integer;
- FFDSet: TFDSet;
- //
- class function FDSelect(AReadSet: PFDSet; AWriteSet: PFDSet; AExceptSet: PFDSet;
- const ATimeout: Integer = IdTimeoutInfinite): Integer;
- function GetItem(AIndex: Integer): TIdStackSocketHandle; override;
- public
- procedure Add(AHandle: TIdStackSocketHandle); override;
- procedure Remove(AHandle: TIdStackSocketHandle); override;
- function Count: Integer; override;
- procedure Clear; override;
- function Clone: TIdSocketList; override;
- function ContainsSocket(AHandle: TIdStackSocketHandle): Boolean; override;
- procedure GetFDSet(var VSet: TFDSet);
- procedure SetFDSet(var VSet: TFDSet);
- class function Select(AReadList: TIdSocketList; AWriteList: TIdSocketList;
- AExceptList: TIdSocketList; const ATimeout: Integer = IdTimeoutInfinite): Boolean; override;
- function SelectRead(const ATimeout: Integer = IdTimeoutInfinite): Boolean; override;
- function SelectReadList(var VSocketList: TIdSocketList;
- const ATimeout: Integer = IdTimeoutInfinite): Boolean; override;
- end;
- TIdStackUnix = class(TIdStackBSDBase)
- protected
- procedure WriteChecksumIPv6(s: TIdStackSocketHandle; var VBuffer: TIdBytes;
- const AOffset: Integer; const AIP: String; const APort: TIdPort);
- function GetLastError: Integer;
- procedure SetLastError(const AError: Integer);
- function HostByName(const AHostName: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string; override;
- function ReadHostName: string; override;
- function WSCloseSocket(ASocket: TIdStackSocketHandle): Integer; override;
- function WSRecv(ASocket: TIdStackSocketHandle; var ABuffer;
- const ABufferLength, AFlags: Integer): Integer; override;
- function WSSend(ASocket: TIdStackSocketHandle; const ABuffer;
- const ABufferLength, AFlags: Integer): Integer; override;
- function WSShutdown(ASocket: TIdStackSocketHandle; AHow: Integer): Integer; override;
- {$IFNDEF VCL_XE3_OR_ABOVE}
- procedure WSGetSocketOption(ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel;
- AOptName: TIdSocketOption; var AOptVal; var AOptLen: Integer); override;
- procedure WSSetSocketOption(ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel;
- AOptName: TIdSocketOption; const AOptVal; const AOptLen: Integer); override;
- {$ENDIF}
- public
- constructor Create; override;
- destructor Destroy; override;
- procedure SetBlocking(ASocket: TIdStackSocketHandle; const ABlocking: Boolean); override;
- function WouldBlock(const AResult: Integer): Boolean; override;
- function WSTranslateSocketErrorMsg(const AErr: Integer): string; override;
- function Accept(ASocket: TIdStackSocketHandle; var VIP: string; var VPort: TIdPort;
- var VIPVersion: TIdIPVersion): TIdStackSocketHandle; override;
- procedure Bind(ASocket: TIdStackSocketHandle; const AIP: string;
- const APort: TIdPort; const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION); override;
- procedure Connect(const ASocket: TIdStackSocketHandle; const AIP: string;
- const APort: TIdPort; const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION); override;
- function HostByAddress(const AAddress: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string; override;
- function WSGetLastError: Integer; override;
- procedure WSSetLastError(const AErr : Integer); override;
- function WSGetServByName(const AServiceName: string): TIdPort; override;
- procedure AddServByPortToList(const APortNumber: TIdPort; AAddresses: TStrings); override;
- procedure GetPeerName(ASocket: TIdStackSocketHandle; var VIP: string;
- var VPort: TIdPort; var VIPVersion: TIdIPVersion); override;
- procedure GetSocketName(ASocket: TIdStackSocketHandle; var VIP: string;
- var VPort: TIdPort; var VIPVersion: TIdIPVersion); override;
- procedure Listen(ASocket: TIdStackSocketHandle; ABackLog: Integer); override;
- function HostToNetwork(AValue: UInt16): UInt16; override;
- function NetworkToHost(AValue: UInt16): UInt16; override;
- function HostToNetwork(AValue: UInt32): UInt32; override;
- function NetworkToHost(AValue: UInt32): UInt32; override;
- function HostToNetwork(AValue: TIdUInt64): TIdUInt64; override;
- function NetworkToHost(AValue: TIdUInt64): TIdUInt64; override;
- function RecvFrom(const ASocket: TIdStackSocketHandle; var VBuffer;
- const ALength, AFlags: Integer; var VIP: string; var VPort: TIdPort;
- var VIPVersion: TIdIPVersion): Integer; override;
- function ReceiveMsg(ASocket: TIdStackSocketHandle;
- var VBuffer: TIdBytes; APkt: TIdPacketInfo): UInt32; override;
- procedure WSSendTo(ASocket: TIdStackSocketHandle; const ABuffer;
- const ABufferLength, AFlags: Integer; const AIP: string; const APort: TIdPort;
- AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION); override;
- function WSSocket(AFamily, AStruct, AProtocol: Integer;
- const ANonBlocking: Boolean = False): TIdStackSocketHandle; override;
- procedure Disconnect(ASocket: TIdStackSocketHandle); override;
- {$IFDEF VCL_XE3_OR_ABOVE}
- procedure GetSocketOption(ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel;
- AOptName: TIdSocketOption; var AOptVal; var AOptLen: Integer); override;
- procedure SetSocketOption(ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel;
- AOptName: TIdSocketOption; const AOptVal; const AOptLen: Integer); override;
- {$ENDIF}
- procedure SetKeepAliveValues(ASocket: TIdStackSocketHandle;
- const AEnabled: Boolean; const ATimeMS, AInterval: Integer); override;
- function SupportsIPv4: Boolean; overload; override;
- function SupportsIPv6: Boolean; overload; override;
- function CheckIPVersionSupport(const AIPVersion: TIdIPVersion): boolean; override;
- //In Windows, this writes a checksum into a buffer. In Linux, it would probably
- //simply have the kernal write the checksum with something like this (RFC 2292):
- //
- // int offset = 2;
- // setsockopt(fd, IPPROTO_IPV6, IPV6_CHECKSUM, &offset, sizeof(offset));
- //
- // Note that this should be called
- //IMMEDIATELY before you do a SendTo because the Local IPv6 address might change
- procedure WriteChecksum(s : TIdStackSocketHandle; var VBuffer : TIdBytes;
- const AOffset : Integer; const AIP : String; const APort : TIdPort;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION); override;
- function IOControl(const s: TIdStackSocketHandle; const cmd: UInt32;
- var arg: UInt32): Integer; override;
- procedure GetLocalAddressList(AAddresses: TIdStackLocalAddressList); override;
- end;
- {$IFNDEF NO_REDECLARE}
- TLinger = record
- l_onoff: UInt16;
- l_linger: UInt16;
- end;
- {$ENDIF}
- TIdLinger = TLinger;
- implementation
- uses
- netdb,
- unix,
- termio,
- IdResourceStrings,
- IdResourceStringsUnix,
- IdException,
- SysUtils;
- //from: netdbh.inc, we can not include it with the I derrective and netdb.pas
- //does not expose it.
- {const
- EAI_SYSTEM = -(11);}
- const
- FD_SETSIZE = FD_MAXFDSET;
- __FD_SETSIZE = FD_MAXFDSET;
- {$IFDEF DARWIN}
- { MSG_NOSIGNAL does not exist in OS X. Instead we have SO_NOSIGPIPE, which we set in Connect. }
- Id_MSG_NOSIGNAL = 0;
- {$ELSE}
- Id_MSG_NOSIGNAL = MSG_NOSIGNAL;
- {$ENDIF}
- ESysEPIPE = ESysEPIPE;
- //helper functions for some structs
- {Note: These hide an API difference in structures.
- BSD 4.4 introduced a minor API change. sa_family was changed from a 16bit
- word to an 8 bit byteee and an 8 bit byte feild named sa_len was added.
- }
- procedure InitSockaddr(var VSock : Sockaddr);
- {$IFDEF USE_INLINE} inline; {$ENDIF}
- begin
- FillChar(VSock, SizeOf(Sockaddr), 0);
- VSock.sin_family := PF_INET;
- {$IFDEF SOCK_HAS_SINLEN}
- VSock.sa_len := SizeOf(Sockaddr);
- {$ENDIF}
- end;
- procedure InitSockAddr_in6(var VSock : SockAddr_in6);
- {$IFDEF USE_INLINE} inline; {$ENDIF}
- begin
- FillChar(VSock, SizeOf(SockAddr_in6), 0);
- {$IFDEF SOCK_HAS_SINLEN}
- VSock.sin6_len := SizeOf(SockAddr_in6);
- {$ENDIF}
- VSock.sin6_family := PF_INET6;
- end;
- //
- constructor TIdStackUnix.Create;
- begin
- inherited Create;
- end;
- destructor TIdStackUnix.Destroy;
- begin
- inherited Destroy;
- end;
- function TIdStackUnix.GetLastError : Integer;
- begin
- Result := SocketError;
- end;
- procedure TIdStackUnix.SetLastError(Const AError : Integer);
- begin
- errno := AError;
- end;
- function TIdStackUnix.Accept(ASocket: TIdStackSocketHandle;
- var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion): TIdStackSocketHandle;
- var
- LA : socklen_t;
- LAddr: sockaddr_in6;
- begin
- LA := SizeOf(LAddr);
- Result := fpaccept(ASocket, @LAddr, @LA);
- //calls prefixed by fp to avoid clashing with libc
- if Result <> ID_SOCKET_ERROR then begin
- case LAddr.sin6_family of
- PF_INET : begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := NetAddrToStr(sin_addr);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPv4;
- end;
- PF_INET6: begin
- with LAddr do
- begin
- VIP := NetAddrToStr6(sin6_addr);
- VPort := Ntohs(sin6_port);
- end;
- VIPVersion := Id_IPv6;
- end;
- else begin
- fpclose(Result);
- Result := Id_INVALID_SOCKET;
- IPVersionUnsupported;
- end;
- end;
- end else begin
- if GetLastError = ESysEBADF then begin
- SetLastError(ESysEINTR);
- end;
- end;
- end;
- procedure TIdStackUnix.Bind(ASocket: TIdStackSocketHandle; const AIP: string;
- const APort: TIdPort; const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
- var
- LAddr: sockaddr_in6;
- begin
- case AIPVersion of
- Id_IPv4: begin
- InitSockAddr(Psockaddr(@LAddr)^);
- with Psockaddr(@LAddr)^ do
- begin
- if AIP <> '' then begin
- sin_addr := StrToNetAddr(AIP);
- //TranslateStringToTInAddr(AIP, sin_addr, Id_IPv4);
- end;
- sin_port := htons(APort);
- end;
- CheckForSocketError(fpBind(ASocket, Psockaddr(@LAddr), SizeOf(sockaddr)));
- end;
- Id_IPv6: begin
- InitSockAddr_in6(LAddr);
- with LAddr do
- begin
- if AIP <> '' then begin
- sin6_addr := StrToNetAddr6(AIP);
- //TranslateStringToTInAddr(AIP, sin6_addr, Id_IPv6);
- end;
- sin6_port := htons(APort);
- end;
- CheckForSocketError(fpBind(ASocket, Psockaddr(@LAddr), SizeOf(Sockaddr_in6)));
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- function TIdStackUnix.WSCloseSocket(ASocket: TIdStackSocketHandle): Integer;
- begin
- Result := fpclose(ASocket);
- end;
- procedure TIdStackUnix.Connect(const ASocket: TIdStackSocketHandle;
- const AIP: string; const APort: TIdPort;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
- var
- LAddr: sockaddr_in6;
- begin
- case AIPVersion of
- Id_IPv4: begin
- InitSockAddr(Psockaddr(@LAddr)^);
- with Psockaddr(@LAddr)^ do
- begin
- sin_addr := StrToNetAddr(AIP);
- //TranslateStringToTInAddr(AIP, sin_addr, Id_IPv4);
- sin_port := htons(APort);
- end;
- CheckForSocketError(fpConnect(ASocket, Psockaddr(@LAddr), SizeOf(sockaddr)));
- end;
- Id_IPv6: begin
- InitSockAddr_in6(LAddr);
- with LAddr do
- begin
- sin6_addr := StrToNetAddr6(AIP);
- //TranslateStringToTInAddr(AIP, LAddr6.sin6_addr, Id_IPv6);
- sin6_port := htons(APort);
- end;
- CheckForSocketError(fpConnect(ASocket, Psockaddr(@LAddr), SizeOf(sockaddr_in6)));
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- {$IFDEF DARWIN} // TODO: use HAS_SOCKET_NOSIGPIPE instead...
- SetSocketOption(ASocket, Id_SOL_SOCKET, SO_NOSIGPIPE, 1);
- {$ENDIF}
- end;
- function TIdStackUnix.HostByName(const AHostName: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string;
- var
- LI4 : array of THostAddr;
- LI6 : array of THostAddr6;
- LH4 : THostEntry;
- LRetVal : Integer;
- begin
- case AIPVersion of
- Id_IPv4 :
- begin
- if GetHostByName(AHostName, LH4) then
- begin
- Result := HostAddrToStr(LH4.Addr);
- Exit;
- end;
- SetLength(LI4, 10);
- LRetVal := ResolveName(AHostName, LI4);
- if LRetVal < 1 then begin
- raise EIdResolveError.CreateFmt(RSResolveError, [AHostName, 'Error', LRetVal]); {do not localize}
- end;
- Result := NetAddrToStr(LI4[0]);
- end;
- Id_IPv6 :
- begin
- SetLength(LI6, 10);
- LRetVal := ResolveName6(AHostName, LI6);
- if LRetVal < 1 then begin
- raise EIdResolveError.CreateFmt(RSResolveError, [AHostName, LRetVal]);
- end;
- Result := NetAddrToStr6(LI6[0]);
- end;
- end;
- end;
- function TIdStackUnix.ReadHostName: string;
- begin
- Result := GetHostName;
- end;
- procedure TIdStackUnix.Listen(ASocket: TIdStackSocketHandle; ABackLog: Integer);
- begin
- CheckForSocketError(fpListen(ASocket, ABacklog));
- end;
- function TIdStackUnix.WSRecv(ASocket: TIdStackSocketHandle; var ABuffer;
- const ABufferLength, AFlags: Integer): Integer;
- begin
- //IdStackWindows is just: Result := Recv(ASocket, ABuffer, ABufferLength, AFlags);
- Result := fpRecv(ASocket, @ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL);
- end;
- function TIdStackUnix.RecvFrom(const ASocket: TIdStackSocketHandle; var VBuffer;
- const ALength, AFlags: Integer; var VIP: string; var VPort: TIdPort;
- var VIPVersion: TIdIPVersion): Integer;
- var
- LiSize: tsocklen;
- LAddr: sockaddr_in6;
- begin
- LiSize := SizeOf(sockaddr_in6);
- Result := fpRecvFrom(ASocket, @VBuffer, ALength, AFlags or Id_MSG_NOSIGNAL, Psockaddr(@LAddr), @LiSize);
- if Result >= 0 then
- begin
- case LAddr.sin6_family of
- Id_PF_INET4 :
- begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := NetAddrToStr(sin_addr);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPV4;
- end;
- Id_PF_INET6:
- begin
- with LAddr do
- begin
- VIP := NetAddrToStr6(sin6_addr);
- VPort := ntohs(sin6_port);
- end;
- VIPVersion := Id_IPV6;
- end;
- end;
- end;
- end;
- function TIdStackUnix.ReceiveMsg(ASocket: TIdStackSocketHandle;
- var VBuffer: TIdBytes; APkt: TIdPacketInfo): UInt32;
- var
- LIP : String;
- LPort : TIdPort;
- LIPVersion : TIdIPVersion;
- begin
- Result := RecvFrom(ASocket, VBuffer, Length(VBuffer), 0, LIP, LPort, LIPVersion);
- APkt.Reset;
- APkt.SourceIP := LIP;
- APkt.SourcePort := LPort;
- APkt.SourceIPVersion := LIPVersion;
- APkt.DestIPVersion := LIPVersion;
- SetLength(VBuffer, Result);
- end;
- {The stuff below is commented out until I figure out what to do}
- {var
- LIP : String;
- LPort : TIdPort;
- LSize: UInt32;
- LAddr: SockAddr_In6;
- LMsg : msghdr;
- LMsgBuf : BUF;
- LControl : TIdBytes;
- LCurCmsg : CMSGHDR; //for iterating through the control buffer
- LCurPt : Pin_pktinfo;
- LCurPt6 : Pin6_pktinfo;
- LByte : PByte;
- LDummy, LDummy2 : UInt32;
- begin
- //we call the macro twice because we specified two possible structures.
- //Id_IPV6_HOPLIMIT and Id_IPV6_PKTINFO
- LSize := CMSG_SPACE(SizeOf(Byte)) + CMSG_SPACE(SizeOf(in6_pktinfo));
- SetLength(LControl, LSize);
- LMsgBuf.len := Length(VBuffer); // Length(VMsgData);
- LMsgBuf.buf := @VBuffer[0]; // @VMsgData[0];
- FillChar(LMsg,SizeOf(LMsg),0);
- LMsg.lpBuffers := @LMsgBuf;
- LMsg.dwBufferCount := 1;
- LMsg.Control.Len := LSize;
- LMsg.Control.buf := @LControl[0];
- LMsg.name := PSOCKADDR(@LAddr);
- LMsg.namelen := SizeOf(LAddr);
- CheckForSocketError( RecvMsg(ASocket,@LMsg,Result,@LDummy,LPwsaoverlapped_COMPLETION_ROUTINE(@LDummy2)));
- case LAddr.sin6_family of
- Id_PF_INET4: begin
- with PSOCKADDR(@LAddr)^do
- begin
- APkt.SourceIP := TranslateTInAddrToString(sin_addr,Id_IPv4);
- APkt.SourcePort := NToHs(sin_port);
- end;
- APkt.SourceIPVersion := Id_IPv4;
- end;
- Id_PF_INET6: begin
- with LAddr do
- begin
- APkt.SourceIP := TranslateTInAddrToString(sin6_addr, Id_IPv6);
- APkt.SourcePort := NToHs(sin6_port);
- end;
- APkt.SourceIPVersion := Id_IPv6;
- end;
- else begin
- Result := 0; // avoid warning
- IPVersionUnsupported;
- end;
- end;
- LCurCmsg := nil;
- repeat
- LCurCmsg := CMSG_NXTHDR(@LMsg,LCurCmsg);
- if LCurCmsg = nil then
- begin
- break;
- end;
- case LCurCmsg^.cmsg_type of
- IP_PKTINFO : //done this way because IPV6_PKTINF and IP_PKTINFO
- //are both 19
- begin
- case LAddr.sin6_family of
- Id_PF_INET4:
- begin
- LCurPt := WSA_CMSG_DATA(LCurCmsg);
- APkt.DestIP := GWindowsStack.TranslateTInAddrToString(LCurPt^.ipi_addr,Id_IPv4);
- APkt.DestIF := LCurPt^.ipi_ifindex;
- APkt.DestIPVersion := Id_IPv4;
- end;
- Id_PF_INET6:
- begin
- LCurPt6 := WSA_CMSG_DATA(LCurCmsg);
- APkt.DestIP := GWindowsStack.TranslateTInAddrToString(LCurPt6^.ipi6_addr,Id_IPv6);
- APkt.DestIF := LCurPt6^.ipi6_ifindex;
- APkt.DestIPVersion := Id_IPv6;
- end;
- end;
- end;
- Id_IPV6_HOPLIMIT :
- begin
- LByte := PByte(WSA_CMSG_DATA(LCurCmsg));
- APkt.TTL := LByte^;
- end;
- end;
- until False; }
- function TIdStackUnix.WSSend(ASocket: TIdStackSocketHandle;
- const ABuffer; const ABufferLength, AFlags: Integer): Integer;
- begin
- //CC: Should Id_MSG_NOSIGNAL be included?
- // Result := Send(ASocket, ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL);
- Result := CheckForSocketError(fpsend(ASocket, @ABuffer, ABufferLength, AFlags));
- end;
- procedure TIdStackUnix.WSSendTo(ASocket: TIdStackSocketHandle;
- const ABuffer; const ABufferLength, AFlags: Integer; const AIP: string;
- const APort: TIdPort; AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
- var
- LAddr : sockaddr_in6;
- LBytesOut: Integer;
- begin
- case AIPVersion of
- Id_IPv4 :
- begin
- InitSockAddr(Psockaddr(@LAddr)^);
- with Psockaddr(@LAddr)^ do
- begin
- sin_addr := StrToNetAddr(AIP);
- sin_port := htons(APort);
- end;
- LBytesOut := fpSendTo(ASocket, @ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL, Psockaddr(@LAddr), SizeOf(sockaddr));
- end;
- Id_IPv6:
- begin
- InitSockAddr_in6(LAddr);
- with LAddr do
- begin
- sin6_addr := StrToHostAddr6(AIP);
- //TranslateStringToTInAddr(AIP, sin6_addr, AIPVersion);
- sin6_port := htons(APort);
- end;
- LBytesOut := fpSendTo(ASocket, @ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL, Psockaddr(@LAddr), SizeOf(sockaddr_in6));
- end;
- else begin
- LBytesOut := 0; // avoid warning
- IPVersionUnsupported;
- end;
- end;
- if LBytesOut = -1 then begin
- // TODO: move this into RaiseLastSocketError() directly
- if WSGetLastError() = Id_WSAEMSGSIZE then begin
- raise EIdPackageSizeTooBig.Create(RSPackageSizeTooBig);
- end else begin
- RaiseLastSocketError;
- end;
- end else if LBytesOut <> ABufferLength then begin
- raise EIdNotAllBytesSent.Create(RSNotAllBytesSent);
- end;
- end;
- procedure TIdStackUnix.{$IFDEF VCL_XE3_OR_ABOVE}GetSocketOption{$ELSE}WSGetSocketOption{$ENDIF}
- (ASocket: TIdStackSocketHandle; ALevel: TIdSocketProtocol; AOptName: TIdSocketOption;
- var AOptVal; var AOptLen: Integer);
- var
- LLen : TSockLen;
- begin
- LLen := AOptLen;
- CheckForSocketError(fpGetSockOpt(ASocket, ALevel, AOptName, PAnsiChar(@AOptVal), @LLen));
- AOptLen := LLen;
- end;
- procedure TIdStackUnix.{$IFDEF VCL_XE3_OR_ABOVE}SetSocketOption{$ELSE}WSSetSocketOption{$ENDIF}
- (ASocket: TIdStackSocketHandle; ALevel: TIdSocketProtocol; AOptName: TIdSocketOption;
- const AOptVal; const AOptLen: Integer);
- begin
- CheckForSocketError(fpSetSockOpt(ASocket, ALevel, AOptName, PAnsiChar(@AOptVal), AOptLen));
- end;
- function TIdStackUnix.WSGetLastError: Integer;
- begin
- //IdStackWindows just uses result := WSAGetLastError;
- Result := GetLastError; //System.GetLastOSError; - FPC doesn't define it in System
- if Result = ESysEPIPE then begin
- Result := Id_WSAECONNRESET;
- end;
- end;
- procedure TIdStackUnix.WSSetLastError(const AErr : Integer);
- begin
- SetLastError(AErr);
- end;
- function TIdStackUnix.WSSocket(AFamily, AStruct, AProtocol: Integer;
- const ANonBlocking: Boolean = False): TIdStackSocketHandle;
- var
- LValue: UInt32;
- begin
- Result := fpsocket(AFamily, AStruct, AProtocol);
- if Result <> Id_INVALID_SOCKET then begin
- //SetBlocking(Result, not ANonBlocking);
- LValue := UInt32(ANonBlocking);
- fpioctl(Result, FIONBIO, @LValue);
- end;
- end;
- function TIdStackUnix.WSGetServByName(const AServiceName: string): TIdPort;
- var
- LS : TServiceEntry;
- begin
- if GetServiceByName(AServiceName, '', LS) then begin
- Result := LS.Port;
- end else begin
- raise EIdInvalidServiceName.CreateFmt(RSInvalidServiceName, [AServiceName]);
- end;
- end;
- function TIdStackUnix.HostToNetwork(AValue: UInt16): UInt16;
- begin
- Result := htons(AValue);
- end;
- function TIdStackUnix.NetworkToHost(AValue: UInt16): UInt16;
- begin
- Result := ntohs(AValue);
- end;
- function TIdStackUnix.HostToNetwork(AValue: UInt32): UInt32;
- begin
- {$i IdRangeCheckingOff.inc} // disable range checking
- Result := htonl(AValue);
- {$i IdRangeCheckingOn.inc} // Restore range checking
- end;
- function TIdStackUnix.NetworkToHost(AValue: UInt32): UInt32;
- begin
- {$i IdRangeCheckingOff.inc} // disable range checking
- Result := ntohl(AValue);
- // Restore range checking
- {$i IdRangeCheckingOn.inc} // Restore range checking
- end;
- { RP - I'm not sure what endian Linux natively uses, thus the
- check to see if the bytes need swapping or not ... }
- function TIdStackUnix.HostToNetwork(AValue: TIdUInt64): TIdUInt64;
- var
- LParts: TIdUInt64Parts;
- L: UInt32;
- begin
- {$i IdRangeCheckingOff.inc} // disable range checking
- if (htonl(1) <> 1) then begin
- LParts.QuadPart := AValue{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF};
- L := htonl(LParts.HighPart);
- LParts.HighPart := htonl(LParts.LowPart);
- LParts.LowPart := L;
- Result{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF} := LParts.QuadPart;
- end else begin
- Result{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF} := AValue{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF};
- end;
- {$i IdRangeCheckingOn.inc} // Restore range checking
- end;
- function TIdStackUnix.NetworkToHost(AValue: TIdUInt64): TIdUInt64;
- var
- LParts: TIdUInt64Parts;
- L: UInt32;
- begin
- {$i IdRangeCheckingOff.inc} // disable range checking
- if (ntohl(1) <> 1) then begin
- LParts.QuadPart := AValue{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF};
- L := ntohl(LParts.HighPart);
- LParts.HighPart := NetworkToHost(LParts.LowPart);
- LParts.LowPart := L;
- Result{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF} := LParts.QuadPart;
- end else begin
- Result{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF} := AValue{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF};
- end;
- {$i IdRangeCheckingOn.inc} // Restore range checking
- end;
- {$IFDEF HAS_getifaddrs}
- // TODO: does FreePascal already define these anywhere?
- type
- pifaddrs = ^ifaddrs;
- ifaddrs = record
- ifa_next: pifaddrs; { Pointer to next struct }
- ifa_name: PIdAnsiChar; { Interface name }
- // Solaris ifaddrs struct implements 64bit ifa_flags. (Details: https://docs.oracle.com/cd/E88353_01/html/E37843/getifaddrs-3c.html)
- {$IFDEF SOLARIS}
- ifa_flags: UInt64; { Interface flags }
- {$ELSE}
- ifa_flags: Cardinal; { Interface flags }
- {$ENDIF}
- ifa_addr: psockaddr; { Interface address }
- ifa_netmask: psockaddr; { Interface netmask }
- ifa_broadaddr: psockaddr; { Interface broadcast address }
- ifa_dstaddr: psockaddr; { P2P interface destination }
- ifa_data: Pointer; { Address specific data }
- end;
- const
- IFF_LOOPBACK = $8;
-
- function getifaddrs(var ifap: pifaddrs): Integer; cdecl; external 'libc.so' name 'getifaddrs'; {do not localize}
- procedure freeifaddrs(ifap: pifaddrs); cdecl; external 'libc.so' name 'freeifaddrs'; {do not localize}
- {$IFDEF HAS_if_nametoindex}
- procedure if_nametoindex(const ifname: PIdAnsiChar): UInt32; cdecl; external 'libc.so' name 'if_nametoindex'; {do not localize}
- {$ENDIF}
- type
- TIdStackLocalAddressAccess = class(TIdStackLocalAddress)
- end;
- {$ENDIF}
- procedure TIdStackUnix.GetLocalAddressList(AAddresses: TIdStackLocalAddressList);
- var
- {$IFDEF HAS_getifaddrs}
- LAddrList, LAddrInfo: pifaddrs;
- LSubNetStr: String;
- LAddress: TIdStackLocalAddress;
- LName: string;
- {$ELSE}
- LI4 : array of THostAddr;
- LI6 : array of THostAddr6;
- i : Integer;
- LHostName : String;
- {$ENDIF}
- begin
- // TODO: Using gethostname() and ResolveName() like this may not always
- // return just the machine's IP addresses. Technically speaking, they will
- // return the local hostname, and then return the address(es) to which that
- // hostname resolves. It is possible for a machine to (a) be configured such
- // that its name does not resolve to an IP, or (b) be configured such that
- // its name resolves to multiple IPs, only one of which belongs to the local
- // machine. For better results, we should use getifaddrs() on platforms that
- // support it...
- {$IFDEF HAS_getifaddrs}
- if getifaddrs(LAddrList) = 0 then // TODO: raise an exception if it fails
- try
- AAddresses.BeginUpdate;
- try
- LAddrInfo := LAddrList;
- repeat
- if (LAddrInfo^.ifa_addr <> nil) and ((LAddrInfo^.ifa_flags and IFF_LOOPBACK) = 0) then
- begin
- LAddress := nil;
- case LAddrInfo^.ifa_addr^.sa_family of
- Id_PF_INET4: begin
- if LAddrInfo^.ifa_netmask <> nil then begin
- LSubNetStr := TranslateTInAddrToString( PSockAddr_In(LAddrInfo^.ifa_netmask)^.sin_addr, Id_IPv4);
- end else begin
- LSubNetStr := '';
- end;
- LAddress := TIdStackLocalAddressIPv4.Create(AAddresses, TranslateTInAddrToString( PSockAddr_In(LAddrInfo^.ifa_addr)^.sin_addr, Id_IPv4), LSubNetStr);
- end;
- Id_PF_INET6: begin
- LAddress := TIdStackLocalAddressIPv6.Create(AAddresses, TranslateTInAddrToString( PSockAddr_In6(LAddrInfo^.ifa_addr)^.sin6_addr, Id_IPv6));
- end;
- end;
- if LAddress <> nil then begin
- LName := String(LAddrInfo^.ifa_name);
- {$I IdObjectChecksOff.inc}
- TIdStackLocalAddressAccess(LAddress).FDescription := LName;
- TIdStackLocalAddressAccess(LAddress).FFriendlyName := LName;
- TIdStackLocalAddressAccess(LAddress).FInterfaceName := LName;
- {$IFDEF HAS_if_nametoindex}
- TIdStackLocalAddressAccess(LAddress).FInterfaceIndex := if_nametoindex(LAddrInfo^.ifa_name);
- {$ENDIF}
- {$I IdObjectChecksOn.inc}
- end;
- end;
- LAddrInfo := LAddrInfo^.ifa_next;
- until LAddrInfo = nil;
- finally
- AAddresses.EndUpdate;
- end;
- finally
- freeifaddrs(LAddrList);
- end;
- {$ELSE}
- LHostName := GetHostName;
- if LHostName = '' then begin
- RaiseLastSocketError;
- end;
- AAddresses.BeginUpdate;
- try
- if ResolveName(LHostName, LI4) = 0 then
- begin
- for i := Low(LI4) to High(LI4) do
- begin
- TIdStackLocalAddressIPv4.Create(AAddresses, NetAddrToStr(LI4[i]), ''); // TODO: SubNet
- end;
- end;
- if ResolveName6(LHostName, LI6) = 0 then
- begin
- for i := Low(LI6) to High(LI6) do
- begin
- TIdStackLocalAddressIPv6.Create(AAddresses, NetAddrToStr6(LI6[i]));
- end;
- end;
- finally
- AAddresses.EndUpdate;
- end;
- {$ENDIF}
- end;
- function TIdStackUnix.HostByAddress(const AAddress: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string;
- var
- LI : Array of string;
- LAddr4: THostAddr;
- LAddr6: THostAddr6;
- begin
- Result := '';
- case AIPVersion of
- Id_IPv4 :
- begin
- LAddr4 := StrToNetAddr(AAddress);
- if ResolveAddress(LAddr4, LI) = 0 then begin
- Result := LI[0];
- end;
- end;
- Id_IPv6 :
- begin
- LAddr6 := StrToNetAddr6(AAddress);
- if ResolveAddress6(LAddr6, LI) = 0 then begin
- Result := LI[0];
- end;
- end;
- end;
- end;
- function TIdStackUnix.WSShutdown(ASocket: TIdStackSocketHandle; AHow: Integer): Integer;
- begin
- Result := fpShutdown(ASocket, AHow);
- end;
- procedure TIdStackUnix.Disconnect(ASocket: TIdStackSocketHandle);
- begin
- // Windows uses Id_SD_Send, Linux should use Id_SD_Both
- WSShutdown(ASocket, Id_SD_Both);
- // SO_LINGER is false - socket may take a little while to actually close after this
- WSCloseSocket(ASocket);
- end;
- procedure TIdStackUnix.GetPeerName(ASocket: TIdStackSocketHandle; var VIP: string;
- var VPort: TIdPort; var VIPVersion: TIdIPVersion);
- var
- i: tsocklen;
- LAddr: sockaddr_in6;
- begin
- i := SizeOf(LAddr);
- CheckForSocketError(fpGetPeerName(ASocket, @LAddr, @i));
- case LAddr.sin6_family of
- PF_INET: begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := NetAddrToStr(sin_addr);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPv4;
- end;
- PF_INET6: begin
- with LAddr do
- begin
- VIP := NetAddrToStr6(sin6_addr);
- VPort := ntohs(sin6_port);
- end;
- VIPVersion := Id_IPv6;
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- procedure TIdStackUnix.GetSocketName(ASocket: TIdStackSocketHandle;
- var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion);
- var
- i: tsocklen;
- LAddr: sockaddr_in6;
- begin
- i := SizeOf(LAddr);
- CheckForSocketError(fpGetSockName(ASocket, @LAddr, @i));
- case LAddr.sin6_family of
- PF_INET : begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := NetAddrToStr(sin_addr);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPV4;
- end;
- PF_INET6: begin
- with LAddr do
- begin
- VIP := NetAddrToStr6(sin6_addr);
- VPort := ntohs(sin6_port);
- end;
- VIPVersion := Id_IPv6;
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- procedure TIdStackUnix.AddServByPortToList(const APortNumber: TIdPort; AAddresses: TStrings);
- var
- LS : TServiceEntry;
- begin
- if GetServiceByPort(APortNumber, '', LS) then begin
- AAddresses.Add(LS.Name);
- end;
- end;
- function TIdStackUnix.WSTranslateSocketErrorMsg(const AErr: Integer): string;
- begin
- //we override this function for the herr constants that
- //are returned by the DNS functions
- //note that this is not really applicable because we are using some
- //FPC functions that do direct DNS lookups without the standard Unix
- //DNS functions. It sounds odd but I think there's a good reason for it.
- Result := inherited WSTranslateSocketErrorMsg(AErr);
- end;
- procedure TIdStackUnix.SetBlocking(ASocket: TIdStackSocketHandle;
- const ABlocking: Boolean);
- var
- LValue: UInt32;
- begin
- LValue := UInt32(not ABlocking);
- CheckForSocketError(fpioctl(ASocket, FIONBIO, @LValue));
- end;
- function TIdStackUnix.WouldBlock(const AResult: Integer): Boolean;
- begin
- // using if-else instead of in..range because EAGAIN and EWOULDBLOCK
- // have often the same value and so FPC might report a range error
- Result := (AResult = Id_WSAEAGAIN) or
- (AResult = Id_WSAEWOULDBLOCK) or
- (AResult = Id_WSAEINPROGRESS);
- end;
- function TIdStackUnix.SupportsIPv4: Boolean;
- //In Windows, this does something else. It checks the LSP's installed.
- begin
- Result := CheckIPVersionSupport(Id_IPv4);
- end;
- function TIdStackUnix.SupportsIPv6: Boolean;
- //In Windows, this does something else. It checks the LSP's installed.
- begin
- Result := CheckIPVersionSupport(Id_IPv6);
- end;
- function TIdStackUnix.CheckIPVersionSupport(const AIPVersion: TIdIPVersion): Boolean;
- var
- LTmpSocket: TIdStackSocketHandle;
- begin
- // TODO: on nix systems (or maybe just Linux?), an alternative would be to
- // check for the existance of the '/proc/net/if_inet6' kernel pseudo-file
- LTmpSocket := WSSocket(IdIPFamily[AIPVersion], Integer(Id_SOCK_STREAM), Id_IPPROTO_IP);
- Result := LTmpSocket <> Id_INVALID_SOCKET;
- if Result then begin
- WSCloseSocket(LTmpSocket);
- end;
- end;
- procedure TIdStackUnix.WriteChecksum(s: TIdStackSocketHandle;
- var VBuffer: TIdBytes; const AOffset: Integer; const AIP: String;
- const APort: TIdPort; const AIPVersion: TIdIPVersion);
- begin
- case AIPVersion of
- Id_IPv4 : CopyTIdUInt16(HostToLittleEndian(CalcCheckSum(VBuffer)), VBuffer, AOffset);
- Id_IPv6 : WriteChecksumIPv6(s, VBuffer, AOffset, AIP, APort);
- else
- IPVersionUnsupported;
- end;
- end;
- procedure TIdStackUnix.WriteChecksumIPv6(s: TIdStackSocketHandle;
- var VBuffer: TIdBytes; const AOffset: Integer; const AIP: String;
- const APort: TIdPort);
- begin
- //we simply request that the kernal write the checksum when the data
- //is sent. All of the parameters required are because Windows is bonked
- //because it doesn't have the IPV6CHECKSUM socket option meaning we have
- //to querry the network interface in TIdStackWindows -- yuck!!
- SetSocketOption(s, Id_IPPROTO_IPV6, IPV6_CHECKSUM, AOffset);
- end;
- function TIdStackUnix.IOControl(const s: TIdStackSocketHandle; const cmd: UInt32;
- var arg: UInt32): Integer;
- begin
- Result := fpioctl(s, cmd, @arg);
- end;
- (*
- Why did I remove this again?
- 1) it sends SIGPIPE even if the socket is created with the no-sigpipe bit set
- that could be solved by blocking sigpipe within this thread
- This is probably a bug in the Linux kernel, but we could work around it
- by blocking that signal for the time of sending the file (just get the
- sigprocmask, see if pipe bit is set, if not set it and remove again after
- sending the file)
- But the more serious reason is another one, which exists in Windows too:
- 2) I think that ServeFile is misdesigned:
- ServeFile does not raise an exception if it didn't send all the bytes.
- Now what happens if I have OnExecute assigned like this
- AThread.Connection.ServeFile('...', True); // <-- true to send via kernel
- is that it will return 0, but notice that in this case I didn't ask for the
- result. Net effect is that the thread will loop in OnExecute even if the
- socket is long gone. This doesn't fit Indy semantics at all, exceptions are
- always raised if the remote end disconnects. Even if I would do
- AThread.Connection.ServeFile('...', False);
- then it would raise an exception.
- I think this is a big flaw in the design of the ServeFile function.
- Maybe GServeFile should only return the bytes sent, but then
- TCPConnection.ServeFile() should raise an exception if GServeFile didn't
- send all the bytes.
- JM Berg, 2002-09-09
- function ServeFile(ASocket: TIdStackSocketHandle; AFileName: string): UInt32;
- var
- LFileHandle: integer;
- offset: integer;
- stat: _stat;
- begin
- LFileHandle := open(PChar(AFileName), O_RDONLY);
- try
- offset := 0;
- fstat(LFileHandle, stat);
- Result := sendfile(ASocket, LFileHandle, offset, stat.st_size);
- //** if Result = UInt32(-1) then RaiseLastOSError;
- finally libc.__close(LFileHandle); end;
- end;
- *)
- procedure TIdStackUnix.SetKeepAliveValues(ASocket: TIdStackSocketHandle;
- const AEnabled: Boolean; const ATimeMS, AInterval: Integer);
- begin
- inherited; // turn SO_KEEPALIVE on/off first...
- // TODO: remove below, as it should be handled by TIdStack.SetKeepAliveValues() now...
- if AEnabled then begin
- {$IFDEF HAS_TCP_KEEPIDLE}
- SetSocketOption(ASocket, Id_SOL_TCP, Id_TCP_KEEPIDLE, ATimeMS div MSecsPerSec);
- {$ENDIF}
- {$IFDEF HAS_TCP_KEEPINTVL}
- SetSocketOption(ASocket, Id_SOL_TCP, Id_TCP_KEEPINTVL, AInterval div MSecsPerSec);
- {$ENDIF}
- end;
- end;
- { TIdSocketListUnix }
- procedure TIdSocketListUnix.Add(AHandle: TIdStackSocketHandle);
- begin
- Lock;
- try
- if fpFD_ISSET(AHandle, FFDSet) = 0 then begin
- if AHandle >= FD_SETSIZE then begin
- raise EIdStackSetSizeExceeded.Create(RSSetSizeExceeded);
- end;
- fpFD_SET(AHandle, FFDSet);
- Inc(FCount);
- end;
- finally
- Unlock;
- end;
- end;//
- procedure TIdSocketListUnix.Clear;
- begin
- Lock;
- try
- fpFD_ZERO(FFDSet);
- FCount := 0;
- finally
- Unlock;
- end;
- end;
- function TIdSocketListUnix.ContainsSocket(AHandle: TIdStackSocketHandle): Boolean;
- begin
- Lock;
- try
- Result := fpFD_ISSET(AHandle, FFDSet) > 0;
- finally
- Unlock;
- end;
- end;
- function TIdSocketListUnix.Count: Integer;
- begin
- Lock;
- try
- Result := FCount;
- finally
- Unlock;
- end;
- end;//
- class function TIdSocketListUnix.FDSelect(AReadSet, AWriteSet, AExceptSet: PFDSet;
- const ATimeout: Integer): Integer;
- var
- LTime: TTimeVal;
- LTimePtr: PTimeVal;
- begin
- if ATimeout = IdTimeoutInfinite then begin
- LTimePtr := nil;
- end else begin
- LTime.tv_sec := ATimeout div 1000;
- LTime.tv_usec := (ATimeout mod 1000) * 1000;
- LTimePtr := @LTime;
- end;
- // TODO: calculate the actual nfds value based on the Sets provided...
- // TODO: use poll() instead of select() to remove limit on how many sockets can be queried
- Result := fpSelect(FD_SETSIZE, AReadSet, AWriteSet, AExceptSet, LTimePtr);
- end;
- procedure TIdSocketListUnix.GetFDSet(var VSet: TFDSet);
- begin
- Lock;
- try
- VSet := FFDSet;
- finally
- Unlock;
- end;
- end;
- function TIdSocketListUnix.GetItem(AIndex: Integer): TIdStackSocketHandle;
- var
- LIndex, i: Integer;
- begin
- Result := 0;
- // TODO: is this missing Lock/Unlock calls?
- LIndex := 0;
- //? use FMaxHandle div x
- for i:= 0 to __FD_SETSIZE - 1 do begin
- if fpFD_ISSET(i, FFDSet) = 1 then begin
- if LIndex = AIndex then begin
- Result := i;
- Break;
- end;
- Inc(LIndex);
- end;
- end;
- end;
- procedure TIdSocketListUnix.Remove(AHandle: TIdStackSocketHandle);
- begin
- Lock;
- try
- if fpFD_ISSET(AHandle, FFDSet) = 1 then
- begin
- Dec(FCount);
- fpFD_CLR(AHandle, FFDSet);
- end;
- finally
- Unlock;
- end;
- end;//
- procedure TIdSocketListUnix.SetFDSet(var VSet: TFDSet);
- begin
- Lock;
- try
- FFDSet := VSet;
- finally
- Unlock;
- end;
- end;
- class function TIdSocketListUnix.Select(AReadList: TIdSocketList; AWriteList: TIdSocketList;
- AExceptList: TIdSocketList; const ATimeout: Integer = IdTimeoutInfinite): Boolean;
- var
- LReadSet: TFDSet;
- LWriteSet: TFDSet;
- LExceptSet: TFDSet;
- LPReadSet: PFDSet;
- LPWriteSet: PFDSet;
- LPExceptSet: PFDSet;
- procedure ReadSet(AList: TIdSocketList; var ASet: TFDSet; var APSet: PFDSet);
- begin
- if AList <> nil then begin
- TIdSocketListUnix(AList).GetFDSet(ASet);
- APSet := @ASet;
- end else begin
- APSet := nil;
- end;
- end;
- begin
- ReadSet(AReadList, LReadSet, LPReadSet);
- ReadSet(AWriteList, LWriteSet, LPWriteSet);
- ReadSet(AExceptList, LExceptSet, LPExceptSet);
- //
- Result := FDSelect(LPReadSet, LPWriteSet, LPExceptSet, ATimeout) <> 0;
- //
- if AReadList <> nil then begin
- TIdSocketListUnix(AReadList).SetFDSet(LReadSet);
- end;
- if AWriteList <> nil then begin
- TIdSocketListUnix(AWriteList).SetFDSet(LWriteSet);
- end;
- if AExceptList <> nil then begin
- TIdSocketListUnix(AExceptList).SetFDSet(LExceptSet);
- end;
- end;
- function TIdSocketListUnix.SelectRead(const ATimeout: Integer): Boolean;
- var
- LSet: TFDSet;
- begin
- Lock;
- try
- LSet := FFDSet;
- // select() updates this structure on return,
- // so we need to copy it each time we need it
- finally
- Unlock;
- end;
- Result := FDSelect(@LSet, nil, nil, ATimeout) > 0;
- end;
- function TIdSocketListUnix.SelectReadList(var VSocketList: TIdSocketList;
- const ATimeout: Integer = IdTimeoutInfinite): Boolean;
- var
- LSet: TFDSet;
- begin
- Lock;
- try
- LSet := FFDSet;
- // select() updates this structure on return,
- // so we need to copy it each time we need it
- finally
- Unlock;
- end;
- Result := FDSelect(@LSet, nil, nil, ATimeout) > 0;
- if Result then begin
- if VSocketList = nil then begin
- VSocketList := TIdSocketList.CreateSocketList;
- end;
- TIdSocketListUnix(VSocketList).SetFDSet(LSet);
- end;
- end;
- function TIdSocketListUnix.Clone: TIdSocketList;
- begin
- Result := TIdSocketListUnix.Create;
- try
- Lock;
- try
- TIdSocketListUnix(Result).SetFDSet(FFDSet);
- finally
- Unlock;
- end;
- except
- FreeAndNil(Result);
- raise;
- end;
- end;
- initialization
- GSocketListClass := TIdSocketListUnix;
- end.
|