diff options
| -rw-r--r-- | git_hooks_handle.go | 48 |
1 files changed, 36 insertions, 12 deletions
diff --git a/git_hooks_handle.go b/git_hooks_handle.go index 841c8d7..00e2627 100644 --- a/git_hooks_handle.go +++ b/git_hooks_handle.go @@ -27,12 +27,16 @@ func hooks_handle_connection(conn net.Conn) { // another user. ucred, err := get_ucred(conn) if err != nil { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Unable to get peer credentials:", err.Error()) return } if ucred.Uid != uint32(os.Getuid()) { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "UID mismatch") return } @@ -40,14 +44,18 @@ func hooks_handle_connection(conn net.Conn) { cookie := make([]byte, 64) _, err = conn.Read(cookie) if err != nil { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Failed to read cookie:", err.Error()) return } pack_to_hook, ok := pack_to_hook_by_cookie.Load(string(cookie)) if !ok { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Invalid handler cookie") return } @@ -55,7 +63,9 @@ func hooks_handle_connection(conn net.Conn) { var argc64 uint64 err = binary.Read(conn, binary.NativeEndian, &argc64) if err != nil { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Failed to read argc:", err.Error()) return } @@ -66,7 +76,9 @@ func hooks_handle_connection(conn net.Conn) { b := make([]byte, 1) n, err := conn.Read(b) if err != nil || n != 1 { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Failed to read arg:", err.Error()) return } @@ -87,7 +99,9 @@ func hooks_handle_connection(conn net.Conn) { switch filepath.Base(args[0]) { case "pre-receive": if pack_to_hook.direct_access { - conn.Write([]byte{0}) + if _, err := conn.Write([]byte{0}); err != nil { + return + } } else { ref_ok := make(map[string]uint8) // 0 for ok @@ -103,14 +117,18 @@ func hooks_handle_connection(conn net.Conn) { old_oid, rest, found := strings.Cut(line, " ") if !found { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Invalid pre-receive line:", line) break } new_oid, ref_name, found := strings.Cut(rest, " ") if !found { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Invalid pre-receive line:", line) break } @@ -129,10 +147,14 @@ func hooks_handle_connection(conn net.Conn) { } if or_all_in_map(ref_ok) == 0 { - conn.Write([]byte{0}) + if _, err := conn.Write([]byte{0}); err != nil { + return + } fmt.Fprintln(conn, "Stuff") } else { - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } for ref, status := range ref_ok { switch status { case 0: @@ -148,7 +170,9 @@ func hooks_handle_connection(conn net.Conn) { } } default: - conn.Write([]byte{1}) + if _, err := conn.Write([]byte{1}); err != nil { + return + } fmt.Fprintln(conn, "Invalid hook:", args[0]) } } |