thread_share.cpp

VC++编写的类似Sniffer的网络数据包监听程序

//---------------------------------------------------------------------------

#include <vcl.h>
#pragma hdrstop

#include "thread_share.h"
#include "main.h"
#pragma package(smart_init)
#include "syncobjs.hpp"
extern int share_thread_no;//share线程数量
extern int share_ip_no;//share工作的位置指示
extern int ping_thread_no;
extern TCriticalSection *ping_CriticalSection;//全局临界变量
extern boolean thread_run;//
extern boolean lookup_user;//查找用户信息
extern boolean lookup_system;//查找系统信息
extern boolean lookup_Services;//查找系统信息
extern boolean do_select;//只对选择的工作

void __fastcall Thread_share::ThreadDone_share(TObject * Sender)
{
  FNetConnection->Free();
  list->Free();
  share_thread_no--; //线程总数减一
  main_f->Edit16->Text=IntToStr(share_thread_no);
  if ((share_thread_no==0)&&(ping_thread_no==0))
   {
    main_f->Edit8->Text="查找共享完成";
    main_f->BitBtn4->Enabled=!main_f->share_check->Checked;
    main_f->BitBtn5->Enabled=!main_f->share_check->Checked;
    main_f->Bar->Position=0;
    do_select=false;
   }
}

__fastcall Thread_share::Thread_share(bool CreateSuspended)
        : TThread(CreateSuspended)
{
   FNetConnection=new TNetConnection(main_f);
   list=new TStringList();

   FNetConnection->OnEchoClose=NetCon_EchoClose;
   FNetConnection->OnEchoOpen=NetCon_Echoopen;
   FNetConnection->OnEchoShareEnum=NetCon_EchoShareEnum;
   FNetConnection->OnOpenSucceed=Netopensucceed;
   FNetConnection->GetInfo=lookup_system;//找系统信息
   OnTerminate =ThreadDone_share;
   FreeOnTerminate = True;
   share_thread_no++;//线程加1
   main_f->Edit16->Text=IntToStr(share_thread_no);
}

void __fastcall Thread_share::Execute()
{
   ping_CriticalSection->Enter();//进入临界状态
   int no=share_ip_no;
      if (no>=main_f->tree_ping->Items->Count)
        {
         ping_CriticalSection->Leave();//析放临界状态
         return ;
        }
       else
       if (do_select)
        {
        if (main_f->tree_ping->Selected==NULL)
          FNetConnection->Address=main_f->tree_ping->Items->Item[0]->Text;
          else
          FNetConnection->Address=main_f->tree_ping->Selected->Text;
        main_f->Edit8->Text=FNetConnection->Address;
        share_ip_no=main_f->tree_ping->Items->Count;//设为最大
        }
        else
       {
        share_ip_no++;
        FNetConnection->Address=main_f->tree_ping->Items->Item[no]->Text;
        main_f->Edit8->Text=FNetConnection->Address;
        }
   ping_CriticalSection->Leave();//析放临界状态

   FNetConnection->open();
}

void __fastcall Thread_share::NetCon_EchoClose(TObject *Sender,
      TObject *IConnection, int Error)
{
 Synchronize(set_system);
 if (lookup_Services)
   {
    FNetConnection->EnumServices(list);
    if (list->Count>1) Synchronize(set_svr);//找服务
   }

  ping_CriticalSection->Enter();//进入临界状态
   int no=share_ip_no;
      if (no>=main_f->tree_ping->Items->Count)
        {
          ping_CriticalSection->Leave();//析放临界状态
          return;
        }
       else{
        share_ip_no++;
        FNetConnection->Address=main_f->tree_ping->Items->Item[no]->Text;
        main_f->Edit8->Text=FNetConnection->Address;
        }
   ping_CriticalSection->Leave();//析放临界状态

 if (thread_run) FNetConnection->open();
}
//---------------------------------------------------------------------------
void __fastcall Thread_share::NetCon_Echoopen(TObject *Sender,
      TObject *IConnection, int Error)
{
  if (Error!=0) NetCon_EchoClose(NULL,NULL,0);
}

void __fastcall Thread_share::NetCon_EchoShareEnum(TObject *Sender,
      TObject *IConnection, int Error)
{
  if ((Error==0)||(Error==234)) Synchronize(set_list);
}

void __fastcall Thread_share::Netopensucceed(TObject *Sender)
{
 if (lookup_user)
   {
     FNetConnection->NetQueryDisplayInformation(1,list);
     if (list->Count>1) Synchronize(set_user);//找用户
   }
}

void __fastcall Thread_share::set_list()
{
  int i;
  TTreeNode *temp_node;
  temp_node=main_f->tree_share->Items->Add(NULL,FNetConnection->list->Strings[0]);

  for (i=1;i<FNetConnection->list->Count;i++)
   {
    main_f->tree_share->Items->AddChild(temp_node,FNetConnection->list->Strings[i]);
    main_f->Edit10->Text=IntToStr(StrToInt(main_f->Edit10->Text)+1);
   }
}

void __fastcall Thread_share::set_svr()
{
  int i;
  TTreeNode *temp_node;
  temp_node=main_f->share_svr->Items->Add(NULL,list->Strings[0]);

  for (i=1;i<list->Count;i++)
   {
    main_f->share_svr->Items->AddChild(temp_node,list->Strings[i]);
//    main_f->Edit10->Text=IntToStr(StrToInt(main_f->Edit10->Text)+1);
   }
}

void __fastcall Thread_share::set_user()
{
  int i;
  TTreeNode *temp_node;
  temp_node=main_f->share_user->Items->Add(NULL,list->Strings[0]);

  for (i=1;i<list->Count;i++)
   {
    main_f->share_user->Items->AddChild(temp_node,list->Strings[i]);
//    main_f->Edit10->Text=IntToStr(StrToInt(main_f->Edit10->Text)+1);
   }
}
void __fastcall Thread_share::set_system()
{
  int i;
  TTreeNode *temp_node;
 if ((lookup_system)&&(FNetConnection->list_info->Count>1))
   {
    temp_node=main_f->share_system->Items->Add(NULL,FNetConnection->list_info->Strings[0]);
    for (i=1;i<FNetConnection->list_info->Count;i++)
     {
      main_f->share_system->Items->AddChild(temp_node,FNetConnection->list_info->Strings[i]);
     }
    FNetConnection->list_info->Clear();
   }
}

//---------------------------------------------------------------------------