基于winsock的简单客户服务器通信程序

#include <winsock2.h> 
#include <stdio.h> 
#include <iostream.h>
#include <string.h>
#pragma comment( lib, "WS2_32.lib" )

//全局参数定义
char host_name[256];
const char *pszAddr;//本地计算机名和计算机IP地址

char other_name[256];
char *ipAddress;//对方计算机名和计算机IP地址

int GetLocal() 
{ 
    
   int WSA_return; 
   WSADATA WSAData; 
   HOSTENT *host_entry; 
   WORD  wVersionRequested; 

   wVersionRequested = MAKEWORD( 2, 0 );  
   WSA_return=WSAStartup(wVersionRequested,&WSAData); //进行WSAStartup函数调用 

   if(WSA_return==0) 
   { 
       gethostname(host_name,sizeof(host_name));  //获取本地主机名称 
       host_entry=gethostbyname(host_name);   //从主机名数据库中得到对应的“主机” 

          for( int i = 0; host_entry!= NULL && host_entry->h_addr_list[i]!= NULL; i++ ) 
                //循环得出本地机器所有IP地址 
          { 
                pszAddr=inet_ntoa (*(struct in_addr *)host_entry->h_addr_list[i]);
    cout<<"本地计算机名:"<<host_name<<endl;
    cout<<"本地IP地址:"<<pszAddr<<endl;
                //MessageBox(NULL,pszAddr,host_name,MB_OK); 
          } 

   } 
   else 
   { 
          //错误提示 
       printf("ERROR\n"); 
   } 

WSACleanup(); 
return 0; 
}
int GetOther(const char*ipAddress)
{
 unsigned long addr;
 addr=inet_addr(ipAddress);
 struct hostent FAR * lpHostEnt=gethostbyaddr((char *)&addr,4,AF_INET);
 if(lpHostEnt==NULL)
 {
  //产生错误
  cout<<"获取对方计算机名失败!"<<endl;
  //MessageBox(NULL,"连接不上!","警告!",MB_OK);//应该取得其错误
  return -1;
 }
 strcpy(other_name,lpHostEnt->h_name);
 return 0;
}

//服务器接收进程
void server()
{
 WSADATA wsaData;
 SOCKET listeningSocket;
 SOCKET newConnection;
 SOCKADDR_IN serverAddr;
 SOCKADDR_IN clientAddr;
 int clientAddrLen=sizeof(serverAddr);
 int port=5150;
 char revbuff[1024];
 int ret,nLeft,idx;
 nLeft=512;
 idx=0;
 GetLocal();
 cout<<endl;

 WSAStartup(MAKEWORD(2,2),&wsaData);
 listeningSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
 if(listeningSocket==0)
 {
  cout<<"Socket()Error!";
  return;
 }
 //出错处理
 cout<<"Socket()Successed!"<<endl;
 serverAddr.sin_family=AF_INET;
 serverAddr.sin_port=htons(port);
 serverAddr.sin_addr.s_addr=htonl(INADDR_ANY);
 bind(listeningSocket,(SOCKADDR*)&serverAddr,sizeof(serverAddr));
 cout<<"bind successed!"<<endl;
 
   listen(listeningSocket,5);
 newConnection=accept(listeningSocket,(SOCKADDR*)&clientAddr,&clientAddrLen);
 ipAddress=inet_ntoa (clientAddr.sin_addr);
 cout<<"接入的IP地址为:"<<ipAddress<<endl;
 GetOther(ipAddress);
 cout<<"接入计算机名为:"<<other_name<<endl;
 //MessageBox(NULL,"accept successed!","成功!",MB_OK);
 while(nLeft>0)
 {
  ret=recv(newConnection,&revbuff[idx],nLeft,0);
  if(ret==SOCKET_ERROR)
  {
   cout<<"SOCKET_ERROR!"<<endl;
  }
  idx+=ret;
  nLeft-=ret;
 }
 cout<<"Message:";
 cout<<revbuff<<endl;
 closesocket(newConnection);
 closesocket(listeningSocket);
 WSACleanup();
}
//客户机发送进程
void client()
{
 WSADATA wsaData;
 SOCKET s;
 SOCKADDR_IN serverAddr;
 int port=5150;
 char sendbuff[1024];
 int nBytes=1024;
 int nLeft,idx;
 nLeft=nBytes;
 idx=0;
 int ret;
 GetLocal();
 cout<<endl;

 cout<<"please input the message sended:";
 cin>>sendbuff;
 WSAStartup(MAKEWORD(2,2),&wsaData);
 s=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
 serverAddr.sin_family=AF_INET;
 serverAddr.sin_port=htons(port);
 serverAddr.sin_addr.s_addr=inet_addr("192.168.0.159");

 if(connect(s,(SOCKADDR *)&serverAddr,sizeof(serverAddr)))cout<<"connect failed!"<<endl;
 else cout<<"connect success!"<<endl;
 while(nLeft>0)
 {
  ret=send(s,&sendbuff[idx],nLeft,0);
  if(ret==SOCKET_ERROR)
  {
   cout<<"SOCKET_ERROR!"<<endl;
  }
  idx+=ret;
  nLeft-=ret;
 }
 cout<<"send successed!"<<endl;
}

void all()
{

}
void main()
{
 int type;
 cout<<"请选择角色:(1.服务器 2.客户机 3.两者)"<<endl;
 cin>>type;
 if(type==1)
 {
  server();
 }
 else if(type==2)
 {
  client();
 }
 else if(type==3)
 {

 }
}