unit netlist;

interface

uses
  Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms,
  Dialogs, ImgList, StdCtrls, Buttons, ComCtrls, ExtCtrls;

type
  TNetResourceArray = ^TnetResource;
  TFm_netlist = class(TForm)
    Panel1: TPanel;
    TreeView1: TTreeView;
    Panel2: TPanel;
    statusbar: TStatusBar;
    BitBtn1: TBitBtn;
    BitBtn2: TBitBtn;
    ImageList1: TImageList;
    procedure FormActivate(Sender: TObject);
    procedure TreeView1Click(Sender: TObject);
    procedure BitBtn1Click(Sender: TObject);
    procedure BitBtn2Click(Sender: TObject);
  private
    procedure getcomputername;
    procedure getgroupname;
    { Private declarations }
  public
    { Public declarations }
  end;
function NetExecute(aFormClass: TFormClass): string;
var
  fm_NetList: Tfm_NetList;
  node: Ttreenode;
  aComputerName : string;//全局變量，存儲選中的主機名

implementation

{$R *.dfm}
//打開窗體，并返回選定的計算機
function NetExecute(aFormClass: TFormClass): string;
begin
  aComputerName := '';
  with aFormClass.Create(Application) do
  begin
    try
      showModal;
    finally
      free;
    end;
    result := aComputerName;
  end;
end;

procedure TFm_netlist.FormActivate(Sender: TObject);
begin
node:=treeview1.items.add(Treeview1.topitem, '整個網絡');
node.imageindex:=0;
treeview1.setfocus;
end;

procedure TFm_netlist.TreeView1Click(Sender: TObject);
begin
  if treeview1.Selected.Level=0 then
  begin
    if treeview1.Selected.Count=0 then
    //添加工作組名
    getgroupname();
  end;
  if treeview1.Selected.Level=1 then
  begin
    if treeview1.Selected.Count=0 then
    //添加計算機名
    getcomputername;
  end;
  if treeview1.Selected.Level=2 then
  bitbtn1.Enabled:=true
  else
  bitbtn1.Enabled:=false;
end;
//取得計算機名
procedure Tfm_netlist.getcomputername;
var
  NetResource: TNetResource;
  Buf: Pointer;
  Count, BufSize, Res: DWord;
  Ind: Integer;
  lphEnum: THandle;
  Temp: TNetResourceArray;
  groupname: string;
  my_node: Ttreenode;
  my_node_2: Ttreenode;
begin
  screen.Cursor := crHourGlass;
  statusbar.panels[0].text := '正在列舉組名...，請稍侯';
  statusbar.refresh;
  my_node := treeview1.Selected;
  groupname := treeview1.Selected.Text;
  FillChar(NetResource, SizeOf(NetResource), 0); //初始化網絡層次信息
  NetResource.lpRemoteName := @GroupName[1]; //指定工作組名稱
  NetResource.dwDisplayType := RESOURCEDISPLAYTYPE_SERVER; //類型為服務器（工作組）
  NetResource.dwUsage := RESOURCEUSAGE_CONTAINER;
  NetResource.dwScope := RESOURCETYPE_DISK; //列舉文件資源信息
  //獲取指定工作組的網絡資源句柄
  Res := WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_DISK,
    RESOURCEUSAGE_CONTAINER, @NetResource, lphEnum);
  if Res <> NO_ERROR then Exit; //執行失敗
  while True do //列舉指定工作組的網絡資源
  begin
    Count := $FFFFFFFF; //不限資源數目
    BufSize := 8192; //緩沖區大小設置為8K
    GetMem(Buf, BufSize); //申請內存，用于獲取工作組信息
   //獲取計算機名稱
    Res := WNetEnumResource(lphEnum, Count, Pointer(Buf), BufSize);
    if Res = ERROR_NO_MORE_ITEMS then break; //資源列舉完畢
    if (Res <> NO_ERROR) then Exit; //執行失敗
    Temp := TNetResourceArray(Buf);
    for Ind := 0 to Count - 1 do //列舉工作組的計算機名稱
    begin
      //獲取工作組的計算機名稱，+2表示刪除"\\"，如\\wangfajun=>wangfajun
      my_node_2 := treeview1.Items.AddChild(my_node, Temp^.lpRemoteName + 2);
      my_node_2.imageindex := 2;
      Inc(Temp);
    end;
  end;
  Res := WNetCloseEnum(lphEnum); //關閉一次列舉
  if Res <> NO_ERROR then exit; //執行失敗
  FreeMem(Buf);
  screen.Cursor := crDefault;
  statusbar.panels[0].text := '';
  statusbar.refresh;
