<commit_before>package main

import (
	"flag"
	"fmt"
	"os"
	"os/exec"
	"os/signal"
	"path/filepath"
	"runtime"
	"runtime/pprof"
	"syscall"
	"time"

	"github.com/rfjakob/gocryptfs/cryptfs"
	"github.com/rfjakob/gocryptfs/pathfs_frontend"

	"github.com/hanwen/go-fuse/fuse"
	"github.com/hanwen/go-fuse/fuse/nodefs"
	"github.com/hanwen/go-fuse/fuse/pathfs"
)

const (
	PROGRAM_NAME = "gocryptfs"

	// Exit codes
	ERREXIT_USAGE      = 1
	ERREXIT_MOUNT      = 3
	ERREXIT_CIPHERDIR  = 6
	ERREXIT_INIT       = 7
	ERREXIT_LOADCONF   = 8
	ERREXIT_PASSWORD   = 9
	ERREXIT_MOUNTPOINT = 10
)

// GitVersion will be set by the build script "build.bash"
var GitVersion = "[version not set - please compile using ./build.bash]"

func initDir(dirArg string, plaintextNames bool) {
	dir, _ := filepath.Abs(dirArg)

	err := checkDirEmpty(dir)
	if err != nil {
		fmt.Printf("Invalid CIPHERDIR: %v\n", err)
		os.Exit(ERREXIT_INIT)
	}

	confName := filepath.Join(dir, cryptfs.ConfDefaultName)
	cryptfs.Info.Printf("Choose a password for protecting your files.\n")
	password := <PASSWORD>()
	err = cryptfs.CreateConfFile(confName, password, plaintextNames)
	if err != nil {
		fmt.Println(err)
		os.Exit(ERREXIT_INIT)
	}
	cryptfs.Info.Printf("The filesystem is now ready for mounting.\n")
	os.Exit(0)
}

func usageText() {
	fmt.Fprintf(os.Stderr, "Usage: %s [OPTIONS] CIPHERDIR MOUNTPOINT\n", PROGRAM_NAME)
	fmt.Fprintf(os.Stderr, "\nOptions:\n")
	flag.PrintDefaults()
}

