Forum: C++ Winsock

// TCP.h
 
#ifndef TCP_H
#define TCP_H
 
#include <iostream>
#include <string>
#include <winsock2.h>
 
using namespace std;
 
class TCP
{
private:
	int sockfd;
	int new_fd;
 
public:
	TCP( void );
	~TCP( void );
	bool Create( void );
	bool Connect( string Adress, unsigned short int Port );
	bool Bind( int localport );
	bool Listen( int backlogs );
	bool Accept( void );
	int Send( string data );
	int Recv( char *buffer, int length );
	void Close( void );
	void Shutdown( void );
};
 
 
#endif

// TCP.cpp
 
#include "TCP.h"
 
 
TCP::TCP( void )
{
	new_fd = 0;
	WSADATA wsa;
	WSAStartup( MAKEWORD( 1,1 ), &wsa );
}
TCP::~TCP( void )
{
	WSACleanup();
}
bool TCP::Create( void )
{
	sockfd = socket( AF_INET, SOCK_STREAM, NULL );
 
	if( sockfd == INVALID_SOCKET )
	{
		cout << "Setup		error" << endl;
		return false;
	}
	else
	{
		cout << "Setup		Ok" << endl;
		return true;
	}
}
bool TCP::Bind( int localport )
{
	struct sockaddr_in addr;
 
	addr.sin_addr.s_addr	= INADDR_ANY;
	addr.sin_family			= AF_INET;
	addr.sin_port			= htons( localport );
	memset( addr.sin_zero, 0, sizeof addr.sin_zero );
 
	if(	bind( sockfd, (struct sockaddr*) &addr, sizeof sockaddr_in ) == SOCKET_ERROR )
	{
		cout << "Bind		error" << endl;
		return false;
	}
 
	cout << "bind		Ok" << endl;
	return true;
}
bool TCP::Listen( int backlogs )
{
	if( listen( sockfd, backlogs ) == SOCKET_ERROR )
	{
		cout << "Listen		error" << endl;
		return false;
	}
	cout << "Listen		Ok" << endl;
	return true;
}
bool TCP::Accept( void )
{
	sockaddr_in their_addr;
	int len = sizeof sockaddr_in;
	memset( &their_addr, 0, sizeof sockaddr_in );
 
	new_fd = accept( sockfd, ( sockaddr* ) &their_addr, &len );
	if( new_fd == SOCKET_ERROR )
	{
		cout << "Accept		error" << endl;
		return false;
	}
	cout << "Accept		Ok" << endl;
	return true;
}
bool TCP::Connect( string Adress, unsigned short Port )
{
	struct sockaddr_in addr;
 
	addr.sin_family			= AF_INET;
	addr.sin_port			= htons( Port );
	addr.sin_addr.s_addr	= inet_addr( Adress.c_str() );
	memset( addr.sin_zero, 0, sizeof addr.sin_zero );
 
	if( connect( sockfd, ( const sockaddr* ) &addr, sizeof sockaddr ) == SOCKET_ERROR )
	{
		cout << "Connection		error" << endl;
		return false;
	}
 
	cout << "Connection	Ok" << endl;
	return true;
}
 
int TCP::Send( string data )
{
	if( new_fd == NULL )
		return send( sockfd, data.c_str(), strlen( data.c_str() ), NULL );
	else
		return send( new_fd, data.c_str(), strlen( data.c_str() ), NULL );
}
int TCP::Recv( char *buffer, int length )
{
	if( new_fd == NULL )
		return recv( sockfd, buffer, length, NULL );
	else
		return recv( new_fd, buffer, length, NULL );
}
void TCP::Close( void )
{
	cout << "Close		Ok" << endl;
	closesocket( sockfd );
}
void TCP::Shutdown( void )
{
	shutdown( new_fd, SD_BOTH );
}

// main.cpp
 
#pragma comment( lib, "WSock32.Lib" )
 
#include <iostream>
#include <winsock2.h>
#include <string>
 
#include "TCP.h"
 
using namespace std;
 
#define PORT 1337
#define ADRESS "127.0.0.1"
 
 
 
int main()
{
	TCP test;
 
	string val;
	char *buffer = "";
 
	cout << "1 - Server" <<endl << "2 - Client" << endl;
	cin >> val;
 
	if( val == "1" )
	{
		cout << "--------SERVER--------" << endl;		
		test.Create();
		test.Bind( PORT );
 
		while( true )
		{
			test.Listen( 10 );
			test.Accept();
			cout << "Received: " << test.Recv( buffer, 512 ) << endl;
			cout << "WSAERROR: " << WSAGetLastError() << endl;
			cout << "Buffer: " << buffer << endl;
		}
 
	}
	else
	{	
		cout << "--------CLIENT--------" << endl;
		test.Create();
		test.Connect( ADRESS, PORT );
		test.Send( "hej" );
		cout << "WSAERROR: " << WSAGetLastError() << endl;
		test.Close();		
	}
 
	cout << "klar" << endl;
	cin.get();
	cin.get();
 
 
	return 0;
}

char buffer[512];