blue
/
croc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added fallback to recieve file in pwd if target_dir doesn't exist

This commit is contained in:
Sahil Sarwar 2024-11-10 21:16:36 +05:30
parent 30684fb383
commit 90fead89a6
1 changed files with 24 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/cli/cli.go
View File

@ -643,7 +643,13 @@ Or you can go back to the classic croc behavior by enabling classic mode:
}
}
if c.String("out") != "" {
if err = os.Chdir(c.String("out")); err != nil {
// validate the path
destPath, err := validateDestPath(c.String("out"))
if err != nil {
return err
}
if err = os.Chdir(destPath); err != nil {
return err
}
}
@ -714,3 +720,20 @@ func relay(c *cli.Context) (err error) {
}
return tcp.Run(debugString, host, ports[0], determinePass(c), tcpPorts)
}
func validateDestPath(path string) (string, error) {
_, err := os.Stat(path)
if err != nil {
// Path doesn't exist, fallback to current directory
currentDir, err := os.Getwd()
if err != nil {
return "", fmt.Errorf("could not get current directory: %w", err)
}
log.Debugf("Path %s does not exist, falling back to current directory: %s", path, currentDir)
fmt.Println("Target directory doesn't exist, receiving in current directory")
// flush the stdout buffer
os.Stdout.Sync()
return currentDir, nil
}
return path, nil
}