auth

Paddy 2014-09-29 Parent:3a6a65ed380c Child:73a9f7a6af54

48:d78418fb9f56 Go to Latest

auth/profile.go

Validate profile changes. Implement validation for ProfileChange and BulkProfileChange structs. Add unit tests to ensure that validation is working as intended.

Download raw file
View source Diff to previous Annotate
History
paddy@27 1 package auth
paddy@27 2
paddy@27 3 import (
paddy@38 4 "errors"
paddy@27 5 "time"
paddy@27 6
paddy@45 7 "code.secondbit.org/uuid"
paddy@27 8 )
paddy@27 9
paddy@48 10 const (
paddy@48 11 MinPassphraseLength = 6
paddy@48 12 MaxPassphraseLength = 64
paddy@48 13 )
paddy@48 14
paddy@38 15 var (
paddy@38 16 ErrProfileAlreadyExists = errors.New("profile already exists in ProfileStore")
paddy@38 17 ErrProfileNotFound = errors.New("profile not found in ProfileStore")
paddy@44 18 ErrLoginAlreadyExists = errors.New("login already exists in ProfileStore")
paddy@44 19 ErrLoginNotFound = errors.New("login not found in ProfileStore")
paddy@48 20
paddy@48 21 ErrMissingPassphrase = errors.New("missing passphrase")
paddy@48 22 ErrMissingPassphraseReset = errors.New("missing passphrase reset")
paddy@48 23 ErrMissingPassphraseResetCreated = errors.New("missing passphrase reset created timestamp")
paddy@48 24 ErrPassphraseTooShort = errors.New("passphrase too short")
paddy@48 25 ErrPassphraseTooLong = errors.New("passphrase too long")
paddy@38 26 )
paddy@38 27
paddy@27 28 type Profile struct {
paddy@38 29 ID uuid.ID
paddy@38 30 Name string
paddy@38 31 Passphrase string
paddy@38 32 Iterations int64
paddy@38 33 Salt string
paddy@38 34 PassphraseScheme int
paddy@38 35 Compromised bool
paddy@38 36 LockedUntil time.Time
paddy@38 37 PassphraseReset string
paddy@38 38 PassphraseResetCreated time.Time
paddy@38 39 Created time.Time
paddy@38 40 LastSeen time.Time
paddy@38 41 }
paddy@38 42
paddy@38 43 func (p *Profile) ApplyChange(change ProfileChange) {
paddy@38 44 if change.Name != nil {
paddy@38 45 p.Name = *change.Name
paddy@38 46 }
paddy@38 47 if change.Passphrase != nil {
paddy@38 48 p.Passphrase = *change.Passphrase
paddy@38 49 }
paddy@38 50 if change.Iterations != nil {
paddy@38 51 p.Iterations = *change.Iterations
paddy@38 52 }
paddy@38 53 if change.Salt != nil {
paddy@38 54 p.Salt = *change.Salt
paddy@38 55 }
paddy@38 56 if change.PassphraseScheme != nil {
paddy@38 57 p.PassphraseScheme = *change.PassphraseScheme
paddy@38 58 }
paddy@38 59 if change.Compromised != nil {
paddy@38 60 p.Compromised = *change.Compromised
paddy@38 61 }
paddy@38 62 if change.LockedUntil != nil {
paddy@38 63 p.LockedUntil = *change.LockedUntil
paddy@38 64 }
paddy@38 65 if change.PassphraseReset != nil {
paddy@38 66 p.PassphraseReset = *change.PassphraseReset
paddy@38 67 }
paddy@38 68 if change.PassphraseResetCreated != nil {
paddy@38 69 p.PassphraseResetCreated = *change.PassphraseResetCreated
paddy@38 70 }
paddy@38 71 if change.LastSeen != nil {
paddy@38 72 p.LastSeen = *change.LastSeen
paddy@38 73 }
paddy@38 74 }
paddy@38 75
paddy@44 76 func (p *Profile) ApplyBulkChange(change BulkProfileChange) {
paddy@44 77 if change.Compromised != nil {
paddy@44 78 p.Compromised = *change.Compromised
paddy@44 79 }
paddy@44 80 }
paddy@44 81
paddy@38 82 type ProfileChange struct {
paddy@38 83 Name *string
paddy@38 84 Passphrase *string
paddy@38 85 Iterations *int64
paddy@38 86 Salt *string
paddy@38 87 PassphraseScheme *int
paddy@38 88 Compromised *bool
paddy@38 89 LockedUntil *time.Time
paddy@38 90 PassphraseReset *string
paddy@38 91 PassphraseResetCreated *time.Time
paddy@38 92 LastSeen *time.Time
paddy@38 93 }
paddy@38 94
paddy@38 95 func (c ProfileChange) Validate() error {
paddy@48 96 if c.Name == nil && c.Passphrase == nil && c.Iterations == nil && c.Salt == nil && c.PassphraseScheme == nil && c.Compromised == nil && c.LockedUntil == nil && c.PassphraseReset == nil && c.PassphraseResetCreated == nil && c.LastSeen == nil {
paddy@48 97 return ErrEmptyChange
paddy@48 98 }
paddy@48 99 if c.PassphraseScheme != nil && c.Passphrase == nil {
paddy@48 100 return ErrMissingPassphrase
paddy@48 101 }
paddy@48 102 if c.PassphraseReset != nil && c.PassphraseResetCreated == nil {
paddy@48 103 return ErrMissingPassphraseResetCreated
paddy@48 104 }
paddy@48 105 if c.PassphraseReset == nil && c.PassphraseResetCreated != nil {
paddy@48 106 return ErrMissingPassphraseReset
paddy@48 107 }
paddy@48 108 if c.Salt != nil && c.Passphrase == nil {
paddy@48 109 return ErrMissingPassphrase
paddy@48 110 }
paddy@48 111 if c.Iterations != nil && c.Passphrase == nil {
paddy@48 112 return ErrMissingPassphrase
paddy@48 113 }
paddy@48 114 if c.Passphrase != nil && len(*c.Passphrase) < MinPassphraseLength {
paddy@48 115 return ErrPassphraseTooShort
paddy@48 116 }
paddy@48 117 if c.Passphrase != nil && len(*c.Passphrase) > MaxPassphraseLength {
paddy@48 118 return ErrPassphraseTooLong
paddy@48 119 }
paddy@38 120 return nil
paddy@27 121 }
paddy@27 122
paddy@44 123 type BulkProfileChange struct {
paddy@44 124 Compromised *bool
paddy@44 125 }
paddy@44 126
paddy@44 127 func (b BulkProfileChange) Validate() error {
paddy@48 128 if b.Compromised == nil {
paddy@48 129 return ErrEmptyChange
paddy@48 130 }
paddy@44 131 return nil
paddy@44 132 }
paddy@44 133
paddy@27 134 type Login struct {
paddy@27 135 Type string
paddy@27 136 Value string
paddy@27 137 ProfileID uuid.ID
paddy@27 138 Created time.Time
paddy@27 139 LastUsed time.Time
paddy@27 140 }
paddy@27 141
paddy@27 142 type ProfileStore interface {
paddy@27 143 GetProfileByID(id uuid.ID) (Profile, error)
paddy@44 144 GetProfileByLogin(loginType, value string) (Profile, error)
paddy@38 145 SaveProfile(profile Profile) error
paddy@38 146 UpdateProfile(id uuid.ID, change ProfileChange) error
paddy@44 147 UpdateProfiles(ids []uuid.ID, change BulkProfileChange) error
paddy@27 148 DeleteProfile(id uuid.ID) error
paddy@44 149
paddy@44 150 AddLogin(login Login) error
paddy@44 151 RemoveLogin(loginType, value string, profile uuid.ID) error
paddy@44 152 RecordLoginUse(loginType, value string, when time.Time) error
paddy@44 153 ListLogins(profile uuid.ID, num, offset int) ([]Login, error)
paddy@38 154 }
paddy@27 155
paddy@38 156 func (m *Memstore) GetProfileByID(id uuid.ID) (Profile, error) {
paddy@38 157 m.profileLock.RLock()
paddy@38 158 defer m.profileLock.RUnlock()
paddy@38 159 p, ok := m.profiles[id.String()]
paddy@38 160 if !ok {
paddy@38 161 return Profile{}, ErrProfileNotFound
paddy@38 162 }
paddy@38 163 return p, nil
paddy@27 164 }
paddy@38 165
paddy@44 166 func (m *Memstore) GetProfileByLogin(loginType, value string) (Profile, error) {
paddy@44 167 m.loginLock.RLock()
paddy@44 168 defer m.loginLock.RUnlock()
paddy@44 169 login, ok := m.logins[loginType+":"+value]
paddy@44 170 if !ok {
paddy@44 171 return Profile{}, ErrLoginNotFound
paddy@44 172 }
paddy@44 173 m.profileLock.RLock()
paddy@44 174 defer m.profileLock.RUnlock()
paddy@44 175 profile, ok := m.profiles[login.ProfileID.String()]
paddy@44 176 if !ok {
paddy@44 177 return Profile{}, ErrProfileNotFound
paddy@44 178 }
paddy@44 179 return profile, nil
paddy@38 180 }
paddy@38 181
paddy@38 182 func (m *Memstore) SaveProfile(profile Profile) error {
paddy@38 183 m.profileLock.Lock()
paddy@38 184 defer m.profileLock.Unlock()
paddy@38 185 _, ok := m.profiles[profile.ID.String()]
paddy@38 186 if ok {
paddy@38 187 return ErrProfileAlreadyExists
paddy@38 188 }
paddy@38 189 m.profiles[profile.ID.String()] = profile
paddy@38 190 return nil
paddy@38 191 }
paddy@38 192
paddy@38 193 func (m *Memstore) UpdateProfile(id uuid.ID, change ProfileChange) error {
paddy@38 194 m.profileLock.Lock()
paddy@38 195 defer m.profileLock.Unlock()
paddy@38 196 p, ok := m.profiles[id.String()]
paddy@38 197 if !ok {
paddy@38 198 return ErrProfileNotFound
paddy@38 199 }
paddy@38 200 p.ApplyChange(change)
paddy@38 201 m.profiles[id.String()] = p
paddy@38 202 return nil
paddy@38 203 }
paddy@38 204
paddy@44 205 func (m *Memstore) UpdateProfiles(ids []uuid.ID, change BulkProfileChange) error {
paddy@44 206 m.profileLock.Lock()
paddy@44 207 defer m.profileLock.Unlock()
paddy@44 208 for id, profile := range m.profiles {
paddy@44 209 for _, i := range ids {
paddy@44 210 if id == i.String() {
paddy@44 211 profile.ApplyBulkChange(change)
paddy@44 212 m.profiles[id] = profile
paddy@44 213 break
paddy@44 214 }
paddy@44 215 }
paddy@44 216 }
paddy@44 217 return nil
paddy@44 218 }
paddy@44 219
paddy@38 220 func (m *Memstore) DeleteProfile(id uuid.ID) error {
paddy@38 221 m.profileLock.Lock()
paddy@38 222 defer m.profileLock.Unlock()
paddy@38 223 _, ok := m.profiles[id.String()]
paddy@38 224 if !ok {
paddy@38 225 return ErrProfileNotFound
paddy@38 226 }
paddy@38 227 delete(m.profiles, id.String())
paddy@38 228 return nil
paddy@38 229 }
paddy@40 230
paddy@44 231 func (m *Memstore) AddLogin(login Login) error {
paddy@44 232 m.loginLock.Lock()
paddy@44 233 defer m.loginLock.Unlock()
paddy@44 234 _, ok := m.logins[login.Type+":"+login.Value]
paddy@44 235 if ok {
paddy@44 236 return ErrLoginAlreadyExists
paddy@44 237 }
paddy@44 238 m.logins[login.Type+":"+login.Value] = login
paddy@44 239 m.profileLoginLookup[login.ProfileID.String()] = append(m.profileLoginLookup[login.ProfileID.String()], login.Type+":"+login.Value)
paddy@44 240 return nil
paddy@44 241 }
paddy@44 242
paddy@44 243 func (m *Memstore) RemoveLogin(loginType, value string, profile uuid.ID) error {
paddy@44 244 m.loginLock.Lock()
paddy@44 245 defer m.loginLock.Unlock()
paddy@44 246 l, ok := m.logins[loginType+":"+value]
paddy@44 247 if !ok {
paddy@44 248 return ErrLoginNotFound
paddy@44 249 }
paddy@44 250 if !l.ProfileID.Equal(profile) {
paddy@44 251 return ErrLoginNotFound
paddy@44 252 }
paddy@44 253 delete(m.logins, loginType+":"+value)
paddy@44 254 pos := -1
paddy@44 255 for p, id := range m.profileLoginLookup[profile.String()] {
paddy@44 256 if id == loginType+":"+value {
paddy@44 257 pos = p
paddy@44 258 break
paddy@44 259 }
paddy@44 260 }
paddy@44 261 if pos >= 0 {
paddy@44 262 m.profileLoginLookup[profile.String()] = append(m.profileLoginLookup[profile.String()][:pos], m.profileLoginLookup[profile.String()][pos+1:]...)
paddy@44 263 }
paddy@44 264 return nil
paddy@44 265 }
paddy@44 266
paddy@44 267 func (m *Memstore) RecordLoginUse(loginType, value string, when time.Time) error {
paddy@44 268 m.loginLock.Lock()
paddy@44 269 defer m.loginLock.Unlock()
paddy@44 270 l, ok := m.logins[loginType+":"+value]
paddy@44 271 if !ok {
paddy@44 272 return ErrLoginNotFound
paddy@44 273 }
paddy@44 274 l.LastUsed = when
paddy@44 275 m.logins[loginType+":"+value] = l
paddy@44 276 return nil
paddy@44 277 }
paddy@44 278
paddy@44 279 func (m *Memstore) ListLogins(profile uuid.ID, num, offset int) ([]Login, error) {
paddy@44 280 m.loginLock.RLock()
paddy@44 281 defer m.loginLock.RUnlock()
paddy@44 282 ids, ok := m.profileLoginLookup[profile.String()]
paddy@44 283 if !ok {
paddy@44 284 return []Login{}, nil
paddy@44 285 }
paddy@44 286 if len(ids) > num+offset {
paddy@44 287 ids = ids[offset : num+offset]
paddy@44 288 } else if len(ids) > offset {
paddy@44 289 ids = ids[offset:]
paddy@44 290 } else {
paddy@44 291 return []Login{}, nil
paddy@44 292 }
paddy@44 293 logins := []Login{}
paddy@44 294 for _, id := range ids {
paddy@44 295 login, ok := m.logins[id]
paddy@44 296 if !ok {
paddy@44 297 continue
paddy@44 298 }
paddy@44 299 logins = append(logins, login)
paddy@44 300 }
paddy@44 301 return logins, nil
paddy@44 302 }