diff options
Diffstat (limited to 'src/proone-mask.c')
| -rw-r--r-- | src/proone-mask.c | 122 |
1 files changed, 61 insertions, 61 deletions
diff --git a/src/proone-mask.c b/src/proone-mask.c index ba138a9..1807f3f 100644 --- a/src/proone-mask.c +++ b/src/proone-mask.c @@ -11,75 +11,75 @@ int main (const int argc, const char **args) { - int exit_code = 0; - ssize_t fd_read_size; - size_t read_size = 0; - uint8_t salt; - prne_dvault_mask_result_t mask_result; - prne_data_type_t type; + int exit_code = 0; + ssize_t fd_read_size; + size_t read_size = 0; + uint8_t salt; + prne_dvault_mask_result_t mask_result; + prne_data_type_t type; - prne_init_dvault_mask_result(&mask_result); + prne_init_dvault_mask_result(&mask_result); - if (argc <= 1) { - fprintf(stderr, - "Usage: %s <type>\n" - "<type>: 'cstr', 'bin'\n", - args[0]); - exit_code = 2; - goto END; - } + if (argc <= 1) { + fprintf(stderr, + "Usage: %s <type>\n" + "<type>: 'cstr', 'bin'\n", + args[0]); + exit_code = 2; + goto END; + } - type = prne_data_type_fstr(args[1]); - switch (type) { - case PRNE_DATA_TYPE_BIN: - case PRNE_DATA_TYPE_CSTR: { - static const size_t buf_size = 0x0000FFFF + 1; - uint8_t buf[buf_size]; + type = prne_data_type_fstr(args[1]); + switch (type) { + case PRNE_DATA_TYPE_BIN: + case PRNE_DATA_TYPE_CSTR: { + static const size_t buf_size = 0x0000FFFF + 1; + uint8_t buf[buf_size]; - do { - fd_read_size = read(STDIN_FILENO, buf + read_size, buf_size - read_size); - if (fd_read_size < 0) { - perror("Error reading stdin"); - exit_code = 1; - goto END; - } - if (fd_read_size > 0) { - read_size += fd_read_size; - if (read_size >= buf_size) { - fprintf(stderr, "Error: data too large\n"); - exit_code = 1; - goto END; - } - } - } while (fd_read_size > 0); + do { + fd_read_size = read(STDIN_FILENO, buf + read_size, buf_size - read_size); + if (fd_read_size < 0) { + perror("Error reading stdin"); + exit_code = 1; + goto END; + } + if (fd_read_size > 0) { + read_size += fd_read_size; + if (read_size >= buf_size) { + fprintf(stderr, "Error: data too large\n"); + exit_code = 1; + goto END; + } + } + } while (fd_read_size > 0); - if (read_size == 0) { - fprintf(stderr, "Error: no data read\n"); - exit_code = 1; - goto END; - } + if (read_size == 0) { + fprintf(stderr, "Error: no data read\n"); + exit_code = 1; + goto END; + } - getrandom(&salt, sizeof(salt), 0); + getrandom(&salt, sizeof(salt), 0); - mask_result = prne_dvault_mask(type, salt, read_size, buf); - if (mask_result.result == PRNE_DVAULT_MASK_OK) { - printf("(uint8_t*)\"%s\",\n", mask_result.str); - } - else { - fprintf(stderr, "Error: prne_dvault_mask() returned %d\n", (int)mask_result.result); - exit_code = 1; - goto END; - } - break; - } - default: - fprintf(stderr, "Error: unknown data type '%s'\n", args[1]); - exit_code = 2; - goto END; - } + mask_result = prne_dvault_mask(type, salt, read_size, buf); + if (mask_result.result == PRNE_DVAULT_MASK_OK) { + printf("(uint8_t*)\"%s\",\n", mask_result.str); + } + else { + fprintf(stderr, "Error: prne_dvault_mask() returned %d\n", (int)mask_result.result); + exit_code = 1; + goto END; + } + break; + } + default: + fprintf(stderr, "Error: unknown data type '%s'\n", args[1]); + exit_code = 2; + goto END; + } END: - prne_free_dvault_mask_result(&mask_result); + prne_free_dvault_mask_result(&mask_result); - return exit_code; + return exit_code; } |