//----- Include files --------------------------------------------------------- #include // Needed for exit() #include // Needed for strcpy() and strlen() #include // Needed for file i/o constants #include // Needed for file i/o constants #include #include #include // #include // #include // #include // #include //----- HTTP response messages ---------------------------------------------- #define OK_IMAGE "HTTP/1.0 200 OK\nContent-Type:image/gif\n\n" #define OK_TEXT "HTTP/1.0 200 OK\nContent-Type:text/html\n\n" #define NOTOK_404 "HTTP/1.0 404 Not Found\nContent-Type:text/html\n\n" #define MESS_404 "

FILE NOT FOUND

" //----- Defines ------------------------------------------------------------- #define BUF_SIZE 4096 // Buffer size (big enough for a GET) #define PORT_NUM 9080 // Port number for a Web server //----- Function prototypes ------------------------------------------------- void BrowserToServer(int browser_s, int server_s); void ServerToBrowser(int server_s, int browser_s); //===== modeule main ======================================================== int main(void) { unsigned int proxy_s; // Server socket descriptor struct sockaddr_in proxy_addr; // Server Internet address unsigned int browser_s; // Client socket descriptor struct sockaddr_in browser_addr; // Client Internet address struct in_addr browser_ip_addr; // Client IP address unsigned int server_s; // Client socket descriptor struct sockaddr_in server_addr; // Client Internet address struct in_addr server_ip_addr; // Client IP address socklen_t addr_len; // Internet address length // Create a socket, fill-in address information, and then bind it proxy_s = socket(AF_INET, SOCK_STREAM, 0); if (proxy_s == -1) { std::cerr << "ERROR - Unable to create socket on server" << std::endl; exit(1); } proxy_addr.sin_family = AF_INET; proxy_addr.sin_port = htons(PORT_NUM); proxy_addr.sin_addr.s_addr = htonl(INADDR_ANY); int bindRet = bind(proxy_s, (struct sockaddr *)&proxy_addr, sizeof(proxy_addr)); if (bindRet == -1) { std::cerr << "ERROR - Unable to bind socket" << std::endl; exit(1); } // Listen for connections and then accept listen(proxy_s, 100); // Main loop to listen, accept, and then spin-off a thread to handle the GET while (1) { addr_len = sizeof(browser_addr); browser_s = accept(proxy_s, (struct sockaddr *)&browser_addr, &addr_len); if (browser_s == -1) { std::cerr << "ERROR - Unable to create socket to client" << std::endl; continue; } std::cout << "Got here" << std::endl; unsigned int server_s = socket(AF_INET, SOCK_STREAM, 0); connect(server_s, (struct sockaddr *)&server_addr, addr_len); std::async(std::launch::async, BrowserToServer, browser_s, server_s); std::async(std::launch::async, ServerToBrowser, server_s, browser_s); } close(proxy_s); return (1); } void BrowserToServer(int browser_s, int server_s) { char in_buf[BUF_SIZE]; // Input buffer for GET resquest 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); } void ServerToBrowser(int server_s, int browser_s) { char out_buf[BUF_SIZE]; // Output buffer for HTML response ssize_t buf_len; // Buffer length for file reads // 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); }