end;

//獲取組名
procedure Tfm_NetList.GetGroupName;
var
  NetResource: TNetResource;
  Buf: Pointer;
  Count, BufSize, Res: DWORD;
  lphEnum: THandle;
  p: TNetResourceArray;
  i, j: SmallInt;
  NetworkTypeList: TList;
  my_node_1: Ttreenode;
begin
  statusbar.panels[0].text := '正在列舉域名...，請稍侯';
  statusbar.refresh;
  screen.Cursor := crHourGlass;
  NetworkTypeList := TList.Create;
  //獲取整個網絡中的文件資源的句柄，lphEnum為返回名柄
  Res := WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_DISK,
    RESOURCEUSAGE_CONTAINER, nil, lphEnum);
  if Res <> NO_ERROR then exit; //執行失敗
  //獲取整個網絡中的網絡類型信息
  Count := $FFFFFFFF; //不限資源數目
  BufSize := 8192; //緩沖區大小設置為8K
  GetMem(Buf, BufSize); //申請內存，用于獲取工作組信息
  Res := WNetEnumResource(lphEnum, Count, Pointer(Buf), BufSize);
  if (Res = ERROR_NO_MORE_ITEMS) //資源列舉完畢
    or (Res <> NO_ERROR) //執行失敗
    then Exit;
  P := TNetResourceArray(Buf);
  for I := 0 to Count - 1 do //記錄各個網絡類型的信息
  begin
    NetworkTypeList.Add(p);
    Inc(P);
  end;
  //WNetCloseEnum關閉一個列舉句柄
  Res := WNetCloseEnum(lphEnum); //關閉一次列舉
  if Res <> NO_ERROR then exit;

  for J := 0 to NetworkTypeList.Count - 1 do //列出各個網絡類型中的所有工作組名稱
  begin //列出一個網絡類型中的所有工作組名稱
    NetResource := TNetResource(NetworkTypeList.Items[J]^); //網絡類型信息
  //獲取某個網絡類型的文件資源的句柄，NetResource為網絡類型信息，lphEnum為返回名柄
    Res := WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_DISK,
      RESOURCEUSAGE_CONTAINER, @NetResource, lphEnum);
    if Res <> NO_ERROR then break; //執行失敗

    while true do //列舉一個網絡類型的所有工作組的信息
    begin
      Count := $FFFFFFFF; //不限資源數目
      BufSize := 8192; //緩沖區大小設置為8K
      GetMem(Buf, BufSize); //申請內存，用于獲取工作組信息
              //獲取一個網絡類型的文件資源信息，
      Res := WNetEnumResource(lphEnum, Count, Pointer(Buf), BufSize);
      if (Res = ERROR_NO_MORE_ITEMS) //資源列舉完畢
        or (Res <> NO_ERROR) //執行失敗
        then break;
      P := TNetResourceArray(Buf);
      for I := 0 to Count - 1 do //列舉各個工作組的信息
      begin
        my_node_1 := treeview1.Items.addchild(node, StrPAS(P^.lpRemoteName)); //取得一個工作組的名稱
        my_node_1.imageindex := 1;
        Inc(P);
      end;
    end;
    Res := WNetCloseEnum(lphEnum); //關閉一次列舉
    if Res <> NO_ERROR then break; //執行失敗
  end;
  FreeMem(Buf);
  NetworkTypeList.Destroy;
  screen.Cursor := crDefault;
  statusbar.panels[0].text := '';
  statusbar.refresh;
end;

procedure TFm_netlist.BitBtn1Click(Sender: TObject);
begin
  aComputerName := treeview1.Selected.Text;
  close;
end;

procedure TFm_netlist.BitBtn2Click(Sender: TObject);
begin
close
end;

end.