KernelLand/Modules/Storage/LVM/volumes.c

   1 /*
   2  * Acess2 Logical Volume Manager
   3  * - By John Hodge (thePowersGang)
   4  *
   5  * volumes.c
   6  * - Volume management
   7  */
   8 #define DEBUG   1
   9 #include "lvm_int.h"
  10
  11 // === PROTOTYPES ===
  12  int    LVM_int_VFSReadEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, void *Dest);
  13  int    LVM_int_VFSWriteEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, const void *Source);
  14
  15 // === CODE ===
  16 // --------------------------------------------------------------------
  17 // Managment / Initialisation
  18 // --------------------------------------------------------------------
  19 int LVM_AddVolumeVFS(const char *Name, int FD)
  20 {
  21         // Assuming 512-byte blocks, not a good idea
  22 //      return LVM_AddVolume(Name, (void*)(Uint)FD, 512, LVM_int_VFSReadEmul, LVM_int_VFSWriteEmul);
  23         return 0;
  24 }
  25
  26 void *LVM_AddVolume(const tLVM_VolType *Type, const char *Name, void *Ptr, size_t BlockSize, size_t BlockCount)
  27 {
  28         tLVM_Vol        dummy_vol;
  29         tLVM_Format     *fmt;
  30
  31         if( BlockCount == 0 || BlockSize == 0 ) {
  32                 Log_Error("LVM", "BlockSize(0x%x)/BlockCount(0x%x) invalid in LVM_AddVolume",
  33                         BlockSize, BlockCount);
  34                 return NULL;
  35         }
  36
  37         dummy_vol.Type = Type;
  38         dummy_vol.Ptr = Ptr;
  39         dummy_vol.BlockCount = BlockCount;
  40         dummy_vol.BlockSize = BlockSize;
  41
  42         // Read the first block of the volume
  43         void *first_block = malloc(BlockSize);
  44         if( !first_block ) {
  45                 Log_Error("LVM", "LVM_AddVolume - malloc error on %i bytes", BlockSize);
  46                 return NULL;
  47         }
  48         if( Type->Read(Ptr, 0, 1, first_block) != 1 ) {
  49                 Log_Error("LVM", "LVM_AddVolume - Failed to read first sector");
  50                 free(first_block);
  51                 return NULL;
  52         }
  53
  54         // Determine Format
  55         // TODO: Determine format
  56         fmt = &gLVM_MBRType;
  57
  58         // Type->CountSubvolumes
  59         dummy_vol.nSubVolumes = fmt->CountSubvolumes(&dummy_vol, first_block);
  60
  61         // Create real volume descriptor
  62         // TODO: If this needs to be rescanned later, having the subvolume list separate might be an idea
  63         size_t  allocsize = sizeof(tLVM_Vol) + strlen(Name) + 1 + sizeof(tLVM_SubVolume*) * dummy_vol.nSubVolumes;
  64         tLVM_Vol        *real_vol = malloc( allocsize );
  65         if( !real_vol ) {
  66                 Log_Error("LVM", "LVM_AddVolume - malloc error on %i bytes", allocsize);
  67                 free(first_block);
  68                 return NULL;
  69         }
  70         real_vol->Next = NULL;
  71         real_vol->Type = Type;
  72         real_vol->Ptr = Ptr;
  73         real_vol->BlockSize = BlockSize;
  74         real_vol->BlockCount = BlockCount;
  75         real_vol->nSubVolumes = dummy_vol.nSubVolumes;
  76         real_vol->SubVolumes = (void*)( real_vol->Name + strlen(Name) + 1 );
  77         real_vol->BlockSize = BlockSize;
  78         strcpy(real_vol->Name, Name);
  79         memset(real_vol->SubVolumes, 0, sizeof(tLVM_SubVolume*) * real_vol->nSubVolumes);
  80         // - VFS Nodes
  81         memset(&real_vol->DirNode, 0, sizeof(tVFS_Node));
  82         real_vol->DirNode.Type = &gLVM_VolNodeType;
  83         real_vol->DirNode.ImplPtr = real_vol;
  84         real_vol->DirNode.Flags = VFS_FFLAG_DIRECTORY;
  85         real_vol->DirNode.Size = -1;
  86         memset(&real_vol->VolNode, 0, sizeof(tVFS_Node));
  87         real_vol->VolNode.Type = &gLVM_VolNodeType;
  88         real_vol->VolNode.ImplPtr = real_vol;
  89         real_vol->VolNode.Size = BlockCount * BlockSize;
  90
  91         // Type->PopulateSubvolumes
  92         fmt->PopulateSubvolumes(real_vol, first_block);
  93         free(first_block);
  94
  95         // Add to volume list
  96         gpLVM_LastVolume->Next = real_vol;
  97         gpLVM_LastVolume = real_vol;
  98
  99         return real_vol;
 100 }
 101
 102 void LVM_int_SetSubvolume_Anon(tLVM_Vol *Volume, int Index, Uint64 FirstBlock, Uint64 BlockCount)
 103 {
 104         tLVM_SubVolume  *sv;
 105          int    namelen;
 106
 107         if( Index < 0 || Index >= Volume->nSubVolumes ) {
 108                 Log_Warning("LVM", "SV ID is out of range (0 < %i < %i)",
 109                         Index, Volume->nSubVolumes);
 110                 return ;
 111         }
 112
 113         if( Volume->SubVolumes[Index] ) {
 114                 Log_Warning("LVM", "Attempt to set SV %i of %p twice", Index, Volume);
 115                 return ;
 116         }
 117
 118         namelen = snprintf(NULL, 0, "%i", Index);
 119
 120         sv = malloc( sizeof(tLVM_SubVolume) + namelen + 1 );
 121         if(!sv) {
 122                 // Oh, f*ck
 123                 return ;
 124         }
 125         Volume->SubVolumes[Index] = sv;
 126
 127         sv->Vol = Volume;
 128         sprintf(sv->Name, "%i", Index);
 129         sv->FirstBlock = FirstBlock;
 130         sv->BlockCount = BlockCount;
 131         memset(&sv->Node, 0, sizeof(tVFS_Node));
 132
 133         sv->Node.ImplPtr = sv;
 134         sv->Node.Type = &gLVM_SubVolNodeType;
 135         sv->Node.Size = BlockCount * Volume->BlockSize;
 136
 137         Log_Log("LVM", "Partition %s/%s - 0x%llx+0x%llx blocks",
 138                 Volume->Name, sv->Name, FirstBlock, BlockCount);
 139 }
 140
 141 // --------------------------------------------------------------------
 142 // IO
 143 // --------------------------------------------------------------------
 144 size_t LVM_int_ReadVolume(tLVM_Vol *Volume, Uint64 BlockNum, size_t BlockCount, void *Dest)
 145 {
 146         return Volume->Type->Read(Volume->Ptr, BlockNum, BlockCount, Dest);
 147 }
 148
 149 size_t LVM_int_WriteVolume(tLVM_Vol *Volume, Uint64 BlockNum, size_t BlockCount, const void *Src)
 150 {
 151         return Volume->Type->Write(Volume->Ptr, BlockNum, BlockCount, Src);
 152 }
 153
 154 int LVM_int_VFSReadEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, void *Dest)
 155 {
 156         size_t  blocksize;
 157         size_t  rv;
 158
 159         blocksize = 512;        // TODO: Don't assume
 160
 161         rv = VFS_ReadAt( (int)(Uint)Arg, BlockStart * blocksize, BlockCount * blocksize, Dest );
 162         rv /= blocksize;
 163         return rv;
 164 }
 165
 166 int LVM_int_VFSWriteEmul(void *Arg, Uint64 BlockStart, size_t BlockCount, const void *Source)
 167 {
 168         size_t  blocksize;
 169         size_t  rv;
 170
 171         blocksize = 512;        // TODO: Don't assume
 172
 173         rv = VFS_WriteAt( (int)(Uint)Arg, BlockStart * blocksize, BlockCount * blocksize, Source );
 174         rv /= blocksize;
 175         return rv;
 176 }
 177