diff options
Diffstat (limited to 'src/pdf.c')
| -rw-r--r-- | src/pdf.c | 228 |
1 files changed, 228 insertions, 0 deletions
diff --git a/src/pdf.c b/src/pdf.c new file mode 100644 index 0000000..92dd717 --- /dev/null +++ b/src/pdf.c @@ -0,0 +1,228 @@ +/* + * Copyright (c) 2020, yzrh <yzrh@tuta.io> + * + * SPDX-License-Identifier: Apache-2.0 + */ + +#include <stdlib.h> +#include <string.h> + +#include "pdf.h" + +static int +_min_id(pdf_object_t **pdf) +{ + int min = 0; + + pdf_object_t *ptr = (*pdf)->next; + while (ptr != NULL) { + if (min == 0 || ptr->id < min) + min = ptr->id; + + ptr = ptr->next; + } + + return min; +} + +int +pdf_obj_create(pdf_object_t **pdf) +{ + if (*pdf != NULL) + return 1; + + *pdf = malloc(sizeof(pdf_object_t)); + + if (*pdf == NULL) + return 1; + + (*pdf)->address = 0; + (*pdf)->size = 0; + (*pdf)->id = 0; + (*pdf)->object_size = 0; + (*pdf)->object = NULL; + (*pdf)->dictionary_size = 0; + (*pdf)->dictionary = NULL; + (*pdf)->stream_size = 0; + (*pdf)->stream= NULL; + (*pdf)->next = NULL; + + return 0; +} + +void +pdf_obj_destroy(pdf_object_t **pdf) +{ + pdf_object_t *ptr; + while ((ptr = *pdf) != NULL) { + *pdf = (*pdf)->next; + free(ptr->object); + free(ptr->dictionary); + free(ptr->stream); + free(ptr); + } +} + +int +pdf_obj_add(pdf_object_t **pdf, int id, + const char * restrict object, + const char * restrict dictionary, + const char * restrict stream) +{ + if (*pdf != NULL || id <= 0 || + (object != NULL && dictionary != NULL)) + return 1; + + *pdf = malloc(sizeof(pdf_object_t)); + + if (*pdf == NULL) + return 1; + + (*pdf)->address = 0; + (*pdf)->size = 0; + + (*pdf)->id = id; + + if (dictionary != NULL) { + (*pdf)->dictionary_size = strlen(dictionary) + 1; + (*pdf)->dictionary = malloc((*pdf)->dictionary_size); + + if ((*pdf)->dictionary == NULL) + return 1; + + strncpy((*pdf)->dictionary, dictionary, (*pdf)->dictionary_size); + + (*pdf)->object_size = 0; + (*pdf)->object = NULL; + } else if (object != NULL) { + (*pdf)->object_size = strlen(object) + 1; + (*pdf)->object = malloc((*pdf)->object_size); + + if ((*pdf)->object == NULL) + return 1; + + strncpy((*pdf)->object, object, (*pdf)->object_size); + + (*pdf)->dictionary_size = 0; + (*pdf)->dictionary = NULL; + } else { + (*pdf)->object_size = 0; + (*pdf)->object = NULL; + (*pdf)->dictionary_size = 0; + (*pdf)->dictionary = NULL; + } + + if (stream != NULL) { + (*pdf)->stream_size = sizeof(stream); + (*pdf)->stream = malloc((*pdf)->stream_size); + + if ((*pdf)->stream == NULL) + return 1; + + memcpy((*pdf)->stream, stream, (*pdf)->stream_size); + } else { + (*pdf)->stream_size = 0; + (*pdf)->stream = NULL; + } + + (*pdf)->next = NULL; + + return 0; +} + +int +pdf_obj_del(pdf_object_t **pdf, int id) +{ + if (*pdf == NULL || id <= 0) + return 1; + + pdf_object_t *ptr = *pdf; + while (ptr->next != NULL) { + if (ptr->next->id == id) { + ptr->next = ptr->next->next; + break; + } + + ptr = ptr->next; + } + + return 0; +} + +int +pdf_obj_prepend(pdf_object_t **pdf, int id, + const char * restrict object, + const char * restrict dictionary, + const char * restrict stream) +{ + if (*pdf == NULL) + return 1; + + if (id <= 0) + id = pdf_get_free_id(pdf); + + pdf_object_t *ptr = NULL; + + if (pdf_obj_add(&ptr, id, object, dictionary, stream) != 0) { + free(ptr); + return 1; + } + + ptr->next = (*pdf)->next; + (*pdf)->next = ptr; + + return 0; +} + +int +pdf_obj_append(pdf_object_t **pdf, int id, + const char * restrict object, + const char * restrict dictionary, + const char * restrict stream) +{ + if (*pdf == NULL) + return 1; + + if (id <= 0) + id = pdf_get_free_id(pdf); + + pdf_object_t *ptr = *pdf; + while (ptr->next != NULL) + ptr = ptr->next; + + if (pdf_obj_add(&ptr->next, id, object, dictionary, stream) != 0) + return 1; + + return 0; +} + +int +pdf_obj_sort(pdf_object_t **pdf) +{ + if (*pdf == NULL) + return 1; + + int id; + pdf_object_t *tmp; + pdf_object_t *ptr; + + ptr = *pdf; + while (ptr->next != NULL) { + id = _min_id(&ptr->next); + + if (id == 0) + return 1; + + if (id < ptr->next->id) { + pdf_get_obj(&ptr->next, id, &tmp); + pdf_obj_del(&ptr->next, id); + + tmp->next = ptr->next; + ptr->next = tmp; + } + + ptr = ptr->next; + } + + return 0; +} |