cmd_sata.c

来自「U-boot源码 ARM7启动代码」· C语言 代码 · 共 713 行 · 第 1/2 页

	return status;
}

u8
sata_chk_status (struct sata_ioports * ioaddr)
{
	return sata_inb (ioaddr->status_addr);
}

void
msleep (int count)
{
	int i;

	for (i = 0; i < count; i++)
		udelay (1000);
}

ulong
sata_read (int device, ulong blknr,lbaint_t blkcnt, void * buff)
{
	ulong n = 0, *buffer = (ulong *)buff;
	u8 dev = 0, num = 0, mask = 0, status = 0;

#ifdef CONFIG_LBA48
	unsigned char lba48 = 0;

	if (blknr & 0x0000fffff0000000) {
		if (!sata_dev_desc[devno].lba48) {
			printf ("Drive doesn't support 48-bit addressing\n");
			return 0;
		}
		/* more than 28 bits used, use 48bit mode */
		lba48 = 1;
	}
#endif
	/*Port Number */
	num = device / CFG_SATA_DEVS_PER_BUS;
	/*dev on the port */
	if (device >= CFG_SATA_DEVS_PER_BUS)
		dev = device - CFG_SATA_DEVS_PER_BUS;
	else
		dev = device;

	if (dev == 0)
		mask = 0x01;
	else
		mask = 0x02;

	if (!(port[num].dev_mask & mask)) {
		printf ("dev%d is not present on port#%d\n", dev, num);
		return 0;
	}

	/* Select device */
	dev_select (&port[num].ioaddr, dev);

	status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 500);
	if (status & ATA_BUSY) {
		printf ("ata%u failed to respond\n", port[num].port_no);
		return n;
	}
	while (blkcnt-- > 0) {
		status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 500);
		if (status & ATA_BUSY) {
			printf ("ata%u failed to respond\n", 0);
			return n;
		}
#ifdef CONFIG_LBA48
		if (lba48) {
			/* write high bits */
			sata_outb (0, port[num].ioaddr.nsect_addr);
			sata_outb ((blknr >> 24) & 0xFF,
				   port[num].ioaddr.lbal_addr);
			sata_outb ((blknr >> 32) & 0xFF,
				   port[num].ioaddr.lbam_addr);
			sata_outb ((blknr >> 40) & 0xFF,
				   port[num].ioaddr.lbah_addr);
		}
#endif
		sata_outb (1, port[num].ioaddr.nsect_addr);
		sata_outb (((blknr) >> 0) & 0xFF,
			   port[num].ioaddr.lbal_addr);
		sata_outb ((blknr >> 8) & 0xFF, port[num].ioaddr.lbam_addr);
		sata_outb ((blknr >> 16) & 0xFF, port[num].ioaddr.lbah_addr);

#ifdef CONFIG_LBA48
		if (lba48) {
			sata_outb (ATA_LBA, port[num].ioaddr.device_addr);
			sata_outb (ATA_CMD_READ_EXT,
				   port[num].ioaddr.command_addr);
		} else
#endif
		{
			sata_outb (ATA_LBA | ((blknr >> 24) & 0xF),
				   port[num].ioaddr.device_addr);
			sata_outb (ATA_CMD_READ,
				   port[num].ioaddr.command_addr);
		}

		msleep (50);
		/*may take up to 4 sec */
		status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 4000);

		if ((status & (ATA_STAT_DRQ | ATA_STAT_BUSY | ATA_STAT_ERR))
		    != ATA_STAT_DRQ) {
			u8 err = 0;

			printf ("Error no DRQ dev %d blk %ld: sts 0x%02x\n",
				device, (ulong) blknr, status);
			err = sata_inb (port[num].ioaddr.error_addr);
			printf ("Error reg = 0x%x\n", err);
			return (n);
		}
		input_data (&port[num].ioaddr, buffer, ATA_SECTORWORDS);
		sata_inb (port[num].ioaddr.altstatus_addr);
		udelay (50);

		++n;
		++blknr;
		buffer += ATA_SECTORWORDS;
	}
	return n;
}

