com.hierynomus.msfscc.fileinformation.FileInformationFactory Maven / Gradle / Ivy
/*
* Copyright (C)2016 - SMBJ Contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hierynomus.msfscc.fileinformation;
import com.hierynomus.msdtyp.FileTime;
import com.hierynomus.msdtyp.MsDataTypes;
import com.hierynomus.msfscc.FileInformationClass;
import com.hierynomus.protocol.commons.Charsets;
import com.hierynomus.protocol.commons.buffer.Buffer;
import com.hierynomus.protocol.commons.buffer.Endian;
import com.hierynomus.smbj.common.SMBRuntimeException;
import java.util.*;
public class FileInformationFactory {
private static final Map, FileInformation.Encoder> encoders;
private static final Map, FileInformation.Decoder> decoders;
private FileInformationFactory() {
}
static {
encoders = new HashMap<>();
decoders = new HashMap<>();
decoders.put(FileAccessInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileAccessInformation;
}
@Override
public FileAccessInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileAccessInformation(inputBuffer);
}
});
decoders.put(FileAlignmentInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileAlignmentInformation;
}
@Override
public FileAlignmentInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileAlignmentInformation(inputBuffer);
}
});
decoders.put(FileAllInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileAllInformation;
}
@Override
public FileAllInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileAllInformation(inputBuffer);
}
});
FileInformation.Codec allocationCodec = new FileInformation.Codec() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileAllocationInformation;
}
@Override
public FileAllocationInformation read(Buffer inputBuffer) throws Buffer.BufferException {
long allocationSize = ((Buffer) inputBuffer).readLong();
return new FileAllocationInformation(allocationSize);
}
@Override
public void write(FileAllocationInformation info, Buffer outputBuffer) {
outputBuffer.putLong(info.getAllocationSize());
}
};
decoders.put(FileAllocationInformation.class, allocationCodec);
encoders.put(FileAllocationInformation.class, allocationCodec);
FileInformation.Codec basicCodec = new FileInformation.Codec() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileBasicInformation;
}
@Override
public FileBasicInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileBasicInformation(inputBuffer);
}
@Override
public void write(FileBasicInformation info, Buffer outputBuffer) {
MsDataTypes.putFileTime(info.getCreationTime(), outputBuffer);
MsDataTypes.putFileTime(info.getLastAccessTime(), outputBuffer);
MsDataTypes.putFileTime(info.getLastWriteTime(), outputBuffer);
MsDataTypes.putFileTime(info.getChangeTime(), outputBuffer);
((Buffer) outputBuffer).putUInt32(info.getFileAttributes());
((Buffer) outputBuffer).putUInt32(0); // Reserved (4 bytes)
}
};
decoders.put(FileBasicInformation.class, basicCodec);
encoders.put(FileBasicInformation.class, basicCodec);
FileInformation.Encoder dispositionCodec = new FileInformation.Encoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileDispositionInformation;
}
@Override
public void write(FileDispositionInformation info, Buffer outputBuffer) {
((Buffer) outputBuffer).putBoolean(info.isDeleteOnClose());
}
};
encoders.put(FileDispositionInformation.class, dispositionCodec);
decoders.put(FileEaInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileEaInformation;
}
@Override
public FileEaInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileEaInformation(inputBuffer);
}
});
decoders.put(FileStreamInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileStreamInformation;
}
@Override
public FileStreamInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileStreamInformation(inputBuffer);
}
});
FileInformation.Encoder endOfFileCodec = new FileInformation.Encoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileEndOfFileInformation;
}
@Override
public void write(FileEndOfFileInformation info, Buffer outputBuffer) {
((Buffer) outputBuffer).putLong(info.getEndOfFile());
}
};
encoders.put(FileEndOfFileInformation.class, endOfFileCodec);
decoders.put(FileInternalInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileInternalInformation;
}
@Override
public FileInternalInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileInternalInformation(inputBuffer);
}
});
FileInformation.Codec modeCodec = new FileInformation.Codec() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileModeInformation;
}
@Override
public FileModeInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileModeInformation(inputBuffer);
}
@Override
public void write(FileModeInformation info, Buffer outputBuffer) {
outputBuffer.putUInt32(info.getMode() & 0xFFFFFFFFL);
}
};
decoders.put(FileModeInformation.class, modeCodec);
encoders.put(FileModeInformation.class, modeCodec);
decoders.put(FilePositionInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FilePositionInformation;
}
@Override
public FilePositionInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFilePositionInformation(inputBuffer);
}
});
decoders.put(FileStandardInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileStandardInformation;
}
@Override
public FileStandardInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileStandardInformation(inputBuffer);
}
});
decoders.put(FileBothDirectoryInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileBothDirectoryInformation;
}
@Override
public FileBothDirectoryInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileBothDirectoryInformation(inputBuffer);
}
});
decoders.put(FileDirectoryInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileDirectoryInformation;
}
@Override
public FileDirectoryInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileDirectoryInformation(inputBuffer);
}
});
decoders.put(FileFullDirectoryInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileFullDirectoryInformation;
}
@Override
public FileFullDirectoryInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileFullDirectoryInformation(inputBuffer);
}
});
decoders.put(FileIdBothDirectoryInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileIdBothDirectoryInformation;
}
@Override
public FileIdBothDirectoryInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileIdBothDirectoryInformation(inputBuffer);
}
});
decoders.put(FileIdFullDirectoryInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileIdFullDirectoryInformation;
}
@Override
public FileIdFullDirectoryInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileIdFullDirectoryInformation(inputBuffer);
}
});
decoders.put(FileNamesInformation.class, new FileInformation.Decoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileNamesInformation;
}
@Override
public FileNamesInformation read(Buffer inputBuffer) throws Buffer.BufferException {
return parseFileNamesInformation(inputBuffer);
}
});
FileInformation.Encoder renameCodec = new FileInformation.Encoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileRenameInformation;
}
@Override
public void write(FileRenameInformation info, Buffer outputBuffer) {
writeFileRenameInformation(info, outputBuffer);
}
};
encoders.put(FileRenameInformation.class, renameCodec);
FileInformation.Encoder linkCodec = new FileInformation.Encoder() {
@Override
public FileInformationClass getInformationClass() {
return FileInformationClass.FileLinkInformation;
}
@Override
public void write(FileLinkInformation info, Buffer outputBuffer) {
writeFileRenameInformation(info, outputBuffer);
}
};
encoders.put(FileLinkInformation.class, linkCodec);
}
@SuppressWarnings("unchecked")
public static FileInformation.Encoder getEncoder(F fileInformation) {
return getEncoder((Class) fileInformation.getClass());
}
@SuppressWarnings("unchecked")
public static FileInformation.Encoder getEncoder(Class fileInformationClass) {
FileInformation.Encoder encoder = (FileInformation.Encoder) encoders.get(fileInformationClass);
if (encoder == null) {
throw new IllegalArgumentException("FileInformationClass not supported - " + fileInformationClass);
} else {
return encoder;
}
}
@SuppressWarnings("unchecked")
public static FileInformation.Decoder getDecoder(Class fileInformationClass) {
FileInformation.Decoder decoder = (FileInformation.Decoder) decoders.get(fileInformationClass);
if (decoder == null) {
throw new IllegalArgumentException("FileInformationClass not supported - " + fileInformationClass);
} else {
return decoder;
}
}
/**
* [MS-SMB2] 2.2.34 SMB2 QUERY_DIRECTORY Response for FileInformationClass->FileIdBothDirectoryInformation
*
* @param data
* @param decoder
* @return
*/
public static List parseFileInformationList(
byte[] data, FileInformation.Decoder decoder) {
List _fileInfoList = new ArrayList<>();
Iterator iterator = createFileInformationIterator(data, decoder);
while (iterator.hasNext()) {
_fileInfoList.add(iterator.next());
}
return _fileInfoList;
}
public static Iterator createFileInformationIterator(byte[] data, FileInformation.Decoder decoder) {
return new FileInfoIterator<>(data, decoder, 0);
}
private static class FileInfoIterator implements Iterator {
private final Buffer.PlainBuffer buffer;
private final FileInformation.Decoder decoder;
private int offsetStart;
private F next;
FileInfoIterator(byte[] data, FileInformation.Decoder decoder, int offsetStart) {
this.buffer = new Buffer.PlainBuffer(data, Endian.LE);
this.decoder = decoder;
this.offsetStart = offsetStart;
this.next = prepareNext();
}
@Override
public boolean hasNext() {
return next != null;
}
@Override
public F next() {
if (next == null) {
throw new NoSuchElementException();
}
F fileInfo = this.next;
this.next = prepareNext();
return fileInfo;
}
private F prepareNext() {
try {
F nxt = null;
while (nxt == null && offsetStart != -1) {
buffer.rpos(offsetStart);
F fileInfo = decoder.read(buffer);
int nextOffset = (int) fileInfo.getNextOffset();
if (nextOffset == 0) {
offsetStart = -1;
} else {
offsetStart += nextOffset;
}
nxt = fileInfo;
}
return nxt;
} catch (Buffer.BufferException e) {
throw new SMBRuntimeException(e);
}
}
@Override
public void remove() {
throw new UnsupportedOperationException();
}
}
/**
* [MS-SMB2] 2.2.38 SMB2 QUERY_INFO Response, SMB2_0_INFO_FILE/FileAllInformation
*
* [MS-FSCC] 2.4.2 FileAllInformation
*/
public static FileAllInformation parseFileAllInformation(Buffer buffer) throws Buffer.BufferException {
FileBasicInformation basicInformation = parseFileBasicInformation(buffer);
FileStandardInformation standardInformation = parseFileStandardInformation(buffer);
FileInternalInformation internalInformation = parseFileInternalInformation(buffer);
FileEaInformation eaInformation = parseFileEaInformation(buffer);
FileAccessInformation accessInformation = parseFileAccessInformation(buffer);
FilePositionInformation positionInformation = parseFilePositionInformation(buffer);
FileModeInformation modeInformation = parseFileModeInformation(buffer);
FileAlignmentInformation alignmentInformation = parseFileAlignmentInformation(buffer);
String nameInformation = parseFileNameInformation(buffer);
return new FileAllInformation(
basicInformation,
standardInformation,
internalInformation,
eaInformation,
accessInformation,
positionInformation,
modeInformation,
alignmentInformation,
nameInformation
);
}
private static String parseFileNameInformation(Buffer buffer) throws Buffer.BufferException {
long fileNameLen = buffer.readUInt32(); // File name length
return buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
}
private static FileBasicInformation parseFileBasicInformation(Buffer buffer) throws Buffer.BufferException {
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long fileAttributes = buffer.readUInt32(); // File Attributes
buffer.skip(4); // Reserved (4 bytes)
return new FileBasicInformation(creationTime, lastAccessTime, lastWriteTime, changeTime, fileAttributes);
}
private static FileStandardInformation parseFileStandardInformation(Buffer buffer) throws Buffer.BufferException {
long allocationSize = buffer.readLong();
long endOfFile = buffer.readUInt64();
long numberOfLinks = buffer.readUInt32();
boolean deletePending = buffer.readBoolean();
boolean directory = buffer.readBoolean();
buffer.skip(2); // Reserved
return new FileStandardInformation(allocationSize, endOfFile, numberOfLinks, deletePending, directory);
}
private static FileInternalInformation parseFileInternalInformation(Buffer buffer) throws Buffer.BufferException {
long indexNumber = buffer.readLong();
return new FileInternalInformation(indexNumber);
}
private static FileEaInformation parseFileEaInformation(Buffer buffer) throws Buffer.BufferException {
long eaSize = buffer.readUInt32();
return new FileEaInformation(eaSize);
}
/**
* 2.4.40 FileStreamInformation
*/
private static FileStreamInformation parseFileStreamInformation(Buffer buffer) throws Buffer.BufferException {
List streamList = new ArrayList<>();
long currEntry = 0;
long nextEntry = 0;
do {
currEntry += nextEntry;
buffer.rpos((int) currEntry);
nextEntry = buffer.readUInt32(); //Unsigned 32
long nameLen = buffer.readUInt32(); //Unsigned 32
long size = buffer.readLong(); //Signed 64
long allocSize = buffer.readLong(); //Signed 64
String name = buffer.readString(Charsets.UTF_16LE, (int) nameLen / 2);
streamList.add(new FileStreamInformationItem(size, allocSize, name));
} while (nextEntry != 0);
return new FileStreamInformation(streamList);
}
private static FileAccessInformation parseFileAccessInformation(Buffer buffer) throws Buffer.BufferException {
int accessFlags = (int) buffer.readUInt32();
return new FileAccessInformation(accessFlags);
}
private static FilePositionInformation parseFilePositionInformation(Buffer buffer) throws Buffer.BufferException {
long currentByteOffset = buffer.readLong();
return new FilePositionInformation(currentByteOffset);
}
private static FileModeInformation parseFileModeInformation(Buffer buffer) throws Buffer.BufferException {
int mode = (int) buffer.readUInt32();
return new FileModeInformation(mode);
}
private static FileAlignmentInformation parseFileAlignmentInformation(Buffer buffer) throws Buffer.BufferException {
long alignmentReq = buffer.readUInt32();
return new FileAlignmentInformation(alignmentReq);
}
/**
* 2.4.8 FileBothDirectoryInformation
*/
public static FileBothDirectoryInformation parseFileBothDirectoryInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long endOfFile = buffer.readUInt64();
long allocationSize = buffer.readUInt64();
long fileAttributes = buffer.readUInt32();
long fileNameLen = buffer.readUInt32();
long eaSize = buffer.readUInt32();
byte shortNameLen = buffer.readByte();
buffer.readByte(); // Reserved1 (1)
byte[] shortNameBytes = buffer.readRawBytes(24);// Shortname
String shortName = new String(shortNameBytes, 0, shortNameLen, Charsets.UTF_16LE);
String fileName = buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
FileBothDirectoryInformation fi = new FileBothDirectoryInformation(
nextOffset, fileIndex, fileName,
creationTime, lastAccessTime, lastWriteTime, changeTime,
endOfFile, allocationSize,
fileAttributes,
eaSize,
shortName
);
return fi;
}
/**
* 2.4.10 FileDirectoryInformation
*/
public static FileDirectoryInformation parseFileDirectoryInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long endOfFile = buffer.readUInt64();
long allocationSize = buffer.readUInt64();
long fileAttributes = buffer.readUInt32();
String fileName = parseFileNameInformation(buffer);
FileDirectoryInformation fi = new FileDirectoryInformation(
nextOffset, fileIndex, fileName,
creationTime, lastAccessTime, lastWriteTime, changeTime,
endOfFile, allocationSize,
fileAttributes
);
return fi;
}
/**
* 2.4.14 FileFullDirectoryInformation
*/
public static FileFullDirectoryInformation parseFileFullDirectoryInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long endOfFile = buffer.readUInt64();
long allocationSize = buffer.readUInt64();
long fileAttributes = buffer.readUInt32();
long fileNameLen = buffer.readUInt32();
long eaSize = buffer.readUInt32();
String fileName = buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
FileFullDirectoryInformation fi = new FileFullDirectoryInformation(
nextOffset, fileIndex, fileName,
creationTime, lastAccessTime, lastWriteTime, changeTime,
endOfFile, allocationSize,
fileAttributes,
eaSize
);
return fi;
}
/**
* 2.4.17 FileIdBothDirectoryInformation
*/
public static FileIdBothDirectoryInformation parseFileIdBothDirectoryInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long endOfFile = buffer.readUInt64();
long allocationSize = buffer.readUInt64();
long fileAttributes = buffer.readUInt32();
long fileNameLen = buffer.readUInt32();
long eaSize = buffer.readUInt32();
byte shortNameLen = buffer.readByte();
buffer.readByte(); // Reserved1 (1)
byte[] shortNameBytes = buffer.readRawBytes(24); // Shortname
String shortName = new String(shortNameBytes, 0, shortNameLen, Charsets.UTF_16LE);
buffer.readUInt16(); // Reserved2
long fileId = buffer.readLong();
String fileName = buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
FileIdBothDirectoryInformation fi = new FileIdBothDirectoryInformation(
nextOffset, fileIndex, fileName,
creationTime, lastAccessTime, lastWriteTime, changeTime,
endOfFile, allocationSize,
fileAttributes,
eaSize,
shortName,
fileId
);
return fi;
}
/**
* 2.4.18 FileIdFullDirectoryInformation
*/
public static FileIdFullDirectoryInformation parseFileIdFullDirectoryInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
FileTime creationTime = MsDataTypes.readFileTime(buffer);
FileTime lastAccessTime = MsDataTypes.readFileTime(buffer);
FileTime lastWriteTime = MsDataTypes.readFileTime(buffer);
FileTime changeTime = MsDataTypes.readFileTime(buffer);
long endOfFile = buffer.readUInt64();
long allocationSize = buffer.readUInt64();
long fileAttributes = buffer.readUInt32();
long fileNameLen = buffer.readUInt32();
long eaSize = buffer.readUInt32();
buffer.skip(4); // Reserved
long fileId = buffer.readLong();
String fileName = buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
FileIdFullDirectoryInformation fi = new FileIdFullDirectoryInformation(
nextOffset, fileIndex, fileName,
creationTime, lastAccessTime, lastWriteTime, changeTime,
endOfFile, allocationSize,
fileAttributes,
eaSize,
fileId
);
return fi;
}
/**
* 2.4.26 FileNamesInformation
*/
public static FileNamesInformation parseFileNamesInformation(Buffer buffer) throws Buffer.BufferException {
long nextOffset = buffer.readUInt32();
long fileIndex = buffer.readUInt32();
long fileNameLen = buffer.readUInt32();
String fileName = buffer.readString(Charsets.UTF_16LE, (int) fileNameLen / 2);
return new FileNamesInformation(nextOffset, fileIndex, fileName);
}
/**
* MS-FSCC 2.4.34.2 FileRenameInformation for SMB2
*/
public static void writeFileRenameInformation(FileRenameInformation information, Buffer buffer) {
buffer.putByte((byte) (information.isReplaceIfExists() ? 1 : 0));
buffer.putRawBytes(new byte[]{0, 0, 0, 0, 0, 0, 0}); // reserved
buffer.putUInt64(information.getRootDirectory());
buffer.putUInt32(information.getFileNameLength() * 2L); // unicode
buffer.putRawBytes(information.getFileName().getBytes(Charsets.UTF_16LE));
}
}