TCP Client

#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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值