shell.c

#include "shell.h"
#include "constants.h"
 
wchar_t *escape_for_batch(const wchar_t *input, size_t n_chars) {
    if (input == nullptr || n_chars == 0) {
        errno = EINVAL;
        return nullptr;
    }
    unsigned i, j;
    size_t new_n_chars = 0;
    for (i = 0; i < n_chars; i++) {
        new_n_chars++;
        // Last condition is to handle REG_MULTI_SZ string sets
        if (input[i] == L'\\' || input[i] == L'%' || input[i] == L'"' ||
            (input[i] == L'\0' && i < (n_chars - 1))) {
            new_n_chars++;
        }
    }
    wchar_t *out = calloc(new_n_chars + 1, WL);
    if (!out) { // LCOV_EXCL_START
        return nullptr;
    } // LCOV_EXCL_STOP
    wmemset(out, L'\0', new_n_chars + 1);
    if (n_chars == new_n_chars) {
        wmemcpy(out, input, new_n_chars + 1);
        return out;
    }
    for (i = 0, j = 0; i < n_chars && j < new_n_chars; i++, j++) {
        // This condition is to handle REG_MULTI_SZ string sets
        if (input[i] == '\0' && i < (n_chars - 1) && j < (new_n_chars - 1)) {
            out[j] = '\\';
            out[++j] = '0';
        } else {
            out[j] = input[i];
            if (input[i] == L'"' || input[i] == L'%') {
                out[++j] = input[i];
            }
        }
    }
    return out;
}