static int initialize() {
int retval;
if (!cc_config.allow_multiple_clients) {
retval = wait_client_mutex(".", 10);
if (retval) {
log_message_error("Another instance of BOINC is running.");
return ERR_EXEC;
}
}
// Initialize WinSock
#if defined(_WIN32) && defined(USE_WINSOCK)
if (WinsockInitialize() != 0) {
log_message_error("Failed to initialize the Windows Sockets interface.");
return ERR_IO;
}
#endif
curl_init();
#ifdef _WIN32
if(!startup_idle_monitor()) {
log_message_error(
"Failed to initialize the BOINC idle monitor interface."
"BOINC will not be able to determine if the user is idle or not...\n"
);
}
#endif
return 0;
}
int main(int argc, char **argv)
{
u_short server_port = 5500;
u_short viewer_port = 5900;
int max_slots = 50;
char * dump_file = NULL;
thread_t hServerThread;
thread_t hViewerThread;
/* Arguments */
if (argc > 1) {
for(int i = 1; i < argc; i++)
{
if( _stricmp( argv[i], "-server" ) == 0 ) {
/* Requires argument */
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
server_port = atoi( argv[(i+1)] );
if( argv[(i+1)][0] == '-' ) {
usage( argv[0] );
return 1;
} else if( server_port == 0 ) {
usage( argv[0] );
return 1;
} else if( server_port > 65535 ) {
usage( argv[0] );
return 1;
}
i++;
} else if( _stricmp( argv[i], "-viewer" ) == 0 ) {
/* Requires argument */
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
viewer_port = atoi( argv[(i+1)] );
if( argv[(i+1)][0] == '-' ) {
usage( argv[0] );
return 1;
} else if( viewer_port == 0 ) {
usage( argv[0] );
return 1;
} else if( viewer_port > 65535 ) {
usage( argv[0] );
return 1;
}
i++;
} else if ( _stricmp( argv[i], "-dump" ) == 0 ) {
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
dump_file = argv[(i+1)];
i++;
} else if ( _stricmp( argv[i], "-loglevel" ) == 0 ) {
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
char level = ::get_log_level(argv[(i+1)]);
if(level == -1) { usage( argv[0] ); return 1; }
::logger_level = level;
i++;
} else if ( _stricmp( argv[i], "-iobuffer_size" ) == 0 ) {
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
int buffer_sz = atoi(argv[(i+1)]);
if(buffer_sz <= 128) { usage( argv[0] ); return 1; }
IOBUFFER_SIZE = buffer_sz;
i++;
} else if ( _stricmp( argv[i], "-max_slots" ) == 0 ) {
if( (i+i) == argc ) {
usage( argv[0] );
return 1;
}
int _max_slots = atoi(argv[(i+1)]);
if(_max_slots < 1) { usage( argv[0] ); return 1; }
max_slots = _max_slots;
i++;
}
else {
usage( argv[0] );
return 1;
}
}
}
#ifdef WIN32
/* Winsock */
if( WinsockInitialize() == 0 )
return 1;
#endif
/* Start */
logp(ERROR, "VNC Repeater [Version %s]", VNCREPEATER_VERSION);
log(INFO, "Copyright (C) 2010 Juan Pedro Gonzalez Gutierrez. Licensed under GPL v2.");
log(INFO, "Copyright (C) 2013 XSoft Ltd. - Andrey Andreev - www.xsoftbg.com. Licensed under GPL v2.");
log(INFO, "Get the latest version at http://code.google.com/p/vncrepeater/ or https://github.com/XSoftBG/repeater\n");
/* Initialize some variables */
notstopped = true;
InitializeSlots(max_slots);
/* Trap signal in order to exit cleanlly */
signal(SIGINT, ExitRepeater);
listener_thread_params * server_thread_params = (listener_thread_params *)malloc(sizeof(listener_thread_params));
memset(server_thread_params, 0, sizeof(listener_thread_params));
listener_thread_params * viewer_thread_params = (listener_thread_params *)malloc(sizeof(listener_thread_params));
memset(viewer_thread_params, 0, sizeof(listener_thread_params));
server_thread_params->port = server_port;
viewer_thread_params->port = viewer_port;
// Start multithreading...
// Tying new threads ;)
if (notstopped) {
if ( thread_create(&hServerThread, NULL, server_listen, (LPVOID)server_thread_params) != 0 ) {
log(FATAL, "Unable to create the thread to listen for servers.");
notstopped = false;
}
}
if (notstopped) {
if ( thread_create(&hViewerThread, NULL, viewer_listen, (LPVOID)viewer_thread_params) != 0 ) {
log(FATAL, "Unable to create the thread to listen for viewers.");
notstopped = false;
}
}
if (notstopped) {
FILE *f = fopen( "pid.repeater", "w" );
if(!f) perror("pid.repeater"), exit(1);
fprintf(f, "%d\n", (int)getpid());
fclose(f);
}
// Main loop
while (notstopped)
{
/* Clean slots: Free slots where the endpoint has disconnected */
CleanupSlots();
if (dump_file != NULL) {
int dump_fd = ::open(dump_file, O_CREAT | O_TRUNC | O_WRONLY, 0644);
std::string json = DumpSlots();
if (!json.empty())
write (dump_fd, json.c_str(), json.length() );
close (dump_fd);
}
/* Take a "nap" so CPU usage doesn't go up. */
#ifdef WIN32
Sleep(5000000);
#else
usleep(5000000);
#endif
}
log(ERROR, "\nExiting VNC Repeater...\n");
notstopped = false;
/* Free the repeater slots */
FreeSlots();
/* Close the sockets used for the listeners */
socket_close( server_thread_params->sock );
socket_close( viewer_thread_params->sock );
/* Free allocated memory for the thread parameters */
free( server_thread_params );
free( viewer_thread_params );
/* Make sure the threads have finalized */
if (thread_cleanup(hServerThread, 30) != 0) log(ERROR, "The server listener thread doesn't seem to exit cleanlly.");
if (thread_cleanup(hViewerThread, 30) != 0) log(ERROR, "The viewer listener thread doesn't seem to exit cleanlly.");
FinalizeSlots();
#ifdef WIN32
WinsockFinalize(); // Cleanup Winsock.
#endif
return 0;
}
