Jonathan Tam / Mbed 2 deprecated BalancingRobot_NOINTERRUPTS_OOP_PS3
Committer:
jcytam
Date:
Thu Feb 23 22:59:51 2012 +0000
Revision:
0:0309bb86b703
balancing robot ps3

Who changed what in which revision?

UserRevisionLine numberNew contents of line
jcytam 0:0309bb86b703 1
jcytam 0:0309bb86b703 2 /*
jcytam 0:0309bb86b703 3 Copyright (c) 2010 Peter Barrett
jcytam 0:0309bb86b703 4
jcytam 0:0309bb86b703 5 Permission is hereby granted, free of charge, to any person obtaining a copy
jcytam 0:0309bb86b703 6 of this software and associated documentation files (the "Software"), to deal
jcytam 0:0309bb86b703 7 in the Software without restriction, including without limitation the rights
jcytam 0:0309bb86b703 8 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
jcytam 0:0309bb86b703 9 copies of the Software, and to permit persons to whom the Software is
jcytam 0:0309bb86b703 10 furnished to do so, subject to the following conditions:
jcytam 0:0309bb86b703 11
jcytam 0:0309bb86b703 12 The above copyright notice and this permission notice shall be included in
jcytam 0:0309bb86b703 13 all copies or substantial portions of the Software.
jcytam 0:0309bb86b703 14
jcytam 0:0309bb86b703 15 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
jcytam 0:0309bb86b703 16 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
jcytam 0:0309bb86b703 17 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
jcytam 0:0309bb86b703 18 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
jcytam 0:0309bb86b703 19 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
jcytam 0:0309bb86b703 20 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
jcytam 0:0309bb86b703 21 THE SOFTWARE.
jcytam 0:0309bb86b703 22 */
jcytam 0:0309bb86b703 23
jcytam 0:0309bb86b703 24 #include "stdlib.h"
jcytam 0:0309bb86b703 25 #include "stdio.h"
jcytam 0:0309bb86b703 26 #include "string.h"
jcytam 0:0309bb86b703 27
jcytam 0:0309bb86b703 28 #include "Utils.h"
jcytam 0:0309bb86b703 29 #include "USBHost.h"
jcytam 0:0309bb86b703 30
jcytam 0:0309bb86b703 31
jcytam 0:0309bb86b703 32 int MassStorage_ReadCapacity(int device, u32* blockCount, u32* blockSize);
jcytam 0:0309bb86b703 33 int MassStorage_ReadBlock(int device, u32 block, u8* dst);
jcytam 0:0309bb86b703 34 int MassStorage_WriteBlock(int device, u32 block, const u8* dst);
jcytam 0:0309bb86b703 35
jcytam 0:0309bb86b703 36
jcytam 0:0309bb86b703 37 #define ERR_BAD_CSW_SIGNATURE -200
jcytam 0:0309bb86b703 38
jcytam 0:0309bb86b703 39 #define CBW_SIGNATURE 0x43425355
jcytam 0:0309bb86b703 40 #define CSW_SIGNATURE 0x53425355
jcytam 0:0309bb86b703 41
jcytam 0:0309bb86b703 42 // Command Block
jcytam 0:0309bb86b703 43 typedef struct
jcytam 0:0309bb86b703 44 {
jcytam 0:0309bb86b703 45 u32 Signature;
jcytam 0:0309bb86b703 46 u32 Tag;
jcytam 0:0309bb86b703 47 u32 TransferLength;
jcytam 0:0309bb86b703 48 u8 Flags;
jcytam 0:0309bb86b703 49 u8 LUN;
jcytam 0:0309bb86b703 50 u8 CBLength;
jcytam 0:0309bb86b703 51 u8 CB[16]; // only 6 really
jcytam 0:0309bb86b703 52 } CBW;
jcytam 0:0309bb86b703 53
jcytam 0:0309bb86b703 54 // Status block
jcytam 0:0309bb86b703 55 typedef struct
jcytam 0:0309bb86b703 56 {
jcytam 0:0309bb86b703 57 u32 Signature;
jcytam 0:0309bb86b703 58 u32 Tag;
jcytam 0:0309bb86b703 59 u32 DataResidue;
jcytam 0:0309bb86b703 60 u8 Status;
jcytam 0:0309bb86b703 61 } CSW;
jcytam 0:0309bb86b703 62
jcytam 0:0309bb86b703 63 int SCSIRequestSense(int device);
jcytam 0:0309bb86b703 64
jcytam 0:0309bb86b703 65 int DoSCSI(int device, const u8* cmd, int cmdLen, int flags, u8* data, u32 transferLen)
jcytam 0:0309bb86b703 66 {
jcytam 0:0309bb86b703 67 CBW cbw;
jcytam 0:0309bb86b703 68 cbw.Signature = CBW_SIGNATURE;
jcytam 0:0309bb86b703 69 cbw.Tag = 0;
jcytam 0:0309bb86b703 70 cbw.TransferLength = transferLen;
jcytam 0:0309bb86b703 71 cbw.Flags = flags;
jcytam 0:0309bb86b703 72 cbw.LUN = 0;
jcytam 0:0309bb86b703 73 cbw.CBLength = cmdLen;
jcytam 0:0309bb86b703 74 memset(cbw.CB,0,sizeof(cbw.CB));
jcytam 0:0309bb86b703 75 memcpy(cbw.CB,cmd,cmdLen);
jcytam 0:0309bb86b703 76
jcytam 0:0309bb86b703 77 int r;
jcytam 0:0309bb86b703 78 r = USBBulkTransfer(device,0x01,(u8*)&cbw,31); // Send the command
jcytam 0:0309bb86b703 79 if (r < 0)
jcytam 0:0309bb86b703 80 return r;
jcytam 0:0309bb86b703 81
jcytam 0:0309bb86b703 82 if (data)
jcytam 0:0309bb86b703 83 {
jcytam 0:0309bb86b703 84 r = USBBulkTransfer(device,flags | 1,data,transferLen);
jcytam 0:0309bb86b703 85 if (r < 0)
jcytam 0:0309bb86b703 86 return r;
jcytam 0:0309bb86b703 87 }
jcytam 0:0309bb86b703 88
jcytam 0:0309bb86b703 89 CSW csw;
jcytam 0:0309bb86b703 90 csw.Signature = 0;
jcytam 0:0309bb86b703 91 r = USBBulkTransfer(device,0x81,(u8*)&csw,13);
jcytam 0:0309bb86b703 92 if (r < 0)
jcytam 0:0309bb86b703 93 return r;
jcytam 0:0309bb86b703 94
jcytam 0:0309bb86b703 95 if (csw.Signature != CSW_SIGNATURE)
jcytam 0:0309bb86b703 96 return ERR_BAD_CSW_SIGNATURE;
jcytam 0:0309bb86b703 97
jcytam 0:0309bb86b703 98 // ModeSense?
jcytam 0:0309bb86b703 99 if (csw.Status == 1 && cmd[0] != 3)
jcytam 0:0309bb86b703 100 return SCSIRequestSense(device);
jcytam 0:0309bb86b703 101
jcytam 0:0309bb86b703 102 return csw.Status;
jcytam 0:0309bb86b703 103 }
jcytam 0:0309bb86b703 104
jcytam 0:0309bb86b703 105 int SCSITestUnitReady(int device)
jcytam 0:0309bb86b703 106 {
jcytam 0:0309bb86b703 107 u8 cmd[6];
jcytam 0:0309bb86b703 108 memset(cmd,0,6);
jcytam 0:0309bb86b703 109 return DoSCSI(device,cmd,6,DEVICE_TO_HOST,0,0);
jcytam 0:0309bb86b703 110 }
jcytam 0:0309bb86b703 111
jcytam 0:0309bb86b703 112 int SCSIRequestSense(int device)
jcytam 0:0309bb86b703 113 {
jcytam 0:0309bb86b703 114 u8 cmd[6] = {0x03,0,0,0,18,0};
jcytam 0:0309bb86b703 115 u8 result[18];
jcytam 0:0309bb86b703 116 int r = DoSCSI(device,cmd,6,DEVICE_TO_HOST,result,18);
jcytam 0:0309bb86b703 117 return r;
jcytam 0:0309bb86b703 118 }
jcytam 0:0309bb86b703 119
jcytam 0:0309bb86b703 120 int SCSIInquiry(int device)
jcytam 0:0309bb86b703 121 {
jcytam 0:0309bb86b703 122 u8 cmd[6] = {0x12,0,0,0,36,0};
jcytam 0:0309bb86b703 123 u8 result[36+2];
jcytam 0:0309bb86b703 124 result[36] = '\n';
jcytam 0:0309bb86b703 125 result[37] = 0;
jcytam 0:0309bb86b703 126 int r = DoSCSI(device,cmd,6,DEVICE_TO_HOST,result,36);
jcytam 0:0309bb86b703 127 if (r == 0)
jcytam 0:0309bb86b703 128 printf((const char*)result + 8);
jcytam 0:0309bb86b703 129 return r;
jcytam 0:0309bb86b703 130 }
jcytam 0:0309bb86b703 131
jcytam 0:0309bb86b703 132 int SCSIReadCapacity(int device, u32* blockCount, u32* blockSize)
jcytam 0:0309bb86b703 133 {
jcytam 0:0309bb86b703 134 u8 cmd[10] = {0x25,0,0,0,8,0,0,0,0,0};
jcytam 0:0309bb86b703 135 u8 result[8];
jcytam 0:0309bb86b703 136 *blockSize = 0;
jcytam 0:0309bb86b703 137 *blockCount = 0;
jcytam 0:0309bb86b703 138 int r = DoSCSI(device,cmd,10,DEVICE_TO_HOST,result,8);
jcytam 0:0309bb86b703 139 if (r == 0)
jcytam 0:0309bb86b703 140 {
jcytam 0:0309bb86b703 141 *blockCount = BE32(result);
jcytam 0:0309bb86b703 142 *blockSize = BE32(result+4);
jcytam 0:0309bb86b703 143 }
jcytam 0:0309bb86b703 144 return r;
jcytam 0:0309bb86b703 145 }
jcytam 0:0309bb86b703 146
jcytam 0:0309bb86b703 147 int SCSITransfer(int device, u32 blockAddr, u32 blockCount, u8* dst, u32 blockSize, int direction)
jcytam 0:0309bb86b703 148 {
jcytam 0:0309bb86b703 149 // USB hardware will only do 4k per transfer
jcytam 0:0309bb86b703 150 while (blockCount*blockSize > 4096)
jcytam 0:0309bb86b703 151 {
jcytam 0:0309bb86b703 152 int count = 4096/blockSize;
jcytam 0:0309bb86b703 153 int r = SCSITransfer(device,blockAddr,count,dst,blockSize,direction);
jcytam 0:0309bb86b703 154 dst += count*blockSize;
jcytam 0:0309bb86b703 155 blockAddr += count;
jcytam 0:0309bb86b703 156 blockCount -= count;
jcytam 0:0309bb86b703 157 }
jcytam 0:0309bb86b703 158
jcytam 0:0309bb86b703 159 u8 cmd[10];
jcytam 0:0309bb86b703 160 memset(cmd,0,10);
jcytam 0:0309bb86b703 161 cmd[0] = (direction == DEVICE_TO_HOST) ? 0x28 : 0x2A;
jcytam 0:0309bb86b703 162 BE32(blockAddr,cmd+2);
jcytam 0:0309bb86b703 163 BE16(blockCount,cmd+7);
jcytam 0:0309bb86b703 164 return DoSCSI(device,cmd,10,direction,dst,blockSize*blockCount);
jcytam 0:0309bb86b703 165 }
jcytam 0:0309bb86b703 166
jcytam 0:0309bb86b703 167 int MassStorage_ReadCapacity(int device, u32* blockCount, u32* blockSize)
jcytam 0:0309bb86b703 168 {
jcytam 0:0309bb86b703 169 return SCSIReadCapacity(device,blockCount,blockSize);
jcytam 0:0309bb86b703 170 }
jcytam 0:0309bb86b703 171
jcytam 0:0309bb86b703 172 int MassStorage_Read(int device, u32 blockAddr, u32 blockCount, u8* dst, u32 blockSize = 512)
jcytam 0:0309bb86b703 173 {
jcytam 0:0309bb86b703 174 return SCSITransfer(device,blockAddr,blockCount,dst,blockSize,DEVICE_TO_HOST);
jcytam 0:0309bb86b703 175 }
jcytam 0:0309bb86b703 176
jcytam 0:0309bb86b703 177 int MassStorage_Write(int device, u32 blockAddr, u32 blockCount, u8* dst, u32 blockSize = 512)
jcytam 0:0309bb86b703 178 {
jcytam 0:0309bb86b703 179 return SCSITransfer(device,blockAddr,blockCount,dst,blockSize,HOST_TO_DEVICE);
jcytam 0:0309bb86b703 180 }