haiku.git

/*
 * Copyright 2002-2014, Axel Dörfler, axeld@pinc-software.de.
 * Distributed under the terms of the MIT License.
 */


#include <errno.h>
#include <fcntl.h>
#include <string.h>
#include <unistd.h>

#include <FindDirectory.h>
#include <StorageDefs.h>

#include <errno_private.h>
#include <find_directory_private.h>


static status_t
get_path(char *path, bool create)
{
	status_t status = __find_directory(B_SYSTEM_SETTINGS_DIRECTORY, -1, create,
		path, B_PATH_NAME_LENGTH);
	if (status != B_OK)
		return status;

	strlcat(path, "/network", B_PATH_NAME_LENGTH);
	if (create)
		mkdir(path, 0755);
	strlcat(path, "/hostname", B_PATH_NAME_LENGTH);
	return B_OK;
}


extern "C" int
sethostname(const char *hostName, size_t nameSize)
{
	if (nameSize > MAXHOSTNAMELEN) {
		__set_errno(EINVAL);
		return -1;
	}

	char path[B_PATH_NAME_LENGTH];
	if (get_path(path, false) != B_OK) {
		__set_errno(B_ERROR);
		return -1;
	}

	int file = open(path, O_WRONLY | O_CREAT, 0644);
	if (file < 0)
		return -1;

	ftruncate(file, nameSize + 1);

	if (write(file, hostName, nameSize) != (ssize_t)nameSize
		|| write(file, "\n", 1) != 1) {
		close(file);
		return -1;
	}

	close(file);
	return 0;
}


extern "C" int
gethostname(char *hostName, size_t nameSize)
{
	// look up hostname from network settings hostname file

	char path[B_PATH_NAME_LENGTH];
	if (get_path(path, false) != B_OK) {
		__set_errno(B_ERROR);
		return -1;
	}

	int file = open(path, O_RDONLY);
	if (file < 0)
		return -1;

	nameSize = min_c(nameSize, MAXHOSTNAMELEN);

	int length = read(file, hostName, nameSize - 1);
	close(file);

	if (length < 0)
		return -1;

	hostName[length] = '\0';

	char *end = strpbrk(hostName, "\r\n\t");
	if (end != NULL)
		end[0] = '\0';

	return 0;
}