在C++ Builder中用socket api來寫網路通訊程式（同時支援TCP和UDP協議） (轉)

在C++ Builder中用socket api來寫網路通訊程式（同時支援TCP和UDP協議） (轉)[@more@]

在7月4日看完sockcomp.pas後，我決定用socket 來寫一個客戶端和並且同時支援TCP,UDP，於是我就去做，現將程式碼貼出來（已透過）
Socket api Client:
#ifndef UDPClientH
#define UDPClientH
#include
#include
#include
#include
#include
#include "CCEdit.h"
#define WM_SOCK WM_USER+100
class TLANFo: public TForm
{
__published: // -managed Components
  TEdit *Port;
  TLabel *Label1;
  TLabel *Label2;
  TComboBox *Prot;
  TButton *Button1;
  TLabel *Label3;
  TEdit *Addr;
  TCCEdit *TxtEdit;
  void __fastcall FormCreate(T *Sender);
  void __fastcall Button1Click(TObject *Sender);
  void __fastcall FormDestroy(TObject *Sender);
private: // User declarations
  void __fastcall OnRecv(TMessage &Message);
public: // User declarations
  __fastcall TLANForm(TComponent* Owner);
  BEGIN_MESSAGE_MAP
  VCL_MESSAGE_HANDLER(WM_SOCK,TMessage,OnRecv);
  END_MESSAGE_MAP(TForm);
};
extern PACKAGE TLANForm *LANForm;
#endif
.cpp File
#include
#pragma hdrstop
#include "UDPClient.h"
#include "WinSock.h"
#pragma package(smart_init)
#pragma link "CCEdit"
#pragma re "*.dfm"
TLANForm *LANForm;
enum PROTO {TCP=0,UDP=1};
SOCKET m_Socket=INVALID_SOCKET;
PROTO m_Protocol=TCP;
__fastcall TLANForm::TLANForm(TComponent* Owner)
  : TForm(Owner)
{

}
void __fastcall TLANForm::FormCreate(TObject *Sender)
{
  ::SendMessage(Prot->Handle,CB_SETCURSEL,0,0);
}
void __fastcall TLANForm::OnRecv(TMessage &Message)
{
  char buf[4096];
  int nLen;
  struct sockaddr_in from;
  int nLength=sizeof(struct sockaddr_in);
  switch(WSAGETEVENT(Message.LParam))
  {
  case FD_READ:
  switch(m_Protocol)
  {
  case TCP:
  nLen=recv(m_Socket,buf,4096,0);
  if(nLen>0){
  buf[nLen]='';
  TxtEdit->Text="Received Length:"+String(nLen)+" "+StrPas(buf);
  }
  break;
  case UDP:
  nLen=recvfrom(m_Socket,buf,4096,0,(struct sockaddr*)&from,&nLength);
  if(nLen>0){
  buf[nLen]='';
  TxtEdit->Text="Received Length:"+String(nLen)+" "+StrPas(buf);
  }
  break;
  }
  break;
  case FD_CLOSE:
  closesocket(m_Socket);
  break;
  }
}

