81 lines
2.5 KiB
C++
81 lines
2.5 KiB
C++
|
#include "network.hpp"
|
||
|
|
#include <arpa/inet.h>
|
||
|
|
#include <stdlib.h> // Needed for exit()
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
void HandleClient(int browser_s, int server_s) {
|
||
|
|
char in_buf[BUF_SIZE]; // Input buffer for GET resquest
|
||
|
|
char out_buf[BUF_SIZE]; // Output buffer for HTML response
|
||
|
|
ssize_t buf_len; // Buffer length for file reads
|
||
|
|
|
||
|
|
// Pass GET along from browser to server
|
||
|
|
do {
|
||
|
|
buf_len = recv(browser_s, in_buf, BUF_SIZE, 0);
|
||
|
|
send(server_s, in_buf, buf_len, 0);
|
||
|
|
} while (buf_len != 0);
|
||
|
|
|
||
|
|
// Pass response along from server to browser
|
||
|
|
do {
|
||
|
|
buf_len = recv(server_s, out_buf, BUF_SIZE, 0);
|
||
|
|
send(browser_s, out_buf, buf_len, 0);
|
||
|
|
} while (buf_len != 0);
|
||
|
|
close(server_s);
|
||
|
|
close(browser_s);
|
||
|
|
}
|
||
|
|
|
||
|
|
Client::Client(void) {
|
||
|
|
addressLength = sizeof(address);
|
||
|
|
address.sin_family = AF_INET;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Client::ConnectFrom(int socketFD) {
|
||
|
|
socketFD = accept(socketFD, (struct sockaddr *)&address, &addressLength);
|
||
|
|
if (socketFD == -1) { return 1; }
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Client::ConnectTo(int portNumber) {
|
||
|
|
std::cout << "Attempting to connect to " << kWebserverIP << " on " << portNumber << std::endl;
|
||
|
|
address.sin_port = htons(portNumber);
|
||
|
|
if (inet_pton(address.sin_family, kWebserverIP, &address.sin_addr) <= 0) {
|
||
|
|
std::cerr << "ERROR (info) - inet_pton" << std::endl;
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
socketFD = socket(address.sin_family, SOCK_STREAM, 0);
|
||
|
|
if (connect(socketFD, (struct sockaddr *)&address, addressLength) == -1) {
|
||
|
|
std::cerr << "ERROR (info) - connect" << std::endl;
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
std::cout << "Connection successful" << std::endl;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
Server::Server(int portNumber) {
|
||
|
|
address.sin_family = AF_INET;
|
||
|
|
address.sin_port = htons(portNumber);
|
||
|
|
address.sin_addr.s_addr = htonl(INADDR_ANY);
|
||
|
|
}
|
||
|
|
|
||
|
|
void Server::Open(void) {
|
||
|
|
std::cout << "Opening the server" << std::endl;
|
||
|
|
// Create a socket, fill-in address information, and then bind it
|
||
|
|
socketFD = socket(address.sin_family, SOCK_STREAM, 0);
|
||
|
|
if (socketFD == -1)
|
||
|
|
{
|
||
|
|
std::cerr << "ERROR - Unable to create socket on server" << std::endl;
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
if (bind(socketFD, (struct sockaddr *)&address, sizeof(address)) == -1)
|
||
|
|
{
|
||
|
|
std::cerr << "ERROR - Unable to bind socket" << std::endl;
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
// Listen for connections and then accept
|
||
|
|
listen(socketFD, 100);
|
||
|
|
}
|
||
|
|
|
||
|
|
void Server::Close(void) {
|
||
|
|
std::cout << "Closing the server" << std::endl;
|
||
|
|
close(socketFD);
|
||
|
|
}
|