mirror of
https://github.com/aaru-dps/aaruremote.git
synced 2025-12-16 19:24:37 +00:00
Implement ATA packets.
This commit is contained in:
@@ -3,7 +3,7 @@ project(dicremote C)
|
||||
|
||||
set(CMAKE_C_STANDARD 90)
|
||||
|
||||
set(MAIN_SOURCES main.c list_devices.c device.c scsi.c hex2bin.c usb.c ieee1394.c pcmcia.c)
|
||||
set(MAIN_SOURCES main.c list_devices.c device.c scsi.c hex2bin.c usb.c ieee1394.c pcmcia.c ata.c)
|
||||
|
||||
if("${CMAKE_SYSTEM}" MATCHES "Linux")
|
||||
set(PLATFORM_SOURCES linux/list_devices.c linux/linux.h linux/device.c linux/scsi.c linux/usb.c linux/ieee1394.c linux/pcmcia.c)
|
||||
|
||||
62
ata.c
Normal file
62
ata.c
Normal file
@@ -0,0 +1,62 @@
|
||||
/*
|
||||
* This file is part of the DiscImageChef Remote Server.
|
||||
* Copyright (c) 2019 Natalia Portillo.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, version 3.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but
|
||||
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
* General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "dicmote.h"
|
||||
|
||||
#include <stddef.h>
|
||||
|
||||
int32_t SendAtaChsCommand(int device_fd,
|
||||
AtaRegistersChs registers,
|
||||
AtaErrorRegistersChs* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
int32_t SendAtaLba28Command(int device_fd,
|
||||
AtaRegistersLba28 registers,
|
||||
AtaErrorRegistersLba28* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
int32_t SendAtaLba48Command(int device_fd,
|
||||
AtaRegistersLba48 registers,
|
||||
AtaErrorRegistersLba48* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
46
dicmote.h
46
dicmote.h
@@ -70,6 +70,22 @@
|
||||
#define DICMOTE_SCSI_DIRECTION_OUT 1
|
||||
#define DICMOTE_SCSI_DIRECTION_IN 2
|
||||
#define DICMOTE_SCSI_DIRECTION_INOUT 3
|
||||
#define DICMOTE_ATA_PROTOCOL_HARD_RESET 0
|
||||
#define DICMOTE_ATA_PROTOCOL_SOFT_RESET 1
|
||||
#define DICMOTE_ATA_PROTOCOL_NO_DATA 3
|
||||
#define DICMOTE_ATA_PROTOCOL_PIO_IN 4
|
||||
#define DICMOTE_ATA_PROTOCOL_PIO_OUT 5
|
||||
#define DICMOTE_ATA_PROTOCOL_DMA 6
|
||||
#define DICMOTE_ATA_PROTOCOL_DMA_QUEUED 7
|
||||
#define DICMOTE_ATA_PROTOCOL_DEVICE_DIAGNOSTIC 8
|
||||
#define DICMOTE_ATA_PROTOCOL_DEVICE_RESET 9
|
||||
#define DICMOTE_ATA_PROTOCOL_UDMA_IN 10
|
||||
#define DICMOTE_ATA_PROTOCOL_UDMA_OUT 11
|
||||
#define DICMOTE_ATA_PROTOCOL_FPDMA 12
|
||||
#define DICMOTE_ATA_TRANSFER_REGISTER_NONE 0
|
||||
#define DICMOTE_ATA_TRANSFER_REGISTER_FEATURE 1
|
||||
#define DICMOTE_ATA_TRANSFER_REGISTER_SECTOR_COUNT 2
|
||||
#define DICMOTE_ATA_TRANSFER_REGISTER_SPTSIU 3
|
||||
|
||||
#pragma pack(push, 1)
|
||||
|
||||
@@ -433,5 +449,35 @@ uint8_t GetFireWireData(const char* devicePath,
|
||||
char* vendor,
|
||||
char* model);
|
||||
uint8_t GetPcmciaData(const char* devicePath, uint16_t* cisLen, char* cis);
|
||||
int32_t SendAtaChsCommand(int device_fd,
|
||||
AtaRegistersChs registers,
|
||||
AtaErrorRegistersChs* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense);
|
||||
int32_t SendAtaLba28Command(int device_fd,
|
||||
AtaRegistersLba28 registers,
|
||||
AtaErrorRegistersLba28* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense);
|
||||
int32_t SendAtaLba48Command(int device_fd,
|
||||
AtaRegistersLba48 registers,
|
||||
AtaErrorRegistersLba48* errorRegisters,
|
||||
uint8_t protocol,
|
||||
uint8_t transferRegister,
|
||||
char* buffer,
|
||||
uint32_t timeout,
|
||||
uint8_t transferBlocks,
|
||||
uint32_t* duration,
|
||||
uint32_t* sense);
|
||||
|
||||
#endif
|
||||
|
||||
207
main.c
207
main.c
@@ -68,6 +68,15 @@ int main()
|
||||
DicPacketResGetUsbData* pkt_res_usb;
|
||||
DicPacketResGetFireWireData* pkt_res_firewire;
|
||||
DicPacketResGetPcmciaData* pkt_res_pcmcia;
|
||||
DicPacketCmdAtaChs* pkt_cmd_ata_chs;
|
||||
DicPacketCmdAtaLba28* pkt_cmd_ata_lba28;
|
||||
DicPacketCmdAtaLba48* pkt_cmd_ata_lba48;
|
||||
DicPacketResAtaChs* pkt_res_ata_chs;
|
||||
DicPacketResAtaLba28* pkt_res_ata_lba28;
|
||||
DicPacketResAtaLba48* pkt_res_ata_lba48;
|
||||
AtaErrorRegistersChs ata_chs_error_regs;
|
||||
AtaErrorRegistersLba28 ata_lba28_error_regs;
|
||||
AtaErrorRegistersLba48 ata_lba48_error_regs;
|
||||
|
||||
printf("DiscImageChef Remote Server %s\n", DICMOTE_VERSION);
|
||||
printf("Copyright (C) 2019 Natalia Portillo\n");
|
||||
@@ -737,8 +746,206 @@ int main()
|
||||
free(pkt_res_pcmcia);
|
||||
continue;
|
||||
case DICMOTE_PACKET_TYPE_COMMAND_ATA_CHS:
|
||||
// Packet contains data after
|
||||
in_buf = malloc(pkt_hdr->len);
|
||||
|
||||
if(!in_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, closing connection...\n", errno);
|
||||
free(pkt_hdr);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
recv(cli_sock, in_buf, pkt_hdr->len, 0);
|
||||
|
||||
pkt_cmd_ata_chs = (DicPacketCmdAtaChs*)in_buf;
|
||||
|
||||
// TODO: Check size of buffers + size of packet is not bigger than size in header
|
||||
|
||||
if(pkt_cmd_ata_chs->buf_len > 0)
|
||||
buffer = in_buf + sizeof(DicPacketCmdAtaChs);
|
||||
else
|
||||
buffer = NULL;
|
||||
|
||||
memset(&ata_chs_error_regs, 0, sizeof(AtaErrorRegistersChs));
|
||||
|
||||
duration = 0;
|
||||
sense = 1;
|
||||
ret = SendAtaChsCommand(device_fd,
|
||||
pkt_cmd_ata_chs->registers,
|
||||
&ata_chs_error_regs,
|
||||
pkt_cmd_ata_chs->protocol,
|
||||
pkt_cmd_ata_chs->transferRegister,
|
||||
buffer,
|
||||
pkt_cmd_ata_chs->timeout,
|
||||
pkt_cmd_ata_chs->transferBlocks,
|
||||
&duration,
|
||||
&sense);
|
||||
|
||||
out_buf = malloc(sizeof(DicPacketResAtaChs) + pkt_cmd_ata_chs->buf_len);
|
||||
|
||||
if(!out_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, continuing...\n", errno);
|
||||
free(pkt_hdr);
|
||||
free(in_buf);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
pkt_res_ata_chs = (DicPacketResAtaChs*)out_buf;
|
||||
if(buffer) memcpy(out_buf + sizeof(DicPacketResAtaChs), buffer, pkt_cmd_ata_chs->buf_len);
|
||||
|
||||
pkt_res_ata_chs->hdr.len = sizeof(DicPacketResAtaChs) + pkt_cmd_ata_chs->buf_len;
|
||||
pkt_res_ata_chs->hdr.packet_type = DICMOTE_PACKET_TYPE_RESPONSE_ATA_CHS;
|
||||
pkt_res_ata_chs->hdr.version = DICMOTE_PACKET_VERSION;
|
||||
pkt_res_ata_chs->hdr.id = DICMOTE_PACKET_ID;
|
||||
|
||||
pkt_res_ata_chs->registers = ata_chs_error_regs;
|
||||
pkt_res_ata_chs->buf_len = pkt_cmd_ata_chs->buf_len;
|
||||
pkt_res_ata_chs->duration = duration;
|
||||
pkt_res_ata_chs->sense = sense;
|
||||
pkt_res_ata_chs->error_no = ret;
|
||||
|
||||
write(cli_sock, pkt_res_ata_chs, pkt_res_ata_chs->hdr.len);
|
||||
free(pkt_cmd_ata_chs);
|
||||
free(pkt_res_ata_chs);
|
||||
continue;
|
||||
case DICMOTE_PACKET_TYPE_COMMAND_ATA_LBA28:
|
||||
// Packet contains data after
|
||||
in_buf = malloc(pkt_hdr->len);
|
||||
|
||||
if(!in_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, closing connection...\n", errno);
|
||||
free(pkt_hdr);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
recv(cli_sock, in_buf, pkt_hdr->len, 0);
|
||||
|
||||
pkt_cmd_ata_lba28 = (DicPacketCmdAtaLba28*)in_buf;
|
||||
|
||||
// TODO: Check size of buffers + size of packet is not bigger than size in header
|
||||
|
||||
if(pkt_cmd_ata_lba28->buf_len > 0)
|
||||
buffer = in_buf + sizeof(DicPacketCmdAtaLba28);
|
||||
else
|
||||
buffer = NULL;
|
||||
|
||||
memset(&ata_lba28_error_regs, 0, sizeof(AtaErrorRegistersLba28));
|
||||
|
||||
duration = 0;
|
||||
sense = 1;
|
||||
ret = SendAtaLba28Command(device_fd,
|
||||
pkt_cmd_ata_lba28->registers,
|
||||
&ata_lba28_error_regs,
|
||||
pkt_cmd_ata_lba28->protocol,
|
||||
pkt_cmd_ata_lba28->transferRegister,
|
||||
buffer,
|
||||
pkt_cmd_ata_lba28->timeout,
|
||||
pkt_cmd_ata_lba28->transferBlocks,
|
||||
&duration,
|
||||
&sense);
|
||||
|
||||
out_buf = malloc(sizeof(DicPacketResAtaLba28) + pkt_cmd_ata_lba28->buf_len);
|
||||
|
||||
if(!out_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, continuing...\n", errno);
|
||||
free(pkt_hdr);
|
||||
free(in_buf);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
pkt_res_ata_lba28 = (DicPacketResAtaLba28*)out_buf;
|
||||
if(buffer) memcpy(out_buf + sizeof(DicPacketResAtaLba28), buffer, pkt_cmd_ata_lba28->buf_len);
|
||||
|
||||
pkt_res_ata_lba28->hdr.len = sizeof(DicPacketResAtaLba28) + pkt_cmd_ata_lba28->buf_len;
|
||||
pkt_res_ata_lba28->hdr.packet_type = DICMOTE_PACKET_TYPE_RESPONSE_ATA_LBA28;
|
||||
pkt_res_ata_lba28->hdr.version = DICMOTE_PACKET_VERSION;
|
||||
pkt_res_ata_lba28->hdr.id = DICMOTE_PACKET_ID;
|
||||
|
||||
pkt_res_ata_lba28->registers = ata_lba28_error_regs;
|
||||
pkt_res_ata_lba28->buf_len = pkt_cmd_ata_lba28->buf_len;
|
||||
pkt_res_ata_lba28->duration = duration;
|
||||
pkt_res_ata_lba28->sense = sense;
|
||||
pkt_res_ata_lba28->error_no = ret;
|
||||
|
||||
write(cli_sock, pkt_res_ata_lba28, pkt_res_ata_lba28->hdr.len);
|
||||
free(pkt_cmd_ata_lba28);
|
||||
free(pkt_res_ata_lba28);
|
||||
continue;
|
||||
case DICMOTE_PACKET_TYPE_COMMAND_ATA_LBA48:
|
||||
// Packet contains data after
|
||||
in_buf = malloc(pkt_hdr->len);
|
||||
|
||||
if(!in_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, closing connection...\n", errno);
|
||||
free(pkt_hdr);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
recv(cli_sock, in_buf, pkt_hdr->len, 0);
|
||||
|
||||
pkt_cmd_ata_lba48 = (DicPacketCmdAtaLba48*)in_buf;
|
||||
|
||||
// TODO: Check size of buffers + size of packet is not bigger than size in header
|
||||
|
||||
if(pkt_cmd_ata_lba48->buf_len > 0)
|
||||
buffer = in_buf + sizeof(DicPacketCmdAtaLba48);
|
||||
else
|
||||
buffer = NULL;
|
||||
|
||||
memset(&ata_lba48_error_regs, 0, sizeof(AtaErrorRegistersLba48));
|
||||
|
||||
duration = 0;
|
||||
sense = 1;
|
||||
ret = SendAtaLba48Command(device_fd,
|
||||
pkt_cmd_ata_lba48->registers,
|
||||
&ata_lba48_error_regs,
|
||||
pkt_cmd_ata_lba48->protocol,
|
||||
pkt_cmd_ata_lba48->transferRegister,
|
||||
buffer,
|
||||
pkt_cmd_ata_lba48->timeout,
|
||||
pkt_cmd_ata_lba48->transferBlocks,
|
||||
&duration,
|
||||
&sense);
|
||||
|
||||
out_buf = malloc(sizeof(DicPacketResAtaLba48) + pkt_cmd_ata_lba48->buf_len);
|
||||
|
||||
if(!out_buf)
|
||||
{
|
||||
printf("Fatal error %d allocating memory for packet, continuing...\n", errno);
|
||||
free(pkt_hdr);
|
||||
free(in_buf);
|
||||
close(cli_sock);
|
||||
continue;
|
||||
}
|
||||
|
||||
pkt_res_ata_lba48 = (DicPacketResAtaLba48*)out_buf;
|
||||
if(buffer) memcpy(out_buf + sizeof(DicPacketResAtaLba48), buffer, pkt_cmd_ata_lba48->buf_len);
|
||||
|
||||
pkt_res_ata_lba48->hdr.len = sizeof(DicPacketResAtaLba48) + pkt_cmd_ata_lba48->buf_len;
|
||||
pkt_res_ata_lba48->hdr.packet_type = DICMOTE_PACKET_TYPE_RESPONSE_ATA_LBA48;
|
||||
pkt_res_ata_lba48->hdr.version = DICMOTE_PACKET_VERSION;
|
||||
pkt_res_ata_lba48->hdr.id = DICMOTE_PACKET_ID;
|
||||
|
||||
pkt_res_ata_lba48->registers = ata_lba48_error_regs;
|
||||
pkt_res_ata_lba48->buf_len = pkt_cmd_ata_lba48->buf_len;
|
||||
pkt_res_ata_lba48->duration = duration;
|
||||
pkt_res_ata_lba48->sense = sense;
|
||||
pkt_res_ata_lba48->error_no = ret;
|
||||
|
||||
write(cli_sock, pkt_res_ata_lba48, pkt_res_ata_lba48->hdr.len);
|
||||
free(pkt_cmd_ata_lba48);
|
||||
free(pkt_res_ata_lba48);
|
||||
continue;
|
||||
case DICMOTE_PACKET_TYPE_COMMAND_SDHCI:
|
||||
pkt_nop->reason_code = DICMOTE_PACKET_NOP_REASON_NOT_IMPLEMENTED;
|
||||
memset(&pkt_nop->reason, 0, 256);
|
||||
|
||||
Reference in New Issue
Block a user