view multipass-checkpassword.go @ 7:406d3cd76739

Report errors to the user.
author Paul Fisher <paul@pfish.zone>
date Sat, 24 Oct 2015 23:40:13 -0400
parents 10613f0f4737
children 4db389f948c9
line wrap: on
line source

package main

import (
	"bufio"
	"os"
	"os/user"
	"syscall"

	"pfish.zone/go/multipass/file"
)

const (
	InternalError = 111
	Failed        = 1
)

func main() {
	infile := os.NewFile(3, "")
	reader := bufio.NewReader(infile)
	username, err := reader.ReadString(0)
	if err != nil {
		os.Exit(InternalError)
	}
	username = username[:len(username)-1]
	pass, err := reader.ReadString(0)
	if err != nil {
		os.Exit(InternalError)
	}
	pass = pass[:len(pass)-1]
	infile.Close()
	passfile, err := file.ForUser(username)
	if err != nil {
		os.Exit(Failed)
	}
	success, _ := passfile.Authenticate(pass)
	if !success {
		os.Exit(Failed)
	}
	user, err := user.Lookup(username)
	if err != nil {
		os.Exit(Failed)
	}
	os.Setenv("userdb_uid", user.Uid)
	os.Setenv("userdb_gid", user.Gid)
	os.Setenv("EXTRA", "userdb_uid userdb_gid")
	environ := []string{
		"USER=" + user.Username,
		"userdb_uid=" + user.Uid,
		"userdb_gid=" + user.Gid,
		"HOME=" + user.HomeDir,
		"EXTRA=userdb_uid userdb_gid",
	}
	syscall.Exec(os.Args[1], os.Args[1:], environ)
}