Subversion Repositories userdetect2

unit NetworkUtils;

interface

uses

  Windows, SysUtils, Classes;

function GetLocalIPAddressList(outsl: TStrings): DWORD;

function GetLocalMACAddressList(outSL: TStrings): DWORD;

function GetDHCPIPAddressList(outsl: TStrings): DWORD;

function GetGatewayIPAddressList(outsl: TStrings): DWORD;

function GetMACAddress(const IPAddress: string; var outAddress: string): DWORD;

function FormatMAC(s: string): string;

function GetDomainName(var outDomainName: WideString): boolean;

implementation

uses

  {$IFDEF UNICODE}

  iphlpapi, IpTypes, Iprtrmib,

  {$ELSE}

  iphlp,

  {$ENDIF}

  WinSock, Registry;

// TODO: Replace GetAdaptersInfo()? A comment at MSDN states that there might be problems with IPv6

//           "GetAdaptersInfo returns ERROR_NO_DATA if there are only IPv6 interfaces

//            configured on system. In that case GetAdapterAddresses has to be used!"

function GetLocalIPAddressList(outsl: TStrings): DWORD;

var

  pAdapterInfo: PIP_ADAPTER_INFO;

  addr: AnsiString;

  addrStr: IP_ADDR_STRING;

  BufLen: ULONG;

begin

  BufLen := SizeOf(IP_ADAPTER_INFO);

  Result := GetAdaptersInfo(nil, BufLen);

  if Result <> ERROR_BUFFER_OVERFLOW then Exit;

  pAdapterInfo := AllocMem(BufLen);

  try

    ZeroMemory(pAdapterInfo, BufLen);

    Result := GetAdaptersInfo(pAdapterInfo, BufLen);

    if Result <> ERROR_SUCCESS then Exit;

    while pAdapterInfo <> nil do

    begin

      addrStr := pAdapterInfo^.IpAddressList;

      repeat

        addr := addrStr.IpAddress.S;

        if (addr <> '') and (outsl.IndexOf(addr) = -1) then

          outsl.Add(addr);

        if addrStr.Next = nil then break;

        AddrStr := addrStr.Next^;

      until false;

      pAdapterInfo := pAdapterInfo^.next;

    end;

  finally

    Freemem(pAdapterInfo);

  end;

end;

function GetDHCPIPAddressList(outsl: TStrings): DWORD;

var

  pAdapterInfo: PIP_ADAPTER_INFO;

  addr: AnsiString;

  addrStr: IP_ADDR_STRING;

  BufLen: ULONG;

begin

  BufLen := SizeOf(IP_ADAPTER_INFO);

  Result := GetAdaptersInfo(nil, BufLen);

  if Result <> ERROR_BUFFER_OVERFLOW then Exit;

  pAdapterInfo := AllocMem(BufLen);

  try

    ZeroMemory(pAdapterInfo, BufLen);

    Result := GetAdaptersInfo(pAdapterInfo, BufLen);

    if Result <> ERROR_SUCCESS then Exit;

    while pAdapterInfo <> nil do

    begin

      addrStr := pAdapterInfo^.DhcpServer;

      repeat

        addr := addrStr.IpAddress.S;

        if (addr <> '') and (outsl.IndexOf(addr) = -1) then

          outsl.Add(addr);

        if addrStr.Next = nil then break;

        AddrStr := addrStr.Next^;

      until false;

      pAdapterInfo := pAdapterInfo^.next;

    end;

  finally

    Freemem(pAdapterInfo);

  end;

end;

function GetGatewayIPAddressList(outsl: TStrings): DWORD;

var

  pAdapterInfo: PIP_ADAPTER_INFO;

  addr: AnsiString;

  addrStr: IP_ADDR_STRING;

  BufLen: ULONG;

begin

  BufLen := SizeOf(IP_ADAPTER_INFO);

  Result := GetAdaptersInfo(nil, BufLen);

  if Result <> ERROR_BUFFER_OVERFLOW then Exit;

  pAdapterInfo := AllocMem(BufLen);

  try

    ZeroMemory(pAdapterInfo, BufLen);

    Result := GetAdaptersInfo(pAdapterInfo, BufLen);

    if Result <> ERROR_SUCCESS then Exit;

    while pAdapterInfo <> nil do

    begin

      addrStr := pAdapterInfo^.GatewayList;

      repeat

        addr := addrStr.IpAddress.S;

        if (addr <> '') and (outsl.IndexOf(addr) = -1) then

          outsl.Add(addr);

        if addrStr.Next = nil then break;

        AddrStr := addrStr.Next^;

      until false;

      pAdapterInfo := pAdapterInfo^.next;

    end;

  finally

    Freemem(pAdapterInfo);

  end;

