ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

server: Pass the file instead of the device pointer in all IRP requests.

This commit is contained in:
Alexandre Julliard 2015-07-08 19:04:32 +09:00
parent 13499a3d11
commit bf26950194
5 changed files with 70 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
dlls/ntoskrnl.exe/ntoskrnl.c
View File

@ -131,14 +131,6 @@ static HANDLE get_device_manager(void)
static NTSTATUS dispatch_irp( DEVICE_OBJECT *device, IRP *irp ) static NTSTATUS dispatch_irp( DEVICE_OBJECT *device, IRP *irp )
{ {
LARGE_INTEGER count; LARGE_INTEGER count;
FILE_OBJECT file;
irp->RequestorMode = UserMode;
irp->Tail.Overlay.OriginalFileObject = &file;
memset( &file, 0x88, sizeof(file) );
file.FsContext = NULL;
file.FsContext2 = NULL;
KeQueryTickCount( &count ); /* update the global KeTickCount */ KeQueryTickCount( &count ); /* update the global KeTickCount */
@ -185,6 +177,7 @@ static NTSTATUS dispatch_create( const irp_params_t *params, void *in_buff, ULON
irpsp->Parameters.Create.EaLength = 0; irpsp->Parameters.Create.EaLength = 0;
irp->Tail.Overlay.OriginalFileObject = file; irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
irp->AssociatedIrp.SystemBuffer = NULL; irp->AssociatedIrp.SystemBuffer = NULL;
irp->UserBuffer = NULL; irp->UserBuffer = NULL;
irp->UserIosb = irp_handle; /* note: we abuse UserIosb to store the server irp handle */ irp->UserIosb = irp_handle; /* note: we abuse UserIosb to store the server irp handle */
@ -227,6 +220,7 @@ static NTSTATUS dispatch_close( const irp_params_t *params, void *in_buff, ULONG
irpsp->Parameters.Create.EaLength = 0; irpsp->Parameters.Create.EaLength = 0;
irp->Tail.Overlay.OriginalFileObject = file; irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
irp->AssociatedIrp.SystemBuffer = NULL; irp->AssociatedIrp.SystemBuffer = NULL;
irp->UserBuffer = NULL; irp->UserBuffer = NULL;
irp->UserIosb = irp_handle; /* note: we abuse UserIosb to store the server irp handle */ irp->UserIosb = irp_handle; /* note: we abuse UserIosb to store the server irp handle */
@ -247,11 +241,15 @@ static NTSTATUS dispatch_read( const irp_params_t *params, void *in_buff, ULONG
void *out_buff; void *out_buff;
LARGE_INTEGER offset; LARGE_INTEGER offset;
IO_STACK_LOCATION *irpsp; IO_STACK_LOCATION *irpsp;
DEVICE_OBJECT *device = wine_server_get_ptr( params->read.device ); DEVICE_OBJECT *device;
FILE_OBJECT *file = wine_server_get_ptr( params->read.file );
if (!file) return STATUS_INVALID_HANDLE;
device = file->DeviceObject;
if (!device->DriverObject->MajorFunction[IRP_MJ_READ]) return STATUS_NOT_SUPPORTED; if (!device->DriverObject->MajorFunction[IRP_MJ_READ]) return STATUS_NOT_SUPPORTED;
TRACE( "device %p size %u\n", device, out_size ); TRACE( "device %p file %p size %u\n", device, file, out_size );
if (!(out_buff = HeapAlloc( GetProcessHeap(), 0, out_size ))) return STATUS_NO_MEMORY; if (!(out_buff = HeapAlloc( GetProcessHeap(), 0, out_size ))) return STATUS_NO_MEMORY;
@ -265,6 +263,9 @@ static NTSTATUS dispatch_read( const irp_params_t *params, void *in_buff, ULONG
return STATUS_NO_MEMORY; return STATUS_NO_MEMORY;
} }
irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
irpsp = IoGetNextIrpStackLocation( irp ); irpsp = IoGetNextIrpStackLocation( irp );
irpsp->Parameters.Read.Key = params->read.key; irpsp->Parameters.Read.Key = params->read.key;
@ -278,11 +279,15 @@ static NTSTATUS dispatch_write( const irp_params_t *params, void *in_buff, ULONG
IRP *irp; IRP *irp;
LARGE_INTEGER offset; LARGE_INTEGER offset;
IO_STACK_LOCATION *irpsp; IO_STACK_LOCATION *irpsp;
DEVICE_OBJECT *device = wine_server_get_ptr( params->write.device ); DEVICE_OBJECT *device;
FILE_OBJECT *file = wine_server_get_ptr( params->write.file );
if (!file) return STATUS_INVALID_HANDLE;
device = file->DeviceObject;
if (!device->DriverObject->MajorFunction[IRP_MJ_WRITE]) return STATUS_NOT_SUPPORTED; if (!device->DriverObject->MajorFunction[IRP_MJ_WRITE]) return STATUS_NOT_SUPPORTED;
TRACE( "device %p size %u\n", device, in_size ); TRACE( "device %p file %p size %u\n", device, file, in_size );
offset.QuadPart = params->write.pos; offset.QuadPart = params->write.pos;
@ -291,6 +296,9 @@ static NTSTATUS dispatch_write( const irp_params_t *params, void *in_buff, ULONG
&offset, NULL, irp_handle ))) &offset, NULL, irp_handle )))
return STATUS_NO_MEMORY; return STATUS_NO_MEMORY;
irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
irpsp = IoGetNextIrpStackLocation( irp ); irpsp = IoGetNextIrpStackLocation( irp );
irpsp->Parameters.Write.Key = params->write.key; irpsp->Parameters.Write.Key = params->write.key;
@ -302,17 +310,24 @@ static NTSTATUS dispatch_flush( const irp_params_t *params, void *in_buff, ULONG
ULONG out_size, HANDLE irp_handle ) ULONG out_size, HANDLE irp_handle )
{ {
IRP *irp; IRP *irp;
DEVICE_OBJECT *device = wine_server_get_ptr( params->flush.device ); DEVICE_OBJECT *device;
FILE_OBJECT *file = wine_server_get_ptr( params->flush.file );
if (!file) return STATUS_INVALID_HANDLE;
device = file->DeviceObject;
if (!device->DriverObject->MajorFunction[IRP_MJ_FLUSH_BUFFERS]) return STATUS_NOT_SUPPORTED; if (!device->DriverObject->MajorFunction[IRP_MJ_FLUSH_BUFFERS]) return STATUS_NOT_SUPPORTED;
TRACE( "device %p\n", device ); TRACE( "device %p file %p\n", device, file );
/* note: we abuse UserIosb to store the server irp handle */ /* note: we abuse UserIosb to store the server irp handle */
if (!(irp = IoBuildSynchronousFsdRequest( IRP_MJ_FLUSH_BUFFERS, device, in_buff, in_size, if (!(irp = IoBuildSynchronousFsdRequest( IRP_MJ_FLUSH_BUFFERS, device, in_buff, in_size,
NULL, NULL, irp_handle ))) NULL, NULL, irp_handle )))
return STATUS_NO_MEMORY; return STATUS_NO_MEMORY;
irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
return dispatch_irp( device, irp ); return dispatch_irp( device, irp );
} }
@ -322,11 +337,16 @@ static NTSTATUS dispatch_ioctl( const irp_params_t *params, void *in_buff, ULONG
{ {
IRP *irp; IRP *irp;
void *out_buff = NULL; void *out_buff = NULL;
DEVICE_OBJECT *device = wine_server_get_ptr( params->ioctl.device ); DEVICE_OBJECT *device;
FILE_OBJECT *file = wine_server_get_ptr( params->ioctl.file );
if (!file) return STATUS_INVALID_HANDLE;
device = file->DeviceObject;
if (!device->DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL]) return STATUS_NOT_SUPPORTED; if (!device->DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL]) return STATUS_NOT_SUPPORTED;
TRACE( "ioctl %x device %p in_size %u out_size %u\n", params->ioctl.code, device, in_size, out_size ); TRACE( "ioctl %x device %p file %p in_size %u out_size %u\n",
params->ioctl.code, device, file, in_size, out_size );
if ((params->ioctl.code & 3) == METHOD_BUFFERED) out_size = max( in_size, out_size ); if ((params->ioctl.code & 3) == METHOD_BUFFERED) out_size = max( in_size, out_size );
@ -348,6 +368,10 @@ static NTSTATUS dispatch_ioctl( const irp_params_t *params, void *in_buff, ULONG
HeapFree( GetProcessHeap(), 0, out_buff ); HeapFree( GetProcessHeap(), 0, out_buff );
return STATUS_NO_MEMORY; return STATUS_NO_MEMORY;
} }
irp->Tail.Overlay.OriginalFileObject = file;
irp->RequestorMode = UserMode;
return dispatch_irp( device, irp ); return dispatch_irp( device, irp );
} }

