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