Changeset - f4df7fdd4e5b
Parent rev.
Child rev.
[Not reviewed]
0 3 0
HanzZ - 14 years ago 2011-09-08 15:10:17
hanzz.k@gmail.com
spectrum is daemon now, hope there's no Buffy
3 files changed with 95 insertions and 9 deletions:
spectrum/src/main.cpp
88
8
spectrum/src/sample.cfg
1
1
src/config.cpp
6
0 comments (0 inline, 0 general)
spectrum/src/main.cpp
➞
Show inline comments
 
@@ -12,99 +12,179 @@
 
#include "sys/signal.h"
 
#else
 
#include <Windows.h>
 
#include <tchar.h>
 
#endif
 
#include "log4cxx/logger.h"
 
#include "log4cxx/patternlayout.h"
 
#include "log4cxx/propertyconfigurator.h"
 
#include "log4cxx/consoleappender.h"
 
 
using namespace log4cxx;
 
 
using namespace Transport;
 
 
Swift::SimpleEventLoop *eventLoop_ = NULL;
 
 
static void spectrum_sigint_handler(int sig) {
 
	eventLoop_->stop();
 
}
 
 
static void spectrum_sigterm_handler(int sig) {
 
	eventLoop_->stop();
 
}
 
 
#ifndef WIN32
 
static void daemonize(const char *cwd, const char *lock_file) {
 
	pid_t pid, sid;
 
	FILE* lock_file_f;
 
	char process_pid[20];
 
 
	/* already a daemon */
 
	if ( getppid() == 1 ) return;
 
 
	/* Fork off the parent process */
 
	pid = fork();
 
	if (pid < 0) {
 
		exit(1);
 
	}
 
	/* If we got a good PID, then we can exit the parent process. */
 
	if (pid > 0) {
 
		exit(0);
 
	}
 
 
	/* At this point we are executing as the child process */
 
 
	/* Change the file mode mask */
 
	umask(0);
 
 
	/* Create a new SID for the child process */
 
	sid = setsid();
 
	if (sid < 0) {
 
		exit(1);
 
	}
 
 
	/* Change the current working directory.  This prevents the current
 
		directory from being locked; hence not being able to remove it. */
 
	if ((chdir(cwd)) < 0) {
 
		exit(1);
 
	}
 
 
	if (lock_file) {
 
		/* write our pid into it & close the file. */
 
		lock_file_f = fopen(lock_file, "w+");
 
		if (lock_file_f == NULL) {
 
			std::cout << "EE cannot create lock file " << lock_file << ". Exiting\n";
 
			exit(1);
 
		}
 
		sprintf(process_pid,"%d\n",getpid());
 
		if (fwrite(process_pid,1,strlen(process_pid),lock_file_f) < strlen(process_pid)) {
 
			std::cout << "EE cannot write to lock file " << lock_file << ". Exiting\n";
 
			exit(1);
 
		}
 
		fclose(lock_file_f);
 
	}
 
 
	if (freopen( "/dev/null", "r", stdin) == NULL) {
 
		std::cout << "EE cannot open /dev/null. Exiting\n";
 
		exit(1);
 
	}
 
}
 
 
#endif
 
 
int main(int argc, char **argv)
 
