sergiotarxz/nuttx-apps
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

system/hostname: Add an option to read the hostname from a file

Browse Source
This commit is contained in:
Norman Rasmussen 2022-07-03 13:57:21 -07:00 committed by Xiang Xiao
parent 71ea8b052d
commit 5c84c47cad
1 changed files with 85 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

92
system/hostname/hostname_main.c
View File

@ -35,6 +35,19 @@
* Private Data
****************************************************************************/
/****************************************************************************
* Private Functions
****************************************************************************/
/****************************************************************************
* Name: show_usage
****************************************************************************/
static void show_usage(FAR const char *progname)
{
printf("Usage: %s [<hostname>|-F <file>]\n", progname);
}
/****************************************************************************
* Public Functions
****************************************************************************/
@ -42,20 +55,85 @@
int main(int argc, FAR char *argv[])
{
int ret;
char hostname[HOST_NAME_MAX + 1];
bool set = false;
FAR FILE *f;
char hostname[HOST_NAME_MAX + 2];
if (argc > 1)
while ((ret = getopt(argc, argv, ":F:h")) != ERROR)
{
if (argc > 2)
switch (ret)
{
fprintf(stderr, "ERROR: Too many arguments\n");
case 'F':
set = true;
f = fopen(optarg, "r");
if (f == NULL)
{
fprintf(stderr, "ERROR: Failed to open '%s': %s\n", optarg,
strerror(errno));
return EXIT_FAILURE;
}
ret = sethostname(argv[1], strlen(argv[1]));
if (fgets(hostname, sizeof(hostname), f) == NULL)
{
if (errno != 0)
{
fprintf(stderr, "ERROR: Failed to read '%s': %s\n",
optarg, strerror(errno));
fclose(f);
return EXIT_FAILURE;
}
hostname[0] = '\0';
}
else
{
*strchrnul(hostname, '\n') = '\0';
}
fclose(f);
break;
case 'h':
show_usage(argv[0]);
return EXIT_SUCCESS;
case ':':
fprintf(stderr, "ERROR: Option needs a value: '%c'\n", optopt);
show_usage(argv[0]);
return EXIT_FAILURE;
default:
case '?':
fprintf(stderr, "ERROR: Unrecognized option: '%c'\n", optopt);
show_usage(argv[0]);
return EXIT_FAILURE;
}
}
if (optind < argc && !set)
{
set = true;
strlcpy(hostname, argv[optind++], sizeof(hostname));
}
if (optind < argc)
{
fprintf(stderr, "ERROR: Too many arguments\n");
return -1;
}
if (set)
{
if (strlen(hostname) == 0 || strlen(hostname) > HOST_NAME_MAX)
{
fprintf(stderr, "ERROR: The specified hostname is invalid\n");
return EXIT_FAILURE;
}
ret = sethostname(hostname, strlen(hostname));
if (ret != 0)
{
printf("sethostname() returned %d\n", ret);
fprintf(stderr, "sethostname() returned %d\n", ret);
return EXIT_FAILURE;
}
}
@ -64,7 +142,7 @@ int main(int argc, FAR char *argv[])
ret = gethostname(hostname, HOST_NAME_MAX);
if (ret != 0)
{
printf("gethostname() returned %d\n", ret);
fprintf(stderr, "gethostname() returned %d\n", ret);
return EXIT_FAILURE;
}