1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
//------------------------------------------------------------------------------
// <copyright file="UnmanagedWebSocketContext.cs" company="Microsoft">
// Copyright (c) Microsoft Corporation. All rights reserved.
// </copyright>
//------------------------------------------------------------------------------
namespace System.Web.WebSockets {
using System;
using System.Runtime.InteropServices;
using System.Security;
using System.Security.Permissions;
// Concrete implementation of IUnmanagedWebSocketContext, implemented by IIS8 unmanaged WebSocket module
// SECURITY NOTE: All parameters are assumed to have been validated by the caller
[SecurityPermission(SecurityAction.LinkDemand, UnmanagedCode = true)]
internal sealed class UnmanagedWebSocketContext : IUnmanagedWebSocketContext {
// the 'this' parameter for the unmanaged interface
private readonly IntPtr _pWebSocketContext;
internal UnmanagedWebSocketContext(IntPtr pWebSocketContext) {
_pWebSocketContext = pWebSocketContext;
}
public int WriteFragment(IntPtr pData, ref int pcbSent, bool fAsync, bool fUtf8Encoded, bool fFinalFragment, IntPtr pfnCompletion, IntPtr pvCompletionContext, out bool pfCompletionExpected) {
return IIS.MgdWebSocketWriteFragment(_pWebSocketContext, pData, ref pcbSent, fAsync, fUtf8Encoded, fFinalFragment, pfnCompletion, pvCompletionContext, out pfCompletionExpected);
}
public int ReadFragment(IntPtr pData, ref int pcbData, bool fAsync, out bool pfUtf8Encoded, out bool pfFinalFragment, out bool pfConnectionClose, IntPtr pfnCompletion, IntPtr pvCompletionContext, out bool pfCompletionExpected) {
return IIS.MgdWebSocketReadFragment(_pWebSocketContext, pData, ref pcbData, fAsync, out pfUtf8Encoded, out pfFinalFragment, out pfConnectionClose, pfnCompletion, pvCompletionContext, out pfCompletionExpected);
}
public int SendConnectionClose(bool fAsync, ushort uStatusCode, string szReason, IntPtr pfnCompletion, IntPtr pvCompletionContext, out bool pfCompletionExpected) {
return IIS.MgdWebSocketSendConnectionClose(_pWebSocketContext, fAsync, uStatusCode, szReason, pfnCompletion, pvCompletionContext, out pfCompletionExpected);
}
public int GetCloseStatus(out ushort pStatusCode, out IntPtr ppszReason, out ushort pcchReason) {
return IIS.MgdWebSocketGetCloseStatus(_pWebSocketContext, out pStatusCode, out ppszReason, out pcchReason);
}
public void CloseTcpConnection() {
IIS.MgdWebSocketCloseTcpConnection(_pWebSocketContext);
}
// API documentation for each method can be found in ndp/fx/src/xsp/webengine/mgdexports.cxx.
[SuppressUnmanagedCodeSecurity]
private static class IIS {
private const string _IIS_NATIVE_DLL = ModName.MGDENG_FULL_NAME;
// Write a data fragment to the provided IWebSocketContext.
[DllImport(_IIS_NATIVE_DLL)]
internal static extern int MgdWebSocketWriteFragment(
IntPtr pContext,
IntPtr pData,
ref int pcbSent,
bool fAsync,
bool fUTF8Encoded,
bool fFinalFragment,
IntPtr pfnCompletion,
IntPtr pvCompletionContext,
out bool pfCompletionExpected);
// Reads a data fragment from the provided IWebSocketContext.
[DllImport(_IIS_NATIVE_DLL)]
internal static extern int MgdWebSocketReadFragment(
IntPtr pContext,
IntPtr pData,
ref int pcbData,
bool fAsync,
out bool pfUTF8Encoded,
out bool pfFinalFragment,
out bool pfConnectionClose,
IntPtr pfnCompletion,
IntPtr pvCompletionContext,
out bool pfCompletionExpected);
// Sends a CLOSE frame to the provided IWebSocketContext.
[DllImport(_IIS_NATIVE_DLL)]
internal static extern int MgdWebSocketSendConnectionClose(
IntPtr pContext,
bool fAsync,
ushort pStatusCode,
[MarshalAs(UnmanagedType.LPWStr)] string pszReason,
IntPtr pfnCompletion,
IntPtr pvCompletionContext,
out bool pfCompletionExpected);
// Gets information on the CLOSE frame sent from the client to the server.
[DllImport(_IIS_NATIVE_DLL)]
internal static extern int MgdWebSocketGetCloseStatus(
IntPtr pContext,
out ushort pStatusCode,
out IntPtr ppszReason,
out ushort pcchReason);
// Closes the TCP connection used by the provided IWebSocketContext.
[DllImport(_IIS_NATIVE_DLL)]
internal static extern void MgdWebSocketCloseTcpConnection(
IntPtr pContext);
}
}
}
|
