go-asink/asinkd/rpc_server.go

package main

import (
	"asink"
	"net"
	"net/http"
	"net/rpc"
)

type UserModifier struct {
	adb *AsinkDB
}

type UserModifierArgs struct {
	Current        *User
	Updated        *User
	UpdateLogin    bool
	UpdateRole     bool
	UpdatePassword bool
}

func (u *UserModifier) AddUser(user *User, result *int) error {
	err := u.adb.DatabaseAddUser(user)
	if err != nil {
		*result = 1
	} else {
		*result = 0
	}
	return err
}

func (u *UserModifier) ModifyUser(args *UserModifierArgs, result *int) error {
	currentUser, err := u.adb.DatabaseGetUser(args.Current.Username)
	if err != nil {
		*result = 1
		return err
	}

	if args.UpdateLogin {
		currentUser.Username = args.Updated.Username
	}
	if args.UpdateRole {
		currentUser.Role = args.Updated.Role
	}
	if args.UpdatePassword {
		currentUser.PWHash = args.Updated.PWHash
	}

	err = u.adb.DatabaseUpdateUser(currentUser)
	if err != nil {
		*result = 1
		return err
	}

	*result = 0
	return nil
}

func (u *UserModifier) RemoveUser(user *User, result *int) error {
	err := u.adb.DatabaseDeleteUser(user)
	if err != nil {
		*result = 1
	} else {
		*result = 0
	}
	return err
}

type ServerStopper int

func (s *ServerStopper) StopServer(code *int, result *int) error {
	asink.Exit(*code)
	*result = 0
	return nil
}

func StartRPC(sock string, tornDown chan int, adb *AsinkDB) {
	defer func() { tornDown <- 0 }() //the main thread waits for this to ensure the socket is closed

	usermod := new(UserModifier)
	usermod.adb = adb
	rpc.Register(usermod)

	serverstop := new(ServerStopper)
	rpc.Register(serverstop)

	rpc.HandleHTTP()
	l, err := net.Listen("unix", sock)
	if err != nil {
		panic(err)
	}
	defer l.Close()

	go http.Serve(l, nil)

	asink.WaitOnExit()
}