1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#ifndef ATA_DRIVER_H
#define ATA_DRIVER_H
#include <kernel/fs/vfs.h>
#include <kernel/memory/paging.h>
#include <kernel/interrupts/interrupts.h>
typedef struct prdt {
uint32_t buffer_phys;
uint16_t transfer_size;
uint16_t mark_end;
} __attribute__((packed)) prdt_t;
typedef struct ata_dev {
uint16_t data;
uint16_t error;
uint16_t sector_count;
union {
uint16_t sector_num;
uint16_t lba_lo;
};
union {
uint16_t cylinder_low;
uint16_t lba_mid;
};
union {
uint16_t cylinder_high;
uint16_t lba_high;
};
union {
uint16_t drive;
uint16_t head;
};
union {
uint16_t command;
uint16_t status;
};
union {
uint16_t control;
uint16_t alt_status;
};
int slave;
uint32_t bar4;
uint32_t BMR_COMMAND;
uint32_t BMR_prdt;
uint32_t BMR_STATUS;
prdt_t *prdt;
uint8_t *prdt_phys;
uint8_t *mem_buffer;
uint8_t *mem_buffer_phys;
char mountpoint[32];
} __attribute__((packed)) ata_dev_t;
// ATA PCI info of QEMU // TODO: Remove Qemu based device id
#define ATA_VENDOR_ID 0x8086
#define ATA_DEVICE_ID 0x7010
// Control reg
#define CONTROL_STOP_INTERRUPT 0x2
#define CONTROL_SOFTWARE_RESET 0x4
#define CONTROL_HIGH_ORDER_BYTE 0x80
#define CONTROL_ZERO 0x00
// Command reg
#define COMMAND_IDENTIFY 0xEC
#define COMMAND_DMA_READ 0xC8
#define ATA_CMD_READ_PIO 0x20
// Status reg
#define STATUS_ERR 0x0
#define STATUS_DRQ 0x8
#define STATUS_SRV 0x10
#define STATUS_DF 0x20
#define STATUS_RDY 0x40
#define STATUS_BSY 0x80
// Bus Master Reg Command
#define BMR_COMMAND_DMA_START 0x1
#define BMR_COMMAND_DMA_STOP 0x0
#define BMR_COMMAND_READ 0x8
#define BMR_STATUS_INT 0x4
#define BMR_STATUS_ERR 0x2
// Prdt
#define SECTOR_SIZE 512
#define MARK_END 0x8000
void io_wait(ata_dev_t *dev);
void software_reset(ata_dev_t *dev);
void ata_handler(struct regs *reg);
void ata_open(vfs_node_t *node, uint32_t flags);
void ata_close(vfs_node_t *node);
uint32_t ata_read(vfs_node_t *node, uint32_t offset, uint32_t size, char *buf);
uint32_t ata_write(vfs_node_t *node, uint32_t offset, uint32_t size, char *buf);
void ata_write_sector(ata_dev_t *dev, uint32_t lba, char *buf);
char *ata_read_sector(ata_dev_t *dev, uint32_t lba);
vfs_node_t *create_ata_device(ata_dev_t *dev);
void ata_device_init(ata_dev_t *dev, int primary);
void ata_device_detect(ata_dev_t *dev, int primary);
void ata_init();
#endif
|
