Code:
/ 4.0 / 4.0 / DEVDIV_TFS / Dev10 / Releases / RTMRel / ndp / fx / src / Net / System / Net / _ListenerAsyncResult.cs / 1305376 / _ListenerAsyncResult.cs
//------------------------------------------------------------------------------
//
// Copyright (c) Microsoft Corporation. All rights reserved.
//
//-----------------------------------------------------------------------------
namespace System.Net {
using System.ComponentModel;
using System.Threading;
using System.Runtime.InteropServices;
unsafe class ListenerAsyncResult : LazyAsyncResult
{
private static readonly IOCompletionCallback s_IOCallback = new IOCompletionCallback(WaitCallback);
internal static IOCompletionCallback IOCallback
{
get
{
return s_IOCallback;
}
}
private AsyncRequestContext m_RequestContext;
internal ListenerAsyncResult(object asyncObject, object userState, AsyncCallback callback) :
base(asyncObject, userState, callback)
{
m_RequestContext = new AsyncRequestContext(this);
}
private static unsafe void WaitCallback(uint errorCode, uint numBytes, NativeOverlapped* nativeOverlapped)
{
// take the ListenerAsyncResult object from the state
Overlapped callbackOverlapped = Overlapped.Unpack(nativeOverlapped);
ListenerAsyncResult asyncResult = (ListenerAsyncResult) callbackOverlapped.AsyncResult;
object result = null;
try
{
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() errorCode:[" + errorCode.ToString() + "] numBytes:[" + numBytes.ToString() + "] nativeOverlapped:[" + ((long)nativeOverlapped).ToString() + "]");
if (errorCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS &&
errorCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_MORE_DATA)
{
asyncResult.ErrorCode = (int)errorCode;
result = new HttpListenerException((int)errorCode);
}
else
{
HttpListener httpWebListener = asyncResult.AsyncObject as HttpListener;
if (errorCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS)
{
// at this point we have received an unmanaged HTTP_REQUEST and memoryBlob
// points to it we need to hook up our authentication handling code here.
bool stoleBlob = false;
try
{
result = httpWebListener.HandleAuthentication(asyncResult.m_RequestContext, out stoleBlob);
}
finally
{
if (stoleBlob)
{
// The request has been handed to the user, which means this code can't reuse the blob. Reset it here.
asyncResult.m_RequestContext = result == null ? new AsyncRequestContext(asyncResult) : null;
}
else
{
asyncResult.m_RequestContext.Reset(0, 0);
}
}
}
else
{
asyncResult.m_RequestContext.Reset(asyncResult.m_RequestContext.RequestBlob->RequestId, numBytes);
}
// We need to issue a new request, either because auth failed, or because our buffer was too small the first time.
if (result==null)
{
uint statusCode = asyncResult.QueueBeginGetContext();
if (statusCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS &&
statusCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_IO_PENDING)
{
// someother bad error, possible(?) return values are:
// ERROR_INVALID_HANDLE, ERROR_INSUFFICIENT_BUFFER, ERROR_OPERATION_ABORTED
result = new HttpListenerException((int)statusCode);
}
}
if (result==null) {
return;
}
}
// complete the async IO and invoke the callback
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() calling Complete()");
}
catch (Exception exception)
{
if (NclUtilities.IsFatal(exception)) throw;
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() Caught exception:" + exception.ToString());
result = exception;
}
asyncResult.InvokeCallback(result);
}
internal uint QueueBeginGetContext()
{
uint statusCode = UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS;
while (true)
{
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(this) + "::QueueBeginGetContext() calling UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest RequestId:" + m_RequestContext.RequestBlob->RequestId + " Buffer:0x" + ((IntPtr) m_RequestContext.RequestBlob).ToString("x") + " Size:" + m_RequestContext.Size.ToString());
(AsyncObject as HttpListener).EnsureBoundHandle();
uint bytesTransferred = 0;
statusCode = UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest(
(AsyncObject as HttpListener).RequestQueueHandle,
m_RequestContext.RequestBlob->RequestId,
(uint) UnsafeNclNativeMethods.HttpApi.HTTP_FLAGS.HTTP_RECEIVE_REQUEST_FLAG_COPY_BODY,
m_RequestContext.RequestBlob,
m_RequestContext.Size,
&bytesTransferred,
m_RequestContext.NativeOverlapped);
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(this) + "::QueueBeginGetContext() call to UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest returned:" + statusCode);
if (statusCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_INVALID_PARAMETER && m_RequestContext.RequestBlob->RequestId != 0)
{
// we might get this if somebody stole our RequestId,
// set RequestId to 0 and start all over again with the buffer we just allocated
//
m_RequestContext.RequestBlob->RequestId = 0;
continue;
}
else if (statusCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_MORE_DATA)
{
// the buffer was not big enough to fit the headers, we need
// to read the RequestId returned, allocate a new buffer of the required size
m_RequestContext.Reset(m_RequestContext.RequestBlob->RequestId, bytesTransferred);
continue;
}
break;
}
return statusCode;
}
// Will be called from the base class upon InvokeCallback()
protected override void Cleanup()
{
if (m_RequestContext != null)
{
m_RequestContext.ReleasePins();
m_RequestContext.Close();
}
base.Cleanup();
}
}
}
// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
//------------------------------------------------------------------------------
//
// Copyright (c) Microsoft Corporation. All rights reserved.
//
//-----------------------------------------------------------------------------
namespace System.Net {
using System.ComponentModel;
using System.Threading;
using System.Runtime.InteropServices;
unsafe class ListenerAsyncResult : LazyAsyncResult
{
private static readonly IOCompletionCallback s_IOCallback = new IOCompletionCallback(WaitCallback);
internal static IOCompletionCallback IOCallback
{
get
{
return s_IOCallback;
}
}
private AsyncRequestContext m_RequestContext;
internal ListenerAsyncResult(object asyncObject, object userState, AsyncCallback callback) :
base(asyncObject, userState, callback)
{
m_RequestContext = new AsyncRequestContext(this);
}
private static unsafe void WaitCallback(uint errorCode, uint numBytes, NativeOverlapped* nativeOverlapped)
{
// take the ListenerAsyncResult object from the state
Overlapped callbackOverlapped = Overlapped.Unpack(nativeOverlapped);
ListenerAsyncResult asyncResult = (ListenerAsyncResult) callbackOverlapped.AsyncResult;
object result = null;
try
{
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() errorCode:[" + errorCode.ToString() + "] numBytes:[" + numBytes.ToString() + "] nativeOverlapped:[" + ((long)nativeOverlapped).ToString() + "]");
if (errorCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS &&
errorCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_MORE_DATA)
{
asyncResult.ErrorCode = (int)errorCode;
result = new HttpListenerException((int)errorCode);
}
else
{
HttpListener httpWebListener = asyncResult.AsyncObject as HttpListener;
if (errorCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS)
{
// at this point we have received an unmanaged HTTP_REQUEST and memoryBlob
// points to it we need to hook up our authentication handling code here.
bool stoleBlob = false;
try
{
result = httpWebListener.HandleAuthentication(asyncResult.m_RequestContext, out stoleBlob);
}
finally
{
if (stoleBlob)
{
// The request has been handed to the user, which means this code can't reuse the blob. Reset it here.
asyncResult.m_RequestContext = result == null ? new AsyncRequestContext(asyncResult) : null;
}
else
{
asyncResult.m_RequestContext.Reset(0, 0);
}
}
}
else
{
asyncResult.m_RequestContext.Reset(asyncResult.m_RequestContext.RequestBlob->RequestId, numBytes);
}
// We need to issue a new request, either because auth failed, or because our buffer was too small the first time.
if (result==null)
{
uint statusCode = asyncResult.QueueBeginGetContext();
if (statusCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS &&
statusCode != UnsafeNclNativeMethods.ErrorCodes.ERROR_IO_PENDING)
{
// someother bad error, possible(?) return values are:
// ERROR_INVALID_HANDLE, ERROR_INSUFFICIENT_BUFFER, ERROR_OPERATION_ABORTED
result = new HttpListenerException((int)statusCode);
}
}
if (result==null) {
return;
}
}
// complete the async IO and invoke the callback
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() calling Complete()");
}
catch (Exception exception)
{
if (NclUtilities.IsFatal(exception)) throw;
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(asyncResult) + "::WaitCallback() Caught exception:" + exception.ToString());
result = exception;
}
asyncResult.InvokeCallback(result);
}
internal uint QueueBeginGetContext()
{
uint statusCode = UnsafeNclNativeMethods.ErrorCodes.ERROR_SUCCESS;
while (true)
{
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(this) + "::QueueBeginGetContext() calling UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest RequestId:" + m_RequestContext.RequestBlob->RequestId + " Buffer:0x" + ((IntPtr) m_RequestContext.RequestBlob).ToString("x") + " Size:" + m_RequestContext.Size.ToString());
(AsyncObject as HttpListener).EnsureBoundHandle();
uint bytesTransferred = 0;
statusCode = UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest(
(AsyncObject as HttpListener).RequestQueueHandle,
m_RequestContext.RequestBlob->RequestId,
(uint) UnsafeNclNativeMethods.HttpApi.HTTP_FLAGS.HTTP_RECEIVE_REQUEST_FLAG_COPY_BODY,
m_RequestContext.RequestBlob,
m_RequestContext.Size,
&bytesTransferred,
m_RequestContext.NativeOverlapped);
GlobalLog.Print("ListenerAsyncResult#" + ValidationHelper.HashString(this) + "::QueueBeginGetContext() call to UnsafeNclNativeMethods.HttpApi.HttpReceiveHttpRequest returned:" + statusCode);
if (statusCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_INVALID_PARAMETER && m_RequestContext.RequestBlob->RequestId != 0)
{
// we might get this if somebody stole our RequestId,
// set RequestId to 0 and start all over again with the buffer we just allocated
//
m_RequestContext.RequestBlob->RequestId = 0;
continue;
}
else if (statusCode == UnsafeNclNativeMethods.ErrorCodes.ERROR_MORE_DATA)
{
// the buffer was not big enough to fit the headers, we need
// to read the RequestId returned, allocate a new buffer of the required size
m_RequestContext.Reset(m_RequestContext.RequestBlob->RequestId, bytesTransferred);
continue;
}
break;
}
return statusCode;
}
// Will be called from the base class upon InvokeCallback()
protected override void Cleanup()
{
if (m_RequestContext != null)
{
m_RequestContext.ReleasePins();
m_RequestContext.Close();
}
base.Cleanup();
}
}
}
// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
Link Menu
This book is available now!
Buy at Amazon US or
Buy at Amazon UK
- KnownTypes.cs
- SecurityTokenAttachmentMode.cs
- WmlImageAdapter.cs
- PolygonHotSpot.cs
- ObjectToIdCache.cs
- PageClientProxyGenerator.cs
- _NegoState.cs
- CapiSymmetricAlgorithm.cs
- Renderer.cs
- METAHEADER.cs
- Evidence.cs
- JsonServiceDocumentSerializer.cs
- SimpleMailWebEventProvider.cs
- XmlComment.cs
- DataGridItemCollection.cs
- ListMarkerSourceInfo.cs
- TimeSpanParse.cs
- ConvertEvent.cs
- ConfigurationManagerInternal.cs
- XmlAttributeCollection.cs
- CheckBoxPopupAdapter.cs
- ViewCellRelation.cs
- CookielessHelper.cs
- CloudCollection.cs
- TableProviderWrapper.cs
- RootProjectionNode.cs
- SafeArrayRankMismatchException.cs
- TargetConverter.cs
- SqlPersonalizationProvider.cs
- RepeaterItemCollection.cs
- HtmlProps.cs
- COAUTHIDENTITY.cs
- Buffer.cs
- WindowsEditBox.cs
- _TimerThread.cs
- FrameworkTextComposition.cs
- Utils.cs
- ContentPathSegment.cs
- ResolveNameEventArgs.cs
- OutputCacheSection.cs
- SqlDataSourceCache.cs
- DbException.cs
- GlyphShapingProperties.cs
- ServerValidateEventArgs.cs
- ConfigXmlText.cs
- _DisconnectOverlappedAsyncResult.cs
- EntityDataSourceViewSchema.cs
- ConfigurationValidatorBase.cs
- StatusBarAutomationPeer.cs
- ErrorReporting.cs
- SocketInformation.cs
- FontCacheLogic.cs
- BezierSegment.cs
- RuntimeConfigurationRecord.cs
- HeaderedContentControl.cs
- LoginCancelEventArgs.cs
- KeyInterop.cs
- EventDescriptorCollection.cs
- SingleAnimation.cs
- LinearGradientBrush.cs
- SafeUserTokenHandle.cs
- SliderAutomationPeer.cs
- ConfigPathUtility.cs
- HttpFileCollectionBase.cs
- WorkflowCompensationBehavior.cs
- DataExpression.cs
- DataServiceQuery.cs
- SqlColumnizer.cs
- BindingParameterCollection.cs
- DesignDataSource.cs
- ByteViewer.cs
- MenuItem.cs
- StackBuilderSink.cs
- IpcClientChannel.cs
- thaishape.cs
- MouseWheelEventArgs.cs
- SamlAssertion.cs
- SoapMessage.cs
- BindingEditor.xaml.cs
- WorkflowPersistenceService.cs
- DateTimeUtil.cs
- ExecutionEngineException.cs
- Trace.cs
- NetCodeGroup.cs
- XmlDataContract.cs
- ParameterRetriever.cs
- ClientRolePrincipal.cs
- ValidationError.cs
- CompositeDuplexBindingElementImporter.cs
- ClientSponsor.cs
- StringToken.cs
- QilDataSource.cs
- UserPersonalizationStateInfo.cs
- ElementHost.cs
- PropertyDescriptorCollection.cs
- ByteViewer.cs
- LayoutDump.cs
- ScriptResourceMapping.cs
- ProviderManager.cs
- UnsignedPublishLicense.cs