ulong
sata_write (int device, ulong blknr,lbaint_t blkcnt, void * buff)
{
	ulong n = 0, *buffer = (ulong *)buff;
	unsigned char status = 0, num = 0, dev = 0, mask = 0;

#ifdef CONFIG_LBA48
	unsigned char lba48 = 0;

	if (blknr & 0x0000fffff0000000) {
		if (!sata_dev_desc[devno].lba48) {
			printf ("Drive doesn't support 48-bit addressing\n");
			return 0;
		}
		/* more than 28 bits used, use 48bit mode */
		lba48 = 1;
	}
#endif
	/*Port Number */
	num = device / CFG_SATA_DEVS_PER_BUS;
	/*dev on the Port */
	if (device >= CFG_SATA_DEVS_PER_BUS)
		dev = device - CFG_SATA_DEVS_PER_BUS;
	else
		dev = device;

	if (dev == 0)
		mask = 0x01;
	else
		mask = 0x02;

	/* Select device */
	dev_select (&port[num].ioaddr, dev);

	status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 500);
	if (status & ATA_BUSY) {
		printf ("ata%u failed to respond\n", port[num].port_no);
		return n;
	}

	while (blkcnt-- > 0) {
		status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 500);
		if (status & ATA_BUSY) {
			printf ("ata%u failed to respond\n",
				port[num].port_no);
			return n;
		}
#ifdef CONFIG_LBA48
		if (lba48) {
			/* write high bits */
			sata_outb (0, port[num].ioaddr.nsect_addr);
			sata_outb ((blknr >> 24) & 0xFF,
				   port[num].ioaddr.lbal_addr);
			sata_outb ((blknr >> 32) & 0xFF,
				   port[num].ioaddr.lbam_addr);
			sata_outb ((blknr >> 40) & 0xFF,
				   port[num].ioaddr.lbah_addr);
		}
#endif
		sata_outb (1, port[num].ioaddr.nsect_addr);
		sata_outb ((blknr >> 0) & 0xFF, port[num].ioaddr.lbal_addr);
		sata_outb ((blknr >> 8) & 0xFF, port[num].ioaddr.lbam_addr);
		sata_outb ((blknr >> 16) & 0xFF, port[num].ioaddr.lbah_addr);
#ifdef CONFIG_LBA48
		if (lba48) {
			sata_outb (ATA_LBA, port[num].ioaddr.device_addr);
			sata_outb (ATA_CMD_WRITE_EXT,
				   port[num].ioaddr.command_addr);
		} else
#endif
		{
			sata_outb (ATA_LBA | ((blknr >> 24) & 0xF),
				   port[num].ioaddr.device_addr);
			sata_outb (ATA_CMD_WRITE,
				   port[num].ioaddr.command_addr);
		}

		msleep (50);
		/*may take up to 4 sec */
		status = sata_busy_wait (&port[num].ioaddr, ATA_BUSY, 4000);
		if ((status & (ATA_STAT_DRQ | ATA_STAT_BUSY | ATA_STAT_ERR))
		    != ATA_STAT_DRQ) {
			printf ("Error no DRQ dev %d blk %ld: sts 0x%02x\n",
				device, (ulong) blknr, status);
			return (n);
		}

		output_data (&port[num].ioaddr, buffer, ATA_SECTORWORDS);
		sata_inb (port[num].ioaddr.altstatus_addr);
		udelay (50);

		++n;
		++blknr;
		buffer += ATA_SECTORWORDS;
	}
	return n;
}

block_dev_desc_t *sata_get_dev (int dev);

block_dev_desc_t *
sata_get_dev (int dev)
{
	return ((block_dev_desc_t *) & sata_dev_desc[dev]);
}

