| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380 |
- {
- $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}
- 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 }
- type
- // TODO: rewrite this to use poll() instead of select(), similar to TIdSocketListVCLPosix
- 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;
- 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.
|