NetBSD/sbin/gpt/label.c

160 lines
3.7 KiB
C
Raw Normal View History

2006-10-16 02:34:15 +04:00
/*-
* Copyright (c) 2005 Marcel Moolenaar
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
2014-09-30 00:28:57 +04:00
#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif
2006-10-16 02:34:15 +04:00
#include <sys/cdefs.h>
2006-10-16 02:36:29 +04:00
#ifdef __FBSDID
2006-10-16 02:34:15 +04:00
__FBSDID("$FreeBSD: src/sbin/gpt/label.c,v 1.3 2006/10/04 18:20:25 marcel Exp $");
2006-10-16 02:36:29 +04:00
#endif
#ifdef __RCSID
2015-12-04 04:47:48 +03:00
__RCSID("$NetBSD: label.c,v 1.28 2015/12/04 01:47:48 christos Exp $");
2006-10-16 02:36:29 +04:00
#endif
2006-10-16 02:34:15 +04:00
#include <sys/types.h>
#include <err.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "map.h"
#include "gpt.h"
#include "gpt_private.h"
#include "gpt_uuid.h"
2006-10-16 02:34:15 +04:00
static int cmd_label(gpt_t, int, char *[]);
static const char *labelhelp[] = {
"-a <-l label | -f file>",
"[-b blocknr] [-i index] [-L label] [-s sectors] [-t uuid] "
"<-l label | -f file>",
};
struct gpt_cmd c_label = {
"label",
cmd_label,
labelhelp, __arraycount(labelhelp),
0,
};
#define usage() gpt_usage(NULL, &c_label)
2006-10-16 02:34:15 +04:00
2015-12-01 22:25:24 +03:00
static void
change(struct gpt_ent *ent, void *v)
2006-10-16 02:34:15 +04:00
{
2015-12-01 22:25:24 +03:00
uint8_t *name = v;
utf8_to_utf16(name, ent->ent_name, __arraycount(ent->ent_name));
2006-10-16 02:34:15 +04:00
}
static int
name_from_file(gpt_t gpt, void *v)
2006-10-16 02:34:15 +04:00
{
FILE *f;
char *p;
size_t maxlen = 1024;
size_t len;
const char *fn = optarg;
char **name = v;
if (*name != NULL)
return -1;
2006-10-16 02:34:15 +04:00
if (strcmp(fn, "-") != 0) {
f = fopen(fn, "r");
if (f == NULL) {
gpt_warn(gpt, "Can't open `%s'", fn);
return -1;
}
2006-10-16 02:34:15 +04:00
} else
f = stdin;
if ((*name = malloc(maxlen)) == NULL) {
gpt_warn(gpt, "Can't copy string");
2015-12-04 00:48:12 +03:00
goto cleanup;
}
len = fread(*name, 1, maxlen - 1, f);
if (ferror(f)) {
gpt_warn(gpt, "Can't label from `%s'", fn);
2015-12-04 00:48:12 +03:00
goto cleanup;
}
2006-10-16 02:34:15 +04:00
if (f != stdin)
fclose(f);
(*name)[len] = '\0';
2006-10-16 02:34:15 +04:00
/* Only keep the first line, excluding the newline character. */
p = strchr(*name, '\n');
2006-10-16 02:34:15 +04:00
if (p != NULL)
*p = '\0';
return 0;
2015-12-04 00:48:12 +03:00
cleanup:
free(*name);
if (f != stdin)
fclose(f);
return -1;
2006-10-16 02:34:15 +04:00
}
static int
cmd_label(gpt_t gpt, int argc, char *argv[])
2006-10-16 02:34:15 +04:00
{
int ch;
2015-12-01 22:25:24 +03:00
struct gpt_find find;
char *name = NULL;
memset(&find, 0, sizeof(find));
find.msg = "label changed";
2006-10-16 02:34:15 +04:00
/* Get the label options */
2015-12-01 22:25:24 +03:00
while ((ch = getopt(argc, argv, GPT_FIND "f:l:")) != -1) {
2006-10-16 02:34:15 +04:00
switch(ch) {
case 'f':
if (name_from_file(gpt, &name) == -1)
2015-12-04 00:51:02 +03:00
goto usage;
break;
2006-10-16 02:34:15 +04:00
case 'l':
if (gpt_name_get(gpt, &name) == -1)
2015-12-04 00:51:02 +03:00
goto usage;
2006-10-16 02:34:15 +04:00
break;
2015-12-01 22:25:24 +03:00
default:
if (gpt_add_find(gpt, &find, ch) == -1)
2015-12-04 00:51:02 +03:00
goto usage;
2006-10-16 02:34:15 +04:00
break;
}
}
if (name == NULL || argc != optind)
2015-12-04 00:51:02 +03:00
goto usage;
2006-10-16 02:34:15 +04:00
2015-12-01 22:25:24 +03:00
return gpt_change_ent(gpt, &find, change, name);
2015-12-04 00:51:02 +03:00
usage:
2015-12-04 04:47:48 +03:00
usage();
2015-12-04 00:51:02 +03:00
free(name);
return -1;
2006-10-16 02:34:15 +04:00
}