jellyfin/Emby.Common.Implementations/IO/SharpCifs/Smb/SmbComTreeConnectAndX.cs

// 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;
using SharpCifs.Util.Sharpen;

namespace SharpCifs.Smb
{
    internal class SmbComTreeConnectAndX : AndXServerMessageBlock
    {
        private static readonly bool DisablePlainTextPasswords
            = Config.GetBoolean("jcifs.smb.client.disablePlainTextPasswords", true);

        private SmbSession _session;

        private bool _disconnectTid = false;

        private string _service;

        private byte[] _password;

        private int _passwordLength;

        internal string path;

        private static byte[] _batchLimits = { 1, 1, 1, 1, 1, 1, 1, 1, 0 };

        static SmbComTreeConnectAndX()
        {
            string s;
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.CheckDirectory")) != null)
            {
                _batchLimits[0] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.CreateDirectory")) != null)
            {
                _batchLimits[2] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.Delete")) != null)
            {
                _batchLimits[3] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.DeleteDirectory")) != null)
            {
                _batchLimits[4] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.OpenAndX")) != null)
            {
                _batchLimits[5] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.Rename")) != null)
            {
                _batchLimits[6] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.Transaction")) != null)
            {
                _batchLimits[7] = byte.Parse(s);
            }
            if ((s = Config.GetProperty("jcifs.smb.client.TreeConnectAndX.QueryInformation")) != null)
            {
                _batchLimits[8] = byte.Parse(s);
            }
        }

        internal SmbComTreeConnectAndX(SmbSession session,
                                       string path,
                                       string service,
                                       ServerMessageBlock andx) : base(andx)
        {
            this._session = session;
            this.path = path;
            this._service = service;
            Command = SmbComTreeConnectAndx;
        }

        internal override int GetBatchLimit(byte command)
        {
            int c = command & unchecked(0xFF);
            switch (c)
            {
                case SmbComCheckDirectory:
                    {
                        // why isn't this just return batchLimits[c]?
                        return _batchLimits[0];
                    }

                case SmbComCreateDirectory:
                    {
                        return _batchLimits[2];
                    }

                case SmbComDelete:
                    {
                        return _batchLimits[3];
                    }

                case SmbComDeleteDirectory:
                    {
                        return _batchLimits[4];
                    }

                case SmbComOpenAndx:
                    {
                        return _batchLimits[5];
                    }

                case SmbComRename:
                    {
                        return _batchLimits[6];
                    }

                case SmbComTransaction:
                    {
                        return _batchLimits[7];
                    }

                case SmbComQueryInformation:
                    {
                        return _batchLimits[8];
                    }
            }
            return 0;
        }

        internal override int WriteParameterWordsWireFormat(byte[] dst, int dstIndex)
        {
            if (_session.transport.Server.Security == SmbConstants.SecurityShare
                && (_session.Auth.HashesExternal || _session.Auth.Password.Length > 0))
            {
                if (_session.transport.Server.EncryptedPasswords)
                {
                    // encrypted
                    _password = _session.Auth.GetAnsiHash(_session.transport.Server.EncryptionKey);
                    _passwordLength = _password.Length;
                }
                else
                {
                    if (DisablePlainTextPasswords)
                    {
                        throw new RuntimeException("Plain text passwords are disabled");
                    }
                    // plain text
                    _password = new byte[(_session.Auth.Password.Length + 1) * 2];
                    _passwordLength = WriteString(_session.Auth.Password, _password, 0);
                }
            }
            else
            {
                // no password in tree connect
                _passwordLength = 1;
            }
            dst[dstIndex++] = _disconnectTid
                                    ? unchecked((byte)unchecked(0x01))
                                    : unchecked((byte)unchecked(0x00));
            dst[dstIndex++] = unchecked(unchecked(0x00));
            WriteInt2(_passwordLength, dst, dstIndex);
            return 4;
        }

        internal override int WriteBytesWireFormat(byte[] dst, int dstIndex)
        {
            int start = dstIndex;
            if (_session.transport.Server.Security == SmbConstants.SecurityShare
                && (_session.Auth.HashesExternal || _session.Auth.Password.Length > 0))
            {
                Array.Copy(_password,
                           0,
                           dst,
                           dstIndex,
                           _passwordLength);
                dstIndex += _passwordLength;
            }
            else
            {
                // no password in tree connect
                dst[dstIndex++] = unchecked(unchecked(0x00));
            }
            dstIndex += WriteString(path, dst, dstIndex);
            try
            {
                //Array.Copy(Runtime.GetBytesForString(_service, "ASCII"), 0, dst, dstIndex
                // , _service.Length);
                Array.Copy(Runtime.GetBytesForString(_service, "UTF-8"),
                           0,
                           dst,
                           dstIndex,
                           _service.Length);
            }
            catch (UnsupportedEncodingException)
            {
                return 0;
            }
            dstIndex += _service.Length;
            dst[dstIndex++] = unchecked((byte)('\0'));
            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 = "SmbComTreeConnectAndX["
                                    + base.ToString()
                                    + ",disconnectTid=" + _disconnectTid
                                    + ",passwordLength=" + _passwordLength
                                    + ",password=" + Hexdump.ToHexString(_password, _passwordLength, 0)
                                    + ",path=" + path
                                    + ",service=" + _service + "]";
            return result;
        }
    }
}