| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895 |
- // Copyright (C) 2019 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- //go:build !nobolt
- // +build !nobolt
- package dataprovider
- import (
- "bytes"
- "crypto/x509"
- "encoding/json"
- "errors"
- "fmt"
- "net/netip"
- "path/filepath"
- "sort"
- "time"
- bolt "go.etcd.io/bbolt"
- "github.com/drakkan/sftpgo/v2/internal/logger"
- "github.com/drakkan/sftpgo/v2/internal/util"
- "github.com/drakkan/sftpgo/v2/internal/version"
- "github.com/drakkan/sftpgo/v2/internal/vfs"
- )
- const (
- boltDatabaseVersion = 28
- )
- var (
- usersBucket = []byte("users")
- groupsBucket = []byte("groups")
- foldersBucket = []byte("folders")
- adminsBucket = []byte("admins")
- apiKeysBucket = []byte("api_keys")
- sharesBucket = []byte("shares")
- actionsBucket = []byte("events_actions")
- rulesBucket = []byte("events_rules")
- rolesBucket = []byte("roles")
- ipListsBucket = []byte("ip_lists")
- configsBucket = []byte("configs")
- dbVersionBucket = []byte("db_version")
- dbVersionKey = []byte("version")
- configsKey = []byte("configs")
- boltBuckets = [][]byte{usersBucket, groupsBucket, foldersBucket, adminsBucket, apiKeysBucket,
- sharesBucket, actionsBucket, rulesBucket, rolesBucket, ipListsBucket, configsBucket, dbVersionBucket}
- )
- // BoltProvider defines the auth provider for bolt key/value store
- type BoltProvider struct {
- dbHandle *bolt.DB
- }
- func init() {
- version.AddFeature("+bolt")
- }
- func initializeBoltProvider(basePath string) error {
- var err error
- dbPath := config.Name
- if !util.IsFileInputValid(dbPath) {
- return fmt.Errorf("invalid database path: %q", dbPath)
- }
- if !filepath.IsAbs(dbPath) {
- dbPath = filepath.Join(basePath, dbPath)
- }
- dbHandle, err := bolt.Open(dbPath, 0600, &bolt.Options{
- NoGrowSync: false,
- FreelistType: bolt.FreelistArrayType,
- Timeout: 5 * time.Second})
- if err == nil {
- providerLog(logger.LevelDebug, "bolt key store handle created")
- for _, bucket := range boltBuckets {
- if err := dbHandle.Update(func(tx *bolt.Tx) error {
- _, e := tx.CreateBucketIfNotExists(bucket)
- return e
- }); err != nil {
- providerLog(logger.LevelError, "error creating bucket %q: %v", string(bucket), err)
- }
- }
- provider = &BoltProvider{dbHandle: dbHandle}
- } else {
- providerLog(logger.LevelError, "error creating bolt key/value store handler: %v", err)
- }
- return err
- }
- func (p *BoltProvider) checkAvailability() error {
- _, err := getBoltDatabaseVersion(p.dbHandle)
- return err
- }
- func (p *BoltProvider) validateUserAndTLSCert(username, protocol string, tlsCert *x509.Certificate) (User, error) {
- var user User
- if tlsCert == nil {
- return user, errors.New("TLS certificate cannot be null or empty")
- }
- user, err := p.userExists(username, "")
- if err != nil {
- providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
- return user, err
- }
- return checkUserAndTLSCertificate(&user, protocol, tlsCert)
- }
- func (p *BoltProvider) validateUserAndPass(username, password, ip, protocol string) (User, error) {
- user, err := p.userExists(username, "")
- if err != nil {
- providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
- return user, err
- }
- return checkUserAndPass(&user, password, ip, protocol)
- }
- func (p *BoltProvider) validateAdminAndPass(username, password, ip string) (Admin, error) {
- admin, err := p.adminExists(username)
- if err != nil {
- providerLog(logger.LevelWarn, "error authenticating admin %q: %v", username, err)
- return admin, err
- }
- err = admin.checkUserAndPass(password, ip)
- return admin, err
- }
- func (p *BoltProvider) validateUserAndPubKey(username string, pubKey []byte, isSSHCert bool) (User, string, error) {
- var user User
- if len(pubKey) == 0 {
- return user, "", errors.New("credentials cannot be null or empty")
- }
- user, err := p.userExists(username, "")
- if err != nil {
- providerLog(logger.LevelWarn, "error authenticating user %q: %v", username, err)
- return user, "", err
- }
- return checkUserAndPubKey(&user, pubKey, isSSHCert)
- }
- func (p *BoltProvider) updateAPIKeyLastUse(keyID string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(keyID)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("key %q does not exist, unable to update last use", keyID))
- }
- var apiKey APIKey
- err = json.Unmarshal(u, &apiKey)
- if err != nil {
- return err
- }
- apiKey.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(apiKey)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(keyID), buf)
- if err != nil {
- providerLog(logger.LevelWarn, "error updating last use for key %q: %v", keyID, err)
- return err
- }
- providerLog(logger.LevelDebug, "last use updated for key %q", keyID)
- return nil
- })
- }
- func (p *BoltProvider) setUpdatedAt(username string) {
- p.dbHandle.Update(func(tx *bolt.Tx) error { //nolint:errcheck
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update updated at", username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(username), buf)
- if err == nil {
- providerLog(logger.LevelDebug, "updated at set for user %q", username)
- setLastUserUpdate()
- } else {
- providerLog(logger.LevelWarn, "error setting updated_at for user %q: %v", username, err)
- }
- return err
- })
- }
- func (p *BoltProvider) updateLastLogin(username string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update last login", username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- user.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(username), buf)
- if err != nil {
- providerLog(logger.LevelWarn, "error updating last login for user %q: %v", username, err)
- } else {
- providerLog(logger.LevelDebug, "last login updated for user %q", username)
- }
- return err
- })
- }
- func (p *BoltProvider) updateAdminLastLogin(username string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(username)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("admin %q does not exist, unable to update last login", username))
- }
- var admin Admin
- err = json.Unmarshal(a, &admin)
- if err != nil {
- return err
- }
- admin.LastLogin = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(admin)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(username), buf)
- if err == nil {
- providerLog(logger.LevelDebug, "last login updated for admin %q", username)
- return err
- }
- providerLog(logger.LevelWarn, "error updating last login for admin %q: %v", username, err)
- return err
- })
- }
- func (p *BoltProvider) updateTransferQuota(username string, uploadSize, downloadSize int64, reset bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update transfer quota",
- username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- if !reset {
- user.UsedUploadDataTransfer += uploadSize
- user.UsedDownloadDataTransfer += downloadSize
- } else {
- user.UsedUploadDataTransfer = uploadSize
- user.UsedDownloadDataTransfer = downloadSize
- }
- user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(username), buf)
- providerLog(logger.LevelDebug, "transfer quota updated for user %q, ul increment: %v dl increment: %v is reset? %v",
- username, uploadSize, downloadSize, reset)
- return err
- })
- }
- func (p *BoltProvider) updateQuota(username string, filesAdd int, sizeAdd int64, reset bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to update quota", username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- if reset {
- user.UsedQuotaSize = sizeAdd
- user.UsedQuotaFiles = filesAdd
- } else {
- user.UsedQuotaSize += sizeAdd
- user.UsedQuotaFiles += filesAdd
- }
- user.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(username), buf)
- providerLog(logger.LevelDebug, "quota updated for user %q, files increment: %v size increment: %v is reset? %v",
- username, filesAdd, sizeAdd, reset)
- return err
- })
- }
- func (p *BoltProvider) getUsedQuota(username string) (int, int64, int64, int64, error) {
- user, err := p.userExists(username, "")
- if err != nil {
- providerLog(logger.LevelError, "unable to get quota for user %v error: %v", username, err)
- return 0, 0, 0, 0, err
- }
- return user.UsedQuotaFiles, user.UsedQuotaSize, user.UsedUploadDataTransfer, user.UsedDownloadDataTransfer, err
- }
- func (p *BoltProvider) adminExists(username string) (Admin, error) {
- var admin Admin
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- a := bucket.Get([]byte(username))
- if a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", username))
- }
- return json.Unmarshal(a, &admin)
- })
- return admin, err
- }
- func (p *BoltProvider) addAdmin(admin *Admin) error {
- err := admin.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- groupBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- if a := bucket.Get([]byte(admin.Username)); a != nil {
- return fmt.Errorf("admin %q already exists", admin.Username)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- admin.ID = int64(id)
- admin.LastLogin = 0
- admin.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- sort.Slice(admin.Groups, func(i, j int) bool {
- return admin.Groups[i].Name < admin.Groups[j].Name
- })
- for idx := range admin.Groups {
- err = p.addAdminToGroupMapping(admin.Username, admin.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- if err = p.addAdminToRole(admin.Username, admin.Role, rolesBucket); err != nil {
- return err
- }
- buf, err := json.Marshal(admin)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(admin.Username), buf)
- })
- }
- func (p *BoltProvider) updateAdmin(admin *Admin) error {
- err := admin.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- groupBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(admin.Username)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", admin.Username))
- }
- var oldAdmin Admin
- err = json.Unmarshal(a, &oldAdmin)
- if err != nil {
- return err
- }
- if err = p.removeAdminFromRole(oldAdmin.Username, oldAdmin.Role, rolesBucket); err != nil {
- return err
- }
- for idx := range oldAdmin.Groups {
- err = p.removeAdminFromGroupMapping(oldAdmin.Username, oldAdmin.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- if err = p.addAdminToRole(admin.Username, admin.Role, rolesBucket); err != nil {
- return err
- }
- sort.Slice(admin.Groups, func(i, j int) bool {
- return admin.Groups[i].Name < admin.Groups[j].Name
- })
- for idx := range admin.Groups {
- err = p.addAdminToGroupMapping(admin.Username, admin.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- admin.ID = oldAdmin.ID
- admin.CreatedAt = oldAdmin.CreatedAt
- admin.LastLogin = oldAdmin.LastLogin
- admin.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(admin)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(admin.Username), buf)
- })
- }
- func (p *BoltProvider) deleteAdmin(admin Admin) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(admin.Username)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", admin.Username))
- }
- var oldAdmin Admin
- err = json.Unmarshal(a, &oldAdmin)
- if err != nil {
- return err
- }
- if len(oldAdmin.Groups) > 0 {
- groupBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- for idx := range oldAdmin.Groups {
- err = p.removeAdminFromGroupMapping(oldAdmin.Username, oldAdmin.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- }
- if oldAdmin.Role != "" {
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- if err = p.removeAdminFromRole(oldAdmin.Username, oldAdmin.Role, rolesBucket); err != nil {
- return err
- }
- }
- if err := p.deleteRelatedAPIKey(tx, admin.Username, APIKeyScopeAdmin); err != nil {
- return err
- }
- return bucket.Delete([]byte(admin.Username))
- })
- }
- func (p *BoltProvider) getAdmins(limit int, offset int, order string) ([]Admin, error) {
- admins := make([]Admin, 0, limit)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var admin Admin
- err = json.Unmarshal(v, &admin)
- if err != nil {
- return err
- }
- admin.HideConfidentialData()
- admins = append(admins, admin)
- if len(admins) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var admin Admin
- err = json.Unmarshal(v, &admin)
- if err != nil {
- return err
- }
- admin.HideConfidentialData()
- admins = append(admins, admin)
- if len(admins) >= limit {
- break
- }
- }
- }
- return err
- })
- return admins, err
- }
- func (p *BoltProvider) dumpAdmins() ([]Admin, error) {
- admins := make([]Admin, 0, 30)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var admin Admin
- err = json.Unmarshal(v, &admin)
- if err != nil {
- return err
- }
- admins = append(admins, admin)
- }
- return err
- })
- return admins, err
- }
- func (p *BoltProvider) userExists(username, role string) (User, error) {
- var user User
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- u := bucket.Get([]byte(username))
- if u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- user, err = p.joinUserAndFolders(u, foldersBucket)
- if err != nil {
- return err
- }
- if !user.hasRole(role) {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
- }
- return nil
- })
- return user, err
- }
- func (p *BoltProvider) addUser(user *User) error {
- err := ValidateUser(user)
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- groupBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- if u := bucket.Get([]byte(user.Username)); u != nil {
- return fmt.Errorf("username %v already exists", user.Username)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- user.ID = int64(id)
- user.LastQuotaUpdate = 0
- user.UsedQuotaSize = 0
- user.UsedQuotaFiles = 0
- user.UsedUploadDataTransfer = 0
- user.UsedDownloadDataTransfer = 0
- user.LastLogin = 0
- user.FirstDownload = 0
- user.FirstUpload = 0
- user.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- if err := p.addUserToRole(user.Username, user.Role, rolesBucket); err != nil {
- return err
- }
- sort.Slice(user.VirtualFolders, func(i, j int) bool {
- return user.VirtualFolders[i].Name < user.VirtualFolders[j].Name
- })
- for idx := range user.VirtualFolders {
- err = p.addRelationToFolderMapping(user.VirtualFolders[idx].Name, user, nil, foldersBucket)
- if err != nil {
- return err
- }
- }
- sort.Slice(user.Groups, func(i, j int) bool {
- return user.Groups[i].Name < user.Groups[j].Name
- })
- for idx := range user.Groups {
- err = p.addUserToGroupMapping(user.Username, user.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(user.Username), buf)
- })
- }
- func (p *BoltProvider) updateUser(user *User) error {
- err := ValidateUser(user)
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(user.Username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", user.Username))
- }
- var oldUser User
- err = json.Unmarshal(u, &oldUser)
- if err != nil {
- return err
- }
- if err = p.updateUserRelations(tx, user, oldUser); err != nil {
- return err
- }
- user.ID = oldUser.ID
- user.LastQuotaUpdate = oldUser.LastQuotaUpdate
- user.UsedQuotaSize = oldUser.UsedQuotaSize
- user.UsedQuotaFiles = oldUser.UsedQuotaFiles
- user.UsedUploadDataTransfer = oldUser.UsedUploadDataTransfer
- user.UsedDownloadDataTransfer = oldUser.UsedDownloadDataTransfer
- user.LastLogin = oldUser.LastLogin
- user.FirstDownload = oldUser.FirstDownload
- user.FirstUpload = oldUser.FirstUpload
- user.CreatedAt = oldUser.CreatedAt
- user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(user.Username), buf)
- if err == nil {
- setLastUserUpdate()
- }
- return err
- })
- }
- func (p *BoltProvider) deleteUser(user User, _ bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- groupBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(user.Username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", user.Username))
- }
- var oldUser User
- err = json.Unmarshal(u, &oldUser)
- if err != nil {
- return err
- }
- if err := p.removeUserFromRole(oldUser.Username, oldUser.Role, rolesBucket); err != nil {
- return err
- }
- for idx := range oldUser.VirtualFolders {
- err = p.removeRelationFromFolderMapping(oldUser.VirtualFolders[idx], oldUser.Username, "", foldersBucket)
- if err != nil {
- return err
- }
- }
- for idx := range oldUser.Groups {
- err = p.removeUserFromGroupMapping(oldUser.Username, oldUser.Groups[idx].Name, groupBucket)
- if err != nil {
- return err
- }
- }
- if err := p.deleteRelatedAPIKey(tx, user.Username, APIKeyScopeUser); err != nil {
- return err
- }
- if err := p.deleteRelatedShares(tx, user.Username); err != nil {
- return err
- }
- return bucket.Delete([]byte(user.Username))
- })
- }
- func (p *BoltProvider) updateUserPassword(username, password string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- user.Password = password
- user.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(username), buf)
- })
- }
- func (p *BoltProvider) dumpUsers() ([]User, error) {
- users := make([]User, 0, 100)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- user, err := p.joinUserAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- users = append(users, user)
- }
- return err
- })
- return users, err
- }
- func (p *BoltProvider) getRecentlyUpdatedUsers(after int64) ([]User, error) {
- if getLastUserUpdate() < after {
- return nil, nil
- }
- users := make([]User, 0, 10)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- groupsBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var user User
- err := json.Unmarshal(v, &user)
- if err != nil {
- return err
- }
- if user.UpdatedAt < after {
- continue
- }
- if len(user.VirtualFolders) > 0 {
- var folders []vfs.VirtualFolder
- for idx := range user.VirtualFolders {
- folder := &user.VirtualFolders[idx]
- baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
- if err != nil {
- continue
- }
- folder.BaseVirtualFolder = baseFolder
- folders = append(folders, *folder)
- }
- user.VirtualFolders = folders
- }
- if len(user.Groups) > 0 {
- groupMapping := make(map[string]Group)
- for idx := range user.Groups {
- group, err := p.groupExistsInternal(user.Groups[idx].Name, groupsBucket)
- if err != nil {
- continue
- }
- groupMapping[group.Name] = group
- }
- user.applyGroupSettings(groupMapping)
- }
- user.SetEmptySecretsIfNil()
- users = append(users, user)
- }
- return err
- })
- return users, err
- }
- func (p *BoltProvider) getUsersForQuotaCheck(toFetch map[string]bool) ([]User, error) {
- users := make([]User, 0, 10)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- groupsBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var user User
- err := json.Unmarshal(v, &user)
- if err != nil {
- return err
- }
- if needFolders, ok := toFetch[user.Username]; ok {
- if needFolders && len(user.VirtualFolders) > 0 {
- var folders []vfs.VirtualFolder
- for idx := range user.VirtualFolders {
- folder := &user.VirtualFolders[idx]
- baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
- if err != nil {
- continue
- }
- folder.BaseVirtualFolder = baseFolder
- folders = append(folders, *folder)
- }
- user.VirtualFolders = folders
- }
- if len(user.Groups) > 0 {
- groupMapping := make(map[string]Group)
- for idx := range user.Groups {
- group, err := p.groupExistsInternal(user.Groups[idx].Name, groupsBucket)
- if err != nil {
- continue
- }
- groupMapping[group.Name] = group
- }
- user.applyGroupSettings(groupMapping)
- }
- user.SetEmptySecretsIfNil()
- user.PrepareForRendering()
- users = append(users, user)
- }
- }
- return nil
- })
- return users, err
- }
- func (p *BoltProvider) getUsers(limit int, offset int, order, role string) ([]User, error) {
- users := make([]User, 0, limit)
- var err error
- if limit <= 0 {
- return users, err
- }
- err = p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- user, err := p.joinUserAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- if !user.hasRole(role) {
- continue
- }
- user.PrepareForRendering()
- users = append(users, user)
- if len(users) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- user, err := p.joinUserAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- if !user.hasRole(role) {
- continue
- }
- user.PrepareForRendering()
- users = append(users, user)
- if len(users) >= limit {
- break
- }
- }
- }
- return err
- })
- return users, err
- }
- func (p *BoltProvider) dumpFolders() ([]vfs.BaseVirtualFolder, error) {
- folders := make([]vfs.BaseVirtualFolder, 0, 50)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var folder vfs.BaseVirtualFolder
- err = json.Unmarshal(v, &folder)
- if err != nil {
- return err
- }
- folders = append(folders, folder)
- }
- return err
- })
- return folders, err
- }
- func (p *BoltProvider) getFolders(limit, offset int, order string, _ bool) ([]vfs.BaseVirtualFolder, error) {
- folders := make([]vfs.BaseVirtualFolder, 0, limit)
- var err error
- if limit <= 0 {
- return folders, err
- }
- err = p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var folder vfs.BaseVirtualFolder
- err = json.Unmarshal(v, &folder)
- if err != nil {
- return err
- }
- folder.PrepareForRendering()
- folders = append(folders, folder)
- if len(folders) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var folder vfs.BaseVirtualFolder
- err = json.Unmarshal(v, &folder)
- if err != nil {
- return err
- }
- folder.PrepareForRendering()
- folders = append(folders, folder)
- if len(folders) >= limit {
- break
- }
- }
- }
- return err
- })
- return folders, err
- }
- func (p *BoltProvider) getFolderByName(name string) (vfs.BaseVirtualFolder, error) {
- var folder vfs.BaseVirtualFolder
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- folder, err = p.folderExistsInternal(name, bucket)
- return err
- })
- return folder, err
- }
- func (p *BoltProvider) addFolder(folder *vfs.BaseVirtualFolder) error {
- err := ValidateFolder(folder)
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- if f := bucket.Get([]byte(folder.Name)); f != nil {
- return fmt.Errorf("folder %v already exists", folder.Name)
- }
- folder.Users = nil
- folder.Groups = nil
- return p.addFolderInternal(*folder, bucket)
- })
- }
- func (p *BoltProvider) updateFolder(folder *vfs.BaseVirtualFolder) error {
- err := ValidateFolder(folder)
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- var f []byte
- if f = bucket.Get([]byte(folder.Name)); f == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("folder %v does not exist", folder.Name))
- }
- var oldFolder vfs.BaseVirtualFolder
- err = json.Unmarshal(f, &oldFolder)
- if err != nil {
- return err
- }
- folder.ID = oldFolder.ID
- folder.LastQuotaUpdate = oldFolder.LastQuotaUpdate
- folder.UsedQuotaFiles = oldFolder.UsedQuotaFiles
- folder.UsedQuotaSize = oldFolder.UsedQuotaSize
- folder.Users = oldFolder.Users
- folder.Groups = oldFolder.Groups
- buf, err := json.Marshal(folder)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(folder.Name), buf)
- })
- }
- func (p *BoltProvider) deleteFolderMappings(folder vfs.BaseVirtualFolder, usersBucket, groupsBucket *bolt.Bucket) error {
- for _, username := range folder.Users {
- var u []byte
- if u = usersBucket.Get([]byte(username)); u == nil {
- continue
- }
- var user User
- err := json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- var folders []vfs.VirtualFolder
- for _, userFolder := range user.VirtualFolders {
- if folder.Name != userFolder.Name {
- folders = append(folders, userFolder)
- }
- }
- user.VirtualFolders = folders
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- err = usersBucket.Put([]byte(user.Username), buf)
- if err != nil {
- return err
- }
- }
- for _, groupname := range folder.Groups {
- var u []byte
- if u = groupsBucket.Get([]byte(groupname)); u == nil {
- continue
- }
- var group Group
- err := json.Unmarshal(u, &group)
- if err != nil {
- return err
- }
- var folders []vfs.VirtualFolder
- for _, groupFolder := range group.VirtualFolders {
- if folder.Name != groupFolder.Name {
- folders = append(folders, groupFolder)
- }
- }
- group.VirtualFolders = folders
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- err = groupsBucket.Put([]byte(group.Name), buf)
- if err != nil {
- return err
- }
- }
- return nil
- }
- func (p *BoltProvider) deleteFolder(baseFolder vfs.BaseVirtualFolder) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- usersBucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- groupsBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- var f []byte
- if f = bucket.Get([]byte(baseFolder.Name)); f == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("folder %v does not exist", baseFolder.Name))
- }
- var folder vfs.BaseVirtualFolder
- err = json.Unmarshal(f, &folder)
- if err != nil {
- return err
- }
- if err = p.deleteFolderMappings(folder, usersBucket, groupsBucket); err != nil {
- return err
- }
- return bucket.Delete([]byte(folder.Name))
- })
- }
- func (p *BoltProvider) updateFolderQuota(name string, filesAdd int, sizeAdd int64, reset bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- var f []byte
- if f = bucket.Get([]byte(name)); f == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("folder %q does not exist, unable to update quota", name))
- }
- var folder vfs.BaseVirtualFolder
- err = json.Unmarshal(f, &folder)
- if err != nil {
- return err
- }
- if reset {
- folder.UsedQuotaSize = sizeAdd
- folder.UsedQuotaFiles = filesAdd
- } else {
- folder.UsedQuotaSize += sizeAdd
- folder.UsedQuotaFiles += filesAdd
- }
- folder.LastQuotaUpdate = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(folder)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(folder.Name), buf)
- })
- }
- func (p *BoltProvider) getUsedFolderQuota(name string) (int, int64, error) {
- folder, err := p.getFolderByName(name)
- if err != nil {
- providerLog(logger.LevelError, "unable to get quota for folder %q error: %v", name, err)
- return 0, 0, err
- }
- return folder.UsedQuotaFiles, folder.UsedQuotaSize, err
- }
- func (p *BoltProvider) getGroups(limit, offset int, order string, _ bool) ([]Group, error) {
- groups := make([]Group, 0, limit)
- var err error
- if limit <= 0 {
- return groups, err
- }
- err = p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var group Group
- group, err = p.joinGroupAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- group.PrepareForRendering()
- groups = append(groups, group)
- if len(groups) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var group Group
- group, err = p.joinGroupAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- group.PrepareForRendering()
- groups = append(groups, group)
- if len(groups) >= limit {
- break
- }
- }
- }
- return err
- })
- return groups, err
- }
- func (p *BoltProvider) getGroupsWithNames(names []string) ([]Group, error) {
- var groups []Group
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- for _, name := range names {
- g := bucket.Get([]byte(name))
- if g == nil {
- continue
- }
- group, err := p.joinGroupAndFolders(g, foldersBucket)
- if err != nil {
- return err
- }
- groups = append(groups, group)
- }
- return nil
- })
- return groups, err
- }
- func (p *BoltProvider) getUsersInGroups(names []string) ([]string, error) {
- var usernames []string
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- for _, name := range names {
- g := bucket.Get([]byte(name))
- if g == nil {
- continue
- }
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return err
- }
- usernames = append(usernames, group.Users...)
- }
- return nil
- })
- return usernames, err
- }
- func (p *BoltProvider) groupExists(name string) (Group, error) {
- var group Group
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- g := bucket.Get([]byte(name))
- if g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", name))
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- group, err = p.joinGroupAndFolders(g, foldersBucket)
- return err
- })
- return group, err
- }
- func (p *BoltProvider) addGroup(group *Group) error {
- if err := group.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- if u := bucket.Get([]byte(group.Name)); u != nil {
- return fmt.Errorf("group %v already exists", group.Name)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- group.ID = int64(id)
- group.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- group.Users = nil
- group.Admins = nil
- sort.Slice(group.VirtualFolders, func(i, j int) bool {
- return group.VirtualFolders[i].Name < group.VirtualFolders[j].Name
- })
- for idx := range group.VirtualFolders {
- err = p.addRelationToFolderMapping(group.VirtualFolders[idx].Name, nil, group, foldersBucket)
- if err != nil {
- return err
- }
- }
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- })
- }
- func (p *BoltProvider) updateGroup(group *Group) error {
- if err := group.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- var g []byte
- if g = bucket.Get([]byte(group.Name)); g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", group.Name))
- }
- var oldGroup Group
- err = json.Unmarshal(g, &oldGroup)
- if err != nil {
- return err
- }
- for idx := range oldGroup.VirtualFolders {
- err = p.removeRelationFromFolderMapping(oldGroup.VirtualFolders[idx], "", oldGroup.Name, foldersBucket)
- if err != nil {
- return err
- }
- }
- sort.Slice(group.VirtualFolders, func(i, j int) bool {
- return group.VirtualFolders[i].Name < group.VirtualFolders[j].Name
- })
- for idx := range group.VirtualFolders {
- err = p.addRelationToFolderMapping(group.VirtualFolders[idx].Name, nil, group, foldersBucket)
- if err != nil {
- return err
- }
- }
- group.ID = oldGroup.ID
- group.CreatedAt = oldGroup.CreatedAt
- group.Users = oldGroup.Users
- group.Admins = oldGroup.Admins
- group.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- })
- }
- func (p *BoltProvider) deleteGroup(group Group) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- var g []byte
- if g = bucket.Get([]byte(group.Name)); g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", group.Name))
- }
- var oldGroup Group
- err = json.Unmarshal(g, &oldGroup)
- if err != nil {
- return err
- }
- if len(oldGroup.Users) > 0 {
- return util.NewValidationError(fmt.Sprintf("the group %q is referenced, it cannot be removed", oldGroup.Name))
- }
- if len(oldGroup.VirtualFolders) > 0 {
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- for idx := range oldGroup.VirtualFolders {
- err = p.removeRelationFromFolderMapping(oldGroup.VirtualFolders[idx], "", oldGroup.Name, foldersBucket)
- if err != nil {
- return err
- }
- }
- }
- if len(oldGroup.Admins) > 0 {
- adminsBucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- for idx := range oldGroup.Admins {
- err = p.removeGroupFromAdminMapping(oldGroup.Name, oldGroup.Admins[idx], adminsBucket)
- if err != nil {
- return err
- }
- }
- }
- return bucket.Delete([]byte(group.Name))
- })
- }
- func (p *BoltProvider) dumpGroups() ([]Group, error) {
- groups := make([]Group, 0, 50)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- group, err := p.joinGroupAndFolders(v, foldersBucket)
- if err != nil {
- return err
- }
- groups = append(groups, group)
- }
- return err
- })
- return groups, err
- }
- func (p *BoltProvider) apiKeyExists(keyID string) (APIKey, error) {
- var apiKey APIKey
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- k := bucket.Get([]byte(keyID))
- if k == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", keyID))
- }
- return json.Unmarshal(k, &apiKey)
- })
- return apiKey, err
- }
- func (p *BoltProvider) addAPIKey(apiKey *APIKey) error {
- err := apiKey.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- if a := bucket.Get([]byte(apiKey.KeyID)); a != nil {
- return fmt.Errorf("API key %v already exists", apiKey.KeyID)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- apiKey.ID = int64(id)
- apiKey.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- apiKey.LastUseAt = 0
- if apiKey.User != "" {
- if err := p.userExistsInternal(tx, apiKey.User); err != nil {
- return util.NewValidationError(fmt.Sprintf("related user %q does not exists", apiKey.User))
- }
- }
- if apiKey.Admin != "" {
- if err := p.adminExistsInternal(tx, apiKey.Admin); err != nil {
- return util.NewValidationError(fmt.Sprintf("related admin %q does not exists", apiKey.User))
- }
- }
- buf, err := json.Marshal(apiKey)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(apiKey.KeyID), buf)
- })
- }
- func (p *BoltProvider) updateAPIKey(apiKey *APIKey) error {
- err := apiKey.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(apiKey.KeyID)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", apiKey.KeyID))
- }
- var oldAPIKey APIKey
- err = json.Unmarshal(a, &oldAPIKey)
- if err != nil {
- return err
- }
- apiKey.ID = oldAPIKey.ID
- apiKey.KeyID = oldAPIKey.KeyID
- apiKey.Key = oldAPIKey.Key
- apiKey.CreatedAt = oldAPIKey.CreatedAt
- apiKey.LastUseAt = oldAPIKey.LastUseAt
- apiKey.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- if apiKey.User != "" {
- if err := p.userExistsInternal(tx, apiKey.User); err != nil {
- return util.NewValidationError(fmt.Sprintf("related user %q does not exists", apiKey.User))
- }
- }
- if apiKey.Admin != "" {
- if err := p.adminExistsInternal(tx, apiKey.Admin); err != nil {
- return util.NewValidationError(fmt.Sprintf("related admin %q does not exists", apiKey.User))
- }
- }
- buf, err := json.Marshal(apiKey)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(apiKey.KeyID), buf)
- })
- }
- func (p *BoltProvider) deleteAPIKey(apiKey APIKey) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- if bucket.Get([]byte(apiKey.KeyID)) == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("API key %v does not exist", apiKey.KeyID))
- }
- return bucket.Delete([]byte(apiKey.KeyID))
- })
- }
- func (p *BoltProvider) getAPIKeys(limit int, offset int, order string) ([]APIKey, error) {
- apiKeys := make([]APIKey, 0, limit)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var apiKey APIKey
- err = json.Unmarshal(v, &apiKey)
- if err != nil {
- return err
- }
- apiKey.HideConfidentialData()
- apiKeys = append(apiKeys, apiKey)
- if len(apiKeys) >= limit {
- break
- }
- }
- return nil
- }
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var apiKey APIKey
- err = json.Unmarshal(v, &apiKey)
- if err != nil {
- return err
- }
- apiKey.HideConfidentialData()
- apiKeys = append(apiKeys, apiKey)
- if len(apiKeys) >= limit {
- break
- }
- }
- return nil
- })
- return apiKeys, err
- }
- func (p *BoltProvider) dumpAPIKeys() ([]APIKey, error) {
- apiKeys := make([]APIKey, 0, 30)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var apiKey APIKey
- err = json.Unmarshal(v, &apiKey)
- if err != nil {
- return err
- }
- apiKeys = append(apiKeys, apiKey)
- }
- return err
- })
- return apiKeys, err
- }
- func (p *BoltProvider) shareExists(shareID, username string) (Share, error) {
- var share Share
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- s := bucket.Get([]byte(shareID))
- if s == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", shareID))
- }
- if err := json.Unmarshal(s, &share); err != nil {
- return err
- }
- if username != "" && share.Username != username {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", shareID))
- }
- return nil
- })
- return share, err
- }
- func (p *BoltProvider) addShare(share *Share) error {
- err := share.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- if a := bucket.Get([]byte(share.ShareID)); a != nil {
- return fmt.Errorf("share %v already exists", share.ShareID)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- share.ID = int64(id)
- if !share.IsRestore {
- share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- share.UpdatedAt = share.CreatedAt
- share.LastUseAt = 0
- share.UsedTokens = 0
- }
- if share.CreatedAt == 0 {
- share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- }
- if share.UpdatedAt == 0 {
- share.UpdatedAt = share.CreatedAt
- }
- if err := p.userExistsInternal(tx, share.Username); err != nil {
- return util.NewValidationError(fmt.Sprintf("related user %q does not exists", share.Username))
- }
- buf, err := json.Marshal(share)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(share.ShareID), buf)
- })
- }
- func (p *BoltProvider) updateShare(share *Share) error {
- if err := share.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- var s []byte
- if s = bucket.Get([]byte(share.ShareID)); s == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
- }
- var oldObject Share
- if err = json.Unmarshal(s, &oldObject); err != nil {
- return err
- }
- if oldObject.Username != share.Username {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
- }
- share.ID = oldObject.ID
- share.ShareID = oldObject.ShareID
- if !share.IsRestore {
- share.UsedTokens = oldObject.UsedTokens
- share.CreatedAt = oldObject.CreatedAt
- share.LastUseAt = oldObject.LastUseAt
- share.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- }
- if share.CreatedAt == 0 {
- share.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- }
- if share.UpdatedAt == 0 {
- share.UpdatedAt = share.CreatedAt
- }
- if err := p.userExistsInternal(tx, share.Username); err != nil {
- return util.NewValidationError(fmt.Sprintf("related user %q does not exists", share.Username))
- }
- buf, err := json.Marshal(share)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(share.ShareID), buf)
- })
- }
- func (p *BoltProvider) deleteShare(share Share) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- var s []byte
- if s = bucket.Get([]byte(share.ShareID)); s == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
- }
- var oldObject Share
- if err = json.Unmarshal(s, &oldObject); err != nil {
- return err
- }
- if oldObject.Username != share.Username {
- return util.NewRecordNotFoundError(fmt.Sprintf("Share %v does not exist", share.ShareID))
- }
- return bucket.Delete([]byte(share.ShareID))
- })
- }
- func (p *BoltProvider) getShares(limit int, offset int, order, username string) ([]Share, error) {
- shares := make([]Share, 0, limit)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var share Share
- if err := json.Unmarshal(v, &share); err != nil {
- return err
- }
- if share.Username != username {
- continue
- }
- itNum++
- if itNum <= offset {
- continue
- }
- share.HideConfidentialData()
- shares = append(shares, share)
- if len(shares) >= limit {
- break
- }
- }
- return nil
- }
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- var share Share
- err = json.Unmarshal(v, &share)
- if err != nil {
- return err
- }
- if share.Username != username {
- continue
- }
- itNum++
- if itNum <= offset {
- continue
- }
- share.HideConfidentialData()
- shares = append(shares, share)
- if len(shares) >= limit {
- break
- }
- }
- return nil
- })
- return shares, err
- }
- func (p *BoltProvider) dumpShares() ([]Share, error) {
- shares := make([]Share, 0, 30)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var share Share
- err = json.Unmarshal(v, &share)
- if err != nil {
- return err
- }
- shares = append(shares, share)
- }
- return err
- })
- return shares, err
- }
- func (p *BoltProvider) updateShareLastUse(shareID string, numTokens int) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(shareID)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("share %q does not exist, unable to update last use", shareID))
- }
- var share Share
- err = json.Unmarshal(u, &share)
- if err != nil {
- return err
- }
- share.LastUseAt = util.GetTimeAsMsSinceEpoch(time.Now())
- share.UsedTokens += numTokens
- buf, err := json.Marshal(share)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(shareID), buf)
- if err != nil {
- providerLog(logger.LevelWarn, "error updating last use for share %q: %v", shareID, err)
- return err
- }
- providerLog(logger.LevelDebug, "last use updated for share %q", shareID)
- return nil
- })
- }
- func (p *BoltProvider) getDefenderHosts(_ int64, _ int) ([]DefenderEntry, error) {
- return nil, ErrNotImplemented
- }
- func (p *BoltProvider) getDefenderHostByIP(_ string, _ int64) (DefenderEntry, error) {
- return DefenderEntry{}, ErrNotImplemented
- }
- func (p *BoltProvider) isDefenderHostBanned(_ string) (DefenderEntry, error) {
- return DefenderEntry{}, ErrNotImplemented
- }
- func (p *BoltProvider) updateDefenderBanTime(_ string, _ int) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) deleteDefenderHost(_ string) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) addDefenderEvent(_ string, _ int) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) setDefenderBanTime(_ string, _ int64) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) cleanupDefender(_ int64) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) addActiveTransfer(_ ActiveTransfer) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) updateActiveTransferSizes(_, _, _ int64, _ string) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) removeActiveTransfer(_ int64, _ string) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) cleanupActiveTransfers(_ time.Time) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) getActiveTransfers(_ time.Time) ([]ActiveTransfer, error) {
- return nil, ErrNotImplemented
- }
- func (p *BoltProvider) addSharedSession(_ Session) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) deleteSharedSession(_ string) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) getSharedSession(_ string) (Session, error) {
- return Session{}, ErrNotImplemented
- }
- func (p *BoltProvider) cleanupSharedSessions(_ SessionType, _ int64) error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) getEventActions(limit, offset int, order string, _ bool) ([]BaseEventAction, error) {
- if limit <= 0 {
- return nil, nil
- }
- actions := make([]BaseEventAction, 0, limit)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- itNum := 0
- cursor := bucket.Cursor()
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var action BaseEventAction
- err = json.Unmarshal(v, &action)
- if err != nil {
- return err
- }
- action.PrepareForRendering()
- actions = append(actions, action)
- if len(actions) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var action BaseEventAction
- err = json.Unmarshal(v, &action)
- if err != nil {
- return err
- }
- action.PrepareForRendering()
- actions = append(actions, action)
- if len(actions) >= limit {
- break
- }
- }
- }
- return nil
- })
- return actions, err
- }
- func (p *BoltProvider) dumpEventActions() ([]BaseEventAction, error) {
- actions := make([]BaseEventAction, 0, 50)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var action BaseEventAction
- err = json.Unmarshal(v, &action)
- if err != nil {
- return err
- }
- actions = append(actions, action)
- }
- return nil
- })
- return actions, err
- }
- func (p *BoltProvider) eventActionExists(name string) (BaseEventAction, error) {
- var action BaseEventAction
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- k := bucket.Get([]byte(name))
- if k == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("action %q does not exist", name))
- }
- return json.Unmarshal(k, &action)
- })
- return action, err
- }
- func (p *BoltProvider) addEventAction(action *BaseEventAction) error {
- err := action.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- if a := bucket.Get([]byte(action.Name)); a != nil {
- return fmt.Errorf("event action %s already exists", action.Name)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- action.ID = int64(id)
- action.Rules = nil
- buf, err := json.Marshal(action)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(action.Name), buf)
- })
- }
- func (p *BoltProvider) updateEventAction(action *BaseEventAction) error {
- err := action.validate()
- if err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(action.Name)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("event action %s does not exist", action.Name))
- }
- var oldAction BaseEventAction
- err = json.Unmarshal(a, &oldAction)
- if err != nil {
- return err
- }
- action.ID = oldAction.ID
- action.Name = oldAction.Name
- action.Rules = nil
- if len(oldAction.Rules) > 0 {
- rulesBucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- var relatedRules []string
- for _, ruleName := range oldAction.Rules {
- r := rulesBucket.Get([]byte(ruleName))
- if r != nil {
- relatedRules = append(relatedRules, ruleName)
- var rule EventRule
- err := json.Unmarshal(r, &rule)
- if err != nil {
- return err
- }
- rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(rule)
- if err != nil {
- return err
- }
- if err = rulesBucket.Put([]byte(rule.Name), buf); err != nil {
- return err
- }
- setLastRuleUpdate()
- }
- }
- action.Rules = relatedRules
- }
- buf, err := json.Marshal(action)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(action.Name), buf)
- })
- }
- func (p *BoltProvider) deleteEventAction(action BaseEventAction) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- var a []byte
- if a = bucket.Get([]byte(action.Name)); a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("action %s does not exist", action.Name))
- }
- var oldAction BaseEventAction
- err = json.Unmarshal(a, &oldAction)
- if err != nil {
- return err
- }
- if len(oldAction.Rules) > 0 {
- return util.NewValidationError(fmt.Sprintf("action %s is referenced, it cannot be removed", oldAction.Name))
- }
- return bucket.Delete([]byte(action.Name))
- })
- }
- func (p *BoltProvider) getEventRules(limit, offset int, order string) ([]EventRule, error) {
- if limit <= 0 {
- return nil, nil
- }
- rules := make([]EventRule, 0, limit)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- itNum := 0
- cursor := bucket.Cursor()
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var rule EventRule
- rule, err = p.joinRuleAndActions(v, actionsBucket)
- if err != nil {
- return err
- }
- rule.PrepareForRendering()
- rules = append(rules, rule)
- if len(rules) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var rule EventRule
- rule, err = p.joinRuleAndActions(v, actionsBucket)
- if err != nil {
- return err
- }
- rule.PrepareForRendering()
- rules = append(rules, rule)
- if len(rules) >= limit {
- break
- }
- }
- }
- return err
- })
- return rules, err
- }
- func (p *BoltProvider) dumpEventRules() ([]EventRule, error) {
- rules := make([]EventRule, 0, 50)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- rule, err := p.joinRuleAndActions(v, actionsBucket)
- if err != nil {
- return err
- }
- rules = append(rules, rule)
- }
- return nil
- })
- return rules, err
- }
- func (p *BoltProvider) getRecentlyUpdatedRules(after int64) ([]EventRule, error) {
- if getLastRuleUpdate() < after {
- return nil, nil
- }
- rules := make([]EventRule, 0, 10)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var rule EventRule
- err := json.Unmarshal(v, &rule)
- if err != nil {
- return err
- }
- if rule.UpdatedAt < after {
- continue
- }
- var actions []EventAction
- for idx := range rule.Actions {
- action := &rule.Actions[idx]
- var baseAction BaseEventAction
- k := actionsBucket.Get([]byte(action.Name))
- if k == nil {
- continue
- }
- err = json.Unmarshal(k, &baseAction)
- if err != nil {
- continue
- }
- baseAction.Options.SetEmptySecretsIfNil()
- action.BaseEventAction = baseAction
- actions = append(actions, *action)
- }
- rule.Actions = actions
- rules = append(rules, rule)
- }
- return nil
- })
- return rules, err
- }
- func (p *BoltProvider) eventRuleExists(name string) (EventRule, error) {
- var rule EventRule
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- r := bucket.Get([]byte(name))
- if r == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", name))
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- rule, err = p.joinRuleAndActions(r, actionsBucket)
- return err
- })
- return rule, err
- }
- func (p *BoltProvider) addEventRule(rule *EventRule) error {
- if err := rule.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- if r := bucket.Get([]byte(rule.Name)); r != nil {
- return fmt.Errorf("event rule %q already exists", rule.Name)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- rule.ID = int64(id)
- rule.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- rule.UpdatedAt = rule.CreatedAt
- for idx := range rule.Actions {
- if err = p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name, actionsBucket); err != nil {
- return err
- }
- }
- sort.Slice(rule.Actions, func(i, j int) bool {
- return rule.Actions[i].Order < rule.Actions[j].Order
- })
- buf, err := json.Marshal(rule)
- if err != nil {
- return err
- }
- err = bucket.Put([]byte(rule.Name), buf)
- if err == nil {
- setLastRuleUpdate()
- }
- return err
- })
- }
- func (p *BoltProvider) updateEventRule(rule *EventRule) error {
- if err := rule.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- var r []byte
- if r = bucket.Get([]byte(rule.Name)); r == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", rule.Name))
- }
- var oldRule EventRule
- if err = json.Unmarshal(r, &oldRule); err != nil {
- return err
- }
- for idx := range oldRule.Actions {
- if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name, actionsBucket); err != nil {
- return err
- }
- }
- for idx := range rule.Actions {
- if err = p.addRuleToActionMapping(rule.Name, rule.Actions[idx].Name, actionsBucket); err != nil {
- return err
- }
- }
- rule.ID = oldRule.ID
- rule.CreatedAt = oldRule.CreatedAt
- rule.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(rule)
- if err != nil {
- return err
- }
- sort.Slice(rule.Actions, func(i, j int) bool {
- return rule.Actions[i].Order < rule.Actions[j].Order
- })
- err = bucket.Put([]byte(rule.Name), buf)
- if err == nil {
- setLastRuleUpdate()
- }
- return err
- })
- }
- func (p *BoltProvider) deleteEventRule(rule EventRule, _ bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRulesBucket(tx)
- if err != nil {
- return err
- }
- var r []byte
- if r = bucket.Get([]byte(rule.Name)); r == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("event rule %q does not exist", rule.Name))
- }
- var oldRule EventRule
- if err = json.Unmarshal(r, &oldRule); err != nil {
- return err
- }
- if len(oldRule.Actions) > 0 {
- actionsBucket, err := p.getActionsBucket(tx)
- if err != nil {
- return err
- }
- for idx := range oldRule.Actions {
- if err = p.removeRuleFromActionMapping(rule.Name, oldRule.Actions[idx].Name, actionsBucket); err != nil {
- return err
- }
- }
- }
- return bucket.Delete([]byte(rule.Name))
- })
- }
- func (*BoltProvider) getTaskByName(_ string) (Task, error) {
- return Task{}, ErrNotImplemented
- }
- func (*BoltProvider) addTask(_ string) error {
- return ErrNotImplemented
- }
- func (*BoltProvider) updateTask(_ string, _ int64) error {
- return ErrNotImplemented
- }
- func (*BoltProvider) updateTaskTimestamp(_ string) error {
- return ErrNotImplemented
- }
- func (*BoltProvider) addNode() error {
- return ErrNotImplemented
- }
- func (*BoltProvider) getNodeByName(_ string) (Node, error) {
- return Node{}, ErrNotImplemented
- }
- func (*BoltProvider) getNodes() ([]Node, error) {
- return nil, ErrNotImplemented
- }
- func (*BoltProvider) updateNodeTimestamp() error {
- return ErrNotImplemented
- }
- func (*BoltProvider) cleanupNodes() error {
- return ErrNotImplemented
- }
- func (p *BoltProvider) roleExists(name string) (Role, error) {
- var role Role
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- r := bucket.Get([]byte(name))
- if r == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("role %q does not exist", name))
- }
- return json.Unmarshal(r, &role)
- })
- return role, err
- }
- func (p *BoltProvider) addRole(role *Role) error {
- if err := role.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- if r := bucket.Get([]byte(role.Name)); r != nil {
- return fmt.Errorf("role %q already exists", role.Name)
- }
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- role.ID = int64(id)
- role.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- role.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- role.Users = nil
- role.Admins = nil
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- })
- }
- func (p *BoltProvider) updateRole(role *Role) error {
- if err := role.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- var r []byte
- if r = bucket.Get([]byte(role.Name)); r == nil {
- return fmt.Errorf("role %q does not exist", role.Name)
- }
- var oldRole Role
- err = json.Unmarshal(r, &oldRole)
- if err != nil {
- return err
- }
- role.ID = oldRole.ID
- role.CreatedAt = oldRole.CreatedAt
- role.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- role.Users = oldRole.Users
- role.Admins = oldRole.Admins
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- })
- }
- func (p *BoltProvider) deleteRole(role Role) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- var r []byte
- if r = bucket.Get([]byte(role.Name)); r == nil {
- return fmt.Errorf("role %q does not exist", role.Name)
- }
- var oldRole Role
- err = json.Unmarshal(r, &oldRole)
- if err != nil {
- return err
- }
- if len(oldRole.Admins) > 0 {
- return util.NewValidationError(fmt.Sprintf("the role %q is referenced, it cannot be removed", oldRole.Name))
- }
- if len(oldRole.Users) > 0 {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- for _, username := range oldRole.Users {
- if err := p.removeRoleFromUser(username, oldRole.Name, bucket); err != nil {
- return err
- }
- }
- }
- return bucket.Delete([]byte(role.Name))
- })
- }
- func (p *BoltProvider) getRoles(limit int, offset int, order string, _ bool) ([]Role, error) {
- roles := make([]Role, 0, limit)
- if limit <= 0 {
- return roles, nil
- }
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- itNum := 0
- if order == OrderASC {
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- itNum++
- if itNum <= offset {
- continue
- }
- var role Role
- err = json.Unmarshal(v, &role)
- if err != nil {
- return err
- }
- roles = append(roles, role)
- if len(roles) >= limit {
- break
- }
- }
- } else {
- for k, v := cursor.Last(); k != nil; k, v = cursor.Prev() {
- itNum++
- if itNum <= offset {
- continue
- }
- var role Role
- err = json.Unmarshal(v, &role)
- if err != nil {
- return err
- }
- roles = append(roles, role)
- if len(roles) >= limit {
- break
- }
- }
- }
- return nil
- })
- return roles, err
- }
- func (p *BoltProvider) dumpRoles() ([]Role, error) {
- roles := make([]Role, 0, 10)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var role Role
- err = json.Unmarshal(v, &role)
- if err != nil {
- return err
- }
- roles = append(roles, role)
- }
- return err
- })
- return roles, err
- }
- func (p *BoltProvider) ipListEntryExists(ipOrNet string, listType IPListType) (IPListEntry, error) {
- entry := IPListEntry{
- IPOrNet: ipOrNet,
- Type: listType,
- }
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- e := bucket.Get([]byte(entry.getKey()))
- if e == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("entry %q does not exist", entry.IPOrNet))
- }
- err = json.Unmarshal(e, &entry)
- if err == nil {
- entry.PrepareForRendering()
- }
- return err
- })
- return entry, err
- }
- func (p *BoltProvider) addIPListEntry(entry *IPListEntry) error {
- if err := entry.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- if e := bucket.Get([]byte(entry.getKey())); e != nil {
- return fmt.Errorf("entry %q already exists", entry.IPOrNet)
- }
- entry.CreatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- entry.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(entry)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(entry.getKey()), buf)
- })
- }
- func (p *BoltProvider) updateIPListEntry(entry *IPListEntry) error {
- if err := entry.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- var e []byte
- if e = bucket.Get([]byte(entry.getKey())); e == nil {
- return fmt.Errorf("entry %q does not exist", entry.IPOrNet)
- }
- var oldEntry IPListEntry
- err = json.Unmarshal(e, &oldEntry)
- if err != nil {
- return err
- }
- entry.CreatedAt = oldEntry.CreatedAt
- entry.UpdatedAt = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(entry)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(entry.getKey()), buf)
- })
- }
- func (p *BoltProvider) deleteIPListEntry(entry IPListEntry, _ bool) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- if e := bucket.Get([]byte(entry.getKey())); e == nil {
- return fmt.Errorf("entry %q does not exist", entry.IPOrNet)
- }
- return bucket.Delete([]byte(entry.getKey()))
- })
- }
- func (p *BoltProvider) getIPListEntries(listType IPListType, filter, from, order string, limit int) ([]IPListEntry, error) {
- entries := make([]IPListEntry, 0, 15)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- prefix := []byte(fmt.Sprintf("%d_", listType))
- acceptKey := func(k []byte) bool {
- return k != nil && bytes.HasPrefix(k, prefix)
- }
- cursor := bucket.Cursor()
- if order == OrderASC {
- for k, v := cursor.Seek(prefix); acceptKey(k); k, v = cursor.Next() {
- var entry IPListEntry
- err = json.Unmarshal(v, &entry)
- if err != nil {
- return err
- }
- if entry.satisfySearchConstraints(filter, from, order) {
- entry.PrepareForRendering()
- entries = append(entries, entry)
- if limit > 0 && len(entries) >= limit {
- break
- }
- }
- }
- } else {
- for k, v := cursor.Last(); acceptKey(k); k, v = cursor.Prev() {
- var entry IPListEntry
- err = json.Unmarshal(v, &entry)
- if err != nil {
- return err
- }
- if entry.satisfySearchConstraints(filter, from, order) {
- entry.PrepareForRendering()
- entries = append(entries, entry)
- if limit > 0 && len(entries) >= limit {
- break
- }
- }
- }
- }
- return nil
- })
- return entries, err
- }
- func (p *BoltProvider) getRecentlyUpdatedIPListEntries(_ int64) ([]IPListEntry, error) {
- return nil, ErrNotImplemented
- }
- func (p *BoltProvider) dumpIPListEntries() ([]IPListEntry, error) {
- entries := make([]IPListEntry, 0, 10)
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- if count := bucket.Stats().KeyN; count > ipListMemoryLimit {
- providerLog(logger.LevelInfo, "IP lists excluded from dump, too many entries: %d", count)
- return nil
- }
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var entry IPListEntry
- err = json.Unmarshal(v, &entry)
- if err != nil {
- return err
- }
- entry.PrepareForRendering()
- entries = append(entries, entry)
- }
- return nil
- })
- return entries, err
- }
- func (p *BoltProvider) countIPListEntries(listType IPListType) (int64, error) {
- var count int64
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- if listType == 0 {
- count = int64(bucket.Stats().KeyN)
- return nil
- }
- prefix := []byte(fmt.Sprintf("%d_", listType))
- cursor := bucket.Cursor()
- for k, _ := cursor.Seek(prefix); k != nil && bytes.HasPrefix(k, prefix); k, _ = cursor.Next() {
- count++
- }
- return nil
- })
- return count, err
- }
- func (p *BoltProvider) getListEntriesForIP(ip string, listType IPListType) ([]IPListEntry, error) {
- entries := make([]IPListEntry, 0, 3)
- ipAddr, err := netip.ParseAddr(ip)
- if err != nil {
- return entries, fmt.Errorf("invalid ip address %s", ip)
- }
- var netType int
- var ipBytes []byte
- if ipAddr.Is4() || ipAddr.Is4In6() {
- netType = ipTypeV4
- as4 := ipAddr.As4()
- ipBytes = as4[:]
- } else {
- netType = ipTypeV6
- as16 := ipAddr.As16()
- ipBytes = as16[:]
- }
- err = p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket, err := p.getIPListsBucket(tx)
- if err != nil {
- return err
- }
- prefix := []byte(fmt.Sprintf("%d_", listType))
- cursor := bucket.Cursor()
- for k, v := cursor.Seek(prefix); k != nil && bytes.HasPrefix(k, prefix); k, v = cursor.Next() {
- var entry IPListEntry
- err = json.Unmarshal(v, &entry)
- if err != nil {
- return err
- }
- if entry.IPType == netType && bytes.Compare(ipBytes, entry.First) >= 0 && bytes.Compare(ipBytes, entry.Last) <= 0 {
- entry.PrepareForRendering()
- entries = append(entries, entry)
- }
- }
- return nil
- })
- return entries, err
- }
- func (p *BoltProvider) getConfigs() (Configs, error) {
- var configs Configs
- err := p.dbHandle.View(func(tx *bolt.Tx) error {
- bucket := tx.Bucket(configsBucket)
- if bucket == nil {
- return fmt.Errorf("unable to find configs bucket")
- }
- data := bucket.Get(configsKey)
- if data != nil {
- return json.Unmarshal(data, &configs)
- }
- return nil
- })
- return configs, err
- }
- func (p *BoltProvider) setConfigs(configs *Configs) error {
- if err := configs.validate(); err != nil {
- return err
- }
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket := tx.Bucket(configsBucket)
- if bucket == nil {
- return fmt.Errorf("unable to find configs bucket")
- }
- buf, err := json.Marshal(configs)
- if err != nil {
- return err
- }
- return bucket.Put(configsKey, buf)
- })
- }
- func (p *BoltProvider) setFirstDownloadTimestamp(username string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to set download timestamp",
- username))
- }
- var user User
- err = json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- if user.FirstDownload > 0 {
- return util.NewGenericError(fmt.Sprintf("first download already set to %v",
- util.GetTimeFromMsecSinceEpoch(user.FirstDownload)))
- }
- user.FirstDownload = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(username), buf)
- })
- }
- func (p *BoltProvider) setFirstUploadTimestamp(username string) error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- var u []byte
- if u = bucket.Get([]byte(username)); u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist, unable to set upload timestamp",
- username))
- }
- var user User
- if err = json.Unmarshal(u, &user); err != nil {
- return err
- }
- if user.FirstUpload > 0 {
- return util.NewGenericError(fmt.Sprintf("first upload already set to %v",
- util.GetTimeFromMsecSinceEpoch(user.FirstUpload)))
- }
- user.FirstUpload = util.GetTimeAsMsSinceEpoch(time.Now())
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(username), buf)
- })
- }
- func (p *BoltProvider) close() error {
- return p.dbHandle.Close()
- }
- func (p *BoltProvider) reloadConfig() error {
- return nil
- }
- // initializeDatabase does nothing, no initilization is needed for bolt provider
- func (p *BoltProvider) initializeDatabase() error {
- return ErrNoInitRequired
- }
- func (p *BoltProvider) migrateDatabase() error {
- dbVersion, err := getBoltDatabaseVersion(p.dbHandle)
- if err != nil {
- return err
- }
- switch version := dbVersion.Version; {
- case version == boltDatabaseVersion:
- providerLog(logger.LevelDebug, "bolt database is up to date, current version: %d", version)
- return ErrNoInitRequired
- case version < 28:
- err = fmt.Errorf("database schema version %d is too old, please see the upgrading docs", version)
- providerLog(logger.LevelError, "%v", err)
- logger.ErrorToConsole("%v", err)
- return err
- default:
- if version > boltDatabaseVersion {
- providerLog(logger.LevelError, "database schema version %d is newer than the supported one: %d", version,
- boltDatabaseVersion)
- logger.WarnToConsole("database schema version %d is newer than the supported one: %d", version,
- boltDatabaseVersion)
- return nil
- }
- return fmt.Errorf("database schema version not handled: %d", version)
- }
- }
- func (p *BoltProvider) revertDatabase(targetVersion int) error { //nolint:gocyclo
- dbVersion, err := getBoltDatabaseVersion(p.dbHandle)
- if err != nil {
- return err
- }
- if dbVersion.Version == targetVersion {
- return errors.New("current version match target version, nothing to do")
- }
- switch dbVersion.Version {
- default:
- return fmt.Errorf("database schema version not handled: %v", dbVersion.Version)
- }
- }
- func (p *BoltProvider) resetDatabase() error {
- return p.dbHandle.Update(func(tx *bolt.Tx) error {
- for _, bucketName := range boltBuckets {
- err := tx.DeleteBucket(bucketName)
- if err != nil && !errors.Is(err, bolt.ErrBucketNotFound) {
- return fmt.Errorf("unable to remove bucket %v: %w", bucketName, err)
- }
- }
- return nil
- })
- }
- func (p *BoltProvider) joinRuleAndActions(r []byte, actionsBucket *bolt.Bucket) (EventRule, error) {
- var rule EventRule
- err := json.Unmarshal(r, &rule)
- if err != nil {
- return rule, err
- }
- var actions []EventAction
- for idx := range rule.Actions {
- action := &rule.Actions[idx]
- var baseAction BaseEventAction
- k := actionsBucket.Get([]byte(action.Name))
- if k == nil {
- continue
- }
- err = json.Unmarshal(k, &baseAction)
- if err != nil {
- continue
- }
- baseAction.Options.SetEmptySecretsIfNil()
- action.BaseEventAction = baseAction
- actions = append(actions, *action)
- }
- rule.Actions = actions
- return rule, nil
- }
- func (p *BoltProvider) joinGroupAndFolders(g []byte, foldersBucket *bolt.Bucket) (Group, error) {
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return group, err
- }
- if len(group.VirtualFolders) > 0 {
- var folders []vfs.VirtualFolder
- for idx := range group.VirtualFolders {
- folder := &group.VirtualFolders[idx]
- baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
- if err != nil {
- continue
- }
- folder.BaseVirtualFolder = baseFolder
- folders = append(folders, *folder)
- }
- group.VirtualFolders = folders
- }
- group.SetEmptySecretsIfNil()
- return group, err
- }
- func (p *BoltProvider) joinUserAndFolders(u []byte, foldersBucket *bolt.Bucket) (User, error) {
- var user User
- err := json.Unmarshal(u, &user)
- if err != nil {
- return user, err
- }
- if len(user.VirtualFolders) > 0 {
- var folders []vfs.VirtualFolder
- for idx := range user.VirtualFolders {
- folder := &user.VirtualFolders[idx]
- baseFolder, err := p.folderExistsInternal(folder.Name, foldersBucket)
- if err != nil {
- continue
- }
- folder.BaseVirtualFolder = baseFolder
- folders = append(folders, *folder)
- }
- user.VirtualFolders = folders
- }
- user.SetEmptySecretsIfNil()
- return user, err
- }
- func (p *BoltProvider) groupExistsInternal(name string, bucket *bolt.Bucket) (Group, error) {
- var group Group
- g := bucket.Get([]byte(name))
- if g == nil {
- err := util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", name))
- return group, err
- }
- err := json.Unmarshal(g, &group)
- return group, err
- }
- func (p *BoltProvider) folderExistsInternal(name string, bucket *bolt.Bucket) (vfs.BaseVirtualFolder, error) {
- var folder vfs.BaseVirtualFolder
- f := bucket.Get([]byte(name))
- if f == nil {
- err := util.NewRecordNotFoundError(fmt.Sprintf("folder %q does not exist", name))
- return folder, err
- }
- err := json.Unmarshal(f, &folder)
- return folder, err
- }
- func (p *BoltProvider) addFolderInternal(folder vfs.BaseVirtualFolder, bucket *bolt.Bucket) error {
- id, err := bucket.NextSequence()
- if err != nil {
- return err
- }
- folder.ID = int64(id)
- buf, err := json.Marshal(folder)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(folder.Name), buf)
- }
- func (p *BoltProvider) removeRoleFromUser(username, role string, bucket *bolt.Bucket) error {
- u := bucket.Get([]byte(username))
- if u == nil {
- providerLog(logger.LevelWarn, "user %q does not exist, cannot remove role %q", username, role)
- return nil
- }
- var user User
- err := json.Unmarshal(u, &user)
- if err != nil {
- return err
- }
- if user.Role == role {
- user.Role = ""
- buf, err := json.Marshal(user)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(user.Username), buf)
- }
- providerLog(logger.LevelError, "user %q does not have the expected role %q, actual %q", username, role, user.Role)
- return nil
- }
- func (p *BoltProvider) addAdminToRole(username, roleName string, bucket *bolt.Bucket) error {
- if roleName == "" {
- return nil
- }
- r := bucket.Get([]byte(roleName))
- if r == nil {
- return util.NewGenericError(fmt.Sprintf("role %q does not exist", roleName))
- }
- var role Role
- err := json.Unmarshal(r, &role)
- if err != nil {
- return err
- }
- if !util.Contains(role.Admins, username) {
- role.Admins = append(role.Admins, username)
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) removeAdminFromRole(username, roleName string, bucket *bolt.Bucket) error {
- if roleName == "" {
- return nil
- }
- r := bucket.Get([]byte(roleName))
- if r == nil {
- providerLog(logger.LevelWarn, "role %q does not exist, cannot remove admin %q", roleName, username)
- return nil
- }
- var role Role
- err := json.Unmarshal(r, &role)
- if err != nil {
- return err
- }
- if util.Contains(role.Admins, username) {
- var admins []string
- for _, admin := range role.Admins {
- if admin != username {
- admins = append(admins, admin)
- }
- }
- role.Admins = util.RemoveDuplicates(admins, false)
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) addUserToRole(username, roleName string, bucket *bolt.Bucket) error {
- if roleName == "" {
- return nil
- }
- r := bucket.Get([]byte(roleName))
- if r == nil {
- return util.NewGenericError(fmt.Sprintf("role %q does not exist", roleName))
- }
- var role Role
- err := json.Unmarshal(r, &role)
- if err != nil {
- return err
- }
- if !util.Contains(role.Users, username) {
- role.Users = append(role.Users, username)
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) removeUserFromRole(username, roleName string, bucket *bolt.Bucket) error {
- if roleName == "" {
- return nil
- }
- r := bucket.Get([]byte(roleName))
- if r == nil {
- providerLog(logger.LevelWarn, "role %q does not exist, cannot remove admin %q", roleName, username)
- return nil
- }
- var role Role
- err := json.Unmarshal(r, &role)
- if err != nil {
- return err
- }
- if util.Contains(role.Users, username) {
- var users []string
- for _, user := range role.Users {
- if user != username {
- users = append(users, user)
- }
- }
- users = util.RemoveDuplicates(users, false)
- role.Users = users
- buf, err := json.Marshal(role)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(role.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) addRuleToActionMapping(ruleName, actionName string, bucket *bolt.Bucket) error {
- a := bucket.Get([]byte(actionName))
- if a == nil {
- return util.NewGenericError(fmt.Sprintf("action %q does not exist", actionName))
- }
- var action BaseEventAction
- err := json.Unmarshal(a, &action)
- if err != nil {
- return err
- }
- if !util.Contains(action.Rules, ruleName) {
- action.Rules = append(action.Rules, ruleName)
- buf, err := json.Marshal(action)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(action.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) removeRuleFromActionMapping(ruleName, actionName string, bucket *bolt.Bucket) error {
- a := bucket.Get([]byte(actionName))
- if a == nil {
- providerLog(logger.LevelWarn, "action %q does not exist, cannot remove from mapping", actionName)
- return nil
- }
- var action BaseEventAction
- err := json.Unmarshal(a, &action)
- if err != nil {
- return err
- }
- if util.Contains(action.Rules, ruleName) {
- var rules []string
- for _, r := range action.Rules {
- if r != ruleName {
- rules = append(rules, r)
- }
- }
- action.Rules = util.RemoveDuplicates(rules, false)
- buf, err := json.Marshal(action)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(action.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) addUserToGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
- g := bucket.Get([]byte(groupname))
- if g == nil {
- return util.NewGenericError(fmt.Sprintf("group %q does not exist", groupname))
- }
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return err
- }
- if !util.Contains(group.Users, username) {
- group.Users = append(group.Users, username)
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) removeUserFromGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
- g := bucket.Get([]byte(groupname))
- if g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
- }
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return err
- }
- var users []string
- for _, u := range group.Users {
- if u != username {
- users = append(users, u)
- }
- }
- group.Users = util.RemoveDuplicates(users, false)
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- }
- func (p *BoltProvider) addAdminToGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
- g := bucket.Get([]byte(groupname))
- if g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
- }
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return err
- }
- if !util.Contains(group.Admins, username) {
- group.Admins = append(group.Admins, username)
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- }
- return nil
- }
- func (p *BoltProvider) removeAdminFromGroupMapping(username, groupname string, bucket *bolt.Bucket) error {
- g := bucket.Get([]byte(groupname))
- if g == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("group %q does not exist", groupname))
- }
- var group Group
- err := json.Unmarshal(g, &group)
- if err != nil {
- return err
- }
- var admins []string
- for _, a := range group.Admins {
- if a != username {
- admins = append(admins, a)
- }
- }
- group.Admins = util.RemoveDuplicates(admins, false)
- buf, err := json.Marshal(group)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(group.Name), buf)
- }
- func (p *BoltProvider) removeGroupFromAdminMapping(groupName, adminName string, bucket *bolt.Bucket) error {
- var a []byte
- if a = bucket.Get([]byte(adminName)); a == nil {
- // the admin does not exist so there is no associated group
- return nil
- }
- var admin Admin
- err := json.Unmarshal(a, &admin)
- if err != nil {
- return err
- }
- var newGroups []AdminGroupMapping
- for _, g := range admin.Groups {
- if g.Name != groupName {
- newGroups = append(newGroups, g)
- }
- }
- admin.Groups = newGroups
- buf, err := json.Marshal(admin)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(adminName), buf)
- }
- func (p *BoltProvider) addRelationToFolderMapping(folderName string, user *User, group *Group, bucket *bolt.Bucket) error {
- f := bucket.Get([]byte(folderName))
- if f == nil {
- return util.NewGenericError(fmt.Sprintf("folder %q does not exist", folderName))
- }
- var folder vfs.BaseVirtualFolder
- err := json.Unmarshal(f, &folder)
- if err != nil {
- return err
- }
- updated := false
- if user != nil && !util.Contains(folder.Users, user.Username) {
- folder.Users = append(folder.Users, user.Username)
- updated = true
- }
- if group != nil && !util.Contains(folder.Groups, group.Name) {
- folder.Groups = append(folder.Groups, group.Name)
- updated = true
- }
- if !updated {
- return nil
- }
- buf, err := json.Marshal(folder)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(folder.Name), buf)
- }
- func (p *BoltProvider) removeRelationFromFolderMapping(folder vfs.VirtualFolder, username, groupname string,
- bucket *bolt.Bucket,
- ) error {
- var f []byte
- if f = bucket.Get([]byte(folder.Name)); f == nil {
- // the folder does not exist so there is no associated user/group
- return nil
- }
- var baseFolder vfs.BaseVirtualFolder
- err := json.Unmarshal(f, &baseFolder)
- if err != nil {
- return err
- }
- found := false
- if username != "" {
- found = true
- var newUserMapping []string
- for _, u := range baseFolder.Users {
- if u != username {
- newUserMapping = append(newUserMapping, u)
- }
- }
- baseFolder.Users = newUserMapping
- }
- if groupname != "" {
- found = true
- var newGroupMapping []string
- for _, g := range baseFolder.Groups {
- if g != groupname {
- newGroupMapping = append(newGroupMapping, g)
- }
- }
- baseFolder.Groups = newGroupMapping
- }
- if !found {
- return nil
- }
- buf, err := json.Marshal(baseFolder)
- if err != nil {
- return err
- }
- return bucket.Put([]byte(folder.Name), buf)
- }
- func (p *BoltProvider) updateUserRelations(tx *bolt.Tx, user *User, oldUser User) error {
- foldersBucket, err := p.getFoldersBucket(tx)
- if err != nil {
- return err
- }
- groupsBucket, err := p.getGroupsBucket(tx)
- if err != nil {
- return err
- }
- rolesBucket, err := p.getRolesBucket(tx)
- if err != nil {
- return err
- }
- for idx := range oldUser.VirtualFolders {
- err = p.removeRelationFromFolderMapping(oldUser.VirtualFolders[idx], oldUser.Username, "", foldersBucket)
- if err != nil {
- return err
- }
- }
- for idx := range oldUser.Groups {
- err = p.removeUserFromGroupMapping(user.Username, oldUser.Groups[idx].Name, groupsBucket)
- if err != nil {
- return err
- }
- }
- if err = p.removeUserFromRole(oldUser.Username, oldUser.Role, rolesBucket); err != nil {
- return err
- }
- sort.Slice(user.VirtualFolders, func(i, j int) bool {
- return user.VirtualFolders[i].Name < user.VirtualFolders[j].Name
- })
- for idx := range user.VirtualFolders {
- err = p.addRelationToFolderMapping(user.VirtualFolders[idx].Name, user, nil, foldersBucket)
- if err != nil {
- return err
- }
- }
- sort.Slice(user.Groups, func(i, j int) bool {
- return user.Groups[i].Name < user.Groups[j].Name
- })
- for idx := range user.Groups {
- err = p.addUserToGroupMapping(user.Username, user.Groups[idx].Name, groupsBucket)
- if err != nil {
- return err
- }
- }
- return p.addUserToRole(user.Username, user.Role, rolesBucket)
- }
- func (p *BoltProvider) adminExistsInternal(tx *bolt.Tx, username string) error {
- bucket, err := p.getAdminsBucket(tx)
- if err != nil {
- return err
- }
- a := bucket.Get([]byte(username))
- if a == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("admin %v does not exist", username))
- }
- return nil
- }
- func (p *BoltProvider) userExistsInternal(tx *bolt.Tx, username string) error {
- bucket, err := p.getUsersBucket(tx)
- if err != nil {
- return err
- }
- u := bucket.Get([]byte(username))
- if u == nil {
- return util.NewRecordNotFoundError(fmt.Sprintf("username %q does not exist", username))
- }
- return nil
- }
- func (p *BoltProvider) deleteRelatedShares(tx *bolt.Tx, username string) error {
- bucket, err := p.getSharesBucket(tx)
- if err != nil {
- return err
- }
- var toRemove []string
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var share Share
- err = json.Unmarshal(v, &share)
- if err != nil {
- return err
- }
- if share.Username == username {
- toRemove = append(toRemove, share.ShareID)
- }
- }
- for _, k := range toRemove {
- if err := bucket.Delete([]byte(k)); err != nil {
- return err
- }
- }
- return nil
- }
- func (p *BoltProvider) deleteRelatedAPIKey(tx *bolt.Tx, username string, scope APIKeyScope) error {
- bucket, err := p.getAPIKeysBucket(tx)
- if err != nil {
- return err
- }
- var toRemove []string
- cursor := bucket.Cursor()
- for k, v := cursor.First(); k != nil; k, v = cursor.Next() {
- var apiKey APIKey
- err = json.Unmarshal(v, &apiKey)
- if err != nil {
- return err
- }
- if scope == APIKeyScopeUser {
- if apiKey.User == username {
- toRemove = append(toRemove, apiKey.KeyID)
- }
- } else {
- if apiKey.Admin == username {
- toRemove = append(toRemove, apiKey.KeyID)
- }
- }
- }
- for _, k := range toRemove {
- if err := bucket.Delete([]byte(k)); err != nil {
- return err
- }
- }
- return nil
- }
- func (p *BoltProvider) getSharesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(sharesBucket)
- if bucket == nil {
- err = errors.New("unable to find shares bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getAPIKeysBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(apiKeysBucket)
- if bucket == nil {
- err = errors.New("unable to find api keys bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getAdminsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(adminsBucket)
- if bucket == nil {
- err = errors.New("unable to find admins bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getUsersBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(usersBucket)
- if bucket == nil {
- err = errors.New("unable to find users bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getGroupsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(groupsBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find groups bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getRolesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(rolesBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find roles bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getIPListsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(rolesBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find IP lists bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getFoldersBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(foldersBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find folders bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getActionsBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(actionsBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find event actions bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func (p *BoltProvider) getRulesBucket(tx *bolt.Tx) (*bolt.Bucket, error) {
- var err error
- bucket := tx.Bucket(rulesBucket)
- if bucket == nil {
- err = fmt.Errorf("unable to find event rules bucket, bolt database structure not correcly defined")
- }
- return bucket, err
- }
- func getBoltDatabaseVersion(dbHandle *bolt.DB) (schemaVersion, error) {
- var dbVersion schemaVersion
- err := dbHandle.View(func(tx *bolt.Tx) error {
- bucket := tx.Bucket(dbVersionBucket)
- if bucket == nil {
- return fmt.Errorf("unable to find database schema version bucket")
- }
- v := bucket.Get(dbVersionKey)
- if v == nil {
- dbVersion = schemaVersion{
- Version: 28,
- }
- return nil
- }
- return json.Unmarshal(v, &dbVersion)
- })
- return dbVersion, err
- }
- /*func updateBoltDatabaseVersion(dbHandle *bolt.DB, version int) error {
- err := dbHandle.Update(func(tx *bolt.Tx) error {
- bucket := tx.Bucket(dbVersionBucket)
- if bucket == nil {
- return fmt.Errorf("unable to find database schema version bucket")
- }
- newDbVersion := schemaVersion{
- Version: version,
- }
- buf, err := json.Marshal(newDbVersion)
- if err != nil {
- return err
- }
- return bucket.Put(dbVersionKey, buf)
- })
- return err
- }*/
|