This commit is contained in:
Kovid Goyal 2011-11-03 07:31:39 +05:30
parent f64a88cd37
commit 646071ecb0

View File

@ -9,6 +9,8 @@
#include <fcntl.h> #include <fcntl.h>
#define MARKER ".created_by_calibre_mount_helper" #define MARKER ".created_by_calibre_mount_helper"
#define DEV "/dev/"
#define MEDIA "/media/"
#define False 0 #define False 0
#define True 1 #define True 1
@ -34,17 +36,17 @@ void ensure_root() {
} }
int check_args(const char *dev, const char *mp) { int check_args(const char *dev, const char *mp) {
if (dev == NULL || strlen(dev) < strlen("/dev/") || mp == NULL || strlen(mp) < strlen("/media/")) { if (dev == NULL || strlen(dev) < strlen(DEV) || mp == NULL || strlen(mp) < strlen(MEDIA)) {
fprintf(stderr, "Invalid arguments\n"); fprintf(stderr, "Invalid arguments\n");
return False; return False;
} }
if (strncmp("/media/", mp, strlen("/media/")) != 0) { if (strncmp(MEDIA, mp, strlen("MEDIA")) != 0) {
fprintf(stderr, "Trying to operate on a mount point not under /media is not allowed\n"); fprintf(stderr, "Trying to operate on a mount point not under /media is not allowed\n");
return False; return False;
} }
if (strncmp("/dev/", dev, strlen("/dev/")) != 0) { if (strncmp(DEV, dev, strlen(DEV)) != 0) {
fprintf(stderr, "Trying to operate on a dev node not under /dev\n"); fprintf(stderr, "Trying to operate on a dev node not under /dev\n");
return False; return False;
} }