memstream.c File Reference

#include <rtl.h>
#include <debug.h>

Include dependency graph for memstream.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
static PRTL_MEMORY_STREAM	IStream_To_RTL_MEMORY_STREAM (_In_ IStream *Interface)
VOID NTAPI	RtlInitMemoryStream (_Out_ PRTL_MEMORY_STREAM Stream)
VOID NTAPI	RtlInitOutOfProcessMemoryStream (_Out_ PRTL_MEMORY_STREAM Stream)
VOID NTAPI	RtlFinalReleaseOutOfProcessMemoryStream (_In_ PRTL_MEMORY_STREAM Stream)
HRESULT NTAPI	RtlQueryInterfaceMemoryStream (_In_ IStream *This, _In_ REFIID RequestedIid, _Outptr_ PVOID *ResultObject)
ULONG NTAPI	RtlAddRefMemoryStream (_In_ IStream *This)
ULONG NTAPI	RtlReleaseMemoryStream (_In_ IStream *This)
HRESULT NTAPI	RtlReadMemoryStream (_In_ IStream *This, _Out_writes_bytes_(Length) PVOID Buffer, _In_ ULONG Length, _Out_opt_ PULONG BytesRead)
HRESULT NTAPI	RtlReadOutOfProcessMemoryStream (_In_ IStream *This, _Out_writes_bytes_(Length) PVOID Buffer, _In_ ULONG Length, _Out_opt_ PULONG BytesRead)
HRESULT NTAPI	RtlSeekMemoryStream (_In_ IStream *This, _In_ LARGE_INTEGER RelativeOffset, _In_ ULONG Origin, _Out_opt_ PULARGE_INTEGER ResultOffset)
HRESULT NTAPI	RtlCopyMemoryStreamTo (_In_ IStream *This, _In_ IStream *Target, _In_ ULARGE_INTEGER Length, _Out_opt_ PULARGE_INTEGER BytesRead, _Out_opt_ PULARGE_INTEGER BytesWritten)
HRESULT NTAPI	RtlStatMemoryStream (_In_ IStream *This, _Out_ STATSTG *Stats, _In_ ULONG Flags)
HRESULT NTAPI	RtlWriteMemoryStream (_In_ IStream *This, _In_reads_bytes_(Length) CONST VOID *Buffer, _In_ ULONG Length, _Out_opt_ PULONG BytesWritten)
HRESULT NTAPI	RtlSetMemoryStreamSize (_In_ IStream *This, _In_ ULARGE_INTEGER NewSize)
HRESULT NTAPI	RtlCommitMemoryStream (_In_ IStream *This, _In_ ULONG CommitFlags)
HRESULT NTAPI	RtlRevertMemoryStream (_In_ IStream *This)
HRESULT NTAPI	RtlLockMemoryStreamRegion (_In_ IStream *This, _In_ ULARGE_INTEGER Offset, _In_ ULARGE_INTEGER Length, _In_ ULONG LockType)
HRESULT NTAPI	RtlUnlockMemoryStreamRegion (_In_ IStream *This, _In_ ULARGE_INTEGER Offset, _In_ ULARGE_INTEGER Length, _In_ ULONG LockType)
HRESULT NTAPI	RtlCloneMemoryStream (_In_ IStream *This, _Outptr_ IStream **ResultStream)
NTSTATUS NTAPI	RtlCopyMappedMemory (_Out_writes_bytes_all_(Size) PVOID Destination, _In_reads_bytes_(Size) const VOID *Source, _In_ SIZE_T Size)

Variables
const struct IStreamVtbl	RtlMemoryStreamVtbl
const struct IStreamVtbl	RtlOutOfProcessMemoryStreamVtbl

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 13 of file memstream.c.

Function Documentation

IStream_To_RTL_MEMORY_STREAM()

static PRTL_MEMORY_STREAM IStream_To_RTL_MEMORY_STREAM ( _In_ IStream *  Interface )

static

Definition at line 58 of file memstream.c.

{
    if (Interface == NULL)
        return NULL;
 
    return CONTAINING_RECORD(Interface, RTL_MEMORY_STREAM, Vtbl);
}

Referenced by RtlAddRefMemoryStream(), RtlReadMemoryStream(), RtlReadOutOfProcessMemoryStream(), RtlReleaseMemoryStream(), RtlSeekMemoryStream(), and RtlStatMemoryStream().

RtlAddRefMemoryStream()

ULONG NTAPI RtlAddRefMemoryStream

