52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/binary"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
|
|
"srs.epita.fr/fic-server/libfic"
|
|
)
|
|
|
|
type wantChoices struct {
|
|
FlagId int `json:"id"`
|
|
}
|
|
|
|
func treatWantChoices(pathname string, team fic.Team) {
|
|
// Generate a unique identifier to follow the request in logs
|
|
bid := make([]byte, 5)
|
|
binary.LittleEndian.PutUint32(bid, rand.Uint32())
|
|
id := "[" + base64.StdEncoding.EncodeToString(bid) + "]"
|
|
log.Println(id, "New wantChoices receive", pathname)
|
|
|
|
var ask wantChoices
|
|
|
|
if cnt_raw, err := ioutil.ReadFile(pathname); err != nil {
|
|
log.Printf("%s [ERR] %s\n", id, err)
|
|
} else if err = json.Unmarshal(cnt_raw, &ask); err != nil {
|
|
log.Printf("%s [ERR] %s\n", id, err)
|
|
} else if ask.FlagId == 0 {
|
|
log.Printf("%s [WRN] Invalid content in wantChoices file: %s\n", id, pathname)
|
|
os.Remove(pathname)
|
|
} else if flag, err := fic.GetFlagKey(ask.FlagId); err != nil {
|
|
log.Printf("%s [ERR] %s\n", id, err)
|
|
} else if !team.CanSeeFlag(flag) {
|
|
log.Printf("%s [!!!] The team asks to display choices whereas it doesn't have access to the flag\n", id)
|
|
} else if exercice, err := flag.GetExercice(); err != nil {
|
|
log.Printf("%s [ERR] Unable to retrieve the flag's underlying exercice: %s\n", id, err)
|
|
} else if !team.HasAccess(exercice) {
|
|
log.Printf("%s [!!!] The team asks to display choices whereas it doesn't have access to the exercice\n", id)
|
|
} else if err = team.DisplayChoices(flag); err != nil {
|
|
log.Printf("%s [ERR] %s\n", id, err)
|
|
} else {
|
|
genTeamQueue <- &team
|
|
if err = os.Remove(pathname); err != nil {
|
|
log.Printf("%s [ERR] %s\n", id, err)
|
|
}
|
|
}
|
|
}
|