webclient.go 74 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233
  1. // Copyright (C) 2019 Nicola Murino
  2. //
  3. // This program is free software: you can redistribute it and/or modify
  4. // it under the terms of the GNU Affero General Public License as published
  5. // by the Free Software Foundation, version 3.
  6. //
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU Affero General Public License for more details.
  11. //
  12. // You should have received a copy of the GNU Affero General Public License
  13. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  14. package httpd
  15. import (
  16. "bytes"
  17. "encoding/json"
  18. "errors"
  19. "fmt"
  20. "html/template"
  21. "io"
  22. "math"
  23. "net/http"
  24. "net/url"
  25. "os"
  26. "path"
  27. "path/filepath"
  28. "strconv"
  29. "strings"
  30. "time"
  31. "github.com/go-chi/render"
  32. "github.com/rs/xid"
  33. "github.com/sftpgo/sdk"
  34. "github.com/drakkan/sftpgo/v2/internal/common"
  35. "github.com/drakkan/sftpgo/v2/internal/dataprovider"
  36. "github.com/drakkan/sftpgo/v2/internal/logger"
  37. "github.com/drakkan/sftpgo/v2/internal/mfa"
  38. "github.com/drakkan/sftpgo/v2/internal/smtp"
  39. "github.com/drakkan/sftpgo/v2/internal/util"
  40. "github.com/drakkan/sftpgo/v2/internal/vfs"
  41. )
  42. const (
  43. templateClientDir = "webclient"
  44. templateClientBase = "base.html"
  45. templateClientFiles = "files.html"
  46. templateClientProfile = "profile.html"
  47. templateClientMFA = "mfa.html"
  48. templateClientEditFile = "editfile.html"
  49. templateClientShare = "share.html"
  50. templateClientShares = "shares.html"
  51. templateClientViewPDF = "viewpdf.html"
  52. templateShareLogin = "sharelogin.html"
  53. templateShareDownload = "sharedownload.html"
  54. templateUploadToShare = "shareupload.html"
  55. )
  56. // condResult is the result of an HTTP request precondition check.
  57. // See https://tools.ietf.org/html/rfc7232 section 3.
  58. type condResult int
  59. const (
  60. condNone condResult = iota
  61. condTrue
  62. condFalse
  63. )
  64. var (
  65. clientTemplates = make(map[string]*template.Template)
  66. unixEpochTime = time.Unix(0, 0)
  67. )
  68. // isZeroTime reports whether t is obviously unspecified (either zero or Unix()=0).
  69. func isZeroTime(t time.Time) bool {
  70. return t.IsZero() || t.Equal(unixEpochTime)
  71. }
  72. type baseClientPage struct {
  73. commonBasePage
  74. Title string
  75. CurrentURL string
  76. FilesURL string
  77. SharesURL string
  78. ShareURL string
  79. ProfileURL string
  80. PingURL string
  81. ChangePwdURL string
  82. LogoutURL string
  83. LoginURL string
  84. EditURL string
  85. MFAURL string
  86. CSRFToken string
  87. LoggedUser *dataprovider.User
  88. Branding UIBranding
  89. }
  90. type dirMapping struct {
  91. DirName string
  92. Href string
  93. }
  94. type viewPDFPage struct {
  95. commonBasePage
  96. Title string
  97. URL string
  98. Branding UIBranding
  99. }
  100. type editFilePage struct {
  101. baseClientPage
  102. CurrentDir string
  103. FileURL string
  104. Path string
  105. Name string
  106. ReadOnly bool
  107. Data string
  108. }
  109. type filesPage struct {
  110. baseClientPage
  111. CurrentDir string
  112. DirsURL string
  113. FileActionsURL string
  114. CheckExistURL string
  115. DownloadURL string
  116. ViewPDFURL string
  117. FileURL string
  118. TasksURL string
  119. CanAddFiles bool
  120. CanCreateDirs bool
  121. CanRename bool
  122. CanDelete bool
  123. CanDownload bool
  124. CanShare bool
  125. CanCopy bool
  126. ShareUploadBaseURL string
  127. Error *util.I18nError
  128. Paths []dirMapping
  129. QuotaUsage *userQuotaUsage
  130. }
  131. type shareLoginPage struct {
  132. commonBasePage
  133. CurrentURL string
  134. Error *util.I18nError
  135. CSRFToken string
  136. Title string
  137. Branding UIBranding
  138. }
  139. type shareDownloadPage struct {
  140. baseClientPage
  141. DownloadLink string
  142. }
  143. type shareUploadPage struct {
  144. baseClientPage
  145. Share *dataprovider.Share
  146. UploadBasePath string
  147. }
  148. type clientMessagePage struct {
  149. baseClientPage
  150. Error *util.I18nError
  151. Success string
  152. Text string
  153. }
  154. type clientProfilePage struct {
  155. baseClientPage
  156. PublicKeys []string
  157. TLSCerts []string
  158. CanSubmit bool
  159. AllowAPIKeyAuth bool
  160. Email string
  161. Description string
  162. Error *util.I18nError
  163. }
  164. type changeClientPasswordPage struct {
  165. baseClientPage
  166. Error *util.I18nError
  167. }
  168. type clientMFAPage struct {
  169. baseClientPage
  170. TOTPConfigs []string
  171. TOTPConfig dataprovider.UserTOTPConfig
  172. GenerateTOTPURL string
  173. ValidateTOTPURL string
  174. SaveTOTPURL string
  175. RecCodesURL string
  176. Protocols []string
  177. RequiredProtocols []string
  178. }
  179. type clientSharesPage struct {
  180. baseClientPage
  181. BasePublicSharesURL string
  182. }
  183. type clientSharePage struct {
  184. baseClientPage
  185. Share *dataprovider.Share
  186. Error *util.I18nError
  187. IsAdd bool
  188. }
  189. type userQuotaUsage struct {
  190. QuotaSize int64
  191. QuotaFiles int
  192. UsedQuotaSize int64
  193. UsedQuotaFiles int
  194. UploadDataTransfer int64
  195. DownloadDataTransfer int64
  196. TotalDataTransfer int64
  197. UsedUploadDataTransfer int64
  198. UsedDownloadDataTransfer int64
  199. }
  200. func (u *userQuotaUsage) HasQuotaInfo() bool {
  201. if dataprovider.GetQuotaTracking() == 0 {
  202. return false
  203. }
  204. if u.HasDiskQuota() {
  205. return true
  206. }
  207. return u.HasTranferQuota()
  208. }
  209. func (u *userQuotaUsage) HasDiskQuota() bool {
  210. if u.QuotaSize > 0 || u.UsedQuotaSize > 0 {
  211. return true
  212. }
  213. return u.QuotaFiles > 0 || u.UsedQuotaFiles > 0
  214. }
  215. func (u *userQuotaUsage) HasTranferQuota() bool {
  216. if u.TotalDataTransfer > 0 || u.UploadDataTransfer > 0 || u.DownloadDataTransfer > 0 {
  217. return true
  218. }
  219. return u.UsedDownloadDataTransfer > 0 || u.UsedUploadDataTransfer > 0
  220. }
  221. func (u *userQuotaUsage) GetQuotaSize() string {
  222. if u.QuotaSize > 0 {
  223. return fmt.Sprintf("%s/%s", util.ByteCountIEC(u.UsedQuotaSize), util.ByteCountIEC(u.QuotaSize))
  224. }
  225. if u.UsedQuotaSize > 0 {
  226. return util.ByteCountIEC(u.UsedQuotaSize)
  227. }
  228. return ""
  229. }
  230. func (u *userQuotaUsage) GetQuotaFiles() string {
  231. if u.QuotaFiles > 0 {
  232. return fmt.Sprintf("%d/%d", u.UsedQuotaFiles, u.QuotaFiles)
  233. }
  234. if u.UsedQuotaFiles > 0 {
  235. return strconv.FormatInt(int64(u.UsedQuotaFiles), 10)
  236. }
  237. return ""
  238. }
  239. func (u *userQuotaUsage) GetQuotaSizePercentage() int {
  240. if u.QuotaSize > 0 {
  241. return int(math.Round(100 * float64(u.UsedQuotaSize) / float64(u.QuotaSize)))
  242. }
  243. return 0
  244. }
  245. func (u *userQuotaUsage) GetQuotaFilesPercentage() int {
  246. if u.QuotaFiles > 0 {
  247. return int(math.Round(100 * float64(u.UsedQuotaFiles) / float64(u.QuotaFiles)))
  248. }
  249. return 0
  250. }
  251. func (u *userQuotaUsage) IsQuotaSizeLow() bool {
  252. return u.GetQuotaSizePercentage() > 85
  253. }
  254. func (u *userQuotaUsage) IsQuotaFilesLow() bool {
  255. return u.GetQuotaFilesPercentage() > 85
  256. }
  257. func (u *userQuotaUsage) IsDiskQuotaLow() bool {
  258. return u.IsQuotaSizeLow() || u.IsQuotaFilesLow()
  259. }
  260. func (u *userQuotaUsage) GetTotalTransferQuota() string {
  261. total := u.UsedUploadDataTransfer + u.UsedDownloadDataTransfer
  262. if u.TotalDataTransfer > 0 {
  263. return fmt.Sprintf("%s/%s", util.ByteCountIEC(total), util.ByteCountIEC(u.TotalDataTransfer*1048576))
  264. }
  265. if total > 0 {
  266. return util.ByteCountIEC(total)
  267. }
  268. return ""
  269. }
  270. func (u *userQuotaUsage) GetUploadTransferQuota() string {
  271. if u.UploadDataTransfer > 0 {
  272. return fmt.Sprintf("%s/%s", util.ByteCountIEC(u.UsedUploadDataTransfer),
  273. util.ByteCountIEC(u.UploadDataTransfer*1048576))
  274. }
  275. if u.UsedUploadDataTransfer > 0 {
  276. return util.ByteCountIEC(u.UsedUploadDataTransfer)
  277. }
  278. return ""
  279. }
  280. func (u *userQuotaUsage) GetDownloadTransferQuota() string {
  281. if u.DownloadDataTransfer > 0 {
  282. return fmt.Sprintf("%s/%s", util.ByteCountIEC(u.UsedDownloadDataTransfer),
  283. util.ByteCountIEC(u.DownloadDataTransfer*1048576))
  284. }
  285. if u.UsedDownloadDataTransfer > 0 {
  286. return util.ByteCountIEC(u.UsedDownloadDataTransfer)
  287. }
  288. return ""
  289. }
  290. func (u *userQuotaUsage) GetTotalTransferQuotaPercentage() int {
  291. if u.TotalDataTransfer > 0 {
  292. return int(math.Round(100 * float64(u.UsedDownloadDataTransfer+u.UsedUploadDataTransfer) / float64(u.TotalDataTransfer*1048576)))
  293. }
  294. return 0
  295. }
  296. func (u *userQuotaUsage) GetUploadTransferQuotaPercentage() int {
  297. if u.UploadDataTransfer > 0 {
  298. return int(math.Round(100 * float64(u.UsedUploadDataTransfer) / float64(u.UploadDataTransfer*1048576)))
  299. }
  300. return 0
  301. }
  302. func (u *userQuotaUsage) GetDownloadTransferQuotaPercentage() int {
  303. if u.DownloadDataTransfer > 0 {
  304. return int(math.Round(100 * float64(u.UsedDownloadDataTransfer) / float64(u.DownloadDataTransfer*1048576)))
  305. }
  306. return 0
  307. }
  308. func (u *userQuotaUsage) IsTotalTransferQuotaLow() bool {
  309. if u.TotalDataTransfer > 0 {
  310. return u.GetTotalTransferQuotaPercentage() > 85
  311. }
  312. return false
  313. }
  314. func (u *userQuotaUsage) IsUploadTransferQuotaLow() bool {
  315. if u.UploadDataTransfer > 0 {
  316. return u.GetUploadTransferQuotaPercentage() > 85
  317. }
  318. return false
  319. }
  320. func (u *userQuotaUsage) IsDownloadTransferQuotaLow() bool {
  321. if u.DownloadDataTransfer > 0 {
  322. return u.GetDownloadTransferQuotaPercentage() > 85
  323. }
  324. return false
  325. }
  326. func (u *userQuotaUsage) IsTransferQuotaLow() bool {
  327. return u.IsTotalTransferQuotaLow() || u.IsUploadTransferQuotaLow() || u.IsDownloadTransferQuotaLow()
  328. }
  329. func (u *userQuotaUsage) IsQuotaLow() bool {
  330. return u.IsDiskQuotaLow() || u.IsTransferQuotaLow()
  331. }
  332. func newUserQuotaUsage(u *dataprovider.User) *userQuotaUsage {
  333. return &userQuotaUsage{
  334. QuotaSize: u.QuotaSize,
  335. QuotaFiles: u.QuotaFiles,
  336. UsedQuotaSize: u.UsedQuotaSize,
  337. UsedQuotaFiles: u.UsedQuotaFiles,
  338. TotalDataTransfer: u.TotalDataTransfer,
  339. UploadDataTransfer: u.UploadDataTransfer,
  340. DownloadDataTransfer: u.DownloadDataTransfer,
  341. UsedUploadDataTransfer: u.UsedUploadDataTransfer,
  342. UsedDownloadDataTransfer: u.UsedDownloadDataTransfer,
  343. }
  344. }
  345. func getFileObjectURL(baseDir, name, baseWebPath string) string {
  346. return fmt.Sprintf("%v?path=%v&_=%v", baseWebPath, url.QueryEscape(path.Join(baseDir, name)), time.Now().UTC().Unix())
  347. }
  348. func getFileObjectModTime(t time.Time) int64 {
  349. if isZeroTime(t) {
  350. return 0
  351. }
  352. return t.UnixMilli()
  353. }
  354. func loadClientTemplates(templatesPath string) {
  355. filesPaths := []string{
  356. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  357. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  358. filepath.Join(templatesPath, templateClientDir, templateClientFiles),
  359. }
  360. editFilePath := []string{
  361. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  362. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  363. filepath.Join(templatesPath, templateClientDir, templateClientEditFile),
  364. }
  365. sharesPaths := []string{
  366. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  367. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  368. filepath.Join(templatesPath, templateClientDir, templateClientShares),
  369. }
  370. sharePaths := []string{
  371. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  372. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  373. filepath.Join(templatesPath, templateClientDir, templateClientShare),
  374. }
  375. profilePaths := []string{
  376. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  377. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  378. filepath.Join(templatesPath, templateClientDir, templateClientProfile),
  379. }
  380. changePwdPaths := []string{
  381. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  382. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  383. filepath.Join(templatesPath, templateCommonDir, templateChangePwd),
  384. }
  385. loginPaths := []string{
  386. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  387. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  388. filepath.Join(templatesPath, templateCommonDir, templateCommonLogin),
  389. }
  390. messagePaths := []string{
  391. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  392. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  393. filepath.Join(templatesPath, templateCommonDir, templateMessage),
  394. }
  395. mfaPaths := []string{
  396. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  397. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  398. filepath.Join(templatesPath, templateClientDir, templateClientMFA),
  399. }
  400. twoFactorPaths := []string{
  401. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  402. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  403. filepath.Join(templatesPath, templateCommonDir, templateTwoFactor),
  404. }
  405. twoFactorRecoveryPaths := []string{
  406. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  407. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  408. filepath.Join(templatesPath, templateCommonDir, templateTwoFactorRecovery),
  409. }
  410. forgotPwdPaths := []string{
  411. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  412. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  413. filepath.Join(templatesPath, templateCommonDir, templateForgotPassword),
  414. }
  415. resetPwdPaths := []string{
  416. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  417. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  418. filepath.Join(templatesPath, templateCommonDir, templateResetPassword),
  419. }
  420. viewPDFPaths := []string{
  421. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  422. filepath.Join(templatesPath, templateClientDir, templateClientViewPDF),
  423. }
  424. shareLoginPath := []string{
  425. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  426. filepath.Join(templatesPath, templateCommonDir, templateCommonBaseLogin),
  427. filepath.Join(templatesPath, templateClientDir, templateShareLogin),
  428. }
  429. shareUploadPath := []string{
  430. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  431. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  432. filepath.Join(templatesPath, templateClientDir, templateUploadToShare),
  433. }
  434. shareDownloadPath := []string{
  435. filepath.Join(templatesPath, templateCommonDir, templateCommonBase),
  436. filepath.Join(templatesPath, templateClientDir, templateClientBase),
  437. filepath.Join(templatesPath, templateClientDir, templateShareDownload),
  438. }
  439. filesTmpl := util.LoadTemplate(nil, filesPaths...)
  440. profileTmpl := util.LoadTemplate(nil, profilePaths...)
  441. changePwdTmpl := util.LoadTemplate(nil, changePwdPaths...)
  442. loginTmpl := util.LoadTemplate(nil, loginPaths...)
  443. messageTmpl := util.LoadTemplate(nil, messagePaths...)
  444. mfaTmpl := util.LoadTemplate(nil, mfaPaths...)
  445. twoFactorTmpl := util.LoadTemplate(nil, twoFactorPaths...)
  446. twoFactorRecoveryTmpl := util.LoadTemplate(nil, twoFactorRecoveryPaths...)
  447. editFileTmpl := util.LoadTemplate(nil, editFilePath...)
  448. shareLoginTmpl := util.LoadTemplate(nil, shareLoginPath...)
  449. sharesTmpl := util.LoadTemplate(nil, sharesPaths...)
  450. shareTmpl := util.LoadTemplate(nil, sharePaths...)
  451. forgotPwdTmpl := util.LoadTemplate(nil, forgotPwdPaths...)
  452. resetPwdTmpl := util.LoadTemplate(nil, resetPwdPaths...)
  453. viewPDFTmpl := util.LoadTemplate(nil, viewPDFPaths...)
  454. shareUploadTmpl := util.LoadTemplate(nil, shareUploadPath...)
  455. shareDownloadTmpl := util.LoadTemplate(nil, shareDownloadPath...)
  456. clientTemplates[templateClientFiles] = filesTmpl
  457. clientTemplates[templateClientProfile] = profileTmpl
  458. clientTemplates[templateChangePwd] = changePwdTmpl
  459. clientTemplates[templateCommonLogin] = loginTmpl
  460. clientTemplates[templateMessage] = messageTmpl
  461. clientTemplates[templateClientMFA] = mfaTmpl
  462. clientTemplates[templateTwoFactor] = twoFactorTmpl
  463. clientTemplates[templateTwoFactorRecovery] = twoFactorRecoveryTmpl
  464. clientTemplates[templateClientEditFile] = editFileTmpl
  465. clientTemplates[templateClientShares] = sharesTmpl
  466. clientTemplates[templateClientShare] = shareTmpl
  467. clientTemplates[templateForgotPassword] = forgotPwdTmpl
  468. clientTemplates[templateResetPassword] = resetPwdTmpl
  469. clientTemplates[templateClientViewPDF] = viewPDFTmpl
  470. clientTemplates[templateShareLogin] = shareLoginTmpl
  471. clientTemplates[templateUploadToShare] = shareUploadTmpl
  472. clientTemplates[templateShareDownload] = shareDownloadTmpl
  473. }
  474. func (s *httpdServer) getBaseClientPageData(title, currentURL string, r *http.Request) baseClientPage {
  475. var csrfToken string
  476. if currentURL != "" {
  477. csrfToken = createCSRFToken(util.GetIPFromRemoteAddress(r.RemoteAddr))
  478. }
  479. data := baseClientPage{
  480. commonBasePage: getCommonBasePage(r),
  481. Title: title,
  482. CurrentURL: currentURL,
  483. FilesURL: webClientFilesPath,
  484. SharesURL: webClientSharesPath,
  485. ShareURL: webClientSharePath,
  486. ProfileURL: webClientProfilePath,
  487. PingURL: webClientPingPath,
  488. ChangePwdURL: webChangeClientPwdPath,
  489. LogoutURL: webClientLogoutPath,
  490. EditURL: webClientEditFilePath,
  491. MFAURL: webClientMFAPath,
  492. CSRFToken: csrfToken,
  493. LoggedUser: getUserFromToken(r),
  494. Branding: s.binding.Branding.WebClient,
  495. }
  496. if !strings.HasPrefix(r.RequestURI, webClientPubSharesPath) {
  497. data.LoginURL = webClientLoginPath
  498. }
  499. return data
  500. }
  501. func (s *httpdServer) renderClientForgotPwdPage(w http.ResponseWriter, r *http.Request, err *util.I18nError, ip string) {
  502. data := forgotPwdPage{
  503. commonBasePage: getCommonBasePage(r),
  504. CurrentURL: webClientForgotPwdPath,
  505. Error: err,
  506. CSRFToken: createCSRFToken(ip),
  507. LoginURL: webClientLoginPath,
  508. Title: util.I18nForgotPwdTitle,
  509. Branding: s.binding.Branding.WebClient,
  510. }
  511. renderClientTemplate(w, templateForgotPassword, data)
  512. }
  513. func (s *httpdServer) renderClientResetPwdPage(w http.ResponseWriter, r *http.Request, err *util.I18nError, ip string) {
  514. data := resetPwdPage{
  515. commonBasePage: getCommonBasePage(r),
  516. CurrentURL: webClientResetPwdPath,
  517. Error: err,
  518. CSRFToken: createCSRFToken(ip),
  519. LoginURL: webClientLoginPath,
  520. Title: util.I18nResetPwdTitle,
  521. Branding: s.binding.Branding.WebClient,
  522. }
  523. renderClientTemplate(w, templateResetPassword, data)
  524. }
  525. func (s *httpdServer) renderShareLoginPage(w http.ResponseWriter, r *http.Request, err *util.I18nError, ip string) {
  526. data := shareLoginPage{
  527. commonBasePage: getCommonBasePage(r),
  528. Title: util.I18nShareLoginTitle,
  529. CurrentURL: r.RequestURI,
  530. Error: err,
  531. CSRFToken: createCSRFToken(ip),
  532. Branding: s.binding.Branding.WebClient,
  533. }
  534. renderClientTemplate(w, templateShareLogin, data)
  535. }
  536. func renderClientTemplate(w http.ResponseWriter, tmplName string, data any) {
  537. err := clientTemplates[tmplName].ExecuteTemplate(w, tmplName, data)
  538. if err != nil {
  539. http.Error(w, err.Error(), http.StatusInternalServerError)
  540. }
  541. }
  542. func (s *httpdServer) renderClientMessagePage(w http.ResponseWriter, r *http.Request, title string, statusCode int, err error, message string) {
  543. data := clientMessagePage{
  544. baseClientPage: s.getBaseClientPageData(title, "", r),
  545. Error: getI18nError(err),
  546. Success: message,
  547. }
  548. w.WriteHeader(statusCode)
  549. renderClientTemplate(w, templateMessage, data)
  550. }
  551. func (s *httpdServer) renderClientInternalServerErrorPage(w http.ResponseWriter, r *http.Request, err error) {
  552. s.renderClientMessagePage(w, r, util.I18nError500Title, http.StatusInternalServerError,
  553. util.NewI18nError(err, util.I18nError500Message), "")
  554. }
  555. func (s *httpdServer) renderClientBadRequestPage(w http.ResponseWriter, r *http.Request, err error) {
  556. s.renderClientMessagePage(w, r, util.I18nError400Title, http.StatusBadRequest,
  557. util.NewI18nError(err, util.I18nError400Message), "")
  558. }
  559. func (s *httpdServer) renderClientForbiddenPage(w http.ResponseWriter, r *http.Request, err error) {
  560. s.renderClientMessagePage(w, r, util.I18nError403Title, http.StatusForbidden,
  561. util.NewI18nError(err, util.I18nError403Message), "")
  562. }
  563. func (s *httpdServer) renderClientNotFoundPage(w http.ResponseWriter, r *http.Request, err error) {
  564. s.renderClientMessagePage(w, r, util.I18nError404Title, http.StatusNotFound,
  565. util.NewI18nError(err, util.I18nError404Message), "")
  566. }
  567. func (s *httpdServer) renderClientTwoFactorPage(w http.ResponseWriter, r *http.Request, err *util.I18nError, ip string) {
  568. data := twoFactorPage{
  569. commonBasePage: getCommonBasePage(r),
  570. Title: pageTwoFactorTitle,
  571. CurrentURL: webClientTwoFactorPath,
  572. Error: err,
  573. CSRFToken: createCSRFToken(ip),
  574. RecoveryURL: webClientTwoFactorRecoveryPath,
  575. Branding: s.binding.Branding.WebClient,
  576. }
  577. if next := r.URL.Query().Get("next"); strings.HasPrefix(next, webClientFilesPath) {
  578. data.CurrentURL += "?next=" + url.QueryEscape(next)
  579. }
  580. renderClientTemplate(w, templateTwoFactor, data)
  581. }
  582. func (s *httpdServer) renderClientTwoFactorRecoveryPage(w http.ResponseWriter, r *http.Request, err *util.I18nError, ip string) {
  583. data := twoFactorPage{
  584. commonBasePage: getCommonBasePage(r),
  585. Title: pageTwoFactorRecoveryTitle,
  586. CurrentURL: webClientTwoFactorRecoveryPath,
  587. Error: err,
  588. CSRFToken: createCSRFToken(ip),
  589. Branding: s.binding.Branding.WebClient,
  590. }
  591. renderClientTemplate(w, templateTwoFactorRecovery, data)
  592. }
  593. func (s *httpdServer) renderClientMFAPage(w http.ResponseWriter, r *http.Request) {
  594. data := clientMFAPage{
  595. baseClientPage: s.getBaseClientPageData(util.I18n2FATitle, webClientMFAPath, r),
  596. TOTPConfigs: mfa.GetAvailableTOTPConfigNames(),
  597. GenerateTOTPURL: webClientTOTPGeneratePath,
  598. ValidateTOTPURL: webClientTOTPValidatePath,
  599. SaveTOTPURL: webClientTOTPSavePath,
  600. RecCodesURL: webClientRecoveryCodesPath,
  601. Protocols: dataprovider.MFAProtocols,
  602. }
  603. user, err := dataprovider.GetUserWithGroupSettings(data.LoggedUser.Username, "")
  604. if err != nil {
  605. s.renderClientInternalServerErrorPage(w, r, err)
  606. return
  607. }
  608. data.TOTPConfig = user.Filters.TOTPConfig
  609. data.RequiredProtocols = user.Filters.TwoFactorAuthProtocols
  610. renderClientTemplate(w, templateClientMFA, data)
  611. }
  612. func (s *httpdServer) renderEditFilePage(w http.ResponseWriter, r *http.Request, fileName, fileData string, readOnly bool) {
  613. title := util.I18nViewFileTitle
  614. if !readOnly {
  615. title = util.I18nEditFileTitle
  616. }
  617. data := editFilePage{
  618. baseClientPage: s.getBaseClientPageData(title, webClientEditFilePath, r),
  619. Path: fileName,
  620. Name: path.Base(fileName),
  621. CurrentDir: path.Dir(fileName),
  622. FileURL: webClientFilePath,
  623. ReadOnly: readOnly,
  624. Data: fileData,
  625. }
  626. renderClientTemplate(w, templateClientEditFile, data)
  627. }
  628. func (s *httpdServer) renderAddUpdateSharePage(w http.ResponseWriter, r *http.Request, share *dataprovider.Share,
  629. err *util.I18nError, isAdd bool) {
  630. currentURL := webClientSharePath
  631. title := util.I18nShareAddTitle
  632. if !isAdd {
  633. currentURL = fmt.Sprintf("%v/%v", webClientSharePath, url.PathEscape(share.ShareID))
  634. title = util.I18nShareUpdateTitle
  635. }
  636. if share.IsPasswordHashed() {
  637. share.Password = redactedSecret
  638. }
  639. data := clientSharePage{
  640. baseClientPage: s.getBaseClientPageData(title, currentURL, r),
  641. Share: share,
  642. Error: err,
  643. IsAdd: isAdd,
  644. }
  645. renderClientTemplate(w, templateClientShare, data)
  646. }
  647. func getDirMapping(dirName, baseWebPath string) []dirMapping {
  648. paths := []dirMapping{}
  649. if dirName != "/" {
  650. paths = append(paths, dirMapping{
  651. DirName: path.Base(dirName),
  652. Href: getFileObjectURL("/", dirName, baseWebPath),
  653. })
  654. for {
  655. dirName = path.Dir(dirName)
  656. if dirName == "/" || dirName == "." {
  657. break
  658. }
  659. paths = append([]dirMapping{{
  660. DirName: path.Base(dirName),
  661. Href: getFileObjectURL("/", dirName, baseWebPath)},
  662. }, paths...)
  663. }
  664. }
  665. return paths
  666. }
  667. func (s *httpdServer) renderSharedFilesPage(w http.ResponseWriter, r *http.Request, dirName string,
  668. err *util.I18nError, share dataprovider.Share,
  669. ) {
  670. currentURL := path.Join(webClientPubSharesPath, share.ShareID, "browse")
  671. baseData := s.getBaseClientPageData(util.I18nSharedFilesTitle, currentURL, r)
  672. baseData.FilesURL = currentURL
  673. baseSharePath := path.Join(webClientPubSharesPath, share.ShareID)
  674. data := filesPage{
  675. baseClientPage: baseData,
  676. Error: err,
  677. CurrentDir: url.QueryEscape(dirName),
  678. DownloadURL: path.Join(baseSharePath, "partial"),
  679. // dirName must be escaped because the router expects the full path as single argument
  680. ShareUploadBaseURL: path.Join(baseSharePath, url.PathEscape(dirName)),
  681. ViewPDFURL: path.Join(baseSharePath, "viewpdf"),
  682. DirsURL: path.Join(baseSharePath, "dirs"),
  683. FileURL: "",
  684. FileActionsURL: "",
  685. CheckExistURL: path.Join(baseSharePath, "browse", "exist"),
  686. TasksURL: "",
  687. CanAddFiles: share.Scope == dataprovider.ShareScopeReadWrite,
  688. CanCreateDirs: false,
  689. CanRename: false,
  690. CanDelete: false,
  691. CanDownload: share.Scope != dataprovider.ShareScopeWrite,
  692. CanShare: false,
  693. CanCopy: false,
  694. Paths: getDirMapping(dirName, currentURL),
  695. QuotaUsage: newUserQuotaUsage(&dataprovider.User{}),
  696. }
  697. renderClientTemplate(w, templateClientFiles, data)
  698. }
  699. func (s *httpdServer) renderShareDownloadPage(w http.ResponseWriter, r *http.Request, downloadLink string) {
  700. data := shareDownloadPage{
  701. baseClientPage: s.getBaseClientPageData(util.I18nShareDownloadTitle, "", r),
  702. DownloadLink: downloadLink,
  703. }
  704. renderClientTemplate(w, templateShareDownload, data)
  705. }
  706. func (s *httpdServer) renderUploadToSharePage(w http.ResponseWriter, r *http.Request, share dataprovider.Share) {
  707. currentURL := path.Join(webClientPubSharesPath, share.ShareID, "upload")
  708. data := shareUploadPage{
  709. baseClientPage: s.getBaseClientPageData(util.I18nShareUploadTitle, currentURL, r),
  710. Share: &share,
  711. UploadBasePath: path.Join(webClientPubSharesPath, share.ShareID),
  712. }
  713. renderClientTemplate(w, templateUploadToShare, data)
  714. }
  715. func (s *httpdServer) renderFilesPage(w http.ResponseWriter, r *http.Request, dirName string,
  716. err *util.I18nError, user *dataprovider.User) {
  717. data := filesPage{
  718. baseClientPage: s.getBaseClientPageData(util.I18nFilesTitle, webClientFilesPath, r),
  719. Error: err,
  720. CurrentDir: url.QueryEscape(dirName),
  721. DownloadURL: webClientDownloadZipPath,
  722. ViewPDFURL: webClientViewPDFPath,
  723. DirsURL: webClientDirsPath,
  724. FileURL: webClientFilePath,
  725. FileActionsURL: webClientFileActionsPath,
  726. CheckExistURL: webClientExistPath,
  727. TasksURL: webClientTasksPath,
  728. CanAddFiles: user.CanAddFilesFromWeb(dirName),
  729. CanCreateDirs: user.CanAddDirsFromWeb(dirName),
  730. CanRename: user.CanRenameFromWeb(dirName, dirName),
  731. CanDelete: user.CanDeleteFromWeb(dirName),
  732. CanDownload: user.HasPerm(dataprovider.PermDownload, dirName),
  733. CanShare: user.CanManageShares(),
  734. CanCopy: user.CanCopyFromWeb(dirName, dirName),
  735. ShareUploadBaseURL: "",
  736. Paths: getDirMapping(dirName, webClientFilesPath),
  737. QuotaUsage: newUserQuotaUsage(user),
  738. }
  739. renderClientTemplate(w, templateClientFiles, data)
  740. }
  741. func (s *httpdServer) renderClientProfilePage(w http.ResponseWriter, r *http.Request, err *util.I18nError) {
  742. data := clientProfilePage{
  743. baseClientPage: s.getBaseClientPageData(util.I18nProfileTitle, webClientProfilePath, r),
  744. Error: err,
  745. }
  746. user, userMerged, errUser := dataprovider.GetUserVariants(data.LoggedUser.Username, "")
  747. if errUser != nil {
  748. s.renderClientInternalServerErrorPage(w, r, errUser)
  749. return
  750. }
  751. data.PublicKeys = user.PublicKeys
  752. data.TLSCerts = user.Filters.TLSCerts
  753. data.AllowAPIKeyAuth = user.Filters.AllowAPIKeyAuth
  754. data.Email = user.Email
  755. data.Description = user.Description
  756. data.CanSubmit = userMerged.CanUpdateProfile()
  757. renderClientTemplate(w, templateClientProfile, data)
  758. }
  759. func (s *httpdServer) renderClientChangePasswordPage(w http.ResponseWriter, r *http.Request, err *util.I18nError) {
  760. data := changeClientPasswordPage{
  761. baseClientPage: s.getBaseClientPageData(util.I18nChangePwdTitle, webChangeClientPwdPath, r),
  762. Error: err,
  763. }
  764. renderClientTemplate(w, templateChangePwd, data)
  765. }
  766. func (s *httpdServer) handleWebClientDownloadZip(w http.ResponseWriter, r *http.Request) {
  767. r.Body = http.MaxBytesReader(w, r.Body, maxMultipartMem)
  768. claims, err := getTokenClaims(r)
  769. if err != nil || claims.Username == "" {
  770. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  771. return
  772. }
  773. if err := r.ParseForm(); err != nil {
  774. s.renderClientBadRequestPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidForm))
  775. return
  776. }
  777. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  778. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  779. s.renderClientForbiddenPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF))
  780. return
  781. }
  782. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  783. if err != nil {
  784. s.renderClientMessagePage(w, r, util.I18nError500Title, getRespStatus(err),
  785. util.NewI18nError(err, util.I18nErrorGetUser), "")
  786. return
  787. }
  788. connID := xid.New().String()
  789. protocol := getProtocolFromRequest(r)
  790. connectionID := fmt.Sprintf("%v_%v", protocol, connID)
  791. if err := checkHTTPClientUser(&user, r, connectionID, false); err != nil {
  792. s.renderClientForbiddenPage(w, r, err)
  793. return
  794. }
  795. connection := &Connection{
  796. BaseConnection: common.NewBaseConnection(connID, protocol, util.GetHTTPLocalAddress(r),
  797. r.RemoteAddr, user),
  798. request: r,
  799. }
  800. if err = common.Connections.Add(connection); err != nil {
  801. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  802. util.NewI18nError(err, util.I18nError429Message), "")
  803. return
  804. }
  805. defer common.Connections.Remove(connection.GetID())
  806. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  807. files := r.Form.Get("files")
  808. var filesList []string
  809. err = json.Unmarshal(util.StringToBytes(files), &filesList)
  810. if err != nil {
  811. s.renderClientBadRequestPage(w, r, err)
  812. return
  813. }
  814. w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"",
  815. getCompressedFileName(connection.GetUsername(), filesList)))
  816. renderCompressedFiles(w, connection, name, filesList, nil)
  817. }
  818. func (s *httpdServer) handleClientSharePartialDownload(w http.ResponseWriter, r *http.Request) {
  819. r.Body = http.MaxBytesReader(w, r.Body, maxMultipartMem)
  820. if err := r.ParseForm(); err != nil {
  821. s.renderClientBadRequestPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidForm))
  822. return
  823. }
  824. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead, dataprovider.ShareScopeReadWrite}
  825. share, connection, err := s.checkPublicShare(w, r, validScopes)
  826. if err != nil {
  827. return
  828. }
  829. if err := validateBrowsableShare(share, connection); err != nil {
  830. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  831. return
  832. }
  833. name, err := getBrowsableSharedPath(share.Paths[0], r)
  834. if err != nil {
  835. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  836. return
  837. }
  838. if err = common.Connections.Add(connection); err != nil {
  839. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  840. util.NewI18nError(err, util.I18nError429Message), "")
  841. return
  842. }
  843. defer common.Connections.Remove(connection.GetID())
  844. transferQuota := connection.GetTransferQuota()
  845. if !transferQuota.HasDownloadSpace() {
  846. err = util.NewI18nError(connection.GetReadQuotaExceededError(), util.I18nErrorQuotaRead)
  847. connection.Log(logger.LevelInfo, "denying share read due to quota limits")
  848. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getMappedStatusCode(err), err, "")
  849. return
  850. }
  851. files := r.Form.Get("files")
  852. var filesList []string
  853. err = json.Unmarshal(util.StringToBytes(files), &filesList)
  854. if err != nil {
  855. s.renderClientBadRequestPage(w, r, err)
  856. return
  857. }
  858. dataprovider.UpdateShareLastUse(&share, 1) //nolint:errcheck
  859. w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=\"%s\"",
  860. getCompressedFileName(fmt.Sprintf("share-%s", share.Name), filesList)))
  861. renderCompressedFiles(w, connection, name, filesList, &share)
  862. }
  863. func (s *httpdServer) handleShareGetDirContents(w http.ResponseWriter, r *http.Request) {
  864. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  865. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead, dataprovider.ShareScopeReadWrite}
  866. share, connection, err := s.checkPublicShare(w, r, validScopes)
  867. if err != nil {
  868. return
  869. }
  870. if err := validateBrowsableShare(share, connection); err != nil {
  871. sendAPIResponse(w, r, err, getI18NErrorString(err, util.I18nError500Message), getRespStatus(err))
  872. return
  873. }
  874. name, err := getBrowsableSharedPath(share.Paths[0], r)
  875. if err != nil {
  876. sendAPIResponse(w, r, err, getI18NErrorString(err, util.I18nError500Message), getRespStatus(err))
  877. return
  878. }
  879. if err = common.Connections.Add(connection); err != nil {
  880. sendAPIResponse(w, r, err, getI18NErrorString(err, util.I18nError429Message), http.StatusTooManyRequests)
  881. return
  882. }
  883. defer common.Connections.Remove(connection.GetID())
  884. lister, err := connection.ReadDir(name)
  885. if err != nil {
  886. sendAPIResponse(w, r, err, getI18NErrorString(err, util.I18nErrorDirListGeneric), getMappedStatusCode(err))
  887. return
  888. }
  889. defer lister.Close()
  890. dataGetter := func(limit, _ int) ([]byte, int, error) {
  891. contents, err := lister.Next(limit)
  892. if errors.Is(err, io.EOF) {
  893. err = nil
  894. }
  895. if err != nil {
  896. return nil, 0, err
  897. }
  898. results := make([]map[string]any, 0, len(contents))
  899. for _, info := range contents {
  900. if !info.Mode().IsDir() && !info.Mode().IsRegular() {
  901. continue
  902. }
  903. res := make(map[string]any)
  904. if info.IsDir() {
  905. res["type"] = "1"
  906. res["size"] = ""
  907. } else {
  908. res["type"] = "2"
  909. res["size"] = info.Size()
  910. }
  911. res["meta"] = fmt.Sprintf("%v_%v", res["type"], info.Name())
  912. res["name"] = info.Name()
  913. res["url"] = getFileObjectURL(share.GetRelativePath(name), info.Name(),
  914. path.Join(webClientPubSharesPath, share.ShareID, "browse"))
  915. res["last_modified"] = getFileObjectModTime(info.ModTime())
  916. results = append(results, res)
  917. }
  918. data, err := json.Marshal(results)
  919. count := limit
  920. if len(results) == 0 {
  921. count = 0
  922. }
  923. return data, count, err
  924. }
  925. streamJSONArray(w, defaultQueryLimit, dataGetter)
  926. }
  927. func (s *httpdServer) handleClientUploadToShare(w http.ResponseWriter, r *http.Request) {
  928. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  929. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeWrite, dataprovider.ShareScopeReadWrite}
  930. share, _, err := s.checkPublicShare(w, r, validScopes)
  931. if err != nil {
  932. return
  933. }
  934. if share.Scope == dataprovider.ShareScopeReadWrite {
  935. http.Redirect(w, r, path.Join(webClientPubSharesPath, share.ShareID, "browse"), http.StatusFound)
  936. return
  937. }
  938. s.renderUploadToSharePage(w, r, share)
  939. }
  940. func (s *httpdServer) handleShareGetFiles(w http.ResponseWriter, r *http.Request) {
  941. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  942. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead, dataprovider.ShareScopeReadWrite}
  943. share, connection, err := s.checkPublicShare(w, r, validScopes)
  944. if err != nil {
  945. return
  946. }
  947. if err := validateBrowsableShare(share, connection); err != nil {
  948. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  949. return
  950. }
  951. name, err := getBrowsableSharedPath(share.Paths[0], r)
  952. if err != nil {
  953. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  954. return
  955. }
  956. if err = common.Connections.Add(connection); err != nil {
  957. s.renderSharedFilesPage(w, r, path.Dir(share.GetRelativePath(name)),
  958. util.NewI18nError(err, util.I18nError429Message), share)
  959. return
  960. }
  961. defer common.Connections.Remove(connection.GetID())
  962. var info os.FileInfo
  963. if name == "/" {
  964. info = vfs.NewFileInfo(name, true, 0, time.Unix(0, 0), false)
  965. } else {
  966. info, err = connection.Stat(name, 1)
  967. }
  968. if err != nil {
  969. s.renderSharedFilesPage(w, r, path.Dir(share.GetRelativePath(name)),
  970. util.NewI18nError(err, i18nFsMsg(getRespStatus(err))), share)
  971. return
  972. }
  973. if info.IsDir() {
  974. s.renderSharedFilesPage(w, r, share.GetRelativePath(name), nil, share)
  975. return
  976. }
  977. dataprovider.UpdateShareLastUse(&share, 1) //nolint:errcheck
  978. if status, err := downloadFile(w, r, connection, name, info, false, &share); err != nil {
  979. dataprovider.UpdateShareLastUse(&share, -1) //nolint:errcheck
  980. if status > 0 {
  981. s.renderSharedFilesPage(w, r, path.Dir(share.GetRelativePath(name)),
  982. util.NewI18nError(err, i18nFsMsg(getRespStatus(err))), share)
  983. }
  984. }
  985. }
  986. func (s *httpdServer) handleShareViewPDF(w http.ResponseWriter, r *http.Request) {
  987. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  988. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead, dataprovider.ShareScopeReadWrite}
  989. share, _, err := s.checkPublicShare(w, r, validScopes)
  990. if err != nil {
  991. return
  992. }
  993. name := util.CleanPath(r.URL.Query().Get("path"))
  994. data := viewPDFPage{
  995. commonBasePage: getCommonBasePage(r),
  996. Title: path.Base(name),
  997. URL: fmt.Sprintf("%s?path=%s&_=%d", path.Join(webClientPubSharesPath, share.ShareID, "getpdf"),
  998. url.QueryEscape(name), time.Now().UTC().Unix()),
  999. Branding: s.binding.Branding.WebClient,
  1000. }
  1001. renderClientTemplate(w, templateClientViewPDF, data)
  1002. }
  1003. func (s *httpdServer) handleShareGetPDF(w http.ResponseWriter, r *http.Request) {
  1004. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1005. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead, dataprovider.ShareScopeReadWrite}
  1006. share, connection, err := s.checkPublicShare(w, r, validScopes)
  1007. if err != nil {
  1008. return
  1009. }
  1010. if err := validateBrowsableShare(share, connection); err != nil {
  1011. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  1012. return
  1013. }
  1014. name, err := getBrowsableSharedPath(share.Paths[0], r)
  1015. if err != nil {
  1016. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, getRespStatus(err), err, "")
  1017. return
  1018. }
  1019. if err = common.Connections.Add(connection); err != nil {
  1020. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  1021. util.NewI18nError(err, util.I18nError429Message), "")
  1022. return
  1023. }
  1024. defer common.Connections.Remove(connection.GetID())
  1025. info, err := connection.Stat(name, 1)
  1026. if err != nil {
  1027. status := getRespStatus(err)
  1028. s.renderClientMessagePage(w, r, util.I18nShareAccessErrorTitle, status,
  1029. util.NewI18nError(err, i18nFsMsg(status)), "")
  1030. return
  1031. }
  1032. if info.IsDir() {
  1033. s.renderClientBadRequestPage(w, r, util.NewI18nError(fmt.Errorf("%q is not a file", name), util.I18nErrorPDFMessage))
  1034. return
  1035. }
  1036. connection.User.CheckFsRoot(connection.ID) //nolint:errcheck
  1037. if err := s.ensurePDF(w, r, name, connection); err != nil {
  1038. return
  1039. }
  1040. dataprovider.UpdateShareLastUse(&share, 1) //nolint:errcheck
  1041. if _, err := downloadFile(w, r, connection, name, info, true, &share); err != nil {
  1042. dataprovider.UpdateShareLastUse(&share, -1) //nolint:errcheck
  1043. }
  1044. }
  1045. func (s *httpdServer) handleClientGetDirContents(w http.ResponseWriter, r *http.Request) {
  1046. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1047. claims, err := getTokenClaims(r)
  1048. if err != nil || claims.Username == "" {
  1049. sendAPIResponse(w, r, nil, util.I18nErrorDirList403, http.StatusForbidden)
  1050. return
  1051. }
  1052. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1053. if err != nil {
  1054. sendAPIResponse(w, r, nil, util.I18nErrorDirListUser, getRespStatus(err))
  1055. return
  1056. }
  1057. connID := xid.New().String()
  1058. protocol := getProtocolFromRequest(r)
  1059. connectionID := fmt.Sprintf("%s_%s", protocol, connID)
  1060. if err := checkHTTPClientUser(&user, r, connectionID, false); err != nil {
  1061. sendAPIResponse(w, r, err, getI18NErrorString(err, util.I18nErrorDirList403), http.StatusForbidden)
  1062. return
  1063. }
  1064. connection := &Connection{
  1065. BaseConnection: common.NewBaseConnection(connID, protocol, util.GetHTTPLocalAddress(r),
  1066. r.RemoteAddr, user),
  1067. request: r,
  1068. }
  1069. if err = common.Connections.Add(connection); err != nil {
  1070. sendAPIResponse(w, r, err, util.I18nErrorDirList429, http.StatusTooManyRequests)
  1071. return
  1072. }
  1073. defer common.Connections.Remove(connection.GetID())
  1074. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1075. lister, err := connection.ReadDir(name)
  1076. if err != nil {
  1077. statusCode := getMappedStatusCode(err)
  1078. sendAPIResponse(w, r, err, i18nListDirMsg(statusCode), statusCode)
  1079. return
  1080. }
  1081. defer lister.Close()
  1082. dirTree := r.URL.Query().Get("dirtree") == "1"
  1083. dataGetter := func(limit, _ int) ([]byte, int, error) {
  1084. contents, err := lister.Next(limit)
  1085. if errors.Is(err, io.EOF) {
  1086. err = nil
  1087. }
  1088. if err != nil {
  1089. return nil, 0, err
  1090. }
  1091. results := make([]map[string]any, 0, len(contents))
  1092. for _, info := range contents {
  1093. res := make(map[string]any)
  1094. res["url"] = getFileObjectURL(name, info.Name(), webClientFilesPath)
  1095. if info.IsDir() {
  1096. res["type"] = "1"
  1097. res["size"] = ""
  1098. res["dir_path"] = url.QueryEscape(path.Join(name, info.Name()))
  1099. } else {
  1100. if dirTree {
  1101. continue
  1102. }
  1103. res["type"] = "2"
  1104. if info.Mode()&os.ModeSymlink != 0 {
  1105. res["size"] = ""
  1106. } else {
  1107. res["size"] = info.Size()
  1108. if info.Size() < httpdMaxEditFileSize {
  1109. res["edit_url"] = strings.Replace(res["url"].(string), webClientFilesPath, webClientEditFilePath, 1)
  1110. }
  1111. }
  1112. }
  1113. res["meta"] = fmt.Sprintf("%v_%v", res["type"], info.Name())
  1114. res["name"] = info.Name()
  1115. res["last_modified"] = getFileObjectModTime(info.ModTime())
  1116. results = append(results, res)
  1117. }
  1118. data, err := json.Marshal(results)
  1119. count := limit
  1120. if len(results) == 0 {
  1121. count = 0
  1122. }
  1123. return data, count, err
  1124. }
  1125. streamJSONArray(w, defaultQueryLimit, dataGetter)
  1126. }
  1127. func (s *httpdServer) handleClientGetFiles(w http.ResponseWriter, r *http.Request) {
  1128. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1129. claims, err := getTokenClaims(r)
  1130. if err != nil || claims.Username == "" {
  1131. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1132. return
  1133. }
  1134. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1135. if err != nil {
  1136. s.renderClientMessagePage(w, r, util.I18nError500Title, getRespStatus(err),
  1137. util.NewI18nError(err, util.I18nErrorGetUser), "")
  1138. return
  1139. }
  1140. connID := xid.New().String()
  1141. protocol := getProtocolFromRequest(r)
  1142. connectionID := fmt.Sprintf("%v_%v", protocol, connID)
  1143. if err := checkHTTPClientUser(&user, r, connectionID, false); err != nil {
  1144. s.renderClientForbiddenPage(w, r, err)
  1145. return
  1146. }
  1147. connection := &Connection{
  1148. BaseConnection: common.NewBaseConnection(connID, protocol, util.GetHTTPLocalAddress(r),
  1149. r.RemoteAddr, user),
  1150. request: r,
  1151. }
  1152. if err = common.Connections.Add(connection); err != nil {
  1153. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  1154. util.NewI18nError(err, util.I18nError429Message), "")
  1155. return
  1156. }
  1157. defer common.Connections.Remove(connection.GetID())
  1158. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1159. var info os.FileInfo
  1160. if name == "/" {
  1161. info = vfs.NewFileInfo(name, true, 0, time.Unix(0, 0), false)
  1162. } else {
  1163. info, err = connection.Stat(name, 0)
  1164. }
  1165. if err != nil {
  1166. s.renderFilesPage(w, r, path.Dir(name), util.NewI18nError(err, i18nFsMsg(getRespStatus(err))), &user)
  1167. return
  1168. }
  1169. if info.IsDir() {
  1170. s.renderFilesPage(w, r, name, nil, &user)
  1171. return
  1172. }
  1173. if status, err := downloadFile(w, r, connection, name, info, false, nil); err != nil && status != 0 {
  1174. if status > 0 {
  1175. if status == http.StatusRequestedRangeNotSatisfiable {
  1176. s.renderClientMessagePage(w, r, util.I18nError416Title, status,
  1177. util.NewI18nError(err, util.I18nError416Message), "")
  1178. return
  1179. }
  1180. s.renderFilesPage(w, r, path.Dir(name), util.NewI18nError(err, i18nFsMsg(status)), &user)
  1181. }
  1182. }
  1183. }
  1184. func (s *httpdServer) handleClientEditFile(w http.ResponseWriter, r *http.Request) {
  1185. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1186. claims, err := getTokenClaims(r)
  1187. if err != nil || claims.Username == "" {
  1188. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1189. return
  1190. }
  1191. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1192. if err != nil {
  1193. s.renderClientMessagePage(w, r, util.I18nError500Title, getRespStatus(err),
  1194. util.NewI18nError(err, util.I18nErrorGetUser), "")
  1195. return
  1196. }
  1197. connID := xid.New().String()
  1198. protocol := getProtocolFromRequest(r)
  1199. connectionID := fmt.Sprintf("%v_%v", protocol, connID)
  1200. if err := checkHTTPClientUser(&user, r, connectionID, false); err != nil {
  1201. s.renderClientForbiddenPage(w, r, err)
  1202. return
  1203. }
  1204. connection := &Connection{
  1205. BaseConnection: common.NewBaseConnection(connID, protocol, util.GetHTTPLocalAddress(r),
  1206. r.RemoteAddr, user),
  1207. request: r,
  1208. }
  1209. if err = common.Connections.Add(connection); err != nil {
  1210. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  1211. util.NewI18nError(err, util.I18nError429Message), "")
  1212. return
  1213. }
  1214. defer common.Connections.Remove(connection.GetID())
  1215. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1216. info, err := connection.Stat(name, 0)
  1217. if err != nil {
  1218. status := getRespStatus(err)
  1219. s.renderClientMessagePage(w, r, util.I18nErrorEditorTitle, status, util.NewI18nError(err, i18nFsMsg(status)), "")
  1220. return
  1221. }
  1222. if info.IsDir() {
  1223. s.renderClientMessagePage(w, r, util.I18nErrorEditorTitle, http.StatusBadRequest,
  1224. util.NewI18nError(
  1225. util.NewValidationError(fmt.Sprintf("The path %q does not point to a file", name)),
  1226. util.I18nErrorEditDir,
  1227. ), "")
  1228. return
  1229. }
  1230. if info.Size() > httpdMaxEditFileSize {
  1231. s.renderClientMessagePage(w, r, util.I18nErrorEditorTitle, http.StatusBadRequest,
  1232. util.NewI18nError(
  1233. util.NewValidationError(fmt.Sprintf("The file size %v for %q exceeds the maximum allowed size",
  1234. util.ByteCountIEC(info.Size()), name)),
  1235. util.I18nErrorEditSize,
  1236. ), "")
  1237. return
  1238. }
  1239. connection.User.CheckFsRoot(connection.ID) //nolint:errcheck
  1240. reader, err := connection.getFileReader(name, 0, r.Method)
  1241. if err != nil {
  1242. s.renderClientMessagePage(w, r, util.I18nErrorEditorTitle, getRespStatus(err),
  1243. util.NewI18nError(err, util.I18nError500Message), "")
  1244. return
  1245. }
  1246. defer reader.Close()
  1247. var b bytes.Buffer
  1248. _, err = io.Copy(&b, reader)
  1249. if err != nil {
  1250. s.renderClientMessagePage(w, r, util.I18nErrorEditorTitle, getRespStatus(err),
  1251. util.NewI18nError(err, util.I18nError500Message), "")
  1252. return
  1253. }
  1254. s.renderEditFilePage(w, r, name, b.String(), !user.CanAddFilesFromWeb(path.Dir(name)))
  1255. }
  1256. func (s *httpdServer) handleClientAddShareGet(w http.ResponseWriter, r *http.Request) {
  1257. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1258. claims, err := getTokenClaims(r)
  1259. if err != nil || claims.Username == "" {
  1260. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1261. return
  1262. }
  1263. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1264. if err != nil {
  1265. s.renderClientMessagePage(w, r, util.I18nError500Title, getRespStatus(err),
  1266. util.NewI18nError(err, util.I18nErrorGetUser), "")
  1267. return
  1268. }
  1269. share := &dataprovider.Share{Scope: dataprovider.ShareScopeRead}
  1270. if user.Filters.DefaultSharesExpiration > 0 {
  1271. share.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(24 * time.Hour * time.Duration(user.Filters.DefaultSharesExpiration)))
  1272. } else if user.Filters.MaxSharesExpiration > 0 {
  1273. share.ExpiresAt = util.GetTimeAsMsSinceEpoch(time.Now().Add(24 * time.Hour * time.Duration(user.Filters.MaxSharesExpiration)))
  1274. }
  1275. dirName := "/"
  1276. if _, ok := r.URL.Query()["path"]; ok {
  1277. dirName = util.CleanPath(r.URL.Query().Get("path"))
  1278. }
  1279. if _, ok := r.URL.Query()["files"]; ok {
  1280. files := r.URL.Query().Get("files")
  1281. var filesList []string
  1282. err := json.Unmarshal(util.StringToBytes(files), &filesList)
  1283. if err != nil {
  1284. s.renderClientBadRequestPage(w, r, err)
  1285. return
  1286. }
  1287. for _, f := range filesList {
  1288. if f != "" {
  1289. share.Paths = append(share.Paths, path.Join(dirName, f))
  1290. }
  1291. }
  1292. }
  1293. s.renderAddUpdateSharePage(w, r, share, nil, true)
  1294. }
  1295. func (s *httpdServer) handleClientUpdateShareGet(w http.ResponseWriter, r *http.Request) {
  1296. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1297. claims, err := getTokenClaims(r)
  1298. if err != nil || claims.Username == "" {
  1299. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1300. return
  1301. }
  1302. shareID := getURLParam(r, "id")
  1303. share, err := dataprovider.ShareExists(shareID, claims.Username)
  1304. if err == nil {
  1305. s.renderAddUpdateSharePage(w, r, &share, nil, false)
  1306. } else if errors.Is(err, util.ErrNotFound) {
  1307. s.renderClientNotFoundPage(w, r, err)
  1308. } else {
  1309. s.renderClientInternalServerErrorPage(w, r, err)
  1310. }
  1311. }
  1312. func (s *httpdServer) handleClientAddSharePost(w http.ResponseWriter, r *http.Request) {
  1313. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1314. claims, err := getTokenClaims(r)
  1315. if err != nil || claims.Username == "" {
  1316. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1317. return
  1318. }
  1319. share, err := getShareFromPostFields(r)
  1320. if err != nil {
  1321. s.renderAddUpdateSharePage(w, r, share, util.NewI18nError(err, util.I18nError500Message), true)
  1322. return
  1323. }
  1324. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  1325. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  1326. s.renderClientForbiddenPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF))
  1327. return
  1328. }
  1329. share.ID = 0
  1330. share.ShareID = util.GenerateUniqueID()
  1331. share.LastUseAt = 0
  1332. share.Username = claims.Username
  1333. if share.Password == "" {
  1334. if util.Contains(claims.Permissions, sdk.WebClientShareNoPasswordDisabled) {
  1335. s.renderAddUpdateSharePage(w, r, share,
  1336. util.NewI18nError(util.NewValidationError("You are not allowed to share files/folders without password"), util.I18nErrorShareNoPwd),
  1337. true)
  1338. return
  1339. }
  1340. }
  1341. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1342. if err != nil {
  1343. s.renderAddUpdateSharePage(w, r, share, util.NewI18nError(err, util.I18nErrorGetUser), true)
  1344. return
  1345. }
  1346. if err := user.CheckMaxShareExpiration(util.GetTimeFromMsecSinceEpoch(share.ExpiresAt)); err != nil {
  1347. s.renderAddUpdateSharePage(w, r, share, util.NewI18nError(
  1348. err,
  1349. util.I18nErrorShareExpirationOutOfRange,
  1350. util.I18nErrorArgs(
  1351. map[string]any{
  1352. "val": time.Now().Add(24 * time.Hour * time.Duration(user.Filters.MaxSharesExpiration+1)).UnixMilli(),
  1353. "formatParams": map[string]string{
  1354. "year": "numeric",
  1355. "month": "numeric",
  1356. "day": "numeric",
  1357. },
  1358. },
  1359. ),
  1360. ), true)
  1361. return
  1362. }
  1363. err = dataprovider.AddShare(share, claims.Username, ipAddr, claims.Role)
  1364. if err == nil {
  1365. http.Redirect(w, r, webClientSharesPath, http.StatusSeeOther)
  1366. } else {
  1367. s.renderAddUpdateSharePage(w, r, share, util.NewI18nError(err, util.I18nErrorShareGeneric), true)
  1368. }
  1369. }
  1370. func (s *httpdServer) handleClientUpdateSharePost(w http.ResponseWriter, r *http.Request) {
  1371. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1372. claims, err := getTokenClaims(r)
  1373. if err != nil || claims.Username == "" {
  1374. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1375. return
  1376. }
  1377. shareID := getURLParam(r, "id")
  1378. share, err := dataprovider.ShareExists(shareID, claims.Username)
  1379. if errors.Is(err, util.ErrNotFound) {
  1380. s.renderClientNotFoundPage(w, r, err)
  1381. return
  1382. } else if err != nil {
  1383. s.renderClientInternalServerErrorPage(w, r, err)
  1384. return
  1385. }
  1386. updatedShare, err := getShareFromPostFields(r)
  1387. if err != nil {
  1388. s.renderAddUpdateSharePage(w, r, updatedShare, util.NewI18nError(err, util.I18nError500Message), false)
  1389. return
  1390. }
  1391. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  1392. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  1393. s.renderClientForbiddenPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF))
  1394. return
  1395. }
  1396. updatedShare.ShareID = shareID
  1397. updatedShare.Username = claims.Username
  1398. if updatedShare.Password == redactedSecret {
  1399. updatedShare.Password = share.Password
  1400. }
  1401. if updatedShare.Password == "" {
  1402. if util.Contains(claims.Permissions, sdk.WebClientShareNoPasswordDisabled) {
  1403. s.renderAddUpdateSharePage(w, r, updatedShare,
  1404. util.NewI18nError(util.NewValidationError("You are not allowed to share files/folders without password"), util.I18nErrorShareNoPwd),
  1405. false)
  1406. return
  1407. }
  1408. }
  1409. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1410. if err != nil {
  1411. s.renderAddUpdateSharePage(w, r, updatedShare, util.NewI18nError(err, util.I18nErrorGetUser), false)
  1412. return
  1413. }
  1414. if err := user.CheckMaxShareExpiration(util.GetTimeFromMsecSinceEpoch(updatedShare.ExpiresAt)); err != nil {
  1415. s.renderAddUpdateSharePage(w, r, updatedShare, util.NewI18nError(
  1416. err,
  1417. util.I18nErrorShareExpirationOutOfRange,
  1418. util.I18nErrorArgs(
  1419. map[string]any{
  1420. "val": time.Now().Add(24 * time.Hour * time.Duration(user.Filters.MaxSharesExpiration+1)).UnixMilli(),
  1421. "formatParams": map[string]string{
  1422. "year": "numeric",
  1423. "month": "numeric",
  1424. "day": "numeric",
  1425. },
  1426. },
  1427. ),
  1428. ), false)
  1429. return
  1430. }
  1431. err = dataprovider.UpdateShare(updatedShare, claims.Username, ipAddr, claims.Role)
  1432. if err == nil {
  1433. http.Redirect(w, r, webClientSharesPath, http.StatusSeeOther)
  1434. } else {
  1435. s.renderAddUpdateSharePage(w, r, updatedShare, util.NewI18nError(err, util.I18nErrorShareGeneric), false)
  1436. }
  1437. }
  1438. func getAllShares(w http.ResponseWriter, r *http.Request) {
  1439. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1440. claims, err := getTokenClaims(r)
  1441. if err != nil || claims.Username == "" {
  1442. sendAPIResponse(w, r, nil, util.I18nErrorInvalidToken, http.StatusForbidden)
  1443. return
  1444. }
  1445. dataGetter := func(limit, offset int) ([]byte, int, error) {
  1446. shares, err := dataprovider.GetShares(limit, offset, dataprovider.OrderASC, claims.Username)
  1447. if err != nil {
  1448. return nil, 0, err
  1449. }
  1450. data, err := json.Marshal(shares)
  1451. return data, len(shares), err
  1452. }
  1453. streamJSONArray(w, defaultQueryLimit, dataGetter)
  1454. }
  1455. func (s *httpdServer) handleClientGetShares(w http.ResponseWriter, r *http.Request) {
  1456. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1457. data := clientSharesPage{
  1458. baseClientPage: s.getBaseClientPageData(util.I18nSharesTitle, webClientSharesPath, r),
  1459. BasePublicSharesURL: webClientPubSharesPath,
  1460. }
  1461. renderClientTemplate(w, templateClientShares, data)
  1462. }
  1463. func (s *httpdServer) handleClientGetProfile(w http.ResponseWriter, r *http.Request) {
  1464. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1465. s.renderClientProfilePage(w, r, nil)
  1466. }
  1467. func (s *httpdServer) handleWebClientChangePwd(w http.ResponseWriter, r *http.Request) {
  1468. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1469. s.renderClientChangePasswordPage(w, r, nil)
  1470. }
  1471. func (s *httpdServer) handleWebClientProfilePost(w http.ResponseWriter, r *http.Request) { //nolint:gocyclo
  1472. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1473. err := r.ParseForm()
  1474. if err != nil {
  1475. s.renderClientProfilePage(w, r, util.NewI18nError(err, util.I18nErrorInvalidForm))
  1476. return
  1477. }
  1478. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  1479. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  1480. s.renderClientForbiddenPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF))
  1481. return
  1482. }
  1483. claims, err := getTokenClaims(r)
  1484. if err != nil || claims.Username == "" {
  1485. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1486. return
  1487. }
  1488. user, userMerged, err := dataprovider.GetUserVariants(claims.Username, "")
  1489. if err != nil {
  1490. s.renderClientProfilePage(w, r, util.NewI18nError(err, util.I18nErrorGetUser))
  1491. return
  1492. }
  1493. if !userMerged.CanUpdateProfile() {
  1494. s.renderClientForbiddenPage(w, r, util.NewI18nError(
  1495. errors.New("you are not allowed to change anything"),
  1496. util.I18nErrorNoPermissions,
  1497. ))
  1498. return
  1499. }
  1500. if userMerged.CanManagePublicKeys() {
  1501. for k := range r.Form {
  1502. if hasPrefixAndSuffix(k, "public_keys[", "][public_key]") {
  1503. r.Form.Add("public_keys", r.Form.Get(k))
  1504. }
  1505. }
  1506. user.PublicKeys = r.Form["public_keys"]
  1507. }
  1508. if userMerged.CanManageTLSCerts() {
  1509. for k := range r.Form {
  1510. if hasPrefixAndSuffix(k, "tls_certs[", "][tls_cert]") {
  1511. r.Form.Add("tls_certs", r.Form.Get(k))
  1512. }
  1513. }
  1514. user.Filters.TLSCerts = r.Form["tls_certs"]
  1515. }
  1516. if userMerged.CanChangeAPIKeyAuth() {
  1517. user.Filters.AllowAPIKeyAuth = r.Form.Get("allow_api_key_auth") != ""
  1518. }
  1519. if userMerged.CanChangeInfo() {
  1520. user.Email = strings.TrimSpace(r.Form.Get("email"))
  1521. user.Description = r.Form.Get("description")
  1522. }
  1523. err = dataprovider.UpdateUser(&user, dataprovider.ActionExecutorSelf, ipAddr, user.Role)
  1524. if err != nil {
  1525. s.renderClientProfilePage(w, r, util.NewI18nError(err, util.I18nError500Message))
  1526. return
  1527. }
  1528. s.renderClientMessagePage(w, r, util.I18nProfileTitle, http.StatusOK, nil, util.I18nProfileUpdated)
  1529. }
  1530. func (s *httpdServer) handleWebClientMFA(w http.ResponseWriter, r *http.Request) {
  1531. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1532. s.renderClientMFAPage(w, r)
  1533. }
  1534. func (s *httpdServer) handleWebClientTwoFactor(w http.ResponseWriter, r *http.Request) {
  1535. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1536. s.renderClientTwoFactorPage(w, r, nil, util.GetIPFromRemoteAddress(r.RemoteAddr))
  1537. }
  1538. func (s *httpdServer) handleWebClientTwoFactorRecovery(w http.ResponseWriter, r *http.Request) {
  1539. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1540. s.renderClientTwoFactorRecoveryPage(w, r, nil, util.GetIPFromRemoteAddress(r.RemoteAddr))
  1541. }
  1542. func getShareFromPostFields(r *http.Request) (*dataprovider.Share, error) {
  1543. share := &dataprovider.Share{}
  1544. if err := r.ParseForm(); err != nil {
  1545. return share, util.NewI18nError(err, util.I18nErrorInvalidForm)
  1546. }
  1547. for k := range r.Form {
  1548. if hasPrefixAndSuffix(k, "paths[", "][path]") {
  1549. r.Form.Add("paths", r.Form.Get(k))
  1550. }
  1551. }
  1552. share.Name = strings.TrimSpace(r.Form.Get("name"))
  1553. share.Description = r.Form.Get("description")
  1554. for _, p := range r.Form["paths"] {
  1555. if strings.TrimSpace(p) != "" {
  1556. share.Paths = append(share.Paths, p)
  1557. }
  1558. }
  1559. share.Password = strings.TrimSpace(r.Form.Get("password"))
  1560. share.AllowFrom = getSliceFromDelimitedValues(r.Form.Get("allowed_ip"), ",")
  1561. scope, err := strconv.Atoi(r.Form.Get("scope"))
  1562. if err != nil {
  1563. return share, util.NewI18nError(err, util.I18nErrorShareScope)
  1564. }
  1565. share.Scope = dataprovider.ShareScope(scope)
  1566. maxTokens, err := strconv.Atoi(r.Form.Get("max_tokens"))
  1567. if err != nil {
  1568. return share, util.NewI18nError(err, util.I18nErrorShareMaxTokens)
  1569. }
  1570. share.MaxTokens = maxTokens
  1571. expirationDateMillis := int64(0)
  1572. expirationDateString := strings.TrimSpace(r.Form.Get("expiration_date"))
  1573. if expirationDateString != "" {
  1574. expirationDate, err := time.Parse(webDateTimeFormat, expirationDateString)
  1575. if err != nil {
  1576. return share, util.NewI18nError(err, util.I18nErrorShareExpiration)
  1577. }
  1578. expirationDateMillis = util.GetTimeAsMsSinceEpoch(expirationDate)
  1579. }
  1580. share.ExpiresAt = expirationDateMillis
  1581. return share, nil
  1582. }
  1583. func (s *httpdServer) handleWebClientForgotPwd(w http.ResponseWriter, r *http.Request) {
  1584. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1585. if !smtp.IsEnabled() {
  1586. s.renderClientNotFoundPage(w, r, errors.New("this page does not exist"))
  1587. return
  1588. }
  1589. s.renderClientForgotPwdPage(w, r, nil, util.GetIPFromRemoteAddress(r.RemoteAddr))
  1590. }
  1591. func (s *httpdServer) handleWebClientForgotPwdPost(w http.ResponseWriter, r *http.Request) {
  1592. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1593. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  1594. err := r.ParseForm()
  1595. if err != nil {
  1596. s.renderClientForgotPwdPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidForm), ipAddr)
  1597. return
  1598. }
  1599. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  1600. s.renderClientForbiddenPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF))
  1601. return
  1602. }
  1603. username := strings.TrimSpace(r.Form.Get("username"))
  1604. err = handleForgotPassword(r, username, false)
  1605. if err != nil {
  1606. s.renderClientForgotPwdPage(w, r, util.NewI18nError(err, util.I18nErrorPwdResetGeneric), ipAddr)
  1607. return
  1608. }
  1609. http.Redirect(w, r, webClientResetPwdPath, http.StatusFound)
  1610. }
  1611. func (s *httpdServer) handleWebClientPasswordReset(w http.ResponseWriter, r *http.Request) {
  1612. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1613. if !smtp.IsEnabled() {
  1614. s.renderClientNotFoundPage(w, r, errors.New("this page does not exist"))
  1615. return
  1616. }
  1617. s.renderClientResetPwdPage(w, r, nil, util.GetIPFromRemoteAddress(r.RemoteAddr))
  1618. }
  1619. func (s *httpdServer) handleClientViewPDF(w http.ResponseWriter, r *http.Request) {
  1620. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1621. name := r.URL.Query().Get("path")
  1622. if name == "" {
  1623. s.renderClientBadRequestPage(w, r, errors.New("no file specified"))
  1624. return
  1625. }
  1626. name = util.CleanPath(name)
  1627. data := viewPDFPage{
  1628. commonBasePage: getCommonBasePage(r),
  1629. Title: path.Base(name),
  1630. URL: fmt.Sprintf("%s?path=%s&_=%d", webClientGetPDFPath, url.QueryEscape(name), time.Now().UTC().Unix()),
  1631. Branding: s.binding.Branding.WebClient,
  1632. }
  1633. renderClientTemplate(w, templateClientViewPDF, data)
  1634. }
  1635. func (s *httpdServer) handleClientGetPDF(w http.ResponseWriter, r *http.Request) {
  1636. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1637. claims, err := getTokenClaims(r)
  1638. if err != nil || claims.Username == "" {
  1639. s.renderClientForbiddenPage(w, r, util.NewI18nError(errInvalidTokenClaims, util.I18nErrorInvalidToken))
  1640. return
  1641. }
  1642. name := r.URL.Query().Get("path")
  1643. if name == "" {
  1644. s.renderClientBadRequestPage(w, r, util.NewI18nError(errors.New("no file specified"), util.I18nError400Message))
  1645. return
  1646. }
  1647. name = util.CleanPath(name)
  1648. user, err := dataprovider.GetUserWithGroupSettings(claims.Username, "")
  1649. if err != nil {
  1650. s.renderClientMessagePage(w, r, util.I18nError500Title, getRespStatus(err),
  1651. util.NewI18nError(err, util.I18nErrorGetUser), "")
  1652. return
  1653. }
  1654. connID := xid.New().String()
  1655. protocol := getProtocolFromRequest(r)
  1656. connectionID := fmt.Sprintf("%v_%v", protocol, connID)
  1657. if err := checkHTTPClientUser(&user, r, connectionID, false); err != nil {
  1658. s.renderClientForbiddenPage(w, r, err)
  1659. return
  1660. }
  1661. connection := &Connection{
  1662. BaseConnection: common.NewBaseConnection(connID, protocol, util.GetHTTPLocalAddress(r),
  1663. r.RemoteAddr, user),
  1664. request: r,
  1665. }
  1666. if err = common.Connections.Add(connection); err != nil {
  1667. s.renderClientMessagePage(w, r, util.I18nError429Title, http.StatusTooManyRequests,
  1668. util.NewI18nError(err, util.I18nError429Message), "")
  1669. return
  1670. }
  1671. defer common.Connections.Remove(connection.GetID())
  1672. info, err := connection.Stat(name, 0)
  1673. if err != nil {
  1674. status := getRespStatus(err)
  1675. s.renderClientMessagePage(w, r, util.I18nErrorPDFTitle, status, util.NewI18nError(err, i18nFsMsg(status)), "")
  1676. return
  1677. }
  1678. if info.IsDir() {
  1679. s.renderClientBadRequestPage(w, r, util.NewI18nError(fmt.Errorf("%q is not a file", name), util.I18nErrorPDFMessage))
  1680. return
  1681. }
  1682. connection.User.CheckFsRoot(connection.ID) //nolint:errcheck
  1683. if err := s.ensurePDF(w, r, name, connection); err != nil {
  1684. return
  1685. }
  1686. downloadFile(w, r, connection, name, info, true, nil) //nolint:errcheck
  1687. }
  1688. func (s *httpdServer) ensurePDF(w http.ResponseWriter, r *http.Request, name string, connection *Connection) error {
  1689. reader, err := connection.getFileReader(name, 0, r.Method)
  1690. if err != nil {
  1691. s.renderClientMessagePage(w, r, util.I18nErrorPDFTitle,
  1692. getRespStatus(err), util.NewI18nError(err, util.I18nError500Message), "")
  1693. return err
  1694. }
  1695. defer reader.Close()
  1696. var b bytes.Buffer
  1697. _, err = io.CopyN(&b, reader, 128)
  1698. if err != nil {
  1699. s.renderClientMessagePage(w, r, util.I18nErrorPDFTitle, getRespStatus(err),
  1700. util.NewI18nError(err, util.I18nErrorPDFMessage), "")
  1701. return err
  1702. }
  1703. if ctype := http.DetectContentType(b.Bytes()); ctype != "application/pdf" {
  1704. connection.Log(logger.LevelDebug, "detected %q content type, expected PDF, file %q", ctype, name)
  1705. err := fmt.Errorf("the file %q does not look like a PDF", name)
  1706. s.renderClientBadRequestPage(w, r, util.NewI18nError(err, util.I18nErrorPDFMessage))
  1707. return err
  1708. }
  1709. return nil
  1710. }
  1711. func (s *httpdServer) handleClientShareLoginGet(w http.ResponseWriter, r *http.Request) {
  1712. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1713. s.renderShareLoginPage(w, r, nil, util.GetIPFromRemoteAddress(r.RemoteAddr))
  1714. }
  1715. func (s *httpdServer) handleClientShareLoginPost(w http.ResponseWriter, r *http.Request) {
  1716. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1717. ipAddr := util.GetIPFromRemoteAddress(r.RemoteAddr)
  1718. if err := r.ParseForm(); err != nil {
  1719. s.renderShareLoginPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidForm), ipAddr)
  1720. return
  1721. }
  1722. if err := verifyCSRFToken(r.Form.Get(csrfFormToken), ipAddr); err != nil {
  1723. s.renderShareLoginPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCSRF), ipAddr)
  1724. return
  1725. }
  1726. shareID := getURLParam(r, "id")
  1727. share, err := dataprovider.ShareExists(shareID, "")
  1728. if err != nil {
  1729. s.renderShareLoginPage(w, r, util.NewI18nError(err, util.I18nErrorInvalidCredentials), ipAddr)
  1730. return
  1731. }
  1732. match, err := share.CheckCredentials(strings.TrimSpace(r.Form.Get("share_password")))
  1733. if !match || err != nil {
  1734. s.renderShareLoginPage(w, r, util.NewI18nError(dataprovider.ErrInvalidCredentials, util.I18nErrorInvalidCredentials),
  1735. ipAddr)
  1736. return
  1737. }
  1738. c := jwtTokenClaims{
  1739. Username: shareID,
  1740. }
  1741. err = c.createAndSetCookie(w, r, s.tokenAuth, tokenAudienceWebShare, ipAddr)
  1742. if err != nil {
  1743. s.renderShareLoginPage(w, r, util.NewI18nError(err, util.I18nError500Message), ipAddr)
  1744. return
  1745. }
  1746. next := path.Clean(r.URL.Query().Get("next"))
  1747. baseShareURL := path.Join(webClientPubSharesPath, share.ShareID)
  1748. isRedirect, redirectTo := checkShareRedirectURL(next, baseShareURL)
  1749. if isRedirect {
  1750. http.Redirect(w, r, redirectTo, http.StatusFound)
  1751. return
  1752. }
  1753. s.renderClientMessagePage(w, r, util.I18nSharedFilesTitle, http.StatusOK, nil, util.I18nShareLoginOK)
  1754. }
  1755. func (s *httpdServer) handleClientSharedFile(w http.ResponseWriter, r *http.Request) {
  1756. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1757. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeRead}
  1758. share, _, err := s.checkPublicShare(w, r, validScopes)
  1759. if err != nil {
  1760. return
  1761. }
  1762. query := ""
  1763. if r.URL.RawQuery != "" {
  1764. query = "?" + r.URL.RawQuery
  1765. }
  1766. s.renderShareDownloadPage(w, r, path.Join(webClientPubSharesPath, share.ShareID)+query)
  1767. }
  1768. func (s *httpdServer) handleClientCheckExist(w http.ResponseWriter, r *http.Request) {
  1769. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1770. connection, err := getUserConnection(w, r)
  1771. if err != nil {
  1772. return
  1773. }
  1774. defer common.Connections.Remove(connection.GetID())
  1775. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1776. doCheckExist(w, r, connection, name)
  1777. }
  1778. func (s *httpdServer) handleClientShareCheckExist(w http.ResponseWriter, r *http.Request) {
  1779. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1780. validScopes := []dataprovider.ShareScope{dataprovider.ShareScopeReadWrite}
  1781. share, connection, err := s.checkPublicShare(w, r, validScopes)
  1782. if err != nil {
  1783. return
  1784. }
  1785. if err := validateBrowsableShare(share, connection); err != nil {
  1786. sendAPIResponse(w, r, err, "", getRespStatus(err))
  1787. return
  1788. }
  1789. name, err := getBrowsableSharedPath(share.Paths[0], r)
  1790. if err != nil {
  1791. sendAPIResponse(w, r, err, "", http.StatusBadRequest)
  1792. return
  1793. }
  1794. if err = common.Connections.Add(connection); err != nil {
  1795. sendAPIResponse(w, r, err, "Unable to add connection", http.StatusTooManyRequests)
  1796. return
  1797. }
  1798. defer common.Connections.Remove(connection.GetID())
  1799. doCheckExist(w, r, connection, name)
  1800. }
  1801. type filesToCheck struct {
  1802. Files []string `json:"files"`
  1803. }
  1804. func doCheckExist(w http.ResponseWriter, r *http.Request, connection *Connection, name string) {
  1805. var filesList filesToCheck
  1806. err := render.DecodeJSON(r.Body, &filesList)
  1807. if err != nil {
  1808. sendAPIResponse(w, r, err, "", http.StatusBadRequest)
  1809. return
  1810. }
  1811. if len(filesList.Files) == 0 {
  1812. sendAPIResponse(w, r, errors.New("files to be checked are mandatory"), "", http.StatusBadRequest)
  1813. return
  1814. }
  1815. lister, err := connection.ListDir(name)
  1816. if err != nil {
  1817. sendAPIResponse(w, r, err, "Unable to get directory contents", getMappedStatusCode(err))
  1818. return
  1819. }
  1820. defer lister.Close()
  1821. dataGetter := func(limit, _ int) ([]byte, int, error) {
  1822. contents, err := lister.Next(limit)
  1823. if errors.Is(err, io.EOF) {
  1824. err = nil
  1825. }
  1826. if err != nil {
  1827. return nil, 0, err
  1828. }
  1829. existing := make([]map[string]any, 0)
  1830. for _, info := range contents {
  1831. if util.Contains(filesList.Files, info.Name()) {
  1832. res := make(map[string]any)
  1833. res["name"] = info.Name()
  1834. if info.IsDir() {
  1835. res["type"] = "1"
  1836. res["size"] = ""
  1837. } else {
  1838. res["type"] = "2"
  1839. res["size"] = info.Size()
  1840. }
  1841. existing = append(existing, res)
  1842. }
  1843. }
  1844. data, err := json.Marshal(existing)
  1845. count := limit
  1846. if len(existing) == 0 {
  1847. count = 0
  1848. }
  1849. return data, count, err
  1850. }
  1851. streamJSONArray(w, defaultQueryLimit, dataGetter)
  1852. }
  1853. func checkShareRedirectURL(next, base string) (bool, string) {
  1854. if !strings.HasPrefix(next, base) {
  1855. return false, ""
  1856. }
  1857. if next == base {
  1858. return true, path.Join(next, "download")
  1859. }
  1860. baseURL, err := url.Parse(base)
  1861. if err != nil {
  1862. return false, ""
  1863. }
  1864. nextURL, err := url.Parse(next)
  1865. if err != nil {
  1866. return false, ""
  1867. }
  1868. if nextURL.Path == baseURL.Path {
  1869. redirectURL := nextURL.JoinPath("download")
  1870. return true, redirectURL.String()
  1871. }
  1872. return true, next
  1873. }
  1874. func getWebTask(w http.ResponseWriter, r *http.Request) {
  1875. r.Body = http.MaxBytesReader(w, r.Body, maxLoginBodySize)
  1876. claims, err := getTokenClaims(r)
  1877. if err != nil || claims.Username == "" {
  1878. sendAPIResponse(w, r, err, "Invalid token claims", http.StatusBadRequest)
  1879. return
  1880. }
  1881. taskID := getURLParam(r, "id")
  1882. task, err := webTaskMgr.Get(taskID)
  1883. if err != nil {
  1884. sendAPIResponse(w, r, err, "Unable to get task", getMappedStatusCode(err))
  1885. return
  1886. }
  1887. if task.User != claims.Username {
  1888. sendAPIResponse(w, r, nil, http.StatusText(http.StatusForbidden), http.StatusForbidden)
  1889. return
  1890. }
  1891. render.JSON(w, r, task)
  1892. }
  1893. func taskDeleteDir(w http.ResponseWriter, r *http.Request) {
  1894. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1895. connection, err := getUserConnection(w, r)
  1896. if err != nil {
  1897. return
  1898. }
  1899. name := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1900. task := webTaskData{
  1901. ID: connection.GetID(),
  1902. User: connection.GetUsername(),
  1903. Path: name,
  1904. Timestamp: util.GetTimeAsMsSinceEpoch(time.Now()),
  1905. Status: 0,
  1906. }
  1907. if err := webTaskMgr.Add(task); err != nil {
  1908. common.Connections.Remove(connection.GetID())
  1909. sendAPIResponse(w, r, nil, "Unable to create task", http.StatusInternalServerError)
  1910. return
  1911. }
  1912. go executeDeleteTask(connection, task)
  1913. sendAPIResponse(w, r, nil, task.ID, http.StatusAccepted)
  1914. }
  1915. func taskRenameFsEntry(w http.ResponseWriter, r *http.Request) {
  1916. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1917. connection, err := getUserConnection(w, r)
  1918. if err != nil {
  1919. return
  1920. }
  1921. oldName := connection.User.GetCleanedPath(r.URL.Query().Get("path"))
  1922. newName := connection.User.GetCleanedPath(r.URL.Query().Get("target"))
  1923. task := webTaskData{
  1924. ID: connection.GetID(),
  1925. User: connection.GetUsername(),
  1926. Path: oldName,
  1927. Target: newName,
  1928. Timestamp: util.GetTimeAsMsSinceEpoch(time.Now()),
  1929. Status: 0,
  1930. }
  1931. if err := webTaskMgr.Add(task); err != nil {
  1932. common.Connections.Remove(connection.GetID())
  1933. sendAPIResponse(w, r, nil, "Unable to create task", http.StatusInternalServerError)
  1934. return
  1935. }
  1936. go executeRenameTask(connection, task)
  1937. sendAPIResponse(w, r, nil, task.ID, http.StatusAccepted)
  1938. }
  1939. func taskCopyFsEntry(w http.ResponseWriter, r *http.Request) {
  1940. r.Body = http.MaxBytesReader(w, r.Body, maxRequestSize)
  1941. connection, err := getUserConnection(w, r)
  1942. if err != nil {
  1943. return
  1944. }
  1945. source := r.URL.Query().Get("path")
  1946. target := r.URL.Query().Get("target")
  1947. copyFromSource := strings.HasSuffix(source, "/")
  1948. copyInTarget := strings.HasSuffix(target, "/")
  1949. source = connection.User.GetCleanedPath(source)
  1950. target = connection.User.GetCleanedPath(target)
  1951. if copyFromSource {
  1952. source += "/"
  1953. }
  1954. if copyInTarget {
  1955. target += "/"
  1956. }
  1957. task := webTaskData{
  1958. ID: connection.GetID(),
  1959. User: connection.GetUsername(),
  1960. Path: source,
  1961. Target: target,
  1962. Timestamp: util.GetTimeAsMsSinceEpoch(time.Now()),
  1963. Status: 0,
  1964. }
  1965. if err := webTaskMgr.Add(task); err != nil {
  1966. common.Connections.Remove(connection.GetID())
  1967. sendAPIResponse(w, r, nil, "Unable to create task", http.StatusInternalServerError)
  1968. return
  1969. }
  1970. go executeCopyTask(connection, task)
  1971. sendAPIResponse(w, r, nil, task.ID, http.StatusAccepted)
  1972. }
  1973. func executeDeleteTask(conn *Connection, task webTaskData) {
  1974. done := make(chan bool)
  1975. defer func() {
  1976. close(done)
  1977. common.Connections.Remove(conn.GetID())
  1978. }()
  1979. go keepAliveTask(task, done, 2*time.Minute)
  1980. status := http.StatusOK
  1981. if err := conn.RemoveAll(task.Path); err != nil {
  1982. status = getMappedStatusCode(err)
  1983. }
  1984. task.Timestamp = util.GetTimeAsMsSinceEpoch(time.Now())
  1985. task.Status = status
  1986. err := webTaskMgr.Add(task)
  1987. conn.Log(logger.LevelDebug, "delete task finished, status: %d, update task err: %v", status, err)
  1988. }
  1989. func executeRenameTask(conn *Connection, task webTaskData) {
  1990. done := make(chan bool)
  1991. defer func() {
  1992. close(done)
  1993. common.Connections.Remove(conn.GetID())
  1994. }()
  1995. go keepAliveTask(task, done, 2*time.Minute)
  1996. status := http.StatusOK
  1997. if !conn.IsSameResource(task.Path, task.Target) {
  1998. if err := conn.Copy(task.Path, task.Target); err != nil {
  1999. status = getMappedStatusCode(err)
  2000. task.Timestamp = util.GetTimeAsMsSinceEpoch(time.Now())
  2001. task.Status = status
  2002. err = webTaskMgr.Add(task)
  2003. conn.Log(logger.LevelDebug, "copy step for rename task finished, status: %d, update task err: %v", status, err)
  2004. return
  2005. }
  2006. if err := conn.RemoveAll(task.Path); err != nil {
  2007. status = getMappedStatusCode(err)
  2008. }
  2009. } else {
  2010. if err := conn.Rename(task.Path, task.Target); err != nil {
  2011. status = getMappedStatusCode(err)
  2012. }
  2013. }
  2014. task.Timestamp = util.GetTimeAsMsSinceEpoch(time.Now())
  2015. task.Status = status
  2016. err := webTaskMgr.Add(task)
  2017. conn.Log(logger.LevelDebug, "rename task finished, status: %d, update task err: %v", status, err)
  2018. }
  2019. func executeCopyTask(conn *Connection, task webTaskData) {
  2020. done := make(chan bool)
  2021. defer func() {
  2022. close(done)
  2023. common.Connections.Remove(conn.GetID())
  2024. }()
  2025. go keepAliveTask(task, done, 2*time.Minute)
  2026. status := http.StatusOK
  2027. if err := conn.Copy(task.Path, task.Target); err != nil {
  2028. status = getMappedStatusCode(err)
  2029. }
  2030. task.Timestamp = util.GetTimeAsMsSinceEpoch(time.Now())
  2031. task.Status = status
  2032. err := webTaskMgr.Add(task)
  2033. conn.Log(logger.LevelDebug, "copy task finished, status: %d, update task err: %v", status, err)
  2034. }
  2035. func keepAliveTask(task webTaskData, done chan bool, interval time.Duration) {
  2036. ticker := time.NewTicker(interval)
  2037. defer func() {
  2038. ticker.Stop()
  2039. }()
  2040. for {
  2041. select {
  2042. case <-done:
  2043. return
  2044. case <-ticker.C:
  2045. task.Timestamp = util.GetTimeAsMsSinceEpoch(time.Now())
  2046. err := webTaskMgr.Add(task)
  2047. logger.Debug(logSender, task.ID, "task timestamp updated, err: %v", err)
  2048. }
  2049. }
  2050. }