/**
* Attempts to unload a module if loaded, for ten seconds before
* giving up
*
* @param driver The name of the driver (not a filename)
* @return 1 if the driver is succesfully unloaded, 0 otherwise
*/
int module_unload(char *driver) {
if (module_is_loaded(driver) == 1) {
bb_log(LOG_INFO, "Unloading %s driver\n", driver);
char *mod_argv[] = {
"rmmod",
"--wait",
driver,
NULL
};
bb_run_fork_wait(mod_argv, 10);
if (module_is_loaded(driver) == 1) {
bb_log(LOG_ERR, "Unloading %s driver timed out.\n", driver);
return 0;
}
}
return 1;
}
/**
* Attempts to load a module. If the module has not been loaded after ten
* seconds, give up
*
* @param module_name The filename of the module to be loaded
* @param driver The name of the driver to be loaded
* @return 1 if the driver is succesfully loaded, 0 otherwise
*/
int module_load(char *module_name, char *driver) {
if (module_is_loaded(driver) == 0) {
/* the module has not loaded yet, try to load it */
bb_log(LOG_INFO, "Loading driver %s (module %s)\n", driver, module_name);
char *mod_argv[] = {
"modprobe",
module_name,
NULL
};
bb_run_fork_wait(mod_argv, 10);
if (module_is_loaded(driver) == 0) {
bb_log(LOG_ERR, "Module %s could not be loaded (timeout?)\n", module_name);
return 0;
}
}
return 1;
}
int main(int argc, char* argv[]) {
/* Setup signal handling before anything else */
signal(SIGHUP, handle_signal);
signal(SIGTERM, handle_signal);
signal(SIGINT, handle_signal);
signal(SIGQUIT, handle_signal);
/* Initializing configuration */
init_config(argc, argv);
/* set runmode depending on leftover arguments */
if (optind >= argc) {
bb_status.runmode = BB_RUN_STATUS;
} else {
bb_status.runmode = BB_RUN_APP;
}
bb_init_log();
bb_log(LOG_DEBUG, "%s version %s starting...\n", bb_status.program_name, GITVERSION);
/* Connect to listening daemon */
bb_status.bb_socket = socketConnect(bb_config.socket_path, SOCK_NOBLOCK);
if (bb_status.bb_socket < 0) {
bb_log(LOG_ERR, "Could not connect to bumblebee daemon - is it running?\n");
bb_closelog();
return EXIT_FAILURE;
}
char buffer[BUFFER_SIZE];
int r;
/* Request status */
if (bb_status.runmode == BB_RUN_STATUS) {
r = snprintf(buffer, BUFFER_SIZE, "Status?");
socketWrite(&bb_status.bb_socket, buffer, r);
while (bb_status.bb_socket != -1) {
r = socketRead(&bb_status.bb_socket, buffer, BUFFER_SIZE);
if (r > 0) {
printf("Bumblebee status: %*s\n", r, buffer);
socketClose(&bb_status.bb_socket);
}
}
}
/* Run given application */
if (bb_status.runmode == BB_RUN_APP) {
int ranapp = 0;
r = snprintf(buffer, BUFFER_SIZE, "Checking availability...");
socketWrite(&bb_status.bb_socket, buffer, r);
while (bb_status.bb_socket != -1) {
r = socketRead(&bb_status.bb_socket, buffer, BUFFER_SIZE);
if (r > 0) {
bb_log(LOG_INFO, "Response: %*s\n", r, buffer);
switch (buffer[0]) {
case 'N': //No, run normally.
socketClose(&bb_status.bb_socket);
if (!bb_config.fallback_start){
bb_log(LOG_ERR, "Cannot access secondary GPU. Aborting.\n");
}
break;
case 'Y': //Yes, run through vglrun
bb_log(LOG_INFO, "Running application through vglrun.\n");
ranapp = 1;
//run vglclient if any method other than proxy is used
if (strncmp(bb_config.vgl_compress, "proxy", BUFFER_SIZE) != 0) {
char * vglclient_args[] = {
"vglclient",
"-detach",
0
};
bb_run_fork(vglclient_args);
}
char ** vglrun_args = malloc(sizeof (char *) * (9 + argc - optind));
vglrun_args[0] = "vglrun";
vglrun_args[1] = "-c";
vglrun_args[2] = bb_config.vgl_compress;
vglrun_args[3] = "-d";
vglrun_args[4] = bb_config.x_display;
vglrun_args[5] = "-ld";
vglrun_args[6] = bb_config.ld_path;
vglrun_args[7] = "--";
for (r = 0; r < argc - optind; r++) {
vglrun_args[8 + r] = argv[optind + r];
}
vglrun_args[8 + r] = 0;
bb_run_fork_wait(vglrun_args);
socketClose(&bb_status.bb_socket);
break;
default: //Something went wrong - output and exit.
bb_log(LOG_ERR, "Problem: %*s\n", r, buffer);
socketClose(&bb_status.bb_socket);
break;
}
}
}
if (!ranapp && bb_config.fallback_start) {
bb_log(LOG_WARNING, "Running application normally.\n");
bb_run_exec(argv + optind);
}
}
bb_closelog();
bb_stop_all(); //stop any started processes that are left
return (EXIT_SUCCESS);
}
