Staging directory filter. Uses 2 regular expressions to filter out the stating directory based on the apache core classes algorithm (LocalJobRunner.java & JobTracker.java)

Author: childsb

src/main/java/org/apache/hadoop/fs/glusterfs/AclPathFilter.java

@@ -0,0 +1,57 @@
+package org.apache.hadoop.fs.glusterfs;
+
+import java.io.IOException;
+import java.util.ListIterator;
+import java.util.Vector;
+import org.apache.hadoop.conf.Configuration;
+import org.apache.hadoop.fs.Path;
+import org.apache.hadoop.security.UserGroupInformation;
+
+
+public class AclPathFilter {
+	
+	Vector<String> paths = null;
+	
+	public AclPathFilter(){
+		paths = new Vector<String>();
+	}
+	/* generates a white list of ACL path regular expressions */
+	public AclPathFilter(Configuration conf){
+		this();
+	    UserGroupInformation ugi = null;
+	    try {
+			ugi = UserGroupInformation.getCurrentUser();
+		} catch (IOException e) {
+			
+		}
+		String stagingRootDir = new Path(conf.get("mapreduce.jobtracker.staging.root.dir", "/tmp/hadoop/mapred/staging")).toString();
+		String user;
+		String randid = "\\d*";
+		if (ugi != null) {
+			user = ugi.getShortUserName();
+		} else {
+		    user = "dummy";
+		}
+		paths.add("^" + new Path(stagingRootDir, user +  randid +"/.staging").toString() + ".*");
+		stagingRootDir = new Path(conf.get("mapreduce.jobtracker.staging.root.dir","/tmp/hadoop/mapred/staging")).toString();
+        paths.add("^" + new Path(stagingRootDir, user+"/.staging").toString() + ".*");
+		
+	}
+	
+	public boolean matches(String path){
+		
+		boolean needsAcl = false;
+		ListIterator<String> list = paths.listIterator();
+		String filterPath = null;
+		while(list.hasNext() && !needsAcl){
+			filterPath = list.next();
+			if(path.matches(filterPath)){
+				needsAcl = true;
+			}
+		}
+		return needsAcl;
+	}
+	public boolean matches(Path path){
+		return matches(path.toString());
+	}
+}

src/main/java/org/apache/hadoop/fs/glusterfs/GlusterVolume.java

@@ -43,6 +43,7 @@ public class GlusterVolume extends RawLocalFileSystem{
 
     protected String root=null;
     protected String superUser=null;
+    protected AclPathFilter aclFilter = null;
 
     protected static GlusterFSXattr attr = null;
 
@@ -83,6 +84,7 @@ public void setConf(Configuration conf){
 
                 superUser =  conf.get("gluster.daemon.user", null);
 
+                aclFilter = new AclPathFilter(conf);
                 //volName=conf.get("fs.glusterfs.volname", null);
                 //remoteGFSServer=conf.get("fs.glusterfs.server", null);
 
@@ -165,7 +167,7 @@ public long getBlockSize(Path path) throws IOException{
      */
 
     private void updateAcl(Path p){
-    	if(superUser!=null){
+    	if(superUser!=null && aclFilter.matches(p)  ){
     		File f = pathToFile(p);
     		String path = f.getAbsolutePath();
     		String command = "setfacl -m u:" + superUser + ":rwx " + path;