READ CAPACITY (16) turns out to be an optional command (even though
[people/andreif/gpxe.git] / src / drivers / block / scsi.c
index 7ddc4a1..1e906e1 100644 (file)
@@ -50,11 +50,12 @@ static int scsi_command ( struct scsi_device *scsi,
  *
  * @v blockdev         Block device
  * @v block            LBA block number
+ * @v count            Block count
  * @v buffer           Data buffer
  * @ret rc             Return status code
  */
 static int scsi_read ( struct block_device *blockdev, uint64_t block,
-                      void *buffer ) {
+                      unsigned long count, userptr_t buffer ) {
        struct scsi_device *scsi = block_to_scsi ( blockdev );
        struct scsi_command command;
        struct scsi_cdb_read_16 *cdb = &command.cdb.read16;
@@ -63,9 +64,9 @@ static int scsi_read ( struct block_device *blockdev, uint64_t block,
        memset ( &command, 0, sizeof ( command ) );
        cdb->opcode = SCSI_OPCODE_READ_16;
        cdb->lba = cpu_to_be64 ( block );
-       cdb->len = cpu_to_be32 ( 1 ); /* always a single block */
+       cdb->len = cpu_to_be32 ( count );
        command.data_in = buffer;
-       command.data_in_len = blockdev->blksize;
+       command.data_in_len = ( count * blockdev->blksize );
        return scsi_command ( scsi, &command );
 }
 
@@ -74,11 +75,12 @@ static int scsi_read ( struct block_device *blockdev, uint64_t block,
  *
  * @v blockdev         Block device
  * @v block            LBA block number
+ * @v count            Block count
  * @v buffer           Data buffer
  * @ret rc             Return status code
  */
 static int scsi_write ( struct block_device *blockdev, uint64_t block,
-                       const void *buffer ) {
+                       unsigned long count, userptr_t buffer ) {
        struct scsi_device *scsi = block_to_scsi ( blockdev );
        struct scsi_command command;
        struct scsi_cdb_write_16 *cdb = &command.cdb.write16;
@@ -87,19 +89,48 @@ static int scsi_write ( struct block_device *blockdev, uint64_t block,
        memset ( &command, 0, sizeof ( command ) );
        cdb->opcode = SCSI_OPCODE_WRITE_16;
        cdb->lba = cpu_to_be64 ( block );
-       cdb->len = cpu_to_be32 ( 1 ); /* always a single block */
+       cdb->len = cpu_to_be32 ( count );
        command.data_out = buffer;
-       command.data_out_len = blockdev->blksize;
+       command.data_out_len = ( count * blockdev->blksize );
        return scsi_command ( scsi, &command );
 }
 
 /**
- * Read capacity of SCSI device
+ * Read capacity of SCSI device via READ CAPACITY (10)
  *
  * @v blockdev         Block device
  * @ret rc             Return status code
  */
-static int scsi_read_capacity ( struct block_device *blockdev ) {
+static int scsi_read_capacity_10 ( struct block_device *blockdev ) {
+       struct scsi_device *scsi = block_to_scsi ( blockdev );
+       struct scsi_command command;
+       struct scsi_cdb_read_capacity_10 *cdb = &command.cdb.readcap10;
+       struct scsi_capacity_10 capacity;
+       int rc;
+
+       /* Issue READ CAPACITY (10) */
+       memset ( &command, 0, sizeof ( command ) );
+       cdb->opcode = SCSI_OPCODE_READ_CAPACITY_10;
+       command.data_in = virt_to_user ( &capacity );
+       command.data_in_len = sizeof ( capacity );
+
+       if ( ( rc = scsi_command ( scsi, &command ) ) != 0 )
+               return rc;
+
+       /* Fill in block device fields */
+       blockdev->blksize = be32_to_cpu ( capacity.blksize );
+       blockdev->blocks = ( be32_to_cpu ( capacity.lba ) + 1 );
+
+       return 0;
+}
+
+/**
+ * Read capacity of SCSI device via READ CAPACITY (16)
+ *
+ * @v blockdev         Block device
+ * @ret rc             Return status code
+ */
+static int scsi_read_capacity_16 ( struct block_device *blockdev ) {
        struct scsi_device *scsi = block_to_scsi ( blockdev );
        struct scsi_command command;
        struct scsi_cdb_read_capacity_16 *cdb = &command.cdb.readcap16;
@@ -111,7 +142,7 @@ static int scsi_read_capacity ( struct block_device *blockdev ) {
        cdb->opcode = SCSI_OPCODE_SERVICE_ACTION_IN;
        cdb->service_action = SCSI_SERVICE_ACTION_READ_CAPACITY_16;
        cdb->len = cpu_to_be32 ( sizeof ( capacity ) );
-       command.data_in = &capacity;
+       command.data_in = virt_to_user ( &capacity );
        command.data_in_len = sizeof ( capacity );
 
        if ( ( rc = scsi_command ( scsi, &command ) ) != 0 )
@@ -123,6 +154,32 @@ static int scsi_read_capacity ( struct block_device *blockdev ) {
        return 0;
 }
 
+/**
+ * Read capacity of SCSI device
+ *
+ * @v blockdev         Block device
+ * @ret rc             Return status code
+ */
+static int scsi_read_capacity ( struct block_device *blockdev ) {
+       int rc;
+
+       /* Try READ CAPACITY (10), which is a mandatory command, first. */
+       if ( ( rc = scsi_read_capacity_10 ( blockdev ) ) != 0 )
+               return rc;
+
+       /* If capacity range was exceeded (i.e. capacity.lba was
+        * 0xffffffff, meaning that blockdev->blocks is now zero), use
+        * READ CAPACITY (16) instead.  READ CAPACITY (16) is not
+        * mandatory, so we can't just use it straight off.
+        */
+       if ( blockdev->blocks == 0 ) {
+               if ( ( rc = scsi_read_capacity_16 ( blockdev ) ) != 0 )
+                       return rc;
+       }
+
+       return 0;
+}
+
 /**
  * Initialise SCSI device
  *