main.cpp:
codice:
#include <iostream>
#include "C:\Users\JeanDenajar\Desktop\socket\Socket.h"
#pragma comment(lib, "ws2_32.lib")
using namespace std;
int main()
{
int scelta;
int port = 80;
//char *ipAddress = "127.0.0.1";
char ipaddress[456];
string ipAddress;
bool done = false;
char recMessage[STRLEN];
char sendMessage[STRLEN];
cout<<"1) fai host"<<endl;
cout<<"2) entra nel server"<<endl;
cout<<"3) esci"<<endl;
cin>>scelta;
if ( scelta == 3 )
exit(0);
else if ( scelta == 2 )
{
//Client
cout<<"Inserisci l'IP: ";
cin>>ipaddress;
ipAddress = ipaddress;
cout << ipaddress;
ClientSocket sockClient;
cout<<"Attendi..."<<endl;
sockClient.ConnectToServer( ipAddress.c_str(), port );
//Connected
while ( !done )
{
sockClient.GetAndSendMessage();
cout<<"\t--aspetta--"<<endl;
sockClient.RecvData( recMessage, STRLEN );
cout<<"ricevuto > "<<recMessage<<endl;
if ( strcmp( recMessage, "end" ) == 0 || strcmp( sendMessage, "end" ) == 0 )
{
done = true;
}
}
sockClient.CloseConnection();
}
else if ( scelta == 1 )
{
//SERVER
ServerSocket sockServer;
cout<<"attendo connessione client..."<<endl;
sockServer.StartHosting( port );
//Connected
while ( !done )
{
cout<<"\t--aspetta--"<<endl;
sockServer.RecvData( recMessage, STRLEN );
cout<<"Ricevuto > "<<recMessage<<endl;
sockServer.GetAndSendMessage();
if ( strcmp( recMessage, "end" ) == 0 ||
strcmp( sendMessage, "end" ) == 0 )
{
done = true;
}
}
}
}
Sokets.cpp:
codice:
#include "Socket.h"
Socket::Socket()
{
if( WSAStartup( MAKEWORD(2, 2), &wsaData ) != NO_ERROR )
{
cerr<<"impossibile inizializzare\n";
system("pause");
WSACleanup();
exit(10);
}
//Create a socket
mySocket = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
if ( mySocket == INVALID_SOCKET )
{
cerr<<"Socket Initialization: Error creating socket"<<endl;
system("pause");
WSACleanup();
exit(11);
}
Backup = mySocket;
}
Socket::~Socket()
{
WSACleanup();
}
bool Socket::SendData( char *buffer )
{
send( mySocket, buffer, strlen( buffer ), 0 );
return true;
}
bool Socket::RecvData( char *buffer, int size )
{
int i = recv( mySocket, buffer, size, 0 );
buffer[i] = '\0';
return true;
}
void Socket::CloseConnection()
{
closesocket( mySocket );
mySocket = Backup;
}
void Socket::GetAndSendMessage()
{
char message[STRLEN];
cin.ignore();
cout<<"Inviato > ";
cin.get( message, STRLEN );
SendData( message );
}
void ServerSocket::StartHosting( int port )
{
Bind( port );
Listen();
}
void ServerSocket::Listen()
{
if ( listen ( mySocket, 1 ) == SOCKET_ERROR )
{
cerr<<"Errore server\n";
system("pause");
WSACleanup();
exit(15);
}
acceptSocket = accept( Backup, NULL, NULL );
while ( acceptSocket == SOCKET_ERROR )
{
acceptSocket = accept( Backup, NULL, NULL );
}
mySocket = acceptSocket;
}
void ServerSocket::Bind( int port )
{
myAddress.sin_family = AF_INET;
myAddress.sin_addr.s_addr = inet_addr( "0.0.0.0" );
myAddress.sin_port = htons( port );
if ( bind ( mySocket, (SOCKADDR*) &myAddress, sizeof( myAddress) ) == SOCKET_ERROR )
{
cerr<<"Errore server\n";
system("pause");
WSACleanup();
exit(14);
}
}
void ClientSocket::ConnectToServer( const char *ipAddress, int port )
{
myAddress.sin_family = AF_INET;
myAddress.sin_addr.s_addr = inet_addr( ipAddress );
myAddress.sin_port = htons( port );
//cout<<"CONNECTED"<<endl;
if ( connect( mySocket, (SOCKADDR*) &myAddress, sizeof( myAddress ) ) == SOCKET_ERROR )
{
cerr<<"impossibil connttersi\n";
system("pause");
WSACleanup();
exit(13);
}
}
Sockets.h :
codice:
#include "WinSock2.h"
using namespace std;
const int STRLEN = 256;
class Socket
{
protected:
WSADATA wsaData;
SOCKET mySocket;
SOCKET Backup;
SOCKET acceptSocket;
sockaddr_in myAddress;
public:
Socket();
~Socket();
bool SendData( char* );
bool RecvData( char*, int );
void CloseConnection();
void GetAndSendMessage();
};
class ServerSocket : public Socket
{
public:
void Listen();
void Bind( int port );
void StartHosting( int port );
};
class ClientSocket : public Socket
{
public:
void ConnectToServer( const char *ipAddress, int port );
};