- get mpu401, parallel and ramdrv to work on msvc.
[reactos.git] / reactos / drivers / dd / ramdrv / ramdrv.c
1 #include <ntddk.h>
2 #include <ntdddisk.h>
3 #include "ramdrv.h"
4 #include <debug.h>
5 #include "../../lib/bzip2/bzlib.h"
6
7 static NTSTATUS STDCALL RamdrvDispatchDeviceControl(PDEVICE_OBJECT DeviceObject,
8 PIRP Irp)
9 {
10 PIO_STACK_LOCATION IrpStack;
11 ULONG ControlCode, InputLength, OutputLength;
12 NTSTATUS Status;
13
14 DPRINT("RamdrvDispatchDeviceControl\n");
15
16 IrpStack = IoGetCurrentIrpStackLocation(Irp);
17 ControlCode = IrpStack->Parameters.DeviceIoControl.IoControlCode;
18 InputLength = IrpStack->Parameters.DeviceIoControl.InputBufferLength;
19 OutputLength = IrpStack->Parameters.DeviceIoControl.OutputBufferLength;
20
21 switch (ControlCode)
22 {
23 case IOCTL_DISK_GET_DRIVE_GEOMETRY:
24 if (OutputLength < sizeof(DISK_GEOMETRY))
25 {
26 Status = STATUS_INVALID_PARAMETER;
27 }
28 else
29 {
30 PDISK_GEOMETRY Geometry = Irp->AssociatedIrp.SystemBuffer;
31 Geometry->MediaType = F3_1Pt44_512;
32 Geometry->Cylinders.QuadPart = 80;
33 Geometry->TracksPerCylinder = 2 * 18;
34 Geometry->SectorsPerTrack = 18;
35 Geometry->BytesPerSector = 512;
36 Status = STATUS_SUCCESS;
37 Irp->IoStatus.Information = sizeof(DISK_GEOMETRY);
38 }
39 break;
40 default:
41 Status = STATUS_INVALID_DEVICE_REQUEST;
42 }
43 Irp->IoStatus.Status = Status;
44 IoCompleteRequest(Irp, NT_SUCCESS(Status) ? IO_DISK_INCREMENT : IO_NO_INCREMENT);
45 return Status;
46 }
47
48 static NTSTATUS STDCALL RamdrvDispatchReadWrite(PDEVICE_OBJECT DeviceObject,
49 PIRP Irp)
50 {
51 PRAMDRV_DEVICE_EXTENSION devext = (PRAMDRV_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
52 PIO_STACK_LOCATION Stk = IoGetCurrentIrpStackLocation( Irp );
53
54 if( Stk->Parameters.Read.ByteOffset.u.HighPart ||
55 Stk->Parameters.Read.ByteOffset.u.LowPart >= devext->Size )
56 {
57 Irp->IoStatus.Status = STATUS_END_OF_FILE;
58 Irp->IoStatus.Information = 0;
59 IoCompleteRequest( Irp, 0 );
60 return STATUS_END_OF_FILE;
61 }
62 if( (Stk->Parameters.Read.ByteOffset.u.LowPart + Stk->Parameters.Read.Length) > devext->Size )
63 Stk->Parameters.Read.Length = devext->Size - Stk->Parameters.Read.ByteOffset.u.LowPart;
64 if( Stk->MajorFunction == IRP_MJ_READ )
65 RtlCopyMemory( MmGetSystemAddressForMdl( Irp->MdlAddress ),
66 (PVOID)((ULONG_PTR)devext->Buffer + Stk->Parameters.Read.ByteOffset.u.LowPart),
67 Stk->Parameters.Read.Length );
68 else RtlCopyMemory( (PVOID)((ULONG_PTR)devext->Buffer + Stk->Parameters.Read.ByteOffset.u.LowPart),
69 MmGetSystemAddressForMdl( Irp->MdlAddress ),
70 Stk->Parameters.Read.Length );
71 Irp->IoStatus.Status = STATUS_SUCCESS;
72 Irp->IoStatus.Information = Stk->Parameters.Read.Length;
73 IoCompleteRequest( Irp, 0 );
74 return STATUS_SUCCESS;
75 }
76
77 static NTSTATUS STDCALL RamdrvDispatchOpenClose(PDEVICE_OBJECT DeviceObject,
78 PIRP Irp)
79 {
80 DPRINT("RamdrvDispatchOpenClose\n");
81 return STATUS_SUCCESS;
82 }
83
84 NTSTATUS STDCALL DriverEntry(IN PDRIVER_OBJECT DriverObject,
85 IN PUNICODE_STRING RegistryPath)
86 {
87 UNICODE_STRING DeviceName = RTL_CONSTANT_STRING(L"\\Device\\Ramdisk");
88 NTSTATUS Status;
89 PDEVICE_OBJECT DeviceObject;
90 PRAMDRV_DEVICE_EXTENSION devext;
91 UNICODE_STRING LinkName = RTL_CONSTANT_STRING(L"\\??\\Z:");
92 UNICODE_STRING ImageName = RTL_CONSTANT_STRING(L"\\Device\\Floppy0\\ramdisk.bz2");
93 HANDLE file;
94 OBJECT_ATTRIBUTES objattr;
95 IO_STATUS_BLOCK iosb;
96 LARGE_INTEGER allocsize;
97 HANDLE event;
98 void *tbuff;
99 unsigned int dstlen = 1024 * 1440;
100 FILE_STANDARD_INFORMATION finfo;
101 ULONG err;
102
103 DPRINT("Ramdisk driver\n");
104
105 /* Export other driver entry points... */
106 DriverObject->MajorFunction[IRP_MJ_CREATE] = RamdrvDispatchOpenClose;
107 DriverObject->MajorFunction[IRP_MJ_CLOSE] = RamdrvDispatchOpenClose;
108 DriverObject->MajorFunction[IRP_MJ_READ] = RamdrvDispatchReadWrite;
109 DriverObject->MajorFunction[IRP_MJ_WRITE] = RamdrvDispatchReadWrite;
110 DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = RamdrvDispatchDeviceControl;
111
112
113 // create device and symbolic link
114 Status = IoCreateDevice( DriverObject,
115 sizeof( RAMDRV_DEVICE_EXTENSION ),
116 &DeviceName,
117 FILE_DEVICE_DISK,
118 0,
119 FALSE,
120 &DeviceObject );
121 if( !NT_SUCCESS( Status ) )
122 return Status;
123 DeviceObject->Flags |= DO_DIRECT_IO;
124 devext = (PRAMDRV_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
125 devext->Size = 1440 * 1024;
126 devext->Buffer = ExAllocatePool( PagedPool, devext->Size );
127 if( !devext->Buffer )
128 {
129 Status = STATUS_INSUFFICIENT_RESOURCES;
130 goto cleandevice;
131 }
132 IoCreateSymbolicLink( &LinkName, &DeviceName );
133
134 InitializeObjectAttributes( &objattr,
135 &ImageName,
136 0,
137 0,
138 0 );
139 allocsize.u.LowPart = allocsize.u.HighPart = 0;
140
141 Status = ZwOpenFile( &file,
142 GENERIC_READ,
143 &objattr,
144 &iosb,
145 FILE_SHARE_READ,
146 FILE_NO_INTERMEDIATE_BUFFERING );
147
148 if( !NT_SUCCESS( Status ) )
149 {
150 DPRINT( "Failed to open floppy\n" );
151 goto cleanbuffer;
152 }
153
154 InitializeObjectAttributes( &objattr,
155 0,
156 0,
157 0,
158 0 );
159 Status = ZwCreateEvent( &event,
160 0,
161 &objattr,
162 NotificationEvent,
163 FALSE );
164 if( !NT_SUCCESS( Status ) )
165 {
166 DPRINT( "Failed to create event\n" );
167 goto cleanfile;
168 }
169
170 Status = ZwQueryInformationFile( file,
171 &iosb,
172 &finfo,
173 sizeof( finfo ),
174 FileStandardInformation );
175
176 if( !NT_SUCCESS( Status ) )
177 {
178 DPRINT1( "Failed to query file information\n" );
179 goto cleanevent;
180 }
181 tbuff = ExAllocatePool( PagedPool, finfo.EndOfFile.u.LowPart );
182 if( !tbuff )
183 {
184 DPRINT1( "Failed to allocate buffer of size %d\n", finfo.EndOfFile.u.LowPart );
185 Status = STATUS_INSUFFICIENT_RESOURCES;
186 goto cleanevent;
187 }
188
189 Status = ZwReadFile( file,
190 event,
191 0,
192 0,
193 &iosb,
194 tbuff,
195 finfo.EndOfFile.u.LowPart,
196 &allocsize,
197 0 );
198
199 if( !NT_SUCCESS( Status ) )
200 {
201 DPRINT( "Failed to read floppy\n" );
202 goto cleantbuff;
203 }
204 Status = ZwWaitForSingleObject( event, FALSE, 0 );
205 if( Status != STATUS_WAIT_0 || !NT_SUCCESS( iosb.Status ) )
206 {
207 DPRINT( "Failed to read floppy\n" );
208 goto cleantbuff;
209 }
210 DPRINT( "RAMDRV: Read in %d bytes, decompressing now\n", iosb.Information );
211 err = BZ2_bzBuffToBuffDecompress( devext->Buffer,
212 &dstlen,
213 tbuff,
214 iosb.Information,
215 1,
216 0 );
217 if( err == 0 )
218 {
219 DPRINT( "RAMDRV: Image Decompressed\n");
220 }
221 else DbgPrint( "RAMDRV: Failed to decomparess image, error: %d\n", err );
222 ExFreePool( tbuff );
223 ZwClose( file );
224 ZwClose( event );
225 return STATUS_SUCCESS;
226
227 cleantbuff:
228 ExFreePool( tbuff );
229 cleanevent:
230 ZwClose( event );
231 cleanfile:
232 ZwClose( file );
233 cleanbuffer:
234 ExFreePool( devext->Buffer );
235
236 cleandevice:
237 IoDeleteDevice( DeviceObject );
238 for(;;);
239
240 return Status;
241 }
242