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
- ContainerControl.cs
- MeasurementDCInfo.cs
- DesignBindingValueUIHandler.cs
- CultureSpecificCharacterBufferRange.cs
- LinkUtilities.cs
- ConfigurationSectionGroup.cs
- TypeValidationEventArgs.cs
- UiaCoreApi.cs
- TraceContext.cs
- WeakHashtable.cs
- MsmqBindingBase.cs
- SafeArrayRankMismatchException.cs
- XmlSchemaCollection.cs
- MenuRendererClassic.cs
- XmlDataDocument.cs
- Vector3DKeyFrameCollection.cs
- BaseParaClient.cs
- InplaceBitmapMetadataWriter.cs
- BulletedList.cs
- LineServicesCallbacks.cs
- TextEditorParagraphs.cs
- ZoneMembershipCondition.cs
- DataChangedEventManager.cs
- DeferredSelectedIndexReference.cs
- Cell.cs
- EdmComplexTypeAttribute.cs
- SchemaInfo.cs
- EndPoint.cs
- XsltFunctions.cs
- PageThemeCodeDomTreeGenerator.cs
- PagePropertiesChangingEventArgs.cs
- OleDbWrapper.cs
- FontWeights.cs
- OleDbTransaction.cs
- RootBrowserWindow.cs
- CurrencyManager.cs
- HttpPostedFile.cs
- FileNotFoundException.cs
- EditorZoneBase.cs
- StringDictionary.cs
- CertificateManager.cs
- PointAnimationClockResource.cs
- Certificate.cs
- InheritanceUI.cs
- ChtmlTextWriter.cs
- BinaryConverter.cs
- EncodingTable.cs
- DesignerDataConnection.cs
- LogReservationCollection.cs
- ProtocolsSection.cs
- CompilerCollection.cs
- EdmProperty.cs
- ToolTipAutomationPeer.cs
- ScaleTransform3D.cs
- ResolveMatchesMessageCD1.cs
- CqlIdentifiers.cs
- ApplicationProxyInternal.cs
- DetailsViewModeEventArgs.cs
- HScrollProperties.cs
- CursorConverter.cs
- JoinGraph.cs
- updateconfighost.cs
- codemethodreferenceexpression.cs
- FontFaceLayoutInfo.cs
- UdpAnnouncementEndpoint.cs
- ControlCachePolicy.cs
- XamlLoadErrorInfo.cs
- WindowsComboBox.cs
- WebEventCodes.cs
- FontStyleConverter.cs
- ChannelCacheDefaults.cs
- Typography.cs
- peersecurityelement.cs
- XmlAnyElementAttribute.cs
- BindingManagerDataErrorEventArgs.cs
- StackSpiller.cs
- HotCommands.cs
- ComponentSerializationService.cs
- DataServiceStreamProviderWrapper.cs
- CharacterMetrics.cs
- Rotation3D.cs
- AsymmetricKeyExchangeDeformatter.cs
- NativeMethodsCLR.cs
- InputLanguageSource.cs
- DropSource.cs
- SiteMap.cs
- ApplicationManager.cs
- StaticFileHandler.cs
- EndOfStreamException.cs
- OdbcDataReader.cs
- PeerNameResolver.cs
- ListenerElementsCollection.cs
- WebRequest.cs
- TemplatedAdorner.cs
- CompiledIdentityConstraint.cs
- AppearanceEditorPart.cs
- GuidTagList.cs
- VerbConverter.cs
- SaveWorkflowCommand.cs
- AutomationPropertyInfo.cs