HADOOP-14274. Azure: Simplify Ranger-WASB policy model. Contributed by Sivaguru Sankaridurg

This commit is contained in:
Mingliang Liu 2017-04-12 16:07:10 -07:00
parent b053fdc547
commit 0cab57223e
7 changed files with 606 additions and 127 deletions

View File

@ -1426,13 +1426,20 @@ public class NativeAzureFileSystem extends FileSystem {
return store; return store;
} }
private void performAuthCheck(String path, String accessType, /**
String operation) throws WasbAuthorizationException, IOException { * @param requestingAccessForPath - The path to the ancestor/parent/subtree/file that needs to be
* checked before granting access to originalPath
* @param accessType - The type of access READ/WRITE being requested
* @param operation - A string describing the operation being performed ("delete", "create" etc.).
* @param originalPath - The originalPath that was being accessed
*/
private void performAuthCheck(String requestingAccessForPath, WasbAuthorizationOperations accessType,
String operation, String originalPath) throws WasbAuthorizationException, IOException {
if (azureAuthorization && this.authorizer != null && if (azureAuthorization && this.authorizer != null &&
!this.authorizer.authorize(path, accessType)) { !this.authorizer.authorize(requestingAccessForPath, accessType.toString())) {
throw new WasbAuthorizationException(operation throw new WasbAuthorizationException(operation
+ " operation for Path : " + path + " not allowed"); + " operation for Path : " + originalPath + " not allowed");
} }
} }
@ -1459,8 +1466,7 @@ public class NativeAzureFileSystem extends FileSystem {
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
performAuthCheck(absolutePath.toString(), performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.WRITE, "append", absolutePath.toString());
WasbAuthorizationOperations.WRITE.toString(), "append");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
FileMetadata meta = null; FileMetadata meta = null;
@ -1663,9 +1669,9 @@ public class NativeAzureFileSystem extends FileSystem {
} }
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
Path ancestor = getAncestor(absolutePath);
performAuthCheck(absolutePath.toString(), performAuthCheck(ancestor.toString(), WasbAuthorizationOperations.WRITE, "create", absolutePath.toString());
WasbAuthorizationOperations.WRITE.toString(), "create");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
@ -1678,6 +1684,9 @@ public class NativeAzureFileSystem extends FileSystem {
if (!overwrite) { if (!overwrite) {
throw new FileAlreadyExistsException("File already exists:" + f); throw new FileAlreadyExistsException("File already exists:" + f);
} }
else {
performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.WRITE, "create", absolutePath.toString());
}
} }
Path parentFolder = absolutePath.getParent(); Path parentFolder = absolutePath.getParent();
@ -1768,7 +1777,7 @@ public class NativeAzureFileSystem extends FileSystem {
/** /**
* Delete the specified file or folder. The parameter * Delete the specified file or folder. The parameter
* skipParentFolderLastModifidedTimeUpdate * skipParentFolderLastModifiedTimeUpdate
* is used in the case of atomic folder rename redo. In that case, there is * is used in the case of atomic folder rename redo. In that case, there is
* a lease on the parent folder, so (without reworking the code) modifying * a lease on the parent folder, so (without reworking the code) modifying
* the parent folder update time will fail because of a conflict with the * the parent folder update time will fail because of a conflict with the
@ -1778,20 +1787,20 @@ public class NativeAzureFileSystem extends FileSystem {
* *
* @param f file path to be deleted. * @param f file path to be deleted.
* @param recursive specify deleting recursively or not. * @param recursive specify deleting recursively or not.
* @param skipParentFolderLastModifidedTimeUpdate If true, don't update the folder last * @param skipParentFolderLastModifiedTimeUpdate If true, don't update the folder last
* modified time. * modified time.
* @return true if and only if the file is deleted * @return true if and only if the file is deleted
* @throws IOException Thrown when fail to delete file or directory. * @throws IOException Thrown when fail to delete file or directory.
*/ */
public boolean delete(Path f, boolean recursive, public boolean delete(Path f, boolean recursive,
boolean skipParentFolderLastModifidedTimeUpdate) throws IOException { boolean skipParentFolderLastModifiedTimeUpdate) throws IOException {
LOG.debug("Deleting file: {}", f.toString()); LOG.debug("Deleting file: {}", f.toString());
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
Path parentPath = absolutePath.getParent();
performAuthCheck(absolutePath.toString(), performAuthCheck(parentPath.toString(), WasbAuthorizationOperations.WRITE, "delete", absolutePath.toString());
WasbAuthorizationOperations.EXECUTE.toString(), "delete");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
@ -1827,7 +1836,7 @@ public class NativeAzureFileSystem extends FileSystem {
// (e.g. the blob store only contains the blob a/b and there's no // (e.g. the blob store only contains the blob a/b and there's no
// corresponding directory blob a) and that would implicitly delete // corresponding directory blob a) and that would implicitly delete
// the directory as well, which is not correct. // the directory as well, which is not correct.
Path parentPath = absolutePath.getParent();
if (parentPath.getParent() != null) {// Not root if (parentPath.getParent() != null) {// Not root
String parentKey = pathToKey(parentPath); String parentKey = pathToKey(parentPath);
@ -1876,7 +1885,7 @@ public class NativeAzureFileSystem extends FileSystem {
store.storeEmptyFolder(parentKey, store.storeEmptyFolder(parentKey,
createPermissionStatus(FsPermission.getDefault())); createPermissionStatus(FsPermission.getDefault()));
} else { } else {
if (!skipParentFolderLastModifidedTimeUpdate) { if (!skipParentFolderLastModifiedTimeUpdate) {
updateParentFolderLastModifiedTime(key); updateParentFolderLastModifiedTime(key);
} }
} }
@ -1903,7 +1912,6 @@ public class NativeAzureFileSystem extends FileSystem {
// The path specifies a folder. Recursively delete all entries under the // The path specifies a folder. Recursively delete all entries under the
// folder. // folder.
LOG.debug("Directory Delete encountered: {}", f.toString()); LOG.debug("Directory Delete encountered: {}", f.toString());
Path parentPath = absolutePath.getParent();
if (parentPath.getParent() != null) { if (parentPath.getParent() != null) {
String parentKey = pathToKey(parentPath); String parentKey = pathToKey(parentPath);
FileMetadata parentMetadata = null; FileMetadata parentMetadata = null;
@ -1981,12 +1989,30 @@ public class NativeAzureFileSystem extends FileSystem {
final FileMetadata[] contents = fileMetadataList.toArray(new FileMetadata[fileMetadataList.size()]); final FileMetadata[] contents = fileMetadataList.toArray(new FileMetadata[fileMetadataList.size()]);
if (!recursive && contents.length > 0) { if (contents.length > 0) {
if (!recursive) {
// The folder is non-empty and recursive delete was not specified. // The folder is non-empty and recursive delete was not specified.
// Throw an exception indicating that a non-recursive delete was // Throw an exception indicating that a non-recursive delete was
// specified for a non-empty folder. // specified for a non-empty folder.
throw new IOException("Non-recursive delete of non-empty directory " throw new IOException("Non-recursive delete of non-empty directory "
+ f.toString()); + f.toString());
}
else {
// Check write-permissions on sub-tree including current folder
// NOTE: Ideally the subtree needs read-write-execute access check.
// But we will simplify it to write-access check.
if (metaFile.isDir()) { // the absolute-path
performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.WRITE, "delete",
absolutePath.toString());
}
for (FileMetadata meta : contents) {
if (meta.isDir()) {
Path subTreeDir = keyToPath(meta.getKey());
performAuthCheck(subTreeDir.toString(), WasbAuthorizationOperations.WRITE, "delete",
absolutePath.toString());
}
}
}
} }
// Delete all files / folders in current directory stored as list in 'contents'. // Delete all files / folders in current directory stored as list in 'contents'.
@ -2014,7 +2040,7 @@ public class NativeAzureFileSystem extends FileSystem {
// Update parent directory last modified time // Update parent directory last modified time
Path parent = absolutePath.getParent(); Path parent = absolutePath.getParent();
if (parent != null && parent.getParent() != null) { // not root if (parent != null && parent.getParent() != null) { // not root
if (!skipParentFolderLastModifidedTimeUpdate) { if (!skipParentFolderLastModifiedTimeUpdate) {
updateParentFolderLastModifiedTime(key); updateParentFolderLastModifiedTime(key);
} }
} }
@ -2064,8 +2090,8 @@ public class NativeAzureFileSystem extends FileSystem {
// Capture the absolute path and the path to key. // Capture the absolute path and the path to key.
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
performAuthCheck(absolutePath.toString(), performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.READ, "getFileStatus",
WasbAuthorizationOperations.EXECUTE.toString(), "getFileStatus"); absolutePath.toString());
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
if (key.length() == 0) { // root always exists if (key.length() == 0) { // root always exists
@ -2166,8 +2192,7 @@ public class NativeAzureFileSystem extends FileSystem {
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
performAuthCheck(absolutePath.toString(), performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.READ, "liststatus", absolutePath.toString());
WasbAuthorizationOperations.EXECUTE.toString(), "list");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
Set<FileStatus> status = new TreeSet<FileStatus>(); Set<FileStatus> status = new TreeSet<FileStatus>();
@ -2375,6 +2400,24 @@ public class NativeAzureFileSystem extends FileSystem {
permission); permission);
} }
private Path getAncestor(Path f) throws IOException {
for (Path current = f.getParent(), parent = current.getParent();
parent != null; // Stop when you get to the root
current = parent, parent = current.getParent()) {
String currentKey = pathToKey(current);
FileMetadata currentMetadata = store.retrieveMetadata(currentKey);
if (currentMetadata != null) {
Path ancestor = keyToPath(currentMetadata.getKey());
LOG.debug("Found ancestor {}, for path: {}", ancestor.toString(), f.toString());
return ancestor;
}
}
return new Path("/");
}
@Override @Override
public boolean mkdirs(Path f, FsPermission permission) throws IOException { public boolean mkdirs(Path f, FsPermission permission) throws IOException {
return mkdirs(f, permission, false); return mkdirs(f, permission, false);
@ -2391,9 +2434,9 @@ public class NativeAzureFileSystem extends FileSystem {
} }
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
Path ancestor = getAncestor(absolutePath);
performAuthCheck(absolutePath.toString(), performAuthCheck(ancestor.toString(), WasbAuthorizationOperations.WRITE, "mkdirs", absolutePath.toString());
WasbAuthorizationOperations.EXECUTE.toString(), "mkdirs");
PermissionStatus permissionStatus = null; PermissionStatus permissionStatus = null;
if(noUmask) { if(noUmask) {
@ -2449,8 +2492,7 @@ public class NativeAzureFileSystem extends FileSystem {
Path absolutePath = makeAbsolute(f); Path absolutePath = makeAbsolute(f);
performAuthCheck(absolutePath.toString(), performAuthCheck(absolutePath.toString(), WasbAuthorizationOperations.READ, "read", absolutePath.toString());
WasbAuthorizationOperations.READ.toString(), "read");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
FileMetadata meta = null; FileMetadata meta = null;
@ -2508,12 +2550,18 @@ public class NativeAzureFileSystem extends FileSystem {
+ " through WASB that has colons in the name"); + " through WASB that has colons in the name");
} }
Path absolutePath = makeAbsolute(src); Path absoluteSrcPath = makeAbsolute(src);
Path srcParentFolder = absoluteSrcPath.getParent();
performAuthCheck(absolutePath.toString(), if (srcParentFolder == null) {
WasbAuthorizationOperations.EXECUTE.toString(), "rename"); // Cannot rename root of file system
return false;
}
String srcKey = pathToKey(absolutePath); performAuthCheck(srcParentFolder.toString(), WasbAuthorizationOperations.WRITE, "rename",
absoluteSrcPath.toString());
String srcKey = pathToKey(absoluteSrcPath);
if (srcKey.length() == 0) { if (srcKey.length() == 0) {
// Cannot rename root of file system // Cannot rename root of file system
@ -2521,8 +2569,13 @@ public class NativeAzureFileSystem extends FileSystem {
} }
// Figure out the final destination // Figure out the final destination
Path absoluteDst = makeAbsolute(dst); Path absoluteDstPath = makeAbsolute(dst);
String dstKey = pathToKey(absoluteDst); Path dstParentFolder = absoluteDstPath.getParent();
performAuthCheck(dstParentFolder.toString(), WasbAuthorizationOperations.WRITE, "rename",
absoluteDstPath.toString());
String dstKey = pathToKey(absoluteDstPath);
FileMetadata dstMetadata = null; FileMetadata dstMetadata = null;
try { try {
dstMetadata = store.retrieveMetadata(dstKey); dstMetadata = store.retrieveMetadata(dstKey);
@ -2530,14 +2583,14 @@ public class NativeAzureFileSystem extends FileSystem {
Throwable innerException = NativeAzureFileSystemHelper.checkForAzureStorageException(ex); Throwable innerException = NativeAzureFileSystemHelper.checkForAzureStorageException(ex);
// A BlobNotFound storage exception in only thrown from retrieveMetdata API when // A BlobNotFound storage exception in only thrown from retrieveMetadata API when
// there is a race condition. If there is another thread which deletes the destination // there is a race condition. If there is another thread which deletes the destination
// file or folder, then this thread calling rename should be able to continue with // file or folder, then this thread calling rename should be able to continue with
// rename gracefully. Hence the StorageException is swallowed here. // rename gracefully. Hence the StorageException is swallowed here.
if (innerException instanceof StorageException) { if (innerException instanceof StorageException) {
if (NativeAzureFileSystemHelper.isFileNotFoundException((StorageException) innerException)) { if (NativeAzureFileSystemHelper.isFileNotFoundException((StorageException) innerException)) {
LOG.debug("BlobNotFound exception encountered for Destination key : {}. " LOG.debug("BlobNotFound exception encountered for Destination key : {}. "
+ "Swallowin the exception to handle race condition gracefully", dstKey); + "Swallowing the exception to handle race condition gracefully", dstKey);
} }
} else { } else {
throw ex; throw ex;
@ -2558,7 +2611,7 @@ public class NativeAzureFileSystem extends FileSystem {
// Check that the parent directory exists. // Check that the parent directory exists.
FileMetadata parentOfDestMetadata = null; FileMetadata parentOfDestMetadata = null;
try { try {
parentOfDestMetadata = store.retrieveMetadata(pathToKey(absoluteDst.getParent())); parentOfDestMetadata = store.retrieveMetadata(pathToKey(absoluteDstPath.getParent()));
} catch (IOException ex) { } catch (IOException ex) {
Throwable innerException = NativeAzureFileSystemHelper.checkForAzureStorageException(ex); Throwable innerException = NativeAzureFileSystemHelper.checkForAzureStorageException(ex);
@ -2816,9 +2869,6 @@ public class NativeAzureFileSystem extends FileSystem {
public void setPermission(Path p, FsPermission permission) throws FileNotFoundException, IOException { public void setPermission(Path p, FsPermission permission) throws FileNotFoundException, IOException {
Path absolutePath = makeAbsolute(p); Path absolutePath = makeAbsolute(p);
performAuthCheck(absolutePath.toString(),
WasbAuthorizationOperations.EXECUTE.toString(), "setPermission");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
FileMetadata metadata = null; FileMetadata metadata = null;
try { try {
@ -2858,9 +2908,6 @@ public class NativeAzureFileSystem extends FileSystem {
throws IOException { throws IOException {
Path absolutePath = makeAbsolute(p); Path absolutePath = makeAbsolute(p);
performAuthCheck(absolutePath.toString(),
WasbAuthorizationOperations.EXECUTE.toString(), "setOwner");
String key = pathToKey(absolutePath); String key = pathToKey(absolutePath);
FileMetadata metadata = null; FileMetadata metadata = null;

View File

@ -132,7 +132,14 @@ public class RemoteWasbAuthorizerImpl implements WasbAuthorizerInterface {
@Override @Override
public boolean authorize(String wasbAbsolutePath, String accessType) public boolean authorize(String wasbAbsolutePath, String accessType)
throws WasbAuthorizationException, IOException { throws WasbAuthorizationException, IOException {
try { try {
/* Make an exception for the internal -RenamePending files */
if (wasbAbsolutePath.endsWith(NativeAzureFileSystem.FolderRenamePending.SUFFIX)) {
return true;
}
URIBuilder uriBuilder = new URIBuilder(remoteAuthorizerServiceUrl); URIBuilder uriBuilder = new URIBuilder(remoteAuthorizerServiceUrl);
uriBuilder.setPath("/" + CHECK_AUTHORIZATION_OP); uriBuilder.setPath("/" + CHECK_AUTHORIZATION_OP);
uriBuilder.addParameter(WASB_ABSOLUTE_PATH_QUERY_PARAM_NAME, uriBuilder.addParameter(WASB_ABSOLUTE_PATH_QUERY_PARAM_NAME,
@ -203,7 +210,7 @@ public class RemoteWasbAuthorizerImpl implements WasbAuthorizerInterface {
return authorizerResponse.getAuthorizationResult(); return authorizerResponse.getAuthorizationResult();
} else { } else {
throw new WasbAuthorizationException("Remote authorization" throw new WasbAuthorizationException("Remote authorization"
+ " serivce encountered an error " + " service encountered an error "
+ authorizerResponse.getResponseMessage()); + authorizerResponse.getResponseMessage());
} }
} catch (URISyntaxException | WasbRemoteCallException } catch (URISyntaxException | WasbRemoteCallException
@ -220,7 +227,7 @@ public class RemoteWasbAuthorizerImpl implements WasbAuthorizerInterface {
* response in the following JSON format * response in the following JSON format
* { * {
* "responseCode" : 0 or non-zero <int>, * "responseCode" : 0 or non-zero <int>,
* "responseMessage" : relavant message of failure <String> * "responseMessage" : relevant message of failure <String>
* "authorizationResult" : authorization result <boolean> * "authorizationResult" : authorization result <boolean>
* true - if auhorization allowed * true - if auhorization allowed
* false - otherwise. * false - otherwise.

View File

@ -34,8 +34,6 @@ public enum WasbAuthorizationOperations {
return "read"; return "read";
case WRITE: case WRITE:
return "write"; return "write";
case EXECUTE:
return "execute";
default: default:
throw new IllegalArgumentException( throw new IllegalArgumentException(
"Invalid Authorization Operation"); "Invalid Authorization Operation");

View File

@ -20,6 +20,7 @@ package org.apache.hadoop.fs.azure;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.regex.Pattern;
import org.apache.hadoop.conf.Configuration; import org.apache.hadoop.conf.Configuration;
@ -38,8 +39,11 @@ public class MockWasbAuthorizerImpl implements WasbAuthorizerInterface {
public void addAuthRule(String wasbAbsolutePath, public void addAuthRule(String wasbAbsolutePath,
String accessType, boolean access) { String accessType, boolean access) {
AuthorizationComponent component =
new AuthorizationComponent(wasbAbsolutePath, accessType); AuthorizationComponent component = wasbAbsolutePath.endsWith("*")
? new AuthorizationComponent("^" + wasbAbsolutePath.replace("*", ".*"), accessType)
: new AuthorizationComponent(wasbAbsolutePath, accessType);
this.authRules.put(component, access); this.authRules.put(component, access);
} }
@ -47,12 +51,26 @@ public class MockWasbAuthorizerImpl implements WasbAuthorizerInterface {
public boolean authorize(String wasbAbsolutePath, String accessType) public boolean authorize(String wasbAbsolutePath, String accessType)
throws WasbAuthorizationException { throws WasbAuthorizationException {
if (wasbAbsolutePath.endsWith(NativeAzureFileSystem.FolderRenamePending.SUFFIX)) {
return true;
}
AuthorizationComponent component = AuthorizationComponent component =
new AuthorizationComponent(wasbAbsolutePath, accessType); new AuthorizationComponent(wasbAbsolutePath, accessType);
if (authRules.containsKey(component)) { if (authRules.containsKey(component)) {
return authRules.get(component); return authRules.get(component);
} else { } else {
// Regex-pattern match if we don't have a straight match
for (Map.Entry<AuthorizationComponent, Boolean> entry : authRules.entrySet()) {
AuthorizationComponent key = entry.getKey();
String keyPath = key.getWasbAbsolutePath();
String keyAccess = key.getAccessType();
if (keyPath.endsWith("*") && Pattern.matches(keyPath, wasbAbsolutePath) && keyAccess.equals(accessType)) {
return entry.getValue();
}
}
return false; return false;
} }
} }

View File

@ -28,11 +28,8 @@ import org.junit.Before;
import org.junit.Rule; import org.junit.Rule;
import org.junit.Test; import org.junit.Test;
import com.sun.tools.javac.util.Assert;
import org.junit.rules.ExpectedException; import org.junit.rules.ExpectedException;
import java.io.Console;
import static org.apache.hadoop.fs.azure.AzureNativeFileSystemStore.KEY_USE_SECURE_MODE; import static org.apache.hadoop.fs.azure.AzureNativeFileSystemStore.KEY_USE_SECURE_MODE;
/** /**
@ -67,32 +64,154 @@ public class TestNativeAzureFileSystemAuthorization
public ExpectedException expectedEx = ExpectedException.none(); public ExpectedException expectedEx = ExpectedException.none();
/** /**
* Positive test to verify Create and delete access check * Setup up permissions to allow a recursive delete for cleanup purposes.
*/
private void allowRecursiveDelete(NativeAzureFileSystem fs, MockWasbAuthorizerImpl authorizer, String path) {
int index = path.lastIndexOf('/');
String parent = (index == 0) ? "/" : path.substring(0, index);
authorizer.init(null);
authorizer.addAuthRule(parent, WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule((path.endsWith("*") ? path : path+"*"), WasbAuthorizationOperations.WRITE.toString(), true);
fs.updateWasbAuthorizer(authorizer);
}
/**
* Positive test to verify Create access check
* The file is created directly under an existing folder.
* No intermediate folders need to be created.
* @throws Throwable * @throws Throwable
*/ */
@Test @Test
public void testCreateAccessCheckPositive() throws Throwable { public void testCreateAccessWithoutCreateIntermediateFoldersCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount(); AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem(); NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testCreateAccessCheckPositive"); Path parentDir = new Path("/");
Path testPath = new Path(parentDir, "test.dat"); Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
fs.create(testPath); try {
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath); fs.create(testPath);
fs.delete(parentDir, true); ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
}
finally {
fs.delete(testPath, false);
}
} }
/** /**
* Negative test to verify Create access check * Positive test to verify Create access check
* The test tries to create a file whose parent is non-existent to ensure that
* the intermediate folders between ancestor and direct parent are being created
* when proper ranger policies are configured.
* @throws Throwable
*/
@Test
public void testCreateAccessWithCreateIntermediateFoldersCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testCreateAccessCheckPositive/1/2/3");
Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
}
finally {
allowRecursiveDelete(fs, authorizer, "/testCreateAccessCheckPositive");
fs.delete(new Path("/testCreateAccessCheckPositive"), true);
}
}
/**
* Negative test to verify that create fails when trying to overwrite an existing file
* without proper write permissions on the file being overwritten.
* @throws Throwable
*/
@Test // (expected=WasbAuthorizationException.class)
public void testCreateAccessWithOverwriteCheckNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("create operation for Path : /test.dat not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/");
Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
boolean initialCreateSucceeded = false;
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
initialCreateSucceeded = true;
fs.create(testPath, true);
}
finally {
ContractTestUtils.assertTrue(initialCreateSucceeded);
fs.delete(testPath, false);
}
}
/**
* Positive test to verify that create succeeds when trying to overwrite an existing file
* when proper write permissions on the file being overwritten are provided.
* @throws Throwable
*/
@Test
public void testCreateAccessWithOverwriteCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/");
Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true);
fs.updateWasbAuthorizer(authorizer);
boolean initialCreateSucceeded = false;
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
initialCreateSucceeded = true;
fs.create(testPath, true);
}
finally {
ContractTestUtils.assertTrue(initialCreateSucceeded);
fs.delete(testPath, false);
}
}
/**
* Negative test to verify that Create fails when appropriate permissions are not provided.
* @throws Throwable * @throws Throwable
*/ */
@ -110,20 +229,21 @@ public class TestNativeAzureFileSystemAuthorization
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(),WasbAuthorizationOperations.WRITE.toString(), false); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), false);
authorizer.addAuthRule(parentDir.toString(),WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
try { try {
fs.create(testPath); fs.create(testPath);
} }
finally { finally {
/* Provide permissions to cleanup in case the file got created */
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
/** /**
* Positive test to verify Create and delete access check * Positive test to verify listStatus access check
* @throws Throwable * @throws Throwable
*/ */
@Test @Test
@ -133,28 +253,27 @@ public class TestNativeAzureFileSystemAuthorization
NativeAzureFileSystem fs = testAccount.getFileSystem(); NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testListAccessCheckPositive"); Path parentDir = new Path("/testListAccessCheckPositive");
Path testPath = new Path(parentDir, "test.dat"); Path intermediateFolders = new Path(parentDir, "1/2/3/");
Path testPath = new Path(intermediateFolders, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath does not exist", testPath);
try { try {
fs.create(testPath);
fs.listStatus(testPath); fs.listStatus(testPath);
} }
finally { finally {
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
/** /**
* Negative test to verify Create access check * Negative test to verify listStatus access check
* @throws Throwable * @throws Throwable
*/ */
@ -162,7 +281,7 @@ public class TestNativeAzureFileSystemAuthorization
public void testListAccessCheckNegative() throws Throwable { public void testListAccessCheckNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class); expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("getFileStatus operation for Path : /testListAccessCheckNegative/test.dat not allowed"); expectedEx.expectMessage("liststatus operation for Path : /testListAccessCheckNegative/test.dat not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount(); AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem(); NativeAzureFileSystem fs = testAccount.getFileSystem();
@ -172,18 +291,16 @@ public class TestNativeAzureFileSystemAuthorization
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), false); authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), false);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath does not exist", testPath);
try { try {
fs.create(testPath);
fs.listStatus(testPath); fs.listStatus(testPath);
} }
finally { finally {
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
@ -199,25 +316,26 @@ public class TestNativeAzureFileSystemAuthorization
NativeAzureFileSystem fs = testAccount.getFileSystem(); NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testRenameAccessCheckPositive"); Path parentDir = new Path("/testRenameAccessCheckPositive");
Path testPath = new Path(parentDir, "test.dat"); Path srcPath = new Path(parentDir, "test1.dat");
Path renamePath = new Path(parentDir, "test2.dat"); Path dstPath = new Path(parentDir, "test2.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true); /* to create parentDir */
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.WRITE.toString(), true); /* for rename */
authorizer.addAuthRule(renamePath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(srcPath.toString(), WasbAuthorizationOperations.READ.toString(), true); /* for exists */
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(dstPath.toString(), WasbAuthorizationOperations.READ.toString(), true); /* for exists */
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", testPath);
try { try {
fs.rename(testPath, renamePath); fs.create(srcPath);
ContractTestUtils.assertPathExists(fs, "destPath does not exist", renamePath); ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", srcPath);
fs.rename(srcPath, dstPath);
ContractTestUtils.assertPathExists(fs, "destPath does not exist", dstPath);
ContractTestUtils.assertPathDoesNotExist(fs, "sourcePath exists after rename!", srcPath);
} }
finally { finally {
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
@ -230,37 +348,112 @@ public class TestNativeAzureFileSystemAuthorization
public void testRenameAccessCheckNegative() throws Throwable { public void testRenameAccessCheckNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class); expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("rename operation for Path : /testRenameAccessCheckNegative/test.dat not allowed"); expectedEx.expectMessage("rename operation for Path : /testRenameAccessCheckNegative/test1.dat not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount(); AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem(); NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testRenameAccessCheckNegative"); Path parentDir = new Path("/testRenameAccessCheckNegative");
Path testPath = new Path(parentDir, "test.dat"); Path srcPath = new Path(parentDir, "test1.dat");
Path renamePath = new Path(parentDir, "test2.dat"); Path dstPath = new Path(parentDir, "test2.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true); /* to create parent dir */
// set EXECUTE to true for initial assert right after creation. authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.WRITE.toString(), false);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(srcPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true); authorizer.addAuthRule(dstPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", testPath);
// Set EXECUTE to false for actual rename-failure test
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), false);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
try { try {
fs.rename(testPath, renamePath); fs.create(srcPath);
ContractTestUtils.assertPathExists(fs, "destPath does not exist", renamePath); ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", srcPath);
fs.rename(srcPath, dstPath);
ContractTestUtils.assertPathExists(fs, "destPath does not exist", dstPath);
} finally { } finally {
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist after rename failure!", srcPath);
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
/**
* Negative test to verify rename access check - the dstFolder disallows rename
* @throws Throwable
*/
@Test //(expected=WasbAuthorizationException.class)
public void testRenameAccessCheckNegativeOnDstFolder() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("rename operation for Path : /testRenameAccessCheckNegativeDst/test2.dat not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentSrcDir = new Path("/testRenameAccessCheckNegativeSrc");
Path srcPath = new Path(parentSrcDir, "test1.dat");
Path parentDstDir = new Path("/testRenameAccessCheckNegativeDst");
Path dstPath = new Path(parentDstDir, "test2.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true); /* to create parent dir */
authorizer.addAuthRule(parentSrcDir.toString(), WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(parentDstDir.toString(), WasbAuthorizationOperations.WRITE.toString(), false);
authorizer.addAuthRule(srcPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(dstPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(srcPath);
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", srcPath);
fs.rename(srcPath, dstPath);
ContractTestUtils.assertPathDoesNotExist(fs, "destPath does not exist", dstPath);
} finally {
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist after rename !", srcPath);
allowRecursiveDelete(fs, authorizer, parentSrcDir.toString());
fs.delete(parentSrcDir, true);
}
}
/**
* Positive test to verify rename access check - the dstFolder allows rename
* @throws Throwable
*/
@Test //(expected=WasbAuthorizationException.class)
public void testRenameAccessCheckPositiveOnDstFolder() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentSrcDir = new Path("/testRenameAccessCheckPositiveSrc");
Path srcPath = new Path(parentSrcDir, "test1.dat");
Path parentDstDir = new Path("/testRenameAccessCheckPositiveDst");
Path dstPath = new Path(parentDstDir, "test2.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true); /* to create parent dirs */
authorizer.addAuthRule(parentSrcDir.toString(), WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(parentDstDir.toString(), WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(srcPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(dstPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(srcPath);
ContractTestUtils.assertPathExists(fs, "sourcePath does not exist", srcPath);
fs.mkdirs(parentDstDir);
fs.rename(srcPath, dstPath);
ContractTestUtils.assertPathDoesNotExist(fs, "sourcePath does not exist", srcPath);
ContractTestUtils.assertPathExists(fs, "destPath does not exist", dstPath);
} finally {
allowRecursiveDelete(fs, authorizer, parentSrcDir.toString());
fs.delete(parentSrcDir, true);
allowRecursiveDelete(fs, authorizer, parentDstDir.toString());
fs.delete(parentDstDir, true);
}
}
/** /**
* Positive test for read access check. * Positive test for read access check.
* @throws Throwable * @throws Throwable
@ -275,27 +468,30 @@ public class TestNativeAzureFileSystemAuthorization
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true); authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
FSDataOutputStream fso = fs.create(testPath);
String data = "Hello World";
fso.writeBytes(data);
fso.close();
ContractTestUtils.assertPathExists(fs, "testPath does not exist", testPath);
FSDataInputStream inputStream = null; FSDataInputStream inputStream = null;
FSDataOutputStream fso = null;
try { try {
fso = fs.create(testPath);
String data = "Hello World";
fso.writeBytes(data);
fso.close();
inputStream = fs.open(testPath); inputStream = fs.open(testPath);
ContractTestUtils.verifyRead(inputStream, data.getBytes(), 0, data.length()); ContractTestUtils.verifyRead(inputStream, data.getBytes(), 0, data.length());
} }
finally { finally {
if (fso != null) {
fso.close();
}
if(inputStream != null) { if(inputStream != null) {
inputStream.close(); inputStream.close();
} }
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
@ -318,27 +514,239 @@ public class TestNativeAzureFileSystemAuthorization
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl(); MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null); authorizer.init(null);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.WRITE.toString(), true); authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), false); authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), false);
authorizer.addAuthRule(parentDir.toString(), WasbAuthorizationOperations.EXECUTE.toString(), true);
fs.updateWasbAuthorizer(authorizer); fs.updateWasbAuthorizer(authorizer);
FSDataOutputStream fso = fs.create(testPath);
String data = "Hello World";
fso.writeBytes(data);
fso.close();
ContractTestUtils.assertPathExists(fs, "testPath does not exist", testPath);
FSDataInputStream inputStream = null; FSDataInputStream inputStream = null;
FSDataOutputStream fso = null;
try { try {
fso = fs.create(testPath);
String data = "Hello World";
fso.writeBytes(data);
fso.close();
inputStream = fs.open(testPath); inputStream = fs.open(testPath);
ContractTestUtils.verifyRead(inputStream, data.getBytes(), 0, data.length()); ContractTestUtils.verifyRead(inputStream, data.getBytes(), 0, data.length());
} finally { } finally {
if (fso != null) {
fso.close();
}
if (inputStream != null) { if (inputStream != null) {
inputStream.close(); inputStream.close();
} }
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true); fs.delete(parentDir, true);
} }
} }
/**
* Positive test to verify file delete access check
* @throws Throwable
*/
@Test
public void testFileDeleteAccessCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/");
Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
}
finally {
fs.delete(testPath, false);
ContractTestUtils.assertPathDoesNotExist(fs, "testPath exists after deletion!", testPath);
}
}
/**
* Negative test to verify file delete access check
* @throws Throwable
*/
@Test //(expected=WasbAuthorizationException.class)
public void testFileDeleteAccessCheckNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("delete operation for Path : /test.dat not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/");
Path testPath = new Path(parentDir, "test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
/* Remove permissions for delete to force failure */
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), false);
fs.updateWasbAuthorizer(authorizer);
fs.delete(testPath, false);
}
finally {
/* Restore permissions to force a successful delete */
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
fs.delete(testPath, false);
ContractTestUtils.assertPathDoesNotExist(fs, "testPath exists after deletion!", testPath);
}
}
/**
* Positive test to verify file delete access check, with intermediate folders
* Uses wildcard recursive permissions
* @throws Throwable
*/
@Test
public void testFileDeleteAccessWithIntermediateFoldersCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path parentDir = new Path("/testDeleteIntermediateFolder");
Path testPath = new Path(parentDir, "1/2/test.dat");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true); // for create and delete
authorizer.addAuthRule("/testDeleteIntermediateFolder*",
WasbAuthorizationOperations.WRITE.toString(), true); // for recursive delete
authorizer.addAuthRule("/*", WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.create(testPath);
ContractTestUtils.assertPathExists(fs, "testPath was not created", testPath);
fs.delete(parentDir, true);
ContractTestUtils.assertPathDoesNotExist(fs, "testPath exists after deletion!", parentDir);
}
finally {
allowRecursiveDelete(fs, authorizer, parentDir.toString());
fs.delete(parentDir, true);
}
}
/**
* Positive test for getFileStatus
* @throws Throwable
*/
@Test
public void testGetFileStatusPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path testPath = new Path("/");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
ContractTestUtils.assertIsDirectory(fs, testPath);
}
/**
* Negative test for getFileStatus
* @throws Throwable
*/
@Test //(expected=WasbAuthorizationException.class)
public void testGetFileStatusNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("getFileStatus operation for Path : / not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path testPath = new Path("/");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.READ.toString(), false);
fs.updateWasbAuthorizer(authorizer);
ContractTestUtils.assertIsDirectory(fs, testPath);
}
/**
* Positive test for mkdirs access check
* @throws Throwable
*/
@Test
public void testMkdirsCheckPositive() throws Throwable {
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path testPath = new Path("/testMkdirsAccessCheckPositive/1/2/3");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), true);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.mkdirs(testPath);
ContractTestUtils.assertIsDirectory(fs, testPath);
}
finally {
allowRecursiveDelete(fs, authorizer, "/testMkdirsAccessCheckPositive");
fs.delete(new Path("/testMkdirsAccessCheckPositive"), true);
}
}
/**
* Negative test for mkdirs access check
* @throws Throwable
*/
@Test //(expected=WasbAuthorizationException.class)
public void testMkdirsCheckNegative() throws Throwable {
expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("mkdirs operation for Path : /testMkdirsAccessCheckNegative/1/2/3 not allowed");
AzureBlobStorageTestAccount testAccount = createTestAccount();
NativeAzureFileSystem fs = testAccount.getFileSystem();
Path testPath = new Path("/testMkdirsAccessCheckNegative/1/2/3");
MockWasbAuthorizerImpl authorizer = new MockWasbAuthorizerImpl();
authorizer.init(null);
authorizer.addAuthRule("/", WasbAuthorizationOperations.WRITE.toString(), false);
authorizer.addAuthRule(testPath.toString(), WasbAuthorizationOperations.READ.toString(), true);
fs.updateWasbAuthorizer(authorizer);
try {
fs.mkdirs(testPath);
ContractTestUtils.assertPathDoesNotExist(fs, "testPath was not created", testPath);
}
finally {
allowRecursiveDelete(fs, authorizer, "/testMkdirsAccessCheckNegative");
fs.delete(new Path("/testMkdirsAccessCheckNegative"), true);
}
}
} }

View File

@ -263,9 +263,9 @@ public class TestWasbRemoteCallHelper
private void setupExpectations() { private void setupExpectations() {
expectedEx.expect(WasbAuthorizationException.class); expectedEx.expect(WasbAuthorizationException.class);
expectedEx.expectMessage("org.apache.hadoop.fs.azure.WasbRemoteCallException: " + expectedEx.expectMessage("org.apache.hadoop.fs.azure.WasbRemoteCallException: "
"http://localhost/CHECK_AUTHORIZATION?wasb_absolute_path=%2Ftest.dat&" + + "http://localhost/CHECK_AUTHORIZATION?wasb_absolute_path=%2F&"
"operation_type=write&delegation_token:Encountered IOException while making remote call"); + "operation_type=write:Encountered IOException while making remote call");
} }
private void performop(HttpClient mockHttpClient) throws Throwable { private void performop(HttpClient mockHttpClient) throws Throwable {

View File

@ -117,7 +117,8 @@ public class TestAzureFileSystemInstrumentation {
// levels, and then 2 requests for checking/stamping the version of AS, // levels, and then 2 requests for checking/stamping the version of AS,
// totaling 11. // totaling 11.
// Also, there's the initial 1 request for container check so total is 12. // Also, there's the initial 1 request for container check so total is 12.
base = assertWebResponsesInRange(base, 1, 12); // The getAncestor call at the very beginning adds another 4 calls, totalling 16.
base = assertWebResponsesInRange(base, 1, 16);
assertEquals(1, assertEquals(1,
AzureMetricsTestUtil.getLongCounterValue(getInstrumentation(), WASB_DIRECTORIES_CREATED)); AzureMetricsTestUtil.getLongCounterValue(getInstrumentation(), WASB_DIRECTORIES_CREATED));