2018-02-08 21:27:30 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
"path/filepath"
|
|
|
|
)
|
|
|
|
|
2018-02-12 10:39:44 +00:00
|
|
|
var tftpDir string
|
|
|
|
|
2018-02-08 21:27:30 +00:00
|
|
|
func main() {
|
2018-02-09 15:26:15 +00:00
|
|
|
var studentsFile string
|
|
|
|
|
2018-02-18 13:41:06 +00:00
|
|
|
var lc loginChecker
|
|
|
|
|
2018-02-08 21:27:30 +00:00
|
|
|
var bind = flag.String("bind", ":8081", "Bind port/socket")
|
2018-02-09 15:26:15 +00:00
|
|
|
flag.StringVar(&studentsFile, "students", "./students.csv", "Path to a CSV file containing students list")
|
2018-02-08 21:27:30 +00:00
|
|
|
flag.StringVar(&ARPTable, "arp", ARPTable, "Path to ARP table")
|
2018-02-12 10:39:44 +00:00
|
|
|
flag.StringVar(&tftpDir, "tftpdir", "/var/tftp/", "Path to TFTPd directory")
|
2018-02-18 16:37:16 +00:00
|
|
|
flag.StringVar(&loginSalt, "loginsalt", "adelina", "secret used in login HMAC")
|
2018-02-18 13:41:06 +00:00
|
|
|
|
|
|
|
flag.StringVar(&lc.ldapAddr, "ldaphost", "auth.cri.epita.fr", "LDAP host")
|
|
|
|
flag.IntVar(&lc.ldapPort, "ldapport", 636, "LDAP port")
|
|
|
|
flag.BoolVar(&lc.ldapIsTLS, "ldaptls", false, "Is LDAP connection LDAPS?")
|
|
|
|
flag.StringVar(&lc.ldapBase, "ldapbase", "dc=epita,dc=net", "LDAP base")
|
|
|
|
flag.StringVar(&lc.ldapBindUsername, "ldapbindusername", "", "LDAP user to use in order to perform bind (optional if search can be made anonymously)")
|
|
|
|
flag.StringVar(&lc.ldapBindPassword, "ldapbindpassword", "", "Password for the bind user")
|
2018-02-08 21:27:30 +00:00
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
var err error
|
|
|
|
|
|
|
|
// Sanitize options
|
|
|
|
log.Println("Checking paths...")
|
2018-02-12 10:39:44 +00:00
|
|
|
if tftpDir, err = filepath.Abs(tftpDir); err != nil {
|
2018-02-08 21:27:30 +00:00
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2018-02-18 13:41:06 +00:00
|
|
|
lc.students, err = readStudentsList(studentsFile)
|
2018-02-09 15:26:15 +00:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2018-02-08 21:27:30 +00:00
|
|
|
log.Println("Registering handlers...")
|
|
|
|
mux := http.NewServeMux()
|
|
|
|
mux.HandleFunc("/", Index)
|
2018-02-18 13:41:06 +00:00
|
|
|
mux.Handle("/login", lc)
|
2018-02-18 13:42:35 +00:00
|
|
|
mux.HandleFunc("/logout", logout)
|
2018-02-08 21:27:30 +00:00
|
|
|
http.HandleFunc("/", mux.ServeHTTP)
|
|
|
|
|
|
|
|
log.Println("Ready, listening on port", *bind)
|
|
|
|
http.ListenAndServe(*bind, nil)
|
|
|
|
}
|