static gboolean
PowerOpsRunScript(PowerOpState *state,
gchar *script)
{
gchar *quoted = NULL;
ProcMgr_ProcArgs procArgs;
/*
* Pass the CREATE_NO_WINDOW flag to CreateProcess so that the
* cmd.exe window will not be visible to the user in the guest.
*/
memset(&procArgs, 0, sizeof procArgs);
procArgs.bInheritHandles = TRUE;
procArgs.dwCreationFlags = CREATE_NO_WINDOW;
/* Quote the path if it's not yet quoted. */
if (script[0] != '"') {
quoted = g_strdup_printf("\"%s\"", script);
}
g_message("Executing script: %s\n", (quoted != NULL) ? quoted : script);
state->pid = ProcMgr_ExecAsync((quoted != NULL) ? quoted : script, &procArgs);
g_free(quoted);
if (state->pid != NULL) {
HANDLE h = ProcMgr_GetAsyncProcSelectable(state->pid);
GSource *watch = VMTools_NewHandleSource(h);
VMTOOLSAPP_ATTACH_SOURCE(state->ctx, watch, PowerOpsScriptCallback, state, NULL);
g_source_unref(watch);
return TRUE;
} else {
g_warning("Failed to start script: out of memory?\n");
return FALSE;
}
}
static gboolean
GuestInfoVMSupport(RpcInData *data)
{
#if defined(_WIN32)
char vmSupportCmd[] = "vm-support.vbs";
char *vmSupportPath = NULL;
gchar *vmSupport = NULL;
SECURITY_ATTRIBUTES saProcess = {0}, saThread = {0};
ProcMgr_AsyncProc *vmSupportProc = NULL;
ProcMgr_ProcArgs vmSupportProcArgs = {0};
/*
* Construct the commandline to be passed during execution
* This will be the path of our vm-support.vbs
*/
vmSupportPath = GuestApp_GetInstallPath();
if (vmSupportPath == NULL) {
return RPCIN_SETRETVALS(data,
"GuestApp_GetInstallPath failed", FALSE);
}
/* Put together absolute vm-support filename. */
vmSupport = g_strdup_printf("cscript \"%s%s%s\" -u",
vmSupportPath, DIRSEPS, vmSupportCmd);
vm_free(vmSupportPath);
saProcess.nLength = sizeof saProcess;
saProcess.bInheritHandle = TRUE;
saThread.nLength = sizeof saThread;
vmSupportProcArgs.lpProcessAttributes = &saProcess;
vmSupportProcArgs.lpThreadAttributes = &saThread;
vmSupportProcArgs.dwCreationFlags = CREATE_NO_WINDOW;
g_message("Starting vm-support script - %s\n", vmSupport);
vmSupportProc = ProcMgr_ExecAsync(vmSupport, &vmSupportProcArgs);
g_free(vmSupport);
if (vmSupportProc == NULL) {
g_warning("Error starting vm-support script\n");
return RPCIN_SETRETVALS(data,
"Error starting vm-support script", FALSE);
}
ProcMgr_Free(vmSupportProc);
return RPCIN_SETRETVALS(data, "", TRUE);
#else
gchar *vmSupportCmdArgv[] = {"vm-support", "-u", NULL};
g_message("Starting vm-support script - %s\n", vmSupportCmdArgv[0]);
if (!g_spawn_async(NULL, vmSupportCmdArgv, NULL,
G_SPAWN_SEARCH_PATH |
G_SPAWN_STDOUT_TO_DEV_NULL |
G_SPAWN_STDERR_TO_DEV_NULL,
NULL, NULL, NULL, NULL)) {
g_warning("Error starting vm-support script\n");
return RPCIN_SETRETVALS(data,
"Error starting vm-support script", FALSE);
}
return RPCIN_SETRETVALS(data, "", TRUE);
#endif
}
static int
VmBackupRunNextScript(VmBackupScriptOp *op) // IN/OUT
{
const char *scriptOp;
int ret = 0;
ssize_t index;
VmBackupScript *scripts = op->state->scripts;
switch (op->type) {
case VMBACKUP_SCRIPT_FREEZE:
index = ++op->state->currentScript;
scriptOp = "freeze";
break;
case VMBACKUP_SCRIPT_FREEZE_FAIL:
index = --op->state->currentScript;
scriptOp = "freezeFail";
break;
case VMBACKUP_SCRIPT_THAW:
index = --op->state->currentScript;
scriptOp = "thaw";
break;
default:
NOT_REACHED();
}
while (index >= 0 && scripts[index].path != NULL) {
char *cmd;
if (File_IsFile(scripts[index].path)) {
if (op->state->scriptArg != NULL) {
cmd = Str_Asprintf(NULL, "\"%s\" %s \"%s\"", scripts[index].path,
scriptOp, op->state->scriptArg);
} else {
cmd = Str_Asprintf(NULL, "\"%s\" %s", scripts[index].path,
scriptOp);
}
if (cmd != NULL) {
g_debug("Running script: %s\n", cmd);
scripts[index].proc = ProcMgr_ExecAsync(cmd, NULL);
} else {
g_debug("Failed to allocate memory to run script: %s\n",
scripts[index].path);
scripts[index].proc = NULL;
}
vm_free(cmd);
if (scripts[index].proc == NULL) {
if (op->type == VMBACKUP_SCRIPT_FREEZE) {
ret = -1;
break;
} else {
op->thawFailed = TRUE;
}
} else {
ret = 1;
break;
}
}
if (op->type == VMBACKUP_SCRIPT_FREEZE) {
index = ++op->state->currentScript;
} else {
index = --op->state->currentScript;
}
/*
* This happens if all thaw/fail scripts failed to start. Since the first
* entry may be a legacy script (which may not exist), need to check
* whether the interesting failure is the first or the second entry in
* the script list.
*/
if (index == -1) {
size_t failIdx = 0;
if (!File_IsFile(scripts[0].path)) {
failIdx = 1;
}
if (scripts[failIdx].proc == NULL && scripts[failIdx].path != NULL) {
ret = -1;
}
}
}
return ret;
}
