veyron/security/agent/agentd: support the veyron.credentials flag
Change-Id: If50764814dd1f05b7c1d9f6ad574db97f477d8dd
diff --git a/security/agent/agentd/main.go b/security/agent/agentd/main.go
index 03a97ff..e15ba81 100644
--- a/security/agent/agentd/main.go
+++ b/security/agent/agentd/main.go
@@ -65,9 +65,7 @@
return
}
- // TODO(ashankar,cnicolaou): Should flags.Parse be used instead? But that adds unnecessary
- // flags like "--veyron.namespace.root", which has no meaning for this binary.
- dir := os.Getenv(consts.VeyronCredentials)
+ dir := flag.Lookup("veyron.credentials").Value.String()
if len(dir) == 0 {
vlog.Fatalf("The %v environment variable must be set to a directory", consts.VeyronCredentials)
}
