forks/molly-brown
1
0
Fork 0
Code Releases Activity

Use standard library logging facilities for error log.

Browse source
This commit is contained in:
Solderpunk 2020-07-01 19:57:39 +02:00
parent 3c4c447bd3
commit 821a862036
3 changed files with 27 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
dynamic.go
View file

@ -5,6 +5,7 @@ import (
"context" "context"
"crypto/tls" "crypto/tls"
"io" "io"
"log"
"net" "net"
"net/url" "net/url"
"os" "os"
@ -14,7 +15,7 @@ import (
"time" "time"
) )
func handleCGI(config Config, path string, cgiPath string, URL *url.URL, log *LogEntry, errorLog chan string, conn net.Conn) { func handleCGI(config Config, path string, cgiPath string, URL *url.URL, log *LogEntry, errorLog *log.Logger, conn net.Conn) {
// Find the shortest leading part of path which maps to an executable file. // Find the shortest leading part of path which maps to an executable file.
// Call this part scriptPath, and everything after it pathInfo. // Call this part scriptPath, and everything after it pathInfo.
components := strings.Split(path, "/") components := strings.Split(path, "/")
@ -57,13 +58,13 @@ func handleCGI(config Config, path string, cgiPath string, URL *url.URL, log *Lo
response, err := cmd.Output() response, err := cmd.Output()
if ctx.Err() == context.DeadlineExceeded { if ctx.Err() == context.DeadlineExceeded {
errorLog <- "Terminating CGI process " + path + " due to exceeding 10 second runtime limit." errorLog.Println("Terminating CGI process " + path + " due to exceeding 10 second runtime limit.")
conn.Write([]byte("42 CGI process timed out!\r\n")) conn.Write([]byte("42 CGI process timed out!\r\n"))
log.Status = 42 log.Status = 42
return return
} }
if err != nil { if err != nil {
errorLog <- "Error starting CGI executable " + path + ": " + err.Error() errorLog.Println("Error starting CGI executable " + path + ": " + err.Error())
conn.Write([]byte("42 CGI error!\r\n")) conn.Write([]byte("42 CGI error!\r\n"))
log.Status = 42 log.Status = 42
return return
@ -72,7 +73,7 @@ func handleCGI(config Config, path string, cgiPath string, URL *url.URL, log *Lo
header, _, err := bufio.NewReader(strings.NewReader(string(response))).ReadLine() header, _, err := bufio.NewReader(strings.NewReader(string(response))).ReadLine()
status, err2 := strconv.Atoi(strings.Fields(string(header))[0]) status, err2 := strconv.Atoi(strings.Fields(string(header))[0])
if err != nil || err2 != nil { if err != nil || err2 != nil {
errorLog <- "Unable to parse first line of output from CGI process " + path + " as valid Gemini response header." errorLog.Println("Unable to parse first line of output from CGI process " + path + " as valid Gemini response header.")
conn.Write([]byte("42 CGI error!\r\n")) conn.Write([]byte("42 CGI error!\r\n"))
log.Status = 42 log.Status = 42
return return

29
handler.go
View file

@ -7,6 +7,7 @@ import (
"fmt" "fmt"
"github.com/BurntSushi/toml" "github.com/BurntSushi/toml"
"io/ioutil" "io/ioutil"
"log"
"mime" "mime"
"net" "net"
"net/url" "net/url"
@ -18,7 +19,7 @@ import (
"time" "time"
) )
func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan LogEntry, errorLogEntries chan string) { func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan LogEntry, errorLog *log.Logger) {
defer conn.Close() defer conn.Close()
var tlsConn (*tls.Conn) = conn.(*tls.Conn) var tlsConn (*tls.Conn) = conn.(*tls.Conn)
var log LogEntry var log LogEntry
@ -29,7 +30,7 @@ func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan Log
defer func() { accessLogEntries <- log }() defer func() { accessLogEntries <- log }()
// Read request // Read request
URL, err := readRequest(conn, &log, errorLogEntries) URL, err := readRequest(conn, &log, errorLog)
if err != nil { if err != nil {
return return
} }
@ -75,7 +76,7 @@ func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan Log
// Read Molly files // Read Molly files
if config.ReadMollyFiles { if config.ReadMollyFiles {
parseMollyFiles(path, &config, errorLogEntries) parseMollyFiles(path, &config, errorLog)
} }
// Check whether this URL is in a certificate zone // Check whether this URL is in a certificate zone
@ -101,7 +102,7 @@ func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan Log
// Check whether this URL is in a configured CGI path // Check whether this URL is in a configured CGI path
for _, cgiPath := range config.CGIPaths { for _, cgiPath := range config.CGIPaths {
if strings.HasPrefix(path, cgiPath) { if strings.HasPrefix(path, cgiPath) {
handleCGI(config, path, cgiPath, URL, &log, errorLogEntries, conn) handleCGI(config, path, cgiPath, URL, &log, errorLog, conn)
if log.Status != 0 { if log.Status != 0 {
return return
} }
@ -126,13 +127,13 @@ func handleGeminiRequest(conn net.Conn, config Config, accessLogEntries chan Log
// Finally, serve the file or directory // Finally, serve the file or directory
if info.IsDir() { if info.IsDir() {
serveDirectory(URL, path, &log, conn, config, errorLogEntries) serveDirectory(URL, path, &log, conn, config, errorLog)
} else { } else {
serveFile(path, &log, conn, config, errorLogEntries) serveFile(path, &log, conn, config, errorLog)
} }
} }
func readRequest(conn net.Conn, log *LogEntry, errorLog chan string) (*url.URL, error) { func readRequest(conn net.Conn, log *LogEntry, errorLog *log.Logger) (*url.URL, error) {
reader := bufio.NewReaderSize(conn, 1024) reader := bufio.NewReaderSize(conn, 1024)
request, overflow, err := reader.ReadLine() request, overflow, err := reader.ReadLine()
if overflow { if overflow {
@ -140,7 +141,7 @@ func readRequest(conn net.Conn, log *LogEntry, errorLog chan string) (*url.URL,
log.Status = 59 log.Status = 59
return nil, errors.New("Request too long") return nil, errors.New("Request too long")
} else if err != nil { } else if err != nil {
errorLog <- "Error reading request: " + err.Error() errorLog.Println("Error reading request: " + err.Error())
conn.Write([]byte("40 Unknown error reading request!\r\n")) conn.Write([]byte("40 Unknown error reading request!\r\n"))
log.Status = 40 log.Status = 40
return nil, errors.New("Error reading request") return nil, errors.New("Error reading request")
@ -149,7 +150,7 @@ func readRequest(conn net.Conn, log *LogEntry, errorLog chan string) (*url.URL,
// Parse request as URL // Parse request as URL
URL, err := url.Parse(string(request)) URL, err := url.Parse(string(request))
if err != nil { if err != nil {
errorLog <- "Error parsing request URL " + string(request) + ": " + err.Error() errorLog.Println("Error parsing request URL " + string(request) + ": " + err.Error())
conn.Write([]byte("59 Error parsing URL!\r\n")) conn.Write([]byte("59 Error parsing URL!\r\n"))
log.Status = 59 log.Status = 59
return nil, errors.New("Bad URL in request") return nil, errors.New("Bad URL in request")
@ -178,7 +179,7 @@ func resolvePath(path string, config Config) string {
return path return path
} }
func parseMollyFiles(path string, config *Config, errorLogEntries chan string) { func parseMollyFiles(path string, config *Config, errorLog *log.Logger) {
// Build list of directories to check // Build list of directories to check
var dirs []string var dirs []string
dirs = append(dirs, path) dirs = append(dirs, path)
@ -214,7 +215,7 @@ func parseMollyFiles(path string, config *Config, errorLogEntries chan string) {
// If the file exists and we can read it, try to parse it // If the file exists and we can read it, try to parse it
_, err = toml.DecodeFile(mollyPath, &mollyFile) _, err = toml.DecodeFile(mollyPath, &mollyFile)
if err != nil { if err != nil {
errorLogEntries <- "Error parsing .molly file " + mollyPath + ": " + err.Error() errorLog.Println("Error parsing .molly file " + mollyPath + ": " + err.Error())
continue continue
} }
// Overwrite main Config using MollyFile // Overwrite main Config using MollyFile
@ -259,7 +260,7 @@ func handleRedirectsInner(URL *url.URL, redirects map[string]string, status int,
} }
} }
func serveDirectory(URL *url.URL, path string, log *LogEntry, conn net.Conn, config Config, errorLogEntries chan string) { func serveDirectory(URL *url.URL, path string, log *LogEntry, conn net.Conn, config Config, errorLog *log.Logger) {
// Redirect to add trailing slash if missing // Redirect to add trailing slash if missing
// (otherwise relative links don't work properly) // (otherwise relative links don't work properly)
if !strings.HasSuffix(URL.Path, "/") { if !strings.HasSuffix(URL.Path, "/") {
@ -271,7 +272,7 @@ func serveDirectory(URL *url.URL, path string, log *LogEntry, conn net.Conn, con
index_path := filepath.Join(path, "index."+config.GeminiExt) index_path := filepath.Join(path, "index."+config.GeminiExt)
index_info, err := os.Stat(index_path) index_info, err := os.Stat(index_path)
if err == nil && uint64(index_info.Mode().Perm())&0444 == 0444 { if err == nil && uint64(index_info.Mode().Perm())&0444 == 0444 {
serveFile(index_path, log, conn, config, errorLogEntries) serveFile(index_path, log, conn, config, errorLog)
// Serve a generated listing // Serve a generated listing
} else { } else {
conn.Write([]byte("20 text/gemini\r\n")) conn.Write([]byte("20 text/gemini\r\n"))
@ -280,7 +281,7 @@ func serveDirectory(URL *url.URL, path string, log *LogEntry, conn net.Conn, con
} }
} }
func serveFile(path string, log *LogEntry, conn net.Conn, config Config, errorLog chan string) { func serveFile(path string, log *LogEntry, conn net.Conn, config Config, errorLog *log.Logger) {
// Get MIME type of files // Get MIME type of files
ext := filepath.Ext(path) ext := filepath.Ext(path)
var mimeType string var mimeType string

16
main.go
View file

@ -6,7 +6,6 @@ import (
"log" "log"
"os" "os"
"strconv" "strconv"
"time"
) )
func main() { func main() {
@ -32,8 +31,11 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
defer errorLogFile.Close() defer errorLogFile.Close()
errorLog := log.New(errorLogFile, "", log.Ldate | log.Ltime)
accessLogFile, err := os.OpenFile(config.AccessLog, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) accessLogFile, err := os.OpenFile(config.AccessLog, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil { if err != nil {
errorLog.Println("Error opening access log file: " + err.Error())
log.Fatal(err) log.Fatal(err)
} }
defer accessLogFile.Close() defer accessLogFile.Close()
@ -41,6 +43,7 @@ func main() {
// Read TLS files, create TLS config // Read TLS files, create TLS config
cert, err := tls.LoadX509KeyPair(config.CertPath, config.KeyPath) cert, err := tls.LoadX509KeyPair(config.CertPath, config.KeyPath)
if err != nil { if err != nil {
errorLog.Println("Error loading TLS keypair: " + err.Error())
log.Fatal(err) log.Fatal(err)
} }
tlscfg := &tls.Config{ tlscfg := &tls.Config{
@ -52,6 +55,7 @@ func main() {
// Create TLS listener // Create TLS listener
listener, err := tls.Listen("tcp", ":"+strconv.Itoa(config.Port), tlscfg) listener, err := tls.Listen("tcp", ":"+strconv.Itoa(config.Port), tlscfg)
if err != nil { if err != nil {
errorLog.Println("Error creating TLS listener: " + err.Error())
log.Fatal(err) log.Fatal(err)
} }
defer listener.Close() defer listener.Close()
@ -64,21 +68,15 @@ func main() {
writeLogEntry(accessLogFile, entry) writeLogEntry(accessLogFile, entry)
} }
}() }()
errorLogEntries := make(chan string, 10)
go func() {
for {
message := <-errorLogEntries
errorLogFile.WriteString(time.Now().Format(time.RFC3339) + " " + message + "\n")
}
}()
// Infinite serve loop // Infinite serve loop
for { for {
conn, err := listener.Accept() conn, err := listener.Accept()
if err != nil { if err != nil {
errorLog.Println("Error accepting connection: " + err.Error())
log.Fatal(err) log.Fatal(err)
} }
go handleGeminiRequest(conn, config, accessLogEntries, errorLogEntries) go handleGeminiRequest(conn, config, accessLogEntries, errorLog)
} }
} }