Foro de elhacker.net

#ifndef _TCPSOCKET_H_
#define _TCPSOCKET_H_

class TCPSocket
{
	private:
		int localSocket;

	public:
		TCPSocket();
		~TCPSocket();
		void Connect(const char *host, int port);
		void Write(const char *data);
		char *Read();
};

#endif 

#include <iostream>
#include <string.h>
#include <unistd.h>
#include <netdb.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include "TCPSocket.h"

using namespace std;

TCPSocket::TCPSocket()
{
	localSocket = socket(AF_INET, SOCK_STREAM, 0);
	if (localSocket == -1) throw "No se puede crear el socket";
}

TCPSocket::~TCPSocket()
{
	int result = close(localSocket);
	if (result == -1) throw "No se puede cerrar el socket";
}

void TCPSocket::Connect(const char *host, int port)
{
	struct hostent *server = gethostbyname(host);
	if (server == NULL) throw "No se encuentra el host";

	struct sockaddr_in serverAddress;
	serverAddress.sin_family = AF_INET;
	serverAddress.sin_port = htons(port);
	serverAddress.sin_addr = *(struct in_addr *)(server->h_addr_list[0]);
	bzero(&(serverAddress.sin_zero), 8);

	cout << "Conectando con " << host << endl;
	cout << "Conectando con " << inet_ntoa(serverAddress.sin_addr) << endl;

	int result = connect(localSocket, (struct sockaddr *) &serverAddress, sizeof(serverAddress));
	if (result == -1) throw "No se puede conectar con el host";
}

void TCPSocket::Write(const char *data)
{
	int dataLength = strlen(data);
	int sent = send(localSocket, data, dataLength, 0);

	if (sent == -1) throw "No se pueden enviar los datos al host";
	if (sent != dataLength) throw "No se han podido enviar todos los datos al host";
}

char *TCPSocket::Read()
{
	const int MAX_DATA_LENGHT = 16384;
	char *data = new char[MAX_DATA_LENGHT];
	
	int received = recv(localSocket, data, MAX_DATA_LENGHT, 0);	
	if (received == -1) throw "No se pueden recibir los datos del host";

	return (data);
}

#include <iostream>
#include "TCPSocket.h"

using namespace std;

int main()
{
	try
	{
		cout << "HTTP REQUEST" << endl;
		cout << "------------" << endl;
	
		// Establecemos la conexion con el servidor
		TCPSocket *tcpSocket = new TCPSocket();	
		tcpSocket->Connect("www.ubuntu.com", 80);

		// Generamos la peticion HTTP
		string request = "";
		request.append("HEAD / HTTP/1.1\r\n");
		request.append("Host: www.ubuntu.com\r\n");
		request.append("User-agent: My User Agent\r\n");
		request.append("\r\n");
	
		for(int i = 0; i < 5; i++)
		{
			// Realizamos la peticion HTTP
			tcpSocket->Write(request.c_str());

			// Obtenemos la respuesta del servidor
			char *data = tcpSocket->Read();
			string response = string(data);
		
			cout << endl << response << endl;
		}
		
		delete tcpSocket;
	
		return (0);
	}
		
	catch (const char *e)
	{
		cout << "Excepcion: " << e << endl;
		return (-1);
	}
}