#include "olm/olm.hh" #include <stdio.h> #include <stdlib.h> #include <unistd.h> #include <stddef.h> #include <string.h> #include <sys/types.h> #include <sys/stat.h> #include <fcntl.h> ssize_t read_file( int fd, uint8_t **buffer ) { size_t buffer_size = 4096; uint8_t * current_buffer = (uint8_t *) malloc(buffer_size); if (current_buffer == NULL) return -1; size_t buffer_pos = 0; while (1) { ssize_t count = read( fd, current_buffer + buffer_pos, buffer_size - buffer_pos ); if (count < 0) break; if (count == 0) { uint8_t * return_buffer = (uint8_t *) realloc(current_buffer, buffer_pos); if (return_buffer == NULL) break; *buffer = return_buffer; return buffer_pos; } buffer_pos += count; if (buffer_pos == buffer_size) { buffer_size *= 2; uint8_t * new_buffer = (uint8_t *) realloc(current_buffer, buffer_size); if (new_buffer == NULL) break; current_buffer = new_buffer; } } free(current_buffer); return -1; } template<typename T> T check_errno( const char * message, T value ) { if (value == T(-1)) { perror(message); exit(1); } return value; } template<typename T, typename F> size_t check_error( F f, T * object, const char * message, size_t value ) { if (value == olm_error()) { const char * olm_message = f(object); ssize_t ignored; ignored = write(STDERR_FILENO, message, strlen(message)); ignored = write(STDERR_FILENO, ": ", 2); ignored = write(STDERR_FILENO, olm_message, strlen(olm_message)); ignored = write(STDERR_FILENO, "\n", 1); exit(2); return ignored; } return value; } size_t check_session( OlmSession * session, const char * message, size_t value ) { return check_error(olm_session_last_error, session, message, value); }