/*

 * Copyright (C) 2008  Kristian Høgsberg <krh@redhat.com>

 * Copyright (C) 2008  Red Hat, Inc

 * Copyright (C) 2009, 2011-2012, 2014  J. Ali Harlow <ali@juiblex.co.uk>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License along

 * with this program; if not, write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#ifndef _RAZOR_INTERNAL_H_

#define _RAZOR_INTERNAL_H_

#ifdef MSWIN_API

#include <windows.h>

#endif

#include <stdlib.h>

#include <stdint.h>

#include <stdarg.h>

#include <unistd.h>

#include "razor.h"

#include "types/types.h"

/* GCC extensions */

#if defined(__GNUC__) && __GNUC__ >= 4

#define RAZOR_EXPORT __attribute__ ((visibility("default")))

#else

#define RAZOR_EXPORT

#endif

#define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))

#define PADDING(value, base) (-(value) & (base - 1))

#define ALIGN(value, base) ((value) + PADDING(value, base))

void *zalloc(size_t size);

struct razor_set_section {

	uint32_t name;

	uint32_t offset;

	uint32_t size;

};

struct razor_set_header {

	uint32_t magic;

	uint32_t version;

	uint32_t num_sections;

};

#define RAZOR_MAGIC 	0x525a4442

#define RAZOR_STRING_POOL		"string_pool"

#define RAZOR_PACKAGES			"packages"

#define RAZOR_PROPERTIES		"properties"

#define RAZOR_PACKAGE_POOL		"package_pool"

#define RAZOR_PROPERTY_POOL		"property_pool"

#define RAZOR_PREFIX_POOL		"prefix_pool"

#define RAZOR_DETAILS_STRING_POOL	"details_string_pool"

#define RAZOR_FILES			"files"

#define RAZOR_FILE_POOL			"file_pool"

#define RAZOR_FILE_STRING_POOL		"file_string_pool"

struct razor_script {

	uint32_t program;

	uint32_t body;

};

struct razor_package {

	unsigned int name  : 24;

	unsigned int flags : 8;

	uint32_t version;

	uint32_t arch;

	uint32_t summary;

	uint32_t description;

	uint32_t url;

	uint32_t license;

	struct list_head properties;

	struct list_head files;

	struct list_head install_prefixes;

	struct razor_script preun;

	struct razor_script postun;

};

struct razor_property {

	uint32_t name;

	uint32_t flags;

	uint32_t version;

	struct list_head packages;

};

struct razor_entry {

	uint32_t name  : 24;

	uint32_t flags : 8;

	uint32_t start;

	struct list_head packages;

};

#define RAZOR_ENTRY_LAST	0x80

struct razor_set {

	uint32_t header_version;

	struct array string_pool;

 	struct array packages;

 	struct array properties;

 	struct array files;

	struct array package_pool;

 	struct array property_pool;

 	struct array file_pool;

 	struct array prefix_pool;

	struct array file_string_pool;

	struct array details_string_pool;

	struct razor_mapped_file *mapped_files;

	int lock_fd, ref_count;

	enum razor_set_flags flags;

};

struct import_entry {

	uint32_t package;

	char *name;

};

struct import_directory {

	uint32_t name, count;

	struct array files;

	struct array packages;

	struct import_directory *last;

};

struct razor_importer {

	struct razor_set *set;

	struct hashtable table;

	struct hashtable file_table;

	struct hashtable details_table;

	struct razor_package *package;

	struct array properties;

	struct array files;

	struct array file_requires;

	struct array install_prefixes;

};

struct razor_package_iterator {

	struct razor_set *set;

	struct razor_package *package, *end;

	struct list *index;

	void *alloced_index;

};

void

razor_package_iterator_init_for_property(struct razor_package_iterator *pi,

					 struct razor_set *set,

					 struct razor_property *property);

struct razor_property_iterator {

	struct razor_set *set;

	struct razor_property *property, *end;

	struct list *index;

};

struct razor_file_iterator {

	struct razor_set *set;

	struct array path;

	struct list *index;

	int post_order;

};

struct razor_string_iterator {

	struct razor_set *set;

	struct list *index;

};

int

razor_set_aquire_lock(struct razor_set *set, const char *path, int exclusive);

struct razor_entry *

razor_set_find_entry(struct razor_set *set,

		     struct razor_entry *dir, const char *pattern);

struct razor_merger *

razor_merger_create(struct razor_set *set1, struct razor_set *set2);

void

razor_merger_add_package(struct razor_merger *merger,

			 struct razor_package *package);

struct razor_set *

razor_merger_commit(struct razor_merger *merger);

void

razor_merger_package_add_script(struct razor_merger *merger,

				struct razor_package *package,

				enum razor_property_flags script,

				const char *program, const char *body);

void

razor_merger_destroy(struct razor_merger *merger);

int run_lua_script(const char *root, const char *name, const char *body,

		   ssize_t len, int arg1);

