internal_test.go 52 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450
  1. package webdavd
  2. import (
  3. "context"
  4. "crypto/tls"
  5. "crypto/x509"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "net/http"
  10. "net/http/httptest"
  11. "os"
  12. "path"
  13. "path/filepath"
  14. "runtime"
  15. "testing"
  16. "time"
  17. "github.com/eikenb/pipeat"
  18. "github.com/stretchr/testify/assert"
  19. "golang.org/x/net/webdav"
  20. "github.com/drakkan/sftpgo/common"
  21. "github.com/drakkan/sftpgo/dataprovider"
  22. "github.com/drakkan/sftpgo/kms"
  23. "github.com/drakkan/sftpgo/vfs"
  24. )
  25. const (
  26. testFile = "test_dav_file"
  27. webDavCert = `-----BEGIN CERTIFICATE-----
  28. MIICHTCCAaKgAwIBAgIUHnqw7QnB1Bj9oUsNpdb+ZkFPOxMwCgYIKoZIzj0EAwIw
  29. RTELMAkGA1UEBhMCQVUxEzARBgNVBAgMClNvbWUtU3RhdGUxITAfBgNVBAoMGElu
  30. dGVybmV0IFdpZGdpdHMgUHR5IEx0ZDAeFw0yMDAyMDQwOTUzMDRaFw0zMDAyMDEw
  31. OTUzMDRaMEUxCzAJBgNVBAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYD
  32. VQQKDBhJbnRlcm5ldCBXaWRnaXRzIFB0eSBMdGQwdjAQBgcqhkjOPQIBBgUrgQQA
  33. IgNiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVqWvrJ51t5OxV0v25NsOgR82CA
  34. NXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIVCzgWkxiz7XE4lgUwX44FCXZM
  35. 3+JeUbKjUzBRMB0GA1UdDgQWBBRhLw+/o3+Z02MI/d4tmaMui9W16jAfBgNVHSME
  36. GDAWgBRhLw+/o3+Z02MI/d4tmaMui9W16jAPBgNVHRMBAf8EBTADAQH/MAoGCCqG
  37. SM49BAMCA2kAMGYCMQDqLt2lm8mE+tGgtjDmtFgdOcI72HSbRQ74D5rYTzgST1rY
  38. /8wTi5xl8TiFUyLMUsICMQC5ViVxdXbhuG7gX6yEqSkMKZICHpO8hqFwOD/uaFVI
  39. dV4vKmHUzwK/eIx+8Ay3neE=
  40. -----END CERTIFICATE-----`
  41. webDavKey = `-----BEGIN EC PARAMETERS-----
  42. BgUrgQQAIg==
  43. -----END EC PARAMETERS-----
  44. -----BEGIN EC PRIVATE KEY-----
  45. MIGkAgEBBDCfMNsN6miEE3rVyUPwElfiJSWaR5huPCzUenZOfJT04GAcQdWvEju3
  46. UM2lmBLIXpGgBwYFK4EEACKhZANiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVq
  47. WvrJ51t5OxV0v25NsOgR82CANXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIV
  48. CzgWkxiz7XE4lgUwX44FCXZM3+JeUbI=
  49. -----END EC PRIVATE KEY-----`
  50. caCRT = `-----BEGIN CERTIFICATE-----
  51. MIIE5jCCAs6gAwIBAgIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0
  52. QXV0aDAeFw0yMTAxMDIyMTIwNTVaFw0yMjA3MDIyMTMwNTJaMBMxETAPBgNVBAMT
  53. CENlcnRBdXRoMIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEA4Tiho5xW
  54. AC15JRkMwfp3/TJwI2As7MY5dele5cmdr5bHAE+sRKqC+Ti88OJWCV5saoyax/1S
  55. CjxJlQMZMl169P1QYJskKjdG2sdv6RLWLMgwSNRRjxp/Bw9dHdiEb9MjLgu28Jro
  56. 9peQkHcRHeMf5hM9WvlIJGrdzbC4hUehmqggcqgARainBkYjf0SwuWxHeu4nMqkp
  57. Ak5tcSTLCjHfEFHZ9Te0TIPG5YkWocQKyeLgu4lvuU+DD2W2lym+YVUtRMGs1Env
  58. k7p+N0DcGU26qfzZ2sF5ZXkqm7dBsGQB9pIxwc2Q8T1dCIyP9OQCKVILdc5aVFf1
  59. cryQFHYzYNNZXFlIBims5VV5Mgfp8ESHQSue+v6n6ykecLEyKt1F1Y/MWY/nWUSI
  60. 8zdq83jdBAZVjo9MSthxVn57/06s/hQca65IpcTZV2gX0a+eRlAVqaRbAhL3LaZe
  61. bYsW3WHKoUOftwemuep3nL51TzlXZVL7Oz/ClGaEOsnGG9KFO6jh+W768qC0zLQI
  62. CdE7v2Zex98sZteHCg9fGJHIaYoF0aJG5P3WI5oZf2fy7UIYN9ADLFZiorCXAZEh
  63. CSU6mDoRViZ4RGR9GZxbDZ9KYn7O8M/KCR72bkQg73TlMsk1zSXEw0MKLUjtsw6c
  64. rZ0Jt8t3sRatHO3JrYHALMt9vZfyNCZp0IsCAwEAAaNFMEMwDgYDVR0PAQH/BAQD
  65. AgEGMBIGA1UdEwEB/wQIMAYBAf8CAQAwHQYDVR0OBBYEFO1yCNAGr/zQTJIi8lw3
  66. w5OiuBvMMA0GCSqGSIb3DQEBCwUAA4ICAQA6gCNuM7r8mnx674dm31GxBjQy5ZwB
  67. 7CxDzYEvL/oiZ3Tv3HlPfN2LAAsJUfGnghh9DOytenL2CTZWjl/emP5eijzmlP+9
  68. zva5I6CIMCf/eDDVsRdO244t0o4uG7+At0IgSDM3bpVaVb4RHZNjEziYChsEYY8d
  69. HK6iwuRSvFniV6yhR/Vj1Ymi9yZ5xclqseLXiQnUB0PkfIk23+7s42cXB16653fH
  70. O/FsPyKBLiKJArizLYQc12aP3QOrYoYD9+fAzIIzew7A5C0aanZCGzkuFpO6TRlD
  71. Tb7ry9Gf0DfPpCgxraH8tOcmnqp/ka3hjqo/SRnnTk0IFrmmLdarJvjD46rKwBo4
  72. MjyAIR1mQ5j8GTlSFBmSgETOQ/EYvO3FPLmra1Fh7L+DvaVzTpqI9fG3TuyyY+Ri
  73. Fby4ycTOGSZOe5Fh8lqkX5Y47mCUJ3zHzOA1vUJy2eTlMRGpu47Eb1++Vm6EzPUP
  74. 2EF5aD+zwcssh+atZvQbwxpgVqVcyLt91RSkKkmZQslh0rnlTb68yxvUnD3zw7So
  75. o6TAf9UvwVMEvdLT9NnFd6hwi2jcNte/h538GJwXeBb8EkfpqLKpTKyicnOdkamZ
  76. 7E9zY8SHNRYMwB9coQ/W8NvufbCgkvOoLyMXk5edbXofXl3PhNGOlraWbghBnzf5
  77. r3rwjFsQOoZotA==
  78. -----END CERTIFICATE-----`
  79. caKey = `-----BEGIN RSA PRIVATE KEY-----
  80. MIIJKQIBAAKCAgEA4Tiho5xWAC15JRkMwfp3/TJwI2As7MY5dele5cmdr5bHAE+s
  81. RKqC+Ti88OJWCV5saoyax/1SCjxJlQMZMl169P1QYJskKjdG2sdv6RLWLMgwSNRR
  82. jxp/Bw9dHdiEb9MjLgu28Jro9peQkHcRHeMf5hM9WvlIJGrdzbC4hUehmqggcqgA
  83. RainBkYjf0SwuWxHeu4nMqkpAk5tcSTLCjHfEFHZ9Te0TIPG5YkWocQKyeLgu4lv
  84. uU+DD2W2lym+YVUtRMGs1Envk7p+N0DcGU26qfzZ2sF5ZXkqm7dBsGQB9pIxwc2Q
  85. 8T1dCIyP9OQCKVILdc5aVFf1cryQFHYzYNNZXFlIBims5VV5Mgfp8ESHQSue+v6n
  86. 6ykecLEyKt1F1Y/MWY/nWUSI8zdq83jdBAZVjo9MSthxVn57/06s/hQca65IpcTZ
  87. V2gX0a+eRlAVqaRbAhL3LaZebYsW3WHKoUOftwemuep3nL51TzlXZVL7Oz/ClGaE
  88. OsnGG9KFO6jh+W768qC0zLQICdE7v2Zex98sZteHCg9fGJHIaYoF0aJG5P3WI5oZ
  89. f2fy7UIYN9ADLFZiorCXAZEhCSU6mDoRViZ4RGR9GZxbDZ9KYn7O8M/KCR72bkQg
  90. 73TlMsk1zSXEw0MKLUjtsw6crZ0Jt8t3sRatHO3JrYHALMt9vZfyNCZp0IsCAwEA
  91. AQKCAgAV+ElERYbaI5VyufvVnFJCH75ypPoc6sVGLEq2jbFVJJcq/5qlZCC8oP1F
  92. Xj7YUR6wUiDzK1Hqb7EZ2SCHGjlZVrCVi+y+NYAy7UuMZ+r+mVSkdhmypPoJPUVv
  93. GOTqZ6VB46Cn3eSl0WknvoWr7bD555yPmEuiSc5zNy74yWEJTidEKAFGyknowcTK
  94. sG+w1tAuPLcUKQ44DGB+rgEkcHL7C5EAa7upzx0C3RmZFB+dTAVyJdkBMbFuOhTS
  95. sB7DLeTplR7/4mp9da7EQw51ZXC1DlZOEZt++4/desXsqATNAbva1OuzrLG7mMKe
  96. N/PCBh/aERQcsCvgUmaXqGQgqN1Jhw8kbXnjZnVd9iE7TAh7ki3VqNy1OMgTwOex
  97. bBYWaCqHuDYIxCjeW0qLJcn0cKQ13FVYrxgInf4Jp82SQht5b/zLL3IRZEyKcLJF
  98. kL6g1wlmTUTUX0z8eZzlM0ZCrqtExjgElMO/rV971nyNV5WU8Og3NmE8/slqMrmJ
  99. DlrQr9q0WJsDKj1IMe46EUM6ix7bbxC5NIfJ96dgdxZDn6ghjca6iZYqqUACvmUj
  100. cq08s3R4Ouw9/87kn11wwGBx2yDueCwrjKEGc0RKjweGbwu0nBxOrkJ8JXz6bAv7
  101. 1OKfYaX3afI9B8x4uaiuRs38oBQlg9uAYFfl4HNBPuQikGLmsQKCAQEA8VjFOsaz
  102. y6NMZzKXi7WZ48uu3ed5x3Kf6RyDr1WvQ1jkBMv9b6b8Gp1CRnPqviRBto9L8QAg
  103. bCXZTqnXzn//brskmW8IZgqjAlf89AWa53piucu9/hgidrHRZobs5gTqev28uJdc
  104. zcuw1g8c3nCpY9WeTjHODzX5NXYRLFpkazLfYa6c8Q9jZR4KKrpdM+66fxL0JlOd
  105. 7dN0oQtEqEAugsd3cwkZgvWhY4oM7FGErrZoDLy273ZdJzi/vU+dThyVzfD8Ab8u
  106. VxxuobVMT/S608zbe+uaiUdov5s96OkCl87403UNKJBH+6LNb3rjBBLE9NPN5ET9
  107. JLQMrYd+zj8jQwKCAQEA7uU5I9MOufo9bIgJqjY4Ie1+Ex9DZEMUYFAvGNCJCVcS
  108. mwOdGF8AWzIavTLACmEDJO7t/OrBdoo4L7IEsCNjgA3WiIwIMiWUVqveAGUMEXr6
  109. TRI5EolV6FTqqIP6AS+BAeBq7G1ELgsTrWNHh11rW3+3kBMuOCn77PUQ8WHwcq/r
  110. teZcZn4Ewcr6P7cBODgVvnBPhe/J8xHS0HFVCeS1CvaiNYgees5yA80Apo9IPjDJ
  111. YWawLjmH5wUBI5yDFVp067wjqJnoKPSoKwWkZXqUk+zgFXx5KT0gh/c5yh1frASp
  112. q6oaYnHEVC5qj2SpT1GFLonTcrQUXiSkiUudvNu1GQKCAQEAmko+5GFtRe0ihgLQ
  113. 4S76r6diJli6AKil1Fg3U1r6zZpBQ1PJtJxTJQyN9w5Z7q6tF/GqAesrzxevQdvQ
  114. rCImAPtA3ZofC2UXawMnIjWHHx6diNvYnV1+gtUQ4nO1dSOFZ5VZFcUmPiZO6boF
  115. oaryj3FcX+71JcJCjEvrlKhA9Es0hXUkvfMxfs5if4he1zlyHpTWYr4oA4egUugq
  116. P0mwskikc3VIyvEO+NyjgFxo72yLPkFSzemkidN8uKDyFqKtnlfGM7OuA2CY1WZa
  117. 3+67lXWshx9KzyJIs92iCYkU8EoPxtdYzyrV6efdX7x27v60zTOut5TnJJS6WiF6
  118. Do5MkwKCAQAxoR9IyP0DN/BwzqYrXU42Bi+t603F04W1KJNQNWpyrUspNwv41yus
  119. xnD1o0hwH41Wq+h3JZIBfV+E0RfWO9Pc84MBJQ5C1LnHc7cQH+3s575+Km3+4tcd
  120. CB8j2R8kBeloKWYtLdn/Mr/ownpGreqyvIq2/LUaZ+Z1aMgXTYB1YwS16mCBzmZQ
  121. mEl62RsAwe4KfSyYJ6OtwqMoOJMxFfliiLBULK4gVykqjvk2oQeiG+KKQJoTUFJi
  122. dRCyhD5bPkqR+qjxyt+HOqSBI4/uoROi05AOBqjpH1DVzk+MJKQOiX1yM0l98CKY
  123. Vng+x+vAla/0Zh+ucajVkgk4mKPxazdpAoIBAQC17vWk4KYJpF2RC3pKPcQ0PdiX
  124. bN35YNlvyhkYlSfDNdyH3aDrGiycUyW2mMXUgEDFsLRxHMTL+zPC6efqO6sTAJDY
  125. cBptsW4drW/qo8NTx3dNOisLkW+mGGJOR/w157hREFr29ymCVMYu/Z7fVWIeSpCq
  126. p3u8YX8WTljrxwSczlGjvpM7uJx3SfYRM4TUoy+8wU8bK74LywLa5f60bQY6Dye0
  127. Gqd9O6OoPfgcQlwjC5MiAofeqwPJvU0hQOPoehZyNLAmOCWXTYWaTP7lxO1r6+NE
  128. M3hGYqW3W8Ixua71OskCypBZg/HVlIP/lzjRzdx+VOB2hbWVth2Iup/Z1egW
  129. -----END RSA PRIVATE KEY-----`
  130. caCRL = `-----BEGIN X509 CRL-----
  131. MIICpzCBkAIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0QXV0aBcN
  132. MjEwMTAyMjEzNDA1WhcNMjMwMTAyMjEzNDA1WjAkMCICEQC+l04DbHWMyC3fG09k
  133. VXf+Fw0yMTAxMDIyMTM0MDVaoCMwITAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJc
  134. N8OTorgbzDANBgkqhkiG9w0BAQsFAAOCAgEAEJ7z+uNc8sqtxlOhSdTGDzX/xput
  135. E857kFQkSlMnU2whQ8c+XpYrBLA5vIZJNSSwohTpM4+zVBX/bJpmu3wqqaArRO9/
  136. YcW5mQk9Anvb4WjQW1cHmtNapMTzoC9AiYt/OWPfy+P6JCgCr4Hy6LgQyIRL6bM9
  137. VYTalolOm1qa4Y5cIeT7iHq/91mfaqo8/6MYRjLl8DOTROpmw8OS9bCXkzGKdCat
  138. AbAzwkQUSauyoCQ10rpX+Y64w9ng3g4Dr20aCqPf5osaqplEJ2HTK8ljDTidlslv
  139. 9anQj8ax3Su89vI8+hK+YbfVQwrThabgdSjQsn+veyx8GlP8WwHLAQ379KjZjWg+
  140. OlOSwBeU1vTdP0QcB8X5C2gVujAyuQekbaV86xzIBOj7vZdfHZ6ee30TZ2FKiMyg
  141. 7/N2OqW0w77ChsjB4MSHJCfuTgIeg62GzuZXLM+Q2Z9LBdtm4Byg+sm/P52adOEg
  142. gVb2Zf4KSvsAmA0PIBlu449/QXUFcMxzLFy7mwTeZj2B4Ln0Hm0szV9f9R8MwMtB
  143. SyLYxVH+mgqaR6Jkk22Q/yYyLPaELfafX5gp/AIXG8n0zxfVaTvK3auSgb1Q6ZLS
  144. 5QH9dSIsmZHlPq7GoSXmKpMdjUL8eaky/IMteioyXgsBiATzl5L2dsw6MTX3MDF0
  145. QbDK+MzhmbKfDxs=
  146. -----END X509 CRL-----`
  147. client1Crt = `-----BEGIN CERTIFICATE-----
  148. MIIEITCCAgmgAwIBAgIRAIppZHoj1hM80D7WzTEKLuAwDQYJKoZIhvcNAQELBQAw
  149. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjEwMTAyMjEyMzEwWhcNMjIwNzAyMjEz
  150. MDUxWjASMRAwDgYDVQQDEwdjbGllbnQxMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  151. MIIBCgKCAQEAoKbYY9MdF2kF/nhBESIiZTdVYtA8XL9xrIZyDj9EnCiTxHiVbJtH
  152. XVwszqSl5TRrotPmnmAQcX3r8OCk+z+RQZ0QQj257P3kG6q4rNnOcWCS5xEd20jP
  153. yhQ3m+hMGfZsotNTQze1ochuQgLUN6IPyPxZkH22ia3jX4iu1eo/QxeLYHj1UHw4
  154. 3Cii9yE+j5kPUC21xmnrGKdUrB55NYLXHx6yTIqYR5znSOVB8oJi18/hwdZmH859
  155. DHhm0Hx1HrS+jbjI3+CMorZJ3WUyNf+CkiVLD3xYutPbxzEpwiqkG/XYzLH0habT
  156. cDcILo18n+o3jvem2KWBrDhyairjIDscwQIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  157. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBSJ5GIv
  158. zIrE4ZSQt2+CGblKTDswizAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJcN8OTorgb
  159. zDANBgkqhkiG9w0BAQsFAAOCAgEALh4f5GhvNYNou0Ab04iQBbLEdOu2RlbK1B5n
  160. K9P/umYenBHMY/z6HT3+6tpcHsDuqE8UVdq3f3Gh4S2Gu9m8PRitT+cJ3gdo9Plm
  161. 3rD4ufn/s6rGg3ppydXcedm17492tbccUDWOBZw3IO/ASVq13WPgT0/Kev7cPq0k
  162. sSdSNhVeXqx8Myc2/d+8GYyzbul2Kpfa7h9i24sK49E9ftnSmsIvngONo08eT1T0
  163. 3wAOyK2981LIsHaAWcneShKFLDB6LeXIT9oitOYhiykhFlBZ4M1GNlSNfhQ8IIQP
  164. xbqMNXCLkW4/BtLhGEEcg0QVso6Kudl9rzgTfQknrdF7pHp6rS46wYUjoSyIY6dl
  165. oLmnoAVJX36J3QPWelePI9e07X2wrTfiZWewwgw3KNRWjd6/zfPLe7GoqXnK1S2z
  166. PT8qMfCaTwKTtUkzXuTFvQ8bAo2My/mS8FOcpkt2oQWeOsADHAUX7fz5BCoa2DL3
  167. k/7Mh4gVT+JYZEoTwCFuYHgMWFWe98naqHi9lB4yR981p1QgXgxO7qBeipagKY1F
  168. LlH1iwXUqZ3MZnkNA+4e1Fglsw3sa/rC+L98HnznJ/YbTfQbCP6aQ1qcOymrjMud
  169. 7MrFwqZjtd/SK4Qx1VpK6jGEAtPgWBTUS3p9ayg6lqjMBjsmySWfvRsDQbq6P5Ct
  170. O/e3EH8=
  171. -----END CERTIFICATE-----`
  172. client1Key = `-----BEGIN RSA PRIVATE KEY-----
  173. MIIEpAIBAAKCAQEAoKbYY9MdF2kF/nhBESIiZTdVYtA8XL9xrIZyDj9EnCiTxHiV
  174. bJtHXVwszqSl5TRrotPmnmAQcX3r8OCk+z+RQZ0QQj257P3kG6q4rNnOcWCS5xEd
  175. 20jPyhQ3m+hMGfZsotNTQze1ochuQgLUN6IPyPxZkH22ia3jX4iu1eo/QxeLYHj1
  176. UHw43Cii9yE+j5kPUC21xmnrGKdUrB55NYLXHx6yTIqYR5znSOVB8oJi18/hwdZm
  177. H859DHhm0Hx1HrS+jbjI3+CMorZJ3WUyNf+CkiVLD3xYutPbxzEpwiqkG/XYzLH0
  178. habTcDcILo18n+o3jvem2KWBrDhyairjIDscwQIDAQABAoIBAEBSjVFqtbsp0byR
  179. aXvyrtLX1Ng7h++at2jca85Ihq//jyqbHTje8zPuNAKI6eNbmb0YGr5OuEa4pD9N
  180. ssDmMsKSoG/lRwwcm7h4InkSvBWpFShvMgUaohfHAHzsBYxfnh+TfULsi0y7c2n6
  181. t/2OZcOTRkkUDIITnXYiw93ibHHv2Mv2bBDu35kGrcK+c2dN5IL5ZjTjMRpbJTe2
  182. 44RBJbdTxHBVSgoGBnugF+s2aEma6Ehsj70oyfoVpM6Aed5kGge0A5zA1JO7WCn9
  183. Ay/DzlULRXHjJIoRWd2NKvx5n3FNppUc9vJh2plRHalRooZ2+MjSf8HmXlvG2Hpb
  184. ScvmWgECgYEA1G+A/2KnxWsr/7uWIJ7ClcGCiNLdk17Pv3DZ3G4qUsU2ITftfIbb
  185. tU0Q/b19na1IY8Pjy9ptP7t74/hF5kky97cf1FA8F+nMj/k4+wO8QDI8OJfzVzh9
  186. PwielA5vbE+xmvis5Hdp8/od1Yrc/rPSy2TKtPFhvsqXjqoUmOAjDP8CgYEAwZjH
  187. 9dt1sc2lx/rMxihlWEzQ3JPswKW9/LJAmbRBoSWF9FGNjbX7uhWtXRKJkzb8ZAwa
  188. 88azluNo2oftbDD/+jw8b2cDgaJHlLAkSD4O1D1RthW7/LKD15qZ/oFsRb13NV85
  189. ZNKtwslXGbfVNyGKUVFm7fVA8vBAOUey+LKDFj8CgYEAg8WWstOzVdYguMTXXuyb
  190. ruEV42FJaDyLiSirOvxq7GTAKuLSQUg1yMRBIeQEo2X1XU0JZE3dLodRVhuO4EXP
  191. g7Dn4X7Th9HSvgvNuIacowWGLWSz4Qp9RjhGhXhezUSx2nseY6le46PmFavJYYSR
  192. 4PBofMyt4PcyA6Cknh+KHmkCgYEAnTriG7ETE0a7v4DXUpB4TpCEiMCy5Xs2o8Z5
  193. ZNva+W+qLVUWq+MDAIyechqeFSvxK6gRM69LJ96lx+XhU58wJiFJzAhT9rK/g+jS
  194. bsHH9WOfu0xHkuHA5hgvvV2Le9B2wqgFyva4HJy82qxMxCu/VG/SMqyfBS9OWbb7
  195. ibQhdq0CgYAl53LUWZsFSZIth1vux2LVOsI8C3X1oiXDGpnrdlQ+K7z57hq5EsRq
  196. GC+INxwXbvKNqp5h0z2MvmKYPDlGVTgw8f8JjM7TkN17ERLcydhdRrMONUryZpo8
  197. 1xTob+8blyJgfxZUIAKbMbMbIiU0WAF0rfD/eJJwS4htOW/Hfv4TGA==
  198. -----END RSA PRIVATE KEY-----`
  199. // client 2 crt is revoked
  200. client2Crt = `-----BEGIN CERTIFICATE-----
  201. MIIEITCCAgmgAwIBAgIRAL6XTgNsdYzILd8bT2RVd/4wDQYJKoZIhvcNAQELBQAw
  202. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjEwMTAyMjEyMzIwWhcNMjIwNzAyMjEz
  203. MDUxWjASMRAwDgYDVQQDEwdjbGllbnQyMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  204. MIIBCgKCAQEA6xjW5KQR3/OFQtV5M75WINqQ4AzXSu6DhSz/yumaaQZP/UxY+6hi
  205. jcrFzGo9MMie/Sza8DhkXOFAl2BelUubrOeB2cl+/Gr8OCyRi2Gv6j3zCsuN/4jQ
  206. tNaoez/IbkDvI3l/ZpzBtnuNY2RiemGgHuORXHRVf3qVlsw+npBIRW5rM2HkO/xG
  207. oZjeBErWVu390Lyn+Gvk2TqQDnkutWnxUC60/zPlHhXZ4BwaFAekbSnjsSDB1YFM
  208. s8HwW4oBryoxdj3/+/qLrBHt75IdLw3T7/V1UDJQM3EvSQOr12w4egpldhtsC871
  209. nnBQZeY6qA5feffIwwg/6lJm70o6S6OX6wIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  210. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBTB84v5
  211. t9HqhLhMODbn6oYkEQt3KzAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJcN8OTorgb
  212. zDANBgkqhkiG9w0BAQsFAAOCAgEALGtBCve5k8tToL3oLuXp/oSik6ovIB/zq4I/
  213. 4zNMYPU31+ZWz6aahysgx1JL1yqTa3Qm8o2tu52MbnV10dM7CIw7c/cYa+c+OPcG
  214. 5LF97kp13X+r2axy+CmwM86b4ILaDGs2Qyai6VB6k7oFUve+av5o7aUrNFpqGCJz
  215. HWdtHZSVA3JMATzy0TfWanwkzreqfdw7qH0yZ9bDURlBKAVWrqnCstva9jRuv+AI
  216. eqxr/4Ro986TFjJdoAP3Vr16CPg7/B6GA/KmsBWJrpeJdPWq4i2gpLKvYZoy89qD
  217. mUZf34RbzcCtV4NvV1DadGnt4us0nvLrvS5rL2+2uWD09kZYq9RbLkvgzF/cY0fz
  218. i7I1bi5XQ+alWe0uAk5ZZL/D+GTRYUX1AWwCqwJxmHrMxcskMyO9pXvLyuSWRDLo
  219. YNBrbX9nLcfJzVCp+X+9sntTHjs4l6Cw+fLepJIgtgqdCHtbhTiv68vSM6cgb4br
  220. 6n2xrXRKuioiWFOrTSRr+oalZh8dGJ/xvwY8IbWknZAvml9mf1VvfE7Ma5P777QM
  221. fsbYVTq0Y3R/5hIWsC3HA5z6MIM8L1oRe/YyhP3CTmrCHkVKyDOosGXpGz+JVcyo
  222. cfYkY5A3yFKB2HaCwZSfwFmRhxkrYWGEbHv3Cd9YkZs1J3hNhGFZyVMC9Uh0S85a
  223. 6zdDidU=
  224. -----END CERTIFICATE-----`
  225. client2Key = `-----BEGIN RSA PRIVATE KEY-----
  226. MIIEpAIBAAKCAQEA6xjW5KQR3/OFQtV5M75WINqQ4AzXSu6DhSz/yumaaQZP/UxY
  227. +6hijcrFzGo9MMie/Sza8DhkXOFAl2BelUubrOeB2cl+/Gr8OCyRi2Gv6j3zCsuN
  228. /4jQtNaoez/IbkDvI3l/ZpzBtnuNY2RiemGgHuORXHRVf3qVlsw+npBIRW5rM2Hk
  229. O/xGoZjeBErWVu390Lyn+Gvk2TqQDnkutWnxUC60/zPlHhXZ4BwaFAekbSnjsSDB
  230. 1YFMs8HwW4oBryoxdj3/+/qLrBHt75IdLw3T7/V1UDJQM3EvSQOr12w4egpldhts
  231. C871nnBQZeY6qA5feffIwwg/6lJm70o6S6OX6wIDAQABAoIBAFatstVb1KdQXsq0
  232. cFpui8zTKOUiduJOrDkWzTygAmlEhYtrccdfXu7OWz0x0lvBLDVGK3a0I/TGrAzj
  233. 4BuFY+FM/egxTVt9in6fmA3et4BS1OAfCryzUdfK6RV//8L+t+zJZ/qKQzWnugpy
  234. QYjDo8ifuMFwtvEoXizaIyBNLAhEp9hnrv+Tyi2O2gahPvCHsD48zkyZRCHYRstD
  235. NH5cIrwz9/RJgPO1KI+QsJE7Nh7stR0sbr+5TPU4fnsL2mNhMUF2TJrwIPrc1yp+
  236. YIUjdnh3SO88j4TQT3CIrWi8i4pOy6N0dcVn3gpCRGaqAKyS2ZYUj+yVtLO4KwxZ
  237. SZ1lNvECgYEA78BrF7f4ETfWSLcBQ3qxfLs7ibB6IYo2x25685FhZjD+zLXM1AKb
  238. FJHEXUm3mUYrFJK6AFEyOQnyGKBOLs3S6oTAswMPbTkkZeD1Y9O6uv0AHASLZnK6
  239. pC6ub0eSRF5LUyTQ55Jj8D7QsjXJueO8v+G5ihWhNSN9tB2UA+8NBmkCgYEA+weq
  240. cvoeMIEMBQHnNNLy35bwfqrceGyPIRBcUIvzQfY1vk7KW6DYOUzC7u+WUzy/hA52
  241. DjXVVhua2eMQ9qqtOav7djcMc2W9RbLowxvno7K5qiCss013MeWk64TCWy+WMp5A
  242. AVAtOliC3hMkIKqvR2poqn+IBTh1449agUJQqTMCgYEAu06IHGq1GraV6g9XpGF5
  243. wqoAlMzUTdnOfDabRilBf/YtSr+J++ThRcuwLvXFw7CnPZZ4TIEjDJ7xjj3HdxeE
  244. fYYjineMmNd40UNUU556F1ZLvJfsVKizmkuCKhwvcMx+asGrmA+tlmds4p3VMS50
  245. KzDtpKzLWlmU/p/RINWlRmkCgYBy0pHTn7aZZx2xWKqCDg+L2EXPGqZX6wgZDpu7
  246. OBifzlfM4ctL2CmvI/5yPmLbVgkgBWFYpKUdiujsyyEiQvWTUKhn7UwjqKDHtcsk
  247. G6p7xS+JswJrzX4885bZJ9Oi1AR2yM3sC9l0O7I4lDbNPmWIXBLeEhGMmcPKv/Kc
  248. 91Ff4wKBgQCF3ur+Vt0PSU0ucrPVHjCe7tqazm0LJaWbPXL1Aw0pzdM2EcNcW/MA
  249. w0kqpr7MgJ94qhXCBcVcfPuFN9fBOadM3UBj1B45Cz3pptoK+ScI8XKno6jvVK/p
  250. xr5cb9VBRBtB9aOKVfuRhpatAfS2Pzm2Htae9lFn7slGPUmu2hkjDw==
  251. -----END RSA PRIVATE KEY-----`
  252. )
  253. var (
  254. errWalkDir = errors.New("err walk dir")
  255. errWalkFile = errors.New("err walk file")
  256. )
  257. // MockOsFs mockable OsFs
  258. type MockOsFs struct {
  259. vfs.Fs
  260. err error
  261. isAtomicUploadSupported bool
  262. reader *pipeat.PipeReaderAt
  263. }
  264. // Name returns the name for the Fs implementation
  265. func (fs *MockOsFs) Name() string {
  266. return "mockOsFs"
  267. }
  268. // Open returns nil
  269. func (fs *MockOsFs) Open(name string, offset int64) (vfs.File, *pipeat.PipeReaderAt, func(), error) {
  270. return nil, fs.reader, nil, nil
  271. }
  272. // IsUploadResumeSupported returns true if upload resume is supported
  273. func (*MockOsFs) IsUploadResumeSupported() bool {
  274. return false
  275. }
  276. // IsAtomicUploadSupported returns true if atomic upload is supported
  277. func (fs *MockOsFs) IsAtomicUploadSupported() bool {
  278. return fs.isAtomicUploadSupported
  279. }
  280. // Remove removes the named file or (empty) directory.
  281. func (fs *MockOsFs) Remove(name string, isDir bool) error {
  282. if fs.err != nil {
  283. return fs.err
  284. }
  285. return os.Remove(name)
  286. }
  287. // Rename renames (moves) source to target
  288. func (fs *MockOsFs) Rename(source, target string) error {
  289. if fs.err != nil {
  290. return fs.err
  291. }
  292. return os.Rename(source, target)
  293. }
  294. // Walk returns a duplicate path for testing
  295. func (fs *MockOsFs) Walk(root string, walkFn filepath.WalkFunc) error {
  296. if fs.err == errWalkDir {
  297. walkFn("fsdpath", vfs.NewFileInfo("dpath", true, 0, time.Now(), false), nil) //nolint:errcheck
  298. walkFn("fsdpath", vfs.NewFileInfo("dpath", true, 0, time.Now(), false), nil) //nolint:errcheck
  299. return nil
  300. }
  301. walkFn("fsfpath", vfs.NewFileInfo("fpath", false, 0, time.Now(), false), nil) //nolint:errcheck
  302. return fs.err
  303. }
  304. // GetMimeType returns the content type
  305. func (fs *MockOsFs) GetMimeType(name string) (string, error) {
  306. return "application/custom-mime", nil
  307. }
  308. func newMockOsFs(err error, atomicUpload bool, connectionID, rootDir string, reader *pipeat.PipeReaderAt) vfs.Fs {
  309. return &MockOsFs{
  310. Fs: vfs.NewOsFs(connectionID, rootDir, ""),
  311. err: err,
  312. isAtomicUploadSupported: atomicUpload,
  313. reader: reader,
  314. }
  315. }
  316. func TestOrderDirsToRemove(t *testing.T) {
  317. user := dataprovider.User{}
  318. fs := vfs.NewOsFs("id", os.TempDir(), "")
  319. connection := &Connection{
  320. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  321. request: nil,
  322. }
  323. dirsToRemove := []objectMapping{}
  324. orderedDirs := connection.orderDirsToRemove(fs, dirsToRemove)
  325. assert.Equal(t, len(dirsToRemove), len(orderedDirs))
  326. dirsToRemove = []objectMapping{
  327. {
  328. fsPath: "dir1",
  329. virtualPath: "",
  330. },
  331. }
  332. orderedDirs = connection.orderDirsToRemove(fs, dirsToRemove)
  333. assert.Equal(t, len(dirsToRemove), len(orderedDirs))
  334. dirsToRemove = []objectMapping{
  335. {
  336. fsPath: "dir1",
  337. virtualPath: "",
  338. },
  339. {
  340. fsPath: "dir12",
  341. virtualPath: "",
  342. },
  343. {
  344. fsPath: filepath.Join("dir1", "a", "b"),
  345. virtualPath: "",
  346. },
  347. {
  348. fsPath: filepath.Join("dir1", "a"),
  349. virtualPath: "",
  350. },
  351. }
  352. orderedDirs = connection.orderDirsToRemove(fs, dirsToRemove)
  353. if assert.Equal(t, len(dirsToRemove), len(orderedDirs)) {
  354. assert.Equal(t, "dir12", orderedDirs[0].fsPath)
  355. assert.Equal(t, filepath.Join("dir1", "a", "b"), orderedDirs[1].fsPath)
  356. assert.Equal(t, filepath.Join("dir1", "a"), orderedDirs[2].fsPath)
  357. assert.Equal(t, "dir1", orderedDirs[3].fsPath)
  358. }
  359. }
  360. func TestUserInvalidParams(t *testing.T) {
  361. u := &dataprovider.User{
  362. Username: "username",
  363. HomeDir: "invalid",
  364. }
  365. c := &Configuration{
  366. Bindings: []Binding{
  367. {
  368. Port: 9000,
  369. },
  370. },
  371. }
  372. server := webDavServer{
  373. config: c,
  374. binding: c.Bindings[0],
  375. }
  376. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", u.Username), nil)
  377. assert.NoError(t, err)
  378. _, err = server.validateUser(u, req, dataprovider.LoginMethodPassword)
  379. if assert.Error(t, err) {
  380. assert.EqualError(t, err, fmt.Sprintf("cannot login user with invalid home dir: %#v", u.HomeDir))
  381. }
  382. req.TLS = &tls.ConnectionState{}
  383. writeLog(req, nil)
  384. }
  385. func TestRemoteAddress(t *testing.T) {
  386. req, err := http.NewRequest(http.MethodGet, "/username", nil)
  387. assert.NoError(t, err)
  388. assert.Empty(t, req.RemoteAddr)
  389. remoteAddr1 := "100.100.100.100"
  390. remoteAddr2 := "172.172.172.172"
  391. req.Header.Set("X-Forwarded-For", remoteAddr1)
  392. checkRemoteAddress(req)
  393. assert.Equal(t, remoteAddr1, req.RemoteAddr)
  394. req.RemoteAddr = ""
  395. req.Header.Set("X-Forwarded-For", fmt.Sprintf("%v, %v", remoteAddr2, remoteAddr1))
  396. checkRemoteAddress(req)
  397. assert.Equal(t, remoteAddr2, req.RemoteAddr)
  398. req.Header.Del("X-Forwarded-For")
  399. req.RemoteAddr = ""
  400. req.Header.Set("X-Real-IP", remoteAddr1)
  401. checkRemoteAddress(req)
  402. assert.Equal(t, remoteAddr1, req.RemoteAddr)
  403. req.RemoteAddr = ""
  404. oldValue := common.Config.ProxyProtocol
  405. common.Config.ProxyProtocol = 1
  406. checkRemoteAddress(req)
  407. assert.Empty(t, req.RemoteAddr)
  408. common.Config.ProxyProtocol = oldValue
  409. }
  410. func TestConnWithNilRequest(t *testing.T) {
  411. c := &Connection{}
  412. assert.Empty(t, c.GetClientVersion())
  413. assert.Empty(t, c.GetCommand())
  414. assert.Empty(t, c.GetRemoteAddress())
  415. }
  416. func TestResolvePathErrors(t *testing.T) {
  417. ctx := context.Background()
  418. user := dataprovider.User{
  419. HomeDir: "invalid",
  420. }
  421. user.Permissions = make(map[string][]string)
  422. user.Permissions["/"] = []string{dataprovider.PermAny}
  423. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  424. connection := &Connection{
  425. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  426. }
  427. err := connection.Mkdir(ctx, "", os.ModePerm)
  428. if assert.Error(t, err) {
  429. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  430. }
  431. err = connection.Rename(ctx, "oldName", "newName")
  432. if assert.Error(t, err) {
  433. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  434. }
  435. _, err = connection.Stat(ctx, "name")
  436. if assert.Error(t, err) {
  437. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  438. }
  439. err = connection.RemoveAll(ctx, "")
  440. if assert.Error(t, err) {
  441. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  442. }
  443. _, err = connection.OpenFile(ctx, "", 0, os.ModePerm)
  444. if assert.Error(t, err) {
  445. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  446. }
  447. if runtime.GOOS != "windows" {
  448. user.HomeDir = filepath.Clean(os.TempDir())
  449. connection.User = user
  450. fs := vfs.NewOsFs("connID", connection.User.HomeDir, "")
  451. subDir := "sub"
  452. testTxtFile := "file.txt"
  453. err = os.MkdirAll(filepath.Join(os.TempDir(), subDir, subDir), os.ModePerm)
  454. assert.NoError(t, err)
  455. err = os.WriteFile(filepath.Join(os.TempDir(), subDir, subDir, testTxtFile), []byte("content"), os.ModePerm)
  456. assert.NoError(t, err)
  457. err = os.Chmod(filepath.Join(os.TempDir(), subDir, subDir), 0001)
  458. assert.NoError(t, err)
  459. err = os.WriteFile(filepath.Join(os.TempDir(), testTxtFile), []byte("test content"), os.ModePerm)
  460. assert.NoError(t, err)
  461. err = connection.Rename(ctx, testTxtFile, path.Join(subDir, subDir, testTxtFile))
  462. if assert.Error(t, err) {
  463. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  464. }
  465. _, err = connection.putFile(fs, filepath.Join(connection.User.HomeDir, subDir, subDir, testTxtFile),
  466. path.Join(subDir, subDir, testTxtFile))
  467. if assert.Error(t, err) {
  468. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  469. }
  470. err = os.Chmod(filepath.Join(os.TempDir(), subDir, subDir), os.ModePerm)
  471. assert.NoError(t, err)
  472. err = os.RemoveAll(filepath.Join(os.TempDir(), subDir))
  473. assert.NoError(t, err)
  474. err = os.Remove(filepath.Join(os.TempDir(), testTxtFile))
  475. assert.NoError(t, err)
  476. }
  477. }
  478. func TestFileAccessErrors(t *testing.T) {
  479. ctx := context.Background()
  480. user := dataprovider.User{
  481. HomeDir: filepath.Clean(os.TempDir()),
  482. }
  483. user.Permissions = make(map[string][]string)
  484. user.Permissions["/"] = []string{dataprovider.PermAny}
  485. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  486. connection := &Connection{
  487. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  488. }
  489. missingPath := "missing path"
  490. fsMissingPath := filepath.Join(user.HomeDir, missingPath)
  491. err := connection.RemoveAll(ctx, missingPath)
  492. if assert.Error(t, err) {
  493. assert.EqualError(t, err, os.ErrNotExist.Error())
  494. }
  495. _, err = connection.getFile(fs, fsMissingPath, missingPath)
  496. if assert.Error(t, err) {
  497. assert.EqualError(t, err, os.ErrNotExist.Error())
  498. }
  499. _, err = connection.getFile(fs, fsMissingPath, missingPath)
  500. if assert.Error(t, err) {
  501. assert.EqualError(t, err, os.ErrNotExist.Error())
  502. }
  503. p := filepath.Join(user.HomeDir, "adir", missingPath)
  504. _, err = connection.handleUploadToNewFile(fs, p, p, path.Join("adir", missingPath))
  505. if assert.Error(t, err) {
  506. assert.EqualError(t, err, os.ErrNotExist.Error())
  507. }
  508. _, err = connection.handleUploadToExistingFile(fs, p, p, 0, path.Join("adir", missingPath))
  509. if assert.Error(t, err) {
  510. assert.EqualError(t, err, os.ErrNotExist.Error())
  511. }
  512. fs = newMockOsFs(nil, false, fs.ConnectionID(), user.HomeDir, nil)
  513. _, err = connection.handleUploadToExistingFile(fs, p, p, 0, path.Join("adir", missingPath))
  514. if assert.Error(t, err) {
  515. assert.EqualError(t, err, os.ErrNotExist.Error())
  516. }
  517. f, err := os.CreateTemp("", "temp")
  518. assert.NoError(t, err)
  519. err = f.Close()
  520. assert.NoError(t, err)
  521. davFile, err := connection.handleUploadToExistingFile(fs, f.Name(), f.Name(), 123, f.Name())
  522. if assert.NoError(t, err) {
  523. transfer := davFile.(*webDavFile)
  524. transfers := connection.GetTransfers()
  525. if assert.Equal(t, 1, len(transfers)) {
  526. assert.Equal(t, transfers[0].ID, transfer.GetID())
  527. assert.Equal(t, int64(123), transfer.InitialSize)
  528. err = transfer.Close()
  529. assert.NoError(t, err)
  530. assert.Equal(t, 0, len(connection.GetTransfers()))
  531. }
  532. }
  533. err = os.Remove(f.Name())
  534. assert.NoError(t, err)
  535. }
  536. func TestRemoveDirTree(t *testing.T) {
  537. user := dataprovider.User{
  538. HomeDir: filepath.Clean(os.TempDir()),
  539. }
  540. user.Permissions = make(map[string][]string)
  541. user.Permissions["/"] = []string{dataprovider.PermAny}
  542. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  543. connection := &Connection{
  544. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  545. }
  546. vpath := path.Join("adir", "missing")
  547. p := filepath.Join(user.HomeDir, "adir", "missing")
  548. err := connection.removeDirTree(fs, p, vpath)
  549. if assert.Error(t, err) {
  550. assert.True(t, os.IsNotExist(err))
  551. }
  552. fs = newMockOsFs(nil, false, "mockID", user.HomeDir, nil)
  553. err = connection.removeDirTree(fs, p, vpath)
  554. if assert.Error(t, err) {
  555. assert.True(t, os.IsNotExist(err), "unexpected error: %v", err)
  556. }
  557. errFake := errors.New("fake err")
  558. fs = newMockOsFs(errFake, false, "mockID", user.HomeDir, nil)
  559. err = connection.removeDirTree(fs, p, vpath)
  560. if assert.Error(t, err) {
  561. assert.EqualError(t, err, errFake.Error())
  562. }
  563. fs = newMockOsFs(errWalkDir, true, "mockID", user.HomeDir, nil)
  564. err = connection.removeDirTree(fs, p, vpath)
  565. if assert.Error(t, err) {
  566. assert.True(t, os.IsPermission(err), "unexpected error: %v", err)
  567. }
  568. fs = newMockOsFs(errWalkFile, false, "mockID", user.HomeDir, nil)
  569. err = connection.removeDirTree(fs, p, vpath)
  570. if assert.Error(t, err) {
  571. assert.EqualError(t, err, errWalkFile.Error())
  572. }
  573. connection.User.Permissions["/"] = []string{dataprovider.PermListItems}
  574. fs = newMockOsFs(nil, false, "mockID", user.HomeDir, nil)
  575. err = connection.removeDirTree(fs, p, vpath)
  576. if assert.Error(t, err) {
  577. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  578. }
  579. }
  580. func TestContentType(t *testing.T) {
  581. user := dataprovider.User{
  582. HomeDir: filepath.Clean(os.TempDir()),
  583. }
  584. user.Permissions = make(map[string][]string)
  585. user.Permissions["/"] = []string{dataprovider.PermAny}
  586. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  587. connection := &Connection{
  588. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  589. }
  590. testFilePath := filepath.Join(user.HomeDir, testFile)
  591. ctx := context.Background()
  592. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  593. common.TransferDownload, 0, 0, 0, false, fs)
  594. fs = newMockOsFs(nil, false, fs.ConnectionID(), user.GetHomeDir(), nil)
  595. err := os.WriteFile(testFilePath, []byte(""), os.ModePerm)
  596. assert.NoError(t, err)
  597. davFile := newWebDavFile(baseTransfer, nil, nil)
  598. davFile.Fs = fs
  599. fi, err := davFile.Stat()
  600. if assert.NoError(t, err) {
  601. ctype, err := fi.(*webDavFileInfo).ContentType(ctx)
  602. assert.NoError(t, err)
  603. assert.Equal(t, "application/custom-mime", ctype)
  604. }
  605. _, err = davFile.Readdir(-1)
  606. assert.Error(t, err)
  607. err = davFile.Close()
  608. assert.NoError(t, err)
  609. davFile = newWebDavFile(baseTransfer, nil, nil)
  610. davFile.Fs = vfs.NewOsFs("id", user.HomeDir, "")
  611. fi, err = davFile.Stat()
  612. if assert.NoError(t, err) {
  613. ctype, err := fi.(*webDavFileInfo).ContentType(ctx)
  614. assert.NoError(t, err)
  615. assert.Equal(t, "text/plain; charset=utf-8", ctype)
  616. }
  617. err = davFile.Close()
  618. assert.NoError(t, err)
  619. fi.(*webDavFileInfo).fsPath = "missing"
  620. _, err = fi.(*webDavFileInfo).ContentType(ctx)
  621. assert.EqualError(t, err, webdav.ErrNotImplemented.Error())
  622. err = os.Remove(testFilePath)
  623. assert.NoError(t, err)
  624. }
  625. func TestTransferReadWriteErrors(t *testing.T) {
  626. user := dataprovider.User{
  627. HomeDir: filepath.Clean(os.TempDir()),
  628. }
  629. user.Permissions = make(map[string][]string)
  630. user.Permissions["/"] = []string{dataprovider.PermAny}
  631. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  632. connection := &Connection{
  633. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  634. }
  635. testFilePath := filepath.Join(user.HomeDir, testFile)
  636. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  637. common.TransferUpload, 0, 0, 0, false, fs)
  638. davFile := newWebDavFile(baseTransfer, nil, nil)
  639. p := make([]byte, 1)
  640. _, err := davFile.Read(p)
  641. assert.EqualError(t, err, common.ErrOpUnsupported.Error())
  642. r, w, err := pipeat.Pipe()
  643. assert.NoError(t, err)
  644. davFile = newWebDavFile(baseTransfer, nil, r)
  645. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  646. davFile = newWebDavFile(baseTransfer, vfs.NewPipeWriter(w), nil)
  647. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  648. err = r.Close()
  649. assert.NoError(t, err)
  650. err = w.Close()
  651. assert.NoError(t, err)
  652. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  653. common.TransferDownload, 0, 0, 0, false, fs)
  654. davFile = newWebDavFile(baseTransfer, nil, nil)
  655. _, err = davFile.Read(p)
  656. assert.True(t, os.IsNotExist(err))
  657. _, err = davFile.Stat()
  658. assert.True(t, os.IsNotExist(err))
  659. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  660. common.TransferDownload, 0, 0, 0, false, fs)
  661. err = os.WriteFile(testFilePath, []byte(""), os.ModePerm)
  662. assert.NoError(t, err)
  663. f, err := os.Open(testFilePath)
  664. if assert.NoError(t, err) {
  665. err = f.Close()
  666. assert.NoError(t, err)
  667. }
  668. davFile = newWebDavFile(baseTransfer, nil, nil)
  669. davFile.reader = f
  670. err = davFile.Close()
  671. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  672. err = davFile.Close()
  673. assert.EqualError(t, err, common.ErrTransferClosed.Error())
  674. _, err = davFile.Read(p)
  675. assert.Error(t, err)
  676. info, err := davFile.Stat()
  677. if assert.NoError(t, err) {
  678. assert.Equal(t, int64(0), info.Size())
  679. }
  680. r, w, err = pipeat.Pipe()
  681. assert.NoError(t, err)
  682. mockFs := newMockOsFs(nil, false, fs.ConnectionID(), user.HomeDir, r)
  683. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  684. common.TransferDownload, 0, 0, 0, false, mockFs)
  685. davFile = newWebDavFile(baseTransfer, nil, nil)
  686. writeContent := []byte("content\r\n")
  687. go func() {
  688. n, err := w.Write(writeContent)
  689. assert.NoError(t, err)
  690. assert.Equal(t, len(writeContent), n)
  691. err = w.Close()
  692. assert.NoError(t, err)
  693. }()
  694. p = make([]byte, 64)
  695. n, err := davFile.Read(p)
  696. assert.EqualError(t, err, io.EOF.Error())
  697. assert.Equal(t, len(writeContent), n)
  698. err = davFile.Close()
  699. assert.NoError(t, err)
  700. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  701. common.TransferDownload, 0, 0, 0, false, fs)
  702. davFile = newWebDavFile(baseTransfer, nil, nil)
  703. davFile.writer = f
  704. err = davFile.Close()
  705. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  706. err = os.Remove(testFilePath)
  707. assert.NoError(t, err)
  708. }
  709. func TestTransferSeek(t *testing.T) {
  710. user := dataprovider.User{
  711. HomeDir: filepath.Clean(os.TempDir()),
  712. }
  713. user.Permissions = make(map[string][]string)
  714. user.Permissions["/"] = []string{dataprovider.PermAny}
  715. fs := vfs.NewOsFs("connID", user.HomeDir, "")
  716. connection := &Connection{
  717. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user),
  718. }
  719. testFilePath := filepath.Join(user.HomeDir, testFile)
  720. testFileContents := []byte("content")
  721. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  722. common.TransferUpload, 0, 0, 0, false, fs)
  723. davFile := newWebDavFile(baseTransfer, nil, nil)
  724. _, err := davFile.Seek(0, io.SeekStart)
  725. assert.EqualError(t, err, common.ErrOpUnsupported.Error())
  726. err = davFile.Close()
  727. assert.NoError(t, err)
  728. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  729. common.TransferDownload, 0, 0, 0, false, fs)
  730. davFile = newWebDavFile(baseTransfer, nil, nil)
  731. _, err = davFile.Seek(0, io.SeekCurrent)
  732. assert.True(t, os.IsNotExist(err))
  733. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  734. err = os.WriteFile(testFilePath, testFileContents, os.ModePerm)
  735. assert.NoError(t, err)
  736. f, err := os.Open(testFilePath)
  737. if assert.NoError(t, err) {
  738. err = f.Close()
  739. assert.NoError(t, err)
  740. }
  741. baseTransfer = common.NewBaseTransfer(f, connection.BaseConnection, nil, testFilePath, testFile,
  742. common.TransferDownload, 0, 0, 0, false, fs)
  743. davFile = newWebDavFile(baseTransfer, nil, nil)
  744. _, err = davFile.Seek(0, io.SeekStart)
  745. assert.Error(t, err)
  746. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  747. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  748. common.TransferDownload, 0, 0, 0, false, fs)
  749. davFile = newWebDavFile(baseTransfer, nil, nil)
  750. res, err := davFile.Seek(0, io.SeekStart)
  751. assert.NoError(t, err)
  752. assert.Equal(t, int64(0), res)
  753. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  754. davFile = newWebDavFile(baseTransfer, nil, nil)
  755. res, err = davFile.Seek(0, io.SeekEnd)
  756. assert.NoError(t, err)
  757. assert.Equal(t, int64(len(testFileContents)), res)
  758. err = davFile.updateStatInfo()
  759. assert.Nil(t, err)
  760. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath+"1", testFile,
  761. common.TransferDownload, 0, 0, 0, false, fs)
  762. davFile = newWebDavFile(baseTransfer, nil, nil)
  763. _, err = davFile.Seek(0, io.SeekEnd)
  764. assert.True(t, os.IsNotExist(err))
  765. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  766. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  767. common.TransferDownload, 0, 0, 0, false, fs)
  768. davFile = newWebDavFile(baseTransfer, nil, nil)
  769. davFile.reader = f
  770. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  771. res, err = davFile.Seek(2, io.SeekStart)
  772. assert.NoError(t, err)
  773. assert.Equal(t, int64(2), res)
  774. davFile = newWebDavFile(baseTransfer, nil, nil)
  775. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  776. res, err = davFile.Seek(2, io.SeekEnd)
  777. assert.NoError(t, err)
  778. assert.Equal(t, int64(5), res)
  779. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath+"1", testFile,
  780. common.TransferDownload, 0, 0, 0, false, fs)
  781. davFile = newWebDavFile(baseTransfer, nil, nil)
  782. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  783. res, err = davFile.Seek(2, io.SeekEnd)
  784. assert.True(t, os.IsNotExist(err))
  785. assert.Equal(t, int64(0), res)
  786. assert.Len(t, common.Connections.GetStats(), 0)
  787. err = os.Remove(testFilePath)
  788. assert.NoError(t, err)
  789. }
  790. func TestBasicUsersCache(t *testing.T) {
  791. username := "webdav_internal_test"
  792. password := "pwd"
  793. u := dataprovider.User{
  794. Username: username,
  795. Password: password,
  796. HomeDir: filepath.Join(os.TempDir(), username),
  797. Status: 1,
  798. ExpirationDate: 0,
  799. }
  800. u.Permissions = make(map[string][]string)
  801. u.Permissions["/"] = []string{dataprovider.PermAny}
  802. err := dataprovider.AddUser(&u)
  803. assert.NoError(t, err)
  804. user, err := dataprovider.UserExists(u.Username)
  805. assert.NoError(t, err)
  806. c := &Configuration{
  807. Bindings: []Binding{
  808. {
  809. Port: 9000,
  810. },
  811. },
  812. Cache: Cache{
  813. Users: UsersCacheConfig{
  814. MaxSize: 50,
  815. ExpirationTime: 1,
  816. },
  817. },
  818. }
  819. server := webDavServer{
  820. config: c,
  821. binding: c.Bindings[0],
  822. }
  823. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user.Username), nil)
  824. assert.NoError(t, err)
  825. ipAddr := "127.0.0.1"
  826. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  827. assert.Error(t, err)
  828. now := time.Now()
  829. req.SetBasicAuth(username, password)
  830. _, isCached, _, loginMethod, err := server.authenticate(req, ipAddr)
  831. assert.NoError(t, err)
  832. assert.False(t, isCached)
  833. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  834. // now the user should be cached
  835. var cachedUser *dataprovider.CachedUser
  836. result, ok := dataprovider.GetCachedWebDAVUser(username)
  837. if assert.True(t, ok) {
  838. cachedUser = result.(*dataprovider.CachedUser)
  839. assert.False(t, cachedUser.IsExpired())
  840. assert.True(t, cachedUser.Expiration.After(now.Add(time.Duration(c.Cache.Users.ExpirationTime)*time.Minute)))
  841. // authenticate must return the cached user now
  842. authUser, isCached, _, _, err := server.authenticate(req, ipAddr)
  843. assert.NoError(t, err)
  844. assert.True(t, isCached)
  845. assert.Equal(t, cachedUser.User, authUser)
  846. }
  847. // a wrong password must fail
  848. req.SetBasicAuth(username, "wrong")
  849. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  850. assert.EqualError(t, err, dataprovider.ErrInvalidCredentials.Error())
  851. req.SetBasicAuth(username, password)
  852. // force cached user expiration
  853. cachedUser.Expiration = now
  854. dataprovider.CacheWebDAVUser(cachedUser, c.Cache.Users.MaxSize)
  855. result, ok = dataprovider.GetCachedWebDAVUser(username)
  856. if assert.True(t, ok) {
  857. cachedUser = result.(*dataprovider.CachedUser)
  858. assert.True(t, cachedUser.IsExpired())
  859. }
  860. // now authenticate should get the user from the data provider and update the cache
  861. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  862. assert.NoError(t, err)
  863. assert.False(t, isCached)
  864. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  865. result, ok = dataprovider.GetCachedWebDAVUser(username)
  866. if assert.True(t, ok) {
  867. cachedUser = result.(*dataprovider.CachedUser)
  868. assert.False(t, cachedUser.IsExpired())
  869. }
  870. // cache is invalidated after a user modification
  871. err = dataprovider.UpdateUser(&user)
  872. assert.NoError(t, err)
  873. _, ok = dataprovider.GetCachedWebDAVUser(username)
  874. assert.False(t, ok)
  875. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  876. assert.NoError(t, err)
  877. assert.False(t, isCached)
  878. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  879. _, ok = dataprovider.GetCachedWebDAVUser(username)
  880. assert.True(t, ok)
  881. // cache is invalidated after user deletion
  882. err = dataprovider.DeleteUser(user.Username)
  883. assert.NoError(t, err)
  884. _, ok = dataprovider.GetCachedWebDAVUser(username)
  885. assert.False(t, ok)
  886. err = os.RemoveAll(u.GetHomeDir())
  887. assert.NoError(t, err)
  888. }
  889. func TestCachedUserWithFolders(t *testing.T) {
  890. username := "webdav_internal_folder_test"
  891. password := "dav_pwd"
  892. folderName := "test_folder"
  893. u := dataprovider.User{
  894. Username: username,
  895. Password: password,
  896. HomeDir: filepath.Join(os.TempDir(), username),
  897. Status: 1,
  898. ExpirationDate: 0,
  899. }
  900. u.Permissions = make(map[string][]string)
  901. u.Permissions["/"] = []string{dataprovider.PermAny}
  902. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  903. BaseVirtualFolder: vfs.BaseVirtualFolder{
  904. Name: folderName,
  905. MappedPath: filepath.Join(os.TempDir(), folderName),
  906. },
  907. VirtualPath: "/vpath",
  908. })
  909. err := dataprovider.AddUser(&u)
  910. assert.NoError(t, err)
  911. user, err := dataprovider.UserExists(u.Username)
  912. assert.NoError(t, err)
  913. c := &Configuration{
  914. Bindings: []Binding{
  915. {
  916. Port: 9000,
  917. },
  918. },
  919. Cache: Cache{
  920. Users: UsersCacheConfig{
  921. MaxSize: 50,
  922. ExpirationTime: 1,
  923. },
  924. },
  925. }
  926. server := webDavServer{
  927. config: c,
  928. binding: c.Bindings[0],
  929. }
  930. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user.Username), nil)
  931. assert.NoError(t, err)
  932. ipAddr := "127.0.0.1"
  933. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  934. assert.Error(t, err)
  935. now := time.Now()
  936. req.SetBasicAuth(username, password)
  937. _, isCached, _, loginMethod, err := server.authenticate(req, ipAddr)
  938. assert.NoError(t, err)
  939. assert.False(t, isCached)
  940. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  941. // now the user should be cached
  942. var cachedUser *dataprovider.CachedUser
  943. result, ok := dataprovider.GetCachedWebDAVUser(username)
  944. if assert.True(t, ok) {
  945. cachedUser = result.(*dataprovider.CachedUser)
  946. assert.False(t, cachedUser.IsExpired())
  947. assert.True(t, cachedUser.Expiration.After(now.Add(time.Duration(c.Cache.Users.ExpirationTime)*time.Minute)))
  948. // authenticate must return the cached user now
  949. authUser, isCached, _, _, err := server.authenticate(req, ipAddr)
  950. assert.NoError(t, err)
  951. assert.True(t, isCached)
  952. assert.Equal(t, cachedUser.User, authUser)
  953. }
  954. folder, err := dataprovider.GetFolderByName(folderName)
  955. assert.NoError(t, err)
  956. // updating a used folder should invalidate the cache
  957. err = dataprovider.UpdateFolder(&folder, folder.Users)
  958. assert.NoError(t, err)
  959. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  960. assert.NoError(t, err)
  961. assert.False(t, isCached)
  962. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  963. result, ok = dataprovider.GetCachedWebDAVUser(username)
  964. if assert.True(t, ok) {
  965. cachedUser = result.(*dataprovider.CachedUser)
  966. assert.False(t, cachedUser.IsExpired())
  967. }
  968. err = dataprovider.DeleteFolder(folderName)
  969. assert.NoError(t, err)
  970. // removing a used folder should invalidate the cache
  971. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  972. assert.NoError(t, err)
  973. assert.False(t, isCached)
  974. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  975. result, ok = dataprovider.GetCachedWebDAVUser(username)
  976. if assert.True(t, ok) {
  977. cachedUser = result.(*dataprovider.CachedUser)
  978. assert.False(t, cachedUser.IsExpired())
  979. }
  980. err = dataprovider.DeleteUser(user.Username)
  981. assert.NoError(t, err)
  982. _, ok = dataprovider.GetCachedWebDAVUser(username)
  983. assert.False(t, ok)
  984. err = os.RemoveAll(u.GetHomeDir())
  985. assert.NoError(t, err)
  986. err = os.RemoveAll(folder.MappedPath)
  987. assert.NoError(t, err)
  988. }
  989. func TestUsersCacheSizeAndExpiration(t *testing.T) {
  990. username := "webdav_internal_test"
  991. password := "pwd"
  992. u := dataprovider.User{
  993. HomeDir: filepath.Join(os.TempDir(), username),
  994. Status: 1,
  995. ExpirationDate: 0,
  996. }
  997. u.Username = username + "1"
  998. u.Password = password + "1"
  999. u.Permissions = make(map[string][]string)
  1000. u.Permissions["/"] = []string{dataprovider.PermAny}
  1001. err := dataprovider.AddUser(&u)
  1002. assert.NoError(t, err)
  1003. user1, err := dataprovider.UserExists(u.Username)
  1004. assert.NoError(t, err)
  1005. u.Username = username + "2"
  1006. u.Password = password + "2"
  1007. err = dataprovider.AddUser(&u)
  1008. assert.NoError(t, err)
  1009. user2, err := dataprovider.UserExists(u.Username)
  1010. assert.NoError(t, err)
  1011. u.Username = username + "3"
  1012. u.Password = password + "3"
  1013. err = dataprovider.AddUser(&u)
  1014. assert.NoError(t, err)
  1015. user3, err := dataprovider.UserExists(u.Username)
  1016. assert.NoError(t, err)
  1017. u.Username = username + "4"
  1018. u.Password = password + "4"
  1019. err = dataprovider.AddUser(&u)
  1020. assert.NoError(t, err)
  1021. user4, err := dataprovider.UserExists(u.Username)
  1022. assert.NoError(t, err)
  1023. c := &Configuration{
  1024. Bindings: []Binding{
  1025. {
  1026. Port: 9000,
  1027. },
  1028. },
  1029. Cache: Cache{
  1030. Users: UsersCacheConfig{
  1031. MaxSize: 3,
  1032. ExpirationTime: 1,
  1033. },
  1034. },
  1035. }
  1036. server := webDavServer{
  1037. config: c,
  1038. binding: c.Bindings[0],
  1039. }
  1040. ipAddr := "127.0.1.1"
  1041. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1042. assert.NoError(t, err)
  1043. req.SetBasicAuth(user1.Username, password+"1")
  1044. _, isCached, _, loginMehod, err := server.authenticate(req, ipAddr)
  1045. assert.NoError(t, err)
  1046. assert.False(t, isCached)
  1047. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1048. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  1049. assert.NoError(t, err)
  1050. req.SetBasicAuth(user2.Username, password+"2")
  1051. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1052. assert.NoError(t, err)
  1053. assert.False(t, isCached)
  1054. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1055. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  1056. assert.NoError(t, err)
  1057. req.SetBasicAuth(user3.Username, password+"3")
  1058. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1059. assert.NoError(t, err)
  1060. assert.False(t, isCached)
  1061. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1062. // the first 3 users are now cached
  1063. _, ok := dataprovider.GetCachedWebDAVUser(user1.Username)
  1064. assert.True(t, ok)
  1065. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1066. assert.True(t, ok)
  1067. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1068. assert.True(t, ok)
  1069. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  1070. assert.NoError(t, err)
  1071. req.SetBasicAuth(user4.Username, password+"4")
  1072. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1073. assert.NoError(t, err)
  1074. assert.False(t, isCached)
  1075. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1076. // user1, the first cached, should be removed now
  1077. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1078. assert.False(t, ok)
  1079. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1080. assert.True(t, ok)
  1081. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1082. assert.True(t, ok)
  1083. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1084. assert.True(t, ok)
  1085. // user1 logins, user2 should be removed
  1086. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1087. assert.NoError(t, err)
  1088. req.SetBasicAuth(user1.Username, password+"1")
  1089. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1090. assert.NoError(t, err)
  1091. assert.False(t, isCached)
  1092. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1093. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1094. assert.False(t, ok)
  1095. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1096. assert.True(t, ok)
  1097. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1098. assert.True(t, ok)
  1099. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1100. assert.True(t, ok)
  1101. // user2 logins, user3 should be removed
  1102. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  1103. assert.NoError(t, err)
  1104. req.SetBasicAuth(user2.Username, password+"2")
  1105. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1106. assert.NoError(t, err)
  1107. assert.False(t, isCached)
  1108. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1109. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1110. assert.False(t, ok)
  1111. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1112. assert.True(t, ok)
  1113. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1114. assert.True(t, ok)
  1115. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1116. assert.True(t, ok)
  1117. // user3 logins, user4 should be removed
  1118. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  1119. assert.NoError(t, err)
  1120. req.SetBasicAuth(user3.Username, password+"3")
  1121. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1122. assert.NoError(t, err)
  1123. assert.False(t, isCached)
  1124. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1125. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1126. assert.False(t, ok)
  1127. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1128. assert.True(t, ok)
  1129. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1130. assert.True(t, ok)
  1131. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1132. assert.True(t, ok)
  1133. // now remove user1 after an update
  1134. err = dataprovider.UpdateUser(&user1)
  1135. assert.NoError(t, err)
  1136. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1137. assert.False(t, ok)
  1138. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  1139. assert.NoError(t, err)
  1140. req.SetBasicAuth(user4.Username, password+"4")
  1141. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1142. assert.NoError(t, err)
  1143. assert.False(t, isCached)
  1144. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1145. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1146. assert.NoError(t, err)
  1147. req.SetBasicAuth(user1.Username, password+"1")
  1148. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1149. assert.NoError(t, err)
  1150. assert.False(t, isCached)
  1151. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1152. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1153. assert.False(t, ok)
  1154. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1155. assert.True(t, ok)
  1156. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1157. assert.True(t, ok)
  1158. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1159. assert.True(t, ok)
  1160. err = dataprovider.DeleteUser(user1.Username)
  1161. assert.NoError(t, err)
  1162. err = dataprovider.DeleteUser(user2.Username)
  1163. assert.NoError(t, err)
  1164. err = dataprovider.DeleteUser(user3.Username)
  1165. assert.NoError(t, err)
  1166. err = dataprovider.DeleteUser(user4.Username)
  1167. assert.NoError(t, err)
  1168. err = os.RemoveAll(u.GetHomeDir())
  1169. assert.NoError(t, err)
  1170. }
  1171. func TestUserCacheIsolation(t *testing.T) {
  1172. username := "webdav_internal_cache_test"
  1173. password := "dav_pwd"
  1174. u := dataprovider.User{
  1175. Username: username,
  1176. Password: password,
  1177. HomeDir: filepath.Join(os.TempDir(), username),
  1178. Status: 1,
  1179. ExpirationDate: 0,
  1180. }
  1181. u.Permissions = make(map[string][]string)
  1182. u.Permissions["/"] = []string{dataprovider.PermAny}
  1183. err := dataprovider.AddUser(&u)
  1184. assert.NoError(t, err)
  1185. user, err := dataprovider.UserExists(u.Username)
  1186. assert.NoError(t, err)
  1187. cachedUser := &dataprovider.CachedUser{
  1188. User: user,
  1189. Expiration: time.Now().Add(24 * time.Hour),
  1190. Password: password,
  1191. LockSystem: webdav.NewMemLS(),
  1192. }
  1193. cachedUser.User.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("test secret")
  1194. err = cachedUser.User.FsConfig.S3Config.AccessSecret.Encrypt()
  1195. assert.NoError(t, err)
  1196. dataprovider.CacheWebDAVUser(cachedUser, 10)
  1197. result, ok := dataprovider.GetCachedWebDAVUser(username)
  1198. if assert.True(t, ok) {
  1199. cachedUser := result.(*dataprovider.CachedUser).User
  1200. _, err = cachedUser.GetFilesystem("")
  1201. assert.NoError(t, err)
  1202. // the filesystem is now cached
  1203. }
  1204. result, ok = dataprovider.GetCachedWebDAVUser(username)
  1205. if assert.True(t, ok) {
  1206. cachedUser := result.(*dataprovider.CachedUser).User
  1207. assert.True(t, cachedUser.FsConfig.S3Config.AccessSecret.IsEncrypted())
  1208. err = cachedUser.FsConfig.S3Config.AccessSecret.Decrypt()
  1209. assert.NoError(t, err)
  1210. cachedUser.FsConfig.Provider = vfs.S3FilesystemProvider
  1211. _, err = cachedUser.GetFilesystem("")
  1212. assert.Error(t, err, "we don't have to get the previously cached filesystem!")
  1213. }
  1214. result, ok = dataprovider.GetCachedWebDAVUser(username)
  1215. if assert.True(t, ok) {
  1216. cachedUser := result.(*dataprovider.CachedUser).User
  1217. assert.Equal(t, vfs.LocalFilesystemProvider, cachedUser.FsConfig.Provider)
  1218. assert.False(t, cachedUser.FsConfig.S3Config.AccessSecret.IsEncrypted())
  1219. }
  1220. err = dataprovider.DeleteUser(username)
  1221. assert.NoError(t, err)
  1222. _, ok = dataprovider.GetCachedWebDAVUser(username)
  1223. assert.False(t, ok)
  1224. }
  1225. func TestRecoverer(t *testing.T) {
  1226. c := &Configuration{
  1227. Bindings: []Binding{
  1228. {
  1229. Port: 9000,
  1230. },
  1231. },
  1232. }
  1233. server := webDavServer{
  1234. config: c,
  1235. binding: c.Bindings[0],
  1236. }
  1237. rr := httptest.NewRecorder()
  1238. server.ServeHTTP(rr, nil)
  1239. assert.Equal(t, http.StatusInternalServerError, rr.Code)
  1240. }
  1241. func TestMimeCache(t *testing.T) {
  1242. cache := mimeCache{
  1243. maxSize: 0,
  1244. mimeTypes: make(map[string]string),
  1245. }
  1246. cache.addMimeToCache(".zip", "application/zip")
  1247. mtype := cache.getMimeFromCache(".zip")
  1248. assert.Equal(t, "", mtype)
  1249. cache.maxSize = 1
  1250. cache.addMimeToCache(".zip", "application/zip")
  1251. mtype = cache.getMimeFromCache(".zip")
  1252. assert.Equal(t, "application/zip", mtype)
  1253. cache.addMimeToCache(".jpg", "image/jpeg")
  1254. mtype = cache.getMimeFromCache(".jpg")
  1255. assert.Equal(t, "", mtype)
  1256. }
  1257. func TestVerifyTLSConnection(t *testing.T) {
  1258. oldCertMgr := certMgr
  1259. caCrlPath := filepath.Join(os.TempDir(), "testcrl.crt")
  1260. certPath := filepath.Join(os.TempDir(), "test.crt")
  1261. keyPath := filepath.Join(os.TempDir(), "test.key")
  1262. err := os.WriteFile(caCrlPath, []byte(caCRL), os.ModePerm)
  1263. assert.NoError(t, err)
  1264. err = os.WriteFile(certPath, []byte(webDavCert), os.ModePerm)
  1265. assert.NoError(t, err)
  1266. err = os.WriteFile(keyPath, []byte(webDavKey), os.ModePerm)
  1267. assert.NoError(t, err)
  1268. certMgr, err = common.NewCertManager(certPath, keyPath, "", "webdav_test")
  1269. assert.NoError(t, err)
  1270. certMgr.SetCARevocationLists([]string{caCrlPath})
  1271. err = certMgr.LoadCRLs()
  1272. assert.NoError(t, err)
  1273. crt, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  1274. assert.NoError(t, err)
  1275. x509crt, err := x509.ParseCertificate(crt.Certificate[0])
  1276. assert.NoError(t, err)
  1277. server := webDavServer{}
  1278. state := tls.ConnectionState{
  1279. PeerCertificates: []*x509.Certificate{x509crt},
  1280. }
  1281. err = server.verifyTLSConnection(state)
  1282. assert.Error(t, err) // no verified certification chain
  1283. crt, err = tls.X509KeyPair([]byte(caCRT), []byte(caKey))
  1284. assert.NoError(t, err)
  1285. x509CAcrt, err := x509.ParseCertificate(crt.Certificate[0])
  1286. assert.NoError(t, err)
  1287. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crt, x509CAcrt})
  1288. err = server.verifyTLSConnection(state)
  1289. assert.NoError(t, err)
  1290. crt, err = tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  1291. assert.NoError(t, err)
  1292. x509crtRevoked, err := x509.ParseCertificate(crt.Certificate[0])
  1293. assert.NoError(t, err)
  1294. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crtRevoked, x509CAcrt})
  1295. state.PeerCertificates = []*x509.Certificate{x509crtRevoked}
  1296. err = server.verifyTLSConnection(state)
  1297. assert.EqualError(t, err, common.ErrCrtRevoked.Error())
  1298. err = os.Remove(caCrlPath)
  1299. assert.NoError(t, err)
  1300. err = os.Remove(certPath)
  1301. assert.NoError(t, err)
  1302. err = os.Remove(keyPath)
  1303. assert.NoError(t, err)
  1304. certMgr = oldCertMgr
  1305. }
  1306. func TestMisc(t *testing.T) {
  1307. oldCertMgr := certMgr
  1308. certMgr = nil
  1309. err := ReloadCertificateMgr()
  1310. assert.Nil(t, err)
  1311. val := getConfigPath("", ".")
  1312. assert.Empty(t, val)
  1313. certMgr = oldCertMgr
  1314. }