From 65a62b05369a2dbb75d26971dc071b28b1119331 Mon Sep 17 00:00:00 2001 From: nixonrodrigues <nixon@apache.org> Date: Fri, 6 Dec 2019 14:49:21 +0530 Subject: [PATCH] ATLAS-3550 :- Add authorize check for partialUpdateEntityAttrByGuid API call --- repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java index bc4cc8b..ea5e6ab 100644 --- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java +++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java @@ -376,6 +376,8 @@ public class AtlasEntityStoreV2 implements AtlasEntityStore { entity.setGuid(guid); + AtlasAuthorizationUtils.verifyAccess(new AtlasEntityAccessRequest(typeRegistry, AtlasPrivilege.ENTITY_UPDATE, new AtlasEntityHeader(entity)), "update entity ByUniqueAttributes"); + return createOrUpdate(new AtlasEntityStream(updatedEntityInfo), true, false); } @@ -391,6 +393,8 @@ public class AtlasEntityStoreV2 implements AtlasEntityStore { AtlasEntityType entityType = (AtlasEntityType) typeRegistry.getType(entity.getTypeName()); AtlasAttribute attr = entityType.getAttribute(attrName); + AtlasAuthorizationUtils.verifyAccess(new AtlasEntityAccessRequest(typeRegistry, AtlasPrivilege.ENTITY_UPDATE, entity), "update entity ByUniqueAttributes : guid=", guid ); + if (attr == null) { attr = entityType.getRelationshipAttribute(attrName, AtlasEntityUtil.getRelationshipType(attrValue)); -- libgit2 0.27.1