#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)
{
}
}