util.c

/*
 *  util.c
 *  FuseHFS
 *
 *  This is fusefs_hfs.util, which is called during the probing/mount process.
 *  See full description in main.c
 *
 *  Created by Zydeco on 2010-03-11.
 *  Copyright 2010 namedfork.net. All rights reserved.
 *  Edited by Joel Cretan 2022-08-24
 *
 *  Licensed under GPLv2: https://www.gnu.org/licenses/gpl-2.0.html
 */

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <libhfs/libhfs.h>
#include <sys/loadable_fs.h>
#include <sys/stat.h>
#include <iconv.h>
#include <sys/errno.h>

#include "common.h"
#include "log.h"

#define FILENAME "[util.c]\t"

/* no longer supported in xnu 3247 / osx 10.11  */
/* from xnu/bsd/sys/loadable_fs.h.auto.html     */
#define    FSUC_INITIALIZE        'i'    /* initialize FS */

int usage() {
	fprintf(stderr, FILENAME "usage: fusefs_hfs.util [-p|-m|-i] <options>\n");
	return EXIT_FAILURE;
}

int have_macfuse() {
	struct stat us;
	return (stat("/usr/local/lib/libfuse.dylib", &us) == 0);
}

#define HFSPLUS_SIGWORD	0x482b /* 'H+' */
#define HFSX_SIGWORD	0x482b /* 'HX' */

int probe (const char *device, int removable, int readonly) {
	char *path;
	int ret = FSUR_RECOGNIZED;
	asprintf(&path, "/dev/%s", device);
	hfsvol * vol = hfs_mount(path, 0, HFS_MODE_RDONLY);
	free(path);
	if (vol == NULL) return FSUR_UNRECOGNIZED;

	/* Refuse to mount HFS+ wrapper volume */
	int embed = vol->mdb.drEmbedSigWord;
	if (embed == HFSPLUS_SIGWORD || embed == HFSX_SIGWORD) {
		ret = FSUR_UNRECOGNIZED;
	} else {
		hfsvolent ent;
		if (hfs_vstat(vol, &ent) == 0)
            // WARNING!! apparently stdout is a pipe!!!
			printf("%s\n", ent.name); // TODO: convert to UTF8
	}
	hfs_umount(vol);
	return ret;
}

int initialize (const char *device, const char *label) {
	// convert label to MacRoman
	char volname[HFS_MAX_VLEN+1];
	size_t len = strlen(label);
	size_t outleft = HFS_MAX_VLEN;
	char *outp = volname;
	iconv_t conv = iconv_open("Macintosh", "UTF-8");
	iconv(conv, (char **restrict)&label, &len, &outp, &outleft);
	volname[HFS_MAX_VLEN-outleft] = '\0';
	volname[HFS_MAX_VLEN] = '\0';
	iconv_close(conv);
	
	// format
	if (hfs_format(device, 0, 0, volname, 0, NULL)) {
		return FSUR_IO_FAIL;
	}
	return EXIT_SUCCESS;
}

/*int mount (const char *device, const char *mountpoint) {
	char *cmd;
	asprintf(&cmd, "/Library/Filesystems/fusefs_hfs.fs/Contents/Resources/fuse_wait \"%s\" %d /Library/Filesystems/fusefs_hfs.fs/Contents/Resources/fusefs_hfs \"%s\" \"%s\"", mountpoint, 5, device, mountpoint);
    fprintf(stderr, FILENAME "mount cmd: %s", cmd);
    fflush(stderr);
    int ret = system(cmd); // TODO: escape input to avoid command injection
	free(cmd);
	return ret?FSUR_IO_FAIL:FSUR_IO_SUCCESS;
}*/

int main (int argc, char * argv[], char * envp[], char * apple[]) {
    log_to_file();
    log_invoking_command(FILENAME, argc, argv);

    
	// check arguments
	if (argc < 3) return usage();
	int ret = 0;
	switch (argv[1][1]) {
		case FSUC_PROBE:
            fprintf(stderr, FILENAME "probing\n");
            fflush(stderr);
			ret = probe(argv[2], !strcmp(argv[3],DEVICE_REMOVABLE), !strcmp(argv[4],DEVICE_READONLY));
			break;
		case FSUC_INITIALIZE: {
            fprintf(stderr, FILENAME "initializing\n");
            fflush(stderr);
			int larg = 2;
			const char *label, *device;
			if (strcmp(argv[2], "-v") == 0) larg = 3;
			label = argv[larg];
			device = argv[larg+1];
			ret = initialize(device, label);
			break; }
		/*case FSUC_MOUNT:
            fprintf(stderr, FILENAME "mounting\n");
            fflush(stderr);
			ret = mount(argv[argc-2], argv[argc-1]);
			break;*/
		case 'k': // get UUID
		case 's': // set UUID
			ret = FSUR_INVAL;
			break;
		default:
			ret = FSUR_INVAL;
			break;
	}
    fprintf(stderr, FILENAME "returning %d\n", ret);
	return ret;
}