unleashed-firmware/lib/toolbox/args.c

#include "args.h"
#include "hex.h"
#include "strint.h"
#include "m-core.h"
#include <errno.h>

size_t args_get_first_word_length(FuriString* args) {
    size_t ws = furi_string_search_char(args, ' ');
    if(ws == FURI_STRING_FAILURE) {
        ws = furi_string_size(args);
    }

    return ws;
}

size_t args_length(FuriString* args) {
    return furi_string_size(args);
}

bool args_read_int_and_trim(FuriString* args, int* value) {
    size_t cmd_length = args_get_first_word_length(args);

    if(cmd_length == 0) {
        return false;
    }

    int32_t temp;
    if(strint_to_int32(furi_string_get_cstr(args), NULL, &temp, 10) == StrintParseNoError) {
        *value = temp;
        furi_string_right(args, cmd_length);
        furi_string_trim(args);
        return true;
    }

    return false;
}

bool args_read_float_and_trim(FuriString* args, float* value) {
    size_t cmd_length = args_get_first_word_length(args);
    if(cmd_length == 0) {
        return false;
    }

    char* end_ptr;
    float temp = strtof(furi_string_get_cstr(args), &end_ptr);
    if(end_ptr == furi_string_get_cstr(args)) {
        return false;
    }

    *value = temp;
    furi_string_right(args, cmd_length);
    furi_string_trim(args);
    return true;
}

bool args_read_string_and_trim(FuriString* args, FuriString* word) {
    size_t cmd_length = args_get_first_word_length(args);

    if(cmd_length == 0) {
        return false;
    }

    furi_string_set_n(word, args, 0, cmd_length);
    furi_string_right(args, cmd_length);
    furi_string_trim(args);

    return true;
}

bool args_read_probably_quoted_string_and_trim(FuriString* args, FuriString* word) {
    if(furi_string_size(args) > 1 && furi_string_get_char(args, 0) == '\"') {
        size_t second_quote_pos = furi_string_search_char(args, '\"', 1);

        // Missing closing quote: furi_string_search_char returns FURI_STRING_FAILURE
        if(second_quote_pos == FURI_STRING_FAILURE) {
            return false;
        }

        furi_string_set_n(word, args, 1, second_quote_pos - 1);
        furi_string_right(args, second_quote_pos + 1);
        furi_string_trim(args);
        return true;
    } else {
        return args_read_string_and_trim(args, word);
    }
}

bool args_char_to_hex(char hi_nibble, char low_nibble, uint8_t* byte) {
    uint8_t hi_nibble_value = 0;
    uint8_t low_nibble_value = 0;
    bool result = false;

    if(hex_char_to_hex_nibble(hi_nibble, &hi_nibble_value)) {
        if(hex_char_to_hex_nibble(low_nibble, &low_nibble_value)) {
            result = true;
            *byte = (hi_nibble_value << 4) | low_nibble_value;
        }
    }

    return result;
}

bool args_read_hex_bytes(FuriString* args, uint8_t* bytes, size_t bytes_count) {
    bool result = true;
    const char* str_pointer = furi_string_get_cstr(args);

    if(args_get_first_word_length(args) == (bytes_count * 2)) {
        for(size_t i = 0; i < bytes_count; i++) {
            if(!args_char_to_hex(str_pointer[i * 2], str_pointer[i * 2 + 1], &(bytes[i]))) {
                result = false;
                break;
            }
        }
    } else {
        result = false;
    }

    return result;
}

bool args_read_duration(FuriString* args, uint32_t* value, const char* default_unit) {
    const char* args_cstr = furi_string_get_cstr(args);

    const char* unit;
    errno = 0;
    double duration_ms = strtod(args_cstr, (char**)&unit);
    if(errno) return false;
    if(duration_ms < 0) return false;
    if(unit == args_cstr) return false;

    if(strcmp(unit, "") == 0) {
        unit = default_unit;
        if(!unit) unit = "ms";
    }

    uint32_t multiplier;
    if(strcasecmp(unit, "ms") == 0) {
        multiplier = 1;
    } else if(strcasecmp(unit, "s") == 0) {
        multiplier = 1000;
    } else if(strcasecmp(unit, "m") == 0) {
        multiplier = 60 * 1000;
    } else if(strcasecmp(unit, "h") == 0) {
        multiplier = 60 * 60 * 1000;
    } else {
        return false;
    }

    const uint32_t max_pre_multiplication = UINT32_MAX / multiplier;
    if(duration_ms > max_pre_multiplication) return false;

    *value = round(duration_ms * multiplier);
    return true;
}