258 lines
7.2 KiB
C
258 lines
7.2 KiB
C
/*
|
|
* imount.c
|
|
*
|
|
* Copyright (C) 2007, 2008, 2009 Red Hat, Inc. All rights reserved.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <errno.h>
|
|
#include <fcntl.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/mount.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
#include <glib.h>
|
|
|
|
#include "imount.h"
|
|
#include "log.h"
|
|
|
|
#define _(foo) foo
|
|
|
|
static int readFD(int fd, char **buf) {
|
|
char *p;
|
|
size_t size = 4096;
|
|
int s, filesize = 0;
|
|
|
|
*buf = calloc(4096, sizeof (char));
|
|
if (*buf == NULL)
|
|
abort();
|
|
|
|
do {
|
|
p = &(*buf)[filesize];
|
|
s = read(fd, p, 4096);
|
|
if (s < 0)
|
|
break;
|
|
|
|
filesize += s;
|
|
if (s == 0)
|
|
break;
|
|
|
|
size += s;
|
|
*buf = realloc(*buf, size);
|
|
if (*buf == NULL)
|
|
abort();
|
|
} while (1);
|
|
|
|
if (filesize == 0 && s < 0) {
|
|
free(*buf);
|
|
*buf = NULL;
|
|
return -1;
|
|
}
|
|
|
|
return filesize;
|
|
}
|
|
|
|
static size_t rstrip(char *str) {
|
|
size_t len = strlen(str);
|
|
if (len > 0 && str[len-1] == '\n') {
|
|
str[len-1] = '\0';
|
|
--len;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
int mountCommandWrapper(int mode, char *dev, char *where, char *fs,
|
|
char *options, char **err) {
|
|
int rc, child, status;
|
|
int stdout_pipe[2], stderr_pipe[2];
|
|
char *opts = NULL, *device = NULL, *cmd = NULL;
|
|
|
|
switch (mode) {
|
|
case IMOUNT_MODE_MOUNT:
|
|
case IMOUNT_MODE_BIND:
|
|
cmd = "/bin/mount";
|
|
if (g_mkdir_with_parents(where, 0755))
|
|
return IMOUNT_ERR_ERRNO;
|
|
break;
|
|
case IMOUNT_MODE_UMOUNT:
|
|
cmd = "/bin/umount";
|
|
break;
|
|
default:
|
|
return IMOUNT_ERR_MODE;
|
|
}
|
|
|
|
if (mode == IMOUNT_MODE_MOUNT) {
|
|
if (strstr(fs, "nfs")) {
|
|
if (options && strlen(options)) { // do not append if the options are empty
|
|
if (asprintf(&opts, "%s,nolock", options) == -1) {
|
|
fprintf(stderr, "%s: %d: %s\n", __func__, __LINE__,
|
|
strerror(errno));
|
|
fflush(stderr);
|
|
abort();
|
|
}
|
|
} else {
|
|
opts = strdup("nolock");
|
|
}
|
|
|
|
device = strdup(dev);
|
|
} else {
|
|
if ((options && strstr(options, "bind") == NULL) &&
|
|
strncmp(dev, "LABEL=", 6) && strncmp(dev, "UUID=", 5) &&
|
|
*dev != '/') {
|
|
if (asprintf(&device, "/dev/%s", dev) == -1) {
|
|
fprintf(stderr, "%s: %d: %s\n", __func__, __LINE__,
|
|
strerror(errno));
|
|
fflush(stderr);
|
|
abort();
|
|
}
|
|
} else {
|
|
device = strdup(dev);
|
|
}
|
|
|
|
if (options)
|
|
opts = strdup(options);
|
|
}
|
|
}
|
|
|
|
if (pipe(stdout_pipe))
|
|
return IMOUNT_ERR_ERRNO;
|
|
if (pipe(stderr_pipe))
|
|
return IMOUNT_ERR_ERRNO;
|
|
|
|
if (!(child = fork())) {
|
|
int tty_fd;
|
|
|
|
close(stdout_pipe[0]);
|
|
close(stderr_pipe[0]);
|
|
|
|
/* Pull stdin from /dev/tty5 and redirect stdout and stderr to the pipes
|
|
* so we can log the output and put error messages into exceptions.
|
|
* We'll only use these messages should mount also return an error
|
|
* code.
|
|
*/
|
|
tty_fd = open("/dev/tty5", O_RDONLY);
|
|
close(STDIN_FILENO);
|
|
dup2(tty_fd, STDIN_FILENO);
|
|
close(tty_fd);
|
|
|
|
close(STDOUT_FILENO);
|
|
dup2(stdout_pipe[1], STDOUT_FILENO);
|
|
close(STDERR_FILENO);
|
|
dup2(stderr_pipe[1], STDERR_FILENO);
|
|
|
|
if (mode == IMOUNT_MODE_MOUNT) {
|
|
if (opts) {
|
|
logProgramMessage(INFO, "Running... %s -n -t %s -o %s %s %s",
|
|
cmd, fs, opts, device, where);
|
|
rc = execl(cmd, cmd,
|
|
"-n", "-t", fs, "-o", opts, device, where, NULL);
|
|
exit(1);
|
|
} else {
|
|
logProgramMessage(INFO, "Running... %s -n -t %s %s %s",
|
|
cmd, fs, device, where);
|
|
rc = execl(cmd, cmd, "-n", "-t", fs, device, where, NULL);
|
|
exit(1);
|
|
}
|
|
} else if (mode == IMOUNT_MODE_BIND) {
|
|
logProgramMessage(INFO, "Running... %s --bind %s %s",
|
|
cmd, dev, where);
|
|
rc = execl(cmd, cmd, "--bind", dev, where, NULL);
|
|
exit(1);
|
|
} else if (mode == IMOUNT_MODE_UMOUNT) {
|
|
logProgramMessage(INFO, "Running... %s %s", cmd, where);
|
|
rc = execl(cmd, cmd, where, NULL);
|
|
exit(1);
|
|
} else {
|
|
logProgramMessage(ERROR, "Running... Unknown imount mode: %d\n", mode);
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
close(stdout_pipe[1]);
|
|
close(stderr_pipe[1]);
|
|
|
|
char *buffer = NULL;
|
|
/* In case when when the stderr pipe gets enough data to fill the kernel
|
|
* buffer we might see a deadlock as this will block the mount program on
|
|
* its next write(). The buffer size is 65kB though so we should be safe.
|
|
*/
|
|
rc = readFD(stdout_pipe[0], &buffer);
|
|
if (rc > 0) {
|
|
rstrip(buffer);
|
|
logProgramMessage(INFO, buffer);
|
|
free(buffer);
|
|
buffer = NULL;
|
|
}
|
|
rc = readFD(stderr_pipe[0], &buffer);
|
|
if (rc > 0) {
|
|
rstrip(buffer);
|
|
logProgramMessage(ERROR, buffer);
|
|
if (err != NULL)
|
|
*err = buffer;
|
|
else
|
|
free(buffer);
|
|
}
|
|
|
|
waitpid(child, &status, 0);
|
|
close(stdout_pipe[0]);
|
|
close(stderr_pipe[0]);
|
|
|
|
if (opts) {
|
|
free(opts);
|
|
}
|
|
|
|
if (device) {
|
|
free(device);
|
|
}
|
|
|
|
if (!WIFEXITED(status))
|
|
return IMOUNT_ERR_OTHER;
|
|
else if ( (rc = WEXITSTATUS(status)) )
|
|
return rc;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int doBindMount(char* path, char *where, char **err) {
|
|
return mountCommandWrapper(IMOUNT_MODE_BIND,
|
|
path, where, NULL, NULL, err);
|
|
}
|
|
|
|
int doPwMount(char *dev, char *where, char *fs, char *options, char **err) {
|
|
return mountCommandWrapper(IMOUNT_MODE_MOUNT,
|
|
dev, where, fs, options, err);
|
|
}
|
|
|
|
int doPwUmount(char *where, char **err) {
|
|
return mountCommandWrapper(IMOUNT_MODE_UMOUNT,
|
|
NULL, where, NULL, NULL, err);
|
|
}
|
|
|
|
/* Returns true iff it is possible that the mount command that have returned
|
|
* 'errno' might succeed at a later time (think e.g. not yet initialized USB
|
|
* device, etc.) */
|
|
int mountMightSucceedLater(int mountRc)
|
|
{
|
|
/* 32 is the mount exit code for "mount failure" */
|
|
if (mountRc > 0 && mountRc & 0x20)
|
|
return 1;
|
|
else
|
|
return 0;
|
|
}
|