fix: Change kmip14/op_locate to be an array of strings

Signed-off-by: Joe Skazinski <[email protected]>
Seagate · Sep 16, 2022 · fc514dd · fc514dd
1 parent a3b47d6
commit fc514dd
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 5 deletions.
diff --git a/op_locate.go b/op_locate.go
@@ -19,8 +19,8 @@ type LocateRequestPayload struct {
 // Table 191
 
 type LocateResponsePayload struct {
-	LocatedItems     uint32 // Required: No
-	UniqueIdentifier string // Required: No
+	LocatedItems     uint32   // Required: No
+	UniqueIdentifier []string // Required: No
 }
 
 type LocateHandler struct {

diff --git a/src/kmipapi/kmip14.go b/src/kmipapi/kmip14.go
@@ -301,9 +301,16 @@ func (kmips *kmip14service) Locate(ctx context.Context, settings *ConfigurationS
 		return nil, fmt.Errorf("unable to decode GetResponsePayload, error: %v", err)
 	}
 
-	uid := respPayload.UniqueIdentifier
+	logger.V(4).Info("XXX Locate response payload", "respPayload", respPayload)
+
+	uids := respPayload.UniqueIdentifier
 	logger.V(4).Info("XXX Locate response payload", "uid", respPayload.UniqueIdentifier)
 
+	uid := ""
+	if len(uids) > 0 {
+		uid = uids[0]
+	}
+
 	return &LocateResponse{UniqueIdentifier: uid}, nil
 }
 

diff --git a/src/kmipapi/kmip20.go b/src/kmipapi/kmip20.go
@@ -333,10 +333,15 @@ func (kmips *kmip20service) Locate(ctx context.Context, settings *ConfigurationS
 		return nil, fmt.Errorf("unable to decode GetResponsePayload, error: %v", err)
 	}
 
-	uid := respPayload.UniqueIdentifier
+	uids := respPayload.UniqueIdentifier
 	logger.V(4).Info("XXX Locate response payload", "uid", respPayload.UniqueIdentifier)
 
-	return &LocateResponse{UniqueIdentifier: uid[0]}, nil
+	uid := ""
+	if len(uids) > 0 {
+		uid = uids[0]
+	}
+
+	return &LocateResponse{UniqueIdentifier: uid}, nil
 }
 
 // SetAttribute: