path: root/src/cnki.c

/*
 * Copyright (c) 2020-2022, yzrh <yzrh@noema.org>
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <stdlib.h>
#include <string.h>

#include "cnki.h"

int
cnki_create(cnki_t **param)
{
	if (*param != NULL)
		return 1;

	*param = malloc(sizeof(cnki_t));

	if (*param == NULL)
		return 1;

	(*param)->stat = 0;
	(*param)->size_buf = 524288;
	(*param)->fp_i = NULL;
	(*param)->fp_o = NULL;

	(*param)->file_stat = malloc(sizeof(file_stat_t));

	if ((*param)->file_stat== NULL)
		return 1;

	memset((*param)->file_stat, 0, sizeof(file_stat_t));

	(*param)->object_outline = NULL;
	(*param)->object_hn = NULL;

	return 0;
}

void
cnki_destroy(cnki_t **param)
{
	if (*param != NULL) {
		if ((*param)->file_stat != NULL)
			free((*param)->file_stat);

		object_outline_t *ptr_outline;
		while ((ptr_outline = (*param)->object_outline) != NULL) {
			(*param)->object_outline = (*param)->object_outline->next;
			free(ptr_outline);
		}

		object_hn_t *ptr_hn;
		while ((ptr_hn = (*param)->object_hn) != NULL) {
			(*param)->object_hn = (*param)->object_hn->next;
			free(ptr_hn->text);
			if (ptr_hn->image_data != NULL)
				for (int i = 0; i < ptr_hn->image_length; i++)
					free(ptr_hn->image_data[i].image);
			free(ptr_hn->image_data);
			free(ptr_hn);
		}

		free(*param);
	}
}

int
cnki_info(cnki_t **param)
{
	if (*param == NULL)
		return 1;

	if ((*param)->stat > 1)
		printf("Reading file header at 0x%x\n", ADDRESS_HEAD);

	int addr[2];
	unsigned char str[2];

	fseek((*param)->fp_i, ADDRESS_HEAD, SEEK_SET);
	fread((*param)->file_stat->type, 4, 1, (*param)->fp_i);

	fread(str, 2, 1, (*param)->fp_i);

	if ((*param)->stat > 0) {
		if ((unsigned char) (*param)->file_stat->type[0] > 0x7f)
			printf("File type is '%02x'\n", (unsigned char) (*param)->file_stat->type[0]);
		else
			printf("File type is '%s'\n", (*param)->file_stat->type);
	}

	if (strncmp((*param)->file_stat->type, "%PDF", 4) == 0) {
		return 0;
	} else if (strncmp((*param)->file_stat->type, "CAJ", 3) == 0) {
		addr[0] = ADDRESS_CAJ_PAGE;
		addr[1] = ADDRESS_CAJ_OUTLINE;
	} else if (strncmp((*param)->file_stat->type, "HN", 2) == 0) {
		addr[0] = ADDRESS_HN_PAGE;
		addr[1] = ADDRESS_HN_OUTLINE;
	} else if ((unsigned char) (*param)->file_stat->type[0] == 0xc8) {
		addr[0] = ADDRESS_C8_PAGE;
		addr[1] = ADDRESS_HN_OUTLINE;
	} else if (strncmp((*param)->file_stat->type, "KDH ", 4) == 0) {
		return 0;
	} else {
		return 1;
	}

	if ((*param)->stat > 1)
		printf("Reading page count at 0x%x\n", addr[0]);

	fseek((*param)->fp_i, addr[0], SEEK_SET);
	fread(&(*param)->file_stat->page, 4, 1, (*param)->fp_i);

	if ((*param)->stat > 0)
		printf("Advised %d page(s)\n",
			(*param)->file_stat->page);

	if (strncmp((*param)->file_stat->type, "HN", 2) == 0 && str[0] == 0xc8 && str[1] == 0x00) {
		fseek((*param)->fp_i, 0xd8, SEEK_SET);
		return 0;
	} else if ((unsigned char) (*param)->file_stat->type[0] == 0xc8) {
		fseek((*param)->fp_i, 0x50, SEEK_SET);
		return 0;
	}

	if ((*param)->stat > 1)
		printf("Reading outline count at 0x%x\n", addr[1]);

	fseek((*param)->fp_i, addr[1], SEEK_SET);
	fread(&(*param)->file_stat->outline, 4, 1, (*param)->fp_i);

	if ((*param)->stat > 0)
		printf("Advised %d outline(s)\n",
			(*param)->file_stat->outline);

	if ((*param)->file_stat->outline > 0) {
		if ((*param)->stat > 1) {
			printf("Loading outline(s)\n");
			printf("\t%19s\t%-24s\t%12s\t%12s\t%5s\n",
				"title",
				"hierarchy",
				"page",
				"text",
				"depth");
		}

		(*param)->object_outline = malloc(sizeof(object_outline_t));

		if ((*param)->object_outline == NULL)
			return 1;

		object_outline_t *ptr = (*param)->object_outline;
		for (int i = 0; i < (*param)->file_stat->outline; i++) {
			fread(ptr->title, 256, 1, (*param)->fp_i);
			fread(ptr->hierarchy, 24, 1, (*param)->fp_i);
			fread(ptr->page, 12, 1, (*param)->fp_i);
			fread(ptr->text, 12, 1, (*param)->fp_i);
			fread(&ptr->depth, 4, 1, (*param)->fp_i);

			ptr->next = NULL;

			if ((*param)->stat > 1) {
				printf("\t");
				for (int j = 1; j <= 256; j++) {
					printf("%02x", (unsigned char) ptr->title[j - 1]);

					if (j % 8 == 0 && ptr->title[j] == '\0')
						break;

					if (j % 8 == 0)
						printf("\n\t");
					else if (j % 2 == 0)
						printf(" ");
				}
				printf("\t%-24s\t%12s\t%12s\t%5d\n",
					ptr->hierarchy,
					ptr->page,
					ptr->text,
					ptr->depth);
			}

			if (i < (*param)->file_stat->outline - 1) {
				ptr->next = malloc(sizeof(object_outline_t));

				if (ptr->next == NULL)
					return 1;
			}

			ptr = ptr->next;
		}

		if ((*param)->stat > 0)
			printf("Loaded %d outline(s)\n",
				(*param)->file_stat->outline);
	}

	return 0;
}