/*
 * Copyright (c) 2015, 2021 Jonas 'Sortie' Termansen.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 * harddisk.c
 * Harddisk abstraction.
 */

#include <sys/types.h>

#include <errno.h>
#include <fcntl.h>
#include <inttypes.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>

#include <mount/harddisk.h>

#include "util.h"

static char* atcgetblob(int fd, const char* name, size_t* size_ptr)
{
	ssize_t size = tcgetblob(fd, name, NULL, 0);
	if ( size < 0 )
		return NULL;
	char* result = (char*) malloc((size_t) size + 1);
	if ( !result )
		return NULL;
	ssize_t second_size = tcgetblob(fd, name, result, (size_t) size);
	if ( second_size != size )
		return free(result), (char*) NULL;
	result[(size_t) size] = '\0';
	if ( size_ptr )
		*size_ptr = (size_t) size;
	return result;
}

struct harddisk* harddisk_openat(int dirfd, const char* relpath, int rdwr)
{
	struct harddisk* hd = CALLOC_TYPE(struct harddisk);
	if ( !hd )
		return (struct harddisk*) NULL;
	memset(&hd->bdev, 0, sizeof(hd->bdev));
	hd->bdev.hd = hd;
	if ( (hd->fd = openat(dirfd, relpath, rdwr)) < 0 )
		return harddisk_close(hd), (struct harddisk*) NULL;
	if ( !(hd->path = canonicalize_file_name_at(dirfd, relpath)) )
		return harddisk_close(hd), (struct harddisk*) NULL;
	if ( fstat(hd->fd, &hd->st) < 0 )
		return harddisk_close(hd), (struct harddisk*) NULL;
	return hd;
}

bool harddisk_inspect_blockdevice(struct harddisk* hd)
{
	if ( !S_ISBLK(hd->st.st_mode) )
		return errno = ENOTBLK, false;
	if ( tcgetblob(hd->fd, "harddisk-driver", NULL, 0) < 0 )
	{
		if ( errno == ENOENT )
			errno = ENOTBLK;
		return false;
	}
	if ( !(hd->driver = atcgetblob(hd->fd, "harddisk-driver", NULL)) ||
	     !(hd->model = atcgetblob(hd->fd, "harddisk-model", NULL)) ||
	     !(hd->serial = atcgetblob(hd->fd, "harddisk-serial", NULL)) )
		return harddisk_close(hd), false;
	char* str;
	if ( !(str = atcgetblob(hd->fd, "harddisk-block-size", NULL)) )
		return harddisk_close(hd), false;
	hd->logical_block_size = (blksize_t) strtoumax(str, NULL, 10);
	free(str);
	if ( !(str = atcgetblob(hd->fd, "harddisk-cylinders", NULL)) )
		return harddisk_close(hd), false;
	hd->cylinders = (uint16_t) strtoul(str, NULL, 10);
	free(str);
	if ( !(str = atcgetblob(hd->fd, "harddisk-heads", NULL)) )
		return harddisk_close(hd), false;
	hd->heads = (uint16_t) strtoul(str, NULL, 10);
	free(str);
	if ( !(str = atcgetblob(hd->fd, "harddisk-sectors", NULL)) )
		return harddisk_close(hd), false;
	hd->sectors = (uint16_t) strtoul(str, NULL, 10);
	free(str);
	if ( !hd->logical_block_size )
		return errno = ENOMEDIUM, false;
	// TODO: To avoid potential overflow bugs (assuming malicious filesystem),
	//       reject ridiculous block sizes.
	if ( 65536 < hd->logical_block_size )
		return harddisk_close(hd), errno = EINVAL, false;
	if ( hd->logical_block_size < 512 )
		return harddisk_close(hd), errno = EINVAL, false;
	return true;
}

void harddisk_close(struct harddisk* hd)
{
	if ( 0 <= hd->fd )
		close(hd->fd);
	free(hd->path);
	free(hd->driver);
	free(hd->model);
	free(hd->serial);
	free(hd);
}