Subversion Repositories XServices

import javax.ws.rs.core.MediaType;

import javax.ws.rs.core.MediaType;

import javax.ws.rs.core.Response;

import javax.ws.rs.core.Response;

import javax.ws.rs.core.StreamingOutput;

import javax.ws.rs.core.StreamingOutput;

import javax.ws.rs.core.UriInfo;

import javax.ws.rs.core.UriInfo;

import lombok.extern.slf4j.Slf4j;

import net.brutex.xservices.security.DirectoryPermission;

import net.brutex.xservices.security.DirectoryPermission;

import net.brutex.xservices.types.FileInfoType;

import net.brutex.xservices.types.FileInfoType;

import net.brutex.xservices.util.FileWalker;

import net.brutex.xservices.util.FileWalker;

import org.apache.commons.jcs.JCS;

import org.apache.commons.jcs.JCS;

import org.apache.logging.log4j.LogManager;

import org.apache.commons.jcs.access.CacheAccess;

import org.apache.logging.log4j.Logger;

import org.apache.commons.jcs.access.exception.CacheException;

import org.apache.shiro.SecurityUtils;

import org.apache.shiro.SecurityUtils;

import org.apache.shiro.authz.UnauthorizedException;

import org.apache.shiro.authz.UnauthorizedException;

/**

/**

 * The Class FileInfoImpl.

 * The Class FileInfoImpl.

 *

 * @author Brian Rosenberger, bru(at)brutex.de

public class FileInfoImpl  implements FileInfo {

@Slf4j

public class FileInfoImpl  implements FileInfo {

	Logger logger = LogManager.getLogger();

  /* (non-Javadoc)

  /* (non-Javadoc)

   */

   */

  public Response getFiles(HttpHeaders h, UriInfo uriInfo, String dir, boolean withDir, boolean withFiles, int level, String search, int count, int page, boolean useCache)

  public Response getFiles(HttpHeaders h, UriInfo uriInfo, String dir, boolean withDir, boolean withFiles, int level, String search, int count, int page, boolean useCache)

  {

  {

		dir = "c:/"; 

		dir = "c:/"; 

		logger.warn("No directory specified. Default is 'c:/'.");

		log.warn("No directory specified. Default is 'c:/'.");

		}

		}

	isPermitted(dir);

	isPermitted(dir);

    URI baseuri = URI.create(uriInfo.getBaseUri()+FileInfo.BASE_PATH+"getFile?file=");

    URI baseuri = URI.create(uriInfo.getBaseUri()+FileInfo.BASE_PATH+"getFile?file=");

    logger.info(String.format("Listing directory '%s'.", dir));

    log.info(String.format("Listing directory '%s'.", dir));

    if (level <= 0) level = 1;

    if (level <= 0) level = 1;

    if ((!withDir) && (!withFiles)) withFiles = true;

    if ((!withDir) && (!withFiles)) withFiles = true;

    try {

    try {

      CacheAccess<Object, Object> jcs = JCS.getInstance("FileCache");

      CacheAccess<Object, Object> jcs = JCS.getInstance("FileCache");

      /*Try to retrieve the file list from the cache*/

      /*Try to retrieve the file list from the cache*/

      List<FileInfoType> list = (List<FileInfoType>)jcs.get(cachekey);

      List<FileInfoType> list = (List<FileInfoType>)jcs.get(cachekey);

      if (list == null || !useCache) {

      if (list == null || !useCache) {

        list = setDirectory(baseuri, dir, withDir, withFiles, level, search);

        list = setDirectory(baseuri, dir, withDir, withFiles, level, search);

        jcs.put(cachekey, list);

        jcs.put(cachekey, list);

      } else {

      } else {

        logger.debug("Got from Cache: " + list.toString());

        log.debug("Got from Cache: " + list.toString());

      }

      }

      int fromIndex = 0;

      int fromIndex = 0;

  {

  {

    if (depth <= 0) return;

    if (depth <= 0) return;

    	if(search==null || search.equals("") ) {

    	if(search==null || search.equals("") ) {

    		logger.info("No search pattern supplied, using default '*'.");

    		log.info("No search pattern supplied, using default '*'.");

    	}

    	}

    	FileWalker finder = new FileWalker(search);

    	FileWalker finder = new FileWalker(search);

			Files.walkFileTree(dir.toPath(), EnumSet.of(FileVisitOption.FOLLOW_LINKS), depth, finder);

			Files.walkFileTree(dir.toPath(), EnumSet.of(FileVisitOption.FOLLOW_LINKS), depth, finder);

			logger.info("FileWalker returned '"+finder.getCount()+"' hits. '" + finder.getTotal() + "' files have been scanned.");

			log.info("FileWalker returned '"+finder.getCount()+"' hits. '" + finder.getTotal() + "' files have been scanned.");

			List<Path> result = finder.getResult();

			List<Path> result = finder.getResult();

	    	for(Path f : result) {

	    	for(Path f : result) {

	    		if(! withDirectories) {

	    		if(! withDirectories) {

	    			if(f.toFile().isFile()) continue;

	    			if(f.toFile().isFile()) continue;

	    		}

	    		}

	    		list.add(new FileInfoType(f, baseuri));

	    		list.add(new FileInfoType(f, baseuri));

	    	}

	    	}

		} catch (IOException e2) {

		} catch (IOException e2) {

		}

		}

  }

  }

  /**

  /**

		mime = MediaType.valueOf(Files.probeContentType(path));

		mime = MediaType.valueOf(Files.probeContentType(path));

	} catch (IllegalArgumentException | IOException e) {

	} catch (IllegalArgumentException | IOException e) {

		//In case we can not find the media type for some reason

		//In case we can not find the media type for some reason

		//the default assignment is taken, so we can

		//the default assignment is taken, so we can

		//ignore this error.

		//ignore this error.

	}

	}

	Response r = Response.ok(path.toFile(), mime).build();

	Response r = Response.ok(path.toFile(), mime).build();

	String fileName = path.getFileName().toString();

	String fileName = path.getFileName().toString();

	if(mime == MediaType.APPLICATION_OCTET_STREAM_TYPE) r.getHeaders().add("Content-Disposition", "attachment; filename=\"" + fileName + "\"");

	if(mime == MediaType.APPLICATION_OCTET_STREAM_TYPE) r.getHeaders().add("Content-Disposition", "attachment; filename=\"" + fileName + "\"");

	return r;

	return r;

		} catch (IOException e1) {

		} catch (IOException e1) {

			// TODO Auto-generated catch block

			// TODO Auto-generated catch block

			return Response.serverError().build();

			return Response.serverError().build();

		}

		}

}

}

	return r;

	return r;

}

}

private boolean isPermitted(String dir) {

private boolean isPermitted(String dir) {

		logger.warn(String.format("User '%s' does not have permission to access '%s'.",SecurityUtils.getSubject().getPrincipal(), dir ));

		log.warn(String.format("User '%s' does not have permission to access '%s'.",SecurityUtils.getSubject().getPrincipal(), dir ));

		throw new NotAuthorizedException(new UnauthorizedException("User does not have permission to access "+ dir));

		throw new NotAuthorizedException(new UnauthorizedException("User does not have permission to access "+ dir));

	}

	}

	return true;

	return true;