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
- figurelengthconverter.cs
- ColumnHeader.cs
- SkipQueryOptionExpression.cs
- DiffuseMaterial.cs
- Models.cs
- PrintDocument.cs
- CachedFontFace.cs
- GridViewUpdatedEventArgs.cs
- SocketInformation.cs
- QuarticEase.cs
- ColorConvertedBitmap.cs
- OrderedDictionaryStateHelper.cs
- CurrentChangingEventArgs.cs
- FormViewPagerRow.cs
- FrugalList.cs
- CompletedAsyncResult.cs
- AdornerDecorator.cs
- ClientOptions.cs
- TextPointer.cs
- columnmapkeybuilder.cs
- TemplatedControlDesigner.cs
- ToolboxDataAttribute.cs
- FormClosingEvent.cs
- Module.cs
- GenerateHelper.cs
- StylusPlugin.cs
- MobileTemplatedControlDesigner.cs
- GenericEnumerator.cs
- NumericExpr.cs
- InvariantComparer.cs
- ProviderUtil.cs
- StateDesigner.cs
- XmlCDATASection.cs
- DataGridViewColumnCollectionEditor.cs
- HashMembershipCondition.cs
- CompilerCollection.cs
- listitem.cs
- TreeNodeEventArgs.cs
- DrawingAttributes.cs
- StreamSecurityUpgradeAcceptorBase.cs
- GridViewUpdatedEventArgs.cs
- AlternateViewCollection.cs
- FunctionImportMapping.cs
- OracleInfoMessageEventArgs.cs
- PreviewPageInfo.cs
- ComAdminWrapper.cs
- LocalFileSettingsProvider.cs
- SmiEventSink_DeferedProcessing.cs
- basecomparevalidator.cs
- PathParser.cs
- WorkflowNamespace.cs
- Config.cs
- OperationBehaviorAttribute.cs
- NumberEdit.cs
- FaultImportOptions.cs
- SafeViewOfFileHandle.cs
- DataGridRelationshipRow.cs
- FileLogRecordEnumerator.cs
- SecurityUniqueId.cs
- ListViewEditEventArgs.cs
- ArcSegment.cs
- IImplicitResourceProvider.cs
- SettingsPropertyNotFoundException.cs
- TypeUtils.cs
- XmlTextReader.cs
- PaintEvent.cs
- GatewayDefinition.cs
- WmlObjectListAdapter.cs
- RuntimeArgument.cs
- Char.cs
- PasswordRecoveryDesigner.cs
- SecurityException.cs
- IDispatchConstantAttribute.cs
- JpegBitmapEncoder.cs
- PersistChildrenAttribute.cs
- EmbeddedMailObjectsCollection.cs
- PingOptions.cs
- ConfigurationSettings.cs
- SafeNativeMethodsMilCoreApi.cs
- CommandHelper.cs
- WebPartChrome.cs
- BulletedList.cs
- CollectionViewProxy.cs
- Margins.cs
- ADConnectionHelper.cs
- RuleValidation.cs
- WebScriptServiceHostFactory.cs
- ServicePoint.cs
- RSACryptoServiceProvider.cs
- HtmlEncodedRawTextWriter.cs
- DispatchOperationRuntime.cs
- ImageAttributes.cs
- AssemblyBuilder.cs
- JsonWriter.cs
- FamilyMap.cs
- Shape.cs
- DirectoryObjectSecurity.cs
- SemanticBasicElement.cs
- TextEvent.cs
- DoubleStorage.cs