internal_test.go 52 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459
  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 resuming uploads 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. dataprovider.InitializeWebDAVUserCache(c.Cache.Users.MaxSize)
  820. server := webDavServer{
  821. config: c,
  822. binding: c.Bindings[0],
  823. }
  824. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user.Username), nil)
  825. assert.NoError(t, err)
  826. ipAddr := "127.0.0.1"
  827. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  828. assert.Error(t, err)
  829. now := time.Now()
  830. req.SetBasicAuth(username, password)
  831. _, isCached, _, loginMethod, err := server.authenticate(req, ipAddr)
  832. assert.NoError(t, err)
  833. assert.False(t, isCached)
  834. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  835. // now the user should be cached
  836. cachedUser, ok := dataprovider.GetCachedWebDAVUser(username)
  837. if assert.True(t, ok) {
  838. assert.False(t, cachedUser.IsExpired())
  839. assert.True(t, cachedUser.Expiration.After(now.Add(time.Duration(c.Cache.Users.ExpirationTime)*time.Minute)))
  840. // authenticate must return the cached user now
  841. authUser, isCached, _, _, err := server.authenticate(req, ipAddr)
  842. assert.NoError(t, err)
  843. assert.True(t, isCached)
  844. assert.Equal(t, cachedUser.User, authUser)
  845. }
  846. // a wrong password must fail
  847. req.SetBasicAuth(username, "wrong")
  848. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  849. assert.EqualError(t, err, dataprovider.ErrInvalidCredentials.Error())
  850. req.SetBasicAuth(username, password)
  851. // force cached user expiration
  852. cachedUser.Expiration = now
  853. dataprovider.CacheWebDAVUser(cachedUser)
  854. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  855. if assert.True(t, ok) {
  856. assert.True(t, cachedUser.IsExpired())
  857. }
  858. // now authenticate should get the user from the data provider and update the cache
  859. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  860. assert.NoError(t, err)
  861. assert.False(t, isCached)
  862. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  863. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  864. if assert.True(t, ok) {
  865. assert.False(t, cachedUser.IsExpired())
  866. }
  867. // cache is not invalidated after a user modification if the fs does not change
  868. err = dataprovider.UpdateUser(&user)
  869. assert.NoError(t, err)
  870. _, ok = dataprovider.GetCachedWebDAVUser(username)
  871. assert.True(t, ok)
  872. folderName := "testFolder"
  873. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  874. BaseVirtualFolder: vfs.BaseVirtualFolder{
  875. Name: folderName,
  876. MappedPath: filepath.Join(os.TempDir(), "mapped"),
  877. },
  878. VirtualPath: "/vdir",
  879. })
  880. err = dataprovider.UpdateUser(&user)
  881. assert.NoError(t, err)
  882. _, ok = dataprovider.GetCachedWebDAVUser(username)
  883. assert.False(t, ok)
  884. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  885. assert.NoError(t, err)
  886. assert.False(t, isCached)
  887. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  888. _, ok = dataprovider.GetCachedWebDAVUser(username)
  889. assert.True(t, ok)
  890. // cache is invalidated after user deletion
  891. err = dataprovider.DeleteUser(user.Username)
  892. assert.NoError(t, err)
  893. _, ok = dataprovider.GetCachedWebDAVUser(username)
  894. assert.False(t, ok)
  895. err = dataprovider.DeleteFolder(folderName)
  896. assert.NoError(t, err)
  897. err = os.RemoveAll(u.GetHomeDir())
  898. assert.NoError(t, err)
  899. }
  900. func TestCachedUserWithFolders(t *testing.T) {
  901. username := "webdav_internal_folder_test"
  902. password := "dav_pwd"
  903. folderName := "test_folder"
  904. u := dataprovider.User{
  905. Username: username,
  906. Password: password,
  907. HomeDir: filepath.Join(os.TempDir(), username),
  908. Status: 1,
  909. ExpirationDate: 0,
  910. }
  911. u.Permissions = make(map[string][]string)
  912. u.Permissions["/"] = []string{dataprovider.PermAny}
  913. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  914. BaseVirtualFolder: vfs.BaseVirtualFolder{
  915. Name: folderName,
  916. MappedPath: filepath.Join(os.TempDir(), folderName),
  917. },
  918. VirtualPath: "/vpath",
  919. })
  920. err := dataprovider.AddUser(&u)
  921. assert.NoError(t, err)
  922. user, err := dataprovider.UserExists(u.Username)
  923. assert.NoError(t, err)
  924. c := &Configuration{
  925. Bindings: []Binding{
  926. {
  927. Port: 9000,
  928. },
  929. },
  930. Cache: Cache{
  931. Users: UsersCacheConfig{
  932. MaxSize: 50,
  933. ExpirationTime: 1,
  934. },
  935. },
  936. }
  937. dataprovider.InitializeWebDAVUserCache(c.Cache.Users.MaxSize)
  938. server := webDavServer{
  939. config: c,
  940. binding: c.Bindings[0],
  941. }
  942. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user.Username), nil)
  943. assert.NoError(t, err)
  944. ipAddr := "127.0.0.1"
  945. _, _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  946. assert.Error(t, err)
  947. now := time.Now()
  948. req.SetBasicAuth(username, password)
  949. _, isCached, _, loginMethod, err := server.authenticate(req, ipAddr)
  950. assert.NoError(t, err)
  951. assert.False(t, isCached)
  952. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  953. // now the user should be cached
  954. cachedUser, ok := dataprovider.GetCachedWebDAVUser(username)
  955. if assert.True(t, ok) {
  956. assert.False(t, cachedUser.IsExpired())
  957. assert.True(t, cachedUser.Expiration.After(now.Add(time.Duration(c.Cache.Users.ExpirationTime)*time.Minute)))
  958. // authenticate must return the cached user now
  959. authUser, isCached, _, _, err := server.authenticate(req, ipAddr)
  960. assert.NoError(t, err)
  961. assert.True(t, isCached)
  962. assert.Equal(t, cachedUser.User, authUser)
  963. }
  964. folder, err := dataprovider.GetFolderByName(folderName)
  965. assert.NoError(t, err)
  966. // updating a used folder should invalidate the cache
  967. err = dataprovider.UpdateFolder(&folder, folder.Users)
  968. assert.NoError(t, err)
  969. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  970. assert.NoError(t, err)
  971. assert.False(t, isCached)
  972. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  973. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  974. if assert.True(t, ok) {
  975. assert.False(t, cachedUser.IsExpired())
  976. }
  977. err = dataprovider.DeleteFolder(folderName)
  978. assert.NoError(t, err)
  979. // removing a used folder should invalidate the cache
  980. _, isCached, _, loginMethod, err = server.authenticate(req, ipAddr)
  981. assert.NoError(t, err)
  982. assert.False(t, isCached)
  983. assert.Equal(t, dataprovider.LoginMethodPassword, loginMethod)
  984. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  985. if assert.True(t, ok) {
  986. assert.False(t, cachedUser.IsExpired())
  987. }
  988. err = dataprovider.DeleteUser(user.Username)
  989. assert.NoError(t, err)
  990. _, ok = dataprovider.GetCachedWebDAVUser(username)
  991. assert.False(t, ok)
  992. err = os.RemoveAll(u.GetHomeDir())
  993. assert.NoError(t, err)
  994. err = os.RemoveAll(folder.MappedPath)
  995. assert.NoError(t, err)
  996. }
  997. func TestUsersCacheSizeAndExpiration(t *testing.T) {
  998. username := "webdav_internal_test"
  999. password := "pwd"
  1000. u := dataprovider.User{
  1001. HomeDir: filepath.Join(os.TempDir(), username),
  1002. Status: 1,
  1003. ExpirationDate: 0,
  1004. }
  1005. u.Username = username + "1"
  1006. u.Password = password + "1"
  1007. u.Permissions = make(map[string][]string)
  1008. u.Permissions["/"] = []string{dataprovider.PermAny}
  1009. err := dataprovider.AddUser(&u)
  1010. assert.NoError(t, err)
  1011. user1, err := dataprovider.UserExists(u.Username)
  1012. assert.NoError(t, err)
  1013. u.Username = username + "2"
  1014. u.Password = password + "2"
  1015. err = dataprovider.AddUser(&u)
  1016. assert.NoError(t, err)
  1017. user2, err := dataprovider.UserExists(u.Username)
  1018. assert.NoError(t, err)
  1019. u.Username = username + "3"
  1020. u.Password = password + "3"
  1021. err = dataprovider.AddUser(&u)
  1022. assert.NoError(t, err)
  1023. user3, err := dataprovider.UserExists(u.Username)
  1024. assert.NoError(t, err)
  1025. u.Username = username + "4"
  1026. u.Password = password + "4"
  1027. err = dataprovider.AddUser(&u)
  1028. assert.NoError(t, err)
  1029. user4, err := dataprovider.UserExists(u.Username)
  1030. assert.NoError(t, err)
  1031. c := &Configuration{
  1032. Bindings: []Binding{
  1033. {
  1034. Port: 9000,
  1035. },
  1036. },
  1037. Cache: Cache{
  1038. Users: UsersCacheConfig{
  1039. MaxSize: 3,
  1040. ExpirationTime: 1,
  1041. },
  1042. },
  1043. }
  1044. dataprovider.InitializeWebDAVUserCache(c.Cache.Users.MaxSize)
  1045. server := webDavServer{
  1046. config: c,
  1047. binding: c.Bindings[0],
  1048. }
  1049. ipAddr := "127.0.1.1"
  1050. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1051. assert.NoError(t, err)
  1052. req.SetBasicAuth(user1.Username, password+"1")
  1053. _, isCached, _, loginMehod, err := server.authenticate(req, ipAddr)
  1054. assert.NoError(t, err)
  1055. assert.False(t, isCached)
  1056. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1057. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  1058. assert.NoError(t, err)
  1059. req.SetBasicAuth(user2.Username, password+"2")
  1060. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1061. assert.NoError(t, err)
  1062. assert.False(t, isCached)
  1063. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1064. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  1065. assert.NoError(t, err)
  1066. req.SetBasicAuth(user3.Username, password+"3")
  1067. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1068. assert.NoError(t, err)
  1069. assert.False(t, isCached)
  1070. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1071. // the first 3 users are now cached
  1072. _, ok := dataprovider.GetCachedWebDAVUser(user1.Username)
  1073. assert.True(t, ok)
  1074. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1075. assert.True(t, ok)
  1076. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1077. assert.True(t, ok)
  1078. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  1079. assert.NoError(t, err)
  1080. req.SetBasicAuth(user4.Username, password+"4")
  1081. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1082. assert.NoError(t, err)
  1083. assert.False(t, isCached)
  1084. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1085. // user1, the first cached, should be removed now
  1086. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1087. assert.False(t, ok)
  1088. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1089. assert.True(t, ok)
  1090. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1091. assert.True(t, ok)
  1092. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1093. assert.True(t, ok)
  1094. // user1 logins, user2 should be removed
  1095. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1096. assert.NoError(t, err)
  1097. req.SetBasicAuth(user1.Username, password+"1")
  1098. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1099. assert.NoError(t, err)
  1100. assert.False(t, isCached)
  1101. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1102. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1103. assert.False(t, ok)
  1104. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1105. assert.True(t, ok)
  1106. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1107. assert.True(t, ok)
  1108. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1109. assert.True(t, ok)
  1110. // user2 logins, user3 should be removed
  1111. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  1112. assert.NoError(t, err)
  1113. req.SetBasicAuth(user2.Username, password+"2")
  1114. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1115. assert.NoError(t, err)
  1116. assert.False(t, isCached)
  1117. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1118. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1119. assert.False(t, ok)
  1120. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1121. assert.True(t, ok)
  1122. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1123. assert.True(t, ok)
  1124. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1125. assert.True(t, ok)
  1126. // user3 logins, user4 should be removed
  1127. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  1128. assert.NoError(t, err)
  1129. req.SetBasicAuth(user3.Username, password+"3")
  1130. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1131. assert.NoError(t, err)
  1132. assert.False(t, isCached)
  1133. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1134. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1135. assert.False(t, ok)
  1136. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1137. assert.True(t, ok)
  1138. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1139. assert.True(t, ok)
  1140. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1141. assert.True(t, ok)
  1142. // now remove user1 after an update
  1143. user1.HomeDir += "_mod"
  1144. err = dataprovider.UpdateUser(&user1)
  1145. assert.NoError(t, err)
  1146. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1147. assert.False(t, ok)
  1148. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  1149. assert.NoError(t, err)
  1150. req.SetBasicAuth(user4.Username, password+"4")
  1151. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1152. assert.NoError(t, err)
  1153. assert.False(t, isCached)
  1154. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1155. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1156. assert.NoError(t, err)
  1157. req.SetBasicAuth(user1.Username, password+"1")
  1158. _, isCached, _, loginMehod, err = server.authenticate(req, ipAddr)
  1159. assert.NoError(t, err)
  1160. assert.False(t, isCached)
  1161. assert.Equal(t, dataprovider.LoginMethodPassword, loginMehod)
  1162. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1163. assert.False(t, ok)
  1164. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1165. assert.True(t, ok)
  1166. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1167. assert.True(t, ok)
  1168. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1169. assert.True(t, ok)
  1170. err = dataprovider.DeleteUser(user1.Username)
  1171. assert.NoError(t, err)
  1172. err = dataprovider.DeleteUser(user2.Username)
  1173. assert.NoError(t, err)
  1174. err = dataprovider.DeleteUser(user3.Username)
  1175. assert.NoError(t, err)
  1176. err = dataprovider.DeleteUser(user4.Username)
  1177. assert.NoError(t, err)
  1178. err = os.RemoveAll(u.GetHomeDir())
  1179. assert.NoError(t, err)
  1180. }
  1181. func TestUserCacheIsolation(t *testing.T) {
  1182. dataprovider.InitializeWebDAVUserCache(10)
  1183. username := "webdav_internal_cache_test"
  1184. password := "dav_pwd"
  1185. u := dataprovider.User{
  1186. Username: username,
  1187. Password: password,
  1188. HomeDir: filepath.Join(os.TempDir(), username),
  1189. Status: 1,
  1190. ExpirationDate: 0,
  1191. }
  1192. u.Permissions = make(map[string][]string)
  1193. u.Permissions["/"] = []string{dataprovider.PermAny}
  1194. err := dataprovider.AddUser(&u)
  1195. assert.NoError(t, err)
  1196. user, err := dataprovider.UserExists(u.Username)
  1197. assert.NoError(t, err)
  1198. cachedUser := &dataprovider.CachedUser{
  1199. User: user,
  1200. Expiration: time.Now().Add(24 * time.Hour),
  1201. Password: password,
  1202. LockSystem: webdav.NewMemLS(),
  1203. }
  1204. cachedUser.User.FsConfig.S3Config.AccessSecret = kms.NewPlainSecret("test secret")
  1205. err = cachedUser.User.FsConfig.S3Config.AccessSecret.Encrypt()
  1206. assert.NoError(t, err)
  1207. dataprovider.CacheWebDAVUser(cachedUser)
  1208. cachedUser, ok := dataprovider.GetCachedWebDAVUser(username)
  1209. if assert.True(t, ok) {
  1210. _, err = cachedUser.User.GetFilesystem("")
  1211. assert.NoError(t, err)
  1212. // the filesystem is now cached
  1213. }
  1214. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  1215. if assert.True(t, ok) {
  1216. assert.True(t, cachedUser.User.FsConfig.S3Config.AccessSecret.IsEncrypted())
  1217. err = cachedUser.User.FsConfig.S3Config.AccessSecret.Decrypt()
  1218. assert.NoError(t, err)
  1219. cachedUser.User.FsConfig.Provider = vfs.S3FilesystemProvider
  1220. _, err = cachedUser.User.GetFilesystem("")
  1221. assert.Error(t, err, "we don't have to get the previously cached filesystem!")
  1222. }
  1223. cachedUser, ok = dataprovider.GetCachedWebDAVUser(username)
  1224. if assert.True(t, ok) {
  1225. assert.Equal(t, vfs.LocalFilesystemProvider, cachedUser.User.FsConfig.Provider)
  1226. assert.False(t, cachedUser.User.FsConfig.S3Config.AccessSecret.IsEncrypted())
  1227. }
  1228. err = dataprovider.DeleteUser(username)
  1229. assert.NoError(t, err)
  1230. _, ok = dataprovider.GetCachedWebDAVUser(username)
  1231. assert.False(t, ok)
  1232. }
  1233. func TestRecoverer(t *testing.T) {
  1234. c := &Configuration{
  1235. Bindings: []Binding{
  1236. {
  1237. Port: 9000,
  1238. },
  1239. },
  1240. }
  1241. server := webDavServer{
  1242. config: c,
  1243. binding: c.Bindings[0],
  1244. }
  1245. rr := httptest.NewRecorder()
  1246. server.ServeHTTP(rr, nil)
  1247. assert.Equal(t, http.StatusInternalServerError, rr.Code)
  1248. }
  1249. func TestMimeCache(t *testing.T) {
  1250. cache := mimeCache{
  1251. maxSize: 0,
  1252. mimeTypes: make(map[string]string),
  1253. }
  1254. cache.addMimeToCache(".zip", "application/zip")
  1255. mtype := cache.getMimeFromCache(".zip")
  1256. assert.Equal(t, "", mtype)
  1257. cache.maxSize = 1
  1258. cache.addMimeToCache(".zip", "application/zip")
  1259. mtype = cache.getMimeFromCache(".zip")
  1260. assert.Equal(t, "application/zip", mtype)
  1261. cache.addMimeToCache(".jpg", "image/jpeg")
  1262. mtype = cache.getMimeFromCache(".jpg")
  1263. assert.Equal(t, "", mtype)
  1264. }
  1265. func TestVerifyTLSConnection(t *testing.T) {
  1266. oldCertMgr := certMgr
  1267. caCrlPath := filepath.Join(os.TempDir(), "testcrl.crt")
  1268. certPath := filepath.Join(os.TempDir(), "test.crt")
  1269. keyPath := filepath.Join(os.TempDir(), "test.key")
  1270. err := os.WriteFile(caCrlPath, []byte(caCRL), os.ModePerm)
  1271. assert.NoError(t, err)
  1272. err = os.WriteFile(certPath, []byte(webDavCert), os.ModePerm)
  1273. assert.NoError(t, err)
  1274. err = os.WriteFile(keyPath, []byte(webDavKey), os.ModePerm)
  1275. assert.NoError(t, err)
  1276. certMgr, err = common.NewCertManager(certPath, keyPath, "", "webdav_test")
  1277. assert.NoError(t, err)
  1278. certMgr.SetCARevocationLists([]string{caCrlPath})
  1279. err = certMgr.LoadCRLs()
  1280. assert.NoError(t, err)
  1281. crt, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  1282. assert.NoError(t, err)
  1283. x509crt, err := x509.ParseCertificate(crt.Certificate[0])
  1284. assert.NoError(t, err)
  1285. server := webDavServer{}
  1286. state := tls.ConnectionState{
  1287. PeerCertificates: []*x509.Certificate{x509crt},
  1288. }
  1289. err = server.verifyTLSConnection(state)
  1290. assert.Error(t, err) // no verified certification chain
  1291. crt, err = tls.X509KeyPair([]byte(caCRT), []byte(caKey))
  1292. assert.NoError(t, err)
  1293. x509CAcrt, err := x509.ParseCertificate(crt.Certificate[0])
  1294. assert.NoError(t, err)
  1295. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crt, x509CAcrt})
  1296. err = server.verifyTLSConnection(state)
  1297. assert.NoError(t, err)
  1298. crt, err = tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  1299. assert.NoError(t, err)
  1300. x509crtRevoked, err := x509.ParseCertificate(crt.Certificate[0])
  1301. assert.NoError(t, err)
  1302. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crtRevoked, x509CAcrt})
  1303. state.PeerCertificates = []*x509.Certificate{x509crtRevoked}
  1304. err = server.verifyTLSConnection(state)
  1305. assert.EqualError(t, err, common.ErrCrtRevoked.Error())
  1306. err = os.Remove(caCrlPath)
  1307. assert.NoError(t, err)
  1308. err = os.Remove(certPath)
  1309. assert.NoError(t, err)
  1310. err = os.Remove(keyPath)
  1311. assert.NoError(t, err)
  1312. certMgr = oldCertMgr
  1313. }
  1314. func TestMisc(t *testing.T) {
  1315. oldCertMgr := certMgr
  1316. certMgr = nil
  1317. err := ReloadCertificateMgr()
  1318. assert.Nil(t, err)
  1319. val := getConfigPath("", ".")
  1320. assert.Empty(t, val)
  1321. certMgr = oldCertMgr
  1322. }