jellyfin-server/Emby.Common.Implementations/IO/SharpCifs/Smb/SmbComSessionSetupAndX.cs
2017-06-21 02:46:57 -04:00

245 lines
9.7 KiB
C#

// This code is derived from jcifs smb client library <jcifs at samba dot org>
//
// 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
//
// Ported to C# by J. Arturo <webmaster at komodosoft.net>
using System;
using SharpCifs.Util.Sharpen;
namespace SharpCifs.Smb
{
internal class SmbComSessionSetupAndX : AndXServerMessageBlock
{
private static readonly int BatchLimit
= Config.GetInt("jcifs.smb.client.SessionSetupAndX.TreeConnectAndX", 1);
private static readonly bool DisablePlainTextPasswords
= Config.GetBoolean("jcifs.smb.client.disablePlainTextPasswords", true);
private byte[] _lmHash;
private byte[] _ntHash;
private byte[] _blob;
private int _sessionKey;
private int _capabilities;
private string _accountName;
private string _primaryDomain;
internal SmbSession Session;
internal object Cred;
/// <exception cref="SharpCifs.Smb.SmbException"></exception>
internal SmbComSessionSetupAndX(SmbSession session,
ServerMessageBlock andx,
object cred) : base(andx)
{
Command = SmbComSessionSetupAndx;
this.Session = session;
this.Cred = cred;
_sessionKey = session.transport.SessionKey;
_capabilities = session.transport.Capabilities;
if (session.transport.Server.Security == SmbConstants.SecurityUser)
{
if (cred is NtlmPasswordAuthentication)
{
NtlmPasswordAuthentication auth = (NtlmPasswordAuthentication)cred;
if (auth == NtlmPasswordAuthentication.Anonymous)
{
_lmHash = new byte[0];
_ntHash = new byte[0];
_capabilities &= ~SmbConstants.CapExtendedSecurity;
}
else
{
if (session.transport.Server.EncryptedPasswords)
{
_lmHash = auth.GetAnsiHash(session.transport.Server.EncryptionKey);
_ntHash = auth.GetUnicodeHash(session.transport.Server.EncryptionKey);
// prohibit HTTP auth attempts for the null session
if (_lmHash.Length == 0 && _ntHash.Length == 0)
{
throw new RuntimeException("Null setup prohibited.");
}
}
else
{
if (DisablePlainTextPasswords)
{
throw new RuntimeException("Plain text passwords are disabled");
}
if (UseUnicode)
{
// plain text
string password = auth.GetPassword();
_lmHash = new byte[0];
_ntHash = new byte[(password.Length + 1) * 2];
WriteString(password, _ntHash, 0);
}
else
{
// plain text
string password = auth.GetPassword();
_lmHash = new byte[(password.Length + 1) * 2];
_ntHash = new byte[0];
WriteString(password, _lmHash, 0);
}
}
}
_accountName = auth.Username;
if (UseUnicode)
{
_accountName = _accountName.ToUpper();
}
_primaryDomain = auth.Domain.ToUpper();
}
else
{
if (cred is byte[])
{
_blob = (byte[])cred;
}
else
{
throw new SmbException("Unsupported credential type");
}
}
}
else
{
if (session.transport.Server.Security == SmbConstants.SecurityShare)
{
if (cred is NtlmPasswordAuthentication)
{
NtlmPasswordAuthentication auth = (NtlmPasswordAuthentication)cred;
_lmHash = new byte[0];
_ntHash = new byte[0];
_accountName = auth.Username;
if (UseUnicode)
{
_accountName = _accountName.ToUpper();
}
_primaryDomain = auth.Domain.ToUpper();
}
else
{
throw new SmbException("Unsupported credential type");
}
}
else
{
throw new SmbException("Unsupported");
}
}
}
internal override int GetBatchLimit(byte command)
{
return command == SmbComTreeConnectAndx ? BatchLimit : 0;
}
internal override int WriteParameterWordsWireFormat(byte[] dst, int dstIndex)
{
int start = dstIndex;
WriteInt2(Session.transport.SndBufSize, dst, dstIndex);
dstIndex += 2;
WriteInt2(Session.transport.MaxMpxCount, dst, dstIndex);
dstIndex += 2;
WriteInt2(SmbConstants.VcNumber, dst, dstIndex);
dstIndex += 2;
WriteInt4(_sessionKey, dst, dstIndex);
dstIndex += 4;
if (_blob != null)
{
WriteInt2(_blob.Length, dst, dstIndex);
dstIndex += 2;
}
else
{
WriteInt2(_lmHash.Length, dst, dstIndex);
dstIndex += 2;
WriteInt2(_ntHash.Length, dst, dstIndex);
dstIndex += 2;
}
dst[dstIndex++] = unchecked(unchecked(0x00));
dst[dstIndex++] = unchecked(unchecked(0x00));
dst[dstIndex++] = unchecked(unchecked(0x00));
dst[dstIndex++] = unchecked(unchecked(0x00));
WriteInt4(_capabilities, dst, dstIndex);
dstIndex += 4;
return dstIndex - start;
}
internal override int WriteBytesWireFormat(byte[] dst, int dstIndex)
{
int start = dstIndex;
if (_blob != null)
{
Array.Copy(_blob, 0, dst, dstIndex, _blob.Length);
dstIndex += _blob.Length;
}
else
{
Array.Copy(_lmHash, 0, dst, dstIndex, _lmHash.Length);
dstIndex += _lmHash.Length;
Array.Copy(_ntHash, 0, dst, dstIndex, _ntHash.Length);
dstIndex += _ntHash.Length;
dstIndex += WriteString(_accountName, dst, dstIndex);
dstIndex += WriteString(_primaryDomain, dst, dstIndex);
}
dstIndex += WriteString(SmbConstants.NativeOs, dst, dstIndex);
dstIndex += WriteString(SmbConstants.NativeLanman, dst, dstIndex);
return dstIndex - start;
}
internal override int ReadParameterWordsWireFormat(byte[] buffer, int bufferIndex)
{
return 0;
}
internal override int ReadBytesWireFormat(byte[] buffer, int bufferIndex)
{
return 0;
}
public override string ToString()
{
string result = "SmbComSessionSetupAndX["
+ base.ToString()
+ ",snd_buf_size=" + Session.transport.SndBufSize
+ ",maxMpxCount=" + Session.transport.MaxMpxCount
+ ",VC_NUMBER=" + SmbConstants.VcNumber
+ ",sessionKey=" + _sessionKey
+ ",lmHash.length=" + (_lmHash == null
? 0
: _lmHash.Length)
+ ",ntHash.length=" + (_ntHash == null
? 0
: _ntHash.Length)
+ ",capabilities=" + _capabilities
+ ",accountName=" + _accountName
+ ",primaryDomain=" + _primaryDomain
+ ",NATIVE_OS=" + SmbConstants.NativeOs
+ ",NATIVE_LANMAN=" + SmbConstants.NativeLanman + "]";
return result;
}
}
}