Write username to file

Author: SeanThomasWilliams

config.go

@@ -9,6 +9,7 @@ import (
 
 const defaultConfigFilePath = "/etc/https-user-management/config.yaml"
 const defaultTokenFilePath = "/etc/https-user-management/user.token"
+const defaultUserFilePath = "/etc/https-user-management/user.name"
 
 var AppConfig *Config
 var ConfigError bool
@@ -22,6 +23,7 @@ type Config struct {
 	Debug     bool
 	URL       string
 	TokenFile string
+	UserFile  string
 }
 
 func NewConfig(configPath string) (*Config, error) {
@@ -67,6 +69,10 @@ func init() {
 		AppConfig.TokenFile = defaultTokenFilePath
 	}
 
+	if AppConfig.UserFile == "" {
+		AppConfig.UserFile = defaultUserFilePath
+	}
+
 	logger.Infof("AppConfig: %+v", *AppConfig)
 
 	if AppConfig.Debug {

pam-https/main.go

@@ -27,6 +27,11 @@ func (mp *mypam) Authenticate(hdl pam.Handle, args pam.Args) pam.Value {
 	humcommon.Log().Infof("Got request for user: %v", user)
 	humcommon.Log().Debugf("Got request for user: %v", user)
 
+	if err := writeUserFile(user); err != nil {
+		humcommon.Log().Warnf("Error writing user file: %v", err)
+		return pam.AuthInfoUnavailable
+	}
+
 	userPassword, err := hdl.GetItem(pam.AuthToken)
 	if err != nil {
 		humcommon.Log().Warnf("Error getting PAM passwd for user: %v", err)
@@ -82,6 +87,13 @@ func writeTokenFile(token string) error {
 	return nil
 }
 
+func writeUserFile(user string) error {
+	if _, err := os.Stat(humcommon.AppConfig.UserFile); os.IsNotExist(err) {
+		return ioutil.WriteFile(humcommon.AppConfig.UserFile, []byte(user), 0644)
+	}
+	return nil
+}
+
 func fileContains(line []byte, filePath string) (bool, error) {
 	data, err := ioutil.ReadFile(filePath)
 	if err != nil {