rjx-mirror/Ryujinx.Core/OsHle/Services/FspSrv/IFile.cs

using ChocolArm64.Memory;
using Ryujinx.Core.OsHle.Ipc;
using System;
using System.Collections.Generic;
using System.IO;

namespace Ryujinx.Core.OsHle.Services.FspSrv
{
    class IFile : IpcService, IDisposable
    {
        private Dictionary<int, ServiceProcessRequest> m_Commands;

        public override IReadOnlyDictionary<int, ServiceProcessRequest> Commands => m_Commands;

        private Stream BaseStream;

        public event EventHandler<EventArgs> Disposed;

        public string HostPath { get; private set; }

        public IFile(Stream BaseStream, string HostPath)
        {
            m_Commands = new Dictionary<int, ServiceProcessRequest>()
            {
                { 0, Read    },
                { 1, Write   },
                { 2, Flush   },
                { 3, SetSize },
                { 4, GetSize }
            };

            this.BaseStream = BaseStream;
            this.HostPath   = HostPath;
        }

        public long Read(ServiceCtx Context)
        {
            long Position = Context.Request.ReceiveBuff[0].Position;

            long Zero   = Context.RequestData.ReadInt64();
            long Offset = Context.RequestData.ReadInt64();
            long Size   = Context.RequestData.ReadInt64();

            byte[] Data = new byte[Size];

            BaseStream.Seek(Offset, SeekOrigin.Begin);

            int ReadSize = BaseStream.Read(Data, 0, (int)Size);

            AMemoryHelper.WriteBytes(Context.Memory, Position, Data);

            Context.ResponseData.Write((long)ReadSize);

            return 0;
        }

        public long Write(ServiceCtx Context)
        {
            long Position = Context.Request.SendBuff[0].Position;

            long Zero   = Context.RequestData.ReadInt64();
            long Offset = Context.RequestData.ReadInt64();
            long Size   = Context.RequestData.ReadInt64();

            byte[] Data = Context.Memory.ReadBytes(Position, Size);

            BaseStream.Seek(Offset, SeekOrigin.Begin);
            BaseStream.Write(Data, 0, (int)Size);

            return 0;
        }

        public long Flush(ServiceCtx Context)
        {
            BaseStream.Flush();

            return 0;
        }

        public long SetSize(ServiceCtx Context)
        {
            long Size = Context.RequestData.ReadInt64();

            BaseStream.SetLength(Size);

            return 0;
        }

        public long GetSize(ServiceCtx Context)
        {
            Context.ResponseData.Write(BaseStream.Length);

            return 0;
        }

        public void Dispose()
        {
            Dispose(true);
        }

        protected virtual void Dispose(bool disposing)
        {
            if (disposing && BaseStream != null)
            {
                BaseStream.Dispose();

                Disposed?.Invoke(this, EventArgs.Empty);
            }
        }
    }
}