134 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			134 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include <sys/mount.h>
 | |
| #include <sys/types.h>
 | |
| #include <sys/stat.h>
 | |
| #include <sys/wait.h>
 | |
| #include <fcntl.h>
 | |
| #include <stdio.h>
 | |
| #include <unistd.h>
 | |
| 
 | |
| int max_error = 0;
 | |
| #include "common.h"
 | |
| 
 | |
| 
 | |
| #define TESTMNT		"testmnt"
 | |
| #define TESTFILE	"test.txt"
 | |
| #define TESTSTRING	"foobar"
 | |
| #define RAMDISK		"/dev/ram5"
 | |
| #define RAMDISK_SIZE	"2048"
 | |
| #define SILENT		" > /dev/null 2>&1"
 | |
| 
 | |
| void basic_test(void);
 | |
| void bomb(char const *msg);
 | |
| void skip(char const *msg);
 | |
| void create_partition(void);
 | |
| void verify_tools(void);
 | |
| 
 | |
| void
 | |
| basic_test(void)
 | |
| {
 | |
| /* Write a string to a file, read it back, and confirm it's identical */
 | |
| 	int status;
 | |
| 	char cmd_buf[1024];
 | |
| 	char file_buf[sizeof(TESTSTRING)*10];
 | |
| 	int fd;
 | |
| 
 | |
| 	subtest = 3;
 | |
| 
 | |
| 	/* Write test string to test file */
 | |
| 	snprintf(cmd_buf, sizeof(cmd_buf), "echo -n %s > %s/%s\n",
 | |
| 		TESTSTRING, TESTMNT, TESTFILE);
 | |
| 	status = system(cmd_buf);
 | |
| 	if (WEXITSTATUS(status) != 0)
 | |
| 		bomb("Unable to echo string to file");
 | |
| 
 | |
| 	/* Flush to disk and unmount, remount */
 | |
| 	system("sync");
 | |
| 	system("umount " RAMDISK SILENT);
 | |
| 	snprintf(cmd_buf, sizeof(cmd_buf), "mount -t ntfs-3g %s %s %s",
 | |
| 		RAMDISK, TESTMNT, SILENT);
 | |
| 	status = system(cmd_buf);
 | |
| 	if (WEXITSTATUS(status) != 0)
 | |
| 		bomb("Unable to mount NTFS partition (1)");
 | |
| 
 | |
| 	/* Open file and verify contents */
 | |
| 	if ((fd = open(TESTMNT "/" TESTFILE, O_RDONLY)) < 0) e(1);
 | |
| 	if (read(fd, file_buf, sizeof(file_buf)) != strlen(TESTSTRING)) e(2);
 | |
| 	(void) close(fd);
 | |
| 	system("umount " RAMDISK SILENT);
 | |
| 	if (strncmp(file_buf, TESTSTRING, strlen(TESTSTRING))) e(3);
 | |
| }
 | |
| 
 | |
| void
 | |
| skip(char const *msg)
 | |
| {
 | |
| 	system("umount " RAMDISK SILENT);
 | |
| 	printf("%s\n", msg);
 | |
| 	quit();
 | |
| }
 | |
| 
 | |
| void
 | |
| bomb(char const *msg)
 | |
| {
 | |
| 	system("umount " RAMDISK SILENT);
 | |
| 	printf("%s\n", msg);
 | |
| 	e(99);
 | |
| 	quit();
 | |
| }
 | |
| 
 | |
| void
 | |
| create_partition(void)
 | |
| {
 | |
| 	int status;
 | |
| 	char mntcmd[1024];
 | |
| 
 | |
| 	subtest = 1;
 | |
| 
 | |
| 	if (getuid() != 0 && setuid(0) != 0) e(1);
 | |
| 	status = system("ramdisk " RAMDISK_SIZE " " RAMDISK SILENT);
 | |
| 	if (WEXITSTATUS(status) != 0)
 | |
| 		bomb("Unable to create ramdisk");
 | |
| 
 | |
| 	status = system("mkntfs " RAMDISK SILENT);
 | |
| 	if (WEXITSTATUS(status) != 0)
 | |
| 		bomb("Unable to create NTFS file system on " RAMDISK);
 | |
| 
 | |
| 	if (mkdir(TESTMNT, 0755) != 0)
 | |
| 		bomb("Unable to create directory for mounting");
 | |
| 
 | |
| 	snprintf(mntcmd, sizeof(mntcmd), "mount -t ntfs-3g %s %s %s",
 | |
| 		RAMDISK, TESTMNT, SILENT);
 | |
| 	status = system(mntcmd);
 | |
| 	if (WEXITSTATUS(status) != 0)
 | |
| 		bomb("Unable to mount NTFS partition (1)");
 | |
| }
 | |
| 
 | |
| void
 | |
| verify_tools(void)
 | |
| {
 | |
| 	int status;
 | |
| 
 | |
| 	subtest = 1;
 | |
| 	status = system("which mkntfs > /dev/null 2>&1");
 | |
| 	if (WEXITSTATUS(status) != 0) {
 | |
| 		skip("mkntfs not found. Please install ntfsprogs (pkgin in "
 | |
| 			"ntfsprogs)");
 | |
| 	}
 | |
| 	status = system("which ntfs-3g > /dev/null 2>&1");
 | |
| 	if (WEXITSTATUS(status) != 0) {
 | |
| 		skip("ntfs-3g not found. Please install fuse-ntfs-3g-1.1120 "
 | |
| 			"(pkgin in fuse-ntfs-3g-1.1120)");
 | |
| 	}
 | |
| }
 | |
| 
 | |
| int
 | |
| main(int argc, char *argv[])
 | |
| {
 | |
| 	start(65);
 | |
| 	verify_tools();
 | |
| 	create_partition();
 | |
| 	basic_test();
 | |
| 	quit();
 | |
| 	return(-1);	/* Unreachable */
 | |
| }
 | |
| 
 | 
