Lindenii Project Forge
Login
Commit info
IDc35b685daf560d7462af5c2332959f6d79fd52ce
AuthorRunxi Yu<me@runxiyu.org>
Author dateMon, 17 Feb 2025 13:54:39 +0800
CommitterRunxi Yu<me@runxiyu.org>
Committer dateMon, 17 Feb 2025 13:54:39 +0800
Actions
Get patch
git_hooks_handle.go: Exit on UID mismatch, handle >1 connections
package main

import (
	"errors"
	"fmt"
	"net"
	"os"
	"syscall"
)

var err_not_unixconn = errors.New("Not a unix connection")

func hooks_handle_connection(conn net.Conn) (err error) {
func hooks_handle_connection(conn net.Conn) {
	defer conn.Close()

	unix_conn, ok := conn.(*net.UnixConn)
	if !ok {
		return err_not_unixconn
	}
	unix_conn := conn.(*net.UnixConn)

	fd, err := unix_conn.File()
	if err != nil {
		return err
		conn.Write([]byte{1})
		fmt.Fprintln(conn, "Unable to get file descriptor")
		return
	}
	defer fd.Close()

	ucred, err := get_ucred(fd)
	if err != nil {
		return err
		conn.Write([]byte{1})
		fmt.Fprintln(conn, "Unable to get peer credentials")
		return
	}

	pid := ucred.Pid
	if ucred.Uid != uint32(os.Getuid()) {
		conn.Write([]byte{1})
		fmt.Fprintln(conn, "UID mismatch")
		return
	}

	conn.Write([]byte{0})
	fmt.Fprintf(conn, "your PID is %d\n", pid)
	fmt.Fprintf(conn, "Your PID is %d\n", ucred.Pid)

	return nil
	return
}

func serve_git_hooks(listener net.Listener) error {
	conn, err := listener.Accept()
	if err != nil {
		return err
	for {
		conn, err := listener.Accept()
		if err != nil {
			return err
		}
		go hooks_handle_connection(conn)
	}

	return hooks_handle_connection(conn)
}

func get_ucred(fd *os.File) (*syscall.Ucred, error) {
	ucred, err := syscall.GetsockoptUcred(int(fd.Fd()), syscall.SOL_SOCKET, syscall.SO_PEERCRED)
	if err != nil {
		return nil, fmt.Errorf("failed to get credentials: %v", err)
	}
	return ucred, nil
}