handle error from osuser.Current (microsoft#503)

shueybubbles · web-flow · commit a02c6d7e548f · 2024-01-25T08:57:43.000-06:00
diff --git a/pkg/sqlcmd/sqlcmd.go b/pkg/sqlcmd/sqlcmd.go
@@ -295,10 +295,10 @@ func (s *Sqlcmd) ConnectDb(connect *ConnectSettings, nopw bool) error {
 		s.vars.Set(SQLCMDUSER, connect.UserName)
 	} else {
 		u, e := osuser.Current()
-		if e != nil {
-			panic("Unable to get user name")
+		// osuser.Current() returns an error in some restricted environments
+		if e == nil {
+			s.vars.Set(SQLCMDUSER, u.Username)
 		}
-		s.vars.Set(SQLCMDUSER, u.Username)
 	}
 	if newConnection {
 		s.Connect = connect

Original file line number	Diff line number	Diff line change
`@@ -295,10 +295,10 @@ func (s Sqlcmd) ConnectDb(connect ConnectSettings, nopw bool) error {`
`295`	`295`	`s.vars.Set(SQLCMDUSER, connect.UserName)`
`296`	`296`	`} else {`
`297`	`297`	`u, e := osuser.Current()`
`298`		`- if e != nil {`
`299`		`- panic("Unable to get user name")`
	`298`	`+ // osuser.Current() returns an error in some restricted environments`
	`299`	`+ if e == nil {`
	`300`	`+ s.vars.Set(SQLCMDUSER, u.Username)`
`300`	`301`	`}`
`301`		`- s.vars.Set(SQLCMDUSER, u.Username)`
`302`	`302`	`}`
`303`	`303`	`if newConnection {`
`304`	`304`	`s.Connect = connect`