boinc/samples/vboxwrapper/deprecated/vm.cpp

142 lines
4.0 KiB
C++

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2010 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
#ifdef _WIN32
#include "boinc_win.h"
#include "win_util.h"
#else
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <stdio.h>
#include <vector>
#include <string>
#include <unistd.h>
#endif
#include "diagnostics.h"
#include "filesys.h"
#include "parse.h"
#include "str_util.h"
#include "str_replace.h"
#include "util.h"
#include "error_numbers.h"
#include "procinfo.h"
#include "boinc_api.h"
#include "vbox.h"
#include "vm.h"
VM vm;
VM::VM() {
stdin_filename.clear();
stdout_filename.clear();
stderr_filename.clear();
vm_os_name.clear();
vm_os_version.clear();
vm_memory_size = 0;
vm_disk_image_name.clear();
vm_disk_image_type.clear();
vm_shared_folder_name.clear();
vm_shared_folder_dir_name.clear();
suspended = false;
enable_network = false;
enable_shared_directory = false;
}
VM::~VM() {
}
int VM::parse(XML_PARSER& xp) {
char tag[1024], buf[8192];
bool is_tag;
while (!xp.get(tag, sizeof(tag), is_tag)) {
if (!is_tag) {
fprintf(stderr, "%s VM::parse(): unexpected text %s\n",
boinc_msg_prefix(buf, sizeof(buf)), tag
);
continue;
}
if (!strcmp(tag, "/vm")) {
return 0;
}
else if (xp.parse_string(tag, "stdin_filename", stdin_filename)) continue;
else if (xp.parse_string(tag, "stdout_filename", stdout_filename)) continue;
else if (xp.parse_string(tag, "stderr_filename", stderr_filename)) continue;
else if (xp.parse_string(tag, "vm_os_name", vm_os_name)) continue;
else if (xp.parse_string(tag, "vm_os_version", vm_os_version)) continue;
else if (xp.parse_int(tag, "vm_memory_size", vm_memory_size)) continue;
else if (xp.parse_string(tag, "vm_disk_image_name", vm_disk_image_name)) continue;
else if (xp.parse_string(tag, "vm_disk_image_type", vm_disk_image_type)) continue;
else if (xp.parse_string(tag, "vm_shared_folder_name", vm_shared_folder_name)) continue;
else if (xp.parse_string(tag, "vm_shared_folder_dir_name", vm_shared_folder_dir_name)) continue;
else if (xp.parse_bool(tag, "enable_network", enable_network)) continue;
else if (xp.parse_bool(tag, "enable_shared_directory", enable_shared_directory)) continue;
}
return ERR_XML_PARSE;
}
int VM::run() {
int retval;
retval = virtualbox_initialize();
if (retval) return retval;
if (!virtualbox_vm_is_registered()) {
retval = virtualbox_register_vm();
if (retval) return retval;
}
retval = virtualbox_startvm();
if (retval) return retval;
return 0;
}
void VM::poll() {
virtualbox_monitor();
}
void VM::stop() {
virtualbox_stopvm();
virtualbox_cleanup();
}
void VM::pause() {
virtualbox_pausevm();
suspended = true;
}
void VM::resume() {
virtualbox_resumevm();
suspended = false;
}
void VM::cleanup() {
virtualbox_stopvm();
virtualbox_deregister_vm();
virtualbox_cleanup();
// Give time enough for external processes to finish the cleanup process
boinc_sleep(5.0);
}
bool VM::is_running() {
return (bool)virtualbox_vm_is_running();
}