Account for the root

[rbdr/mobius] / hotline / file_path.go

diff --git a/hotline/file_path.go b/hotline/file_path.go
index 46ef97c95915bd786917df4028b6883d2defbd4e..bb6bbd312b62864591b893774930d23da5b19227 100644 (file)
--- a/hotline/file_path.go
+++ b/hotline/file_path.go
@@ -5,6 +5,7 @@ import (
        "bytes"
        "encoding/binary"
        "errors"
        "bytes"
        "encoding/binary"
        "errors"

+       "fmt"

        "io"
        "path/filepath"
        "strings"
        "io"
        "path/filepath"
        "strings"


@@ -34,6 +35,9 @@ func fileItemScanner(data []byte, _ bool) (advance int, token []byte, err error)
 
 // Write implements the io.Writer interface for FilePathItem
 func (fpi *FilePathItem) Write(b []byte) (n int, err error) {
 
 // Write implements the io.Writer interface for FilePathItem
 func (fpi *FilePathItem) Write(b []byte) (n int, err error) {

+       if len(b) < 3 {
+               return n, errors.New("buflen too small")
+       }

        fpi.Len = b[2]
        fpi.Name = b[fileItemMinLen : fpi.Len+fileItemMinLen]
 
        fpi.Len = b[2]
        fpi.Name = b[fileItemMinLen : fpi.Len+fileItemMinLen]
 


@@ -45,6 +49,7 @@ type FilePath struct {
        Items     []FilePathItem
 }
 
        Items     []FilePathItem
 }
 

+// Write implements io.Writer interface for FilePath

 func (fp *FilePath) Write(b []byte) (n int, err error) {
        reader := bytes.NewReader(b)
        err = binary.Read(reader, binary.BigEndian, &fp.ItemCount)
 func (fp *FilePath) Write(b []byte) (n int, err error) {
        reader := bytes.NewReader(b)
        err = binary.Read(reader, binary.BigEndian, &fp.ItemCount)


@@ -61,7 +66,13 @@ func (fp *FilePath) Write(b []byte) (n int, err error) {
        for i := 0; i < int(binary.BigEndian.Uint16(fp.ItemCount[:])); i++ {
                var fpi FilePathItem
                scanner.Scan()
        for i := 0; i < int(binary.BigEndian.Uint16(fp.ItemCount[:])); i++ {
                var fpi FilePathItem
                scanner.Scan()

-               if _, err := fpi.Write(scanner.Bytes()); err != nil {
+
+               // Make a new []byte slice and copy the scanner bytes to it.  This is critical to avoid a data race as the
+               // scanner re-uses the buffer for subsequent scans.
+               buf := make([]byte, len(scanner.Bytes()))
+               copy(buf, scanner.Bytes())
+
+               if _, err := fpi.Write(buf); err != nil {

                        return n, err
                }
                fp.Items = append(fp.Items, fpi)
                        return n, err
                }
                fp.Items = append(fp.Items, fpi)


@@ -87,11 +98,20 @@ func (fp *FilePath) IsUploadDir() bool {
        return strings.Contains(strings.ToLower(string(fp.Items[fp.Len()-1].Name)), "upload")
 }
 
        return strings.Contains(strings.ToLower(string(fp.Items[fp.Len()-1].Name)), "upload")
 }
 

+func (fp *FilePath) IsUserDir() bool {
+    if fp.Len() == 0 {
+        return false
+    }
+
+    return strings.HasPrefix(string(fp.Items[0].Name), "~")
+}
+
+

 func (fp *FilePath) Len() uint16 {
        return binary.BigEndian.Uint16(fp.ItemCount[:])
 }
 
 func (fp *FilePath) Len() uint16 {
        return binary.BigEndian.Uint16(fp.ItemCount[:])
 }
 

-func readPath(fileRoot string, filePath, fileName []byte) (fullPath string, err error) {
+func ReadPath(fileRoot string, filePath, fileName []byte) (fullPath string, err error) {

        var fp FilePath
        if filePath != nil {
                if _, err = fp.Write(filePath); err != nil {
        var fp FilePath
        if filePath != nil {
                if _, err = fp.Write(filePath); err != nil {


@@ -109,6 +129,9 @@ func readPath(fileRoot string, filePath, fileName []byte) (fullPath string, err
                subPath,
                filepath.Join("/", string(fileName)),
        )
                subPath,
                filepath.Join("/", string(fileName)),
        )

-
+       fullPath, err = txtDecoder.String(fullPath)
+       if err != nil {
+               return "", fmt.Errorf("invalid filepath encoding: %w", err)
+       }

        return fullPath, nil
 }
        return fullPath, nil
 }