ACE3/extensions/clipboard/ace_clipboard.cpp

/*
 * ace_clipboard.cpp
 *
 * Takes a string and copies it to the clipboard; bypasses arma 8k clippy limit. Windows only.
 *
 * Takes:
 * Localized string as string
 *
 * Returns:
 * None
 */

#include "shared.hpp"

#include <stdlib.h>
#include <vector>
#include <string>

extern "C" {
    EXPORT void __stdcall RVExtension(char *output, int outputSize, const char *function);
}

std::string gClipboardData;

void __stdcall RVExtension(char *output, int outputSize, const char *function) {
    std::string cur_input(function);
    std::string result;

    ZERO_OUTPUT();

    if (cur_input.length() < 1) {
        EXTENSION_RETURN();
    }

    if (!strcmp(function, "version")) {
        strncpy_s(output, outputSize, ACE_FULL_VERSION_STR, _TRUNCATE);
        EXTENSION_RETURN();
    }

    #ifdef _WIN32

    if (!strcmp(function, "--COMPLETE--"))
    {
        if (OpenClipboard(NULL) == 0)
        {
            result = "OpenClipboard() failed, GetLastError=" + GetLastError();
        }
        else
        {
            if (EmptyClipboard() == 0)
            {
                result = "EmptyClipboard() failed, GetLastError=" + GetLastError();
            }
            else
            {
                // GPTR = GMEM_FIXED + GMEM_ZEROINIT, returns a ptr, no need for GlobalLock/GlobalUnlock
                char *pClipboardData = (char *)GlobalAlloc(GPTR, gClipboardData.length());
                if (pClipboardData == NULL)
                {
                    result = "GlobalAlloc() failed, GetLastError=" + GetLastError();
                    EXTENSION_RETURN();
                }

                strncpy_s(pClipboardData, gClipboardData.length(), gClipboardData.c_str(), _TRUNCATE);

                // if success, system owns the memory, if fail, free it from the heap
                if (SetClipboardData(CF_TEXT, pClipboardData) == NULL)
                {
                    result = "SetClipboardData() failed, GetLastError=" + GetLastError();

                    GlobalFree(pClipboardData);
                }
                else
                {
                    if (CloseClipboard() == 0)
                    {
                        result = "CloseClipboard() failed, GetLastError=" + GetLastError();
                    }
                }
            }
        }

        gClipboardData = "";
    }
    else
    {
        gClipboardData = gClipboardData + cur_input;
    }

    if (result.length() > 1)
    {
        strncpy_s(output, outputSize, result.c_str(), _TRUNCATE);
    }

    #endif

    EXTENSION_RETURN();
}