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
- ToolboxComponentsCreatingEventArgs.cs
- DataGrid.cs
- Pair.cs
- BasicExpandProvider.cs
- PLINQETWProvider.cs
- TextSearch.cs
- XamlPoint3DCollectionSerializer.cs
- ContentType.cs
- TopClause.cs
- EditingCommands.cs
- XmlSignificantWhitespace.cs
- RecordsAffectedEventArgs.cs
- SizeF.cs
- ObjectCloneHelper.cs
- VirtualizingPanel.cs
- InvalidComObjectException.cs
- AuthenticationSection.cs
- WebEventTraceProvider.cs
- TraceContext.cs
- Rotation3D.cs
- NetNamedPipeBindingElement.cs
- SystemFonts.cs
- AdCreatedEventArgs.cs
- EntityDataSourceDesigner.cs
- UdpDiscoveryEndpointElement.cs
- WeakReference.cs
- ControlEvent.cs
- BackgroundWorker.cs
- TraceListener.cs
- RoleManagerModule.cs
- SchemaMapping.cs
- DirectoryNotFoundException.cs
- DataGridTextBoxColumn.cs
- ParentControlDesigner.cs
- MessageParameterAttribute.cs
- ToolStripContainer.cs
- HighlightVisual.cs
- SystemPens.cs
- Message.cs
- XXXOnTypeBuilderInstantiation.cs
- CommandEventArgs.cs
- EntitySetBase.cs
- SystemIcons.cs
- XamlStream.cs
- CompressStream.cs
- ToolStripItemImageRenderEventArgs.cs
- StaticFileHandler.cs
- CustomError.cs
- TextServicesCompartment.cs
- UpdatePanelControlTrigger.cs
- DnsPermission.cs
- ToolStripContainerActionList.cs
- UserPreferenceChangedEventArgs.cs
- ComponentDispatcher.cs
- AssertValidation.cs
- WebConfigurationFileMap.cs
- ProfileBuildProvider.cs
- ExpressionBinding.cs
- DeflateStream.cs
- EmptyControlCollection.cs
- nulltextcontainer.cs
- XmlSchemaComplexContentRestriction.cs
- WebPartConnection.cs
- BaseDataListComponentEditor.cs
- MediaElementAutomationPeer.cs
- ChtmlLinkAdapter.cs
- x509store.cs
- XmlSchemaComplexContentRestriction.cs
- ModelPropertyDescriptor.cs
- CompatibleComparer.cs
- SystemEvents.cs
- InstanceCollisionException.cs
- ConnectionPoint.cs
- IPipelineRuntime.cs
- HandlerFactoryWrapper.cs
- RegisteredScript.cs
- SecurityElement.cs
- WriteableBitmap.cs
- QuaternionAnimationBase.cs
- SqlDataSourceEnumerator.cs
- NameService.cs
- sitestring.cs
- TextEmbeddedObject.cs
- HtmlLink.cs
- SortQuery.cs
- WebPartVerbCollection.cs
- SByteStorage.cs
- DatatypeImplementation.cs
- ConfigXmlCDataSection.cs
- TypeUsageBuilder.cs
- ParseChildrenAsPropertiesAttribute.cs
- OleAutBinder.cs
- EntityTransaction.cs
- RegionData.cs
- DrawingGroupDrawingContext.cs
- XslException.cs
- Identity.cs
- BufferAllocator.cs
- wpf-etw.cs
- CharUnicodeInfo.cs