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( GetExceptionCode(), GetExceptionInformation()) )
 128     {
 129
 130         AFSDbgLogMsg( 0,
 131                       0,
 132                       "EXCEPTION - AFSQueryFileInfo\n");
 133
 134         ntStatus = STATUS_UNSUCCESSFUL;
 135     }
 136
 137     return ntStatus;
 138 }
 139
 140 //
 141 // Function: AFSSetFileInfo
 142 //
 143 // Description:
 144 //
 145 //      This function is the dispatch handler for the IRP_MJ_SET_FILE_INFORMATION request
 146 //
 147 // Return:
 148 //
 149 //      A status is returned for the function
 150 //
 151
 152 NTSTATUS
 153 AFSSetFileInfo( IN PDEVICE_OBJECT DeviceObject,
 154                 IN PIRP Irp)
 155 {
 156
 157     NTSTATUS ntStatus = STATUS_SUCCESS;
 158     AFSDeviceExt *pDeviceExt = (AFSDeviceExt *)DeviceObject->DeviceExtension;
 159     AFSDeviceExt *pControlDeviceExt = (AFSDeviceExt *)AFSDeviceObject->DeviceExtension;
 160     IO_STACK_LOCATION *pIrpSp = IoGetCurrentIrpStackLocation( Irp);
 161     AFSFcb* pFcb = NULL;
 162
 163     __try
 164     {
 165
 166         if( DeviceObject == AFSDeviceObject)
 167         {
 168
 169             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
 170
 171             AFSCompleteRequest( Irp,
 172                                 ntStatus);
 173
 174             try_return( ntStatus);
 175         }
 176
 177         pFcb = (AFSFcb*) pIrpSp->FileObject->FsContext;
 178
 179         if( pFcb == NULL ||
 180             pFcb->Header.NodeTypeCode == AFS_REDIRECTOR_FCB)
 181         {
 182
 183             //
 184             // Root open
 185             //
 186
 187             ntStatus = STATUS_INVALID_DEVICE_REQUEST;
 188
 189             AFSCompleteRequest( Irp,
 190                                 ntStatus);
 191
 192             try_return( ntStatus);
 193         }
 194
 195         //
 196         // Check the state of the library
 197         //
 198
 199         ntStatus = AFSCheckLibraryState( Irp);
 200
 201         if( !NT_SUCCESS( ntStatus) ||
 202             ntStatus == STATUS_PENDING)
 203         {
 204
 205             if( ntStatus != STATUS_PENDING)
 206             {
 207                 AFSCompleteRequest( Irp, ntStatus);
 208             }
 209
 210             try_return( ntStatus);
 211         }
 212
 213         IoSkipCurrentIrpStackLocation( Irp);
 214
 215         ntStatus = IoCallDriver( pControlDeviceExt->Specific.Control.LibraryDeviceObject,
 216                                  Irp);
 217
 218         //
 219         // Indicate the library is done with the request
 220         //
 221
 222         AFSClearLibraryRequest();
 223
 224 try_exit:
 225
 226         NOTHING;
 227     }
 228     __except( AFSExceptionFilter( GetExceptionCode(), GetExceptionInformation()) )
 229     {
 230
 231         AFSDbgLogMsg( 0,
 232                       0,
 233                       "EXCEPTION - AFSSetFileInfo\n");
 234
 235         ntStatus = STATUS_UNSUCCESSFUL;
 236     }
 237
 238     return ntStatus;
 239 }