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
- ClientBuildManagerCallback.cs
- ColumnPropertiesGroup.cs
- TemplateBamlTreeBuilder.cs
- ComboBoxHelper.cs
- XmlSchemaAttributeGroupRef.cs
- CodeSnippetCompileUnit.cs
- ServiceMetadataBehavior.cs
- EmptyControlCollection.cs
- ProgressBarHighlightConverter.cs
- GridLength.cs
- DataSourceCache.cs
- BoundColumn.cs
- CheckBoxRenderer.cs
- TdsParserStaticMethods.cs
- EntryWrittenEventArgs.cs
- GlobalItem.cs
- CodeCatchClauseCollection.cs
- BaseTemplateBuildProvider.cs
- BypassElementCollection.cs
- RootBrowserWindowAutomationPeer.cs
- PersonalizationProvider.cs
- MapPathBasedVirtualPathProvider.cs
- ActivityExecutor.cs
- HandlerMappingMemo.cs
- UserControlParser.cs
- WebServiceClientProxyGenerator.cs
- FormatterServices.cs
- Win32PrintDialog.cs
- OperatingSystem.cs
- WmpBitmapDecoder.cs
- Emitter.cs
- ImpersonateTokenRef.cs
- HWStack.cs
- HttpWriter.cs
- SettingsBase.cs
- MenuItemStyleCollection.cs
- PropertyInformation.cs
- CustomError.cs
- RotateTransform.cs
- ScriptingSectionGroup.cs
- SqlClientWrapperSmiStream.cs
- UserNameSecurityToken.cs
- DocumentOrderComparer.cs
- StyleXamlParser.cs
- UnsafeNativeMethods.cs
- ObservableDictionary.cs
- TabControl.cs
- InvokeHandlers.cs
- SmtpCommands.cs
- OleDbRowUpdatingEvent.cs
- HwndProxyElementProvider.cs
- ReadContentAsBinaryHelper.cs
- Visual3D.cs
- DataGridItem.cs
- SafeBitVector32.cs
- SiteMapDataSource.cs
- ToolStripPanelCell.cs
- InternalControlCollection.cs
- X509SubjectKeyIdentifierClause.cs
- SizeConverter.cs
- ColumnResizeUndoUnit.cs
- VisualTarget.cs
- ClosureBinding.cs
- SizeAnimation.cs
- TogglePattern.cs
- TypeToArgumentTypeConverter.cs
- CopyNodeSetAction.cs
- SchemaObjectWriter.cs
- AssociatedControlConverter.cs
- BrushValueSerializer.cs
- TreeWalkHelper.cs
- Ipv6Element.cs
- XamlReaderHelper.cs
- DesignerSerializationOptionsAttribute.cs
- IntSecurity.cs
- DataGridPagerStyle.cs
- InfoCardXmlSerializer.cs
- UTF7Encoding.cs
- _LocalDataStore.cs
- PageCodeDomTreeGenerator.cs
- DataGridViewHitTestInfo.cs
- XAMLParseException.cs
- TimeSpanSecondsOrInfiniteConverter.cs
- RandomNumberGenerator.cs
- DataColumnPropertyDescriptor.cs
- ImageDrawing.cs
- TextViewDesigner.cs
- DtdParser.cs
- ModelTreeEnumerator.cs
- DataGridViewUtilities.cs
- LocatorGroup.cs
- WrappedIUnknown.cs
- CollaborationHelperFunctions.cs
- ItemMap.cs
- CompareValidator.cs
- XslException.cs
- ContextMenu.cs
- LongMinMaxAggregationOperator.cs
- WebUtil.cs
- FlowPosition.cs