diff --git a/go.mod b/go.mod index 118961a3..2ef988c9 100644 --- a/go.mod +++ b/go.mod @@ -15,6 +15,7 @@ replace ( k8s.io/component-base => k8s.io/component-base v0.26.4 sigs.k8s.io/kustomize/api => sigs.k8s.io/kustomize/api v0.12.1 sigs.k8s.io/kustomize/kyaml => sigs.k8s.io/kustomize/kyaml v0.13.9 + github.com/kubearmor/KubeArmor/protobuf => /home/vagrant/xyz3/KubeArmor/protobuf/ ) require ( diff --git a/probe/probe.go b/probe/probe.go index 8101c8ab..f1b99fe8 100644 --- a/probe/probe.go +++ b/probe/probe.go @@ -25,6 +25,8 @@ import ( tp "github.com/kubearmor/KubeArmor/KubeArmor/types" "github.com/kubearmor/kubearmor-client/deployment" "github.com/kubearmor/kubearmor-client/k8s" + "google.golang.org/grpc" + "google.golang.org/protobuf/types/known/emptypb" "golang.org/x/exp/slices" "golang.org/x/mod/semver" @@ -35,6 +37,7 @@ import ( "errors" + pb "github.com/kubearmor/KubeArmor/protobuf" "golang.org/x/sys/unix" ) @@ -565,9 +568,29 @@ func probeSystemdMode() error { return err } printKubeArmorProbeOutput(kd) + + err = printContainers() + return nil } +func printContainers() error { + gRPC := "" + conn, err := grpc.Dial(gRPC, grpc.WithInsecure()) + if err != nil { + return err + } + + client := pb.NewKarmorClient(conn) + + resp, err := client.GetKarmorData(context.Background(), *emptypb.Empty) + + fmt.Println(resp.Containers) + + return err + +} + func getAnnotatedPodLabels(m map[string]string) mapset.Set[string] { var a []string for key, value := range m {