Address all security review comments: sanitize remaining error messages and handle errors properly

Co-authored-by: krau <71133316+krau@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-01-19 04:50:48 +00:00
co-authored by krau
parent 9dcb5201e1
commit 30c165033e
2 changed files with 16 additions and 7 deletions
+10 -6
View File
@@ -4,7 +4,6 @@ import (
"bytes" "bytes"
"context" "context"
"encoding/json" "encoding/json"
"fmt"
"io" "io"
"net/http" "net/http"
"path" "path"
@@ -171,7 +170,7 @@ func handleCreateTask(w http.ResponseWriter, r *http.Request) {
}) })
if err != nil { if err != nil {
logger.Errorf("Failed to create task: %v", err) logger.Errorf("Failed to create task: %v", err)
respondError(w, fmt.Sprintf("failed to create task: %v", err), http.StatusInternalServerError) respondError(w, "failed to create task", http.StatusInternalServerError)
return return
} }
@@ -182,7 +181,7 @@ func handleCreateTask(w http.ResponseWriter, r *http.Request) {
if err := core.AddTask(injectCtx, task); err != nil { if err := core.AddTask(injectCtx, task); err != nil {
logger.Errorf("Failed to add task: %v", err) logger.Errorf("Failed to add task: %v", err)
updateTaskStatus(taskID, "failed", err.Error()) updateTaskStatus(taskID, "failed", err.Error())
respondError(w, fmt.Sprintf("failed to add task: %v", err), http.StatusInternalServerError) respondError(w, "failed to add task to queue", http.StatusInternalServerError)
return return
} }
@@ -242,7 +241,8 @@ func handleCancelTask(w http.ResponseWriter, r *http.Request) {
} }
if err := core.CancelTask(r.Context(), taskID); err != nil { if err := core.CancelTask(r.Context(), taskID); err != nil {
respondError(w, fmt.Sprintf("failed to cancel task: %v", err), http.StatusInternalServerError) log.FromContext(r.Context()).Errorf("Failed to cancel task %s: %v", taskID, err)
respondError(w, "failed to cancel task", http.StatusInternalServerError)
return return
} }
@@ -365,7 +365,11 @@ func sendWebhook(taskID, status, errorMsg string) {
defer resp.Body.Close() defer resp.Body.Close()
if resp.StatusCode >= 400 { if resp.StatusCode >= 400 {
body, _ := io.ReadAll(resp.Body) body, err := io.ReadAll(resp.Body)
log.Errorf("Webhook returned error status %d: %s", resp.StatusCode, string(body)) if err != nil {
log.Errorf("Webhook returned error status %d, failed to read response body: %v", resp.StatusCode, err)
} else {
log.Errorf("Webhook returned error status %d: %s", resp.StatusCode, string(body))
}
} }
} }
+6 -1
View File
@@ -91,7 +91,12 @@ func getClientIP(r *http.Request) string {
} }
// Fall back to RemoteAddr // Fall back to RemoteAddr
ip, _, _ := net.SplitHostPort(r.RemoteAddr) ip, _, err := net.SplitHostPort(r.RemoteAddr)
if err != nil {
// If SplitHostPort fails, RemoteAddr might not have a port
// In this case, just return RemoteAddr as is
return r.RemoteAddr
}
return ip return ip
} }