view multipass-checkpassword.go @ 15:9b4ec6b5c23e

Add tests for multipass files.
author Paul Fisher <paul@pfish.zone>
date Thu, 29 Oct 2015 23:56:53 -0400
parents 4db389f948c9
children
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("USER", user.Username)
	os.Setenv("HOME", user.HomeDir)
	os.Setenv("userdb_uid", user.Uid)
	os.Setenv("userdb_gid", user.Gid)
	os.Setenv("EXTRA", "userdb_uid userdb_gid")
	syscall.Exec(os.Args[1], os.Args[1:], os.Environ())
}