src/WINNT/afsrdr/kernel/fs/AFSFileInfo.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: AFSFileInfo.cpp
  37 //
  38
  39 #include "AFSCommon.h"
  40
  41 //
  42 // Function: AFSQueryFileInfo
  43 //
  44 // Description:
  45 //
  46 //      This function is the dispatch handler for the IRP_MJ_QUERY_FILE_INFORMATION request
  47 //
  48 // Return:
  49 //
  50 //      A status is returned for the function
  51 //
  52
  53 NTSTATUS
  54 AFSQueryFileInfo( IN PDEVICE_OBJECT DeviceObject,
  55                   IN PIRP Irp)
  56 {
  57
  58     NTSTATUS ntStatus = STATUS_SUCCESS;
  59     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
  60     AFSDeviceExt *pControlDeviceExt = (AFSDeviceExt *)AFSDeviceObject->DeviceExtension;
  61     AFSFcb* pFcb = NULL;
  62     __try
  63     {
  64
  65         if( DeviceObject == AFSDeviceObject)
  66         {
  67
  68             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
  69
  70             AFSCompleteRequest( Irp,
  71                                 ntStatus);
  72
  73             try_return( ntStatus);
  74         }
  75
  76         pFcb = (AFSFcb*) pIrpSp->FileObject->FsContext;
  77
  78         if( pFcb == NULL ||
  79             pFcb->Header.NodeTypeCode == AFS_REDIRECTOR_FCB)
  80         {
  81
  82             //
  83             // Root open
  84             //
  85
  86             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
  87
  88             AFSCompleteRequest( Irp,
  89                                 ntStatus);
  90
  91             try_return( ntStatus);
  92         }
  93
  94         //
  95         // Check the state of the library
  96         //
  97
  98         ntStatus = AFSCheckLibraryState( Irp);
  99
 100         if( !NT_SUCCESS( ntStatus) ||
 101             ntStatus == STATUS_PENDING)
 102         {
 103
 104             if( ntStatus != STATUS_PENDING)
 105             {
 106                 AFSCompleteRequest( Irp, ntStatus);
 107             }
 108
 109             try_return( ntStatus);
 110         }
 111
 112         IoSkipCurrentIrpStackLocation( Irp);
 113
 114         ntStatus = IoCallDriver( pControlDeviceExt->Specific.Control.LibraryDeviceObject,
 115                                  Irp);
 116
 117         //
 118         // Indicate the library is done with the request
 119         //
 120
 121         AFSClearLibraryRequest();
 122
 123 try_exit:
 124
 125         NOTHING;
 126     }
 127     __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
 128     {
 129
 130         AFSDbgTrace(( 0,
 131                       0,
 132                       "EXCEPTION - AFSQueryFileInfo\n"));
 133
 134         ntStatus = STATUS_UNSUCCESSFUL;
 135
 136         AFSDumpTraceFilesFnc();
 137     }
 138
 139     return ntStatus;
 140 }
 141
 142 //
 143 // Function: AFSSetFileInfo
 144 //
 145 // Description:
 146 //
 147 //      This function is the dispatch handler for the IRP_MJ_SET_FILE_INFORMATION request
 148 //
 149 // Return:
 150 //
 151 //      A status is returned for the function
 152 //
 153
 154 NTSTATUS
 155 AFSSetFileInfo( IN PDEVICE_OBJECT DeviceObject,
 156                 IN PIRP Irp)
 157 {
 158
 159     NTSTATUS ntStatus = STATUS_SUCCESS;
 160     AFSDeviceExt *pControlDeviceExt = (AFSDeviceExt *)AFSDeviceObject->DeviceExtension;
 161     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
 162     AFSFcb* pFcb = NULL;
 163
 164     __try
 165     {
 166
 167         if( DeviceObject == AFSDeviceObject)
 168         {
 169
 170             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
 171
 172             AFSCompleteRequest( Irp,
 173                                 ntStatus);
 174
 175             try_return( ntStatus);
 176         }
 177
 178         pFcb = (AFSFcb*) pIrpSp->FileObject->FsContext;
 179
 180         if( pFcb == NULL ||
 181             pFcb->Header.NodeTypeCode == AFS_REDIRECTOR_FCB)
 182         {
 183
 184             //
 185             // Root open
 186             //
 187
 188             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
 189
 190             AFSCompleteRequest( Irp,
 191                                 ntStatus);
 192
 193             try_return( ntStatus);
 194         }
 195
 196         //
 197         // Check the state of the library
 198         //
 199
 200         ntStatus = AFSCheckLibraryState( Irp);
 201
 202         if( !NT_SUCCESS( ntStatus) ||
 203             ntStatus == STATUS_PENDING)
 204         {
 205
 206             if( ntStatus != STATUS_PENDING)
 207             {
 208                 AFSCompleteRequest( Irp, ntStatus);
 209             }
 210
 211             try_return( ntStatus);
 212         }
 213
 214         IoSkipCurrentIrpStackLocation( Irp);
 215
 216         ntStatus = IoCallDriver( pControlDeviceExt->Specific.Control.LibraryDeviceObject,
 217                                  Irp);
 218
 219         //
 220         // Indicate the library is done with the request
 221         //
 222
 223         AFSClearLibraryRequest();
 224
 225 try_exit:
 226
 227         NOTHING;
 228     }
 229     __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
 230     {
 231
 232         AFSDbgTrace(( 0,
 233                       0,
 234                       "EXCEPTION - AFSSetFileInfo\n"));
 235
 236         ntStatus = STATUS_UNSUCCESSFUL;
 237
 238         AFSDumpTraceFilesFnc();
 239     }
 240
 241     return ntStatus;
 242 }