helper.c

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include "vix.h"
#include "helper.h"
#include "_cgo_export.h"

VixError vix_job_wait(VixHandle jobHandle) {
    return VixJob_Wait(jobHandle, VIX_PROPERTY_NONE);
}

VixError get_vix_handle(
    VixHandle jobHandle,
    VixPropertyID prop1,
    VixHandle* handle,
    VixPropertyID prop2) {

    return VixJob_Wait(jobHandle,
        prop1,
        handle,
        prop2);
}

VixError alloc_vm_pwd_proplist(
    VixHandle handle,
    VixHandle* resultHandle,
    char* password
) {
    return VixPropertyList_AllocPropertyList(handle,
                                        resultHandle,
                                        VIX_PROPERTY_VM_ENCRYPTION_PASSWORD,
                                        password,
                                        VIX_PROPERTY_NONE);
}

VixError get_screenshot_bytes(
    VixHandle handle,
    int* byte_count,
    char* screen_bits) {

    return VixJob_Wait(handle,
        VIX_PROPERTY_JOB_RESULT_SCREEN_IMAGE_DATA,
        byte_count, screen_bits,
        VIX_PROPERTY_NONE);
}

VixError get_num_shared_folders(VixHandle jobHandle, int* numSharedFolders) {
    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_SHARED_FOLDER_COUNT,
        numSharedFolders,
        VIX_PROPERTY_NONE);
}

VixError read_variable(VixHandle jobHandle, char** readValue) {
    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_VM_VARIABLE_STRING,
        readValue,
        VIX_PROPERTY_NONE);
}

VixError get_temp_filepath(VixHandle jobHandle, char* tempFilePath) {
    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_ITEM_NAME,
        tempFilePath,
        VIX_PROPERTY_NONE);
}

VixError is_file_or_dir(VixHandle jobHandle, int* result) {
    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_GUEST_OBJECT_EXISTS,
        result,
        VIX_PROPERTY_NONE);
}

VixError get_program_output(
    VixHandle jobHandle,
    uint64* pid,
    int* elapsedTime,
    int* exitCode) {
    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_PROCESS_ID,
        pid,
        VIX_PROPERTY_JOB_RESULT_GUEST_PROGRAM_ELAPSED_TIME,
        elapsedTime,
        VIX_PROPERTY_JOB_RESULT_GUEST_PROGRAM_EXIT_CODE,
        exitCode,
        VIX_PROPERTY_NONE);
}

VixError get_shared_folder(
    VixHandle jobHandle,
    char* folderName,
    char* folderHostPath,
    int* folderFlags) {

    return VixJob_Wait(jobHandle,
        VIX_PROPERTY_JOB_RESULT_ITEM_NAME, folderName,
        VIX_PROPERTY_JOB_RESULT_SHARED_FOLDER_HOST, folderHostPath,
        VIX_PROPERTY_JOB_RESULT_SHARED_FOLDER_FLAGS, folderFlags,
        VIX_PROPERTY_NONE);
}

VixError get_property(VixHandle handle, VixPropertyID id, void* value) {
    return Vix_GetProperties(handle, id, value, VIX_PROPERTY_NONE);
}

VixError get_vm_url(char* url, VixHandle moreEvtInfo) {
    return  Vix_GetProperties(  moreEvtInfo,
                                VIX_PROPERTY_FOUND_ITEM_LOCATION,
                                url,
                                VIX_PROPERTY_NONE);
}

VixError get_file_info(VixHandle jobHandle,
                     int64* fsize,
                     int* flags,
                     int64* modtime) {

    return Vix_GetProperties(jobHandle,
                        VIX_PROPERTY_JOB_RESULT_FILE_SIZE,
                        fsize,
                        VIX_PROPERTY_JOB_RESULT_FILE_FLAGS,
                        flags,
                        VIX_PROPERTY_JOB_RESULT_FILE_MOD_TIME,
                        modtime,
                        VIX_PROPERTY_NONE);
}

void find_items_callback(
    VixHandle jobHandle,
    VixEventType eventType,
    VixHandle moreEventInfo,
    void* hostPtr) {

    VixError err = VIX_OK;
    char* url = NULL;

    //Check callback event; ignore progress reports.
    if (VIX_EVENTTYPE_FIND_ITEM != eventType) {
        return;
    }

    // Found a virtual machine.
    err = Vix_GetProperties(moreEventInfo,
                            VIX_PROPERTY_FOUND_ITEM_LOCATION,
                            &url,
                            VIX_PROPERTY_NONE);

    if (VIX_OK != err) {
        // TODO(c4milo): forward the error to Go land so it can be handled properly.
        printf("Error %s\n", Vix_GetErrorText(err, NULL));
    }

    addItem(hostPtr, url);

    Vix_FreeBuffer(url);
}

VixError get_guest_file(VixHandle jobHandle, int i, char* name, int64* size, int64* modtime, int* flags) {
    return VixJob_GetNthProperties( jobHandle,
                                    i,
                                    VIX_PROPERTY_JOB_RESULT_ITEM_NAME, name,
                                    VIX_PROPERTY_JOB_RESULT_FILE_SIZE, size,
                                    VIX_PROPERTY_JOB_RESULT_FILE_FLAGS, flags,
                                    VIX_PROPERTY_JOB_RESULT_FILE_MOD_TIME, modtime,
                                    VIX_PROPERTY_NONE);
}

VixError get_guest_process(VixHandle jobHandle, int i, char* name, uint64* pid, char* owner, char* cmdline, Bool* is_debugged, int* start_time) {
    return VixJob_GetNthProperties( jobHandle,
                                    i,
                                    VIX_PROPERTY_JOB_RESULT_ITEM_NAME, name,
                                    VIX_PROPERTY_JOB_RESULT_PROCESS_ID, pid,
                                    VIX_PROPERTY_JOB_RESULT_PROCESS_OWNER, owner,
                                    VIX_PROPERTY_JOB_RESULT_PROCESS_COMMAND, cmdline,
                                    VIX_PROPERTY_JOB_RESULT_PROCESS_BEING_DEBUGGED, is_debugged,
                                    VIX_PROPERTY_JOB_RESULT_PROCESS_START_TIME, start_time,
                                    VIX_PROPERTY_NONE);
}