From d84884f26b85fe9aee9655b65d0603aeb5c33362 Mon Sep 17 00:00:00 2001
From: Milan Nikolic <gen2brain@gmail.com>
Date: Sun, 15 Jun 2025 08:34:19 +0200
Subject: [PATCH] Update server

---
 handlers/jpeg.go | 2 ++
 server/server.go | 6 +++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/handlers/jpeg.go b/handlers/jpeg.go
index 06f1fb2..67c24c3 100644
--- a/handlers/jpeg.go
+++ b/handlers/jpeg.go
@@ -33,6 +33,7 @@ func (j *JPEG) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	img, err := j.reader.Read()
 	if err != nil {
 		log.Printf("jpeg: read: %v", err)
+		http.Error(w, "500 Internal Server Error", http.StatusInternalServerError)
 
 		return
 	}
@@ -40,6 +41,7 @@ func (j *JPEG) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	err = image.NewEncoder(w, j.quality).Encode(img)
 	if err != nil {
 		log.Printf("jpeg: encode: %v", err)
+		http.Error(w, "500 Internal Server Error", http.StatusInternalServerError)
 
 		return
 	}
diff --git a/server/server.go b/server/server.go
index 94d7ea6..a940937 100644
--- a/server/server.go
+++ b/server/server.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net"
 	"net/http"
+	"time"
 
 	"github.com/abbot/go-http-auth"
 
@@ -63,7 +64,10 @@ func (s *Server) ListenAndServe() error {
 
 	http.Handle("/", newAuthHandler(handlers.NewIndex(), basic))
 
-	srv := &http.Server{}
+	srv := &http.Server{
+		ReadTimeout:  5 * time.Second,
+		WriteTimeout: 5 * time.Second,
+	}
 
 	listener, err := net.Listen("tcp", s.Bind)
 	if err != nil {