10
include/wine/server_protocol.h
View File

@ -652,27 +652,27 @@ typedef union
{ {
unsigned int major; unsigned int major;
unsigned int key; unsigned int key;
client_ptr_t device; client_ptr_t file;
file_pos_t pos; file_pos_t pos;
} read; } read;
struct struct
{ {
unsigned int major; unsigned int major;
unsigned int key; unsigned int key;
client_ptr_t device; client_ptr_t file;
file_pos_t pos; file_pos_t pos;
} write; } write;
struct struct
{ {
unsigned int major; unsigned int major;
int __pad; int __pad;
client_ptr_t device; client_ptr_t file;
} flush; } flush;
struct struct
{ {
unsigned int major; unsigned int major;
ioctl_code_t code; ioctl_code_t code;
client_ptr_t device; client_ptr_t file;
} ioctl; } ioctl;
} irp_params_t; } irp_params_t;
@ -6096,6 +6096,6 @@ union generic_reply
struct terminate_job_reply terminate_job_reply; struct terminate_job_reply terminate_job_reply;
}; };
#define SERVER_PROTOCOL_VERSION 480 #define SERVER_PROTOCOL_VERSION 481
#endif /* __WINE_WINE_SERVER_PROTOCOL_H */ #endif /* __WINE_WINE_SERVER_PROTOCOL_H */

