2016-01-07 17:27:53 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
|
|
|
"log"
|
|
|
|
"net/http"
|
2016-01-07 17:28:16 +00:00
|
|
|
"os"
|
2016-11-19 15:26:23 +00:00
|
|
|
"path"
|
2016-01-20 21:44:34 +00:00
|
|
|
"path/filepath"
|
2016-11-19 15:26:23 +00:00
|
|
|
"text/template"
|
2016-01-13 19:25:25 +00:00
|
|
|
|
|
|
|
"srs.epita.fr/fic-server/libfic"
|
2016-01-07 17:27:53 +00:00
|
|
|
)
|
|
|
|
|
2016-01-20 18:56:08 +00:00
|
|
|
var PKIDir string
|
2016-01-07 17:27:53 +00:00
|
|
|
var SubmissionDir string
|
|
|
|
var BaseURL string
|
2016-01-20 21:44:34 +00:00
|
|
|
var CloudDAVBase string
|
|
|
|
var CloudUsername string
|
|
|
|
var CloudPassword string
|
2016-11-19 15:35:39 +00:00
|
|
|
var StaticDir string
|
2016-01-07 17:27:53 +00:00
|
|
|
|
|
|
|
func main() {
|
2016-11-19 15:17:53 +00:00
|
|
|
var bind = flag.String("bind", "127.0.0.1:8081", "Bind port/socket")
|
2016-01-23 12:16:31 +00:00
|
|
|
var dsn = flag.String("dsn", "fic:fic@/fic", "DSN to connect to the MySQL server")
|
2016-11-19 15:26:23 +00:00
|
|
|
flag.StringVar(&BaseURL, "baseurl", "/", "URL prepended to each URL")
|
2016-01-07 17:27:53 +00:00
|
|
|
flag.StringVar(&SubmissionDir, "submission", "./submissions/", "Base directory where save submissions")
|
2016-01-20 18:56:08 +00:00
|
|
|
flag.StringVar(&PKIDir, "pki", "./pki/", "Base directory where found PKI scripts")
|
2016-11-19 15:35:39 +00:00
|
|
|
flag.StringVar(&StaticDir, "static", "./htdocs-admin/", "Directory containing static files")
|
2016-01-20 21:44:34 +00:00
|
|
|
flag.StringVar(&fic.FilesDir, "files", "./FILES/", "Base directory where found challenges files, local part")
|
|
|
|
flag.StringVar(&CloudDAVBase, "clouddav", "https://srs.epita.fr/owncloud/remote.php/webdav/FIC 2016",
|
|
|
|
"Base directory where found challenges files, cloud part")
|
|
|
|
flag.StringVar(&CloudUsername, "clouduser", "fic", "Username used to sync")
|
|
|
|
flag.StringVar(&CloudPassword, "cloudpass", "", "Password used to sync")
|
2016-01-07 17:27:53 +00:00
|
|
|
flag.Parse()
|
|
|
|
|
2016-10-13 17:02:41 +00:00
|
|
|
log.SetPrefix("[admin] ")
|
|
|
|
|
2016-01-20 21:44:34 +00:00
|
|
|
var err error
|
|
|
|
log.Println("Checking paths...")
|
2016-11-19 15:35:39 +00:00
|
|
|
if StaticDir, err = filepath.Abs(StaticDir); err != nil {
|
2016-01-22 16:34:34 +00:00
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2016-01-20 21:44:34 +00:00
|
|
|
if fic.FilesDir, err = filepath.Abs(fic.FilesDir); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
if PKIDir, err = filepath.Abs(PKIDir); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
if SubmissionDir, err = filepath.Abs(SubmissionDir); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2016-01-22 16:34:34 +00:00
|
|
|
if fic.FilesDir, err = filepath.Abs(fic.FilesDir); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2016-01-20 21:44:34 +00:00
|
|
|
|
2016-01-07 17:28:16 +00:00
|
|
|
log.Println("Opening database...")
|
2016-01-23 12:16:31 +00:00
|
|
|
if err := fic.DBInit(fmt.Sprintf("%s?parseTime=true", *dsn)); err != nil {
|
2016-01-07 17:28:16 +00:00
|
|
|
log.Fatal("Cannot open the database: ", err)
|
|
|
|
}
|
2016-01-13 19:25:25 +00:00
|
|
|
defer fic.DBClose()
|
2016-01-07 17:28:16 +00:00
|
|
|
|
|
|
|
log.Println("Creating database...")
|
2016-01-13 19:25:25 +00:00
|
|
|
if err := fic.DBCreate(); err != nil {
|
2016-01-07 17:28:16 +00:00
|
|
|
log.Fatal("Cannot create database: ", err)
|
|
|
|
}
|
|
|
|
|
2016-11-19 15:26:23 +00:00
|
|
|
log.Println("Changing base url...")
|
2016-11-19 15:35:39 +00:00
|
|
|
if file, err := os.OpenFile(path.Join(StaticDir, "index.html"), os.O_CREATE|os.O_WRONLY|os.O_TRUNC, os.FileMode(0644)); err != nil {
|
2016-11-19 15:26:23 +00:00
|
|
|
log.Println("Unable to open index.html: ", err)
|
|
|
|
} else if indexTmpl, err := template.New("index").Parse(indextpl); err != nil {
|
|
|
|
log.Println("Cannot create template: ", err)
|
2016-11-19 15:35:39 +00:00
|
|
|
} else if err := indexTmpl.Execute(file, map[string]string{"urlbase": path.Clean(path.Join(baseURL + "/", "nuke"))[:len(path.Clean(path.Join(baseURL + "/", "nuke"))) - 4]}); err != nil {
|
2016-11-19 15:26:23 +00:00
|
|
|
log.Println("An error occurs during template execution: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-20 18:56:08 +00:00
|
|
|
os.Chdir(PKIDir)
|
|
|
|
|
2016-01-07 17:27:53 +00:00
|
|
|
log.Println("Registering handlers...")
|
2016-01-13 00:22:01 +00:00
|
|
|
mux := http.NewServeMux()
|
2016-11-19 15:26:23 +00:00
|
|
|
mux.Handle(path.Join(BaseURL, "api") + "/", http.StripPrefix(path.Join(BaseURL, "api"), ApiHandler()))
|
|
|
|
mux.Handle(path.Join(BaseURL, "teams") + "/", http.StripPrefix(BaseURL, StaticHandler(staticDir)))
|
|
|
|
mux.Handle(path.Join(BaseURL, "themes") + "/", http.StripPrefix(BaseURL, StaticHandler(staticDir)))
|
|
|
|
mux.Handle(BaseURL, http.StripPrefix(BaseURL, http.FileServer(http.Dir(staticDir))))
|
2016-01-07 17:27:53 +00:00
|
|
|
|
|
|
|
log.Println(fmt.Sprintf("Ready, listening on %s", *bind))
|
2016-06-25 14:48:16 +00:00
|
|
|
if err := http.ListenAndServe(*bind, mux); err != nil {
|
2016-01-07 17:27:53 +00:00
|
|
|
log.Fatal("Unable to listen and serve: ", err)
|
|
|
|
}
|
|
|
|
}
|