toaruos/apps/hostname.c

37 lines
815 B
C

/**
* @brief hostname - Prints or sets the system hostname.
*
* @copyright
* This file is part of ToaruOS and is released under the terms
* of the NCSA / University of Illinois License - see LICENSE.md
* Copyright (C) 2013-2018 K. Lange
*/
#include <stdio.h>
#include <unistd.h>
#include <string.h>
int main(int argc, char * argv[]) {
if ((argc > 1 && argv[1][0] == '-') || (argc < 2)) {
char tmp[256] = {0};
gethostname(tmp, 255);
printf("%s\n", tmp);
return 0;
} else {
if (geteuid() != 0) {
fprintf(stderr,"Must be root to set hostname.\n");
return 1;
} else {
sethostname(argv[1], strlen(argv[1]));
FILE * file = fopen("/etc/hostname", "w");
if (!file) {
return 1;
} else {
fprintf(file, "%s\n", argv[1]);
fclose(file);
return 0;
}
}
}
return 0;
}