mirror of
https://github.com/jellyfin/jellyfin.git
synced 2026-03-22 01:56:40 +00:00
merge common implementations and server implementations
This commit is contained in:
Luke Pulverenti
@@ -0,0 +1,346 @@
|
||||
// This code is derived from jcifs smb client library <jcifs at samba dot org>
|
||||
// Ported by J. Arturo <webmaster at komodosoft dot net>
|
||||
//
|
||||
// This library is free software; you can redistribute it and/or
|
||||
// modify it under the terms of the GNU Lesser General Public
|
||||
// License as published by the Free Software Foundation; either
|
||||
// version 2.1 of the License, or (at your option) any later version.
|
||||
//
|
||||
// This library is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
// Lesser General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU Lesser General Public
|
||||
// License along with this library; if not, write to the Free Software
|
||||
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
using System;
|
||||
using SharpCifs.Util;
|
||||
|
||||
namespace SharpCifs.Smb
|
||||
{
|
||||
internal abstract class SmbComTransaction : ServerMessageBlock
|
||||
{
|
||||
private static readonly int DefaultMaxDataCount = Config.GetInt("jcifs.smb.client.transaction_buf_size"
|
||||
, TransactionBufSize) - 512;
|
||||
|
||||
private const int PrimarySetupOffset = 61;
|
||||
|
||||
private const int SecondaryParameterOffset = 51;
|
||||
|
||||
private const int DisconnectTid = unchecked(0x01);
|
||||
|
||||
private const int OneWayTransaction = unchecked(0x02);
|
||||
|
||||
private const int PaddingSize = 2;
|
||||
|
||||
private int _flags = unchecked(0x00);
|
||||
|
||||
private int _fid;
|
||||
|
||||
private int _pad;
|
||||
|
||||
private int _pad1;
|
||||
|
||||
private bool _hasMore = true;
|
||||
|
||||
private bool _isPrimary = true;
|
||||
|
||||
private int _bufParameterOffset;
|
||||
|
||||
private int _bufDataOffset;
|
||||
|
||||
internal const int TransactionBufSize = unchecked(0xFFFF);
|
||||
|
||||
internal const byte Trans2FindFirst2 = unchecked(unchecked(0x01));
|
||||
|
||||
internal const byte Trans2FindNext2 = unchecked(unchecked(0x02));
|
||||
|
||||
internal const byte Trans2QueryFsInformation = unchecked(unchecked(0x03));
|
||||
|
||||
internal const byte Trans2QueryPathInformation = unchecked(unchecked(0x05));
|
||||
|
||||
internal const byte Trans2GetDfsReferral = unchecked(unchecked(0x10));
|
||||
|
||||
internal const byte Trans2SetFileInformation = unchecked(unchecked(0x08));
|
||||
|
||||
internal const int NetShareEnum = unchecked(0x0000);
|
||||
|
||||
internal const int NetServerEnum2 = unchecked(0x0068);
|
||||
|
||||
internal const int NetServerEnum3 = unchecked(0x00D7);
|
||||
|
||||
internal const byte TransPeekNamedPipe = unchecked(unchecked(0x23
|
||||
));
|
||||
|
||||
internal const byte TransWaitNamedPipe = unchecked(unchecked(0x53
|
||||
));
|
||||
|
||||
internal const byte TransCallNamedPipe = unchecked(unchecked(0x54
|
||||
));
|
||||
|
||||
internal const byte TransTransactNamedPipe = unchecked(unchecked(0x26));
|
||||
|
||||
protected internal int primarySetupOffset;
|
||||
|
||||
protected internal int secondaryParameterOffset;
|
||||
|
||||
protected internal int ParameterCount;
|
||||
|
||||
protected internal int ParameterOffset;
|
||||
|
||||
protected internal int ParameterDisplacement;
|
||||
|
||||
protected internal int DataCount;
|
||||
|
||||
protected internal int DataOffset;
|
||||
|
||||
protected internal int DataDisplacement;
|
||||
|
||||
internal int TotalParameterCount;
|
||||
|
||||
internal int TotalDataCount;
|
||||
|
||||
internal int MaxParameterCount;
|
||||
|
||||
internal int MaxDataCount = DefaultMaxDataCount;
|
||||
|
||||
internal byte MaxSetupCount;
|
||||
|
||||
internal int Timeout = 0;
|
||||
|
||||
internal int SetupCount = 1;
|
||||
|
||||
internal byte SubCommand;
|
||||
|
||||
internal string Name = string.Empty;
|
||||
|
||||
internal int MaxBufferSize;
|
||||
|
||||
internal byte[] TxnBuf;
|
||||
|
||||
public SmbComTransaction()
|
||||
{
|
||||
// relative to headerStart
|
||||
// set in SmbTransport.sendTransaction() before nextElement called
|
||||
MaxParameterCount = 1024;
|
||||
primarySetupOffset = PrimarySetupOffset;
|
||||
secondaryParameterOffset = SecondaryParameterOffset;
|
||||
}
|
||||
|
||||
internal override void Reset()
|
||||
{
|
||||
base.Reset();
|
||||
_isPrimary = _hasMore = true;
|
||||
}
|
||||
|
||||
internal virtual void Reset(int key, string lastName)
|
||||
{
|
||||
Reset();
|
||||
}
|
||||
|
||||
public virtual bool MoveNext()
|
||||
{
|
||||
return _hasMore;
|
||||
}
|
||||
|
||||
public virtual object Current()
|
||||
{
|
||||
if (_isPrimary)
|
||||
{
|
||||
_isPrimary = false;
|
||||
ParameterOffset = primarySetupOffset + (SetupCount * 2) + 2;
|
||||
if (Command != SmbComNtTransact)
|
||||
{
|
||||
if (Command == SmbComTransaction && IsResponse() == false)
|
||||
{
|
||||
ParameterOffset += StringWireLength(Name, ParameterOffset);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (Command == SmbComNtTransact)
|
||||
{
|
||||
ParameterOffset += 2;
|
||||
}
|
||||
}
|
||||
_pad = ParameterOffset % PaddingSize;
|
||||
_pad = _pad == 0 ? 0 : PaddingSize - _pad;
|
||||
ParameterOffset += _pad;
|
||||
TotalParameterCount = WriteParametersWireFormat(TxnBuf, _bufParameterOffset);
|
||||
_bufDataOffset = TotalParameterCount;
|
||||
// data comes right after data
|
||||
int available = MaxBufferSize - ParameterOffset;
|
||||
ParameterCount = Math.Min(TotalParameterCount, available);
|
||||
available -= ParameterCount;
|
||||
DataOffset = ParameterOffset + ParameterCount;
|
||||
_pad1 = DataOffset % PaddingSize;
|
||||
_pad1 = _pad1 == 0 ? 0 : PaddingSize - _pad1;
|
||||
DataOffset += _pad1;
|
||||
TotalDataCount = WriteDataWireFormat(TxnBuf, _bufDataOffset);
|
||||
DataCount = Math.Min(TotalDataCount, available);
|
||||
}
|
||||
else
|
||||
{
|
||||
if (Command != SmbComNtTransact)
|
||||
{
|
||||
Command = SmbComTransactionSecondary;
|
||||
}
|
||||
else
|
||||
{
|
||||
Command = SmbComNtTransactSecondary;
|
||||
}
|
||||
// totalParameterCount and totalDataCount are set ok from primary
|
||||
ParameterOffset = SecondaryParameterOffset;
|
||||
if ((TotalParameterCount - ParameterDisplacement) > 0)
|
||||
{
|
||||
_pad = ParameterOffset % PaddingSize;
|
||||
_pad = _pad == 0 ? 0 : PaddingSize - _pad;
|
||||
ParameterOffset += _pad;
|
||||
}
|
||||
// caclulate parameterDisplacement before calculating new parameterCount
|
||||
ParameterDisplacement += ParameterCount;
|
||||
int available = MaxBufferSize - ParameterOffset - _pad;
|
||||
ParameterCount = Math.Min(TotalParameterCount - ParameterDisplacement, available);
|
||||
available -= ParameterCount;
|
||||
DataOffset = ParameterOffset + ParameterCount;
|
||||
_pad1 = DataOffset % PaddingSize;
|
||||
_pad1 = _pad1 == 0 ? 0 : PaddingSize - _pad1;
|
||||
DataOffset += _pad1;
|
||||
DataDisplacement += DataCount;
|
||||
available -= _pad1;
|
||||
DataCount = Math.Min(TotalDataCount - DataDisplacement, available);
|
||||
}
|
||||
if ((ParameterDisplacement + ParameterCount) >= TotalParameterCount && (DataDisplacement
|
||||
+ DataCount) >= TotalDataCount)
|
||||
{
|
||||
_hasMore = false;
|
||||
}
|
||||
return this;
|
||||
|
||||
}
|
||||
|
||||
internal override int WriteParameterWordsWireFormat(byte[] dst, int dstIndex)
|
||||
{
|
||||
int start = dstIndex;
|
||||
WriteInt2(TotalParameterCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
WriteInt2(TotalDataCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
if (Command != SmbComTransactionSecondary)
|
||||
{
|
||||
WriteInt2(MaxParameterCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
WriteInt2(MaxDataCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
dst[dstIndex++] = MaxSetupCount;
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
// Reserved1
|
||||
WriteInt2(_flags, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
WriteInt4(Timeout, dst, dstIndex);
|
||||
dstIndex += 4;
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
// Reserved2
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
}
|
||||
WriteInt2(ParameterCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
// writeInt2(( parameterCount == 0 ? 0 : parameterOffset ), dst, dstIndex );
|
||||
WriteInt2(ParameterOffset, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
if (Command == SmbComTransactionSecondary)
|
||||
{
|
||||
WriteInt2(ParameterDisplacement, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
}
|
||||
WriteInt2(DataCount, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
WriteInt2((DataCount == 0 ? 0 : DataOffset), dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
if (Command == SmbComTransactionSecondary)
|
||||
{
|
||||
WriteInt2(DataDisplacement, dst, dstIndex);
|
||||
dstIndex += 2;
|
||||
}
|
||||
else
|
||||
{
|
||||
dst[dstIndex++] = unchecked((byte)SetupCount);
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
// Reserved3
|
||||
dstIndex += WriteSetupWireFormat(dst, dstIndex);
|
||||
}
|
||||
return dstIndex - start;
|
||||
}
|
||||
|
||||
internal override int WriteBytesWireFormat(byte[] dst, int dstIndex)
|
||||
{
|
||||
int start = dstIndex;
|
||||
int p = _pad;
|
||||
if (Command == SmbComTransaction && IsResponse() == false)
|
||||
{
|
||||
dstIndex += WriteString(Name, dst, dstIndex);
|
||||
}
|
||||
if (ParameterCount > 0)
|
||||
{
|
||||
while (p-- > 0)
|
||||
{
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
}
|
||||
// Pad
|
||||
Array.Copy(TxnBuf, _bufParameterOffset, dst, dstIndex, ParameterCount);
|
||||
dstIndex += ParameterCount;
|
||||
}
|
||||
if (DataCount > 0)
|
||||
{
|
||||
p = _pad1;
|
||||
while (p-- > 0)
|
||||
{
|
||||
dst[dstIndex++] = unchecked(unchecked(0x00));
|
||||
}
|
||||
// Pad1
|
||||
Array.Copy(TxnBuf, _bufDataOffset, dst, dstIndex, DataCount);
|
||||
_bufDataOffset += DataCount;
|
||||
dstIndex += DataCount;
|
||||
}
|
||||
return dstIndex - start;
|
||||
}
|
||||
|
||||
internal override int ReadParameterWordsWireFormat(byte[] buffer, int bufferIndex
|
||||
)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
internal override int ReadBytesWireFormat(byte[] buffer, int bufferIndex)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
|
||||
internal abstract int WriteSetupWireFormat(byte[] dst, int dstIndex);
|
||||
|
||||
internal abstract int WriteParametersWireFormat(byte[] dst, int dstIndex);
|
||||
|
||||
internal abstract int WriteDataWireFormat(byte[] dst, int dstIndex);
|
||||
|
||||
internal abstract int ReadSetupWireFormat(byte[] buffer, int bufferIndex, int len
|
||||
);
|
||||
|
||||
internal abstract int ReadParametersWireFormat(byte[] buffer, int bufferIndex, int
|
||||
len);
|
||||
|
||||
internal abstract int ReadDataWireFormat(byte[] buffer, int bufferIndex, int len);
|
||||
|
||||
public override string ToString()
|
||||
{
|
||||
return base.ToString() + ",totalParameterCount=" + TotalParameterCount
|
||||
+ ",totalDataCount=" + TotalDataCount + ",maxParameterCount=" + MaxParameterCount
|
||||
+ ",maxDataCount=" + MaxDataCount + ",maxSetupCount=" + (int)MaxSetupCount + ",flags=0x"
|
||||
+ Hexdump.ToHexString(_flags, 2) + ",timeout=" + Timeout + ",parameterCount=" +
|
||||
ParameterCount + ",parameterOffset=" + ParameterOffset + ",parameterDisplacement="
|
||||
+ ParameterDisplacement + ",dataCount=" + DataCount + ",dataOffset=" + DataOffset
|
||||
+ ",dataDisplacement=" + DataDisplacement + ",setupCount=" + SetupCount + ",pad="
|
||||
+ _pad + ",pad1=" + _pad1;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user