Windows: Wait for memory allocation if necessary
[openafs.git] / src / WINNT / afsrdr / kernel / fs / AFSVolumeInfo.cpp
1 /*
2  * Copyright (c) 2008, 2009, 2010, 2011 Kernel Drivers, LLC.
3  * Copyright (c) 2009, 2010, 2011 Your File System, Inc.
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  *
10  * - Redistributions of source code must retain the above copyright notice,
11  *   this list of conditions and the following disclaimer.
12  * - Redistributions in binary form must reproduce the above copyright
13  *   notice,
14  *   this list of conditions and the following disclaimer in the
15  *   documentation
16  *   and/or other materials provided with the distribution.
17  * - Neither the names of Kernel Drivers, LLC and Your File System, Inc.
18  *   nor the names of their contributors may be used to endorse or promote
19  *   products derived from this software without specific prior written
20  *   permission from Kernel Drivers, LLC and Your File System, Inc.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
23  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
24  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
25  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
26  * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
27  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
28  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
29  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
30  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
31  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
32  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
33  */
34
35 //
36 // File: AFSVolumeInfo.cpp
37 //
38
39 #include "AFSCommon.h"
40
41 NTSTATUS
42 AFSQueryVolumeInfo( IN PDEVICE_OBJECT DeviceObject,
43                     IN PIRP Irp)
44 {
45
46     NTSTATUS ntStatus = STATUS_SUCCESS;
47     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
48     IO_STACK_LOCATION *pIrpSp;
49     AFSDeviceExt *pControlDeviceExt = (AFSDeviceExt *)AFSDeviceObject->DeviceExtension;
50
51     pIrpSp = IoGetCurrentIrpStackLocation( Irp);
52
53     __try
54     {
55
56         if( DeviceObject == AFSDeviceObject)
57         {
58
59             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
60
61             AFSCompleteRequest( Irp,
62                                 ntStatus);
63
64             try_return( ntStatus);
65         }
66
67         //
68         // Check the state of the library
69         //
70
71         ntStatus = AFSCheckLibraryState( Irp);
72
73         if( !NT_SUCCESS( ntStatus) ||
74             ntStatus == STATUS_PENDING)
75         {
76
77             if( ntStatus != STATUS_PENDING)
78             {
79                 AFSCompleteRequest( Irp, ntStatus);
80             }
81
82             try_return( ntStatus);
83         }
84
85         IoSkipCurrentIrpStackLocation( Irp);
86
87         ntStatus = IoCallDriver( pControlDeviceExt->Specific.Control.LibraryDeviceObject,
88                                  Irp);
89
90         //
91         // Indicate the library is done with the request
92         //
93
94         AFSClearLibraryRequest();
95
96 try_exit:
97
98         NOTHING;
99     }
100     __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
101     {
102
103         AFSDbgLogMsg( 0,
104                       0,
105                       "EXCEPTION - AFSQueryVolumeInfo\n");
106     }
107
108     return ntStatus;
109 }
110
111 NTSTATUS
112 AFSSetVolumeInfo( IN PDEVICE_OBJECT DeviceObject,
113                   IN PIRP Irp)
114 {
115
116     NTSTATUS ntStatus = STATUS_INVALID_DEVICE_REQUEST;
117     IO_STACK_LOCATION *pIrpSp;
118
119     pIrpSp = IoGetCurrentIrpStackLocation( Irp);
120
121     __try
122     {
123
124         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
125                       AFS_TRACE_LEVEL_WARNING,
126                       "AFSSetVolumeInfo Entry for FO %08lX\n", pIrpSp->FileObject);
127
128         AFSCompleteRequest( Irp,
129                             ntStatus);
130
131     }
132     __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
133     {
134
135         AFSDbgLogMsg( 0,
136                       0,
137                       "EXCEPTION - AFSSetVolumeInfo\n");
138     }
139
140     return ntStatus;
141 }