mirror of
https://github.com/infinite-iroha/touka.git
synced 2026-02-03 08:51:11 +08:00
Compare commits
No commits in common. "0ed9fa3290b7755212cfb18b03c7b567795a74f4" and "b92f1face5d19fa1f6800a3aa6c2f8f80e3f2278" have entirely different histories.
0ed9fa3290
...
b92f1face5
7 changed files with 28 additions and 70 deletions
|
|
@ -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 {
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
|
|
|
||||||
|
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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) {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue