jellyfin/Emby.Common.Implementations/IO/SharpCifs/Smb/SmbRandomAccessFile.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 System.IO;
using System.Text;
using SharpCifs.Util;
using SharpCifs.Util.Sharpen;

namespace SharpCifs.Smb
{
    public class SmbRandomAccessFile //: DataOutput, DataInput
    {
        private const int WriteOptions = unchecked(0x0842);

        private SmbFile _file;

        private long _fp;

        private int _openFlags;

        private int _access;

        private int _readSize;

        private int _writeSize;

        private int _ch;

        private int _options;

        private byte[] _tmp = new byte[8];

        private SmbComWriteAndXResponse _writeAndxResp;

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        /// <exception cref="System.UriFormatException"></exception>
        /// <exception cref="UnknownHostException"></exception>
        public SmbRandomAccessFile(string url, string mode, int shareAccess) 
            : this(new SmbFile(url, string.Empty, null, shareAccess), mode)
        {
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        /// <exception cref="System.UriFormatException"></exception>
        /// <exception cref="UnknownHostException"></exception>
        public SmbRandomAccessFile(SmbFile file, string mode)
        {
            this._file = file;
            if (mode.Equals("r"))
            {
                _openFlags = SmbFile.OCreat | SmbFile.ORdonly;
            }
            else
            {
                if (mode.Equals("rw"))
                {
                    _openFlags = SmbFile.OCreat | SmbFile.ORdwr | SmbFile.OAppend;
                    _writeAndxResp = new SmbComWriteAndXResponse();
                    _options = WriteOptions;
                    _access = SmbConstants.FileReadData | SmbConstants.FileWriteData;
                }
                else
                {
                    throw new ArgumentException("Invalid mode");
                }
            }
            file.Open(_openFlags, _access, SmbFile.AttrNormal, _options);
            _readSize = file.Tree.Session.transport.RcvBufSize - 70;
            _writeSize = file.Tree.Session.transport.SndBufSize - 70;
            _fp = 0L;
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual int Read()
        {
            if (Read(_tmp, 0, 1) == -1)
            {
                return -1;
            }
            return _tmp[0] & unchecked(0xFF);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual int Read(byte[] b)
        {
            return Read(b, 0, b.Length);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual int Read(byte[] b, int off, int len)
        {
            if (len <= 0)
            {
                return 0;
            }
            long start = _fp;
            // ensure file is open
            if (_file.IsOpen() == false)
            {
                _file.Open(_openFlags, 0, SmbFile.AttrNormal, _options);
            }
            int r;
            int n;
            SmbComReadAndXResponse response = new SmbComReadAndXResponse(b, off);
            do
            {
                r = len > _readSize ? _readSize : len;
                _file.Send(new SmbComReadAndX(_file.Fid, _fp, r, null), response);
                if ((n = response.DataLength) <= 0)
                {
                    return (int)((_fp - start) > 0L ? _fp - start : -1);
                }
                _fp += n;
                len -= n;
                response.Off += n;
            }
            while (len > 0 && n == r);
            return (int)(_fp - start);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void ReadFully(byte[] b)
        {
            ReadFully(b, 0, b.Length);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void ReadFully(byte[] b, int off, int len)
        {
            int n = 0;
            int count;
            do
            {
                count = Read(b, off + n, len - n);
                if (count < 0)
                {
                    throw new SmbException("EOF");
                }
                n += count;
                _fp += count;
            }
            while (n < len);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual int SkipBytes(int n)
        {
            if (n > 0)
            {
                _fp += n;
                return n;
            }
            return 0;
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void Write(int b)
        {
            _tmp[0] = unchecked((byte)b);
            Write(_tmp, 0, 1);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void Write(byte[] b)
        {
            Write(b, 0, b.Length);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void Write(byte[] b, int off, int len)
        {
            if (len <= 0)
            {
                return;
            }
            // ensure file is open
            if (_file.IsOpen() == false)
            {
                _file.Open(_openFlags, 0, SmbFile.AttrNormal, _options);
            }
            int w;
            do
            {
                w = len > _writeSize ? _writeSize : len;
                _file.Send(new SmbComWriteAndX(_file.Fid, _fp, len - w, b, off, w, null), 
                           _writeAndxResp);
                _fp += _writeAndxResp.Count;
                len -= (int)_writeAndxResp.Count;
                off += (int)_writeAndxResp.Count;
            }
            while (len > 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual long GetFilePointer()
        {
            return _fp;
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void Seek(long pos)
        {
            _fp = pos;
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual long Length()
        {
            return _file.Length();
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void SetLength(long newLength)
        {
            // ensure file is open
            if (_file.IsOpen() == false)
            {
                _file.Open(_openFlags, 0, SmbFile.AttrNormal, _options);
            }
            SmbComWriteResponse rsp = new SmbComWriteResponse();
            _file.Send(new SmbComWrite(_file.Fid, 
                                       (int)(newLength & unchecked(0xFFFFFFFFL)), 
                                       0, 
                                       _tmp, 
                                       0, 
                                       0), 
                       rsp);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public virtual void Close()
        {
            _file.Close();
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public bool ReadBoolean()
        {
            if ((Read(_tmp, 0, 1)) < 0)
            {
                throw new SmbException("EOF");
            }
            return _tmp[0] != unchecked(unchecked(0x00));
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public byte ReadByte()
        {
            if ((Read(_tmp, 0, 1)) < 0)
            {
                throw new SmbException("EOF");
            }
            return _tmp[0];
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public int ReadUnsignedByte()
        {
            if ((Read(_tmp, 0, 1)) < 0)
            {
                throw new SmbException("EOF");
            }
            return _tmp[0] & unchecked(0xFF);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public short ReadShort()
        {
            if ((Read(_tmp, 0, 2)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_uint16be(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public int ReadUnsignedShort()
        {
            if ((Read(_tmp, 0, 2)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_uint16be(_tmp, 0) & unchecked(0xFFFF);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public char ReadChar()
        {
            if ((Read(_tmp, 0, 2)) < 0)
            {
                throw new SmbException("EOF");
            }
            return (char)Encdec.Dec_uint16be(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public int ReadInt()
        {
            if ((Read(_tmp, 0, 4)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_uint32be(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public long ReadLong()
        {
            if ((Read(_tmp, 0, 8)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_uint64be(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public float ReadFloat()
        {
            if ((Read(_tmp, 0, 4)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_floatbe(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public double ReadDouble()
        {
            if ((Read(_tmp, 0, 8)) < 0)
            {
                throw new SmbException("EOF");
            }
            return Encdec.Dec_doublebe(_tmp, 0);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public string ReadLine()
        {
            StringBuilder input = new StringBuilder();
            int c = -1;
            bool eol = false;
            while (!eol)
            {
                switch (c = Read())
                {
                    case -1:
                    case '\n':
                        {
                            eol = true;
                            break;
                        }

                    case '\r':
                        {
                            eol = true;
                            long cur = _fp;
                            if (Read() != '\n')
                            {
                                _fp = cur;
                            }
                            break;
                        }

                    default:
                        {
                            input.Append((char)c);
                            break;
                        }
                }
            }
            if ((c == -1) && (input.Length == 0))
            {
                return null;
            }
            return input.ToString();
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public string ReadUtf()
        {
            int size = ReadUnsignedShort();
            byte[] b = new byte[size];
            Read(b, 0, size);
            try
            {
                return Encdec.Dec_utf8(b, 0, size);
            }
            catch (IOException ioe)
            {
                throw new SmbException(string.Empty, ioe);
            }
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteBoolean(bool v)
        {
            _tmp[0] = unchecked((byte)(v ? 1 : 0));
            Write(_tmp, 0, 1);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteByte(int v)
        {
            _tmp[0] = unchecked((byte)v);
            Write(_tmp, 0, 1);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteShort(int v)
        {
            Encdec.Enc_uint16be((short)v, _tmp, 0);
            Write(_tmp, 0, 2);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteChar(int v)
        {
            Encdec.Enc_uint16be((short)v, _tmp, 0);
            Write(_tmp, 0, 2);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteInt(int v)
        {
            Encdec.Enc_uint32be(v, _tmp, 0);
            Write(_tmp, 0, 4);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteLong(long v)
        {
            Encdec.Enc_uint64be(v, _tmp, 0);
            Write(_tmp, 0, 8);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteFloat(float v)
        {
            Encdec.Enc_floatbe(v, _tmp, 0);
            Write(_tmp, 0, 4);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteDouble(double v)
        {
            Encdec.Enc_doublebe(v, _tmp, 0);
            Write(_tmp, 0, 8);
        }

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteBytes(string s)
        {
            byte[] b = Runtime.GetBytesForString(s);
            Write(b, 0, b.Length);
        }


        /*	
        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteChars(string s)
        {
            int clen = s.Length;
            int blen = 2 * clen;
            byte[] b = new byte[blen];
            char[] c = new char[clen];
            Sharpen.Runtime.GetCharsForString(s, 0, clen, c, 0);
            for (int i = 0, j = 0; i < clen; i++)
            {
                b[j++] = unchecked((byte)((char)(((uchar)c[i]) >> 8)));
                b[j++] = unchecked((byte)((char)(((uchar)c[i]) >> 0)));
            }
            Write(b, 0, blen);
        }
        */

        /// <exception cref="SharpCifs.Smb.SmbException"></exception>
        public void WriteUtf(string str)
        {
            int len = str.Length;
            int ch;
            int size = 0;
            byte[] dst;
            for (int i = 0; i < len; i++)
            {
                ch = str[i];
                size += ch > unchecked(0x07F) 
                            ? (ch > unchecked(0x7FF) 
                                    ? 3 
                                    : 2) 
                            : 1;
            }
            dst = new byte[size];
            WriteShort(size);
            try
            {
                Encdec.Enc_utf8(str, dst, 0, size);
            }
            catch (IOException ioe)
            {
                throw new SmbException(string.Empty, ioe);
            }
            Write(dst, 0, size);
        }
    }
}