(

_In_ IStream *

This

)

Definition at line 131 of file memstream.c.

{
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
 
    return InterlockedIncrement(&Stream->RefCount);
}

RtlCloneMemoryStream()

HRESULT NTAPI RtlCloneMemoryStream	(	_In_ IStream *	This,
		_Outptr_ IStream **	ResultStream
	)

Definition at line 464 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(ResultStream);
 
    return E_NOTIMPL;
}

RtlCommitMemoryStream()

HRESULT NTAPI RtlCommitMemoryStream	(	_In_ IStream *	This,
		_In_ ULONG	CommitFlags
	)

Definition at line 398 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(CommitFlags);
 
    return E_NOTIMPL;
}

RtlCopyMappedMemory()

NTSTATUS NTAPI RtlCopyMappedMemory	(	_Out_writes_bytes_all_(Size) PVOID	Destination,
		_In_reads_bytes_(Size) const VOID *	Source,
		_In_ SIZE_T	Size
	)

Definition at line 479 of file memstream.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
    _SEH2_TRY
    {
        RtlCopyMemory(Destination, Source, Size);
    }
    _SEH2_EXCEPT(_SEH2_GetExceptionCode() == STATUS_IN_PAGE_ERROR
                    ? EXCEPTION_EXECUTE_HANDLER
                    : EXCEPTION_CONTINUE_SEARCH)
    {
        Status = _SEH2_GetExceptionCode();
    }
    _SEH2_END;
    return Status;
}

RtlCopyMemoryStreamTo()

HRESULT NTAPI RtlCopyMemoryStreamTo	(	_In_ IStream *	This,
		_In_ IStream *	Target,
		_In_ ULARGE_INTEGER	Length,
		_Out_opt_ PULARGE_INTEGER	BytesRead,
		_Out_opt_ PULARGE_INTEGER	BytesWritten
	)

Definition at line 282 of file memstream.c.

{
    CHAR Buffer[1024];
    ULONGLONG TotalSize;
    ULONG Left, Amount;
    HRESULT Result;
 
    if (BytesRead)
        BytesRead->QuadPart = 0;
    if (BytesWritten)
        BytesWritten->QuadPart = 0;
 
    if (!Target)
        return S_OK;
 
    if (!Length.QuadPart)
        return S_OK;
 
    /* Copy data */
    TotalSize = Length.QuadPart;
    while (TotalSize)
    {
        Left = (ULONG)min(TotalSize, sizeof(Buffer));
 
        /* Read */
        Result = IStream_Read(This, Buffer, Left, &Amount);
        if (BytesRead)
            BytesRead->QuadPart += Amount;
        if (FAILED(Result) || Amount == 0)
            break;
 
        Left = Amount;
 
        /* Write */
        Result = IStream_Write(Target, Buffer, Left, &Amount);
        if (BytesWritten)
            BytesWritten->QuadPart += Amount;
        if (FAILED(Result) || Amount != Left)
            break;
 
        TotalSize -= Left;
    }
    return Result;
}

RtlFinalReleaseOutOfProcessMemoryStream()

VOID NTAPI RtlFinalReleaseOutOfProcessMemoryStream

(

_In_ PRTL_MEMORY_STREAM

Stream

)

Definition at line 97 of file memstream.c.

{
    UNIMPLEMENTED;
}

Referenced by CustomFinalReleaseOutOfProcessMemoryStream(), RtlInitOutOfProcessMemoryStream(), and test_OutOfProcess().

RtlInitMemoryStream()

VOID NTAPI RtlInitMemoryStream

(

_Out_ PRTL_MEMORY_STREAM

Stream

)

Definition at line 72 of file memstream.c.

{
    RtlZeroMemory(Stream, sizeof(RTL_MEMORY_STREAM));
    Stream->Vtbl = &RtlMemoryStreamVtbl;
}

Referenced by test_InProcess().

RtlInitOutOfProcessMemoryStream()

VOID NTAPI RtlInitOutOfProcessMemoryStream

(

_Out_ PRTL_MEMORY_STREAM

Stream

)

Definition at line 84 of file memstream.c.

{
    RtlZeroMemory(Stream, sizeof(RTL_MEMORY_STREAM));
    Stream->Vtbl = &RtlOutOfProcessMemoryStreamVtbl;
    Stream->FinalRelease = RtlFinalReleaseOutOfProcessMemoryStream;
}

Referenced by test_OutOfProcess().

RtlLockMemoryStreamRegion()

