Compare commits

..

No commits in common. "0ed9fa3290b7755212cfb18b03c7b567795a74f4" and "b92f1face5d19fa1f6800a3aa6c2f8f80e3f2278" have entirely different histories.

7 changed files with 28 additions and 70 deletions

View file

@ -18,11 +18,9 @@ func main() {
} }
// Serve the "public" directory on the "/webdav/" route. // Serve the "public" directory on the "/webdav/" route.
closer, err := webdav.Serve(r, "/webdav", "public") if err := webdav.Serve(r, "/webdav", "public"); err != nil {
if err != nil {
log.Fatal(err) log.Fatal(err)
} }
defer closer.Close()
log.Println("Touka WebDAV Server starting on :8080...") log.Println("Touka WebDAV Server starting on :8080...")
if err := r.RunShutdown(":8080", 10*time.Second); err != nil { if err := r.RunShutdown(":8080", 10*time.Second); err != nil {

View file

@ -5,7 +5,6 @@
package webdav package webdav
import ( import (
"io"
"log" "log"
"os" "os"
@ -21,6 +20,10 @@ type Config struct {
// Register registers a WebDAV handler on the given router. // Register registers a WebDAV handler on the given router.
func Register(engine *touka.Engine, prefix string, cfg *Config) { func Register(engine *touka.Engine, prefix string, cfg *Config) {
if cfg.LockSystem == nil {
cfg.LockSystem = NewMemLock()
}
handler := NewHandler(prefix, cfg.FileSystem, cfg.LockSystem, cfg.Logger) handler := NewHandler(prefix, cfg.FileSystem, cfg.LockSystem, cfg.Logger)
webdavMethods := []string{ webdavMethods := []string{
@ -30,18 +33,16 @@ func Register(engine *touka.Engine, prefix string, cfg *Config) {
} }
// Serve serves a local directory via WebDAV. // Serve serves a local directory via WebDAV.
func Serve(engine *touka.Engine, prefix string, rootDir string) (io.Closer, error) { func Serve(engine *touka.Engine, prefix string, rootDir string) error {
fs, err := NewOSFS(rootDir) fs, err := NewOSFS(rootDir)
if err != nil { if err != nil {
return nil, err return err
} }
ls := NewMemLock()
cfg := &Config{ cfg := &Config{
FileSystem: fs, FileSystem: fs,
LockSystem: ls,
Logger: log.New(os.Stdout, "", 0), Logger: log.New(os.Stdout, "", 0),
} }
Register(engine, prefix, cfg) Register(engine, prefix, cfg)
return ls, nil return nil
} }

View file

@ -17,7 +17,6 @@ func TestRegister(t *testing.T) {
r := touka.New() r := touka.New()
cfg := &Config{ cfg := &Config{
FileSystem: NewMemFS(), FileSystem: NewMemFS(),
LockSystem: NewMemLock(),
} }
Register(r, "/dav", cfg) Register(r, "/dav", cfg)
@ -36,11 +35,9 @@ func TestServe(t *testing.T) {
dir, _ := os.MkdirTemp("", "webdav") dir, _ := os.MkdirTemp("", "webdav")
defer os.RemoveAll(dir) defer os.RemoveAll(dir)
closer, err := Serve(r, "/serve", dir) if err := Serve(r, "/serve", dir); err != nil {
if err != nil {
t.Fatalf("Serve failed: %v", err) t.Fatalf("Serve failed: %v", err)
} }
defer closer.Close()
// Check if a WebDAV method is registered // Check if a WebDAV method is registered
req, _ := http.NewRequest("OPTIONS", "/serve/", nil) req, _ := http.NewRequest("OPTIONS", "/serve/", nil)

View file

@ -131,35 +131,16 @@ func (fs *MemFS) RemoveAll(ctx context.Context, name string) error {
fs.mu.Lock() fs.mu.Lock()
defer fs.mu.Unlock() defer fs.mu.Unlock()
cleanPath := path.Clean(name) dir, base := path.Split(name)
if cleanPath == "/" {
return os.ErrInvalid
}
dir, base := path.Split(cleanPath)
parent, err := fs.findNode(dir) parent, err := fs.findNode(dir)
if err != nil { if err != nil {
return err return err
} }
node, exists := parent.children[base] if _, exists := parent.children[base]; !exists {
if !exists {
return os.ErrNotExist return os.ErrNotExist
} }
var recursiveDelete func(*memNode)
recursiveDelete = func(n *memNode) {
if n.isDir {
for _, child := range n.children {
recursiveDelete(child)
}
}
n.parent = nil
n.children = nil
n.data = nil
}
recursiveDelete(node)
delete(parent.children, base) delete(parent.children, base)
return nil return nil
} }
@ -259,34 +240,17 @@ func (f *memFile) Read(p []byte) (n int, err error) {
func (f *memFile) Write(p []byte) (n int, err error) { func (f *memFile) Write(p []byte) (n int, err error) {
f.fs.mu.Lock() f.fs.mu.Lock()
defer f.fs.mu.Unlock() defer f.fs.mu.Unlock()
newSize := f.offset + int64(len(p))
writeEnd := f.offset + int64(len(p)) if newSize > int64(cap(f.node.data)) {
newData := make([]byte, newSize)
// Grow slice if necessary
if writeEnd > int64(cap(f.node.data)) {
newCap := int64(cap(f.node.data)) * 2
if newCap < writeEnd {
newCap = writeEnd
}
newData := make([]byte, len(f.node.data), newCap)
copy(newData, f.node.data) copy(newData, f.node.data)
f.node.data = newData f.node.data = newData
} else {
f.node.data = f.node.data[:newSize]
} }
// Extend slice length if write goes past the end
if writeEnd > int64(len(f.node.data)) {
f.node.data = f.node.data[:writeEnd]
}
n = copy(f.node.data[f.offset:], p) n = copy(f.node.data[f.offset:], p)
f.offset += int64(n) f.offset += int64(n)
atomic.StoreInt64(&f.node.size, newSize)
// Update size only if the file has grown
if f.offset > atomic.LoadInt64(&f.node.size) {
atomic.StoreInt64(&f.node.size, f.offset)
}
f.node.modTime = time.Now()
return n, nil return n, nil
} }

View file

@ -38,9 +38,8 @@ func NewMemLock() *MemLock {
} }
// Close stops the cleanup goroutine. // Close stops the cleanup goroutine.
func (l *MemLock) Close() error { func (l *MemLock) Close() {
close(l.stop) close(l.stop)
return nil
} }
func (l *MemLock) cleanup() { func (l *MemLock) cleanup() {
@ -67,13 +66,6 @@ func (l *MemLock) Create(ctx context.Context, path string, info LockInfo) (strin
l.mu.Lock() l.mu.Lock()
defer l.mu.Unlock() defer l.mu.Unlock()
// Check for conflicting locks
for _, v := range l.locks {
if v.path == path {
return "", os.ErrExist
}
}
token := make([]byte, 16) token := make([]byte, 16)
if _, err := rand.Read(token); err != nil { if _, err := rand.Read(token); err != nil {
return "", err return "", err

View file

@ -28,7 +28,7 @@ func NewOSFS(rootDir string) (*OSFS, error) {
} }
func (fs *OSFS) resolve(name string) (string, error) { func (fs *OSFS) resolve(name string) (string, error) {
if strings.Contains(name, "..") { if filepath.IsAbs(name) || strings.Contains(name, "..") {
return "", os.ErrPermission return "", os.ErrPermission
} }

View file

@ -588,8 +588,11 @@ func (h *Handler) handlePropfind(c *touka.Context) {
} }
func (h *Handler) createPropfindResponse(p string, info ObjectInfo, propfind Propfind) *Response { func (h *Handler) createPropfindResponse(path string, info ObjectInfo, propfind Propfind) *Response {
fullPath := path.Join(h.Prefix, p) fullPath := path
if h.Prefix != "/" {
fullPath = h.Prefix + path
}
resp := &Response{ resp := &Response{
Href: []string{fullPath}, Href: []string{fullPath},
@ -638,7 +641,10 @@ func (h *Handler) handleProppatch(c *touka.Context) {
} }
func (h *Handler) stripPrefix(p string) string { func (h *Handler) stripPrefix(p string) string {
return strings.TrimPrefix(strings.TrimPrefix(p, h.Prefix), "/") if h.Prefix == "/" {
return p
}
return strings.TrimPrefix(p, h.Prefix)
} }
func (h *Handler) handleLock(c *touka.Context) { func (h *Handler) handleLock(c *touka.Context) {