restemplate/indy/System/IdStackLinux.pas

1410 lines
43 KiB
Plaintext
Raw Normal View History

{
$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 IdStackLinux;
interface
{$i IdCompilerDefines.inc}
uses
Classes,
Libc,
IdStack,
IdStackConsts,
IdGlobal,
IdStackBSDBase;
type
TIdSocketListLinux = 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
TIdStackLinux = 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;
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 : Integer; AStruct : TIdSocketType; AProtocol: Integer;
const AOverlapped: 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}
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,
IdResourceStringsKylixCompat,
IdResourceStringsUnix,
IdException,
SysUtils;
type
psockaddr_in6 = ^sockaddr_in6;
const
Id_MSG_NOSIGNAL = MSG_NOSIGNAL;
Id_WSAEPIPE = EPIPE;
constructor TIdStackLinux.Create;
begin
inherited Create;
end;
destructor TIdStackLinux.Destroy;
begin
inherited Destroy;
end;
function TIdStackLinux.GetLastError : Integer;
begin
Result := errno;
end;
procedure TIdStackLinux.SetLastError(Const AError : Integer);
begin
__errno_location^ := AError;
end;
procedure TIdStackLinux.WSSetLastError(const AErr : Integer);
begin
SetLastError(AErr);
end;
function TIdStackLinux.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 TIdStackLinux.Bind(ASocket: TIdStackSocketHandle;
const AIP: string; const APort: TIdPort;
const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
var
LAddr: sockaddr_in6;
LSize: UInt32;
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;
LSize := 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;
LSize := SizeOf(sockaddr_in6);
end;
else begin
LSize := 0; // avoid warning
IPVersionUnsupported;
end;
end;
CheckForSocketError(Libc.bind(ASocket, Psockaddr(@LAddr), LSize);
end;
function TIdStackLinux.WSCloseSocket(ASocket: TIdStackSocketHandle): Integer;
begin
Result := Libc.__close(ASocket);
end;
procedure TIdStackLinux.Connect(const ASocket: TIdStackSocketHandle;
const AIP: string; const APort: TIdPort;
const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION);
var
LAddr: sockaddr_in6;
LSize: UInt32;
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;
LSize := 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;
LSize := SizeOf(sockaddr_in6);
end;
else begin
LSize := 0; // avoid warning
IPVersionUnsupported;
end;
end;
CheckForSocketError(Libc.connect(
ASocket,
{$IFDEF KYLIX}Psockaddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
LSize));
end;
function TIdStackLinux.HostByName(const AHostName: string;
const AIPVersion: TIdIPVersion = ID_DEFAULT_IP_VERSION): string;
var
Lpa: PAnsiChar;
Lsa: TInAddr;
LHost: PHostEnt;
// ipv6
LHints: TAddressInfo;
{$IFDEF KYLIX}
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: gethostbyname() 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 KYLIX}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 TIdStackLinux.ReadHostName: string;
var
LStr: AnsiString;
begin
SetLength(LStr, 250);
Libc.gethostname(PAnsiChar(LStr), 250);
Result := PAnsiChar(LStr);
end;
procedure TIdStackLinux.Listen(ASocket: TIdStackSocketHandle; ABackLog: Integer);
begin
CheckForSocketError(Libc.listen(ASocket, ABacklog));
end;
function TIdStackLinux.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 TIdStackLinux.RecvFrom(const ASocket: TIdStackSocketHandle;
var VBuffer; const ALength, AFlags: Integer; var VIP: string;
var VPort: TIdPort; var VIPVersion: TIdIPVersion): Integer;
var
LiSize: UInt32;
LAddr: sockaddr_in6;
begin
LiSize := SizeOf(sockaddr_in6);
Result := Libc.recvfrom(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 := 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
Result := 0;
IPVersionUnsupported;
end;
end;
end;
end;
function TIdStackLinux.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
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_LEN(CMSG_LEN(Length(VBuffer)));
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
with Pin_pktinfo(WSA_CMSG_DATA(LCurCmsg))^ do
begin
APkt.DestIP := GWindowsStack.TranslateTInAddrToString(ipi_addr, Id_IPv4);
APkt.DestIF := ipi_ifindex;
end;
APkt.DestIPVersion := Id_IPv4;
end;
Id_PF_INET6: begin
with Pin6_pktinfo(WSA_CMSG_DATA(LCurCmsg))^ do
begin
APkt.DestIP := GWindowsStack.TranslateTInAddrToString(ipi6_addr, Id_IPv6);
APkt.DestIF := ipi6_ifindex;
end;
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 TIdStackLinux.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 TIdStackLinux.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: Integer;
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;
LiSize := 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;
LiSize := SizeOf(sockaddr_in6);
end;
else begin
LiSize := 0; // avoid warning
IPVersionUnsupported;
end;
end;
LiSize := Libc.sendto(
ASocket, ABuffer, ABufferLength, AFlags or Id_MSG_NOSIGNAL,
{$IFDEF KYLIX}Psockaddr(@LAddr)^{$ELSE}Psockaddr(@LAddr){$ENDIF},
LiSize);
end;
if LiSize = 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 LiSize <> ABufferLength then begin
raise EIdNotAllBytesSent.Create(RSNotAllBytesSent);
end;
end;
procedure TIdStackLinux.{$IFDEF VCL_XE3_OR_ABOVE}GetSocketOption{$ELSE}WSGetSocketOption{$ENDIF}
(ASocket: TIdStackSocketHandle; ALevel: TIdSocketProtocol; AOptName: TIdSocketOption;
var AOptVal; var AOptLen: Integer);
var
LLen: UInt32;
begin
LLen := AOptLen;
CheckForSocketError(Libc.getsockopt(ASocket, ALevel, AOptName, PAnsiChar(@AOptVal), LLen));
AOptLen := LLen;
end;
procedure TIdStackLinux.{$IFDEF VCL_XE3_OR_ABOVE}SetSocketOption{$ELSE}WSSetSocketOption{$ENDIF}
(ASocket: TIdStackSocketHandle; ALevel: TIdSocketProtocol; AOptName: TIdSocketOption;
const AOptVal; const AOptLen: Integer);
begin
CheckForSocketError(Libc.setsockopt(ASocket, ALevel, AOptName, PAnsiChar(@AOptVal), AOptLen));
end;
function TIdStackLinux.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 TIdStackLinux.WSSocket(AFamily : Integer; AStruct : TIdSocketType; AProtocol: Integer;
const AOverlapped: Boolean = False): TIdStackSocketHandle;
begin
Result := Libc.socket(AFamily, AStruct, AProtocol);
end;
function TIdStackLinux.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 TIdStackLinux.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;
Lp := nil;
try
Lps := Libc.getservbyport(htons(APortNumber), nil);
if Lps <> nil then begin
Result.Add(Lps^.s_name);
Li := 0;
Lp := Pointer(Lps^.s_aliases);
while Lp[Li] <> nil do begin
Result.Add(Lp[Li]);
Inc(Li);
end;
end;
except
FreeAndNil(Result);
raise;
end;
end;
function TIdStackLinux.HostToNetwork(AValue: UInt16): UInt16;
begin
Result := htons(AValue);
end;
function TIdStackLinux.NetworkToHost(AValue: UInt16): UInt16;
begin
Result := ntohs(AValue);
end;
function TIdStackLinux.HostToNetwork(AValue: UInt32): UInt32;
begin
Result := htonl(AValue);
end;
function TIdStackLinux.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 TIdStackLinux.HostToNetwork(AValue: TIdUInt64): TIdUInt64;
var
LParts: TIdUInt64Parts;
L: UInt32;
begin
LParts.QuadPart := AValue{$IFDEF TIdUInt64_IS_NOT_NATIVE}.QuadPart{$ENDIF};
L := htonl(LParts.HighPart);
if (L <> LParts.HighPart) then begin
LParts.HighPart := htonl(LParts.LowPart);
LParts.LowPart := L;
end;
Result{$IFDEF TIdUInt64_IS_NOT_NATIVE}.QuadPart{$ENDIF} := LParts.QuadPart;
end;
function TIdStackLinux.NetworkToHost(AValue: TIdUInt64): TIdUInt64;
var
LParts: TIdUInt64Parts;
L: UInt32;
begin
LParts.QuadPart := AValue{$IFDEF TIdUInt64_IS_NOT_NATIVE}.QuadPart{$ENDIF};
L := ntohl(LParts.HighPart);
if (L <> LParts.HighPart) then begin
LParts.HighPart := ntohl(LParts.LowPart);
LParts.LowPart := L;
end;
Result{$IFDEF TIdUInt64_IS_NOT_NATIVE}.QuadPart{$ENDIF} := LParts.QuadPart;
end;
procedure TIdStackLinux.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;
{$ELSE}
Li: Integer;
LAHost: PHostEnt;
LPAdrPtr: PaPInAddr;
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
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;
TIdStackLocalAddressIPv4.Create(AAddresses, TranslateTInAddrToString(PSockAddr_In(LAddrInfo^.ifa_addr)^.sin_addr, Id_IPv4), LSubNetStr);
end;
Id_PF_INET6: begin
TIdStackLocalAddressIPv6.Create(AAddresses, TranslateTInAddrToString(PSockAddr_In6(LAddrInfo^.ifa_addr)^.sin6_addr, Id_IPv6));
end;
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 := 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(AAddresses, 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(AAddresses, TranslateTInAddrToString(LPAdr6Ptr^[Li]^, Id_IPv6));
Inc(Li);
until LPAdr6Ptr^[Li] = nil;
finally
AAddresses.EndUpdate;
end;
end;
end;
end;
{$ENDIF}
end;
function TIdStackLinux.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 KYLIX}
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(Psockaddr(@LAddr)^, 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 TIdStackLinux.WSShutdown(ASocket: TIdStackSocketHandle; AHow: Integer): Integer;
begin
Result := Libc.shutdown(ASocket, AHow);
end;
procedure TIdStackLinux.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 TIdStackLinux.GetPeerName(ASocket: TIdStackSocketHandle;
var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion);
var
LiSize: UInt32;
LAddr: sockaddr_in6;
begin
LiSize := SizeOf(LAddr);
CheckForSocketError(Libc.getpeername(ASocket, Psockaddr(@LAddr)^, LiSize));
case LAddr.sin6_family of
Id_PF_INET4: begin
with Psockaddr(@LAddr6)^ 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 TIdStackLinux.GetSocketName(ASocket: TIdStackSocketHandle;
var VIP: string; var VPort: TIdPort; var VIPVersion: TIdIPVersion);
var
LiSize: UInt32;
LAddr: sockaddr_in6;
begin
LiSize := SizeOf(LAddr);
CheckForSocketError(Libc.getsockname(ASocket, Psockaddr(@LAddr)^, LiSize));
case LAddr.sin6_family of
Id_PF_INET4: begin
with Psockaddr(@LAddr6)^ 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 TIdStackLinux.WouldBlock(const AResult: Integer): Boolean;
begin
//non-blocking does not exist in Linux, always indicate things will block
Result := True;
end;
function TIdStackLinux.SupportsIPv6: Boolean;
begin
//In Windows, this does something else. It checks the LSP's installed.
Result := CheckIPVersionSupport(Id_IPv6);
end;
function TIdStackLinux.CheckIPVersionSupport(const AIPVersion: TIdIPVersion): Boolean;
var
LTmpSocket: TIdStackSocketHandle;
begin
// TODO: an alternative would be to check for the existance of the '/proc/net/if_inet6' kernel pseudo-file
LTmpSocket := WSSocket(IdIPFamily[AIPVersion], Id_SOCK_STREAM, Id_IPPROTO_IP );
Result := LTmpSocket <> Id_INVALID_SOCKET;
if Result then begin
WSCloseSocket(LTmpSocket);
end;
end;
procedure TIdStackLinux.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 TIdStackLinux.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 TIdStackLinux.IOControl(const s: TIdStackSocketHandle;
const cmd: UInt32; var arg: UInt32): Integer;
begin
Result := ioctl(s, cmd, @arg);
end;
{ TIdSocketListLinux }
procedure TIdSocketListLinux.Add(AHandle: TIdStackSocketHandle);
begin
Lock;
try
if not FD_ISSET(AHandle, FFDSet) then begin
if Count >= __FD_SETSIZE then begin
raise EIdStackSetSizeExceeded.Create(RSSetSizeExceeded);
end;
FD_SET(AHandle, FFDSet);
Inc(FCount);
end;
finally
Unlock;
end;
end;//
procedure TIdSocketListLinux.Clear;
begin
Lock;
try
FD_ZERO(FFDSet);
FCount := 0;
finally
Unlock;
end;
end;
function TIdSocketListLinux.ContainsSocket(
AHandle: TIdStackSocketHandle): boolean;
begin
Lock;
try
Result := FD_ISSET(AHandle, FFDSet);
finally
Unlock;
end;
end;
function TIdSocketListLinux.Count: Integer;
begin
Lock;
try
Result := FCount;
finally
Unlock;
end;
end;//
class function TIdSocketListLinux.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...
Result := Libc.select(MaxLongint, AReadSet, AWriteSet, AExceptSet, LTimePtr);
end;
procedure TIdSocketListLinux.GetFDSet(var VSet: TFDSet);
begin
Lock;
try
VSet := FFDSet;
finally
Unlock;
end;
end;
function TIdSocketListLinux.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 TIdSocketListLinux.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 TIdStackLinux.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 TIdSocketListLinux.SetFDSet(var VSet: TFDSet);
begin
Lock;
try
FFDSet := VSet;
finally
Unlock;
end;
end;
class function TIdSocketListLinux.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
TIdSocketListLinux(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
TIdSocketListLinux(AReadList).SetFDSet(LReadSet);
end;
if AWriteList <> nil then begin
TIdSocketListLinux(AWriteList).SetFDSet(LWriteSet);
end;
if AExceptList <> nil then begin
TIdSocketListLinux(AExceptList).SetFDSet(LExceptSet);
end;
end;
function TIdSocketListLinux.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 TIdSocketListLinux.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;
TIdSocketListLinux(VSocketList).SetFDSet(LSet);
end;
end;
procedure TIdStackLinux.SetBlocking(ASocket: TIdStackSocketHandle;
const ABlocking: Boolean);
begin
if not ABlocking then begin
raise EIdNonBlockingNotSupported.Create(RSStackNonBlockingNotSupported);
end;
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 TIdSocketListLinux.Clone: TIdSocketList;
begin
Result := TIdSocketListLinux.Create;
try
Lock;
try
TIdSocketListLinux(Result).SetFDSet(FFDSet);
finally
Unlock;
end;
except
FreeAndNil(Result);
raise;
end;
end;
initialization
GSocketListClass := TIdSocketListLinux;
end.