Changeset - b6dd8ad6fd17
[Not reviewed]
0 1 0
Jan Kaluza - 14 years ago 2011-08-02 11:29:52
hanzz.k@gmail.com
Better spawning
1 file changed with 52 insertions and 5 deletions:
0 comments (0 inline, 0 general)
src/networkpluginserver.cpp
Show inline comments
 
@@ -89,23 +89,70 @@ class NetworkFactory : public Factory {
 

	
 
#define WRAP(MESSAGE, TYPE) 	pbnetwork::WrapperMessage wrap; \
 
	wrap.set_type(TYPE); \
 
	wrap.set_payload(MESSAGE); \
 
	wrap.SerializeToString(&MESSAGE);
 
	
 
static pid_t exec_(const char *path, const char *host, const char *port, const char *config) {
 
static std::vector<std::string> &split(const std::string &s, char delim, std::vector<std::string> &elems) {
 
    std::stringstream ss(s);
 
    std::string item;
 
    while(std::getline(ss, item, delim)) {
 
        elems.push_back(item);
 
    }
 
    return elems;
 
}
 

	
 

	
 
static std::vector<std::string> split(const std::string &s, char delim) {
 
    std::vector<std::string> elems;
 
    return split(s, delim, elems);
 
}
 
	
 
static pid_t exec_(std::string path, const char *host, const char *port, const char *config) {
 
	LOG4CXX_INFO(logger, "Starting new backend " << path);
 
	std::vector<std::string> cmd = split(path, ' ');
 
	char **argv = (char **) malloc(sizeof(char *) * (cmd.size() + 6));
 
	int i = 0;
 
	BOOST_FOREACH(std::string c, cmd) {
 
		argv[i] = (char *) malloc(c.size() + 1);
 
		strcpy(argv[i++], c.c_str());
 
// 		std::cout << i << "\n";
 
	}
 

	
 
	argv[i] = (char *) malloc(100);
 
	strcpy(argv[i++], "--host");
 

	
 
	argv[i] = (char *) malloc(100);
 
	strcpy(argv[i++], host);
 

	
 
	argv[i] = (char *) malloc(100);
 
	strcpy(argv[i++], "--port");
 

	
 
	argv[i] = (char *) malloc(100);
 
	strcpy(argv[i++], port);
 

	
 
	argv[i] = (char *) malloc(100);
 
	strcpy(argv[i++], config);
 

	
 
	argv[i] = 0;
 

	
 
// 	char *argv[] = {(char*)script_name, '\0'}; 
 
	pid_t pid = fork();
 
	if ( pid == 0 ) {
 
		// child process
 
		exit(execlp(path, path, "--host", host, "--port", port, config, NULL));
 
		exit(execv(cmd[0].c_str(), argv));
 
	} else if ( pid < 0 ) {
 
		// fork failed
 
	}
 

	
 
	i = 0;
 
	while (argv[i] != 0) {
 
		free(argv[i++]);
 
	}
 
	free(argv);
 

	
 
	return pid;
 
}
 

	
 
static void SigCatcher(int n) {
 
	pid_t result;
 
	int status;
 
@@ -170,13 +217,13 @@ NetworkPluginServer::NetworkPluginServer(Component *component, Config *config, U
 
	m_server->start();
 

	
 
	LOG4CXX_INFO(logger, "Listening on host " << CONFIG_STRING(m_config, "service.backend_host") << " port " << CONFIG_STRING(m_config, "service.backend_port"));
 

	
 
	signal(SIGCHLD, SigCatcher);
 

	
 
	exec_(CONFIG_STRING(m_config, "service.backend").c_str(), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
	exec_(CONFIG_STRING(m_config, "service.backend"), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
}
 

	
 
NetworkPluginServer::~NetworkPluginServer() {
 
	m_pingTimer->stop();
 
	m_server->stop();
 
	m_server.reset();
 
@@ -249,13 +296,13 @@ void NetworkPluginServer::handleSessionFinished(Backend *c) {
 
	// Execute new session only if there's no free one after this crash/disconnection
 
	for (std::list<Backend *>::const_iterator it = m_clients.begin(); it != m_clients.end(); it++) {
 
		if ((*it)->users.size() < CONFIG_INT(m_config, "service.users_per_backend")) {
 
			return;
 
		}
 
	}
 
	exec_(CONFIG_STRING(m_config, "service.backend").c_str(), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
	exec_(CONFIG_STRING(m_config, "service.backend"), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
}
 

	
 
void NetworkPluginServer::handleConnectedPayload(const std::string &data) {
 
	pbnetwork::Connected payload;
 
	if (payload.ParseFromString(data) == false) {
 
		// TODO: ERROR
 
@@ -975,13 +1022,13 @@ NetworkPluginServer::Backend *NetworkPluginServer::getFreeClient() {
 
			}
 
			break;
 
		}
 
	}
 

	
 
	if (c == NULL) {
 
		exec_(CONFIG_STRING(m_config, "service.backend").c_str(), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
		exec_(CONFIG_STRING(m_config, "service.backend"), CONFIG_STRING(m_config, "service.backend_host").c_str(), CONFIG_STRING(m_config, "service.backend_port").c_str(), m_config->getConfigFile().c_str());
 
	}
 

	
 
	return c;
 
}
 

	
 
}
0 comments (0 inline, 0 general)