summaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
Diffstat (limited to 'include')
-rw-r--r--include/kit.inl.h50
1 files changed, 44 insertions, 6 deletions
diff --git a/include/kit.inl.h b/include/kit.inl.h
index c50dd61..968d973 100644
--- a/include/kit.inl.h
+++ b/include/kit.inl.h
@@ -1259,6 +1259,7 @@ static kit_bigint_t kit_bi_from_base58(kit_str_t base58) {
*********************************************************************/
#ifndef KIT_INPUT_STREAM_H
#define KIT_INPUT_STREAM_H
+#include <stdio.h>
#ifdef __cplusplus
extern "C" {
#endif
@@ -1270,17 +1271,21 @@ typedef struct {
} kit_is_handle_t;
kit_is_handle_t kit_is_wrap_string(kit_str_t string,
kit_allocator_t alloc);
+kit_is_handle_t kit_is_wrap_file(FILE *f, kit_allocator_t alloc);
void kit_is_destroy(kit_is_handle_t in);
#define KIT_IS_WRAP_STRING(string) \
kit_is_wrap_string((string), kit_alloc_default())
+#define KIT_IS_WRAP_FILE(f) kit_is_wrap_file((f), kit_alloc_default())
#define KIT_IS_READ(in, destination) \
(in).read((in).state, (destination))
#ifndef KIT_DISABLE_SHORT_NAMES
# define is_read_fn kit_is_read_fn
# define is_handle_t kit_is_handle_t
# define is_wrap_string kit_is_wrap_string
+# define is_wrap_file kit_is_wrap_file
# define is_destroy kit_is_destroy
# define IS_WRAP_STRING KIT_IS_WRAP_STRING
+# define IS_WRAP_FILE KIT_IS_WRAP_FILE
# define IS_READ KIT_IS_READ
#endif
#ifdef __cplusplus
@@ -2644,7 +2649,7 @@ void kit_da_resize(kit_da_void_t *array, ptrdiff_t element_size,
* *
*********************************************************************/
#include <string.h>
-enum { input_stream_str };
+enum { KIT_INPUT_STREAM_STR, KIT_INPUT_STREAM_FILE };
typedef struct {
ptrdiff_t type;
kit_allocator_t alloc;
@@ -2654,12 +2659,17 @@ typedef struct {
kit_allocator_t alloc;
kit_str_t string;
} kit_is_state_str_t;
-static int check_type(void *state, ptrdiff_t type) {
+typedef struct {
+ ptrdiff_t type;
+ kit_allocator_t alloc;
+ FILE *file;
+} kit_is_state_file_t;
+static int kit_is_check_type_(void *state, ptrdiff_t type) {
kit_is_state_basic_t *basic = (kit_is_state_basic_t *) state;
return basic != NULL && basic->type == type;
}
-static ptrdiff_t read_str(void *state, kit_str_t destination) {
- if (!check_type(state, input_stream_str))
+static ptrdiff_t kit_read_str_(void *state, kit_str_t destination) {
+ if (!kit_is_check_type_(state, KIT_INPUT_STREAM_STR))
return 0;
kit_is_state_str_t *str = (kit_is_state_str_t *) state;
ptrdiff_t size = destination.size < str->string.size
@@ -2670,6 +2680,18 @@ static ptrdiff_t read_str(void *state, kit_str_t destination) {
str->string.size -= size;
return size;
}
+static ptrdiff_t kit_read_file_(void *state, kit_str_t destination) {
+ if (!kit_is_check_type_(state, KIT_INPUT_STREAM_FILE))
+ return 0;
+ kit_is_state_file_t *f = (kit_is_state_file_t *) state;
+ if (f->file == NULL || feof(f->file))
+ return 0;
+ ptrdiff_t size = (ptrdiff_t) fread(destination.values, 1,
+ destination.size, f->file);
+ if (size <= 0)
+ return 0;
+ return size;
+}
kit_is_handle_t kit_is_wrap_string(kit_str_t string,
kit_allocator_t alloc) {
kit_is_handle_t in;
@@ -2679,11 +2701,27 @@ kit_is_handle_t kit_is_wrap_string(kit_str_t string,
sizeof(kit_is_state_str_t), 0, NULL);
if (state != NULL) {
memset(state, 0, sizeof *state);
- state->type = input_stream_str;
+ state->type = KIT_INPUT_STREAM_STR;
state->string = string;
state->alloc = alloc;
in.state = state;
- in.read = read_str;
+ in.read = kit_read_str_;
+ }
+ return in;
+}
+kit_is_handle_t kit_is_wrap_file(FILE *f, kit_allocator_t alloc) {
+ kit_is_handle_t in;
+ memset(&in, 0, sizeof in);
+ kit_is_state_file_t *state = (kit_is_state_file_t *)
+ kit_alloc_dispatch(alloc, KIT_ALLOCATE,
+ sizeof(kit_is_state_file_t), 0, NULL);
+ if (state != NULL) {
+ memset(state, 0, sizeof *state);
+ state->type = KIT_INPUT_STREAM_FILE;
+ state->file = f;
+ state->alloc = alloc;
+ in.state = state;
+ in.read = kit_read_file_;
}
return in;
}