12
server/device.c
View File

@ -475,6 +475,10 @@ static void set_file_user_ptr( struct device_file *file, client_ptr_t ptr )
switch (irp->params.major) switch (irp->params.major)
{ {
case IRP_MJ_CLOSE: irp->params.close.file = ptr; break; case IRP_MJ_CLOSE: irp->params.close.file = ptr; break;
case IRP_MJ_READ: irp->params.read.file = ptr; break;
case IRP_MJ_WRITE: irp->params.write.file = ptr; break;
case IRP_MJ_FLUSH_BUFFERS: irp->params.flush.file = ptr; break;
case IRP_MJ_DEVICE_CONTROL: irp->params.ioctl.file = ptr; break;
} }
} }
} }
@ -514,7 +518,7 @@ static obj_handle_t device_file_read( struct fd *fd, const async_data_t *async_d
params.read.major = IRP_MJ_READ; params.read.major = IRP_MJ_READ;
params.read.key = 0; params.read.key = 0;
params.read.pos = pos; params.read.pos = pos;
params.read.device = file->device->user_ptr; params.read.file = file->user_ptr;
irp = create_irp( file, &params, NULL, 0, get_reply_max_size() ); irp = create_irp( file, &params, NULL, 0, get_reply_max_size() );
if (!irp) return 0; if (!irp) return 0;
@ -535,7 +539,7 @@ static obj_handle_t device_file_write( struct fd *fd, const async_data_t *async_
params.write.major = IRP_MJ_WRITE; params.write.major = IRP_MJ_WRITE;
params.write.key = 0; params.write.key = 0;
params.write.pos = pos; params.write.pos = pos;
params.write.device = file->device->user_ptr; params.write.file = file->user_ptr;
irp = create_irp( file, &params, get_req_data(), get_req_data_size(), 0 ); irp = create_irp( file, &params, get_req_data(), get_req_data_size(), 0 );
if (!irp) return 0; if (!irp) return 0;
@ -553,7 +557,7 @@ static obj_handle_t device_file_flush( struct fd *fd, const async_data_t *async_
irp_params_t params; irp_params_t params;
params.flush.major = IRP_MJ_FLUSH_BUFFERS; params.flush.major = IRP_MJ_FLUSH_BUFFERS;
params.flush.device = file->device->user_ptr; params.flush.file = file->user_ptr;
irp = create_irp( file, &params, NULL, 0, 0 ); irp = create_irp( file, &params, NULL, 0, 0 );
if (!irp) return 0; if (!irp) return 0;
@ -573,7 +577,7 @@ static obj_handle_t device_file_ioctl( struct fd *fd, ioctl_code_t code, const a
params.ioctl.major = IRP_MJ_DEVICE_CONTROL; params.ioctl.major = IRP_MJ_DEVICE_CONTROL;
params.ioctl.code = code; params.ioctl.code = code;
params.ioctl.device = file->device->user_ptr; params.ioctl.file = file->user_ptr;
irp = create_irp( file, &params, get_req_data(), get_req_data_size(), irp = create_irp( file, &params, get_req_data(), get_req_data_size(),
get_reply_max_size() ); get_reply_max_size() );

8
server/protocol.def
View File

@ -668,27 +668,27 @@ typedef union
{ {
unsigned int major; /* IRP_MJ_READ */ unsigned int major; /* IRP_MJ_READ */
unsigned int key; /* driver key */ unsigned int key; /* driver key */
client_ptr_t device; /* opaque ptr for the device */ client_ptr_t file; /* opaque ptr for the file object */
file_pos_t pos; /* file position */ file_pos_t pos; /* file position */
} read; } read;
struct struct
{ {
unsigned int major; /* IRP_MJ_WRITE */ unsigned int major; /* IRP_MJ_WRITE */
unsigned int key; /* driver key */ unsigned int key; /* driver key */
client_ptr_t device; /* opaque ptr for the device */ client_ptr_t file; /* opaque ptr for the file object */
file_pos_t pos; /* file position */ file_pos_t pos; /* file position */
} write; } write;
struct struct
{ {
unsigned int major; /* IRP_MJ_FLUSH_BUFFERS */ unsigned int major; /* IRP_MJ_FLUSH_BUFFERS */
int __pad; int __pad;
client_ptr_t device; /* opaque ptr for the device */ client_ptr_t file; /* opaque ptr for the file object */
} flush; } flush;
struct struct
{ {
unsigned int major; /* IRP_MJ_DEVICE_CONTROL */ unsigned int major; /* IRP_MJ_DEVICE_CONTROL */
ioctl_code_t code; /* ioctl code */ ioctl_code_t code; /* ioctl code */
client_ptr_t device; /* opaque ptr for the device */ client_ptr_t file; /* opaque ptr for the file object */
} ioctl; } ioctl;
} irp_params_t; } irp_params_t;

8
server/trace.c
View File

@ -330,24 +330,24 @@ static void dump_irp_params( const char *prefix, const irp_params_t *data )
case IRP_MJ_READ: case IRP_MJ_READ:
fprintf( stderr, "%s{major=READ,key=%08x", prefix, data->read.key ); fprintf( stderr, "%s{major=READ,key=%08x", prefix, data->read.key );
dump_uint64( ",pos=", &data->read.pos ); dump_uint64( ",pos=", &data->read.pos );
dump_uint64( ",device=", &data->read.device ); dump_uint64( ",file=", &data->read.file );
fputc( '}', stderr ); fputc( '}', stderr );
break; break;
case IRP_MJ_WRITE: case IRP_MJ_WRITE:
fprintf( stderr, "%s{major=WRITE,key=%08x", prefix, data->write.key ); fprintf( stderr, "%s{major=WRITE,key=%08x", prefix, data->write.key );
dump_uint64( ",pos=", &data->write.pos ); dump_uint64( ",pos=", &data->write.pos );
dump_uint64( ",device=", &data->write.device ); dump_uint64( ",file=", &data->write.file );
fputc( '}', stderr ); fputc( '}', stderr );
break; break;
case IRP_MJ_FLUSH_BUFFERS: case IRP_MJ_FLUSH_BUFFERS:
fprintf( stderr, "%s{major=FLUSH_BUFFERS", prefix ); fprintf( stderr, "%s{major=FLUSH_BUFFERS", prefix );
dump_uint64( ",device=", &data->flush.device ); dump_uint64( ",file=", &data->flush.file );
fputc( '}', stderr ); fputc( '}', stderr );
break; break;
case IRP_MJ_DEVICE_CONTROL: case IRP_MJ_DEVICE_CONTROL:
fprintf( stderr, "%s{major=DEVICE_CONTROL", prefix ); fprintf( stderr, "%s{major=DEVICE_CONTROL", prefix );
dump_ioctl_code( ",code=", &data->ioctl.code ); dump_ioctl_code( ",code=", &data->ioctl.code );
dump_uint64( ",device=", &data->ioctl.device ); dump_uint64( ",file=", &data->ioctl.file );
fputc( '}', stderr ); fputc( '}', stderr );
break; break;
case IRP_MJ_MAXIMUM_FUNCTION + 1: /* invalid */ case IRP_MJ_MAXIMUM_FUNCTION + 1: /* invalid */