void __fastcall TLANForm::Button1Click(TObject *Sender)
{
  char szTmp[256],buf[4096];
  int nSize=0;
  UINT m_Port;
  AnsiString addr;
  addr=Addr->Text.Trim();
  if(addr.IsEmpty()){
  ::MessageBox(0,"Please enter the server IP!","Error",MB_OK+MB_ICONERROR);
  return;
  }
  unsigned long nAddr=_addr(addr.c_str());
  if(nAddr==INADDR_NONE){
  ::MessageBox(0,"Bad Internet IP!","Error",MB_OK+MB_ICONERROR);
  return;}

  try
  {
  m_Port=Port->Text.ToInt();
  }
  catch(Exception &e)
  {
  ::MessageBox(0,e.Message.c_str(),"Error",MB_OK+MB_ICONERROR);
  return;
  }
  switch(Prot->ItemIndex)
  {
  case 0:
  m_Protocol=TCP;
  break;
  case 1:
  m_Protocol=UDP;
  break;
  }
  if(TxtEdit->Text.IsEmpty()){
  ::MessageBox(0,"Please enter the text you want to send!","Error",MB_OK+MB_ICONERROR);
  return;}

  Winsocket
  WSAData wsaData;
  ::ZeroMemory(&wsaData,sizeof(WSAData));
  version=MAKEWORD(2,0);
  if(::Wtartup(version,&wsaData)){
  sprintf(szTmp,"Failed to initial winsock enviroment!,error no:%d",::WSAGetLastError());
  return;}

  the active connection
  char ComputerName[255];
  gethostname(ComputerName,255);
  struct hostent* he=gethostbyname(ComputerName);
  if(!he){
  sprintf(szTmp,"Failed to get information to host!","Error",MB_OK+MB_ICONERROR);
  ::WSACleanup();
  return;
  }
  new socket
  m_Socket=INVALID_SOCKET;
  switch(m_Protocol)
  {
  case TCP:
  m_Socket=socket(AF_INET,SOCK_STREAM,0);
  break;
  case UDP:
  m_Socket=socket(AF_INET,SOCK_DGRAM,0);
  break;
  }
  if(m_Socket==INVALID_SOCKET){
  sprintf(szTmp,"Failed to create a new socket!,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  ::WSACleanup();
  return;
  }
  socket
  struct sockaddr_in client;
  unsigned long nClient;
  memcpy(&nClient,he->h_addr_list[0],sizeof(int));
  if(nClient==INADDR_NONE){
  sprintf(szTmp,"Failed to obtain the local machine's IP!","Error",MB_OK+MB_ICONERROR);
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  closesocket(m_Socket);
  ::WSACleanup();
  return;
  }
  client.sin_family=AF_INET;
  client.sin_port=0;
  client.sin_addr.S_un.S_addr=(int)nClient;
  if(bind(m_Socket,(struct sockaddr*)&client,sizeof(struct sockaddr))){
  sprintf(szTmp,"Failed to bind socket!","Error",MB_OK+MB_ICONERROR);
  closesocket(m_Socket);
  ::WSACleanup();
  return;}
  socket
  struct sockaddr_in To;
  To.sin_family=AF_INET;
  To.sin_port=htons(m_Port);
  To.sin_addr.S_un.S_addr=(int)nAddr;
  fd_set FDSET;
  FD_ZERO(&FDSET);
  FD_SET(m_Socket,&FDSET);

  if(m_Protocol==TCP){
  if(connect(m_Socket,(struct sockaddr*)&To,sizeof(struct sockaddr))){
  sprintf(szTmp,"Failed to connect the object!,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  closesocket(m_Socket);
  ::WSACleanup();
  return;
  }
  int nError=select(1,0,&FDSET,0,0);
  if(nError<=0){
  sprintf(szTmp,"Failed to select socket!,error no:%d",::WSAGetLastError());
  closesocket(m_Socket);
  ::WSACleanup();
  return;}
  }
  data
  int nLen=TxtEdit->Text.Length();
  if(nLen>4096){
  sprintf(szTmp,"The buffer is too size to send,it shoud not be more than 4096 bytes!");
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  closesocket(m_Socket);
  ::WSACleanup();
  return;
  }
  strncpy(buf,TxtEdit->Text.c_str(),nLen);
  switch(m_Protocol)
  {
  case TCP:
  nSize=send(m_Socket,buf,nLen,0);
  (nSize);
  break;
  case UDP:
  nSize=sendto(m_Socket,buf,nLen,0,(struct sockaddr*)&To,sizeof(struct sockaddr));
  (nSize);
  break;

  }
  if(::WSAAsyncSelect(m_Socket,Handle,WM_SOCK,FD_READ|FD_CLOSE)){
  sprintf(szTmp,"Failed to register socket event!,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  closesocket(m_Socket);
  ::WSACleanup();
  return;}
}
void __fastcall TLANForm::FormDestroy(TObject *Sender)
{
  closesocket(m_Socket);
  ::WSACleanup();
}

Socket api Server:
.h File
#ifndef UDPServerH
#define UDPServerH
#include
#include
#include
#include
#include
#include "WinSock.h"
/*#define WM_SOCKET WM_USER+1000
#define INITSOCKETSUCCESS 0
#define INITSOCKETFAILURE 1
#define CREATELISTENSOCKETSUCCESS 2
#define CREATELISTENSOCKETFAILURE 3
#define SETLISTENSOCKETSUCCESS 4
#define SETLISTENSOCKETFAILURE 5
#define BINDLISTENSOCKETSUCCESS 6
#define BINDLISTENSOCKETFAILURE 7
#define LISTENSOCKETSUCCESS 8
#define LISTENSOCKETFAILURE 9
#define ACCEPTSOCKETSUCCESS 10
#define ACCEPTSOCKETFAILURE 11
*/
class TPSTNForm : public TForm
{
__published: // IDE-managed Components
  TEdit *Port;
  TLabel *Label1;
  TMemo *Memo1;
  TButton *Button1;
  TButton *Button2;
  TLabel *Label2;
  TComboBox *Prot;
  void __fastcall Button1Click(TObject *Sender);
  void __fastcall Button2Click(TObject *Sender);
  void __fastcall FormCreate(TObject *Sender);
  void __fastcall FormDestroy(TObject *Sender);
private: // User declarations
public: // User declarations
  __fastcall TPSTNForm(TComponent* Owner);
};
enum PROTO {TCP,UDP};
class TCommunication:public TThread  Thread
{
private:
  SOCKET m_AcceptSocket;
  char szTmp[256];//ERROR MESSAGE
public:
  __fastcall TCommunication(SOCKET m_Socket,bool CreateSuspended);
  __fastcall ~TCommunication();
protected:
  virtual void __fastcall Execute();
};
class TListenThread:public TThread Thread
{
private:
  WSAData wsaData;
  struct sockaddr_in server;
  fd_set FDS;
  UINT m_Port;
  PROTO m_Protocol;
  char szTmp[256];//Error Message
public:
  SOCKET m_Socket;
  void __fastcall DoError();
  void __fastcall InitSocket();
  void __fastcall CreateListenSocket();
  void __fastcall SetListenSocket();
  void __fastcall BindListenSocket();
  void __fastcall ListenSocket();

public:
  __fastcall TListenThread(PROTO m_ProtocolA,UINT m_PortA,bool CreateSuspended);
  virtual __fastcall ~TListenThread();
protected:
  virtual void __fastcall Execute();
};
extern PACKAGE TPSTNForm *PSTNForm;
#endif
.cpp File
#include
#pragma hdrstop
#include "UDPServer.h"
#pragma package(smart_init)
#pragma resource "*.dfm"
TPSTNForm *PSTNForm;
TListenThread *pThread=0;
CLASS TCommunication For TCP****************************************************
__fastcall TCommunication::TCommunication(SOCKET m_Socket,bool CreateSuspended):TThread(FALSE)
{
  m_AcceptSocket=m_Socket;
  szTmp[0]='';
  FreeOnTenate=true;
}
__fastcall TCommunication::~TCommunication()
{
  // closesocket(m_AcceptSocket);
}
void __fastcall TCommunication::Execute()
{
  char buf[4096];
  int nSize=0;
  nSize=recv(m_AcceptSocket,(char FAR*)buf,4096,0);
  if(nSize>0)
  {
  buf[nSize]='';
 
  PSTNForm->Memo1->Lines->Add("Received Length:"+String(nSize));
  PSTNForm->Memo1->Lines->Add("Received:"+StrPas(buf));
 
  ::Sleep(100);
  send(m_AcceptSocket,buf,nSize,0);
  }

}
CLASS TListenThread*****************************************************
__fastcall TListenThread::TListenThread(PROTO m_ProtocolA,UINT m_PortA,bool CreateSuspended):TThread(FALSE)
{
  m_Socket=INVALID_SOCKET;
  m_Port=m_PortA;
  m_Protocol=m_ProtocolA;
  szTmp[0]='';
  ::ZeroMemory(&wsaData,sizeof(WSAData));
  ::ZeroMemory(&server,sizeof(struct sockaddr_in));
  FreeOnTerminate=TRUE;//Automatically delete while terminating.
}
__fastcall TListenThread::~TListenThread()
{
  closesocket(m_Socket);
  ::WSACleanup();
  m_Socket=INVALID_SOCKET;
  m_Port=0;
  m_Protocol=TCP;
  szTmp[0]='';
  ::ZeroMemory(&wsaData,sizeof(WSAData));
  ::ZeroMemory(&server,sizeof(struct sockaddr_in));
}
void __fastcall TListenThread::DoError()
{
  if(m_Socket!=INVALID_SOCKET) closesocket(m_Socket);
  WSACleanup();
  return;
}
void __fastcall TListenThread::InitSocket()
{
  WORD version=MAKEWORD(2,0);
  if(::WSAStartup(version,&wsaData)){
  sprintf(szTmp,"Failed to intiailize socket,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  return;
  }
}
void __fastcall TListenThread::CreateListenSocket()
{
  switch(m_Protocol)
  {
  case UDP:
  m_Socket=socket(AF_INET,SOCK_DGRAM,0);
  break;
  case TCP:
  m_Socket=socket(AF_INET,SOCK_STREAM,0);
  break;
  default:
  sprintf(szTmp,"Error protocol!");
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  break;
  }
  if(m_Socket==INVALID_SOCKET){
  sprintf(szTmp,"Failed to create socket!");
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  return;
  }
}
void __fastcall TListenThread::SetListenSocket()
{
  server.sin_family=AF_INET;
  server.sin_port=htons(m_Port);
  server.sin_addr.S_un.S_addr=INADDR_ANY;
  int NewOpenType=SO_SYNCHRONOUS_NONALERT;
  if(setsockopt(INVALID_SOCKET,SOL_SOCKET,SO_OPENTYPE,(char*)&NewOpenType,4)){
  sprintf(szTmp,"Set socket option error,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  return;
  }
}
void __fastcall TListenThread::BindListenSocket()
{
  if(bind(m_Socket,(sockaddr*)&server,sizeof(struct sockaddr_in))){
  sprintf(szTmp,"Failed to bind socket,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  return;
  }

}
void __fastcall TListenThread::ListenSocket()
{
  if(listen(m_Socket,SOMAXCONN)){
  sprintf(szTmp,"listen error,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  return;
  }
  whether there is any connection
  FD_ZERO(&FDS);
  FD_SET(m_Socket,&FDS);
}
void __fastcall TListenThread::Execute()
{
  char buf[4096];
  struct sockaddr_in from; UDP
  int nLen=sizeof(from),nSize=0;  UDP
  InitSocket();
  CreateListenSocket();
  SetListenSocket();
  BindListenSocket();
  if(m_Protocol==UDP){
  while(!Terminated){
  int nSize=recvfrom(m_Socket,buf,4096,0,(struct sockaddr*)&from,&nLen);
  if(nSize>0){
  buf[nSize]='';
  PSTNForm->Memo1->Lines->Add("Received Length:"+String(nSize));
  PSTNForm->Memo1->Lines->Add("Received:"+StrPas(buf));
  ::Sleep(100);
  sendto(m_Socket,buf,nSize,0,(struct sockaddr*)&from,sizeof(struct sockaddr_in));
  }
  else return;

  }
  }
  ListenSocket();
  struct sockaddr_in client;
  int nLength=sizeof(struct sockaddr_in);
  while(!Terminated){
  int nError=select(1,&FDS,0,0,0);
  if(nError<=0) Terminate();
  SOCKET m_AcceptSocket=accept(m_Socket,(struct sockaddr*)&client,&nLength);
  if(m_AcceptSocket==INVALID_SOCKET){
  sprintf(szTmp,"Failed to execute accept,error no:%d",::WSAGetLastError());
  ::MessageBox(0,szTmp,"Error",MB_OK+MB_ICONERROR);
  DoError();
  Terminate();
  return;
  }
  TCommunication *pCThread=new TCommunication(m_AcceptSocket,FALSE);
  pCThread->Terminate();
  pCThread->WaitFor();
  }
}
PSTNForm*********************************************//
__fastcall TPSTNForm::TPSTNForm(TComponent* Owner)
  : TForm(Owner)
{
}
void __fastcall TPSTNForm::Button1Click(TObject *Sender)
{
  Close();
}
void __fastcall TPSTNForm::Button2Click(TObject *Sender)
{
  if(pThread){
  pThread->Suspend();
  pThread->Terminate();
  delete pThread;
  pThread=0;
  }
  UINT m_Port;
  try
  {
  m_Port=Port->Text.ToInt();
  }
  catch(Exception &e)
  {
  ::MessageBox(0,e.Message.c_str(),"Error",MB_OK+MB_ICONERROR);
  return;
  }
  PROTO m_Protocol;
  switch(Prot->ItemIndex)
  {
  case 0:
  m_Protocol=TCP;
  break;
  case 1:
  m_Protocol=UDP;
  break;
  default:
  break;
  }
  pThread=new TListenThread(m_Protocol,m_Port,FALSE);
  >Terminate();
}
void __fastcall TPSTNForm::FormCreate(TObject *Sender)
{
  ::SendMessage(Prot->Handle,CB_SETCURSEL,0,1);
}

void __fastcall TPSTNForm::FormDestroy(TObject *Sender)
{
  if(pThread){
  pThread->Suspend();
  pThread->Terminate();}
}
上面的程式碼，各位可根據自已的需要和針對自已的應用，在資料處理方面加以改進就可以了。

再次宣告，上述程式碼中出現的file:字首這是csdn文件編輯器自動加的，各位凡是見到file:字首就表示它是註釋部分，特此宣告，以免誤解。