Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
8431b6adf5
|
|||
24e923fe84
|
|||
10ddc7c190
|
142
cmdext/cmdrunner.go
Normal file
142
cmdext/cmdrunner.go
Normal file
@@ -0,0 +1,142 @@
|
|||||||
|
package cmdext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"os/exec"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CommandResult struct {
|
||||||
|
StdOut string
|
||||||
|
StdErr string
|
||||||
|
StdCombined string
|
||||||
|
ExitCode int
|
||||||
|
CommandTimedOut bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func RunCommand(program string, args []string, timeout *time.Duration) (CommandResult, error) {
|
||||||
|
|
||||||
|
cmd := exec.Command(program, args...)
|
||||||
|
|
||||||
|
stdoutPipe, err := cmd.StdoutPipe()
|
||||||
|
if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
stderrPipe, err := cmd.StderrPipe()
|
||||||
|
if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = cmd.Start()
|
||||||
|
if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
errch := make(chan error, 1)
|
||||||
|
go func() { errch <- cmd.Wait() }()
|
||||||
|
|
||||||
|
combch := make(chan string, 32)
|
||||||
|
stopCombch := make(chan bool)
|
||||||
|
|
||||||
|
stdout := ""
|
||||||
|
go func() {
|
||||||
|
scanner := bufio.NewScanner(stdoutPipe)
|
||||||
|
for scanner.Scan() {
|
||||||
|
txt := scanner.Text()
|
||||||
|
stdout += txt
|
||||||
|
combch <- txt
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
stderr := ""
|
||||||
|
go func() {
|
||||||
|
scanner := bufio.NewScanner(stderrPipe)
|
||||||
|
for scanner.Scan() {
|
||||||
|
txt := scanner.Text()
|
||||||
|
stderr += txt
|
||||||
|
combch <- txt
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
stopCombch <- true
|
||||||
|
}()
|
||||||
|
|
||||||
|
stdcombined := ""
|
||||||
|
go func() {
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case txt := <-combch:
|
||||||
|
stdcombined += txt
|
||||||
|
case <-stopCombch:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
if timeout != nil {
|
||||||
|
|
||||||
|
select {
|
||||||
|
|
||||||
|
case <-time.After(*timeout):
|
||||||
|
_ = cmd.Process.Kill()
|
||||||
|
return CommandResult{
|
||||||
|
StdOut: stdout,
|
||||||
|
StdErr: stderr,
|
||||||
|
StdCombined: stdcombined,
|
||||||
|
ExitCode: -1,
|
||||||
|
CommandTimedOut: true,
|
||||||
|
}, nil
|
||||||
|
|
||||||
|
case err := <-errch:
|
||||||
|
if exiterr, ok := err.(*exec.ExitError); ok {
|
||||||
|
return CommandResult{
|
||||||
|
StdOut: stdout,
|
||||||
|
StdErr: stderr,
|
||||||
|
StdCombined: stdcombined,
|
||||||
|
ExitCode: exiterr.ExitCode(),
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}, nil
|
||||||
|
} else if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
} else {
|
||||||
|
return CommandResult{
|
||||||
|
StdOut: stdout,
|
||||||
|
StdErr: stderr,
|
||||||
|
StdCombined: stdcombined,
|
||||||
|
ExitCode: 0,
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
select {
|
||||||
|
|
||||||
|
case err := <-errch:
|
||||||
|
if exiterr, ok := err.(*exec.ExitError); ok {
|
||||||
|
return CommandResult{
|
||||||
|
StdOut: stdout,
|
||||||
|
StdErr: stderr,
|
||||||
|
StdCombined: stdcombined,
|
||||||
|
ExitCode: exiterr.ExitCode(),
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}, nil
|
||||||
|
} else if err != nil {
|
||||||
|
return CommandResult{}, err
|
||||||
|
} else {
|
||||||
|
return CommandResult{
|
||||||
|
StdOut: stdout,
|
||||||
|
StdErr: stderr,
|
||||||
|
StdCombined: stdcombined,
|
||||||
|
ExitCode: 0,
|
||||||
|
CommandTimedOut: false,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@@ -150,9 +150,9 @@ func (ph PassHash) Verify(plainpass string, totp *string) bool {
|
|||||||
|
|
||||||
if version == 4 {
|
if version == 4 {
|
||||||
if !hastotp {
|
if !hastotp {
|
||||||
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) != nil
|
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) == nil
|
||||||
} else {
|
} else {
|
||||||
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) != nil && totpext.Validate(totpsecret, *totp)
|
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) == nil && totpext.Validate(totpsecret, *totp)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@@ -50,7 +50,6 @@ func ScanSingle[TData any](rows *sqlx.Rows, mode StructScanMode, sec StructScanS
|
|||||||
return *new(TData), err
|
return *new(TData), err
|
||||||
}
|
}
|
||||||
} else if mode == SModeExtended {
|
} else if mode == SModeExtended {
|
||||||
var data TData
|
|
||||||
err := strscan.StructScanExt(&data)
|
err := strscan.StructScanExt(&data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return *new(TData), err
|
return *new(TData), err
|
||||||
|
@@ -113,7 +113,7 @@ func (r *StructScanner) StructScanExt(dest any) error {
|
|||||||
|
|
||||||
if _, ok := forcenulled[k]; !ok {
|
if _, ok := forcenulled[k]; !ok {
|
||||||
f := reflectx.FieldByIndexes(v, traversal[0:i])
|
f := reflectx.FieldByIndexes(v, traversal[0:i])
|
||||||
f.Set(reflect.New(f.Type().Elem())) // set to nil
|
f.Set(reflect.Zero(f.Type())) // set to nil
|
||||||
forcenulled[k] = true
|
forcenulled[k] = true
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -138,7 +138,7 @@ func (r *StructScanner) StructScanExt(dest any) error {
|
|||||||
return errors.New(fmt.Sprintf("Cannot set field %v to NULL value from column '%s' (type: %s)", traversal, r.columns[i], f.Type().String()))
|
return errors.New(fmt.Sprintf("Cannot set field %v to NULL value from column '%s' (type: %s)", traversal, r.columns[i], f.Type().String()))
|
||||||
}
|
}
|
||||||
|
|
||||||
f.Set(reflect.New(f.Type().Elem())) // set to nil
|
f.Set(reflect.Zero(f.Type())) // set to nil
|
||||||
} else {
|
} else {
|
||||||
f.Set(val3)
|
f.Set(val3)
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user