HRESULT NTAPI RtlLockMemoryStreamRegion	(	_In_ IStream *	This,
		_In_ ULARGE_INTEGER	Offset,
		_In_ ULARGE_INTEGER	Length,
		_In_ ULONG	LockType
	)

Definition at line 426 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(Offset);
    UNREFERENCED_PARAMETER(Length);
    UNREFERENCED_PARAMETER(LockType);
 
    return E_NOTIMPL;
}

RtlQueryInterfaceMemoryStream()

HRESULT NTAPI RtlQueryInterfaceMemoryStream	(	_In_ IStream *	This,
		_In_ REFIID	RequestedIid,
		_Outptr_ PVOID *	ResultObject
	)

Definition at line 108 of file memstream.c.

{
    if (IsEqualGUID(RequestedIid, &IID_IUnknown) ||
        IsEqualGUID(RequestedIid, &IID_ISequentialStream) ||
        IsEqualGUID(RequestedIid, &IID_IStream))
    {
        IStream_AddRef(This);
        *ResultObject = This;
        return S_OK;
    }
 
    *ResultObject = NULL;
    return E_NOINTERFACE;
}

RtlReadMemoryStream()

HRESULT NTAPI RtlReadMemoryStream	(	_In_ IStream *	This,
		_Out_writes_bytes_(Length) PVOID	Buffer,
		_In_ ULONG	Length,
		_Out_opt_ PULONG	BytesRead
	)

Definition at line 166 of file memstream.c.

{
    ULONG CopyLength;
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
    SIZE_T Available = (PUCHAR)Stream->End - (PUCHAR)Stream->Current;
 
    if (BytesRead)
        *BytesRead = 0;
 
    if (!Length)
        return S_OK;
 
    CopyLength = min(Available, Length);
 
    RtlMoveMemory(Buffer, Stream->Current, CopyLength);
 
    Stream->Current = (PUCHAR)Stream->Current + CopyLength;
 
    *BytesRead = CopyLength;
 
    return S_OK;
}

RtlReadOutOfProcessMemoryStream()

HRESULT NTAPI RtlReadOutOfProcessMemoryStream	(	_In_ IStream *	This,
		_Out_writes_bytes_(Length) PVOID	Buffer,
		_In_ ULONG	Length,
		_Out_opt_ PULONG	BytesRead
	)

Definition at line 198 of file memstream.c.

{
    NTSTATUS Status;
    ULONG CopyLength;
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
    SIZE_T Available = (PUCHAR)Stream->End - (PUCHAR)Stream->Current;
    SIZE_T LocalBytesRead = 0;
 
    if (BytesRead)
        *BytesRead = 0;
 
    if (!Length)
        return S_OK;
 
    CopyLength = min(Available, Length);
 
    Status = NtReadVirtualMemory(Stream->ProcessHandle,
                                 Stream->Current,
                                 Buffer,
                                 CopyLength,
                                 &LocalBytesRead);
 
    if (NT_SUCCESS(Status))
    {
        Stream->Current = (PUCHAR)Stream->Current + LocalBytesRead;
        if (BytesRead)
            *BytesRead = (ULONG)LocalBytesRead;
    }
 
    return HRESULT_FROM_WIN32(RtlNtStatusToDosError(Status));
}

RtlReleaseMemoryStream()

ULONG NTAPI RtlReleaseMemoryStream

(

_In_ IStream *

This

)

Definition at line 144 of file memstream.c.

{
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
    LONG Result;
 
    Result = InterlockedDecrement(&Stream->RefCount);
 
    if (Result == 0)
    {
        if (Stream->FinalRelease)
            Stream->FinalRelease(Stream);
    }
 
    return Result;
}

RtlRevertMemoryStream()

HRESULT NTAPI RtlRevertMemoryStream

(

_In_ IStream *

This

)

Definition at line 413 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
 
    return E_NOTIMPL;
}

RtlSeekMemoryStream()

HRESULT NTAPI RtlSeekMemoryStream	(	_In_ IStream *	This,
		_In_ LARGE_INTEGER	RelativeOffset,
		_In_ ULONG	Origin,
		_Out_opt_ PULARGE_INTEGER	ResultOffset
	)

Definition at line 239 of file memstream.c.

{
    PVOID NewPosition;
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
 
    switch (Origin)
    {
        case STREAM_SEEK_SET:
            NewPosition = (PUCHAR)Stream->Start + RelativeOffset.QuadPart;
            break;
 
        case STREAM_SEEK_CUR:
            NewPosition = (PUCHAR)Stream->Current + RelativeOffset.QuadPart;
            break;
 
        case STREAM_SEEK_END:
            NewPosition = (PUCHAR)Stream->End - RelativeOffset.QuadPart;
            break;
 
        default:
            return E_INVALIDARG;
    }
 
    if (NewPosition < Stream->Start || NewPosition > Stream->End)
        return STG_E_INVALIDPOINTER;
 
    Stream->Current = NewPosition;
 
    if (ResultOffset)
        ResultOffset->QuadPart = (PUCHAR)Stream->Current - (PUCHAR)Stream->Start;
 
    return S_OK;
}

RtlSetMemoryStreamSize()

HRESULT NTAPI RtlSetMemoryStreamSize	(	_In_ IStream *	This,
		_In_ ULARGE_INTEGER	NewSize
	)

Definition at line 383 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(NewSize);
 
    return E_NOTIMPL;
}

RtlStatMemoryStream()

HRESULT NTAPI RtlStatMemoryStream	(	_In_ IStream *	This,
		_Out_ STATSTG *	Stats,
		_In_ ULONG	Flags
	)

Definition at line 337 of file memstream.c.

{
    PRTL_MEMORY_STREAM Stream = IStream_To_RTL_MEMORY_STREAM(This);
 
    if (!Stats)
        return STG_E_INVALIDPOINTER;
 
    RtlZeroMemory(Stats, sizeof(STATSTG));
    Stats->type = STGTY_STREAM;
    Stats->cbSize.QuadPart = (PUCHAR)Stream->End - (PUCHAR)Stream->Start;
 
    return S_OK;
}

RtlUnlockMemoryStreamRegion()

HRESULT NTAPI RtlUnlockMemoryStreamRegion	(	_In_ IStream *	This,
		_In_ ULARGE_INTEGER	Offset,
		_In_ ULARGE_INTEGER	Length,
		_In_ ULONG	LockType
	)

Definition at line 445 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(Offset);
    UNREFERENCED_PARAMETER(Length);
    UNREFERENCED_PARAMETER(LockType);
 
    return E_NOTIMPL;
}

RtlWriteMemoryStream()

HRESULT NTAPI RtlWriteMemoryStream	(	_In_ IStream *	This,
		_In_reads_bytes_(Length) CONST VOID *	Buffer,
		_In_ ULONG	Length,
		_Out_opt_ PULONG	BytesWritten
	)

Definition at line 364 of file memstream.c.

{
    UNREFERENCED_PARAMETER(This);
    UNREFERENCED_PARAMETER(Buffer);
    UNREFERENCED_PARAMETER(Length);
    UNREFERENCED_PARAMETER(BytesWritten);
 
    return E_NOTIMPL;
}

Variable Documentation

RtlMemoryStreamVtbl

const struct IStreamVtbl RtlMemoryStreamVtbl

Initial value:

=
{
    RtlQueryInterfaceMemoryStream,
    RtlAddRefMemoryStream,
    RtlReleaseMemoryStream,
    RtlReadMemoryStream,
    RtlWriteMemoryStream,
    RtlSeekMemoryStream,
    RtlSetMemoryStreamSize,
    RtlCopyMemoryStreamTo,
    RtlCommitMemoryStream,
    RtlRevertMemoryStream,
    RtlLockMemoryStreamRegion,
    RtlUnlockMemoryStreamRegion,
    RtlStatMemoryStream,
    RtlCloneMemoryStream,
}

Definition at line 18 of file memstream.c.

Referenced by RtlInitMemoryStream().

RtlOutOfProcessMemoryStreamVtbl

const struct IStreamVtbl RtlOutOfProcessMemoryStreamVtbl

Initial value:

=
{
    RtlQueryInterfaceMemoryStream,
    RtlAddRefMemoryStream,
    RtlReleaseMemoryStream,
    RtlReadOutOfProcessMemoryStream,
    RtlWriteMemoryStream,
    RtlSeekMemoryStream,
    RtlSetMemoryStreamSize,
    RtlCopyMemoryStreamTo,
    RtlCommitMemoryStream,
    RtlRevertMemoryStream,
    RtlLockMemoryStreamRegion,
    RtlUnlockMemoryStreamRegion,
    RtlStatMemoryStream,
    RtlCloneMemoryStream,
}

Definition at line 36 of file memstream.c.

Referenced by RtlInitOutOfProcessMemoryStream().