bolt.go 98 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920
  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. //go:build !nobolt
  15. // +build !nobolt
  16. package dataprovider
  17. import (
  18. "bytes"
  19. "crypto/x509"
  20. "encoding/json"
  21. "errors"
  22. "fmt"
  23. "net/netip"
  24. "path/filepath"
  25. "slices"
  26. "sort"
  27. "time"
  28. bolt "go.etcd.io/bbolt"
  29. "github.com/drakkan/sftpgo/v2/internal/logger"
  30. "github.com/drakkan/sftpgo/v2/internal/util"
  31. "github.com/drakkan/sftpgo/v2/internal/version"
  32. "github.com/drakkan/sftpgo/v2/internal/vfs"
  33. )
  34. const (
  35. boltDatabaseVersion = 29
  36. )
  37. var (
  38. usersBucket = []byte("users")
  39. groupsBucket = []byte("groups")
  40. foldersBucket = []byte("folders")
  41. adminsBucket = []byte("admins")
  42. apiKeysBucket = []byte("api_keys")
  43. sharesBucket = []byte("shares")
  44. actionsBucket = []byte("events_actions")
  45. rulesBucket = []byte("events_rules")
  46. rolesBucket = []byte("roles")
  47. ipListsBucket = []byte("ip_lists")
  48. configsBucket = []byte("configs")
  49. dbVersionBucket = []byte("db_version")
  50. dbVersionKey = []byte("version")
  51. configsKey = []byte("configs")
  52. boltBuckets = [][]byte{usersBucket, groupsBucket, foldersBucket, adminsBucket, apiKeysBucket,
  53. sharesBucket, actionsBucket, rulesBucket, rolesBucket, ipListsBucket, configsBucket, dbVersionBucket}
  54. )
  55. // BoltProvider defines the auth provider for bolt key/value store
  56. type BoltProvider struct {
  57. dbHandle *bolt.DB
  58. }
  59. func init() {
  60. version.AddFeature("+bolt")
  61. }
  62. func initializeBoltProvider(basePath string) error {
  63. var err error
  64. dbPath := config.Name
  65. if !util.IsFileInputValid(dbPath) {
  66. return fmt.Errorf("invalid database path: %q", dbPath)
  67. }
  68. if !filepath.IsAbs(dbPath) {
  69. dbPath = filepath.Join(basePath, dbPath)
  70. }
  71. dbHandle, err := bolt.Open(dbPath, 0600, &bolt.Options{
  72. NoGrowSync: false,
  73. FreelistType: bolt.FreelistArrayType,
  74. Timeout: 5 * time.Second})
  75. if err == nil {
  76. providerLog(logger.LevelDebug, "bolt key store handle created")
  77. for _, bucket := range boltBuckets {
  78. if err := dbHandle.Update(func(tx *bolt.Tx) error {
  79. _, e := tx.CreateBucketIfNotExists(bucket)
  80. return e
  81. }); err != nil {
  82. providerLog(logger.LevelError, "error creating bucket %q: %v", string(bucket), err)
  83. }
  84. }
  85. provider = &BoltProvider{dbHandle: dbHandle}
  86. } else {
  87. providerLog(logger.LevelError, "error creating bolt key/value store handler: %v", err)
  88. }
  89. return err
  90. }
  91. func (p *BoltProvider) checkAvailability() error {
  92. _, err := getBoltDatabaseVersion(p.dbHandle)
  93. return err
  94. }
  95. func (p *BoltProvider) validateUserAndTLSCert(username, protocol string, tlsCert *x509.Certificate) (User, error) {
  96. var user User
  97. if tlsCert == nil {
  98. return user, errors.New("TLS certificate cannot be null or empty")
  99. }
  100. user, err := p.userExists(username, "")
  101. if err != nil {
  102. providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
  103. return user, err
  104. }
  105. return checkUserAndTLSCertificate(&user, protocol, tlsCert)
  106. }
  107. func (p *BoltProvider) validateUserAndPass(username, password, ip, protocol string) (User, error) {
  108. user, err := p.userExists(username, "")
  109. if err != nil {
  110. providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
  111. return user, err
  112. }
  113. return checkUserAndPass(&user, password, ip, protocol)
  114. }
  115. func (p *BoltProvider) validateAdminAndPass(username, password, ip string) (Admin, error) {
  116. admin, err := p.adminExists(username)
  117. if err != nil {
  118. providerLog(logger.LevelWarn, "error authenticating admin %q: %v", username, err)
  119. return admin, err
  120. }
  121. err = admin.checkUserAndPass(password, ip)
  122. return admin, err
  123. }
  124. func (p *BoltProvider) validateUserAndPubKey(username string, pubKey []byte, isSSHCert bool) (User, string, error) {
  125. var user User
  126. if len(pubKey) == 0 {
  127. return user, "", errors.New("credentials cannot be null or empty")
  128. }
  129. user, err := p.userExists(username, "")
  130. if err != nil {
  131. providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
  132. return user, "", err
  133. }
  134. return checkUserAndPubKey(&user, pubKey, isSSHCert)
  135. }
  136. func (p *BoltProvider) updateAPIKeyLastUse(keyID string) error {
  137. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  138. bucket, err := p.getAPIKeysBucket(tx)
  139. if err != nil {
  140. return err
  141. }
  142. var u []byte
  143. if u = bucket.Get([]byte(keyID)); u == nil {
  144. return util.NewRecordNotFoundError(fmt.Sprintf("key %q does not exist, unable to update last use", keyID))
  145. }
  146. var apiKey APIKey
  147. err = json.Unmarshal(u, &apiKey)
  148. if err != nil {
  149. return err
  150. }
  151. apiKey.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
  152. buf, err := json.Marshal(apiKey)
  153. if err != nil {
  154. return err
  155. }
  156. err = bucket.Put([]byte(keyID), buf)
  157. if err != nil {
  158. providerLog(logger.LevelWarn, "error updating last use for key %q: %v", keyID, err)
  159. return err
  160. }
  161. providerLog(logger.LevelDebug, "last use updated for key %q", keyID)
  162. return nil
  163. })
  164. }
  165. func (p *BoltProvider) setUpdatedAt(username string) {
  166. p.dbHandle.Update(func(tx *bolt.Tx) error { //nolint:errcheck
  167. bucket, err := p.getUsersBucket(tx)
  168. if err != nil {
  169. return err
  170. }
  171. var u []byte
  172. if u = bucket.Get([]byte(username)); u == nil {
  173. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update updated at", username))
  174. }
  175. var user User
  176. err = json.Unmarshal(u, &user)
  177. if err != nil {
  178. return err
  179. }
  180. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  181. buf, err := json.Marshal(user)
  182. if err != nil {
  183. return err
  184. }
  185. err = bucket.Put([]byte(username), buf)
  186. if err == nil {
  187. providerLog(logger.LevelDebug, "updated at set for user %q", username)
  188. setLastUserUpdate()
  189. } else {
  190. providerLog(logger.LevelWarn, "error setting updated_at for user %q: %v", username, err)
  191. }
  192. return err
  193. })
  194. }
  195. func (p *BoltProvider) updateLastLogin(username string) error {
  196. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  197. bucket, err := p.getUsersBucket(tx)
  198. if err != nil {
  199. return err
  200. }
  201. var u []byte
  202. if u = bucket.Get([]byte(username)); u == nil {
  203. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update last login", username))
  204. }
  205. var user User
  206. err = json.Unmarshal(u, &user)
  207. if err != nil {
  208. return err
  209. }
  210. user.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
  211. buf, err := json.Marshal(user)
  212. if err != nil {
  213. return err
  214. }
  215. err = bucket.Put([]byte(username), buf)
  216. if err != nil {
  217. providerLog(logger.LevelWarn, "error updating last login for user %q: %v", username, err)
  218. } else {
  219. providerLog(logger.LevelDebug, "last login updated for user %q", username)
  220. }
  221. return err
  222. })
  223. }
  224. func (p *BoltProvider) updateAdminLastLogin(username string) error {
  225. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  226. bucket, err := p.getAdminsBucket(tx)
  227. if err != nil {
  228. return err
  229. }
  230. var a []byte
  231. if a = bucket.Get([]byte(username)); a == nil {
  232. return util.NewRecordNotFoundError(fmt.Sprintf("admin %q does not exist, unable to update last login", username))
  233. }
  234. var admin Admin
  235. err = json.Unmarshal(a, &admin)
  236. if err != nil {
  237. return err
  238. }
  239. admin.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
  240. buf, err := json.Marshal(admin)
  241. if err != nil {
  242. return err
  243. }
  244. err = bucket.Put([]byte(username), buf)
  245. if err == nil {
  246. providerLog(logger.LevelDebug, "last login updated for admin %q", username)
  247. return err
  248. }
  249. providerLog(logger.LevelWarn, "error updating last login for admin %q: %v", username, err)
  250. return err
  251. })
  252. }
  253. func (p *BoltProvider) updateTransferQuota(username string, uploadSize, downloadSize int64, reset bool) error {
  254. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  255. bucket, err := p.getUsersBucket(tx)
  256. if err != nil {
  257. return err
  258. }
  259. var u []byte
  260. if u = bucket.Get([]byte(username)); u == nil {
  261. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update transfer quota",
  262. username))
  263. }
  264. var user User
  265. err = json.Unmarshal(u, &user)
  266. if err != nil {
  267. return err
  268. }
  269. if !reset {
  270. user.UsedUploadDataTransfer += uploadSize
  271. user.UsedDownloadDataTransfer += downloadSize
  272. } else {
  273. user.UsedUploadDataTransfer = uploadSize
  274. user.UsedDownloadDataTransfer = downloadSize
  275. }
  276. user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  277. buf, err := json.Marshal(user)
  278. if err != nil {
  279. return err
  280. }
  281. err = bucket.Put([]byte(username), buf)
  282. providerLog(logger.LevelDebug, "transfer quota updated for user %q, ul increment: %v dl increment: %v is reset? %v",
  283. username, uploadSize, downloadSize, reset)
  284. return err
  285. })
  286. }
  287. func (p *BoltProvider) updateQuota(username string, filesAdd int, sizeAdd int64, reset bool) error {
  288. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  289. bucket, err := p.getUsersBucket(tx)
  290. if err != nil {
  291. return err
  292. }
  293. var u []byte
  294. if u = bucket.Get([]byte(username)); u == nil {
  295. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update quota", username))
  296. }
  297. var user User
  298. err = json.Unmarshal(u, &user)
  299. if err != nil {
  300. return err
  301. }
  302. if reset {
  303. user.UsedQuotaSize = sizeAdd
  304. user.UsedQuotaFiles = filesAdd
  305. } else {
  306. user.UsedQuotaSize += sizeAdd
  307. user.UsedQuotaFiles += filesAdd
  308. }
  309. user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  310. buf, err := json.Marshal(user)
  311. if err != nil {
  312. return err
  313. }
  314. err = bucket.Put([]byte(username), buf)
  315. providerLog(logger.LevelDebug, "quota updated for user %q, files increment: %v size increment: %v is reset? %v",
  316. username, filesAdd, sizeAdd, reset)
  317. return err
  318. })
  319. }
  320. func (p *BoltProvider) getUsedQuota(username string) (int, int64, int64, int64, error) {
  321. user, err := p.userExists(username, "")
  322. if err != nil {
  323. providerLog(logger.LevelError, "unable to get quota for user %v error: %v", username, err)
  324. return 0, 0, 0, 0, err
  325. }
  326. return user.UsedQuotaFiles, user.UsedQuotaSize, user.UsedUploadDataTransfer, user.UsedDownloadDataTransfer, err
  327. }
  328. func (p *BoltProvider) adminExists(username string) (Admin, error) {
  329. var admin Admin
  330. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  331. bucket, err := p.getAdminsBucket(tx)
  332. if err != nil {
  333. return err
  334. }
  335. a := bucket.Get([]byte(username))
  336. if a == nil {
  337. return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", username))
  338. }
  339. return json.Unmarshal(a, &admin)
  340. })
  341. return admin, err
  342. }
  343. func (p *BoltProvider) addAdmin(admin *Admin) error {
  344. err := admin.validate()
  345. if err != nil {
  346. return err
  347. }
  348. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  349. bucket, err := p.getAdminsBucket(tx)
  350. if err != nil {
  351. return err
  352. }
  353. groupBucket, err := p.getGroupsBucket(tx)
  354. if err != nil {
  355. return err
  356. }
  357. rolesBucket, err := p.getRolesBucket(tx)
  358. if err != nil {
  359. return err
  360. }
  361. if a := bucket.Get([]byte(admin.Username)); a != nil {
  362. return util.NewI18nError(
  363. fmt.Errorf("%w: admin %q already exists", ErrDuplicatedKey, admin.Username),
  364. util.I18nErrorDuplicatedUsername,
  365. )
  366. }
  367. id, err := bucket.NextSequence()
  368. if err != nil {
  369. return err
  370. }
  371. admin.ID = int64(id)
  372. admin.LastLogin = 0
  373. admin.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  374. admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  375. sort.Slice(admin.Groups, func(i, j int) bool {
  376. return admin.Groups[i].Name < admin.Groups[j].Name
  377. })
  378. for idx := range admin.Groups {
  379. err = p.addAdminToGroupMapping(admin.Username, admin.Groups[idx].Name, groupBucket)
  380. if err != nil {
  381. return err
  382. }
  383. }
  384. if err = p.addAdminToRole(admin.Username, admin.Role, rolesBucket); err != nil {
  385. return err
  386. }
  387. buf, err := json.Marshal(admin)
  388. if err != nil {
  389. return err
  390. }
  391. return bucket.Put([]byte(admin.Username), buf)
  392. })
  393. }
  394. func (p *BoltProvider) updateAdmin(admin *Admin) error {
  395. err := admin.validate()
  396. if err != nil {
  397. return err
  398. }
  399. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  400. bucket, err := p.getAdminsBucket(tx)
  401. if err != nil {
  402. return err
  403. }
  404. groupBucket, err := p.getGroupsBucket(tx)
  405. if err != nil {
  406. return err
  407. }
  408. rolesBucket, err := p.getRolesBucket(tx)
  409. if err != nil {
  410. return err
  411. }
  412. var a []byte
  413. if a = bucket.Get([]byte(admin.Username)); a == nil {
  414. return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", admin.Username))
  415. }
  416. var oldAdmin Admin
  417. err = json.Unmarshal(a, &oldAdmin)
  418. if err != nil {
  419. return err
  420. }
  421. if err = p.removeAdminFromRole(oldAdmin.Username, oldAdmin.Role, rolesBucket); err != nil {
  422. return err
  423. }
  424. for idx := range oldAdmin.Groups {
  425. err = p.removeAdminFromGroupMapping(oldAdmin.Username, oldAdmin.Groups[idx].Name, groupBucket)
  426. if err != nil {
  427. return err
  428. }
  429. }
  430. if err = p.addAdminToRole(admin.Username, admin.Role, rolesBucket); err != nil {
  431. return err
  432. }
  433. sort.Slice(admin.Groups, func(i, j int) bool {
  434. return admin.Groups[i].Name < admin.Groups[j].Name
  435. })
  436. for idx := range admin.Groups {
  437. err = p.addAdminToGroupMapping(admin.Username, admin.Groups[idx].Name, groupBucket)
  438. if err != nil {
  439. return err
  440. }
  441. }
  442. admin.ID = oldAdmin.ID
  443. admin.CreatedAt = oldAdmin.CreatedAt
  444. admin.LastLogin = oldAdmin.LastLogin
  445. admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  446. buf, err := json.Marshal(admin)
  447. if err != nil {
  448. return err
  449. }
  450. return bucket.Put([]byte(admin.Username), buf)
  451. })
  452. }
  453. func (p *BoltProvider) deleteAdmin(admin Admin) error {
  454. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  455. bucket, err := p.getAdminsBucket(tx)
  456. if err != nil {
  457. return err
  458. }
  459. var a []byte
  460. if a = bucket.Get([]byte(admin.Username)); a == nil {
  461. return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", admin.Username))
  462. }
  463. var oldAdmin Admin
  464. err = json.Unmarshal(a, &oldAdmin)
  465. if err != nil {
  466. return err
  467. }
  468. if len(oldAdmin.Groups) > 0 {
  469. groupBucket, err := p.getGroupsBucket(tx)
  470. if err != nil {
  471. return err
  472. }
  473. for idx := range oldAdmin.Groups {
  474. err = p.removeAdminFromGroupMapping(oldAdmin.Username, oldAdmin.Groups[idx].Name, groupBucket)
  475. if err != nil {
  476. return err
  477. }
  478. }
  479. }
  480. if oldAdmin.Role != "" {
  481. rolesBucket, err := p.getRolesBucket(tx)
  482. if err != nil {
  483. return err
  484. }
  485. if err = p.removeAdminFromRole(oldAdmin.Username, oldAdmin.Role, rolesBucket); err != nil {
  486. return err
  487. }
  488. }
  489. if err := p.deleteRelatedAPIKey(tx, admin.Username, APIKeyScopeAdmin); err != nil {
  490. return err
  491. }
  492. return bucket.Delete([]byte(admin.Username))
  493. })
  494. }
  495. func (p *BoltProvider) getAdmins(limit int, offset int, order string) ([]Admin, error) {
  496. admins := make([]Admin, 0, limit)
  497. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  498. bucket, err := p.getAdminsBucket(tx)
  499. if err != nil {
  500. return err
  501. }
  502. cursor := bucket.Cursor()
  503. itNum := 0
  504. if order == OrderASC {
  505. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  506. itNum++
  507. if itNum <= offset {
  508. continue
  509. }
  510. var admin Admin
  511. err = json.Unmarshal(v, &admin)
  512. if err != nil {
  513. return err
  514. }
  515. admin.HideConfidentialData()
  516. admins = append(admins, admin)
  517. if len(admins) >= limit {
  518. break
  519. }
  520. }
  521. } else {
  522. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  523. itNum++
  524. if itNum <= offset {
  525. continue
  526. }
  527. var admin Admin
  528. err = json.Unmarshal(v, &admin)
  529. if err != nil {
  530. return err
  531. }
  532. admin.HideConfidentialData()
  533. admins = append(admins, admin)
  534. if len(admins) >= limit {
  535. break
  536. }
  537. }
  538. }
  539. return err
  540. })
  541. return admins, err
  542. }
  543. func (p *BoltProvider) dumpAdmins() ([]Admin, error) {
  544. admins := make([]Admin, 0, 30)
  545. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  546. bucket, err := p.getAdminsBucket(tx)
  547. if err != nil {
  548. return err
  549. }
  550. cursor := bucket.Cursor()
  551. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  552. var admin Admin
  553. err = json.Unmarshal(v, &admin)
  554. if err != nil {
  555. return err
  556. }
  557. admins = append(admins, admin)
  558. }
  559. return err
  560. })
  561. return admins, err
  562. }
  563. func (p *BoltProvider) userExists(username, role string) (User, error) {
  564. var user User
  565. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  566. bucket, err := p.getUsersBucket(tx)
  567. if err != nil {
  568. return err
  569. }
  570. u := bucket.Get([]byte(username))
  571. if u == nil {
  572. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
  573. }
  574. foldersBucket, err := p.getFoldersBucket(tx)
  575. if err != nil {
  576. return err
  577. }
  578. user, err = p.joinUserAndFolders(u, foldersBucket)
  579. if err != nil {
  580. return err
  581. }
  582. if !user.hasRole(role) {
  583. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
  584. }
  585. return nil
  586. })
  587. return user, err
  588. }
  589. func (p *BoltProvider) addUser(user *User) error {
  590. err := ValidateUser(user)
  591. if err != nil {
  592. return err
  593. }
  594. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  595. bucket, err := p.getUsersBucket(tx)
  596. if err != nil {
  597. return err
  598. }
  599. foldersBucket, err := p.getFoldersBucket(tx)
  600. if err != nil {
  601. return err
  602. }
  603. groupBucket, err := p.getGroupsBucket(tx)
  604. if err != nil {
  605. return err
  606. }
  607. rolesBucket, err := p.getRolesBucket(tx)
  608. if err != nil {
  609. return err
  610. }
  611. if u := bucket.Get([]byte(user.Username)); u != nil {
  612. return util.NewI18nError(
  613. fmt.Errorf("%w: username %v already exists", ErrDuplicatedKey, user.Username),
  614. util.I18nErrorDuplicatedUsername,
  615. )
  616. }
  617. id, err := bucket.NextSequence()
  618. if err != nil {
  619. return err
  620. }
  621. user.ID = int64(id)
  622. user.LastQuotaUpdate = 0
  623. user.UsedQuotaSize = 0
  624. user.UsedQuotaFiles = 0
  625. user.UsedUploadDataTransfer = 0
  626. user.UsedDownloadDataTransfer = 0
  627. user.LastLogin = 0
  628. user.FirstDownload = 0
  629. user.FirstUpload = 0
  630. user.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  631. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  632. if err := p.addUserToRole(user.Username, user.Role, rolesBucket); err != nil {
  633. return err
  634. }
  635. sort.Slice(user.VirtualFolders, func(i, j int) bool {
  636. return user.VirtualFolders[i].Name < user.VirtualFolders[j].Name
  637. })
  638. for idx := range user.VirtualFolders {
  639. err = p.addRelationToFolderMapping(user.VirtualFolders[idx].Name, user, nil, foldersBucket)
  640. if err != nil {
  641. return err
  642. }
  643. }
  644. sort.Slice(user.Groups, func(i, j int) bool {
  645. return user.Groups[i].Name < user.Groups[j].Name
  646. })
  647. for idx := range user.Groups {
  648. err = p.addUserToGroupMapping(user.Username, user.Groups[idx].Name, groupBucket)
  649. if err != nil {
  650. return err
  651. }
  652. }
  653. buf, err := json.Marshal(user)
  654. if err != nil {
  655. return err
  656. }
  657. return bucket.Put([]byte(user.Username), buf)
  658. })
  659. }
  660. func (p *BoltProvider) updateUser(user *User) error {
  661. err := ValidateUser(user)
  662. if err != nil {
  663. return err
  664. }
  665. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  666. bucket, err := p.getUsersBucket(tx)
  667. if err != nil {
  668. return err
  669. }
  670. var u []byte
  671. if u = bucket.Get([]byte(user.Username)); u == nil {
  672. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", user.Username))
  673. }
  674. var oldUser User
  675. err = json.Unmarshal(u, &oldUser)
  676. if err != nil {
  677. return err
  678. }
  679. if err = p.updateUserRelations(tx, user, oldUser); err != nil {
  680. return err
  681. }
  682. user.ID = oldUser.ID
  683. user.LastQuotaUpdate = oldUser.LastQuotaUpdate
  684. user.UsedQuotaSize = oldUser.UsedQuotaSize
  685. user.UsedQuotaFiles = oldUser.UsedQuotaFiles
  686. user.UsedUploadDataTransfer = oldUser.UsedUploadDataTransfer
  687. user.UsedDownloadDataTransfer = oldUser.UsedDownloadDataTransfer
  688. user.LastLogin = oldUser.LastLogin
  689. user.FirstDownload = oldUser.FirstDownload
  690. user.FirstUpload = oldUser.FirstUpload
  691. user.CreatedAt = oldUser.CreatedAt
  692. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  693. buf, err := json.Marshal(user)
  694. if err != nil {
  695. return err
  696. }
  697. err = bucket.Put([]byte(user.Username), buf)
  698. if err == nil {
  699. setLastUserUpdate()
  700. }
  701. return err
  702. })
  703. }
  704. func (p *BoltProvider) deleteUser(user User, _ bool) error {
  705. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  706. bucket, err := p.getUsersBucket(tx)
  707. if err != nil {
  708. return err
  709. }
  710. foldersBucket, err := p.getFoldersBucket(tx)
  711. if err != nil {
  712. return err
  713. }
  714. groupBucket, err := p.getGroupsBucket(tx)
  715. if err != nil {
  716. return err
  717. }
  718. rolesBucket, err := p.getRolesBucket(tx)
  719. if err != nil {
  720. return err
  721. }
  722. var u []byte
  723. if u = bucket.Get([]byte(user.Username)); u == nil {
  724. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", user.Username))
  725. }
  726. var oldUser User
  727. err = json.Unmarshal(u, &oldUser)
  728. if err != nil {
  729. return err
  730. }
  731. if err := p.removeUserFromRole(oldUser.Username, oldUser.Role, rolesBucket); err != nil {
  732. return err
  733. }
  734. for idx := range oldUser.VirtualFolders {
  735. err = p.removeRelationFromFolderMapping(oldUser.VirtualFolders[idx], oldUser.Username, "", foldersBucket)
  736. if err != nil {
  737. return err
  738. }
  739. }
  740. for idx := range oldUser.Groups {
  741. err = p.removeUserFromGroupMapping(oldUser.Username, oldUser.Groups[idx].Name, groupBucket)
  742. if err != nil {
  743. return err
  744. }
  745. }
  746. if err := p.deleteRelatedAPIKey(tx, user.Username, APIKeyScopeUser); err != nil {
  747. return err
  748. }
  749. if err := p.deleteRelatedShares(tx, user.Username); err != nil {
  750. return err
  751. }
  752. return bucket.Delete([]byte(user.Username))
  753. })
  754. }
  755. func (p *BoltProvider) updateUserPassword(username, password string) error {
  756. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  757. bucket, err := p.getUsersBucket(tx)
  758. if err != nil {
  759. return err
  760. }
  761. var u []byte
  762. if u = bucket.Get([]byte(username)); u == nil {
  763. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
  764. }
  765. var user User
  766. err = json.Unmarshal(u, &user)
  767. if err != nil {
  768. return err
  769. }
  770. user.Password = password
  771. user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  772. buf, err := json.Marshal(user)
  773. if err != nil {
  774. return err
  775. }
  776. return bucket.Put([]byte(username), buf)
  777. })
  778. }
  779. func (p *BoltProvider) dumpUsers() ([]User, error) {
  780. users := make([]User, 0, 100)
  781. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  782. bucket, err := p.getUsersBucket(tx)
  783. if err != nil {
  784. return err
  785. }
  786. foldersBucket, err := p.getFoldersBucket(tx)
  787. if err != nil {
  788. return err
  789. }
  790. cursor := bucket.Cursor()
  791. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  792. user, err := p.joinUserAndFolders(v, foldersBucket)
  793. if err != nil {
  794. return err
  795. }
  796. users = append(users, user)
  797. }
  798. return err
  799. })
  800. return users, err
  801. }
  802. func (p *BoltProvider) getRecentlyUpdatedUsers(after int64) ([]User, error) {
  803. if getLastUserUpdate() < after {
  804. return nil, nil
  805. }
  806. users := make([]User, 0, 10)
  807. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  808. bucket, err := p.getUsersBucket(tx)
  809. if err != nil {
  810. return err
  811. }
  812. foldersBucket, err := p.getFoldersBucket(tx)
  813. if err != nil {
  814. return err
  815. }
  816. groupsBucket, err := p.getGroupsBucket(tx)
  817. if err != nil {
  818. return err
  819. }
  820. cursor := bucket.Cursor()
  821. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  822. var user User
  823. err := json.Unmarshal(v, &user)
  824. if err != nil {
  825. return err
  826. }
  827. if user.UpdatedAt < after {
  828. continue
  829. }
  830. if len(user.VirtualFolders) > 0 {
  831. var folders []vfs.VirtualFolder
  832. for idx := range user.VirtualFolders {
  833. folder := &user.VirtualFolders[idx]
  834. baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
  835. if err != nil {
  836. continue
  837. }
  838. folder.BaseVirtualFolder = baseFolder
  839. folders = append(folders, *folder)
  840. }
  841. user.VirtualFolders = folders
  842. }
  843. if len(user.Groups) > 0 {
  844. groupMapping := make(map[string]Group)
  845. for idx := range user.Groups {
  846. group, err := p.groupExistsInternal(user.Groups[idx].Name, groupsBucket)
  847. if err != nil {
  848. continue
  849. }
  850. groupMapping[group.Name] = group
  851. }
  852. user.applyGroupSettings(groupMapping)
  853. }
  854. user.SetEmptySecretsIfNil()
  855. users = append(users, user)
  856. }
  857. return err
  858. })
  859. return users, err
  860. }
  861. func (p *BoltProvider) getUsersForQuotaCheck(toFetch map[string]bool) ([]User, error) {
  862. users := make([]User, 0, 10)
  863. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  864. bucket, err := p.getUsersBucket(tx)
  865. if err != nil {
  866. return err
  867. }
  868. foldersBucket, err := p.getFoldersBucket(tx)
  869. if err != nil {
  870. return err
  871. }
  872. groupsBucket, err := p.getGroupsBucket(tx)
  873. if err != nil {
  874. return err
  875. }
  876. cursor := bucket.Cursor()
  877. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  878. var user User
  879. err := json.Unmarshal(v, &user)
  880. if err != nil {
  881. return err
  882. }
  883. if needFolders, ok := toFetch[user.Username]; ok {
  884. if needFolders && len(user.VirtualFolders) > 0 {
  885. var folders []vfs.VirtualFolder
  886. for idx := range user.VirtualFolders {
  887. folder := &user.VirtualFolders[idx]
  888. baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
  889. if err != nil {
  890. continue
  891. }
  892. folder.BaseVirtualFolder = baseFolder
  893. folders = append(folders, *folder)
  894. }
  895. user.VirtualFolders = folders
  896. }
  897. if len(user.Groups) > 0 {
  898. groupMapping := make(map[string]Group)
  899. for idx := range user.Groups {
  900. group, err := p.groupExistsInternal(user.Groups[idx].Name, groupsBucket)
  901. if err != nil {
  902. continue
  903. }
  904. groupMapping[group.Name] = group
  905. }
  906. user.applyGroupSettings(groupMapping)
  907. }
  908. user.SetEmptySecretsIfNil()
  909. user.PrepareForRendering()
  910. users = append(users, user)
  911. }
  912. }
  913. return nil
  914. })
  915. return users, err
  916. }
  917. func (p *BoltProvider) getUsers(limit int, offset int, order, role string) ([]User, error) {
  918. users := make([]User, 0, limit)
  919. var err error
  920. if limit <= 0 {
  921. return users, err
  922. }
  923. err = p.dbHandle.View(func(tx *bolt.Tx) error {
  924. bucket, err := p.getUsersBucket(tx)
  925. if err != nil {
  926. return err
  927. }
  928. foldersBucket, err := p.getFoldersBucket(tx)
  929. if err != nil {
  930. return err
  931. }
  932. cursor := bucket.Cursor()
  933. itNum := 0
  934. if order == OrderASC {
  935. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  936. itNum++
  937. if itNum <= offset {
  938. continue
  939. }
  940. user, err := p.joinUserAndFolders(v, foldersBucket)
  941. if err != nil {
  942. return err
  943. }
  944. if !user.hasRole(role) {
  945. continue
  946. }
  947. user.PrepareForRendering()
  948. users = append(users, user)
  949. if len(users) >= limit {
  950. break
  951. }
  952. }
  953. } else {
  954. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  955. itNum++
  956. if itNum <= offset {
  957. continue
  958. }
  959. user, err := p.joinUserAndFolders(v, foldersBucket)
  960. if err != nil {
  961. return err
  962. }
  963. if !user.hasRole(role) {
  964. continue
  965. }
  966. user.PrepareForRendering()
  967. users = append(users, user)
  968. if len(users) >= limit {
  969. break
  970. }
  971. }
  972. }
  973. return err
  974. })
  975. return users, err
  976. }
  977. func (p *BoltProvider) dumpFolders() ([]vfs.BaseVirtualFolder, error) {
  978. folders := make([]vfs.BaseVirtualFolder, 0, 50)
  979. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  980. bucket, err := p.getFoldersBucket(tx)
  981. if err != nil {
  982. return err
  983. }
  984. cursor := bucket.Cursor()
  985. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  986. var folder vfs.BaseVirtualFolder
  987. err = json.Unmarshal(v, &folder)
  988. if err != nil {
  989. return err
  990. }
  991. folders = append(folders, folder)
  992. }
  993. return err
  994. })
  995. return folders, err
  996. }
  997. func (p *BoltProvider) getFolders(limit, offset int, order string, _ bool) ([]vfs.BaseVirtualFolder, error) {
  998. folders := make([]vfs.BaseVirtualFolder, 0, limit)
  999. var err error
  1000. if limit <= 0 {
  1001. return folders, err
  1002. }
  1003. err = p.dbHandle.View(func(tx *bolt.Tx) error {
  1004. bucket, err := p.getFoldersBucket(tx)
  1005. if err != nil {
  1006. return err
  1007. }
  1008. cursor := bucket.Cursor()
  1009. itNum := 0
  1010. if order == OrderASC {
  1011. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1012. itNum++
  1013. if itNum <= offset {
  1014. continue
  1015. }
  1016. var folder vfs.BaseVirtualFolder
  1017. err = json.Unmarshal(v, &folder)
  1018. if err != nil {
  1019. return err
  1020. }
  1021. folder.PrepareForRendering()
  1022. folders = append(folders, folder)
  1023. if len(folders) >= limit {
  1024. break
  1025. }
  1026. }
  1027. } else {
  1028. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  1029. itNum++
  1030. if itNum <= offset {
  1031. continue
  1032. }
  1033. var folder vfs.BaseVirtualFolder
  1034. err = json.Unmarshal(v, &folder)
  1035. if err != nil {
  1036. return err
  1037. }
  1038. folder.PrepareForRendering()
  1039. folders = append(folders, folder)
  1040. if len(folders) >= limit {
  1041. break
  1042. }
  1043. }
  1044. }
  1045. return err
  1046. })
  1047. return folders, err
  1048. }
  1049. func (p *BoltProvider) getFolderByName(name string) (vfs.BaseVirtualFolder, error) {
  1050. var folder vfs.BaseVirtualFolder
  1051. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1052. bucket, err := p.getFoldersBucket(tx)
  1053. if err != nil {
  1054. return err
  1055. }
  1056. folder, err = p.folderExistsInternal(name, bucket)
  1057. return err
  1058. })
  1059. return folder, err
  1060. }
  1061. func (p *BoltProvider) addFolder(folder *vfs.BaseVirtualFolder) error {
  1062. err := ValidateFolder(folder)
  1063. if err != nil {
  1064. return err
  1065. }
  1066. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1067. bucket, err := p.getFoldersBucket(tx)
  1068. if err != nil {
  1069. return err
  1070. }
  1071. if f := bucket.Get([]byte(folder.Name)); f != nil {
  1072. return util.NewI18nError(
  1073. fmt.Errorf("%w: folder %q already exists", ErrDuplicatedKey, folder.Name),
  1074. util.I18nErrorDuplicatedUsername,
  1075. )
  1076. }
  1077. folder.Users = nil
  1078. folder.Groups = nil
  1079. return p.addFolderInternal(*folder, bucket)
  1080. })
  1081. }
  1082. func (p *BoltProvider) updateFolder(folder *vfs.BaseVirtualFolder) error {
  1083. err := ValidateFolder(folder)
  1084. if err != nil {
  1085. return err
  1086. }
  1087. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1088. bucket, err := p.getFoldersBucket(tx)
  1089. if err != nil {
  1090. return err
  1091. }
  1092. var f []byte
  1093. if f = bucket.Get([]byte(folder.Name)); f == nil {
  1094. return util.NewRecordNotFoundError(fmt.Sprintf("folder %v does not exist", folder.Name))
  1095. }
  1096. var oldFolder vfs.BaseVirtualFolder
  1097. err = json.Unmarshal(f, &oldFolder)
  1098. if err != nil {
  1099. return err
  1100. }
  1101. folder.ID = oldFolder.ID
  1102. folder.LastQuotaUpdate = oldFolder.LastQuotaUpdate
  1103. folder.UsedQuotaFiles = oldFolder.UsedQuotaFiles
  1104. folder.UsedQuotaSize = oldFolder.UsedQuotaSize
  1105. folder.Users = oldFolder.Users
  1106. folder.Groups = oldFolder.Groups
  1107. buf, err := json.Marshal(folder)
  1108. if err != nil {
  1109. return err
  1110. }
  1111. return bucket.Put([]byte(folder.Name), buf)
  1112. })
  1113. }
  1114. func (p *BoltProvider) deleteFolderMappings(folder vfs.BaseVirtualFolder, usersBucket, groupsBucket *bolt.Bucket) error {
  1115. for _, username := range folder.Users {
  1116. var u []byte
  1117. if u = usersBucket.Get([]byte(username)); u == nil {
  1118. continue
  1119. }
  1120. var user User
  1121. err := json.Unmarshal(u, &user)
  1122. if err != nil {
  1123. return err
  1124. }
  1125. var folders []vfs.VirtualFolder
  1126. for _, userFolder := range user.VirtualFolders {
  1127. if folder.Name != userFolder.Name {
  1128. folders = append(folders, userFolder)
  1129. }
  1130. }
  1131. user.VirtualFolders = folders
  1132. buf, err := json.Marshal(user)
  1133. if err != nil {
  1134. return err
  1135. }
  1136. err = usersBucket.Put([]byte(user.Username), buf)
  1137. if err != nil {
  1138. return err
  1139. }
  1140. }
  1141. for _, groupname := range folder.Groups {
  1142. var u []byte
  1143. if u = groupsBucket.Get([]byte(groupname)); u == nil {
  1144. continue
  1145. }
  1146. var group Group
  1147. err := json.Unmarshal(u, &group)
  1148. if err != nil {
  1149. return err
  1150. }
  1151. var folders []vfs.VirtualFolder
  1152. for _, groupFolder := range group.VirtualFolders {
  1153. if folder.Name != groupFolder.Name {
  1154. folders = append(folders, groupFolder)
  1155. }
  1156. }
  1157. group.VirtualFolders = folders
  1158. buf, err := json.Marshal(group)
  1159. if err != nil {
  1160. return err
  1161. }
  1162. err = groupsBucket.Put([]byte(group.Name), buf)
  1163. if err != nil {
  1164. return err
  1165. }
  1166. }
  1167. return nil
  1168. }
  1169. func (p *BoltProvider) deleteFolder(baseFolder vfs.BaseVirtualFolder) error {
  1170. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1171. bucket, err := p.getFoldersBucket(tx)
  1172. if err != nil {
  1173. return err
  1174. }
  1175. usersBucket, err := p.getUsersBucket(tx)
  1176. if err != nil {
  1177. return err
  1178. }
  1179. groupsBucket, err := p.getGroupsBucket(tx)
  1180. if err != nil {
  1181. return err
  1182. }
  1183. var f []byte
  1184. if f = bucket.Get([]byte(baseFolder.Name)); f == nil {
  1185. return util.NewRecordNotFoundError(fmt.Sprintf("folder %v does not exist", baseFolder.Name))
  1186. }
  1187. var folder vfs.BaseVirtualFolder
  1188. err = json.Unmarshal(f, &folder)
  1189. if err != nil {
  1190. return err
  1191. }
  1192. if err = p.deleteFolderMappings(folder, usersBucket, groupsBucket); err != nil {
  1193. return err
  1194. }
  1195. return bucket.Delete([]byte(folder.Name))
  1196. })
  1197. }
  1198. func (p *BoltProvider) updateFolderQuota(name string, filesAdd int, sizeAdd int64, reset bool) error {
  1199. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1200. bucket, err := p.getFoldersBucket(tx)
  1201. if err != nil {
  1202. return err
  1203. }
  1204. var f []byte
  1205. if f = bucket.Get([]byte(name)); f == nil {
  1206. return util.NewRecordNotFoundError(fmt.Sprintf("folder %q does not exist, unable to update quota", name))
  1207. }
  1208. var folder vfs.BaseVirtualFolder
  1209. err = json.Unmarshal(f, &folder)
  1210. if err != nil {
  1211. return err
  1212. }
  1213. if reset {
  1214. folder.UsedQuotaSize = sizeAdd
  1215. folder.UsedQuotaFiles = filesAdd
  1216. } else {
  1217. folder.UsedQuotaSize += sizeAdd
  1218. folder.UsedQuotaFiles += filesAdd
  1219. }
  1220. folder.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
  1221. buf, err := json.Marshal(folder)
  1222. if err != nil {
  1223. return err
  1224. }
  1225. return bucket.Put([]byte(folder.Name), buf)
  1226. })
  1227. }
  1228. func (p *BoltProvider) getUsedFolderQuota(name string) (int, int64, error) {
  1229. folder, err := p.getFolderByName(name)
  1230. if err != nil {
  1231. providerLog(logger.LevelError, "unable to get quota for folder %q error: %v", name, err)
  1232. return 0, 0, err
  1233. }
  1234. return folder.UsedQuotaFiles, folder.UsedQuotaSize, err
  1235. }
  1236. func (p *BoltProvider) getGroups(limit, offset int, order string, _ bool) ([]Group, error) {
  1237. groups := make([]Group, 0, limit)
  1238. var err error
  1239. if limit <= 0 {
  1240. return groups, err
  1241. }
  1242. err = p.dbHandle.View(func(tx *bolt.Tx) error {
  1243. bucket, err := p.getGroupsBucket(tx)
  1244. if err != nil {
  1245. return err
  1246. }
  1247. foldersBucket, err := p.getFoldersBucket(tx)
  1248. if err != nil {
  1249. return err
  1250. }
  1251. cursor := bucket.Cursor()
  1252. itNum := 0
  1253. if order == OrderASC {
  1254. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1255. itNum++
  1256. if itNum <= offset {
  1257. continue
  1258. }
  1259. var group Group
  1260. group, err = p.joinGroupAndFolders(v, foldersBucket)
  1261. if err != nil {
  1262. return err
  1263. }
  1264. group.PrepareForRendering()
  1265. groups = append(groups, group)
  1266. if len(groups) >= limit {
  1267. break
  1268. }
  1269. }
  1270. } else {
  1271. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  1272. itNum++
  1273. if itNum <= offset {
  1274. continue
  1275. }
  1276. var group Group
  1277. group, err = p.joinGroupAndFolders(v, foldersBucket)
  1278. if err != nil {
  1279. return err
  1280. }
  1281. group.PrepareForRendering()
  1282. groups = append(groups, group)
  1283. if len(groups) >= limit {
  1284. break
  1285. }
  1286. }
  1287. }
  1288. return err
  1289. })
  1290. return groups, err
  1291. }
  1292. func (p *BoltProvider) getGroupsWithNames(names []string) ([]Group, error) {
  1293. var groups []Group
  1294. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1295. bucket, err := p.getGroupsBucket(tx)
  1296. if err != nil {
  1297. return err
  1298. }
  1299. foldersBucket, err := p.getFoldersBucket(tx)
  1300. if err != nil {
  1301. return err
  1302. }
  1303. for _, name := range names {
  1304. g := bucket.Get([]byte(name))
  1305. if g == nil {
  1306. continue
  1307. }
  1308. group, err := p.joinGroupAndFolders(g, foldersBucket)
  1309. if err != nil {
  1310. return err
  1311. }
  1312. groups = append(groups, group)
  1313. }
  1314. return nil
  1315. })
  1316. return groups, err
  1317. }
  1318. func (p *BoltProvider) getUsersInGroups(names []string) ([]string, error) {
  1319. var usernames []string
  1320. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1321. bucket, err := p.getGroupsBucket(tx)
  1322. if err != nil {
  1323. return err
  1324. }
  1325. for _, name := range names {
  1326. g := bucket.Get([]byte(name))
  1327. if g == nil {
  1328. continue
  1329. }
  1330. var group Group
  1331. err := json.Unmarshal(g, &group)
  1332. if err != nil {
  1333. return err
  1334. }
  1335. usernames = append(usernames, group.Users...)
  1336. }
  1337. return nil
  1338. })
  1339. return usernames, err
  1340. }
  1341. func (p *BoltProvider) groupExists(name string) (Group, error) {
  1342. var group Group
  1343. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1344. bucket, err := p.getGroupsBucket(tx)
  1345. if err != nil {
  1346. return err
  1347. }
  1348. g := bucket.Get([]byte(name))
  1349. if g == nil {
  1350. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", name))
  1351. }
  1352. foldersBucket, err := p.getFoldersBucket(tx)
  1353. if err != nil {
  1354. return err
  1355. }
  1356. group, err = p.joinGroupAndFolders(g, foldersBucket)
  1357. return err
  1358. })
  1359. return group, err
  1360. }
  1361. func (p *BoltProvider) addGroup(group *Group) error {
  1362. if err := group.validate(); err != nil {
  1363. return err
  1364. }
  1365. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1366. bucket, err := p.getGroupsBucket(tx)
  1367. if err != nil {
  1368. return err
  1369. }
  1370. foldersBucket, err := p.getFoldersBucket(tx)
  1371. if err != nil {
  1372. return err
  1373. }
  1374. if u := bucket.Get([]byte(group.Name)); u != nil {
  1375. return util.NewI18nError(
  1376. fmt.Errorf("%w: group %q already exists", ErrDuplicatedKey, group.Name),
  1377. util.I18nErrorDuplicatedUsername,
  1378. )
  1379. }
  1380. id, err := bucket.NextSequence()
  1381. if err != nil {
  1382. return err
  1383. }
  1384. group.ID = int64(id)
  1385. group.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1386. group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1387. group.Users = nil
  1388. group.Admins = nil
  1389. sort.Slice(group.VirtualFolders, func(i, j int) bool {
  1390. return group.VirtualFolders[i].Name < group.VirtualFolders[j].Name
  1391. })
  1392. for idx := range group.VirtualFolders {
  1393. err = p.addRelationToFolderMapping(group.VirtualFolders[idx].Name, nil, group, foldersBucket)
  1394. if err != nil {
  1395. return err
  1396. }
  1397. }
  1398. buf, err := json.Marshal(group)
  1399. if err != nil {
  1400. return err
  1401. }
  1402. return bucket.Put([]byte(group.Name), buf)
  1403. })
  1404. }
  1405. func (p *BoltProvider) updateGroup(group *Group) error {
  1406. if err := group.validate(); err != nil {
  1407. return err
  1408. }
  1409. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1410. bucket, err := p.getGroupsBucket(tx)
  1411. if err != nil {
  1412. return err
  1413. }
  1414. foldersBucket, err := p.getFoldersBucket(tx)
  1415. if err != nil {
  1416. return err
  1417. }
  1418. var g []byte
  1419. if g = bucket.Get([]byte(group.Name)); g == nil {
  1420. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", group.Name))
  1421. }
  1422. var oldGroup Group
  1423. err = json.Unmarshal(g, &oldGroup)
  1424. if err != nil {
  1425. return err
  1426. }
  1427. for idx := range oldGroup.VirtualFolders {
  1428. err = p.removeRelationFromFolderMapping(oldGroup.VirtualFolders[idx], "", oldGroup.Name, foldersBucket)
  1429. if err != nil {
  1430. return err
  1431. }
  1432. }
  1433. sort.Slice(group.VirtualFolders, func(i, j int) bool {
  1434. return group.VirtualFolders[i].Name < group.VirtualFolders[j].Name
  1435. })
  1436. for idx := range group.VirtualFolders {
  1437. err = p.addRelationToFolderMapping(group.VirtualFolders[idx].Name, nil, group, foldersBucket)
  1438. if err != nil {
  1439. return err
  1440. }
  1441. }
  1442. group.ID = oldGroup.ID
  1443. group.CreatedAt = oldGroup.CreatedAt
  1444. group.Users = oldGroup.Users
  1445. group.Admins = oldGroup.Admins
  1446. group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1447. buf, err := json.Marshal(group)
  1448. if err != nil {
  1449. return err
  1450. }
  1451. return bucket.Put([]byte(group.Name), buf)
  1452. })
  1453. }
  1454. func (p *BoltProvider) deleteGroup(group Group) error {
  1455. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1456. bucket, err := p.getGroupsBucket(tx)
  1457. if err != nil {
  1458. return err
  1459. }
  1460. var g []byte
  1461. if g = bucket.Get([]byte(group.Name)); g == nil {
  1462. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", group.Name))
  1463. }
  1464. var oldGroup Group
  1465. err = json.Unmarshal(g, &oldGroup)
  1466. if err != nil {
  1467. return err
  1468. }
  1469. if len(oldGroup.Users) > 0 {
  1470. return util.NewValidationError(fmt.Sprintf("the group %q is referenced, it cannot be removed", oldGroup.Name))
  1471. }
  1472. if len(oldGroup.VirtualFolders) > 0 {
  1473. foldersBucket, err := p.getFoldersBucket(tx)
  1474. if err != nil {
  1475. return err
  1476. }
  1477. for idx := range oldGroup.VirtualFolders {
  1478. err = p.removeRelationFromFolderMapping(oldGroup.VirtualFolders[idx], "", oldGroup.Name, foldersBucket)
  1479. if err != nil {
  1480. return err
  1481. }
  1482. }
  1483. }
  1484. if len(oldGroup.Admins) > 0 {
  1485. adminsBucket, err := p.getAdminsBucket(tx)
  1486. if err != nil {
  1487. return err
  1488. }
  1489. for idx := range oldGroup.Admins {
  1490. err = p.removeGroupFromAdminMapping(oldGroup.Name, oldGroup.Admins[idx], adminsBucket)
  1491. if err != nil {
  1492. return err
  1493. }
  1494. }
  1495. }
  1496. return bucket.Delete([]byte(group.Name))
  1497. })
  1498. }
  1499. func (p *BoltProvider) dumpGroups() ([]Group, error) {
  1500. groups := make([]Group, 0, 50)
  1501. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1502. bucket, err := p.getGroupsBucket(tx)
  1503. if err != nil {
  1504. return err
  1505. }
  1506. foldersBucket, err := p.getFoldersBucket(tx)
  1507. if err != nil {
  1508. return err
  1509. }
  1510. cursor := bucket.Cursor()
  1511. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1512. group, err := p.joinGroupAndFolders(v, foldersBucket)
  1513. if err != nil {
  1514. return err
  1515. }
  1516. groups = append(groups, group)
  1517. }
  1518. return err
  1519. })
  1520. return groups, err
  1521. }
  1522. func (p *BoltProvider) apiKeyExists(keyID string) (APIKey, error) {
  1523. var apiKey APIKey
  1524. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1525. bucket, err := p.getAPIKeysBucket(tx)
  1526. if err != nil {
  1527. return err
  1528. }
  1529. k := bucket.Get([]byte(keyID))
  1530. if k == nil {
  1531. return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", keyID))
  1532. }
  1533. return json.Unmarshal(k, &apiKey)
  1534. })
  1535. return apiKey, err
  1536. }
  1537. func (p *BoltProvider) addAPIKey(apiKey *APIKey) error {
  1538. err := apiKey.validate()
  1539. if err != nil {
  1540. return err
  1541. }
  1542. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1543. bucket, err := p.getAPIKeysBucket(tx)
  1544. if err != nil {
  1545. return err
  1546. }
  1547. if a := bucket.Get([]byte(apiKey.KeyID)); a != nil {
  1548. return fmt.Errorf("API key %v already exists", apiKey.KeyID)
  1549. }
  1550. id, err := bucket.NextSequence()
  1551. if err != nil {
  1552. return err
  1553. }
  1554. apiKey.ID = int64(id)
  1555. apiKey.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1556. apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1557. apiKey.LastUseAt = 0
  1558. if apiKey.User != "" {
  1559. if err := p.userExistsInternal(tx, apiKey.User); err != nil {
  1560. return fmt.Errorf("%w: related user %q does not exists", ErrForeignKeyViolated, apiKey.User)
  1561. }
  1562. }
  1563. if apiKey.Admin != "" {
  1564. if err := p.adminExistsInternal(tx, apiKey.Admin); err != nil {
  1565. return fmt.Errorf("%w: related admin %q does not exists", ErrForeignKeyViolated, apiKey.Admin)
  1566. }
  1567. }
  1568. buf, err := json.Marshal(apiKey)
  1569. if err != nil {
  1570. return err
  1571. }
  1572. return bucket.Put([]byte(apiKey.KeyID), buf)
  1573. })
  1574. }
  1575. func (p *BoltProvider) updateAPIKey(apiKey *APIKey) error {
  1576. err := apiKey.validate()
  1577. if err != nil {
  1578. return err
  1579. }
  1580. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1581. bucket, err := p.getAPIKeysBucket(tx)
  1582. if err != nil {
  1583. return err
  1584. }
  1585. var a []byte
  1586. if a = bucket.Get([]byte(apiKey.KeyID)); a == nil {
  1587. return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", apiKey.KeyID))
  1588. }
  1589. var oldAPIKey APIKey
  1590. err = json.Unmarshal(a, &oldAPIKey)
  1591. if err != nil {
  1592. return err
  1593. }
  1594. apiKey.ID = oldAPIKey.ID
  1595. apiKey.KeyID = oldAPIKey.KeyID
  1596. apiKey.Key = oldAPIKey.Key
  1597. apiKey.CreatedAt = oldAPIKey.CreatedAt
  1598. apiKey.LastUseAt = oldAPIKey.LastUseAt
  1599. apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1600. if apiKey.User != "" {
  1601. if err := p.userExistsInternal(tx, apiKey.User); err != nil {
  1602. return fmt.Errorf("%w: related user %q does not exists", ErrForeignKeyViolated, apiKey.User)
  1603. }
  1604. }
  1605. if apiKey.Admin != "" {
  1606. if err := p.adminExistsInternal(tx, apiKey.Admin); err != nil {
  1607. return fmt.Errorf("%w: related admin %q does not exists", ErrForeignKeyViolated, apiKey.Admin)
  1608. }
  1609. }
  1610. buf, err := json.Marshal(apiKey)
  1611. if err != nil {
  1612. return err
  1613. }
  1614. return bucket.Put([]byte(apiKey.KeyID), buf)
  1615. })
  1616. }
  1617. func (p *BoltProvider) deleteAPIKey(apiKey APIKey) error {
  1618. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1619. bucket, err := p.getAPIKeysBucket(tx)
  1620. if err != nil {
  1621. return err
  1622. }
  1623. if bucket.Get([]byte(apiKey.KeyID)) == nil {
  1624. return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", apiKey.KeyID))
  1625. }
  1626. return bucket.Delete([]byte(apiKey.KeyID))
  1627. })
  1628. }
  1629. func (p *BoltProvider) getAPIKeys(limit int, offset int, order string) ([]APIKey, error) {
  1630. apiKeys := make([]APIKey, 0, limit)
  1631. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1632. bucket, err := p.getAPIKeysBucket(tx)
  1633. if err != nil {
  1634. return err
  1635. }
  1636. cursor := bucket.Cursor()
  1637. itNum := 0
  1638. if order == OrderASC {
  1639. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1640. itNum++
  1641. if itNum <= offset {
  1642. continue
  1643. }
  1644. var apiKey APIKey
  1645. err = json.Unmarshal(v, &apiKey)
  1646. if err != nil {
  1647. return err
  1648. }
  1649. apiKey.HideConfidentialData()
  1650. apiKeys = append(apiKeys, apiKey)
  1651. if len(apiKeys) >= limit {
  1652. break
  1653. }
  1654. }
  1655. return nil
  1656. }
  1657. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  1658. itNum++
  1659. if itNum <= offset {
  1660. continue
  1661. }
  1662. var apiKey APIKey
  1663. err = json.Unmarshal(v, &apiKey)
  1664. if err != nil {
  1665. return err
  1666. }
  1667. apiKey.HideConfidentialData()
  1668. apiKeys = append(apiKeys, apiKey)
  1669. if len(apiKeys) >= limit {
  1670. break
  1671. }
  1672. }
  1673. return nil
  1674. })
  1675. return apiKeys, err
  1676. }
  1677. func (p *BoltProvider) dumpAPIKeys() ([]APIKey, error) {
  1678. apiKeys := make([]APIKey, 0, 30)
  1679. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1680. bucket, err := p.getAPIKeysBucket(tx)
  1681. if err != nil {
  1682. return err
  1683. }
  1684. cursor := bucket.Cursor()
  1685. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1686. var apiKey APIKey
  1687. err = json.Unmarshal(v, &apiKey)
  1688. if err != nil {
  1689. return err
  1690. }
  1691. apiKeys = append(apiKeys, apiKey)
  1692. }
  1693. return err
  1694. })
  1695. return apiKeys, err
  1696. }
  1697. func (p *BoltProvider) shareExists(shareID, username string) (Share, error) {
  1698. var share Share
  1699. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1700. bucket, err := p.getSharesBucket(tx)
  1701. if err != nil {
  1702. return err
  1703. }
  1704. s := bucket.Get([]byte(shareID))
  1705. if s == nil {
  1706. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", shareID))
  1707. }
  1708. if err := json.Unmarshal(s, &share); err != nil {
  1709. return err
  1710. }
  1711. if username != "" && share.Username != username {
  1712. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", shareID))
  1713. }
  1714. return nil
  1715. })
  1716. return share, err
  1717. }
  1718. func (p *BoltProvider) addShare(share *Share) error {
  1719. err := share.validate()
  1720. if err != nil {
  1721. return err
  1722. }
  1723. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1724. bucket, err := p.getSharesBucket(tx)
  1725. if err != nil {
  1726. return err
  1727. }
  1728. if a := bucket.Get([]byte(share.ShareID)); a != nil {
  1729. return fmt.Errorf("share %q already exists", share.ShareID)
  1730. }
  1731. id, err := bucket.NextSequence()
  1732. if err != nil {
  1733. return err
  1734. }
  1735. share.ID = int64(id)
  1736. if !share.IsRestore {
  1737. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1738. share.UpdatedAt = share.CreatedAt
  1739. share.LastUseAt = 0
  1740. share.UsedTokens = 0
  1741. }
  1742. if share.CreatedAt == 0 {
  1743. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1744. }
  1745. if share.UpdatedAt == 0 {
  1746. share.UpdatedAt = share.CreatedAt
  1747. }
  1748. if err := p.userExistsInternal(tx, share.Username); err != nil {
  1749. return util.NewValidationError(fmt.Sprintf("related user %q does not exists", share.Username))
  1750. }
  1751. buf, err := json.Marshal(share)
  1752. if err != nil {
  1753. return err
  1754. }
  1755. return bucket.Put([]byte(share.ShareID), buf)
  1756. })
  1757. }
  1758. func (p *BoltProvider) updateShare(share *Share) error {
  1759. if err := share.validate(); err != nil {
  1760. return err
  1761. }
  1762. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1763. bucket, err := p.getSharesBucket(tx)
  1764. if err != nil {
  1765. return err
  1766. }
  1767. var s []byte
  1768. if s = bucket.Get([]byte(share.ShareID)); s == nil {
  1769. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
  1770. }
  1771. var oldObject Share
  1772. if err = json.Unmarshal(s, &oldObject); err != nil {
  1773. return err
  1774. }
  1775. if oldObject.Username != share.Username {
  1776. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
  1777. }
  1778. share.ID = oldObject.ID
  1779. share.ShareID = oldObject.ShareID
  1780. if !share.IsRestore {
  1781. share.UsedTokens = oldObject.UsedTokens
  1782. share.CreatedAt = oldObject.CreatedAt
  1783. share.LastUseAt = oldObject.LastUseAt
  1784. share.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1785. }
  1786. if share.CreatedAt == 0 {
  1787. share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1788. }
  1789. if share.UpdatedAt == 0 {
  1790. share.UpdatedAt = share.CreatedAt
  1791. }
  1792. if err := p.userExistsInternal(tx, share.Username); err != nil {
  1793. return util.NewValidationError(fmt.Sprintf("related user %q does not exists", share.Username))
  1794. }
  1795. buf, err := json.Marshal(share)
  1796. if err != nil {
  1797. return err
  1798. }
  1799. return bucket.Put([]byte(share.ShareID), buf)
  1800. })
  1801. }
  1802. func (p *BoltProvider) deleteShare(share Share) error {
  1803. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1804. bucket, err := p.getSharesBucket(tx)
  1805. if err != nil {
  1806. return err
  1807. }
  1808. var s []byte
  1809. if s = bucket.Get([]byte(share.ShareID)); s == nil {
  1810. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
  1811. }
  1812. var oldObject Share
  1813. if err = json.Unmarshal(s, &oldObject); err != nil {
  1814. return err
  1815. }
  1816. if oldObject.Username != share.Username {
  1817. return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
  1818. }
  1819. return bucket.Delete([]byte(share.ShareID))
  1820. })
  1821. }
  1822. func (p *BoltProvider) getShares(limit int, offset int, order, username string) ([]Share, error) {
  1823. shares := make([]Share, 0, limit)
  1824. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1825. bucket, err := p.getSharesBucket(tx)
  1826. if err != nil {
  1827. return err
  1828. }
  1829. cursor := bucket.Cursor()
  1830. itNum := 0
  1831. if order == OrderASC {
  1832. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1833. var share Share
  1834. if err := json.Unmarshal(v, &share); err != nil {
  1835. return err
  1836. }
  1837. if share.Username != username {
  1838. continue
  1839. }
  1840. itNum++
  1841. if itNum <= offset {
  1842. continue
  1843. }
  1844. share.HideConfidentialData()
  1845. shares = append(shares, share)
  1846. if len(shares) >= limit {
  1847. break
  1848. }
  1849. }
  1850. return nil
  1851. }
  1852. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  1853. var share Share
  1854. err = json.Unmarshal(v, &share)
  1855. if err != nil {
  1856. return err
  1857. }
  1858. if share.Username != username {
  1859. continue
  1860. }
  1861. itNum++
  1862. if itNum <= offset {
  1863. continue
  1864. }
  1865. share.HideConfidentialData()
  1866. shares = append(shares, share)
  1867. if len(shares) >= limit {
  1868. break
  1869. }
  1870. }
  1871. return nil
  1872. })
  1873. return shares, err
  1874. }
  1875. func (p *BoltProvider) dumpShares() ([]Share, error) {
  1876. shares := make([]Share, 0, 30)
  1877. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1878. bucket, err := p.getSharesBucket(tx)
  1879. if err != nil {
  1880. return err
  1881. }
  1882. cursor := bucket.Cursor()
  1883. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1884. var share Share
  1885. err = json.Unmarshal(v, &share)
  1886. if err != nil {
  1887. return err
  1888. }
  1889. shares = append(shares, share)
  1890. }
  1891. return err
  1892. })
  1893. return shares, err
  1894. }
  1895. func (p *BoltProvider) updateShareLastUse(shareID string, numTokens int) error {
  1896. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  1897. bucket, err := p.getSharesBucket(tx)
  1898. if err != nil {
  1899. return err
  1900. }
  1901. var u []byte
  1902. if u = bucket.Get([]byte(shareID)); u == nil {
  1903. return util.NewRecordNotFoundError(fmt.Sprintf("share %q does not exist, unable to update last use", shareID))
  1904. }
  1905. var share Share
  1906. err = json.Unmarshal(u, &share)
  1907. if err != nil {
  1908. return err
  1909. }
  1910. share.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
  1911. share.UsedTokens += numTokens
  1912. buf, err := json.Marshal(share)
  1913. if err != nil {
  1914. return err
  1915. }
  1916. err = bucket.Put([]byte(shareID), buf)
  1917. if err != nil {
  1918. providerLog(logger.LevelWarn, "error updating last use for share %q: %v", shareID, err)
  1919. return err
  1920. }
  1921. providerLog(logger.LevelDebug, "last use updated for share %q", shareID)
  1922. return nil
  1923. })
  1924. }
  1925. func (p *BoltProvider) getDefenderHosts(_ int64, _ int) ([]DefenderEntry, error) {
  1926. return nil, ErrNotImplemented
  1927. }
  1928. func (p *BoltProvider) getDefenderHostByIP(_ string, _ int64) (DefenderEntry, error) {
  1929. return DefenderEntry{}, ErrNotImplemented
  1930. }
  1931. func (p *BoltProvider) isDefenderHostBanned(_ string) (DefenderEntry, error) {
  1932. return DefenderEntry{}, ErrNotImplemented
  1933. }
  1934. func (p *BoltProvider) updateDefenderBanTime(_ string, _ int) error {
  1935. return ErrNotImplemented
  1936. }
  1937. func (p *BoltProvider) deleteDefenderHost(_ string) error {
  1938. return ErrNotImplemented
  1939. }
  1940. func (p *BoltProvider) addDefenderEvent(_ string, _ int) error {
  1941. return ErrNotImplemented
  1942. }
  1943. func (p *BoltProvider) setDefenderBanTime(_ string, _ int64) error {
  1944. return ErrNotImplemented
  1945. }
  1946. func (p *BoltProvider) cleanupDefender(_ int64) error {
  1947. return ErrNotImplemented
  1948. }
  1949. func (p *BoltProvider) addActiveTransfer(_ ActiveTransfer) error {
  1950. return ErrNotImplemented
  1951. }
  1952. func (p *BoltProvider) updateActiveTransferSizes(_, _, _ int64, _ string) error {
  1953. return ErrNotImplemented
  1954. }
  1955. func (p *BoltProvider) removeActiveTransfer(_ int64, _ string) error {
  1956. return ErrNotImplemented
  1957. }
  1958. func (p *BoltProvider) cleanupActiveTransfers(_ time.Time) error {
  1959. return ErrNotImplemented
  1960. }
  1961. func (p *BoltProvider) getActiveTransfers(_ time.Time) ([]ActiveTransfer, error) {
  1962. return nil, ErrNotImplemented
  1963. }
  1964. func (p *BoltProvider) addSharedSession(_ Session) error {
  1965. return ErrNotImplemented
  1966. }
  1967. func (p *BoltProvider) deleteSharedSession(_ string) error {
  1968. return ErrNotImplemented
  1969. }
  1970. func (p *BoltProvider) getSharedSession(_ string) (Session, error) {
  1971. return Session{}, ErrNotImplemented
  1972. }
  1973. func (p *BoltProvider) cleanupSharedSessions(_ SessionType, _ int64) error {
  1974. return ErrNotImplemented
  1975. }
  1976. func (p *BoltProvider) getEventActions(limit, offset int, order string, _ bool) ([]BaseEventAction, error) {
  1977. if limit <= 0 {
  1978. return nil, nil
  1979. }
  1980. actions := make([]BaseEventAction, 0, limit)
  1981. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  1982. bucket, err := p.getActionsBucket(tx)
  1983. if err != nil {
  1984. return err
  1985. }
  1986. itNum := 0
  1987. cursor := bucket.Cursor()
  1988. if order == OrderASC {
  1989. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  1990. itNum++
  1991. if itNum <= offset {
  1992. continue
  1993. }
  1994. var action BaseEventAction
  1995. err = json.Unmarshal(v, &action)
  1996. if err != nil {
  1997. return err
  1998. }
  1999. action.PrepareForRendering()
  2000. actions = append(actions, action)
  2001. if len(actions) >= limit {
  2002. break
  2003. }
  2004. }
  2005. } else {
  2006. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  2007. itNum++
  2008. if itNum <= offset {
  2009. continue
  2010. }
  2011. var action BaseEventAction
  2012. err = json.Unmarshal(v, &action)
  2013. if err != nil {
  2014. return err
  2015. }
  2016. action.PrepareForRendering()
  2017. actions = append(actions, action)
  2018. if len(actions) >= limit {
  2019. break
  2020. }
  2021. }
  2022. }
  2023. return nil
  2024. })
  2025. return actions, err
  2026. }
  2027. func (p *BoltProvider) dumpEventActions() ([]BaseEventAction, error) {
  2028. actions := make([]BaseEventAction, 0, 50)
  2029. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2030. bucket, err := p.getActionsBucket(tx)
  2031. if err != nil {
  2032. return err
  2033. }
  2034. cursor := bucket.Cursor()
  2035. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2036. var action BaseEventAction
  2037. err = json.Unmarshal(v, &action)
  2038. if err != nil {
  2039. return err
  2040. }
  2041. actions = append(actions, action)
  2042. }
  2043. return nil
  2044. })
  2045. return actions, err
  2046. }
  2047. func (p *BoltProvider) eventActionExists(name string) (BaseEventAction, error) {
  2048. var action BaseEventAction
  2049. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2050. bucket, err := p.getActionsBucket(tx)
  2051. if err != nil {
  2052. return err
  2053. }
  2054. k := bucket.Get([]byte(name))
  2055. if k == nil {
  2056. return util.NewRecordNotFoundError(fmt.Sprintf("action %q does not exist", name))
  2057. }
  2058. return json.Unmarshal(k, &action)
  2059. })
  2060. return action, err
  2061. }
  2062. func (p *BoltProvider) addEventAction(action *BaseEventAction) error {
  2063. err := action.validate()
  2064. if err != nil {
  2065. return err
  2066. }
  2067. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2068. bucket, err := p.getActionsBucket(tx)
  2069. if err != nil {
  2070. return err
  2071. }
  2072. if a := bucket.Get([]byte(action.Name)); a != nil {
  2073. return util.NewI18nError(
  2074. fmt.Errorf("%w: event action %q already exists", ErrDuplicatedKey, action.Name),
  2075. util.I18nErrorDuplicatedName,
  2076. )
  2077. }
  2078. id, err := bucket.NextSequence()
  2079. if err != nil {
  2080. return err
  2081. }
  2082. action.ID = int64(id)
  2083. action.Rules = nil
  2084. buf, err := json.Marshal(action)
  2085. if err != nil {
  2086. return err
  2087. }
  2088. return bucket.Put([]byte(action.Name), buf)
  2089. })
  2090. }
  2091. func (p *BoltProvider) updateEventAction(action *BaseEventAction) error {
  2092. err := action.validate()
  2093. if err != nil {
  2094. return err
  2095. }
  2096. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2097. bucket, err := p.getActionsBucket(tx)
  2098. if err != nil {
  2099. return err
  2100. }
  2101. var a []byte
  2102. if a = bucket.Get([]byte(action.Name)); a == nil {
  2103. return util.NewRecordNotFoundError(fmt.Sprintf("event action %s does not exist", action.Name))
  2104. }
  2105. var oldAction BaseEventAction
  2106. err = json.Unmarshal(a, &oldAction)
  2107. if err != nil {
  2108. return err
  2109. }
  2110. action.ID = oldAction.ID
  2111. action.Name = oldAction.Name
  2112. action.Rules = nil
  2113. if len(oldAction.Rules) > 0 {
  2114. rulesBucket, err := p.getRulesBucket(tx)
  2115. if err != nil {
  2116. return err
  2117. }
  2118. var relatedRules []string
  2119. for _, ruleName := range oldAction.Rules {
  2120. r := rulesBucket.Get([]byte(ruleName))
  2121. if r != nil {
  2122. relatedRules = append(relatedRules, ruleName)
  2123. var rule EventRule
  2124. err := json.Unmarshal(r, &rule)
  2125. if err != nil {
  2126. return err
  2127. }
  2128. rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2129. buf, err := json.Marshal(rule)
  2130. if err != nil {
  2131. return err
  2132. }
  2133. if err = rulesBucket.Put([]byte(rule.Name), buf); err != nil {
  2134. return err
  2135. }
  2136. setLastRuleUpdate()
  2137. }
  2138. }
  2139. action.Rules = relatedRules
  2140. }
  2141. buf, err := json.Marshal(action)
  2142. if err != nil {
  2143. return err
  2144. }
  2145. return bucket.Put([]byte(action.Name), buf)
  2146. })
  2147. }
  2148. func (p *BoltProvider) deleteEventAction(action BaseEventAction) error {
  2149. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2150. bucket, err := p.getActionsBucket(tx)
  2151. if err != nil {
  2152. return err
  2153. }
  2154. var a []byte
  2155. if a = bucket.Get([]byte(action.Name)); a == nil {
  2156. return util.NewRecordNotFoundError(fmt.Sprintf("action %s does not exist", action.Name))
  2157. }
  2158. var oldAction BaseEventAction
  2159. err = json.Unmarshal(a, &oldAction)
  2160. if err != nil {
  2161. return err
  2162. }
  2163. if len(oldAction.Rules) > 0 {
  2164. return util.NewValidationError(fmt.Sprintf("action %s is referenced, it cannot be removed", oldAction.Name))
  2165. }
  2166. return bucket.Delete([]byte(action.Name))
  2167. })
  2168. }
  2169. func (p *BoltProvider) getEventRules(limit, offset int, order string) ([]EventRule, error) {
  2170. if limit <= 0 {
  2171. return nil, nil
  2172. }
  2173. rules := make([]EventRule, 0, limit)
  2174. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2175. bucket, err := p.getRulesBucket(tx)
  2176. if err != nil {
  2177. return err
  2178. }
  2179. actionsBucket, err := p.getActionsBucket(tx)
  2180. if err != nil {
  2181. return err
  2182. }
  2183. itNum := 0
  2184. cursor := bucket.Cursor()
  2185. if order == OrderASC {
  2186. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2187. itNum++
  2188. if itNum <= offset {
  2189. continue
  2190. }
  2191. var rule EventRule
  2192. rule, err = p.joinRuleAndActions(v, actionsBucket)
  2193. if err != nil {
  2194. return err
  2195. }
  2196. rule.PrepareForRendering()
  2197. rules = append(rules, rule)
  2198. if len(rules) >= limit {
  2199. break
  2200. }
  2201. }
  2202. } else {
  2203. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  2204. itNum++
  2205. if itNum <= offset {
  2206. continue
  2207. }
  2208. var rule EventRule
  2209. rule, err = p.joinRuleAndActions(v, actionsBucket)
  2210. if err != nil {
  2211. return err
  2212. }
  2213. rule.PrepareForRendering()
  2214. rules = append(rules, rule)
  2215. if len(rules) >= limit {
  2216. break
  2217. }
  2218. }
  2219. }
  2220. return err
  2221. })
  2222. return rules, err
  2223. }
  2224. func (p *BoltProvider) dumpEventRules() ([]EventRule, error) {
  2225. rules := make([]EventRule, 0, 50)
  2226. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2227. bucket, err := p.getRulesBucket(tx)
  2228. if err != nil {
  2229. return err
  2230. }
  2231. actionsBucket, err := p.getActionsBucket(tx)
  2232. if err != nil {
  2233. return err
  2234. }
  2235. cursor := bucket.Cursor()
  2236. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2237. rule, err := p.joinRuleAndActions(v, actionsBucket)
  2238. if err != nil {
  2239. return err
  2240. }
  2241. rules = append(rules, rule)
  2242. }
  2243. return nil
  2244. })
  2245. return rules, err
  2246. }
  2247. func (p *BoltProvider) getRecentlyUpdatedRules(after int64) ([]EventRule, error) {
  2248. if getLastRuleUpdate() < after {
  2249. return nil, nil
  2250. }
  2251. rules := make([]EventRule, 0, 10)
  2252. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2253. bucket, err := p.getRulesBucket(tx)
  2254. if err != nil {
  2255. return err
  2256. }
  2257. actionsBucket, err := p.getActionsBucket(tx)
  2258. if err != nil {
  2259. return err
  2260. }
  2261. cursor := bucket.Cursor()
  2262. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2263. var rule EventRule
  2264. err := json.Unmarshal(v, &rule)
  2265. if err != nil {
  2266. return err
  2267. }
  2268. if rule.UpdatedAt < after {
  2269. continue
  2270. }
  2271. var actions []EventAction
  2272. for idx := range rule.Actions {
  2273. action := &rule.Actions[idx]
  2274. var baseAction BaseEventAction
  2275. k := actionsBucket.Get([]byte(action.Name))
  2276. if k == nil {
  2277. continue
  2278. }
  2279. err = json.Unmarshal(k, &baseAction)
  2280. if err != nil {
  2281. continue
  2282. }
  2283. baseAction.Options.SetEmptySecretsIfNil()
  2284. action.BaseEventAction = baseAction
  2285. actions = append(actions, *action)
  2286. }
  2287. rule.Actions = actions
  2288. rules = append(rules, rule)
  2289. }
  2290. return nil
  2291. })
  2292. return rules, err
  2293. }
  2294. func (p *BoltProvider) eventRuleExists(name string) (EventRule, error) {
  2295. var rule EventRule
  2296. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2297. bucket, err := p.getRulesBucket(tx)
  2298. if err != nil {
  2299. return err
  2300. }
  2301. r := bucket.Get([]byte(name))
  2302. if r == nil {
  2303. return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", name))
  2304. }
  2305. actionsBucket, err := p.getActionsBucket(tx)
  2306. if err != nil {
  2307. return err
  2308. }
  2309. rule, err = p.joinRuleAndActions(r, actionsBucket)
  2310. return err
  2311. })
  2312. return rule, err
  2313. }
  2314. func (p *BoltProvider) addEventRule(rule *EventRule) error {
  2315. if err := rule.validate(); err != nil {
  2316. return err
  2317. }
  2318. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2319. bucket, err := p.getRulesBucket(tx)
  2320. if err != nil {
  2321. return err
  2322. }
  2323. actionsBucket, err := p.getActionsBucket(tx)
  2324. if err != nil {
  2325. return err
  2326. }
  2327. if r := bucket.Get([]byte(rule.Name)); r != nil {
  2328. return util.NewI18nError(
  2329. fmt.Errorf("%w: event rule %q already exists", ErrDuplicatedKey, rule.Name),
  2330. util.I18nErrorDuplicatedName,
  2331. )
  2332. }
  2333. id, err := bucket.NextSequence()
  2334. if err != nil {
  2335. return err
  2336. }
  2337. rule.ID = int64(id)
  2338. rule.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2339. rule.UpdatedAt = rule.CreatedAt
  2340. for idx := range rule.Actions {
  2341. if err = p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name, actionsBucket); err != nil {
  2342. return err
  2343. }
  2344. }
  2345. sort.Slice(rule.Actions, func(i, j int) bool {
  2346. return rule.Actions[i].Order < rule.Actions[j].Order
  2347. })
  2348. buf, err := json.Marshal(rule)
  2349. if err != nil {
  2350. return err
  2351. }
  2352. err = bucket.Put([]byte(rule.Name), buf)
  2353. if err == nil {
  2354. setLastRuleUpdate()
  2355. }
  2356. return err
  2357. })
  2358. }
  2359. func (p *BoltProvider) updateEventRule(rule *EventRule) error {
  2360. if err := rule.validate(); err != nil {
  2361. return err
  2362. }
  2363. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2364. bucket, err := p.getRulesBucket(tx)
  2365. if err != nil {
  2366. return err
  2367. }
  2368. actionsBucket, err := p.getActionsBucket(tx)
  2369. if err != nil {
  2370. return err
  2371. }
  2372. var r []byte
  2373. if r = bucket.Get([]byte(rule.Name)); r == nil {
  2374. return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", rule.Name))
  2375. }
  2376. var oldRule EventRule
  2377. if err = json.Unmarshal(r, &oldRule); err != nil {
  2378. return err
  2379. }
  2380. for idx := range oldRule.Actions {
  2381. if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name, actionsBucket); err != nil {
  2382. return err
  2383. }
  2384. }
  2385. for idx := range rule.Actions {
  2386. if err = p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name, actionsBucket); err != nil {
  2387. return err
  2388. }
  2389. }
  2390. rule.ID = oldRule.ID
  2391. rule.CreatedAt = oldRule.CreatedAt
  2392. rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2393. buf, err := json.Marshal(rule)
  2394. if err != nil {
  2395. return err
  2396. }
  2397. sort.Slice(rule.Actions, func(i, j int) bool {
  2398. return rule.Actions[i].Order < rule.Actions[j].Order
  2399. })
  2400. err = bucket.Put([]byte(rule.Name), buf)
  2401. if err == nil {
  2402. setLastRuleUpdate()
  2403. }
  2404. return err
  2405. })
  2406. }
  2407. func (p *BoltProvider) deleteEventRule(rule EventRule, _ bool) error {
  2408. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2409. bucket, err := p.getRulesBucket(tx)
  2410. if err != nil {
  2411. return err
  2412. }
  2413. var r []byte
  2414. if r = bucket.Get([]byte(rule.Name)); r == nil {
  2415. return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", rule.Name))
  2416. }
  2417. var oldRule EventRule
  2418. if err = json.Unmarshal(r, &oldRule); err != nil {
  2419. return err
  2420. }
  2421. if len(oldRule.Actions) > 0 {
  2422. actionsBucket, err := p.getActionsBucket(tx)
  2423. if err != nil {
  2424. return err
  2425. }
  2426. for idx := range oldRule.Actions {
  2427. if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name, actionsBucket); err != nil {
  2428. return err
  2429. }
  2430. }
  2431. }
  2432. return bucket.Delete([]byte(rule.Name))
  2433. })
  2434. }
  2435. func (*BoltProvider) getTaskByName(_ string) (Task, error) {
  2436. return Task{}, ErrNotImplemented
  2437. }
  2438. func (*BoltProvider) addTask(_ string) error {
  2439. return ErrNotImplemented
  2440. }
  2441. func (*BoltProvider) updateTask(_ string, _ int64) error {
  2442. return ErrNotImplemented
  2443. }
  2444. func (*BoltProvider) updateTaskTimestamp(_ string) error {
  2445. return ErrNotImplemented
  2446. }
  2447. func (*BoltProvider) addNode() error {
  2448. return ErrNotImplemented
  2449. }
  2450. func (*BoltProvider) getNodeByName(_ string) (Node, error) {
  2451. return Node{}, ErrNotImplemented
  2452. }
  2453. func (*BoltProvider) getNodes() ([]Node, error) {
  2454. return nil, ErrNotImplemented
  2455. }
  2456. func (*BoltProvider) updateNodeTimestamp() error {
  2457. return ErrNotImplemented
  2458. }
  2459. func (*BoltProvider) cleanupNodes() error {
  2460. return ErrNotImplemented
  2461. }
  2462. func (p *BoltProvider) roleExists(name string) (Role, error) {
  2463. var role Role
  2464. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2465. bucket, err := p.getRolesBucket(tx)
  2466. if err != nil {
  2467. return err
  2468. }
  2469. r := bucket.Get([]byte(name))
  2470. if r == nil {
  2471. return util.NewRecordNotFoundError(fmt.Sprintf("role %q does not exist", name))
  2472. }
  2473. return json.Unmarshal(r, &role)
  2474. })
  2475. return role, err
  2476. }
  2477. func (p *BoltProvider) addRole(role *Role) error {
  2478. if err := role.validate(); err != nil {
  2479. return err
  2480. }
  2481. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2482. bucket, err := p.getRolesBucket(tx)
  2483. if err != nil {
  2484. return err
  2485. }
  2486. if r := bucket.Get([]byte(role.Name)); r != nil {
  2487. return util.NewI18nError(
  2488. fmt.Errorf("%w: role %q already exists", ErrDuplicatedKey, role.Name),
  2489. util.I18nErrorDuplicatedName,
  2490. )
  2491. }
  2492. id, err := bucket.NextSequence()
  2493. if err != nil {
  2494. return err
  2495. }
  2496. role.ID = int64(id)
  2497. role.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2498. role.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2499. role.Users = nil
  2500. role.Admins = nil
  2501. buf, err := json.Marshal(role)
  2502. if err != nil {
  2503. return err
  2504. }
  2505. return bucket.Put([]byte(role.Name), buf)
  2506. })
  2507. }
  2508. func (p *BoltProvider) updateRole(role *Role) error {
  2509. if err := role.validate(); err != nil {
  2510. return err
  2511. }
  2512. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2513. bucket, err := p.getRolesBucket(tx)
  2514. if err != nil {
  2515. return err
  2516. }
  2517. var r []byte
  2518. if r = bucket.Get([]byte(role.Name)); r == nil {
  2519. return fmt.Errorf("role %q does not exist", role.Name)
  2520. }
  2521. var oldRole Role
  2522. err = json.Unmarshal(r, &oldRole)
  2523. if err != nil {
  2524. return err
  2525. }
  2526. role.ID = oldRole.ID
  2527. role.CreatedAt = oldRole.CreatedAt
  2528. role.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2529. role.Users = oldRole.Users
  2530. role.Admins = oldRole.Admins
  2531. buf, err := json.Marshal(role)
  2532. if err != nil {
  2533. return err
  2534. }
  2535. return bucket.Put([]byte(role.Name), buf)
  2536. })
  2537. }
  2538. func (p *BoltProvider) deleteRole(role Role) error {
  2539. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2540. bucket, err := p.getRolesBucket(tx)
  2541. if err != nil {
  2542. return err
  2543. }
  2544. var r []byte
  2545. if r = bucket.Get([]byte(role.Name)); r == nil {
  2546. return fmt.Errorf("role %q does not exist", role.Name)
  2547. }
  2548. var oldRole Role
  2549. err = json.Unmarshal(r, &oldRole)
  2550. if err != nil {
  2551. return err
  2552. }
  2553. if len(oldRole.Admins) > 0 {
  2554. return util.NewValidationError(fmt.Sprintf("the role %q is referenced, it cannot be removed", oldRole.Name))
  2555. }
  2556. if len(oldRole.Users) > 0 {
  2557. bucket, err := p.getUsersBucket(tx)
  2558. if err != nil {
  2559. return err
  2560. }
  2561. for _, username := range oldRole.Users {
  2562. if err := p.removeRoleFromUser(username, oldRole.Name, bucket); err != nil {
  2563. return err
  2564. }
  2565. }
  2566. }
  2567. return bucket.Delete([]byte(role.Name))
  2568. })
  2569. }
  2570. func (p *BoltProvider) getRoles(limit int, offset int, order string, _ bool) ([]Role, error) {
  2571. roles := make([]Role, 0, limit)
  2572. if limit <= 0 {
  2573. return roles, nil
  2574. }
  2575. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2576. bucket, err := p.getRolesBucket(tx)
  2577. if err != nil {
  2578. return err
  2579. }
  2580. cursor := bucket.Cursor()
  2581. itNum := 0
  2582. if order == OrderASC {
  2583. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2584. itNum++
  2585. if itNum <= offset {
  2586. continue
  2587. }
  2588. var role Role
  2589. err = json.Unmarshal(v, &role)
  2590. if err != nil {
  2591. return err
  2592. }
  2593. roles = append(roles, role)
  2594. if len(roles) >= limit {
  2595. break
  2596. }
  2597. }
  2598. } else {
  2599. for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
  2600. itNum++
  2601. if itNum <= offset {
  2602. continue
  2603. }
  2604. var role Role
  2605. err = json.Unmarshal(v, &role)
  2606. if err != nil {
  2607. return err
  2608. }
  2609. roles = append(roles, role)
  2610. if len(roles) >= limit {
  2611. break
  2612. }
  2613. }
  2614. }
  2615. return nil
  2616. })
  2617. return roles, err
  2618. }
  2619. func (p *BoltProvider) dumpRoles() ([]Role, error) {
  2620. roles := make([]Role, 0, 10)
  2621. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2622. bucket, err := p.getRolesBucket(tx)
  2623. if err != nil {
  2624. return err
  2625. }
  2626. cursor := bucket.Cursor()
  2627. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2628. var role Role
  2629. err = json.Unmarshal(v, &role)
  2630. if err != nil {
  2631. return err
  2632. }
  2633. roles = append(roles, role)
  2634. }
  2635. return err
  2636. })
  2637. return roles, err
  2638. }
  2639. func (p *BoltProvider) ipListEntryExists(ipOrNet string, listType IPListType) (IPListEntry, error) {
  2640. entry := IPListEntry{
  2641. IPOrNet: ipOrNet,
  2642. Type: listType,
  2643. }
  2644. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2645. bucket, err := p.getIPListsBucket(tx)
  2646. if err != nil {
  2647. return err
  2648. }
  2649. e := bucket.Get([]byte(entry.getKey()))
  2650. if e == nil {
  2651. return util.NewRecordNotFoundError(fmt.Sprintf("entry %q does not exist", entry.IPOrNet))
  2652. }
  2653. err = json.Unmarshal(e, &entry)
  2654. if err == nil {
  2655. entry.PrepareForRendering()
  2656. }
  2657. return err
  2658. })
  2659. return entry, err
  2660. }
  2661. func (p *BoltProvider) addIPListEntry(entry *IPListEntry) error {
  2662. if err := entry.validate(); err != nil {
  2663. return err
  2664. }
  2665. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2666. bucket, err := p.getIPListsBucket(tx)
  2667. if err != nil {
  2668. return err
  2669. }
  2670. if e := bucket.Get([]byte(entry.getKey())); e != nil {
  2671. return util.NewI18nError(
  2672. fmt.Errorf("%w: entry %q already exists", ErrDuplicatedKey, entry.IPOrNet),
  2673. util.I18nErrorDuplicatedIPNet,
  2674. )
  2675. }
  2676. entry.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2677. entry.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2678. buf, err := json.Marshal(entry)
  2679. if err != nil {
  2680. return err
  2681. }
  2682. return bucket.Put([]byte(entry.getKey()), buf)
  2683. })
  2684. }
  2685. func (p *BoltProvider) updateIPListEntry(entry *IPListEntry) error {
  2686. if err := entry.validate(); err != nil {
  2687. return err
  2688. }
  2689. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2690. bucket, err := p.getIPListsBucket(tx)
  2691. if err != nil {
  2692. return err
  2693. }
  2694. var e []byte
  2695. if e = bucket.Get([]byte(entry.getKey())); e == nil {
  2696. return fmt.Errorf("entry %q does not exist", entry.IPOrNet)
  2697. }
  2698. var oldEntry IPListEntry
  2699. err = json.Unmarshal(e, &oldEntry)
  2700. if err != nil {
  2701. return err
  2702. }
  2703. entry.CreatedAt = oldEntry.CreatedAt
  2704. entry.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
  2705. buf, err := json.Marshal(entry)
  2706. if err != nil {
  2707. return err
  2708. }
  2709. return bucket.Put([]byte(entry.getKey()), buf)
  2710. })
  2711. }
  2712. func (p *BoltProvider) deleteIPListEntry(entry IPListEntry, _ bool) error {
  2713. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2714. bucket, err := p.getIPListsBucket(tx)
  2715. if err != nil {
  2716. return err
  2717. }
  2718. if e := bucket.Get([]byte(entry.getKey())); e == nil {
  2719. return fmt.Errorf("entry %q does not exist", entry.IPOrNet)
  2720. }
  2721. return bucket.Delete([]byte(entry.getKey()))
  2722. })
  2723. }
  2724. func (p *BoltProvider) getIPListEntries(listType IPListType, filter, from, order string, limit int) ([]IPListEntry, error) {
  2725. entries := make([]IPListEntry, 0, 15)
  2726. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2727. bucket, err := p.getIPListsBucket(tx)
  2728. if err != nil {
  2729. return err
  2730. }
  2731. prefix := []byte(fmt.Sprintf("%d_", listType))
  2732. acceptKey := func(k []byte) bool {
  2733. return k != nil && bytes.HasPrefix(k, prefix)
  2734. }
  2735. cursor := bucket.Cursor()
  2736. if order == OrderASC {
  2737. for k, v := cursor.Seek(prefix); acceptKey(k); k, v = cursor.Next() {
  2738. var entry IPListEntry
  2739. err = json.Unmarshal(v, &entry)
  2740. if err != nil {
  2741. return err
  2742. }
  2743. if entry.satisfySearchConstraints(filter, from, order) {
  2744. entry.PrepareForRendering()
  2745. entries = append(entries, entry)
  2746. if limit > 0 && len(entries) >= limit {
  2747. break
  2748. }
  2749. }
  2750. }
  2751. } else {
  2752. for k, v := cursor.Last(); acceptKey(k); k, v = cursor.Prev() {
  2753. var entry IPListEntry
  2754. err = json.Unmarshal(v, &entry)
  2755. if err != nil {
  2756. return err
  2757. }
  2758. if entry.satisfySearchConstraints(filter, from, order) {
  2759. entry.PrepareForRendering()
  2760. entries = append(entries, entry)
  2761. if limit > 0 && len(entries) >= limit {
  2762. break
  2763. }
  2764. }
  2765. }
  2766. }
  2767. return nil
  2768. })
  2769. return entries, err
  2770. }
  2771. func (p *BoltProvider) getRecentlyUpdatedIPListEntries(_ int64) ([]IPListEntry, error) {
  2772. return nil, ErrNotImplemented
  2773. }
  2774. func (p *BoltProvider) dumpIPListEntries() ([]IPListEntry, error) {
  2775. entries := make([]IPListEntry, 0, 10)
  2776. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2777. bucket, err := p.getIPListsBucket(tx)
  2778. if err != nil {
  2779. return err
  2780. }
  2781. if count := bucket.Stats().KeyN; count > ipListMemoryLimit {
  2782. providerLog(logger.LevelInfo, "IP lists excluded from dump, too many entries: %d", count)
  2783. return nil
  2784. }
  2785. cursor := bucket.Cursor()
  2786. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  2787. var entry IPListEntry
  2788. err = json.Unmarshal(v, &entry)
  2789. if err != nil {
  2790. return err
  2791. }
  2792. entry.PrepareForRendering()
  2793. entries = append(entries, entry)
  2794. }
  2795. return nil
  2796. })
  2797. return entries, err
  2798. }
  2799. func (p *BoltProvider) countIPListEntries(listType IPListType) (int64, error) {
  2800. var count int64
  2801. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2802. bucket, err := p.getIPListsBucket(tx)
  2803. if err != nil {
  2804. return err
  2805. }
  2806. if listType == 0 {
  2807. count = int64(bucket.Stats().KeyN)
  2808. return nil
  2809. }
  2810. prefix := []byte(fmt.Sprintf("%d_", listType))
  2811. cursor := bucket.Cursor()
  2812. for k, _ := cursor.Seek(prefix); k != nil && bytes.HasPrefix(k, prefix); k, _ = cursor.Next() {
  2813. count++
  2814. }
  2815. return nil
  2816. })
  2817. return count, err
  2818. }
  2819. func (p *BoltProvider) getListEntriesForIP(ip string, listType IPListType) ([]IPListEntry, error) {
  2820. entries := make([]IPListEntry, 0, 3)
  2821. ipAddr, err := netip.ParseAddr(ip)
  2822. if err != nil {
  2823. return entries, fmt.Errorf("invalid ip address %s", ip)
  2824. }
  2825. var netType int
  2826. var ipBytes []byte
  2827. if ipAddr.Is4() || ipAddr.Is4In6() {
  2828. netType = ipTypeV4
  2829. as4 := ipAddr.As4()
  2830. ipBytes = as4[:]
  2831. } else {
  2832. netType = ipTypeV6
  2833. as16 := ipAddr.As16()
  2834. ipBytes = as16[:]
  2835. }
  2836. err = p.dbHandle.View(func(tx *bolt.Tx) error {
  2837. bucket, err := p.getIPListsBucket(tx)
  2838. if err != nil {
  2839. return err
  2840. }
  2841. prefix := []byte(fmt.Sprintf("%d_", listType))
  2842. cursor := bucket.Cursor()
  2843. for k, v := cursor.Seek(prefix); k != nil && bytes.HasPrefix(k, prefix); k, v = cursor.Next() {
  2844. var entry IPListEntry
  2845. err = json.Unmarshal(v, &entry)
  2846. if err != nil {
  2847. return err
  2848. }
  2849. if entry.IPType == netType && bytes.Compare(ipBytes, entry.First) >= 0 && bytes.Compare(ipBytes, entry.Last) <= 0 {
  2850. entry.PrepareForRendering()
  2851. entries = append(entries, entry)
  2852. }
  2853. }
  2854. return nil
  2855. })
  2856. return entries, err
  2857. }
  2858. func (p *BoltProvider) getConfigs() (Configs, error) {
  2859. var configs Configs
  2860. err := p.dbHandle.View(func(tx *bolt.Tx) error {
  2861. bucket := tx.Bucket(configsBucket)
  2862. if bucket == nil {
  2863. return fmt.Errorf("unable to find configs bucket")
  2864. }
  2865. data := bucket.Get(configsKey)
  2866. if data != nil {
  2867. return json.Unmarshal(data, &configs)
  2868. }
  2869. return nil
  2870. })
  2871. return configs, err
  2872. }
  2873. func (p *BoltProvider) setConfigs(configs *Configs) error {
  2874. if err := configs.validate(); err != nil {
  2875. return err
  2876. }
  2877. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2878. bucket := tx.Bucket(configsBucket)
  2879. if bucket == nil {
  2880. return fmt.Errorf("unable to find configs bucket")
  2881. }
  2882. buf, err := json.Marshal(configs)
  2883. if err != nil {
  2884. return err
  2885. }
  2886. return bucket.Put(configsKey, buf)
  2887. })
  2888. }
  2889. func (p *BoltProvider) setFirstDownloadTimestamp(username string) error {
  2890. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2891. bucket, err := p.getUsersBucket(tx)
  2892. if err != nil {
  2893. return err
  2894. }
  2895. var u []byte
  2896. if u = bucket.Get([]byte(username)); u == nil {
  2897. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to set download timestamp",
  2898. username))
  2899. }
  2900. var user User
  2901. err = json.Unmarshal(u, &user)
  2902. if err != nil {
  2903. return err
  2904. }
  2905. if user.FirstDownload > 0 {
  2906. return util.NewGenericError(fmt.Sprintf("first download already set to %v",
  2907. util.GetTimeFromMsecSinceEpoch(user.FirstDownload)))
  2908. }
  2909. user.FirstDownload = util.GetTimeAsMsSinceEpoch(time.Now())
  2910. buf, err := json.Marshal(user)
  2911. if err != nil {
  2912. return err
  2913. }
  2914. return bucket.Put([]byte(username), buf)
  2915. })
  2916. }
  2917. func (p *BoltProvider) setFirstUploadTimestamp(username string) error {
  2918. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2919. bucket, err := p.getUsersBucket(tx)
  2920. if err != nil {
  2921. return err
  2922. }
  2923. var u []byte
  2924. if u = bucket.Get([]byte(username)); u == nil {
  2925. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to set upload timestamp",
  2926. username))
  2927. }
  2928. var user User
  2929. if err = json.Unmarshal(u, &user); err != nil {
  2930. return err
  2931. }
  2932. if user.FirstUpload > 0 {
  2933. return util.NewGenericError(fmt.Sprintf("first upload already set to %v",
  2934. util.GetTimeFromMsecSinceEpoch(user.FirstUpload)))
  2935. }
  2936. user.FirstUpload = util.GetTimeAsMsSinceEpoch(time.Now())
  2937. buf, err := json.Marshal(user)
  2938. if err != nil {
  2939. return err
  2940. }
  2941. return bucket.Put([]byte(username), buf)
  2942. })
  2943. }
  2944. func (p *BoltProvider) close() error {
  2945. return p.dbHandle.Close()
  2946. }
  2947. func (p *BoltProvider) reloadConfig() error {
  2948. return nil
  2949. }
  2950. // initializeDatabase does nothing, no initilization is needed for bolt provider
  2951. func (p *BoltProvider) initializeDatabase() error {
  2952. return ErrNoInitRequired
  2953. }
  2954. func (p *BoltProvider) migrateDatabase() error {
  2955. dbVersion, err := getBoltDatabaseVersion(p.dbHandle)
  2956. if err != nil {
  2957. return err
  2958. }
  2959. switch version := dbVersion.Version; {
  2960. case version == boltDatabaseVersion:
  2961. providerLog(logger.LevelDebug, "bolt database is up to date, current version: %d", version)
  2962. return ErrNoInitRequired
  2963. case version < 29:
  2964. err = fmt.Errorf("database schema version %d is too old, please see the upgrading docs", version)
  2965. providerLog(logger.LevelError, "%v", err)
  2966. logger.ErrorToConsole("%v", err)
  2967. return err
  2968. default:
  2969. if version > boltDatabaseVersion {
  2970. providerLog(logger.LevelError, "database schema version %d is newer than the supported one: %d", version,
  2971. boltDatabaseVersion)
  2972. logger.WarnToConsole("database schema version %d is newer than the supported one: %d", version,
  2973. boltDatabaseVersion)
  2974. return nil
  2975. }
  2976. return fmt.Errorf("database schema version not handled: %d", version)
  2977. }
  2978. }
  2979. func (p *BoltProvider) revertDatabase(targetVersion int) error { //nolint:gocyclo
  2980. dbVersion, err := getBoltDatabaseVersion(p.dbHandle)
  2981. if err != nil {
  2982. return err
  2983. }
  2984. if dbVersion.Version == targetVersion {
  2985. return errors.New("current version match target version, nothing to do")
  2986. }
  2987. switch dbVersion.Version {
  2988. default:
  2989. return fmt.Errorf("database schema version not handled: %v", dbVersion.Version)
  2990. }
  2991. }
  2992. func (p *BoltProvider) resetDatabase() error {
  2993. return p.dbHandle.Update(func(tx *bolt.Tx) error {
  2994. for _, bucketName := range boltBuckets {
  2995. err := tx.DeleteBucket(bucketName)
  2996. if err != nil && !errors.Is(err, bolt.ErrBucketNotFound) {
  2997. return fmt.Errorf("unable to remove bucket %v: %w", bucketName, err)
  2998. }
  2999. }
  3000. return nil
  3001. })
  3002. }
  3003. func (p *BoltProvider) joinRuleAndActions(r []byte, actionsBucket *bolt.Bucket) (EventRule, error) {
  3004. var rule EventRule
  3005. err := json.Unmarshal(r, &rule)
  3006. if err != nil {
  3007. return rule, err
  3008. }
  3009. var actions []EventAction
  3010. for idx := range rule.Actions {
  3011. action := &rule.Actions[idx]
  3012. var baseAction BaseEventAction
  3013. k := actionsBucket.Get([]byte(action.Name))
  3014. if k == nil {
  3015. continue
  3016. }
  3017. err = json.Unmarshal(k, &baseAction)
  3018. if err != nil {
  3019. continue
  3020. }
  3021. baseAction.Options.SetEmptySecretsIfNil()
  3022. action.BaseEventAction = baseAction
  3023. actions = append(actions, *action)
  3024. }
  3025. rule.Actions = actions
  3026. return rule, nil
  3027. }
  3028. func (p *BoltProvider) joinGroupAndFolders(g []byte, foldersBucket *bolt.Bucket) (Group, error) {
  3029. var group Group
  3030. err := json.Unmarshal(g, &group)
  3031. if err != nil {
  3032. return group, err
  3033. }
  3034. if len(group.VirtualFolders) > 0 {
  3035. var folders []vfs.VirtualFolder
  3036. for idx := range group.VirtualFolders {
  3037. folder := &group.VirtualFolders[idx]
  3038. baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
  3039. if err != nil {
  3040. continue
  3041. }
  3042. folder.BaseVirtualFolder = baseFolder
  3043. folders = append(folders, *folder)
  3044. }
  3045. group.VirtualFolders = folders
  3046. }
  3047. group.SetEmptySecretsIfNil()
  3048. return group, err
  3049. }
  3050. func (p *BoltProvider) joinUserAndFolders(u []byte, foldersBucket *bolt.Bucket) (User, error) {
  3051. var user User
  3052. err := json.Unmarshal(u, &user)
  3053. if err != nil {
  3054. return user, err
  3055. }
  3056. if len(user.VirtualFolders) > 0 {
  3057. var folders []vfs.VirtualFolder
  3058. for idx := range user.VirtualFolders {
  3059. folder := &user.VirtualFolders[idx]
  3060. baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
  3061. if err != nil {
  3062. continue
  3063. }
  3064. folder.BaseVirtualFolder = baseFolder
  3065. folders = append(folders, *folder)
  3066. }
  3067. user.VirtualFolders = folders
  3068. }
  3069. user.SetEmptySecretsIfNil()
  3070. return user, err
  3071. }
  3072. func (p *BoltProvider) groupExistsInternal(name string, bucket *bolt.Bucket) (Group, error) {
  3073. var group Group
  3074. g := bucket.Get([]byte(name))
  3075. if g == nil {
  3076. err := util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", name))
  3077. return group, err
  3078. }
  3079. err := json.Unmarshal(g, &group)
  3080. return group, err
  3081. }
  3082. func (p *BoltProvider) folderExistsInternal(name string, bucket *bolt.Bucket) (vfs.BaseVirtualFolder, error) {
  3083. var folder vfs.BaseVirtualFolder
  3084. f := bucket.Get([]byte(name))
  3085. if f == nil {
  3086. err := util.NewRecordNotFoundError(fmt.Sprintf("folder %q does not exist", name))
  3087. return folder, err
  3088. }
  3089. err := json.Unmarshal(f, &folder)
  3090. return folder, err
  3091. }
  3092. func (p *BoltProvider) addFolderInternal(folder vfs.BaseVirtualFolder, bucket *bolt.Bucket) error {
  3093. id, err := bucket.NextSequence()
  3094. if err != nil {
  3095. return err
  3096. }
  3097. folder.ID = int64(id)
  3098. buf, err := json.Marshal(folder)
  3099. if err != nil {
  3100. return err
  3101. }
  3102. return bucket.Put([]byte(folder.Name), buf)
  3103. }
  3104. func (p *BoltProvider) removeRoleFromUser(username, role string, bucket *bolt.Bucket) error {
  3105. u := bucket.Get([]byte(username))
  3106. if u == nil {
  3107. providerLog(logger.LevelWarn, "user %q does not exist, cannot remove role %q", username, role)
  3108. return nil
  3109. }
  3110. var user User
  3111. err := json.Unmarshal(u, &user)
  3112. if err != nil {
  3113. return err
  3114. }
  3115. if user.Role == role {
  3116. user.Role = ""
  3117. buf, err := json.Marshal(user)
  3118. if err != nil {
  3119. return err
  3120. }
  3121. return bucket.Put([]byte(user.Username), buf)
  3122. }
  3123. providerLog(logger.LevelError, "user %q does not have the expected role %q, actual %q", username, role, user.Role)
  3124. return nil
  3125. }
  3126. func (p *BoltProvider) addAdminToRole(username, roleName string, bucket *bolt.Bucket) error {
  3127. if roleName == "" {
  3128. return nil
  3129. }
  3130. r := bucket.Get([]byte(roleName))
  3131. if r == nil {
  3132. return fmt.Errorf("%w: role %q does not exist", ErrForeignKeyViolated, roleName)
  3133. }
  3134. var role Role
  3135. err := json.Unmarshal(r, &role)
  3136. if err != nil {
  3137. return err
  3138. }
  3139. if !slices.Contains(role.Admins, username) {
  3140. role.Admins = append(role.Admins, username)
  3141. buf, err := json.Marshal(role)
  3142. if err != nil {
  3143. return err
  3144. }
  3145. return bucket.Put([]byte(role.Name), buf)
  3146. }
  3147. return nil
  3148. }
  3149. func (p *BoltProvider) removeAdminFromRole(username, roleName string, bucket *bolt.Bucket) error {
  3150. if roleName == "" {
  3151. return nil
  3152. }
  3153. r := bucket.Get([]byte(roleName))
  3154. if r == nil {
  3155. providerLog(logger.LevelWarn, "role %q does not exist, cannot remove admin %q", roleName, username)
  3156. return nil
  3157. }
  3158. var role Role
  3159. err := json.Unmarshal(r, &role)
  3160. if err != nil {
  3161. return err
  3162. }
  3163. if slices.Contains(role.Admins, username) {
  3164. var admins []string
  3165. for _, admin := range role.Admins {
  3166. if admin != username {
  3167. admins = append(admins, admin)
  3168. }
  3169. }
  3170. role.Admins = util.RemoveDuplicates(admins, false)
  3171. buf, err := json.Marshal(role)
  3172. if err != nil {
  3173. return err
  3174. }
  3175. return bucket.Put([]byte(role.Name), buf)
  3176. }
  3177. return nil
  3178. }
  3179. func (p *BoltProvider) addUserToRole(username, roleName string, bucket *bolt.Bucket) error {
  3180. if roleName == "" {
  3181. return nil
  3182. }
  3183. r := bucket.Get([]byte(roleName))
  3184. if r == nil {
  3185. return fmt.Errorf("%w: role %q does not exist", ErrForeignKeyViolated, roleName)
  3186. }
  3187. var role Role
  3188. err := json.Unmarshal(r, &role)
  3189. if err != nil {
  3190. return err
  3191. }
  3192. if !slices.Contains(role.Users, username) {
  3193. role.Users = append(role.Users, username)
  3194. buf, err := json.Marshal(role)
  3195. if err != nil {
  3196. return err
  3197. }
  3198. return bucket.Put([]byte(role.Name), buf)
  3199. }
  3200. return nil
  3201. }
  3202. func (p *BoltProvider) removeUserFromRole(username, roleName string, bucket *bolt.Bucket) error {
  3203. if roleName == "" {
  3204. return nil
  3205. }
  3206. r := bucket.Get([]byte(roleName))
  3207. if r == nil {
  3208. providerLog(logger.LevelWarn, "role %q does not exist, cannot remove admin %q", roleName, username)
  3209. return nil
  3210. }
  3211. var role Role
  3212. err := json.Unmarshal(r, &role)
  3213. if err != nil {
  3214. return err
  3215. }
  3216. if slices.Contains(role.Users, username) {
  3217. var users []string
  3218. for _, user := range role.Users {
  3219. if user != username {
  3220. users = append(users, user)
  3221. }
  3222. }
  3223. users = util.RemoveDuplicates(users, false)
  3224. role.Users = users
  3225. buf, err := json.Marshal(role)
  3226. if err != nil {
  3227. return err
  3228. }
  3229. return bucket.Put([]byte(role.Name), buf)
  3230. }
  3231. return nil
  3232. }
  3233. func (p *BoltProvider) addRuleToActionMapping(ruleName, actionName string, bucket *bolt.Bucket) error {
  3234. a := bucket.Get([]byte(actionName))
  3235. if a == nil {
  3236. return util.NewGenericError(fmt.Sprintf("action %q does not exist", actionName))
  3237. }
  3238. var action BaseEventAction
  3239. err := json.Unmarshal(a, &action)
  3240. if err != nil {
  3241. return err
  3242. }
  3243. if !slices.Contains(action.Rules, ruleName) {
  3244. action.Rules = append(action.Rules, ruleName)
  3245. buf, err := json.Marshal(action)
  3246. if err != nil {
  3247. return err
  3248. }
  3249. return bucket.Put([]byte(action.Name), buf)
  3250. }
  3251. return nil
  3252. }
  3253. func (p *BoltProvider) removeRuleFromActionMapping(ruleName, actionName string, bucket *bolt.Bucket) error {
  3254. a := bucket.Get([]byte(actionName))
  3255. if a == nil {
  3256. providerLog(logger.LevelWarn, "action %q does not exist, cannot remove from mapping", actionName)
  3257. return nil
  3258. }
  3259. var action BaseEventAction
  3260. err := json.Unmarshal(a, &action)
  3261. if err != nil {
  3262. return err
  3263. }
  3264. if slices.Contains(action.Rules, ruleName) {
  3265. var rules []string
  3266. for _, r := range action.Rules {
  3267. if r != ruleName {
  3268. rules = append(rules, r)
  3269. }
  3270. }
  3271. action.Rules = util.RemoveDuplicates(rules, false)
  3272. buf, err := json.Marshal(action)
  3273. if err != nil {
  3274. return err
  3275. }
  3276. return bucket.Put([]byte(action.Name), buf)
  3277. }
  3278. return nil
  3279. }
  3280. func (p *BoltProvider) addUserToGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
  3281. g := bucket.Get([]byte(groupname))
  3282. if g == nil {
  3283. return util.NewGenericError(fmt.Sprintf("group %q does not exist", groupname))
  3284. }
  3285. var group Group
  3286. err := json.Unmarshal(g, &group)
  3287. if err != nil {
  3288. return err
  3289. }
  3290. if !slices.Contains(group.Users, username) {
  3291. group.Users = append(group.Users, username)
  3292. buf, err := json.Marshal(group)
  3293. if err != nil {
  3294. return err
  3295. }
  3296. return bucket.Put([]byte(group.Name), buf)
  3297. }
  3298. return nil
  3299. }
  3300. func (p *BoltProvider) removeUserFromGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
  3301. g := bucket.Get([]byte(groupname))
  3302. if g == nil {
  3303. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
  3304. }
  3305. var group Group
  3306. err := json.Unmarshal(g, &group)
  3307. if err != nil {
  3308. return err
  3309. }
  3310. var users []string
  3311. for _, u := range group.Users {
  3312. if u != username {
  3313. users = append(users, u)
  3314. }
  3315. }
  3316. group.Users = util.RemoveDuplicates(users, false)
  3317. buf, err := json.Marshal(group)
  3318. if err != nil {
  3319. return err
  3320. }
  3321. return bucket.Put([]byte(group.Name), buf)
  3322. }
  3323. func (p *BoltProvider) addAdminToGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
  3324. g := bucket.Get([]byte(groupname))
  3325. if g == nil {
  3326. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
  3327. }
  3328. var group Group
  3329. err := json.Unmarshal(g, &group)
  3330. if err != nil {
  3331. return err
  3332. }
  3333. if !slices.Contains(group.Admins, username) {
  3334. group.Admins = append(group.Admins, username)
  3335. buf, err := json.Marshal(group)
  3336. if err != nil {
  3337. return err
  3338. }
  3339. return bucket.Put([]byte(group.Name), buf)
  3340. }
  3341. return nil
  3342. }
  3343. func (p *BoltProvider) removeAdminFromGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
  3344. g := bucket.Get([]byte(groupname))
  3345. if g == nil {
  3346. return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
  3347. }
  3348. var group Group
  3349. err := json.Unmarshal(g, &group)
  3350. if err != nil {
  3351. return err
  3352. }
  3353. var admins []string
  3354. for _, a := range group.Admins {
  3355. if a != username {
  3356. admins = append(admins, a)
  3357. }
  3358. }
  3359. group.Admins = util.RemoveDuplicates(admins, false)
  3360. buf, err := json.Marshal(group)
  3361. if err != nil {
  3362. return err
  3363. }
  3364. return bucket.Put([]byte(group.Name), buf)
  3365. }
  3366. func (p *BoltProvider) removeGroupFromAdminMapping(groupName, adminName string, bucket *bolt.Bucket) error {
  3367. var a []byte
  3368. if a = bucket.Get([]byte(adminName)); a == nil {
  3369. // the admin does not exist so there is no associated group
  3370. return nil
  3371. }
  3372. var admin Admin
  3373. err := json.Unmarshal(a, &admin)
  3374. if err != nil {
  3375. return err
  3376. }
  3377. var newGroups []AdminGroupMapping
  3378. for _, g := range admin.Groups {
  3379. if g.Name != groupName {
  3380. newGroups = append(newGroups, g)
  3381. }
  3382. }
  3383. admin.Groups = newGroups
  3384. buf, err := json.Marshal(admin)
  3385. if err != nil {
  3386. return err
  3387. }
  3388. return bucket.Put([]byte(adminName), buf)
  3389. }
  3390. func (p *BoltProvider) addRelationToFolderMapping(folderName string, user *User, group *Group, bucket *bolt.Bucket) error {
  3391. f := bucket.Get([]byte(folderName))
  3392. if f == nil {
  3393. return util.NewGenericError(fmt.Sprintf("folder %q does not exist", folderName))
  3394. }
  3395. var folder vfs.BaseVirtualFolder
  3396. err := json.Unmarshal(f, &folder)
  3397. if err != nil {
  3398. return err
  3399. }
  3400. updated := false
  3401. if user != nil && !slices.Contains(folder.Users, user.Username) {
  3402. folder.Users = append(folder.Users, user.Username)
  3403. updated = true
  3404. }
  3405. if group != nil && !slices.Contains(folder.Groups, group.Name) {
  3406. folder.Groups = append(folder.Groups, group.Name)
  3407. updated = true
  3408. }
  3409. if !updated {
  3410. return nil
  3411. }
  3412. buf, err := json.Marshal(folder)
  3413. if err != nil {
  3414. return err
  3415. }
  3416. return bucket.Put([]byte(folder.Name), buf)
  3417. }
  3418. func (p *BoltProvider) removeRelationFromFolderMapping(folder vfs.VirtualFolder, username, groupname string,
  3419. bucket *bolt.Bucket,
  3420. ) error {
  3421. var f []byte
  3422. if f = bucket.Get([]byte(folder.Name)); f == nil {
  3423. // the folder does not exist so there is no associated user/group
  3424. return nil
  3425. }
  3426. var baseFolder vfs.BaseVirtualFolder
  3427. err := json.Unmarshal(f, &baseFolder)
  3428. if err != nil {
  3429. return err
  3430. }
  3431. found := false
  3432. if username != "" {
  3433. found = true
  3434. var newUserMapping []string
  3435. for _, u := range baseFolder.Users {
  3436. if u != username {
  3437. newUserMapping = append(newUserMapping, u)
  3438. }
  3439. }
  3440. baseFolder.Users = newUserMapping
  3441. }
  3442. if groupname != "" {
  3443. found = true
  3444. var newGroupMapping []string
  3445. for _, g := range baseFolder.Groups {
  3446. if g != groupname {
  3447. newGroupMapping = append(newGroupMapping, g)
  3448. }
  3449. }
  3450. baseFolder.Groups = newGroupMapping
  3451. }
  3452. if !found {
  3453. return nil
  3454. }
  3455. buf, err := json.Marshal(baseFolder)
  3456. if err != nil {
  3457. return err
  3458. }
  3459. return bucket.Put([]byte(folder.Name), buf)
  3460. }
  3461. func (p *BoltProvider) updateUserRelations(tx *bolt.Tx, user *User, oldUser User) error {
  3462. foldersBucket, err := p.getFoldersBucket(tx)
  3463. if err != nil {
  3464. return err
  3465. }
  3466. groupsBucket, err := p.getGroupsBucket(tx)
  3467. if err != nil {
  3468. return err
  3469. }
  3470. rolesBucket, err := p.getRolesBucket(tx)
  3471. if err != nil {
  3472. return err
  3473. }
  3474. for idx := range oldUser.VirtualFolders {
  3475. err = p.removeRelationFromFolderMapping(oldUser.VirtualFolders[idx], oldUser.Username, "", foldersBucket)
  3476. if err != nil {
  3477. return err
  3478. }
  3479. }
  3480. for idx := range oldUser.Groups {
  3481. err = p.removeUserFromGroupMapping(user.Username, oldUser.Groups[idx].Name, groupsBucket)
  3482. if err != nil {
  3483. return err
  3484. }
  3485. }
  3486. if err = p.removeUserFromRole(oldUser.Username, oldUser.Role, rolesBucket); err != nil {
  3487. return err
  3488. }
  3489. sort.Slice(user.VirtualFolders, func(i, j int) bool {
  3490. return user.VirtualFolders[i].Name < user.VirtualFolders[j].Name
  3491. })
  3492. for idx := range user.VirtualFolders {
  3493. err = p.addRelationToFolderMapping(user.VirtualFolders[idx].Name, user, nil, foldersBucket)
  3494. if err != nil {
  3495. return err
  3496. }
  3497. }
  3498. sort.Slice(user.Groups, func(i, j int) bool {
  3499. return user.Groups[i].Name < user.Groups[j].Name
  3500. })
  3501. for idx := range user.Groups {
  3502. err = p.addUserToGroupMapping(user.Username, user.Groups[idx].Name, groupsBucket)
  3503. if err != nil {
  3504. return err
  3505. }
  3506. }
  3507. return p.addUserToRole(user.Username, user.Role, rolesBucket)
  3508. }
  3509. func (p *BoltProvider) adminExistsInternal(tx *bolt.Tx, username string) error {
  3510. bucket, err := p.getAdminsBucket(tx)
  3511. if err != nil {
  3512. return err
  3513. }
  3514. a := bucket.Get([]byte(username))
  3515. if a == nil {
  3516. return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", username))
  3517. }
  3518. return nil
  3519. }
  3520. func (p *BoltProvider) userExistsInternal(tx *bolt.Tx, username string) error {
  3521. bucket, err := p.getUsersBucket(tx)
  3522. if err != nil {
  3523. return err
  3524. }
  3525. u := bucket.Get([]byte(username))
  3526. if u == nil {
  3527. return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
  3528. }
  3529. return nil
  3530. }
  3531. func (p *BoltProvider) deleteRelatedShares(tx *bolt.Tx, username string) error {
  3532. bucket, err := p.getSharesBucket(tx)
  3533. if err != nil {
  3534. return err
  3535. }
  3536. var toRemove []string
  3537. cursor := bucket.Cursor()
  3538. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  3539. var share Share
  3540. err = json.Unmarshal(v, &share)
  3541. if err != nil {
  3542. return err
  3543. }
  3544. if share.Username == username {
  3545. toRemove = append(toRemove, share.ShareID)
  3546. }
  3547. }
  3548. for _, k := range toRemove {
  3549. if err := bucket.Delete([]byte(k)); err != nil {
  3550. return err
  3551. }
  3552. }
  3553. return nil
  3554. }
  3555. func (p *BoltProvider) deleteRelatedAPIKey(tx *bolt.Tx, username string, scope APIKeyScope) error {
  3556. bucket, err := p.getAPIKeysBucket(tx)
  3557. if err != nil {
  3558. return err
  3559. }
  3560. var toRemove []string
  3561. cursor := bucket.Cursor()
  3562. for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
  3563. var apiKey APIKey
  3564. err = json.Unmarshal(v, &apiKey)
  3565. if err != nil {
  3566. return err
  3567. }
  3568. if scope == APIKeyScopeUser {
  3569. if apiKey.User == username {
  3570. toRemove = append(toRemove, apiKey.KeyID)
  3571. }
  3572. } else {
  3573. if apiKey.Admin == username {
  3574. toRemove = append(toRemove, apiKey.KeyID)
  3575. }
  3576. }
  3577. }
  3578. for _, k := range toRemove {
  3579. if err := bucket.Delete([]byte(k)); err != nil {
  3580. return err
  3581. }
  3582. }
  3583. return nil
  3584. }
  3585. func (p *BoltProvider) getSharesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3586. var err error
  3587. bucket := tx.Bucket(sharesBucket)
  3588. if bucket == nil {
  3589. err = errors.New("unable to find shares bucket, bolt database structure not correcly defined")
  3590. }
  3591. return bucket, err
  3592. }
  3593. func (p *BoltProvider) getAPIKeysBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3594. var err error
  3595. bucket := tx.Bucket(apiKeysBucket)
  3596. if bucket == nil {
  3597. err = errors.New("unable to find api keys bucket, bolt database structure not correcly defined")
  3598. }
  3599. return bucket, err
  3600. }
  3601. func (p *BoltProvider) getAdminsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3602. var err error
  3603. bucket := tx.Bucket(adminsBucket)
  3604. if bucket == nil {
  3605. err = errors.New("unable to find admins bucket, bolt database structure not correcly defined")
  3606. }
  3607. return bucket, err
  3608. }
  3609. func (p *BoltProvider) getUsersBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3610. var err error
  3611. bucket := tx.Bucket(usersBucket)
  3612. if bucket == nil {
  3613. err = errors.New("unable to find users bucket, bolt database structure not correcly defined")
  3614. }
  3615. return bucket, err
  3616. }
  3617. func (p *BoltProvider) getGroupsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3618. var err error
  3619. bucket := tx.Bucket(groupsBucket)
  3620. if bucket == nil {
  3621. err = fmt.Errorf("unable to find groups bucket, bolt database structure not correcly defined")
  3622. }
  3623. return bucket, err
  3624. }
  3625. func (p *BoltProvider) getRolesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3626. var err error
  3627. bucket := tx.Bucket(rolesBucket)
  3628. if bucket == nil {
  3629. err = fmt.Errorf("unable to find roles bucket, bolt database structure not correcly defined")
  3630. }
  3631. return bucket, err
  3632. }
  3633. func (p *BoltProvider) getIPListsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3634. var err error
  3635. bucket := tx.Bucket(rolesBucket)
  3636. if bucket == nil {
  3637. err = fmt.Errorf("unable to find IP lists bucket, bolt database structure not correcly defined")
  3638. }
  3639. return bucket, err
  3640. }
  3641. func (p *BoltProvider) getFoldersBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3642. var err error
  3643. bucket := tx.Bucket(foldersBucket)
  3644. if bucket == nil {
  3645. err = fmt.Errorf("unable to find folders bucket, bolt database structure not correcly defined")
  3646. }
  3647. return bucket, err
  3648. }
  3649. func (p *BoltProvider) getActionsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3650. var err error
  3651. bucket := tx.Bucket(actionsBucket)
  3652. if bucket == nil {
  3653. err = fmt.Errorf("unable to find event actions bucket, bolt database structure not correcly defined")
  3654. }
  3655. return bucket, err
  3656. }
  3657. func (p *BoltProvider) getRulesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
  3658. var err error
  3659. bucket := tx.Bucket(rulesBucket)
  3660. if bucket == nil {
  3661. err = fmt.Errorf("unable to find event rules bucket, bolt database structure not correcly defined")
  3662. }
  3663. return bucket, err
  3664. }
  3665. func getBoltDatabaseVersion(dbHandle *bolt.DB) (schemaVersion, error) {
  3666. var dbVersion schemaVersion
  3667. err := dbHandle.View(func(tx *bolt.Tx) error {
  3668. bucket := tx.Bucket(dbVersionBucket)
  3669. if bucket == nil {
  3670. return fmt.Errorf("unable to find database schema version bucket")
  3671. }
  3672. v := bucket.Get(dbVersionKey)
  3673. if v == nil {
  3674. dbVersion = schemaVersion{
  3675. Version: 29,
  3676. }
  3677. return nil
  3678. }
  3679. return json.Unmarshal(v, &dbVersion)
  3680. })
  3681. return dbVersion, err
  3682. }
  3683. /*func updateBoltDatabaseVersion(dbHandle *bolt.DB, version int) error {
  3684. err := dbHandle.Update(func(tx *bolt.Tx) error {
  3685. bucket := tx.Bucket(dbVersionBucket)
  3686. if bucket == nil {
  3687. return fmt.Errorf("unable to find database schema version bucket")
  3688. }
  3689. newDbVersion := schemaVersion{
  3690. Version: version,
  3691. }
  3692. buf, err := json.Marshal(newDbVersion)
  3693. if err != nil {
  3694. return err
  3695. }
  3696. return bucket.Put(dbVersionKey, buf)
  3697. })
  3698. return err
  3699. }*/