com.hierynomus.smbj.share.File Java Examples

@Override
public OutputStream createFile(String f) throws FileSystemException, IOException {
	Set<AccessMask> accessMask = new HashSet<AccessMask>(EnumSet.of(AccessMask.FILE_ADD_FILE));
	Set<SMB2CreateOptions> createOptions = new HashSet<SMB2CreateOptions>(
			EnumSet.of(SMB2CreateOptions.FILE_NON_DIRECTORY_FILE, SMB2CreateOptions.FILE_WRITE_THROUGH));
	
	final File file = diskShare.openFile(f, accessMask, null, SMB2ShareAccess.ALL,
			SMB2CreateDisposition.FILE_OVERWRITE_IF, createOptions);
	OutputStream out = file.getOutputStream();
	FilterOutputStream fos = new FilterOutputStream(out) {

		boolean isOpen = true;
		@Override
		public void close() throws IOException {
			if(isOpen) {
				super.close();
				isOpen=false;
			}
			file.close();
		}
	};
	return fos;
}
 

@Override
public long getFileLength(String fileName) {
    String filePath = getPathNotShare(fileName);

    try {
        File file = openFile(diskShare, filePath);
        FileStandardInformation standardInfo = file.getFileInformation(FileStandardInformation.class);
        return standardInfo.getEndOfFile();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return 0;
}
 

/**
 * get smb file, just need reed permission
 *
 * @param share    share
 * @param filePath file path not share name
 * @return smb file
 */
private File openFile(DiskShare share, String filePath) {
    return share.openFile(filePath,
            EnumSet.of(AccessMask.FILE_READ_DATA),
            null,
            SMB2ShareAccess.ALL,
            FILE_OPEN,
            null);
}
 

@Override
public InputStream getFileInputStream(String fileName) {
    String filePath = getPathNotShare(fileName);

    try {
        File file = openFile(diskShare, filePath);
        inputStream = file.getInputStream();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return inputStream;
}
 

@Override
public long getFileLength(String fileName) {
    String filePath = getPathNotShare(fileName);

    try {
        File file = openFile(diskShare, filePath);
        FileStandardInformation standardInfo = file.getFileInformation(FileStandardInformation.class);
        return standardInfo.getEndOfFile();
    } catch (Exception e) {
        e.printStackTrace();
    }
    return 0;
}
 

/**
 * get smb file, just need reed permission
 *
 * @param share    share
 * @param filePath file path not share name
 * @return smb file
 */
private File openFile(DiskShare share, String filePath) {
    return share.openFile(filePath,
            EnumSet.of(AccessMask.FILE_READ_DATA),
            null,
            SMB2ShareAccess.ALL,
            FILE_OPEN,
            null);
}
 

@Override
public InputStream getFileInputStream(String fileName) {
    String filePath = getPathNotShare(fileName);

    try {
        File file = openFile(diskShare, filePath);
        inputStream = file.getInputStream();
        return inputStream;
    } catch (Exception e) {
        e.printStackTrace();
    }
    return null;
}
 

private static OutputStream getOutputStream(final PooledSMB2Connection connection, String path, boolean append) throws IOException {
	try {
		DiskShare share = connection.getShare();
		
		Set<AccessMask> accessMask = append ? EnumSet.of(AccessMask.FILE_APPEND_DATA) : EnumSet.of(AccessMask.FILE_WRITE_DATA);
		SMB2CreateDisposition createDisposition = append ? SMB2CreateDisposition.FILE_OPEN_IF : SMB2CreateDisposition.FILE_OVERWRITE_IF;
		
		final com.hierynomus.smbj.share.File file = openFile(share, path, accessMask, createDisposition);
		OutputStream os = append ? new AppendOutputStream(file) : file.getOutputStream();
		if (append) { // add buffering, AppendOutputStream is slow
			os = new BufferedOutputStream(os, connection.getWriteBufferSize());
		}
		os = new CloseOnceOutputStream(os) {

			@Override
			protected void doClose() throws IOException {
				try {
					super.doClose();
				} finally {
					FileUtils.closeAll(file, connection);
				}
			}
			
		};
		return os;
	} catch (Throwable t) {
		connection.returnToPool();
		throw ExceptionUtils.getIOException(t);
	}
}
 

@Override
public OutputStream appendFile(String f) throws FileSystemException, IOException {
	final File file = getFile(f, AccessMask.FILE_APPEND_DATA, SMB2CreateDisposition.FILE_OPEN_IF);
	OutputStream out = file.getOutputStream();
	FilterOutputStream fos = new FilterOutputStream(out) {

		boolean isOpen = true;
		@Override
		public void close() throws IOException {
			if(isOpen) {
				super.close();
				isOpen=false;
			}
			file.close();
		}
	};
	return fos;
}
 

@Override
public InputStream readFile(String filename) throws FileSystemException, IOException {
	final File file = getFile(filename, AccessMask.GENERIC_READ, SMB2CreateDisposition.FILE_OPEN);
	InputStream is = file.getInputStream();
	FilterInputStream fis = new FilterInputStream(is) {

		boolean isOpen = true;
		@Override
		public void close() throws IOException {
			if(isOpen) {
				super.close();
				isOpen=false;
			}
			file.close();
		}
	};
	return fis;
}
 

@Override
public long getFileSize(String f) throws FileSystemException {
	long size;
	if (isFolder(f)) {
		try (Directory dir = getFolder(f, AccessMask.FILE_READ_ATTRIBUTES, SMB2CreateDisposition.FILE_OPEN)) {
			size = dir.getFileInformation().getStandardInformation().getAllocationSize();
			return size;
		}
	} else {
		try (File file = getFile(f, AccessMask.FILE_READ_ATTRIBUTES, SMB2CreateDisposition.FILE_OPEN)) {
			size = file.getFileInformation().getStandardInformation().getAllocationSize();
			return size;
		}
	}
}
 

private File getFile(String filename, AccessMask accessMask, SMB2CreateDisposition createDisposition) {
	Set<SMB2ShareAccess> shareAccess = new HashSet<SMB2ShareAccess>();
	shareAccess.addAll(SMB2ShareAccess.ALL);

	Set<SMB2CreateOptions> createOptions = new HashSet<SMB2CreateOptions>();
	createOptions.add(SMB2CreateOptions.FILE_WRITE_THROUGH);
	
	Set<AccessMask> accessMaskSet = new HashSet<AccessMask>();
	accessMaskSet.add(accessMask);
	File file;

	file = diskShare.openFile(filename, accessMaskSet, null, shareAccess, createDisposition, createOptions);
	return file;
}
 

@Override
public String copyFile(String f, String destinationFolder, boolean createFolder) throws FileSystemException {
	try (File file = getFile(f, AccessMask.GENERIC_ALL, SMB2CreateDisposition.FILE_OPEN)) {
		String destination = getDestinationFile(f, destinationFolder, createFolder, "copy");
		try (File destinationFile = getFile(f, AccessMask.GENERIC_ALL, SMB2CreateDisposition.FILE_OVERWRITE)) {
			file.remoteCopyTo(destinationFile);
		} catch (TransportException | BufferException e) {
			throw new FileSystemException("cannot copy file ["+f+"] to ["+destinationFolder+"]",e);
		}
		return destination;
	}
}
 

@Override
public String moveFile(String f, String destinationFolder, boolean createFolder) throws FileSystemException {
	try (File file = getFile(f, AccessMask.GENERIC_ALL, SMB2CreateDisposition.FILE_OPEN)) {
		String destination = getDestinationFile(f, destinationFolder, createFolder, "move");
		file.rename(destination, false);
		return destination;
	}
}
 

@Override
public String renameFile(String f, String newName, boolean force) throws FileSystemException {
	try (File file = getFile(f, AccessMask.GENERIC_ALL, SMB2CreateDisposition.FILE_OPEN)) {
		if (exists(newName) && !force) {
			throw new FileSystemException("Cannot rename file. Destination file already exists.");
		}
		file.rename(newName, force);
	}
	return newName;
}
 

private static InputStream getInputStream(final PooledSMB2Connection connection, String path) throws IOException {
	try {
		DiskShare share = connection.getShare();
		
		Set<AccessMask> accessMask = EnumSet.of(AccessMask.FILE_READ_DATA);
		SMB2CreateDisposition createDisposition = SMB2CreateDisposition.FILE_OPEN;
		
		final com.hierynomus.smbj.share.File file = openFile(share, path, accessMask, createDisposition);
		InputStream is = file.getInputStream();
		is = new CloseOnceInputStream(is) {

			@Override
			protected void doClose() throws IOException {
				try {
					super.doClose();
				} finally {
					FileUtils.closeAll(file, connection);
				}
			}
			
		};
		return is;
	} catch (Throwable t) {
		connection.returnToPool();
		throw ExceptionUtils.getIOException(t);
	}
}
 

public AppendOutputStream(File file) {
	this.file = Objects.requireNonNull(file);
	this.fileOffset = file.getFileInformation(FileStandardInformation.class).getEndOfFile();
}