BCB 用P2P实现的一个聊天例子,可传文件

//--------------------------------------------------------------------------- #include <vcl.h> #pragma hdrstop #include "ChaUnittMain.h" //--------------------------------------------------------------------------- #pragma package(smart_init) #pragma resource "*.dfm" TMainForm *MainForm; //--------------------------------------------------------------------------- __fastcall TMainForm::TMainForm(TComponent* Owner) : TForm(Owner) { } //--------------------------------------------------------------------------- void __fastcall TMainForm::FormCreate(TObject *Sender) { TWSAData aWSAData; TSockAddrIn addr; TIniFile *ini; WORD w=MAKEWORD(1,1); ini=new TIniFile(ExtractFileDir(Application->ExeName)+"\config.ini"); EdtIP->Text=ini->ReadString("Config","Server","127.0.0.1"); EdtPort->Text=IntToStr(SERVER_PORT); EdtName->Text=ini->ReadString("Config","Name",""); EdtPort->Text=ini->ReadString("Config","Port",IntToStr(SERVER_PORT)); delete ini; sendInfo.IsWorking=false; recvInfo.IsWorking=false; if(WSAStartup(w,&aWSAData)!=0) ShowMessage("WinSock Version Error~!"); sock = socket(AF_INET,SOCK_DGRAM,0); addr.sin_family = AF_INET; addr.sin_port=0; addr.sin_addr.s_addr = INADDR_ANY; if(SOCKET_ERROR==WSAAsyncSelect(sock,MainForm->Handle,WM_SOCKETCLI,FD_READ)) ShowMessage("WM_SOCKET Error"); } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnConnectClick(TObject *Sender) { TP2PLoginPack pack; int BufferSize = sizeof(pack)+1; char *Buffer = new char[ BufferSize ]; Buffer[ sizeof(pack) ]='\0'; if(EdtName->Text=="") { ShowMessage("请输入名字"); EdtName->SetFocus(); exit(0); } addrSrv.sin_addr.S_un.S_addr = inet_addr((EdtIP->Text).c_str()); addrSrv.sin_family=AF_INET; addrSrv.sin_port=htons(StrToInt(EdtPort->Text)); pack.head.Command=cmdLogin; StrPCopy(pack.body.name,EdtName->Text); memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer,BufferSize,0,(LPSOCKADDR)&addrSrv,sizeof(addrSrv)); LoginTickCount=::GetTickCount(); SetTimer(MainForm->Handle,2,2000,(TIMERPROC)&OnCheckLoginResp); delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::ListBox1Click(TObject *Sender) { TP2PUserInfoPack pack; char *Buffer = new char[ sizeof(pack)+1 ]; Buffer[ sizeof(pack) ]='\0'; if(ListBox1->ItemIndex<0)exit(0); pack.head.Command=cmdUserInfo; StrPCopy(pack.body.name1,EdtName->Text);//自己 StrPCopy(pack.body.name2,(ListBox1->Items->Strings[ListBox1->ItemIndex]));//对方 memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer,sizeof(pack)+1,0,(LPSOCKADDR)&addrSrv,sizeof(addrSrv)); delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::TimerMakeHoleTimer(TObject *Sender) { TP2PHolePack pack; char *Buffer = new char[ sizeof(pack)+1 ]; Buffer[ sizeof(pack)]='\0'; pack.head.Command=cmdHole; memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer,sizeof(pack)+1,0,(LPSOCKADDR)&addrP2P,sizeof(addrP2P)); if(recvInfo.IsWorking) { Label5->Caption=IntToStr(recvInfo.ID-recvInfo.progress)+"KB/s"; } if(sendInfo.IsWorking) { Label5->Caption=IntToStr(sendInfo.ID-sendInfo.progress)+"KB/s"; } recvInfo.progress=recvInfo.ID; sendInfo.progress=sendInfo.ID; delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnSendClick(TObject *Sender) { TP2PMessagePack pack; char *Buffer = new char[ sizeof(pack)+1 ]; Buffer[ sizeof(pack) ]='\0'; if(EdtMessage->Text=="")return; pack.head.Command=cmdMessage; StrPCopy(pack.body.name,EdtName->Text); StrPCopy(pack.body.Text,EdtMessage->Text); memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer ,sizeof(pack)+1,0,(LPSOCKADDR)&addrP2P,sizeof(addrP2P)); Memo1->Lines->Add(EdtName->Text+":"+EdtMessage->Text); EdtMessage->Clear(); delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnBrowseClick(TObject *Sender) { if(OpenDialog1->Execute())EdtFile->Text=OpenDialog1->FileName; } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnSendFileClick(TObject *Sender) { TFileStream *stream=NULL; TP2PInquireAcceptFilePack pack; char *Buffer = new char[ sizeof(pack)+1 ]; Buffer[ sizeof(pack) ]='\0'; if(FileExists(EdtFile->Text)) { stream=new TFileStream(EdtFile->Text,fmOpenRead); sendInfo.FileName=EdtFile->Text; sendInfo.FileSize=stream->Size; sendInfo.BlockCount=GetBlockCount(sendInfo.FileSize);//文件分为多少块 delete stream; pack.head.Command=cmdInquireAcceptFile; StrPCopy(pack.body.name,EdtName->Text); StrPCopy(pack.body.FileName,ExtractFileName(EdtFile->Text)); pack.body.FileSize=sendInfo.FileSize; memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer ,sizeof(pack)+1,0,(LPSOCKADDR)&addrP2P,sizeof(addrP2P)); } else ShowMessage("请确认要发送的文件"); delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::EdtMessageKeyPress(TObject *Sender, char &Key) { if(int(Key)==13)btnSendClick(Sender); } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnClearClick(TObject *Sender) { Memo1->Clear(); } //--------------------------------------------------------------------------- void __fastcall TMainForm::btnRefreshClick(TObject *Sender) { TP2PUserListPack pack; char *Buffer = new char[ sizeof(pack) ]; Buffer[ sizeof(pack)+1 ]='\0'; pack.head.Command=cmdUserList; memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer,sizeof(pack),0,(LPSOCKADDR)&addrSrv,sizeof(addrSrv)); delete[] Buffer; } //--------------------------------------------------------------------------- void __fastcall TMainForm::TimerKeepOnlineTimer(TObject *Sender) { TP2POnlinePack pack; char *Buffer = new char[ sizeof(pack)+1 ]; Buffer[ sizeof(pack)]='\0'; pack.head.Command=cmdOnline; StrPCopy(pack.body.name,EdtName->Text); memcpy( Buffer, &pack, sizeof( pack ) ); sendto(sock,Buffer,sizeof(pack)+1,0,(LPSOCKADDR)&addrSrv,sizeof(addrSrv)); delete[] Buffer; } //--------------------------------------------------------------------------- int __fastcall TMainForm::GetBlockCount(int Size) { if(Size%BlockSize==0) { return Size/BlockSize; } else { return (Size/BlockSize)+1; } } //--------------------------------------------------------------------------- void __fastcall TMainForm::WMSOCKETCLI(Messages::TMessage& msg) { TSockAddrIn addr;//接收到的主机信息 int addrlen;//地址结构长度 int i;//for循环专用 byte buffer[1500];//数据包 TP2PHead head;//包头 //i开头的是接收的消息结构，o开头的是发出的消息结构 TP2PLoginResp iP2PLoginResp;//server登录回应 TP2PUserListPack oP2PUserListPack;//请求用户列表 TStringList *users;//在线用户信息临时变量，用于分割字符 TP2PUserListResp iP2PUserListResp;//server返回的在线用户信息 TP2PUserInfoResp iP2PUserInfoResp;//server返回的指定用户信息 TP2PMakeHole iP2PMakeHole;//server要求你打洞，包含对方的主机信息 TP2PMessage iP2PMessage;//p2p之间的聊天信息 TP2PInquireAcceptFile iP2PInquireAcceptFile;//询问是否接收文件 TP2PInquireAcceptFileResp iP2PInquireAcceptFileResp;//同上，只是方向不同 PChar fileBuffer;//用来预先分配硬盘空间，来创建一个文件 TP2