int razor_run_script(const char *root, enum razor_property_flags script,

		     const char *program, const char *body, int arg1);

/* Utility functions */

void

razor_package_get_details_varg(struct razor_set *set,

			       struct razor_package *package,

			       va_list args);

void razor_rpm_get_details_varg(struct razor_rpm *rpm, va_list args);

int razor_create_dir(const char *root, const char *path);

int razor_remove(const char *path);

int razor_write(int fd, const void *data, size_t size);

void *

razor_file_get_contents(const char *filename, size_t *length, int private,

			struct razor_error **error);

int razor_file_free_contents(void *addr, size_t length);

typedef int (*razor_compare_with_data_func_t)(const void *p1,

					      const void *p,

					      void *data);

uint32_t *

razor_qsort_with_data(void *base, size_t nelem, size_t size,

		      razor_compare_with_data_func_t compare, void *data);

struct environment {

	int is_set;

	struct array vars, string_pool;

};

void environment_init(struct environment *env);

void environment_add_variable(struct environment *env,

			      const char *variable, const char *value);

void environment_set(struct environment *env);

void environment_unset(struct environment *env);

void environment_release(struct environment *env);

#ifdef MSWIN_API

char *razor_utf16_to_utf8(const wchar_t *utf16, int len);

wchar_t *razor_utf8_to_utf16(const char *utf8, int len);

#endif

/* Error functions */

struct razor_error {

	int domain;

	int code;

	char *object;

	char *str;

	char *obj_str;

	char *summary;

	char *msg;

};

#define razor_error_new_posix(object) \

	razor_error_new_str(RAZOR_POSIX_ERROR, errno, object, strerror(errno))

#define razor_set_error_posix(error, object) \

	if (error) \

		*(error) = razor_error_new_posix(object); \

	else

#ifdef MSWIN_API

struct razor_error *razor_error_new_mswin(const wchar_t *object, DWORD error);

struct razor_error *razor_error_new_str2(int domain, int code,

					 const wchar_t *object,

					 const char *str);

#define razor_set_error_mswin(error, object, err) \

	if (error) \

		*(error) = razor_error_new_mswin(object, err); \

	else

#define razor_set_error2(error, domain, code, object, str) \

	if (error) \

		*(error) = razor_error_new_str2(domain, code, object, str); \

	else

#endif	/* MSWIN_API */

/* Atomic functions */

#if HAVE_WINDOWS_KTM

struct razor_atomic {

	HANDLE transaction;

	int n_files;

	struct razor_atomic_file {

		wchar_t *path;

		HANDLE h;

	} *files;

	int in_undo;

	struct razor_error *error;

};

#elif ENABLE_ATOMIC

struct atomic_action {

	struct atomic_action *next;

	enum atomic_action_type {

		/* Complex actions */

		ACTION_MAKE_DIRS,

		ACTION_REMOVE,

		/* Primitive actions */

		ACTION_CREATE_DIR,

#if HAVE_SYMLINK

		ACTION_CREATE_SYMLINK,

#endif

		ACTION_MOVE,

	} type;

	struct {

		char *path;

		union atomic_action_args {

			struct {

				char *root;

			} make_dirs;

			struct {

				mode_t mode;

			} create_dir;

#if HAVE_SYMLINK

			struct {

				char *target;

			} create_symlink;

#endif

			struct {

				char *dest;

			} move;

		} u;

	} args;

};

struct razor_atomic {

	struct atomic_action *actions;

	char *description;

	char *toplevel;

	unsigned next_file_tag;

	int in_undo;

	struct razor_error *error;

};

char *atomic_action_attic_tmpnam(struct razor_atomic *atomic);

struct atomic_action *

atomic_action_list_prepend(struct atomic_action *list,

			   struct atomic_action *action);

struct atomic_action *atomic_action_new(enum atomic_action_type type);

void atomic_action_free(struct atomic_action *action);

struct atomic_action *atomic_action_list_reverse(struct atomic_action *list);

struct atomic_action *

atomic_action_do(struct razor_atomic *atomic, struct atomic_action *action);

void

atomic_action_undo(struct razor_atomic *atomic, struct atomic_action *action);

#else	/* !HAVE_WINDOWS_KTM && !ENABLE_ATOMIC */

struct razor_atomic {

	int in_undo;

	struct razor_error *error;

};

#endif

int razor_allow_all_root_names(void);

int razor_valid_root_name(const char *name);

#define RAZOR_ASCII_ISALPHA(c) \

	((c) >= 'A' && (c) <= 'Z' || (c) >= 'a' && (c) <= 'z')

#ifdef MSWIN_API

#define SKIP_DRIVE_LETTER(path) \

	((RAZOR_ASCII_ISALPHA(path[0]) && (path)[1] == ':') ? \

	   (path) + 2 : (path))

#else

#define SKIP_DRIVE_LETTER(path) (path)

#endif /* MSWIN_API */

#endif /* _RAZOR_INTERNAL_H_ */