1 | /*
|
---|
2 | * Copyright (c) 2009 Jiri Svoboda
|
---|
3 | * All rights reserved.
|
---|
4 | *
|
---|
5 | * Redistribution and use in source and binary forms, with or without
|
---|
6 | * modification, are permitted provided that the following conditions
|
---|
7 | * are met:
|
---|
8 | *
|
---|
9 | * - Redistributions of source code must retain the above copyright
|
---|
10 | * notice, this list of conditions and the following disclaimer.
|
---|
11 | * - Redistributions in binary form must reproduce the above copyright
|
---|
12 | * notice, this list of conditions and the following disclaimer in the
|
---|
13 | * documentation and/or other materials provided with the distribution.
|
---|
14 | * - The name of the author may not be used to endorse or promote products
|
---|
15 | * derived from this software without specific prior written permission.
|
---|
16 | *
|
---|
17 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
---|
18 | * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
---|
19 | * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
---|
20 | * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
---|
21 | * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
---|
22 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
---|
23 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
---|
24 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
25 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
---|
26 | * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
27 | */
|
---|
28 |
|
---|
29 | /** @addtogroup bd
|
---|
30 | * @{
|
---|
31 | */
|
---|
32 |
|
---|
33 | /**
|
---|
34 | * @file
|
---|
35 | * @brief ATA disk driver
|
---|
36 | *
|
---|
37 | * This driver currently works only with CHS addressing and uses PIO.
|
---|
38 | * Currently based on the (now obsolete) ANSI X3.221-1994 (ATA-1) standard.
|
---|
39 | * At this point only reading is possible, not writing.
|
---|
40 | *
|
---|
41 | * The driver services a single controller which can have up to two disks
|
---|
42 | * attached.
|
---|
43 | */
|
---|
44 |
|
---|
45 | #include <stdio.h>
|
---|
46 | #include <libarch/ddi.h>
|
---|
47 | #include <ddi.h>
|
---|
48 | #include <ipc/ipc.h>
|
---|
49 | #include <ipc/bd.h>
|
---|
50 | #include <async.h>
|
---|
51 | #include <as.h>
|
---|
52 | #include <fibril_sync.h>
|
---|
53 | #include <devmap.h>
|
---|
54 | #include <sys/types.h>
|
---|
55 | #include <errno.h>
|
---|
56 | #include <bool.h>
|
---|
57 |
|
---|
58 | #include "ata_bd.h"
|
---|
59 |
|
---|
60 | #define NAME "ata_bd"
|
---|
61 |
|
---|
62 | static const size_t block_size = 512;
|
---|
63 | static size_t comm_size;
|
---|
64 |
|
---|
65 | static uintptr_t cmd_physical = 0x1f0;
|
---|
66 | static uintptr_t ctl_physical = 0x170;
|
---|
67 | static ata_cmd_t *cmd;
|
---|
68 | static ata_ctl_t *ctl;
|
---|
69 |
|
---|
70 | /** Per-disk state. */
|
---|
71 | static disk_t disk[MAX_DISKS];
|
---|
72 |
|
---|
73 | static int ata_bd_init(void);
|
---|
74 | static void ata_bd_connection(ipc_callid_t iid, ipc_call_t *icall);
|
---|
75 | static int ata_bd_rdwr(int disk_id, ipcarg_t method, off_t offset, size_t size,
|
---|
76 | void *buf);
|
---|
77 | static int ata_bd_read_block(int disk_id, uint64_t blk_idx, size_t blk_cnt,
|
---|
78 | void *buf);
|
---|
79 | static int ata_bd_write_block(int disk_id, uint64_t blk_idx, size_t blk_cnt,
|
---|
80 | const void *buf);
|
---|
81 | static int drive_identify(int drive_id, disk_t *d);
|
---|
82 |
|
---|
83 | int main(int argc, char **argv)
|
---|
84 | {
|
---|
85 | uint8_t status;
|
---|
86 | char name[16];
|
---|
87 | int i, rc;
|
---|
88 | int n_disks;
|
---|
89 |
|
---|
90 | printf(NAME ": ATA disk driver\n");
|
---|
91 |
|
---|
92 | printf("I/O address 0x%x\n", cmd_physical);
|
---|
93 |
|
---|
94 | if (ata_bd_init() != EOK)
|
---|
95 | return -1;
|
---|
96 |
|
---|
97 | /* Put drives to reset, disable interrupts. */
|
---|
98 | printf("Reset drives...\n");
|
---|
99 | pio_write_8(&ctl->device_control, DCR_SRST);
|
---|
100 | /* FIXME: Find out how to do this properly. */
|
---|
101 | async_usleep(100);
|
---|
102 | pio_write_8(&ctl->device_control, 0);
|
---|
103 |
|
---|
104 | do {
|
---|
105 | status = pio_read_8(&cmd->status);
|
---|
106 | } while ((status & SR_BSY) != 0);
|
---|
107 | printf("Done\n");
|
---|
108 |
|
---|
109 | printf("Status = 0x%x\n", pio_read_8(&cmd->status));
|
---|
110 |
|
---|
111 | (void) drive_identify(0, &disk[0]);
|
---|
112 | (void) drive_identify(1, &disk[1]);
|
---|
113 |
|
---|
114 | n_disks = 0;
|
---|
115 |
|
---|
116 | for (i = 0; i < MAX_DISKS; i++) {
|
---|
117 | /* Skip unattached drives. */
|
---|
118 | if (disk[i].present == false)
|
---|
119 | continue;
|
---|
120 |
|
---|
121 | snprintf(name, 16, "disk%d", i);
|
---|
122 | rc = devmap_device_register(name, &disk[i].dev_handle);
|
---|
123 | if (rc != EOK) {
|
---|
124 | devmap_hangup_phone(DEVMAP_DRIVER);
|
---|
125 | printf(NAME ": Unable to register device %s.\n",
|
---|
126 | name);
|
---|
127 | return rc;
|
---|
128 | }
|
---|
129 | ++n_disks;
|
---|
130 | }
|
---|
131 |
|
---|
132 | if (n_disks == 0) {
|
---|
133 | printf("No disks detected.\n");
|
---|
134 | return -1;
|
---|
135 | }
|
---|
136 |
|
---|
137 | printf(NAME ": Accepting connections\n");
|
---|
138 | async_manager();
|
---|
139 |
|
---|
140 | /* Not reached */
|
---|
141 | return 0;
|
---|
142 | }
|
---|
143 |
|
---|
144 | static int drive_identify(int disk_id, disk_t *d)
|
---|
145 | {
|
---|
146 | uint16_t data;
|
---|
147 | uint8_t status;
|
---|
148 | size_t i;
|
---|
149 |
|
---|
150 | printf("Identify drive %d\n", disk_id);
|
---|
151 | pio_write_8(&cmd->drive_head, ((disk_id != 0) ? DHR_DRV : 0));
|
---|
152 | async_usleep(100);
|
---|
153 | pio_write_8(&cmd->command, CMD_IDENTIFY_DRIVE);
|
---|
154 |
|
---|
155 | status = pio_read_8(&cmd->status);
|
---|
156 | printf("Status = 0x%x\n", status);
|
---|
157 |
|
---|
158 | d->present = false;
|
---|
159 |
|
---|
160 | /*
|
---|
161 | * Detect if drive is present. This is Qemu only! Need to
|
---|
162 | * do the right thing to work with real drives.
|
---|
163 | */
|
---|
164 | if ((status & SR_DRDY) == 0) {
|
---|
165 | printf("None attached.\n");
|
---|
166 | return ENOENT;
|
---|
167 | }
|
---|
168 |
|
---|
169 | for (i = 0; i < block_size / 2; i++) {
|
---|
170 | do {
|
---|
171 | status = pio_read_8(&cmd->status);
|
---|
172 | } while ((status & SR_DRDY) == 0);
|
---|
173 |
|
---|
174 | data = pio_read_16(&cmd->data_port);
|
---|
175 |
|
---|
176 | switch (i) {
|
---|
177 | case 1: d->cylinders = data; break;
|
---|
178 | case 3: d->heads = data; break;
|
---|
179 | case 6: d->sectors = data; break;
|
---|
180 | }
|
---|
181 | }
|
---|
182 |
|
---|
183 | d->blocks = d->cylinders * d->heads * d->sectors;
|
---|
184 |
|
---|
185 | printf("Geometry: %u cylinders, %u heads, %u sectors\n",
|
---|
186 | d->cylinders, d->heads, d->sectors);
|
---|
187 |
|
---|
188 | d->present = true;
|
---|
189 | fibril_mutex_initialize(&d->lock);
|
---|
190 |
|
---|
191 | return EOK;
|
---|
192 | }
|
---|
193 |
|
---|
194 | static int ata_bd_init(void)
|
---|
195 | {
|
---|
196 | void *vaddr;
|
---|
197 | int rc;
|
---|
198 |
|
---|
199 | rc = devmap_driver_register(NAME, ata_bd_connection);
|
---|
200 | if (rc < 0) {
|
---|
201 | printf(NAME ": Unable to register driver.\n");
|
---|
202 | return rc;
|
---|
203 | }
|
---|
204 |
|
---|
205 | rc = pio_enable((void *) cmd_physical, sizeof(ata_cmd_t), &vaddr);
|
---|
206 | if (rc != EOK) {
|
---|
207 | printf(NAME ": Could not initialize device I/O space.\n");
|
---|
208 | return rc;
|
---|
209 | }
|
---|
210 |
|
---|
211 | cmd = vaddr;
|
---|
212 |
|
---|
213 | rc = pio_enable((void *) ctl_physical, sizeof(ata_ctl_t), &vaddr);
|
---|
214 | if (rc != EOK) {
|
---|
215 | printf(NAME ": Could not initialize device I/O space.\n");
|
---|
216 | return rc;
|
---|
217 | }
|
---|
218 |
|
---|
219 | ctl = vaddr;
|
---|
220 |
|
---|
221 |
|
---|
222 | return EOK;
|
---|
223 | }
|
---|
224 |
|
---|
225 | static void ata_bd_connection(ipc_callid_t iid, ipc_call_t *icall)
|
---|
226 | {
|
---|
227 | void *fs_va = NULL;
|
---|
228 | ipc_callid_t callid;
|
---|
229 | ipc_call_t call;
|
---|
230 | ipcarg_t method;
|
---|
231 | dev_handle_t dh;
|
---|
232 | int flags;
|
---|
233 | int retval;
|
---|
234 | off_t idx;
|
---|
235 | size_t size;
|
---|
236 | int disk_id, i;
|
---|
237 |
|
---|
238 | /* Get the device handle. */
|
---|
239 | dh = IPC_GET_ARG1(*icall);
|
---|
240 |
|
---|
241 | /* Determine which disk device is the client connecting to. */
|
---|
242 | disk_id = -1;
|
---|
243 | for (i = 0; i < MAX_DISKS; i++)
|
---|
244 | if (disk[i].dev_handle == dh)
|
---|
245 | disk_id = i;
|
---|
246 |
|
---|
247 | if (disk_id < 0 || disk[disk_id].present == false) {
|
---|
248 | ipc_answer_0(iid, EINVAL);
|
---|
249 | return;
|
---|
250 | }
|
---|
251 |
|
---|
252 | /* Answer the IPC_M_CONNECT_ME_TO call. */
|
---|
253 | ipc_answer_0(iid, EOK);
|
---|
254 |
|
---|
255 | if (!ipc_share_out_receive(&callid, &comm_size, &flags)) {
|
---|
256 | ipc_answer_0(callid, EHANGUP);
|
---|
257 | return;
|
---|
258 | }
|
---|
259 |
|
---|
260 | fs_va = as_get_mappable_page(comm_size);
|
---|
261 | if (fs_va == NULL) {
|
---|
262 | ipc_answer_0(callid, EHANGUP);
|
---|
263 | return;
|
---|
264 | }
|
---|
265 |
|
---|
266 | (void) ipc_share_out_finalize(callid, fs_va);
|
---|
267 |
|
---|
268 | while (1) {
|
---|
269 | callid = async_get_call(&call);
|
---|
270 | method = IPC_GET_METHOD(call);
|
---|
271 | switch (method) {
|
---|
272 | case IPC_M_PHONE_HUNGUP:
|
---|
273 | /* The other side has hung up. */
|
---|
274 | ipc_answer_0(callid, EOK);
|
---|
275 | return;
|
---|
276 | case BD_READ_BLOCK:
|
---|
277 | case BD_WRITE_BLOCK:
|
---|
278 | idx = IPC_GET_ARG1(call);
|
---|
279 | size = IPC_GET_ARG2(call);
|
---|
280 | if (size > comm_size) {
|
---|
281 | retval = EINVAL;
|
---|
282 | break;
|
---|
283 | }
|
---|
284 | retval = ata_bd_rdwr(disk_id, method, idx,
|
---|
285 | size, fs_va);
|
---|
286 | break;
|
---|
287 | default:
|
---|
288 | retval = EINVAL;
|
---|
289 | break;
|
---|
290 | }
|
---|
291 | ipc_answer_0(callid, retval);
|
---|
292 | }
|
---|
293 | }
|
---|
294 |
|
---|
295 | static int ata_bd_rdwr(int disk_id, ipcarg_t method, off_t blk_idx, size_t size,
|
---|
296 | void *buf)
|
---|
297 | {
|
---|
298 | int rc;
|
---|
299 | size_t now;
|
---|
300 |
|
---|
301 | while (size > 0) {
|
---|
302 | now = size < block_size ? size : block_size;
|
---|
303 | if (now != block_size)
|
---|
304 | return EINVAL;
|
---|
305 |
|
---|
306 | if (method == BD_READ_BLOCK)
|
---|
307 | rc = ata_bd_read_block(disk_id, blk_idx, 1, buf);
|
---|
308 | else
|
---|
309 | rc = ata_bd_write_block(disk_id, blk_idx, 1, buf);
|
---|
310 |
|
---|
311 | if (rc != EOK)
|
---|
312 | return rc;
|
---|
313 |
|
---|
314 | buf += block_size;
|
---|
315 | blk_idx++;
|
---|
316 |
|
---|
317 | if (size > block_size)
|
---|
318 | size -= block_size;
|
---|
319 | else
|
---|
320 | size = 0;
|
---|
321 | }
|
---|
322 |
|
---|
323 | return EOK;
|
---|
324 | }
|
---|
325 |
|
---|
326 |
|
---|
327 | static int ata_bd_read_block(int disk_id, uint64_t blk_idx, size_t blk_cnt,
|
---|
328 | void *buf)
|
---|
329 | {
|
---|
330 | size_t i;
|
---|
331 | uint16_t data;
|
---|
332 | uint8_t status;
|
---|
333 | uint64_t c, h, s;
|
---|
334 | uint64_t idx;
|
---|
335 | uint8_t drv_head;
|
---|
336 | disk_t *d;
|
---|
337 |
|
---|
338 | d = &disk[disk_id];
|
---|
339 |
|
---|
340 | /* Check device bounds. */
|
---|
341 | if (blk_idx >= d->blocks)
|
---|
342 | return EINVAL;
|
---|
343 |
|
---|
344 | /* Compute CHS. */
|
---|
345 | c = blk_idx / (d->heads * d->sectors);
|
---|
346 | idx = blk_idx % (d->heads * d->sectors);
|
---|
347 |
|
---|
348 | h = idx / d->sectors;
|
---|
349 | s = 1 + (idx % d->sectors);
|
---|
350 |
|
---|
351 | /* New value for Drive/Head register */
|
---|
352 | drv_head =
|
---|
353 | ((disk_id != 0) ? DHR_DRV : 0) |
|
---|
354 | (h & 0x0f);
|
---|
355 |
|
---|
356 | fibril_mutex_lock(&d->lock);
|
---|
357 |
|
---|
358 | /* Program a Read Sectors operation. */
|
---|
359 |
|
---|
360 | pio_write_8(&cmd->drive_head, drv_head);
|
---|
361 | pio_write_8(&cmd->sector_count, 1);
|
---|
362 | pio_write_8(&cmd->sector_number, s);
|
---|
363 | pio_write_8(&cmd->cylinder_low, c & 0xff);
|
---|
364 | pio_write_8(&cmd->cylinder_high, c >> 16);
|
---|
365 | pio_write_8(&cmd->command, CMD_READ_SECTORS);
|
---|
366 |
|
---|
367 | /* Read data from the disk buffer. */
|
---|
368 |
|
---|
369 | for (i = 0; i < block_size / 2; i++) {
|
---|
370 | do {
|
---|
371 | status = pio_read_8(&cmd->status);
|
---|
372 | } while ((status & SR_DRDY) == 0);
|
---|
373 |
|
---|
374 | data = pio_read_16(&cmd->data_port);
|
---|
375 | ((uint16_t *) buf)[i] = data;
|
---|
376 | }
|
---|
377 |
|
---|
378 | fibril_mutex_unlock(&d->lock);
|
---|
379 | return EOK;
|
---|
380 | }
|
---|
381 |
|
---|
382 | static int ata_bd_write_block(int disk_id, uint64_t blk_idx, size_t blk_cnt,
|
---|
383 | const void *buf)
|
---|
384 | {
|
---|
385 | size_t i;
|
---|
386 | uint8_t status;
|
---|
387 | uint64_t c, h, s;
|
---|
388 | uint64_t idx;
|
---|
389 | uint8_t drv_head;
|
---|
390 | disk_t *d;
|
---|
391 |
|
---|
392 | d = &disk[disk_id];
|
---|
393 |
|
---|
394 | /* Check device bounds. */
|
---|
395 | if (blk_idx >= d->blocks)
|
---|
396 | return EINVAL;
|
---|
397 |
|
---|
398 | /* Compute CHS. */
|
---|
399 | c = blk_idx / (d->heads * d->sectors);
|
---|
400 | idx = blk_idx % (d->heads * d->sectors);
|
---|
401 |
|
---|
402 | h = idx / d->sectors;
|
---|
403 | s = 1 + (idx % d->sectors);
|
---|
404 |
|
---|
405 | /* New value for Drive/Head register */
|
---|
406 | drv_head =
|
---|
407 | ((disk_id != 0) ? DHR_DRV : 0) |
|
---|
408 | (h & 0x0f);
|
---|
409 |
|
---|
410 | fibril_mutex_lock(&d->lock);
|
---|
411 |
|
---|
412 | /* Program a Read Sectors operation. */
|
---|
413 |
|
---|
414 | pio_write_8(&cmd->drive_head, drv_head);
|
---|
415 | pio_write_8(&cmd->sector_count, 1);
|
---|
416 | pio_write_8(&cmd->sector_number, s);
|
---|
417 | pio_write_8(&cmd->cylinder_low, c & 0xff);
|
---|
418 | pio_write_8(&cmd->cylinder_high, c >> 16);
|
---|
419 | pio_write_8(&cmd->command, CMD_WRITE_SECTORS);
|
---|
420 |
|
---|
421 | /* Write data to the disk buffer. */
|
---|
422 |
|
---|
423 | for (i = 0; i < block_size / 2; i++) {
|
---|
424 | do {
|
---|
425 | status = pio_read_8(&cmd->status);
|
---|
426 | } while ((status & SR_DRDY) == 0);
|
---|
427 |
|
---|
428 | pio_write_16(&cmd->data_port, ((uint16_t *) buf)[i]);
|
---|
429 | }
|
---|
430 |
|
---|
431 | fibril_mutex_unlock(&d->lock);
|
---|
432 | return EOK;
|
---|
433 | }
|
---|
434 |
|
---|
435 |
|
---|
436 | /**
|
---|
437 | * @}
|
---|
438 | */
|
---|