Change Key.Value to Key.Checksum

This commit is contained in:
nemunaire 2017-12-04 08:40:17 +01:00
parent ea12b6a0d2
commit b701aa1710
2 changed files with 14 additions and 13 deletions

View file

@ -7,10 +7,10 @@ import (
) )
type Key struct { type Key struct {
Id int64 `json:"id"` Id int64 `json:"id"`
IdExercice int64 `json:"idExercice"` IdExercice int64 `json:"idExercice"`
Type string `json:"type"` Type string `json:"type"`
Value [blake2b.KeySize]byte `json:"value"` Checksum []byte `json:"value"`
} }
func (e Exercice) GetKeys() ([]Key, error) { func (e Exercice) GetKeys() ([]Key, error) {
@ -24,7 +24,7 @@ func (e Exercice) GetKeys() ([]Key, error) {
var k Key var k Key
k.IdExercice = e.Id k.IdExercice = e.Id
if err := rows.Scan(&k.Id, &k.IdExercice, &k.Type, &k.Value); err != nil { if err := rows.Scan(&k.Id, &k.IdExercice, &k.Type, &k.Checksum); err != nil {
return nil, err return nil, err
} }
keys = append(keys, k) keys = append(keys, k)
@ -43,21 +43,22 @@ func getHashedKey(raw_value string) [blake2b.KeySize]byte {
} }
func (e Exercice) AddRawKey(name string, raw_value string) (Key, error) { func (e Exercice) AddRawKey(name string, raw_value string) (Key, error) {
return e.AddKey(name, getHashedKey(raw_value)) hash := getHashedKey(raw_value)
return e.AddKey(name, hash[:])
} }
func (e Exercice) AddKey(name string, value [blake2b.KeySize]byte) (Key, error) { func (e Exercice) AddKey(name string, checksum []byte) (Key, error) {
if res, err := DBExec("INSERT INTO exercice_keys (id_exercice, type, cksum) VALUES (?, ?, ?)", e.Id, name, value); err != nil { if res, err := DBExec("INSERT INTO exercice_keys (id_exercice, type, cksum) VALUES (?, ?, ?)", e.Id, name, checksum); err != nil {
return Key{}, err return Key{}, err
} else if kid, err := res.LastInsertId(); err != nil { } else if kid, err := res.LastInsertId(); err != nil {
return Key{}, err return Key{}, err
} else { } else {
return Key{kid, e.Id, name, value}, nil return Key{kid, e.Id, name, checksum}, nil
} }
} }
func (k Key) Update() (int64, error) { func (k Key) Update() (int64, error) {
if res, err := DBExec("UPDATE exercice_keys SET id_exercice = ?, type = ?, cksum = ? WHERE id_key = ?", k.IdExercice, k.Type, k.Value, k.Id); err != nil { if res, err := DBExec("UPDATE exercice_keys SET id_exercice = ?, type = ?, cksum = ? WHERE id_key = ?", k.IdExercice, k.Type, k.Checksum, k.Id); err != nil {
return 0, err return 0, err
} else if nb, err := res.RowsAffected(); err != nil { } else if nb, err := res.RowsAffected(); err != nil {
return 0, err return 0, err
@ -78,12 +79,12 @@ func (k Key) Delete() (int64, error) {
func (k Key) Check(val string) bool { func (k Key) Check(val string) bool {
hash := getHashedKey(val) hash := getHashedKey(val)
if len(k.Value) != len(hash) { if len(k.Checksum) != len(hash) {
return false return false
} }
for i := range hash { for i := range hash {
if k.Value[i] != hash[i] { if k.Checksum[i] != hash[i] {
return false return false
} }
} }

View file

@ -130,7 +130,7 @@ func MyJSONTeam(t *Team, started bool) (interface{}, error) {
} else { } else {
for _, k := range keys { for _, k := range keys {
if t == nil { if t == nil {
exercice.Keys = append(exercice.Keys, fmt.Sprintf("%x", k.Value)+k.Type) exercice.Keys = append(exercice.Keys, fmt.Sprintf("%x", k.Checksum)+k.Type)
} else { } else {
exercice.Keys = append(exercice.Keys, k.Type) exercice.Keys = append(exercice.Keys, k.Type)
if PartialValidation { if PartialValidation {