MethodPost.cpp

#include"includes/Connection.hpp"
#include"includes/Request.hpp"

std::string ExtGet(std::string type){
    std::string ext;
    if(type == "image/png" || type == "image/gif" || type == "image/webp" 
        || type == "image/jpeg" || type == "video/mp4")
        ext = "." + type.substr(6);
    else if( type == "text/plain" || type == "application/x-www-form-urlencoded")
        ext =  ".txt";
    else if (type == "application/octet-stream")
        ext = ".bin";
    else if (type == "text/html" || type == "text/css" || type == "text/pdf" || type == "text/json")
        ext = "." + type.substr(5);
	else if (type == "text/markdown")
		ext = ".md";
	else if (type == "application/pdf")
		ext = ".pdf";
	else if (type == "text/x-csrc" )
		ext = ".c";
    return ext;
}

std::string generate_name(){

    std::string name;
    std::string str;
    int pos;

    srand(time(NULL));
    int s ;

    str = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
    s = (rand() % 3) + 4;
    for(int j = 0; j < s; j++){
        pos = rand() % str.size();
        name.push_back(str[pos]);    
    }
    return name;
}

void sign_handler(int i){
	throw MyError ("exit");
	(void)i;
}


void Connection::run_cgi_post(std::string url_path, std::string path, Location &lc,char **env){
    std::string pth = lc.getCgi()["." + get_extension(url_path)];
	time_t start = time(NULL);
	int	status , fd,  body;
	pid_t	pid;
	pid_t	w = 0;
	char *scri[3];

	cgi = generate_name();
	signal(SIGINT,SIG_IGN);
	
	body = open(path.c_str(), O_RDONLY);
	if(body == -1)
		return get_err(500);
	fd = open(cgi.c_str(), O_CREAT | O_TRUNC | O_WRONLY, 0644);
	if(fd ==-1 ){
		close(body);
		std::remove(cgi.c_str());
		return get_err(500);
	}
	pid = fork();
	if(pid == -1){
		close(fd);
		close(body);
		std::remove(cgi.c_str());
		return get_err(500);
	}
	if(pid == 0){
		signal(SIGINT,sign_handler);
		scri[0] = strdup(pth.c_str());
		scri[1] = strdup(url_path.c_str());
		scri[2] = NULL;
		dup2(body, STDIN_FILENO);
		close(body);
		dup2(fd,STDOUT_FILENO);
		close(fd);
		execve(pth.c_str(), scri, env);
		free(scri[0]);
		free(scri[1]);
		free_env(env);
		throw MyError ("exit");
	}
	while(w == 0 && time(NULL) - start < 7){
		w = waitpid(pid, &status, WNOHANG);
		if(w == pid &&  WIFEXITED(status) && !WEXITSTATUS(status) ){
			close(body);
			close(fd);
			if (!get_respo(cgi, true, path)){
				std::remove(cgi.c_str());
				return get_err(500);
			}
			return;
		}
		if(WIFEXITED(status) && WEXITSTATUS(status) == 1){
			close(fd);
			std::remove(cgi.c_str());
			return get_err(502);
		}
	}
	if(w == 0){
		kill(pid,SIGKILL) ;
		waitpid(pid,&status,0);
		close(body);
		close(fd);
		std::remove(cgi.c_str());
		return get_err(504);	
	}
	close(fd);
	close(body);
	std::remove(cgi.c_str());
	return get_err(500);
}

std::string check_dir_cgi(std::string url_path,Location &lc){
	std::string tmp;
	size_t i = 0;
	if(url_path[url_path.size() - 1] != '/')
		url_path += '/';
	while(i < lc.getIndex().size()){
		tmp.clear();
		tmp = url_path + lc.getIndex()[i];
		if(!access(tmp.c_str(), F_OK) && check_cgi(tmp, lc)){
			return tmp;
		}
		i++;
	}
	tmp.clear();
	return tmp;
}
void	Connection::MethodPost(Location &lc){
    struct  stat buf ;
	std::string tmp;
    std::string path = lc.getRoot() + lc.getUpload();
    std::vector<std::string> vect = lc.getMethods();
	std::string	url_path = lc.getRoot() + this->route;
    std::string name = generate_name() + ExtGet(_request.getHeaders()["Content-Type"]);

    std::vector<std::string>::iterator it = std::find(vect.begin(), vect.end(), "POST");
	if(it == vect.end()){
		std::remove(_request.getBody().c_str());
		get_err(405);
		return ;
	}
	if(ExtGet(_request.getHeaders()["Content-Type"]).empty()){
		std::remove(_request.getBody().c_str());
		get_err(415);
		return;
	}
    int err = stat(path.c_str(),&buf);
    if ( !err && S_ISDIR(buf.st_mode)){
		path += "/" + name;
		if(std::rename(_request.getBody().c_str(), path.c_str())){
			get_err(500);
			std::remove(_request.getBody().c_str());
			return;
		}
		err = stat(url_path.c_str(),&buf);
        if(err){
			get_err(404);
			 std::remove(path.c_str());
			return;
		}
		else if(S_ISREG(buf.st_mode) && check_cgi(url_path, lc)){
			char **env = initialize_env(url_path,lc);
			run_cgi_post(url_path,path,lc,env);
			std::remove(path.c_str());
			free_env(env);
		}
		else if (S_ISDIR(buf.st_mode) ){
			tmp = check_dir_cgi(url_path,lc);
			if (!tmp.empty()){
				char **env = initialize_env(tmp,lc);
				run_cgi_post(tmp,path,lc,env);
				std::remove(path.c_str());
				free_env(env);
			}
			else{
				this->upload_name = lc.getUpload()+ "/"  + name;
				get_err(201);
			}
		}
        else{
			this->upload_name = lc.getUpload() +"/" + name;
        	get_err(201);
		}
    }
    else {
	    get_err(500);
	    std::remove(_request.getBody().c_str());
    }
}