internal_test.go 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275
  1. package webdavd
  2. import (
  3. "context"
  4. "crypto/tls"
  5. "crypto/x509"
  6. "errors"
  7. "fmt"
  8. "io"
  9. "io/ioutil"
  10. "net/http"
  11. "net/http/httptest"
  12. "os"
  13. "path"
  14. "path/filepath"
  15. "runtime"
  16. "testing"
  17. "time"
  18. "github.com/eikenb/pipeat"
  19. "github.com/stretchr/testify/assert"
  20. "golang.org/x/net/webdav"
  21. "github.com/drakkan/sftpgo/common"
  22. "github.com/drakkan/sftpgo/dataprovider"
  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, nil),
  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(), nil)
  319. connection := &Connection{
  320. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  321. request: nil,
  322. }
  323. dirsToRemove := []objectMapping{}
  324. orderedDirs := connection.orderDirsToRemove(dirsToRemove)
  325. assert.Equal(t, len(dirsToRemove), len(orderedDirs))
  326. dirsToRemove = []objectMapping{
  327. {
  328. fsPath: "dir1",
  329. virtualPath: "",
  330. },
  331. }
  332. orderedDirs = connection.orderDirsToRemove(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(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)
  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. u.HomeDir = filepath.Clean(os.TempDir())
  383. subDir := "subdir"
  384. mappedPath1 := filepath.Join(os.TempDir(), "vdir1")
  385. vdirPath1 := "/vdir1"
  386. mappedPath2 := filepath.Join(os.TempDir(), "vdir1", subDir)
  387. vdirPath2 := "/vdir2"
  388. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  389. BaseVirtualFolder: vfs.BaseVirtualFolder{
  390. MappedPath: mappedPath1,
  391. },
  392. VirtualPath: vdirPath1,
  393. })
  394. u.VirtualFolders = append(u.VirtualFolders, vfs.VirtualFolder{
  395. BaseVirtualFolder: vfs.BaseVirtualFolder{
  396. MappedPath: mappedPath2,
  397. },
  398. VirtualPath: vdirPath2,
  399. })
  400. _, err = server.validateUser(u, req)
  401. if assert.Error(t, err) {
  402. assert.EqualError(t, err, "overlapping mapped folders are allowed only with quota tracking disabled")
  403. }
  404. req.TLS = &tls.ConnectionState{}
  405. writeLog(req, nil)
  406. }
  407. func TestRemoteAddress(t *testing.T) {
  408. req, err := http.NewRequest(http.MethodGet, "/username", nil)
  409. assert.NoError(t, err)
  410. assert.Empty(t, req.RemoteAddr)
  411. remoteAddr1 := "100.100.100.100"
  412. remoteAddr2 := "172.172.172.172"
  413. req.Header.Set("X-Forwarded-For", remoteAddr1)
  414. checkRemoteAddress(req)
  415. assert.Equal(t, remoteAddr1, req.RemoteAddr)
  416. req.RemoteAddr = ""
  417. req.Header.Set("X-Forwarded-For", fmt.Sprintf("%v, %v", remoteAddr2, remoteAddr1))
  418. checkRemoteAddress(req)
  419. assert.Equal(t, remoteAddr2, req.RemoteAddr)
  420. req.Header.Del("X-Forwarded-For")
  421. req.RemoteAddr = ""
  422. req.Header.Set("X-Real-IP", remoteAddr1)
  423. checkRemoteAddress(req)
  424. assert.Equal(t, remoteAddr1, req.RemoteAddr)
  425. req.RemoteAddr = ""
  426. oldValue := common.Config.ProxyProtocol
  427. common.Config.ProxyProtocol = 1
  428. checkRemoteAddress(req)
  429. assert.Empty(t, req.RemoteAddr)
  430. common.Config.ProxyProtocol = oldValue
  431. }
  432. func TestConnWithNilRequest(t *testing.T) {
  433. c := &Connection{}
  434. assert.Empty(t, c.GetClientVersion())
  435. assert.Empty(t, c.GetCommand())
  436. assert.Empty(t, c.GetRemoteAddress())
  437. }
  438. func TestResolvePathErrors(t *testing.T) {
  439. ctx := context.Background()
  440. user := dataprovider.User{
  441. HomeDir: "invalid",
  442. }
  443. user.Permissions = make(map[string][]string)
  444. user.Permissions["/"] = []string{dataprovider.PermAny}
  445. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  446. connection := &Connection{
  447. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  448. }
  449. err := connection.Mkdir(ctx, "", os.ModePerm)
  450. if assert.Error(t, err) {
  451. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  452. }
  453. err = connection.Rename(ctx, "oldName", "newName")
  454. if assert.Error(t, err) {
  455. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  456. }
  457. _, err = connection.Stat(ctx, "name")
  458. if assert.Error(t, err) {
  459. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  460. }
  461. err = connection.RemoveAll(ctx, "")
  462. if assert.Error(t, err) {
  463. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  464. }
  465. _, err = connection.OpenFile(ctx, "", 0, os.ModePerm)
  466. if assert.Error(t, err) {
  467. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  468. }
  469. if runtime.GOOS != "windows" {
  470. connection.User.HomeDir = filepath.Clean(os.TempDir())
  471. connection.Fs = vfs.NewOsFs("connID", connection.User.HomeDir, nil)
  472. subDir := "sub"
  473. testTxtFile := "file.txt"
  474. err = os.MkdirAll(filepath.Join(os.TempDir(), subDir, subDir), os.ModePerm)
  475. assert.NoError(t, err)
  476. err = ioutil.WriteFile(filepath.Join(os.TempDir(), subDir, subDir, testTxtFile), []byte("content"), os.ModePerm)
  477. assert.NoError(t, err)
  478. err = os.Chmod(filepath.Join(os.TempDir(), subDir, subDir), 0001)
  479. assert.NoError(t, err)
  480. err = connection.Rename(ctx, testTxtFile, path.Join(subDir, subDir, testTxtFile))
  481. if assert.Error(t, err) {
  482. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  483. }
  484. _, err = connection.putFile(filepath.Join(connection.User.HomeDir, subDir, subDir, testTxtFile),
  485. path.Join(subDir, subDir, testTxtFile))
  486. if assert.Error(t, err) {
  487. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  488. }
  489. err = os.Chmod(filepath.Join(os.TempDir(), subDir, subDir), os.ModePerm)
  490. assert.NoError(t, err)
  491. err = os.RemoveAll(filepath.Join(os.TempDir(), subDir))
  492. assert.NoError(t, err)
  493. }
  494. }
  495. func TestFileAccessErrors(t *testing.T) {
  496. ctx := context.Background()
  497. user := dataprovider.User{
  498. HomeDir: filepath.Clean(os.TempDir()),
  499. }
  500. user.Permissions = make(map[string][]string)
  501. user.Permissions["/"] = []string{dataprovider.PermAny}
  502. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  503. connection := &Connection{
  504. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  505. }
  506. missingPath := "missing path"
  507. fsMissingPath := filepath.Join(user.HomeDir, missingPath)
  508. err := connection.RemoveAll(ctx, missingPath)
  509. if assert.Error(t, err) {
  510. assert.EqualError(t, err, os.ErrNotExist.Error())
  511. }
  512. _, err = connection.getFile(fsMissingPath, missingPath)
  513. if assert.Error(t, err) {
  514. assert.EqualError(t, err, os.ErrNotExist.Error())
  515. }
  516. _, err = connection.getFile(fsMissingPath, missingPath)
  517. if assert.Error(t, err) {
  518. assert.EqualError(t, err, os.ErrNotExist.Error())
  519. }
  520. p := filepath.Join(user.HomeDir, "adir", missingPath)
  521. _, err = connection.handleUploadToNewFile(p, p, path.Join("adir", missingPath))
  522. if assert.Error(t, err) {
  523. assert.EqualError(t, err, os.ErrNotExist.Error())
  524. }
  525. _, err = connection.handleUploadToExistingFile(p, p, 0, path.Join("adir", missingPath))
  526. if assert.Error(t, err) {
  527. assert.EqualError(t, err, os.ErrNotExist.Error())
  528. }
  529. connection.Fs = newMockOsFs(nil, false, fs.ConnectionID(), user.HomeDir, nil)
  530. _, err = connection.handleUploadToExistingFile(p, p, 0, path.Join("adir", missingPath))
  531. if assert.Error(t, err) {
  532. assert.EqualError(t, err, os.ErrNotExist.Error())
  533. }
  534. f, err := ioutil.TempFile("", "temp")
  535. assert.NoError(t, err)
  536. err = f.Close()
  537. assert.NoError(t, err)
  538. davFile, err := connection.handleUploadToExistingFile(f.Name(), f.Name(), 123, f.Name())
  539. if assert.NoError(t, err) {
  540. transfer := davFile.(*webDavFile)
  541. transfers := connection.GetTransfers()
  542. if assert.Equal(t, 1, len(transfers)) {
  543. assert.Equal(t, transfers[0].ID, transfer.GetID())
  544. assert.Equal(t, int64(123), transfer.InitialSize)
  545. err = transfer.Close()
  546. assert.NoError(t, err)
  547. assert.Equal(t, 0, len(connection.GetTransfers()))
  548. }
  549. }
  550. err = os.Remove(f.Name())
  551. assert.NoError(t, err)
  552. }
  553. func TestRemoveDirTree(t *testing.T) {
  554. user := dataprovider.User{
  555. HomeDir: filepath.Clean(os.TempDir()),
  556. }
  557. user.Permissions = make(map[string][]string)
  558. user.Permissions["/"] = []string{dataprovider.PermAny}
  559. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  560. connection := &Connection{
  561. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  562. }
  563. vpath := path.Join("adir", "missing")
  564. p := filepath.Join(user.HomeDir, "adir", "missing")
  565. err := connection.removeDirTree(p, vpath)
  566. if assert.Error(t, err) {
  567. assert.True(t, os.IsNotExist(err))
  568. }
  569. connection.Fs = newMockOsFs(nil, false, "mockID", user.HomeDir, nil)
  570. err = connection.removeDirTree(p, vpath)
  571. if assert.Error(t, err) {
  572. assert.True(t, os.IsNotExist(err))
  573. }
  574. errFake := errors.New("fake err")
  575. connection.Fs = newMockOsFs(errFake, false, "mockID", user.HomeDir, nil)
  576. err = connection.removeDirTree(p, vpath)
  577. if assert.Error(t, err) {
  578. assert.EqualError(t, err, errFake.Error())
  579. }
  580. connection.Fs = newMockOsFs(errWalkDir, true, "mockID", user.HomeDir, nil)
  581. err = connection.removeDirTree(p, vpath)
  582. if assert.Error(t, err) {
  583. assert.True(t, os.IsNotExist(err))
  584. }
  585. connection.Fs = newMockOsFs(errWalkFile, false, "mockID", user.HomeDir, nil)
  586. err = connection.removeDirTree(p, vpath)
  587. if assert.Error(t, err) {
  588. assert.EqualError(t, err, errWalkFile.Error())
  589. }
  590. connection.User.Permissions["/"] = []string{dataprovider.PermListItems}
  591. connection.Fs = newMockOsFs(nil, false, "mockID", user.HomeDir, nil)
  592. err = connection.removeDirTree(p, vpath)
  593. if assert.Error(t, err) {
  594. assert.EqualError(t, err, common.ErrPermissionDenied.Error())
  595. }
  596. }
  597. func TestContentType(t *testing.T) {
  598. user := dataprovider.User{
  599. HomeDir: filepath.Clean(os.TempDir()),
  600. }
  601. user.Permissions = make(map[string][]string)
  602. user.Permissions["/"] = []string{dataprovider.PermAny}
  603. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  604. connection := &Connection{
  605. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  606. }
  607. testFilePath := filepath.Join(user.HomeDir, testFile)
  608. ctx := context.Background()
  609. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  610. common.TransferDownload, 0, 0, 0, false, fs)
  611. fs = newMockOsFs(nil, false, fs.ConnectionID(), user.GetHomeDir(), nil)
  612. err := ioutil.WriteFile(testFilePath, []byte(""), os.ModePerm)
  613. assert.NoError(t, err)
  614. davFile := newWebDavFile(baseTransfer, nil, nil)
  615. davFile.Fs = fs
  616. fi, err := davFile.Stat()
  617. if assert.NoError(t, err) {
  618. ctype, err := fi.(*webDavFileInfo).ContentType(ctx)
  619. assert.NoError(t, err)
  620. assert.Equal(t, "application/custom-mime", ctype)
  621. }
  622. _, err = davFile.Readdir(-1)
  623. assert.Error(t, err)
  624. err = davFile.Close()
  625. assert.NoError(t, err)
  626. davFile = newWebDavFile(baseTransfer, nil, nil)
  627. davFile.Fs = vfs.NewOsFs("id", user.HomeDir, nil)
  628. fi, err = davFile.Stat()
  629. if assert.NoError(t, err) {
  630. ctype, err := fi.(*webDavFileInfo).ContentType(ctx)
  631. assert.NoError(t, err)
  632. assert.Equal(t, "text/plain; charset=utf-8", ctype)
  633. }
  634. err = davFile.Close()
  635. assert.NoError(t, err)
  636. fi.(*webDavFileInfo).fsPath = "missing"
  637. _, err = fi.(*webDavFileInfo).ContentType(ctx)
  638. assert.EqualError(t, err, webdav.ErrNotImplemented.Error())
  639. err = os.Remove(testFilePath)
  640. assert.NoError(t, err)
  641. }
  642. func TestTransferReadWriteErrors(t *testing.T) {
  643. user := dataprovider.User{
  644. HomeDir: filepath.Clean(os.TempDir()),
  645. }
  646. user.Permissions = make(map[string][]string)
  647. user.Permissions["/"] = []string{dataprovider.PermAny}
  648. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  649. connection := &Connection{
  650. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  651. }
  652. testFilePath := filepath.Join(user.HomeDir, testFile)
  653. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  654. common.TransferUpload, 0, 0, 0, false, fs)
  655. davFile := newWebDavFile(baseTransfer, nil, nil)
  656. p := make([]byte, 1)
  657. _, err := davFile.Read(p)
  658. assert.EqualError(t, err, common.ErrOpUnsupported.Error())
  659. r, w, err := pipeat.Pipe()
  660. assert.NoError(t, err)
  661. davFile = newWebDavFile(baseTransfer, nil, r)
  662. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  663. davFile = newWebDavFile(baseTransfer, vfs.NewPipeWriter(w), nil)
  664. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  665. err = r.Close()
  666. assert.NoError(t, err)
  667. err = w.Close()
  668. assert.NoError(t, err)
  669. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  670. common.TransferDownload, 0, 0, 0, false, fs)
  671. davFile = newWebDavFile(baseTransfer, nil, nil)
  672. _, err = davFile.Read(p)
  673. assert.True(t, os.IsNotExist(err))
  674. _, err = davFile.Stat()
  675. assert.True(t, os.IsNotExist(err))
  676. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  677. common.TransferDownload, 0, 0, 0, false, fs)
  678. err = ioutil.WriteFile(testFilePath, []byte(""), os.ModePerm)
  679. assert.NoError(t, err)
  680. f, err := os.Open(testFilePath)
  681. if assert.NoError(t, err) {
  682. err = f.Close()
  683. assert.NoError(t, err)
  684. }
  685. davFile = newWebDavFile(baseTransfer, nil, nil)
  686. davFile.reader = f
  687. err = davFile.Close()
  688. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  689. err = davFile.Close()
  690. assert.EqualError(t, err, common.ErrTransferClosed.Error())
  691. _, err = davFile.Read(p)
  692. assert.Error(t, err)
  693. info, err := davFile.Stat()
  694. if assert.NoError(t, err) {
  695. assert.Equal(t, int64(0), info.Size())
  696. }
  697. r, w, err = pipeat.Pipe()
  698. assert.NoError(t, err)
  699. mockFs := newMockOsFs(nil, false, fs.ConnectionID(), user.HomeDir, r)
  700. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  701. common.TransferDownload, 0, 0, 0, false, mockFs)
  702. davFile = newWebDavFile(baseTransfer, nil, nil)
  703. writeContent := []byte("content\r\n")
  704. go func() {
  705. n, err := w.Write(writeContent)
  706. assert.NoError(t, err)
  707. assert.Equal(t, len(writeContent), n)
  708. err = w.Close()
  709. assert.NoError(t, err)
  710. }()
  711. p = make([]byte, 64)
  712. n, err := davFile.Read(p)
  713. assert.EqualError(t, err, io.EOF.Error())
  714. assert.Equal(t, len(writeContent), n)
  715. err = davFile.Close()
  716. assert.NoError(t, err)
  717. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  718. common.TransferDownload, 0, 0, 0, false, fs)
  719. davFile = newWebDavFile(baseTransfer, nil, nil)
  720. davFile.writer = f
  721. err = davFile.Close()
  722. assert.EqualError(t, err, common.ErrGenericFailure.Error())
  723. err = os.Remove(testFilePath)
  724. assert.NoError(t, err)
  725. }
  726. func TestTransferSeek(t *testing.T) {
  727. user := dataprovider.User{
  728. HomeDir: filepath.Clean(os.TempDir()),
  729. }
  730. user.Permissions = make(map[string][]string)
  731. user.Permissions["/"] = []string{dataprovider.PermAny}
  732. fs := vfs.NewOsFs("connID", user.HomeDir, nil)
  733. connection := &Connection{
  734. BaseConnection: common.NewBaseConnection(fs.ConnectionID(), common.ProtocolWebDAV, user, fs),
  735. }
  736. testFilePath := filepath.Join(user.HomeDir, testFile)
  737. testFileContents := []byte("content")
  738. baseTransfer := common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  739. common.TransferUpload, 0, 0, 0, false, fs)
  740. davFile := newWebDavFile(baseTransfer, nil, nil)
  741. _, err := davFile.Seek(0, io.SeekStart)
  742. assert.EqualError(t, err, common.ErrOpUnsupported.Error())
  743. err = davFile.Close()
  744. assert.NoError(t, err)
  745. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  746. common.TransferDownload, 0, 0, 0, false, fs)
  747. davFile = newWebDavFile(baseTransfer, nil, nil)
  748. _, err = davFile.Seek(0, io.SeekCurrent)
  749. assert.True(t, os.IsNotExist(err))
  750. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  751. err = ioutil.WriteFile(testFilePath, testFileContents, os.ModePerm)
  752. assert.NoError(t, err)
  753. f, err := os.Open(testFilePath)
  754. if assert.NoError(t, err) {
  755. err = f.Close()
  756. assert.NoError(t, err)
  757. }
  758. baseTransfer = common.NewBaseTransfer(f, connection.BaseConnection, nil, testFilePath, testFile,
  759. common.TransferDownload, 0, 0, 0, false, fs)
  760. davFile = newWebDavFile(baseTransfer, nil, nil)
  761. _, err = davFile.Seek(0, io.SeekStart)
  762. assert.Error(t, err)
  763. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  764. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  765. common.TransferDownload, 0, 0, 0, false, fs)
  766. davFile = newWebDavFile(baseTransfer, nil, nil)
  767. res, err := davFile.Seek(0, io.SeekStart)
  768. assert.NoError(t, err)
  769. assert.Equal(t, int64(0), res)
  770. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  771. davFile = newWebDavFile(baseTransfer, nil, nil)
  772. res, err = davFile.Seek(0, io.SeekEnd)
  773. assert.NoError(t, err)
  774. assert.Equal(t, int64(len(testFileContents)), res)
  775. err = davFile.updateStatInfo()
  776. assert.Nil(t, err)
  777. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath+"1", testFile,
  778. common.TransferDownload, 0, 0, 0, false, fs)
  779. davFile = newWebDavFile(baseTransfer, nil, nil)
  780. _, err = davFile.Seek(0, io.SeekEnd)
  781. assert.True(t, os.IsNotExist(err))
  782. davFile.Connection.RemoveTransfer(davFile.BaseTransfer)
  783. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath, testFile,
  784. common.TransferDownload, 0, 0, 0, false, fs)
  785. davFile = newWebDavFile(baseTransfer, nil, nil)
  786. davFile.reader = f
  787. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  788. res, err = davFile.Seek(2, io.SeekStart)
  789. assert.NoError(t, err)
  790. assert.Equal(t, int64(2), res)
  791. davFile = newWebDavFile(baseTransfer, nil, nil)
  792. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  793. res, err = davFile.Seek(2, io.SeekEnd)
  794. assert.NoError(t, err)
  795. assert.Equal(t, int64(5), res)
  796. baseTransfer = common.NewBaseTransfer(nil, connection.BaseConnection, nil, testFilePath+"1", testFile,
  797. common.TransferDownload, 0, 0, 0, false, fs)
  798. davFile = newWebDavFile(baseTransfer, nil, nil)
  799. davFile.Fs = newMockOsFs(nil, true, fs.ConnectionID(), user.GetHomeDir(), nil)
  800. res, err = davFile.Seek(2, io.SeekEnd)
  801. assert.True(t, os.IsNotExist(err))
  802. assert.Equal(t, int64(0), res)
  803. assert.Len(t, common.Connections.GetStats(), 0)
  804. err = os.Remove(testFilePath)
  805. assert.NoError(t, err)
  806. }
  807. func TestBasicUsersCache(t *testing.T) {
  808. username := "webdav_internal_test"
  809. password := "pwd"
  810. u := dataprovider.User{
  811. Username: username,
  812. Password: password,
  813. HomeDir: filepath.Join(os.TempDir(), username),
  814. Status: 1,
  815. ExpirationDate: 0,
  816. }
  817. u.Permissions = make(map[string][]string)
  818. u.Permissions["/"] = []string{dataprovider.PermAny}
  819. err := dataprovider.AddUser(&u)
  820. assert.NoError(t, err)
  821. user, err := dataprovider.UserExists(u.Username)
  822. assert.NoError(t, err)
  823. c := &Configuration{
  824. Bindings: []Binding{
  825. {
  826. Port: 9000,
  827. },
  828. },
  829. Cache: Cache{
  830. Users: UsersCacheConfig{
  831. MaxSize: 50,
  832. ExpirationTime: 1,
  833. },
  834. },
  835. }
  836. server := webDavServer{
  837. config: c,
  838. binding: c.Bindings[0],
  839. }
  840. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user.Username), nil)
  841. assert.NoError(t, err)
  842. ipAddr := "127.0.0.1"
  843. _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  844. assert.Error(t, err)
  845. now := time.Now()
  846. req.SetBasicAuth(username, password)
  847. _, isCached, _, err := server.authenticate(req, ipAddr)
  848. assert.NoError(t, err)
  849. assert.False(t, isCached)
  850. // now the user should be cached
  851. var cachedUser *dataprovider.CachedUser
  852. result, ok := dataprovider.GetCachedWebDAVUser(username)
  853. if assert.True(t, ok) {
  854. cachedUser = result.(*dataprovider.CachedUser)
  855. assert.False(t, cachedUser.IsExpired())
  856. assert.True(t, cachedUser.Expiration.After(now.Add(time.Duration(c.Cache.Users.ExpirationTime)*time.Minute)))
  857. // authenticate must return the cached user now
  858. authUser, isCached, _, err := server.authenticate(req, ipAddr)
  859. assert.NoError(t, err)
  860. assert.True(t, isCached)
  861. assert.Equal(t, cachedUser.User, authUser)
  862. }
  863. // a wrong password must fail
  864. req.SetBasicAuth(username, "wrong")
  865. _, _, _, err = server.authenticate(req, ipAddr) //nolint:dogsled
  866. assert.EqualError(t, err, dataprovider.ErrInvalidCredentials.Error())
  867. req.SetBasicAuth(username, password)
  868. // force cached user expiration
  869. cachedUser.Expiration = now
  870. dataprovider.CacheWebDAVUser(cachedUser, c.Cache.Users.MaxSize)
  871. result, ok = dataprovider.GetCachedWebDAVUser(username)
  872. if assert.True(t, ok) {
  873. cachedUser = result.(*dataprovider.CachedUser)
  874. assert.True(t, cachedUser.IsExpired())
  875. }
  876. // now authenticate should get the user from the data provider and update the cache
  877. _, isCached, _, err = server.authenticate(req, ipAddr)
  878. assert.NoError(t, err)
  879. assert.False(t, isCached)
  880. result, ok = dataprovider.GetCachedWebDAVUser(username)
  881. if assert.True(t, ok) {
  882. cachedUser = result.(*dataprovider.CachedUser)
  883. assert.False(t, cachedUser.IsExpired())
  884. }
  885. // cache is invalidated after a user modification
  886. err = dataprovider.UpdateUser(&user)
  887. assert.NoError(t, err)
  888. _, ok = dataprovider.GetCachedWebDAVUser(username)
  889. assert.False(t, ok)
  890. _, isCached, _, err = server.authenticate(req, ipAddr)
  891. assert.NoError(t, err)
  892. assert.False(t, isCached)
  893. _, ok = dataprovider.GetCachedWebDAVUser(username)
  894. assert.True(t, ok)
  895. // cache is invalidated after user deletion
  896. err = dataprovider.DeleteUser(&user)
  897. assert.NoError(t, err)
  898. _, ok = dataprovider.GetCachedWebDAVUser(username)
  899. assert.False(t, ok)
  900. err = os.RemoveAll(u.GetHomeDir())
  901. assert.NoError(t, err)
  902. }
  903. func TestUsersCacheSizeAndExpiration(t *testing.T) {
  904. username := "webdav_internal_test"
  905. password := "pwd"
  906. u := dataprovider.User{
  907. HomeDir: filepath.Join(os.TempDir(), username),
  908. Status: 1,
  909. ExpirationDate: 0,
  910. }
  911. u.Username = username + "1"
  912. u.Password = password + "1"
  913. u.Permissions = make(map[string][]string)
  914. u.Permissions["/"] = []string{dataprovider.PermAny}
  915. err := dataprovider.AddUser(&u)
  916. assert.NoError(t, err)
  917. user1, err := dataprovider.UserExists(u.Username)
  918. assert.NoError(t, err)
  919. u.Username = username + "2"
  920. u.Password = password + "2"
  921. err = dataprovider.AddUser(&u)
  922. assert.NoError(t, err)
  923. user2, err := dataprovider.UserExists(u.Username)
  924. assert.NoError(t, err)
  925. u.Username = username + "3"
  926. u.Password = password + "3"
  927. err = dataprovider.AddUser(&u)
  928. assert.NoError(t, err)
  929. user3, err := dataprovider.UserExists(u.Username)
  930. assert.NoError(t, err)
  931. u.Username = username + "4"
  932. u.Password = password + "4"
  933. err = dataprovider.AddUser(&u)
  934. assert.NoError(t, err)
  935. user4, err := dataprovider.UserExists(u.Username)
  936. assert.NoError(t, err)
  937. c := &Configuration{
  938. Bindings: []Binding{
  939. {
  940. Port: 9000,
  941. },
  942. },
  943. Cache: Cache{
  944. Users: UsersCacheConfig{
  945. MaxSize: 3,
  946. ExpirationTime: 1,
  947. },
  948. },
  949. }
  950. server := webDavServer{
  951. config: c,
  952. binding: c.Bindings[0],
  953. }
  954. ipAddr := "127.0.1.1"
  955. req, err := http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  956. assert.NoError(t, err)
  957. req.SetBasicAuth(user1.Username, password+"1")
  958. _, isCached, _, err := server.authenticate(req, ipAddr)
  959. assert.NoError(t, err)
  960. assert.False(t, isCached)
  961. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  962. assert.NoError(t, err)
  963. req.SetBasicAuth(user2.Username, password+"2")
  964. _, isCached, _, err = server.authenticate(req, ipAddr)
  965. assert.NoError(t, err)
  966. assert.False(t, isCached)
  967. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  968. assert.NoError(t, err)
  969. req.SetBasicAuth(user3.Username, password+"3")
  970. _, isCached, _, err = server.authenticate(req, ipAddr)
  971. assert.NoError(t, err)
  972. assert.False(t, isCached)
  973. // the first 3 users are now cached
  974. _, ok := dataprovider.GetCachedWebDAVUser(user1.Username)
  975. assert.True(t, ok)
  976. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  977. assert.True(t, ok)
  978. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  979. assert.True(t, ok)
  980. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  981. assert.NoError(t, err)
  982. req.SetBasicAuth(user4.Username, password+"4")
  983. _, isCached, _, err = server.authenticate(req, ipAddr)
  984. assert.NoError(t, err)
  985. assert.False(t, isCached)
  986. // user1, the first cached, should be removed now
  987. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  988. assert.False(t, ok)
  989. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  990. assert.True(t, ok)
  991. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  992. assert.True(t, ok)
  993. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  994. assert.True(t, ok)
  995. // user1 logins, user2 should be removed
  996. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  997. assert.NoError(t, err)
  998. req.SetBasicAuth(user1.Username, password+"1")
  999. _, isCached, _, err = server.authenticate(req, ipAddr)
  1000. assert.NoError(t, err)
  1001. assert.False(t, isCached)
  1002. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1003. assert.False(t, ok)
  1004. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1005. assert.True(t, ok)
  1006. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1007. assert.True(t, ok)
  1008. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1009. assert.True(t, ok)
  1010. // user2 logins, user3 should be removed
  1011. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user2.Username), nil)
  1012. assert.NoError(t, err)
  1013. req.SetBasicAuth(user2.Username, password+"2")
  1014. _, isCached, _, err = server.authenticate(req, ipAddr)
  1015. assert.NoError(t, err)
  1016. assert.False(t, isCached)
  1017. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1018. assert.False(t, ok)
  1019. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1020. assert.True(t, ok)
  1021. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1022. assert.True(t, ok)
  1023. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1024. assert.True(t, ok)
  1025. // user3 logins, user4 should be removed
  1026. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user3.Username), nil)
  1027. assert.NoError(t, err)
  1028. req.SetBasicAuth(user3.Username, password+"3")
  1029. _, isCached, _, err = server.authenticate(req, ipAddr)
  1030. assert.NoError(t, err)
  1031. assert.False(t, isCached)
  1032. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1033. assert.False(t, ok)
  1034. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1035. assert.True(t, ok)
  1036. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1037. assert.True(t, ok)
  1038. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1039. assert.True(t, ok)
  1040. // now remove user1 after an update
  1041. err = dataprovider.UpdateUser(&user1)
  1042. assert.NoError(t, err)
  1043. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1044. assert.False(t, ok)
  1045. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user4.Username), nil)
  1046. assert.NoError(t, err)
  1047. req.SetBasicAuth(user4.Username, password+"4")
  1048. _, isCached, _, err = server.authenticate(req, ipAddr)
  1049. assert.NoError(t, err)
  1050. assert.False(t, isCached)
  1051. req, err = http.NewRequest(http.MethodGet, fmt.Sprintf("/%v", user1.Username), nil)
  1052. assert.NoError(t, err)
  1053. req.SetBasicAuth(user1.Username, password+"1")
  1054. _, isCached, _, err = server.authenticate(req, ipAddr)
  1055. assert.NoError(t, err)
  1056. assert.False(t, isCached)
  1057. _, ok = dataprovider.GetCachedWebDAVUser(user2.Username)
  1058. assert.False(t, ok)
  1059. _, ok = dataprovider.GetCachedWebDAVUser(user1.Username)
  1060. assert.True(t, ok)
  1061. _, ok = dataprovider.GetCachedWebDAVUser(user3.Username)
  1062. assert.True(t, ok)
  1063. _, ok = dataprovider.GetCachedWebDAVUser(user4.Username)
  1064. assert.True(t, ok)
  1065. err = dataprovider.DeleteUser(&user1)
  1066. assert.NoError(t, err)
  1067. err = dataprovider.DeleteUser(&user2)
  1068. assert.NoError(t, err)
  1069. err = dataprovider.DeleteUser(&user3)
  1070. assert.NoError(t, err)
  1071. err = dataprovider.DeleteUser(&user4)
  1072. assert.NoError(t, err)
  1073. err = os.RemoveAll(u.GetHomeDir())
  1074. assert.NoError(t, err)
  1075. }
  1076. func TestRecoverer(t *testing.T) {
  1077. c := &Configuration{
  1078. Bindings: []Binding{
  1079. {
  1080. Port: 9000,
  1081. },
  1082. },
  1083. }
  1084. server := webDavServer{
  1085. config: c,
  1086. binding: c.Bindings[0],
  1087. }
  1088. rr := httptest.NewRecorder()
  1089. server.ServeHTTP(rr, nil)
  1090. assert.Equal(t, http.StatusInternalServerError, rr.Code)
  1091. }
  1092. func TestMimeCache(t *testing.T) {
  1093. cache := mimeCache{
  1094. maxSize: 0,
  1095. mimeTypes: make(map[string]string),
  1096. }
  1097. cache.addMimeToCache(".zip", "application/zip")
  1098. mtype := cache.getMimeFromCache(".zip")
  1099. assert.Equal(t, "", mtype)
  1100. cache.maxSize = 1
  1101. cache.addMimeToCache(".zip", "application/zip")
  1102. mtype = cache.getMimeFromCache(".zip")
  1103. assert.Equal(t, "application/zip", mtype)
  1104. cache.addMimeToCache(".jpg", "image/jpeg")
  1105. mtype = cache.getMimeFromCache(".jpg")
  1106. assert.Equal(t, "", mtype)
  1107. }
  1108. func TestVerifyTLSConnection(t *testing.T) {
  1109. oldCertMgr := certMgr
  1110. caCrlPath := filepath.Join(os.TempDir(), "testcrl.crt")
  1111. certPath := filepath.Join(os.TempDir(), "test.crt")
  1112. keyPath := filepath.Join(os.TempDir(), "test.key")
  1113. err := ioutil.WriteFile(caCrlPath, []byte(caCRL), os.ModePerm)
  1114. assert.NoError(t, err)
  1115. err = ioutil.WriteFile(certPath, []byte(webDavCert), os.ModePerm)
  1116. assert.NoError(t, err)
  1117. err = ioutil.WriteFile(keyPath, []byte(webDavKey), os.ModePerm)
  1118. assert.NoError(t, err)
  1119. certMgr, err = common.NewCertManager(certPath, keyPath, "", "webdav_test")
  1120. assert.NoError(t, err)
  1121. certMgr.SetCARevocationLists([]string{caCrlPath})
  1122. err = certMgr.LoadCRLs()
  1123. assert.NoError(t, err)
  1124. crt, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  1125. assert.NoError(t, err)
  1126. x509crt, err := x509.ParseCertificate(crt.Certificate[0])
  1127. assert.NoError(t, err)
  1128. server := webDavServer{}
  1129. state := tls.ConnectionState{
  1130. PeerCertificates: []*x509.Certificate{x509crt},
  1131. }
  1132. err = server.verifyTLSConnection(state)
  1133. assert.Error(t, err) // no verified certification chain
  1134. crt, err = tls.X509KeyPair([]byte(caCRT), []byte(caKey))
  1135. assert.NoError(t, err)
  1136. x509CAcrt, err := x509.ParseCertificate(crt.Certificate[0])
  1137. assert.NoError(t, err)
  1138. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crt, x509CAcrt})
  1139. err = server.verifyTLSConnection(state)
  1140. assert.NoError(t, err)
  1141. crt, err = tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  1142. assert.NoError(t, err)
  1143. x509crtRevoked, err := x509.ParseCertificate(crt.Certificate[0])
  1144. assert.NoError(t, err)
  1145. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crtRevoked, x509CAcrt})
  1146. state.PeerCertificates = []*x509.Certificate{x509crtRevoked}
  1147. err = server.verifyTLSConnection(state)
  1148. assert.EqualError(t, err, common.ErrCrtRevoked.Error())
  1149. err = os.Remove(caCrlPath)
  1150. assert.NoError(t, err)
  1151. err = os.Remove(certPath)
  1152. assert.NoError(t, err)
  1153. err = os.Remove(keyPath)
  1154. assert.NoError(t, err)
  1155. certMgr = oldCertMgr
  1156. }