Socket编程 简易测试socket TCP程序

简易的利用socket 实现TCP连接的程序,旨在简化程序结构,给初学者一个启示的作用.

       

        服务器端程序:

// SocketSrv.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <WinSock2.h>
#include <Windows.h>
#include <iostream>
using namespace std;
#pragma comment(lib,"ws2_32.lib")	//添加ws2_32.lib

int _tmain(int argc, _TCHAR* argv[])
{
	//初始化 ws2_32.lib
	WSADATA wsaData;
	WORD wVerionRequested;

	wVerionRequested = MAKEWORD(2,2);
	if (WSAStartup(wVerionRequested,&wsaData) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"WSAStarup Error","Demon",MB_OK | MB_ICONERROR);
		return -1;
	}
	cout << "Initialize ws2_32.lib...done..." << endl;


	//创建socket
	SOCKET srv_socket = socket(AF_INET,SOCK_STREAM,0);
	if (srv_socket == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"socket error","demon",MB_ICONERROR|MB_OK);
		return -1;
	}
	cout << "create socket...done..." << endl;

	//填充sockaddr_in结构体用于bind
	sockaddr_in srv_addr;
	srv_addr.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
	srv_addr.sin_family = AF_INET;
	srv_addr.sin_port = htons(6500);

	if (bind(srv_socket,(SOCKADDR*)&srv_addr,sizeof(srv_addr)) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"bind error","demon",MB_OK|MB_ICONERROR);
		return -1;
	}
	cout << "bind...done..." << endl;

	//使本地socket处于监听状态
	int backlog = 3;
	if (listen(srv_socket,backlog) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"listen error","demon",MB_ICONERROR|MB_OK);
		return -1;
	}
	cout << "listen...done..." << endl;

	//开启接受状态accept
	//存储发送方的地址和端口信息
	SOCKET from_socket;
	sockaddr_in from_addr;
	int fromlen = sizeof(SOCKADDR);
	
	if ((from_socket = accept(srv_socket,(SOCKADDR*)&from_addr,&fromlen)) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"accept error","demon",MB_OK|MB_ICONERROR);
		return -1;
	}
	cout << "accept...done..." << endl;

	//接受发送方的数据
	char recvBuf[256];
	if (recv(from_socket,recvBuf,256,0) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"recv error","demon",MB_ICONERROR|MB_OK);
		return -1;
	}
	else
		cout << recvBuf << endl;

	cout << "recv...done..." << endl;

	closesocket(srv_socket);
	closesocket(from_socket);

	WSACleanup();

	
	return 0;
}

    

        客户端程序:

// SocketClient.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <WinSock2.h>
#include <Windows.h>
#include <iostream>
using namespace std;
#pragma comment(lib,"ws2_32.lib")	//添加ws2_32.lib

#define DEMON "Demon"
#define WTYPE	MB_OK | MB_ICONERROR

int _tmain(int argc, _TCHAR* argv[])
{

	//初始化 ws2_32.lib
	WSADATA wsaData;
	WORD wVerionRequested;

	wVerionRequested = MAKEWORD(2,2);
	if (WSAStartup(wVerionRequested,&wsaData) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"WSAStartup error",DEMON,WTYPE);
		return -1;
	}
	cout << "Initialize ws2_32.lib...done..." << endl;

	//创建本地套接字
	SOCKET local_socket;
	if ((local_socket = socket(AF_INET,SOCK_STREAM,0)) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"socket error",DEMON,WTYPE);
		return -1;
	}
	cout << "create socket...done..." << endl;

	//填充sockaddr_in,用于connect
	sockaddr_in cnt_addr;
	cnt_addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
	cnt_addr.sin_family = AF_INET;
	cnt_addr.sin_port = htons(6500);

	if (connect(local_socket,(SOCKADDR*)&cnt_addr,sizeof(SOCKADDR)) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"connect error",DEMON,WTYPE);
		return -1;
	}
	cout << "connect...done..." << endl;

	//发送数据
	char buf[256] = "Hello...";
	if (send(local_socket,buf,strlen(buf)+1,0) == SOCKET_ERROR)
	{
		cout << WSAGetLastError() << endl;
		::MessageBoxA(NULL,"send error",DEMON,WTYPE);
		return -1;
	}
	cout << "send...done..." << endl;


	
	closesocket(local_socket);
	WSACleanup();
	return 0;
}