#include <stdio.h>
#include <string.h>
#include <winsock2.h>
#pragma comment(lib,"Ws2_32.lib")
#define BUFFER_SIZE 50
DWORD WINAPI RecvThread(LPVOID lpParam)
{
printf("start to Recv\n");
LPSTR lpRecv = (LPSTR)HeapAlloc(GetProcessHeap(), 0, BUFFER_SIZE);
SOCKET sock = (SOCKET)lpParam;
int nRecv = 0;
BOOL bRead = TRUE;
while(bRead)
{
while(1)
{
nRecv = recv(sock, lpRecv, BUFFER_SIZE, 0);
if (nRecv==BUFFER_SIZE)
{
printf("%s", lpRecv);
}
else if (nRecv<BUFFER_SIZE && nRecv>0)
{
printf("%s\n", lpRecv);
break;
}
else if(nRecv==0)
{
printf("\nReceive Over.");
break;
}
else
{
int error = WSAGetLastError();
switch(error)
{
case WSAECONNRESET:
printf("\nerror:%d, %s\n", GetLastError(), "socket is closed at remote");
break;
case WSAECONNABORTED:
printf("\nerror:%d, %s\n", GetLastError(), "socket is closed at local");
break;
}
closesocket(sock);
bRead = FALSE;
break;
}
}
}
HeapFree(GetProcessHeap(),0,lpRecv);
closesocket(sock);
return 0;
}
DWORD WINAPI SendThread(LPVOID lpParam)
{
printf("start to send\n");
SOCKET sock = (SOCKET)lpParam;
char buf[1024] = {0};
while(1)
{
printf("Client Input:");
scanf("%s", buf);
if (send(sock, buf, sizeof(buf), 0)!=sizeof(buf))
{
printf("send error: %d\n", WSAGetLastError());
break;
}
}
closesocket(sock);
return 0;
}
int main()
{
HANDLE child[2];
int err;
SOCKET sClient;
char ip[20] = "127.0.0.1"; /*IP地址*/
char buf[BUFFER_SIZE]; /*数据发送缓冲区*/
int port = 10000; /*端口*/
WSADATA wsaD;
struct sockaddr_in servaddr; /*存放服务器端网络地址的结构*/
// printf("Please input target IP and Port(eg:127.0.0.1 5555):/n");
// scanf("%s%hu",ip,&port);
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = inet_addr(ip);
servaddr.sin_port = htons(port);
/*Winsock初始化*/
err = WSAStartup(MAKEWORD(1,1),&wsaD);
/*创建Socket*/
if((sClient = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP)) == SOCKET_ERROR){
err = WSAGetLastError();
printf("Create socket error %d",err);
return -1;
}
/*连接到服务器*/
if(connect(sClient,(struct sockaddr *) &servaddr,sizeof(servaddr)) == SOCKET_ERROR){
err = WSAGetLastError();
printf("Failed to connect to server!%d\n",err);
return -1;
}
printf("Connected to server %s\n",inet_ntoa(servaddr.sin_addr));
//
child[0]=CreateThread(NULL, 0, RecvThread, (LPVOID)sClient, 0, NULL);
if (!child[0] )
{
printf("Create recv thread error: %d\n", WSAGetLastError());
}
if (! ( child[1]=CreateThread(NULL,
0,
SendThread,
(LPVOID)sClient,
0,
NULL) ) )
{
printf("Create send thread error: %d\n", WSAGetLastError());
}
//
// while(1)
// {
// gets(buf);
// if (strcmp(buf,"quit")==0 || strcmp(buf,"exit")==0)
// break;
// send(sClient,buf,strlen(buf)+1,0) ;/*发送数据*/
// }
// WaitForSingleObject(child[0], INFINITE);
WaitForMultipleObjects(2, child, TRUE, INFINITE);
closesocket(sClient);
WSACleanup();
return 0;
}