{##########################################
         旁注入侵專用程序 3.0升級版
 -----------------------------------------
  模塊：網(wǎng)絡查詢 - 讀取網(wǎng)頁代碼
  描述：該單元主要用于讀取指定URL的HTML代碼
  作者：2005.3.8日下午  明小子
##########################################}

unit GetHtmlCode;

interface

uses
  SysUtils,Wininet,Windows,Classes;

function UrlGetStr(const URL: string; ShowHeaders: boolean = false): string;

implementation

function UrlGetStr(const URL: string; ShowHeaders: boolean = false): string;
const
  Agent = 'Internet Explorer 6.0';
var
  hFile, HInet: HINTERNET;
  Buffer: array[0..32767] of Char;
  BufRead: Cardinal;
  BufSize: Cardinal;
  TempStream: TStringStream;
  dwIndex: dword;
begin
  HInet := InternetOpen(PChar(Agent), INTERNET_OPEN_TYPE_PRECONFIG, nil, nil, 0);
  if Assigned(HInet) then
  try
    if LowerCase(Copy(URL,1,7)) <> 'http://' then
    hFile := InternetOpenUrl(HInet, PChar('http://' + URL), nil, 0, 0, 0)
    else
    hFile := InternetOpenUrl(HInet, PChar(URL), nil, 0, 0, 0);
    TempStream := TStringStream.Create('');
    dwIndex := 0;
    BufSize := SizeOf(Buffer);
    HttpQueryInfo(hfile, HTTP_QUERY_RAW_HEADERS_CRLF, @Buffer, BufSize, dwIndex);
    if ShowHeaders then TempStream.Write(Buffer, BufSize);
    if Assigned(hFile) then
    try
      with TempStream do
      try
        while InternetReadFile(hFile, @Buffer, BufSize, BufRead) and (BufRead > 0) do
          Write(Buffer, BufRead);
        Result := DataString;
      finally
        Free;
      end;
    finally
      InternetCloseHandle(hFile);
    end;
  finally
    InternetCloseHandle(hinet);
  end;
end;

end.