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