kill: add kill -a option

Adds the -a parameter to the kill runv command, allowing to
kill all processes in the container.

Signed-off-by: Antonios Motakis <antonios.motakis@huawei.com>

Author: tvelocity

kill.go

@@ -69,6 +69,12 @@ For example, if the container id is "ubuntu01" the following will send a "KILL"
 signal to the init process of the "ubuntu01" container:
 
        # runv kill ubuntu01 KILL`,
+	Flags: []cli.Flag{
+		cli.BoolFlag{
+			Name:  "all, a",
+			Usage: "send the signal to all processes in the container",
+		},
+	},
 	Action: func(context *cli.Context) error {
 		container := context.Args().First()
 		if container == "" {
@@ -88,17 +94,50 @@ signal to the init process of the "ubuntu01" container:
 		if err != nil {
 			return cli.NewExitError(fmt.Sprintf("failed to get client: %v", err), -1)
 		}
-		if _, err = c.Signal(netcontext.Background(), &types.SignalRequest{
-			Id:     container,
-			Pid:    "init",
-			Signal: uint32(signal),
-		}); err != nil {
-			return cli.NewExitError(fmt.Sprintf("kill signal failed, %v", err), -1)
+
+		plist := make([]string, 0)
+
+		if context.Bool("all") {
+			if plist, err = getProcessList(c, container); err != nil {
+				return cli.NewExitError(fmt.Sprintf("can't get process list, %v", err), -1)
+			}
+		} else {
+			plist = append(plist, "init")
+		}
+
+		for _, p := range plist {
+			if _, err = c.Signal(netcontext.Background(), &types.SignalRequest{
+				Id:     container,
+				Pid:    p,
+				Signal: uint32(signal),
+			}); err != nil {
+				return cli.NewExitError(fmt.Sprintf("kill signal failed, %v", err), -1)
+			}
 		}
 		return nil
 	},
 }
 
+func getProcessList(c types.APIClient, container string) ([]string, error) {
+	s, err := c.State(netcontext.Background(), &types.StateRequest{Id: container})
+	if err != nil {
+		return nil, fmt.Errorf("get container state failed, %v", err)
+	}
+
+	for _, cc := range s.Containers {
+		if cc.Id == container {
+			plist := make([]string, 0)
+			for _, p := range cc.Processes {
+				plist = append(plist, p.Pid)
+			}
+
+			return plist, nil
+		}
+	}
+
+	return nil, fmt.Errorf("container %s not found", container)
+}
+
 func parseSignal(rawSignal string) (syscall.Signal, error) {
 	s, err := strconv.Atoi(rawSignal)
 	if err == nil {

supervisor/process.go

@@ -93,8 +93,7 @@ func (p *Process) signal(sig int) error {
 		// TODO: change vm.KillContainer()
 		return p.ownerCont.ownerPod.vm.KillContainer(p.ownerCont.Id, syscall.Signal(sig))
 	} else {
-		// TODO support it
-		return fmt.Errorf("Kill to non-init process of container is unsupported")
+		return p.ownerCont.ownerPod.vm.SignalProcess(p.ownerCont.Id, p.Id, syscall.Signal(sig))
 	}
 }