add: CgiHandle class

src/CGI/CgiHandle.cpp

@@ -0,0 +1,192 @@
+#include "CgiHandle.hpp"
+
+CgiHandle::CgiHandle(Request &request, Response &response):
+	_request(request), _response(response)
+{
+	initEnvVariables();
+}
+
+std::string toString(unsigned long var)
+{
+	std::stringstream ss;
+	ss << var;
+
+	return (ss.str());
+}
+
+std::string getEnvFormat(std:: string tmp)
+{
+	std::string str = tmp;
+	std::replace(str.begin(), str.end(), '-', '_');
+	std::transform(str.begin(), str.end(), str.begin(), ::toupper);
+	return (str);
+}
+
+char    **CgiHandle::mapToCharPtr(void)
+{
+	char **tmp = new char *[_variable.size() + 1];
+	std::map<std::string, std::string>::iterator    it;
+
+	int i = 0;
+	std::string value;
+	for (it = _variable.begin(); it != _variable.end(); it++)
+	{
+		if (it->second.empty())
+			continue;
+		value = it->first + "=" + it->second;
+		tmp[i] = new char[value.size() + 1];
+		strcpy(tmp[i], value.data());
+		i++;
+	}
+	tmp[i] = NULL;
+
+	return (tmp);    
+}
+
+void	CgiHandle::freeEnvVariables()
+{
+	for (size_t i = 0; env[i]; i++)
+	{
+		delete[] env[i];
+	}
+	delete[] env;
+	
+}
+
+void	CgiHandle::printSSmap(std::map<std::string, std::string> &m)
+{
+	std::map<std::string, std::string>::iterator it;
+
+	DBOUT << RED << "print MAP" << ENDL;
+	it = m.begin();
+	for (; it != m.end(); it++)
+	{
+		DBOUT << BLUE << it->first << PINK << it->second << ENDL;
+	}
+}
+
+void	printenv(char **env)
+{
+	for(size_t i = 0; env[i]; i++)
+	{
+		DBOUT << RED << env[i] << ENDL;
+	}
+}
+
+std::string CgiHandle::executeCgi()
+{
+	char		*argv[2];
+	pid_t		pid;
+	int			sI;
+	int			sO;
+	int			byte_read = 1;
+	std::string body;
+
+	argv[0] = const_cast<char *>(_response.getCgiPass().data());
+	argv[1] = NULL;
+	try
+	{
+		env = mapToCharPtr();
+	}
+	catch(const std::exception& e)
+	{
+		std::cerr << RED << e.what()  << RESET << '\n';
+	}
+	printenv(env);
+	sI = dup(STDIN_FILENO);
+	sO = dup(STDOUT_FILENO);
+
+	FILE *fIn = tmpfile();
+	FILE *fOt = tmpfile();
+	long	fdin = fileno(fIn);
+	long	fdOut = fileno(fOt);
+	DBOUT <<  BLUE << "in CGI exe" << ENDL;
+	// DBOUT << "BODY[\n" << _request.getBody() << "\n]" << ENDL;
+	write(fdin, _request.getBody().data(), _request.getBody().size());
+	lseek(fdin, 0, SEEK_SET);
+	pid = fork();
+	if (pid == -1)
+	{
+		std::cerr << RED << "Pid = -1. Fork error." << ENDL;
+	}
+	else if (pid == 0)
+	{
+		dup2(fdin, STDIN_FILENO);
+		dup2(fdOut, STDOUT_FILENO);
+		execve(_response.getCgiPass().c_str(), argv, env);
+		std::cerr << RED << "Execve error." << RESET << ENDL;
+		write(STDOUT_FILENO, "Status: 500\r\n\r\n", 15);
+	}
+	else
+	{
+		char buffer[BUFFSIZE + 1] = {0};
+
+		waitpid(-1, NULL, 0);
+		lseek(fdOut, 0, SEEK_SET);
+
+		while (byte_read)
+		{
+			bzero(buffer, BUFFSIZE);
+			byte_read = read(fdOut, buffer, BUFFSIZE);
+			body.append(buffer, byte_read);
+		}
+	}
+	dup2(sI, STDIN_FILENO);
+	dup2(sO, STDOUT_FILENO);
+	fclose(fIn);
+	fclose(fOt);
+	close(fdin);
+	close(fdOut);
+	close(sI);
+	close(sO);
+	freeEnvVariables();	
+	return (body);
+}
+
+void    CgiHandle::initEnvVariables()
+{
+	std::map<std::string, std::string>::iterator    it;
+	std::map<std::string, std::string> tmp1 =  _request.getClientFields();
+	_scriptName = _response.getFullURI().substr(_response.getFullURI().rfind("/"));
+	it = tmp1.find("content_type");
+	if (it != tmp1.end())
+		_variable["AUTH TYPE"] = it->second;
+	else
+		_variable["AUTH TYPE"] = "";
+	_variable["CONTENT_LENGTH"] = toString(_request.getContentLength());
+	it = _request.getClientFields().find("content-type");
+	_variable["CONTENT_TYPE"] = "";
+	_variable["GATEWAY_INTERFACE"] = std::string("CGI/1.1");
+
+	_variable["SCRIPT_NAME"] = _response.getFullURI();
+	_variable["SCRIPT_FILENAME"] = _response.getFullURI();
+
+	_variable["PATH_TRANSLATED"] = _response.getFullURI();
+	_variable["REQUEST_URI"] = _request.getURI();
+	_variable["PATH_INFO"] = _request.getURI();
+	
+	_variable["QUERY_STRING"] = _request.getQuery();
+	_variable["REMOTE_ADDR"] = _response.getListen().ip;
+	_variable["REMOTE_HOST"] = _response.getListen().ip;
+	_variable["REMOTE_IDENT"] = "";
+	_variable["REMOTE_USER"] = "";
+	_variable["REQUEST_METHOD"] = _request.getMethod();
+	_variable["SERVER_NAME"] = _request.getHost();
+	_variable["SERVER_PORT"] = toString(_response.getListen().port);
+	_variable["SERVER_PROTOCOL"] = "HTTP/1.1";
+	_variable["REDIRECT_STATUS"] = "200";
+	_variable["SERVER_SOFTVARE"] = "poheck/1.0.0";
+
+	it = tmp1.begin();
+	std::string tmp;
+	for (; it != tmp1.end(); it++)
+	{
+		tmp = getEnvFormat(it->first);
+		_variable["HTTP_" + tmp] = it->second;
+	}
+	printSSmap(_variable);
+}
+
+CgiHandle::~CgiHandle()
+{
+}

src/CGI/CgiHandle.hpp

@@ -0,0 +1,30 @@
+#ifndef CGIHANDLE_HPP
+#define CGIHANDLE_HPP
+
+#include "webserv.hpp"
+#include "Request.hpp"
+#include "ServerConfig.hpp"
+#include "Response.hpp"
+
+class Response;
+
+class CgiHandle
+{
+private:
+    Request                             &_request;
+    Response                            &_response;
+    std::map<std::string, std::string>  _variable;
+    std::string                         _scriptName;
+    char                                **env;
+public:
+    void    printSSmap(std::map<std::string, std::string> &);
+    CgiHandle(Request &request, Response &response);
+    void    initEnvVariables();
+    void    freeEnvVariables();
+    char    **mapToCharPtr();
+    std::string executeCgi();
+    ~CgiHandle();
+};
+
+
+#endif