| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484 |
- {
- $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 IdStackLibc;
- interface
- {$i IdCompilerDefines.inc}
- uses
- Classes,
- Libc,
- IdStack,
- IdStackConsts,
- IdGlobal,
- IdStackBSDBase;
- {$UNDEF LIBCPASS_STRUCT}
- {$IFDEF KYLIX}
- {$DEFINE LIBCPASS_STRUCT}
- {$ENDIF}
- {$IFDEF DELPHI_CROSS}
- {$DEFINE LIBCPASS_STRUCT}
- {$ENDIF}
- type
- // TODO: move this class into the implementation section! It is not used outside of this unit
- TIdSocketListLibc = 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;//TIdSocketList
- TIdStackLibc = class(TIdStackBSDBase)
- private
- procedure WriteChecksumIPv6(s: TIdStackSocketHandle;
- var VBuffer: TIdBytes; const AOffset: Integer; const AIP: String;
- const APort: TIdPort);
- protected
- 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
- 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;
- function WSGetServByPort(const APortNumber: TIdPort): 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;
- AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): 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 : Integer; AStruct : TIdSocketType; 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;
- constructor Create; override;
- destructor Destroy; 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;
- TLinger = record
- l_onoff: UInt16;
- l_linger: UInt16;
- end;
- TIdLinger = TLinger;
- implementation
- uses
- IdResourceStrings,
- IdException,
- SysUtils;
- type
- psockaddr_in6 = ^sockaddr_in6;
- const
- Id_MSG_NOSIGNAL = MSG_NOSIGNAL;
- Id_WSAEPIPE = EPIPE;
- constructor TIdStackLibc.Create;
- begin
- inherited Create;
- end;
- destructor TIdStackLibc.Destroy;
- begin
- inherited Destroy;
- end;
- function TIdStackLibc.GetLastError : Integer;
- begin
- Result := errno;
- end;
- procedure TIdStackLibc.SetLastError(Const AError : Integer);
- begin
- __errno_location^ := AError;
- end;
- procedure TIdStackLibc.WSSetLastError(const AErr : Integer);
- begin
- SetLastError(AErr);
- end;
- function TIdStackLibc.Accept(ASocket: TIdStackSocketHandle;
- var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion): TIdStackSocketHandle;
- var
- LN: UInt32;
- LAddr: sockaddr_in6;
- begin
- LN := SizeOf(LAddr);
- Result := Libc.accept(ASocket, PSockAddr(@LAddr), @LN);
- if Result <> SOCKET_ERROR then begin
- case LAddr.sin6_family of
- Id_PF_INET4: begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := TranslateTInAddrToString(sin_addr, Id_IPv4);
- VPort := Ntohs(sin_port);
- end;
- VIPVersion := Id_IPV4;
- end;
- Id_PF_INET6: begin
- with LAddr do
- begin
- VIP := TranslateTInAddrToString(sin6_addr, Id_IPv6);
- VPort := ntohs(sin6_port);
- end;
- VIPVersion := Id_IPV6;
- end;
- else begin
- Libc.__close(Result);
- Result := Id_INVALID_SOCKET;
- IPVersionUnsupported;
- end;
- end;
- end else begin
- if GetLastError = EBADF then begin
- SetLastError(EINTR);
- end;
- end;
- end;
- procedure TIdStackLibc.Bind(ASocket: TIdStackSocketHandle;
- const AIP: string; const APort: TIdPort;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
- var
- LAddr: sockaddr_in6;
- begin
- FillChar(LAddr, SizeOf(LAddr), 0);
- case AIPVersion of
- Id_IPv4: begin
- with Psockaddr(@LAddr)^ do
- begin
- sin_family := Id_PF_INET4;
- if AIP <> '' then begin
- TranslateStringToTInAddr(AIP, sin_addr, Id_IPv4);
- end;
- sin_port := htons(APort);
- end;
- CheckForSocketError(Libc.bind(ASocket, {$IFDEF LIBCPASS_STRUCT}PSockAddr(@LAddr)^ {$ELSE} Psockaddr(@LAddr) {$ENDIF},SizeOf(sockaddr)));
- end;
- Id_IPv6: begin
- with LAddr do
- begin
- sin6_family := Id_PF_INET6;
- if AIP <> '' then begin
- TranslateStringToTInAddr(AIP, sin6_addr, Id_IPv6);
- end;
- sin6_port := htons(APort);
- end;
- CheckForSocketError(Libc.bind(ASocket, {$IFDEF LIBCPASS_STRUCT}Psockaddr(@LAddr)^ {$ELSE}Psockaddr(@LAddr){$ENDIF}, SizeOf(sockaddr_in6)));
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- function TIdStackLibc.WSCloseSocket(ASocket: TIdStackSocketHandle): Integer;
- begin
- Result := Libc.__close(ASocket);
- end;
- procedure TIdStackLibc.Connect(const ASocket: TIdStackSocketHandle;
- const AIP: string; const APort: TIdPort;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
- var
- LAddr: sockAddr_in6;
- begin
- FillChar(LAddr, SizeOf(LAddr), 0);
- case AIPVersion of
- Id_IPv4: begin
- with Psockaddr(@LAddr)^ do
- begin
- sin_family := Id_PF_INET4;
- TranslateStringToTInAddr(AIP, sin_addr, Id_IPv4);
- sin_port := htons(APort);
- end;
- CheckForSocketError(Libc.connect(
- ASocket,
- {$IFDEF LIBCPASS_STRUCT}PSockAddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
- SizeOf(sockaddr)));
- end;
- Id_IPv6: begin
- with LAddr do
- begin
- sin6_family := Id_PF_INET6;
- TranslateStringToTInAddr(AIP, sin6_addr, Id_IPv6);
- sin6_port := htons(APort);
- end;
- CheckForSocketError(Libc.connect(
- ASocket,
- {$IFDEF LIBCPASS_STRUCT}Psockaddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
- SizeOf(sockaddr_in6)));
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- function TIdStackLibc.HostByName(const AHostName: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string;
- var
- Lpa: PAnsiChar;
- Lsa: TInAddr;
- LHost: PHostEnt;
- // ipv6
- LHints: TAddressInfo;
- {$IFDEF LIBCPASS_STRUCT}
- LAddrInfo: PAddressInfo;
- {$ELSE}
- LAddrInfo: PAddrInfo;
- {$ENDIF}
- LRetVal: Integer;
- {$IFDEF STRING_IS_UNICODE}
- LAStr: AnsiString;
- {$ENDIF}
- begin
- case AIPVersion of
- Id_IPv4: begin
- {$IFDEF STRING_IS_UNICODE}
- LAStr := AnsiString(AHostName); // explicit convert to Ansi
- {$ENDIF}
- // TODO: use getaddrinfo() instead for IPv4 as well...
- LHost := Libc.gethostbyname(
- PAnsiChar({$IFDEF STRING_IS_UNICODE}LAStr{$ELSE}AHostName{$ENDIF}));
- if LHost <> nil then begin
- // TODO: gethostbynaame() might return other things besides IPv4
- // addresses, so we should be validating the address type before
- // attempting the conversion...
- Lpa := LHost^.h_addr_list^;
- Lsa.S_un_b.s_b1 := Ord(Lpa[0]);
- Lsa.S_un_b.s_b2 := Ord(Lpa[1]);
- Lsa.S_un_b.s_b3 := Ord(Lpa[2]);
- Lsa.S_un_b.s_b4 := Ord(Lpa[3]);
- Result := TranslateTInAddrToString(Lsa, Id_IPv4);
- end else begin
- //RaiseSocketError(h_errno);
- RaiseLastSocketError;
- end;
- end;
- Id_IPv6: begin
- FillChar(LHints, SizeOf(LHints), 0);
- LHints.ai_family := IdIPFamily[AIPVersion];
- LHints.ai_socktype := Integer(SOCK_STREAM);
- LAddrInfo := nil;
- {$IFDEF STRING_IS_UNICODE}
- LAStr := AnsiString(AHostName); // explicit convert to Ansi
- {$ENDIF}
- LRetVal := getaddrinfo(
- PAnsiChar({$IFDEF STRING_IS_UNICODE}LAStr{$ELSE}AHostName{$ENDIF}),
- nil, @LHints, {$IFDEF LIBCPASS_STRUCT}LAddrInfo{$ELSE}@LAddrInfo{$ENDIF});
- if LRetVal <> 0 then begin
- if LRetVal = EAI_SYSTEM then begin
- IndyRaiseLastError;
- end else begin
- raise EIdResolveError.CreateFmt(RSResolveError, [AHostName, gai_strerror(LRetVal), LRetVal]);
- end;
- end;
- try
- Result := TranslateTInAddrToString(LAddrInfo^.ai_addr^.sin_zero, Id_IPv6);
- finally
- freeaddrinfo(LAddrInfo);
- end;
- end
- else
- Result := ''; // avoid warning
- IPVersionUnsupported;
- end;
- end;
- function TIdStackLibc.ReadHostName: string;
- var
- LStr: array[0..250] of AnsiChar;
- begin
- if Libc.gethostname(LStr, 250) = 0 then begin
- LStr[250] := #0;
- Result := String(LStr);
- end else begin
- Result := '';
- end;
- end;
- procedure TIdStackLibc.Listen(ASocket: TIdStackSocketHandle; ABackLog: Integer);
- begin
- CheckForSocketError(Libc.listen(ASocket, ABacklog));
- end;
- function TIdStackLibc.WSRecv(ASocket: TIdStackSocketHandle; var ABuffer;
- const ABufferLength, AFlags: Integer): Integer;
- begin
- //IdStackWindows is just: Result := Recv(ASocket, ABuffer, ABufferLength, AFlags);
- Result := Recv(ASocket, ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL);
- end;
- function TIdStackLibc.RecvFrom(const ASocket: TIdStackSocketHandle;
- var VBuffer; const ALength, AFlags: Integer; var VIP: string;
- var VPort: TIdPort; AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION ): Integer;
- var
- LiSize: UInt32;
- LAddr: sockaddr_in6;
- begin
- case AIPVersion of
- Id_IPv4,
- Id_IPv6: begin
- if AIPVersion = Id_IPv4 then begin
- LiSize := SizeOf(sockaddr);
- end else begin
- LiSize := SizeOf(sockaddr_in6);
- end;
- Result := Libc.recvfrom(ASocket, VBuffer, ALength, AFlags or Id_MSG_NOSIGNAL, PSockAddr(@LAddr), @LiSize);
- if AIPVersion = Id_IPv4 then begin
- with Psockaddr(@LAddr)^ do begin
- VIP := TranslateTInAddrToString(sin_addr, Id_IPv4);
- VPort := ntohs(sin_port);
- end;
- end else begin
- with LAddr do begin
- VIP := TranslateTInAddrToString(sin6_addr, Id_IPv6);
- VPort := ntohs(sin6_port);
- end;
- end;
- end;
- else begin
- Result := 0;
- IPVersionUnsupported;
- end;
- end;
- end;
- function TIdStackLibc.ReceiveMsg(ASocket: TIdStackSocketHandle;
- var VBuffer: TIdBytes; APkt: TIdPacketInfo): UInt32;
- {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)));
- APkt.Reset;
- 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; }
- begin
- APkt.Reset;
- Result := 0; // avoid warning
- end;
- function TIdStackLibc.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(Libc.send(ASocket, ABuffer, ABufferLength, AFlags));
- end;
- procedure TIdStackLibc.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;
- LiSize, LBytesOut: Integer;
- begin
- case AIPVersion of
- Id_IPv4, Id_IPv6:
- begin
- FillChar(LAddr, SizeOf(LAddr), 0);
- if AIPVersion = Id_IPv4 then begin
- with PsockAddr(@LAddr)^ do begin
- sin_family := Id_PF_INET4;
- TranslateStringToTInAddr(AIP, sin_addr, Id_IPV4);
- sin_port := htons(APort);
- end;
- LiSize := SizeOf(sockaddr);
- end else begin
- with LAddr do begin
- sin6_family := Id_PF_INET6;
- TranslateStringToTInAddr(AIP, sin6_addr, AIPVersion);
- sin6_port := htons(APort);
- end;
- LiSize := SizeOf(sockaddr_in6);
- end;
- LBytesOut := Libc.sendto(
- ASocket, ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL,
- {$IFDEF LIBCPASS_STRUCT}Psockaddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
- LiSize);
- end;
- else begin
- LBytesOut := 0; // avoid warning
- IPVersionUnsupported;
- end;
- end;
- if LBytesOut = Id_SOCKET_ERROR 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 TIdStackLibc.{$IFDEF VCL_XE3_OR_ABOVE}GetSocketOption{$ELSE}WSGetSocketOption{$ENDIF}
- (ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel; AOptName: TIdSocketOption;
- var AOptVal; var AOptLen: Integer);
- var
- LLen: UInt32;
- begin
- LLen := AOptLen;
- CheckForSocketError(Libc.getsockopt(ASocket, ALevel, AOptName, PIdAnsiChar(@AOptVal), LLen));
- AOptLen := LLen;
- end;
- procedure TIdStackLibc.{$IFDEF VCL_XE3_OR_ABOVE}SetSocketOption{$ELSE}WSSetSocketOption{$ENDIF}
- (ASocket: TIdStackSocketHandle; ALevel: TIdSocketOptionLevel; AOptName: TIdSocketOption;
- const AOptVal; const AOptLen: Integer);
- begin
- CheckForSocketError(Libc.setsockopt(ASocket, ALevel, AOptName, PIdAnsiChar(@AOptVal), AOptLen));
- end;
- function TIdStackLibc.WSGetLastError: Integer;
- begin
- //IdStackWindows just uses result := WSAGetLastError;
- Result := GetLastError; //System.GetLastOSError; - FPC doesn't define it in System
- if Result = Id_WSAEPIPE then begin
- Result := Id_WSAECONNRESET;
- end;
- end;
- function TIdStackLibc.WSSocket(AFamily : Integer; AStruct : TIdSocketType; AProtocol: Integer;
- const ANonBlocking: Boolean = False): TIdStackSocketHandle; override;
- begin
- Result := Libc.socket(AFamily, AStruct or iif(ANonBlocking, SOCK_NONBLOCK, 0), AProtocol);
- end;
- function TIdStackLibc.WSGetServByName(const AServiceName: string): TIdPort;
- var
- Lps: PServEnt;
- {$IFDEF STRING_IS_UNICODE}
- LAStr: AnsiString;
- {$ENDIF}
- begin
- {$IFDEF STRING_IS_UNICODE}
- LAStr := AnsiString(AServiceName); // explicit convert to Ansi
- {$ENDIF}
- Lps := Libc.getservbyname(
- PAnsiChar({$IFDEF STRING_IS_UNICODE}LAStr{$ELSE}AServiceName{$ENDIF}),
- nil);
- if Lps <> nil then begin
- Result := ntohs(Lps^.s_port);
- end else begin
- try
- Result := IndyStrToInt(AServiceName);
- except
- on EConvertError do begin
- IndyRaiseOuterException(EIdInvalidServiceName.CreateFmt(RSInvalidServiceName, [AServiceName]));
- end;
- end;
- end;
- end;
- function TIdStackLibc.WSGetServByPort(const APortNumber: TIdPort): TStrings;
- type
- PPAnsiCharArray = ^TPAnsiCharArray;
- TPAnsiCharArray = packed array[0..(Maxint div SizeOf(PAnsiChar))-1] of PAnsiChar;
- var
- Lps: PServEnt;
- Li: Integer;
- Lp: PPAnsiCharArray;
- begin
- Result := TStringList.Create;
- try
- Lps := Libc.getservbyport(htons(APortNumber), nil);
- if Lps <> nil then begin
- Result.Add(String(Lps^.s_name));
- Li := 0;
- Lp := Pointer(Lps^.s_aliases);
- while Lp[Li] <> nil do begin
- Result.Add(String(Lp[Li]));
- Inc(Li);
- end;
- end;
- except
- FreeAndNil(Result);
- raise;
- end;
- end;
- function TIdStackLibc.HostToNetwork(AValue: UInt16): UInt16;
- begin
- Result := htons(AValue);
- end;
- function TIdStackLibc.NetworkToHost(AValue: UInt16): UInt16;
- begin
- Result := ntohs(AValue);
- end;
- function TIdStackLibc.HostToNetwork(AValue: UInt32): UInt32;
- begin
- Result := htonl(AValue);
- end;
- function TIdStackLibc.NetworkToHost(AValue: UInt32): UInt32;
- begin
- Result := ntohl(AValue);
- end;
- { RP - I'm not sure what endian Linux natively uses, thus the
- check to see if the bytes need swapping or not ... }
- function TIdStackLibc.HostToNetwork(AValue: TIdUInt64): TIdUInt64;
- var
- LParts: TIdUInt64Parts;
- L: UInt32;
- begin
- 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;
- end;
- function TIdStackLibc.NetworkToHost(AValue: TIdUInt64): TIdUInt64;
- var
- LParts: TIdUInt64Parts;
- L: UInt32;
- begin
- if (ntohl(1) <> 1) then begin
- LParts.QuadPart := AValue{$IFDEF TIdUInt64_HAS_QuadPart}.QuadPart{$ENDIF};
- L := ntohl(LParts.HighPart);
- LParts.HighPart := ntohl(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;
- end;
- {$IFDEF HAS_getifaddrs}
- type
- TIdStackLocalAddressAccess = class(TIdStackLocalAddress)
- end;
- {$ENDIF}
- procedure TIdStackLibc.GetLocalAddressList(AAddresses: TIdStackLocalAddressList);
- {$IFNDEF HAS_getifaddrs}
- type
- TaPInAddr = array[0..250] of PInAddr;
- PaPInAddr = ^TaPInAddr;
- TaPIn6Addr = array[0..250] of PIn6Addr;
- PaPIn6Addr = ^TaPIn6Addr;
- {$ENDIF}
- var
- {$IFDEF HAS_getifaddrs}
- LAddrList, LAddrInfo: pifaddrs;
- LSubNetStr: string;
- LAddress: TIdStackLocalAddress;
- LName: string;
- {$ELSE}
- Li: Integer;
- LAHost: PHostEnt;
- LPAdrPtr: PaPInAddr;
- LPAdr6Ptr: PaPIn6Addr;
- LHostName: AnsiString;
- {$ENDIF}
- begin
- // TODO: Using gethostname() and gethostbyname() 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 := 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}
- // this won't get IPv6 addresses as I didn't find a way
- // to enumerate IPv6 addresses on a linux machine
- LHostName := AnsiString(HostName);
- LAHost := Libc.gethostbyname(PAnsiChar(LHostName));
- if LAHost = nil then begin
- RaiseLastSocketError;
- end;
- // gethostbyname() might return other things besides IPv4 addresses, so we
- // need to validate the address type before attempting the conversion...
- case LAHost^.h_addrtype of
- Id_PF_INET4: begin
- LPAdrPtr := PAPInAddr(LAHost^.h_addr_list);
- Li := 0;
- if LPAdrPtr^[Li] <> nil then begin
- AAddresses.BeginUpdate;
- try
- repeat
- TIdStackLocalAddressIPv4.Create(Addresses, TranslateTInAddrToString(LPAdrPtr^[Li]^, Id_IPv4), ''); // TODO: SubNet
- Inc(Li);
- until LPAdrPtr^[Li] = nil;
- finally
- AAddresses.EndUpdate;
- end;
- end;
- end;
- Id_PF_INET6: begin
- LPAdr6Ptr := PAPIn6Addr(LAHost^.h_addr_list);
- Li := 0;
- if LPAdr6Ptr^[Li] <> nil then begin
- AAddresses.BeginUpdate;
- try
- repeat
- TIdStackLocalAddressIPv6.Create(Addresses, TranslateTInAddrToString(LPAdr6Ptr^[Li]^, Id_IPv6));
- Inc(Li);
- until LPAdr6Ptr^[Li] = nil;
- finally
- AAddresses.EndUpdate;
- end;
- end;
- end;
- end;
- {$ENDIF}
- end;
- function TIdStackLibc.HostByAddress(const AAddress: string;
- const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string;
- var
- LAddr: sockaddr_in6;
- LSize: UInt32;
- LHostName : array[0..NI_MAXHOST] of TIdAnsiChar;
- {$IFDEF USE_MARSHALLED_PTRS}
- LHostNamePtr: TPtrWrapper;
- {$ENDIF}
- LRet : Integer;
- {$IFDEF LIBCPASS_STRUCT}
- LHints: TAddressInfo;
- LAddrInfo: PAddressInfo;
- {$ELSE}
- LHints: AddrInfo; //The T is no omission - that's what I found in the header
- LAddrInfo: PAddrInfo;
- {$ENDIF}
- begin
- FillChar(LAddr, SizeOf(LAddr), 0);
- case AIPVersion of
- Id_IPv4: begin
- with Psockaddr(@LAddr)^ do begin
- sin_family := Id_PF_INET4;
- TranslateStringToTInAddr(AAddress, sin_addr, Id_IPv4);
- end;
- LSize := SizeOf(sockaddr);
- end;
- Id_IPv6: begin
- with LAddr do begin
- sin6_family := Id_PF_INET6;
- TranslateStringToTInAddr(AAddress, sin6_addr, Id_IPv6);
- end;
- LSize := SizeOf(sockaddr_in6);
- end;
- else begin
- LSize := 0; // avoid warning
- IPVersionUnsupported;
- end;
- end;
- FillChar(LHostName[0],Length(LHostName),0);
- {$IFDEF USE_MARSHALLED_PTRS}
- LHostNamePtr := TPtrWrapper.Create(@LHostName[0]);
- {$ENDIF}
- LRet := getnameinfo(
- {$IFDEF LIBCPASS_STRUCT}Psockaddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
- LSize,
- {$IFDEF USE_MARSHALLED_PTRS}
- LHostNamePtr.ToPointer
- {$ELSE}
- LHostName
- {$ENDIF},
- NI_MAXHOST,nil,0,NI_NAMEREQD );
- if LRet <> 0 then begin
- if LRet = EAI_SYSTEM then begin
- RaiseLastOSError;
- end else begin
- raise EIdReverseResolveError.CreateFmt(RSReverseResolveError, [AAddress, gai_strerror(LRet), LRet]);
- end;
- end;
- {
- IMPORTANT!!!
- getnameinfo can return either results from a numeric to text conversion or
- results from a DNS reverse lookup. Someone could make a malicous PTR record
- such as
- 1.0.0.127.in-addr.arpa. IN PTR 10.1.1.1
- and trick a caller into beleiving the socket address is 10.1.1.1 instead of
- 127.0.0.1. If there is a numeric host in LAddr, than this is the case and
- we disregard the result and raise an exception.
- }
- FillChar(LHints,SizeOf(LHints),0);
- LHints.ai_socktype := SOCK_DGRAM; //*dummy*/
- LHints.ai_flags := AI_NUMERICHOST;
- if getaddrinfo(
- {$IFDEF USE_MARSHALLED_PTRS}
- LHostNamePtr.ToPointer
- {$ELSE}
- LHostName
- {$ENDIF},
- '0', LHints, LAddrInfo) = 0 then
- begin
- freeaddrinfo(LAddrInfo^);
- Result := '';
- raise EIdMaliciousPtrRecord.Create(RSMaliciousPtrRecord);
- end;
- {$IFDEF USE_MARSHALLED_PTRS}
- Result := TMarshal.ReadStringAsAnsi(LHostNamePtr);
- {$ELSE}
- Result := String(LHostName);
- {$ENDIF}
- (* JMB: I left this in here just in case someone
- complains, but the other code works on all
- linux systems for all addresses and is thread-safe
- variables for it:
- Host: PHostEnt;
- LAddr: u_long;
- Id_IPv4: begin
- // GetHostByAddr is thread-safe in Linux.
- // It might not be safe in Solaris or BSD Unix
- LAddr := inet_addr(PAnsiChar(AAddress));
- Host := GetHostByAddr(@LAddr,SizeOf(LAddr),AF_INET);
- if (Host <> nil) then begin
- Result := Host^.h_name;
- end else begin
- RaiseSocketError(h_errno);
- end;
- end;
- *)
- end;
- function TIdStackLibc.WSShutdown(ASocket: TIdStackSocketHandle; AHow: Integer): Integer;
- begin
- Result := Libc.shutdown(ASocket, AHow);
- end;
- procedure TIdStackLibc.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 TIdStackLibc.GetPeerName(ASocket: TIdStackSocketHandle;
- var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion);
- var
- i: UInt32;
- LAddr: sockaddr_in6;
- begin
- i := SizeOf(LAddr);
- CheckForSocketError(Libc.getpeername(ASocket, Psockaddr(@LAddr)^, i));
- case LAddr.sin6_family of
- Id_PF_INET4: begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := TranslateTInAddrToString(sin_addr, Id_IPv4);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPV4;
- end;
- Id_PF_INET6: begin
- with LAddr do
- begin
- VIP := TranslateTInAddrToString(sin6_addr, Id_IPv6);
- VPort := Ntohs(sin6_port);
- end;
- VIPVersion := Id_IPV6;
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- procedure TIdStackLibc.GetSocketName(ASocket: TIdStackSocketHandle;
- var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion);
- var
- i: UInt32;
- LAddr: sockaddr_in6;
- begin
- i := SizeOf(LAddr);
- CheckForSocketError(Libc.getsockname(ASocket, Psockaddr(@LAddr)^, i));
- case LAddr.sin6_family of
- Id_PF_INET4: begin
- with Psockaddr(@LAddr)^ do
- begin
- VIP := TranslateTInAddrToString(sin_addr, Id_IPv4);
- VPort := ntohs(sin_port);
- end;
- VIPVersion := Id_IPV4;
- end;
- Id_PF_INET6: begin
- with LAddr do
- begin
- VIP := TranslateTInAddrToString(sin6_addr, Id_IPv6);
- VPort := ntohs(sin6_port);
- end;
- VIPVersion := Id_IPV6;
- end;
- else begin
- IPVersionUnsupported;
- end;
- end;
- end;
- function TIdStackLibc.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 TIdStackLibc.SupportsIPv4: Boolean;
- begin
- //In Windows, this does something else. It checks the LSP's installed.
- Result := CheckIPVersionSupport(Id_IPv4);
- end;
- function TIdStackLibc.SupportsIPv6: Boolean;
- begin
- //In Windows, this does something else. It checks the LSP's installed.
- Result := CheckIPVersionSupport(Id_IPv6);
- end;
- function TIdStackLibc.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 TIdStackLibc.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 TIdStackLibc.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, IPPROTO_IPV6, IPV6_CHECKSUM, AOffset);
- end;
- function TIdStackLibc.IOControl(const s: TIdStackSocketHandle;
- const cmd: UInt32; var arg: UInt32): Integer;
- begin
- Result := ioctl(s, cmd, @arg);
- end;
- procedure TIdStackLibc.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
- SetSocketOption(ASocket, Id_SOL_TCP, Id_TCP_KEEPIDLE, ATimeMS div MSecsPerSec);
- SetSocketOption(ASocket, Id_SOL_TCP, Id_TCP_KEEPINTVL, AInterval div MSecsPerSec);
- end;
- end;
- { TIdSocketListLibc }
- procedure TIdSocketListLibc.Add(AHandle: TIdStackSocketHandle);
- begin
- Lock;
- try
- if not FD_ISSET(AHandle, FFDSet) then begin
- if AHandle >= __FD_SETSIZE then begin
- raise EIdStackSetSizeExceeded.Create(RSSetSizeExceeded);
- end;
- FD_SET(AHandle, FFDSet);
- Inc(FCount);
- end;
- finally
- Unlock;
- end;
- end;//
- procedure TIdSocketListLibc.Clear;
- begin
- Lock;
- try
- FD_ZERO(FFDSet);
- FCount := 0;
- finally
- Unlock;
- end;
- end;
- function TIdSocketListLibc.ContainsSocket(
- AHandle: TIdStackSocketHandle): boolean;
- begin
- Lock;
- try
- Result := FD_ISSET(AHandle, FFDSet);
- finally
- Unlock;
- end;
- end;
- function TIdSocketListLibc.Count: Integer;
- begin
- Lock;
- try
- Result := FCount;
- finally
- Unlock;
- end;
- end;//
- class function TIdSocketListLibc.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 := Libc.select(__FD_SETSIZE, AReadSet, AWriteSet, AExceptSet, LTimePtr);
- end;
- procedure TIdSocketListLibc.GetFDSet(var VSet: TFDSet);
- begin
- Lock;
- try
- VSet := FFDSet;
- finally
- Unlock;
- end;
- end;
- function TIdSocketListLibc.GetItem(AIndex: Integer): TIdStackSocketHandle;
- var
- LIndex, i: Integer;
- begin
- Result := 0;
- Lock;
- try
- LIndex := 0;
- //? use FMaxHandle div x
- for i:= 0 to __FD_SETSIZE - 1 do begin
- if FD_ISSET(i, FFDSet) then begin
- if LIndex = AIndex then begin
- Result := i;
- Break;
- end;
- Inc(LIndex);
- end;
- end;
- finally
- Unlock;
- end;
- end;//
- procedure TIdSocketListLibc.Remove(AHandle: TIdStackSocketHandle);
- begin
- Lock;
- try
- if FD_ISSET(AHandle, FFDSet) then begin
- Dec(FCount);
- FD_CLR(AHandle, FFDSet);
- end;
- finally
- Unlock;
- end;
- end;//
- function TIdStackLibc.WSTranslateSocketErrorMsg(const AErr: Integer): string;
- begin
- //we override this function for the herr constants that
- //are returned by the DNS functions
- case AErr of
- Libc.HOST_NOT_FOUND: Result := RSStackHOST_NOT_FOUND;
- Libc.TRY_AGAIN: Result := RSStackTRY_AGAIN;
- Libc.NO_RECOVERY: Result := RSStackNO_RECOVERY;
- Libc.NO_DATA: Result := RSStackNO_DATA;
- else
- Result := inherited WSTranslateSocketErrorMsg(AErr);
- end;
- end;
- procedure TIdSocketListLibc.SetFDSet(var VSet: TFDSet);
- begin
- Lock;
- try
- FFDSet := VSet;
- finally
- Unlock;
- end;
- end;
- class function TIdSocketListLibc.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
- TIdSocketListLibc(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
- TIdSocketListLibc(AReadList).SetFDSet(LReadSet);
- end;
- if AWriteList <> nil then begin
- TIdSocketListLibc(AWriteList).SetFDSet(LWriteSet);
- end;
- if AExceptList <> nil then begin
- TIdSocketListLibc(AExceptList).SetFDSet(LExceptSet);
- end;
- end;
- function TIdSocketListLibc.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 TIdSocketListLibc.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;
- TIdSocketListLibc(VSocketList).SetFDSet(LSet);
- end;
- end;
- procedure TIdStackLibc.SetBlocking(ASocket: TIdStackSocketHandle;
- const ABlocking: Boolean);
- var
- LFlags: Integer;
- //LValue: UInt32;
- begin
- LFlags := CheckForSocketError(Libc.fcntl(ASocket, F_GETFL, 0));
- if ABlocking then begin
- LFlags := LFlags and not O_NONBLOCK;
- end else begin
- LFlags := LFlags or O_NONBLOCK;
- end;
- CheckForSocketError(Libc.fcntl(ASocket, F_SETFL, LFlags));
- {
- LValue := UInt32(not ABlocking);
- CheckForSocketError(Libc.ioctl(ASocket, FIONBIO, @LValue));
- }
- 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(PAnsiChar(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;
- *)
- function TIdSocketListLibc.Clone: TIdSocketList;
- begin
- Result := TIdSocketListLibc.Create;
- try
- Lock;
- try
- TIdSocketListLibc(Result).SetFDSet(FFDSet);
- finally
- Unlock;
- end;
- except
- FreeAndNil(Result);
- raise;
- end;
- end;
- initialization
- GSocketListClass := TIdSocketListLibc;
- end.
|