internal_test.go 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786
  1. package httpd
  2. import (
  3. "bytes"
  4. "context"
  5. "crypto/tls"
  6. "crypto/x509"
  7. "encoding/json"
  8. "errors"
  9. "fmt"
  10. "html/template"
  11. "io/ioutil"
  12. "net/http"
  13. "net/http/httptest"
  14. "net/url"
  15. "os"
  16. "path"
  17. "path/filepath"
  18. "runtime"
  19. "strings"
  20. "testing"
  21. "time"
  22. "github.com/go-chi/chi"
  23. "github.com/go-chi/jwtauth"
  24. "github.com/lestrrat-go/jwx/jwt"
  25. "github.com/stretchr/testify/assert"
  26. "github.com/stretchr/testify/require"
  27. "github.com/drakkan/sftpgo/common"
  28. "github.com/drakkan/sftpgo/dataprovider"
  29. "github.com/drakkan/sftpgo/kms"
  30. "github.com/drakkan/sftpgo/utils"
  31. "github.com/drakkan/sftpgo/vfs"
  32. )
  33. const (
  34. httpdCert = `-----BEGIN CERTIFICATE-----
  35. MIICHTCCAaKgAwIBAgIUHnqw7QnB1Bj9oUsNpdb+ZkFPOxMwCgYIKoZIzj0EAwIw
  36. RTELMAkGA1UEBhMCQVUxEzARBgNVBAgMClNvbWUtU3RhdGUxITAfBgNVBAoMGElu
  37. dGVybmV0IFdpZGdpdHMgUHR5IEx0ZDAeFw0yMDAyMDQwOTUzMDRaFw0zMDAyMDEw
  38. OTUzMDRaMEUxCzAJBgNVBAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMSEwHwYD
  39. VQQKDBhJbnRlcm5ldCBXaWRnaXRzIFB0eSBMdGQwdjAQBgcqhkjOPQIBBgUrgQQA
  40. IgNiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVqWvrJ51t5OxV0v25NsOgR82CA
  41. NXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIVCzgWkxiz7XE4lgUwX44FCXZM
  42. 3+JeUbKjUzBRMB0GA1UdDgQWBBRhLw+/o3+Z02MI/d4tmaMui9W16jAfBgNVHSME
  43. GDAWgBRhLw+/o3+Z02MI/d4tmaMui9W16jAPBgNVHRMBAf8EBTADAQH/MAoGCCqG
  44. SM49BAMCA2kAMGYCMQDqLt2lm8mE+tGgtjDmtFgdOcI72HSbRQ74D5rYTzgST1rY
  45. /8wTi5xl8TiFUyLMUsICMQC5ViVxdXbhuG7gX6yEqSkMKZICHpO8hqFwOD/uaFVI
  46. dV4vKmHUzwK/eIx+8Ay3neE=
  47. -----END CERTIFICATE-----`
  48. httpdKey = `-----BEGIN EC PARAMETERS-----
  49. BgUrgQQAIg==
  50. -----END EC PARAMETERS-----
  51. -----BEGIN EC PRIVATE KEY-----
  52. MIGkAgEBBDCfMNsN6miEE3rVyUPwElfiJSWaR5huPCzUenZOfJT04GAcQdWvEju3
  53. UM2lmBLIXpGgBwYFK4EEACKhZANiAARCjRMqJ85rzMC998X5z761nJ+xL3bkmGVq
  54. WvrJ51t5OxV0v25NsOgR82CANXUgvhVYs7vNFN+jxtb2aj6Xg+/2G/BNxkaFspIV
  55. CzgWkxiz7XE4lgUwX44FCXZM3+JeUbI=
  56. -----END EC PRIVATE KEY-----`
  57. caCRT = `-----BEGIN CERTIFICATE-----
  58. MIIE5jCCAs6gAwIBAgIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0
  59. QXV0aDAeFw0yMTAxMDIyMTIwNTVaFw0yMjA3MDIyMTMwNTJaMBMxETAPBgNVBAMT
  60. CENlcnRBdXRoMIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEA4Tiho5xW
  61. AC15JRkMwfp3/TJwI2As7MY5dele5cmdr5bHAE+sRKqC+Ti88OJWCV5saoyax/1S
  62. CjxJlQMZMl169P1QYJskKjdG2sdv6RLWLMgwSNRRjxp/Bw9dHdiEb9MjLgu28Jro
  63. 9peQkHcRHeMf5hM9WvlIJGrdzbC4hUehmqggcqgARainBkYjf0SwuWxHeu4nMqkp
  64. Ak5tcSTLCjHfEFHZ9Te0TIPG5YkWocQKyeLgu4lvuU+DD2W2lym+YVUtRMGs1Env
  65. k7p+N0DcGU26qfzZ2sF5ZXkqm7dBsGQB9pIxwc2Q8T1dCIyP9OQCKVILdc5aVFf1
  66. cryQFHYzYNNZXFlIBims5VV5Mgfp8ESHQSue+v6n6ykecLEyKt1F1Y/MWY/nWUSI
  67. 8zdq83jdBAZVjo9MSthxVn57/06s/hQca65IpcTZV2gX0a+eRlAVqaRbAhL3LaZe
  68. bYsW3WHKoUOftwemuep3nL51TzlXZVL7Oz/ClGaEOsnGG9KFO6jh+W768qC0zLQI
  69. CdE7v2Zex98sZteHCg9fGJHIaYoF0aJG5P3WI5oZf2fy7UIYN9ADLFZiorCXAZEh
  70. CSU6mDoRViZ4RGR9GZxbDZ9KYn7O8M/KCR72bkQg73TlMsk1zSXEw0MKLUjtsw6c
  71. rZ0Jt8t3sRatHO3JrYHALMt9vZfyNCZp0IsCAwEAAaNFMEMwDgYDVR0PAQH/BAQD
  72. AgEGMBIGA1UdEwEB/wQIMAYBAf8CAQAwHQYDVR0OBBYEFO1yCNAGr/zQTJIi8lw3
  73. w5OiuBvMMA0GCSqGSIb3DQEBCwUAA4ICAQA6gCNuM7r8mnx674dm31GxBjQy5ZwB
  74. 7CxDzYEvL/oiZ3Tv3HlPfN2LAAsJUfGnghh9DOytenL2CTZWjl/emP5eijzmlP+9
  75. zva5I6CIMCf/eDDVsRdO244t0o4uG7+At0IgSDM3bpVaVb4RHZNjEziYChsEYY8d
  76. HK6iwuRSvFniV6yhR/Vj1Ymi9yZ5xclqseLXiQnUB0PkfIk23+7s42cXB16653fH
  77. O/FsPyKBLiKJArizLYQc12aP3QOrYoYD9+fAzIIzew7A5C0aanZCGzkuFpO6TRlD
  78. Tb7ry9Gf0DfPpCgxraH8tOcmnqp/ka3hjqo/SRnnTk0IFrmmLdarJvjD46rKwBo4
  79. MjyAIR1mQ5j8GTlSFBmSgETOQ/EYvO3FPLmra1Fh7L+DvaVzTpqI9fG3TuyyY+Ri
  80. Fby4ycTOGSZOe5Fh8lqkX5Y47mCUJ3zHzOA1vUJy2eTlMRGpu47Eb1++Vm6EzPUP
  81. 2EF5aD+zwcssh+atZvQbwxpgVqVcyLt91RSkKkmZQslh0rnlTb68yxvUnD3zw7So
  82. o6TAf9UvwVMEvdLT9NnFd6hwi2jcNte/h538GJwXeBb8EkfpqLKpTKyicnOdkamZ
  83. 7E9zY8SHNRYMwB9coQ/W8NvufbCgkvOoLyMXk5edbXofXl3PhNGOlraWbghBnzf5
  84. r3rwjFsQOoZotA==
  85. -----END CERTIFICATE-----`
  86. caKey = `-----BEGIN RSA PRIVATE KEY-----
  87. MIIJKQIBAAKCAgEA4Tiho5xWAC15JRkMwfp3/TJwI2As7MY5dele5cmdr5bHAE+s
  88. RKqC+Ti88OJWCV5saoyax/1SCjxJlQMZMl169P1QYJskKjdG2sdv6RLWLMgwSNRR
  89. jxp/Bw9dHdiEb9MjLgu28Jro9peQkHcRHeMf5hM9WvlIJGrdzbC4hUehmqggcqgA
  90. RainBkYjf0SwuWxHeu4nMqkpAk5tcSTLCjHfEFHZ9Te0TIPG5YkWocQKyeLgu4lv
  91. uU+DD2W2lym+YVUtRMGs1Envk7p+N0DcGU26qfzZ2sF5ZXkqm7dBsGQB9pIxwc2Q
  92. 8T1dCIyP9OQCKVILdc5aVFf1cryQFHYzYNNZXFlIBims5VV5Mgfp8ESHQSue+v6n
  93. 6ykecLEyKt1F1Y/MWY/nWUSI8zdq83jdBAZVjo9MSthxVn57/06s/hQca65IpcTZ
  94. V2gX0a+eRlAVqaRbAhL3LaZebYsW3WHKoUOftwemuep3nL51TzlXZVL7Oz/ClGaE
  95. OsnGG9KFO6jh+W768qC0zLQICdE7v2Zex98sZteHCg9fGJHIaYoF0aJG5P3WI5oZ
  96. f2fy7UIYN9ADLFZiorCXAZEhCSU6mDoRViZ4RGR9GZxbDZ9KYn7O8M/KCR72bkQg
  97. 73TlMsk1zSXEw0MKLUjtsw6crZ0Jt8t3sRatHO3JrYHALMt9vZfyNCZp0IsCAwEA
  98. AQKCAgAV+ElERYbaI5VyufvVnFJCH75ypPoc6sVGLEq2jbFVJJcq/5qlZCC8oP1F
  99. Xj7YUR6wUiDzK1Hqb7EZ2SCHGjlZVrCVi+y+NYAy7UuMZ+r+mVSkdhmypPoJPUVv
  100. GOTqZ6VB46Cn3eSl0WknvoWr7bD555yPmEuiSc5zNy74yWEJTidEKAFGyknowcTK
  101. sG+w1tAuPLcUKQ44DGB+rgEkcHL7C5EAa7upzx0C3RmZFB+dTAVyJdkBMbFuOhTS
  102. sB7DLeTplR7/4mp9da7EQw51ZXC1DlZOEZt++4/desXsqATNAbva1OuzrLG7mMKe
  103. N/PCBh/aERQcsCvgUmaXqGQgqN1Jhw8kbXnjZnVd9iE7TAh7ki3VqNy1OMgTwOex
  104. bBYWaCqHuDYIxCjeW0qLJcn0cKQ13FVYrxgInf4Jp82SQht5b/zLL3IRZEyKcLJF
  105. kL6g1wlmTUTUX0z8eZzlM0ZCrqtExjgElMO/rV971nyNV5WU8Og3NmE8/slqMrmJ
  106. DlrQr9q0WJsDKj1IMe46EUM6ix7bbxC5NIfJ96dgdxZDn6ghjca6iZYqqUACvmUj
  107. cq08s3R4Ouw9/87kn11wwGBx2yDueCwrjKEGc0RKjweGbwu0nBxOrkJ8JXz6bAv7
  108. 1OKfYaX3afI9B8x4uaiuRs38oBQlg9uAYFfl4HNBPuQikGLmsQKCAQEA8VjFOsaz
  109. y6NMZzKXi7WZ48uu3ed5x3Kf6RyDr1WvQ1jkBMv9b6b8Gp1CRnPqviRBto9L8QAg
  110. bCXZTqnXzn//brskmW8IZgqjAlf89AWa53piucu9/hgidrHRZobs5gTqev28uJdc
  111. zcuw1g8c3nCpY9WeTjHODzX5NXYRLFpkazLfYa6c8Q9jZR4KKrpdM+66fxL0JlOd
  112. 7dN0oQtEqEAugsd3cwkZgvWhY4oM7FGErrZoDLy273ZdJzi/vU+dThyVzfD8Ab8u
  113. VxxuobVMT/S608zbe+uaiUdov5s96OkCl87403UNKJBH+6LNb3rjBBLE9NPN5ET9
  114. JLQMrYd+zj8jQwKCAQEA7uU5I9MOufo9bIgJqjY4Ie1+Ex9DZEMUYFAvGNCJCVcS
  115. mwOdGF8AWzIavTLACmEDJO7t/OrBdoo4L7IEsCNjgA3WiIwIMiWUVqveAGUMEXr6
  116. TRI5EolV6FTqqIP6AS+BAeBq7G1ELgsTrWNHh11rW3+3kBMuOCn77PUQ8WHwcq/r
  117. teZcZn4Ewcr6P7cBODgVvnBPhe/J8xHS0HFVCeS1CvaiNYgees5yA80Apo9IPjDJ
  118. YWawLjmH5wUBI5yDFVp067wjqJnoKPSoKwWkZXqUk+zgFXx5KT0gh/c5yh1frASp
  119. q6oaYnHEVC5qj2SpT1GFLonTcrQUXiSkiUudvNu1GQKCAQEAmko+5GFtRe0ihgLQ
  120. 4S76r6diJli6AKil1Fg3U1r6zZpBQ1PJtJxTJQyN9w5Z7q6tF/GqAesrzxevQdvQ
  121. rCImAPtA3ZofC2UXawMnIjWHHx6diNvYnV1+gtUQ4nO1dSOFZ5VZFcUmPiZO6boF
  122. oaryj3FcX+71JcJCjEvrlKhA9Es0hXUkvfMxfs5if4he1zlyHpTWYr4oA4egUugq
  123. P0mwskikc3VIyvEO+NyjgFxo72yLPkFSzemkidN8uKDyFqKtnlfGM7OuA2CY1WZa
  124. 3+67lXWshx9KzyJIs92iCYkU8EoPxtdYzyrV6efdX7x27v60zTOut5TnJJS6WiF6
  125. Do5MkwKCAQAxoR9IyP0DN/BwzqYrXU42Bi+t603F04W1KJNQNWpyrUspNwv41yus
  126. xnD1o0hwH41Wq+h3JZIBfV+E0RfWO9Pc84MBJQ5C1LnHc7cQH+3s575+Km3+4tcd
  127. CB8j2R8kBeloKWYtLdn/Mr/ownpGreqyvIq2/LUaZ+Z1aMgXTYB1YwS16mCBzmZQ
  128. mEl62RsAwe4KfSyYJ6OtwqMoOJMxFfliiLBULK4gVykqjvk2oQeiG+KKQJoTUFJi
  129. dRCyhD5bPkqR+qjxyt+HOqSBI4/uoROi05AOBqjpH1DVzk+MJKQOiX1yM0l98CKY
  130. Vng+x+vAla/0Zh+ucajVkgk4mKPxazdpAoIBAQC17vWk4KYJpF2RC3pKPcQ0PdiX
  131. bN35YNlvyhkYlSfDNdyH3aDrGiycUyW2mMXUgEDFsLRxHMTL+zPC6efqO6sTAJDY
  132. cBptsW4drW/qo8NTx3dNOisLkW+mGGJOR/w157hREFr29ymCVMYu/Z7fVWIeSpCq
  133. p3u8YX8WTljrxwSczlGjvpM7uJx3SfYRM4TUoy+8wU8bK74LywLa5f60bQY6Dye0
  134. Gqd9O6OoPfgcQlwjC5MiAofeqwPJvU0hQOPoehZyNLAmOCWXTYWaTP7lxO1r6+NE
  135. M3hGYqW3W8Ixua71OskCypBZg/HVlIP/lzjRzdx+VOB2hbWVth2Iup/Z1egW
  136. -----END RSA PRIVATE KEY-----`
  137. caCRL = `-----BEGIN X509 CRL-----
  138. MIICpzCBkAIBATANBgkqhkiG9w0BAQsFADATMREwDwYDVQQDEwhDZXJ0QXV0aBcN
  139. MjEwMTAyMjEzNDA1WhcNMjMwMTAyMjEzNDA1WjAkMCICEQC+l04DbHWMyC3fG09k
  140. VXf+Fw0yMTAxMDIyMTM0MDVaoCMwITAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJc
  141. N8OTorgbzDANBgkqhkiG9w0BAQsFAAOCAgEAEJ7z+uNc8sqtxlOhSdTGDzX/xput
  142. E857kFQkSlMnU2whQ8c+XpYrBLA5vIZJNSSwohTpM4+zVBX/bJpmu3wqqaArRO9/
  143. YcW5mQk9Anvb4WjQW1cHmtNapMTzoC9AiYt/OWPfy+P6JCgCr4Hy6LgQyIRL6bM9
  144. VYTalolOm1qa4Y5cIeT7iHq/91mfaqo8/6MYRjLl8DOTROpmw8OS9bCXkzGKdCat
  145. AbAzwkQUSauyoCQ10rpX+Y64w9ng3g4Dr20aCqPf5osaqplEJ2HTK8ljDTidlslv
  146. 9anQj8ax3Su89vI8+hK+YbfVQwrThabgdSjQsn+veyx8GlP8WwHLAQ379KjZjWg+
  147. OlOSwBeU1vTdP0QcB8X5C2gVujAyuQekbaV86xzIBOj7vZdfHZ6ee30TZ2FKiMyg
  148. 7/N2OqW0w77ChsjB4MSHJCfuTgIeg62GzuZXLM+Q2Z9LBdtm4Byg+sm/P52adOEg
  149. gVb2Zf4KSvsAmA0PIBlu449/QXUFcMxzLFy7mwTeZj2B4Ln0Hm0szV9f9R8MwMtB
  150. SyLYxVH+mgqaR6Jkk22Q/yYyLPaELfafX5gp/AIXG8n0zxfVaTvK3auSgb1Q6ZLS
  151. 5QH9dSIsmZHlPq7GoSXmKpMdjUL8eaky/IMteioyXgsBiATzl5L2dsw6MTX3MDF0
  152. QbDK+MzhmbKfDxs=
  153. -----END X509 CRL-----`
  154. client1Crt = `-----BEGIN CERTIFICATE-----
  155. MIIEITCCAgmgAwIBAgIRAIppZHoj1hM80D7WzTEKLuAwDQYJKoZIhvcNAQELBQAw
  156. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjEwMTAyMjEyMzEwWhcNMjIwNzAyMjEz
  157. MDUxWjASMRAwDgYDVQQDEwdjbGllbnQxMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  158. MIIBCgKCAQEAoKbYY9MdF2kF/nhBESIiZTdVYtA8XL9xrIZyDj9EnCiTxHiVbJtH
  159. XVwszqSl5TRrotPmnmAQcX3r8OCk+z+RQZ0QQj257P3kG6q4rNnOcWCS5xEd20jP
  160. yhQ3m+hMGfZsotNTQze1ochuQgLUN6IPyPxZkH22ia3jX4iu1eo/QxeLYHj1UHw4
  161. 3Cii9yE+j5kPUC21xmnrGKdUrB55NYLXHx6yTIqYR5znSOVB8oJi18/hwdZmH859
  162. DHhm0Hx1HrS+jbjI3+CMorZJ3WUyNf+CkiVLD3xYutPbxzEpwiqkG/XYzLH0habT
  163. cDcILo18n+o3jvem2KWBrDhyairjIDscwQIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  164. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBSJ5GIv
  165. zIrE4ZSQt2+CGblKTDswizAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJcN8OTorgb
  166. zDANBgkqhkiG9w0BAQsFAAOCAgEALh4f5GhvNYNou0Ab04iQBbLEdOu2RlbK1B5n
  167. K9P/umYenBHMY/z6HT3+6tpcHsDuqE8UVdq3f3Gh4S2Gu9m8PRitT+cJ3gdo9Plm
  168. 3rD4ufn/s6rGg3ppydXcedm17492tbccUDWOBZw3IO/ASVq13WPgT0/Kev7cPq0k
  169. sSdSNhVeXqx8Myc2/d+8GYyzbul2Kpfa7h9i24sK49E9ftnSmsIvngONo08eT1T0
  170. 3wAOyK2981LIsHaAWcneShKFLDB6LeXIT9oitOYhiykhFlBZ4M1GNlSNfhQ8IIQP
  171. xbqMNXCLkW4/BtLhGEEcg0QVso6Kudl9rzgTfQknrdF7pHp6rS46wYUjoSyIY6dl
  172. oLmnoAVJX36J3QPWelePI9e07X2wrTfiZWewwgw3KNRWjd6/zfPLe7GoqXnK1S2z
  173. PT8qMfCaTwKTtUkzXuTFvQ8bAo2My/mS8FOcpkt2oQWeOsADHAUX7fz5BCoa2DL3
  174. k/7Mh4gVT+JYZEoTwCFuYHgMWFWe98naqHi9lB4yR981p1QgXgxO7qBeipagKY1F
  175. LlH1iwXUqZ3MZnkNA+4e1Fglsw3sa/rC+L98HnznJ/YbTfQbCP6aQ1qcOymrjMud
  176. 7MrFwqZjtd/SK4Qx1VpK6jGEAtPgWBTUS3p9ayg6lqjMBjsmySWfvRsDQbq6P5Ct
  177. O/e3EH8=
  178. -----END CERTIFICATE-----`
  179. client1Key = `-----BEGIN RSA PRIVATE KEY-----
  180. MIIEpAIBAAKCAQEAoKbYY9MdF2kF/nhBESIiZTdVYtA8XL9xrIZyDj9EnCiTxHiV
  181. bJtHXVwszqSl5TRrotPmnmAQcX3r8OCk+z+RQZ0QQj257P3kG6q4rNnOcWCS5xEd
  182. 20jPyhQ3m+hMGfZsotNTQze1ochuQgLUN6IPyPxZkH22ia3jX4iu1eo/QxeLYHj1
  183. UHw43Cii9yE+j5kPUC21xmnrGKdUrB55NYLXHx6yTIqYR5znSOVB8oJi18/hwdZm
  184. H859DHhm0Hx1HrS+jbjI3+CMorZJ3WUyNf+CkiVLD3xYutPbxzEpwiqkG/XYzLH0
  185. habTcDcILo18n+o3jvem2KWBrDhyairjIDscwQIDAQABAoIBAEBSjVFqtbsp0byR
  186. aXvyrtLX1Ng7h++at2jca85Ihq//jyqbHTje8zPuNAKI6eNbmb0YGr5OuEa4pD9N
  187. ssDmMsKSoG/lRwwcm7h4InkSvBWpFShvMgUaohfHAHzsBYxfnh+TfULsi0y7c2n6
  188. t/2OZcOTRkkUDIITnXYiw93ibHHv2Mv2bBDu35kGrcK+c2dN5IL5ZjTjMRpbJTe2
  189. 44RBJbdTxHBVSgoGBnugF+s2aEma6Ehsj70oyfoVpM6Aed5kGge0A5zA1JO7WCn9
  190. Ay/DzlULRXHjJIoRWd2NKvx5n3FNppUc9vJh2plRHalRooZ2+MjSf8HmXlvG2Hpb
  191. ScvmWgECgYEA1G+A/2KnxWsr/7uWIJ7ClcGCiNLdk17Pv3DZ3G4qUsU2ITftfIbb
  192. tU0Q/b19na1IY8Pjy9ptP7t74/hF5kky97cf1FA8F+nMj/k4+wO8QDI8OJfzVzh9
  193. PwielA5vbE+xmvis5Hdp8/od1Yrc/rPSy2TKtPFhvsqXjqoUmOAjDP8CgYEAwZjH
  194. 9dt1sc2lx/rMxihlWEzQ3JPswKW9/LJAmbRBoSWF9FGNjbX7uhWtXRKJkzb8ZAwa
  195. 88azluNo2oftbDD/+jw8b2cDgaJHlLAkSD4O1D1RthW7/LKD15qZ/oFsRb13NV85
  196. ZNKtwslXGbfVNyGKUVFm7fVA8vBAOUey+LKDFj8CgYEAg8WWstOzVdYguMTXXuyb
  197. ruEV42FJaDyLiSirOvxq7GTAKuLSQUg1yMRBIeQEo2X1XU0JZE3dLodRVhuO4EXP
  198. g7Dn4X7Th9HSvgvNuIacowWGLWSz4Qp9RjhGhXhezUSx2nseY6le46PmFavJYYSR
  199. 4PBofMyt4PcyA6Cknh+KHmkCgYEAnTriG7ETE0a7v4DXUpB4TpCEiMCy5Xs2o8Z5
  200. ZNva+W+qLVUWq+MDAIyechqeFSvxK6gRM69LJ96lx+XhU58wJiFJzAhT9rK/g+jS
  201. bsHH9WOfu0xHkuHA5hgvvV2Le9B2wqgFyva4HJy82qxMxCu/VG/SMqyfBS9OWbb7
  202. ibQhdq0CgYAl53LUWZsFSZIth1vux2LVOsI8C3X1oiXDGpnrdlQ+K7z57hq5EsRq
  203. GC+INxwXbvKNqp5h0z2MvmKYPDlGVTgw8f8JjM7TkN17ERLcydhdRrMONUryZpo8
  204. 1xTob+8blyJgfxZUIAKbMbMbIiU0WAF0rfD/eJJwS4htOW/Hfv4TGA==
  205. -----END RSA PRIVATE KEY-----`
  206. // client 2 crt is revoked
  207. client2Crt = `-----BEGIN CERTIFICATE-----
  208. MIIEITCCAgmgAwIBAgIRAL6XTgNsdYzILd8bT2RVd/4wDQYJKoZIhvcNAQELBQAw
  209. EzERMA8GA1UEAxMIQ2VydEF1dGgwHhcNMjEwMTAyMjEyMzIwWhcNMjIwNzAyMjEz
  210. MDUxWjASMRAwDgYDVQQDEwdjbGllbnQyMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8A
  211. MIIBCgKCAQEA6xjW5KQR3/OFQtV5M75WINqQ4AzXSu6DhSz/yumaaQZP/UxY+6hi
  212. jcrFzGo9MMie/Sza8DhkXOFAl2BelUubrOeB2cl+/Gr8OCyRi2Gv6j3zCsuN/4jQ
  213. tNaoez/IbkDvI3l/ZpzBtnuNY2RiemGgHuORXHRVf3qVlsw+npBIRW5rM2HkO/xG
  214. oZjeBErWVu390Lyn+Gvk2TqQDnkutWnxUC60/zPlHhXZ4BwaFAekbSnjsSDB1YFM
  215. s8HwW4oBryoxdj3/+/qLrBHt75IdLw3T7/V1UDJQM3EvSQOr12w4egpldhtsC871
  216. nnBQZeY6qA5feffIwwg/6lJm70o6S6OX6wIDAQABo3EwbzAOBgNVHQ8BAf8EBAMC
  217. A7gwHQYDVR0lBBYwFAYIKwYBBQUHAwEGCCsGAQUFBwMCMB0GA1UdDgQWBBTB84v5
  218. t9HqhLhMODbn6oYkEQt3KzAfBgNVHSMEGDAWgBTtcgjQBq/80EySIvJcN8OTorgb
  219. zDANBgkqhkiG9w0BAQsFAAOCAgEALGtBCve5k8tToL3oLuXp/oSik6ovIB/zq4I/
  220. 4zNMYPU31+ZWz6aahysgx1JL1yqTa3Qm8o2tu52MbnV10dM7CIw7c/cYa+c+OPcG
  221. 5LF97kp13X+r2axy+CmwM86b4ILaDGs2Qyai6VB6k7oFUve+av5o7aUrNFpqGCJz
  222. HWdtHZSVA3JMATzy0TfWanwkzreqfdw7qH0yZ9bDURlBKAVWrqnCstva9jRuv+AI
  223. eqxr/4Ro986TFjJdoAP3Vr16CPg7/B6GA/KmsBWJrpeJdPWq4i2gpLKvYZoy89qD
  224. mUZf34RbzcCtV4NvV1DadGnt4us0nvLrvS5rL2+2uWD09kZYq9RbLkvgzF/cY0fz
  225. i7I1bi5XQ+alWe0uAk5ZZL/D+GTRYUX1AWwCqwJxmHrMxcskMyO9pXvLyuSWRDLo
  226. YNBrbX9nLcfJzVCp+X+9sntTHjs4l6Cw+fLepJIgtgqdCHtbhTiv68vSM6cgb4br
  227. 6n2xrXRKuioiWFOrTSRr+oalZh8dGJ/xvwY8IbWknZAvml9mf1VvfE7Ma5P777QM
  228. fsbYVTq0Y3R/5hIWsC3HA5z6MIM8L1oRe/YyhP3CTmrCHkVKyDOosGXpGz+JVcyo
  229. cfYkY5A3yFKB2HaCwZSfwFmRhxkrYWGEbHv3Cd9YkZs1J3hNhGFZyVMC9Uh0S85a
  230. 6zdDidU=
  231. -----END CERTIFICATE-----`
  232. client2Key = `-----BEGIN RSA PRIVATE KEY-----
  233. MIIEpAIBAAKCAQEA6xjW5KQR3/OFQtV5M75WINqQ4AzXSu6DhSz/yumaaQZP/UxY
  234. +6hijcrFzGo9MMie/Sza8DhkXOFAl2BelUubrOeB2cl+/Gr8OCyRi2Gv6j3zCsuN
  235. /4jQtNaoez/IbkDvI3l/ZpzBtnuNY2RiemGgHuORXHRVf3qVlsw+npBIRW5rM2Hk
  236. O/xGoZjeBErWVu390Lyn+Gvk2TqQDnkutWnxUC60/zPlHhXZ4BwaFAekbSnjsSDB
  237. 1YFMs8HwW4oBryoxdj3/+/qLrBHt75IdLw3T7/V1UDJQM3EvSQOr12w4egpldhts
  238. C871nnBQZeY6qA5feffIwwg/6lJm70o6S6OX6wIDAQABAoIBAFatstVb1KdQXsq0
  239. cFpui8zTKOUiduJOrDkWzTygAmlEhYtrccdfXu7OWz0x0lvBLDVGK3a0I/TGrAzj
  240. 4BuFY+FM/egxTVt9in6fmA3et4BS1OAfCryzUdfK6RV//8L+t+zJZ/qKQzWnugpy
  241. QYjDo8ifuMFwtvEoXizaIyBNLAhEp9hnrv+Tyi2O2gahPvCHsD48zkyZRCHYRstD
  242. NH5cIrwz9/RJgPO1KI+QsJE7Nh7stR0sbr+5TPU4fnsL2mNhMUF2TJrwIPrc1yp+
  243. YIUjdnh3SO88j4TQT3CIrWi8i4pOy6N0dcVn3gpCRGaqAKyS2ZYUj+yVtLO4KwxZ
  244. SZ1lNvECgYEA78BrF7f4ETfWSLcBQ3qxfLs7ibB6IYo2x25685FhZjD+zLXM1AKb
  245. FJHEXUm3mUYrFJK6AFEyOQnyGKBOLs3S6oTAswMPbTkkZeD1Y9O6uv0AHASLZnK6
  246. pC6ub0eSRF5LUyTQ55Jj8D7QsjXJueO8v+G5ihWhNSN9tB2UA+8NBmkCgYEA+weq
  247. cvoeMIEMBQHnNNLy35bwfqrceGyPIRBcUIvzQfY1vk7KW6DYOUzC7u+WUzy/hA52
  248. DjXVVhua2eMQ9qqtOav7djcMc2W9RbLowxvno7K5qiCss013MeWk64TCWy+WMp5A
  249. AVAtOliC3hMkIKqvR2poqn+IBTh1449agUJQqTMCgYEAu06IHGq1GraV6g9XpGF5
  250. wqoAlMzUTdnOfDabRilBf/YtSr+J++ThRcuwLvXFw7CnPZZ4TIEjDJ7xjj3HdxeE
  251. fYYjineMmNd40UNUU556F1ZLvJfsVKizmkuCKhwvcMx+asGrmA+tlmds4p3VMS50
  252. KzDtpKzLWlmU/p/RINWlRmkCgYBy0pHTn7aZZx2xWKqCDg+L2EXPGqZX6wgZDpu7
  253. OBifzlfM4ctL2CmvI/5yPmLbVgkgBWFYpKUdiujsyyEiQvWTUKhn7UwjqKDHtcsk
  254. G6p7xS+JswJrzX4885bZJ9Oi1AR2yM3sC9l0O7I4lDbNPmWIXBLeEhGMmcPKv/Kc
  255. 91Ff4wKBgQCF3ur+Vt0PSU0ucrPVHjCe7tqazm0LJaWbPXL1Aw0pzdM2EcNcW/MA
  256. w0kqpr7MgJ94qhXCBcVcfPuFN9fBOadM3UBj1B45Cz3pptoK+ScI8XKno6jvVK/p
  257. xr5cb9VBRBtB9aOKVfuRhpatAfS2Pzm2Htae9lFn7slGPUmu2hkjDw==
  258. -----END RSA PRIVATE KEY-----`
  259. )
  260. func TestShouldBind(t *testing.T) {
  261. c := Conf{
  262. Bindings: []Binding{
  263. {
  264. Port: 10000,
  265. },
  266. },
  267. }
  268. require.True(t, c.ShouldBind())
  269. c.Bindings[0].Port = 0
  270. require.False(t, c.ShouldBind())
  271. if runtime.GOOS != osWindows {
  272. c.Bindings[0].Address = "/absolute/path"
  273. require.True(t, c.ShouldBind())
  274. }
  275. }
  276. func TestGetRespStatus(t *testing.T) {
  277. var err error
  278. err = &dataprovider.MethodDisabledError{}
  279. respStatus := getRespStatus(err)
  280. assert.Equal(t, http.StatusForbidden, respStatus)
  281. err = fmt.Errorf("generic error")
  282. respStatus = getRespStatus(err)
  283. assert.Equal(t, http.StatusInternalServerError, respStatus)
  284. }
  285. func TestGCSWebInvalidFormFile(t *testing.T) {
  286. form := make(url.Values)
  287. form.Set("username", "test_username")
  288. form.Set("fs_provider", "2")
  289. req, _ := http.NewRequest(http.MethodPost, webUserPath, strings.NewReader(form.Encode()))
  290. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  291. err := req.ParseForm()
  292. assert.NoError(t, err)
  293. _, err = getFsConfigFromUserPostFields(req)
  294. assert.EqualError(t, err, http.ErrNotMultipart.Error())
  295. }
  296. func TestInvalidToken(t *testing.T) {
  297. admin := dataprovider.Admin{
  298. Username: "admin",
  299. }
  300. errFake := errors.New("fake error")
  301. asJSON, err := json.Marshal(admin)
  302. assert.NoError(t, err)
  303. req, _ := http.NewRequest(http.MethodPut, path.Join(adminPath, admin.Username), bytes.NewBuffer(asJSON))
  304. rctx := chi.NewRouteContext()
  305. rctx.URLParams.Add("username", admin.Username)
  306. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  307. req = req.WithContext(context.WithValue(req.Context(), jwtauth.ErrorCtxKey, errFake))
  308. rr := httptest.NewRecorder()
  309. updateAdmin(rr, req)
  310. assert.Equal(t, http.StatusBadRequest, rr.Code)
  311. rr = httptest.NewRecorder()
  312. deleteAdmin(rr, req)
  313. assert.Equal(t, http.StatusBadRequest, rr.Code)
  314. adminPwd := adminPwd{
  315. CurrentPassword: "old",
  316. NewPassword: "new",
  317. }
  318. asJSON, err = json.Marshal(adminPwd)
  319. assert.NoError(t, err)
  320. req, _ = http.NewRequest(http.MethodPut, "", bytes.NewBuffer(asJSON))
  321. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  322. req = req.WithContext(context.WithValue(req.Context(), jwtauth.ErrorCtxKey, errFake))
  323. rr = httptest.NewRecorder()
  324. changeAdminPassword(rr, req)
  325. assert.Equal(t, http.StatusInternalServerError, rr.Code)
  326. adm := getAdminFromToken(req)
  327. assert.Empty(t, adm.Username)
  328. }
  329. func TestUpdateWebAdminInvalidClaims(t *testing.T) {
  330. server := httpdServer{}
  331. server.initializeRouter()
  332. rr := httptest.NewRecorder()
  333. admin := dataprovider.Admin{
  334. Username: "",
  335. Password: "password",
  336. }
  337. c := jwtTokenClaims{
  338. Username: admin.Username,
  339. Permissions: admin.Permissions,
  340. Signature: admin.GetSignature(),
  341. }
  342. token, err := c.createTokenResponse(server.tokenAuth)
  343. assert.NoError(t, err)
  344. form := make(url.Values)
  345. form.Set("status", "1")
  346. req, _ := http.NewRequest(http.MethodPost, path.Join(webAdminPath, "admin"), bytes.NewBuffer([]byte(form.Encode())))
  347. rctx := chi.NewRouteContext()
  348. rctx.URLParams.Add("username", "admin")
  349. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  350. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  351. req.Header.Set("Cookie", fmt.Sprintf("jwt=%v", token["access_token"]))
  352. handleWebUpdateAdminPost(rr, req)
  353. assert.Equal(t, http.StatusOK, rr.Code)
  354. assert.Contains(t, rr.Body.String(), "Invalid token claims")
  355. }
  356. func TestCreateTokenError(t *testing.T) {
  357. server := httpdServer{
  358. tokenAuth: jwtauth.New("PS256", utils.GenerateRandomBytes(32), nil),
  359. }
  360. rr := httptest.NewRecorder()
  361. admin := dataprovider.Admin{
  362. Username: "admin",
  363. Password: "password",
  364. }
  365. req, _ := http.NewRequest(http.MethodGet, tokenPath, nil)
  366. server.checkAddrAndSendToken(rr, req, admin)
  367. assert.Equal(t, http.StatusInternalServerError, rr.Code)
  368. rr = httptest.NewRecorder()
  369. form := make(url.Values)
  370. form.Set("username", admin.Username)
  371. form.Set("password", admin.Password)
  372. req, _ = http.NewRequest(http.MethodPost, webLoginPath, bytes.NewBuffer([]byte(form.Encode())))
  373. req.RemoteAddr = "127.0.0.1:1234"
  374. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  375. server.handleWebLoginPost(rr, req)
  376. assert.Equal(t, http.StatusOK, rr.Code, rr.Body.String())
  377. // req with no content type
  378. req, _ = http.NewRequest(http.MethodPost, webLoginPath, nil)
  379. rr = httptest.NewRecorder()
  380. server.handleWebLoginPost(rr, req)
  381. assert.Equal(t, http.StatusOK, rr.Code, rr.Body.String())
  382. // req with no POST body
  383. req, _ = http.NewRequest(http.MethodGet, webLoginPath+"?a=a%C3%AO%GG", nil)
  384. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  385. rr = httptest.NewRecorder()
  386. server.handleWebLoginPost(rr, req)
  387. assert.Equal(t, http.StatusOK, rr.Code, rr.Body.String())
  388. req, _ = http.NewRequest(http.MethodGet, webLoginPath+"?a=a%C3%A1%G2", nil)
  389. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  390. rr = httptest.NewRecorder()
  391. handleWebAdminChangePwdPost(rr, req)
  392. assert.Equal(t, http.StatusOK, rr.Code, rr.Body.String())
  393. req, _ = http.NewRequest(http.MethodGet, webLoginPath+"?a=a%C3%A2%G3", nil)
  394. req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  395. _, err := getAdminFromPostFields(req)
  396. assert.Error(t, err)
  397. }
  398. func TestJWTTokenValidation(t *testing.T) {
  399. tokenAuth := jwtauth.New("HS256", utils.GenerateRandomBytes(32), nil)
  400. claims := make(map[string]interface{})
  401. claims["username"] = "admin"
  402. claims[jwt.ExpirationKey] = time.Now().UTC().Add(-1 * time.Hour)
  403. token, _, err := tokenAuth.Encode(claims)
  404. assert.NoError(t, err)
  405. r := GetHTTPRouter()
  406. fn := jwtAuthenticator(r)
  407. rr := httptest.NewRecorder()
  408. req, _ := http.NewRequest(http.MethodGet, userPath, nil)
  409. ctx := jwtauth.NewContext(req.Context(), token, nil)
  410. fn.ServeHTTP(rr, req.WithContext(ctx))
  411. assert.Equal(t, http.StatusUnauthorized, rr.Code)
  412. fn = jwtAuthenticatorWeb(r)
  413. rr = httptest.NewRecorder()
  414. req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
  415. ctx = jwtauth.NewContext(req.Context(), token, nil)
  416. fn.ServeHTTP(rr, req.WithContext(ctx))
  417. assert.Equal(t, http.StatusFound, rr.Code)
  418. errTest := errors.New("test error")
  419. permFn := checkPerm(dataprovider.PermAdminAny)
  420. fn = permFn(r)
  421. rr = httptest.NewRecorder()
  422. req, _ = http.NewRequest(http.MethodGet, userPath, nil)
  423. ctx = jwtauth.NewContext(req.Context(), token, errTest)
  424. fn.ServeHTTP(rr, req.WithContext(ctx))
  425. assert.Equal(t, http.StatusBadRequest, rr.Code)
  426. permFn = checkPerm(dataprovider.PermAdminAny)
  427. fn = permFn(r)
  428. rr = httptest.NewRecorder()
  429. req, _ = http.NewRequest(http.MethodGet, webUserPath, nil)
  430. req.RequestURI = webUserPath
  431. ctx = jwtauth.NewContext(req.Context(), token, errTest)
  432. fn.ServeHTTP(rr, req.WithContext(ctx))
  433. assert.Equal(t, http.StatusBadRequest, rr.Code)
  434. }
  435. func TestAdminAllowListConnAddr(t *testing.T) {
  436. server := httpdServer{}
  437. admin := dataprovider.Admin{
  438. Filters: dataprovider.AdminFilters{
  439. AllowList: []string{"192.168.1.0/24"},
  440. },
  441. }
  442. rr := httptest.NewRecorder()
  443. req, _ := http.NewRequest(http.MethodGet, tokenPath, nil)
  444. ctx := context.WithValue(req.Context(), connAddrKey, "127.0.0.1:4567")
  445. req.RemoteAddr = "192.168.1.16:1234"
  446. server.checkAddrAndSendToken(rr, req.WithContext(ctx), admin)
  447. assert.Equal(t, http.StatusForbidden, rr.Code, rr.Body.String())
  448. }
  449. func TestUpdateContextFromCookie(t *testing.T) {
  450. server := httpdServer{
  451. tokenAuth: jwtauth.New("HS256", utils.GenerateRandomBytes(32), nil),
  452. }
  453. req, _ := http.NewRequest(http.MethodGet, tokenPath, nil)
  454. claims := make(map[string]interface{})
  455. claims["a"] = "b"
  456. token, _, err := server.tokenAuth.Encode(claims)
  457. assert.NoError(t, err)
  458. ctx := jwtauth.NewContext(req.Context(), token, nil)
  459. server.updateContextFromCookie(req.WithContext(ctx))
  460. }
  461. func TestCookieExpiration(t *testing.T) {
  462. server := httpdServer{
  463. tokenAuth: jwtauth.New("HS256", utils.GenerateRandomBytes(32), nil),
  464. }
  465. err := errors.New("test error")
  466. rr := httptest.NewRecorder()
  467. req, _ := http.NewRequest(http.MethodGet, tokenPath, nil)
  468. ctx := jwtauth.NewContext(req.Context(), nil, err)
  469. server.checkCookieExpiration(rr, req.WithContext(ctx))
  470. cookie := rr.Header().Get("Set-Cookie")
  471. assert.Empty(t, cookie)
  472. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  473. claims := make(map[string]interface{})
  474. claims["a"] = "b"
  475. token, _, err := server.tokenAuth.Encode(claims)
  476. assert.NoError(t, err)
  477. ctx = jwtauth.NewContext(req.Context(), token, nil)
  478. server.checkCookieExpiration(rr, req.WithContext(ctx))
  479. cookie = rr.Header().Get("Set-Cookie")
  480. assert.Empty(t, cookie)
  481. admin := dataprovider.Admin{
  482. Username: "newtestadmin",
  483. Password: "password",
  484. Permissions: []string{dataprovider.PermAdminAny},
  485. }
  486. claims = make(map[string]interface{})
  487. claims[claimUsernameKey] = admin.Username
  488. claims[claimPermissionsKey] = admin.Permissions
  489. claims[jwt.SubjectKey] = admin.GetSignature()
  490. claims[jwt.ExpirationKey] = time.Now().Add(1 * time.Minute)
  491. token, _, err = server.tokenAuth.Encode(claims)
  492. assert.NoError(t, err)
  493. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  494. ctx = jwtauth.NewContext(req.Context(), token, nil)
  495. server.checkCookieExpiration(rr, req.WithContext(ctx))
  496. cookie = rr.Header().Get("Set-Cookie")
  497. assert.Empty(t, cookie)
  498. admin.Status = 0
  499. err = dataprovider.AddAdmin(&admin)
  500. assert.NoError(t, err)
  501. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  502. ctx = jwtauth.NewContext(req.Context(), token, nil)
  503. server.checkCookieExpiration(rr, req.WithContext(ctx))
  504. cookie = rr.Header().Get("Set-Cookie")
  505. assert.Empty(t, cookie)
  506. admin.Status = 1
  507. admin.Filters.AllowList = []string{"172.16.1.0/24"}
  508. err = dataprovider.UpdateAdmin(&admin)
  509. assert.NoError(t, err)
  510. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  511. ctx = jwtauth.NewContext(req.Context(), token, nil)
  512. server.checkCookieExpiration(rr, req.WithContext(ctx))
  513. cookie = rr.Header().Get("Set-Cookie")
  514. assert.Empty(t, cookie)
  515. admin, err = dataprovider.AdminExists(admin.Username)
  516. assert.NoError(t, err)
  517. claims = make(map[string]interface{})
  518. claims[claimUsernameKey] = admin.Username
  519. claims[claimPermissionsKey] = admin.Permissions
  520. claims[jwt.SubjectKey] = admin.GetSignature()
  521. claims[jwt.ExpirationKey] = time.Now().Add(1 * time.Minute)
  522. token, _, err = server.tokenAuth.Encode(claims)
  523. assert.NoError(t, err)
  524. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  525. req.RemoteAddr = "192.168.8.1:1234"
  526. ctx = jwtauth.NewContext(req.Context(), token, nil)
  527. server.checkCookieExpiration(rr, req.WithContext(ctx))
  528. cookie = rr.Header().Get("Set-Cookie")
  529. assert.Empty(t, cookie)
  530. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  531. req.RemoteAddr = "172.16.1.2:1234"
  532. ctx = jwtauth.NewContext(req.Context(), token, nil)
  533. ctx = context.WithValue(ctx, connAddrKey, "10.9.9.9")
  534. server.checkCookieExpiration(rr, req.WithContext(ctx))
  535. cookie = rr.Header().Get("Set-Cookie")
  536. assert.Empty(t, cookie)
  537. req, _ = http.NewRequest(http.MethodGet, tokenPath, nil)
  538. req.RemoteAddr = "172.16.1.12:4567"
  539. ctx = jwtauth.NewContext(req.Context(), token, nil)
  540. server.checkCookieExpiration(rr, req.WithContext(ctx))
  541. cookie = rr.Header().Get("Set-Cookie")
  542. assert.True(t, strings.HasPrefix(cookie, "jwt="))
  543. err = dataprovider.DeleteAdmin(admin.Username)
  544. assert.NoError(t, err)
  545. }
  546. func TestGetURLParam(t *testing.T) {
  547. req, _ := http.NewRequest(http.MethodGet, adminPwdPath, nil)
  548. rctx := chi.NewRouteContext()
  549. rctx.URLParams.Add("val", "testuser%C3%A0")
  550. rctx.URLParams.Add("inval", "testuser%C3%AO%GG")
  551. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  552. escaped := getURLParam(req, "val")
  553. assert.Equal(t, "testuserà", escaped)
  554. escaped = getURLParam(req, "inval")
  555. assert.Equal(t, "testuser%C3%AO%GG", escaped)
  556. }
  557. func TestChangePwdValidationErrors(t *testing.T) {
  558. err := doChangeAdminPassword(nil, "", "", "")
  559. require.Error(t, err)
  560. err = doChangeAdminPassword(nil, "a", "b", "c")
  561. require.Error(t, err)
  562. err = doChangeAdminPassword(nil, "a", "a", "a")
  563. require.Error(t, err)
  564. req, _ := http.NewRequest(http.MethodPut, adminPwdPath, nil)
  565. err = doChangeAdminPassword(req, "currentpwd", "newpwd", "newpwd")
  566. assert.Error(t, err)
  567. }
  568. func TestRenderUnexistingFolder(t *testing.T) {
  569. rr := httptest.NewRecorder()
  570. req, _ := http.NewRequest(http.MethodPost, folderPath, nil)
  571. renderFolder(rr, req, "path not mapped")
  572. assert.Equal(t, http.StatusNotFound, rr.Code)
  573. }
  574. func TestCloseConnectionHandler(t *testing.T) {
  575. req, _ := http.NewRequest(http.MethodDelete, activeConnectionsPath+"/connectionID", nil)
  576. rctx := chi.NewRouteContext()
  577. rctx.URLParams.Add("connectionID", "")
  578. req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  579. rr := httptest.NewRecorder()
  580. handleCloseConnection(rr, req)
  581. assert.Equal(t, http.StatusBadRequest, rr.Code)
  582. }
  583. func TestRenderInvalidTemplate(t *testing.T) {
  584. tmpl, err := template.New("test").Parse("{{.Count}}")
  585. if assert.NoError(t, err) {
  586. templates["no_match"] = tmpl
  587. rw := httptest.NewRecorder()
  588. renderTemplate(rw, "no_match", map[string]string{})
  589. assert.Equal(t, http.StatusInternalServerError, rw.Code)
  590. }
  591. }
  592. func TestQuotaScanInvalidFs(t *testing.T) {
  593. user := dataprovider.User{
  594. Username: "test",
  595. HomeDir: os.TempDir(),
  596. FsConfig: dataprovider.Filesystem{
  597. Provider: dataprovider.S3FilesystemProvider,
  598. },
  599. }
  600. common.QuotaScans.AddUserQuotaScan(user.Username)
  601. err := doQuotaScan(user)
  602. assert.Error(t, err)
  603. }
  604. func TestVerifyTLSConnection(t *testing.T) {
  605. oldCertMgr := certMgr
  606. caCrlPath := filepath.Join(os.TempDir(), "testcrl.crt")
  607. certPath := filepath.Join(os.TempDir(), "testh.crt")
  608. keyPath := filepath.Join(os.TempDir(), "testh.key")
  609. err := ioutil.WriteFile(caCrlPath, []byte(caCRL), os.ModePerm)
  610. assert.NoError(t, err)
  611. err = ioutil.WriteFile(certPath, []byte(httpdCert), os.ModePerm)
  612. assert.NoError(t, err)
  613. err = ioutil.WriteFile(keyPath, []byte(httpdKey), os.ModePerm)
  614. assert.NoError(t, err)
  615. certMgr, err = common.NewCertManager(certPath, keyPath, "", "webdav_test")
  616. assert.NoError(t, err)
  617. certMgr.SetCARevocationLists([]string{caCrlPath})
  618. err = certMgr.LoadCRLs()
  619. assert.NoError(t, err)
  620. crt, err := tls.X509KeyPair([]byte(client1Crt), []byte(client1Key))
  621. assert.NoError(t, err)
  622. x509crt, err := x509.ParseCertificate(crt.Certificate[0])
  623. assert.NoError(t, err)
  624. server := httpdServer{}
  625. state := tls.ConnectionState{
  626. PeerCertificates: []*x509.Certificate{x509crt},
  627. }
  628. err = server.verifyTLSConnection(state)
  629. assert.Error(t, err) // no verified certification chain
  630. crt, err = tls.X509KeyPair([]byte(caCRT), []byte(caKey))
  631. assert.NoError(t, err)
  632. x509CAcrt, err := x509.ParseCertificate(crt.Certificate[0])
  633. assert.NoError(t, err)
  634. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crt, x509CAcrt})
  635. err = server.verifyTLSConnection(state)
  636. assert.NoError(t, err)
  637. crt, err = tls.X509KeyPair([]byte(client2Crt), []byte(client2Key))
  638. assert.NoError(t, err)
  639. x509crtRevoked, err := x509.ParseCertificate(crt.Certificate[0])
  640. assert.NoError(t, err)
  641. state.VerifiedChains = append(state.VerifiedChains, []*x509.Certificate{x509crtRevoked, x509CAcrt})
  642. state.PeerCertificates = []*x509.Certificate{x509crtRevoked}
  643. err = server.verifyTLSConnection(state)
  644. assert.EqualError(t, err, common.ErrCrtRevoked.Error())
  645. err = os.Remove(caCrlPath)
  646. assert.NoError(t, err)
  647. err = os.Remove(certPath)
  648. assert.NoError(t, err)
  649. err = os.Remove(keyPath)
  650. assert.NoError(t, err)
  651. certMgr = oldCertMgr
  652. }
  653. func TestGetUserFromTemplate(t *testing.T) {
  654. user := dataprovider.User{
  655. Status: 1,
  656. }
  657. user.VirtualFolders = append(user.VirtualFolders, vfs.VirtualFolder{
  658. BaseVirtualFolder: vfs.BaseVirtualFolder{
  659. MappedPath: "dir%username%",
  660. },
  661. })
  662. username := "userTemplate"
  663. password := "pwdTemplate"
  664. templateFields := userTemplateFields{
  665. Username: username,
  666. Password: password,
  667. }
  668. userTemplate := getUserFromTemplate(user, templateFields)
  669. require.Len(t, userTemplate.VirtualFolders, 1)
  670. require.Equal(t, "dir"+username, userTemplate.VirtualFolders[0].MappedPath)
  671. user.FsConfig.Provider = dataprovider.CryptedFilesystemProvider
  672. user.FsConfig.CryptConfig.Passphrase = kms.NewPlainSecret("%password%")
  673. userTemplate = getUserFromTemplate(user, templateFields)
  674. require.Equal(t, password, userTemplate.FsConfig.CryptConfig.Passphrase.GetPayload())
  675. user.FsConfig.Provider = dataprovider.GCSFilesystemProvider
  676. user.FsConfig.GCSConfig.KeyPrefix = "%username%%password%"
  677. userTemplate = getUserFromTemplate(user, templateFields)
  678. require.Equal(t, username+password, userTemplate.FsConfig.GCSConfig.KeyPrefix)
  679. user.FsConfig.Provider = dataprovider.AzureBlobFilesystemProvider
  680. user.FsConfig.AzBlobConfig.KeyPrefix = "a%username%"
  681. user.FsConfig.AzBlobConfig.AccountKey = kms.NewPlainSecret("pwd%password%%username%")
  682. userTemplate = getUserFromTemplate(user, templateFields)
  683. require.Equal(t, "a"+username, userTemplate.FsConfig.AzBlobConfig.KeyPrefix)
  684. require.Equal(t, "pwd"+password+username, userTemplate.FsConfig.AzBlobConfig.AccountKey.GetPayload())
  685. user.FsConfig.Provider = dataprovider.SFTPFilesystemProvider
  686. user.FsConfig.SFTPConfig.Prefix = "%username%"
  687. user.FsConfig.SFTPConfig.Username = "sftp_%username%"
  688. user.FsConfig.SFTPConfig.Password = kms.NewPlainSecret("sftp%password%")
  689. userTemplate = getUserFromTemplate(user, templateFields)
  690. require.Equal(t, username, userTemplate.FsConfig.SFTPConfig.Prefix)
  691. require.Equal(t, "sftp_"+username, userTemplate.FsConfig.SFTPConfig.Username)
  692. require.Equal(t, "sftp"+password, userTemplate.FsConfig.SFTPConfig.Password.GetPayload())
  693. }
  694. func TestJWTTokenCleanup(t *testing.T) {
  695. server := httpdServer{
  696. tokenAuth: jwtauth.New("HS256", utils.GenerateRandomBytes(32), nil),
  697. }
  698. admin := dataprovider.Admin{
  699. Username: "newtestadmin",
  700. Password: "password",
  701. Permissions: []string{dataprovider.PermAdminAny},
  702. }
  703. claims := make(map[string]interface{})
  704. claims[claimUsernameKey] = admin.Username
  705. claims[claimPermissionsKey] = admin.Permissions
  706. claims[jwt.SubjectKey] = admin.GetSignature()
  707. claims[jwt.ExpirationKey] = time.Now().Add(1 * time.Minute)
  708. _, token, err := server.tokenAuth.Encode(claims)
  709. assert.NoError(t, err)
  710. req, _ := http.NewRequest(http.MethodGet, versionPath, nil)
  711. assert.True(t, isTokenInvalidated(req))
  712. req.Header.Set("Authorization", fmt.Sprintf("Bearer %v", token))
  713. invalidatedJWTTokens.Store(token, time.Now().UTC().Add(-tokenDuration))
  714. require.True(t, isTokenInvalidated(req))
  715. startJWTTokensCleanupTicker(100 * time.Millisecond)
  716. assert.Eventually(t, func() bool { return !isTokenInvalidated(req) }, 1*time.Second, 200*time.Millisecond)
  717. stopJWTTokensCleanupTicker()
  718. }