// SPDX-License-Identifier: BSD-3-Clause OR Apache-2.0
#include "bundle6/create.h"

#include "ud3tn/bundle.h"
#include "ud3tn/common.h"

#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>

struct bundle *bundle6_create_local(
	void *payload, size_t payload_length,
	const char *source, const char *destination,
	uint64_t creation_time_ms, uint64_t sequence_number,
	uint64_t lifetime_ms, enum bundle_proc_flags proc_flags)
{
	struct bundle *bundle = bundle_init();

	if (bundle == NULL) {
		free(payload);
		return NULL;
	}

	bundle->protocol_version = 0x6;
	bundle->proc_flags = proc_flags | BUNDLE_V6_FLAG_SINGLETON_ENDPOINT;

	// Creation time
	bundle->creation_timestamp_ms = creation_time_ms;
	bundle->sequence_number = sequence_number;
	bundle->lifetime_ms = lifetime_ms;

	// Create payload block and block list
	bundle->payload_block = bundle_block_create(BUNDLE_BLOCK_TYPE_PAYLOAD);
	bundle->payload_block->flags = BUNDLE_V6_BLOCK_FLAG_LAST_BLOCK;
	bundle->blocks = bundle_block_entry_create(bundle->payload_block);

	if (bundle->payload_block == NULL || bundle->blocks == NULL)
		goto fail;

	bundle->source = strdup(source);
	if (bundle->source == NULL || strchr(source, ':') == NULL)
		goto fail;

	bundle->destination = strdup(destination);
	if (bundle->destination == NULL || strchr(destination, ':') == NULL)
		goto fail;

	bundle->report_to = strdup("dtn:none");
	bundle->current_custodian = strdup("dtn:none");

	bundle->payload_block->data = payload;
	bundle->payload_block->length = payload_length;

	if (bundle_recalculate_header_length(bundle) == UD3TN_FAIL) {
		bundle->payload_block->data = NULL; // prevent double-free
		goto fail;
	}

	return bundle;

fail:
	free(payload);
	bundle_free(bundle);
	return NULL;
}