Connection.cpp

#include "Connection.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string>

Connection::Connection()  {
    int PORT;
    std::cout << "What port number? ";
    std::cin >> PORT;
    
    int tempSockfd, portno;
    socklen_t clilen;
    char buffer[256];
    struct sockaddr_in serv_addr, cli_addr;
    int n;
    tempSockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (tempSockfd < 0)
       std::cout << "ERROR opening socket";
    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = INADDR_ANY;
    serv_addr.sin_port = htons(PORT);
    if (bind(tempSockfd, (struct sockaddr *) &serv_addr,
             sizeof(serv_addr)) < 0)
    		std::cout << "ERROR on binding";
    listen(tempSockfd,5);
    clilen = sizeof(cli_addr);
    std::cout << "Waiting for connection\n";
    socketFd = accept(tempSockfd,
                (struct sockaddr *) &cli_addr,
                &clilen);

    std::cout << "Recieved connection\n";
    if (socketFd < 0)
    	std::cout << "ERROR on accept";
}

Connection::~Connection() {
	disconnect();
}

double Connection::getReward() {
	std::string currentLine = getString();
	std::cout << "lineL " << currentLine << "\n";
	while(currentLine.compare("4") != 0) {
		std::cout << "line: " << currentLine << "\n";
	}
	return atof(getString().c_str());
}

void Connection::disconnect() {
	close(socketFd);
}

std::string Connection::getString() {
	std::string newString = "";
	char buffer;

	// Keep reading char by char until a newline
	while(true) {
		read(socketFd, &buffer, 1);

		std::cout << buffer;

		if (buffer == '\n') break;
		else newString += buffer;
	}

	return newString;
}