unit FSockets; {(c)Flenov forum.antichat}

interface

uses
  SysUtils, WinSock2;

type
  TFSocket = TSocket;


function FSocketsInicialize(Version: Word=$0202): Integer;
function FSockOpen(Host: String; Port: Word): TFSocket;
function FPuts(Sock: TFSocket; Str: String): Integer;
function FGets(Sock: TFSocket; var Size: Integer): String;
function FRedySend(Sock: TFSocket; Timeout: LongWord=10): Boolean;
function FRedyRead(Sock: TFSocket; Timeout: LongWord=10): Boolean;
function FClose(Sock: TFSocket): Integer;
function FSocketsDeinicialize: Integer;

implementation


function FSocketsInicialize(Version: Word=$0202): Integer;
var
  wData: TWSAData;
begin
  Result:=WSAStartup(Version, wData);
end;

function FSockOpen(Host: String; Port: Word): TFSocket;
var
  Sock: TFSocket;
  SockAddr: TSockAddrIn;
  HostEnt: PHostEnt;
  InAddr: TInAddr;
  FDSet: TFDSet;
begin
  Result:=0;
  HostEnt:=GetHostByName(PChar(Host));
  if (HostEnt=nil) then Exit;
  FillChar(InAddr, SizeOf(InAddr), 0);
  With InAddr, HostEnt^ do
                          begin
                            S_un_b.s_b1:=Byte(h_addr^[0]);
                            S_un_b.s_b2:=Byte(h_addr^[1]);
                            S_un_b.s_b3:=Byte(h_addr^[2]);
                            S_un_b.s_b4:=Byte(h_addr^[3]);
                          end;
  SockAddr.sin_family:=AF_INET;
  SockAddr.sin_addr:=InAddr;
  SockAddr.sin_port:=HToNS(Port);
  FillChar(SockAddr.sin_zero, SizeOf(SockAddr.sin_zero), 0);
  Sock:=Socket(AF_INET, SOCK_STREAM, 0);
  if (Sock<>INVALID_SOCKET) then
    if (Connect(Sock, @SockAddr, Sizeof(SockAddr))<>SOCKET_ERROR) then Result:=Sock
                                                                  else CloseSocket(Sock);
end;

function FPuts(Sock: TFSocket; Str: String): Integer;
begin
  Result:=Send(Sock, PChar(Str)^, Length(Str), 0);
end;

function FGets(Sock: TFSocket; var Size: Integer): String;
var
  Buff: PChar;
begin
  Buff:=GetMemory(Size);
  try
    Size:=Recv(Sock, Buff^, Size, 0);
    Result:=Copy(Buff, 1, Size);
  finally
    FreeMemory(Buff);
  end;
end;

function FRedySend(Sock: TFSocket; Timeout: LongWord=10): Boolean;
var
  WFDS: TFDSet;
  tv: TTimeVal;
begin
  Result:=False;
  tv.tv_sec:=Timeout;
  tv.tv_usec:=0;
  FD_ZERO(WFDS);
  FD_SET(Sock, WFDS);
  if FD_ISSET(Sock, WFDS) then
    if (Select(0, nil, @WFDS, nil, @tv)=1) then Result:=True;
end;

function FRedyRead(Sock: TFSocket; Timeout: LongWord=10): Boolean;
var
  RFDS: TFDSet;
  tv: TTimeVal;
begin
  Result:=False;
  tv.tv_sec:=Timeout;
  tv.tv_usec:=0;
  FD_ZERO(RFDS);
  FD_SET(Sock, RFDS);
  if FD_ISSET(Sock, RFDS) then
    if (Select(0, @RFDS, nil, nil, @tv)=1) then Result:=True;
end;

function FClose(Sock: TFSocket): Integer;
begin
  Result:=CloseSocket(Sock);
end;

function FSocketsDeinicialize: Integer;
begin
  Result:=WSACleanup;
end;
end.