main.cpp

00001 /*
00002 Copyright (c) 2010 Peter Barrett
00003 
00004 Permission is hereby granted, free of charge, to any person obtaining a copy
00005 of this software and associated documentation files (the "Software"), to deal
00006 in the Software without restriction, including without limitation the rights
00007 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
00008 copies of the Software, and to permit persons to whom the Software is
00009 furnished to do so, subject to the following conditions:
00010 
00011 The above copyright notice and this permission notice shall be included in
00012 all copies or substantial portions of the Software.
00013 
00014 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
00015 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
00016 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
00017 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
00018 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
00019 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
00020 THE SOFTWARE.
00021 */
00022 
00023 #include "mbed.h"
00024 #include "USBHost.h"
00025 #include "Utils.h"
00026 #include "FATFileSystem.h"
00027 
00028 int MassStorage_ReadCapacity(int device, u32* blockCount, u32* blockSize);
00029 int MassStorage_Read(int device, u32 blockAddr, u32 blockCount, u8* dst, u32 blockSize);
00030 int MassStorage_Write(int device, u32 blockAddr, u32 blockCount, u8* dst, u32 blockSize);
00031 
00032 class USBFileSystem : public FATFileSystem
00033 {
00034     int _device;
00035     u32 _blockSize;
00036     u32 _blockCount;
00037     
00038 public:
00039     USBFileSystem() : FATFileSystem("usb"),_device(0),_blockSize(0),_blockCount(0)
00040     {
00041     }
00042     
00043     void SetDevice(int device)
00044     {
00045         _device = device;
00046     }
00047     
00048     virtual int disk_initialize()
00049     {
00050         return MassStorage_ReadCapacity(_device,&_blockCount,&_blockSize);
00051     }
00052     
00053     virtual int disk_write(const char *buffer, int block_number)
00054     {
00055         return MassStorage_Write(_device,block_number,1,(u8*)buffer,_blockSize);
00056     }
00057     
00058     virtual int disk_read(char *buffer, int block_number)
00059     {
00060         return MassStorage_Read(_device,block_number,1,(u8*)buffer,_blockSize);
00061     }
00062         
00063     virtual int disk_sectors()
00064     {
00065         return _blockCount;
00066     }
00067 };
00068 
00069 Serial pc(USBTX, USBRX);
00070 
00071 void DumpFS(int depth, int count)
00072 {
00073     DIR *d = opendir("/usb");
00074     if (!d)
00075     {
00076         printf("USB file system borked\n");
00077         return;
00078     }
00079 
00080     printf("\nDumping root dir\n");
00081     struct dirent *p;
00082     while ((p = readdir(d)))
00083     {
00084         int len = sizeof( dirent);
00085         printf("%s %d\n", p->d_name, len);
00086     }
00087     closedir(d);
00088 }
00089 
00090 int OnDiskInsert(int device)
00091 {
00092     USBFileSystem fs;
00093     fs.SetDevice(device);
00094     DumpFS(0,0);
00095     return 0;
00096 }
00097 
00098 /*
00099     Simple test shell to exercise mouse,keyboard,mass storage and hubs.
00100     Add 2 15k pulldown resistors between D+/D- and ground, attach a usb socket and have at it.
00101 */
00102 
00103 int main()
00104 {
00105     pc.baud(460800);
00106     printf("USBShell\nNow get a bunch of usb things and plug them in\n");
00107     USBInit();
00108     for (;;)
00109         USBLoop();
00110 }