end;

function GetMACAddress(const IPAddress: string; var outAddress: string): DWORD;

// http://stackoverflow.com/questions/4550672/delphi-get-mac-of-router

var

  MacAddr    : Array[0..5] of Byte;

  DestIP     : ULONG;

  PhyAddrLen : ULONG;

  WSAData    : TWSAData;

  j: integer;

begin

  outAddress := '';

  WSAStartup($0101, WSAData);

  try

    ZeroMemory(@MacAddr, SizeOf(MacAddr));

    DestIP     := inet_addr(PAnsiChar(IPAddress));

    PhyAddrLen := SizeOf(MacAddr); // TODO: more ?

    Result     := SendArp(DestIP, 0, @MacAddr, PhyAddrLen);

    if Result = S_OK then

    begin

      outAddress := '';

      for j := 0 to PhyAddrLen-1 do

      begin

        outAddress := outAddress + format('%.2x', [MacAddr[j]]);

      end;

      outAddress := FormatMAC(outAddress);

    end;

  finally

    WSACleanup;

  end;

end;

function GetLocalMACAddressList(outSL: TStrings): DWORD;

var

  pIfTable: PMIB_IFTABLE;

  TableSize: ULONG;

  tmp: String;

  i, j: Integer;

begin

  pIfTable := nil;

  try

    // First: just get the buffer size.

    // TableSize returns the size needed.

    TableSize := 0; // Set to zero so the GetIfTable function won't try to fill the buffer yet,

                    // but only return the actual size it needs.

    GetIfTable(pIfTable, TableSize, true);

    if (TableSize < SizeOf(MIB_IFROW)+SizeOf(LongInt)) then

    begin

      Result := ERROR_NO_DATA;

      Exit; // less than 1 table entry?!

    end;

    // Second:

    // allocate memory for the buffer and retrieve the

    // entire table.

    GetMem(pIfTable, TableSize);

    Result := GetIfTable(pIfTable, TableSize, true);

    if Result <> NO_ERROR then Exit;

    // Read the ETHERNET addresses.

    for i := 1 to pIfTable^.dwNumEntries do

    begin

      //if pIfTable^.table[i].dwType = MIB_IF_TYPE_ETHERNET then

      begin

        tmp := '';

        for j := 0 to pIfTable^.table[i].dwPhysAddrLen-1 do

        begin

          tmp := tmp + format('%.2x', [pIfTable^.table[i].bPhysAddr[j]]);

        end;

        tmp := FormatMAC(tmp);

        if (tmp <> '') and (outSL.IndexOf(tmp) = -1) then

          outSL.Add(tmp);

      end;

    end;

  finally

    if Assigned(pIfTable) then FreeMem(pIfTable, TableSize);

  end;

end;

function FormatMAC(s: string): string;

var

  m: integer;

begin

  result := '';

  m := 1;

  s := UpperCase(s);

  repeat

    if m > 1 then result := result + '-';

    result := result + Copy(s, m, 2);

    inc(m, 2);

  until m > Length(s);

end;

(*

type

  WKSTA_INFO_100   = Record

      wki100_platform_id  : DWORD;

      wki100_computername : LPWSTR;

      wki100_langroup     : LPWSTR;

      wki100_ver_major    : DWORD;

      wki100_ver_minor    : DWORD;

  End;

   LPWKSTA_INFO_100 = ^WKSTA_INFO_100;

function GetNetParam(AParam: integer): string;

Var

  PBuf  : LPWKSTA_INFO_100;

  Res   : LongInt;

begin

  result := '';

  Res := NetWkstaGetInfo(nil, 100, @PBuf);

  If Res = NERR_Success Then

    begin

      case AParam of

       0:   Result := string(PBuf^.wki100_computername);

       1:   Result := string(PBuf^.wki100_langroup);

      end;

    end;

end;

function GetTheComputerName: string;

begin

  Result := GetNetParam(0);

end;

function GetTheDomainName: string;

begin

  Result := GetNetParam(1);

end;

*)

function GetDomainName(var outDomainName: WideString): boolean;

var

  reg: TRegistry;

begin

  outDomainName := '';

  reg := TRegistry.Create;

  try

    reg.RootKey := HKEY_LOCAL_MACHINE;

    result := reg.OpenKeyReadOnly('\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters');

    if result then

    begin

      outDomainName := reg.ReadString('Domain');

      reg.CloseKey;

    end;

  finally

    reg.Free;

  end;

end;

end.