{
 
	Config config;
 
 
	boost::program_options::variables_map vm;
 
	bool no_daemon = false;
 
	std::string config_file;
 
 
 
#ifndef WIN32
 
	if (signal(SIGINT, spectrum_sigint_handler) == SIG_ERR) {
 
		std::cout << "SIGINT handler can't be set\n";
 
		return -1;
 
	}
 
 
	if (signal(SIGTERM, spectrum_sigterm_handler) == SIG_ERR) {
 
		std::cout << "SIGTERM handler can't be set\n";
 
		return -1;
 
	}
 
#endif
 
	boost::program_options::options_description desc("Usage: spectrum [OPTIONS] <config_file.cfg>\nAllowed options");
 
	desc.add_options()
 
		("help,h", "help")
 
		("no-daemonize,n", "Do not run spectrum as daemon")
 
		("config", boost::program_options::value<std::string>(&config_file)->default_value(""), "Config file")
 
		;
 
	try
 
	{
 
		boost::program_options::variables_map vm;
 
		boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);
 
		boost::program_options::positional_options_description p;
 
		p.add("config", -1);
 
		boost::program_options::store(boost::program_options::command_line_parser(argc, argv).
 
          options(desc).positional(p).run(), vm);
 
		boost::program_options::notify(vm);
 
 
 
 
		if(vm.count("help"))
 
		{
 
			std::cout << desc << "\n";
 
			return 1;
 
		}
 
 
		if(vm.count("config") == 0) {
 
			std::cout << desc << "\n";
 
			return 1;
 
		}
 
 
		if(vm.count("no-daemonize")) {
 
			no_daemon = true;
 
		}
 
	}
 
	catch (std::runtime_error& e)
 
	{
 
		std::cout << desc << "\n";
 
		return 1;
 
	}
 
	catch (...)
 
	{
 
		std::cout << desc << "\n";
 
		return 1;
 
	}
 
 
	if (argc != 2) {
 
		std::cout << desc << "\n";
 
	if (!config.load(vm["config"].as<std::string>())) {
 
		std::cerr << "Can't load configuration file.\n";
 
		return 1;
 
	}
 
 
 
	if (!config.load(argv[1])) {
 
		std::cerr << "Can't load configuration file.\n";
 
		return 1;
 
#ifndef WIN32
 
	if (!no_daemon) {
 
		daemonize("/", NULL);
 
	}
 
#endif
 
 
	if (CONFIG_STRING(&config, "logging.config").empty()) {
 
		LoggerPtr root = log4cxx::Logger::getRootLogger();
 
#ifdef WIN32
 
		root->addAppender(new ConsoleAppender(new PatternLayout(L"%d %-5p %c: %m%n")));
 
#else
 
		root->addAppender(new ConsoleAppender(new PatternLayout("%d %-5p %c: %m%n")));
 
#endif
 
	}
 
	else {
 
		log4cxx::PropertyConfigurator::configure(CONFIG_STRING(&config, "logging.config"));
 
	}
 
 
	Swift::SimpleEventLoop eventLoop;
 
 
	Swift::BoostNetworkFactories *factories = new Swift::BoostNetworkFactories(&eventLoop);
 
	UserRegistry userRegistry(&config, factories);
 
 
	Component transport(&eventLoop, factories, &config, NULL, &userRegistry);
 
// 	Logger logger(&transport);
 
 
	StorageBackend *storageBackend = NULL;
 
 
#ifdef WITH_SQLITE
spectrum/src/sample.cfg
➞
Show inline comments
 
[service]
 
jid = localhost
 
password = secret
 
server = 127.0.0.1
 
port = 5222
 
server_mode = 1
 
backend_host=localhost # < this option doesn't work yet
 
backend_port=10001
 
admin_username=admin
 
admin_password=test
 
#idle_reconnect_time=10
 
cert=server.pfx #patch to PKCS#12 certificate
 
cert_password=test #password to that certificate if any
 
users_per_backend=10
 
backend=../../backends/libpurple/spectrum_libpurple_backend
 
backend=/home/hanzz/code/libtransport/backends/libpurple/spectrum_libpurple_backend
 
#backend=../../backends/libircclient-qt/spectrum_libircclient-qt_backend
 
#protocol=prpl-msn
 
protocol=any
 
#protocol=prpl-icq
 
#idle_reconnect_time=10
 

			




	
	
	
		
 
[backend]

			




	
	
	
		
 
#default_avatar=catmelonhead.jpg

			




	
	
	
		
 
#no_vcard_fetch=true

			




	
	
	
		
 

			




	
	
	
		
 
[logging]

			




	
	
	
		
 
#config=logging.cfg # log4cxx/log4j logging configuration file

			




	
	
	
		
 
#backend_config=backend_logging.cfg # log4cxx/log4j logging configuration file for backends

			




	
	
	
		
 

			




	
	
	
		
 
[database]

			




	
	
	
		
 
#type = mysql # or "none" without database backend

			




	
	
	
		
 
database = test.sql

			




	
	
	
		
 
prefix=icq

			




        

    


    
    

    

        

                

                    src/config.cpp
                

                

                  
                      
                        

                      ➞

                      
                        
                  

                  
                      
                  
                      
                  
                      
                  
                      
                  
                  
                

                

                    Show inline comments
                    
                

        

        

            



	
	
		
 
@@ -13,48 +13,54 @@

			




	
	
	
		
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

			




	
	
	
		
 
 * GNU General Public License for more details.

			




	
	
	
		
 
 *

			




	
	
	
		
 
 * You should have received a copy of the GNU General Public License

			




	
	
	
		
 
 * along with this program; if not, write to the Free Software

			




	
	
	
		
 
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111-1301  USA

			




	
	
	
		
 
 */

			




	
	
	
		
 

			




	
	
	
		
 
#include "transport/config.h"

			




	
	
	
		
 
#include <fstream>

			




	
	
	
		
 

			




	
	
	
		
 
using namespace boost::program_options;

			




	
	
	
		
 

			




	
	
	
		
 
namespace Transport {

			




	
	
	
		
 

			




	
	
	
		
 
bool Config::load(const std::string &configfile, boost::program_options::options_description &opts) {

			




	
	
	
		
 
	std::ifstream ifs(configfile.c_str());

			




	
	
	
		
 
	if (!ifs.is_open())

			




	
	
	
		
 
		return false;

			




	
	
	
		
 

			




	
	
	
		
 
	m_file = configfile;

			




	
	
	
		
 
	bool ret = load(ifs, opts);

			




	
	
	
		
 
	ifs.close();

			




	
	
	
		
 

			




	
	
	
		
 
	char path[PATH_MAX] = "";

			




	
	
	
		
 
	if (m_file.find_first_of("/") != 0) {

			




	
	
	
		
 
		getcwd(path, PATH_MAX);

			




	
	
	
		
 
		m_file = std::string(path) + "/" + m_file;

			




	
	
	
		
 
	}

			




	
	
	
		
 

			




	
	
	
		
 
	return ret;

			




	
	
	
		
 
}

			




	
	
	
		
 

			




	
	
	
		
 
bool Config::load(std::istream &ifs, boost::program_options::options_description &opts) {

			




	
	
	
		
 
	m_unregistered.clear();

			




	
	
	
		
 
	opts.add_options()

			




	
	
	
		
 
		("service.jid", value<std::string>()->default_value(""), "Transport Jabber ID")

			




	
	
	
		
 
		("service.server", value<std::string>()->default_value(""), "Server to connect to")

			




	
	
	
		
 
		("service.password", value<std::string>()->default_value(""), "Password used to auth the server")

			




	
	
	
		
 
		("service.port", value<int>()->default_value(0), "Port the server is listening on")

			




	
	
	
		
 
		("service.backend", value<std::string>()->default_value("libpurple_backend"), "Backend")

			




	
	
	
		
 
		("service.protocol", value<std::string>()->default_value(""), "Protocol")

			




	
	
	
		
 
		("service.allowed_servers", value<std::string>()->default_value(""), "Only users from these servers can connect")

			




	
	
	
		
 
		("service.server_mode", value<bool>()->default_value(false), "True if Spectrum should behave as server")

			




	
	
	
		
 
		("service.users_per_backend", value<int>()->default_value(100), "Number of users per one legacy network backend")

			




	
	
	
		
 
		("service.backend_host", value<std::string>()->default_value("localhost"), "Host to bind backend server to")

			




	
	
	
		
 
		("service.backend_port", value<std::string>()->default_value("10000"), "Port to bind backend server to")

			




	
	
	
		
 
		("service.cert", value<std::string>()->default_value(""), "PKCS#12 Certificate.")

			




	
	
	
		
 
		("service.cert_password", value<std::string>()->default_value(""), "PKCS#12 Certificate password.")

			




	
	
	
		
 
		("service.admin_username", value<std::string>()->default_value(""), "Administrator username.")

			




	
	
	
		
 
		("service.admin_password", value<std::string>()->default_value(""), "Administrator password.")

			




	
	
	
		
 
		("service.reuse_old_backends", value<bool>()->default_value(true), "True if Spectrum should use old backends which were full in the past.")

			




	
	
	
		
 
		("service.idle_reconnect_time", value<int>()->default_value(4*3600), "Time in seconds after which idle users are reconnected to let their backend die.")

			




	
	
	
		
 
		("service.more_resources", value<bool>()->default_value(false), "Allow more resources to be connected in server mode at the same time.")

			




        

    



    


    



    



      

      





    
    0 comments (0 inline, 0 general)
    





    


  

  







  


    




    








    
        Server instance: srv01-3934
    
    
        This site is powered by
            Kallithea,
        which is
        © 2010–2021 by various authors & licensed under GPLv3.
            – Support