mirror of
https://github.com/containers/podman
synced 2024-10-20 17:23:30 +00:00
Restrict top output to container's pids only
Due to the way ps arguments work, it was possible to display pids that dont below to the container in top output. We now filter pids that dont belong to the container out of the output. This also means the pid column must be present in the output or we throw an error. This resolves issue #391 Signed-off-by: baude <bbaude@redhat.com> Closes: #400 Approved by: rhatdan
This commit is contained in:
parent
f47a5be60d
commit
6cb1c31d3f
|
@ -70,11 +70,11 @@ func topCmd(c *cli.Context) error {
|
|||
|
||||
psArgs = append(psArgs, psOpts...)
|
||||
|
||||
results, err := container.GetContainerPidInformation(psArgs)
|
||||
psOutput, err := container.GetContainerPidInformation(psArgs)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, line := range results {
|
||||
for _, line := range psOutput {
|
||||
fmt.Println(line)
|
||||
}
|
||||
return nil
|
||||
|
|
|
@ -38,7 +38,7 @@ func (c *Container) getContainerPids() ([]string, error) {
|
|||
}
|
||||
|
||||
// GetContainerPidInformation calls ps with the appropriate options and returns
|
||||
// the results as a string
|
||||
// the results as a string and the container's PIDs as a []string
|
||||
func (c *Container) GetContainerPidInformation(args []string) ([]string, error) {
|
||||
if !c.locked {
|
||||
c.lock.Lock()
|
||||
|
@ -57,5 +57,55 @@ func (c *Container) GetContainerPidInformation(args []string) ([]string, error)
|
|||
if err != nil {
|
||||
return []string{}, errors.Wrapf(err, "unable to obtain information about pids")
|
||||
}
|
||||
return strings.Split(results, "\n"), nil
|
||||
|
||||
filteredOutput, err := filterPids(results, pids)
|
||||
if err != nil {
|
||||
return []string{}, err
|
||||
}
|
||||
return filteredOutput, nil
|
||||
}
|
||||
|
||||
func filterPids(psOutput string, pids []string) ([]string, error) {
|
||||
var output []string
|
||||
results := strings.Split(psOutput, "\n")
|
||||
// The headers are in the first line of the results
|
||||
headers := fieldsASCII(results[0])
|
||||
// We need to make sure PID in headers, so that we can filter
|
||||
// Pids that don't belong.
|
||||
|
||||
// append the headers back in
|
||||
output = append(output, results[0])
|
||||
|
||||
pidIndex := -1
|
||||
for i, header := range headers {
|
||||
if header == "PID" {
|
||||
pidIndex = i
|
||||
}
|
||||
}
|
||||
if pidIndex == -1 {
|
||||
return []string{}, errors.Errorf("unable to find PID field in ps output. try a different set of ps arguments")
|
||||
}
|
||||
for _, l := range results[1:] {
|
||||
if l == "" {
|
||||
continue
|
||||
}
|
||||
cols := fieldsASCII(l)
|
||||
pid := cols[pidIndex]
|
||||
if StringInSlice(pid, pids) {
|
||||
output = append(output, l)
|
||||
}
|
||||
}
|
||||
return output, nil
|
||||
}
|
||||
|
||||
// Detects ascii whitespaces
|
||||
func fieldsASCII(s string) []string {
|
||||
fn := func(r rune) bool {
|
||||
switch r {
|
||||
case '\t', '\n', '\f', '\r', ' ':
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
return strings.FieldsFunc(s, fn)
|
||||
}
|
||||
|
|
|
@ -59,14 +59,26 @@ var _ = Describe("Podman top", func() {
|
|||
Expect(len(result.OutputToStringArray())).To(BeNumerically(">", 1))
|
||||
})
|
||||
|
||||
It("podman top on non-running container", func() {
|
||||
It("podman top with options", func() {
|
||||
session := podmanTest.Podman([]string{"run", "-d", ALPINE, "top", "-d", "2"})
|
||||
session.WaitWithDefaultTimeout()
|
||||
Expect(session.ExitCode()).To(Equal(0))
|
||||
|
||||
result := podmanTest.Podman([]string{"top", session.OutputToString(), "-o", "fuser,f,comm,label"})
|
||||
result := podmanTest.Podman([]string{"top", session.OutputToString(), "-o", "pid,fuser,f,comm,label"})
|
||||
result.WaitWithDefaultTimeout()
|
||||
Expect(result.ExitCode()).To(Equal(0))
|
||||
Expect(len(result.OutputToStringArray())).To(BeNumerically(">", 1))
|
||||
})
|
||||
|
||||
It("podman top on container invalid options", func() {
|
||||
sleep := podmanTest.RunSleepContainer("")
|
||||
sleep.WaitWithDefaultTimeout()
|
||||
Expect(sleep.ExitCode()).To(Equal(0))
|
||||
cid := sleep.OutputToString()
|
||||
|
||||
result := podmanTest.Podman([]string{"top", cid, "-o time"})
|
||||
result.WaitWithDefaultTimeout()
|
||||
Expect(result.ExitCode()).To(Equal(125))
|
||||
})
|
||||
|
||||
})
|
||||
|
|
Loading…
Reference in a new issue