support header flags

minio · Feb 11, 2025 · c8eed15 · c8eed15
1 parent 10c5036
commit c8eed15
Show file tree

Hide file tree

Showing 192 changed files with 578 additions and 461 deletions.
diff --git a/cmd/admin-accesskey-list.go b/cmd/admin-accesskey-list.go
@@ -141,7 +141,7 @@ func mainAdminAccesskeyList(ctx *cli.Context) error {
 	aliasedURL, tentativeAll, users, opts := commonAccesskeyList(ctx)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	accessKeysMap, e := client.ListAccessKeysBulk(globalContext, users, opts)

diff --git a/cmd/admin-cluster-bucket-export.go b/cmd/admin-cluster-bucket-export.go
@@ -73,7 +73,7 @@ func mainClusterBucketExport(ctx *cli.Context) error {
 	console.SetColor("File", color.New(color.FgWhite, color.Bold))
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil

diff --git a/cmd/admin-cluster-bucket-import.go b/cmd/admin-cluster-bucket-import.go
@@ -98,7 +98,7 @@ func mainClusterBucketImport(ctx *cli.Context) error {
 	fatalIf(probe.NewError(e).Trace(args...), "Unable to get bucket metadata")
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil

diff --git a/cmd/admin-cluster-iam-export.go b/cmd/admin-cluster-iam-export.go
@@ -87,7 +87,7 @@ func mainClusterIAMExport(ctx *cli.Context) error {
 	console.SetColor("File", color.New(color.FgWhite, color.Bold))
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil

diff --git a/cmd/admin-cluster-iam-import.go b/cmd/admin-cluster-iam-import.go
@@ -209,7 +209,7 @@ func mainClusterIAMImport(ctx *cli.Context) error {
 	fatalIf(probe.NewError(e).Trace(args...), "Unable to get IAM info")
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil

diff --git a/cmd/admin-config-export.go b/cmd/admin-config-export.go
@@ -109,7 +109,7 @@ func mainAdminConfigExport(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	// Call get config API

diff --git a/cmd/admin-config-get.go b/cmd/admin-config-get.go
@@ -123,7 +123,7 @@ func mainAdminConfigGet(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	if len(ctx.Args()) == 1 {

diff --git a/cmd/admin-config-history.go b/cmd/admin-config-history.go
@@ -124,7 +124,7 @@ func mainAdminConfigHistory(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	if ctx.Bool("clear") {

diff --git a/cmd/admin-config-import.go b/cmd/admin-config-import.go
@@ -93,7 +93,7 @@ func mainAdminConfigImport(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	// Call set config API

diff --git a/cmd/admin-config-reset.go b/cmd/admin-config-reset.go
@@ -111,7 +111,7 @@ func mainAdminConfigReset(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	if len(ctx.Args()) == 1 {

diff --git a/cmd/admin-config-restore.go b/cmd/admin-config-restore.go
@@ -91,7 +91,7 @@ func mainAdminConfigRestore(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	// Call get config API

diff --git a/cmd/admin-config-set.go b/cmd/admin-config-set.go
@@ -105,7 +105,7 @@ func mainAdminConfigSet(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	input := strings.Join(args.Tail(), " ")

diff --git a/cmd/admin-decom-cancel.go b/cmd/admin-decom-cancel.go
@@ -70,7 +70,7 @@ func mainAdminDecommissionCancel(ctx *cli.Context) error {
 	aliasedURL = filepath.Clean(aliasedURL)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	if pool := args.Get(1); pool != "" {

diff --git a/cmd/admin-decom-start.go b/cmd/admin-decom-start.go
@@ -88,7 +88,7 @@ func mainAdminDecommissionStart(ctx *cli.Context) error {
 	aliasedURL = filepath.Clean(aliasedURL)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	e := client.DecommissionPool(globalContext, args.Get(1))

diff --git a/cmd/admin-decom-status.go b/cmd/admin-decom-status.go
@@ -71,7 +71,7 @@ func mainAdminDecommissionStatus(ctx *cli.Context) error {
 	aliasedURL = filepath.Clean(aliasedURL)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	if pool := args.Get(1); pool != "" {

diff --git a/cmd/admin-group-add.go b/cmd/admin-group-add.go
@@ -125,7 +125,7 @@ func mainAdminGroupAdd(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	members := []string{}

diff --git a/cmd/admin-group-enable.go b/cmd/admin-group-enable.go
@@ -65,7 +65,7 @@ func mainAdminGroupEnableDisable(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	group := args.Get(1)

diff --git a/cmd/admin-group-info.go b/cmd/admin-group-info.go
@@ -64,7 +64,7 @@ func mainAdminGroupInfo(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	group := args.Get(1)

diff --git a/cmd/admin-group-list.go b/cmd/admin-group-list.go
@@ -65,7 +65,7 @@ func mainAdminGroupList(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	gs, e := client.ListGroups(globalContext)

diff --git a/cmd/admin-group-remove.go b/cmd/admin-group-remove.go
@@ -69,7 +69,7 @@ func mainAdminGroupRemove(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	members := []string{}

diff --git a/cmd/admin-heal.go b/cmd/admin-heal.go
@@ -669,7 +669,7 @@ func mainAdminHeal(ctx *cli.Context) error {
 	console.SetColor("NodeFailed", color.New(color.FgRed, color.Bold))
 
 	// Create a new MinIO Admin Client
-	adminClnt, err := newAdminClient(aliasedURL)
+	adminClnt, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil
@@ -680,7 +680,7 @@ func mainAdminHeal(ctx *cli.Context) error {
 	splits := splitStr(aliasedURL, "/", 3)
 	bucket, prefix := splits[1], splits[2]
 
-	clnt, err := newClient(aliasedURL)
+	clnt, err := newClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(clnt.GetURL().String()), "Unable to create client for URL ", aliasedURL)
 		return nil

diff --git a/cmd/admin-info.go b/cmd/admin-info.go
@@ -384,7 +384,7 @@ func mainAdminInfo(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	clusterInfo := clusterStruct{

diff --git a/cmd/admin-kms-key-create.go b/cmd/admin-kms-key-create.go
@@ -56,7 +56,7 @@ func mainAdminKMSCreateKey(ctx *cli.Context) error {
 		showCommandHelpAndExit(ctx, 1) // last argument is exit code
 	}
 
-	client, err := newAdminClient(ctx.Args().Get(0))
+	client, err := newAdminClient(newClientParams{urlStr: ctx.Args().Get(0), customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Cannot get a configured admin connection.")
 
 	keyID := ctx.Args().Get(1)

diff --git a/cmd/admin-kms-key-list.go b/cmd/admin-kms-key-list.go
@@ -65,7 +65,7 @@ func mainAdminKMSKeyList(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	keys, e := client.ListKeys(globalContext, "*")

diff --git a/cmd/admin-kms-key-status.go b/cmd/admin-kms-key-status.go
@@ -61,7 +61,7 @@ func mainAdminKMSKeyStatus(ctx *cli.Context) error {
 	console.SetColor("StatusError", color.New(color.FgRed, color.Bold))
 	console.SetColor("StatusUnknown", color.New(color.FgYellow, color.Bold))
 
-	client, err := newAdminClient(ctx.Args().Get(0))
+	client, err := newAdminClient(newClientParams{urlStr: ctx.Args().Get(0), customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to get a configured admin connection.")
 
 	var keyID string

diff --git a/cmd/admin-logs.go b/cmd/admin-logs.go
@@ -190,7 +190,7 @@ func mainAdminLogs(ctx *cli.Context) error {
 		fatalIf(errInvalidArgument().Trace(ctx.Args()...), "Invalid value for --type flag. Valid options are [minio, application, all]")
 	}
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client.")
 		return nil

diff --git a/cmd/admin-policy-attach.go b/cmd/admin-policy-attach.go
@@ -91,7 +91,7 @@ func userAttachOrDetachPolicy(ctx *cli.Context, attach bool) error {
 	}
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	var e error

diff --git a/cmd/admin-policy-create.go b/cmd/admin-policy-create.go
@@ -130,7 +130,7 @@ func mainAdminPolicyCreate(ctx *cli.Context) error {
 	fatalIf(probe.NewError(e).Trace(args...), "Unable to get policy")
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	fatalIf(probe.NewError(client.AddCannedPolicy(globalContext, args.Get(1), policy)).Trace(args...), "Unable to create new policy")

diff --git a/cmd/admin-policy-entities.go b/cmd/admin-policy-entities.go
@@ -84,7 +84,7 @@ func mainAdminPolicyEntities(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	res, e := client.GetPolicyEntities(globalContext,

diff --git a/cmd/admin-policy-info.go b/cmd/admin-policy-info.go
@@ -100,7 +100,7 @@ func mainAdminPolicyInfo(ctx *cli.Context) error {
 	policyName := args.Get(1)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection")
 
 	pinfo, e := getPolicyInfo(client, policyName)

diff --git a/cmd/admin-policy-list.go b/cmd/admin-policy-list.go
@@ -65,7 +65,7 @@ func mainAdminPolicyList(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	policies, e := client.ListCannedPolicies(globalContext)

diff --git a/cmd/admin-policy-remove.go b/cmd/admin-policy-remove.go
@@ -68,7 +68,7 @@ func mainAdminPolicyRemove(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	fatalIf(probe.NewError(client.RemoveCannedPolicy(globalContext, args.Get(1))).Trace(args...), "Unable to remove policy")

diff --git a/cmd/admin-rebalance-start.go b/cmd/admin-rebalance-start.go
@@ -77,7 +77,7 @@ func mainAdminRebalanceStart(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client")
 
 	id, e := client.RebalanceStart(globalContext)

diff --git a/cmd/admin-rebalance-status.go b/cmd/admin-rebalance-status.go
@@ -60,7 +60,7 @@ func mainAdminRebalanceStatus(ctx *cli.Context) error {
 	args := ctx.Args()
 	aliasedURL := args.Get(0)
 
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	if err != nil {
 		fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client")
 		return err.ToGoError()

diff --git a/cmd/admin-rebalance-stop.go b/cmd/admin-rebalance-stop.go
@@ -76,7 +76,7 @@ func mainAdminRebalanceStop(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err.Trace(aliasedURL), "Unable to initialize admin client")
 
 	fatalIf(probe.NewError(client.RebalanceStop(globalContext)), "Unable to stop rebalance operation")

diff --git a/cmd/admin-replicate-add.go b/cmd/admin-replicate-add.go
@@ -99,12 +99,12 @@ func mainAdminReplicateAdd(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	ps := make([]madmin.PeerSite, 0, len(ctx.Args()))
 	for _, clusterName := range ctx.Args() {
-		admClient, err := newAdminClient(clusterName)
+		admClient, err := newAdminClient(newClientParams{urlStr: clusterName, customHeader: getCustomHeader(ctx)})
 		fatalIf(err, "unable to initialize admin connection")
 
 		ak, sk := admClient.GetAccessAndSecretKey()

diff --git a/cmd/admin-replicate-info.go b/cmd/admin-replicate-info.go
@@ -136,7 +136,7 @@ func mainAdminReplicationInfo(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	info, e := client.SiteReplicationInfo(globalContext)

diff --git a/cmd/admin-replicate-remove.go b/cmd/admin-replicate-remove.go
@@ -120,7 +120,7 @@ func mainAdminReplicationRemoveStatus(ctx *cli.Context) error {
 	rreq.SiteNames = append(rreq.SiteNames, args.Tail()...)
 	rreq.RemoveAll = ctx.Bool("all")
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 
 	st, e := client.SiteReplicationRemove(globalContext, rreq)

diff --git a/cmd/admin-replicate-resync-cancel.go b/cmd/admin-replicate-resync-cancel.go
@@ -88,12 +88,12 @@ func mainAdminReplicateResyncCancel(ctx *cli.Context) error {
 	aliasedURL := args.Get(0)
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(aliasedURL)
+	client, err := newAdminClient(newClientParams{urlStr: aliasedURL, customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 	info, e := client.SiteReplicationInfo(globalContext)
 	fatalIf(probe.NewError(e), "Unable to fetch site replication info.")
 
-	peerClient := getClient(args.Get(1))
+	peerClient := getClient(args.Get(1), getCustomHeader(ctx))
 	peerAdmInfo, e := peerClient.ServerInfo(globalContext)
 	fatalIf(probe.NewError(e), "Unable to fetch server info of the peer.")
 

diff --git a/cmd/admin-replicate-resync-start.go b/cmd/admin-replicate-resync-start.go
@@ -89,12 +89,12 @@ func mainAdminReplicateResyncStart(ctx *cli.Context) error {
 	args := ctx.Args()
 
 	// Create a new MinIO Admin Client
-	client, err := newAdminClient(args.Get(0))
+	client, err := newAdminClient(newClientParams{urlStr: args.Get(0), customHeader: getCustomHeader(ctx)})
 	fatalIf(err, "Unable to initialize admin connection.")
 	info, e := client.SiteReplicationInfo(globalContext)
 	fatalIf(probe.NewError(e), "Unable to fetch site replication info.")
 
-	peerClient := getClient(args.Get(1))
+	peerClient := getClient(args.Get(1), getCustomHeader(ctx))
 	peerAdmInfo, e := peerClient.ServerInfo(globalContext)
 	fatalIf(probe.NewError(e), "Unable to fetch server info of the peer.")