src/WINNT/afsrdr/kernel/lib/AFSSecurity.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: AFSSecurity.cpp
  37 //
  38
  39 #include "AFSCommon.h"
  40
  41 NTSTATUS
  42 AFSSetSecurity( IN PDEVICE_OBJECT LibDeviceObject,
  43                 IN PIRP Irp)
  44 {
  45
  46     NTSTATUS ntStatus = STATUS_SUCCESS;
  47     IO_STACK_LOCATION *pIrpSp;
  48
  49     pIrpSp = IoGetCurrentIrpStackLocation( Irp);
  50
  51     __try
  52     {
  53
  54         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
  55                       AFS_TRACE_LEVEL_ERROR,
  56                       "AFSSetSecurity Entry for FO %08lX\n",
  57                       pIrpSp->FileObject);
  58
  59         AFSCompleteRequest( Irp,
  60                             ntStatus);
  61     }
  62     __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
  63     {
  64
  65         AFSDbgLogMsg( 0,
  66                       0,
  67                       "EXCEPTION - AFSSetSecurity\n");
  68
  69         AFSDumpTraceFilesFnc();
  70     }
  71
  72     return ntStatus;
  73 }
  74
  75 NTSTATUS
  76 AFSQuerySecurity( IN PDEVICE_OBJECT LibDeviceObject,
  77                   IN PIRP Irp)
  78 {
  79
  80     NTSTATUS ntStatus = STATUS_SUCCESS;
  81     PIO_STACK_LOCATION pIrpSp;
  82     PMDL pUserBufferMdl = NULL;
  83     void *pLockedUserBuffer = NULL;
  84     ULONG ulSDLength = 0;
  85
  86     __try
  87     {
  88
  89         pIrpSp = IoGetCurrentIrpStackLocation( Irp);
  90
  91         AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
  92                       AFS_TRACE_LEVEL_VERBOSE,
  93                       "AFSQuerySecurity Entry for FO %08lX\n",
  94                       pIrpSp->FileObject);
  95
  96         if( AFSDefaultSD == NULL)
  97         {
  98
  99             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
 100                           AFS_TRACE_LEVEL_ERROR,
 101                           "AFSQuerySecurity No default SD allocated\n");
 102
 103             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 104         }
 105
 106         ulSDLength = RtlLengthSecurityDescriptor( AFSDefaultSD);
 107
 108         if( pIrpSp->Parameters.QuerySecurity.Length < ulSDLength ||
 109             Irp->UserBuffer == NULL)
 110         {
 111
 112             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
 113                           AFS_TRACE_LEVEL_VERBOSE,
 114                           "AFSQuerySecurity Buffer too small\n");
 115
 116             Irp->IoStatus.Information = (ULONG_PTR)ulSDLength;
 117
 118             try_return( ntStatus = STATUS_BUFFER_OVERFLOW);
 119         }
 120
 121         pLockedUserBuffer = AFSLockUserBuffer( Irp->UserBuffer,
 122                                                pIrpSp->Parameters.QuerySecurity.Length,
 123                                                &pUserBufferMdl);
 124
 125         if( pLockedUserBuffer == NULL)
 126         {
 127
 128             AFSDbgLogMsg( AFS_SUBSYSTEM_FILE_PROCESSING,
 129                           AFS_TRACE_LEVEL_ERROR,
 130                           "AFSQuerySecurity Failed to lock user buffer\n");
 131
 132             try_return( ntStatus = STATUS_INSUFFICIENT_RESOURCES);
 133         }
 134
 135         RtlCopyMemory( pLockedUserBuffer,
 136                        AFSDefaultSD,
 137                        ulSDLength);
 138
 139         Irp->IoStatus.Information = (ULONG_PTR)ulSDLength;
 140
 141 try_exit:
 142
 143         if( pUserBufferMdl != NULL)
 144         {
 145             MmUnlockPages( pUserBufferMdl);
 146             IoFreeMdl( pUserBufferMdl);
 147         }
 148     }
 149     __except( AFSExceptionFilter( __FUNCTION__, GetExceptionCode(), GetExceptionInformation()) )
 150     {
 151
 152         AFSDbgLogMsg( 0,
 153                       0,
 154                       "EXCEPTION - AFSQuerySecurity\n");
 155
 156         AFSDumpTraceFilesFnc();
 157     }
 158
 159     AFSCompleteRequest( Irp,
 160                         ntStatus);
 161
 162     return ntStatus;
 163 }