int
do_sata (cmd_tbl_t * cmdtp, int flag, int argc, char *argv[])
{

	switch (argc) {
	case 0:
	case 1:
		printf ("Usage:\n%s\n", cmdtp->usage);
		return 1;
	case 2:
		if (strncmp (argv[1], "init", 4) == 0) {
			int rcode = 0;

			rcode = init_sata ();
			if (rcode)
				printf ("Sata initialization Failed\n");
			return rcode;
		} else if (strncmp (argv[1], "inf", 3) == 0) {
			int i;

			putc ('\n');
			for (i = 0; i < CFG_SATA_MAXDEVICES; ++i) {
				/*List only known devices */
				if (sata_dev_desc[i].type ==
				    DEV_TYPE_UNKNOWN)
					continue;
				printf ("sata dev %d: ", i);
				dev_print (&sata_dev_desc[i]);
			}
			return 0;
		}
		printf ("Usage:\n%s\n", cmdtp->usage);
		return 1;
	case 3:
		if (strcmp (argv[1], "dev") == 0) {
			int dev = (int) simple_strtoul (argv[2], NULL, 10);

			if (dev >= CFG_SATA_MAXDEVICES) {
				printf ("\nSata dev %d not available\n",
					dev);
				return 1;
			}
			printf ("\nSATA dev %d: ", dev);
			dev_print (&sata_dev_desc[dev]);
			if (sata_dev_desc[dev].type == DEV_TYPE_UNKNOWN)
				return 1;
			curr_dev = dev;
			return 0;
		} else if (strcmp (argv[1], "part") == 0) {
			int dev = (int) simple_strtoul (argv[2], NULL, 10);

			if (dev >= CFG_SATA_MAXDEVICES) {
				printf ("\nSata dev %d not available\n",
					dev);
				return 1;
			}
			PRINTF ("\nSATA dev %d: ", dev);
			if (sata_dev_desc[dev].part_type !=
			    PART_TYPE_UNKNOWN) {
				print_part (&sata_dev_desc[dev]);
			} else {
				printf ("\nSata dev %d partition type "
					"unknown\n", dev);
				return 1;
			}
			return 0;
		}
		printf ("Usage:\n%s\n", cmdtp->usage);
		return 1;
	default:
		if (argc < 5) {
			printf ("Usage:\n%s\n", cmdtp->usage);
			return 1;
		}
		if (strcmp (argv[1], "read") == 0) {
			ulong addr = simple_strtoul (argv[2], NULL, 16);
			ulong cnt = simple_strtoul (argv[4], NULL, 16);
			ulong n;
			lbaint_t blk = simple_strtoul (argv[3], NULL, 16);

			memset ((int *) addr, 0, cnt * 512);
			printf ("\nSATA read: dev %d blk # %ld,"
				"count %ld ... ", curr_dev, blk, cnt);
			n = sata_read (curr_dev, blk, cnt, (ulong *) addr);
			/* flush cache after read */
			flush_cache (addr, cnt * 512);
			printf ("%ld blocks read: %s\n", n,
				(n == cnt) ? "OK" : "ERR");
			if (n == cnt)
				return 1;
			else
				return 0;
		} else if (strcmp (argv[1], "write") == 0) {
			ulong addr = simple_strtoul (argv[2], NULL, 16);
			ulong cnt = simple_strtoul (argv[4], NULL, 16);
			ulong n;
			lbaint_t blk = simple_strtoul (argv[3], NULL, 16);

			printf ("\nSata write: dev %d blk # %ld,"
				"count %ld ... ", curr_dev, blk, cnt);
			n = sata_write (curr_dev, blk, cnt, (ulong *) addr);
			printf ("%ld blocks written: %s\n", n,
				(n == cnt) ? "OK" : "ERR");
			if (n == cnt)
				return 1;
			else
				return 0;
		} else {
			printf ("Usage:\n%s\n", cmdtp->usage);
			return 1;
		}
	}			/*End OF SWITCH */
}

U_BOOT_CMD (sata, 5, 1, do_sata,
	    "sata init\n"
	    "sata info\n"
	    "sata part device\n"
	    "sata dev device\n"
	    "sata read  addr blk# cnt\n"
	    "sata write  addr blk# cnt\n", "cmd for init,rw and dev-info\n");

#endif