func main() {
	runtime.GOMAXPROCS(4)

	// Parse command line arguments
	var args struct {
		debug, init, zerokey, fusedebug, openssl, passwd, foreground, version,
			plaintextnames, quiet bool
		masterkey, mountpoint, cipherdir string
		cpuprofile *string
	}

	flag.Usage = usageText
	flag.BoolVar(&args.debug, "debug", false, "Enable debug output")
	flag.BoolVar(&args.fusedebug, "fusedebug", false, "Enable fuse library debug output")
	flag.BoolVar(&args.init, "init", false, "Initialize encrypted directory")
	flag.BoolVar(&args.zerokey, "zerokey", false, "Use all-zero dummy master key")
	flag.BoolVar(&args.openssl, "openssl", true, "Use OpenSSL instead of built-in Go crypto")
	flag.BoolVar(&args.passwd, "passwd", false, "Change password")
	flag.BoolVar(&args.foreground, "f", false, "Stay in the foreground")
	flag.BoolVar(&args.version, "version", false, "Print version and exit")
	flag.BoolVar(&args.plaintextnames, "plaintextnames", false,
		"Do not encrypt file names - can only be used together with -init")
	flag.BoolVar(&args.quiet, "q", false, "Quiet - silence informational messages")
	flag.StringVar(&args.masterkey, "masterkey", "", "Mount with explicit master key")
	args.cpuprofile = flag.String("cpuprofile", "", "Write cpu profile to specified file")

	flag.Parse()
	if args.version {
		fmt.Printf("%s %s; on-disk format %d\n", PROGRAM_NAME, GitVersion, cryptfs.HEADER_CURRENT_VERSION)
		os.Exit(0)
	}
	if args.quiet {
		cryptfs.Info.Disable()
	}
	if !args.foreground {
		daemonize() // does not return
	}
	if *args.cpuprofile != "" {
		f, err := os.Create(*args.cpuprofile)
		if err != nil {
			fmt.Println(err)
			os.Exit(ERREXIT_INIT)
		}
		cryptfs.Info.Printf("Writing CPU profile to %s\n", *args.cpuprofile)
		pprof.StartCPUProfile(f)
		defer pprof.StopCPUProfile()
	}
	if args.debug {
		cryptfs.Debug.Enable()
		cryptfs.Debug.Printf("Debug output enabled\n")
	}
	if args.openssl == false {
		cryptfs.Info.Printf("Openssl disabled\n")
	}
	if args.init {
		if flag.NArg() != 1 && args.plaintextnames == false {
			fmt.Printf("Usage: %s --init [--plaintextnames] CIPHERDIR\n", PROGRAM_NAME)
			os.Exit(ERREXIT_USAGE)
		}
		initDir(flag.Arg(0), args.plaintextnames) // does not return
	}
	if args.passwd {
		if flag.NArg() != 1 {
			fmt.Printf("Usage: %s --passwd CIPHERDIR\n", PROGRAM_NAME)
			os.Exit(ERREXIT_USAGE)
		}
	} else {
		// Normal mount
		if flag.NArg() < 2 {
			usageText()
			os.Exit(ERREXIT_USAGE)
		}
		args.mountpoint, _ = filepath.Abs(flag.Arg(1))
		err := checkDirEmpty(args.mountpoint)
		if err != nil {
			fmt.Printf("Invalid MOUNTPOINT: %v\n", err)
			os.Exit(ERREXIT_MOUNTPOINT)
		}
	}
	args.cipherdir, _ = filepath.Abs(flag.Arg(0))
	err := checkDir(args.cipherdir)
	if err != nil {
		fmt.Printf("Invalid CIPHERDIR: %v\n", err)
		os.Exit(ERREXIT_CIPHERDIR)
	}

	var plaintextNames bool
	var cf *cryptfs.ConfFile
	var currentPassword string
	key := make([]byte, cryptfs.KEY_LEN)
	if args.zerokey {
		cryptfs.Info.Printf("Zerokey mode active: using all-zero dummy master key.\n")
		plaintextNames = args.plaintextnames
	} else if len(args.masterkey) > 0 {
		key = parseMasterKey(args.masterkey)
		cryptfs.Info.Printf("Using explicit master key.\n")
	} else {
		// Load config file
		cfname := filepath.Join(args.cipherdir, cryptfs.ConfDefaultName)
		_, err = os.Stat(cfname)
		if err != nil {
			fmt.Printf("Error: %s not found in CIPHERDIR\n", cryptfs.ConfDefaultName)
			fmt.Printf("Please run \"%s --init %s\" first\n", os.Args[0], flag.Arg(0))
			os.Exit(ERREXIT_LOADCONF)
		}
		if args.passwd == true {
			fmt.Printf("Old password: ")
		} else {
			fmt.Printf("Password: ")
		}
		currentPassword = readPassword()
		cryptfs.Info.Printf("\nDecrypting master key... ")
		cryptfs.Warn.Disable() // Silence DecryptBlock() error messages on incorrect password
		key, cf, err = cryptfs.LoadConfFile(cfname, currentPassword)
		cryptfs.Warn.Enable()
		if err != nil {
			fmt.Println(err)
			os.Exit(ERREXIT_LOADCONF)
		}
		cryptfs.Info.Printf("done.\n")
	}<commit_msg>cli: Create our own FlagSet so we can set the policy to ExitOnError<commit_after>package main

import (
	"flag"
	"fmt"
	"os"
	"os/exec"
	"os/signal"
	"path/filepath"
	"runtime"
	"runtime/pprof"
	"syscall"
	"time"

	"github.com/rfjakob/gocryptfs/cryptfs"
	"github.com/rfjakob/gocryptfs/pathfs_frontend"

	"github.com/hanwen/go-fuse/fuse"
	"github.com/hanwen/go-fuse/fuse/nodefs"
	"github.com/hanwen/go-fuse/fuse/pathfs"
)

const (
	PROGRAM_NAME = "gocryptfs"

	// Exit codes
	ERREXIT_USAGE      = 1
	ERREXIT_MOUNT      = 3
	ERREXIT_CIPHERDIR  = 6
	ERREXIT_INIT       = 7
	ERREXIT_LOADCONF   = 8
	ERREXIT_PASSWORD   = 9
	ERREXIT_MOUNTPOINT = 10
)

// GitVersion will be set by the build script "build.bash"
var GitVersion = "[version not set - please compile using ./build.bash]"

func initDir(dirArg string, plaintextNames bool) {
	dir, _ := filepath.Abs(dirArg)

	err := checkDirEmpty(dir)
	if err != nil {
		fmt.Printf("Invalid CIPHERDIR: %v\n", err)
		os.Exit(ERREXIT_INIT)
	}

	confName := filepath.Join(dir, cryptfs.ConfDefaultName)
	cryptfs.Info.Printf("Choose a password for protecting your files.\n")
	password := <PASSWORD>()
	err = cryptfs.CreateConfFile(confName, password, plaintextNames)
	if err != nil {
		fmt.Println(err)
		os.Exit(ERREXIT_INIT)
	}
	cryptfs.Info.Printf("The filesystem is now ready for mounting.\n")
	os.Exit(0)
}

func usageText() {
	fmt.Fprintf(os.Stderr, "Usage: %s [OPTIONS] CIPHERDIR MOUNTPOINT\n", PROGRAM_NAME)
	fmt.Fprintf(os.Stderr, "\nOptions:\n")
	flagSet.PrintDefaults()
}

type argContainer struct {
		debug, init, zerokey, fusedebug, openssl, passwd, foreground, version,
			plaintextnames, quiet bool
		masterkey, mountpoint, cipherdir string
		cpuprofile *string
		sendusr1 int
}

var flagSet *flag.FlagSet

func main() {
	runtime.GOMAXPROCS(4)

	// Parse command line arguments
	var args argContainer
	flagSet = flag.NewFlagSet(PROGRAM_NAME, flag.ExitOnError)
	flagSet.Usage = usageText
	flagSet.BoolVar(&args.debug, "debug", false, "Enable debug output")
	flagSet.BoolVar(&args.fusedebug, "fusedebug", false, "Enable fuse library debug output")
	flagSet.BoolVar(&args.init, "init", false, "Initialize encrypted directory")
	flagSet.BoolVar(&args.zerokey, "zerokey", false, "Use all-zero dummy master key")
	flagSet.BoolVar(&args.openssl, "openssl", true, "Use OpenSSL instead of built-in Go crypto")
	flagSet.BoolVar(&args.passwd, "passwd", false, "Change password")
	flagSet.BoolVar(&args.foreground, "f", false, "Stay in the foreground")
	flagSet.BoolVar(&args.version, "version", false, "Print version and exit")
	flagSet.BoolVar(&args.plaintextnames, "plaintextnames", false,
		"Do not encrypt file names - can only be used together with -init")
	flagSet.BoolVar(&args.quiet, "q", false, "Quiet - silence informational messages")
	flagSet.StringVar(&args.masterkey, "masterkey", "", "Mount with explicit master key")
	args.cpuprofile = flagSet.String("cpuprofile", "", "Write cpu profile to specified file")
	flagSet.IntVar(&args.sendusr1, "sendusr1", 0,
		"Send USR1 to the specified process after successful mount - used internally for daemonization")
	flagSet.Parse(os.Args[1:])
	if args.version {
		fmt.Printf("%s %s; on-disk format %d\n", PROGRAM_NAME, GitVersion, cryptfs.HEADER_CURRENT_VERSION)
		os.Exit(0)
	}
	if args.quiet {
		cryptfs.Info.Disable()
	}
	if !args.foreground {
		daemonize() // does not return
	}
	if *args.cpuprofile != "" {
		f, err := os.Create(*args.cpuprofile)
		if err != nil {
			fmt.Println(err)
			os.Exit(ERREXIT_INIT)
		}
		cryptfs.Info.Printf("Writing CPU profile to %s\n", *args.cpuprofile)
		pprof.StartCPUProfile(f)
		defer pprof.StopCPUProfile()
	}
	if args.debug {
		cryptfs.Debug.Enable()
		cryptfs.Debug.Printf("Debug output enabled\n")
	}
	if args.openssl == false {
		cryptfs.Info.Printf("Openssl disabled\n")
	}
	if args.init {
		if flagSet.NArg() != 1 && args.plaintextnames == false {
			fmt.Printf("Usage: %s --init [--plaintextnames] CIPHERDIR\n", PROGRAM_NAME)
			os.Exit(ERREXIT_USAGE)
		}
		initDir(flagSet.Arg(0), args.plaintextnames) // does not return
	}
	if args.passwd {
		if flagSet.NArg() != 1 {
			fmt.Printf("Usage: %s --passwd CIPHERDIR\n", PROGRAM_NAME)
			os.Exit(ERREXIT_USAGE)
		}
	} else {
		// Normal mount
		if flagSet.NArg() < 2 {
			usageText()
			os.Exit(ERREXIT_USAGE)
		}
		args.mountpoint, _ = filepath.Abs(flagSet.Arg(1))
		err := checkDirEmpty(args.mountpoint)
		if err != nil {
			fmt.Printf("Invalid MOUNTPOINT: %v\n", err)
			os.Exit(ERREXIT_MOUNTPOINT)
		}
	}
	args.cipherdir, _ = filepath.Abs(flagSet.Arg(0))
	err := checkDir(args.cipherdir)
	if err != nil {
		fmt.Printf("Invalid CIPHERDIR: %v\n", err)
		os.Exit(ERREXIT_CIPHERDIR)
	}

	var plaintextNames bool
	var cf *cryptfs.ConfFile
	var currentPassword string
	key := make([]byte, cryptfs.KEY_LEN)
	if args.zerokey {
		cryptfs.Info.Printf("Zerokey mode active: using all-zero dummy master key.\n")
		plaintextNames = args.plaintextnames
	} else if len(args.masterkey) > 0 {
		key = parseMasterKey(args.masterkey)
		cryptfs.Info.Printf("Using explicit master key.\n")
	} else {
		// Load config file
		cfname := filepath.Join(args.cipherdir, cryptfs.ConfDefaultName)
		_, err = os.Stat(cfname)
		if err != nil {
			fmt.Printf("Error: %s not found in CIPHERDIR\n", cryptfs.ConfDefaultName)
			fmt.Printf("Please run \"%s --init %s\" first\n", os.Args[0], flagSet.Arg(0))
			os.Exit(ERREXIT_LOADCONF)
		}
		if args.passwd == true {
			fmt.Printf("Old password: ")
		} else {
			fmt.Printf("Password: ")
		}
		currentPassword = readPassword()
		cryptfs.Info.Printf("\nDecrypting master key... ")
		cryptfs.Warn.Disable() // Silence DecryptBlock() error messages on incorrect password
		key, cf, err = cryptfs.LoadConfFile(cfname, currentPassword)
		cryptfs.Warn.Enable()
		if err != nil {
			fmt.Println(err)
			os.Exit(ERREXIT_LOADCONF)
		}
		cryptfs.Info.Printf("done.\n")
	}
{'fraction_non_alphanumeric': 0.14251655629139073, 'fraction_numerical': 0.004061810154525386, 'mean_word_length': 4.129528985507246, 'pattern_counts': {'":': 0, '<': 7, '<?xml version=': 0, '>': 7, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6024509', 'n_tokens_mistral': 4547, 'n_tokens_neox': 4273, 'n_words': 925}