blob: df2e3cfad5cf18583864bb9a743dc5174585ae35 [file] [log] [blame]
/*
* SPDX-License-Identifier: GPL-2.0-only
*/
#define _GNU_SOURCE
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <stdarg.h>
#include <stdlib.h>
#include <ctype.h>
#include <fcntl.h>
#include <dirent.h>
#include <unistd.h>
#include <time.h>
#include <getopt.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/stat.h>
#define MINORBITS 8
#define MKDEV(ma,mi) (((ma) << MINORBITS) | (mi))
#define MAX_ID_LEN 40
#define MAX_NAME_LEN 40
#ifndef PATH_MAX
#define PATH_MAX 4096
#endif
#define VERSION "1.0.1"
/* These are all stolen from busybox's libbb to make
* error handling simpler (and since I maintain busybox,
* I'm rather partial to these for error handling).
* -Erik
*/
static const char *const app_name = "makedevs";
static const char *const memory_exhausted = "memory exhausted";
static char default_rootdir[]=".";
static char *rootdir = default_rootdir;
static int trace = 0;
struct name_id {
char name[MAX_NAME_LEN+1];
unsigned long id;
struct name_id *next;
};
static struct name_id *usr_list = NULL;
static struct name_id *grp_list = NULL;
static void verror_msg(const char *s, va_list p)
{
fflush(stdout);
fprintf(stderr, "%s: ", app_name);
vfprintf(stderr, s, p);
}
static void error_msg_and_die(const char *s, ...)
{
va_list p;
va_start(p, s);
verror_msg(s, p);
va_end(p);
putc('\n', stderr);
exit(EXIT_FAILURE);
}
static void vperror_msg(const char *s, va_list p)
{
int err = errno;
if (s == 0)
s = "";
verror_msg(s, p);
if (*s)
s = ": ";
fprintf(stderr, "%s%s\n", s, strerror(err));
}
static void perror_msg_and_die(const char *s, ...)
{
va_list p;
va_start(p, s);
vperror_msg(s, p);
va_end(p);
exit(EXIT_FAILURE);
}
static FILE *xfopen(const char *path, const char *mode)
{
FILE *fp;
if ((fp = fopen(path, mode)) == NULL)
perror_msg_and_die("%s", path);
return fp;
}
static char *xstrdup(const char *s)
{
char *t;
if (s == NULL)
return NULL;
t = strdup(s);
if (t == NULL)
error_msg_and_die(memory_exhausted);
return t;
}
static struct name_id* alloc_node(void)
{
struct name_id *node;
node = (struct name_id*)malloc(sizeof(struct name_id));
if (node == NULL) {
error_msg_and_die(memory_exhausted);
}
memset((void *)node->name, 0, MAX_NAME_LEN+1);
node->id = 0xffffffff;
node->next = NULL;
return node;
}
static struct name_id* parse_line(char *line)
{
char *p;
int i;
char id_buf[MAX_ID_LEN+1];
struct name_id *node;
node = alloc_node();
p = line;
i = 0;
// Get name field
while (*p != ':') {
if (i > MAX_NAME_LEN)
error_msg_and_die("Name field too long");
node->name[i++] = *p++;
}
node->name[i] = '\0';
p++;
// Skip the second field
while (*p != ':')
p++;
p++;
// Get id field
i = 0;
while (*p != ':') {
if (i > MAX_ID_LEN)
error_msg_and_die("ID filed too long");
id_buf[i++] = *p++;
}
id_buf[i] = '\0';
node->id = atol(id_buf);
return node;
}
static void get_list_from_file(FILE *file, struct name_id **plist)
{
char *line;
int len = 0;
size_t length = 256;
struct name_id *node, *cur;
if((line = (char *)malloc(length)) == NULL) {
error_msg_and_die(memory_exhausted);
}
while ((len = getline(&line, &length, file)) != -1) {
node = parse_line(line);
if (*plist == NULL) {
*plist = node;
cur = *plist;
} else {
cur->next = node;
cur = cur->next;
}
}
if (line)
free(line);
}
static unsigned long convert2guid(char *id_buf, struct name_id *search_list)
{
char *p;
int isnum;
struct name_id *node;
p = id_buf;
isnum = 1;
while (*p != '\0') {
if (!isdigit(*p)) {
isnum = 0;
break;
}
p++;
}
if (isnum) {
// Check for bad user/group name
node = search_list;
while (node != NULL) {
if (!strncmp(node->name, id_buf, strlen(id_buf))) {
fprintf(stderr, "WARNING: Bad user/group name %s detected\n", id_buf);
break;
}
node = node->next;
}
return (unsigned long)atol(id_buf);
} else {
node = search_list;
while (node != NULL) {
if (!strncmp(node->name, id_buf, strlen(id_buf)))
return node->id;
node = node->next;
}
error_msg_and_die("No entry for %s in search list", id_buf);
}
}
static void free_list(struct name_id *list)
{
struct name_id *cur;
cur = list;
while (cur != NULL) {
list = cur;
cur = cur->next;
free(list);
}
}
static void add_new_directory(char *name, char *path,
unsigned long uid, unsigned long gid, unsigned long mode)
{
if (trace)
fprintf(stderr, "Directory: %s %s UID: %lu GID %lu MODE: %04lo", path, name, uid, gid, mode);
if (mkdir(path, mode) < 0) {
if (EEXIST == errno) {
/* Unconditionally apply the mode setting to the existing directory.
* XXX should output something when trace */
chmod(path, mode & ~S_IFMT);
}
}
if (trace)
putc('\n', stderr);
chown(path, uid, gid);
}
static void add_new_device(char *name, char *path, unsigned long uid,
unsigned long gid, unsigned long mode, dev_t rdev)
{
int status;
struct stat sb;
if (trace) {
fprintf(stderr, "Device: %s %s UID: %lu GID: %lu MODE: %04lo MAJOR: %d MINOR: %d",
path, name, uid, gid, mode, (short)(rdev >> 8), (short)(rdev & 0xff));
}
memset(&sb, 0, sizeof(struct stat));
status = lstat(path, &sb);
if (status >= 0) {
/* It is ok for some types of files to not exit on disk (such as
* device nodes), but if they _do_ exist, the file type bits had
* better match those of the actual file or strange things will happen... */
if ((mode & S_IFMT) != (sb.st_mode & S_IFMT)) {
if (trace)
putc('\n', stderr);
error_msg_and_die("%s: existing file (04%o) type does not match specified file type (04%lo)!",
path, (sb.st_mode & S_IFMT), (mode & S_IFMT));
}
if (mode != sb.st_mode) {
if (trace)
fprintf(stderr, " -- applying new mode 04%lo (old was 04%o)\n", mode & ~S_IFMT, sb.st_mode & ~S_IFMT);
/* Apply the mode setting to the existing device node */
chmod(path, mode & ~S_IFMT);
}
else {
if (trace)
fprintf(stderr, " -- extraneous entry in table\n", path);
}
}
else {
mknod(path, mode, rdev);
if (trace)
putc('\n', stderr);
}
chown(path, uid, gid);
}
static void add_new_file(char *name, char *path, unsigned long uid,
unsigned long gid, unsigned long mode)
{
if (trace) {
fprintf(stderr, "File: %s %s UID: %lu GID: %lu MODE: %04lo\n",
path, name, gid, uid, mode);
}
int fd = open(path,O_CREAT | O_WRONLY, mode);
if (fd < 0) {
error_msg_and_die("%s: file can not be created!", path);
} else {
close(fd);
}
chmod(path, mode);
chown(path, uid, gid);
}
static void add_new_fifo(char *name, char *path, unsigned long uid,
unsigned long gid, unsigned long mode)
{
if (trace) {
printf("Fifo: %s %s UID: %lu GID: %lu MODE: %04lo\n",
path, name, gid, uid, mode);
}
int status;
struct stat sb;
memset(&sb, 0, sizeof(struct stat));
status = stat(path, &sb);
/* Update the mode if we exist and are a fifo already */
if (status >= 0 && S_ISFIFO(sb.st_mode)) {
chmod(path, mode);
} else {
if (mknod(path, mode, 0))
error_msg_and_die("%s: file can not be created with mknod!", path);
}
chown(path, uid, gid);
}
/* device table entries take the form of:
<path> <type> <mode> <usr> <grp> <major> <minor> <start> <inc> <count>
/dev/mem c 640 0 0 1 1 0 0 -
/dev/zero c 644 root root 1 5 - - -
type can be one of:
f A regular file
d Directory
c Character special device file
b Block special device file
p Fifo (named pipe)
I don't bother with symlinks (permissions are irrelevant), hard
links (special cases of regular files), or sockets (why bother).
Regular files must exist in the target root directory. If a char,
block, fifo, or directory does not exist, it will be created.
*/
static int interpret_table_entry(char *line)
{
char *name;
char usr_buf[MAX_ID_LEN];
char grp_buf[MAX_ID_LEN];
char path[4096], type;
unsigned long mode = 0755, uid = 0, gid = 0, major = 0, minor = 0;
unsigned long start = 0, increment = 1, count = 0;
if (0 > sscanf(line, "%4095s %c %lo %39s %39s %lu %lu %lu %lu %lu", path,
&type, &mode, usr_buf, grp_buf, &major, &minor, &start,
&increment, &count))
{
fprintf(stderr, "%s: sscanf returned < 0 for line '%s'\n", app_name, line);
return 1;
}
uid = convert2guid(usr_buf, usr_list);
gid = convert2guid(grp_buf, grp_list);
if (strncmp(path, "/", 1)) {
error_msg_and_die("Device table entries require absolute paths");
}
name = xstrdup(path + 1);
/* prefix path with rootdir */
sprintf(path, "%s/%s", rootdir, name);
/* XXX Why is name passed into all of the add_new_*() routines? */
switch (type) {
case 'd':
mode |= S_IFDIR;
add_new_directory(name, path, uid, gid, mode);
break;
case 'f':
mode |= S_IFREG;
add_new_file(name, path, uid, gid, mode);
break;
case 'p':
mode |= S_IFIFO;
add_new_fifo(name, path, uid, gid, mode);
break;
case 'c':
case 'b':
mode |= (type == 'c') ? S_IFCHR : S_IFBLK;
if (count > 0) {
int i;
dev_t rdev;
char buf[80];
for (i = start; i < start + count; i++) {
sprintf(buf, "%s%d", name, i);
sprintf(path, "%s/%s%d", rootdir, name, i);
/* FIXME: MKDEV uses illicit insider knowledge of kernel
* major/minor representation... */
rdev = MKDEV(major, minor + (i - start) * increment);
sprintf(path, "%s/%s\0", rootdir, buf);
add_new_device(buf, path, uid, gid, mode, rdev);
}
} else {
/* FIXME: MKDEV uses illicit insider knowledge of kernel
* major/minor representation... */
dev_t rdev = MKDEV(major, minor);
add_new_device(name, path, uid, gid, mode, rdev);
}
break;
default:
error_msg_and_die("Unsupported file type");
}
if (name) free(name);
return 0;
}
static void parse_device_table(FILE * file)
{
char *line;
size_t length = 256;
int len = 0;
if((line = (char *)malloc(length)) == NULL) {
error_msg_and_die(memory_exhausted);
}
/* Looks ok so far. The general plan now is to read in one
* line at a time, trim off leading and trailing whitespace,
* check for leading comment delimiters ('#') or a blank line,
* then try and parse the line as a device table entry. If we fail
* to parse things, try and help the poor fool to fix their
* device table with a useful error msg... */
while ((len = getline(&line, &length, file)) != -1) {
/* First trim off any whitespace */
/* trim trailing whitespace */
while (len > 0 && isspace(line[len - 1]))
line[--len] = '\0';
/* trim leading whitespace */
memmove(line, &line[strspn(line, " \n\r\t\v")], len + 1);
/* If this is NOT a comment or an empty line, try to interpret it */
if (*line != '#' && *line != '\0') interpret_table_entry(line);
}
if (line)
free(line);
}
static int parse_devtable(FILE * devtable)
{
struct stat sb;
if (lstat(rootdir, &sb)) {
perror_msg_and_die("%s", rootdir);
}
if (chdir(rootdir))
perror_msg_and_die("%s", rootdir);
if (devtable)
parse_device_table(devtable);
return 0;
}
static struct option long_options[] = {
{"root", 1, NULL, 'r'},
{"help", 0, NULL, 'h'},
{"trace", 0, NULL, 't'},
{"version", 0, NULL, 'v'},
{"devtable", 1, NULL, 'D'},
{NULL, 0, NULL, 0}
};
static char *helptext =
"Usage: makedevs [OPTIONS]\n"
"Build entries based upon device_table.txt\n\n"
"Options:\n"
" -r, -d, --root=DIR Build filesystem from directory DIR (default: cwd)\n"
" -D, --devtable=FILE Use the named FILE as a device table file\n"
" -h, --help Display this help text\n"
" -t, --trace Be verbose\n"
" -v, --version Display version information\n\n";
int main(int argc, char **argv)
{
int c, opt;
extern char *optarg;
struct stat statbuf;
char passwd_path[PATH_MAX];
char group_path[PATH_MAX];
FILE *passwd_file = NULL;
FILE *group_file = NULL;
FILE *devtable = NULL;
DIR *dir = NULL;
umask (0);
if (argc==1) {
fputs( helptext , stderr );
exit(1);
}
while ((opt = getopt_long(argc, argv, "D:d:r:htv",
long_options, &c)) >= 0) {
switch (opt) {
case 'D':
devtable = xfopen(optarg, "r");
if (fstat(fileno(devtable), &statbuf) < 0)
perror_msg_and_die(optarg);
if (statbuf.st_size < 10)
error_msg_and_die("%s: not a proper device table file", optarg);
break;
case 'h':
puts(helptext);
exit(0);
case 'r':
case 'd': /* for compatibility with mkfs.jffs, genext2fs, etc... */
if (rootdir != default_rootdir) {
error_msg_and_die("root directory specified more than once");
}
if ((dir = opendir(optarg)) == NULL) {
perror_msg_and_die(optarg);
} else {
closedir(dir);
}
/* If "/" is specified, use "" because rootdir is always prepended to a
* string that starts with "/" */
if (0 == strcmp(optarg, "/"))
rootdir = xstrdup("");
else
rootdir = xstrdup(optarg);
break;
case 't':
trace = 1;
break;
case 'v':
printf("%s: %s\n", app_name, VERSION);
exit(0);
default:
fputs(helptext,stderr);
exit(1);
}
}
if (argv[optind] != NULL) {
fputs(helptext,stderr);
exit(1);
}
// Get name-id mapping
sprintf(passwd_path, "%s/etc/passwd", rootdir);
sprintf(group_path, "%s/etc/group", rootdir);
if ((passwd_file = fopen(passwd_path, "r")) != NULL) {
get_list_from_file(passwd_file, &usr_list);
fclose(passwd_file);
}
if ((group_file = fopen(group_path, "r")) != NULL) {
get_list_from_file(group_file, &grp_list);
fclose(group_file);
}
// Parse devtable
if(devtable) {
parse_devtable(devtable);
fclose(devtable);
}
// Free list
free_list(usr_list);
free_list(grp_list);
return 0;
}