PUNICODE_STRING

SfGetFileName(

    IN PFILE_OBJECT FileObject,

    IN NTSTATUS CreateStatus,

    IN OUT PGET_NAME_CONTROL NameControl

    )

/*++



Routine Description:



    This routine will try and get the name of the given file object.  This

    is guaranteed to always return a printable string (though it may be NULL).

    This will allocate a buffer if it needs to.



Arguments:

    FileObject - the file object we want the name for



    CreateStatus - status of the create operation



    NameControl - control structure used for retrieving the name.  It keeps

        track if a buffer was allocated or if we are using the internal

        buffer.



Return Value:



    Pointer to the unicode string with the name



--*/

{

    POBJECT_NAME_INFORMATION nameInfo;

    NTSTATUS status;

    ULONG size;

    ULONG bufferSize;



    //

    //  Mark we have not allocated the buffer

    //



    NameControl->allocatedBuffer = NULL;



    //

    //  Use the small buffer in the structure (that will handle most cases)

    //  for the name

    //



    nameInfo = (POBJECT_NAME_INFORMATION)NameControl->smallBuffer;

    bufferSize = sizeof(NameControl->smallBuffer);



    //

    //  If the open succeeded, get the name of the file, if it

    //  failed, get the name of the device.

    //

        

    status = ObQueryNameString(

                  (NT_SUCCESS( CreateStatus ) ?

                    (PVOID)FileObject :

                    (PVOID)FileObject->DeviceObject),

                  nameInfo,

                  bufferSize,

                  &size );



    //

    //  See if the buffer was to small

    //



    if (status == STATUS_BUFFER_OVERFLOW) {



        //

        //  The buffer was too small, allocate one big enough

        //



        bufferSize = size + sizeof(WCHAR);



        NameControl->allocatedBuffer = ExAllocatePoolWithTag( 

                                            NonPagedPool,

                                            bufferSize,

                                            SFLT_POOL_TAG );



        if (NULL == NameControl->allocatedBuffer) {



            //

            //  Failed allocating a buffer, return an empty string for the name

            //



            RtlInitEmptyUnicodeString(

                (PUNICODE_STRING)&NameControl->smallBuffer,

                (PWCHAR)(NameControl->smallBuffer + sizeof(UNICODE_STRING)),

                (USHORT)(sizeof(NameControl->smallBuffer) - sizeof(UNICODE_STRING)) );



            return (PUNICODE_STRING)&NameControl->smallBuffer;

        }



        //

        //  Set the allocated buffer and get the name again

        //



        nameInfo = (POBJECT_NAME_INFORMATION)NameControl->allocatedBuffer;



        status = ObQueryNameString(

                      FileObject,

                      nameInfo,

                      bufferSize,

                      &size );

    }



    //

    //  If we got a name and an error opening the file then we

    //  just received the device name.  Grab the rest of the name

    //  from the FileObject (note that this can only be done if being called

    //  from Create).  This only happens if we got an error back from the

    //  create.

    //



    if (NT_SUCCESS( status ) && 

                    !NT_SUCCESS( CreateStatus )) {



        ULONG newSize;

        PCHAR newBuffer;

        POBJECT_NAME_INFORMATION newNameInfo;



        //

        //  Calculate the size of the buffer we will need to hold

        //  the combined names

        //



        newSize = size + FileObject->FileName.Length;



        //

        //  If there is a related file object add in the length

        //  of that plus space for a separator

        //



        if (NULL != FileObject->RelatedFileObject) {



            newSize += FileObject->RelatedFileObject->FileName.Length + 

                       sizeof(WCHAR);

        }



        //

        //  See if it will fit in the existing buffer

        //



        if (newSize > bufferSize) {



            //

            //  It does not fit, allocate a bigger buffer

            //



            newBuffer = ExAllocatePoolWithTag( 

                                    NonPagedPool,

                                    newSize,

                                    SFLT_POOL_TAG );



            if (NULL == newBuffer) {



                //

                //  Failed allocating a buffer, return an empty string for the name

                //



                RtlInitEmptyUnicodeString(

                    (PUNICODE_STRING)&NameControl->smallBuffer,

                    (PWCHAR)(NameControl->smallBuffer + sizeof(UNICODE_STRING)),

                    (USHORT)(sizeof(NameControl->smallBuffer) - sizeof(UNICODE_STRING)) );



                return (PUNICODE_STRING)&NameControl->smallBuffer;

            }



            //

            //  Now initialize the new buffer with the information

            //  from the old buffer.

            //



            newNameInfo = (POBJECT_NAME_INFORMATION)newBuffer;



            RtlInitEmptyUnicodeString(

                &newNameInfo->Name,

                (PWCHAR)(newBuffer + sizeof(OBJECT_NAME_INFORMATION)),

                (USHORT)(newSize - sizeof(OBJECT_NAME_INFORMATION)) );



            RtlCopyUnicodeString( &newNameInfo->Name, 

                                  &nameInfo->Name );



            //

            //  Free the old allocated buffer (if there is one)

            //  and save off the new allocated buffer address.  It

            //  would be very rare that we should have to free the

            //  old buffer because device names should always fit

            //  inside it.

            //



            if (NULL != NameControl->allocatedBuffer) {



                ExFreePool( NameControl->allocatedBuffer );

            }



            //

            //  Readjust our pointers

            //



            NameControl->allocatedBuffer = newBuffer;

            bufferSize = newSize;

            nameInfo = newNameInfo;



        } else {



            //

            //  The MaximumLength was set by ObQueryNameString to

            //  one char larger then the length.  Set it to the

            //  true size of the buffer (so we can append the names)

            //



            nameInfo->Name.MaximumLength = (USHORT)(bufferSize - 

                                  sizeof(OBJECT_NAME_INFORMATION));

        }



        //

        //  If there is a related file object, append that name

        //  first onto the device object along with a separator

        //  character

        //



        if (NULL != FileObject->RelatedFileObject) {



            RtlAppendUnicodeStringToString(

                    &nameInfo->Name,

                    &FileObject->RelatedFileObject->FileName );



            RtlAppendUnicodeToString( &nameInfo->Name, L"\\" );

        }



        //

        //  Append the name from the file object

        //



        RtlAppendUnicodeStringToString(

                &nameInfo->Name,

                &FileObject->FileName );



        ASSERT(nameInfo->Name.Length <= nameInfo->Name.MaximumLength);

    }



    //

    //  Return the name

    //



    return &nameInfo->Name;

}

typedef struct _GET_NAME_CONTROL {



    PCHAR allocatedBuffer;

    CHAR smallBuffer[256];

    

} GET_NAME_CONTROL, *PGET_NAME_CONTROL;



调用时候

GET_NAME_CONTROL nameControl;

PUNICODE_STRING name = SfGetFileName(FileObject, 

                          STATUS_SUCCESS, 

                          &nameControl );

调用完，在调用这个函数，释放内存

SfGetFileNameCleanup( &nameControl );





VOID

SfGetFileNameCleanup(

    IN OUT PGET_NAME_CONTROL NameControl

    )

/*++



Routine Description:



    This will see if a buffer was allocated and will free it if it was



Arguments:



    NameControl - control structure used for retrieving the name.  It keeps

        track if a buffer was allocated or if we are using the internal

        buffer.



Return Value:



    None



--*/

{



    if (NULL != NameControl->allocatedBuffer) {



        ExFreePool( NameControl->allocatedBuffer);

        NameControl->allocatedBuffer = NULL;

    }

}